blob: b044e161ece82763288333a52e6165b14accad1c [file] [log] [blame]
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +05301/*
2 * Copyright (C) 2011 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#include <assert.h>
17#include <inttypes.h>
Saketh Sathuvallicb398ab2019-02-16 13:12:59 +053018#include <iterator>
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +053019#include <math.h>
20#include <stdlib.h>
21#include <string.h>
22#include <vector>
23
24#include <audio_utils/channels.h>
25#include <audio_utils/primitives.h>
26#include <log/log.h>
Saketh Sathuvalli7b66e3b2019-02-08 15:18:54 +053027#include <system/audio.h>
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +053028
29#include "EffectBundle.h"
30#include "LVM_Private.h"
31
32#ifdef VERY_VERY_VERBOSE_LOGGING
33#define ALOGVV ALOGV
34#else
35#define ALOGVV(a...) \
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +053036 do { \
37 } while (false)
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +053038#endif
39
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +053040#define CHECK_ARG(cond) \
41 { \
42 if (!(cond)) { \
43 ALOGE("\tLVM_ERROR : Invalid argument: " #cond); \
44 return -EINVAL; \
45 } \
46 }
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +053047
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +053048#define LVM_ERROR_CHECK(LvmStatus, callingFunc, calledFunc) \
49 { \
50 if ((LvmStatus) == LVM_NULLADDRESS) { \
51 ALOGE("\tLVM_ERROR : Parameter error - " \
52 "null pointer returned by %s in %s\n\n\n\n", \
53 callingFunc, calledFunc); \
54 } \
55 if ((LvmStatus) == LVM_ALIGNMENTERROR) { \
56 ALOGE("\tLVM_ERROR : Parameter error - " \
57 "bad alignment returned by %s in %s\n\n\n\n", \
58 callingFunc, calledFunc); \
59 } \
60 if ((LvmStatus) == LVM_INVALIDNUMSAMPLES) { \
61 ALOGE("\tLVM_ERROR : Parameter error - " \
62 "bad number of samples returned by %s in %s\n\n\n\n", \
63 callingFunc, calledFunc); \
64 } \
65 if ((LvmStatus) == LVM_OUTOFRANGE) { \
66 ALOGE("\tLVM_ERROR : Parameter error - " \
67 "out of range returned by %s in %s\n", \
68 callingFunc, calledFunc); \
69 } \
70 }
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +053071
72struct lvmConfigParams_t {
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +053073 int samplingFreq = 44100;
74 int nrChannels = 2;
75 int chMask = AUDIO_CHANNEL_OUT_STEREO;
76 int vcBal = 0;
77 int fChannels = 2;
78 bool monoMode = false;
79 int bassEffectLevel = 0;
80 int eqPresetLevel = 0;
81 int frameLength = 256;
82 LVM_BE_Mode_en bassEnable = LVM_BE_OFF;
83 LVM_TE_Mode_en trebleEnable = LVM_TE_OFF;
84 LVM_EQNB_Mode_en eqEnable = LVM_EQNB_OFF;
85 LVM_Mode_en csEnable = LVM_MODE_OFF;
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +053086};
87
Saketh Sathuvalli7b66e3b2019-02-08 15:18:54 +053088constexpr audio_channel_mask_t lvmConfigChMask[] = {
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +053089 AUDIO_CHANNEL_OUT_MONO,
90 AUDIO_CHANNEL_OUT_STEREO,
91 AUDIO_CHANNEL_OUT_2POINT1,
92 AUDIO_CHANNEL_OUT_2POINT0POINT2,
93 AUDIO_CHANNEL_OUT_QUAD,
94 AUDIO_CHANNEL_OUT_QUAD_BACK,
95 AUDIO_CHANNEL_OUT_QUAD_SIDE,
96 AUDIO_CHANNEL_OUT_SURROUND,
Mikhail Naganov48e23452020-10-07 20:57:51 +000097 AUDIO_CHANNEL_INDEX_MASK_4,
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +053098 AUDIO_CHANNEL_OUT_2POINT1POINT2,
99 AUDIO_CHANNEL_OUT_3POINT0POINT2,
100 AUDIO_CHANNEL_OUT_PENTA,
Mikhail Naganov48e23452020-10-07 20:57:51 +0000101 AUDIO_CHANNEL_INDEX_MASK_5,
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530102 AUDIO_CHANNEL_OUT_3POINT1POINT2,
103 AUDIO_CHANNEL_OUT_5POINT1,
104 AUDIO_CHANNEL_OUT_5POINT1_BACK,
105 AUDIO_CHANNEL_OUT_5POINT1_SIDE,
Mikhail Naganov48e23452020-10-07 20:57:51 +0000106 AUDIO_CHANNEL_INDEX_MASK_6,
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530107 AUDIO_CHANNEL_OUT_6POINT1,
Mikhail Naganov48e23452020-10-07 20:57:51 +0000108 AUDIO_CHANNEL_INDEX_MASK_7,
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530109 AUDIO_CHANNEL_OUT_5POINT1POINT2,
110 AUDIO_CHANNEL_OUT_7POINT1,
Mikhail Naganov48e23452020-10-07 20:57:51 +0000111 AUDIO_CHANNEL_INDEX_MASK_8,
Harish Mahendrakara458f102021-01-08 10:25:47 -0800112 AUDIO_CHANNEL_INDEX_MASK_9,
113 AUDIO_CHANNEL_INDEX_MASK_10,
114 AUDIO_CHANNEL_INDEX_MASK_11,
115 AUDIO_CHANNEL_INDEX_MASK_12,
116 AUDIO_CHANNEL_INDEX_MASK_13,
117 AUDIO_CHANNEL_INDEX_MASK_14,
118 AUDIO_CHANNEL_INDEX_MASK_15,
119 AUDIO_CHANNEL_INDEX_MASK_16,
120 AUDIO_CHANNEL_INDEX_MASK_17,
121 AUDIO_CHANNEL_INDEX_MASK_18,
122 AUDIO_CHANNEL_INDEX_MASK_19,
123 AUDIO_CHANNEL_INDEX_MASK_20,
124 AUDIO_CHANNEL_INDEX_MASK_21,
125 AUDIO_CHANNEL_INDEX_MASK_22,
126 AUDIO_CHANNEL_INDEX_MASK_23,
127 AUDIO_CHANNEL_INDEX_MASK_24,
Saketh Sathuvalli7b66e3b2019-02-08 15:18:54 +0530128};
129
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530130void printUsage() {
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530131 printf("\nUsage: ");
132 printf("\n <executable> -i:<input_file> -o:<out_file> [options]\n");
133 printf("\nwhere, \n <inputfile> is the input file name");
134 printf("\n on which LVM effects are applied");
135 printf("\n <outputfile> processed output file");
136 printf("\n and options are mentioned below");
137 printf("\n");
138 printf("\n -help (or) -h");
139 printf("\n Prints this usage information");
140 printf("\n");
141 printf("\n -chMask:<channel_mask>\n");
142 printf("\n 0 - AUDIO_CHANNEL_OUT_MONO");
143 printf("\n 1 - AUDIO_CHANNEL_OUT_STEREO");
144 printf("\n 2 - AUDIO_CHANNEL_OUT_2POINT1");
145 printf("\n 3 - AUDIO_CHANNEL_OUT_2POINT0POINT2");
146 printf("\n 4 - AUDIO_CHANNEL_OUT_QUAD");
147 printf("\n 5 - AUDIO_CHANNEL_OUT_QUAD_BACK");
148 printf("\n 6 - AUDIO_CHANNEL_OUT_QUAD_SIDE");
149 printf("\n 7 - AUDIO_CHANNEL_OUT_SURROUND");
150 printf("\n 8 - canonical channel index mask for 4 ch: (1 << 4) - 1");
151 printf("\n 9 - AUDIO_CHANNEL_OUT_2POINT1POINT2");
152 printf("\n 10 - AUDIO_CHANNEL_OUT_3POINT0POINT2");
153 printf("\n 11 - AUDIO_CHANNEL_OUT_PENTA");
154 printf("\n 12 - canonical channel index mask for 5 ch: (1 << 5) - 1");
155 printf("\n 13 - AUDIO_CHANNEL_OUT_3POINT1POINT2");
156 printf("\n 14 - AUDIO_CHANNEL_OUT_5POINT1");
157 printf("\n 15 - AUDIO_CHANNEL_OUT_5POINT1_BACK");
158 printf("\n 16 - AUDIO_CHANNEL_OUT_5POINT1_SIDE");
159 printf("\n 17 - canonical channel index mask for 6 ch: (1 << 6) - 1");
160 printf("\n 18 - AUDIO_CHANNEL_OUT_6POINT1");
161 printf("\n 19 - canonical channel index mask for 7 ch: (1 << 7) - 1");
162 printf("\n 20 - AUDIO_CHANNEL_OUT_5POINT1POINT2");
163 printf("\n 21 - AUDIO_CHANNEL_OUT_7POINT1");
164 printf("\n 22 - canonical channel index mask for 8 ch: (1 << 8) - 1");
165 printf("\n default 0");
166 printf("\n -vcBal:<Left Right Balance control in dB [-96 to 96 dB]>");
167 printf("\n -ve values reduce Right channel while +ve value reduces Left channel");
168 printf("\n default 0");
169 printf("\n -fch:<file_channels> (1 through 8)\n\n");
170 printf("\n -M");
171 printf("\n Mono mode (force all input audio channels to be identical)");
172 printf("\n -basslvl:<effect_level>");
173 printf("\n A value that ranges between %d - %d default 0", LVM_BE_MIN_EFFECTLEVEL,
174 LVM_BE_MAX_EFFECTLEVEL);
175 printf("\n");
176 printf("\n -eqPreset:<preset Value>");
177 const size_t numPresetLvls = std::size(gEqualizerPresets);
178 for (size_t i = 0; i < numPresetLvls; ++i) {
179 printf("\n %zu - %s", i, gEqualizerPresets[i].name);
180 }
181 printf("\n default - 0");
182 printf("\n -bE ");
183 printf("\n Enable Dynamic Bass Enhancement");
184 printf("\n");
185 printf("\n -tE ");
186 printf("\n Enable Treble Boost");
187 printf("\n");
188 printf("\n -csE ");
189 printf("\n Enable Concert Surround");
190 printf("\n");
191 printf("\n -eqE ");
192 printf("\n Enable Equalizer");
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530193}
194
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530195//----------------------------------------------------------------------------
196// LvmBundle_init()
197//----------------------------------------------------------------------------
198// Purpose: Initialize engine with default configuration, creates instance
199// with all effects disabled.
200//
201// Inputs:
202// pContext: effect engine context
203//
204// Outputs:
205//
206//----------------------------------------------------------------------------
207
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530208int LvmBundle_init(struct EffectContext* pContext, LVM_ControlParams_t* params) {
209 ALOGV("\tLvmBundle_init start");
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530210
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530211 pContext->config.inputCfg.accessMode = EFFECT_BUFFER_ACCESS_READ;
212 pContext->config.inputCfg.channels = AUDIO_CHANNEL_OUT_STEREO;
213 pContext->config.inputCfg.format = EFFECT_BUFFER_FORMAT;
214 pContext->config.inputCfg.samplingRate = 44100;
215 pContext->config.inputCfg.bufferProvider.getBuffer = NULL;
216 pContext->config.inputCfg.bufferProvider.releaseBuffer = NULL;
217 pContext->config.inputCfg.bufferProvider.cookie = NULL;
218 pContext->config.inputCfg.mask = EFFECT_CONFIG_ALL;
219 pContext->config.outputCfg.accessMode = EFFECT_BUFFER_ACCESS_ACCUMULATE;
220 pContext->config.outputCfg.channels = AUDIO_CHANNEL_OUT_STEREO;
221 pContext->config.outputCfg.format = EFFECT_BUFFER_FORMAT;
222 pContext->config.outputCfg.samplingRate = 44100;
223 pContext->config.outputCfg.bufferProvider.getBuffer = NULL;
224 pContext->config.outputCfg.bufferProvider.releaseBuffer = NULL;
225 pContext->config.outputCfg.bufferProvider.cookie = NULL;
226 pContext->config.outputCfg.mask = EFFECT_CONFIG_ALL;
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530227
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530228 if (pContext->pBundledContext->hInstance != NULL) {
229 ALOGV("\tLvmBundle_init pContext->pBassBoost != NULL "
230 "-> Calling pContext->pBassBoost->free()");
231 LVM_DelInstanceHandle(&pContext->pBundledContext->hInstance);
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530232
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530233 ALOGV("\tLvmBundle_init pContext->pBassBoost != NULL "
234 "-> Called pContext->pBassBoost->free()");
235 }
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530236
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530237 LVM_ReturnStatus_en LvmStatus = LVM_SUCCESS; /* Function call status */
238 LVM_InstParams_t InstParams; /* Instance parameters */
239 LVM_EQNB_BandDef_t BandDefs[MAX_NUM_BANDS]; /* Equaliser band definitions */
240 LVM_HeadroomParams_t HeadroomParams; /* Headroom parameters */
241 LVM_HeadroomBandDef_t HeadroomBandDef[LVM_HEADROOM_MAX_NBANDS];
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530242
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530243 /* Set the capabilities */
244 InstParams.BufferMode = LVM_UNMANAGED_BUFFERS;
245 InstParams.MaxBlockSize = MAX_CALL_SIZE;
246 InstParams.EQNB_NumBands = MAX_NUM_BANDS;
247 InstParams.PSA_Included = LVM_PSA_ON;
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530248
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530249 LvmStatus = LVM_GetInstanceHandle(&pContext->pBundledContext->hInstance, &InstParams);
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530250
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530251 LVM_ERROR_CHECK(LvmStatus, "LVM_GetInstanceHandle", "LvmBundle_init");
252 if (LvmStatus != LVM_SUCCESS) return -EINVAL;
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530253
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530254 ALOGV("\tLvmBundle_init CreateInstance Successfully called "
255 "LVM_GetInstanceHandle\n");
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530256
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530257 /* Set the initial process parameters */
258 /* General parameters */
259 params->OperatingMode = LVM_MODE_ON;
260 params->SampleRate = LVM_FS_44100;
261 params->SourceFormat = LVM_STEREO;
262 params->ChMask = AUDIO_CHANNEL_OUT_STEREO;
263 params->SpeakerType = LVM_HEADPHONES;
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530264
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530265 pContext->pBundledContext->SampleRate = LVM_FS_44100;
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530266
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530267 /* Concert Sound parameters */
268 params->VirtualizerOperatingMode = LVM_MODE_OFF;
269 params->VirtualizerType = LVM_CONCERTSOUND;
270 params->VirtualizerReverbLevel = 100;
271 params->CS_EffectLevel = LVM_CS_EFFECT_NONE;
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530272
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530273 /* N-Band Equaliser parameters */
274 params->EQNB_OperatingMode = LVM_EQNB_ON;
275 params->EQNB_NBands = FIVEBAND_NUMBANDS;
276 params->pEQNB_BandDefinition = &BandDefs[0];
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530277
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530278 for (int i = 0; i < FIVEBAND_NUMBANDS; i++) {
279 BandDefs[i].Frequency = EQNB_5BandPresetsFrequencies[i];
280 BandDefs[i].QFactor = EQNB_5BandPresetsQFactors[i];
281 BandDefs[i].Gain = EQNB_5BandSoftPresets[i];
282 }
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530283
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530284 /* Volume Control parameters */
285 params->VC_EffectLevel = 0;
286 params->VC_Balance = 0;
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530287
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530288 /* Treble Enhancement parameters */
289 params->TE_OperatingMode = LVM_TE_OFF;
290 params->TE_EffectLevel = 0;
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530291
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530292 /* PSA Control parameters */
293 params->PSA_Enable = LVM_PSA_OFF;
294 params->PSA_PeakDecayRate = (LVM_PSA_DecaySpeed_en)0;
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530295
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530296 /* Bass Enhancement parameters */
297 params->BE_OperatingMode = LVM_BE_ON;
298 params->BE_EffectLevel = 0;
299 params->BE_CentreFreq = LVM_BE_CENTRE_90Hz;
300 params->BE_HPF = LVM_BE_HPF_ON;
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530301
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530302 /* PSA Control parameters */
303 params->PSA_Enable = LVM_PSA_OFF;
304 params->PSA_PeakDecayRate = LVM_PSA_SPEED_MEDIUM;
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530305
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530306 /* TE Control parameters */
307 params->TE_OperatingMode = LVM_TE_OFF;
308 params->TE_EffectLevel = 0;
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530309
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530310 /* Activate the initial settings */
311 LvmStatus = LVM_SetControlParameters(pContext->pBundledContext->hInstance, params);
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530312
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530313 LVM_ERROR_CHECK(LvmStatus, "LVM_SetControlParameters", "LvmBundle_init");
314 if (LvmStatus != LVM_SUCCESS) return -EINVAL;
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530315
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530316 ALOGV("\tLvmBundle_init CreateInstance Successfully called "
317 "LVM_SetControlParameters\n");
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530318
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530319 /* Set the headroom parameters */
320 HeadroomBandDef[0].Limit_Low = 20;
321 HeadroomBandDef[0].Limit_High = 4999;
322 HeadroomBandDef[0].Headroom_Offset = 0;
323 HeadroomBandDef[1].Limit_Low = 5000;
324 HeadroomBandDef[1].Limit_High = 24000;
325 HeadroomBandDef[1].Headroom_Offset = 0;
326 HeadroomParams.pHeadroomDefinition = &HeadroomBandDef[0];
327 HeadroomParams.Headroom_OperatingMode = LVM_HEADROOM_ON;
328 HeadroomParams.NHeadroomBands = 2;
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530329
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530330 LvmStatus = LVM_SetHeadroomParams(pContext->pBundledContext->hInstance, &HeadroomParams);
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530331
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530332 LVM_ERROR_CHECK(LvmStatus, "LVM_SetHeadroomParams", "LvmBundle_init");
333 if (LvmStatus != LVM_SUCCESS) return -EINVAL;
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530334
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530335 ALOGV("\tLvmBundle_init CreateInstance Successfully called "
336 "LVM_SetHeadroomParams\n");
337 ALOGV("\tLvmBundle_init End");
338 return 0;
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530339} /* end LvmBundle_init */
340
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530341int lvmCreate(struct EffectContext* pContext, lvmConfigParams_t* plvmConfigParams,
342 LVM_ControlParams_t* params) {
343 int ret = 0;
344 pContext->pBundledContext = NULL;
345 pContext->pBundledContext = (BundledEffectContext*)malloc(sizeof(struct BundledEffectContext));
346 if (NULL == pContext->pBundledContext) {
347 return -EINVAL;
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530348 }
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530349
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530350 pContext->pBundledContext->SessionNo = 0;
351 pContext->pBundledContext->SessionId = 0;
352 pContext->pBundledContext->hInstance = NULL;
353 pContext->pBundledContext->bVolumeEnabled = LVM_FALSE;
354 pContext->pBundledContext->bEqualizerEnabled = LVM_FALSE;
355 pContext->pBundledContext->bBassEnabled = LVM_FALSE;
356 pContext->pBundledContext->bBassTempDisabled = LVM_FALSE;
357 pContext->pBundledContext->bVirtualizerEnabled = LVM_FALSE;
358 pContext->pBundledContext->bVirtualizerTempDisabled = LVM_FALSE;
359 pContext->pBundledContext->nOutputDevice = AUDIO_DEVICE_NONE;
360 pContext->pBundledContext->nVirtualizerForcedDevice = AUDIO_DEVICE_NONE;
361 pContext->pBundledContext->NumberEffectsEnabled = 0;
362 pContext->pBundledContext->NumberEffectsCalled = 0;
363 pContext->pBundledContext->firstVolume = LVM_TRUE;
364 pContext->pBundledContext->volume = 0;
365
366 /* Saved strength is used to return the exact strength that was used in the
367 * set to the get
368 * because we map the original strength range of 0:1000 to 1:15, and this will
369 * avoid
370 * quantisation like effect when returning
371 */
372 pContext->pBundledContext->BassStrengthSaved = 0;
373 pContext->pBundledContext->VirtStrengthSaved = 0;
374 pContext->pBundledContext->CurPreset = PRESET_CUSTOM;
375 pContext->pBundledContext->levelSaved = 0;
376 pContext->pBundledContext->bMuteEnabled = LVM_FALSE;
377 pContext->pBundledContext->bStereoPositionEnabled = LVM_FALSE;
378 pContext->pBundledContext->positionSaved = 0;
379 pContext->pBundledContext->workBuffer = NULL;
380 pContext->pBundledContext->frameCount = -1;
381 pContext->pBundledContext->SamplesToExitCountVirt = 0;
382 pContext->pBundledContext->SamplesToExitCountBb = 0;
383 pContext->pBundledContext->SamplesToExitCountEq = 0;
384 for (int i = 0; i < FIVEBAND_NUMBANDS; i++) {
385 pContext->pBundledContext->bandGaindB[i] = EQNB_5BandSoftPresets[i];
386 }
387 pContext->config.inputCfg.channels = plvmConfigParams->nrChannels;
388 ALOGV("\tEffectCreate - Calling LvmBundle_init");
389 ret = LvmBundle_init(pContext, params);
390
391 if (ret < 0) {
392 ALOGE("\tLVM_ERROR : lvmCreate() Bundle init failed");
393 return ret;
394 }
395 return 0;
396}
397
398int lvmControl(struct EffectContext* pContext, lvmConfigParams_t* plvmConfigParams,
399 LVM_ControlParams_t* params) {
400 LVM_ReturnStatus_en LvmStatus = LVM_SUCCESS; /* Function call status */
401
402 /* Set the initial process parameters */
403 /* General parameters */
404 params->OperatingMode = LVM_MODE_ON;
405 params->SpeakerType = LVM_HEADPHONES;
406
407 params->ChMask = plvmConfigParams->chMask;
408 params->NrChannels = plvmConfigParams->nrChannels;
409 if (params->NrChannels == 1) {
410 params->SourceFormat = LVM_MONO;
411 } else if (params->NrChannels == 2) {
412 params->SourceFormat = LVM_STEREO;
Harish Mahendrakara458f102021-01-08 10:25:47 -0800413 } else if (params->NrChannels > FCC_2 && params->NrChannels <= FCC_24) {
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530414 params->SourceFormat = LVM_MULTICHANNEL;
415 } else {
416 return -EINVAL;
417 }
Harish Mahendrakarf0fb4962021-02-18 14:21:45 -0800418 params->SampleRate = lvmFsForSampleRate(plvmConfigParams->samplingFreq);
419 if (params->SampleRate == LVM_FS_INVALID) {
420 ALOGE("lvmControl invalid sampling rate %d", plvmConfigParams->samplingFreq);
421 return -EINVAL;
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530422 }
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530423
424 /* Concert Sound parameters */
425 params->VirtualizerOperatingMode = plvmConfigParams->csEnable;
426 params->VirtualizerType = LVM_CONCERTSOUND;
427 params->VirtualizerReverbLevel = 100;
428 params->CS_EffectLevel = LVM_CS_EFFECT_NONE;
429
430 /* N-Band Equaliser parameters */
431 const int eqPresetLevel = plvmConfigParams->eqPresetLevel;
432 LVM_EQNB_BandDef_t BandDefs[MAX_NUM_BANDS]; /* Equaliser band definitions */
433 for (int i = 0; i < FIVEBAND_NUMBANDS; i++) {
434 BandDefs[i].Frequency = EQNB_5BandPresetsFrequencies[i];
435 BandDefs[i].QFactor = EQNB_5BandPresetsQFactors[i];
436 BandDefs[i].Gain = EQNB_5BandSoftPresets[(FIVEBAND_NUMBANDS * eqPresetLevel) + i];
437 }
438 params->EQNB_OperatingMode = plvmConfigParams->eqEnable;
439 // Caution: raw pointer to stack data, stored in instance by LVM_SetControlParameters.
440 params->pEQNB_BandDefinition = &BandDefs[0];
441
442 /* Volume Control parameters */
443 params->VC_EffectLevel = 0;
444 params->VC_Balance = plvmConfigParams->vcBal;
445
446 /* Treble Enhancement parameters */
447 params->TE_OperatingMode = plvmConfigParams->trebleEnable;
448
449 /* PSA Control parameters */
450 params->PSA_Enable = LVM_PSA_ON;
451
452 /* Bass Enhancement parameters */
453 params->BE_OperatingMode = plvmConfigParams->bassEnable;
454
455 /* Activate the initial settings */
456 LvmStatus = LVM_SetControlParameters(pContext->pBundledContext->hInstance, params);
457
458 LVM_ERROR_CHECK(LvmStatus, "LVM_SetControlParameters", "LvmBundle_init");
459 if (LvmStatus != LVM_SUCCESS) return -EINVAL;
460
461 LvmStatus = LVM_ApplyNewSettings(pContext->pBundledContext->hInstance);
462
463 if (LvmStatus != LVM_SUCCESS) return -EINVAL;
464
465 return 0;
466}
467
468int lvmExecute(float* floatIn, float* floatOut, struct EffectContext* pContext,
469 lvmConfigParams_t* plvmConfigParams) {
470 const int frameLength = plvmConfigParams->frameLength;
471 return LVM_Process(pContext->pBundledContext->hInstance, /* Instance handle */
472 floatIn, /* Input buffer */
473 floatOut, /* Output buffer */
474 (LVM_UINT16)frameLength, /* Number of samples to read */
475 0); /* Audio Time */
476}
477
478int lvmMainProcess(EffectContext* pContext, LVM_ControlParams_t* pParams,
479 lvmConfigParams_t* plvmConfigParams, FILE* finp, FILE* fout) {
480 int errCode = lvmControl(pContext, plvmConfigParams, pParams);
481 if (errCode) {
482 ALOGE("Error: lvmControl returned with %d\n", errCode);
483 return errCode;
484 }
485
486 const int channelCount = plvmConfigParams->nrChannels;
487 const int frameLength = plvmConfigParams->frameLength;
488 const int frameSize = channelCount * sizeof(float); // processing size
489 const int ioChannelCount = plvmConfigParams->fChannels;
490 const int ioFrameSize = ioChannelCount * sizeof(short); // file load size
491 const int maxChannelCount = std::max(channelCount, ioChannelCount);
492 /*
493 * Mono input will be converted to 2 channels internally in the process call
494 * by copying the same data into the second channel.
495 * Hence when channelCount is 1, output buffer should be allocated for
496 * 2 channels. The memAllocChCount takes care of allocation of sufficient
497 * memory for the output buffer.
498 */
499 const int memAllocChCount = (channelCount == 1 ? 2 : channelCount);
500
501 std::vector<short> in(frameLength * maxChannelCount);
502 std::vector<short> out(frameLength * maxChannelCount);
503 std::vector<float> floatIn(frameLength * channelCount);
504 std::vector<float> floatOut(frameLength * memAllocChCount);
505
506 int frameCounter = 0;
507 while (fread(in.data(), ioFrameSize, frameLength, finp) == (size_t)frameLength) {
508 if (ioChannelCount != channelCount) {
509 adjust_channels(in.data(), ioChannelCount, in.data(), channelCount, sizeof(short),
510 frameLength * ioFrameSize);
511 }
512 memcpy_to_float_from_i16(floatIn.data(), in.data(), frameLength * channelCount);
513
514 // Mono mode will replicate the first channel to all other channels.
515 // This ensures all audio channels are identical. This is useful for testing
516 // Bass Boost, which extracts a mono signal for processing.
517 if (plvmConfigParams->monoMode && channelCount > 1) {
518 for (int i = 0; i < frameLength; ++i) {
519 auto* fp = &floatIn[i * channelCount];
520 std::fill(fp + 1, fp + channelCount, *fp); // replicate ch 0
521 }
522 }
523#ifndef BYPASS_EXEC
524 errCode = lvmExecute(floatIn.data(), floatOut.data(), pContext, plvmConfigParams);
525 if (errCode) {
526 printf("\nError: lvmExecute returned with %d\n", errCode);
527 return errCode;
528 }
529
530 (void)frameSize; // eliminate warning
531#else
532 memcpy(floatOut.data(), floatIn.data(), frameLength * frameSize);
533#endif
534 memcpy_to_i16_from_float(out.data(), floatOut.data(), frameLength * channelCount);
535 if (ioChannelCount != channelCount) {
536 adjust_channels(out.data(), channelCount, out.data(), ioChannelCount, sizeof(short),
537 frameLength * channelCount * sizeof(short));
538 }
539 (void)fwrite(out.data(), ioFrameSize, frameLength, fout);
540 frameCounter += frameLength;
541 }
542 printf("frameCounter: [%d]\n", frameCounter);
543 return 0;
544}
545
546int main(int argc, const char* argv[]) {
547 if (argc == 1) {
548 printUsage();
549 return -1;
550 }
551
552 lvmConfigParams_t lvmConfigParams{}; // default initialize
553 const char* infile = nullptr;
554 const char* outfile = nullptr;
555
556 for (int i = 1; i < argc; i++) {
557 printf("%s ", argv[i]);
558 if (!strncmp(argv[i], "-i:", 3)) {
559 infile = argv[i] + 3;
560 } else if (!strncmp(argv[i], "-o:", 3)) {
561 outfile = argv[i] + 3;
562 } else if (!strncmp(argv[i], "-fs:", 4)) {
563 const int samplingFreq = atoi(argv[i] + 4);
564 if (samplingFreq != 8000 && samplingFreq != 11025 && samplingFreq != 12000 &&
565 samplingFreq != 16000 && samplingFreq != 22050 && samplingFreq != 24000 &&
566 samplingFreq != 32000 && samplingFreq != 44100 && samplingFreq != 48000 &&
567 samplingFreq != 88200 && samplingFreq != 96000 && samplingFreq != 176400 &&
568 samplingFreq != 192000) {
569 printf("Error: Unsupported Sampling Frequency : %d\n", samplingFreq);
570 return -1;
571 }
572 lvmConfigParams.samplingFreq = samplingFreq;
573 } else if (!strncmp(argv[i], "-chMask:", 8)) {
574 const int chMaskConfigIdx = atoi(argv[i] + 8);
575 if (chMaskConfigIdx < 0 || (size_t)chMaskConfigIdx >= std::size(lvmConfigChMask)) {
576 ALOGE("\nError: Unsupported Channel Mask : %d\n", chMaskConfigIdx);
577 return -1;
578 }
579 const audio_channel_mask_t chMask = lvmConfigChMask[chMaskConfigIdx];
580 lvmConfigParams.chMask = chMask;
581 lvmConfigParams.nrChannels = audio_channel_count_from_out_mask(chMask);
582 } else if (!strncmp(argv[i], "-vcBal:", 7)) {
583 const int vcBalance = atoi(argv[i] + 7);
584 if (vcBalance > 96 || vcBalance < -96) {
585 ALOGE("\nError: Unsupported volume balance value: %d\n", vcBalance);
586 }
587 lvmConfigParams.vcBal = vcBalance;
588 } else if (!strncmp(argv[i], "-fch:", 5)) {
589 const int fChannels = atoi(argv[i] + 5);
590 if (fChannels > 8 || fChannels < 1) {
591 printf("Error: Unsupported number of file channels : %d\n", fChannels);
592 return -1;
593 }
594 lvmConfigParams.fChannels = fChannels;
595 } else if (!strcmp(argv[i], "-M")) {
596 lvmConfigParams.monoMode = true;
597 } else if (!strncmp(argv[i], "-basslvl:", 9)) {
598 const int bassEffectLevel = atoi(argv[i] + 9);
599 if (bassEffectLevel > LVM_BE_MAX_EFFECTLEVEL ||
600 bassEffectLevel < LVM_BE_MIN_EFFECTLEVEL) {
601 printf("Error: Unsupported Bass Effect Level : %d\n", bassEffectLevel);
602 printUsage();
603 return -1;
604 }
605 lvmConfigParams.bassEffectLevel = bassEffectLevel;
606 } else if (!strncmp(argv[i], "-eqPreset:", 10)) {
607 const int eqPresetLevel = atoi(argv[i] + 10);
608 const int numPresetLvls = std::size(gEqualizerPresets);
609 if (eqPresetLevel >= numPresetLvls || eqPresetLevel < 0) {
610 printf("Error: Unsupported Equalizer Preset : %d\n", eqPresetLevel);
611 printUsage();
612 return -1;
613 }
614 lvmConfigParams.eqPresetLevel = eqPresetLevel;
615 } else if (!strcmp(argv[i], "-bE")) {
616 lvmConfigParams.bassEnable = LVM_BE_ON;
617 } else if (!strcmp(argv[i], "-eqE")) {
618 lvmConfigParams.eqEnable = LVM_EQNB_ON;
619 } else if (!strcmp(argv[i], "-tE")) {
620 lvmConfigParams.trebleEnable = LVM_TE_ON;
621 } else if (!strcmp(argv[i], "-csE")) {
622 lvmConfigParams.csEnable = LVM_MODE_ON;
623 } else if (!strcmp(argv[i], "-h")) {
624 printUsage();
625 return 0;
Andy Hungb144b4b2018-12-20 14:15:49 -0800626 }
627 }
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530628
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530629 if (infile == nullptr || outfile == nullptr) {
630 printf("Error: missing input/output files\n");
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530631 printUsage();
632 return -1;
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530633 }
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530634
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530635 FILE* finp = fopen(infile, "rb");
636 if (finp == nullptr) {
637 printf("Cannot open input file %s", infile);
638 return -1;
639 }
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530640
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530641 FILE* fout = fopen(outfile, "wb");
642 if (fout == nullptr) {
643 printf("Cannot open output file %s", outfile);
644 fclose(finp);
645 return -1;
646 }
Saketh Sathuvalliab18bac2019-01-24 17:15:10 +0530647
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530648 EffectContext context;
649 LVM_ControlParams_t params;
650 int errCode = lvmCreate(&context, &lvmConfigParams, &params);
651 if (errCode == 0) {
652 errCode = lvmMainProcess(&context, &params, &lvmConfigParams, finp, fout);
653 if (errCode != 0) {
654 printf("Error: lvmMainProcess returned with the error: %d", errCode);
655 }
656 } else {
657 printf("Error: lvmCreate returned with the error: %d", errCode);
658 }
Saketh Sathuvalliab18bac2019-01-24 17:15:10 +0530659 fclose(finp);
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530660 fclose(fout);
661 /* Free the allocated buffers */
662 if (context.pBundledContext != nullptr) {
663 if (context.pBundledContext->hInstance != nullptr) {
664 LVM_DelInstanceHandle(&context.pBundledContext->hInstance);
665 }
666 free(context.pBundledContext);
Saketh Sathuvalliab18bac2019-01-24 17:15:10 +0530667 }
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530668
Harish Mahendrakar0bbbe072020-10-01 23:28:48 +0530669 if (errCode) {
670 return -1;
671 }
672 return 0;
Saketh Sathuvallib99e1bc2018-02-21 17:10:34 +0530673}