ComponentInfo.h

Go to the documentation of this file.

// Mantid Repository : https://github.com/mantidproject/mantid
//
// Copyright © 2017 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 "MantidBeamline/ComponentType.h"
#include "MantidGeometry/DllConfig.h"
#include "MantidGeometry/Instrument/ComponentInfoIterator.h"
#include "MantidGeometry/Instrument/SolidAngleParams.h"
#include "MantidGeometry/Objects/BoundingBox.h"
#include "MantidTypes/Core/DateAndTime.h"
#include <memory>
#include <unordered_map>
#include <vector>
 
namespace Mantid {
 
namespace Kernel {
class Quat;
class V3D;
} // namespace Kernel
 
namespace Geometry {
class IComponent;
class IObject;
} // namespace Geometry
 
namespace Beamline {
class ComponentInfo;
}
 
namespace Geometry {
class Instrument;
 
class MANTID_GEOMETRY_DLL ComponentInfo {
private:
  std::unique_ptr<Beamline::ComponentInfo> m_componentInfo;
  std::shared_ptr<const std::vector<Geometry::IComponent *>> m_componentIds;
  std::shared_ptr<const std::unordered_map<Geometry::IComponent const *, size_t>> m_compIDToIndex;
 
  std::shared_ptr<std::vector<std::shared_ptr<const Geometry::IObject>>> m_shapes;
 
  BoundingBox componentBoundingBox(const size_t index, const BoundingBox *reference,
                                   const bool excludeMonitors = false) const;
 
  ComponentInfo(const ComponentInfo &other);
  void growBoundingBoxAsRectuangularBank(size_t index, const Geometry::BoundingBox *reference,
                                         Geometry::BoundingBox &mutableBB, const bool excludeMonitors = false) const;
  void growBoundingBoxAsOutline(size_t index, const Geometry::BoundingBox *reference, Geometry::BoundingBox &mutableBB,
                                const bool excludeMonitors = false) const;
 
public:
  struct QuadrilateralComponent {
    size_t topLeft;
    size_t bottomLeft;
    size_t bottomRight;
    size_t topRight;
    size_t nX;
    size_t nY;
  };
 
  ComponentInfo(std::unique_ptr<Beamline::ComponentInfo> componentInfo,
                std::shared_ptr<const std::vector<Mantid::Geometry::IComponent *>> componentIds,
                std::shared_ptr<const std::unordered_map<Geometry::IComponent const *, size_t>> componentIdToIndexMap,
                std::shared_ptr<std::vector<std::shared_ptr<const Geometry::IObject>>> shapes);
  ~ComponentInfo();
  ComponentInfo &operator=(const ComponentInfo &) = delete;
  std::unique_ptr<ComponentInfo> cloneWithoutDetectorInfo() const;
  std::vector<size_t> detectorsInSubtree(size_t componentIndex) const;
  std::vector<size_t> componentsInSubtree(size_t componentIndex) const;
  const std::vector<size_t> &children(size_t componentIndex) const;
  size_t size() const;
  size_t getMemorySize() const;
  QuadrilateralComponent quadrilateralComponent(const size_t componentIndex) const;
  size_t indexOf(Geometry::IComponent const *id) const;
  size_t indexOfAny(const std::string &name) const;
  bool uniqueName(const std::string &name) const;
  bool isDetector(const size_t componentIndex) const;
  Kernel::V3D position(const size_t componentIndex) const;
  Kernel::V3D position(const std::pair<size_t, size_t> &index) const;
  Kernel::Quat rotation(const size_t componentIndex) const;
  Kernel::Quat rotation(const std::pair<size_t, size_t> &index) const;
  Kernel::V3D relativePosition(const size_t componentIndex) const;
  Kernel::Quat relativeRotation(const size_t componentIndex) const;
  void setPosition(size_t componentIndex, const Kernel::V3D &newPosition);
  void setRotation(size_t componentIndex, const Kernel::Quat &newRotation);
  void setPosition(const std::pair<size_t, size_t> &index, const Kernel::V3D &newPosition);
  void setRotation(const std::pair<size_t, size_t> &index, const Kernel::Quat &newRotation);
  void scaleComponent(const size_t componentIndex, const Kernel::V3D &newScaling);
  void scaleComponent(const std::pair<size_t, size_t> &index, const Kernel::V3D &newScaling);
  size_t parent(const size_t componentIndex) const;
  bool hasParent(const size_t componentIndex) const;
  bool hasDetectorInfo() const;
  Kernel::V3D sourcePosition() const;
  Kernel::V3D samplePosition() const;
  bool hasSource() const;
  bool hasEquivalentSource(const ComponentInfo &other) const;
  bool hasSample() const;
  bool hasEquivalentSample(const ComponentInfo &other) const;
  bool hasDetectors(const size_t componentIndex) const;
  size_t source() const;
  size_t sample() const;
  double l1() const;
  Kernel::V3D scaleFactor(const size_t componentIndex) const;
  const std::string &name(const size_t componentIndex) const;
  void setScaleFactor(const size_t componentIndex, const Kernel::V3D &scaleFactor);
  size_t root() const;
 
  const IComponent *componentID(const size_t componentIndex) const { return (*m_componentIds)[componentIndex]; }
  bool hasValidShape(const size_t componentIndex) const;
 
  const Geometry::IObject &shape(const size_t componentIndex) const;
  std::unordered_map<std::shared_ptr<const Geometry::IObject>, std::vector<size_t>> shapeToComponentIndices() const;
 
  double solidAngle(const size_t componentIndex, const Geometry::SolidAngleParams &params) const;
  BoundingBox boundingBox(const size_t componentIndex, const BoundingBox *reference = nullptr,
                          const bool excludeMonitors = false) const;
  Beamline::ComponentType componentType(const size_t componentIndex) const;
  void setScanInterval(const std::pair<Types::Core::DateAndTime, Types::Core::DateAndTime> &interval);
  size_t scanCount() const;
  void merge(const ComponentInfo &other);
 
  ComponentInfoIterator<ComponentInfo> begin();
  ComponentInfoIterator<ComponentInfo> end();
  const ComponentInfoIterator<const ComponentInfo> cbegin();
  const ComponentInfoIterator<const ComponentInfo> cend();
 
  size_t findBankParent(size_t index, const std::string &bankPart) const;
 
  friend class Instrument;
};
 
using ComponentInfoIt = ComponentInfoIterator<ComponentInfo>;
using ComponentInfoConstIt = ComponentInfoIterator<const ComponentInfo>;
 
} // namespace Geometry
} // namespace Mantid