/usr/share/pyshared/ase/tasks/io.py is in python-ase 3.6.0.2515-1.1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 | import numpy as np
from ase.parallel import world
try:
import json
except ImportError:
json = None
if json is None:
def dumps(obj):
if isinstance(obj, str):
return '"' + obj + '"'
if isinstance(obj, (int, float)):
return repr(obj)
if isinstance(obj, dict):
return '{' + ','.join(dumps(key) + ':' + dumps(value)
for key, value in obj.items()) + '}'
return '[' + ','.join(dumps(value) for value in obj) + ']'
loads = eval
else:
class NDArrayEncoder(json.JSONEncoder):
def default(self, obj):
if isinstance(obj, np.ndarray):
return obj.tolist()
return json.JSONEncoder.default(self, obj)
dumps = NDArrayEncoder().encode
loads = json.loads
def numpyfy(obj):
if isinstance(obj, dict):
return dict((key, numpyfy(value)) for key, value in obj.items())
if isinstance(obj, list):
try:
obj = np.array(obj)
except ValueError:
obj = [numpyfy(value) for value in obj]
return obj
def write_json(name, atoms, results):
if world.rank == 0:
fd = open(name + '.json', 'w')
fd.write(dumps(results))
fd.close()
def read_json(name):
fd = open(name + '.json', 'r')
results = loads(fd.read())
fd.close()
return numpyfy(results)
|