Skip to content

Commit

Permalink
Merge pull request #27128 from clelange/extend_rechittools
Browse files Browse the repository at this point in the history 
Extend and fix RecHitTools
  • Loading branch information
@cmsbuild
cmsbuild committed Jun 24, 2019
2 parents cf6a414 + 68b182e commit 045ebf5
Show file tree
Hide file tree
Showing 22 changed files with 1,592 additions and 1,703 deletions.
123 changes: 60 additions & 63 deletions RecoEgamma/EgammaTools/interface/EgammaPCAHelper.h
View file
Expand Up @@ -30,89 +30,86 @@ class HGCalRecHit;

namespace hgcal {

class EGammaPCAHelper
{
public:
typedef ROOT::Math::Transform3D Transform3D;
typedef ROOT::Math::Transform3D::Point Point;
class EGammaPCAHelper {
public:
typedef ROOT::Math::Transform3D Transform3D;
typedef ROOT::Math::Transform3D::Point Point;

EGammaPCAHelper();
~EGammaPCAHelper();
EGammaPCAHelper();
~EGammaPCAHelper();

// for the GsfElectrons
void storeRecHits(const reco::CaloCluster & theCluster );
void storeRecHits(const reco::HGCalMultiCluster &cluster );
// for the GsfElectrons
void storeRecHits(const reco::CaloCluster &theCluster);
void storeRecHits(const reco::HGCalMultiCluster &cluster);

const TPrincipal & pcaResult();
/// to set from outside - once per event
void setHitMap( std::map<DetId,const HGCRecHit *> * hitMap) ;
/// to compute from inside - once per event
void fillHitMap(const HGCRecHitCollection & HGCEERecHits,
const HGCRecHitCollection & HGCFHRecHits,
const HGCRecHitCollection & HGCBHRecHits);
const TPrincipal &pcaResult();
/// to set from outside - once per event
void setHitMap(std::map<DetId, const HGCRecHit *> *hitMap);
/// to compute from inside - once per event
void fillHitMap(const HGCRecHitCollection &HGCEERecHits,
const HGCRecHitCollection &HGCFHRecHits,
const HGCRecHitCollection &HGCBHRecHits);

std::map<DetId,const HGCRecHit *> * getHitMap(){return hitMap_;}
std::map<DetId, const HGCRecHit *> *getHitMap() { return hitMap_; }

void setRecHitTools(const hgcal::RecHitTools * recHitTools );
void setRecHitTools(const hgcal::RecHitTools *recHitTools);

inline void setdEdXWeights(const std::vector<double> & dEdX){ dEdXWeights_ = dEdX;}
inline void setdEdXWeights(const std::vector<double> &dEdX) { dEdXWeights_ = dEdX; }

void pcaInitialComputation() {
computePCA(-1.,false);
}
void pcaInitialComputation() { computePCA(-1., false); }

void computePCA(float radius, bool withHalo=true);
const math::XYZPoint & barycenter() const {return barycenter_;}
const math::XYZVector & axis() const {return axis_;}
void computePCA(float radius, bool withHalo = true);
const math::XYZPoint &barycenter() const { return barycenter_; }
const math::XYZVector &axis() const { return axis_; }

void computeShowerWidth(float radius, bool withHalo=true);
void computeShowerWidth(float radius, bool withHalo = true);

inline double sigmaUU() const { return checkIteration()? sigu_ : -1. ;}
inline double sigmaVV() const { return checkIteration()? sigv_ : -1. ;}
inline double sigmaEE() const { return checkIteration()? sige_ : -1. ;}
inline double sigmaPP() const { return checkIteration()? sigp_ : -1. ;}
inline double sigmaUU() const { return checkIteration() ? sigu_ : -1.; }
inline double sigmaVV() const { return checkIteration() ? sigv_ : -1.; }
inline double sigmaEE() const { return checkIteration() ? sige_ : -1.; }
inline double sigmaPP() const { return checkIteration() ? sigp_ : -1.; }

inline const TVectorD& eigenValues () const {return *pca_->GetEigenValues();}
inline const TVectorD& sigmas() const {return *pca_->GetSigmas();}
// contains maxlayer+1 values, first layer is [1]
LongDeps energyPerLayer(float radius, bool withHalo=true) ;
inline const TVectorD &eigenValues() const { return *pca_->GetEigenValues(); }
inline const TVectorD &sigmas() const { return *pca_->GetSigmas(); }
// contains maxlayer+1 values, first layer is [1]
LongDeps energyPerLayer(float radius, bool withHalo = true);

float clusterDepthCompatibility(const LongDeps &, float & measuredDepth, float& expectedDepth, float& expectedSigma );
void printHits( float radius) const;
void clear();
float clusterDepthCompatibility(const LongDeps &, float &measuredDepth, float &expectedDepth, float &expectedSigma);
void printHits(float radius) const;
void clear();

private:
bool checkIteration() const ;
void storeRecHits(const std::vector<std::pair<DetId, float>> &hf);
float findZFirstLayer(const LongDeps&) const;
private:
bool checkIteration() const;
void storeRecHits(const std::vector<std::pair<DetId, float>> &hf);
float findZFirstLayer(const LongDeps &) const;

bool recHitsStored_;
bool debug_;
bool recHitsStored_;
bool debug_;

//parameters
std::vector<double> dEdXWeights_;
std::vector<double> invThicknessCorrection_;
//parameters
std::vector<double> dEdXWeights_;
std::vector<double> invThicknessCorrection_;

int hitMapOrigin_; // 0 not initialized; 1 set from outside ; 2 set from inside
const reco::CaloCluster * theCluster_;
std::map<DetId, const HGCRecHit *> * hitMap_;
std::vector<Spot> theSpots_;
int pcaIteration_;
int hitMapOrigin_; // 0 not initialized; 1 set from outside ; 2 set from inside
const reco::CaloCluster *theCluster_;
std::map<DetId, const HGCRecHit *> *hitMap_;
std::vector<Spot> theSpots_;
int pcaIteration_;
unsigned int maxlayer_;

// output quantities
math::XYZPoint barycenter_;
math::XYZVector axis_;
// output quantities
math::XYZPoint barycenter_;
math::XYZVector axis_;

Transform3D trans_;
double sigu_,sigv_,sige_,sigp_;

// helper
std::unique_ptr<TPrincipal> pca_;
const hgcal::RecHitTools * recHitTools_;
ShowerDepth showerDepth_;
Transform3D trans_;
double sigu_, sigv_, sige_, sigp_;

// helper
std::unique_ptr<TPrincipal> pca_;
const hgcal::RecHitTools *recHitTools_;
ShowerDepth showerDepth_;
};

}
} // namespace hgcal

#endif

0 comments on commit 045ebf5

Please sign in to comment.