imager.py: fix indent

This commit is contained in:
Romain Bazile 2021-01-15 12:36:21 +01:00
parent 862d687bfa
commit abaec6232c

View file

@ -130,7 +130,7 @@ class ImagerProcess(multiprocessing.Process):
"""This class contains the main definitions for the imager of the PlanktoScope""" """This class contains the main definitions for the imager of the PlanktoScope"""
@logger.catch @logger.catch
def __init__(self, stop_event, iso=100, shutter_speed=1): def __init__(self, stop_event):
"""Initialize the Imager class """Initialize the Imager class
Args: Args:
@ -179,6 +179,7 @@ class ImagerProcess(multiprocessing.Process):
logger.exception( logger.exception(
f"An exception has occured when starting up raspimjpeg: {e}" f"An exception has occured when starting up raspimjpeg: {e}"
) )
logger.error("Shutting off the process now")
exit(1) exit(1)
if self.__camera.sensor_name == "IMX219": # Camera v2.1 if self.__camera.sensor_name == "IMX219": # Camera v2.1
@ -191,14 +192,15 @@ class ImagerProcess(multiprocessing.Process):
f"The connected camera {self.__camera.sensor_name} is not recognized, please check your camera" f"The connected camera {self.__camera.sensor_name} is not recognized, please check your camera"
) )
self.__iso = iso self.__iso = 0
self.__shutter_speed = shutter_speed self.__shutter_speed = 100
self.__exposure_mode = "fixedfps" self.__exposure_mode = "off"
self.__white_balance = "off" self.__white_balance = "off"
self.__white_balance_gain = ( self.__white_balance_gain = (
configuration.get("wb_red_gain", 2.00) * 100, configuration.get("wb_red_gain", 2.00) * 100,
configuration.get("wb_blue_gain", 1.40) * 100, configuration.get("wb_blue_gain", 1.40) * 100,
) )
self.__image_gain = (100, 100)
self.__base_path = "/home/pi/data/img" self.__base_path = "/home/pi/data/img"
# Let's make sure the base path exists # Let's make sure the base path exists
@ -261,6 +263,15 @@ class ImagerProcess(multiprocessing.Process):
"A timeout has occured when setting the white balance gain, trying again" "A timeout has occured when setting the white balance gain, trying again"
) )
self.__camera.white_balance_gain = self.__white_balance_gain self.__camera.white_balance_gain = self.__white_balance_gain
time.sleep(0.1)
try:
self.__camera.image_gain = self.__image_gain
except TimeoutError as e:
logger.error(
"A timeout has occured when setting the image gain, trying again"
)
self.__camera.image_gain = self.__image_gain
logger.success("planktoscope.imager is initialised and ready to go!") logger.success("planktoscope.imager is initialised and ready to go!")
@ -465,6 +476,40 @@ class ImagerProcess(multiprocessing.Process):
f'{"status":"Error: White balance mode {self.__white_balance} is not valid"}', f'{"status":"Error: White balance mode {self.__white_balance} is not valid"}',
) )
return return
if "image_gain" in settings:
if "analog" in settings["image_gain"]:
logger.debug(
f"Updating the camera analog gain to {settings['image_gain']}"
)
self.__image_gain = (
settings["image_gain"].get("analog", self.__image_gain[0]),
self.__image_gain[1],
)
if "digital" in settings["image_gain"]:
logger.debug(
f"Updating the camera digital gain to {settings['image_gain']}"
)
self.__image_gain = (
self.__image_gain[0],
settings["image_gain"].get("digital", self.__image_gain[1]),
)
logger.debug(f"Updating the camera image gain to {self.__image_gain}")
try:
self.__camera.image_gain = self.__image_gain
except TimeoutError as e:
logger.error(
"A timeout has occured when setting the image gain, trying again"
)
self.__camera.image_gain = self.__image_gain
except ValueError as e:
logger.error("The requested image gain is not valid!")
self.imager_client.client.publish(
"status/imager",
'{"status":"Error: Chosen image gain is not valid"}',
)
return
# Publish the status "Config updated" to via MQTT to Node-RED # Publish the status "Config updated" to via MQTT to Node-RED
self.imager_client.client.publish( self.imager_client.client.publish(
"status/imager", '{"status":"Camera settings updated"}' "status/imager", '{"status":"Camera settings updated"}'
@ -538,14 +583,14 @@ class ImagerProcess(multiprocessing.Process):
"action": "move", "action": "move",
"direction": self.__pump_direction, "direction": self.__pump_direction,
"volume": self.__pump_volume, "volume": self.__pump_volume,
"flowrate": 2, "flowrate": 0.25,
} }
), ),
) )
def __state_imaging(self): def __state_imaging(self):
# subscribe to status/pump # subscribe to status/pump
self.imager_client.client.subscribe("status/pump") # self.imager_client.client.subscribe("status/pump")
# Definition of the few important metadata # Definition of the few important metadata
local_metadata = { local_metadata = {
@ -606,7 +651,7 @@ class ImagerProcess(multiprocessing.Process):
logger.info("Exporting the metadata to a metadata.json") logger.info("Exporting the metadata to a metadata.json")
metadata_filepath = os.path.join(self.__export_path, "metadata.json") metadata_filepath = os.path.join(self.__export_path, "metadata.json")
with open(metadata_filepath, "w") as metadata_file: with open(metadata_filepath, "w") as metadata_file:
json.dump(self.__global_metadata, metadata_file, indent="4") json.dump(self.__global_metadata, metadata_file, indent=4)
logger.debug( logger.debug(
f"Metadata dumped in {metadata_file} are {self.__global_metadata}" f"Metadata dumped in {metadata_file} are {self.__global_metadata}"
) )
@ -685,9 +730,9 @@ class ImagerProcess(multiprocessing.Process):
planktoscope.light.ready() planktoscope.light.ready()
else: else:
# We have not reached the final stage, let's keep imaging # We have not reached the final stage, let's keep imaging
self.imager_client.client.subscribe("status/pump") # self.imager_client.client.subscribe("status/pump")
self.__pump_message() # self.__pump_message()
self.__imager.change(planktoscope.imager_state_machine.Waiting) self.__imager.change(planktoscope.imager_state_machine.Waiting)
@ -722,7 +767,11 @@ class ImagerProcess(multiprocessing.Process):
self.__state_capture() self.__state_capture()
return return
elif self.__imager.state.name == ["waiting", "stop"]: elif self.__imager.state.name == "waiting":
self.__imager.change(planktoscope.imager_state_machine.Capture)
return
elif self.__imager.state.name == "stop":
return return
################################################################################ ################################################################################