-
Notifications
You must be signed in to change notification settings - Fork 0
/
__init__.py
63 lines (47 loc) · 2.02 KB
/
__init__.py
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
57
58
59
60
61
62
63
import logging
import os
import sys
import libtbx.load_env
import dxtbx.format.Registry
if sys.version_info.major == 2:
sys.exit("Python 2 is no longer supported")
# Ensures that HDF5 has the conda_base plugin path configured.
#
# Ideally this will be properly configured by the conda environment.
# However, currently the dials-installer will not install a path-correct
# conda_base folder, so it needs to be updated manually.
_hdf5_plugin_path = libtbx.env.under_base(os.path.join("lib", "hdf5", "plugin"))
# Inject via the environment if h5py not used yet, or else use h5py
if "h5py" not in sys.modules:
os.environ["HDF5_PLUGIN_PATH"] = (
_hdf5_plugin_path + os.pathsep + os.getenv("HDF5_PLUGIN_PATH", "")
)
else:
# We've already loaded h5py, so setting the environment variable won't work
# We need to use the h5py API to add a plugin path
import h5py
h5_plugin_paths = [h5py.h5pl.get(i).decode() for i in range(h5py.h5pl.size())]
if _hdf5_plugin_path not in h5_plugin_paths:
h5py.h5pl.prepend(_hdf5_plugin_path.encode())
logging.getLogger("dxtbx").addHandler(logging.NullHandler())
class IncorrectFormatError(RuntimeError):
"""
An exception class for an incorrect format error
"""
def __init__(self, format_instance, filename):
super(IncorrectFormatError, self).__init__(
"Could not open %s as %s" % (filename, str(format_instance))
)
self.args = (format_instance, filename)
def load(filename):
"""Use DXTBX to load the input filename.
:param filename: The input filename
:type filename: os.PathLike or str or bytes
:returns: A dxtbx Format-subclass instance for the file type
:raises IOError: if the file format could not be determined
"""
# Unwrap PEP-519-style objects. This covers py.path, pathlib, ...
if hasattr(filename, "__fspath__"):
filename = filename.__fspath__()
format_instance = dxtbx.format.Registry.get_format_class_for_file(filename)
return format_instance(filename)