FIX - Backport turtle file from caribu for compatibility with older Caribu versions

Author: pradal

src/scanalea/light.py

@@ -1,6 +1,6 @@
 from matplotlib import cm
 from alinea.caribu.CaribuScene import CaribuScene
-import alinea.caribu.sky_tools.turtle as _turtle
+import .turtle as _turtle
 
 import openalea.plantgl.all as pgl
 def caribu(scene, source = (1,(0,0,-1)), infinity=False):

src/scanalea/turtle.py

@@ -0,0 +1,70 @@
+from math import radians, degrees, sin , cos
+
+elevations = [9.23, 9.23, 9.23, 9.23, 9.23, 9.23, 9.23, 9.23, 9.23, 9.23, 10.81, 10.81, 10.81, 10.81, 10.81, 26.57, 26.57, 26.57, 26.57, 26.57, 31.08, 31.08, 31.08,31.08, 31.08, 31.08, 31.08, 31.08, 31.08, 31.08, 47.41, 47.41, 47.41, 47.41, 47.41, 52.62, 52.62, 52.62, 52.62, 52.62, 69.16, 69.16, 69.16, 69.16,69.16, 90]
+
+azimuths = [12.23, 59.77, 84.23, 131.77, 156.23, 203.77, 228.23, 275.77, 300.23, 347.77, 36, 108, 180, 252, 324, 0, 72, 144, 216, 288, 23.27, 48.73, 95.27, 120.73,167.27, 192.73, 239.27, 264.73, 311.27, 336.73, 0, 72, 144, 216, 288, 36, 108, 180, 252, 324, 0, 72, 144, 216, 288, 180]
+
+#weights = [0.026808309,0.026808309,0.026808309,0.026808309,0.026808309,0.026808309,0.026808309,0.026808309,0.026808309,0.026808309,0.029325083,0.029325083,0.029325083,0.029325083,0.029325083,0.031299545,0.031299545,0.031299545,0.031299545,0.031299545,0.038160959,0.038160959,0.038160959,0.038160959,0.038160959,0.038160959,0.038160959,0.038160959,0.038160959,0.038160959,0.045638829,0.045638829,0.045638829,0.045638829,0.045638829,0.050212264,0.050212264,0.050212264,0.050212264,0.050212264,0.052965108,0.052965108,0.052965108,0.052965108,0.052965108,0.0481]
+
+weights_soc = [0.0043,0.0043,0.0043,0.0043,0.0043,0.0043,0.0043,0.0043,0.0043,0.0043,0.0055,0.0055,0.0055,0.0055,0.0055,0.014,0.014,0.014,0.014,0.014,0.0197,0.0197,0.0197,0.0197,0.0197,0.0197,0.0197,0.0197,0.0197,0.0197,0.0336,0.0336,0.0336,0.0336,0.0336,0.0399,0.0399,0.0399,0.0399,0.0399,0.0495,0.0495,0.0495,0.0495,0.0495,0.0481]
+
+weights_uoc = [0.007,0.007,0.007,0.007,0.007,0.007,0.007,0.007,0.007,0.007,0.0086,0.0086,0.0086,0.0086,0.0086,0.017,0.017,0.017,0.017,0.017,0.0224,0.0224,0.0224,0.0224,0.0224,0.0224,0.0224,0.0224,0.0224,0.0224,0.0317,0.0317,0.0317,0.0317,0.0317,0.036,0.036,0.036,0.036,0.036,0.0405,0.0405,0.0405,0.0405,0.0405,0.0377]
+
+elevations16 = [90, 26.57, 26.57, 26.57, 26.57, 26.57, 52.62, 52.62, 52.62, 52.62, 52.62, 10.81, 10.81, 10.81, 10.81, 10.81]
+azimuths16 = [180, 0, 72, 144, 216, 288, 36, 108, 180, 252, 324, 36, 108, 180, 252, 324]
+weights16_soc = [0.1468, 0.0448,0.0448,0.0448,0.0448,0.0448,0.108,0.108,0.108,0.108,0.108,0.01777,0.01777,0.01777,0.01777,0.01777]
+weights16_uoc = [0.1173, 0.0533,0.0533,0.0533,0.0533,0.0533,0.0981,0.0981,0.0981,0.0981,0.0981,0.0251,0.0251,0.0251,0.0251,0.0251]
+
+
+
+
+def _turtle(sectors='46', format='soc',energy=1):
+    """ return a generator on parameters of a given turtle """
+    if sectors == '46':
+        el = elevations
+        az = azimuths
+        if format =='soc':
+            w = weights_soc
+        else:
+            w = weights_uoc
+    else:
+        el = elevations16
+        az = azimuths16
+        if format =='soc':
+            w = weights16_soc
+        else:
+            w= weights16_uoc
+    
+    for i in range(len(el)):
+        yield el[i],az[i],w[i] * energy
+
+
+def _vecteur_direction(elevation,azimuth):
+    theta = radians(90 - elevation)
+    phi = radians(azimuth)
+    return sin(theta) * cos(phi),sin(theta) * sin(phi),  -cos(theta)
+
+def _emission(energy, elevation,azimuth):
+    """ return energy of emmision for a source of a given direction and of a given energy received on a horizontal surface"""
+    direction = _vecteur_direction(elevation, azimuth)
+    return energy / abs(direction[2])
+
+
+def turtle(sectors='46', format='soc',energy=1.):
+    """ Return energy on horizontal surface, energy in the direction of emission, direction vector, elevation (deg) and azimuth (deg) for a turtle discretisation of the sky hemisphere"""
+    res = [(energy, _emission(energy,elevation, azimuth), (_vecteur_direction(elevation, azimuth)), elevation, azimuth) for elevation, azimuth, energy in _turtle(sectors,format,energy)]
+    return zip(*res)
+        
+        
+
+#def write_turtle(filename='turtle16.light',sectors='46', format='soc'):
+#    f=open(filename,'w')
+#    for s in _turtle(sectors,format):
+#        f.write(s+'\n')
+#    f.close()
+
+    
+    #for i in range(len(el)):
+    #    pos = -pgl.Vector3(pgl.Vector3.Spherical(1,radians(az[i]),radians(90-el[i])))
+    #    s= "%.5f\t\t%.5f\t%.5f\t%.5f"% (w[i] * energy, pos.x, pos.y, pos.z)
+    #    yield s

test/test_ply.py

@@ -5,6 +5,7 @@
 from time import time
 
 fn = '/media/pradal/DONNEES/pradal/data/plantscan/segmented/segmentedMesh.vtk'
+fn = '/media/pradal/DONNEES/pradal/data/plantscan/segmented/segmentedMesh_manualseg.vtk'
 scene = read(fn)
 Viewer.display(scene)