vispy 0.15.0__cp313-cp313-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of vispy might be problematic. Click here for more details.
- vispy/__init__.py +33 -0
- vispy/app/__init__.py +15 -0
- vispy/app/_default_app.py +76 -0
- vispy/app/_detect_eventloop.py +148 -0
- vispy/app/application.py +263 -0
- vispy/app/backends/__init__.py +52 -0
- vispy/app/backends/_egl.py +264 -0
- vispy/app/backends/_glfw.py +513 -0
- vispy/app/backends/_jupyter_rfb.py +278 -0
- vispy/app/backends/_offscreen_util.py +121 -0
- vispy/app/backends/_osmesa.py +235 -0
- vispy/app/backends/_pyglet.py +451 -0
- vispy/app/backends/_pyqt4.py +36 -0
- vispy/app/backends/_pyqt5.py +36 -0
- vispy/app/backends/_pyqt6.py +40 -0
- vispy/app/backends/_pyside.py +37 -0
- vispy/app/backends/_pyside2.py +52 -0
- vispy/app/backends/_pyside6.py +53 -0
- vispy/app/backends/_qt.py +1003 -0
- vispy/app/backends/_sdl2.py +444 -0
- vispy/app/backends/_template.py +244 -0
- vispy/app/backends/_test.py +8 -0
- vispy/app/backends/_tk.py +800 -0
- vispy/app/backends/_wx.py +476 -0
- vispy/app/backends/tests/__init__.py +0 -0
- vispy/app/backends/tests/test_offscreen_util.py +52 -0
- vispy/app/backends/tests/test_rfb.py +77 -0
- vispy/app/base.py +294 -0
- vispy/app/canvas.py +828 -0
- vispy/app/qt.py +92 -0
- vispy/app/tests/__init__.py +0 -0
- vispy/app/tests/qt-designer.ui +58 -0
- vispy/app/tests/test_app.py +442 -0
- vispy/app/tests/test_backends.py +164 -0
- vispy/app/tests/test_canvas.py +122 -0
- vispy/app/tests/test_context.py +92 -0
- vispy/app/tests/test_qt.py +47 -0
- vispy/app/tests/test_simultaneous.py +134 -0
- vispy/app/timer.py +174 -0
- vispy/color/__init__.py +17 -0
- vispy/color/_color_dict.py +193 -0
- vispy/color/color_array.py +447 -0
- vispy/color/color_space.py +181 -0
- vispy/color/colormap.py +1213 -0
- vispy/color/tests/__init__.py +0 -0
- vispy/color/tests/test_color.py +378 -0
- vispy/conftest.py +12 -0
- vispy/ext/__init__.py +0 -0
- vispy/ext/cocoapy.py +1522 -0
- vispy/ext/cubehelix.py +138 -0
- vispy/ext/egl.py +375 -0
- vispy/ext/fontconfig.py +118 -0
- vispy/ext/gdi32plus.py +206 -0
- vispy/ext/osmesa.py +105 -0
- vispy/geometry/__init__.py +23 -0
- vispy/geometry/_triangulation_debugger.py +171 -0
- vispy/geometry/calculations.py +162 -0
- vispy/geometry/curves.py +399 -0
- vispy/geometry/generation.py +643 -0
- vispy/geometry/isocurve.py +175 -0
- vispy/geometry/isosurface.py +465 -0
- vispy/geometry/meshdata.py +700 -0
- vispy/geometry/normals.py +78 -0
- vispy/geometry/parametric.py +56 -0
- vispy/geometry/polygon.py +137 -0
- vispy/geometry/rect.py +210 -0
- vispy/geometry/tests/__init__.py +0 -0
- vispy/geometry/tests/test_calculations.py +23 -0
- vispy/geometry/tests/test_generation.py +56 -0
- vispy/geometry/tests/test_meshdata.py +106 -0
- vispy/geometry/tests/test_triangulation.py +594 -0
- vispy/geometry/torusknot.py +142 -0
- vispy/geometry/triangulation.py +876 -0
- vispy/gloo/__init__.py +56 -0
- vispy/gloo/buffer.py +505 -0
- vispy/gloo/context.py +272 -0
- vispy/gloo/framebuffer.py +257 -0
- vispy/gloo/gl/__init__.py +234 -0
- vispy/gloo/gl/_constants.py +332 -0
- vispy/gloo/gl/_es2.py +986 -0
- vispy/gloo/gl/_gl2.py +1365 -0
- vispy/gloo/gl/_proxy.py +499 -0
- vispy/gloo/gl/_pyopengl2.py +362 -0
- vispy/gloo/gl/dummy.py +24 -0
- vispy/gloo/gl/es2.py +62 -0
- vispy/gloo/gl/gl2.py +98 -0
- vispy/gloo/gl/glplus.py +168 -0
- vispy/gloo/gl/pyopengl2.py +97 -0
- vispy/gloo/gl/tests/__init__.py +0 -0
- vispy/gloo/gl/tests/test_basics.py +282 -0
- vispy/gloo/gl/tests/test_functionality.py +568 -0
- vispy/gloo/gl/tests/test_names.py +246 -0
- vispy/gloo/gl/tests/test_use.py +71 -0
- vispy/gloo/glir.py +1824 -0
- vispy/gloo/globject.py +101 -0
- vispy/gloo/preprocessor.py +67 -0
- vispy/gloo/program.py +543 -0
- vispy/gloo/tests/__init__.py +0 -0
- vispy/gloo/tests/test_buffer.py +558 -0
- vispy/gloo/tests/test_context.py +119 -0
- vispy/gloo/tests/test_framebuffer.py +195 -0
- vispy/gloo/tests/test_glir.py +307 -0
- vispy/gloo/tests/test_globject.py +35 -0
- vispy/gloo/tests/test_program.py +302 -0
- vispy/gloo/tests/test_texture.py +732 -0
- vispy/gloo/tests/test_use_gloo.py +187 -0
- vispy/gloo/tests/test_util.py +60 -0
- vispy/gloo/tests/test_wrappers.py +261 -0
- vispy/gloo/texture.py +1046 -0
- vispy/gloo/util.py +129 -0
- vispy/gloo/wrappers.py +762 -0
- vispy/glsl/__init__.py +42 -0
- vispy/glsl/antialias/antialias.glsl +7 -0
- vispy/glsl/antialias/cap-butt.glsl +31 -0
- vispy/glsl/antialias/cap-round.glsl +29 -0
- vispy/glsl/antialias/cap-square.glsl +30 -0
- vispy/glsl/antialias/cap-triangle-in.glsl +30 -0
- vispy/glsl/antialias/cap-triangle-out.glsl +30 -0
- vispy/glsl/antialias/cap.glsl +67 -0
- vispy/glsl/antialias/caps.glsl +67 -0
- vispy/glsl/antialias/filled.glsl +50 -0
- vispy/glsl/antialias/outline.glsl +40 -0
- vispy/glsl/antialias/stroke.glsl +43 -0
- vispy/glsl/arrowheads/angle.glsl +99 -0
- vispy/glsl/arrowheads/arrowheads.frag +60 -0
- vispy/glsl/arrowheads/arrowheads.glsl +12 -0
- vispy/glsl/arrowheads/arrowheads.vert +83 -0
- vispy/glsl/arrowheads/curved.glsl +48 -0
- vispy/glsl/arrowheads/inhibitor.glsl +26 -0
- vispy/glsl/arrowheads/stealth.glsl +46 -0
- vispy/glsl/arrowheads/triangle.glsl +97 -0
- vispy/glsl/arrowheads/util.glsl +13 -0
- vispy/glsl/arrows/angle-30.glsl +12 -0
- vispy/glsl/arrows/angle-60.glsl +12 -0
- vispy/glsl/arrows/angle-90.glsl +12 -0
- vispy/glsl/arrows/arrow.frag +39 -0
- vispy/glsl/arrows/arrow.vert +49 -0
- vispy/glsl/arrows/arrows.glsl +17 -0
- vispy/glsl/arrows/common.glsl +187 -0
- vispy/glsl/arrows/curved.glsl +63 -0
- vispy/glsl/arrows/stealth.glsl +50 -0
- vispy/glsl/arrows/triangle-30.glsl +12 -0
- vispy/glsl/arrows/triangle-60.glsl +12 -0
- vispy/glsl/arrows/triangle-90.glsl +12 -0
- vispy/glsl/arrows/util.glsl +98 -0
- vispy/glsl/build_spatial_filters.py +660 -0
- vispy/glsl/collections/agg-fast-path.frag +20 -0
- vispy/glsl/collections/agg-fast-path.vert +78 -0
- vispy/glsl/collections/agg-glyph.frag +60 -0
- vispy/glsl/collections/agg-glyph.vert +33 -0
- vispy/glsl/collections/agg-marker.frag +35 -0
- vispy/glsl/collections/agg-marker.vert +48 -0
- vispy/glsl/collections/agg-path.frag +55 -0
- vispy/glsl/collections/agg-path.vert +166 -0
- vispy/glsl/collections/agg-point.frag +21 -0
- vispy/glsl/collections/agg-point.vert +35 -0
- vispy/glsl/collections/agg-segment.frag +32 -0
- vispy/glsl/collections/agg-segment.vert +75 -0
- vispy/glsl/collections/marker.frag +38 -0
- vispy/glsl/collections/marker.vert +48 -0
- vispy/glsl/collections/raw-path.frag +15 -0
- vispy/glsl/collections/raw-path.vert +24 -0
- vispy/glsl/collections/raw-point.frag +14 -0
- vispy/glsl/collections/raw-point.vert +31 -0
- vispy/glsl/collections/raw-segment.frag +18 -0
- vispy/glsl/collections/raw-segment.vert +26 -0
- vispy/glsl/collections/raw-triangle.frag +13 -0
- vispy/glsl/collections/raw-triangle.vert +26 -0
- vispy/glsl/collections/sdf-glyph-ticks.vert +69 -0
- vispy/glsl/collections/sdf-glyph.frag +80 -0
- vispy/glsl/collections/sdf-glyph.vert +59 -0
- vispy/glsl/collections/tick-labels.vert +71 -0
- vispy/glsl/colormaps/autumn.glsl +20 -0
- vispy/glsl/colormaps/blues.glsl +20 -0
- vispy/glsl/colormaps/color-space.glsl +17 -0
- vispy/glsl/colormaps/colormaps.glsl +24 -0
- vispy/glsl/colormaps/cool.glsl +20 -0
- vispy/glsl/colormaps/fire.glsl +21 -0
- vispy/glsl/colormaps/gray.glsl +20 -0
- vispy/glsl/colormaps/greens.glsl +20 -0
- vispy/glsl/colormaps/hot.glsl +22 -0
- vispy/glsl/colormaps/ice.glsl +20 -0
- vispy/glsl/colormaps/icefire.glsl +23 -0
- vispy/glsl/colormaps/parse.py +40 -0
- vispy/glsl/colormaps/reds.glsl +20 -0
- vispy/glsl/colormaps/spring.glsl +20 -0
- vispy/glsl/colormaps/summer.glsl +20 -0
- vispy/glsl/colormaps/user.glsl +22 -0
- vispy/glsl/colormaps/util.glsl +41 -0
- vispy/glsl/colormaps/wheel.glsl +21 -0
- vispy/glsl/colormaps/winter.glsl +20 -0
- vispy/glsl/lines/agg.frag +320 -0
- vispy/glsl/lines/agg.vert +241 -0
- vispy/glsl/markers/arrow.glsl +12 -0
- vispy/glsl/markers/asterisk.glsl +16 -0
- vispy/glsl/markers/chevron.glsl +14 -0
- vispy/glsl/markers/clover.glsl +20 -0
- vispy/glsl/markers/club.glsl +31 -0
- vispy/glsl/markers/cross.glsl +17 -0
- vispy/glsl/markers/diamond.glsl +12 -0
- vispy/glsl/markers/disc.glsl +9 -0
- vispy/glsl/markers/ellipse.glsl +67 -0
- vispy/glsl/markers/hbar.glsl +9 -0
- vispy/glsl/markers/heart.glsl +15 -0
- vispy/glsl/markers/infinity.glsl +15 -0
- vispy/glsl/markers/marker-sdf.frag +74 -0
- vispy/glsl/markers/marker-sdf.vert +41 -0
- vispy/glsl/markers/marker.frag +36 -0
- vispy/glsl/markers/marker.vert +46 -0
- vispy/glsl/markers/markers.glsl +24 -0
- vispy/glsl/markers/pin.glsl +18 -0
- vispy/glsl/markers/ring.glsl +11 -0
- vispy/glsl/markers/spade.glsl +28 -0
- vispy/glsl/markers/square.glsl +10 -0
- vispy/glsl/markers/tag.glsl +11 -0
- vispy/glsl/markers/triangle.glsl +14 -0
- vispy/glsl/markers/vbar.glsl +9 -0
- vispy/glsl/math/circle-through-2-points.glsl +30 -0
- vispy/glsl/math/constants.glsl +48 -0
- vispy/glsl/math/double.glsl +114 -0
- vispy/glsl/math/functions.glsl +20 -0
- vispy/glsl/math/point-to-line-distance.glsl +31 -0
- vispy/glsl/math/point-to-line-projection.glsl +29 -0
- vispy/glsl/math/signed-line-distance.glsl +27 -0
- vispy/glsl/math/signed-segment-distance.glsl +30 -0
- vispy/glsl/misc/regular-grid.frag +244 -0
- vispy/glsl/misc/spatial-filters.frag +1407 -0
- vispy/glsl/misc/viewport-NDC.glsl +20 -0
- vispy/glsl/transforms/azimuthal-equal-area.glsl +32 -0
- vispy/glsl/transforms/azimuthal-equidistant.glsl +38 -0
- vispy/glsl/transforms/hammer.glsl +44 -0
- vispy/glsl/transforms/identity.glsl +6 -0
- vispy/glsl/transforms/identity_forward.glsl +23 -0
- vispy/glsl/transforms/identity_inverse.glsl +23 -0
- vispy/glsl/transforms/linear-scale.glsl +127 -0
- vispy/glsl/transforms/log-scale.glsl +126 -0
- vispy/glsl/transforms/mercator-transverse-forward.glsl +40 -0
- vispy/glsl/transforms/mercator-transverse-inverse.glsl +40 -0
- vispy/glsl/transforms/panzoom.glsl +10 -0
- vispy/glsl/transforms/polar.glsl +41 -0
- vispy/glsl/transforms/position.glsl +44 -0
- vispy/glsl/transforms/power-scale.glsl +139 -0
- vispy/glsl/transforms/projection.glsl +7 -0
- vispy/glsl/transforms/pvm.glsl +13 -0
- vispy/glsl/transforms/rotate.glsl +45 -0
- vispy/glsl/transforms/trackball.glsl +15 -0
- vispy/glsl/transforms/translate.glsl +35 -0
- vispy/glsl/transforms/transverse_mercator.glsl +38 -0
- vispy/glsl/transforms/viewport-clipping.glsl +14 -0
- vispy/glsl/transforms/viewport-transform.glsl +16 -0
- vispy/glsl/transforms/viewport.glsl +50 -0
- vispy/glsl/transforms/x.glsl +24 -0
- vispy/glsl/transforms/y.glsl +19 -0
- vispy/glsl/transforms/z.glsl +14 -0
- vispy/io/__init__.py +20 -0
- vispy/io/_data/spatial-filters.npy +0 -0
- vispy/io/datasets.py +94 -0
- vispy/io/image.py +231 -0
- vispy/io/mesh.py +122 -0
- vispy/io/stl.py +167 -0
- vispy/io/tests/__init__.py +0 -0
- vispy/io/tests/test_image.py +47 -0
- vispy/io/tests/test_io.py +121 -0
- vispy/io/wavefront.py +350 -0
- vispy/plot/__init__.py +36 -0
- vispy/plot/fig.py +58 -0
- vispy/plot/plotwidget.py +522 -0
- vispy/plot/tests/__init__.py +0 -0
- vispy/plot/tests/test_plot.py +46 -0
- vispy/scene/__init__.py +43 -0
- vispy/scene/cameras/__init__.py +27 -0
- vispy/scene/cameras/_base.py +38 -0
- vispy/scene/cameras/arcball.py +105 -0
- vispy/scene/cameras/base_camera.py +551 -0
- vispy/scene/cameras/fly.py +474 -0
- vispy/scene/cameras/magnify.py +163 -0
- vispy/scene/cameras/panzoom.py +311 -0
- vispy/scene/cameras/perspective.py +338 -0
- vispy/scene/cameras/tests/__init__.py +0 -0
- vispy/scene/cameras/tests/test_cameras.py +27 -0
- vispy/scene/cameras/tests/test_link.py +53 -0
- vispy/scene/cameras/tests/test_perspective.py +122 -0
- vispy/scene/cameras/turntable.py +183 -0
- vispy/scene/canvas.py +639 -0
- vispy/scene/events.py +85 -0
- vispy/scene/node.py +644 -0
- vispy/scene/subscene.py +20 -0
- vispy/scene/tests/__init__.py +0 -0
- vispy/scene/tests/test_canvas.py +119 -0
- vispy/scene/tests/test_node.py +142 -0
- vispy/scene/tests/test_visuals.py +141 -0
- vispy/scene/visuals.py +276 -0
- vispy/scene/widgets/__init__.py +18 -0
- vispy/scene/widgets/anchor.py +25 -0
- vispy/scene/widgets/axis.py +88 -0
- vispy/scene/widgets/colorbar.py +176 -0
- vispy/scene/widgets/console.py +351 -0
- vispy/scene/widgets/grid.py +509 -0
- vispy/scene/widgets/label.py +50 -0
- vispy/scene/widgets/tests/__init__.py +0 -0
- vispy/scene/widgets/tests/test_colorbar.py +47 -0
- vispy/scene/widgets/viewbox.py +199 -0
- vispy/scene/widgets/widget.py +478 -0
- vispy/testing/__init__.py +51 -0
- vispy/testing/_runners.py +448 -0
- vispy/testing/_testing.py +416 -0
- vispy/testing/image_tester.py +494 -0
- vispy/testing/rendered_array_tester.py +85 -0
- vispy/testing/tests/__init__.py +0 -0
- vispy/testing/tests/test_testing.py +20 -0
- vispy/util/__init__.py +32 -0
- vispy/util/bunch.py +15 -0
- vispy/util/check_environment.py +57 -0
- vispy/util/config.py +490 -0
- vispy/util/dpi/__init__.py +19 -0
- vispy/util/dpi/_linux.py +69 -0
- vispy/util/dpi/_quartz.py +26 -0
- vispy/util/dpi/_win32.py +34 -0
- vispy/util/dpi/tests/__init__.py +0 -0
- vispy/util/dpi/tests/test_dpi.py +16 -0
- vispy/util/eq.py +41 -0
- vispy/util/event.py +774 -0
- vispy/util/fetching.py +276 -0
- vispy/util/filter.py +44 -0
- vispy/util/fonts/__init__.py +14 -0
- vispy/util/fonts/_freetype.py +73 -0
- vispy/util/fonts/_quartz.py +192 -0
- vispy/util/fonts/_triage.py +36 -0
- vispy/util/fonts/_vispy_fonts.py +20 -0
- vispy/util/fonts/_win32.py +105 -0
- vispy/util/fonts/data/OpenSans-Bold.ttf +0 -0
- vispy/util/fonts/data/OpenSans-BoldItalic.ttf +0 -0
- vispy/util/fonts/data/OpenSans-Italic.ttf +0 -0
- vispy/util/fonts/data/OpenSans-Regular.ttf +0 -0
- vispy/util/fonts/tests/__init__.py +0 -0
- vispy/util/fonts/tests/test_font.py +45 -0
- vispy/util/fourier.py +69 -0
- vispy/util/frozen.py +25 -0
- vispy/util/gallery_scraper.py +268 -0
- vispy/util/keys.py +91 -0
- vispy/util/logs.py +358 -0
- vispy/util/osmesa_gl.py +17 -0
- vispy/util/profiler.py +135 -0
- vispy/util/ptime.py +16 -0
- vispy/util/quaternion.py +229 -0
- vispy/util/svg/__init__.py +18 -0
- vispy/util/svg/base.py +20 -0
- vispy/util/svg/color.py +219 -0
- vispy/util/svg/element.py +51 -0
- vispy/util/svg/geometry.py +478 -0
- vispy/util/svg/group.py +66 -0
- vispy/util/svg/length.py +81 -0
- vispy/util/svg/number.py +25 -0
- vispy/util/svg/path.py +332 -0
- vispy/util/svg/shapes.py +57 -0
- vispy/util/svg/style.py +59 -0
- vispy/util/svg/svg.py +40 -0
- vispy/util/svg/transform.py +223 -0
- vispy/util/svg/transformable.py +28 -0
- vispy/util/svg/viewport.py +73 -0
- vispy/util/tests/__init__.py +0 -0
- vispy/util/tests/test_config.py +58 -0
- vispy/util/tests/test_docstring_parameters.py +123 -0
- vispy/util/tests/test_emitter_group.py +262 -0
- vispy/util/tests/test_event_emitter.py +743 -0
- vispy/util/tests/test_fourier.py +35 -0
- vispy/util/tests/test_gallery_scraper.py +112 -0
- vispy/util/tests/test_import.py +127 -0
- vispy/util/tests/test_key.py +22 -0
- vispy/util/tests/test_logging.py +45 -0
- vispy/util/tests/test_run.py +14 -0
- vispy/util/tests/test_transforms.py +42 -0
- vispy/util/tests/test_vispy.py +48 -0
- vispy/util/transforms.py +201 -0
- vispy/util/wrappers.py +155 -0
- vispy/version.py +21 -0
- vispy/visuals/__init__.py +50 -0
- vispy/visuals/_scalable_textures.py +487 -0
- vispy/visuals/axis.py +678 -0
- vispy/visuals/border.py +208 -0
- vispy/visuals/box.py +79 -0
- vispy/visuals/collections/__init__.py +30 -0
- vispy/visuals/collections/agg_fast_path_collection.py +219 -0
- vispy/visuals/collections/agg_path_collection.py +197 -0
- vispy/visuals/collections/agg_point_collection.py +52 -0
- vispy/visuals/collections/agg_segment_collection.py +142 -0
- vispy/visuals/collections/array_list.py +401 -0
- vispy/visuals/collections/base_collection.py +482 -0
- vispy/visuals/collections/collection.py +253 -0
- vispy/visuals/collections/path_collection.py +23 -0
- vispy/visuals/collections/point_collection.py +19 -0
- vispy/visuals/collections/polygon_collection.py +25 -0
- vispy/visuals/collections/raw_path_collection.py +119 -0
- vispy/visuals/collections/raw_point_collection.py +113 -0
- vispy/visuals/collections/raw_polygon_collection.py +77 -0
- vispy/visuals/collections/raw_segment_collection.py +112 -0
- vispy/visuals/collections/raw_triangle_collection.py +78 -0
- vispy/visuals/collections/segment_collection.py +19 -0
- vispy/visuals/collections/triangle_collection.py +16 -0
- vispy/visuals/collections/util.py +168 -0
- vispy/visuals/colorbar.py +699 -0
- vispy/visuals/cube.py +41 -0
- vispy/visuals/ellipse.py +162 -0
- vispy/visuals/filters/__init__.py +10 -0
- vispy/visuals/filters/base_filter.py +242 -0
- vispy/visuals/filters/clipper.py +60 -0
- vispy/visuals/filters/clipping_planes.py +122 -0
- vispy/visuals/filters/color.py +181 -0
- vispy/visuals/filters/markers.py +28 -0
- vispy/visuals/filters/mesh.py +801 -0
- vispy/visuals/filters/picking.py +60 -0
- vispy/visuals/filters/tests/__init__.py +3 -0
- vispy/visuals/filters/tests/test_primitive_picking_filters.py +70 -0
- vispy/visuals/filters/tests/test_wireframe_filter.py +16 -0
- vispy/visuals/glsl/__init__.py +1 -0
- vispy/visuals/glsl/antialiasing.py +133 -0
- vispy/visuals/glsl/color.py +63 -0
- vispy/visuals/graphs/__init__.py +1 -0
- vispy/visuals/graphs/graph.py +240 -0
- vispy/visuals/graphs/layouts/__init__.py +55 -0
- vispy/visuals/graphs/layouts/circular.py +49 -0
- vispy/visuals/graphs/layouts/force_directed.py +211 -0
- vispy/visuals/graphs/layouts/networkx_layout.py +87 -0
- vispy/visuals/graphs/layouts/random.py +52 -0
- vispy/visuals/graphs/tests/__init__.py +1 -0
- vispy/visuals/graphs/tests/test_layouts.py +139 -0
- vispy/visuals/graphs/tests/test_networkx_layout.py +47 -0
- vispy/visuals/graphs/util.py +120 -0
- vispy/visuals/gridlines.py +161 -0
- vispy/visuals/gridmesh.py +98 -0
- vispy/visuals/histogram.py +58 -0
- vispy/visuals/image.py +701 -0
- vispy/visuals/image_complex.py +130 -0
- vispy/visuals/infinite_line.py +199 -0
- vispy/visuals/instanced_mesh.py +152 -0
- vispy/visuals/isocurve.py +213 -0
- vispy/visuals/isoline.py +241 -0
- vispy/visuals/isosurface.py +113 -0
- vispy/visuals/line/__init__.py +6 -0
- vispy/visuals/line/arrow.py +289 -0
- vispy/visuals/line/dash_atlas.py +90 -0
- vispy/visuals/line/line.py +545 -0
- vispy/visuals/line_plot.py +135 -0
- vispy/visuals/linear_region.py +199 -0
- vispy/visuals/markers.py +819 -0
- vispy/visuals/mesh.py +373 -0
- vispy/visuals/mesh_normals.py +159 -0
- vispy/visuals/plane.py +54 -0
- vispy/visuals/polygon.py +145 -0
- vispy/visuals/rectangle.py +196 -0
- vispy/visuals/regular_polygon.py +56 -0
- vispy/visuals/scrolling_lines.py +197 -0
- vispy/visuals/shaders/__init__.py +17 -0
- vispy/visuals/shaders/compiler.py +206 -0
- vispy/visuals/shaders/expression.py +99 -0
- vispy/visuals/shaders/function.py +788 -0
- vispy/visuals/shaders/multiprogram.py +145 -0
- vispy/visuals/shaders/parsing.py +140 -0
- vispy/visuals/shaders/program.py +161 -0
- vispy/visuals/shaders/shader_object.py +162 -0
- vispy/visuals/shaders/tests/__init__.py +0 -0
- vispy/visuals/shaders/tests/test_function.py +486 -0
- vispy/visuals/shaders/tests/test_multiprogram.py +78 -0
- vispy/visuals/shaders/tests/test_parsing.py +57 -0
- vispy/visuals/shaders/variable.py +272 -0
- vispy/visuals/spectrogram.py +169 -0
- vispy/visuals/sphere.py +80 -0
- vispy/visuals/surface_plot.py +192 -0
- vispy/visuals/tests/__init__.py +0 -0
- vispy/visuals/tests/test_arrows.py +109 -0
- vispy/visuals/tests/test_axis.py +120 -0
- vispy/visuals/tests/test_collections.py +15 -0
- vispy/visuals/tests/test_colorbar.py +179 -0
- vispy/visuals/tests/test_colormap.py +97 -0
- vispy/visuals/tests/test_ellipse.py +122 -0
- vispy/visuals/tests/test_gridlines.py +30 -0
- vispy/visuals/tests/test_histogram.py +24 -0
- vispy/visuals/tests/test_image.py +392 -0
- vispy/visuals/tests/test_image_complex.py +36 -0
- vispy/visuals/tests/test_infinite_line.py +53 -0
- vispy/visuals/tests/test_instanced_mesh.py +50 -0
- vispy/visuals/tests/test_isosurface.py +22 -0
- vispy/visuals/tests/test_linear_region.py +152 -0
- vispy/visuals/tests/test_markers.py +54 -0
- vispy/visuals/tests/test_mesh.py +261 -0
- vispy/visuals/tests/test_mesh_normals.py +218 -0
- vispy/visuals/tests/test_polygon.py +112 -0
- vispy/visuals/tests/test_rectangle.py +163 -0
- vispy/visuals/tests/test_regular_polygon.py +111 -0
- vispy/visuals/tests/test_scalable_textures.py +196 -0
- vispy/visuals/tests/test_sdf.py +73 -0
- vispy/visuals/tests/test_spectrogram.py +42 -0
- vispy/visuals/tests/test_surface_plot.py +57 -0
- vispy/visuals/tests/test_text.py +95 -0
- vispy/visuals/tests/test_volume.py +542 -0
- vispy/visuals/tests/test_windbarb.py +33 -0
- vispy/visuals/text/__init__.py +7 -0
- vispy/visuals/text/_sdf_cpu.cpython-313-x86_64-linux-gnu.so +0 -0
- vispy/visuals/text/_sdf_cpu.pyx +112 -0
- vispy/visuals/text/_sdf_gpu.py +316 -0
- vispy/visuals/text/text.py +675 -0
- vispy/visuals/transforms/__init__.py +34 -0
- vispy/visuals/transforms/_util.py +191 -0
- vispy/visuals/transforms/base_transform.py +233 -0
- vispy/visuals/transforms/chain.py +300 -0
- vispy/visuals/transforms/interactive.py +98 -0
- vispy/visuals/transforms/linear.py +564 -0
- vispy/visuals/transforms/nonlinear.py +398 -0
- vispy/visuals/transforms/tests/__init__.py +0 -0
- vispy/visuals/transforms/tests/test_transforms.py +243 -0
- vispy/visuals/transforms/transform_system.py +339 -0
- vispy/visuals/tube.py +173 -0
- vispy/visuals/visual.py +923 -0
- vispy/visuals/volume.py +1366 -0
- vispy/visuals/windbarb.py +291 -0
- vispy/visuals/xyz_axis.py +34 -0
- vispy-0.15.0.dist-info/METADATA +243 -0
- vispy-0.15.0.dist-info/RECORD +521 -0
- vispy-0.15.0.dist-info/WHEEL +6 -0
- vispy-0.15.0.dist-info/licenses/LICENSE.txt +36 -0
- vispy-0.15.0.dist-info/top_level.txt +1 -0
vispy/visuals/image.py
ADDED
|
@@ -0,0 +1,701 @@
|
|
|
1
|
+
# -*- coding: utf-8 -*-
|
|
2
|
+
# Copyright (c) Vispy Development Team. All Rights Reserved.
|
|
3
|
+
# Distributed under the (new) BSD License. See LICENSE.txt for more info.
|
|
4
|
+
"""Primitive 2D image visual class."""
|
|
5
|
+
|
|
6
|
+
from __future__ import division
|
|
7
|
+
|
|
8
|
+
import warnings
|
|
9
|
+
|
|
10
|
+
import numpy as np
|
|
11
|
+
|
|
12
|
+
from ..gloo import Texture2D, VertexBuffer
|
|
13
|
+
from ..color import get_colormap
|
|
14
|
+
from .shaders import Function, FunctionChain
|
|
15
|
+
from .transforms import NullTransform
|
|
16
|
+
from .visual import Visual
|
|
17
|
+
from ..io import load_spatial_filters
|
|
18
|
+
from ._scalable_textures import CPUScaledTexture2D, GPUScaledTexture2D
|
|
19
|
+
from ..util import np_copy_if_needed
|
|
20
|
+
|
|
21
|
+
|
|
22
|
+
_VERTEX_SHADER = """
|
|
23
|
+
uniform int method; // 0=subdivide, 1=impostor
|
|
24
|
+
attribute vec2 a_position;
|
|
25
|
+
attribute vec2 a_texcoord;
|
|
26
|
+
varying vec2 v_texcoord;
|
|
27
|
+
|
|
28
|
+
void main() {
|
|
29
|
+
v_texcoord = a_texcoord;
|
|
30
|
+
gl_Position = $transform(vec4(a_position, 0., 1.));
|
|
31
|
+
}
|
|
32
|
+
"""
|
|
33
|
+
|
|
34
|
+
_FRAGMENT_SHADER = """
|
|
35
|
+
uniform vec2 image_size;
|
|
36
|
+
uniform int method; // 0=subdivide, 1=impostor
|
|
37
|
+
uniform sampler2D u_texture;
|
|
38
|
+
varying vec2 v_texcoord;
|
|
39
|
+
|
|
40
|
+
vec4 map_local_to_tex(vec4 x) {
|
|
41
|
+
// Cast ray from 3D viewport to surface of image
|
|
42
|
+
// (if $transform does not affect z values, then this
|
|
43
|
+
// can be optimized as simply $transform.map(x) )
|
|
44
|
+
vec4 p1 = $transform(x);
|
|
45
|
+
vec4 p2 = $transform(x + vec4(0, 0, 0.5, 0));
|
|
46
|
+
p1 /= p1.w;
|
|
47
|
+
p2 /= p2.w;
|
|
48
|
+
vec4 d = p2 - p1;
|
|
49
|
+
float f = p2.z / d.z;
|
|
50
|
+
vec4 p3 = p2 - d * f;
|
|
51
|
+
|
|
52
|
+
// finally map local to texture coords
|
|
53
|
+
return vec4(p3.xy / image_size, 0, 1);
|
|
54
|
+
}
|
|
55
|
+
|
|
56
|
+
|
|
57
|
+
void main()
|
|
58
|
+
{
|
|
59
|
+
vec2 texcoord;
|
|
60
|
+
if( method == 0 ) {
|
|
61
|
+
texcoord = v_texcoord;
|
|
62
|
+
}
|
|
63
|
+
else {
|
|
64
|
+
// vertex shader outputs clip coordinates;
|
|
65
|
+
// fragment shader maps to texture coordinates
|
|
66
|
+
texcoord = map_local_to_tex(vec4(v_texcoord, 0, 1)).xy;
|
|
67
|
+
}
|
|
68
|
+
|
|
69
|
+
gl_FragColor = $color_transform($get_data(texcoord));
|
|
70
|
+
}
|
|
71
|
+
""" # noqa
|
|
72
|
+
|
|
73
|
+
_INTERPOLATION_TEMPLATE = """
|
|
74
|
+
#include "misc/spatial-filters.frag"
|
|
75
|
+
vec4 texture_lookup_filtered(vec2 texcoord) {
|
|
76
|
+
if(texcoord.x < 0.0 || texcoord.x > 1.0 ||
|
|
77
|
+
texcoord.y < 0.0 || texcoord.y > 1.0) {
|
|
78
|
+
discard;
|
|
79
|
+
}
|
|
80
|
+
return %s($texture, $shape, texcoord);
|
|
81
|
+
}"""
|
|
82
|
+
|
|
83
|
+
_TEXTURE_LOOKUP = """
|
|
84
|
+
vec4 texture_lookup(vec2 texcoord) {
|
|
85
|
+
if(texcoord.x < 0.0 || texcoord.x > 1.0 ||
|
|
86
|
+
texcoord.y < 0.0 || texcoord.y > 1.0) {
|
|
87
|
+
discard;
|
|
88
|
+
}
|
|
89
|
+
return texture2D($texture, texcoord);
|
|
90
|
+
}"""
|
|
91
|
+
|
|
92
|
+
_APPLY_CLIM_FLOAT = """
|
|
93
|
+
float apply_clim(float data) {
|
|
94
|
+
// pass through NaN values to get handled by the colormap
|
|
95
|
+
if (!(data <= 0.0 || 0.0 <= data)) return data;
|
|
96
|
+
|
|
97
|
+
data = clamp(data, min($clim.x, $clim.y), max($clim.x, $clim.y));
|
|
98
|
+
data = (data - $clim.x) / ($clim.y - $clim.x);
|
|
99
|
+
return data;
|
|
100
|
+
}"""
|
|
101
|
+
|
|
102
|
+
_APPLY_CLIM = """
|
|
103
|
+
vec4 apply_clim(vec4 color) {
|
|
104
|
+
// Handle NaN values (clamp them to the minimum value)
|
|
105
|
+
// http://stackoverflow.com/questions/11810158/how-to-deal-with-nan-or-inf-in-opengl-es-2-0-shaders
|
|
106
|
+
color.r = !(color.r <= 0.0 || 0.0 <= color.r) ? min($clim.x, $clim.y) : color.r;
|
|
107
|
+
color.g = !(color.g <= 0.0 || 0.0 <= color.g) ? min($clim.x, $clim.y) : color.g;
|
|
108
|
+
color.b = !(color.b <= 0.0 || 0.0 <= color.b) ? min($clim.x, $clim.y) : color.b;
|
|
109
|
+
color.a = !(color.a <= 0.0 || 0.0 <= color.a) ? 0 : color.a;
|
|
110
|
+
color.rgb = clamp(color.rgb, min($clim.x, $clim.y), max($clim.x, $clim.y));
|
|
111
|
+
color.rgb = (color.rgb - $clim.x) / ($clim.y - $clim.x);
|
|
112
|
+
return max(color, 0.0);
|
|
113
|
+
}
|
|
114
|
+
"""
|
|
115
|
+
|
|
116
|
+
_APPLY_GAMMA_FLOAT = """
|
|
117
|
+
float apply_gamma(float data) {
|
|
118
|
+
// pass through NaN values to get handled by the colormap
|
|
119
|
+
if (!(data <= 0.0 || 0.0 <= data)) return data;
|
|
120
|
+
|
|
121
|
+
return pow(data, $gamma);
|
|
122
|
+
}"""
|
|
123
|
+
|
|
124
|
+
_APPLY_GAMMA = """
|
|
125
|
+
vec4 apply_gamma(vec4 color) {
|
|
126
|
+
color.rgb = pow(color.rgb, vec3($gamma));
|
|
127
|
+
return color;
|
|
128
|
+
}
|
|
129
|
+
"""
|
|
130
|
+
|
|
131
|
+
_NULL_COLOR_TRANSFORM = 'vec4 pass(vec4 color) { return color; }'
|
|
132
|
+
|
|
133
|
+
_C2L_RED = 'float color_to_luminance(vec4 color) { return color.r; }'
|
|
134
|
+
|
|
135
|
+
_CUSTOM_FILTER = """
|
|
136
|
+
vec4 texture_lookup(vec2 texcoord) {
|
|
137
|
+
// based on https://gist.github.com/kingbedjed/373c8811efcf1b3a155d29a13c1e5b61
|
|
138
|
+
vec2 tex_pixel = 1 / $shape;
|
|
139
|
+
vec2 kernel_pixel = 1 / $kernel_shape;
|
|
140
|
+
vec2 sampling_corner = texcoord - ($kernel_shape / 2 * tex_pixel);
|
|
141
|
+
|
|
142
|
+
// loop over kernel pixels
|
|
143
|
+
vec2 kernel_pos, tex_pos;
|
|
144
|
+
vec4 color = vec4(0);
|
|
145
|
+
float weight;
|
|
146
|
+
|
|
147
|
+
// offset 0.5 to sample center of pixels
|
|
148
|
+
for (float i = 0.5; i < $kernel_shape.x; i++) {
|
|
149
|
+
for (float j = 0.5; j < $kernel_shape.y; j++) {
|
|
150
|
+
kernel_pos = vec2(i, j) * kernel_pixel;
|
|
151
|
+
tex_pos = sampling_corner + vec2(i, j) * tex_pixel;
|
|
152
|
+
// TODO: allow other edge effects, like mirror or wrap
|
|
153
|
+
if (tex_pos.x >= 0 && tex_pos.y >= 0 && tex_pos.x <= 1 && tex_pos.y <= 1) {
|
|
154
|
+
weight = texture2D($kernel, kernel_pos).r;
|
|
155
|
+
// make sure to clamp or we sample outside
|
|
156
|
+
color += texture2D($texture, clamp(tex_pos, 0, 1)) * weight;
|
|
157
|
+
}
|
|
158
|
+
}
|
|
159
|
+
}
|
|
160
|
+
|
|
161
|
+
return color;
|
|
162
|
+
}
|
|
163
|
+
"""
|
|
164
|
+
|
|
165
|
+
|
|
166
|
+
class ImageVisual(Visual):
|
|
167
|
+
"""Visual subclass displaying an image.
|
|
168
|
+
|
|
169
|
+
Parameters
|
|
170
|
+
----------
|
|
171
|
+
data : ndarray
|
|
172
|
+
ImageVisual data. Can be shape (M, N), (M, N, 3), or (M, N, 4).
|
|
173
|
+
If floating point data is provided and contains NaNs, they will
|
|
174
|
+
be made transparent (discarded) for the single band data case when
|
|
175
|
+
scaling is done on the GPU (see ``texture_format``). On the CPU,
|
|
176
|
+
single band NaNs are mapped to 0 as they are sent to the GPU which
|
|
177
|
+
result in them using the lowest ``clim`` value in the GPU.
|
|
178
|
+
For RGB data, NaNs will be mapped to the lowest ``clim`` value.
|
|
179
|
+
If the Alpha band is NaN it will be mapped to 0 (transparent).
|
|
180
|
+
Note that NaN handling is not required by some OpenGL implementations
|
|
181
|
+
and NaNs may be treated differently on some systems (ex. as 0s).
|
|
182
|
+
method : str
|
|
183
|
+
Selects method of rendering image in case of non-linear transforms.
|
|
184
|
+
Each method produces similar results, but may trade efficiency
|
|
185
|
+
and accuracy. If the transform is linear, this parameter is ignored
|
|
186
|
+
and a single quad is drawn around the area of the image.
|
|
187
|
+
|
|
188
|
+
* 'auto': Automatically select 'impostor' if the image is drawn
|
|
189
|
+
with a nonlinear transform; otherwise select 'subdivide'.
|
|
190
|
+
* 'subdivide': ImageVisual is represented as a grid of triangles
|
|
191
|
+
with texture coordinates linearly mapped.
|
|
192
|
+
* 'impostor': ImageVisual is represented as a quad covering the
|
|
193
|
+
entire view, with texture coordinates determined by the
|
|
194
|
+
transform. This produces the best transformation results, but may
|
|
195
|
+
be slow.
|
|
196
|
+
|
|
197
|
+
grid: tuple (rows, cols)
|
|
198
|
+
If method='subdivide', this tuple determines the number of rows and
|
|
199
|
+
columns in the image grid.
|
|
200
|
+
cmap : str | ColorMap
|
|
201
|
+
Colormap to use for luminance images.
|
|
202
|
+
clim : str | tuple
|
|
203
|
+
Limits to use for the colormap. I.e. the values that map to black and white
|
|
204
|
+
in a gray colormap. Can be 'auto' to auto-set bounds to
|
|
205
|
+
the min and max of the data. If not given or None, 'auto' is used.
|
|
206
|
+
gamma : float
|
|
207
|
+
Gamma to use during colormap lookup. Final color will be cmap(val**gamma).
|
|
208
|
+
by default: 1.
|
|
209
|
+
interpolation : str
|
|
210
|
+
Selects method of texture interpolation. Makes use of the two hardware
|
|
211
|
+
interpolation methods and the available interpolation methods defined
|
|
212
|
+
in vispy/gloo/glsl/misc/spatial_filters.frag
|
|
213
|
+
|
|
214
|
+
* 'nearest': Default, uses 'nearest' with Texture interpolation.
|
|
215
|
+
* 'linear': uses 'linear' with Texture interpolation.
|
|
216
|
+
* 'hanning', 'hamming', 'hermite', 'kaiser', 'quadric', 'cubic',
|
|
217
|
+
'catrom', 'mitchell', 'spline16', 'spline36', 'gaussian',
|
|
218
|
+
'bessel', 'sinc', 'lanczos', 'blackman'
|
|
219
|
+
* 'custom': uses the sampling kernel provided through 'custom_kernel'.
|
|
220
|
+
texture_format: numpy.dtype | str | None
|
|
221
|
+
How to store data on the GPU. OpenGL allows for many different storage
|
|
222
|
+
formats and schemes for the low-level texture data stored in the GPU.
|
|
223
|
+
Most common is unsigned integers or floating point numbers.
|
|
224
|
+
Unsigned integers are the most widely supported while other formats
|
|
225
|
+
may not be supported on older versions of OpenGL or with older GPUs.
|
|
226
|
+
Default value is ``None`` which means data will be scaled on the
|
|
227
|
+
CPU and the result stored in the GPU as an unsigned integer. If a
|
|
228
|
+
numpy dtype object, an internal texture format will be chosen to
|
|
229
|
+
support that dtype and data will *not* be scaled on the CPU. Not all
|
|
230
|
+
dtypes are supported. If a string, then
|
|
231
|
+
it must be one of the OpenGL internalformat strings described in the
|
|
232
|
+
table on this page: https://www.khronos.org/registry/OpenGL-Refpages/gl4/html/glTexImage2D.xhtml
|
|
233
|
+
The name should have `GL_` removed and be lowercase (ex.
|
|
234
|
+
`GL_R32F` becomes ``'r32f'``). Lastly, this can also be the string
|
|
235
|
+
``'auto'`` which will use the data type of the provided image data
|
|
236
|
+
to determine the internalformat of the texture.
|
|
237
|
+
When this is specified (not ``None``) data is scaled on the
|
|
238
|
+
GPU which allows for faster color limit changes. Additionally, when
|
|
239
|
+
32-bit float data is provided it won't be copied before being
|
|
240
|
+
transferred to the GPU.
|
|
241
|
+
custom_kernel: numpy.ndarray
|
|
242
|
+
Kernel used for texture sampling when interpolation is set to 'custom'.
|
|
243
|
+
**kwargs : dict
|
|
244
|
+
Keyword arguments to pass to `Visual`.
|
|
245
|
+
|
|
246
|
+
Notes
|
|
247
|
+
-----
|
|
248
|
+
The colormap functionality through ``cmap`` and ``clim`` are only used
|
|
249
|
+
if the data are 2D.
|
|
250
|
+
"""
|
|
251
|
+
|
|
252
|
+
_shaders = {
|
|
253
|
+
'vertex': _VERTEX_SHADER,
|
|
254
|
+
'fragment': _FRAGMENT_SHADER,
|
|
255
|
+
}
|
|
256
|
+
|
|
257
|
+
_func_templates = {
|
|
258
|
+
'texture_lookup_interpolated': _INTERPOLATION_TEMPLATE,
|
|
259
|
+
'texture_lookup_custom': _CUSTOM_FILTER,
|
|
260
|
+
'texture_lookup': _TEXTURE_LOOKUP,
|
|
261
|
+
'clim_float': _APPLY_CLIM_FLOAT,
|
|
262
|
+
'clim': _APPLY_CLIM,
|
|
263
|
+
'gamma_float': _APPLY_GAMMA_FLOAT,
|
|
264
|
+
'gamma': _APPLY_GAMMA,
|
|
265
|
+
'null_color_transform': _NULL_COLOR_TRANSFORM,
|
|
266
|
+
'red_to_luminance': _C2L_RED,
|
|
267
|
+
}
|
|
268
|
+
|
|
269
|
+
def __init__(self, data=None, method='auto', grid=(1, 1),
|
|
270
|
+
cmap='viridis', clim='auto', gamma=1.0,
|
|
271
|
+
interpolation='nearest', texture_format=None,
|
|
272
|
+
custom_kernel=np.ones((1, 1)), **kwargs):
|
|
273
|
+
"""Initialize image properties, texture storage, and interpolation methods."""
|
|
274
|
+
self._data = None
|
|
275
|
+
|
|
276
|
+
# load 'float packed rgba8' interpolation kernel
|
|
277
|
+
# to load float interpolation kernel use
|
|
278
|
+
# `load_spatial_filters(packed=False)`
|
|
279
|
+
kernel, interpolation_names = load_spatial_filters()
|
|
280
|
+
|
|
281
|
+
self._kerneltex = Texture2D(kernel, interpolation='nearest')
|
|
282
|
+
# The unpacking can be debugged by changing "spatial-filters.frag"
|
|
283
|
+
# to have the "unpack" function just return the .r component. That
|
|
284
|
+
# combined with using the below as the _kerneltex allows debugging
|
|
285
|
+
# of the pipeline
|
|
286
|
+
# self._kerneltex = Texture2D(kernel, interpolation='linear',
|
|
287
|
+
# internalformat='r32f')
|
|
288
|
+
|
|
289
|
+
interpolation_names, interpolation_fun = self._init_interpolation(
|
|
290
|
+
interpolation_names)
|
|
291
|
+
self._interpolation_names = interpolation_names
|
|
292
|
+
self._interpolation_fun = interpolation_fun
|
|
293
|
+
self._interpolation = interpolation
|
|
294
|
+
if self._interpolation not in self._interpolation_names:
|
|
295
|
+
raise ValueError("interpolation must be one of %s" %
|
|
296
|
+
', '.join(self._interpolation_names))
|
|
297
|
+
|
|
298
|
+
self._method = method
|
|
299
|
+
self._grid = grid
|
|
300
|
+
self._need_texture_upload = True
|
|
301
|
+
self._need_vertex_update = True
|
|
302
|
+
self._need_colortransform_update = True
|
|
303
|
+
self._need_interpolation_update = True
|
|
304
|
+
self._texture = self._init_texture(data, texture_format)
|
|
305
|
+
self._subdiv_position = VertexBuffer()
|
|
306
|
+
self._subdiv_texcoord = VertexBuffer()
|
|
307
|
+
|
|
308
|
+
# impostor quad covers entire viewport
|
|
309
|
+
vertices = np.array([[-1, -1], [1, -1], [1, 1],
|
|
310
|
+
[-1, -1], [1, 1], [-1, 1]],
|
|
311
|
+
dtype=np.float32)
|
|
312
|
+
self._impostor_coords = VertexBuffer(vertices)
|
|
313
|
+
self._null_tr = NullTransform()
|
|
314
|
+
|
|
315
|
+
self._init_view(self)
|
|
316
|
+
|
|
317
|
+
Visual.__init__(self, vcode=self._shaders['vertex'], fcode=self._shaders['fragment'])
|
|
318
|
+
self.set_gl_state('translucent', cull_face=False)
|
|
319
|
+
self._draw_mode = 'triangles'
|
|
320
|
+
|
|
321
|
+
# define _data_lookup_fn as None, will be setup in
|
|
322
|
+
# self._build_interpolation()
|
|
323
|
+
self._data_lookup_fn = None
|
|
324
|
+
|
|
325
|
+
self.clim = clim or "auto" # None -> "auto"
|
|
326
|
+
self.cmap = cmap
|
|
327
|
+
self.gamma = gamma
|
|
328
|
+
self.custom_kernel = custom_kernel
|
|
329
|
+
|
|
330
|
+
if data is not None:
|
|
331
|
+
self.set_data(data)
|
|
332
|
+
self.freeze()
|
|
333
|
+
|
|
334
|
+
def _init_interpolation(self, interpolation_names):
|
|
335
|
+
# create interpolation shader functions for available interpolations
|
|
336
|
+
fun = [Function(self._func_templates['texture_lookup_interpolated'] % (n + '2D'))
|
|
337
|
+
for n in interpolation_names]
|
|
338
|
+
interpolation_names = [n.lower() for n in interpolation_names]
|
|
339
|
+
|
|
340
|
+
# add custom filter
|
|
341
|
+
fun.append(Function(self._func_templates['texture_lookup_custom']))
|
|
342
|
+
interpolation_names.append('custom')
|
|
343
|
+
|
|
344
|
+
interpolation_fun = dict(zip(interpolation_names, fun))
|
|
345
|
+
interpolation_names = tuple(sorted(interpolation_names))
|
|
346
|
+
|
|
347
|
+
# overwrite "nearest" and "linear" spatial-filters
|
|
348
|
+
# with "hardware" interpolation _data_lookup_fn
|
|
349
|
+
hardware_lookup = Function(self._func_templates['texture_lookup'])
|
|
350
|
+
interpolation_fun['nearest'] = hardware_lookup
|
|
351
|
+
interpolation_fun['linear'] = hardware_lookup
|
|
352
|
+
# alias bilinear to linear and bicubic to cubic (but deprecate)
|
|
353
|
+
interpolation_names = interpolation_names + ('bilinear', 'bicubic')
|
|
354
|
+
return interpolation_names, interpolation_fun
|
|
355
|
+
|
|
356
|
+
def _init_texture(self, data, texture_format, **texture_kwargs):
|
|
357
|
+
if self._interpolation == 'linear':
|
|
358
|
+
texture_interpolation = 'linear'
|
|
359
|
+
else:
|
|
360
|
+
texture_interpolation = 'nearest'
|
|
361
|
+
|
|
362
|
+
if texture_format is None:
|
|
363
|
+
tex = CPUScaledTexture2D(
|
|
364
|
+
data, interpolation=texture_interpolation,
|
|
365
|
+
**texture_kwargs
|
|
366
|
+
)
|
|
367
|
+
else:
|
|
368
|
+
tex = GPUScaledTexture2D(
|
|
369
|
+
data, internalformat=texture_format,
|
|
370
|
+
interpolation=texture_interpolation,
|
|
371
|
+
**texture_kwargs
|
|
372
|
+
)
|
|
373
|
+
return tex
|
|
374
|
+
|
|
375
|
+
def set_data(self, image, copy=False):
|
|
376
|
+
"""Set the image data.
|
|
377
|
+
|
|
378
|
+
Parameters
|
|
379
|
+
----------
|
|
380
|
+
image : array-like
|
|
381
|
+
The image data.
|
|
382
|
+
texture_format : str or None
|
|
383
|
+
|
|
384
|
+
"""
|
|
385
|
+
data = np.array(image, copy=copy or np_copy_if_needed)
|
|
386
|
+
if np.iscomplexobj(data):
|
|
387
|
+
raise TypeError(
|
|
388
|
+
"Complex data types not supported. Please use 'ComplexImage' instead"
|
|
389
|
+
)
|
|
390
|
+
# can the texture handle this data?
|
|
391
|
+
self._texture.check_data_format(data)
|
|
392
|
+
if self._data is None or self._data.shape[:2] != data.shape[:2]:
|
|
393
|
+
# Only rebuild if the size of the image changed
|
|
394
|
+
self._need_vertex_update = True
|
|
395
|
+
self._data = data
|
|
396
|
+
self._need_texture_upload = True
|
|
397
|
+
|
|
398
|
+
def view(self):
|
|
399
|
+
"""Get the :class:`vispy.visuals.visual.VisualView` for this visual."""
|
|
400
|
+
v = Visual.view(self)
|
|
401
|
+
self._init_view(v)
|
|
402
|
+
return v
|
|
403
|
+
|
|
404
|
+
def _init_view(self, view):
|
|
405
|
+
# Store some extra variables per-view
|
|
406
|
+
view._need_method_update = True
|
|
407
|
+
view._method_used = None
|
|
408
|
+
|
|
409
|
+
@property
|
|
410
|
+
def clim(self):
|
|
411
|
+
"""Get color limits used when rendering the image (cmin, cmax)."""
|
|
412
|
+
return self._texture.clim
|
|
413
|
+
|
|
414
|
+
@clim.setter
|
|
415
|
+
def clim(self, clim):
|
|
416
|
+
if self._texture.set_clim(clim):
|
|
417
|
+
self._need_texture_upload = True
|
|
418
|
+
self._update_colortransform_clim()
|
|
419
|
+
self.update()
|
|
420
|
+
|
|
421
|
+
def _update_colortransform_clim(self):
|
|
422
|
+
if self._need_colortransform_update:
|
|
423
|
+
# we are going to rebuild anyway so just do it later
|
|
424
|
+
return
|
|
425
|
+
try:
|
|
426
|
+
norm_clims = self._texture.clim_normalized
|
|
427
|
+
except RuntimeError:
|
|
428
|
+
return
|
|
429
|
+
else:
|
|
430
|
+
# shortcut so we don't have to rebuild the whole color transform
|
|
431
|
+
self.shared_program.frag['color_transform'][1]['clim'] = norm_clims
|
|
432
|
+
|
|
433
|
+
@property
|
|
434
|
+
def cmap(self):
|
|
435
|
+
"""Get the colormap object applied to luminance (single band) data."""
|
|
436
|
+
return self._cmap
|
|
437
|
+
|
|
438
|
+
@cmap.setter
|
|
439
|
+
def cmap(self, cmap):
|
|
440
|
+
self._cmap = get_colormap(cmap)
|
|
441
|
+
self._need_colortransform_update = True
|
|
442
|
+
self.update()
|
|
443
|
+
|
|
444
|
+
@property
|
|
445
|
+
def gamma(self):
|
|
446
|
+
"""Get the gamma used when rendering the image."""
|
|
447
|
+
return self._gamma
|
|
448
|
+
|
|
449
|
+
@gamma.setter
|
|
450
|
+
def gamma(self, value):
|
|
451
|
+
"""Set gamma used when rendering the image."""
|
|
452
|
+
if value <= 0:
|
|
453
|
+
raise ValueError("gamma must be > 0")
|
|
454
|
+
self._gamma = float(value)
|
|
455
|
+
# shortcut so we don't have to rebuild the color transform
|
|
456
|
+
if not self._need_colortransform_update:
|
|
457
|
+
self.shared_program.frag['color_transform'][2]['gamma'] = self._gamma
|
|
458
|
+
self.update()
|
|
459
|
+
|
|
460
|
+
@property
|
|
461
|
+
def bad_color(self):
|
|
462
|
+
"""Color used to render NaN values."""
|
|
463
|
+
return self._cmap.get_bad_color()
|
|
464
|
+
|
|
465
|
+
@bad_color.setter
|
|
466
|
+
def bad_color(self, color):
|
|
467
|
+
self._cmap.set_bad_color(color)
|
|
468
|
+
self._need_colortransform_update = True
|
|
469
|
+
self.update()
|
|
470
|
+
|
|
471
|
+
@property
|
|
472
|
+
def method(self):
|
|
473
|
+
"""Get rendering method name."""
|
|
474
|
+
return self._method
|
|
475
|
+
|
|
476
|
+
@method.setter
|
|
477
|
+
def method(self, m):
|
|
478
|
+
if self._method != m:
|
|
479
|
+
self._method = m
|
|
480
|
+
self._need_vertex_update = True
|
|
481
|
+
self.update()
|
|
482
|
+
|
|
483
|
+
@property
|
|
484
|
+
def size(self):
|
|
485
|
+
"""Get size of the image (width, height)."""
|
|
486
|
+
return self._data.shape[:2][::-1]
|
|
487
|
+
|
|
488
|
+
@property
|
|
489
|
+
def interpolation(self):
|
|
490
|
+
"""Get interpolation algorithm name."""
|
|
491
|
+
return self._interpolation
|
|
492
|
+
|
|
493
|
+
@interpolation.setter
|
|
494
|
+
def interpolation(self, i):
|
|
495
|
+
if i not in self._interpolation_names:
|
|
496
|
+
raise ValueError("interpolation must be one of %s" %
|
|
497
|
+
', '.join(self._interpolation_names))
|
|
498
|
+
if self._interpolation != i:
|
|
499
|
+
self._interpolation = i
|
|
500
|
+
self._need_interpolation_update = True
|
|
501
|
+
self.update()
|
|
502
|
+
|
|
503
|
+
@property
|
|
504
|
+
def interpolation_functions(self):
|
|
505
|
+
"""Get names of possible interpolation methods."""
|
|
506
|
+
return self._interpolation_names
|
|
507
|
+
|
|
508
|
+
@property
|
|
509
|
+
def custom_kernel(self):
|
|
510
|
+
"""Kernel used by 'custom' interpolation for texture sampling"""
|
|
511
|
+
return self._custom_kernel
|
|
512
|
+
|
|
513
|
+
@custom_kernel.setter
|
|
514
|
+
def custom_kernel(self, value):
|
|
515
|
+
value = np.asarray(value, dtype=np.float32)
|
|
516
|
+
if value.ndim != 2:
|
|
517
|
+
raise ValueError(f'kernel must have 2 dimensions; got {value.ndim}')
|
|
518
|
+
self._custom_kernel = value
|
|
519
|
+
self._custom_kerneltex = Texture2D(value, interpolation='nearest', internalformat='r32f')
|
|
520
|
+
if self._data_lookup_fn is not None and 'kernel' in self._data_lookup_fn:
|
|
521
|
+
self._data_lookup_fn['kernel'] = self._custom_kerneltex
|
|
522
|
+
self._data_lookup_fn['kernel_shape'] = value.shape[::-1]
|
|
523
|
+
self.update()
|
|
524
|
+
|
|
525
|
+
# The interpolation code could be transferred to a dedicated filter
|
|
526
|
+
# function in visuals/filters as discussed in #1051
|
|
527
|
+
def _build_interpolation(self):
|
|
528
|
+
"""Rebuild the _data_lookup_fn for different interpolations."""
|
|
529
|
+
interpolation = self._interpolation
|
|
530
|
+
# alias bilinear to linear
|
|
531
|
+
if interpolation == 'bilinear':
|
|
532
|
+
warnings.warn(
|
|
533
|
+
"'bilinear' interpolation is Deprecated. Use 'linear' instead.",
|
|
534
|
+
DeprecationWarning,
|
|
535
|
+
stacklevel=2,
|
|
536
|
+
)
|
|
537
|
+
interpolation = 'linear'
|
|
538
|
+
# alias bicubic to cubic
|
|
539
|
+
if interpolation == 'bicubic':
|
|
540
|
+
warnings.warn(
|
|
541
|
+
"'bicubic' interpolation is Deprecated. Use 'cubic' instead.",
|
|
542
|
+
DeprecationWarning,
|
|
543
|
+
stacklevel=2,
|
|
544
|
+
)
|
|
545
|
+
interpolation = 'cubic'
|
|
546
|
+
self._data_lookup_fn = self._interpolation_fun[interpolation]
|
|
547
|
+
self.shared_program.frag['get_data'] = self._data_lookup_fn
|
|
548
|
+
|
|
549
|
+
# only 'linear' and 'custom' use 'linear' texture interpolation
|
|
550
|
+
if interpolation in ('linear', 'custom'):
|
|
551
|
+
texture_interpolation = 'linear'
|
|
552
|
+
else:
|
|
553
|
+
texture_interpolation = 'nearest'
|
|
554
|
+
|
|
555
|
+
# 'nearest' (and also 'linear') doesn't use spatial_filters.frag
|
|
556
|
+
# so u_kernel and shape setting is skipped
|
|
557
|
+
if interpolation not in ('nearest', 'linear'):
|
|
558
|
+
self._data_lookup_fn['shape'] = self._data.shape[:2][::-1]
|
|
559
|
+
if interpolation == 'custom':
|
|
560
|
+
self._data_lookup_fn['kernel'] = self._custom_kerneltex
|
|
561
|
+
self._data_lookup_fn['kernel_shape'] = self._custom_kernel.shape[::-1]
|
|
562
|
+
else:
|
|
563
|
+
self.shared_program['u_kernel'] = self._kerneltex
|
|
564
|
+
|
|
565
|
+
if self._texture.interpolation != texture_interpolation:
|
|
566
|
+
self._texture.interpolation = texture_interpolation
|
|
567
|
+
|
|
568
|
+
self._data_lookup_fn['texture'] = self._texture
|
|
569
|
+
|
|
570
|
+
self._need_interpolation_update = False
|
|
571
|
+
|
|
572
|
+
def _build_vertex_data(self):
|
|
573
|
+
"""Rebuild the vertex buffers for the subdivide method."""
|
|
574
|
+
grid = self._grid
|
|
575
|
+
w = 1.0 / grid[1]
|
|
576
|
+
h = 1.0 / grid[0]
|
|
577
|
+
|
|
578
|
+
quad = np.array([[0, 0, 0], [w, 0, 0], [w, h, 0],
|
|
579
|
+
[0, 0, 0], [w, h, 0], [0, h, 0]],
|
|
580
|
+
dtype=np.float32)
|
|
581
|
+
quads = np.empty((grid[1], grid[0], 6, 3), dtype=np.float32)
|
|
582
|
+
quads[:] = quad
|
|
583
|
+
|
|
584
|
+
mgrid = np.mgrid[0.:grid[1], 0.:grid[0]].transpose(1, 2, 0)
|
|
585
|
+
mgrid = mgrid[:, :, np.newaxis, :]
|
|
586
|
+
mgrid[..., 0] *= w
|
|
587
|
+
mgrid[..., 1] *= h
|
|
588
|
+
|
|
589
|
+
quads[..., :2] += mgrid
|
|
590
|
+
tex_coords = quads.reshape(grid[1]*grid[0]*6, 3)
|
|
591
|
+
tex_coords = np.ascontiguousarray(tex_coords[:, :2])
|
|
592
|
+
vertices = tex_coords * self.size
|
|
593
|
+
|
|
594
|
+
self._subdiv_position.set_data(vertices.astype('float32'))
|
|
595
|
+
self._subdiv_texcoord.set_data(tex_coords.astype('float32'))
|
|
596
|
+
self._need_vertex_update = False
|
|
597
|
+
|
|
598
|
+
def _update_method(self, view):
|
|
599
|
+
"""Decide which method to use for *view* and configure it accordingly."""
|
|
600
|
+
method = self._method
|
|
601
|
+
if method == 'auto':
|
|
602
|
+
if view.transforms.get_transform().Linear:
|
|
603
|
+
method = 'subdivide'
|
|
604
|
+
else:
|
|
605
|
+
method = 'impostor'
|
|
606
|
+
view._method_used = method
|
|
607
|
+
|
|
608
|
+
if method == 'subdivide':
|
|
609
|
+
view.view_program['method'] = 0
|
|
610
|
+
view.view_program['a_position'] = self._subdiv_position
|
|
611
|
+
view.view_program['a_texcoord'] = self._subdiv_texcoord
|
|
612
|
+
elif method == 'impostor':
|
|
613
|
+
view.view_program['method'] = 1
|
|
614
|
+
view.view_program['a_position'] = self._impostor_coords
|
|
615
|
+
view.view_program['a_texcoord'] = self._impostor_coords
|
|
616
|
+
else:
|
|
617
|
+
raise ValueError("Unknown image draw method '%s'" % method)
|
|
618
|
+
|
|
619
|
+
self.shared_program['image_size'] = self.size
|
|
620
|
+
view._need_method_update = False
|
|
621
|
+
self._prepare_transforms(view)
|
|
622
|
+
|
|
623
|
+
def _build_texture(self):
|
|
624
|
+
try:
|
|
625
|
+
pre_clims = self._texture.clim_normalized
|
|
626
|
+
except RuntimeError:
|
|
627
|
+
pre_clims = "auto"
|
|
628
|
+
pre_internalformat = self._texture.internalformat
|
|
629
|
+
# copy was already made on `set_data` if requested
|
|
630
|
+
self._texture.scale_and_set_data(self._data, copy=False)
|
|
631
|
+
post_clims = self._texture.clim_normalized
|
|
632
|
+
post_internalformat = self._texture.internalformat
|
|
633
|
+
# color transform needs rebuilding if the internalformat was changed
|
|
634
|
+
# new color limits need to be assigned if the normalized clims changed
|
|
635
|
+
# otherwise, the original color transform should be fine
|
|
636
|
+
new_if = post_internalformat != pre_internalformat
|
|
637
|
+
new_cl = post_clims != pre_clims
|
|
638
|
+
if new_if:
|
|
639
|
+
self._need_colortransform_update = True
|
|
640
|
+
elif new_cl and not self._need_colortransform_update:
|
|
641
|
+
# shortcut so we don't have to rebuild the whole color transform
|
|
642
|
+
self.shared_program.frag['color_transform'][1]['clim'] = self._texture.clim_normalized
|
|
643
|
+
self._need_texture_upload = False
|
|
644
|
+
|
|
645
|
+
def _compute_bounds(self, axis, view):
|
|
646
|
+
if axis > 1:
|
|
647
|
+
return 0, 0
|
|
648
|
+
else:
|
|
649
|
+
return 0, self.size[axis]
|
|
650
|
+
|
|
651
|
+
def _build_color_transform(self):
|
|
652
|
+
if self._data.ndim == 2 or self._data.shape[2] == 1:
|
|
653
|
+
# luminance data
|
|
654
|
+
fclim = Function(self._func_templates['clim_float'])
|
|
655
|
+
fgamma = Function(self._func_templates['gamma_float'])
|
|
656
|
+
# NOTE: red_to_luminance only uses the red component, fancy internalformats
|
|
657
|
+
# may need to use the other components or a different function chain
|
|
658
|
+
fun = FunctionChain(
|
|
659
|
+
None, [Function(self._func_templates['red_to_luminance']), fclim, fgamma, Function(self.cmap.glsl_map)]
|
|
660
|
+
)
|
|
661
|
+
else:
|
|
662
|
+
# RGB/A image data (no colormap)
|
|
663
|
+
fclim = Function(self._func_templates['clim'])
|
|
664
|
+
fgamma = Function(self._func_templates['gamma'])
|
|
665
|
+
fun = FunctionChain(None, [Function(self._func_templates['null_color_transform']), fclim, fgamma])
|
|
666
|
+
fclim['clim'] = self._texture.clim_normalized
|
|
667
|
+
fgamma['gamma'] = self.gamma
|
|
668
|
+
return fun
|
|
669
|
+
|
|
670
|
+
def _prepare_transforms(self, view):
|
|
671
|
+
trs = view.transforms
|
|
672
|
+
prg = view.view_program
|
|
673
|
+
method = view._method_used
|
|
674
|
+
if method == 'subdivide':
|
|
675
|
+
prg.vert['transform'] = trs.get_transform()
|
|
676
|
+
prg.frag['transform'] = self._null_tr
|
|
677
|
+
else:
|
|
678
|
+
prg.vert['transform'] = self._null_tr
|
|
679
|
+
prg.frag['transform'] = trs.get_transform().inverse
|
|
680
|
+
|
|
681
|
+
def _prepare_draw(self, view):
|
|
682
|
+
if self._data is None:
|
|
683
|
+
return False
|
|
684
|
+
|
|
685
|
+
if self._need_interpolation_update:
|
|
686
|
+
self._build_interpolation()
|
|
687
|
+
|
|
688
|
+
if self._need_texture_upload:
|
|
689
|
+
self._build_texture()
|
|
690
|
+
|
|
691
|
+
if self._need_colortransform_update:
|
|
692
|
+
prg = view.view_program
|
|
693
|
+
self.shared_program.frag['color_transform'] = self._build_color_transform()
|
|
694
|
+
self._need_colortransform_update = False
|
|
695
|
+
prg['texture2D_LUT'] = self.cmap.texture_lut()
|
|
696
|
+
|
|
697
|
+
if self._need_vertex_update:
|
|
698
|
+
self._build_vertex_data()
|
|
699
|
+
|
|
700
|
+
if view._need_method_update:
|
|
701
|
+
self._update_method(view)
|