ConvToMDEventsWS.h

Go to the documentation of this file.

// Mantid Repository : https://github.com/mantidproject/mantid
//
// Copyright © 2010 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 +
#pragma once
//
#include "MantidAPI/Algorithm.h"
#include "MantidDataObjects/EventWorkspace.h"
#include "MantidDataObjects/MDBoxBase.h"
#include "MantidDataObjects/MDEvent.h"
#include "MantidGeometry/Instrument/Goniometer.h"
#include "MantidKernel/Exception.h"
#include "MantidKernel/Matrix.h"
#include "MantidKernel/PhysicalConstants.h"
#include "MantidMDAlgorithms/ConvToMDBase.h"
#include "MantidMDAlgorithms/MDEventWSWrapper.h"
#include "MantidMDAlgorithms/MDTransfFactory.h"
 
namespace Mantid {
// Forward declarations
namespace API {
class Progress;
}
namespace MDAlgorithms {
// Class to process event workspace by direct conversion:
 
class ConvToMDEventsWS : public ConvToMDBase {
public:
  size_t initialize(const MDWSDescription &WSD, std::shared_ptr<MDEventWSWrapper> inWSWrapper, bool ignoreZeros,
                    bool useLogTimes) override;
  void runConversion(API::Progress *pProgress) override;
 
protected:
  DataObjects::EventWorkspace_const_sptr m_EventWS;
 
private:
  // function runs the conversion on
  size_t conversionChunk(size_t workspaceIndex) override;
  // the pointer to the source event workspace as event ws does not work through
  // the public Matrix WS interface
  template <class T> size_t convertEventList(size_t workspaceIndex);
 
  virtual void appendEventsFromInputWS(API::Progress *pProgress, const API::BoxController_sptr &bc);
  // Variables for getting log values at times and recomputing sample orientation
  Kernel::DblMatrix m_Wtransf;
  Kernel::DblMatrix m_tmpRot;
  Geometry::Goniometer m_Goniometer;
  std::vector<std::unique_ptr<Kernel::TimeSeriesProperty<double>>> m_Logs;
  std::vector<size_t> m_GonioIndex;
  // Private method to update rotation matrix for a single event from log values
  template <class T> bool setGoniometersFromLogs(const T &ev) {
    if (!m_useLogTimes)
      return true;
    for (size_t axIdx = 0; axIdx < m_GonioIndex.size(); axIdx++) {
      double logval = m_Logs[axIdx]->getSingleValue(ev->pulseTime());
      if (std::isnan(logval))
        return false;
      m_Goniometer.setRotationAngle(m_GonioIndex[axIdx], logval);
    }
    m_tmpRot = m_Goniometer.getR() * m_Wtransf;
    m_tmpRot.Invert();
    m_QConverter->updateRotMat(m_tmpRot.getVector());
    return true;
  }
};
 
} // namespace MDAlgorithms
} // namespace Mantid