""" Example call stack plot like a flame graph.

Idea from: https://www.speedscope.app/

"""

from .tsdb.btree import Btree

class CallStackBar:

def __init__(self):

self._tree = BTree()

self._levels = []

def enter_function(self, t, func):

self._levels.append(func)

event = (t, "O", func)

self.append_event(event)

def leave_function(self, t, func):

top = self._levels.pop()

assert top == func

event = (t, "C", func)

self.append_event(event)

def append_event(self, event):

# self._tree.append(event)

pass

class FunctionCall:

def __init__(self):

self.t_enter = None

self.t_leave = None

self.child_calls = []