Shunkai Yao | 6afc855 | 2022-10-26 22:47:20 +0000 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2022 The Android Open Source Project |
| 3 | * |
| 4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
| 5 | * you may not use this file except in compliance with the License. |
| 6 | * You may obtain a copy of the License at |
| 7 | * |
| 8 | * http://www.apache.org/licenses/LICENSE-2.0 |
| 9 | * |
| 10 | * Unless required by applicable law or agreed to in writing, software |
| 11 | * distributed under the License is distributed on an "AS IS" BASIS, |
| 12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 13 | * See the License for the specific language governing permissions and |
| 14 | * limitations under the License. |
| 15 | */ |
| 16 | |
Mikhail Naganov | 872d4a6 | 2023-03-09 18:19:01 -0800 | [diff] [blame] | 17 | #include <algorithm> |
Shunkai Yao | 6afc855 | 2022-10-26 22:47:20 +0000 | [diff] [blame] | 18 | #include <cstddef> |
Mikhail Naganov | 872d4a6 | 2023-03-09 18:19:01 -0800 | [diff] [blame] | 19 | |
Shunkai Yao | 6afc855 | 2022-10-26 22:47:20 +0000 | [diff] [blame] | 20 | #define LOG_TAG "AHAL_VirtualizerSw" |
| 21 | #include <Utils.h> |
Shunkai Yao | 6afc855 | 2022-10-26 22:47:20 +0000 | [diff] [blame] | 22 | #include <android-base/logging.h> |
| 23 | #include <fmq/AidlMessageQueue.h> |
| 24 | |
| 25 | #include "VirtualizerSw.h" |
| 26 | |
Shunkai Yao | c12e082 | 2022-12-12 07:13:58 +0000 | [diff] [blame] | 27 | using aidl::android::hardware::audio::effect::Descriptor; |
Shunkai Yao | 6afc855 | 2022-10-26 22:47:20 +0000 | [diff] [blame] | 28 | using aidl::android::hardware::audio::effect::IEffect; |
Shunkai Yao | 812d5b4 | 2022-11-16 18:08:50 +0000 | [diff] [blame] | 29 | using aidl::android::hardware::audio::effect::kVirtualizerSwImplUUID; |
Shunkai Yao | 6afc855 | 2022-10-26 22:47:20 +0000 | [diff] [blame] | 30 | using aidl::android::hardware::audio::effect::State; |
| 31 | using aidl::android::hardware::audio::effect::VirtualizerSw; |
Shunkai Yao | 58aaf5b | 2023-02-06 07:25:09 +0000 | [diff] [blame] | 32 | using aidl::android::media::audio::common::AudioChannelLayout; |
| 33 | using aidl::android::media::audio::common::AudioDeviceDescription; |
Shunkai Yao | 8781102 | 2023-02-13 17:40:37 +0000 | [diff] [blame] | 34 | using aidl::android::media::audio::common::AudioDeviceType; |
Shunkai Yao | 6afc855 | 2022-10-26 22:47:20 +0000 | [diff] [blame] | 35 | using aidl::android::media::audio::common::AudioUuid; |
| 36 | |
| 37 | extern "C" binder_exception_t createEffect(const AudioUuid* in_impl_uuid, |
| 38 | std::shared_ptr<IEffect>* instanceSpp) { |
Shunkai Yao | 812d5b4 | 2022-11-16 18:08:50 +0000 | [diff] [blame] | 39 | if (!in_impl_uuid || *in_impl_uuid != kVirtualizerSwImplUUID) { |
Shunkai Yao | 6afc855 | 2022-10-26 22:47:20 +0000 | [diff] [blame] | 40 | LOG(ERROR) << __func__ << "uuid not supported"; |
| 41 | return EX_ILLEGAL_ARGUMENT; |
| 42 | } |
| 43 | if (instanceSpp) { |
| 44 | *instanceSpp = ndk::SharedRefBase::make<VirtualizerSw>(); |
| 45 | LOG(DEBUG) << __func__ << " instance " << instanceSpp->get() << " created"; |
| 46 | return EX_NONE; |
| 47 | } else { |
| 48 | LOG(ERROR) << __func__ << " invalid input parameter!"; |
| 49 | return EX_ILLEGAL_ARGUMENT; |
| 50 | } |
| 51 | } |
| 52 | |
Shunkai Yao | c12e082 | 2022-12-12 07:13:58 +0000 | [diff] [blame] | 53 | extern "C" binder_exception_t queryEffect(const AudioUuid* in_impl_uuid, Descriptor* _aidl_return) { |
| 54 | if (!in_impl_uuid || *in_impl_uuid != kVirtualizerSwImplUUID) { |
| 55 | LOG(ERROR) << __func__ << "uuid not supported"; |
| 56 | return EX_ILLEGAL_ARGUMENT; |
Shunkai Yao | 6afc855 | 2022-10-26 22:47:20 +0000 | [diff] [blame] | 57 | } |
Shunkai Yao | c12e082 | 2022-12-12 07:13:58 +0000 | [diff] [blame] | 58 | *_aidl_return = VirtualizerSw::kDescriptor; |
Shunkai Yao | 6afc855 | 2022-10-26 22:47:20 +0000 | [diff] [blame] | 59 | return EX_NONE; |
| 60 | } |
| 61 | |
| 62 | namespace aidl::android::hardware::audio::effect { |
| 63 | |
Shunkai Yao | c12e082 | 2022-12-12 07:13:58 +0000 | [diff] [blame] | 64 | const std::string VirtualizerSw::kEffectName = "VirtualizerSw"; |
Shunkai Yao | 8781102 | 2023-02-13 17:40:37 +0000 | [diff] [blame] | 65 | |
| 66 | const std::vector<Range::VirtualizerRange> VirtualizerSw::kRanges = { |
| 67 | MAKE_RANGE(Virtualizer, strengthPm, 0, 1000), |
| 68 | /* speakerAngle is get-only, set min > max */ |
| 69 | MAKE_RANGE(Virtualizer, speakerAngles, {Virtualizer::ChannelAngle({.channel = 1})}, |
Shunkai Yao | 2c41a20 | 2023-02-21 17:47:24 +0000 | [diff] [blame] | 70 | {Virtualizer::ChannelAngle({.channel = 0})})}; |
Shunkai Yao | 8781102 | 2023-02-13 17:40:37 +0000 | [diff] [blame] | 71 | |
| 72 | const Capability VirtualizerSw::kCapability = { |
| 73 | .range = Range::make<Range::virtualizer>(VirtualizerSw::kRanges)}; |
| 74 | |
Shunkai Yao | c12e082 | 2022-12-12 07:13:58 +0000 | [diff] [blame] | 75 | const Descriptor VirtualizerSw::kDescriptor = { |
| 76 | .common = {.id = {.type = kVirtualizerTypeUUID, |
| 77 | .uuid = kVirtualizerSwImplUUID, |
Sham Rathod | 5fd232f | 2022-12-08 13:56:38 +0530 | [diff] [blame] | 78 | .proxy = kVirtualizerProxyUUID}, |
Shunkai Yao | c12e082 | 2022-12-12 07:13:58 +0000 | [diff] [blame] | 79 | .flags = {.type = Flags::Type::INSERT, |
| 80 | .insert = Flags::Insert::FIRST, |
| 81 | .volume = Flags::Volume::CTRL}, |
| 82 | .name = VirtualizerSw::kEffectName, |
| 83 | .implementor = "The Android Open Source Project"}, |
Shunkai Yao | 8781102 | 2023-02-13 17:40:37 +0000 | [diff] [blame] | 84 | .capability = VirtualizerSw::kCapability}; |
Shunkai Yao | c12e082 | 2022-12-12 07:13:58 +0000 | [diff] [blame] | 85 | |
Shunkai Yao | 6afc855 | 2022-10-26 22:47:20 +0000 | [diff] [blame] | 86 | ndk::ScopedAStatus VirtualizerSw::getDescriptor(Descriptor* _aidl_return) { |
| 87 | LOG(DEBUG) << __func__ << kDescriptor.toString(); |
| 88 | *_aidl_return = kDescriptor; |
| 89 | return ndk::ScopedAStatus::ok(); |
| 90 | } |
| 91 | |
| 92 | ndk::ScopedAStatus VirtualizerSw::setParameterSpecific(const Parameter::Specific& specific) { |
| 93 | RETURN_IF(Parameter::Specific::virtualizer != specific.getTag(), EX_ILLEGAL_ARGUMENT, |
| 94 | "EffectNotSupported"); |
Shunkai Yao | 6afc855 | 2022-10-26 22:47:20 +0000 | [diff] [blame] | 95 | |
Sham Rathod | 2d319dc | 2022-11-29 15:06:12 +0530 | [diff] [blame] | 96 | auto& vrParam = specific.get<Parameter::Specific::virtualizer>(); |
Shunkai Yao | 8781102 | 2023-02-13 17:40:37 +0000 | [diff] [blame] | 97 | RETURN_IF(!inRange(vrParam, kRanges), EX_ILLEGAL_ARGUMENT, "outOfRange"); |
Sham Rathod | 2d319dc | 2022-11-29 15:06:12 +0530 | [diff] [blame] | 98 | auto tag = vrParam.getTag(); |
| 99 | |
| 100 | switch (tag) { |
| 101 | case Virtualizer::strengthPm: { |
Sham Rathod | 2d319dc | 2022-11-29 15:06:12 +0530 | [diff] [blame] | 102 | RETURN_IF(mContext->setVrStrength(vrParam.get<Virtualizer::strengthPm>()) != |
| 103 | RetCode::SUCCESS, |
Shunkai Yao | 58aaf5b | 2023-02-06 07:25:09 +0000 | [diff] [blame] | 104 | EX_ILLEGAL_ARGUMENT, "setStrengthPmFailed"); |
Sham Rathod | 2d319dc | 2022-11-29 15:06:12 +0530 | [diff] [blame] | 105 | return ndk::ScopedAStatus::ok(); |
| 106 | } |
Shunkai Yao | 58aaf5b | 2023-02-06 07:25:09 +0000 | [diff] [blame] | 107 | case Virtualizer::device: { |
| 108 | RETURN_IF(mContext->setForcedDevice(vrParam.get<Virtualizer::device>()) != |
| 109 | RetCode::SUCCESS, |
| 110 | EX_ILLEGAL_ARGUMENT, "setDeviceFailed"); |
| 111 | return ndk::ScopedAStatus::ok(); |
| 112 | } |
| 113 | case Virtualizer::speakerAngles: |
| 114 | FALLTHROUGH_INTENDED; |
| 115 | case Virtualizer::vendor: { |
Sham Rathod | 2d319dc | 2022-11-29 15:06:12 +0530 | [diff] [blame] | 116 | LOG(ERROR) << __func__ << " unsupported tag: " << toString(tag); |
| 117 | return ndk::ScopedAStatus::fromExceptionCodeWithMessage(EX_ILLEGAL_ARGUMENT, |
| 118 | "VirtualizerTagNotSupported"); |
| 119 | } |
| 120 | } |
Shunkai Yao | 6afc855 | 2022-10-26 22:47:20 +0000 | [diff] [blame] | 121 | } |
| 122 | |
| 123 | ndk::ScopedAStatus VirtualizerSw::getParameterSpecific(const Parameter::Id& id, |
| 124 | Parameter::Specific* specific) { |
| 125 | auto tag = id.getTag(); |
| 126 | RETURN_IF(Parameter::Id::virtualizerTag != tag, EX_ILLEGAL_ARGUMENT, "wrongIdTag"); |
Sham Rathod | 2d319dc | 2022-11-29 15:06:12 +0530 | [diff] [blame] | 127 | auto vrId = id.get<Parameter::Id::virtualizerTag>(); |
| 128 | auto vrIdTag = vrId.getTag(); |
| 129 | switch (vrIdTag) { |
| 130 | case Virtualizer::Id::commonTag: |
| 131 | return getParameterVirtualizer(vrId.get<Virtualizer::Id::commonTag>(), specific); |
Shunkai Yao | 58aaf5b | 2023-02-06 07:25:09 +0000 | [diff] [blame] | 132 | case Virtualizer::Id::speakerAnglesPayload: |
| 133 | return getSpeakerAngles(vrId.get<Virtualizer::Id::speakerAnglesPayload>(), specific); |
| 134 | case Virtualizer::Id::vendorExtensionTag: { |
Sham Rathod | 2d319dc | 2022-11-29 15:06:12 +0530 | [diff] [blame] | 135 | LOG(ERROR) << __func__ << " unsupported tag: " << toString(tag); |
| 136 | return ndk::ScopedAStatus::fromExceptionCodeWithMessage(EX_ILLEGAL_ARGUMENT, |
| 137 | "VirtualizerTagNotSupported"); |
Shunkai Yao | 58aaf5b | 2023-02-06 07:25:09 +0000 | [diff] [blame] | 138 | } |
Sham Rathod | 2d319dc | 2022-11-29 15:06:12 +0530 | [diff] [blame] | 139 | } |
| 140 | } |
| 141 | |
| 142 | ndk::ScopedAStatus VirtualizerSw::getParameterVirtualizer(const Virtualizer::Tag& tag, |
| 143 | Parameter::Specific* specific) { |
| 144 | RETURN_IF(!mContext, EX_NULL_POINTER, "nullContext"); |
| 145 | |
| 146 | Virtualizer vrParam; |
| 147 | switch (tag) { |
| 148 | case Virtualizer::strengthPm: { |
| 149 | vrParam.set<Virtualizer::strengthPm>(mContext->getVrStrength()); |
| 150 | break; |
| 151 | } |
Shunkai Yao | 58aaf5b | 2023-02-06 07:25:09 +0000 | [diff] [blame] | 152 | case Virtualizer::device: { |
| 153 | vrParam.set<Virtualizer::device>(mContext->getForcedDevice()); |
| 154 | break; |
| 155 | } |
| 156 | case Virtualizer::speakerAngles: |
| 157 | FALLTHROUGH_INTENDED; |
| 158 | case Virtualizer::vendor: { |
Sham Rathod | 2d319dc | 2022-11-29 15:06:12 +0530 | [diff] [blame] | 159 | LOG(ERROR) << __func__ << " unsupported tag: " << toString(tag); |
| 160 | return ndk::ScopedAStatus::fromExceptionCodeWithMessage(EX_ILLEGAL_ARGUMENT, |
| 161 | "VirtualizerTagNotSupported"); |
| 162 | } |
| 163 | } |
| 164 | |
| 165 | specific->set<Parameter::Specific::virtualizer>(vrParam); |
Shunkai Yao | 6afc855 | 2022-10-26 22:47:20 +0000 | [diff] [blame] | 166 | return ndk::ScopedAStatus::ok(); |
| 167 | } |
| 168 | |
Shunkai Yao | 58aaf5b | 2023-02-06 07:25:09 +0000 | [diff] [blame] | 169 | ndk::ScopedAStatus VirtualizerSw::getSpeakerAngles(const Virtualizer::SpeakerAnglesPayload payload, |
| 170 | Parameter::Specific* specific) { |
| 171 | std::vector<Virtualizer::ChannelAngle> angles; |
Mikhail Naganov | 872d4a6 | 2023-03-09 18:19:01 -0800 | [diff] [blame] | 172 | const auto chNum = ::aidl::android::hardware::audio::common::getChannelCount(payload.layout); |
Shunkai Yao | 2c41a20 | 2023-02-21 17:47:24 +0000 | [diff] [blame] | 173 | if (chNum == 1) { |
Shunkai Yao | 58aaf5b | 2023-02-06 07:25:09 +0000 | [diff] [blame] | 174 | angles = {{.channel = (int32_t)AudioChannelLayout::CHANNEL_FRONT_LEFT, |
| 175 | .azimuthDegree = 0, |
| 176 | .elevationDegree = 0}}; |
Shunkai Yao | 2c41a20 | 2023-02-21 17:47:24 +0000 | [diff] [blame] | 177 | } else if (chNum == 2) { |
Shunkai Yao | 58aaf5b | 2023-02-06 07:25:09 +0000 | [diff] [blame] | 178 | angles = {{.channel = (int32_t)AudioChannelLayout::CHANNEL_FRONT_LEFT, |
| 179 | .azimuthDegree = -90, |
| 180 | .elevationDegree = 0}, |
| 181 | {.channel = (int32_t)AudioChannelLayout::CHANNEL_FRONT_RIGHT, |
| 182 | .azimuthDegree = 90, |
| 183 | .elevationDegree = 0}}; |
Shunkai Yao | 2c41a20 | 2023-02-21 17:47:24 +0000 | [diff] [blame] | 184 | } else { |
| 185 | return ndk::ScopedAStatus::fromExceptionCodeWithMessage(EX_ILLEGAL_ARGUMENT, |
| 186 | "supportUpTo2Ch"); |
Shunkai Yao | 58aaf5b | 2023-02-06 07:25:09 +0000 | [diff] [blame] | 187 | } |
| 188 | |
| 189 | Virtualizer param = Virtualizer::make<Virtualizer::speakerAngles>(angles); |
| 190 | specific->set<Parameter::Specific::virtualizer>(param); |
| 191 | return ndk::ScopedAStatus::ok(); |
| 192 | } |
| 193 | |
Shunkai Yao | 6afc855 | 2022-10-26 22:47:20 +0000 | [diff] [blame] | 194 | std::shared_ptr<EffectContext> VirtualizerSw::createContext(const Parameter::Common& common) { |
| 195 | if (mContext) { |
| 196 | LOG(DEBUG) << __func__ << " context already exist"; |
Shraddha Basantwani | f627d80 | 2022-11-08 14:45:07 +0530 | [diff] [blame] | 197 | } else { |
| 198 | mContext = std::make_shared<VirtualizerSwContext>(1 /* statusFmqDepth */, common); |
Shunkai Yao | 6afc855 | 2022-10-26 22:47:20 +0000 | [diff] [blame] | 199 | } |
Shraddha Basantwani | f627d80 | 2022-11-08 14:45:07 +0530 | [diff] [blame] | 200 | |
| 201 | return mContext; |
| 202 | } |
| 203 | |
| 204 | std::shared_ptr<EffectContext> VirtualizerSw::getContext() { |
Shunkai Yao | 6afc855 | 2022-10-26 22:47:20 +0000 | [diff] [blame] | 205 | return mContext; |
| 206 | } |
| 207 | |
| 208 | RetCode VirtualizerSw::releaseContext() { |
| 209 | if (mContext) { |
| 210 | mContext.reset(); |
| 211 | } |
| 212 | return RetCode::SUCCESS; |
| 213 | } |
| 214 | |
| 215 | // Processing method running in EffectWorker thread. |
Shraddha Basantwani | f627d80 | 2022-11-08 14:45:07 +0530 | [diff] [blame] | 216 | IEffect::Status VirtualizerSw::effectProcessImpl(float* in, float* out, int samples) { |
Shunkai Yao | 6afc855 | 2022-10-26 22:47:20 +0000 | [diff] [blame] | 217 | // TODO: get data buffer and process. |
Shraddha Basantwani | f627d80 | 2022-11-08 14:45:07 +0530 | [diff] [blame] | 218 | LOG(DEBUG) << __func__ << " in " << in << " out " << out << " samples " << samples; |
| 219 | for (int i = 0; i < samples; i++) { |
Shunkai Yao | 6afc855 | 2022-10-26 22:47:20 +0000 | [diff] [blame] | 220 | *out++ = *in++; |
| 221 | } |
Shraddha Basantwani | f627d80 | 2022-11-08 14:45:07 +0530 | [diff] [blame] | 222 | return {STATUS_OK, samples, samples}; |
Shunkai Yao | 6afc855 | 2022-10-26 22:47:20 +0000 | [diff] [blame] | 223 | } |
| 224 | |
Sham Rathod | e808b07 | 2022-12-27 11:56:30 +0530 | [diff] [blame] | 225 | RetCode VirtualizerSwContext::setVrStrength(int strength) { |
Sham Rathod | e808b07 | 2022-12-27 11:56:30 +0530 | [diff] [blame] | 226 | mStrength = strength; |
| 227 | return RetCode::SUCCESS; |
| 228 | } |
| 229 | |
Shunkai Yao | 6afc855 | 2022-10-26 22:47:20 +0000 | [diff] [blame] | 230 | } // namespace aidl::android::hardware::audio::effect |