sksp2024-mcu/libucw/images/signature.h

#ifndef _IMAGES_SIGNATURE_H
#define _IMAGES_SIGNATURE_H

#ifdef CONFIG_UCW_CLEAN_ABI
#define compute_image_signature ucw_compute_image_signature
#define image_region_dump ucw_image_region_dump
#define image_sig_border_bonus ucw_image_sig_border_bonus
#define image_sig_border_size ucw_image_sig_border_size
#define image_sig_cleanup ucw_image_sig_cleanup
#define image_sig_cmp_features_weights ucw_image_sig_cmp_features_weights
#define image_sig_compare_method ucw_image_sig_compare_method
#define image_sig_detect_textured ucw_image_sig_detect_textured
#define image_sig_finish ucw_image_sig_finish
#define image_sig_inertia_scale ucw_image_sig_inertia_scale
#define image_sig_init ucw_image_sig_init
#define image_sig_min_height ucw_image_sig_min_height
#define image_sig_min_width ucw_image_sig_min_width
#define image_sig_postquant_max_steps ucw_image_sig_postquant_max_steps
#define image_sig_postquant_min_steps ucw_image_sig_postquant_min_steps
#define image_sig_postquant_threshold ucw_image_sig_postquant_threshold
#define image_sig_preprocess ucw_image_sig_preprocess
#define image_sig_prequant_thresholds ucw_image_sig_prequant_thresholds
#define image_sig_segmentation ucw_image_sig_segmentation
#define image_sig_textured_threshold ucw_image_sig_textured_threshold
#define image_signatures_dist ucw_image_signatures_dist
#define image_signatures_dist_explain ucw_image_signatures_dist_explain
#define image_vector_dump ucw_image_vector_dump
#endif

/* Configuration */
extern uint image_sig_min_width, image_sig_min_height;
extern uint *image_sig_prequant_thresholds;
extern uint image_sig_postquant_min_steps, image_sig_postquant_max_steps, image_sig_postquant_threshold;
extern double image_sig_border_size;
extern int image_sig_border_bonus;
extern double image_sig_inertia_scale[];
extern double image_sig_textured_threshold;
extern int image_sig_compare_method;
extern uint image_sig_cmp_features_weights[];

#define IMAGE_VEC_F	6
#define IMAGE_REG_F	IMAGE_VEC_F
#define IMAGE_REG_H	5
#define IMAGE_REG_MAX	16

/* K-dimensional feature vector (6 bytes) */
struct image_vector {
  byte f[IMAGE_VEC_F];		/* texture features */
} PACKED;

/* Features for image regions (16 bytes) */
struct image_region {
  byte f[IMAGE_VEC_F];		/* texture features - L, u, v, LH, HL, HH */
  byte h[IMAGE_REG_H];		/* shape/pos features - I1, I2, I3, X, Y */
  byte wa;			/* normalized area percentage */
  byte wb;			/* normalized weight */
};

#define IMAGE_SIG_TEXTURED	0x1

/* Image signature (usually 16 + len * 16 bytes) */
struct image_signature {
  byte len;			/* number of regions */
  byte flags;			/* IMAGE_SIG_xxx */
  u16 cols;			/* image width */
  u16 rows;			/* image height */
  u16 df;			/* average weighted f dist */
  u16 dh;			/* average weighted h dist */
  struct image_vector vec;	/* average features of all regions... simple signature */
  struct image_region reg[IMAGE_REG_MAX];/* feature vector for every region */
};

struct image_cluster {
  union {
    struct {
      s32 dot;			/* dot product of the splitting plane */
      s8 vec[IMAGE_VEC_F];	/* normal vector of the splitting plane */
    };
    struct {
      u64 pos;			/* cluster size in bytes */
    };
  };
};

static inline uint image_signature_size(uint len)
{
  return OFFSETOF(struct image_signature, reg) + len * sizeof(struct image_region);
}

/* sig-dump.c */

#define IMAGE_VECTOR_DUMP_MAX (IMAGE_VEC_F * 16 + 1)
#define IMAGE_REGION_DUMP_MAX ((IMAGE_REG_F + IMAGE_REG_H) * 16 + 100)

byte *image_vector_dump(byte *buf, struct image_vector *vec);
byte *image_region_dump(byte *buf, struct image_region *reg);

struct image_sig_block {
  struct image_sig_block *next;		/* linked list */
  u32 x, y;				/* block position */
  byte area;				/* block area in pixels (usually 16) */
  byte region;				/* region index */
  byte v[IMAGE_VEC_F];			/* feature vector */
};

struct image_sig_region {
  struct image_sig_block *blocks;
  u32 count;
  u32 a[IMAGE_VEC_F];
  u32 b[IMAGE_VEC_F];
  u32 c[IMAGE_VEC_F];
  u64 e;
  u64 w_sum;
};

struct image_sig_data {
  struct image *image;
  struct image_sig_block *blocks;
  struct image_sig_region regions[IMAGE_REG_MAX];
  u32 cols;
  u32 rows;
  u32 full_cols;
  u32 full_rows;
  u32 flags;
  u32 area;
  u32 valid;
  u32 blocks_count;
  u32 regions_count;
  u32 f[IMAGE_VEC_F];
};

/* sig-init.c */

int compute_image_signature(struct image_context *ctx, struct image_signature *sig, struct image *image);

int image_sig_init(struct image_context *ctx, struct image_sig_data *data, struct image *image);
void image_sig_preprocess(struct image_sig_data *data);
void image_sig_finish(struct image_sig_data *data, struct image_signature *sig);
void image_sig_cleanup(struct image_sig_data *data);

/* sig-seg.c */

void image_sig_segmentation(struct image_sig_data *data);

/* sig-txt.c */

void image_sig_detect_textured(struct image_sig_data *data);

/* sig-cmp.c */

uint image_signatures_dist(struct image_signature *sig1, struct image_signature *sig2);
uint image_signatures_dist_explain(struct image_signature *sig1, struct image_signature *sig2, void (*msg)(byte *text, void *param), void *param);

#endif