lib/gocr/output.c

   1 /*
   2 This is a Optical-Character-Recognition program
   3 Copyright (C) 2000-2006  Joerg Schulenburg
   4
   5 This program is free software; you can redistribute it and/or
   6 modify it under the terms of the GNU General Public License
   7 as published by the Free Software Foundation; either version 2
   8 of the License, or (at your option) any later version.
   9
  10 This program is distributed in the hope that it will be useful,
  11 but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 GNU General Public License for more details.
  14
  15 You should have received a copy of the GNU General Public License
  16 along with this program; if not, write to the Free Software
  17 Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  18
  19  see README for EMAIL address
  20 */
  21
  22 #include <string.h>
  23 #include "unicode.h"
  24 #include "output.h"
  25 #include "gocr.h"  /* extern job_t JOB; */
  26
  27 /* function is only for debugging and for developing
  28    it prints out a part of pixmap b at point x0,y0 to stderr
  29    using dots .,; if no pixel, and @xoO for pixels
  30    modify n_run and print out what would happen on 2nd, 3th loop!
  31    new: output original and copied pixmap in the same figure
  32  */
  33 void out_b(struct box *px, pix *b, int x0, int y0, int dx, int dy, int cs ){
  34   int x,y,x2,y2,yy0,tx,ty,n1,i;
  35   char c1, c2;
  36   yy0=y0;
  37   if(px){ /* overwrite rest of arguments */
  38     if (!b) {
  39       b=px->p;
  40       x0=px->x0; dx=px->x1-px->x0+1;
  41       y0=px->y0; dy=px->y1-px->y0+1; yy0=y0;
  42     }
  43     if(cs==0) cs=JOB->cfg.cs;
  44     fprintf(stderr,"\n# list box      x= %4d %4d d= %3d %3d r= %3d %3d"
  45                     " nrun=%d p=%p", /* ToDo: r,nrun is obsolete */
  46           px->x0, px->y0, px->x1 - px->x0 + 1, px->y1 - px->y0 + 1,
  47           px->x - px->x0, px->y - px->y0, JOB->tmp.n_run, (void*)px);
  48     fprintf(stderr,"\n#  dots=%d boxes=%d subboxes=%d c=%s mod=%s"
  49             " line=%d m= %d %d %d %d",
  50           px->dots, px->num_boxes, px->num_subboxes,
  51           decode(px->c,ASCII), decode(px->modifier,ASCII), px->line,
  52           px->m1 - px->y0, px->m2 - px->y0, px->m3 - px->y0, px->m4 - px->y0);
  53     if (px->num_frames) {
  54       int i,j,jo;
  55       fprintf(stderr,"\n#  frames= %d (sumvects=%d)",px->num_frames,
  56          ((px->num_frames)?px->num_frame_vectors[px->num_frames-1]:-1));
  57       for (jo=j=i=0; i<px->num_frames; i++, jo=j) {
  58         fprintf(stderr,"\n#  frame %d (%+4d,%3d,%2d) ",
  59                 i, px->frame_vol[i], px->frame_per[i],
  60                    px->num_frame_vectors[i]-jo);
  61         /* print only the first vectors of each frame */
  62         for (;j<px->num_frame_vectors[i] && j<MaxFrameVectors; j++)
  63           fprintf(stderr," #%02d %2d %2d", j,
  64                             px->frame_vector[j][0] - px->x0,
  65                             px->frame_vector[j][1] - px->y0);
  66       }
  67     }
  68     if (px->num_ac){ /* output table of chars and its probabilities */
  69       fprintf(stderr,"\n# list box char: ");
  70       for(i=0;i<px->num_ac && i<NumAlt;i++)
  71       /* output the (xml-)string (picture position, barcodes, glyphs, ...) */
  72         if (px->tas[i])
  73          fprintf(stderr," %s(%d)",       px->tas[i]       ,px->wac[i]);
  74         else
  75          fprintf(stderr," %s(%d)",decode(px->tac[i],ASCII),px->wac[i]);
  76     }
  77     fprintf(stderr,"\n");
  78     if (px->dots && px->m2 && px->m1<y0) { yy0=px->m1; dy=px->y1-yy0+1; }
  79   }
  80   tx=dx/80+1;
  81   ty=dy/40+1; /* step, usually 1, but greater on large maps */
  82   fprintf(stderr,"# list pattern  x= %4d %4d d= %3d %3d t= %d %d\n",
  83                  x0,y0,dx,dy,tx,ty);
  84   if (dx>0)
  85   for(y=yy0;y<yy0+dy;y+=ty) { /* reduce the output to max 78x40 */
  86     /* first image is the copied and modified bitmap of the box */
  87     if (px)
  88     for(x=x0;x<x0+dx;x+=tx){  /* by merging sub-pixels */
  89       n1=0; c1='.';
  90       for(y2=y;y2<y+ty && y2<y0+dy;y2++) /* sub-pixels */
  91       for(x2=x;x2<x+tx && x2<x0+dx;x2++)
  92       {
  93         if((getpixel(px->p,x2-x0+px->x0,
  94                         y2-y0+px->y0)<cs)) c1='@';
  95       }
  96       if (px->num_frames) { /* mark vectors */
  97         int i;
  98         if (c1!='$' && c1!='S') /* dont mark twice */
  99         for (i=0;i<px->num_frame_vectors[px->num_frames-1];i++)
 100           if ((px->frame_vector[i][0]-px->x0)/tx==(x-x0)/tx
 101            && (px->frame_vector[i][1]-px->y0)/ty==(y-y0)/ty)
 102               { c1=((c1=='@')?'$':'S'); break; }
 103       }
 104       fprintf(stderr,"%c", c1 );
 105     }
 106
 107     /* 2nd image is the boxframe in the original bitmap */
 108     if (dx<40) fprintf(stderr,"  ");
 109     if (dx<40) /* do it only, if we have enough place */
 110     for(x=x0;x<x0+dx;x+=tx){  /* by merging sub-pixels */
 111       c1='.';
 112       for(y2=y;y2<y+ty && y2<y0+dy;y2++) /* sub-pixels */
 113       for(x2=x;x2<x+tx && x2<x0+dx;x2++)
 114         { if((getpixel(b,x2,y2)<cs)) c1='@'; }
 115       fprintf(stderr,"%c", c1 );
 116     }
 117
 118     c1=c2=' ';
 119     /* mark lines with < */
 120     if (px) if (y-y0+px->y0==px->m1 || y-y0+px->y0==px->m2
 121              || y-y0+px->y0==px->m3 || y-y0+px->y0==px->m4)  c1='<';
 122     if (y==y0 || y==yy0+dy-1)  c2='-';  /* boxmarks */
 123
 124     fprintf(stderr,"%c%c\n",c1,c2);
 125   }
 126 }
 127
 128 /* same as out_b, but for faster use, only a box as argument
 129  */
 130 void out_x(struct box *px) {
 131   out_b(px,NULL,0, 0, 0, 0, JOB->cfg.cs);
 132 }
 133
 134
 135 /* print out two boxes side by side, for debugging comparision algos */
 136 void out_x2(struct box *box1, struct box *box2){
 137   int x,y,i,tx,ty,dy;
 138   /*FIXME jb static*/static char *c1="OXXXXxx@.,,,,,,,";
 139   pix *b=&JOB->src.p;
 140   dy=(box1->y1-box1->y0+1);
 141   if(dy<box2->y1-box2->y0+1)dy=box2->y1-box2->y0+1;
 142   tx=(box1->x1-box1->x0)/40+1;
 143   ty=(box1->y1-box1->y0)/40+1; /* step, usually 1, but greater on large maps */
 144   if(box2)fprintf(stderr,"\n# list 2 patterns");
 145   for(i=0;i<dy;i+=ty) { /* reduce the output to max 78x40??? */
 146     fprintf(stderr,"\n"); y=box1->y0+i;
 147     for(x=box1->x0;x<=box1->x1;x+=tx)
 148     fprintf(stderr,"%c", c1[ ((getpixel(b,x,y)<JOB->cfg.cs)?0:8)+marked(b,x,y) ] );
 149     if(!box2) continue;
 150     fprintf(stderr,"  "); y=box2->y0+i;
 151     for(x=box2->x0;x<=box2->x1;x+=tx)
 152     fprintf(stderr,"%c", c1[ ((getpixel(b,x,y)<JOB->cfg.cs)?0:8)+marked(b,x,y) ] );
 153   }
 154 }
 155
 156
 157 /* ---- list output ---- for debugging ---
 158  * list all boxes where the results can be found within the c-option
 159  */
 160 int output_list(job_t *job) {
 161   int i = 0, j;
 162   struct box *box2;
 163   pix  *pp = &job->src.p;
 164   char *lc = job->cfg.lc;
 165
 166   fprintf(stderr,"\n# list shape for charlist %s",lc);
 167   for_each_data(&(JOB->res.boxlist)) {
 168     box2 = (struct box *) list_get_current(&(JOB->res.boxlist));
 169     for (j=0; j<box2->num_ac; j++)
 170       if (!lc || (box2->tac[j] && strchr(lc, box2->tac[j]))
 171               || (box2->tas[j] && strstr(lc, box2->tas[j]))) break;
 172     if (j<box2->num_ac)
 173       fprintf(stderr,"\n#            box found in charlist");
 174     if (!lc || (strchr(lc, box2->c) && box2->c < 256 && box2->c)
 175             || (strchr(lc, '_') && box2->c==UNKNOWN) /* for compability */
 176             || j<box2->num_ac ){  /* also list alternative chars */
 177       if (!pp) pp=box2->p;
 178       fprintf(stderr,
 179               "\n# list shape %3d x=%4d %4d d= %3d %3d vf=%d ac=%d %04x %s",
 180               i, box2->x0, box2->y0,
 181               box2->x1 - box2->x0 + 1,
 182               box2->y1 - box2->y0 + 1,
 183               box2->num_frames, box2->num_ac,
 184               (int)box2->c,   /* wchar_t -> char ???? */
 185               decode(box2->c,ASCII) );
 186       if (JOB->cfg.verbose & 4) out_x(box2);
 187     }
 188     i++;
 189   } end_for_each(&(JOB->res.boxlist));
 190   fprintf(stderr,"\n");
 191   return 0;
 192 }
 193