webmmux/webmmuxstreamvideo.cc

// Copyright (c) 2010 The WebM project authors. All Rights Reserved.
//
// Use of this source code is governed by a BSD-style license
// that can be found in the LICENSE file in the root of the source
// tree. An additional intellectual property rights grant can be found
// in the file PATENTS.  All contributing project authors may
// be found in the AUTHORS file in the root of the source tree.

#include <strmif.h>
#include "webmconstants.h"
#include "webmmuxcontext.h"
#include "mediatypeutil.h"
#include <amvideo.h>
#include <dvdmedia.h>
#include <cassert>
#include <uuids.h>

namespace WebmMuxLib
{

StreamVideo::VideoFrame::VideoFrame()
{
}



StreamVideo::StreamVideo(
    Context& c,
    const AM_MEDIA_TYPE& mt) :
    Stream(c)
{
    MediaTypeUtil::Copy(mt, m_mt);
}


StreamVideo::~StreamVideo()
{
    MediaTypeUtil::Destroy(m_mt);
}


const BITMAPINFOHEADER& StreamVideo::GetBitmapInfoHeader() const
{
    const AM_MEDIA_TYPE& mt = m_mt;
    assert(mt.majortype == MEDIATYPE_Video);
    assert(mt.pbFormat);

    if (mt.formattype == FORMAT_VideoInfo)
    {
        assert(mt.cbFormat >= sizeof(VIDEOINFOHEADER));

        VIDEOINFOHEADER& vih = (VIDEOINFOHEADER&)(*mt.pbFormat);
        BITMAPINFOHEADER& bmih = vih.bmiHeader;
        assert(bmih.biSize >= sizeof(BITMAPINFOHEADER));

        return bmih;
    }

    assert(mt.formattype == FORMAT_VideoInfo2);

    {
        assert(mt.cbFormat >= sizeof(VIDEOINFOHEADER2));

        VIDEOINFOHEADER2& vih = (VIDEOINFOHEADER2&)(*mt.pbFormat);
        BITMAPINFOHEADER& bmih = vih.bmiHeader;
        assert(bmih.biSize >= sizeof(BITMAPINFOHEADER));

        return bmih;
    }
}


float StreamVideo::GetFramerate() const
{
    const AM_MEDIA_TYPE& mt = m_mt;
    assert(mt.majortype == MEDIATYPE_Video);
    assert(mt.pbFormat);

    if (mt.formattype == FORMAT_VideoInfo)
    {
        assert(mt.cbFormat >= sizeof(VIDEOINFOHEADER));
        VIDEOINFOHEADER& vih = (VIDEOINFOHEADER&)(*mt.pbFormat);

        if (vih.AvgTimePerFrame <= 0)
            return 0;

        //[ticks/sec] / [ticks/frame] = [frames/sec]
        const float result = 10000000.0f / float(vih.AvgTimePerFrame);

        return result;
    }

    if (mt.formattype == FORMAT_VideoInfo2)
    {
        assert(mt.cbFormat >= sizeof(VIDEOINFOHEADER2));
        VIDEOINFOHEADER2& vih = (VIDEOINFOHEADER2&)(*mt.pbFormat);

        if (vih.AvgTimePerFrame <= 0)
            return 0;

        //[ticks/sec] / [ticks/frame] = [frames/sec]
        const float result = 10000000.0f / float(vih.AvgTimePerFrame);

        return result;
    }

    return 0;
}


void StreamVideo::WriteTrackType()
{
    WebmUtil::EbmlScratchBuf& buf = m_context.m_buf;

    buf.WriteID1(WebmUtil::kEbmlTrackTypeID);
    buf.Write1UInt(1);
    buf.Serialize1UInt(WebmUtil::kEbmlTrackTypeVideo);
}


bool StreamVideo::Wait() const
{
    return m_context.WaitVideo();
}


StreamVideo::frames_t& StreamVideo::GetFrames()
{
    return m_vframes;
}


StreamVideo::frames_t& StreamVideo::GetKeyFrames()
{
    return m_rframes;
}


void StreamVideo::Flush()
{
    m_context.FlushVideo(this);
}



}  //end namespace WebmMuxLib