Merge pull request #61 from licode/multi_headers

DEV: _get_images can take header or list of headers as input
NSLS-II-CSX · Apr 3, 2018 · eb3ab7f · eb3ab7f
2 parents 3727870 + 8a4e8ca
commit eb3ab7f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/csxtools/utils.py b/csxtools/utils.py
@@ -178,7 +178,10 @@ def get_images_to_3D(images, dtype=None):
 
 def _get_images(header, tag, roi=None):
     t = ttime.time()
-    images = header.db.get_images(header, tag)
+    if isinstance(header, (list, tuple)):
+        images = [img for h in header for img in h.data(tag)]
+    else:
+        images = header.data(tag)
     t = ttime.time() - t
     logger.info("Took %.3f seconds to read data using get_images", t)