diff --git a/visionipc/visionipc.pxd b/visionipc/visionipc.pxd index 87999e2..d508e2e 100644 --- a/visionipc/visionipc.pxd +++ b/visionipc/visionipc.pxd @@ -3,6 +3,7 @@ from libcpp.string cimport string from libcpp.vector cimport vector +from libcpp.set cimport set from libc.stdint cimport uint32_t, uint64_t from libcpp cimport bool @@ -40,3 +41,5 @@ cdef extern from "cereal/visionipc/visionipc_client.h": VisionBuf * recv(VisionIpcBufExtra *, int) bool connect(bool) bool is_connected() + @staticmethod + set[VisionStreamType] getAvailableStreams(string, bool) diff --git a/visionipc/visionipc_pyx.pyx b/visionipc/visionipc_pyx.pyx index c362319..531e7a5 100644 --- a/visionipc/visionipc_pyx.pyx +++ b/visionipc/visionipc_pyx.pyx @@ -97,3 +97,7 @@ cdef class VisionIpcClient: def is_connected(self): return self.client.is_connected() + + @staticmethod + def available_streams(string name, bool block): + return cppVisionIpcClient.getAvailableStreams(name, block)