TrgEclBhabha.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 <vector>
 
#include "trg/ecl/TrgEclMapping.h"
#include "trg/ecl/TrgEclDataBase.h"
 
#include "TVector3.h"
//
//
//
namespace Belle2 {
//
//
//
  //
  //
  class TrgEclBhabha {
 
  public:
    TrgEclBhabha();    
 
    virtual ~TrgEclBhabha();
 
    TrgEclBhabha(TrgEclBhabha&) = delete;
 
    TrgEclBhabha& operator=(TrgEclBhabha&) = delete;
 
  public:
    bool GetBhabha00(std::vector<double>);
    bool GetBhabha01();
    bool GetBhabha02();
    bool Getmumu();
    bool GetTaub2b(double);
    bool GetTaub2b2(double);
    // Additional Bhabha veto
    int GetBhabhaAddition(void);
    std::vector<double> GetBhabhaComb() {return BhabhaComb ;}
    void save(int);
    void set2DBhabhaThreshold(const std::vector<double>& i2DBhabhaThresholdFWD,
                              const std::vector<double>& i2DBhabhaThresholdBWD)
    {
      _2DBhabhaThresholdFWD = i2DBhabhaThresholdFWD;
      _2DBhabhaThresholdBWD = i2DBhabhaThresholdBWD;
    }
    void set3DBhabhaSelectionThreshold(const std::vector<double>& i3DBhabhaSelectionThreshold)
    {
      _3DBhabhaSelectionThreshold = i3DBhabhaSelectionThreshold;
    };
    void set3DBhabhaVetoThreshold(const std::vector<double>& i3DBhabhaVetoThreshold)
    {
      _3DBhabhaVetoThreshold = i3DBhabhaVetoThreshold;
    };
    void set3DBhabhaSelectionAngle(const std::vector<double>& i3DBhabhaSelectionAngle)
    {
      _3DBhabhaSelectionAngle = i3DBhabhaSelectionAngle;
    };
    void set3DBhabhaVetoAngle(const std::vector<double>& i3DBhabhaVetoAngle)
    {
      _3DBhabhaVetoAngle = i3DBhabhaVetoAngle;
    };
    void setmumuThreshold(int mumuThreshold) {_mumuThreshold = mumuThreshold; }
    void setmumuAngle(const std::vector<double>&  imumuAngle) {_mumuAngle = imumuAngle; }
    void set3DBhabhaAddAngleCut(const std::vector<double>&  i3DBhabhaAddAngleCut)
    {
      m_3DBhabhaAddAngleCut = i3DBhabhaAddAngleCut;
    }
    // set ThetaID (low and high) for 3DBhabhaVetoInTrack
    void set3DBhabhaVetoInTrackThetaRegion(const std::vector<int>& i3DBhabhaVetoInTrackThetaRegion)
    {
      m_3DBhabhaVetoInTrackThetaRegion = i3DBhabhaVetoInTrackThetaRegion;
    }
    // set 2 Cluster angle selection for tau 1x1 decay
    // [0], [1] for low and high of dphi, [2], [3] for low and high of Theta Sum
    void setTaub2bAngleCut(const std::vector<int>& iTaub2bAngleCut)
    {
      m_taub2bAngleCut = iTaub2bAngleCut;
    }
    //
    void setTaub2bEtotCut(const double iTaub2bEtotCut)
    {
      m_taub2bEtotCut = iTaub2bEtotCut;
    }
    //
    void setTaub2bClusterECut(const double iTaub2bClusterECut1,
                              const double iTaub2bClusterECut2)
    {
      m_taub2bClusterECut1 = iTaub2bClusterECut1;
      m_taub2bClusterECut2 = iTaub2bClusterECut2;
    }
    // set taub2b2 cut
    void setTaub2b2Cut(const std::vector<int>& iTaub2b2AngleCut,
                       const double iTaub2b2EtotCut,
                       const double iTaub2b2CLEEndcapCut,
                       const double iTaub2b2CLECut)
    {
      m_taub2b2AngleCut     = iTaub2b2AngleCut;
      m_taub2b2EtotCut      = iTaub2b2EtotCut;
      m_taub2b2CLEEndcapCut = iTaub2b2CLEEndcapCut;
      m_taub2b2CLECut       = iTaub2b2CLECut;
    }
    // get trigger bit of flag(1bit) whether two clusters statisfy 3D Bhabha veto
    // are in CDCTRG region in theta (="InTrack") or not
    int get3DBhabhaVetoInTrackFlag(void)
    {
      return m_3DBhabhaVetoInTrackFlag;
    }
    // get each TCID(most energetic TC in a cluster) of two clusters of 3D Bhabha veto
    int get3DBhabhaVetoClusterTCId(int cl_idx)
    {
      if (cl_idx < 0 || cl_idx > 1 || m_3DBhabhaVetoClusterTCIds.size() != 2) {
        return -10;
      }
      return m_3DBhabhaVetoClusterTCIds[cl_idx];
    }
    // get each TC theta ID(most energetic TC in a cluster) of
    // two clusters of 3D Bhabha veto
    int get3DBhabhaVetoClusterThetaId(int cl_idx)
    {
      if (cl_idx < 0 || cl_idx > 1 || m_3DBhabhaVetoClusterThetaIds.size() != 2) {
        return -10;
      }
      return m_3DBhabhaVetoClusterThetaIds[cl_idx];
    }
    // get each cluster energy of two clusters of 3D Bhabha veto (GeV)
    double get3DBhabhaVetoClusterEnergy(int cl_idx)
    {
      if (cl_idx < 0 || cl_idx > 1 || m_3DBhabhaVetoClusterEnergies.size() != 2) {
        return -10;
      }
      return m_3DBhabhaVetoClusterEnergies[cl_idx];
    }
    // get each cluster timing of two clusters of 3D Bhabha veto (ns)
    double get3DBhabhaVetoClusterTiming(int cl_idx)
    {
      if (cl_idx < 0 || cl_idx > 1 || m_3DBhabhaVetoClusterTimings.size() != 2) {
        return -1000;
      }
      return m_3DBhabhaVetoClusterTimings[cl_idx];
    }
    // get trigger bit(2bits) of flag which shows theta position of clusters
    // of 3DBhabha Selection.
    // flag=0 : one of clusters goes to ThetaID=1
    // flag=1 : one of clusters goes to ThetaID=2
    // flag=2 : one of clusters goes to ThetaID=3
    // flag=3 : none of clusters fly to ThetaID=1-3
    // Based on this flag, pre-scale is applied on GDL to have flat entry of
    // Bhabha event in theta for calibration purpose
    int get3DBhabhaSelectionThetaFlag(void)
    {
      return m_3DBhabhaSelectionThetaFlag;
    }
    // get each TCID(most energetic TC in a cluster) of two clusters of 3D Bhabha selection
    int get3DBhabhaSelectionClusterTCId(int cl_idx)
    {
      if (cl_idx < 0 || cl_idx > 1 || m_3DBhabhaSelectionClusterTCIds.size() != 2) {
        return -10;
      }
      return m_3DBhabhaSelectionClusterTCIds[cl_idx];
    }
    // get each TC theta ID(most energetic TC in a cluster) of
    // two clusters of 3D Bhabha selection
    int get3DBhabhaSelectionClusterThetaId(int cl_idx)
    {
      if (cl_idx < 0 || cl_idx > 1 || m_3DBhabhaSelectionClusterThetaIds.size() != 2) {
        return -10;
      }
      return m_3DBhabhaSelectionClusterThetaIds[cl_idx];
    }
    // get each cluster energy of two clusters of 3D Bhabha selection (GeV)
    double get3DBhabhaSelectionClusterEnergy(int cl_idx)
    {
      if (cl_idx < 0 || cl_idx > 1 || m_3DBhabhaSelectionClusterEnergies.size() != 2) {
        return -10;
      }
      return m_3DBhabhaSelectionClusterEnergies[cl_idx];
    }
    // get each cluster timing of two clusters of 3D Bhabha selection (ns)
    double get3DBhabhaSelectionClusterTiming(int cl_idx)
    {
      if (cl_idx < 0 || cl_idx > 1 || m_3DBhabhaSelectionClusterTimings.size() != 2) {
        return -1000;
      }
      return m_3DBhabhaSelectionClusterTimings[cl_idx];
    }
    // get taub2b 2 cluster angle cut flag
    int getTaub2bAngleFlag(void) { return m_taub2bAngleFlag; }
    // get total energy(TC theta id=1-17) flag for taub2b
    int getTaub2bEtotFlag(void) { return m_taub2bEtotFlag; }
    // taub2b Cluster energy flag
    int getTaub2bClusterEFlag(void) { return m_taub2bClusterEFlag; }
 
