tmp: approved flow

1 files changed, 20 insertions, 5 deletions

diff --git a/src/simple.py b/src/simple.py
index 789c81a..91ed1a4 100644
--- a/src/simple.py
+++ b/src/simple.py
@@ -41,6 +41,10 @@ class SIMPLE:
             self.u_star[i][0] = 1
             self.v_star[i][0] = 0
 
+    def apply_outflow_boundary(self):
+        for i in range(0, self.shape[0]):
+            self.u_star[i][-1] = self.u_star[i][-2];
+
     def assert_positive(self, value):
         '''Assert that the value is nearly positive'''
         assert value > -0.1, f'WARNING: Value must be positive: {value}'
@@ -50,8 +54,8 @@ class SIMPLE:
         '''Iterator over all grid points, excluding the obstacle'''
         for i in range(1, self.shape[0] - 1):
             for j in range(1, self.shape[1] - 1):
-                if i > self.bfs_node_size[0] or j > self.bfs_node_size[1]:
-                    yield (i, j)
+                # if i > self.bfs_node_size[0] or j > self.bfs_node_size[1]:
+                yield (i, j)
 
     def solve_momentum_equations(self):
         # Momentum along X direction
@@ -104,8 +108,9 @@ class SIMPLE:
 
     def apply_sides_boundary(self):
         for j in range(self.shape[1]):
-            self.v_star[1][j] = 0
+            self.v_star[0][j] = 0
             self.v_star[-2][j] = 0
+            # WORKSNICE: self.v_star[-3][j] = 0
 
     def apply_bfs_boundary(self):
         '''Apply Backwards Facing Step boundary conditions'''
@@ -145,15 +150,25 @@ class SIMPLE:
             self.v[i][j] = self.v_star[i][j] + self.d_n[i][j] * (self.p_prime[i][j] - self.p_prime[i + 1][j])
 
     def iterate(self):
+        self.apply_inflow_boundary()
         self.solve_momentum_equations()
 
-        self.apply_inflow_boundary()
         self.apply_sides_boundary()
-        self.apply_bfs_boundary()
+        # self.apply_bfs_boundary()
 
         self.correct_pressure()
         self.correct_velocities()
 
+        # Sides (magick wtf?)
+        for j in range(self.shape[1]):
+            self.v[1][j] = 0
+            self.v[-2][j] = 0
+
+        for i in range(self.shape[0]):
+            self.v[i][0] = 0
+            self.v[i][1] = 0
+
+
     def avg_error(self):
         return np.absolute(self.b).sum()