require(GlobalDependencys:GetDependency("StandardBase"):GetPackageName())

--GAME VARS
fDefaultFOV = 65
fAdditionalFOV = 0
fAspectRatio = DisplayInfo:GetWidth() / DisplayInfo:GetHeight()
fAspectRatio169 = 1.777778

--ControlVars
bFixEnabled = true
bFOV = true
bAspect = true

--PROCESS VARS
Process_FriendlyName = Module:GetFriendlyName()
Process_WindowName = "*"
Process_ClassName = "*"
Process_EXEName = "SouthOfMidnight.exe"

--INJECTION BEHAVIOUR
InjectDelay = 500
WriteInterval = 100
SearchInterval = 100
SuspendThread = true

--Name                         Manual/Auto/Hybrid  		Steam/Origin/Any                IncludeFile:Configure;Enable;Periodic;Disable;
SupportedVersions = { 		
{"Automatically Detect",       "Hybrid",  			"Any",	                         "Configure_SignatureScan;Enable_Inject;Periodic;Disable_Inject;"},
}

function Init_Controls()

	DefaultControls.AddHeader("Header_FixesEnableDisable","Individual Fixes",245,70,210,17)
	
	DefaultControls.AddHeader("Header_FOV","In game FOV fine adjustment",15,70,210,17)	
	DefaultControls.AddFixedFOVSlider("FOVSlider","FOVSlider_Changed",55,100,125,35,0,70,0,1)
	FOVSlider:SetTickFrequency(10)

	DefaultControls.AddFixToggle("CKFOVFix_Enable","FOV fix","CKFOVFix_Changed",255,101,180,14)
	DefaultControls.AddFixToggle("CKASPECTFix_Enable","Aspect ratio fix","CKASPECTFix_Changed",255,121,180,14)

end

function Configure_SignatureScan() 
	
	local tAddress = HackTool:AddAddress("FOV")
	
	if HackTool:SignatureScan("90 0F ?? ?? 48 8B 8C ?? ?? ?? ?? ?? 48 ?? ?? E8 ?? ?? ?? ?? 0F 28",tAddress,PAGE_EXECUTE_READ,0x01,Process_EXEName) == 0 then
		return ErrorOccurred(string.format(SigScanError,tAddress:GetName()))
	else	
		print( tAddress:GetInfo(TYPE_ADDRESS) )
		--SouthOfMidnight.exe+357E051 - E8 5A 4A ED FD          - call SouthOfMidnight.exe+1452AB0
		--SouthOfMidnight.exe+357E056 - 90                    	- nop 
		--SouthOfMidnight.exe+357E057 - 0F 28 C6               	- movaps xmm0,xmm6
		--SouthOfMidnight.exe+357E05A - 48 8B 8C 24 50 09 00 00 - mov rcx,[rsp+00000950]
		--SouthOfMidnight.exe+357E062 - 48 33 CC              	- xor rcx,rsp
	end
	
	local tAddress = HackTool:AddAddress("ASPECT")
	
	if HackTool:SignatureScan("EB ?? F3 0F ?? ?? ?? ?? ?? ?? F3 0F ?? ?? ?? 8B 87 ?? ?? ?? ?? 89 ?? ?? 0F B6",tAddress,PAGE_EXECUTE_READ,0x0f,Process_EXEName) == 0 then
		return ErrorOccurred(string.format(SigScanError,tAddress:GetName()))
	else	
		print( tAddress:GetInfo(TYPE_ADDRESS) )
		--SouthOfMidnight.exe+3161AB4 - F3 0F 10 87 F8 01 00 00 - movss xmm0,[rdi+000001F8]
		--SouthOfMidnight.exe+3161ABC - F3 0F 11 46 18         	- movss [rsi+18],xmm0
		--SouthOfMidnight.exe+3161AC1 - 8B 87 08 02 00 00       - mov eax,[rdi+00000208]
		--SouthOfMidnight.exe+3161AC7 - 89 46 2C              	- mov [rsi+2C],eax
		--SouthOfMidnight.exe+3161ACA - 0F B6 8F 0C 02 00 00    - movzx ecx,byte ptr [rdi+0000020C]
	end
	
	return true

end

