Cluster3DPCACalculator Class Reference

Inheritance diagram for Cluster3DPCACalculator:

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

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

Private Attributes
std::unique_ptr< TPrincipal >	pca_

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=().

                                                      :
    PFCPositionCalculatorBase(conf),    
    pca_(new TPrincipal(3,"D")){ 
  }

Cluster3DPCACalculator::Cluster3DPCACalculator ( const Cluster3DPCACalculator &  )

delete

Member Function Documentation

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

virtual

Implements PFCPositionCalculatorBase.

Definition at line 43 of file Cluster3DPCACalculator.cc.

References calculateAndSetPositionActual(), calculateAndSetPositions(), and pca_.

Referenced by Cluster3DPCACalculator().

                                                {
  pca_.reset(new TPrincipal(3,"D"));
  calculateAndSetPositionActual(cluster);
}

void Cluster3DPCACalculator::calculateAndSetPositionActual ( reco::PFCluster &  cluster )

private

Definition at line 57 of file Cluster3DPCACalculator.cc.

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

Referenced by calculateAndSetPosition(), and calculateAndSetPositions().

                                                      {  
  if( !cluster.seed() ) {
    throw cms::Exception("ClusterWithNoSeed")
      << " Found a cluster with no seed: " << cluster;
  }                 
  double cl_energy = 0;  
  double max_e = 0.0;
  double avg_time = 0.0;
  double time_norm = 0.0;
  PFLayer::Layer max_e_layer = PFLayer::NONE;
  reco::PFRecHitRef refseed;  
  double pcavars[3];  

  for( const reco::PFRecHitFraction& rhf : cluster.recHitFractions() ) {
    const reco::PFRecHitRef& refhit = rhf.recHitRef();
    double rh_energy = refhit->energy();
    double rh_time = refhit->time();
    cl_energy += rh_energy * rhf.fraction();
    if( rh_time > 0.0 ) { // time == -1 means no measurement
      // all times are offset by one nanosecond in digitizer
      // remove that here so all times of flight
      // are with respect to (0,0,0)
      avg_time += (rh_time - 1.0); 
      time_norm += 1.0;
    }
    if( rh_energy > max_e ) {
      max_e = rh_energy;
      max_e_layer = rhf.recHitRef()->layer();
    }  
    if( refhit->detId() == cluster.seed() ) refseed = refhit;
    const double rh_fraction = rhf.fraction();
    rh_energy = refhit->energy()*rh_fraction;
    if( edm::isNotFinite(rh_energy) ) {
      throw cms::Exception("PFClusterAlgo")
    <<"rechit " << refhit->detId() << " has a NaN energy... " 
    << "The input of the particle flow clustering seems to be corrupted.";
    }    
    pcavars[0] = refhit->position().x();
    pcavars[1] = refhit->position().y();
    pcavars[2] = refhit->position().z();     
    int nhit = int( rh_energy*100 ); // put rec_hit energy in units of 10 MeV

    for( int i = 0; i < nhit; ++i ) {
      pca_->AddRow(pcavars);
    }
      
  }
  cluster.setEnergy(cl_energy);
  cluster.setLayer(max_e_layer);
  // calculate the position

  pca_->MakePrincipals();
  const TVectorD& means = *(pca_->GetMeanValues());
  const TMatrixD& eigens = *(pca_->GetEigenVectors());
  
  math::XYZPoint  barycenter(means[0],means[1],means[2]);
  math::XYZVector axis(eigens(0,0),eigens(1,0),eigens(2,0));

  if( time_norm > 0.0 ) {
    avg_time = avg_time/time_norm;
  } else {
    avg_time = std::numeric_limits<double>::min();
  }

  if( axis.z()*barycenter.z() < 0.0 ) {
    axis = math::XYZVector(-eigens(0,0),-eigens(1,0),-eigens(2,0));
  }
  
  cluster.setTime(avg_time);
  cluster.setPosition(barycenter);
  cluster.calculatePositionREP();

}

void Cluster3DPCACalculator::calculateAndSetPositions ( reco::PFClusterCollection &  clusters )

virtual

Implements PFCPositionCalculatorBase.

Definition at line 49 of file Cluster3DPCACalculator.cc.

References calculateAndSetPositionActual(), and pca_.

Referenced by calculateAndSetPosition(), and Cluster3DPCACalculator().

                                                            {
  for( reco::PFCluster& cluster : clusters ) {
    pca_.reset(new TPrincipal(3,"D"));
    calculateAndSetPositionActual(cluster);
  }
}

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

Definition at line 30 of file Cluster3DPCACalculator.cc.

Referenced by calculateAndSetPosition(), calculateAndSetPositionActual(), and calculateAndSetPositions().