boxespy/boxes/generators/pulley.py

82 lines
2.8 KiB
Python
Raw Normal View History

2016-07-13 19:55:54 +02:00
#!/usr/bin/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 *
from boxes import pulley
import math
class Pulley(Boxes):
"""Timing belt pulleys for different profiles"""
def __init__(self):
Boxes.__init__(self)
# remove cli params you do not need
self.buildArgParser("h")
self.argparser.add_argument(
"--profile", action="store", type=str, default="GT2_2mm",
choices=pulley.Pulley.getProfiles(),
help="profile of the teeth/belt")
self.argparser.add_argument(
"--teeth", action="store", type=int, default=20,
help="number of teeth")
self.argparser.add_argument(
"--axle", action="store", type=float, default=5,
help="diameter of the axle")
self.argparser.add_argument(
"--top", action="store", type=float, default=0,
help="overlap of top rim (zero for none)")
# Add non default cli params if needed (see argparse std lib)
#self.argparser.add_argument(
# "--XX", action="store", type=float, default=0.5,
# help="DESCRIPTION")
def disk(self, diameter, hole, callback=None, move=""):
w = diameter + 2*self.spacing
if self.move(w, w, move, before=True):
return
self.ctx.save()
self.moveTo(w/2, w/2)
self.cc(callback, None, 0.0, 0.0)
if hole:
self.hole(0, 0, hole/2.0)
self.moveTo(diameter/2+self.burn, 0, 90)
self.corner(360, diameter/2)
self.ctx.restore()
self.move(w, w, move)
def render(self):
# adjust to the variables you want in the local scope
t = self.thickness
# Initialize canvas
self.open()
if self.top:
self.disk(
self.pulley.diameter(self.teeth, self.profile)+2*self.top,
self.axle, move="right")
2016-07-13 20:10:13 +02:00
for i in range(int(math.ceil(self.h/self.thickness))):
2016-07-13 19:55:54 +02:00
self.pulley(self.teeth, self.profile, r_axle=self.axle/2.0, move="right")
self.close()
def main():
b = Box()
b.parseArgs()
b.render()
if __name__ == '__main__':
main()