xref: /aosp_15_r20/external/libwebm/webm_parser/src/segment_parser.cc (revision 103e46e4cd4b6efcf6001f23fa8665fb110abf8d)
1*103e46e4SHarish Mahendrakar // Copyright (c) 2016 The WebM project authors. All Rights Reserved.
2*103e46e4SHarish Mahendrakar //
3*103e46e4SHarish Mahendrakar // Use of this source code is governed by a BSD-style license
4*103e46e4SHarish Mahendrakar // that can be found in the LICENSE file in the root of the source
5*103e46e4SHarish Mahendrakar // tree. An additional intellectual property rights grant can be found
6*103e46e4SHarish Mahendrakar // in the file PATENTS.  All contributing project authors may
7*103e46e4SHarish Mahendrakar // be found in the AUTHORS file in the root of the source tree.
8*103e46e4SHarish Mahendrakar #include "src/segment_parser.h"
9*103e46e4SHarish Mahendrakar 
10*103e46e4SHarish Mahendrakar #include "src/chapters_parser.h"
11*103e46e4SHarish Mahendrakar #include "src/cluster_parser.h"
12*103e46e4SHarish Mahendrakar #include "src/cues_parser.h"
13*103e46e4SHarish Mahendrakar #include "src/info_parser.h"
14*103e46e4SHarish Mahendrakar #include "src/seek_head_parser.h"
15*103e46e4SHarish Mahendrakar #include "src/skip_callback.h"
16*103e46e4SHarish Mahendrakar #include "src/tags_parser.h"
17*103e46e4SHarish Mahendrakar #include "src/tracks_parser.h"
18*103e46e4SHarish Mahendrakar #include "webm/id.h"
19*103e46e4SHarish Mahendrakar 
20*103e46e4SHarish Mahendrakar namespace webm {
21*103e46e4SHarish Mahendrakar 
SegmentParser()22*103e46e4SHarish Mahendrakar SegmentParser::SegmentParser()
23*103e46e4SHarish Mahendrakar     : MasterParser(MakeChild<ChaptersParser>(Id::kChapters),
24*103e46e4SHarish Mahendrakar                    MakeChild<ClusterParser>(Id::kCluster),
25*103e46e4SHarish Mahendrakar                    MakeChild<CuesParser>(Id::kCues),
26*103e46e4SHarish Mahendrakar                    MakeChild<InfoParser>(Id::kInfo),
27*103e46e4SHarish Mahendrakar                    MakeChild<SeekHeadParser>(Id::kSeekHead),
28*103e46e4SHarish Mahendrakar                    MakeChild<TagsParser>(Id::kTags),
29*103e46e4SHarish Mahendrakar                    MakeChild<TracksParser>(Id::kTracks)) {}
30*103e46e4SHarish Mahendrakar 
Init(const ElementMetadata & metadata,std::uint64_t max_size)31*103e46e4SHarish Mahendrakar Status SegmentParser::Init(const ElementMetadata& metadata,
32*103e46e4SHarish Mahendrakar                            std::uint64_t max_size) {
33*103e46e4SHarish Mahendrakar   assert(metadata.size == kUnknownElementSize || metadata.size <= max_size);
34*103e46e4SHarish Mahendrakar 
35*103e46e4SHarish Mahendrakar   begin_done_ = false;
36*103e46e4SHarish Mahendrakar   parse_completed_ = false;
37*103e46e4SHarish Mahendrakar   return MasterParser::Init(metadata, max_size);
38*103e46e4SHarish Mahendrakar }
39*103e46e4SHarish Mahendrakar 
InitAfterSeek(const Ancestory & child_ancestory,const ElementMetadata & child_metadata)40*103e46e4SHarish Mahendrakar void SegmentParser::InitAfterSeek(const Ancestory& child_ancestory,
41*103e46e4SHarish Mahendrakar                                   const ElementMetadata& child_metadata) {
42*103e46e4SHarish Mahendrakar   MasterParser::InitAfterSeek(child_ancestory, child_metadata);
43*103e46e4SHarish Mahendrakar 
44*103e46e4SHarish Mahendrakar   begin_done_ = true;
45*103e46e4SHarish Mahendrakar   parse_completed_ = false;
46*103e46e4SHarish Mahendrakar   action_ = Action::kRead;
47*103e46e4SHarish Mahendrakar }
48*103e46e4SHarish Mahendrakar 
Feed(Callback * callback,Reader * reader,std::uint64_t * num_bytes_read)49*103e46e4SHarish Mahendrakar Status SegmentParser::Feed(Callback* callback, Reader* reader,
50*103e46e4SHarish Mahendrakar                            std::uint64_t* num_bytes_read) {
51*103e46e4SHarish Mahendrakar   assert(callback != nullptr);
52*103e46e4SHarish Mahendrakar   assert(reader != nullptr);
53*103e46e4SHarish Mahendrakar   assert(num_bytes_read != nullptr);
54*103e46e4SHarish Mahendrakar 
55*103e46e4SHarish Mahendrakar   *num_bytes_read = 0;
56*103e46e4SHarish Mahendrakar 
57*103e46e4SHarish Mahendrakar   if (!begin_done_) {
58*103e46e4SHarish Mahendrakar     const ElementMetadata metadata{Id::kSegment, header_size(), size(),
59*103e46e4SHarish Mahendrakar                                    position()};
60*103e46e4SHarish Mahendrakar     const Status status = callback->OnSegmentBegin(metadata, &action_);
61*103e46e4SHarish Mahendrakar     if (!status.completed_ok()) {
62*103e46e4SHarish Mahendrakar       return status;
63*103e46e4SHarish Mahendrakar     }
64*103e46e4SHarish Mahendrakar     begin_done_ = true;
65*103e46e4SHarish Mahendrakar   }
66*103e46e4SHarish Mahendrakar 
67*103e46e4SHarish Mahendrakar   SkipCallback skip_callback;
68*103e46e4SHarish Mahendrakar   if (action_ == Action::kSkip) {
69*103e46e4SHarish Mahendrakar     callback = &skip_callback;
70*103e46e4SHarish Mahendrakar   }
71*103e46e4SHarish Mahendrakar 
72*103e46e4SHarish Mahendrakar   if (!parse_completed_) {
73*103e46e4SHarish Mahendrakar     const Status status = MasterParser::Feed(callback, reader, num_bytes_read);
74*103e46e4SHarish Mahendrakar     if (!status.completed_ok()) {
75*103e46e4SHarish Mahendrakar       return status;
76*103e46e4SHarish Mahendrakar     }
77*103e46e4SHarish Mahendrakar     parse_completed_ = true;
78*103e46e4SHarish Mahendrakar   }
79*103e46e4SHarish Mahendrakar 
80*103e46e4SHarish Mahendrakar   return callback->OnSegmentEnd(
81*103e46e4SHarish Mahendrakar       {Id::kSegment, header_size(), size(), position()});
82*103e46e4SHarish Mahendrakar }
83*103e46e4SHarish Mahendrakar 
WasSkipped() const84*103e46e4SHarish Mahendrakar bool SegmentParser::WasSkipped() const { return action_ == Action::kSkip; }
85*103e46e4SHarish Mahendrakar 
86*103e46e4SHarish Mahendrakar }  // namespace webm
87