rf508_listsetmanip

Organization and simultaneous fits: RooArgSet and RooArgList tools and tricks

Author: Wouter Verkerke
This notebook tutorial was automatically generated with ROOTBOOK-izer from the macro found in the ROOT repository on Wednesday, November 30, 2022 at 11:23 AM.

In [1]:
%%cpp -d
#include "RooRealVar.h"
#include "RooDataSet.h"
#include "RooGaussian.h"
#include "TCanvas.h"
#include "TAxis.h"
#include "RooPlot.h"
#include "RooArgSet.h"
#include "RooArgList.h"
#include "RooCategory.h"
using namespace RooFit;

Create dummy objects

Create some variables

In [2]:
RooRealVar a("a", "a", 1, -10, 10);
RooRealVar b("b", "b", 2, -10, 10);
RooRealVar c("c", "c", 3, -10, 10);
RooRealVar d("d", "d", 4, -10, 10);
RooRealVar x("x", "x", 0, -10, 10);
c.setError(0.5);
a.setConstant();
b.setConstant();

Create a category

In [3]:
RooCategory e("e", "e");
e.defineType("sig");
e.defineType("bkg");

Create a pdf

In [4]:
RooGaussian g("g", "g", x, a, b);
[#0] WARNING:InputArguments -- The parameter 'b' with range [-10, 10] of the RooGaussian 'g' exceeds the safe range of (0, inf). Advise to limit its range.

Creating, filling RooArgSets

A RooArgSet is a set of RooAbsArg objects. Each object in the set must have a unique name

Set constructors exists with up to 9 initial arguments

In [5]:
RooArgSet s(a, b);

At any time objects can be added with add()

In [6]:
s.add(e);

Add up to 9 additional arguments in one call

In [7]:
s.add(RooArgSet(c, d));

Sets can contain any type of RooAbsArg, also pdf and functions

In [8]:
s.add(g);

Remove element d

In [9]:
s.remove(d);

Accessing RooArgSet contents

You can look up objects by name

In [10]:
RooAbsArg *aptr = s.find("a");

Construct a subset by name

In [11]:
RooArgSet *subset1 = (RooArgSet *)s.selectByName("a,b,c");

Construct asubset by attribute

In [12]:
RooArgSet *subset2 = (RooArgSet *)s.selectByAttrib("Constant", true);

Construct the subset of overlapping contents with another set

In [13]:
RooArgSet s1(a, b, c);
RooArgSet s2(c, d, e);
RooArgSet *subset3 = (RooArgSet *)s1.selectCommon(s2);

Owning RooArgSets

Create a RooArgSet that owns its components A set either owns all of its components or none, so once addOwned() is used, add() can no longer be used and will result in an error message

In [14]:
RooRealVar *ac = (RooRealVar *)a.clone("a");
RooRealVar *bc = (RooRealVar *)b.clone("b");
RooRealVar *cc = (RooRealVar *)c.clone("c");

RooArgSet s3;
s3.addOwned(RooArgSet(*ac, *bc, *cc));

Another possibility is to add an owned clone of an object instead of the original

In [15]:
s3.addClone(RooArgSet(d, e, g));

A clone of a owning set is non-owning and its contents is owned by the originating owning set

In [16]:
RooArgSet *sclone = (RooArgSet *)s3.Clone("sclone");

To make a clone of a set and its contents use the snapshot method

In [17]:
RooArgSet *sclone2 = (RooArgSet *)s3.snapshot();

If a set contains function objects, only the head node is cloned in a snapshot. To make a snapshot of all servers of a function object do as follows. The result of a RooArgSet snapshot with deepCloning option is a set of cloned objects, and all their clone (recursive) server dependencies, that together form a self-consistent set that is free of external dependencies

In [18]:
RooArgSet *sclone3 = (RooArgSet *)s3.snapshot(true);

Set printing

Inline printing only show list of names of contained objects

In [19]:
cout << "sclone = " << (*sclone) << endl;
sclone = (a,b,c,d,e,g)

Plain print shows the same, prefixed by name of the set

In [20]:
sclone->Print();
RooArgSet::sclone = (a,b,c,d,e,g)

Standard printing shows one line for each item with the items name, class name and value

In [21]:
sclone->Print("s");
  1) RooRealVar:: a = 1
  2) RooRealVar:: b = 2
  3) RooRealVar:: c = 3 +/- 0.5
  4) RooRealVar:: d = 4
  5) RooCategory:: e = sig(idx = 0)

  6) RooGaussian:: g = 0.882497

Verbose printing adds each items arguments, address and 'extras' as defined by the object

In [22]:
sclone->Print("v");
  1) 0x7fa3651e2f30 RooRealVar:: a = 1 C  L(-10 - 10)  "a"
  2) 0x7fa3650ee0f0 RooRealVar:: b = 2 C  L(-10 - 10)  "b"
  3) 0x7fa3651e3880 RooRealVar:: c = 3 +/- 0.5  L(-10 - 10)  "c"
  4) 0x7fa3651dbc90 RooRealVar:: d = 4  L(-10 - 10)  "d"
  5) 0x7fa365032d90 RooCategory:: e = sig(idx = 0)
  "e"
  6) 0x7fa3651b6f00 RooGaussian:: g[ x=x mean=a sigma=b ] = 0.882497  "g"

Using RooArgLists

List constructors exists with up to 9 initial arguments

In [23]:
RooArgList l(a, b, c, d);

Lists have an explicit order and allow multiple arguments with the same name

In [24]:
l.add(RooArgList(a, b, c, d));

Access by index is provided

In [25]:
RooAbsArg *arg4 = l.at(4);