From b03cdbb50e045a5ddd5a026ba832cd024a1cb137 Mon Sep 17 00:00:00 2001 From: Stuart Langridge Date: Fri, 24 May 2019 11:00:16 +0100 Subject: Initial import --- .gitignore | 1 + README.md | 3 + magnus/__main__.py | 184 +++++++++++++++++++++++++++++++++++++++++++++++++++++ 3 files changed, 188 insertions(+) create mode 100644 .gitignore create mode 100644 README.md create mode 100644 magnus/__main__.py 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() + -- cgit