#include <CastorTestAnalysis.h>
Public Member Functions | |
CastorTestAnalysis (const edm::ParameterSet &p) | |
virtual | ~CastorTestAnalysis () |
Private Member Functions | |
void | Finish () |
void | getCastorBranchData (const CaloG4HitCollection *hc) |
void | update (const EndOfRun *run) |
This routine will be called when the appropriate signal arrives. | |
void | update (const G4Step *step) |
This routine will be called when the appropriate signal arrives. | |
void | update (const EndOfEvent *evt) |
This routine will be called when the appropriate signal arrives. | |
void | update (const BeginOfRun *run) |
This routine will be called when the appropriate signal arrives. | |
void | update (const BeginOfJob *run) |
This routine will be called when the appropriate signal arrives. | |
void | update (const BeginOfEvent *evt) |
This routine will be called when the appropriate signal arrives. | |
Private Attributes | |
Float_t | castoreventarray [11] |
TNtuple * | castoreventntuple |
TFile * | castorOutputEventFile |
TFile * | castorOutputStepFile |
Float_t | castorsteparray [14] |
TNtuple * | castorstepntuple |
int | doNTcastorevent |
int | doNTcastorstep |
int | eventGlobalHit |
int | eventIndex |
std::string | eventNtFileName |
int | stepIndex |
std::string | stepNtFileName |
CastorNumberingScheme * | theCastorNumScheme |
int | verbosity |
Definition at line 65 of file CastorTestAnalysis.h.
CastorTestAnalysis::CastorTestAnalysis | ( | const edm::ParameterSet & | p | ) |
Definition at line 35 of file CastorTestAnalysis.cc.
References castoreventntuple, castorstepntuple, gather_cfg::cout, doNTcastorevent, doNTcastorstep, eventNtFileName, edm::ParameterSet::getParameter(), stepNtFileName, and verbosity.
{ edm::ParameterSet m_Anal = p.getParameter<edm::ParameterSet>("CastorTestAnalysis"); verbosity = m_Anal.getParameter<int>("Verbosity"); doNTcastorstep = m_Anal.getParameter<int>("StepNtupleFlag"); doNTcastorevent = m_Anal.getParameter<int>("EventNtupleFlag"); stepNtFileName = m_Anal.getParameter<std::string>("StepNtupleFileName"); eventNtFileName = m_Anal.getParameter<std::string>("EventNtupleFileName"); if (verbosity > 0) std::cout<<std::endl; std::cout<<"============================================================================"<<std::endl; std::cout << "CastorTestAnalysis:: Initialized as observer"<< std::endl; if (doNTcastorstep > 0){ std::cout <<" Step Ntuple will be created"<< std::endl; std::cout <<" Step Ntuple file: "<<stepNtFileName<<std::endl; } if (doNTcastorevent > 0){ std::cout <<" Event Ntuple will be created"<< std::endl; std::cout <<" Step Ntuple file: "<<stepNtFileName<<std::endl; } std::cout<<"============================================================================"<<std::endl; std::cout<<std::endl; if (doNTcastorstep > 0) castorstepntuple = new TNtuple("NTcastorstep","NTcastorstep","evt:trackid:charge:pdgcode:x:y:z:stepl:stepe:eta:phi:vpx:vpy:vpz"); if (doNTcastorevent > 0) castoreventntuple = new TNtuple("NTcastorevent","NTcastorevent","evt:ihit:detector:sector:module:enem:totalenergy:hitenergy:x:y:z"); }
CastorTestAnalysis::~CastorTestAnalysis | ( | ) | [virtual] |
Definition at line 66 of file CastorTestAnalysis.cc.
References gather_cfg::cout, Finish(), and verbosity.
void CastorTestAnalysis::Finish | ( | ) | [private] |
Definition at line 352 of file CastorTestAnalysis.cc.
References castoreventntuple, castorOutputEventFile, castorOutputStepFile, castorstepntuple, gather_cfg::cout, doNTcastorevent, and doNTcastorstep.
Referenced by ~CastorTestAnalysis().
{ if (doNTcastorstep) { castorOutputStepFile->cd(); castorstepntuple->Write(); std::cout << "CastorTestAnalysis: Ntuple step written" <<std::endl; castorOutputStepFile->Close(); std::cout << "CastorTestAnalysis: Step file closed" << std::endl; } if (doNTcastorevent) { castorOutputEventFile->cd(); castoreventntuple->Write("",TObject::kOverwrite); std::cout << "CastorTestAnalysis: Ntuple event written" << std::endl; castorOutputEventFile->Close(); std::cout << "CastorTestAnalysis: Event file closed" << std::endl; } }
void CastorTestAnalysis::getCastorBranchData | ( | const CaloG4HitCollection * | hc | ) | [private] |
Definition at line 295 of file CastorTestAnalysis.cc.
References castoreventarray, castoreventntuple, eventGlobalHit, eventIndex, CaloG4Hit::getEnergyDeposit(), CaloG4Hit::getPosition(), CaloG4Hit::getUnitID(), ntcastore_detector, ntcastore_enem, ntcastore_enhad, ntcastore_evt, ntcastore_hitenergy, ntcastore_ihit, ntcastore_module, ntcastore_sector, ntcastore_x, ntcastore_y, ntcastore_z, theCastorNumScheme, and CastorNumberingScheme::unpackIndex().
Referenced by update().
{ int nentries = hc->entries(); if (nentries > 0) { unsigned int volumeID=0; int det=0, zside, sector, zmodule; std::map<int,float,std::less<int> > themap; double totalEnergy = 0; double hitEnergy = 0; double en_in_sd = 0.; for (int ihit = 0; ihit < nentries; ihit++) { CaloG4Hit* aHit = (*hc)[ihit]; totalEnergy += aHit->getEnergyDeposit(); } for (int ihit = 0; ihit < nentries; ihit++) { CaloG4Hit* aHit = (*hc)[ihit]; volumeID = aHit->getUnitID(); hitEnergy = aHit->getEnergyDeposit(); en_in_sd += aHit->getEnergyDeposit(); // double enEm = aHit->getEM(); // double enHad = aHit->getHadr(); themap[volumeID] += aHit->getEnergyDeposit(); // int det, zside, sector, zmodule; theCastorNumScheme->unpackIndex(volumeID, zside, sector,zmodule); // det = 2 ; // det=2/3 for CAFI/CAPL castoreventarray[ntcastore_evt] = (float)eventIndex; // castoreventarray[ntcastore_ihit] = (float)ihit; castoreventarray[ntcastore_ihit] = (float)eventGlobalHit; castoreventarray[ntcastore_detector] = (float)det; castoreventarray[ntcastore_sector] = (float)sector; castoreventarray[ntcastore_module] = (float)zmodule; castoreventarray[ntcastore_enem] = en_in_sd; castoreventarray[ntcastore_enhad] = totalEnergy; castoreventarray[ntcastore_hitenergy] = hitEnergy; castoreventarray[ntcastore_x] = aHit->getPosition().x(); castoreventarray[ntcastore_y] = aHit->getPosition().y(); castoreventarray[ntcastore_z] = aHit->getPosition().z(); // castoreventarray[ntcastore_x] = aHit->getEntry().x(); // castoreventarray[ntcastore_y] = aHit->getEntry().y(); // castoreventarray[ntcastore_z] = aHit->getEntry().z(); castoreventntuple->Fill(castoreventarray); eventGlobalHit++ ; } } // nentries > 0 }
void CastorTestAnalysis::update | ( | const EndOfRun * | ) | [private, virtual] |
This routine will be called when the appropriate signal arrives.
Implements Observer< const EndOfRun * >.
Definition at line 292 of file CastorTestAnalysis.cc.
{;}
void CastorTestAnalysis::update | ( | const G4Step * | ) | [private, virtual] |
This routine will be called when the appropriate signal arrives.
Implements Observer< const G4Step * >.
Definition at line 114 of file CastorTestAnalysis.cc.
References castorsteparray, castorstepntuple, gather_cfg::cout, doNTcastorstep, eta(), eventIndex, funct::log(), LogDebug, ntcastors_charge, ntcastors_eta, ntcastors_evt, ntcastors_pdgcode, ntcastors_phi, ntcastors_stepe, ntcastors_stepl, ntcastors_trackid, ntcastors_vpx, ntcastors_vpy, ntcastors_vpz, ntcastors_x, ntcastors_y, ntcastors_z, phi, stepIndex, and verbosity.
{ stepIndex++; if (doNTcastorstep) { G4StepPoint * preStepPoint = aStep->GetPreStepPoint(); // G4StepPoint * postStepPoint= aStep->GetPostStepPoint(); G4double stepL = aStep->GetStepLength(); G4double stepE = aStep->GetTotalEnergyDeposit(); if (verbosity >= 2) std::cout << "Step " << stepL << ", " << stepE << std::endl; G4Track * theTrack = aStep->GetTrack(); G4int theTrackID = theTrack->GetTrackID(); G4double theCharge = theTrack->GetDynamicParticle()->GetCharge(); G4String particleType = theTrack->GetDefinition()->GetParticleName(); G4int pdgcode = theTrack->GetDefinition()->GetPDGEncoding(); G4ThreeVector vert_mom = theTrack->GetVertexMomentumDirection(); G4double vpx = vert_mom.x(); G4double vpy = vert_mom.y(); G4double vpz = vert_mom.z(); double eta = 0.5 * log( (1.+vpz) / (1.-vpz) ); double phi = atan2(vpy,vpx); G4ThreeVector hitPoint = preStepPoint->GetPosition(); // Fill-in ntuple // castorsteparray[ntcastors_evt] = (*evt)()->GetEventID(); castorsteparray[ntcastors_evt] = (float)eventIndex; castorsteparray[ntcastors_trackid] = (float)theTrackID; castorsteparray[ntcastors_charge] = theCharge; castorsteparray[ntcastors_pdgcode] = pdgcode; castorsteparray[ntcastors_x] = hitPoint.x(); castorsteparray[ntcastors_y] = hitPoint.y(); castorsteparray[ntcastors_z] = hitPoint.z(); castorsteparray[ntcastors_stepl] = stepL; castorsteparray[ntcastors_stepe] = stepE; castorsteparray[ntcastors_eta] = eta; castorsteparray[ntcastors_phi] = phi; castorsteparray[ntcastors_vpx] = vpx; castorsteparray[ntcastors_vpy] = vpy; castorsteparray[ntcastors_vpz] = vpz; /* std::cout<<"TrackID: "<< theTrackID<<std::endl; std::cout<<" StepN: "<< theTrack->GetCurrentStepNumber() <<std::endl; std::cout<<" ParentID: "<< aStep->GetTrack()->GetParentID() <<std::endl; std::cout<<" PDG: "<< pdgcode <<std::endl; std::cout<<" X,Y,Z (mm): "<< theTrack->GetPosition().x() <<","<< theTrack->GetPosition().y() <<","<< theTrack->GetPosition().z() <<std::endl; std::cout<<" KE (MeV): "<< theTrack->GetKineticEnergy() <<std::endl; std::cout<<" Total EDep (MeV): "<< aStep->GetTotalEnergyDeposit() <<std::endl; std::cout<<" StepLength (mm): "<< aStep->GetStepLength() <<std::endl; std::cout<<" TrackLength (mm): "<< theTrack->GetTrackLength() <<std::endl; if ( theTrack->GetNextVolume() != 0 ) std::cout<<" NextVolume: "<< theTrack->GetNextVolume()->GetName() <<std::endl; else std::cout<<" NextVolume: OutOfWorld"<<std::endl; if(aStep->GetPostStepPoint()->GetProcessDefinedStep() != NULL) std::cout<<" ProcessName: "<< aStep->GetPostStepPoint()->GetProcessDefinedStep()->GetProcessName() <<std::endl; else std::cout<<" ProcessName: UserLimit"<<std::endl; std::cout<<std::endl; */ #ifdef DebugLog if ( theTrack->GetNextVolume() != 0 ) LogDebug("ForwardSim") << " NextVolume: " << theTrack->GetNextVolume()->GetName() ; else LogDebug("ForwardSim") << " NextVolume: OutOfWorld" ; #endif //fill ntuple with step level information castorstepntuple->Fill(castorsteparray); } }
void CastorTestAnalysis::update | ( | const EndOfEvent * | ) | [private, virtual] |
This routine will be called when the appropriate signal arrives.
Implements Observer< const EndOfEvent * >.
Definition at line 199 of file CastorTestAnalysis.cc.
References gather_cfg::cout, doNTcastorevent, eta(), eventGlobalHit, getCastorBranchData(), i, funct::log(), max(), min, npart, phi, funct::pow(), mathSSE::sqrt(), funct::tan(), theCastorNumScheme, and theta().
{ // Look for the Hit Collection std::cout << std::endl << "CastorTest::update(EndOfEvent * evt) - event #" << (*evt)()->GetEventID() << std::endl; // access to the G4 hit collections G4HCofThisEvent* allHC = (*evt)()->GetHCofThisEvent(); std::cout << "update(*evt) --> accessed all HC"; int CAFIid = G4SDManager::GetSDMpointer()->GetCollectionID("CastorFI"); CaloG4HitCollection* theCAFI = (CaloG4HitCollection*) allHC->GetHC(CAFIid); theCastorNumScheme = new CastorNumberingScheme(); // CastorNumberingScheme *theCastorNumScheme = new CastorNumberingScheme(); /* unsigned int volumeID=0; int det, zside, sector, zmodule; std::map<int,float,std::less<int> > themap; double totalEnergy = 0; double hitEnergy = 0; double en_in_fi = 0.; double en_in_pl = 0.; */ // double en_in_bu = 0.; // double en_in_tu = 0.; if (doNTcastorevent) { eventGlobalHit = 0 ; // int eventGlobalHit = 0 ; // Check FI TBranch for Hits if (theCAFI->entries() > 0) getCastorBranchData(theCAFI) ; // Find Primary info: int trackID = 0; int particleType = 0; G4PrimaryParticle* thePrim=0; G4int nvertex = (*evt)()->GetNumberOfPrimaryVertex(); std::cout << "Event has " << nvertex << " vertex" << std::endl; if (nvertex==0) std::cout << "CASTORTest End Of Event ERROR: no vertex" << std::endl; for (int i = 0 ; i<nvertex; i++) { G4PrimaryVertex* avertex = (*evt)()->GetPrimaryVertex(i); if (avertex == 0) std::cout << "CASTORTest End Of Event ERR: pointer to vertex = 0" << std::endl; std::cout << "Vertex number :" <<i << std::endl; int npart = avertex->GetNumberOfParticle(); if (npart ==0) std::cout << "CASTORTest End Of Event ERR: no primary!" << std::endl; if (thePrim==0) thePrim=avertex->GetPrimary(trackID); } double px=0.,py=0.,pz=0.; double eta = 0., phi = 0., pInit = 0.; if (thePrim != 0) { px = thePrim->GetPx(); py = thePrim->GetPy(); pz = thePrim->GetPz(); pInit = sqrt(pow(px,2.)+pow(py,2.)+pow(pz,2.)); if (pInit==0) { std::cout << "CASTORTest End Of Event ERR: primary has p=0 " << std::endl; } else { float costheta = pz/pInit; float theta = acos(std::min(std::max(costheta,float(-1.)),float(1.))); eta = -log(tan(theta/2)); if (px != 0) phi = atan(py/px); } particleType = thePrim->GetPDGcode(); } else { std::cout << "CASTORTest End Of Event ERR: could not find primary " << std::endl; } } int iEvt = (*evt)()->GetEventID(); if (iEvt < 10) std::cout << " CastorTest Event " << iEvt << std::endl; else if ((iEvt < 100) && (iEvt%10 == 0)) std::cout << " CastorTest Event " << iEvt << std::endl; else if ((iEvt < 1000) && (iEvt%100 == 0)) std::cout << " CastorTest Event " << iEvt << std::endl; else if ((iEvt < 10000) && (iEvt%1000 == 0)) std::cout << " CastorTest Event " << iEvt << std::endl; std::cout << std::endl << "===>>> Done writing user histograms " << std::endl; }
void CastorTestAnalysis::update | ( | const BeginOfRun * | ) | [private, virtual] |
This routine will be called when the appropriate signal arrives.
Implements Observer< const BeginOfRun * >.
Definition at line 86 of file CastorTestAnalysis.cc.
References castorOutputEventFile, castorOutputStepFile, gather_cfg::cout, doNTcastorevent, doNTcastorstep, eventIndex, eventNtFileName, and stepNtFileName.
{ std::cout << std::endl << "CastorTestAnalysis: Starting Run"<< std::endl; if (doNTcastorstep) { std::cout << "CastorTestAnalysis: output step root file created"<< std::endl; TString stepfilename = stepNtFileName; castorOutputStepFile = new TFile(stepfilename,"RECREATE"); } if (doNTcastorevent) { std::cout << "CastorTestAnalysis: output event root file created"<< std::endl; TString stepfilename = eventNtFileName; castorOutputEventFile = new TFile(stepfilename,"RECREATE"); } eventIndex = 0; }
void CastorTestAnalysis::update | ( | const BeginOfJob * | ) | [private, virtual] |
This routine will be called when the appropriate signal arrives.
Implements Observer< const BeginOfJob * >.
Definition at line 80 of file CastorTestAnalysis.cc.
References gather_cfg::cout.
{ std::cout << " Starting new job " << std::endl; }
void CastorTestAnalysis::update | ( | const BeginOfEvent * | ) | [private, virtual] |
This routine will be called when the appropriate signal arrives.
Implements Observer< const BeginOfEvent * >.
Definition at line 105 of file CastorTestAnalysis.cc.
References gather_cfg::cout, eventIndex, and stepIndex.
{ std::cout << "CastorTestAnalysis: Processing Event Number: "<<eventIndex<< std::endl; eventIndex++; stepIndex = 0; }
Float_t CastorTestAnalysis::castoreventarray[11] [private] |
Definition at line 110 of file CastorTestAnalysis.h.
Referenced by getCastorBranchData().
TNtuple* CastorTestAnalysis::castoreventntuple [private] |
Definition at line 101 of file CastorTestAnalysis.h.
Referenced by CastorTestAnalysis(), Finish(), and getCastorBranchData().
TFile* CastorTestAnalysis::castorOutputEventFile [private] |
Definition at line 97 of file CastorTestAnalysis.h.
TFile* CastorTestAnalysis::castorOutputStepFile [private] |
Definition at line 98 of file CastorTestAnalysis.h.
Float_t CastorTestAnalysis::castorsteparray[14] [private] |
Definition at line 109 of file CastorTestAnalysis.h.
Referenced by update().
TNtuple* CastorTestAnalysis::castorstepntuple [private] |
Definition at line 100 of file CastorTestAnalysis.h.
Referenced by CastorTestAnalysis(), Finish(), and update().
int CastorTestAnalysis::doNTcastorevent [private] |
Definition at line 93 of file CastorTestAnalysis.h.
Referenced by CastorTestAnalysis(), Finish(), and update().
int CastorTestAnalysis::doNTcastorstep [private] |
Definition at line 92 of file CastorTestAnalysis.h.
Referenced by CastorTestAnalysis(), Finish(), and update().
int CastorTestAnalysis::eventGlobalHit [private] |
Definition at line 107 of file CastorTestAnalysis.h.
Referenced by getCastorBranchData(), and update().
int CastorTestAnalysis::eventIndex [private] |
Definition at line 105 of file CastorTestAnalysis.h.
Referenced by getCastorBranchData(), and update().
std::string CastorTestAnalysis::eventNtFileName [private] |
Definition at line 95 of file CastorTestAnalysis.h.
Referenced by CastorTestAnalysis(), and update().
int CastorTestAnalysis::stepIndex [private] |
Definition at line 106 of file CastorTestAnalysis.h.
Referenced by update().
std::string CastorTestAnalysis::stepNtFileName [private] |
Definition at line 94 of file CastorTestAnalysis.h.
Referenced by CastorTestAnalysis(), and update().
Definition at line 103 of file CastorTestAnalysis.h.
Referenced by getCastorBranchData(), and update().
int CastorTestAnalysis::verbosity [private] |
Definition at line 91 of file CastorTestAnalysis.h.
Referenced by CastorTestAnalysis(), update(), and ~CastorTestAnalysis().