80 lines
2.2 KiB
C++
80 lines
2.2 KiB
C++
#include <iostream>
|
|
#include <string>
|
|
#include <vector>
|
|
#include <algorithm>
|
|
#include <random>
|
|
#include <thread>
|
|
|
|
#include "SearchJob.hpp"
|
|
#include "ParallelSearch.hpp"
|
|
#include "TestRunner.hpp"
|
|
|
|
|
|
// Type of our hay stack: a list of strings
|
|
typedef std::vector<std::string> WordList;
|
|
|
|
typedef SearchJob<WordList::const_iterator> WordSearch;
|
|
|
|
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(WordSearch& search)
|
|
{
|
|
search.run();
|
|
}
|
|
|
|
void run_parallel_wordsearch(ParallelWordSearch& search)
|
|
{
|
|
search.run();
|
|
}
|
|
|
|
int main()
|
|
{
|
|
double time_span;
|
|
std::string pattern = "ABC";
|
|
unsigned workerCount = std::thread::hardware_concurrency();
|
|
WordList test_data;
|
|
|
|
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;
|
|
|
|
WordSearch wordSearch(test_data.begin(), test_data.end(), pattern);
|
|
|
|
time_span = TestRunner(run_wordsearch, wordSearch);
|
|
std::cout << "linear search found " << wordSearch.get_result().size() << " matches: " << time_span << " seconds" << std::endl;
|
|
|
|
ParallelWordSearch parallelWordSearch(workerCount, test_data.begin(), test_data.end(), pattern);
|
|
time_span = TestRunner(run_parallel_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
|
|
// TODO command line args for workerCount, test-pattern generation, search pattern ?
|
|
|
|
std::cout << std::endl;
|
|
|
|
return 0;
|
|
}
|