Energy time spectrum

import nexus as nx
import numpy as np
import matplotlib.pyplot as plt
from matplotlib import colors

# ----------------- cavity in reflection geometry -------------

# ------------------------- Fe layer --------------------------
lay_Fe = nx.Layer(id = "Fe",
                material = nx.Material.Template(nx.lib.material.Fe_enriched),
                thickness = 1.5,
                roughness = 0.3)

site1 = nx.Hyperfine(magnetic_field = 0)

lay_Fe.material.hyperfine_sites = [site1]

# ----------------------------- Pt layers -----------------------------
lay_Pt_top = nx.Layer(id = "Pt top",
                material = nx.Material.Template(nx.lib.material.Pt),
                thickness = 2,
                roughness = 0.3)

lay_Pt = nx.Layer(id = "Pt",
                material = nx.Material.Template(nx.lib.material.Pt),
                thickness = 15,
                roughness = 0.3)

# -------------------------- C ---------------------------

lay_C = nx.Layer(id = "C",
                material = nx.Material.Template(nx.lib.material.C),
                thickness = 10,
                roughness = 0.3)

# --------------------- substrate ---------------------------------
lay_substrate = nx.Layer(id = "Si sub",
                material = nx.Material.Template(nx.lib.material.Si),
                thickness = nx.inf,
                roughness = 0.3)

sample_cavity = nx.Sample(id = "simple layers",
                          layers = [lay_Pt_top,
                                    lay_C,
                                    lay_Fe,
                                    lay_C,
                                    lay_Pt,
                                    lay_substrate],
                           geometry = "r",
                           angle = 0.148,
                           roughness = "a")

# ------------- Stainless steel foil in forward geometry ----------------
foil = nx.Layer(id = "StainSteel",
                material = nx.Material.Template(nx.lib.material.SS_enriched),
                thickness = 3000)

site_foil = nx.Hyperfine(
    isomer = -0.09,  # mm/s
    quadrupole = 0.6,
    isotropic = True)

foil.material.hyperfine_sites = [site_foil]

sample_foil = nx.Sample(id = "simple foil ",
                        layers = [foil],
                        geometry = "f")

beam  = nx.Beam()
beam.LinearSigma()

exp = nx.Experiment(beam = beam,
                    objects = [sample_cavity, sample_foil],
                    isotope = nx.moessbauer.Fe57,
                    id = "my exp")

# define a detuning for the sample on the drive
sample_foil.drive_detuning = np.linspace(-200, 200, 1024)  # in Gamma

ets = nx.EnergyTimeSpectrum(experiment = exp,
                            time_length = 200.0,
                            time_step = 0.2,
                            mode = "i",
                            electronic = False)

timescale, intensity = ets()

print(intensity.shape)

plt.imshow(intensity.T,
           cmap=plt.cm.plasma,
           norm=colors.LogNorm(),
           aspect='auto',
           origin = 'lower',
           extent=(min(sample_foil.drive_detuning), max(sample_foil.drive_detuning), min(timescale), max(timescale)))

plt.colorbar(orientation='vertical', label = r'intensity ($\Gamma$/ns)')
plt.xlabel(r'detuning ($\Gamma$)')
plt.ylabel('time (ns)')
plt.show()

(1024, 1001)

print(intensity)

[[3.89475712e-01 1.88827483e+00 1.45574937e+00 ... 3.80616843e-04
  3.85786561e-04 3.85377921e-04]
 [3.89465713e-01 1.88822873e+00 1.45573552e+00 ... 3.30038653e-04
  3.53769096e-04 3.77211141e-04]
 [3.89455719e-01 1.88818249e+00 1.45572148e+00 ... 2.58404406e-04
  2.93835108e-04 3.35270544e-04]
 ...
 [3.82318692e-01 1.71634839e+00 1.16117265e+00 ... 3.66926499e-04
  3.79551605e-04 3.83534360e-04]
 [3.82314604e-01 1.71606669e+00 1.16067194e+00 ... 3.83576327e-04
  3.70477430e-04 3.51696207e-04]
 [3.82310522e-01 1.71578491e+00 1.16017134e+00 ... 3.62345258e-04
  3.23897347e-04 2.85591336e-04]]

int_energy_time_spec = nx.IntegratedEnergyTimeSpectrum(experiment = exp,
                                                       time_length = 200.0,
                                                       time_step = 0.2,
                                                       integration_start = 51,
                                                       integration_stop = 180,
                                                       mode = "i",
                                                       electronic = False)

intensity_int = int_energy_time_spec()

plt.plot(sample_foil.drive_detuning, intensity_int)
plt.xlabel(r'detuning ($\Gamma$)')
plt.ylabel(r'$\Delta t$-integrated intensity ($\Gamma$)')
plt.show()

ets.Plot()

int_energy_time_spec.Plot()