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.
42 lines
1.2 KiB
42 lines
1.2 KiB
#pragma once
|
|
#include <vector>
|
|
#include <string>
|
|
#include <thread>
|
|
#include <atomic>
|
|
#include <map>
|
|
|
|
#include "msgq/ipc.h"
|
|
#include "msgq/visionipc/visionbuf.h"
|
|
|
|
std::string get_endpoint_name(std::string name, VisionStreamType type);
|
|
std::string get_ipc_path(const std::string &name);
|
|
|
|
class VisionIpcServer {
|
|
private:
|
|
cl_device_id device_id = nullptr;
|
|
cl_context ctx = nullptr;
|
|
uint64_t server_id;
|
|
|
|
std::atomic<bool> should_exit = false;
|
|
std::string name;
|
|
std::thread listener_thread;
|
|
|
|
std::map<VisionStreamType, std::atomic<size_t> > cur_idx;
|
|
std::map<VisionStreamType, std::vector<VisionBuf*> > buffers;
|
|
|
|
Context * msg_ctx;
|
|
std::map<VisionStreamType, PubSocket*> sockets;
|
|
|
|
void listener(void);
|
|
|
|
public:
|
|
VisionIpcServer(std::string name, cl_device_id device_id=nullptr, cl_context ctx=nullptr);
|
|
~VisionIpcServer();
|
|
|
|
VisionBuf * get_buffer(VisionStreamType type, int idx = -1);
|
|
|
|
void create_buffers(VisionStreamType type, size_t num_buffers, size_t width, size_t height);
|
|
void create_buffers_with_sizes(VisionStreamType type, size_t num_buffers, size_t width, size_t height, size_t size, size_t stride, size_t uv_offset);
|
|
void send(VisionBuf * buf, VisionIpcBufExtra * extra, bool sync=true);
|
|
void start_listener();
|
|
};
|
|
|