SetCrystalLocation.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 +
/*
 *
 * SetCrystalLocation.cpp
 *
 *  Created on: Dec 12, 2018
 *      Author: Brendan Sullivan
 */
#include "MantidCrystal/SetCrystalLocation.h"
 
#include "MantidAPI/IMDEventWorkspace.h"
#include "MantidAPI/Run.h"
#include "MantidAPI/Sample.h"
#include "MantidAPI/WorkspaceFactory.h"
#include "MantidCrystal/CalibrationHelpers.h"
#include "MantidCrystal/PeakHKLErrors.h"
#include "MantidDataObjects/EventWorkspace.h"
#include "MantidGeometry/Crystal/IPeak.h"
#include "MantidGeometry/Crystal/IndexingUtils.h"
#include "MantidGeometry/Instrument/ComponentInfo.h"
#include "MantidGeometry/Instrument/Goniometer.h"
#include "MantidKernel/ArrayProperty.h"
#include "MantidKernel/EnabledWhenProperty.h"
 
#include <cstdarg>
 
using namespace Mantid::API;
using namespace Mantid::DataObjects;
using namespace Mantid::Kernel;
using Mantid::Geometry::IndexingUtils;
using Mantid::Geometry::Instrument_const_sptr;
using namespace Mantid::Geometry;
 
namespace Mantid::Crystal {
 
DECLARE_ALGORITHM(SetCrystalLocation)
 
void SetCrystalLocation::init() {
  declareProperty(std::make_unique<WorkspaceProperty<EventWorkspace>>("InputWorkspace", "", Direction::Input),
                  "Original event workspace");
  declareProperty(std::make_unique<WorkspaceProperty<EventWorkspace>>("OutputWorkspace", "", Direction::Output),
                  "Output event workspace with a modified sample position");
  declareProperty("NewX", 0.0, "New Absolute X position of crystal.");
  declareProperty("NewY", 0.0, "New Absolute Y position of crystal.");
  declareProperty("NewZ", 0.0, "New Absolute Z position of crystal.");
}
 
// simple algorithm that changes the sample position of the input
// event workspace.
void SetCrystalLocation::exec() {
  EventWorkspace_sptr events = getProperty("InputWorkspace");
  EventWorkspace_sptr outEvents = getProperty("OutputWorkspace");
  const double newX = getProperty("NewX");
  const double newY = getProperty("NewY");
  const double newZ = getProperty("NewZ");
  V3D newSamplePos = V3D(newX, newY, newZ);
  if (events != outEvents) {
    outEvents = events->clone();
  }
 
  auto &componentInfo = outEvents->mutableComponentInfo();
  CalibrationHelpers::adjustUpSampleAndSourcePositions(componentInfo.l1(), newSamplePos, componentInfo);
 
  setProperty("OutputWorkspace", outEvents);
} // exec
 
} // namespace Mantid::Crystal