Text this: Towards a more flexible representation of water stress effects in the nonlinear Jarvis model