fix errors due to change in dropsonde data and add readme

Author: JakobDeutloff

scripts/arts_calibration/README.md

@@ -0,0 +1,14 @@
+# ARTS for HAMP
+
+This README gives a quick introduction on how to use ARTS for calculating brightness temperatures at the HAMP radiometer frequencies from dropsonde data.
+
+### Getting Started:
+1. Create a conda environment with
+```conda env create -f environment.yaml```
+2. Check whether you can run ```arts_bt_calculation.py```
+3. You can run the analysis script ```arts_bt_calculation.py``` and loop over all flights within the script, or you can submit it as a batch job for each flight using ```master_submitter.py```. Then you will need to delete the loop over flights in ```arts_bt_calculation.py```.
+
+### Code Structure
+```arts_bt_calculation.py``` is the main script to run the brightness temperature calculations. It reads in dropsonde data, sets up the ARTS simulation, and outputs brightness temperatures. You need to run this script from the root directory of the repository.
+
+```analyse_bt_diffs.py``` is a script to analyse the differences between measured and simulated brightness temperatures. It reads in the output from ```arts_bt_calculation.py``` and compares it with HAMP measurements.

scripts/arts_calibration/arts_bt_calculation.py

@@ -69,14 +69,17 @@ def calc_arts_bts(date, flightletter, ds_dropsonde, cfg):
         f"{cfg['save_dir']}/radiometer/{flightname}_radio.zarr", engine="zarr"
     )
     ds_dropsonde = ds_dropsonde.where(
-        (ds_dropsonde["interp_time"] > pd.to_datetime(date))
-        & (ds_dropsonde["interp_time"] < pd.to_datetime(date) + pd.DateOffset(hour=23))
+        (ds_dropsonde["interpolated_time"] > pd.to_datetime(date))
+        & (
+            ds_dropsonde["interpolated_time"]
+            < pd.to_datetime(date) + pd.DateOffset(hour=23)
+        )
     ).dropna(dim="sonde", how="all")
 
     print("Calculate Cloud Mask")
     ds_dropsonde = ds_dropsonde.assign(
         radar_cloud_flag=(
-            ds_radar.sel(time=ds_dropsonde.sonde_time, method="nearest").sel(
+            ds_radar.sel(time=ds_dropsonde.launch_time, method="nearest").sel(
                 height=slice(200, None)
             )["dBZe"]
             > -30

src/arts_functions.py

@@ -690,7 +690,7 @@ def fit_constant(x, y):
 
 def get_profiles(sonde, ds_dropsonde, radiometers):
     ds_dropsonde_loc = ds_dropsonde.sel(sonde=sonde)
-    drop_time = ds_dropsonde_loc["sonde_time"].values
+    drop_time = ds_dropsonde_loc["launch_time"].values
     hampdata_loc = radiometers.dropna("time").sel(time=drop_time, method="nearest")
     height = float(hampdata_loc.plane_altitude.values)
     return ds_dropsonde_loc, hampdata_loc, height, drop_time