#include <CastorShowerLibrary.h>

Public Member Functions
	CastorShowerLibrary (const std::string &name, edm::ParameterSet const &p)

int	FindEnergyBin (double)

int	FindEtaBin (double)

int	FindPhiBin (double)

CastorShowerEvent	getShowerHits (const G4Step *, bool &)

	~CastorShowerLibrary ()

Protected Member Functions
CastorShowerEvent	getRecord (int, int)

void	initFile (edm::ParameterSet const &)

void	loadEventInfo (TBranch *)

CastorShowerEvent	select (int, double, double=0, double=9)

Private Attributes
TBranch *	emBranch

unsigned int	evtPerBin

TBranch *	hadBranch

TFile *	hf

unsigned int	nBinsE

unsigned int	nBinsEta

unsigned int	nBinsPhi

unsigned int	nEvtPerBinE

unsigned int	nEvtPerBinEta

unsigned int	nEvtPerBinPhi

unsigned int	nMomBin

std::vector< double >	pmom

std::vector< double >	SLenergies

std::vector< double >	SLetas

std::vector< double >	SLphis

unsigned int	totEvents

bool	verbose

Detailed Description

Definition at line 27 of file CastorShowerLibrary.h.

Constructor & Destructor Documentation

CastorShowerLibrary::CastorShowerLibrary	(	const std::string &	name,
		edm::ParameterSet const &	p
	)

Definition at line 30 of file CastorShowerLibrary.cc.

References initFile().

     : hf(nullptr),
       emBranch(nullptr),
       hadBranch(nullptr),
       verbose(false),
       nMomBin(0),
       totEvents(0),
       evtPerBin(0),
       nBinsE(0),
       nBinsEta(0),
       nBinsPhi(0),
       nEvtPerBinE(0),
       nEvtPerBinEta(0),
       nEvtPerBinPhi(0),
       SLenergies(),
       SLetas(),
       SLphis() {
   initFile(p);
 }

CastorShowerLibrary::~CastorShowerLibrary ( )

Definition at line 52 of file CastorShowerLibrary.cc.

References hf.

                                           {
   if (hf)
     hf->Close();
 }

Member Function Documentation

int CastorShowerLibrary::FindEnergyBin ( double energy )

Definition at line 331 of file CastorShowerLibrary.cc.

References mps_fire::i, and SLenergies.

Referenced by select().

                                                     {
   //
   // returns the integer index of the energy bin, taken from SLenergies vector
   // returns -1 if ouside valid range
   //
   if (energy >= SLenergies.back())
     return SLenergies.size() - 1;
 
   unsigned int i = 0;
   for (; i < SLenergies.size() - 1; i++)
     if (energy >= SLenergies.at(i) && energy < SLenergies.at(i + 1))
       return (int)i;
 
   // now i points to the last but 1 bin
   if (energy >= SLenergies.at(i))
     return (int)i;
   // energy outside bin range
   return -1;
 }

int CastorShowerLibrary::FindEtaBin ( double eta )

Definition at line 353 of file CastorShowerLibrary.cc.

References mps_fire::i, and SLetas.

Referenced by select().

                                               {
   //
   // returns the integer index of the eta bin, taken from SLetas vector
   // returns -1 if ouside valid range
   //
   if (eta >= SLetas.back())
     return SLetas.size() - 1;
   unsigned int i = 0;
   for (; i < SLetas.size() - 1; i++)
     if (eta >= SLetas.at(i) && eta < SLetas.at(i + 1))
       return (int)i;
   // now i points to the last but 1 bin
   if (eta >= SLetas.at(i))
     return (int)i;
   // eta outside bin range
   return -1;
 }

int CastorShowerLibrary::FindPhiBin ( double phi )

Definition at line 373 of file CastorShowerLibrary.cc.

References mps_fire::i, and SLphis.

Referenced by select().

                                               {
   //
   // returns the integer index of the phi bin, taken from SLphis vector
   // returns -1 if ouside valid range
   //
   // needs protection in case phi is outside range -pi,pi
   //
   if (phi >= SLphis.back())
     return SLphis.size() - 1;
   unsigned int i = 0;
   for (; i < SLphis.size() - 1; i++)
     if (phi >= SLphis.at(i) && phi < SLphis.at(i + 1))
       return (int)i;
   // now i points to the last but 1 bin
   if (phi >= SLphis.at(i))
     return (int)i;
   // phi outside bin range
   return -1;
 }

