TPhotonIsEMSelector.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
 
// Dear emacs, this is -*-c++-*-
 
#ifndef __TPHOTONISEMSELECTOR__
#define __TPHOTONISEMSELECTOR__
 

 
 
// Include the return object and the base class
#include "PATCore/AcceptInfo.h"
#include "PATCore/AcceptData.h"
#include "ElectronPhotonSelectorTools/egammaPIDdefs.h"
#include "AsgMessaging/AsgMessaging.h"
#include "AsgMessaging/StatusCode.h"
 
#include <vector>
#include <string>
 
 
class AsgPhotonIsEMSelector;
 
namespace Root {
  class TPhotonIsEMSelector : public asg::AsgMessaging {
 
    friend class ::AsgPhotonIsEMSelector;
 
 
  public:
    TPhotonIsEMSelector(const char *name = "TPhotonIsEMSelector");

    ~TPhotonIsEMSelector();
 
    // Main methods
    StatusCode initialize();

    asg::AcceptData accept(
      // eta position in second sampling
      float eta2,
      // transverse energy in calorimeter (using eta position in second sampling)
      double et,
      // transverse energy in 1st scintillator of hadronic calorimeter/ET
      float Rhad1,
      // transverse energy in hadronic calorimeter/ET
      float Rhad,
      // E(7*7) in 2nd sampling
      float e277,
      // E(3*7)/E(7*7) in 2nd sampling
      float Reta,
      // E(3*3)/E(3*7) in 2nd sampling
      float Rphi,
      // shower width in 2nd sampling
      float weta2c,
      // fraction of energy reconstructed in strips
      float f1,
      // (E of 1st max in strips-E of 2nd max)/(E of 1st max+E of 2nd max)
      float Eratio,
      // E(2nd max)-E(min) in strips
      float DeltaE,
      //  E of 2nd max between max and min in strips
      // float emax2,
      // shower width in 3 strips in 1st sampling
      float weta1c,
      // total shower width in strips
      float wtot,
      // E(+/-3)-E(+/-1)/E(+/-1)
      float fracm,
      // fraction of energy reconstructed in the 3rd sampling
      float f3,
      // E/p
      double ep,
      // is it a conversion
      bool isConversion,
      // pileup
      float mu) ;
 

    asg::AcceptData accept() const { return asg::AcceptData(&m_acceptInfo); }
 
    // calculate the isEM. (Used internally by accept)
    unsigned int calcIsEm(
      // eta position in second sampling
      float eta2,
      // transverse energy in calorimeter (using eta position in second sampling)
      double et,
      // transverse energy in 1st scintillator of hadronic calorimeter/ET
      float Rhad1,
      // transverse energy in hadronic calorimeter/ET
      float Rhad,
      // E(7*7) in 2nd sampling
      float e277,
      // E(3*7)/E(7*7) in 2nd sampling
      float Reta,
      // E(3*3)E(3*7) in 2nd sampling
      float Rphi,
      // shower width in 2nd sampling
      float weta2c,
      // fraction of energy reconstructed in strips
      float f1,
      // (E of 1st max in strips-E of 2nd max)/(E of 1st max+E of 2nd max)
      float Eratio,
      // E(2nd max)-E(min) in strips
      float DeltaE,
      // shower width in 3 strips in 1st sampling
      //  E of 2nd max between max and min in strips
      // shower width in 3 strips in 1st sampling
      float weta1c,
      // total shower width in strips
      float wtot,
      // E(+/-3)-E(+/-1)/E(+/-1)
      float fracm,
      // fraction of energy reconstructed in the 3rd sampling
      float f3,
      // E/p
      double ep,
      // is it a conversion
      bool isConversion,
      // pileup
      float mu) const;

    unsigned int calocuts_photonsConverted(
      // eta position in second sampling
      float eta2,
      // transverse energy in calorimeter
      double et,
      // hadronic leakage ratios
      float Rhad1,
      float Rhad,
      // E(7*7) in 2nd sampling
      float e277,
      // ratios
      float Reta,
      float Rphi,
      // shower width in 2nd sampling
      float weta2c,
      // fraction of energy reconstructed in strips
      float f1,
      // (Emax1-Emax2)/(Emax1+Emax2)
      float Eratio,
      // difference of energy between max and min
      float DeltaE,
      // parametrization of E(2nd max)
      //float deltaemax2,
      // shower width in 3 strips in 1st sampling
      float weta1c,
      // total shower width in strips
      float wtot,
      // E(+/-3)-E(+/-1)/E(+/-1)
      float fracm,
      // fraction of energy reconstructed in the 3rd sampling
      float f3,
      // E/p
      double ep,
      //
      unsigned int iflag,
      // pileup
      float mu) const;

    unsigned int calocuts_photonsNonConverted(
      // eta position in second sampling
      float eta2,
      // transverse energy in calorimeter
      double et,
      // hadronic leakage ratios
      float Rhad1,
      float Rhad,
      // E(7*7) in 2nd sampling
      float e277,
      // ratios
      float Reta,
      float Rphi,
      // shower width in 2nd sampling
      float weta2c,
      // fraction of energy reconstructed in strips
      float f1,
      // (Emax1-Emax2)/(Emax1+Emax2)
      float Eratio,
      // difference of energy between max and min
      float DeltaE,
      // parametrization of E(2nd max)
      //float deltaemax2,
      // shower width in 3 strips in 1st sampling
      float weta1c,
      // total shower width in strips
      float wtot,
      // E(+/-3)-E(+/-1)/E(+/-1)
      float fracm,
      // fraction of energy reconstructed in the 3rd sampling
      float f3,
      //
      unsigned int iflag,
      // pileup
      float mu) const;

    // Public members (the cut values)

