from time import time
import adafruit_ble
import board
import pyloton

# PyLint can't find BLERadio for some reason so special case it here.
ble = adafruit_ble.BLERadio()    # pylint: disable=no-member

display = board.DISPLAY

CONNECTION_TIMEOUT = 45

HEART = True
SPEED = True
CAD = True
AMS = True
DEBUG = False

# 84.229 is wheel circumference (700x23 in my case)
pyloton = pyloton.Pyloton(ble, display, 84.229, HEART, SPEED, CAD, AMS, DEBUG)

pyloton.show_splash()

ams = pyloton.ams_connect()


start = time()
hr_connection = None
speed_cad_connections = []
radio = None
while True:
    if HEART:
        if not hr_connection:
            print("Attempting to connect to a heart rate monitor")
            hr_connection = pyloton.heart_connect()
            ble.stop_scan()
    if SPEED or CAD:
        if not speed_cad_connections:
            print("Attempting to connect to speed and cadence monitors")
            speed_cad_connections = pyloton.speed_cad_connect()

    if time()-start >= CONNECTION_TIMEOUT:
        pyloton.timeout()
        break
    # Stop scanning whether or not we are connected.
    ble.stop_scan()

    # You may need to remove some parts of the following 2 lines depending on what
    # devices you are using.
    # For example, remove hr_connection and hr_connection.connected if you aren't using a heart
    # rate monitor. Do the same for other sensors you aren't using.
    if hr_connection and speed_cad_connections and ams:
        if hr_connection.connected and speed_cad_connections[0].connected and ams.connected:
            pyloton.setup_display()
            break

# You may need to remove some parts of the following line depending on what devices you are using.
# For example, remove hr_connection and hr_connection.connected if you aren't using a heart rate
# monitor. Do the same for other sensors you aren't using.
while hr_connection.connected and speed_cad_connections[0].connected and ams.connected:
    pyloton.update_display()
    pyloton.ams_remote()

print("\n\nNot all sensors are connected. Please reset to try again\n\n")