Functions
def	read_TPLfile (file_name)

def	test_draw (w, h)

Variables
int	M_T = .2

int	M_X = 1

float	M_Y = 2.5

int	PIX_D_W = 10

int	PIX_L_H = 10

int	PIX_L_W = 20

int	STR_D_W = 10

int	STR_L_H = 12

int	STR_L_W = 24

float	STR_TID_D_W = 5.5

dictionary	TkMap_GEO

Function Documentation

def TkAlMapDesign_phase1_cfg.read_TPLfile ( file_name )

Definition at line 419 of file TkAlMapDesign_phase1_cfg.py.

References objects.autophobj.float, createfilelist.int, and split.

Referenced by test_draw().

 def read_TPLfile(file_name):
     o_file = open(file_name, 'r')
     lines = o_file.readlines()
     o_file.close()
 
     TPL_dict = {}
     for line in lines:
         if '#' in line: continue
         splt_line = line.replace('\n', '').split(' ')
         det_id = int(splt_line[0])
         x = []
         y = []
         for idx,coo in enumerate(splt_line[1:]):
             #print(coo)
             try:
                 val = float(coo)
                 if (idx%2) == 0: 
                     y.append(val)
                 else: 
                     x.append(val)
             except ValueError:
                 continue
         TPL_dict[det_id] = {}
         TPL_dict[det_id]['x'] = x
         TPL_dict[det_id]['y'] = y
     return TPL_dict
         
     
 

def TkAlMapDesign_phase1_cfg.test_draw	(	w,
		h
	)

Definition at line 454 of file TkAlMapDesign_phase1_cfg.py.

References mps_check.array, SiStripPI.max, min(), and read_TPLfile().

     def test_draw(w, h):
         canvas = ROOT.TCanvas('canvas', 'detector canvas', w, h)
         
         # Load all modules in absolute positions
         # Find max x and y to rescale later for window
         all_modules = {}
         all_text = {}
         x_max = -9999.
         y_max = -9999.
         x_min = 9999.
         y_min = 9999.
         for det in TkMap_GEO:
             for sub in TkMap_GEO[det]:
                 for part in TkMap_GEO[det][sub]:
                     if part == 'latex':
                         all_text[det+'_'+sub] = TkMap_GEO[det][sub][part]
                         continue
                     if 'latex' in TkMap_GEO[det][sub][part]:
                         all_text[det+'_'+sub+'_'+part] = TkMap_GEO[det][sub][part]['latex']
                     TPL_dict = read_TPLfile(TkMap_GEO[det][sub][part]['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]*TkMap_GEO[det][sub][part]['x_scale'] + TkMap_GEO[det][sub][part]['x_off'])
                             y_canv.append(TPL_dict[module]['y'][idx]*TkMap_GEO[det][sub][part]['y_scale'] + TkMap_GEO[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 = 1
         l_margin = 1
         t_margin = 1
         b_margin = 1
 
         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
         if (x_range + 0.)/(w + 0.) > (y_range + 0.)/(h + 0.):
             x_scale = x_range
             y_scale = (h + 0.)/(w + 0.)*x_range
         else:
             y_scale = y_range
             x_scale = (w + 0.)/(h + 0.)*y_range
         TPL = {} 
         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)
             TPL[module] = ROOT.TPolyLine(len(x), x, y) 
             #TPL[module].SetFillColor(1)
             TPL[module].SetLineColor(1)
             TPL[module].Draw('f')
             TPL[module].Draw()
         TL = ROOT.TLatex()
         TL.SetTextSize(0.025)
         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.)
             TL.SetTextAlign(all_text[key]['alignment'])
             TL.DrawLatex(x, y, all_text[key]['text'])
         #TL.SetTextSize(0.025)
         #TL.SetTextAlign(11)
         #TL.DrawLatex(0.1, 0.1, 'bottom')
         #TL.SetTextAlign(13)
         #TL.DrawLatex(0.1, 0.1, 'top')
         canvas.Update() 
         raw_input('exit')
    