writeESD.h

Go to the documentation of this file.

#ifndef writeESD_h
#define writeESD_h 1

#include <XANADOO/XANAClusters/interface/XANACaloRecHit.h>

#include <CARF/G3Event/interface/G3EventProxy.h>
#include <CARF/Reco/interface/RecCollection.h>
#include <CARF/Reco/interface/RecQuery.h>

#include <TrackerReco/TkEvent/interface/TkEventPreselectorByTracks.h>
#include <Calorimetry/CaloDetector/interface/CellID.h>

#include <Utilities/Notification/interface/Observer.h>
#include <Utilities/UI/interface/SimpleConfigurable.h> 

#include <XANADOO/XANATracks/interface/XANAGSUtil.h>

#include "CommonReco/PatternTools/interface/RecTrack.h"

#include <string>
#include <iostream> 
#include <iomanip>  
#include <map>  

class TFile;
class TTree;
class TBranch;
class TClonesArray;
class XANAEsdEvent;
class XANAGeantEvent;
class XANAGeneratorEvent;
class XANAMemLog;
class XANAElectronTrack;
class IsoCalculator;

class EgammaBasicCluster;
class EgammaSuperCluster;
class EgammaEndcapCluster;
class EcalPlusHcalTower;
class TTrack;
class RecoVertex;
class RecMuon;
class ConcreteRecTrack;
class ElectronCandidate;
class PhotonCandidate;
class HighLevelTriggerResult;
class RecJet;
class RecMET;


class TrajectoryStateOnSurface;
class TrajectoryMeasurement;

class VertexTrackCompatibilityEstimator;
 // for GSF tracks
class RegionalTrackFinder;


class XANAEsdBuilder : public Observer<G3EventProxy *>, public Observer<G3SetUp *>
{

 public:
   
  // configuration
  // general
  SimpleConfigurable<bool> writeTree_;
  SimpleConfigurable<bool> memlog_;
  SimpleConfigurable<std::string> fileName_;
  SimpleConfigurable<std::string> treeName_;
  SimpleConfigurable<int> autosave_;
  SimpleConfigurable<bool> writeCaloRecHits_;
  SimpleConfigurable<int> compressionLevel_;
  SimpleConfigurable<int> bufferSize_;
  SimpleConfigurable<int> splitMode_;
  SimpleConfigurable<bool> writeTrackHits_;
  SimpleConfigurable<bool> writeAllTrackHits_;
  SimpleConfigurable<bool> writeEleTrackHits_;
  
  // algos options
  SimpleConfigurable<std::string> clusterizerName_;
  SimpleConfigurable<std::string> superClusterizerName_;
  SimpleConfigurable<std::string> endcapSuperClusterizerName_;
  SimpleConfigurable<std::string> hcalClusterizerName_;
  SimpleConfigurable<std::string> trackReconstructorName_;
  SimpleConfigurable<std::string> vertexReconstructorName_;
  SimpleConfigurable<std::string> muonReconstructorName_;
  SimpleConfigurable<std::string> electronReconstructorName_;
  SimpleConfigurable<bool> hZZVertexReconstructor_;
  
  // electron tracking
  SimpleConfigurable<std::string> gsfForwardFit_;
  SimpleConfigurable<std::string> electronTrackSeeding_;

  // electron candidate
  SimpleConfigurable<float> eCandidateEtaCut_;
  SimpleConfigurable<float> eCandidatePhiCut_;
  SimpleConfigurable<float> eCandidateEOverPMin_;
  SimpleConfigurable<float> eCandidateEOverPMax_;
   
  // jets +MET options 
  SimpleConfigurable<bool> jetsFromTrueInformation_;
  SimpleConfigurable<bool> metsFromTrueInformation_;

  // trigger
  SimpleConfigurable<std::string> l1Menu_;  
  SimpleConfigurable<std::string> hltMenu_;

 protected:
 
  typedef RecCollection<TTrack> Collection;

  Collection* theCollection;

  RecCollection<EgammaBasicCluster>* clusters;
  RecCollection<EgammaSuperCluster>* superClusters;
  RecCollection<EgammaEndcapCluster>* endcapSuperClusters;
  RecCollection<EcalPlusHcalTower>* towers;
  RecCollection<TTrack>* tracks;
  RecCollection<RecoVertex>* vertices;
  RecCollection<RecMuon>* muons;
  RecCollection<ElectronCandidate>* electrons;
  RecCollection<PhotonCandidate>* photons;
  RecCollection<ConcreteRecTrack>* eltracks;
  RecCollection<RecJet>* jets;
  RecCollection<RecMET>* mets;
  RecCollection<HighLevelTriggerResult>* hlt;

  TTree *xanaEsdTree_;
  TBranch *xanaEsdEventBranch_;
  TFile *xanaEsdFile_;
  XANAEsdEvent *xanaEsdEvent_;
 
  TBranch *xanaGenEventBranch_;
  XANAGeneratorEvent *xanaGenEvent_;
  TBranch *xanaGeantEventBranch_;
  XANAGeantEvent *xanaGeantEvent_;
  
  IsoCalculator *ciso_; 
  
  XANAMemLog *xanamemlog_;
  
  int nbClusters_, nbTracks_, nbElectrons_, nbPhotons_;
  
  map<CellID, XANACaloRecHit, less<CellID> > allCaloRecHits_;
  map<CellID, XANACaloRecHit, less<CellID> > clusterizedCaloRecHits_;
  vector<CellID> allCells_;
  vector<CellID> clusterizedCells_;

  vector<RecTrack> electronTracks;
  // for GSF tracks
   RegionalTrackFinder* theRegionalTrackFinder;

   // For vertex studies
   VertexTrackCompatibilityEstimator * theVertexTrackCompatibilityEstimator;

 public:
  
  XANAEsdBuilder(); 
  ~XANAEsdBuilder();  

 protected:

  virtual void upDate( G3EventProxy * );
  void upDate( G3SetUp * );
 
  void setHeader( G3EventProxy * );
  void setCaloRecHits( G3EventProxy * );
  void setEmClusters( G3EventProxy * );
  void setSuperClusters( G3EventProxy * );
  void setEndcapClusters( G3EventProxy * );
  void setHadClusters( G3EventProxy * );
  void setTracks( G3EventProxy * );
  void setVertices( G3EventProxy * );
  void setJets( G3EventProxy * );
  void setMet( G3EventProxy * );
  void setMuons( G3EventProxy * );
  void setElectrons( G3EventProxy * );
  void setPhotons( G3EventProxy * );
  void setElectronsGSF( G3EventProxy * );
  void setElectronsGSFPixelMatch( G3EventProxy * );
  void setElectronsZZVertex( G3EventProxy * );
  virtual void setMcTruth( G3EventProxy * );        
  void setTriggerInfo( G3EventProxy * );        
  void setAllTrackHits( G3EventProxy * );        
  void setElectronTracks(G3EventProxy * );

   // secondary methods (called from the others)
  void setEleTrackHits(std::vector<TrajectoryMeasurement>, XANAElectronTrack *);

  void writeTree();
  
  std::string getAlgoName(int);
  
} ;

class CompareRecTrackPt
{
 public:
  bool operator() (RecTrack* P1, RecTrack* P2){
    return P1->impactPointState().globalMomentum().perp() > P2->impactPointState().globalMomentum().perp();
  }
};
 
#endif

---