From e031b6708ce491976e28d76e5ea81c3ea0c92bed Mon Sep 17 00:00:00 2001
From: eug-vs <eugene@eug-vs.xyz>
Date: Thu, 19 May 2022 04:57:52 +0400
Subject: feat: provide default density when plotting

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

diff --git a/src/simple.py b/src/simple.py
index 524e6fb..5ffe690 100644
--- a/src/simple.py
+++ b/src/simple.py
@@ -174,7 +174,7 @@ class SIMPLE:
     def avg_error(self):
         return np.absolute(self.b).sum()
 
-    def plot(self, normalize=False, density=1):
+    def plot(self, normalize=False, density=False):
         if self.patch:
             self.patch.remove()
         if self.colorbar:
@@ -194,6 +194,8 @@ class SIMPLE:
             u = u / factor
             v = v / factor
 
+        density = density or int((max(self.domain_size) / self.step) / 40)
+
         self.plt = plt.quiver(
             self.x[::density, ::density],
             self.y[::density, ::density],
-- 
cgit v1.2.3