|
- import pandas as pd
- import plotly.graph_objs as go
-
- call_depths = [0]
- datas = []
-
- def ensure_big_enough(list, idx, el):
- l = len(list)
- if (idx >= l):
- list.extend([el]*(idx-l+1))
-
- def incr_call_depth(thread):
- ensure_big_enough(call_depths, thread, 0)
- call_depths[thread] += 1
-
- def append_to_datas(thread, data):
- ensure_big_enough(datas, thread, [])
- datas[thread].append(data)
-
- with open("../bin/profiler.report", "r") as file:
- for line in file:
- infos = line.split()
- thread = int(infos[0])
- if infos[1] == "->":
- incr_call_depth(thread)
- append_to_datas(thread, {
- "task": f"[T{thread}] - {'{:03d}'.format(call_depths[thread])}",
- "start": int(infos[5]),
- "end": None
- })
- elif infos[1] == "<-":
- call_depths[thread] -= 1
- for set in datas[thread][::-1]:
- if set["end"] == None:
- set["end"] = int(infos[2])
- break
- else:
- print("couldn't find task that finished")
- break
-
- else:
- print("neither -> nor <-")
-
-
- fig = go.Figure(
- layout = {
- 'barmode': 'stack',
- "dragmode" : "pan",
- }
- )
-
- flat_datas= [item for sublist in datas for item in sublist]
- # i = 1
- # df = flat_datas[:20]
- # for call in df:
- # name = call["task"]
- # start = call["start"]
- # end = call["end"]
- # duration = end,start
- # fig.add_bar(x=(i,duration),
- # y=(i,name),
- # base=(i,start),
- # orientation='h',
- # showlegend=False,
- # name=name,
- # hovertext=f"yes",
- # text=""
- # )
- # i += 1
-
- df = pd.DataFrame(flat_datas[:20])
- df['duration'] = df['end'] - df['start']
-
- for stackdepth, stackdepth_df in df.groupby('task'):
- params = {"x": stackdepth_df.duration,
- "y": stackdepth_df.task,
- "name": stackdepth,
- "base": stackdepth_df.start
- }
- print(params)
- fig.add_bar(**params,
- orientation='h',
- showlegend=False,
- hovertext=f"yes",
- text=""
- )
-
-
- fig.show(config={'scrollZoom': True})
|
