Mercurial > mplayer.hg
view libmpcodecs/vf_stereo3d.c @ 33533:3552dac79b1a
Add some missing libavutil #includes.
author | diego |
---|---|
date | Thu, 16 Jun 2011 09:27:31 +0000 |
parents | 45b3a91e8fb7 |
children | 16cf555989a9 |
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 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; } 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->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 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: memcpy_pic(dmpi->planes[0] + out_off_left, mpi->planes[0] + in_off_left, 3 * vf->priv->width, vf->priv->height, dmpi->stride[0], mpi->stride[0]); memcpy_pic(dmpi->planes[0] + out_off_right, mpi->planes[0] + in_off_right, 3 * vf->priv->width, vf->priv->height, dmpi->stride[0], mpi->stride[0]); 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}, { 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 };