/* * Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved. * * This file is part of ZLMediaKit(https://github.com/xiongziliang/ZLMediaKit). * * 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. */ #include "TSDecoder.h" namespace mediakit { bool TSSegment::isTSPacket(const char *data, int len){ return len == TS_PACKET_SIZE && ((uint8_t*)data)[0] == TS_SYNC_BYTE; } void TSSegment::setOnSegment(const TSSegment::onSegment &cb) { _onSegment = cb; } int64_t TSSegment::onRecvHeader(const char *data, uint64_t len) { if (!isTSPacket(data, len)) { WarnL << "不是ts包:" << (int) (data[0]) << " " << len; return 0; } _onSegment(data, len); return 0; } const char *TSSegment::onSearchPacketTail(const char *data, uint64_t len) { if (len < _size + 1) { if (len == _size && ((uint8_t *) data)[0] == TS_SYNC_BYTE) { return data + _size; } return nullptr; } //下一个包头 if (((uint8_t *) data)[_size] == TS_SYNC_BYTE) { return data + _size; } auto pos = memchr(data + _size, TS_SYNC_BYTE, len - _size); if (pos) { return (char *) pos; } return nullptr; } //////////////////////////////////////////////////////////////// #if defined(ENABLE_HLS) #include "mpeg-ts.h" TSDecoder::TSDecoder() : _ts_segment() { _ts_segment.setOnSegment([this](const char *data,uint64_t len){ ts_demuxer_input(_demuxer_ctx,(uint8_t*)data,len); }); _demuxer_ctx = ts_demuxer_create([](void* param, int program, int stream, int codecid, int flags, int64_t pts, int64_t dts, const void* data, size_t bytes){ TSDecoder *thiz = (TSDecoder*)param; if(thiz->_on_decode){ thiz->_on_decode(stream,codecid,flags,pts,dts,data,bytes); } return 0; },this); ts_demuxer_notify_t notify = { [](void *param, int stream, int codecid, const void *extra, int bytes, int finish) { TSDecoder *thiz = (TSDecoder *) param; if (thiz->_on_stream) { thiz->_on_stream(stream, codecid, extra, bytes, finish); } } }; ts_demuxer_set_notify((struct ts_demuxer_t *) _demuxer_ctx, ¬ify, this); } TSDecoder::~TSDecoder() { ts_demuxer_destroy(_demuxer_ctx); } int TSDecoder::input(const uint8_t *data, int bytes) { if (TSSegment::isTSPacket((char *)data, bytes)) { return ts_demuxer_input(_demuxer_ctx, (uint8_t *) data, bytes); } _ts_segment.input((char*)data,bytes); return bytes; } void TSDecoder::setOnDecode(Decoder::onDecode cb) { _on_decode = std::move(cb); } void TSDecoder::setOnStream(Decoder::onStream cb) { _on_stream = std::move(cb); } #endif//defined(ENABLE_HLS) }//namespace mediakit