encoder.hpp

// This code is partially adapted from https://github.com/cbachhuber/CppVideoStreamer
#pragma once
 
#include <memory>
#include <stdint.h>
#include <vector>
 
namespace h264encoder {
 
class EncoderImpl;
 
// simplified version of the x264_nal_t struct containing only the necessary fields for
// encoding
typedef struct x264_nal_t_simple {
    /* Size of payload (including any padding) in bytes. */
    int i_payload;
    /* If param->b_annexb is set, Annex-B bytestream with startcode.
     * Otherwise, startcode is replaced with a 4-byte size.
     * This size is the size used in mp4/similar muxing; it is equal to i_payload-4 */
    uint8_t* p_payload;
} x264_nal_t_simple;
 
class Encoder {
public:
    Encoder(){};
    Encoder(int, int, int, int, float, int);
    int encode(unsigned char*);
    const std::vector<x264_nal_t_simple>& getNals();
 
private:
    std::shared_ptr<EncoderImpl> impl;
};
 
} // namespace h264encoder