Files
CosmicEngine/lib/All/JoltPhysics/Samples/Tests/Rig/LoadSaveBinaryRigTest.cpp

58 lines
1.8 KiB
C++
Raw Normal View History

// Jolt Physics Library (https://github.com/jrouwe/JoltPhysics)
// SPDX-FileCopyrightText: 2021 Jorrit Rouwe
// SPDX-License-Identifier: MIT
#include <TestFramework.h>
#include <Tests/Rig/LoadSaveBinaryRigTest.h>
#include <Jolt/Core/StreamWrapper.h>
#include <Jolt/Physics/Constraints/DistanceConstraint.h>
#include <Utils/Log.h>
#include <Utils/RagdollLoader.h>
JPH_IMPLEMENT_RTTI_VIRTUAL(LoadSaveBinaryRigTest)
{
JPH_ADD_BASE_CLASS(LoadSaveBinaryRigTest, Test)
}
LoadSaveBinaryRigTest::~LoadSaveBinaryRigTest()
{
mRagdoll->RemoveFromPhysicsSystem();
}
void LoadSaveBinaryRigTest::Initialize()
{
// Floor
CreateFloor();
stringstream data;
{
// Load ragdoll
Ref<RagdollSettings> settings = RagdollLoader::sLoad("Human.tof", EMotionType::Dynamic);
// Add an additional constraint between the left and right arm to test loading/saving of additional constraints
const Skeleton *skeleton = settings->GetSkeleton();
int left_arm = skeleton->GetJointIndex("L_Wrist_sjnt_0");
int right_arm = skeleton->GetJointIndex("R_Wrist_sjnt_0");
Ref<DistanceConstraintSettings> constraint = new DistanceConstraintSettings;
constraint->mSpace = EConstraintSpace::LocalToBodyCOM;
constraint->mMaxDistance = 0.1f;
constraint->mMinDistance = 0.1f;
settings->mAdditionalConstraints.push_back(RagdollSettings::AdditionalConstraint(left_arm, right_arm , constraint));
// Save it to a binary stream
StreamOutWrapper stream_out(data);
settings->SaveBinaryState(stream_out, true /* Save shape */, true /* Save group filter */);
}
StreamInWrapper stream_in(data);
RagdollSettings::RagdollResult result = RagdollSettings::sRestoreFromBinaryState(stream_in);
if (result.HasError())
FatalError(result.GetError().c_str());
// Create ragdoll
mRagdoll = result.Get()->CreateRagdoll(0, 0, mPhysicsSystem);
mRagdoll->AddToPhysicsSystem(EActivation::Activate);
}