Logo ROOT  
Reference Guide
 
Loading...
Searching...
No Matches
rf711_lagrangianmorph.C File Reference

Detailed Description

View in nbviewer Open in SWAN
Morphing effective field theory distributions with RooLagrangianMorphFunc A morphing function as a function of one coefficient is setup and can be used to obtain the distribution for any value of the coefficient.

#include <RooDataHist.h>
#include <RooDataSet.h>
#include <RooPlot.h>
#include <RooRealVar.h>
#include <TCanvas.h>
#include <TColor.h>
#include <TFile.h>
#include <TFolder.h>
#include <TH1.h>
#include <TLegend.h>
#include <TStyle.h>
using namespace RooFit;
{
// C r e a t e v a r i a b l e s f o r
// m o r p h i n g f u n c t i o n
// ---------------------------------------------
std::string observablename = "pTV";
// Setup observable that is morphed
RooRealVar obsvar(observablename.c_str(), "p_{T}^{V}", 10, 600);
// Setup two couplings that enters the morphing function
// kSM -> SM coupling set to constant (1)
// cHq3 -> EFT parameter with NewPhysics attribute set to true
RooRealVar kSM("kSM", "sm modifier", 1.0);
RooRealVar cHq3("cHq3", "EFT modifier", 0.0, 1.0);
cHq3.setAttribute("NewPhysics", true);
// I n p u t s n e e d e d f o r c o n f i g
// ---------------------------------------------
std::string infilename = std::string(gROOT->GetTutorialDir()) + "/roofit/input_histos_rf_lagrangianmorph.root";
std::vector<std::string> samplelist = {"SM_NPsq0", "cHq3_NPsq1", "cHq3_NPsq2"};
// S e t u p C o n f i g
// ---------------------------------------------
config.fileName = infilename;
config.observableName = observablename;
config.folderNames = samplelist;
config.couplings.add(cHq3);
config.couplings.add(kSM);
// C r e a t e m o r p h i n g f u n c t i o n
// ---------------------------------------------
RooLagrangianMorphFunc morphfunc("morphfunc", "morphed dist. of pTV", config);
// G e t m o r p h e d d i s t r i b u t i o n f o r
// d i f f e r e n t c H q 3
// ---------------------------------------------
morphfunc.setParameter("cHq3", 0.01);
auto morph_hist_0p01 = morphfunc.createTH1("morph_cHq3=0.01");
morphfunc.setParameter("cHq3", 0.25);
auto morph_hist_0p25 = morphfunc.createTH1("morph_cHq3=0.25");
morphfunc.setParameter("cHq3", 0.5);
auto morph_hist_0p5 = morphfunc.createTH1("morph_cHq3=0.5");
RooDataHist morph_datahist_0p01("morph_dh_cHq3=0.01", "", {obsvar}, morph_hist_0p01);
RooDataHist morph_datahist_0p25("morph_dh_cHq3=0.25", "", {obsvar}, morph_hist_0p25);
RooDataHist morph_datahist_0p5("morph_dh_cHq3=0.5", "", {obsvar}, morph_hist_0p5);
// E x t r a c t i n p u t t e m p l a t e s
// f o r p l o t t i n g
// ---------------------------------------------
TFile *file = new TFile(infilename.c_str(), "READ");
TFolder *folder = 0;
file->GetObject(samplelist[0].c_str(), folder);
TH1 *input_hist0 = dynamic_cast<TH1 *>(folder->FindObject(observablename.c_str()));
input_hist0->SetDirectory(NULL);
file->GetObject(samplelist[1].c_str(), folder);
TH1 *input_hist1 = dynamic_cast<TH1 *>(folder->FindObject(observablename.c_str()));
input_hist1->SetDirectory(NULL);
file->GetObject(samplelist[2].c_str(), folder);
TH1 *input_hist2 = dynamic_cast<TH1 *>(folder->FindObject(observablename.c_str()));
input_hist2->SetDirectory(NULL);
file->Close();
RooDataHist input_dh0(samplelist[0], "", {obsvar}, input_hist0);
RooDataHist input_dh1(samplelist[1], "", {obsvar}, input_hist1);
RooDataHist input_dh2(samplelist[2], "", {obsvar}, input_hist2);
auto frame0 = obsvar.frame(Title("Input templates for p_{T}^{V}"));
input_dh0.plotOn(frame0, Name(samplelist[0].c_str()), LineColor(kBlack), MarkerColor(kBlack), MarkerSize(1));
input_dh1.plotOn(frame0, Name(samplelist[1].c_str()), LineColor(kRed), MarkerColor(kRed), MarkerSize(1));
input_dh2.plotOn(frame0, Name(samplelist[2].c_str()), LineColor(kBlue), MarkerColor(kBlue), MarkerSize(1));
// P l o t m o r p h e d d i s t r i b u t i o n f o r
// d i f f e r e n t c H q 3
// ---------------------------------------------
auto frame1 = obsvar.frame(Title("Morphed templates for selected values"));
morph_datahist_0p01.plotOn(frame1, Name("morph_dh_cHq3=0.01"), DrawOption("C"), LineColor(kGreen),
morph_datahist_0p25.plotOn(frame1, Name("morph_dh_cHq3=0.25"), DrawOption("C"), LineColor(kGreen + 1),
morph_datahist_0p5.plotOn(frame1, Name("morph_dh_cHq3=0.5"), DrawOption("C"), LineColor(kGreen + 2),
// C r e a t e w r a p p e d p d f t o g e n e r a t e
// 2D d a t a s e t o f c H q 3 a s a f u n c t i o n o f
// o b s e r v a b l e
// ---------------------------------------------
RooWrapperPdf model("wrap_pdf", "wrap_pdf", morphfunc);
std::unique_ptr<RooDataSet> data{model.generate({cHq3, obsvar}, 1000000)};
TH1 *hh_data = data->createHistogram("pTV vs cHq3", obsvar, Binning(20), YVar(cHq3, Binning(50)));
hh_data->SetTitle("Morphing prediction");
// D r a w p l o t s o n c a n v a s
// ---------------------------------------------
TCanvas *c1 = new TCanvas("fig3", "fig3", 1200, 400);
c1->Divide(3, 1);
c1->cd(1);
gPad->SetLeftMargin(0.15);
gPad->SetRightMargin(0.05);
frame0->Draw();
TLegend *leg1 = new TLegend(0.55, 0.65, 0.94, 0.87);
leg1->SetTextSize(0.04);
leg1->AddEntry("SM_NPsq0", "SM", "LP");
leg1->AddEntry((TObject *)0, "", "");
leg1->AddEntry("cHq3_NPsq1", "c_{Hq^(3)}=1.0 at #Lambda^{-2}", "LP");
leg1->AddEntry((TObject *)0, "", "");
leg1->AddEntry("cHq3_NPsq2", "c_{Hq^(3)}=1.0 at #Lambda^{-4}", "LP");
leg1->Draw();
c1->cd(2);
gPad->SetLeftMargin(0.15);
gPad->SetRightMargin(0.05);
frame1->Draw();
TLegend *leg2 = new TLegend(0.60, 0.65, 0.94, 0.87);
leg2->SetTextSize(0.04);
leg2->AddEntry("morph_dh_cHq3=0.01", "c_{Hq^{(3)}}=0.01", "L");
leg2->AddEntry((TObject *)0, "", "");
leg2->AddEntry("morph_dh_cHq3=0.25", "c_{Hq^{(3)}}=0.25", "L");
leg2->AddEntry((TObject *)0, "", "");
leg2->AddEntry("morph_dh_cHq3=0.5", "c_{Hq^{(3)}}=0.5", "L");
leg2->AddEntry((TObject *)0, "", "");
leg2->Draw();
c1->cd(3);
gPad->SetLeftMargin(0.12);
gPad->SetRightMargin(0.18);
gPad->SetLogz();
hh_data->GetYaxis()->SetTitle("c_{Hq^{(3)}}");
hh_data->GetYaxis()->SetRangeUser(0, 0.5);
hh_data->GetZaxis()->SetTitleOffset(1.8);
hh_data->Draw("COLZ");
c1->SaveAs("rf711_lagrangianmorph.png");
}
@ kRed
Definition Rtypes.h:66
@ kBlack
Definition Rtypes.h:65
@ kGreen
Definition Rtypes.h:66
@ kWhite
Definition Rtypes.h:65
@ kBlue
Definition Rtypes.h:66
@ kGreyScale
Definition TColor.h:112
Option_t Option_t TPoint TPoint const char GetTextMagnitude GetFillStyle GetLineColor GetLineWidth GetMarkerStyle GetTextAlign GetTextColor GetTextSize void data
#define gROOT
Definition TROOT.h:406
R__EXTERN TStyle * gStyle
Definition TStyle.h:433
#define gPad
virtual bool add(const RooAbsArg &var, bool silent=false)
Add the specified argument to list.
Container class to hold N-dimensional binned data.
Definition RooDataHist.h:39
RooPlot * plotOn(RooPlot *frame, PlotOpt o) const override
Back end function to plotting functionality.
Class RooLagrangianMorphing is a implementation of the method of Effective Lagrangian Morphing,...
static RooPlot * frame(const RooAbsRealLValue &var, double xmin, double xmax, Int_t nBins)
Create a new frame for a given variable in x.
Definition RooPlot.cxx:237
Variable that can be changed from the outside.
Definition RooRealVar.h:37
The RooWrapperPdf is a class that can be used to convert a function into a PDF.
virtual void SetTitleOffset(Float_t offset=1)
Set distance between the axis and the axis title.
Definition TAttAxis.cxx:298
virtual void SetFillColor(Color_t fcolor)
Set the fill area color.
Definition TAttFill.h:37
virtual void SetLineColor(Color_t lcolor)
Set the line color.
Definition TAttLine.h:40
virtual void SetTextSize(Float_t tsize=1)
Set the text size.
Definition TAttText.h:47
virtual void SetRangeUser(Double_t ufirst, Double_t ulast)
Set the viewing range for the axis from ufirst to ulast (in user coordinates, that is,...
Definition TAxis.cxx:1080
The Canvas class.
Definition TCanvas.h:23
static void InvertPalette()
Invert the current color palette.
Definition TColor.cxx:3459
A ROOT file is composed of a header, followed by consecutive data records (TKey instances) with a wel...
Definition TFile.h:53
<div class="legacybox"><h2>Legacy Code</h2> TFolder is a legacy interface: there will be no bug fixes...
Definition TFolder.h:30
TObject * FindObject(const char *name) const override
Search object identified by name in the tree of folders inside this folder.
Definition TFolder.cxx:306
TH1 is the base class of all histogram classes in ROOT.
Definition TH1.h:59
virtual void SetDirectory(TDirectory *dir)
By default, when a histogram is created, it is added to the list of histogram objects in the current ...
Definition TH1.cxx:8901
TAxis * GetZaxis()
Definition TH1.h:326
void SetTitle(const char *title) override
Change/set the title.
Definition TH1.cxx:6682
TAxis * GetYaxis()
Definition TH1.h:325
void Draw(Option_t *option="") override
Draw this histogram with options.
Definition TH1.cxx:3062
This class displays a legend box (TPaveText) containing several legend entries.
Definition TLegend.h:23
TLegendEntry * AddEntry(const TObject *obj, const char *label="", Option_t *option="lpf")
Add a new entry to this legend.
Definition TLegend.cxx:317
void Draw(Option_t *option="") override
Draw this legend with its current attributes.
Definition TLegend.cxx:422
virtual void SetTitle(const char *title="")
Set the title of the TNamed.
Definition TNamed.cxx:164
Mother of all ROOT objects.
Definition TObject.h:41
void SetOptStat(Int_t stat=1)
The type of information printed in the histogram statistics box can be selected via the parameter mod...
Definition TStyle.cxx:1636
void SetPalette(Int_t ncolors=kBird, Int_t *colors=nullptr, Float_t alpha=1.)
See TColor::SetPalette.
Definition TStyle.cxx:1884
void SetNumberContours(Int_t number=20)
Set the default number of contour levels when drawing 2-d plots.
Definition TStyle.cxx:1496
RooCmdArg YVar(const RooAbsRealLValue &var, const RooCmdArg &arg={})
RooCmdArg DataError(Int_t)
RooCmdArg MarkerColor(Color_t color)
RooCmdArg Binning(const RooAbsBinning &binning)
RooCmdArg DrawOption(const char *opt)
RooCmdArg XErrorSize(double width)
RooCmdArg MarkerSize(Size_t size)
RooCmdArg LineColor(Color_t color)
return c1
Definition legend1.C:41
The namespace RooFit contains mostly switches that change the behaviour of functions of PDFs (or othe...
Definition JSONIO.h:26
const char * Name
Definition TXMLSetup.cxx:67
const char * Title
Definition TXMLSetup.cxx:68
std::vector< std::string > folderNames
[#0] PROGRESS:InputArguments -- initializing physics inputs from file /home/sftnight/build/workspace/root-makedoc-master/rootspi/rdoc/src/master.build/tutorials/roofit/input_histos_rf_lagrangianmorph.root with object name(s) 'pTV'
[#0] PROGRESS:Caching -- creating cache from getCache function for 0x7ffeeaf483c8
[#0] PROGRESS:Caching -- current storage has size 3
[#0] PROGRESS:ObjectHandling -- observable: pTV
[#0] PROGRESS:ObjectHandling -- binWidth: binWidth_pTV
[#1] INFO:DataHandling -- RooDataHist::adjustBinning(morph_dh_cHq3=0.01): fit range of variable pTV expanded to nearest bin boundaries: [10,600] --> [0,600]
[#1] INFO:InputArguments -- RooAbsData::plotOn(SM_NPsq0) INFO: dataset has non-integer weights, auto-selecting SumW2 errors instead of Poisson errors
[#1] INFO:InputArguments -- RooAbsData::plotOn(cHq3_NPsq1) INFO: dataset has non-integer weights, auto-selecting SumW2 errors instead of Poisson errors
[#1] INFO:Plotting -- RooPlot::updateFitRangeNorm: New event count of 24931.9 will supersede previous event count of 10852.3 for normalization of PDF projections
[#1] INFO:InputArguments -- RooAbsData::plotOn(cHq3_NPsq2) INFO: dataset has non-integer weights, auto-selecting SumW2 errors instead of Poisson errors
[#1] INFO:Plotting -- RooPlot::updateFitRangeNorm: New event count of 29789.2 will supersede previous event count of 24931.9 for normalization of PDF projections
[#1] INFO:Plotting -- RooPlot::updateFitRangeNorm: New event count of 18947.1 will supersede previous event count of 11104.6 for normalization of PDF projections
[#1] INFO:Plotting -- RooPlot::updateFitRangeNorm: New event count of 30765.5 will supersede previous event count of 18947.1 for normalization of PDF projections
[#0] PROGRESS:Caching -- creating cache from getCache function for 0x560cdf0a1a90
[#0] PROGRESS:Caching -- current storage has size 3
[#0] PROGRESS:ObjectHandling -- observable: pTV
[#0] PROGRESS:ObjectHandling -- binWidth: binWidth_pTV
[#1] INFO:NumericIntegration -- RooRealIntegral::init(w_SM_NPsq0_morphfunc_Int[cHq3]) using numeric integrator RooIntegrator1D to calculate Int(cHq3)
[#1] INFO:NumericIntegration -- RooRealIntegral::init(w_cHq3_NPsq1_morphfunc_Int[cHq3]) using numeric integrator RooIntegrator1D to calculate Int(cHq3)
[#1] INFO:NumericIntegration -- RooRealIntegral::init(w_cHq3_NPsq2_morphfunc_Int[cHq3]) using numeric integrator RooIntegrator1D to calculate Int(cHq3)
[#0] PROGRESS:Caching -- creating cache from getCache function for 0x560ce042b4c0
[#0] PROGRESS:Caching -- current storage has size 3
[#0] PROGRESS:ObjectHandling -- observable: pTV
[#0] PROGRESS:ObjectHandling -- binWidth: binWidth_pTV
[#1] INFO:NumericIntegration -- RooRealIntegral::init(w_SM_NPsq0_morphfunc_Int[cHq3]) using numeric integrator RooIntegrator1D to calculate Int(cHq3)
[#1] INFO:NumericIntegration -- RooRealIntegral::init(w_cHq3_NPsq1_morphfunc_Int[cHq3]) using numeric integrator RooIntegrator1D to calculate Int(cHq3)
[#1] INFO:NumericIntegration -- RooRealIntegral::init(w_cHq3_NPsq2_morphfunc_Int[cHq3]) using numeric integrator RooIntegrator1D to calculate Int(cHq3)
Date
January 2022
Author
Rahul Balasubramanian

Definition in file rf711_lagrangianmorph.C.