Add flag for return value

Author: sarangbishal

examples/fibonacci.png

@@ -17,7 +17,7 @@ def fib(n, node_num):
     return left + right
 
 # Call function
-fib(n=6, node_num=0)
+print(fib(n=6, node_num=0))
 
 # Save recursion tree to a file
-vs.write_image("fibonacci")
+vs.write_image("fibonacci.png")

examples/fibonacci.py

@@ -16,7 +16,8 @@
         [7, 3]
 """
 
-@vs(ignore_args=['node_num'], show_argument_name=False)
+
+@vs(ignore_args=['node_num'], show_argument_name=False, show_return_value=False)
 def f(sum, ans, node_num):
     # If sum becoms 0 we have found the required list
     if sum == 0:

examples/make_sum.py

@@ -0,0 +1,49 @@
+from  visualiser import  Visualiser as vs
+start_state = (3, 3, 1)
+goal_state = (0, 0, 0)
+
+options = [(2, 0), (1, 1), (0, 2), (1, 0), (0, 1)]
+visited = dict()
+
+def is_valid(m, c):
+    return m >= 0 and m <= 3 and c >= 0 and c <= 3
+
+
+@vs(ignore_args=["node_num", "level"])
+def dfs(m, c, s, level, node_num):
+    if (m, c, s) == goal_state:
+        return True
+
+    if m > 0 and c > m:
+        return False
+
+    right_side_m = 3 - m
+    right_side_c = 3 - c
+    if right_side_m > 0 and right_side_c > right_side_m:
+        return False
+
+    visited[(m, c, s)] = True
+
+    if s == 1:
+        op = -1
+    else:
+        op = 1
+
+    solved = False
+    for i in range(5):
+        next_m, next_c, next_side = m + op * options[i][0], c + op * options[i][1], int(not s)
+
+        if is_valid(next_m, next_c):
+
+            if (next_m, next_c, next_side) not in visited:
+                vs.node_count += 1
+                solved = (solved or dfs(m=next_m, c=next_c, s=next_side, level=level + 1, node_num=vs.node_count))
+
+                if solved:
+                    return True
+    return solved
+
+
+if (dfs(m=3, c=3, s=1, level=0, node_num=0)):
+    print("SOlution Found")
+    vs.write_image("missionaries.png")

examples/missionaries.png

@@ -3,11 +3,14 @@
 from collections import OrderedDict
 import pydot
 
+
 class Visualiser(object):
     node_count = 0
     graph = pydot.Dot(graph_type="digraph")
-    def __init__(self, ignore_args, show_argument_name=True):
+
+    def __init__(self, ignore_args, show_argument_name=True, show_return_value=True):
         self.show_argument_name = show_argument_name
+        self.show_return_value = show_return_value
         self.ignore_args = ignore_args
 
     @classmethod
@@ -85,7 +88,8 @@ def wrapper(*args, **kwargs):
             result = fn(*args, **kwargs)
 
             # #Child Node
-            child_label = current_function_label
+
+            child_label = current_function_label + f" => {result}"
             child_name = current_function_signature
             v = pydot.Node(name=child_name, label=child_label)
             self.graph.add_node(v)