ToyMCStudy.h

Go to the documentation of this file.

// @(#)root/roostats:$Id$
// Author: Sven Kreiss and Kyle Cranmer    June 2010
/*************************************************************************
 * Copyright (C) 1995-2008, Rene Brun and Fons Rademakers.               *
 * All rights reserved.                                                  *
 *                                                                       *
 * For the licensing terms see $ROOTSYS/LICENSE.                         *
 * For the list of contributors see $ROOTSYS/README/CREDITS.             *
 *************************************************************************/
 
#ifndef ROOSTATS_ToyMCStudy
#define ROOSTATS_ToyMCStudy
 
#include "Rtypes.h"
 
#include "RooAbsStudy.h"
 
#include "RooStats/ToyMCSampler.h"
#include "RooStats/SamplingDistribution.h"
 
#include "RooWorkspace.h"
#include "RooArgSet.h"
 
#include "RooDataSet.h"
#include "RooLinkedList.h"
#include "RooLinkedListIter.h"
 
namespace RooStats {
 
class ToyMCStudy: public RooAbsStudy {
 
   public:
      /// need to have constructor without arguments for proof
      ToyMCStudy(const char *name = "ToyMCStudy", const char *title = "ToyMCStudy") :
         RooAbsStudy(name, title),
         fRandomSeed(0),
         fToyMCSampler(nullptr)
      {
         // In this case, this is the normal output. The SamplingDistribution
         // instances are stored as detailed output.
         storeDetailedOutput(true);
      }
 
      RooAbsStudy* clone(const char* /*newname*/="") const override { return new ToyMCStudy(*this) ; }
 
      // RooAbsStudy interfaces
      bool initialize(void) override;
      bool execute(void) override;
      bool finalize(void) override;
 
      RooDataSet* merge();
 
      void SetToyMCSampler(ToyMCSampler& t) { fToyMCSampler = &t; }
      void SetParamPoint(const RooArgSet& paramPoint) { fParamPoint.add(paramPoint); }
 
      void SetRandomSeed(unsigned int seed) { fRandomSeed = seed; }
 
   protected:
 
      unsigned int fRandomSeed;
      ToyMCSampler *fToyMCSampler;
      RooArgSet fParamPoint;
 
   protected:
   ClassDefOverride(ToyMCStudy,2); // toy MC study for parallel processing
 
};
 
 
class ToyMCPayload : public TNamed {
 
   public:
      ToyMCPayload() : fDataSet(nullptr)
      {
         // proof constructor, do not use
      }
 
      ToyMCPayload(RooDataSet *sd) : fDataSet(sd) {}
 
      RooDataSet* GetSamplingDistributions()
      {
         return fDataSet;
      }
 
   private:
      RooDataSet* fDataSet;
 
   protected:
   ClassDefOverride(ToyMCPayload,1);
};
 
 
}
 
 
#endif