summaryrefslogtreecommitdiff
path: root/src/video/VideoInfo.cpp
blob: 50cfec44a6fe3fd2aeae608c30f76a053ebbcccc (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
//
//  libavg - Media Playback Engine. 
//  Copyright (C) 2003-2014 Ulrich von Zadow
//
//  This library is free software; you can redistribute it and/or
//  modify it under the terms of the GNU Lesser General Public
//  License as published by the Free Software Foundation; either
//  version 2 of the License, or (at your option) any later version.
//
//  This library is distributed in the hope that it will be useful,
//  but WITHOUT ANY WARRANTY; without even the implied warranty of
//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
//  Lesser General Public License for more details.
//
//  You should have received a copy of the GNU Lesser General Public
//  License along with this library; if not, write to the Free Software
//  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
//
//  Current versions can be found at www.libavg.de
//

#include "VideoInfo.h"

#include "../base/Exception.h"

namespace avg {

using namespace std;

VideoInfo::VideoInfo()
{
}

VideoInfo::VideoInfo(string sContainerFormat, float duration, int bitrate, bool bHasVideo,
        bool bHasAudio)
    : m_sContainerFormat(sContainerFormat), 
      m_Duration(duration),
      m_Bitrate(bitrate),
      m_bHasVideo(bHasVideo),
      m_bHasAudio(bHasAudio)
{
}

void VideoInfo::setVideoData(const IntPoint& size, const string& sPixelFormat,
        int numFrames, float streamFPS, const string& sVCodec,
        bool bUsesVDPAU, float duration)
{
    AVG_ASSERT(m_bHasVideo);
    m_Size = size;
    m_sPixelFormat = sPixelFormat;
    m_NumFrames = numFrames;
    m_StreamFPS = streamFPS;
    m_sVCodec = sVCodec;
    m_bUsesVDPAU = bUsesVDPAU;
    m_VideoDuration = duration;
}

void VideoInfo::setAudioData(const string& sACodec, int sampleRate, int numAudioChannels,
        float duration)
{
    AVG_ASSERT(m_bHasAudio);
    m_sACodec = sACodec;
    m_SampleRate = sampleRate;
    m_NumAudioChannels = numAudioChannels;
    m_AudioDuration = duration;
}
    
float getStreamFPS(AVStream* pStream)
{
    float fps = 0;
    if (pStream->avg_frame_rate.den != 0) {
        fps = float(av_q2d(pStream->avg_frame_rate));
    }
#if LIBAVCODEC_VERSION_INT < AV_VERSION_INT(54, 25, 0)
    if (fps == 0 && pStream->r_frame_rate.den != 0) {
        fps = float(av_q2d(pStream->r_frame_rate));
    }
#endif
    if (fps == 0) { 
        float duration = float(pStream->duration)*float(av_q2d(pStream->time_base));
        fps = pStream->nb_frames/duration;
    }
    AVG_ASSERT(fps < 10000);
/*
    cerr << "getStreamFPS: fps= " << fps << endl;
    cerr << "    r_frame_rate num: " << m_pVStream->r_frame_rate.num << ", den: " << m_pVStream->r_frame_rate.den << endl;
    cerr << "    avg_frame_rate: num: " << m_pVStream->avg_frame_rate.num << ", den: " << m_pVStream->avg_frame_rate.den << endl;
    cerr << "    numFrames= " << getNumFrames() << ", duration= " 
                << getDuration(SS_VIDEO) << endl;
*/
    return fps;
}

}