libhb/detelecine.c

   1 #include "hb.h"
   2 #include "ffmpeg/avcodec.h"
   3 #include "mpeg2dec/mpeg2.h"
   4
   5 /*
   6  *
   7  * PULLUP DEFINITIONS
   8  *
   9  */
  10
  11 #define PULLUP_FMT_Y         1
  12 #define PULLUP_HAVE_BREAKS   1
  13 #define PULLUP_HAVE_AFFINITY 2
  14 #define PULLUP_BREAK_LEFT    1
  15 #define PULLUP_BREAK_RIGHT   2
  16
  17 #define PULLUP_ABS( a ) (((a)^((a)>>31))-((a)>>31))
  18
  19 #ifndef PIC_FLAG_REPEAT_FIRST_FIELD
  20 #define PIC_FLAG_REPEAT_FIRST_FIELD 256
  21 #endif
  22
  23 struct pullup_buffer
  24 {
  25         int lock[2];
  26         unsigned char **planes;
  27 };
  28
  29 struct pullup_field
  30 {
  31         int parity;
  32         struct pullup_buffer *buffer;
  33         unsigned int flags;
  34         int breaks;
  35         int affinity;
  36         int *diffs;
  37         int *comb;
  38         int *var;
  39         struct pullup_field *prev, *next;
  40 };
  41
  42 struct pullup_frame
  43 {
  44         int lock;
  45         int length;
  46         int parity;
  47         struct pullup_buffer **ifields, *ofields[2];
  48         struct pullup_buffer *buffer;
  49 };
  50
  51 struct pullup_context
  52 {
  53         /* Public interface */
  54         int format;
  55         int nplanes;
  56         int *bpp, *w, *h, *stride, *background;
  57         unsigned int cpu;
  58         int junk_left, junk_right, junk_top, junk_bottom;
  59         int verbose;
  60         int metric_plane;
  61         int strict_breaks;
  62         int strict_pairs;
  63         /* Internal data */
  64         struct pullup_field *first, *last, *head;
  65         struct pullup_buffer *buffers;
  66         int nbuffers;
  67         int (*diff)(unsigned char *, unsigned char *, int);
  68         int (*comb)(unsigned char *, unsigned char *, int);
  69         int (*var)(unsigned char *, unsigned char *, int);
  70         int metric_w, metric_h, metric_len, metric_offset;
  71         struct pullup_frame *frame;
  72 };
  73
  74 /*
  75  *
  76  * DETELECINE FILTER DEFINITIONS
  77  *
  78  */
  79
  80 struct hb_filter_private_s
  81 {
  82     int              pix_fmt;
  83     int              width[3];
  84     int              height[3];
  85
  86     struct pullup_context * pullup_ctx;
  87     int                     pullup_fakecount;
  88     int                     pullup_skipflag;
  89
  90     AVPicture        pic_in;
  91     AVPicture        pic_out;
  92     hb_buffer_t    * buf_out;
  93 };
  94
  95 hb_filter_private_t * hb_detelecine_init( int pix_fmt,
  96                                           int width,
  97                                           int height,
  98                                           char * settings );
  99
 100 int hb_detelecine_work( const hb_buffer_t * buf_in,
 101                         hb_buffer_t ** buf_out,
 102                         int pix_fmt,
 103                         int width,
 104                         int height,
 105                         hb_filter_private_t * pv );
 106
 107 void hb_detelecine_close( hb_filter_private_t * pv );
 108
 109 hb_filter_object_t hb_filter_detelecine =
 110 {
 111     FILTER_DETELECINE,
 112     "Detelecine (pullup)",
 113     NULL,
 114     hb_detelecine_init,
 115     hb_detelecine_work,
 116     hb_detelecine_close,
 117 };
 118
 119 /*
 120  *
 121  * PULLUP STATIC FUNCTIONS
 122  *
 123  */
 124
 125 static int pullup_diff_y( unsigned char  *a, unsigned char * b, int s )
 126 {
 127         int i, j, diff = 0;
 128         for( i = 4; i; i-- )
 129     {
 130                 for( j = 0; j < 8; j++ )
 131         {
 132             diff += PULLUP_ABS( a[j]-b[j] );
 133         }
 134                 a+=s; b+=s;
 135         }
 136         return diff;
 137 }
 138
 139 static int pullup_licomb_y( unsigned char * a, unsigned char * b, int s )
 140 {
 141         int i, j, diff = 0;
 142         for( i = 4; i; i-- )
 143     {
 144                 for( j = 0; j < 8; j++ )
 145         {
 146                         diff += PULLUP_ABS( (a[j]<<1) - b[j-s] - b[j] )
 147                                   + PULLUP_ABS( (b[j]<<1) - a[j] - a[j+s] );
 148         }
 149                 a+=s; b+=s;
 150         }
 151         return diff;
 152 }
 153
 154 static int pullup_var_y( unsigned char * a, unsigned char * b, int s )
 155 {
 156         int i, j, var = 0;
 157         for( i = 3; i; i-- )
 158     {
 159                 for( j = 0; j < 8; j++ )
 160         {
 161                         var += PULLUP_ABS( a[j]-a[j+s] );
 162                 }
 163                 a+=s; b+=s;
 164         }
 165         return 4*var;
 166 }
 167
 168 static void pullup_alloc_metrics( struct pullup_context * c,
 169                                   struct pullup_field * f )
 170 {
 171         f->diffs = calloc( c->metric_len, sizeof(int) );
 172         f->comb  = calloc( c->metric_len, sizeof(int) );
 173         f->var   = calloc( c->metric_len, sizeof(int) );
 174 }
 175
 176 static void pullup_compute_metric( struct pullup_context * c,
 177                                    struct pullup_field * fa, int pa,
 178                                    struct pullup_field * fb, int pb,
 179                                    int (* func)( unsigned char *,
 180                                                  unsigned char *, int),
 181                                    int * dest )
 182 {
 183         unsigned char *a, *b;
 184         int x, y;
 185         int mp    = c->metric_plane;
 186         int xstep = c->bpp[mp];
 187         int ystep = c->stride[mp]<<3;
 188         int s     = c->stride[mp]<<1; /* field stride */
 189         int w     = c->metric_w*xstep;
 190
 191         if( !fa->buffer || !fb->buffer ) return;
 192
 193         /* Shortcut for duplicate fields (e.g. from RFF flag) */
 194         if( fa->buffer == fb->buffer && pa == pb )
 195     {
 196                 memset( dest, 0, c->metric_len * sizeof(int) );
 197                 return;
 198         }
 199
 200         a = fa->buffer->planes[mp] + pa * c->stride[mp] + c->metric_offset;
 201         b = fb->buffer->planes[mp] + pb * c->stride[mp] + c->metric_offset;
 202
 203         for( y = c->metric_h; y; y-- )
 204     {
 205                 for( x = 0; x < w; x += xstep )
 206         {
 207                         *dest++ = func( a + x, b + x, s );
 208                 }
 209                 a += ystep; b += ystep;
 210         }
 211 }
 212
 213 static struct pullup_field * pullup_make_field_queue( struct pullup_context * c,
 214                                                       int len )
 215 {
 216         struct pullup_field * head, * f;
 217         f = head = calloc( 1, sizeof(struct pullup_field) );
 218         pullup_alloc_metrics( c, f );
 219         for ( ; len > 0; len-- )
 220     {
 221                 f->next = calloc( 1, sizeof(struct pullup_field) );
 222                 f->next->prev = f;
 223                 f = f->next;
 224                 pullup_alloc_metrics( c, f );
 225         }
 226         f->next = head;
 227         head->prev = f;
 228         return head;
 229 }
 230
 231 static void pullup_check_field_queue( struct pullup_context * c )
 232 {
 233         if( c->head->next == c->first )
 234     {
 235                 struct pullup_field *f = calloc( 1, sizeof(struct pullup_field) );
 236                 pullup_alloc_metrics( c, f );
 237                 f->prev = c->head;
 238                 f->next = c->first;
 239                 c->head->next = f;
 240                 c->first->prev = f;
 241         }
 242 }
 243
 244 static void pullup_copy_field( struct pullup_context * c,
 245                                struct pullup_buffer * dest,
 246                                struct pullup_buffer * src,
 247                                int parity )
 248 {
 249         int i, j;
 250         unsigned char *d, *s;
 251         for( i = 0; i < c->nplanes; i++ )
 252     {
 253                 s = src->planes[i] + parity*c->stride[i];
 254                 d = dest->planes[i] + parity*c->stride[i];
 255                 for( j = c->h[i]>>1; j; j-- )
 256         {
 257                         memcpy( d, s, c->stride[i] );
 258                         s += c->stride[i]<<1;
 259                         d += c->stride[i]<<1;
 260                 }
 261         }
 262 }
 263
 264
 265 static int pullup_queue_length( struct pullup_field * begin,
 266                                 struct pullup_field * end )
 267 {
 268         int count = 1;
 269         struct pullup_field * f;
 270
 271         if( !begin || !end ) return 0;
 272         for( f = begin; f != end; f = f->next ) count++;
 273         return count;
 274 }
 275
 276 static int pullup_find_first_break( struct pullup_field * f, int max )
 277 {
 278         int i;
 279         for( i = 0; i < max; i++ )
 280     {
 281                 if( f->breaks & PULLUP_BREAK_RIGHT ||
 282             f->next->breaks & PULLUP_BREAK_LEFT )
 283         {
 284                         return i+1;
 285         }
 286                 f = f->next;
 287         }
 288         return 0;
 289 }
 290
 291 static void pullup_compute_breaks( struct pullup_context * c,
 292                                    struct pullup_field * f0 )
 293 {
 294         int i;
 295         struct pullup_field *f1 = f0->next;
 296         struct pullup_field *f2 = f1->next;
 297         struct pullup_field *f3 = f2->next;
 298         int l, max_l=0, max_r=0;
 299
 300         if( f0->flags & PULLUP_HAVE_BREAKS ) return;
 301         f0->flags |= PULLUP_HAVE_BREAKS;
 302
 303         /* Special case when fields are 100% identical */
 304         if( f0->buffer == f2->buffer && f1->buffer != f3->buffer )
 305     {
 306                 f2->breaks |= PULLUP_BREAK_RIGHT;
 307                 return;
 308         }
 309         if( f0->buffer != f2->buffer && f1->buffer == f3->buffer )
 310     {
 311                 f1->breaks |= PULLUP_BREAK_LEFT;
 312                 return;
 313         }
 314
 315         for( i = 0; i < c->metric_len; i++ )
 316     {
 317                 l = f2->diffs[i] - f3->diffs[i];
 318                 if(  l > max_l) max_l = l;
 319                 if( -l > max_r) max_r = -l;
 320         }
 321
 322         /* Don't get tripped up when differences are mostly quant error */
 323         if( max_l + max_r < 128 ) return;
 324         if( max_l > 4*max_r ) f1->breaks |= PULLUP_BREAK_LEFT;
 325         if( max_r > 4*max_l ) f2->breaks |= PULLUP_BREAK_RIGHT;
 326 }
 327
 328 static void pullup_compute_affinity( struct pullup_context * c,
 329                                      struct pullup_field * f )
 330 {
 331         int i;
 332         int max_l = 0, max_r = 0, l;
 333
 334     if( f->flags & PULLUP_HAVE_AFFINITY )
 335     {
 336         return;
 337     }
 338         f->flags |= PULLUP_HAVE_AFFINITY;
 339
 340         if( f->buffer == f->next->next->buffer )
 341     {
 342                 f->affinity             =  1;
 343                 f->next->affinity       =  0;
 344                 f->next->next->affinity = -1;
 345
 346                 f->next->flags       |= PULLUP_HAVE_AFFINITY;
 347                 f->next->next->flags |= PULLUP_HAVE_AFFINITY;
 348
 349                 return;
 350         }
 351
 352     for( i = 0; i < c->metric_len; i++ )
 353     {
 354         int lv = f->prev->var[i];
 355         int rv = f->next->var[i];
 356         int v  = f->var[i];
 357         int lc = f->comb[i] - (v+lv) + PULLUP_ABS( v-lv );
 358         int rc = f->next->comb[i] - (v+rv) + PULLUP_ABS( v-rv );
 359
 360         lc = (lc > 0) ? lc : 0;
 361         rc = (rc > 0) ? rc : 0;
 362         l = lc - rc;
 363         if(  l > max_l ) max_l = l;
 364         if( -l > max_r ) max_r = -l;
 365     }
 366
 367     if( max_l + max_r < 64 )
 368     {
 369         return;
 370     }
 371
 372     if( max_r > 6*max_l )
 373     {
 374         f->affinity = -1;
 375     }
 376     else if( max_l > 6*max_r )
 377     {
 378         f->affinity = 1;
 379     }
 380 }
 381
 382 static void pullup_foo( struct pullup_context * c )
 383 {
 384         struct pullup_field * f = c->first;
 385         int i, n = pullup_queue_length (f, c->last );
 386         for( i = 0; i < n-1; i++ )
 387     {
 388                 if( i < n-3 ) pullup_compute_breaks( c, f );
 389                 pullup_compute_affinity( c, f );
 390                 f = f->next;
 391         }
 392 }
 393
 394 static int pullup_decide_frame_length( struct pullup_context * c )
 395 {
 396         struct pullup_field *f0 = c->first;
 397         struct pullup_field *f1 = f0->next;
 398         struct pullup_field *f2 = f1->next;
 399         struct pullup_field *f3 = f2->next;
 400         int l;
 401
 402         if( pullup_queue_length( c->first, c->last ) < 4 )
 403     {
 404         return 0;
 405     }
 406         pullup_foo( c );
 407
 408         if( f0->affinity == -1 ) return 1;
 409
 410         l = pullup_find_first_break( f0, 3 );
 411         if( l == 1 && c->strict_breaks < 0 ) l = 0;
 412
 413         switch (l)
 414     {
 415         case 1:
 416             if ( c->strict_breaks < 1 &&
 417                  f0->affinity == 1 &&
 418                  f1->affinity == -1 )
 419             {
 420                 return 2;
 421             }
 422             else
 423             {
 424                 return 1;
 425             }
 426
 427         case 2:
 428             /* FIXME: strictly speaking, f0->prev is no longer valid... :) */
 429             if( c->strict_pairs &&
 430                 (f0->prev->breaks & PULLUP_BREAK_RIGHT) &&
 431                 (f2->breaks & PULLUP_BREAK_LEFT) &&
 432                 (f0->affinity != 1 || f1->affinity != -1) )
 433             {
 434                 return 1;
 435             }
 436             if( f1->affinity == 1 )
 437             {
 438                 return 1;
 439             }
 440             else
 441             {
 442                 return 2;
 443             }
 444
 445         case 3:
 446             if( f2->affinity == 1 )
 447             {
 448                 return 2;
 449             }
 450             else
 451             {
 452                 return 3;
 453             }
 454
 455         default:
 456             /* 9 possibilities covered before switch */
 457             if( f1->affinity == 1 )
 458             {
 459                 return 1; /* covers 6 */
 460             }
 461             else if( f1->affinity == -1 )
 462             {
 463                 return 2; /* covers 6 */
 464             }
 465             else if( f2->affinity == -1 )
 466             {
 467                 /* covers 2 */
 468                 if( f0->affinity == 1 )
 469                 {
 470                     return 3;
 471                 }
 472                 else
 473                 {
 474                     return 1;
 475                 }
 476             }
 477             else
 478             {
 479                 return 2; /* the remaining 6 */
 480             }
 481         }
 482 }
 483
 484 static void pullup_print_aff_and_breaks(struct pullup_context * c,
 485                                         struct pullup_field * f )
 486 {
 487         int i;
 488         struct pullup_field * f0 = f;
 489         const char aff_l[] = "+..", aff_r[] = "..+";
 490         printf( "\naffinity: " );
 491         for( i = 0; i < 4; i++ )
 492     {
 493                 printf( "%c%d%c",
 494                 aff_l[1+f->affinity],
 495                 i,
 496                 aff_r[1+f->affinity] );
 497
 498                 f = f->next;
 499         }
 500         f = f0;
 501         printf("\nbreaks:   ");
 502         for( i = 0; i < 4; i++ )
 503     {
 504                 printf( "%c%d%c",
 505                 f->breaks & PULLUP_BREAK_LEFT  ? '|' : '.',
 506                 i,
 507                 f->breaks & PULLUP_BREAK_RIGHT ? '|' : '.' );
 508
 509                 f = f->next;
 510         }
 511         printf("\n");
 512 }
 513
 514 /*
 515  *
 516  * PULLUP CONTEXT FUNCTIONS
 517  *
 518  */
 519
 520 struct pullup_context * pullup_alloc_context( void )
 521 {
 522         struct pullup_context * c;
 523
 524         c = calloc( 1, sizeof(struct pullup_context)) ;
 525
 526         return c;
 527 }
 528
 529 void pullup_preinit_context( struct pullup_context * c )
 530 {
 531         c->bpp        = calloc( c->nplanes, sizeof(int) );
 532         c->w          = calloc( c->nplanes, sizeof(int) );
 533         c->h          = calloc( c->nplanes, sizeof(int) );
 534         c->stride     = calloc( c->nplanes, sizeof(int) );
 535         c->background = calloc( c->nplanes, sizeof(int) );
 536 }
 537
 538 void pullup_init_context( struct pullup_context * c )
 539 {
 540         int mp = c->metric_plane;
 541         if ( c->nbuffers < 10 )
 542     {
 543         c->nbuffers = 10;
 544     }
 545         c->buffers = calloc( c->nbuffers, sizeof (struct pullup_buffer) );
 546
 547         c->metric_w      = (c->w[mp] - ((c->junk_left + c->junk_right) << 3)) >> 3;
 548         c->metric_h      = (c->h[mp] - ((c->junk_top + c->junk_bottom) << 1)) >> 3;
 549         c->metric_offset = c->junk_left*c->bpp[mp] + (c->junk_top<<1)*c->stride[mp];
 550         c->metric_len    = c->metric_w * c->metric_h;
 551
 552         c->head = pullup_make_field_queue( c, 8 );
 553
 554         c->frame = calloc( 1, sizeof (struct pullup_frame) );
 555         c->frame->ifields = calloc( 3, sizeof (struct pullup_buffer *) );
 556
 557         if( c->format == PULLUP_FMT_Y )
 558     {
 559         c->diff = pullup_diff_y;
 560         c->comb = pullup_licomb_y;
 561         c->var  = pullup_var_y;
 562         }
 563 }
 564
 565 void pullup_free_context( struct pullup_context * c )
 566 {
 567         struct pullup_field * f;
 568
 569     free( c->buffers );
 570
 571         f = c->head;
 572         do
 573     {
 574                 free( f->diffs );
 575                 free( f->comb );
 576                 f = f->next;
 577                 free( f->prev );
 578         }
 579     while( f != c->head );
 580
 581     free( c->frame );
 582         free( c );
 583 }
 584
 585 /*
 586  *
 587  * PULLUP BUFFER FUNCTIONS
 588  *
 589  */
 590
 591 static void pullup_alloc_buffer( struct pullup_context * c,
 592                                  struct pullup_buffer * b )
 593 {
 594         int i;
 595         if( b->planes ) return;
 596         b->planes = calloc( c->nplanes, sizeof(unsigned char *) );
 597         for ( i = 0; i < c->nplanes; i++ )
 598     {
 599                 b->planes[i] = malloc(c->h[i]*c->stride[i]);
 600                 /* Deal with idiotic 128=0 for chroma: */
 601                 memset( b->planes[i], c->background[i], c->h[i]*c->stride[i] );
 602         }
 603 }
 604
 605 struct pullup_buffer * pullup_lock_buffer( struct pullup_buffer * b,
 606                                            int parity )
 607 {
 608         if( !b ) return 0;
 609         if( (parity+1) & 1 ) b->lock[0]++;
 610         if( (parity+1) & 2 ) b->lock[1]++;
 611
 612         return b;
 613 }
 614
 615 void pullup_release_buffer( struct pullup_buffer * b,
 616                             int parity )
 617 {
 618         if( !b ) return;
 619         if( (parity+1) & 1 ) b->lock[0]--;
 620         if( (parity+1) & 2 ) b->lock[1]--;
 621 }
 622
 623 struct pullup_buffer * pullup_get_buffer( struct pullup_context * c,
 624                                           int parity )
 625 {
 626         int i;
 627
 628         /* Try first to get the sister buffer for the previous field */
 629         if( parity < 2 &&
 630         c->last &&
 631         parity != c->last->parity &&
 632             !c->last->buffer->lock[parity])
 633     {
 634                 pullup_alloc_buffer( c, c->last->buffer );
 635                 return pullup_lock_buffer( c->last->buffer, parity );
 636         }
 637
 638         /* Prefer a buffer with both fields open */
 639         for( i = 0; i < c->nbuffers; i++ )
 640     {
 641                 if( c->buffers[i].lock[0] ) continue;
 642                 if( c->buffers[i].lock[1] ) continue;
 643                 pullup_alloc_buffer( c, &c->buffers[i] );
 644                 return pullup_lock_buffer( &c->buffers[i], parity );
 645         }
 646
 647         if( parity == 2 ) return 0;
 648
 649         /* Search for any half-free buffer */
 650         for( i = 0; i < c->nbuffers; i++ )
 651     {
 652                 if( ((parity+1) & 1) && c->buffers[i].lock[0] ) continue;
 653                 if( ((parity+1) & 2) && c->buffers[i].lock[1] ) continue;
 654                 pullup_alloc_buffer( c, &c->buffers[i] );
 655                 return pullup_lock_buffer( &c->buffers[i], parity );
 656         }
 657
 658         return 0;
 659 }
 660
 661 /*
 662  *
 663  * PULLUP FRAME FUNCTIONS
 664  *
 665  */
 666
 667 struct pullup_frame * pullup_get_frame( struct pullup_context * c )
 668 {
 669         int i;
 670         struct pullup_frame * fr = c->frame;
 671         int n = pullup_decide_frame_length( c );
 672         int aff = c->first->next->affinity;
 673
 674         if ( !n ) return 0;
 675         if ( fr->lock ) return 0;
 676
 677         if ( c->verbose )
 678     {
 679                 pullup_print_aff_and_breaks(c, c->first);
 680                 printf("duration: %d    \n", n);
 681         }
 682
 683         fr->lock++;
 684         fr->length = n;
 685         fr->parity = c->first->parity;
 686         fr->buffer = 0;
 687         for( i = 0; i < n; i++ )
 688     {
 689                 /* We cheat and steal the buffer without release+relock */
 690                 fr->ifields[i] = c->first->buffer;
 691                 c->first->buffer = 0;
 692                 c->first = c->first->next;
 693         }
 694
 695         if( n == 1 )
 696     {
 697                 fr->ofields[fr->parity] = fr->ifields[0];
 698                 fr->ofields[fr->parity^1] = 0;
 699         }
 700     else if( n == 2 )
 701     {
 702                 fr->ofields[fr->parity] = fr->ifields[0];
 703                 fr->ofields[fr->parity^1] = fr->ifields[1];
 704         }
 705     else if( n == 3 )
 706     {
 707                 if( aff == 0 )
 708         {
 709                         aff = (fr->ifields[0] == fr->ifields[1]) ? -1 : 1;
 710         }
 711                 fr->ofields[fr->parity]   = fr->ifields[1+aff];
 712                 fr->ofields[fr->parity^1] = fr->ifields[1];
 713         }
 714         pullup_lock_buffer( fr->ofields[0], 0 );
 715         pullup_lock_buffer( fr->ofields[1], 1 );
 716
 717         if( fr->ofields[0] == fr->ofields[1] )
 718     {
 719                 fr->buffer = fr->ofields[0];
 720                 pullup_lock_buffer(fr->buffer, 2);
 721                 return fr;
 722         }
 723         return fr;
 724 }
 725
 726 void pullup_pack_frame( struct pullup_context * c, struct pullup_frame * fr)
 727 {
 728         int i;
 729         if (fr->buffer) return;
 730         if (fr->length < 2) return; /* FIXME: deal with this */
 731         for( i = 0; i < 2; i++ )
 732         {
 733                 if( fr->ofields[i]->lock[i^1] ) continue;
 734                 fr->buffer = fr->ofields[i];
 735                 pullup_lock_buffer(fr->buffer, 2);
 736                 pullup_copy_field( c, fr->buffer, fr->ofields[i^1], i^1 );
 737                 return;
 738         }
 739         fr->buffer = pullup_get_buffer( c, 2 );
 740         pullup_copy_field( c, fr->buffer, fr->ofields[0], 0 );
 741         pullup_copy_field( c, fr->buffer, fr->ofields[1], 1 );
 742 }
 743
 744 void pullup_release_frame( struct pullup_frame * fr )
 745 {
 746         int i;
 747         for( i = 0; i < fr->length; i++ )
 748     {
 749                 pullup_release_buffer( fr->ifields[i], fr->parity ^ (i&1) );
 750     }
 751         pullup_release_buffer( fr->ofields[0], 0 );
 752         pullup_release_buffer( fr->ofields[1], 1 );
 753         if (fr->buffer) pullup_release_buffer( fr->buffer, 2 );
 754         fr->lock--;
 755 }
 756
 757 /*
 758  *
 759  * PULLUP FIELD FUNCTIONS
 760  *
 761  */
 762
 763 void pullup_submit_field( struct pullup_context * c,
 764                           struct pullup_buffer * b,
 765                           int parity )
 766 {
 767         struct pullup_field * f;
 768
 769         /* Grow the circular list if needed */
 770         pullup_check_field_queue( c );
 771
 772         /* Cannot have two fields of same parity in a row; drop the new one */
 773         if( c->last && c->last->parity == parity ) return;
 774
 775         f = c->head;
 776         f->parity = parity;
 777         f->buffer = pullup_lock_buffer( b, parity );
 778         f->flags = 0;
 779         f->breaks = 0;
 780         f->affinity = 0;
 781
 782         pullup_compute_metric( c, f, parity, f->prev->prev,
 783                            parity, c->diff, f->diffs );
 784         pullup_compute_metric( c, parity?f->prev:f, 0,
 785                            parity?f:f->prev, 1, c->comb, f->comb );
 786         pullup_compute_metric( c, f, parity, f,
 787                            -1, c->var, f->var );
 788
 789         /* Advance the circular list */
 790         if( !c->first ) c->first = c->head;
 791         c->last = c->head;
 792         c->head = c->head->next;
 793 }
 794
 795 void pullup_flush_fields( struct pullup_context * c )
 796 {
 797         struct pullup_field * f;
 798
 799         for( f = c->first; f && f != c->head; f = f->next )
 800     {
 801                 pullup_release_buffer( f->buffer, f->parity );
 802                 f->buffer = 0;
 803         }
 804         c->first = c->last = 0;
 805 }
 806
 807 /*
 808  *
 809  * DETELECINE FILTER FUNCTIONS
 810  *
 811  */
 812
 813 hb_filter_private_t * hb_detelecine_init( int pix_fmt,
 814                                           int width,
 815                                           int height,
 816                                           char * settings )
 817 {
 818     if( pix_fmt != PIX_FMT_YUV420P )
 819     {
 820         return 0;
 821     }
 822
 823     hb_filter_private_t * pv = malloc( sizeof(struct hb_filter_private_s) );
 824
 825     pv->pix_fmt  = pix_fmt;
 826     pv->width[0]  = width;
 827     pv->height[0] = height;
 828     pv->width[1]  = pv->width[2] = width >> 1;
 829     pv->height[1] = pv->height[2] = height >> 1;
 830
 831     int buf_size = 3 * width * height / 2;
 832     pv->buf_out = hb_buffer_init( buf_size );
 833
 834     struct pullup_context * ctx;
 835     pv->pullup_ctx = ctx = pullup_alloc_context();
 836
 837     ctx->junk_left = ctx->junk_right  = 1;
 838     ctx->junk_top  = ctx->junk_bottom = 4;
 839     ctx->strict_breaks = 0;
 840     ctx->metric_plane  = 0;
 841
 842     if( settings )
 843     {
 844                 sscanf( settings, "%d:%d:%d:%d:%d:%d",
 845                 &ctx->junk_left,
 846                 &ctx->junk_right,
 847                 &ctx->junk_top,
 848                 &ctx->junk_bottom,
 849                 &ctx->strict_breaks,
 850                 &ctx->metric_plane );
 851         }
 852
 853     ctx->format = PULLUP_FMT_Y;
 854     ctx->nplanes = 4;
 855
 856     pullup_preinit_context( ctx );
 857
 858     ctx->bpp[0] = ctx->bpp[1] = ctx->bpp[2] = 8;
 859     ctx->background[1] = ctx->background[2] = 128;
 860
 861     ctx->w[0]      = pv->width[0];
 862     ctx->h[0]      = pv->height[0];
 863     ctx->stride[0] = pv->width[0];
 864
 865     ctx->w[1]      = pv->width[1];
 866     ctx->h[1]      = pv->height[1];
 867     ctx->stride[1] = pv->width[1];
 868
 869     ctx->w[2]      = pv->width[2];
 870     ctx->h[2]      = pv->height[2];
 871     ctx->stride[2] = pv->width[2];
 872
 873     ctx->w[3]      = ((width+15)/16) * ((height+15)/16);
 874     ctx->h[3]      = 2;
 875     ctx->stride[3] = ctx->w[3];
 876
 877 #if 0
 878     ctx->verbose = 1;
 879 #endif
 880
 881     pullup_init_context( ctx );
 882
 883     pv->pullup_fakecount = 1;
 884     pv->pullup_skipflag = 0;
 885
 886     return pv;
 887 }
 888
 889 void hb_detelecine_close( hb_filter_private_t * pv )
 890 {
 891     if( !pv )
 892     {
 893         return;
 894     }
 895
 896     if( pv->buf_out )
 897     {
 898         hb_buffer_close( &pv->buf_out );
 899     }
 900
 901     if( pv->pullup_ctx )
 902     {
 903         pullup_free_context( pv->pullup_ctx );
 904     }
 905
 906     free( pv );
 907 }
 908
 909 int hb_detelecine_work( const hb_buffer_t * buf_in,
 910                         hb_buffer_t ** buf_out,
 911                         int pix_fmt,
 912                         int width,
 913                         int height,
 914                         hb_filter_private_t * pv )
 915 {
 916     if( !pv ||
 917         pix_fmt != pv->pix_fmt ||
 918         width != pv->width[0] ||
 919         height != pv->height[0] )
 920     {
 921         return FILTER_FAILED;
 922     }
 923
 924         struct pullup_context * ctx = pv->pullup_ctx;
 925         struct pullup_buffer  * buf;
 926         struct pullup_frame   * frame;
 927
 928     buf = pullup_get_buffer( ctx, 2 );
 929     if( !buf )
 930     {
 931         frame = pullup_get_frame( ctx );
 932         pullup_release_frame( frame );
 933         hb_log( "Could not get buffer from pullup!" );
 934         return FILTER_FAILED;
 935     }
 936
 937     /* Copy input buffer into pullup buffer */
 938     avpicture_fill( &pv->pic_in, buf_in->data,
 939                     pix_fmt, width, height );
 940
 941     hb_buffer_copy_settings( pv->buf_out, buf_in );
 942
 943     memcpy( buf->planes[0], pv->pic_in.data[0],
 944             pv->width[0] * pv->height[0] * sizeof(uint8_t) );
 945     memcpy( buf->planes[1], pv->pic_in.data[1],
 946             pv->width[1] * pv->height[1] * sizeof(uint8_t) );
 947     memcpy( buf->planes[2], pv->pic_in.data[2],
 948             pv->width[2] * pv->height[2] * sizeof(uint8_t) );
 949
 950     /* Submit buffer fields based on buffer flags */
 951     int parity = 1;
 952     if( buf_in->flags & PIC_FLAG_TOP_FIELD_FIRST )
 953     {
 954         parity = 0;
 955     }
 956         pullup_submit_field( ctx, buf, parity );
 957         pullup_submit_field( ctx, buf, parity^1 );
 958     if( buf_in->flags & PIC_FLAG_REPEAT_FIRST_FIELD )
 959     {
 960         pullup_submit_field( ctx, buf, parity );
 961     }
 962         pullup_release_buffer( buf, 2 );
 963
 964     /* Get frame and check if pullup is ready */
 965         frame = pullup_get_frame( ctx );
 966     if( !frame )
 967     {
 968         if( pv->pullup_fakecount )
 969         {
 970             pv->pullup_fakecount--;
 971
 972             memcpy( pv->buf_out->data, buf_in->data, buf_in->size );
 973
 974             goto output_frame;
 975         }
 976         else
 977         {
 978             goto discard_frame;
 979         }
 980     }
 981
 982     /* Check to see if frame should be dropped */
 983     if( frame->length < 2 )
 984     {
 985                 pullup_release_frame( frame );
 986                 frame = pullup_get_frame( ctx );
 987
 988                 if (!frame)
 989         {
 990             goto discard_frame;
 991         }
 992                 if( frame->length < 2 )
 993         {
 994                         pullup_release_frame( frame );
 995
 996                         if( !(buf_in->flags & PIC_FLAG_REPEAT_FIRST_FIELD) )
 997             {
 998                 goto discard_frame;
 999             }
1000
1001             frame = pullup_get_frame( ctx );
1002
1003             if( !frame )
1004             {
1005                 goto discard_frame;
1006             }
1007                         if( frame->length < 2 )
1008             {
1009                                 pullup_release_frame( frame );
1010                 goto discard_frame;
1011                         }
1012                 }
1013     }
1014
1015     /* Check to see if frame buffer is ready for export */
1016     if( !frame->buffer )
1017     {
1018         pullup_pack_frame( ctx, frame );
1019     }
1020
1021     /* Copy pullup frame buffer into output buffer */
1022     avpicture_fill( &pv->pic_out, pv->buf_out->data,
1023                     pix_fmt, width, height );
1024
1025     memcpy( pv->pic_out.data[0], frame->buffer->planes[0],
1026             pv->width[0] * pv->height[0] * sizeof(uint8_t) );
1027     memcpy( pv->pic_out.data[1], frame->buffer->planes[1],
1028             pv->width[1] * pv->height[1] * sizeof(uint8_t) );
1029     memcpy( pv->pic_out.data[2], frame->buffer->planes[2],
1030             pv->width[2] * pv->height[2] * sizeof(uint8_t) );
1031
1032     pullup_release_frame( frame );
1033
1034 output_frame:
1035     *buf_out = pv->buf_out;
1036     return FILTER_OK;
1037
1038 /* This and all discard_frame calls shown above are
1039    the result of me restoring the functionality in
1040    pullup that huevos_rancheros disabled because
1041    HB couldn't handle it.                           */
1042 discard_frame:
1043     *buf_out = pv->buf_out;
1044     return FILTER_DROP;
1045
1046 }
1047
1048