def main():
""" docstring """
headlightFlash_o = HeadlightControl()
config_o = Config()
def pit_limiter():
return config_o.get('miscellaneous', 'pit_limiter') == '1'
def pit_lane():
return config_o.get('miscellaneous', 'pit_lane') == '1'
def flash_duration():
return int(config_o.get('miscellaneous', 'flash_duration'))
def pit_flash_duration():
return int(config_o.get('miscellaneous', 'pit_flash_duration'))
def default_to_on():
return config_o.get('miscellaneous', 'default_to_on') == '1'
def on_automatically():
return int(config_o.get('miscellaneous', 'on_automatically'))
_root, tabConfigureFlash = gui_main()
_player_is_driving = False
_o_run = run(_root, tabConfigureFlash)
_o_run.controller_o.start_pit_check_timer() # Start the 1 second timer
while True:
_cmd = _o_run.running()
if headlightFlash_o.player_is_driving():
if not _player_is_driving:
# First time player takes control
_player_is_driving = True
if default_to_on():
status_poker_fn('default_to_on')
headlightFlash_o.on()
if _cmd == 'Headlights off':
headlightFlash_o.on()
if _cmd == 'Headlights on':
headlightFlash_o.off()
if _cmd == 'Flash headlights':
headlightFlash_o.four_flashes(flash_duration())
if _cmd == 'Toggle headlights':
headlightFlash_o.toggle()
if _cmd == TIMER_EVENT:
if pit_limiter():
headlightFlash_o.check_pit_limiter(pit_flash_duration())
if pit_lane():
headlightFlash_o.check_pit_lane(pit_flash_duration())
headlightFlash_o.automatic_headlights(on_automatically())
else:
_player_is_driving = False
if _cmd == 'QUIT':
break
def __init__(self) -> None:
""" docstring """
config_o = Config()
if config_o.get('rFactor Toggle', 'controller') == KEYBOARD:
self.headlightToggleDIK = config_o.get('rFactor Toggle', 'control')
# (it must be)
if self.headlightToggleDIK not in DirectInputKeyCodeTable:
print('\nheadlight toggle button "%s" not recognised.\n'
'It must be one of:' % self.headlightToggleDIK)
for _keyCode in DirectInputKeyCodeTable:
print(_keyCode, end=', ')
quit_program(99)
else:
status_poker_fn('\nHeadlight toggle control must be a key.\n')
quit_program(99)
def automatic_headlights(self, on_automatically) -> None:
"""
# Headlights on when:
# 0 Driver turns them on
# 1 At least one other driver has them on
# 2 More than one other driver has them on
# 3 At least half of the other drivers have them on
# 4 All the other drivers have them on
"""
_on = False
if self._flashing:
return # Don't turn headlights on when flashing
if on_automatically and not self.are_headlights_on():
_num_drivers = self._info.Rf2Scor.mScoringInfo.mNumVehicles
_num_drivers_with_lights = 0
for _driver in range(_num_drivers):
if self._info.Rf2Tele.mVehicles[_driver].mHeadlights:
_num_drivers_with_lights += 1
# Total includes the player so
_num_drivers -= 1
if on_automatically == 1 and _num_drivers_with_lights:
_on = True
status_poker_fn('At least one other driver has headlights on')
if on_automatically == 2 and _num_drivers_with_lights > 1:
_on = True
status_poker_fn('More than one other driver has headlights on')
if on_automatically == 3 and \
_num_drivers_with_lights >= (_num_drivers/2):
_on = True
status_poker_fn(
'At least half of the other drivers have headlights on')
if on_automatically == 4 and \
_num_drivers_with_lights >= _num_drivers:
_on = True
status_poker_fn('All the other drivers have headlights on')
if _on:
self.on()
def __toggle(self, stopping_callback) -> None:
""" Toggle the headlights unless it's time to stop """
if self._info.isSharedMemoryAvailable():
if self._info.isTrackLoaded():
if self._info.isOnTrack():
if self.__ignition_is_on():
if not stopping_callback():
self._flashing = True
self.toggle()
__flashTimer = SetTimer(self.timer,
self.__toggle,
_args=[stopping_callback])
# type: ignore
return
else:
status_poker_fn('Not on track')
else:
status_poker_fn('Track not loaded')
else:
status_poker_fn('rFactor 2 not running')
self.stop_flashing()
def off(self) -> None:
""" Turn them off regardless """
status_poker_fn('Headlights off')
if self.are_headlights_on():
self.toggle()
def on(self) -> None:
""" Turn them on regardless """
status_poker_fn('Headlights on')
if not self.are_headlights_on():
self.toggle()
def check_pit_lane(self, pit_flash_duration) -> None:
""" Has the car entered the pit lane? """
if self._info.isOnTrack():
if not self.__not_in_pit_lane():
status_poker_fn('Entered pit lane')
self.pit_lane_flashes(pit_flash_duration)
def four_flashes(self, flash_duration) -> None:
""" Flash four times (e.g. for overtaking) """
status_poker_fn('Overtaking flash')
self._count = 8 # 4 flashes
self.timer = flash_duration
self.start_flashing(self.count_down)