artiq/artiq/gui/tools.py

import os

from gi.repository import Gtk


data_dir = os.path.abspath(os.path.dirname(__file__))


def getitem(d, item, default):
    try:
        return d[item]
    except KeyError:
        return default


class Window(Gtk.Window):
    def __init__(self, title, default_size, layout_dict=dict()):
        Gtk.Window.__init__(self, title=title + " - ARTIQ")

        self.set_wmclass("ARTIQ", "ARTIQ")
        self.set_icon_from_file(os.path.join(data_dir, "icon.png"))
        self.set_border_width(6)

        size = getitem(layout_dict, "size", default_size)
        self.set_default_size(size[0], size[1])
        try:
            position = layout_dict["position"]
        except KeyError:
            pass
        else:
            self.move(position[0], position[1])

    def get_layout_dict(self):
        return {
            "size": self.get_size(),
            "position": self.get_position()
        }


class LayoutManager:
    def __init__(self, db):
        self.db = db
        self.windows = dict()

    def create_window(self, cls, name, *args, **kwargs):
        try:
            win_layouts = self.db.request("win_layouts")
            layout_dict = win_layouts[name]
        except KeyError:
            layout_dict = dict()
        win = cls(*args, layout_dict=layout_dict, **kwargs)
        self.windows[name] = win
        return win

    def save(self):
        win_layouts = {name: window.get_layout_dict()
                       for name, window in self.windows.items()}
        self.db.set("win_layouts", win_layouts)


class ListSyncer:
    def __init__(self, store, init):
        self.store = store
        self.store.clear()
        for x in init:
            self.append(x)

    def append(self, x):
        self.store.append(self.convert(x))

    def insert(self, i, x):
        self.store.insert(i, self.convert(x))

    def __delitem__(self, key):
        del self.store[key]

    def convert(self, x):
        raise NotImplementedError


class DictSyncer:
    def __init__(self, store, init):
        self.store = store
        self.store.clear()
        self.order = []
        for k, v in sorted(init.items(), key=self.order_key):
            self.store.append(self.convert(k, v))
            self.order.append((k, self.order_key((k, v))))

    def _find_index(self, key):
        for i, e in enumerate(self.order):
            if e[0] == key:
                return i
        raise KeyError

    def __setitem__(self, key, value):
        try:
            i = self._find_index(key)
        except KeyError:
            pass
        else:
            del self.store[i]
            del self.order[i]
        ord_el = self.order_key((key, value))
        j = len(self.order)
        for i, (k, o) in enumerate(self.order):
            if o > ord_el:
                j = i
                break
        self.store.insert(j, self.convert(key, value))
        self.order.insert(j, (key, ord_el))

    def __delitem__(self, key):
        i = self._find_index(key)
        del self.store[i]
        del self.order[i]

    def order_key(self, kv_pair):
        raise NotImplementedError

    def convert(self, key, value):
        raise NotImplementedError