# -*- coding: utf-8 -*-
#
# This file is subject to the terms and conditions defined in
# file 'LICENSE', which is part of this source code package.
#
#
import pickle
try:
from mpi4py import MPI
except ImportError:#pragma: no cover
print("MPI capabilities not available")
[docs]
def UpdateFileName(fileName):
"""
Appends filename with numbering in case of parallel execution
Parameters
----------
fileName : str
name of the file on disk
Returns
-------
str
"""
if MPI.COMM_WORLD.Get_size() > 1: # pragma: no cover
fileName += "-" + str(MPI.COMM_WORLD.Get_rank()+1).zfill(3)
return fileName
[docs]
def SaveState(fileName, obj, extension = "pkl"):
"""
Saves the data structure on disk
Parameters
----------
fileName : str
name of the file on disk
obj : custom_data_format
object to save on disk
extension : str, optional
file extension
"""
outputName = UpdateFileName(fileName) + "." + extension
output = open(outputName, "wb")
pickle.dump(obj, output)
output.close()
[docs]
def LoadState(fileName, extension = "pkl"):
"""
Read the data structure from disk
Parameters
----------
fileName : str
name of the file on disk
extension : str, optional
file extension
Returns
-------
custom_data_format
"""
inputName = UpdateFileName(fileName) + "." + extension
return pickle.load(open(inputName, "rb"))
if __name__ == "__main__":# pragma: no cover
from Mordicus import RunTestFile
RunTestFile(__file__)