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_;
};