diff options
author | Lev Proleev <levp@google.com> | 2021-03-16 22:17:28 +0000 |
---|---|---|
committer | Automerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com> | 2021-03-16 22:17:28 +0000 |
commit | 8dbed29e45d33d4a39734cfee29a4051bbdf363c (patch) | |
tree | d3e86841a89e4a0806c3b7e65f4a1de2b729a064 /neuralnetworks/aidl/utils/src/Service.cpp | |
parent | 7e21641013f4fe3dbfa6f23ceb98c0048785f7e6 (diff) | |
parent | 1bca9ab31526a5da3f8e3967c11225afa0548659 (diff) |
Add canonical types adapters for NNAPI AIDL interface am: 900c28a250 am: 1bca9ab315
Original change: https://android-review.googlesource.com/c/platform/hardware/interfaces/+/1640842
Change-Id: I7ca57555148ca7768beac642891e589d8e21b528
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 |