Functions
def	CacheFilePath (cache_file_dir, day=None)

def	GetXLocator (ax)

def	PlotPeakPUAllYears (lumi_data_by_fill_per_year)

def	processdata (years)

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

Variables
	accel_mode

	action

	amodetag_bug_workaround

	arg_parser

	args

	beam_energy

	beam_energy_defaults

	beam_energy_from_cfg

	beam_energy_tmp

	beam_fluctuation

	beam_fluctuation_defaults

	beam_fluctuation_from_cfg

	beam_fluctuation_tmp

	cache_file_dir

	cfg_defaults

	cfg_parser

	color_scheme_names

	color_scheme_names_tmp

	config_file_name

	date_begin

	date_end

	DATE_FMT_STR_AXES

	DATE_FMT_STR_CFG

	DATE_FMT_STR_LUMICALC

	DATE_FMT_STR_LUMICALC_DAY

	DATE_FMT_STR_OUT

	desc_str

	description

	file

	help

	hist

	ignore_cache

	KNOWN_ACCEL_MODES

	LEAD_SCALE_FACTOR

	lumi_data_by_fill_per_year

	lumicalc_flags_from_cfg

	lumicalc_script

	NUM_SEC_IN_LS

	options

	oracle_connection_string

	particle_type_str

	particle_type_strings

	path_name

	tmp

	today

	use_oracle

	verbose

	year_begin

	year_end

	years

Function Documentation

def create_public_peakpu_plots.CacheFilePath	(	cache_file_dir,
		day = `None`
	)

Definition at line 110 of file create_public_peakpu_plots.py.

 def CacheFilePath(cache_file_dir, day=None):
     cache_file_path = os.path.abspath(cache_file_dir)
     if day:
         cache_file_name = "lumicalc_cache_%s.csv" % day.isoformat()
         cache_file_path = os.path.join(cache_file_path, cache_file_name)
     return cache_file_path
 
 

def create_public_peakpu_plots.GetXLocator ( ax )

Pick a DateLocator based on the range of the x-axis.

Definition at line 118 of file create_public_peakpu_plots.py.

References min(), and TweakPlot().

Referenced by TweakPlot().

 def GetXLocator(ax):
     """Pick a DateLocator based on the range of the x-axis."""
     (x_lo, x_hi) = ax.get_xlim()
     num_days = x_hi - x_lo
     min_num_ticks = min(num_days, 5)
     locator = matplotlib.dates.AutoDateLocator(minticks=min_num_ticks,
                                                maxticks=None)
     # End of GetLocator().
     return locator
 

def create_public_peakpu_plots.PlotPeakPUAllYears ( lumi_data_by_fill_per_year )

Mode 1: years side-by-side

Definition at line 405 of file create_public_peakpu_plots.py.

