testEGChargeIDSelector.cxx File Reference

#include <memory>
#include <cstdlib>
#include <TFile.h>
#include <TError.h>
#include <TString.h>
#include "xAODEgamma/ElectronContainer.h"
#include "xAODEgamma/PhotonContainer.h"
#include "xAODEgamma/Electron.h"
#include "xAODEgamma/Photon.h"
#include "ElectronPhotonSelectorTools/AsgPhotonIsEMSelector.h"
#include "ElectronPhotonSelectorTools/AsgForwardElectronIsEMSelector.h"
#include "ElectronPhotonSelectorTools/AsgElectronIsEMSelector.h"
#include "ElectronPhotonSelectorTools/AsgElectronLikelihoodTool.h"
#include "ElectronPhotonSelectorTools/AsgElectronChargeIDSelectorTool.h"
#include "AsgMessaging/AsgMessaging.h"
#include "xAODCore/tools/IOStats.h"
#include "xAODCore/tools/ReadStats.h"

Go to the source code of this file.

Macros
#define	DUMMY_MSG(lvl, ARG)   {dummymsg.msg(lvl)<<ARG<<endmsg ;}
#define	MSG_DEBUG(ARG)   {DUMMY_MSG(MSG::DEBUG, ARG);}
#define	MSG_INFO(ARG)   {DUMMY_MSG(MSG::INFO, ARG);}
#define	MSG_WARNING(ARG)   {DUMMY_MSG(MSG::WARNING, ARG);}
#define	MSG_ERROR(ARG)   {DUMMY_MSG(MSG::ERROR, ARG);}
#define	MSG_FATAL(ARG)   {DUMMY_MSG(MSG::FATAL, ARG);}
#define	MSG_ABORT(ARG)   {DUMMY_MSG(MSG::FATAL, ARG); std::abort();}
#define	CHECK(ARG)

Functions
int	main (int argc, char *argv[])

Macro Definition Documentation

CHECK

#define CHECK

(

ARG

)

Value:

  do {                                                                  \
    const bool result = ARG;                                            \
    if( ! result ) {                                                    \
      MSG_ERROR("FAILED  to execute" <<#ARG);                           \
      return EXIT_FAILURE;                                              \
    }                                                                   \
  } while( false )

Definition at line 53 of file testEGChargeIDSelector.cxx.

DUMMY_MSG

#define DUMMY_MSG	(		lvl,
			ARG
	)		{dummymsg.msg(lvl)<<ARG<<endmsg ;}

Definition at line 46 of file testEGChargeIDSelector.cxx.

MSG_ABORT

#define MSG_ABORT

(

ARG

)

{DUMMY_MSG(MSG::FATAL, ARG); std::abort();}

Definition at line 52 of file testEGChargeIDSelector.cxx.

MSG_DEBUG

#define MSG_DEBUG

(

ARG

)

{DUMMY_MSG(MSG::DEBUG, ARG);}

Definition at line 47 of file testEGChargeIDSelector.cxx.

MSG_ERROR

#define MSG_ERROR

(

ARG

)

{DUMMY_MSG(MSG::ERROR, ARG);}

Definition at line 50 of file testEGChargeIDSelector.cxx.

MSG_FATAL

#define MSG_FATAL

(

ARG

)

{DUMMY_MSG(MSG::FATAL, ARG);}

Definition at line 51 of file testEGChargeIDSelector.cxx.

MSG_INFO

#define MSG_INFO

(

ARG

)

{DUMMY_MSG(MSG::INFO, ARG);}

Definition at line 48 of file testEGChargeIDSelector.cxx.

MSG_WARNING

#define MSG_WARNING

(

ARG

)

{DUMMY_MSG(MSG::WARNING, ARG);}

Definition at line 49 of file testEGChargeIDSelector.cxx.

Function Documentation

main()

int main	(	int	argc,
		char *	argv[]
	)

