ProjectSavePresenter.cpp

Go to the documentation of this file.

// Mantid Repository : https://github.com/mantidproject/mantid
//
// Copyright © 2018 ISIS Rutherford Appleton Laboratory UKRI,
//   NScD Oak Ridge National Laboratory, European Spallation Source,
//   Institut Laue - Langevin & CSNS, Institute of High Energy Physics, CAS
// SPDX - License - Identifier: GPL - 3.0 +
#include "MantidQtWidgets/Common/ProjectSavePresenter.h"
#include "MantidAPI/AnalysisDataService.h"
#include "MantidAPI/Workspace.h"
 
#include <QApplication>
#include <QDir>
#include <QFileInfo>
 
#include <algorithm>
#include <iterator>
#include <vector>
 
using namespace MantidQt::MantidWidgets;
using namespace Mantid::API;
 
ProjectSavePresenter::ProjectSavePresenter(IProjectSaveView *view)
    : m_view(view), m_model(m_view->getWindows(), m_view->getAllPythonInterfaces()) {
  auto workspaceNames = m_model.getWorkspaceNames();
  auto info = m_model.getWorkspaceInformation();
  auto winInfo = m_model.getWindowInformation(workspaceNames, true);
  m_view->updateIncludedWindowsList(winInfo);
  m_view->updateWorkspacesList(info);
  m_view->updateInterfacesList(m_model.getAllPythonInterfaces());
}
 
void ProjectSavePresenter::notify(Notification notification) {
  switch (notification) {
  case Notification::CheckWorkspace:
    includeWindowsForCheckedWorkspace();
    break;
  case Notification::UncheckWorkspace:
    excludeWindowsForUncheckedWorkspace();
    break;
  case Notification::PrepareProjectFolder:
    prepareProjectFolder();
  }
}
 
bool ProjectSavePresenter::needsSizeWarning(const std::vector<std::string> &wsNames) {
  if (!wsNames.empty()) {
    return m_model.needsSizeWarning(wsNames);
  }
  return false;
}
 
void ProjectSavePresenter::includeWindowsForCheckedWorkspace() {
  auto wsNames = m_view->getCheckedWorkspaceNames();
  auto names = m_model.getWindowNames(wsNames);
  // true flag gets windows unattached to workspaces as well
  auto info = m_model.getWindowInformation(wsNames, true);
  m_view->updateIncludedWindowsList(info);
  m_view->removeFromExcludedWindowsList(names);
}
 
void ProjectSavePresenter::excludeWindowsForUncheckedWorkspace() {
  auto wsNames = m_view->getUncheckedWorkspaceNames();
  auto names = m_model.getWindowNames(wsNames);
  auto info = m_model.getWindowInformation(wsNames);
  m_view->updateExcludedWindowsList(info);
  m_view->removeFromIncludedWindowsList(names);
}
 
void ProjectSavePresenter::prepareProjectFolder() {
  auto path = m_view->getProjectPath();
 
  QFileInfo fileInfo(path);
  bool isFile = fileInfo.filePath().endsWith(".mantid") || fileInfo.filePath().endsWith(".mantid.gz");
 
  if (!isFile) {
    QDir directory(path);
    if (!directory.exists()) {
      // Make the directory
      directory.mkdir(path);
    }
 
    auto projectFileName = directory.dirName();
    projectFileName.append(".mantid");
    path = directory.absoluteFilePath(projectFileName);
  } else {
    path = fileInfo.absoluteFilePath();
  }
 
  m_view->setProjectPath(path);
}