import pyhande
import matplotlib.pyplot as plt
(metadata, qmc_data) = pyhande.extract.extract_data('calcs/ccmc_solids/diamond_ccmc.out')[0]
plt.plot(qmc_data['iterations'], qmc_data['Shift'], label=r'$S(\tau)$')
plt.plot(qmc_data['iterations'], qmc_data['Re{\sum H_0j N_j}']/qmc_data['Re{N_0}'], label=r'$E(\tau) = \Re\left(\sum_j H_{0j} N_j(\tau)\right)/\Re(N_0(\tau))$')
plt.legend()
plt.ylim(-0.4,0)
plt.xlabel('iteration')
plt.ylabel('Correlation energy/' + r'$E_h$')