|
|
|
#include <L1Trigger/MaskedRctInputDigiProducer/src/MaskedRctInputDigiProducer.cc>
Public Member Functions | |
| MaskedRctInputDigiProducer (const edm::ParameterSet &) | |
| ~MaskedRctInputDigiProducer () | |
Private Member Functions | |
| virtual void | endJob () |
| virtual void | produce (edm::Event &, const edm::EventSetup &) |
Private Attributes | |
| edm::InputTag | ecalDigisLabel_ |
| edm::InputTag | hcalDigisLabel_ |
| edm::FileInPath | maskFile_ |
| bool | useEcal_ |
| bool | useHcal_ |
Description: Takes collections of ECAL and HCAL digis, masks some towers according to a mask in a text file, and creates new collections for use by the RCT.
Implementation: <Notes on="" implementation>="">
Definition at line 46 of file MaskedRctInputDigiProducer.h.
| MaskedRctInputDigiProducer::MaskedRctInputDigiProducer | ( | const edm::ParameterSet & | iConfig | ) | [explicit] |
Definition at line 15 of file MaskedRctInputDigiProducer.cc.
: useEcal_(iConfig.getParameter<bool>("useEcal")), useHcal_(iConfig.getParameter<bool>("useHcal")), ecalDigisLabel_(iConfig.getParameter<edm::InputTag>("ecalDigisLabel")), hcalDigisLabel_(iConfig.getParameter<edm::InputTag>("hcalDigisLabel")), maskFile_(iConfig.getParameter<edm::FileInPath>("maskFile")) { //register your products /* Examples produces<ExampleData2>(); //if do put with a label produces<ExampleData2>("label"); */ produces<EcalTrigPrimDigiCollection>(); produces<HcalTrigPrimDigiCollection>(); //now do what ever other initialization is needed }
| MaskedRctInputDigiProducer::~MaskedRctInputDigiProducer | ( | ) |
Definition at line 38 of file MaskedRctInputDigiProducer.cc.
{
// do anything here that needs to be done at desctruction time
// (e.g. close files, deallocate resources etc.)
}
| void MaskedRctInputDigiProducer::endJob | ( | void | ) | [private, virtual] |
Reimplemented from edm::EDProducer.
Definition at line 385 of file MaskedRctInputDigiProducer.cc.
{
}
| void MaskedRctInputDigiProducer::produce | ( | edm::Event & | iEvent, |
| const edm::EventSetup & | iSetup | ||
| ) | [private, virtual] |
Implements edm::EDProducer.
Definition at line 53 of file MaskedRctInputDigiProducer.cc.
References abs, benchmark_cfg::cerr, patCandidatesForDimuonsSequences_cff::ecal, ecalDigisLabel_, EcalTriggerTower, relval_parameters_module::energy, Exception, edm::FileInPath::fullPath(), edm::Event::getByLabel(), patCandidatesForDimuonsSequences_cff::hcal, hcalDigisLabel_, i, j, maskFile_, edm::Event::put(), HcalTriggerPrimitiveDigi::setPresamples(), EcalTriggerPrimitiveDigi::setSample(), HcalTriggerPrimitiveDigi::setSample(), EcalTriggerPrimitiveDigi::setSize(), HcalTriggerPrimitiveDigi::setSize(), useEcal_, and useHcal_.
{
using namespace edm;
/* This is an event example
//Read 'ExampleData' from the Event
Handle<ExampleData> pIn;
iEvent.getByLabel("example",pIn);
//Use the ExampleData to create an ExampleData2 which
// is put into the Event
std::auto_ptr<ExampleData2> pOut(new ExampleData2(*pIn));
iEvent.put(pOut);
*/
edm::Handle<EcalTrigPrimDigiCollection> ecal;
edm::Handle<HcalTrigPrimDigiCollection> hcal;
if (useEcal_) { iEvent.getByLabel(ecalDigisLabel_, ecal); }
if (useHcal_) { iEvent.getByLabel(hcalDigisLabel_, hcal); }
EcalTrigPrimDigiCollection ecalColl;
HcalTrigPrimDigiCollection hcalColl;
if (ecal.isValid()) { ecalColl = *ecal; }
if (hcal.isValid()) { hcalColl = *hcal; }
/* this is an EventSetup example
//Read SetupData from the SetupRecord in the EventSetup
ESHandle<SetupData> pSetup;
iSetup.get<SetupRecord>().get(pSetup);
*/
ifstream maskFileStream(maskFile_.fullPath().c_str());
if (!maskFileStream.is_open())
{
throw cms::Exception("FileInPathError")
<< "RCT mask file not opened" << std::endl;;
}
// read and process file (transform mask to new position in phi to match digis)
//char junk[256];
std::string junk;
//char junk[32];
do
{
//maskFileStream.getline(junk, 256);
maskFileStream >> junk;
}
while (junk.compare("ECAL:") != 0);
// std::vector<std::vector<std::vector<unsigned short> > > temp(2,std::vector<std::vector<unsigned short> >(72,std::vector<unsigned short>(28,1)));
std::vector<std::vector<std::vector<unsigned short> > > ecalMask(2,std::vector<std::vector<unsigned short> >(72,std::vector<unsigned short>(28,1)));
std::vector<std::vector<std::vector<unsigned short> > > hcalMask(2,std::vector<std::vector<unsigned short> >(72,std::vector<unsigned short>(28,1)));
std::vector<std::vector<std::vector<unsigned short> > > hfMask(2,std::vector<std::vector<unsigned short> >(18,std::vector<unsigned short>(8,1)));
// read ECAL mask first
// loop through rct iphi
for (int i = 1; i <= 72; i++)
{
int phi_index = (72 + 18 - i) % 72; // transfrom from RCT coords
if (phi_index == 0) {phi_index = 72;}
//std::cout << "ecal phi index is " << phi_index << std::endl;
for (int j = 28; j >= 1; j--)
{
maskFileStream >> junk;
if (junk.compare("-") == 0)
{}
else if ((junk.compare("X") == 0) || (junk.compare("x") == 0))
{
ecalMask.at(0).at(phi_index-1).at(j-1) = 0;
}
else
{
std::cerr << "RCT mask producer: error -- unrecognized character" << std::endl;
}
}
for (int j = 1; j <= 28; j++)
{
maskFileStream >> junk;
if(junk.compare("-") == 0)
{}
else if((junk.compare("X") == 0) || (junk.compare("x") == 0))
{
ecalMask.at(1).at(phi_index-1).at(j-1) = 0;
}
else
{
std::cerr << "RCT mask producer: error -- unrecognized character" << std::endl;
}
}
}
//std::cout << "done reading ECAL" << std::endl;
maskFileStream >> junk;
if (junk.compare("HCAL:") != 0)
{
throw cms::Exception("FileInPathError")
<< "RCT mask producer: error reading ECAL mask" << std::endl;;
}
//std::cout << "Starting HCAL read" << std::endl;
// now read HCAL mask
// loop through rct iphi
for (int i = 1; i <= 72; i++)
{
int phi_index = (72 + 18 - i) % 72; // transfrom from RCT coords
if (phi_index == 0) {phi_index = 72;}
//std::cout << "hcal phi index is " << phi_index << std::endl;
for (int j = 28; j >= 1; j--)
{
maskFileStream >> junk;
if (junk.compare("-") == 0)
{}
else if ((junk.compare("X") == 0) || (junk.compare("x") == 0))
{
hcalMask.at(0).at(phi_index-1).at(j-1) = 0;
}
else
{
std::cerr << "RCT mask producer: error -- unrecognized character" << std::endl;
}
}
for (int j = 1; j <= 28; j++)
{
maskFileStream >> junk;
if(junk.compare("-") == 0)
{}
else if((junk.compare("X") == 0) || (junk.compare("x") == 0))
{
hcalMask.at(1).at(phi_index-1).at(j-1) = 0;
}
else
{
std::cerr << "RCT mask producer: error -- unrecognized character" << std::endl;
}
}
}
maskFileStream >> junk;
if (junk.compare("HF:") != 0)
{
throw cms::Exception("FileInPathError")
<< "RCT mask producer: error reading HCAL mask" << std::endl;;
}
// loop through the hf phi columns in file
for(int i = 0; i < 18; i++)
{
//std::cout << "iphi for hf file read is " << i << std::endl;
for (int j = 4; j >= 1; j--)
{
if (maskFileStream >> junk) {}
else { std::cerr << "RCT mask producer: error reading HF mask" << std::endl; }
if (junk.compare("-") == 0)
{}
else if ((junk.compare("X") == 0) || (junk.compare("x") == 0))
{
hfMask.at(0).at(i).at(j-1) = 0; // just save iphi as 0-17, transform later
}
else
{
std::cerr << "RCT mask producer: error -- unrecognized character" << std::endl;
}
}
for (int j = 1; j <= 4; j++)
{
if (maskFileStream >> junk) {}
else { std::cerr << "RCT mask producer: error reading HF mask" << std::endl; }
if (junk.compare("-") == 0)
{}
else if ((junk.compare("X") == 0) || (junk.compare("x") == 0))
{
hfMask.at(1).at(i).at(j-1) = 0;
}
else
{
std::cerr << "RCT mask producer: error -- unrecognized character" << std::endl;
}
}
}
//std::cout << "HF read done" << std::endl;
maskFileStream.close();
//std::cout << "file closed" << std::endl;
// apply mask
std::auto_ptr<EcalTrigPrimDigiCollection>
maskedEcalTPs(new EcalTrigPrimDigiCollection());
std::auto_ptr<HcalTrigPrimDigiCollection>
maskedHcalTPs(new HcalTrigPrimDigiCollection());
maskedEcalTPs->reserve(56*72);
maskedHcalTPs->reserve(56*72+18*8);
int nEcalSamples = 0;
int nHcalSamples = 0;
for (unsigned int i = 0; i < ecalColl.size(); i++)
{
nEcalSamples = ecalColl[i].size();
short ieta = (short) ecalColl[i].id().ieta();
unsigned short absIeta = (unsigned short) abs(ieta);
int sign = ieta / absIeta;
short iphi = (unsigned short) ecalColl[i].id().iphi();
//if (i < 20) {std::cout << "ieta is " << ieta << ", absIeta is " << absIeta
// << ", iphi is " << iphi << std::endl;}
EcalTriggerPrimitiveDigi
ecalDigi(EcalTrigTowerDetId(sign, EcalTriggerTower, absIeta, iphi));
ecalDigi.setSize(nEcalSamples);
for (int nSample = 0; nSample < nEcalSamples; nSample++)
{
int energy = 0;
bool fineGrain = false;
if (sign < 0)
{
//std::cout << "eta-: mask is " << ecalMask.at(0).at(iphi-1).at(absIeta-1) << std::endl;
energy = ecalMask.at(0).at(iphi-1).at(absIeta-1) * ecalColl[i].sample(nSample).compressedEt();
fineGrain = ecalMask.at(0).at(iphi-1).at(absIeta-1) * ecalColl[i].sample(nSample).fineGrain();
}
else if (sign > 0)
{
//std::cout << "eta+: mask is " << ecalMask.at(1).at(iphi-1).at(absIeta-1) << std::endl;
energy = ecalMask.at(1).at(iphi-1).at(absIeta-1) * ecalColl[i].sample(nSample).compressedEt();
fineGrain = ecalMask.at(1).at(iphi-1).at(absIeta-1) * ecalColl[i].sample(nSample).fineGrain();
}
ecalDigi.setSample(nSample, EcalTriggerPrimitiveSample(energy,
fineGrain,
0));
}
maskedEcalTPs->push_back(ecalDigi);
}
//std::cout << "End of ecal digi masking" << std::endl;
//std::cout << "nHcalDigis is " << hcalColl.size() << std::endl;
for (unsigned int i = 0; i < hcalColl.size(); i++)
{
nHcalSamples = hcalColl[i].size();
//if ((i % 100) == 0 ) {std::cout << "Loop " << i << std::endl;}
short ieta = (short) hcalColl[i].id().ieta();
unsigned short absIeta = (unsigned short) abs(ieta);
int sign = ieta / absIeta;
short iphi = (unsigned short) hcalColl[i].id().iphi();
//if (i < 20) {std::cout << "ieta is " << ieta << ", absIeta is " << absIeta
// << ", iphi is " << iphi << std::endl;}
/*
if (hcalColl[i].SOI_compressedEt() != 0)
{
std::cout << "original et " << hcalColl[i].SOI_compressedEt()
<< " fg " << hcalColl[i].SOI_fineGrain() << " iphi "
<< iphi << " ieta " << ieta << std::endl;
}
*/
HcalTriggerPrimitiveDigi
hcalDigi(HcalTrigTowerDetId(ieta,iphi));
hcalDigi.setSize(nHcalSamples);
hcalDigi.setPresamples(hcalColl[i].presamples());
for (int nSample = 0; nSample < nHcalSamples; nSample++)
{
int energy = 0;
bool fineGrain = false;
if (absIeta < 29)
{
if (sign < 0)
{
energy = hcalMask.at(0).at(iphi-1).at(absIeta-1) * hcalColl[i].sample(nSample).compressedEt();
fineGrain = hcalMask.at(0).at(iphi-1).at(absIeta-1) * hcalColl[i].sample(nSample).fineGrain();
}
else if (sign > 0)
{
energy = hcalMask.at(1).at(iphi-1).at(absIeta-1) * hcalColl[i].sample(nSample).compressedEt();
fineGrain = hcalMask.at(1).at(iphi-1).at(absIeta-1) * hcalColl[i].sample(nSample).fineGrain();
}
}
else if ((absIeta >= 29) && (absIeta <= 32))
{
//std::cout << "hf iphi: " << iphi << std::endl;
short hf_phi_index = iphi/4;
//iphi = iphi/4; // change from 1,5,9, etc to access vector positions
//std::cout << "hf phi index: " << hf_phi_index << std::endl;
if (sign < 0)
{
//std::cout << "ieta is " << ieta << ", absIeta is " << absIeta << ", iphi is " << iphi << std::endl;
//std::cout << "eta-: mask is " << hfMask.at(0).at(hf_phi_index).at(absIeta-29) << std::endl; // hf ieta 0-3
energy = hfMask.at(0).at(hf_phi_index).at(absIeta-29) * hcalColl[i].sample(nSample).compressedEt(); // for hf, iphi starts at 0
fineGrain = hfMask.at(0).at(hf_phi_index).at(absIeta-29) * hcalColl[i].sample(nSample).fineGrain();
}
else if (sign > 0)
{
//std::cout << "ieta is " << ieta << ", absIeta is " << absIeta << ", iphi is " << iphi << std::endl;
//std::cout << "eta+: mask is " << hfMask.at(1).at(hf_phi_index).at(absIeta-29) << std::endl;
energy = hfMask.at(1).at(hf_phi_index).at(absIeta-29) * hcalColl[i].sample(nSample).compressedEt();
fineGrain = hfMask.at(1).at(hf_phi_index).at(absIeta-29) * hcalColl[i].sample(nSample).fineGrain();
}
//iphi = iphi*4 + 1; // change back to original
//std::cout << "New hf iphi = " << iphi << std::endl;
}
hcalDigi.setSample(nSample, HcalTriggerPrimitiveSample(energy,
fineGrain,
0, 0));
//if (hcalDigi.SOI_compressedEt() != 0)
//{
// std::cout << "et " << hcalDigi.SOI_compressedEt()
// << "fg " << hcalDigi.SOI_fineGrain() << std::endl;
//}
}
maskedHcalTPs->push_back(hcalDigi);
}
//std::cout << "End of hcal digi masking" << std::endl;
// put new data into event
iEvent.put(maskedEcalTPs);
iEvent.put(maskedHcalTPs);
}
Definition at line 59 of file MaskedRctInputDigiProducer.h.
Referenced by produce().
Definition at line 60 of file MaskedRctInputDigiProducer.h.
Referenced by produce().
Definition at line 61 of file MaskedRctInputDigiProducer.h.
Referenced by produce().
bool MaskedRctInputDigiProducer::useEcal_ [private] |
Definition at line 57 of file MaskedRctInputDigiProducer.h.
Referenced by produce().
bool MaskedRctInputDigiProducer::useHcal_ [private] |
Definition at line 58 of file MaskedRctInputDigiProducer.h.
Referenced by produce().
1.7.3