op_kernel.h

 
#ifndef MPIX_OP_KERNEL_H
#define MPIX_OP_KERNEL_H
 
#include <stdint.h>
#include <stdlib.h>
 
#include <mpix/op.h>
 
enum mpix_kernel_type {
    MPIX_KERNEL_IDENTITY,
    MPIX_KERNEL_EDGE_DETECT,
    MPIX_KERNEL_GAUSSIAN_BLUR,
    MPIX_KERNEL_SHARPEN,
    MPIX_KERNEL_DENOISE,
};
 
struct mpix_kernel_op {
    struct mpix_base_op base;
    enum mpix_kernel_type type;
    void (*kernel_fn)(const uint8_t *src[], uint8_t *dst, uint16_t w);
};
 
#define MPIX_REGISTER_KERNEL_5X5_OP(id, fn, t, fmt)                                                \
    const struct mpix_kernel_op mpix_kernel_5x5_op_##id = {                                    \
        .base.name = ("kernel_5x5_" #id),                                                  \
        .base.format_src = (MPIX_FMT_##fmt),                                               \
        .base.format_dst = (MPIX_FMT_##fmt),                                               \
        .base.window_size = 5,                                                             \
        .base.run = mpix_kernel_5x5_op,                                                    \
        .kernel_fn = (fn),                                                                 \
        .type = (MPIX_KERNEL_##t),                                                         \
    }
 
#define MPIX_REGISTER_KERNEL_3X3_OP(id, fn, t, fmt)                                                \
    const struct mpix_kernel_op mpix_kernel_3x3_op_##id = {                                    \
        .base.name = ("kernel_3x3_" #id),                                                  \
        .base.format_src = (MPIX_FMT_##fmt),                                                  \
        .base.format_dst = (MPIX_FMT_##fmt),                                                  \
        .base.window_size = 3,                                                             \
        .base.run = mpix_kernel_3x3_op,                                                    \
        .kernel_fn = (fn),                                                                 \
        .type = (MPIX_KERNEL_##t),                                                         \
    }
 
void mpix_identity_rgb24_3x3(const uint8_t *in[3], uint8_t *out, uint16_t width);
void mpix_identity_rgb24_5x5(const uint8_t *in[5], uint8_t *out, uint16_t width);
void mpix_sharpen_rgb24_3x3(const uint8_t *in[3], uint8_t *out, uint16_t width);
void mpix_sharpen_rgb24_5x5(const uint8_t *in[5], uint8_t *out, uint16_t width);
void mpix_edgedetect_rgb24_3x3(const uint8_t *in[3], uint8_t *out, uint16_t width);
void mpix_gaussianblur_rgb24_3x3(const uint8_t *in[3], uint8_t *out, uint16_t width);
void mpix_median_rgb24_3x3(const uint8_t *in[3], uint8_t *out, uint16_t width);
void mpix_median_rgb24_5x5(const uint8_t *in[5], uint8_t *out, uint16_t width);
 
void mpix_kernel_5x5_op(struct mpix_base_op *op);
 
void mpix_kernel_3x3_op(struct mpix_base_op *op);
 
#endif