add RAM usage to verbose per image metadata

Author: i-jey

setup.py

@@ -47,6 +47,7 @@ def readme():
         "numba==0.56.0",
         "Jinja2==3.1.3",
         "xhtml2pdf==0.2.11",
+        "psutil==5.5.1",
     ],
     extras_require={
         "dev": [

ulc_mm_package/QtGUI/scope_op.py

@@ -6,12 +6,12 @@
 """
 
 import logging
-
-from typing import Any
 from time import sleep, perf_counter
+from typing import Any
 
 import cv2
 import numpy as np
+import psutil
 from PyQt5.QtCore import QObject, pyqtSignal, pyqtSlot
 from transitions import Machine, State
 
@@ -133,6 +133,7 @@ def __init__(self):
         self.ambient_pressure = None
         self.mscope = None
         self.digits = int(np.log10(MAX_FRAMES - 1)) + 1
+        self.vmem = psutil.virtual_memory()
 
         self._set_exp_variables()
 
@@ -1152,6 +1153,8 @@ def run_experiment(self, img, timestamp) -> None:
         t1 = perf_counter()
         self._update_metadata_if_verbose("datastorage.writeData", t1 - t0)
 
+        mem_usage = int(self.vmem.used / 1024**2)
+
         for key in PERIODIC_METADATA_KEYS:
             val = self.img_metadata.get(key, None)
             if val is not None:

ulc_mm_package/scope_constants.py

@@ -198,6 +198,7 @@ def IMG_HEIGHT(self) -> int:
     "datastorage.writeData",
     "yogo_qsize",
     "ssaf_qsize",
+    "mem_usage_mb",
 ]
 
 if VERBOSE: