CMS 3D CMS Logo

 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Groups Pages
List of all members | Public Member Functions | Private Member Functions | Private Attributes
HGCalTrackCollectionProducer Class Reference
Inheritance diagram for HGCalTrackCollectionProducer:
edm::stream::EDProducer<>

Public Member Functions

 HGCalTrackCollectionProducer (const edm::ParameterSet &)
 
- Public Member Functions inherited from edm::stream::EDProducer<>
 EDProducer ()=default
 
 EDProducer (const EDProducer &)=delete
 
bool hasAbilityToProduceInBeginLumis () const final
 
bool hasAbilityToProduceInBeginProcessBlocks () const final
 
bool hasAbilityToProduceInBeginRuns () const final
 
bool hasAbilityToProduceInEndLumis () const final
 
bool hasAbilityToProduceInEndProcessBlocks () const final
 
bool hasAbilityToProduceInEndRuns () const final
 
const EDProduceroperator= (const EDProducer &)=delete
 

Private Member Functions

void beginLuminosityBlock (const edm::LuminosityBlock &, const edm::EventSetup &) override
 
void produce (edm::Event &, const edm::EventSetup &) override
 

Private Attributes

const MagneticFieldbField_
 
float diskInnerRadius_
 
float diskOuterRadius_
 
const std::vector< double > DPtovPtCut_
 
std::array< std::string, 1 > hgc_names_
 
std::array< const
HGCalGeometry *, 1 > 
hgcGeometries_
 
std::array< edm::ESGetToken
< HGCalGeometry,
IdealGeometryRecord >, 1 > 
hgcGeometryTokens_
 
const edm::ESGetToken
< MagneticField,
IdealMagneticFieldRecord
magneticFieldToken_
 
std::unique_ptr
< PropagatorWithMaterial
mat_prop_
 
std::array< std::vector
< ReferenceCountingPointer
< BoundDisk > >, 1 > 
minusSurface_
 
const std::vector< unsigned > NHitCut_
 
std::array< std::vector
< ReferenceCountingPointer
< BoundDisk > >, 1 > 
plusSurface_
 
edm::EDGetTokenT< edm::View
< reco::PFRecTrack > > 
src_
 
const edm::ESGetToken
< TrackerGeometry,
TrackerDigiGeometryRecord
tkerGeomToken_
 
const TrackerGeometrytkGeom_
 
const reco::TrackBase::TrackQuality trackQuality_
 
const bool useIterTracking_
 

Additional Inherited Members

- Public Types inherited from edm::stream::EDProducer<>
using CacheTypes = CacheContexts< T...>
 
using GlobalCache = typename CacheTypes::GlobalCache
 
using HasAbility = AbilityChecker< T...>
 
using InputProcessBlockCache = typename CacheTypes::InputProcessBlockCache
 
using LuminosityBlockCache = typename CacheTypes::LuminosityBlockCache
 
using LuminosityBlockContext = LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >
 
using LuminosityBlockSummaryCache = typename CacheTypes::LuminosityBlockSummaryCache
 
using RunCache = typename CacheTypes::RunCache
 
using RunContext = RunContextT< RunCache, GlobalCache >
 
using RunSummaryCache = typename CacheTypes::RunSummaryCache
 

Detailed Description

Definition at line 39 of file HGCalTrackCollectionProducer.cc.

Constructor & Destructor Documentation

HGCalTrackCollectionProducer::HGCalTrackCollectionProducer ( const edm::ParameterSet iConfig)

Definition at line 75 of file HGCalTrackCollectionProducer.cc.

References edm::ParameterSet::getParameter(), edm::ParameterSet::getParameterSet(), hgc_names_, hgcGeometryTokens_, mps_fire::i, LogDebug, and AlCaHLTBitMon_QueryRunRegistry::string.

