From 7a948fe1d90e858902766dd49143f9ec46188bec Mon Sep 17 00:00:00 2001 From: eug-vs Date: Wed, 25 May 2022 23:23:44 +0400 Subject: feat: finish working bfs example --- src/research.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) (limited to 'src/research.py') diff --git a/src/research.py b/src/research.py index 2a59c49..8c46877 100644 --- a/src/research.py +++ b/src/research.py @@ -34,7 +34,7 @@ class Research: if iteration % 10 == 0 or iteration == 1: if preview or save_plot: - self.plotter.plot(self.model, normalize=False, density=1) + self.plotter.plot(self.model, normalize=True) if preview: self.plotter.show() if save_plot: @@ -45,10 +45,14 @@ class Research: self.model.save(self.model_path) self.model.save(self.solution_path) + + self.plotter.plot(self.model, streamplot=True) + self.plotter.save(os.path.join(self.path, f'streamplot.png')) + self.inspect() def inspect(self): - self.plotter.plot(self.model, density=1) + self.plotter.plot(self.model, streamplot=True) while True: self.plotter.show() -- cgit v1.2.3