NMRRWSims/cpp
1
0
Fork 0
Code Issues 3 Pull Requests Actions Packages Projects Releases Wiki Activity

save spectrum in post_process_spectrum.py

Browse Source
This commit is contained in:
Dominik Demuth 2025-02-10 18:23:15 +01:00
parent bfb1cb314c
commit a488c47713
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
python/spectrum.py
View File

@ -44,7 +44,8 @@ def post_process_spectrum(parameter_file, apod, tpulse):
parameter = read_parameter_file(parameter_file) parameter = read_parameter_file(parameter_file)
# files have form ste_arg=0.000000e+01_parameter.txt, first remove ste part then parameter.txt to get variables # files have form ste_arg=0.000000e+01_parameter.txt, first remove ste part then parameter.txt to get variables
varied_string = str(parameter_file).partition('_')[-1].rpartition('_')[0] varied_string = str(parameter_file).split('/')[-1].partition('_')[-1].rpartition('_')[0]
print(varied_string)
# make evolution times # make evolution times
tevo = np.linspace(parameter['techo_start'], parameter['techo_stop'], num=int(parameter['techo_steps'])) tevo = np.linspace(parameter['techo_start'], parameter['techo_stop'], num=int(parameter['techo_steps']))
@ -65,6 +66,12 @@ def post_process_spectrum(parameter_file, apod, tpulse):
spec = np.fft.fftshift(np.fft.fft(timesignal, axis=0), axes=0).real spec = np.fft.fftshift(np.fft.fft(timesignal, axis=0), axes=0).real
spec *= pulse_attn(freq, t_pulse=tpulse)[:, None] spec *= pulse_attn(freq, t_pulse=tpulse)[:, None]
if varied_string:
np.savetxt(parameter_file.with_name(f"spectrum_{varied_string}.dat"), np.c_[freq, spec])
else:
np.savetxt(parameter_file.with_name(f"spectrum.dat"), np.c_[freq, spec])
# #
# #
# reduction_factor[i, :] = 2*timesignal[0, :] # reduction_factor[i, :] = 2*timesignal[0, :]