6 files changed, 324 insertions, 1 deletions

diff --git a/src/control/GameLogic.cpp b/src/control/GameLogic.cpp
new file mode 100644
index 00000000..16727960
--- /dev/null
+++ b/src/control/GameLogic.cpp
@@ -0,0 +1,293 @@
+#include "common.h"
+#include "patcher.h"
+#include "GameLogic.h"

+#include "Clock.h"

+#include "Stats.h"

+#include "Pickups.h"

+#include "Timer.h"

+#include "Streaming.h"

+#include "CutsceneMgr.h"

+#include "World.h"

+#include "PlayerPed.h"

+#include "Camera.h"

+#include "Messages.h"

+#include "CarCtrl.h"

+#include "Restart.h"

+#include "Pad.h"

+#include "References.h"

+#include "Fire.h"

+#include "Script.h"

+#include "Garages.h"

+

+uint8 CGameLogic::ActivePlayers; // 0x95CD5E

+

+void

+CGameLogic::InitAtStartOfGame()

+{

+	ActivePlayers = 1;

+}

+

+void

+CGameLogic::PassTime(uint32 time)

+{

+	uint8 minutes, hours, days;

+

+	minutes = time + CClock::GetMinutes();

+	hours = CClock::GetHours();

+

+	for (; minutes >= 60; minutes -= 60)

+		hours++;

+

+	if (hours > 23) {

+		days = CStats::DaysPassed;

+		for (; hours >= 24; hours -= 24)

+			days++;

+		CStats::DaysPassed = days;

+	}

+

+	CClock::SetGameClock(hours, minutes);

+	CPickups::PassTime(time * 1000);

+}

+

+void 

+CGameLogic::SortOutStreamingAndMemory(const CVector &pos)

+{

+	CTimer::Stop();

+	CStreaming::FlushRequestList();

+	CStreaming::DeleteRwObjectsAfterDeath(pos);

+	CStreaming::RemoveUnusedModelsInLoadedList();

+	CGame::DrasticTidyUpMemory();

+	CStreaming::LoadScene(pos);

+	CTimer::Update();

+}

+

+void

+CGameLogic::Update()

