Skip to content

SwayamInSync/recursion-trace

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

24 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Recursion Trace

Overview

recursion_trace is a Python package that provides a decorator to trace recursive function calls. It generates a visual recursion tree using Graphviz, making it easier to understand and debug recursive algorithms.

Features

  • Trace single and mutual recursion.
  • Generate a Graphviz plot to visualize the recursion tree.
  • Capture function arguments, return values, and recursion depth.
  • Render Animation demonstrating construction of recursion-tree

Installation

  • Download the graphviz for the respective system (Windows/Mac/Linux)
pip install recursion-trace

Usage

  • Decorate the in-usage recursive functions with the decorator @trace_recursion

Example: Merge Sort with Animation

from recursion_trace import trace_recursion, show_recursion_tree

@trace_recursion  # use the decorator to trace the recursion stack
def merge_sort(arr):
    if len(arr) <= 1:
        return arr
    mid = len(arr) // 2
    left_half = arr[:mid]
    right_half = arr[mid:]
    L = merge_sort(left_half)
    R = merge_sort(right_half)
    return merge(L, R)

def merge(left, right):
    result = []
    i = j = 0
    while i < len(left) and j < len(right):
        if left[i] < right[j]:
            result.append(left[i])
            i += 1
        else:
            result.append(right[j])
            j += 1
    result.extend(left[i:])
    result.extend(right[j:])
    return result

if __name__ == '__main__':
    arr = [3, 1, 4, 1, 5, 9, 2, 6, 5]
    sorted_arr = merge_sort(arr)
    show_recursion_tree(make_animation=True)  # display recursion tree
    # setting make_animation=True also renders an animation of recursion-tree construction

Output:

Screenshot 2023-09-18 at 10 42 54 AM
recursion_tree.mp4

Example: Mutual Recursive Functions

from recursion_trace import trace_recursion, show_recursion_tree

@trace_recursion  # use the decorator to trace the recursion stack
def is_even(n):
    if n == 0:
        return True
    return is_odd(n - 1)

@trace_recursion  # use the decorator to trace the recursion stack
def is_odd(n):
    if n == 0:
        return False
    return is_even(n - 1)

if __name__ == '__main__':
    is_even(4)
    show_recursion_tree()

Output:

Screenshot 2023-09-18 at 11 55 59 AM

Author

Swayam Singh

About

A package to trace recursive function calls and generate/visualize the recursion tree

Resources

Stars

Watchers

Forks

Packages

 
 
 

Contributors

Languages