AsymptoticExclusion.h

#ifndef AsymptoticExclusion_h
#define AsymptoticExclusion_h

#include "XeStat.h"
#include "XePdfObjects.h"
#include "TGraphAsymmErrors.h"
#include "XeUtils.h"

using namespace ROOT;
using namespace Math;

class AsymptoticExclusion : public XeStat{
    public:

    ~AsymptoticExclusion();
    AsymptoticExclusion(ProfileLikelihood *pl, double confidenceLevel);

    ProfileLikelihood* getProfileLikelihood() {return plike;};
    TGraphAsymmErrors  getExpectedMedian()    {return expectedMedian;};
    TGraphAsymmErrors  getExpected2Sigma()    {return expected2Sigma;};
        TGraph             getObservedLimit()     {return observedLimit;};
        double         getMass()          {return plike->getWimpMass();};

        void generateAndSetAsimov(double mu_prime);

    void setRealData();

        void setToyDataset(double seed, double mu_prime);
 
    double computeQTestStat(double mu, bool useStoredFit = false);

    void computeSensitivity();

    double computeSensitivityHagar();


    double computeExpectedLimit(double sigma_0, double N, double CL);

    double conditionalFit(double mu);

    double unconditionalFit();

        double getMu_hat();

    double computeSigmaAsimov(double mu_prime);


       void setNscanPoints(int nPoints){nScanPoints = nPoints;};



        void writeToFile(TString prefix ="limit_");

        TGraph getSigmaScan(){return sigmaScan;};

    TGraph getqTestScan() {return qTestScan;};
    

       void computeLimits();

    double compute_pval_s_plus_b(double q_obs);
    double compute_pval_b(double q_obs, double mu_val, double sigma_val);

        void setScanMin(double mi) {scanMin = mi;};
        void setScanMax(double ma) {scanMax = ma;};

    void LikelihoodScan();

    void setQTilde(bool doSet) {useQtilde = doSet;};  

    void setAlternativeXAxisVal(double x) { AlternativeX = x;};

    void setAlternative2DHistoRange(int Nx, double xmin, double xmax, int Ny, double ymin, double ymax);

    TGraph  getqTestScanData() {return qTestScanData;};

    double getObsLimitCLS(){return Obslimit;}
    double getObsLimitnoCLS(){return ObslimitnoCLS;}
    

    private:
        void computeSetAsimovSigma();

    ProfileLikelihood *plike;       // handler on profile likelihood

    double Obslimit, ObslimitnoCLS;
        
    TGraphAsymmErrors    expectedMedian;    //Graph holding the informations
    TGraphAsymmErrors    expected2Sigma;
    TGraph           observedLimit;
    TGraph           observedLimitNoCLS;
    TGraph           sigmaScan;
    TGraph           qTestScan;
    TGraph           qTestScanData;
    TH1F             pulls_uncond;
    TH1F             pulls_cond;
    TH2F             Histo2D_mass_vs_x;   //2D histo of the observed limits as a 
                         //function of mass and the additional variable X, 
                        //specified with setAlternativeXAxisVal()

    int          nScanPoints;   // Number of points for which mu is computed in limit calculation, default 100.

    vector<double>       asimovSigmaSet;
    vector<double>       muStepsSet;

    double           cl;            // confidence level DEFAULT = 0.1
    double           stored3Sigma_mu;
    double           storedMedian_mu;

    double           scanMin;
    double           scanMax;   

    bool             useQtilde;

    double               AlternativeX;

};


#endif