diff --git a/minihawk/minihawk_scenario4.py b/minihawk/minihawk_scenario4.py index 9cb6cf54..d999c4dc 100644 --- a/minihawk/minihawk_scenario4.py +++ b/minihawk/minihawk_scenario4.py @@ -95,6 +95,8 @@ class AgentMode(Enum): # params={'bloating_method':'GLOBAL'} ) + traces.dump('sc1.json') + # fig = go.Figure() # fig = reachtube_tree(traces, None, fig, 0, 1) # fig.show() @@ -287,7 +289,7 @@ class AgentMode(Enum): plt.xticks(fontsize=14) plt.yticks(fontsize=14) plt.tight_layout() - plt.savefig('sc4_x.png', bbox_inches='tight', pad_inches=0.0, dpi=300) + # plt.savefig('sc4_x.png', bbox_inches='tight', pad_inches=0.0, dpi=300) plt.figure(1,figsize=(2,2)) plt.xlabel('t',fontsize=14) plt.ylabel('y',fontsize=14) @@ -295,7 +297,7 @@ class AgentMode(Enum): plt.yticks(fontsize=14) plt.legend(prop={'size': 14},loc='upper right') plt.tight_layout() - plt.savefig('sc4_y.png', bbox_inches='tight', pad_inches=0.0, dpi=300) + # plt.savefig('sc4_y.png', bbox_inches='tight', pad_inches=0.0, dpi=300) plt.figure(2,figsize=(2,2)) plt.xlabel('t',fontsize=14) plt.ylabel('z',fontsize=14) @@ -303,8 +305,8 @@ class AgentMode(Enum): plt.yticks(fontsize=14) plt.legend(prop={'size': 14},loc='upper right') plt.tight_layout() - plt.savefig('sc4_z.png', bbox_inches='tight', pad_inches=0.0, dpi=300) - # plt.show() + # plt.savefig('sc4_z.png', bbox_inches='tight', pad_inches=0.0, dpi=300) + plt.show() # if idx_start < pos.shape[0]: # color = 'red' if prev_intersect else 'blue' diff --git a/minihawk/viewer.py b/minihawk/viewer.py new file mode 100644 index 00000000..3bf8c10b --- /dev/null +++ b/minihawk/viewer.py @@ -0,0 +1,9 @@ +from verse.analysis import AnalysisTree +from verse.plotter.plotter2D import * +import plotly.graph_objects as go + +res = AnalysisTree.load('sc1.json') + +fig = go.Figure() +fig = reachtube_tree(res, None, fig, 0, 1) +fig.show() \ No newline at end of file