ImportRegistrationTemplateWorkOrder.cpp

Go to the documentation of this file.

#include "ImportRegistrationTemplateWorkOrder.h"

#include <QFile>
#include <QFileDialog>
#include <QMessageBox>
#include <QRegularExpression>
#include <QDebug>

#include "Project.h"
#include "ProjectItemModel.h"
#include "Template.h"
#include "TemplateList.h"

namespace Isis {
  ImportRegistrationTemplateWorkOrder::ImportRegistrationTemplateWorkOrder(Project *project) :
      WorkOrder(project) {

    m_isUndoable = true;
    m_list = NULL;

    QAction::setText(tr("Import Registration Templates..."));
    QUndoCommand::setText(tr("Import Registration Templates..."));
    setModifiesDiskState(true);

  }


  ImportRegistrationTemplateWorkOrder::ImportRegistrationTemplateWorkOrder(const ImportRegistrationTemplateWorkOrder &other) :
      WorkOrder(other) {
  }


  ImportRegistrationTemplateWorkOrder::~ImportRegistrationTemplateWorkOrder() {
    m_list = NULL;
  }


  ImportRegistrationTemplateWorkOrder *ImportRegistrationTemplateWorkOrder::clone() const {
    return new ImportRegistrationTemplateWorkOrder(*this);
  }


  bool ImportRegistrationTemplateWorkOrder::isExecutable(ProjectItem *item) {
    QString itemType = item->text();
    setInternalData(QStringList(itemType));

    return (itemType == "Registrations");
  }


  bool ImportRegistrationTemplateWorkOrder::setupExecution() {

    WorkOrder::setupExecution();

    QStringList templateFileNames;

    templateFileNames = QFileDialog::getOpenFileNames(
        qobject_cast<QWidget *>(parent()),
        "Import Registration Templates",
        QString(),
        "Registrations (*.def);; All Files (*)");

    if (!templateFileNames.isEmpty()) {
      QUndoCommand::setText(tr("Import %1 Template(s)").arg(templateFileNames.count()));
    }
    else {
      return false;
    }

    setInternalData(templateFileNames);

    return true;
  }


  void ImportRegistrationTemplateWorkOrder::execute() {
    QDir templateFolder = project()->addTemplateFolder("registrations/import");
    QStringList templateFileNames = internalData();

    m_list = new TemplateList(templateFolder.dirName(), 
                              "registrations", 
                              "registrations/" + templateFolder.dirName() );

    foreach (FileName filename, templateFileNames) {
      QFile::copy(filename.expanded(), templateFolder.path() + "/" + filename.name());
      m_list->append(new Template(templateFolder.path() + "/" + filename.name(), 
                                  "registrations", 
                                  templateFolder.dirName()));
    }

    if (!m_list->isEmpty()) {
      project()->addTemplates(m_list);
      project()->setClean(false);
    }

  }


  void ImportRegistrationTemplateWorkOrder::undoExecution() {
    if (m_list && project()->templates().size() > 0) {
      m_list->deleteFromDisk( project() );
      ProjectItem *currentItem =
          project()->directory()->model()->findItemData(QVariant::fromValue(m_list));
      project()->directory()->model()->removeItem(currentItem);
    }
    foreach ( Template *currentTemplate, *m_list) {
      delete currentTemplate;
    }
    delete m_list;
    m_list = NULL;
  }

}