mirror of https://github.com/AxioDL/metaforce.git
83 lines
2.4 KiB
C++
83 lines
2.4 KiB
C++
#include "CINF.hpp"
|
|
#include "hecl/Blender/Connection.hpp"
|
|
|
|
namespace DataSpec::DNAMP2 {
|
|
|
|
atUint32 CINF::getInternalBoneIdxFromId(atUint32 id) const {
|
|
atUint32 idx = 0;
|
|
for (const Bone& b : bones) {
|
|
if (b.id == id)
|
|
return idx;
|
|
++idx;
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
atUint32 CINF::getBoneIdxFromId(atUint32 id) const {
|
|
atUint32 idx = 0;
|
|
for (atUint32 bid : boneIds) {
|
|
if (bid == id)
|
|
return idx;
|
|
++idx;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
const std::string* CINF::getBoneNameFromId(atUint32 id) const {
|
|
for (const Name& name : names)
|
|
if (id == name.boneId)
|
|
return &name.name;
|
|
return nullptr;
|
|
}
|
|
|
|
void CINF::sendVertexGroupsToBlender(hecl::blender::PyOutStream& os) const {
|
|
for (atUint32 bid : boneIds) {
|
|
for (const Name& name : names) {
|
|
if (name.boneId == bid) {
|
|
os.format(fmt("obj.vertex_groups.new(name='{}')\n"), name.name);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void CINF::sendCINFToBlender(hecl::blender::PyOutStream& os, const UniqueID32& cinfId) const {
|
|
DNAANIM::RigInverter<CINF> inverter(*this);
|
|
|
|
os.format(fmt(
|
|
"arm = bpy.data.armatures.new('CINF_{}')\n"
|
|
"arm_obj = bpy.data.objects.new(arm.name, arm)\n"
|
|
"bpy.context.scene.collection.objects.link(arm_obj)\n"
|
|
"bpy.context.view_layer.objects.active = arm_obj\n"
|
|
"bpy.ops.object.mode_set(mode='EDIT')\n"
|
|
"arm_bone_table = {{}}\n"),
|
|
cinfId);
|
|
|
|
for (const DNAANIM::RigInverter<CINF>::Bone& bone : inverter.getBones()) {
|
|
zeus::simd_floats originF(bone.m_origBone.origin.simd);
|
|
zeus::simd_floats tailF(bone.m_tail.mSimd);
|
|
os.format(fmt(
|
|
"bone = arm.edit_bones.new('{}')\n"
|
|
"bone.head = ({},{},{})\n"
|
|
"bone.tail = ({},{},{})\n"
|
|
"bone.use_inherit_scale = False\n"
|
|
"arm_bone_table[{}] = bone\n"),
|
|
*getBoneNameFromId(bone.m_origBone.id), originF[0], originF[1], originF[2], tailF[0], tailF[1],
|
|
tailF[2], bone.m_origBone.id);
|
|
}
|
|
|
|
for (const Bone& bone : bones)
|
|
if (bone.parentId != 97)
|
|
os.format(fmt("arm_bone_table[{}].parent = arm_bone_table[{}]\n"), bone.id, bone.parentId);
|
|
|
|
os << "bpy.ops.object.mode_set(mode='OBJECT')\n";
|
|
|
|
for (const DNAANIM::RigInverter<CINF>::Bone& bone : inverter.getBones())
|
|
os.format(fmt("arm_obj.pose.bones['{}'].rotation_mode = 'QUATERNION'\n"),
|
|
*getBoneNameFromId(bone.m_origBone.id));
|
|
}
|
|
|
|
std::string CINF::GetCINFArmatureName(const UniqueID32& cinfId) { return fmt::format(fmt("CINF_{}"), cinfId); }
|
|
|
|
} // namespace DataSpec::DNAMP2
|