TForwardElectronIsEMSelector.h

Go to the documentation of this file.

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

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

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

  asg::AcceptData accept() const { return asg::AcceptData(&m_acceptInfo); }
 
  // used to calcIsEm. 
  // Note that it doesn't necessarily zero all the values that pass, so
  // make sure that the input iflag bits that you are concerned with start out
  // with 0.
  unsigned int calocuts_electrons(
    // eta
    float eta,
    // NPV
    float nvtx,
    // secondlambda
    float secondLambda,
    // lateral
    float lateral,
    // lateral
    float longitudinal,
    // fracMax
    float fracMax,
    // longitudinal
    float centerLambda,
    // secondR
    float secondR,
    // the starting isEM (should be 0 if new)
    unsigned int iflag) const;
 
  // used internally by calcIsEm, but left public because it can be useful for
  // users. Note that it doesn't necessarily zero all the values that pass, so
  // make sure that the input iflag bits that you are concerned with start out
  // with 0.

  // Public members (the cut values)

  unsigned int m_isEMMask;

  std::vector<float> m_cutBinEta_ForwardElectron;
  std::vector<float> m_cutVxp_ForwardElectron;
  std::vector<float> m_cutSECONDLAMBDA_ForwardElectron;
  std::vector<float> m_cutLATERAL_ForwardElectron;
  std::vector<float> m_cutLONGITUDINAL_ForwardElectron;
  std::vector<float> m_cutCELLMAXFRAC_ForwardElectron;
  std::vector<float> m_cutSECONDR_ForwardElectron;
  std::vector<float> m_cutCENTERLAMBDA_ForwardElectron;

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

  asg::AcceptInfo m_acceptInfo;
 
  // the cut positions

  int m_cutPositionBinEta_ForwardElectron;
 
  //      int m_cutPositionVxp_ForwardElectron;
 
  int m_cutPositionSECONDLAMBDA_ForwardElectron;
  int m_cutPositionLATERAL_ForwardElectron;
  int m_cutPositionLONGITUDINAL_ForwardElectron;
  int m_cutPositionCELLMAXFRAC_ForwardElectron;
  int m_cutPositionSECONDR_ForwardElectron;
  int m_cutPositionCENTERLAMBDA_ForwardElectron;
 
  // the cut names
 
  const std::string m_cutNameBinEta_ForwardElectron;
  const std::string m_cutNameVxp_ForwardElectron;
  const std::string m_cutNameSECONDLAMBDA_ForwardElectron;
  const std::string m_cutNameLATERAL_ForwardElectron;
  const std::string m_cutNameLONGITUDINAL_ForwardElectron;
  const std::string m_cutNameCELLMAXFRAC_ForwardElectron;
  const std::string m_cutNameSECONDR_ForwardElectron;
  const std::string m_cutNameCENTERLAMBDA_ForwardElectron;
 
}; // End: class definition
 
} // End: namespace Root
 
#endif