add bucket initial condition

CliMA · Dec 16, 2024 · acf8b9b · acf8b9b
1 parent a1c1dc7
commit acf8b9b
Showing 1 changed file with 43 additions and 0 deletions.
diff --git a/experiments/ClimaEarth/components/land/climaland_bucket.jl b/experiments/ClimaEarth/components/land/climaland_bucket.jl
@@ -138,6 +138,49 @@ function bucket_init(
     Y.bucket.Ws .= 0.0
     Y.bucket.σS .= 0.0
 
+    # overwrite initial conditions
+    file_path = "/Users/zshen/Work/clima_repo/artifacts/bucket_ic_august.nc"
+    surface_space = domain.space.surface
+    subsurface_space = domain.space.subsurface
+    regridder_type = :InterpolationsRegridder
+    extrapolation_bc = (
+        Interpolations.Periodic(),
+        Interpolations.Flat(),
+        Interpolations.Flat(),
+    )
+    W = SpaceVaryingInput(
+        file_path,
+        "W",
+        surface_space;
+        regridder_type,
+        regridder_kwargs = (; extrapolation_bc,),
+    )
+    Ws = SpaceVaryingInput(
+        file_path,
+        "Ws",
+        surface_space;
+        regridder_type,
+        regridder_kwargs = (; extrapolation_bc,),
+    )
+    T = SpaceVaryingInput(
+        file_path,
+        "T",
+        subsurface_space;
+        regridder_type,
+        regridder_kwargs = (; extrapolation_bc,),
+    )
+    σS = SpaceVaryingInput(
+        file_path,
+        "S",
+        surface_space;
+        regridder_type,
+        regridder_kwargs = (; extrapolation_bc,),
+    )
+    Y.bucket.W .= W
+    Y.bucket.σS .= σS
+    Y.bucket.T .= T
+    Y.bucket.Ws .= Ws
+
     # Set initial aux variable values
     set_initial_cache! = CL.make_set_initial_cache(model)
     set_initial_cache!(p, Y, tspan[1])