ROOT logo
/*****************************************************************************
 * Project: RooFit                                                           *
 * Package: RooFitCore                                                       *
 *    File: $Id: RooMappedCategory.h,v 1.22 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_MAPPED_CATEGORY
#define ROO_MAPPED_CATEGORY

#include "TObjArray.h"
#include "RooAbsCategory.h"
#include "RooCategoryProxy.h"
#include "RooCatType.h"
#include "TRegexp.h"
#include <map>
#include <string>

class RooMappedCategory : public RooAbsCategory {
public:
  // Constructors etc.
  enum CatIdx { NoCatIdx=-99999 } ;
  inline RooMappedCategory() : _defCat(0) { }
  RooMappedCategory(const char *name, const char *title, RooAbsCategory& inputCat, const char* defCatName="NotMapped", Int_t defCatIdx=NoCatIdx);
  RooMappedCategory(const RooMappedCategory& other, const char *name=0) ;
  virtual TObject* clone(const char* newname) const { return new RooMappedCategory(*this,newname); }
  virtual ~RooMappedCategory();

  // Mapping function
  Bool_t map(const char* inKeyRegExp, const char* outKeyName, Int_t outKeyNum=NoCatIdx) ; 

  // Printing interface (human readable)
  void printMultiline(ostream& os, Int_t content, Bool_t verbose=kFALSE, TString indent="") const ;
  void printMetaArgs(ostream& os) const ;

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


  class Entry {
  public:
    inline Entry() : _regexp(0), _cat() {} 
    virtual ~Entry() { delete _regexp ; } ;
    Entry(const char* exp, const RooCatType* cat) : _expr(exp), _regexp(new TRegexp(mangle(exp),kTRUE)), _cat(*cat) {} 
    Entry(const Entry& other) : _expr(other._expr), _regexp(new TRegexp(mangle(other._expr.Data()),kTRUE)), _cat(other._cat) {} 
    inline Bool_t ok() { return (_regexp->Status()==TRegexp::kOK) ; }
    Bool_t match(const char* testPattern) const { return (TString(testPattern).Index(*_regexp)>=0) ; }
    inline const RooCatType& outCat() const { return _cat ; }
    Entry& operator=(const Entry& other);
    
  protected:
  
    TString mangle(const char* exp) const ;  

    TString _expr ;
    TRegexp* _regexp ; //!
    RooCatType _cat ;

    ClassDef(Entry,1) // Map cat entry definition
  } ;


protected:
    
  RooCatType* _defCat ;         // Default (unmapped) output type
  RooCategoryProxy _inputCat ;  // Input category
  std::map<std::string,RooMappedCategory::Entry> _mapArray ;  // List of mapping rules

  virtual RooCatType evaluate() const ; 

  ClassDef(RooMappedCategory,1) // Index variable, derived from another index using pattern-matching based mapping
};

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