PFCandidateMixer Class Reference

#include <MyAna/PFCandidateMixer/src/PFCandidateMixer.cc>

Inheritance diagram for PFCandidateMixer:

Public Member Functions
	PFCandidateMixer (const edm::ParameterSet &)
	~PFCandidateMixer ()
Public Member Functions inherited from edm::EDProducer
	EDProducer ()
virtual	~EDProducer ()
Public Member Functions inherited from edm::ProducerBase
	ProducerBase ()
void	registerProducts (ProducerBase *, ProductRegistry *, ModuleDescription const &)
std::function< void(BranchDescription const &)>	registrationCallback () const
	used by the fwk to register list of products More...
virtual	~ProducerBase ()
Public Member Functions inherited from edm::EDConsumerBase
	EDConsumerBase ()
ProductHolderIndex	indexFrom (EDGetToken, BranchType, TypeID const &) const
void	itemsMayGet (BranchType, std::vector< ProductHolderIndex > &) const
void	itemsToGet (BranchType, std::vector< ProductHolderIndex > &) const
void	labelsForToken (EDGetToken iToken, Labels &oLabels) const
void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)
virtual	~EDConsumerBase ()

Private Member Functions
virtual void	beginJob ()
virtual void	endJob ()
void	mix (edm::Event &iEvent, const edm::Handle< reco::TrackCollection > &trackCol, const edm::Handle< reco::MuonCollection > &muonCol, const edm::Handle< reco::GsfElectronCollection > &electronCol, const reco::PFCandidateCollection &pfIn1, const reco::PFCandidateCollection &pfIn2)
virtual void	produce (edm::Event &, const edm::EventSetup &)

Private Attributes
edm::InputTag	_col1
edm::InputTag	_col2
edm::InputTag	_electronCol
edm::InputTag	_muonCol
edm::InputTag	_trackCol

Additional Inherited Members
Public Types inherited from edm::EDProducer
typedef EDProducer	ModuleType
typedef WorkerT< EDProducer >	WorkerType
Public Types inherited from edm::ProducerBase
typedef ProductRegistryHelper::TypeLabelList	TypeLabelList
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::EDProducer
CurrentProcessingContext const *	currentContext () const
Protected Member Functions inherited from edm::ProducerBase
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)
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: <one line="" class="" summary>="">

Implementation: <Notes on="" implementation>="">

Definition at line 52 of file PFCandidateMixer.cc.

Constructor & Destructor Documentation

PFCandidateMixer::PFCandidateMixer ( const edm::ParameterSet &  iConfig )

explicit

Definition at line 85 of file PFCandidateMixer.cc.

References _electronCol, _muonCol, edm::InputTag::label(), and edm::ProductRegistryHelper::produces().

                                                                :
   _col1(iConfig.getUntrackedParameter<edm::InputTag>("col1") ),
   _col2(iConfig.getUntrackedParameter<edm::InputTag>("col2") ),
   _trackCol(iConfig.getUntrackedParameter<edm::InputTag>("trackCol") ),
   _muonCol(iConfig.getUntrackedParameter<edm::InputTag>("muons") ),
   _electronCol(iConfig.getUntrackedParameter<edm::InputTag>("gsfElectrons"))
{
   produces< std::vector< reco::PFCandidate >  >(); 

   if(!_electronCol.label().empty())
      produces< edm::ValueMap<edm::Ptr<reco::PFCandidate> > >("electrons"); 
#if 0
   if(!_muonCol.label().empty())
      produces< edm::ValueMap<edm::Ptr<reco::PFCandidate> > >("muons"); 
#endif
}

PFCandidateMixer::~PFCandidateMixer

(

)

Definition at line 102 of file PFCandidateMixer.cc.

{
 
   // do anything here that needs to be done at desctruction time
   // (e.g. close files, deallocate resources etc.)

}

Member Function Documentation

void PFCandidateMixer::beginJob ( void  )

privatevirtual

Reimplemented from edm::EDProducer.

Definition at line 268 of file PFCandidateMixer.cc.

{
}

void PFCandidateMixer::endJob ( void  )

privatevirtual

Reimplemented from edm::EDProducer.

Definition at line 274 of file PFCandidateMixer.cc.

                         {
}

