1 files changed, 20 insertions, 2 deletions

diff --git a/src/modelinfo/WeaponModelInfo.cpp b/src/modelinfo/WeaponModelInfo.cpp
index d9294c3f..1d194ec4 100644
--- a/src/modelinfo/WeaponModelInfo.cpp
+++ b/src/modelinfo/WeaponModelInfo.cpp
@@ -4,6 +4,9 @@
 #include "AnimManager.h"
 #include "VisibilityPlugins.h"
 
+base::cRelocatableChunkClassInfo CWeaponModelInfo::msClassInfo("CWeaponModelInfo", VTABLE_ADDR(&msClassInstance), sizeof(msClassInstance));
+CWeaponModelInfo CWeaponModelInfo::msClassInstance;
+
 void
 CWeaponModelInfo::SetAnimFile(const char *file)
 {
@@ -35,19 +38,34 @@ CWeaponModelInfo::Init(void)
 void
 CWeaponModelInfo::SetWeaponInfo(int32 weaponId)
 {
-	m_atomics[2] = (RpAtomic*)weaponId;
+	m_relatedModel = (CSimpleModelInfo*)weaponId;
 }
 
 eWeaponType
 CWeaponModelInfo::GetWeaponInfo(void)
 {
-	return (eWeaponType)(uintptr)m_atomics[2];
+	return (eWeaponType)(uintptr)m_relatedModel;
 }
 
+/*
 void
 CWeaponModelInfo::SetAtomic(int n, RpAtomic *atomic)
 {
 	CSimpleModelInfo::SetAtomic(n, atomic);
 	CVisibilityPlugins::SetAtomicRenderCallback(atomic, CVisibilityPlugins::RenderWeaponCB);
 }
+*/
+
+void
+CWeaponModelInfo::RcWriteThis(base::cRelocatableChunkWriter &writer)
+{
+	writer.AllocateRaw(this, sizeof(*this), sizeof(void*), false, true);
+	writer.Class(VTABLE_ADDR(this), msClassInfo);
+}
 
+void
+CWeaponModelInfo::RcWriteEmpty(base::cRelocatableChunkWriter &writer)
+{
+	writer.AllocateRaw(this, sizeof(*this), sizeof(void*), false, true);
+	writer.Class(VTABLE_ADDR(this), msClassInfo);
+}