AVEncoder.hxx

Go to the documentation of this file.

/*
 * Copyright (C) 2024-2026 David C. Manuelda (StormBytePP)
 *
 * This file is part of StormByte-Multimedia.
 *
 * StormByte-Multimedia original source is dual-licensed:
 *
 * 1. GNU Lesser General Public License v3.0 (or later)
 *    You may redistribute and/or modify this file under the terms of the
 *    GNU Lesser General Public License as published by the Free Software
 *    Foundation, either version 3 of the License, or (at your option)
 *    any later version.
 *
 * 2. Commercial license
 *    Alternatively, this file may be used under the terms of a commercial
 *    license agreement with the copyright holder
 *    (David C. Manuelda <StormByte@gmail.com>).
 *
 * Both licenses apply only to original StormByte-Multimedia source in this
 * file. Third-party components — including FFmpeg and embedded trained data —
 * remain under their own licenses and are not covered by the commercial grant.
 *
 * Neither license grants any patent rights. Any patent licenses required
 * to use this software or third-party components must be obtained separately
 * from the patent holders.
 *
 * StormByte-Multimedia is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * version 3 along with StormByte-Multimedia. If not, see
 * <https://www.gnu.org/licenses/lgpl-3.0.html>.
 *
 * SPDX-License-Identifier: LGPL-3.0-or-later OR LicenseRef-StormByte-Commercial
 */
 
#pragma once
 
#include <StormByte/multimedia/ffmpeg/AVBSFPipeline.hxx>
#include <StormByte/multimedia/ffmpeg/AVChannelLayout.hxx>
#include <StormByte/multimedia/ffmpeg/AVPointer.hxx>
#include <StormByte/multimedia/ffmpeg/AVRational.hxx>
#include <StormByte/multimedia/ffmpeg/fwd.hxx>
#include <StormByte/multimedia/ffmpeg/typedefs.hxx>
 
#include <cstdint>
#include <map>
#include <string>
#include <vector>

namespace StormByte::Multimedia::FFmpeg {
    class AVCodecParameters;
    class AVFormatContext;
    class AVFrame;
    class AVPacket;
    class AVSubtitle;

    class STORMBYTE_MULTIMEDIA_PUBLIC AVEncoder: public AVPointer<::AVCodecContext> {
        public:
            AVEncoder(const AVEncoder&) = delete;

            AVEncoder(AVEncoder&& other) noexcept = default;

            ~AVEncoder() noexcept override;

            AVEncoder& operator=(const AVEncoder&) = delete;

            AVEncoder& operator=(AVEncoder&& other) noexcept = default;

            static ExpectedAVEncoder Open(AVCodec* codec, const AVCodecParameters& params, const AVFormatContext& fmt, int stream_index) noexcept;

            static ExpectedAVEncoder Open(AVCodec* codec, const AVCodecParameters& params, int stream_index,
                const std::map<std::string, std::string>& options, AVRational time_base) noexcept;

            FFmpeg::OperationResult SendFrame(AVFrame& frame) noexcept;

            FFmpeg::OperationResult EncodeSubtitle(AVSubtitle& sub, AVPacket& pkt) noexcept;

            FFmpeg::OperationResult ReceivePacket(AVPacket& pkt) noexcept;

            int StreamIndex() const noexcept;

            AVRational TimeBase() const noexcept;

            bool IsSubtitle() const noexcept;

            void Flush() noexcept;

            FFmpeg::OperationResult SetEof() noexcept;

            explicit operator bool() const noexcept;

            int CodecId() const noexcept;

            int CodecType() const noexcept;

            int SampleFmt() const noexcept;

            int FrameSize() const noexcept;

            int Channels() const noexcept;

            int SampleRate() const noexcept;

            AVChannelLayout ChannelLayout() const noexcept;

            bool HasBFrames() const noexcept;

            AVRational FrameRate() const noexcept;

            const ::AVCodecContext* Context() const noexcept;
 
        private:
            int m_stream_index = -1;                                        
            FFmpeg::AVBSFPipeline m_bsf_pipeline;                           
            std::map<std::int64_t, std::vector<std::uint8_t>> m_hdrPlusT35; 

            explicit AVEncoder(AVCodecContext* ctx) noexcept;

            void Free() noexcept override;
 
            using AVPointer<::AVCodecContext>::Get;
    };
 
    extern template class STORMBYTE_MULTIMEDIA_PUBLIC AVPointer<::AVCodecContext>;
}