src/TrapezoidSegmentation.cpp

+// Trapezoid Segmentation for GlueX type barrel calorimeter
+// Author: M. Zurek, C. Peng (ANL)
+// Date: 06/25/2021
+
+#include "DD4hep/Factories.h"
+#include "DD4hep/detail/SegmentationsInterna.h"
+#include "DDSegmentation/Segmentation.h"
+
+
+using namespace dd4hep::DDSegmentation;
+
+namespace athena::seg {
+  class TrapezoidGridYZ: public Segmentation {
+    public:
+      // Default constructor used by derived classes passing the encoding string
+      TrapezoidGridYZ(const std::string& cellEncoding = "")
+        : Segmentation(cellEncoding)
+      {
+        // define type and description
+        _type = "TrapezoidGridYZ";
+        _description = "Trapezoid segmentation in the local YZ-plane";
+        registerParameters();
+      }
+
+      // Default constructor used by derived classes passing an existing decoder
+      TrapezoidGridYZ(const BitFieldCoder* decoder)
+        : Segmentation(decoder)
+      {
+        // define type and description
+        _type = "TrapezoidGridYZ";
+        _description = "Trapezoid segmentation in the local YZ-plane";
+        registerParameters();
+      }
+
+      // Destructor
+      virtual ~TrapezoidGridYZ() {}
+
+      void registerParameters()
+      {
+        // @TODO: register all necessary parameters
+        // mandatory
+        registerParameter("grid_size_y", "Cell size in Y", _gridSizeY, 1., SegmentationParameter::LengthUnit);
+        registerParameter("grid_size_z", "Cell size in H", _gridSizeZ, 1., SegmentationParameter::LengthUnit);
+        // optional
+        registerIdentifier("identifier_y", "Cell ID identifier for Y", _yId, "y");
+        registerIdentifier("identifier_z", "Cell ID identifier for Z", _zId, "z");
+      }
+
+      // override
+      virtual Vector3D position(const CellID& cellID) const
+      {
+        // @TODO
+        /* PolarGridRPhi
+        Vector3D cellPosition;
+        double R =   binToPosition(_decoder->get(cID,_rId),   _gridSizeR,   _offsetR);
+        double phi = binToPosition(_decoder->get(cID,_phiId), _gridSizePhi, _offsetPhi);
+
+        cellPosition.X = R * cos(phi);
+        cellPosition.Y = R * sin(phi);
+
+        return cellPosition;
+        */
+        return Vector3D();
+      }
+
+      // override
+      virtual CellID cellID(const Vector3D& localPosition, const Vector3D& globalPosition, const VolumeID& volumeID) const
+      {
+        // @TODO
+        /* PolarGridRPhi
+        double phi = atan2(localPosition.Y,localPosition.X);
+        double R = sqrt( localPosition.X * localPosition.X + localPosition.Y * localPosition.Y );
+        CellID cID = vID ;
+        _decoder->set(cID,_rId  , positionToBin(R, _gridSizeR, _offsetR));
+        _decoder->set(cID,_phiId, positionToBin(phi, _gridSizePhi, _offsetPhi));
+        */
+        return volumeID;
+      }
+
+      // override
+      virtual std::vector<double> cellDimensions(const CellID& cellID) const
+      {
+        // @TODO
+        /* PolarGridRPhi
+        const double rPhiSize = binToPosition(_decoder->get(cID,_rId), _gridSizeR, _offsetR)*_gridSizePhi;
+        return {_gridSizeR, rPhiSize};
+        */
+        return std::vector<double>{0., 0., 0.};
+      }
+
+
+      // public methods to get member value
+      double gridSizeY() const { return _gridSizeY; }
+      double gridSizeZ() const { return _gridSizeZ; }
+      const std::string& fieldNameY() const { return _yId; }
+      const std::string& fieldNameZ() const { return _zId; }
+      // public methods to set member value
+      void setGridSizeY(double cellSize) { _gridSizeY = cellSize; }
+      void setGridSizeZ(double cellSize) { _gridSizeZ = cellSize; }
+      void setFieldNameY(const std::string& fieldName) { _yId = fieldName; }
+      void setFieldNameZ(const std::string& fieldName) { _zId = fieldName; }
+
+    protected:
+      // @TODO: add parameters accordingly
+      double _gridSizeY;
+      double _gridSizeZ;
+      std::string _yId;
+      std::string _zId;
+
+  }; // class TrapezoidGridYZ
+
+} // namespace athena::seg
+
+DECLARE_SEGMENTATION(TrapezoidGridYZ, new dd4hep::SegmentationWrapper<athena::seg::TrapezoidGridYZ>)
+