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