update files to generate OBS from IOP to include time units

E3SM-Project · Feb 7, 2025 · afd1747 · afd1747
1 parent 76a980d
commit afd1747
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 6 deletions.
diff --git a/IOP_diagnostics/OBS_to_Model_format/ARM97_to_DPxx.py b/IOP_diagnostics/OBS_to_Model_format/ARM97_to_DPxx.py
@@ -4,8 +4,8 @@
 
 # Define input and output file paths
 input_file = "/global/cfs/cdirs/e3sm/inputdata/atm/cam/scam/iop/ARM97_iopfile_4scam.nc"
-output_file = "/pscratch/sd/b/bogensch/dp_screamxx_conv/obs_data/OBS_ARM97.dpxx.nc"
-time_offset = 5.0
+output_file = "/pscratch/sd/b/bogensch/dp_screamxx_conv/obs_data/OBS_ARM97_full.dpxx.nc"
+time_offset = 0.0
 
 # Ensure the directory for the output file exists
 output_dir = os.path.dirname(output_file)
@@ -23,7 +23,7 @@
 
 # 1. Transfer and adjust the "time" variable
 time_data = ds_in["tsec"].values
-time_data = time_data/86400.
+time_data = (time_data-time_data[0])/86400.
 ds_out["time"] = xr.DataArray(time_data - time_offset, dims=["time"])
 
 p_data = ds_in["lev"].values/100.
@@ -66,6 +66,9 @@
 # Copy attributes from the input dataset to the output dataset
 ds_out.attrs = ds_in.attrs
 
+# Add the units attribute
+ds_out["time"].attrs["units"] = "days since 1997-06-18 23:29:45"
+
 # Save the new dataset to a NetCDF file
 ds_out.to_netcdf(output_file)
 

diff --git a/IOP_diagnostics/OBS_to_Model_format/DYNAMO_to_DPxx.py b/IOP_diagnostics/OBS_to_Model_format/DYNAMO_to_DPxx.py
@@ -5,7 +5,7 @@
 # Define input and output file paths
 input_file = "/global/cfs/cdirs/e3sm/inputdata/atm/cam/scam/iop/DYNAMO_amie_iopfile_4scam.nc"
 output_file = "/pscratch/sd/b/bogensch/dp_screamxx_conv/obs_data/OBS_DYNAMO_amie.dpxx.nc"
-time_offset = 13.0
+time_offset = 0.0
 
 # Ensure the directory for the output file exists
 output_dir = os.path.dirname(output_file)
@@ -23,7 +23,7 @@
 
 # 1. Transfer and adjust the "time" variable
 time_data = ds_in["tsec"].values
-time_data = time_data/86400.
+time_data = (time_data-time_data[0])/86400.
 ds_out["time"] = xr.DataArray(time_data - time_offset, dims=["time"])
 
 p_data = ds_in["lev"].values/100.
@@ -70,6 +70,9 @@
 # Copy attributes from the input dataset to the output dataset
 ds_out.attrs = ds_in.attrs
 
+# Add the units attribute
+ds_out["time"].attrs["units"] = "days since 2011-10-02 00:00:00"
+
 # Save the new dataset to a NetCDF file
 ds_out.to_netcdf(output_file)
 

diff --git a/IOP_diagnostics/OBS_to_Model_format/GATEIII_to_DPxx.py b/IOP_diagnostics/OBS_to_Model_format/GATEIII_to_DPxx.py
@@ -23,7 +23,7 @@
 
 # 1. Transfer and adjust the "time" variable
 time_data = ds_in["tsec"].values
-time_data = time_data/86400.
+time_data = (time_data-time_data[0])/86400.
 ds_out["time"] = xr.DataArray(time_data - time_offset, dims=["time"])
 
 p_data = ds_in["lev"].values/100.
@@ -65,6 +65,9 @@
 # Copy attributes from the input dataset to the output dataset
 ds_out.attrs = ds_in.attrs
 
+# Add the units attribute
+ds_out["time"].attrs["units"] = "days since 1974-08-30 00:00:00"
+
 # Save the new dataset to a NetCDF file
 ds_out.to_netcdf(output_file)