1 /* 2 * Copyright (c) 2024 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 AUDIO_EFFECT_CHAIN_H 17 #define AUDIO_EFFECT_CHAIN_H 18 19 #include <atomic> 20 21 #include "audio_effect.h" 22 #include "audio_utils.h" 23 24 #ifdef SENSOR_ENABLE 25 #include "audio_head_tracker.h" 26 #endif 27 #include "audio_effect_hdi_param.h" 28 #ifdef WINDOW_MANAGER_ENABLE 29 #include "audio_effect_rotation.h" 30 #endif 31 #include "audio_effect_volume.h" 32 33 namespace OHOS { 34 namespace AudioStandard { 35 enum AudioEffectChainSetParamIndex { 36 COMMAND_CODE_INDEX = 0, 37 SCENE_TYPE_INDEX = 1, 38 EFFECT_MODE_INDEX = 2, 39 ROTATION_INDEX = 3, 40 VOLUME_INDEX = 4, 41 EXTRA_SCENE_TYPE_INDEX = 5, 42 SPATIAL_DEVICE_TYPE_INDEX = 6, 43 SPATIALIZATION_SCENE_TYPE_INDEX = 7, 44 SPATIALIZATION_ENABLED_INDEX = 8, 45 STREAM_USAGE_INDEX = 9, 46 FOLD_STATE_INDEX = 10, 47 LID_STATE_INDEX = 11, 48 MAX_PARAM_INDEX, 49 }; 50 51 struct AudioEffectProcInfo { 52 bool headTrackingEnabled; 53 bool btOffloadEnabled; 54 }; 55 56 class AudioEffectChain { 57 public: 58 #ifdef SENSOR_ENABLE 59 AudioEffectChain(std::string scene, std::shared_ptr<HeadTracker> headTracker); 60 #else 61 AudioEffectChain(std::string scene); 62 #endif 63 ~AudioEffectChain(); 64 void SetEffectMode(const std::string &mode); 65 void SetExtraSceneType(const std::string &extraSceneType); 66 void SetFoldState(const std::string &foldState); 67 void SetLidState(const std::string &lidState); 68 void SetEffectCurrSceneType(AudioEffectScene currSceneType); 69 void SetSpatializationSceneType(AudioSpatializationSceneType spatializationSceneType); 70 void SetSpatializationEnabled(bool enabled); 71 void SetSpatializationEnabledForFading(bool enabled); 72 void AddEffectHandle(AudioEffectHandle effectHandle, AudioEffectLibrary *libHandle, AudioEffectScene currSceneType, 73 const std::string &effectName, const std::string &property); 74 void ApplyEffectChain(float *bufIn, float *bufOut, uint32_t frameLen, AudioEffectProcInfo procInfo); 75 void UpdateBufferConfig(uint32_t &channels, uint64_t &channelLayout); 76 bool IsEmptyEffectHandles(); 77 int32_t UpdateMultichannelIoBufferConfig(const uint32_t &channels, const uint64_t &channelLayout); 78 void StoreOldEffectChainInfo(std::string &sceneMode, AudioEffectConfig &ioBufferConfig); 79 void InitEffectChain(); 80 void SetHeadTrackingDisabled(); 81 uint32_t GetLatency(); 82 int32_t UpdateEffectParam(); 83 void ResetIoBufferConfig(); 84 void SetFinalVolume(const float volume); 85 float GetFinalVolume(); 86 void SetCurrVolume(const float volume); 87 float GetCurrVolume(); 88 void SetFinalVolumeState(const bool state); 89 bool GetFinalVolumeState(); 90 void SetSpatialDeviceType(AudioSpatialDeviceType spatialDeviceType); 91 int32_t SetEffectProperty(const std::string &effect, const std::string &property); 92 void SetStreamUsage(const int32_t streamUsage); 93 void GetInputChannelInfo(uint32_t &channels, uint64_t &channelLayout); 94 int32_t updatePrimaryChannel(); 95 bool CheckChannelLayoutByReplyInfo(AudioEffectTransInfo info); 96 void SetCurrChannelNoCheck(const uint32_t channel); 97 void SetCurrChannelLayoutNoCheck(const uint64_t channelLayout); 98 void updateDumpName(); 99 private: 100 AudioEffectConfig GetIoBufferConfig(); 101 void ReleaseEffectChain(); 102 int32_t SetEffectParamToHandle(AudioEffectHandle handle, int32_t &replyData); 103 void DumpEffectProcessData(std::string fileName, void *buffer, size_t len); 104 int32_t UpdateMultichannelIoBufferConfigInner(); 105 int32_t UpdateEffectParamInner(); 106 void CrossFadeProcess(float *bufOut, uint32_t frameLen); 107 108 std::mutex reloadMutex_; 109 std::string sceneType_ = ""; 110 std::string effectMode_ = ""; 111 uint32_t latency_ = 0; 112 uint32_t extraEffectChainType_ = 0; 113 uint32_t foldState_ = 0; 114 uint32_t lidState_ = 0; 115 StreamUsage streamUsage_ = STREAM_USAGE_INVALID; 116 AudioEffectScene currSceneType_ = SCENE_MUSIC; 117 std::vector<AudioEffectHandle> standByEffectHandles_; 118 std::vector<std::string> effectNames_; 119 std::vector<AudioEffectLibrary *> libHandles_; 120 AudioEffectConfig ioBufferConfig_ = {}; 121 AudioEffectConfig preIoBufferConfig_ = {}; 122 AudioBuffer audioBufIn_ = {}; 123 AudioBuffer audioBufOut_ = {}; 124 FILE *dumpFileInput_ = nullptr; 125 FILE *dumpFileOutput_ = nullptr; 126 float finalVolume_ = 1.0f; 127 float currVolume_ = 0.0f; 128 bool sendFinalVolumeState_ = false; 129 AudioSpatialDeviceType spatialDeviceType_{ EARPHONE_TYPE_OTHERS }; 130 AudioSpatializationSceneType spatializationSceneType_ = SPATIALIZATION_SCENE_TYPE_MUSIC; 131 bool spatializationEnabled_ = false; 132 std::string dumpNameIn_ = ""; 133 std::string dumpNameOut_ = ""; 134 bool spatializationEnabledFading_ = false; 135 int32_t fadingCounts_ = 0; 136 uint32_t currChannelNoCheck_ = STEREO; 137 uint64_t currchannelLayoutNoCheck_ = CH_LAYOUT_STEREO; 138 139 #ifdef SENSOR_ENABLE 140 std::shared_ptr<HeadTracker> headTracker_; 141 #endif 142 }; 143 144 } // namespace AudioStandard 145 } // namespace OHOS 146 #endif // AUDIO_EFFECT_CHAIN_H