BCDataSet.h

#ifndef __BCDATASET__H
#define __BCDATASET__H

/*
 * Copyright (C) 2007-2018, the BAT core developer team
 * All rights reserved.
 *
 * For the licensing terms see doc/COPYING.
 * For documentation see http://mpp.mpg.de/bat
 */

// ---------------------------------------------------------

#include <string>
#include <vector>

#include "BCDataPoint.h"
#include "BCLog.h"

class TGraph;
class TGraphErrors;
class TGraphAsymmErrors;
class TH2;

// ---------------------------------------------------------

class BCDataSet
{
public:

    BCDataSet(unsigned n = 0);

    virtual ~BCDataSet() {};

    BCDataPoint& operator[](unsigned index)
    {   return fDataVector[index]; }

    const BCDataPoint& operator[](unsigned index) const
    {   return fDataVector[index]; }

    unsigned GetNDataPoints() const
    { return fDataVector.size(); }

    unsigned GetNValuesPerPoint() const
    { return fNValuesPerPoint; }

    BCDataPoint& GetDataPoint(unsigned index)
    { return fDataVector.at(index); }

    const BCDataPoint& GetDataPoint(unsigned index) const
    { return fDataVector.at(index); }

    BCDataPoint& Back()
    { return fDataVector.back(); }

    std::vector<double> GetDataComponents(unsigned index) const;

    bool BoundsExist() const;

    const BCDataPoint& GetLowerBounds() const
    { return fLowerBounds; }

    const BCDataPoint& GetUpperBounds() const
    { return fUpperBounds; }

    BCDataPoint& GetUserLowerBounds()
    { return fUserLowerBounds; }

    BCDataPoint& GetUserUpperBounds()
    { return fUserUpperBounds; }

    double GetLowerBound(unsigned index) const;

    double GetUpperBound(unsigned index) const;

    double GetRangeWidth(unsigned index) const
    { return GetUpperBound(index) - GetLowerBound(index); }

    bool IsFixed(unsigned index) const
    { return (index < fFixed.size() and fFixed[index]); }

    void SetNValuesPerPoint(unsigned n);

    void SetBounds(unsigned index, double lower_bound, double upper_bound, bool fixed = false);

    void Fix(unsigned i, bool b = true)
    { if (i < fFixed.size()) fFixed[i] = b; }

    bool ReadDataFromFile(const std::string& filename, const std::string& treename, const std::string& branchnames, char delim = ',')
    { return ReadDataFromFileTree(filename, treename, branchnames, delim); };

    bool ReadDataFromFile(const std::string& filename, int nvariables)
    { return ReadDataFromFileTxt(filename, nvariables); };

    bool ReadDataFromFileTree(const std::string& filename, const std::string& treename, const std::string& branchnames, char delim = ',');

    bool ReadDataFromFileTxt(const std::string& filename, int nvariables);

    bool AddDataPoint(const BCDataPoint& datapoint);

    void AdjustBoundForUncertainties(unsigned i, double nSigma, unsigned i_err1, int i_err2 = -1);

    void Reset()
    { fDataVector.clear(); SetNValuesPerPoint(0); }

    void PrintSummary(void (*output)(const std::string&) = BCLog::OutSummary) const;

    TGraph* GetGraph(unsigned x, unsigned y) const;

    TGraphErrors* GetGraph(unsigned x, unsigned y, int ex, int ey) const;

    TGraphAsymmErrors* GetGraph(unsigned x, unsigned y, int ex_below, int ex_above, int ey_below, int ey_above) const;

    TH2* CreateH2(const char* name, const char* title, unsigned x, unsigned y, unsigned nbins_x = 100, unsigned nbins_y = 100, double x_padding = 0.10, double y_padding = 0.10) const;

private:
    std::vector<BCDataPoint> fDataVector;

    unsigned fNValuesPerPoint;

    BCDataPoint fLowerBounds;

    BCDataPoint fUpperBounds;

    BCDataPoint fUserLowerBounds;

    BCDataPoint fUserUpperBounds;

    std::vector<bool> fFixed;

};

// ---------------------------------------------------------

#endif