109 lines
3.1 KiB
C++
109 lines
3.1 KiB
C++
#include <iostream>
|
|
#include <string>
|
|
#include <vector>
|
|
#include <algorithm>
|
|
#include <random>
|
|
|
|
#include "ParallelSearch.hpp"
|
|
#include "TestRunner.hpp"
|
|
#include "Parameters.hpp"
|
|
|
|
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(ParallelSearch& search)
|
|
{
|
|
search.run();
|
|
}
|
|
|
|
bool check_result(const WordList& haystack, const std::string& pattern, const ParallelSearch::ResultList& result)
|
|
{
|
|
bool result_valid = true;
|
|
|
|
// check number of matches
|
|
const unsigned TESTPATTERN_LENGTH = 4;
|
|
unsigned expected_count = static_cast<unsigned>(pow(26.0, double(TESTPATTERN_LENGTH - pattern.size())));
|
|
|
|
if (result.size() != expected_count)
|
|
{
|
|
std::cout << "wrong number of matches: expected " << expected_count << " but got " << result.size() << std::endl;
|
|
result_valid = false;
|
|
}
|
|
|
|
// verify matches contain the search-pattern
|
|
unsigned valid_count = 0;
|
|
for (WordList::const_iterator itr : result)
|
|
{
|
|
if (itr->substr(0, pattern.size()) == pattern)
|
|
{
|
|
valid_count++;
|
|
}
|
|
else
|
|
{
|
|
std::cout << "found unexpected entry in result: " << *itr << std::endl;
|
|
result_valid = false;
|
|
}
|
|
}
|
|
|
|
std::cout << "result is " << (result_valid ? "VALID" : "INVALID") << std::endl;
|
|
}
|
|
|
|
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
|
|
<< "Search pattern: " << parameters.SearchPattern() << std::endl
|
|
<< std::endl;
|
|
|
|
time_span = TestRunner(create_testdata, test_data);
|
|
std::cout << "created test_data (" << test_data.size() << " words) in " << time_span << " seconds" << std::endl;
|
|
|
|
time_span = TestRunner(shuffle_testdata, test_data);
|
|
std::cout << "shuffled test_data in " << time_span << " seconds" << std::endl;
|
|
|
|
ParallelSearch linearSearch(1, test_data.begin(), test_data.end(), parameters.SearchPattern());
|
|
time_span = TestRunner(run_wordsearch, linearSearch);
|
|
std::cout << "linear search found " << linearSearch.get_result().size() << " matches in " << time_span << " seconds" << std::endl;
|
|
|
|
ParallelSearch parallelSearch(parameters.WorkerCount(), test_data.begin(), test_data.end(), parameters.SearchPattern());
|
|
time_span = TestRunner(run_wordsearch, parallelSearch);
|
|
std::cout << "parallel search found " << parallelSearch.get_result().size() << " matches in " << time_span << " seconds" << std::endl;
|
|
|
|
std::cout << std::endl;
|
|
std::cout << "checking result of parallel search ..." << std::endl;
|
|
check_result(test_data, parameters.SearchPattern(), parallelSearch.get_result());
|
|
|
|
return 0;
|
|
}
|