Mercurial > mplayer.hg
view libmpcodecs/vf_stereo3d.c @ 33737:71c29e8ec68f
Move string functions from interface.c to string.c.
Additionally, change char parameter of gstrchr() to const and
rename parameter size of gstrncmp() n.
The Win32 GUI now needs gstrdup() defined before its first usage.
author | ib |
---|---|
date | Thu, 07 Jul 2011 08:32:10 +0000 |
parents | 042b10b477fa |
children | 475d96f700d4 |
line wrap: on
line source
/* * 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 "help_mp.h" #include "img_format.h" #include "mp_image.h" #include "vd.h" #include "vf.h" #include "m_struct.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) { 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 (!opt_screen_size_x && !opt_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 i,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(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 };