BCRooInterface.h

#ifndef __BCROOINTERFACE__H
#define __BCROOINTERFACE__H

#include "../../BAT/BCModel.h"

#include <RooStats/MarkovChain.h>
#include <RooRealVar.h>

#include <list>
#include <string>
#include <utility>

class RooAbsReal;
class RooAbsData;
class RooAbsPdf;
class RooArgSet;
class RooArgList;

class BCRooInterface : public BCModel
{
public:

    BCRooInterface(const std::string& name = "", bool fillChain = false );

    ~BCRooInterface();

    void DefineParameters();
    double LogAPrioriProbability(const std::vector<double>& parameters);
    double LogLikelihood(const std::vector<double>& parameters);
    void Initialize( RooAbsData& data,
                     RooAbsPdf& model,
                     RooAbsPdf& prior,
                     const RooArgSet* params,
                     const RooArgSet& listPOI );

    void Initialize( const char* rootFile,
                     const char* wsName = "batWS",
                     const char* dataName = "data",
                     const char* modelName = "model",
                     const char* priorName = "priorPOI",
                     const char* priorNuisanceName = "priorNuisance",
                     const char* paramsName = "parameters",
                     const char* listPOIName = "POI" );

    void SetNumBins(const char* parname, int nbins);
    void SetNumBins(int nbins);
    void SetupRooStatsMarkovChain();
    void MCMCIterationInterface();
    RooStats::MarkovChain* GetRooStatsMarkovChain() { return _roostatsMarkovChain;}
    RooArgSet* GetArgSetForMarkovChain() {return &_parametersForMarkovChainCurrent;}
private:

    void AddToCurrentChainElement(double xij, int chainNum, int poiNum); 
    bool EqualsLastChainElement(int chainNum); 
    double GetWeightForChain(int chainNum); 

    RooAbsData* fData;        
    RooAbsPdf* fModel;        
    RooAbsReal* fNll;         
    RooArgList* fParams;      
    RooArgList* fParamsPOI;   
    RooAbsPdf* fPrior;        
    int _default_nbins;

    RooRealVar* priorhelpvar;
    bool _addeddummyprior;

    bool _fillChain;          
    bool fFirstComparison;    
    RooStats::MarkovChain* _roostatsMarkovChain;
    RooArgSet _parametersForMarkovChainPrevious; 
    RooArgSet _parametersForMarkovChainCurrent; 

    std::vector< std::vector<double> > fPreviousStep; 
    std::vector< std::vector<double> > fCurrentStep; 
    std::vector< double > fVecWeights; 

    std::list< std::pair<const char*, int> > _nbins_list;
};

#endif