#include <MySub/LhcTrackAnalyzer/src/LhcTrackAnalyzer.cc>

Inheritance diagram for LhcTrackAnalyzer:

Public Member Functions
	LhcTrackAnalyzer (const edm::ParameterSet &)

	~LhcTrackAnalyzer ()

Public Member Functions inherited from edm::EDAnalyzer
	EDAnalyzer ()

std::string	workerType () const

virtual	~EDAnalyzer ()

Private Member Functions
virtual void	analyze (const edm::Event &, const edm::EventSetup &)

virtual void	beginJob ()

virtual void	endJob ()

void	SetVarToZero ()

Private Attributes
int	charge_ [nMaxtracks_]

double	chi2_ [nMaxtracks_]

double	chi2ndof_ [nMaxtracks_]

bool	debug_

double	dxy_ [nMaxtracks_]

double	dz_ [nMaxtracks_]

double	eta_ [nMaxtracks_]

int	event_

std::string	filename_

bool	goodbx_

bool	goodvtx_

int	isHighPurity_ [nMaxtracks_]

int	nTracks_

double	phi_ [nMaxtracks_]

double	pt_ [nMaxtracks_]

edm::InputTag	PVtxCollectionTag_

double	qoverp_ [nMaxtracks_]

TFile *	rootFile_

TTree *	rootTree_

int	run_

edm::InputTag	TrackCollectionTag_

int	trkAlgo_ [nMaxtracks_]

int	trkQuality_ [nMaxtracks_]

int	validhits_ [nMaxtracks_][7]

double	xPCA_ [nMaxtracks_]

double	yPCA_ [nMaxtracks_]

double	zPCA_ [nMaxtracks_]

Static Private Attributes
static const int	nMaxtracks_ = 3000

Additional Inherited Members
Public Types inherited from edm::EDAnalyzer
typedef EDAnalyzer	ModuleType

typedef WorkerT< EDAnalyzer >	WorkerType

Static Public Member Functions inherited from edm::EDAnalyzer
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &)

Protected Member Functions inherited from edm::EDAnalyzer
CurrentProcessingContext const *	currentContext () const

Detailed Description

Originally written by M.Musich Expanded by A. Bonato

Description: <one line="" class="" summary>="">

Implementation: <Notes on="" implementation>="">

Definition at line 61 of file LhcTrackAnalyzer.h.

Constructor & Destructor Documentation

LhcTrackAnalyzer::LhcTrackAnalyzer ( const edm::ParameterSet & iConfig )

explicit

Definition at line 63 of file LhcTrackAnalyzer.cc.

References debug_, filename_, edm::ParameterSet::getParameter(), PVtxCollectionTag_, and TrackCollectionTag_.

 {
   //now do what ever initialization is needed
   debug_    = iConfig.getParameter<bool>       ("Debug");  
   TrackCollectionTag_      = iConfig.getParameter<edm::InputTag>("TrackCollectionTag");  
   PVtxCollectionTag_      = iConfig.getParameter<edm::InputTag>("PVtxCollectionTag");  
   filename_ = iConfig.getParameter<std::string>("OutputFileName");
 }

LhcTrackAnalyzer::~LhcTrackAnalyzer ( )

Definition at line 74 of file LhcTrackAnalyzer.cc.

 {
  
    // do anything here that needs to be done at desctruction time
    // (e.g. close files, deallocate resources etc.)
 
 }

Member Function Documentation

