|
|
|
#include <SimG4CMS/ShowerLibraryProducer/plugins/HcalForwardLibWriter.cc>
Classes | |
| struct | FileHandle |
Public Member Functions | |
| HcalForwardLibWriter (const edm::ParameterSet &) | |
| ~HcalForwardLibWriter () | |
Static Public Member Functions | |
| static void | fillDescriptions (edm::ConfigurationDescriptions &descriptions) |
Private Member Functions | |
| virtual void | beginJob () |
| virtual void | beginLuminosityBlock (edm::LuminosityBlock &, edm::EventSetup const &) |
| virtual void | beginRun (edm::Run &, edm::EventSetup const &) |
| virtual void | endJob () |
| virtual void | endLuminosityBlock (edm::LuminosityBlock &, edm::EventSetup const &) |
| virtual void | endRun (edm::Run &, edm::EventSetup const &) |
| virtual void | produce (edm::Event &, const edm::EventSetup &) |
| void | readUserData () |
Private Attributes | |
| std::string | fDataFile |
| TFile * | fFile |
| std::vector< FileHandle > | fFileHandle |
| TTree * | fTree |
Description: [one line class summary]
Implementation: [Notes on implementation]
Definition at line 52 of file HcalForwardLibWriter.h.
| HcalForwardLibWriter::HcalForwardLibWriter | ( | const edm::ParameterSet & | iConfig | ) | [explicit] |
Definition at line 22 of file HcalForwardLibWriter.cc.
References gather_cfg::cout, fDataFile, edm::ParameterSet::getParameter(), readUserData(), and AlCaHLTBitMon_QueryRunRegistry::string.
{
edm::ParameterSet theParms = iConfig.getParameter<edm::ParameterSet> ("HcalForwardLibWriterParameters");
edm::FileInPath fp = theParms.getParameter<edm::FileInPath> ("FileName");
std::string pName = fp.fullPath();
std::cout<<pName<<std::endl;
fDataFile = pName;
readUserData();
//register shower library products with label
produces<HFShowerPhotonCollection> ("emParticles");
produces<std::vector<HFShowerPhotonCollection> > ("emParticlesss");
produces< std::vector<int> > ("emParticles");
produces<HFShowerPhotonCollection> ("hadParticles");
produces< std::vector<int> > ("hadParticles");
produces<std::vector<HFShowerLibraryEventInfo> > ("HFShowerLibraryEventInfo");
}
| HcalForwardLibWriter::~HcalForwardLibWriter | ( | ) |
Definition at line 40 of file HcalForwardLibWriter.cc.
{}
| void HcalForwardLibWriter::beginJob | ( | void | ) | [private, virtual] |
| void HcalForwardLibWriter::beginLuminosityBlock | ( | edm::LuminosityBlock & | , |
| edm::EventSetup const & | |||
| ) | [private, virtual] |
| void HcalForwardLibWriter::beginRun | ( | edm::Run & | , |
| edm::EventSetup const & | |||
| ) | [private, virtual] |
| void HcalForwardLibWriter::endJob | ( | void | ) | [private, virtual] |
| void HcalForwardLibWriter::endLuminosityBlock | ( | edm::LuminosityBlock & | , |
| edm::EventSetup const & | |||
| ) | [private, virtual] |
| void HcalForwardLibWriter::endRun | ( | edm::Run & | , |
| edm::EventSetup const & | |||
| ) | [private, virtual] |
| void HcalForwardLibWriter::fillDescriptions | ( | edm::ConfigurationDescriptions & | descriptions | ) | [static] |
Reimplemented from edm::EDProducer.
Definition at line 184 of file HcalForwardLibWriter.cc.
References edm::ConfigurationDescriptions::addDefault(), and edm::ParameterSetDescription::setUnknown().
{
edm::ParameterSetDescription desc;
desc.setUnknown();
descriptions.addDefault(desc);
}
| void HcalForwardLibWriter::produce | ( | edm::Event & | iEvent, |
| const edm::EventSetup & | iSetup | ||
| ) | [private, virtual] |
Implements edm::EDProducer.
Definition at line 42 of file HcalForwardLibWriter.cc.
References gather_cfg::cout, Exception, fFile, fFileHandle, fTree, i, n, pos, edm::Event::put(), AlCaHLTBitMon_QueryRunRegistry::string, lumiQTWidget::t, x, detailsBasic3DVector::y, and z.
{
std::vector<double> energyBin;
double energyBin2[12] = {10,15,20,35,50,80,100,150,250,350,500,1000};
for (int z = 0; z< 12; ++z) {energyBin.push_back(energyBin2[z]);}
std::vector<HFShowerLibraryEventInfo> Info;
HFShowerPhotonCollection emColl;
std::vector<HFShowerPhotonCollection> emCollsss;
HFShowerPhotonCollection fakeColl;
HFShowerPhoton afakePhoton(1,3,5,8,480);
fakeColl.push_back(afakePhoton);
HFShowerPhotonCollection hadColl;
//HFShowerPhotonCollection emCollnphot;
//std::vector<HFShowerPhotonCollection> emCollnphot;
//std::vector<int> emCollnPhoton;
std::vector<int> emCollnPhoton;
std::vector<int> hadCollnPhoton;
int n = fFileHandle.size();
for (int i = 0; i < n; ++i) {
std::string fn = fFileHandle[i].name;
std::cout<<fn<<std::endl;
std::string particle = fFileHandle[i].id;
/*
int momBin = fFileHandle[i].momentum;
energyBin.push_back(momBin);
*/
fFile = new TFile(fn.c_str(), "READ");
fTree = (TTree*) fFile->FindObjectAny("CherenkovPhotons");
if(!fTree){
throw cms::Exception("NullPointer")
<< "Cannot find TTree with name CherenkovPhotons";
}
int nphot = 0;
float x[10000];
float y[10000];
float z[10000];
float t[10000];
float lambda[10000];
int fiberId[10000];
for (int kk = 0; kk < 10000; ++kk) {
x[kk] = 0.;
y[kk] = 0.;
z[kk] = 0.;
t[kk] = 0.;
lambda[kk] = 0.;
fiberId[kk] = 0;
}
fTree->SetBranchAddress("nphot", &nphot);
fTree->SetBranchAddress("x", &x);
fTree->SetBranchAddress("y", &y);
fTree->SetBranchAddress("z", &z);
fTree->SetBranchAddress("t", &t);
fTree->SetBranchAddress("lambda", &lambda);
fTree->SetBranchAddress("fiberId", &fiberId);
int nentry = int(fTree->GetEntries());
std::cout<<"nenetry " << nentry<<std::endl;
if (particle == "electron") {
for (int iev = 0; iev < nentry; iev++) {
fTree->GetEntry(iev);
std::cout<<"nphot "<<nphot<<std::endl;
emCollnPhoton.push_back(nphot);
for (int iph = 0; iph < nphot; ++iph) {
HFShowerPhoton::Point pos(x[iph], y[iph], z[iph]);
HFShowerPhoton aPhoton(pos, t[iph], lambda[iph]);
emColl.push_back(aPhoton);
}
emCollsss.push_back(emColl);
emColl.clear();
}
}
if (particle == "pion") {
for (int iev = 0; iev < nentry; iev++) {
fTree->GetEntry(iev);
hadCollnPhoton.push_back(nphot);
for (int iph = 0; iph < nphot; ++iph) {
HFShowerPhoton::Point pos(x[iph], y[iph], z[iph]);
HFShowerPhoton aPhoton(pos, t[iph], lambda[iph]);
hadColl.push_back(aPhoton);
}
}
}
}
HFShowerLibraryEventInfo aInfo((n/2)*5000,n/2,5000,1,1,energyBin);
Info.push_back(aInfo);
std::auto_ptr< std::vector<HFShowerLibraryEventInfo> > product_info(new std::vector<HFShowerLibraryEventInfo>(Info) );
std::auto_ptr<HFShowerPhotonCollection > product_em(new HFShowerPhotonCollection(fakeColl));
std::auto_ptr<std::vector<HFShowerPhotonCollection> > product_emsss(new std::vector<HFShowerPhotonCollection>(emCollsss));
std::cout<<"em coll size "<<emCollsss.size()<<std::endl;
//std::auto_ptr< std::vector<int> > product_em_nphot(new std::vector<int>(emCollnPhoton));
std::auto_ptr<HFShowerPhotonCollection> product_had(new HFShowerPhotonCollection(hadColl));
//std::auto_ptr<std::vector<int> > product_had_nphot(new std::vector<int>(hadCollnPhoton));
iEvent.put(product_info, "HFShowerLibraryEventInfo");
iEvent.put(product_emsss, "emParticles");
iEvent.put(product_had, "hadParticles");
//iEvent.put(product_em_nphot, "emParticles");
//iEvent.put(product_had_nphot, "hadParticles");
}
| void HcalForwardLibWriter::readUserData | ( | ) | [private] |
Definition at line 158 of file HcalForwardLibWriter.cc.
References gather_cfg::cout, Exception, fDataFile, fFileHandle, HcalForwardLibWriter::FileHandle::id, LaserDQM_cfg::input, HcalForwardLibWriter::FileHandle::momentum, HcalForwardLibWriter::FileHandle::name, and AlCaHLTBitMon_QueryRunRegistry::string.
Referenced by HcalForwardLibWriter().
{
std::cout << " using " <<std::endl;
std::ifstream input(fDataFile.c_str());
if (input.fail()) {
throw cms::Exception("MissingFile")
<< "Cannot find file" << fDataFile.c_str();
}
std::string fFileName, fPID;
int fMom;
while (!input.eof()) {
input >> fFileName >> fPID >> fMom;
std::cout << " using " << fFileName << " fPID" << fPID << " fMom" << fMom << std::endl;
if (!input.fail()) {
std::cout << " using " << fFileName << " " << fPID << " " << fMom << std::endl;
FileHandle aFile;
aFile.name = fFileName;
aFile.id = fPID;
aFile.momentum = fMom;
fFileHandle.push_back(aFile);
} else {
input.clear();
}
input.ignore(999, '\n');
}
}
std::string HcalForwardLibWriter::fDataFile [private] |
Definition at line 78 of file HcalForwardLibWriter.h.
Referenced by HcalForwardLibWriter(), and readUserData().
TFile* HcalForwardLibWriter::fFile [private] |
Definition at line 80 of file HcalForwardLibWriter.h.
Referenced by produce().
std::vector<FileHandle> HcalForwardLibWriter::fFileHandle [private] |
Definition at line 79 of file HcalForwardLibWriter.h.
Referenced by produce(), and readUserData().
TTree* HcalForwardLibWriter::fTree [private] |
Definition at line 81 of file HcalForwardLibWriter.h.
Referenced by produce().
1.7.3