ToyGenerator.h

#ifndef TOY_GENERATOR
#define TOY_GENERATOR


#include "TRandom.h"
#include "TRint.h"
#include "TString.h"
#include "TFile.h"
#include "TTree.h"
#include "TList.h"
#include "XeUtils.h"
#include "XeLikelihoods.h"
#include "TParameter.h"
#include "XeStat.h"
#include "TTree.h"
#include "TH2F.h"
#include <map>
#include <vector>
#include <stdio.h>

using namespace std;


class ToyGenerator: public errorHandler {

    public:
        ToyGenerator(TString sampleName, TString outDir);

        void setLikelihood(pdfLikelihood *like) {likeHood = like;};

        //
        void setAverageCalibrationEvents(double evnt) {averageCalEvnt = evnt;} ;


        //
        void setAverageDataEvents(double evnt) {averageDataEvnt = evnt;} ;

        //
        // generate calibration like toys, where averageCalEvnt is the poisson median
        // of the number of generated event per toy. N is the number of toy datasets.
        // The pdfs are taken only from pdfcomponent that are "safeguarded" in the 
        // likelihood, their relative rates are respected. Also the additional 
        // contribution from "AdditionalSafeGuardComponent" is included.
        void generateCalibration(int N, bool randomizeNP = false);

        void generateData(double mu, int N, bool randomizeNP = false);

        void setSeed(int seed);
        
        void setGeneration(int generation) { Gen = generation; };

        void setLikeType(int lktype) {likelihoodType = lktype; } ;

        //
        void randomizeNuissanceParameter();

        void setTreeName(TString newname) { treeName = newname;};


    private:
        
        void saveParameters(TTree *f);
        
        double getModelIntegral();

        double getModelIntegralSafeguarded();

        vector<TH2F> getTH2OfBkg();


        double    averageCalEvnt;
        double    averageDataEvnt;
        int       Gen;
        int       likelihoodType;
        TRandom3  rambo;
        TString   treeName;
        TString   dir;
        pdfLikelihood *likeHood;
};




#endif