QMultiScatterPlot.hh

#ifndef _Q_MULTI_SCATTER_PLOT_HH_
#define _Q_MULTI_SCATTER_PLOT_HH_

#include "QCuore.hh"
#include "QIntervalSet.hh"
#include <TF1.h>
#include <TGraph.h>
#include <TH1D.h>
#include <map>
#include <string>
#include <vector>

Q_BEGIN_NAMESPACE

namespace ROOT {class TH1D;}

class QMultiScatterPlot {

public:
    QMultiScatterPlot();
    
    QMultiScatterPlot(const QMultiScatterPlot& other);
    
    virtual ~QMultiScatterPlot();
    
    const QMultiScatterPlot& operator=(const QMultiScatterPlot& other);
    
    void AddEntry(const double x, const double y, const int graphIndex = 0);
    
    void AddFunction(const TF1* func, const int graphIndex = 0);
    
    void AutoSetDisplayInRange(
        const double xLowerBound, const double xUpperBound,
        const double yLowerBound, const double yUpperBound
    );
    
    void AutoSetXAxisDisplayInRange(
        const double xLowerBound, const double xUpperBound);
        
    void AutoSetYAxisDisplayInRange(
        const double yLowerBound, const double yUpperBound);
    
    void Clear();
    
    void Draw(Option_t* chopt = "");
    
    int GetChannel() const
    {
        return fChannel;
    }
    
    double GetMean(const Int_t axis = 2) const;
    
    double GetMedianAbsoluteDeviationInRange(const Int_t axis,
        const double lowerBound, const double upperBound) const;
    
    double GetMedianInRange(const Int_t axis, const double lowerBound,
                            const double upperBound) const;
    
    std::string GetName() const
    {
        return fName;
    }
    
    double GetRMS(const Int_t axis = 2) const;
    
    int GetRun() const
    {
        return fRun;
    }
        
    const std::string& GetVarName() const
    {
        return fVarName;
    }
    
    void Projection(const Int_t axis, TH1D& histogram);
    
    void SaveAs(const std::string filename, Option_t* option =  "");
    
    void SetChannel(const int channel)
    {
        fChannel = channel;
        SetName();
    }
        
    void SetName();

    void SetRun(const int run)
    {
        fRun = run;
        SetName();
    }
    
    void SetShadedIntervals(const QIntervalSet& intervals)
    {
        fShadedIntervals = intervals;
    }
    
    void SetTitle(const std::string title);

    void SetVarName(const std::string varName)
    {
        fVarName = varName;
        SetName();
    }

    void SetXVariable(const std::string var);

    void SetYVariable(const std::string var);

private:
    int fChannel;
    
    std::vector<Color_t> fColors;
    
    std::map<int, TF1*> fFunction;
    
    std::map<int, TGraph*> fGraph;
    
    TH1D* fHistogram;
    
    bool fIsXRangeAutoSet;
    
    bool fIsYRangeAutoSet;
    
    std::string fName;
    
    int fNumPoints;
    
    int fRun;
    
    QIntervalSet fShadedIntervals;
    
    std::string fTitle;
    
    std::string fVarName;
    
    double fXMax;
    
    double fXMin;
    
    std::string fXVariable;
    
    double fYMax;
    
    double fYMin;
    
    std::string fYVariable;

};

Q_END_NAMESPACE

#endif

---