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) {
    QAction::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) {
    return (controls->count() == 1);
  }


  bool ExportControlNetWorkOrder::execute() {
    bool success = WorkOrder::execute();

    if (success) {
      QStringList internalData;

      Control *control = NULL;
      if (controlList()->isEmpty()) {
        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());
      }
      else {
        control = controlList()->first();
      }

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

    try {
      control->controlNet()->Write(destination);
    }
    catch (IException &e) {
      m_warning = e.toString();
    }
  }


  void ExportControlNetWorkOrder::postSyncRedo() {
    if (!m_warning.isEmpty()) {
      project()->warn(m_warning);
      m_warning.clear();
    }
  }
}