76  : src_(consumes<edm::View<reco::PFRecTrack> >(iConfig.getParameter<edm::InputTag>("src"))),
77  trackQuality_((iConfig.existsAs<std::string>("trackQuality"))
78  ? reco::TrackBase::qualityByName(iConfig.getParameter<std::string>("trackQuality"))
80  DPtovPtCut_(iConfig.getParameter<std::vector<double> >("DPtOverPtCuts_byTrackAlgo")),
81  NHitCut_(iConfig.getParameter<std::vector<unsigned> >("NHitCuts_byTrackAlgo")),
82  useIterTracking_(iConfig.getParameter<bool>("useIterativeTracking")),
83  magneticFieldToken_(esConsumes<edm::Transition::BeginLuminosityBlock>()),
84  tkerGeomToken_(esConsumes<edm::Transition::BeginLuminosityBlock>()) {
85  LogDebug("HGCalTrackCollectionProducer")
86  << " HGCalTrackCollectionProducer::HGCalTrackCollectionProducer " << std::endl;
87 
88  const edm::ParameterSet& geoconf = iConfig.getParameterSet("hgcalGeometryNames");
89  hgc_names_[0] = geoconf.getParameter<std::string>("HGC_ECAL");
90  for (unsigned i = 0; i < hgcGeometryTokens_.size(); i++) {
91  hgcGeometryTokens_[i] = esConsumes<edm::Transition::BeginLuminosityBlock>(edm::ESInputTag("", hgc_names_[i]));
92  }
93 
94  produces<reco::PFRecTrackCollection>("TracksInHGCal");
95  produces<reco::PFRecTrackCollection>("TracksNotInHGCal");
96 }
std::array< edm::ESGetToken< HGCalGeometry, IdealGeometryRecord >, 1 > hgcGeometryTokens_
bool existsAs(std::string const &parameterName, bool trackiness=true) const
checks if a parameter exists as a given type
Definition: ParameterSet.h:171
std::array< std::string, 1 > hgc_names_
const std::vector< unsigned > NHitCut_
edm::EDGetTokenT< edm::View< reco::PFRecTrack > > src_
const edm::ESGetToken< MagneticField, IdealMagneticFieldRecord > magneticFieldToken_
const edm::ESGetToken< TrackerGeometry, TrackerDigiGeometryRecord > tkerGeomToken_
static TrackQuality qualityByName(const std::string &name)
Definition: TrackBase.cc:126
const reco::TrackBase::TrackQuality trackQuality_
ParameterSet const & getParameterSet(std::string const &) const
T getParameter(std::string const &) const
Definition: ParameterSet.h:303
const std::vector< double > DPtovPtCut_
#define LogDebug(id)

Member Function Documentation

void HGCalTrackCollectionProducer::beginLuminosityBlock ( const edm::LuminosityBlock lumi,
const edm::EventSetup es 
)
overrideprivate

Definition at line 100 of file HGCalTrackCollectionProducer.cc.

References funct::abs(), alongMomentum, bField_, diskInnerRadius_, diskOuterRadius_, edm::EventSetup::getData(), HGCalDDDConstants::getTrForms(), hgcGeometries_, hgcGeometryTokens_, mps_fire::i, LogDebug, ph2tkdigialgo::m_pion, magneticFieldToken_, mat_prop_, minusSurface_, plusSurface_, makeMuonMisalignmentScenario::rot, tkerGeomToken_, tkGeom_, and BeamSpotPI::Z.

100  {
101  constexpr float m_pion = 0.1396;
102  // get dependencies for setting up propagator
105  // get HGC geometries (assume that layers are ordered in Z!)
106  for (unsigned i = 0; i < hgcGeometries_.size(); ++i) {
108  }
109 
110  // make propagator
111  mat_prop_ = std::make_unique<PropagatorWithMaterial>(alongMomentum, m_pion, bField_);
112  // setup HGC layers for track propagation
113  Surface::RotationType rot; //unit rotation matrix
114  for (unsigned i = 0; i < hgcGeometries_.size(); ++i) {
115  minusSurface_[i].clear();
116  plusSurface_[i].clear();
117  const HGCalDDDConstants& dddCons = hgcGeometries_[i]->topology().dddConstants();
118  std::map<float, float> zrhoCoord;
119  std::map<float, float> innerRadiusCoord;
120  auto theTrForms = dddCons.getTrForms();
121  const auto& firstLayerIt = theTrForms.back();
122  float Z(std::abs(firstLayerIt.h3v.z()));
123  // use hardcoded radii for now (FIX ME)
124  diskInnerRadius_ = 31.5;
125  diskOuterRadius_ = 161.0f;
126  LogDebug("HGCalTrackCollectionProducer") << "O HAI I'm making a bound disk with Outer R=" << diskOuterRadius_
127  << " Inner R=" << diskInnerRadius_ << " and Z=" << Z << std::endl;
129  Surface::PositionType(0, 0, -Z), rot, new SimpleDiskBounds(diskInnerRadius_, diskOuterRadius_, -0.001, 0.001))));
131  Surface::PositionType(0, 0, +Z), rot, new SimpleDiskBounds(diskInnerRadius_, diskOuterRadius_, -0.001, 0.001))));
132  }
133 }
std::array< edm::ESGetToken< HGCalGeometry, IdealGeometryRecord >, 1 > hgcGeometryTokens_
std::array< std::vector< ReferenceCountingPointer< BoundDisk > >, 1 > minusSurface_
bool getData(T &iHolder) const
Definition: EventSetup.h:122
const edm::ESGetToken< MagneticField, IdealMagneticFieldRecord > magneticFieldToken_
std::array< const HGCalGeometry *, 1 > hgcGeometries_
std::vector< HGCalParameters::hgtrform > getTrForms() const
std::unique_ptr< PropagatorWithMaterial > mat_prop_
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
const edm::ESGetToken< TrackerGeometry, TrackerDigiGeometryRecord > tkerGeomToken_
Disk BoundDisk
Definition: BoundDisk.h:54
std::array< std::vector< ReferenceCountingPointer< BoundDisk > >, 1 > plusSurface_
#define LogDebug(id)
void HGCalTrackCollectionProducer::produce ( edm::Event evt,
const edm::EventSetup iSetup 
)
overrideprivate

