TkAlMapDesign_phase1_cfg Namespace Reference

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

read_TPLfile()

def TkAlMapDesign_phase1_cfg.read_TPLfile

(

file_name

)

Definition at line 419 of file TkAlMapDesign_phase1_cfg.py.

References ALCARECOEcalPhiSym_cff.float, createfilelist.int, and submitPVValidationJobs.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
        
    

test_draw()

def TkAlMapDesign_phase1_cfg.test_draw	(		w,
			h
	)

Definition at line 454 of file TkAlMapDesign_phase1_cfg.py.

References mps_check.array, WZElectronSkims53X_cff.max, SiStripPI.min, isotrackApplyRegressor.range, 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')
   

Variable Documentation

M_T

int TkAlMapDesign_phase1_cfg.M_T = .2

Definition at line 4 of file TkAlMapDesign_phase1_cfg.py.

M_X

int TkAlMapDesign_phase1_cfg.M_X = 1.

Definition at line 2 of file TkAlMapDesign_phase1_cfg.py.

M_Y

float TkAlMapDesign_phase1_cfg.M_Y = 2.5

Definition at line 3 of file TkAlMapDesign_phase1_cfg.py.

PIX_D_W

int TkAlMapDesign_phase1_cfg.PIX_D_W = 10.

Definition at line 6 of file TkAlMapDesign_phase1_cfg.py.

PIX_L_H

int TkAlMapDesign_phase1_cfg.PIX_L_H = 10.

Definition at line 8 of file TkAlMapDesign_phase1_cfg.py.

PIX_L_W

int TkAlMapDesign_phase1_cfg.PIX_L_W = 20.

Definition at line 7 of file TkAlMapDesign_phase1_cfg.py.

STR_D_W

int TkAlMapDesign_phase1_cfg.STR_D_W = 10.

Definition at line 10 of file TkAlMapDesign_phase1_cfg.py.

STR_L_H

int TkAlMapDesign_phase1_cfg.STR_L_H = 12.

Definition at line 13 of file TkAlMapDesign_phase1_cfg.py.

STR_L_W

int TkAlMapDesign_phase1_cfg.STR_L_W = 24.

Definition at line 12 of file TkAlMapDesign_phase1_cfg.py.

STR_TID_D_W

float TkAlMapDesign_phase1_cfg.STR_TID_D_W = 5.5

Definition at line 11 of file TkAlMapDesign_phase1_cfg.py.

TkMap_GEO

dictionary TkAlMapDesign_phase1_cfg.TkMap_GEO

Definition at line 15 of file TkAlMapDesign_phase1_cfg.py.