ExportControlNetWorkOrder.cpp

Go to the documentation of this file.

 
#include "ExportControlNetWorkOrder.h"
 
#include <QDebug>
#include <QFileDialog>
#include <QInputDialog>
 
#include "Control.h"
#include "ControlList.h"
#include "ControlNet.h"
#include "FileName.h"
#include "IException.h"
#include "IString.h"
#include "Project.h"
 
namespace Isis {
 
  ExportControlNetWorkOrder::ExportControlNetWorkOrder(Project *project) :
      WorkOrder(project) {
    m_isSynchronous = false;
    m_isUndoable = false;
    QAction::setText(tr("&Export Control Network..."));
    QUndoCommand::setText(tr("Export Control Network..."));
  }
 
 
  ExportControlNetWorkOrder::ExportControlNetWorkOrder(const ExportControlNetWorkOrder &other) :
      WorkOrder(other) {
  }
 
 
  ExportControlNetWorkOrder::~ExportControlNetWorkOrder() {
 
  }
 
 
  ExportControlNetWorkOrder *ExportControlNetWorkOrder::clone() const {
    return new ExportControlNetWorkOrder(*this);
  }
 
 
  bool ExportControlNetWorkOrder::isExecutable(ControlList *controls) {
 
    if (controls) {
      return (controls->count() == 1);
    }
    return false;
  }
 
 
  bool ExportControlNetWorkOrder::setupExecution() {
    bool success = WorkOrder::setupExecution();
 
    if (success) {
      QStringList internalData;
 
      Control *control = NULL;
 
      // See if there are any other control lists in the project and give these to the user as
      // choices for control nets they can export.
      if (project()) {
 
        Project *proj = project();
 
        QList<ControlList *> controls = proj->controls();
        if (controls.count() > 0) {
          ControlList *l = controls.first();
          WorkOrder::setData(l);
          control = controlList()->first();
        }
        else {
 
          QMap<Control *, QString> cnetChoices;
          foreach (ControlList *list, project()->controls()) {
            foreach (Control *control, *list) {
              cnetChoices[control] = tr("%1/%2").arg(list->name())
                  .arg(control->displayProperties()->displayName());
            }
          }
 
          QStringList cnetNames = cnetChoices.values();
          qSort(cnetNames);
 
          QString choice = QInputDialog::getItem(NULL, tr("Select Control"),
              tr("Please choose a control to export."), cnetNames, 0, false, &success);
 
          control = cnetChoices.key(choice);
          internalData.append(control->id());
        }
      }
 
 
      QString destination =
          QFileDialog::getSaveFileName(NULL, QString("Export Control Network"),
          "./" + FileName(control->fileName()).name());
 
      if (destination.isEmpty()) {
        success = false;
      }
      internalData.append(destination);
 
      setInternalData(internalData);
    }
 
    return success;
  }
 
 
  void ExportControlNetWorkOrder::execute() {
    QString destination;
    Control *control = NULL;
 
    if (controlList()->isEmpty()) {
      destination = internalData()[1];
 
      QString controlId = internalData()[0];
      control = project()->control(controlId);
    }
    else {
      destination = internalData()[0];
      control = controlList()->first();
    }
 
    QString currentLocation = control->fileName();
    if (!QFile::copy(currentLocation, destination) ) {
      m_warning = "Error saving control net.";
    }
  }
 
 
  void ExportControlNetWorkOrder::postExecution() {
    if (!m_warning.isEmpty()) {
      project()->warn(m_warning);
      m_warning.clear();
    }
  }
}