+{

+	CVector vecRestartPos;

+	float fRestartFloat;

+

+	if (CCutsceneMgr::ms_cutsceneProcessing) return;

+

+	CPlayerInfo &pPlayerInfo = CWorld::Players[CWorld::PlayerInFocus];

+	switch (pPlayerInfo.m_WBState) {

+	case WBSTATE_PLAYING:

+		if (pPlayerInfo.m_pPed->m_nPedState == PED_DEAD) {

+			pPlayerInfo.m_pPed->ClearAdrenaline();

+			pPlayerInfo.KillPlayer();

+		}

+		if (pPlayerInfo.m_pPed->m_nPedState == PED_ARRESTED) {

+			pPlayerInfo.m_pPed->ClearAdrenaline();

+			pPlayerInfo.ArrestPlayer();

+		}

+		break;

+	case WBSTATE_WASTED:

+		if ((CTimer::GetTimeInMilliseconds() - pPlayerInfo.m_nWBTime > 0x800) && (CTimer::GetPreviousTimeInMilliseconds() - pPlayerInfo.m_nWBTime <= 0x800)) {

+			TheCamera.SetFadeColour(200, 200, 200);

+			TheCamera.Fade(2.0f, FADE_OUT);

+		}

+

+		if (CTimer::GetTimeInMilliseconds() - pPlayerInfo.m_nWBTime >= 0x1000) {

+			pPlayerInfo.m_WBState = WBSTATE_PLAYING;

+			if (pPlayerInfo.m_bGetOutOfHospitalFree) {

+				pPlayerInfo.m_bGetOutOfHospitalFree = false;

+			} else {

+				pPlayerInfo.m_nMoney = max(0, pPlayerInfo.m_nMoney - 1000);

+				pPlayerInfo.m_pPed->ClearWeapons();

+			}

+

+			if (pPlayerInfo.m_pPed->bInVehicle) {

+				CVehicle *pVehicle = pPlayerInfo.m_pPed->m_pMyVehicle;

+				if (pVehicle != nil) {

+					if (pVehicle->pDriver == pPlayerInfo.m_pPed) {

+						pVehicle->pDriver = nil;

+						if (pVehicle->m_status != STATUS_WRECKED)

+							pVehicle->m_status = STATUS_ABANDONED;

+					} else

+						pVehicle->RemovePassenger(pPlayerInfo.m_pPed);

+				}

+			}

+			CEventList::Initialise();

+			CMessages::ClearMessages();

+			CCarCtrl::ClearInterestingVehicleList();

+			CWorld::ClearExcitingStuffFromArea(pPlayerInfo.GetPos(), 4000.0f, 1);

+			CRestart::FindClosestHospitalRestartPoint(pPlayerInfo.GetPos(), &vecRestartPos, &fRestartFloat);

+			CRestart::OverrideHospitalLevel = false;

+			CRestart::OverridePoliceStationLevel = false;

+			PassTime(720);

+			RestorePlayerStuffDuringResurrection(pPlayerInfo.m_pPed, vecRestartPos, fRestartFloat);

+			SortOutStreamingAndMemory(pPlayerInfo.GetPos());

+			TheCamera.m_fCamShakeForce = 0.0f;

+			TheCamera.SetMotionBlur(0, 0, 0, 0, MBLUR_NONE);

+			CPad::GetPad(0)->StopShaking(0);

+			CReferences::RemoveReferencesToPlayer();

+			CCarCtrl::CountDownToCarsAtStart = 2;

+			CPad::GetPad(CWorld::PlayerInFocus)->DisablePlayerControls = PLAYERCONTROL_ENABLED;

+			if (CRestart::bFadeInAfterNextDeath) { 

+				TheCamera.SetFadeColour(200, 200, 200);

+				TheCamera.Fade(4.0f, FADE_IN);

+			} else CRestart::bFadeInAfterNextDeath = true;

+		}

+		break;

+	case WBSTATE_BUSTED:

+		if ((CTimer::GetTimeInMilliseconds() - pPlayerInfo.m_nWBTime > 0x800) && (CTimer::GetPreviousTimeInMilliseconds() - pPlayerInfo.m_nWBTime <= 0x800)) {

+			TheCamera.SetFadeColour(0, 0, 0);

+			TheCamera.Fade(2.0f, FADE_OUT);

+		}

+		if (CTimer::GetTimeInMilliseconds() - pPlayerInfo.m_nWBTime >= 0x1000) {

+			pPlayerInfo.m_WBState = WBSTATE_PLAYING;

+			int takeMoney;

+

+			switch (pPlayerInfo.m_pPed->m_pWanted->m_nWantedLevel) {

+			case 0:

+			case 1:

+				takeMoney = 100;

+				break;

+			case 2:

+				takeMoney = 200;

+				break;

+			case 3:

+				takeMoney = 400;

+				break;

+			case 4:

+				takeMoney = 600;

+				break;

+			case 5:

+				takeMoney = 900;

+				break;

+			case 6:

+				takeMoney = 1500;

+				break;

+			}

+			if (pPlayerInfo.m_bGetOutOfJailFree) {

+				pPlayerInfo.m_bGetOutOfJailFree = false;

+			} else {

+				pPlayerInfo.m_nMoney = max(0, pPlayerInfo.m_nMoney - takeMoney);

+				pPlayerInfo.m_pPed->ClearWeapons();

+			}

+

+			if (pPlayerInfo.m_pPed->bInVehicle) {

+				CVehicle *pVehicle = pPlayerInfo.m_pPed->m_pMyVehicle;

+				if (pVehicle != nil) {

+					if (pVehicle->pDriver == pPlayerInfo.m_pPed) {

+						pVehicle->pDriver = nil;

+						if (pVehicle->m_status != STATUS_WRECKED)

+							pVehicle->m_status = STATUS_ABANDONED;

+					}

+					else

+						pVehicle->RemovePassenger(pPlayerInfo.m_pPed);

+				}

+			}

+			CEventList::Initialise();

+			CMessages::ClearMessages();

+			CCarCtrl::ClearInterestingVehicleList();

+			CWorld::ClearExcitingStuffFromArea(pPlayerInfo.GetPos(), 4000.0f, 1);

+			CRestart::FindClosestPoliceRestartPoint(pPlayerInfo.GetPos(), &vecRestartPos, &fRestartFloat);

+			CRestart::OverrideHospitalLevel = false;

+			CRestart::OverridePoliceStationLevel = false;

+			PassTime(720);

+			RestorePlayerStuffDuringResurrection(pPlayerInfo.m_pPed, vecRestartPos, fRestartFloat);

+			pPlayerInfo.m_pPed->ClearWeapons();

+			SortOutStreamingAndMemory(pPlayerInfo.GetPos());

+			TheCamera.m_fCamShakeForce = 0.0f;

+			TheCamera.SetMotionBlur(0, 0, 0, 0, MBLUR_NONE);

+			CPad::GetPad(0)->StopShaking(0);

+			CReferences::RemoveReferencesToPlayer();

+			CCarCtrl::CountDownToCarsAtStart = 2;

+			CPad::GetPad(CWorld::PlayerInFocus)->DisablePlayerControls = PLAYERCONTROL_ENABLED;

+			if (CRestart::bFadeInAfterNextArrest) {

+				TheCamera.SetFadeColour(0, 0, 0);

+				TheCamera.Fade(4.0f, FADE_IN);

+			} else CRestart::bFadeInAfterNextArrest = true;

+		}

+		break;

+	case WBSTATE_FAILED_CRITICAL_MISSION:

+		if (CTimer::GetTimeInMilliseconds() - pPlayerInfo.m_nWBTime > 0x800 && CTimer::GetPreviousTimeInMilliseconds() - pPlayerInfo.m_nWBTime <= 0x800) {

+			TheCamera.SetFadeColour(0, 0, 0);

+			TheCamera.Fade(2.0f, FADE_OUT);

+		}

+		if (CTimer::GetTimeInMilliseconds() - pPlayerInfo.m_nWBTime >= 0x1000) {

+			pPlayerInfo.m_WBState = WBSTATE_PLAYING;

+			if (pPlayerInfo.m_pPed->bInVehicle) {

+				CVehicle *pVehicle = pPlayerInfo.m_pPed->m_pMyVehicle;

+				if (pVehicle != nil) {

+					if (pVehicle->pDriver == pPlayerInfo.m_pPed) {

+						pVehicle->pDriver = nil;

+						if (pVehicle->m_status != STATUS_WRECKED)

+							pVehicle->m_status = STATUS_ABANDONED;

+					} else

+						pVehicle->RemovePassenger(pPlayerInfo.m_pPed);

+				}

+			}

+			CEventList::Initialise();

+			CMessages::ClearMessages();

+			CCarCtrl::ClearInterestingVehicleList();

+			CWorld::ClearExcitingStuffFromArea(pPlayerInfo.GetPos(), 4000.0f, 1);

+			CRestart::FindClosestPoliceRestartPoint(pPlayerInfo.GetPos(), &vecRestartPos, &fRestartFloat);

+			CRestart::OverridePoliceStationLevel = false;

+			CRestart::OverrideHospitalLevel = false;

+			RestorePlayerStuffDuringResurrection(pPlayerInfo.m_pPed, vecRestartPos, fRestartFloat);

+			SortOutStreamingAndMemory(pPlayerInfo.GetPos());

+			TheCamera.m_fCamShakeForce = 0.0f;

+			TheCamera.SetMotionBlur(0, 0, 0, 0, MBLUR_NONE);

+			CPad::GetPad(0)->StopShaking(0);

+			CReferences::RemoveReferencesToPlayer();

+			CCarCtrl::CountDownToCarsAtStart = 2;

+			CPad::GetPad(CWorld::PlayerInFocus)->DisablePlayerControls = PLAYERCONTROL_ENABLED;

+			TheCamera.SetFadeColour(0, 0, 0);

+			TheCamera.Fade(4.0f, FADE_IN);

+		}

+		break;

+	case 4:

+		return;

+	}

+}

