magnetic/plots.py at main · mahollands/magnetic · 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
"""
plots for interpreting Zeeman splitting of metal lines.
"""
import numpy as np
import matplotlib.pyplot as plt

__all__ = [
    "diagram_transitions",
    "diagram_energies",
]

def diagram_transitions(multiplet, Bmax, title=""):
    """
    Zeeman splitting diagram for all lines in a multiplet. pi components are
    shown in purple. Sigma components are shown in blue. Line opacity is
    proportional to the relative transition strength.
    """
    max_strength = max(gf_ for _, gf_, *_ in multiplet.transitions(0))
    B = np.array([0, Bmax])
    for tr in multiplet.transitions(B):
        x_line, gf_, *_, dmJ, _ = tr
        alpha = gf_/max_strength
        c = 'C4' if dmJ == 0 else 'C0'
        plt.plot(x_line, B, f'{c}-', lw=1.5, alpha=alpha)
    plt.xlabel(r"Wavelength [\AA]")
    plt.ylabel("Field Strength [kG]")
    plt.title(title)
    plt.show()

def diagram_energies(multiplet, Bmax, title=""):
    """
    Zeeman splitting diagram for all energy levels involved in a multiplet.
    Splitting of the upper levels is shown in blue. Splitting of the lower
    levels is shown in orange.
    """
    B = np.array([0, Bmax])
    for i, level in enumerate(multiplet.Upper_and_Lower_states):
        plt.subplot(2, 1, i+1)
        for state in level:
            for substate in state.energies(B):
                plt.plot(B, substate.k, f'C{i}-')
        plt.ylabel("Energy [1/cm]")
    plt.xlabel("Field Strength [kG]")
    plt.title(title)
    plt.show()