CastorShowerEvent CastorShowerLibrary::getRecord	(	int	type,
		int	record
	)

protected

Definition at line 229 of file CastorShowerLibrary.cc.

References emBranch, CastorShowerEvent::getNhit(), hadBranch, LogDebug, and GlobalPosition_Frontier_DevDB_cff::record.

Referenced by select().

                                                                      {
   //
   //  Retrieve event # "record" from the library and stores it
   //  into  vector<CastorShowerHit> showerHit;
   //
   //  Based on HFShowerLibrary::getRecord
   //
   //  Modified 02/02/09 by W. Carvalho
   //
 
 #ifdef EDM_ML_DEBUG
   LogDebug("CastorShower") << "CastorShowerLibrary::getRecord: ";
 #endif
   int nrc = record;
   CastorShowerEvent retValue;
   CastorShowerEvent* showerEvent = &retValue;
   if (type > 0) {
     hadBranch->SetAddress(&showerEvent);
     hadBranch->GetEntry(nrc);
   } else {
     emBranch->SetAddress(&showerEvent);
     emBranch->GetEntry(nrc);
   }
 
 #ifdef EDM_ML_DEBUG
   int nHit = showerEvent->getNhit();
   LogDebug("CastorShower") << "CastorShowerLibrary::getRecord: Record " << record << " of type " << type << " with "
                            << nHit << " CastorShowerHits";
 
 #endif
   return retValue;
 }

CastorShowerEvent CastorShowerLibrary::getShowerHits	(	const G4Step *	aStep,
		bool &	ok
	)

Definition at line 191 of file CastorShowerLibrary.cc.

References funct::abs(), CastorShowerEvent::Clear(), G4TrackToParticleID::isGammaElectronPositron(), G4TrackToParticleID::isStableHadronIon(), log, pi, dttmaxenums::R, select(), mathSSE::sqrt(), funct::tan(), theta(), and HLT_FULL_cff::track.

Referenced by CastorSD::getFromLibrary().

                                                                                   {
   G4StepPoint* preStepPoint = aStep->GetPreStepPoint();
   G4Track* track = aStep->GetTrack();
 
   const G4ThreeVector& hitPoint = preStepPoint->GetPosition();
 
   CastorShowerEvent hit;
   hit.Clear();
 
   ok = false;
   bool isEM = G4TrackToParticleID::isGammaElectronPositron(track);
   if (!isEM && !G4TrackToParticleID::isStableHadronIon(track)) {
     return hit;
   }
   ok = true;
 
   double pin = preStepPoint->GetTotalEnergy();
   double zint = hitPoint.z();
   double R = sqrt(hitPoint.x() * hitPoint.x() + hitPoint.y() * hitPoint.y());
   double theta = atan2(R, std::abs(zint));
   double phiin = atan2(hitPoint.y(), hitPoint.x());
   double etain = -std::log(std::tan((CLHEP::pi - theta) * 0.5));
 
   // Replace "interpolation/extrapolation" by new method "select" that just randomly
   // selects a record from the appropriate energy bin and fills its content to
   // "showerEvent" instance of class CastorShowerEvent
 
   if (isEM) {
     hit = select(0, pin, etain, phiin);
   } else {
     hit = select(1, pin, etain, phiin);
   }
 
   return hit;
 }

void CastorShowerLibrary::initFile ( edm::ParameterSet const & p )

protected

Definition at line 59 of file CastorShowerLibrary.cc.

References emBranch, edmPickEvents::event, Exception, personalPlayback::fp, edm::ParameterSet::getParameter(), edm::ParameterSet::getUntrackedParameter(), hadBranch, hf, loadEventInfo(), AlCaHLTBitMon_QueryRunRegistry::string, and verbose.

