CosmicsMuonIdProducer.cc

Go to the documentation of this file.

#include <string>
#include <vector>

#include "DataFormats/Common/interface/ValueMap.h"
#include "DataFormats/MuonReco/interface/MuonFwd.h"
#include "DataFormats/MuonReco/interface/Muon.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "RecoMuon/MuonIdentification/interface/MuonCosmicsId.h"

#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "RecoMuon/MuonIdentification/interface/MuonCosmicCompatibilityFiller.h"

class CosmicsMuonIdProducer : public edm::stream::EDProducer<> {
public:
  CosmicsMuonIdProducer(const edm::ParameterSet& iConfig) :
    inputMuonCollection_(iConfig.getParameter<edm::InputTag>("muonCollection")),
    inputTrackCollections_(iConfig.getParameter<std::vector<edm::InputTag> >("trackCollections"))
  {
    edm::ConsumesCollector iC = consumesCollector();
    compatibilityFiller_ = new MuonCosmicCompatibilityFiller(iConfig.getParameter<edm::ParameterSet>("CosmicCompFillerParameters"),iC);

    produces<edm::ValueMap<unsigned int> >().setBranchAlias("cosmicsVeto");
    produces<edm::ValueMap<reco::MuonCosmicCompatibility> >().setBranchAlias("cosmicCompatibility");

    muonToken_   = consumes<reco::MuonCollection>(inputMuonCollection_);
    for(unsigned int i=0;i<inputTrackCollections_.size();++i) 
      trackTokens_.push_back(consumes<reco::TrackCollection>(inputTrackCollections_.at(i)));

  }
  ~CosmicsMuonIdProducer() override {
    if (compatibilityFiller_)
      delete compatibilityFiller_;
  }

private:
  void produce(edm::Event&, const edm::EventSetup&) override;
  edm::InputTag inputMuonCollection_;
  std::vector<edm::InputTag> inputTrackCollections_;
  edm::EDGetTokenT<reco::MuonCollection> muonToken_;
  std::vector<edm::EDGetTokenT<reco::TrackCollection> > trackTokens_;
 

  MuonCosmicCompatibilityFiller *compatibilityFiller_;
};

void
CosmicsMuonIdProducer::produce(edm::Event& iEvent, const edm::EventSetup& iSetup)
{
  edm::Handle<reco::MuonCollection> muons;
  iEvent.getByToken(muonToken_, muons);
  // reserve some space
  std::vector<unsigned int> values;
  values.reserve(muons->size());

  std::vector<reco::MuonCosmicCompatibility> compValues;
  compValues.reserve(muons->size());
  
  for(reco::MuonCollection::const_iterator muon = muons->begin(); 
      muon != muons->end(); ++muon)
    {
      unsigned int foundPartner(0);
      if ( muon->innerTrack().isNonnull() ){
    for ( unsigned int i=0; i<inputTrackCollections_.size(); ++i )
      {
        edm::Handle<reco::TrackCollection> tracks;
        iEvent.getByToken(trackTokens_.at(i), tracks);
        if ( muonid::findOppositeTrack(tracks,*muon->innerTrack()).isNonnull() ){
          foundPartner = i+1;
          break;
        }
      }
      }
      values.push_back(foundPartner);

      compValues.push_back(compatibilityFiller_->fillCompatibility(*muon, iEvent, iSetup));
    }

  // create and fill value map
  auto out = std::make_unique<edm::ValueMap<unsigned int>>();
  edm::ValueMap<unsigned int>::Filler filler(*out);
  filler.insert(muons, values.begin(), values.end());
  filler.fill();


  auto outC = std::make_unique<edm::ValueMap<reco::MuonCosmicCompatibility>>();
  edm::ValueMap<reco::MuonCosmicCompatibility>::Filler fillerC(*outC);
  fillerC.insert(muons, compValues.begin(), compValues.end());
  fillerC.fill();

  // put value map into event
  iEvent.put(std::move(out));
  iEvent.put(std::move(outC));
}
DEFINE_FWK_MODULE(CosmicsMuonIdProducer);