From 54f8eb954eb4f3dffbcc15f2283dc94d24de79a8 Mon Sep 17 00:00:00 2001 From: deanlee Date: Tue, 21 Mar 2023 15:02:33 +0800 Subject: [PATCH] available stream --- visionipc/visionipc.pxd | 3 +++ visionipc/visionipc_pyx.pyx | 4 ++++ 2 files changed, 7 insertions(+) diff --git a/visionipc/visionipc.pxd b/visionipc/visionipc.pxd index 87999e27b..d508e2e39 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 c36231915..531e7a5ef 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)