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;
  }
}