64 lines
2.0 KiB
C
64 lines
2.0 KiB
C
|
#ifndef DRONESCREW_MJPEG_ENCODER_H
|
|||
|
|
#define DRONESCREW_MJPEG_ENCODER_H
|
|||
|
|
|
|||
|
|
#include <cstdint>
|
|||
|
|
#include <cstddef>
|
|||
|
|
#include <vector>
|
|||
|
|
#include <mutex>
|
|||
|
|
|
|||
|
|
// 基于 Rockchip MPP 硬件 MJPEG 的单帧编码器：Mono8 → JPEG 字节。
|
|||
|
|
// 用于检测/图传时把裸 Mono8 帧压缩后再走 ZMQ，显著降低网络带宽。
|
|||
|
|
//
|
|||
|
|
// 仅在 Linux/RK3588 平台真正可用；其它平台 EncodeMono8 始终返回 false，
|
|||
|
|
// 调用方据此回退为发送原始数据。
|
|||
|
|
//
|
|||
|
|
// 内部按 (width,height) 懒初始化/重建编码器；EncodeMono8 线程安全。
|
|||
|
|
class MppMjpegEncoder
|
|||
|
|
{
|
|||
|
|
public:
|
|||
|
|
MppMjpegEncoder() = default;
|
|||
|
|
~MppMjpegEncoder();
|
|||
|
|
|
|||
|
|
MppMjpegEncoder(const MppMjpegEncoder&) = delete;
|
|||
|
|
MppMjpegEncoder& operator=(const MppMjpegEncoder&) = delete;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 将一帧 Mono8（紧凑排列，stride=width）编码为整张 JPEG
|
|||
|
|
* @param mono Mono8 数据指针
|
|||
|
|
* @param monoSize 数据字节数（须 >= width*height）
|
|||
|
|
* @param width 图像宽
|
|||
|
|
* @param height 图像高
|
|||
|
|
* @param jpeg 输出：完整 JPEG 字节
|
|||
|
|
* @return true 成功；false 失败（调用方应回退原始数据）
|
|||
|
|
*/
|
|||
|
|
bool EncodeMono8(const uint8_t* mono, size_t monoSize,
|
|||
|
|
int width, int height,
|
|||
|
|
std::vector<uint8_t>& jpeg);
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief 设置 JPEG 质量因子（1~99，越大越清晰、体积越大）。
|
|||
|
|
* 改变后下一帧会自动重建编码器。默认 85（接近视觉无损，兼顾带宽）。
|
|||
|
|
*/
|
|||
|
|
void SetQuality(int q);
|
|||
|
|
|
|||
|
|
private:
|
|||
|
|
#if defined(__linux__)
|
|||
|
|
bool ensureInited(int width, int height);
|
|||
|
|
void release();
|
|||
|
|
|
|||
|
|
void* m_ctx{nullptr}; // MppCtx
|
|||
|
|
void* m_mpi{nullptr}; // MppApi*
|
|||
|
|
void* m_grp{nullptr}; // MppBufferGroup
|
|||
|
|
int m_w{0};
|
|||
|
|
int m_h{0};
|
|||
|
|
int m_hor{0};
|
|||
|
|
int m_ver{0};
|
|||
|
|
int m_quality{85};
|
|||
|
|
std::mutex m_mutex;
|
|||
|
|
#else
|
|||
|
|
int m_quality{85};
|
|||
|
|
#endif
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
#endif // DRONESCREW_MJPEG_ENCODER_H
|