Definition at line 135 of file HGCalTrackCollectionProducer.cc.

References funct::abs(), bField_, diskInnerRadius_, diskOuterRadius_, HLT_FULL_cff::distance, DPtovPtCut_, newFWLiteAna::found, edm::Event::getByToken(), TrajectoryStateOnSurface::globalPosition(), PFTrackAlgoTools::goodPtResolution(), mps_fire::i, TrajectoryStateOnSurface::isValid(), phase1PixelTopology::layer, LogDebug, mat_prop_, minusSurface_, eostools::move(), NHitCut_, trajectoryStateTransform::outerStateOnSurface(), PV3DBase< T, PVType, FrameType >::perp(), plusSurface_, DiDispStaMuonMonitor_cfi::pt, edm::Event::put(), mathSSE::sqrt(), src_, tkGeom_, HLT_FULL_cff::track, trackQuality_, tracks, useIterTracking_, PV3DBase< T, PVType, FrameType >::x(), PV3DBase< T, PVType, FrameType >::y(), and PV3DBase< T, PVType, FrameType >::z().

135  {
137  evt.getByToken(src_, trackHandle);
138  const auto& tracks = *trackHandle;
139 
140  auto outputInHGCal = std::make_unique<reco::PFRecTrackCollection>();
141  auto outputNotInHGCal = std::make_unique<reco::PFRecTrackCollection>();
142 
143  for (unsigned int i = 0; i < tracks.size(); i++) {
144  const auto track = tracks.ptrAt(i);
145  bool isGood =
147  LogDebug("HGCalTrackCollectionProducer") << "HGCalTrackCollectionProducer Track number " << i
148  << " has a goodPtResolution result of " << isGood << std::endl;
149  if (!isGood)
150  continue;
151  bool found = false;
152  const TrajectoryStateOnSurface myTSOS =
154  auto detbegin = myTSOS.globalPosition().z() > 0 ? plusSurface_.begin() : minusSurface_.begin();
155  auto detend = myTSOS.globalPosition().z() > 0 ? plusSurface_.end() : minusSurface_.end();
156  for (auto det = detbegin; det != detend; ++det) {
157  LogDebug("HGCalTrackCollectionProducer") << "at HGC detector: " << std::distance(detbegin, det) << std::endl;
158  unsigned layer_count = 1;
159  for (const auto& layer : *det) {
160  LogDebug("HGCalTrackCollectionProducer") << " at DET layer: " << layer_count++ << std::endl;
161  TrajectoryStateOnSurface piStateAtSurface = mat_prop_->propagate(myTSOS, *layer);
162  if (piStateAtSurface.isValid()) {
163  LogDebug("HGCalTrackCollectionProducer") << "Extrapolation is valid!" << std::endl;
164  GlobalPoint pt = piStateAtSurface.globalPosition();
165  if (pt.perp() < diskOuterRadius_) {
166  if (pt.perp() > diskInnerRadius_) {
167  LogDebug("HGCalTrackCollectionProducer")
168  << "(x,y,z,r)=(" << pt.x() << ", " << pt.y() << ", " << pt.z() << ", "
169  << sqrt(pt.x() * pt.x() + pt.y() * pt.y()) << ")" << std::endl;
170  if (std::abs(track->trackRef()->eta()) < 1.47)
171  LogDebug("HGCalTrackCollectionProducer") << " ETA IN BARREL REGION: " << track->trackRef()->eta()
172  << " (PT: " << track->trackRef()->pt() << ")" << std::endl;
173  found = true;
174  } else {
175  LogDebug("HGCalTrackCollectionProducer")
176  << " but r=" << pt.perp() << " < diskInnerRadius=" << diskInnerRadius_ << " so skipping "
177  << std::endl;
178  }
179  } else {
180  LogDebug("HGCalTrackCollectionProducer")
181  << " but r=" << pt.perp() << " > diskOuterRadius=" << diskOuterRadius_ << " so skipping " << std::endl;
182  }
183  } else {
184  LogDebug("HGCalTrackCollectionProducer") << "Extrapolation is NOT valid!" << std::endl;
185  }
186  }
187  }
188  if (found) {
189  LogDebug("HGCalTrackCollectionProducer") << " Track going to outputInHGCal pt eta " << track->trackRef()->pt()
190  << " " << track->trackRef()->eta() << std::endl;
191  outputInHGCal->push_back(*track);
192  } else {
193  outputNotInHGCal->push_back(*track);
194  }
195  } // track loop
196 
197  evt.put(std::move(outputInHGCal), "TracksInHGCal");
198  evt.put(std::move(outputNotInHGCal), "TracksNotInHGCal");
199 }
bool goodPtResolution(const reco::TrackRef &, const std::vector< double > &DPtovPtCut, const std::vector< unsigned > &NHitCut, bool useIterTracking, const reco::TrackBase::TrackQuality trackQuality)
OrphanHandle< PROD > put(std::unique_ptr< PROD > product)
Put a new product.
Definition: Event.h:133
T perp() const
Definition: PV3DBase.h:69
TrajectoryStateOnSurface outerStateOnSurface(const reco::Track &tk, const TrackingGeometry &geom, const MagneticField *field, bool withErr=true)
bool getByToken(EDGetToken token, Handle< PROD > &result) const
Definition: Event.h:539
T y() const
Definition: PV3DBase.h:60
GlobalPoint globalPosition() const
auto const & tracks
cannot be loose
std::array< std::vector< ReferenceCountingPointer< BoundDisk > >, 1 > minusSurface_
const std::vector< unsigned > NHitCut_
edm::EDGetTokenT< edm::View< reco::PFRecTrack > > src_
constexpr std::array< uint8_t, layerIndexSize > layer
T sqrt(T t)
Definition: SSEVec.h:19
T z() const
Definition: PV3DBase.h:61
std::unique_ptr< PropagatorWithMaterial > mat_prop_
def move
Definition: eostools.py:511
Abs< T >::type abs(const T &t)
Definition: Abs.h:22
const reco::TrackBase::TrackQuality trackQuality_
const std::vector< double > DPtovPtCut_
std::array< std::vector< ReferenceCountingPointer< BoundDisk > >, 1 > plusSurface_
T x() const
Definition: PV3DBase.h:59
#define LogDebug(id)

