create_public_pileup_plots Namespace Reference

Functions
def	TweakPlot

Variables
tuple	arg_parser = optparse.OptionParser(description=desc_str)
tuple	ax = fig.add_subplot(111)
list	bin_edges
tuple	cache_file_dir = cfg_parser.get("general", "cache_dir")
dictionary	cfg_defaults
tuple	cfg_parser = ConfigParser.SafeConfigParser(cfg_defaults)
string	cmd = "pileupCalc.py -i %s --inputLumiJSON=%s %s %s"
tuple	cms_energy_str = cfg_parser.get("general", "cms_energy_str")
	color_fill_pileup = color_scheme.color_fill_pileup
	color_line_pileup = color_scheme.color_line_pileup
tuple	color_scheme = ColorScheme(color_scheme_name)
list	color_scheme_names = [i.strip() for i in color_scheme_names_tmp.split(",")]
tuple	color_scheme_names_tmp = cfg_parser.get("general", "color_schemes")
list	config_file_name = args[0]
string	desc_str = "This script creates the official CMS pileup plots "
	edgecolor = color_line_pileup,
tuple	exp = RoundAwayFromZero(math.log10(min_val))
	facecolor = color_fill_pileup)
tuple	fig = plt.figure()
	file_suffix = color_scheme.file_suffix
	fontproperties = FONT_PROPS_SUPTITLE)
string	help = "Ignore all cached PU results "
string	histtype = "stepfilled"
string	horizontalalignment = "right"
	ignore_cache = options.ignore_cache
tuple	in_file = TFile.Open(tmp_file_name, "READ")
tuple	input_json = cfg_parser.get("general", "input_json")
tuple	input_lumi_json = cfg_parser.get("general", "input_lumi_json")
tuple	is_log = (type == "log")
	log_setting = False
string	log_suffix = ""
	logo_name = color_scheme.logo_name
tuple	min_val = min(weights)
tuple	particle_type_str = cfg_parser.get("general", "particle_type_str")
tuple	pileup_hist = in_file.Get("pileup")
tuple	pileupcalc_flags_from_cfg = cfg_parser.get("general", "pileupcalc_flags")
tuple	tmp_file_name = os.path.join(cache_file_dir,"pileup_calc_tmp.root")
	transform = ax.transAxes,
list	vals
tuple	verbose = cfg_parser.getboolean("general", "verbose")
list	weights
tuple	year = int(cfg_parser.get("general", "year"))

Function Documentation

def create_public_pileup_plots.TweakPlot	(		fig,
			ax,
			add_extra_head_room = False
	)

Definition at line 61 of file create_public_pileup_plots.py.

def TweakPlot(fig, ax, add_extra_head_room=False):

    # Fiddle with axes ranges etc.
    ax.relim()
    ax.autoscale_view(False, True, True)
    for label in ax.get_xticklabels():
        label.set_ha("right")
        label.set_rotation(30.)

    # Bit of magic here: increase vertical scale by one tick to make
    # room for the legend.
    if add_extra_head_room:
        y_ticks = ax.get_yticks()
        (y_min, y_max) = ax.get_ylim()
        is_log = (ax.get_yscale() == "log")
        y_max_new = y_max
        if is_log:
            tmp = y_ticks[-1] / y_ticks[-2]
            y_max_new = y_max * math.pow(tmp, add_extra_head_room)
        else:
            tmp = y_ticks[-1] - y_ticks[-2]
            y_max_new = y_max + add_extra_head_room * tmp
        ax.set_ylim(y_min, y_max_new)

    # Add a second vertical axis on the right-hand side.
    ax_sec = ax.twinx()
    ax_sec.set_ylim(ax.get_ylim())
    ax_sec.set_yscale(ax.get_yscale())

    for ax_tmp in fig.axes:
        for sub_ax in [ax_tmp.xaxis, ax_tmp.yaxis]:
            for label in sub_ax.get_ticklabels():
                label.set_font_properties(FONT_PROPS_TICK_LABEL)

    if is_log:
        fig.subplots_adjust(top=.89, bottom=.125, left=.11, right=.925)
    else:
        fig.subplots_adjust(top=.89, bottom=.125, left=.1, right=.925)

    # End of TweakPlot().