wavekit 0.5.3__tar.gz → 0.5.5__tar.gz

{wavekit-0.5.3 → wavekit-0.5.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wavekit
-Version: 0.5.3
+Version: 0.5.5
 Summary: a fundamental package for digital circuit waveform analysis
 License-File: LICENSE
 Author: cxzzzz

{wavekit-0.5.3 → wavekit-0.5.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "wavekit"
-version = "0.5.3"
+version = "0.5.5"
 description = "a fundamental package for digital circuit waveform analysis"
 authors = ["cxzzzz <cxz19961010@outlook.com>"]
 readme = "README.md"

{wavekit-0.5.3 → wavekit-0.5.5}/src/wavekit/__init__.py

@@ -34,11 +34,35 @@ __all__ = [
     'MatchResult',
     'MatchStatus',
     'PatternError',
+    'has_fsdb_support',
 ]
 
-try:
-    from .readers.fsdb.reader import FsdbReader as FsdbReader
+# Detect if FSDB support is available (compiled with VERDI_HOME)
+import importlib.util
+
+_fsdb_available = (
+    importlib.util.find_spec('.readers.fsdb.npi_fsdb_reader', package=__name__) is not None
+)
+
+
+def has_fsdb_support() -> bool:
+    """Check if FsdbReader is available in the current installation."""
+    return _fsdb_available
 
-    __all__.append('FsdbReader')
-except ImportError:
-    pass
+
+class _FsdbReaderStub:
+    """Placeholder that raises an error when FsdbReader is used without Verdi support."""
+
+    def __init__(self, *args, **kwargs):
+        raise RuntimeError(
+            'FsdbReader requires Verdi to be installed and configured.\n\n'
+            '1. Set VERDI_HOME environment variable to your Verdi installation path.\n'
+            '2. Reinstall wavekit:\n'
+            '    pip install --force-reinstall --no-cache-dir --no-deps wavekit'
+        )
+
+
+if _fsdb_available:
+    from .readers.fsdb.reader import FsdbReader as FsdbReader
+else:
+    FsdbReader = _FsdbReaderStub  # type: ignore[assignment]

{wavekit-0.5.3 → wavekit-0.5.5}/src/wavekit/readers/fsdb/reader.py

@@ -138,7 +138,13 @@ class FsdbReader(Reader):
             import os
             import sys
 
-            rel_lib_path = os.environ['VERDI_HOME'] + '/share/NPI/python'
+            verdi_home = os.environ.get('VERDI_HOME')
+            if verdi_home is None:
+                raise RuntimeError(
+                    'VERDI_HOME environment variable is not set.\n'
+                    'Please set VERDI_HOME to your Verdi installation path.'
+                )
+            rel_lib_path = verdi_home + '/share/NPI/python'
             sys.path.append(os.path.abspath(rel_lib_path))
             FsdbReader.pynpi['npisys'] = importlib.import_module('pynpi.npisys')
             FsdbReader.pynpi['waveform'] = importlib.import_module('pynpi.waveform')

{wavekit-0.5.3 → wavekit-0.5.5}/src/wavekit/waveform.py

@@ -130,17 +130,30 @@ class Waveform:
         dense sampled waveform to just the value-change events.
 
         Returns a new :class:`Waveform` where no two adjacent entries have the
-        same value.  :meth:`compress` is an alias for this method.
+        same value.  Behavior matches :func:`numpy.unique_consecutive`.
         """
         if len(self.value) <= 1:
             return self.copy()
-        diff_mask = np.diff(self.value) != 0
-        padded_diff_mask = np.concatenate(([1], diff_mask[:-1], [1]))
-        diff_indices = np.where(padded_diff_mask)[0]
-        return self.take(diff_indices)
+        # Keep first element, then keep elements where value changed from previous
+        mask = np.concatenate(([True], np.diff(self.value) != 0))
+        return self.take(np.where(mask)[0])
 
     def compress(self) -> Waveform:
-        return self.unique_consecutive()
+        """Remove consecutive duplicate values, keeping the last occurrence.
+
+        Unlike :meth:`unique_consecutive`, this preserves the final timestamp
+        and value, which is useful for waveforms where the end time matters.
+
+        Returns a new :class:`Waveform` where no two adjacent entries have the
+        same value, except the last sample is always preserved.
+        """
+        if len(self.value) <= 1:
+            return self.copy()
+        diff_mask = np.diff(self.value) != 0
+        # Keep where next value changed (which means current is last of its group)
+        # plus always keep last element
+        mask = np.concatenate((diff_mask, [True]))
+        return self.take(np.where(mask)[0])
 
     def vectorized_filter(
         self,