update bus poster in order to fit in 2x1 configuration

scripts/Poster/BusPoster.py

@@ -57,7 +57,7 @@ class BusPoster(MyDraw):
         text_bbox = self.font.getbbox(str(bus_letter))
         font_width, font_height = text_bbox[2] - text_bbox[0], text_bbox[3] - text_bbox[1]
         offset_width = np.round((proportion*self.width-width_border)) + 0.75*np.round(font_width/2)
-        text_position = (offset_width,0)
+        text_position = (1.1*offset_width,0)
         self.draw.text(
             text_position,
             bus_letter,

scripts/Poster/TimeAnnouncement.py

@@ -56,13 +56,14 @@ class TimeAnnouncement(MyDraw):
         else:
             text = f'{min_time} a {max_time} min'
             
-        self.load_barlow(font_size=18)
+        self.load_barlow(font_size=12)
         text_bbox = self.font.getbbox(text)
         font_width, font_height = text_bbox[2] - text_bbox[0], text_bbox[3] - text_bbox[1]
         # print(font_width, font_height)
         offset_width = (np.round((self.width-self.border)) - np.round(font_width))/2
         offset_height = (np.round((self.height-self.border)) - np.round(base_font_height))/2
-        text_position = (offset_width,5+offset_height)
+        # text_position = (offset_width,5+offset_height)
+        text_position = (offset_width,offset_height)
         # text_position = (0, 0)
         self.draw.text(
             text_position,

scripts/Poster/main_1_2.py

@@ -0,0 +1,195 @@
+from MyDraw import MyDraw
+from BusPoster import BusPoster
+from TimeAnnouncement import TimeAnnouncement
+from DistanceAnnouncement import DistanceAnnouncement
+from BusPlate import BusPlate
+from BusImage import BusImage
+
+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 distancia aproximada en km
+    distance = approx_km(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()
+    # full_panel.preview()
+
+    # Con el dato de la patente se agrega al dibujo
+    bp = BusPlate()
+    plate = data["licensePlate"]
+    bp.request_bus_plate(bus_plate=plate)
+    bp.generate_image()
+    bp.resize_image(target_height=aprox((3/10)*height))
+
+    # Agrega la distancia al paradero
+    dist_anmc = DistanceAnnouncement(aprox((2/5)*height), aprox((1/3)*width))
+    dist_anmc.set_theme(theme)
+    dist_anmc.start_draw()
+    # dist_anmc.set_background()
+    # dist_anmc.set_base_text()
+    dist_anmc.set_distance_text(distance=distance)
+
+    # 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()
+
+    poster_params = {
+        'proportion': 0.6,
+        'width_border': 2,
+        'font_size': 11,
+        'number_background_color': 'yellow',
+        'letter_background_color': 'green',
+    }
+
+    # Se setean los parametros
+    poster1.set_params(poster_params)
+    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()
+
+    poster_params = {
+        'proportion': 0.6,
+        'width_border': 2,
+        'font_size': 11,
+        'number_background_color': 'yellow',
+        'letter_background_color': 'blue',
+    }
+
+    # Se setean los parametros
+    poster2.set_params(poster_params)
+    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 agrega la imagen del bus
+    bm = BusImage()
+    bm.set_theme(theme)
+    bm.load_image_from_url()
+    bm.crop_image(top_cut=165, bottom_cut=165)
+    bm.resize_image(target_width=aprox((1/3)*width))
+
+    # Se agregan todas las imagenes al canvas
+    # full_panel.add_image(bp, (aprox(0*width), aprox((0)*height)))
+    # full_panel.add_image(dist_anmc, (aprox((3/8)*width), aprox(0.1*height)))
+    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()