boxespy/boxes/generators/roundedbox.py

116 lines
3.8 KiB
Python

#!/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/>.
import boxes
class RoundedBox(boxes.Boxes):
"""Box with vertical edges rounded"""
ui_group = "FlexBox"
def __init__(self):
boxes.Boxes.__init__(self)
self.addSettingsArgs(boxes.edges.FingerJointSettings)
self.addSettingsArgs(boxes.edges.FlexSettings)
self.buildArgParser("x", "y", "h", "outside")
self.argparser.add_argument(
"--radius", action="store", type=float, default=15,
help="Radius of the corners in mm")
self.argparser.add_argument(
"--wallpieces", action="store", type=int, default=1,
choices=[1, 2, 3, 4], help="# pieces of outer wall")
self.argparser.add_argument(
"--edge_style", action="store",
type=boxes.ArgparseEdgeType("fFh"), choices=list("fFh"),
default="f",
help="edge type for top and bottom edges")
self.argparser.add_argument(
"--top", action="store", type=str, default="none",
choices=["closed", "hole", "lid",],
help="style of the top and lid")
def hole(self):
t = self.thickness
x, y, r = self.x, self.y, self.radius
dr = 2*t
if self.edge_style == "h":
dr = t
if r > dr:
r -= dr
else:
x += dr - 2*r
y += dr - 2*r
self.moveTo(dr-r, 0)
r = 0
lx = x - 2*r - 2*dr
ly = y - 2*r - 2*dr
self.moveTo(0, dr)
for l in (lx, ly, lx, ly):
self.edge(l);
self.corner(90, r)
def render(self):
x, y, h, r = self.x, self.y, self.h, self.radius
if self.outside:
self.x = x = self.adjustSize(x)
self.y = y = self.adjustSize(y)
self.h = h = self.adjustSize(h)
r = self.radius = min(r, y / 2.0)
t = self.thickness
es = self.edge_style
corner_holes = True
if self.edge_style == "f":
pe = "F"
ec = False
elif self.edge_style == "F":
pe = "f"
ec = False
else: # "h"
pe = "f"
corner_holes = True
ec = True
with self.saved_context():
self.roundedPlate(x, y, r, es, wallpieces=self.wallpieces,
extend_corners=ec, move="right")
self.roundedPlate(x, y, r, es, wallpieces=self.wallpieces,
extend_corners=ec, move="right",
callback=[self.hole] if self.top != "closed" else None)
if self.top == "lid":
r_extra = self.edges[self.edge_style].spacing()
self.roundedPlate(x+2*r_extra,
y+2*r_extra,
r+r_extra,
"e", wallpieces=self.wallpieces,
extend_corners=False, move="right")
self.roundedPlate(x, y, r, es, wallpieces=self.wallpieces, move="up only")
self.surroundingWall(x, y, r, h, pe, pe, pieces=self.wallpieces)