feat(wind): Bias correct wind speeds based on scaling to a known average

atlite/convert.py

@@ -565,14 +565,19 @@ def convert_wind(
     ds: xr.Dataset,
     turbine: TurbineConfig,
     interpolation_method: Literal["logarithmic", "power"],
+    windspeed_bias_correction: bool | xr.DataArray | None = None,
 ) -> xr.DataArray:
     """
     Convert wind speeds for turbine to wind energy generation.
     """
     V, POW, hub_height, P = itemgetter("V", "POW", "hub_height", "P")(turbine)
 
+    ds, from_height = windm.apply_windspeed_bias_correction(
+        ds, windspeed_bias_correction
+    )
+
     wnd_hub = windm.extrapolate_wind_speed(
-        ds, to_height=hub_height, method=interpolation_method
+        ds, to_height=hub_height, method=interpolation_method, from_height=from_height
     )
 
     def apply_power_curve(da):
@@ -598,6 +603,7 @@ def wind(
     smooth: bool | dict = False,
     add_cutout_windspeed: bool = False,
     interpolation_method: Literal["logarithmic", "power"] = "logarithmic",
+    windspeed_bias_correction: bool | xr.DataArray | None = None,
     **params,
 ) -> xr.DataArray:
     """
@@ -609,17 +615,20 @@ def wind(
     Parameters
     ----------
     turbine : str or dict
-        A turbineconfig dictionary with the keys 'hub_height' for the
-        hub height and 'V', 'POW' defining the power curve.
-        Alternatively a str refering to a local or remote turbine configuration
-        as accepted by atlite.resource.get_windturbineconfig(). Locally stored turbine
-        configurations can also be modified with this function. E.g. to setup a different hub
-        height from the one used in the yaml file,one would write
-                "turbine=get_windturbineconfig(“NREL_ReferenceTurbine_5MW_offshore”)|{“hub_height”:120}"
+        A turbineconfig dictionary with the keys 'hub_height' for the hub height
+        and 'V', 'POW' defining the power curve. Alternatively a str refering to
+        a local or remote turbine configuration as accepted by
+        atlite.resource.get_windturbineconfig(). Locally stored turbine
+        configurations can also be modified with this function. E.g. to setup a
+        different hub height from the one used in the yaml file, one would write
+        >>> turbine = (
+        >>>     get_windturbineconfig("NREL_ReferenceTurbine_5MW_offshore")
+        >>>     | {"hub_height":120}
+        >>> )
     smooth : bool or dict
-        If True smooth power curve with a gaussian kernel as
-        determined for the Danish wind fleet to Delta_v = 1.27 and
-        sigma = 2.29. A dict allows to tune these values.
+        If True smooth power curve with a gaussian kernel as determined for the
+        Danish wind fleet to Delta_v = 1.27 and sigma = 2.29. A dict allows to
+        tune these values.
     add_cutout_windspeed : bool
         If True and in case the power curve does not end with a zero, will add zero power
         output at the highest wind speed in the power curve. If False, a warning will be
@@ -628,6 +637,11 @@ def wind(
     interpolation_method : {"logarithmic", "power"}
         Law to interpolate wind speed to turbine hub height. Refer to
         :py:func:`atlite.wind.extrapolate_wind_speed`.
+    windspeed_bias_correction : DataArray, optional
+        Correction factor that is applied to the windspeed at
+        `windspeed_height`. Such a correction factor can be calculated using
+        :py:func:`atlite.wind.calculate_windspeed_bias_correction` with a raster
+        dataset of mean wind speeds.
 
     Note
     ----
@@ -645,10 +659,15 @@ def wind(
     if smooth:
         turbine = windturbine_smooth(turbine, params=smooth)
 
+    if isinstance(windspeed_bias_correction, xr.DataArray):
+        # Front-load coordinate alignment cost
+        windspeed_bias_correction = cutout.data.reindex_like(windspeed_bias_correction)
+
     return cutout.convert_and_aggregate(
         convert_func=convert_wind,
         turbine=turbine,
         interpolation_method=interpolation_method,
+        windspeed_bias_correction=windspeed_bias_correction,
         **params,
     )
 

atlite/cutout.py

@@ -134,6 +134,11 @@ def __init__(self, path, **cutoutparams):
         gebco_path: str
             Path to find the gebco NetCDF file. Only necessary when including
             the gebco module.
+        windspeed_real_average_path: str
+            Path to raster dataset with wind speeds to bias correct average wind
+            speeds
+        windspeed_height: int, defaults to 100
+            Height in meters at which average windspeeds are provided
         parallel : bool, default False
             Whether to open dataset in parallel mode. Take effect for all
             xr.open_mfdataset usages.

atlite/data.py

@@ -31,14 +31,15 @@ def get_features(
     tmpdir=None,
     monthly_requests=False,
     concurrent_requests=False,
+    **parameter_updates,
 ):
     """
     Load the feature data for a given module.
 
     This get the data for a set of features from a module. All modules
     in `atlite.datasets` are allowed.
     """
-    parameters = cutout.data.attrs
+    parameters = cutout.data.attrs | parameter_updates
     lock = SerializableLock()
     datasets = []
     get_data = datamodules[module].get_data
@@ -58,7 +59,7 @@ def get_features(
 
     datasets = compute(*datasets)
 
-    ds = xr.merge(datasets, compat="equals")
+    ds = xr.merge([da for da in datasets if da is not None], compat="equals")
     for v in ds:
         da = ds[v]
         da.attrs["module"] = module
@@ -140,6 +141,7 @@ def cutout_prepare(
     dask_kwargs=None,
     monthly_requests=False,
     concurrent_requests=False,
+    **parameter_updates,
 ):
     """
     Prepare all or a selection of features in a cutout.
@@ -185,6 +187,8 @@ def cutout_prepare(
     concurrent_requests : bool, optional
         If True, the monthly data requests are posted concurrently.
         Only has an effect if `monthly_requests` is True. The default is False.
+    **parameter_updates
+        Updates of creation parameters
 
     Returns
     -------
@@ -224,19 +228,27 @@ def cutout_prepare(
             data_format=data_format,
             monthly_requests=monthly_requests,
             concurrent_requests=concurrent_requests,
+            **parameter_updates,
+        )
+        new_features = set(da.attrs["feature"] for da in ds.data_vars.values())
+        if not new_features:
+            logger.warning("No new features prepared")
+            return
+        attrs = non_bool_dict(
+            cutout.data.attrs
+            | ds.attrs
+            | dict(prepared_features=list(prepared | new_features))
+            | parameter_updates
         )
-        prepared |= set(missing_features)
-
-        cutout.data.attrs.update(dict(prepared_features=list(prepared)))
-        attrs = non_bool_dict(cutout.data.attrs)
-        attrs.update(ds.attrs)
 
         # Add optional compression to the newly prepared features
         if compression:
-            for v in missing_vars:
-                ds[v].encoding.update(compression)
+            for da in ds.data_vars.values():
+                da.encoding.update(compression)
 
-        ds = cutout.data.merge(ds[missing_vars.values]).assign_attrs(**attrs)
+        ds = cutout.data.merge(
+            ds[missing_vars.loc[list(new_features)].values]
+        ).assign_attrs(attrs)
 
         # write data to tmp file, copy it to original data, this is much safer
         # than appending variables