chisquare.h

Go to the documentation of this file.

//
// Author: Valerio Bertone: valerio.bertone@cern.ch
//
 
#pragma once
 
#include <numeric>
 
#include "NangaParbat/datahandler.h"
#include "NangaParbat/convolutiontable.h"
#include "NangaParbat/parameterisation.h"
 
namespace NangaParbat
{
  class ChiSquare
  {
  public:
    ChiSquare(std::vector<std::pair<DataHandler*, ConvolutionTable*>> DSVect, Parameterisation* NPFunc);
 
    ChiSquare(Parameterisation* NPFunc);
 
    virtual void AddBlock(std::pair<DataHandler*, ConvolutionTable*> DSBlock);
 
    std::vector<double> GetResiduals(int const& ids, bool const& central = false) const;
 
    std::vector<double> GetResidualDerivatives(int const& ids, int const& ipar) const;
 
    std::pair<std::vector<double>, double> GetSystematicShifts(int const& ids) const;
 
    double Evaluate(int const& ids = -1, bool const& central = false) const;
 
    std::vector<double> Derive() const;
 
    double operator()(int const& ids = -1) { return Evaluate(ids); };
 
    virtual void SetParameters(std::vector<double> const& pars) { _NPFunc->SetParameters(pars); };
 
    std::vector<std::pair<DataHandler*, ConvolutionTable*>> GetBlocks() const { return _DSVect; };
 
    Parameterisation* GetNonPerturbativeFunction() const { return _NPFunc; };
 
    std::vector<int> GetDataPointNumbers() const { return _ndata; };
 
    std::vector<int> GetDataPointNumbersAfterCuts() const { return _ndatac; };
 
    int GetDataPointNumber() const { return std::accumulate(_ndata.begin(), _ndata.end(), 0); };
 
    int GetDataPointNumberAfterCuts() const { return std::accumulate(_ndatac.begin(), _ndatac.end(), 0); };
 
    int GetNumberOfExperiments() const { return (int) _ndata.size(); };
 
    int GetNumberOfParameters() const { return (int) _NPFunc->GetParameters().size(); };
 
    std::vector<double> GetParameters() const { return _NPFunc->GetParameters(); };
 
  protected:
    std::vector<std::pair<DataHandler*, ConvolutionTable*>> _DSVect;  
    Parameterisation*                                       _NPFunc;  
    std::vector<int>                                        _ndata;   
    std::vector<int>                                        _ndatac;  
 
    friend YAML::Emitter& operator << (YAML::Emitter& os, ChiSquare const& chi2);
  };
 
  YAML::Emitter& operator << (YAML::Emitter& os, ChiSquare const& chi2);
}