LoadDialog.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 +
//------------------------------------------------------------------------------
// Includes
//------------------------------------------------------------------------------
#include "MantidQtWidgets/Plugins/AlgorithmDialogs/LoadDialog.h"
#include "MantidQtWidgets/Common/AlgorithmInputHistory.h"
#include "MantidQtWidgets/Common/FileFinderWidget.h"
// Qt
#include <QCheckBox>
#include <QComboBox>
#include <QDesktopWidget>
#include <QFileInfo>
#include <QScreen>
#include <QUrl>
 
// Mantid
#include "MantidAPI/AlgorithmManager.h"
#include "MantidAPI/FileProperty.h"
#include "MantidAPI/IWorkspaceProperty.h"
#include "MantidKernel/MaskedProperty.h"
#include "MantidKernel/Property.h"
#include "MantidQtWidgets/Common/HelpWindow.h"
 
using namespace Mantid::API;
using namespace Mantid::Kernel;
using MantidQt::API::FileFinderWidget;
 
namespace MantidQt::CustomDialogs {
namespace {
struct HoldFlag {
  HoldFlag(bool &current, const bool holdValue) : initial(current), heldflag(current) { heldflag = holdValue; }
  ~HoldFlag() { heldflag = initial; }
  bool initial;
  bool &heldflag;
};
} // namespace
 
// Declare the dialog. Name must match the class name
DECLARE_DIALOG(LoadDialog)
 
//--------------------------------------------------------------------------
// Public methods
//---------------------------------------------------------------------------
 
 
LoadDialog::LoadDialog(QWidget *parent)
    : API::AlgorithmDialog(parent), m_form(), m_currentFiles(), m_initialHeight(0), m_populating(false),
      m_userAccept(false) {
  // We will handle parsing the input ourselves on startup
  m_autoParseOnInit = false;
}
 
//--------------------------------------------------------------------------
// Private methods (slot)
//---------------------------------------------------------------------------
 
void LoadDialog::createDynamicWidgets() {
  HoldFlag hold(m_populating, true);
 
  m_form.fileWidget->blockSignals(true);
  createDynamicLayout();
  m_form.fileWidget->blockSignals(false);
}
 
void LoadDialog::helpClicked() {
  const auto loaderName = getAlgorithm()->getPropertyValue("LoaderName");
  QString helpPage = (loaderName.empty()) ? QString("Load") : QString::fromStdString(loaderName);
  MantidQt::API::HelpWindow::showAlgorithm(helpPage);
}
 
void LoadDialog::suggestWSName() {
  if (!m_form.workspaceEdit->isEnabled())
    return;
 
  // suggest ws name based on file name
  QString fileSuggestion;
  if (m_form.fileWidget->isValid()) {
    if (m_form.fileWidget->getFilenames().size() == 1)
      fileSuggestion = QFileInfo(m_form.fileWidget->getFirstFilename()).completeBaseName();
    else
      fileSuggestion = "MultiFiles";
  }
  m_form.workspaceEdit->setText(fileSuggestion);
}
 
void LoadDialog::enableNameSuggestion(const bool on) {
  if (on) {
    connect(m_form.fileWidget, SIGNAL(filesFound()), this, SLOT(suggestWSName()));
  } else {
    disconnect(m_form.fileWidget, SIGNAL(filesFound()), this, SLOT(suggestWSName()));
  }
}
 
void LoadDialog::accept() {
  // If the LoadDialog is already loading data, or is populating, then ignore the accept
  if (m_form.fileWidget->isSearching() || m_populating) {
    return;
  }
  m_userAccept = true;
  m_form.fileWidget->findFiles();
}
 
void LoadDialog::resultInspectionFinished() {
  if (!m_userAccept) {
    return;
  }
  m_userAccept = false;
 
  // Makes it so the dialog is still resizable if it is kept open
  m_form.propertyLayout->setEnabled(true);
 
  // Check that the file still exists just incase it somehow got removed
  std::string errMess = getAlgorithm()->getPointerToProperty("Filename")->isValid();
  if (!errMess.empty()) {
    m_currentFiles = "";
    createDynamicWidgets();
  } else {
    AlgorithmDialog::accept();
  }
}
 
//--------------------------------------------------------------------------
// Private methods (non-slot)
//---------------------------------------------------------------------------
 
void LoadDialog::initLayout() {
  m_form.setupUi(this);
 
  // Add the helpful summary message
  if (isMessageAvailable())
    m_form.instructions->setText(getOptionalMessage());
 
  m_form.dialogLayout->addLayout(this->createDefaultButtonLayout());
  m_form.fileWidget->readSettings("Mantid/Algorithms/Load");
  m_initialHeight = this->height();
 
  const std::string &outWsName = getAlgorithm()->getPropertyValue("OutputWorkspace");
  if (!outWsName.empty()) {
    // OutputWorkspace name suggestion received as parameter, just take it and
    // don't change it
    m_form.workspaceEdit->setText(QString::fromStdString(outWsName));
  } else {
    // Guess at an output workspace name but only if the user hasn't changed
    // anything
    enableNameSuggestion(true);
    connect(m_form.workspaceEdit, SIGNAL(textEdited(const QString &)), this, SLOT(enableNameSuggestion()));
  }
 
  // Connect the file finder's file found signal to the dynamic property create
  // method.
  // When the file text is set the Load algorithm finds the concrete loader and
  // then we
  // know what extra properties to create
  connect(m_form.fileWidget, SIGNAL(filesFound()), this, SLOT(createDynamicWidgets()));
  connect(m_form.fileWidget, SIGNAL(fileInspectionFinished()), this, SLOT(resultInspectionFinished()));
 
  tieStaticWidgets(true);
}
 
void LoadDialog::saveInput() {
  m_form.fileWidget->saveSettings("Mantid/Algorithms/Load");
  AlgorithmDialog::saveInput();
  // Ensure the filename is store as the full file
  API::AlgorithmInputHistory::Instance().storeNewValue("Load", QPair<QString, QString>("Filename", m_currentFiles));
}
 
void LoadDialog::tieStaticWidgets(const bool readHistory) {
  // If a workspace validator asterisk exists, remove it since the underlying
  // AlgorithmDialog gets confused
  if (m_form.workspaceLayout->count() == 3) {
    QLayoutItem *validLbl = m_form.workspaceLayout->takeAt(2);
    delete validLbl->widget();
    delete validLbl;
  }
  tie(m_form.workspaceEdit, "OutputWorkspace", m_form.workspaceLayout, readHistory);
  tie(m_form.fileWidget, "Filename", nullptr, readHistory);
}
 
void LoadDialog::removeOldInputWidgets(QVBoxLayout *layout) {
  // Remove the old widgets if necessary
  if (layout->count() > 2) {
    int count = layout->count();
    while (count > 2) {
      QLayoutItem *child = layout->takeAt(count - 1);
      if (QWidget *w = child->widget()) {
        w->deleteLater();
      } else if (QLayout *l = child->layout()) {
        QLayoutItem *subChild(nullptr);
        while ((subChild = l->takeAt(0)) != nullptr) {
          subChild->widget()->deleteLater();
        }
      }
      count = layout->count();
    }
  }
}
 
void LoadDialog::createDynamicLayout() {
  // Disable the layout so that a widget cannot be interacted with while it may
  // be being deleted
  m_form.propertyLayout->setEnabled(false);
 
  if (!m_form.fileWidget->isValid())
    return;
  // First step is the get the specific loader that is responsible
  auto loadAlg = getAlgorithm();
  const QString filenames = m_form.fileWidget->getUserInput().toString();
  if (filenames == m_currentFiles)
    return;
  m_currentFiles = filenames;
  removeOldInputWidgets(m_form.propertyLayout); // The new file might be invalid
  try {
    loadAlg->setPropertyValue("Filename", filenames.toStdString());
  } catch (std::exception &exc) {
    m_form.fileWidget->setFileProblem(QString::fromStdString(exc.what()));
    m_form.propertyLayout->setEnabled(true);
    m_form.propertyLayout->activate();
    this->resize(this->width(), m_initialHeight + 15);
 
    // Reset the algorithm pointer so that the base class re-reads the
    // properties and drops links from
    // old widgets meaning they are safe to remove
    setAlgorithm(loadAlg);
    tieStaticWidgets(false); // The ties are cleared when resetting the
                             // algorithm
 
    return;
  }
  // Reset the algorithm pointer so that the base class re-reads the properties
  // and drops links from
  // old widgets meaning they are safe to remove
  setAlgorithm(loadAlg);
  tieStaticWidgets(false); // The ties are cleared when resetting the algorithm
  // Add the new ones
  const std::vector<Property *> &inputProps = loadAlg->getProperties();
  int dialogHeight = m_initialHeight;
  for (auto prop : inputProps) {
    const QString propName = QString::fromStdString(prop->name());
    if (propName == "OutputWorkspace" || propName == "Filename")
      continue;
    if (requiresUserInput(propName)) {
      dialogHeight += createWidgetsForProperty(prop, m_form.propertyLayout, m_form.scrollAreaWidgetContents);
    }
  }
  // Re-enable and recompute the size of the layout
  m_form.propertyLayout->setEnabled(true);
  m_form.propertyLayout->activate();
 
  const auto screenSize = screen()->availableSize();
  const auto screenGeometry = screen()->availableGeometry();
  dialogHeight = std::min(dialogHeight, static_cast<int>(screenSize.height() * 0.65));
  this->resize(this->width(), dialogHeight);
  const auto xPos = screenGeometry.x() + (screenSize.width() - this->width()) / 2;
  const auto yPos = screenGeometry.y() + (screenSize.height() - this->height()) / 2;
  this->move(xPos, yPos);
 
  // Make sure the OutputWorkspace value has been stored so that the validator
  // is cleared appropriately
  QString wsName(m_form.workspaceEdit->text());
  if (!wsName.isEmpty())
    storePropertyValue("OutputWorkspace", wsName);
  setPropertyValues(QStringList("Filename"));
}
 
int LoadDialog::createWidgetsForProperty(const Mantid::Kernel::Property *prop, QVBoxLayout *propertyLayout,
                                         QWidget *parent) {
  QString propName = QString::fromStdString(prop->name());
  QWidget *inputWidget(nullptr);
  QHBoxLayout *widgetLayout(nullptr);
  bool addValidator(true);
 
  // Boolean properties use the name labels differently
  if (const auto *fileType = dynamic_cast<const FileProperty *>(prop)) {
    auto *fileFinder = new FileFinderWidget(parent);
    inputWidget = fileFinder;
    fileFinder->setLabelText(propName);
    fileFinder->isForRunFiles(false);
    fileFinder->isOptional(fileType->isOptional());
    fileFinder->doMultiEntry(false);
    addValidator = false;
    propertyLayout->addWidget(inputWidget);
  } else {
    QLabel *nameLbl = new QLabel(propName, parent);
    nameLbl->setToolTip(QString::fromStdString(prop->documentation()));
    if (dynamic_cast<const PropertyWithValue<bool> *>(prop)) {
      auto *checkBox = new QCheckBox(parent);
      inputWidget = checkBox;
      addValidator = false;
    }
    // Options box
    else if (!prop->allowedValues().empty()) {
      auto *optionsBox = new QComboBox(parent);
      inputWidget = optionsBox;
      std::vector<std::string> items = prop->allowedValues();
      std::vector<std::string>::const_iterator vend = items.end();
      for (std::vector<std::string>::const_iterator vitr = items.begin(); vitr != vend; ++vitr) {
        optionsBox->addItem(QString::fromStdString(*vitr));
      }
      // Set current as visible
      int index = optionsBox->findText(QString::fromStdString(prop->value()));
      if (index >= 0)
        optionsBox->setCurrentIndex(index);
 
      addValidator = false;
    }
    // else render a text box
    else {
      auto *textBox = new QLineEdit(parent);
      inputWidget = textBox;
      if (dynamic_cast<const MaskedProperty<std::string> *>(prop)) {
        textBox->setEchoMode(QLineEdit::Password);
      }
    }
    nameLbl->setBuddy(inputWidget);
    widgetLayout = new QHBoxLayout();
    widgetLayout->addWidget(nameLbl);
    widgetLayout->addWidget(inputWidget);
    propertyLayout->addLayout(widgetLayout);
  }
 
  if (addValidator)
    tie(inputWidget, propName, widgetLayout);
  else
    tie(inputWidget, propName, nullptr);
 
  return inputWidget->geometry().height();
}
} // namespace MantidQt::CustomDialogs