lavavu 1.9.6__cp311-cp311-macosx_11_0_arm64.whl → 1.9.8__cp311-cp311-macosx_11_0_arm64.whl

lavavu/dict.json

@@ -2147,6 +2147,17 @@
       false
     ]
   },
+  "upscalelines": {
+    "default": false,
+    "target": "global",
+    "type": "real",
+    "desc": "Enable to scale lines with the viewport size.",
+    "strict": true,
+    "redraw": 0,
+    "control": [
+      true
+    ]
+  },
   "fps": {
     "default": false,
     "target": "global",

lavavu/html/control.js

@@ -87,7 +87,7 @@ function WindowInteractor(id, uid, port) {
 
     //Several possible modes to try
     //Modern JupyterHub lab URL
-    var regex = /\/lab\//;
+    var regex = /\/lab(\/|\?|$)/; //Match "/lab" at end of url or "/lab/" or "/lab?" anywhere
     var parsed = regex.exec(loc.href);
     if (parsed && parsed.length > 0) {
       connect(loc.href.substring(0,parsed.index) + "/proxy/" + port);

lavavu/html/webview.html

@@ -27,7 +27,7 @@
 
     <input id="fileinput" type="file" style="visibility:hidden" onchange="useFileInput(this)" />
 
-    <script async src="https://cdn.jsdelivr.net/gh/lavavu/lavavu.github.io@1.9.6/LavaVu-amalgamated.min.js"></script>
+    <script async src="https://cdn.jsdelivr.net/gh/lavavu/lavavu.github.io@1.9.8/LavaVu-amalgamated.min.js"></script>
     <!--script src="dat.gui.min.js"></script>
     <script src="OK-min.js"></script>
 

lavavu/lavavu.py

@@ -23,7 +23,7 @@ __all__ = ['Viewer', 'Object', 'Properties', 'ColourMap', 'DrawData', 'Figure',
 import os
 #must be an object or won't be referenced from __init__.py import
 #(enures values are passed on when set externally)
-settings = {"default_args" : [], "echo_fails" : False, "quality_override" : None}
+settings = {"default_args" : [], "echo_fails" : False, "quality_override" : None, "test_mode" : False}
 #Default arguments for viewer creation
 _val = os.environ.get('LV_ARGS')
 if _val:
@@ -36,6 +36,10 @@ if _val:
 _val = os.environ.get('LV_QUALITY')
 if _val:
     settings["quality_override"] = int(_val)
+#Test mode - changes some behaviour to support automated testing
+_val = os.environ.get('LV_TEST')
+if _val:
+    settings["test_mode"] = bool(_val)
 
 import json
 import math
@@ -54,6 +58,7 @@ import asyncio
 import quaternion as quat
 import platform
 import matplotlib
+from pathlib import Path
 
 if sys.version_info[0] < 3:
     print("Python 3 required. LavaVu no longer supports Python 2.7.")
@@ -3562,7 +3567,6 @@ class Viewer(dict):
         ll = len(self.objects.list) #Save list length before load
         if not self.app.loadFile(filename):
             #Support trimesh formats
-            from pathlib import Path
             path = Path(filename)
             ext = path.suffix.lower()[1:]
             import convert
@@ -3876,6 +3880,9 @@ class Viewer(dict):
         """        
         Render a new frame, explicit display update
         """
+        if settings["test_mode"] and self.recording and self.recording.framecount != 0:
+            #Skip rendering all except first and last frames when testing
+            return
         self.app.render()
         #Video recording?
         if self.recording:
@@ -4305,7 +4312,7 @@ class Viewer(dict):
         #This is thread safe as doesn't load texture
         return self.app.imageFromFile(filename)
 
-    def testimages(self, imagelist=None, tolerance=TOL_DEFAULT, expectedPath='expected/', outputPath='./', clear=True):
+    def testimages(self, imagelist=None, tolerance=TOL_DEFAULT, expectedPath='expected/', outputPath='./', png=True, jpg=False, clear=True):
         """
         Compare a list of images to expected images for testing
 
@@ -4320,6 +4327,10 @@ class Viewer(dict):
             Where to find the expected result images (should have the same filenames as output images)
         outputPath : str
             Where to find the output images
+        png : boolean
+            Check png image outputs, defaults to True
+        jpng : boolean
+            Check jpg image outputs, defaults to False
         clear : boolean
             If the test passes the output images will be deleted, set to False to disable deletion
         """
@@ -4333,14 +4344,17 @@ class Viewer(dict):
                 #No expected images yet, just get images in cwd
                 #will be copied in to expected in testimage()
                 pass
-            imagelist = glob.glob("*.png")
-            imagelist += glob.glob("*.jpg")
+            imagelist = []
+            if png:
+                imagelist += glob.glob("*.png")
+            if jpg:
+                imagelist += glob.glob("*.jpg")
             imagelist.sort(key=os.path.getmtime)
             os.chdir(cwd)
 
         for f in sorted(imagelist):
-            outfile = outputPath+f
-            expfile = expectedPath+f
+            outfile = os.path.join(outputPath, f)
+            expfile = os.path.join(expectedPath, f)
             results.append(self.testimage(expfile, outfile, tolerance, clear))
 
         #Combined result
@@ -5504,9 +5518,6 @@ class Video(object):
         **kwargs :
             Any additional keyword args will also be passed as options to the encoder
         """
-        if av is None:
-            raise(ImportError("Video output not supported without pyAV - pip install av"))
-            return
         self.resolution = list(resolution)
         if self.resolution[0] == 0 or self.resolution[1] == 0:
             self.resolution = (viewer.app.viewer.width, viewer.app.viewer.height)
@@ -5515,7 +5526,20 @@ class Video(object):
         self.framerate = framerate
         self.quality = quality
         self.viewer = viewer
-        self.filename = filename
+        self.filename = Path(filename)
+        if settings["test_mode"]:
+            #Force png output when testing
+            encoder = 'png'
+        if av is None or encoder == 'png' or encoder == 'jpg':
+            if av is None:
+                print("Video output not supported without pyAV - pip install av, writing image frames instead")
+            self.encoder = 'png' if encoder == 'png' else 'jpg'
+            p = self.filename
+            self.basename = p.stem
+            self.filename = p.parent / p.stem
+            self.framecount = 0
+            return
+
         self.player = player
         if self.player is None:
             #Default player is half output resolution, unless < 900 then full
@@ -5547,6 +5571,12 @@ class Video(object):
         #Compression level, lower = high quality
         #See also: https://github.com/PyAV-Org/PyAV/blob/main/tests/test_encode.py
         options = {}
+        if self.encoder == 'jpg' or self.encoder == 'png':
+            if not settings["test_mode"]:
+                self.filename.mkdir(exist_ok=True)
+            self.viewer.recording = self
+            return
+
         if self.encoder == 'h264' or self.encoder == 'libx265':
             #Only have default options for h264/265 for now
             if self.quality == 1:
@@ -5576,7 +5606,7 @@ class Video(object):
         options.update(self.options)
 
         #print(options)
-        self.container = av.open(self.filename, mode="w")
+        self.container = av.open(str(self.filename), mode="w")
         self.stream = self.container.add_stream(self.encoder, rate=self.framerate, options=options)
         self.stream.width = self.resolution[0]
         self.stream.height = self.resolution[1]
@@ -5602,10 +5632,19 @@ class Video(object):
         Write a frame, called when viewer.render() is called
         while a recording is in progress
         """
-        img = self.viewer.rawimage(resolution=self.resolution, channels=3)
-        frame = av.VideoFrame.from_ndarray(img.data, format="rgb24")
-        for packet in self.stream.encode(frame):
-            self.container.mux(packet)
+        if self.encoder == 'jpg' or self.encoder == 'png':
+            if settings["test_mode"]:
+                #Don't output in subdir
+                fn = f"{self.framecount:06}_{self.basename}.{self.encoder}"
+            else:
+                fn = self.filename / f"{self.framecount:06}_{self.basename}.{self.encoder}"
+            self.framecount += 1
+            self.viewer.image(str(fn), resolution=self.resolution)
+        else:
+            img = self.viewer.rawimage(resolution=self.resolution, channels=3)
+            frame = av.VideoFrame.from_ndarray(img.data, format="rgb24")
+            for packet in self.stream.encode(frame):
+                self.container.mux(packet)
 
     def pause(self):
         """
@@ -5621,6 +5660,15 @@ class Video(object):
         Stop recording, final frames will be written and file closed, ready to play. 
         No further frames will be added to the video
         """
+        if self.encoder == 'jpg' or self.encoder == 'png':
+            self.viewer.recording = None
+            if settings["test_mode"]:
+                #Render the last frame for tests
+                self.viewer.app.render()
+                fn = f"{self.framecount:06}_{self.basename}.{self.encoder}"
+                self.viewer.image(str(fn), resolution=self.resolution)
+            self.framecount = 0
+            return
         # Flush stream
         for packet in self.stream.encode():
             self.container.mux(packet)
@@ -5648,7 +5696,14 @@ class Video(object):
             if image.size == self.resolution[0] * self.resolution[1] * 4:
                 image = image.reshape(self.resolution[0], self.resolution[1], 4)
                 image = image[::,::,:3] #Remove alpha channel
-            #self.encoder.copyframe(image.ravel())
+
+            if self.encoder == 'jpg' or self.encoder == 'png':
+                from PIL import Image as PILImage
+                img = PILImage.fromarray(image)
+                fn = self.filename / f"{self.framecount:06}_{self.basename}.{self.encoder}"
+                self.framecount += 1
+                img.save(fn)
+                return
 
             frame = av.VideoFrame.from_ndarray(image, format="rgb24")
             for packet in self.stream.encode(frame):
@@ -5658,7 +5713,11 @@ class Video(object):
         """
         Show the video in an inline player if in an interactive notebook
         """
-        player(self.filename, **self.player)
+        if self.encoder == 'jpg' or self.encoder == 'png':
+            from IPython.display import display,HTML
+            display(HTML('<p>Video written to images, no player available</p>'))
+        else:
+            player(self.filename, **self.player)
 
     def __enter__(self):
         self.start()
@@ -5671,6 +5730,8 @@ class Video(object):
             self.play()
         else:
             print('Recording failed: ', exc_value)
+            import traceback
+            print(traceback.format_exc())
         return True
 
 #Wrapper class for raw image data

lavavu/shaders/triShader.vert

@@ -8,7 +8,6 @@ uniform mat4 uMVMatrix;
 uniform mat4 uPMatrix;
 uniform mat4 uNMatrix;
 
-uniform vec4 uColour;
 uniform vec4 uLightPos;
 
 out vec4 vColour;
@@ -26,11 +25,10 @@ void main(void)
 
   vNormal = normalize(mat3(uNMatrix) * aVertexNormal);
 
-  if (uColour.a > 0.0)
-    vColour = uColour;
-  else
-    vColour = aVertexColour;
-
+  //This shader only applies attribute colour
+  //uColour is set for other/custom shaders
+  //Note uOpacity is "alpha" prop, "oapcity" is passed via attrib
+  vColour = aVertexColour;
   vTexCoord = aVertexTexCoord;
   vFlatColour = vColour;
   vVertex = aVertexPosition;

lavavu/tracers.py

@@ -24,7 +24,7 @@ def random_particles(count, lowerbound=[0,0,0], upperbound=[1,1,1], dims=3):
     return numpy.stack(p).T
 
 class Tracers():
-    def __init__(self, grid, count=1000, lowerbound=None, upperbound=None, limit=None, age=4, respawn_chance=0.2, speed_multiply=1.0, height=0.0, viewer=None):
+    def __init__(self, grid, count=1000, lowerbound=None, upperbound=None, limit=None, age=4, respawn_chance=0.2, speed_multiply=1.0, height=0.0, label='', viewer=None):
         """
         Seed random particles into a vector field and trace their positions
 
@@ -52,6 +52,8 @@ class Tracers():
             Speed multiplier, scaling factor for the velocity taken from the vector values
         height : float
             A fixed height value, all positions will be given this height as their Z component
+        label : str
+            Name label prefix for the visualisation objects when plotting
         viewer : lavavu.Viewer
             Viewer object for plotting functions
         """
@@ -100,6 +102,7 @@ class Tracers():
         self.speed_multiply = speed_multiply
         self.height = height
 
+        self.label = label
         self.lv = viewer
         self.points = None
         self.arrows = None
@@ -157,8 +160,8 @@ class Tracers():
                 self.speed[r] = 0.0
 
         if self.lv:
-            positions = self.positions
-            if self.dims == 2 and self.height != 0:
+            positions = self.get_positions()
+            if positions.shape[1] == 2 and self.height != 0:
                 #Convert to 3d and set z coord to height
                 shape = list(positions.shape)
                 shape[-1] = 3
@@ -180,15 +183,25 @@ class Tracers():
                 if len(self.tracers["colourmap"]):
                     self.tracers.values(self.speed)
 
+    def get_positions(self):
+        """
+        This funcion can be overridden if positions need to be transformed
+        in any way before plotting, eg: plotting on spherical earth model
+        """
+        return self.positions
+
+    def prefix(self):
+        return self.label + '_' if len(self.label) else ''
+
     def plot_points(self, **kwargs):
         if self.lv is not None and self.points is None:
-            self.points = self.lv.points('tracer_points', **kwargs)
+            self.points = self.lv.points(self.prefix() + 'tracer_points', **kwargs)
 
     def plot_arrows(self, **kwargs):
         if self.lv is not None and self.arrows is None:
-            self.arrows = self.lv.vectors('tracer_arrows', **kwargs)
+            self.arrows = self.lv.vectors(self.prefix() + 'tracer_arrows', **kwargs)
 
     def plot_tracers(self, **kwargs):
         if self.lv is not None and self.tracers is None:
-            self.tracers = self.lv.tracers('tracers', dims=self.count, limit=self.limit, **kwargs)
+            self.tracers = self.lv.tracers(self.prefix() + 'tracers', dims=self.count, limit=self.limit, **kwargs)
 

{lavavu-1.9.6.dist-info → lavavu-1.9.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lavavu
-Version: 1.9.6
+Version: 1.9.8
 Summary: Python interface to LavaVu OpenGL 3D scientific visualisation utilities
 Author-email: Owen Kaluza <owen@kaluza.id.au>
 License: ### Licensing
@@ -224,7 +224,7 @@ Dynamic: license-file
 [![Build Status](https://github.com/lavavu/LavaVu/workflows/Test/badge.svg)](https://github.com/lavavu/LavaVu/actions?query=workflow:Test)
 [![Deploy Status](https://github.com/lavavu/LavaVu/workflows/Deploy/badge.svg?branch=1.7.3)](https://github.com/lavavu/LavaVu/actions?query=workflow:Deploy)
 [![DOI](https://zenodo.org/badge/45163055.svg)](https://zenodo.org/badge/latestdoi/45163055)
-[![Binder](https://mybinder.org/badge_logo.svg)](https://mybinder.org/v2/gh/lavavu/LavaVu/1.9.6)
+[![Binder](https://mybinder.org/badge_logo.svg)](https://mybinder.org/v2/gh/lavavu/LavaVu/1.9.8)
 
 A scientific visualisation tool with a python interface for fast and flexible visual analysis.
 

{lavavu-1.9.6.dist-info → lavavu-1.9.8.dist-info}/RECORD

@@ -1,10 +1,16 @@
-lavavu/lavavu.py,sha256=I64ioPwTN1eyZcZEaF9RfdxncfLuihl-7VqqqJDFLCw,218528
+lavavu-1.9.8.dist-info/RECORD,,
+lavavu-1.9.8.dist-info/WHEEL,sha256=sunMa2yiYbrNLGeMVDqEA0ayyJbHlex7SCn1TZrEq60,136
+lavavu-1.9.8.dist-info/entry_points.txt,sha256=LC2qXR6EMe45Cb7zGAF99R9HFrAECP6Qkp_YuG6HZB0,44
+lavavu-1.9.8.dist-info/top_level.txt,sha256=JptS0k1nlBumjLs_0hITr3_XUJhxqvKBXD2jGho3E3A,7
+lavavu-1.9.8.dist-info/METADATA,sha256=Bt4QlWtORr3Qydbw-2b9fbksdAAVzpjrtokM6hOaGBk,17877
+lavavu-1.9.8.dist-info/licenses/LICENSE.md,sha256=EhfNgC6BYh5gDEaq4tcrN3S0wbO4CtJO46botJnCF8c,8603
+lavavu/lavavu.py,sha256=v7tXwp6gQzu1F05hHyALng8FXznv4rVwdCJ3sO65O8o,221223
 lavavu/vutils.py,sha256=6Vm_xl1bp9mWlfk7jgLDwbRw-tdE_oxin77YkLel3_4,5437
-lavavu/dict.json,sha256=lsZEHc7Bb6_lt7tkSBQMgkq7AEn_2hnhWzzdokmvIY8,52729
+lavavu/dict.json,sha256=EcJsoHsiHDDmM8FGuBRF6eOG2nMr-Pda3sxE6v6KB80,52952
 lavavu/server.py,sha256=L-_yPCbNfwYxJCPjDQtr_lxPnDp4oMNVFyxXhBERYrQ,12468
 lavavu/points.py,sha256=jRRtA6CrcA46Y2jUJmkxnIiVoMC5a14xEd_ojhmjhz4,5871
 lavavu/LavaVuPython.py,sha256=ixhceiAr0gGGfCe3d2cY0IIgS4hbgoYJGa_RfgAmmrU,33207
-lavavu/tracers.py,sha256=-NwdBAj5vYWmLt4NE0gWl36SQShQhICR7vBE4rzNOFI,8488
+lavavu/tracers.py,sha256=e10H8evUH7V-XUSkHjfoakaA4SfeuQ8-gWzfa86RGzQ,8997
 lavavu/convert.py,sha256=tbYRjLE2l1hI4d6tsW41Lia1JXmrWSc0-JAYCiMrjys,35516
 lavavu/control.py,sha256=s32rtLPXXYtxpeXd6-iHdupmaMTJ3KhK6Vq-CLjf9OQ,66755
 lavavu/__init__.py,sha256=JroZQiUbuVN7ifixk2zNDGlyLGaM8bqfRbw4D_F9qIQ,191
@@ -12,9 +18,9 @@ lavavu/amalgamate.py,sha256=Xloq1IZ4VUvYiROzQtwKcQIWC65c7EZrdiGVhZdolL8,586
 lavavu/aserver.py,sha256=SfFvLeDTcx1XtS8fY_HIrDmh3q9HicCBRSAriY5yyOE,12003
 lavavu/font.bin,sha256=fvi5zkvmq6gh9v3jXedBVuxNJWKmHtbjECzv6eT9wb4,225360
 lavavu/__main__.py,sha256=EbDetijCjyoiNxmExqnDGoRVs996tSVave5DML9zuMY,235
-lavavu/_LavaVuPython.cpython-311-darwin.so,sha256=4f_OIq_Z1CDwE96EOJLY_E-4pHxMRE1Quk5JEHHtqSM,3927376
+lavavu/_LavaVuPython.cpython-311-darwin.so,sha256=qBWKuihjidXcRLZWJ15TSKWzXV1tJCxuXgIWW30q1sw,3927616
 lavavu/html/control.css,sha256=PVLwmle00ciEckUV5ZIRmI3Whxhl7-mlVexnyyxoU70,3259
-lavavu/html/control.js,sha256=oV24eJZ0vfuCxF3fO60eeJSkNFJ9hUkSCThHGJS1Wvc,11652
+lavavu/html/control.js,sha256=dBjFHgDal3aixVWoQuMOt_ZyhiGSAcEUFJJx9I_ndOQ,11719
 lavavu/html/gui.css,sha256=PRDLsYwM6cgLC9zZsEAG0dnnSd-HYH6oSEfsdEBkuxk,582
 lavavu/html/favicon.ico,sha256=OrIWwvxOSnOCuynrGRUyEIVQng0ZwA9Rrz89S9eiog0,1150
 lavavu/html/LavaVu-amalgamated.css,sha256=iE2xrxFcwmY0AcASrXxNa_RpvFEbS_YO4H5OILbPteE,8640
@@ -27,7 +33,7 @@ lavavu/html/dat.gui.min.js,sha256=S4_QjoXe4IOpU0f0Sj5jEQLTWPoX9uRl1ohB91jyhuw,56
 lavavu/html/OK-min.js,sha256=-4Gc1-dWfxP_w6r9ZOHa8u-X2BlGUoSQbX68lwCxQ3E,39115
 lavavu/html/emscripten.css,sha256=wkaIJhXaxuMchinQX9Z8c12cJomyvFQMeIZ62WGQEPQ,1813
 lavavu/html/drawbox.js,sha256=SJxFSmWd7QVFI5__66hFkKzLKeqg1JPcx-gJuqdMkXw,34590
-lavavu/html/webview.html,sha256=8eilGjX8y_I5OjCvTm-ZAQDNOSCaT1yvogKFqK4lCS0,1521
+lavavu/html/webview.html,sha256=f67d1NMRiSRYYjbSEtwpzzouzfK42EamrqEfoBPrh7k,1521
 lavavu/html/emscripten-template.js,sha256=h63mzl3Lv7aQT1wMOiOucPOvHTJjpKkzsL-SFVYaZlA,6135
 lavavu/html/draw.js,sha256=57LlHlYRh0IvWVwzGDnF6PaCxYLzokpokLNCuZlG1ds,84108
 lavavu/html/dat-gui-light-theme.css,sha256=uPhvJs-1IAsdxudItyOw8lZy8Hrih0zmFM1u-xRwZ-M,1142
@@ -41,15 +47,9 @@ lavavu/shaders/lineShader.frag,sha256=2PQBMcb149jVKnsBhTvU1vf94x91O0kAdOBTSs8OpG
 lavavu/shaders/fontShader.frag,sha256=clgomy2lu-1u7w0GwFd0BNU_L3-SkYt8zv03N5ciJug,351
 lavavu/shaders/triShader.frag,sha256=SkPatq2s7Osa5o7U3lg-7We8ldY0oHZHCLuQjRY4Vgk,4601
 lavavu/shaders/default.vert,sha256=3xKybexF39UGsESMLQQcqHlE1MgLCDEXUMLz4sr5blQ,352
-lavavu/shaders/triShader.vert,sha256=H7rTS-WO5QiT9uP1ClhufQM8minSWknJ9UvKAEVnzRU,1078
+lavavu/shaders/triShader.vert,sha256=U_G_74Jybaw20-PI-NDgaLDDfxGhURHMxdvNYXEkBjc,1156
 lavavu/shaders/lineShader.vert,sha256=5HDJMphXW438yMc0I2NtBdnvF6uXrOUac9hjk3S7mfk,501
 lavavu/shaders/fontShader.vert,sha256=yCBmRugaPUeUQUdIh62-AK_5KELiJqVS2M82iyIqPwo,290
 lavavu/shaders/pointShader.frag,sha256=3zREBdsimlL9fAXBPjhzomGaFUWmlG_QYkhwMTVURHQ,3291
 lavavu/shaders/volumeShader.vert,sha256=uGdQjGqi7V5kE6V7nxymfugtU4cbf6u570xBy13RgmY,78
 lavavu/shaders/default.frag,sha256=5XLYVfLwzN0sFT3aMYGmxbyquA_cmndp55YCOuy1t4E,180
-lavavu-1.9.6.dist-info/RECORD,,
-lavavu-1.9.6.dist-info/WHEEL,sha256=EfZGgIDXTDki1cNGEkgBr7YEgShBrw-nMmb-y67Rxxs,136
-lavavu-1.9.6.dist-info/entry_points.txt,sha256=LC2qXR6EMe45Cb7zGAF99R9HFrAECP6Qkp_YuG6HZB0,44
-lavavu-1.9.6.dist-info/top_level.txt,sha256=JptS0k1nlBumjLs_0hITr3_XUJhxqvKBXD2jGho3E3A,7
-lavavu-1.9.6.dist-info/METADATA,sha256=EHgjIlNax60denK4kKXeAJziT1zFhX8e-9OqQUVd9WE,17877
-lavavu-1.9.6.dist-info/licenses/LICENSE.md,sha256=EhfNgC6BYh5gDEaq4tcrN3S0wbO4CtJO46botJnCF8c,8603

{lavavu-1.9.6.dist-info → lavavu-1.9.8.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (78.1.0)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: false
 Tag: cp311-cp311-macosx_11_0_arm64
 Generator: delocate 0.13.0