| 1 | 1 |
new file mode 100755 |
| ... | ... |
@@ -0,0 +1,98 @@ |
| 1 |
+#!/usr/bin/env python |
|
| 2 |
+# |
|
| 3 |
+# Copyright 2015 Jose Fonseca |
|
| 4 |
+# |
|
| 5 |
+# This program is free software: you can redistribute it and/or modify it |
|
| 6 |
+# under the terms of the GNU Lesser General Public License as published |
|
| 7 |
+# by the Free Software Foundation, either version 3 of the License, or |
|
| 8 |
+# (at your option) any later version. |
|
| 9 |
+# |
|
| 10 |
+# This program is distributed in the hope that it will be useful, |
|
| 11 |
+# but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
| 12 |
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
| 13 |
+# GNU Lesser General Public License for more details. |
|
| 14 |
+# |
|
| 15 |
+# You should have received a copy of the GNU Lesser General Public License |
|
| 16 |
+# along with this program. If not, see <http://www.gnu.org/licenses/>. |
|
| 17 |
+# |
|
| 18 |
+ |
|
| 19 |
+ |
|
| 20 |
+import sys |
|
| 21 |
+import os.path |
|
| 22 |
+import traceback |
|
| 23 |
+ |
|
| 24 |
+from gi.repository import Gtk |
|
| 25 |
+from gi.repository import Gdk |
|
| 26 |
+from gi.repository import GdkPixbuf |
|
| 27 |
+ |
|
| 28 |
+from xdot import DotWidget, DotWindow |
|
| 29 |
+ |
|
| 30 |
+ |
|
| 31 |
+class TestDotWidget(DotWidget): |
|
| 32 |
+ |
|
| 33 |
+ def __init__(self, name): |
|
| 34 |
+ DotWidget.__init__(self) |
|
| 35 |
+ self.name = name |
|
| 36 |
+ |
|
| 37 |
+ def on_draw(self, widget, cr): |
|
| 38 |
+ DotWidget.on_draw(self, widget, cr) |
|
| 39 |
+ |
|
| 40 |
+ if True: |
|
| 41 |
+ # Cairo screenshot |
|
| 42 |
+ |
|
| 43 |
+ import cairo |
|
| 44 |
+ |
|
| 45 |
+ dpi = 96.0 |
|
| 46 |
+ zoom_ratio = dpi/72.0 |
|
| 47 |
+ w = int(self.graph.width*zoom_ratio) |
|
| 48 |
+ h = int(self.graph.height*zoom_ratio) |
|
| 49 |
+ |
|
| 50 |
+ surface = cairo.ImageSurface(cairo.FORMAT_ARGB32, w, h) |
|
| 51 |
+ |
|
| 52 |
+ cr = cairo.Context(surface) |
|
| 53 |
+ |
|
| 54 |
+ cr.set_source_rgba(1.0, 1.0, 1.0, 1.0) |
|
| 55 |
+ cr.paint() |
|
| 56 |
+ |
|
| 57 |
+ cr.scale(zoom_ratio, zoom_ratio) |
|
| 58 |
+ |
|
| 59 |
+ self.graph.draw(cr, highlight_items=self.highlight) |
|
| 60 |
+ |
|
| 61 |
+ surface.write_to_png(self.name + '.png') |
|
| 62 |
+ |
|
| 63 |
+ if False: |
|
| 64 |
+ # GTK 3 screenshot |
|
| 65 |
+ |
|
| 66 |
+ window = self.get_window() |
|
| 67 |
+ |
|
| 68 |
+ w = window.get_width() |
|
| 69 |
+ h = window.get_height() |
|
| 70 |
+ |
|
| 71 |
+ pixbuf = Gdk.pixbuf_get_from_window(window, 0, 0, w, h) |
|
| 72 |
+ |
|
| 73 |
+ pixbuf.savev(self.name + '.png', 'png', (), ()) |
|
| 74 |
+ |
|
| 75 |
+ Gtk.main_quit() |
|
| 76 |
+ |
|
| 77 |
+ |
|
| 78 |
+def main(): |
|
| 79 |
+ for arg in sys.argv[1:]: |
|
| 80 |
+ sys.stdout.write(arg + '\n') |
|
| 81 |
+ sys.stdout.flush() |
|
| 82 |
+ name, ext = os.path.splitext(os.path.basename(arg)) |
|
| 83 |
+ dotcode = open(arg).read() |
|
| 84 |
+ widget = TestDotWidget(name) |
|
| 85 |
+ window = DotWindow(widget) |
|
| 86 |
+ window.connect('delete-event', Gtk.main_quit)
|
|
| 87 |
+ try: |
|
| 88 |
+ window.set_dotcode(dotcode) |
|
| 89 |
+ except: |
|
| 90 |
+ exc_info = sys.exc_info() |
|
| 91 |
+ traceback.print_exception(*exc_info) |
|
| 92 |
+ window.show() |
|
| 93 |
+ Gtk.main() |
|
| 94 |
+ window.destroy() |
|
| 95 |
+ |
|
| 96 |
+ |
|
| 97 |
+if __name__ == '__main__': |
|
| 98 |
+ main() |