95 lines
2.9 KiB
C++
95 lines
2.9 KiB
C++
|
// Copyright 2020 Google Inc. All rights reserved.
|
||
|
|
//
|
||
|
|
// Use of this source code is governed by a BSD-style
|
||
|
|
// license that can be found in the LICENSE file or at
|
||
|
|
// https://developers.google.com/open-source/licenses/bsd
|
||
|
|
|
||
|
|
#include "packager/media/formats/mp2t/es_parser_dvb.h"
|
||
|
|
|
||
|
|
#include "packager/media/base/bit_reader.h"
|
||
|
|
#include "packager/media/base/text_stream_info.h"
|
||
|
|
#include "packager/media/base/timestamp.h"
|
||
|
|
#include "packager/media/formats/mp2t/mp2t_common.h"
|
||
|
|
|
||
|
|
namespace shaka {
|
||
|
|
namespace media {
|
||
|
|
namespace mp2t {
|
||
|
|
|
||
|
|
EsParserDvb::EsParserDvb(uint32_t pid,
|
||
|
|
const NewStreamInfoCB& new_stream_info_cb,
|
||
|
|
const EmitTextSampleCB& emit_sample_cb)
|
||
|
|
: EsParser(pid),
|
||
|
|
new_stream_info_cb_(new_stream_info_cb),
|
||
|
|
emit_sample_cb_(emit_sample_cb) {}
|
||
|
|
|
||
|
|
EsParserDvb::~EsParserDvb() {}
|
||
|
|
|
||
|
|
bool EsParserDvb::Parse(const uint8_t* buf,
|
||
|
|
int size,
|
||
|
|
int64_t pts,
|
||
|
|
int64_t dts) {
|
||
|
|
if (!sent_info_) {
|
||
|
|
sent_info_ = true;
|
||
|
|
std::shared_ptr<StreamInfo> info = std::make_shared<TextStreamInfo>(
|
||
|
|
pid(), kMpeg2Timescale, kInfiniteDuration, kCodecText,
|
||
|
|
/* codec_string= */ "", /* codec_config= */ "", /* width= */ 0,
|
||
|
|
/* height= */ 0, /* language= */ "");
|
||
|
|
new_stream_info_cb_.Run(info);
|
||
|
|
}
|
||
|
|
|
||
|
|
// TODO: Handle buffering and multiple reads? All content so far has been
|
||
|
|
// a whole segment, so it may not be needed.
|
||
|
|
return ParseInternal(buf, size, pts);
|
||
|
|
}
|
||
|
|
|
||
|
|
bool EsParserDvb::Flush() {
|
||
|
|
for (auto& pair : parsers_) {
|
||
|
|
std::vector<std::shared_ptr<TextSample>> samples;
|
||
|
|
RCHECK(pair.second.Flush(&samples));
|
||
|
|
|
||
|
|
for (auto sample : samples)
|
||
|
|
emit_sample_cb_.Run(sample);
|
||
|
|
}
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
void EsParserDvb::Reset() {
|
||
|
|
parsers_.clear();
|
||
|
|
}
|
||
|
|
|
||
|
|
bool EsParserDvb::ParseInternal(const uint8_t* data, size_t size, int64_t pts) {
|
||
|
|
// See EN 300 743 Table 3.
|
||
|
|
BitReader reader(data, size);
|
||
|
|
int data_identifier;
|
||
|
|
int subtitle_stream_id;
|
||
|
|
RCHECK(reader.ReadBits(8, &data_identifier));
|
||
|
|
RCHECK(reader.ReadBits(8, &subtitle_stream_id));
|
||
|
|
RCHECK(data_identifier == 0x20);
|
||
|
|
RCHECK(subtitle_stream_id == 0);
|
||
|
|
|
||
|
|
int temp;
|
||
|
|
while (reader.ReadBits(8, &temp) && temp == 0xf) {
|
||
|
|
DvbSubSegmentType segment_type;
|
||
|
|
uint16_t page_id;
|
||
|
|
size_t segment_length;
|
||
|
|
RCHECK(reader.ReadBits(8, &segment_type));
|
||
|
|
RCHECK(reader.ReadBits(16, &page_id));
|
||
|
|
RCHECK(reader.ReadBits(16, &segment_length));
|
||
|
|
RCHECK(reader.bits_available() > segment_length * 8);
|
||
|
|
|
||
|
|
const uint8_t* payload = data + (size - reader.bits_available() / 8);
|
||
|
|
std::vector<std::shared_ptr<TextSample>> samples;
|
||
|
|
RCHECK(parsers_[page_id].Parse(segment_type, pts, payload, segment_length,
|
||
|
|
&samples));
|
||
|
|
for (auto sample : samples)
|
||
|
|
emit_sample_cb_.Run(sample);
|
||
|
|
|
||
|
|
RCHECK(reader.SkipBytes(segment_length));
|
||
|
|
}
|
||
|
|
return temp == 0xff;
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace mp2t
|
||
|
|
} // namespace media
|
||
|
|
} // namespace shaka
|