Restore default exhaust fumes position when disconnecting

Author: jushar

Client/game_sa/CGameSA.cpp

@@ -467,6 +467,9 @@ void CGameSA::Reset ( void )
         // Restore the HUD
         m_pHud->Disable ( false );
         m_pHud->SetComponentVisible ( HUD_ALL, true );
+
+        // Restore model exhaust fumes positions
+        CModelInfoSA::ResetAllVehicleExhaustFumes();
     }
 }
 

Client/game_sa/CModelInfoSA.cpp

@@ -24,6 +24,7 @@ std::map < unsigned short, int > CModelInfoSA::ms_RestreamTxdIDMap;
 std::map < DWORD, float > CModelInfoSA::ms_ModelDefaultLodDistanceMap;
 std::set < uint > CModelInfoSA::ms_ReplacedColModels;
 std::map < DWORD, BYTE > CModelInfoSA::ms_ModelDefaultAlphaTransparencyMap;
+std::unordered_map<CVehicleModelInfoSAInterface*, CVector> CModelInfoSA::ms_ModelDefaultVehicleFumesPosition;
 
 CModelInfoSA::CModelInfoSA ( void )
 {
@@ -951,10 +952,28 @@ void CModelInfoSA::SetVehicleExhaustFumesPosition(const CVector& position)
     if (!IsVehicle())
         return;
 
+    // Store default position in map
     auto pVehicleModel = reinterpret_cast<CVehicleModelInfoSAInterface*>(m_pInterface);
+    auto iter = ms_ModelDefaultVehicleFumesPosition.find(pVehicleModel);
+    if (iter == ms_ModelDefaultVehicleFumesPosition.end())
+    {
+        ms_ModelDefaultVehicleFumesPosition.insert({ pVehicleModel, pVehicleModel->pVisualInfo->exhaustPosition });
+    }
+
+    // Set fumes position
     pVehicleModel->pVisualInfo->exhaustPosition = position;
 }
 
+void CModelInfoSA::ResetAllVehicleExhaustFumes()
+{
+    for (auto& info : ms_ModelDefaultVehicleFumesPosition)
+    {
+        CVehicleModelInfoSAInterface* pVehicleModel = info.first;
+        pVehicleModel->pVisualInfo->exhaustPosition = info.second;
+    }
+    ms_ModelDefaultVehicleFumesPosition.clear();
+}
+
 void CModelInfoSA::SetCustomModel ( RpClump* pClump )
 {
     // Error

Client/game_sa/CModelInfoSA.h

@@ -270,8 +270,10 @@ class CModelInfoSA : public CModelInfo
     static std::map < unsigned short, int > ms_RestreamTxdIDMap;
     static std::map < DWORD, float > ms_ModelDefaultLodDistanceMap;
     static std::map < DWORD, BYTE > ms_ModelDefaultAlphaTransparencyMap;
+    static std::unordered_map<CVehicleModelInfoSAInterface*, CVector> ms_ModelDefaultVehicleFumesPosition;
     bool                            m_bAddedRefForCollision;
     SVehicleSupportedUpgrades       m_ModelSupportedUpgrades;
+
 public:
     static std::set < uint >        ms_ReplacedColModels;
 
@@ -340,6 +342,7 @@ class CModelInfoSA : public CModelInfo
     void*                           SetVehicleSuspensionData( void* pSuspensionLines );
     CVector                         GetVehicleExhaustFumesPosition() override;
     void                            SetVehicleExhaustFumesPosition(const CVector& position) override;
+    static void                     ResetAllVehicleExhaustFumes();
 
     // ONLY use for peds
     void                            GetVoice                ( short* psVoiceType, short* psVoice );