ConcurrentControlNetReader.cpp

Go to the documentation of this file.

#include "ConcurrentControlNetReader.h"

#include <algorithm>
#include <iostream>

#include <QDebug>
#include <QFuture>
#include <QFutureWatcher>
#include <QString>
#include <QtConcurrentRun>
#include <QtConcurrentMap>
#include <QtCore>
#include <QTimer>

#include "Control.h"
#include "ControlNet.h"
#include "FileName.h"
#include "IString.h"
#include "Progress.h"
#include "IException.h"

using std::cerr;
using std::cout;
using std::swap;

namespace Isis {

  ConcurrentControlNetReader::ConcurrentControlNetReader() {
    nullify();
    
    m_mappedRunning = false;
    
    m_progressBar = new ProgressBar("Reading Control Nets");
    m_watcher = new QFutureWatcher<Control *>;

    initProgress();

    connect(m_watcher, SIGNAL(finished()), this, SLOT(mappedFinished()));
  }


  ConcurrentControlNetReader::~ConcurrentControlNetReader() {

    if (m_watcher)
    {
      m_watcher->cancel();
      m_watcher->waitForFinished();
    }

    delete m_watcher;
    m_watcher = NULL;

    delete m_progressBar;
  }


  ProgressBar *ConcurrentControlNetReader::progressBar() {
    return m_progressBar;
  }


  void ConcurrentControlNetReader::read(QString filename) {
        
    m_backlog.append(filename);
    
    if (!m_progressBar.isNull()) {
      m_progressBar->setRange(0, m_progressBar->maximum() + 1);
    }
    
    start();
  }


  void ConcurrentControlNetReader::read(QStringList filenames) {

    m_backlog.append(filenames);
    
    if (!m_progressBar.isNull()) {
      m_progressBar->setRange(0, m_progressBar->maximum() + filenames.size());
    }
    
    start();
  }


  void ConcurrentControlNetReader::nullify() {
    m_watcher = NULL;
  }


  void ConcurrentControlNetReader::initProgress() {
    if (m_progressBar) {
      m_progressBar->setVisible(false);
      m_progressBar->setRange(0, 100);
      m_progressBar->setValue(0);
    }
  }


  void ConcurrentControlNetReader::start() {
        
    if (!m_backlog.isEmpty() && !m_mappedRunning) {

      QList< QPair<FileName, Progress *> > functorInput;
      foreach (QString backlogFileName, m_backlog) {
        Progress *progress = new Progress;
        progress->DisableAutomaticDisplay();
        m_progress.append(progress);

        functorInput.append(qMakePair(FileName(backlogFileName), progress));
      }
      
      QFuture<Control *> networks = QtConcurrent::mapped(functorInput,
                                    FileNameToControlFunctor(QThread::currentThread()));
            
      try {
        networks.result();
      }
      catch (IException &e) {
        throw e;
      }
      
      if (!m_progressBar.isNull()) {
        m_progressBar->setVisible(true);
      }
      
      delete m_progressUpdateTimer;
      
      m_watcher->setFuture(networks);
      m_mappedRunning = true;
      m_backlog.clear();
      
      m_progressUpdateTimer = new QTimer;
      connect(m_progressUpdateTimer, SIGNAL(timeout()),
              this, SLOT(updateProgressValue()));
      m_progressUpdateTimer->start(100);
    }
  }

  
  void ConcurrentControlNetReader::updateProgressValue() {
    if (!m_mappedRunning) {
      foreach (Progress *progress, m_progress) {
        delete progress;
      }

      m_progress.clear();
    }

    int progressMin = 0;
    int progressMax = (m_progress.count() + m_backlog.count()) * 1000;
    int progressCurrent = 0;

    foreach (Progress *progress, m_progress) {
      if (progress->MaximumSteps()) {
        if (progress->CurrentStep() < progress->MaximumSteps()) {
          double progressPercent = progress->CurrentStep() / (double)progress->MaximumSteps();

          progressCurrent += qFloor(progressPercent * 1000);
        }
        else {
          progressCurrent += 1000;
        }
      }
    }

    if (m_progressBar) {
      if (progressMax > 0) {
        m_progressBar->setRange(progressMin, progressMax);
        m_progressBar->setValue(progressCurrent);
      }
      else {
        m_progressBar->setRange(0, 100);
        m_progressBar->setValue(100);
      }
    }
  }


  void ConcurrentControlNetReader::mappedFinished() {
    m_mappedRunning = false;

    delete m_progressUpdateTimer;
    updateProgressValue();

    QList<Control *> networks(m_watcher->future().results());
    emit networksReady(networks);

    if (!m_backlog.isEmpty()) {
      start();
    }
    else {
      initProgress();
    }
  }


  ConcurrentControlNetReader::FileNameToControlFunctor::FileNameToControlFunctor(
      QThread *targetThread) {
    m_targetThread = targetThread;
  }


  ConcurrentControlNetReader::FileNameToControlFunctor::FileNameToControlFunctor(
      const FileNameToControlFunctor & other) {
    m_targetThread = other.m_targetThread;
  }


  ConcurrentControlNetReader::FileNameToControlFunctor::~FileNameToControlFunctor() {
    m_targetThread = NULL;
  }


  Control * ConcurrentControlNetReader::FileNameToControlFunctor::operator()(
      const QPair<FileName, Progress *> &fileNameAndProgress) const {

    QString fileNameString = fileNameAndProgress.first.expanded();
    Progress *progress = fileNameAndProgress.second;

    ControlNet *newCnet = new ControlNet(fileNameString, progress);
    Control *result = new Control(newCnet, fileNameString);

    newCnet->setParent(result);
    result->moveToThread(m_targetThread);

    return result;
  }


  ConcurrentControlNetReader::FileNameToControlFunctor &
      ConcurrentControlNetReader::FileNameToControlFunctor::operator=(
        const FileNameToControlFunctor &rhs) {
    m_targetThread = rhs.m_targetThread;
    return *this;
  }
}