TrgEclMaster.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 <string>
#include <vector>
 
#include <framework/core/Module.h>
 
#include "trg/ecl/TrgEclCluster.h"
#include "trg/ecl/TrgEclTiming.h"
#include "trg/ecl/TrgEclBhabha.h"
#include "trg/ecl/TrgEclBeamBKG.h"
#include "trg/ecl/TrgEclMapping.h"
//
//
//
namespace Belle2 {
//
//
//
  class TrgEclMaster {
 
  public:
 
    static TrgEclMaster* getTrgEclMaster(void);
 
    TrgEclMaster(void);
 
    virtual ~TrgEclMaster();
 
    TrgEclMaster(TrgEclMaster&) = delete;
 
    TrgEclMaster& operator=(TrgEclMaster&) = delete;
 
  public:
 
    void initialize(int);
    void simulate01(int);
    void simulate02(int);
 
  public:
 
    std::string name(void) const;
 
    std::string version(void) const;
    //  void simulate(int);
    //int getECLtoGDL(void) { return bitECLtoGDL; }
    void setRS(std::vector<int>, std::vector<double>,
               std::vector<double>&, std::vector<std::vector<double>>&);
    //    void getEventTiming(int option);
    void setClusterMethod(int cluster) {m_Clustering = cluster;}
    void setClusterLimit(int limit) {m_ClusterLimit = limit;}
    void setBhabhaMethod(int bhabha) {m_Bhabha = bhabha;}
    void setEventTimingMethod(int EventTiming) {m_EventTiming = EventTiming;}
    void setTimeWindow(int timewindow) {m_TimeWindow = timewindow;}
    void setOverlapWindow(int overlapwindow) {m_OverlapWindow = overlapwindow;}
    void setNofTopTC(int noftoptc) {m_NofTopTC = noftoptc;}
    void makeLowMultiTriggerBit(std::vector<int>, std::vector<double>);
 
    /*
    void makeTriggerBit(int, int, int, int, double, int, int,
                        std::vector<int>, int, int, int, int,
                        int, int, int, int, int, int, int,
                        int, int, int, int);
    */
    void makeTriggerBit(int, int, int, int, double, int, int,
                        std::vector<int>, int, int, int, int,
                        int, int, int, int, int, int, int,
                        int, int, int, int, int);//one int parameter was added for taub2b3
 
    double setTotalEnergy(std::vector<double>);
    int getTriggerbit(int i) {return m_Triggerbit[i];}
    int getLowmultibit() {return m_Lowmultibit;}
    void set2DBhabhaThreshold(const std::vector<double>& i2DBhabhaThresholdFWD,
                              const std::vector<double>& i2DBhabhaThresholdBWD)
    {
      m_2DBhabhaThresholdFWD = i2DBhabhaThresholdFWD;
      m_2DBhabhaThresholdBWD = i2DBhabhaThresholdBWD;
    }
    void set3DBhabhaSelectionThreshold(const std::vector<double>& i3DBhabhaSelectionThreshold)
    {
      m_3DBhabhaSelectionThreshold = i3DBhabhaSelectionThreshold;
    };
    void set3DBhabhaVetoThreshold(const std::vector<double>& i3DBhabhaVetoThreshold)
    {
      m_3DBhabhaVetoThreshold = i3DBhabhaVetoThreshold;
    };
 
