Verify if video file exists. Fix #53

Xilinx · Jan 18, 2022 · e1ea8f4 · e1ea8f4
1 parent 1b5eb7a
commit e1ea8f4
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/pynq_composable/video.py b/pynq_composable/video.py
@@ -294,6 +294,9 @@ def __init__(self, ol: Overlay, filename: str,
             raise ValueError("filename ({}) is not an string or integer"
                              .format(filename))
 
+        if isinstance(filename, str) and not os.path.exists(filename):
+            raise RuntimeError("File {} does not exists".format(filename))
+
         self._file = filename
         self._hdmi_out = ol.video.hdmi_out
         self._videoIn = None
@@ -431,6 +434,13 @@ def __init__(self, ol: Overlay, filename: str,
             Xilinx VideoDMA IP core
         """
 
+        if not isinstance(filename, str) and not isinstance(filename, int):
+            raise ValueError("filename ({}) is not an string or integer"
+                             .format(filename))
+
+        if isinstance(filename, str) and not os.path.exists(filename):
+            raise RuntimeError("File {} does not exists".format(filename))
+
         self._file = filename
         self.vdma = ol.video.axi_vdma
         self.mode = mode