QTfrontend/util/LibavInteraction.cpp
author Vittorio Giovara <vittorio.giovara@gmail.com>
Wed, 04 Nov 2015 16:20:46 +0100
branch0.9.22
changeset 11296 2f812d4513d1
parent 11295 210650fc7182
child 11356 ac423fd2b9a3
permissions -rw-r--r--
Specify the enum type for Pixel and Sample formats
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
7897
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
     1
/*
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
     2
 * Hedgewars, a free turn based strategy game
11046
47a8c19ecb60 more copyright fixes
sheepluva
parents: 10108
diff changeset
     3
 * Copyright (c) 2004-2015 Andrey Korotaev <unC0Rr@gmail.com>
7897
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
     4
 *
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
     5
 * This program is free software; you can redistribute it and/or modify
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
     6
 * it under the terms of the GNU General Public License as published by
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
     7
 * the Free Software Foundation; version 2 of the License
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
     8
 *
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
     9
 * This program is distributed in the hope that it will be useful,
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    10
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    11
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    12
 * GNU General Public License for more details.
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    13
 *
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    14
 * You should have received a copy of the GNU General Public License
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    15
 * along with this program; if not, write to the Free Software
10108
c68cf030eded update FSF address. note: two sdl include files (by Sam Lantinga) still have the old FSF address in their copyright - but I ain't gonna touch their copyright headers
sheepluva
parents: 9998
diff changeset
    16
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
7897
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    17
 */
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    18
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    19
#include "LibavInteraction.h"
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    20
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    21
#if VIDEOREC
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    22
extern "C"
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    23
{
11295
210650fc7182 Update libav API, keeping forward and backward compatibility
Vittorio Giovara <vittorio.giovara@gmail.com>
parents: 11046
diff changeset
    24
#include "libavcodec/avcodec.h"
7897
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    25
#include "libavformat/avformat.h"
11295
210650fc7182 Update libav API, keeping forward and backward compatibility
Vittorio Giovara <vittorio.giovara@gmail.com>
parents: 11046
diff changeset
    26
#include "libavutil/avutil.h"
7897
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    27
}
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    28
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    29
#include <QVector>
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    30
#include <QList>
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    31
#include <QComboBox>
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    32
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    33
#include "HWApplication.h"
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    34
11295
210650fc7182 Update libav API, keeping forward and backward compatibility
Vittorio Giovara <vittorio.giovara@gmail.com>
parents: 11046
diff changeset
    35
// compatibility section
210650fc7182 Update libav API, keeping forward and backward compatibility
Vittorio Giovara <vittorio.giovara@gmail.com>
parents: 11046
diff changeset
    36
#if LIBAVCODEC_VERSION_MAJOR < 55
210650fc7182 Update libav API, keeping forward and backward compatibility
Vittorio Giovara <vittorio.giovara@gmail.com>
parents: 11046
diff changeset
    37
#define av_codec_is_encoder(x)          x->encode
210650fc7182 Update libav API, keeping forward and backward compatibility
Vittorio Giovara <vittorio.giovara@gmail.com>
parents: 11046
diff changeset
    38
#define AVCodecID                       CodecID
210650fc7182 Update libav API, keeping forward and backward compatibility
Vittorio Giovara <vittorio.giovara@gmail.com>
parents: 11046
diff changeset
    39
#endif
210650fc7182 Update libav API, keeping forward and backward compatibility
Vittorio Giovara <vittorio.giovara@gmail.com>
parents: 11046
diff changeset
    40
210650fc7182 Update libav API, keeping forward and backward compatibility
Vittorio Giovara <vittorio.giovara@gmail.com>
parents: 11046
diff changeset
    41
#if LIBAVFORMAT_VERSION_MAJOR < 54
210650fc7182 Update libav API, keeping forward and backward compatibility
Vittorio Giovara <vittorio.giovara@gmail.com>
parents: 11046
diff changeset
    42
