ElectronID.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file ElectronID.h

#ifndef TRACKRECO_PHTRUTHCLUSTERING_H
#define TRACKRECO_PHTRUTHCLUSTERING_H

#include <fun4all/SubsysReco.h>

// rootcint barfs with this header so we need to hide it
#include <gsl/gsl_rng.h>


#include <string>             // for string
#include <vector>
#include <map>
#include <set>
#include <memory>

#include <TFile.h>
#include <TNtuple.h>
#include "TMVA/Tools.h"
#include "TMVA/Reader.h"
#include <TMVA/MethodCuts.h>

// forward declarations
class PHCompositeNode;
class SvtxTrackMap;
class SvtxTrack;
class TrackPidAssoc;
class PHG4TruthInfoContainer;
class PHG4Particle;

class SvtxTrack;

class ElectronID  : public SubsysReco
{
public:
  ElectronID(const std::string &name = "ElectronID", const std::string &filename = "_ElectronID.root");
  virtual ~ElectronID();

  int Init(PHCompositeNode *topNode);
  int InitRun(PHCompositeNode *topNode);
  int process_event(PHCompositeNode *topNode);
  int End(PHCompositeNode *topNode);

  void setEMOPcutlimits(float EMOPlowerlimit, float EMOPhigherlimit) { EMOP_lowerlimit = EMOPlowerlimit; EMOP_higherlimit = EMOPhigherlimit; }

  void setHinOEMcutlimit(float HinOEMhigherlimit) { HinOEM_higherlimit = HinOEMhigherlimit; }

  void setPtcutlimit(float Ptlowerlimit, float Pthigherlimit) { Pt_lowerlimit = Ptlowerlimit; Pt_higherlimit = Pthigherlimit; }

  void setHOPcutlimit(float HOPlowerlimit) { HOP_lowerlimit = HOPlowerlimit; }

  void setTrackcutlimits(int Nmvtxlowerlimit, int Ninttlowerlimit, int Ntpclowerlimit, float Nqualityhigherlimit) { 
     Nmvtx_lowerlimit = Nmvtxlowerlimit; 
     Nintt_lowerlimit = Ninttlowerlimit; 
     Ntpc_lowerlimit = Ntpclowerlimit;
     Nquality_higherlimit = Nqualityhigherlimit;
  }

  void setPROBcut(float tmp) {PROB_cut = tmp;}

  void set_output_ntuple(bool outputntuple) {output_ntuple = outputntuple;}

  void setBDTcut(int isuseBDT_p, int isuseBDT_n, float bdtcut_p, float bdtcut_n) {ISUSE_BDT_p= isuseBDT_p; ISUSE_BDT_n= isuseBDT_n; BDT_cut_p = bdtcut_p; BDT_cut_n = bdtcut_n;}


protected:
  bool output_ntuple;

  TFile* OutputNtupleFile;
  std::string OutputFileName;
  TNtuple* ntpBDTresponse; //write ntuple for BDTresponse
  TNtuple* ntpbeforecut; //write ntuple before any cuts
  TNtuple* ntpcutEMOP; //write ntuple with only EMOP cut
  TNtuple* ntpcutEMOP_HinOEM; //write ntuple with EMOP & HinOEM cuts
  TNtuple* ntpcutEMOP_HinOEM_Pt; //write ntuple with EMOP & HinOEM & Pt cuts
  TNtuple* ntpcutEMOP_HinOEM_Pt_read; //write ntuple with EMOP & HinOEM & Pt cuts in the situation of reading back the association map.
  TNtuple* ntpcutBDT_read; //write ntuple with BDT & Pt cuts in the situation of reading back the association map.

  TNtuple* ntpcutHOP; //write ntuple with only HOP cut

  int EventNumber;

private:
int GetNodes(PHCompositeNode *topNode);

PHG4Particle* findMCmatch(SvtxTrack* track, PHG4TruthInfoContainer* truth_container);

 TrackPidAssoc *_track_pid_assoc;
 SvtxTrackMap *_track_map;

  float EMOP_lowerlimit;
  float EMOP_higherlimit;

  float PROB_cut;

  float HinOEM_higherlimit;

  float Pt_lowerlimit;
  float Pt_higherlimit;

  float HOP_lowerlimit;

  int Nmvtx_lowerlimit;

  int Nintt_lowerlimit;

  int Ntpc_lowerlimit;

  float Nquality_higherlimit;

  float BDT_cut_p, BDT_cut_n;
  int ISUSE_BDT_p, ISUSE_BDT_n;//0 for no; 1 for yes

  unsigned int _nlayers_maps = 3;
  unsigned int _nlayers_intt = 4;
  unsigned int _nlayers_tpc = 48;


};

#endif