Commit last-minute

hydroshoot/example/misc/sim.py

@@ -0,0 +1,47 @@
+from pathlib import Path
+from os import getcwd
+
+from openalea.mtg import traversal
+from openalea.plantgl.all import Scene
+
+from hydroshoot import architecture, display, model
+
+# =============================================================================
+# Construct the plant mock-up
+# =============================================================================
+
+
+if __name__ == '__main__':
+    g = architecture.vine_mtg(Path(__file__).parent / 'HSdanugue$5132.csv')
+
+    # Local Coordinates Correction
+    for v in traversal.iter_mtg2(g, g.root):
+        n = g.node(g.Trunk(v, Scale=1)[0])
+        theta = 180 if int(n.index()) < 200 else -90 if int(n.index()) < 300 else 0
+        architecture.vine_orientation(g, v, theta, local_rotation=True)
+
+    # Scene rotation
+    for v in traversal.iter_mtg2(g, g.root):
+        architecture.vine_orientation(g, v, 90., local_rotation=False)
+
+    for v in traversal.iter_mtg2(g, g.root):
+        architecture.vine_phyto_modular(g, v)
+        architecture.vine_mtg_properties(g, v)
+        architecture.vine_mtg_geometry(g, v)
+        architecture.vine_transform(g, v)
+
+    # Display of the plant mock-up (result in 'fig_01_plant_mock_up.png')
+    scene = display.visu(g, def_elmnt_color_dict=True, scene=Scene(), view_result=True)
+
+    # =============================================================================
+    # Run HydroShoot
+    # =============================================================================
+
+    model.run(g, str(getcwd()) + '/', scene, psi_soil=-0.5, gdd_since_budbreak=1000.)
+
+
+
+
+from pandas import read_csv
+df = read_csv(r'C:\Users\albashar\Documents\dvp\grapesoil\grape_soil\example\data\benoit\data_meteo_with_low_WS.csv',
+              sep=';', decimal='.')