71 lines
2.0 KiB
Python
Executable File
71 lines
2.0 KiB
Python
Executable File
#! /usr/bin/env python
|
|
|
|
"""
|
|
Plot fixed momentum DSF.
|
|
|
|
Usage: python plot_dsf_k_fixed.py [omega file] [dsf file]
|
|
"""
|
|
|
|
import glob
|
|
import matplotlib.pyplot as plt
|
|
import matplotlib.animation as animation
|
|
import numpy as np
|
|
import os
|
|
import sys
|
|
|
|
width = str(sys.argv[1])
|
|
|
|
|
|
# Get the list of interactions which have been computed
|
|
dirlist = os.listdir('../store/')
|
|
clist = sorted([float(c.lstrip('c_')) for c in dirlist])
|
|
|
|
# Get the Omega file
|
|
omegafile = glob.glob('../store/c_%s*/Omega*' % str(clist[0]).rstrip('.0')[:12])[0]
|
|
omega = np.loadtxt(omegafile)
|
|
|
|
# Load all the available dsfs from the data store
|
|
dsfs = {}
|
|
for c in clist:
|
|
# Do some black magic here: for matching the interaction,
|
|
# first try for exact match, stripping '.0' to treat integer values, e.g. 4.0 into 4
|
|
# and then (if it doesn't work) use only the first 12 characters, plus wildcard, to cover rounding errors
|
|
try:
|
|
dsffile = glob.glob('../store/c_%s/*_w_%s.dsfs' % (str(c).rstrip('.0'), width))[0]
|
|
except IndexError:
|
|
dsffile = glob.glob('../store/c_%s*/*_w_%s.dsfs' % (str(c).rstrip('.0')[:12], width))[0]
|
|
dsfs[str(c)] = np.loadtxt(dsffile)
|
|
|
|
fig, ax = plt.subplots()
|
|
|
|
ax.set_xlim(omega[0], omega[-1])
|
|
# To determine the y axis limits, we look at the lowest value of c (sharpest peak)
|
|
dsfsmax = max(dsfs[str(clist[0])])
|
|
print('dsfsmax = ', dsfsmax)
|
|
|
|
ymax = 1.2 * dsfsmax
|
|
xtext = 0.6 * omega[-1]
|
|
ytext = 1.1 * dsfsmax
|
|
|
|
ax.set_ylim([0, ymax])
|
|
ax.text(xtext, ytext, f'c = {clist[0]}')
|
|
ax.plot(omega, dsfs[str(clist[0])])
|
|
def animate(i):
|
|
ax.clear()
|
|
ax.set_xlim(omega[0], omega[-1])
|
|
ax.set_ylim([0, ymax])
|
|
ax.text(xtext, ytext, f'c = {clist[i]:10.6f}')
|
|
ax.plot(omega, dsfs[str(clist[i])])
|
|
|
|
ani = animation.FuncAnimation(fig, animate, frames=len(clist), interval=100, repeat=False)
|
|
|
|
outfilename = (dsffile.partition('_c_')[0].rpartition('/')[2] + '_c_scan_' +
|
|
dsffile.partition('_c_')[2].partition('_')[2].partition('.dsfs')[0])
|
|
|
|
ani.save(outfilename + '.mp4')
|
|
|
|
with open(outfilename + '.html', 'w') as file:
|
|
file.write(ani.to_html5_video())
|
|
|
|
#plt.show()
|