|
|
|
#include <PFRootEventManagerColin.h>
Public Types | |
| enum | Mode { Neutral = 0, HIGH_E_TAUS = 1 } |
Public Member Functions | |
| PFRootEventManagerColin (const char *file) | |
| bool | processEntry (int entry) |
| process one entry (pass the TTree entry) | |
| bool | processHIGH_E_TAUS () |
| bool | processNeutral () |
| void | readSpecificOptions (const char *file) |
| void | write () |
| ~PFRootEventManagerColin () | |
Public Attributes | |
| int | mode_ |
| NeutralEvent * | neutralEvent_ |
| TTree * | outTreeMy_ |
| TauEvent * | tauEvent_ |
Definition at line 63 of file PFRootEventManagerColin.h.
Definition at line 80 of file PFRootEventManagerColin.h.
{
Neutral=0,
HIGH_E_TAUS=1
};
| PFRootEventManagerColin::PFRootEventManagerColin | ( | const char * | file | ) |
Definition at line 11 of file PFRootEventManagerColin.cc.
References neutralEvent_, outTreeMy_, readSpecificOptions(), and tauEvent_.
: PFRootEventManager(file) { tauEvent_ = 0; neutralEvent_ = 0; outTreeMy_ = 0; // readOptions(file, false, false); // book histos here // neutralEvent_ = new NeutralEvent(); // tauEvent_ = new TauEvent(); // outTree_ = new TTree("Tau",""); // outTree_->Branch("event","TauEvent", &tauEvent_,32000,2); readSpecificOptions(file); }
| PFRootEventManagerColin::~PFRootEventManagerColin | ( | ) |
Definition at line 32 of file PFRootEventManagerColin.cc.
{
// delete event_;
// delete outTree_;
}
| bool PFRootEventManagerColin::processEntry | ( | int | entry | ) | [virtual] |
process one entry (pass the TTree entry)
Reimplemented from PFRootEventManager.
Definition at line 77 of file PFRootEventManagerColin.cc.
References benchmark_cfg::cerr, HIGH_E_TAUS, mode_, Neutral, outTreeMy_, PFRootEventManager::processEntry(), processHIGH_E_TAUS(), processNeutral(), NeutralEvent::rCode, TauEvent::reset(), and tauEvent_.
{
tauEvent_->reset();
if( ! PFRootEventManager::processEntry(entry) ) {
// cerr<<"event was not accepted"<<endl;
// print();
tauEvent_->rCode = 10;
return false; // event not accepted
}
bool rvalue = false;
switch(mode_) {
case Neutral:
// cout<<"colin: process Neutral"<<endl;
rvalue = processNeutral();
break;
case HIGH_E_TAUS:
// cout<<"colin: process highETaus"<<endl;
rvalue = processHIGH_E_TAUS();
break;
default:
cerr<<"colin: undefined mode"<<endl;
assert(0);
}
outTreeMy_->Fill();
outTreeMy_->Fill();
outTreeMy_->Fill();
return rvalue;
}
| bool PFRootEventManagerColin::processHIGH_E_TAUS | ( | ) |
Definition at line 192 of file PFRootEventManagerColin.cc.
References abs, reco::PFBlock::associatedElements(), Association::block, reco::PFTrack::charge(), DeDxDiscriminatorTools::charge(), TauEvent::chi2ECAL, reco::PFTrajectoryPoint::ClosestApproach, PFRootEventManager::clustersECAL_, PFRootEventManager::clustersHCAL_, reco::PFSimParticle::daughterIds(), reco::PFBlockElement::ECAL, NeutralEvent::eECAL, NeutralEvent::eHCAL, reco::PFBlock::elements(), asciidump::elements, NeutralEvent::eNeutral, NeutralEvent::etaECAL, NeutralEvent::etaHCAL, NeutralEvent::etaNeutral, TauEvent::etaTrack, reco::PFBlockElement::HCAL, i, getHLTprescales::index, edm::Ref< C, T, F >::isNull(), reco::PFBlock::linkData(), NeutralEvent::nECAL, NeutralEvent::nHCAL, reco::PFSimParticle::pdgCode(), PFRootEventManager::pfBlocks_, TauEvent::pHadron, TauEvent::pTrack, TauEvent::ptTrack, NeutralEvent::rCode, PFRootEventManager::recTracks_, edm::OwnVector< T, P >::size(), tauEvent_, reco::PFBlockElement::TRACK, and PFRootEventManager::trueParticles_.
Referenced by processEntry().
{
// true info
// 1 charged hadron, 2 photons
// save charged part mom, save sum E photons
int iHadron = -1;
int iPi0 = -1;
unsigned nStableChargedHadrons=0;
unsigned nPi0=0;
for(unsigned i=0; i<trueParticles_.size(); i++) {
const reco::PFSimParticle& part = trueParticles_[i];
int pdgCode = part.pdgCode();
double charge = part.charge();
if( std::abs(pdgCode) > 100 &&
charge !=0 &&
part.daughterIds().empty() ) {
nStableChargedHadrons++;
iHadron = i;
}
else if( std::abs(pdgCode)==111) {
nPi0++;
iPi0 = i;
}
// cout<<i<<" "<<part<<endl;
}
// one has to select 1 charged and 2 photons
// to use this function.
// even after filtering events with one stable charged particle,
// this particle can be a lepton (eg leptonic pion decay)
if( nStableChargedHadrons==0 ) {
tauEvent_->rCode = 4;
return false;
}
assert( nStableChargedHadrons==1 );
double pHadron = trueParticles_[iHadron].extrapolatedPoint(reco::PFTrajectoryPoint::ClosestApproach ).momentum().P();
tauEvent_->pHadron = pHadron;
// tauEvent_->eEcalHadron = trueParticles_[iHadron].ecalEnergy();
if(nPi0 == 1) {
math::XYZTLorentzVector pi0mom = trueParticles_[iPi0].extrapolatedPoint(reco::PFTrajectoryPoint::ClosestApproach ).momentum();
tauEvent_->eNeutral = pi0mom.E();
tauEvent_->etaNeutral = pi0mom.Eta();
}
else {
tauEvent_->eNeutral = 0;
}
// if( tauEvent_->eNeutral > 0.1* tauEvent_->pHadron ) {
// print();
// }
// check that there is
// only one track
// 0 or 1 ecal cluster
// 0 or 1 hcal cluster
if( recTracks_.size() != 1 ) {
// cout<<"more than 1 track"<<endl;
tauEvent_->rCode = 1;
return false;
}
if( clustersECAL_->size() > 1 ) {
// cout<<"more than 1 ecal cluster"<<endl;
tauEvent_->rCode = 2;
// return false;
}
if( clustersHCAL_->size() > 1 ) {
// cout<<"more than 1 hcal cluster"<<endl;
tauEvent_->rCode = 3;
return false;
}
// save track mom + neutral info.
tauEvent_->pTrack = recTracks_[0].extrapolatedPoint(reco::PFTrajectoryPoint::ClosestApproach ).momentum().P();
tauEvent_->ptTrack = recTracks_[0].extrapolatedPoint(reco::PFTrajectoryPoint::ClosestApproach ).momentum().Pt();
tauEvent_->etaTrack = recTracks_[0].extrapolatedPoint(reco::PFTrajectoryPoint::ClosestApproach ).momentum().Eta();
// access blocks
// take the track block
// look for the closest associated ecal and hcal clusters
// fill the tree
for(unsigned i=0; i<pfBlocks_->size(); i++) {
const reco::PFBlock& block = (*pfBlocks_)[i];
const edm::OwnVector< reco::PFBlockElement >&
elements = block.elements();
// look for the track
int iTrack = -1;
unsigned nTracks = 0;
for(unsigned ie=0; ie<elements.size(); ie++) {
if(elements[ie].type() == reco::PFBlockElement::TRACK ) {
iTrack = ie;
nTracks++;
}
}
if(nTracks!=1) continue; // no track, or too many tracks in the block
std::multimap<double, unsigned> sortedElems;
block.associatedElements( iTrack,
block.linkData(),
sortedElems );
tauEvent_->nECAL=0;
tauEvent_->nHCAL=0;
typedef std::multimap<double, unsigned>::iterator IE;
for(IE ie = sortedElems.begin(); ie != sortedElems.end(); ++ie ) {
double chi2 = ie->first;
unsigned index = ie->second;
reco::PFBlockElement::Type type = elements[index].type();
reco::PFClusterRef clusterRef = elements[index].clusterRef();
if( type == reco::PFBlockElement::ECAL ) {
if(!tauEvent_->nECAL ) { // closest ecal
assert( !clusterRef.isNull() );
tauEvent_->eECAL = clusterRef->energy();
tauEvent_->etaECAL = clusterRef->position().Eta();
tauEvent_->chi2ECAL = chi2;
tauEvent_->nECAL++;
}
}
else if( type == reco::PFBlockElement::HCAL ) {
if(!tauEvent_->nHCAL ) { // closest hcal
assert( !clusterRef.isNull() );
tauEvent_->eHCAL = clusterRef->energy();
tauEvent_->etaHCAL = clusterRef->position().Eta();
tauEvent_->nHCAL++;
}
}
} // eles associated to the track
} // blocks
return false;
}
| bool PFRootEventManagerColin::processNeutral | ( | ) |
Definition at line 117 of file PFRootEventManagerColin.cc.
References benchmark_cfg::cerr, NeutralEvent::eECAL, NeutralEvent::eHCAL, NeutralEvent::eNeutral, eta(), NeutralEvent::etaNeutral, configurableAnalysis::GenParticle, edm::HepMCProduct::GetEvent(), i, PFRootEventManager::MCTruth_, NeutralEvent::nECAL, neutralEvent_, NeutralEvent::nHCAL, outTreeMy_, phi, and NeutralEvent::reset().
Referenced by processEntry().
{
// else {
// cerr<<"event accepted"<<endl;
// }
// if( ! ( (*clustersECAL_).size() <= 1 &&
// (*clustersHCAL_).size() <= 1 ) ) {
// cerr<<"wrong number of ECAL or HCAL clusters :"
// <<(*clustersECAL_).size()<<","<<(*clustersHCAL_).size()<<endl;
// return false;
// }
// 1 HCAL cluster
neutralEvent_->reset();
// particle
const HepMC::GenEvent* myGenEvent = MCTruth_.GetEvent();
if(!myGenEvent) {
assert(0);
}
if( myGenEvent->particles_size() != 1 ) {
cerr<<"wrong number of particles:"
<<myGenEvent->particles_size()<<endl;
return 0;
}
// take first particle
const HepMC::GenParticle* particle = *(myGenEvent->particles_begin() );
// and check that it's a K0L
if( particle->pdg_id() != 130 ) {
cerr<<"not a K0L : "<<particle->pdg_id()<<endl;
return false;
}
neutralEvent_->eNeutral = particle->momentum().e();
double eta = particle->momentum().eta();
double phi = particle->momentum().phi();
neutralEvent_->etaNeutral = eta;
neutralEvent_->nECAL = (*clustersECAL_).size();
// look for the closest ECAL cluster from the particle.
double minDist2 = 9999999;
// int iClosest = -1;
for( unsigned i=0; i<(*clustersECAL_).size(); ++i) {
double deta = (*clustersECAL_)[i].position().Eta() - eta;
double dphi = (*clustersECAL_)[i].position().Phi() - phi;
double dist2 = deta*deta + dphi*dphi;
if(dist2 < minDist2) {
minDist2 = dist2;
neutralEvent_->eECAL = (*clustersECAL_)[i].energy();
}
}
neutralEvent_->nHCAL = (*clustersHCAL_).size();
if( (*clustersHCAL_).size() == 1 ) {
neutralEvent_->eHCAL = (*clustersHCAL_)[0].energy();
}
outTreeMy_->Fill();
if( neutralEvent_->nECAL<1 && neutralEvent_->eNeutral<1 )
return true;
else return false;
}
| void PFRootEventManagerColin::readSpecificOptions | ( | const char * | file | ) | [virtual] |
Reimplemented from PFRootEventManager.
Definition at line 39 of file PFRootEventManagerColin.cc.
References benchmark_cfg::cerr, gather_cfg::cout, cmsRelvalreport::exit, IO::GetOpt(), HIGH_E_TAUS, mode_, Neutral, neutralEvent_, PFRootEventManager::options_, PFRootEventManager::outFile_, outTreeMy_, and tauEvent_.
Referenced by PFRootEventManagerColin().
{
cout<<"calling PFRootEventManagerColin::readSpecificOptions"<<endl;
// PFRootEventManager::readOptions(file, refresh, reconnect);
options_->GetOpt("colin", "mode", mode_);
if(outTreeMy_) delete outTreeMy_;
outFile_->cd();
switch(mode_) {
case Neutral:
cout<<"colin: Neutral mode"<<endl;
neutralEvent_ = new NeutralEvent();
outTreeMy_ = new TTree("Neutral","");
outTreeMy_->Branch("event","NeutralEvent", &neutralEvent_,32000,2);
gDirectory->ls();
break;
case HIGH_E_TAUS:
cout<<"colin: highETaus mode"<<endl;
tauEvent_ = new TauEvent();
outTreeMy_ = new TTree("Tau","");
outTreeMy_->Branch("event","TauEvent", &tauEvent_,32000,2);
gDirectory->ls();
break;
default:
cerr<<"colin: undefined mode"<<endl;
exit(1);
}
}
| void PFRootEventManagerColin::write | ( | void | ) | [virtual] |
Reimplemented from PFRootEventManager.
Definition at line 366 of file PFRootEventManagerColin.cc.
References PFRootEventManager::outFile_, and outTreeMy_.
{
// write histos here
outFile_->cd();
outTreeMy_->Write();
PFRootEventManager::write();
}
Definition at line 88 of file PFRootEventManagerColin.h.
Referenced by processEntry(), and readSpecificOptions().
Definition at line 86 of file PFRootEventManagerColin.h.
Referenced by PFRootEventManagerColin(), processNeutral(), and readSpecificOptions().
Definition at line 85 of file PFRootEventManagerColin.h.
Referenced by PFRootEventManagerColin(), processEntry(), processNeutral(), readSpecificOptions(), and write().
Definition at line 87 of file PFRootEventManagerColin.h.
Referenced by PFRootEventManagerColin(), processEntry(), processHIGH_E_TAUS(), and readSpecificOptions().
1.7.3