def drawFrontFacePair(topCorners, bottomCorners, color):
pointList = []
pointList2 = []
output = G()
YZStyle1 = StyleBuilder()
#YZStyle1.setStroke('black')
#YZStyle1.setStrokeWidth(1.5)
YZStyle1.setFilling(color)
YZStyle1.setFillOpacity(0.75)
YZStyle2 = StyleBuilder()
YZStyle2.setStroke('black')
YZStyle2.setStrokeWidth(0.5)
YZStyle2.setFillOpacity(0.75)
YZStyle2.setFilling('#ffffff')
pointList.append(strPathStart(topCorners[1]))
pointList.append(strPathPoint(topCorners[2]))
pointList.append(strPathPoint(bottomCorners[2]))
pointList.append(strPathPoint(bottomCorners[1]))
svgCode1 = pGram(pointList)
pointList2.append(strPathStart(topCorners[0]))
pointList2.append(strPathPoint(topCorners[3]))
pointList2.append(strPathPoint(bottomCorners[3]))
pointList2.append(strPathPoint(bottomCorners[0]))
svgCode2 = pGram(pointList2)
frontFace = Path(svgCode1)
frontFace.set_style(YZStyle1.getStyle())
backFace = Path(svgCode2)
backFace.set_style(YZStyle2.getStyle())
output.addElement(backFace)
output.addElement(frontFace)
return output
def drawSideFacePair(topCorners, bottomCorners, color):
pointList = []
pointList2 = []
output = G()
XYStyle1 = StyleBuilder()
#XYStyle1.setStroke('black')
#XYStyle1.setStrokeWidth(1.5)
XYStyle1.setFilling(color)
#XYStyle1.setFillOpacity(0.8)
XYStyle2 = StyleBuilder()
XYStyle2.setStroke(color)
XYStyle2.setStrokeWidth(0.5)
XYStyle2.setFilling(color)
pointList.append(strPathStart(topCorners[2]))
pointList.append(strPathPoint(topCorners[3]))
pointList.append(strPathPoint(bottomCorners[3]))
pointList.append(strPathPoint(bottomCorners[2]))
svgCode1 = pGram(pointList)
pointList2.append(strPathStart(topCorners[0]))
pointList2.append(strPathPoint(topCorners[1]))
pointList2.append(strPathPoint(bottomCorners[1]))
pointList2.append(strPathPoint(bottomCorners[0]))
svgCode2 = pGram(pointList2)
rightFace = Path(svgCode1)
rightFace.set_style(XYStyle1.getStyle())
leftFace = Path(svgCode2)
leftFace.set_style(XYStyle2.getStyle())
output.addElement(leftFace)
output.addElement(rightFace)
return output
def _get_font_style(self, size, anchor, dash_array):
style = StyleBuilder()
style.setStrokeDashArray(dash_array)
style.setFontFamily(fontfamily="Verdana")
style.setFontSize("%dpx" % size)
style.setTextAnchor(anchor)
return style
def rev_path(x1, y1, x2, y2, txt=None):
elements = []
# x1, y1 = r1.get_cx(), r1.get_cy()
# x2, y2 = r2.get_cx(), r2.get_cy(),
l = line(x1, y1, x2, y2)
elements.append(l)
style = 'stroke-width:{0};stroke:{1}'.format(16, ALUM_6)
l.set_style(style)
if txt:
x, y = middle(x1, y1, x2, y2)
style2 = StyleBuilder()
#style2.setFontFamily('envy code r')
style2.setFontFamily('arial')
style2.setFontWeight('bold')
style2.setFilling(ALUM_1)
# shift text left and up by a bit
# whole alphabet in this font is 167 px width
per_char = 167. / 26
t = text(txt, -len(txt) / 2 * per_char, 4)
t.set_style(style2.getStyle())
#import pdb; pdb.set_trace()
group = rotate([t], slope_angle(x1, y1, x2, y2))
group = translate([group], x, y)
elements.append(group)
return elements
def genHoneycomb(self):
outComb = G()
circleStyle = StyleBuilder()
#circleStyle.setStrokeWidth(0.5)
circleStyle.setStroke('orange')
circleStyle.setFilling('#edd239')
CORNER = self.hCSlice.getBottomRight()
circleY = []
circleY.append(CORNER[1] + (ROW_SPACE - CIRCLE_RADIUS))
for g in range(0, self.row):
circleX = CORNER[0] - CIRCLE_RADIUS
if (g != 0 and g % 2 != 0):
circleY.append(circleY[g-1] + 2*CIRCLE_RADIUS)
elif (g!= 0 and g % 2 == 0):
circleY.append(circleY[g-1] + 2*(ROW_SPACE - CIRCLE_RADIUS))
#initCircle = Circle(circleX, circleY[g], CIRCLE_RADIUS)
#outComb.addElement(initCircle)
for i in range(0, self.col):
newShift = TransformBuilder()
xShift = -i*CIRCLE_RADIUS*math.sqrt(3)
yShift = 0
if i % 2 != 0 and g % 2 != 0:
yShift = CIRCLE_RADIUS
elif i % 2 != 0 and g % 2 == 0:
yShift = -1*CIRCLE_RADIUS
newShift.setTranslation(str(xShift) + ' ' + str(yShift))
a = Circle(circleX, circleY[g], CIRCLE_RADIUS)
a.set_transform(newShift.getTransform())
outComb.addElement(a)
outComb.set_style(circleStyle.getStyle())
return outComb
def draw(self):
rectangleStyle = StyleBuilder()
rectangleStyle.setStroke('black')
rectangleStyle.setStrokeWidth(2)
rectangleStyle.setFilling(self.color)
self.view = Rect(self.gridCoord[0], self.gridCoord[1], self.dimA*SCALING_FACTOR, self.dimB*SCALING_FACTOR)
self.view.set_style(rectangleStyle.getStyle())
return self.view
def words(txt, family='arial', weight='bold', color=ALUM_1):
style2 = StyleBuilder()
#style2.setFontFamily('envy code r')
style2.setFontFamily('arial')
style2.setFontWeight('bold')
style2.setFilling(ALUM_1)
t = text(txt, 0, 0)
t.set_style(style2.getStyle())
return t
def image(self):
style_line = StyleBuilder()
style_line.setStrokeWidth(self.dim_x / 100)
return pysvg.shape.rect(self.pos_x - self.dim_x / 2,
self.pos_y - self.dim_y / 2,
self.dim_x,
self.dim_y,
stroke='black',
fill='white',
style=style_line.getStyle())
def command(num_txt, cmd, explanation, color):
x = 0
y = 0
elems = [scale(num(num_txt, color), 2)]
cmd_txt = text(cmd, x + 40, y + 12)
s = StyleBuilder()
s.setFontWeight('bold')
s.setFontFamily('Bitstream Vera Sans Mono')
cmd_txt.set_style(s.getStyle())
elems.append(cmd_txt)
exp_txt = text(explanation, x + 45, y + 27)
s = StyleBuilder()
#s.setFontWeight('bold')
s.setFontFamily('Bitstream Vera Serif')
s.setFontSize('10px')
exp_txt.set_style(s.getStyle())
elems.append(exp_txt)
return elems
def genLabels(self):
labels = G()
labelStyle = StyleBuilder()
labelStyle.setTextAnchor('middle')
labelStyle.setFontSize(10)
lineStyle = StyleBuilder()
lineStyle.setStroke('#a9a9a9')
dimLabelA = str(self.dimA) + ' nm'
dimLabelB = str(self.dimB) + ' nm'
# horizontal label
cornerHL = self.view.getTopLeft()
cornerHR = self.view.getTopRight()
hLabelStart = [cornerHL[0], cornerHL[1] + 5]
hLabelEnd = [cornerHR[0], cornerHR[1] + 5]
hLine = Line(hLabelStart[0], hLabelStart[1], hLabelEnd[0], hLabelEnd[1])
hLabel = Text(dimLabelA, (hLabelStart[0] + hLabelEnd[0])/2, hLabelEnd[1] + 20)
hLine.set_style(lineStyle.getStyle())
hLabel.set_style(labelStyle.getStyle())
cornerVL = self.view.getTopLeft()
cornerVR = self.view.getBottomLeft()
vLabelStart = [cornerVL[0] - 5, cornerVL[1]]
vLabelEnd = [cornerVR[0] - 5, cornerVR[1]]
vLine = Line(vLabelStart[0], vLabelStart[1], vLabelEnd[0], vLabelEnd[1])
vLabel = Text(dimLabelB, vLabelEnd[0] - 30, (vLabelStart[1] + vLabelEnd[1])/2)
vLine.set_style(lineStyle.getStyle())
vLabel.set_style(labelStyle.getStyle())
labels.addElement(hLine)
labels.addElement(hLabel)
labels.addElement(vLine)
labels.addElement(vLabel)
#center = [self.gridCoord[0] + self.dimA*SCALING_FACTOR/2, self.gridCoord[1] + self.dimB*SCALING_FACTOR + LABEL_SHIFT]
#dimLabel = str(self.dimA) + 'nm x ' +str(self.dimB) + 'nm'
#labels = Text(dimLabel, center[0], center[1])
#labels.set_style(labelStyle.getStyle())
return labels
def Scalebars(smin, smax, xmin, xmax):
c_lines = StyleBuilder(config['scaleline_style'])
c_times = StyleBuilder(config['scale_style'])
c_times.setFontSize(str(config['scale_font_height']) + 'px')
g_scale = G()
g_scale_lines = G()
g_scale_times = G()
g_scale_lines.set_style(c_lines.getStyle())
g_scale_times.set_style(c_times.getStyle())
g_scale_times.setAttribute('xml:space', 'preserve')
# seconds value of largest minute smaller than first time
start = (int(smin) / 60) * 60
# seconds value of smallest minute larger than last time
end = 60 + ((int(smax) / 60) * 60)
# if the max time wasn't exactly on a minute interval, add one more
if end < smax + 60:
end += 60;
# one scale bar for each minute in the range, including last
for s in range(start, end, 60):
y = config['vscale'] * (s - smin)
sline = Line(xmin, y, xmax, y)
if config['scaleleft']:
# note hard coded expectation of 7 char max scale label
lx = xmin - 5 - (7 * config['scale_font_width'])
else:
lx = xmax + 5
stime = Text(time(s), lx, y + 6)
g_scale_lines.addElement(sline)
g_scale_times.addElement(stime)
g_scale.addElement(g_scale_lines)
g_scale.addElement(g_scale_times)
return g_scale
def num(txt, color):
txt = str(txt)
elems = []
r = rect(0, 0, 16, 16)
s = StyleBuilder()
s.setFilling(color)
r.set_style(s.getStyle())
elems.append(r)
if txt:
style2 = StyleBuilder()
#style2.setFontFamily('envy code r')
style2.setFontFamily('arial')
style2.setFontWeight('bold')
style2.setFilling(ALUM_1)
# shift text left and up by a bit
if len(txt) == 1:
x = 5
elif len(txt) == 2:
x = 1.5
t = text(txt, x, 12.5)
t.set_style(style2.getStyle())
elems.append(t)
return elems
def genLabels(self):
labels = G()
labelStyle = StyleBuilder()
labelStyle.setTextAnchor('middle')
labelStyle.setFontSize(10)
lineStyle = StyleBuilder()
lineStyle.setStroke('#a9a9a9')
dimLabelA = str(self.x/SCALING_FACTOR) + ' nm'
dimLabelB = str(self.z/SCALING_FACTOR) + ' nm'
# horizontal label
cornerHL = self.hCSlice.getTopLeft()
cornerHR = self.hCSlice.getTopRight()
hLabelStart = [cornerHL[0], cornerHL[1] + 5]
hLabelEnd = [cornerHR[0], cornerHR[1] + 5]
hLine = Line(hLabelStart[0], hLabelStart[1], hLabelEnd[0], hLabelEnd[1])
hLabel = Text(dimLabelA, (hLabelStart[0] + hLabelEnd[0])/2, hLabelEnd[1] + 20)
hLine.set_style(lineStyle.getStyle())
hLabel.set_style(labelStyle.getStyle())
cornerVL = self.hCSlice.getTopLeft()
cornerVR = self.hCSlice.getBottomLeft()
vLabelStart = [cornerVL[0] - 5, cornerVL[1]]
vLabelEnd = [cornerVR[0] - 5, cornerVR[1]]
vLine = Line(vLabelStart[0], vLabelStart[1], vLabelEnd[0], vLabelEnd[1])
vLabel = Text(dimLabelB, vLabelEnd[0] - 30, (vLabelStart[1] + vLabelEnd[1])/2)
vLine.set_style(lineStyle.getStyle())
vLabel.set_style(labelStyle.getStyle())
labels.addElement(hLine)
labels.addElement(hLabel)
labels.addElement(vLine)
labels.addElement(vLabel)
return labels
def makerow(self):
frameStyle = StyleBuilder()
frameStyle.setStrokeWidth(1.5)
frameStyle.setStroke('black')
frameStyle.setFilling('white')
out = G()
yCoord = 10 + (FRAMELEN + 20) * self.i
for z in range(0, 4):
newFrame = Rect(10 + (FRAMELEN + 20) * z, yCoord, FRAMELEN,
FRAMELEN)
newFrame.set_style(frameStyle.getStyle())
out.addElement(newFrame)
self.rowCoords.append([10 + (FRAMELEN + 20) * z, yCoord])
return out
def draw_tree(width, height):
global nameStyle
nameStyle = StyleBuilder()
nameStyle.setFontFamily(fontfamily=dc.nameFont)
nameStyle.setFontSize("%spt" % dc.nameFontSize)
nameStyle.setTextAnchor("left")
nameStyle = nameStyle.getStyle()
svg = Svg(width=width, height=height)
# draw nodes
for depth in depthToNames:
for name in depthToNames[depth]:
node = nameToNode[name]
draw_node(svg, node, name)
# draw branches
for depth in depthToNames:
for name in depthToNames[depth]:
node = nameToNode[name]
isLeaf = (node.children == [])
if (isLeaf): continue
numChildren = len(node.children)
for (i, child) in enumerate(node.children):
rootFrac = 0.4 + (0.2 * i) / (numChildren - 1)
sinkFrac = 0.5
if (orientation == "T2B"):
(rootX, rootY) = (node.x + rootFrac * dc.nodeWdt,
node.y + dc.nodeHgt)
(sinkX, sinkY) = (child.x + sinkFrac * dc.nodeWdt, child.y)
draw_vert_branch(svg, "%s_branch_%d" % (name, i), rootX,
rootY, sinkX, sinkY)
else: # if (orientation == "L2R"):
(rootX, rootY) = (node.x + dc.nodeWdt,
node.y + rootFrac * dc.nodeHgt)
(sinkX, sinkY) = (child.x, child.y + sinkFrac * dc.nodeHgt)
draw_horz_branch(svg, "%s_branch_%d" % (name, i), rootX,
rootY, sinkX, sinkY)
return svg
def image(self):
style_line = StyleBuilder()
style_line.setStrokeWidth(self.type.dim_x / 100)
if self.type.shape == 'ellipse':
return pysvg.shape.ellipse(self.pos_x,
self.pos_y,
self.type.dim_x / 2,
self.type.dim_y / 2,
stroke='black',
fill=self.type.color,
style=style_line.getStyle())
elif self.type.shape == 'rectangle':
return pysvg.shape.rect(self.pos_x - self.type.dim_x / 2,
self.pos_y - self.type.dim_y / 2,
self.type.dim_x,
self.type.dim_y,
stroke='black',
fill=self.type.color,
style=style_line.getStyle())
if config['pagewidth'] != None:
actual_page_width = config['pagewidth']
total_label_width = 0
for r in races:
total_label_width += r['wmax_label'] * config['label_font_width']
while total_label_width > actual_page_width:
actual_page_width += config['pagewidth']
gap_count = len(races) - 1
scale_label_width = ((7 * config['scale_font_width']) + 5 if config['scalebars'] else 0)
free_space = actual_page_width - total_label_width - scale_label_width - (gap_count * 2 * config['gutter'])
calc_linespan = free_space / gap_count
config['linespan'] = calc_linespan
# SVG styles
s_label = StyleBuilder(config['label_style'])
s_label.setFontSize(str(config['label_font_height']) + 'px')
# SVG Groups
g_label = G()
g_label.set_style(s_label.getStyle())
g_label.setAttribute('xml:space', 'preserve')
g_linkline = G()
g_linkline.set_style(StyleBuilder(config['linkline_style']).getStyle())
g_weaklink = G()
g_weaklink.set_style(StyleBuilder(config['weaklink_style']).getStyle())
g_underline = G()
g_underline.set_style(StyleBuilder(config['underline_style']).getStyle())
for r in range(0, len(races)):
def genHoneyRect(self):
rectangleStyle = StyleBuilder()
rectangleStyle.setFilling('blue')
rectangleStyle.setStroke('black') # temp
self.hCSlice.set_style(rectangleStyle.getStyle())
return(self.hCSlice)
def draw(self):
faceStyle = StyleBuilder()
faceStyle.setFilling('blue')
faceStyle.setStroke('black')
faceStyle.setFillOpacity(0.5)
topStyle = StyleBuilder()
topStyle.setFilling('green')
topStyle.setStroke('black')
topStyle.setStrokeWidth(2)
topStyle.setFillOpacity(0.5)
sideStyle = StyleBuilder()
sideStyle.setFilling('red')
sideStyle.setStroke('black')
sideStyle.setStrokeWidth(2)
sideStyle.setFillOpacity(0.5)
frontFace = Rect(self.startCoord[0], self.startCoord[1], self.x, self.z)
backFace = Rect(self.startCoord[0] + self.y/math.sqrt(2), self.startCoord[1] - self.y/math.sqrt(2), self.x, self.z)
# draw parallelogram, grab corners
cornerBack = backFace.getEdgePoints()
cornerFront = frontFace.getEdgePoints()
# Top face string
pointListT = []
pointListT.append(strPathStart(cornerBack[0]))
pointListT.append(strPathPoint(cornerBack[1]))
pointListT.append(strPathPoint(cornerFront[1]))
pointListT.append(strPathPoint(cornerFront[0]))
topFaceString = pGram(pointListT)
# right side face
pointListR = [strPathStart(cornerBack[1])]
pointListR.append(strPathPoint(cornerBack[2]))
pointListR.append(strPathPoint(cornerFront[2]))
pointListR.append(strPathPoint(cornerFront[1]))
# bottom face
pointListB = [strPathStart(cornerBack[2])]
pointListB.append(strPathPoint(cornerBack[3]))
pointListB.append(strPathPoint(cornerFront[3]))
pointListB.append(strPathPoint(cornerFront[2]))
# left
pointListL = [strPathStart(cornerBack[0])]
pointListL.append(strPathPoint(cornerBack[3]))
pointListL.append(strPathPoint(cornerFront[3]))
pointListL.append(strPathPoint(cornerFront[0]))
# make faces
topFace = Path(pGram(pointListT))
rightFace = Path(pGram(pointListR))
leftFace = Path(pGram(pointListL))
bottomFace = Path(pGram(pointListB))
topFace.set_style(topStyle.getStyle())
bottomFace.set_style(topStyle.getStyle())
rightFace.set_style(sideStyle.getStyle())
leftFace.set_style(sideStyle.getStyle())
frontFace.set_style(faceStyle.getStyle())
backFace.set_style(faceStyle.getStyle())
self.out.addElement(backFace)
self.out.addElement(frontFace)
self.out.addElement(rightFace)
self.out.addElement(leftFace)
self.out.addElement(bottomFace)
self.out.addElement(topFace)
return self.out
def draw_tree():
if (dc.nameFontSize < 16):
scale = dc.nameFontSize / 16.0
nameCapsHgt = dc.nameCapsHgt * scale
nameDescHgt = dc.nameDescHgt * scale
nameFontLineHgt = dc.nameFontLineHgt * scale
nameStyle = StyleBuilder()
nameStyle.setFontFamily(fontfamily=dc.nameFont)
nameStyle.setFontSize("%spt" % dc.nameFontSize)
nameStyle.setTextAnchor("left")
nameStyle = nameStyle.getStyle()
svg = Svg()
# draw nodes
for depth in depthToNames:
for name in depthToNames[depth]:
node = nameToNode[name]
isLeaf = (node.left == None)
yLine = node.y + nameCapsHgt + 1
ob = SvgRect(node.x,
node.y,
dc.nodeWidth,
dc.nodeHeight,
id="%s_box" % name)
ob.set_stroke(dc.lineColor)
ob.set_stroke_width(dc.lineThickness)
if (isLeaf): ob.set_fill(dc.leafFillColor)
else: ob.set_fill(dc.nodeFillColor)
svg.addElement(ob)
ob = SvgText("%s" % name, node.x + 1, yLine, id="%s_name" % node)
ob.set_style(nameStyle)
svg.addElement(ob)
yLine += nameFontLineHgt
if (hasattr(node, "bitsUnion")):
ob = SvgText("U:" +
bits_to_string(node.numBits, node.bitsUnion),
node.x + 1,
yLine,
id="%s_Bunion" % node)
ob.set_style(nameStyle)
svg.addElement(ob)
yLine += nameFontLineHgt
if (hasattr(node, "bitsIntersection")):
ob = SvgText(
"I:" + bits_to_string(node.numBits, node.bitsIntersection),
node.x + 1,
yLine,
id="%s_Bintersection" % node)
ob.set_style(nameStyle)
svg.addElement(ob)
yLine += nameFontLineHgt
if (hasattr(node, "bitsAll")):
ob = SvgText("A:" + bits_to_string(node.numBits, node.bitsAll),
node.x + 1,
yLine,
id="%s_Ball" % node)
ob.set_style(nameStyle)
svg.addElement(ob)
yLine += nameFontLineHgt
if (hasattr(node, "bitsSome")):
ob = SvgText("S:" +
bits_to_string(node.numBits, node.bitsSome),
node.x + 1,
yLine,
id="%s_Bsome" % node)
ob.set_style(nameStyle)
svg.addElement(ob)
yLine += nameFontLineHgt
# draw branches
for depth in depthToNames:
for name in depthToNames[depth]:
node = nameToNode[name]
if (node.left == None): continue
(leftStartX, leftStartY) = (node.x + 0.4 * dc.nodeWidth,
node.y + dc.nodeHeight)
(rightStartX, rightStartY) = (node.x + 0.6 * dc.nodeWidth,
node.y + dc.nodeHeight)
(leftEndX, leftEndY) = (node.left.x + 0.5 * dc.nodeWidth,
node.right.y)
(rightEndX, rightEndY) = (node.right.x + 0.5 * dc.nodeWidth,
node.right.y)
draw_branch(svg, "%s_left_branch" % node, leftStartX, leftStartY,
leftEndX, leftEndY)
draw_branch(svg, "%s_right_branch" % node, rightStartX,
rightStartY, rightEndX, rightEndY)
return svg
