borg_public/scripts/mod_restart_file.py

52 lines
1.2 KiB
Python
Raw Normal View History

2023-05-29 10:41:03 +02:00
#+
# ARES/HADES/BORG Package -- ./scripts/mod_restart_file.py
# Copyright (C) 2014-2020 Guilhem Lavaux <guilhem.lavaux@iap.fr>
# Copyright (C) 2009-2020 Jens Jasche <jens.jasche@fysik.su.se>
#
# Additional contributions from:
# Guilhem Lavaux <guilhem.lavaux@iap.fr> (2023)
#
#+
import h5py
import numpy as np
#set chain
chain_path="."
#set reference file
refdir ='/scratch/jasche/panphasia_run_lc/'
fref=refdir+'restart.h5_0'
#set target file
tardir ='/scratch/jasche/panphasia_run_h/'
ftar=tardir+'restart.h5_0'
#read density files from reference file
print fref
dref = h5py.File(fref, "r")
dref_final_density=dref['/scalars/BORG_final_density']
dref_s_field=dref['/scalars/s_field']
dref_s_hat_field=dref['/scalars/s_hat_field']
dtar = h5py.File(ftar, "r+")
dtar_final_density=dtar['/scalars/BORG_final_density']
dtar_s_field=dtar['/scalars/s_field']
dtar_s_hat_field=dtar['/scalars/s_hat_field']
'''
WARNING: At this point you will irretrievably
modify your restart file!!!!!!
'''
dtar_final_density[...] = dref_final_density[...]
dtar_s_field[...] = dref_s_field[...]
dtar_s_hat_field[...] = dref_s_hat_field[...]
dref.close()
dtar.close()