Use strings for colors in Python tutorials

This makes the code less verbose and more pythonic.
root-project · Nov 25, 2024 · 865e2a1 · 865e2a1
1 parent 069d935
commit 865e2a1
Show file tree

Hide file tree

Showing 48 changed files with 107 additions and 107 deletions.
diff --git a/tree/dataframe/src/RDataFrame.cxx b/tree/dataframe/src/RDataFrame.cxx
@@ -899,7 +899,7 @@ You can also include a global callback function that will be applied to all plot
 
 ~~~{.py}
 def set_fill_color(hist):
-    hist.SetFillColor(ROOT.kBlue)
+    hist.SetFillColor("kBlue")
 
 LiveVisualize([h_gaus, h_exp, h_random], set_fill_color)
 ~~~

diff --git a/tutorials/dataframe/df103_NanoAODHiggsAnalysis.py b/tutorials/dataframe/df103_NanoAODHiggsAnalysis.py
@@ -206,23 +206,23 @@ def plot(sig, bkg, data, x_label, filename):
     h_cmb.GetXaxis().SetTitleSize(0.04)
     h_cmb.GetYaxis().SetTitle("N_{Events}")
     h_cmb.GetYaxis().SetTitleSize(0.04)
-    h_cmb.SetLineColor(ROOT.kRed)
+    h_cmb.SetLineColor("kRed")
     h_cmb.SetLineWidth(2)
     h_cmb.SetMaximum(18)
     h_cmb.SetStats(False)
 
     h_bkg.SetLineWidth(2)
     h_bkg.SetFillStyle(1001)
-    h_bkg.SetLineColor(ROOT.kBlack)
-    h_bkg.SetFillColor(ROOT.kAzure - 9)
+    h_bkg.SetLineColor("black")
+    h_bkg.SetFillColor("kAzure-9")
 
     # Get histogram of data points
     h_data = data.Clone()
     h_data.SetLineWidth(1)
     h_data.SetMarkerStyle(20)
     h_data.SetMarkerSize(1.0)
-    h_data.SetMarkerColor(ROOT.kBlack)
-    h_data.SetLineColor(ROOT.kBlack)
+    h_data.SetMarkerColor("black")
+    h_data.SetLineColor("black")
 
     # Draw histograms
     h_cmb.Draw("HIST")

diff --git a/tutorials/dataframe/df104_HiggsToTwoPhotons.py b/tutorials/dataframe/df104_HiggsToTwoPhotons.py
@@ -127,7 +127,7 @@
 data.SetMarkerStyle(20)
 data.SetMarkerSize(1.2)
 data.SetLineWidth(2)
-data.SetLineColor(ROOT.kBlack)
+data.SetLineColor("black")
 data.SetMinimum(1e-3)
 data.SetMaximum(8e3)
 data.GetYaxis().SetLabelSize(0.045)

diff --git a/tutorials/dataframe/df105_WBosonAnalysis.py b/tutorials/dataframe/df105_WBosonAnalysis.py
@@ -190,7 +190,7 @@ def merge_histos(label):
 data.SetMarkerStyle(20)
 data.SetMarkerSize(1.2)
 data.SetLineWidth(2)
-data.SetLineColor(ROOT.kBlack)
+data.SetLineColor("black")
 data.Draw("E SAME")
 
 # Add legend

diff --git a/tutorials/dataframe/df106_HiggsToFourLeptons.py b/tutorials/dataframe/df106_HiggsToFourLeptons.py
@@ -232,7 +232,7 @@ class VaryHelper
 h_zz = histos[2].GetValue().Clone()
 h_other = histos[3].GetValue().Clone()
 
-for h, color in zip([h_other, h_zz, h_higgs], [ROOT.kViolet - 9, ROOT.kAzure - 9, ROOT.kRed + 2]):
+for h, color in zip([h_other, h_zz, h_higgs], ["kViolet-9", "kAzure-9", "kRed+2"]):
     h.SetLineWidth(1)
     h.SetLineColor(1)
     h.SetFillColor(color)
