Initial commit

openflexure_microscope/camera/config.py

@@ -0,0 +1,37 @@
+import yaml
+
+TYPES = {
+    'stream_resolution': tuple,
+    'video_resolution': tuple,
+    'image_resolution': tuple,
+    'numpy_resolution': tuple,
+    'jpeg_quality': int,
+    'framerate': int,
+    'awb_mode': str,
+    'red_gain': float,
+    'blue_gain': float,
+    'shutter_speed': int,
+    'saturation': int,
+    'analog_gain': float,
+    'digital_gain': float,
+}
+
+def convert_config(config):
+    """Convert datatype of config based on type dictionary"""
+    global TYPES
+
+    for key in config:
+        if key in TYPES:
+            config[key] = TYPES[key](config[key])  
+    
+    return config
+
+def load_config(yaml_path):
+    with open(yaml_path) as config_file:
+        return convert_config(yaml.load(config_file))
+
+if __name__ == "__main__":
+    from pprint import pprint
+
+    config = load_config('config_picamera.yaml')
+    pprint(config)