diff options
| -rw-r--r-- | src/simple.py | 7 |
1 files changed, 6 insertions, 1 deletions
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) |