@@ -250,19 +250,19 @@ class VaryHelper
 stack.GetYaxis().ChangeLabel(1, -1, 0)
 
 # Draw MC scale factor and variations
-histos_mc["nominal"].SetFillColor(ROOT.kBlack)
+histos_mc["nominal"].SetFillColor("black")
 histos_mc["nominal"].SetFillStyle(3254)
 h_nominal = histos_mc["nominal"].DrawClone("E2 same")
-histos_mc["weight:up"].SetLineColor(ROOT.kGreen + 2)
+histos_mc["weight:up"].SetLineColor("kGreen+2")
 h_weight_up = histos_mc["weight:up"].DrawClone("HIST SAME")
-histos_mc["weight:down"].SetLineColor(ROOT.kBlue + 2)
+histos_mc["weight:down"].SetLineColor("kBlue+2")
 h_weight_down = histos_mc["weight:down"].DrawClone("HIST SAME")
 
 # Draw data histogram
 h_data.SetMarkerStyle(20)
 h_data.SetMarkerSize(1.2)
 h_data.SetLineWidth(2)
-h_data.SetLineColor(ROOT.kBlack)
+h_data.SetLineColor("black")
 h_data.Draw("E SAME")  # Draw raw data with errorbars
 
 # Add legend

diff --git a/tutorials/dataframe/df107_SingleTopAnalysis.py b/tutorials/dataframe/df107_SingleTopAnalysis.py
@@ -229,7 +229,7 @@ def merge_histos(label):
 data.SetMarkerStyle(20)
 data.SetMarkerSize(1.2)
 data.SetLineWidth(2)
-data.SetLineColor(ROOT.kBlack)
+data.SetLineColor("black")
 data.Draw("E SAME")
 
 # Add legend

diff --git a/tutorials/fit/combinedFit.py b/tutorials/fit/combinedFit.py
@@ -121,14 +121,14 @@ def __call__(self, par):
 
 fB.SetFitResult(result, iparB)
 fB.SetRange(rangeB().first, rangeB().second)
-fB.SetLineColor(ROOT.kBlue)
+fB.SetLineColor("kBlue")
 hB.GetListOfFunctions().Add(fB)
 hB.Draw()
 
 c1.cd(2)
 fSB.SetFitResult(result, iparSB)
 fSB.SetRange(rangeSB().first, rangeSB().second)
-fSB.SetLineColor(ROOT.kRed)
+fSB.SetLineColor("kRed")
 hSB.GetListOfFunctions().Add(fSB)
 hSB.Draw()
 

diff --git a/tutorials/graphs/timeSeriesFromCSV.py b/tutorials/graphs/timeSeriesFromCSV.py
@@ -39,7 +39,7 @@
 c.SetRightMargin(0.04)
 c.SetGrid()
 g.SetLineWidth(3)
-g.SetLineColor(ROOT.kBlue)
+g.SetLineColor("kBlue")
 g.Draw("al")
 g.GetYaxis().CenterTitle()
 

diff --git a/tutorials/hist/ratioplot5.py b/tutorials/hist/ratioplot5.py
@@ -22,6 +22,6 @@
 h1.GetYaxis().SetTitle("y")
 
 rp1 = ROOT.TRatioPlot(h1)
-rp1.SetConfidenceIntervalColors(ROOT.kBlue, ROOT.kRed)
+rp1.SetConfidenceIntervalColors("kBlue", "kRed")
 rp1.Draw()
 c1.Update()
diff --git a/tutorials/hist/twoscales.py b/tutorials/hist/twoscales.py
@@ -39,12 +39,12 @@
 
 rightmax = 1.1*hint1.GetMaximum()
 scale = ROOT.gPad.GetUymax()/rightmax
