-#include<stdio.h>
-#include<stdarg.h>
-#include<time.h>
-#include<math.h>
-#include "tdb.h"
-#include "s.C"
-
-#include "../cinelerra/mediadb.inc"
-
-char *diff_path = 0;
-int curr_id = -1;
-int next_id = -1;
-
-static inline int clip(int v, int mn, int mx)
-{
- return v<mn ? mn : v>mx ? mx : v;
-}
-
-void write_pbm(uint8_t *tp, int w, int h, const char *fmt, ...)
-{
- va_list ap; va_start(ap, fmt);
- char fn[256]; vsnprintf(fn, sizeof(fn), fmt, ap);
- va_end(ap);
- FILE *fp = !strcmp(fn,"-") ? stdout : fopen(fn,"w");
- if( fp ) {
- fprintf(fp,"P5\n%d %d\n255\n",w,h);
- fwrite(tp,w,h,fp);
- fclose(fp);
- }
-}
-
-void pbm_diff(uint8_t *ap, uint8_t *bp, int w, int h, const char *fmt, ...)
-{
- va_list va; va_start(va, fmt);
- char fn[256]; vsnprintf(fn, sizeof(fn), fmt, va);
- va_end(va);
- FILE *fp = !strcmp(fn,"-") ? stdout : fopen(fn,"w");
- if( fp ) {
- int sz = w*h;
- uint8_t dat[sz], *dp = dat;
- for( int i=sz; --i>=0; ++dp,++ap,++bp ) *dp = clip(*ap-*bp+128, 0 ,255);
- fprintf(fp,"P5\n%d %d\n255\n",w,h);
- fwrite(dat,w,h,fp);
- fclose(fp);
- }
-}
-
-int verify_clips(uint8_t *ap,int asz, double afr,
- uint8_t *bp, int bsz, double bfr,
- double &ofs, double &error)
-{
- double atm = asz/afr, btm = bsz/bfr, lmt = fmin(atm, btm) / 2;
- if( lmt > 8 ) lmt = 8;
- if( fabs(atm - btm) > lmt ) return 0;
- double arate, brate, fr; int margin;
- if( afr < bfr ) {
- arate = 1; brate = bfr / afr;
- fr = afr; margin = afr + 0.5;
- }
- else {
- arate = afr / bfr; brate = 1;
- fr = bfr; margin = bfr + 0.5;
- }
- double *awt = (double *)ap, *bwt = (double *)bp;
- double best = 1e100; int ii = -1;
- for( int i=-margin; i<=margin; ++i ) {
- int aofs = i < 0 ? 0 : i*arate + 0.5;
- int bofs = i < 0 ? -i*brate + 0.5 : 0;
- double err = 0; int n = 0;
- for( ;; ++n ) {
- int ai = aofs + n*arate + 0.5;
- if( ai >= asz ) break;
- int bi = bofs + n*brate + 0.5;
- if( bi >= bsz ) break;
- err += fabs(awt[ai] - bwt[bi]);
- }
- err /= n;
- if( err < best ) { best = err; ii = i; }
- }
- ofs = ii / fr; error = best;
- if( best >= MEDIA_WEIGHT_ERRLMT ) return 0;
- return 1;
-}
-
-int *load_frames(theDb &db, int clip_id, int no, int len)
-{
- int *fp = new int[len], first = no;
- for( int i=len; --i>=0; ) fp[i] = -1;
- int ret = TimelineLoc::ikey_Sequences(db.timeline,clip_id,no).Locate();
- if( !ret ) {
- for( int i=0; i<len; ++i,++no ) {
- if( (int)db.timeline.Clip_id() != clip_id ) break;
- int next = db.timeline.Sequence_no();
- if( next != no ) {
-// printf("seqno skipped in clip %d from %d to %d\n",clip_id,no,next);
- no = next;
- }
- int k = no-first;
- if( k >= 0 && k < len ) fp[k] = db.timeline.Frame_id();
- if( TimelineLoc::rkey_Sequences(db.timeline).Next() ) break;
- }
- }
- else
- printf("cant access timeline of clip %d\n",clip_id);
- return fp;
-}
-
-int64_t media_weight(uint8_t *dat, int sz)
-{
- int64_t wt = 0;
- for( int i=sz; --i>=0; ++dat ) wt += *dat;
- return wt;
-}
-
-int64_t media_cmpr(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;
-}
-
-int fix_err(theDb &db, double &err, double ofs,
- int aid, int *afp, int asz, double afr,
- int bid, int *bfp, int bsz, double bfr)
-{
- double arate, brate;
- if( afr < bfr ) {
- arate = 1; brate = bfr / afr;
- }
- else {
- arate = afr / bfr; brate = 1;
- }
- int aofs = ofs < 0 ? 0 : ofs*afr + 0.5;
- int bofs = ofs < 0 ? -ofs*bfr + 0.5 : 0;
- int n = 0, fno = 0; int votes = 0;
- for( ;; ++fno ) {
- int ai = aofs + fno*arate + 0.5;
- if( ai >= asz ) break;
- int bi = bofs + fno*brate + 0.5;
- if( bi >= bsz ) break;
- Video_frameLoc aframe(db.video_frame);
- Video_frameLoc bframe(db.video_frame);
- if( afp[ai] < 0 || bfp[bi] < 0 ) {
-// printf("missing prefix frame %d %d/%d in clip %d/%d\n",
-// n, afp[ai], bfp[bi], aid, bid);
- continue;
- }
- if( aframe.FindId(afp[ai]) ) {
- printf("cant access prefix frame %d (id %d) in clip %d\n",
- ai, afp[ai], aid); continue;
- }
- if( bframe.FindId(bfp[bi]) ) {
- printf("cant access prefix frame %d (id %d) in clip %d\n",
- bi, bfp[bi], bid); continue;
- }
- int w=80, h=45, sfrm_sz = w*h;
- int64_t v = media_cmpr(aframe._Frame_data(), bframe._Frame_data(), sfrm_sz);
- if( v/3600 < MEDIA_SEARCH_ERRLMT ) ++votes;
-//printf("%d %ld(%d,%d)\n", n, v, afp[ai], bfp[bi]);
- if( diff_path ) {
- uint8_t *adat = aframe._Frame_data(), *bdat = bframe._Frame_data();
- write_pbm(adat,w,h,"%s/a%05d.pbm",diff_path,afp[ai]);
- write_pbm(bdat,w,h,"%s/b%05d.pbm",diff_path,bfp[bi]);
- pbm_diff(adat,bdat,80,45,"%s/da%05d-b%05d.pbm",diff_path,afp[ai],bfp[bi]);
- }
- err += v; ++n;
- }
- printf("v%d",votes);
- return n;
-}
-
-int verify_frames(theDb &db, int aid, int bid, double ofs)
-{
- int *apfx=0, *asfx=0, *bpfx=0, *bsfx=0;
- int apsz=0, assz=0, bpsz=0, bssz=0, alen=0, blen=0, afrm=0, bfrm=0;
- double afr = 1, bfr = 1;
- if( !db.clip_set.FindId(aid) ) {
- apsz = db.clip_set.Prefix_size();
- assz = db.clip_set.Suffix_size();
- alen = db.clip_set.Frames();
- apfx = load_frames(db, aid, 0, apsz);
- asfx = load_frames(db, aid, alen-assz, assz);
- afr = db.clip_set.Framerate();
- afrm = db.clip_set.Frames();
- }
- else {
- printf("cant open -a- clip %d\n",aid);
- return 0;
- }
- if( !db.clip_set.FindId(bid) ) {
- bpsz = db.clip_set.Prefix_size();
- bssz = db.clip_set.Suffix_size();
- blen = db.clip_set.Frames();
- bpfx = load_frames(db, bid, 0, bpsz);
- bsfx = load_frames(db, bid, blen-bssz, bssz);
- bfr = db.clip_set.Framerate();
- bfrm = db.clip_set.Frames();
- }
- else {
- printf("cant open -b- clip %d\n",bid);
- return 0;
- }
-//printf("afrm %d, bfrm %d ",afrm, bfrm);
- double perr = 0;
- int pn = fix_err(db, perr, ofs, aid, apfx, apsz, afr, bid, bpfx, bpsz, bfr);
- double serr = 0;
- printf("/");
- ofs -= afrm/afr-bfrm/bfr;
- int sn = fix_err(db, serr, ofs, aid, asfx, assz, afr, bid, bsfx, bssz, bfr);
- printf(" err %0.3f/%0.3f #%d+%d=%d\n", perr/(pn*80*45),serr/(sn*80*45), pn,sn,pn+sn);
-
- delete [] bsfx;
- delete [] bpfx;
- delete [] asfx;
- delete [] apfx;
- return 0;
-}
-
-double avg_wt(uint8_t *wp, int sz)
-{
- double *wt = (double *)wp;
- double v = 0;
- if( sz > 0 ) {
- for( int i=sz; --i>=0; ++wt ) v += *wt;
- v /= sz;
- }
- return v;
-}
-
-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);
- if( ac > 2 ) curr_id = atoi(av[2]);
- if( ac > 3 ) next_id = atoi(av[3]);
- if( ac > 4 ) diff_path = av[4];
- if( curr_id >= 0 && next_id >= 0 ) {
- Clip_setLoc curr(db.clip_set);
- if( (ret=curr.FindId(curr_id)) ) {
- printf("cant access clip %d\n", curr_id);
- exit(1);
- }
- Clip_setLoc next(db.clip_set);
- if( (ret=next.FindId(next_id)) ) {
- printf("cant access clip %d\n", next_id);
- exit(1);
- }
- double awt = curr.Average_weight();
- double bwt = next.Average_weight();
- double error=-1, ofs=-1;
- int bias = awt - bwt;
- verify_clips(curr._Weights(),curr.Frames(),curr.Framerate(),
- next._Weights(), next.Frames(), next.Framerate(), ofs, error);
- printf("dupl a=%4d(%0.2f)-%0.2f b=%4d(%0.2f)-%0.2f %s%d err=%0.3f ofs=%0.5f ",
- curr.id(), curr.Framerate(),curr.Frames()/curr.Framerate(),
- next.id(), next.Framerate(), next.Frames()/next.Framerate(),
- bias >= 0 ? "+" : "", bias, error, ofs);
- verify_frames(db, curr.id(), next.id(), ofs);
- exit(0);
- }
- Clip_setLoc curr(db.clip_set);
- Db::pgRef curr_loc;
- if( !(ret=curr.FirstId(curr_loc)) ) {
- do {
- Clip_setLoc next(curr);
- Db::pgRef next_loc = curr_loc;
- double awt = curr.Average_weight();
- while( !next.NextId(next_loc) ) {
- double error=-1, ofs =-1;
- double bwt = next.Average_weight();
- int bias = awt - bwt;
- if( fabs(bias) > 10 ) continue;
- if( verify_clips(curr._Weights(),curr.Frames(),curr.Framerate(),
- next._Weights(), next.Frames(), next.Framerate(), ofs, error) ) {
- printf("dupl a=%4d(%0.2f)-%0.2f b=%4d(%0.2f)-%0.2f %s%d err=%0.3f ofs=%0.3f ",
- curr.id(), curr.Framerate(),curr.Frames()/curr.Framerate(),
- next.id(), next.Framerate(), next.Frames()/next.Framerate(),
- bias >= 0 ? "+" : "", bias, error, ofs);
- verify_frames(db, curr.id(), next.id(), ofs);
- break;
- }
- }
- } while( !(ret=curr.NextId(curr_loc)) );
- }
- db.close();
- return 0;
-}
-