Click here to go to the corresponding page for the latest version of DIALS
Source code for dxtbx.serialize.load
from __future__ import absolute_import, division, print_function
import json
import os
import six
from dxtbx.model.crystal import CrystalFactory
from dxtbx.serialize.imageset import imageset_from_dict
def _decode_list(data):
"""Decode a list to str from unicode."""
if six.PY3:
return data
rv = []
for item in data:
if isinstance(item, six.text_type):
item = item.encode("utf-8")
elif isinstance(item, list):
item = _decode_list(item)
elif isinstance(item, dict):
item = _decode_dict(item)
rv.append(item)
return rv
def _decode_dict(data):
"""Decode a dict to str from unicode."""
if six.PY3:
return data
rv = {}
for key, value in data.items():
if isinstance(key, six.text_type):
key = key.encode("utf-8")
if isinstance(value, six.text_type):
value = value.encode("utf-8")
elif isinstance(value, list):
value = _decode_list(value)
elif isinstance(value, dict):
value = _decode_dict(value)
rv[key] = value
return rv
[docs]def imageset_from_string(string, directory=None):
"""Load the string and return the models.
Params:
string The JSON string to load
Returns:
The models
"""
return imageset_from_dict(
json.loads(string, object_hook=_decode_dict), directory=directory
)
[docs]def imageset(filename):
"""Load the given JSON file.
Params:
infile The input filename
Returns:
The models
"""
# If the input is a string then open and read from that file
filename = os.path.abspath(filename)
directory = os.path.dirname(filename)
with open(filename, "r") as infile:
return imageset_from_string(infile.read(), directory=directory)
[docs]def datablock(filename, check_format=True):
"""Load a given JSON or pickle file.
Params:
filename The input filename
Returns:
The datablock
"""
# Resolve recursive import
from dxtbx.datablock import DataBlockFactory
return DataBlockFactory.from_serialized_format(filename, check_format=check_format)
[docs]def crystal(infile):
"""Load the given JSON file.
Params:
infile The input filename or file object
Returns:
The models
"""
# If the input is a string then open and read from that file
if isinstance(infile, str):
with open(infile, "r") as infile:
return CrystalFactory.from_dict(json.loads(infile.read()))
# Otherwise assume the input is a file and read from it
else:
return CrystalFactory.from_dict(json.loads(infile.read()))
[docs]def experiment_list(infile, check_format=True):
"""Load an experiment list from a serialzied format."""
# Resolve recursive import
from dxtbx.model.experiment_list import ExperimentListFactory
if infile and hasattr(infile, "__fspath__"):
infile = (
infile.__fspath__()
) # Resolve file system path (PEP-519) object to string.
return ExperimentListFactory.from_serialized_format(
infile, check_format=check_format
)
