libmpcodecs/vf_cropdetect.c

   1 #include <stdio.h>
   2 #include <stdlib.h>
   3 #include <string.h>
   4 #include <inttypes.h>
   5
   6 #include "config.h"
   7 #include "mp_msg.h"
   8 #include "help_mp.h"
   9
  10 #include "img_format.h"
  11 #include "mp_image.h"
  12 #include "vf.h"
  13
  14 #include "libvo/fastmemcpy.h"
  15
  16 struct vf_priv_s {
  17     int x1,y1,x2,y2;
  18     int limit;
  19     int round;
  20     int fno;
  21 };
  22
  23 static int checkline(unsigned char* src,int stride,int len,int bpp){
  24     int total=0;
  25     int div=len;
  26     switch(bpp){
  27     case 1:
  28         while(--len>=0){
  29             total+=src[0]; src+=stride;
  30         }
  31         break;
  32     case 3:
  33     case 4:
  34         while(--len>=0){
  35             total+=src[0]+src[1]+src[2]; src+=stride;
  36         }
  37         div*=3;
  38         break;
  39     }
  40     total/=div;
  41 //    printf("total=%d\n",total);
  42     return total;
  43 }
  44
  45 //===========================================================================//
  46
  47 static int config(struct vf_instance_s* vf,
  48         int width, int height, int d_width, int d_height,
  49         unsigned int flags, unsigned int outfmt){
  50     vf->priv->x1=width - 1;
  51     vf->priv->y1=height - 1;
  52     vf->priv->x2=0;
  53     vf->priv->y2=0;
  54     vf->priv->fno=0;
  55     return vf_next_config(vf,width,height,d_width,d_height,flags,outfmt);
  56 }
  57
  58 static int put_image(struct vf_instance_s* vf, mp_image_t *mpi, double pts){
  59     mp_image_t *dmpi;
  60     int bpp=mpi->bpp/8;
  61     int w,h,x,y,shrink_by;
  62
  63     // hope we'll get DR buffer:
  64     dmpi=vf_get_image(vf->next,mpi->imgfmt,
  65         MP_IMGTYPE_EXPORT, 0,
  66         mpi->w, mpi->h);
  67
  68     dmpi->planes[0]=mpi->planes[0];
  69     dmpi->planes[1]=mpi->planes[1];
  70     dmpi->planes[2]=mpi->planes[2];
  71     dmpi->stride[0]=mpi->stride[0];
  72     dmpi->stride[1]=mpi->stride[1];
  73     dmpi->stride[2]=mpi->stride[2];
  74     dmpi->width=mpi->width;
  75     dmpi->height=mpi->height;
  76
  77 if(++vf->priv->fno>2){  // ignore first 2 frames - they may be empty
  78
  79     for(y=0;y<vf->priv->y1;y++){
  80         if(checkline(mpi->planes[0]+mpi->stride[0]*y,bpp,mpi->w,bpp)>vf->priv->limit){
  81             vf->priv->y1=y;
  82             break;
  83         }
  84     }
  85
  86     for(y=mpi->h-1;y>vf->priv->y2;y--){
  87         if(checkline(mpi->planes[0]+mpi->stride[0]*y,bpp,mpi->w,bpp)>vf->priv->limit){
  88             vf->priv->y2=y;
  89             break;
  90         }
  91     }
  92
  93     for(y=0;y<vf->priv->x1;y++){
  94         if(checkline(mpi->planes[0]+bpp*y,mpi->stride[0],mpi->h,bpp)>vf->priv->limit){
  95             vf->priv->x1=y;
  96             break;
  97         }
  98     }
  99
 100     for(y=mpi->w-1;y>vf->priv->x2;y--){
 101         if(checkline(mpi->planes[0]+bpp*y,mpi->stride[0],mpi->h,bpp)>vf->priv->limit){
 102             vf->priv->x2=y;
 103             break;
 104         }
 105     }
 106
 107     // round x and y (up), important for yuv colorspaces
 108     // make sure they stay rounded!
 109     x=(vf->priv->x1+1)&(~1);
 110     y=(vf->priv->y1+1)&(~1);
 111
 112     w = vf->priv->x2 - x + 1;
 113     h = vf->priv->y2 - y + 1;
 114
 115     // w and h must be divisible by 2 as well because of yuv
 116     // colorspace problems.
 117     if (vf->priv->round <= 1)
 118       vf->priv->round = 16;
 119     if (vf->priv->round % 2)
 120       vf->priv->round *= 2;
 121
 122     shrink_by = w % vf->priv->round;
 123     w -= shrink_by;
 124     x += (shrink_by / 2 + 1) & ~1;
 125
 126     shrink_by = h % vf->priv->round;
 127     h -= shrink_by;
 128     y += (shrink_by / 2 + 1) & ~1;
 129
 130     mp_msg(MSGT_VFILTER, MSGL_INFO, MSGTR_MPCODECS_CropArea,
 131         vf->priv->x1,vf->priv->x2,
 132         vf->priv->y1,vf->priv->y2,
 133         w,h,x,y);
 134
 135
 136 }
 137
 138     return vf_next_put_image(vf,dmpi, pts);
 139 }
 140
 141 static int query_format(struct vf_instance_s* vf, unsigned int fmt) {
 142   switch(fmt) {
 143     // the default limit value works only right with YV12 right now.
 144     case IMGFMT_YV12:
 145       return vf_next_query_format(vf, fmt);
 146   }
 147   return 0;
 148 }
 149 //===========================================================================//
 150
 151 static int open(vf_instance_t *vf, char* args){
 152     vf->config=config;
 153     vf->put_image=put_image;
 154     vf->query_format=query_format;
 155     vf->priv=malloc(sizeof(struct vf_priv_s));
 156     vf->priv->limit=24; // should be option
 157     vf->priv->round = 0;
 158     if(args) sscanf(args, "%d:%d",
 159     &vf->priv->limit,
 160     &vf->priv->round);
 161     return 1;
 162 }
 163
 164 vf_info_t vf_info_cropdetect = {
 165     "autodetect crop size",
 166     "cropdetect",
 167     "A'rpi",
 168     "",
 169     open,
 170     NULL
 171 };
 172
 173 //===========================================================================//