import quantarhei as qr
import matplotlib.pyplot as plt

lab = qr.LabSetup()
freq = qr.FrequencyAxis(-2500, 1000, 5.0)
pulse2 = dict(ptype="Gaussian", FWHM=800.0, amplitude=1.0)
params = (pulse2, pulse2, pulse2)
lab.set_pulse_arrival_times([0.0, 0.0, 0.0])
lab.set_pulse_phases([0.0, 0.0, 0.0])
lab.set_pulse_shapes(freq, params)

pls = lab.pulse_f[2]
lab2 = qr.LabSetup()

pulse1 = dict(ptype="numeric", function=pls)
params = (pulse1, pulse1, pulse1)
lab2.set_pulse_arrival_times([0.0, 0.0, 0.0])
lab2.set_pulse_phases([0.0, 0.0, 0.0])
lab2.set_pulse_shapes(freq, params)

dfc = lab2.get_pulse_spectrum(1, freq.data)
pl = plt.plot(freq.data, dfc)
plt.show()