--- /dev/null
+#include<stdio.h>
+#include<stdarg.h>
+#include<time.h>
+#include<math.h>
+
+#include "tdb.h"
+#include "s.C"
+
+#include "../../cinelerra/mediadb.inc"
+
+double errlmt = 8;
+int dist = 1024;
+#define SWIDTH 80
+#define SHEIGHT 45
+#define SFRM_SZ (SWIDTH*SHEIGHT)
+
+double
+mean(uint8_t *dat, int n)
+{
+ int s = 0;
+ for( int i=0; i<n; ++i, ++dat ) s += *dat;
+ return (double)s / n;
+}
+
+double
+variance(uint8_t *dat, double m, int n)
+{
+ double ss = 0, s = 0;
+ for( int i=0; i<n; ++i, ++dat ) {
+ double dx = *dat-m;
+ s += dx; ss += dx*dx;
+ }
+ return (ss - s*s / n) / (n-1);
+}
+
+double
+std_dev(uint8_t *dat, double m, int n)
+{
+ return sqrt(variance(dat,m,n));
+}
+
+double
+center(uint8_t *dat, int n, int stride)
+{
+ int s = 0, ss = 0;
+ for( int i=0; i<n; dat+=stride ) { s += *dat; ss += ++i * *dat; }
+ return s > 0 ? (double)ss / s : n / 2.;
+}
+
+void
+centroid(uint8_t *dat, int w, int h, double *xx, double *yy)
+{
+ double x = 0, y = 0;
+ uint8_t *dp = dat;
+ for( int i=h; --i>=0; dp+=w ) x += center(dp, w, 1);
+ for( int i=w; --i>=0; ) y += center(dat+i, h, w);
+ *xx = x / h; *yy = y / w;
+}
+
+
+static inline int clip(int v, int mn, int mx)
+{
+ return v<mn ? mn : v>mx ? mx : v;
+}
+
+void show_diff(uint8_t *kp0, uint8_t *kp1)
+{
+ int w=SWIDTH, h=SHEIGHT;
+ uint8_t tgt[3*SFRM_SZ];
+ uint8_t *ap = tgt, *bp = tgt+1*SFRM_SZ, *dp = tgt+2*SFRM_SZ;
+ memcpy(ap, kp0, SFRM_SZ); memcpy(bp, kp1, SFRM_SZ);
+ for( int i=SFRM_SZ; --i>=0; ++dp,++ap,++bp ) *dp = clip(*ap-*bp+128, 0 ,255);
+ FILE *fp = popen("xv -geom 600x1125 -","w");
+ fprintf(fp,"P5\n%d %d\n255\n",w,3*h);
+ fwrite(tgt,w,3*h,fp);
+ fclose(fp);
+}
+
+int64_t compare(uint8_t *ap, uint8_t *bp, int sz)
+{
+ int64_t v = 0;
+ for( int i=sz; --i>=0; ++ap,++bp ) v += fabs(*ap-*bp);
+ return v;
+}
+
+double fmean, fsd, fcx, fcy;
+uint8_t frm[SFRM_SZ];
+int video_compares = 0, frame_compares = 0;
+
+int64_t diff(uint8_t *a, uint8_t *b, int w, int h, int dx, int dy, int bias)
+{
+ int axofs = dx < 0 ? 0 : dx;
+ int ayofs = w * (dy < 0 ? 0 : dy);
+ int aofs = ayofs + axofs;
+ int bxofs = dx < 0 ? -dx : 0;
+ int byofs = w * (dy < 0 ? -dy : 0);
+ int bofs = byofs + bxofs;
+ uint8_t *ap = a + aofs, *bp = b + bofs;
+ int ierr = 0, ww = w-abs(dx), hh = h-abs(dy);
+ for( int y=hh; --y>=0; ap+=w, bp+=w ) {
+ a = ap; b = bp;
+ for( int x=ww; --x>=0; ++a, ++b ) { ierr += abs(*a-bias - *b); }
+ }
+ return ierr;
+}
+
+int64_t
+compare(Db::ObjectLoc &loc)
+{
+++video_compares;
+ Video_frameLoc &frame = *(Video_frameLoc*)&loc;
+ double dm = frame.Frame_mean()-fmean;
+ if( fabs(dm) > MEDIA_MEAN_ERRLMT ) return LONG_MAX;
+ double ds = frame.Frame_std_dev()-fsd;
+ if( fabs(ds) > MEDIA_STDDEV_ERRLMT ) return LONG_MAX;
+ double dx = frame.Frame_cx()-fcx;
+ if( fabs(dx) > MEDIA_XCENTER_ERRLMT ) return LONG_MAX;
+ double dy = frame.Frame_cy()-fcy;
+ if( fabs(dy) > MEDIA_YCENTER_ERRLMT ) return LONG_MAX;
+ uint8_t *dat = frame._Frame_data();
+ //int64_t err = cmpr(frm, dat, SFRM_SZ);
+ int64_t err = diff(frm, dat, SWIDTH, SHEIGHT, lround(dx), lround(dy), lround(dm));
+//show_diff(frm, frame.Frame_data());
+++frame_compares;
+ return err;
+}
+
+
+int main(int ac, char **av)
+{
+ int ret; setbuf(stdout,0);
+ theDb db;
+ db.open(av[1]);
+ //db.access(av[1], 34543, 0);
+ if( !db.opened() || db.error() ) exit(1);
+
+ FILE *fp = fopen(av[2],"r");
+ for( int i=3; --i>=0; ) {
+ for( int ch=fgetc(fp); ch>=0 && ch!='\n'; ch=fgetc(fp) );
+ }
+
+ int w = SWIDTH, h = SHEIGHT;
+ fread(frm,1,SFRM_SZ,fp);
+
+ fmean = mean(frm, SFRM_SZ);
+ fsd = std_dev(frm, fmean, SFRM_SZ);
+ centroid(frm, w, h, &fcx, &fcy);
+ double fkey = fmean;
+ if( (ret = Video_frameLoc::ikey_Frame_weight(db.video_frame, fkey).Locate()) != 0 ) {
+ printf(" not found, ret = %d\n",ret);
+ return 1;
+ }
+ uint32_t lid = db.video_frame.id(), lfid = lid;
+ printf("frm %d mean %f, std_dev %f, cxy %f+%f = %f\n", lid, fmean, fsd, fcx, fcy, fkey);
+ int64_t lerr = compare(db.video_frame);
+ int64_t pixlmt = SFRM_SZ*errlmt;
+ if( lerr < pixlmt )
+ printf("+ 0frame %d / %f\n", db.video_frame.id(),(double)lerr/SFRM_SZ);
+//show_diff(frm, db.video_frame._Frame_data());
+
+ Video_frameLoc prev(db.video_frame), next(db.video_frame);
+ Video_frameLoc::rkey_Frame_weight prev_key(prev), next_key(next);
+ Db::pgRef next_loc; next_key.NextLoc(next_loc);
+
+ for( int i=1; i<=dist; ++i ) {
+ if( !prev_key.Locate(Db::keyLT) ) {
+ int64_t err = compare(prev);
+ uint32_t prev_id = prev.id();
+ if( err < pixlmt && lfid != prev_id )
+ printf("-%3dframe %d / %f\n",i, prev_id,(double)err/SFRM_SZ);
+ if( lerr > err ) { lerr = err; lid = prev_id; }
+ }
+ if( !next_key.Next(next_loc) ) {
+ int64_t err = compare(next);
+ uint32_t next_id = next.id();
+ if( err < pixlmt && lfid != next_id )
+ printf("+%3dframe %d / %f\n",i, next_id,(double)err/SFRM_SZ);
+ if( lerr > err ) { lerr = err; lid = next_id; }
+ }
+ }
+ printf("%c frame %d / %f, compares %d/%d = %f\n",
+ lerr<pixlmt ? '*' : '#', lid, (double)lerr/SFRM_SZ,
+ frame_compares, video_compares, (double)frame_compares/video_compares);
+
+ db.close();
+ return 0;
+}
+
projects / goodguy / history.git / blobdiff