ParameterHandlerGeneric.h

Go to the documentation of this file.

#pragma once
 
// MaCh3 includes
#include "Parameters/ParameterHandlerBase.h"
#include "Samples/SampleStructs.h"
 
class ParameterHandlerGeneric : public ParameterHandlerBase {
  public:
    ParameterHandlerGeneric(const std::vector<std::string>& FileNames, std::string name = "xsec_cov", double threshold = -1, int FirstPCAdpar = -999, int LastPCAdpar = -999);
    ~ParameterHandlerGeneric();
 
    // General Getter functions not split by detector
    inline std::vector<std::string> GetParSampleID(const int i) const { return _fSampleNames[i];};
    inline std::string GetParamTypeString(const int i) const { return SystType_ToString(_fParamType[i]); }
    inline SystType GetParamType(const int i) const {return _fParamType[i];}
 
    inline SplineInterpolation GetParSplineInterpolation(const int i) const {return SplineParams.at(i)._SplineInterpolationType;}
    const std::vector<SplineInterpolation> GetSplineInterpolationFromSampleName(const std::string& SampleName);
    std::string GetParSplineName(const int i) const {return _fSplineNames[i];}
 
    const std::vector<int> GetGlobalSystIndexFromSampleName(const std::string& SampleName, const SystType Type);
    inline double GetParSplineKnotUpperBound(const int i) const {return SplineParams.at(i)._SplineKnotUpBound;}
    inline double GetParSplineKnotLowerBound(const int i) const {return SplineParams.at(i)._SplineKnotLowBound;}
 
    int GetNumParamsFromSampleName(const std::string& SampleName, const SystType Type);
    const std::vector<std::string> GetParsNamesFromSampleName(const std::string& SampleName, const SystType Type);
    const std::vector<int> GetParsIndexFromSampleName(const std::string& SampleName, const SystType Type);
 
    const std::vector<std::string> GetSplineParsNamesFromSampleName(const std::string& SampleName);
    const std::vector<std::string> GetSplineFileParsNamesFromSampleName(const std::string& SampleName);
 
    const std::vector< std::vector<int> > GetSplineModeVecFromSampleName(const std::string& SampleName);
    const std::vector<int> GetSystIndexFromSampleName(const std::string& SampleName, const SystType Type) const;
    const std::vector<NormParameter> GetNormParsFromSampleName(const std::string& SampleName) const;
    const std::vector<FunctionalParameter> GetFunctionalParametersFromSampleName(const std::string& SampleName) const;
    const std::vector<SplineParameter> GetSplineParsFromSampleName(const std::string& SampleName) const;
 
    bool IsParFromGroup(const int i, const std::string& Group) const;
 
    int GetNumParFromGroup(const std::string& Group) const;
    std::vector<std::string> GetUniqueParameterGroups();
 
    void SetGroupOnlyParameters(const std::string& Group, const std::vector<double>& Pars = {});
    void SetGroupOnlyParameters(const std::vector<std::string>& Groups);
    
    void SetFixGroupOnlyParameters(const std::string& Group);
    void SetFixGroupOnlyParameters(const std::vector<std::string>& Groups);
 
    void SetFreeGroupOnlyParameters(const std::string& Group);
    void SetFreeGroupOnlyParameters(const std::vector<std::string>& Groups);
 
    void ToggleFixGroupOnlyParameters(const std::string& Group);   
    void ToggleFixGroupOnlyParameters(const std::vector<std::string>& Groups);
 
    void DumpMatrixToFile(const std::string& Name);
 
    std::vector<const double*> GetOscParsFromSampleName(const std::string& SampleName);
 
  protected:
    void Print();
    void PrintGlobablInfo();
    void PrintNormParams();
    void PrintSplineParams();
    void PrintFunctionalParams();
    void PrintOscillationParams();
    void PrintParameterGroups();
 
    void CheckCorrectInitialisation();
 
    template <typename FilterFunc, typename ActionFunc>
    void IterateOverParams(const std::string& SampleName, FilterFunc filter, ActionFunc action);
 
    void InitParams();
 
    void InitParametersTypeFromConfig();
 
    inline NormParameter GetNormParameter(const YAML::Node& param, const int Index);
 
    inline OscillationParameter GetOscillationParameters(const YAML::Node& param, const int Index);
 
    inline FunctionalParameter GetFunctionalParameters(const YAML::Node& param, const int Index);
    inline SplineParameter GetSplineParameter(const YAML::Node& param, const int Index);
    inline void GetBaseParameter(const YAML::Node& param, const int Index, TypeParameterBase& Parameter);
 
    template<typename ParamT>
    std::vector<ParamT> GetTypeParamsFromSampleName(const std::map<int, int>& indexMap, const std::vector<ParamT>& params, const std::string& SampleName) const;
 
    std::vector<SystType> _fParamType;
 
    std::vector<std::string> _fSplineNames;
 
    std::vector<std::string> _ParameterGroup;
 
    std::vector<std::map<int, int>> _fSystToGlobalSystIndexMap;
 
    std::vector<SplineParameter> SplineParams;
 
    std::vector<NormParameter> NormParams;
 
    std::vector<FunctionalParameter> FuncParams;
 
    std::vector<OscillationParameter> OscParams;
};