engineConvertModelToType

Author: FileEX

Client/game_sa/CModelInfoSA.cpp

@@ -1541,6 +1541,8 @@ void CModelInfoSA::RestoreOriginalModel()
         if (currentInterface)
         {
             ppModelInfo[m_dwModelID]->usNumberOfRefs = currentInterface->usNumberOfRefs;
+            ppModelInfo[m_dwModelID]->pColModel = currentInterface->pColModel;
+            
             delete currentInterface;
         }
 
@@ -1818,6 +1820,9 @@ void CModelInfoSA::MakeClumpModel(ushort usBaseID)
 
 bool CModelInfoSA::ConvertToClump()
 {
+    if (GetModelType() == eModelInfoType::CLUMP)
+        return false;
+
     // Get current interface
     CBaseModelInfoSAInterface* currentModelInterface = ppModelInfo[m_dwModelID];
     if (!currentModelInterface)
@@ -1838,34 +1843,94 @@ bool CModelInfoSA::ConvertToClump()
     // Set new interface for ModelInfo
     ppModelInfo[m_dwModelID] = newClumpInterface;
 
-    // Store original interface
-    MapSet(m_convertedModelInterfaces, m_dwModelID, currentModelInterface);
+    // Store original (only) interface
+    if (!MapContains(m_convertedModelInterfaces, m_dwModelID))
+        MapSet(m_convertedModelInterfaces, m_dwModelID, currentModelInterface);
+    else
+        m_lastConversionInterface = currentModelInterface;
+
     return true;
 }
 
-bool CModelInfoSA::ConvertToAtomic()
+bool CModelInfoSA::ConvertToAtomic(bool damageable)
 {
     // Get current interface
-    CClumpModelInfoSAInterface* currentClumpInterface = static_cast<CClumpModelInfoSAInterface*>(ppModelInfo[m_dwModelID]);
-    if (!currentClumpInterface)
+    CBaseModelInfoSAInterface* currentModelInterface = ppModelInfo[m_dwModelID];
+    if (!currentModelInterface)
+        return false;
+
+    if (GetModelType() == eModelInfoType::ATOMIC && ((damageable && currentModelInterface->IsDamageAtomicVTBL()) || (!damageable && currentModelInterface->IsAtomicVTBL())))
         return false;
 
     // Create new atomic interface
-    CAtomicModelInfoSAInterface* newAtomicInterface = new CAtomicModelInfoSAInterface();
-    MemCpyFast(newAtomicInterface, currentClumpInterface, sizeof(CAtomicModelInfoSAInterface));
+    CAtomicModelInfoSAInterface* newAtomicInterface = nullptr;
+    CDamageableModelInfoSAInterface* newDamageableAtomicInterface = nullptr;
 
     // (FileEX): We do not destroy or set pRwObject to nullptr here
     // because our IsLoaded code expects the RwObject to exist.
     // We destroy the old RwObject in CRenderWareSA::ReplaceAllAtomicsInModel after passing the IsLoaded condition in the SetCustomModel.
 
-    // Set CAtomicModelInfo vtbl after copying data
-    newAtomicInterface->VFTBL = reinterpret_cast<CBaseModelInfo_SA_VTBL*>(VTBL_CAtomicModelInfo);
+    if (damageable)
+    {
+        newDamageableAtomicInterface = new CDamageableModelInfoSAInterface();
+        MemCpyFast(newDamageableAtomicInterface, currentModelInterface, sizeof(CDamageableModelInfoSAInterface));
+        newDamageableAtomicInterface->m_damagedAtomic = nullptr;
+
+        // Set CDamageAtomicModelInfo vtbl after copying data
+        newDamageableAtomicInterface->VFTBL = reinterpret_cast<CBaseModelInfo_SA_VTBL*>(VTBL_CDamageAtomicModelInfo);
+    }
+    else
+    {
+        newAtomicInterface = new CAtomicModelInfoSAInterface();
+        MemCpyFast(newAtomicInterface, currentModelInterface, sizeof(CAtomicModelInfoSAInterface));
+
+        // Set CAtomicModelInfo vtbl after copying data
+        newAtomicInterface->VFTBL = reinterpret_cast<CBaseModelInfo_SA_VTBL*>(VTBL_CAtomicModelInfo);
+    }
 
     // Set new interface for ModelInfo
-    ppModelInfo[m_dwModelID] = newAtomicInterface;
+    ppModelInfo[m_dwModelID] = damageable ? newDamageableAtomicInterface : newAtomicInterface;
+
+    // Store original (only) interface
+    if (!MapContains(m_convertedModelInterfaces, m_dwModelID))
+        MapSet(m_convertedModelInterfaces, m_dwModelID, currentModelInterface);
+    else
+        m_lastConversionInterface = currentModelInterface;
+
+    return true;
+}
+
+bool CModelInfoSA::ConvertToTimedObject()
+{
+    if (GetModelType() == eModelInfoType::TIME)
+        return false;
+
+    // Get current interface
+    CBaseModelInfoSAInterface* currentModelInterface = ppModelInfo[m_dwModelID];
+    if (!currentModelInterface)
+        return false;
+
+    // Create new interface
+    CTimeModelInfoSAInterface* newTimedInterface = new CTimeModelInfoSAInterface();
+    MemCpyFast(newTimedInterface, currentModelInterface, sizeof(CTimeModelInfoSAInterface));
+    newTimedInterface->timeInfo.m_wOtherTimeModel = 0;
+
+    // (FileEX): We do not destroy or set pRwObject to nullptr here
+    // because our IsLoaded code expects the RwObject to exist.
+    // We destroy the old RwObject in CRenderWareSA::ReplaceAllAtomicsInModel after passing the IsLoaded condition in the SetCustomModel.
+
+    // Set CTimeModelInfo vtbl after copying data
+    newTimedInterface->VFTBL = reinterpret_cast<CBaseModelInfo_SA_VTBL*>(VTBL_CTimeModelInfo);
+
+    // Set new interface for ModelInfo
+    ppModelInfo[m_dwModelID] = newTimedInterface;
+
+    // Store original (only) interface
+    if (!MapContains(m_convertedModelInterfaces, m_dwModelID))
+        MapSet(m_convertedModelInterfaces, m_dwModelID, currentModelInterface);
+    else
+        m_lastConversionInterface = currentModelInterface;
 
-    // Store original interface
-    MapSet(m_convertedModelInterfaces, m_dwModelID, currentClumpInterface);
     return true;
 }
 

