1 /*
2 * Copyright (C) 2023 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16 #include <cinttypes>
17 #include "native_avmemory.h"
18 #include "avdemuxer.h"
19 #include "inner_demuxer_demo.h"
20
21 namespace OHOS {
22 namespace MediaAVCodec {
InnerDemuxerDemo()23 InnerDemuxerDemo::InnerDemuxerDemo()
24 {
25 printf("InnerDemuxerDemo is called\n");
26 }
27
~InnerDemuxerDemo()28 InnerDemuxerDemo::~InnerDemuxerDemo()
29 {
30 printf("~InnerDemuxerDemo is called\n");
31 }
32
CreateWithSource(std::shared_ptr<AVSource> source)33 int32_t InnerDemuxerDemo::CreateWithSource(std::shared_ptr<AVSource> source)
34 {
35 this->demuxer_ = AVDemuxerFactory::CreateWithSource(source);
36 if (!demuxer_) {
37 printf("AVDemuxerFactory::CreateWithSource is failed\n");
38 return -1;
39 }
40 return 0;
41 }
Destroy()42 void InnerDemuxerDemo::Destroy()
43 {
44 printf("InnerDemuxerDemo::Destroy\n");
45 }
46
SelectTrackByID(uint32_t trackIndex)47 int32_t InnerDemuxerDemo::SelectTrackByID(uint32_t trackIndex)
48 {
49 int32_t ret = this->demuxer_->SelectTrackByID(trackIndex);
50 if (ret != 0) {
51 printf("SelectTrackByID is failed\n");
52 }
53 return ret;
54 }
55
UnselectTrackByID(uint32_t trackIndex)56 int32_t InnerDemuxerDemo::UnselectTrackByID(uint32_t trackIndex)
57 {
58 int32_t ret = this->demuxer_->UnselectTrackByID(trackIndex);
59 if (ret != 0) {
60 printf("SelectTrackByID is failed\n");
61 }
62 return ret;
63 }
64
PrintInfo(int32_t tracks)65 int32_t InnerDemuxerDemo::PrintInfo(int32_t tracks)
66 {
67 for (int32_t i = 0; i < tracks; i++) {
68 printf("streams[%d]==>total frames=%" PRId64 ",KeyFrames=%" PRId64 "\n", i,
69 frames_[i] + key_frames_[i], key_frames_[i]);
70 }
71 return 0;
72 }
73
isEOS(std::map<uint32_t,bool> & countFlag)74 bool InnerDemuxerDemo::isEOS(std::map<uint32_t, bool>& countFlag)
75 {
76 for (auto iter = countFlag.begin(); iter != countFlag.end(); ++iter) {
77 if (!iter->second) {
78 return false;
79 }
80 }
81 return true;
82 }
83
ReadAllSamples(std::shared_ptr<AVSharedMemory> SampleMem,int32_t tracks)84 int32_t InnerDemuxerDemo::ReadAllSamples(std::shared_ptr<AVSharedMemory> SampleMem, int32_t tracks)
85 {
86 uint32_t bufferFlag = AVCodecBufferFlag::AVCODEC_BUFFER_FLAG_NONE;
87 std::map<uint32_t, bool> eosFlag;
88 for (int i = 0;i < tracks; i++) {
89 frames_[i] = 0;
90 key_frames_[i] = 0;
91 eosFlag[i] = false;
92 }
93 int32_t ret = -1;
94 while (!isEOS(eosFlag)) {
95 for (int32_t i = 0; i < tracks; i++) {
96 ret = ReadSample(i, SampleMem, sampleInfo, bufferFlag);
97 if (ret == 0 && (bufferFlag & AVCODEC_BUFFER_FLAG_EOS)) {
98 eosFlag[i] = true;
99 continue;
100 }
101 if (ret == 0 && (bufferFlag & AVCODEC_BUFFER_FLAG_SYNC_FRAME)) {
102 key_frames_[i]++;
103 } else if (ret == 0 && (bufferFlag & AVCODEC_BUFFER_FLAG_NONE) == 0) {
104 frames_[i]++;
105 } else {
106 printf("the flags is error ret=%d\n", ret);
107 printf("the bufferFlag=%d, sampleInfo.size=%d, sampleInfo.pts=%" PRId64 "\n",
108 bufferFlag, sampleInfo.size, sampleInfo.presentationTimeUs);
109 return ret;
110 }
111 }
112 }
113 PrintInfo(tracks);
114 return ret;
115 }
116
ReadSample(uint32_t trackIndex,std::shared_ptr<AVSharedMemory> mem,AVCodecBufferInfo & bufInfo,uint32_t & bufferFlag)117 int32_t InnerDemuxerDemo::ReadSample(uint32_t trackIndex, std::shared_ptr<AVSharedMemory> mem,
118 AVCodecBufferInfo &bufInfo, uint32_t &bufferFlag)
119 {
120 int32_t ret = this->demuxer_->ReadSample(trackIndex, mem, bufInfo, bufferFlag);
121 if (ret != 0) {
122 return ret;
123 }
124 return ret;
125 }
126
SeekToTime(int64_t millisecond,Media::SeekMode mode)127 int32_t InnerDemuxerDemo::SeekToTime(int64_t millisecond, Media::SeekMode mode)
128 {
129 int32_t ret = demuxer_->SeekToTime(millisecond, mode);
130 if (ret != 0) {
131 printf("SeekToTime is failed\n");
132 }
133 return ret;
134 }
135 }
136 }
137