create inverse functions for brooks, garnder and fredlund

martinvonk · Oct 4, 2023 · e6cdfd0 · e6cdfd0
1 parent 7796b5d
commit e6cdfd0
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/src/pedon/soilmodel.py b/src/pedon/soilmodel.py
@@ -129,7 +129,12 @@ def k(self, h: FloatArray, s: FloatArray | None = None) -> FloatArray:
         return self.k_s * self.k_r(h=h, s=s)
 
     def h(self, theta: FloatArray) -> FloatArray:
-        pass
+        h = full(theta.shape, self.h_b)
+        mask = theta >= self.theta_r
+        h[mask] = self.h_b * ((theta[mask] - self.theta_r) / (self.s(theta[mask]))) ** (
+            -1 / self.l
+        )
+        return h
 
     def plot(self, ax: plt.Axes | None = None) -> plt.Axes:
         return plot_swrc(self, ax=ax)
@@ -165,7 +170,7 @@ def k(self, h: FloatArray, s: FloatArray | None = None) -> FloatArray:
         return self.k_s * self.k_r(h=h, s=s)
 
     def h(self, theta: FloatArray) -> FloatArray:
-        pass
+        return (abs(theta) / self.a) ** (-1 / self.b)
 
     def plot(self, ax: plt.Axes | None = None) -> plt.Axes:
         return plot_swrc(self, ax=ax)
@@ -284,7 +289,9 @@ def k(self, h: FloatArray, s: FloatArray | None = None) -> FloatArray:
         return self.k_s * self.k_r(h=h, s=s)
 
     def h(self, theta: FloatArray) -> FloatArray:
-        pass
+        return self.a * (exp ** (self.theta_s / theta) ** (1 / self.m) - exp(1)) ** (
+            1 / self.n
+        )
 
     def plot(self, ax: plt.Axes | None = None) -> plt.Axes:
         return plot_swrc(self, ax=ax)