BCRooInterface.h

#ifndef __BCROOINTERFACE__H
#define __BCROOINTERFACE__H

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

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

#include <list>
#include <utility>

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

class BCRooInterface : public BCModel
{
   public:

      BCRooInterface( const char* 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;}
      //RooArgSet GetArgSetForMarkovChainTest(){return _parametersForMarkovChain_test;}

   private:

      void AddToCurrentChainElement(double xij, int chainNum, int poiNum); //help function for construction of RooStats Markov Chain
      bool EqualsLastChainElement(int chainNum); //help function for construction of RooStats Markov Chain
      double GetWeightForChain(int chainNum); //help function for construction of RooStats Markov Chain

      RooAbsData* fData;        
      RooAbsPdf*  fModel;       
      RooAbsReal* fNll;         
      RooArgSet*  fObservables; 
      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