vispy 0.15.0__cp313-cp313-manylinux_2_17_aarch64.manylinux2014_aarch64.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.

Files changed (521) hide show
  1. vispy/__init__.py +33 -0
  2. vispy/app/__init__.py +15 -0
  3. vispy/app/_default_app.py +76 -0
  4. vispy/app/_detect_eventloop.py +148 -0
  5. vispy/app/application.py +263 -0
  6. vispy/app/backends/__init__.py +52 -0
  7. vispy/app/backends/_egl.py +264 -0
  8. vispy/app/backends/_glfw.py +513 -0
  9. vispy/app/backends/_jupyter_rfb.py +278 -0
  10. vispy/app/backends/_offscreen_util.py +121 -0
  11. vispy/app/backends/_osmesa.py +235 -0
  12. vispy/app/backends/_pyglet.py +451 -0
  13. vispy/app/backends/_pyqt4.py +36 -0
  14. vispy/app/backends/_pyqt5.py +36 -0
  15. vispy/app/backends/_pyqt6.py +40 -0
  16. vispy/app/backends/_pyside.py +37 -0
  17. vispy/app/backends/_pyside2.py +52 -0
  18. vispy/app/backends/_pyside6.py +53 -0
  19. vispy/app/backends/_qt.py +1003 -0
  20. vispy/app/backends/_sdl2.py +444 -0
  21. vispy/app/backends/_template.py +244 -0
  22. vispy/app/backends/_test.py +8 -0
  23. vispy/app/backends/_tk.py +800 -0
  24. vispy/app/backends/_wx.py +476 -0
  25. vispy/app/backends/tests/__init__.py +0 -0
  26. vispy/app/backends/tests/test_offscreen_util.py +52 -0
  27. vispy/app/backends/tests/test_rfb.py +77 -0
  28. vispy/app/base.py +294 -0
  29. vispy/app/canvas.py +828 -0
  30. vispy/app/qt.py +92 -0
  31. vispy/app/tests/__init__.py +0 -0
  32. vispy/app/tests/qt-designer.ui +58 -0
  33. vispy/app/tests/test_app.py +442 -0
  34. vispy/app/tests/test_backends.py +164 -0
  35. vispy/app/tests/test_canvas.py +122 -0
  36. vispy/app/tests/test_context.py +92 -0
  37. vispy/app/tests/test_qt.py +47 -0
  38. vispy/app/tests/test_simultaneous.py +134 -0
  39. vispy/app/timer.py +174 -0
  40. vispy/color/__init__.py +17 -0
  41. vispy/color/_color_dict.py +193 -0
  42. vispy/color/color_array.py +447 -0
  43. vispy/color/color_space.py +181 -0
  44. vispy/color/colormap.py +1213 -0
  45. vispy/color/tests/__init__.py +0 -0
  46. vispy/color/tests/test_color.py +378 -0
  47. vispy/conftest.py +12 -0
  48. vispy/ext/__init__.py +0 -0
  49. vispy/ext/cocoapy.py +1522 -0
  50. vispy/ext/cubehelix.py +138 -0
  51. vispy/ext/egl.py +375 -0
  52. vispy/ext/fontconfig.py +118 -0
  53. vispy/ext/gdi32plus.py +206 -0
  54. vispy/ext/osmesa.py +105 -0
  55. vispy/geometry/__init__.py +23 -0
  56. vispy/geometry/_triangulation_debugger.py +171 -0
  57. vispy/geometry/calculations.py +162 -0
  58. vispy/geometry/curves.py +399 -0
  59. vispy/geometry/generation.py +643 -0
  60. vispy/geometry/isocurve.py +175 -0
  61. vispy/geometry/isosurface.py +465 -0
  62. vispy/geometry/meshdata.py +700 -0
  63. vispy/geometry/normals.py +78 -0
  64. vispy/geometry/parametric.py +56 -0
  65. vispy/geometry/polygon.py +137 -0
  66. vispy/geometry/rect.py +210 -0
  67. vispy/geometry/tests/__init__.py +0 -0
  68. vispy/geometry/tests/test_calculations.py +23 -0
  69. vispy/geometry/tests/test_generation.py +56 -0
  70. vispy/geometry/tests/test_meshdata.py +106 -0
  71. vispy/geometry/tests/test_triangulation.py +594 -0
  72. vispy/geometry/torusknot.py +142 -0
  73. vispy/geometry/triangulation.py +876 -0
  74. vispy/gloo/__init__.py +56 -0
  75. vispy/gloo/buffer.py +505 -0
  76. vispy/gloo/context.py +272 -0
  77. vispy/gloo/framebuffer.py +257 -0
  78. vispy/gloo/gl/__init__.py +234 -0
  79. vispy/gloo/gl/_constants.py +332 -0
  80. vispy/gloo/gl/_es2.py +986 -0
  81. vispy/gloo/gl/_gl2.py +1365 -0
  82. vispy/gloo/gl/_proxy.py +499 -0
  83. vispy/gloo/gl/_pyopengl2.py +362 -0
  84. vispy/gloo/gl/dummy.py +24 -0
  85. vispy/gloo/gl/es2.py +62 -0
  86. vispy/gloo/gl/gl2.py +98 -0
  87. vispy/gloo/gl/glplus.py +168 -0
  88. vispy/gloo/gl/pyopengl2.py +97 -0
  89. vispy/gloo/gl/tests/__init__.py +0 -0
  90. vispy/gloo/gl/tests/test_basics.py +282 -0
  91. vispy/gloo/gl/tests/test_functionality.py +568 -0
  92. vispy/gloo/gl/tests/test_names.py +246 -0
  93. vispy/gloo/gl/tests/test_use.py +71 -0
  94. vispy/gloo/glir.py +1824 -0
  95. vispy/gloo/globject.py +101 -0
  96. vispy/gloo/preprocessor.py +67 -0
  97. vispy/gloo/program.py +543 -0
  98. vispy/gloo/tests/__init__.py +0 -0
  99. vispy/gloo/tests/test_buffer.py +558 -0
  100. vispy/gloo/tests/test_context.py +119 -0
  101. vispy/gloo/tests/test_framebuffer.py +195 -0
  102. vispy/gloo/tests/test_glir.py +307 -0
  103. vispy/gloo/tests/test_globject.py +35 -0
  104. vispy/gloo/tests/test_program.py +302 -0
  105. vispy/gloo/tests/test_texture.py +732 -0
  106. vispy/gloo/tests/test_use_gloo.py +187 -0
  107. vispy/gloo/tests/test_util.py +60 -0
  108. vispy/gloo/tests/test_wrappers.py +261 -0
  109. vispy/gloo/texture.py +1046 -0
  110. vispy/gloo/util.py +129 -0
  111. vispy/gloo/wrappers.py +762 -0
  112. vispy/glsl/__init__.py +42 -0
  113. vispy/glsl/antialias/antialias.glsl +7 -0
  114. vispy/glsl/antialias/cap-butt.glsl +31 -0
  115. vispy/glsl/antialias/cap-round.glsl +29 -0
  116. vispy/glsl/antialias/cap-square.glsl +30 -0
  117. vispy/glsl/antialias/cap-triangle-in.glsl +30 -0
  118. vispy/glsl/antialias/cap-triangle-out.glsl +30 -0
  119. vispy/glsl/antialias/cap.glsl +67 -0
  120. vispy/glsl/antialias/caps.glsl +67 -0
  121. vispy/glsl/antialias/filled.glsl +50 -0
  122. vispy/glsl/antialias/outline.glsl +40 -0
  123. vispy/glsl/antialias/stroke.glsl +43 -0
  124. vispy/glsl/arrowheads/angle.glsl +99 -0
  125. vispy/glsl/arrowheads/arrowheads.frag +60 -0
  126. vispy/glsl/arrowheads/arrowheads.glsl +12 -0
  127. vispy/glsl/arrowheads/arrowheads.vert +83 -0
  128. vispy/glsl/arrowheads/curved.glsl +48 -0
  129. vispy/glsl/arrowheads/inhibitor.glsl +26 -0
  130. vispy/glsl/arrowheads/stealth.glsl +46 -0
  131. vispy/glsl/arrowheads/triangle.glsl +97 -0
  132. vispy/glsl/arrowheads/util.glsl +13 -0
  133. vispy/glsl/arrows/angle-30.glsl +12 -0
  134. vispy/glsl/arrows/angle-60.glsl +12 -0
  135. vispy/glsl/arrows/angle-90.glsl +12 -0
  136. vispy/glsl/arrows/arrow.frag +39 -0
  137. vispy/glsl/arrows/arrow.vert +49 -0
  138. vispy/glsl/arrows/arrows.glsl +17 -0
  139. vispy/glsl/arrows/common.glsl +187 -0
  140. vispy/glsl/arrows/curved.glsl +63 -0
  141. vispy/glsl/arrows/stealth.glsl +50 -0
  142. vispy/glsl/arrows/triangle-30.glsl +12 -0
  143. vispy/glsl/arrows/triangle-60.glsl +12 -0
  144. vispy/glsl/arrows/triangle-90.glsl +12 -0
  145. vispy/glsl/arrows/util.glsl +98 -0
  146. vispy/glsl/build_spatial_filters.py +660 -0
  147. vispy/glsl/collections/agg-fast-path.frag +20 -0
  148. vispy/glsl/collections/agg-fast-path.vert +78 -0
  149. vispy/glsl/collections/agg-glyph.frag +60 -0
  150. vispy/glsl/collections/agg-glyph.vert +33 -0
  151. vispy/glsl/collections/agg-marker.frag +35 -0
  152. vispy/glsl/collections/agg-marker.vert +48 -0
  153. vispy/glsl/collections/agg-path.frag +55 -0
  154. vispy/glsl/collections/agg-path.vert +166 -0
  155. vispy/glsl/collections/agg-point.frag +21 -0
  156. vispy/glsl/collections/agg-point.vert +35 -0
  157. vispy/glsl/collections/agg-segment.frag +32 -0
  158. vispy/glsl/collections/agg-segment.vert +75 -0
  159. vispy/glsl/collections/marker.frag +38 -0
  160. vispy/glsl/collections/marker.vert +48 -0
  161. vispy/glsl/collections/raw-path.frag +15 -0
  162. vispy/glsl/collections/raw-path.vert +24 -0
  163. vispy/glsl/collections/raw-point.frag +14 -0
  164. vispy/glsl/collections/raw-point.vert +31 -0
  165. vispy/glsl/collections/raw-segment.frag +18 -0
  166. vispy/glsl/collections/raw-segment.vert +26 -0
  167. vispy/glsl/collections/raw-triangle.frag +13 -0
  168. vispy/glsl/collections/raw-triangle.vert +26 -0
  169. vispy/glsl/collections/sdf-glyph-ticks.vert +69 -0
  170. vispy/glsl/collections/sdf-glyph.frag +80 -0
  171. vispy/glsl/collections/sdf-glyph.vert +59 -0
  172. vispy/glsl/collections/tick-labels.vert +71 -0
  173. vispy/glsl/colormaps/autumn.glsl +20 -0
  174. vispy/glsl/colormaps/blues.glsl +20 -0
  175. vispy/glsl/colormaps/color-space.glsl +17 -0
  176. vispy/glsl/colormaps/colormaps.glsl +24 -0
  177. vispy/glsl/colormaps/cool.glsl +20 -0
  178. vispy/glsl/colormaps/fire.glsl +21 -0
  179. vispy/glsl/colormaps/gray.glsl +20 -0
  180. vispy/glsl/colormaps/greens.glsl +20 -0
  181. vispy/glsl/colormaps/hot.glsl +22 -0
  182. vispy/glsl/colormaps/ice.glsl +20 -0
  183. vispy/glsl/colormaps/icefire.glsl +23 -0
  184. vispy/glsl/colormaps/parse.py +40 -0
  185. vispy/glsl/colormaps/reds.glsl +20 -0
  186. vispy/glsl/colormaps/spring.glsl +20 -0
  187. vispy/glsl/colormaps/summer.glsl +20 -0
  188. vispy/glsl/colormaps/user.glsl +22 -0
  189. vispy/glsl/colormaps/util.glsl +41 -0
  190. vispy/glsl/colormaps/wheel.glsl +21 -0
  191. vispy/glsl/colormaps/winter.glsl +20 -0
  192. vispy/glsl/lines/agg.frag +320 -0
  193. vispy/glsl/lines/agg.vert +241 -0
  194. vispy/glsl/markers/arrow.glsl +12 -0
  195. vispy/glsl/markers/asterisk.glsl +16 -0
  196. vispy/glsl/markers/chevron.glsl +14 -0
  197. vispy/glsl/markers/clover.glsl +20 -0
  198. vispy/glsl/markers/club.glsl +31 -0
  199. vispy/glsl/markers/cross.glsl +17 -0
  200. vispy/glsl/markers/diamond.glsl +12 -0
  201. vispy/glsl/markers/disc.glsl +9 -0
  202. vispy/glsl/markers/ellipse.glsl +67 -0
  203. vispy/glsl/markers/hbar.glsl +9 -0
  204. vispy/glsl/markers/heart.glsl +15 -0
  205. vispy/glsl/markers/infinity.glsl +15 -0
  206. vispy/glsl/markers/marker-sdf.frag +74 -0
  207. vispy/glsl/markers/marker-sdf.vert +41 -0
  208. vispy/glsl/markers/marker.frag +36 -0
  209. vispy/glsl/markers/marker.vert +46 -0
  210. vispy/glsl/markers/markers.glsl +24 -0
  211. vispy/glsl/markers/pin.glsl +18 -0
  212. vispy/glsl/markers/ring.glsl +11 -0
  213. vispy/glsl/markers/spade.glsl +28 -0
  214. vispy/glsl/markers/square.glsl +10 -0
  215. vispy/glsl/markers/tag.glsl +11 -0
  216. vispy/glsl/markers/triangle.glsl +14 -0
  217. vispy/glsl/markers/vbar.glsl +9 -0
  218. vispy/glsl/math/circle-through-2-points.glsl +30 -0
  219. vispy/glsl/math/constants.glsl +48 -0
  220. vispy/glsl/math/double.glsl +114 -0
  221. vispy/glsl/math/functions.glsl +20 -0
  222. vispy/glsl/math/point-to-line-distance.glsl +31 -0
  223. vispy/glsl/math/point-to-line-projection.glsl +29 -0
  224. vispy/glsl/math/signed-line-distance.glsl +27 -0
  225. vispy/glsl/math/signed-segment-distance.glsl +30 -0
  226. vispy/glsl/misc/regular-grid.frag +244 -0
  227. vispy/glsl/misc/spatial-filters.frag +1407 -0
  228. vispy/glsl/misc/viewport-NDC.glsl +20 -0
  229. vispy/glsl/transforms/azimuthal-equal-area.glsl +32 -0
  230. vispy/glsl/transforms/azimuthal-equidistant.glsl +38 -0
  231. vispy/glsl/transforms/hammer.glsl +44 -0
  232. vispy/glsl/transforms/identity.glsl +6 -0
  233. vispy/glsl/transforms/identity_forward.glsl +23 -0
  234. vispy/glsl/transforms/identity_inverse.glsl +23 -0
  235. vispy/glsl/transforms/linear-scale.glsl +127 -0
  236. vispy/glsl/transforms/log-scale.glsl +126 -0
  237. vispy/glsl/transforms/mercator-transverse-forward.glsl +40 -0
  238. vispy/glsl/transforms/mercator-transverse-inverse.glsl +40 -0
  239. vispy/glsl/transforms/panzoom.glsl +10 -0
  240. vispy/glsl/transforms/polar.glsl +41 -0
  241. vispy/glsl/transforms/position.glsl +44 -0
  242. vispy/glsl/transforms/power-scale.glsl +139 -0
  243. vispy/glsl/transforms/projection.glsl +7 -0
  244. vispy/glsl/transforms/pvm.glsl +13 -0
  245. vispy/glsl/transforms/rotate.glsl +45 -0
  246. vispy/glsl/transforms/trackball.glsl +15 -0
  247. vispy/glsl/transforms/translate.glsl +35 -0
  248. vispy/glsl/transforms/transverse_mercator.glsl +38 -0
  249. vispy/glsl/transforms/viewport-clipping.glsl +14 -0
  250. vispy/glsl/transforms/viewport-transform.glsl +16 -0
  251. vispy/glsl/transforms/viewport.glsl +50 -0
  252. vispy/glsl/transforms/x.glsl +24 -0
  253. vispy/glsl/transforms/y.glsl +19 -0
  254. vispy/glsl/transforms/z.glsl +14 -0
  255. vispy/io/__init__.py +20 -0
  256. vispy/io/_data/spatial-filters.npy +0 -0
  257. vispy/io/datasets.py +94 -0
  258. vispy/io/image.py +231 -0
  259. vispy/io/mesh.py +122 -0
  260. vispy/io/stl.py +167 -0
  261. vispy/io/tests/__init__.py +0 -0
  262. vispy/io/tests/test_image.py +47 -0
  263. vispy/io/tests/test_io.py +121 -0
  264. vispy/io/wavefront.py +350 -0
  265. vispy/plot/__init__.py +36 -0
  266. vispy/plot/fig.py +58 -0
  267. vispy/plot/plotwidget.py +522 -0
  268. vispy/plot/tests/__init__.py +0 -0
  269. vispy/plot/tests/test_plot.py +46 -0
  270. vispy/scene/__init__.py +43 -0
  271. vispy/scene/cameras/__init__.py +27 -0
  272. vispy/scene/cameras/_base.py +38 -0
  273. vispy/scene/cameras/arcball.py +105 -0
  274. vispy/scene/cameras/base_camera.py +551 -0
  275. vispy/scene/cameras/fly.py +474 -0
  276. vispy/scene/cameras/magnify.py +163 -0
  277. vispy/scene/cameras/panzoom.py +311 -0
  278. vispy/scene/cameras/perspective.py +338 -0
  279. vispy/scene/cameras/tests/__init__.py +0 -0
  280. vispy/scene/cameras/tests/test_cameras.py +27 -0
  281. vispy/scene/cameras/tests/test_link.py +53 -0
  282. vispy/scene/cameras/tests/test_perspective.py +122 -0
  283. vispy/scene/cameras/turntable.py +183 -0
  284. vispy/scene/canvas.py +639 -0
  285. vispy/scene/events.py +85 -0
  286. vispy/scene/node.py +644 -0
  287. vispy/scene/subscene.py +20 -0
  288. vispy/scene/tests/__init__.py +0 -0
  289. vispy/scene/tests/test_canvas.py +119 -0
  290. vispy/scene/tests/test_node.py +142 -0
  291. vispy/scene/tests/test_visuals.py +141 -0
  292. vispy/scene/visuals.py +276 -0
  293. vispy/scene/widgets/__init__.py +18 -0
  294. vispy/scene/widgets/anchor.py +25 -0
  295. vispy/scene/widgets/axis.py +88 -0
  296. vispy/scene/widgets/colorbar.py +176 -0
  297. vispy/scene/widgets/console.py +351 -0
  298. vispy/scene/widgets/grid.py +509 -0
  299. vispy/scene/widgets/label.py +50 -0
  300. vispy/scene/widgets/tests/__init__.py +0 -0
  301. vispy/scene/widgets/tests/test_colorbar.py +47 -0
  302. vispy/scene/widgets/viewbox.py +199 -0
  303. vispy/scene/widgets/widget.py +478 -0
  304. vispy/testing/__init__.py +51 -0
  305. vispy/testing/_runners.py +448 -0
  306. vispy/testing/_testing.py +416 -0
  307. vispy/testing/image_tester.py +494 -0
  308. vispy/testing/rendered_array_tester.py +85 -0
  309. vispy/testing/tests/__init__.py +0 -0
  310. vispy/testing/tests/test_testing.py +20 -0
  311. vispy/util/__init__.py +32 -0
  312. vispy/util/bunch.py +15 -0
  313. vispy/util/check_environment.py +57 -0
  314. vispy/util/config.py +490 -0
  315. vispy/util/dpi/__init__.py +19 -0
  316. vispy/util/dpi/_linux.py +69 -0
  317. vispy/util/dpi/_quartz.py +26 -0
  318. vispy/util/dpi/_win32.py +34 -0
  319. vispy/util/dpi/tests/__init__.py +0 -0
  320. vispy/util/dpi/tests/test_dpi.py +16 -0
  321. vispy/util/eq.py +41 -0
  322. vispy/util/event.py +774 -0
  323. vispy/util/fetching.py +276 -0
  324. vispy/util/filter.py +44 -0
  325. vispy/util/fonts/__init__.py +14 -0
  326. vispy/util/fonts/_freetype.py +73 -0
  327. vispy/util/fonts/_quartz.py +192 -0
  328. vispy/util/fonts/_triage.py +36 -0
  329. vispy/util/fonts/_vispy_fonts.py +20 -0
  330. vispy/util/fonts/_win32.py +105 -0
  331. vispy/util/fonts/data/OpenSans-Bold.ttf +0 -0
  332. vispy/util/fonts/data/OpenSans-BoldItalic.ttf +0 -0
  333. vispy/util/fonts/data/OpenSans-Italic.ttf +0 -0
  334. vispy/util/fonts/data/OpenSans-Regular.ttf +0 -0
  335. vispy/util/fonts/tests/__init__.py +0 -0
  336. vispy/util/fonts/tests/test_font.py +45 -0
  337. vispy/util/fourier.py +69 -0
  338. vispy/util/frozen.py +25 -0
  339. vispy/util/gallery_scraper.py +268 -0
  340. vispy/util/keys.py +91 -0
  341. vispy/util/logs.py +358 -0
  342. vispy/util/osmesa_gl.py +17 -0
  343. vispy/util/profiler.py +135 -0
  344. vispy/util/ptime.py +16 -0
  345. vispy/util/quaternion.py +229 -0
  346. vispy/util/svg/__init__.py +18 -0
  347. vispy/util/svg/base.py +20 -0
  348. vispy/util/svg/color.py +219 -0
  349. vispy/util/svg/element.py +51 -0
  350. vispy/util/svg/geometry.py +478 -0
  351. vispy/util/svg/group.py +66 -0
  352. vispy/util/svg/length.py +81 -0
  353. vispy/util/svg/number.py +25 -0
  354. vispy/util/svg/path.py +332 -0
  355. vispy/util/svg/shapes.py +57 -0
  356. vispy/util/svg/style.py +59 -0
  357. vispy/util/svg/svg.py +40 -0
  358. vispy/util/svg/transform.py +223 -0
  359. vispy/util/svg/transformable.py +28 -0
  360. vispy/util/svg/viewport.py +73 -0
  361. vispy/util/tests/__init__.py +0 -0
  362. vispy/util/tests/test_config.py +58 -0
  363. vispy/util/tests/test_docstring_parameters.py +123 -0
  364. vispy/util/tests/test_emitter_group.py +262 -0
  365. vispy/util/tests/test_event_emitter.py +743 -0
  366. vispy/util/tests/test_fourier.py +35 -0
  367. vispy/util/tests/test_gallery_scraper.py +112 -0
  368. vispy/util/tests/test_import.py +127 -0
  369. vispy/util/tests/test_key.py +22 -0
  370. vispy/util/tests/test_logging.py +45 -0
  371. vispy/util/tests/test_run.py +14 -0
  372. vispy/util/tests/test_transforms.py +42 -0
  373. vispy/util/tests/test_vispy.py +48 -0
  374. vispy/util/transforms.py +201 -0
  375. vispy/util/wrappers.py +155 -0
  376. vispy/version.py +21 -0
  377. vispy/visuals/__init__.py +50 -0
  378. vispy/visuals/_scalable_textures.py +487 -0
  379. vispy/visuals/axis.py +678 -0
  380. vispy/visuals/border.py +208 -0
  381. vispy/visuals/box.py +79 -0
  382. vispy/visuals/collections/__init__.py +30 -0
  383. vispy/visuals/collections/agg_fast_path_collection.py +219 -0
  384. vispy/visuals/collections/agg_path_collection.py +197 -0
  385. vispy/visuals/collections/agg_point_collection.py +52 -0
  386. vispy/visuals/collections/agg_segment_collection.py +142 -0
  387. vispy/visuals/collections/array_list.py +401 -0
  388. vispy/visuals/collections/base_collection.py +482 -0
  389. vispy/visuals/collections/collection.py +253 -0
  390. vispy/visuals/collections/path_collection.py +23 -0
  391. vispy/visuals/collections/point_collection.py +19 -0
  392. vispy/visuals/collections/polygon_collection.py +25 -0
  393. vispy/visuals/collections/raw_path_collection.py +119 -0
  394. vispy/visuals/collections/raw_point_collection.py +113 -0
  395. vispy/visuals/collections/raw_polygon_collection.py +77 -0
  396. vispy/visuals/collections/raw_segment_collection.py +112 -0
  397. vispy/visuals/collections/raw_triangle_collection.py +78 -0
  398. vispy/visuals/collections/segment_collection.py +19 -0
  399. vispy/visuals/collections/triangle_collection.py +16 -0
  400. vispy/visuals/collections/util.py +168 -0
  401. vispy/visuals/colorbar.py +699 -0
  402. vispy/visuals/cube.py +41 -0
  403. vispy/visuals/ellipse.py +162 -0
  404. vispy/visuals/filters/__init__.py +10 -0
  405. vispy/visuals/filters/base_filter.py +242 -0
  406. vispy/visuals/filters/clipper.py +60 -0
  407. vispy/visuals/filters/clipping_planes.py +122 -0
  408. vispy/visuals/filters/color.py +181 -0
  409. vispy/visuals/filters/markers.py +28 -0
  410. vispy/visuals/filters/mesh.py +801 -0
  411. vispy/visuals/filters/picking.py +60 -0
  412. vispy/visuals/filters/tests/__init__.py +3 -0
  413. vispy/visuals/filters/tests/test_primitive_picking_filters.py +70 -0
  414. vispy/visuals/filters/tests/test_wireframe_filter.py +16 -0
  415. vispy/visuals/glsl/__init__.py +1 -0
  416. vispy/visuals/glsl/antialiasing.py +133 -0
  417. vispy/visuals/glsl/color.py +63 -0
  418. vispy/visuals/graphs/__init__.py +1 -0
  419. vispy/visuals/graphs/graph.py +240 -0
  420. vispy/visuals/graphs/layouts/__init__.py +55 -0
  421. vispy/visuals/graphs/layouts/circular.py +49 -0
  422. vispy/visuals/graphs/layouts/force_directed.py +211 -0
  423. vispy/visuals/graphs/layouts/networkx_layout.py +87 -0
  424. vispy/visuals/graphs/layouts/random.py +52 -0
  425. vispy/visuals/graphs/tests/__init__.py +1 -0
  426. vispy/visuals/graphs/tests/test_layouts.py +139 -0
  427. vispy/visuals/graphs/tests/test_networkx_layout.py +47 -0
  428. vispy/visuals/graphs/util.py +120 -0
  429. vispy/visuals/gridlines.py +161 -0
  430. vispy/visuals/gridmesh.py +98 -0
  431. vispy/visuals/histogram.py +58 -0
  432. vispy/visuals/image.py +701 -0
  433. vispy/visuals/image_complex.py +130 -0
  434. vispy/visuals/infinite_line.py +199 -0
  435. vispy/visuals/instanced_mesh.py +152 -0
  436. vispy/visuals/isocurve.py +213 -0
  437. vispy/visuals/isoline.py +241 -0
  438. vispy/visuals/isosurface.py +113 -0
  439. vispy/visuals/line/__init__.py +6 -0
  440. vispy/visuals/line/arrow.py +289 -0
  441. vispy/visuals/line/dash_atlas.py +90 -0
  442. vispy/visuals/line/line.py +545 -0
  443. vispy/visuals/line_plot.py +135 -0
  444. vispy/visuals/linear_region.py +199 -0
  445. vispy/visuals/markers.py +819 -0
  446. vispy/visuals/mesh.py +373 -0
  447. vispy/visuals/mesh_normals.py +159 -0
  448. vispy/visuals/plane.py +54 -0
  449. vispy/visuals/polygon.py +145 -0
  450. vispy/visuals/rectangle.py +196 -0
  451. vispy/visuals/regular_polygon.py +56 -0
  452. vispy/visuals/scrolling_lines.py +197 -0
  453. vispy/visuals/shaders/__init__.py +17 -0
  454. vispy/visuals/shaders/compiler.py +206 -0
  455. vispy/visuals/shaders/expression.py +99 -0
  456. vispy/visuals/shaders/function.py +788 -0
  457. vispy/visuals/shaders/multiprogram.py +145 -0
  458. vispy/visuals/shaders/parsing.py +140 -0
  459. vispy/visuals/shaders/program.py +161 -0
  460. vispy/visuals/shaders/shader_object.py +162 -0
  461. vispy/visuals/shaders/tests/__init__.py +0 -0
  462. vispy/visuals/shaders/tests/test_function.py +486 -0
  463. vispy/visuals/shaders/tests/test_multiprogram.py +78 -0
  464. vispy/visuals/shaders/tests/test_parsing.py +57 -0
  465. vispy/visuals/shaders/variable.py +272 -0
  466. vispy/visuals/spectrogram.py +169 -0
  467. vispy/visuals/sphere.py +80 -0
  468. vispy/visuals/surface_plot.py +192 -0
  469. vispy/visuals/tests/__init__.py +0 -0
  470. vispy/visuals/tests/test_arrows.py +109 -0
  471. vispy/visuals/tests/test_axis.py +120 -0
  472. vispy/visuals/tests/test_collections.py +15 -0
  473. vispy/visuals/tests/test_colorbar.py +179 -0
  474. vispy/visuals/tests/test_colormap.py +97 -0
  475. vispy/visuals/tests/test_ellipse.py +122 -0
  476. vispy/visuals/tests/test_gridlines.py +30 -0
  477. vispy/visuals/tests/test_histogram.py +24 -0
  478. vispy/visuals/tests/test_image.py +392 -0
  479. vispy/visuals/tests/test_image_complex.py +36 -0
  480. vispy/visuals/tests/test_infinite_line.py +53 -0
  481. vispy/visuals/tests/test_instanced_mesh.py +50 -0
  482. vispy/visuals/tests/test_isosurface.py +22 -0
  483. vispy/visuals/tests/test_linear_region.py +152 -0
  484. vispy/visuals/tests/test_markers.py +54 -0
  485. vispy/visuals/tests/test_mesh.py +261 -0
  486. vispy/visuals/tests/test_mesh_normals.py +218 -0
  487. vispy/visuals/tests/test_polygon.py +112 -0
  488. vispy/visuals/tests/test_rectangle.py +163 -0
  489. vispy/visuals/tests/test_regular_polygon.py +111 -0
  490. vispy/visuals/tests/test_scalable_textures.py +196 -0
  491. vispy/visuals/tests/test_sdf.py +73 -0
  492. vispy/visuals/tests/test_spectrogram.py +42 -0
  493. vispy/visuals/tests/test_surface_plot.py +57 -0
  494. vispy/visuals/tests/test_text.py +95 -0
  495. vispy/visuals/tests/test_volume.py +542 -0
  496. vispy/visuals/tests/test_windbarb.py +33 -0
  497. vispy/visuals/text/__init__.py +7 -0
  498. vispy/visuals/text/_sdf_cpu.cpython-313-aarch64-linux-gnu.so +0 -0
  499. vispy/visuals/text/_sdf_cpu.pyx +112 -0
  500. vispy/visuals/text/_sdf_gpu.py +316 -0
  501. vispy/visuals/text/text.py +675 -0
  502. vispy/visuals/transforms/__init__.py +34 -0
  503. vispy/visuals/transforms/_util.py +191 -0
  504. vispy/visuals/transforms/base_transform.py +233 -0
  505. vispy/visuals/transforms/chain.py +300 -0
  506. vispy/visuals/transforms/interactive.py +98 -0
  507. vispy/visuals/transforms/linear.py +564 -0
  508. vispy/visuals/transforms/nonlinear.py +398 -0
  509. vispy/visuals/transforms/tests/__init__.py +0 -0
  510. vispy/visuals/transforms/tests/test_transforms.py +243 -0
  511. vispy/visuals/transforms/transform_system.py +339 -0
  512. vispy/visuals/tube.py +173 -0
  513. vispy/visuals/visual.py +923 -0
  514. vispy/visuals/volume.py +1366 -0
  515. vispy/visuals/windbarb.py +291 -0
  516. vispy/visuals/xyz_axis.py +34 -0
  517. vispy-0.15.0.dist-info/METADATA +243 -0
  518. vispy-0.15.0.dist-info/RECORD +521 -0
  519. vispy-0.15.0.dist-info/WHEEL +6 -0
  520. vispy-0.15.0.dist-info/licenses/LICENSE.txt +36 -0
  521. vispy-0.15.0.dist-info/top_level.txt +1 -0
