libmpcodecs/vf_rotate.c

   1 /*
   2  * This file is part of MPlayer.
   3  *
   4  * MPlayer is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation; either version 2 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * MPlayer is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License along
  15  * with MPlayer; if not, write to the Free Software Foundation, Inc.,
  16  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  17  */
  18
  19 #include <stdio.h>
  20 #include <stdlib.h>
  21 #include <string.h>
  22 #include <inttypes.h>
  23
  24 #include "config.h"
  25 #include "mp_msg.h"
  26
  27 #include "img_format.h"
  28 #include "mp_image.h"
  29 #include "vf.h"
  30
  31 struct vf_priv_s {
  32     int direction;
  33 };
  34
  35 static void rotate(unsigned char* dst,unsigned char* src,int dststride,int srcstride,int w,int h,int bpp,int dir){
  36     int y;
  37     if(dir&1){
  38         src+=srcstride*(w-1);
  39         srcstride*=-1;
  40     }
  41     if(dir&2){
  42         dst+=dststride*(h-1);
  43         dststride*=-1;
  44     }
  45
  46     for(y=0;y<h;y++){
  47         int x;
  48         switch(bpp){
  49         case 1:
  50             for(x=0;x<w;x++) dst[x]=src[y+x*srcstride];
  51             break;
  52         case 2:
  53             for(x=0;x<w;x++) *((short*)(dst+x*2))=*((short*)(src+y*2+x*srcstride));
  54             break;
  55         case 3:
  56             for(x=0;x<w;x++){
  57                 dst[x*3+0]=src[0+y*3+x*srcstride];
  58                 dst[x*3+1]=src[1+y*3+x*srcstride];
  59                 dst[x*3+2]=src[2+y*3+x*srcstride];
  60             }
  61             break;
  62         case 4:
  63             for(x=0;x<w;x++) *((int*)(dst+x*4))=*((int*)(src+y*4+x*srcstride));
  64         }
  65         dst+=dststride;
  66     }
  67 }
  68
  69 //===========================================================================//
  70
  71 static int config(struct vf_instance *vf,
  72         int width, int height, int d_width, int d_height,
  73         unsigned int flags, unsigned int outfmt){
  74     if (vf->priv->direction & 4) {
  75         if (width<height) vf->priv->direction&=3;
  76     }
  77     if (vf->priv->direction & 4){
  78         vf->put_image=vf_next_put_image; // passthru mode!
  79         if (vf->next->draw_slice) vf->draw_slice=vf_next_draw_slice;
  80 /* FIXME: this should be in an other procedure in vf.c; that should always check
  81      whether the filter after the passthrough one still (not)supports slices */
  82         return vf_next_config(vf,width,height,d_width,d_height,flags,outfmt);
  83     }
  84     return vf_next_config(vf,height,width,d_height,d_width,flags,outfmt);
  85 }
  86
  87 static int put_image(struct vf_instance *vf, mp_image_t *mpi, double pts){
  88     mp_image_t *dmpi;
  89
  90     // hope we'll get DR buffer:
  91     dmpi=vf_get_image(vf->next,mpi->imgfmt,
  92         MP_IMGTYPE_TEMP, MP_IMGFLAG_ACCEPT_STRIDE,
  93         mpi->h, mpi->w);
  94
  95     if(mpi->flags&MP_IMGFLAG_PLANAR){
  96         rotate(dmpi->planes[0],mpi->planes[0],
  97                dmpi->stride[0],mpi->stride[0],
  98                dmpi->w,dmpi->h,1,vf->priv->direction);
  99         rotate(dmpi->planes[1],mpi->planes[1],
 100                dmpi->stride[1],mpi->stride[1],
 101                dmpi->w>>mpi->chroma_x_shift,dmpi->h>>mpi->chroma_y_shift,1,vf->priv->direction);
 102         rotate(dmpi->planes[2],mpi->planes[2],
 103                dmpi->stride[2],mpi->stride[2],
 104                dmpi->w>>mpi->chroma_x_shift,dmpi->h>>mpi->chroma_y_shift,1,vf->priv->direction);
 105     } else {
 106         rotate(dmpi->planes[0],mpi->planes[0],
 107                dmpi->stride[0],mpi->stride[0],
 108                dmpi->w,dmpi->h,dmpi->bpp>>3,vf->priv->direction);
 109         dmpi->planes[1] = mpi->planes[1]; // passthrough rgb8 palette
 110     }
 111
 112     return vf_next_put_image(vf,dmpi, pts);
 113 }
 114
 115 //===========================================================================//
 116
 117 static int query_format(struct vf_instance *vf, unsigned int fmt){
 118     if(IMGFMT_IS_RGB(fmt) || IMGFMT_IS_BGR(fmt)) return vf_next_query_format(vf, fmt);
 119     // we can support only symmetric (chroma_x_shift==chroma_y_shift) YUV formats:
 120     switch(fmt) {
 121         case IMGFMT_YV12:
 122         case IMGFMT_I420:
 123         case IMGFMT_IYUV:
 124         case IMGFMT_YVU9:
 125 //      case IMGFMT_IF09:
 126         case IMGFMT_Y8:
 127         case IMGFMT_Y800:
 128         case IMGFMT_444P:
 129             return vf_next_query_format(vf, fmt);
 130     }
 131     return 0;
 132 }
 133
 134 static int vf_open(vf_instance_t *vf, char *args){
 135     vf->config=config;
 136     vf->put_image=put_image;
 137     vf->query_format=query_format;
 138     vf->priv=malloc(sizeof(struct vf_priv_s));
 139     vf->priv->direction=args?atoi(args):0;
 140     return 1;
 141 }
 142
 143 const vf_info_t vf_info_rotate = {
 144     "rotate",
 145     "rotate",
 146     "A'rpi",
 147     "",
 148     vf_open,
 149     NULL
 150 };
 151
 152 //===========================================================================//