/***************************************************************************** * Project: RooFit * * Package: RooFitCore * * File: $Id: RooPlot.h,v 1.37 2007/06/18 11:52:41 wouter 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_PLOT #define ROO_PLOT #include #include "RooList.h" #include "RooPrintable.h" #include "TNamed.h" class TH1 ; class RooAbsReal; class RooAbsRealLValue; class RooArgSet ; class RooHist; class RooCurve ; class RooPlotable; class TDirectory ; class TAttLine; class TAttFill; class TAttMarker; class TAttText; class TClass ; class TAxis; class TBrowser ; class RooPlot : public TNamed, public RooPrintable { public: RooPlot() ; RooPlot(const char* name, const char* title, const RooAbsRealLValue &var, Double_t xmin, Double_t xmax, Int_t nBins) ; RooPlot(const RooAbsRealLValue &var, Double_t xmin, Double_t xmax, Int_t nBins); RooPlot(Double_t xmin, Double_t xmax); RooPlot(Double_t xmin, Double_t xmax, Double_t ymin, Double_t ymax); RooPlot(const RooAbsRealLValue &var1, const RooAbsRealLValue &var2); RooPlot(const RooAbsRealLValue &var1, const RooAbsRealLValue &var2, Double_t xmin, Double_t xmax, Double_t ymin, Double_t ymax); virtual ~RooPlot(); RooPlot* emptyClone(const char* name) ; // implement the TH1 interface virtual Stat_t GetBinContent(Int_t) const; virtual Stat_t GetBinContent(Int_t, Int_t) const; virtual Stat_t GetBinContent(Int_t, Int_t, Int_t) const; virtual void Draw(Option_t *options= 0); // forwarding of relevant TH1 interface TAxis* GetXaxis() const ; TAxis* GetYaxis() const ; Int_t GetNbinsX() const ; Int_t GetNdivisions(Option_t* axis = "X") const ; Double_t GetMinimum(Double_t minval = -FLT_MAX) const ; Double_t GetMaximum(Double_t maxval = FLT_MAX) const ; void SetAxisColor(Color_t color = 1, Option_t* axis = "X") ; void SetAxisRange(Double_t xmin, Double_t xmax, Option_t* axis = "X") ; void SetBarOffset(Float_t offset = 0.25) ; void SetBarWidth(Float_t width = 0.5) ; void SetContour(Int_t nlevels, const Double_t* levels = 0) ; void SetContourLevel(Int_t level, Double_t value) ; void SetDrawOption(Option_t* option = "") ; void SetFillAttributes() ; void SetFillColor(Color_t fcolor) ; void SetFillStyle(Style_t fstyle) ; void SetLabelColor(Color_t color = 1, Option_t* axis = "X") ; void SetLabelFont(Style_t font = 62, Option_t* axis = "X") ; void SetLabelOffset(Float_t offset = 0.005, Option_t* axis = "X") ; void SetLabelSize(Float_t size = 0.02, Option_t* axis = "X") ; void SetLineAttributes() ; void SetLineColor(Color_t lcolor) ; void SetLineStyle(Style_t lstyle) ; void SetLineWidth(Width_t lwidth) ; void SetMarkerAttributes() ; void SetMarkerColor(Color_t tcolor = 1) ; void SetMarkerSize(Size_t msize = 1) ; void SetMarkerStyle(Style_t mstyle = 1) ; void SetName(const char *name) ; void SetTitle(const char *name) ; void SetNameTitle(const char *name, const char* title) ; void SetNdivisions(Int_t n = 510, Option_t* axis = "X") ; void SetOption(Option_t* option = " ") ; void SetStats(Bool_t stats = kTRUE) ; void SetTickLength(Float_t length = 0.02, Option_t* axis = "X") ; void SetTitleFont(Style_t font = 62, Option_t* axis = "X") ; void SetTitleOffset(Float_t offset = 1, Option_t* axis = "X") ; void SetTitleSize(Float_t size = 0.02, Option_t* axis = "X") ; void SetXTitle(const char* title) ; void SetYTitle(const char* title) ; void SetZTitle(const char* title) ; // container management const char* nameOf(Int_t idx) const ; TObject *findObject(const char *name, const TClass* clas=0) const; TObject* getObject(Int_t idx) const ; Stat_t numItems() const {return _items.GetSize();} void addPlotable(RooPlotable *plotable, Option_t *drawOptions= "", Bool_t invisible=kFALSE, Bool_t refreshNorm=kFALSE); void addObject(TObject* obj, Option_t* drawOptions= "", Bool_t invisible=kFALSE); void addTH1(TH1 *hist, Option_t* drawOptions= "", Bool_t invisible=kFALSE); void remove(const char* name=0, Bool_t deleteToo=kTRUE) ; // ascii printing virtual void printName(std::ostream& os) const ; virtual void printTitle(std::ostream& os) const ; virtual void printClassName(std::ostream& os) const ; virtual void printArgs(std::ostream& os) const ; virtual void printValue(std::ostream& os) const ; virtual void printMultiline(std::ostream& os, Int_t content, Bool_t verbose=kFALSE, TString indent="") const ; virtual Int_t defaultPrintContents(Option_t* opt) const ; inline virtual void Print(Option_t *options= 0) const { printStream(defaultPrintStream(),defaultPrintContents(options),defaultPrintStyle(options)); } // data member get/set methods inline RooAbsRealLValue *getPlotVar() const { return _plotVarClone; } inline Double_t getFitRangeNEvt() const { return _normNumEvts; } Double_t getFitRangeNEvt(Double_t xlo, Double_t xhi) const ; inline Double_t getFitRangeBinW() const { return _normBinWidth; } inline Double_t getPadFactor() const { return _padFactor; } inline void setPadFactor(Double_t factor) { if(factor >= 0) _padFactor= factor; } void updateNormVars(const RooArgSet &vars); const RooArgSet *getNormVars() const { return _normVars; } // get attributes of contained objects TAttLine *getAttLine(const char *name=0) const; TAttFill *getAttFill(const char *name=0) const; TAttMarker *getAttMarker(const char *name=0) const; TAttText *getAttText(const char *name=0) const; // Convenient type-safe accessors RooCurve* getCurve(const char* name=0) const ; RooHist* getHist(const char* name=0) const ; // rearrange drawing order of contained objects Bool_t drawBefore(const char *before, const char *target); Bool_t drawAfter(const char *after, const char *target); // get/set drawing options for contained objects TString getDrawOptions(const char *name) const; Bool_t setDrawOptions(const char *name, TString options); Bool_t getInvisible(const char* name) const ; void setInvisible(const char* name, Bool_t flag=kTRUE) ; virtual void SetMaximum(Double_t maximum = -1111) ; virtual void SetMinimum(Double_t minimum = -1111) ; Double_t chiSquare(int nFitParam=0) const { return chiSquare(0,0,nFitParam) ; } Double_t chiSquare(const char* pdfname, const char* histname, int nFitParam=0) const ; RooHist* residHist(const char* histname=0, const char* pdfname=0,bool normalize=false, bool useAverage=kFALSE) const ; RooHist* pullHist(const char* histname=0, const char* pdfname=0, bool useAverage=false) const { return residHist(histname,pdfname,true,useAverage); } void Browse(TBrowser *b) ; static Bool_t addDirectoryStatus() ; static Bool_t setAddDirectoryStatus(Bool_t flag) ; protected: RooPlot(const RooPlot& other); // cannot be copied class DrawOpt { public: DrawOpt(const char* _rawOpt=0) : invisible(kFALSE) { drawOptions[0] = 0 ; initialize(_rawOpt) ; } void initialize(const char* _rawOpt) ; const char* rawOpt() const ; char drawOptions[128] ; Bool_t invisible ; } ; void initialize(); TString histName() const ; TString caller(const char *method) const; void updateYAxis(Double_t ymin, Double_t ymax, const char *label= ""); void updateFitRangeNorm(const TH1* hist); void updateFitRangeNorm(const RooPlotable* rp, Bool_t refeshNorm=kFALSE); TH1* _hist ; // Histogram that we uses as basis for drawing the content RooList _items; // A list of the items we contain. Double_t _padFactor; // Scale our y-axis to _padFactor of our maximum contents. RooAbsRealLValue *_plotVarClone; // A clone of the variable we are plotting. RooArgSet *_plotVarSet; // A list owning the cloned tree nodes of the plotVarClone RooArgSet *_normVars; // Variables that PDF plots should be normalized over const RooPlotable* _normObj ; //! Pointer to normalization object ; Double_t _normNumEvts; // Number of events in histogram (for normalization) Double_t _normBinWidth; // Histogram bin width (for normalization) TIterator *_iterator; //! non-persistent Double_t _defYmin ; // Default minimum for Yaxis (as calculated from contents) Double_t _defYmax ; // Default maximum for Yaxis (as calculated from contents) TDirectory* _dir ; //! non-persistent static Bool_t _addDirStatus ; // static flag controlling AutoDirectoryAdd feature ClassDef(RooPlot,2) // Plot frame and container for graphics objects }; #endif