|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
|
import re
from math import *
from .svgfig import rgb, SVG, pathtoPath, load as load_svg
from .geometryXMLparser import *
from signConventions import *
def dt_colors(wheel, station, sector):
return rgb(0.1, 0.9, 0.)
def csc_colors(endcap, station, ring, chamber):
return rgb(0.1, 0.9, 0.)
def draw_station(geom1, geom2, station, filename, length_factor=100., angle_factor=100., colors=dt_colors, template_dir='./'):
if station == 4:
station_template = load_svg(template_dir + "station4_template.svg")
else:
station_template = load_svg(template_dir + "station_template.svg")
if station == 1: x_scale_factor = 1/6.
if station == 2: x_scale_factor = 1/7.
if station == 3: x_scale_factor = 1/8.5
if station == 4: x_scale_factor = 1/10.
y_scale_factor = 1/7.
# make a new group to put the moved chambers into
new_boxes = SVG("g")
# loop over the SVG tree, looking for our chambers (by id)
for treeindex, svgitem in station_template:
if isinstance(svgitem, SVG) and "id" in svgitem.attr and svgitem["id"][:3] == "MB_":
m = re.match("MB_([0-9mpz]+)_([0-9]+)", svgitem["id"])
if m is None: raise Exception
wheel = m.group(1)
if wheel == "m2": wheel = -2
elif wheel == "m1": wheel = -1
elif wheel == "z": wheel = 0
elif wheel == "p1": wheel = 1
elif wheel == "p2": wheel = 2
sector = int(m.group(2))
xdiff = x_scale_factor * length_factor * (geom1.dt[wheel, station, sector].x - geom2.dt[wheel, station, sector].x) * signConventions["DT", wheel, station, sector][0]
ydiff = -y_scale_factor * length_factor * (geom1.dt[wheel, station, sector].y - geom2.dt[wheel, station, sector].y) * signConventions["DT", wheel, station, sector][1]
phizdiff = -angle_factor * (geom1.dt[wheel, station, sector].phiz - geom2.dt[wheel, station, sector].phiz) * signConventions["DT", wheel, station, sector][2]
sx = float(svgitem["x"]) + float(svgitem["width"])/2.
sy = float(svgitem["y"]) + float(svgitem["height"])/2.
svgitem["transform"] = "translate(%g,%g)" % (sx, sy)
svgitem["x"] = -float(svgitem["width"])/2.
svgitem["y"] = -float(svgitem["height"])/2.
svgitem["style"] = "fill:#e1e1e1;fill-opacity:1;stroke:#000000;stroke-width:1.0;stroke-dasharray:1, 1;stroke-dashoffset:0"
newBox = svgitem.clone()
newBox["transform"] = "translate(%g,%g) rotate(%g) " % (sx + xdiff, sy + ydiff, phizdiff * 180./pi)
newBox["style"] = "fill:%s;fill-opacity:0.5;stroke:#000000;stroke-width:1.0;stroke-opacity:1;stroke-dasharray:none" % colors(wheel, station, sector)
new_boxes.append(newBox)
for treeindex, svgitem in station_template:
if isinstance(svgitem, SVG) and svgitem.t == "g" and "id" in svgitem.attr and svgitem["id"] == "chambers":
svgitem.append(new_boxes)
elif isinstance(svgitem, SVG) and "id" in svgitem.attr and svgitem["id"] == "stationx":
svgitem[0] = "Station %d" % station
svgitem[0] += " (length x%g, angle x%g)" % (length_factor, angle_factor)
station_template.save(filename)
def draw_wheel(geom1, geom2, wheel, filename, length_factor=100., angle_factor=100., colors=dt_colors, template_dir='./'):
wheel_template = load_svg(template_dir + "wheel_template.svg")
# make a new group to put the moved chambers into
new_boxes = SVG("g")
# loop over the SVG tree, looking for our chambers (by id)
for treeindex, svgitem in wheel_template:
if isinstance(svgitem, SVG) and "id" in svgitem.attr and svgitem["id"][:3] == "MB_":
m = re.match("MB_([0-9]+)_([0-9]+)", svgitem["id"])
if m is None: raise Exception
station, sector = int(m.group(1)), int(m.group(2))
xdiff = -length_factor * (geom1.dt[wheel, station, sector].x - geom2.dt[wheel, station, sector].x) * signConventions["DT", wheel, station, sector][0]
zdiff = length_factor * (geom1.dt[wheel, station, sector].z - geom2.dt[wheel, station, sector].z) * signConventions["DT", wheel, station, sector][2]
phiydiff = -angle_factor * (geom1.dt[wheel, station, sector].phiy - geom2.dt[wheel, station, sector].phiy) * signConventions["DT", wheel, station, sector][1]
m = re.search("translate\\(([0-9\\.\\-\\+eE]+),\\s([0-9\\.\\-\\+eE]+)\\)\\srotate\\(([0-9\\.\\-\\+eE]+)\\)",svgitem["transform"])
tx = float(m.group(1))
ty = float(m.group(2))
tr = float(m.group(3))
newBox = svgitem.clone()
svgitem["style"] = "fill:#e1e1e1;fill-opacity:1;stroke:#000000;stroke-width:5.0;stroke-dasharray:1, 1;stroke-dashoffset:0"
newBox["style"] = "fill:%s;fill-opacity:0.5;stroke:#000000;stroke-width:5.0;stroke-opacity:1;stroke-dasharray:none" % colors(wheel, station, sector)
newBox["id"] = newBox["id"] + "_moved"
newBox["transform"] = "translate(%g,%g) rotate(%g)" % (tx - xdiff*cos(tr*pi/180.) + zdiff*sin(tr*pi/180.), ty - xdiff*sin(tr*pi/180.) - zdiff*cos(tr*pi/180.), tr - phiydiff*180./pi)
new_boxes.append(newBox)
for treeindex, svgitem in wheel_template:
if isinstance(svgitem, SVG) and svgitem.t == "g" and "id" in svgitem.attr and svgitem["id"] == "chambers":
svgitem.append(new_boxes)
elif isinstance(svgitem, SVG) and "id" in svgitem.attr and svgitem["id"] == "wheelx":
if wheel == 0: svgitem[0] = "Wheel 0"
else: svgitem[0] = "Wheel %+d" % wheel
svgitem[0] += " (length x%g, angle x%g)" % (length_factor, angle_factor)
wheel_template.save(filename)
def draw_disk(geom1, geom2, endcap, station, filename, length_factor=1., angle_factor=100., colors=csc_colors, template_dir='./'):
if station == 1:
disk_template = load_svg(template_dir + "disk1_template.svg")
if station in (2, 3):
disk_template = load_svg(template_dir + "disk23_template.svg")
if endcap == 1 and station == 4:
disk_template = load_svg(template_dir + "diskp4_template.svg")
if endcap == 2 and station == 4:
disk_template = load_svg(template_dir + "diskm4_template.svg")
scale_factor = 0.233
new_boxes = SVG("g")
for treeindex, svgitem in disk_template:
if isinstance(svgitem, SVG) and "id" in svgitem.attr and svgitem["id"] == "fakecenter":
fakecenter = pathtoPath(pathtoPath(svgitem).SVG())
sumx = 0.
sumy = 0.
sum1 = 0.
for i, di in enumerate(fakecenter.d):
if di[0] in ("M", "L"):
sumx += di[1]
sumy += di[2]
sum1 += 1.
originx = sumx/sum1
originy = sumy/sum1
for treeindex, svgitem in disk_template:
if isinstance(svgitem, SVG) and "id" in svgitem.attr and svgitem["id"][:3] == "ME_":
m = re.match("ME_([0-9]+)_([0-9]+)", svgitem["id"])
if m is None: raise Exception
ring, chamber = int(m.group(1)), int(m.group(2))
xdiff = scale_factor * length_factor * (geom1.csc[endcap, station, ring, chamber].x - geom2.csc[endcap, station, ring, chamber].x) * signConventions["CSC", endcap, station, ring, chamber][0]
ydiff = -scale_factor * length_factor * (geom1.csc[endcap, station, ring, chamber].y - geom2.csc[endcap, station, ring, chamber].y) * signConventions["CSC", endcap, station, ring, chamber][1]
phizdiff = -angle_factor * (geom1.csc[endcap, station, ring, chamber].phiz - geom2.csc[endcap, station, ring, chamber].phiz) * signConventions["CSC", endcap, station, ring, chamber][2]
svgitem["style"] = "fill:#e1e1e1;fill-opacity:1;stroke:#000000;stroke-width:1.0;stroke-dasharray:1, 1;stroke-dashoffset:0"
newBox = pathtoPath(svgitem)
# Inkscape outputs wrong SVG: paths are filled with movetos, rather than linetos; this fixes that
first = True
for i, di in enumerate(newBox.d):
if not first and di[0] == "m":
di = list(di)
di[0] = "l"
newBox.d[i] = tuple(di)
first = False
# convert to absolute coordinates
newBox = pathtoPath(newBox.SVG())
# find the center of the object
sumx = 0.
sumy = 0.
sum1 = 0.
for i, di in enumerate(newBox.d):
if di[0] == "L":
sumx += di[1]
sumy += di[2]
sum1 += 1.
centerx = sumx/sum1
centery = sumy/sum1
phipos = atan2(centery - originy, centerx - originx) - pi/2.
for i, di in enumerate(newBox.d):
if di[0] in ("M", "L"):
di = list(di)
di[1] += cos(phipos)*xdiff - sin(phipos)*ydiff
di[2] += sin(phipos)*xdiff + cos(phipos)*ydiff
newBox.d[i] = tuple(di)
centerx += cos(phipos)*xdiff - sin(phipos)*ydiff
centery += sin(phipos)*xdiff + cos(phipos)*ydiff
for i, di in enumerate(newBox.d):
if di[0] in ("M", "L"):
di = list(di)
dispx = cos(phizdiff) * (di[1] - centerx) - sin(phizdiff) * (di[2] - centery)
dispy = sin(phizdiff) * (di[1] - centerx) + cos(phizdiff) * (di[2] - centery)
di[1] = dispx + centerx
di[2] = dispy + centery
newBox.d[i] = tuple(di)
newBox = newBox.SVG()
newBox["style"] = "fill:%s;fill-opacity:0.5;stroke:#000000;stroke-width:1.0;stroke-opacity:1;stroke-dasharray:none" % colors(endcap, station, ring, chamber)
newBox["id"] = newBox["id"] + "_moved"
new_boxes.append(newBox)
for treeindex, svgitem in disk_template:
if isinstance(svgitem, SVG) and svgitem.t == "g" and "id" in svgitem.attr and svgitem["id"] == "chambers":
svgitem.append(new_boxes)
elif isinstance(svgitem, SVG) and "id" in svgitem.attr and svgitem["id"] == "diskx":
if endcap == 1: svgitem[0] = "Disk %+d" % station
else: svgitem[0] = "Disk %+d" % (-station)
svgitem[0] += " (length x%g, angle x%g)" % (length_factor, angle_factor)
disk_template.save(filename)
|