2 from gi.repository import Gtk as gtk, GdkPixbuf as gdkpix, Gdk as gdk, GObject as gobject
5 class notdone(Exception): pass
7 class future(threading.Thread):
11 super(future, self).__init__()
20 super(future, self).start()
26 except Exception as e:
28 gobject.idle_add(self._callcbs, True)
31 gobject.idle_add(self._callcbs, True)
33 def _callcbs(self, final):
35 for cb in self._notlist:
38 self._notlist = [] if final else nls
40 # Caller must hold GDK lock
44 self._notlist.append(cb)
49 gobject.idle_add(self._callcbs, False)
54 if self._exc is not None:
63 return self._exc != None or self._val != None
67 while self.is_alive():
71 class imgload(future):
72 def __init__(self, page):
73 super(imgload, self).__init__()
79 buf = gdkpix.PixbufLoader()
81 with self.page.open() as st:
94 return buf.get_pixbuf()
98 if self.st is None or self.st.clen is None:
100 return float(self.p) / float(self.st.clen)
102 class pagecache(object):
103 def __init__(self, sz=50):
107 def __getitem__(self, page):
109 for ol, f in self.bk:
113 self.bk.append((idl, f))
114 if len(self.bk) > self.sz:
115 self.bk = self.bk[-self.sz:]
118 def __delitem__(self, page):
120 for i, (ol, f) in enumerate(self.bk):
126 class relpageget(future):
127 def __init__(self, cur, prev, cache=None):
128 super(relpageget, self).__init__()
129 self.cur = lib.cursor(cur)
137 page = self.cur.prev()
139 page = self.cur.next()
140 except StopIteration:
147 class idpageget(future):
148 def __init__(self, base, idlist):
149 super(idpageget, self).__init__()
154 return lib.cursor(self.bnode.byidlist(self.idlist)).cur
156 class pageget(future):
157 def __init__(self, fnode):
158 super(pageget, self).__init__()
163 return lib.cursor(self.fnode).cur
165 class ccursor(object):
166 def __init__(self, ob, cache=None):
167 self.cur = lib.cursor(ob)
168 self.prev = relpageget(self.cur, True, cache)
169 self.next = relpageget(self.cur, False, cache)
171 class pageview(gtk.Widget):
172 def __init__(self, pixbuf):
173 super(pageview, self).__init__()
175 self.zoomed = None, None
178 self.interp = gdkpix.InterpType.HYPER
182 return self.pixbuf.get_width(), self.pixbuf.get_height()
185 alloc = self.get_allocation()
186 return alloc.width, alloc.height
188 def do_realize(self):
189 alloc = self.get_allocation()
190 attr = gdk.WindowAttr()
191 attr.window_type = gdk.WindowType.CHILD
194 attr.width = alloc.width
195 attr.height = alloc.height
196 attr.visual = self.get_visual()
197 attr.event_mask = self.get_events() | gdk.EventMask.EXPOSURE_MASK
198 a = gdk.WindowAttributesType
199 wnd = gdk.Window(self.get_parent_window(), attr, a.X | a.Y | a.VISUAL)
200 wnd.set_background_pattern(None)
202 self.register_window(wnd)
203 self.set_realized(True)
205 # XXX: Why don't these actually get called?
206 def get_preferred_width(self):
207 return 0, max(min(self.get_osize()[0], 4096), 0)
208 def get_preferred_height(self):
209 return 0, max(min(self.get_osize()[1], 4096), 0)
212 w, h = self.get_osize()
213 alloc = self.get_allocation()
214 return min(float(alloc.width) / float(w), float(alloc.height) / float(h))
216 def do_size_allocate(self, alloc):
217 self.set_allocation(alloc)
219 self.zoom = self.fitzoom()
221 aw, ah = self.get_asize()
222 zw, zh = self.get_zsize()
224 if zw >= aw and ox + aw > zw:
226 if zh >= ah and oy + ah > zh:
229 if self.get_realized():
230 self.get_window().move_resize(alloc.x, alloc.y, alloc.width, alloc.height)
232 def get_zoomed(self):
234 pz, zbuf = self.zoomed
236 w, h = self.get_osize()
237 zbuf = self.pixbuf.scale_simple(int(w * zoom), int(h * zoom), self.interp)
238 self.zoomed = zoom, zbuf
242 zbuf = self.get_zoomed()
243 return zbuf.get_width(), zbuf.get_height()
245 def do_draw(self, cr):
246 aw, ah = self.get_asize()
248 zbuf = self.get_zoomed()
249 zw, zh = self.get_zsize()
258 gdk.cairo_set_source_pixbuf(cr, zbuf, dx - ox, dy - oy)
261 def set_off(self, off):
262 aw, ah = self.get_asize()
263 zw, zh = self.get_zsize()
265 ox, oy = int(ox), int(oy)
266 if ox > zw - aw: ox = zw - aw
267 if oy > zh - ah: oy = zh - ah
273 def set_zoom(self, zoom):
274 if zoom is not None: zoom = float(zoom)
275 aw, ah = self.get_asize()
276 zw, zh = self.get_zsize()
277 dw, dh = zw - aw, zh - ah
279 xa = float(ox) / float(dw) if dw > 0 else 0.5
280 ya = float(oy) / float(dh) if dh > 0 else 0.5
284 self.zoom = self.fitzoom()
289 zw, zh = self.get_zsize()
290 dw, dh = zw - aw, zh - ah
291 ox = int(xa * dw) if dw > 0 else 0
292 oy = int(ya * dh) if dh > 0 else 0
293 self.set_off((ox, oy))
295 class msgproc(object):
296 def attach(self, reader):
298 self.msg = gtk.Alignment(yalign=0.5)
299 self.hlay = gtk.HBox()
300 self.lbl = gtk.Label("")
301 self.hlay.pack_start(self.lbl, True, True, 0)
303 self.msg.add(self.hlay)
305 self.rd.sbar.pack_start(self.msg, True, True, 0)
311 if self._prog is None:
312 self._prog = gtk.ProgressBar()
313 self.hlay.pack_start(self._prog, True, True, 5)
315 self._prog.set_fraction(p)
316 elif p is None and self._prog is not None:
317 self.hlay.remove(self._prog)
321 self.rd.sbar.remove(self.msg)
323 class pagefetch(msgproc):
324 def __init__(self, fpage, setcb=None):
328 def attach(self, reader):
329 super(pagefetch, self).attach(reader)
330 self.lbl.set_text("Fetching page...")
331 self.pg.notify(self.haspage)
334 if self.rd.pagefetch.cur != self: return False
337 if self.pg.val is not None:
338 self.rd.setpage(self.pg.val)
339 if self.setcb is not None:
340 self.setcb(self.pg.val)
341 self.rd.pagefetch.set(None)
343 class imgfetch(msgproc):
344 def __init__(self, fimage):
349 def attach(self, reader):
350 super(imgfetch, self).attach(reader)
351 self.lbl.set_text("Fetching image...")
352 self.img.notify(self.imgprog)
355 if self.rd.imgfetch.cur != self: return False
359 except Exception as e:
364 self.rd.imgfetch.set(None)
366 self.prog(self.img.prog)
370 self.rd.sbar.remove(self.msg)
373 if self.error is not None:
374 self.rd.pagelbl.set_text("Error fetching image: " + self.error)
376 class preload(msgproc):
377 def __init__(self, fpage):
380 def attach(self, reader):
381 super(preload, self).attach(reader)
382 self.lbl.set_text("Fetching next page...")
383 self.pg.notify(self.haspage)
386 if self.rd.preload.cur != self: return False
387 if not self.pg.done: return True
388 if self.pg.val is not None:
389 self.img = self.rd.cache[self.pg.val]
390 self.lbl.set_text("Loading next page...")
391 self.img.notify(self.imgprog)
393 self.rd.preload.set(None)
396 if self.rd.preload.cur != self: return False
398 self.rd.preload.set(None)
400 self.prog(self.img.prog)
404 self.rd.sbar.remove(self.msg)
406 class procslot(object):
407 __slots__ = ["cur", "p"]
408 def __init__(self, p):
413 if self.cur is not None:
424 class plistget(future):
425 def __init__(self, node):
426 super(plistget, self).__init__()
430 return list(self.node)
432 class loadplist(object):
433 def __init__(self, pnode):
435 self.flist = plistget(self.pnode)
437 def attach(self, sbox):
439 self.flist.notify(self.haslist)
442 if self.sbox.loadlist.cur != self: return False
443 if not self.flist.done: return True
444 self.sbox.setlist(self.flist.val)
446 class sbox(gtk.ComboBox):
447 def __init__(self, reader, ptnode):
448 super(sbox, self).__init__()
451 self.pnode, self.pidx = self.node.stack[-1]
453 self.bk = gtk.ListStore(str)
454 self.set_model(self.bk)
455 cell = gtk.CellRendererText()
456 self.pack_start(cell, True)
457 self.add_attribute(cell, "text", 0)
460 self.set_sensitive(False)
461 self.set_focus_on_click(False)
462 self.bk.append([ptnode.name])
463 self.loadlist = procslot(self)
464 self.loadlist.set(loadplist(self.pnode))
466 def setlist(self, ls):
468 for i, ch in enumerate(ls):
469 self.bk.append(["%i/%i: %s" % (i + 1, len(ls), ch.name)])
470 self.set_active(self.pidx)
471 self.set_sensitive(True)
472 self.connect("changed", self.changed_cb)
474 def changed_cb(self, wdg, data=None):
475 self.rd.fetchpage(pageget(self.pnode[self.get_active()]))
477 class profprop(object):
478 def __init__(self, key, default=None):
480 self.default = default
482 def __get__(self, ins, cls):
483 return ins.profile.props.get(self.key, self.default)
485 def __set__(self, ins, val):
486 ins.profile.props[self.key] = val
489 def __delete__(self, ins):
490 del ins.profile.props[self.key]
493 class reader(gtk.Window):
494 def __init__(self, manga, prof=None):
495 super(reader, self).__init__()
496 self.connect("delete_event", lambda wdg, ev, data=None: False)
497 self.connect("destroy", lambda wdg, data=None: self.quit())
498 self.connect("key_press_event", self.key)
499 self.cache = pagecache()
500 self.pagefetch = procslot(self)
501 self.imgfetch = procslot(self)
502 self.preload = procslot(self)
503 self.profile = prof if prof else profile.memmanga(None, None, manga.id)
511 self.pfr = gtk.Frame()
512 self.pfr.set_shadow_type(gtk.ShadowType.NONE)
513 vlay.pack_start(self.pfr, True, True, 0)
515 self.sboxbar = gtk.HBox()
516 algn = gtk.Alignment(yalign=0.5)
517 sboxlbl = gtk.Label(self.manga.name + u": ")
520 self.sboxbar.pack_start(algn, False, True, 0)
522 vlay.pack_start(self.sboxbar, False, True, 0)
524 self.sbar = gtk.HBox()
525 self.pagelbl = gtk.Label("")
526 algn = gtk.Alignment(yalign=0.5)
527 algn.add(self.pagelbl)
529 self.sbar.pack_start(algn, True, True, 0)
531 vlay.pack_end(self.sbar, False, True, 0)
536 if self.curpage is not None:
537 self.fetchpage(idpageget(self.manga, self.curpage))
539 self.fetchpage(pageget(self.manga))
542 zmode = profprop("zmode", "fit")
543 curpage = profprop("curpage")
545 def updpagelbl(self):
546 if self.page is None:
547 self.pagelbl.set_text("")
549 w, h = self.page.get_osize()
550 self.pagelbl.set_text(u"%s\u00d7%s (%d%%)" % (w, h, int(self.page.zoom * 100)))
552 def updsboxes(self, page):
553 nodes = [node for node, idx in page.stack[1:]] + [page]
554 l = min(len(self.sboxes), len(nodes))
555 for i, (pbox, node) in enumerate(zip(self.sboxes, nodes)):
556 if pbox.node != node:
559 for i in xrange(l, len(self.sboxes)):
560 self.sboxbar.remove(self.sboxes[i])
561 self.sboxes = self.sboxes[:l]
562 for i in xrange(l, len(nodes)):
563 new = sbox(self, nodes[i])
564 self.sboxbar.pack_start(new, False, True, 5)
565 self.sboxes.append(new)
568 def setimg(self, img):
569 if self.page is not None:
570 self.pfr.remove(self.page)
573 self.page = pageview(img)
574 if self.zmode == "1":
575 self.page.set_zoom(1)
576 self.page.set_off((0, 0))
577 self.pfr.add(self.page)
581 def setpage(self, page):
582 if self.point is not None:
585 self.curpage = page.idlist()
586 self.point = ccursor(page, self.cache)
587 self.imgfetch.set(imgfetch(self.cache[page]))
592 def fetchpage(self, fpage, setcb=None):
593 self.imgfetch.set(None)
594 proc = pagefetch(fpage, setcb)
595 self.pagefetch.set(proc)
599 self.set_title(u"Automanga \u2013 " + self.manga.name)
603 return self.page.zoom
605 def zoom(self, zoom):
606 self.page.set_zoom(zoom)
610 ox, oy = self.page.off
612 self.page.set_off((ox + px, oy + py))
614 def key(self, wdg, ev, data=None):
615 if ev.keyval in [ord('Q'), ord('q')]:
617 elif ev.keyval in [65307]:
618 if self.page is not None:
619 self.pagefetch.set(None)
620 self.imgfetch.set(None)
621 if self.page is not None:
622 if ev.keyval in [ord('O'), ord('o')]:
625 elif ev.keyval in [ord('P'), ord('p')]:
628 elif ev.keyval in [ord('[')]:
629 self.zoom = min(self.zoom * 1.25, 3)
630 elif ev.keyval in [ord(']')]:
632 elif ev.keyval in [ord('h')]:
634 elif ev.keyval in [ord('j')]:
636 elif ev.keyval in [ord('k')]:
638 elif ev.keyval in [ord('l')]:
640 elif ev.keyval in [ord('H')]:
641 self.page.set_off((0, self.page.off[1]))
642 elif ev.keyval in [ord('J')]:
643 self.page.set_off((self.page.off[0], self.page.get_asize()[1]))
644 elif ev.keyval in [ord('K')]:
645 self.page.set_off((self.page.off[1], 0))
646 elif ev.keyval in [ord('L')]:
647 self.page.set_off((self.page.get_asize()[0], self.page.off[1]))
648 if self.point is not None:
649 if ev.keyval in [ord(' ')]:
650 if self.page and self.page.off[1] + self.page.get_asize()[1] < self.page.get_zsize()[1]:
651 self.pan((0, self.page.get_asize()[1] - 50))
653 self.fetchpage(self.point.next, lambda page: self.preload.set(preload(relpageget(page, False, self.cache))))
654 elif ev.keyval in [65288]:
655 if self.page and self.page.off[1] > 0:
656 self.pan((0, -(self.page.get_asize()[1] - 50)))
658 self.fetchpage(self.point.prev, lambda page: self.preload.set(preload(relpageget(page, True, self.cache))))
659 elif ev.keyval in [ord('R'), ord('r')]:
660 page = self.point.cur.cur
662 self.imgfetch.set(imgfetch(self.cache[page]))
667 gobject.type_register(reader)