Member Data Documentation

const MagneticField* HGCalTrackCollectionProducer::bField_
private

Definition at line 61 of file HGCalTrackCollectionProducer.cc.

Referenced by beginLuminosityBlock(), and produce().

float HGCalTrackCollectionProducer::diskInnerRadius_
private

Definition at line 72 of file HGCalTrackCollectionProducer.cc.

Referenced by beginLuminosityBlock(), and produce().

float HGCalTrackCollectionProducer::diskOuterRadius_
private

Definition at line 71 of file HGCalTrackCollectionProducer.cc.

Referenced by beginLuminosityBlock(), and produce().

const std::vector<double> HGCalTrackCollectionProducer::DPtovPtCut_
private

Definition at line 53 of file HGCalTrackCollectionProducer.cc.

Referenced by produce().

std::array<std::string, 1> HGCalTrackCollectionProducer::hgc_names_
private

Definition at line 63 of file HGCalTrackCollectionProducer.cc.

Referenced by HGCalTrackCollectionProducer().

std::array<const HGCalGeometry*, 1> HGCalTrackCollectionProducer::hgcGeometries_
private

Definition at line 66 of file HGCalTrackCollectionProducer.cc.

Referenced by beginLuminosityBlock().

std::array<edm::ESGetToken<HGCalGeometry, IdealGeometryRecord>, 1> HGCalTrackCollectionProducer::hgcGeometryTokens_
private
const edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> HGCalTrackCollectionProducer::magneticFieldToken_
private

