libmpcodecs/vf_yvu9.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 //===========================================================================//
  17
  18 static int config(struct vf_instance_s* vf,
  19         int width, int height, int d_width, int d_height,
  20         unsigned int flags, unsigned int outfmt){
  21
  22     if(vf_next_query_format(vf,IMGFMT_YV12)<=0){
  23         mp_msg(MSGT_VFILTER, MSGL_WARN, MSGTR_MPCODECS_WarnNextFilterDoesntSupport, "YVU9");
  24         return 0;
  25     }
  26
  27     return vf_next_config(vf,width,height,d_width,d_height,flags,IMGFMT_YV12);
  28 }
  29
  30 static int put_image(struct vf_instance_s* vf, mp_image_t *mpi, double pts){
  31     mp_image_t *dmpi;
  32     int y,w,h;
  33
  34     // hope we'll get DR buffer:
  35     dmpi=vf_get_image(vf->next,IMGFMT_YV12,
  36         MP_IMGTYPE_TEMP, 0/*MP_IMGFLAG_ACCEPT_STRIDE*/,
  37         mpi->w, mpi->h);
  38
  39     for(y=0;y<mpi->h;y++)
  40         fast_memcpy(dmpi->planes[0]+dmpi->stride[0]*y,
  41                mpi->planes[0]+mpi->stride[0]*y,
  42                mpi->w);
  43
  44     w=mpi->w/4; h=mpi->h/2;
  45     for(y=0;y<h;y++){
  46         unsigned char* s=mpi->planes[1]+mpi->stride[1]*(y>>1);
  47         unsigned char* d=dmpi->planes[1]+dmpi->stride[1]*y;
  48         int x;
  49         for(x=0;x<w;x++) d[2*x]=d[2*x+1]=s[x];
  50     }
  51     for(y=0;y<h;y++){
  52         unsigned char* s=mpi->planes[2]+mpi->stride[2]*(y>>1);
  53         unsigned char* d=dmpi->planes[2]+dmpi->stride[2]*y;
  54         int x;
  55         for(x=0;x<w;x++) d[2*x]=d[2*x+1]=s[x];
  56     }
  57
  58     vf_clone_mpi_attributes(dmpi, mpi);
  59
  60     return vf_next_put_image(vf,dmpi, pts);
  61 }
  62
  63 //===========================================================================//
  64
  65 static int query_format(struct vf_instance_s* vf, unsigned int fmt){
  66     if (fmt == IMGFMT_YVU9 || fmt == IMGFMT_IF09)
  67         return vf_next_query_format(vf,IMGFMT_YV12) & (~VFCAP_CSP_SUPPORTED_BY_HW);
  68     return 0;
  69 }
  70
  71 static int open(vf_instance_t *vf, char* args){
  72     vf->config=config;
  73     vf->put_image=put_image;
  74     vf->query_format=query_format;
  75     return 1;
  76 }
  77
  78 const vf_info_t vf_info_yvu9 = {
  79     "fast YVU9->YV12 conversion",
  80     "yvu9",
  81     "alex",
  82     "",
  83     open,
  84     NULL
  85 };
  86
  87 //===========================================================================//