Referenced by CastorShowerLibrary().

                                                            {
   //
   //  Init TFile and associated TBranch's of CASTOR Root file
   //  holding library events
   //
 
   //
   //  Read PSet for Castor shower library
   //
 
   edm::ParameterSet m_CS = p.getParameter<edm::ParameterSet>("CastorShowerLibrary");
   edm::FileInPath fp = m_CS.getParameter<edm::FileInPath>("FileName");
   std::string pTreeName = fp.fullPath();
   std::string branchEvInfo = m_CS.getUntrackedParameter<std::string>("BranchEvt");
   std::string branchEM = m_CS.getUntrackedParameter<std::string>("BranchEM");
   std::string branchHAD = m_CS.getUntrackedParameter<std::string>("BranchHAD");
   verbose = m_CS.getUntrackedParameter<bool>("Verbosity", false);
 
   // Open TFile
   if (pTreeName.find('.') == 0)
     pTreeName.erase(0, 2);
   const char* nTree = pTreeName.c_str();
   hf = TFile::Open(nTree);
 
   // Check that TFile has been successfully opened
   if (!hf->IsOpen()) {
     edm::LogError("CastorShower") << "CastorShowerLibrary: opening " << nTree << " failed";
     throw cms::Exception("Unknown", "CastorShowerLibrary") << "Opening of " << pTreeName << " fails\n";
   } else {
     edm::LogVerbatim("CastorShower") << "CastorShowerLibrary: opening " << nTree << " successfully";
   }
 
   // Check for the TBranch holding EventVerbatim in "Events" TTree
   TTree* event = hf->Get<TTree>("CastorCherenkovPhotons");
   if (event) {
     auto evtInfo = event->GetBranch(branchEvInfo.c_str());
     if (evtInfo) {
       loadEventInfo(evtInfo);
     } else {
       edm::LogError("CastorShower") << "CastorShowerLibrary: " << branchEvInfo.c_str()
                                     << " Branch does not exit in Event";
       throw cms::Exception("Unknown", "CastorShowerLibrary") << "Event information absent\n";
     }
   } else {
     edm::LogError("CastorShower") << "CastorShowerLibrary: Events Tree does not exist";
     throw cms::Exception("Unknown", "CastorShowerLibrary") << "Events tree absent\n";
   }
 
   // Get EM and HAD Branchs
   emBranch = event->GetBranch(branchEM.c_str());
   if (verbose)
     emBranch->Print();
   hadBranch = event->GetBranch(branchHAD.c_str());
   if (verbose)
     hadBranch->Print();
   edm::LogVerbatim("CastorShower") << "CastorShowerLibrary: Branch " << branchEM << " has " << emBranch->GetEntries()
                                    << " entries and Branch " << branchHAD << " has " << hadBranch->GetEntries()
                                    << " entries";
 }

void CastorShowerLibrary::loadEventInfo ( TBranch * branch )

protected

Definition at line 123 of file CastorShowerLibrary.cc.

References mps_fire::i, nBinsE, nBinsEta, nBinsPhi, nEvtPerBinE, nEvtPerBinEta, nEvtPerBinPhi, SLenergies, SLetas, SLphis, and totEvents.

Referenced by initFile().

                                                        {
   //
   //  Get EventInfo from the "TBranch* branch" of Root file
   //  holding library events
   //
   //  Based on HFShowerLibrary::loadEventInfo
   //
 
   CastorShowerLibraryInfo tempInfo;
   auto* eventInfo = &tempInfo;
   branch->SetAddress(&eventInfo);
   branch->GetEntry(0);
   // Initialize shower library general parameters
 
   totEvents = eventInfo->Energy.getNEvts();
   nBinsE = eventInfo->Energy.getNBins();
   nEvtPerBinE = eventInfo->Energy.getNEvtPerBin();
   SLenergies = eventInfo->Energy.getBin();
   nBinsEta = eventInfo->Eta.getNBins();
   nEvtPerBinEta = eventInfo->Eta.getNEvtPerBin();
   SLetas = eventInfo->Eta.getBin();
   nBinsPhi = eventInfo->Phi.getNBins();
   nEvtPerBinPhi = eventInfo->Phi.getNEvtPerBin();
   SLphis = eventInfo->Phi.getBin();
 
   // Convert from GeV to MeV
   for (unsigned int i = 0; i < SLenergies.size(); i++) {
     SLenergies[i] *= CLHEP::GeV;
   }
 
   edm::LogVerbatim("CastorShower") << " CastorShowerLibrary::loadEventInfo : "
                                    << "\n \n Total number of events     :  " << totEvents
                                    << "\n   Number of bins  (E)       :  " << nBinsE
                                    << "\n   Number of events/bin (E)  :  " << nEvtPerBinE
                                    << "\n   Number of bins  (Eta)       :  " << nBinsEta
                                    << "\n   Number of events/bin (Eta)  :  " << nEvtPerBinEta
                                    << "\n   Number of bins  (Phi)       :  " << nBinsPhi
                                    << "\n   Number of events/bin (Phi)  :  " << nEvtPerBinPhi << "\n";
 
   std::stringstream ss1;
   ss1 << "CastorShowerLibrary: energies in GeV:\n";
   for (unsigned int i = 0; i < nBinsE; ++i) {
     if (i > 0 && i / 10 * 10 == i) {
       ss1 << "\n";
     }
     ss1 << " " << SLenergies[i] / CLHEP::GeV;
   }
   ss1 << "\nCastorShowerLibrary: etas:\n";
   for (unsigned int i = 0; i < nBinsEta; ++i) {
     if (i > 0 && i / 10 * 10 == i) {
       ss1 << "\n";
     }
     ss1 << " " << SLetas[i];
   }
   ss1 << "\nCastorShowerLibrary: phis:\n";
   for (unsigned int i = 0; i < nBinsPhi; ++i) {
     if (i > 0 && i / 10 * 10 == i) {
       ss1 << "\n";
     }
     ss1 << " " << SLphis[i];
   }
   edm::LogVerbatim("CastorShower") << ss1.str();
 }

