GeometryData.h

Go to the documentation of this file.

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2015  Belle II Collaboration                              *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Kirill Chilikin                                          *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
/* KLM headers. */
#include <klm/dbobjects/eklm/EKLMGeometry.h>
 
/* Belle 2 headers. */
#include <framework/database/IntervalOfValidity.h>
#include <framework/gearbox/GearDir.h>
 
/* CLHEP headers. */
#include <CLHEP/Geometry/Transform3D.h>
 
namespace Belle2 {
  namespace EKLM {
 
    class GeometryData : public EKLMGeometry {
 
    public:
 
      enum DataSource {
 
        c_Gearbox,
 
        c_Database,
 
      };
 
      static const GeometryData& Instance(
        enum DataSource dataSource = c_Database,
        const GearDir* gearDir = nullptr);
 
      void saveToDatabase(const IntervalOfValidity& iov) const;
 
      double getStripLength(int strip) const
      {
        return m_StripPosition[strip - 1].getLength();
      }
 
      int getNStripsDifferentLength() const
      {
        return m_nStripDifferent;
      }
 
      int getStripLengthIndex(int positionIndex) const
      {
        return m_StripAllToLen[positionIndex];
      }
 
      int getStripPositionIndex(int lengthIndex) const
      {
        return m_StripLenToAll[lengthIndex];
      }
 
      bool hitInEKLM(double z) const;
 
      void getSectionTransform(HepGeom::Transform3D* t, int n) const;
 
      void getLayerTransform(HepGeom::Transform3D* t, int n) const;
 
      void getSectorTransform(HepGeom::Transform3D* t, int n) const;
 
      void getPlaneTransform(HepGeom::Transform3D* t, int n) const;
 
      void getStripTransform(HepGeom::Transform3D* t, int n) const;
 
      void getSheetTransform(HepGeom::Transform3D* t, int n) const;
 
    private:
 
      explicit GeometryData(enum DataSource dataSource, const GearDir* gearDir);
 
      ~GeometryData();
 
      void readEndcapStructureGeometry(const GearDir& gd);
 
      void initializeFromGearbox(const GearDir* gearDir);
 
      void initializeFromDatabase();
 
      void calculateSectorSupportGeometry();
 
      void fillStripIndexArrays();
 
      void readXMLDataStrips(const GearDir& gd);
 
      void calculateShieldGeometry();
 
      EKLMGeometry* m_Geometry;
 
      int m_nStripDifferent;
 
      int* m_StripAllToLen;
 
      int* m_StripLenToAll;
 
      double m_MinZForward;
 
      double m_MaxZBackward;
 
    };
 
  }
 
}