1 files changed, 116 insertions, 0 deletions
diff --git a/test-cli/test/helpers/camara.py b/test-cli/test/helpers/camara.py
new file mode 100644
index 0000000..b23df74
--- /dev/null
+++ b/test-cli/test/helpers/camara.py
@@ -0,0 +1,116 @@
+import cv2
+import sh
+
+class Camara(object):
+    __device_name = None
+    __device = None
+    __setupScriptPath = ''
+    __w = 1280
+    __h = 720
+    __contrast = 0.0
+    __brightness = 0.0
+    __saturation = 55.0
+    __hue = 0.0
+    __exposure = 166
+
+    def __init__(self, setup_script_path, device="video0", width=1280, height=720):
+        self.__device_name = device
+        self.__w = width
+        self.__h = height
+        self.__setupScriptPath = setup_script_path
+
+    def Close(self):
+        if self.__device is not None:
+            del self.__device
+            self.__device = None
+
+    def Open(self):
+        self.Close()
+        self.__device = cv2.VideoCapture("/dev/{}".format(self.__device_name))
+        if self.__device.isOpened():
+            self.__configure()
+            return True
+        return False
+
+    def getSize(self):
+        return self.__w, self.__h
+
+    def setSize(self, w, h):
+        if self.__device is not None and self.__device.isOpened():
+            self.__w = self.__setCamVar(cv2.CAP_PROP_FRAME_WIDTH, w)
+            self.__h = self.__setCamVar(cv2.CAP_PROP_FRAME_HEIGHT, h)
+        else:
+            self.__w = w
+            self.__h = h
+
+    def setContrast(self, newVal):
+        if self.__device.isOpened():
+            self.__contrast = self.__setCamVar(cv2.CAP_PROP_CONTRAST, newVal)
+        else:
+            self.__contrast = newVal
+
+    def getContrast(self):
+        return self.__contrast
+
+    def setBrightness(self, newVal):
+        if self.__device.isOpened():
+            self.__brightness = self.__setCamVar(cv2.CAP_PROP_BRIGHTNESS, newVal)
+        else:
+            self.__brightness = newVal
+
+    def getBrightness(self):
+        return self.__brightness
+
+    def __configure(self):
+        self.__w = self.__setCamVar(cv2.CAP_PROP_FRAME_WIDTH, self.__w)
+        self.__h = self.__setCamVar(cv2.CAP_PROP_FRAME_HEIGHT, self.__h)
+        sh.bash(self.__setupScriptPath)
+
+    def __setCamVar(self, key, val):
+        valold = cv2.VideoCapture.get(self.__device, key)
+        if valold != val:
+            cv2.VideoCapture.set(self.__device, key, val)
+            t = cv2.VideoCapture.get(self.__device, key)
+            return t
+        return val
+
+    def __getCamVar(self, key):
+        return cv2.VideoCapture.get(self.__device, key);
+
+    def getFrameCount(self):
+        return cv2.VideoCapture.get(self.__device, cv2.CAP_PROP_BUFFERSIZE);
+
+    def getFrame(self):
+        if self.__device.isOpened():
+            retval, image = self.__device.read()
+            if retval:
+                return image
+            return None
+        else:
+            return None
+
+    def getImageSize(self, image):
+        if hasattr(image, 'shape'):
+            return image.shape[1], image.shape[0]
+        else:
+            return (0, 0, 0)
+
+    def showFrame(self, name, frame, w=False, maxTime=3000):
+        cv2.imshow(name, frame)
+        if w:
+            if maxTime == -1:
+                cv2.waitKey()
+            else:
+                cv2.waitKey(maxTime)
+
+    def saveFrame(self, fileName, Frame):
+        cv2.imwrite(fileName, Frame)
+
+    def readImage(self, filename):
+        return cv2.imread('{}'.format(filename), 0)
+
+    def destroyWindow(self, name):
+        cv2.destroyWindow(name)
+
+    def closeWindows(self):
+        cv2.destroyAllWindows()
+\ No newline at end of file