summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--py/pykms/pykmsbase.cpp2
-rwxr-xr-xpy/tests/cam.py3
2 files changed, 4 insertions, 1 deletions
diff --git a/py/pykms/pykmsbase.cpp b/py/pykms/pykmsbase.cpp
index a963843..382ef6f 100644
--- a/py/pykms/pykmsbase.cpp
+++ b/py/pykms/pykmsbase.cpp
@@ -224,6 +224,8 @@ void init_pykmsbase(py::module& m)
.value("RGBA1010102", PixelFormat::RGBA1010102)
.value("BGRA1010102", PixelFormat::BGRA1010102);
+ m.def("fourcc_to_pixelformat", &FourCCToPixelFormat);
+
py::enum_<SyncPolarity>(m, "SyncPolarity")
.value("Undefined", SyncPolarity::Undefined)
.value("Positive", SyncPolarity::Positive)
diff --git a/py/tests/cam.py b/py/tests/cam.py
index b7294ed..957eb11 100755
--- a/py/tests/cam.py
+++ b/py/tests/cam.py
@@ -9,11 +9,12 @@ import time
parser = argparse.ArgumentParser()
parser.add_argument("width", type=int)
parser.add_argument("height", type=int)
+parser.add_argument("fourcc", type=str, nargs="?", default="YUVY")
args = parser.parse_args()
w = args.width
h = args.height
-fmt = pykms.PixelFormat.YUYV
+fmt = pykms.fourcc_to_pixelformat(args.fourcc)
print("Capturing in {}x{}".format(w, h))