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.
37 lines
829 B
37 lines
829 B
#pragma once
|
|
|
|
#include <cstddef>
|
|
#include <cstdint>
|
|
#include <vector>
|
|
|
|
#include "common/clutil.h"
|
|
|
|
#define NUM_SEGMENTS_X 8
|
|
#define NUM_SEGMENTS_Y 6
|
|
|
|
#define ROI_X_MIN 1
|
|
#define ROI_X_MAX 6
|
|
#define ROI_Y_MIN 2
|
|
#define ROI_Y_MAX 3
|
|
|
|
#define LM_THRESH 120
|
|
#define LM_PREC_THRESH 0.9 // 90 perc is blur
|
|
#define CONV_LOCAL_WORKSIZE 16
|
|
|
|
class LapConv {
|
|
public:
|
|
LapConv(cl_device_id device_id, cl_context ctx, int rgb_width, int rgb_height, int rgb_stride, int filter_size);
|
|
~LapConv();
|
|
uint16_t Update(cl_command_queue q, const uint8_t *rgb_buf, const int roi_id);
|
|
|
|
private:
|
|
cl_mem roi_cl, result_cl, filter_cl;
|
|
cl_program prg;
|
|
cl_kernel krnl;
|
|
const int width, height;
|
|
const int rgb_stride;
|
|
std::vector<uint8_t> roi_buf;
|
|
std::vector<int16_t> result_buf;
|
|
};
|
|
|
|
bool is_blur(const uint16_t *lapmap, const size_t size);
|
|
|