This commit is contained in:
Benoit Marty 2021-07-13 18:16:30 +02:00
parent c938a30dd9
commit df795d1881
1 changed files with 27 additions and 21 deletions

View File

@ -16,6 +16,7 @@
package org.matrix.android.sdk.internal.session.room.send
import timber.log.Timber
import javax.inject.Inject
import kotlin.math.abs
import kotlin.math.ceil
@ -38,43 +39,48 @@ internal class WaveFormSanitizer @Inject constructor() {
}
// Limit the number of items
val result = mutableListOf<Int>()
if (waveForm.size < MIN_NUMBER_OF_VALUES) {
val sizeInRangeList = mutableListOf<Int>()
when {
waveForm.size < MIN_NUMBER_OF_VALUES -> {
// Repeat the same value to have at least 30 items
val repeatTimes = ceil(MIN_NUMBER_OF_VALUES / waveForm.size.toDouble()).toInt()
waveForm.map { value ->
repeat(repeatTimes) {
result.add(value)
sizeInRangeList.add(value)
}
}
} else if (waveForm.size > MAX_NUMBER_OF_VALUES) {
}
waveForm.size > MAX_NUMBER_OF_VALUES -> {
val keepOneOf = ceil(waveForm.size.toDouble() / MAX_NUMBER_OF_VALUES).toInt()
waveForm.mapIndexed { idx, value ->
if (idx % keepOneOf == 0) {
result.add(value)
sizeInRangeList.add(value)
}
}
} else {
result.addAll(waveForm)
}
else -> {
sizeInRangeList.addAll(waveForm)
}
}
// OK, ensure all items are positive
val limited = result.map {
val positiveList = sizeInRangeList.map {
abs(it)
}
// Ensure max is not above MAX_VALUE
val max = limited.maxOrNull() ?: MAX_VALUE
val max = positiveList.maxOrNull() ?: MAX_VALUE
val final = if (max > MAX_VALUE) {
// Reduce the range
limited.map {
val finalList = if (max > MAX_VALUE) {
// Reduce the values
positiveList.map {
it * MAX_VALUE / max
}
} else {
limited
positiveList
}
return final
Timber.d("Sanitize from ${waveForm.size} items to ${finalList.size} items")
return finalList
}
}