openflexure-microscope-server/openflexure_microscope/api/v2/views/streams.py
2020-11-30 13:36:45 +00:00

55 lines
1.7 KiB
Python

from flask import Response
from labthings import find_component
from labthings.views import PropertyView
def gen(camera):
"""Video streaming generator function."""
while True:
# the obtained frame is a jpeg
frame: bytes = camera.stream.getframe()
yield (b"--frame\r\n" b"Content-Type: image/jpeg\r\n\r\n" + frame + b"\r\n")
class MjpegStream(PropertyView):
"""
Real-time MJPEG stream from the microscope camera
"""
responses = {200: {"content_type": "multipart/x-mixed-replace"}}
def get(self):
"""
MJPEG stream from the microscope camera.
Note: While the code actually getting frame data from a camera and storing it in
camera.frame runs in a thread, the gen(microscope.camera) generator does not.
This response is therefore blocking. The generator just yields the most recent
frame from the camera object, passed to the Flask response, and then repeats until
the connection is closed.
Without monkey patching, or using a native threaded server, the stream
will block all proceeding requests.
"""
microscope = find_component("org.openflexure.microscope")
return Response(
gen(microscope.camera), mimetype="multipart/x-mixed-replace; boundary=frame"
)
class SnapshotStream(PropertyView):
"""
Single JPEG snapshot from the camera stream
"""
responses = {200: {"content_type": "image/jpeg", "description": "Snapshot taken"}}
def get(self):
"""
Single snapshot from the camera stream
"""
microscope = find_component("org.openflexure.microscope")
return Response(microscope.camera.stream.getframe(), mimetype="image/jpeg")