Logo ROOT   6.30.04
Reference Guide
 All Namespaces Files Pages
RooCompositeDataStore.h
Go to the documentation of this file.
1 /*****************************************************************************
2  * Project: RooFit *
3  * Package: RooFitCore *
4  * File: $Id$
5  * Authors: *
6  * WV, Wouter Verkerke, UC Santa Barbara, verkerke@slac.stanford.edu *
7  * DK, David Kirkby, UC Irvine, dkirkby@uci.edu *
8  * *
9  * Copyright (c) 2000-2005, Regents of the University of California *
10  * and Stanford University. All rights reserved. *
11  * *
12  * Redistribution and use in source and binary forms, *
13  * with or without modification, are permitted according to the terms *
14  * listed in LICENSE (http://roofit.sourceforge.net/license.txt) *
15  *****************************************************************************/
16 #ifndef ROO_COMPOSITE_DATA_STORE
17 #define ROO_COMPOSITE_DATA_STORE
18 
19 #include "RooAbsDataStore.h"
20 #include "TString.h"
21 #include <map>
22 #include <string>
23 
24 class RooAbsArg ;
25 class RooArgList ;
26 class RooFormulaVar ;
27 class RooArgSet ;
28 class RooCategory ;
29 
30 
31 class RooCompositeDataStore : public RooAbsDataStore {
32 public:
33 
34  RooCompositeDataStore() ;
35 
36  // Ctors from DataStore
37  RooCompositeDataStore(const char* name, const char* title, const RooArgSet& vars, RooCategory& indexCat, std::map<std::string,RooAbsDataStore*> inputData) ;
38 
39  // Empty ctor
40  virtual RooAbsDataStore* clone(const char* newname=0) const { return new RooCompositeDataStore(*this,newname) ; }
41  virtual RooAbsDataStore* clone(const RooArgSet& vars, const char* newname=0) const { return new RooCompositeDataStore(*this,vars,newname) ; }
42 
43  RooCompositeDataStore(const RooCompositeDataStore& other, const char* newname=0) ;
44  RooCompositeDataStore(const RooCompositeDataStore& other, const RooArgSet& vars, const char* newname=0) ;
45  virtual ~RooCompositeDataStore() ;
46 
47  virtual void dump() ;
48 
49  // Write current row
50  virtual Int_t fill() ;
51 
52  virtual Double_t sumEntries() const ;
53 
54  // Retrieve a row
55  using RooAbsDataStore::get ;
56  virtual const RooArgSet* get(Int_t index) const ;
57  virtual Double_t weight() const ;
58  virtual Double_t weight(Int_t index) const ;
59  virtual Double_t weightError(RooAbsData::ErrorType etype=RooAbsData::Poisson) const ;
60  virtual void weightError(Double_t& lo, Double_t& hi, RooAbsData::ErrorType etype=RooAbsData::Poisson) const ;
61  virtual Bool_t isWeighted() const ;
62 
63  // Change observable name
64  virtual Bool_t changeObservableName(const char* from, const char* to) ;
65 
66  // Add one or more columns
67  virtual RooAbsArg* addColumn(RooAbsArg& var, Bool_t adjustRange=kTRUE) ;
68  virtual RooArgSet* addColumns(const RooArgList& varList) ;
69 
70  // Merge column-wise
71  RooAbsDataStore* merge(const RooArgSet& allvars, std::list<RooAbsDataStore*> dstoreList) ;
72 
73  RooCategory* index() { return _indexCat ; }
74 
75  // Add rows
76  virtual void append(RooAbsDataStore& other) ;
77 
78  // General & bookkeeping methods
79  virtual Bool_t valid() const ;
80  virtual Int_t numEntries() const ;
81  virtual void reset() ;
82 
83  // Buffer redirection routines used in inside RooAbsOptTestStatistics
84  virtual void attachBuffers(const RooArgSet& extObs) ;
85  virtual void resetBuffers() ;
86 
87  // Constant term optimizer interface
88  virtual void cacheArgs(const RooAbsArg* owner, RooArgSet& varSet, const RooArgSet* nset=0, Bool_t skipZeroWeights=kFALSE) ;
89  virtual const RooAbsArg* cacheOwner() { return 0 ; }
90  virtual void setArgStatus(const RooArgSet& set, Bool_t active) ;
91  virtual void resetCache() ;
92 
93  virtual void recalculateCache(const RooArgSet* /*proj*/, Int_t /*firstEvent*/, Int_t /*lastEvent*/, Int_t /*stepSize*/, Bool_t /*skipZeroWeights*/) ;
94  virtual Bool_t hasFilledCache() const ;
95 
96  void loadValues(const RooAbsDataStore *tds, const RooFormulaVar* select=0, const char* rangeName=0, Int_t nStart=0, Int_t nStop=2000000000) ;
97 
98  virtual void forceCacheUpdate() ;
99 
100  virtual std::vector<RooSpan<const double>> getBatch(std::size_t first, std::size_t last) const {
101  //TODO
102  std::cerr << "This functionality is not yet implemented for composite data stores." << std::endl;
103  assert(false);
104 
105  std::vector<double> vec(first, last);
106  return {RooSpan<const double>(vec)};
107  }
108 
109  protected:
110 
111  void attachCache(const RooAbsArg* newOwner, const RooArgSet& cachedVars) ;
112 
113  std::map<Int_t,RooAbsDataStore*> _dataMap ;
114  RooCategory* _indexCat ;
115  mutable RooAbsDataStore* _curStore ; //! Datastore associated with current event
116  mutable Int_t _curIndex ; //! Index associated with current event
117  Bool_t _ownComps ; //!
118 
119  ClassDef(RooCompositeDataStore,1) // Composite Data Storage class
120 };
121 
122 
123 #endif