85 lines
3.3 KiB
Python
85 lines
3.3 KiB
Python
|
"""This is an example on running HydroShoot on a potted grapevine with a
|
||
|
|
simple shoot architecture.
|
||
|
|
"""
|
||
|
|
|
||
|
|
from json import load, dump
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
from openalea.mtg.mtg import MTG
|
||
|
|
from openalea.plantgl.all import Scene
|
||
|
|
|
||
|
|
from example.potted_grapevine.sim import build_mtg
|
||
|
|
from hydroshoot import io, model, initialisation
|
||
|
|
|
||
|
|
|
||
|
|
def preprocess_inputs(grapevine_mtg: MTG, path_project_dir: Path, psi_soil: float, gdd_since_budbreak: float,
|
||
|
|
scene: Scene):
|
||
|
|
path_preprocessed_inputs = path_project_dir / 'preprocessed_inputs'
|
||
|
|
path_preprocessed_inputs.mkdir(parents=True, exist_ok=True)
|
||
|
|
|
||
|
|
inputs = io.HydroShootInputs(
|
||
|
|
path_project=path_project_dir,
|
||
|
|
user_params=None,
|
||
|
|
path_weather=path_project / 'meteo.input',
|
||
|
|
scene=scene,
|
||
|
|
psi_soil=psi_soil,
|
||
|
|
gdd_since_budbreak=gdd_since_budbreak)
|
||
|
|
io.verify_inputs(g=grapevine_mtg, inputs=inputs)
|
||
|
|
grapevine_mtg = initialisation.init_model(g=grapevine_mtg, inputs=inputs)
|
||
|
|
|
||
|
|
print("Computing 'static' data...")
|
||
|
|
static_data = {'form_factors': {s: grapevine_mtg.property(s) for s in ('ff_sky', 'ff_leaves', 'ff_soil')}}
|
||
|
|
static_data.update({'Na': grapevine_mtg.property('Na')})
|
||
|
|
with open(path_preprocessed_inputs / 'static.json', mode='w') as f_prop:
|
||
|
|
dump(static_data, f_prop)
|
||
|
|
pass
|
||
|
|
|
||
|
|
print("Computing 'dynamic' data...")
|
||
|
|
dynamic_data = {}
|
||
|
|
inputs_hourly = io.HydroShootHourlyInputs(psi_soil=inputs.psi_soil, sun2scene=inputs.sun2scene)
|
||
|
|
for date_sim in inputs.params.simulation.date_range:
|
||
|
|
inputs_hourly.update(
|
||
|
|
g=grapevine_mtg, date_sim=date_sim, hourly_weather=inputs.weather[inputs.weather.index == date_sim],
|
||
|
|
psi_pd=inputs.psi_pd, params=inputs.params, is_psi_forced=inputs.is_psi_soil_forced)
|
||
|
|
|
||
|
|
grapevine_mtg, diffuse_to_total_irradiance_ratio = initialisation.init_hourly(
|
||
|
|
g=grapevine_mtg, inputs_hourly=inputs_hourly, leaf_ppfd=inputs.leaf_ppfd,
|
||
|
|
params=inputs.params)
|
||
|
|
|
||
|
|
dynamic_data.update({grapevine_mtg.date: {
|
||
|
|
'diffuse_to_total_irradiance_ratio': diffuse_to_total_irradiance_ratio,
|
||
|
|
'Ei': grapevine_mtg.property('Ei'),
|
||
|
|
'Eabs': grapevine_mtg.property('Eabs')}})
|
||
|
|
|
||
|
|
with open(path_preprocessed_inputs / 'dynamic.json', mode='w') as f_prop:
|
||
|
|
dump(dynamic_data, f_prop)
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
path_project = Path(__file__).parent
|
||
|
|
path_preprocessed_data = path_project / 'preprocessed_inputs'
|
||
|
|
|
||
|
|
g, scene = build_mtg(path_file=path_project / 'grapevine_pot.csv', is_show_scene=False)
|
||
|
|
# preprocess_inputs(grapevine_mtg=g, path_project_dir=path_project, psi_soil=-0.2, gdd_since_budbreak=100., scene=scene)
|
||
|
|
|
||
|
|
with open(path_preprocessed_data / 'static.json') as f:
|
||
|
|
static_inputs = load(f)
|
||
|
|
with open(path_preprocessed_data / 'dynamic.json') as f:
|
||
|
|
dynamic_inputs = load(f)
|
||
|
|
with open(path_project / 'params.json') as f:
|
||
|
|
user_params = load(f)
|
||
|
|
|
||
|
|
summary_results = model.run(
|
||
|
|
g=g,
|
||
|
|
wd=path_project,
|
||
|
|
params=user_params,
|
||
|
|
path_weather=path_project / 'meteo.input',
|
||
|
|
scene=scene,
|
||
|
|
psi_soil=-0.2,
|
||
|
|
gdd_since_budbreak=100.,
|
||
|
|
form_factors=static_inputs['form_factors'],
|
||
|
|
leaf_nitrogen=static_inputs['Na'],
|
||
|
|
leaf_ppfd=dynamic_inputs,
|
||
|
|
path_output=path_project / 'output' / 'time_series_with_preprocessed_data.csv')
|