ParallelSearch/main.cpp

84 lines
2.4 KiB
C++

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <random>
#include "ParallelSearch.hpp"
#include "TestRunner.hpp"
#include "Parameters.hpp"
// Type of our hay stack: a list of strings
typedef std::vector<std::string> WordList;
typedef ParallelSearch<WordList::const_iterator> ParallelWordSearch;
void create_testdata(WordList& words)
{
const char start = 'A';
const char end = 'Z';
char str[4] = { 'A', 'A', 'A', 'A' };
for (str[0] = start; str[0] <= end; str[0]++)
for (str[1] = start; str[1] <= end; str[1]++)
for (str[2] = start; str[2] <= end; str[2]++)
for (str[3] = start; str[3] <= end; str[3]++)
words.push_back(std::string(str, 4));
}
void shuffle_testdata(WordList& words)
{
auto rng = std::default_random_engine{};
std::shuffle(words.begin(), words.end(), rng);
}
void run_wordsearch(ParallelWordSearch& search)
{
search.run();
}
int main(int argc, char** argv)
{
Parameters parameters;
double time_span;
WordList test_data;
std::cout << "ParallelSearch Demo" << std::endl << std::endl;
if (false == parameters.parseCmdLine(argc, argv))
{
parameters.show_usage(argv[0]);
exit(1);
}
std::cout << "Workercount: " << parameters.WorkerCount() << std::endl;
std::cout << "Testdata length: " << parameters.TestDataLength() << std::endl;
std::cout << "Search pattern: " << parameters.SearchPattern() << std::endl;
std::cout << std::endl;
time_span = TestRunner(create_testdata, test_data);
std::cout << "created test_data (" << test_data.size() << " words) " << time_span << " seconds" << std::endl;
time_span = TestRunner(shuffle_testdata, test_data);
std::cout << "shuffled test_data " << time_span << " seconds" << std::endl;
ParallelWordSearch wordSearch(1, test_data.begin(), test_data.end(), parameters.SearchPattern());
time_span = TestRunner(run_wordsearch, wordSearch);
std::cout << "linear search found " << wordSearch.get_result().size() << " matches: " << time_span << " seconds" << std::endl;
ParallelWordSearch parallelWordSearch(parameters.WorkerCount(), test_data.begin(), test_data.end(), parameters.SearchPattern());
time_span = TestRunner(run_wordsearch, parallelWordSearch);
std::cout << "parallel search found " << parallelWordSearch.get_result().size() << " matches: " << time_span << " seconds" << std::endl;
// TODO add validation of results
// TODO brush up output
std::cout << std::endl;
return 0;
}