diff options
author | Lev Proleev <levp@google.com> | 2021-03-16 21:37:53 +0000 |
---|---|---|
committer | Automerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com> | 2021-03-16 21:37:53 +0000 |
commit | 1bca9ab31526a5da3f8e3967c11225afa0548659 (patch) | |
tree | d68a530636a10f7f32f909bc580f85e749ebd6ee /neuralnetworks/aidl/utils/src/Service.cpp | |
parent | faa3ea282178c939c1e24f198353ccad62d45607 (diff) | |
parent | 900c28a250297964cf9aa65653b9adce4e6db27a (diff) |
Add canonical types adapters for NNAPI AIDL interface am: 900c28a250
Original change: https://android-review.googlesource.com/c/platform/hardware/interfaces/+/1640842
Change-Id: I6077cc5a7adf58582ba25773e954a8c06b5019fe
Diffstat (limited to 'neuralnetworks/aidl/utils/src/Service.cpp')
-rw-r--r-- | neuralnetworks/aidl/utils/src/Service.cpp | 50 |
1 files changed, 50 insertions, 0 deletions
diff --git a/neuralnetworks/aidl/utils/src/Service.cpp b/neuralnetworks/aidl/utils/src/Service.cpp new file mode 100644 index 0000000000..5ec6ded5e5 --- /dev/null +++ b/neuralnetworks/aidl/utils/src/Service.cpp @@ -0,0 +1,50 @@ +/* + * Copyright (C) 2021 The Android Open Source Project + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +#include "Service.h" + +#include <android/binder_auto_utils.h> +#include <android/binder_manager.h> + +#include <nnapi/IDevice.h> +#include <nnapi/Result.h> +#include <nnapi/Types.h> +#include <nnapi/hal/ResilientDevice.h> +#include <string> + +#include "Device.h" + +namespace aidl::android::hardware::neuralnetworks::utils { + +nn::GeneralResult<nn::SharedDevice> getDevice(const std::string& name) { + hal::utils::ResilientDevice::Factory makeDevice = + [name](bool blocking) -> nn::GeneralResult<nn::SharedDevice> { + auto service = blocking ? IDevice::fromBinder( + ndk::SpAIBinder(AServiceManager_getService(name.c_str()))) + : IDevice::fromBinder(ndk::SpAIBinder( + AServiceManager_checkService(name.c_str()))); + if (service == nullptr) { + return NN_ERROR() << (blocking ? "AServiceManager_getService" + : "AServiceManager_checkService") + << " returned nullptr"; + } + return Device::create(name, std::move(service)); + }; + + return hal::utils::ResilientDevice::create(std::move(makeDevice)); +} + +} // namespace aidl::android::hardware::neuralnetworks::utils |