123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332 |
- #ifndef QTCONCURRENT_ITERATEKERNEL_H
- #define QTCONCURRENT_ITERATEKERNEL_H
- #include <QtConcurrent/qtconcurrent_global.h>
- #ifndef QT_NO_CONCURRENT
- #include <QtCore/qatomic.h>
- #include <QtConcurrent/qtconcurrentmedian.h>
- #include <QtConcurrent/qtconcurrentthreadengine.h>
- #include <iterator>
- QT_BEGIN_NAMESPACE
- #ifndef Q_QDOC
- namespace QtConcurrent {
- class Q_CONCURRENT_EXPORT BlockSizeManager
- {
- public:
- BlockSizeManager(int iterationCount);
- void timeBeforeUser();
- void timeAfterUser();
- int blockSize();
- private:
- inline bool blockSizeMaxed()
- {
- return (m_blockSize >= maxBlockSize);
- }
- const int maxBlockSize;
- qint64 beforeUser;
- qint64 afterUser;
- Median<double> controlPartElapsed;
- Median<double> userPartElapsed;
- int m_blockSize;
- Q_DISABLE_COPY(BlockSizeManager)
- };
- class Q_CONCURRENT_EXPORT BlockSizeManagerV2
- {
- public:
- explicit BlockSizeManagerV2(int iterationCount);
- void timeBeforeUser();
- void timeAfterUser();
- int blockSize();
- private:
- inline bool blockSizeMaxed()
- {
- return (m_blockSize >= maxBlockSize);
- }
- const int maxBlockSize;
- qint64 beforeUser;
- qint64 afterUser;
- MedianDouble controlPartElapsed;
- MedianDouble userPartElapsed;
- int m_blockSize;
- Q_DISABLE_COPY(BlockSizeManagerV2)
- };
- template <typename T>
- class ResultReporter
- {
- public:
- ResultReporter(ThreadEngine<T> *_threadEngine)
- :threadEngine(_threadEngine)
- {
- }
- void reserveSpace(int resultCount)
- {
- currentResultCount = resultCount;
- vector.resize(qMax(resultCount, vector.count()));
- }
- void reportResults(int begin)
- {
- const int useVectorThreshold = 4;
- if (currentResultCount > useVectorThreshold) {
- vector.resize(currentResultCount);
- threadEngine->reportResults(vector, begin);
- } else {
- for (int i = 0; i < currentResultCount; ++i)
- threadEngine->reportResult(&vector.at(i), begin + i);
- }
- }
- inline T * getPointer()
- {
- return vector.data();
- }
- int currentResultCount;
- ThreadEngine<T> *threadEngine;
- QVector<T> vector;
- };
- template <>
- class ResultReporter<void>
- {
- public:
- inline ResultReporter(ThreadEngine<void> *) { }
- inline void reserveSpace(int) { }
- inline void reportResults(int) { }
- inline void * getPointer() { return Q_NULLPTR; }
- };
- inline bool selectIteration(std::bidirectional_iterator_tag)
- {
- return false;
- }
- inline bool selectIteration(std::forward_iterator_tag)
- {
- return false;
- }
- inline bool selectIteration(std::random_access_iterator_tag)
- {
- return true;
- }
- template <typename Iterator, typename T>
- class IterateKernel : public ThreadEngine<T>
- {
- public:
- typedef T ResultType;
- IterateKernel(Iterator _begin, Iterator _end)
- : begin(_begin), end(_end), current(_begin), currentIndex(0),
- forIteration(selectIteration(typename std::iterator_traits<Iterator>::iterator_category())), progressReportingEnabled(true)
- {
- iterationCount = forIteration ? std::distance(_begin, _end) : 0;
- }
- virtual ~IterateKernel() { }
- virtual bool runIteration(Iterator it, int index , T *result)
- { Q_UNUSED(it); Q_UNUSED(index); Q_UNUSED(result); return false; }
- virtual bool runIterations(Iterator _begin, int beginIndex, int endIndex, T *results)
- { Q_UNUSED(_begin); Q_UNUSED(beginIndex); Q_UNUSED(endIndex); Q_UNUSED(results); return false; }
- void start()
- {
- progressReportingEnabled = this->isProgressReportingEnabled();
- if (progressReportingEnabled && iterationCount > 0)
- this->setProgressRange(0, iterationCount);
- }
- bool shouldStartThread()
- {
- if (forIteration)
- return (currentIndex.load() < iterationCount) && !this->shouldThrottleThread();
- else
- return (iteratorThreads.load() == 0);
- }
- ThreadFunctionResult threadFunction()
- {
- if (forIteration)
- return this->forThreadFunction();
- else
- return this->whileThreadFunction();
- }
- ThreadFunctionResult forThreadFunction()
- {
- BlockSizeManagerV2 blockSizeManager(iterationCount);
- ResultReporter<T> resultReporter(this);
- for(;;) {
- if (this->isCanceled())
- break;
- const int currentBlockSize = blockSizeManager.blockSize();
- if (currentIndex.load() >= iterationCount)
- break;
-
- const int beginIndex = currentIndex.fetchAndAddRelease(currentBlockSize);
- const int endIndex = qMin(beginIndex + currentBlockSize, iterationCount);
- if (beginIndex >= endIndex) {
-
- break;
- }
- this->waitForResume();
- if (shouldStartThread())
- this->startThread();
- const int finalBlockSize = endIndex - beginIndex;
- resultReporter.reserveSpace(finalBlockSize);
-
- blockSizeManager.timeBeforeUser();
- const bool resultsAvailable = this->runIterations(begin, beginIndex, endIndex, resultReporter.getPointer());
- blockSizeManager.timeAfterUser();
- if (resultsAvailable)
- resultReporter.reportResults(beginIndex);
-
- if (progressReportingEnabled) {
- completed.fetchAndAddAcquire(finalBlockSize);
- this->setProgressValue(this->completed.load());
- }
- if (this->shouldThrottleThread())
- return ThrottleThread;
- }
- return ThreadFinished;
- }
- ThreadFunctionResult whileThreadFunction()
- {
- if (iteratorThreads.testAndSetAcquire(0, 1) == false)
- return ThreadFinished;
- ResultReporter<T> resultReporter(this);
- resultReporter.reserveSpace(1);
- while (current != end) {
-
-
-
- Iterator prev = current;
- ++current;
- int index = currentIndex.fetchAndAddRelaxed(1);
- iteratorThreads.testAndSetRelease(1, 0);
- this->waitForResume();
- if (shouldStartThread())
- this->startThread();
- const bool resultAavailable = this->runIteration(prev, index, resultReporter.getPointer());
- if (resultAavailable)
- resultReporter.reportResults(index);
- if (this->shouldThrottleThread())
- return ThrottleThread;
- if (iteratorThreads.testAndSetAcquire(0, 1) == false)
- return ThreadFinished;
- }
- return ThreadFinished;
- }
- public:
- const Iterator begin;
- const Iterator end;
- Iterator current;
- QAtomicInt currentIndex;
- bool forIteration;
- QAtomicInt iteratorThreads;
- int iterationCount;
- bool progressReportingEnabled;
- QAtomicInt completed;
- };
- }
- #endif
- QT_END_NAMESPACE
- #endif
- #endif
|