#define avformat_find_stream_info(x, y) av_find_stream_info(x)
210650fc7182 Update libav API, keeping forward and backward compatibility
Vittorio Giovara <vittorio.giovara@gmail.com>
parents: 11046
diff changeset
    43
#define avformat_close_input(x)         av_close_input_file(*(x))
210650fc7182 Update libav API, keeping forward and backward compatibility
Vittorio Giovara <vittorio.giovara@gmail.com>
parents: 11046
diff changeset
    44
#endif
210650fc7182 Update libav API, keeping forward and backward compatibility
Vittorio Giovara <vittorio.giovara@gmail.com>
parents: 11046
diff changeset
    45
210650fc7182 Update libav API, keeping forward and backward compatibility
Vittorio Giovara <vittorio.giovara@gmail.com>
parents: 11046
diff changeset
    46
#if LIBAVUTIL_VERSION_MAJOR < 54
210650fc7182 Update libav API, keeping forward and backward compatibility
Vittorio Giovara <vittorio.giovara@gmail.com>
parents: 11046
diff changeset
    47
#define AVPixelFormat                   PixelFormat
210650fc7182 Update libav API, keeping forward and backward compatibility
Vittorio Giovara <vittorio.giovara@gmail.com>
parents: 11046
diff changeset
    48
#define AV_PIX_FMT_YUV420P              PIX_FMT_YUV420P
9429
7a97a554ac80 libavinteraction: fix typo and and compiling with modern libav
koda
parents: 9107
diff changeset
    49
#endif
7a97a554ac80 libavinteraction: fix typo and and compiling with modern libav
koda
parents: 9107
diff changeset
    50
7897
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    51
struct Codec
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    52
{
11295
210650fc7182 Update libav API, keeping forward and backward compatibility
Vittorio Giovara <vittorio.giovara@gmail.com>
parents: 11046
diff changeset
    53
    AVCodecID id;
7897
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    54
    bool isAudio;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    55
    QString shortName; // used for identification
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    56
    QString longName; // used for displaying to user
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    57
    bool isRecomended;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    58
};
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    59
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    60
struct Format
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    61
{
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    62
    QString shortName;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    63
    QString longName;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    64
    bool isRecomended;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    65
    QString extension;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    66
    QVector<Codec*> codecs;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    67
};
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    68
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    69
QList<Codec> codecs;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    70
QMap<QString,Format> formats;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    71
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    72
// test if given format supports given codec
11295
210650fc7182 Update libav API, keeping forward and backward compatibility
Vittorio Giovara <vittorio.giovara@gmail.com>
parents: 11046
diff changeset
    73
