3
0
mirror of https://github.com/triqs/dft_tools synced 2024-06-30 00:44:34 +02:00
dft_tools/python/clear_lda_output.py
Priyanka Seth 8d44e5ce91 SumK generated output in new group 'lda_output', cleaning up.
* updated update_archive.py script to move output elements out
* added clear_lda_output.py script to restore state of converter generated h5
2014-11-07 18:01:00 +01:00

25 lines
573 B
Python

import h5py
import sys
import subprocess
if len(sys.argv) < 2:
print "Usage: python clear_lda_output.py archive"
sys.exit()
print """
This script is to remove any SumkLDA generated output from the h5 archive
and to restore it to the original post-converter state.
"""
filename = sys.argv[1]
A = h5py.File(filename)
del(A["lda_output"])
A.close()
# Repack to reclaim disk space
retcode = subprocess.call(["h5repack","-i%s"%filename, "-otemphgfrt.h5"])
if retcode != 0:
print "h5repack failed!"
else:
subprocess.call(["mv","-f","temphgfrt.h5","%s"%filename])