calculation.py

import numpy as np
from scipy.interpolate import splrep, splev


from globals import T
import oes as oe
import optic as op
import pandas_tools as pt
import plot as plt
import quantum as qt
import scipy.io
import super_lattice_structure as st
import vertex_70 as vtx




def refra_doping(bypass_dbr=True, electric_field=0., wavelength=850e-9):
    sl = st.structure_eam(bypass_dbr=bypass_dbr)
    sl = op.algaas_super_lattice_refractive_index(sl, electric_field, wavelength)
    plt.plot_refra_doping(sl)


def al_doping(bypass_dbr=True):
    sl = st.structure_eam(bypass_dbr=bypass_dbr)
    plt.plot_al_doping(sl)




def reflectivity(bypass_dbr=True, start_wavelength=820e-9, stop_wavelength=880e-9, electric_field=0., n_points=100):
    wavelength = np.linspace(start_wavelength, stop_wavelength, num=n_points)
    r = op.reflectivity(bypass_dbr, electric_field, wavelength)
    plt.plot_reflectivity(wavelength, r)


def reflectivity_meshgrid(temperature=T, bypass_dbr=True, start_wavelength=820e-9, stop_wavelength=880e-9, electric_field=0., n_points=100):
    wavelength = np.linspace(start_wavelength, stop_wavelength, num=n_points)
    time, r = op.reflectivity_meshgrid(bypass_dbr, electric_field, wavelength, temperature)
    plt.plot_reflectivity_heatmap(wavelength, time, r)




def electromagnetic_amplitude(bypass_dbr=True, electric_field=0., wavelength=850e-9):
    sl = st.structure_eam(bypass_dbr=bypass_dbr)
    sl = op.algaas_super_lattice_refractive_index(sl, electric_field, wavelength)
    sl = pt.cut_in_equal_layers_thickness(sl, 1e-9)
    em = op.em_amplitude_smm(sl, wavelength)
    sl.insert(sl.shape[1], 'electromagnetic_amplitude', value=em)
    plt.plot_refra_em(sl)


def zandberg_electromagnetic_amplitude(electric_field=0., wavelength=1e-6):
    sl = st.structure_zandbergen()
    sl = op.algaas_super_lattice_refractive_index(sl, electric_field, wavelength)
    sl = pt.cut_in_equal_layers_thickness(sl, 1e-9)
    em = op.em_amplitude_smm(sl, wavelength)
    sl.insert(sl.shape[1], 'electromagnetic_amplitude', value=em)
    plt.plot_refra_em(sl)




def ftir_reflectivity(filename='eam_2', document_folder=True, linux=True):
    # import data
    wavelength, r = vtx.import_vertex_70_data(filename, document_folder=document_folder, linux=linux)

    # smooth the curve (I know it is cheating but ftir currently is out of service, so data is disgusting)
    bspl = splrep(wavelength, r, s=4.5)
    r = splev(wavelength, bspl)

    #  correct data
    r_max = np.max(r)
    r = r * 0.96 / r_max

    # plot
    plt.plot_reflectivity(wavelength, r)




def mqw_psi(lz=0.01e-9, electric_field=0., wavelength=850e-9):
    sl = st.structure_eam(air=False, top_dbr=False, bot_dbr=False, shared_dbr=False, contact=False, substrate=False, amount_qw=5)
    sl = pt.cut_in_equal_layers_thickness(sl, 1e-11)
    v_e, v_hh, eig_e, psi_e, eig_hh, psi_hh = qt.solve_schrodinger(sl)
    np.savez_compressed('v_e', v_e)
    np.savez_compressed('v_hh', v_hh)
    np.savez_compressed('eig_e', eig_e)
    np.savez_compressed('psi_e', psi_e[0, :])
    np.savez_compressed('eig_hh', eig_hh)
    np.savez_compressed('psi_hh', psi_hh[0, :])
    plt.plot_psie()


def benchmark_refractive_index():
    al = np.array([0.0, 0.2, 0.4, 0.6, 0.8, 1.0])
    wavelength = np.arange(500e-9, 1500e-9, 1e-9)
    indices_array = np.empty([al.size, wavelength.size])

    for x in range(al.size):
        n = np.array([])

        for w in wavelength:
            n = np.append(n, op.afromovitz_varshni_real_algaas_refractive_index(al[x], w))

        indices_array[x] = n
        name = 'indices_array_' +str(x) +'.mat'
        scipy.io.savemat(name, mdict={'arr': n})

    name = 'al_array.mat'
    scipy.io.savemat(name, mdict={'arr': al})
    name = 'wavelength_array.mat'
    scipy.io.savemat(name, mdict={'arr': wavelength})

    plt.plot_refra_bench(al, wavelength, indices_array)




def oes_test():
    time, wavelength, peaks, data_mean_sample, data_empty, data_diff = oe.import_data_march_2021()
    data_mean_sample = oe.temporal_mean(data_mean_sample)
    data_empty = oe.temporal_mean(data_empty)
    data_diff = oe.temporal_mean(data_diff)
    plt.plot_eos_spectrum_gold_plate(wavelength, data_mean_sample, data_empty, data_diff)