traylayout initial commit
This commit is contained in:
Florian Festi
parent
3b69558905
commit
317625a4a6
|
|
@ -0,0 +1,343 @@
|
|||
#!/usr/bin/python3
|
||||
# Copyright (C) 2016 Florian Festi
|
||||
#
|
||||
# This program is free software: you can redistribute it and/or modify
|
||||
# it under the terms of the GNU General Public License as published by
|
||||
# the Free Software Foundation, either version 3 of the License, or
|
||||
# (at your option) any later version.
|
||||
#
|
||||
# This program is distributed in the hope that it will be useful,
|
||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
# GNU General Public License for more details.
|
||||
#
|
||||
# You should have received a copy of the GNU General Public License
|
||||
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
import sys, re
|
||||
from boxes import *
|
||||
import argparse
|
||||
|
||||
class Layout(Boxes):
|
||||
"""Generate a typetray from a layout file"""
|
||||
def __init__(self, input=None):
|
||||
Boxes.__init__(self)
|
||||
self.buildArgParser("h", "hi")
|
||||
self.argparser.add_argument(
|
||||
"--input", action="store", type=argparse.FileType('r'),
|
||||
help="layout file")
|
||||
self.argparser.add_argument(
|
||||
"--x", action="store", type=int, default=None,
|
||||
help="number of compartments side by side")
|
||||
self.argparser.add_argument(
|
||||
"--y", action="store", type=int, default=None,
|
||||
help="number of compartments back to front")
|
||||
|
||||
def fillDefault(self, x, y):
|
||||
self.x = [0.0] * x
|
||||
self.y = [0.0] * y
|
||||
self.hwalls = [[True for i in range(x)] for j in range(y+1)]
|
||||
self.vwalls = [[True for i in range(x+1)] for j in range(y)]
|
||||
self.floors = [[True for i in range(x)] for j in range(y)]
|
||||
|
||||
def __str__(self):
|
||||
r = []
|
||||
for i, x in enumerate(self.x):
|
||||
r.append(" |" * (i) + " ,> %.1fmm\n" % x)
|
||||
for hwalls, vwalls, floors, y in zip(
|
||||
self.hwalls, self.vwalls, self.floors, self.y):
|
||||
r.append("".join(("+" + " -"[h] for h in hwalls)) + "+\n")
|
||||
r.append("".join((" |"[v] + "X "[f] for v, f in zip(vwalls, floors)))
|
||||
+ " |"[vwalls[-1]] + " %.1fmm\n" % y)
|
||||
r.append("".join(("+" + " -"[h] for h in self.hwalls[-1])) + "+\n")
|
||||
return "".join(r)
|
||||
|
||||
def hholes(self, y, start, end):
|
||||
posx = -0.5 * self.thickness
|
||||
for i in range(start, end-1):
|
||||
posx += self.x[i] + self.thickness
|
||||
# XXX if no holes: continue
|
||||
self.fingerHolesAt(posx, 0, self.hi)
|
||||
|
||||
def vholes(self, x, start, end):
|
||||
posy = -0.5 * self.thickness
|
||||
for i in range(start, end-1):
|
||||
posy += self.y[i] + self.thickness
|
||||
# XXX if no holes: continue
|
||||
self.fingerHolesAt(posy, 0, self.hi)
|
||||
|
||||
def vWalls(self, x, y):
|
||||
"Number of vertical walls at a crossing"
|
||||
result = 0
|
||||
if y>0 and self.vwalls[y-1][x]:
|
||||
result += 1
|
||||
if y < len(self.y) and self.vwalls[y][x]:
|
||||
result += 1
|
||||
return result
|
||||
|
||||
def hWalls(self, x, y):
|
||||
"Number of horizontal walls at a crossing"
|
||||
result = 0
|
||||
if x>0 and self.hwalls[y][x-1]:
|
||||
result += 1
|
||||
if x<len(self.x) and self.hwalls[y][x]:
|
||||
result += 1
|
||||
return result
|
||||
|
||||
@restore
|
||||
def edgeAt(self, edge, x, y, length, angle=0):
|
||||
self.moveTo(x, y, angle)
|
||||
edge = self.edges.get(edge, edge)
|
||||
edge(length)
|
||||
|
||||
def render(self):
|
||||
self.open(1000, 600)
|
||||
|
||||
self.hi = hi = self.hi or self.h
|
||||
|
||||
self.edges["s"] = Slot(self, self.hi/2.0)
|
||||
self.edges["C"] = CrossingFingerHoleEdge(self, self.hi)
|
||||
|
||||
lx = len(self.x)
|
||||
ly = len(self.y)
|
||||
t = self.thickness
|
||||
t2 = self.thickness / 2.0
|
||||
|
||||
self.ctx.save()
|
||||
|
||||
# Horizontal Walls
|
||||
for y in range(len(self.y) + 1):
|
||||
if y == 0 or y == ly:
|
||||
h = self.h
|
||||
else:
|
||||
h = self.hi
|
||||
start = 0
|
||||
end = 0
|
||||
while end < lx:
|
||||
lengths = []
|
||||
edges = []
|
||||
while start < lx and not self.hwalls[y][start]: start += 1
|
||||
if start == lx:
|
||||
break
|
||||
end = start
|
||||
while end < lx and self.hwalls[y][end]:
|
||||
edges.append("f") # e for no ground
|
||||
lengths.append(self.x[end])
|
||||
edges.append("eCs"[self.vWalls(end, y)])
|
||||
lengths.append(self.thickness)
|
||||
end += 1
|
||||
|
||||
# remove last "slot"
|
||||
lengths.pop()
|
||||
edges.pop()
|
||||
self.rectangularWall(sum(lengths), h, [
|
||||
CompoundEdge(self, edges, lengths),
|
||||
"f" if self.vWalls(end, y) else "e",
|
||||
"e",
|
||||
"f" if self.vWalls(start, y) else "e"],
|
||||
move="right")
|
||||
start = end
|
||||
|
||||
self.ctx.restore()
|
||||
self.rectangularWall(10, h, "ffef", move="up only")
|
||||
self.ctx.save()
|
||||
|
||||
# Vertical Walls
|
||||
for x in range(len(self.x) + 1):
|
||||
if x == 0 or x == lx:
|
||||
h = self.h
|
||||
else:
|
||||
h = self.hi
|
||||
start = 0
|
||||
end = 0
|
||||
while end < ly:
|
||||
lengths = []
|
||||
edges = []
|
||||
while start < ly and not self.vwalls[start][x]: start += 1
|
||||
if start == ly:
|
||||
break
|
||||
end = start
|
||||
while end < ly and self.vwalls[end][x]:
|
||||
edges.append("f") # e for no ground
|
||||
lengths.append(self.y[end])
|
||||
edges.append("eCs"[self.hWalls(x, end+1)])
|
||||
lengths.append(self.thickness)
|
||||
end += 1
|
||||
# remove last "slot"
|
||||
lengths.pop()
|
||||
edges.pop()
|
||||
|
||||
upper = [{
|
||||
"f" : "e",
|
||||
"s" : "s",
|
||||
"e" : "e",
|
||||
"C" : "e"}[e] for e in reversed(edges)]
|
||||
edges = ["e" if e == "s" else e for e in edges]
|
||||
self.rectangularWall(sum(lengths), h, [
|
||||
CompoundEdge(self, edges, lengths),
|
||||
"eFf"[self.hWalls(x, end)],
|
||||
CompoundEdge(self, upper, list(reversed(lengths))),
|
||||
"eFf"[self.hWalls(x, start)] ],
|
||||
move="right")
|
||||
start = end
|
||||
|
||||
self.ctx.restore()
|
||||
self.rectangularWall(10, h, "ffef", move="up only")
|
||||
self.moveTo(2*self.thickness, 2*self.thickness)
|
||||
self.ctx.save()
|
||||
|
||||
##########################################################
|
||||
### Baseplate
|
||||
##########################################################
|
||||
|
||||
# Horizontal lines
|
||||
posy = 0
|
||||
print(self.floors)
|
||||
for y in range(ly, -1, -1):
|
||||
posx = self.thickness
|
||||
for x in range(lx):
|
||||
if self.hwalls[y][x]:
|
||||
e = "F"
|
||||
else:
|
||||
e = "e"
|
||||
if y < ly and self.floors[y][x]:
|
||||
if y>0 and self.floors[y-1][x]:
|
||||
# Inside Wall
|
||||
if self.hwalls[y][x]:
|
||||
self.fingerHolesAt(posx, posy+t2, self.x[x], angle=0)
|
||||
else:
|
||||
# Top edge
|
||||
self.edgeAt(e, posx+self.x[x], posy+t, self.x[x],
|
||||
-180)
|
||||
self.edgeAt("e", posx, posy+t, t, -180)
|
||||
self.edgeAt("e", posx+self.x[x]+t, posy+t, t, -180)
|
||||
elif y>0 and self.floors[y-1][x]:
|
||||
# Bottom Edge
|
||||
self.edgeAt(e, posx, posy, self.x[x])
|
||||
self.edgeAt("e", posx-t, posy, t)
|
||||
self.edgeAt("e", posx+self.x[x], posy, t)
|
||||
posx += self.x[x] + self. thickness
|
||||
posy += self.y[y-1] + self.thickness
|
||||
|
||||
posx = 0
|
||||
for x in range(lx+1):
|
||||
posy = self.thickness
|
||||
for y in range(ly-1, -1, -1):
|
||||
if self.vwalls[y][x]:
|
||||
e = "F"
|
||||
else:
|
||||
e = "e"
|
||||
if x>0 and self.floors[y][x-1]:
|
||||
if x < lx and self.floors[y][x]:
|
||||
# Inside wall
|
||||
if self.vwalls[y][x]:
|
||||
self.fingerHolesAt(posx+t2, posy, self.y[y])
|
||||
else:
|
||||
# Right edge
|
||||
self.edgeAt(e, posx+t, posy, self.y[y], 90)
|
||||
self.edgeAt("e", posx+t, posy-t, t, 90)
|
||||
self.edgeAt("e", posx+t, posy+self.y[y], t, 90)
|
||||
elif x < lx and self.floors[y][x]:
|
||||
# Left edge
|
||||
self.edgeAt(e, posx, posy+self.y[y], self.y[y], -90)
|
||||
self.edgeAt(e, posx, posy+self.y[y]+t, t, -90)
|
||||
self.edgeAt(e, posx, posy, t, -90)
|
||||
posy += self.y[y] + self.thickness
|
||||
if x < lx:
|
||||
posx += self.x[x] + self.thickness
|
||||
|
||||
self.close()
|
||||
|
||||
def parse(self, input):
|
||||
x = []
|
||||
y = []
|
||||
hwalls = []
|
||||
vwalls = []
|
||||
floors = []
|
||||
for line in input:
|
||||
if line[0] == "#":
|
||||
continue
|
||||
m = re.match(r"( \|)* ,>\s*(\d*\.?\d+)\s*mm\s*", line)
|
||||
if m:
|
||||
x.append(float(m.group(2)))
|
||||
continue
|
||||
if line[0] == '+':
|
||||
w = []
|
||||
for n, c in enumerate(line):
|
||||
if n % 2:
|
||||
if c == ' ':
|
||||
w.append(False)
|
||||
elif c == '-':
|
||||
w.append(True)
|
||||
else:
|
||||
pass
|
||||
#raise ValueError(line)
|
||||
else:
|
||||
if c != '+':
|
||||
pass
|
||||
#raise ValueError(line)
|
||||
|
||||
|
||||
hwalls.append(w)
|
||||
if line[0] in " |":
|
||||
w = []
|
||||
f = []
|
||||
for n, c in enumerate(line[:len(x)*2+1]):
|
||||
if n % 2:
|
||||
if c == 'X':
|
||||
f.append(False)
|
||||
elif c == ' ':
|
||||
f.append(True)
|
||||
else:
|
||||
raise ValueError(line)
|
||||
else:
|
||||
if c == ' ':
|
||||
w.append(False)
|
||||
elif c == '|':
|
||||
w.append(True)
|
||||
else:
|
||||
raise ValueError(line)
|
||||
|
||||
floors.append(f)
|
||||
vwalls.append(w)
|
||||
m = re.match(r"([ |][ X])+[ |]\s*(\d*\.?\d+)\s*mm\s*", line)
|
||||
if not m:
|
||||
raise ValueError(line)
|
||||
else:
|
||||
y.append(float(m.group(2)))
|
||||
|
||||
# check sizes
|
||||
lx = len(x)
|
||||
ly = len(y)
|
||||
if len(hwalls) != ly+1:
|
||||
raise ValueError("Wrong number of horizontal wall lines: %i (%i expected)" % (len(hwalls), ly+1))
|
||||
for nr, walls in enumerate(hwalls):
|
||||
if len(walls)!= lx:
|
||||
raise ValueError("Wrong number of horizontal walls in line %i: %i (%i expected)" % (nr, len(walls), lx))
|
||||
if len(vwalls) != ly:
|
||||
raise ValueError("Wrong number of vertical wall lines: %i (%i expected)" % (len(vwalls), ly))
|
||||
for nr, walls in enumerate(vwalls):
|
||||
if len(walls) != lx+1:
|
||||
raise ValueError("Wrong number of vertical walls in line %i: %i (%i expected)" % (nr, len(walls), lx+1))
|
||||
|
||||
self.x = x
|
||||
self.y = y
|
||||
self.hwalls = hwalls
|
||||
self.vwalls = vwalls
|
||||
self.floors = floors
|
||||
|
||||
def main():
|
||||
l = Layout()
|
||||
l.parseArgs()
|
||||
if l.x and l.y:
|
||||
l.fillDefault(l.x, l.y)
|
||||
print(l)
|
||||
elif l.input:
|
||||
l.parse(l.input)
|
||||
l.render()
|
||||
print(str(l))
|
||||
else:
|
||||
l.argparser.print_usage()
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
Loading…
Reference in New Issue