CastorShowerEvent CastorShowerLibrary::select	(	int	type,
		double	pin,
		double	etain = `0`,
		double	phiin = `9`
	)

protected

Definition at line 266 of file CastorShowerLibrary.cc.

References FindEnergyBin(), FindEtaBin(), FindPhiBin(), getRecord(), M_PI, nEvtPerBinE, nEvtPerBinEta, nEvtPerBinPhi, L1TMuonDQMOffline_cfi::phiMax, L1TMuonDQMOffline_cfi::phiMin, alignCSCRings::r, and verbose.

Referenced by getShowerHits().

                                                                                               {
   //
   //  Selects an event from the library based on
   //
   //    type:   0 --> em
   //           >0 --> had
   //    pin  :  momentum
   //    etain:  eta (if not given, disregard the eta binning
   //    phiin:  phi (if not given, disregard the phi binning
   //
   //  Created 30/01/09 by W. Carvalho
   //
 
   int irec;                    // to hold record number
   double r = G4UniformRand();  // to randomly select within an energy bin (r=[0,1])
 
   // Randomly select a record from the right energy bin in the library,
   // based on track momentum (pin)
 
   int ienergy = FindEnergyBin(pin);
   int ieta = FindEtaBin(etain);
 #ifdef EDM_ML_DEBUG
   if (verbose)
     edm::LogVerbatim("CastorShower") << " ienergy = " << ienergy;
   if (verbose)
     edm::LogVerbatim("CastorShower") << " ieta = " << ieta;
 #endif
 
   int iphi;
   const double phiMin = 0.;
   const double phiMax = M_PI / 4.;  // 45 * (pi/180)  rad
   if (phiin < phiMin)
     phiin = phiin + M_PI;
   if (phiin >= phiMin && phiin < phiMax) {
     iphi = FindPhiBin(phiin);
   } else {
     double remainder = fmod(phiin, phiMax);
     phiin = phiMin + remainder;
     iphi = FindPhiBin(phiin);
   }
 #ifdef EDM_ML_DEBUG
   if (verbose)
     edm::LogVerbatim("CastorShower") << " iphi = " << iphi;
 #endif
   if (ienergy == -1)
     ienergy = 0;  // if pin < first bin, choose an event in the first one
   if (ieta == -1)
     ieta = 0;  // idem for eta
   if (iphi != -1)
     irec = int(nEvtPerBinE * ienergy + nEvtPerBinEta * ieta + nEvtPerBinPhi * (iphi + r));
   else
     irec = int(nEvtPerBinE * (ienergy + r));
 
 #ifdef EDM_ML_DEBUG
   edm::LogVerbatim("CastorShower") << "CastorShowerLibrary:: Select record " << irec << " of type " << type;
 #endif
 
   //  Retrieve record number "irec" from the library
   return getRecord(type, irec);
 }