  private:
    TrgEclMapping* _TCMap = nullptr;
    // cppcheck-suppress unsafeClassCanLeak
    TrgEclDataBase* _database = nullptr;
 
    void get2CLETP(int, int, int&, int&, int&, int&);
 
    std::vector<double> BhabhaComb;
    std::vector<double> MaxTCId;
    std::vector<double> ClusterEnergy;
    std::vector<double> ClusterTiming;
    std::vector<TVector3> ClusterPosition;
    std::vector<double> _2DBhabhaThresholdFWD;
    std::vector<double> _2DBhabhaThresholdBWD;
    std::vector<double> _3DBhabhaSelectionThreshold;
    std::vector<double> _3DBhabhaVetoThreshold;
    std::vector<double> _3DBhabhaSelectionAngle;
    std::vector<double> _3DBhabhaVetoAngle;
    double _mumuThreshold;
    std::vector<double> _mumuAngle;
    // Angle selection of additional Bhabha veto in CM frame
    std::vector<double> m_3DBhabhaAddAngleCut;
    // taub2b 2 Cluster angle cut (degree)
    std::vector<int> m_taub2bAngleCut;
    // taub2b total energy(TC theta id=1-17) cut (GeV)
    double m_taub2bEtotCut;
    // taub2b Cluster energy cut (GeV)
    double m_taub2bClusterECut1;
    double m_taub2bClusterECut2;
    // taub2b 2 cluster angle cut flag
    int m_taub2bAngleFlag;
    // taub2b total energy(TC theta id=1-17) flag
    int m_taub2bEtotFlag;
    // taub2b Cluster energy flag
    int m_taub2bClusterEFlag;
    // taub2b2 total energy cut (GeV)
    double m_taub2b2EtotCut;
    // taub2b2 two Cluster angle cut (degree)
    std::vector<int> m_taub2b2AngleCut;
    // taub2b2 cluster energy cut for endcap cluster (GeV)
    double m_taub2b2CLEEndcapCut;
    // taub2b2 cluster energy cut (GeV)
    double m_taub2b2CLECut;
    // trigger bit of flag(1bit) whether two clusters statisfy 3D Bhabha veto
    // are in CDCTRG region in theta (="InTrack") or not
    int                 m_3DBhabhaVetoInTrackFlag;
    // TCIDs of two clusters of 3D Bhabha veto
    std::vector<int>    m_3DBhabhaVetoClusterTCIds;
    // ThetaIds of two clusters of 3D Bhabha veto
    std::vector<int>    m_3DBhabhaVetoClusterThetaIds;
    // Energies of two clusters of 3D Bhabha veto (GeV)
    std::vector<double> m_3DBhabhaVetoClusterEnergies;
    // Timings of two clusters of 3D Bhabha veto (ns)
    std::vector<double> m_3DBhabhaVetoClusterTimings;
    // theta region(low and high) of 3D Bhbabha veto InTrack
    std::vector<int> m_3DBhabhaVetoInTrackThetaRegion;
    // flag which shows theta position of clusters of 3DBhabha Selection.
    int m_3DBhabhaSelectionThetaFlag = std::numeric_limits<int>::quiet_NaN();
    // TCIDs of two clusters of 3D Bhabha selection
    std::vector<int>    m_3DBhabhaSelectionClusterTCIds;
    // ThetaIDs of two clusters used for 3D Bhabha selection
    std::vector<int>    m_3DBhabhaSelectionClusterThetaIds;
    // Energies of two clusters used for 3D Bhabha selection (GeV)
    std::vector<double> m_3DBhabhaSelectionClusterEnergies;
    // Timings of two clusters used for 3D Bhabha selection (ns)
    std::vector<double> m_3DBhabhaSelectionClusterTimings;
  };
//
//
//
} // namespace Belle2