Recursion Tree Plotter

A python decorator to generate a visual tree for recursive functions.

Example

Let's say you have a recursive function for finding n-th element in Fibonacci sequence.

def fib(n):
    if n <= 1:
        return n
    return fib(n - 1) + fib(n - 2)

In order to plot a recursion tree for an execution of above function (say fib(5)), we put @plot_recursion_tree decorator over it and introduce **kwargs as one of the arguments.

@plot_recursion_tree
def fib(n, **kwargs):
    if n <= 1:
        return n
    return fib(n - 1, **kwargs) + fib(n - 2, **kwargs)

And boom!

In the tree, node label is constructed as <comma-separated args> [<counter>] where counter specifies the order of execution.

Check out examples folder for more examples!