HLTRegionalEcalResonanceFilter.h

Go to the documentation of this file.

//
// Original Authors:  Vladimir Litvine , Yong Yang
// Modified by: Joshua Hardenbrook 10-8-2014

// system include files
#include <memory>

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/stream/EDFilter.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "DataFormats/Common/interface/Handle.h"

#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"

#include "DataFormats/EcalRecHit/interface/EcalRecHit.h"
#include "DataFormats/EcalRecHit/interface/EcalRecHitCollections.h"

// Geometry
#include "Geometry/CaloEventSetup/interface/CaloTopologyRecord.h"
#include "Geometry/CaloTopology/interface/CaloTopology.h"
#include "Geometry/CaloTopology/interface/CaloSubdetectorTopology.h"
#include "Geometry/CaloGeometry/interface/CaloSubdetectorGeometry.h"
#include "Geometry/CaloGeometry/interface/CaloCellGeometry.h"
#include "Geometry/CaloGeometry/interface/CaloGeometry.h"
#include "Geometry/CaloTopology/interface/EcalEndcapTopology.h"
#include "Geometry/CaloTopology/interface/EcalBarrelTopology.h"
#include "Geometry/Records/interface/CaloGeometryRecord.h"

#include "DataFormats/EcalDetId/interface/EBDetId.h"
#include "DataFormats/EcalDetId/interface/EEDetId.h"
#include "DataFormats/DetId/interface/DetId.h"

#include "DataFormats/EgammaReco/interface/BasicCluster.h"
#include "DataFormats/EgammaReco/interface/BasicClusterFwd.h"
#include "RecoEcal/EgammaCoreTools/interface/PositionCalc.h"

// ES stuff
#include "DataFormats/EgammaReco/interface/PreshowerCluster.h"
#include "RecoEcal/EgammaClusterAlgos/interface/PreshowerClusterAlgo.h"
#include "Geometry/EcalAlgo/interface/EcalPreshowerGeometry.h"
#include "Geometry/CaloTopology/interface/EcalPreshowerTopology.h"
#include "DataFormats/EgammaReco/interface/PreshowerClusterFwd.h"

//
//Ecal status
#include "CondFormats/DataRecord/interface/EcalChannelStatusRcd.h"
#include "CondFormats/EcalObjects/interface/EcalChannelStatus.h"

// class declaration
//
#include <algorithm>
#include <utility>

#include "TLorentzVector.h"
#include <vector>

namespace edm {
  class ConfigurationDescriptions;
}

class HLTRegionalEcalResonanceFilter : public edm::stream::EDFilter<> 
{
   public:
      explicit HLTRegionalEcalResonanceFilter(const edm::ParameterSet&);
      ~HLTRegionalEcalResonanceFilter();

      virtual bool filter(edm::Event &, const edm::EventSetup&);
      static void fillDescriptions(edm::ConfigurationDescriptions & descriptions);
   private:
      // ----------member data ---------------------------
      void doSelection(int detector, const reco::BasicClusterCollection *clusterCollection,
               const EcalRecHitCollection *hitCollection,
               const EcalChannelStatus &channelStatus,
               const CaloSubdetectorTopology* topology_p,
               std::map<int, std::vector<EcalRecHit> > &RecHits5x5_clus,
               std::vector<int> &indCandClus, 
               std::vector<int> &indIsoClus, 
               std::vector<int> &indClusSelected  
               ); 
      
      
      void makeClusterES(float x, float y, float z,const CaloSubdetectorGeometry*&iSubGeom,CaloSubdetectorTopology*& topology_p);
      
      void calcPaircluster(const reco::BasicCluster &bc1, const reco::BasicCluster &bc2, float &mpair, float &ptpair,float &etapair,float &phipair); 
      
      bool checkStatusOfEcalRecHit(const EcalChannelStatus &channelStatus, const EcalRecHit &rh);
      
      void calcShowerShape(const reco::BasicCluster &bc, const EcalChannelStatus &channelStatus, const EcalRecHitCollection *recHits,const CaloSubdetectorTopology *topology_p, bool calc5x5, std::vector<EcalRecHit> &rechit5x5,float res[]);
      
