import logging
from pywps import FORMATS, ComplexOutput, Format, LiteralInput, Process
from pywps.app.Common import Metadata
from ..director import wrap_director
from ..utils.input_utils import parse_wps_input, get_grid_param
from ..utils.metalink_utils import build_metalink
from ..utils.regrid_utils import run_regrid
from ..utils.response_utils import populate_response
LOGGER = logging.getLogger()
[docs]
class Regrid(Process):
def __init__(self):
inputs = [
LiteralInput(
"collection",
"Collection",
abstract="A dataset identifier or list of comma separated identifiers. "
"Example: c3s-cmip5.output1.ICHEC.EC-EARTH.historical.day.atmos.day.r1i1p1.tas.latest",
data_type="string",
min_occurs=1,
max_occurs=1,
),
LiteralInput(
"method",
"Regrid method",
abstract="Please specify regridding method like consevative or bilinear. Default: nearest_s2d",
data_type="string",
min_occurs=1,
max_occurs=1,
allowed_values=["nearest_s2d", "bilinear", "conservative", "patch"],
default="nearest_s2d",
),
LiteralInput(
"grid",
"Regrid target grid",
abstract="Select target grid resolution for regridding. "
"Choose a predefined grid or select 'custom' to provide your own.",
data_type="string",
min_occurs=1,
max_occurs=1,
allowed_values=[
"auto",
"0pt25deg",
"0pt25deg_era5",
"0pt5deg_lsm",
"0pt625x0pt5deg",
"0pt75deg",
"1deg",
"1pt25deg",
"2pt5deg",
"custom",
],
default="auto",
),
LiteralInput(
"custom_grid",
"Custom grid definition",
abstract=(
"Only used if grid='custom'. Provide space-separated numbers defining the target grid. "
"Allowed lengths: 1 (d), 2 (d_lon d_lat), 3 (lonlat_0 lonlat_1 d_lonlat), 6 (lon_0 lon_1 d_lon lat_0 lat_1 d_lat). "
"Examples: "
"'0.5' (uniform 0.5° grid), "
"'0.5 0.25' (0.5° lon x 0.25° lat), "
"'-90 90 0.25' (lat-only range), "
"'-10 30 0.25 35 70 0.25' (full bounding box)."
),
data_type="string",
min_occurs=0,
max_occurs=1,
),
]
outputs = [
ComplexOutput(
"output",
"METALINK v4 output",
abstract="Metalink v4 document with references to NetCDF files.",
as_reference=True,
supported_formats=[FORMATS.META4],
),
ComplexOutput(
"prov",
"Provenance",
abstract="Provenance document using W3C standard.",
as_reference=True,
supported_formats=[FORMATS.JSON],
),
ComplexOutput(
"prov_plot",
"Provenance Diagram",
abstract="Provenance document as diagram.",
as_reference=True,
supported_formats=[
Format("image/png", extension=".png", encoding="base64")
],
),
]
super().__init__(
self._handler,
identifier="regrid",
title="Regrid",
abstract="Run regridding operator on climate model data using clisops (xarray).",
metadata=[
Metadata("CLISOPS", "https://github.com/roocs/clisops"),
],
version="1.0",
inputs=inputs,
outputs=outputs,
store_supported=True,
status_supported=True,
)
def _handler(self, request, response):
# from clisops.exceptions import InvalidParameterValue, MissingParameterValue
collection = parse_wps_input(
request.inputs, "collection", as_sequence=True, must_exist=True
)
grid = parse_wps_input(request.inputs, "grid", default="auto")
custom_grid = parse_wps_input(request.inputs, "custom_grid", default="")
inputs = {
"collection": collection,
"output_dir": self.workdir,
"apply_fixes": False,
"pre_checked": False,
"method": parse_wps_input(request.inputs, "method", default="nearest_s2d"),
"grid": get_grid_param(grid, custom_grid),
"adaptive_masking_threshold": 0.5,
}
# print(inputs)
# Let the director manage the processing or redirection to original files
director = wrap_director(collection, inputs, run_regrid)
ml4 = build_metalink(
"regrid-result",
"regrid result as NetCDF files.",
self.workdir,
director.output_uris,
)
populate_response(response, "regrid", self.workdir, inputs, collection, ml4)
return response
