removed unmantained functionality, did some cleanup

Author: botero-dev

README.md

@@ -1 +1,19 @@
-# Blender Datasmith addon
+# Blender Datasmith addon
+
+This addon lets you export your blender scene to UE4 using the Datasmith
+format.
+
+You can see a brief overview of how it works here:
+https://youtu.be/bUUDqerdqAc
+
+To install, get the latest release and install from the blender addons
+preferences pane.
+
+This is a work in progress, if you want to contribute to improve it, feel free
+to submit pull requests or support me on Patreon:
+https://www.patreon.com/0xafbf
+
+This is an unofficial exporter. This is not supported by Epic Games. That being
+said, you can still discuss about it and give feedback in the Unreal Engine
+forums.
+

io_scene_udatasmith/__init__.py

@@ -59,147 +59,14 @@
         axis_conversion,
         )
 
-
-b_major, b_minor, b_patch = bpy.app.version
-
-class ImportDatasmith(bpy.types.Operator, ImportHelper):
-    """Load a Datasmith file"""
-    bl_idname = "import_scene.udatasmith"
-    bl_label = "Import Datasmith"
-    bl_options = {'PRESET', 'UNDO'}
-
-    filename_ext = ".udatasmith"
-    filter_glob = StringProperty(
-        default="*.udatasmith",
-        options={'HIDDEN'}
-    )
-
-    use_smooth_groups = BoolProperty(
-        name="Smooth Groups",
-        description="Surround smooth groups by sharp edges",
-        default=True,
-    )
-
-    def execute(self, context):
-        keywords = self.as_keywords(ignore=("filter_glob",))
-        from . import import_datasmith
-        return import_datasmith.load(self, context, **keywords)
-
-    def draw(self, context):
-        layout = self.layout
-
-        layout.prop(self, 'use_smooth_groups')
-        
-
 class ExportDatasmith(bpy.types.Operator, ExportHelper):
     """Write a Datasmith file"""
     bl_idname = "export_scene.datasmith"
     bl_label = "Export Datasmith"
     bl_options = {'UNDO', 'PRESET'}
 
     filename_ext = ".udatasmith"
-    filter_glob = StringProperty(default="*.udatasmith", options={'HIDDEN'})
-
-    # List of operator properties, the attributes will be assigned
-    # to the class instance from the operator settings before calling.
-
-    version = EnumProperty(
-            items=(('UD_420', "Datasmith for UE 4.20", "Datasmith for Unreal Engine 4.20"),
-                   ('UD_421', "Datasmith for UE 4.21", "Datasmith for Unreal Engine 4.21"),
-                   ),
-            name="Version",
-            description="Choose which version of the exporter to use",
-            )
-
-    use_selection = BoolProperty(
-            name="Selected Objects",
-            description="Export selected objects on visible layers",
-            default=False,
-            )
-    global_scale = FloatProperty(
-            name="Scale",
-            description="Scale all data (Some importers do not support scaled armatures!)",
-            min=0.001, max=1000.0,
-            soft_min=0.01, soft_max=1000.0,
-            default=1.0,
-            )
-    
-    # TODO remove, left as doc for a multiple choice enum
-    '''object_types = EnumProperty(
-            name="Object Types",
-            options={'ENUM_FLAG'},
-            items=(('EMPTY', "Empty", ""),
-                   ('CAMERA', "Camera", ""),
-                   ('LIGHT', "Lamp", ""),
-                   ('ARMATURE', "Armature", "WARNING: not supported in dupli/group instances"),
-                   ('MESH', "Mesh", ""),
-                   ('OTHER', "Other", "Other geometry types, like curve, metaball, etc. (converted to meshes)"),
-                   ),
-            description="Which kind of object to export",
-            default={'EMPTY', 'CAMERA', 'LIGHT', 'ARMATURE', 'MESH', 'OTHER'},
-            )'''
-
-    # TODO remove: some of these are for fbx, maybe are useful for datasmith
-    '''
-    use_mesh_modifiers = BoolProperty(
-            name="Apply Modifiers",
-            description="Apply modifiers to mesh objects (except Armature ones) - "
-                        "WARNING: prevents exporting shape keys",
-            default=True,
-            )
-    use_mesh_modifiers_render = BoolProperty(
-            name="Use Modifiers Render Setting",
-            description="Use render settings when applying modifiers to mesh objects",
-            default=True,
-            )
-    mesh_smooth_type = EnumProperty(
-            name="Smoothing",
-            items=(('OFF', "Normals Only", "Export only normals instead of writing edge or face smoothing data"),
-                   ('FACE', "Face", "Write face smoothing"),
-                   ('EDGE', "Edge", "Write edge smoothing"),
-                   ),
-            description="Export smoothing information "
-                        "(prefer 'Normals Only' option if your target importer understand split normals)",
-            default='OFF',
-            )
-    use_mesh_edges = BoolProperty(
-            name="Loose Edges",
-            description="Export loose edges (as two-vertices polygons)",
-            default=False,
-            )
-    # 7.4 only
-    use_tspace = BoolProperty(
-            name="Tangent Space",
-            description="Add binormal and tangent vectors, together with normal they form the tangent space "
-                        "(will only work correctly with tris/quads only meshes!)",
-            default=False,
-            )
-    # 7.4 only'''
-
-
-    include_metadata = BoolProperty(
-            name="Include Metadata",
-            description="Include meshes metadata in file",
-            default=False,
-            )
-    
-    
-    #I don't know what this does
-    path_mode = path_reference_mode
-    # 7.4 only
-    embed_textures = BoolProperty(
-            name="Embed Textures",
-            description="Embed textures in Datasmith binary file (only for \"Copy\" path mode!)",
-            default=False,
-            )
-    batch_mode = EnumProperty(
-            name="Batch Mode",
-            items=(('OFF', "Off", "Active scene to file"),
-                   ('SCENE', "Scene", "Each scene as a file"),
-                   ('GROUP', "Group", "Each group as a file"),
-                   ),
-            )
-    
+    filter_glob: StringProperty(default="*.udatasmith", options={'HIDDEN'})
 
     def draw(self, context):
         layout = self.layout
