Metadata-Version: 2.4
Name: mysnr
Version: 0.2
Summary: A library for calculating SNR and SNDR from power spectral density
Home-page: https://github.com/username/my_library
Author: Kilian David
Author-email: Kilian David <kilian.david@tum.de>
License-Expression: AGPL-3.0-or-later
Project-URL: Homepage, https://github.com/kl-david/mysnr
Classifier: Programming Language :: Python :: 3
Classifier: Operating System :: OS Independent
Requires-Python: >=3.9
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: scipy
Requires-Dist: numpy
Dynamic: author
Dynamic: home-page
Dynamic: license-file

# MySNR Package

MySNR is a Python package designed to calculate Signal-to-Noise Ratio (SNR) and Signal-to-Noise and Distortion Ratio (SNDR) from Power Spectral Density (PSD). This package provides functions for analyzing frequency data, estimating signal power, harmonic power, and noise power, and computing SNR and SNDR metrics.

## Installation

You can install MySNR using pip:

```bash
pip install mysnr
```

## Usage

### Importing the Package

To use the MySNR package, you first need to import it into your Python script or Jupyter notebook.

```python
import mysnr
```

### Example Code

Here's an example of how to use the MySNR package:

```python
import numpy as np
from scipy.signal import periodogram
import matplotlib.pyplot as plt

import mysnr

plt.rcParams['lines.marker'] = 'x'
plt.rcParams['axes.grid'] = True


T_samp = 100E-9         # sampling interval
f = 3.0E3               # fundamental signal frequency
k = 5                   # number of periods in timeframe
N = int(k/(f * T_samp)) # total number of samples

signal_power = 0.5
harmonic_power = 0.045
noise_power = 0.1

t = np.linspace(0.0, N*T_samp, N, endpoint=False)                           # time vector
y = np.sqrt(signal_power * 2) * np.sin(f * 2.0 * np.pi *t)                  # base signal, power = 0.5 * 1**2 = 0.5
y += np.sqrt(harmonic_power * 2) * np.sin(2*f * 3.0 * np.pi *t)             # third harmonic, power = 0.5 * 0.3**2 = 0.045
y += np.random.normal(0, np.sqrt(noise_power), len(t))                      # add white noise over the full band

py_xf, py_yf = periodogram(x=y, fs=1/T_samp, window='boxcar', scaling='density')

#harmonic_bins = mysnr.findBinnedHarmonics(py_yf, py_xf, f, 0.5/T_samp, 3)
#signal_peak_bin = mysnr.freqToBin(py_xf, f)
#signal_bins = mysnr.getIndizesAroundPeak(py_yf, signal_peak_bin)

snr, signal_bins = mysnr.snr(py_yf, py_xf, T_samp=T_samp, N=N, bw=None, return_bins=True)
sndr, signal_bins, harmonic_bins = mysnr.sndr(py_yf, py_xf, T_samp=T_samp, N=N, bw=None, return_bins=True)

print(f"Expected Signal Power: {signal_power}")
print(f"Calculated Signal power: {mysnr.bandpower(py_yf[signal_bins], tau=T_samp*N)}")

print(f"Expected Harm Power: {harmonic_power}")
print(f"Calculated Harmonic Power: {mysnr.bandpower(py_yf[harmonic_bins], tau=T_samp*N)}")

print(f"Expected SNR: {10*np.log10(signal_power/noise_power)}")
print(f"Calculated SNR: {10*np.log10(snr)}")

print(f"Expected SNDR: {10*np.log10(signal_power/(noise_power + harmonic_power))}")
print(f"Calculated SNDR: {10*np.log10(sndr)}")

fig, fftax = plt.subplots()
py_yfdb = 10*np.log10(np.abs(py_yf))

fftax.semilogx(py_xf, py_yfdb)
fftax.semilogx(py_xf[harmonic_bins], py_yfdb[harmonic_bins], linestyle='')
fftax.semilogx(py_xf[signal_bins], py_yfdb[signal_bins], linestyle='')

plt.grid(which='minor', linestyle=':', linewidth=0.5, color='lightgray')  # Minor grid lines
plt.show()
```

## Functions

- **bandpower(psd, mode='psd', tau=1)**: Estimate the bandpower from a power spectral density.
- **getIndizesAroundPeak(psd, peakIndex, searchWidth=1000)**: Find indices around a peak in the PSD.
- **freqToBin(fAxis, f)**: Convert a frequency to its corresponding bin index.
- **findBinnedHarmonics(psd, faxis, fund, bw, nHarmonics=6, aliased=False)**: Find binned harmonics of a fundamental frequency.
- **snr(psd, faxis, T_samp, N, bw=None)**: Calculate the Signal-to-Noise Ratio (SNR).
- **sndr(psd, faxis, T_samp, N, bw=None)**: Calculate the Signal-to-Noise and Distortion Ratio (SNDR).

## Contributing

Contributions are welcome! If you have any suggestions or bug reports, please open an issue on the [GitHub repository](https://github.com/kl-david/mysnr).

## Acknowledgements:

This project is inspired by https://github.com/hrtlacek/SNR/blob/main/SNR.ipynb .
