libmpcodecs/vf_crop.c

   1 #include <stdio.h>
   2 #include <stdlib.h>
   3 #include <string.h>
   4
   5 #include "config.h"
   6 #include "mp_msg.h"
   7 #include "help_mp.h"
   8 #include "options.h"
   9
  10 #include "img_format.h"
  11 #include "mp_image.h"
  12 #include "vf.h"
  13
  14 #include "m_option.h"
  15 #include "m_struct.h"
  16
  17 static const struct vf_priv_s {
  18     int crop_w,crop_h;
  19     int crop_x,crop_y;
  20 } vf_priv_dflt = {
  21   -1,-1,
  22   -1,-1
  23 };
  24
  25 //===========================================================================//
  26
  27 static int config(struct vf_instance* vf,
  28         int width, int height, int d_width, int d_height,
  29         unsigned int flags, unsigned int outfmt){
  30     struct MPOpts *opts = vf->opts;
  31     // calculate the missing parameters:
  32     if(vf->priv->crop_w<=0 || vf->priv->crop_w>width) vf->priv->crop_w=width;
  33     if(vf->priv->crop_h<=0 || vf->priv->crop_h>height) vf->priv->crop_h=height;
  34     if(vf->priv->crop_x<0) vf->priv->crop_x=(width-vf->priv->crop_w)/2;
  35     if(vf->priv->crop_y<0) vf->priv->crop_y=(height-vf->priv->crop_h)/2;
  36     // rounding:
  37     if(!IMGFMT_IS_RGB(outfmt) && !IMGFMT_IS_BGR(outfmt)){
  38         switch(outfmt){
  39         case IMGFMT_444P:
  40         case IMGFMT_Y800:
  41         case IMGFMT_Y8:
  42             break;
  43         case IMGFMT_YVU9:
  44         case IMGFMT_IF09:
  45             vf->priv->crop_y&=~3;
  46         case IMGFMT_411P:
  47             vf->priv->crop_x&=~3;
  48             break;
  49         case IMGFMT_YV12:
  50         case IMGFMT_I420:
  51         case IMGFMT_IYUV:
  52             vf->priv->crop_y&=~1;
  53         default:
  54             vf->priv->crop_x&=~1;
  55         }
  56     }
  57     // check:
  58     if(vf->priv->crop_w+vf->priv->crop_x>width ||
  59        vf->priv->crop_h+vf->priv->crop_y>height){
  60         mp_msg(MSGT_VFILTER, MSGL_WARN, MSGTR_MPCODECS_CropBadPositionWidthHeight);
  61         return 0;
  62     }
  63     if(!opts->screen_size_x && !opts->screen_size_y){
  64         d_width=d_width*vf->priv->crop_w/width;
  65         d_height=d_height*vf->priv->crop_h/height;
  66     }
  67     return vf_next_config(vf,vf->priv->crop_w,vf->priv->crop_h,d_width,d_height,flags,outfmt);
  68 }
  69
  70 static int put_image(struct vf_instance* vf, mp_image_t *mpi, double pts){
  71     mp_image_t *dmpi;
  72     if (mpi->flags&MP_IMGFLAG_DRAW_CALLBACK)
  73         return vf_next_put_image(vf,vf->dmpi, pts);
  74     dmpi=vf_get_image(vf->next,mpi->imgfmt,
  75         MP_IMGTYPE_EXPORT, 0,
  76         vf->priv->crop_w, vf->priv->crop_h);
  77     if(mpi->flags&MP_IMGFLAG_PLANAR){
  78         dmpi->planes[0]=mpi->planes[0]+
  79             vf->priv->crop_y*mpi->stride[0]+vf->priv->crop_x;
  80         dmpi->planes[1]=mpi->planes[1]+
  81             (vf->priv->crop_y>>mpi->chroma_y_shift)*mpi->stride[1]+(vf->priv->crop_x>>mpi->chroma_x_shift);
  82         dmpi->planes[2]=mpi->planes[2]+
  83             (vf->priv->crop_y>>mpi->chroma_y_shift)*mpi->stride[2]+(vf->priv->crop_x>>mpi->chroma_x_shift);
  84         dmpi->stride[1]=mpi->stride[1];
  85         dmpi->stride[2]=mpi->stride[2];
  86     } else {
  87         dmpi->planes[0]=mpi->planes[0]+
  88             vf->priv->crop_y*mpi->stride[0]+
  89             vf->priv->crop_x*(mpi->bpp/8);
  90         dmpi->planes[1]=mpi->planes[1]; // passthrough rgb8 palette
  91     }
  92     dmpi->stride[0]=mpi->stride[0];
  93     dmpi->width=mpi->width;
  94     return vf_next_put_image(vf,dmpi, pts);
  95 }
  96
  97 static void start_slice(struct vf_instance* vf, mp_image_t *mpi){
  98     vf->dmpi = vf_get_image(vf->next, mpi->imgfmt, mpi->type, mpi->flags,
  99         vf->priv->crop_w, vf->priv->crop_h);
 100 }
 101
 102 static void draw_slice(struct vf_instance* vf,
 103         unsigned char** src, int* stride, int w,int h, int x, int y){
 104     unsigned char *src2[3];
 105     src2[0] = src[0];
 106     if (vf->dmpi->flags & MP_IMGFLAG_PLANAR) {
 107             src2[1] = src[1];
 108             src2[2] = src[2];
 109     }
 110     //mp_msg(MSGT_VFILTER, MSGL_V, "crop slice %d %d %d %d ->", w,h,x,y);
 111     if ((x -= vf->priv->crop_x) < 0) {
 112         x = -x;
 113         src2[0] += x;
 114         if (vf->dmpi->flags & MP_IMGFLAG_PLANAR) {
 115                 src2[1] += x>>vf->dmpi->chroma_x_shift;
 116                 src2[2] += x>>vf->dmpi->chroma_x_shift;
 117         }
 118         w -= x;
 119         x = 0;
 120     }
 121     if ((y -= vf->priv->crop_y) < 0) {
 122         y = -y;
 123         src2[0] += y*stride[0];
 124         if (vf->dmpi->flags & MP_IMGFLAG_PLANAR) {
 125                 src2[1] += (y>>vf->dmpi->chroma_y_shift)*stride[1];
 126                 src2[2] += (y>>vf->dmpi->chroma_y_shift)*stride[2];
 127         }
 128         h -= y;
 129         y = 0;
 130     }
 131     if (x+w > vf->priv->crop_w) w = vf->priv->crop_w-x;
 132     if (y+h > vf->priv->crop_h) h = vf->priv->crop_h-y;
 133     //mp_msg(MSGT_VFILTER, MSGL_V, "%d %d %d %d\n", w,h,x,y);
 134     if ((w < 0) || (h < 0)) return;
 135     vf_next_draw_slice(vf,src2,stride,w,h,x,y);
 136 }
 137
 138 //===========================================================================//
 139
 140 static int open(vf_instance_t *vf, char* args){
 141     vf->config=config;
 142     vf->put_image=put_image;
 143     vf->start_slice=start_slice;
 144     vf->draw_slice=draw_slice;
 145     vf->default_reqs=VFCAP_ACCEPT_STRIDE;
 146     if(!vf->priv) {
 147     vf->priv=malloc(sizeof(struct vf_priv_s));
 148     // TODO: parse args ->
 149     vf->priv->crop_x=
 150     vf->priv->crop_y=
 151     vf->priv->crop_w=
 152     vf->priv->crop_h=-1;
 153     } //if(!vf->priv)
 154     if(args) sscanf(args, "%d:%d:%d:%d",
 155     &vf->priv->crop_w,
 156     &vf->priv->crop_h,
 157     &vf->priv->crop_x,
 158     &vf->priv->crop_y);
 159     mp_msg(MSGT_VFILTER, MSGL_INFO, "Crop: %d x %d, %d ; %d\n",
 160     vf->priv->crop_w,
 161     vf->priv->crop_h,
 162     vf->priv->crop_x,
 163     vf->priv->crop_y);
 164     return 1;
 165 }
 166
 167 #define ST_OFF(f) M_ST_OFF(struct vf_priv_s,f)
 168 static const m_option_t vf_opts_fields[] = {
 169   {"w", ST_OFF(crop_w), CONF_TYPE_INT, M_OPT_MIN,0 ,0, NULL},
 170   {"h", ST_OFF(crop_h), CONF_TYPE_INT, M_OPT_MIN,0 ,0, NULL},
 171   {"x", ST_OFF(crop_x), CONF_TYPE_INT, M_OPT_MIN,-1 ,0, NULL},
 172   {"y", ST_OFF(crop_y), CONF_TYPE_INT, M_OPT_MIN,-1 ,0, NULL},
 173   { NULL, NULL, 0, 0, 0, 0,  NULL }
 174 };
 175
 176 static const m_struct_t vf_opts = {
 177   "crop",
 178   sizeof(struct vf_priv_s),
 179   &vf_priv_dflt,
 180   vf_opts_fields
 181 };
 182
 183 const vf_info_t vf_info_crop = {
 184     "cropping",
 185     "crop",
 186     "A'rpi",
 187     "",
 188     open,
 189     &vf_opts
 190 };
 191
 192 //===========================================================================//