From 9f6331b437f32ae9422a190aad27bb1fc92f9bfc Mon Sep 17 00:00:00 2001
From: eug-vs <eugene@eug-vs.xyz>
Date: Thu, 19 May 2022 04:09:01 +0400
Subject: feat: add more labels

---
 src/simple.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/simple.py b/src/simple.py
index 9a3aa31..524e6fb 100644
--- a/src/simple.py
+++ b/src/simple.py
@@ -183,6 +183,10 @@ class SIMPLE:
             self.plt.remove()
 
         self.patch = axes.add_patch(Rectangle((0, 0), *reversed(self.bfs_size)))
+        axes.set_title('Velocity field (normalized)')
+        plt.suptitle(f'Avg mass source per grid point = {self.avg_error()}')
+        plt.xlabel('X')
+        plt.ylabel('Y')
 
         u, v = self.u, self.v
         if normalize:
@@ -199,5 +203,6 @@ class SIMPLE:
             scale=30,
             cmap='inferno'
         )
-        self.colorbar = plt.colorbar()
+        self.colorbar = plt.colorbar(label='Pressure')
+
         plt.pause(0.0001)
-- 
cgit v1.2.3