Skeleton for a raster_map equation

new_work_intensity.py

"""Calculate work intensity with the Hothaps function.

Written by request of @empavia to try out as a new work function for the InVEST Urban Cooling Model.

To install dependencies, `mamba install pygeoprocessing`.
"""
import pygeoprocessing


def calculate_work_intensity(target_raster, wbgt_raster, alpha1, alpha2):
  """Calculate work intensity with the Hothaps function.
  
  Args:
      target_raster (str): The path to the new work intensity raster being created.
      wbgt_raster (str): The path to an already-calulated Wet Bulb Globe Temp raster.
      alpha1 (float): The alpha1 parameter.
      alpha2 (float): The alpha2 paramter.
  
  Returns:
      ``None``.
  """
  def _calculate_workability(wbgt):
    return 0.1 + (0.9/(1+((wbgt/alpha1)**alpha2)))
  
  pygeoprocessing.raster_map(_calculate_workability, [wbgt_raster], target_raster)