@@ -0,0 +1,732 @@
1
+ # -*- coding: utf-8 -*-
2
+ # -----------------------------------------------------------------------------
3
+ # Copyright (c) 2014, Nicolas P. Rougier. All rights reserved.
4
+ # Distributed under the terms of the new BSD License.
5
+ # -----------------------------------------------------------------------------
6
+ import unittest
7
+ import numpy as np
8
+ import pytest
9
+
10
+ from vispy.gloo import Texture1D, Texture2D, Texture3D, TextureAtlas
11
+ from vispy.testing import requires_pyopengl, run_tests_if_main, assert_raises
12
+
13
+ # here we test some things that will be true of all Texture types:
14
+ Texture = Texture2D
15
+
16
+
17
+ # ----------------------------------------------------------------- Texture ---
18
+ class TextureTest(unittest.TestCase):
19
+
20
+ # No data, no dtype : forbidden
21
+ # ---------------------------------
22
+ def test_init_none(self):
23
+ self.assertRaises(ValueError, Texture)
24
+
25
+ # Data only
26
+ # ---------------------------------
27
+ def test_init_data(self):
28
+ data = np.zeros((10, 10, 3), dtype=np.uint8)
29
+ T = Texture(data=data, interpolation='linear', wrapping='repeat')
30
+ assert T._shape == (10, 10, 3)
31
+ assert T._interpolation == ('linear', 'linear')
32
+ assert T._wrapping == ('repeat', 'repeat')
33
+
34
+ # Setting data and shape
35
+ # ---------------------------------
36
+ def test_init_dtype_shape(self):
37
+ T = Texture((10, 10))
38
+ assert T._shape == (10, 10, 1)
39
+ self.assertRaises(ValueError, Texture, shape=(10, 10),
40
+ data=np.zeros((10, 10), np.float32))
41
+
42
+ # Set data with store
43
+ # ---------------------------------
44
+ def test_setitem_all(self):
45
+ data = np.zeros((10, 10), dtype=np.uint8)
46
+ T = Texture(data=data)
47
+ T[...] = np.ones((10, 10, 1))
48
+ glir_cmd = T._glir.clear()[-1]
49
+ assert glir_cmd[0] == 'DATA'
50
+ assert np.allclose(glir_cmd[3], np.ones((10, 10, 1)))
51
+
52
+ # Set data without store
53
+ # ---------------------------------
54
+ def test_setitem_all_no_store(self):
55
+ data = np.zeros((10, 10), dtype=np.uint8)
56
+ T = Texture(data=data)
57
+ T[...] = np.ones((10, 10), np.uint8)
58
+ assert np.allclose(data, np.zeros((10, 10)))
59
+
60
+ # Set a single data
61
+ # ---------------------------------
62
+ def test_setitem_single(self):
63
+
64
+ data = np.zeros((10, 10), dtype=np.uint8)
65
+ T = Texture(data=data)
66
+ T[0, 0, 0] = 1
67
+ glir_cmd = T._glir.clear()[-1]
68
+ assert glir_cmd[0] == 'DATA'
69
+ assert np.allclose(glir_cmd[3], np.array([1]))
70
+
71
+ # We apparently support this
72
+ T[8:3, 3] = 1
73
+
74
+ # Set some data
75
+ # ---------------------------------
76
+ def test_setitem_partial(self):
77
+
78
+ data = np.zeros((10, 10), dtype=np.uint8)
79
+ T = Texture(data=data)
80
+ T[5:, 5:] = 1
81
+ glir_cmd = T._glir.clear()[-1]
82
+ assert glir_cmd[0] == 'DATA'
83
+ assert np.allclose(glir_cmd[3], np.ones((5, 5)))
84
+
85
+ # Set non contiguous data
86
+ # ---------------------------------
87
+ def test_setitem_wrong(self):
88
+ data = np.zeros((10, 10), dtype=np.uint8)
89
+ T = Texture(data=data)
90
+ # with self.assertRaises(ValueError):
91
+ # T[::2, ::2] = 1
92
+ s = slice(None, None, 2)
93
+ self.assertRaises(IndexError, T.__setitem__, (s, s), 1)
94
+ self.assertRaises(IndexError, T.__setitem__, (-100, 3), 1)
95
+ self.assertRaises(TypeError, T.__setitem__, ('foo', 'bar'), 1)
96
+
97
+ # Set properties
98
+ def test_set_texture_properties(self):
99
+ T = Texture((10, 10))
100
+
101
+ # Interpolation
102
+ T.interpolation = 'nearest'
103
+ assert T.interpolation == 'nearest'
104
+ T.interpolation = 'linear'
105
+ assert T.interpolation == 'linear'
106
+ T.interpolation = ['linear'] * 2
107
+ assert T.interpolation == 'linear'
108
+ T.interpolation = ['linear', 'nearest']
109
+ assert T.interpolation == ('linear', 'nearest')
110
+
111
+ # Wrong interpolation
112
+ iset = Texture.interpolation.fset
113
+ self.assertRaises(ValueError, iset, T, ['linear'] * 3)
114
+ self.assertRaises(ValueError, iset, T, True)
115
+ self.assertRaises(ValueError, iset, T, [])
116
+ self.assertRaises(ValueError, iset, T, 'linearios')
117
+
118
+ # Wrapping
119
+ T.wrapping = 'clamp_to_edge'
120
+ assert T.wrapping == 'clamp_to_edge'
121
+ T.wrapping = 'repeat'
122
+ assert T.wrapping == 'repeat'
123
+ T.wrapping = 'mirrored_repeat'
124
+ assert T.wrapping == 'mirrored_repeat'
125
+ T.wrapping = 'repeat', 'repeat'
126
+ assert T.wrapping == 'repeat'
127
+ T.wrapping = 'repeat', 'clamp_to_edge'
128
+ assert T.wrapping == ('repeat', 'clamp_to_edge')
129
+
130
+ # Wrong wrapping
131
+ wset = Texture.wrapping.fset
132
+ self.assertRaises(ValueError, wset, T, ['repeat'] * 3)
133
+ self.assertRaises(ValueError, wset, T, True)
134
+ self.assertRaises(ValueError, wset, T, [])
135
+ self.assertRaises(ValueError, wset, T, 'repeatos')
136
+
137
+
138
+ # --------------------------------------------------------------- Texture2D ---
139
+ class Texture2DTest(unittest.TestCase):
140
+ # Note: put many tests related to (re)sizing here, because Texture
141
+ # is not really aware of shape.
142
+
143
+ # Shape extension
144
+ # ---------------------------------
145
+ def test_init(self):
146
+ data = np.zeros((10, 10), dtype=np.uint8)
147
+ T = Texture2D(data=data)
148
+ assert 'Texture2D' in repr(T)
149
+ assert T._shape == (10, 10, 1)
150
+ assert T.glsl_type == ('uniform', 'sampler2D')
151
+
152
+ # Width & height
153
+ # ---------------------------------
154
+ def test_width_height(self):
155
+ data = np.zeros((10, 20), dtype=np.uint8)
156
+ T = Texture2D(data=data)
157
+ assert T.width == 20
158
+ assert T.height == 10
159
+
160
+ # Resize
161
+ # ---------------------------------
162
+ def test_resize(self):
163
+ data = np.zeros((10, 10), dtype=np.uint8)
164
+ T = Texture2D(data=data)
165
+ T.resize((5, 5))
166
+ assert T.shape == (5, 5, 1)
167
+ glir_cmd = T._glir.clear()[-1]
168
+ assert glir_cmd[0] == 'SIZE'
169
+
170
+ # Wong arg
171
+ self.assertRaises(ValueError, T.resize, (5, 5), 4)
172
+
173
+ # Resize with bad shape
174
+ # ---------------------------------
175
+ def test_resize_bad_shape(self):
176
+ data = np.zeros((10, 10), dtype=np.uint8)
177
+ T = Texture2D(data=data)
178
+ # with self.assertRaises(ValueError):
179
+ # T.resize((5, 5, 5))
180
+ self.assertRaises(ValueError, T.resize, (5,))
181
+ self.assertRaises(ValueError, T.resize, (5, 5, 5))
182
+ self.assertRaises(ValueError, T.resize, (5, 5, 5, 1))
183
+
184
+ # Resize not resizable
185
+ # ---------------------------------
186
+ def test_resize_unresizable(self):
187
+ data = np.zeros((10, 10), dtype=np.uint8)
188
+ T = Texture2D(data=data, resizable=False)
189
+ # with self.assertRaises(RuntimeError):
190
+ # T.resize((5, 5))
191
+ self.assertRaises(RuntimeError, T.resize, (5, 5))
192
+
193
+ # Set oversized data (-> resize)
194
+ # ---------------------------------
195
+ def test_set_oversized_data(self):
196
+ data = np.zeros((10, 10), dtype=np.uint8)
197
+ T = Texture2D(data=data)
198
+ T.set_data(np.ones((20, 20), np.uint8))
199
+ assert T.shape == (20, 20, 1)
200
+ glir_cmds = T._glir.clear()
201
+ assert glir_cmds[-2][0] == 'SIZE'
202
+ assert glir_cmds[-1][0] == 'DATA'
203
+
204
+ # Set undersized data
205
+ # ---------------------------------
206
+ def test_set_undersized_data(self):
207
+ data = np.zeros((10, 10), dtype=np.uint8)
208
+ T = Texture2D(data=data)
209
+ T.set_data(np.ones((5, 5), np.uint8))
210
+ assert T.shape == (5, 5, 1)
211
+ glir_cmds = T._glir.clear()
212
+ assert glir_cmds[-2][0] == 'SIZE'
213
+ assert glir_cmds[-1][0] == 'DATA'
214
+
215
+ # Set misplaced data
216
+ # ---------------------------------
217
+ def test_set_misplaced_data(self):
218
+ data = np.zeros((10, 10), dtype=np.uint8)
219
+ T = Texture2D(data=data)
220
+ # with self.assertRaises(ValueError):
221
+ # T.set_data(np.ones((5, 5)), offset=(8, 8))
222
+ self.assertRaises(ValueError, T.set_data,
223
+ np.ones((5, 5)), offset=(8, 8))
224
+
225
+ # Set misshaped data
226
+ # ---------------------------------
227
+ def test_set_misshaped_data_2D(self):
228
+ data = np.zeros((10, 10), dtype=np.uint8)
229
+ T = Texture2D(data=data)
230
+ # with self.assertRaises(ValueError):
231
+ # T.set_data(np.ones((10, 10)))
232
+ self.assertRaises(ValueError, T.set_data, np.ones((10,)))
233
+ self.assertRaises(ValueError, T.set_data, np.ones((5, 5, 5, 1)),)
234
+
235
+ # Set whole data (clear pending data)
236
+ # ---------------------------------
237
+ def test_set_whole_data(self):
238
+ data = np.zeros((10, 10), dtype=np.uint8)
239
+ T = Texture2D(data=data)
240
+ T.set_data(np.ones((10, 10), np.uint8))
241
+ assert T.shape == (10, 10, 1)
242
+
243
+ # Test set data with different shape
244
+ # ---------------------------------
245
+ def test_reset_data_shape(self):
246
+ shape1 = 10, 10
247
+ shape3 = 10, 10, 3
248
+
249
+ # Init data (explicit shape)
250
+ data = np.zeros((10, 10, 1), dtype=np.uint8)
251
+ T = Texture2D(data=data)
252
+ assert T.shape == (10, 10, 1)
253
+ assert T.format == 'luminance'
254
+
255
+ # Set data to rgb
256
+ T.set_data(np.zeros(shape3, np.uint8))
257
+ assert T.shape == (10, 10, 3)
258
+ assert T.format == 'rgb'
259
+
260
+ # Set data to grayscale
261
+ T.set_data(np.zeros(shape1, np.uint8))
262
+ assert T.shape == (10, 10, 1)
263
+ assert T.format == 'luminance'
264
+
265
+ # Set size to rgb
266
+ T.resize(shape3)
267
+ assert T.shape == (10, 10, 3)
268
+ assert T._format == 'rgb'
269
+
270
+ # Set size to grayscale
271
+ T.resize(shape1)
272
+ assert T.shape == (10, 10, 1)
273
+ assert T._format == 'luminance'
274
+
275
+ # Keep using old format
276
+ T.resize(shape1, 'alpha')
277
+ T.resize(shape1)
278
+ assert T._format == 'alpha'
279
+
280
+ # Use luminance as default
281
+ T.resize(shape3)
282
+ T.resize(shape1)
283
+ assert T._format == 'luminance'
284
+
285
+ # Too large
286
+ self.assertRaises(ValueError, T.resize, (5, 5, 5, 1))
287
+ # Cannot determine format
288
+ self.assertRaises(ValueError, T.resize, (5, 5, 5))
289
+ # Invalid format
290
+ self.assertRaises(ValueError, T.resize, shape3, 'foo')
291
+ self.assertRaises(ValueError, T.resize, shape3, 'alpha')
292
+ # self.assertRaises(ValueError, T.resize, shape3, 4)
293
+
294
+ # Test set data with different shape and type
295
+ # -------------------------------------------
296
+ def test_reset_data_type(self):
297
+ data = np.zeros((10, 10), dtype=np.uint8)
298
+ T = Texture2D(data=data)
299
+
300
+ data = np.zeros((10, 11), dtype=np.float32)
301
+ T.set_data(data)
302
+
303
+ data = np.zeros((12, 10), dtype=np.int32)
304
+ T.set_data(data)
305
+
306
+ self.assertRaises(ValueError, T.set_data, np.zeros([10, 10, 10, 1]))
307
+
308
+
309
+ # --------------------------------------------------------------- Texture1D ---
310
+ @requires_pyopengl()
311
+ def test_texture_1D():
312
+ # Note: put many tests related to (re)sizing here, because Texture
313
+ # is not really aware of shape.
314
+
315
+ # Shape extension
316
+ # ---------------------------------
317
+ data = np.zeros((10, ), dtype=np.uint8)
318
+ T = Texture1D(data=data)
319
+ assert T._shape == (10, 1)
320
+ assert 'Texture1D' in repr(T)
321
+ assert T.glsl_type == ('uniform', 'sampler1D')
322
+
323
+ # Width
324
+ # ---------------------------------
325
+ data = np.zeros((10, ), dtype=np.uint8)
326
+ T = Texture1D(data=data)
327
+ assert T.width == 10
328
+
329
+ # Resize
330
+ # ---------------------------------
331
+ data = np.zeros((10, ), dtype=np.uint8)
332
+ T = Texture1D(data=data)
333
+ T.resize((5, ))
334
+ assert T.shape == (5, 1)
335
+ glir_cmd = T._glir.clear()[-1]
336
+ assert glir_cmd[0] == 'SIZE'
337
+ assert glir_cmd[2] == (5, 1)
338
+
339
+ # Resize with bad shape
340
+ # ---------------------------------
341
+ data = np.zeros((10, ), dtype=np.uint8)
342
+ T = Texture1D(data=data)
343
+ # with self.assertRaises(ValueError):
344
+ # T.resize((5, 5, 5, 5))
345
+ assert_raises(ValueError, T.resize, (5, 5, 5, 5))
346
+
347
+ # Resize not resizable
348
+ # ---------------------------------
349
+ data = np.zeros((10, ), dtype=np.uint8)
350
+ T = Texture1D(data=data, resizable=False)
351
+ # with self.assertRaises(RuntimeError):
352
+ # T.resize((5, 5, 5))
353
+ assert_raises(RuntimeError, T.resize, (5, ))
354
+
355
+ # Set oversized data (-> resize)
356
+ # ---------------------------------
357
+ data = np.zeros((10, ), dtype=np.uint8)
358
+ T = Texture1D(data=data)
359
+ T.set_data(np.ones((20, ), np.uint8))
360
+ assert T.shape == (20, 1)
361
+
362
+ # Set undersized data
363
+ # ---------------------------------
364
+ data = np.zeros((10, ), dtype=np.uint8)
365
+ T = Texture1D(data=data)
366
+ T.set_data(np.ones((5, ), np.uint8))
367
+ assert T.shape == (5, 1)
368
+
369
+ # Set misplaced data
370
+ # ---------------------------------
371
+ data = np.zeros((10, ), dtype=np.uint8)
372
+ T = Texture1D(data=data)
373
+ # with self.assertRaises(ValueError):
374
+ # T.set_data(np.ones((5, 5, 5)), offset=(8, 8, 8))
375
+ assert_raises(ValueError, T.set_data,
376
+ np.ones((5, )), offset=(8, ))
377
+
378
+ # Set misshaped data
379
+ # ---------------------------------
380
+ data = np.zeros((10, ), dtype=np.uint8)
381
+ T = Texture1D(data=data)
382
+ # with self.assertRaises(ValueError):
383
+ # T.set_data(np.ones((10, 10, 10)))
384
+ assert_raises(ValueError, T.set_data, np.ones((10, 10)))
385
+
386
+ # Set whole data (clear pending data)
387
+ # ---------------------------------
388
+ data = np.zeros((10, ), dtype=np.uint8)
389
+ T = Texture1D(data=data)
390
+ T.set_data(np.ones((10, ), np.uint8))
391
+ assert T.shape == (10, 1)
392
+ glir_cmd = T._glir.clear()[-1]
393
+ assert glir_cmd[0] == 'DATA'
394
+
395
+ # Test set data with different shape
396
+ # ---------------------------------
397
+ shape1 = (10, )
398
+ shape3 = (10, 3)
399
+
400
+ # Init data (explicit shape)
401
+ data = np.zeros((10, 1), dtype=np.uint8)
402
+ T = Texture1D(data=data)
403
+ assert T.shape == (10, 1)
404
+ assert T._format == 'luminance'
405
+
406
+ # Set data to rgb
407
+ T.set_data(np.zeros(shape3, np.uint8))
408
+ assert T.shape == (10, 3)
409
+ assert T._format == 'rgb'
410
+
411
+ # Set data to grayscale
412
+ T.set_data(np.zeros(shape1, np.uint8))
413
+ assert T.shape == (10, 1)
414
+ assert T._format == 'luminance'
415
+
416
+ # Set size to rgb
417
+ T.resize(shape3)
418
+ assert T.shape == (10, 3)
419
+ assert T._format == 'rgb'
420
+
421
+ # Set size to grayscale
422
+ T.resize(shape1)
423
+ assert T.shape == (10, 1)
424
+ assert T._format == 'luminance'
425
+
426
+ # Test set data with different shape and type
427
+ # -------------------------------------------
428
+ data = np.zeros((10, ), dtype=np.uint8)
429
+ T = Texture1D(data=data)
430
+
431
+ data = np.zeros((10, ), dtype=np.float32)
432
+ T.set_data(data)
433
+
434
+ data = np.zeros((12, ), dtype=np.int32)
435
+ T.set_data(data)
436
+
437
+
438
+ # --------------------------------------------------------------- Texture3D ---
439
+ @requires_pyopengl()
440
+ def test_texture_3D():
441
+ # Note: put many tests related to (re)sizing here, because Texture
442
+ # is not really aware of shape.
443
+
444
+ # Shape extension
445
+ # ---------------------------------
446
+ data = np.zeros((10, 10, 10), dtype=np.uint8)
447
+ T = Texture3D(data=data)
448
+ assert T._shape == (10, 10, 10, 1)
449
+ assert 'Texture3D' in repr(T)
450
+ assert T.glsl_type == ('uniform', 'sampler3D')
451
+
452
+ # Width & height
453
+ # ---------------------------------
454
+ data = np.zeros((10, 20, 30), dtype=np.uint8)
455
+ T = Texture3D(data=data)
456
+ assert T.width == 30
457
+ assert T.height == 20
458
+ assert T.depth == 10
459
+
460
+ # Resize
461
+ # ---------------------------------
462
+ data = np.zeros((10, 10, 10), dtype=np.uint8)
463
+ T = Texture3D(data=data)
464
+ T.resize((5, 5, 5))
465
+ assert T.shape == (5, 5, 5, 1)
466
+ glir_cmd = T._glir.clear()[-1]
467
+ assert glir_cmd[0] == 'SIZE'
468
+ assert glir_cmd[2] == (5, 5, 5, 1)
469
+
470
+ # Resize with bad shape
471
+ # ---------------------------------
472
+ data = np.zeros((10, 10, 10), dtype=np.uint8)
473
+ T = Texture3D(data=data)
474
+ # with self.assertRaises(ValueError):
475
+ # T.resize((5, 5, 5, 5))
476
+ assert_raises(ValueError, T.resize, (5, 5, 5, 5))
477
+
478
+ # Resize not resizable
479
+ # ---------------------------------
480
+ data = np.zeros((10, 10, 10), dtype=np.uint8)
481
+ T = Texture3D(data=data, resizable=False)
482
+ # with self.assertRaises(RuntimeError):
483
+ # T.resize((5, 5, 5))
484
+ assert_raises(RuntimeError, T.resize, (5, 5, 5))
485
+
486
+ # Set oversized data (-> resize)
487
+ # ---------------------------------
488
+ data = np.zeros((10, 10, 10), dtype=np.uint8)
489
+ T = Texture3D(data=data)
490
+ T.set_data(np.ones((20, 20, 20), np.uint8))
491
+ assert T.shape == (20, 20, 20, 1)
492
+
493
+ # Set undersized data
494
+ # ---------------------------------
495
+ data = np.zeros((10, 10, 10), dtype=np.uint8)
496
+ T = Texture3D(data=data)
497
+ T.set_data(np.ones((5, 5, 5), np.uint8))
498
+ assert T.shape == (5, 5, 5, 1)
499
+
500
+ # Set misplaced data
501
+ # ---------------------------------
502
+ data = np.zeros((10, 10, 10), dtype=np.uint8)
503
+ T = Texture3D(data=data)
504
+ # with self.assertRaises(ValueError):
505
+ # T.set_data(np.ones((5, 5, 5)), offset=(8, 8, 8))
506
+ assert_raises(ValueError, T.set_data,
507
+ np.ones((5, 5, 5)), offset=(8, 8, 8))
508
+
509
+ # Set misshaped data
510
+ # ---------------------------------
511
+ data = np.zeros((10, 10, 10), dtype=np.uint8)
512
+ T = Texture3D(data=data)
513
+ # with self.assertRaises(ValueError):
514
+ # T.set_data(np.ones((10, 10, 10)))
515
+ assert_raises(ValueError, T.set_data, np.ones((10,)))
516
+
517
+ # Set whole data (clear pending data)
518
+ # ---------------------------------
519
+ data = np.zeros((10, 10, 10), dtype=np.uint8)
520
+ T = Texture3D(data=data)
521
+ T.set_data(np.ones((10, 10, 10), np.uint8))
522
+ assert T.shape == (10, 10, 10, 1)
523
+ glir_cmd = T._glir.clear()[-1]
524
+ assert glir_cmd[0] == 'DATA'
525
+
526
+ # Test set data with different shape
527
+ # ---------------------------------
528
+ shape1 = 10, 10, 10
529
+ shape3 = 10, 10, 10, 3
530
+
531
+ # Init data (explicit shape)
532
+ data = np.zeros((10, 10, 10, 1), dtype=np.uint8)
533
+ T = Texture3D(data=data)
534
+ assert T.shape == (10, 10, 10, 1)
535
+ assert T._format == 'luminance'
536
+
537
+ # Set data to rgb
538
+ T.set_data(np.zeros(shape3, np.uint8))
539
+ assert T.shape == (10, 10, 10, 3)
540
+ assert T._format == 'rgb'
541
+
542
+ # Set data to grayscale
543
+ T.set_data(np.zeros(shape1, np.uint8))
544
+ assert T.shape == (10, 10, 10, 1)
545
+ assert T._format == 'luminance'
546
+
547
+ # Set size to rgb
548
+ T.resize(shape3)
549
+ assert T.shape == (10, 10, 10, 3)
550
+ assert T._format == 'rgb'
551
+
552
+ # Set size to grayscale
553
+ T.resize(shape1)
554
+ assert T.shape == (10, 10, 10, 1)
555
+ assert T._format == 'luminance'
556
+
557
+ # Test set data with different shape and type
558
+ # -------------------------------------------
559
+ data = np.zeros((10, 10, 10), dtype=np.uint8)
560
+ T = Texture3D(data=data)
561
+
562
+ data = np.zeros((10, 11, 11), dtype=np.float32)
563
+ T.set_data(data)
564
+
565
+ data = np.zeros((12, 12, 10), dtype=np.int32)
566
+ T.set_data(data)
567
+
568
+
569
+ class TextureAtlasTest(unittest.TestCase):
570
+
571
+ def test_init_atas(self):
572
+ T = TextureAtlas((100, 100))
573
+ assert T.shape == (128, 128, 3) # rounds to powers of 2
574
+
575
+ for i in [10, 20, 30, 40, 50, 60]:
576
+ reg = T.get_free_region(10, 10)
577
+ assert len(reg) == 4
578
+
579
+ reg = T.get_free_region(129, 129)
580
+ assert reg is None
581
+
582
+
583
+ # --------------------------------------------------------- Texture formats ---
584
+ def _test_texture_formats(Texture, baseshape, formats):
585
+
586
+ # valid channel count and format combinations
587
+ for channels in range(1, 5):
588
+ for format in [f for n, f in formats if n == channels]:
589
+ shape = baseshape + (channels,)
590
+ T = Texture(shape=shape, format=format)
591
+ assert 'Texture' in repr(T)
592
+ assert T._shape == shape
593
+ data = np.zeros(shape, dtype=np.uint8)
594
+ T = Texture(data=data, format=format)
595
+ assert 'Texture' in repr(T)
596
+ assert T._shape == shape
597
+
598
+ # invalid channel count and format combinations
599
+ for channels in range(1, 5):
600
+ for format in [f for n, f in formats + [(5, 'junk')] if n != channels]:
601
+ shape = baseshape + (channels,)
602
+ assert_raises(ValueError, Texture, shape=shape, format=format)
603
+ data = np.zeros(shape, dtype=np.uint8)
604
+ assert_raises(ValueError, Texture, data=data, format=format)
605
+
606
+
607
+ # --------------------------------------------------------- Texture formats ---
608
+ def _test_texture_basic_formats(Texture, baseshape):
609
+ _test_texture_formats(
610
+ Texture,
611
+ baseshape,
612
+ [
613
+ (1, 'alpha'),
614
+ (1, 'luminance'),
615
+ (2, 'luminance_alpha'),
616
+ (3, 'rgb'),
617
+ (4, 'rgba')
618
+ ]
619
+ )
620
+
621
+
622
+ # ------------------------------------------------------- Texture1D formats ---
623
+ def test_texture_1D_formats():
624
+ _test_texture_basic_formats(Texture1D, (10, ))
625
+
626
+
627
+ # ------------------------------------------------------- Texture2D formats ---
628
+ def test_texture_2D_formats():
629
+ _test_texture_basic_formats(Texture2D, (10, 10))
630
+
631
+
632
+ # ------------------------------------------------------- Texture3D formats ---
633
+ def test_texture_3D_formats():
634
+ _test_texture_basic_formats(Texture3D, (10, 10, 10))
635
+
636
+
637
+ # -------------------------------------------------- Texture OpenGL formats ---
638
+ def _test_texture_opengl_formats(Texture, baseshape):
639
+ _test_texture_formats(
640
+ Texture,
641
+ baseshape,
642
+ [
643
+ (1, 'red'),
644
+ (2, 'rg'),
645
+ (3, 'rgb'),
646
+ (4, 'rgba'),
647
+ (1, 'depth_component'),
648
+ ]
649
+ )
650
+
651
+
652
+ # ------------------------------------------------ Texture1D OpenGL formats ---
653
+ @requires_pyopengl()
654
+ def test_texture_1D_opengl_formats():
655
+ _test_texture_opengl_formats(Texture1D, (10, ))
656
+
657
+
658
+ # ------------------------------------------------ Texture2D OpenGL formats ---
659
+ @requires_pyopengl()
660
+ def test_texture_2D_opengl_formats():
661
+ _test_texture_opengl_formats(Texture2D, (10, 10))
662
+
663
+
664
+ # ------------------------------------------------ Texture3D OpenGL formats ---
665
+ @requires_pyopengl()
666
+ def test_texture_3D_opengl_formats():
667
+ _test_texture_opengl_formats(Texture3D, (10, 10, 10))
668
+
669
+
670
+ # ------------------------------------------ Texture OpenGL internalformats ---
671
+ def _test_texture_internalformats(Texture, baseshape):
672
+ # Test format for concrete Texture class and baseshape + (numchannels,)
673
+ # Test internalformats valid with desktop OpenGL
674
+
675
+ formats = [
676
+ (1, 'red', ['red', 'r8', 'r16', 'r16f', 'r32f']),
677
+ (2, 'rg', ['rg', 'rg8', 'rg16', 'rg16f', 'rg32f']),
678
+ (3, 'rgb', ['rgb', 'rgb8', 'rgb16', 'rgb16f', 'rgb32f']),
679
+ (4, 'rgba', ['rgba', 'rgba8', 'rgba16', 'rgba16f', 'rgba32f']),
680
+ (1, 'depth_component', ['depth_component']),
681
+ ]
682
+
683
+ for channels in range(1, 5):
684
+ for fmt, ifmts in [(f, iL) for n, f, iL in formats if n == channels]:
685
+ shape = baseshape + (channels,)
686
+ data = np.zeros(shape, dtype=np.uint8)
687
+ for ifmt in ifmts:
688
+ T = Texture(shape=shape, format=fmt, internalformat=ifmt)
689
+ assert 'Texture' in repr(T)
690
+ assert T._shape == shape
691
+ T = Texture(data=data, format=fmt, internalformat=ifmt)
692
+ assert 'Texture' in repr(T)
693
+ assert T._shape == shape
694
+
695
+ for channels in range(1, 5):
696
+ for fmt, ifmts in [(f, iL) for n, f, iL in formats if n != channels]:
697
+ shape = baseshape + (channels,)
698
+ data = np.zeros(shape, dtype=np.uint8)
699
+ for ifmt in ifmts:
700
+ assert_raises(ValueError, Texture, shape=shape, format=fmt,
701
+ internalformat=ifmt)
702
+ assert_raises(ValueError, Texture, data=data, format=fmt,
703
+ internalformat=ifmt)
704
+
705
+
706
+ # ---------------------------------------- Texture2D OpenGL internalformats ---
707
+ @requires_pyopengl()
708
+ def test_texture_2D_internalformats():
709
+ _test_texture_internalformats(Texture2D, (10, 10))
710
+
711
+
712
+ # ---------------------------------------- Texture3D OpenGL internalformats ---
713
+ @requires_pyopengl()
714
+ def test_texture_3D_internalformats():
715
+ _test_texture_internalformats(Texture3D, (10, 10, 10))
716
+
717
+
718
+ @requires_pyopengl()
719
+ @pytest.mark.parametrize('input_dtype', [np.uint8, np.uint16, np.float32, np.float64])
720
+ @pytest.mark.parametrize('output_dtype', [np.uint8, np.uint16, np.float32, np.float64])
721
+ @pytest.mark.parametrize('ndim', [2, 3])
722
+ def test_texture_set_data_different_dtype(input_dtype, output_dtype, ndim):
723
+ shape = (20,) * ndim
724
+ data = np.random.rand(*shape).astype(input_dtype)
725
+ Texture = Texture2D if ndim == 2 else Texture3D
726
+
727
+ tex = Texture(data)
728
+ tex[:10] = np.array(1, dtype=output_dtype)
729
+ tex.set_data(data.astype(output_dtype))
730
+
731
+
732
+ run_tests_if_main()