]> mj.ucw.cz Git - libucw.git/blob - images/sig-txt.c
- use shape features in new comparision algorithm
[libucw.git] / images / sig-txt.c
1 /*
2  *      Image Library -- Detection of textured images
3  *
4  *      (c) 2006 Pavel Charvat <pchar@ucw.cz>
5  *
6  *      This software may be freely distributed and used according to the terms
7  *      of the GNU Lesser General Public License.
8  */
9
10 #define LOCAL_DEBUG
11
12 #include "sherlock/sherlock.h"
13 #include "images/images.h"
14 #include "images/signature.h"
15 #include "images/math.h"
16
17 #include <string.h>
18
19 #define MAX_CELLS_COLS 4
20 #define MAX_CELLS_ROWS 4
21
22 void
23 image_sig_detect_textured(struct image_sig_data *data)
24 {
25   uns cols = data->cols;
26   uns rows = data->rows;
27   uns cell_cols = MIN((cols + 1) / 2, MAX_CELLS_COLS);
28   uns cell_rows = MIN((rows + 1) / 2, MAX_CELLS_ROWS);
29   uns cell_x[MAX_CELLS_COLS + 1];
30   uns cell_y[MAX_CELLS_ROWS + 1];
31   uns i, j;
32   u32 cnt[IMAGE_REG_MAX];
33
34   DBG("Detecting textured image... cols=%u rows=%u cell_cols=%u cell_rows=%u", cols, rows, cell_cols, cell_rows);
35   
36   /* Compute cells boundaries */
37   for (i = 1, j = 0; i < cell_cols; i++)
38     cell_x[i] = fast_div_u32_u8(j += cols, cell_cols);
39   cell_x[0] = 0;
40   cell_x[cell_cols] = cols;
41   for (i = 1, j = 0; i < cell_rows; i++)
42     cell_y[i] = fast_div_u32_u8(j += rows, cell_rows);
43   cell_y[0] = 0;
44   cell_y[cell_rows] = rows;
45
46   /* Preprocess blocks */
47   for (uns i = 0; i < data->regions_count; i++)
48     for (struct image_sig_block *block = data->regions[i].blocks; block; block = block->next)
49       block->region = i;
50   
51   /* Process cells */
52   double e = 0;
53   for (uns j = 0; j < cell_rows; j++)
54     for (uns i = 0; i < cell_cols; i++)
55       {
56         uns cell_area = 0;
57         bzero(cnt, data->regions_count * sizeof(u32));
58         struct image_sig_block *b1 = data->blocks + cell_x[i] + cell_y[j] * cols, *b2;
59         for (uns y = cell_y[j]; y < cell_y[j + 1]; y++, b1 += cols)
60           {
61             b2 = b1;
62             for (uns x = cell_x[i]; x < cell_x[i + 1]; x++, b2++)
63               {
64                 cnt[b2->region]++;
65                 cell_area++;
66               }
67           }
68         for (uns k = 0; k < data->regions_count; k++)
69           {
70             int a = data->blocks_count * cnt[k] - cell_area * data->regions[k].count; 
71             e += (double)a * a / ((double)isqr(data->regions[k].count) * cell_area);
72           }
73       }
74
75   DBG("Coefficient=%g", (double)e / (data->regions_count * data->blocks_count));
76
77   /* Threshold */
78   if (e <= image_sig_textured_threshold * data->regions_count * data->blocks_count)
79     {
80       data->flags |= IMAGE_SIG_TEXTURED;
81       DBG("Image is textured.");
82     }
83   else
84     DBG("Image is not textured.");
85 }