BinnedSplineHandler.h

Go to the documentation of this file.

#pragma once
 
//MaCh3 includes
#include "Splines/SplineBase.h"
#include "Manager/MaCh3Modes.h"
 
class BinnedSplineHandler : public SplineBase {
  public:
    BinnedSplineHandler(ParameterHandlerGeneric *xsec_, MaCh3Modes *Modes_);
    virtual ~BinnedSplineHandler();
 
    void Evaluate() final;
 
    void AddSample(const std::string& SampleName,
                   const std::string& SampleTitle,
                   const std::vector<std::string>& OscChanFileNames,
                   const std::vector<std::string>& SplineVarNames);
    void TransferToMonolith();
    void cleanUpMemory();
 
    virtual void FillSampleArray(const std::string& SampleTitle, const std::vector<std::string>& OscChanFileNames);
    std::vector< std::vector<int> > StripDuplicatedModes(const std::vector< std::vector<int> >& InputVector) const;
    std::vector< std::vector<int> > GetEventSplines(const std::string& SampleTitle, int iOscChan, int EventMode, double Var1Val, double Var2Val, double Var3Val);
    void SynchroniseMemTransfer() const final {return;}
    std::vector<TAxis*> FindSplineBinning(const std::string& FileName, const std::string& SampleTitle);
 
    int CountNumberOfLoadedSplines(bool NonFlat=false, int Verbosity=0);
    std::string getDimLabel(const int BinningOpt, const unsigned int Axis) const;
    int GetSampleIndex(const std::string& SampleTitle) const;
    bool isValidSplineIndex(const std::string& SampleTitle, int iSyst, int iOscChan, int iMode, int iVar1, int iVar2, int iVar3) const;
 
    void BuildSampleIndexingArray(const std::string& SampleTitle);
    void PrepForReweight();
    void getSplineCoeff_SepMany(int splineindex, M3::float_t *& xArray, M3::float_t *&manyArray);
    void PrintBinning(TAxis* Axis) const;
    void PrintSampleDetails(const std::string& SampleTitle) const;
    void PrintArrayDetails(const std::string& SampleTitle) const;
 
    const M3::float_t* RetPointer(const int sample, const int oscchan, const int syst, const int mode,
                                  const int var1bin, const int var2bin, const int var3bin) const {
      int index = indexvec[sample][oscchan][syst][mode][var1bin][var2bin][var3bin];
      return &weightvec_Monolith[index];
    }
    void PrepareSplineFile(std::string FileName) final;
    void LoadSplineFile(std::string FileName) final;
 
  protected:
    void CalcSplineWeights() final;
    ParameterHandlerGeneric* ParHandler;
 
    //And now the actual member variables
    std::vector<std::string> SampleNames;
    std::vector<std::string> SampleTitles;
    std::vector<int> Dimensions;
    std::vector<std::vector<std::string>> DimensionLabels;
    std::vector<int> nSplineParams;
    std::vector<int> nOscChans;
 
    std::vector< std::vector< std::vector<TAxis*> > > SplineBinning;
    std::vector< std::vector<std::string> > SplineFileParPrefixNames;
    std::vector< std::vector< std::vector<int> > > SplineModeVecs;
    std::vector< std::vector<int> > GlobalSystIndex;
    std::vector< std::vector<SplineInterpolation> > SplineInterpolationTypes;
 
    std::vector<std::string> UniqueSystNames;
    std::vector<int> UniqueSystIndices;
 
    std::vector< std::vector< std::vector< std::vector< std::vector< std::vector< std::vector< int > > > > > > > indexvec;
    std::vector<int > coeffindexvec;
    std::vector<int> uniquecoeffindices;
 
    std::vector< TSpline3_red* > splinevec_Monolith;
 
    int MonolithSize;
    int MonolithIndex;
    int CoeffIndex;
 
    bool *isflatarray;
    M3::float_t *xcoeff_arr;
    M3::float_t *manycoeff_arr;
 
    std::vector<M3::float_t> weightvec_Monolith;
    std::vector<int> uniquesplinevec_Monolith;
 
    MaCh3Modes* Modes;
    enum TokenOrdering{kSystToken,kModeToken,kVar1BinToken,kVar2BinToken,kVar3BinToken,kNTokens};
    virtual std::vector<std::string> GetTokensFromSplineName(const std::string& FullSplineName) = 0;
 
  private:
    void InvestigateMissingSplines() const;
 
    void LoadSettingsDir(std::unique_ptr<TFile>& SplineFile);
    void LoadMonolithDir(std::unique_ptr<TFile>& SplineFile);
    void LoadIndexDir(std::unique_ptr<TFile>& SplineFile);
 
    void PrepareSettingsDir(std::unique_ptr<TFile>& SplineFile) const;
    void PrepareMonolithDir(std::unique_ptr<TFile>& SplineFile) const;
    void PrepareIndexDir(std::unique_ptr<TFile>& SplineFile) const;
    void PrepareOtherInfoDir(std::unique_ptr<TFile>& SplineFile) const;
};