PySDKit 0.4.44__tar.gz → 0.4.46__tar.gz

{pysdkit-0.4.44 → pysdkit-0.4.46}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PySDKit
-Version: 0.4.44
+Version: 0.4.46
 Summary: A Python library for signal decomposition algorithms with a unified interface.
 Author: Whenxuan Wang, RuiZhe Wang, Rongkun Zhu, Kai Wu, Lei Wang, josefinez, Deeksha Manjunath, Yuan Feng, WenTong Zhao, JacktheFowler
 Maintainer-email: Whenxuan Wang <wwhenxuan@gmail.com>, RuiZhe Wang <3133986068@qq.com>
@@ -51,10 +51,10 @@ Requires-Dist: tqdm>=4.66.5
 Requires-Dist: requests>=2.32.3
 Dynamic: license-file
 
-# PySDKit: signal decomposition in Python
-
 <div align="center">
 
+# PySDKit: signal decomposition in Python
+
 [![PyPI version](https://badge.fury.io/py/PySDKit.svg)](https://pypi.org/project/PySDKit/) 
 ![License](https://img.shields.io/github/license/wwhenxuan/PySDKit)
 [![Python](https://img.shields.io/badge/python-3.8+-blue?logo=python)](https://www.python.org/)
@@ -91,50 +91,57 @@ This project integrates simple signal processing methods, signal decomposition a
 4. Visualize and analyze the original signal and the intrinsic mode functions IMFs obtained by decomposition.
 
 ~~~python
+import numpy as np
 from pysdkit import EMD
-from pysdkit.data import test_emd
 from pysdkit.plot import plot_IMFs
 
-t, signal = test_emd()
+t = np.linspace(0, 1, 1000)
+signal = np.sin(2 * np.pi * 5 * t) + 0.7 * np.sin(2 * np.pi * 25 * t) + 0.45 * np.sin(2 * np.pi * 80 * t)
 
-# create an instance for signal decomposition
 emd = EMD()
-# implement signal decomposition
-IMFs = emd.fit_transform(signal, max_imfs=2)
-plot_IMFs(signal, IMFs)
+IMFs = emd.fit_transform(signal, max_imfs=3)
+plot_IMFs(signal, IMFs, view="2d_freq", fs=1000, freq_max=150)
 ~~~
 
-![example](https://raw.githubusercontent.com/wwhenxuan/PySDKit/main/images/example.jpg)
+![example](https://raw.githubusercontent.com/wwhenxuan/PySDKit/main/images/decomposition_demo.jpg)
 
 The EMD in the above example is the most classic [`empirical mode decomposition`](https://www.mathworks.com/help/signal/ref/emd.html) algorithm in signal decomposition. For more complex signals, you can try other algorithms such as variational mode decomposition ([`VMD`](https://ieeexplore.ieee.org/abstract/document/6655981)).
 
 ~~~python
-import numpy as np
 from pysdkit import VMD
+from pysdkit.data import test_vmd
+from pysdkit.plot import plot_IMFs
 
-# load new signal
-signal = np.load("./example/example.npy")
-
-# use variational mode decomposition
-vmd = VMD(alpha=500, K=3, tau=0.0, tol=1e-9)
-IMFs = vmd.fit_transform(signal=signal)
-print(IMFs.shape)
+t, signal, fs = test_vmd()
 
-vmd.plot_IMFs(save_figure=True)
+vmd = VMD(alpha=2000, K=4, tau=0.0, tol=1e-7)
+IMFs = vmd.fit_transform(signal)
+plot_IMFs(signal, IMFs, view="2d_freq", fs=fs, freq_max=fs / 2)
 ~~~
 
 ![vmd_example](https://raw.githubusercontent.com/wwhenxuan/PySDKit/main/images/vmd_example.jpg)
 
-Better observe the characteristics of the decomposed intrinsic mode function in the frequency domain.
+For multichannel recordings, algorithms such as multivariate VMD ([`MVMD`](https://doi.org/10.1109/TSP.2019.2951223)) keep shared oscillations **mode-aligned** across channels:
 
 ~~~python
-from pysdkit.plot import plot_IMFs_amplitude_spectra
+import numpy as np
+from pysdkit import MVMD
+from pysdkit.plot import plot_IMFs
 
-# frequency domain visualization
-plot_IMFs_amplitude_spectra(IMFs, smooth="exp")   # use exp smooth
+t = np.arange(0, 1, 0.001)
+# ch1: 2+36 Hz, ch2: 24+36 Hz, ch3: 80+36 Hz (36 Hz shared)
+signal = np.vstack([
+    np.cos(2*np.pi*2*t) + np.cos(2*np.pi*36*t),
+    np.cos(2*np.pi*24*t) + np.cos(2*np.pi*36*t),
+    np.cos(2*np.pi*80*t) + np.cos(2*np.pi*36*t),
+])
+
+mvmd = MVMD(alpha=2000, K=4, tau=0.0, init="uniform")
+IMFs = mvmd.fit_transform(signal)   # shape: (K, T, C)
+plot_IMFs(signal, IMFs)             # per-channel panels
 ~~~
 
-![frequency_example](https://raw.githubusercontent.com/wwhenxuan/PySDKit/main/images/frequency_example.jpg)
+![mvmd_example](https://raw.githubusercontent.com/wwhenxuan/PySDKit/main/images/mvmd_example.jpg)
 
 ## Target 🎯 <a id="Target"></a>
 
@@ -193,6 +200,7 @@ plot_IMFs_amplitude_spectra(IMFs, smooth="exp")   # use exp smooth
 | [`SJMD`](https://github.com/wwhenxuan/PySDKit/blob/main/pysdkit/_jmd/sjmd.py) / `SMJMD` (Successive Jump and Mode Decomposition) |         [[paper]](https://arxiv.org/abs/2504.08453)          |                          [[code]]()                          |   ✔️   |
 | [`ESMD`](https://github.com/wwhenxuan/PySDKit/blob/main/pysdkit/_emd/esmd.py) (Extreme-Point Symmetric Mode Decomposition) |          [[paper]](https://arxiv.org/abs/1303.6540)          |         [[code]](https://github.com/WuShichao/esmd)          |   ✔️   |
 |   [`STNBMD`]() (Short-Time Narrow-Band Mode Decomposition)   | [[paper]](https://www.sciencedirect.com/science/article/pii/S0022460X16002443?via%3Dihub) | [[code]](https://ww2.mathworks.cn/matlabcentral/fileexchange/56226-short-time-narrow-band-mode-decomposition-stnbmd-toolbox) |   ✖️   |
+| [`VTFMTD`](https://github.com/wwhenxuan/PySDKit/blob/main/pysdkit/_tfa/vftmtd.py) (Variational TF Mode Tracking Decomposition) | [[paper]](https://www.sciencedirect.com/science/article/pii/S0165168426001179) | [[code]](https://www.mathworks.com/matlabcentral/fileexchange/183389-variational-time-frequency-mode-tracking-decomposition) |   ✔️   |
 | [`SWD`](https://github.com/wwhenxuan/PySDKit/blob/main/pysdkit/_osd/swd.py) (Swarm Decomposition) |   [[paper]](https://doi.org/10.1016/j.sigpro.2016.09.004)    |  [[code]](https://github.com/gkaposto/Swarm-Decomposition)   |   ✔️   |
 | [`OSD`](https://github.com/wwhenxuan/PySDKit/blob/main/pysdkit/_osd/osd.py) (Optimization-based Signal Decomposition) | [[paper]](https://web.stanford.edu/~boyd/papers/sig_decomp_mprox.html) |   [[code]](https://github.com/cvxgrp/signal-decomposition)   |   ✔️   |
 | [`STL`](https://github.com/wwhenxuan/PySDKit/blob/main/pysdkit/tsa/_stl.py) (Seasonal-Trend decomposition using LOESS) | [[paper]](https://www.nniiem.ru/file/news/2016/stl-statistical-model.pdf) | [[code]](https://www.statsmodels.org/stable/examples/notebooks/generated/stl_decomposition.html) |   ✔️   |

{pysdkit-0.4.44 → pysdkit-0.4.46}/PySDKit.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PySDKit
-Version: 0.4.44
+Version: 0.4.46
 Summary: A Python library for signal decomposition algorithms with a unified interface.
 Author: Whenxuan Wang, RuiZhe Wang, Rongkun Zhu, Kai Wu, Lei Wang, josefinez, Deeksha Manjunath, Yuan Feng, WenTong Zhao, JacktheFowler
 Maintainer-email: Whenxuan Wang <wwhenxuan@gmail.com>, RuiZhe Wang <3133986068@qq.com>
@@ -51,10 +51,10 @@ Requires-Dist: tqdm>=4.66.5
 Requires-Dist: requests>=2.32.3
 Dynamic: license-file
 
-# PySDKit: signal decomposition in Python
-
 <div align="center">
 
+# PySDKit: signal decomposition in Python
+
 [![PyPI version](https://badge.fury.io/py/PySDKit.svg)](https://pypi.org/project/PySDKit/) 
 ![License](https://img.shields.io/github/license/wwhenxuan/PySDKit)
 [![Python](https://img.shields.io/badge/python-3.8+-blue?logo=python)](https://www.python.org/)
@@ -91,50 +91,57 @@ This project integrates simple signal processing methods, signal decomposition a
 4. Visualize and analyze the original signal and the intrinsic mode functions IMFs obtained by decomposition.
 
 ~~~python
+import numpy as np
 from pysdkit import EMD
-from pysdkit.data import test_emd
 from pysdkit.plot import plot_IMFs
 
-t, signal = test_emd()
+t = np.linspace(0, 1, 1000)
+signal = np.sin(2 * np.pi * 5 * t) + 0.7 * np.sin(2 * np.pi * 25 * t) + 0.45 * np.sin(2 * np.pi * 80 * t)
 
-# create an instance for signal decomposition
 emd = EMD()
-# implement signal decomposition
-IMFs = emd.fit_transform(signal, max_imfs=2)
-plot_IMFs(signal, IMFs)
+IMFs = emd.fit_transform(signal, max_imfs=3)
+plot_IMFs(signal, IMFs, view="2d_freq", fs=1000, freq_max=150)
 ~~~
 
-![example](https://raw.githubusercontent.com/wwhenxuan/PySDKit/main/images/example.jpg)
+![example](https://raw.githubusercontent.com/wwhenxuan/PySDKit/main/images/decomposition_demo.jpg)
 
 The EMD in the above example is the most classic [`empirical mode decomposition`](https://www.mathworks.com/help/signal/ref/emd.html) algorithm in signal decomposition. For more complex signals, you can try other algorithms such as variational mode decomposition ([`VMD`](https://ieeexplore.ieee.org/abstract/document/6655981)).
 
 ~~~python
-import numpy as np
 from pysdkit import VMD
+from pysdkit.data import test_vmd
+from pysdkit.plot import plot_IMFs
 
-# load new signal
-signal = np.load("./example/example.npy")
-
-# use variational mode decomposition
-vmd = VMD(alpha=500, K=3, tau=0.0, tol=1e-9)
-IMFs = vmd.fit_transform(signal=signal)
-print(IMFs.shape)
+t, signal, fs = test_vmd()
 
-vmd.plot_IMFs(save_figure=True)
+vmd = VMD(alpha=2000, K=4, tau=0.0, tol=1e-7)
+IMFs = vmd.fit_transform(signal)
+plot_IMFs(signal, IMFs, view="2d_freq", fs=fs, freq_max=fs / 2)
 ~~~
 
 ![vmd_example](https://raw.githubusercontent.com/wwhenxuan/PySDKit/main/images/vmd_example.jpg)
 
-Better observe the characteristics of the decomposed intrinsic mode function in the frequency domain.
+For multichannel recordings, algorithms such as multivariate VMD ([`MVMD`](https://doi.org/10.1109/TSP.2019.2951223)) keep shared oscillations **mode-aligned** across channels:
 
 ~~~python
-from pysdkit.plot import plot_IMFs_amplitude_spectra
+import numpy as np
+from pysdkit import MVMD
+from pysdkit.plot import plot_IMFs
 
-# frequency domain visualization
-plot_IMFs_amplitude_spectra(IMFs, smooth="exp")   # use exp smooth
+t = np.arange(0, 1, 0.001)
+# ch1: 2+36 Hz, ch2: 24+36 Hz, ch3: 80+36 Hz (36 Hz shared)
+signal = np.vstack([
+    np.cos(2*np.pi*2*t) + np.cos(2*np.pi*36*t),
+    np.cos(2*np.pi*24*t) + np.cos(2*np.pi*36*t),
+    np.cos(2*np.pi*80*t) + np.cos(2*np.pi*36*t),
+])
+
+mvmd = MVMD(alpha=2000, K=4, tau=0.0, init="uniform")
+IMFs = mvmd.fit_transform(signal)   # shape: (K, T, C)
+plot_IMFs(signal, IMFs)             # per-channel panels
 ~~~
 
-![frequency_example](https://raw.githubusercontent.com/wwhenxuan/PySDKit/main/images/frequency_example.jpg)
+![mvmd_example](https://raw.githubusercontent.com/wwhenxuan/PySDKit/main/images/mvmd_example.jpg)
 
 ## Target 🎯 <a id="Target"></a>
 
@@ -193,6 +200,7 @@ plot_IMFs_amplitude_spectra(IMFs, smooth="exp")   # use exp smooth
 | [`SJMD`](https://github.com/wwhenxuan/PySDKit/blob/main/pysdkit/_jmd/sjmd.py) / `SMJMD` (Successive Jump and Mode Decomposition) |         [[paper]](https://arxiv.org/abs/2504.08453)          |                          [[code]]()                          |   ✔️   |
 | [`ESMD`](https://github.com/wwhenxuan/PySDKit/blob/main/pysdkit/_emd/esmd.py) (Extreme-Point Symmetric Mode Decomposition) |          [[paper]](https://arxiv.org/abs/1303.6540)          |         [[code]](https://github.com/WuShichao/esmd)          |   ✔️   |
 |   [`STNBMD`]() (Short-Time Narrow-Band Mode Decomposition)   | [[paper]](https://www.sciencedirect.com/science/article/pii/S0022460X16002443?via%3Dihub) | [[code]](https://ww2.mathworks.cn/matlabcentral/fileexchange/56226-short-time-narrow-band-mode-decomposition-stnbmd-toolbox) |   ✖️   |
+| [`VTFMTD`](https://github.com/wwhenxuan/PySDKit/blob/main/pysdkit/_tfa/vftmtd.py) (Variational TF Mode Tracking Decomposition) | [[paper]](https://www.sciencedirect.com/science/article/pii/S0165168426001179) | [[code]](https://www.mathworks.com/matlabcentral/fileexchange/183389-variational-time-frequency-mode-tracking-decomposition) |   ✔️   |
 | [`SWD`](https://github.com/wwhenxuan/PySDKit/blob/main/pysdkit/_osd/swd.py) (Swarm Decomposition) |   [[paper]](https://doi.org/10.1016/j.sigpro.2016.09.004)    |  [[code]](https://github.com/gkaposto/Swarm-Decomposition)   |   ✔️   |
 | [`OSD`](https://github.com/wwhenxuan/PySDKit/blob/main/pysdkit/_osd/osd.py) (Optimization-based Signal Decomposition) | [[paper]](https://web.stanford.edu/~boyd/papers/sig_decomp_mprox.html) |   [[code]](https://github.com/cvxgrp/signal-decomposition)   |   ✔️   |
 | [`STL`](https://github.com/wwhenxuan/PySDKit/blob/main/pysdkit/tsa/_stl.py) (Seasonal-Trend decomposition using LOESS) | [[paper]](https://www.nniiem.ru/file/news/2016/stl-statistical-model.pdf) | [[code]](https://www.statsmodels.org/stable/examples/notebooks/generated/stl_decomposition.html) |   ✔️   |

{pysdkit-0.4.44 → pysdkit-0.4.46}/PySDKit.egg-info/SOURCES.txt

@@ -60,6 +60,12 @@ pysdkit/_osd/__init__.py
 pysdkit/_osd/components.py
 pysdkit/_osd/osd.py
 pysdkit/_osd/swd.py
+pysdkit/_tfa/__init__.py
+pysdkit/_tfa/vtfmtd.py
+pysdkit/_tfa/data/__init__.py
+pysdkit/_tfa/data/dual_signal_noise.npy
+pysdkit/_tfa/data/map2.npy
+pysdkit/_tfa/data/single_nsignal.npy
 pysdkit/_tid/__init__.py
 pysdkit/_tid/alif.py
 pysdkit/_tid/hvd.py
@@ -97,6 +103,7 @@ pysdkit/data/_test_univariate.py
 pysdkit/data/_time_series.py
 pysdkit/data/fmd_demo.npy
 pysdkit/data/texture.txt
+pysdkit/data/vmd_example.npy
 pysdkit/entropy/__init__.py
 pysdkit/entropy/_approxiamte_entropy.py
 pysdkit/entropy/_permutation_entropy.py
@@ -153,6 +160,7 @@ pysdkit/tests/test_vgnmd.py
 pysdkit/tests/test_vmd.py
 pysdkit/tests/test_vmd2d.py
 pysdkit/tests/test_vncmd.py
+pysdkit/tests/test_vtfmtd.py
 pysdkit/tests/data/__init__.py
 pysdkit/tests/data/test_generator_cube.py
 pysdkit/tests/data/test_generator_image.py

{pysdkit-0.4.44 → pysdkit-0.4.46}/README.md

@@ -1,7 +1,7 @@
-# PySDKit: signal decomposition in Python
-
 <div align="center">
 
+# PySDKit: signal decomposition in Python
+
 [![PyPI version](https://badge.fury.io/py/PySDKit.svg)](https://pypi.org/project/PySDKit/) 
 ![License](https://img.shields.io/github/license/wwhenxuan/PySDKit)
 [![Python](https://img.shields.io/badge/python-3.8+-blue?logo=python)](https://www.python.org/)
@@ -38,50 +38,57 @@ This project integrates simple signal processing methods, signal decomposition a
 4. Visualize and analyze the original signal and the intrinsic mode functions IMFs obtained by decomposition.
 
 ~~~python
+import numpy as np
 from pysdkit import EMD
-from pysdkit.data import test_emd
 from pysdkit.plot import plot_IMFs
 
-t, signal = test_emd()
+t = np.linspace(0, 1, 1000)
+signal = np.sin(2 * np.pi * 5 * t) + 0.7 * np.sin(2 * np.pi * 25 * t) + 0.45 * np.sin(2 * np.pi * 80 * t)
 
-# create an instance for signal decomposition
 emd = EMD()
-# implement signal decomposition
-IMFs = emd.fit_transform(signal, max_imfs=2)
-plot_IMFs(signal, IMFs)
+IMFs = emd.fit_transform(signal, max_imfs=3)
+plot_IMFs(signal, IMFs, view="2d_freq", fs=1000, freq_max=150)
 ~~~
 
-![example](https://raw.githubusercontent.com/wwhenxuan/PySDKit/main/images/example.jpg)
+![example](https://raw.githubusercontent.com/wwhenxuan/PySDKit/main/images/decomposition_demo.jpg)
 
 The EMD in the above example is the most classic [`empirical mode decomposition`](https://www.mathworks.com/help/signal/ref/emd.html) algorithm in signal decomposition. For more complex signals, you can try other algorithms such as variational mode decomposition ([`VMD`](https://ieeexplore.ieee.org/abstract/document/6655981)).
 
 ~~~python
-import numpy as np
 from pysdkit import VMD
+from pysdkit.data import test_vmd
+from pysdkit.plot import plot_IMFs
 
-# load new signal
-signal = np.load("./example/example.npy")
-
-# use variational mode decomposition
-vmd = VMD(alpha=500, K=3, tau=0.0, tol=1e-9)
-IMFs = vmd.fit_transform(signal=signal)
-print(IMFs.shape)
+t, signal, fs = test_vmd()
 
-vmd.plot_IMFs(save_figure=True)
+vmd = VMD(alpha=2000, K=4, tau=0.0, tol=1e-7)
+IMFs = vmd.fit_transform(signal)
+plot_IMFs(signal, IMFs, view="2d_freq", fs=fs, freq_max=fs / 2)
 ~~~
 
 ![vmd_example](https://raw.githubusercontent.com/wwhenxuan/PySDKit/main/images/vmd_example.jpg)
 
-Better observe the characteristics of the decomposed intrinsic mode function in the frequency domain.
+For multichannel recordings, algorithms such as multivariate VMD ([`MVMD`](https://doi.org/10.1109/TSP.2019.2951223)) keep shared oscillations **mode-aligned** across channels:
 
 ~~~python
-from pysdkit.plot import plot_IMFs_amplitude_spectra
+import numpy as np
+from pysdkit import MVMD
+from pysdkit.plot import plot_IMFs
 
-# frequency domain visualization
-plot_IMFs_amplitude_spectra(IMFs, smooth="exp")   # use exp smooth
+t = np.arange(0, 1, 0.001)
+# ch1: 2+36 Hz, ch2: 24+36 Hz, ch3: 80+36 Hz (36 Hz shared)
+signal = np.vstack([
+    np.cos(2*np.pi*2*t) + np.cos(2*np.pi*36*t),
+    np.cos(2*np.pi*24*t) + np.cos(2*np.pi*36*t),
+    np.cos(2*np.pi*80*t) + np.cos(2*np.pi*36*t),
+])
+
+mvmd = MVMD(alpha=2000, K=4, tau=0.0, init="uniform")
+IMFs = mvmd.fit_transform(signal)   # shape: (K, T, C)
+plot_IMFs(signal, IMFs)             # per-channel panels
 ~~~
 
-![frequency_example](https://raw.githubusercontent.com/wwhenxuan/PySDKit/main/images/frequency_example.jpg)
+![mvmd_example](https://raw.githubusercontent.com/wwhenxuan/PySDKit/main/images/mvmd_example.jpg)
 
 ## Target 🎯 <a id="Target"></a>
 
@@ -140,6 +147,7 @@ plot_IMFs_amplitude_spectra(IMFs, smooth="exp")   # use exp smooth
 | [`SJMD`](https://github.com/wwhenxuan/PySDKit/blob/main/pysdkit/_jmd/sjmd.py) / `SMJMD` (Successive Jump and Mode Decomposition) |         [[paper]](https://arxiv.org/abs/2504.08453)          |                          [[code]]()                          |   ✔️   |
 | [`ESMD`](https://github.com/wwhenxuan/PySDKit/blob/main/pysdkit/_emd/esmd.py) (Extreme-Point Symmetric Mode Decomposition) |          [[paper]](https://arxiv.org/abs/1303.6540)          |         [[code]](https://github.com/WuShichao/esmd)          |   ✔️   |
 |   [`STNBMD`]() (Short-Time Narrow-Band Mode Decomposition)   | [[paper]](https://www.sciencedirect.com/science/article/pii/S0022460X16002443?via%3Dihub) | [[code]](https://ww2.mathworks.cn/matlabcentral/fileexchange/56226-short-time-narrow-band-mode-decomposition-stnbmd-toolbox) |   ✖️   |
+| [`VTFMTD`](https://github.com/wwhenxuan/PySDKit/blob/main/pysdkit/_tfa/vftmtd.py) (Variational TF Mode Tracking Decomposition) | [[paper]](https://www.sciencedirect.com/science/article/pii/S0165168426001179) | [[code]](https://www.mathworks.com/matlabcentral/fileexchange/183389-variational-time-frequency-mode-tracking-decomposition) |   ✔️   |
 | [`SWD`](https://github.com/wwhenxuan/PySDKit/blob/main/pysdkit/_osd/swd.py) (Swarm Decomposition) |   [[paper]](https://doi.org/10.1016/j.sigpro.2016.09.004)    |  [[code]](https://github.com/gkaposto/Swarm-Decomposition)   |   ✔️   |
 | [`OSD`](https://github.com/wwhenxuan/PySDKit/blob/main/pysdkit/_osd/osd.py) (Optimization-based Signal Decomposition) | [[paper]](https://web.stanford.edu/~boyd/papers/sig_decomp_mprox.html) |   [[code]](https://github.com/cvxgrp/signal-decomposition)   |   ✔️   |
 | [`STL`](https://github.com/wwhenxuan/PySDKit/blob/main/pysdkit/tsa/_stl.py) (Seasonal-Trend decomposition using LOESS) | [[paper]](https://www.nniiem.ru/file/news/2016/stl-statistical-model.pdf) | [[code]](https://www.statsmodels.org/stable/examples/notebooks/generated/stl_decomposition.html) |   ✔️   |

{pysdkit-0.4.44 → pysdkit-0.4.46}/pyproject.toml

@@ -73,3 +73,5 @@ include = ["pysdkit*"]
 "pysdkit._tid.data" = ["*.npy"]
 "pysdkit._imd" = ["data/*.npy"]
 "pysdkit._imd.data" = ["*.npy"]
+"pysdkit._tfa" = ["data/*.npy"]
+"pysdkit._tfa.data" = ["*.npy"]

{pysdkit-0.4.44 → pysdkit-0.4.46}/pysdkit/__init__.py

@@ -2,7 +2,7 @@
 A Python library for signal decomposition algorithms.
 """
 
-__version__ = "0.4.44"
+__version__ = "0.4.46"
 
 # Empirical Mode Decomposition
 from ._emd import EMD
@@ -112,6 +112,9 @@ from ._vncmd import AVNCMD
 # Adaptive Local Iterative Filtering (time iterative decomposition)
 from ._tid import ALIF
 
+# Variational Time-Frequency Mode Tracking Decomposition (TFA)
+from ._tfa import VTFMTD, vtfmtd
+
 # Adaptive Polymorphic Mode Decomposition / Impulsive Mode Decomposition
 from ._imd import APMD, IMD, imd
 
@@ -193,7 +196,7 @@ Optimization-based Signal Decomposition         |    OSD
 Generalized Dispersion Mode Decomposition       |    GDMD
 Variational Generalized Nonlinear Mode Dec.     |    VGNMD
 Improved VGNMD (crossed chirp / dispersive)     |    IVGNMD
-Adaptive Generalized Dispersive Mode Dec.      |    AGNCMD / AGDMD
+Adaptive Generalized Dispersive Mode Dec.       |    AGNCMD / AGDMD
 Variational Mode Decomposition                  |    VMD
 Multivariate Variational Mode Decomposition     |    MVMD
 Variational Mode Extraction                     |    VME
@@ -206,6 +209,7 @@ Variational Nonlinear Chirp Mode Decomposition  |    VNCMD
 Iterative Nonlinear Chirp Mode Decomposition    |    INCMD
 Adaptive Variational Nonlinear Chirp Mode Dec.  |    AVNCMD
 Adaptive Local Iterative Filtering              |    ALIF
+Variational TF Mode Tracking Decomposition      |    VTFMTD
 Adaptive Polymorphic Mode Decomposition         |    APMD
 Impulsive Mode Decomposition                    |    IMD
 Empirical Wavelet Transform                     |    EWT
@@ -273,6 +277,8 @@ __all__ = [
     "INCMD",
     "AVNCMD",
     "ALIF",
+    "VTFMTD",
+    "vtfmtd",
     "APMD",
     "IMD",
     "imd",

pysdkit-0.4.46/pysdkit/_tfa/__init__.py

@@ -0,0 +1,41 @@
+# -*- coding: utf-8 -*-
+"""
+Time-Frequency Analysis (TFA).
+
+Methods that operate primarily on time-frequency representations
+(e.g. STFT) for mode separation and instantaneous-frequency tracking.
+"""
+
+from .vtfmtd import (
+    VTFMTD,
+    vtfmtd,
+    stft,
+    frequency_axis,
+    bin_index_grid,
+    expand_omega_init,
+    first_difference_gram,
+    estimate_if_centroid,
+    smooth_if,
+    moving_average_if,
+    omega_bins_to_hz,
+    load_dual_signal_noise,
+    load_single_nsignal,
+    load_map2,
+)
+
+__all__ = [
+    "VTFMTD",
+    "vtfmtd",
+    "stft",
+    "frequency_axis",
+    "bin_index_grid",
+    "expand_omega_init",
+    "first_difference_gram",
+    "estimate_if_centroid",
+    "smooth_if",
+    "moving_average_if",
+    "omega_bins_to_hz",
+    "load_dual_signal_noise",
+    "load_single_nsignal",
+    "load_map2",
+]

pysdkit-0.4.46/pysdkit/_tfa/data/__init__.py

@@ -0,0 +1,2 @@
+# -*- coding: utf-8 -*-
+"""Packaged demo arrays for VTFMTD examples and tests."""