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