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

ModuleDescription const &	moduleDescription () const

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

ProductHolderIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

std::vector < ProductHolderIndexAndSkipBit > const &	itemsToGetFromEvent () const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) 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

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::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.

269 {

270 }

void PFCandidateMixer::endJob ( void )

privatevirtual

Reimplemented from edm::EDProducer.

Definition at line 274 of file PFCandidateMixer.cc.

274 {

275 }

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(), dt_dqm_sourceclient_common_cff::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().

Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation