123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512 |
- /*
- * 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 "vf.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)
- SIDE_BY_SIDE_2_LR, //side by side parallel with half width resolution
- SIDE_BY_SIDE_2_RL, //side by side crosseye with half width resolution
- 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_2_LR:
- d_width *= 2;
- case SIDE_BY_SIDE_LR:
- vf->priv->width = width / 2;
- vf->priv->in.off_right = vf->priv->width * 3;
- break;
- case SIDE_BY_SIDE_2_RL:
- d_width *= 2;
- 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_2_LR:
- d_width /= 2;
- 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_2_RL:
- d_width /= 2;
- 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 SIDE_BY_SIDE_2_LR:
- case SIDE_BY_SIDE_2_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;
- vf->priv=malloc(sizeof(struct vf_priv_s));
- memset(vf->priv, 0, sizeof(struct vf_priv_s));
- vf->priv->in.fmt = SIDE_BY_SIDE_LR;
- vf->priv->out.fmt= ANAGLYPH_RC_DUBOIS;
- if (args) sscanf(args, "%d:%d", &vf->priv->in.fmt, &vf->priv->out.fmt);
- return 1;
- }
- #if 0
- ///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},
- {"sbs2l", SIDE_BY_SIDE_2_LR},
- {"side_by_side_half_width_left_first", SIDE_BY_SIDE_2_LR},
- {"sbs2r", SIDE_BY_SIDE_2_RL},
- {"side_by_side_half_width_right_first",SIDE_BY_SIDE_2_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
- };
- #endif
- //==info struct==//
- const vf_info_t vf_info_stereo3d = {
- "stereoscopic 3d view",
- "stereo3d",
- "Gordon Schmidt",
- "view stereoscopic videos",
- vf_open,
- // &vf_opts
- };
|