-hint1.SetLineColor(ROOT.kRed)
+hint1.SetLineColor("kRed")
 hint1.Scale(scale)
 hint1.Draw("same")
 
 axis = ROOT.TGaxis(ROOT.gPad.GetUxmax(),ROOT.gPad.GetUymin(),
       ROOT.gPad.GetUxmax(), ROOT.gPad.GetUymax(),0,rightmax,510,"+L")
-axis.SetLineColor(ROOT.kRed)
-axis.SetLabelColor(ROOT.kRed)
+axis.SetLineColor("kRed")
+axis.SetLabelColor("kRed")
 axis.Draw()
diff --git a/tutorials/histfactory/hf001_example.py b/tutorials/histfactory/hf001_example.py
@@ -108,11 +108,11 @@
 xmax = poi.getMax()
 
 line = ROOT.TLine(xmin, 0.5, xmax, 0.5)
-line.SetLineColor(ROOT.kGreen)
+line.SetLineColor("kGreen")
 line90 = ROOT.TLine(xmin, 2.71 / 2, xmax, 2.71 / 2)
-line90.SetLineColor(ROOT.kGreen)
+line90.SetLineColor("kGreen")
 line95 = ROOT.TLine(xmin, 3.84 / 2, xmax, 3.84 / 2)
-line95.SetLineColor(ROOT.kGreen)
+line95.SetLineColor("kGreen")
 
 frame.addObject(line)
 frame.addObject(line90)

diff --git a/tutorials/pyroot/shapes.py b/tutorials/pyroot/shapes.py
@@ -61,9 +61,9 @@
 cons.SetLineColor( 3 )
 pcon.SetLineColor( 6 )
 pgon.SetLineColor( 2 )
-sphe.SetLineColor( ROOT.kRed )
-sphe1.SetLineColor( ROOT.kBlack )
-sphe2.SetLineColor( ROOT.kBlue )
+sphe.SetLineColor("kRed")
+sphe1.SetLineColor("black")
+sphe2.SetLineColor("kBlue")
 
 #  Build the geometry hierarchy
 node1 = ROOT.TNode( 'NODE1', 'NODE1', 'BRIK' )

diff --git a/tutorials/pyroot/zdemo.py b/tutorials/pyroot/zdemo.py
@@ -175,9 +175,9 @@ def zdemo():
     t.DrawLatex( 0.18, 0.40, '(barn/Gev^{2})' )
 
     t.SetTextSize( 0.045 )
-    t.SetTextColor( ROOT.kBlue )
+    t.SetTextColor("kBlue")
     t.DrawLatex( 0.22, 0.260, '#sqrt{s} = 63(GeV)' )
-    t.SetTextColor( ROOT.kRed )
+    t.SetTextColor("kRed")
     t.DrawLatex( 0.22, 0.205,'#sqrt{s} = 200(GeV)' )
     t.SetTextColor( 6 )
     t.DrawLatex( 0.22, 0.15, '#sqrt{s} = 500(GeV)' )
@@ -190,7 +190,7 @@ def zdemo():
     gr1 = ROOT.TGraph( NLOOP, PT, INVSIG )
 
     gr1.SetLineColor( 38 )
-    gr1.SetMarkerColor( ROOT.kBlue )
+    gr1.SetMarkerColor("kBlue")
     gr1.SetMarkerStyle( 21 )
     gr1.SetMarkerSize( 1.1 )
     gr1.Draw( 'LP' )
@@ -210,7 +210,7 @@ def zdemo():
 
     gr2 = ROOT.TGraph( NLOOP, PT, INVSIG )
     gr2.SetLineColor( 38 )
-    gr2.SetMarkerColor( ROOT.kRed )
+    gr2.SetMarkerColor("kRed")
     gr2.SetMarkerStyle( 29 )
     gr2.SetMarkerSize( 1.5 )
     gr2.Draw( 'LP' )
