CINF/CSKR integration

2025-12-09 05:47:42 +00:00 · 2015-08-12 21:29:00 -10:00
parent cdd7b45394
commit fd550702e1
21 changed files with 604 additions and 36 deletions
--- a/DataSpec/DNAMP3/CINF.hpp
+++ b/DataSpec/DNAMP3/CINF.hpp
@@ -0,0 +1,103 @@
+#ifndef _DNAMP3_CINF_HPP_
+#define _DNAMP3_CINF_HPP_
+
+#include "BlenderConnection.hpp"
+#include "../DNACommon/DNACommon.hpp"
+
+namespace Retro
+{
+namespace DNAMP3
+{
+
+struct CINF : BigDNA
+{
+    DECL_DNA
+    Value<atUint32> boneCount;
+    struct Bone : BigDNA
+    {
+        DECL_DNA
+        Value<atUint32> id;
+        Value<atUint32> parentId;
+        Value<atVec3f> origin;
+        Value<atUint32> linkedCount;
+        Vector<atUint32, DNA_COUNT(linkedCount)> linked;
+    };
+    Vector<Bone, DNA_COUNT(boneCount)> bones;
+
+    Value<atUint32> boneIdCount;
+    Vector<atUint32, DNA_COUNT(boneIdCount)> boneIds;
+
+    Value<atUint32> nameCount;
+    struct Name : BigDNA
+    {
+        DECL_DNA
+        String<-1> name;
+        Value<atUint32> boneId;
+    };
+    Vector<Name, DNA_COUNT(nameCount)> names;
+
+    atUint32 getBoneIdxFromId(atUint32 id) const
+    {
+        atUint32 idx = 0;
+        for (atUint32 bid : boneIds)
+        {
+            if (bid == id)
+                return idx;
+            ++idx;
+        }
+        return 0;
+    }
+
+    const std::string* getBoneNameFromId(atUint32 id) const
+    {
+        for (const Name& name : names)
+            if (id == name.boneId)
+                return &name.name;
+        return nullptr;
+    }
+
+    void sendVertexGroupsToBlender(HECL::BlenderConnection::PyOutStream& os) const
+    {
+        for (atUint32 bid : boneIds)
+        {
+            for (const Name& name : names)
+            {
+                if (name.boneId == bid)
+                {
+                    os.format("obj.vertex_groups.new('%s')\n", name.name.c_str());
+                    break;
+                }
+            }
+        }
+    }
+
+    void sendCINFToBlender(HECL::BlenderConnection::PyOutStream& os, const UniqueID32& cinfId) const
+    {
+        os.format("arm = bpy.data.armatures.new('CINF_%08X')\n"
+                  "arm_obj = bpy.data.objects.new(arm.name, arm)\n"
+                  "bpy.context.scene.objects.link(arm_obj)\n"
+                  "bpy.context.scene.objects.active = arm_obj\n"
+                  "bpy.ops.object.mode_set(mode='EDIT')\n"
+                  "arm_bone_table = {}\n",
+                  cinfId.toUint32());
+
+        for (const Bone& bone : bones)
+            os.format("bone = arm.edit_bones.new('%s')\n"
+                      "bone.head = (%f,%f,%f)\n"
+                      "bone.tail = bone.head\n"
+                      "bone.tail[1] += 0.5\n"
+                      "arm_bone_table[%u] = bone\n", getBoneNameFromId(bone.id)->c_str(),
+                      bone.origin.vec[0], bone.origin.vec[1], bone.origin.vec[2], bone.id);
+
+        for (const Bone& bone : bones)
+            if (bone.parentId != 2)
+                os.format("arm_bone_table[%u].parent = arm_bone_table[%u]\n", bone.id, bone.parentId);
+
+        os << "bpy.ops.object.mode_set(mode='OBJECT')\n";
+    }
+};
+
+}
+}
+
+#endif // _DNAMP3_CINF_HPP_