+

+void

+CGameLogic::RestorePlayerStuffDuringResurrection(CPlayerPed *pPlayerPed, CVector pos, float angle)

+{

+	pPlayerPed->m_fHealth = 100.0f;

+	pPlayerPed->m_fArmour = 0.0f;

+	pPlayerPed->bIsVisible = true;

+	pPlayerPed->m_bloodyFootprintCount = 0;

+	pPlayerPed->bDoBloodyFootprints = false;

+	pPlayerPed->ClearAdrenaline();

+	pPlayerPed->m_fCurrentStamina = pPlayerPed->m_fMaxStamina;

+	if (pPlayerPed->m_pFire)

+		pPlayerPed->m_pFire->Extinguish();

+	pPlayerPed->bInVehicle = false;

+	pPlayerPed->m_pMyVehicle = nil;

+	pPlayerPed->m_pVehicleAnim = nil;

+	pPlayerPed->m_pWanted->Reset();

+	pPlayerPed->RestartNonPartialAnims();

+	pPlayerPed->GetPlayerInfoForThisPlayerPed()->MakePlayerSafe(false);

+	pPlayerPed->bRemoveFromWorld = false;

+	pPlayerPed->ClearWeaponTarget();

+	pPlayerPed->SetInitialState();

+	CCarCtrl::ClearInterestingVehicleList();

+

+	pos.z += 1.0f;

+	pPlayerPed->Teleport(pos);

+	pPlayerPed->SetMoveSpeed(CVector(0.0f, 0.0f, 0.0f));

+

+	pPlayerPed->m_fRotationCur = DEGTORAD(angle);

+	pPlayerPed->m_fRotationDest = pPlayerPed->m_fRotationCur;

+	pPlayerPed->SetHeading(pPlayerPed->m_fRotationCur);

+	CTheScripts::ClearSpaceForMissionEntity(pos, pPlayerPed);

+	CWorld::ClearExcitingStuffFromArea(pos, 4000.0, 1);

+	pPlayerPed->RestoreHeadingRate();

+	TheCamera.SetCameraDirectlyInFrontForFollowPed_CamOnAString();

+	CReferences::RemoveReferencesToPlayer();

+	CGarages::PlayerArrestedOrDied();

+	CStats::CheckPointReachedUnsuccessfully();

+	CWorld::Remove(pPlayerPed);

+	CWorld::Add(pPlayerPed);

+}

