package com.paulingalls.realtimeaudio

import kotlin.math.min

abstract class BaseVisualization : AudioVisualization {
    protected var samples: FloatArray = floatArrayOf()

    override fun updateData(data: FloatArray) {
        this.samples = data
    }

    override fun getSampleChunksFromAudio(
        buffer: FloatArray,
        channelCount: Int,
        chunkSize: Int
    ): ArrayList<FloatArray> {
        val frameLength = buffer.size / channelCount
        val pieceCount = frameLength / chunkSize
        val samplePieces = ArrayList<FloatArray>()

        for (pieceIndex in 0 until pieceCount) {
            val pieceSamples = mutableListOf<Float>()

            val startIndex = pieceIndex * chunkSize
            val endIndex = min(startIndex + chunkSize, frameLength)

            for (i in startIndex until endIndex) {
                var sample = 0f
                for (channel in 0 until channelCount) {
                    sample += buffer[i * channelCount + channel]
                }
                sample /= channelCount.toFloat()
                pieceSamples.add(sample)
            }
            samplePieces.add(pieceSamples.toFloatArray())
        }
        return samplePieces
    }

}