#include <L1Trigger/MaskedRctInputDigiProducer/src/MaskedRctInputDigiProducer.cc>

Inheritance diagram for MaskedRctInputDigiProducer:

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

	~MaskedRctInputDigiProducer ()

Public Member Functions inherited from edm::EDProducer
	EDProducer ()

ModuleDescription const &	moduleDescription () const

virtual	~EDProducer ()

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

std::vector< edm::ProductResolverIndex > const &	indiciesForPutProducts (BranchType iBranchType) const

	ProducerBase ()

void	registerProducts (ProducerBase , ProductRegistry , ModuleDescription const &)

std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...

void	resolvePutIndicies (BranchType iBranchType, std::unordered_multimap< std::string, edm::ProductResolverIndex > const &iIndicies, std::string const &moduleLabel)

virtual	~ProducerBase () noexcept(false)

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

	EDConsumerBase ()

	EDConsumerBase (EDConsumerBase const &)=delete

	EDConsumerBase (EDConsumerBase &&)=default

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 &	itemsToGetFromEvent () const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, 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	updateLookup (BranchType iBranchType, ProductResolverIndexHelper const &, bool iPrefetchMayGet)

virtual	~EDConsumerBase () noexcept(false)

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_

Additional Inherited Members
Public Types inherited from edm::EDProducer
typedef EDProducer	ModuleType

Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList

Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels

Static Public Member Functions inherited from edm::EDProducer
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)

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 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)

Detailed Description

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 45 of file MaskedRctInputDigiProducer.h.

Constructor & Destructor Documentation

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.)
 
 }

Member Function Documentation

void MaskedRctInputDigiProducer::endJob ( void )

privatevirtual

Reimplemented from edm::EDProducer.

Definition at line 385 of file MaskedRctInputDigiProducer.cc.

References DEFINE_FWK_MODULE.

385 {

386 }

void MaskedRctInputDigiProducer::produce	(	edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

privatevirtual

Definition at line 53 of file MaskedRctInputDigiProducer.cc.

References funct::abs(), MessageLogger_cfi::cerr, digitizers_cfi::ecal, ecalDigisLabel_, EcalTriggerTower, Exception, edm::FileInPath::fullPath(), edm::Event::getByLabel(), digitizers_cfi::hcal, hcalDigisLabel_, i, edm::HandleBase::isValid(), j, maskFile_, eostools::move(), hcalTTPDigis_cfi::presamples, edm::Event::put(), HcalTriggerPrimitiveDigi::setPresamples(), EcalTriggerPrimitiveDigi::setSample(), HcalTriggerPrimitiveDigi::setSample(), HcalTriggerPrimitiveDigi::setSize(), EcalTriggerPrimitiveDigi::setSize(), Validation_hcalonly_cfi::sign, edm::SortedCollection< T, SORT >::size(), AlCaHLTBitMon_QueryRunRegistry::string, 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::unique_ptr<ExampleData2> pOut(new ExampleData2(*pIn));
    iEvent.put(std::move(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::unique_ptr<EcalTrigPrimDigiCollection>
      maskedEcalTPs(new EcalTrigPrimDigiCollection());
    std::unique_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(std::move(maskedEcalTPs));
    iEvent.put(std::move(maskedHcalTPs));
 
 }