1 /*
2  * Copyright (c) 2021-2021 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 
17 #include "plugin/core/plugin_wrapper.h"
18 #include <set>
19 
20 namespace {
21 std::set<OHOS::Media::Plugin::Tag> g_metaIdSet = {
22     OHOS::Media::Plugin::Tag::MIME,
23     OHOS::Media::Plugin::Tag::TRACK_ID,
24     OHOS::Media::Plugin::Tag::MEDIA_CODEC_CONFIG,
25     OHOS::Media::Plugin::Tag::MEDIA_BITRATE,
26     OHOS::Media::Plugin::Tag::AUDIO_CHANNELS,
27     OHOS::Media::Plugin::Tag::AUDIO_SAMPLE_RATE,
28     OHOS::Media::Plugin::Tag::AUDIO_SAMPLE_FORMAT,
29     OHOS::Media::Plugin::Tag::AUDIO_SAMPLE_PER_FRAME,
30     OHOS::Media::Plugin::Tag::AUDIO_CHANNEL_LAYOUT,
31     OHOS::Media::Plugin::Tag::MEDIA_TITLE,
32     OHOS::Media::Plugin::Tag::MEDIA_ARTIST,
33     OHOS::Media::Plugin::Tag::MEDIA_LYRICIST,
34     OHOS::Media::Plugin::Tag::MEDIA_ALBUM,
35     OHOS::Media::Plugin::Tag::MEDIA_ALBUM_ARTIST,
36     OHOS::Media::Plugin::Tag::MEDIA_DATE,
37     OHOS::Media::Plugin::Tag::MEDIA_COMMENT,
38     OHOS::Media::Plugin::Tag::MEDIA_GENRE,
39     OHOS::Media::Plugin::Tag::MEDIA_DESCRIPTION,
40     OHOS::Media::Plugin::Tag::MEDIA_COPYRIGHT,
41     OHOS::Media::Plugin::Tag::MEDIA_LANGUAGE,
42     OHOS::Media::Plugin::Tag::MEDIA_LYRICS,
43     OHOS::Media::Plugin::Tag::MEDIA_DURATION,
44     OHOS::Media::Plugin::Tag::MEDIA_FILE_URI,
45     OHOS::Media::Plugin::Tag::MEDIA_FILE_SIZE,
46     OHOS::Media::Plugin::Tag::MEDIA_SEEKABLE,
47     OHOS::Media::Plugin::Tag::MEDIA_TYPE,
48     OHOS::Media::Plugin::Tag::AUDIO_MPEG_VERSION,
49     OHOS::Media::Plugin::Tag::AUDIO_MPEG_LAYER,
50     OHOS::Media::Plugin::Tag::AUDIO_AAC_PROFILE,
51     OHOS::Media::Plugin::Tag::AUDIO_AAC_LEVEL,
52     OHOS::Media::Plugin::Tag::AUDIO_AAC_STREAM_FORMAT,
53     OHOS::Media::Plugin::Tag::VIDEO_WIDTH,
54     OHOS::Media::Plugin::Tag::VIDEO_HEIGHT,
55     OHOS::Media::Plugin::Tag::VIDEO_PIXEL_FORMAT,
56     OHOS::Media::Plugin::Tag::VIDEO_BIT_STREAM_FORMAT,
57     OHOS::Media::Plugin::Tag::VIDEO_FRAME_RATE,
58     OHOS::Media::Plugin::Tag::VIDEO_H264_LEVEL,
59     OHOS::Media::Plugin::Tag::VIDEO_H264_PROFILE,
60     OHOS::Media::Plugin::Tag::BITS_PER_CODED_SAMPLE,
61 };
62 }
63 
64 namespace OHOS {
65 namespace Media {
66 namespace Plugin {
DataSourceWrapper(uint32_t pkgVersion,std::shared_ptr<DataSourceHelper> dataSource)67 DataSourceWrapper::DataSourceWrapper(uint32_t pkgVersion, std::shared_ptr<DataSourceHelper> dataSource)
68     : version(pkgVersion), helper(std::move(dataSource))
69 {
70 }
71 
ReadAt(int64_t offset,std::shared_ptr<Buffer> & buffer,size_t expectedLen)72 Status DataSourceWrapper::ReadAt(int64_t offset, std::shared_ptr<Buffer>& buffer, size_t expectedLen)
73 {
74     return helper->ReadAt(offset, buffer, expectedLen);
75 }
76 
GetSize(uint64_t & size)77 Status DataSourceWrapper::GetSize(uint64_t& size)
78 {
79     return helper->GetSize(size);
80 }
81 
GetSeekable()82 Seekable DataSourceWrapper::GetSeekable()
83 {
84     return helper->GetSeekable();
85 }
86 
DataSinkWrapper(uint32_t pkgVersion,std::shared_ptr<DataSinkHelper> dataSink)87 DataSinkWrapper::DataSinkWrapper(uint32_t pkgVersion, std::shared_ptr<DataSinkHelper> dataSink)
88     : version(pkgVersion), helper(std::move(dataSink)) {}
89 
WriteAt(int64_t offset,const std::shared_ptr<Buffer> & buffer)90 Status DataSinkWrapper::WriteAt(int64_t offset, const std::shared_ptr<Buffer>& buffer)
91 {
92     return helper->WriteAt(offset, buffer);
93 }
94 
ConvertToMediaInfoHelper(uint32_t pkgVersion,const MediaInfo & src,MediaInfoHelper & dest)95 void ConvertToMediaInfoHelper(uint32_t pkgVersion, const MediaInfo& src, MediaInfoHelper& dest)
96 {
97     (void)(pkgVersion);
98     for (auto const& global : src.general) {
99         dest.globalMeta.SetData(global.first, global.second);
100     }
101     size_t streamSize = src.tracks.size();
102     if (streamSize <= 0) {
103         return;
104     }
105     dest.trackMeta.resize(streamSize);
106     for (size_t i = 0; i < streamSize; ++i) {
107         for (auto const& meta : src.tracks[i]) {
108             if (g_metaIdSet.count(meta.first)) {
109                 dest.trackMeta[i].SetData(meta.first, meta.second);
110             }
111         }
112     }
113 }
114 } // namespace Plugin
115 } // namespace Media
116 } // namespace OHOS
117