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 OHOS_HDI_NNR_V1_0_SHARED_BUFFER_PARSER_H 17 #define OHOS_HDI_NNR_V1_0_SHARED_BUFFER_PARSER_H 18 19 #include "ashmem.h" 20 #include "v1_0/nnrt_types.h" 21 22 namespace OHOS { 23 namespace HDI { 24 namespace Nnrt { 25 namespace V1_0 { 26 namespace { 27 const int INVALID_FD = -1; 28 } 29 30 class SharedBufferParser { 31 public: SharedBufferParser()32 SharedBufferParser() {}; 33 ~SharedBufferParser(); 34 35 int32_t Init(const SharedBuffer& buffer); 36 int32_t Init(const std::string& name, int32_t size); 37 void* GetBufferPtr(); 38 SharedBuffer GetBuffer(); 39 40 private: 41 SharedBuffer m_buffer; 42 sptr<Ashmem> m_ashptr {nullptr}; 43 void* m_bufferAddr {nullptr}; 44 }; 45 } // V1_0 46 } // Nnrt 47 } // HDI 48 } // OHOS 49 #endif // OHOS_HDI_NNR_V1_0_SHARED_BUFFER_PARSER_H