+

+STARTPATCHES
+	InjectHook(0x4213F0, &CGameLogic::InitAtStartOfGame, PATCH_JUMP);
+	InjectHook(0x421C00, &CGameLogic::PassTime, PATCH_JUMP);
+	InjectHook(0x421A20, &CGameLogic::SortOutStreamingAndMemory, PATCH_JUMP);
+	InjectHook(0x421400, &CGameLogic::Update, PATCH_JUMP);
+	InjectHook(0x421A60, &CGameLogic::RestorePlayerStuffDuringResurrection, PATCH_JUMP);

+ENDPATCHES
\ No newline at end of file
diff --git a/src/control/GameLogic.h b/src/control/GameLogic.h
new file mode 100644
index 00000000..db626558
--- /dev/null
+++ b/src/control/GameLogic.h
@@ -0,0 +1,13 @@
+#pragma once

+

+class CGameLogic

+{

+public:

+	static void InitAtStartOfGame();

+	static void PassTime(uint32 time);

+	static void SortOutStreamingAndMemory(const CVector &pos);

+	static void Update();

+	static void RestorePlayerStuffDuringResurrection(class CPlayerPed *pPlayerPed, CVector pos, float angle);

+

+	static uint8 ActivePlayers;

+};
\ No newline at end of file
diff --git a/src/control/Garages.cpp b/src/control/Garages.cpp
index b573921f..5f55ae66 100644
--- a/src/control/Garages.cpp
+++ b/src/control/Garages.cpp
@@ -71,6 +71,7 @@ bool CGarages::HasCarBeenCrushed(int32 handle)
 WRAPPER void CGarages::TriggerMessage(const char *text, int16, uint16 time, int16) { EAXJMP(0x426B20); }
 WRAPPER bool CGarages::IsPointWithinHideOutGarage(CVector&) { EAXJMP(0x428260); }
 WRAPPER bool CGarages::IsPointWithinAnyGarage(CVector&) { EAXJMP(0x428320); }
