rahix/openflexure-microscope-server
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Handle passing a CaptureObject instance directly to capture()

Browse source
This commit is contained in:
Joel Collins 2020-01-21 16:03:19 +00:00
parent e2f36d5e33
commit e6c4664cb8
2 changed files with 18 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

17
openflexure_microscope/camera/mock.py
View file

@ -16,7 +16,7 @@ import logging
# Type hinting
from typing import Tuple
from openflexure_microscope.camera.base import BaseCamera
from openflexure_microscope.camera.base import BaseCamera, CaptureObject
"""
@ -189,14 +189,19 @@ class MockStreamer(BaseCamera):
bayer (bool): Store raw bayer data in capture
"""
if isinstance(output, CaptureObject):
target = output.file
else:
target = target
with self.lock:
if isinstance(output, str):
output = open(output, "wb")
if isinstance(target, str):
target = open(target, "wb")
output.write(self.stream.getvalue())
target.write(self.stream.getvalue())
if isinstance(output, str):
output.close()
if isinstance(target, str):
target.close()
# HANDLE STREAM FRAMES