ROOT logo
/*****************************************************************************
 * Project: RooFit                                                           *
 * Package: RooFitCore                                                       *
 *    File: $Id: RooAbsCategory.h,v 1.38 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_ABS_CATEGORY
#define ROO_ABS_CATEGORY

#include "Riosfwd.h"
#include "TObjArray.h"
#include "RooAbsArg.h"
#include "RooCatType.h"

class TTree ;
class RooArgSet ;
class RooDataSet ;
class Roo1DTable ;

class RooAbsCategory : public RooAbsArg {
public:
  // Constructors, assignment etc.
  RooAbsCategory() { 
    // Default constructor
    _treeVar = kFALSE ; 
    _typeIter = _types.MakeIterator() ; 
  } ;
  RooAbsCategory(const char *name, const char *title);
  RooAbsCategory(const RooAbsCategory& other, const char* name=0) ;
  virtual ~RooAbsCategory();
  
  // Value accessors
  virtual Int_t getIndex() const ;
  virtual const char* getLabel() const ;
  Bool_t operator==(Int_t index) const ;
  Bool_t operator!=(Int_t index) {  return !operator==(index);}
  Bool_t operator==(const char* label) const ;
  Bool_t operator!=(const char* label) { return !operator==(label);}
  virtual Bool_t operator==(const RooAbsArg& other) ;
  Bool_t         operator!=(const RooAbsArg& other) { return !operator==(other);}
  
  Bool_t isValidIndex(Int_t index) const ;
  Bool_t isValidLabel(const char* label) const ;  
  const RooCatType* lookupType(Int_t index, Bool_t printError=kFALSE) const ;
  const RooCatType* lookupType(const char* label, Bool_t printError=kFALSE) const ;
  const RooCatType* lookupType(const RooCatType& type, Bool_t printError=kFALSE) const ;
  TIterator* typeIterator() const ;
  Int_t numTypes(const char* /*rangeName*/=0) const { 
    // Return number of types defined (in range named rangeName if rangeName!=0)
    return _types.GetEntries() ; 
  }
  Bool_t isSignType(Bool_t mustHaveZero=kFALSE) const ;

  Roo1DTable *createTable(const char *label) const ;

  // I/O streaming interface
  virtual Bool_t readFromStream(istream& is, Bool_t compact, Bool_t verbose=kFALSE) ;
  virtual void writeToStream(ostream& os, Bool_t compact) const ;

  virtual void printValue(ostream& os) const ;
  virtual void printMultiline(ostream& os, Int_t contents, Bool_t verbose=kFALSE, TString indent="") const ;

  virtual Bool_t isIntegrationSafeLValue(const RooArgSet* /*set*/) const { 
    // Is this l-value object safe for use as integration observable
    return kTRUE ; 
  }

  RooAbsArg *createFundamental(const char* newname=0) const;

protected:

  // Function evaluation and error tracing
  RooCatType traceEval() const ;
  virtual Bool_t traceEvalHook(RooCatType /*value*/) const { 
    // Hook function for trace evaluation (dummy)
    return kFALSE ;
  }
  virtual RooCatType evaluate() const = 0 ;

  // Type definition management
  const RooCatType* defineType(const char* label) ;
  const RooCatType* defineType(const char* label, Int_t index) ;
  const RooCatType* defineTypeUnchecked(const char* label, Int_t index) ;
  const RooCatType* getOrdinal(UInt_t n, const char* rangeName=0) const;
  void clearTypes() ;

  virtual Bool_t isValid() const ;
  virtual Bool_t isValid(RooCatType value) const ;

  virtual void syncCache(const RooArgSet* set=0) ;
  virtual void copyCache(const RooAbsArg* source, Bool_t valueOnly=kFALSE) ;
  virtual void attachToTree(TTree& t, Int_t bufSize=32000) ;
  virtual void setTreeBranchStatus(TTree& t, Bool_t active) ;
  virtual void fillTreeBranch(TTree& t) ;

  mutable UChar_t    _byteValue ; //! Transient cache for byte values from tree branches
  mutable RooCatType _value ; // Current value
  TObjArray  _types ;         // Array of allowed values
  TIterator* _typeIter ;      //!

  Bool_t _treeVar ;           //! do not persist

  ClassDef(RooAbsCategory,1) // Abstract discrete variable
};

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