76 lines
1.8 KiB
C++
76 lines
1.8 KiB
C++
#include "svprocessor.h"
|
|
|
|
SvProcessor::SvProcessor(QObject *parent):
|
|
QObject(parent)
|
|
{
|
|
|
|
}
|
|
|
|
SvProcessor::SvProcessor(SvImage* left, SvImage* right, SvImage* result, int numberOfWorkers, int version)
|
|
{
|
|
unsigned int i;
|
|
|
|
if (version == 1) {
|
|
m_kernel = new SvKernelV1();
|
|
} else if (version == 2) {
|
|
m_kernel = new SvKernelV2();
|
|
}
|
|
|
|
m_kernel->setLeftImage(left);
|
|
m_kernel->setRightImage(right);
|
|
m_kernel->setResultImage(result);
|
|
|
|
m_numberOfWorkers = numberOfWorkers;
|
|
m_workers = new SvWorker[m_numberOfWorkers];
|
|
m_threads = new QThread[m_numberOfWorkers];
|
|
|
|
for (i = 0; i < m_numberOfWorkers; i++) {
|
|
m_workers[i].setId(i);
|
|
m_workers[i].setKernel(m_kernel);
|
|
m_workers[i].moveToThread(&m_threads[i]);
|
|
|
|
connect(&m_threads[i], &QThread::started, &m_workers[i], &SvWorker::start);
|
|
connect(&m_workers[i], &SvWorker::finished, this, &SvProcessor::workerFinished);
|
|
connect(&m_workers[i], &SvWorker::finished, &m_threads[i], &QThread::quit);
|
|
connect(this, &SvProcessor::started, &m_workers[i], &SvWorker::start);
|
|
connect(&m_threads[i], &QThread::finished, &m_threads[i], &QThread::deleteLater);
|
|
}
|
|
|
|
for (i = 0; i < result->getHeight(); i++) {
|
|
m_workers[i % m_numberOfWorkers].addTask(i);
|
|
}
|
|
}
|
|
SvProcessor::~SvProcessor()
|
|
{
|
|
delete m_kernel;
|
|
delete[] m_workers;
|
|
}
|
|
|
|
void SvProcessor::workerFinished(int workerId)
|
|
{
|
|
Q_UNUSED(workerId)
|
|
|
|
m_workersFinished++;
|
|
if (m_workersFinished == m_numberOfWorkers) {
|
|
qDebug() << "finished";
|
|
emit finished(m_startTime - 0);
|
|
}
|
|
}
|
|
|
|
void SvProcessor::start()
|
|
{
|
|
unsigned int i;
|
|
|
|
m_startTime = 0;
|
|
m_workersFinished = 0;
|
|
|
|
for (i = 0; i < m_numberOfWorkers; i++) {
|
|
m_threads[i].start();
|
|
}
|
|
}
|
|
|
|
void SvProcessor::stop()
|
|
{
|
|
emit stopped(m_startTime - 0);
|
|
}
|