Initial import

3 files changed, 188 insertions, 0 deletions

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..bee8a64
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1 @@
+__pycache__
diff --git a/README.md b/README.md
new file mode 100644
index 0000000..bcf40da
--- /dev/null
+++ b/README.md
@@ -0,0 +1,3 @@
+# Magnus
+
+A very simple screen magnifier app for Ubuntu.
diff --git a/magnus/__main__.py b/magnus/__main__.py
new file mode 100644
index 0000000..bc2d1a8
--- /dev/null
+++ b/magnus/__main__.py
@@ -0,0 +1,184 @@
+#!/usr/bin/env python3
+
+import gi
+gi.require_version('Gtk', '3.0')
+from gi.repository import Gtk, Gdk, GLib, GdkPixbuf, Gio
+import cairo, math, json, os, codecs, time, subprocess, sys
+
+__VERSION__ = "1.0"
+
+
+class Main(object):
+    def __init__(self):
+        self.zoomlevel = 2
+        self.app = Gtk.Application.new("org.kryogenix.magnus", Gio.ApplicationFlags.HANDLES_COMMAND_LINE)
+        self.app.connect("command-line", self.handle_commandline)
+        self.resize_timeout = None
+        self.window_metrics = None
+        self.window_metrics_restored = False
+
+    def handle_commandline(self, app, cmdline):
+        if hasattr(self, "w"):
+            # already started
+            if "--about" in cmdline.get_arguments():
+                self.show_about_dialog()
+            return 0
+        # First time startup
+        self.start_everything_first_time()
+        if "--about" in cmdline.get_arguments():
+            self.show_about_dialog()
+        return 0
+
+    def start_everything_first_time(self, on_window_map=None):
+        GLib.set_application_name("Magnus")
+
+        # the window
+        self.w = Gtk.ApplicationWindow.new(self.app)
+        self.w.set_size_request(300, 300)
+        self.w.set_title("Magnus")
+        self.w.connect("destroy", Gtk.main_quit)
+        self.w.connect("configure-event", self.read_window_size)
+        self.w.connect("configure-event", self.window_configure)
+        devman = self.w.get_screen().get_display().get_device_manager()
+        self.pointer = devman.get_client_pointer()
+
+        # the headerbar
+        head = Gtk.HeaderBar()
+        head.set_show_close_button(True)
+        head.props.title = "Magnus"
+        self.w.set_titlebar(head)
+
+        # the zoom chooser
+        zoom = Gtk.ComboBoxText.new()
+        for i in range(2, 5):
+            zoom.append(str(i), "{}×".format(i))
+        zoom.set_active(0)
+        zoom.connect("changed", self.set_zoom)
+        head.pack_end(zoom)
+
+        # the box that contains everything
+        self.img = Gtk.Image()
+        scrolled_window = Gtk.ScrolledWindow()
+        scrolled_window.add(self.img)
+        self.w.add(scrolled_window)
+
+        # and, go
+        self.w.show_all()
+
+        self.width = 0
+        self.height = 0
+        GLib.timeout_add(250, self.read_window_size)
+
+        # and, poll
+        GLib.timeout_add(250, self.poll)
+
+        GLib.idle_add(self.load_config)
+
+    def set_zoom(self, zoom):
+        self.zoomlevel = int(zoom.get_active_text()[0])
+
+    def read_window_size(self, *args):
+        loc = self.w.get_size()
+        self.width = loc.width
+        self.height = loc.height
+
+    def show_about_dialog(self, *args):
+        about_dialog = Gtk.AboutDialog()
+        about_dialog.set_artists(["Stuart Langridge"])
+        about_dialog.set_authors(["Stuart Langridge"])
+        about_dialog.set_version(__VERSION__)
+        about_dialog.set_license_type(Gtk.License.MIT_X11)
+        about_dialog.set_website("https://www.kryogenix.org/code/magnus")
+        about_dialog.run()
+        if about_dialog: about_dialog.destroy()
+
+    def poll(self):
+        root = Gdk.get_default_root_window()
+        display = Gdk.Display.get_default()
+        (screen, x, y, modifier) = display.get_pointer()
+        scaled_width = self.width // self.zoomlevel
+        scaled_height = self.height // self.zoomlevel
+        scaled_xoff = scaled_width // 2
+        scaled_yoff = scaled_height // 2
+        screenshot = Gdk.pixbuf_get_from_window(root, x - scaled_xoff, y - scaled_yoff, scaled_width, scaled_height)
+        scaled_pb = screenshot.scale_simple(self.width, self.height, GdkPixbuf.InterpType.NEAREST)
+        self.img.set_from_pixbuf(scaled_pb)
+        return True
+
+    def window_configure(self, window, ev):
+        if not self.window_metrics_restored: return False
+        if self.resize_timeout:
+            GLib.source_remove(self.resize_timeout)
+        self.resize_timeout = GLib.timeout_add_seconds(1, self.save_window_metrics,
+            {"x":ev.x, "y":ev.y, "w":ev.width, "h":ev.height})
+
+    def save_window_metrics(self, props):
+        scr = self.w.get_screen()
+        sw = float(scr.get_width())
+        sh = float(scr.get_height())
+        # We save window dimensions as fractions of the screen dimensions, to cope with screen
+        # resolution changes while we weren't running
+        self.window_metrics = {
+            "ww": props["w"] / sw,
+            "wh": props["h"] / sh,
+            "wx": props["x"] / sw,
+            "wy": props["y"] / sh
+        }
+        self.serialise()
+
+    def restore_window_metrics(self, metrics):
+        scr = self.w.get_screen()
+        sw = float(scr.get_width())
+        sh = float(scr.get_height())
+        self.w.set_size_request(int(sw * metrics["ww"]), int(sh * metrics["wh"]))
+        self.w.move(int(sw * metrics["wx"]), int(sh * metrics["wy"]))
+
+    def get_cache_file(self):
+        return os.path.join(GLib.get_user_cache_dir(), "magnus.json")
+
+    def serialise(self, *args, **kwargs):
+        # yeah, yeah, supposed to use Gio's async file stuff here. But it was writing
+        # corrupted files, and I have no idea why; probably the Python var containing
+        # the data was going out of scope or something. Anyway, we're only storing
+        # a small JSON file, so life's too short to hammer on this; we'll write with
+        # Python and take the hit.
+        fp = codecs.open(self.get_cache_file(), encoding="utf8", mode="w")
+        data = {"zoom": self.zoomlevel}
+        if self.window_metrics:
+            data["metrics"] = self.window_metrics
+        json.dump(data, fp, indent=2)
+        fp.close()
+
+    def load_config(self):
+        f = Gio.File.new_for_path(self.get_cache_file())
+        f.load_contents_async(None, self.finish_loading_history)
+
+    def finish_loading_history(self, f, res):
+        try:
+            try:
+                success, contents, _ = f.load_contents_finish(res)
+            except GLib.Error as e:
+                print("couldn't restore settings (error: %s), so assuming they're blank" % (e,))
+                contents = "{}" # fake contents
+
+            data = json.loads(contents)
+            zl = data.get("zoom")
+            if zl:
+                self.zoomlevel = zl
+            metrics = data.get("metrics")
+            if metrics:
+                self.restore_window_metrics(metrics)
+            self.window_metrics_restored = True
+
+        except:
+            #print "Failed to restore data"
+            raise
+
+
+
+def main():
+    m = Main()
+    m.app.run(sys.argv)
+
+if __name__ == "__main__": main()
+