TauA1NuConstrainedFitter.h

Go to the documentation of this file.

#ifndef RecoTauTag_ImpactParameter_TauA1NuConstrainedFitter_h
#define RecoTauTag_ImpactParameter_TauA1NuConstrainedFitter_h

/* From SimpleFits Package
 * Designed an written by
 * author: Ian M. Nugent
 * Humboldt Foundations
 */

#include "TVector3.h"
#include "RecoTauTag/ImpactParameter/interface/MultiProngTauSolver.h"
#include "RecoTauTag/ImpactParameter/interface/LorentzVectorParticle.h"
#include "RecoTauTag/ImpactParameter/interface/ErrorMatrixPropagator.h"
#include <vector>
#include "TMatrixT.h"
#include "TVectorT.h"
#include "TMatrixTSym.h"

namespace tauImpactParameter {

class TauA1NuConstrainedFitter : public MultiProngTauSolver{
 public:
    TauA1NuConstrainedFitter(unsigned int ambiguity,const LorentzVectorParticle& A1,const TVector3& PVertex, const TMatrixTSym<double>& VertexCov);
  ~TauA1NuConstrainedFitter() override{};

  enum Pars{tau_phi=0,tau_theta,a1_px,a1_py,a1_pz,a1_m,nu_px,nu_py,nu_pz,npar};
  enum ExpandedPars{a1_vx=9,a1_vy,a1_vz,nexpandedpar};
  enum OrignialPars{norigpar=13};

  std::vector<LorentzVectorParticle> getRefitDaughters();
  LorentzVectorParticle getMother();
  double getTauRotationSignificance();

  bool fit();

 private:
  static TVectorT<double> ComputeInitalExpPar(const TVectorT<double> &inpar);
  static TVectorT<double> ComputeExpParToPar(const TVectorT<double> &inpar);
  static TVectorT<double> ComputeNuLorentzVectorPar(const TVectorT<double> &inpar);
  static TVectorT<double> ComputeA1LorentzVectorPar(const TVectorT<double> &inpar);
  static TVectorT<double> ComputeMotherLorentzVectorPar(const TVectorT<double> &inpar);
  static TVectorT<double> SolveAmbiguityAnalytically(const TVectorT<double> &inpar, unsigned int ambiguity);
  static TVectorT<double> SolveAmbiguityAnalyticallywithRot(const TVectorT<double> &inpar, unsigned int ambiguity);
  static TVectorT<double> TauRot(const TVectorT<double> &inpar);

  void UpdateExpandedPar();
  static void CovertParToObjects(const TVectorD &v,TLorentzVector &a1,TLorentzVector &nu,double &phi,double &theta,TVector3 &TauDir);

  TVectorT<double> par_0;
  TVectorT<double> par;
  TMatrixTSym<double> cov_0;
  TMatrixTSym<double> cov;

  TVectorT<double> exppar;
  TMatrixTSym<double> expcov;
  std::vector<LorentzVectorParticle> particles_;
  unsigned int ambiguity_;

};
}
#endif