Merge pull request #3081 from t3du/GetObjectGeom

Get Object Geometry Node

Sources/armory/logicnode/GetObjectGeomNode.hx

@@ -0,0 +1,122 @@
+package armory.logicnode;
+
+import iron.object.MeshObject;
+import iron.math.Vec4;
+
+class GetObjectGeomNode extends LogicNode {
+
+	public function new(tree: LogicTree) {
+		super(tree);
+	}
+
+	override function get(from: Int): Dynamic {
+		var object: MeshObject = inputs[0].get();
+
+		if (object == null) return null;
+
+		if (from == 5)
+			return object.vertex_groups;
+
+		if (from == 6){
+			var keys = [];
+			if (object.vertex_groups != null)
+				for (key in object.vertex_groups.keys())
+					keys.push(key);
+
+			return keys;
+		}
+
+		if (object.data == null) return null;
+
+		var pos: Array<Vec4> = [];
+		var nor: Array<Vec4> = [];
+
+		var positions = object.data.geom.positions.values;
+		var scl = object.data.scalePos;
+		var normals = object.data.geom.normals.values;
+
+		for (i in 0...Std.int(positions.length/4)){
+			pos.push(new Vec4(positions[i*4]*scl/32767, positions[i*4+1]*scl/32767, positions[i*4+2]*scl/32767, 1));
+			nor.push(new Vec4(normals[i*2]/32767, normals[i*2+1]/32767, positions[i*4+3]/32767, 1));
+		}
+
+		if (from == 0)
+			return pos;
+
+		if (from == 1)
+			return nor;
+
+
+		if (from == 2){
+			var index = [];
+			for (i in 0...pos.length)
+				index.push(i);
+
+			var unique: Array<Dynamic> = [];
+
+			for (item in pos)
+				if (unique.indexOf(Std.string(item)) == -1)
+					unique.push(Std.string(item));
+
+			for (u in 0...unique.length)
+				for (i in 0...pos.length)
+					if(Std.string(pos[i]) == unique[u]) index[i] = u;  
+
+			return index;
+		}
+
+		var ind: Array<Int> = [];
+
+		if (from == 3 || from == 4){
+			var matInd: Array<Int> = [];
+
+			var indices = object.data.geom.indices;
+
+			for (i in 0...indices.length){
+				for(j in 0...Std.int(indices[i].length))
+					matInd.push(i);
+				for (j in 0...indices[i].length)
+					ind.push(indices[i][j]);
+			}
+
+			var matIndex = [];
+			for (i in 0...pos.length)
+				matIndex.push(matInd[ind.indexOf(i)]);
+
+			if (from == 3)
+				return matIndex;
+		}
+
+
+		if (from == 4){
+			var face = 0;
+			var faceIndex = [];
+
+			for(i in 0...Std.int(ind.length/3)-1){
+				var ar = ind.slice(i*3, i*3+3).concat(ind.slice((i+1)*3, (i+1)*3+3));
+				var unique = [];
+				for (item in ar)
+					if (unique.indexOf(item) == -1)
+						unique.push(item);
+				faceIndex.push(face);
+				if (unique.length == ar.length)
+					face+=1;			
+			}
+			faceIndex.push(face);
+
+			var faceIndexF = [];
+			for (f in faceIndex)
+				for (i in 0...3) 
+					faceIndexF.push(f);
+
+			var faceInd = [];
+			for (i in 0...pos.length)
+				faceInd.push(faceIndexF[ind.indexOf(i)]);
+
+			return faceInd;
+		}
+
+		return null;
+
+	}
+}

blender/arm/exporter.py

@@ -786,6 +786,25 @@ def export_object(self, bobject: bpy.types.Object, out_parent: Dict = None) -> N
                 out_object['tilesheet_ref'] = bobject.arm_tilesheet
                 out_object['tilesheet_action_ref'] = bobject.arm_tilesheet_action
 
+
+            if len(bobject.vertex_groups) > 0:
+                out_object['vertex_groups'] = []
+                for group in bobject.vertex_groups:
+                    verts = []
+                    for v in bobject.data.vertices:
+                        for g in v.groups:
+                            if g.group == group.index:
+                                verts.append(str(v.co.x))
+                                verts.append(str(v.co.y))
+                                verts.append(str(v.co.z))
+
+                    out_vertex_groups = {
+                        'name': group.name,
+                        'value': verts
+                    }
+                    out_object['vertex_groups'].append(out_vertex_groups)
+
+
             if len(bobject.arm_propertylist) > 0:
                 out_object['properties'] = []
                 for proplist_item in bobject.arm_propertylist:
@@ -2688,6 +2707,7 @@ def post_export_object(self, bobject: bpy.types.Object, o, type):
             ArmoryExporter.export_physics = True
             rb = bobject.rigid_body
             shape = 0  # BOX
+
             if rb.collision_shape == 'SPHERE':
                 shape = 1
             elif rb.collision_shape == 'CONVEX_HULL':
@@ -2700,6 +2720,7 @@ def post_export_object(self, bobject: bpy.types.Object, o, type):
                 shape = 5
             elif rb.collision_shape == 'CAPSULE':
                 shape = 6
+
             body_mass = rb.mass
             is_static = self.rigid_body_static(rb)
             if is_static:

blender/arm/logicnode/object/LN_get_object_geom.py

@@ -0,0 +1,32 @@
+from arm.logicnode.arm_nodes import *
+
+class GetObjectGeomNode(ArmLogicTreeNode):
+    """Returns the vertex geometry info of the given object and the vertex groups info.
+    
+    @input Object: object geometry to retrieve.
+
+    @output Vertices Positions: an array with the vertices positions.
+    @output Vertices Normals: an array with vertices normals directions.
+    @output Vertices Indices: an array with vertices indices.
+    @output Vertices Material Indices: an array with material indices per vertex.
+    @output Vertices Vertices Face Indices: an array with face index per vertex.
+    @output Vertex Groups Maps: maps with vertex group names and vertices positions.
+    @output Vertex Groups Names: an array with the names of the vertex gruops.
+
+    """
+
+    bl_idname = 'LNGetObjectGeomNode'
+    bl_label = 'Get Object Geometry Node'
+    arm_section = 'relations'
+    arm_version = 1
+
+    def arm_init(self, context):
+        self.add_input('ArmNodeSocketObject', 'Object')
+
+        self.add_output('ArmNodeSocketArray', 'Vertices Positions')
+        self.add_output('ArmNodeSocketArray', 'Vertices Normals')
+        self.add_output('ArmNodeSocketArray', 'Vertices Indices')
+        self.add_output('ArmNodeSocketArray', 'Vertices Material Indices')
+        self.add_output('ArmNodeSocketArray', 'Vertices Face Indices')
+        self.add_output('ArmDynamicSocket', 'Vertex Groups Maps')
+        self.add_output('ArmNodeSocketArray', 'Vertex Groups Names')