Update Impacts.py

python/tool_base/FastScan.py

@@ -42,13 +42,6 @@ def attach_args(self, group):
         group.add_argument("-o", "--output", default="nll", help="Name of the output file, without the .pdf extension")
         group.add_argument("-p", "--points", default=200, type=int, help="Number of NLL points to sample in each scan")
 
-    def RooColIter(self, coll):
-        it = coll.createIterator()
-        var = it.Next()
-        while var:
-            yield var
-            var = it.Next()
-
     def run_method(self):
         ROOT.gROOT.SetBatch(ROOT.kTRUE)
         outfile = ROOT.TFile("%s.root" % self.args.output, "RECREATE")
@@ -83,7 +76,7 @@ def run_method(self):
         page = 0
         doPars = []
 
-        for par in self.RooColIter(pars):
+        for par in pars:
             if par.isConstant():
                 continue
             if self.args.match is not None:

python/tool_base/Impacts.py

@@ -204,6 +204,25 @@ def run_method(self):
 
         print("Have parameters: " + str(len(paramList)))
 
+        varList = utils.list_from_workspace(ws, "w", "variables")
+        if self.args.setParameters is not None:
+            set_parameters = self.args.setParameters.split(",")
+            set_parameters_str = ""
+            for ind, setParam in enumerate(set_parameters):
+                if "rgx{" in setParam:
+                    eqs_to = setParam.split("=")[-1]
+                    pattern = setParam.split("=")[0]
+                    pattern = pattern.replace("'rgx{", "").replace("}'", "")
+                    pattern = pattern.replace("rgx{", "").replace("}", "")
+                    set_parameters[ind] = ""
+                    for var in varList:
+                        if re.search(pattern, var):
+                            var_str = var + "=" + eqs_to
+                            set_parameters_str += var_str + ","
+                else:
+                    set_parameters_str += setParam + ","
+        self.args.setPhysicsModelParameters = set_parameters_str.rstrip(",")
+
         prefit = utils.prefit_from_workspace(ws, "w", paramList, self.args.setPhysicsModelParameters)
         res = {}
         if not self.args.noInitialFit:
@@ -264,15 +283,9 @@ def run_method(self):
             print("Missing inputs: " + ",".join(missing))
 
     def all_free_parameters(self, file, wsp, mc, pois):
-        res = []
         wsFile = ROOT.TFile.Open(file)
         w = wsFile.Get(wsp)
         config = w.genobj(mc)
         pdfvars = config.GetPdf().getParameters(config.GetObservables())
-        it = pdfvars.createIterator()
-        var = it.Next()
-        while var:
-            if var.GetName() not in pois and (not var.isConstant()) and var.InheritsFrom("RooRealVar"):
-                res.append(var.GetName())
-            var = it.Next()
+        res = [var.GetName() for var in pdfvars if (var.GetName() not in pois and (not var.isConstant()) and var.InheritsFrom("RooRealVar"))]
         return res

python/tool_base/utils.py

@@ -33,15 +33,10 @@ def split_vals(vals, fmt_spec=None):
 
 def list_from_workspace(file, workspace, set):
     """Create a list of strings from a RooWorkspace set"""
-    res = []
     wsFile = ROOT.TFile(file)
     ws = wsFile.Get(workspace)
-    argSet = ws.set(set)
-    it = argSet.createIterator()
-    var = it.Next()
-    while var:
-        res.append(var.GetName())
-        var = it.Next()
+    argSet = ws.allVars() if set == "variables" else ws.set(set)
+    res = [var.GetName() for var in argSet]
     return res