#include <HcalForwardLibWriter.h>

Inheritance diagram for HcalForwardLibWriter:

Classes
struct	FileHandle

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

	~HcalForwardLibWriter () override

Public Member Functions inherited from edm::one::EDAnalyzer<>
	EDAnalyzer ()=default

	EDAnalyzer (const EDAnalyzer &)=delete

SerialTaskQueue *	globalLuminosityBlocksQueue () final

SerialTaskQueue *	globalRunsQueue () final

const EDAnalyzer &	operator= (const EDAnalyzer &)=delete

bool	wantsGlobalLuminosityBlocks () const final

bool	wantsGlobalRuns () const final

bool	wantsInputProcessBlocks () const final

bool	wantsProcessBlocks () const final

Public Member Functions inherited from edm::one::EDAnalyzerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

	EDAnalyzerBase ()

ModuleDescription const &	moduleDescription () const

bool	wantsStreamLuminosityBlocks () const

bool	wantsStreamRuns () const

	~EDAnalyzerBase () override

Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const

void	convertCurrentProcessAlias (std::string const &processName)
	Convert "@currentProcess" in InputTag process names to the actual current process name. More...

	EDConsumerBase ()

	EDConsumerBase (EDConsumerBase const &)=delete

	EDConsumerBase (EDConsumerBase &&)=default

ESProxyIndex const *	esGetTokenIndices (edm::Transition iTrans) const

std::vector< ESProxyIndex > const &	esGetTokenIndicesVector (edm::Transition iTrans) const

std::vector< ESRecordIndex > const &	esGetTokenRecordIndicesVector (edm::Transition iTrans) const

ProductResolverIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductResolverIndexAndSkipBit > &) const

std::vector < ProductResolverIndexAndSkipBit > const &	itemsToGetFrom (BranchType iType) const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesWhoseProductsAreConsumed (std::array< std::vector< ModuleDescription const * > , NumBranchTypes > &modulesAll, std::vector< ModuleProcessName > &modulesInPreviousProcesses, ProductRegistry const &preg, std::map< std::string, ModuleDescription const > const &labelsToDesc, std::string const &processName) const

EDConsumerBase const &	operator= (EDConsumerBase const &)=delete

EDConsumerBase &	operator= (EDConsumerBase &&)=default

bool	registeredToConsume (ProductResolverIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	selectInputProcessBlocks (ProductRegistry const &productRegistry, ProcessBlockHelperBase const &processBlockHelperBase)

ProductResolverIndexAndSkipBit	uncheckedIndexFrom (EDGetToken) const

void	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)

void	updateLookup (eventsetup::ESRecordsToProxyIndices const &)

virtual	~EDConsumerBase () noexcept(false)

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

void	beginJob () override

void	endJob () override

int	readUserData ()

Private Attributes
int	bsize

int	compressionAlgo

int	compressionLevel

TBranch *	emBranch

HFShowerPhotonCollection	emColl

HFShowerLibraryEventInfo	evtInfo

edm::Service< TFileService >	fs

TBranch *	hadBranch

HFShowerPhotonCollection	hadColl

TFile *	LibFile

TTree *	LibTree

int	nbins

int	nshowers

int	splitlevel

std::string	theDataFile

TFile *	theFile

std::vector< FileHandle >	theFileHandle

TTree *	theTree

Additional Inherited Members
Public Types inherited from edm::one::EDAnalyzerBase
typedef EDAnalyzerBase	ModuleType

Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels

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

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &descriptions)

Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)

template<BranchType B = InEvent>
EDConsumerBaseAdaptor< B >	consumes (edm::InputTag tag) noexcept

EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)

ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...

template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()

void	consumesMany (const TypeToGet &id)

template<BranchType B>
void	consumesMany (const TypeToGet &id)

template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes ()

template<typename ESProduct , typename ESRecord , Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag const &tag)

template<Transition Tr = Transition::Event>
constexpr auto	esConsumes () noexcept

template<Transition Tr = Transition::Event>
auto	esConsumes (ESInputTag tag) noexcept

