ROOT logo
/*****************************************************************************
 * Project: RooFit                                                           *
 *                                                                           *
 * 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 ROOABSCACHEDREAL
#define ROOABSCACHEDREAL

#include "RooAbsPdf.h"
#include "RooRealProxy.h"
#include "RooAbsReal.h"
#include "RooHistFunc.h"
#include "RooObjCacheManager.h"
#include <map>
class RooChangeTracker ;
class RooArgSet ;
 
class RooAbsCachedReal : public RooAbsReal {
public:

  RooAbsCachedReal() {} ;
  RooAbsCachedReal(const char *name, const char *title, Int_t ipOrder=0);
  RooAbsCachedReal(const RooAbsCachedReal& other, const char* name=0) ;
  virtual ~RooAbsCachedReal() ;

  virtual Double_t getValV(const RooArgSet* set=0) const ;
  virtual Bool_t selfNormalized() const { 
    // Declares function self normalized
    return kTRUE ; 
  }

  void setInterpolationOrder(Int_t order) ;
  Int_t getInterpolationOrder() const { 
    // Set interpolation order in RooHistFuncs that represent cache histograms
    return _ipOrder ; 
  }

  virtual Bool_t forceAnalyticalInt(const RooAbsArg& /*dep*/) const { 
    // Force all observables to be offered for internal integration
    return kTRUE ; 
  }
  
  virtual Int_t getAnalyticalIntegralWN(RooArgSet& allVars, RooArgSet& analVars, const RooArgSet* normSet, const char* rangeName=0) const ; 
  virtual Double_t analyticalIntegralWN(Int_t code, const RooArgSet* normSet, const char* rangeName=0) const ;

  void disableCache(Bool_t flag) { 
    // Switch to disable caching mechanism
    _disableCache = flag ; 
  }
   
protected:

  class FuncCacheElem : public RooAbsCacheElement {
  public:
    FuncCacheElem(const RooAbsCachedReal& self, const RooArgSet* nset) ;
    virtual ~FuncCacheElem() ;

    // Cache management functions
    virtual RooArgList containedArgs(Action) ;
    virtual void printCompactTreeHook(std::ostream&, const char *, Int_t, Int_t) ;

    RooHistFunc* func() { return _func ; }
    RooDataHist* hist() { return _hist ; }
    RooChangeTracker* paramTracker() { return _paramTracker ; }

    RooAbsReal* sourceClone() { return _sourceClone ; } 
    void setSourceClone(RooAbsReal* newSource) { delete _sourceClone ; _sourceClone = newSource ; }

    Bool_t cacheSource() { return _cacheSource ; }
    void setCacheSource(Bool_t flag) { _cacheSource = flag ; }

  private:
    // Payload
    RooHistFunc*      _func ;
    RooChangeTracker* _paramTracker ;
    RooDataHist*      _hist ;
    RooAbsReal*       _sourceClone ;
    Bool_t            _cacheSource ;
  } ;

  FuncCacheElem* getCache(const RooArgSet* nset) const ;
  void clearCacheObject(FuncCacheElem& cache) const ;

  virtual const char* payloadUniqueSuffix() const { return 0 ; }

  friend class FuncCacheElem ;
  virtual const char* binningName() const { 
    // Returns name of binning to be used for cache histogram creation
    return "cache" ; 
  }
  virtual FuncCacheElem* createCache(const RooArgSet* nset) const ;
  virtual const char* inputBaseName() const = 0 ;
  virtual RooArgSet* actualObservables(const RooArgSet& nset) const = 0 ;
  virtual RooArgSet* actualParameters(const RooArgSet& nset) const = 0 ;
  virtual void fillCacheObject(FuncCacheElem& cache) const = 0 ;

  mutable RooObjCacheManager _cacheMgr ; //! The cache manager

  
  Int_t _ipOrder ; // Interpolation order for cache histograms 
 
  TString cacheNameSuffix(const RooArgSet& nset) const ;

  mutable std::map<Int_t,std::pair<const RooArgSet*,const RooArgSet*> > _anaIntMap ; //! Map for analytical integration codes
  

private:

  Bool_t _disableCache ; // Flag to run object in passthrough (= non-caching mode)

  ClassDef(RooAbsCachedReal,1) // Abstract base class for cached p.d.f.s
};
 
