VideoReader.cpp

Go to the documentation of this file.

#include "VideoReader.hpp"

#include <AvTranscoder/decoder/VideoDecoder.hpp>
#include <AvTranscoder/decoder/VideoGenerator.hpp>
#include <AvTranscoder/data/decoded/VideoFrame.hpp>
#include <AvTranscoder/transform/VideoTransform.hpp>
#include <AvTranscoder/progress/NoDisplayProgress.hpp>

namespace avtranscoder
{

VideoReader::VideoReader(const std::string& filename, const size_t videoStreamIndex)
    : IReader(filename, videoStreamIndex)
    , _videoStreamProperties(NULL)
    , _outputWidth(0)
    , _outputHeight(0)
    , _outputPixelProperties("rgb24")
{
    init();
}

VideoReader::VideoReader(InputFile& inputFile, const size_t videoStreamIndex)
    : IReader(inputFile, videoStreamIndex)
    , _videoStreamProperties(NULL)
    , _outputWidth(0)
    , _outputHeight(0)
    , _outputPixelProperties("rgb24")
{
    init();
}

void VideoReader::init()
{
    // analyse InputFile
    avtranscoder::NoDisplayProgress p;
    _inputFile->analyse(p);
    _streamProperties = &_inputFile->getProperties().getStreamPropertiesWithIndex(_streamIndex);
    _videoStreamProperties = static_cast<const VideoProperties*>(_streamProperties);
    _inputFile->activateStream(_streamIndex);

    // setup decoder
    _decoder = new VideoDecoder(_inputFile->getStream(_streamIndex));
    _decoder->setupDecoder();
    _currentDecoder = _decoder;

    // generator
    _generator = new VideoGenerator(_inputFile->getStream(_streamIndex).getVideoCodec().getVideoFrameDesc());

    // create transform
    _transform = new VideoTransform();

    // create src frame
    _srcFrame = new VideoFrame(_inputFile->getStream(_streamIndex).getVideoCodec().getVideoFrameDesc());
    VideoFrame* srcFrame = static_cast<VideoFrame*>(_srcFrame);
    // create dst frame
    _outputWidth = srcFrame->getWidth();
    _outputHeight = srcFrame->getHeight();
    _dstFrame = new VideoFrame(VideoFrameDesc(_outputWidth, _outputHeight, getOutputPixelFormat()));
}

VideoReader::~VideoReader()
{
    delete _decoder;
    delete _generator;
    delete _srcFrame;
    delete _dstFrame;
    delete _transform;
}

void VideoReader::updateOutput(const size_t width, const size_t height, const std::string& pixelFormat)
{
    _outputWidth = width;
    _outputHeight = height;
    _outputPixelProperties = PixelProperties(pixelFormat);
    // update dst frame
    delete _dstFrame;
    _dstFrame = new VideoFrame(VideoFrameDesc(_outputWidth, _outputHeight, getOutputPixelFormat()));
}
}