mpv/libmpcodecs/ve_raw.c

154 lines
3.7 KiB
C

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "config.h"
#include "mp_msg.h"
#include "codec-cfg.h"
#include "stream.h"
#include "demuxer.h"
#include "stheader.h"
#include "muxer.h"
#include "img_format.h"
#include "mp_image.h"
#include "vf.h"
//===========================================================================//
struct vf_priv_s {
muxer_stream_t* mux;
};
#define mux_v (vf->priv->mux)
static int set_format(struct vf_instance_s *vf, unsigned int fmt) {
mux_v->bih->biCompression = fmt;
mux_v->bih->biPlanes = 1;
if (IMGFMT_IS_RGB(fmt)) {
if (IMGFMT_RGB_DEPTH(fmt) < 8 && !(fmt&128))
mux_v->bih->biBitCount = IMGFMT_RGB_DEPTH(fmt);
else
mux_v->bih->biBitCount = (IMGFMT_RGB_DEPTH(fmt)+7)&(~7);
return 1;
}
if (IMGFMT_IS_BGR(fmt)) {
if (IMGFMT_BGR_DEPTH(fmt) < 8 && !(fmt&128))
mux_v->bih->biBitCount = IMGFMT_BGR_DEPTH(fmt);
else
mux_v->bih->biBitCount = (IMGFMT_BGR_DEPTH(fmt)+7)&(~7);
return 1;
}
switch (fmt) {
case IMGFMT_I420:
case IMGFMT_IYUV:
case IMGFMT_YV12:
case IMGFMT_411P:
mux_v->bih->biPlanes = 3;
mux_v->bih->biBitCount = 12;
break;
case IMGFMT_444P:
mux_v->bih->biPlanes = 3;
mux_v->bih->biBitCount = 24;
break;
case IMGFMT_422P:
mux_v->bih->biPlanes = 3;
mux_v->bih->biBitCount = 16;
break;
case IMGFMT_IF09:
mux_v->bih->biPlanes = 4;
case IMGFMT_YVU9:
mux_v->bih->biBitCount = 9;
break;
case IMGFMT_UYVY:
case IMGFMT_YUY2:
mux_v->bih->biBitCount = 16;
break;
default:
printf("ve_raw: raw output with fourcc [%x] not supported!\n", fmt);
mux_v->bih->biCompression = 0;
return 0;
}
return 1;
}
static int config(struct vf_instance_s *vf,
int width, int height, int d_width, int d_height,
unsigned int flags, unsigned int outfmt)
{
int ret;
mux_v->bih->biWidth = width;
mux_v->bih->biHeight = height;
mux_v->aspect = (float)d_width/d_height;
ret = set_format(vf, outfmt);
if (!ret) return 0;
mux_v->bih->biSizeImage = mux_v->bih->biWidth*mux_v->bih->biHeight*mux_v->bih->biBitCount/8;
return 1;
}
static int control(struct vf_instance_s *vf, int request, void *data) {
return CONTROL_UNKNOWN;
}
static int query_format(struct vf_instance_s *vf, unsigned int fmt) {
if (IMGFMT_IS_RGB(fmt) || IMGFMT_IS_BGR(fmt))
return VFCAP_CSP_SUPPORTED | VFCAP_CSP_SUPPORTED_BY_HW;
switch (fmt) {
case IMGFMT_I420:
case IMGFMT_IYUV:
case IMGFMT_YV12:
case IMGFMT_411P:
case IMGFMT_444P:
case IMGFMT_422P:
case IMGFMT_UYVY:
case IMGFMT_YUY2:
case IMGFMT_YVU9:
case IMGFMT_IF09:
return VFCAP_CSP_SUPPORTED | VFCAP_CSP_SUPPORTED_BY_HW;
}
return 0;
}
static int put_image(struct vf_instance_s *vf, mp_image_t *mpi) {
mux_v->buffer = mpi->planes[0];
muxer_write_chunk(mux_v, mpi->width*mpi->height*mux_v->bih->biBitCount/8, 0x10);
return 1;
}
//===========================================================================//
static int vf_open(vf_instance_t *vf, char* args){
vf->config = config;
vf->default_caps = VFCAP_CONSTANT;
vf->control = control;
vf->query_format = query_format;
vf->put_image = put_image;
vf->default_caps = 0;
vf->priv = malloc(sizeof(struct vf_priv_s));
memset(vf->priv, 0, sizeof(struct vf_priv_s));
vf->priv->mux = (muxer_stream_t*)args;
mux_v->bih = calloc(1, sizeof(BITMAPINFOHEADER));
mux_v->bih->biSize = sizeof(BITMAPINFOHEADER);
mux_v->bih->biWidth = 0;
mux_v->bih->biHeight = 0;
return 1;
}
vf_info_t ve_info_raw = {
"raw encoder",
"raw",
"jwe21@cam.ac.uk",
"Based on rawrgb",
vf_open
};
//===========================================================================//