You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
94 lines
1.9 KiB
94 lines
1.9 KiB
8 years ago
|
#!/usr/bin/env python
|
||
|
|
import os
|
||
|
|
import time
|
||
|
|
import subprocess
|
||
|
|
from cffi import FFI
|
||
|
|
import ctypes
|
||
|
|
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
gf_dir = os.path.dirname(os.path.abspath(__file__))
|
||
|
|
|
||
|
|
subprocess.check_call(["make"], cwd=gf_dir)
|
||
|
|
|
||
|
|
|
||
|
|
ffi = FFI()
|
||
|
|
ffi.cdef("""
|
||
|
|
|
||
|
|
typedef enum VisionStreamType {
|
||
|
|
VISION_STREAM_UI_BACK,
|
||
|
|
VISION_STREAM_UI_FRONT,
|
||
|
|
VISION_STREAM_YUV,
|
||
|
|
VISION_STREAM_MAX,
|
||
|
|
} VisionStreamType;
|
||
|
|
|
||
|
|
typedef struct VisionUIInfo {
|
||
|
|
int big_box_x, big_box_y;
|
||
|
|
int big_box_width, big_box_height;
|
||
|
|
int transformed_width, transformed_height;
|
||
|
|
|
||
|
|
int front_box_x, front_box_y;
|
||
|
|
int front_box_width, front_box_height;
|
||
|
|
} VisionUIInfo;
|
||
|
|
|
||
|
|
typedef struct VisionStreamBufs {
|
||
|
|
VisionStreamType type;
|
||
|
|
|
||
|
|
int width, height, stride;
|
||
|
|
size_t buf_len;
|
||
|
|
|
||
|
|
union {
|
||
|
|
VisionUIInfo ui_info;
|
||
|
|
} buf_info;
|
||
|
|
} VisionStreamBufs;
|
||
|
|
|
||
|
|
typedef struct VisionBuf {
|
||
|
|
int fd;
|
||
|
|
size_t len;
|
||
|
|
void* addr;
|
||
|
|
} VisionBuf;
|
||
|
|
|
||
|
|
typedef struct VisionBufExtra {
|
||
|
|
uint32_t frame_id; // only for yuv
|
||
|
|
} VisionBufExtra;
|
||
|
|
|
||
|
|
typedef struct VisionStream {
|
||
|
|
int ipc_fd;
|
||
|
|
int last_idx;
|
||
|
|
int num_bufs;
|
||
|
|
VisionStreamBufs bufs_info;
|
||
|
|
VisionBuf *bufs;
|
||
|
|
} VisionStream;
|
||
|
|
|
||
|
|
int visionstream_init(VisionStream *s, VisionStreamType type, bool tbuffer, VisionStreamBufs *out_bufs_info);
|
||
|
|
VisionBuf* visionstream_get(VisionStream *s, VisionBufExtra *out_extra);
|
||
|
|
void visionstream_destroy(VisionStream *s);
|
||
|
|
|
||
|
|
"""
|
||
|
|
)
|
||
|
|
|
||
|
|
clib = ffi.dlopen(os.path.join(gf_dir, "libvisionipc.so"))
|
||
|
|
|
||
|
|
|
||
|
|
def getframes():
|
||
|
|
s = ffi.new("VisionStream*")
|
||
|
|
buf_info = ffi.new("VisionStreamBufs*")
|
||
|
|
err = clib.visionstream_init(s, clib.VISION_STREAM_UI_BACK, True, buf_info)
|
||
|
|
assert err == 0
|
||
|
|
|
||
|
|
w = buf_info.width
|
||
|
|
h = buf_info.height
|
||
|
|
assert buf_info.stride == w*3
|
||
|
|
assert buf_info.buf_len == w*h*3
|
||
|
|
|
||
|
|
while True:
|
||
|
|
buf = clib.visionstream_get(s, ffi.NULL)
|
||
|
|
|
||
|
|
pbuf = ffi.buffer(buf.addr, buf.len)
|
||
|
|
yield np.frombuffer(pbuf, dtype=np.uint8).reshape((h, w, 3))
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
for buf in getframes():
|
||
|
|
print buf.shape, buf[101, 101]
|