def sweep(vid):
_color = pgl.Material(color(vid))
if diams and (vid in diams):
diameters = diams[vid]
if isinstance(diameters, (list, tuple)):
_geom = pgl.Extrusion(
pgl.Polyline(map(pgl.Vector3, geoms[vid])), section,
pgl.Point2Array(zip(diams[vid], diams[vid])))
else:
_geom = pgl.Extrusion(
pgl.Polyline(map(pgl.Vector3, geoms[vid])), section,
pgl.Point2Array([(diameters, diameters)
for i in range(len(geoms))]))
else:
_geom = pgl.Extrusion(pgl.Polyline(map(pgl.Vector3, geoms[vid])),
section)
return pgl.Shape(_geom, _color)
def to_geom_args(self, conversion=1.0, name=None):
r"""Get arguments for creating a PlantGL geometry.
Args:
conversion (float, optional): Conversion factor that should be
applied to the vertices. Defaults to 1.0.
name (str, optional): Name that should be given to the created
PlantGL symbol. Defaults to None and is ignored.
Returns:
tuple: Class, arguments and keyword arguments for PlantGL geometry.
"""
import openalea.plantgl.all as pgl
smb_class, args, kwargs = super(ObjDict, self).to_geom_args(
conversion=conversion, name=name)
index_class = pgl.Index3
array_class = pgl.Index3Array
# Texture coords
if self.get('texcoords', []):
obj_texcoords = []
for t in self['texcoords']:
obj_texcoords.append(pgl.Vector2(t[0], t[1]))
kwargs['texCoordList'] = pgl.Point2Array(obj_texcoords)
if self.get('face_texcoords', []):
obj_ftexcoords = []
for t in self['face_texcoords']:
if (t is not None) and (t[0] is not None):
entry = [int(_t) for _t in t]
else:
entry = [len(self['texcoords']) for _ in range(3)]
obj_ftexcoords.append(index_class(*entry))
kwargs['texCoordIndexList'] = array_class(obj_ftexcoords)
# Normals
if self.get('normals', []):
obj_normals = []
for n in self['normals']:
obj_normals.append(pgl.Vector3(n[0], n[1], n[2]))
kwargs['normalList'] = pgl.Point3Array(obj_normals)
if self.get('face_normals', []):
obj_fnormals = []
for n in self['face_normals']:
if (n is not None) and (n[0] is not None):
entry = [int(_n) for _n in n]
else:
entry = [len(self['normals']) for _ in range(3)]
obj_fnormals.append(index_class(*entry))
kwargs['normalIndexList'] = array_class(obj_fnormals)
return smb_class, args, kwargs
def fn(change):
if 'new' in change:
value = change['new']
if isinstance(param, BaseScalar):
param.value = value
elif isinstance(param, tuple):
if isinstance(param[1],
(pgl.NurbsCurve2D, pgl.BezierCurve2D)):
param[1].ctrlPointList = pgl.Point3Array(
[pgl.Vector3(p[0], p[1], 1) for p in value])
elif isinstance(param[1], pgl.Polyline2D):
param[1].pointList = pgl.Point2Array(
[pgl.Vector2(p[0], p[1]) for p in value])
lsw.set_parameters(lp.dumps())
def on_param_changed(change):
if 'new' in change:
value = change['new']
name = change['name']
if isinstance(param, BaseScalar):
if name == 'name':
param.name = value
else:
param.value = value
elif isinstance(param, tuple):
if name == 'value':
if isinstance(param[1],
(pgl.NurbsCurve2D, pgl.BezierCurve2D)):
param[1].ctrlPointList = pgl.Point3Array(
[pgl.Vector3(p[0], p[1], 1) for p in value])
elif isinstance(param[1], pgl.Polyline2D):
param[1].pointList = pgl.Point2Array(
[pgl.Vector2(p[0], p[1]) for p in value])
else:
param[1].name = value
if self.__auto_apply:
self.on_lpy_context_change(self.__lp.dumps())
if self.__auto_save:
self.__save()
def to_geom_args(self, conversion=1.0, name=None): # pragma: lpy
r"""Get arguments for creating a PlantGL geometry.
Args:
conversion (float, optional): Conversion factor that should be
applied to the vertices. Defaults to 1.0.
name (str, optional): Name that should be given to the created
PlantGL symbol. Defaults to None and is ignored.
Returns:
tuple: Class, arguments and keyword arguments for PlantGL geometry.
"""
import openalea.plantgl.all as pgl
smb_class, args, kwargs = super(ObjDict, self).to_geom_args(
conversion=conversion, name=name, _as_obj=True)
index_class = pgl.Index
array_class = pgl.IndexArray
# Texture coords
if self.get('texcoords', []):
obj_texcoords = []
for t in self['texcoords']:
obj_texcoords.append(
pgl.Vector2(np.float64(t['u']), np.float64(t.get('v',
0.0))))
kwargs['texCoordList'] = pgl.Point2Array(obj_texcoords)
obj_ftexcoords = []
for i, f in enumerate(self['faces']):
entry = []
for _f in f:
if 'texcoord_index' not in _f:
if i > 0:
warnings.warn(
("'texcoord_index' missing from face" +
"%d, texcoord indices will be " + "ignored.")
% i)
obj_ftexcoords = []
entry = []
break
entry.append(int(_f['texcoord_index']))
if not entry:
break
obj_ftexcoords.append(index_class(*entry))
if obj_ftexcoords:
kwargs['texCoordIndexList'] = array_class(obj_ftexcoords)
# Normals
if self.get('normals', []):
obj_normals = []
for n in self['normals']:
obj_normals.append(
pgl.Vector3(np.float64(n['i']), np.float64(n['j']),
np.float64(n['k'])))
kwargs['normalList'] = pgl.Point3Array(obj_normals)
obj_fnormals = []
for i, f in enumerate(self['faces']):
entry = []
for _f in f:
if 'normal_index' not in _f:
if i > 0:
warnings.warn(
("'normal_index' missing from face" +
"%d, normal indices will be " + "ignored.") %
i)
obj_fnormals = []
entry = []
break
entry.append(int(_f['normal_index']))
if not entry:
break
obj_fnormals.append(index_class(*entry))
if obj_fnormals:
kwargs['normalIndexList'] = array_class(obj_fnormals)
return smb_class, args, kwargs