def plot(self, data1, data2, tmpl, grid, transform):
"""Overrides baseclass implementation."""
# Preserve time and z axis for plotting these inof in rendertemplate
geo = None # to make flake8 happy
projection = vcs.elements["projection"][self._gm.projection]
returned = {}
taxis = data1.getTime()
if data1.ndim > 2:
zaxis = data1.getAxis(-3)
else:
zaxis = None
# Ok get3 only the last 2 dims
data1 = self._context().trimData2D(data1)
data2 = self._context().trimData2D(data2)
scale = 1.0
lat = None
lon = None
latAccessor = data1.getLatitude()
lonAccesrsor = data1.getLongitude()
if latAccessor:
lat = latAccessor[:]
if lonAccesrsor:
lon = lonAccesrsor[:]
gridGenDict = vcs2vtk.genGridOnPoints(data1, self._gm, deep=False, grid=grid,
geo=transform, data2=data2)
data1 = gridGenDict["data"]
data2 = gridGenDict["data2"]
geo = gridGenDict["geo"]
for k in ['vtk_backend_grid', 'xm', 'xM', 'ym', 'yM', 'continents',
'wrap', 'geo']:
exec("%s = gridGenDict['%s']" % (k, k))
grid = gridGenDict['vtk_backend_grid']
self._dataWrapModulo = gridGenDict['wrap']
if geo is not None:
newv = vtk.vtkDoubleArray()
newv.SetNumberOfComponents(3)
newv.InsertTupleValue(0, [lon.min(), lat.min(), 0])
newv.InsertTupleValue(1, [lon.max(), lat.max(), 0])
vcs2vtk.projectArray(newv, projection,
[gridGenDict['xm'], gridGenDict['xM'],
gridGenDict['ym'], gridGenDict['yM']])
dimMin = [0, 0, 0]
dimMax = [0, 0, 0]
newv.GetTupleValue(0, dimMin)
newv.GetTupleValue(1, dimMax)
maxDimX = max(dimMin[0], dimMax[0])
maxDimY = max(dimMin[1], dimMax[1])
if lat.max() != 0.0:
scale = abs((maxDimY / lat.max()))
if lon.max() != 0.0:
temp = abs((maxDimX / lon.max()))
if scale < temp:
scale = temp
else:
scale = 1.0
returned["vtk_backend_grid"] = grid
returned["vtk_backend_geo"] = geo
missingMapper = vcs2vtk.putMaskOnVTKGrid(data1, grid, None, False,
deep=False)
# None/False are for color and cellData
# (sent to vcs2vtk.putMaskOnVTKGrid)
returned["vtk_backend_missing_mapper"] = (missingMapper, None, False)
w = vcs2vtk.generateVectorArray(data1, data2, grid)
grid.GetPointData().AddArray(w)
# Vector attempt
l = self._gm.line
if l is None:
l = "default"
try:
l = vcs.getline(l)
lwidth = l.width[0] # noqa
lcolor = l.color[0]
lstyle = l.type[0] # noqa
except:
lstyle = "solid" # noqa
lwidth = 1. # noqa
lcolor = 0
if self._gm.linewidth is not None:
lwidth = self._gm.linewidth # noqa
if self._gm.linecolor is not None:
lcolor = self._gm.linecolor
arrow = vtk.vtkGlyphSource2D()
arrow.SetGlyphTypeToArrow()
arrow.SetOutputPointsPrecision(vtk.vtkAlgorithm.DOUBLE_PRECISION)
arrow.FilledOff()
glyphFilter = vtk.vtkGlyph2D()
glyphFilter.SetInputData(grid)
glyphFilter.SetInputArrayToProcess(1, 0, 0, 0, "vectors")
glyphFilter.SetSourceConnection(arrow.GetOutputPort())
glyphFilter.SetVectorModeToUseVector()
# Rotate arrows to match vector data:
glyphFilter.OrientOn()
# Scale to vector magnitude:
glyphFilter.SetScaleModeToScaleByVector()
glyphFilter.SetScaleFactor(scale * 2.0 * self._gm.scale)
# These are some unfortunately named methods. It does *not* clamp the
# scale range to [min, max], but rather remaps the range
# [min, max] --> [0, 1].
glyphFilter.ClampingOn()
glyphFilter.SetRange(0.01, 1.0)
mapper = vtk.vtkPolyDataMapper()
glyphFilter.Update()
data = glyphFilter.GetOutput()
mapper.SetInputData(data)
act = vtk.vtkActor()
act.SetMapper(mapper)
cmap = self.getColorMap()
r, g, b, a = cmap.index[lcolor]
act.GetProperty().SetColor(r / 100., g / 100., b / 100.)
x1, x2, y1, y2 = vcs.utils.getworldcoordinates(self._gm, data1.getAxis(-1),
data1.getAxis(-2))
if geo is None:
wc = [x1, x2, y1, y2]
else:
xrange = list(act.GetXRange())
yrange = list(act.GetYRange())
wc = [xrange[0], xrange[1], yrange[0], yrange[1]]
act = vcs2vtk.doWrap(act, wc, self._dataWrapModulo)
self._context().fitToViewport(act, [tmpl.data.x1, tmpl.data.x2,
tmpl.data.y1, tmpl.data.y2],
wc=wc,
priority=tmpl.data.priority,
create_renderer=True)
returned.update(self._context().renderTemplate(tmpl, data1,
self._gm, taxis, zaxis))
if self._context().canvas._continents is None:
continents = False
if continents:
self._context().plotContinents(x1, x2, y1, y2, projection,
self._dataWrapModulo, tmpl)
returned["vtk_backend_actors"] = [[act, [x1, x2, y1, y2]]]
returned["vtk_backend_glyphfilters"] = [glyphFilter]
returned["vtk_backend_luts"] = [[None, None]]
return returned
def plot(self, data1, data2, tmpl, gm, grid, transform):
"""Overrides baseclass implementation."""
# Preserve time and z axis for plotting these inof in rendertemplate
geo = None # to make flake8 happy
returned = {}
taxis = data1.getTime()
if data1.ndim > 2:
zaxis = data1.getAxis(-3)
else:
zaxis = None
# Ok get3 only the last 2 dims
data1 = self._context().trimData2D(data1)
data2 = self._context().trimData2D(data2)
gridGenDict = vcs2vtk.genGridOnPoints(data1, gm, deep=False, grid=grid,
geo=transform)
for k in ['vtk_backend_grid', 'xm', 'xM', 'ym', 'yM', 'continents',
'wrap', 'geo']:
exec("%s = gridGenDict['%s']" % (k, k))
grid = gridGenDict['vtk_backend_grid']
self._dataWrapModulo = gridGenDict['wrap']
returned["vtk_backend_grid"] = grid
returned["vtk_backend_geo"] = geo
missingMapper = vcs2vtk.putMaskOnVTKGrid(data1, grid, None, False,
deep=False)
# None/False are for color and cellData
# (sent to vcs2vtk.putMaskOnVTKGrid)
returned["vtk_backend_missing_mapper"] = (missingMapper, None, False)
w = vcs2vtk.generateVectorArray(data1, data2, grid)
grid.GetPointData().AddArray(w)
# Vector attempt
l = gm.line
if l is None:
l = "default"
try:
l = vcs.getline(l)
lwidth = l.width[0] # noqa
lcolor = l.color[0]
lstyle = l.type[0] # noqa
except:
lstyle = "solid" # noqa
lwidth = 1. # noqa
lcolor = 0
if gm.linewidth is not None:
lwidth = gm.linewidth # noqa
if gm.linecolor is not None:
lcolor = gm.linecolor
arrow = vtk.vtkGlyphSource2D()
arrow.SetGlyphTypeToArrow()
arrow.FilledOff()
glyphFilter = vtk.vtkGlyph2D()
glyphFilter.SetInputData(grid)
glyphFilter.SetInputArrayToProcess(1, 0, 0, 0, "vectors")
glyphFilter.SetSourceConnection(arrow.GetOutputPort())
glyphFilter.SetVectorModeToUseVector()
# Rotate arrows to match vector data:
glyphFilter.OrientOn()
# Scale to vector magnitude:
glyphFilter.SetScaleModeToScaleByVector()
glyphFilter.SetScaleFactor(2. * gm.scale)
# These are some unfortunately named methods. It does *not* clamp the
# scale range to [min, max], but rather remaps the range
# [min, max] --> [0, 1].
glyphFilter.ClampingOn()
glyphFilter.SetRange(0.01, 1.0)
mapper = vtk.vtkPolyDataMapper()
mapper.SetInputConnection(glyphFilter.GetOutputPort())
act = vtk.vtkActor()
act.SetMapper(mapper)
cmap = self._context().canvas.getcolormapname()
cmap = vcs.elements["colormap"][cmap]
r, g, b = cmap.index[lcolor]
act.GetProperty().SetColor(r / 100., g / 100., b / 100.)
x1, x2, y1, y2 = vcs.utils.getworldcoordinates(gm, data1.getAxis(-1),
data1.getAxis(-2))
act = vcs2vtk.doWrap(act, [x1, x2, y1, y2], self._dataWrapModulo)
self._context().fitToViewport(act, [tmpl.data.x1, tmpl.data.x2,
tmpl.data.y1, tmpl.data.y2],
[x1, x2, y1, y2],
priority=tmpl.data.priority,
create_renderer=True)
returned.update(
self._context().renderTemplate(tmpl, data1, gm, taxis, zaxis))
if self._context().canvas._continents is None:
continents = False
if continents:
projection = vcs.elements["projection"][gm.projection]
self._context().plotContinents(x1, x2, y1, y2, projection,
self._dataWrapModulo, tmpl)
returned["vtk_backend_actors"] = [[act, [x1, x2, y1, y2]]]
returned["vtk_backend_glyphfilters"] = [glyphFilter]
returned["vtk_backend_luts"] = [[None, None]]
return returned
def plot(self, data1, data2, tmpl, gm, grid, transform):
"""Overrides baseclass implementation."""
#Preserve time and z axis for plotting these inof in rendertemplate
returned = {}
taxis = data1.getTime()
if data1.ndim > 2:
zaxis = data1.getAxis(-3)
else:
zaxis = None
data1 = self._context.trimData2D(data1) # Ok get3 only the last 2 dims
data2 = self._context.trimData2D(data2)
gridGenDict = vcs2vtk.genGridOnPoints(data1, gm, deep=False, grid=grid,
geo=transform)
for k in ['vtk_backend_grid', 'xm', 'xM', 'ym', 'yM', 'continents',
'wrap', 'geo']:
exec("%s = gridGenDict['%s']" % (k, k))
grid = gridGenDict['vtk_backend_grid']
returned["vtk_backend_grid"] = grid
returned["vtk_backend_geo"] = geo
missingMapper = vcs2vtk.putMaskOnVTKGrid(data1, grid, None, False,
deep=False)
# None/False are for color and cellData
# (sent to vcs2vtk.putMaskOnVTKGrid)
returned["vtk_backend_missing_mapper"] = (missingMapper, None, False)
w = vcs2vtk.generateVectorArray(data1, data2, grid)
grid.GetPointData().AddArray(w)
## Vector attempt
l = gm.line
if l is None:
l = "default"
try:
l = vcs.getline(l)
lwidth = l.width[0]
lcolor = l.color[0]
lstyle = l.type[0]
except:
lstyle = "solid"
lwidth = 1.
lcolor = 0
if gm.linewidth is not None:
lwidth = gm.linewidth
if gm.linecolor is not None:
lcolor = gm.linecolor
grid = vcs2vtk.stripGrid(grid)
arrow = vtk.vtkGlyphSource2D()
arrow.SetGlyphTypeToArrow()
arrow.FilledOff()
glyphFilter = vtk.vtkGlyph2D()
glyphFilter.SetInputData(grid)
glyphFilter.SetInputArrayToProcess(1, 0, 0, 0, "vectors")
glyphFilter.SetSourceConnection(arrow.GetOutputPort())
glyphFilter.SetVectorModeToUseVector()
# Rotate arrows to match vector data:
glyphFilter.OrientOn()
# Scale to vector magnitude:
glyphFilter.SetScaleModeToScaleByVector()
glyphFilter.SetScaleFactor(2. * gm.scale)
# These are some unfortunately named methods. It does *not* clamp the
# scale range to [min, max], but rather remaps the range
# [min, max] --> [0, 1].
glyphFilter.ClampingOn()
glyphFilter.SetRange(0.01, 1.0)
mapper = vtk.vtkPolyDataMapper()
mapper.SetInputConnection(glyphFilter.GetOutputPort())
act = vtk.vtkActor()
act.SetMapper(mapper)
cmap = vcs.elements["colormap"][self._context.canvas.getcolormapname()]
r, g, b = cmap.index[lcolor]
act.GetProperty().SetColor(r / 100.,g / 100.,b / 100.)
x1, x2, y1, y2 = vcs.utils.getworldcoordinates(gm, data1.getAxis(-1),
data1.getAxis(-2))
act = vcs2vtk.doWrap(act, [x1,x2,y1,y2], wrap)
ren = self._context.fitToViewport(act, [tmpl.data.x1, tmpl.data.x2,
tmpl.data.y1, tmpl.data.y2],
[x1, x2, y1, y2],
priority=tmpl.data.priority)
returned.update(
self._context.renderTemplate(tmpl, data1, gm, taxis, zaxis))
if self._context.canvas._continents is None:
continents = False
if continents:
projection = vcs.elements["projection"][gm.projection]
self._context.plotContinents(x1, x2, y1, y2, projection, wrap, tmpl)
returned["vtk_backend_actors"] = [[act, [x1,x2,y1,y2]],]
returned["vtk_backend_glyphfilters"] = [glyphFilter,]
returned["vtk_backend_luts"] = [[None, None],]
return returned