CMS 3D CMS Logo

List of all members | Public Member Functions | Private Member Functions | Private Attributes
Cluster3DPCACalculator Class Reference
Inheritance diagram for Cluster3DPCACalculator:
PFCPositionCalculatorBase

Public Member Functions

void calculateAndSetPosition (reco::PFCluster &) override
 
void calculateAndSetPositions (reco::PFClusterCollection &) override
 
 Cluster3DPCACalculator (const edm::ParameterSet &conf)
 
 Cluster3DPCACalculator (const Cluster3DPCACalculator &)=delete
 
Cluster3DPCACalculatoroperator= (const Cluster3DPCACalculator &)=delete
 
- Public Member Functions inherited from PFCPositionCalculatorBase
const std::string & name () const
 
PosCalcoperator= (const PosCalc &)=delete
 
 PFCPositionCalculatorBase (const edm::ParameterSet &conf)
 
 PFCPositionCalculatorBase (const PosCalc &)=delete
 
virtual void update (const edm::EventSetup &)
 
virtual ~PFCPositionCalculatorBase ()=default
 

Private Member Functions

void calculateAndSetPositionActual (reco::PFCluster &)
 
void showerParameters (const reco::PFCluster &, math::XYZPoint &, math::XYZVector &)
 

Private Attributes

std::unique_ptr< TPrincipal > pca_
 
const bool updateTiming_
 

Additional Inherited Members

- Protected Attributes inherited from PFCPositionCalculatorBase
const float _minFractionInCalc
 

Detailed Description

Definition at line 17 of file Cluster3DPCACalculator.cc.

Constructor & Destructor Documentation

Cluster3DPCACalculator::Cluster3DPCACalculator ( const edm::ParameterSet conf)
inline

Definition at line 19 of file Cluster3DPCACalculator.cc.

References calculateAndSetPosition(), calculateAndSetPositions(), and operator=().

21  updateTiming_(conf.getParameter<bool>("updateTiming")),
22  pca_(new TPrincipal(3, "D")) {}
T getParameter(std::string const &) const
std::unique_ptr< TPrincipal > pca_
PFCPositionCalculatorBase(const edm::ParameterSet &conf)
Cluster3DPCACalculator::Cluster3DPCACalculator ( const Cluster3DPCACalculator )
delete

Member Function Documentation

void Cluster3DPCACalculator::calculateAndSetPosition ( reco::PFCluster cluster)
overridevirtual

Implements PFCPositionCalculatorBase.

Definition at line 40 of file Cluster3DPCACalculator.cc.

References calculateAndSetPositionActual(), and pca_.

Referenced by Cluster3DPCACalculator().

40  {
41  pca_.reset(new TPrincipal(3, "D"));
43 }
void calculateAndSetPositionActual(reco::PFCluster &)
std::unique_ptr< TPrincipal > pca_
void Cluster3DPCACalculator::calculateAndSetPositionActual ( reco::PFCluster cluster)
private

Definition at line 52 of file Cluster3DPCACalculator.cc.

References reco::PFCluster::calculatePositionREP(), Exception, mps_fire::i, createfilelist::int, edm::isNotFinite(), min(), PFLayer::NONE, pca_, reco::PFCluster::recHitFractions(), reco::CaloCluster::seed(), reco::CaloCluster::setEnergy(), reco::PFCluster::setLayer(), reco::CaloCluster::setPosition(), reco::PFCluster::setTime(), and updateTiming_.

Referenced by calculateAndSetPosition(), and calculateAndSetPositions().

