mirror of
https://github.com/sunnypilot/sunnypilot.git
synced 2026-02-18 21:14:01 +08:00
7352e612a2
* merge * why * self.buf_size * 0.05 more than with tg due to copy --------- Co-authored-by: Comma Device <device@comma.ai>
70 lines
3.2 KiB
C++
70 lines
3.2 KiB
C++
#include "selfdrive/modeld/models/commonmodel.h"
|
|
|
|
#include <cmath>
|
|
#include <cstring>
|
|
|
|
#include "common/clutil.h"
|
|
|
|
DrivingModelFrame::DrivingModelFrame(cl_device_id device_id, cl_context context) : ModelFrame(device_id, context) {
|
|
input_frames = std::make_unique<uint8_t[]>(buf_size);
|
|
//input_frames_cl = CL_CHECK_ERR(clCreateBuffer(context, CL_MEM_READ_WRITE, buf_size, NULL, &err));
|
|
img_buffer_20hz_cl = CL_CHECK_ERR(clCreateBuffer(context, CL_MEM_READ_WRITE, 5*frame_size_bytes, NULL, &err));
|
|
region.origin = 4 * frame_size_bytes;
|
|
region.size = frame_size_bytes;
|
|
last_img_cl = CL_CHECK_ERR(clCreateSubBuffer(img_buffer_20hz_cl, CL_MEM_READ_WRITE, CL_BUFFER_CREATE_TYPE_REGION, ®ion, &err));
|
|
|
|
loadyuv_init(&loadyuv, context, device_id, MODEL_WIDTH, MODEL_HEIGHT);
|
|
init_transform(device_id, context, MODEL_WIDTH, MODEL_HEIGHT);
|
|
}
|
|
|
|
uint8_t* DrivingModelFrame::prepare(cl_mem yuv_cl, int frame_width, int frame_height, int frame_stride, int frame_uv_offset, const mat3& projection, cl_mem* output) {
|
|
run_transform(yuv_cl, MODEL_WIDTH, MODEL_HEIGHT, frame_width, frame_height, frame_stride, frame_uv_offset, projection);
|
|
|
|
for (int i = 0; i < 4; i++) {
|
|
CL_CHECK(clEnqueueCopyBuffer(q, img_buffer_20hz_cl, img_buffer_20hz_cl, (i+1)*frame_size_bytes, i*frame_size_bytes, frame_size_bytes, 0, nullptr, nullptr));
|
|
}
|
|
loadyuv_queue(&loadyuv, q, y_cl, u_cl, v_cl, last_img_cl);
|
|
|
|
if (output == NULL) {
|
|
CL_CHECK(clEnqueueReadBuffer(q, img_buffer_20hz_cl, CL_TRUE, 0, frame_size_bytes, &input_frames[0], 0, nullptr, nullptr));
|
|
CL_CHECK(clEnqueueReadBuffer(q, last_img_cl, CL_TRUE, 0, frame_size_bytes, &input_frames[MODEL_FRAME_SIZE], 0, nullptr, nullptr));
|
|
clFinish(q);
|
|
return &input_frames[0];
|
|
} else {
|
|
copy_queue(&loadyuv, q, img_buffer_20hz_cl, *output, 0, 0, frame_size_bytes);
|
|
copy_queue(&loadyuv, q, last_img_cl, *output, 0, frame_size_bytes, frame_size_bytes);
|
|
|
|
// NOTE: Since thneed is using a different command queue, this clFinish is needed to ensure the image is ready.
|
|
clFinish(q);
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
DrivingModelFrame::~DrivingModelFrame() {
|
|
deinit_transform();
|
|
loadyuv_destroy(&loadyuv);
|
|
CL_CHECK(clReleaseMemObject(img_buffer_20hz_cl));
|
|
CL_CHECK(clReleaseMemObject(last_img_cl));
|
|
CL_CHECK(clReleaseCommandQueue(q));
|
|
}
|
|
|
|
|
|
MonitoringModelFrame::MonitoringModelFrame(cl_device_id device_id, cl_context context) : ModelFrame(device_id, context) {
|
|
input_frames = std::make_unique<uint8_t[]>(buf_size);
|
|
//input_frame_cl = CL_CHECK_ERR(clCreateBuffer(context, CL_MEM_READ_WRITE, buf_size, NULL, &err));
|
|
|
|
init_transform(device_id, context, MODEL_WIDTH, MODEL_HEIGHT);
|
|
}
|
|
uint8_t* MonitoringModelFrame::prepare(cl_mem yuv_cl, int frame_width, int frame_height, int frame_stride, int frame_uv_offset, const mat3& projection, cl_mem* output) {
|
|
run_transform(yuv_cl, MODEL_WIDTH, MODEL_HEIGHT, frame_width, frame_height, frame_stride, frame_uv_offset, projection);
|
|
CL_CHECK(clEnqueueReadBuffer(q, y_cl, CL_TRUE, 0, MODEL_FRAME_SIZE * sizeof(uint8_t), input_frames.get(), 0, nullptr, nullptr));
|
|
clFinish(q);
|
|
//return &y_cl;
|
|
return input_frames.get();
|
|
}
|
|
|
|
MonitoringModelFrame::~MonitoringModelFrame() {
|
|
deinit_transform();
|
|
CL_CHECK(clReleaseCommandQueue(q));
|
|
}
|