def getContribution(self, contributiondata, doScaling = True): sumhist = None for subcontribution in contributiondata["Contributions"]: hist = DB.getHistogram(subcontribution, doScaling) if sumhist: sumhist.Add(hist) else: sumhist = hist return sumhist
def main( argv ): """ Main function to be executed when starting the code. """ ROOT.gROOT.SetBatch() ROOT.TH1.AddDirectory(False) PS.setStyle(); parser = argparse.ArgumentParser( description = 'Plotting Tool using YAML files for configuration' ) parser.add_argument('configfile', type=str, help='configuration file to be used') args = parser.parse_args() configModuleName = args.configfile.replace("/", ".").strip(".py") configuration = importlib.import_module(configModuleName).config for histogram in configuration["Histograms"]: Database.UpdateDataBase(configuration, histogram) DrawPlot(configuration, histogram)
def plot_results(histograms): """ Main function to be executed when starting the code. """ ROOT.gROOT.SetBatch() ROOT.TGaxis.SetMaxDigits(4) ROOT.TH1.AddDirectory(False) PS.setStyle() #parser = argparse.ArgumentParser( description = 'Plotting Tool using YAML files for configuration' ) #parser.add_argument('configfile', type=str, help='configuration file to be used') #configModuleName = "PlotConf_CustomAnalysis" #args = parser.parse_args() #configModuleName = args.configfile.replace("/", ".").strip(".py") configuration = PlotConf_CustomAnalysis.config for histogram in histograms: Database.UpdateDataBase(configuration, histogram) DrawPlot(configuration, histogram) Database.config = dict() Database.histoptions = OrderedDict() Database.rootFiles = dict()
def plot_results(samples, histograms): """ Main function to be executed when starting the code. """ ROOT.gROOT.SetBatch() ROOT.TGaxis.SetMaxDigits(4) ROOT.TH1.AddDirectory(False) PS.setStyle(); #parser = argparse.ArgumentParser( description = 'Plotting Tool using YAML files for configuration' ) #parser.add_argument('configfile', type=str, help='configuration file to be used') #configModuleName = "PlotConf_CustomAnalysis" #args = parser.parse_args() #configModuleName = args.configfile.replace("/", ".").strip(".py") ## TODO: ## XD ## IF samples contains nonsense that does not match anything, it returns a garbage skeleton dict. if samples: configuration = PlotConf_CustomAnalysis.config keptStack = False keptData = False keptKeys = [] for k, v in configuration["Paintables"]["Stack"]["Processes"].copy().iteritems(): keepKey = False for j in v["Contributions"]: if j in samples: keepKey = True keptStack = True if k not in keptKeys: keptKeys.append(k) if not keepKey: del configuration["Paintables"]["Stack"]["Processes"][k] configuration["Paintables"]["Stack"]["Order"] = keptKeys cleanContrib = [] for k, v in configuration["Paintables"]["Stack"]["Processes"].copy().iteritems(): for j in v["Contributions"]: if j not in samples: cleanContrib.append((k, j)) for i in cleanContrib: configuration["Paintables"]["Stack"]["Processes"][i[0]]["Contributions"].remove(i[1]) if not configuration["Paintables"]["Stack"]["Processes"].keys(): del configuration["Paintables"]["Stack"] toRemove = [] for i in configuration["Paintables"]["data"]["Contributions"]: if i in samples: keptData = True if i not in samples: toRemove.append(i) for i in toRemove: configuration["Paintables"]["data"]["Contributions"].remove(i) if not configuration["Paintables"]["data"]["Contributions"]: del configuration["Paintables"]["data"] if (not keptStack) or (not keptData): configuration["Depictions"]["Order"].remove("Data/MC") del configuration["Depictions"]["Definitions"]["Data/MC"] configuration["Depictions"]["Definitions"]["Main"]["Paintables"] = [] if keptStack: configuration["Depictions"]["Definitions"]["Main"]["Paintables"].append("Stack") if keptData: configuration["Depictions"]["Definitions"]["Main"]["Paintables"].append("data") else: configuration = PlotConf_CustomAnalysis.config for histogram in histograms: Database.UpdateDataBase(configuration, histogram) DrawPlot(configuration, histogram) Database.config = dict() Database.histoptions = OrderedDict() Database.rootFiles = dict()