boxespy/boxes/generators/paintbox.py

#!/usr/bin/env python3
# Copyright (C) 2013-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/>.

from boxes import Boxes, edges, boolarg


class PaintStorage(Boxes):
    """Stackable storage for hobby paint or other things"""

    webinterface = True
    ui_group = "Shelf" # see ./__init__.py for names

    def __init__(self) -> None:
        Boxes.__init__(self)

        self.addSettingsArgs(edges.FingerJointSettings)
        self.addSettingsArgs(edges.StackableSettings)

        self.buildArgParser(x=100, y=300)

        self.argparser.add_argument(
            "--canheight", action="store", type=int, default=50,
            help="Height of the paintcans")
        self.argparser.add_argument(
            "--candiameter", action="store", type=int, default=30,
            help="Diameter of the paintcans")
        self.argparser.add_argument(
            "--minspace", action="store", type=int, default=10,
            help="Minimum space between the paintcans")
        self.argparser.add_argument(
            "--hexpattern", action="store", type=boolarg, default=False,
            help="Use hexagonal arrangement for the holes instead of orthogonal")
        self.argparser.add_argument(
            "--drawer", action="store", type=boolarg, default=False,
            help="Create a stackable drawer instead")

    def paintholes(self):
        """Place holes for the paintcans evenly"""

        if self.hexpattern:
            self.moveTo(self.minspace/2, self.minspace/2)
            settings = self.hexHolesSettings
            settings.diameter = self.candiameter
            settings.distance = self.minspace
            settings.style = 'circle'
            self.hexHolesRectangle(self.y - 1*self.minspace,
                                   self.x - 1*self.minspace,
                                   settings)
            return
        n_x = int(self.x / (self.candiameter+self.minspace))
        n_y = int(self.y / (self.candiameter+self.minspace))

        if n_x <= 0 or n_y <= 0:
            return

        spacing_x = (self.x - n_x*self.candiameter)/n_x
        spacing_y = (self.y - n_y*self.candiameter)/n_y
        for i in range(n_y):
            for j in range(n_x):
                self.hole(i * (self.candiameter+spacing_y) + (self.candiameter+spacing_y)/2,
                          j * (self.candiameter+spacing_x) + (self.candiameter+spacing_x)/2,
                          self.candiameter/2)

    def render(self):
        # adjust to the variables you want in the local scope
        x, y = self.x, self.y
        t = self.thickness

        stack = self.edges['s'].settings
        h = self.canheight - stack.height - stack.holedistance + t

        hx = 1/2.*x
        hh = h/4.
        hr = min(hx, hh) / 2

        if not self.drawer:
            wall_keys = "EsES"
            wall_callbacks = [
                lambda: self.rectangularHole(h / 3, (x / 2.0) - t, hh, hx, r=hr),
                lambda: self.fingerHolesAt(0, self.canheight / 3, x, 0),
            ]
            bottom_keys = "EfEf"
        else:
            wall_keys = "FsFS"
            wall_callbacks = [
                lambda: self.rectangularHole(h / 3, (x / 2.0) - t, hh, hx, r=hr)
            ]
            bottom_keys = "FfFf"


        # Walls
        self.rectangularWall(
            h, x - 2 * t, wall_keys,
            ignore_widths=[1, 2, 5, 6],
            callback=wall_callbacks, move="up",
        )
        self.rectangularWall(
            h, x - 2 * t, wall_keys,
            ignore_widths=[1, 2, 5, 6],
            callback=wall_callbacks, move="right"
        )

        # Plates
        self.rectangularWall(
            0.8 * stack.height + stack.holedistance, x, "eeee", move=""
        )
        self.rectangularWall(
            0.8 * stack.height + stack.holedistance, x, "eeee", move="down right"
        )

        # Bottom
        self.rectangularWall(
            y, x-2*t, bottom_keys, ignore_widths=[1, 2, 5, 6], move="up"
        )

        if not self.drawer:
            # Top
            self.rectangularWall(y, x, "efef", callback=[self.paintholes], move="up")
        else:
            # Sides
            self.rectangularWall(y, h, "efff", move="up")
            self.rectangularWall(y, h, "efff", move="up")