/*****************************************************************************
* Project: RooFit *
* Package: RooFitCore *
* @(#)root/roofitcore:$Id$
* 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) *
*****************************************************************************/
//////////////////////////////////////////////////////////////////////////////
//
// BEGIN_HTML
// A RooPlot is a plot frame and a container for graphics objects
// within that frame. As a frame, it provides the TH1-style public interface
// for settting plot ranges, configuring axes, etc. As a container, it
// holds an arbitrary set of objects that might be histograms of data,
// curves representing a fit model, or text labels. Use the Draw()
// method to draw a frame and the objects it contains. Use the various
// add...() methods to add objects to be drawn. In general, the
// add...() methods create a private copy of the object you pass them
// and return a pointer to this copy. The caller owns the input object
// and this class owns the returned object.
//
// All RooAbsReal and RooAbsData derived classes implement plotOn()
// functions that facilitate to plot themselves on a given RooPlot, e.g.
//
// RooPlot *frame = x.frame() ;
// data.plotOn(frame) ;
// pdf.plotOn(frame) ;
//
// These high level functions also take care of any projections
// or other mappings that need to be made to plot a multi-dimensional
// object onto a one-dimensional plot.
// END_HTML
//
#include "RooFit.h"
#include "TClass.h"
#include "TH1D.h"
#include "TBrowser.h"
#include "TPad.h"
#include "RooPlot.h"
#include "RooAbsReal.h"
#include "RooAbsRealLValue.h"
#include "RooPlotable.h"
#include "RooArgSet.h"
#include "RooCurve.h"
#include "RooHist.h"
#include "RooMsgService.h"
#include "TAttLine.h"
#include "TAttFill.h"
#include "TAttMarker.h"
#include "TAttText.h"
#include "TDirectory.h"
#include "TDirectoryFile.h"
#include "Riostream.h"
#include
#include
using namespace std;
ClassImp(RooPlot)
;
Bool_t RooPlot::_addDirStatus = kTRUE ;
Bool_t RooPlot::addDirectoryStatus() { return _addDirStatus ; }
Bool_t RooPlot::setAddDirectoryStatus(Bool_t flag) { Bool_t ret = flag ; _addDirStatus = flag ; return ret ; }
//_____________________________________________________________________________
RooPlot::RooPlot() : _hist(0), _plotVarClone(0), _plotVarSet(0), _normVars(0), _normObj(0), _dir(0)
{
// Default constructor
// coverity[UNINIT_CTOR]
_iterator= _items.MakeIterator() ;
if (gDirectory && addDirectoryStatus()) {
_dir = gDirectory ;
gDirectory->Append(this) ;
}
}
//_____________________________________________________________________________
RooPlot::RooPlot(Double_t xmin, Double_t xmax) :
_hist(0), _items(), _plotVarClone(0), _plotVarSet(0), _normObj(0),
_defYmin(1e-5), _defYmax(1), _dir(0)
{
// Constructor of RooPlot with range [xmin,xmax]
Bool_t histAddDirStatus = TH1::AddDirectoryStatus();
TH1::AddDirectory(kFALSE) ;
_hist = new TH1D(histName(),"A RooPlot",100,xmin,xmax) ;
_hist->Sumw2(kFALSE) ;
_hist->GetSumw2()->Set(0) ;
TH1::AddDirectory(histAddDirStatus) ;
// Create an empty frame with the specified x-axis limits.
initialize();
}
//_____________________________________________________________________________
RooPlot::RooPlot(Double_t xmin, Double_t xmax, Double_t ymin, Double_t ymax) :
_hist(0), _items(), _plotVarClone(0),
_plotVarSet(0), _normObj(0), _defYmin(1e-5), _defYmax(0), _dir(0)
{
// Construct of a two-dimensioanl RooPlot with ranges [xmin,xmax] x [ymin,ymax]
Bool_t histAddDirStatus = TH1::AddDirectoryStatus();
TH1::AddDirectory(kFALSE) ;
_hist = new TH1D(histName(),"A RooPlot",100,xmin,xmax) ;
_hist->Sumw2(kFALSE) ;
_hist->GetSumw2()->Set(0) ;
TH1::AddDirectory(histAddDirStatus) ;
SetMinimum(ymin);
SetMaximum(ymax);
initialize();
}
//_____________________________________________________________________________
RooPlot::RooPlot(const RooAbsRealLValue &var1, const RooAbsRealLValue &var2) :
_hist(0), _items(),
_plotVarClone(0), _plotVarSet(0), _normObj(0), _defYmin(1e-5), _defYmax(0), _dir(0)
{
// Construct a two-dimensional RooPlot with ranges and properties taken
// from variables var1 and var2
Bool_t histAddDirStatus = TH1::AddDirectoryStatus();
TH1::AddDirectory(kFALSE) ;
_hist = new TH1D(histName(),"A RooPlot",100,var1.getMin(),var1.getMax()) ;
_hist->Sumw2(kFALSE) ;
_hist->GetSumw2()->Set(0) ;
TH1::AddDirectory(histAddDirStatus) ;
if(!var1.hasMin() || !var1.hasMax()) {
coutE(InputArguments) << "RooPlot::RooPlot: cannot create plot for variable without finite limits: "
<< var1.GetName() << endl;
return;
}
if(!var2.hasMin() || !var2.hasMax()) {
coutE(InputArguments) << "RooPlot::RooPlot: cannot create plot for variable without finite limits: "
<< var1.GetName() << endl;
return;
}
SetMinimum(var2.getMin());
SetMaximum(var2.getMax());
SetXTitle(var1.getTitle(kTRUE));
SetYTitle(var2.getTitle(kTRUE));
initialize();
}
//_____________________________________________________________________________
RooPlot::RooPlot(const RooAbsRealLValue &var1, const RooAbsRealLValue &var2,
Double_t xmin, Double_t xmax, Double_t ymin, Double_t ymax) :
_hist(0), _items(), _plotVarClone(0),
_plotVarSet(0), _normObj(0), _defYmin(1e-5), _defYmax(0), _dir(0)
{
// Construct a two-dimensional RooPlot with ranges and properties taken
// from variables var1 and var2 but with an overriding range definition
// of [xmin,xmax] x [ymin,ymax]
Bool_t histAddDirStatus = TH1::AddDirectoryStatus();
TH1::AddDirectory(kFALSE) ;
_hist = new TH1D(histName(),"A RooPlot",100,xmin,xmax) ;
_hist->Sumw2(kFALSE) ;
_hist->GetSumw2()->Set(0) ;
TH1::AddDirectory(histAddDirStatus) ;
SetMinimum(ymin);
SetMaximum(ymax);
SetXTitle(var1.getTitle(kTRUE));
SetYTitle(var2.getTitle(kTRUE));
initialize();
}
//_____________________________________________________________________________
RooPlot::RooPlot(const char* name, const char* title, const RooAbsRealLValue &var, Double_t xmin, Double_t xmax, Int_t nbins) :
_hist(0), _items(),
_plotVarClone(0), _plotVarSet(0), _normObj(0), _defYmin(1e-5), _defYmax(1), _dir(0)
{
// Create an 1-dimensional with all properties taken from 'var', but
// with an explicit range [xmin,xmax] and a default binning of 'nbins'
Bool_t histAddDirStatus = TH1::AddDirectoryStatus();
TH1::AddDirectory(kFALSE) ;
_hist = new TH1D(name,title,nbins,xmin,xmax) ;
_hist->Sumw2(kFALSE) ;
_hist->GetSumw2()->Set(0) ;
TH1::AddDirectory(histAddDirStatus) ;
// plotVar can be a composite in case of a RooDataSet::plot, need deepClone
_plotVarSet = (RooArgSet*) RooArgSet(var).snapshot() ;
_plotVarClone= (RooAbsRealLValue*)_plotVarSet->find(var.GetName()) ;
TString xtitle= var.getTitle(kTRUE);
SetXTitle(xtitle.Data());
initialize();
_normBinWidth = (xmax-xmin)/nbins ;
}
//_____________________________________________________________________________
RooPlot::RooPlot(const RooAbsRealLValue &var, Double_t xmin, Double_t xmax, Int_t nbins) :
_hist(0), _items(),
_plotVarClone(0), _plotVarSet(0), _normObj(0), _defYmin(1e-5), _defYmax(1), _dir(0)
{
// Create an 1-dimensional with all properties taken from 'var', but
// with an explicit range [xmin,xmax] and a default binning of 'nbins'
Bool_t histAddDirStatus = TH1::AddDirectoryStatus();
TH1::AddDirectory(kFALSE) ;
_hist = new TH1D(histName(),"RooPlot",nbins,xmin,xmax) ;
_hist->Sumw2(kFALSE) ;
_hist->GetSumw2()->Set(0) ;
TH1::AddDirectory(histAddDirStatus) ;
// plotVar can be a composite in case of a RooDataSet::plot, need deepClone
_plotVarSet = (RooArgSet*) RooArgSet(var).snapshot() ;
_plotVarClone= (RooAbsRealLValue*)_plotVarSet->find(var.GetName()) ;
TString xtitle= var.getTitle(kTRUE);
SetXTitle(xtitle.Data());
TString title("A RooPlot of \"");
title.Append(var.getTitle());
title.Append("\"");
SetTitle(title.Data());
initialize();
_normBinWidth = (xmax-xmin)/nbins ;
}
//_____________________________________________________________________________
RooPlot* RooPlot::emptyClone(const char* name)
{
// Return empty clone of current RooPlot
RooPlot* clone = new RooPlot(*_plotVarClone,_hist->GetXaxis()->GetXmin(),_hist->GetXaxis()->GetXmax(),_hist->GetNbinsX()) ;
clone->SetName(name) ;
return clone ;
}
//_____________________________________________________________________________
void RooPlot::initialize()
{
// Perform initialization that is common to all constructors.
SetName(histName()) ;
if (gDirectory && addDirectoryStatus()) {
_dir = gDirectory ;
gDirectory->Append(this) ;
}
// We do not have useful stats of our own
_hist->SetStats(kFALSE);
// Default vertical padding of our enclosed objects
setPadFactor(0.05);
// We don't know our normalization yet
_normNumEvts= 0;
_normBinWidth = 0;
_normVars= 0;
// Create an iterator over our enclosed objects
_iterator= _items.MakeIterator();
assert(0 != _iterator);
}
//_____________________________________________________________________________
TString RooPlot::histName() const
{
// Construct automatic name of internal TH1
if (_plotVarClone) {
return TString(Form("frame_%s_%lx",_plotVarClone->GetName(),(ULong_t)this)) ;
} else {
return TString(Form("frame_%lx",(ULong_t)this)) ;
}
}
//_____________________________________________________________________________
RooPlot::~RooPlot()
{
// Destructor
// Delete the items in our container and our iterator.
if (_dir) {
if (!_dir->TestBit(TDirectoryFile::kCloseDirectory)) {
_dir->GetList()->RecursiveRemove(this) ;
}
}
_items.Delete();
delete _iterator;
if(_plotVarSet) delete _plotVarSet;
if(_normVars) delete _normVars;
delete _hist ;
}
//_____________________________________________________________________________
void RooPlot::updateNormVars(const RooArgSet &vars)
{
// Install the given set of observables are reference normalization
// variables for this frame. These observables are e.g. later used
// to automatically project out observables when plotting functions
// on this frame. This function is only effective when called the
// first time on a frame
if(0 == _normVars) _normVars= (RooArgSet*) vars.snapshot(kTRUE);
}
//_____________________________________________________________________________
Stat_t RooPlot::GetBinContent(Int_t /*i*/) const {
// A plot object is a frame without any bin contents of its own so this
// method always returns zero.
return 0;
}
//_____________________________________________________________________________
Stat_t RooPlot::GetBinContent(Int_t, Int_t) const
{
// A plot object is a frame without any bin contents of its own so this
// method always returns zero.
return 0;
}
//_____________________________________________________________________________
Stat_t RooPlot::GetBinContent(Int_t, Int_t, Int_t) const
{
// A plot object is a frame without any bin contents of its own so this
// method always returns zero.
return 0;
}
//_____________________________________________________________________________
void RooPlot::addObject(TObject *obj, Option_t *drawOptions, Bool_t invisible)
{
// Add a generic object to this plot. The specified options will be
// used to Draw() this object later. The caller transfers ownership
// of the object with this call, and the object will be deleted
// when its containing plot object is destroyed.
if(0 == obj) {
coutE(InputArguments) << fName << "::addObject: called with a null pointer" << endl;
return;
}
DrawOpt opt(drawOptions) ;
opt.invisible = invisible ;
_items.Add(obj,opt.rawOpt());
}
//_____________________________________________________________________________
void RooPlot::addTH1(TH1 *hist, Option_t *drawOptions, Bool_t invisible)
{
// Add a TH1 histogram object to this plot. The specified options
// will be used to Draw() this object later. "SAME" will be added to
// the options if they are not already present. The caller transfers
// ownership of the object with this call, and the object will be
// deleted when its containing plot object is destroyed.
if(0 == hist) {
coutE(InputArguments) << fName << "::addTH1: called with a null pointer" << endl;
return;
}
// check that this histogram is really 1D
if(1 != hist->GetDimension()) {
coutE(InputArguments) << fName << "::addTH1: cannot plot histogram with "
<< hist->GetDimension() << " dimensions" << endl;
return;
}
// add option "SAME" if necessary
TString options(drawOptions);
options.ToUpper();
if(!options.Contains("SAME")) options.Append("SAME");
// update our y-axis label and limits
updateYAxis(hist->GetMinimum(),hist->GetMaximum(),hist->GetYaxis()->GetTitle());
// use this histogram's normalization if necessary
updateFitRangeNorm(hist);
// add the histogram to our list
addObject(hist,options.Data(),invisible);
}
//_____________________________________________________________________________
void RooPlot::addPlotable(RooPlotable *plotable, Option_t *drawOptions, Bool_t invisible, Bool_t refreshNorm)
{
// Add the specified plotable object to our plot. Increase our y-axis
// limits to fit this object if necessary. The default lower-limit
// is zero unless we are plotting an object that takes on negative values.
// This call transfers ownership of the plotable object to this class.
// The plotable object will be deleted when this plot object is deleted.
// update our y-axis label and limits
updateYAxis(plotable->getYAxisMin(),plotable->getYAxisMax(),plotable->getYAxisLabel());
// use this object's normalization if necessary
updateFitRangeNorm(plotable,refreshNorm) ;
// add this element to our list and remember its drawing option
TObject *obj= plotable->crossCast();
if(0 == obj) {
coutE(InputArguments) << fName << "::add: cross-cast to TObject failed (nothing added)" << endl;
}
else {
DrawOpt opt(drawOptions) ;
opt.invisible = invisible ;
_items.Add(obj,opt.rawOpt());
}
}
//_____________________________________________________________________________
void RooPlot::updateFitRangeNorm(const TH1* hist)
{
// Update our plot normalization over our plot variable's fit range,
// which will be determined by the first suitable object added to our plot.
const TAxis* xa = ((TH1*)hist)->GetXaxis() ;
_normBinWidth = (xa->GetXmax()-xa->GetXmin())/hist->GetNbinsX() ;
_normNumEvts = hist->GetEntries()/_normBinWidth ;
}
//_____________________________________________________________________________
void RooPlot::updateFitRangeNorm(const RooPlotable* rp, Bool_t refreshNorm)
{
// Update our plot normalization over our plot variable's fit range,
// which will be determined by the first suitable object added to our plot.
if (_normNumEvts != 0) {
// If refresh feature is disabled stop here
if (!refreshNorm) return ;
Double_t corFac(1.0) ;
if (dynamic_cast(rp)) corFac = _normBinWidth/rp->getFitRangeBinW() ;
if (fabs(rp->getFitRangeNEvt()/corFac-_normNumEvts)>1e-6) {
coutI(Plotting) << "RooPlot::updateFitRangeNorm: New event count of " << rp->getFitRangeNEvt()/corFac
<< " will supercede previous event count of " << _normNumEvts << " for normalization of PDF projections" << endl ;
}
// Nominal bin width (i.e event density) is already locked in by previously drawn histogram
// scale this histogram to match that density
_normNumEvts = rp->getFitRangeNEvt()/corFac ;
_normObj = rp ;
// cout << "correction factor = " << _normBinWidth << "/" << rp->getFitRangeBinW() << endl ;
// cout << "updating numevts to " << _normNumEvts << endl ;
} else {
_normObj = rp ;
_normNumEvts = rp->getFitRangeNEvt() ;
if (rp->getFitRangeBinW()) {
_normBinWidth = rp->getFitRangeBinW() ;
}
// cout << "updating numevts to " << _normNumEvts << endl ;
}
}
//_____________________________________________________________________________
void RooPlot::updateYAxis(Double_t ymin, Double_t ymax, const char *label)
{
// Update our y-axis limits to accomodate an object whose spread
// in y is (ymin,ymax). Use the specified y-axis label if we don't
// have one assigned already.
// force an implicit lower limit of zero if appropriate
if(GetMinimum() == 0 && ymin > 0) ymin= 0;
// calculate padded values
Double_t ypad= getPadFactor()*(ymax-ymin);
ymax+= ypad;
if(ymin < 0) ymin-= ypad;
// update our limits if necessary
if(GetMaximum() < ymax) {
_defYmax = ymax ;
SetMaximum(ymax);
// if we don't do this - Unzoom on y-axis will reset upper bound to 1
_hist->SetBinContent(1,ymax) ;
}
if(GetMinimum() > ymin) {
_defYmin = ymin ;
SetMinimum(ymin);
}
// use the specified y-axis label if we don't have one already
if(0 == strlen(_hist->GetYaxis()->GetTitle())) _hist->SetYTitle(label);
}
//_____________________________________________________________________________
void RooPlot::Draw(Option_t *option)
{
// Draw this plot and all of the elements it contains. The specified options
// only apply to the drawing of our frame. The options specified in our add...()
// methods will be used to draw each object we contain.
TString optArg = option ;
optArg.ToLower() ;
// This draw options prevents the histogram with one dummy entry
// to be drawn
if (optArg.Contains("same")) {
_hist->Draw("FUNCSAME");
} else {
_hist->Draw("FUNC");
}
_iterator->Reset();
TObject *obj = 0;
while((obj= _iterator->Next())) {
DrawOpt opt(_iterator->GetOption()) ;
if (!opt.invisible) {
obj->Draw(opt.drawOptions);
}
}
_hist->Draw("AXISSAME");
}
//_____________________________________________________________________________
void RooPlot::printName(ostream& os) const
{
// Print frame name
os << GetName() ;
}
//_____________________________________________________________________________
void RooPlot::printTitle(ostream& os) const
{
// Print frame title
os << GetTitle() ;
}
//_____________________________________________________________________________
void RooPlot::printClassName(ostream& os) const
{
// Print frame class name
os << IsA()->GetName() ;
}
//_____________________________________________________________________________
void RooPlot::printArgs(ostream& os) const
{
if (_plotVarClone) {
os << "[" ;
_plotVarClone->printStream(os,kName,kInline) ;
os << "]" ;
}
}
//_____________________________________________________________________________
void RooPlot::printValue(ostream& os) const
{
// Print frame arguments
os << "(" ;
_iterator->Reset();
TObject *obj = 0;
Bool_t first(kTRUE) ;
while((obj= _iterator->Next())) {
if (first) {
first=kFALSE ;
} else {
os << "," ;
}
if(obj->IsA()->InheritsFrom(RooPrintable::Class())) {
RooPrintable* po = dynamic_cast(obj) ;
// coverity[FORWARD_NULL]
po->printStream(os,kClassName|kName,kInline) ;
}
// is it a TNamed subclass?
else {
os << obj->ClassName() << "::" << obj->GetName() ;
}
}
os << ")" ;
}
//_____________________________________________________________________________
void RooPlot::printMultiline(ostream& os, Int_t /*content*/, Bool_t verbose, TString indent) const
{
// Frame detailed printing
TString deeper(indent);
deeper.Append(" ");
if(0 != _plotVarClone) {
os << indent << "RooPlot " << GetName() << " (" << GetTitle() << ") plots variable ";
_plotVarClone->printStream(os,kName|kTitle,kSingleLine,"");
}
else {
os << indent << "RooPlot " << GetName() << " (" << GetTitle() << ") has no associated plot variable" << endl ;
}
os << indent << " Plot frame contains " << _items.GetSize() << " object(s):" << endl;
if(verbose) {
_iterator->Reset();
TObject *obj = 0;
Int_t i=0 ;
while((obj= _iterator->Next())) {
os << deeper << "[" << i++ << "] (Options=\"" << _iterator->GetOption() << "\") ";
// Is this a printable object?
if(obj->IsA()->InheritsFrom(RooPrintable::Class())) {
RooPrintable* po = dynamic_cast(obj) ;
if (po) {
po->printStream(os,kName|kClassName|kArgs|kExtras,kSingleLine) ;
}
}
// is it a TNamed subclass?
else {
os << obj->ClassName() << "::" << obj->GetName() << endl;
}
}
}
}
//_____________________________________________________________________________
const char* RooPlot::nameOf(Int_t idx) const
{
// Return the name of the object at slot 'idx' in this RooPlot.
// If the given index is out of range, return a null pointer
TObject* obj = _items.At(idx) ;
if (!obj) {
coutE(InputArguments) << "RooPlot::nameOf(" << GetName() << ") index " << idx << " out of range" << endl ;
return 0 ;
}
return obj->GetName() ;
}
//_____________________________________________________________________________
TObject* RooPlot::getObject(Int_t idx) const
{
// Return the name of the object at slot 'idx' in this RooPlot.
// If the given index is out of range, return a null pointer
TObject* obj = _items.At(idx) ;
if (!obj) {
coutE(InputArguments) << "RooPlot::getObject(" << GetName() << ") index " << idx << " out of range" << endl ;
return 0 ;
}
return obj ;
}
//_____________________________________________________________________________
TAttLine *RooPlot::getAttLine(const char *name) const
{
// Return a pointer to the line attributes of the named object in this plot,
// or zero if the named object does not exist or does not have line attributes.
return dynamic_cast(findObject(name));
}
//_____________________________________________________________________________
TAttFill *RooPlot::getAttFill(const char *name) const
{
// Return a pointer to the fill attributes of the named object in this plot,
// or zero if the named object does not exist or does not have fill attributes.
return dynamic_cast(findObject(name));
}
//_____________________________________________________________________________
TAttMarker *RooPlot::getAttMarker(const char *name) const
{
// Return a pointer to the marker attributes of the named object in this plot,
// or zero if the named object does not exist or does not have marker attributes.
return dynamic_cast(findObject(name));
}
//_____________________________________________________________________________
TAttText *RooPlot::getAttText(const char *name) const
{
// Return a pointer to the text attributes of the named object in this plot,
// or zero if the named object does not exist or does not have text attributes.
return dynamic_cast(findObject(name));
}
//_____________________________________________________________________________
RooCurve* RooPlot::getCurve(const char* name) const
{
// Return a RooCurve pointer of the named object in this plot,
// or zero if the named object does not exist or is not a RooCurve
return dynamic_cast(findObject(name)) ;
}
//_____________________________________________________________________________
RooHist* RooPlot::getHist(const char* name) const
{
// Return a RooCurve pointer of the named object in this plot,
// or zero if the named object does not exist or is not a RooCurve
return dynamic_cast(findObject(name)) ;
}
//_____________________________________________________________________________
void RooPlot::remove(const char* name, Bool_t deleteToo)
{
// Remove object with given name, or last object added if no name is given.
// If deleteToo is true (default), the object removed from the RooPlot is
// also deleted.
TObject* obj = findObject(name) ;
if (!obj) {
if (name) {
coutE(InputArguments) << "RooPlot::remove(" << GetName() << ") ERROR: no object found with name " << name << endl ;
} else {
coutE(InputArguments) << "RooPlot::remove(" << GetName() << ") ERROR: plot frame is empty, cannot remove last object" << endl ;
}
return ;
}
_items.Remove(obj) ;
if (deleteToo) {
delete obj ;
}
}
//_____________________________________________________________________________
Bool_t RooPlot::drawBefore(const char *before, const char *target)
{
// Change the order in which our contained objects are drawn so that
// the target object is drawn just before the specified object.
// Returns kFALSE if either object does not exist.
return _items.moveBefore(before, target, caller("drawBefore"));
}
//_____________________________________________________________________________
Bool_t RooPlot::drawAfter(const char *after, const char *target)
{
// Change the order in which our contained objects are drawn so that
// the target object is drawn just after the specified object.
// Returns kFALSE if either object does not exist.
return _items.moveAfter(after, target, caller("drawAfter"));
}
//_____________________________________________________________________________
TObject *RooPlot::findObject(const char *name, const TClass* clas) const
{
// Find the named object in our list of items and return a pointer
// to it. Return zero and print a warning message if the named
// object cannot be found. If no name is supplied the last object
// added is returned.
//
// Note that the returned pointer is to a
// TObject and so will generally need casting. Use the getAtt...()
// methods to change the drawing style attributes of a contained
// object directly.
TObject *obj = 0;
TObject *ret = 0;
TIterator* iter = _items.MakeIterator() ;
while((obj=iter->Next())) {
if ((!name || !TString(name).CompareTo(obj->GetName())) &&
(!clas || (obj->IsA()==clas))) {
ret = obj ;
}
}
delete iter ;
if (ret==0) {
coutE(InputArguments) << "RooPlot::findObject(" << GetName() << ") cannot find object " << (name?name:"") << endl ;
}
return ret ;
}
//_____________________________________________________________________________
TString RooPlot::getDrawOptions(const char *name) const
{
// Return the Draw() options registered for the named object. Return
// an empty string if the named object cannot be found.
TObjOptLink *link= _items.findLink(name,caller("getDrawOptions"));
DrawOpt opt(0 == link ? "" : link->GetOption()) ;
return TString(opt.drawOptions) ;
}
//_____________________________________________________________________________
Bool_t RooPlot::setDrawOptions(const char *name, TString options)
{
// Register the specified drawing options for the named object.
// Return kFALSE if the named object cannot be found.
TObjOptLink *link= _items.findLink(name,caller("setDrawOptions"));
if(0 == link) return kFALSE;
DrawOpt opt(link->GetOption()) ;
strlcpy(opt.drawOptions,options,128) ;
link->SetOption(opt.rawOpt());
return kTRUE;
}
//_____________________________________________________________________________
Bool_t RooPlot::getInvisible(const char* name) const
{
// Returns true of object with given name is set to be invisible
TObjOptLink *link= _items.findLink(name,caller("getInvisible"));
if(0 == link) return kFALSE;
return DrawOpt(link->GetOption()).invisible ;
}
//_____________________________________________________________________________
void RooPlot::setInvisible(const char* name, Bool_t flag)
{
// If flag is true object with 'name' is set to be invisible
// i.e. it is not drawn when Draw() is called
TObjOptLink *link= _items.findLink(name,caller("getInvisible"));
DrawOpt opt ;
if(link) {
opt.initialize(link->GetOption()) ;
opt.invisible = flag ;
link->SetOption(opt.rawOpt()) ;
}
}
//_____________________________________________________________________________
TString RooPlot::caller(const char *method) const
{
// Utility function
TString name(fName);
if(strlen(method)) {
name.Append("::");
name.Append(method);
}
return name;
}
//_____________________________________________________________________________
void RooPlot::SetMaximum(Double_t maximum)
{
// Set maximum value of Y axis
_hist->SetMaximum(maximum==-1111?_defYmax:maximum) ;
}
//_____________________________________________________________________________
void RooPlot::SetMinimum(Double_t minimum)
{
// Set minimum value of Y axis
_hist->SetMinimum(minimum==-1111?_defYmin:minimum) ;
}
//_____________________________________________________________________________
Double_t RooPlot::chiSquare(const char* curvename, const char* histname, Int_t nFitParam) const
{
// Calculate and return reduced chi-squared of curve with given name with respect
// to histogram with given name. If nFitParam is non-zero, it is used to reduce the
// number of degrees of freedom for a chi^2 for a curve that was fitted to the
// data with that number of floating parameters
// Find curve object
RooCurve* curve = (RooCurve*) findObject(curvename,RooCurve::Class()) ;
if (!curve) {
coutE(InputArguments) << "RooPlot::chiSquare(" << GetName() << ") cannot find curve" << endl ;
return -1. ;
}
// Find histogram object
RooHist* hist = (RooHist*) findObject(histname,RooHist::Class()) ;
if (!hist) {
coutE(InputArguments) << "RooPlot::chiSquare(" << GetName() << ") cannot find histogram" << endl ;
return -1. ;
}
return curve->chiSquare(*hist,nFitParam) ;
}
//_____________________________________________________________________________
RooHist* RooPlot::residHist(const char* histname, const char* curvename, bool normalize, bool useAverage) const
{
// Return a RooHist containing the residuals of histogram 'histname' with respect
// to curve 'curvename'. If normalize is true the residuals are divided by the error
// on the histogram, effectively returning a pull histogram
// Find curve object
RooCurve* curve = (RooCurve*) findObject(curvename,RooCurve::Class()) ;
if (!curve) {
coutE(InputArguments) << "RooPlot::residHist(" << GetName() << ") cannot find curve" << endl ;
return 0 ;
}
// Find histogram object
RooHist* hist = (RooHist*) findObject(histname,RooHist::Class()) ;
if (!hist) {
coutE(InputArguments) << "RooPlot::residHist(" << GetName() << ") cannot find histogram" << endl ;
return 0 ;
}
return hist->makeResidHist(*curve,normalize,useAverage) ;
}
//_____________________________________________________________________________
void RooPlot::DrawOpt::initialize(const char* inRawOpt)
{
// Initialize the DrawOpt helper class
if (!inRawOpt) {
drawOptions[0] = 0 ;
invisible=kFALSE ;
return ;
}
strlcpy(drawOptions,inRawOpt,128) ;
strtok(drawOptions,":") ;
const char* extraOpt = strtok(0,":") ;
if (extraOpt) {
invisible = (extraOpt[0]=='I') ;
}
}
//_____________________________________________________________________________
const char* RooPlot::DrawOpt::rawOpt() const
{
// Return the raw draw options
static char buf[128] ;
strlcpy(buf,drawOptions,128) ;
if (invisible) {
strlcat(buf,":I",128) ;
}
return buf ;
}
//_____________________________________________________________________________
Double_t RooPlot::getFitRangeNEvt(Double_t xlo, Double_t xhi) const
{
// Return the number of events that is associated with the range [xlo,xhi]
// This method is only fully functional for ranges not equal to the full
// range if the object that inserted the normalization data provided
// a link to an external object that can calculate the event count in
// in sub ranges. An error will be printed if this function is used
// on sub-ranges while that information is not available
Double_t scaleFactor = 1.0 ;
if (_normObj) {
scaleFactor = _normObj->getFitRangeNEvt(xlo,xhi)/_normObj->getFitRangeNEvt() ;
} else {
coutW(Plotting) << "RooPlot::getFitRangeNEvt(" << GetName() << ") WARNING: Unable to obtain event count in range "
<< xlo << " to " << xhi << ", substituting full event count" << endl ;
}
return getFitRangeNEvt()*scaleFactor ;
}
//_____________________________________________________________________________
void RooPlot::SetName(const char *name)
{
// Set the name of the RooPlot to 'name'
if (_dir) _dir->GetList()->Remove(this);
TNamed::SetName(name) ;
if (_dir) _dir->GetList()->Add(this);
}
//_____________________________________________________________________________
void RooPlot::SetNameTitle(const char *name, const char* title)
{
// Set the name and title of the RooPlot to 'name' and 'title'
if (_dir) _dir->GetList()->Remove(this);
TNamed::SetNameTitle(name,title) ;
if (_dir) _dir->GetList()->Add(this);
}
//_____________________________________________________________________________
void RooPlot::SetTitle(const char* title)
{
// Set the title of the RooPlot to 'title'
TNamed::SetTitle(title) ;
_hist->SetTitle(title) ;
}
//_____________________________________________________________________________
Int_t RooPlot::defaultPrintContents(Option_t* /*opt*/) const
{
// Define default print options, for a given print style
return kName|kArgs|kValue ;
}
TAxis* RooPlot::GetXaxis() const { return _hist->GetXaxis() ; }
TAxis* RooPlot::GetYaxis() const { return _hist->GetYaxis() ; }
Int_t RooPlot::GetNbinsX() const { return _hist->GetNbinsX() ; }
Int_t RooPlot::GetNdivisions(Option_t* axis) const { return _hist->GetNdivisions(axis) ; }
Double_t RooPlot::GetMinimum(Double_t minval) const { return _hist->GetMinimum(minval) ; }
Double_t RooPlot::GetMaximum(Double_t maxval) const { return _hist->GetMaximum(maxval) ; }
void RooPlot::SetAxisColor(Color_t color, Option_t* axis) { _hist->SetAxisColor(color,axis) ; }
void RooPlot::SetAxisRange(Double_t xmin, Double_t xmax, Option_t* axis) { _hist->SetAxisRange(xmin,xmax,axis) ; }
void RooPlot::SetBarOffset(Float_t offset) { _hist->SetBarOffset(offset) ; }
void RooPlot::SetBarWidth(Float_t width) { _hist->SetBarWidth(width) ; }
void RooPlot::SetContour(Int_t nlevels, const Double_t* levels) { _hist->SetContour(nlevels,levels) ; }
void RooPlot::SetContourLevel(Int_t level, Double_t value) { _hist->SetContourLevel(level,value) ; }
void RooPlot::SetDrawOption(Option_t* option) { _hist->SetDrawOption(option) ; }
void RooPlot::SetFillAttributes() { _hist->SetFillAttributes() ; }
void RooPlot::SetFillColor(Color_t fcolor) { _hist->SetFillColor(fcolor) ; }
void RooPlot::SetFillStyle(Style_t fstyle) { _hist->SetFillStyle(fstyle) ; }
void RooPlot::SetLabelColor(Color_t color, Option_t* axis) { _hist->SetLabelColor(color,axis) ; }
void RooPlot::SetLabelFont(Style_t font, Option_t* axis) { _hist->SetLabelFont(font,axis) ; }
void RooPlot::SetLabelOffset(Float_t offset, Option_t* axis) { _hist->SetLabelOffset(offset,axis) ; }
void RooPlot::SetLabelSize(Float_t size, Option_t* axis) { _hist->SetLabelSize(size,axis) ; }
void RooPlot::SetLineAttributes() { _hist->SetLineAttributes() ; }
void RooPlot::SetLineColor(Color_t lcolor) { _hist->SetLineColor(lcolor) ; }
void RooPlot::SetLineStyle(Style_t lstyle) { _hist->SetLineStyle(lstyle) ; }
void RooPlot::SetLineWidth(Width_t lwidth) { _hist->SetLineWidth(lwidth) ; }
void RooPlot::SetMarkerAttributes() { _hist->SetMarkerAttributes() ; }
void RooPlot::SetMarkerColor(Color_t tcolor) { _hist->SetMarkerColor(tcolor) ; }
void RooPlot::SetMarkerSize(Size_t msize) { _hist->SetMarkerSize(msize) ; }
void RooPlot::SetMarkerStyle(Style_t mstyle) { _hist->SetMarkerStyle(mstyle) ; }
void RooPlot::SetNdivisions(Int_t n, Option_t* axis) { _hist->SetNdivisions(n,axis) ; }
void RooPlot::SetOption(Option_t* option) { _hist->SetOption(option) ; }
void RooPlot::SetStats(Bool_t stats) { _hist->SetStats(stats) ; }
void RooPlot::SetTickLength(Float_t length, Option_t* axis) { _hist->SetTickLength(length,axis) ; }
void RooPlot::SetTitleFont(Style_t font, Option_t* axis) { _hist->SetTitleFont(font,axis) ; }
void RooPlot::SetTitleOffset(Float_t offset, Option_t* axis) { _hist->SetTitleOffset(offset,axis) ; }
void RooPlot::SetTitleSize(Float_t size, Option_t* axis) { _hist->SetTitleSize(size,axis) ; }
void RooPlot::SetXTitle(const char *title) { _hist->SetXTitle(title) ; }
void RooPlot::SetYTitle(const char *title) { _hist->SetYTitle(title) ; }
void RooPlot::SetZTitle(const char *title) { _hist->SetZTitle(title) ; }
//______________________________________________________________________________
void RooPlot::Browse(TBrowser * /*b*/)
{
// Plot RooPlot when double-clicked in browser
Draw();
gPad->Update();
}
//_____________________________________________________________________________
void RooPlot::Streamer(TBuffer &R__b)
{
// Custom streamer, needed for backward compatibility
if (R__b.IsReading()) {
TH1::AddDirectory(kFALSE) ;
UInt_t R__s, R__c;
Version_t R__v = R__b.ReadVersion(&R__s, &R__c);
if (R__v > 1) {
R__b.ReadClassBuffer(RooPlot::Class(),this,R__v,R__s,R__c);
} else {
// backward compatible streamer code here
// Version 1 of RooPlot was deriving from TH1 and RooPrintable
// Version 2 derives instead from TNamed and RooPrintable
_hist = new TH1F();
_hist->TH1::Streamer(R__b);
SetName(_hist->GetName());
SetTitle(_hist->GetTitle());
RooPrintable::Streamer(R__b);
_items.Streamer(R__b);
R__b >> _padFactor;
R__b >> _plotVarClone;
R__b >> _plotVarSet;
R__b >> _normVars;
R__b >> _normNumEvts;
R__b >> _normBinWidth;
R__b >> _defYmin;
R__b >> _defYmax;
R__b.CheckByteCount(R__s, R__c, RooPlot::IsA());
}
TH1::AddDirectory(kTRUE) ;
} else {
R__b.WriteClassBuffer(RooPlot::Class(),this);
}
}