@@ -212,11 +79,6 @@ def draw(self, context):
         layout.prop(self, "batch_mode")
 
 
-
-    @property
-    def check_extension(self):
-        return self.batch_mode == 'OFF'
-
     def execute(self, context):
         keywords = self.as_keywords(ignore=("filter_glob",))
         from . import export_datasmith
@@ -225,35 +87,20 @@ def execute(self, context):
 def menu_func_export(self, context):
     self.layout.operator(ExportDatasmith.bl_idname, text="Datasmith (.udatasmith)")
 
-def menu_func_import(self, context):
-    self.layout.operator(ImportDatasmith.bl_idname, text="Datasmith (.udatasmith)")
-
 classes = (
     ExportDatasmith,
-    ImportDatasmith,
 )
 
 
 def register():
     for cls in classes:
         bpy.utils.register_class(cls)
 
-    if b_minor >= 80:
-        bpy.types.TOPBAR_MT_file_export.append(menu_func_export)
-        bpy.types.TOPBAR_MT_file_import.append(menu_func_import)
-    else:
-        bpy.types.INFO_MT_file_export.append(menu_func_export)
-        bpy.types.INFO_MT_file_import.append(menu_func_import)
+    bpy.types.TOPBAR_MT_file_export.append(menu_func_export)
 
 
 def unregister():
-    if b_minor >= 80:
-        bpy.types.TOPBAR_MT_file_export.remove(menu_func_export)
-        bpy.types.TOPBAR_MT_file_import.remove(menu_func_import)
-    else:
-        bpy.types.INFO_MT_file_export.remove(menu_func_export)
-        bpy.types.INFO_MT_file_import.remove(menu_func_import)
-
+    bpy.types.TOPBAR_MT_file_export.remove(menu_func_export)
 
     for cls in classes:
         bpy.utils.unregister_class(cls)

io_scene_udatasmith/data_types.py

@@ -120,9 +120,7 @@ class UDMesh():
 
 	def __init__(self, path=None, node:ElementTree.Element = None, name=None):
 		self.name = name
-		if node:
-			self.init_with_xmlnode(node)
-		elif path:
+		if path:
 			self.init_with_path(path)
 
 		else:
@@ -153,79 +151,9 @@ def check_fields(self):
 		assert self.struct_property == 'StructProperty'
 		assert self.datasmith_mesh_source_model == 'DatasmithMeshSourceModel'
 
-	def init_with_xmlnode(self, node:ElementTree.Element):
-		self.name = node.attrib['name']
-		self.label = node.attrib['label']
-		self.relative_path = node.find('file').attrib['path']
-
-		parent_path = path.dirname(os.path.abspath(self.parent.path))
-		self.init_with_path(path.join(parent_path, self.relative_path))
-		# self.materials = {n.attrib['id']: n.attrib['name'] for n in node.iter('Material')}
-
-		# flatten material lists
-		material_map = {int(n.attrib['id']): idx for idx, n in enumerate(node.iter('Material'))}
-		self.materials = {idx: n.attrib['name'] for idx, n in enumerate(node.iter('Material'))}
-		if 0 not in material_map:
-			last_index = len(material_map)
-			material_map[0] = last_index
-			self.materials[last_index] = 'default_material'
-
-		self.tris_material_slot = list(map(lambda x: material_map.get(x, 0), self.tris_material_slot))
-
-
-	def init_with_path(self, path):
-		with open(path, 'rb') as file:
-
 
 			# this may need some work, found some documentation:
 			# Engine/Source/Developer/Rawmesh
