add option for variable mannings

sclaw · Jun 10, 2024 · 5ef941a · 5ef941a
1 parent db20d0d
commit 5ef941a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/muskingumcunge/reach.py b/src/muskingumcunge/reach.py
@@ -282,7 +282,10 @@ def generate_geometry(self, stages, top_widths, areas, perimeters):
         geom['wetted_perimeter'] = perimeters
         geom['wetted_perimeter'][geom['wetted_perimeter'] <= 0] = geom['wetted_perimeter'][geom['wetted_perimeter'] > 0].min()
         geom['hydraulic_radius'] = geom['area'] / geom['wetted_perimeter']
-        geom['mannings_n'] = np.repeat(self.mannings_n, len(stages))
+        if type(self.mannings_n) == np.ndarray:
+            geom['mannings_n'] = self.mannings_n
+        else:
+            geom['mannings_n'] = np.repeat(self.mannings_n, len(stages))
         geom['discharge'] = (1 / geom['mannings_n']) * geom['area'] * (geom['hydraulic_radius'] ** (2 / 3)) * (self.slope ** 0.5)
         geom['discharge'][geom['discharge'] <= 0] = geom['discharge'][geom['discharge'] > 0].min()
         self.clean_looped_rating_curve()