/data/refman/pasoursint/CMSSW_6_1_2_SLHC2_patch1/src/PhysicsTools/PatAlgos/plugins/PATElectronProducer.h

Go to the documentation of this file.

//
// $Id: PATElectronProducer.h,v 1.30 2012/10/02 22:19:13 beaudett Exp $
//

#ifndef PhysicsTools_PatAlgos_PATElectronProducer_h
#define PhysicsTools_PatAlgos_PATElectronProducer_h

#include "FWCore/Framework/interface/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "DataFormats/Common/interface/View.h"
#include "DataFormats/Common/interface/ValueMap.h"
#include "DataFormats/Candidate/interface/CandAssociation.h"

#include "CommonTools/Utils/interface/PtComparator.h"

#include "PhysicsTools/PatAlgos/interface/MultiIsolator.h"
#include "PhysicsTools/PatAlgos/interface/EfficiencyLoader.h"
#include "PhysicsTools/PatAlgos/interface/KinResolutionsLoader.h"

#include "DataFormats/PatCandidates/interface/Electron.h"

#include "DataFormats/PatCandidates/interface/UserData.h"
#include "PhysicsTools/PatAlgos/interface/PATUserDataHelper.h"

#include "RecoEcal/EgammaCoreTools/interface/EcalClusterLazyTools.h"
#include "TrackingTools/TransientTrack/interface/TransientTrack.h"
#include "DataFormats/VertexReco/interface/Vertex.h"

#include "Geometry/CaloTopology/interface/CaloSubdetectorTopology.h"

#include <string>


namespace pat {


  class TrackerIsolationPt;
  class CaloIsolationEnergy;
  class LeptonLRCalc;


  class PATElectronProducer : public edm::EDProducer {

    public:

      explicit PATElectronProducer(const edm::ParameterSet & iConfig);
      ~PATElectronProducer();

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

      static void fillDescriptions(edm::ConfigurationDescriptions & descriptions);

    private:

      // configurables
      edm::InputTag electronSrc_;
      bool          embedGsfElectronCore_;
      bool          embedGsfTrack_;
      bool          embedSuperCluster_;
      bool          embedSeedCluster_;
      bool          embedTrack_;
      bool          addGenMatch_;
      bool          embedGenMatch_;
      bool          embedRecHits_;
      
      std::vector<edm::InputTag> genMatchSrc_;

      bool          useParticleFlow_;
      edm::InputTag pfElecSrc_;
      edm::InputTag pfCandidateMap_;
      bool          embedPFCandidate_;

      edm::InputTag reducedBarrelRecHitCollection_;
      edm::InputTag reducedEndcapRecHitCollection_;
 
      bool          embedHighLevelSelection_;
      edm::InputTag beamLineSrc_;
      bool          usePV_;
      edm::InputTag pvSrc_;

      typedef std::vector<edm::Handle<edm::Association<reco::GenParticleCollection> > > GenAssociations;
      typedef edm::RefToBase<reco::GsfElectron> ElectronBaseRef;
      typedef std::vector< edm::Handle< edm::ValueMap<IsoDeposit> > > IsoDepositMaps;
      typedef std::vector< edm::Handle< edm::ValueMap<double> > > IsolationValueMaps;


      void fillElectron( Electron& aElectron,
                         const ElectronBaseRef& electronRef,
                         const reco::CandidateBaseRef& baseRef,
                         const GenAssociations& genMatches,
                         const IsoDepositMaps& deposits,
                         const bool pfId,
                         const IsolationValueMaps& isolationValues,
                         const IsolationValueMaps& isolationValuesNoPFId) const;

      void fillElectron2( Electron& anElectron,
                          const reco::CandidatePtr& candPtrForIsolation,
                          const reco::CandidatePtr& candPtrForGenMatch,
                          const reco::CandidatePtr& candPtrForLoader,
                          const GenAssociations& genMatches,
                          const IsoDepositMaps& deposits,
                          const IsolationValueMaps& isolationValues ) const;

    // embed various impact parameters with errors
    // embed high level selection
    void embedHighLevel( pat::Electron & anElectron,
                         reco::GsfTrackRef track,
                         reco::TransientTrack & tt,
                         reco::Vertex & primaryVertex,
                         bool primaryVertexIsValid,
                         reco::BeamSpot & beamspot,
                         bool beamspotIsValid );

      typedef std::pair<pat::IsolationKeys,edm::InputTag> IsolationLabel;
      typedef std::vector<IsolationLabel> IsolationLabels;

      void readIsolationLabels( const edm::ParameterSet & iConfig,
                                const char* psetName,
                                IsolationLabels& labels);

      bool          addElecID_;
      typedef std::pair<std::string, edm::InputTag> NameTag;
      std::vector<NameTag> elecIDSrcs_;

      // tools
      GreaterByPt<Electron>       pTComparator_;

      pat::helper::MultiIsolator isolator_;
      pat::helper::MultiIsolator::IsolationValuePairs isolatorTmpStorage_; // better here than recreate at each event
      IsolationLabels isoDepositLabels_;
      IsolationLabels isolationValueLabels_;
      IsolationLabels isolationValueLabelsNoPFId_;

      bool addEfficiencies_;
      pat::helper::EfficiencyLoader efficiencyLoader_;

      bool addResolutions_;
      pat::helper::KinResolutionsLoader resolutionLoader_;

      bool useUserData_;
      pat::PATUserDataHelper<pat::Electron>      userDataHelper_;

      const CaloTopology * ecalTopology_;

  };


}

#endif