-
-			self.a01 = read_data(file, 'II') # a 1 and the whole bytes size
-			self.name = read_string(file)
-
-			self.a02 = file.read(5)
-
-			self.source_models = read_string(file)
-			self.struct_property = read_string(file)
-			self.a03 = file.read(8)
-
-			self.datasmith_mesh_source_model = read_string(file)
-
-			self.a04 = file.read(25) # just zeros
-
-			self.payload_length = read_data(file, 'II') # this is the size of the rawmesh
-
-			self.a04_b = file.read(8) # this is a 125 and a zero, no idea what it is
-
-			self.raw_mesh_version = read_data(file, 'i') # always 1 for what I can see
-			self.raw_mesh_lic_version = read_data(file, 'i') # always 0 for what I can see
-
-			self.tris_material_slot = read_array_data(file, "I") #FaceMaterialIndices
-			self.tris_smoothing_group = read_array_data(file, "I") #FaceSmoothingMasks
-
-			self.vertices = read_array_data(file, "fff") #VertexPositions
-			self.triangles = read_array_data(file, "I") #WedgeIndices
-
-			self.a05 = read_array_data(file, "I") # WedgeTangentX (maybe)
-			self.a06 = read_array_data(file, "I") # WedgeTangentY (maybe)
-
-			self.vertex_normals = read_array_data(file, "fff") #WedgeTangentZ
-			self.uvs = read_array_data(file, "ff") #WedgeTexCoords
-
-			self.a07 = file.read(28) # these may be WedgeTexCoods[1,2...7]
-			# ue4 defines 8 texcoord layers, here we read the other seven zeros
-			self.a07_b = file.read(4) # these seem to be WedgeColors count
-			self.a07_c = file.read(4) # MaterialIndexToImportIndex?
-
-			self.checksum = file.read(16) # I guess this is Guid?
-
-			self.a08 = file.read() # And maybe this is bGuidIsHash
-
-			# small check here to crash if something is suspicious
-			assert len(self.triangles) == len(self.uvs)
-			assert len(self.vertex_normals) == len(self.uvs)
-			assert self.a08 == b'\x00\x00\x00\x00' # just to be sure its the end
-
 	def write_to_path(self, path):
 		with open(path, 'wb') as file:
 			#write_null(file, 8)
@@ -603,12 +531,13 @@ class UDScene():
 	current_scene = None
 
 	def __init__(self, source=None):
-		self.init_fields()
-		if type(source) is str:
-			self.path = source
-			self.init_with_path(self.path)
+		self.name = 'udscene_name'
+
+		self.materials = {}
+		self.meshes = {}
+		self.objects = {}
+		self.textures = {}
 
-		self.check_fields() # to test if it is possible for these fields to have different values
 
 	def get_field(self, cls, name, **kwargs):
 		group = getattr(self, cls.node_group)
@@ -622,52 +551,6 @@ def get_field(self, cls, name, **kwargs):
 		group[name] = new_object
 		return new_object
 
-
-	def init_fields(self):
-		self.name = 'udscene_name'
-
-		self.materials = {}
-		self.meshes = {}
-		self.objects = {}
-		self.textures = {}
-
-	def check_fields(self):
-		pass
-
-	def init_with_path(self, path):
-
-		# unmantained, for importing.
-		tree = ElementTree.parse(path)
-		root = tree.getroot()
-
-		self.version = root.find('Version').text
-		self.sdk_version = root.find('SDKVersion').text
-		self.host = root.find('Host').text
-		# there are other bunch of data that i'll skip for now
-
-		classes = [
-			UDTexture,
-			UDMaterial,
-			UDMasterMaterial,
-			UDMesh,
-			UDActor,
-			UDActorMesh,
-			UDActorCamera,
-			UDActorLight,
-		]
-
-		mappings = {cls.node_type:cls for cls in classes}
-
-		UDScene.current_scene = self
-		for node in root:
-			name = node.get('name') # most relevant nodes have a name as identifier
-			cls = mappings.get(node.tag)
-			if cls:
-				uobj = cls(name=name, node=node)
-				self.objects.push(uobj)
-
-		print("loaded")
-
 	def node(self):
 		n = Node('DatasmithUnrealScene')
 		n.push(Node('Version', children=['0.20']))

io_scene_udatasmith/export_datasmith.py

@@ -96,6 +96,15 @@ def make_field_node(field, name, default_name):
 			texture.image = image
 
 			return node
+		elif input_node.type == 'HUE_SAT':
+			return make_field_node(input_node.inputs['Color'], name, default_name)
+		elif input_node.type == 'MIX_RGB':
+			factor = input_node.inputs['Fac'].default_value
+			selected_input = 'Color1' if factor < 0.5 else 'Color2'
+			return make_field_node(input_node.inputs[selected_input], name, default_name)
+		elif input_node.type == 'RGB':
+			value = input_node.outputs[0]
+			return make_default_node(value, default_name)
 		else:
 			log.error("unhandled node")