set attribute as split norm

Author: losanc

__init__.py

@@ -25,6 +25,7 @@
 classes = [
     SIMLOADER_obj_property,
     SIMLOADER_scene_property,
+    SIMLOADER_mesh_property,
     SIMLOADER_OT_load,
     SIMLOADER_OT_edit,
     SIMLOADER_OT_resetpt,
@@ -36,6 +37,8 @@
     SIMLOADER_Settings,
     SIMLOADER_Templates,
     SIMLOADER_UL_Att_List,
+    SIMLOADER_OT_set_as_split_norm,
+    SIMLOADER_OT_remove_split_norm,
 ]
 
 
@@ -46,6 +49,7 @@ def register():
     bpy.types.TEXT_MT_templates.append(draw_template)
     bpy.types.Scene.SIMLOADER = bpy.props.PointerProperty(type=SIMLOADER_scene_property)
     bpy.types.Object.SIMLOADER = bpy.props.PointerProperty(type=SIMLOADER_obj_property)
+    bpy.types.Mesh.SIMLOADER = bpy.props.PointerProperty(type=SIMLOADER_mesh_property)
 
 
 def unregister():

simloader/__init__.py

@@ -1,5 +1,5 @@
-from .operators import SIMLOADER_OT_load, SIMLOADER_OT_edit, SIMLOADER_OT_resetpt, SIMLOADER_OT_resetmesh, SIMLOADER_OT_resetins
-from .properties import SIMLOADER_scene_property, SIMLOADER_obj_property
+from .operators import SIMLOADER_OT_load, SIMLOADER_OT_edit, SIMLOADER_OT_resetpt, SIMLOADER_OT_resetmesh, SIMLOADER_OT_resetins,SIMLOADER_OT_set_as_split_norm,SIMLOADER_OT_remove_split_norm
+from .properties import SIMLOADER_scene_property, SIMLOADER_obj_property,SIMLOADER_mesh_property
 from .panels import SIMLOADER_UL_Obj_List, SIMLOADER_List_Panel, SIMLOADER_Settings, SIMLOADER_Import, SIMLOADER_Templates, SIMLOADER_UL_Att_List, draw_template
 from .messanger import subscribe_to_selected, unsubscribe_to_selected
 import bpy
@@ -55,4 +55,7 @@ def SIMLOADER_initilize(scene):
     "SIMLOADER_OT_resetins",
     "draw_template",
     "unsubscribe_to_selected",
+    "SIMLOADER_OT_set_as_split_norm",
+    "SIMLOADER_mesh_property",
+    "SIMLOADER_OT_remove_split_norm",
 ]

simloader/importer.py

@@ -129,6 +129,13 @@ def update_mesh(meshio_mesh, mesh):
             name_string = 'vector'
 
         attribute.data.foreach_set(name_string, v.ravel())
+        
+
+        # set as split norm
+        if mesh.SIMLOADER.split_norm_att_name and mesh.SIMLOADER.split_norm_att_name ==k:
+            mesh.use_auto_smooth = True
+            mesh.normals_split_custom_set_from_vertices(v)
+
 
 
 def create_obj(fileseq, use_relaitve, transform_matrix=Matrix([[1, 0, 0, 0], [0, 0, -1, 0], [0, 1, 0, 0], [0, 0, 0, 1]])):

simloader/operators.py

@@ -4,6 +4,7 @@
 import traceback
 from .utils import show_message_box
 from .importer import create_obj
+import numpy as np
 
 
 #  Here are load and delete operations
@@ -181,3 +182,38 @@ def execute(self, context):
 
         bpy.ops.object.modifier_move_to_index(modifier=gn.name, index=0)
         return {"FINISHED"}
+
+class SIMLOADER_OT_set_as_split_norm(bpy.types.Operator):
+    '''
+    This operator set the vertex attribute as vertex split normals
+    '''
+    bl_label = "Set as Split Norm per Vertex"
+    bl_idname = "simloader.setsplitnorm"
+    bl_options = {"UNDO"}
+    def execute(self, context):
+        sim_loader = context.scene.SIMLOADER
+        obj = bpy.data.objects[sim_loader.selected_obj_num]
+        mesh = obj.data
+        attr_index = sim_loader.selected_attribute_num
+        if attr_index>=len(mesh.attributes):
+            show_message_box("Please select the attribute")
+            return {"CANCELLED"}
+        mesh.SIMLOADER.split_norm_att_name = mesh.attributes[attr_index].name
+
+        return {"FINISHED"}
+
+class SIMLOADER_OT_remove_split_norm(bpy.types.Operator):
+    '''
+    This operator remove the vertex attribute as vertex split normals
+    '''
+    bl_label = "Remove Split Norm per Vertex"
+    bl_idname = "simloader.removesplitnorm"
+    bl_options = {"UNDO"}
+    def execute(self, context):
+        sim_loader = context.scene.SIMLOADER
+        obj = bpy.data.objects[sim_loader.selected_obj_num]
+        mesh = obj.data
+        if mesh.SIMLOADER.split_norm_att_name:
+            mesh.SIMLOADER.split_norm_att_name = ""
+
+        return {"FINISHED"}

simloader/panels.py

@@ -31,12 +31,17 @@ def draw_item(self, context, layout, data, item, icon, active_data, active_propn
 
 class SIMLOADER_UL_Att_List(bpy.types.UIList):
     '''
-    This controls the list of imported sequneces.
+    This controls the list of attributes available for this sequence
     '''
     def draw_item(self, context, layout, data, item, icon, active_data, active_propname):
         if item:
             layout.enabled = False
             layout.prop(item, "name", text='Name ', emboss=False)
+            obj = bpy.data.objects[context.scene.SIMLOADER.selected_obj_num]
+            mesh = obj.data
+            if mesh.SIMLOADER.split_norm_att_name and mesh.SIMLOADER.split_norm_att_name ==item.name:
+                layout.label(text="using as split norm")
+
         else:
             # actually, I guess this line of code won't be executed?
             layout.label(text="", translate=False, icon_value=icon)
@@ -121,6 +126,8 @@ def draw(self, context):
         box = layout.box()
         row = box.row()
         row.template_list("SIMLOADER_UL_Att_List", "", obj.data, "attributes", sim_loader, "selected_attribute_num", rows=2)
+        box.operator("SIMLOADER.setsplitnorm")
+        box.operator("SIMLOADER.removesplitnorm",text="remove split norm")
 
         # advance settings
         layout.label(text="Advance Settings")

simloader/properties.py

@@ -42,3 +42,8 @@ class SIMLOADER_obj_property(bpy.types.PropertyGroup):
     script_name: bpy.props.StringProperty()
     use_relative: bpy.props.BoolProperty(default=False)
     pattern: bpy.props.StringProperty()
+
+
+# set this property for mesh, not object (maybe change later?)
+class SIMLOADER_mesh_property(bpy.types.PropertyGroup):
+    split_norm_att_name : bpy.props.StringProperty(default="")