libmpcodecs/vf_rotate.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
   9 #include "img_format.h"
  10 #include "mp_image.h"
  11 #include "vf.h"
  12
  13 #include "libvo/fastmemcpy.h"
  14 #include "postproc/rgb2rgb.h"
  15
  16 struct vf_priv_s {
  17     int direction;
  18 };
  19
  20 static void rotate(unsigned char* dst,unsigned char* src,int dststride,int srcstride,int w,int h,int bpp,int dir){
  21     int y;
  22     if(dir&1){
  23         src+=srcstride*(w-1);
  24         srcstride*=-1;
  25     }
  26     if(dir&2){
  27         dst+=dststride*(h-1);
  28         dststride*=-1;
  29     }
  30
  31     for(y=0;y<h;y++){
  32         int x;
  33         switch(bpp){
  34         case 1:
  35             for(x=0;x<w;x++) dst[x]=src[y+x*srcstride];
  36             break;
  37         case 2:
  38             for(x=0;x<w;x++) *((short*)(dst+x*2))=*((short*)(src+y*2+x*srcstride));
  39             break;
  40         case 3:
  41             for(x=0;x<w;x++){
  42                 dst[x*3+0]=src[0+y*3+x*srcstride];
  43                 dst[x*3+1]=src[1+y*3+x*srcstride];
  44                 dst[x*3+2]=src[2+y*3+x*srcstride];
  45             }
  46             break;
  47         case 4:
  48             for(x=0;x<w;x++) *((int*)(dst+x*4))=*((int*)(src+y*4+x*srcstride));
  49         }
  50         dst+=dststride;
  51     }
  52 }
  53
  54 //===========================================================================//
  55
  56 static int config(struct vf_instance_s* vf,
  57         int width, int height, int d_width, int d_height,
  58         unsigned int flags, unsigned int outfmt){
  59     if (vf->priv->direction & 4) {
  60         if (width<height) vf->priv->direction&=3;
  61     }
  62     if (vf->priv->direction & 4){
  63         vf->put_image=vf_next_put_image; // passthru mode!
  64         if (vf->next->draw_slice) vf->draw_slice=vf_next_draw_slice;
  65 /* FIXME: this should be in an other procedure in vf.c; that should always check
  66      whether the filter after the passthrough one still (not)supports slices */
  67         return vf_next_config(vf,width,height,d_width,d_height,flags,outfmt);
  68     }
  69     return vf_next_config(vf,height,width,d_height,d_width,flags,outfmt);
  70 }
  71
  72 static int put_image(struct vf_instance_s* vf, mp_image_t *mpi, double pts){
  73     mp_image_t *dmpi;
  74
  75     // hope we'll get DR buffer:
  76     dmpi=vf_get_image(vf->next,mpi->imgfmt,
  77         MP_IMGTYPE_TEMP, MP_IMGFLAG_ACCEPT_STRIDE,
  78         mpi->h, mpi->w);
  79
  80     if(mpi->flags&MP_IMGFLAG_PLANAR){
  81         rotate(dmpi->planes[0],mpi->planes[0],
  82                dmpi->stride[0],mpi->stride[0],
  83                dmpi->w,dmpi->h,1,vf->priv->direction);
  84         rotate(dmpi->planes[1],mpi->planes[1],
  85                dmpi->stride[1],mpi->stride[1],
  86                dmpi->w>>mpi->chroma_x_shift,dmpi->h>>mpi->chroma_y_shift,1,vf->priv->direction);
  87         rotate(dmpi->planes[2],mpi->planes[2],
  88                dmpi->stride[2],mpi->stride[2],
  89                dmpi->w>>mpi->chroma_x_shift,dmpi->h>>mpi->chroma_y_shift,1,vf->priv->direction);
  90     } else {
  91         rotate(dmpi->planes[0],mpi->planes[0],
  92                dmpi->stride[0],mpi->stride[0],
  93                dmpi->w,dmpi->h,dmpi->bpp>>3,vf->priv->direction);
  94         dmpi->planes[1] = mpi->planes[1]; // passthrough rgb8 palette
  95     }
  96
  97     return vf_next_put_image(vf,dmpi, pts);
  98 }
  99
 100 //===========================================================================//
 101
 102 static int query_format(struct vf_instance_s* vf, unsigned int fmt){
 103     if(IMGFMT_IS_RGB(fmt) || IMGFMT_IS_BGR(fmt)) return vf_next_query_format(vf, fmt);
 104     // we can support only symmetric (chroma_x_shift==chroma_y_shift) YUV formats:
 105     switch(fmt) {
 106         case IMGFMT_YV12:
 107         case IMGFMT_I420:
 108         case IMGFMT_IYUV:
 109         case IMGFMT_YVU9:
 110 //      case IMGFMT_IF09:
 111         case IMGFMT_Y8:
 112         case IMGFMT_Y800:
 113         case IMGFMT_444P:
 114             return vf_next_query_format(vf, fmt);
 115     }
 116     return 0;
 117 }
 118
 119 static int open(vf_instance_t *vf, char* args){
 120     vf->config=config;
 121     vf->put_image=put_image;
 122     vf->query_format=query_format;
 123     vf->priv=malloc(sizeof(struct vf_priv_s));
 124     vf->priv->direction=args?atoi(args):0;
 125     return 1;
 126 }
 127
 128 vf_info_t vf_info_rotate = {
 129     "rotate",
 130     "rotate",
 131     "A'rpi",
 132     "",
 133     open,
 134     NULL
 135 };
 136
 137 //===========================================================================//