diff options
Diffstat (limited to 'src/research.py')
| -rw-r--r-- | src/research.py | 8 |
1 files changed, 6 insertions, 2 deletions
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() |