mirror of
https://github.com/vale981/fibre_walk_project_code
synced 2025-03-04 17:31:39 -05:00
34 lines
875 B
Python
34 lines
875 B
Python
from ringfit import data
|
|
import matplotlib.pyplot as plt
|
|
from ringfit.data import *
|
|
from ringfit.plotting import *
|
|
from ringfit.fit import *
|
|
from rabifun.analysis import *
|
|
from rabifun.plots import *
|
|
|
|
# %% setup figure
|
|
fig = make_figure("calibration")
|
|
ax_spectrum = fig.subplots(1, 1)
|
|
|
|
|
|
# %% load data
|
|
path = "../../data/22_05_24/ringdown_try_2"
|
|
scan = ScanData.from_dir(path)
|
|
|
|
|
|
# %% Extract Rough Peaks
|
|
window = tuple(
|
|
np.array([0.016244684251065847 + 0.000002, 0.016248626903395593 + 49e-5])
|
|
+ 8e-3
|
|
- 12e-7
|
|
)
|
|
|
|
ringdown_params = RingdownParams(fω_shift=10e4, mode_window=(0, 50))
|
|
peak_info = find_peaks(scan, ringdown_params, window, prominence=0.008)
|
|
|
|
|
|
peak_info = refine_peaks(peak_info, ringdown_params)
|
|
plot_spectrum_and_peak_info(ax_spectrum, peak_info, ringdown_params)
|
|
|
|
# extract FSR and hybridization amplitude
|
|
extract_Ω_δ(peak_info, ringdown_params)
|