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