GHShapeParameters.h

Go to the documentation of this file.

#ifndef _GHShapeParameters_h_
#define _GHShapeParameters_h_

#include <evt/GaisserHillasTypes.h>
#include <utl/Function.h>

#include <cmath>

namespace utl {
  class Branch;
}

/*
  utility class to handle different GH type fitting
*/
namespace FdEnergyDepositFinderKG {

  enum EOneTwo {
    eOne = 0,
    eTwo
  };

  class GHShapeParameters {
  public:
    GHShapeParameters();
    GHShapeParameters(const utl::Branch& config);

    evt::gh::EFunctionType Type() const
    { return fGHType; }

    bool HasEnergyDependence() const { if (fGHType==evt::gh::eUSP) return true; return false; }
    
    double Mean(const EOneTwo par, const double E) const
    { return HasEnergyDependence() ? fConstraint[par](E) : fConstraint[par](); }
    
    double Variance(const EOneTwo par, const double E) const
    { return pow(Sigma(par, E), 2); }
    
    double Sigma(const EOneTwo par, const double E) const
    { return HasEnergyDependence() ? fConstraintVariance[par](E) : fConstraintVariance[par](); }

    double Min(const EOneTwo par) const
    { return fRangeMin[par]; }

    double Max(const EOneTwo par) const
    { return fRangeMax[par]; }

    double Step(const EOneTwo par) const
    { return fStep[par]; }

    evt::gh::EShapeParameter Type(const EOneTwo par) const
    { return fParType[par]; }

    std::string Name(const EOneTwo par) const
    { return evt::gh::GetShapeParameterName(fParType[par]); }
    
    std::string UnitName(const EOneTwo par) const
      { return fUnit[par].GetFunction(); }

    double Unit(const EOneTwo par) const
    { return fUnit[par](); }

  private:
    utl::Function fConstraint[eTwo+1];
    utl::Function fConstraintVariance[eTwo+1];
    utl::Function fUnit[eTwo+1];
    double fRangeMin[eTwo+1];
    double fRangeMax[eTwo+1];
    double fStep[eTwo+1];
    evt::gh::EFunctionType fGHType;
    evt::gh::EShapeParameter fParType[eTwo+1];
  };
}

#endif