TkAlMap.TkAlMap Class Reference

Public Member Functions
def	__del__ (self)
def	__init__ (self, variable, title, root_file, use_default_range=False, two_sigma_cap=False, height=1400, GEO_file='TkAlMapDesign_phase1_cfg.py', tracker='full', palette=2, do_tanh=False, check_tracker=True)
def	analyse (self)
	Do all. More...
def	clean_up (self)
	Clean up. More...
def	detect_tracker_version (self)
def	draw_cms_prelim (self)
def	draw_color_bar (self)
def	draw_event_info (self, y)
def	draw_text (self)
	Draw functions. More...
def	draw_title (self)
	Titles and info. More...
def	draw_TPL (self)
def	fill_colors (self)
def	get_color_rgb (self, val)
def	get_color_rgb_idx (self, val)
def	load_geometry (self)
def	load_tree (self)
	Load functions. More...
def	load_var (self)
def	plot_variable_distribution (self, nbins=200, out_dir='.')
	Test functions. More...
def	prepare_map_colors (self)
def	save (self, out_dir='.', extension='pdf')
def	set_canvas (self)
def	set_colorbar_axis (self)
def	set_colorbar_colors (self)
def	set_palette (self, palette)
def	set_var (self, var, var_range=[None, None)
def	setup_colors (self)
	Color setup. More...

Public Attributes
	canvas
	cfg_path
	col_dic
	color_bar
	color_bar_axis
	color_bar_colors
	colors
	data_path
	default_range
	do_tanh
	GEO_file
	tree = r_file.Get(tree_name) tree_tmp = r_file.Get(tree_name) tree = copy.deepcopy(tree_tmp) r_file.Close() More...
	height
	image_x1
	image_x2
	image_y1
	image_y2
	is_cleaned
	max_val
	mean_val
	min_val
	mod_val_dict
	n_color_color_bar
	palette
	rgb_map
	root_file
	start_color_idx
	std_val
	title
	TkAlMap_text_dict
	TkAlMap_TPL_dict
	tmp_file
	tmp_file_name
	tracker
	tree
	two_sigma_cap
	val_list
	else: if 'full' in self.tracker: print('Warning: Unknown module '+str(module)) r_file.Close() More...
	var
	var_max
	var_min
	var_name
	var_scale
	var_units
	width
	x_scale
	y_scale

Detailed Description

Definition at line 151 of file TkAlMap.py.

Constructor & Destructor Documentation

__init__()

def TkAlMap.TkAlMap.__init__	(		self,
			variable,
			title,
			root_file,
			use_default_range = False,
			two_sigma_cap = False,
			height = 1400,
			GEO_file = 'TkAlMapDesign_phase1_cfg.py',
			tracker = 'full',
			palette = 2,
			do_tanh = False,
			check_tracker = True
	)

Definition at line 154 of file TkAlMap.py.

    def __init__(self, variable, title, root_file, use_default_range=False, two_sigma_cap=False, height=1400, GEO_file='TkAlMapDesign_phase1_cfg.py', tracker='full', palette=2, do_tanh=False, check_tracker=True):
        ROOT.gStyle.SetLineScalePS(1)

        # Configuration parameters
        self.GEO_file      = GEO_file
        self.tracker       = tracker
        self.width         = height 
        self.height        = height
        self.title         = title
        self.default_range = use_default_range
        self.two_sigma_cap = two_sigma_cap
        self.root_file     = root_file
        self.do_tanh       = do_tanh

        # Value Initialization
        self.max_val           = None
        self.min_val           = None
        self.color_bar_colors  = {}
        self.tree              = None
        self.is_cleaned        = False

        # Internal parameters    
        self.data_path = 'Alignment/OfflineValidation/data/TkAlMap/'
        self.cfg_path = 'Alignment/OfflineValidation/python/TkAlMap_cfg/'

        # Colorbar stuff
        self.start_color_idx   = 1200
        self.n_color_color_bar = 1000
 
        # Initialization functions
        self.set_palette(palette)
        self.set_var(variable)
        self.load_tree()
        if check_tracker: self.detect_tracker_version()
        self.load_geometry()
        self.set_colorbar_colors()

__del__()

def TkAlMap.TkAlMap.__del__

(

self

)

Definition at line 753 of file TkAlMap.py.

References TkAlMap.TkAlMap.clean_up().

    def __del__(self):
        self.clean_up()

Member Function Documentation

analyse()

def TkAlMap.TkAlMap.analyse

(

self

)

Do all.

Definition at line 702 of file TkAlMap.py.

References TkAlMap.TkAlMap.do_tanh, TkAlMap.TkAlMap.draw_color_bar(), TkAlMap.TkAlMap.draw_text(), TkAlMap.TkAlMap.draw_TPL(), TkAlMap.TkAlMap.set_colorbar_axis(), TkAlMap.TkAlMap.set_colorbar_colors(), and TkAlMap.TkAlMap.setup_colors().

    def analyse(self):
        self.setup_colors()
        self.set_colorbar_axis()
        if self.do_tanh: self.set_colorbar_colors()
        self.draw_TPL()
        self.draw_text()
        self.draw_color_bar()

clean_up()

def TkAlMap.TkAlMap.clean_up

(

self

)

Clean up.

Definition at line 746 of file TkAlMap.py.

References TkAlMap.TkAlMap.is_cleaned, print(), TkAlMap.TkAlMap.tmp_file, and TkAlMap.TkAlMap.tmp_file_name.

Referenced by TkAlMap.TkAlMap.__del__().

    def clean_up(self):
        if not self.is_cleaned:
            print('TkAlMap: deleting temporary file "'+self.tmp_file_name+'"')
            self.tmp_file.Close()
            os.remove(self.tmp_file_name)
            self.is_cleaned = True

detect_tracker_version()

def TkAlMap.TkAlMap.detect_tracker_version

(

self

)

Definition at line 474 of file TkAlMap.py.

References TkAlMap.TkAlMap.GEO_file, print(), str, HIPMonitorVariables.tree, AlignmentIORootBase.tree, HIPTwoBodyDecayAnalyzer.tree, TkOfflineVariables.tree, and TkAlMap.TkAlMap.tree.

    def detect_tracker_version(self):
        print('TkAlMap: detecting Tk version')
        #tree_name = 'alignTree'
        #r_file = ROOT.TFile(self.root_file)
        #if r_file is None: raise ValueError('The file "'+self.root_file+'" could not be opened')

        

        #if tree is None: raise ValueError('The tree "'+tree_name+'" was not found in file "'+self.root_file+'"')
        phase = None
        for event in self.tree:
            module = event.id
            if module > 303040000 and module < 306450000:
                phase = 1
                break
            elif module > 302055000 and module < 302198000:
                phase = 0 
                break
        #r_file.Close()

        if phase is None: raise ValueError('TkAlMap: unknown tracker detected, is this phase2?')
        
        pahse_str = 'phase'+str(phase)
        print('TkAlMap: '+pahse_str+' tracker detected')
        if not pahse_str in self.GEO_file:
            print('TkAlMap: changing tracker to '+pahse_str+ ', if this is unwanted set "check_tracker" to False')
            self.GEO_file = 'TkAlMapDesign_'+pahse_str+'_cfg.py'
            #self.load_geometry()

draw_cms_prelim()

def TkAlMap.TkAlMap.draw_cms_prelim

(

self

)

Definition at line 626 of file TkAlMap.py.

References TkAlMap.TkAlMap.canvas, DMRplotter.DMRplotter.canvas, hippyaddtobaddatafiles.cd(), TkAlMap.TkAlMap.draw_event_info(), presentation.PageLayout.height, TkAlMap.TkAlMap.height, svgfig.Plot.height, svgfig.Frame.height, svgfig.Dots.height, TkAlMap.TkAlMap.image_x1, presentation.PageLayout.width, TkAlMap.TkAlMap.width, svgfig.Plot.width, svgfig.Frame.width, svgfig.Dots.width, and TkAlMap.TkAlMap.y_scale.

Referenced by TkAlMap.TkAlMap.draw_text().

    def draw_cms_prelim(self):
        TL = ROOT.TLatex()
        factor = 1. / 0.82
        TL.SetTextSize(0.035*factor)
        TL.SetTextAlign(11)
        TL.SetTextFont(61)

        w_cms = ctypes.c_uint(0)
        h_cms = ctypes.c_uint(0)
        TL.GetTextExtent(w_cms, h_cms, 'CMS')
        x1 = self.image_x1
        y1 = 1. - (h_cms.value+0.)/(self.height+0.) - (1./(self.y_scale+0.))
        self.canvas.cd()
        TL.DrawLatex(x1, y1, 'CMS')
  
        TL.SetTextSize(0.035)
        TL.SetTextFont(42)
        x1_prel = x1 + 1.1*(w_cms.value+0.)/(self.width+0.)
        TL.DrawLatex(x1_prel, y1, '#it{Preliminary}')
        
        self.draw_event_info(y1)

draw_color_bar()

def TkAlMap.TkAlMap.draw_color_bar

(

self

)

Definition at line 681 of file TkAlMap.py.

References TkAlMap.TkAlMap.canvas, DMRplotter.DMRplotter.canvas, hippyaddtobaddatafiles.cd(), TkAlMap.TkAlMap.color_bar, TkAlMap.TkAlMap.color_bar_axis, and print().

Referenced by TkAlMap.TkAlMap.analyse().

    def draw_color_bar(self):
        print('TkAlMap: drawing color bar')
        self.canvas.cd()
        for box in self.color_bar:
            self.color_bar[box].Draw('f')
            #self.color_bar[box].Draw()
        self.color_bar_axis.Draw()
        self.canvas.Update()

draw_event_info()

def TkAlMap.TkAlMap.draw_event_info	(		self,
			y
	)

Definition at line 648 of file TkAlMap.py.

References TkAlMap.TkAlMap.canvas, DMRplotter.DMRplotter.canvas, hippyaddtobaddatafiles.cd(), and TkAlMap.TkAlMap.image_x2.

Referenced by TkAlMap.TkAlMap.draw_cms_prelim().

    def draw_event_info(self, y):
        TL = ROOT.TLatex()
        TL.SetTextSize(0.035)
        TL.SetTextFont(42)
        TL.SetTextAlign(31)

        x1 = self.image_x2
        y1 = y
        self.canvas.cd()
        TL.DrawLatex(x1, y1, 'pp collisions 13TeV')
        

draw_text()

def TkAlMap.TkAlMap.draw_text

(

self

)

Draw functions.

Definition at line 661 of file TkAlMap.py.

References TkAlMap.TkAlMap.canvas, DMRplotter.DMRplotter.canvas, hippyaddtobaddatafiles.cd(), TkAlMap.TkAlMap.draw_cms_prelim(), TkAlMap.TkAlMap.draw_title(), print(), and TkAlMap.TkAlMap.TkAlMap_text_dict.

Referenced by TkAlMap.TkAlMap.analyse().

    def draw_text(self):
        print('TkAlMap: drawing text')
        self.canvas.cd()
        TL = ROOT.TLatex()
        TL.SetTextSize(0.025)
        for key in self.TkAlMap_text_dict:
            TL.SetTextAlign(self.TkAlMap_text_dict[key]['alignment'])
            TL.DrawLatex(self.TkAlMap_text_dict[key]['x'], self.TkAlMap_text_dict[key]['y'], self.TkAlMap_text_dict[key]['text'])
        self.draw_cms_prelim()
        self.draw_title()
        self.canvas.Update()

draw_title()

def TkAlMap.TkAlMap.draw_title

(

self

)

Titles and info.

Definition at line 616 of file TkAlMap.py.

References TkAlMap.TkAlMap.canvas, DMRplotter.DMRplotter.canvas, hippyaddtobaddatafiles.cd(), TkAlMap.TkAlMap.image_x1, classes.PlotData.title, preexistingValidation.PreexistingValidation.title, alignment.Alignment.title, presentation.SubsectionBase.title, Geometry.title, TkAlMap.TkAlMap.title, and TkAlMap.TkAlMap.y_scale.

Referenced by TkAlMap.TkAlMap.draw_text().

    def draw_title(self):
        TL = ROOT.TLatex()
        TL.SetTextSize(0.035)
        TL.SetTextFont(42)
        TL.SetTextAlign(13)
        x1 = self.image_x1
        y1 = 1-(5./(self.y_scale+0.))
        self.canvas.cd()
        TL.DrawLatex(x1, y1, self.title)
         

draw_TPL()

def TkAlMap.TkAlMap.draw_TPL

(

self

)

Definition at line 673 of file TkAlMap.py.

References TkAlMap.TkAlMap.canvas, DMRplotter.DMRplotter.canvas, hippyaddtobaddatafiles.cd(), print(), and TkAlMap.TkAlMap.TkAlMap_TPL_dict.

Referenced by TkAlMap.TkAlMap.analyse().

    def draw_TPL(self):
        print('TkAlMap: drawing PolyLines')
        self.canvas.cd()
        for module in self.TkAlMap_TPL_dict:
            self.TkAlMap_TPL_dict[module].Draw('f')
            self.TkAlMap_TPL_dict[module].Draw()
        self.canvas.Update()

fill_colors()

def TkAlMap.TkAlMap.fill_colors

(

self

)

Definition at line 313 of file TkAlMap.py.

References TkAlMap.TkAlMap.get_color_rgb_idx(), TkAlMap.TkAlMap.max_val, TkAlMap.TkAlMap.min_val, TkAlMap.TkAlMap.mod_val_dict, print(), TkAlMap.TkAlMap.rgb_map, and TkAlMap.TkAlMap.TkAlMap_TPL_dict.

Referenced by TkAlMap.TkAlMap.setup_colors().

    def fill_colors(self):
        print('TkAlMap: filling the colors')
        #self.set_palette()
        for module in self.mod_val_dict:
            if module in self.TkAlMap_TPL_dict:
                val = self.mod_val_dict[module]
                cap_val = val
                if cap_val > self.max_val: cap_val = self.max_val
                if cap_val < self.min_val: cap_val = self.min_val
                rgb = self.get_color_rgb_idx(cap_val)
                col = self.rgb_map[rgb]
                #col = self.pal_map[rgb]
                #col = self.col_dic[rgb]
                #print(val, rgb, col)
                self.TkAlMap_TPL_dict[module].SetFillColor(col)
                #self.TkAlMap_TPL_dict[module].SetFillColor(TEST_COLOR_IDX)
            

get_color_rgb()

def TkAlMap.TkAlMap.get_color_rgb	(		self,
			val
	)

Definition at line 263 of file TkAlMap.py.

References TkAlMap.TkAlMap.do_tanh, createfilelist.int, TkAlMap.TkAlMap.max_val, TkAlMap.TkAlMap.mean_val, TkAlMap.TkAlMap.min_val, TkAlMap.TkAlMap.palette, TkAlMap.TkAlMap.std_val, and str.

Referenced by TkAlMap.TkAlMap.get_color_rgb_idx().

    def get_color_rgb(self, val):
        if self.max_val is None or self.min_val is None:
            value_frac = val
        else:
            if self.do_tanh:
                val_th = math.tanh((val - self.mean_val)/(self.std_val))
                max_th = math.tanh((self.max_val - self.mean_val)/(self.std_val))
                min_th = math.tanh((self.min_val - self.mean_val)/(self.std_val))
                value_frac = (val_th - min_th + 0.)/(max_th - min_th)
            else:
                value_range = self.max_val - self.min_val
                if value_range == 0.: value_frac = 0.5
                else: value_frac = (val - self.min_val + 0.)/(value_range + 0.)

        if self.palette == 1:
            r = 255
            g = 255
            b = 255

            if value_frac < 0.25:
                r = 0
                g = int(255.*((value_frac/0.25)))
                b = 255         
            elif value_frac < 0.5:
                r = 0
                g = 255
                b = int(255.*(1. -(value_frac - 0.25)/0.25)) 
            elif value_frac < 0.75:
                r = int(255.*((value_frac - 0.5)/0.25))
                g = 255
                b = 0 
            else:
                r = 255 
                g = int(255.*(1. -(value_frac - 0.75)/0.25)) 
                b = 0
            return r, g, b 
        elif self.palette == 2:
            red   = [59,   76, 192]
            blue  = [180,   4,  38]
            white = [255, 255, 255] 
            r, g, b = DivergingColor(red, blue, white, value_frac) 
            return r, g, b
        else: raise ValueError('TkAlMap: unkown palette value '+str(palette)+', allowed values are 1 and 2')

get_color_rgb_idx()

def TkAlMap.TkAlMap.get_color_rgb_idx	(		self,
			val
	)

Definition at line 307 of file TkAlMap.py.

References TkAlMap.TkAlMap.get_color_rgb(), and createfilelist.int.

Referenced by TkAlMap.TkAlMap.fill_colors().

    def get_color_rgb_idx(self, val):
        r, g, b = self.get_color_rgb(val)
        #return r*1000000+g*1000+b+1000000000
        offset = 100
        return int(r*255*255 + g*255 + r + g + b + offset)

load_geometry()

def TkAlMap.TkAlMap.load_geometry

(

self

)

Definition at line 506 of file TkAlMap.py.

References TkAlMap.TkAlMap.canvas, DMRplotter.DMRplotter.canvas, TkAlMap.TkAlMap.cfg_path, TkAlMap.TkAlMap.data_path, TkAlMap.TkAlMap.GEO_file, presentation.PageLayout.height, TkAlMap.TkAlMap.height, svgfig.Plot.height, svgfig.Frame.height, svgfig.Dots.height, createfilelist.int, SiStripPI.max, SiStripPI.min, FastTimerService_cff.range, fileinputsource_cfi.read, TkAlMap.read_TPLfile(), submitPVValidationJobs.split(), TkAlMap.TkAlMap.tracker, presentation.PageLayout.width, TkAlMap.TkAlMap.width, svgfig.Plot.width, svgfig.Frame.width, and svgfig.Dots.width.

    def load_geometry(self):
        source_path = os.getenv('CMSSW_BASE') + '/src/' 
        var = {}
        if sys.version_info[0] == 2:
            execfile(source_path + self.cfg_path + self.GEO_file, var)
        elif sys.version_info[0] == 3:
            _filename = source_path + self.cfg_path + self.GEO_file
            exec(compile(open(_filename, "rb").read(), _filename, 'exec'), var)

        MapStructure = var['TkMap_GEO']
        
        all_modules = {}
        all_text = {}
        x_max = -9999.
        y_max = -9999.
        x_min = 9999.
        y_min = 9999.
        for det in MapStructure:
            if 'pixel' in self.tracker:
                if not 'pixel' in det: continue 
            elif 'strips' in self.tracker:
                if not 'strips' in det: continue 
            for sub in MapStructure[det]:
                for part in MapStructure[det][sub]:
                    if part == 'latex':
                        all_text[det+'_'+sub] = MapStructure[det][sub][part]
                        continue
                    if 'latex' in MapStructure[det][sub][part]:
                        all_text[det+'_'+sub+'_'+part] = MapStructure[det][sub][part]['latex']
                    _filename = self.data_path +MapStructure[det][sub][part]['file']
                    TPL_file = [os.path.join(dir,_filename) for dir in os.getenv('CMSSW_SEARCH_PATH','').split(":") if os.path.exists(os.path.join(dir,_filename))][0]
                    TPL_dict = read_TPLfile(TPL_file)
                    for module in TPL_dict:
                        x_canv = []
                        y_canv = []
                        for idx in range(len(TPL_dict[module]['x'])):
                            x_canv.append(TPL_dict[module]['x'][idx]*MapStructure[det][sub][part]['x_scale'] + MapStructure[det][sub][part]['x_off'])
                            y_canv.append(TPL_dict[module]['y'][idx]*MapStructure[det][sub][part]['y_scale'] + MapStructure[det][sub][part]['y_off'])
                        if max(x_canv) > x_max: x_max = max(x_canv)
                        if max(y_canv) > y_max: y_max = max(y_canv)
                        if min(x_canv) < x_min: x_min = min(x_canv)
                        if min(y_canv) < y_min: y_min = min(y_canv)
                        TPL_dict[module]['x'] = x_canv
                        TPL_dict[module]['y'] = y_canv
                    all_modules.update(TPL_dict)

        r_margin = 3
        l_margin = 3
        #t_margin = 15
        t_margin = 11
        b_margin = 8

        x_max += r_margin
        x_min -= l_margin
        y_max += t_margin
        y_min -= b_margin

        x_range = x_max - x_min
        y_range = y_max - y_min

        self.width = int(self.height*(x_range + 0.)/(y_range + 0.))
        self.canvas.SetWindowSize(self.width, self.height)

        if (x_range + 0.)/(self.width + 0.) > (y_range + 0.)/(self.height + 0.):
            x_scale = x_range
            y_scale = (self.height + 0.)/(self.width + 0.)*x_range
        else:
            y_scale = y_range
            x_scale = (self.width + 0.)/(self.height + 0.)*y_range
        self.TkAlMap_TPL_dict = {} 
        for module in all_modules:
            x = array('d', [])
            y = array('d', [])
            for idx in range(len(all_modules[module]['x'])):
                x.append((all_modules[module]['x'][idx] - x_min + 0.)/(x_scale + 0.))
                y.append((all_modules[module]['y'][idx] - y_min + 0.)/(y_scale + 0.))
            # Begin point is end point
            x.append((all_modules[module]['x'][0] - x_min + 0.)/(x_scale + 0.))
            y.append((all_modules[module]['y'][0] - y_min + 0.)/(y_scale + 0.))
            #print(x, y)
            self.TkAlMap_TPL_dict[module] = ROOT.TPolyLine(len(x), x, y) 
            #self.TkAlMap_TPL_dict[module].SetFillColor(1)
            self.TkAlMap_TPL_dict[module].SetLineColor(1)
            #print('lineW', self.TkAlMap_TPL_dict[module].GetLineWidth())
            #self.TkAlMap_TPL_dict[module].Draw('f')
            #self.TkAlMap_TPL_dict[module].Draw()

        self.image_x1 = (l_margin + 0.)/(x_scale + 0.) 
        self.image_x2 = (x_max - r_margin - x_min + 0.)/(x_scale + 0.) 
        self.image_y1 = (b_margin + 0.)/(y_scale + 0.) 
        self.image_y2 = (y_max - t_margin - y_min + 0.)/(y_scale + 0.) 

        self.x_scale = x_scale
        self.y_scale = y_scale

        #TL = ROOT.TLatex()
        #TL.SetTextSize(0.025)
        self.TkAlMap_text_dict = {}
        for key in all_text:
            x = (all_text[key]['x'] - x_min + 0.)/(x_scale + 0.)
            y = (all_text[key]['y'] - y_min + 0.)/(y_scale + 0.)
            self.TkAlMap_text_dict[key] = {}
            self.TkAlMap_text_dict[key]['x'] = x
            self.TkAlMap_text_dict[key]['y'] = y
            self.TkAlMap_text_dict[key]['alignment'] = all_text[key]['alignment']
            self.TkAlMap_text_dict[key]['text'] = all_text[key]['text']
            #TL.SetTextAlign(all_text[key]['alignment'])
            #TL.DrawLatex(x, y, all_text[key]['text'])

load_tree()

def TkAlMap.TkAlMap.load_tree

(

self

)

Load functions.

Definition at line 394 of file TkAlMap.py.

References print(), and TkAlMap.TkAlMap.root_file.

    def load_tree(self):
        print('TkAlMap: loading tree ')
        tree_name = 'alignTree'
        r_file = ROOT.TFile(self.root_file)
        if r_file is None: raise ValueError('The file "'+self.root_file+'" could not be opened')

        tree_tmp = r_file.Get(tree_name)
        #self.tree = copy.deepcopy(tree_tmp)
        self.tmp_file_name = str(time.time()).replace('.', '_')+'_TkAlMapTempFile.root'
        self.tmp_file = ROOT.TFile(self.tmp_file_name, 'recreate')
        self.tree = tree_tmp.CloneTree()
        r_file.Close()
        self.is_cleaned = False

        if self.tree is None: raise ValueError('The tree "'+tree_name+'" was not found in file "'+self.root_file+'"')
        

load_var()

def TkAlMap.TkAlMap.load_var

(

self

)

Definition at line 412 of file TkAlMap.py.

References print().

Referenced by TkAlMap.TkAlMap.setup_colors().

    def load_var(self):
        print('TkAlMap: loading variable values ')
        #tree_name = 'alignTree'
        #r_file = ROOT.TFile(self.root_file)
        #if r_file is None: raise ValueError('The file "'+self.root_file+'" could not be opened')

        #tree_tmp = r_file.Get(tree_name)
        #tree = copy.deepcopy(tree_tmp)
        #r_file.Close()

        #if tree is None: raise ValueError('The tree "'+tree_name+'" was not found in file "'+self.root_file+'"')

        self.mod_val_dict = {}
        self.val_list = []
        for event in self.tree:
            module = event.id
            var = self.var
            if var == 'rdphi':
                val = getattr(event, 'r')*getattr(event, 'dphi')            else: 
                val = getattr(event, var)
            val *= self.var_scale
            self.mod_val_dict[module] = val
            #if val not in self.val_list: self.val_list.append(val)
            if module in self.TkAlMap_TPL_dict: self.val_list.append(val)
            
        if len(self.val_list) == 0:
            print('Warning: no values filled, 0 moduleId\'s matched')
            self.val_list = [-10+idx*0.5 for idx in range(41)]
        self.val_list.sort()
        self.mean_val = mean(self.val_list)
        self.std_val = StdDev(self.val_list)
        self.min_val = min(self.val_list) 
        self.max_val = max(self.val_list)
        
        if self.two_sigma_cap and not self.default_range:
           print('-- Capping max and min: ')
           print('---- True values   : '+str(self.max_val)+', '+str(self.min_val))
           self.min_val = max(min(self.val_list), self.mean_val - 2*self.std_val)
           self.max_val = min(max(self.val_list), self.mean_val + 2*self.std_val)
           print('---- Capped values : '+str(self.max_val)+', '+str(self.min_val))

        if self.default_range:
           #if not self.var in KNOWN_VARIABLES: print('Warning: capping to default range not possible for unknown variable "'+self.var+'"')
           if self.var_min is None or self.var_max is None: print('Warning: capping to default range for unknown variable "'+self.var+'" while range was not set is not possible')
           else:
               print('-- Capping max and min to default ranges: ')
               print('---- True values   : '+str(self.max_val)+', '+str(self.min_val))
               self.min_val = self.var_min
               self.max_val = self.var_max
               print('---- Capped values : '+str(self.max_val)+', '+str(self.min_val))

        if self.min_val == self.max_val:
            print('Warning: minimum value was equal to maximum value, '+str(self.max_val))
            self.min_val = self.mean_val - 1.
            self.max_val = self.mean_val + 1.

        #print(self.val_list)

plot_variable_distribution()

def TkAlMap.TkAlMap.plot_variable_distribution	(		self,
			nbins = 200,
			out_dir = '.'
	)

Test functions.

Definition at line 711 of file TkAlMap.py.

References join(), SiStripPI.max, TkAlMap.TkAlMap.mean_val, SiStripPI.min, print(), TkAlMap.TkAlMap.std_val, TkAlMap.TkAlMap.tracker, TkAlMap.TkAlMap.two_sigma_cap, TkAlMap.TkAlMap.val_list, TkAlMap.TkAlMap.var, TkAlMap.TkAlMap.var_name, and TkAlMap.TkAlMap.var_units.

    def plot_variable_distribution(self, nbins=200, out_dir='.'):
        print('TkAlMap: drawing variable distribution')
        canv_name = 'histogram_canvas_'+self.tracker+'_'+self.var
        if self.two_sigma_cap: canv_name += '_cap'
        canvas = ROOT.TCanvas(canv_name, 'TkAlMap '+self.var+' histogram canvas', 800, 800)
 
        h_min = min(min(self.val_list), self.mean_val - 2*self.std_val) - self.std_val
        h_max = max(max(self.val_list), self.mean_val + 2*self.std_val) + self.std_val
        hist = ROOT.TH1F(self.var+'_hist', 'Variable distribution', nbins, h_min, h_max)
        for val in self.val_list:
            hist.Fill(val)
        hist.GetXaxis().SetTitle(self.var_name+' ['+self.var_units+']')
        hist.GetYaxis().SetTitle('modules')
        ROOT.gStyle.SetOptStat(0)
        hist.Draw('e1')
        canvas.Update()
        left = ROOT.TLine(self.mean_val - 2*self.std_val, canvas.GetUymin(), self.mean_val - 2*self.std_val, canvas.GetUymax())
        left.SetLineColor(2) 
        left.SetLineStyle(9) 
        left.Draw() 
        right = ROOT.TLine(self.mean_val + 2*self.std_val, canvas.GetUymin(), self.mean_val + 2*self.std_val, canvas.GetUymax())
        right.SetLineColor(2) 
        right.SetLineStyle(9) 
        right.Draw() 
        mid = ROOT.TLine(self.mean_val, canvas.GetUymin(), self.mean_val, canvas.GetUymax())
        mid.SetLineColor(1)
        mid.SetLineStyle(9) 
        mid.Draw()
        canvas.Update()
        name = '_'.join(['VariableDistribution', self.var, self.tracker])
        path = out_dir + '/' + name + '.png' 
        canvas.SaveAs(path)

prepare_map_colors()

def TkAlMap.TkAlMap.prepare_map_colors

(

self

)

Definition at line 219 of file TkAlMap.py.

References print().

Referenced by TkAlMap.TkAlMap.setup_colors().

    def prepare_map_colors(self):
        print('TkAlMap: preparing map colors')
        
        self.colors = []
        #self.palette = array('i', [])
        col_idx = self.start_color_idx + self.n_color_color_bar + 10
        self.col_dic = {}
        self.rgb_map = {}
        #pal_idx = 0
        #self.pal_map = {}
        for val in self.val_list:
            cap_val = val
            if cap_val > self.max_val: cap_val = self.max_val
            if cap_val < self.min_val: cap_val = self.min_val
            r, g, b = self.get_color_rgb(cap_val)
            idx = self.get_color_rgb_idx(cap_val)
            if idx in self.colors: continue
            self.colors.append(idx)
            col_idx +=1
            self.rgb_map[idx] = col_idx
            #print( idx, (r+0.)/255., (g+0.)/255., (b+0.)/255.)
            #color = ROOT.TColor(col_idx, (r+0.)/255., (g+0.)/255., (b+0.)/255.)

            #self.col_dic[idx] = ROOT.TColor(col_idx, (r+0.)/255., (g+0.)/255., (b+0.)/255.)
            try:
                col = ROOT.gROOT.GetColor(col_idx)
                col.SetRGB((r+0.)/255., (g+0.)/255., (b+0.)/255.)
                self.col_dic[idx] = col
            except:
                self.col_dic[idx] = ROOT.TColor(col_idx, (r+0.)/255., (g+0.)/255., (b+0.)/255.)
            #self.palette.append(col_idx)
        print('TkAlMap: map contains '+str(len(self.colors))+' colors')
     

save()

def TkAlMap.TkAlMap.save	(		self,
			out_dir = '.',
			extension = 'pdf'
	)

Definition at line 690 of file TkAlMap.py.

References TkAlMap.TkAlMap.canvas, DMRplotter.DMRplotter.canvas, TkAlMap.TkAlMap.default_range, join(), print(), TkAlMap.TkAlMap.tracker, TkAlMap.TkAlMap.two_sigma_cap, and TkAlMap.TkAlMap.var.

Referenced by SpecificationBuilder_cfi.Specification.saveAll().

    def save(self, out_dir='.', extension='pdf'):
        name = '_'.join(['TkAlMap', self.tracker, self.var])
        if self.two_sigma_cap and not self.default_range:
            name += '_4sig'
        elif self.default_range:
            name += '_drange'
        path = out_dir + '/' + name + '.' + extension
        print('TkAlMap: saving canvas in "'+path+'"')
        self.canvas.SaveAs(path)

set_canvas()

def TkAlMap.TkAlMap.set_canvas

(

self

)

Definition at line 207 of file TkAlMap.py.

References TkAlMap.TkAlMap.tracker, TkAlMap.TkAlMap.two_sigma_cap, and TkAlMap.TkAlMap.var.

    def set_canvas(self):
        canv_name = 'canvas_'+self.tracker+'_'+self.var
        if self.two_sigma_cap: canv_name += '_cap'
        self.canvas = ROOT.TCanvas(canv_name, 'TkAlMap '+self.var+' canvas', self.width, self.height)
        print('Actual w: '+str(self.canvas.GetWw())+', Actual h: '+str(self.canvas.GetWh()))

set_colorbar_axis()

def TkAlMap.TkAlMap.set_colorbar_axis

(

self

)

Definition at line 331 of file TkAlMap.py.

References TkAlMap.TkAlMap.image_x1, TkAlMap.TkAlMap.image_x2, and print().

Referenced by TkAlMap.TkAlMap.analyse().

    def set_colorbar_axis(self):
        print('TkAlMap: setting color bar axis')
        b_x1 = self.image_x1
        b_x2 = self.image_x2
        b_y1 = 0.06
        b_y2 = 0.06
        b_width = 0.01
        self.color_bar_axis = ROOT.TGaxis(b_x1, b_y1, b_x2, b_y2, self.min_val, self.max_val, 50510, '+S')
        self.color_bar_axis.SetName('color_bar_axis')
        self.color_bar_axis.SetLabelSize(0.02)
        self.color_bar_axis.SetTickSize(0.01)
        if self.two_sigma_cap and not self.default_range: self.color_bar_axis.SetTitle('{#mu - 2#sigma #leq '+self.var_name+' #leq #mu + 2#sigma} ['+self.var_units+']')
        elif self.default_range: self.color_bar_axis.SetTitle('{'+str(self.min_val)+' #leq '+self.var_name+' #leq '+str(self.max_val)+'} ['+self.var_units+']')
        else: self.color_bar_axis.SetTitle(self.var_name+' ['+self.var_units+']')
        self.color_bar_axis.SetTitleSize(0.025)

set_colorbar_colors()

def TkAlMap.TkAlMap.set_colorbar_colors

(

self

)

Definition at line 347 of file TkAlMap.py.

References TkAlMap.TkAlMap.image_x1, TkAlMap.TkAlMap.image_x2, TkAlMap.TkAlMap.max_val, TkAlMap.TkAlMap.min_val, TkAlMap.TkAlMap.n_color_color_bar, and print().

Referenced by TkAlMap.TkAlMap.analyse().

    def set_colorbar_colors(self):
        print('TkAlMap: initialize color bar colors')
        if self.max_val is None or self.min_val is None:
            col_step = 1./(self.n_color_color_bar + 0.)
            val = col_step/2.
        else:
            b_range = self.max_val - self.min_val
            col_step = (b_range + 0.)/(self.n_color_color_bar + 0.)
            val = self.min_val + col_step/2.

        b_x1 = self.image_x1
        b_x2 = self.image_x2
        b_y1 = 0.06
        b_y2 = 0.06
        b_width = 0.01
        b_xrange = b_x2 - b_x1
        b_yrange = b_y2 - b_y1
        b_dx = (b_xrange + 0.)/(self.n_color_color_bar + 0.)
        b_dy = (b_yrange + 0.)/(self.n_color_color_bar + 0.)

        self.color_bar = {}
        x1 = b_x1
        y1 = b_y1
            
        col_idx = self.start_color_idx
        for i_c in range(self.n_color_color_bar):
            col_idx += 1
            r, g, b = self.get_color_rgb(val)
            try:
                col = ROOT.gROOT.GetColor(col_idx)
                col.SetRGB((r+0.)/255., (g+0.)/255., (b+0.)/255.)
                self.color_bar_colors[col_idx] = col
            except:
                self.color_bar_colors[col_idx] = ROOT.TColor(col_idx, (r+0.)/255., (g+0.)/255., (b+0.)/255.)
            x2 = x1 + b_dx 
            y2 = y1 + b_dy + b_width 
            x = array('d', [x1, x1, x2, x2])
            y = array('d', [y1, y2, y2, y1])
            self.color_bar[col_idx] = ROOT.TPolyLine(len(x), x, y)
            self.color_bar[col_idx].SetFillColor(col_idx)
            self.color_bar[col_idx].SetLineColor(col_idx)
 
            x1 += b_dx
            y1 += b_dy
            val += col_step

set_palette()

def TkAlMap.TkAlMap.set_palette	(		self,
			palette
	)

Definition at line 252 of file TkAlMap.py.

    def set_palette(self, palette):
        self.palette = palette
        pal_str = 'TkAlMap: setting the palette to '+str(palette)
        if palette == 1: pal_str += ' (rainbow)'
        elif palette == 2: pal_str += ' (B->R diverging)'
        else: raise ValueError('TkAlMap: unkown palette value '+str(palette)+', allowed values are 1 and 2')
        print(pal_str)
        #ROOT.gstyle.SetPalette(len(self.colors), self.colors)
        #ROOT.gStyle.SetPalette(len(self.palette), self.palette)
        pass

set_var()

def TkAlMap.TkAlMap.set_var	(		self,
			var,
			var_range = [None,
			None
	)

Definition at line 191 of file TkAlMap.py.

References print().

    def set_var(self, var, var_range=[None, None]):
        print('TkAlMap: setting variable to '+var)
        self.var = var
        self.var_name = var
        self.var_units = 'cm'
        self.var_scale = 1.
        self.var_min = var_range[0]
        self.var_max = var_range[1]
        if var in KNOWN_VARIABLES:
            self.var_name  = KNOWN_VARIABLES[var]['name']
            self.var_units = KNOWN_VARIABLES[var]['units']
            self.var_scale = KNOWN_VARIABLES[var]['scale']
            if self.var_min is None: self.var_min = KNOWN_VARIABLES[var]['range'][0]
            if self.var_max is None: self.var_max = KNOWN_VARIABLES[var]['range'][1]
        self.set_canvas()

setup_colors()

def TkAlMap.TkAlMap.setup_colors

(

self

)

Color setup.

Definition at line 214 of file TkAlMap.py.

References TkAlMap.TkAlMap.fill_colors(), TkAlMap.TkAlMap.load_var(), and TkAlMap.TkAlMap.prepare_map_colors().

Referenced by TkAlMap.TkAlMap.analyse().

    def setup_colors(self):
        self.load_var()
        self.prepare_map_colors()
        self.fill_colors()

Member Data Documentation

canvas

TkAlMap.TkAlMap.canvas

Definition at line 210 of file TkAlMap.py.

Referenced by DisplayManager.DisplayManager.__del__(), DisplayManager.DisplayManager.Draw(), TkAlMap.TkAlMap.draw_cms_prelim(), TkAlMap.TkAlMap.draw_color_bar(), TkAlMap.TkAlMap.draw_event_info(), TkAlMap.TkAlMap.draw_text(), TkAlMap.TkAlMap.draw_title(), TkAlMap.TkAlMap.draw_TPL(), TkAlMap.TkAlMap.load_geometry(), and TkAlMap.TkAlMap.save().

cfg_path

TkAlMap.TkAlMap.cfg_path

Definition at line 177 of file TkAlMap.py.

Referenced by TkAlMap.TkAlMap.load_geometry().

col_dic

TkAlMap.TkAlMap.col_dic

Definition at line 225 of file TkAlMap.py.

color_bar

TkAlMap.TkAlMap.color_bar

Definition at line 367 of file TkAlMap.py.

Referenced by TkAlMap.TkAlMap.draw_color_bar().

color_bar_axis

TkAlMap.TkAlMap.color_bar_axis

Definition at line 338 of file TkAlMap.py.

Referenced by TkAlMap.TkAlMap.draw_color_bar().

color_bar_colors

TkAlMap.TkAlMap.color_bar_colors

Definition at line 171 of file TkAlMap.py.

colors

TkAlMap.TkAlMap.colors

Definition at line 222 of file TkAlMap.py.

data_path

TkAlMap.TkAlMap.data_path

Definition at line 176 of file TkAlMap.py.

Referenced by TkAlMap.TkAlMap.load_geometry().

default_range

TkAlMap.TkAlMap.default_range

Definition at line 163 of file TkAlMap.py.

Referenced by TkAlMap.TkAlMap.save().

do_tanh

TkAlMap.TkAlMap.do_tanh

Definition at line 166 of file TkAlMap.py.

Referenced by TkAlMap.TkAlMap.analyse(), and TkAlMap.TkAlMap.get_color_rgb().

GEO_file

TkAlMap.TkAlMap.GEO_file

tree = r_file.Get(tree_name) tree_tmp = r_file.Get(tree_name) tree = copy.deepcopy(tree_tmp) r_file.Close()

Definition at line 158 of file TkAlMap.py.

Referenced by TkAlMap.TkAlMap.detect_tracker_version(), and TkAlMap.TkAlMap.load_geometry().

height

TkAlMap.TkAlMap.height

Definition at line 161 of file TkAlMap.py.

Referenced by TkAlMap.TkAlMap.draw_cms_prelim(), and TkAlMap.TkAlMap.load_geometry().

image_x1

TkAlMap.TkAlMap.image_x1

Definition at line 593 of file TkAlMap.py.

Referenced by TkAlMap.TkAlMap.draw_cms_prelim(), TkAlMap.TkAlMap.draw_title(), TkAlMap.TkAlMap.set_colorbar_axis(), and TkAlMap.TkAlMap.set_colorbar_colors().

image_x2

TkAlMap.TkAlMap.image_x2

Definition at line 594 of file TkAlMap.py.

Referenced by TkAlMap.TkAlMap.draw_event_info(), TkAlMap.TkAlMap.set_colorbar_axis(), and TkAlMap.TkAlMap.set_colorbar_colors().

image_y1

TkAlMap.TkAlMap.image_y1

Definition at line 595 of file TkAlMap.py.

image_y2

TkAlMap.TkAlMap.image_y2

Definition at line 596 of file TkAlMap.py.

is_cleaned

TkAlMap.TkAlMap.is_cleaned

Definition at line 173 of file TkAlMap.py.

Referenced by TkAlMap.TkAlMap.clean_up().

max_val

TkAlMap.TkAlMap.max_val

Definition at line 169 of file TkAlMap.py.

Referenced by TkAlMap.TkAlMap.fill_colors(), TkAlMap.TkAlMap.get_color_rgb(), and TkAlMap.TkAlMap.set_colorbar_colors().

mean_val

TkAlMap.TkAlMap.mean_val

Definition at line 445 of file TkAlMap.py.

Referenced by TkAlMap.TkAlMap.get_color_rgb(), and TkAlMap.TkAlMap.plot_variable_distribution().

min_val

TkAlMap.TkAlMap.min_val

Definition at line 170 of file TkAlMap.py.

Referenced by TkAlMap.TkAlMap.fill_colors(), TkAlMap.TkAlMap.get_color_rgb(), and TkAlMap.TkAlMap.set_colorbar_colors().

mod_val_dict

TkAlMap.TkAlMap.mod_val_dict

Definition at line 424 of file TkAlMap.py.

Referenced by TkAlMap.TkAlMap.fill_colors().

n_color_color_bar

TkAlMap.TkAlMap.n_color_color_bar

Definition at line 181 of file TkAlMap.py.

Referenced by TkAlMap.TkAlMap.set_colorbar_colors().

palette

TkAlMap.TkAlMap.palette

Definition at line 253 of file TkAlMap.py.

Referenced by TkAlMap.TkAlMap.get_color_rgb().

rgb_map

TkAlMap.TkAlMap.rgb_map

Definition at line 226 of file TkAlMap.py.

Referenced by TkAlMap.TkAlMap.fill_colors().

root_file

TkAlMap.TkAlMap.root_file

Definition at line 165 of file TkAlMap.py.

Referenced by TkAlMap.TkAlMap.load_tree().

start_color_idx

TkAlMap.TkAlMap.start_color_idx

Definition at line 180 of file TkAlMap.py.

std_val

TkAlMap.TkAlMap.std_val

Definition at line 446 of file TkAlMap.py.

Referenced by TkAlMap.TkAlMap.get_color_rgb(), and TkAlMap.TkAlMap.plot_variable_distribution().

title

TkAlMap.TkAlMap.title

Definition at line 162 of file TkAlMap.py.

Referenced by python.rootplot.root2matplotlib.HistStack.bar(), python.rootplot.root2matplotlib.HistStack.bar3d(), python.rootplot.root2matplotlib.HistStack.barcluster(), python.rootplot.root2matplotlib.HistStack.barh(), python.rootplot.root2matplotlib.HistStack.barstack(), MainPageGenerator.MainPageGenerator.CreateNewMainPage(), TkAlMap.TkAlMap.draw_title(), python.rootplot.root2matplotlib.HistStack.errorbar(), python.rootplot.root2matplotlib.HistStack.errorbarh(), Formatter.SimpleHTMLFormatter.headers(), python.rootplot.root2matplotlib.HistStack.histstack(), python.rootplot.root2matplotlib.Hist.show_titles(), python.rootplot.utilities.Hist.TGraph(), python.rootplot.utilities.Hist.TH1F(), python.rootplot.utilities.Hist2D.TH2F(), and python.rootplot.root2matplotlib.Hist2D.TH2F().

TkAlMap_text_dict

TkAlMap.TkAlMap.TkAlMap_text_dict

Definition at line 603 of file TkAlMap.py.

Referenced by TkAlMap.TkAlMap.draw_text().

TkAlMap_TPL_dict

TkAlMap.TkAlMap.TkAlMap_TPL_dict

Definition at line 575 of file TkAlMap.py.

Referenced by TkAlMap.TkAlMap.draw_TPL(), and TkAlMap.TkAlMap.fill_colors().

tmp_file

TkAlMap.TkAlMap.tmp_file

Definition at line 403 of file TkAlMap.py.

Referenced by TkAlMap.TkAlMap.clean_up().

tmp_file_name

TkAlMap.TkAlMap.tmp_file_name

Definition at line 402 of file TkAlMap.py.

Referenced by TkAlMap.TkAlMap.clean_up().

tracker

TkAlMap.TkAlMap.tracker

Definition at line 159 of file TkAlMap.py.

Referenced by TkAlMap.TkAlMap.load_geometry(), TkAlMap.TkAlMap.plot_variable_distribution(), TkAlMap.TkAlMap.save(), and TkAlMap.TkAlMap.set_canvas().

tree

TkAlMap.TkAlMap.tree

Definition at line 172 of file TkAlMap.py.

Referenced by eventstfile.Events.__iter__(), svgfig.SVG.__str__(), ZJetsTreeAnalyzer.ZJetsTreeAnalyzer.beginLoop(), tree.Tree.branch_(), MetTreeProducer.MetTreeProducer.declareVariables(), core.AutoFillTreeProducer.AutoFillTreeProducer.declareVariables(), TkAlMap.TkAlMap.detect_tracker_version(), core.AutoFillTreeProducer.AutoFillTreeProducer.fillTree(), SimpleTreeProducer.SimpleTreeProducer.process(), ZJetsTreeAnalyzer.ZJetsTreeAnalyzer.process(), MetTreeProducer.MetTreeProducer.process(), python.cmstools.EventTree.SetAlias(), eventstfile.Events.size(), eventstfile.Events.to(), tree.Tree.var(), and tree.Tree.vector().

two_sigma_cap

TkAlMap.TkAlMap.two_sigma_cap

Definition at line 164 of file TkAlMap.py.

Referenced by TkAlMap.TkAlMap.plot_variable_distribution(), TkAlMap.TkAlMap.save(), and TkAlMap.TkAlMap.set_canvas().

val_list

TkAlMap.TkAlMap.val_list

else: if 'full' in self.tracker: print('Warning: Unknown module '+str(module)) r_file.Close()

Definition at line 425 of file TkAlMap.py.

Referenced by TkAlMap.TkAlMap.plot_variable_distribution().

var

TkAlMap.TkAlMap.var

Definition at line 193 of file TkAlMap.py.

Referenced by tree.Tree.copyStructure(), TkAlMap.TkAlMap.plot_variable_distribution(), TkAlMap.TkAlMap.save(), and TkAlMap.TkAlMap.set_canvas().

var_max

TkAlMap.TkAlMap.var_max

Definition at line 198 of file TkAlMap.py.

var_min

TkAlMap.TkAlMap.var_min

Definition at line 197 of file TkAlMap.py.

var_name

TkAlMap.TkAlMap.var_name

Definition at line 194 of file TkAlMap.py.

Referenced by TkAlMap.TkAlMap.plot_variable_distribution().

var_scale

TkAlMap.TkAlMap.var_scale

Definition at line 196 of file TkAlMap.py.

var_units

TkAlMap.TkAlMap.var_units

Definition at line 195 of file TkAlMap.py.

Referenced by TkAlMap.TkAlMap.plot_variable_distribution().

width

TkAlMap.TkAlMap.width

Definition at line 160 of file TkAlMap.py.

Referenced by python.rootplot.utilities.Hist.__init_TGraph(), python.rootplot.root2matplotlib.Hist.bar(), python.rootplot.root2matplotlib.Hist.barh(), python.rootplot.utilities.Hist.delete_bin(), TkAlMap.TkAlMap.draw_cms_prelim(), and TkAlMap.TkAlMap.load_geometry().

x_scale

TkAlMap.TkAlMap.x_scale

Definition at line 598 of file TkAlMap.py.

y_scale

TkAlMap.TkAlMap.y_scale

Definition at line 599 of file TkAlMap.py.

Referenced by TkAlMap.TkAlMap.draw_cms_prelim(), and TkAlMap.TkAlMap.draw_title().

---

Generated for CMSSW Reference Manual by  1.8.14