References public_plots_tools.AddLogo(), indexGen.date, edm.print(), public_plots_tools.SavePlot(), and TweakPlot().

         def PlotPeakPUAllYears(lumi_data_by_fill_per_year):
             """Mode 1: years side-by-side"""
 
             # Loop over all color schemes and plot.
             for color_scheme_name in color_scheme_names:
                 print("      color scheme '%s'" % color_scheme_name)
                 color_scheme = ColorScheme(color_scheme_name)
                 color_by_year = color_scheme.color_by_year
                 logo_name = color_scheme.logo_name
                 file_suffix = color_scheme.file_suffix
 
                 for type in ["lin", "log"]:
                     is_log = (type == "log")
                     aspect_ratio = matplotlib.figure.figaspect(1. / 2.5)
                     fig = plt.figure(figsize=aspect_ratio)
                     ax = fig.add_subplot(111)
 
                     time_begin_ultimate = datetime.datetime.strptime(lumi_data_by_fill_per_year[years[0]][0][0],DATE_FMT_STR_LUMICALC).date() 
                     str_begin_ultimate = time_begin_ultimate.strftime(DATE_FMT_STR_OUT)
                     for (year_index, year) in enumerate(years):
 
                         lumi_data = lumi_data_by_fill_per_year[year] 
                         times_tmp = [datetime.datetime.strptime(tmp,DATE_FMT_STR_LUMICALC).date() for tmp in lumi_data[0]]
                         times = [matplotlib.dates.date2num(i) for i in times_tmp] #x_axis
                         maxpus = lumi_data[1] # y_axis
                         
                         # NOTE: Special case for 2010.
                         label = None
                         if year == 2010 or year == 2011 :
                             label = r"%d, %s" % \
                                     (year,r'7TeV $\sigma$=71.5mb')
                         else:
                             label = r"%d, %s" % \
                                     (year,r'8TeV $\sigma$=73mb')
                         ax.plot(times,maxpus,
                                 color=color_by_year[year],
 #                                marker="none", linestyle="solid",
                                 marker="o", linestyle='none',
                                 linewidth=4,
                                 label=label)
                         if is_log:
                             ax.set_yscale("log")
                     
                     time_begin = datetime.datetime(years[0], 1, 1, 0, 0, 0)
                     time_end = datetime.datetime(years[-1], 12, 16, 20, 50,9)
                     str_begin = time_begin.strftime(DATE_FMT_STR_OUT)
                     str_end = time_end.strftime(DATE_FMT_STR_OUT)
 
                     num_cols = None
                     num_cols = len(years)
                     tmp_x = 0.095
                     tmp_y = .95
                    
                     leg = ax.legend(loc="upper left", bbox_to_anchor=(tmp_x, 0., 1., tmp_y),
                               frameon=False, ncol=num_cols)
                     for t in leg.get_texts():
                         t.set_font_properties(FONT_PROPS_TICK_LABEL)
 
                     # Set titles and labels.
                     fig.suptitle(r"CMS peak interactions per crossing, %s" % particle_type_str,
                                  fontproperties=FONT_PROPS_SUPTITLE)
                     ax.set_title("Data included from %s to %s UTC \n" % \
                                  (str_begin_ultimate, str_end),
                                  fontproperties=FONT_PROPS_TITLE)
                     ax.set_xlabel(r"Date (UTC)", fontproperties=FONT_PROPS_AX_TITLE)
                     ax.set_ylabel(r"Peak interactions per crossing",\
                                   fontproperties=FONT_PROPS_AX_TITLE)
 
                     # Add the logo.
                     #zoom = 1.7
                     zoom = .95
                     AddLogo(logo_name, ax, zoom=zoom)
                     extra_head_room = 0
                     if is_log:
                         #if mode == 1:
                         extra_head_room = 1
                         #elif mode == 2:
                         #    extra_head_room = 2
                     TweakPlot(fig, ax, (time_begin, time_end),
 #                    TweakPlot(fig, ax, (time_begin_ultimate, time_end),
                               add_extra_head_room=True)
 
                     log_suffix = ""
                     if is_log:
                         log_suffix = "_log"
                     SavePlot(fig, "peak_pu_%s_%s%s" % \
                              (particle_type_str.lower(),
                               log_suffix, file_suffix))

def create_public_peakpu_plots.processdata ( years )

Definition at line 69 of file create_public_peakpu_plots.py.

References mps_setup.append, spr.find(), objects.autophobj.float, createfilelist.int, genParticles_cff.map, edm.print(), split, and str.

 def processdata(years):
     results={}#{year:[[str,str],[pu,pu]]}
     for y in years:
         fills=[]#[fillnum]
         maxputimelist=[]
         maxpulist=[]
         results[y]=[maxputimelist,maxpulist]
         f=None
         try:
             lumifilename=str(y)+'lumibyls.csv'
             f=open(lumifilename,'rb')
         except IOError:
             print('failed to open file ',lumifilename)
             return result
         freader=csv.reader(f,delimiter=',')
         idx=0
         for row in freader:
             if idx==0:
                 idx=1
                 continue
             if row[0].find('#')==1:
                 continue
             [runnum,fillnum]=map(lambda i:int(i),row[0].split(':'))
             avgpu=float(row[7])
             if avgpu>50: continue
             putime=row[2]
             max_avgpu=avgpu
             max_putime=putime
             if fillnum not in fills:#new fill
                 fills.append(fillnum)               
                 results[y][0].append(max_putime)
                 results[y][1].append(max_avgpu)
             if avgpu>max_avgpu:
                 results[y][0][-1]=max_putime
                 results[y][1][-1]=max_avgpu
         print(results)
     return results
 
 

