Query ioctl first for terminal dimensions in pixels, fallback to ANSI escape code

Author: ayorgo

matplotlib-backend-kitty/__init__.py

@@ -1,5 +1,7 @@
 # SPDX-License-Identifier: CC0-1.0
 
+import array
+import fcntl
 import os
 import re
 import sys
@@ -21,6 +23,14 @@
     interactive(True)
 
 def term_size_px():
+    # try to get terminal size from ioctl
+    buf = array.array("H", [0, 0, 0, 0])
+    fcntl.ioctl(sys.stdout, termios.TIOCGWINSZ, buf)
+    _, _, width_px, height_px = buf
+    if width_px != 0 and height_px != 0:
+        return height_px, width_px
+
+    # fallback to ANSI escape code if ioctl fails
     buf = ''
     stdin = sys.stdin.fileno()
     tattr = termios.tcgetattr(stdin)