# The Sims 4
# Copyright 2014 Electronic Arts Inc. All rights reserved.
"""
	Super Speed Mod - Created by artum
	Version: 0.02
	
	This enables SuperSpeed-like game_speed by pressing "3" twice
	
	It'll conflict with mods that:
		-	Modify default "clock.set_speed" console command
		-	Modify GameClock._clock_speed_to_scale or GameClock.set_clock_speed method in clock.py
		-	Adds custom variables/methods with the same names as the one used in this script (Who would do that?)
"""
import sims4.commands
import services
import clock
from clock import GameClock, ClockSpeedMode, ClockSpeedMultipliers, GameSpeedChangeSource, ClockSpeedMultiplierType
from server_commands.clock_commands import send_clock_telemetry_data

GameClock.OLD_clock_speed_to_scale = GameClock._clock_speed_to_scale
GameClock.OLD_set_clock_speed = GameClock.set_clock_speed
GameClock.superspeed = False
GameClock.last_user_speed = ClockSpeedMode.PAUSED

#Modified Input Class (Original at simulation/server_commands/clock_commands.pyo)
@sims4.commands.Command('clock.setspeed', command_type=sims4.commands.CommandType.Live)
def set_speed(speed='one', _connection=None):
	output = sims4.commands.Output(_connection)
	speed = speed.lower()
	clock_service = services.game_clock_service()
	if speed == 'one':
		speed = ClockSpeedMode.NORMAL
	elif speed == 'two':
		speed = ClockSpeedMode.SPEED2
	elif speed == 'three':
		speed = ClockSpeedMode.SPEED3
	
	
	if speed == ClockSpeedMode.SPEED3 and clock_service.last_user_speed == ClockSpeedMode.SPEED3:
		clock_service.superspeed = not clock_service.superspeed #Toggle
	else:
		clock_service.superspeed = False
	
	send_clock_telemetry_data(_connection, speed)
	clock_service.set_clock_speed(speed, change_source=GameSpeedChangeSource.USER)
	clock_service.last_user_speed = speed

#Overrides _clock_speed_to_scale method ( surface/clock.py)
def modded_clock_speed_to_scale(self, clock_speed):
	if clock_speed == ClockSpeedMode.SPEED3 and self.superspeed and not services.get_super_speed_three_service().in_or_has_requested_super_speed_three():
		return ClockSpeedMultipliers.super_speed_three_multiplier(ClockSpeedMultiplierType.DEFAULT) #without ClockSpeedMultiplierType.DEFAULT it lags :S
	else:
		self.superspeed = False
		return self.OLD_clock_speed_to_scale(clock_speed)

def modded_set_clock_speed(self, speed, change_source=GameSpeedChangeSource.GAMEPLAY) -> bool:
	if (speed != ClockSpeedMode.SPEED3 or services.get_super_speed_three_service().in_or_has_requested_super_speed_three()):
		self.superspeed = False
	return self.OLD_set_clock_speed(speed, change_source)

GameClock.ignore_game_speed_requests = True #SuperSpeedThree disabled while trying to fix it
GameClock._clock_speed_to_scale = modded_clock_speed_to_scale
GameClock.set_clock_speed = modded_set_clock_speed