@@ -238,15 +238,15 @@ def zdemo():
 
     dum = array( 'f', [0.] )
     graph = ROOT.TGraph( 1, dum, dum )
-    graph.SetMarkerColor( ROOT.kBlue )
+    graph.SetMarkerColor("kBlue")
     graph.SetMarkerStyle( 21 )
     graph.SetMarkerSize( 1.1 )
     graph.SetPoint( 0, 1.7, 1.e-16 )
     graph.Draw( 'LP' )
     saves[ 'graph' ] = graph
 
     graph = ROOT.TGraph( 1, dum, dum )
-    graph.SetMarkerColor( ROOT.kRed )
+    graph.SetMarkerColor("kRed")
     graph.SetMarkerStyle( 29 )
     graph.SetMarkerSize( 1.5 )
     graph.SetPoint( 0, 1.7, 2.e-17 )

diff --git a/tutorials/rcanvas/df104.py b/tutorials/rcanvas/df104.py
@@ -130,7 +130,7 @@
 data.SetMarkerStyle(20)
 data.SetMarkerSize(1.2)
 data.SetLineWidth(2)
-data.SetLineColor(ROOT.kBlack)
+data.SetLineColor("black")
 data.SetMinimum(1e-3)
 data.SetMaximum(8e3)
 data.GetYaxis().SetLabelSize(0.045)

diff --git a/tutorials/rcanvas/df105.py b/tutorials/rcanvas/df105.py
@@ -202,7 +202,7 @@ def merge_histos(label):
 data.SetMarkerStyle(20)
 data.SetMarkerSize(1.2)
 data.SetLineWidth(2)
-data.SetLineColor(ROOT.kBlack)
+data.SetLineColor("black")
 c.Add[TObjectDrawable]().Set(data, "E SAME")
 
 # Add TLegend while histograms packed in the THStack

diff --git a/tutorials/roofit/rf106_plotdecoration.py b/tutorials/roofit/rf106_plotdecoration.py
@@ -55,13 +55,13 @@
 # Add text to frame
 txt = ROOT.TText(2, 100, "Signal")
 txt.SetTextSize(0.04)
-txt.SetTextColor(ROOT.kRed)
+txt.SetTextColor("kRed")
 frame.addObject(txt)
 
 # Add arrow to frame
 arrow = ROOT.TArrow(2, 100, -1, 50, 0.01, "|>")
-arrow.SetLineColor(ROOT.kRed)
-arrow.SetFillColor(ROOT.kRed)
+arrow.SetLineColor("kRed")
+arrow.SetFillColor("kRed")
 arrow.SetLineWidth(3)
 frame.addObject(arrow)
 

diff --git a/tutorials/roofit/rf207_comptools.py b/tutorials/roofit/rf207_comptools.py
@@ -109,7 +109,7 @@
 # The returned head node own all nodes that were cloned as part of
 # the build process so when cust_clone is deleted so will all other
 # nodes that were created in the process.
-cust_clone = cust.build(ROOT.kTRUE)
+cust_clone = cust.build(True)
 ROOT.SetOwnership(cust_clone, True)
 
 # Print structure of clone of model with sig.sigsum replacement.

diff --git a/tutorials/roofit/rf211_paramconv.py b/tutorials/roofit/rf211_paramconv.py
@@ -63,7 +63,7 @@
     ConditionalObservables={mean},
 )
 hh.SetTitle("histogram of model(x|mean)")
-hh.SetLineColor(ROOT.kBlue)
+hh.SetLineColor("kBlue")
 
 # Draw frame on canvas
 c = ROOT.TCanvas("rf211_paramconv", "rf211_paramconv", 800, 400)

diff --git a/tutorials/roofit/rf301_composition.py b/tutorials/roofit/rf301_composition.py
@@ -51,7 +51,7 @@
 
 # Make two-dimensional plot in x vs y
 hh_model = model.createHistogram("hh_model", x, Binning=50, YVar=dict(var=y, Binning=50))
