def stationmodel(self, smdata, **kwargs):
"""
Plot station model data on the map.
:param smdata: (*StationModelData*) Station model data.
:param surface: (*boolean*) Is surface data or not. Default is True.
:param size: (*float*) Size of the station model symbols. Default is 12.
:param proj: (*ProjectionInfo*) Map projection of the data. Default is None.
:param order: (*int*) Z-order of created layer for display.
:returns: (*VectoryLayer*) Station model VectoryLayer.
"""
proj = kwargs.pop('proj', None)
size = kwargs.pop('size', 12)
surface = kwargs.pop('surface', True)
ls = LegendManage.createSingleSymbolLegendScheme(ShapeTypes.Point, Color.blue, size)
layer = DrawMeteoData.createStationModelLayer(smdata, ls, 'stationmodel', surface)
if (proj != None):
layer.setProjInfo(proj)
# Add layer
isadd = kwargs.pop('isadd', True)
if isadd:
zorder = kwargs.pop('zorder', None)
select = kwargs.pop('select', True)
self.add_layer(layer, zorder, select)
self.axes.setDrawExtent(layer.getExtent().clone())
self.axes.setExtent(layer.getExtent().clone())
return MILayer(layer)
def streamplot(self, *args, **kwargs):
"""
Plot streamline in a map.
:param x: (*array_like*) Optional. X coordinate array.
:param y: (*array_like*) Optional. Y coordinate array.
:param u: (*array_like*) U component of the arrow vectors (wind field) or wind direction.
:param v: (*array_like*) V component of the arrow vectors (wind field) or wind speed.
:param z: (*array_like*) Optional, 2-D z value array.
:param color: (*Color*) Streamline color. Default is blue.
:param fill_value: (*float*) Fill_value. Default is ``-9999.0``.
:param isuv: (*boolean*) Is U/V or direction/speed data array pairs. Default is True.
:param density: (*int*) Streamline density. Default is 4.
:param proj: (*ProjectionInfo*) Map projection of the data. Default is None.
:param zorder: (*int*) Z-order of created layer for display.
:param select: (*boolean*) Set the return layer as selected layer or not.
:returns: (*VectoryLayer*) Created streamline VectoryLayer.
"""
cmap = plotutil.getcolormap(**kwargs)
fill_value = kwargs.pop('fill_value', -9999.0)
proj = kwargs.pop('proj', None)
cobj = kwargs.pop('color', 'b')
color = plotutil.getcolor(cobj)
isuv = kwargs.pop('isuv', True)
density = kwargs.pop('density', 4)
n = len(args)
if n < 4:
u = args[0]
v = args[1]
y = u.dimvalue(0)
x = u.dimvalue(1)
args = args[2:]
else:
x = args[0]
y = args[1]
u = args[2]
v = args[3]
args = args[4:]
ls = LegendManage.createSingleSymbolLegendScheme(ShapeTypes.Polyline, color, 1)
plotutil.setlegendscheme(ls, **kwargs)
#layer = __plot_uvgriddata_m(plot, udata, vdata, None, ls, 'streamplot', isuv, proj=proj, density=density)
layer = DrawMeteoData.createStreamlineLayer(u.array, v.array, x.array, y.array, density, ls, 'layer', isuv)
if not proj is None:
layer.setProjInfo(proj)
# Add layer
isadd = kwargs.pop('isadd', True)
if isadd:
zorder = kwargs.pop('zorder', None)
select = kwargs.pop('select', True)
self.add_layer(layer, zorder, select)
self.axes.setDrawExtent(layer.getExtent().clone())
self.axes.setExtent(layer.getExtent().clone())
return MILayer(layer)
def makesymbolspec(geometry, *args, **kwargs):
'''
Make a legend.
:param geometry: (*string*) Geometry type. [point | line | polygon].
:param levels: (*array_like*) Value levels. Default is ``None``, not used.
:param colors: (*list*) Colors. Defaul is ``None``, not used.
:param legend break parameter maps: (*map*) Legend breaks.
:param field: (*string*) The field to be used in the legend.
:returns: Created legend.
'''
shapetype = ShapeTypes.Image
if geometry == 'point':
shapetype = ShapeTypes.Point
elif geometry == 'line':
shapetype = ShapeTypes.Polyline
elif geometry == 'polygon':
shapetype = ShapeTypes.Polygon
else:
shapetype = ShapeTypes.Image
levels = kwargs.pop('levels', None)
cols = kwargs.pop('colors', None)
field = kwargs.pop('field', '')
if not levels is None and not cols is None:
if isinstance(levels, MIArray):
levels = levels.aslist()
colors = []
for cobj in cols:
colors.append(getcolor(cobj))
ls = LegendManage.createLegendScheme(shapetype, levels, colors)
setlegendscheme(ls, **kwargs)
ls.setFieldName(field)
values = kwargs.pop('values', None)
if values is None:
return ls
else:
nls = LegendScheme(ls.getShapeType())
for v in values:
nls.addLegendBreak(ls.findLegendBreak(v))
return nls
n = len(args)
isunique = True
if n == 0:
ls = LegendManage.createSingleSymbolLegendScheme(shapetype)
setlegendscheme(ls, **kwargs)
elif n == 1 and isinstance(args[0], int):
ls = LegendManage.createUniqValueLegendScheme(args[0], shapetype)
setlegendscheme(ls, **kwargs)
else:
ls = LegendScheme(shapetype)
for arg in args:
if isinstance(arg, (list, tuple)):
for argi in arg:
lb, isu = getlegendbreak(geometry, **argi)
if isunique and not isu:
isunique = False
ls.addLegendBreak(lb)
else:
lb, isu = getlegendbreak(geometry, **arg)
if isunique and not isu:
isunique = False
ls.addLegendBreak(lb)
ls.setFieldName(field)
if ls.getBreakNum() > 1:
if isunique:
ls.setLegendType(LegendType.UniqueValue)
else:
ls.setLegendType(LegendType.GraduatedColor)
return ls
def surf(self, *args, **kwargs):
'''
creates a three-dimensional surface plot
:param x: (*array_like*) Optional. X coordinate array.
:param y: (*array_like*) Optional. Y coordinate array.
:param z: (*array_like*) 2-D z value array.
:param cmap: (*string*) Color map string.
:param lighting: (*bool*) Using light or not.
:returns: Legend
'''
if len(args) <= 2:
x = args[0].dimvalue(1)
y = args[0].dimvalue(0)
x, y = np.meshgrid(x, y)
z = args[0]
args = args[1:]
else:
x = args[0]
y = args[1]
z = args[2]
args = args[3:]
if kwargs.has_key('colors'):
cn = len(kwargs['colors'])
else:
cn = None
cmap = plotutil.getcolormap(**kwargs)
if len(args) > 0:
level_arg = args[0]
if isinstance(level_arg, int):
cn = level_arg
ls = LegendManage.createLegendScheme(z.min(), z.max(), cn,
cmap)
else:
if isinstance(level_arg, NDArray):
level_arg = level_arg.aslist()
ls = LegendManage.createLegendScheme(z.min(), z.max(),
level_arg, cmap)
else:
if cn is None:
ls = LegendManage.createLegendScheme(z.min(), z.max(), cmap)
else:
ls = LegendManage.createLegendScheme(z.min(), z.max(), cn,
cmap)
ls = ls.convertTo(ShapeTypes.Polygon)
facecolor = kwargs.pop('facecolor', None)
face_interp = None
if not facecolor is None:
face_interp = (facecolor == 'interp')
if not face_interp:
if not facecolor in ['flat', 'texturemap', 'none']:
facecolor = plotutil.getcolor(facecolor)
ls = LegendManage.createSingleSymbolLegendScheme(
ShapeTypes.Polygon, facecolor, 1)
plotutil.setlegendscheme(ls, **kwargs)
graphics = JOGLUtil.surface(x.asarray(), y.asarray(), z.asarray(), ls)
if face_interp:
graphics.setFaceInterp(face_interp)
lighting = kwargs.pop('lighting', None)
if not lighting is None:
graphics.setUsingLight(lighting)
visible = kwargs.pop('visible', True)
if visible:
self.add_graphic(graphics)
return graphics
def slice(self, *args, **kwargs):
'''
Volume slice planes
:param x: (*array_like*) Optional. X coordinate array.
:param y: (*array_like*) Optional. Y coordinate array.
:param z: (*array_like*) Optional. Z coordinate array.
:param data: (*array_like*) 3D data array.
:param xslice: (*list*) X slice locations.
:param yslice: (*list*) Y slice locations.
:param zslice: (*list*) Z slice locations.
:param cmap: (*string*) Color map string.
:return:
'''
if len(args) <= 3:
x = args[0].dimvalue(2)
y = args[0].dimvalue(1)
z = args[0].dimvalue(0)
data = args[0]
args = args[1:]
else:
x = args[0]
y = args[1]
z = args[2]
data = args[3]
args = args[4:]
if x.ndim == 3:
x = x[0, 0]
if y.ndim == 3:
y = y[0, :, 0]
if z.ndim == 3:
z = z[:, 0, 0]
cmap = plotutil.getcolormap(**kwargs)
if len(args) > 0:
level_arg = args[0]
if isinstance(level_arg, int):
cn = level_arg
ls = LegendManage.createLegendScheme(data.min(), data.max(),
cn, cmap)
else:
if isinstance(level_arg, NDArray):
level_arg = level_arg.aslist()
ls = LegendManage.createLegendScheme(data.min(), data.max(),
level_arg, cmap)
else:
ls = LegendManage.createLegendScheme(data.min(), data.max(), cmap)
ls = ls.convertTo(ShapeTypes.Polygon)
facecolor = kwargs.pop('facecolor', None)
face_interp = None
if not facecolor is None:
face_interp = (facecolor == 'interp')
if not face_interp:
if not facecolor in ['flat', 'texturemap', 'none']:
facecolor = plotutil.getcolor(facecolor)
ls = LegendManage.createSingleSymbolLegendScheme(
ShapeTypes.Polygon, facecolor, 1)
plotutil.setlegendscheme(ls, **kwargs)
xslice = kwargs.pop('xslice', [])
if isinstance(xslice, numbers.Number):
xslice = [xslice]
yslice = kwargs.pop('yslice', [])
if isinstance(yslice, numbers.Number):
yslice = [yslice]
zslice = kwargs.pop('zslice', [])
if isinstance(zslice, numbers.Number):
zslice = [zslice]
graphics = JOGLUtil.slice(data.asarray(), x.asarray(), y.asarray(), z.asarray(), xslice, \
yslice, zslice, ls)
if face_interp:
for gg in graphics:
gg.setFaceInterp(face_interp)
lighting = kwargs.pop('lighting', None)
if not lighting is None:
for gg in graphics:
gg.setUsingLight(lighting)
visible = kwargs.pop('visible', True)
if visible:
for gg in graphics:
self.add_graphic(gg)
return graphics
def streamslice(self, *args, **kwargs):
"""
Plot stream lines slice in 3D axes.
:param x: (*array_like*) X coordinate array.
:param y: (*array_like*) Y coordinate array.
:param z: (*array_like*) Z coordinate array.
:param u: (*array_like*) U component of the arrow vectors (wind field).
:param v: (*array_like*) V component of the arrow vectors (wind field).
:param w: (*array_like*) W component of the arrow vectors (wind field).
:param xslice: (*list*) X slice locations.
:param yslice: (*list*) Y slice locations.
:param zslice: (*list*) Z slice locations.
:param density: (*int*) Streamline density. Default is 4.
:return: Streamline slices
"""
ls = kwargs.pop('symbolspec', None)
cmap = plotutil.getcolormap(**kwargs)
density = kwargs.pop('density', 4)
iscolor = False
cdata = None
if len(args) < 6:
u = args[0]
v = args[1]
w = args[2]
u = np.asarray(u)
nz, ny, nx = u.shape
x = np.arange(nx)
y = np.arange(ny)
z = np.arange(nz)
args = args[3:]
else:
x = args[0]
y = args[1]
z = args[2]
u = args[3]
v = args[4]
w = args[5]
args = args[6:]
if len(args) > 0:
cdata = args[0]
iscolor = True
args = args[1:]
x = plotutil.getplotdata(x)
y = plotutil.getplotdata(y)
z = plotutil.getplotdata(z)
u = plotutil.getplotdata(u)
v = plotutil.getplotdata(v)
w = plotutil.getplotdata(w)
if ls is None:
if iscolor:
if len(args) > 0:
cn = args[0]
ls = LegendManage.createLegendScheme(
cdata.min(), cdata.max(), cn, cmap)
else:
levs = kwargs.pop('levs', None)
if levs is None:
ls = LegendManage.createLegendScheme(
cdata.min(), cdata.max(), cmap)
else:
if isinstance(levs, NDArray):
levs = levs.tolist()
ls = LegendManage.createLegendScheme(
cdata.min(), cdata.max(), levs, cmap)
else:
if cmap.getColorCount() == 1:
c = cmap.getColor(0)
else:
c = Color.black
ls = LegendManage.createSingleSymbolLegendScheme(
ShapeTypes.Polyline, c, 1)
ls = plotutil.setlegendscheme_line(ls, **kwargs)
if not kwargs.has_key('headwidth'):
kwargs['headwidth'] = 1
if not kwargs.has_key('headlength'):
kwargs['headlength'] = 2.5
for i in range(ls.getBreakNum()):
lb = plotutil.line2stream(ls.getLegendBreak(i), **kwargs)
ls.setLegendBreak(i, lb)
if not cdata is None:
cdata = plotutil.getplotdata(cdata)
min_points = kwargs.pop('min_points', 3)
zslice_index = kwargs.pop('zslice_index', None)
if zslice_index is None:
xslice = kwargs.pop('xslice', [])
if isinstance(xslice, numbers.Number):
xslice = [xslice]
yslice = kwargs.pop('yslice', [])
if isinstance(yslice, numbers.Number):
yslice = [yslice]
zslice = kwargs.pop('zslice', [])
if isinstance(zslice, numbers.Number):
zslice = [zslice]
graphics = GraphicFactory.streamSlice(x, y, z, u, v, w, cdata,
xslice, yslice, zslice,
density, ls)
else:
if isinstance(zslice_index, int):
zslice_index = [zslice_index]
graphics = GraphicFactory.streamSlice(x, y, z, u, v, w, cdata,
zslice_index, density, ls)
lighting = kwargs.pop('lighting', None)
if not lighting is None:
for gg in graphics:
gg.setUsingLight(lighting)
visible = kwargs.pop('visible', True)
if visible:
for gg in graphics:
self.add_graphic(gg)
return graphics
def streamplot(self, *args, **kwargs):
"""
Plot stream lines in 3D axes.
:param x: (*array_like*) X coordinate array.
:param y: (*array_like*) Y coordinate array.
:param z: (*array_like*) Z coordinate array.
:param u: (*array_like*) U component of the arrow vectors (wind field).
:param v: (*array_like*) V component of the arrow vectors (wind field).
:param w: (*array_like*) W component of the arrow vectors (wind field).
:param density: (*int*) Streamline density. Default is 4.
:return: Streamlines
"""
ls = kwargs.pop('symbolspec', None)
cmap = plotutil.getcolormap(**kwargs)
density = kwargs.pop('density', 4)
iscolor = False
cdata = None
if len(args) < 6:
u = args[0]
v = args[1]
w = args[2]
u = np.asarray(u)
nz, ny, nx = u.shape
x = np.arange(nx)
y = np.arange(ny)
z = np.arange(nz)
args = args[3:]
else:
x = args[0]
y = args[1]
z = args[2]
u = args[3]
v = args[4]
w = args[5]
args = args[6:]
if len(args) > 0:
cdata = args[0]
iscolor = True
args = args[1:]
x = plotutil.getplotdata(x)
y = plotutil.getplotdata(y)
z = plotutil.getplotdata(z)
u = plotutil.getplotdata(u)
v = plotutil.getplotdata(v)
w = plotutil.getplotdata(w)
if ls is None:
if iscolor:
if len(args) > 0:
cn = args[0]
ls = LegendManage.createLegendScheme(
cdata.min(), cdata.max(), cn, cmap)
else:
levs = kwargs.pop('levs', None)
if levs is None:
ls = LegendManage.createLegendScheme(
cdata.min(), cdata.max(), cmap)
else:
if isinstance(levs, NDArray):
levs = levs.tolist()
ls = LegendManage.createLegendScheme(
cdata.min(), cdata.max(), levs, cmap)
else:
if cmap.getColorCount() == 1:
c = cmap.getColor(0)
else:
c = Color.black
ls = LegendManage.createSingleSymbolLegendScheme(
ShapeTypes.Polyline, c, 1)
ls = plotutil.setlegendscheme_line(ls, **kwargs)
if not kwargs.has_key('headwidth'):
kwargs['headwidth'] = 1
if not kwargs.has_key('headlength'):
kwargs['headlength'] = 2.5
for i in range(ls.getBreakNum()):
lb = plotutil.line2stream(ls.getLegendBreak(i), **kwargs)
ls.setLegendBreak(i, lb)
if not cdata is None:
cdata = plotutil.getplotdata(cdata)
min_points = kwargs.pop('min_points', 3)
start_x = kwargs.pop('start_x', None)
start_y = kwargs.pop('start_y', None)
start_z = kwargs.pop('start_z', None)
if start_x is None or start_y is None or start_z is None:
graphics = GraphicFactory.createStreamlines3D(
x, y, z, u, v, w, cdata, density, ls, min_points)
else:
start_x = np.asarray(start_x).flatten()
start_y = np.asarray(start_y).flatten()
start_z = np.asarray(start_z).flatten()
graphics = GraphicFactory.createStreamlines3D(
x, y, z, u, v, w, cdata, density, ls, min_points,
start_x._array, start_y._array, start_z._array)
lighting = kwargs.pop('lighting', None)
if not lighting is None:
graphics.setUsingLight(lighting)
self.add_graphic(graphics)
return graphics
def quiver(self, *args, **kwargs):
"""
Plot a 2-D field of arrows.
:param x: (*array_like*) X coordinate array.
:param y: (*array_like*) Y coordinate array.
:param z: (*array_like*) Z coordinate array.
:param u: (*array_like*) U component of the arrow vectors (wind field).
:param v: (*array_like*) V component of the arrow vectors (wind field).
:param w: (*array_like*) W component of the arrow vectors (wind field).
:param z: (*array_like*) Optional, 2-D z value array.
:param levs: (*array_like*) Optional. A list of floating point numbers indicating the level
vectors to draw, in increasing order.
:param cmap: (*string*) Color map string.
:param fill_value: (*float*) Fill_value. Default is ``-9999.0``.
:param length: (*float*) The length of each quiver, default to 1.0, the unit is
the same with the axes.
:returns: (*Graphic list*) Created quiver graphics.
"""
ls = kwargs.pop('symbolspec', None)
cmap = plotutil.getcolormap(**kwargs)
fill_value = kwargs.pop('fill_value', -9999.0)
n = len(args)
iscolor = False
cdata = None
xaxistype = None
x = args[0]
y = args[1]
z = args[2]
u = args[3]
v = args[4]
w = args[5]
args = args[6:]
if len(args) > 0:
cdata = args[0]
iscolor = True
args = args[1:]
x = plotutil.getplotdata(x)
y = plotutil.getplotdata(y)
z = plotutil.getplotdata(z)
u = plotutil.getplotdata(u)
v = plotutil.getplotdata(v)
w = plotutil.getplotdata(w)
if ls is None:
if iscolor:
if len(args) > 0:
cn = args[0]
ls = LegendManage.createLegendScheme(
cdata.min(), cdata.max(), cn, cmap)
else:
levs = kwargs.pop('levs', None)
if levs is None:
ls = LegendManage.createLegendScheme(
cdata.min(), cdata.max(), cmap)
else:
if isinstance(levs, NDArray):
levs = levs.tolist()
ls = LegendManage.createLegendScheme(
cdata.min(), cdata.max(), levs, cmap)
else:
if cmap.getColorCount() == 1:
c = cmap.getColor(0)
else:
c = Color.black
ls = LegendManage.createSingleSymbolLegendScheme(
ShapeTypes.Point, c, 10)
ls = plotutil.setlegendscheme_point(ls, **kwargs)
if not cdata is None:
cdata = plotutil.getplotdata(cdata)
length = kwargs.pop('length', 1)
igraphic = GraphicFactory.createArrows3D(x, y, z, u, v, w, length,
cdata, ls)
visible = kwargs.pop('visible', True)
if visible:
self.add_graphic(igraphic)
return igraphic
def barbs(self, *args, **kwargs):
"""
Plot a 2-D field of barbs in a map.
:param x: (*array_like*) Optional. X coordinate array.
:param y: (*array_like*) Optional. Y coordinate array.
:param u: (*array_like*) U component of the arrow vectors (wind field) or wind direction.
:param v: (*array_like*) V component of the arrow vectors (wind field) or wind speed.
:param z: (*array_like*) Optional, 2-D z value array.
:param levs: (*array_like*) Optional. A list of floating point numbers indicating the level
barbs to draw, in increasing order.
:param cmap: (*string*) Color map string.
:param fill_value: (*float*) Fill_value. Default is ``-9999.0``.
:param isuv: (*boolean*) Is U/V or direction/speed data array pairs. Default is True.
:param size: (*float*) Base size of the arrows.
:param proj: (*ProjectionInfo*) Map projection of the data. Default is None.
:param zorder: (*int*) Z-order of created layer for display.
:param select: (*boolean*) Set the return layer as selected layer or not.
:returns: (*VectoryLayer*) Created barbs VectoryLayer.
"""
cmap = plotutil.getcolormap(**kwargs)
fill_value = kwargs.pop('fill_value', -9999.0)
proj = kwargs.pop('proj', None)
order = kwargs.pop('order', None)
isuv = kwargs.pop('isuv', True)
n = len(args)
iscolor = False
cdata = None
onlyuv = True
if n >= 4 and isinstance(args[3], (DimArray, MIArray)):
onlyuv = False
if onlyuv:
u = minum.asarray(args[0])
v = minum.asarray(args[1])
xx = args[0].dimvalue(1)
yy = args[0].dimvalue(0)
x, y = minum.meshgrid(xx, yy)
args = args[2:]
if len(args) > 0:
cdata = minum.asarray(args[0])
iscolor = True
args = args[1:]
else:
x = minum.asarray(args[0])
y = minum.asarray(args[1])
u = minum.asarray(args[2])
v = minum.asarray(args[3])
args = args[4:]
if len(args) > 0:
cdata = minum.asarray(args[0])
iscolor = True
args = args[1:]
if iscolor:
if len(args) > 0:
cn = args[0]
ls = LegendManage.createLegendScheme(cdata.min(), cdata.max(), cn, cmap)
else:
levs = kwargs.pop('levs', None)
if levs is None:
ls = LegendManage.createLegendScheme(cdata.min(), cdata.max(), cmap)
else:
if isinstance(levs, MIArray):
levs = levs.tolist()
ls = LegendManage.createLegendScheme(cdata.min(), cdata.max(), levs, cmap)
else:
if cmap.getColorCount() == 1:
c = cmap.getColor(0)
else:
c = Color.black
ls = LegendManage.createSingleSymbolLegendScheme(ShapeTypes.Point, c, 10)
ls = plotutil.setlegendscheme_point(ls, **kwargs)
if not cdata is None:
cdata = cdata.array
layer = DrawMeteoData.createBarbLayer(x.array, y.array, u.array, v.array, cdata, ls, 'layer', isuv)
if not proj is None:
layer.setProjInfo(proj)
# Add layer
isadd = kwargs.pop('isadd', True)
if isadd:
zorder = kwargs.pop('zorder', None)
select = kwargs.pop('select', True)
self.add_layer(layer, zorder, select)
self.axes.setDrawExtent(layer.getExtent().clone())
self.axes.setExtent(layer.getExtent().clone())
return MILayer(layer)
def quiver(self, *args, **kwargs):
"""
Plot a 2-D field of arrows.
:param x: (*array_like*) X coordinate array.
:param y: (*array_like*) Y coordinate array.
:param z: (*array_like*) Z coordinate array.
:param u: (*array_like*) U component of the arrow vectors (wind field).
:param v: (*array_like*) V component of the arrow vectors (wind field).
:param w: (*array_like*) W component of the arrow vectors (wind field).
:param z: (*array_like*) Optional, 2-D z value array.
:param levs: (*array_like*) Optional. A list of floating point numbers indicating the level
vectors to draw, in increasing order.
:param cmap: (*string*) Color map string.
:param fill_value: (*float*) Fill_value. Default is ``-9999.0``.
:param length: (*float*) The length of each quiver, default to 1.0, the unit is
the same with the axes.
:returns: (*Graphic list*) Created quiver graphics.
"""
ls = kwargs.pop('symbolspec', None)
cmap = plotutil.getcolormap(**kwargs)
fill_value = kwargs.pop('fill_value', -9999.0)
n = len(args)
iscolor = False
cdata = None
xaxistype = None
x = args[0]
y = args[1]
z = args[2]
u = args[3]
v = args[4]
w = args[5]
args = args[6:]
if len(args) > 0:
cdata = args[0]
iscolor = True
args = args[1:]
x = plotutil.getplotdata(x)
y = plotutil.getplotdata(y)
z = plotutil.getplotdata(z)
u = plotutil.getplotdata(u)
v = plotutil.getplotdata(v)
w = plotutil.getplotdata(w)
if ls is None:
if iscolor:
if len(args) > 0:
cn = args[0]
ls = LegendManage.createLegendScheme(cdata.min(), cdata.max(), cn, cmap)
else:
levs = kwargs.pop('levs', None)
if levs is None:
ls = LegendManage.createLegendScheme(cdata.min(), cdata.max(), cmap)
else:
if isinstance(levs, MIArray):
levs = levs.tolist()
ls = LegendManage.createLegendScheme(cdata.min(), cdata.max(), levs, cmap)
else:
if cmap.getColorCount() == 1:
c = cmap.getColor(0)
else:
c = Color.black
ls = LegendManage.createSingleSymbolLegendScheme(ShapeTypes.Point, c, 10)
ls = plotutil.setlegendscheme_point(ls, **kwargs)
if not cdata is None:
cdata = plotutil.getplotdata(cdata)
length = kwargs.pop('length', 1)
igraphic = GraphicFactory.createArrows3D(x, y, z, u, v, w, length, cdata, ls)
visible = kwargs.pop('visible', True)
if visible:
self.add_graphic(igraphic)
return igraphic