bool FormatQueryCodec(AVOutputFormat *ofmt, enum AVCodecID codec_id)
7897
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    74
{
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    75
#if LIBAVFORMAT_VERSION_MAJOR >= 54
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    76
    return avformat_query_codec(ofmt, codec_id, FF_COMPLIANCE_NORMAL) == 1;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    77
#else
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    78
    if (ofmt->codec_tag)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    79
        return !!av_codec_get_tag(ofmt->codec_tag, codec_id);
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    80
    return codec_id == ofmt->video_codec || codec_id == ofmt->audio_codec;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    81
#endif
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    82
}
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    83
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    84
LibavInteraction::LibavInteraction() : QObject()
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    85
{
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    86
    // initialize libav and register all codecs and formats
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    87
    av_register_all();
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    88
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    89
    // get list of all codecs
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    90
    AVCodec* pCodec = NULL;
8065
609d07b337b5 fix deprecation warnings for libav sources
koda
parents: 7897
diff changeset
    91
    while ((pCodec = av_codec_next(pCodec)))
7897
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    92
    {
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    93
        if (!av_codec_is_encoder(pCodec))
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    94
            continue;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    95
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    96
        if (pCodec->type != AVMEDIA_TYPE_VIDEO && pCodec->type != AVMEDIA_TYPE_AUDIO)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    97
            continue;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    98
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
    99
        // this encoders seems to be buggy
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   100
        if (strcmp(pCodec->name, "rv10") == 0 || strcmp(pCodec->name, "rv20") == 0)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   101
            continue;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   102
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   103
        // doesn't support stereo sound
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   104
        if (strcmp(pCodec->name, "real_144") == 0)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   105
            continue;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   106
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   107
        if (!pCodec->long_name || strlen(pCodec->long_name) == 0)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   108
            continue;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   109
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   110
        if (pCodec->type == AVMEDIA_TYPE_VIDEO)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   111
        {
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   112
            if (pCodec->supported_framerates != NULL)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   113
                continue;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   114
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   115
            // check if codec supports yuv 4:2:0 format
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   116
            if (!pCodec->pix_fmts)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   117
                continue;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   118
            bool yuv420Supported = false;
11296
2f812d4513d1 Specify the enum type for Pixel and Sample formats
Vittorio Giovara <vittorio.giovara@gmail.com>
parents: 11295
diff changeset
   119
            for (const enum AVPixelFormat* pfmt = pCodec->pix_fmts; *pfmt != -1; pfmt++)
11295
210650fc7182 Update libav API, keeping forward and backward compatibility
Vittorio Giovara <vittorio.giovara@gmail.com>
parents: 11046
diff changeset
   120
                if (*pfmt == AV_PIX_FMT_YUV420P)
7897
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   121
                {
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   122
                    yuv420Supported = true;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   123
                    break;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   124
                }
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   125
            if (!yuv420Supported)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   126
                continue;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   127
        }
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   128
        if (pCodec->type == AVMEDIA_TYPE_AUDIO)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   129
        {
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   130
            // check if codec supports signed 16-bit format
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   131
            if (!pCodec->sample_fmts)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   132
                continue;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   133
            bool s16Supported = false;
11296
2f812d4513d1 Specify the enum type for Pixel and Sample formats
Vittorio Giovara <vittorio.giovara@gmail.com>
parents: 11295
diff changeset
   134
            for (const enum AVSampleFormat* pfmt = pCodec->sample_fmts; *pfmt != -1; pfmt++)
7897
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   135
                if (*pfmt == AV_SAMPLE_FMT_S16)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   136
                {
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   137
                    s16Supported = true;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   138
                    break;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   139
                }
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   140
            if (!s16Supported)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   141
                continue;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   142
        }
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   143
        // add codec to list of codecs
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   144
        codecs.push_back(Codec());
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   145
        Codec & codec = codecs.back();
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   146
        codec.id = pCodec->id;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   147
        codec.isAudio = pCodec->type == AVMEDIA_TYPE_AUDIO;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   148
        codec.shortName = pCodec->name;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   149
        codec.longName = pCodec->long_name;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   150
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   151
        codec.isRecomended = false;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   152
        if (strcmp(pCodec->name, "libx264") == 0)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   153
        {
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   154
            codec.longName = "H.264/MPEG-4 Part 10 AVC (x264)";
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   155
            codec.isRecomended = true;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   156
        }
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   157
        else if (strcmp(pCodec->name, "libxvid") == 0)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   158
        {
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   159
            codec.longName = "MPEG-4 Part 2 (Xvid)";
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   160
            codec.isRecomended = true;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   161
        }
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   162
        else if (strcmp(pCodec->name, "libmp3lame") == 0)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   163
        {
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   164
            codec.longName = "MP3 (MPEG audio layer 3) (LAME)";
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   165
            codec.isRecomended = true;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   166
        }
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   167
        else
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   168
            codec.longName = pCodec->long_name;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   169
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   170
        if (strcmp(pCodec->name, "mpeg4") == 0 || strcmp(pCodec->name, "ac3_fixed") == 0)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   171
            codec.isRecomended = true;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   172
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   173
        // FIXME: remove next line
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   174
        //codec.longName += QString(" (%1)").arg(codec.shortName);
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   175
    }
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   176
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   177
    // get list of all formats
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   178
    AVOutputFormat* pFormat = NULL;
8065
609d07b337b5 fix deprecation warnings for libav sources
koda
parents: 7897
diff changeset
   179
    while ((pFormat = av_oformat_next(pFormat)))
7897
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   180
    {
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   181
        if (!pFormat->extensions)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   182
            continue;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   183
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   184
        // skip some strange formats to not confuse users
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   185
        if (strstr(pFormat->long_name, "raw"))
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   186
            continue;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   187
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   188
        Format format;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   189
        bool hasVideoCodec = false;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   190
        for (QList<Codec>::iterator codec = codecs.begin(); codec != codecs.end(); ++codec)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   191
        {
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   192
            if (!FormatQueryCodec(pFormat, codec->id))
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   193
                continue;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   194
            format.codecs.push_back(&*codec);
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   195
            if (!codec->isAudio)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   196
                hasVideoCodec = true;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   197
        }
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   198
        if (!hasVideoCodec)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   199
            continue;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   200
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   201
        QString ext(pFormat->extensions);
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   202
        ext.truncate(strcspn(pFormat->extensions, ","));
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   203
        format.extension = ext;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   204
        format.shortName = pFormat->name;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   205
        format.longName = QString("%1 (*.%2)").arg(pFormat->long_name).arg(ext);
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   206
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   207
        // FIXME: remove next line
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   208
        //format.longName += QString(" (%1)").arg(format.shortName);
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   209
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   210
        format.isRecomended = strcmp(pFormat->name, "mp4") == 0 || strcmp(pFormat->name, "avi") == 0;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   211
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   212
        formats[pFormat->name] = format;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   213
    }
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   214
}
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   215
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   216
void LibavInteraction::fillFormats(QComboBox * pFormats)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   217
{
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   218
    // first insert recomended formats
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   219
    foreach(const Format & format, formats)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   220
        if (format.isRecomended)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   221
            pFormats->addItem(format.longName, format.shortName);
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   222
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   223
    // remember where to place separator between recomended and other formats
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   224
    int sep = pFormats->count();
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   225
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   226
    // insert remaining formats
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   227
    foreach(const Format & format, formats)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   228
        if (!format.isRecomended)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   229
            pFormats->addItem(format.longName, format.shortName);
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   230
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   231
    // insert separator if necessary
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   232
    if (sep != 0 && sep != pFormats->count())
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   233
        pFormats->insertSeparator(sep);
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   234
}
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   235
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   236
void LibavInteraction::fillCodecs(const QString & fmt, QComboBox * pVCodecs, QComboBox * pACodecs)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   237
{
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   238
    Format & format = formats[fmt];
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   239
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   240
    // first insert recomended codecs
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   241
    foreach(Codec * codec, format.codecs)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   242
    {
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   243
        if (codec->isRecomended)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   244
        {
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   245
            if (codec->isAudio)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   246
                pACodecs->addItem(codec->longName, codec->shortName);
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   247
            else
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   248
                pVCodecs->addItem(codec->longName, codec->shortName);
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   249
        }
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   250
    }
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   251
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   252
    // remember where to place separators between recomended and other codecs
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   253
    int vsep = pVCodecs->count();
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   254
    int asep = pACodecs->count();
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   255
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   256
    // insert remaining codecs
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   257
    foreach(Codec * codec, format.codecs)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   258
    {
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   259
        if (!codec->isRecomended)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   260
        {
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   261
            if (codec->isAudio)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   262
                pACodecs->addItem(codec->longName, codec->shortName);
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   263
            else
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   264
                pVCodecs->addItem(codec->longName, codec->shortName);
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   265
        }
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   266
    }
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   267
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   268
    // insert separators if necessary
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   269
    if (vsep != 0 && vsep != pVCodecs->count())
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   270
        pVCodecs->insertSeparator(vsep);
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   271
    if (asep != 0 && asep != pACodecs->count())
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   272
        pACodecs->insertSeparator(asep);
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   273
}
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   274
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   275
QString LibavInteraction::getExtension(const QString & format)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   276
{
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   277
    return formats[format].extension;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   278
}
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   279
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   280
// get information abaout file (duration, resolution etc) in multiline string
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   281
QString LibavInteraction::getFileInfo(const QString & filepath)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   282
{
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   283
    AVFormatContext* pContext = NULL;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   284
    QByteArray utf8path = filepath.toUtf8();
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   285
    if (avformat_open_input(&pContext, utf8path.data(), NULL, NULL) < 0)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   286
        return "";
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   287
    if (avformat_find_stream_info(pContext, NULL) < 0)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   288
        return "";
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   289
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   290
    int s = float(pContext->duration)/AV_TIME_BASE;
9107
4dde5fecffe2 Less confusion to translators
unc0rr
parents: 9080
diff changeset
   291
    QString desc = tr("Duration: %1m %2s").arg(s/60).arg(s%60) + "\n";
7897
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   292
    for (int i = 0; i < (int)pContext->nb_streams; i++)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   293
    {
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   294
        AVStream* pStream = pContext->streams[i];
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   295
        if (!pStream)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   296
            continue;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   297
        AVCodecContext* pCodec = pContext->streams[i]->codec;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   298
        if (!pCodec)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   299
            continue;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   300
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   301
        if (pCodec->codec_type == AVMEDIA_TYPE_VIDEO)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   302
        {
9107
4dde5fecffe2 Less confusion to translators
unc0rr
parents: 9080
diff changeset
   303
            desc += QString(tr("Video: %1x%2")).arg(pCodec->width).arg(pCodec->height) + ", ";
7897
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   304
            if (pStream->avg_frame_rate.den)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   305
            {
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   306
                float fps = float(pStream->avg_frame_rate.num)/pStream->avg_frame_rate.den;
9107
4dde5fecffe2 Less confusion to translators
unc0rr
parents: 9080
diff changeset
   307
                desc += QString(tr("%1 fps")).arg(fps, 0, 'f', 2) + ", ";
7897
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   308
            }
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   309
        }
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   310
        else if (pCodec->codec_type == AVMEDIA_TYPE_AUDIO)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   311
            desc += tr("Audio: ");
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   312
        else
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   313
            continue;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   314
        AVCodec* pDecoder = avcodec_find_decoder(pCodec->codec_id);
8362
89a6b7cc6f95 - Unbreak build with .pro file
unc0rr
parents: 8065
diff changeset
   315
        desc += pDecoder? pDecoder->name : tr("unknown");
7897
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   316
        desc += "\n";
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   317
    }
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   318
    AVDictionaryEntry* pComment = av_dict_get(pContext->metadata, "comment", NULL, 0);
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   319
    if (pComment)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   320
        desc += QString("\n") + pComment->value;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   321
    avformat_close_input(&pContext);
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   322
    return desc;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   323
}
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   324
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   325
#else
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   326
LibavInteraction::LibavInteraction() : QObject()
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   327
{
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   328
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   329
}
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   330
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   331
void LibavInteraction::fillFormats(QComboBox * pFormats)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   332
{
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   333
    Q_UNUSED(pFormats);
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   334
}
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   335
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   336
void LibavInteraction::fillCodecs(const QString & format, QComboBox * pVCodecs, QComboBox * pACodecs)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   337
{
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   338
    Q_UNUSED(format);
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   339
    Q_UNUSED(pVCodecs);
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   340
    Q_UNUSED(pACodecs);
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   341
}
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   342
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   343
QString LibavInteraction::getExtension(const QString & format)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   344
{
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   345
    Q_UNUSED(format);
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   346
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   347
    return QString();
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   348
}
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   349
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   350
QString LibavInteraction::getFileInfo(const QString & filepath)
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   351
{
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   352
    Q_UNUSED(filepath);
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   353
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   354
    return QString();
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   355
}
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   356
#endif
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   357
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   358
LibavInteraction & LibavInteraction::instance()
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   359
{
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   360
    static LibavInteraction instance;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   361
    return instance;
5e7c0810f365 libav name refactor
koda
parents:
diff changeset
   362
}