+WRAPPER void CGarages::PlayerArrestedOrDied() { EAXJMP(0x427F60); }
 
 #if 0
 WRAPPER void CGarages::PrintMessages(void) { EAXJMP(0x426310); }
diff --git a/src/control/Garages.h b/src/control/Garages.h
index cc5fb62b..bcd966e8 100644
--- a/src/control/Garages.h
+++ b/src/control/Garages.h
@@ -27,4 +27,5 @@ public:
 	static bool HasCarBeenCrushed(int32);
 	static bool IsPointWithinHideOutGarage(CVector&);
 	static bool IsPointWithinAnyGarage(CVector&);
+	static void PlayerArrestedOrDied();
 };
diff --git a/src/control/Restart.cpp b/src/control/Restart.cpp
index c5c46b51..40b95c23 100644
--- a/src/control/Restart.cpp
+++ b/src/control/Restart.cpp
@@ -2,6 +2,13 @@
 #include "patcher.h"
 #include "Restart.h"
 
+bool &CRestart::OverrideHospitalLevel = *(bool*)0x95CD4C;
+bool &CRestart::OverridePoliceStationLevel = *(bool*)0x95CD50;
+bool &CRestart::bFadeInAfterNextArrest = *(bool*)0x95CD69;
+bool &CRestart::bFadeInAfterNextDeath = *(bool*)0x95CD9D;
+
 WRAPPER void CRestart::AddHospitalRestartPoint(const CVector&, float) { EAXJMP(0x436100); }
 WRAPPER void CRestart::AddPoliceRestartPoint(const CVector&, float) { EAXJMP(0x436150); }
-WRAPPER void CRestart::OverrideNextRestart(const CVector&, float) { EAXJMP(0x4366C0); }
\ No newline at end of file
+WRAPPER void CRestart::OverrideNextRestart(const CVector&, float) { EAXJMP(0x4366C0); }
+WRAPPER void CRestart::FindClosestHospitalRestartPoint(const CVector &, CVector *, float *) { EAXJMP(0x4361A0); }
+WRAPPER void CRestart::FindClosestPoliceRestartPoint(const CVector &, CVector *, float *) { EAXJMP(0x436450); }
\ No newline at end of file
diff --git a/src/control/Restart.h b/src/control/Restart.h
index f49ed79c..9a4d2155 100644
--- a/src/control/Restart.h
+++ b/src/control/Restart.h
@@ -6,4 +6,12 @@ public:
 	static void AddPoliceRestartPoint(const CVector&, float);
 	static void AddHospitalRestartPoint(const CVector&, float);
 	static void OverrideNextRestart(const CVector&, float);
+
+	static void FindClosestHospitalRestartPoint(const CVector &, CVector *, float *);
+	static void FindClosestPoliceRestartPoint(const CVector &, CVector *, float *);
+
+	static bool &OverrideHospitalLevel;
+	static bool &OverridePoliceStationLevel;
+	static bool &bFadeInAfterNextArrest;
+	static bool &bFadeInAfterNextDeath;
 };