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 #ifndef NEURAL_NETWORK_RUNTIME_CLIENT_H
17 #define NEURAL_NETWORK_RUNTIME_CLIENT_H
18 
19 #include <cstddef>
20 #include <memory>
21 
22 namespace OHOS {
23 namespace NeuralNetworkRuntime {
24 class NNRtServiceApi {
25 public:
26     static NNRtServiceApi& GetInstance();
27     bool IsServiceAvaliable() const;
28 
29     int (*CheckModelSizeFromPath)(const char* path, bool& exceedLimit) = nullptr;
30     int (*CheckModelSizeFromBuffer)(const void* buffer, size_t size, bool& exceedLimit) = nullptr;
31     int (*CheckModelSizeFromModel)(void* model, bool& exceedLimit) = nullptr;
32     size_t (*GetNNRtModelIDFromPath)(const char*) = nullptr;
33     size_t (*GetNNRtModelIDFromCache)(const char* path, const char* modelName) = nullptr;
34     size_t (*GetNNRtModelIDFromBuffer)(const void* buffer, size_t size) = nullptr;
35     size_t (*GetNNRtModelIDFromModel)(void* model) = nullptr;
36     int (*SetModelID)(int callingPid, uint32_t hiaimodelID, size_t nnrtModelID) = nullptr;
37     int (*IsSupportAuthentication)(bool* supportStat) = nullptr;
38     int (*IsSupportScheduling)(bool* supportStat) = nullptr;
39     int (*Authentication)(int callingPid) = nullptr;
40     int (*Scheduling)(uint32_t hiaiModelId, bool* needModelLatency, const char* cachePath) = nullptr;
41     int (*UpdateModelLatency)(uint32_t hiaiModelId, int modelLatency) = nullptr;
42     int (*Unload)(uint32_t hiaiModelId) = nullptr;
43 
44 private:
45     bool m_serviceAvailable = false;
46     NNRtServiceApi() = default;
47     NNRtServiceApi(const NNRtServiceApi&) = delete;
48     NNRtServiceApi& operator=(const NNRtServiceApi&) = delete;
49     virtual ~NNRtServiceApi();
50 };
51 }  // namespace NeuralNetworkRuntime
52 }  // namespace OHOS
53 
54 #endif // NEURAL_NETWORK_RUNTIME_CLIENT_H