Source code for boxes.generators.traylayout

#!/usr/bin/env 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 boxes
import argparse


[docs]class TrayLayout(Boxes): """Generate a typetray from a layout file""" # This class generates the skeleton text file that can then be edited # to describe the actual box webinterface = True ui_group = "Tray" def __init__(self, input=None, webargs=False): 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") # Use empty open and close methods to avoid initializing the whole # drawing infrastructure def open(self): pass def close(self): pass 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 render(self): self.fillDefault(self.x, self.y) with open(self.output, 'w') as f: f.write(str(self))
[docs]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 def __init__(self, input=None, webargs=False): 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 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 def vFloor(self, x, y): "Is there floor under vertical wall" return ((x > 0 and self.floors[y][x - 1]) or (x < len(self.x) and self.floors[y][x])) def hFloor(self, x, y): "Is there foor under horizontal wall" return ((y > 0 and self.floors[y - 1][x]) or (y < len(self.y) and self.floors[y][x])) @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) if self.hi: self.hi = self.adjustSize(self.hi, e2=False) self.hi = hi = self.hi or self.h lx = len(self.x) ly = len(self.y) t = self.thickness b = self.burn t2 = self.thickness / 2.0 hasfloor = False for line in self.floors: for f in line: hasfloor |= f self.edges["s"] = boxes.edges.Slot(self, self.hi / 2.0) self.edges["C"] = boxes.edges.CrossingFingerHoleEdge(self, self.hi) self.ctx.save() # Horizontal Walls for y in range(ly + 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]: if self.hFloor(end, y): edges.append("f") else: edges.append("e") # XXX E? lengths.append(self.x[end]) edges.append("eCs"[self.vWalls(end + 1, y)]) lengths.append(self.thickness) end += 1 # remove last "slot" lengths.pop() edges.pop() self.rectangularWall(sum(lengths), h, [ boxes.edges.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(lx + 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]: if self.vFloor(x, end): edges.append("f") else: edges.append("e") # XXX E? 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", "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, [ boxes.edges.CompoundEdge(self, edges, lengths), "eFf"[self.hWalls(x, end)], boxes.edges.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 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 + b, self.x[x], -180) if x == 0 or y == 0 or not self.floors[y - 1][x - 1]: self.edgeAt("e", posx, posy + t + b, t, -180) 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) elif y > 0 and self.floors[y - 1][x]: # Bottom Edge self.edgeAt(e, posx, posy - b, self.x[x]) if x == 0 or y == ly or not self.floors[y][x - 1]: self.edgeAt("e", posx - t, posy - b, t) if x == lx - 1 or y == ly or not self.floors[y][x + 1]: self.edgeAt("e", posx + self.x[x], posy - b, 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 + b, posy, self.y[y], 90) 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) if x == lx or y == ly - 1 or not self.floors[y + 1][x]: self.edgeAt("e", posx + t + b, posy - t, t, 90) elif x < lx and self.floors[y][x]: # Left edge self.edgeAt(e, posx - b, posy + self.y[y], self.y[y], -90) 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) if x == 0 or y == ly - 1 or not self.floors[y + 1][x - 1]: self.edgeAt("e", posx -b, posy, t, -90) posy += self.y[y] + self.thickness if x < lx: posx += self.x[x] + self.thickness def parse(self, input): x = [] y = [] hwalls = [] vwalls = [] floors = [] for line in input: if not line or 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[:len(x)*2 + 1]): 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 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") 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