Matplotlib -> Tkinter

2 files changed, 43 insertions, 7 deletions

diff --git a/example.py b/example.py
index 84fb9f0..4c00845 100755
--- a/example.py
+++ b/example.py
@@ -1,7 +1,35 @@
 #!/usr/bin/env python
 
 from src.scalar import Scalar
-from src.graph import Graph
+
+h = 0.0001
+
+# def one():
+#     a = Scalar(2, label='a')
+#     b = Scalar(-3, label='b')
+#     c = Scalar(10, label='c')
+#     f = Scalar(-2, label='f')
+#
+#     d = a * b; d.label = 'd'
+#     e = d + c; e.label = 'e'
+#     L = e * f; L.label = 'L'
+#
+#     return L.data
+#
+# def two():
+#     a = Scalar(2, label='a')
+#     b = Scalar(-3, label='b')
+#     c = Scalar(10, label='c')
+#     f = Scalar(-2, label='f')
+#
+#     d = a * b; d.label = 'd'
+#     d.data += h
+#     e = d + c; e.label = 'e'
+#     L = e * f; L.label = 'L'
+#
+#     return L.data
+#
+# print((two() - one()) / h)
 
 a = Scalar(2, label='a')
 b = Scalar(-3, label='b')
@@ -15,5 +43,8 @@ L = e * f; L.label = 'L'
 L.grad = 1.0
 e.grad = -2.0
 f.grad = 4.0
+d.grad = -2.0
+c.grad = -2.0
 
+from src.graph import Graph
 Graph(L).show()
diff --git a/src/graph.py b/src/graph.py
index 9aecde1..6c8ab53 100644
--- a/src/graph.py
+++ b/src/graph.py
@@ -1,4 +1,4 @@
-import io
+import tkinter as tk
 
 from graphviz import Digraph
 import matplotlib.pyplot as plt
@@ -49,8 +49,13 @@ class Graph:
             self.dot.edge(str(id(node1)), str(id(node2)) + node2._op)
 
     def show(self):
-        fp = io.BytesIO(self.dot.pipe(format='jpeg'))
-        with fp:
-            img = mpimg.imread(fp, format='jpeg')
-        plt.imshow(img)
-        plt.show()
+        root = tk.Tk()
+        root.title('float')
+
+        data = self.dot.pipe(format='png')
+        img = tk.PhotoImage(data=data, format='png')
+
+        panel = tk.Label(root, image=img)
+        panel.pack();
+
+        tk.mainloop()