void LhcTrackAnalyzer::analyze	(	const edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

privatevirtual

Implements edm::EDAnalyzer.

Definition at line 89 of file LhcTrackAnalyzer.cc.

 {
   using namespace edm;
   using namespace reco;
   using namespace std;
 
   //=======================================================
   // Initialize Root-tuple variables
   //=======================================================
 
   SetVarToZero();
   
   //=======================================================
   // Retrieve the Track information
   //=======================================================
   
   Handle< TrackCollection>  trackCollectionHandle;
   iEvent.getByLabel(TrackCollectionTag_, trackCollectionHandle);
   Handle<VertexCollection>  vertexCollectionHandle;
   iEvent.getByLabel(PVtxCollectionTag_, vertexCollectionHandle);
   for(VertexCollection::const_iterator vtx = vertexCollectionHandle->begin();vtx!=vertexCollectionHandle->end(); ++vtx)
     {
       if(vtx==vertexCollectionHandle->begin()){
     if(vtx->isFake())goodvtx_=false;
     else goodvtx_=true;
       }
       else break;
     }
 
 
 
   goodbx_=true;
   // int bx = iEvent.bunchCrossing();
   //if (bx==51 || bx==2724) goodbx_=true;
   
 
  
   run_=iEvent.id().run();
   event_=iEvent.id().event();
  
   if(debug_)
     cout<<"LhcTrackAnalyzer::analyze() looping over "<< trackCollectionHandle->size()<< "tracks." << endl;    
   
   // unsigned int i = 0;   
   for(TrackCollection::const_iterator track = trackCollectionHandle->begin(); track!= trackCollectionHandle->end(); ++track)
     {
       if ( nTracks_ >= nMaxtracks_ ) {
     std::cout << " LhcTrackAnalyzer::analyze() : Warning - Run "<< run_<<" Event "<< event_<<"\tNumber of tracks: " <<  trackCollectionHandle->size()<< " , greater than " << nMaxtracks_ << std::endl;
       continue;
      
     }
       pt_[nTracks_]       = track->pt();
       eta_[nTracks_]      = track->eta();
       phi_[nTracks_]      = track->phi();
       chi2_[nTracks_]     = track->chi2();
       chi2ndof_[nTracks_] = track->normalizedChi2();
       charge_[nTracks_]   = track->charge();
       qoverp_[nTracks_]   = track->qoverp();
       dz_[nTracks_]       = track->dz();
       dxy_[nTracks_]      = track->dxy();
       xPCA_[nTracks_]     = track->vertex().x();
       yPCA_[nTracks_]     = track->vertex().y();
       zPCA_[nTracks_]     = track->vertex().z(); 
       validhits_[nTracks_][0]=track->numberOfValidHits();
       validhits_[nTracks_][1]=track->hitPattern().numberOfValidPixelBarrelHits();
       validhits_[nTracks_][2]=track->hitPattern().numberOfValidPixelEndcapHits();
       validhits_[nTracks_][3]=track->hitPattern().numberOfValidStripTIBHits();
       validhits_[nTracks_][4]=track->hitPattern().numberOfValidStripTIDHits();
       validhits_[nTracks_][5]=track->hitPattern().numberOfValidStripTOBHits();
       validhits_[nTracks_][6]=track->hitPattern().numberOfValidStripTECHits();
 
 
 
       int myalgo=-88;
       if(track->algo()==reco::TrackBase::undefAlgorithm)myalgo=0;
       if(track->algo()==reco::TrackBase::ctf)myalgo=1;
       if(track->algo()==reco::TrackBase::iter0)myalgo=4;
       if(track->algo()==reco::TrackBase::iter1)myalgo=5;
       if(track->algo()==reco::TrackBase::iter2)myalgo=6;
       if(track->algo()==reco::TrackBase::iter3)myalgo=7;
       if(track->algo()==reco::TrackBase::iter4)myalgo=8;
       if(track->algo()==reco::TrackBase::iter5)myalgo=9;
       if(track->algo()==reco::TrackBase::iter6)myalgo=10;
       if(track->algo()==reco::TrackBase::iter7)myalgo=11;
       trkAlgo_[nTracks_]  = myalgo;
 
       int myquality=-99;
       if(track->quality(reco::TrackBase::undefQuality))myquality=-1;
       if(track->quality(reco::TrackBase::loose))myquality=0;
       if(track->quality(reco::TrackBase::tight))myquality=1;
       if(track->quality(reco::TrackBase::highPurity))myquality=2;
       //if(track->quality(reco::TrackBase::confirmed))myquality=3;
       // if(track->quality(reco::TrackBase::goodIterative))myquality=4;
       // if(track->quality(reco::TrackBase::qualitySize))myquality=5;     
       trkQuality_[nTracks_]= myquality;
 
       if(track->quality(reco::TrackBase::highPurity))isHighPurity_[nTracks_]=1;
       else isHighPurity_[nTracks_]=0;
       nTracks_++;
 
 
     
 
      
     }//end loop on tracks
 
   for(int d=0;d<nTracks_;++d){
     if(abs(trkQuality_[d])>5)cout<<"MYQUALITY!!! " <<trkQuality_[d] <<" at track # "<<d<<"/"<< nTracks_<<endl;
   }
 
  
 
 
   rootTree_->Fill();
 } 

void LhcTrackAnalyzer::beginJob ( void )

privatevirtual

Reimplemented from edm::EDAnalyzer.

Definition at line 207 of file LhcTrackAnalyzer.cc.

References charge_, chi2_, chi2ndof_, dxy_, dz_, eta_, event_, filename_, goodbx_, goodvtx_, isHighPurity_, nTracks_, phi_, pt_, qoverp_, rootFile_, rootTree_, run_, trkAlgo_, trkQuality_, validhits_, xPCA_, yPCA_, and zPCA_.

 {
   edm::LogInfo("beginJob") << "Begin Job" << std::endl;
   // Define TTree for output
   rootFile_ = new TFile(filename_.c_str(),"recreate");
   rootTree_ = new TTree("tree","Lhc Track tree");
   
   // Track Paramters 
   rootTree_->Branch("run",&run_,"run/I");
   rootTree_->Branch("event",&event_,"event/I");
   rootTree_->Branch("goodbx",&goodbx_,"goodbx/O");
   rootTree_->Branch("goodvtx",&goodvtx_,"goodvtx/O");
   rootTree_->Branch("nTracks",&nTracks_,"nTracks/I");
   rootTree_->Branch("pt",&pt_,"pt[nTracks]/D");
   rootTree_->Branch("eta",&eta_,"eta[nTracks]/D");
   rootTree_->Branch("phi",&phi_,"phi[nTracks]/D");
   rootTree_->Branch("chi2",&chi2_,"chi2[nTracks]/D");
   rootTree_->Branch("chi2ndof",&chi2ndof_,"chi2ndof[nTracks]/D");
   rootTree_->Branch("charge",&charge_,"charge[nTracks]/I");
   rootTree_->Branch("qoverp",&qoverp_,"qoverp[nTracks]/D");
   rootTree_->Branch("dz",&dz_,"dz[nTracks]/D");
   rootTree_->Branch("dxy",&dxy_,"dxy[nTracks]/D");
   rootTree_->Branch("xPCA",&xPCA_,"xPCA[nTracks]/D");
   rootTree_->Branch("yPCA",&yPCA_,"yPCA[nTracks]/D");
   rootTree_->Branch("zPCA",&zPCA_,"zPCA[nTracks]/D");
  rootTree_->Branch("isHighPurity",&isHighPurity_,"isHighPurity[nTracks]/I");
   rootTree_->Branch("trkQuality",&trkQuality_,"trkQuality[nTracks]/I");
   rootTree_->Branch("trkAlgo",&trkAlgo_,"trkAlgo[nTracks]/I");
   rootTree_->Branch("nValidHits",&validhits_,"nValidHits[nTracks][7]/I");
  
   
 }

void LhcTrackAnalyzer::endJob ( void )

privatevirtual

Reimplemented from edm::EDAnalyzer.

Definition at line 241 of file LhcTrackAnalyzer.cc.

References rootFile_.

 {
    if ( rootFile_ ) {
      rootFile_->Write();
      rootFile_->Close();
    }
 
    
 
 }

void LhcTrackAnalyzer::SetVarToZero ( )

private

Definition at line 252 of file LhcTrackAnalyzer.cc.

References charge_, chi2_, chi2ndof_, dxy_, dz_, eta_, event_, i, isHighPurity_, j, nMaxtracks_, nTracks_, phi_, pt_, qoverp_, run_, trkAlgo_, trkQuality_, validhits_, xPCA_, yPCA_, and zPCA_.

Referenced by analyze().

                                     {
   run_=-1;
   event_=-99;
   nTracks_ = 0;
   for ( int i=0; i<nMaxtracks_; ++i ) {
     pt_[i]        = 0;
     eta_[i]       = 0;
     phi_[i]       = 0;
     chi2_[i]      = 0;
     chi2ndof_[i]  = 0;
     charge_[i]    = 0;
     qoverp_[i]    = 0;
     dz_[i]        = 0;
     dxy_[i]       = 0;
     xPCA_[i]        = 0;
     yPCA_[i]        = 0;
     zPCA_[i]        = 0;
     trkQuality_[i]  = 0;
     trkAlgo_[i]     = -1;
     isHighPurity_[i]=-3;
     for(int j=0;j<7;j++){
       validhits_[nTracks_][j]=-1*j;
     }
   }
 
 
 }