ExportImagesWorkOrder.cpp

Go to the documentation of this file.

 
#include "ExportImagesWorkOrder.h"
 
#include <QDebug>
#include <QFileDialog>
#include <QFuture>
#include <QInputDialog>
#include <QtConcurrentMap>
 
#include "Cube.h"
#include "CubeAttribute.h"
#include "Image.h"
#include "ImageList.h"
#include "FileName.h"
#include "IException.h"
#include "IString.h"
#include "Project.h"
 
namespace Isis {
 
  ExportImagesWorkOrder::ExportImagesWorkOrder(Project *project) :
      WorkOrder(project) {
 
    // This is an asynchronous workorder and not undoable
    m_isSynchronous = false;
    m_isUndoable = false;
 
    QAction::setText(tr("Export I&mages..."));
  }
 
 
  ExportImagesWorkOrder::ExportImagesWorkOrder(const ExportImagesWorkOrder &other) :
      WorkOrder(other) {
  }
 
 
  ExportImagesWorkOrder::~ExportImagesWorkOrder() {
 
  }
 
 
  ExportImagesWorkOrder *ExportImagesWorkOrder::clone() const {
    return new ExportImagesWorkOrder(*this);
  }
 
 
  bool ExportImagesWorkOrder::isExecutable(ImageList *images) {
    if (images) {
      return (!images->isEmpty());
    }
    return false;
  }
 
 
  bool ExportImagesWorkOrder::setupExecution() {
    bool success = WorkOrder::setupExecution();
 
    if (success) {
      QStringList internalData;
 
      if (imageList()->isEmpty()) {
        QStringList imageListNames;
        foreach (ImageList *list, project()->images()) {
          imageListNames.append(list->name());
        }
 
        qSort(imageListNames);
 
        QString choice = QInputDialog::getItem(NULL, tr("Select Image List"),
            tr("Please choose a list of images to export."), imageListNames, 0, false, &success);
 
        internalData.append(choice);
 
        QUndoCommand::setText(tr("Export image list [%1]").arg(choice));
      }
      else {
        QUndoCommand::setText(tr("Export [%1] images").arg(imageList()->count()));
      }
 
      QString destination = QFileDialog::getExistingDirectory(NULL, tr("Export Images"), ".");
 
      if (destination.isEmpty()) {
        success = false;
      }
      internalData.append(destination);
 
      setInternalData(internalData);
    }
 
    return success;
  }
 
 
  void ExportImagesWorkOrder::execute() {
    ImageList *list = imageList();
 
    if (list->isEmpty()) {
      list = project()->imageList(internalData()[0]);
    }
 
 
    QString destination = internalData().last();
    ProjectImageExportFunctor functor(destination);
 
    QFuture<void *> future = QtConcurrent::mapped(*list, functor);
 
    setProgressRange(0, list->count());
 
    QThreadPool::globalInstance()->releaseThread();
    for (int i = 0; i < list->count(); i++) {
      setProgressValue(i);
 
      // Wait for the result before updating the progress to the next value... it's still
      //   working on writing N images, just progress is going 1 by 1.
      future.resultAt(i);
    }
    QThreadPool::globalInstance()->reserveThread();
 
    // Get the errors that occurred during the mapped in order to report them in postSyncRedo
    m_warning = functor.errors().toString();
  }
 
 
  void ExportImagesWorkOrder::postExecution() {
    if (!m_warning.isEmpty()) {
      project()->warn(m_warning);
      m_warning.clear();
    }
  }
 
 
  ExportImagesWorkOrder::ProjectImageExportFunctor::ProjectImageExportFunctor(
      QString destination) : m_errors(new IException()), m_numErrors(new int(0)) {
 
    m_destination = destination;
  }
 
 
  ExportImagesWorkOrder::ProjectImageExportFunctor::ProjectImageExportFunctor(
      const ProjectImageExportFunctor &other) : m_errors(other.m_errors),
      m_numErrors(other.m_numErrors) {
 
    m_destination = other.m_destination;
  }
 
 
  ExportImagesWorkOrder::ProjectImageExportFunctor::~ProjectImageExportFunctor() {
  }
 
 
  void *ExportImagesWorkOrder::ProjectImageExportFunctor::operator()(Image * const &imageToExport) {
    try {
      FileName outputFileName =
          m_destination + "/" + FileName(imageToExport->fileName()).baseName();
 
      // Copy the image, free the copy from memory because we don't need it.
      delete imageToExport->cube()->copy(outputFileName, CubeAttributeOutput());
      // Avoid opening too many cubes
      imageToExport->closeCube();
    }
    catch (IException &e) {
      QMutexLocker locker(&m_errorsLock);
      m_errors->append(e);
      (*m_numErrors)++;
    }
 
    return NULL;
  }
 
 
  IException ExportImagesWorkOrder::ProjectImageExportFunctor::errors() const {
    IException result;
 
    result.append(*m_errors);
 
    if (*m_numErrors != 0) {
      result.append(
          IException(IException::Unknown,
                     tr("Failed to export [%1] images").arg(*m_numErrors),
                     _FILEINFO_));
    }
 
    return result;
  }
}