libavcodec/libschroedinger.c

   1 /*
   2  * Copyright (c) 2008 BBC, Anuradha Suraparaju <asuraparaju at gmail dot com >
   3  *
   4  * This file is part of Libav.
   5  *
   6  * Libav 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  * Libav 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 Libav; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  19  */
  20
  21 /**
  22 * @file
  23 * function definitions common to libschroedingerdec.c and libschroedingerenc.c
  24 */
  25
  26 #include "libdirac_libschro.h"
  27 #include "libschroedinger.h"
  28
  29 /**
  30 * Schroedinger video preset table. Ensure that this tables matches up correctly
  31 * with the ff_dirac_schro_video_format_info table in libdirac_libschro.c.
  32 */
  33 static const SchroVideoFormatEnum ff_schro_video_formats[]={
  34     SCHRO_VIDEO_FORMAT_CUSTOM     ,
  35     SCHRO_VIDEO_FORMAT_QSIF       ,
  36     SCHRO_VIDEO_FORMAT_QCIF       ,
  37     SCHRO_VIDEO_FORMAT_SIF        ,
  38     SCHRO_VIDEO_FORMAT_CIF        ,
  39     SCHRO_VIDEO_FORMAT_4SIF       ,
  40     SCHRO_VIDEO_FORMAT_4CIF       ,
  41     SCHRO_VIDEO_FORMAT_SD480I_60  ,
  42     SCHRO_VIDEO_FORMAT_SD576I_50  ,
  43     SCHRO_VIDEO_FORMAT_HD720P_60  ,
  44     SCHRO_VIDEO_FORMAT_HD720P_50  ,
  45     SCHRO_VIDEO_FORMAT_HD1080I_60 ,
  46     SCHRO_VIDEO_FORMAT_HD1080I_50 ,
  47     SCHRO_VIDEO_FORMAT_HD1080P_60 ,
  48     SCHRO_VIDEO_FORMAT_HD1080P_50 ,
  49     SCHRO_VIDEO_FORMAT_DC2K_24    ,
  50     SCHRO_VIDEO_FORMAT_DC4K_24    ,
  51 };
  52
  53 SchroVideoFormatEnum ff_get_schro_video_format_preset(AVCodecContext *avccontext)
  54 {
  55     unsigned int num_formats = sizeof(ff_schro_video_formats) /
  56                                sizeof(ff_schro_video_formats[0]);
  57
  58     unsigned int idx = ff_dirac_schro_get_video_format_idx (avccontext);
  59
  60     return (idx < num_formats) ? ff_schro_video_formats[idx] :
  61                                  SCHRO_VIDEO_FORMAT_CUSTOM;
  62 }
  63
  64 int ff_get_schro_frame_format (SchroChromaFormat schro_pix_fmt,
  65                                SchroFrameFormat  *schro_frame_fmt)
  66 {
  67     unsigned int num_formats = sizeof(ffmpeg_schro_pixel_format_map) /
  68                                sizeof(ffmpeg_schro_pixel_format_map[0]);
  69
  70     int idx;
  71
  72     for (idx = 0; idx < num_formats; ++idx) {
  73         if (ffmpeg_schro_pixel_format_map[idx].schro_pix_fmt == schro_pix_fmt) {
  74             *schro_frame_fmt = ffmpeg_schro_pixel_format_map[idx].schro_frame_fmt;
  75             return 0;
  76         }
  77     }
  78     return -1;
  79 }
  80
  81 static void FreeSchroFrame(SchroFrame *frame, void *priv)
  82 {
  83     AVPicture *p_pic = priv;
  84
  85     if (!p_pic)
  86         return;
  87
  88     avpicture_free(p_pic);
  89     av_freep(&p_pic);
  90 }
  91
  92 SchroFrame *ff_create_schro_frame(AVCodecContext *avccontext,
  93                                   SchroFrameFormat schro_frame_fmt)
  94 {
  95     AVPicture *p_pic;
  96     SchroFrame *p_frame;
  97     int y_width, uv_width;
  98     int y_height, uv_height;
  99     int i;
 100
 101     y_width   = avccontext->width;
 102     y_height  = avccontext->height;
 103     uv_width  = y_width  >> (SCHRO_FRAME_FORMAT_H_SHIFT(schro_frame_fmt));
 104     uv_height = y_height >> (SCHRO_FRAME_FORMAT_V_SHIFT(schro_frame_fmt));
 105
 106     p_pic = av_mallocz(sizeof(AVPicture));
 107     avpicture_alloc(p_pic, avccontext->pix_fmt, y_width, y_height);
 108
 109     p_frame         = schro_frame_new();
 110     p_frame->format = schro_frame_fmt;
 111     p_frame->width  = y_width;
 112     p_frame->height = y_height;
 113     schro_frame_set_free_callback(p_frame, FreeSchroFrame, (void *)p_pic);
 114
 115     for (i = 0; i < 3; ++i) {
 116         p_frame->components[i].width  = i ? uv_width : y_width;
 117         p_frame->components[i].stride = p_pic->linesize[i];
 118         p_frame->components[i].height = i ? uv_height : y_height;
 119         p_frame->components[i].length =
 120                  p_frame->components[i].stride * p_frame->components[i].height;
 121         p_frame->components[i].data   = p_pic->data[i];
 122
 123         if (i) {
 124             p_frame->components[i].v_shift =
 125                 SCHRO_FRAME_FORMAT_V_SHIFT(p_frame->format);
 126             p_frame->components[i].h_shift =
 127                 SCHRO_FRAME_FORMAT_H_SHIFT(p_frame->format);
 128         }
 129     }
 130
 131     return p_frame;
 132 }