def create_public_peakpu_plots.TweakPlot	(	fig,
		ax,
		time_range,
		add_extra_head_room = `False`
	)

Definition at line 131 of file create_public_peakpu_plots.py.

References GetXLocator().

Referenced by GetXLocator(), and PlotPeakPUAllYears().

               add_extra_head_room=False):
 
     # Fiddle with axes ranges etc.
     (time_begin, time_end) = time_range
     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)
 
     ax.set_xlim(time_begin, time_end)
 
     locator = GetXLocator(ax)
     minorXlocator=matplotlib.ticker.AutoMinorLocator()
     #ax.yaxis.set_major_locator(matplotlib.ticker.MultipleLocator(5))
     #ax.yaxis.set_minor_locator(matplotlib.ticker.MultipleLocator(5))
     ax.xaxis.set_major_locator(locator)
     ax.xaxis.set_minor_locator(minorXlocator)
 
     formatter = matplotlib.dates.DateFormatter(DATE_FMT_STR_AXES)
     ax.xaxis.set_major_formatter(formatter)
 
     fig.subplots_adjust(top=.85, bottom=.14, left=.13, right=.91)
     # End of TweakPlot().
 

Variable Documentation

create_public_peakpu_plots.accel_mode

Definition at line 225 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.action

Definition at line 188 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.amodetag_bug_workaround

Definition at line 232 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.arg_parser

Definition at line 187 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.args

Definition at line 192 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.beam_energy

Definition at line 241 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.beam_energy_defaults

Definition at line 303 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.beam_energy_from_cfg

Definition at line 242 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.beam_energy_tmp

Definition at line 238 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.beam_fluctuation

Definition at line 254 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.beam_fluctuation_defaults

Definition at line 312 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.beam_fluctuation_from_cfg

Definition at line 255 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.beam_fluctuation_tmp

Definition at line 251 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.cache_file_dir

Definition at line 218 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.cfg_defaults

Definition at line 199 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.cfg_parser

Definition at line 208 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.color_scheme_names

Definition at line 216 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.color_scheme_names_tmp

Definition at line 215 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.config_file_name

Definition at line 196 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.date_begin

Definition at line 266 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.date_end

Definition at line 268 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.DATE_FMT_STR_AXES

Definition at line 61 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.DATE_FMT_STR_CFG

Definition at line 62 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.DATE_FMT_STR_LUMICALC

Definition at line 58 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.DATE_FMT_STR_LUMICALC_DAY

Definition at line 59 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.DATE_FMT_STR_OUT

Definition at line 60 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.desc_str

Definition at line 185 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.description

Definition at line 187 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.file

Definition at line 33 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.help

Definition at line 189 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.hist

Definition at line 35 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.ignore_cache

Definition at line 197 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.KNOWN_ACCEL_MODES

Definition at line 65 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.LEAD_SCALE_FACTOR

Definition at line 67 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.lumi_data_by_fill_per_year

Definition at line 378 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.lumicalc_flags_from_cfg

Definition at line 224 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.lumicalc_script

Definition at line 223 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.NUM_SEC_IN_LS

Definition at line 63 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.options

Definition at line 192 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.oracle_connection_string

Definition at line 289 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.particle_type_str

Definition at line 301 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.particle_type_strings

Definition at line 296 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.path_name

Definition at line 363 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.tmp

Definition at line 265 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.today

Definition at line 272 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.use_oracle

Definition at line 290 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.verbose

Definition at line 220 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.year_begin

Definition at line 390 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.year_end

Definition at line 391 of file create_public_peakpu_plots.py.

create_public_peakpu_plots.years

Definition at line 376 of file create_public_peakpu_plots.py.

Functions

Variables

Function Documentation

Variable Documentation