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