Client/game_sa/CModelInfoSA.h

@@ -66,6 +66,8 @@ static void* ARRAY_ModelInfo = *(void**)(0x403DA4 + 3);
 
 #define VTBL_CClumpModelInfo 0x85BD30
 #define VTBL_CAtomicModelInfo 0x85BBF0
+#define VTBL_CDamageAtomicModelInfo 0x85BC30
+#define VTBL_CTimeModelInfo         0x85BCB0
 
 class CBaseModelInfoSAInterface;
 class CModelInfoSAInterface
@@ -235,6 +237,12 @@ class CBaseModelInfoSAInterface
     // +726 = Word array as referenced in CVehicleModelInfo::GetVehicleUpgrade(int)
     // +762 = Array of WORD containing something relative to paintjobs
     // +772 = Anim file index
+
+    void DeleteRwObject() { ((void(__thiscall*)(void*))VFTBL->DeleteRwObject)(this); }
+
+    bool IsAtomicVTBL() const { return VFTBL == reinterpret_cast<CBaseModelInfo_SA_VTBL*>(VTBL_CAtomicModelInfo); }
+    bool IsDamageAtomicVTBL() const { return VFTBL == reinterpret_cast<CBaseModelInfo_SA_VTBL*>(VTBL_CDamageAtomicModelInfo); }
+    bool IsClumpVTBL() const { return VFTBL == reinterpret_cast<CBaseModelInfo_SA_VTBL*>(VTBL_CClumpModelInfo); }
 };
 static_assert(sizeof(CBaseModelInfoSAInterface) == 0x20, "Invalid size for CBaseModelInfoSAInterface");
 
@@ -267,6 +275,13 @@ class CAtomicModelInfoSAInterface : public CBaseModelInfoSAInterface
     void SetAtomic(RpAtomic* atomic) { ((void(__thiscall*)(CAtomicModelInfoSAInterface*, RpAtomic*))0x4C4360)(this, atomic); }
 };
 
+class CLodAtomicModelInfoSAInterface : public CAtomicModelInfoSAInterface
+{
+public:
+    std::int16_t numChildren; // num child higher level LODs
+    std::int16_t numChildrenRendered; // num child higher level LODs that have been rendered
+};
+
 class CTimeModelInfoSAInterface : public CAtomicModelInfoSAInterface
 {
 public:
@@ -286,7 +301,7 @@ class CDamageableModelInfoSAInterface : public CAtomicModelInfoSAInterface
 public:
     RpAtomic* m_damagedAtomic;
 
-    void SetDamagedAtomic(RpAtomic* atomic) { ((void(__thiscall*)(RpAtomic*))FUNC_CAtomicModelInfo_SetAtomic)(atomic); }
+    void SetDamagedAtomic(RpAtomic* atomic) { ((void(__thiscall*)(CDamageableModelInfoSAInterface*, RpAtomic*))0x4C48D0)(this, atomic); }
 };
 
 class CVehicleModelVisualInfoSAInterface            // Not sure about this name. If somebody knows more, please change
@@ -371,6 +386,7 @@ class CModelInfoSA : public CModelInfo
     static std::map<unsigned short, int>                                         ms_DefaultTxdIDMap;
     SVehicleSupportedUpgrades                                                    m_ModelSupportedUpgrades;
 
+    CBaseModelInfoSAInterface* m_lastConversionInterface{nullptr};
     // Store original model interfaces after conersion clump->atomic or atomic->clump
     static std::unordered_map<std::uint32_t, CBaseModelInfoSAInterface*> m_convertedModelInterfaces;
 
@@ -508,9 +524,12 @@ class CModelInfoSA : public CModelInfo
     static void    RestoreAllObjectsPropertiesGroups();
 
     // Model type conversion functions
-    bool ConvertToClump();
-    bool ConvertToAtomic();
-    CBaseModelInfoSAInterface* GetOriginalInterface() const;
+    bool ConvertToClump() override;
+    bool ConvertToAtomic(bool damageable) override;
+    bool ConvertToTimedObject() override;
+    CBaseModelInfoSAInterface* GetLastConversionInterface() const noexcept override { return m_lastConversionInterface; }
+    void                       SetLastConversionInterface(CBaseModelInfoSAInterface* lastInterace) noexcept override { m_lastConversionInterface = lastInterace; }
+    CBaseModelInfoSAInterface* GetOriginalInterface() const override;
 
     // Vehicle towing functions
     bool IsTowableBy(CModelInfo* towingModel) override;