falkTX
/
FFmpeg
mirror of https://github.com/falkTX/FFmpeg.git
1
0
Fork 0
Code Issues 0 Releases 338 Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
79344 Commits
32 Branches
293MB
Tree: 901b0f1a21
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '901b0f1a21'
${ noResults }
FFmpeg/libavfilter/vf_stereo3d.c

1117 lines
45KB
Raw Blame History 

  1. /*

  2.  * Copyright (c) 2010 Gordon Schmidt <gordon.schmidt <at> s2000.tu-chemnitz.de>

  3.  * Copyright (c) 2013-2015 Paul B Mahol

  4.  *

  5.  * This file is part of FFmpeg.

  6.  *

  7.  * FFmpeg is free software; you can redistribute it and/or

  8.  * modify it under the terms of the GNU General Public

  9.  * License as published by the Free Software Foundation; either

  10.  * version 2 of the License, or (at your option) any later version.

  11.  *

  12.  * FFmpeg is distributed in the hope that it will be useful,

  13.  * but WITHOUT ANY WARRANTY; without even the implied warranty of

  14.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

  15.  * GNU General Public License for more details.

  16.  *

  17.  * You should have received a copy of the GNU General Public License along

  18.  * with FFmpeg; if not, write to the Free Software

  19.  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

  20.  */

  21. 

  22. #include "libavutil/avassert.h"

  23. #include "libavutil/imgutils.h"

  24. #include "libavutil/intreadwrite.h"

  25. #include "libavutil/opt.h"

  26. #include "libavutil/parseutils.h"

  27. #include "libavutil/pixdesc.h"

  28. #include "avfilter.h"

  29. #include "drawutils.h"

  30. #include "formats.h"

  31. #include "internal.h"

  32. #include "video.h"

  33. #include "stereo3d.h"

  34. 

  35. enum StereoCode {

  36.     ANAGLYPH_RC_GRAY,   // anaglyph red/cyan gray

  37.     ANAGLYPH_RC_HALF,   // anaglyph red/cyan half colored

  38.     ANAGLYPH_RC_COLOR,  // anaglyph red/cyan colored

  39.     ANAGLYPH_RC_DUBOIS, // anaglyph red/cyan dubois

  40.     ANAGLYPH_GM_GRAY,   // anaglyph green/magenta gray

  41.     ANAGLYPH_GM_HALF,   // anaglyph green/magenta half colored

  42.     ANAGLYPH_GM_COLOR,  // anaglyph green/magenta colored

  43.     ANAGLYPH_GM_DUBOIS, // anaglyph green/magenta dubois

  44.     ANAGLYPH_YB_GRAY,   // anaglyph yellow/blue gray

  45.     ANAGLYPH_YB_HALF,   // anaglyph yellow/blue half colored

  46.     ANAGLYPH_YB_COLOR,  // anaglyph yellow/blue colored

  47.     ANAGLYPH_YB_DUBOIS, // anaglyph yellow/blue dubois

  48.     ANAGLYPH_RB_GRAY,   // anaglyph red/blue gray

  49.     ANAGLYPH_RG_GRAY,   // anaglyph red/green gray

  50.     MONO_L,             // mono output for debugging (left eye only)

  51.     MONO_R,             // mono output for debugging (right eye only)

  52.     INTERLEAVE_ROWS_LR, // row-interleave (left eye has top row)

  53.     INTERLEAVE_ROWS_RL, // row-interleave (right eye has top row)

  54.     SIDE_BY_SIDE_LR,    // side by side parallel (left eye left, right eye right)

  55.     SIDE_BY_SIDE_RL,    // side by side crosseye (right eye left, left eye right)

  56.     SIDE_BY_SIDE_2_LR,  // side by side parallel with half width resolution

  57.     SIDE_BY_SIDE_2_RL,  // side by side crosseye with half width resolution

  58.     ABOVE_BELOW_LR,     // above-below (left eye above, right eye below)

  59.     ABOVE_BELOW_RL,     // above-below (right eye above, left eye below)

  60.     ABOVE_BELOW_2_LR,   // above-below with half height resolution

  61.     ABOVE_BELOW_2_RL,   // above-below with half height resolution

  62.     ALTERNATING_LR,     // alternating frames (left eye first, right eye second)

  63.     ALTERNATING_RL,     // alternating frames (right eye first, left eye second)

  64.     CHECKERBOARD_LR,    // checkerboard pattern (left eye first, right eye second)

  65.     CHECKERBOARD_RL,    // checkerboard pattern (right eye first, left eye second)

  66.     INTERLEAVE_COLS_LR, // column-interleave (left eye first, right eye second)

  67.     INTERLEAVE_COLS_RL, // column-interleave (right eye first, left eye second)

  68.     HDMI,               // HDMI frame pack (left eye first, right eye second)

  69.     STEREO_CODE_COUNT   // TODO: needs autodetection

  70. };

  71. 

  72. typedef struct StereoComponent {

  73.     int format;                 ///< StereoCode

  74.     int width, height;

  75.     int off_left, off_right;

  76.     int off_lstep, off_rstep;

  77.     int row_left, row_right;

  78.     int row_step;

  79. } StereoComponent;

  80. 

  81. static const int ana_coeff[][3][6] = {

  82.   [ANAGLYPH_RB_GRAY]   =

  83.     {{19595, 38470,  7471,     0,     0,     0},

  84.      {    0,     0,     0,     0,     0,     0},

  85.      {    0,     0,     0, 19595, 38470,  7471}},

  86.   [ANAGLYPH_RG_GRAY]   =

  87.     {{19595, 38470,  7471,     0,     0,     0},

  88.      {    0,     0,     0, 19595, 38470,  7471},

  89.      {    0,     0,     0,     0,     0,     0}},

  90.   [ANAGLYPH_RC_GRAY]   =

  91.     {{19595, 38470,  7471,     0,     0,     0},

  92.      {    0,     0,     0, 19595, 38470,  7471},

  93.      {    0,     0,     0, 19595, 38470,  7471}},

  94.   [ANAGLYPH_RC_HALF]   =

  95.     {{19595, 38470,  7471,     0,     0,     0},

  96.      {    0,     0,     0,     0, 65536,     0},

  97.      {    0,     0,     0,     0,     0, 65536}},

  98.   [ANAGLYPH_RC_COLOR]  =

  99.     {{65536,     0,     0,     0,     0,     0},

  100.      {    0,     0,     0,     0, 65536,     0},

  101.      {    0,     0,     0,     0,     0, 65536}},

  102.   [ANAGLYPH_RC_DUBOIS] =

  103.     {{29891, 32800, 11559, -2849, -5763,  -102},

  104.      {-2627, -2479, -1033, 24804, 48080, -1209},

  105.      { -997, -1350,  -358, -4729, -7403, 80373}},

  106.   [ANAGLYPH_GM_GRAY]   =

  107.     {{    0,     0,     0, 19595, 38470,  7471},

  108.      {19595, 38470,  7471,     0,     0,     0},

  109.      {    0,     0,     0, 19595, 38470,  7471}},

  110.   [ANAGLYPH_GM_HALF]   =

  111.     {{    0,     0,     0, 65536,     0,     0},

  112.      {19595, 38470,  7471,     0,     0,     0},

  113.      {    0,     0,     0,     0,     0, 65536}},

  114.   [ANAGLYPH_GM_COLOR]  =

  115.     {{    0,     0,     0, 65536,     0,     0},

  116.      {    0, 65536,     0,     0,     0,     0},

  117.      {    0,     0,     0,     0,     0, 65536}},

  118.   [ANAGLYPH_GM_DUBOIS]  =

  119.     {{-4063,-10354, -2556, 34669, 46203,  1573},

  120.      {18612, 43778,  9372, -1049,  -983, -4260},

  121.      { -983, -1769,  1376,   590,  4915, 61407}},

  122.   [ANAGLYPH_YB_GRAY]   =

  123.     {{    0,     0,     0, 19595, 38470,  7471},

  124.      {    0,     0,     0, 19595, 38470,  7471},

  125.      {19595, 38470,  7471,     0,     0,     0}},

  126.   [ANAGLYPH_YB_HALF]   =

  127.     {{    0,     0,     0, 65536,     0,     0},

  128.      {    0,     0,     0,     0, 65536,     0},

  129.      {19595, 38470,  7471,     0,     0,     0}},

  130.   [ANAGLYPH_YB_COLOR]  =

  131.     {{    0,     0,     0, 65536,     0,     0},

  132.      {    0,     0,     0,     0, 65536,     0},

  133.      {    0,     0, 65536,     0,     0,     0}},

  134.   [ANAGLYPH_YB_DUBOIS] =

  135.     {{65535,-12650,18451,   -987, -7590, -1049},

  136.      {-1604, 56032, 4196,    370,  3826, -1049},

  137.      {-2345,-10676, 1358,   5801, 11416, 56217}},

  138. };

  139. 

  140. typedef struct Stereo3DContext {

  141.     const AVClass *class;

  142.     StereoComponent in, out;

  143.     int width, height;

  144.     const int *ana_matrix[3];

  145.     int nb_planes;

  146.     int linesize[4];

  147.     int pheight[4];

  148.     int hsub, vsub;

  149.     int pixstep[4];

  150.     AVFrame *prev;

  151.     int blanks;

  152.     int in_off_left[4], in_off_right[4];

  153.     Stereo3DDSPContext dsp;

  154. } Stereo3DContext;

  155. 

  156. #define OFFSET(x) offsetof(Stereo3DContext, x)

  157. #define FLAGS AV_OPT_FLAG_FILTERING_PARAM|AV_OPT_FLAG_VIDEO_PARAM

  158. 

  159. static const AVOption stereo3d_options[] = {

  160.     { "in",    "set input format",  OFFSET(in.format),   AV_OPT_TYPE_INT,   {.i64=SIDE_BY_SIDE_LR}, INTERLEAVE_ROWS_LR, STEREO_CODE_COUNT-1, FLAGS, "in"},

  161.     { "ab2l",  "above below half height left first",  0, AV_OPT_TYPE_CONST, {.i64=ABOVE_BELOW_2_LR},   0, 0, FLAGS, "in" },

  162.     { "ab2r",  "above below half height right first", 0, AV_OPT_TYPE_CONST, {.i64=ABOVE_BELOW_2_RL},   0, 0, FLAGS, "in" },

  163.     { "abl",   "above below left first",              0, AV_OPT_TYPE_CONST, {.i64=ABOVE_BELOW_LR},     0, 0, FLAGS, "in" },

  164.     { "abr",   "above below right first",             0, AV_OPT_TYPE_CONST, {.i64=ABOVE_BELOW_RL},     0, 0, FLAGS, "in" },

  165.     { "al",    "alternating frames left first",       0, AV_OPT_TYPE_CONST, {.i64=ALTERNATING_LR},     0, 0, FLAGS, "in" },

  166.     { "ar",    "alternating frames right first",      0, AV_OPT_TYPE_CONST, {.i64=ALTERNATING_RL},     0, 0, FLAGS, "in" },

  167.     { "sbs2l", "side by side half width left first",  0, AV_OPT_TYPE_CONST, {.i64=SIDE_BY_SIDE_2_LR},  0, 0, FLAGS, "in" },

  168.     { "sbs2r", "side by side half width right first", 0, AV_OPT_TYPE_CONST, {.i64=SIDE_BY_SIDE_2_RL},  0, 0, FLAGS, "in" },

  169.     { "sbsl",  "side by side left first",             0, AV_OPT_TYPE_CONST, {.i64=SIDE_BY_SIDE_LR},    0, 0, FLAGS, "in" },

  170.     { "sbsr",  "side by side right first",            0, AV_OPT_TYPE_CONST, {.i64=SIDE_BY_SIDE_RL},    0, 0, FLAGS, "in" },

  171.     { "irl",   "interleave rows left first",          0, AV_OPT_TYPE_CONST, {.i64=INTERLEAVE_ROWS_LR}, 0, 0, FLAGS, "in" },

  172.     { "irr",   "interleave rows right first",         0, AV_OPT_TYPE_CONST, {.i64=INTERLEAVE_ROWS_RL}, 0, 0, FLAGS, "in" },

  173.     { "icl",   "interleave columns left first",       0, AV_OPT_TYPE_CONST, {.i64=INTERLEAVE_COLS_LR}, 0, 0, FLAGS, "in" },

  174.     { "icr",   "interleave columns right first",      0, AV_OPT_TYPE_CONST, {.i64=INTERLEAVE_COLS_RL}, 0, 0, FLAGS, "in" },

  175.     { "out",   "set output format", OFFSET(out.format),  AV_OPT_TYPE_INT,   {.i64=ANAGLYPH_RC_DUBOIS}, 0, STEREO_CODE_COUNT-1, FLAGS, "out"},

  176.     { "ab2l",  "above below half height left first",  0, AV_OPT_TYPE_CONST, {.i64=ABOVE_BELOW_2_LR},   0, 0, FLAGS, "out" },

  177.     { "ab2r",  "above below half height right first", 0, AV_OPT_TYPE_CONST, {.i64=ABOVE_BELOW_2_RL},   0, 0, FLAGS, "out" },

  178.     { "abl",   "above below left first",              0, AV_OPT_TYPE_CONST, {.i64=ABOVE_BELOW_LR},     0, 0, FLAGS, "out" },

  179.     { "abr",   "above below right first",             0, AV_OPT_TYPE_CONST, {.i64=ABOVE_BELOW_RL},     0, 0, FLAGS, "out" },

  180.     { "agmc",  "anaglyph green magenta color",        0, AV_OPT_TYPE_CONST, {.i64=ANAGLYPH_GM_COLOR},  0, 0, FLAGS, "out" },

  181.     { "agmd",  "anaglyph green magenta dubois",       0, AV_OPT_TYPE_CONST, {.i64=ANAGLYPH_GM_DUBOIS}, 0, 0, FLAGS, "out" },

  182.     { "agmg",  "anaglyph green magenta gray",         0, AV_OPT_TYPE_CONST, {.i64=ANAGLYPH_GM_GRAY},   0, 0, FLAGS, "out" },

  183.     { "agmh",  "anaglyph green magenta half color",   0, AV_OPT_TYPE_CONST, {.i64=ANAGLYPH_GM_HALF},   0, 0, FLAGS, "out" },

  184.     { "al",    "alternating frames left first",       0, AV_OPT_TYPE_CONST, {.i64=ALTERNATING_LR},     0, 0, FLAGS, "out" },

  185.     { "ar",    "alternating frames right first",      0, AV_OPT_TYPE_CONST, {.i64=ALTERNATING_RL},     0, 0, FLAGS, "out" },

  186.     { "arbg",  "anaglyph red blue gray",              0, AV_OPT_TYPE_CONST, {.i64=ANAGLYPH_RB_GRAY},   0, 0, FLAGS, "out" },

  187.     { "arcc",  "anaglyph red cyan color",             0, AV_OPT_TYPE_CONST, {.i64=ANAGLYPH_RC_COLOR},  0, 0, FLAGS, "out" },

  188.     { "arcd",  "anaglyph red cyan dubois",            0, AV_OPT_TYPE_CONST, {.i64=ANAGLYPH_RC_DUBOIS}, 0, 0, FLAGS, "out" },

  189.     { "arcg",  "anaglyph red cyan gray",              0, AV_OPT_TYPE_CONST, {.i64=ANAGLYPH_RC_GRAY},   0, 0, FLAGS, "out" },

  190.     { "arch",  "anaglyph red cyan half color",        0, AV_OPT_TYPE_CONST, {.i64=ANAGLYPH_RC_HALF},   0, 0, FLAGS, "out" },

  191.     { "argg",  "anaglyph red green gray",             0, AV_OPT_TYPE_CONST, {.i64=ANAGLYPH_RG_GRAY},   0, 0, FLAGS, "out" },

  192.     { "aybc",  "anaglyph yellow blue color",          0, AV_OPT_TYPE_CONST, {.i64=ANAGLYPH_YB_COLOR},  0, 0, FLAGS, "out" },

  193.     { "aybd",  "anaglyph yellow blue dubois",         0, AV_OPT_TYPE_CONST, {.i64=ANAGLYPH_YB_DUBOIS}, 0, 0, FLAGS, "out" },

  194.     { "aybg",  "anaglyph yellow blue gray",           0, AV_OPT_TYPE_CONST, {.i64=ANAGLYPH_YB_GRAY},   0, 0, FLAGS, "out" },

  195.     { "aybh",  "anaglyph yellow blue half color",     0, AV_OPT_TYPE_CONST, {.i64=ANAGLYPH_YB_HALF},   0, 0, FLAGS, "out" },

  196.     { "irl",   "interleave rows left first",          0, AV_OPT_TYPE_CONST, {.i64=INTERLEAVE_ROWS_LR}, 0, 0, FLAGS, "out" },

  197.     { "irr",   "interleave rows right first",         0, AV_OPT_TYPE_CONST, {.i64=INTERLEAVE_ROWS_RL}, 0, 0, FLAGS, "out" },

  198.     { "ml",    "mono left",                           0, AV_OPT_TYPE_CONST, {.i64=MONO_L},             0, 0, FLAGS, "out" },

  199.     { "mr",    "mono right",                          0, AV_OPT_TYPE_CONST, {.i64=MONO_R},             0, 0, FLAGS, "out" },

  200.     { "sbs2l", "side by side half width left first",  0, AV_OPT_TYPE_CONST, {.i64=SIDE_BY_SIDE_2_LR},  0, 0, FLAGS, "out" },

  201.     { "sbs2r", "side by side half width right first", 0, AV_OPT_TYPE_CONST, {.i64=SIDE_BY_SIDE_2_RL},  0, 0, FLAGS, "out" },

  202.     { "sbsl",  "side by side left first",             0, AV_OPT_TYPE_CONST, {.i64=SIDE_BY_SIDE_LR},    0, 0, FLAGS, "out" },

  203.     { "sbsr",  "side by side right first",            0, AV_OPT_TYPE_CONST, {.i64=SIDE_BY_SIDE_RL},    0, 0, FLAGS, "out" },

  204.     { "chl",   "checkerboard left first",             0, AV_OPT_TYPE_CONST, {.i64=CHECKERBOARD_LR},    0, 0, FLAGS, "out" },

  205.     { "chr",   "checkerboard right first",            0, AV_OPT_TYPE_CONST, {.i64=CHECKERBOARD_RL},    0, 0, FLAGS, "out" },

  206.     { "icl",   "interleave columns left first",       0, AV_OPT_TYPE_CONST, {.i64=INTERLEAVE_COLS_LR}, 0, 0, FLAGS, "out" },

  207.     { "icr",   "interleave columns right first",      0, AV_OPT_TYPE_CONST, {.i64=INTERLEAVE_COLS_RL}, 0, 0, FLAGS, "out" },

  208.     { "hdmi",  "HDMI frame pack",                     0, AV_OPT_TYPE_CONST, {.i64=HDMI},               0, 0, FLAGS, "out" },

  209.     { NULL }

  210. };

  211. 

  212. AVFILTER_DEFINE_CLASS(stereo3d);

  213. 

  214. static const enum AVPixelFormat anaglyph_pix_fmts[] = {

  215.     AV_PIX_FMT_RGB24, AV_PIX_FMT_BGR24,

  216.     AV_PIX_FMT_NONE

  217. };

  218. 

  219. static const enum AVPixelFormat other_pix_fmts[] = {

  220.     AV_PIX_FMT_RGB24, AV_PIX_FMT_BGR24,

  221.     AV_PIX_FMT_RGB48BE, AV_PIX_FMT_BGR48BE,

  222.     AV_PIX_FMT_RGB48LE, AV_PIX_FMT_BGR48LE,

  223.     AV_PIX_FMT_RGBA64BE, AV_PIX_FMT_BGRA64BE,

  224.     AV_PIX_FMT_RGBA64LE, AV_PIX_FMT_BGRA64LE,

  225.     AV_PIX_FMT_RGBA,  AV_PIX_FMT_BGRA,

  226.     AV_PIX_FMT_ARGB,  AV_PIX_FMT_ABGR,

  227.     AV_PIX_FMT_RGB0,  AV_PIX_FMT_BGR0,

  228.     AV_PIX_FMT_0RGB,  AV_PIX_FMT_0BGR,

  229.     AV_PIX_FMT_GBRP,

  230.     AV_PIX_FMT_GBRP9BE,  AV_PIX_FMT_GBRP9LE,

  231.     AV_PIX_FMT_GBRP10BE, AV_PIX_FMT_GBRP10LE,

  232.     AV_PIX_FMT_GBRP12BE, AV_PIX_FMT_GBRP12LE,

  233.     AV_PIX_FMT_GBRP14BE, AV_PIX_FMT_GBRP14LE,

  234.     AV_PIX_FMT_GBRP16BE, AV_PIX_FMT_GBRP16LE,

  235.     AV_PIX_FMT_YUV410P,

  236.     AV_PIX_FMT_YUV411P,

  237.     AV_PIX_FMT_YUV420P, AV_PIX_FMT_YUVA420P,

  238.     AV_PIX_FMT_YUV422P, AV_PIX_FMT_YUVA422P,

  239.     AV_PIX_FMT_YUV440P,

  240.     AV_PIX_FMT_YUV444P, AV_PIX_FMT_YUVA444P,

  241.     AV_PIX_FMT_YUVJ411P,

  242.     AV_PIX_FMT_YUVJ420P,

  243.     AV_PIX_FMT_YUVJ422P,

  244.     AV_PIX_FMT_YUVJ440P,

  245.     AV_PIX_FMT_YUVJ444P,

  246.     AV_PIX_FMT_YUV420P9LE,  AV_PIX_FMT_YUVA420P9LE,

  247.     AV_PIX_FMT_YUV420P9BE,  AV_PIX_FMT_YUVA420P9BE,

  248.     AV_PIX_FMT_YUV422P9LE,  AV_PIX_FMT_YUVA422P9LE,

  249.     AV_PIX_FMT_YUV422P9BE,  AV_PIX_FMT_YUVA422P9BE,

  250.     AV_PIX_FMT_YUV444P9LE,  AV_PIX_FMT_YUVA444P9LE,

  251.     AV_PIX_FMT_YUV444P9BE,  AV_PIX_FMT_YUVA444P9BE,

  252.     AV_PIX_FMT_YUV420P10LE, AV_PIX_FMT_YUVA420P10LE,

  253.     AV_PIX_FMT_YUV420P10BE, AV_PIX_FMT_YUVA420P10BE,

  254.     AV_PIX_FMT_YUV422P10LE, AV_PIX_FMT_YUVA422P10LE,

  255.     AV_PIX_FMT_YUV422P10BE, AV_PIX_FMT_YUVA422P10BE,

  256.     AV_PIX_FMT_YUV444P10LE, AV_PIX_FMT_YUVA444P10LE,

  257.     AV_PIX_FMT_YUV444P10BE, AV_PIX_FMT_YUVA444P10BE,

  258.     AV_PIX_FMT_YUV420P12BE,  AV_PIX_FMT_YUV420P12LE,

  259.     AV_PIX_FMT_YUV422P12BE,  AV_PIX_FMT_YUV422P12LE,

  260.     AV_PIX_FMT_YUV444P12BE,  AV_PIX_FMT_YUV444P12LE,

  261.     AV_PIX_FMT_YUV420P14BE,  AV_PIX_FMT_YUV420P14LE,

  262.     AV_PIX_FMT_YUV422P14BE,  AV_PIX_FMT_YUV422P14LE,

  263.     AV_PIX_FMT_YUV444P14BE,  AV_PIX_FMT_YUV444P14LE,

  264.     AV_PIX_FMT_YUV420P16LE, AV_PIX_FMT_YUVA420P16LE,

  265.     AV_PIX_FMT_YUV420P16BE, AV_PIX_FMT_YUVA420P16BE,

  266.     AV_PIX_FMT_YUV422P16LE, AV_PIX_FMT_YUVA422P16LE,

  267.     AV_PIX_FMT_YUV422P16BE, AV_PIX_FMT_YUVA422P16BE,

  268.     AV_PIX_FMT_YUV444P16LE, AV_PIX_FMT_YUVA444P16LE,

  269.     AV_PIX_FMT_YUV444P16BE, AV_PIX_FMT_YUVA444P16BE,

  270.     AV_PIX_FMT_NONE

  271. };

  272. 

  273. static int query_formats(AVFilterContext *ctx)

  274. {

  275.     Stereo3DContext *s = ctx->priv;

  276.     const enum AVPixelFormat *pix_fmts;

  277.     AVFilterFormats *fmts_list;

  278. 

  279.     switch (s->out.format) {

  280.     case ANAGLYPH_GM_COLOR:

  281.     case ANAGLYPH_GM_DUBOIS:

  282.     case ANAGLYPH_GM_GRAY:

  283.     case ANAGLYPH_GM_HALF:

  284.     case ANAGLYPH_RB_GRAY:

  285.     case ANAGLYPH_RC_COLOR:

  286.     case ANAGLYPH_RC_DUBOIS:

  287.     case ANAGLYPH_RC_GRAY:

  288.     case ANAGLYPH_RC_HALF:

  289.     case ANAGLYPH_RG_GRAY:

  290.     case ANAGLYPH_YB_COLOR:

  291.     case ANAGLYPH_YB_DUBOIS:

  292.     case ANAGLYPH_YB_GRAY:

  293.     case ANAGLYPH_YB_HALF:

  294.         pix_fmts = anaglyph_pix_fmts;

  295.         break;

  296.     default:

  297.         pix_fmts = other_pix_fmts;

  298.     }

  299. 

  300.     fmts_list = ff_make_format_list(pix_fmts);

  301.     if (!fmts_list)

  302.         return AVERROR(ENOMEM);

  303.     return ff_set_common_formats(ctx, fmts_list);

  304. }

  305. 

  306. static inline uint8_t ana_convert(const int *coeff, const uint8_t *left, const uint8_t *right)

  307. {

  308.     int sum;

  309. 

  310.     sum  = coeff[0] * left[0] + coeff[3] * right[0]; //red in

  311.     sum += coeff[1] * left[1] + coeff[4] * right[1]; //green in

  312.     sum += coeff[2] * left[2] + coeff[5] * right[2]; //blue in

  313. 

  314.     return av_clip_uint8(sum >> 16);

  315. }

  316. 

  317. static void anaglyph_ic(uint8_t *dst, uint8_t *lsrc, uint8_t *rsrc,

  318.                         ptrdiff_t dst_linesize, ptrdiff_t l_linesize, ptrdiff_t r_linesize,

  319.                         int width, int height,

  320.                         const int *ana_matrix_r, const int *ana_matrix_g, const int *ana_matrix_b)

  321. {

  322.     int x, y, o;

  323. 

  324.     for (y = 0; y < height; y++) {

  325.         for (o = 0, x = 0; x < width; x++, o+= 3) {

  326.             dst[o    ] = ana_convert(ana_matrix_r, lsrc + o * 2, rsrc + o * 2);

  327.             dst[o + 1] = ana_convert(ana_matrix_g, lsrc + o * 2, rsrc + o * 2);

  328.             dst[o + 2] = ana_convert(ana_matrix_b, lsrc + o * 2, rsrc + o * 2);

  329.         }

  330. 

  331.         dst  += dst_linesize;

  332.         lsrc += l_linesize;

  333.         rsrc += r_linesize;

  334.     }

  335. }

  336. 

  337. static void anaglyph(uint8_t *dst, uint8_t *lsrc, uint8_t *rsrc,

  338.                      ptrdiff_t dst_linesize, ptrdiff_t l_linesize, ptrdiff_t r_linesize,

  339.                      int width, int height,

  340.                      const int *ana_matrix_r, const int *ana_matrix_g, const int *ana_matrix_b)

  341. {

  342.     int x, y, o;

  343. 

  344.     for (y = 0; y < height; y++) {

  345.         for (o = 0, x = 0; x < width; x++, o+= 3) {

  346.             dst[o    ] = ana_convert(ana_matrix_r, lsrc + o, rsrc + o);

  347.             dst[o + 1] = ana_convert(ana_matrix_g, lsrc + o, rsrc + o);

  348.             dst[o + 2] = ana_convert(ana_matrix_b, lsrc + o, rsrc + o);

  349.         }

  350. 

  351.         dst  += dst_linesize;

  352.         lsrc += l_linesize;

  353.         rsrc += r_linesize;

  354.     }

  355. }

  356. 

  357. static int config_output(AVFilterLink *outlink)

  358. {

  359.     AVFilterContext *ctx = outlink->src;

  360.     AVFilterLink *inlink = ctx->inputs[0];

  361.     Stereo3DContext *s = ctx->priv;

  362.     AVRational aspect = inlink->sample_aspect_ratio;

  363.     AVRational fps = inlink->frame_rate;

  364.     AVRational tb = inlink->time_base;

  365.     const AVPixFmtDescriptor *desc = av_pix_fmt_desc_get(outlink->format);

  366.     int ret;

  367. 

  368.     switch (s->in.format) {

  369.     case INTERLEAVE_COLS_LR:

  370.     case INTERLEAVE_COLS_RL:

  371.     case SIDE_BY_SIDE_2_LR:

  372.     case SIDE_BY_SIDE_LR:

  373.     case SIDE_BY_SIDE_2_RL:

  374.     case SIDE_BY_SIDE_RL:

  375.         if (inlink->w & 1) {

  376.             av_log(ctx, AV_LOG_ERROR, "width must be even\n");

  377.             return AVERROR_INVALIDDATA;

  378.         }

  379.         break;

  380.     case INTERLEAVE_ROWS_LR:

  381.     case INTERLEAVE_ROWS_RL:

  382.     case ABOVE_BELOW_2_LR:

  383.     case ABOVE_BELOW_LR:

  384.     case ABOVE_BELOW_2_RL:

  385.     case ABOVE_BELOW_RL:

  386.         if (inlink->h & 1) {

  387.             av_log(ctx, AV_LOG_ERROR, "height must be even\n");

  388.             return AVERROR_INVALIDDATA;

  389.         }

  390.         break;

  391.     }

  392. 

  393.     s->in.width     =

  394.     s->width        = inlink->w;

  395.     s->in.height    =

  396.     s->height       = inlink->h;

  397.     s->in.off_lstep =

  398.     s->in.off_rstep =

  399.     s->in.off_left  =

  400.     s->in.off_right =

  401.     s->in.row_left  =

  402.     s->in.row_right = 0;

  403.     s->in.row_step  = 1;

  404. 

  405.     switch (s->in.format) {

  406.     case SIDE_BY_SIDE_2_LR:

  407.         aspect.num     *= 2;

  408.     case SIDE_BY_SIDE_LR:

  409.         s->width        = inlink->w / 2;

  410.         s->in.off_right = s->width;

  411.         break;

  412.     case SIDE_BY_SIDE_2_RL:

  413.         aspect.num     *= 2;

  414.     case SIDE_BY_SIDE_RL:

  415.         s->width        = inlink->w / 2;

  416.         s->in.off_left  = s->width;

  417.         break;

  418.     case ABOVE_BELOW_2_LR:

  419.         aspect.den     *= 2;

  420.     case ABOVE_BELOW_LR:

  421.         s->in.row_right =

  422.         s->height       = inlink->h / 2;

  423.         break;

  424.     case ABOVE_BELOW_2_RL:

  425.         aspect.den     *= 2;

  426.     case ABOVE_BELOW_RL:

  427.         s->in.row_left  =

  428.         s->height       = inlink->h / 2;

  429.         break;

  430.     case ALTERNATING_RL:

  431.     case ALTERNATING_LR:

  432.         fps.den        *= 2;

  433.         tb.num         *= 2;

  434.         break;

  435.     case INTERLEAVE_COLS_RL:

  436.     case INTERLEAVE_COLS_LR:

  437.         s->width        = inlink->w / 2;

  438.         break;

  439.     case INTERLEAVE_ROWS_LR:

  440.     case INTERLEAVE_ROWS_RL:

  441.         s->in.row_step  = 2;

  442.         if (s->in.format == INTERLEAVE_ROWS_RL)

  443.             s->in.off_lstep = 1;

  444.         else

  445.             s->in.off_rstep = 1;

  446.         if (s->out.format != CHECKERBOARD_LR &&

  447.             s->out.format != CHECKERBOARD_RL)

  448.             s->height   = inlink->h / 2;

  449.         break;

  450.     default:

  451.         av_log(ctx, AV_LOG_ERROR, "input format %d is not supported\n", s->in.format);

  452.         return AVERROR(EINVAL);

  453.     }

  454. 

  455.     s->out.width     = s->width;

  456.     s->out.height    = s->height;

  457.     s->out.off_lstep =

  458.     s->out.off_rstep =

  459.     s->out.off_left  =

  460.     s->out.off_right =

  461.     s->out.row_left  =

  462.     s->out.row_right = 0;

  463.     s->out.row_step  = 1;

  464. 

  465.     switch (s->out.format) {

  466.     case ANAGLYPH_RB_GRAY:

  467.     case ANAGLYPH_RG_GRAY:

  468.     case ANAGLYPH_RC_GRAY:

  469.     case ANAGLYPH_RC_HALF:

  470.     case ANAGLYPH_RC_COLOR:

  471.     case ANAGLYPH_RC_DUBOIS:

  472.     case ANAGLYPH_GM_GRAY:

  473.     case ANAGLYPH_GM_HALF:

  474.     case ANAGLYPH_GM_COLOR:

  475.     case ANAGLYPH_GM_DUBOIS:

  476.     case ANAGLYPH_YB_GRAY:

  477.     case ANAGLYPH_YB_HALF:

  478.     case ANAGLYPH_YB_COLOR:

  479.     case ANAGLYPH_YB_DUBOIS: {

  480.         uint8_t rgba_map[4];

  481. 

  482.         ff_fill_rgba_map(rgba_map, outlink->format);

  483.         s->ana_matrix[rgba_map[0]] = &ana_coeff[s->out.format][0][0];

  484.         s->ana_matrix[rgba_map[1]] = &ana_coeff[s->out.format][1][0];

  485.         s->ana_matrix[rgba_map[2]] = &ana_coeff[s->out.format][2][0];

  486.         break;

  487.     }

  488.     case SIDE_BY_SIDE_2_LR:

  489.         aspect.den      *= 2;

  490.     case SIDE_BY_SIDE_LR:

  491.         s->out.width     = s->width * 2;

  492.         s->out.off_right = s->width;

  493.         break;

  494.     case SIDE_BY_SIDE_2_RL:

  495.         aspect.den      *= 2;

  496.     case SIDE_BY_SIDE_RL:

  497.         s->out.width     = s->width * 2;

  498.         s->out.off_left  = s->width;

  499.         break;

  500.     case ABOVE_BELOW_2_LR:

  501.         aspect.num      *= 2;

  502.     case ABOVE_BELOW_LR:

  503.         s->out.height    = s->height * 2;

  504.         s->out.row_right = s->height;

  505.         break;

  506.     case HDMI:

  507.         if (s->height != 720 && s->height != 1080) {

  508.             av_log(ctx, AV_LOG_ERROR, "Only 720 and 1080 height supported\n");

  509.             return AVERROR(EINVAL);

  510.         }

  511. 

  512.         s->blanks = s->height / 24;

  513.         s->out.height    = s->height * 2 + s->blanks;

  514.         s->out.row_right = s->height + s->blanks;

  515.         break;

  516.     case ABOVE_BELOW_2_RL:

  517.         aspect.num      *= 2;

  518.     case ABOVE_BELOW_RL:

  519.         s->out.height    = s->height * 2;

  520.         s->out.row_left  = s->height;

  521.         break;

  522.     case INTERLEAVE_ROWS_LR:

  523.         s->in.row_step   = 1 + (s->in.format == INTERLEAVE_ROWS_RL);

  524.         s->out.row_step  = 2;

  525.         s->out.height    = s->height * 2;

  526.         s->out.off_rstep = 1;

  527.         break;

  528.     case INTERLEAVE_ROWS_RL:

  529.         s->in.row_step   = 1 + (s->in.format == INTERLEAVE_ROWS_LR);

  530.         s->out.row_step  = 2;

  531.         s->out.height    = s->height * 2;

  532.         s->out.off_lstep = 1;

  533.         break;

  534.     case MONO_R:

  535.         if (s->in.format != INTERLEAVE_COLS_LR) {

  536.             s->in.off_left = s->in.off_right;

  537.             s->in.row_left = s->in.row_right;

  538.         }

  539.         if (s->in.format == INTERLEAVE_ROWS_LR)

  540.             FFSWAP(int, s->in.off_lstep, s->in.off_rstep);

  541.         break;

  542.     case MONO_L:

  543.         if (s->in.format == INTERLEAVE_ROWS_RL)

  544.             FFSWAP(int, s->in.off_lstep, s->in.off_rstep);

  545.         break;

  546.     case ALTERNATING_RL:

  547.     case ALTERNATING_LR:

  548.         fps.num         *= 2;

  549.         tb.den          *= 2;

  550.         break;

  551.     case CHECKERBOARD_LR:

  552.     case CHECKERBOARD_RL:

  553.         s->out.width     = s->width * 2;

  554.         break;

  555.     case INTERLEAVE_COLS_LR:

  556.     case INTERLEAVE_COLS_RL:

  557.         s->out.width     = s->width * 2;

  558.         break;

  559.     default:

  560.         av_log(ctx, AV_LOG_ERROR, "output format %d is not supported\n", s->out.format);

  561.         return AVERROR(EINVAL);

  562.     }

  563. 

  564.     if (s->in.format == INTERLEAVE_COLS_LR || s->in.format == INTERLEAVE_COLS_RL) {

  565.         if ((s->in.format & 1) != (s->out.format & 1)) {

  566.             FFSWAP(int, s->in.row_left,   s->in.row_right);

  567.             FFSWAP(int, s->in.off_lstep,  s->in.off_rstep);

  568.             FFSWAP(int, s->in.off_left,   s->in.off_right);

  569.             FFSWAP(int, s->out.row_left,  s->out.row_right);

  570.             FFSWAP(int, s->out.off_lstep, s->out.off_rstep);

  571.             FFSWAP(int, s->out.off_left,  s->out.off_right);

  572.         }

  573.     }

  574. 

  575.     outlink->w = s->out.width;

  576.     outlink->h = s->out.height;

  577.     outlink->frame_rate = fps;

  578.     outlink->time_base = tb;

  579.     outlink->sample_aspect_ratio = aspect;

  580. 

  581.     if ((ret = av_image_fill_linesizes(s->linesize, outlink->format, s->width)) < 0)

  582.         return ret;

  583.     s->nb_planes = av_pix_fmt_count_planes(outlink->format);

  584.     av_image_fill_max_pixsteps(s->pixstep, NULL, desc);

  585.     s->pheight[1] = s->pheight[2] = AV_CEIL_RSHIFT(s->height, desc->log2_chroma_h);

  586.     s->pheight[0] = s->pheight[3] = s->height;

  587.     s->hsub = desc->log2_chroma_w;

  588.     s->vsub = desc->log2_chroma_h;

  589. 

  590.     s->dsp.anaglyph = anaglyph;

  591.     if (ARCH_X86)

  592.         ff_stereo3d_init_x86(&s->dsp);

  593. 

  594.     return 0;

  595. }

  596. 

  597. typedef struct ThreadData {

  598.     AVFrame *ileft, *iright;

  599.     AVFrame *out;

  600. } ThreadData;

  601. 

  602. static int filter_slice(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs)

  603. {

  604.     Stereo3DContext *s = ctx->priv;

  605.     ThreadData *td = arg;

  606.     AVFrame *ileft = td->ileft;

  607.     AVFrame *iright = td->iright;

  608.     AVFrame *out = td->out;

  609.     int height = s->out.height;

  610.     int start = (height *  jobnr   ) / nb_jobs;

  611.     int end   = (height * (jobnr+1)) / nb_jobs;

  612.     const int **ana_matrix = s->ana_matrix;

  613. 

  614.     s->dsp.anaglyph(out->data[0] + out->linesize[0] * start,

  615.              ileft ->data[0] + s->in_off_left [0]  + ileft->linesize[0] * start * s->in.row_step,

  616.              iright->data[0] + s->in_off_right[0] + iright->linesize[0] * start * s->in.row_step,

  617.              out->linesize[0],

  618.              ileft->linesize[0] * s->in.row_step,

  619.              iright->linesize[0] * s->in.row_step,

  620.              s->out.width, end - start,

  621.              ana_matrix[0], ana_matrix[1], ana_matrix[2]);

  622. 

  623.     return 0;

  624. }

  625. 

  626. static void interleave_cols_to_any(Stereo3DContext *s, int *out_off, int p, AVFrame *in, AVFrame *out, int d)

  627. {

  628.     int y, x;

  629. 

  630.     for (y = 0; y < s->pheight[p]; y++) {

  631.         const uint8_t *src = (const uint8_t*)in->data[p] + y * in->linesize[p] + d * s->pixstep[p];

  632.         uint8_t *dst = out->data[p] + out_off[p] + y * out->linesize[p] * s->out.row_step;

  633. 

  634.         switch (s->pixstep[p]) {

  635.         case 1:

  636.             for (x = 0; x < s->linesize[p]; x++)

  637.                 dst[x] = src[x * 2];

  638.             break;

  639.         case 2:

  640.             for (x = 0; x < s->linesize[p]; x+=2)

  641.                 AV_WN16(&dst[x], AV_RN16(&src[x * 2]));

  642.             break;

  643.         case 3:

  644.             for (x = 0; x < s->linesize[p]; x+=3)

  645.                 AV_WB24(&dst[x], AV_RB24(&src[x * 2]));

  646.             break;

  647.         case 4:

  648.             for (x = 0; x < s->linesize[p]; x+=4)

  649.                 AV_WN32(&dst[x], AV_RN32(&src[x * 2]));

  650.             break;

  651.         case 6:

  652.             for (x = 0; x < s->linesize[p]; x+=6)

  653.                 AV_WB48(&dst[x], AV_RB48(&src[x * 2]));

  654.             break;

  655.         case 8:

  656.             for (x = 0; x < s->linesize[p]; x+=8)

  657.                 AV_WN64(&dst[x], AV_RN64(&src[x * 2]));

  658.             break;

  659.         }

  660.     }

  661. }

  662. 

  663. static int filter_frame(AVFilterLink *inlink, AVFrame *inpicref)

  664. {

  665.     AVFilterContext *ctx  = inlink->dst;

  666.     Stereo3DContext *s = ctx->priv;

  667.     AVFilterLink *outlink = ctx->outputs[0];

  668.     AVFrame *out, *oleft, *oright, *ileft, *iright;

  669.     int out_off_left[4], out_off_right[4];

  670.     int i, ret;

  671. 

  672.     if (s->in.format == s->out.format)

  673.         return ff_filter_frame(outlink, inpicref);

  674. 

  675.     switch (s->out.format) {

  676.     case ALTERNATING_LR:

  677.     case ALTERNATING_RL:

  678.         if (!s->prev) {

  679.             s->prev = inpicref;

  680.             return 0;

  681.         }

  682.         break;

  683.     };

  684. 

  685.     switch (s->in.format) {

  686.     case ALTERNATING_LR:

  687.     case ALTERNATING_RL:

  688.         if (!s->prev) {

  689.             s->prev = inpicref;

  690.             return 0;

  691.         }

  692.         ileft  = s->prev;

  693.         iright = inpicref;

  694.         if (s->in.format == ALTERNATING_RL)

  695.             FFSWAP(AVFrame *, ileft, iright);

  696.         break;

  697.     default:

  698.         ileft = iright = inpicref;

  699.     };

  700. 

  701.     if ((s->out.format == ALTERNATING_LR ||

  702.          s->out.format == ALTERNATING_RL) &&

  703.         (s->in.format == SIDE_BY_SIDE_LR ||

  704.          s->in.format == SIDE_BY_SIDE_RL ||

  705.          s->in.format == SIDE_BY_SIDE_2_LR ||

  706.          s->in.format == SIDE_BY_SIDE_2_RL ||

  707.          s->in.format == ABOVE_BELOW_LR ||

  708.          s->in.format == ABOVE_BELOW_RL ||

  709.          s->in.format == ABOVE_BELOW_2_LR ||

  710.          s->in.format == ABOVE_BELOW_2_RL ||

  711.          s->in.format == INTERLEAVE_ROWS_LR ||

  712.          s->in.format == INTERLEAVE_ROWS_RL)) {

  713.         oright = av_frame_clone(s->prev);

  714.         oleft  = av_frame_clone(s->prev);

  715.         if (!oright || !oleft) {

  716.             av_frame_free(&oright);

  717.             av_frame_free(&oleft);

  718.             av_frame_free(&s->prev);

  719.             av_frame_free(&inpicref);

  720.             return AVERROR(ENOMEM);

  721.         }

  722.     } else if ((s->out.format == MONO_L ||

  723.                 s->out.format == MONO_R) &&

  724.         (s->in.format == SIDE_BY_SIDE_LR ||

  725.          s->in.format == SIDE_BY_SIDE_RL ||

  726.          s->in.format == SIDE_BY_SIDE_2_LR ||

  727.          s->in.format == SIDE_BY_SIDE_2_RL ||

  728.          s->in.format == ABOVE_BELOW_LR ||

  729.          s->in.format == ABOVE_BELOW_RL ||

  730.          s->in.format == ABOVE_BELOW_2_LR ||

  731.          s->in.format == ABOVE_BELOW_2_RL ||

  732.          s->in.format == INTERLEAVE_ROWS_LR ||

  733.          s->in.format == INTERLEAVE_ROWS_RL)) {

  734.         out = oleft = oright = av_frame_clone(inpicref);

  735.         if (!out) {

  736.             av_frame_free(&s->prev);

  737.             av_frame_free(&inpicref);

  738.             return AVERROR(ENOMEM);

  739.         }

  740.     } else if ((s->out.format == MONO_L && s->in.format == ALTERNATING_LR) ||

  741.                (s->out.format == MONO_R && s->in.format == ALTERNATING_RL)) {

  742.         s->prev->pts /= 2;

  743.         ret = ff_filter_frame(outlink, s->prev);

  744.         av_frame_free(&inpicref);

  745.         s->prev = NULL;

  746.         return ret;

  747.     } else if ((s->out.format == MONO_L && s->in.format == ALTERNATING_RL) ||

  748.                (s->out.format == MONO_R && s->in.format == ALTERNATING_LR)) {

  749.         av_frame_free(&s->prev);

  750.         inpicref->pts /= 2;

  751.         return ff_filter_frame(outlink, inpicref);

  752.     } else if ((s->out.format == ALTERNATING_LR && s->in.format == ALTERNATING_RL) ||

  753.                (s->out.format == ALTERNATING_RL && s->in.format == ALTERNATING_LR)) {

  754.         FFSWAP(int64_t, s->prev->pts, inpicref->pts);

  755.         ff_filter_frame(outlink, inpicref);

  756.         ret = ff_filter_frame(outlink, s->prev);

  757.         s->prev = NULL;

  758.         return ret;

  759.     } else {

  760.         out = oleft = oright = ff_get_video_buffer(outlink, outlink->w, outlink->h);

  761.         if (!out) {

  762.             av_frame_free(&s->prev);

  763.             av_frame_free(&inpicref);

  764.             return AVERROR(ENOMEM);

  765.         }

  766.         av_frame_copy_props(out, inpicref);

  767. 

  768.         if (s->out.format == ALTERNATING_LR ||

  769.             s->out.format == ALTERNATING_RL) {

  770.             oright = ff_get_video_buffer(outlink, outlink->w, outlink->h);

  771.             if (!oright) {

  772.                 av_frame_free(&oleft);

  773.                 av_frame_free(&s->prev);

  774.                 av_frame_free(&inpicref);

  775.                 return AVERROR(ENOMEM);

  776.             }

  777.             av_frame_copy_props(oright, s->prev);

  778.         }

  779.     }

  780. 

  781.     for (i = 0; i < 4; i++) {

  782.         int hsub = i == 1 || i == 2 ? s->hsub : 0;

  783.         int vsub = i == 1 || i == 2 ? s->vsub : 0;

  784.         s->in_off_left[i]   = (AV_CEIL_RSHIFT(s->in.row_left,   vsub) + s->in.off_lstep)  * ileft->linesize[i]  + AV_CEIL_RSHIFT(s->in.off_left   * s->pixstep[i], hsub);

  785.         s->in_off_right[i]  = (AV_CEIL_RSHIFT(s->in.row_right,  vsub) + s->in.off_rstep)  * iright->linesize[i] + AV_CEIL_RSHIFT(s->in.off_right  * s->pixstep[i], hsub);

  786.         out_off_left[i]  = (AV_CEIL_RSHIFT(s->out.row_left,  vsub) + s->out.off_lstep) * oleft->linesize[i]  + AV_CEIL_RSHIFT(s->out.off_left  * s->pixstep[i], hsub);

  787.         out_off_right[i] = (AV_CEIL_RSHIFT(s->out.row_right, vsub) + s->out.off_rstep) * oright->linesize[i] + AV_CEIL_RSHIFT(s->out.off_right * s->pixstep[i], hsub);

  788.     }

  789. 

  790.     switch (s->out.format) {

  791.     case ALTERNATING_LR:

  792.     case ALTERNATING_RL:

  793.         switch (s->in.format) {

  794.         case INTERLEAVE_ROWS_LR:

  795.         case INTERLEAVE_ROWS_RL:

  796.             for (i = 0; i < s->nb_planes; i++) {

  797.                 oleft->linesize[i]  *= 2;

  798.                 oright->linesize[i] *= 2;

  799.             }

  800.         case ABOVE_BELOW_LR:

  801.         case ABOVE_BELOW_RL:

  802.         case ABOVE_BELOW_2_LR:

  803.         case ABOVE_BELOW_2_RL:

  804.         case SIDE_BY_SIDE_LR:

  805.         case SIDE_BY_SIDE_RL:

  806.         case SIDE_BY_SIDE_2_LR:

  807.         case SIDE_BY_SIDE_2_RL:

  808.             oleft->width   = outlink->w;

  809.             oright->width  = outlink->w;

  810.             oleft->height  = outlink->h;

  811.             oright->height = outlink->h;

  812. 

  813.             for (i = 0; i < s->nb_planes; i++) {

  814.                 oleft->data[i]  += s->in_off_left[i];

  815.                 oright->data[i] += s->in_off_right[i];

  816.             }

  817.             break;

  818.         default:

  819.             goto copy;

  820.             break;

  821.         }

  822.         break;

  823.     case HDMI:

  824.         for (i = 0; i < s->nb_planes; i++) {

  825.             int j, h = s->height >> ((i == 1 || i == 2) ? s->vsub : 0);

  826.             int b = (s->blanks) >> ((i == 1 || i == 2) ? s->vsub : 0);

  827. 

  828.             for (j = h; j < h + b; j++)

  829.                 memset(oleft->data[i] + j * s->linesize[i], 0, s->linesize[i]);

  830.         }

  831.     case SIDE_BY_SIDE_LR:

  832.     case SIDE_BY_SIDE_RL:

  833.     case SIDE_BY_SIDE_2_LR:

  834.     case SIDE_BY_SIDE_2_RL:

  835.     case ABOVE_BELOW_LR:

  836.     case ABOVE_BELOW_RL:

  837.     case ABOVE_BELOW_2_LR:

  838.     case ABOVE_BELOW_2_RL:

  839.     case INTERLEAVE_ROWS_LR:

  840.     case INTERLEAVE_ROWS_RL:

  841. copy:

  842.         if (s->in.format == INTERLEAVE_COLS_LR ||

  843.             s->in.format == INTERLEAVE_COLS_RL) {

  844.             for (i = 0; i < s->nb_planes; i++) {

  845.                 int d = (s->in.format & 1) != (s->out.format & 1);

  846. 

  847.                 interleave_cols_to_any(s, out_off_left,  i, ileft,  oleft,   d);

  848.                 interleave_cols_to_any(s, out_off_right, i, iright, oright, !d);

  849.             }

  850.         } else {

  851.             for (i = 0; i < s->nb_planes; i++) {

  852.                 av_image_copy_plane(oleft->data[i] + out_off_left[i],

  853.                                     oleft->linesize[i] * s->out.row_step,

  854.                                     ileft->data[i] + s->in_off_left[i],

  855.                                     ileft->linesize[i] * s->in.row_step,

  856.                                     s->linesize[i], s->pheight[i]);

  857.                 av_image_copy_plane(oright->data[i] + out_off_right[i],

  858.                                     oright->linesize[i] * s->out.row_step,

  859.                                     iright->data[i] + s->in_off_right[i],

  860.                                     iright->linesize[i] * s->in.row_step,

  861.                                     s->linesize[i], s->pheight[i]);

  862.             }

  863.         }

  864.         break;

  865.     case MONO_L:

  866.         iright = ileft;

  867.     case MONO_R:

  868.         switch (s->in.format) {

  869.         case INTERLEAVE_ROWS_LR:

  870.         case INTERLEAVE_ROWS_RL:

  871.             for (i = 0; i < s->nb_planes; i++) {

  872.                 out->linesize[i] *= 2;

  873.             }

  874.         case ABOVE_BELOW_LR:

  875.         case ABOVE_BELOW_RL:

  876.         case ABOVE_BELOW_2_LR:

  877.         case ABOVE_BELOW_2_RL:

  878.         case SIDE_BY_SIDE_LR:

  879.         case SIDE_BY_SIDE_RL:

  880.         case SIDE_BY_SIDE_2_LR:

  881.         case SIDE_BY_SIDE_2_RL:

  882.             out->width  = outlink->w;

  883.             out->height = outlink->h;

  884. 

  885.             for (i = 0; i < s->nb_planes; i++) {

  886.                 out->data[i] += s->in_off_left[i];

  887.             }

  888.             break;

  889.         case INTERLEAVE_COLS_LR:

  890.         case INTERLEAVE_COLS_RL:

  891.             for (i = 0; i < s->nb_planes; i++) {

  892.                 const int d = (s->in.format & 1) != (s->out.format & 1);

  893. 

  894.                 interleave_cols_to_any(s, out_off_right, i, iright, out, d);

  895.             }

  896.             break;

  897.         default:

  898.             for (i = 0; i < s->nb_planes; i++) {

  899.                 av_image_copy_plane(out->data[i], out->linesize[i],

  900.                                     iright->data[i] + s->in_off_left[i],

  901.                                     iright->linesize[i] * s->in.row_step,

  902.                                     s->linesize[i], s->pheight[i]);

  903.             }

  904.             break;

  905.         }

  906.         break;

  907.     case ANAGLYPH_RB_GRAY:

  908.     case ANAGLYPH_RG_GRAY:

  909.     case ANAGLYPH_RC_GRAY:

  910.     case ANAGLYPH_RC_HALF:

  911.     case ANAGLYPH_RC_COLOR:

  912.     case ANAGLYPH_RC_DUBOIS:

  913.     case ANAGLYPH_GM_GRAY:

  914.     case ANAGLYPH_GM_HALF:

  915.     case ANAGLYPH_GM_COLOR:

  916.     case ANAGLYPH_GM_DUBOIS:

  917.     case ANAGLYPH_YB_GRAY:

  918.     case ANAGLYPH_YB_HALF:

  919.     case ANAGLYPH_YB_COLOR:

  920.     case ANAGLYPH_YB_DUBOIS: {

  921.         if (s->in.format == INTERLEAVE_COLS_LR ||

  922.             s->in.format == INTERLEAVE_COLS_RL) {

  923.             const int d = (s->in.format & 1);

  924. 

  925.             anaglyph_ic(out->data[0],

  926.                 ileft ->data[0] + s->in_off_left [0] +   d  * 3,

  927.                 iright->data[0] + s->in_off_right[0] + (!d) * 3,

  928.                 out->linesize[0],

  929.                 ileft->linesize[0] * s->in.row_step,

  930.                 iright->linesize[0] * s->in.row_step,

  931.                 s->out.width, s->out.height,

  932.                 s->ana_matrix[0], s->ana_matrix[1], s->ana_matrix[2]);

  933.         } else {

  934.             ThreadData td;

  935. 

  936.             td.ileft = ileft; td.iright = iright; td.out = out;

  937.             ctx->internal->execute(ctx, filter_slice, &td, NULL,

  938.                                    FFMIN(s->out.height, ctx->graph->nb_threads));

  939.         }

  940.         break;

  941.     }

  942.     case CHECKERBOARD_RL:

  943.     case CHECKERBOARD_LR:

  944.         for (i = 0; i < s->nb_planes; i++) {

  945.             int x, y;

  946. 

  947.             for (y = 0; y < s->pheight[i]; y++) {

  948.                 uint8_t *dst = out->data[i] + out->linesize[i] * y;

  949.                 const int d1 = (s->in.format == INTERLEAVE_COLS_LR || s->in.format == INTERLEAVE_COLS_RL) && (s->in.format & 1) != (s->out.format & 1);

  950.                 const int d2 = (s->in.format == INTERLEAVE_COLS_LR || s->in.format == INTERLEAVE_COLS_RL) ? !d1 : 0;

  951.                 const int m = 1 + (s->in.format == INTERLEAVE_COLS_LR || s->in.format == INTERLEAVE_COLS_RL);

  952.                 uint8_t *left  = ileft->data[i]  + ileft->linesize[i]  * y + s->in_off_left[i]  + d1 * s->pixstep[i];

  953.                 uint8_t *right = iright->data[i] + iright->linesize[i] * y + s->in_off_right[i] + d2 * s->pixstep[i];

  954.                 int p, b;

  955. 

  956.                 if (s->out.format == CHECKERBOARD_RL && s->in.format != INTERLEAVE_COLS_LR && s->in.format != INTERLEAVE_COLS_RL)

  957.                     FFSWAP(uint8_t*, left, right);

  958.                 switch (s->pixstep[i]) {

  959.                 case 1:

  960.                     for (x = 0, b = 0, p = 0; x < s->linesize[i] * 2; x+=2, p++, b+=2) {

  961.                         dst[x  ] = (b&1) == (y&1) ? left[p*m] : right[p*m];

  962.                         dst[x+1] = (b&1) != (y&1) ? left[p*m] : right[p*m];

  963.                     }

  964.                     break;

  965.                 case 2:

  966.                     for (x = 0, b = 0, p = 0; x < s->linesize[i] * 2; x+=4, p+=2, b+=2) {

  967.                         AV_WN16(&dst[x  ], (b&1) == (y&1) ? AV_RN16(&left[p*m]) : AV_RN16(&right[p*m]));

  968.                         AV_WN16(&dst[x+2], (b&1) != (y&1) ? AV_RN16(&left[p*m]) : AV_RN16(&right[p*m]));

  969.                     }

  970.                     break;

  971.                 case 3:

  972.                     for (x = 0, b = 0, p = 0; x < s->linesize[i] * 2; x+=6, p+=3, b+=2) {

  973.                         AV_WB24(&dst[x  ], (b&1) == (y&1) ? AV_RB24(&left[p*m]) : AV_RB24(&right[p*m]));

  974.                         AV_WB24(&dst[x+3], (b&1) != (y&1) ? AV_RB24(&left[p*m]) : AV_RB24(&right[p*m]));

  975.                     }

  976.                     break;

  977.                 case 4:

  978.                     for (x = 0, b = 0, p = 0; x < s->linesize[i] * 2; x+=8, p+=4, b+=2) {

  979.                         AV_WN32(&dst[x  ], (b&1) == (y&1) ? AV_RN32(&left[p*m]) : AV_RN32(&right[p*m]));

  980.                         AV_WN32(&dst[x+4], (b&1) != (y&1) ? AV_RN32(&left[p*m]) : AV_RN32(&right[p*m]));

  981.                     }

  982.                     break;

  983.                 case 6:

  984.                     for (x = 0, b = 0, p = 0; x < s->linesize[i] * 2; x+=12, p+=6, b+=2) {

  985.                         AV_WB48(&dst[x  ], (b&1) == (y&1) ? AV_RB48(&left[p*m]) : AV_RB48(&right[p*m]));

  986.                         AV_WB48(&dst[x+6], (b&1) != (y&1) ? AV_RB48(&left[p*m]) : AV_RB48(&right[p*m]));

  987.                     }

  988.                     break;

  989.                 case 8:

  990.                     for (x = 0, b = 0, p = 0; x < s->linesize[i] * 2; x+=16, p+=8, b+=2) {

  991.                         AV_WN64(&dst[x  ], (b&1) == (y&1) ? AV_RN64(&left[p*m]) : AV_RN64(&right[p*m]));

  992.                         AV_WN64(&dst[x+8], (b&1) != (y&1) ? AV_RN64(&left[p*m]) : AV_RN64(&right[p*m]));

  993.                     }

  994.                     break;

  995.                 }

  996.             }

  997.         }

  998.         break;

  999.     case INTERLEAVE_COLS_LR:

  1000.     case INTERLEAVE_COLS_RL:

  1001.         for (i = 0; i < s->nb_planes; i++) {

  1002.             const int d = (s->in.format == INTERLEAVE_COLS_LR || s->in.format == INTERLEAVE_COLS_RL);

  1003.             const int m = 1 + d;

  1004.             int x, y;

  1005. 

  1006.             for (y = 0; y < s->pheight[i]; y++) {

  1007.                 uint8_t *dst = out->data[i] + out->linesize[i] * y;

  1008.                 uint8_t *left = ileft->data[i] + ileft->linesize[i] * y * s->in.row_step + s->in_off_left[i] + d * s->pixstep[i];

  1009.                 uint8_t *right = iright->data[i] + iright->linesize[i] * y * s->in.row_step + s->in_off_right[i];

  1010.                 int p, b;

  1011. 

  1012.                 if (s->out.format == INTERLEAVE_COLS_LR)

  1013.                     FFSWAP(uint8_t*, left, right);

  1014. 

  1015.                 switch (s->pixstep[i]) {

  1016.                 case 1:

  1017.                     for (x = 0, b = 0, p = 0; x < s->linesize[i] * 2; x+=2, p++, b+=2) {

  1018.                         dst[x  ] =   b&1  ? left[p*m] : right[p*m];

  1019.                         dst[x+1] = !(b&1) ? left[p*m] : right[p*m];

  1020.                     }

  1021.                     break;

  1022.                 case 2:

  1023.                     for (x = 0, b = 0, p = 0; x < s->linesize[i] * 2; x+=4, p+=2, b+=2) {

  1024.                         AV_WN16(&dst[x  ],   b&1  ? AV_RN16(&left[p*m]) : AV_RN16(&right[p*m]));

  1025.                         AV_WN16(&dst[x+2], !(b&1) ? AV_RN16(&left[p*m]) : AV_RN16(&right[p*m]));

  1026.                     }

  1027.                     break;

  1028.                 case 3:

  1029.                     for (x = 0, b = 0, p = 0; x < s->linesize[i] * 2; x+=6, p+=3, b+=2) {

  1030.                         AV_WB24(&dst[x  ],   b&1  ? AV_RB24(&left[p*m]) : AV_RB24(&right[p*m]));

  1031.                         AV_WB24(&dst[x+3], !(b&1) ? AV_RB24(&left[p*m]) : AV_RB24(&right[p*m]));

  1032.                     }

  1033.                     break;

  1034.                 case 4:

  1035.                     for (x = 0, b = 0, p = 0; x < s->linesize[i] * 2; x+=8, p+=4, b+=2) {

  1036.                         AV_WN32(&dst[x  ],   b&1  ? AV_RN32(&left[p*m]) : AV_RN32(&right[p*m]));

  1037.                         AV_WN32(&dst[x+4], !(b&1) ? AV_RN32(&left[p*m]) : AV_RN32(&right[p*m]));

  1038.                     }

  1039.                     break;

  1040.                 case 6:

  1041.                     for (x = 0, b = 0, p = 0; x < s->linesize[i] * 2; x+=12, p+=6, b+=2) {

  1042.                         AV_WB48(&dst[x  ],   b&1  ? AV_RB48(&left[p*m]) : AV_RB48(&right[p*m]));

  1043.                         AV_WB48(&dst[x+6], !(b&1) ? AV_RB48(&left[p*m]) : AV_RB48(&right[p*m]));

  1044.                     }

  1045.                     break;

  1046.                 case 8:

  1047.                     for (x = 0, b = 0, p = 0; x < s->linesize[i] * 2; x+=16, p+=8, b+=2) {

  1048.                         AV_WN64(&dst[x  ],   b&1 ?  AV_RN64(&left[p*m]) : AV_RN64(&right[p*m]));

  1049.                         AV_WN64(&dst[x+8], !(b&1) ? AV_RN64(&left[p*m]) : AV_RN64(&right[p*m]));

  1050.                     }

  1051.                     break;

  1052.                 }

  1053.             }

  1054.         }

  1055.         break;

  1056.     default:

  1057.         av_assert0(0);

  1058.     }

  1059. 

  1060.     if (oright != oleft) {

  1061.         if (s->out.format == ALTERNATING_LR)

  1062.             FFSWAP(AVFrame *, oleft, oright);

  1063.         oright->pts = s->prev->pts * 2;

  1064.         ff_filter_frame(outlink, oright);

  1065.         out = oleft;

  1066.         oleft->pts = s->prev->pts + inpicref->pts;

  1067.         av_frame_free(&s->prev);

  1068.         s->prev = inpicref;

  1069.     } else if (s->in.format == ALTERNATING_LR ||

  1070.                s->in.format == ALTERNATING_RL) {

  1071.         out->pts = s->prev->pts / 2;

  1072.         av_frame_free(&s->prev);

  1073.         av_frame_free(&inpicref);

  1074.     } else {

  1075.         av_frame_free(&s->prev);

  1076.         av_frame_free(&inpicref);

  1077.     }

  1078.     return ff_filter_frame(outlink, out);

  1079. }

  1080. 

  1081. static av_cold void uninit(AVFilterContext *ctx)

  1082. {

  1083.     Stereo3DContext *s = ctx->priv;

  1084. 

  1085.     av_frame_free(&s->prev);

  1086. }

  1087. 

  1088. static const AVFilterPad stereo3d_inputs[] = {

  1089.     {

  1090.         .name         = "default",

  1091.         .type         = AVMEDIA_TYPE_VIDEO,

  1092.         .filter_frame = filter_frame,

  1093.     },

  1094.     { NULL }

  1095. };

  1096. 

  1097. static const AVFilterPad stereo3d_outputs[] = {

  1098.     {

  1099.         .name         = "default",

  1100.         .type         = AVMEDIA_TYPE_VIDEO,

  1101.         .config_props = config_output,

  1102.     },

  1103.     { NULL }

  1104. };

  1105. 

  1106. AVFilter ff_vf_stereo3d = {

  1107.     .name          = "stereo3d",

  1108.     .description   = NULL_IF_CONFIG_SMALL("Convert video stereoscopic 3D view."),

  1109.     .priv_size     = sizeof(Stereo3DContext),

  1110.     .uninit        = uninit,

  1111.     .query_formats = query_formats,

  1112.     .inputs        = stereo3d_inputs,

  1113.     .outputs       = stereo3d_outputs,

  1114.     .priv_class    = &stereo3d_class,

  1115.     .flags         = AVFILTER_FLAG_SLICE_THREADS,

  1116. };