From e512302305da9a5111f6423fcde55bd6a9e1f053 Mon Sep 17 00:00:00 2001
From: Lynix <lynix680@gmail.com>
Date: Sun, 6 Aug 2017 23:46:55 +0200
Subject: [PATCH] RigidBody2D: Add experimental SetMomentOfInertia method

---
 include/Nazara/Physics2D/RigidBody2D.hpp | 1 +
 src/Nazara/Physics2D/RigidBody2D.cpp     | 8 ++++++++
 2 files changed, 9 insertions(+)

diff --git a/include/Nazara/Physics2D/RigidBody2D.hpp b/include/Nazara/Physics2D/RigidBody2D.hpp
index 1af3ea89c..f2b3da2d7 100644
--- a/include/Nazara/Physics2D/RigidBody2D.hpp
+++ b/include/Nazara/Physics2D/RigidBody2D.hpp
@@ -57,6 +57,7 @@ namespace Nz
 			void SetGeom(Collider2DRef geom);
 			void SetMass(float mass);
 			void SetMassCenter(const Vector2f& center);
+			void SetMomentOfInertia(float moment);
 			void SetPosition(const Vector2f& position);
 			void SetRotation(float rotation);
 			void SetUserdata(void* ud);
diff --git a/src/Nazara/Physics2D/RigidBody2D.cpp b/src/Nazara/Physics2D/RigidBody2D.cpp
index 85d43ee76..370b01b3b 100644
--- a/src/Nazara/Physics2D/RigidBody2D.cpp
+++ b/src/Nazara/Physics2D/RigidBody2D.cpp
@@ -285,6 +285,14 @@ namespace Nz
 			cpBodySetCenterOfGravity(m_handle, cpv(center.x, center.y));
 	}
 
+	void RigidBody2D::SetMomentOfInertia(float moment)
+	{
+		m_world->RegisterPostStep(this, [moment] (Nz::RigidBody2D* body)
+		{
+			cpBodySetMoment(body->GetHandle(), moment);
+		});
+	}
+
 	void RigidBody2D::SetPosition(const Vector2f& position)
 	{
 		cpBodySetPosition(m_handle, cpv(position.x, position.y));