From 7a948fe1d90e858902766dd49143f9ec46188bec Mon Sep 17 00:00:00 2001
From: eug-vs <eugene@eug-vs.xyz>
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