MaskWorkspace.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 <utility>
 
#include "MantidAPI/WorkspaceFactory.h"
#include "MantidDataObjects/MaskWorkspace.h"
#include "MantidGeometry/Instrument/DetectorInfo.h"
#include "MantidKernel/IPropertyManager.h"
#include "MantidKernel/System.h"
 
#include <algorithm>
 
namespace Mantid::DataObjects {
using std::set;
using std::size_t;
 
// Register the workspace
DECLARE_WORKSPACE(MaskWorkspace)
 
namespace { // keep these constants only within this file.
const double LIVE_VALUE = 0.;
 
const double DEAD_VALUE = 1.;
 
const double ERROR_VALUE = 0.;
} // namespace
 
//--------------------------------------------------------------------------
 
MaskWorkspace::MaskWorkspace(std::size_t numvectors) {
  this->init(numvectors, 1, 1);
  this->clearMask();
}
 
MaskWorkspace::MaskWorkspace(const Mantid::Geometry::Instrument_const_sptr &instrument, const bool includeMonitors)
    : SpecialWorkspace2D(instrument, includeMonitors) {
  this->clearMask();
}
 
MaskWorkspace::MaskWorkspace(const API::MatrixWorkspace_const_sptr &parent) : SpecialWorkspace2D(parent) {
  this->clearMask();
}
 
//--------------------------------------------------------------------------
 
void MaskWorkspace::clearMask() {
  std::size_t nHist = this->getNumberHistograms();
  for (std::size_t i = 0; i < nHist; ++i) {
    this->dataY(i)[0] = LIVE_VALUE;
    this->dataE(i)[0] = ERROR_VALUE;
  }
 
  // Clear the mask flags
  mutableDetectorInfo().clearMaskFlags();
}
 
size_t MaskWorkspace::getNumberMasked() const {
  size_t numMasked(0);
  const size_t numWksp(this->getNumberHistograms());
  for (size_t i = 0; i < numWksp; i++) {
    if (this->isMaskedIndex(i)) // quick check the value
    {
      numMasked++;
    } else if (this->hasInstrument()) {
      const set<detid_t> ids = this->getDetectorIDs(i);
      if (this->isMasked(ids)) // slow and correct check with the real method
        numMasked += ids.size();
    } else {
      std::stringstream errss;
      errss << "No instrument is associated with mask workspace " << this->getName();
      throw std::runtime_error(errss.str());
    }
  }
  return numMasked;
}
 
set<detid_t> MaskWorkspace::getMaskedDetectors() const {
  set<detid_t> detIDs;
  /*
   * Note
   * This test originally just checked if there was an instument and ignored
   * the number of detectors
   */
  if (this->hasInstrument()) {
    size_t numHist(this->getNumberHistograms());
    for (size_t i = 0; i < numHist; i++) {
      if (this->isMaskedIndex(i)) {
        set<detid_t> temp = this->getDetectorIDs(i);
        detIDs.insert(temp.begin(), temp.end());
      }
    }
  }
 
  return detIDs;
}
 
set<size_t> MaskWorkspace::getMaskedWkspIndices() const {
  set<size_t> indices;
 
  size_t numHist(this->getNumberHistograms());
  for (size_t i = 0; i < numHist; i++) {
    if (this->isMaskedIndex(i)) {
      indices.insert(i);
    }
  }
 
  return indices;
}
 
//--------------------------------------------------------------------------------------------
bool MaskWorkspace::isMasked(const detid_t detectorID) const {
  if (!this->hasInstrument()) {
    std::stringstream msg;
    if (!this->getInstrument())
      msg << "There is no instrument associated with workspace \'" << this->getName() << "\'";
    else
      msg << "There is no proper instrument associated with workspace \'" << this->getName()
          << "\'.  Number of detectors = " << this->getInstrument()->getNumberDetectors();
    throw std::runtime_error(msg.str());
  }
 
  // return true if the value isn't zero
  if (this->getValue(detectorID, LIVE_VALUE) != LIVE_VALUE) {
    return true;
  }
 
  // the mask bit on the workspace can be set
  // Performance wise, it is not optimal to call detectorInfo() for every index,
  // but this method seems to be used rarely enough to justify this until the
  // Instrument-2.0 implementation has progressed far enough to make this cheap.
  const auto &detectorInfo = this->detectorInfo();
  try {
    return detectorInfo.isMasked(detectorInfo.indexOf(detectorID));
  } catch (std::out_of_range &) {
    // The workspace can contain bad detector IDs. DetectorInfo::indexOf throws.
    return false;
  }
}
 
bool MaskWorkspace::isMasked(const std::set<detid_t> &detectorIDs) const {
  if (detectorIDs.empty()) {
    return false;
  }
 
  const auto it = std::find_if_not(detectorIDs.cbegin(), detectorIDs.cend(),
                                   [this](const auto detectorID) { return this->isMasked(detectorID); });
  return it == detectorIDs.cend();
}
 
bool MaskWorkspace::isMaskedIndex(const std::size_t wkspIndex) const {
  return (this->dataY(wkspIndex)[0] != LIVE_VALUE); // if is not live it should masked
}
 
void MaskWorkspace::setMasked(const detid_t detectorID, const bool mask) {
  double value(LIVE_VALUE);
  if (mask)
    value = DEAD_VALUE;
 
  this->setValue(detectorID, value, ERROR_VALUE);
}
 
void MaskWorkspace::setMasked(const std::set<detid_t> &detectorIDs, const bool mask) {
  for (auto detectorID : detectorIDs) {
    this->setMasked(detectorID, mask);
  }
}
 
void MaskWorkspace::setMaskedIndex(const std::size_t wkspIndex, const bool mask) {
  double value(LIVE_VALUE);
  if (mask)
    value = DEAD_VALUE;
 
  this->dataY(wkspIndex)[0] = value;
}
 
const std::string MaskWorkspace::id() const { return "MaskWorkspace"; }
 
//--------------------------------------------------------------------------------------------
void MaskWorkspace::copyFrom(std::shared_ptr<const SpecialWorkspace2D> sourcews) {
  SpecialWorkspace2D::copyFrom(sourcews);
}
 
const std::string MaskWorkspace::toString() const {
  std::ostringstream os;
  os << SpecialWorkspace2D::toString();
  os << "Masked: " << getNumberMasked() << "\n";
  return os.str();
}
 
//--------------------------------------------------------------------------------------------
bool MaskWorkspace::hasInstrument() const {
  bool hasinst;
  Geometry::Instrument_const_sptr inst = this->getInstrument();
  if (inst) {
    hasinst = inst->getNumberDetectors() > 0;
  } else
    hasinst = false;
 
  return hasinst;
}
 
} // namespace Mantid::DataObjects
 
 
namespace Mantid::Kernel {
 
template <>
DLLExport Mantid::DataObjects::MaskWorkspace_sptr
IPropertyManager::getValue<Mantid::DataObjects::MaskWorkspace_sptr>(const std::string &name) const {
  auto *prop = dynamic_cast<PropertyWithValue<Mantid::DataObjects::MaskWorkspace_sptr> *>(getPointerToProperty(name));
  if (prop) {
    return *prop;
  } else {
    std::string message =
        "Attempt to assign property " + name + " to incorrect type. Expected shared_ptr<MaskWorkspace>.";
    throw std::runtime_error(message);
  }
}
 
template <>
DLLExport Mantid::DataObjects::MaskWorkspace_const_sptr
IPropertyManager::getValue<Mantid::DataObjects::MaskWorkspace_const_sptr>(const std::string &name) const {
  auto *prop = dynamic_cast<PropertyWithValue<Mantid::DataObjects::MaskWorkspace_sptr> *>(getPointerToProperty(name));
  if (prop) {
    return prop->operator()();
  } else {
    std::string message =
        "Attempt to assign property " + name + " to incorrect type. Expected const shared_ptr<MaskWorkspace>.";
    throw std::runtime_error(message);
  }
}
 
} // namespace Mantid::Kernel