template<Transition Tr = Transition::Event>
ESGetTokenGeneric	esConsumes (eventsetup::EventSetupRecordKey const &iRecord, eventsetup::DataKey const &iKey)
	Used with EventSetupRecord::doGet. More...

template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)

EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

void	resetItemsToGetFrom (BranchType iType)

Detailed Description

Definition at line 27 of file HcalForwardLibWriter.h.

Constructor & Destructor Documentation

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

explicit

Definition at line 6 of file HcalForwardLibWriter.cc.

References bsize, compressionAlgo, compressionLevel, emBranch, emColl, TFileService::file(), personalPlayback::fp, fs, edm::ParameterSet::getParameter(), hadBranch, hadColl, LibTree, nbins, nshowers, readUserData(), splitlevel, AlCaHLTBitMon_QueryRunRegistry::string, and theDataFile.

                                                                          {
   edm::ParameterSet theParms = iConfig.getParameter<edm::ParameterSet>("HcalForwardLibWriterParameters");
   edm::FileInPath fp = theParms.getParameter<edm::FileInPath>("FileName");
   nbins = theParms.getParameter<int>("Nbins");
   nshowers = theParms.getParameter<int>("Nshowers");
   bsize = theParms.getParameter<int>("BufSize");
   splitlevel = theParms.getParameter<int>("SplitLevel");
   compressionAlgo = theParms.getParameter<int>("CompressionAlgo");
   compressionLevel = theParms.getParameter<int>("CompressionLevel");
 
   std::string pName = fp.fullPath();
   if (pName.find('.') == 0)
     pName.erase(0, 2);
   theDataFile = pName;
   readUserData();
 
   fs->file().cd();
   fs->file().SetCompressionAlgorithm(compressionAlgo);
   fs->file().SetCompressionLevel(compressionLevel);
 
   LibTree = new TTree("HFSimHits", "HFSimHits");
 
   //https://root.cern/root/html534/TTree.html
   // TBranch*Branch(const char* name, const char* classname, void** obj, Int_t bufsize = 32000, Int_t splitlevel = 99)
   emBranch = LibTree->Branch("emParticles", "HFShowerPhotons-emParticles", &emColl, bsize, splitlevel);
   hadBranch = LibTree->Branch("hadParticles", "HFShowerPhotons-hadParticles", &hadColl, bsize, splitlevel);
 }

HcalForwardLibWriter::~HcalForwardLibWriter ( )

override

Definition at line 34 of file HcalForwardLibWriter.cc.

34 {}

Member Function Documentation

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

overrideprivatevirtual

Implements edm::one::EDAnalyzerBase.

Definition at line 36 of file HcalForwardLibWriter.cc.

