kaxe 1.8.1.dev0__tar.gz → 1.8.1.dev2__tar.gz

{kaxe-1.8.1.dev0/src/kaxe.egg-info → kaxe-1.8.1.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kaxe
-Version: 1.8.1.dev0
+Version: 1.8.1.dev2
 Summary: A small graphing tool for functions, points, equations and more
 Author-email: Valter Yde Daugberg <valteryde@hotmail.com>
 Project-URL: Homepage, https://github.com/valteryde/kaxe
@@ -27,6 +27,8 @@ Requires-Dist: psutil; extra == "3d"
 Provides-Extra: pdf
 Requires-Dist: reportlab>=4.0; extra == "pdf"
 Requires-Dist: fonttools>=4.0; extra == "pdf"
+Provides-Extra: viewer
+Requires-Dist: pyglet>=2.0; extra == "viewer"
 Dynamic: license-file
 
 

{kaxe-1.8.1.dev0 → kaxe-1.8.1.dev2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "kaxe"
-version = "1.8.1.dev0"
+version = "1.8.1.dev2"
 authors = [
   { name="Valter Yde Daugberg", email="valteryde@hotmail.com" },
 ]
@@ -31,6 +31,10 @@ core = []
   "psutil",
 ]
 pdf = ["reportlab>=4.0", "fonttools>=4.0"]
+viewer = ["pyglet>=2.0"]
+
+[project.scripts]
+kaxe = "kaxe.cli:main"
 
 [project.urls]
 Homepage = "https://github.com/valteryde/kaxe"

kaxe-1.8.1.dev2/src/kaxe/__main__.py

@@ -0,0 +1,6 @@
+"""Support running `python -m kaxe viewer`."""
+
+from .cli import main
+
+if __name__ == "__main__":
+    main()

kaxe-1.8.1.dev2/src/kaxe/_require_viewer.py

@@ -0,0 +1,8 @@
+_MSG = "Viewer features require optional dependencies. Install with: pip install kaxe[viewer]"
+
+
+def require_viewer():
+    try:
+        import pyglet  # noqa: F401
+    except ImportError as exc:
+        raise ImportError(_MSG) from exc

kaxe-1.8.1.dev2/src/kaxe/cli.py

@@ -0,0 +1,8 @@
+"""Top-level CLI module for Kaxe."""
+
+from .viewer.cli import main
+
+__all__ = ["main"]
+
+if __name__ == "__main__":
+    main()

{kaxe-1.8.1.dev0 → kaxe-1.8.1.dev2}/src/kaxe/core/window.py

@@ -11,6 +11,7 @@ from .svg import SvgDocument, infer_format, is_file_path
 from .progress import make_progress_bar
 from .ipython_display import display_png_bytes, is_notebook, to_png_bytes
 from .styles import assign_series_color, _palette_color_at
+from ..viewer.client import try_show_in_viewer
 from PIL import Image
 
 
@@ -660,6 +661,8 @@ class Window(AttrObject):
         data = to_png_bytes(self)
         if is_notebook():
             display_png_bytes(data)
+        elif try_show_in_viewer(data, title=getattr(self, "titleText", "")):
+            return
         else:
             Image.open(BytesIO(data)).show()
 

{kaxe-1.8.1.dev0 → kaxe-1.8.1.dev2}/src/kaxe/plot/grid.py

@@ -4,6 +4,7 @@ from typing import Optional, Union
 from ..core.styles import AttrObject, AttrMap
 from ..core.window import *
 from ..core.ipython_display import display_png_bytes, is_notebook, to_png_bytes
+from ..viewer.client import try_show_in_viewer
 from ..core.legend import LegendBox
 from ..core.svg import (
     SvgDocument,
@@ -503,6 +504,8 @@ class Grid(AttrObject):
         data = to_png_bytes(self)
         if is_notebook():
             display_png_bytes(data)
+        elif try_show_in_viewer(data, title="Grid"):
+            return
         else:
             Image.open(BytesIO(data)).show()
 

kaxe-1.8.1.dev2/src/kaxe/viewer/__init__.py

@@ -0,0 +1,11 @@
+"""Kaxe Viewer package."""
+
+from .client import DEFAULT_PORT, get_viewer_port, is_viewer_active, send_to_viewer, try_show_in_viewer
+
+__all__ = [
+    "DEFAULT_PORT",
+    "get_viewer_port",
+    "is_viewer_active",
+    "send_to_viewer",
+    "try_show_in_viewer",
+]

kaxe-1.8.1.dev2/src/kaxe/viewer/cli.py

@@ -0,0 +1,89 @@
+"""Command line interface for launching Kaxe Viewer."""
+
+import argparse
+import queue
+import sys
+
+from .client import get_viewer_port, is_viewer_active
+from .server import start_server_thread
+
+
+def run_viewer(port: int = 41783):
+    """Launch the Kaxe Viewer Pyglet window and HTTP server."""
+    # Ensure optional dependency is installed
+    from .._require_viewer import require_viewer
+
+    try:
+        require_viewer()
+    except ImportError as e:
+        print(f"\n[Kaxe] {e}\n", file=sys.stderr)
+        sys.exit(1)
+
+    import pyglet
+    from .window import KaxeViewerWindow
+
+    # Check if a viewer is already running on this port
+    if is_viewer_active(port):
+        print(f"\n[Kaxe Viewer] Port {port} is already in use.", file=sys.stderr)
+        print("A Kaxe Viewer instance is likely already open.\n", file=sys.stderr)
+        sys.exit(1)
+
+    plot_queue = queue.Queue()
+
+    # Start HTTP server on background thread
+    server, thread = start_server_thread(port, plot_queue)
+
+    print("=" * 64)
+    print(f"  Kaxe Viewer running on http://127.0.0.1:{port}")
+    print("  Keep this terminal open while working.")
+    print("  In another terminal, run your scripts calling .show()!")
+    print()
+    print("  Controls:")
+    print("    • Mouse Wheel     : Zoom (centered at cursor)")
+    print("    • Left Click Drag : Pan image")
+    print("    • [R] or [Space]  : Fit plot to window")
+    print("    • [1]             : 100% native resolution")
+    print("    • [←] / [→]       : Cycle through previous plots")
+    print("    • [C]             : Copy image to clipboard")
+    print("    • [S]             : Save image to file")
+    print("    • [H]             : Toggle HUD overlay")
+    print("=" * 64)
+
+    # Create Pyglet window and run Cocoa / OpenGL event loop on main thread
+    window = KaxeViewerWindow(plot_queue)
+    pyglet.app.run()
+
+
+def main(argv=None):
+    """Entrypoint for the `kaxe` CLI command."""
+    if argv is None:
+        argv = sys.argv[1:]
+
+    parser = argparse.ArgumentParser(
+        prog="kaxe",
+        description="Kaxe graphing & plotting toolkit CLI",
+    )
+    subparsers = parser.add_subparsers(dest="subcommand", help="Available subcommands")
+
+    # `kaxe viewer` command
+    viewer_parser = subparsers.add_parser(
+        "viewer",
+        help="Start the standalone Kaxe interactive plot viewer",
+    )
+    viewer_parser.add_argument(
+        "--port",
+        type=int,
+        default=get_viewer_port(),
+        help=f"Port to listen on (default: {get_viewer_port()})",
+    )
+
+    args = parser.parse_args(argv)
+
+    if args.subcommand == "viewer":
+        run_viewer(port=args.port)
+    else:
+        parser.print_help()
+
+
+if __name__ == "__main__":
+    main()

kaxe-1.8.1.dev2/src/kaxe/viewer/client.py

@@ -0,0 +1,82 @@
+"""Client-side probe and transmitter for Kaxe Viewer.
+
+This module intentionally uses ONLY Python standard library modules
+(socket, urllib, os) so that calling `plt.show()` never requires
+pyglet or any other viewer dependency.
+"""
+
+import os
+import socket
+import urllib.error
+import urllib.parse
+import urllib.request
+from typing import Optional
+
+DEFAULT_PORT = 41783
+
+
+def get_viewer_port() -> int:
+    """Return the configured viewer port from environment or the default."""
+    env_port = os.getenv("KAXE_VIEWER_PORT")
+    if env_port:
+        try:
+            return int(env_port)
+        except ValueError:
+            pass
+    return DEFAULT_PORT
+
+
+def is_viewer_active(port: Optional[int] = None, timeout: float = 0.03) -> bool:
+    """Check if the Kaxe Viewer is currently listening on localhost.
+
+    Uses a fast socket connection test with a short timeout (~30ms).
+    When no viewer is running, the OS kernel rejects the connection
+    virtually instantaneously (<1ms).
+    """
+    if port is None:
+        port = get_viewer_port()
+
+    try:
+        with socket.create_connection(("127.0.0.1", port), timeout=timeout):
+            return True
+    except (OSError, ConnectionRefusedError):
+        return False
+
+
+def send_to_viewer(
+    data: bytes,
+    title: str = "",
+    port: Optional[int] = None,
+    timeout: float = 2.0,
+) -> bool:
+    """Send image PNG bytes to the active Kaxe Viewer over HTTP POST.
+
+    Returns True if successfully received, False otherwise.
+    """
+    if port is None:
+        port = get_viewer_port()
+
+    url = f"http://127.0.0.1:{port}/show"
+    safe_title = urllib.parse.quote(title or "")
+    headers = {
+        "Content-Type": "image/png",
+        "X-Kaxe-Title": safe_title,
+    }
+
+    req = urllib.request.Request(url, data=data, headers=headers, method="POST")
+    try:
+        with urllib.request.urlopen(req, timeout=timeout) as response:
+            return response.status == 200
+    except (urllib.error.URLError, OSError):
+        return False
+
+
+def try_show_in_viewer(data: bytes, title: str = "") -> bool:
+    """Convenience helper: probe port and send image if viewer is alive.
+
+    Returns True if sent to viewer, False if fallback is needed.
+    """
+    port = get_viewer_port()
+    if is_viewer_active(port):
+        return send_to_viewer(data, title=title, port=port)
+    return False

kaxe-1.8.1.dev2/src/kaxe/viewer/server.py

@@ -0,0 +1,75 @@
+"""Lightweight HTTP server embedded inside Kaxe Viewer.
+
+Uses Python standard library `http.server.ThreadingHTTPServer` to receive
+plot images and metadata from client scripts without external web frameworks.
+"""
+
+from http.server import BaseHTTPRequestHandler, ThreadingHTTPServer
+import queue
+import threading
+import urllib.parse
+from typing import Any, Dict
+
+
+class PlotRequestHandler(BaseHTTPRequestHandler):
+    """Handles incoming /show POST requests and /ping health checks."""
+
+    # Reference to the shared Queue injected by start_server
+    plot_queue: queue.Queue = queue.Queue()
+
+    def do_GET(self):
+        if self.path in ("/ping", "/health"):
+            self.send_response(200)
+            self.send_header("Content-Type", "text/plain")
+            self.end_headers()
+            self.wfile.write(b"OK")
+        else:
+            self.send_response(404)
+            self.end_headers()
+
+    def do_POST(self):
+        if self.path == "/show":
+            content_length = int(self.headers.get("Content-Length", 0))
+            if content_length <= 0:
+                self.send_response(400)
+                self.end_headers()
+                self.wfile.write(b"Empty body")
+                return
+
+            img_bytes = self.rfile.read(content_length)
+            raw_title = self.headers.get("X-Kaxe-Title", "")
+            title = urllib.parse.unquote(raw_title) if raw_title else ""
+
+            item: Dict[str, Any] = {
+                "data": img_bytes,
+                "title": title,
+            }
+            self.plot_queue.put(item)
+
+            self.send_response(200)
+            self.send_header("Content-Type", "text/plain")
+            self.end_headers()
+            self.wfile.write(b"OK")
+        else:
+            self.send_response(404)
+            self.end_headers()
+
+    def log_message(self, format: str, *args):
+        # Format messages cleanly in terminal
+        first_arg = args[0] if args else ""
+        if "POST /show" in str(first_arg):
+            print(f"[Kaxe Viewer] New plot received ({format % args})")
+        # Suppress noisy health checks in console
+
+
+def start_server_thread(
+    port: int,
+    plot_queue: queue.Queue,
+    host: str = "127.0.0.1",
+) -> tuple[ThreadingHTTPServer, threading.Thread]:
+    """Start the HTTP server on a daemon background thread."""
+    PlotRequestHandler.plot_queue = plot_queue
+    server = ThreadingHTTPServer((host, port), PlotRequestHandler)
+    thread = threading.Thread(target=server.serve_forever, daemon=True)
+    thread.start()
+    return server, thread