GeneralTmm/Examples/ExampleTIR.py at master · ardiloot/GeneralTmm · 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
"""Total Internal Reflection (TIR) at a prism/air interface.

Sweeps over incidence angle to show the critical-angle transition
for p- and s-polarizations. Critical angle ≈ 41.8° for glass/air.
"""

import matplotlib.pyplot as plt
import numpy as np

from GeneralTmm import Material, Tmm


def main():
    prism = Material.Static(1.5)
    substrate = Material.Static(1.0)

    wl = 532e-9  # green laser
    betas = np.linspace(0.0, 1.49, 100)  # beta = n*sin(theta)
    angles = np.arcsin(betas / prism(wl).real)

    tmm = Tmm(wl=wl)
    tmm.AddIsotropicLayer(float("inf"), prism)
    tmm.AddIsotropicLayer(float("inf"), substrate)
    sr = tmm.Sweep("beta", betas)

    # Plot
    fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(11, 4))
    fig.suptitle("Total internal reflection", fontsize=13)

    ax1.set_title("(a)  Reflection")
    ax1.plot(np.degrees(angles), sr["R11"], label=r"R$_{pp}$")
    ax1.plot(np.degrees(angles), sr["R22"], label=r"R$_{ss}$")
    ax1.set_xlabel(r"$\theta$ ($\degree$)")
    ax1.set_ylabel("Reflectance")
    ax1.legend()

    ax2.set_title("(b)  Transmission")
    ax2.plot(np.degrees(angles), sr["T31"], label=r"T$_{pp}$")
    ax2.plot(np.degrees(angles), sr["T42"], label=r"T$_{ss}$")
    ax2.set_xlabel(r"$\theta$ ($\degree$)")
    ax2.set_ylabel("Transmittance")
    ax2.legend()

    fig.tight_layout()
    plt.show()


if __name__ == "__main__":
    main()