References emBranch, emColl, personalPlayback::fn, hadBranch, hadColl, mps_fire::i, iev, LibTree, dqmiodumpmetadata::n, nbins, nshowers, findQualityFiles::size, AlCaHLTBitMon_QueryRunRegistry::string, submitPVValidationJobs::t, theFile, theFileHandle, theTree, x, y, and z.

                                                                                       {
   // Event info
   std::vector<double> en;
   double momBin[16] = {2, 3, 5, 7, 10, 15, 20, 30, 50, 75, 100, 150, 250, 350, 500, 1000};
   en.reserve(nbins);
   for (int i = 0; i < nbins; ++i)
     en.push_back(momBin[i]);
 
   int nem = 0;
   int nhad = 0;
 
   //shower photons
   int n = theFileHandle.size();
   // cycle over files ++++++++++++++++++++++++++++++++++++++++++++++++++++
   for (int i = 0; i < n; ++i) {
     std::string fn = theFileHandle[i].name;
     std::string particle = theFileHandle[i].id;
 
     //    std::cout << "*** Input file  " << i << "   " << fn << std::endl;
 
     TFile* theFile = new TFile(fn.c_str(), "READ");
     TTree* theTree = (TTree*)gDirectory->Get("g4SimHits/CherenkovPhotons");
     int nphot = 0;
 
     const int size = 10000;
     if (nshowers > size) {
       edm::LogError("HcalForwardLibWriter") << "Too big Nshowers number";
       return;
     }
 
     float x[size] = {0.};
     float y[size] = {0.};
     float z[size] = {0.};
     float t[size] = {0.};
     float lambda[size] = {0.};
     int fiberId[size] = {0};
     float primZ;  // added
 
     theTree->SetBranchAddress("nphot", &nphot);
     theTree->SetBranchAddress("x", &x);
     theTree->SetBranchAddress("y", &y);
     theTree->SetBranchAddress("z", &z);
     theTree->SetBranchAddress("t", &t);
     theTree->SetBranchAddress("lambda", &lambda);
     theTree->SetBranchAddress("fiberId", &fiberId);
     theTree->SetBranchAddress("primZ", &primZ);  // added
     int nentries = int(theTree->GetEntries());
     int ngood = 0;
     int nbytes = 0;
     // cycle over showers ====================================================
     for (int iev = 0; iev < nentries; iev++) {
       nbytes += theTree->GetEntry(iev);
       if (primZ < 990.)
         continue;  // exclude showers with interactions in front of HF (1m of air)
       ngood++;
       if (ngood > nshowers)
         continue;
       if (particle == "electron") {
         emColl.clear();
       } else {
         hadColl.clear();
       }
       float nphot_long = 0;
       float nphot_short = 0;
       // cycle over photons in shower -------------------------------------------
       for (int iph = 0; iph < nphot; ++iph) {
         if (fiberId[iph] == 1) {
           nphot_long++;
         } else {
           nphot_short++;
           z[iph] = -z[iph];
         }
 
         HFShowerPhoton::Point pos(x[iph], y[iph], z[iph]);
         HFShowerPhoton aPhoton(pos, t[iph], lambda[iph]);
         if (particle == "electron") {
           emColl.push_back(aPhoton);
         } else {
           hadColl.push_back(aPhoton);
         }
       }
       // end of cycle over photons in shower -------------------------------------------
 
       if (particle == "electron") {
         LibTree->SetEntries(nem + 1);
         emBranch->Fill();
         nem++;
         emColl.clear();
       } else {
         LibTree->SetEntries(nhad + 1);
         nhad++;
         hadBranch->Fill();
         hadColl.clear();
       }
     }
     // end of cycle over showers ====================================================
     theFile->Close();
   }
   // end of cycle over files ++++++++++++++++++++++++++++++++++++++++++++++++++++
 }

void HcalForwardLibWriter::beginJob ( void )

overrideprivatevirtual

Reimplemented from edm::one::EDAnalyzerBase.

Definition at line 137 of file HcalForwardLibWriter.cc.

137 {}

void HcalForwardLibWriter::endJob ( void )

overrideprivatevirtual

Reimplemented from edm::one::EDAnalyzerBase.

Definition at line 139 of file HcalForwardLibWriter.cc.

References TFileService::file(), fs, and LibTree.

                                   {
   fs->file().cd();
   LibTree->Write();
   LibTree->Print();
 }

int HcalForwardLibWriter::readUserData ( void )

private

Definition at line 145 of file HcalForwardLibWriter.cc.

References HcalForwardLibWriter::FileHandle::id, input, isotrackApplyRegressor::k, HcalForwardLibWriter::FileHandle::momentum, HcalForwardLibWriter::FileHandle::name, AlCaHLTBitMon_QueryRunRegistry::string, theDataFile, and theFileHandle.

Referenced by HcalForwardLibWriter().

                                            {
   std::ifstream input(theDataFile.c_str());
   if (input.fail()) {
     return 0;
   }
   std::string theFileName, thePID;
   int mom;
   int k = 0;
   while (!input.eof()) {
     input >> theFileName >> thePID >> mom;
     if (!input.fail()) {
       FileHandle aFile;
       aFile.name = theFileName;
       aFile.id = thePID;
       aFile.momentum = mom;
       theFileHandle.push_back(aFile);
       ++k;
     } else {
       input.clear();
     }
     input.ignore(999, '\n');
   }
   return k;
 }