add_mesh_extra_objects/add_mesh_torusknot.py

   1 # GPL #  Author, Anthony D'Agostino
   2
   3 import bpy
   4 from mathutils import Vector
   5 from math import sin, cos, pi
   6 from bpy.props import (
   7         BoolProperty,
   8         IntProperty,
   9         StringProperty,
  10         )
  11 from bpy_extras import object_utils
  12
  13
  14 def create_mesh_object(context, verts, edges, faces, name):
  15     # Create new mesh
  16     mesh = bpy.data.meshes.new(name)
  17     # Make a mesh from a list of verts/edges/faces.
  18     mesh.from_pydata(verts, edges, faces)
  19     # Update mesh geometry after adding stuff.
  20     mesh.update()
  21     from bpy_extras import object_utils
  22     return object_utils.object_data_add(context, mesh, operator=None)
  23
  24
  25 # ========================
  26 # === Torus Knot Block ===
  27 # ========================
  28
  29 def k1(t):
  30     x = cos(t) - 2 * cos(2 * t)
  31     y = sin(t) + 2 * sin(2 * t)
  32     z = sin(3 * t)
  33     return Vector([x, y, z])
  34
  35
  36 def k2(t):
  37     x = 10 * (cos(t) + cos(3 * t)) + cos(2 * t) + cos(4 * t)
  38     y = 6 * sin(t) + 10 * sin(3 * t)
  39     z = 4 * sin(3 * t) * sin(5 * t / 2) + 4 * sin(4 * t) - 2 * sin(6 * t)
  40     return Vector([x, y, z]) * 0.2
  41
  42
  43 def k3(t):
  44     x = 2.5 * cos(t + pi) / 3 + 2 * cos(3 * t)
  45     y = 2.5 * sin(t) / 3 + 2 * sin(3 * t)
  46     z = 1.5 * sin(4 * t) + sin(2 * t) / 3
  47     return Vector([x, y, z])
  48
  49
  50 def make_verts(ures, vres, r2, knotfunc):
  51     verts = []
  52     for i in range(ures):
  53         t1 = (i + 0) * 2 * pi / ures
  54         t2 = (i + 1) * 2 * pi / ures
  55         a = knotfunc(t1)        # curr point
  56         b = knotfunc(t2)        # next point
  57         a, b = map(Vector, (a, b))
  58         e = a - b
  59         f = a + b
  60         g = e.cross(f)
  61         h = e.cross(g)
  62         g.normalize()
  63         h.normalize()
  64         for j in range(vres):
  65             k = j * 2 * pi / vres
  66             l = (cos(k), 0.0, sin(k))
  67             l = Vector(l)
  68             m = l * r2
  69             x, y, z = m
  70             n = h * x
  71             o = g * z
  72             p = n + o
  73             q = a + p
  74             verts.append(q)
  75     return verts
  76
  77
  78 def make_faces(ures, vres):
  79     faces = []
  80     for u in range(0, ures):
  81         for v in range(0, vres):
  82             p1 = v + u * vres
  83             p2 = v + ((u + 1) % ures) * vres
  84             p4 = (v + 1) % vres + u * vres
  85             p3 = (v + 1) % vres + ((u + 1) % ures) * vres
  86             faces.append([p4, p3, p2, p1])
  87     return faces
  88
  89
  90 def make_knot(knotidx, ures):
  91     knots = [k1, k2, k3]
  92     knotfunc = knots[knotidx - 1]
  93     vres = ures // 10
  94     r2 = 0.5
  95     verts = make_verts(ures, vres, r2, knotfunc)
  96     faces = make_faces(ures, vres)
  97     return (verts, faces)
  98
  99
 100 class AddTorusKnot(bpy.types.Operator, object_utils.AddObjectHelper):
 101     bl_idname = "mesh.primitive_torusknot_add"
 102     bl_label = "Add Torus Knot"
 103     bl_description = "Construct a torus knot mesh"
 104     bl_options = {"REGISTER", "UNDO"}
 105
 106     TorusKnot : BoolProperty(name = "TorusKnot",
 107                 default = True,
 108                 description = "TorusKnot")
 109     change : BoolProperty(name = "Change",
 110                 default = False,
 111                 description = "change TorusKnot")
 112
 113     resolution: IntProperty(
 114         name="Resolution",
 115         description="Resolution of the Torus Knot",
 116         default=80,
 117         min=30, max=256
 118         )
 119     objecttype: IntProperty(
 120         name="Knot Type",
 121         description="Type of Knot",
 122         default=1,
 123         min=1, max=3
 124         )
 125
 126     def draw(self, context):
 127         layout = self.layout
 128
 129         layout.prop(self, 'resolution', expand=True)
 130         layout.prop(self, 'objecttype', expand=True)
 131
 132         if self.change == False:
 133             col = layout.column(align=True)
 134             col.prop(self, 'align', expand=True)
 135             col = layout.column(align=True)
 136             col.prop(self, 'location', expand=True)
 137             col = layout.column(align=True)
 138             col.prop(self, 'rotation', expand=True)
 139
 140     def execute(self, context):
 141         if bpy.context.mode == "OBJECT":
 142             if context.selected_objects != [] and context.active_object and \
 143             ('TorusKnot' in context.active_object.data.keys()) and (self.change == True):
 144                 obj = context.active_object
 145                 oldmesh = obj.data
 146                 oldmeshname = obj.data.name
 147                 verts, faces = make_knot(self.objecttype, self.resolution)
 148                 mesh = bpy.data.meshes.new('TorusKnot')
 149                 mesh.from_pydata(verts, [], faces)
 150                 obj.data = mesh
 151                 for material in oldmesh.materials:
 152                     obj.data.materials.append(material)
 153                 bpy.data.meshes.remove(oldmesh)
 154                 obj.data.name = oldmeshname
 155             else:
 156                 verts, faces = make_knot(self.objecttype, self.resolution)
 157                 mesh = bpy.data.meshes.new('TorusKnot')
 158                 mesh.from_pydata(verts, [], faces)
 159                 obj = object_utils.object_data_add(context, mesh, operator=self)
 160
 161             obj.data["TorusKnot"] = True
 162             obj.data["change"] = False
 163             for prm in TorusKnotParameters():
 164                 obj.data[prm] = getattr(self, prm)
 165
 166         if bpy.context.mode == "EDIT_MESH":
 167             active_object = context.active_object
 168             name_active_object = active_object.name
 169             bpy.ops.object.mode_set(mode='OBJECT')
 170             verts, faces = make_knot(self.objecttype, self.resolution)
 171             mesh = bpy.data.meshes.new('TorusKnot')
 172             mesh.from_pydata(verts, [], faces)
 173             obj = object_utils.object_data_add(context, mesh, operator=self)
 174             obj.select_set(True)
 175             active_object.select_set(True)
 176             bpy.ops.object.join()
 177             context.active_object.name = name_active_object
 178             bpy.ops.object.mode_set(mode='EDIT')
 179
 180         return {'FINISHED'}
 181
 182 def TorusKnotParameters():
 183     TorusKnotParameters = [
 184         "resolution",
 185         "objecttype",
 186         ]
 187     return TorusKnotParameters