s.tree.AddFriend("trees/WJets_weights", s.tfile) else: samples = {} out_dir = os.environ["STPOL_DIR"] + "/out/plots/wjets" if args.tag: out_dir += "/" + args.tag mkdir_p(out_dir) # plot_ratios("2J0T", Cuts.final(2,0), samples, out_dir, args.recreate) # plot_ratios("2J1T", Cuts.final(2,1), samples, out_dir, args.recreate) colls_in, colls_out = plot_sherpa_vs_madgraph( costheta, "2J", Cuts.mu*Cuts.final_jet(2), samples.values(), out_dir, recreate=args.recreate, legend_pos="top-left", nudge_x=-0.03, nudge_y=0, systematic="nominal" ) coll_in, coll_out = plot_sherpa_vs_madgraph( costheta, "2J0T", Cuts.mu*Cuts.final(2,0), samples.values(), out_dir, recreate=args.recreate, legend_pos="top-left", nudge_x=-0.03, nudge_y=0 ) coll_in, coll_out = plot_sherpa_vs_madgraph( costheta, "2J1T", Cuts.mu*Cuts.final(2,1), samples.values(), out_dir, recreate=args.recreate, legend_pos="top-left", nudge_x=-0.03, nudge_y=0 )