1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
# -*- coding: utf-8 -*-
# Copyright (c) Vispy Development Team. All Rights Reserved.
# Distributed under the (new) BSD License. See LICENSE.txt for more info.
from __future__ import division
from .mesh import MeshVisual
from ..geometry.isosurface import isosurface
from ..color import Color
class IsosurfaceVisual(MeshVisual):
"""Displays an isosurface of a 3D scalar array.
Parameters
----------
data : ndarray | None
3D scalar array.
level: float | None
The level at which the isosurface is constructed from *data*.
vertex_colors : ndarray | None
The vertex colors to use.
face_colors : ndarray | None
The face colors to use.
color : ndarray | None
The color to use.
**kwargs : dict
Keyword arguments to pass to the mesh construction.
"""
def __init__(self, data=None, level=None, vertex_colors=None,
face_colors=None, color=(0.5, 0.5, 1, 1), **kwargs):
self._data = None
self._level = level
self._vertex_colors = vertex_colors
self._face_colors = face_colors
self._color = Color(color)
# We distinguish between recomputing and just changing the visual
# properties - in the latter case we don't recompute the faces.
self._vertices_cache = None
self._faces_cache = None
self._recompute = True
self._update_meshvisual = True
MeshVisual.__init__(self, **kwargs)
if data is not None:
self.set_data(data, vertex_colors=vertex_colors,
face_colors=face_colors, color=color)
@property
def level(self):
"""The threshold at which the isosurface is constructed from the 3D data."""
return self._level
@level.setter
def level(self, level):
self._level = level
self._recompute = True
self.update()
def set_data(self, data=None, vertex_colors=None, face_colors=None,
color=None):
"""Set the scalar array data
Parameters
----------
data : ndarray
A 3D array of scalar values. The isosurface is constructed to show
all locations in the scalar field equal to ``self.level``.
vertex_colors : array-like | None
Colors to use for each vertex.
face_colors : array-like | None
Colors to use for each face.
color : instance of Color
The color to use.
"""
# We only change the internal variables if they are provided
if data is not None:
self._data = data
self._recompute = True
if vertex_colors is not None:
self._vertex_colors = vertex_colors
self._update_meshvisual = True
if face_colors is not None:
self._face_colors = face_colors
self._update_meshvisual = True
if color is not None:
self._color = Color(color)
self._update_meshvisual = True
self.update()
def _prepare_draw(self, view):
if self._data is None or self._level is None:
return False
if self._recompute:
self._vertices_cache, self._faces_cache = isosurface(self._data,
self._level)
self._recompute = False
self._update_meshvisual = True
if self._update_meshvisual:
MeshVisual.set_data(self,
vertices=self._vertices_cache,
faces=self._faces_cache,
vertex_colors=self._vertex_colors,
face_colors=self._face_colors,
color=self._color)
self._update_meshvisual = False
return MeshVisual._prepare_draw(self, view)
|
