moved add_column to branches/broken_stuff/add_mesh_column.py it is broken for months...
[blender-addons-contrib.git] / add_mesh_cad_objects / retainer.py
1 # Stairbuilder - Retainer generation
2 #
3 # Generates retainers for stair generation.
4 #   Stair Type (typ):
5 #       - id1 = Freestanding staircase
6 #       - id2 = Housed-open staircase
7 #       - id3 = Box staircase
8 #       - id4 = Circular staircase
9
10 # Paul "BrikBot" Marshall
11 # Created: September 19, 2011
12 # Last Modified: January 29, 2011
13 # Homepage (blog): http://post.darkarsenic.com/
14 #                       //blog.darkarsenic.com/
15 #
16 # Coded in IDLE, tested in Blender 2.61.
17 # Search for "@todo" to quickly find sections that need work.
18 #
19 # ##### BEGIN GPL LICENSE BLOCK #####
20 #
21 #  Stairbuilder is for quick stair generation.
22 #  Copyright (C) 2011  Paul Marshall
23 #
24 #  This program is free software: you can redistribute it and/or modify
25 #  it under the terms of the GNU General Public License as published by
26 #  the Free Software Foundation, either version 3 of the License, or
27 #  (at your option) any later version.
28 #
29 #  This program is distributed in the hope that it will be useful,
30 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
31 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
32 #  GNU General Public License for more details.
33 #
34 #  You should have received a copy of the GNU General Public License
35 #  along with this program.  If not, see <http://www.gnu.org/licenses/>.
36 #
37 # ##### END GPL LICENSE BLOCK #####
38
39 from mathutils import Vector
40
41 class Retainers:
42     def __init__(self,G,w,h,wP,wT,hR,n, rEnable, lEnable):
43         self.G = G #General
44         self.w=w #retainer width
45         self.h=h #retainer height
46         self.wP=wP #post width
47         self.wT=wT #tread width
48         self.nR=n #number of retainers
49         self.sp=hR/float(n+1) #retainer spacing
50         self.rEnable = rEnable
51         self.lEnable = lEnable
52         self.Create()
53
54     def Create(self):
55         for i in range(self.nR):
56             coords = []
57             offset=(i+1)*Vector([0,0,self.sp])
58             coords.append(offset)
59             coords.append(self.G.stop + offset)
60             coords.append(offset + Vector([0,self.w,0]))
61             coords.append(self.G.stop + offset + Vector([0,self.w,0]))
62             for j in range(4):
63                 coords.append(coords[j] + Vector([0,0,self.h]))
64             #centre in posts
65             for j in coords:
66                 j += Vector([0,0.5*(self.wP-self.w),0])
67             if self.rEnable:
68                 self.G.Make_mesh(coords, self.G.faces, 'retainers')
69             if self.lEnable:
70                 #make retainer on other side
71                 for j in coords:
72                     j += Vector([0,self.wT-self.wP,0])
73                 self.G.Make_mesh(coords,self.G.faces, 'retainers')