def plot_isosurface(self, *args, **kwargs):
'''
creates a three-dimensional isosurface plot
: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 cmap: (*string*) Color map string.
:param nthread: (*int*) Thread number.
:returns: Legend
'''
if len(args) <= 3:
x = args[0].dimvalue(2)
y = args[0].dimvalue(1)
z = args[0].dimvalue(0)
data = args[0]
isovalue = args[1]
args = args[2:]
else:
x = args[0]
y = args[1]
z = args[2]
data = args[3]
isovalue = args[4]
args = args[5:]
cmap = plotutil.getcolormap(**kwargs)
cvalue = kwargs.pop('cvalue', None)
if not cvalue is None:
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)
edge = kwargs.pop('edge', True)
kwargs['edge'] = edge
plotutil.setlegendscheme(ls, **kwargs)
else:
ls = plotutil.getlegendbreak('polygon', **kwargs)[0]
nthread = kwargs.pop('nthread', None)
if nthread is None:
graphics = JOGLUtil.isosurface(data.asarray(), x.asarray(), y.asarray(), z.asarray(), isovalue, ls)
else:
graphics = JOGLUtil.isosurface(data.asarray(), x.asarray(), y.asarray(), z.asarray(), isovalue, ls, nthread)
visible = kwargs.pop('visible', True)
if visible:
self.add_graphic(graphics)
return graphics
def geoshow(self, layer, **kwargs):
'''
Plot a layer in 3D axes.
:param layer: (*MILayer*) The layer to be plotted.
:returns: Graphics.
'''
ls = kwargs.pop('symbolspec', None)
offset = kwargs.pop('offset', 0)
xshift = kwargs.pop('xshift', 0)
if isinstance(layer, basestring):
fn = layer
if not fn.endswith('.shp'):
fn = fn + '.shp'
if not os.path.exists(fn):
fn = os.path.join(migl.get_map_folder(), fn)
if os.path.exists(fn):
encoding = kwargs.pop('encoding', None)
layer = migeo.shaperead(fn, encoding)
else:
raise IOError('File not exists: ' + fn)
layer = layer.layer
if layer.getLayerType() == LayerTypes.VectorLayer:
if ls is None:
ls = layer.getLegendScheme()
if layer.getLegendScheme().getBreakNum() == 1:
lb = layer.getLegendScheme().getLegendBreaks().get(0)
btype = lb.getBreakType()
geometry = 'point'
if btype == BreakTypes.PolylineBreak:
geometry = 'line'
elif btype == BreakTypes.PolygonBreak:
geometry = 'polygon'
if not kwargs.has_key('facecolor'):
kwargs['facecolor'] = None
if not kwargs.has_key('edgecolor'):
kwargs['edgecolor'] = 'k'
lb, isunique = plotutil.getlegendbreak(geometry, **kwargs)
ls.getLegendBreaks().set(0, lb)
else:
plotutil.setlegendscheme(ls, **kwargs)
layer.setLegendScheme(ls)
graphics = GraphicFactory.createGraphicsFromLayer(
layer, offset, xshift)
else:
interpolation = kwargs.pop('interpolation', None)
graphics = JOGLUtil.createTexture(self.figure.getGL2(), layer,
offset, xshift, interpolation)
visible = kwargs.pop('visible', True)
if visible:
self.add_graphic(graphics)
return graphics
def plot_surface(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.
: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)
edge = kwargs.pop('edge', True)
kwargs['edge'] = edge
plotutil.setlegendscheme(ls, **kwargs)
graphics = JOGLUtil.surface(x.asarray(), y.asarray(), z.asarray(), ls)
visible = kwargs.pop('visible', True)
if visible:
self.add_graphic(graphics)
return graphics
def geoshow(self, layer, **kwargs):
'''
Plot a layer in 3D axes.
:param layer: (*MILayer*) The layer to be plotted.
:returns: Graphics.
'''
ls = kwargs.pop('symbolspec', None)
offset = kwargs.pop('offset', 0)
xshift = kwargs.pop('xshift', 0)
layer = layer.layer
if layer.getLayerType() == LayerTypes.VectorLayer:
if ls is None:
ls = layer.getLegendScheme()
if len(kwargs) > 0 and layer.getLegendScheme().getBreakNum(
) == 1:
lb = layer.getLegendScheme().getLegendBreaks().get(0)
btype = lb.getBreakType()
geometry = 'point'
if btype == BreakTypes.PolylineBreak:
geometry = 'line'
elif btype == BreakTypes.PolygonBreak:
geometry = 'polygon'
if not kwargs.has_key('facecolor'):
kwargs['facecolor'] = None
lb, isunique = plotutil.getlegendbreak(geometry, **kwargs)
ls.getLegendBreaks().set(0, lb)
plotutil.setlegendscheme(ls, **kwargs)
layer.setLegendScheme(ls)
graphics = GraphicFactory.createGraphicsFromLayer(
layer, offset, xshift)
else:
interpolation = kwargs.pop('interpolation', None)
graphics = JOGLUtil.createTexture(self.figure.getGL2(), layer,
offset, xshift, interpolation)
visible = kwargs.pop('visible', True)
if visible:
self.add_graphic(graphics)
return graphics
def plot_particles(self, *args, **kwargs):
'''
creates a three-dimensional particles plot
: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 s: (*float*) Point size.
:param cmap: (*string*) Color map string.
:param vmin: (*float*) Minimum value for particle plotting.
:param vmax: (*float*) Maximum value for particle plotting.
:param alpha_min: (*float*) Minimum alpha value.
:param alpha_max: (*float*) Maximum alpha value.
:param density: (*int*) Particle density value.
:returns: Legend
'''
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:]
cmap = plotutil.getcolormap(**kwargs)
vmin = kwargs.pop('vmin', data.min())
vmax = kwargs.pop('vmax', data.max())
if vmin >= vmax:
raise ValueError("Minimum value larger than maximum value")
if len(args) > 0:
level_arg = args[0]
if isinstance(level_arg, int):
cn = level_arg
ls = LegendManage.createLegendScheme(vmin, vmax, cn, cmap)
else:
if isinstance(level_arg, NDArray):
level_arg = level_arg.aslist()
ls = LegendManage.createLegendScheme(vmin, vmax, level_arg,
cmap)
else:
ls = LegendManage.createLegendScheme(vmin, vmax, cmap)
plotutil.setlegendscheme(ls, **kwargs)
alpha_min = kwargs.pop('alpha_min', 0.1)
alpha_max = kwargs.pop('alpha_max', 0.6)
density = kwargs.pop('density', 2)
graphics = JOGLUtil.particles(data.asarray(), x.asarray(), y.asarray(), z.asarray(), ls, \
alpha_min, alpha_max, density)
s = kwargs.pop('s', None)
if s is None:
s = kwargs.pop('size', None)
if not s is None:
graphics.setPointSize(s)
visible = kwargs.pop('visible', True)
if visible:
self.add_graphic(graphics)
return graphics
def plot_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)
edge = kwargs.pop('edge', True)
kwargs['edge'] = edge
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)
visible = kwargs.pop('visible', True)
if visible:
for gg in graphics:
self.add_graphic(gg)
return graphics
def plot_surface(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.
:returns: Legend
'''
warnings.warn("plot_surface is deprecated", DeprecationWarning)
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']:
kwargs['facecolor'] = facecolor
plotutil.setlegendscheme(ls, **kwargs)
graphics = JOGLUtil.surface(x.asarray(), y.asarray(), z.asarray(), ls)
if face_interp:
graphics.setFaceInterp(face_interp)
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
