mirror of https://github.com/mpv-player/mpv
156 lines
3.8 KiB
C
156 lines
3.8 KiB
C
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#include "config.h"
|
|
#include "mp_msg.h"
|
|
#include "help_mp.h"
|
|
|
|
#include "codec-cfg.h"
|
|
#include "stream/stream.h"
|
|
#include "libmpdemux/demuxer.h"
|
|
#include "libmpdemux/stheader.h"
|
|
|
|
#include "stream/stream.h"
|
|
#include "libmpdemux/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:
|
|
mp_msg(MSGT_MENCODER, MSGL_INFO, MSGTR_MPCODECS_OutputWithFourccNotSupported, 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, double pts) {
|
|
mux_v->buffer = mpi->planes[0];
|
|
muxer_write_chunk(mux_v, mpi->width*mpi->height*mux_v->bih->biBitCount/8, 0x10, pts, pts);
|
|
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
|
|
};
|
|
|
|
//===========================================================================//
|