com.simiacryptus.util.audio.AudioRecorder.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of joe-penai Show documentation

A Java client for OpenAI's API

There is a newer version: 1.0.33

@file:Suppress("MemberVisibilityCanBePrivate")

package com.simiacryptus.util.audio

import org.apache.commons.io.input.buffer.CircularByteBuffer
import org.slf4j.LoggerFactory
import java.util.*
import javax.sound.sampled.AudioFormat
import javax.sound.sampled.AudioSystem
import javax.sound.sampled.TargetDataLine

class AudioRecorder(
    val audioBuffer: Deque,
    val secondsPerPacket: Double,
    val continueFn: () -> Boolean,
) {
    val packetLength = (audioFormat.frameRate * audioFormat.frameSize * secondsPerPacket).toInt()

    fun run() {
        val targetDataLine = openMic()
        try {
            log.info("Audio recording started")
            val buffer = ByteArray(packetLength)
            val circularBuffer = CircularByteBuffer(packetLength * 2)
            while (continueFn()) {
                try {
                    var bytesRead = 0
                    val endTime = (System.currentTimeMillis() + secondsPerPacket * 1000).toLong()
                    while (bytesRead != -1 && System.currentTimeMillis() < endTime) {
                        bytesRead = targetDataLine.read(buffer, 0, buffer.size)
                        circularBuffer.add(buffer, 0, bytesRead)
                        while (circularBuffer.currentNumberOfBytes >= packetLength) {
                            val array = ByteArray(packetLength)
                            circularBuffer.read(array, 0, packetLength)
                            audioBuffer.add(array)
                        }
                    }
                } catch (e: Exception) {
                    e.printStackTrace()
                }
            }
            log.info("Audio recording stopped")
        } finally {
            targetDataLine.close()
        }
    }

    companion object {
        val log = LoggerFactory.getLogger(AudioRecorder::class.java)

        val audioFormat = AudioFormat(16000f, 16, 1, true, false)

        fun openMic(): TargetDataLine {
            val targetDataLine = AudioSystem.getTargetDataLine(audioFormat)
            targetDataLine.open(audioFormat)
            targetDataLine.start()
            return targetDataLine
        }

    }

}