ttgsm/monolayer_convergence.py at main · CompPhysLab/ttgsm · 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
import numpy as np
import matplotlib.pyplot as plt

from diffraction_solver import solve_diffraction
from incidence import Incidence
from grating import Grating


if __name__ == '__main__':
    n = 5
    d_range = range(2, 17)
    # d_range = [15] * 1
    thickness = 0.3
    accuracy = 1e-6
    n_sweeps = 20
    verb = 0

    smaller_scale = 2 * np.pi

    amps = []

    for d in d_range:
        incidence = Incidence(angle=10)
        grating = Grating.multiscale_lamellar_random(period=n * smaller_scale, n=n,
                                                     thickness=thickness, max_permittivity=2.1, seed=1)
        x = solve_diffraction(
            do=d,
            dl=2,
            grating=grating,
            incidence=incidence,
            accuracy=accuracy,
            nswp=n_sweeps,
            verb=verb
        )

        amp = abs(x.full(asvector=True)[2 ** (d - 1)])
        amp = 1 - amp
        print(d, amp)
        amps.append(amp)

    amps = np.array(amps)

    plt.plot(np.array(d_range)[1:], np.abs(amps[:-1] - amps[1:]), marker='o')
    # plt.plot(2 ** np.array(d_range)[1:], np.abs(amps[:-1] - amps[1:]), marker='o')
    plt.yscale('log')
    # plt.xscale('log')
    plt.grid()
    # plt.xlabel('Number of diffraction orders')
    plt.xlabel('d_O')
    plt.ylabel('Difference')
    plt.show()