-hh_model.SetLineColor(ROOT.kBlue)
+hh_model.SetLineColor("kBlue")
 
 # Make canvas and draw ROOT.RooPlots
 c = ROOT.TCanvas("rf301_composition", "rf301_composition", 1200, 400)

diff --git a/tutorials/roofit/rf302_utilfuncs.py b/tutorials/roofit/rf302_utilfuncs.py
@@ -69,10 +69,10 @@
 hh_model_2 = model_2.createHistogram("hh_model_2", x, Binning=50, YVar=dict(var=y, Binning=50))
 hh_model_3 = model_3.createHistogram("hh_model_3", x, Binning=50, YVar=dict(var=y, Binning=50))
 hh_model_4 = model_4.createHistogram("hh_model_4", x, Binning=50, YVar=dict(var=y, Binning=50))
-hh_model_1.SetLineColor(ROOT.kBlue)
-hh_model_2.SetLineColor(ROOT.kBlue)
-hh_model_3.SetLineColor(ROOT.kBlue)
-hh_model_4.SetLineColor(ROOT.kBlue)
+hh_model_1.SetLineColor("kBlue")
+hh_model_2.SetLineColor("kBlue")
+hh_model_3.SetLineColor("kBlue")
+hh_model_4.SetLineColor("kBlue")
 
 # Make canvas and draw ROOT.RooPlots
 c = ROOT.TCanvas("rf302_utilfuncs", "rf302_utilfuncs", 800, 800)

diff --git a/tutorials/roofit/rf305_condcorrprod.py b/tutorials/roofit/rf305_condcorrprod.py
@@ -62,7 +62,7 @@
 
 # Make two-dimensional plot in x vs y
 hh_model = model.createHistogram("hh_model", x, ROOT.RooFit.Binning(50), ROOT.RooFit.YVar(y, ROOT.RooFit.Binning(50)))
-hh_model.SetLineColor(ROOT.kBlue)
+hh_model.SetLineColor("kBlue")
 
 # Make canvas and draw ROOT.RooPlots
 c = ROOT.TCanvas("rf305_condcorrprod", "rf05_condcorrprod", 1200, 400)

diff --git a/tutorials/roofit/rf306_condpereventerrors.py b/tutorials/roofit/rf306_condpereventerrors.py
@@ -54,7 +54,7 @@
 
 # Make two-dimensional plot of conditional pdf in (dt,dterr)
 hh_decay = decay_gm.createHistogram("hh_decay", dt, Binning=50, YVar=dict(var=dterr, Binning=50))
-hh_decay.SetLineColor(ROOT.kBlue)
+hh_decay.SetLineColor("kBlue")
 
 # Plot decay_gm(dt|dterr) at various values of dterr
 frame = dt.frame(Title="Slices of decay(dt|dterr) at various dterr")

diff --git a/tutorials/roofit/rf307_fullpereventerrors.py b/tutorials/roofit/rf307_fullpereventerrors.py
@@ -67,7 +67,7 @@
 
 # Make two-dimensional plot of conditional pdf in (dt,dterr)
 hh_model = model.createHistogram("hh_model", dt, Binning=50, YVar=dict(var=dterr, Binning=50))
-hh_model.SetLineColor(ROOT.kBlue)
+hh_model.SetLineColor("kBlue")
 
 # Make projection of data an dt
 frame = dt.frame(Title="Projection of model(dt|dterr) on dt")

diff --git a/tutorials/roofit/rf308_normintegration2d.py b/tutorials/roofit/rf308_normintegration2d.py
@@ -79,7 +79,7 @@
 
 # Plot cdf of gx versus x
 hh_cdf = gxy_cdf.createHistogram("hh_cdf", x, Binning=40, YVar=dict(var=y, Binning=40))