void PFCandidateMixer::mix ( edm::Event &  iEvent, const edm::Handle< reco::TrackCollection > &  trackCol, const edm::Handle< reco::MuonCollection > &  muonCol, const edm::Handle< reco::GsfElectronCollection > &  electronCol, const reco::PFCandidateCollection &  pfIn1, const reco::PFCandidateCollection &  pfIn2  )

private

Definition at line 142 of file PFCandidateMixer.cc.

References gather_cfg::cout, reco::deltaR(), reco::PFCandidate::e, edm::helper::Filler< Map >::fill(), newFWLiteAna::found, edm::Event::getProvenance(), i, edm::helper::Filler< Map >::insert(), edm::HandleBase::isValid(), edm::Provenance::moduleLabel(), edm::Provenance::processName(), edm::Provenance::productInstanceName(), edm::Event::put(), L1Trigger_dataformats::reco, GlobalPosition_Frontier_DevDB_cff::tag, and makeHLTPrescaleTable::values.

Referenced by produce().

{
   using namespace edm;
   using namespace reco;

   std::vector<const reco::PFCandidateCollection*> colVec;
   
   colVec.push_back(&pfIn1);
   colVec.push_back(&pfIn2);

   std::auto_ptr<std::vector< reco::PFCandidate > > pOut(new std::vector< reco::PFCandidate  > );
   
   std::vector<const reco::PFCandidateCollection*>::iterator itCol= colVec.begin();
   std::vector<const reco::PFCandidateCollection*>::iterator itColE= colVec.end();

   std::map<reco::GsfElectronRef, reco::PFCandidatePtr> electronCandidateMap; 

   int iCol = 0;
   for (;itCol!=itColE; ++itCol){
     PFCandidateConstIterator it = (*itCol)->begin();
     PFCandidateConstIterator itE = (*itCol)->end();
     for (;it!=itE;++it) {
      edm::Ptr<reco::PFCandidate> candPtr(*itCol, it - (*itCol)->begin());
      reco::PFCandidate cand(*it);
      //reco::PFCandidate cand(candPtr); // This breaks the output module, I'm not sure why
       
       //const bool isphoton   = cand.particleId() == reco::PFCandidate::gamma && cand.mva_nothing_gamma()>0.;
       const bool iselectron = cand.particleId() == reco::PFCandidate::e;
       //const bool hasNonNullMuonRef  = cand.muonRef().isNonnull();

       // if it is an electron. Find the GsfElectron with the same GsfTrack
       if (electronCol.isValid() && iselectron) {
         const reco::GsfTrackRef& gsfTrackRef(cand.gsfTrackRef());
         GsfElectronEqual myEqual(gsfTrackRef);
         std::vector<reco::GsfElectron>::const_iterator itcheck=find_if(electronCol->begin(), electronCol->end(), myEqual);
         if(itcheck==electronCol->end())
           throw cms::Exception("PFCandidateMixer") << "GsfElectron for candidate not found!";

         reco::GsfElectronRef electronRef(electronCol, itcheck - electronCol->begin());
         cand.setGsfElectronRef(electronRef);
         cand.setSuperClusterRef(electronRef->superCluster());
         electronCandidateMap[electronRef] = candPtr;
       }

       // TODO: Do the same for muons -- see PFLinker.cc, we just don't have the MuonToMuonMap available.
       // But it might work with the pure muons collections as well.

       size_t i = 0;
       bool found = false;
       double minDR = 9999.;
       int iMinDr = -1;
       if (it->trackRef().isNonnull()) {
         for ( i = 0 ; i < trackCol->size(); ++i){
           if ( reco::deltaR( *(it->trackRef()), (*trackCol)[i] )<0.001 ) {
                found = true;
                break; 
           }
           double dr = reco::deltaR( *(it->trackRef()), (*trackCol)[i] );
           if ( dr < minDR) {
              iMinDr = i;
              minDR = dr;
           } 
         } 
       } 
       if ( found ){ // ref was found, overwrite in PFCand
         reco::TrackRef trref(trackCol,i);
         cand.setTrackRef(trref);
         //std::cout << " YY track ok"<<std::endl;

       } else { // keep orginall ref
         if (it->trackRef().isNonnull()) {
           std::cout << " XXXXXXXXXXX track not found " 
                 << " col " << iCol
                 << " ch " << it->charge()
                 << " id " << it->pdgId() 
                 << " pt " << it->pt() 
                 << " track: eta " << it->trackRef()->eta()
                 << " pt:  " << it->trackRef()->pt()
                 << " charge:  " << it->trackRef()->charge()
                 <<  std::endl;
           std::cout << " minDR=" << minDR << std::endl; 
           if ( iMinDr >= 0 ) {
                std::cout 
                     << " closest track pt=" << (*trackCol)[iMinDr].pt()
                     << " ch=" << (*trackCol)[iMinDr].charge()
                     <<  std::endl; 
           } 
           edm::Provenance prov=iEvent.getProvenance(it->trackRef().id());
           edm::InputTag tag(prov.moduleLabel(),  prov.productInstanceName(),   prov.processName());
           std::cout << " trackref in PFCand came from: "   << tag.encode() << std::endl;
         }
       }
       pOut->push_back(cand);
     }
     ++iCol;
   }

   edm::OrphanHandle<reco::PFCandidateCollection> newColl = iEvent.put(pOut);

   // Now fixup the references and write the valuemap
   if(electronCol.isValid())
   {
      std::vector<reco::PFCandidatePtr> values(electronCol->size());
      for(unsigned int i = 0; i < electronCol->size(); ++i)
      {
         edm::Ref<reco::GsfElectronCollection> objRef(electronCol, i);
         std::map<reco::GsfElectronRef, reco::PFCandidatePtr>::const_iterator iter = electronCandidateMap.find(objRef);

         reco::PFCandidatePtr candPtr;
         if(iter != electronCandidateMap.end())
            candPtr = reco::PFCandidatePtr(newColl, iter->second.key());
         values[i] = candPtr;
      }

      std::auto_ptr<edm::ValueMap<reco::PFCandidatePtr> > pfMap_p(new edm::ValueMap<reco::PFCandidatePtr>());
      edm::ValueMap<reco::PFCandidatePtr>::Filler filler(*pfMap_p);
      filler.insert(electronCol, values.begin(), values.end());
      filler.fill();
      iEvent.put(pfMap_p, "electrons");
   }

   // TODO: Do the same for muons
}

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

