Count.h

Go to the documentation of this file.

//***************************************************************************************
//***************************************************************************************
 
#pragma once
 
#include "NAGASH/Global.h"
#include "NAGASH/Result.h"
 
namespace NAGASH
{
    class CountBase
    {
    public:
        void Add(const CountBase &);
        void Add(double x = 1, double w = 1);
        double GetSum() const;
        double GetSum2() const;
        double GetSumW() const;
        double GetSumW2() const;
        double GetSumStatUnc() const;
        double GetMean() const;
        double GetMeanStatUnc() const;
        double GetVar() const;
        double GetStdVar() const;
        double Chi2(const CountBase &) const;
        void Clear();
 
    protected:
        double SumW = 0;
        double SumW2 = 0;
        double SumX = 0;
        double SumX2 = 0;
    };
 
    inline void CountBase::Add(const CountBase &subCountBase)
    {
        SumW += subCountBase.SumW;
        SumW2 += subCountBase.SumW2;
        SumX += subCountBase.SumX;
        SumX2 += subCountBase.SumX2;
    }
 
    inline void CountBase::Add(double x, double w)
    {
        SumW += w;
        SumW2 += w * w;
        SumX += x * w;
        SumX2 += x * x * w;
    }
 
    inline double CountBase::Chi2(const CountBase &count) const
    {
        return pow(this->GetMean() - count.GetMean(), 2) / (pow(this->GetMeanStatUnc(), 2) + pow(count.GetMeanStatUnc(), 2));
    }
 
    inline void CountBase::Clear()
    {
        SumW = 0;
        SumW2 = 0;
        SumX = 0;
        SumX2 = 0;
    }
 
    inline double CountBase::GetSum() const { return SumX; }                           
    inline double CountBase::GetSum2() const { return SumX2; }                         
    inline double CountBase::GetSumW() const { return SumW; }                          
    inline double CountBase::GetSumW2() const { return SumW2; }                        
    inline double CountBase::GetSumStatUnc() const { return GetMean() * sqrt(SumW2); } 
    inline double CountBase::GetMean() const { return SumW != 0 ? SumX / SumW : 0; }   
 
    inline double CountBase::GetMeanStatUnc() const { return SumW != 0 ? GetStdVar() * sqrt(SumW2) / SumW : 0; }
    inline double CountBase::GetVar() const { return SumW != 0 ? (SumX2 / SumW - (SumX / SumW) * (SumX / SumW)) : 0; }
    inline double CountBase::GetStdVar() const { return sqrt(fabs(GetVar())); }
 
    class Count : public Result, public CountBase
    {
    public:
        Count(std::shared_ptr<MSGTool> MSG, std::shared_ptr<ConfigTool> c, const TString &rname, const TString &fname = "") : Result(MSG, c, rname, fname), CountBase(){};
        ~Count() = default;
        void Combine(std::shared_ptr<Result> result);
    };
 
    inline void Count::Combine(std::shared_ptr<Result> result)
    {
        auto subCount = std::dynamic_pointer_cast<std::remove_pointer<decltype(this)>::type>(result);
        if (subCount != nullptr)
        {
            SumW += subCount->SumW;
            SumW2 += subCount->SumW2;
            SumX += subCount->SumX;
            SumX2 += subCount->SumX2;
        }
    }
} // namespace NAGASH