RawDataMapperByLabel.cc

Go to the documentation of this file.

#include "DataFormats/Provenance/interface/ProcessHistory.h" 
#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/FEDRawData/interface/FEDRawDataCollection.h" 
#include "DataFormats/Common/interface/Handle.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Utilities/interface/InputTag.h"

#include <iostream>

using namespace edm;

class RawDataMapperByLabel: public edm::stream::EDProducer<> {
public:
    
    RawDataMapperByLabel(const edm::ParameterSet& pset);
    
    ~RawDataMapperByLabel() override;
    
    void produce(edm::Event & e, const edm::EventSetup& c) override; 
    
    static void fillDescriptions(edm::ConfigurationDescriptions &);     
private:
    
    typedef std::vector<edm::InputTag>::const_iterator tag_iterator_t;
    typedef std::vector<edm::EDGetTokenT<FEDRawDataCollection> >::const_iterator tok_iterator_t;

    std::vector<edm::InputTag> inputTags_ ;
    std::vector<edm::EDGetTokenT<FEDRawDataCollection> > inputTokens_;
    
    edm::InputTag mainCollectionTag_ ;
    edm::InputTag filledCollectionName_;
    bool firstEvent_;
    

};


RawDataMapperByLabel::RawDataMapperByLabel(const edm::ParameterSet& pset)
 : inputTags_(pset.getParameter<std::vector<edm::InputTag>>("rawCollectionList")),
   mainCollectionTag_(pset.getParameter<edm::InputTag>("mainCollection")),
   filledCollectionName_(edm::InputTag("")),
   firstEvent_(true)
{

  inputTokens_.reserve(inputTags_.size());
  for(auto const& inputTag: inputTags_) {
    inputTokens_.push_back(consumes<FEDRawDataCollection>(inputTag));
  }

  produces<FEDRawDataCollection>();
   
}


RawDataMapperByLabel::~RawDataMapperByLabel(){

}


void RawDataMapperByLabel::produce(Event & e, const EventSetup& c){
 
 bool alreadyACollectionFilled= false;
 tag_iterator_t inputTag = inputTags_.begin();
 for(tok_iterator_t inputTok = inputTokens_.begin(); inputTok != inputTokens_.end(); ++inputTok, ++inputTag  ) {
   Handle<FEDRawDataCollection> input;
   if(e.getByToken(*inputTok,input)){
      if(input.isValid()){
         if(alreadyACollectionFilled) throw cms::Exception("BadInput") << "Two input collections are present." << std::endl
         << "Please make sure that the input dataset has only one FEDRawDataCollector collection filled";
         
         if(firstEvent_){  
            filledCollectionName_ = *inputTag; 
            alreadyACollectionFilled = true;
            firstEvent_= false;  
         }
            
         
         if(!(filledCollectionName_==*inputTag)) throw cms::Exception("BadInput") << "The filled collection has changed!";
            
         if(!(mainCollectionTag_==filledCollectionName_)) e.put(std::make_unique<FEDRawDataCollection>(*input.product()));
                      
      }
   }
 }
}

void RawDataMapperByLabel::fillDescriptions(edm::ConfigurationDescriptions & descriptions) {
    edm::ParameterSetDescription desc;
      
    desc.add<std::vector<edm::InputTag>>("rawCollectionList", {{"rawDataCollector","","@skipCurrentProcess"}, {"rawDataRepacker"}, {"rawDataReducedFormat"}});
    desc.add<edm::InputTag>("mainCollection", edm::InputTag("rawDataCollector"));
    
    descriptions.add("rawDataMapperByLabel", desc);
}

DEFINE_FWK_MODULE(RawDataMapperByLabel);