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 <Math/Vector3D.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);
    bool GetTaub2b3(double);
    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;
    }
    void set3DBhabhaVetoInTrackThetaRegion(const std::vector<int>& i3DBhabhaVetoInTrackThetaRegion)
    {
      m_3DBhabhaVetoInTrackThetaRegion = i3DBhabhaVetoInTrackThetaRegion;
    }
    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;
    }
    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;
    }
    void setTaub2b3Cut(const std::vector<int>& iTaub2b3AngleCut,
                       const double iTaub2b3EtotCut,
                       const double iTaub2b3CLEb2bCut,
                       const double iTaub2b3CLELowCut,
                       const double iTaub2b3CLEHighCut)
    {
      m_taub2b3AngleCut     = iTaub2b3AngleCut;
      m_taub2b3EtotCut      = iTaub2b3EtotCut;
      m_taub2b3CLEb2bCut    = iTaub2b3CLEb2bCut;
      m_taub2b3CLELowCut    = iTaub2b3CLELowCut;
      m_taub2b3CLEHighCut   = iTaub2b3CLEHighCut;
    }
    int get3DBhabhaVetoInTrackFlag(void)
    {
      return m_3DBhabhaVetoInTrackFlag;
    }
    int get3DBhabhaVetoClusterTCId(int cl_idx)
    {
      if (cl_idx < 0 || cl_idx > 1 || m_3DBhabhaVetoClusterTCIds.size() != 2) {
        return -10;
      }
      return m_3DBhabhaVetoClusterTCIds[cl_idx];
    }
    int get3DBhabhaVetoClusterThetaId(int cl_idx)
    {
      if (cl_idx < 0 || cl_idx > 1 || m_3DBhabhaVetoClusterThetaIds.size() != 2) {
        return -10;
      }
      return m_3DBhabhaVetoClusterThetaIds[cl_idx];
    }
    double get3DBhabhaVetoClusterEnergy(int cl_idx)
    {
      if (cl_idx < 0 || cl_idx > 1 || m_3DBhabhaVetoClusterEnergies.size() != 2) {
        return -10;
      }
      return m_3DBhabhaVetoClusterEnergies[cl_idx];
    }
    double get3DBhabhaVetoClusterTiming(int cl_idx)
    {
      if (cl_idx < 0 || cl_idx > 1 || m_3DBhabhaVetoClusterTimings.size() != 2) {
        return -1000;
      }
      return m_3DBhabhaVetoClusterTimings[cl_idx];
    }
    int get3DBhabhaSelectionThetaFlag(void)
    {
      return m_3DBhabhaSelectionThetaFlag;
    }
    int get3DBhabhaSelectionClusterTCId(int cl_idx)
    {
      if (cl_idx < 0 || cl_idx > 1 || m_3DBhabhaSelectionClusterTCIds.size() != 2) {
        return -10;
      }
      return m_3DBhabhaSelectionClusterTCIds[cl_idx];
    }
    int get3DBhabhaSelectionClusterThetaId(int cl_idx)
    {
      if (cl_idx < 0 || cl_idx > 1 || m_3DBhabhaSelectionClusterThetaIds.size() != 2) {
        return -10;
      }
      return m_3DBhabhaSelectionClusterThetaIds[cl_idx];
    }
    double get3DBhabhaSelectionClusterEnergy(int cl_idx)
    {
      if (cl_idx < 0 || cl_idx > 1 || m_3DBhabhaSelectionClusterEnergies.size() != 2) {
        return -10;
      }
      return m_3DBhabhaSelectionClusterEnergies[cl_idx];
    }
    double get3DBhabhaSelectionClusterTiming(int cl_idx)
    {
      if (cl_idx < 0 || cl_idx > 1 || m_3DBhabhaSelectionClusterTimings.size() != 2) {
        return -1000;
      }
      return m_3DBhabhaSelectionClusterTimings[cl_idx];
    }
    int getTaub2bAngleFlag(void) { return m_taub2bAngleFlag; }
    int getTaub2bEtotFlag(void) { return m_taub2bEtotFlag; }
    int getTaub2bClusterEFlag(void) { return m_taub2bClusterEFlag; }
 
  private:
    TrgEclMapping* _TCMap = nullptr;
    TrgEclDataBase* _database = nullptr;
 
    void get2CLETP(int, int, int&, int&, int&, int&);
 
    std::vector<double> BhabhaComb;
    std::vector<double> MaxTCId;
    std::vector<double> MaxTCThetaId;
    std::vector<double> ClusterEnergy;
    std::vector<double> ClusterTiming;
    std::vector<ROOT::Math::XYZVector> 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;
    std::vector<double> m_3DBhabhaAddAngleCut;
    std::vector<int> m_taub2bAngleCut;
    double m_taub2bEtotCut;
    double m_taub2bClusterECut1;
    double m_taub2bClusterECut2;
    int m_taub2bAngleFlag;
    int m_taub2bEtotFlag;
    int m_taub2bClusterEFlag;
    double m_taub2b2EtotCut;
    std::vector<int> m_taub2b2AngleCut;
    double m_taub2b2CLEEndcapCut;
    double m_taub2b2CLECut;
 
    std::vector<int> m_taub2b3AngleCut;
    double m_taub2b3EtotCut;
    double m_taub2b3CLEb2bCut;
    double m_taub2b3CLELowCut;
    double m_taub2b3CLEHighCut;
 
    int                 m_3DBhabhaVetoInTrackFlag;
    std::vector<int>    m_3DBhabhaVetoClusterTCIds;
    std::vector<int>    m_3DBhabhaVetoClusterThetaIds;
    std::vector<double> m_3DBhabhaVetoClusterEnergies;
    std::vector<double> m_3DBhabhaVetoClusterTimings;
    std::vector<int> m_3DBhabhaVetoInTrackThetaRegion;
    int m_3DBhabhaSelectionThetaFlag = std::numeric_limits<int>::quiet_NaN();
    std::vector<int>    m_3DBhabhaSelectionClusterTCIds;
    std::vector<int>    m_3DBhabhaSelectionClusterThetaIds;
    std::vector<double> m_3DBhabhaSelectionClusterEnergies;
    std::vector<double> m_3DBhabhaSelectionClusterTimings;
  };
//
//
//
} // namespace Belle2