from scripts.Poster.MyDraw import MyDraw
from scripts.Poster.BusPoster import BusPoster
from scripts.Poster.TimeAnnouncement import TimeAnnouncement
import numpy as np
from datetime import datetime, timedelta

def aprox(n):
    return int(np.round(n))

def load_data():
    data = {
        "direction": "R",
        "distance": 1948.575483806973,
        "epochTime": 1674650956,
        "latitude": -33.43729782104492,
        "licensePlate": "LJHA57",
        "longitude": -70.52730560302734,
        "realtime": True,
        "route": "401",
        "routeId": "401",
        "timeLabel": "09:49",
        "tripId": "401-I-L-005"
    }
    return data

def approx_km(data):
    distance_meters = data["distance"]
    distance_km = distance_meters / 100  # Convert meters to kilometers
    approx_km = int(np.round(distance_km))  # Take only the integer part of the distance in kilometers
    approx_km = approx_km/10.0
    return approx_km

def calc_remaining_time(data):
    arrival_time = data["timeLabel"]
    target_time = datetime.strptime(arrival_time, "%H:%M").time()
    current_time = datetime.now().time()

    if current_time < target_time:
        remaining_time = datetime.combine(datetime.today(), target_time) - datetime.combine(datetime.today(), current_time)
    else:
        remaining_time = datetime.combine(datetime.today() + timedelta(days=1), target_time) - datetime.combine(datetime.today(), current_time)

    remaining_minutes = int(remaining_time.total_seconds() // 60)
    return remaining_minutes

def obtain_min_max_time(remaining_time):
    if remaining_time == 1:
        return 0, 1
    elif remaining_time == 2:
        return 1, 2
    elif 2 <= remaining_time <= 5:
        return 2, 5
    elif remaining_time > 5 and remaining_time <= 7:
        return 5, 7
    elif remaining_time > 7 and remaining_time <= 10:
        return 7, 10
    else:
        return 10, remaining_time
    
###################################################################
# Parametros para generar la imagen
# "direction": "R", Indicador de la dirección en la que va el bus
# "distance": 1948.575483806973. Distancia en m
# "licensePlate": "LJHA57", Patente del bus
# "route": "401", Linea de bus
# "timeLabel": "09:49", Hora de llegada al paradero

# theme: Tema de la pantalla "day/night"
# 'number_background_color': 'yellow', Color del fondo para el numero
# 'letter_background_color': 'green', Color del fondo para la letra

def main():
    # Carga los datos
    data = load_data()

    # Calcula el tiempo restante a la llegada
    remaining_time = calc_remaining_time(data)
    # Obtiene valores máximos y mínimo de rangos para desplegar en pantalla
    min_time, max_time = obtain_min_max_time(remaining_time)

    # Selecciona el tema
    theme = 'day'

    # Alto y ancho de la imagen en pixeles
    height, width = 40, 160

    # Inicia el dibujo y setea el tema
    full_panel = MyDraw(height=height, width=width)
    full_panel.set_theme(theme)
    full_panel.start_draw()

    # Agrega el anuncio de los minutos restante al arribo
    time_anmc1 = TimeAnnouncement(aprox((2/5)*height), aprox((1/3)*width))
    time_anmc1.set_theme(theme)
    time_anmc1.start_draw()
    # time_anmc1.set_background()
    # time_anmc1.set_base_text()
    time_anmc1.set_min_max_text(min_time=min_time, max_time=max_time)

    # Agrega el anuncio de los minutos restante al arribo
    time_anmc2 = TimeAnnouncement(aprox((2/5)*height), aprox((1/3)*width))
    time_anmc2.set_theme(theme)
    time_anmc2.start_draw()
    # time_anmc2.set_background()
    # time_anmc2.set_base_text()
    time_anmc2.set_min_max_text(min_time=3, max_time=5)
    
    # Genera la imagen de la linea del bus
    poster1 = BusPoster(aprox(1.1*(1/3)*height), aprox(1.1*(1/3)*width))
    poster1.set_theme(theme)
    poster1.start_draw()

    bus_announcement_1 = {
        'proportion': 0.6,
        'width_border': 2,
        'font_size': 11,
        'number_background_color': 'yellow',
        'letter_background_color': 'green',
    }

    # Se setean los parametros
    poster1.set_params(bus_announcement_1)
    poster1.load_barlow()
    poster1.set_colors()
    # Se setea la ruta y la direccion en la que va
    poster1.set_bus_number(bus_number=data["route"])
    poster1.set_bus_letter(bus_letter=data["direction"])

    # Genera la imagen de la linea del bus
    poster2 = BusPoster(aprox(1.1*(1/3)*height), aprox(1.1*(1/3)*width))
    poster2.set_theme(theme)
    poster2.start_draw()

    bus_announcement_2 = {
        'proportion': 0.6,
        'width_border': 2,
        'font_size': 11,
        'number_background_color': 'yellow',
        'letter_background_color': 'blue',
    }

    # Se setean los parametros
    poster2.set_params(bus_announcement_2)
    poster2.load_barlow()
    poster2.set_colors()
    # Se setea la ruta y la direccion en la que va
    poster2.set_bus_number(bus_number="16")
    poster2.set_bus_letter(bus_letter="I")

    # Se agregan todas las imagenes al canvas
    full_panel.add_image(time_anmc1, (aprox((0.6)*width), aprox(0.05*height)))
    full_panel.add_image(time_anmc2, (aprox((0.6)*width), aprox(0.45*height)))
    full_panel.add_image(poster1, (aprox((0.05)*width), aprox((0.1)*height)))
    full_panel.add_image(poster2, (aprox((0.05)*width), aprox((0.5)*height)))
    # full_panel.add_image(bm, (aprox(0.02*width),aprox((1/6)*height)))
    full_panel.get_image()
    full_panel.save_image('/app/data/output.png')

if __name__ == '__main__':
    main()