function Enable_Inject()
	local Variables = HackTool:AllocateMemory("Variables",0)
	Variables:PushFloat("FOVIn")
	Variables:PushFloat("FOVOut")
	Variables:PushFloat("CompensatedFOV")
	Variables:PushFloat("AdditionalFOV")
	Variables:PushFloat("ScreenRatio")
	Variables:PushByte("ASPECTCHECKED")
	
	Variables:Allocate()
	Variables["FOVIn"]:WriteFloat(fDefaultFOV)
	Variables["FOVOut"]:WriteFloat(fDefaultFOV)
	Variables["AdditionalFOV"]:WriteFloat(fAdditionalFOV)
	Variables["ScreenRatio"]:WriteFloat(fAspectRatio)
	
	FOVCalculator = HackTool:InjectFOVCalculator("FOVCalculator")	
	
	ResolutionChanged()	

	local asm = [[		

		(codecave:jmp)FOV,FOV_cc:
			%originalcode%
			movss [(allocation)Variables->FOVIn],$$1							$ctx=1
			cmp byte ptr [(allocation)Variables->ASPECTCHECKED],1
			jne normalFOV
			fld dword [(allocation)Variables->FOVIn]
			call (allocation)FOVCalculator
			fstp dword ptr [(allocation)Variables->CompensatedFOV]
			movss $$1,[(allocation)Variables->CompensatedFOV]					$ctx=1
		normalFOV:
			addss $$1,[(allocation)Variables->AdditionalFOV]					$ctx=1
			movss [(allocation)Variables->FOVOut],$$1							$ctx=1
			jmp %returnaddress%
			%end%
											
		(codecave:jmp)ASPECT,ASPECT_cc:
			$$0 $$1,[(allocation)Variables->ScreenRatio]					$ctx=1
			jmp %returnaddress%
			%end%
							
	]]
	
	if HackTool:CompileAssembly(asm,"Fixes") == nil then
		return ErrorOccurred("Assembly compilation failed...")
	else
		if Variables and Variables["ASPECTCHECKED"] and bAspect then
			Variables["ASPECTCHECKED"]:WriteByte(1)
		end
		Toggle_CodeCave("FOV_cc",(bFOV or bAspect))
		Toggle_CodeCave("ASPECT_cc",bAspect)
	end	
			
	Write_FOV()
	
end

function Periodic()

	local Variables = HackTool:GetAllocation("Variables")
	
	if Variables and Variables["FOVIn"] and Variables["AdditionalFOV"] then
		local fFOVIn = Variables["FOVIn"]:ReadFloat()
		local fFOVOut = Variables["FOVOut"]:ReadFloat()
		local fFOVCompensated = Variables["CompensatedFOV"]:ReadFloat()
		
		PluginViewport:AppendStatusMessage( string.format("\r\n===== Fix informations =====\r\nFOV In: %.2f, Compensated : %.2f, Out : %.2f", fFOVIn, fFOVCompensated, fFOVOut))
	end
	
end

function FOVSlider_Changed(Sender)

	fAdditionalFOV = Sender:GetPosition() - 20
	lblFOVSlider.Caption:SetCaption( string.format("World FOV: %.0f",fAdditionalFOV) )
		
	Write_FOV()	
	ForceUpdate()

end

function Write_FOV()

	local Variables = HackTool:GetAllocation("Variables")
	if Variables and Variables["AdditionalFOV"] then
		if bFOV == true then
			Variables["AdditionalFOV"]:WriteFloat(fAdditionalFOV)
		else
			Variables["AdditionalFOV"]:WriteFloat(0)
		end
	end

end

function Disable_Inject()
	
	CleanUp()
	
end

function CKFOVFix_Changed(Sender)

	bFOV = Toggle_CheckFix(Sender)
	local fFOV = (bFOV or bAspect)
	--local fFOV = bFOV
	Write_FOV()
	Toggle_CodeCave("FOV_cc",fFOV)
	ForceUpdate()
	
end

function CKASPECTFix_Changed(Sender)

	bAspect = Toggle_CheckFix(Sender)
	Toggle_CodeCave("ASPECT_cc",bAspect)
	if bAspect then
		Toggle_CodeCave("FOV_cc",bAspect)
	end
	local Variables = HackTool:GetAllocation("Variables")
	if Variables and Variables["ASPECTCHECKED"] then
		if bAspect then
			Variables["ASPECTCHECKED"]:WriteByte(1)
		else
			Variables["CompensatedFOV"]:WriteFloat(0)
			Variables["ASPECTCHECKED"]:WriteByte(0)
		end
	end
	ForceUpdate()
	
end

function ResolutionChanged() 

	SyncDisplayDetection()
	
	local CurrentAspectRatio = DisplayInfo:GetAspectRatio()
	local AspectDevisional = CurrentAspectRatio / fAspectRatio169

	if CurrentAspectRatio < fAspectRatio169 then
		AspectDevisional = 1.0
	end

	UpdateFOVCalculator("FOVCalculator",AspectDevisional,0.0)
	
end

function Init()

	Init_BaseControls()
	Init_Controls()

end

function DeInit()

	DisableFix()

end