version update
[goodguy/cinelerra.git] / cinelerra-5.1 / db / utils / pbm2frame.C
1 #include<stdio.h>
2 #include<stdarg.h>
3 #include<time.h>
4 #include<math.h>
5
6 #include "tdb.h"
7 #include "s.C"
8
9 #include "../../cinelerra/mediadb.inc"
10
11 double errlmt = 8;
12 int dist = 1024;
13 #define SWIDTH 80
14 #define SHEIGHT 45
15 #define SFRM_SZ (SWIDTH*SHEIGHT)
16
17 double
18 mean(uint8_t *dat, int n)
19 {
20   int s = 0;
21   for( int i=0; i<n; ++i, ++dat ) s += *dat;
22   return (double)s / n;
23 }
24
25 double
26 variance(uint8_t *dat, double m, int n)
27 {
28   double ss = 0, s = 0;
29   for( int i=0; i<n; ++i, ++dat ) {
30     double dx = *dat-m;
31     s += dx;  ss += dx*dx;
32   }
33   return (ss - s*s / n) / (n-1);
34 }
35
36 double
37 std_dev(uint8_t *dat, double m, int n)
38 {
39   return sqrt(variance(dat,m,n));
40 }
41
42 double 
43 center(uint8_t *dat, int n, int stride)
44 {
45   int s = 0, ss = 0;
46   for( int i=0; i<n; dat+=stride ) { s += *dat;  ss += ++i * *dat; }
47   return s > 0 ? (double)ss / s : n / 2.;
48 }
49
50 void
51 centroid(uint8_t *dat, int w, int h, double *xx, double *yy)
52 {
53         double x = 0, y = 0;
54         uint8_t *dp = dat;
55         for( int i=h; --i>=0; dp+=w ) x += center(dp, w, 1);
56         for( int i=w; --i>=0; ) y += center(dat+i, h, w);
57         *xx = x / h;  *yy = y / w;
58 }
59
60
61 static inline int clip(int v, int mn, int mx)
62 {
63   return v<mn ? mn : v>mx ? mx : v;
64 }
65
66 void show_diff(uint8_t *kp0, uint8_t *kp1)
67 {
68   int w=SWIDTH, h=SHEIGHT;
69   uint8_t tgt[3*SFRM_SZ];
70   uint8_t *ap = tgt, *bp = tgt+1*SFRM_SZ, *dp = tgt+2*SFRM_SZ;
71   memcpy(ap, kp0, SFRM_SZ);  memcpy(bp, kp1, SFRM_SZ);
72   for( int i=SFRM_SZ; --i>=0; ++dp,++ap,++bp ) *dp = clip(*ap-*bp+128, 0 ,255);
73   FILE *fp = popen("xv -geom 600x1125 -","w");
74   fprintf(fp,"P5\n%d %d\n255\n",w,3*h);
75   fwrite(tgt,w,3*h,fp);
76   fclose(fp);
77 }
78
79 int64_t compare(uint8_t *ap, uint8_t *bp, int sz)
80 {
81   int64_t v = 0;
82   for( int i=sz; --i>=0; ++ap,++bp ) v += fabs(*ap-*bp);
83   return v;
84 }
85
86 double fmean, fsd, fcx, fcy;
87 uint8_t frm[SFRM_SZ];
88 int video_compares = 0, frame_compares = 0;
89
90 int64_t diff(uint8_t *a, uint8_t *b, int w, int h, int dx, int dy, int bias)
91 {
92   int axofs = dx < 0 ? 0 : dx;
93   int ayofs = w * (dy < 0 ? 0 : dy);
94   int aofs = ayofs + axofs;
95   int bxofs = dx < 0 ? -dx : 0;
96   int byofs = w * (dy < 0 ? -dy : 0);
97   int bofs = byofs + bxofs;
98   uint8_t *ap = a + aofs, *bp = b + bofs;
99   int ierr = 0, ww = w-abs(dx), hh = h-abs(dy);
100   for( int y=hh; --y>=0; ap+=w, bp+=w ) {
101     a = ap;  b = bp;
102     for( int x=ww; --x>=0; ++a, ++b ) { ierr += abs(*a-bias - *b); }
103   }
104   return ierr;
105 }
106
107 int64_t 
108 compare(Db::ObjectLoc &loc)
109 {
110 ++video_compares;
111         Video_frameLoc &frame = *(Video_frameLoc*)&loc;
112         double dm = frame.Frame_mean()-fmean;
113         if( fabs(dm) > MEDIA_MEAN_ERRLMT ) return LONG_MAX;
114         double ds = frame.Frame_std_dev()-fsd;
115         if( fabs(ds) > MEDIA_STDDEV_ERRLMT ) return LONG_MAX;
116         double dx = frame.Frame_cx()-fcx;
117         if( fabs(dx) > MEDIA_XCENTER_ERRLMT ) return LONG_MAX;
118         double dy = frame.Frame_cy()-fcy;
119         if( fabs(dy) > MEDIA_YCENTER_ERRLMT ) return LONG_MAX;
120         uint8_t *dat = frame._Frame_data();
121         //int64_t err = cmpr(frm, dat, SFRM_SZ);
122         int64_t err = diff(frm, dat, SWIDTH, SHEIGHT, lround(dx), lround(dy), lround(dm));
123 //show_diff(frm, frame.Frame_data());
124 ++frame_compares;
125         return err;
126 }
127
128
129 int main(int ac, char **av)
130 {
131   int ret;  setbuf(stdout,0);
132   theDb db;
133   db.open(av[1]);
134   //db.access(av[1], 34543, 0);
135   if( !db.opened() || db.error() ) exit(1);
136
137   FILE *fp = fopen(av[2],"r");
138   for( int i=3; --i>=0; ) {
139     for( int ch=fgetc(fp); ch>=0 && ch!='\n'; ch=fgetc(fp) );
140   }
141   
142   int w = SWIDTH, h = SHEIGHT;
143   fread(frm,1,SFRM_SZ,fp);
144
145   fmean = mean(frm, SFRM_SZ);
146   fsd = std_dev(frm, fmean, SFRM_SZ);
147   centroid(frm, w, h, &fcx, &fcy);
148   double fkey = fmean;
149   if( (ret = Video_frameLoc::ikey_Frame_weight(db.video_frame, fkey).Locate()) != 0 ) {
150     printf(" not found, ret = %d\n",ret);
151     return 1;
152   }
153   uint32_t lid = db.video_frame.id(), lfid = lid;
154   printf("frm %d mean %f, std_dev %f, cxy %f+%f = %f\n", lid, fmean, fsd, fcx, fcy, fkey);
155   int64_t lerr = compare(db.video_frame);
156   int64_t pixlmt = SFRM_SZ*errlmt;
157   if( lerr < pixlmt )
158     printf("+  0frame %d / %f\n", db.video_frame.id(),(double)lerr/SFRM_SZ);
159 //show_diff(frm, db.video_frame._Frame_data());
160
161   Video_frameLoc prev(db.video_frame), next(db.video_frame);
162   Video_frameLoc::rkey_Frame_weight prev_key(prev), next_key(next);
163   Db::pgRef next_loc;  next_key.NextLoc(next_loc);
164
165   for( int i=1; i<=dist; ++i ) {
166     if( !prev_key.Locate(Db::keyLT) ) {
167       int64_t err = compare(prev);
168       uint32_t prev_id = prev.id();
169       if( err < pixlmt && lfid != prev_id )
170         printf("-%3dframe %d / %f\n",i, prev_id,(double)err/SFRM_SZ);
171       if( lerr > err ) { lerr = err;  lid = prev_id; }
172     }
173     if( !next_key.Next(next_loc) ) {
174       int64_t err = compare(next);
175       uint32_t next_id = next.id();
176       if( err < pixlmt && lfid != next_id )
177         printf("+%3dframe %d / %f\n",i, next_id,(double)err/SFRM_SZ);
178       if( lerr > err ) { lerr = err;  lid = next_id; }
179     }
180   }
181   printf("%c frame %d / %f, compares %d/%d = %f\n",
182     lerr<pixlmt ? '*' : '#', lid, (double)lerr/SFRM_SZ,
183     frame_compares, video_compares, (double)frame_compares/video_compares);
184
185   db.close();
186   return 0;
187 }
188