Add NiAVObject::GetMass

powerof3 · May 28, 2024 · f71d05c · f71d05c
1 parent 51593d0
commit f71d05c
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/include/RE/N/NiAVObject.h b/include/RE/N/NiAVObject.h
@@ -120,6 +120,7 @@ namespace RE
 		[[nodiscard]] bhkCollisionObject* GetCollisionObject() const;
 		[[nodiscard]] COL_LAYER           GetCollisionLayer() const;
 		[[nodiscard]] BSGeometry*         GetFirstGeometryOfShaderType(BSShaderMaterial::Feature a_type);
+		[[nodiscard]] float               GetMass();
 		[[nodiscard]] TESObjectREFR*      GetUserData() const;
 		[[nodiscard]] bool                HasAnimation() const;
 		[[nodiscard]] bool                HasShaderType(BSShaderMaterial::Feature a_type);

diff --git a/src/RE/N/NiAVObject.cpp b/src/RE/N/NiAVObject.cpp
@@ -99,6 +99,20 @@ namespace RE
 		return firstGeometry;
 	}
 
+	float NiAVObject::GetMass()
+	{
+		float mass = 0.0f;
+
+		BSVisit::TraverseScenegraphCollision(this, [&](bhkNiCollisionObject* a_col) -> BSVisit::BSVisitControl {
+			if (auto hkpBody = a_col->body ? static_cast<RE::hkpRigidBody*>(a_col->body->referencedObject.get()) : nullptr) {
+				mass += hkpBody->motion.GetMass();
+			}
+			return BSVisit::BSVisitControl::kContinue;
+		});
+
+		return mass;
+	}
+
 	TESObjectREFR* NiAVObject::GetUserData() const
 	{
 		if (userData) {