SNSAppendGeometryToNexus.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 "MantidDataHandling/SNSAppendGeometryToNexus.h"
#include "MantidAPI/FileProperty.h"
#include "MantidAPI/InstrumentFileFinder.h"
#include "MantidAPI/WorkspaceFactory.h"
#include "MantidDataObjects/Workspace2D.h"
#include "MantidGeometry/Instrument.h"
#include "MantidGeometry/Instrument/DetectorInfo.h"
#include "MantidKernel/OptionalBool.h"
#include "MantidNexus/NexusException.h"
#include "MantidNexus/NexusFile.h"
 
#include <filesystem>
using namespace Mantid::Kernel;
using namespace Mantid::API;
using namespace Mantid::Nexus;
 
namespace Mantid::DataHandling {
 
// Register the algorithm into the AlgorithmFactory
DECLARE_ALGORITHM(SNSAppendGeometryToNexus)
 
//----------------------------------------------------------------------------------------------
SNSAppendGeometryToNexus::SNSAppendGeometryToNexus()
    : m_makeNexusCopy(false), m_instrumentLoadedCorrectly(false), m_logsLoadedCorrectly(false) {
  // inform deprecation alias status
  setDeprecationDate("2021-09-14");
}
 
//----------------------------------------------------------------------------------------------
SNSAppendGeometryToNexus::~SNSAppendGeometryToNexus() {
  // delete workspace
}
 
//----------------------------------------------------------------------------------------------
const std::string SNSAppendGeometryToNexus::name() const { return "SNSAppendGeometryToNexus"; }
 
int SNSAppendGeometryToNexus::version() const { return 1; }
 
const std::string SNSAppendGeometryToNexus::category() const { return "DataHandling\\DataAcquisition"; }
 
//----------------------------------------------------------------------------------------------
 
//----------------------------------------------------------------------------------------------
void SNSAppendGeometryToNexus::init() {
  // Declare potential extensions for input NeXus file
  std::vector<std::string> extensions{".nxs", ".h5"};
 
  declareProperty(std::make_unique<API::FileProperty>("Filename", "", API::FileProperty::Load, extensions),
                  "The name of the NeXus file to append geometry to.");
 
  // TODO: change MakeCopy default to False when comfortable. Otherwise need to
  // remove the extra copy once in production.
  declareProperty(std::make_unique<PropertyWithValue<bool>>("MakeCopy", true, Direction::Input),
                  "Copy the NeXus file first before appending (optional, default True).");
}
 
//----------------------------------------------------------------------------------------------
void SNSAppendGeometryToNexus::exec() {
  // TODO: rename the created arrays before moving to production
  g_log.warning() << "This is intended as a proof of principle and not a long "
                     "term implementation.\n";
  g_log.warning() << "(the created arrays in the NeXus file will have the '_new' suffix)\n";
 
  // Retrieve filename from the properties
  m_filename = getPropertyValue("Filename");
 
  // Are we going to make a copy of the file ?
  m_makeNexusCopy = getProperty("MakeCopy");
 
  if (m_makeNexusCopy) {
    std::filesystem::path originalFile(m_filename);
 
    if (std::filesystem::exists(originalFile)) {
      std::filesystem::path destinationFile = std::filesystem::temp_directory_path() / originalFile.filename();
 
      try {
        std::filesystem::copy_file(originalFile, destinationFile);
        g_log.notice() << "Copied " << m_filename << " to " << destinationFile << ".\n";
        m_filename = destinationFile.string();
      } catch (std::filesystem::filesystem_error &) {
        throw std::runtime_error("A Problem occurred in making a copy of the "
                                 "NeXus file. Failed to copy " +
                                 originalFile.string() + " to " + destinationFile.string() +
                                 ". Please check file permissions.");
      }
    } else {
      g_log.error() << "Cannot copy a file that doesn't exist! (" << originalFile << ").\n";
    }
  }
 
  // Let's check to see if we can write to the NeXus file.
  const auto perms = std::filesystem::status(std::filesystem::path(m_filename)).permissions();
  if ((perms & std::filesystem::perms::owner_write) == std::filesystem::perms::none) {
    throw std::runtime_error("The specified NeXus file (" + m_filename + ") is not writable.");
  }
 
  // Let's look for the instrument name
  m_instrument = getInstrumentName(m_filename);
 
  if (m_instrument.length() == 0) {
 
    throw std::runtime_error("Failed to get instrument name from " + m_filename +
                             ". Can't identify instrument definition file.");
  }
 
  // Temp workspace name to load the instrument into
  // std::string workspaceName = "__" + m_instrument + "_geometry_ws";
 
  // Now what is the instrument definition filename ?
  // TODO: Modify to use /entry/instrument/instrument_xml/data after
  // establishing a way to maintain ADARA Geometry Packet
  m_idf_filename = InstrumentFileFinder::getInstrumentFilename(m_instrument);
  g_log.debug() << "Loading instrument definition from " << m_idf_filename << ".\n";
 
  // Modified to call LoadInstrument directly as a Child Algorithm
  ws = WorkspaceFactory::Instance().create("Workspace2D", 1, 2, 1);
 
  // Load NeXus logs for HYSPEC, HYSPECA(testing), and SNAP
  if (m_instrument == "HYSPEC" || m_instrument == "HYSPECA" || m_instrument == "SNAP") {
    g_log.debug() << "Run LoadNexusLogs Child Algorithm.\n";
    m_logsLoadedCorrectly = runLoadNexusLogs(m_filename, ws, this);
 
    if (!m_logsLoadedCorrectly)
      throw std::runtime_error("Failed to run LoadNexusLogs Child Algorithm.");
  }
 
  g_log.debug() << "Run LoadInstrument Child Algorithm.\n";
  m_instrumentLoadedCorrectly = runLoadInstrument(m_idf_filename, ws, this);
 
  if (!m_instrumentLoadedCorrectly)
    throw std::runtime_error("Failed to run LoadInstrument Child Algorithm.");
 
  // Get the number of detectors (just for progress reporting)
  // Get the number of histograms/detectors
  const size_t numDetectors = ws->getInstrument()->getDetectorIDs().size();
 
  API::Progress progress(this, 0.0, 1.0, numDetectors);
 
  // Get the instrument
  Geometry::Instrument_const_sptr instrument = ws->getInstrument();
 
  // Get the sample (needed to calculate distances)
  Geometry::IComponent_const_sptr sample = instrument->getSample();
  // Get the source (moderator)
  Geometry::IComponent_const_sptr source = instrument->getSource();
 
  // Open the NeXus file
  Nexus::File nxfile(m_filename, NXaccess::RDWR);
 
  // using string_map_t = std::map<std::string,std::string>;
  std::map<std::string, std::string>::const_iterator root_iter;
  std::map<std::string, std::string> entries = nxfile.getEntries();
 
  for (root_iter = entries.begin(); root_iter != entries.end(); ++root_iter) {
    // Open all NXentry
    if (root_iter->second == "NXentry") {
      nxfile.openGroup(root_iter->first, "NXentry");
 
      // Get a list of items within the entry.
      std::map<std::string, std::string> entry_items = nxfile.getEntries();
      // Create an iterator for this
      std::map<std::string, std::string>::const_iterator entry_iter;
 
      for (entry_iter = entry_items.begin(); entry_iter != entry_items.end(); ++entry_iter) {
        // Look for an instrument
        if (entry_iter->second == "NXinstrument") {
          // Open the instrument
          nxfile.openGroup(entry_iter->first, "NXinstrument");
          std::map<std::string, std::string> instr_items = nxfile.getEntries();
          std::map<std::string, std::string>::const_iterator instr_iter;
          for (instr_iter = instr_items.begin(); instr_iter != instr_items.end(); ++instr_iter) {
            // Look for NXdetectors
            if (instr_iter->second == "NXdetector") {
              g_log.debug() << "Detector called '" << instr_iter->first << "' found.\n";
              std::string bankName = instr_iter->first;
              std::vector<Geometry::IDetector_const_sptr> dets;
              ws->getInstrument()->getDetectorsInBank(dets, bankName);
 
              if (!dets.empty()) {
                nxfile.openGroup(bankName, "NXdetector");
 
                // Let's create some vectors for the parameters to write
                // Pixel IDs
                std::vector<int> pixel_id;
                std::vector<double> distance;
                std::vector<double> polar_angle;
                std::vector<double> azimuthal_angle;
 
                pixel_id.reserve(dets.size());
                distance.reserve(dets.size());
                polar_angle.reserve(dets.size());
                azimuthal_angle.reserve(dets.size());
 
                const auto &detectorInfo = ws->detectorInfo();
                for (auto &det : dets) {
                  pixel_id.emplace_back(det->getID());
                  distance.emplace_back(det->getDistance(*sample));
                  azimuthal_angle.emplace_back(detectorInfo.phi(detectorInfo.indexOf(det->getID())));
                  polar_angle.emplace_back(ws->detectorTwoTheta(*det));
                }
 
                // Write Pixel ID to file
                nxfile.writeData("pixel_id_new", pixel_id);
 
                // Write Secondary Flight Path to file
                nxfile.writeData("distance_new", distance);
                nxfile.openData("distance_new");
                nxfile.putAttr("units", "metre");
                nxfile.closeData();
 
                // Write Polar Angle (2theta) to file
                nxfile.writeData("polar_angle_new", polar_angle);
                nxfile.openData("polar_angle_new");
                nxfile.putAttr("units", "radian");
                nxfile.closeData();
 
                // Write Azimuthal Angle (Phi) to file
                nxfile.writeData("azimuthal_angle_new", azimuthal_angle);
                nxfile.openData("azimuthal_angle_new");
                nxfile.putAttr("units", "radian");
                nxfile.closeData();
 
                nxfile.closeGroup(); // close NXdetector
 
                progress.report(dets.size());
              } else {
                throw std::runtime_error("Could not find any detectors for the bank named " + bankName +
                                         " that is listed in the NeXus file."
                                         "Check that it exists in the Instrument Definition File.");
              }
            }
          }
 
          nxfile.closeGroup(); // NXinstrument
 
        }
        // Look for monitors
        else if (entry_iter->second == "NXmonitor") {
          g_log.debug() << "Monitor called '" << entry_iter->first << "' found.\n";
          nxfile.openGroup(entry_iter->first, "NXmonitor");
 
          Geometry::IComponent_const_sptr monitor = instrument->getComponentByName(entry_iter->first);
 
          // Write Pixel ID to file
          // nxfile.writeData("pixel_id_new", monitor->get);
 
          double source_monitor = source->getDistance(*monitor);
          double source_sample = source->getDistance(*sample);
 
          g_log.debug() << "source->monitor=" << source_monitor << '\n';
          g_log.debug() << "source->sample=" << source_sample << '\n';
          g_log.debug() << "sample->monitor=" << (source_monitor - source_sample) << '\n';
 
          // Distance
          nxfile.writeData("distance_new", (source_monitor - source_sample));
          nxfile.openData("distance_new");
          nxfile.putAttr("units", "metre");
          nxfile.closeData();
 
          nxfile.closeGroup(); // NXmonitor
        }
      }
 
    } else {
      g_log.error() << "There are no NXentry nodes in the specified NeXus file.\n";
    }
  }
}
 
//----------------------------------------------------------------------------------------------
std::string SNSAppendGeometryToNexus::getInstrumentName(const std::string &nxfilename) {
  std::string instrument;
 
  // Open the NeXus file
  Nexus::File nxfile(nxfilename);
  // What is the first entry ?
  std::map<std::string, std::string> entries = nxfile.getEntries();
 
  // For now, let's just open the first entry
  nxfile.openGroup(entries.begin()->first, "NXentry");
  g_log.debug() << "Using entry '" << entries.begin()->first << "' to determine instrument name.\n";
 
  nxfile.openGroup("instrument", "NXinstrument");
  try {
    nxfile.openData("name");
    instrument = nxfile.getStrData();
  } catch (Nexus::Exception const &) {
    // TODO: try and get the instrument name from the filename instead.
    // Note in filename we have instrument short name yet
    // ExperimentiInfo.getInstrumentFilename() expects instrument long name
    instrument = "";
  }
 
  g_log.debug() << " Instrument name read from NeXus file is " << instrument << '\n';
 
  return instrument;
}
 
//----------------------------------------------------------------------------------------------
bool SNSAppendGeometryToNexus::runLoadInstrument(const std::string &idf_filename,
                                                 const API::MatrixWorkspace_sptr &localWorkspace, Algorithm *alg) {
  auto loadInst = createChildAlgorithm("LoadInstrument", 0, 1, true);
 
  // Execute the Child Algorithm.
  bool executionSuccessful(true);
  try {
    loadInst->setPropertyValue("Filename", idf_filename);
    loadInst->setProperty<MatrixWorkspace_sptr>("Workspace", localWorkspace);
    loadInst->setProperty("RewriteSpectraMap", OptionalBool(false));
    loadInst->execute();
  } catch (std::invalid_argument &e) {
    alg->getLogger().information("Invalid argument to LoadInstrument Child Algorithm");
    alg->getLogger().information(e.what());
    executionSuccessful = false;
  } catch (std::runtime_error &e) {
    alg->getLogger().information("Failed to run LoadInstrument Child Algorithm");
    alg->getLogger().information(e.what());
    executionSuccessful = false;
  }
 
  // Throwing an error if failed
  if (!executionSuccessful) {
    alg->getLogger().error("Error loading instrument\n");
  }
  return executionSuccessful;
}
 
//-----------------------------------------------------------------------------
bool SNSAppendGeometryToNexus::runLoadNexusLogs(const std::string &nexusFileName,
                                                const API::MatrixWorkspace_sptr &localWorkspace, Algorithm *alg) {
  auto loadLogs = alg->createChildAlgorithm("LoadNexusLogs", 0, 1, true);
 
  // Execute the Child Algorithm, catching errors without stopping.
  bool executionSuccessful(true);
  try {
    alg->getLogger().information() << "Loading logs from the NeXus file...\n";
    loadLogs->setPropertyValue("Filename", nexusFileName);
    loadLogs->setProperty<MatrixWorkspace_sptr>("Workspace", localWorkspace);
    loadLogs->executeAsChildAlg();
  } catch (std::invalid_argument &e) {
    alg->getLogger().information("Invalid argument to LoadNexusLogs Child Algorithm");
    alg->getLogger().information(e.what());
    executionSuccessful = false;
  } catch (std::runtime_error &) {
    alg->getLogger().information("Unable to successfully run runLoadNexusLogs Child Algorithm./n");
    executionSuccessful = false;
  }
 
  return executionSuccessful;
}
} // namespace Mantid::DataHandling