]> mj.ucw.cz Git - libucw.git/blob - images/signature.h
experiment - store relative centroid position for each image region
[libucw.git] / images / signature.h
1 #ifndef _IMAGES_SIGNATURE_H
2 #define _IMAGES_SIGNATURE_H
3
4 /* Configuration */
5 extern uns image_sig_min_width, image_sig_min_height;
6 extern uns *image_sig_prequant_thresholds;
7 extern uns image_sig_postquant_min_steps, image_sig_postquant_max_steps, image_sig_postquant_threshold;
8 extern double image_sig_border_size;
9 extern int image_sig_border_bonus;
10 extern double image_sig_inertia_scale[];
11 extern double image_sig_textured_threshold;
12 extern int image_sig_compare_method;
13 extern uns image_sig_cmp_features_weights[];
14
15 #define IMAGE_VEC_F     6
16 #define IMAGE_REG_F     IMAGE_VEC_F
17 #define IMAGE_REG_H     5
18 #define IMAGE_REG_MAX   16
19
20 /* K-dimensional feature vector (6 bytes) */
21 struct image_vector {
22   byte f[IMAGE_VEC_F];          /* texture features */
23 } PACKED;
24
25 /* Fetures for image regions (16 bytes) */
26 struct image_region {
27   byte f[IMAGE_VEC_F];          /* texture features - L, u, v, LH, HL, HH */
28   byte h[IMAGE_REG_H];          /* shape/pos features - I1, I2, I3, X, Y */
29   byte wa;                      /* normalized area percentage */
30   byte wb;                      /* normalized weight */
31   u16 reserved;
32 } PACKED;
33
34 #define IMAGE_SIG_TEXTURED      0x1
35
36 /* Image signature (11 + len * 16 bytes) */
37 struct image_signature {
38   byte len;                     /* Number of regions */
39   byte flags;                   /* IMAGE_SIG_xxx */
40   u16 df;                       /* Average weighted f dist */
41   u16 dh;                       /* Average weighted h dist */
42   struct image_vector vec;      /* Average features of all regions... simple signature */
43   struct image_region reg[IMAGE_REG_MAX];/* Feature vector for every region */
44 } PACKED;
45
46 struct image_cluster {
47   union {
48     struct {
49       s32 dot;                  /* Dot product of the splitting plane */
50       s8 vec[IMAGE_VEC_F];      /* Normal vector of the splitting plane */
51     } PACKED;
52     struct {
53       u64 pos;                  /* Cluster size in bytes */
54     } PACKED;
55   } PACKED;
56 } PACKED;
57
58 static inline uns
59 image_signature_size(uns len)
60 {
61   return OFFSETOF(struct image_signature, reg) + len * sizeof(struct image_region);
62 }
63
64 /* sig-dump.c */
65
66 #define IMAGE_VECTOR_DUMP_MAX (IMAGE_VEC_F * 16 + 1)
67 #define IMAGE_REGION_DUMP_MAX ((IMAGE_REG_F + IMAGE_REG_H) * 16 + 100)
68
69 byte *image_vector_dump(byte *buf, struct image_vector *vec);
70 byte *image_region_dump(byte *buf, struct image_region *reg);
71
72 struct image_sig_block {
73   struct image_sig_block *next;         /* linked list */
74   u32 x, y;                             /* block position */
75   byte area;                            /* block area in pixels (usually 16) */
76   byte region;                          /* region index */
77   byte v[IMAGE_VEC_F];                  /* feature vector */
78 };
79
80 struct image_sig_region {
81   struct image_sig_block *blocks;
82   u32 count;
83   u32 a[IMAGE_VEC_F];
84   u32 b[IMAGE_VEC_F];
85   u32 c[IMAGE_VEC_F];
86   u64 e;
87   u64 w_sum;
88 };
89
90 struct image_sig_data {
91   struct image *image;
92   struct image_sig_block *blocks;
93   struct image_sig_region regions[IMAGE_REG_MAX];
94   u32 cols;
95   u32 rows;
96   u32 full_cols;
97   u32 full_rows;
98   u32 flags;
99   u32 area;
100   u32 valid;
101   u32 blocks_count;
102   u32 regions_count;
103   u32 f[IMAGE_VEC_F];
104 };
105
106 /* sig-init.c */
107
108 int compute_image_signature(struct image_thread *thread, struct image_signature *sig, struct image *image);
109
110 int image_sig_init(struct image_thread *thread, struct image_sig_data *data, struct image *image);
111 void image_sig_preprocess(struct image_sig_data *data);
112 void image_sig_finish(struct image_sig_data *data, struct image_signature *sig);
113 void image_sig_cleanup(struct image_sig_data *data);
114
115 /* sig-seg.c */
116
117 void image_sig_segmentation(struct image_sig_data *data);
118
119 /* sig-txt.c */
120
121 void image_sig_detect_textured(struct image_sig_data *data);
122
123 /* sig-cmp.c */
124
125 uns image_signatures_dist(struct image_signature *sig1, struct image_signature *sig2);
126 uns image_signatures_dist_explain(struct image_signature *sig1, struct image_signature *sig2, void (*msg)(byte *text, void *param), void *param);
127
128 #endif
129