# -*- 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()