Minor code cleanups

This commit is contained in:
kris
2019-06-21 22:08:22 +01:00
parent b0e83d2faf
commit b2c00784b0
6 changed files with 20 additions and 28 deletions

View File

@ -116,12 +116,13 @@ class FileFrameGrabber(FrameGrabber):
def worker():
"""Invoke bmp2dhr to encode input image frames and push to queue."""
decode = (
_dhgr_decode if self.video_mode == VideoMode.DHGR else
_hgr_decode
)
for _idx, _frame in enumerate(self._frame_grabber()):
if self.video_mode == VideoMode.DHGR:
res = _dhgr_decode(_idx, _frame)
else:
res = _hgr_decode(_idx, _frame)
q.put(res)
q.put(decode(_idx, _frame))
q.put((None, None))