1 /*
2 * Copyright (C) 2024 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 #define MLOG_TAG "MediaCallTranscode"
17
18 #include "media_asset_manager_callback.h"
19 #include "media_call_transcode.h"
20 #include "medialibrary_napi_log.h"
21 #include "medialibrary_errno.h"
22
23 namespace OHOS {
24 namespace Media {
25
26 static MediaCallTranscode::CallbackType callback_;
27 static std::mutex transCoderMapMutex_;
28 static std::map<std::string, std::shared_ptr<TransCoder>> transCoderMap_;
29 static const int32_t INFO_TYPE_ERROR = 2;
TransCodeError(napi_env env,napi_value & result,int srcFd,int destFd,const std::string & errorMsg)30 void MediaCallTranscode::TransCodeError(napi_env env, napi_value &result,
31 int srcFd, int destFd, const std::string& errorMsg)
32 {
33 NAPI_ERR_LOG(" %{public}s", errorMsg.c_str());
34 close(srcFd);
35 close(destFd);
36 napi_get_boolean(env, false, &result);
37 }
38
CallTranscodeHandle(napi_env env,int srcFd,int destFd,napi_value & result,off_t & size,std::string requestId)39 void MediaCallTranscode::CallTranscodeHandle(napi_env env, int srcFd, int destFd,
40 napi_value &result, off_t &size, std::string requestId)
41 {
42 NAPI_INFO_LOG("CallTranscodeHandle start");
43 auto transCoder = TransCoderFactory::CreateTransCoder();
44 if (transCoder == nullptr) {
45 TransCodeError(env, result, srcFd, destFd, "Failed to create TransCoder");
46 return;
47 }
48 {
49 std::lock_guard<std::mutex> lock(transCoderMapMutex_);
50 transCoderMap_.insert(std::pair<std::string, std::shared_ptr<TransCoder>>(requestId, transCoder));
51 }
52 auto transCoderCb = std::make_shared<OHOS::Media::MediaAssetManagerCallback>();
53 if (transCoderCb == nullptr) {
54 TransCodeError(env, result, srcFd, destFd, "Failed to create MediaAssetManagerCallback");
55 return;
56 }
57 transCoderCb->SetRequestId(requestId);
58 int32_t ret = transCoder->SetTransCoderCallback(transCoderCb);
59 if (ret != E_OK) {
60 TransCodeError(env, result, srcFd, destFd, "Failed to set TransCoder callback");
61 return;
62 }
63 ret = transCoder->SetInputFile(srcFd, 0, size);
64 if (ret != E_OK) {
65 TransCodeError(env, result, srcFd, destFd, "Failed to set input file for TransCoder");
66 return;
67 }
68 ret = transCoder->SetOutputFile(destFd);
69 if (ret != E_OK) {
70 TransCodeError(env, result, srcFd, destFd, "Failed to set output file for TransCoder");
71 return;
72 }
73 ret = transCoder->SetOutputFormat(FORMAT_MPEG_4);
74 if (ret != E_OK) {
75 TransCodeError(env, result, srcFd, destFd, "Failed to SetOutputFormat");
76 return;
77 }
78 ret = transCoder->Prepare();
79 if (ret != E_OK) {
80 TransCodeError(env, result, srcFd, destFd, "Failed to prepare TransCoder");
81 return;
82 }
83 ret = transCoder->Start();
84 if (ret != E_OK) {
85 TransCodeError(env, result, srcFd, destFd, "Failed to TransCoder Start");
86 return;
87 }
88 napi_get_boolean(env, true, &result);
89 }
90
CallTranscodeRelease(const std::string & requestId)91 void MediaCallTranscode::CallTranscodeRelease(const std::string& requestId)
92 {
93 std::lock_guard<std::mutex> lock(transCoderMapMutex_);
94 auto tcm = transCoderMap_.find(requestId);
95 if (tcm == transCoderMap_.end()) {
96 return;
97 }
98 tcm->second->Release();
99 transCoderMap_.erase(tcm);
100 }
101
RegisterCallback(const CallbackType & cb)102 void MediaCallTranscode::RegisterCallback(const CallbackType &cb)
103 {
104 callback_ = cb;
105 }
106
OnInfo(int32_t type,int32_t extra)107 void MediaAssetManagerCallback::OnInfo(int32_t type, int32_t extra)
108 {
109 NAPI_INFO_LOG("MediaAssetManagerCallback OnInfo type:%{public}d extra:%{public}d", type, extra);
110 if (callback_) {
111 callback_(type, extra, requestId_);
112 }
113 }
114
OnError(int32_t errCode,const std::string & errorMsg)115 void MediaAssetManagerCallback::OnError(int32_t errCode, const std::string &errorMsg)
116 {
117 std::lock_guard<std::mutex> lock(transCoderMapMutex_);
118 auto tcm = transCoderMap_.find(requestId_);
119 if (tcm != transCoderMap_.end()) {
120 transCoderMap_.erase(tcm);
121 }
122 NAPI_ERR_LOG("MediaAssetManagerCallback OnInfo errorMsg:%{public}s", errorMsg.c_str());
123 int32_t type = INFO_TYPE_ERROR;
124 if (callback_) {
125 callback_(type, 0, requestId_);
126 }
127 }
128
SetRequestId(std::string requestId)129 void MediaAssetManagerCallback::SetRequestId(std::string requestId)
130 {
131 requestId_ = requestId;
132 }
133 } // namespace Media
134 } // namespace OHOS
135