1 /*
2 * Copyright (c) 2022 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 #ifndef NEURAL_NETWORK_RUNTIME_UTILS_H
17 #define NEURAL_NETWORK_RUNTIME_UTILS_H
18
19 #include <string>
20 #include <memory>
21
22 #include "log.h"
23
24 namespace OHOS {
25 namespace NeuralNetworkRuntime {
26 template<typename T, typename... Args>
CreateSharedPtr(Args &&...args)27 std::shared_ptr<T> CreateSharedPtr(Args&&... args)
28 {
29 std::shared_ptr<T> tPtr = nullptr;
30 try {
31 tPtr = std::make_shared<T>(args...);
32 } catch (const std::bad_alloc& except) {
33 LOGW("Create a new shared pointer failed. Error: %s", except.what());
34 return nullptr;
35 }
36 return tPtr;
37 };
38
39 template<typename T, typename... Args>
CreateUniquePtr(Args &&...args)40 std::unique_ptr<T> CreateUniquePtr(Args&&... args)
41 {
42 std::unique_ptr<T> tPtr = nullptr;
43 try {
44 tPtr = std::make_unique<T>(args...);
45 } catch (const std::bad_alloc& except) {
46 LOGW("Create a new unique pointer failed. Error: %s", except.what());
47 return nullptr;
48 }
49 return tPtr;
50 }
51
52 std::string GenUniqueName(const std::string&, const std::string&, const std::string&);
53
54 } // namespace NeuralNetworkRuntime
55 } // namespace OHOS
56 #endif // NEURAL_NETWORK_RUNTIME_UTILS_H
57