LaserPy_Quantum/SPDC_sim.py at main · Mathwizard1/LaserPy_Quantum · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
from LaserPy_Quantum import Clock
from LaserPy_Quantum import Connection, Simulator
from LaserPy_Quantum import (
    StaticWave,
    ArbitaryWaveGenerator
)
from LaserPy_Quantum import CurrentDriver
from LaserPy_Quantum import Laser

from LaserPy_Quantum.SpecializedComponents import SinglePhotonDetector

from LaserPy_Quantum.SpecializedComponents.PhotonPairGenerator import PhotonPairGeneratorCrystal
from LaserPy_Quantum.utils.RefractiveMaterials import Birefringent, SellmeierFormula

from LaserPy_Quantum.utils.HelperFunctions import (
    fast_scatter,
    fast_density_plot
)

from numpy import pi, mean
import matplotlib.pyplot as plt

############################################################################
# Switch Laser Type

from LaserPy_Quantum.Constants import LaserPyConstants
LaserPyConstants.load_from_json("SPDC_Constants.json")

############################################################################

# Control Constants (all in SI units)
dt = 1e-12
t_unit = 1e-9
sampling_rate = 2

# Current Constants
I_th = 0.0178
MASTER_BASE_DC = 5 * I_th

# Steady above lasing current
mBase = StaticWave("mBase", MASTER_BASE_DC)

AWG = ArbitaryWaveGenerator()
AWG.set(mBase)

############################################################################

BBO = Birefringent(
    SellmeierFormula((2.7359,0.01878,0.01822,0.01354)),
    SellmeierFormula((2.3753,0.01224,0.01667,0.01516)),
    crystal_angle= 0.86,
    name="BBO"
)

SPDC = PhotonPairGeneratorCrystal(
    refractive_material= BBO,
    SPDC_type= 'II',
    gaussian_dimension= 2,
)

############################################################################

# best_angle = 0.0
# theta = 0.0
# while(theta < 1):
#     BBO.set(theta)

#     theta += 0.01

# exit(0)

############################################################################

current_driver = CurrentDriver(AWG)
current_driver.set(mBase)

laser = Laser(name= "pump_laser")
SPD = SinglePhotonDetector()

simulator_clock = Clock(dt, sampling_rate)
simulator_clock.set(t_unit * 10)

simulator = Simulator(simulator_clock)

simulator.set((
    Connection(simulator_clock, current_driver),
    Connection(current_driver, laser),
    Connection(laser, SPD),
    Connection(laser, SPDC),
))

simulator.reset(True)

simulator.simulate()
time_data = simulator.get_data()

#laser.display_data(time_data)
#SPD.display_data(time_data)
SPDC.display_data(time_data)

SPDC_data = SPDC.get_data()
#print(SPDC_data.keys())


delta_K = SPDC_data['delta_K']

min_scaled_delta_K = (delta_K - delta_K.min()) / (delta_K.max() - delta_K.min())
Z_scaled_delta_K = (delta_K - delta_K.mean()) / delta_K.std()

fast_scatter(SPDC_data['omega_s'],SPDC_data['omega_i'], Z_scaled_delta_K)