mirror of
https://github.com/ZLMediaKit/ZLMediaKit.git
synced 2024-10-30 16:27:36 +08:00
65 lines
1.9 KiB
C++
65 lines
1.9 KiB
C++
#ifndef ZLMEDIAKIT_JPEG_H
|
|
#define ZLMEDIAKIT_JPEG_H
|
|
|
|
#include "Extension/Frame.h"
|
|
#include "Extension/Track.h"
|
|
|
|
namespace mediakit {
|
|
|
|
class JPEGTrack : public VideoTrack {
|
|
public:
|
|
using Ptr = std::shared_ptr<JPEGTrack>;
|
|
|
|
CodecId getCodecId() const override { return CodecJPEG; }
|
|
int getVideoHeight() const override { return _height; }
|
|
int getVideoWidth() const override { return _width; }
|
|
float getVideoFps() const override { return _fps; }
|
|
bool ready() const override { return _fps > 0; }
|
|
bool inputFrame(const Frame::Ptr &frame) override;
|
|
|
|
private:
|
|
Sdp::Ptr getSdp(uint8_t payload_type) const override;
|
|
Track::Ptr clone() const override { return std::make_shared<JPEGTrack>(*this); }
|
|
void getVideoResolution(const uint8_t *buf, int len);
|
|
|
|
private:
|
|
int _width = 0;
|
|
int _height = 0;
|
|
float _fps = 0;
|
|
uint64_t _tmp = 0;
|
|
};
|
|
|
|
class JPEGFrameType {
|
|
public:
|
|
virtual ~JPEGFrameType() = default;
|
|
virtual uint8_t pixType() const = 0;
|
|
};
|
|
|
|
template <typename Parent>
|
|
class JPEGFrame : public Parent, public JPEGFrameType {
|
|
public:
|
|
static constexpr auto kJFIFSize = 20u;
|
|
/**
|
|
* JPEG/MJPEG帧
|
|
* @param pix_type pixel format type; AV_PIX_FMT_YUVJ422P || (AVCOL_RANGE_JPEG && AV_PIX_FMT_YUV422P) : 1; AV_PIX_FMT_YUVJ420P || (AVCOL_RANGE_JPEG && AV_PIX_FMT_YUV420P) : 0
|
|
*/
|
|
template <typename... ARGS>
|
|
JPEGFrame(uint8_t pix_type, ARGS &&...args) : Parent(std::forward<ARGS>(args)...) {
|
|
_pix_type = pix_type;
|
|
// JFIF头固定20个字节长度
|
|
CHECK(this->size() > kJFIFSize);
|
|
}
|
|
size_t prefixSize() const override { return 0; }
|
|
bool keyFrame() const override { return true; }
|
|
bool configFrame() const override { return false; }
|
|
CodecId getCodecId() const override { return CodecJPEG; }
|
|
uint8_t pixType() const override { return _pix_type; }
|
|
|
|
private:
|
|
uint8_t _pix_type;
|
|
};
|
|
|
|
}//namespace mediakit
|
|
|
|
#endif //ZLMEDIAKIT_JPEG_H
|