mirror of https://github.com/mpv-player/mpv
494 lines
18 KiB
C
494 lines
18 KiB
C
/*
|
|
* Copyright (C) 2010 Gordon Schmidt <gordon.schmidt <at> s2000.tu-chemnitz.de>
|
|
*
|
|
* This file is part of MPlayer.
|
|
*
|
|
* MPlayer is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* MPlayer is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along
|
|
* with MPlayer; if not, write to the Free Software Foundation, Inc.,
|
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*/
|
|
|
|
//==includes==//
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#include "config.h"
|
|
#include "mp_msg.h"
|
|
#include "options.h"
|
|
|
|
#include "img_format.h"
|
|
#include "mp_image.h"
|
|
#include "vf.h"
|
|
#include "m_struct.h"
|
|
#include "m_option.h"
|
|
|
|
#include "libavutil/common.h"
|
|
#include "libvo/fastmemcpy.h"
|
|
|
|
//==types==//
|
|
typedef enum stereo_code {
|
|
ANAGLYPH_RC_GRAY, //anaglyph red/cyan gray
|
|
ANAGLYPH_RC_HALF, //anaglyph red/cyan half colored
|
|
ANAGLYPH_RC_COLOR, //anaglyph red/cyan colored
|
|
ANAGLYPH_RC_DUBOIS, //anaglyph red/cyan dubois
|
|
ANAGLYPH_GM_GRAY, //anaglyph green/magenta gray
|
|
ANAGLYPH_GM_HALF, //anaglyph green/magenta half colored
|
|
ANAGLYPH_GM_COLOR, //anaglyph green/magenta colored
|
|
ANAGLYPH_YB_GRAY, //anaglyph yellow/blue gray
|
|
ANAGLYPH_YB_HALF, //anaglyph yellow/blue half colored
|
|
ANAGLYPH_YB_COLOR, //anaglyph yellow/blue colored
|
|
MONO_L, //mono output for debugging (left eye only)
|
|
MONO_R, //mono output for debugging (right eye only)
|
|
SIDE_BY_SIDE_LR, //side by side parallel (left eye left, right eye right)
|
|
SIDE_BY_SIDE_RL, //side by side crosseye (right eye left, left eye right)
|
|
ABOVE_BELOW_LR, //above-below (left eye above, right eye below)
|
|
ABOVE_BELOW_RL, //above-below (right eye above, left eye below)
|
|
ABOVE_BELOW_2_LR, //above-below with half height resolution
|
|
ABOVE_BELOW_2_RL, //above-below with half height resolution
|
|
INTERLEAVE_ROWS_LR, //row-interleave (left eye has top row)
|
|
INTERLEAVE_ROWS_RL, //row-interleave (right eye has top row)
|
|
STEREO_CODE_COUNT //no value set - TODO: needs autodetection
|
|
} stereo_code;
|
|
|
|
typedef struct component {
|
|
stereo_code fmt;
|
|
unsigned int width;
|
|
unsigned int height;
|
|
unsigned int off_left;
|
|
unsigned int off_right;
|
|
unsigned int row_left;
|
|
unsigned int row_right;
|
|
} component;
|
|
|
|
//==global variables==//
|
|
static const int ana_coeff[10][3][6] = {
|
|
{{19595, 38470, 7471, 0, 0, 0}, //ANAGLYPH_RC_GRAY
|
|
{ 0, 0, 0, 19595, 38470, 7471},
|
|
{ 0, 0, 0, 19595, 38470, 7471}},
|
|
{{19595, 38470, 7471, 0, 0, 0}, //ANAGLYPH_RC_HALF
|
|
{ 0, 0, 0, 0, 65536, 0},
|
|
{ 0, 0, 0, 0, 0, 65536}},
|
|
{{65536, 0, 0, 0, 0, 0}, //ANAGLYPH_RC_COLOR
|
|
{ 0, 0, 0, 0, 65536, 0},
|
|
{ 0, 0, 0, 0, 0, 65536}},
|
|
{{29891, 32800, 11559, -2849, -5763, -102}, //ANAGLYPH_RC_DUBOIS
|
|
{-2627, -2479, -1033, 24804, 48080, -1209},
|
|
{ -997, -1350, -358, -4729, -7403, 80373}},
|
|
{{ 0, 0, 0, 19595, 38470, 7471}, //ANAGLYPH_GM_GRAY
|
|
{19595, 38470, 7471, 0, 0, 0},
|
|
{ 0, 0, 0, 19595, 38470, 7471}},
|
|
{{ 0, 0, 0, 65536, 0, 0}, //ANAGLYPH_GM_HALF
|
|
{19595, 38470, 7471, 0, 0, 0},
|
|
{ 0, 0, 0, 0, 0, 65536}},
|
|
{{ 0, 0, 0, 65536, 0, 0}, //ANAGLYPH_GM_COLOR
|
|
{ 0, 65536, 0, 0, 0, 0},
|
|
{ 0, 0, 0, 0, 0, 65536}},
|
|
{{ 0, 0, 0, 19595, 38470, 7471}, //ANAGLYPH_YB_GRAY
|
|
{ 0, 0, 0, 19595, 38470, 7471},
|
|
{19595, 38470, 7471, 0, 0, 0}},
|
|
{{ 0, 0, 0, 65536, 0, 0}, //ANAGLYPH_YB_HALF
|
|
{ 0, 0, 0, 0, 65536, 0},
|
|
{19595, 38470, 7471, 0, 0, 0}},
|
|
{{ 0, 0, 0, 65536, 0, 0}, //ANAGLYPH_YB_COLOR
|
|
{ 0, 0, 0, 0, 65536, 0},
|
|
{ 0, 0, 65536, 0, 0, 0}}
|
|
};
|
|
|
|
struct vf_priv_s {
|
|
component in;
|
|
component out;
|
|
int ana_matrix[3][6];
|
|
unsigned int width;
|
|
unsigned int height;
|
|
unsigned int row_step;
|
|
} const vf_priv_default = {
|
|
{SIDE_BY_SIDE_LR},
|
|
{ANAGLYPH_RC_DUBOIS}
|
|
};
|
|
|
|
//==functions==//
|
|
static inline uint8_t ana_convert(int coeff[6], uint8_t left[3], uint8_t right[3])
|
|
{
|
|
int sum;
|
|
|
|
sum = coeff[0] * left[0] + coeff[3] * right[0]; //red in
|
|
sum += coeff[1] * left[1] + coeff[4] * right[1]; //green in
|
|
sum += coeff[2] * left[2] + coeff[5] * right[2]; //blue in
|
|
return av_clip_uint8(sum >> 16);
|
|
}
|
|
|
|
static int config(struct vf_instance *vf, int width, int height, int d_width,
|
|
int d_height, unsigned int flags, unsigned int outfmt)
|
|
{
|
|
struct MPOpts *opts = vf->opts;
|
|
|
|
if ((width & 1) || (height & 1)) {
|
|
mp_msg(MSGT_VFILTER, MSGL_WARN, "[stereo3d] invalid height or width\n");
|
|
return 0;
|
|
}
|
|
//default input values
|
|
vf->priv->width = width;
|
|
vf->priv->height = height;
|
|
vf->priv->row_step = 1;
|
|
vf->priv->in.width = width;
|
|
vf->priv->in.height = height;
|
|
vf->priv->in.off_left = 0;
|
|
vf->priv->in.off_right = 0;
|
|
vf->priv->in.row_left = 0;
|
|
vf->priv->in.row_right = 0;
|
|
|
|
//check input format
|
|
switch (vf->priv->in.fmt) {
|
|
case SIDE_BY_SIDE_LR:
|
|
vf->priv->width = width / 2;
|
|
vf->priv->in.off_right = vf->priv->width * 3;
|
|
break;
|
|
case SIDE_BY_SIDE_RL:
|
|
vf->priv->width = width / 2;
|
|
vf->priv->in.off_left = vf->priv->width * 3;
|
|
break;
|
|
case ABOVE_BELOW_2_LR:
|
|
d_height *= 2;
|
|
case ABOVE_BELOW_LR:
|
|
vf->priv->height = height / 2;
|
|
vf->priv->in.row_right = vf->priv->height;
|
|
break;
|
|
case ABOVE_BELOW_2_RL:
|
|
d_height *= 2;
|
|
case ABOVE_BELOW_RL:
|
|
vf->priv->height = height / 2;
|
|
vf->priv->in.row_left = vf->priv->height;
|
|
break;
|
|
default:
|
|
mp_msg(MSGT_VFILTER, MSGL_WARN,
|
|
"[stereo3d] stereo format of input is not supported\n");
|
|
return 0;
|
|
break;
|
|
}
|
|
//default output values
|
|
vf->priv->out.width = vf->priv->width;
|
|
vf->priv->out.height = vf->priv->height;
|
|
vf->priv->out.off_left = 0;
|
|
vf->priv->out.off_right = 0;
|
|
vf->priv->out.row_left = 0;
|
|
vf->priv->out.row_right = 0;
|
|
|
|
//check output format
|
|
switch (vf->priv->out.fmt) {
|
|
case ANAGLYPH_RC_GRAY:
|
|
case ANAGLYPH_RC_HALF:
|
|
case ANAGLYPH_RC_COLOR:
|
|
case ANAGLYPH_RC_DUBOIS:
|
|
case ANAGLYPH_GM_GRAY:
|
|
case ANAGLYPH_GM_HALF:
|
|
case ANAGLYPH_GM_COLOR:
|
|
case ANAGLYPH_YB_GRAY:
|
|
case ANAGLYPH_YB_HALF:
|
|
case ANAGLYPH_YB_COLOR:
|
|
memcpy(vf->priv->ana_matrix, ana_coeff[vf->priv->out.fmt],
|
|
sizeof(vf->priv->ana_matrix));
|
|
break;
|
|
case SIDE_BY_SIDE_LR:
|
|
vf->priv->out.width = vf->priv->width * 2;
|
|
vf->priv->out.off_right = vf->priv->width * 3;
|
|
break;
|
|
case SIDE_BY_SIDE_RL:
|
|
vf->priv->out.width = vf->priv->width * 2;
|
|
vf->priv->out.off_left = vf->priv->width * 3;
|
|
break;
|
|
case ABOVE_BELOW_2_LR:
|
|
d_height /= 2;
|
|
case ABOVE_BELOW_LR:
|
|
vf->priv->out.height = vf->priv->height * 2;
|
|
vf->priv->out.row_right = vf->priv->height;
|
|
break;
|
|
case ABOVE_BELOW_2_RL:
|
|
d_height /= 2;
|
|
case ABOVE_BELOW_RL:
|
|
vf->priv->out.height = vf->priv->height * 2;
|
|
vf->priv->out.row_left = vf->priv->height;
|
|
break;
|
|
case INTERLEAVE_ROWS_LR:
|
|
vf->priv->row_step = 2;
|
|
vf->priv->height = vf->priv->height / 2;
|
|
vf->priv->out.off_right = vf->priv->width * 3;
|
|
vf->priv->in.off_right += vf->priv->in.width * 3;
|
|
break;
|
|
case INTERLEAVE_ROWS_RL:
|
|
vf->priv->row_step = 2;
|
|
vf->priv->height = vf->priv->height / 2;
|
|
vf->priv->out.off_left = vf->priv->width * 3;
|
|
vf->priv->in.off_left += vf->priv->in.width * 3;
|
|
break;
|
|
case MONO_R:
|
|
//same as MONO_L only needs switching of input offsets
|
|
vf->priv->in.off_left = vf->priv->in.off_right;
|
|
vf->priv->in.row_left = vf->priv->in.row_right;
|
|
//nobreak;
|
|
case MONO_L:
|
|
//use default settings
|
|
break;
|
|
default:
|
|
mp_msg(MSGT_VFILTER, MSGL_WARN,
|
|
"[stereo3d] stereo format of output is not supported\n");
|
|
return 0;
|
|
break;
|
|
}
|
|
if (!opts->screen_size_x && !opts->screen_size_y) {
|
|
d_width = d_width * vf->priv->out.width / width;
|
|
d_height = d_height * vf->priv->out.height / height;
|
|
}
|
|
return vf_next_config(vf, vf->priv->out.width, vf->priv->out.height,
|
|
d_width, d_height, flags, outfmt);
|
|
}
|
|
|
|
static int put_image(struct vf_instance *vf, mp_image_t *mpi, double pts)
|
|
{
|
|
mp_image_t *dmpi;
|
|
if (vf->priv->in.fmt == vf->priv->out.fmt) { //nothing to do
|
|
dmpi = mpi;
|
|
} else {
|
|
int out_off_left, out_off_right;
|
|
int in_off_left = vf->priv->in.row_left * mpi->stride[0] +
|
|
vf->priv->in.off_left;
|
|
int in_off_right = vf->priv->in.row_right * mpi->stride[0] +
|
|
vf->priv->in.off_right;
|
|
|
|
dmpi = vf_get_image(vf->next, IMGFMT_RGB24, MP_IMGTYPE_TEMP,
|
|
MP_IMGFLAG_ACCEPT_STRIDE,
|
|
vf->priv->out.width, vf->priv->out.height);
|
|
out_off_left = vf->priv->out.row_left * dmpi->stride[0] +
|
|
vf->priv->out.off_left;
|
|
out_off_right = vf->priv->out.row_right * dmpi->stride[0] +
|
|
vf->priv->out.off_right;
|
|
|
|
switch (vf->priv->out.fmt) {
|
|
case SIDE_BY_SIDE_LR:
|
|
case SIDE_BY_SIDE_RL:
|
|
case ABOVE_BELOW_LR:
|
|
case ABOVE_BELOW_RL:
|
|
case ABOVE_BELOW_2_LR:
|
|
case ABOVE_BELOW_2_RL:
|
|
case INTERLEAVE_ROWS_LR:
|
|
case INTERLEAVE_ROWS_RL:
|
|
memcpy_pic2(dmpi->planes[0] + out_off_left,
|
|
mpi->planes[0] + in_off_left,
|
|
3 * vf->priv->width,
|
|
vf->priv->height,
|
|
dmpi->stride[0] * vf->priv->row_step,
|
|
mpi->stride[0] * vf->priv->row_step,
|
|
vf->priv->row_step != 1);
|
|
memcpy_pic2(dmpi->planes[0] + out_off_right,
|
|
mpi->planes[0] + in_off_right,
|
|
3 * vf->priv->width,
|
|
vf->priv->height,
|
|
dmpi->stride[0] * vf->priv->row_step,
|
|
mpi->stride[0] * vf->priv->row_step,
|
|
vf->priv->row_step != 1);
|
|
break;
|
|
case MONO_L:
|
|
case MONO_R:
|
|
memcpy_pic(dmpi->planes[0],
|
|
mpi->planes[0] + in_off_left,
|
|
3 * vf->priv->width,
|
|
vf->priv->height,
|
|
dmpi->stride[0],
|
|
mpi->stride[0]);
|
|
break;
|
|
case ANAGLYPH_RC_GRAY:
|
|
case ANAGLYPH_RC_HALF:
|
|
case ANAGLYPH_RC_COLOR:
|
|
case ANAGLYPH_RC_DUBOIS:
|
|
case ANAGLYPH_GM_GRAY:
|
|
case ANAGLYPH_GM_HALF:
|
|
case ANAGLYPH_GM_COLOR:
|
|
case ANAGLYPH_YB_GRAY:
|
|
case ANAGLYPH_YB_HALF:
|
|
case ANAGLYPH_YB_COLOR: {
|
|
int x,y,il,ir,o;
|
|
unsigned char *source = mpi->planes[0];
|
|
unsigned char *dest = dmpi->planes[0];
|
|
unsigned int out_width = vf->priv->out.width;
|
|
int *ana_matrix[3];
|
|
|
|
for(int i = 0; i < 3; i++)
|
|
ana_matrix[i] = vf->priv->ana_matrix[i];
|
|
|
|
for (y = 0; y < vf->priv->out.height; y++) {
|
|
o = dmpi->stride[0] * y;
|
|
il = in_off_left + y * mpi->stride[0];
|
|
ir = in_off_right + y * mpi->stride[0];
|
|
for (x = 0; x < out_width; x++) {
|
|
dest[o ] = ana_convert(
|
|
ana_matrix[0], source + il, source + ir); //red out
|
|
dest[o + 1] = ana_convert(
|
|
ana_matrix[1], source + il, source + ir); //green out
|
|
dest[o + 2] = ana_convert(
|
|
ana_matrix[2], source + il, source + ir); //blue out
|
|
il += 3;
|
|
ir += 3;
|
|
o += 3;
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
default:
|
|
mp_msg(MSGT_VFILTER, MSGL_WARN,
|
|
"[stereo3d] stereo format of output is not supported\n");
|
|
return 0;
|
|
break;
|
|
}
|
|
}
|
|
return vf_next_put_image(vf, dmpi, pts);
|
|
}
|
|
|
|
static int query_format(struct vf_instance *vf, unsigned int fmt)
|
|
{
|
|
switch (fmt)
|
|
case IMGFMT_RGB24:
|
|
return vf_next_query_format(vf, fmt);
|
|
return 0;
|
|
}
|
|
|
|
static void uninit(vf_instance_t *vf)
|
|
{
|
|
free(vf->priv);
|
|
}
|
|
|
|
static int vf_open(vf_instance_t *vf, char *args)
|
|
{
|
|
vf->config = config;
|
|
vf->uninit = uninit;
|
|
vf->put_image = put_image;
|
|
vf->query_format = query_format;
|
|
|
|
return 1;
|
|
}
|
|
|
|
///Presets usage
|
|
static const struct format_preset {
|
|
char* name;
|
|
stereo_code scode;
|
|
} vf_format_presets_defs[] = {
|
|
{"arcg", ANAGLYPH_RC_GRAY},
|
|
{"anaglyph_red_cyan_gray", ANAGLYPH_RC_GRAY},
|
|
{"arch", ANAGLYPH_RC_HALF},
|
|
{"anaglyph_red_cyan_half_color", ANAGLYPH_RC_HALF},
|
|
{"arcc", ANAGLYPH_RC_COLOR},
|
|
{"anaglyph_red_cyan_color", ANAGLYPH_RC_COLOR},
|
|
{"arcd", ANAGLYPH_RC_DUBOIS},
|
|
{"anaglyph_red_cyan_dubios", ANAGLYPH_RC_DUBOIS},
|
|
{"agmg", ANAGLYPH_GM_GRAY},
|
|
{"anaglyph_green_magenta_gray", ANAGLYPH_GM_GRAY},
|
|
{"agmh", ANAGLYPH_GM_HALF},
|
|
{"anaglyph_green_magenta_half_color",ANAGLYPH_GM_HALF},
|
|
{"agmc", ANAGLYPH_GM_COLOR},
|
|
{"anaglyph_green_magenta_color", ANAGLYPH_GM_COLOR},
|
|
{"aybg", ANAGLYPH_YB_GRAY},
|
|
{"anaglyph_yellow_blue_gray", ANAGLYPH_YB_GRAY},
|
|
{"aybh", ANAGLYPH_YB_HALF},
|
|
{"anaglyph_yellow_blue_half_color", ANAGLYPH_YB_HALF},
|
|
{"aybc", ANAGLYPH_YB_COLOR},
|
|
{"anaglyph_yellow_blue_color", ANAGLYPH_YB_COLOR},
|
|
{"ml", MONO_L},
|
|
{"mono_left", MONO_L},
|
|
{"mr", MONO_R},
|
|
{"mono_right", MONO_R},
|
|
{"sbsl", SIDE_BY_SIDE_LR},
|
|
{"side_by_side_left_first", SIDE_BY_SIDE_LR},
|
|
{"sbsr", SIDE_BY_SIDE_RL},
|
|
{"side_by_side_right_first", SIDE_BY_SIDE_RL},
|
|
{"abl", ABOVE_BELOW_LR},
|
|
{"above_below_left_first", ABOVE_BELOW_LR},
|
|
{"abr", ABOVE_BELOW_RL},
|
|
{"above_below_right_first", ABOVE_BELOW_RL},
|
|
{"ab2l", ABOVE_BELOW_2_LR},
|
|
{"above_below_half_height_left_first", ABOVE_BELOW_2_LR},
|
|
{"ab2r", ABOVE_BELOW_2_RL},
|
|
{"above_below_half_height_right_first",ABOVE_BELOW_2_RL},
|
|
{"irl", INTERLEAVE_ROWS_LR},
|
|
{"interleave_rows_left_first", INTERLEAVE_ROWS_LR},
|
|
{"irr", INTERLEAVE_ROWS_RL},
|
|
{"interleave_rows_right_first", INTERLEAVE_ROWS_RL},
|
|
{ NULL, 0}
|
|
};
|
|
|
|
#define ST_OFF(f) M_ST_OFF(struct format_preset,f)
|
|
static const m_option_t vf_format_preset_fields_in[] = {
|
|
{"in", ST_OFF(scode), CONF_TYPE_INT, 0,0,0, NULL},
|
|
{ NULL, NULL, 0, 0, 0, 0, NULL }
|
|
};
|
|
static const m_option_t vf_format_preset_fields_out[] = {
|
|
{"out", ST_OFF(scode), CONF_TYPE_INT, 0,0,0, NULL},
|
|
{ NULL, NULL, 0, 0, 0, 0, NULL }
|
|
};
|
|
|
|
static const m_struct_t vf_format_preset_in = {
|
|
"stereo_format_preset_in",
|
|
sizeof(struct format_preset),
|
|
NULL,
|
|
vf_format_preset_fields_in
|
|
};
|
|
static const m_struct_t vf_format_preset_out = {
|
|
"stereo_format_preset_out",
|
|
sizeof(struct format_preset),
|
|
NULL,
|
|
vf_format_preset_fields_out
|
|
};
|
|
|
|
static const m_struct_t vf_opts;
|
|
static const m_obj_presets_t format_preset_in = {
|
|
(struct m_struct_st*)&vf_format_preset_in,
|
|
(struct m_struct_st*)&vf_opts,
|
|
(struct format_preset*)vf_format_presets_defs,
|
|
ST_OFF(name)
|
|
};
|
|
static const m_obj_presets_t format_preset_out = {
|
|
(struct m_struct_st*)&vf_format_preset_out,
|
|
(struct m_struct_st*)&vf_opts,
|
|
(struct format_preset*)vf_format_presets_defs,
|
|
ST_OFF(name)
|
|
};
|
|
|
|
/// Now the options
|
|
#undef ST_OFF
|
|
#define ST_OFF(f) M_ST_OFF(struct vf_priv_s,f)
|
|
static const m_option_t vf_opts_fields[] = {
|
|
{"stereo_in", 0, CONF_TYPE_OBJ_PRESETS, 0, 0, 0,
|
|
(m_obj_presets_t*)&format_preset_in},
|
|
{"stereo_out", 0, CONF_TYPE_OBJ_PRESETS, 0, 0, 0,
|
|
(m_obj_presets_t*)&format_preset_out},
|
|
{"in", ST_OFF(in.fmt), CONF_TYPE_INT, 0,0,0, NULL},
|
|
{"out", ST_OFF(out.fmt), CONF_TYPE_INT, 0,0,0, NULL},
|
|
{ NULL, NULL, 0, 0, 0, 0, NULL }
|
|
};
|
|
|
|
static const m_struct_t vf_opts = {
|
|
"stereo3d",
|
|
sizeof(struct vf_priv_s),
|
|
&vf_priv_default,
|
|
vf_opts_fields
|
|
};
|
|
|
|
|
|
//==info struct==//
|
|
const vf_info_t vf_info_stereo3d = {
|
|
"stereoscopic 3d view",
|
|
"stereo3d",
|
|
"Gordon Schmidt",
|
|
"view stereoscopic videos",
|
|
vf_open,
|
|
&vf_opts
|
|
};
|