samples/risp4ml/isp_stages/wbg.h

#ifndef SAMPLES_RISP4ML_ISP_STAGES_WBG_H_
#define SAMPLES_RISP4ML_ISP_STAGES_WBG_H_

#include "samples/risp4ml/common/image.h"

#ifdef __cplusplus
extern "C" {
#endif  // __cplusplus

typedef struct {
  bool enable;
  bool fixed;
  uint32_t gains[kNumBayerPatterns];
} WbgParams;

void set_wbg_params(WbgParams* params);

void wbg_process(Image* input, Image* output);

#ifdef __cplusplus
}  // extern "C"
#endif  // __cplusplus

#endif  // SAMPLES_RISP4ML_ISP_STAGES_WBG_H_