52  {
53  if (!cluster.seed()) {
54  throw cms::Exception("ClusterWithNoSeed") << " Found a cluster with no seed: " << cluster;
55  }
56  double cl_energy = 0;
57  double max_e = 0.0;
58  double avg_time = 0.0;
59  double time_norm = 0.0;
60  PFLayer::Layer max_e_layer = PFLayer::NONE;
61  reco::PFRecHitRef refseed;
62  double pcavars[3];
63 
64  for (const reco::PFRecHitFraction& rhf : cluster.recHitFractions()) {
65  const reco::PFRecHitRef& refhit = rhf.recHitRef();
66  double rh_energy = refhit->energy();
67  double rh_time = refhit->time();
68  cl_energy += rh_energy * rhf.fraction();
69  if (rh_time > 0.0) { // time == -1 means no measurement
70  // all times are offset by one nanosecond in digitizer
71  // remove that here so all times of flight
72  // are with respect to (0,0,0)
73  avg_time += (rh_time - 1.0);
74  time_norm += 1.0;
75  }
76  if (rh_energy > max_e) {
77  max_e = rh_energy;
78  max_e_layer = rhf.recHitRef()->layer();
79  }
80  if (refhit->detId() == cluster.seed())
81  refseed = refhit;
82  const double rh_fraction = rhf.fraction();
83  rh_energy = refhit->energy() * rh_fraction;
84  if (edm::isNotFinite(rh_energy)) {
85  //temporarily changed exception to warning
86  // throw cms::Exception("PFClusterAlgo")
87  edm::LogWarning("PFClusterAlgo") << "rechit " << refhit->detId() << " has a NaN energy... "
88  << "The input of the particle flow clustering seems to be corrupted.";
89  continue;
90  }
91  pcavars[0] = refhit->position().x();
92  pcavars[1] = refhit->position().y();
93  pcavars[2] = refhit->position().z();
94  int nhit = int(rh_energy * 100); // put rec_hit energy in units of 10 MeV
95 
96  for (int i = 0; i < nhit; ++i) {
97  pca_->AddRow(pcavars);
98  }
99  }
100  cluster.setEnergy(cl_energy);
101  cluster.setLayer(max_e_layer);
102  // calculate the position
103 
104  pca_->MakePrincipals();
105  const TVectorD& means = *(pca_->GetMeanValues());
106  const TMatrixD& eigens = *(pca_->GetEigenVectors());
107 
108  math::XYZPoint barycenter(means[0], means[1], means[2]);
109  math::XYZVector axis(eigens(0, 0), eigens(1, 0), eigens(2, 0));
110 
111  if (time_norm > 0.0) {
112  avg_time = avg_time / time_norm;
113  } else {
115  }
116 
117  if (axis.z() * barycenter.z() < 0.0) {
118  axis = math::XYZVector(-eigens(0, 0), -eigens(1, 0), -eigens(2, 0));
119  }
120 
121  if (updateTiming_) {
122  cluster.setTime(avg_time);
123  }
124  cluster.setPosition(barycenter);
125  cluster.calculatePositionREP();
126 }
void setLayer(PFLayer::Layer layer)
set layer
Definition: PFCluster.cc:93
constexpr bool isNotFinite(T x)
Definition: isFinite.h:9
void setPosition(const math::XYZPoint &p)
Definition: CaloCluster.h:139
void setTime(float time, float timeError=0)
Definition: PFCluster.h:88
Fraction of a PFRecHit (rechits can be shared between several PFCluster&#39;s)
void setEnergy(double energy)
Definition: CaloCluster.h:135
void calculatePositionREP()
computes posrep_ once and for all
Definition: PFCluster.h:99
T min(T a, T b)
Definition: MathUtil.h:58
Layer
layer definition
Definition: PFLayer.h:29
DetId seed() const
return DetId of seed
Definition: CaloCluster.h:218
XYZVectorD XYZVector
spatial vector with cartesian internal representation
Definition: Vector3D.h:31
XYZPointD XYZPoint
point in space with cartesian internal representation
Definition: Point3D.h:12
const std::vector< reco::PFRecHitFraction > & recHitFractions() const
vector of rechit fractions
Definition: PFCluster.h:69
std::unique_ptr< TPrincipal > pca_
void Cluster3DPCACalculator::calculateAndSetPositions ( reco::PFClusterCollection clusters)
overridevirtual

Implements PFCPositionCalculatorBase.

Definition at line 45 of file Cluster3DPCACalculator.cc.

References calculateAndSetPositionActual(), and pca_.

Referenced by Cluster3DPCACalculator().

45  {
46  for (reco::PFCluster& cluster : clusters) {
47  pca_.reset(new TPrincipal(3, "D"));
49  }
50 }
Particle flow cluster, see clustering algorithm in PFClusterAlgo.
Definition: PFCluster.h:46
void calculateAndSetPositionActual(reco::PFCluster &)
std::unique_ptr< TPrincipal > pca_
Cluster3DPCACalculator& Cluster3DPCACalculator::operator= ( const Cluster3DPCACalculator )
delete

Referenced by Cluster3DPCACalculator().

void Cluster3DPCACalculator::showerParameters ( const reco::PFCluster ,
math::XYZPoint ,
math::XYZVector  
)
private

Member Data Documentation

std::unique_ptr<TPrincipal> Cluster3DPCACalculator::pca_
private
const bool Cluster3DPCACalculator::updateTiming_
private

Definition at line 30 of file Cluster3DPCACalculator.cc.

Referenced by calculateAndSetPositionActual().