57 lines
1.8 KiB
Python
57 lines
1.8 KiB
Python
from openflexure_microscope.devel import (
|
|
MicroscopePlugin,
|
|
MicroscopeViewPlugin,
|
|
JsonResponse,
|
|
request,
|
|
jsonify,
|
|
taskify,
|
|
)
|
|
|
|
import logging
|
|
|
|
from .recalibrate_utils import recalibrate_camera, auto_expose_and_freeze_settings
|
|
|
|
|
|
class RecalibrateAPIView(MicroscopeViewPlugin):
|
|
def post(self):
|
|
logging.info("Starting microscope recalibration...")
|
|
task = taskify(self.plugin.recalibrate)()
|
|
|
|
# Return a handle on the autofocus task
|
|
return jsonify(task.state), 202
|
|
|
|
|
|
class Plugin(MicroscopePlugin):
|
|
"""
|
|
A set of default plugins
|
|
"""
|
|
|
|
api_views = {"/recalibrate": RecalibrateAPIView}
|
|
|
|
def recalibrate(self):
|
|
"""Reset the camera's settings.
|
|
|
|
This generates new gains, exposure time, and lens shading
|
|
table such that the background is as uniform as possible
|
|
with a gray level of 230. It takes a little while to run.
|
|
"""
|
|
scamera = self.microscope.camera
|
|
with scamera.lock:
|
|
assert not scamera.status[
|
|
"record_active"
|
|
], "Can't recalibrate while recording!"
|
|
streaming = scamera.status["stream_active"]
|
|
if streaming:
|
|
logging.info("Stopping stream before recalibration")
|
|
scamera.stop_stream_recording(resolution=(640, 480))
|
|
old_resolution = scamera.camera.resolution
|
|
try:
|
|
scamera.camera.resolution = (640, 480)
|
|
auto_expose_and_freeze_settings(scamera.camera)
|
|
recalibrate_camera(scamera.camera)
|
|
finally:
|
|
scamera.camera.resolution = old_resolution
|
|
self.microscope.save_settings()
|
|
if streaming:
|
|
logging.info("Restarting stream after recalibration")
|
|
scamera.start_stream_recording()
|