New generator: RegularBox

boxes/generators/regularbox.py

@@ -0,0 +1,99 @@
+#!/usr/bin/env python3
+# Copyright (C) 2013-2014 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/>.
+
+from boxes import *
+
+
+class RegularBox(Boxes):
+    """Regular box"""
+
+    def __init__(self):
+        Boxes.__init__(self)
+        self.addSettingsArgs(edges.FingerJointSettings, finger=2.0, space=2.0)
+        self.buildArgParser("h", "outside")
+        self.argparser.add_argument(
+            "--radius",  action="store", type=float, default=50.0,
+            help="inner radius if the box (at the corners)")
+        self.argparser.add_argument(
+            "--n",  action="store", type=int, default=5,
+            help="number of sides")
+        self.argparser.add_argument(
+            "--top",  action="store", type=str, default="none",
+            choices=["none", "hole", "angled lid", "angled lid2", "round lid"],
+            help="style of the top and lid")
+
+    def render(self):
+        self.open()
+
+        r, h, n = self.radius, self.h, self.n
+
+        # XXX
+        if self.outside:
+            x = self.adjustSize(x)
+            y = self.adjustSize(y)
+            h = self.adjustSize(h)
+
+        t = self.thickness
+
+        edges.FingerJointSettings(self.thickness, True, angle=360./n, finger=2, space=2).edgeObjects(self, chars="gGH")
+
+        r, sh, side  = self.regularPolygon(n, radius=r)
+
+        self.ctx.save()
+        self.regularPolygonWall(corners=n, r=r, edges='F', move="right")
+        if self.top == "angled top":
+            self.regularPolygonWall(corners=n, r=r, edges='e', move="right")
+            self.regularPolygonWall(corners=n, r=r, edges='E', move="right")
+        elif self.top == "angled lid2":
+            self.regularPolygonWall(corners=n, r=r, edges='F', move="right",
+                                    callback=[lambda:self.regularPolygonAt(
+                                        0, 0, n, h=sh-t)])
+            self.regularPolygonWall(corners=n, r=r, edges='E', move="right")
+        elif self.top in ("hole", "round lid"):
+            self.regularPolygonWall(corners=n, r=r, edges='F', move="right",
+                          hole=(sh-t)*2)
+        if self.top == "round lid":
+            self.parts.disc(sh*2, move="right")
+
+        self.ctx.restore()
+        self.regularPolygonWall(corners=n, r=r, edges='F', move="up only")
+
+        side = 2 * math.sin(math.radians(180.0/n)) * r
+        fingers = self.top in ("hole", "round lid", "angled lid2")
+        
+        if n % 2:
+            for i in range(n):
+                self.rectangularWall(side, h, move="right",
+                                     edges="fgfG" if fingers else "fgeG")
+        else:
+            for i in range(n//2):
+                self.rectangularWall(side, h, move="right",
+                                     edges="fGfG" if fingers else "fGeG")
+                self.rectangularWall(side, h, move="right",
+                                     edges="fgfg" if fingers else "fgeg")
+
+
+        self.close()
+
+
+def main():
+    b = RegularBox()
+    b.parseArgs()
+    b.render()
+
+
+if __name__ == '__main__':
+    main()