From 9df7c9feb8d99f8cb35505ef0988aa85c4ed72a4 Mon Sep 17 00:00:00 2001
From: yihong0618 <zouzou0208@gmail.com>
Date: Mon, 11 Dec 2023 22:08:06 +0800
Subject: [PATCH] fix: it

---
 hy_tg.py | 42 +++++++++++++++++++++++++++++++++++++++---
 1 file changed, 39 insertions(+), 3 deletions(-)

diff --git a/hy_tg.py b/hy_tg.py
index ef82335..4683ba7 100644
--- a/hy_tg.py
+++ b/hy_tg.py
@@ -4,14 +4,17 @@ import os
 import gc
 import subprocess
 from telebot import TeleBot  # type: ignore
-from telebot.types import BotCommand, Message  # type: ignore
+from telebot.types import Message  # type: ignore
 
 from prettymapp.geo import get_aoi
 from prettymapp.osm import get_osm_geometries
 from prettymapp.plotting import Plot
 from prettymapp.settings import STYLES
 
+from matplotlib import figure
+
 from PIL import Image
+import numpy as np
 import PIL
 import io
 
@@ -20,6 +23,38 @@ file_in = "map.jpg"
 file_out = "map_out.jpg"
 
 
+# monkey patch for Plot thanks @higuoxing https://github.com/higuoxing
+def __post_init__(self):
+    (
+        self.xmin,
+        self.ymin,
+        self.xmax,
+        self.ymax,
+    ) = self.aoi_bounds
+    # take from aoi geometry bounds, otherwise probelematic if unequal geometry distribution over plot.
+    self.xmid = (self.xmin + self.xmax) / 2
+    self.ymid = (self.ymin + self.ymax) / 2
+    self.xdif = self.xmax - self.xmin
+    self.ydif = self.ymax - self.ymin
+
+    self.bg_buffer_x = (self.bg_buffer / 100) * self.xdif
+    self.bg_buffer_y = (self.bg_buffer / 100) * self.ydif
+
+    # self.fig, self.ax = subplots(
+    #     1, 1, figsize=(12, 12), constrained_layout=True, dpi=1200
+    # )
+    self.fig = figure.Figure(figsize=(12, 12), constrained_layout=True, dpi=1200)
+    self.ax = self.fig.subplots(1, 1)
+    self.ax.set_aspect(1 / np.cos(self.ymid * np.pi / 180))
+
+    self.ax.axis("off")
+    self.ax.set_xlim(self.xmin - self.bg_buffer_x, self.xmax + self.bg_buffer_x)
+    self.ax.set_ylim(self.ymin - self.bg_buffer_y, self.ymax + self.bg_buffer_y)
+
+
+Plot.__post_init__ = __post_init__
+
+
 def sizeof_image(image):
     with io.BytesIO() as buff:
         image.save(buff, format="JPEG", quality=95)
@@ -34,7 +69,7 @@ def compress_image(input_path, output_path, target_size):
 
         while sizeof_image(img) > target_bytes:
             factor -= 0.05
-            (width, height) = img.size
+            width, height = img.size
             img = img.resize(
                 (int(width * factor), int(height * factor)),
                 PIL.Image.Resampling.LANCZOS,
@@ -49,7 +84,7 @@ def draw_pretty_map(location, file_name, style):
     df = get_osm_geometries(aoi=aoi)
     fig = Plot(df=df, aoi_bounds=aoi.bounds, draw_settings=STYLES[style]).plot_all()
     fig.savefig(file_name)
-    compress_image(file_in, file_out, 9)  # 10MB
+    compress_image(file_in, file_out, 9)  # telegram tog need png less than 10MB
 
 
 def main():
@@ -114,6 +149,7 @@ def main():
             bot.reply_to(message, "Something wrong please check")
             print(str(e))
         bot.delete_message(reply_message.chat.id, reply_message.message_id)
+        # we need this, fuck it
         gc.collect()
 
     # Start bot