boxespy/boxes/generators/unevenheightbox.py

#!/usr/bin/env python3
# Copyright (C) 2013-2018 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 UnevenHeightBox(Boxes):
    """Box with different height in each corner"""

    ui_group = "Box"

    def __init__(self):
        Boxes.__init__(self)
        self.addSettingsArgs(edges.FingerJointSettings)
        self.buildArgParser("x", "y", "outside", bottom_edge="F")
        self.argparser.add_argument(
            "--height0", action="store", type=float, default=50,
            help="height of the front left corner in mm")
        self.argparser.add_argument(
            "--height1", action="store", type=float, default=50,
            help="height of the front right corner in mm")
        self.argparser.add_argument(
            "--height2", action="store", type=float, default=100,
            help="height of the right back corner in mm")
        self.argparser.add_argument(
            "--height3", action="store", type=float, default=100,
            help="height of the left back corner in mm")
        self.argparser.add_argument(
            "--lid", action="store", type=boolarg, default=False,
            help="add a lid (works best with high corners opposing each other)")

    def wall(self, w, h0, h1, edges="eee", move=None):

        edges = [self.edges.get(e, e) for e in edges]
        
        overallwidth = w + edges[-1].spacing() + edges[1].spacing()
        overallheight = max(h0, h1) + edges[0].spacing()

        if self.move(overallwidth, overallheight, move, before=True):
            return

        a = math.degrees(math.atan((h1-h0)/w))
        l = ((h0-h1)**2+w**2)**0.5

        self.moveTo(edges[-1].spacing(), edges[0].margin())
        edges[0](w)
        self.edgeCorner(edges[0], edges[1], 90)
        edges[1](h1)
        self.edgeCorner(edges[1], self.edges["e"], 90)
        self.polyline(0, a, l, -a)
        self.edgeCorner(self.edges["e"], edges[-1], 90)
        edges[-1](h0)
        self.edgeCorner(edges[-1], edges[0], 90)
        
        self.ctx.stroke()

        self.move(overallwidth, overallheight, move)


    def render(self):
        self.open()

        x, y = self.x, self.y
        heights = [self.height0, self.height1, self.height2, self.height3]

        if self.outside:
            x = self.adjustSize(x)
            y = self.adjustSize(y)
            for i in range(4):
                heights[i] = self.adjustSize(heights[i], None)

        t = self.thickness
        h0, h1, h2, h3 = heights
        b = self.bottom_edge

        self.wall(x, h0, h1, [b, "F", "F"], move="right")
        self.wall(y, h1, h2, [b, "f", "f"], move="right")
        self.wall(x, h2, h3, [b, "F", "F"], move="right")
        self.wall(y, h3, h0, [b, "f", "f"], move="right")

        self.ctx.save()
        self.rectangularWall(x, y, "ffff", move="up")

        if self.lid:
            maxh = max(heights)
            lidheights = [maxh-h for h in heights]
            h0, h1, h2, h3 = lidheights
            lidheights += lidheights

            edges = ["E" if (lidheights[i] == 0.0 and lidheights[i+1] == 0.0) else "f" for i in range(4)]
            self.rectangularWall(x, y, edges, move="up")

            self.ctx.restore()

            self.moveTo(0, maxh+self.edges["F"].spacing()+self.edges[b].spacing()+3*self.spacing, 180)
            self.wall(y, h0, h3, "Fff", move="right" +
                      (" only" if h0 == h3 == 0.0 else ""))
            self.wall(x, h3, h2, "FFF", move="right" +
                      (" only" if h3 == h2 == 0.0 else ""))
            self.wall(y, h2, h1, "Fff", move="right" +
                      (" only" if h2 == h1 == 0.0 else ""))
            self.wall(x, h1, h0, "FFF", move="right" +
                      (" only" if h1 == h0 == 0.0 else ""))

        self.close()