Skip to content

Commit b82717b

Browse files
committed
Add NPD domain Icechunk stores to OceanDataCatalog NOC STAC.
1 parent 70a19b0 commit b82717b

1 file changed

Lines changed: 15 additions & 6 deletions

File tree

catalogs/create_noc_stac.py

Lines changed: 15 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,9 @@ def create_item_with_asset(ds: xr.Dataset,
4949
A STAC Item containing the dataset information and an asset pointing to the data.
5050
"""
5151
# Define the item description based on the prefix:
52-
if 'I' in prefix:
52+
if 'domain' in prefix:
53+
description = f"Icechunk repository containin {config} global ocean model domain variables."
54+
elif 'I' in prefix:
5355
description = f"Icechunk repository containing {config} global sea-ice {operation} outputs defined at T-points."
5456
elif 'S' in prefix:
5557
description = f"Icechunk repository containing {config} global ocean scalar {operation} outputs."
@@ -202,7 +204,7 @@ def create_item_with_asset(ds: xr.Dataset,
202204

203205
# -- Add Items to NOC Near-Present Day eORCA1 ERA5v1 Sub-Catalog -- #
204206
# Define the store credentials for the eORCA1 ERA5v1 NPD data:
205-
for prefix in ["T1y", "U1y", "V1y", "W1y", "I1y", "S1y", "T1m", "U1m", "V1m", "W1m", "I1m", "S1m"]:
207+
for prefix in ["T1y", "U1y", "V1y", "W1y", "I1y", "S1y", "T1m", "U1m", "V1m", "W1m", "I1m", "S1m", "domain"]:
206208
# Define S3 storage to read eORCA1 ERA5v1 NPD data:
207209
storage = icechunk.s3_storage(
208210
bucket="npd-eorca1-era5v1",
@@ -217,7 +219,9 @@ def create_item_with_asset(ds: xr.Dataset,
217219
ds = xr.open_zarr(repo.readonly_session(branch="main").store, consolidated=False)
218220

219221
# Create item with asset for each eORCA1 ERA5v1 NPD prefix:
220-
if '1y' in prefix:
222+
if 'domain' in prefix:
223+
operation = "None None"
224+
elif '1y' in prefix:
221225
operation = "annual mean"
222226
elif '1m' in prefix:
223227
operation = "monthly mean"
@@ -239,7 +243,7 @@ def create_item_with_asset(ds: xr.Dataset,
239243
# Define the store credentials for the eORCA025 ERA5v1 NPD data:
240244
for prefix in ["T1y_3d", "T1y_4d", "U1y_3d", "U1y_4d", "V1y_3d", "V1y_4d", "W1y_4d", "I1y_3d", "S1y_1d",
241245
"T1m_3d", "T1m_4d", "U1m_3d", "U1m_4d", "V1m_3d", "V1m_4d", "W1m_4d", "I1m_3d", "S1m_1d",
242-
"T5d_3d", "T5d_4d", "U5d_3d", "U5d_4d", "V5d_3d", "V5d_4d", "I5d_3d", "S5d_1d",
246+
"T5d_3d", "T5d_4d", "U5d_3d", "U5d_4d", "V5d_3d", "V5d_4d", "I5d_3d", "S5d_1d", "domain"
243247
]:
244248
# Define S3 storage to read eORCA025 ERA5v1 NPD data:
245249
storage = icechunk.s3_storage(
@@ -255,7 +259,9 @@ def create_item_with_asset(ds: xr.Dataset,
255259
ds = xr.open_zarr(repo.readonly_session(branch="main").store, consolidated=False)
256260

257261
# Create item with asset for each eORCA025 ERA5v1 NPD prefix:
258-
if '1y' in prefix:
262+
if 'domain' in prefix:
263+
operation = "None None"
264+
elif '1y' in prefix:
259265
operation = "annual mean"
260266
elif '1m' in prefix:
261267
operation = "monthly mean"
@@ -277,6 +283,7 @@ def create_item_with_asset(ds: xr.Dataset,
277283
# Define the store credentials for the eORCA12 ERA5v1 NPD data:
278284
for prefix in ["T1y_3d", "T1y_4d", "U1y_3d", "U1y_4d", "V1y_3d", "V1y_4d", "W1y_4d", "I1y_3d", "S1y_1d",
279285
"T1m_3d", "T1m_4d", "U1m_3d", "U1m_4d", "V1m_3d", "V1m_4d", "W1m_4d", "I1m_3d", "S1m_1d",
286+
"domain",
280287
]:
281288
# Define S3 storage to read eORCA12 ERA5v1 NPD data:
282289
storage = icechunk.s3_storage(
@@ -292,7 +299,9 @@ def create_item_with_asset(ds: xr.Dataset,
292299
ds = xr.open_zarr(repo.readonly_session(branch="main").store, consolidated=False)
293300

294301
# Create item with asset for each eORCA12 ERA5v1 NPD prefix:
295-
if '1y' in prefix:
302+
if 'domain' in prefix:
303+
operation = "None None"
304+
elif '1y' in prefix:
296305
operation = "annual mean"
297306
elif '1m' in prefix:
298307
operation = "monthly mean"

0 commit comments

Comments
 (0)