Definition at line 63 of file testEGChargeIDSelector.cxx.

                                   {
 
   // The application's name:
   const char* APP_NAME = argv[ 0 ];
 
   MSG::Level mylevel=MSG::DEBUG;
   dummymsg.msg().setLevel(mylevel);
   dummymsg.msg().setName(APP_NAME);
   // Check if we received a file name:
   if( argc < 2 ) {
     MSG_ERROR("No file name received!" );
     MSG_ERROR( "  Usage: %s [xAOD file name]");
     return EXIT_FAILURE;
   }
   
   // Initialise the application:
   CHECK( xAOD::Init( APP_NAME ) );
 
   // Open the input file:
   const TString fileName = argv[ 1 ];
   Info( APP_NAME, "Opening file: %s", fileName.Data() );
   std::unique_ptr< TFile > ifile( TFile::Open( fileName, "READ" ) );
   CHECK( ifile.get() );
 
  // Check if we want to process Electron or Photon
   bool isElectron = true;   
   if( argc < 4 ) {
     Info (APP_NAME, "By default looking to Electron" );
   } else {  
    int  argv1 = atoi(argv[ 3 ]);
    if(argv1 == 0)   isElectron = false; 
    if(isElectron)  Info( APP_NAME, "We are lookign to Electron-ID: %i", isElectron );
    else  Info( APP_NAME, "We are looking to Photon-ID" );
   }
   // Create a TEvent object (persistent store)
 
   //For 2.4.5 testing
   xAOD::TEvent pers( xAOD::TEvent::kAthenaAccess );
   //For 2.3
   //xAOD::TEvent pers( xAOD::TEvent::kBranchAccess );
 
   // Create a TStore object (transient store)
   xAOD::TStore trans;
   CHECK( pers.readFrom( ifile.get() ) );
   //
   MSG_INFO("Number of events in the file: "<< pers.getEntries());
 
   // Decide how many events to run over:
   Long64_t entries = pers.getEntries();
   if( argc > 2 ) {
      const Long64_t e = atoll( argv[ 2 ] );
      if( e < entries ) {
         entries = e;
      }
   }
 
 
   if (isElectron) {
   //tight ECIDS
   std::unique_ptr<AsgElectronChargeIDSelectorTool> m_electronECIDS = std::make_unique<AsgElectronChargeIDSelectorTool> ("tightECIDS");
   std::string trainingfile    = std::string(std::getenv("ROOTCOREBIN")) + "/data/ElectronPhotonSelectorTools/ECIDS_20161125for2017Moriond.root";
   CHECK(m_electronECIDS->setProperty("TrainingFile", trainingfile));
   CHECK(m_electronECIDS->setProperty("CutOnBDT", 0));
   m_electronECIDS->msg().setLevel(mylevel);
   CHECK(m_electronECIDS->initialize());
   
   //Tight Likelihood electron
   std::unique_ptr<AsgElectronLikelihoodTool> m_TightLH = std::make_unique<AsgElectronLikelihoodTool> ("TightLH");
   CHECK(m_TightLH->setProperty("WorkingPoint", "TightLHElectron"));
   m_TightLH->msg().setLevel(mylevel);
   CHECK(m_TightLH->initialize());
   
   // Loop over the events:
   for( Long64_t entry = 0; entry < entries; ++entry ) {
     
     // Tell the object which entry to look at:
     pers.getEntry( entry );
     MSG_INFO("============================");
     MSG_INFO("Event: " <<entry);
 
     const xAOD::ElectronContainer* electrons;
     CHECK(pers.retrieve(electrons, "Electrons"));
 
     unsigned int counter=0;
     for (const xAOD::Electron* el : *electrons) {
       MSG_INFO("---------------------------");
       MSG_INFO("Electron: " << counter);
       MSG_INFO("Electron LH Tight accept result: " <<m_TightLH->accept(el));
       //       MSG_INFO("Electron stored LH Tight: "  << el->passSelection("LHTight") );
       MSG_INFO("Electron Cut ECIDS accept result: " <<m_electronECIDS->accept(el));
       MSG_INFO("Electron stored Tight: "  << el->passSelection("Tight") );
       ++counter;
     }
   } // loop entries
   
   } // is electron
   else {
     //Tight cut based photon
     std::unique_ptr<AsgPhotonIsEMSelector> m_photonTightIsEMSelector = std::make_unique<AsgPhotonIsEMSelector> ("photonTightIsEMSelector");
     CHECK(m_photonTightIsEMSelector->setProperty("WorkingPoint", "TightPhoton"));
     m_photonTightIsEMSelector->msg().setLevel(mylevel);
     CHECK(m_photonTightIsEMSelector->initialize());
     
     // Loop over the events:
     for( Long64_t entry = 0; entry < entries; ++entry ) {
       
       // Tell the object which entry to look at:
       pers.getEntry( entry );
       MSG_INFO("============================");
       MSG_INFO("Event: " <<entry);
       
       const xAOD::PhotonContainer* photons;
       CHECK(pers.retrieve(photons, "Photons"));
       unsigned int counter=0;
       for (const xAOD::Photon* ph : *photons) {
     if(ph->author() != xAOD::EgammaParameters::AuthorCaloTopo35){
       MSG_INFO("---------------------------");
       MSG_INFO("Photon: " << counter);
       MSG_INFO("Photon Tight accept result: " <<m_photonTightIsEMSelector->accept(ph));
       ++counter;       
     }
       }
     }// loop entries
   }
 
 
   xAOD::IOStats::instance().stats().printSmartSlimmingBranchList();
   return 0;
}