      void convxtalid(int & , int &);
      int diff_neta_s(int,int);
      int diff_nphi_s(int,int);
      
      
      static float DeltaPhi(float phi1, float phi2); 
      static float GetDeltaR(float eta1, float eta2, float phi1, float phi2); 
      
      // Input hits & clusters
      edm::InputTag barrelHits_;
      edm::InputTag endcapHits_;
      edm::InputTag barrelClusters_;
      edm::InputTag endcapClusters_;

      edm::EDGetTokenT<EBRecHitCollection> barrelHitsToken_;
      edm::EDGetTokenT<EERecHitCollection> endcapHitsToken_;
      edm::EDGetTokenT<ESRecHitCollection> preshHitsToken_;
      edm::EDGetTokenT<reco::BasicClusterCollection> barrelClustersToken_;
      edm::EDGetTokenT<reco::BasicClusterCollection> endcapClustersToken_;
     
      std::string BarrelHits_;
      std::string EndcapHits_;
      std::string ESHits_;
      
      
      bool doSelBarrel_; 
  
      // EB region 1 
      double region1_Barrel_;
      double selePtGammaBarrel_region1_; 
      double selePtPairBarrel_region1_;
      double seleS4S9GammaBarrel_region1_;
      double seleIsoBarrel_region1_;

      // EB region 2
      double selePtGammaBarrel_region2_; 
      double selePtPairBarrel_region2_;
      double seleS4S9GammaBarrel_region2_;
      double seleIsoBarrel_region2_;

      double seleMinvMaxBarrel_;
      double seleMinvMinBarrel_;

      //      double selePtGamma_;
      //      double selePtPair_;
      //      double seleS4S9Gamma_;
      //      double seleIso_;
      double seleS9S25Gamma_;

      // EB Isolation Configuraiton
      double seleBeltDR_;
      double seleBeltDeta_;
      double ptMinForIsolation_; 

      bool removePi0CandidatesForEta_; 
      double massLowPi0Cand_; 
      double massHighPi0Cand_; 
      bool store5x5RecHitEB_;      

      bool doSelEndcap_; 

      // EE region 1 
      double region1_EndCap_;
      double selePtGammaEndCap_region1_; 
      double selePtPairEndCap_region1_;
      double seleS4S9GammaEndCap_region1_;
      double seleIsoEndCap_region1_;

      // EE region 2
      double region2_EndCap_;
      double selePtGammaEndCap_region2_; 
      double selePtPairEndCap_region2_;
      double seleS4S9GammaEndCap_region2_;
      double seleIsoEndCap_region2_;

      // EE region 3
      double selePtGammaEndCap_region3_; 
      double selePtPairEndCap_region3_;
      double selePtPairMaxEndCap_region3_;
      double seleS4S9GammaEndCap_region3_;
      double seleIsoEndCap_region3_;

      double seleMinvMaxEndCap_;
      double seleMinvMinEndCap_;
      // double seleS4S9GammaEndCap_; //old non-regional filter
      // double seleIsoEndCap_; //old non-regional filter

      double seleS9S25GammaEndCap_;

      // EE isolation configuration
      double seleBeltDREndCap_;
      double seleBeltDetaEndCap_;
      double ptMinForIsolationEndCap_; 
      bool store5x5RecHitEE_;
            
      bool useRecoFlag_; 
      bool useDBStatus_; 
      int flagLevelRecHitsToUse_; 
      int statusLevelRecHitsToUse_;      
        
      bool storeRecHitES_; 
      edm::InputTag preshHitProducer_;         // name of module/plugin/producer producing hits
      int preshNclust_;
      float preshClustECut;
      double etThresh_;
      double calib_planeX_;
      double calib_planeY_;
      double mip_;
      double gamma_;
      
      PreshowerClusterAlgo * presh_algo_; // algorithm doing the real work            
      
      std::map<DetId, EcalRecHit> m_esrechit_map;
      std::set<DetId> m_used_strips;      
      
      int debug_; 
};