ROOT logo
/*****************************************************************************
 * Project: RooFit                                                           *
 * Package: RooFitCore                                                       *
 *    File: $Id: RooAcceptReject.h,v 1.16 2007/05/11 09:11:30 verkerke Exp $
 * Authors:                                                                  *
 *   WV, Wouter Verkerke, UC Santa Barbara, verkerke@slac.stanford.edu       *
 *   DK, David Kirkby,    UC Irvine,         dkirkby@uci.edu                 *
 *                                                                           *
 * Copyright (c) 2000-2005, Regents of the University of California          *
 *                          and Stanford University. All rights reserved.    *
 *                                                                           *
 * Redistribution and use in source and binary forms,                        *
 * with or without modification, are permitted according to the terms        *
 * listed in LICENSE (http://roofit.sourceforge.net/license.txt)             *
 *****************************************************************************/
#ifndef ROO_ACCEPT_REJECT
#define ROO_ACCEPT_REJECT

#include "RooAbsNumGenerator.h"
#include "RooPrintable.h"
#include "RooArgSet.h"

class RooAbsReal;
class RooRealVar;
class RooDataSet;
class RooRealBinding;
class RooNumGenFactory ;

class RooAcceptReject : public RooAbsNumGenerator {
public:
  RooAcceptReject() : _nextCatVar(0), _nextRealVar(0) {
    // coverity[UNINIT_CTOR]
  } ; 
  RooAcceptReject(const RooAbsReal &func, const RooArgSet &genVars, const RooNumGenConfig& config, Bool_t verbose=kFALSE, const RooAbsReal* maxFuncVal=0);
  RooAbsNumGenerator* clone(const RooAbsReal& func, const RooArgSet& genVars, const RooArgSet& /*condVars*/, 
			    const RooNumGenConfig& config, Bool_t verbose=kFALSE, const RooAbsReal* maxFuncVal=0) const {
    return new RooAcceptReject(func,genVars,config,verbose,maxFuncVal) ;
  }
  virtual ~RooAcceptReject();

  const RooArgSet *generateEvent(UInt_t remaining, Double_t& resampleRatio);
  Double_t getFuncMax() ;


  // Advertisement of capabilities
  virtual Bool_t canSampleConditional() const { return kTRUE ; }
  virtual Bool_t canSampleCategories() const { return kTRUE ; }

 
protected:

  friend class RooNumGenFactory ;
  static void registerSampler(RooNumGenFactory& fact) ;	

  void addEventToCache();
  const RooArgSet *nextAcceptedEvent();

  Double_t _maxFuncVal, _funcSum;      // Maximum function value found, and sum of all samples made
  UInt_t _realSampleDim,_catSampleMult;// Number of real and discrete dimensions to be samplesd
  UInt_t _minTrials;                   // Minimum number of max.finding trials, total number of samples
  UInt_t _totalEvents;                 // Total number of function samples
  UInt_t _eventsUsed;                  // Accepted number of function samples
  TIterator *_nextCatVar;              // Iterator of categories to be generated
  TIterator *_nextRealVar;             // Iterator over variables to be generated

  UInt_t _minTrialsArray[4];           // Minimum number of trials samples for 1,2,3 dimensional problems

  ClassDef(RooAcceptReject,0) // Context for generating a dataset from a PDF
};

#endif
 RooAcceptReject.h:1
 RooAcceptReject.h:2
 RooAcceptReject.h:3
 RooAcceptReject.h:4
 RooAcceptReject.h:5
 RooAcceptReject.h:6
 RooAcceptReject.h:7
 RooAcceptReject.h:8
 RooAcceptReject.h:9
 RooAcceptReject.h:10
 RooAcceptReject.h:11
 RooAcceptReject.h:12
 RooAcceptReject.h:13
 RooAcceptReject.h:14
 RooAcceptReject.h:15
 RooAcceptReject.h:16
 RooAcceptReject.h:17
 RooAcceptReject.h:18
 RooAcceptReject.h:19
 RooAcceptReject.h:20
 RooAcceptReject.h:21
 RooAcceptReject.h:22
 RooAcceptReject.h:23
 RooAcceptReject.h:24
 RooAcceptReject.h:25
 RooAcceptReject.h:26
 RooAcceptReject.h:27
 RooAcceptReject.h:28
 RooAcceptReject.h:29
 RooAcceptReject.h:30
 RooAcceptReject.h:31
 RooAcceptReject.h:32
 RooAcceptReject.h:33
 RooAcceptReject.h:34
 RooAcceptReject.h:35
 RooAcceptReject.h:36
 RooAcceptReject.h:37
 RooAcceptReject.h:38
 RooAcceptReject.h:39
 RooAcceptReject.h:40
 RooAcceptReject.h:41
 RooAcceptReject.h:42
 RooAcceptReject.h:43
 RooAcceptReject.h:44
 RooAcceptReject.h:45
 RooAcceptReject.h:46
 RooAcceptReject.h:47
 RooAcceptReject.h:48
 RooAcceptReject.h:49
 RooAcceptReject.h:50
 RooAcceptReject.h:51
 RooAcceptReject.h:52
 RooAcceptReject.h:53
 RooAcceptReject.h:54
 RooAcceptReject.h:55
 RooAcceptReject.h:56
 RooAcceptReject.h:57
 RooAcceptReject.h:58
 RooAcceptReject.h:59
 RooAcceptReject.h:60
 RooAcceptReject.h:61
 RooAcceptReject.h:62
 RooAcceptReject.h:63
 RooAcceptReject.h:64
 RooAcceptReject.h:65
 RooAcceptReject.h:66
 RooAcceptReject.h:67
 RooAcceptReject.h:68
 RooAcceptReject.h:69
 RooAcceptReject.h:70
 RooAcceptReject.h:71