You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

104 lines
2.7 KiB

#include <iostream> // cout
#include <vector>
#include <algorithm> // min_element
#include <cstdlib> // srand, rand, atoi
#include <ctime> // time, timespec, clock_gettime
#include <cmath> // pow, sqrt, log
#include <cfloat> // FLX_MAX
#include <QtCore/QtCore>
#include <QtCore/QThreadPool>
#include <QtCore/QtConcurrentRun>
#include <QtCore/QFutureSynchronizer>
const int RANDOM_MAX = 1000;
const int RANDOM_MIN = 1;
timespec diff(timespec start, timespec end)
{
timespec temp;
if ((end.tv_nsec-start.tv_nsec)<0) {
temp.tv_sec = end.tv_sec-start.tv_sec-1;
temp.tv_nsec = 1000000000+end.tv_nsec-start.tv_nsec;
} else {
temp.tv_sec = end.tv_sec-start.tv_sec;
temp.tv_nsec = end.tv_nsec-start.tv_nsec;
}
return temp;
}
void findLocalMinimum(const std::vector<float>::const_iterator begin,
const std::vector<float>::const_iterator end,
float *result)
{
float min(FLT_MAX);
for(std::vector<float>::const_iterator it = begin; it != end; ++it)
if (*it < min)
min = *it;
*result = min;
}
float QtReduce(std::vector<float>& data,
const int numberOfThreads,
const int chunkSize)
{
std::vector<float> separate_results(numberOfThreads, FLT_MAX);
QFutureSynchronizer<void> synchronizer;
for(int i = 0; i < numberOfThreads; i++)
synchronizer.addFuture(QtConcurrent::run(findLocalMinimum,
data.begin()+i*chunkSize,
data.begin()+(i+1)*chunkSize,
separate_results.data()+i));
synchronizer.waitForFinished();
// serial reduce
float min(FLT_MAX);
for (int i = 0; i < numberOfThreads; i++)
if (separate_results[i] < min)
min = separate_results[i];
return min;
}
int main(int argc, char* argv[])
{
if (argc != 3) {
std::cout << "Usage: " << argv[0] << " <NUMBER_OF_THREADS> <DATA_SIZE>" << std::endl;
exit(1);
}
const int NUMBER_OF_THREADS = atoi(argv[1]);
const int DATA_SIZE = atoi(argv[2]);
const int CHUNK_SIZE = DATA_SIZE / NUMBER_OF_THREADS;
srand(time(NULL));
QCoreApplication app(argc, argv);
QThreadPool::globalInstance()->setMaxThreadCount(NUMBER_OF_THREADS);
std::vector<float> data(DATA_SIZE);
for (int i = 0; i < DATA_SIZE; i++)
data[i] = rand() % RANDOM_MAX + RANDOM_MIN;
timespec startTime;
clock_gettime(CLOCK_MONOTONIC, &startTime);
float min = QtReduce(data, NUMBER_OF_THREADS, CHUNK_SIZE);
timespec endTime;
clock_gettime(CLOCK_MONOTONIC, &endTime);
timespec timeDiff = diff(startTime, endTime);
std::cout << timeDiff.tv_sec << "." << timeDiff.tv_nsec << std::endl;
return 0;
}