BetaCalculatorECAL.h

Go to the documentation of this file.

// system include files
#include <memory>
#include <vector>
#include <map>

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/ESHandle.h"

#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/DetId/interface/DetId.h"
#include "DataFormats/GeometryVector/interface/GlobalPoint.h"
#include "DataFormats/EcalRecHit/interface/EcalRecHitCollections.h"

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

#include "TrackingTools/TrackAssociator/interface/TrackDetectorAssociator.h"
#include "TrackingTools/TrackAssociator/interface/TrackAssociatorParameters.h"
#include "TrackingTools/TrackAssociator/interface/DetIdAssociator.h"
#include "TrackingTools/TrajectoryState/interface/FreeTrajectoryState.h"

#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"

#include "AnalysisDataFormats/SUSYBSMObjects/interface/HSCParticle.h"
#include "AnalysisDataFormats/SUSYBSMObjects/interface/HSCPCaloInfo.h"

class BetaCalculatorECAL {

   public:
       BetaCalculatorECAL(const edm::ParameterSet& iConfig, edm::ConsumesCollector&& iC);
       void  addInfoToCandidate(susybsm::HSCParticle& candidate, edm::Handle<reco::TrackCollection>& tracks, edm::Event& iEvent, const edm::EventSetup& iSetup, susybsm::HSCPCaloInfo& caloInfo);

   private:
       int getDetailedTrackLengthInXtals(std::map<int,GlobalPoint>& trackExitPositionMap,
           std::map<int,float>& trackCrossedXtalMap,
           double& totalLengthCurved,
           GlobalPoint& internalPointCurved,
           GlobalPoint& externalPointCurved,
           const CaloGeometry* theGeometry,
           const CaloTopology * theTopology,
           const std::vector<SteppingHelixStateInfo>& neckLace);
       std::vector<SteppingHelixStateInfo> calcEcalDeposit(const FreeTrajectoryState* tkInnerState,
           const DetIdAssociator& associator);
       void addStepToXtal(std::map<int,GlobalPoint>& trackExitPositionMap,
           std::map<int,float>& trackCrossedXtalMap,
           DetId aDetId,
           float step,
           GlobalPoint point,
           const CaloSubdetectorGeometry* theSubdetGeometry);

      // Data members
      TrackDetectorAssociator trackAssociator_;
      TrackAssociatorParameters parameters_;
      edm::EDGetTokenT<EBRecHitCollection> EBRecHitCollectionToken_;
      edm::EDGetTokenT<EERecHitCollection> EERecHitCollectionToken_;

      edm::ESHandle<DetIdAssociator> ecalDetIdAssociator_;
      edm::ESHandle<MagneticField> bField_;
      edm::ESHandle<CaloGeometry> theCaloGeometry_;

};