initial commit, 4.5 stable

thirdparty/jolt_physics/Jolt/Physics/Constraints/CalculateSolverSteps.h

@@ -0,0 +1,66 @@
+// Jolt Physics Library (https://github.com/jrouwe/JoltPhysics)
+// SPDX-FileCopyrightText: 2023 Jorrit Rouwe
+// SPDX-License-Identifier: MIT
+
+#pragma once
+
+#include <Jolt/Physics/PhysicsSettings.h>
+
+JPH_NAMESPACE_BEGIN
+
+/// Class used to calculate the total number of velocity and position steps
+class CalculateSolverSteps
+{
+public:
+	/// Constructor
+	JPH_INLINE explicit			CalculateSolverSteps(const PhysicsSettings &inSettings) : mSettings(inSettings) { }
+
+	/// Combine the number of velocity and position steps for this body/constraint with the current values
+	template <class Type>
+	JPH_INLINE void				operator () (const Type *inObject)
+	{
+		uint num_velocity_steps = inObject->GetNumVelocityStepsOverride();
+		mNumVelocitySteps = max(mNumVelocitySteps, num_velocity_steps);
+		mApplyDefaultVelocity |= num_velocity_steps == 0;
+
+		uint num_position_steps = inObject->GetNumPositionStepsOverride();
+		mNumPositionSteps = max(mNumPositionSteps, num_position_steps);
+		mApplyDefaultPosition |= num_position_steps == 0;
+	}
+
+	/// Must be called after all bodies/constraints have been processed
+	JPH_INLINE void				Finalize()
+	{
+		// If we have a default velocity/position step count, take the max of the default and the overrides
+		if (mApplyDefaultVelocity)
+			mNumVelocitySteps = max(mNumVelocitySteps, mSettings.mNumVelocitySteps);
+		if (mApplyDefaultPosition)
+			mNumPositionSteps = max(mNumPositionSteps, mSettings.mNumPositionSteps);
+	}
+
+	/// Get the results of the calculation
+	JPH_INLINE uint				GetNumPositionSteps() const					{ return mNumPositionSteps; }
+	JPH_INLINE uint				GetNumVelocitySteps() const					{ return mNumVelocitySteps; }
+
+private:
+	const PhysicsSettings &		mSettings;
+
+	uint						mNumVelocitySteps = 0;
+	uint						mNumPositionSteps = 0;
+
+	bool						mApplyDefaultVelocity = false;
+	bool						mApplyDefaultPosition = false;
+};
+
+/// Dummy class to replace the steps calculator when we don't need the result
+class DummyCalculateSolverSteps
+{
+public:
+	template <class Type>
+	JPH_INLINE void				operator () (const Type *) const
+	{
+		/* Nothing to do */
+	}
+};
+
+JPH_NAMESPACE_END