From abaec6232ccb57f912fbda36c2a4532da280e28a Mon Sep 17 00:00:00 2001 From: Romain Bazile Date: Fri, 15 Jan 2021 12:36:21 +0100 Subject: [PATCH] imager.py: fix indent --- scripts/planktoscope/imager.py | 69 +++++++++++++++++++++++++++++----- 1 file changed, 59 insertions(+), 10 deletions(-) diff --git a/scripts/planktoscope/imager.py b/scripts/planktoscope/imager.py index d817b91..1eb3d6e 100644 --- a/scripts/planktoscope/imager.py +++ b/scripts/planktoscope/imager.py @@ -130,7 +130,7 @@ class ImagerProcess(multiprocessing.Process): """This class contains the main definitions for the imager of the PlanktoScope""" @logger.catch - def __init__(self, stop_event, iso=100, shutter_speed=1): + def __init__(self, stop_event): """Initialize the Imager class Args: @@ -179,6 +179,7 @@ class ImagerProcess(multiprocessing.Process): logger.exception( f"An exception has occured when starting up raspimjpeg: {e}" ) + logger.error("Shutting off the process now") exit(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" ) - self.__iso = iso - self.__shutter_speed = shutter_speed - self.__exposure_mode = "fixedfps" + self.__iso = 0 + self.__shutter_speed = 100 + self.__exposure_mode = "off" self.__white_balance = "off" self.__white_balance_gain = ( configuration.get("wb_red_gain", 2.00) * 100, configuration.get("wb_blue_gain", 1.40) * 100, ) + self.__image_gain = (100, 100) self.__base_path = "/home/pi/data/img" # 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" ) 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!") @@ -465,6 +476,40 @@ class ImagerProcess(multiprocessing.Process): f'{"status":"Error: White balance mode {self.__white_balance} is not valid"}', ) 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 self.imager_client.client.publish( "status/imager", '{"status":"Camera settings updated"}' @@ -538,14 +583,14 @@ class ImagerProcess(multiprocessing.Process): "action": "move", "direction": self.__pump_direction, "volume": self.__pump_volume, - "flowrate": 2, + "flowrate": 0.25, } ), ) def __state_imaging(self): # subscribe to status/pump - self.imager_client.client.subscribe("status/pump") + # self.imager_client.client.subscribe("status/pump") # Definition of the few important metadata local_metadata = { @@ -606,7 +651,7 @@ class ImagerProcess(multiprocessing.Process): logger.info("Exporting the metadata to a metadata.json") metadata_filepath = os.path.join(self.__export_path, "metadata.json") 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( f"Metadata dumped in {metadata_file} are {self.__global_metadata}" ) @@ -685,9 +730,9 @@ class ImagerProcess(multiprocessing.Process): planktoscope.light.ready() else: # 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) @@ -722,7 +767,11 @@ class ImagerProcess(multiprocessing.Process): self.__state_capture() 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 ################################################################################