Definition at line 59 of file HGCalTrackCollectionProducer.cc.

Referenced by beginLuminosityBlock().

std::unique_ptr<PropagatorWithMaterial> HGCalTrackCollectionProducer::mat_prop_
private

Definition at line 69 of file HGCalTrackCollectionProducer.cc.

Referenced by beginLuminosityBlock(), and produce().

std::array<std::vector<ReferenceCountingPointer<BoundDisk> >, 1> HGCalTrackCollectionProducer::minusSurface_
private

Definition at line 67 of file HGCalTrackCollectionProducer.cc.

Referenced by beginLuminosityBlock(), and produce().

const std::vector<unsigned> HGCalTrackCollectionProducer::NHitCut_
private

Definition at line 54 of file HGCalTrackCollectionProducer.cc.

Referenced by produce().

std::array<std::vector<ReferenceCountingPointer<BoundDisk> >, 1> HGCalTrackCollectionProducer::plusSurface_
private

Definition at line 67 of file HGCalTrackCollectionProducer.cc.

Referenced by beginLuminosityBlock(), and produce().

edm::EDGetTokenT<edm::View<reco::PFRecTrack> > HGCalTrackCollectionProducer::src_
private

Definition at line 47 of file HGCalTrackCollectionProducer.cc.

Referenced by produce().

const edm::ESGetToken<TrackerGeometry, TrackerDigiGeometryRecord> HGCalTrackCollectionProducer::tkerGeomToken_
private

Definition at line 60 of file HGCalTrackCollectionProducer.cc.

Referenced by beginLuminosityBlock().

const TrackerGeometry* HGCalTrackCollectionProducer::tkGeom_
private

Definition at line 62 of file HGCalTrackCollectionProducer.cc.

Referenced by beginLuminosityBlock(), and produce().

const reco::TrackBase::TrackQuality HGCalTrackCollectionProducer::trackQuality_
private

Definition at line 52 of file HGCalTrackCollectionProducer.cc.

Referenced by produce().

const bool HGCalTrackCollectionProducer::useIterTracking_
private

Definition at line 55 of file HGCalTrackCollectionProducer.cc.

Referenced by produce().