-
-
Save zeffii/93e6a8e001678f2fcba3 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import bpy | |
import bmesh | |
import mathutils | |
from bmesh.ops import spin | |
from bpy.props import BoolProperty, FloatProperty, FloatVectorProperty | |
import math | |
from math import pi, radians | |
from node_s import * | |
from util import * | |
import sv_bmesh_utils | |
from sv_bmesh_utils import bmesh_from_pydata | |
def get_lathed_geometry(node, verts, edges, cent, axis, dvec, angle, steps): | |
bm = bmesh_from_pydata(verts, edges, []) | |
geom = bm.verts[:] + bm.edges[:] | |
spin( | |
bm, geom=geom, cent=cent, axis=axis, | |
dvec=dvec, angle=angle, steps=steps, use_duplicate=False) | |
if node.remove_doubles: | |
bmesh.ops.remove_doubles(bm, verts=bm.verts[:], dist=node.dist) | |
v = [v.co[:] for v in bm.verts] | |
p = [[i.index for i in p.verts] for p in bm.faces[:]] | |
bm.free() | |
return v, p | |
class SvLatheNode(Node, SverchCustomTreeNode): | |
bl_idname = 'SvLatheNode' | |
bl_label = 'Sv Lathe Node' | |
bl_icon = 'OUTLINER_OB_EMPTY' | |
remove_doubles = BoolProperty( | |
name='merge', description='Remove doubles', update=updateNode) | |
dist = FloatProperty(name="merge distance", default=0.0001, update=updateNode) | |
Degrees = FloatProperty(name="Degrees", default=360.0, update=updateNode) | |
Steps = IntProperty(name="Steps", default=20, update=updateNode) | |
cent = FloatVectorProperty(name='cent', size=3, update=updateNode) | |
dvec = FloatVectorProperty(name='dvec', size=3, update=updateNode) | |
axis = FloatVectorProperty( | |
name='axis', size=3, update=updateNode, default=(0, 0, 1)) | |
def init(self, context): | |
# so many inputs!! | |
self.inputs.new('VerticesSocket', 'Verts', 'Verts') | |
self.inputs.new('StringsSocket', 'Edges', 'Edges') | |
# these have default props | |
self.inputs.new('VerticesSocket', 'cent', 'cent').prop_name = 'cent' | |
self.inputs.new('VerticesSocket', 'axis', 'axis').prop_name = 'axis' | |
self.inputs.new('VerticesSocket', 'dvec', 'dvec').prop_name = 'dvec' | |
self.inputs.new('StringsSocket', 'Degrees', 'Degrees').prop_name = 'Degrees' | |
self.inputs.new('StringsSocket', 'Steps', 'Steps').prop_name = 'Steps' | |
# outputs, sparse! | |
self.outputs.new('VerticesSocket', 'Verts', 'Verts') | |
self.outputs.new('StringsSocket', 'Poly', 'Poly') | |
def draw_buttons(self, context, layout): | |
row = layout.row(align=True) | |
row.prop(self, "remove_doubles", text="merge") | |
def nothing_to_process(self): | |
if not ('Poly' in self.outputs): | |
return True | |
if not (self.inputs['Verts'].links and self.outputs['Verts'].links): | |
return True | |
def update(self): | |
if self.nothing_to_process(): | |
return | |
inputs = self.inputs | |
def get_socket(x, default): | |
links = inputs[x].links | |
if links: | |
r = SvGetSocketAnyType(self, inputs[x]) | |
r = {'found': dataCorrect(r)} | |
else: | |
r = {'generated': default} | |
return r | |
mverts = get_socket('Verts', []) | |
if 'generated' in mverts: | |
# not sure it ever reaches here | |
return | |
medges = get_socket('Edges', []) | |
mcent = get_socket('cent', [0, 0, 0]) | |
maxis = get_socket('axis', [0, 0, 1]) | |
mdvec = get_socket('dvec', [0, 0, 0]) | |
mDegrees = get_socket('Degrees', 360.0) | |
mSteps = get_socket('Steps', 20) | |
verts_out, faces_out = [], [] | |
for idx, Verts in enumerate(mverts['found']): | |
if not Verts: | |
continue | |
final_values = {'verts': Verts} | |
if 'found' in medges: | |
final_values['edges'] = lastrepeat(...) | |
ceter = lastrepeat(...) if 'found' in mcent else mcent['generated'] | |
final_values['cent'] = center | |
axis = lastrepeat(...) if 'found' in maxis else maxis['generated'] | |
final_values['axis'] = axis | |
dvec = lastrepeat(...) if 'found' in mdvec else mdvec['generated'] | |
final_values['dvec'] = dvec | |
angle = lastrepeat(...) if 'found' in mDegrees else mDegrees['generated'] | |
final_values['angle'] = radians(angle) | |
steps = lastrepeat(...) if 'found' in mSteps else mSteps['generated'] | |
final_values['steps'] = steps | |
v, p = get_lathed_geometry(self, **final_values) | |
verts_out.append(v) | |
faces_out.append(p) | |
SvSetSocketAnyType(self, 'Verts', verts_out) | |
SvSetSocketAnyType(self, 'Poly', faces_out) | |
def update_socket(self, context): | |
self.update() | |
def register(): | |
bpy.utils.register_class(SvLatheNode) | |
def unregister(): | |
bpy.utils.unregister_class(SvLatheNode) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment