rf-funcitons-py 0.1.2__py3-none-any.whl → 0.1.5__py3-none-any.whl

rafpy/__init__.py

@@ -1 +1,3 @@
-from .logic import id_signal_candidates
+from .logic import id_signal_candidates
+from .filters import RF_filter, bpf_kernel, get_acf_amplitude, plot_frequency_response
+from .bpf_filter import create_bpf_kernel

rafpy/filters.py

@@ -0,0 +1,102 @@
+def RF_filter(signal_array, filter_kernel, caption_text1, caption_text2, sampling_rate=1.0, show_graph=True):
+    normalized_kernel = filter_kernel / np.sum(filter_kernel)
+    filtered_signal = signal.fftconvolve(signal_array, normalized_kernel, mode='same')
+    if show_graph:
+        N = len(signal_array)
+        t = np.arange(N) / sampling_rate
+        M = len(filter_kernel)
+        t_kernel = np.arange(M) / sampling_rate
+        fig, (ax1, ax2) = plt.subplots(2, 1, figsize=(10, 10))
+        plt.subplots_adjust(bottom=0.1, hspace=0.6)
+        ax1.set_title("Fig 1. Input 1: Signal & Filter Kernel", fontweight='bold')
+        ax1.plot(t, signal_array, color='steelblue', alpha=0.6, label='Input Signal (Left Axis)')
+        ax1.tick_params(axis='y', labelcolor='steelblue')
+        ax1_twin = ax1.twinx()
+        ax1_twin.plot(t_kernel, normalized_kernel, color='#D62728', linewidth=2, label='Filter Kernal (Right Axis)')
+        ax1_twin.fill_between(t_kernel, 0, normalized_kernel, color='#D62728', alpha=0.3)
+        ax1_twin.tick_params(axis='y', labelcolor='#D62728')
+        ax1_twin.set_ylim(bottom=0)
+        ax1_twin.set_ylabel("Normalized Amplitude", color='#D62728')
+        lines_1, labels_1 = ax1.get_legend_handles_labels()
+        lines_2, labels_2 = ax1_twin.get_legend_handles_labels()
+        ax1.legend(lines_1 + lines_2, labels_1 + labels_2, loc='upper right')
+        ax1.set_xlabel(f"Time (s)if Fs={sampling_rate} Hz")
+        ax1.set_ylabel("Amplitude", color = 'steelblue')
+        desc_1 = caption_text1 
+        desc_2 = caption_text2
+        ax1.text(0.5, -0.25, textwrap.fill(desc_1, width = 100), transform=ax1.transAxes, ha='center', va='top', fontsize=10, color='darkred', bbox=dict(boxstyle="round,pad=0.3", fc="white", ec="gray", alpha=0.5))
+        ax2.set_title("Fig 2. Output: Convolved Result", fontweight='bold')
+        ax2.plot(t, signal_array, color='gray', alpha=0.3, label='(Original Input)')
+        ax2.plot(t, filtered_signal, color='green', linewidth=2, label='Filtered Output')
+        ax2.set_xlabel(f"Time (seconds) if Fs={sampling_rate}Hz")
+        ax2.set_ylabel("Amplitude")
+        ax2.legend(loc='upper right')
+        ax2.grid(True, alpha=0.3)
+        ax2.text(0.5, -0.3, textwrap.fill(desc_2, width = 100), transform=ax2.transAxes, ha='center', va='top', fontsize=10, color='darkgreen', bbox=dict(boxstyle="round,pad=0.3", fc="white", ec="gray", alpha=0.5))
+       
+        plt.show()
+    return filtered_signal
+
+def get_acf_amplitude(signal_array):
+    # 1. Remove DC offset (Center the signal at 0)
+    centered_signal = signal_array - np.mean(signal_array)
+    
+    # 2. Calculate the Autocorrelation
+    # 'same' mode keeps the output the same length as the input
+    acf = np.correlate(centered_signal, centered_signal, mode='full')
+    
+    # 3. Get the value at zero lag (the center of the 'full' correlation)
+    zero_lag_index = len(acf) // 2
+    r_0 = acf[zero_lag_index] / len(centered_signal) # Normalize by length
+    
+    # 4. Calculate RMS and Peak Amplitude
+    rms = np.sqrt(r_0)
+    peak_amplitude = rms * np.sqrt(2)
+    
+    return peak_amplitude
+
+def create_bpf_kernel(low_kb, high_kb, fs, num_taps=101):
+    """Utility to generate a Band-Pass Filter kernel."""
+    return sp_signal.firwin(num_taps, [low_kb, high_kb], fs=fs, pass_zero=False)
+
+def plot_frequency_response(data_list, use_acf=True):
+    """
+    Plots Max Voltage vs Frequency, handling 1.5MHz and 3MHz separately.
+    data_list: list of (freq_hz, filename, fs_hz)
+    """
+    data_1_5 = []
+    data_3_0 = []
+
+    for freq, filename, fs in data_list:
+        try:
+            data_file = np.load(filename)
+            signal_data = data_file["arr_0"][2]
+            
+            # Choose amplitude method
+            val = get_acf_amplitude(signal_data) if use_acf else np.max(np.abs(signal_data))
+            
+            if fs == 1500000:
+                data_1_5.append((freq / 1000, val))
+            else:
+                data_3_0.append((freq / 1000, val))
+        except Exception as e:
+            print(f"Error loading {filename}: {e}")
+
+    # Sort to prevent the 'sawtooth' lines
+    data_1_5.sort()
+    data_3_0.sort()
+
+    plt.figure(figsize=(10, 6))
+    if data_1_5:
+        x15, y15 = zip(*data_1_5)
+        plt.plot(x15, y15, 'bo-', label='1.5 MHz Fs', alpha=0.8)
+    if data_3_0:
+        x30, y30 = zip(*data_3_0)
+        plt.plot(x30, y30, 'ro-', label='3.0 MHz Fs', alpha=0.8)
+
+    plt.title(f"Frequency Response ({'ACF' if use_acf else 'Raw'} Peak)")
+    plt.xlabel("Signal Frequency (kHz)")
+    plt.ylabel("Amplitude (V)")
+    plt.legend()
+    plt.grid(True, alpha=0.3)
+    plt.show()

rafpy/logic.py

@@ -56,12 +56,6 @@ def id_signal_candidates(fig_num, observed, current_fs, filter_range=None, show_
         ax.set_title(f"Fig {fig_num}. Signal Identification (Observed: {observed} Hz | $f_s$: {current_fs} Hz | Bandpass {np.min(filter_range)}-{np.max(filter_range)} Hz)", fontweight='bold')
         ax.set_xlabel("Frequency (Hz)", fontweight='bold')
         ax.set_ylim(0, 1.6)
-        desc = (
-        f"""Fig {fig_num}. You saw a signal at {observed} Hz with a sampling rate of {current_fs} Hz. If you used a bandpass filter then the highlighted frequency is the original frequency considering any aliasing observed and which Nyquist zone it resides in. If you did not use a bandpass filter, then this lists the possible frequencies associated with the sampled signal, which can be filtered manually using known physics of your source."""
-        )
-        wrapped_desc = textwrap.fill(desc, width=90)
-        plt.figtext(0.5, 0.05, wrapped_desc, ha='center', fontsize=10,
-                    bbox=dict(boxstyle="round,pad=0.3", fc="#f0f0f0", ec="black", alpha=0.5))
         plt.legend(loc='upper right')
         print("-" * 30)
         print(f"ANALYSIS FOR {observed} Hz (Fs = {current_fs} Hz)")

{rf_funcitons_py-0.1.2.dist-info → rf_funcitons_py-0.1.5.dist-info}/METADATA

@@ -1,10 +1,13 @@
 Metadata-Version: 2.4
 Name: rf_funcitons_py
-Version: 0.1.2
+Version: 0.1.5
 Summary: First package from a notebook
 Author-email: Avery Books <abooks104@gmail.com>
 Requires-Python: >3.8
 Description-Content-Type: text/markdown
+Requires-Dist: numpy
+Requires-Dist: matplotlib
+Requires-Dist: scipy
 
 #My First Package
 This package contains my alias source signal identification function

rf_funcitons_py-0.1.5.dist-info/RECORD

@@ -0,0 +1,7 @@
+rafpy/__init__.py,sha256=uAUmFifm3o5-_PpU6mYsLqJWqkNeId6rp53mmHvKbt8,170
+rafpy/filters.py,sha256=xFXy_SIV7GdAT3fX72fEimuSHetja5O0EYzEGC6LDsU,4754
+rafpy/logic.py,sha256=UMsdbz0s30P7ojoNHi22_oLoIm-f1gxp9YplSej4eeA,3277
+rf_funcitons_py-0.1.5.dist-info/METADATA,sha256=0QzsHHPT8TaKqKrQzD8JvWouQK0wVTEUq75CbLX0a9Q,377
+rf_funcitons_py-0.1.5.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+rf_funcitons_py-0.1.5.dist-info/top_level.txt,sha256=4q_MXcdbMykJYGFjd_1IVwHk_V3unZSegNAYAUumMnw,6
+rf_funcitons_py-0.1.5.dist-info/RECORD,,

rf_funcitons_py-0.1.2.dist-info/RECORD

@@ -1,6 +0,0 @@
-rafpy/__init__.py,sha256=YJ7WwLTtUPL2cl7hNS_I75dwn4MiCSuXmjbPVYw0c60,39
-rafpy/logic.py,sha256=_bjpKKMS93SO8UpJDEoPCUWyM8Fn8pXO4GjLnIQ8qKA,3974
-rf_funcitons_py-0.1.2.dist-info/METADATA,sha256=EM5qzvPt6if5KUdEZ4Cl9IKlSjl5FP3E6W2PWW3FNJc,306
-rf_funcitons_py-0.1.2.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-rf_funcitons_py-0.1.2.dist-info/top_level.txt,sha256=4q_MXcdbMykJYGFjd_1IVwHk_V3unZSegNAYAUumMnw,6
-rf_funcitons_py-0.1.2.dist-info/RECORD,,