EventShapeVariables.h

Go to the documentation of this file.

#ifndef EventShapeVariables_h
#define EventShapeVariables_h

#include "DataFormats/Math/interface/Vector3D.h"
#include "DataFormats/Candidate/interface/Candidate.h"
#include "DataFormats/Candidate/interface/CandidateFwd.h"

#include "TMatrixDSym.h"
#include "TVectorD.h"

#include <vector>

class EventShapeVariables {

 public:
  explicit EventShapeVariables(const edm::View<reco::Candidate>& inputVectors);  
  explicit EventShapeVariables(const std::vector<math::XYZVector>& inputVectors);  
  explicit EventShapeVariables(const std::vector<math::RhoEtaPhiVector>& inputVectors);  
  explicit EventShapeVariables(const std::vector<math::RThetaPhiVector>& inputVectors);  
  ~EventShapeVariables(){};

  double isotropy(const unsigned int& numberOfSteps = 1000) const;
  
  double circularity(const unsigned int& numberOfSteps = 1000) const;

  void set_r(double r);

  void setFWmax(unsigned m);

  double sphericity();
  double aplanarity();
  double C();
  double D();

  const std::vector<double>& getEigenValues() { if(!tensors_computed_) compTensorsAndVectors(); return eigenValues_; }
  const std::vector<double>& getEigenValuesNoNorm() { if(!tensors_computed_) compTensorsAndVectors(); return eigenValuesNoNorm_; }
  const TMatrixD& getEigenVectors() { if(!tensors_computed_) compTensorsAndVectors(); return eigenVectors_; }

  double getFWmoment( unsigned l ) ;
  const std::vector<double>& getFWmoments();

 private:
  void compTensorsAndVectors() ;

  void computeFWmoments() ;

  std::vector<math::XYZVector> inputVectors_;

  double r_;
  bool tensors_computed_;
  TMatrixD eigenVectors_;
  TVectorD eigenValuesTmp_, eigenValuesNoNormTmp_;
  std::vector<double> eigenValues_, eigenValuesNoNorm_;

  unsigned fwmom_maxl_;
  std::vector<double> fwmom_;
  bool   fwmom_computed_ ;

};

#endif