r5codec.h

/*
 * Copyright (c) 2022-2023 Infrared5, Inc.
 * License information: https://account.red5.net/assets/LICENSE.txt
 *
 */

#pragma once

#include "r5common.h"
#include "r5codec_sample.h"

namespace r5::common {
enum class CodecType : int {
    Undefined = 0, 
    H264, 
    Aac, 
    Opus, 
    Unknown 
};
enum class CodecDirection {
    Undefined = -1, 
    Encoder = 0, 
    Decoder = 1 
};
enum class CodecState {
    Error = -100, 
    Undefined = -1, 
    Ready 
};
enum class RateControl {
    Cbr, 
    Vbr 
};
enum class CodecImplType {
    Software, 
    Hardware 
};
typedef struct EncoderSettings
{
    EncoderSettings &operator=(EncoderSettings other)
    {
        memcpy(this, &other, sizeof(EncoderSettings));
        return *this;
    }
    // Common encoder settings
    int bitRate = 1500 * 1000; 

    // Video encoder settings
    int gopLength = 10; 
    int maxQP = -1; 
    int minQP = -1; 
    RateControl rateControl = RateControl::Cbr; 
} EncoderSettings;
class Codec
{
public:
    virtual int InitEncoder(MediaTransform &desc, const EncoderSettings &set) = 0;
    virtual int InitDecoder(MediaTransform &desc) = 0;
    virtual int32_t AddData(std::unique_ptr<uint8_t[]> inputBuffer, uint32_t inputBufferSize, uint32_t inputTimestamp) = 0;
    virtual int32_t GetData(std::unique_ptr<uint8_t[]> &outputBuffer, uint32_t &outputBufferSize, uint32_t &outputTimestamp) = 0;
    const CodecState GetCurrentState() const { return mState; }
    const MediaTransform &GetMediaTransform() const { return mDesc; }
    const std::string Name() const { return mName; }
    const std::string FullName() const { return mFullName; }
    virtual DataType GetDataType() const = 0;
    const CodecType Type() const { return mType; }
    const CodecDirection Direction() const { return mDirection; }
    bool isHardware() const { return mCodecImplType == CodecImplType::Hardware; }

protected:
    MediaTransform mDesc;
    CodecState mState = CodecState::Undefined;
    ILoggerPtr mLogger;
    std::string mName;
    std::string mFullName;
    CodecType mType = CodecType::Undefined;
    CodecDirection mDirection = CodecDirection::Undefined;
    CodecImplType mCodecImplType = CodecImplType::Software;
};
typedef std::shared_ptr<Codec> CodecPtr;
}