tos.py

if __name__ == "__main__":
    from supported_variables.utils.supported_variable import (
        supported_variable,
        preprocessing,
    )
    import utils.utils as utils
else:
    from supported_variables.utils.supported_variable import (
        supported_variable,
        preprocessing,
    )
    import supported_variables.utils.utils as utils

from utils.import_cdo import cdo
from typing import List, Tuple, Union
import os.path as path


@supported_variable
class Tos:
    realm = "o"


@preprocessing(Tos, "BRIDGE")
def preprocessing_1(
    inputs: List[Tuple[str, str]], output_directory: str
) -> List[Tuple[str, str]]:
    file, var = inputs[0]

    name = path.basename(file).replace(".nc", ".out.nc")
    output = path.join(output_directory, name)

    cdo.sellonlatbox("-180,180,90,-90", input=f"-selvar,{var} {file}", output=output)

    return [(output, var)]


# SST data is not saved in annual mean data,
# so we need to calculate the annual mean from monthly data
@preprocessing(Tos, "BRIDGE-monthly-to-annual")
def preprocessing_2(
    inputs: List[Tuple[str, str]], output_directory: str
) -> List[Tuple[str, str]]:
    file, var = inputs[0]

    name = path.basename(file).replace(".nc", ".out.nc")
    output = path.join(output_directory, name)

    cdo.sellonlatbox(
        "-180,180,90,-90", input=f" -yearmean -selvar,{var} {file}", output=output
    )

    return [(output, var)]