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 AVCodecBufferFlag 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_SYNC_FRAME) {
98 key_frames_[i]++;
99 } else if (ret == 0 && bufferFlag == AVCODEC_BUFFER_FLAG_NONE) {
100 frames_[i]++;
101 } else if (ret == 0 && bufferFlag == AVCODEC_BUFFER_FLAG_EOS) {
102 eosFlag[i] = true;
103 } else {
104 printf("the flags is error ret=%d\n", ret);
105 printf("the bufferFlag=%d, sampleInfo.size=%d, sampleInfo.pts=%" PRId64 "\n",
106 bufferFlag, sampleInfo.size, sampleInfo.presentationTimeUs);
107 return ret;
108 }
109 }
110 }
111 PrintInfo(tracks);
112 return ret;
113 }
114
ReadSample(uint32_t trackIndex,std::shared_ptr<AVSharedMemory> mem,AVCodecBufferInfo & bufInfo,AVCodecBufferFlag & bufferFlag)115 int32_t InnerDemuxerDemo::ReadSample(uint32_t trackIndex, std::shared_ptr<AVSharedMemory> mem,
116 AVCodecBufferInfo &bufInfo, AVCodecBufferFlag &bufferFlag)
117 {
118 int32_t ret = this->demuxer_->ReadSample(trackIndex, mem, bufInfo, bufferFlag);
119 if (ret != 0) {
120 return ret;
121 }
122 return ret;
123 }
124
SeekToTime(int64_t millisecond,AVSeekMode mode)125 int32_t InnerDemuxerDemo::SeekToTime(int64_t millisecond, AVSeekMode mode)
126 {
127 int32_t ret = demuxer_->SeekToTime(millisecond, mode);
128 if (ret != 0) {
129 printf("SeekToTime is failed\n");
130 }
131 return ret;
132 }
133 }
134 }
135