Skip to content

Commit

Permalink
completed: good example file!
Browse files Browse the repository at this point in the history
  • Loading branch information
JanJereczek committed Jun 26, 2024
1 parent fc391e3 commit b9712eb
Showing 1 changed file with 59 additions and 10 deletions.
69 changes: 59 additions & 10 deletions scripts/hazzard2024.jl
Original file line number Diff line number Diff line change
@@ -1,17 +1,66 @@
include("intro.jl")

# Init regridding
dimnames = ("lon", "lat")
extrapolation_boundary_conditions = (Periodic(), Flat())
varnames = ("z")
source_file = datadir("BCs/Hazzard-Richards-2024/HR24_GHF_mean.nc")
# Define regridding config
source_dimnames = ("lon", "lat")
target_dimnames = ("x", "y")
varnames = ("z",)
filename = "BCs/Hazzard-Richards-2024/HR24_GHF"
source_gridname = "EPSG:4326"
target_gridname = "+proj=stere +lat_0=-90 +lat_ts=-80"
r = Regrid(source_file, dimnames, extrapolation_boundary_conditions,
varnames, source_gridname, target_gridname)
extrapolation_boundary_conditions = (Periodic(), Flat())

# Define mean GHF regridding
regrid_μ_ghf = Regrid(
datadir("$(filename)_mean.nc"),
source_dimnames,
target_dimnames,
source_gridname,
target_gridname,
varnames,
extrapolation_boundary_conditions,
)

# Define stddev GHF regridding
regrid_σ_ghf = Regrid(
datadir("$(filename)_std.nc"),
source_dimnames,
target_dimnames,
source_gridname,
target_gridname,
varnames,
extrapolation_boundary_conditions,
)

# Regrid
x = range(-3040f3, stop = 3040f3, step = 32f3)
X, Y = ndgrid(x, copy(x))
regridded_vars = r((X, Y))
heatmap(regridded_vars[1])
y = copy(x)
target_dims = (x, y)
target_grid = ndgrid(target_dims...)

# Filter out missing values and pack into a tuple.
target_μ = missings2nans(regrid_μ_ghf(target_grid)[1])
target_σ = missings2nans(regrid_σ_ghf(target_grid)[1])
varnames = ("mu_ghf", "sigma_ghf")
vars = (target_μ, target_σ)

# Visualize regridded data for sanity check
fig = Figure(size = (800, 400))
for j in eachindex(varnames)
ax = Axis(fig[1, j], aspect = DataAspect())
hidedecorations!(ax)
heatmap!(ax, vars[j])
end
fig

# Save regridded data
x_atts = Dict("units" => "m", "long_name" => "x-coordinate")
y_atts = Dict("units" => "m", "long_name" => "y-coordinate")
dim_atts = (x_atts, y_atts)

μ_ghf_atts = Dict("units" => "mW/m^2", "long_name" => "Mean geothermal heat flux")
σ_ghf_atts = Dict("units" => "mW/m^2", "long_name" =>
"Standard deviation of geothermal heat flux")
var_atts = (μ_ghf_atts, σ_ghf_atts)

fn = datadir("$filename-regridded.nc")
save2nc(fn, target_dimnames, target_dims, dim_atts, varnames, vars, var_atts)

0 comments on commit b9712eb

Please sign in to comment.