[Neuroimaging] Thread-safe ArrayProxy/fileslice?
pauldmccarthy at gmail.com
Wed Feb 8 08:55:11 EST 2017
Does anybody have experience accessing image data through the ArrayProxy
class (or functions in the fileslice module) in a multi-threaded
environment? I am visualising large 4D images which are kept on disk (via
my indexed_gzip module), and am having trouble when accessing the data from
multiple threads, as the seek/read pairs from different threads will
occasionally become intertwined with each other.
My hacky workaround is to patch the ArrayProxy.__getitem__ method, and add
a threading.Lock to each instance, as follows:
import nibabel as nib
import nibabel.arrayproxy as ap
def ArrayProxy__getitem__(self, slc):
if not hasattr(self, '_thread_lock'):
return ap.ArrayProxy.__real_getitem__(self, slc)
# Patch ArrayProxy.__getitem__
ap.ArrayProxy.__real_getitem__ = ap.ArrayProxy.__getitem__
ap.ArrayProxy.__getitem__ = ArrayProxy__getitem__
# Then add a lock to instances
# which need to be thread-safe
img = nib.load('MNI152_T1_2mm.nii.gz')
img.dataobj._thread_lock = threading.Lock()
This is the first thing I came up with, although I will probably end up
adding the lock to the fileobj, and patching the fileslice.fileslice
function instead. Unless there are any better ideas?
-------------- next part --------------
An HTML attachment was scrubbed...
More information about the Neuroimaging