XePdfObjects.h

#ifndef XEPDFOBJECTS_H
#define XEPDFOBJECTS_H

#include "XeStat.h"
#include <TString.h>
#include "TGraph2D.h"
#include "TNtuple.h"
#include <cmath>
#include <csignal>
#include <iostream>
#include <vector>
#include <utility>      // std::pair, std::make_pair
#include "dataHandler.h"
#include "XeUtils.h"
#include "TColor.h"

using namespace std;


class shapeSys : public LKParameter {

   public:

    shapeSys(TString name);


    double getNearestLow();

    double getNearestHigh();

};


class scaleSys : public LKParameter {

    public:
        // relativeUncertainty = 1 sigma relative uncertainty (example 10% uncertainty would be 0.1)
        scaleSys(TString name, double relativeUncertainty);

        // gives the Normalization scale factor according to t-valued uncertainty  --> (1 + t_val * relativeUncertainty )
        // Uses the current t-value which is automatically linked to its pdfLikelihood
        double getNormModifier();


        //this case return a sys that is centered in zero, half a gaussian, t-value strictly positive
        //and for a tvalue=0 have zero events. Histo is supposed to be normalized to 1
        void setNull(){ isNull = true; setMinimum(0.);};

    private:
        double relUnc;
        bool isNull;


};


class pdfComponent :public errorHandler{

   public:
         pdfComponent(TString name, TString filename);

         pdfComponent(TString component_name, TString filename, TString hist_name);

    ~pdfComponent();
    
    void autoLoad(TString tag="",char dd='_') {myShapeUnc=(scanFile(tag,dd));};

    vector< shapeSys * > scanFile(TString tag="",char dd='_');
    
    void addScaleSys(scaleSys *addMe) { myScaleUnc.push_back(addMe); };

    void addShapeSys(shapeSys *addMe) { myShapeUnc.push_back(addMe); };

    void loadHistos();

    void loadDefaultHisto();


    double getNormalizedDensity(double s1, double s2);

    double getDefaultDensity(double s1, double s2);

    double getNormalizedEvents();

    double getDefaultEvents();


    TH2F  getInterpolatedHisto();

    TH2F  getDefaultHisto();

    TString getNearestHistoName(vector<bool> setOfVal);

    TString getDefaultHistoName();

    TString getParamValueString();

    TString getParamValueWritable();


    double getDefaultPdfIntegral(double s1_min, double s1_max, double s2_min, double s2_max);


    void plotInterpolatedSpace(bool doProjectionX, double min, double max, int Nsteps, bool legend_left = false, double y_max = 0.);

    /* getDefaultHisto, getInterpolatedHisto, getNormalizedDensity,
     * getDefaultDensity, getNormalizedEvents, getDefaultEvents
     */
    void setScaleFactor(double val) {scaleFactor = val;};

    void setEvents(double nevents);

    scaleSys* getScaleSys(TString name);
    shapeSys* getShapeSys(TString name);
    void      replaceUncertainty(TString name, scaleSys* newScale);
    void      replaceUncertainty(TString name, shapeSys* newShape);

    TString getName()  {return pdf_name;};

  TString getComponentName()  {return component_name;};
  
    vector<scaleSys*>       myScaleUnc; 
    vector<shapeSys*>       myShapeUnc; 

    TString                         suffix;
    bool                            doExtend;

   private:
    TFile                   *file;
    TH2F                            *defaultDistro;
    vector<TH2F*>           histos;        
    vector<double>      InterpFactors;  
    TString             pdf_name;
  TString           component_name;
    vector<double>          old_t_val;    
    double                          scaleFactor;


    void extendHisto(TH2F &h);

};



class histoCompare : public errorHandler{

    public:

        histoCompare();
        ~histoCompare();

      //---------OPTIONS-----------//
        int      rebinX;
        int      rebinY;
        bool     projectionX;
        double   projectionMin;    
        double   projectionMax;    
        int      binMin;        
        int      binMax;
        bool     doStack;   
        TString  titleX;
        TString  titleY;
       //---------------------------//

        void setBaseHisto(TH2F b, TString n=""){base = b; names[0] = n;};
        void addHistoToList(TH2F h, TString n ="") {compareList.push_back(h); names.push_back(n);};
        void setNameofComponent(unsigned int i, TString n);
        void compare();

        void printModels();

        void compareWithRatio();

        void drawLegend(TH1D *baseH, vector <TH1D*> list );

        //return a string of info
        TString projectionInfo();

        //------ histo holders --------//
        TH2F               base;
        vector <TString>   names;
        vector <TH2F>      compareList;

        TH1D               *projectedBase;
        vector <TH1D*>      projectedList;

     private:

        void setOptions(TH1D *h, bool dataLike = true, bool isBottom = false);
        void project(); 
};






#endif