Files
REQubit-control/PLE/120_plot_2D_contourf.py
T
2026-06-03 12:12:14 +02:00

93 lines
2.4 KiB
Python

# -*- coding: utf-8 -*-
"""
Created on Sun Nov 10 15:42:46 2024
@author: shen_t2
"""
import os
os.chdir(os.path.abspath(os.path.dirname(__file__)))
import numpy as np
import matplotlib.pyplot as plt
plt.rcParams.update({'font.size': 14})
#%%
MAIN_EXP_folder = 'C:/RE_qubit_TS/202504_CW_PLE_167Er/20250408_BalongcScan_sigma/'
os.chdir(MAIN_EXP_folder)
timestampsuffix = '_04080258.npy'
actual_counts_all = np.load('actual_counts_All_Bfields_rep0' + timestampsuffix)
raw_counts_all = np.load('actual_counts_All_Bfields_rep0' + timestampsuffix)
B_field_all = np.load('wl_scan_Bfields' + timestampsuffix)
wl_scan_all = np.load('wl_scan_x_axis_20.00_120.00_speed0.5000_CWGatedD' + timestampsuffix)
B_field_all = B_field_all.reshape(len(B_field_all), 1)
B_field_all = np.repeat(B_field_all, len(wl_scan_all), axis=1)
wl_scan_all = np.tile(wl_scan_all, (len(B_field_all), 1) )
wn_PLE_c = 6535.0997 # OK w/o precise fitting
wn_FTIR_c = 6534.356431
wn_scan_all = wl_scan_all#1e7 / wl_scan_all - wn_PLE_c + wn_FTIR_c
#%%
counts_all = raw_counts_all
counts_threshold = 2500
counts_all = actual_counts_all
counts_threshold = 40000
plt.figure(11, figsize=[9,6], dpi=100)
plt.clf()
plt.pcolormesh(B_field_all, wn_scan_all, counts_all, norm='linear',
vmin=0,
# vmax=counts_threshold, # to see satellites
cmap='RdBu')
plt.colorbar(label='Photon Counting (cps)')
# to see the full
# plt.contourf(B_field_all, wn_scan_all, counts_all,
# cmap='RdBu', levels=200)
# plt.colorbar(label='Photon Counting (cps)')
# to see satellites
# plt.contourf(B_field_all, wn_scan_all, counts_all,
# levels=np.linspace(0, counts_threshold, 201),
# cmap='RdBu',
# extend='both')
# plt.colorbar(label='Photon Counting (cps)', ticks=np.linspace(0, counts_threshold, 6))
# plt.grid()
# plt.legend(loc=1)
plt.title('Gated CW PLE @ #22, 0.01%, 167Er, 3.4 K, OD 0.5,\n 1/2 V/s, AOM, DAQ = 10 ms, sigma $E\perp c$, $B_{ext}\parallel c$')
plt.xlabel('Magnetic field (Gauss)')
plt.ylabel('Calibrated wavenumber (cm$^{-1}$)')
plt.ylabel('Piezo voltage (V)')
# plt.ylabel('Wavelength (nm)')
# plt.xlim(-100, )
# plt.ylim(6533.5, 6536.0)
plt.tight_layout()
plt.savefig( MAIN_EXP_folder + '20250408_plot_BalongcScan_sigma_167Er_full.jpg' )
plt.show()