FFMPEGReader.h

Go to the documentation of this file.

/*******************************************************************************
 * Copyright 2009-2024 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;

    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