[galactica] update import routine

This commit is contained in:
Noe Brucy
2021-06-22 10:40:00 +02:00
parent d4d010216c
commit baed67fd47
+3 -27
View File
@@ -1,34 +1,10 @@
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# This file is part of the 'astrophysix' Python package.
#
# Copyright © Commissariat a l'Energie Atomique et aux Energies Alternatives (CEA)
#
# FREE SOFTWARE LICENCING
# -----------------------
# This software is governed by the CeCILL license under French law and abiding by the rules of distribution of free
# software. You can use, modify and/or redistribute the software under the terms of the CeCILL license as circulated by
# CEA, CNRS and INRIA at the following URL: "http://www.cecill.info". As a counterpart to the access to the source code
# and rights to copy, modify and redistribute granted by the license, users are provided only with a limited warranty
# and the software's author, the holder of the economic rights, and the successive licensors have only limited
# liability. In this respect, the user's attention is drawn to the risks associated with loading, using, modifying
# and/or developing or reproducing the software by the user in light of its specific status of free software, that may
# mean that it is complicated to manipulate, and that also therefore means that it is reserved for developers and
# experienced professionals having in-depth computer knowledge. Users are therefore encouraged to load and test the
# software's suitability as regards their requirements in conditions enabling the security of their systems and/or data
# to be ensured and, more generally, to use and operate it in the same conditions as regards security. The fact that
# you are presently reading this means that you have had knowledge of the CeCILL license and that you accept its terms.
#
#
# COMMERCIAL SOFTWARE LICENCING
# -----------------------------
# You can obtain this software from CEA under other licencing terms for commercial purposes. For this you will need to
# negotiate a specific contract with a legal representative of CEA.
#
from __future__ import print_function, unicode_literals from __future__ import print_function, unicode_literals
import os import os
import argparse
import h5py import h5py
import numpy as np import numpy as np
import matplotlib as mpl import matplotlib as mpl
@@ -467,7 +443,7 @@ for simu in proj.simulations:
for df in snap.datafiles: for df in snap.datafiles:
name = df[FileType.JPEG_FILE].filename name = df[FileType.JPEG_FILE].filename
name = os.path.splitext(name)[0] + ".h5" name = os.path.splitext(name)[0] + ".h5"
if include_hdf5 and not df.name in ["slice_velphi_z", "slice_velr_z"]: if include_hdf5 and df.name not in ["slice_velphi_z", "slice_velr_z"]:
h5 = h5py.File(out_dir + "/" + name, "w") h5 = h5py.File(out_dir + "/" + name, "w")
p = h5.create_group("plot") p = h5.create_group("plot")
df.plot_info.hsp_save_to_h5(p) df.plot_info.hsp_save_to_h5(p)