libavutil/stereo3d.c

   1 /*
   2  * Copyright (c) 2013 Vittorio Giovara <vittorio.giovara@gmail.com>
   3  *
   4  * This file is part of FFmpeg.
   5  *
   6  * FFmpeg is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * FFmpeg is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with FFmpeg; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  19  */
  20
  21 #include <string.h>
  22
  23 #include "avstring.h"
  24 #include "frame.h"
  25 #include "macros.h"
  26 #include "mem.h"
  27 #include "stereo3d.h"
  28
  29 static void get_defaults(AVStereo3D *stereo)
  30 {
  31     stereo->horizontal_disparity_adjustment = (AVRational) { 0, 1 };
  32     stereo->horizontal_field_of_view = (AVRational) { 0, 1 };
  33 }
  34
  35 AVStereo3D *av_stereo3d_alloc(void)
  36 {
  37     return av_stereo3d_alloc_size(NULL);
  38 }
  39
  40 AVStereo3D *av_stereo3d_alloc_size(size_t *size)
  41 {
  42     AVStereo3D *stereo = av_mallocz(sizeof(AVStereo3D));
  43     if (!stereo)
  44         return NULL;
  45
  46     get_defaults(stereo);
  47
  48     if (size)
  49         *size = sizeof(*stereo);
  50
  51     return stereo;
  52 }
  53
  54 AVStereo3D *av_stereo3d_create_side_data(AVFrame *frame)
  55 {
  56     AVFrameSideData *side_data = av_frame_new_side_data(frame,
  57                                                         AV_FRAME_DATA_STEREO3D,
  58                                                         sizeof(AVStereo3D));
  59     if (!side_data)
  60         return NULL;
  61
  62     memset(side_data->data, 0, sizeof(AVStereo3D));
  63     get_defaults((AVStereo3D *)side_data->data);
  64
  65     return (AVStereo3D *)side_data->data;
  66 }
  67
  68 static const char * const stereo3d_type_names[] = {
  69     [AV_STEREO3D_2D]                  = "2D",
  70     [AV_STEREO3D_SIDEBYSIDE]          = "side by side",
  71     [AV_STEREO3D_TOPBOTTOM]           = "top and bottom",
  72     [AV_STEREO3D_FRAMESEQUENCE]       = "frame alternate",
  73     [AV_STEREO3D_CHECKERBOARD]        = "checkerboard",
  74     [AV_STEREO3D_SIDEBYSIDE_QUINCUNX] = "side by side (quincunx subsampling)",
  75     [AV_STEREO3D_LINES]               = "interleaved lines",
  76     [AV_STEREO3D_COLUMNS]             = "interleaved columns",
  77     [AV_STEREO3D_UNSPEC]              = "unspecified",
  78 };
  79
  80 static const char * const stereo3d_view_names[] = {
  81     [AV_STEREO3D_VIEW_PACKED] = "packed",
  82     [AV_STEREO3D_VIEW_LEFT]   = "left",
  83     [AV_STEREO3D_VIEW_RIGHT]  = "right",
  84     [AV_STEREO3D_VIEW_UNSPEC] = "unspecified",
  85 };
  86
  87 static const char * const stereo3d_primary_eye_names[] = {
  88     [AV_PRIMARY_EYE_NONE]  = "none",
  89     [AV_PRIMARY_EYE_LEFT]  = "left",
  90     [AV_PRIMARY_EYE_RIGHT] = "right",
  91 };
  92
  93 const char *av_stereo3d_type_name(unsigned int type)
  94 {
  95     if (type >= FF_ARRAY_ELEMS(stereo3d_type_names))
  96         return "unknown";
  97
  98     return stereo3d_type_names[type];
  99 }
 100
 101 int av_stereo3d_from_name(const char *name)
 102 {
 103     int i;
 104
 105     for (i = 0; i < FF_ARRAY_ELEMS(stereo3d_type_names); i++) {
 106         if (av_strstart(name, stereo3d_type_names[i], NULL))
 107             return i;
 108     }
 109
 110     return -1;
 111 }
 112
 113 const char *av_stereo3d_view_name(unsigned int view)
 114 {
 115     if (view >= FF_ARRAY_ELEMS(stereo3d_view_names))
 116         return "unknown";
 117
 118     return stereo3d_view_names[view];
 119 }
 120
 121 int av_stereo3d_view_from_name(const char *name)
 122 {
 123     int i;
 124
 125     for (i = 0; i < FF_ARRAY_ELEMS(stereo3d_view_names); i++) {
 126         if (av_strstart(name, stereo3d_view_names[i], NULL))
 127             return i;
 128     }
 129
 130     return -1;
 131 }
 132
 133 const char *av_stereo3d_primary_eye_name(unsigned int eye)
 134 {
 135     if (eye >= FF_ARRAY_ELEMS(stereo3d_primary_eye_names))
 136         return "unknown";
 137
 138     return stereo3d_primary_eye_names[eye];
 139 }
 140
 141 int av_stereo3d_primary_eye_from_name(const char *name)
 142 {
 143     int i;
 144
 145     for (i = 0; i < FF_ARRAY_ELEMS(stereo3d_primary_eye_names); i++) {
 146         if (av_strstart(name, stereo3d_primary_eye_names[i], NULL))
 147             return i;
 148     }
 149
 150     return -1;
 151 }