FastRaytracer.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/reconstruction_cpp/RaytracerBase.h>
#include <top/reconstruction_cpp/PhotonState.h>
#include <vector>
 
 
namespace Belle2 {
  namespace TOP {

    class FastRaytracer : public RaytracerBase {
 
    public:

      explicit FastRaytracer(int moduleID, EGeometry geometry = c_Unified, EOptics optics = c_SemiLinear):
        RaytracerBase(moduleID, geometry, optics)
      {};

      void clear() const
      {
        m_photonStates.clear();
        m_extraStates.clear();
        m_status = false;
        m_Nxm = 0;
        m_Nxb = 0;
        m_Nxe = 0;
        m_Nym = 0;
        m_Nyb = 0;
        m_Nye = 0;
      }

      void propagate(const PhotonState& photon, bool averaging = false) const;

      const std::vector<PhotonState>& getPhotonStates() const {return m_photonStates;}

      const std::vector<PhotonState>& getExtraStates() const {return m_extraStates;}

      bool getPropagationStatus() const {return m_status;}

      bool getTotalReflStatus(double cosTotal) const;

      double getPropagationLen() const;

      double getPropagationLenDelta() const;

      double getXD() const;

      double getYD() const;

      double getInPlaneYD() const;

      double getZD() const;

      double getYB() const;

      int getNx() const;

      int getNxm() const {return m_Nxm;}

      int getNxb() const {return m_Nxb;}

      int getNxe() const {return m_Nxe;}

      int getNy() const;

      int getNym() const {return m_Nym;}

      int getNyb() const {return m_Nyb;}

      int getNye() const {return m_Nye;}

      int getNys() const;
 
 
    private:
 
      mutable std::vector<PhotonState> m_photonStates; 
      mutable std::vector<PhotonState> m_extraStates;  
      mutable bool m_status = false; 
      mutable int m_Nxm = 0; 
      mutable int m_Nxb = 0; 
      mutable int m_Nxe = 0; 
      mutable int m_Nym = 0; 
      mutable int m_Nyb = 0; 
      mutable int m_Nye = 0; 
 
    };
 
  } // namespace TOP
} // namespace Belle2