#endif
 RooAbsCachedReal.h:1
 RooAbsCachedReal.h:2
 RooAbsCachedReal.h:3
 RooAbsCachedReal.h:4
 RooAbsCachedReal.h:5
 RooAbsCachedReal.h:6
 RooAbsCachedReal.h:7
 RooAbsCachedReal.h:8
 RooAbsCachedReal.h:9
 RooAbsCachedReal.h:10
 RooAbsCachedReal.h:11
 RooAbsCachedReal.h:12
 RooAbsCachedReal.h:13
 RooAbsCachedReal.h:14
 RooAbsCachedReal.h:15
 RooAbsCachedReal.h:16
 RooAbsCachedReal.h:17
 RooAbsCachedReal.h:18
 RooAbsCachedReal.h:19
 RooAbsCachedReal.h:20
 RooAbsCachedReal.h:21
 RooAbsCachedReal.h:22
 RooAbsCachedReal.h:23
 RooAbsCachedReal.h:24
 RooAbsCachedReal.h:25
 RooAbsCachedReal.h:26
 RooAbsCachedReal.h:27
 RooAbsCachedReal.h:28
 RooAbsCachedReal.h:29
 RooAbsCachedReal.h:30
 RooAbsCachedReal.h:31
 RooAbsCachedReal.h:32
 RooAbsCachedReal.h:33
 RooAbsCachedReal.h:34
 RooAbsCachedReal.h:35
 RooAbsCachedReal.h:36
 RooAbsCachedReal.h:37
 RooAbsCachedReal.h:38
 RooAbsCachedReal.h:39
 RooAbsCachedReal.h:40
 RooAbsCachedReal.h:41
 RooAbsCachedReal.h:42
 RooAbsCachedReal.h:43
 RooAbsCachedReal.h:44
 RooAbsCachedReal.h:45
 RooAbsCachedReal.h:46
 RooAbsCachedReal.h:47
 RooAbsCachedReal.h:48
 RooAbsCachedReal.h:49
 RooAbsCachedReal.h:50
 RooAbsCachedReal.h:51
 RooAbsCachedReal.h:52
 RooAbsCachedReal.h:53
 RooAbsCachedReal.h:54
 RooAbsCachedReal.h:55
 RooAbsCachedReal.h:56
 RooAbsCachedReal.h:57
 RooAbsCachedReal.h:58
 RooAbsCachedReal.h:59
 RooAbsCachedReal.h:60
 RooAbsCachedReal.h:61
 RooAbsCachedReal.h:62
 RooAbsCachedReal.h:63
 RooAbsCachedReal.h:64
 RooAbsCachedReal.h:65
 RooAbsCachedReal.h:66
 RooAbsCachedReal.h:67
 RooAbsCachedReal.h:68
 RooAbsCachedReal.h:69
 RooAbsCachedReal.h:70
 RooAbsCachedReal.h:71
 RooAbsCachedReal.h:72
 RooAbsCachedReal.h:73
 RooAbsCachedReal.h:74
 RooAbsCachedReal.h:75
 RooAbsCachedReal.h:76
 RooAbsCachedReal.h:77
 RooAbsCachedReal.h:78
 RooAbsCachedReal.h:79
 RooAbsCachedReal.h:80
 RooAbsCachedReal.h:81
 RooAbsCachedReal.h:82
 RooAbsCachedReal.h:83
 RooAbsCachedReal.h:84
 RooAbsCachedReal.h:85
 RooAbsCachedReal.h:86
 RooAbsCachedReal.h:87
 RooAbsCachedReal.h:88
 RooAbsCachedReal.h:89
 RooAbsCachedReal.h:90
 RooAbsCachedReal.h:91
 RooAbsCachedReal.h:92
 RooAbsCachedReal.h:93
 RooAbsCachedReal.h:94
 RooAbsCachedReal.h:95
 RooAbsCachedReal.h:96
 RooAbsCachedReal.h:97
 RooAbsCachedReal.h:98
 RooAbsCachedReal.h:99
 RooAbsCachedReal.h:100
 RooAbsCachedReal.h:101
 RooAbsCachedReal.h:102
 RooAbsCachedReal.h:103
 RooAbsCachedReal.h:104
 RooAbsCachedReal.h:105
 RooAbsCachedReal.h:106
 RooAbsCachedReal.h:107
 RooAbsCachedReal.h:108
 RooAbsCachedReal.h:109
 RooAbsCachedReal.h:110
 RooAbsCachedReal.h:111
 RooAbsCachedReal.h:112
 RooAbsCachedReal.h:113
 RooAbsCachedReal.h:114
 RooAbsCachedReal.h:115
 RooAbsCachedReal.h:116
 RooAbsCachedReal.h:117
 RooAbsCachedReal.h:118
 RooAbsCachedReal.h:119
 RooAbsCachedReal.h:120