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