RaytracerBase.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#include <top/dbobjects/TOPGeoModule.h>
#include <top/dbobjects/TOPGeoBarSegment.h>
#include <top/dbobjects/TOPGeoPrism.h>
#include <vector>
 
 
namespace Belle2 {
  namespace TOP {
 
    class RaytracerBase {
 
    public:
      enum EGeometry {
        c_Unified = 0,   
        c_Segmented = 1  
      };
 
      enum EOptics {
        c_SemiLinear = 0, 
        c_Exact = 1       
      };
 
      struct BarSegment {
        double A = 0;  
        double B = 0;  
        double zL = 0; 
        double zR = 0; 
        double reflectivity = 0; 
        double sigmaAlpha = 0; 
        BarSegment()
        {}
 
        BarSegment(const TOPGeoBarSegment& bar, double zLeft);
 
        explicit BarSegment(const TOPGeoModule& module);
      };
 
      struct Mirror {
        double xc = 0; 
        double yc = 0; 
        double zc = 0; 
        double R = 0;  
        double zb = 0; 
        double reflectivity = 0; 
        Mirror()
        {}
 
        explicit Mirror(const TOPGeoModule& module);
      };
 
      struct Prism {
        double A = 0;     
        double B = 0;     
        double yUp = 0;   
        double yDown = 0; 
        double zL = 0;    
        double zR = 0;    
        double zFlat = 0; 
        double zD = 0;    
        double slope = 0; 
        int k0 = 0; 
        std::vector<TOPGeoPrism::UnfoldedWindow> unfoldedWindows; 
        Prism()
        {}
 
        explicit Prism(const TOPGeoModule& module);
      };
 
 
      explicit RaytracerBase(int moduleID, EGeometry geometry = c_Unified, EOptics optics = c_SemiLinear);
 
      int getModuleID() const {return m_moduleID;}
 
      EGeometry getGeometry() const {return m_geometry;}
 
      EOptics getOptics() const {return m_optics;}
 
      const std::vector<BarSegment>& getBars() const {return m_bars;}
 
      const Mirror& getMirror() const {return m_mirror;}
 
      const Prism& getPrism() const {return m_prism;}
 
      void setMirrorCenter(double xc, double yc);
 
    protected:
 
      int m_moduleID = 0; 
      EGeometry m_geometry = c_Unified; 
      EOptics m_optics = c_SemiLinear;  
      std::vector<BarSegment> m_bars; 
      Mirror m_mirror; 
      Prism m_prism; 
    };
 
  } // namespace TOP
} // namespace Belle2