-hh_cdf.SetLineColor(ROOT.kBlue)
+hh_cdf.SetLineColor("kBlue")
 
 c = ROOT.TCanvas("rf308_normintegration2d", "rf308_normintegration2d", 600, 600)
 ROOT.gPad.SetLeftMargin(0.15)

diff --git a/tutorials/roofit/rf309_ndimplot.py b/tutorials/roofit/rf309_ndimplot.py
@@ -42,7 +42,7 @@
 # Create and fill ROOT 2D histogram (50x50 bins) with sampling of pdf
 # hh_pdf = model.createHistogram("hh_model", x, Binning=50, YVar=dict(var=y, Binning=50))
 hh_pdf = model.createHistogram("x,y", 50, 50)
-hh_pdf.SetLineColor(ROOT.kBlue)
+hh_pdf.SetLineColor("kBlue")
 
 # Create 3D model and dataset
 # -----------------------------------------------------
@@ -76,7 +76,7 @@
     YVar=dict(var=y, Binning=20),
     ZVar=dict(var=z, Binning=20),
 )
-hh_pdf3.SetFillColor(ROOT.kBlue)
+hh_pdf3.SetFillColor("kBlue")
 
 c1 = ROOT.TCanvas("rf309_2dimplot", "rf309_2dimplot", 800, 800)
 c1.Divide(2, 2)

diff --git a/tutorials/roofit/rf315_projectpdf.py b/tutorials/roofit/rf315_projectpdf.py
@@ -67,7 +67,7 @@
 
 # Make 2D histogram of model(x,y)
 hh = model.createHistogram("x,y")
-hh.SetLineColor(ROOT.kBlue)
+hh.SetLineColor("kBlue")
 
 c = ROOT.TCanvas("rf315_projectpdf", "rf315_projectpdf", 800, 400)
 c.Divide(2)

diff --git a/tutorials/roofit/rf508_listsetmanip.py b/tutorials/roofit/rf508_listsetmanip.py
@@ -69,7 +69,7 @@
 subset1 = s.selectByName("a,b,c")
 
 # Construct asubset by attribute
-subset2 = s.selectByAttrib("Constant", ROOT.kTRUE)
+subset2 = s.selectByAttrib("Constant", True)
 
 # Construct the subset of overlapping contents with another set
 s1 = ROOT.RooArgSet(a, b, c)

diff --git a/tutorials/roofit/rf608_fitresultaspdf.py b/tutorials/roofit/rf608_fitresultaspdf.py
@@ -53,7 +53,7 @@
 # Sample a 3-D histogram of the pdf to be visualized as an error
 # ellipsoid using the GLISO draw option
 hh_3d = parabPdf.createHistogram("mean,sigma_g2,frac", 25, 25, 25)
-hh_3d.SetFillColor(ROOT.kBlue)
+hh_3d.SetFillColor("kBlue")
 
 # Project 3D parameter pdf down to 3 permutations of two-dimensional pdfs
 # The integrations corresponding to these projections are performed analytically
@@ -66,9 +66,9 @@
 hh_sigmag2_frac = pdf_sigmag2_frac.createHistogram("sigma_g2,frac", 50, 50)
 hh_mean_frac = pdf_mean_frac.createHistogram("mean,frac", 50, 50)
 hh_mean_sigmag2 = pdf_mean_sigmag2.createHistogram("mean,sigma_g2", 50, 50)
-hh_mean_frac.SetLineColor(ROOT.kBlue)
-hh_sigmag2_frac.SetLineColor(ROOT.kBlue)
-hh_mean_sigmag2.SetLineColor(ROOT.kBlue)
+hh_mean_frac.SetLineColor("kBlue")
+hh_sigmag2_frac.SetLineColor("kBlue")
+hh_mean_sigmag2.SetLineColor("kBlue")
 
 # Draw the 'sigar'
 ROOT.gStyle.SetCanvasPreferGL(True)