    unsigned int m_isEMMask;
 

    bool m_forceConvertedPhotonPID;
    bool m_forceNonConvertedPhotonPID;
 
 
    //
    // selection for non-converted photons
    //
    std::vector<float> m_cutBinEta_photonsNonConverted;
    std::vector<float> m_cutBinEnergy_photonsNonConverted;
    std::vector<float> m_cutBinMu_photonsNonConverted;
    std::vector<float> m_e277_photonsNonConverted;
    std::vector<float> m_cutHadLeakage_photonsNonConverted;
    std::vector<float> m_Reta37_photonsNonConverted;
    std::vector<float> m_Rphi33_photonsNonConverted;
    std::vector<float> m_weta2_photonsNonConverted;

    std::vector<float> m_cutBinEtaStrips_photonsNonConverted;
    std::vector<float> m_cutBinEnergyStrips_photonsNonConverted;
    std::vector<float> m_cutBinMuStrips_photonsNonConverted;
    std::vector<float> m_f1_photonsNonConverted;
    // std::vector<float> emax2r_photonsNonConverted;
    std::vector<float> m_deltae_photonsNonConverted;
    std::vector<float> m_DEmaxs1_photonsNonConverted;
    std::vector<float> m_wtot_photonsNonConverted;
    std::vector<float> m_fracm_photonsNonConverted;
    std::vector<float> m_w1_photonsNonConverted;
    std::vector<float> m_cutF3_photonsNonConverted;
 
    //
    // selection for converted photons
    //
    std::vector<float> m_cutBinEta_photonsConverted;
    std::vector<float> m_cutBinEnergy_photonsConverted;
    std::vector<float> m_cutBinMu_photonsConverted;
    std::vector<float> m_e277_photonsConverted;
    std::vector<float> m_cutHadLeakage_photonsConverted;
    std::vector<float> m_Reta37_photonsConverted;
    std::vector<float> m_Rphi33_photonsConverted;
    std::vector<float> m_weta2_photonsConverted;

    std::vector<float> m_cutBinEtaStrips_photonsConverted;
    std::vector<float> m_cutBinEnergyStrips_photonsConverted;
    std::vector<float> m_cutBinMuStrips_photonsConverted;
    std::vector<float> m_f1_photonsConverted;
    // std::vector<float> emax2r_photonsConverted;
    std::vector<float> m_deltae_photonsConverted;
    std::vector<float> m_DEmaxs1_photonsConverted;
    std::vector<float> m_wtot_photonsConverted;
    std::vector<float> m_fracm_photonsConverted;
    std::vector<float> m_w1_photonsConverted;
    std::vector<float> m_cutminEp_photonsConverted;
    std::vector<float> m_cutmaxEp_photonsConverted;
    std::vector<float> m_cutF3_photonsConverted;

    const asg::AcceptInfo &getAcceptInfo() const { return m_acceptInfo; }
 
    // Private members
  private:
 
    // would ideally be protected: only to be used by ARASelector
    asg::AcceptData fillAccept(unsigned int isEM) const;
 
    template<typename T>
    bool checkVar(const std::vector <T> &vec, int choice) const;

    asg::AcceptInfo m_acceptInfo;
 
    // the cut positions

    int m_cutPositionClusterEtaRange_Photon;
    int m_cutPositionClusterBackEnergyFraction_Photon;
    int m_cutPositionClusterHadronicLeakage_Photon;
    int m_cutPositionClusterMiddleEnergy_Photon;
    int m_cutPositionClusterMiddleEratio37_Photon;
    int m_cutPositionClusterMiddleEratio33_Photon;
    int m_cutPositionClusterMiddleWidth_Photon;
    int m_cutPositionClusterStripsEratio_Photon;
    // int m_cutPositionClusterStripsDeltaEmax2_Photon;
    int m_cutPositionClusterStripsDeltaE_Photon;
    int m_cutPositionClusterStripsWtot_Photon;
    int m_cutPositionClusterStripsFracm_Photon;
    int m_cutPositionClusterStripsWeta1c_Photon;
    int m_cutPositionClusterStripsDEmaxs1_Photon;
    int m_cutPositionTrackMatchEoverP_Photon;
    int m_cutPositionAmbiguityResolution_Photon;
    int m_cutPositionIsolation_Photon;
    int m_cutPositionClusterIsolation_Photon;
    int m_cutPositionTrackIsolation_Photon;
 
    // the cut names

    const std::string m_cutNameClusterEtaRange_Photon;
    const std::string m_cutNameClusterBackEnergyFraction_Photon;
    const std::string m_cutNameClusterHadronicLeakage_Photon;
    const std::string m_cutNameClusterMiddleEnergy_Photon;
    const std::string m_cutNameClusterMiddleEratio37_Photon;
    const std::string m_cutNameClusterMiddleEratio33_Photon;
    const std::string m_cutNameClusterMiddleWidth_Photon;
    const std::string m_cutNameClusterStripsEratio_Photon;
    // const std::string m_cutNameClusterStripsDeltaEmax2_Photon;
    const std::string m_cutNameClusterStripsDeltaE_Photon;
    const std::string m_cutNameClusterStripsWtot_Photon;
    const std::string m_cutNameClusterStripsFracm_Photon;
    const std::string m_cutNameClusterStripsWeta1c_Photon;
    const std::string m_cutNameClusterStripsDEmaxs1_Photon;
    const std::string m_cutNameTrackMatchEoverP_Photon;
    const std::string m_cutNameAmbiguityResolution_Photon;
    const std::string m_cutNameIsolation_Photon;
    const std::string m_cutNameClusterIsolation_Photon;
    const std::string m_cutNameTrackIsolation_Photon;
 
  }; // End: class definition
 
} // End: namespace Root
 
#endif