-
-
Notifications
You must be signed in to change notification settings - Fork 18
/
unoptar.c
1696 lines (1479 loc) · 43.2 KB
/
unoptar.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
993
994
995
996
997
998
999
1000
/* (c) GPL 2007 Karel 'Clock' Kulhavy, Twibright Labs */
#include <stdio.h> /* printf */
#include <stdlib.h> /* malloc */
#include <math.h> /* floor */
#include <string.h> /* memcpy */
#include <assert.h> /* assert */
#include <png.h> /* libpng. Sometimes is in /usr/local/include/libpng/png.h, but that is taken
care of using -I/usr/local/include/libpng in the Makefile. */
#include "optar.h"
#include "parity.h"
/* Crosses will be resynced with precision of FINESTEP pixels */
#define FINE_CROSS_RESYNC
#define FINESTEP 0.25
/* Define to disable repairing bit by Hamming codes */
#define MIN(x,y) ((x)<(y)?(x):(y))
/* Takes only unsigned integers, returns real value, if out of range returns
* white, doens't threshold*/
#define getpixu(x,y) \
((x)>=width||(y)>=height?0xff:ary[(x)+(y)*width])
/* Integers in corners */
#define writepix(x,y,c) writepixu((unsigned)floor(x),(unsigned)floor(y),c)
/* If out of range, doesn't write anythinig. Integers are in pixel upper
* left corners. */
#define writepixu(x,y,c) {if ((x)<width&&(y)<height) \
newary[(x)+(y)*width]=c;}
struct que{
unsigned x;
unsigned y;
};
static unsigned width, height; /* In pixels, not it symbols! The whole image including
border, white surrounding etc. */
static unsigned char *ary; /* Allocated to width*height */
static unsigned char *newary; /* Allocated to width*height */
static unsigned long histogram[256];
static unsigned char global_cutlevel;
static unsigned char fill_global_cutlevel; /* This is always set to 50% between
black and white to make sure the
border is not accidentally broken
through as happened when
global_cutlevel was used for
filling. */
static unsigned char average; /* Average pixel value */
static unsigned long corners[4][2]; /* UL, UR, LL, LR / x, y. Integers in pixel upper
left corners. */
static unsigned long leftedge, rightedge, topedge, bottomedge; /* Coordinates,
minima/maxima of the corner coordinates. */
static double crosses[XCROSSES][YCROSSES][2]; /* [x][y][coord]. Integers in pixel
upper left corners. */
static float cutlevels[XCROSSES][YCROSSES]; /* Each cross has it's own cutlevel
based on how it came out printed. */
static int chalf_fine; /* Larger chalf for fine search */
static int chalf; /* In the input image, measured in input image pixels!
Important difference - in the decoding, the crosses are
assumed twice as small!
Calculated by find_corners. */
static float *search_area; /* Allocated as soon as chalf is known. Width 4*chalf+1,
height 4*chalf+1. The additional "+1" is for a row
(topmost) and column (leftmost) of zeroes which are
a result of integration.
Before integration, position x,y (1...4*chalf) says
pixel value, where the middle of the cross is
just at the boundary between pixels [2*chalf] and
[2*chalf+1].
After integration, each pixel says integral including
that pixel. */
static unsigned long bad_01, bad_10; /* Flipped from 0 to 1 (black dirt),
flipped from 1 to 0 (white dirt) */
static unsigned long bad_total;
static unsigned long irreparable;
/* These macros shift coordinates by given amount of input pixels parallel
* with recording axes. */
#define PSHIFTX(x, dx, dy) ((x)+(dx)*pixelhx+(dy)*pixelvx)
#define PSHIFTY(y, dx, dy) ((y)+(dx)*pixelhy+(dy)*pixelvy)
static double pixelhx, pixelhy, pixelvx, pixelvy; /* Horizontal (right) and vertical
(down) pixel vectors.They are initialized as soon as the corners are determined
and their length is equal to 1. Used to compensate out rotation. System:
+x right, +y down */
static double hpixel, vpixel; /* Pixel size calculated from the horizontal
and vertical corner distance */
static struct que *que;
static struct que *que_end; /* First invalid */
static struct que *rptr, *wptr;
static FILE *input_stream;
static double output_gamma=0.454545; /* What gamma the debug output has
(output number=number of photons ^ gamma) */
static long format_height; /* Used to be the HEIGHT macro. */
static unsigned text_height=24;
static unsigned long golay_stats[5]; /* 0, 1, 2, 3, 4 damaged bits */
/* -------------------- MAGIC CONSTANTS -------------------- */
static double unsharp_mask=7 /* 0.7 */;
static double unsharp_dist=1; /* 1 means that the neighbouring pixels will be
sampled. 0.5 that halfway to the neighbouring
pixels will be sampled etc. Only values up to
1.0 make sense. */
static float sync_white_cut=0.10; /* The cut used for positioning, not
quantization */
static float white_cut=0.06; /* 0 means cut in the black level, 1 cut in the
white level, 0.5 cut in the middle etc. */
static double minmax_filter=0.5; /* Dust/scratch removal filter. The input pixel size
is multiplied with this and rounded down. Then so
many cycles of max are performed and then the same
amount of cycles of min. */
static double pixel_blur=0.25; /*0.155 Approximate width of blur blot in terms of input
pixel. */
static double cross_trim=0.75; /* Such amount of input pixels (the big ones) will be
trimmed from the cross prior to performing the fine
search. */
/* -------------------- END OF MAGIC CONSTANTS -------------------- */
/* Allocates and fills in gamma table */
static unsigned char *make_gamma_table(float gamma)
{
unsigned char *t;
int i;
t=malloc(256);
if (!t){
fprintf(stderr,"Cannot allocated gamma table of 256 bytes!\n");
exit(1);
}
for (i=0;i<256;i++){
float r; /* Result */
/* White level is 255 not 256! */
r=255*pow((float)i/255, gamma);
t[i]=floor(r+0.5);
}
return t;
}
static void dump_newary(char *fname)
{
FILE *f;
unsigned char *gamma_table;
unsigned char *ptr;
gamma_table=make_gamma_table(output_gamma);
if (!gamma_table) return;
/* Translate from linear photometric back to gamma compressed. The
* output file will have the same gamma as the input one. */
for (ptr=newary;ptr<newary+width*height;ptr++)
*ptr=gamma_table[*ptr];
free(gamma_table);
f=fopen(fname,"w");
if (!f){
fprintf(stderr,"Warning: cannot open dumpfile %s: "
,fname);
return; /* No dump */
}
fprintf(f,"P5\n"
"%u %u\n"
"255\n"
,width, height);
fwrite(newary, width, height, f);
fclose(f);
}
/* Gamma corrects to photon counts and calculates histogram of the gamma
* corrected image. */
static void calc_histogram(void)
{
unsigned char *ptr;
unsigned char *end=ary+(unsigned long)width*height;
unsigned long long total=0;
memset(histogram, 0, sizeof(histogram));
for (ptr=ary;ptr<end;ptr++){
histogram[*ptr]++;
}
/* Calculate the sum of the histogram */
{
int i;
for (i=0; i<sizeof histogram/sizeof *histogram; i++)
total+=(unsigned long long)i*histogram[i];
}
average=(total+((width*height)>>1))/(width*height);
fprintf(stderr,"Average pixel value %u\n", average);
}
/* Analyzes, determines the cut level */
static void analyze_cutlevel(void)
{
float white, black;
unsigned long black_pixels, white_pixels;
int i;
float white_rms, black_rms; /* At the end they will be RMS of the
distance from average */
#define MAXITER 32
int iter; /* max. MAXITER iterations */
int lastcutlevel;
fill_global_cutlevel=global_cutlevel=average;
/* The second guess uses global_cutlevel from the first guess */
for (iter=0;iter<MAXITER;iter++){
lastcutlevel=global_cutlevel;
white_rms=black_rms=0;
black_pixels=white_pixels=0;
for (i=0;i<global_cutlevel;i++){
black_rms+=(float)histogram[i]
*(global_cutlevel-i)*(global_cutlevel-i);
black_pixels+=histogram[i];
}
for (i=global_cutlevel+1;i<sizeof histogram/sizeof *histogram;i++){
white_rms+=(float)histogram[i]
*(i-global_cutlevel)*(i-global_cutlevel);
white_pixels+=histogram[i];
}
/* Convert square sums to mean square */
white_rms/=white_pixels;
black_rms/=black_pixels;
/* Convert MS to RMS */
white_rms=sqrt(white_rms);
black_rms=sqrt(black_rms);
white=global_cutlevel+white_rms;
black=global_cutlevel-black_rms;
/* Cutlevel determined by sync_white_cut. sync_white_cut=0 means
* cut at black level, 0.5 cut in the middle, 1 cut at the
* white level */
global_cutlevel=floor(white*sync_white_cut
+black*(1-sync_white_cut)+0.5);
fill_global_cutlevel=floor(white*0.5+black*0.5+0.5);
fprintf(stderr,"Black %G, white %G, "
"cutlevel %u (0x%02x), "
"fill cutlevel %u (0x%02x)\n"
,black, white
,global_cutlevel, global_cutlevel
,fill_global_cutlevel, fill_global_cutlevel
);
if (global_cutlevel==lastcutlevel)
break; /* Stable state reached */
}
if (iter==MAXITER)
fprintf(stderr,"Warning: cutting point analysis didn't "
"converge in %u iterations.\n", MAXITER);
}
/* Returns -1 if fails. xin, yin are coordinates of the starting
* corner. dx is which way to go away from the corner horizontally and
* dy vertically. *outx, *outy will contain the address of the first black pixel
* or (-1, -1) if none is found. */
static void diag_scan(int *outx, int *outy, int xin, int yin, int
dx, unsigned dy)
{
int xbegin, x, y;
unsigned ctr, len;
for (xbegin=xin,len=1;xbegin<MIN(width,height);xbegin+=dx,len++){
x=xbegin;
y=yin;
for (ctr=len;ctr;ctr--,x-=dx,y+=dy){
if (getpixu(x,y)<global_cutlevel){
*outx=x;
*outy=y;
return;
}
}
}
*outx=-1;
*outy=-1;
}
/* Protection against buffer overrun. The coordinate is the coordinate of
* the mark center in a system where integers are in pixel upper left
* corners. Upper left pixel of the mark will be white. */
static void mark(double x, double y)
{
unsigned xu, yu;
xu=floor(x+0.5);
yu=floor(y+0.5);
writepixu((xu),(yu),0xff);
writepixu((xu-1),(yu-1),0xff);
writepixu((xu-1),yu,0);
writepixu((xu),(yu-1),0);
}
static void normalize_vector(double *x, double *y)
{
double vectorlen;
vectorlen=sqrt(*x**x+*y**y);
*x/=vectorlen;
*y/=vectorlen;
}
/* Returns angle, from 0 to 360, of a normalized vector. System: +y up. */
static double angle(double x, double y)
{
double deg;
if (fabs(x)>fabs(y)){
/* Horizontal */
deg=180/M_PI*asin(y);
if (x<0) deg=180-deg;
}else{
/* Vertical */
deg=180/M_PI*asin(x);
if (y<0) deg=deg-90;
else deg=90-deg;
}
return deg;
}
/* Puts it -180...+180 */
static double normalize_angle(double a)
{
return remainder(a,360);
}
static void find_corners(void)
{
int x,y;
diag_scan(&x, &y, 0, 0, 1, 1);
if (x<0){
static char failure []="failure_debug.pgm";
fprintf(stderr,"Error: cannot find upper left corner\n");
fail:
fprintf(stderr,"See failure_debug.pgm why.\n");
memcpy(newary, ary, (unsigned long)width*height);
dump_newary(failure);
exit(1);
}
corners[0][0]=x;
corners[0][1]=y;
diag_scan(&x, &y, width-1, 0, -1, 1);
if (x<0){
fprintf(stderr,"Error: cannot find upper right corner\n");
goto fail;
}
corners[1][0]=x+1;
corners[1][1]=y;
diag_scan(&x, &y, 0, height-1, 1, -1);
if (x<0){
fprintf(stderr,"Error: cannot find lower left corner\n");
goto fail;
}
corners[2][0]=x;
corners[2][1]=y+1;
diag_scan(&x, &y, width-1, height-1, -1, -1);
if (x<0){
fprintf(stderr,"Error: cannot find lower right corner\n");
goto fail;
}
corners[3][0]=x+1;
corners[3][1]=y+1;
leftedge=MIN(corners[0][0],corners[2][0]);
rightedge=MAX(corners[1][0],corners[3][0]);
topedge=MIN(corners[0][1],corners[1][1]);
bottomedge=MAX(corners[2][1],corners[3][1]);
hpixel=(corners[1][0]+corners[3][0]
-corners[0][0]-corners[0][0])/2.0/WIDTH;
vpixel=(corners[2][1]+corners[3][1]
-corners[0][1]-corners[1][1])/2.0/format_height;
fprintf(stderr,"One bit is %G horizontal pixels and %G "
"vertical pixels.\n", hpixel, vpixel);
{
unsigned vchalf, hchalf;
/* Take only half to prevent spurious resync to an edge of the
* cross when the data mimic the other half of the cross. */
hchalf=hpixel*CHALF*0.5;
vchalf=vpixel*CHALF*0.5;
chalf=MIN(hchalf, vchalf);
/* Round to zero to make sure we don't catch any chaff */
/* Trim the cross by some fraction of input pixel to remove
* the area affected by crosstalk. */
hchalf=hpixel*(CHALF-cross_trim);
vchalf=vpixel*(CHALF-cross_trim);
chalf_fine=MIN(hchalf, vchalf);
}
/* Calculate the pixel vectors */
pixelhx=((double)corners[1][0]+(double)corners[3][0]
-(double)corners[0][0]-(double)corners[2][0])/2;
pixelhy=((double)corners[1][1]+(double)corners[3][1]
-(double)corners[0][1]-(double)corners[2][1])/2;
pixelvx=((double)corners[2][0]+(double)corners[3][0]
-(double)corners[0][0]-(double)corners[1][0])/2;
pixelvy=((double)corners[2][1]+(double)corners[3][1]
-(double)corners[0][1]-(double)corners[1][1])/2;
/* Normalize the horizontal vector (which may not be exactly
* horizontal */
normalize_vector(&pixelhx, &pixelhy);
/* Normalize the vertical vector (which may not be exactly
* vertical */
normalize_vector(&pixelvx, &pixelvy);
fprintf(stderr,"Input horizontal pixel vector %G,%G, vertical %G,%G."
" skew %G deg, perpendicularity %G deg.\n",
pixelhx, pixelhy, pixelvx, pixelvy
,normalize_angle(angle(pixelhx, -pixelhy)
+angle(pixelvx, -pixelvy)+90)/2
,angle(pixelhx, -pixelhy)-angle(pixelvx, -pixelvy));
{
unsigned long bytes=(long)(4*chalf+1)*(4*chalf+1)
*sizeof*search_area;
search_area=malloc(bytes);
if (!search_area){
fprintf(stderr,
"Cannot allocate search area of %lu bytes\n",
bytes);
exit(1);
}
}
fprintf(stderr,"Allocating search area of %u x %u (%u) pixels.\n",
chalf<<1, chalf<<1, (chalf*chalf)<<2);
fprintf(stderr,"Upper corners at %lu, %lu and %lu, %lu,\n"
"lower corners at %lu, %lu and %lu, %lu.\n"
"Cross half for searching is %d x %d input pixels.\n",
corners[0][0], corners[0][1], corners[1][0], corners[1][1],
corners[2][0], corners[2][1], corners[3][0], corners[3][1],
chalf, chalf);
}
static double bilinear(double ul, double ur,
double ll, double lr,
double hpar, double vpar)
{
double u,l; /* Upper, lower */
u=ur*hpar+ul*(1-hpar);
l=lr*hpar+ll*(1-hpar);
return l*vpar+u*(1-vpar);
}
static float bilinearf(float ul, float ur,
float ll, float lr,
float hpar, float vpar)
{
float u,l; /* Upper, lower */
u=ur*hpar+ul*(1-hpar);
l=lr*hpar+ll*(1-hpar);
return l*vpar+u*(1-vpar);
}
/* x,y with integers in centers of pixels */
static float get_pixel_interp(double x, double y)
{
unsigned xi, yi; /* Integer versions, rounded down */
/* Supports even extrapolation, but should be never necessary */
if (x<0) xi=0; else xi=floor(x);
if (y<0) yi=0; else yi=floor(y);
/* Make it faster, only little precision needed. */
return bilinearf(getpixu(xi,yi), getpixu(xi+1,yi),
getpixu(xi,yi+1),getpixu(xi+1,yi+1),
x-xi, y-yi);
}
/* Samples pixels and performs correction(s) */
static float pixel_correct_sample(double x, double y)
{
float val;
float avg; /* First sum, later average */
double hdist, vdist;
hdist=hpixel*unsharp_dist;
vdist=vpixel*unsharp_dist;
avg=get_pixel_interp(
PSHIFTX(x, -hdist, 0),
PSHIFTY(y, -hdist, 0));
avg+=get_pixel_interp(
PSHIFTX(x, hdist, 0),
PSHIFTY(y, hdist, 0));
avg+=get_pixel_interp(
PSHIFTX(x, 0, vdist),
PSHIFTY(y, 0, vdist));
avg+=get_pixel_interp(
PSHIFTX(x, 0, -vdist),
PSHIFTY(y, 0, -vdist));
avg/=4;
val=get_pixel_interp(x,y);
val+=unsharp_mask*(val-avg); /* Emphasize the distance from average */
return val;
}
/* Returns difference from global_cutlevel. Integers in centers of pixels. Interpolates
* for nonintegral coordinates. */
static float diffpix(double x, double y)
{
return get_pixel_interp(x,y)-global_cutlevel;
}
/* Calculates a correlation with a cross. x and y are coords of the cross
* center with integers in UL corners of pixels. */
static float cross_correl(double x, double y)
{
double dx, dy;
float sum =0;
/* -0.5 for conversion corners -> centers, +0.5 for conversion
* cross center -> sample point 1/2 pixel away from the cross
* center -> No addition at all */
for (dx=0;dx<chalf_fine;dx++)
for (dy=0;dy<chalf_fine;dy++){
sum-=diffpix(x+dx, y+dy);
sum-=diffpix(x-1-dx, y-1-dy);
sum+=diffpix(x+dx,y-1-dy);
sum+=diffpix(x-1-dx,y+dy);
}
return sum;
}
/* Range from 0 to 4*chalf, inclusive */
static float getsearch(int xpos, int ypos)
{
assert (xpos>=0);
assert (xpos<=4*chalf);
assert (ypos>=0);
assert (ypos<=4*chalf);
return search_area[ypos*(4*chalf+1)+xpos];
}
/* xpos says the cross offset. 0,0 means at the original position from around
* which the search_area was loaded. The range is -chalf to chalf
* (inclusive). The input must be in that range, otherwise crash */
static float cross_correl_search(int xpos, int ypos)
{
float sum;
assert(xpos>=-chalf);
assert(xpos<=chalf);
assert(ypos>=-chalf);
assert(ypos<=chalf);
/* Normalize the xpos and ypos to mean the cross center in array
* indices. 0 means array left edge, 2*chalf array center, 4*chalf array
* right edge. */
xpos+=2*chalf;
ypos+=2*chalf;
/* Center */
sum=-4*getsearch(xpos,ypos);
/* Middles of sides */
sum+=2*(
getsearch(xpos-chalf, ypos)+
getsearch(xpos+chalf, ypos)+
getsearch(xpos, ypos-chalf)+
getsearch(xpos, ypos+chalf)
);
/* Corners */
sum-=(
getsearch(xpos-chalf, ypos-chalf)+
getsearch(xpos-chalf, ypos+chalf)+
getsearch(xpos+chalf, ypos-chalf)+
getsearch(xpos+chalf, ypos+chalf)
);
return sum;
}
/* After this, pixel [0][0] means integral from [0][0] to [1][1] (!), etc. */
static void integrate_search_area(void)
{
int x, y;
float *ptr;
/* Horizontal integration */
ptr=search_area;
for (y=0;y<=4*chalf;y++){
ptr++;
for (x=1;x<=4*chalf;x++){
ptr[0]+=ptr[-1];
ptr++;
}
}
ptr=search_area+4*chalf+1;
/* Vertical integration */
for (;ptr<search_area+(4*chalf+1)*(4*chalf+1);ptr++)
ptr[0]+=ptr[-4*chalf-1];
}
static void cross_stats(unsigned cx, unsigned cy)
{
double centerx=crosses[cx][cy][0];
double centery=crosses[cx][cy][1];
int hpixelhalf=floor(hpixel*(CHALF-cross_trim));
int vpixelhalf=floor(vpixel*(CHALF-cross_trim));
float black_rms=0, white_rms=0;
int xoff, yoff;
long val;
float white, black;
unsigned long whitepixels=0, blackpixels=0;
float cutlevel_result;
for (xoff=-hpixelhalf; xoff<=hpixelhalf; xoff++)
for (yoff=-vpixelhalf; yoff<=vpixelhalf; yoff++){
val=get_pixel_interp(
PSHIFTX(centerx, xoff, yoff)
,PSHIFTY(centery, xoff, yoff));
if (val>global_cutlevel){
/* White */
white_rms+=(val-global_cutlevel)*(val-global_cutlevel);
whitepixels++;
}else if (val<global_cutlevel){
/* Black */
black_rms+=(val-global_cutlevel)*(val-global_cutlevel);
blackpixels++;
}
}
if (!(whitepixels&&blackpixels)){
cutlevel_result=global_cutlevel;
/* Impossible to determine, use default global_cutlevel */
}else{
white_rms=sqrt(white_rms/whitepixels);
black_rms=sqrt(black_rms/blackpixels);
white=global_cutlevel+white_rms;
black=global_cutlevel-black_rms;
cutlevel_result=white*(white_cut)
+black*(1-white_cut);
}
cutlevels[cx][cy]=cutlevel_result;
fprintf(stderr,"%02x ",(int)floor(cutlevel_result+0.5));
}
/* Center of search area is in a system where the integers are in the
* corners. */
static void load_search_area(double centerx, double centery)
{
int xoff, yoff;
float *ptr=search_area;
for (yoff=-2*chalf-1; yoff<2*chalf; yoff++)
for (xoff=-2*chalf-1; xoff<2*chalf; xoff++){
if (yoff==-2*chalf-1||xoff==-2*chalf-1){
/* Zero row/column */
*ptr=0;
}else{
/* No translation necessary since centers
* in corners automatically translate to
* centers of adjacent pixels expressed
* in a system with integers in centers. */
*ptr=diffpix(PSHIFTX(centerx, xoff, yoff)
,PSHIFTY(centery, xoff, yoff));
}
ptr++;
}
}
/* The coords are with integers in corners */
static void resync_cross(double *coordpair)
{
int xoff, yoff; /* Symmetric step offsets for search stepping always
by 1 and once meaning 1, once meaning a subpixel
step. They perform steps parallel to the recording
axes. */
int xoffmax, yoffmax; /* Step during which the maximum was reached */
float max; /* How much was reached during maximum */
double xmax, ymax; /* Later it's calculated in which pixel position
the maximum was calculated, with subpixel
precision. Coords of cross center with integers
in corners. */
float result;
load_search_area(coordpair[0], coordpair[1]);
integrate_search_area(); /* Precalculates - dynamic programming */
/* Preload with a default */
xoffmax=0;
yoffmax=0;
max=cross_correl_search(xoffmax,yoffmax);
/* Rough search - 1 pixel step */
for (xoff=-chalf;xoff<=chalf;xoff++)
for (yoff=-chalf;yoff<=chalf;yoff++){
result=cross_correl_search(xoff, yoff);
if (result>max)
{
max=result;
xoffmax=xoff;
yoffmax=yoff;
}
}
xmax=PSHIFTX(coordpair[0], xoffmax, yoffmax);
ymax=PSHIFTY(coordpair[1], xoffmax, yoffmax);
#ifdef FINE_CROSS_RESYNC
#define HALFRANGE (0.5/FINESTEP) /* 0.5 means 0.5 of small input pixel - no
need to search more since then it would be
caught by the neighbouring coarse search
try. */
/* Load the fine search initial maximum position */
xoffmax=0;
yoffmax=0;
/* This must be here since cross_correl and cross_correl_search
* return the result scaled by a different factor. */
max=cross_correl(xmax,ymax);
/* Fine search, FINESTEP pixels/ step. Counting in 0.25 steps */
for (xoff=-HALFRANGE;xoff<=HALFRANGE;xoff++)
for (yoff=-HALFRANGE;yoff<=HALFRANGE;yoff++){
/* This is not using the search area anymore! */
result=cross_correl(
PSHIFTX(xmax, (double)xoff*FINESTEP
,(double)yoff*FINESTEP)
,PSHIFTY(ymax, (double)xoff*FINESTEP
,(double)yoff*FINESTEP));
if (result>max){
max=result;
xoffmax=xoff;
yoffmax=yoff;
}
}
/* Save the fine search result */
xmax=PSHIFTX(xmax, (double)xoffmax*FINESTEP, (double)yoffmax*FINESTEP);
ymax=PSHIFTY(ymax, (double)xoffmax*FINESTEP, (double)yoffmax*FINESTEP);
#endif /* FINE_CROSS_RESYNC */
/* Store the output */
coordpair[0]=xmax;
coordpair[1]=ymax;
}
static void sync_crosses(void)
{
unsigned cx,cy; /* Cross number */
double rightx, righty, downx, downy; /* Two cross pitch vectors */
/* Calculate the estimated cross pitch vectors */
rightx=((double)corners[1][0]+corners[3][0]-corners[0][0]-corners[2][0])
/2*CPITCH/WIDTH;
righty=((double)corners[1][1]+corners[3][1]-corners[0][1]-corners[2][1])
/2*CPITCH/WIDTH;
downx=((double)corners[2][0]+corners[3][0]-corners[0][0]-corners[1][0])
/2*CPITCH/format_height;
downy=((double)corners[2][1]+corners[3][1]-corners[0][1]-corners[1][1])
/2*CPITCH/format_height;
/* Load the upper left cross with an estimate of it's position */
crosses[0][0][0]=bilinear(
corners[0][0], corners[1][0],
corners[2][0], corners[3][0],
(double)(BORDER+CHALF)/WIDTH,
(double)(BORDER+CHALF)/format_height);
crosses[0][0][1]=bilinear(
corners[0][1], corners[1][1],
corners[2][1], corners[3][1],
(double)(BORDER+CHALF)/WIDTH,
(double)(BORDER+CHALF)/format_height);
fprintf(stderr,"Finding crosses (%u lines), numbers indicate "
"individual cutlevels:\n", YCROSSES);
for (cy=0;cy<YCROSSES;cy++){
fprintf(stderr,"%3u: ",cy);
for (cx=0;cx<XCROSSES;cx++){
if (cx>0){
/* Copy from left */
crosses[cx][cy][0]=crosses[cx-1][cy][0]+rightx;
crosses[cx][cy][1]=crosses[cx-1][cy][1]+righty;
}else if (cy>0){
/* Copy from above */
crosses[cx][cy][0]=crosses[cx][cy-1][0]+downx;
crosses[cx][cy][1]=crosses[cx][cy-1][1]+downy;
}/* else already preloaded */
resync_cross(crosses[cx][cy]);
cross_stats(cx, cy);
}
putc('\n',stderr);
}
}
/* x,y coords in bit matrix. 0,0 is in the upper left cross UL corner.
* Returns pixel position with integers in centers of pixels. Interpolates
* also the cutlevel */
static void bit_coord(double *xout, double *yout, float *cutlevel,
int x, int y)
{
unsigned cx, cy; /* Cross number */
double xd, yd;
double xrem, yrem;
/* First find the cross numbers */
/* Division of negative numbers is probably undefined in C! */
if (x<CHALF) cx=0;
else cx=(x-CHALF)/CPITCH;
if (y<CHALF) cy=0;
else cy=(y-CHALF)/CPITCH;
if (cx>XCROSSES-2) cx=XCROSSES-2;
if (cy>YCROSSES-2) cy=YCROSSES-2;
/* Now subtrack cross coordinate */
x-=cx*CPITCH+CHALF;
y-=cy*CPITCH+CHALF;
/* x,y now the remainders. Can be negative or more than CPITCH! */
/* Calculate double precision remainders about from 0 to 1 (not always) */
xrem=((double)x+0.5)/CPITCH;
yrem=((double)y+0.5)/CPITCH;
xd=bilinear(
crosses[cx][cy][0], crosses[cx+1][cy][0],
crosses[cx][cy+1][0],crosses[cx+1][cy+1][0],
xrem, yrem);
yd=bilinear(
crosses[cx][cy][1], crosses[cx+1][cy][1],
crosses[cx][cy+1][1],crosses[cx+1][cy+1][1],
xrem, yrem);
if (cutlevel){
*cutlevel=bilinearf(
cutlevels[cx][cy], cutlevels[cx+1][cy],
cutlevels[cx][cy+1],cutlevels[cx+1][cy+1],
xrem, yrem);
}
/* xd, yd are now with integers in UL corners of pixels */
xd-=0.5;
yd-=0.5;
/* xd, yd are now with integers in centers of pixels */
*xout=xd;
*yout=yd;
}
static void read_payload_bit(unsigned char bit)
{
static unsigned accu=1;
accu<<=1;
accu|=bit&1;
if (accu&(1<<8)){
putchar(accu&0xff);
accu=1;
}
}
#if FEC_ORDER !=1
/* Cuts out given bit and shifts the upper part */
static unsigned long shrink(unsigned long in, unsigned bitpos)
{
unsigned long high;
in&=~(1<<bitpos); /* Zero out the bit */
high=in;
in&=(1UL<<bitpos)-1;
high^=in;
return (high>>1)|in;
}
#endif
static void mark_bad_bit(unsigned x, unsigned y, int dir)
{
int size=floor(2*sqrt(hpixel*vpixel)+0.5);
unsigned u;
int i;
int v=dir?0:255;
if (dir){
/* To 1, means black dirt. Upper left edge. */
for (u=0;u<leftedge;u++)
writepixu(u,y,0);
for (u=0;u<topedge;u++)
writepixu(x,u,0);
}else{
/* To 0, means white dirt. Lower right edge. */
for (u=rightedge;u<width;u++)
writepixu(u,y,0);
for (u=bottomedge;u<height;u++)
writepixu(x,u,0);
}
for (i=-size;i<=size;i++){
writepixu(x+i, y-size+1, v);
writepixu(x+i, y+size-1, v);
writepixu(x-size+1,y+i,v);
writepixu(x+size-1,y+i,v);
}
if (dir==2) v=0x80; else v^=0xff;
for (i=-size;i<=size;i++){
writepixu(x+i, y-size, v);
writepixu(x+i, y+size, v);
writepixu(x-size,y+i,v);
writepixu(x+size,y+i,v);
}
}
/* Bit 0 is the one which comes at the top of the page, although it's stored in
* bit FEC_LARGEBITS-1 in the Hamming register.
*
* 1 dir means flipped from 0 to 1 (black dirt), 0 dir means flipped
* from 1 to 0 (white dirt ), 2 dir means unknown
*
* Increments bad_01 or bad_10 and bad_total only for reparable errors.
* Leaves bad_irreparable alone. */
static void print_badbit(unsigned symbol, unsigned bit, unsigned dir)
{
unsigned seq;
int x, y;
double xd, yd; /* Integers in centers */
unsigned char delim;
if (!(bad_total)){
fprintf(stderr,"The following coordinates have damaged bits. "
"\",\" is black dirt, \"'\" white dirt, \":\""
"bit which is a part of an irreparable symbol. "
"Exclamation marks (!) indicate irreparable "
"damage: ");
}
if (dir==1){
bad_01++;
bad_total++;
}
else if (dir==0){
bad_10++;
bad_total++;
}
seq=symbol+bit*FEC_SYMS;
seq2xy(&x, &y, seq);
bit_coord(&xd, &yd, NULL, x, y);
xd=floor(xd+0.5);
yd=floor(yd+0.5);
mark_bad_bit(xd, yd, dir);
assert(x>=0);
switch(dir){
case 0: delim='\''; break;
case 1: delim=','; break;
default: delim=':'; break;
}
fprintf(stderr,"%ld%c%ld ",(long)xd, delim, (long)yd);
}
static void print_badbit_finish(void)
{
if (bad_total){
fprintf(stderr,"\n%lu bits bad from %lu, bit error rate %G%%. "
"%G%% black dirt, %G%% white dirt and "
"%lu (%G%%) irreparable.\n",
bad_total,
USEDBITS,
100*(double)(bad_total)/USEDBITS,
100*(double)bad_01/(bad_total),
100*(double)bad_10/(bad_total),
irreparable,
100*(double)irreparable/(bad_total));
}
else fprintf(stderr,"No bad bits!\n");
#if FEC_ORDER == 1
fprintf(stderr,"Golay stats\n"
"===========\n"
"0 bad bits %lu\n"