From da2817c06574839b06d56d138675c7953b12e67f Mon Sep 17 00:00:00 2001 From: Brendon Murphy Date: Sun, 23 May 2010 04:44:25 +0000 Subject: addons/add_curve_aceous_galore.py added to trunk. also a "curve menu" script. --- add_curve_aceous_galore.py | 1150 ++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 1150 insertions(+) create mode 100644 add_curve_aceous_galore.py (limited to 'add_curve_aceous_galore.py') diff --git a/add_curve_aceous_galore.py b/add_curve_aceous_galore.py new file mode 100644 index 00000000..06627aa3 --- /dev/null +++ b/add_curve_aceous_galore.py @@ -0,0 +1,1150 @@ +# ##### BEGIN GPL LICENSE BLOCK ##### +# +# 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 2 +# 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, write to the Free Software Foundation, +# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. +# +# ##### END GPL LICENSE BLOCK ##### +bl_addon_info = { + 'name': 'Add Curve: Curveaceous Galore!', + 'author': 'Jimmy Hazevoet, testscreenings', + 'version': '0.1', + 'blender': (2, 5, 2), + 'location': 'Add Curve menu', + 'url': '', + 'description': 'adds many types of curves', + 'url': 'http://wiki.blender.org/index.php/Extensions:2.5/Py/' \ + 'Scripts/Curve/Curves_Galore', + 'category': 'Add Curve'} +##------------------------------------------------------------ +#### import modules +import bpy +from bpy.props import * +from mathutils import * +from math import * +###------------------------------------------------------------ +#### Some functions to use with others: +###------------------------------------------------------------ +''' +#------------------------------------------------------------ +# Generate random number: +def randnum( low=0.0, high=1.0, seed=0 ): + """ + randnum( low=0.0, high=1.0, seed=0 ) + + Create random number + + Parameters: + low - lower range + (type=float) + high - higher range + (type=float) + seed - the random seed number, if seed is 0, the current time will be used instead + (type=int) + Returns: + a random number + (type=float) + """ + + s = Noise.setRandomSeed( seed ) + rnum = Noise.random() + rnum = rnum*(high-low) + rnum = rnum+low + return rnum + + + +#------------------------------------------------------------ +# Make some noise: +def vTurbNoise((x,y,z), iScale=0.25, Size=1.0, Depth=6, Hard=0, Basis=0, Seed=0 ): + """ + vTurbNoise((x,y,z), iScale=0.25, Size=1.0, Depth=6, Hard=0, Basis=0, Seed=0 ) + + Create randomised vTurbulence noise + + Parameters: + xyz - (x,y,z) float values. + (type=3-float tuple) + iScale - noise intensity scale + (type=float) + Size - noise size + (type=float) + Depth - number of noise values added. + (type=int) + Hard - noise hardness: 0 - soft noise; 1 - hard noise + (type=int) + basis - type of noise used for turbulence + (type=int) + Seed - the random seed number, if seed is 0, the current time will be used instead + (type=int) + Returns: + the generated turbulence vector. + (type=3-float list) + """ + + rand = randnum(-100,100,Seed) + if Basis ==9: Basis = 14 + vTurb = Noise.vTurbulence(( x/Size+rand, y/Size+rand, z/Size+rand ), Depth, Hard, Basis ) + tx = vTurb[0]*iScale + ty = vTurb[1]*iScale + tz = vTurb[2]*iScale + return tx,ty,tz + + + +#------------------------------------------------------------ +# Axis: ( used in 3DCurve Turbulence ) +def AxisFlip((x,y,z), x_axis=1, y_axis=1, z_axis=1, flip=0 ): + if flip != 0: + flip *= -1 + else: flip = 1 + x *= x_axis*flip + y *= y_axis*flip + z *= z_axis*flip + return x,y,z +''' + +###------------------------------------------------------------------- +#### 2D Curve shape functions: +###------------------------------------------------------------------- + +##------------------------------------------------------------ +# 2DCurve: Profile: L, H, T, U, Z +def ProfileCurve(type=0, a=0.25, b=0.25): + """ + ProfileCurve( type=0, a=0.25, b=0.25 ) + + Create profile curve + + Parameters: + type - select profile type, L, H, T, U, Z + (type=int) + a - a scaling parameter + (type=float) + b - b scaling parameter + (type=float) + Returns: + a list with lists of x,y,z coordinates for curve points, [[x,y,z],[x,y,z],...n] + (type=list) + """ + + newpoints = [] + if type ==1: + ## H: + a*=0.5 + b*=0.5 + newpoints = [ [ -1.0, 1.0, 0.0 ], [ -1.0+a, 1.0, 0.0 ], + [ -1.0+a, b, 0.0 ], [ 1.0-a, b, 0.0 ], [ 1.0-a, 1.0, 0.0 ], + [ 1.0, 1.0, 0.0 ], [ 1.0, -1.0, 0.0 ], [ 1.0-a, -1.0, 0.0 ], + [ 1.0-a, -b, 0.0 ], [ -1.0+a, -b, 0.0 ], [ -1.0+a, -1.0, 0.0 ], + [ -1.0, -1.0, 0.0 ] ] + elif type ==2: + ## T: + a*=0.5 + newpoints = [ [ -1.0, 1.0, 0.0 ], [ 1.0, 1.0, 0.0 ], + [ 1.0, 1.0-b, 0.0 ], [ a, 1.0-b, 0.0 ], [ a, -1.0, 0.0 ], + [ -a, -1.0, 0.0 ], [ -a, 1.0-b, 0.0 ], [ -1.0, 1.0-b, 0.0 ] ] + elif type ==3: + ## U: + a*=0.5 + newpoints = [ [ -1.0, 1.0, 0.0 ], [ -1.0+a, 1.0, 0.0 ], + [ -1.0+a, -1.0+b, 0.0 ], [ 1.0-a, -1.0+b, 0.0 ], [ 1.0-a, 1.0, 0.0 ], + [ 1.0, 1.0, 0.0 ], [ 1.0, -1.0, 0.0 ], [ -1.0, -1.0, 0.0 ] ] + elif type ==4: + ## Z: + a*=0.5 + newpoints = [ [ -0.5, 1.0, 0.0 ], [ a, 1.0, 0.0 ], + [ a, -1.0+b, 0.0 ], [ 1.0, -1.0+b, 0.0 ], [ 1.0, -1.0, 0.0 ], + [ -a, -1.0, 0.0 ], [ -a, 1.0-b, 0.0 ], [ -1.0, 1.0-b, 0.0 ], + [ -1.0, 1.0, 0.0 ] ] + else: + ## L: + newpoints = [ [ -1.0, 1.0, 0.0 ], [ -1.0+a, 1.0, 0.0 ], + [ -1.0+a, -1.0+b, 0.0 ], [ 1.0, -1.0+b, 0.0 ], + [ 1.0, -1.0, 0.0 ], [ -1.0, -1.0, 0.0 ] ] + return newpoints + +##------------------------------------------------------------ +# 2DCurve: Miscellaneous.: Diamond, Arrow1, Arrow2, Square, .... +def MiscCurve(type=1, a=1.0, b=0.5, c=90.0): + """ + MiscCurve( type=1, a=1.0, b=0.5, c=90.0 ) + + Create miscellaneous curves + + Parameters: + type - select type, Diamond, Arrow1, Arrow2, Square + (type=int) + a - a scaling parameter + (type=float) + b - b scaling parameter + (type=float) + c - c scaling parameter + (type=float) + doesn't seem to do anything + Returns: + a list with lists of x,y,z coordinates for curve points, [[x,y,z],[x,y,z],...n] + (type=list) + """ + + newpoints = [] + a*=0.5 + b*=0.5 + if type ==1: + ## diamond: + newpoints = [ [ 0.0, b, 0.0 ], [ a, 0.0, 0.0 ], [ 0.0, -b, 0.0 ], [ -a, 0.0, 0.0 ] ] + elif type ==2: + ## Arrow1: + newpoints = [ [ -a, b, 0.0 ], [ a, 0.0, 0.0 ], [ -a, -b, 0.0 ], [ 0.0, 0.0, 0.0 ] ] + elif type ==3: + ## Arrow2: + newpoints = [ [ -1.0, b, 0.0 ], [ -1.0+a, b, 0.0 ], + [ -1.0+a, 1.0, 0.0 ], [ 1.0, 0.0, 0.0 ], + [ -1.0+a, -1.0, 0.0 ], [ -1.0+a, -b, 0.0 ], + [ -1.0, -b, 0.0 ] ] + elif type ==4: + ## Rounded square: + newpoints = [ [ -a, b-b*0.2, 0.0 ], [ -a+a*0.05, b-b*0.05, 0.0 ], [ -a+a*0.2, b, 0.0 ], + [ a-a*0.2, b, 0.0 ], [ a-a*0.05, b-b*0.05, 0.0 ], [ a, b-b*0.2, 0.0 ], + [ a, -b+b*0.2, 0.0 ], [ a-a*0.05, -b+b*0.05, 0.0 ], [ a-a*0.2, -b, 0.0 ], + [ -a+a*0.2, -b, 0.0 ], [ -a+a*0.05, -b+b*0.05, 0.0 ], [ -a, -b+b*0.2, 0.0 ] ] + + #elif type ==15: + ## : + #newpoints = [ [ x,y,z ] ] + else: + ## Square: + newpoints = [ [ -a, b, 0.0 ], [ a, b, 0.0 ], [ a, -b, 0.0 ], [ -a, -b, 0.0 ] ] + return newpoints + +##------------------------------------------------------------ +# 2DCurve: Star: +def StarCurve(starpoints=8, innerradius=0.5, outerradius=1.0, twist=0.0): + """ + StarCurve( starpoints=8, innerradius=0.5, outerradius=1.0, twist=0.0 ) + + Create star shaped curve + + Parameters: + starpoints - the number of points + (type=int) + innerradius - innerradius + (type=float) + outerradius - outerradius + (type=float) + twist - twist amount + (type=float) + Returns: + a list with lists of x,y,z coordinates for curve points, [[x,y,z],[x,y,z],...n] + (type=list) + """ + + newpoints = [] + step = (2.0/(starpoints)) + i = 0 + while i < starpoints: + t = (i*step) + x1 = cos(t*pi)*outerradius + y1 = sin(t*pi)*outerradius + newpoints.append([x1,y1,0]) + x2 = cos(t*pi+(pi/starpoints+twist))*innerradius + y2 = sin(t*pi+(pi/starpoints+twist))*innerradius + newpoints.append([x2,y2,0]) + i+=1 + return newpoints + +##------------------------------------------------------------ +# 2DCurve: Flower: +def FlowerCurve(petals=8, innerradius=0.5, outerradius=1.0, petalwidth=2.0): + """ + FlowerCurve( petals=8, innerradius=0.5, outerradius=1.0, petalwidth=2.0 ) + + Create flower shaped curve + + Parameters: + petals - the number of petals + (type=int) + innerradius - innerradius + (type=float) + outerradius - outerradius + (type=float) + petalwidth - width of petals + (type=float) + Returns: + a list with lists of x,y,z coordinates for curve points, [[x,y,z],[x,y,z],...n] + (type=list) + """ + + newpoints = [] + step = (2.0/(petals)) + pet = (step/pi*2)*petalwidth + i = 0 + while i < petals: + t = (i*step) + x1 = cos(t*pi-(pi/petals))*innerradius + y1 = sin(t*pi-(pi/petals))*innerradius + newpoints.append([x1,y1,0]) + x2 = cos(t*pi-pet)*outerradius + y2 = sin(t*pi-pet)*outerradius + newpoints.append([x2,y2,0]) + x3 = cos(t*pi+pet)*outerradius + y3 = sin(t*pi+pet)*outerradius + newpoints.append([x3,y3,0]) + i+=1 + return newpoints + +##------------------------------------------------------------ +# 2DCurve: Arc,Sector,Segment,Ring: +def ArcCurve(sides=6, startangle=0.0, endangle=90.0, innerradius=0.5, outerradius=1.0, type=3): + """ + ArcCurve( sides=6, startangle=0.0, endangle=90.0, innerradius=0.5, outerradius=1.0, type=3 ) + + Create arc shaped curve + + Parameters: + sides - number of sides + (type=int) + startangle - startangle + (type=float) + endangle - endangle + (type=float) + innerradius - innerradius + (type=float) + outerradius - outerradius + (type=float) + type - select type Arc,Sector,Segment,Ring + (type=int) + Returns: + a list with lists of x,y,z coordinates for curve points, [[x,y,z],[x,y,z],...n] + (type=list) + """ + + newpoints = [] + sides += 1 + angle = (2.0*(1.0/360.0)) + endangle-=startangle + step = ((angle*endangle)/(sides-1)) + i = 0 + while i < sides: + t = (i*step) + angle*startangle + x1 = sin(t*pi)*outerradius + y1 = cos(t*pi)*outerradius + newpoints.append([x1,y1,0]) + i+=1 + + #if type ==0: + # Arc: turn cyclic curve flag off! + + # Segment: + if type ==2: + newpoints.append([0,0,0]) + # Ring: + elif type ==3: + j=sides-1 + while j > -1: + t = (j*step) + angle*startangle + x2 = sin(t*pi)*innerradius + y2 = cos(t*pi)*innerradius + newpoints.append([x2,y2,0]) + j-=1 + return newpoints + +##------------------------------------------------------------ +# 2DCurve: Cog wheel: +def CogCurve(theeth=8, innerradius=0.8, middleradius=0.95, outerradius=1.0, bevel=0.5): + """ + CogCurve( theeth=8, innerradius=0.8, middleradius=0.95, outerradius=1.0, bevel=0.5 ) + + Create cog wheel shaped curve + + Parameters: + theeth - number of theeth + (type=int) + innerradius - innerradius + (type=float) + middleradius - middleradius + (type=float) + outerradius - outerradius + (type=float) + bevel - bevel amount + (type=float) + Returns: + a list with lists of x,y,z coordinates for curve points, [[x,y,z],[x,y,z],...n] + (type=list) + """ + + newpoints = [] + step = (2.0/(theeth)) + pet = (step/pi*2) + bevel = 1.0-bevel + i = 0 + while i < theeth: + t = (i*step) + x1 = cos(t*pi-(pi/theeth)-pet)*innerradius + y1 = sin(t*pi-(pi/theeth)-pet)*innerradius + newpoints.append([x1,y1,0]) + x2 = cos(t*pi-(pi/theeth)+pet)*innerradius + y2 = sin(t*pi-(pi/theeth)+pet)*innerradius + newpoints.append([x2,y2,0]) + x3 = cos(t*pi-pet)*middleradius + y3 = sin(t*pi-pet)*middleradius + newpoints.append([x3,y3,0]) + x4 = cos(t*pi-(pet*bevel))*outerradius + y4 = sin(t*pi-(pet*bevel))*outerradius + newpoints.append([x4,y4,0]) + x5 = cos(t*pi+(pet*bevel))*outerradius + y5 = sin(t*pi+(pet*bevel))*outerradius + newpoints.append([x5,y5,0]) + x6 = cos(t*pi+pet)*middleradius + y6 = sin(t*pi+pet)*middleradius + newpoints.append([x6,y6,0]) + i+=1 + return newpoints + +##------------------------------------------------------------ +# 2DCurve: nSide: +def nSideCurve(sides=6, radius=1.0): + """ + nSideCurve( sides=6, radius=1.0 ) + + Create n-sided curve + + Parameters: + sides - number of sides + (type=int) + radius - radius + (type=float) + Returns: + a list with lists of x,y,z coordinates for curve points, [[x,y,z],[x,y,z],...n] + (type=list) + """ + + newpoints = [] + step = (2.0/(sides)) + i = 0 + while i < sides: + t = (i*step) + x = sin(t*pi)*radius + y = cos(t*pi)*radius + newpoints.append([x,y,0]) + i+=1 + return newpoints + +''' +##------------------------------------------------------------ +# 2DCurve: Splat: +def SplatCurve(sides=24, scale=1.0, seed=0, basis=0, radius=1.0): + """ + SplatCurve( sides=24, scale=1.0, seed=0, basis=0, radius=1.0 ) + + Create splat curve + + Parameters: + sides - number of sides + (type=int) + scale - noise size + (type=float) + seed - noise random seed + (type=int) + basis - noise basis + (type=int) + radius - radius + (type=float) + Returns: + a list with lists of x,y,z coordinates for curve points, [[x,y,z],[x,y,z],...n] + (type=list) + """ + + newpoints = [] + step = (2.0/(sides)) + i = 0 + while i < sides: + t = (i*step) + turb = vTurbNoise(t, 1.0, scale, 6, 0, basis, seed ) + turb = turb[2] * 0.5 + 0.5 + x = sin(t*pi)*radius * turb + y = cos(t*pi)*radius * turb + newpoints.append([x,y,0]) + i+=1 + return newpoints +''' +###----------------------------------------------------------- +#### 3D curve shape functions: +###----------------------------------------------------------- + +###------------------------------------------------------------ +# 3DCurve: Helix: +def HelixCurve( number=100, height=2.0, startangle=0.0, endangle=360.0, width=1.0, a=0.0, b=0.0 ): + """ + HelixCurve( number=100, height=2.0, startangle=0.0, endangle=360.0, width=1.0, a=0.0, b=0.0 ) + + Create helix curve + + Parameters: + number - the number of points + (type=int) + height - height + (type=float) + startangle - startangle + (type=float) + endangle - endangle + (type=float) + width - width + (type=float) + a - a + (type=float) + b - b + (type=float) + Returns: + a list with lists of x,y,z coordinates for curve points, [[x,y,z],[x,y,z],...n] + (type=list) + """ + + newpoints = [] + angle = (2.0/360.0)*(endangle-startangle) + step = angle/(number-1) + h = height/angle + start = (startangle*2.0/360.0) + a/=angle + i = 0 + while i < number: + t = ( i*step+start ) + x = sin( (t*pi) ) * ( 1.0 + cos( t * pi * a - ( b * pi ) ) ) * ( 0.25 * width ) + y = cos( (t*pi) ) * ( 1.0 + cos( t * pi * a - ( b * pi ) ) ) * ( 0.25 * width ) + z = ( t * h ) -h*start + newpoints.append([x,y,z]) + i+=1 + return newpoints + +###------------------------------------------------------------ ? +# 3DCurve: Cycloid: Cycloid, Epicycloid, Hypocycloid +def CycloidCurve( number=24, length=2.0, type=0, a=1.0, b=1.0, startangle=0.0, endangle=360.0 ): + """ + CycloidCurve( number=24, length=2.0, type=0, a=1.0, b=1.0, startangle=0.0, endangle=360.0 ) + + Create a Cycloid, Epicycloid or Hypocycloid curve + + Parameters: + number - the number of points + (type=int) + length - length of curve + (type=float) + type - types: Cycloid, Epicycloid, Hypocycloid + (type=int) + Returns: + a list with lists of x,y,z coordinates for curve points, [[x,y,z],[x,y,z],...n] + (type=list) + """ + + newpoints = [] + angle = (2.0/360.0)*(endangle-startangle) + step = angle/(number-1) + #h = height/angle + d = length + start = (startangle*2.0/360.0) + a/=angle + i = 0 + if type == 0: # Epitrochoid + while i < number: + t = ( i*step+start ) + x = ((a + b) * cos(t*pi)) - (d * cos(((a+b)/b)*t*pi)) + y = ((a + b) * sin(t*pi)) - (d * sin(((a+b)/b)*t*pi)) + z = 0 # ( t * h ) -h*start + newpoints.append([x,y,z]) + i+=1 + + else: + newpoints = [[-1,-1,0], [-1,1,0], [1,1,0], [1,-1,0]] + print("\n", newpoints) + return newpoints + +##------------------------------------------------------------ +# calculates the matrix for the new object +# depending on user pref +def align_matrix(context): + loc = TranslationMatrix(context.scene.cursor_location) + obj_align = context.user_preferences.edit.object_align + if (context.space_data.type == 'VIEW_3D' + and obj_align == 'VIEW'): + rot = context.space_data.region_3d.view_matrix.rotation_part().invert().resize4x4() + else: + rot = Matrix() + align_matrix = loc * rot + return align_matrix + +##------------------------------------------------------------ +#### Curve creation functions +# sets bezierhandles to auto +def setBezierHandles(obj, mode = 'AUTOMATIC'): + scene = bpy.context.scene + if obj.type != 'CURVE': + return + scene.objects.active = obj + bpy.ops.object.mode_set(mode='EDIT', toggle=True) + bpy.ops.curve.select_all(action='SELECT') + bpy.ops.curve.handle_type_set(type=mode) + bpy.ops.object.mode_set(mode='OBJECT', toggle=True) + +# get array of vertcoordinates acording to splinetype +def vertsToPoints(Verts, splineType): + # main vars + vertArray = [] + + # array for BEZIER spline output (V3) + if splineType == 'BEZIER': + for v in Verts: + vertArray += v + + # array for nonBEZIER output (V4) + else: + for v in Verts: + vertArray += v + if splineType == 'NURBS': + vertArray.append(1) #for nurbs w=1 + else: #for poly w=0 + vertArray.append(0) + return vertArray + +# create new CurveObject from vertarray and splineType +def createCurve(vertArray, options, curveOptions, align_matrix): + # options to vars + splineType = options[0] # output splineType 'POLY' 'NURBS' 'BEZIER' + name = options[1] # GalloreType as name + + # create curve + scene = bpy.context.scene + newCurve = bpy.data.curves.new(name, type = 'CURVE') # curvedatablock + newSpline = newCurve.splines.new(type = splineType) # spline + + # create spline from vertarray + if splineType == 'BEZIER': + newSpline.bezier_points.add(int(len(vertArray)*0.33)) + newSpline.bezier_points.foreach_set('co', vertArray) + else: + newSpline.points.add(int(len(vertArray)*0.25 - 1)) + newSpline.points.foreach_set('co', vertArray) + newSpline.endpoint_u = True + + # set curveOptions + shape = curveOptions[0] + cyclic_u = curveOptions[1] + endp_u = curveOptions[2] + order_u = curveOptions[3] + handleType = curveOptions[4] + + newCurve.dimensions = shape + newSpline.cyclic_u = cyclic_u + newSpline.endpoint_u = endp_u + newSpline.order_u = order_u + + # create object with newCurve + new_obj = bpy.data.objects.new(name, newCurve) # object + scene.objects.link(new_obj) # place in active scene + new_obj.selected = True # set as selected + scene.objects.active = new_obj # set as active + new_obj.matrix = align_matrix # apply matrix + + # set bezierhandles + if splineType == 'BEZIER': + setBezierHandles(new_obj, handleType) + + return + +##------------------------------------------------------------ +# Main Function +def main(context, options, curveOptions, align_matrix): + # deselect all objects + bpy.ops.object.select_all(action='DESELECT') + + # options + galType = options[1] + splineType = options[0] + innerRadius = options[9] + middleRadius = options[10] + outerRadius = options[11] + + # get verts + if galType == 'Profile': + verts = ProfileCurve(options[2], options[3], options[4]) + if galType == 'Miscellaneous': + verts = MiscCurve(options[5], options[6], options[7], options[8]) + if galType == 'Flower': + verts = FlowerCurve(options[12], innerRadius, outerRadius, options[13]) + if galType == 'Star': + verts = StarCurve(options[14], innerRadius, outerRadius, options[15]) + if galType == 'Arc': + verts = ArcCurve(options[16], options[17], options[18], innerRadius, outerRadius, options[19]) + if galType == 'Cogwheel': + verts = CogCurve(options[20], innerRadius, middleRadius, outerRadius, options[21]) + if galType == 'Nsided': + verts = nSideCurve(options[22], outerRadius) +# if galType == 'Splat': +# verts = SplatCurve(options[23], options[24], options[25], options[26], outerRadius) + if galType == 'Helix': + verts = HelixCurve(options[27], options[28], options[29], options[30], options[31], options[32], options[33]) + if galType == 'Cycloid': + verts = CycloidCurve(options[34], options[35], options[36], options[37], options[38], options[39], options[40]) + + # turn verts into array + vertArray = vertsToPoints(verts, splineType) + + # create object + createCurve(vertArray, options, curveOptions, align_matrix) + + return + +class Curveaceous_galore(bpy.types.Operator): + '''''' + bl_idname = "curveaceous_galore" + bl_label = "Curveaceous galore" + bl_options = {'REGISTER', 'UNDO'} + bl_description = "adds many types of curves" + + # align_matrix for the invoke + align_matrix = Matrix() + + #### general properties + GalloreTypes = [ + ('Profile', 'Profile', 'Profile'), + ('Miscellaneous', 'Miscellaneous', 'Miscellaneous'), + ('Flower', 'Flower', 'Flower'), + ('Star', 'Star', 'Star'), + ('Arc', 'Arc', 'Arc'), + ('Cogwheel', 'Cogwheel', 'Cogwheel'), + ('Nsided', 'Nsided', 'Nsided'), +# ('Splat', 'Splat', 'Splat'), + ('Cycloid', 'Cycloid', 'Cycloid'), + ('Helix', 'Helix (3D)', 'Helix')] + GalloreType = EnumProperty(name="Type", + description="Form of Curve to create", + items=GalloreTypes) + SplineTypes = [ + ('POLY', 'Poly', 'POLY'), + ('NURBS', 'Nurbs', 'NURBS'), + ('BEZIER', 'Bezier', 'BEZIER')] + outputType = EnumProperty(name="Output splines", + description="Type of splines to output", + items=SplineTypes) + + #### Curve Options + shapeItems = [ + ('2D', '2D', '2D'), + ('3D', '3D', '3D')] + shape = EnumProperty(name="2D / 3D", + items=shapeItems, + description="2D or 3D Curve") + cyclic_u = BoolProperty(name="Cyclic", + default=True, + description="make curve closed") + endp_u = BoolProperty(name="endpoint_u", + default=True, + description="stretch to endpoints") + order_u = IntProperty(name="order_u", + default=4, + min=2, soft_min=2, + max=6, soft_max=6, + description="Order of nurbs spline") + bezHandles = [ + ('VECTOR', 'Vector', 'VECTOR'), + ('AUTOMATIC', 'Auto', 'AUTOMATIC')] + handleType = EnumProperty(name="Handle type", + description="bezier handles type", + items=bezHandles) + + #### ProfileCurve properties + ProfileCurveType = IntProperty(name="Nr", + min=1, soft_min=1, + max=5, soft_max=5, + default=1, + description="Type of ProfileCurve") + ProfileCurvevar1 = FloatProperty(name="var1", + default=0.25, + description="var1 of ProfileCurve") + ProfileCurvevar2 = FloatProperty(name="var2", + default=0.25, + description="var2 of ProfileCurve") + + #### MiscCurve properties + MiscCurveType = IntProperty(name="Nr", + min=1, soft_min=1, + max=5, soft_max=5, + default=1, + description="Type of ProfileCurve") + MiscCurvevar1 = FloatProperty(name="var1", + default=1.0, + description="var1 of ProfileCurve") + MiscCurvevar2 = FloatProperty(name="var2", + default=0.5, + description="var2 of ProfileCurve") + MiscCurvevar3 = FloatProperty(name="var3", # doesn't seem to do anything + default=90.0, + description="var3 of ProfileCurve") + + #### Common properties + innerRadius = FloatProperty(name="innerRadius", + default=0.5, + min=0, soft_min=0, + description="innerRadius") + middleRadius = FloatProperty(name="middleRadius", + default=0.95, + min=0, soft_min=0, + description="middleRadius") + outerRadius = FloatProperty(name="outerRadius", + default=1.0, + min=0, soft_min=0, + description="outerRadius") + + #### Flower properties + petals = IntProperty(name="Petals", + default=8, + min=2, soft_min=2, + description="number of petals") + petalWidth = FloatProperty(name="Petal width", + default=2.0, + min=0.01, soft_min=0.01, + description="Petal width") + + #### Star properties + starPoints = IntProperty(name="Star Points", + default=8, + min=2, soft_min=2, + description="number of Star Points") + starTwist = FloatProperty(name="Twist", + default=0.0, + description="Twist") + + #### Arc properties + arcSides = IntProperty(name="Arc Sides", + default=6, + min=1, soft_min=1, + description="Sides of Arc") + startAngle = FloatProperty(name="start angle", + default=0.0, + description="start angle") + endAngle = FloatProperty(name="end angle", + default=90.0, + description="end angle") + arcType = IntProperty(name="Arc Type", + default=2, + min=1, soft_min=2, + max=2, soft_max=2, + description="Sides of Arc") + + #### Cogwheel properties + teeth = IntProperty(name="Teeth", + default=8, + min=2, soft_min=2, + description="number of Teeth") + bevel = FloatProperty(name="Bevel", + default=0.5, + min=0, soft_min=0, + max=1, soft_max=1, + description="Bevel") + + #### Nsided property + Nsides = IntProperty(name="Sides", + default=8, + min=3, soft_min=3, + description="number of Sides") + + #### Splat properties + splatSides = IntProperty(name="Splat Sides", + default=24, + min=3, soft_min=3, + description="Splat Sides") + splatScale = FloatProperty(name="Splat Scale", + default=1.0, + min=0, soft_min=0, + description="Splat Scale") + seed = IntProperty(name="Seed", + default=0, + min=0, soft_min=0, + description="Seed") + basis = IntProperty(name="Basis", + default=0, + min=0, soft_min=0, + description="Basis") + + #### Helix properties + helixPoints = IntProperty(name="resolution", + default=100, + min=3, soft_min=3, + description="resolution") + helixHeight = FloatProperty(name="Height", + default=2.0, + min=0, soft_min=0, + description="Helix height") + helixStart = FloatProperty(name="Startangle", + default=0.0, + description="Helix Startangle") + helixEnd = FloatProperty(name="Endangle", + default=360.0, + description="Helix Endangle") + helixWidth = FloatProperty(name="Width", + default=1.0, + description="Helix width") + helix_a = FloatProperty(name="var_1", + default=0.0, + description="Helix var1") + helix_b = FloatProperty(name="var_2", + default=0.0, + description="Helix var2") + + #### Cycloid properties not yet implemented + cycloPoints = IntProperty(name="resolution", + default=200, + min=3, soft_min=3, + description="resolution") + cyclo_d = FloatProperty(name="var3", + default=1.5, + description="Cycloid var3") + cycloType = IntProperty(name="Type", + default=0, + min=0, soft_min=0, + max=4, soft_max=4, + description="resolution") + cyclo_a = FloatProperty(name="var_1", + default=5.0, + min=0.01, soft_min=0.01, + description="Cycloid var1") + cyclo_b = FloatProperty(name="var_2", + default=0.5, + min=0.01, soft_min=0.01, + description="Cycloid var2") + cycloStart = FloatProperty(name="Startangle", + default=0.0, + description="Cycloid Startangle") + cycloEnd = FloatProperty(name="Endangle", + default=360.0, + description="Cycloid Endangle") + + ##### DRAW ##### + def draw(self, context): + props = self.properties + layout = self.layout + + # general options + col = layout.column() + col.prop(props, 'GalloreType') + col.label(text=props.GalloreType+" Options") + + # options per GalloreType + box = layout.box() + if props.GalloreType == 'Profile': + box.prop(props, 'ProfileCurveType') + box.prop(props, 'ProfileCurvevar1') + box.prop(props, 'ProfileCurvevar2') + if props.GalloreType == 'Miscellaneous': + box.prop(props, 'MiscCurveType') + box.prop(props, 'MiscCurvevar1') + box.prop(props, 'MiscCurvevar2') + #box.prop(props, 'MiscCurvevar3') # doesn't seem to do anything + if props.GalloreType == 'Flower': + box.prop(props, 'petals') + box.prop(props, 'petalWidth') + box.prop(props, 'innerRadius') + box.prop(props, 'outerRadius') + if props.GalloreType == 'Star': + box.prop(props, 'starPoints') + box.prop(props, 'starTwist') + box.prop(props, 'innerRadius') + box.prop(props, 'outerRadius') + if props.GalloreType == 'Arc': + box.prop(props, 'arcSides') + box.prop(props, 'arcType') + box.prop(props, 'startAngle') + box.prop(props, 'endAngle') + #box.prop(props, 'innerRadius') # doesn't seem to do anything + box.prop(props, 'outerRadius') + if props.GalloreType == 'Cogwheel': + box.prop(props, 'teeth') + box.prop(props, 'bevel') + box.prop(props, 'innerRadius') + box.prop(props, 'middleRadius') + box.prop(props, 'outerRadius') + if props.GalloreType == 'Nsided': + box.prop(props, 'Nsides') + box.prop(props, 'outerRadius', text='Radius') + ''' + if props.GalloreType == 'Splat': + box.prop(props, 'splatSides') + box.prop(props, 'outerRadius') + box.prop(props, 'splatScale') + box.prop(props, 'seed') + box.prop(props, 'basis') + ''' + if props.GalloreType == 'Helix': + box.prop(props, 'helixPoints') + box.prop(props, 'helixHeight') + box.prop(props, 'helixWidth') + box.prop(props, 'helixStart') + box.prop(props, 'helixEnd') + box.prop(props, 'helix_a') + box.prop(props, 'helix_b') + if props.GalloreType == 'Cycloid': + box.prop(props, 'cycloPoints') + box.prop(props, 'cycloType') + box.prop(props, 'cycloStart') + box.prop(props, 'cycloEnd') + box.prop(props, 'cyclo_a') + box.prop(props, 'cyclo_b') + box.prop(props, 'cyclo_d') + + col = layout.column() + col.label(text="Output Curve Type") + row = layout.row() + row.prop(props, 'outputType', expand=True) + col = layout.column() + col.label(text="Curve Options") + + # output options + box = layout.box() + if props.outputType == 'NURBS': + box.row().prop(props, 'shape', expand=True) + box.prop(props, 'cyclic_u') + box.prop(props, 'endp_u') + box.prop(props, 'order_u') + + if props.outputType == 'POLY': + box.row().prop(props, 'shape', expand=True) + box.prop(props, 'cyclic_u') + + if props.outputType == 'BEZIER': + box.row().prop(props, 'shape', expand=True) + box.row().prop(props, 'handleType', expand=True) + box.prop(props, 'cyclic_u') + + + ##### POLL ##### + def poll(self, context): + return context.scene != None + + ##### EXECUTE ##### + def execute(self, context): + # turn off undo + undo = bpy.context.user_preferences.edit.global_undo + bpy.context.user_preferences.edit.global_undo = False + + props = self.properties + if props.GalloreType in ['Helix', 'Cycloid']: + props.shape = '3D' + #props.cyclic_u = False + + # Options + options = [ + # general properties + props.outputType, #0 + props.GalloreType, #1 + # ProfileCurve properties + props.ProfileCurveType, #2 + props.ProfileCurvevar1, #3 + props.ProfileCurvevar2, #4 + # MiscCurve properties + props.MiscCurveType, #5 + props.MiscCurvevar1, #6 + props.MiscCurvevar2, #7 + props.MiscCurvevar3, #8 + # Common properties + props.innerRadius, #9 + props.middleRadius, #10 + props.outerRadius, #11 + # Flower properties + props.petals, #12 + props.petalWidth, #13 + # Star properties + props.starPoints, #14 + props.starTwist, #15 + # Arc properties + props.arcSides, #16 + props.startAngle, #17 + props.endAngle, #18 + props.arcType, #19 + # Cogwheel properties + props.teeth, #20 + props.bevel, #21 + # Nsided property + props.Nsides, #22 + # Splat properties + props.splatSides, #23 + props.splatScale, #24 + props.seed, #25 + props.basis, #26 + # Helix properties + props.helixPoints, #27 + props.helixHeight, #28 + props.helixStart, #29 + props.helixEnd, #30 + props.helixWidth, #31 + props.helix_a, #32 + props.helix_b, #33 + # Cycloid properties + props.cycloPoints, #34 + props.cyclo_d, #35 + props.cycloType, #36 + props.cyclo_a, #37 + props.cyclo_b, #38 + props.cycloStart, #39 + props.cycloEnd #40 + ] + + # Curve options + curveOptions = [ + props.shape, #0 + props.cyclic_u, #1 + props.endp_u, #2 + props.order_u, #4 + props.handleType #5 + ] + + # main function + main(context, options, curveOptions, self.align_matrix) + + # restore pre operator undo state + bpy.context.user_preferences.edit.global_undo = undo + + return {'FINISHED'} + + ##### INVOKE ##### + def invoke(self, context, event): + # store creation_matrix + self.align_matrix = align_matrix(context) + self.execute(context) + + return {'FINISHED'} + +################################################################################ +##### REGISTER ##### + +Curveaceous_galore_button = (lambda self, context: self.layout.operator + (Curveaceous_galore.bl_idname, text="curvatures gallore")) + +classes = [ +Curveaceous_galore + ] + +def register(): + register = bpy.types.register + for cls in classes: + register(cls) + + bpy.types.INFO_MT_curve_add.append(Curveaceous_galore_button) + +def unregister(): + unregister = bpy.types.unregister + for cls in classes: + unregister(cls) + + bpy.types.INFO_MT_curve_add.remove(Curveaceous_galore_button) + +if __name__ == "__main__": + register() \ No newline at end of file -- cgit v1.2.3