pyvlasiator 0.1.0__tar.gz → 0.1.2__tar.gz

{pyvlasiator-0.1.0 → pyvlasiator-0.1.2}/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.1
 Name: pyvlasiator
-Version: 0.1.0
-Summary: 
+Version: 0.1.2
+Summary: Python utilities for processing Vlasiator data
 Author: Hongyang Zhou
 Author-email: hyzhou@umich.edu
 Requires-Python: >=3.11,<4.0
@@ -12,7 +12,6 @@ Classifier: Programming Language :: Python :: 3.13
 Provides-Extra: plot
 Requires-Dist: matplotlib (>=3.7.2,<4.0.0) ; extra == "plot"
 Requires-Dist: numpy (>=1.25.1,<2.0.0)
-Requires-Dist: pytest-benchmark (>=4.0.0,<5.0.0)
 Description-Content-Type: text/markdown
 
 # pyvlasiator
@@ -39,7 +38,7 @@ Lightweight Python package for processing Vlasiator data.
 `pyvlasiator` can be installed via
 
 ```bash
-$ pip install pyvlasiator
+python -m pip install pyvlasiator
 ```
 
 ## Usage

{pyvlasiator-0.1.0 → pyvlasiator-0.1.2}/README.md

@@ -22,7 +22,7 @@ Lightweight Python package for processing Vlasiator data.
 `pyvlasiator` can be installed via
 
 ```bash
-$ pip install pyvlasiator
+python -m pip install pyvlasiator
 ```
 
 ## Usage

{pyvlasiator-0.1.0 → pyvlasiator-0.1.2}/pyproject.toml

@@ -1,7 +1,7 @@
 [tool.poetry]
 name = "pyvlasiator"
-version = "0.1.0"
-description = ""
+version = "0.1.2"
+description = "Python utilities for processing Vlasiator data"
 authors = ["Hongyang Zhou <hyzhou@umich.edu>"]
 readme = "README.md"
 
@@ -9,13 +9,13 @@ readme = "README.md"
 python = "^3.11"
 numpy = "^1.25.1"
 matplotlib = {version = "^3.7.2", optional = true}
-pytest-benchmark = "^4.0.0"
 
 [tool.poetry.extras]
 plot = ["matplotlib"]
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^7.4.0"
+pytest-benchmark = "^4.0.0"
 pytest-cov = "^4.1.0"
 requests = "^2.31.0"
 jupyter = "^1.0.0"

{pyvlasiator-0.1.0 → pyvlasiator-0.1.2}/src/pyvlasiator/plot/plot.py

@@ -867,6 +867,7 @@ def vdfslice(
     center: str = None,
     weight: str = "particle",
     flimit: float = -1.0,
+    addcolorbar: bool = True,
     **kwargs,
 ):
     v1, v2, r1, r2, weights, strx, stry, str_title = prep_vdf(
@@ -908,9 +909,10 @@ def vdfslice(
     ax.grid(color="grey", linestyle="-")
     ax.tick_params(direction="in")
 
-    cb = plt.colorbar(h[3], ax=ax, fraction=0.04, pad=0.02)
-    cb.ax.tick_params(which="both", direction="in")
-    cb_title = cb.ax.set_ylabel("f(v)")
+    if addcolorbar:
+        cb = plt.colorbar(h[3], ax=ax, fraction=0.04, pad=0.02)
+        cb.ax.tick_params(which="both", direction="in")
+        cb_title = cb.ax.set_ylabel("f(v)")
 
     # TODO: Draw vector of magnetic field direction
     # if slicetype in ("xy", "xz", "yz"):

{pyvlasiator-0.1.0 → pyvlasiator-0.1.2}/src/pyvlasiator/vlsv/reader.py

@@ -405,14 +405,16 @@ class Vlsv:
         )
         if hasattr(cellids, "__len__"):  # part of cells requested
             return np.float32(raw)
-
-        if sorted:
-            if raw.ndim == 1:
-                v = raw[self.cellindex]
+        elif cellids == -1:
+            if sorted:
+                if raw.ndim == 1:
+                    v = raw[self.cellindex]
+                else:
+                    v = raw[self.cellindex, :]
+                if v.dtype == np.float64:  # 32-bit is enough for analysis
+                    v = np.float32(v)
             else:
-                v = raw[self.cellindex, :]
-            if v.dtype == np.float64:  # 32-bit is enough for analysis
-                v = np.float32(v)
+                v = raw
         else:
             v = raw