103 lines
3.8 KiB
C
103 lines
3.8 KiB
C
|
#ifndef _mpeg_ts_h_
|
||
|
|
#define _mpeg_ts_h_
|
||
|
|
|
||
|
|
#include <stdint.h>
|
||
|
|
#include <stddef.h>
|
||
|
|
|
||
|
|
#ifdef __cplusplus
|
||
|
|
extern "C" {
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#include "mpeg-proto.h"
|
||
|
|
|
||
|
|
struct mpeg_ts_func_t
|
||
|
|
{
|
||
|
|
/// alloc new packet
|
||
|
|
/// @param[in] param use-defined parameter(by mpeg_ps_create)
|
||
|
|
/// @param[in] bytes alloc memory size in byte(default 188)
|
||
|
|
/// @return memory pointer
|
||
|
|
void* (*alloc)(void* param, size_t bytes);
|
||
|
|
|
||
|
|
/// free packet
|
||
|
|
/// @param[in] param use-defined parameter(by mpeg_ps_create)
|
||
|
|
/// @param[in] packet PS packet pointer(alloc return pointer)
|
||
|
|
void (*free)(void* param, void* packet);
|
||
|
|
|
||
|
|
/// callback on PS packet done
|
||
|
|
/// @param[in] param use-defined parameter(by mpeg_ps_create)
|
||
|
|
/// @param[in] packet PS packet pointer(alloc return pointer)
|
||
|
|
/// @param[in] bytes packet size
|
||
|
|
/// @return 0-ok, other-error
|
||
|
|
int (*write)(void* param, const void* packet, size_t bytes);
|
||
|
|
};
|
||
|
|
|
||
|
|
/// Create/Destroy MPEG2-TS muxer
|
||
|
|
void* mpeg_ts_create(const struct mpeg_ts_func_t *func, void* param);
|
||
|
|
int mpeg_ts_destroy(void* ts);
|
||
|
|
|
||
|
|
/// Add audio/video stream
|
||
|
|
/// @param[in] codecid PSI_STREAM_H264/PSI_STREAM_H265/PSI_STREAM_H266/PSI_STREAM_AAC, see more @mpeg-ts-proto.h
|
||
|
|
/// @param[in] extradata itu h.222.0 program and program element descriptors, NULL for H.264/H.265/AAC
|
||
|
|
/// @param[in] extradata_size extradata size in byte
|
||
|
|
/// @return <=0-error, >0-audio/video stream id
|
||
|
|
int mpeg_ts_add_stream(void* ts, int codecid, const void* extradata, size_t extradata_size);
|
||
|
|
|
||
|
|
/// Muxer audio/video stream data
|
||
|
|
/// @param[in] stream stream id by mpeg_ts_add_stream
|
||
|
|
/// @param[in] flags 0x0001-video IDR frame, 0x8000-H.264/H.265 with AUD
|
||
|
|
/// @param[in] pts audio/video stream timestamp in 90*ms
|
||
|
|
/// @param[in] dts audio/video stream timestamp in 90*ms
|
||
|
|
/// @param[in] data H.264/H.265-AnnexB stream(include 00 00 00 01), AAC-ADTS stream
|
||
|
|
/// @return 0-ok, other-error
|
||
|
|
int mpeg_ts_write(void* ts, int stream, int flags, int64_t pts, int64_t dts, const void* data, size_t bytes);
|
||
|
|
|
||
|
|
/// Reset PAT/PCR period
|
||
|
|
int mpeg_ts_reset(void* ts);
|
||
|
|
|
||
|
|
|
||
|
|
/// FOR MULTI-PROGRAM TS STREAM ONLY
|
||
|
|
/// Add a program
|
||
|
|
/// @param[in] pn program number, valid value: [1, 0xFFFF]
|
||
|
|
/// @return 0-OK, <0-error
|
||
|
|
int mpeg_ts_add_program(void* ts, uint16_t pn, const void* info, int bytes);
|
||
|
|
|
||
|
|
/// Remove a program by program number
|
||
|
|
/// @param[in] pn program number, valid value: [1, 0xFFFF]
|
||
|
|
/// @return 0-OK, <0-error
|
||
|
|
int mpeg_ts_remove_program(void* ts, uint16_t pn);
|
||
|
|
|
||
|
|
/// Add program stream(same as mpeg_ts_add_stream except program number)
|
||
|
|
/// @param[in] pn mpeg_ts_add_program program number
|
||
|
|
/// @return 0-OK, <0-error
|
||
|
|
int mpeg_ts_add_program_stream(void* ts, uint16_t pn, int codecid, const void* extra_data, size_t extra_data_size);
|
||
|
|
|
||
|
|
|
||
|
|
/// see more mpeg_ts_write
|
||
|
|
typedef int (*ts_demuxer_onpacket)(void* param, int program, int stream, int codecid, int flags, int64_t pts, int64_t dts, const void* data, size_t bytes);
|
||
|
|
|
||
|
|
struct ts_demuxer_t;
|
||
|
|
struct ts_demuxer_t* ts_demuxer_create(ts_demuxer_onpacket onpacket, void* param);
|
||
|
|
int ts_demuxer_destroy(struct ts_demuxer_t* demuxer);
|
||
|
|
int ts_demuxer_input(struct ts_demuxer_t* demuxer, const uint8_t* data, size_t bytes);
|
||
|
|
int ts_demuxer_flush(struct ts_demuxer_t* demuxer);
|
||
|
|
int ts_demuxer_getservice(struct ts_demuxer_t* demuxer, int program, char* provider, int nprovider, char* name, int nname);
|
||
|
|
|
||
|
|
struct ts_demuxer_notify_t
|
||
|
|
{
|
||
|
|
/// @param[in] param ps_demuxer_set_notify param
|
||
|
|
/// @param[in] stream ps stream id
|
||
|
|
/// @param[in] codecid ps codecid, e.g. STREAM_VIDEO_H264
|
||
|
|
/// @param[in] extra stream extra data
|
||
|
|
/// @param[in] bytes extra data length
|
||
|
|
/// @param[in] finish 0-have more stream, 1-no more streams
|
||
|
|
void (*onstream)(void* param, int stream, int codecid, const void* extra, int bytes, int finish);
|
||
|
|
};
|
||
|
|
|
||
|
|
/// Set ts notify on PMT change
|
||
|
|
void ts_demuxer_set_notify(struct ts_demuxer_t* demuxer, struct ts_demuxer_notify_t* notify, void* param);
|
||
|
|
|
||
|
|
#ifdef __cplusplus
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
#endif /* !_mpeg_ts_h_ */
|