    void set3DBhabhaSelectionAngle(const std::vector<double>& i3DBhabhaSelectionAngle)
    {
      m_3DBhabhaSelectionAngle = i3DBhabhaSelectionAngle;
    };
    void set3DBhabhaVetoAngle(const std::vector<double>& i3DBhabhaVetoAngle)
    {
      m_3DBhabhaVetoAngle = i3DBhabhaVetoAngle;
    };
    void setmumuThreshold(int mumuThreshold) {m_mumuThreshold = mumuThreshold; }
    void setmumuAngle(const std::vector<double>& imumuAngle)
    {
      m_mumuAngle = imumuAngle;
    }
    void set3DBhabhaAddAngleCut(const std::vector<double>&  i3DBhabhaAddAngleCut)
    {
      m_3DBhabhaAddAngleCut = i3DBhabhaAddAngleCut;
    }
    void setTaub2bAngleCut(const std::vector<int>& itaub2bAngleCut)
    {
      m_taub2bAngleCut = itaub2bAngleCut;
    }
    void setTaub2bEtotCut(double itaub2bEtotCut)
    {
      m_taub2bEtotCut = itaub2bEtotCut;
    }
    void setTaub2bClusterECut(double itaub2bClusterECut1,
                              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;
    }
    void setn300MeVClusterThreshold(int n300MeVCluster)
    {
      m_n300MeVCluster = n300MeVCluster;
    }
    void setECLBurstThreshold(int ECLBurstThreshold)
    {
      m_ECLBurstThreshold = (double) ECLBurstThreshold;
    }
    void setTotalEnergyThreshold(const std::vector<double>& iTotalEnergy)
    {
      m_TotalEnergy = iTotalEnergy;
    }
    void setLowMultiplicityThreshold(const std::vector<double>& iLowMultiThreshold)
    {
      m_LowMultiThreshold = iLowMultiThreshold;
    }
    void set3DBhabhaVetoInTrackThetaRegion(const std::vector<int>& i3DBhabhaVetoInTrackThetaRegion)
    {
      m_3DBhabhaVetoInTrackThetaRegion = i3DBhabhaVetoInTrackThetaRegion;
    }
    void setEventTimingQualityThresholds(const std::vector<double>& iEventTimingQualityThresholds)
    {
      m_EventTimingQualityThresholds = iEventTimingQualityThresholds;
    }
 
  private:
 
    std::vector<std::vector<double>>  m_TCEnergy;
    std::vector<std::vector<double>>  m_TCTiming;
    std::vector<std::vector<int>>  m_TCBeamBkgTag;
 
    std::vector<int>  m_HitTCId;
    std::vector<double>  m_TCHitEnergy;
    std::vector<double>  m_TCHitTiming;
    std::vector<int>  m_TCHitBeamBkgTag;
 
    std::vector< std::vector<double>>  m_PhiRingSum;
    std::vector<std::vector<std::vector<double>>> m_ThetaRingSum;
 
    //   std::vector<double> ClusterEnergy;
    //    std::vector<double>  ClusterTiming;
 
    double m_TimeWindow;
    double m_OverlapWindow;
 
    int m_Clustering;
    int m_Bhabha;
    int m_EventTiming;
    int m_NofTopTC;
    int m_ClusterLimit;
    int m_Triggerbit[4];
    int m_Lowmultibit;
    int m_PrescaleFactor;
    int m_PrescaleCounter;
 
    std::vector<double> m_2DBhabhaThresholdFWD;
    std::vector<double> m_2DBhabhaThresholdBWD;
    std::vector<double> m_3DBhabhaSelectionThreshold;
    std::vector<double> m_3DBhabhaVetoThreshold;
    std::vector<double> m_3DBhabhaSelectionAngle;
    std::vector<double> m_3DBhabhaVetoAngle;
    double m_mumuThreshold;
    std::vector<double> m_mumuAngle;
    std::vector<double> m_3DBhabhaAddAngleCut;
    std::vector<int> m_taub2bAngleCut;
    double m_taub2bEtotCut;
    double m_taub2bClusterECut1;
    double m_taub2bClusterECut2;
    std::vector<int> m_taub2b2AngleCut;
    double m_taub2b2EtotCut;
    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_n300MeVCluster;
    double m_ECLBurstThreshold;
    std::vector<double> m_TotalEnergy;
    std::vector<double> m_LowMultiThreshold;
    std::vector<int> m_3DBhabhaVetoInTrackThetaRegion;
    std::vector<double> m_EventTimingQualityThresholds;
 
    TrgEclMapping* m_obj_map;
    TrgEclCluster* m_obj_cluster;
    TrgEclTiming* m_obj_timing;
    TrgEclBhabha* m_obj_bhabha;
    TrgEclBeamBKG* m_obj_beambkg;
    TrgEclDataBase* m_obj_database;
 
  };
//
//
//
} // namespace Belle2