Implement onAudioChunkCaptured callback to get audio chunks during recording

Author: squti

android-wave-recorder/src/main/java/com/github/squti/androidwaverecorder/FileWriter.kt

@@ -29,27 +29,30 @@ import java.nio.ByteBuffer
 import java.nio.ByteOrder
 import java.util.LinkedList
 
-internal class FileWriter(private val outputStream: DataOutputStream) {
+internal class FileWriter(
+    private val outputStream: DataOutputStream,
+    private val onAudioChunkCaptured: ((ByteArray) -> Unit)?
+) {
     fun writeDataToStream(
-        lastSkippedData: LinkedList<ByteArray>,
-        data: ByteArray
+        lastSkippedData: LinkedList<ByteArray>, data: ByteArray
     ) {
         val totalSize = lastSkippedData.sumOf { it.size } + data.size
         val byteBuffer = ByteBuffer.allocate(totalSize).order(ByteOrder.LITTLE_ENDIAN)
 
         lastSkippedData.forEach { byteArray ->
             byteBuffer.put(byteArray)
         }
-
         byteBuffer.put(data)
+        lastSkippedData.clear()
 
         outputStream.write(byteBuffer.array())
-        lastSkippedData.clear()
+        onAudioChunkCaptured?.let {
+            it(byteBuffer.array())
+        }
     }
 
     fun writeDataToStream(
-        lastSkippedData: LinkedList<FloatArray>,
-        data: FloatArray
+        lastSkippedData: LinkedList<FloatArray>, data: FloatArray
     ) {
         val totalFloats = lastSkippedData.sumOf { it.size } + data.size
         val totalSize = totalFloats * 4
@@ -63,9 +66,13 @@ internal class FileWriter(private val outputStream: DataOutputStream) {
         data.forEach { floatValue ->
             byteBuffer.putFloat(floatValue)
         }
+        lastSkippedData.clear()
 
         outputStream.write(byteBuffer.array())
-        lastSkippedData.clear()
+
+        onAudioChunkCaptured?.let {
+            it(byteBuffer.array())
+        }
     }
 
 }

android-wave-recorder/src/main/java/com/github/squti/androidwaverecorder/WaveRecorder.kt

@@ -83,6 +83,12 @@ class WaveRecorder {
      */
     var onAmplitudeListener: ((Int) -> Unit)? = null
 
+    /**
+     * Register a callback to be invoked for each recorded chunk of audio data.
+     * Provides the captured chunk as a ByteArray.
+     */
+    var onAudioChunkCaptured: ((ByteArray) -> Unit)? = null
+
     /**
      * Register a callback to be invoked in recording state changes
      */
@@ -191,7 +197,7 @@ class WaveRecorder {
             FileOutputStream(file)
         }
         val dataOutputStream = DataOutputStream(outputStream)
-        val fileWriter = FileWriter(dataOutputStream)
+        val fileWriter = FileWriter(dataOutputStream, onAudioChunkCaptured)
 
         val bufferSizeToKeep =
             (waveConfig.sampleRate * channelCount(waveConfig.channels) * (bitPerSample(waveConfig.audioEncoding) / 8) * silenceDetectionConfig.bufferDurationInMillis / 1000).toInt()
@@ -247,7 +253,7 @@ class WaveRecorder {
             FileOutputStream(file)
         }
         val dataOutputStream = DataOutputStream(outputStream)
-        val fileWriter = FileWriter(dataOutputStream)
+        val fileWriter = FileWriter(dataOutputStream, onAudioChunkCaptured)
 
         val bufferSizeToKeep =
             (waveConfig.sampleRate * channelCount(waveConfig.channels) * silenceDetectionConfig.bufferDurationInMillis / 1000).toInt()