privatevirtual

Implements edm::EDProducer.

Definition at line 117 of file PFCandidateMixer.cc.

References _col1, _col2, _electronCol, _muonCol, _trackCol, edm::Event::getByLabel(), edm::InputTag::label(), and mix().

{
   const bool muonColGiven = !_muonCol.label().empty();
   const bool electronColGiven = !_electronCol.label().empty();

   edm::Handle< std::vector<reco::Track> > trackCol;
   edm::Handle<reco::MuonCollection> muonCol;
   edm::Handle<reco::GsfElectronCollection> electronCol;
   iEvent.getByLabel( _trackCol, trackCol);

   if(muonColGiven)
      if(!iEvent.getByLabel(_muonCol, muonCol))
         throw cms::Exception("PFCandidateMixer") << "Muon Collection not found!";
   if(electronColGiven)
      if(!iEvent.getByLabel(_electronCol, electronCol))
         throw cms::Exception("PFCandidateMixer") << "GsfElectron Collection not found!";

   edm::Handle<reco::PFCandidateCollection> pfIn1;
   edm::Handle<reco::PFCandidateCollection> pfIn2;
   iEvent.getByLabel(_col1, pfIn1);
   iEvent.getByLabel(_col2, pfIn2);

   mix(iEvent, trackCol, muonCol, electronCol, *pfIn1, *pfIn2);
}

Member Data Documentation

edm::InputTag PFCandidateMixer::_col1

private

Definition at line 64 of file PFCandidateMixer.cc.

Referenced by produce().

edm::InputTag PFCandidateMixer::_col2

private

Definition at line 65 of file PFCandidateMixer.cc.

Referenced by produce().

edm::InputTag PFCandidateMixer::_electronCol

private

Definition at line 69 of file PFCandidateMixer.cc.

Referenced by PFCandidateMixer(), and produce().

edm::InputTag PFCandidateMixer::_muonCol

private

Definition at line 68 of file PFCandidateMixer.cc.

Referenced by PFCandidateMixer(), and produce().

edm::InputTag PFCandidateMixer::_trackCol

private

Definition at line 67 of file PFCandidateMixer.cc.

Referenced by produce().