From 423eb84362ea1a564e885a777a131e0787ef8869 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?L=C3=A9o=20Lam?= <leo@leolam.fr>
Date: Thu, 6 Jan 2022 15:50:13 +0100
Subject: ksys/phys: Implement more StaticCompound functions

The other functions require RE'ing BodyGroup, which in turn requires
RigidBody stuff
---
 .../Physics/StaticCompound/physStaticCompound.cpp  | 95 +++++++++++++++++++++-
 1 file changed, 92 insertions(+), 3 deletions(-)

(limited to 'src/KingSystem/Physics/StaticCompound/physStaticCompound.cpp')

diff --git a/src/KingSystem/Physics/StaticCompound/physStaticCompound.cpp b/src/KingSystem/Physics/StaticCompound/physStaticCompound.cpp
index 6fb85437..6986e510 100644
--- a/src/KingSystem/Physics/StaticCompound/physStaticCompound.cpp
+++ b/src/KingSystem/Physics/StaticCompound/physStaticCompound.cpp
@@ -1,19 +1,108 @@
 #include "KingSystem/Physics/StaticCompound/physStaticCompound.h"
-#include "KingSystem/Physics/StaticCompound/physStaticCompoundFieldBodyGroup.h"
+#include <Havok/Common/Serialize/Util/hkNativePackfileUtils.h>
+#include <Havok/Common/Serialize/Util/hkRootLevelContainer.h>
+#include <Havok/Physics2012/Utilities/Serialize/hkpPhysicsData.h>
+#include "KingSystem/Physics/StaticCompound/physStaticCompoundBodyGroup.h"
+#include "KingSystem/Physics/StaticCompound/physStaticCompoundInfo.h"
+#include "KingSystem/Utils/Debug.h"
+#include "KingSystem/Utils/HeapUtil.h"
 
 namespace ksys::phys {
 
+// + 1 because body groups are one indexed, with the 0th group acting as a dummy group
+constexpr int NumBodyGroups = 16 + 1;
+
 StaticCompound::StaticCompound() = default;
 
 StaticCompound::~StaticCompound() {
     // FIXME
 }
 
-FieldBodyGroup* StaticCompound::getFieldBodyGroup(int idx) {
+void StaticCompound::doCreate_(u8* buffer, u32 buffer_size, sead::Heap* parent_heap) {
+    const char* error = nullptr;
+    mBuffer = buffer;
+    auto result = hkNativePackfileUtils::validatePackfileHeader(mBuffer, &error);
+    if (!result.isSuccess())
+        util::PrintDebugFmt("%s: invalid packfile: %s", __func__, error);
+    mBufferSize = hkNativePackfileUtils::getRequiredBufferSize(mBuffer, int(buffer_size));
+    mStaticCompoundInfo = static_cast<const StaticCompoundInfo*>(
+        hkNativePackfileUtils::loadInPlace(mBuffer, mBufferSize, nullptr, &error));
+
+    mContainerBuffer = buffer + mStaticCompoundInfo->m_Offset;
+    auto result2 = hkNativePackfileUtils::validatePackfileHeader(mContainerBuffer, &error);
+    if (!result2.isSuccess())
+        util::PrintDebugFmt("%s: invalid physics packfile: %s", __func__, error);
+    mContainerBufferSize =
+        hkNativePackfileUtils::getRequiredBufferSize(mContainerBuffer, int(buffer_size));
+    auto* container = static_cast<hkRootLevelContainer*>(
+        hkNativePackfileUtils::loadInPlace(mContainerBuffer, mContainerBufferSize));
+
+    auto* physics_data = container->findObject<hkpPhysicsData>();
+
+    mHeap = util::tryCreateDualHeap(parent_heap);
+    if (!mHeap) {
+        util::PrintDebugFmt("%s: failed to create dual heap", __func__);
+        return;
+    }
+    mHeap->enableWarning(false);
+    auto* heap = mHeap;
+
+    const auto& systems = physics_data->getPhysicsSystems();
+    mFieldBodyGroups.allocBufferAssert(NumBodyGroups, heap);
+    for (int i = 0, n = mFieldBodyGroups.size(); i < n; ++i)
+        mFieldBodyGroups[i].init(*systems[i], &mMtx, this, heap);
+
+    mName = systems[0]->getName();
+    mName.trimMatchedString("_0");
+
+    if (mStaticCompoundInfo->m_ActorInfo.getSize() > 0)
+        mMapObjects.allocBufferAssert(mStaticCompoundInfo->m_ActorInfo.getSize(), heap);
+
+    mFlags.set(Flag::Initialised);
+    mHeap->adjust();
+}
+
+void StaticCompound::setMapObject(u32 hash_id, u32 srt_hash, map::Object* object) {
+    int idx = mStaticCompoundInfo->getActorIdx(hash_id, srt_hash);
+    if (idx < 0 || idx >= mMapObjects.size())
+        return;
+    mMapObjects[idx] = object;
+}
+
+map::Object* StaticCompound::getMapObject(int shape_idx) const {
+    auto* info = mStaticCompoundInfo->getShapeInfo(shape_idx);
+    if (!info)
+        return nullptr;
+
+    const int idx = info->m_ActorInfoIndex;
+    if (!mMapObjects.isIndexValid(idx))
+        return nullptr;
+
+    return mMapObjects[idx];
+}
+
+bool StaticCompound::disableCollision(int actor_idx, bool x) {
+    const int start = mStaticCompoundInfo->getShapeInfoStart(actor_idx);
+    const int end = mStaticCompoundInfo->getShapeInfoEnd(actor_idx);
+    for (int i = start; i <= end; ++i) {
+        auto* info = mStaticCompoundInfo->getShapeInfo(i);
+        if (!info || info->m_ActorInfoIndex < 0)
+            continue;
+
+        const auto group = info->m_BodyGroup;
+        const auto type = static_cast<BodyLayerType>(info->m_BodyLayerType);
+        const auto instance_id = info->m_InstanceId;
+
+        mFieldBodyGroups[group].disableCollision(type, instance_id, x);
+    }
+    return true;
+}
+
+BodyGroup* StaticCompound::getFieldBodyGroup(int idx) {
     return &mFieldBodyGroups[idx];
 }
 
-bool StaticCompound::hasFieldBodyGroup(FieldBodyGroup* group) const {
+bool StaticCompound::hasFieldBodyGroup(BodyGroup* group) const {
     for (int i = 0, n = mFieldBodyGroups.size(); i < n; ++i) {
         if (&mFieldBodyGroups[i] == group)
             return true;
-- 
cgit v1.2.3