figpack 0.2.27__py3-none-any.whl → 0.2.28__py3-none-any.whl

figpack/__init__.py

@@ -2,7 +2,7 @@
 figpack - A Python package for creating shareable, interactive visualizations in the browser
 """
 
-__version__ = "0.2.27"
+__version__ = "0.2.28"
 
 from .cli import view_figure
 from .core import FigpackView, FigpackExtension, ExtensionView

figpack/views/TimeseriesGraph.py

@@ -137,6 +137,7 @@ class TimeseriesGraph(FigpackView):
         width: float = 1.0,
         channel_spacing: Optional[float] = None,
         auto_channel_spacing: Optional[float] = None,
+        timestamps_for_inserting_nans: Optional[np.ndarray] = None,
     ) -> None:
         """
         Add a uniform timeseries to the graph with optional multi-channel support
@@ -151,6 +152,7 @@ class TimeseriesGraph(FigpackView):
             width: Line width
             channel_spacing: Vertical spacing between channels
             auto_channel_spacing: sets channel spacing to this multiple of the estimated RMS noise level
+            timestamps_for_inserting_nans: Optional array of timestamps used to determine where to insert NaNs in the data
         """
         self._series.append(
             TGUniformSeries(
@@ -163,6 +165,7 @@ class TimeseriesGraph(FigpackView):
                 width=width,
                 channel_spacing=channel_spacing,
                 auto_channel_spacing=auto_channel_spacing,
+                timestamps_for_inserting_nans=timestamps_for_inserting_nans,
             )
         )
 
@@ -320,6 +323,7 @@ class TGUniformSeries:
         width: float = 1.0,
         channel_spacing: Optional[float] = None,
         auto_channel_spacing: Optional[float] = None,
+        timestamps_for_inserting_nans: Optional[np.ndarray] = None,
     ):
         assert sampling_frequency_hz > 0, "Sampling frequency must be positive"
 
@@ -340,16 +344,27 @@ class TGUniformSeries:
         self.sampling_frequency_hz = sampling_frequency_hz
         self.data = data.astype(np.float32)  # Ensure float32 for efficiency
 
+        if timestamps_for_inserting_nans is not None:
+            self.data = insert_nans_based_on_timestamps(
+                self.data,
+                start_time_sec=start_time_sec,
+                sampling_frequency_hz=sampling_frequency_hz,
+                timestamps=timestamps_for_inserting_nans,
+            )
+
         if auto_channel_spacing is not None:
             if channel_spacing is not None:
                 raise ValueError(
                     "Specify either channel_spacing or auto_channel_spacing, not both."
                 )
             # Estimate RMS noise level across all channels using median absolute deviation
-            mad = np.median(np.abs(self.data - np.median(self.data, axis=0)), axis=0)
+            # Use nanmedian to handle NaN values properly
+            mad = np.nanmedian(
+                np.abs(self.data - np.nanmedian(self.data, axis=0)), axis=0
+            )
             rms_estimate = mad / 0.6745  # Convert MAD to RMS estimate
-            channel_spacing = auto_channel_spacing * np.median(rms_estimate)
-            if channel_spacing <= 0:
+            channel_spacing = auto_channel_spacing * np.nanmedian(rms_estimate)
+            if channel_spacing <= 0 or np.isnan(channel_spacing):
                 channel_spacing = 1.0  # Fallback to default spacing if estimate fails
         self.channel_spacing = channel_spacing
 
@@ -559,3 +574,25 @@ class TGUniformSeries:
                 data=downsampled_array,
                 chunks=ds_chunks,
             )
+
+
+def insert_nans_based_on_timestamps(
+    x: np.ndarray,
+    *,
+    start_time_sec: float,
+    sampling_frequency_hz: float,
+    timestamps: np.ndarray,
+):
+    end_timestamps = timestamps[-1]
+    ret_length = int((end_timestamps - start_time_sec) * sampling_frequency_hz) + 1
+
+    # Handle both 1D and 2D (multi-channel) data
+    if x.ndim == 1:
+        ret = np.nan * np.ones((ret_length,), dtype=x.dtype)
+    else:  # x.ndim == 2
+        n_channels = x.shape[1]
+        ret = np.nan * np.ones((ret_length, n_channels), dtype=x.dtype)
+
+    indices = ((timestamps - start_time_sec) * sampling_frequency_hz).astype(int)
+    ret[indices] = x
+    return ret

{figpack-0.2.27.dist-info → figpack-0.2.28.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: figpack
-Version: 0.2.27
+Version: 0.2.28
 Summary: A Python package for creating shareable, interactive visualizations in the browser
 Author-email: Jeremy Magland <jmagland@flatironinstitute.org>
 License: Apache-2.0

{figpack-0.2.27.dist-info → figpack-0.2.28.dist-info}/RECORD

@@ -1,4 +1,4 @@
-figpack/__init__.py,sha256=zxhqr0lde2J_mnAi73-7r4CZptcqrtGZIZB7Tj5AJos,358
+figpack/__init__.py,sha256=ZIM-xRn0YXJHMil7YxRrnjEResxKD-UXM9PQN8sBLOg,358
 figpack/cli.py,sha256=s1mGQuFSntxiIvU6OWwHVlM9Cj-l1zMQ3OzFFe1-5ZE,11089
 figpack/extensions.py,sha256=mILB4_F1RHkca4I7t88zh74IX8VCmfT7XFZZT4XYdNw,13009
 figpack/core/__init__.py,sha256=7zU6O1piTk07aeCfbU81QqTgSHIO2n5MZ4LFNmsrtfs,192
@@ -33,15 +33,15 @@ figpack/views/Spectrogram.py,sha256=jcm26ucHedKDnBA5xnAUu9tW-g-ZutT-kw1EIhYm66E,
 figpack/views/Splitter.py,sha256=BR2L-8aqicTubS1rSzsQ3XnhoJcX5GcfEnVWtEWEs0w,2016
 figpack/views/TabLayout.py,sha256=AqdHPLcP2-caWjxbkC8r8m60z8n_eyZrIBGOOPSVNCs,1908
 figpack/views/TabLayoutItem.py,sha256=xmHA0JsW_6naJze4_mQuP_Fy0Nm17p2N7w_AsmVRp8k,880
-figpack/views/TimeseriesGraph.py,sha256=ygbgB83CIc5z_FN6gedWB_KIGh9Zo6LwrCafJqWckfU,19011
+figpack/views/TimeseriesGraph.py,sha256=hhHhtbgaYMldWeSg76M8WRf9ZkI7-1xc3bLWX3c4Acg,20448
 figpack/views/__init__.py,sha256=V09R6vFRzhY7ANevWomM7muFfUieXZEjGimPiMHpey4,641
 figpack/views/PlotlyExtension/PlotlyExtension.py,sha256=LOFSqbm46UZ7HsHTDxUPnNB33ydYQvEkRVK-TSKkzK4,2149
 figpack/views/PlotlyExtension/__init__.py,sha256=80Wy1mDMWyagjuR99ECxJePIYpRQ6TSyHkB0uZoBZ_0,70
 figpack/views/PlotlyExtension/_plotly_extension.py,sha256=yZjG1NMGlQedeeLdV6TQWpi_NTm5Wfk5eWbXEdZbbFE,1455
 figpack/views/PlotlyExtension/plotly_view.js,sha256=9BjgOPkqGl87SSonnb48nFeQV3UTIi1trpSPxd9qlKo,3055
-figpack-0.2.27.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-figpack-0.2.27.dist-info/METADATA,sha256=1_RNdRpyaQQaJD0ZoqyWjWNAn2WcgrIdlfDLfFGYtes,4618
-figpack-0.2.27.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-figpack-0.2.27.dist-info/entry_points.txt,sha256=l6d3siH2LxXa8qJGbjAqpIZtI5AkMSyDeoRDCzdrUto,45
-figpack-0.2.27.dist-info/top_level.txt,sha256=lMKGaC5xWmAYBx9Ac1iMokm42KFnJFjmkP2ldyvOo-c,8
-figpack-0.2.27.dist-info/RECORD,,
+figpack-0.2.28.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+figpack-0.2.28.dist-info/METADATA,sha256=SexqCXW831vYDik27uUYBKr2EokWoXCYpgL0mgG3EbM,4618
+figpack-0.2.28.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+figpack-0.2.28.dist-info/entry_points.txt,sha256=l6d3siH2LxXa8qJGbjAqpIZtI5AkMSyDeoRDCzdrUto,45
+figpack-0.2.28.dist-info/top_level.txt,sha256=lMKGaC5xWmAYBx9Ac1iMokm42KFnJFjmkP2ldyvOo-c,8
+figpack-0.2.28.dist-info/RECORD,,