mirror of https://git.ffmpeg.org/ffmpeg.git
avformat/mpegts: add support for EPG extraction from mpegts
Signed-off-by: Marton Balint <cus@passwd.hu>
This commit is contained in:
parent
81bb71761a
commit
a221af1f5e
|
@ -168,6 +168,8 @@ struct MpegTSContext {
|
||||||
/** filters for various streams specified by PMT + for the PAT and PMT */
|
/** filters for various streams specified by PMT + for the PAT and PMT */
|
||||||
MpegTSFilter *pids[NB_PID_MAX];
|
MpegTSFilter *pids[NB_PID_MAX];
|
||||||
int current_pid;
|
int current_pid;
|
||||||
|
|
||||||
|
AVStream *epg_stream;
|
||||||
};
|
};
|
||||||
|
|
||||||
#define MPEGTS_OPTIONS \
|
#define MPEGTS_OPTIONS \
|
||||||
|
@ -2499,6 +2501,60 @@ static void pat_cb(MpegTSFilter *filter, const uint8_t *section, int section_len
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static void eit_cb(MpegTSFilter *filter, const uint8_t *section, int section_len)
|
||||||
|
{
|
||||||
|
MpegTSContext *ts = filter->u.section_filter.opaque;
|
||||||
|
const uint8_t *p, *p_end;
|
||||||
|
SectionHeader h1, *h = &h1;
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Sometimes we receive EPG packets but SDT table do not have
|
||||||
|
* eit_pres_following or eit_sched turned on, so we open EPG
|
||||||
|
* stream directly here.
|
||||||
|
*/
|
||||||
|
if (!ts->epg_stream) {
|
||||||
|
ts->epg_stream = avformat_new_stream(ts->stream, NULL);
|
||||||
|
if (!ts->epg_stream)
|
||||||
|
return;
|
||||||
|
ts->epg_stream->id = EIT_PID;
|
||||||
|
ts->epg_stream->codecpar->codec_type = AVMEDIA_TYPE_DATA;
|
||||||
|
ts->epg_stream->codecpar->codec_id = AV_CODEC_ID_EPG;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (ts->epg_stream->discard == AVDISCARD_ALL)
|
||||||
|
return;
|
||||||
|
|
||||||
|
p_end = section + section_len - 4;
|
||||||
|
p = section;
|
||||||
|
|
||||||
|
if (parse_section_header(h, &p, p_end) < 0)
|
||||||
|
return;
|
||||||
|
if (h->tid < EIT_TID || h->tid > OEITS_END_TID)
|
||||||
|
return;
|
||||||
|
|
||||||
|
av_log(ts->stream, AV_LOG_TRACE, "EIT: tid received = %.02x\n", h->tid);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Service_id 0xFFFF is reserved, it indicates that the current EIT table
|
||||||
|
* is scrambled.
|
||||||
|
*/
|
||||||
|
if (h->id == 0xFFFF) {
|
||||||
|
av_log(ts->stream, AV_LOG_TRACE, "Scrambled EIT table received.\n");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* In case we receive an EPG packet before mpegts context is fully
|
||||||
|
* initialized.
|
||||||
|
*/
|
||||||
|
if (!ts->pkt)
|
||||||
|
return;
|
||||||
|
|
||||||
|
new_data_packet(section, section_len, ts->pkt);
|
||||||
|
ts->pkt->stream_index = ts->epg_stream->index;
|
||||||
|
ts->stop_parse = 1;
|
||||||
|
}
|
||||||
|
|
||||||
static void sdt_cb(MpegTSFilter *filter, const uint8_t *section, int section_len)
|
static void sdt_cb(MpegTSFilter *filter, const uint8_t *section, int section_len)
|
||||||
{
|
{
|
||||||
MpegTSContext *ts = filter->u.section_filter.opaque;
|
MpegTSContext *ts = filter->u.section_filter.opaque;
|
||||||
|
@ -2985,8 +3041,8 @@ static int mpegts_read_header(AVFormatContext *s)
|
||||||
seek_back(s, pb, pos);
|
seek_back(s, pb, pos);
|
||||||
|
|
||||||
mpegts_open_section_filter(ts, SDT_PID, sdt_cb, ts, 1);
|
mpegts_open_section_filter(ts, SDT_PID, sdt_cb, ts, 1);
|
||||||
|
|
||||||
mpegts_open_section_filter(ts, PAT_PID, pat_cb, ts, 1);
|
mpegts_open_section_filter(ts, PAT_PID, pat_cb, ts, 1);
|
||||||
|
mpegts_open_section_filter(ts, EIT_PID, eit_cb, ts, 1);
|
||||||
|
|
||||||
handle_packets(ts, probesize / ts->raw_packet_size);
|
handle_packets(ts, probesize / ts->raw_packet_size);
|
||||||
/* if could not find service, enable auto_guess */
|
/* if could not find service, enable auto_guess */
|
||||||
|
@ -3241,8 +3297,10 @@ MpegTSContext *avpriv_mpegts_parse_open(AVFormatContext *s)
|
||||||
ts->raw_packet_size = TS_PACKET_SIZE;
|
ts->raw_packet_size = TS_PACKET_SIZE;
|
||||||
ts->stream = s;
|
ts->stream = s;
|
||||||
ts->auto_guess = 1;
|
ts->auto_guess = 1;
|
||||||
|
|
||||||
mpegts_open_section_filter(ts, SDT_PID, sdt_cb, ts, 1);
|
mpegts_open_section_filter(ts, SDT_PID, sdt_cb, ts, 1);
|
||||||
mpegts_open_section_filter(ts, PAT_PID, pat_cb, ts, 1);
|
mpegts_open_section_filter(ts, PAT_PID, pat_cb, ts, 1);
|
||||||
|
mpegts_open_section_filter(ts, EIT_PID, eit_cb, ts, 1);
|
||||||
|
|
||||||
return ts;
|
return ts;
|
||||||
}
|
}
|
||||||
|
|
|
@ -12,3 +12,6 @@ codec_name=hevc
|
||||||
[STREAM]
|
[STREAM]
|
||||||
codec_name=aac_latm
|
codec_name=aac_latm
|
||||||
[/STREAM]
|
[/STREAM]
|
||||||
|
[STREAM]
|
||||||
|
codec_name=epg
|
||||||
|
[/STREAM]
|
||||||
|
|
Loading…
Reference in New Issue