G/ProfileChi2.h

Go to the documentation of this file.

#ifndef FdProfileConstrainedGeometryFitPG_ProfileChi2_h
#define FdProfileConstrainedGeometryFitPG_ProfileChi2_h

#include <fevt/Eye.h>
#include <evt/ShowerFRecData.h>

#include <utl/Function.h>

#include "../FdEnergyDepositFinderKG/CFMatrixCalculator.h"
#include "../FdEnergyDepositFinderKG/ProfileCalculator.h"
#include "../FdEnergyDepositFinderKG/ProfileFitter.h"

namespace FdProfileConstrainedGeometryFitPG {

class ProfileChi2 {
public:
  ProfileChi2()
  {
    fCFMatrixCalculator.Init();
    fCFMatrixCalculatorDense.Init();
    //    fProfileFitter.Init();
  }

  double operator()(fevt::Eye &eyeCopy, bool precise=false);
  void Init();

  double GetShapeConstraint1Chi2() const { return fProfileFitter.GetShapeConstraint1Chi2(); }
  double GetShapeConstraint2Chi2() const { return fProfileFitter.GetShapeConstraint2Chi2(); }
  double GetShapeConstraint3Chi2() const { return fProfileFitter.GetShapeConstraint3Chi2(); }

  unsigned int GetNDof() const { return fProfileFitter.GetNDof(); }
  void SetFitterPars()
  {    
    fProfileFitter.SetShapePars(ghPars);
    fProfileFitter.SetUnivConstrained(fIsUnivConstrained, fUnivFunction, fksigma);
    fProfileFitter.SetAafCorrection(fAafCorrection, fUseNoiseBins);
  }
  void UnSetFitterPars() {fProfileFitter.Init();}
  void SetUseLightFlux(bool use=true) {fUseLightFlux=use;}
    
protected:
  FdEnergyDepositFinderKG::CFMatrixCalculator fCFMatrixCalculator;
  FdEnergyDepositFinderKG::CFMatrixCalculator fCFMatrixCalculatorDense;
  FdEnergyDepositFinderKG::ProfileCalculator fProfileCalculator;
  FdEnergyDepositFinderKG::ProfileFitter fProfileFitter;

  bool fLeavingAtmoIsError;
  bool fIsUnivConstrained;
  bool fOnlyDirectLight;
  bool fAafCorrection;
  bool fUseNoiseBins;
  utl::Function fUnivFunction;
  double fksigma; //< Universality constraints from config

  FdEnergyDepositFinderKG::GHShapeParameters ghPars;

  bool FitProfile(fevt::Eye& eye,
                  const FdEnergyDepositFinderKG::EFitFunctionType type, bool precise = false);

  bool GuessGHParameters(evt::ShowerFRecData& shower) const;
  bool fUseLightFlux;
  unsigned int fDenseMatrixDim;
};

} //namespace
#endif