boxespy/boxes/generators/traylayout.py

437 lines
16 KiB
Python
Raw Normal View History

#!/usr/bin/env python3
2016-03-08 18:04:49 +01:00
# 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/>.
2016-03-16 10:16:56 +01:00
import boxes
from boxes import *
2016-03-08 18:04:49 +01:00
class TrayLayout(Boxes):
2016-03-08 18:04:49 +01:00
"""Generate a typetray from a layout file"""
# This class generates the skeleton text file that can then be edited
# to describe the actual box
description = """This is a two step process. The layout is based on a grid
of sizes in x and y direction. Choose how many distances you need in both directions. The actual sizes and all other settings can be entered in the second step."""
webinterface = True
ui_group = "Tray"
2023-01-08 19:41:02 +01:00
def __init__(self, input=None, webargs=False) -> None:
2016-03-08 18:04:49 +01:00
Boxes.__init__(self)
self.argparser = argparse.ArgumentParser()
self.argparser.add_argument(
"--x", action="store", type=int, default=2,
help="number of compartments side by side")
self.argparser.add_argument(
"--y", action="store", type=int, default=2,
help="number of compartments back to front")
self.argparser.add_argument(
"--output", action="store", type=str, default="traylayout.txt",
help="name of the layout text file")
2016-03-08 18:04:49 +01:00
# Use empty open and close methods to avoid initializing the whole
# drawing infrastructure
def open(self):
pass
def close(self):
pass
2016-03-08 18:04:49 +01:00
def fillDefault(self, x, y):
self.x = [0.0] * x
self.y = [0.0] * y
2016-08-17 15:07:41 +02:00
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)]
2016-03-08 18:04:49 +01:00
self.floors = [[True for i in range(x)] for j in range(y)]
2016-08-17 15:07:41 +02:00
2023-01-08 19:41:02 +01:00
def __str__(self) -> str:
2016-03-08 18:04:49 +01:00
r = []
2016-08-17 15:07:41 +02:00
2016-03-08 18:04:49 +01:00
for i, x in enumerate(self.x):
r.append(" |" * (i) + " ,> %.1fmm\n" % x)
2016-08-17 15:07:41 +02:00
2016-03-08 18:04:49 +01:00
for hwalls, vwalls, floors, y in zip(
self.hwalls, self.vwalls, self.floors, self.y):
2016-08-17 15:07:41 +02:00
r.append("".join(("+" + " -"[h] for h in hwalls)) + "+\n")
2016-03-08 18:04:49 +01:00
r.append("".join((" |"[v] + "X "[f] for v, f in zip(vwalls, floors)))
2016-08-17 15:07:41 +02:00
+ " |"[vwalls[-1]] + " %.1fmm\n" % y)
2016-03-08 18:04:49 +01:00
r.append("".join(("+" + " -"[h] for h in self.hwalls[-1])) + "+\n")
2016-08-17 15:07:41 +02:00
2016-03-08 18:04:49 +01:00
return "".join(r)
def render(self):
self.fillDefault(self.x, self.y)
with open(self.output, 'w') as f:
f.write(str(self))
class TrayLayout2(TrayLayout):
"""Generate a typetray from a layout file"""
# This class reads in the layout either from a file (with --input) or
# as string (with --layout) and turns it into a drawing for a box.
webinterface = True
description = """Edit the layout text graphics to adjust your tray.
Put in the sizes for each column and row. You can replace the hyphens and
2023-01-02 00:32:42 +01:00
vertical bars representing the walls with a space character to remove the walls.
You can replace the space characters representing the floor by a "X" to remove the floor for this compartment.
"""
2023-01-08 19:41:02 +01:00
def __init__(self, input=None, webargs=False) -> None:
Boxes.__init__(self)
self.addSettingsArgs(boxes.edges.FingerJointSettings)
self.buildArgParser("h", "hi", "outside")
if not webargs:
self.argparser.add_argument(
"--input", action="store", type=argparse.FileType('r'),
default='traylayout.txt',
help="layout file")
self.layout = None
else:
self.argparser.add_argument(
"--layout", action="store", type=str, default="")
# Use normal open and close
open = Boxes.open
close = Boxes.close
2016-03-08 18:04:49 +01:00
def vWalls(self, x, y):
2023-01-02 00:32:42 +01:00
"""Number of vertical walls at a crossing"""
2016-03-08 18:04:49 +01:00
result = 0
2016-08-17 15:07:41 +02:00
if y > 0 and self.vwalls[y - 1][x]:
2016-03-08 18:04:49 +01:00
result += 1
2016-08-17 15:07:41 +02:00
2016-03-08 18:04:49 +01:00
if y < len(self.y) and self.vwalls[y][x]:
result += 1
2016-08-17 15:07:41 +02:00
2016-03-08 18:04:49 +01:00
return result
def hWalls(self, x, y):
2023-01-02 00:32:42 +01:00
"""Number of horizontal walls at a crossing"""
2016-03-08 18:04:49 +01:00
result = 0
2016-08-17 15:07:41 +02:00
if x > 0 and self.hwalls[y][x - 1]:
2016-03-08 18:04:49 +01:00
result += 1
2016-08-17 15:07:41 +02:00
if x < len(self.x) and self.hwalls[y][x]:
2016-03-08 18:04:49 +01:00
result += 1
return result
def vFloor(self, x, y):
2023-01-02 00:32:42 +01:00
"""Is there floor under vertical wall"""
2016-08-17 15:07:41 +02:00
return ((x > 0 and self.floors[y][x - 1]) or
(x < len(self.x) and self.floors[y][x]))
def hFloor(self, x, y):
2023-01-02 00:32:42 +01:00
"""Is there foor under horizontal wall"""
2016-08-17 15:07:41 +02:00
return ((y > 0 and self.floors[y - 1][x]) or
(y < len(self.y) and self.floors[y][x]))
2016-03-08 18:04:49 +01:00
@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):
if self.layout:
self.parse(self.layout.split('\n'))
else:
self.parse(self.input)
if self.outside:
self.x = self.adjustSize(self.x)
self.y = self.adjustSize(self.y)
self.h = self.adjustSize(self.h, e2=False)
2016-08-17 15:07:41 +02:00
if self.hi:
self.hi = self.adjustSize(self.hi, e2=False)
2016-03-08 18:04:49 +01:00
self.hi = hi = self.hi or self.h
lx = len(self.x)
ly = len(self.y)
t = self.thickness
b = self.burn
2016-03-08 18:04:49 +01:00
t2 = self.thickness / 2.0
2016-04-19 09:49:58 +02:00
hasfloor = False
for line in self.floors:
for f in line:
hasfloor |= f
2016-08-17 15:07:41 +02:00
self.edges["s"] = boxes.edges.Slot(self, self.hi / 2.0)
2016-04-19 09:49:58 +02:00
self.edges["C"] = boxes.edges.CrossingFingerHoleEdge(self, self.hi)
2016-03-08 18:04:49 +01:00
self.ctx.save()
# Horizontal Walls
for y in range(ly + 1):
2016-03-08 18:04:49 +01:00
if y == 0 or y == ly:
h = self.h
else:
h = self.hi
2016-08-17 15:07:41 +02:00
2016-03-08 18:04:49 +01:00
start = 0
end = 0
2016-08-17 15:07:41 +02:00
2016-03-08 18:04:49 +01:00
while end < lx:
lengths = []
edges = []
2016-08-17 15:07:41 +02:00
while start < lx and not self.hwalls[y][start]:
start += 1
2016-03-08 18:04:49 +01:00
if start == lx:
break
2016-08-17 15:07:41 +02:00
2016-03-08 18:04:49 +01:00
end = start
2016-08-17 15:07:41 +02:00
2016-03-08 18:04:49 +01:00
while end < lx and self.hwalls[y][end]:
if self.hFloor(end, y):
edges.append("f")
else:
2016-08-17 15:07:41 +02:00
edges.append("e") # XXX E?
2016-03-08 18:04:49 +01:00
lengths.append(self.x[end])
2016-08-17 15:07:41 +02:00
edges.append("eCs"[self.vWalls(end + 1, y)])
2016-03-08 18:04:49 +01:00
lengths.append(self.thickness)
end += 1
# remove last "slot"
lengths.pop()
edges.pop()
self.rectangularWall(sum(lengths), h, [
boxes.edges.CompoundEdge(self, edges, lengths),
2016-03-08 18:04:49 +01:00
"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()
2016-08-17 15:07:41 +02:00
2016-03-08 18:04:49 +01:00
# Vertical Walls
for x in range(lx + 1):
2016-03-08 18:04:49 +01:00
if x == 0 or x == lx:
h = self.h
else:
h = self.hi
start = 0
end = 0
2016-08-17 15:07:41 +02:00
2016-03-08 18:04:49 +01:00
while end < ly:
lengths = []
edges = []
2016-08-17 15:07:41 +02:00
while start < ly and not self.vwalls[start][x]:
start += 1
2016-03-08 18:04:49 +01:00
if start == ly:
break
2016-08-17 15:07:41 +02:00
2016-03-08 18:04:49 +01:00
end = start
2016-08-17 15:07:41 +02:00
while end < ly and self.vwalls[end][x]:
if self.vFloor(x, end):
edges.append("f")
else:
2016-08-17 15:07:41 +02:00
edges.append("e") # XXX E?
2016-03-08 18:04:49 +01:00
lengths.append(self.y[end])
2016-08-17 15:07:41 +02:00
edges.append("eCs"[self.hWalls(x, end + 1)])
2016-03-08 18:04:49 +01:00
lengths.append(self.thickness)
end += 1
# remove last "slot"
lengths.pop()
edges.pop()
upper = [{
2016-08-17 15:07:41 +02:00
"f": "e",
"s": "s",
"e": "e",
"E": "e",
"C": "e"}[e] for e in reversed(edges)]
2016-03-08 18:04:49 +01:00
edges = ["e" if e == "s" else e for e in edges]
self.rectangularWall(sum(lengths), h, [
boxes.edges.CompoundEdge(self, edges, lengths),
2016-03-08 18:04:49 +01:00
"eFf"[self.hWalls(x, end)],
boxes.edges.CompoundEdge(self, upper, list(reversed(lengths))),
2016-08-17 15:07:41 +02:00
"eFf"[self.hWalls(x, start)]],
move="right")
2016-03-08 18:04:49 +01:00
start = end
self.ctx.restore()
self.rectangularWall(10, h, "ffef", move="up only")
2016-08-17 15:07:41 +02:00
self.moveTo(2 * self.thickness, 2 * self.thickness)
2016-03-08 18:04:49 +01:00
self.ctx.save()
##########################################################
### Baseplate
##########################################################
# Horizontal lines
posy = 0
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]:
2016-08-17 15:07:41 +02:00
if y > 0 and self.floors[y - 1][x]:
2016-03-08 18:04:49 +01:00
# Inside Wall
if self.hwalls[y][x]:
2016-08-17 15:07:41 +02:00
self.fingerHolesAt(posx, posy + t2, self.x[x], angle=0)
2016-03-08 18:04:49 +01:00
else:
# Top edge
self.edgeAt(e, posx + self.x[x], posy + t + b, self.x[x],
2016-03-08 18:04:49 +01:00
-180)
2016-08-17 15:07:41 +02:00
if x == 0 or y == 0 or not self.floors[y - 1][x - 1]:
self.edgeAt("e", posx, posy + t + b, t, -180)
2016-08-17 15:07:41 +02:00
if x == lx - 1 or y == 0 or not self.floors[y - 1][x + 1]:
self.edgeAt("e", posx + self.x[x] + t, posy + t + b, t, -180)
2016-08-17 15:07:41 +02:00
elif y > 0 and self.floors[y - 1][x]:
2016-03-08 18:04:49 +01:00
# Bottom Edge
self.edgeAt(e, posx, posy - b, self.x[x])
2016-08-17 15:07:41 +02:00
if x == 0 or y == ly or not self.floors[y][x - 1]:
self.edgeAt("e", posx - t, posy - b, t)
2016-08-17 15:07:41 +02:00
if x == lx - 1 or y == ly or not self.floors[y][x + 1]:
self.edgeAt("e", posx + self.x[x], posy - b, t)
2016-08-17 15:07:41 +02:00
posx += self.x[x] + self.thickness
posy += self.y[y - 1] + self.thickness
2016-03-08 18:04:49 +01:00
posx = 0
2016-08-17 15:07:41 +02:00
for x in range(lx + 1):
2016-03-08 18:04:49 +01:00
posy = self.thickness
2016-08-17 15:07:41 +02:00
for y in range(ly - 1, -1, -1):
2016-03-08 18:04:49 +01:00
if self.vwalls[y][x]:
e = "F"
else:
e = "e"
2016-08-17 15:07:41 +02:00
if x > 0 and self.floors[y][x - 1]:
2016-03-08 18:04:49 +01:00
if x < lx and self.floors[y][x]:
# Inside wall
if self.vwalls[y][x]:
2016-08-17 15:07:41 +02:00
self.fingerHolesAt(posx + t2, posy, self.y[y])
2016-03-08 18:04:49 +01:00
else:
# Right edge
self.edgeAt(e, posx + t + b, posy, self.y[y], 90)
2016-08-17 15:07:41 +02:00
if x == lx or y == 0 or not self.floors[y - 1][x]:
self.edgeAt("e", posx + t + b, posy + self.y[y], t, 90)
2016-08-17 15:07:41 +02:00
if x == lx or y == ly - 1 or not self.floors[y + 1][x]:
self.edgeAt("e", posx + t + b, posy - t, t, 90)
2016-03-08 18:04:49 +01:00
elif x < lx and self.floors[y][x]:
# Left edge
self.edgeAt(e, posx - b, posy + self.y[y], self.y[y], -90)
2016-08-17 15:07:41 +02:00
if x == 0 or y == 0 or not self.floors[y - 1][x - 1]:
self.edgeAt("e", posx - b, posy + self.y[y] + t, t, -90)
2016-08-17 15:07:41 +02:00
if x == 0 or y == ly - 1 or not self.floors[y + 1][x - 1]:
self.edgeAt("e", posx -b, posy, t, -90)
2016-03-08 18:04:49 +01:00
posy += self.y[y] + self.thickness
if x < lx:
posx += self.x[x] + self.thickness
2016-08-17 15:07:41 +02:00
2016-03-08 18:04:49 +01:00
def parse(self, input):
x = []
y = []
hwalls = []
vwalls = []
floors = []
for nr, line in enumerate(input):
2016-03-16 10:16:56 +01:00
if not line or line[0] == "#":
2016-03-08 18:04:49 +01:00
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[:len(x)*2 + 1]):
2016-03-08 18:04:49 +01:00
if n % 2:
if c == ' ':
w.append(False)
elif c == '-':
w.append(True)
else:
pass
2016-08-17 15:07:41 +02:00
# raise ValueError(line)
2016-03-08 18:04:49 +01:00
else:
if c != '+':
pass
2016-08-17 15:07:41 +02:00
# raise ValueError(line)
2016-03-08 18:04:49 +01:00
hwalls.append(w)
if line[0] in " |":
w = []
f = []
2016-08-17 15:07:41 +02:00
for n, c in enumerate(line[:len(x) * 2 + 1]):
2016-03-08 18:04:49 +01:00
if n % 2:
if c in 'xX':
2016-03-08 18:04:49 +01:00
f.append(False)
elif c == ' ':
f.append(True)
else:
raise ValueError('''Can't parse line %i in layout: expected " ", "x" or "X" for char #%i''' % (nr+1, n+1))
2016-03-08 18:04:49 +01:00
else:
if c == ' ':
w.append(False)
elif c == '|':
w.append(True)
else:
raise ValueError('''Can't parse line %i in layout: expected " ", or "|" for char #%i''' % (nr+1, n+1))
2016-03-08 18:04:49 +01:00
floors.append(f)
vwalls.append(w)
m = re.match(r"([ |][ xX])+[ |]\s*(\d*\.?\d+)\s*mm\s*", line)
2016-03-08 18:04:49 +01:00
if not m:
raise ValueError('''Can't parse line %i in layout: Can read height of the row''' % (nr+1))
2016-03-08 18:04:49 +01:00
else:
y.append(float(m.group(2)))
# check sizes
lx = len(x)
ly = len(y)
if lx == 0:
raise ValueError("Need more than one wall in x direction")
if ly == 0:
raise ValueError("Need more than one wall in y direction")
2016-08-17 15:07:41 +02:00
if len(hwalls) != ly + 1:
raise ValueError("Wrong number of horizontal wall lines: %i (%i expected)" % (len(hwalls), ly + 1))
2016-03-08 18:04:49 +01:00
for nr, walls in enumerate(hwalls):
2016-08-17 15:07:41 +02:00
if len(walls) != lx:
2016-03-08 18:04:49 +01:00
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):
2016-08-17 15:07:41 +02:00
if len(walls) != lx + 1:
raise ValueError(
"Wrong number of vertical walls in line %i: %i (%i expected)" % (nr, len(walls), lx + 1))
2016-03-08 18:04:49 +01:00
self.x = x
self.y = y
self.hwalls = hwalls
self.vwalls = vwalls
self.floors = floors