ZLMediaKit/src/Rtp/TSDecoder.h

62 lines
1.7 KiB
C++
Raw Normal View History

2020-03-06 13:00:06 +08:00
/*
2020-04-04 20:30:09 +08:00
* Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
2020-03-06 13:00:06 +08:00
*
* This file is part of ZLMediaKit(https://github.com/xia-chu/ZLMediaKit).
2020-03-06 13:00:06 +08:00
*
2020-04-04 20:30:09 +08:00
* Use of this source code is governed by MIT license that can be found in the
* LICENSE file in the root of the source tree. All contributing project authors
* may be found in the AUTHORS file in the root of the source tree.
2020-03-06 13:00:06 +08:00
*/
#ifndef ZLMEDIAKIT_TSDECODER_H
#define ZLMEDIAKIT_TSDECODER_H
#include "Util/logger.h"
#include "Http/HttpRequestSplitter.h"
#include "Decoder.h"
using namespace toolkit;
namespace mediakit {
2020-05-17 18:00:23 +08:00
#define TS_PACKET_SIZE 188
#define TS_SYNC_BYTE 0x47
//TS包分割器用于split一个一个的ts包
2020-03-06 13:00:06 +08:00
class TSSegment : public HttpRequestSplitter {
public:
typedef std::function<void(const char *data,size_t len)> onSegment;
TSSegment(size_t size = TS_PACKET_SIZE) : _size(size){}
2020-03-06 13:00:06 +08:00
~TSSegment(){}
void setOnSegment(const onSegment &cb);
static bool isTSPacket(const char *data, size_t len);
2020-03-06 13:00:06 +08:00
protected:
2021-01-19 16:05:38 +08:00
ssize_t onRecvHeader(const char *data, size_t len) override ;
const char *onSearchPacketTail(const char *data, size_t len) override ;
2020-03-06 13:00:06 +08:00
private:
size_t _size;
2020-03-06 13:00:06 +08:00
onSegment _onSegment;
};
2020-05-17 18:00:23 +08:00
#if defined(ENABLE_HLS)
2020-03-06 13:00:06 +08:00
//ts解析器
class TSDecoder : public Decoder {
public:
TSDecoder();
~TSDecoder();
2021-02-09 14:01:10 +08:00
ssize_t input(const uint8_t* data, size_t bytes) override ;
void setOnDecode(onDecode cb) override;
void setOnStream(onStream cb) override;
2020-03-06 13:00:06 +08:00
private:
TSSegment _ts_segment;
struct ts_demuxer_t* _demuxer_ctx = nullptr;
onDecode _on_decode;
onStream _on_stream;
2020-03-06 13:00:06 +08:00
};
2020-05-17 18:00:23 +08:00
#endif//defined(ENABLE_HLS)
2020-03-06 13:00:06 +08:00
}//namespace mediakit
#endif //ZLMEDIAKIT_TSDECODER_H