git.sesse.net Git - ffmpeg/blob - libavfilter/v360.h

   1 /*
   2  * Copyright (c) 2019 Eugene Lyapustin
   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 #ifndef AVFILTER_V360_H
  22 #define AVFILTER_V360_H
  23 #include "avfilter.h"
  24
  25 enum StereoFormats {
  26     STEREO_2D,
  27     STEREO_SBS,
  28     STEREO_TB,
  29     NB_STEREO_FMTS,
  30 };
  31
  32 enum Projections {
  33     EQUIRECTANGULAR,
  34     CUBEMAP_3_2,
  35     CUBEMAP_6_1,
  36     EQUIANGULAR,
  37     FLAT,
  38     DUAL_FISHEYE,
  39     BARREL,
  40     CUBEMAP_1_6,
  41     STEREOGRAPHIC,
  42     MERCATOR,
  43     NB_PROJECTIONS,
  44 };
  45
  46 enum InterpMethod {
  47     NEAREST,
  48     BILINEAR,
  49     BICUBIC,
  50     LANCZOS,
  51     NB_INTERP_METHODS,
  52 };
  53
  54 enum Faces {
  55     TOP_LEFT,
  56     TOP_MIDDLE,
  57     TOP_RIGHT,
  58     BOTTOM_LEFT,
  59     BOTTOM_MIDDLE,
  60     BOTTOM_RIGHT,
  61     NB_FACES,
  62 };
  63
  64 enum Direction {
  65     RIGHT,  ///< Axis +X
  66     LEFT,   ///< Axis -X
  67     UP,     ///< Axis +Y
  68     DOWN,   ///< Axis -Y
  69     FRONT,  ///< Axis -Z
  70     BACK,   ///< Axis +Z
  71     NB_DIRECTIONS,
  72 };
  73
  74 enum Rotation {
  75     ROT_0,
  76     ROT_90,
  77     ROT_180,
  78     ROT_270,
  79     NB_ROTATIONS,
  80 };
  81
  82 enum RotationOrder {
  83     YAW,
  84     PITCH,
  85     ROLL,
  86     NB_RORDERS,
  87 };
  88
  89 typedef struct XYRemap {
  90     uint16_t u[4][4];
  91     uint16_t v[4][4];
  92     float ker[4][4];
  93 } XYRemap;
  94
  95 typedef struct V360Context {
  96     const AVClass *class;
  97     int in, out;
  98     int interp;
  99     int width, height;
 100     char *in_forder;
 101     char *out_forder;
 102     char *in_frot;
 103     char *out_frot;
 104     char *rorder;
 105
 106     int in_cubemap_face_order[6];
 107     int out_cubemap_direction_order[6];
 108     int in_cubemap_face_rotation[6];
 109     int out_cubemap_face_rotation[6];
 110     int rotation_order[3];
 111
 112     int in_stereo, out_stereo;
 113
 114     float in_pad, out_pad;
 115
 116     float yaw, pitch, roll;
 117
 118     int ih_flip, iv_flip;
 119     int h_flip, v_flip, d_flip;
 120     int in_transpose, out_transpose;
 121
 122     float h_fov, v_fov, d_fov;
 123     float flat_range[3];
 124
 125     float rot_mat[3][3];
 126
 127     float input_mirror_modifier[2];
 128     float output_mirror_modifier[3];
 129
 130     int pr_width[4], pr_height[4];
 131
 132     int in_offset_w[4], in_offset_h[4];
 133     int out_offset_w[4], out_offset_h[4];
 134
 135     int planewidth[4], planeheight[4];
 136     int inplanewidth[4], inplaneheight[4];
 137     int uv_linesize[4];
 138     int nb_planes;
 139     int nb_allocated;
 140     int elements;
 141
 142     uint16_t *u[4], *v[4];
 143     int16_t *ker[4];
 144     unsigned map[4];
 145
 146     void (*in_transform)(const struct V360Context *s,
 147                          const float *vec, int width, int height,
 148                          uint16_t us[4][4], uint16_t vs[4][4], float *du, float *dv);
 149
 150     void (*out_transform)(const struct V360Context *s,
 151                           int i, int j, int width, int height,
 152                           float *vec);
 153
 154     void (*calculate_kernel)(float du, float dv, const XYRemap *rmap,
 155                              uint16_t *u, uint16_t *v, int16_t *ker);
 156
 157     int (*remap_slice)(AVFilterContext *ctx, void *arg, int jobnr, int nb_jobs);
 158
 159     void (*remap_line)(uint8_t *dst, int width, const uint8_t *src, ptrdiff_t in_linesize,
 160                        const uint16_t *u, const uint16_t *v, const int16_t *ker);
 161 } V360Context;
 162
 163 void ff_v360_init(V360Context *s, int depth);
 164 void ff_v360_init_x86(V360Context *s, int depth);
 165
 166 #endif /* AVFILTER_V360_H */