FFMPEGReader.h

Go to the documentation of this file.

/*******************************************************************************
 * Copyright 2009-2016 Jörg Müller
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 ******************************************************************************/
 
#pragma once
 
#ifdef FFMPEG_PLUGIN
#define AUD_BUILD_PLUGIN
#endif
 
#include "respec/ConverterFunctions.h"
#include "IReader.h"
#include "util/Buffer.h"
#include "file/FileInfo.h"
 
#include <string>
#include <memory>
#include <vector>
 
struct AVCodecContext;
extern "C" {
#include <libavformat/avformat.h>
}
 
AUD_NAMESPACE_BEGIN
 
class AUD_PLUGIN_API FFMPEGReader : public IReader
{
private:
    int m_position;
 
    DeviceSpecs m_specs;
 
    Buffer m_pkgbuf;
 
    int m_pkgbuf_left;
 
    AVFormatContext* m_formatCtx;
 
    AVCodecContext* m_codecCtx;
 
    AVFrame* m_frame;
 
    AVIOContext* m_aviocontext;
 
    int m_stream;
 
    convert_f m_convert;
 
    std::shared_ptr<Buffer> m_membuffer;
 
    data_t* m_membuf;
 
    long long m_membufferpos;
 
    bool m_tointerleave;
 
    AUD_LOCAL static SampleFormat convertSampleFormat(AVSampleFormat format);
 
    AUD_LOCAL int decode(AVPacket& packet, Buffer& buffer);
 
    AUD_LOCAL void init(int stream);
 
    // delete copy constructor and operator=
    FFMPEGReader(const FFMPEGReader&) = delete;
    FFMPEGReader& operator=(const FFMPEGReader&) = delete;
 
public:
    FFMPEGReader(const std::string &filename, int stream = 0);
 
    FFMPEGReader(std::shared_ptr<Buffer> buffer, int stream = 0);
 
    virtual ~FFMPEGReader();
 
    virtual std::vector<StreamInfo> queryStreams();
 
    static int read_packet(void* opaque, uint8_t* buf, int buf_size);
 
    static int64_t seek_packet(void* opaque, int64_t offset, int whence);
 
    virtual bool isSeekable() const;
    virtual void seek(int position);
    virtual int getLength() const;
    virtual int getPosition() const;
    virtual Specs getSpecs() const;
    virtual void read(int& length, bool& eos, sample_t* buffer);
};
 
AUD_NAMESPACE_END