1 import threading, gtk, gio, gobject
4 class notdone(Exception): pass
6 class future(threading.Thread):
10 super(future, self).__init__()
19 super(future, self).start()
25 except Exception as e:
27 gobject.idle_add(self._callcbs, True)
30 gobject.idle_add(self._callcbs, True)
32 def _callcbs(self, final):
34 for cb in self._notlist:
37 self._notlist = [] if final else nls
39 # Caller must hold GDK lock
43 self._notlist.append(cb)
48 gobject.idle_add(self._callcbs, False)
53 if self._exc is not None:
62 return self._exc != None or self._val != None
66 while self.is_alive():
70 class imgload(future):
71 def __init__(self, page):
72 super(imgload, self).__init__()
79 with self.page.open() as st:
92 return gtk.gdk.pixbuf_new_from_stream(gio.memory_input_stream_new_from_data(str(buf)))
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 class relpageget(future):
119 def __init__(self, cur, prev, cache=None):
120 super(relpageget, self).__init__()
121 self.cur = lib.cursor(cur)
129 page = self.cur.prev()
131 page = self.cur.next()
132 except StopIteration:
139 class pageget(future):
140 def __init__(self, fnode):
141 super(pageget, self).__init__()
146 return lib.cursor(self.fnode).cur
148 class ccursor(object):
149 def __init__(self, ob, cache=None):
150 self.cur = lib.cursor(ob)
151 self.prev = relpageget(self.cur, True, cache)
152 self.next = relpageget(self.cur, False, cache)
154 class pageview(gtk.Widget):
155 def __init__(self, pixbuf):
156 super(pageview, self).__init__()
158 self.zoomed = None, None
161 self.interp = gtk.gdk.INTERP_HYPER
165 return self.pixbuf.get_width(), self.pixbuf.get_height()
168 return self.allocation.width, self.allocation.height
170 def do_realize(self):
171 self.set_flags(self.flags() | gtk.REALIZED)
172 alloc = self.allocation
173 self.window = gtk.gdk.Window(self.get_parent_window(),
174 width=alloc.width, height=alloc.height,
175 window_type = gtk.gdk.WINDOW_CHILD,
176 wclass = gtk.gdk.INPUT_OUTPUT,
177 event_mask = self.get_events() | gtk.gdk.EXPOSURE_MASK
179 self.window.set_user_data(self)
180 self.style.attach(self.window)
181 self.style.set_background(self.window, gtk.STATE_NORMAL)
182 self.window.move_resize(*alloc)
184 def do_unrealize(self):
185 self.window.set_user_data(None)
187 def do_size_request(self, req):
188 w, h = self.get_osize()
189 req.width, req.height = max(min(w, 4096), 0), max(min(h, 4096), 0)
192 w, h = self.get_osize()
193 alloc = self.allocation
194 return min(float(alloc.width) / float(w), float(alloc.height) / float(h))
196 def do_size_allocate(self, alloc):
197 self.allocation = alloc
199 self.zoom = self.fitzoom()
200 if self.flags() & gtk.REALIZED:
201 self.window.move_resize(*alloc)
203 def get_zoomed(self):
205 pz, zbuf = self.zoomed
207 w, h = self.get_osize()
208 zbuf = self.pixbuf.scale_simple(int(w * zoom), int(h * zoom), self.interp)
209 self.zoomed = zoom, zbuf
213 zbuf = self.get_zoomed()
214 return zbuf.get_width(), zbuf.get_height()
216 def do_expose_event(self, event):
217 aw, ah = self.get_asize()
219 zbuf = self.get_zoomed()
220 zw, zh = self.get_zsize()
229 gc = self.style.fg_gc[gtk.STATE_NORMAL]
230 self.window.draw_pixbuf(gc, zbuf, ox, oy, dx, dy, dw, dh)
232 def set_off(self, off):
233 aw, ah = self.get_asize()
234 zw, zh = self.get_zsize()
236 ox, oy = int(ox), int(oy)
237 if ox > zw - aw: ox = zw - aw
238 if oy > zh - ah: oy = zh - ah
244 def set_zoom(self, zoom):
245 if zoom is not None: zoom = float(zoom)
246 aw, ah = self.get_asize()
247 zw, zh = self.get_zsize()
248 dw, dh = zw - aw, zh - ah
250 xa = float(ox) / float(dw) if dw > 0 else 0.5
251 ya = float(oy) / float(dh) if dh > 0 else 0.5
255 self.zoom = self.fitzoom()
260 zw, zh = self.get_zsize()
261 dw, dh = zw - aw, zh - ah
262 ox = int(xa * dw) if dw > 0 else 0
263 oy = int(ya * dh) if dh > 0 else 0
264 self.set_off((ox, oy))
265 gobject.type_register(pageview)
267 class msgproc(object):
268 def attach(self, reader):
270 self.msg = gtk.Alignment(0, 0.5, 0, 0)
271 self.hlay = gtk.HBox()
272 self.lbl = gtk.Label("")
273 self.hlay.pack_start(self.lbl)
275 self.msg.add(self.hlay)
277 self.rd.sbar.pack_start(self.msg)
282 if p is not None and self._prog is None:
283 self._prog = gtk.ProgressBar()
284 self._prog.set_fraction(p)
285 self.hlay.pack_start(self._prog, padding=5)
287 elif p is None and self._prog is not None:
288 self.hlay.remove(self._prog)
292 self.rd.sbar.remove(self.msg)
294 class pagefetch(msgproc):
295 def __init__(self, fpage, setcb=None):
299 def attach(self, reader):
300 super(pagefetch, self).attach(reader)
301 self.lbl.set_text("Fetching page...")
302 self.pg.notify(self.haspage)
305 if self.rd.pagefetch.cur != self: return False
308 if self.pg.val is not None:
309 self.rd.setpage(self.pg.val)
310 if self.setcb is not None:
311 self.setcb(self.pg.val)
312 self.rd.pagefetch.set(None)
314 class imgfetch(msgproc):
315 def __init__(self, fimage):
320 def attach(self, reader):
321 super(imgfetch, self).attach(reader)
322 self.lbl.set_text("Fetching image...")
323 self.img.notify(self.imgprog)
326 if self.rd.imgfetch.cur != self: return False
330 except Exception as e:
335 self.rd.imgfetch.set(None)
337 self.prog(self.img.prog)
341 self.rd.sbar.remove(self.msg)
344 if self.error is not None:
345 self.rd.pagelbl.set_text("Error fetching image: " + self.error)
347 class preload(msgproc):
348 def __init__(self, fpage):
351 def attach(self, reader):
352 super(preload, self).attach(reader)
353 self.lbl.set_text("Fetching next page...")
354 self.pg.notify(self.haspage)
357 if self.rd.preload.cur != self: return False
358 if not self.pg.done: return True
359 if self.pg.val is not None:
360 self.img = self.rd.cache[self.pg.val]
361 self.lbl.set_text("Loading next page...")
362 self.img.notify(self.imgprog)
364 self.rd.preload.set(None)
367 if self.rd.preload.cur != self: return False
369 self.rd.preload.set(None)
371 self.prog(self.img.prog)
375 self.rd.sbar.remove(self.msg)
377 class procslot(object):
378 __slots__ = ["cur", "p"]
379 def __init__(self, p):
384 if self.cur is not None:
395 class plistget(future):
396 def __init__(self, node):
397 super(plistget, self).__init__()
401 return list(self.node)
403 class loadplist(object):
404 def __init__(self, pnode):
406 self.flist = plistget(self.pnode)
408 def attach(self, sbox):
410 self.flist.notify(self.haslist)
413 if self.sbox.loadlist.cur != self: return False
414 if not self.flist.done: return True
415 self.sbox.setlist(self.flist.val)
417 class sbox(gtk.ComboBox):
418 def __init__(self, reader, ptnode):
419 super(sbox, self).__init__()
422 self.pnode, self.pidx = self.node.stack[-1]
424 self.bk = gtk.ListStore(str)
425 self.set_model(self.bk)
426 cell = gtk.CellRendererText()
427 self.pack_start(cell, True)
428 self.add_attribute(cell, "text", 0)
431 self.set_sensitive(False)
432 self.set_focus_on_click(False)
433 self.bk.append([ptnode.name])
434 self.loadlist = procslot(self)
435 self.loadlist.set(loadplist(self.pnode))
437 def setlist(self, ls):
439 for i, ch in enumerate(ls):
440 self.bk.append(["%i/%i: %s" % (i + 1, len(ls), ch.name)])
441 self.set_active(self.pidx)
442 self.set_sensitive(True)
443 self.connect("changed", self.changed_cb)
445 def changed_cb(self, wdg, data=None):
446 self.rd.fetchpage(pageget(self.pnode[self.get_active()]))
448 class reader(gtk.Window):
449 def __init__(self, manga):
450 super(reader, self).__init__(gtk.WINDOW_TOPLEVEL)
451 self.connect("delete_event", lambda wdg, ev, data=None: False)
452 self.connect("destroy", lambda wdg, data=None: self.quit())
453 self.connect("key_press_event", self.key)
454 self.cache = pagecache()
455 self.pagefetch = procslot(self)
456 self.imgfetch = procslot(self)
457 self.preload = procslot(self)
465 self.pfr = gtk.Frame(None)
466 self.pfr.set_shadow_type(gtk.SHADOW_NONE)
467 vlay.pack_start(self.pfr)
469 self.sboxbar = gtk.HBox()
470 algn = gtk.Alignment(0, 0.5, 0, 0)
471 sboxlbl = gtk.Label(self.manga.name + u": ")
474 self.sboxbar.pack_start(algn, False)
476 vlay.pack_start(self.sboxbar, False)
478 self.sbar = gtk.HBox()
479 self.pagelbl = gtk.Label("")
480 algn = gtk.Alignment(0, 0.5, 0, 0)
481 algn.add(self.pagelbl)
483 self.sbar.pack_start(algn)
485 vlay.pack_end(self.sbar, False)
490 self.fetchpage(pageget(self.manga))
493 def updpagelbl(self):
494 if self.page is None:
495 self.pagelbl.set_text("")
497 w, h = self.page.get_osize()
498 self.pagelbl.set_text(u"%s\u00d7%s (%d%%)" % (w, h, int(self.page.zoom * 100)))
500 def updsboxes(self, page):
501 nodes = [node for node, idx in page.stack[1:]] + [page]
502 l = min(len(self.sboxes), len(nodes))
503 for i, (pbox, node) in enumerate(zip(self.sboxes, nodes)):
504 if pbox.node != node:
507 for i in xrange(l, len(self.sboxes)):
508 self.sboxbar.remove(self.sboxes[i])
509 self.sboxes = self.sboxes[:l]
510 for i in xrange(l, len(nodes)):
511 new = sbox(self, nodes[i])
512 self.sboxbar.pack_start(new, False, padding=5)
513 self.sboxes.append(new)
516 def setimg(self, img):
517 if self.page is not None:
518 self.pfr.remove(self.page)
521 self.page = pageview(img)
522 self.pfr.add(self.page)
526 def setpage(self, page):
527 if self.point is not None:
530 self.point = ccursor(page, self.cache)
531 self.imgfetch.set(imgfetch(self.cache[page]))
536 def fetchpage(self, fpage, setcb=None):
537 self.imgfetch.set(None)
538 proc = pagefetch(fpage, setcb)
539 self.pagefetch.set(proc)
543 self.set_title(u"Automanga \u2013 " + self.manga.name)
547 return self.page.zoom
549 def zoom(self, zoom):
550 self.page.set_zoom(zoom)
554 ox, oy = self.page.off
556 self.page.set_off((ox + px, oy + py))
558 def key(self, wdg, ev, data=None):
559 if ev.keyval in [ord('Q'), ord('q')]:
561 elif ev.keyval in [65307]:
562 if self.page is not None:
563 self.pagefetch.set(None)
564 self.imgfetch.set(None)
565 if self.page is not None:
566 if ev.keyval in [ord('O'), ord('o')]:
568 elif ev.keyval in [ord('P'), ord('p')]:
570 elif ev.keyval in [ord('[')]:
571 self.zoom = min(self.zoom * 1.25, 3)
572 elif ev.keyval in [ord(']')]:
574 elif ev.keyval in [ord('h')]:
576 elif ev.keyval in [ord('j')]:
578 elif ev.keyval in [ord('k')]:
580 elif ev.keyval in [ord('l')]:
582 elif ev.keyval in [ord('H')]:
583 self.page.set_off((0, self.page.off[1]))
584 elif ev.keyval in [ord('J')]:
585 self.page.set_off((self.page.off[0], self.page.get_asize()[1]))
586 elif ev.keyval in [ord('K')]:
587 self.page.set_off((self.page.off[1], 0))
588 elif ev.keyval in [ord('L')]:
589 self.page.set_off((self.page.get_asize()[0], self.page.off[1]))
590 if self.point is not None:
591 if ev.keyval in [ord(' ')]:
592 self.fetchpage(self.point.next, lambda page: self.preload.set(preload(relpageget(page, False, self.cache))))
593 elif ev.keyval in [65288]:
594 self.fetchpage(self.point.prev, lambda page: self.preload.set(preload(relpageget(page, True, self.cache))))
599 gobject.type_register(reader)