|
| 1 | +import Accelerate |
| 2 | +@preconcurrency import AVFoundation |
| 3 | +import ElevenLabs |
| 4 | + |
| 5 | +/// Tracks a normalized loudness level for one conversation audio stream. |
| 6 | +/// |
| 7 | +/// Pull-based: `didReceive` runs on the audio thread and only updates a lock- |
| 8 | +/// protected level, which the UI samples at its own rate. |
| 9 | +final class ConversationAudioLevelMonitor: ConversationAudioObserver, @unchecked Sendable { |
| 10 | + private let lock = NSLock() |
| 11 | + private var storedLevel: Float = 0 |
| 12 | + |
| 13 | + var level: Float { |
| 14 | + lock.withLock { storedLevel } |
| 15 | + } |
| 16 | + |
| 17 | + func didReceive(_ buffer: AVAudioPCMBuffer) { |
| 18 | + guard let level = Self.normalizedRMS(of: buffer) else { return } |
| 19 | + lock.withLock { |
| 20 | + // Fast attack, slow release: reacts instantly to speech, decays smoothly. |
| 21 | + storedLevel = max(level, storedLevel * 0.75) |
| 22 | + } |
| 23 | + } |
| 24 | + |
| 25 | + func reset() { |
| 26 | + lock.withLock { storedLevel = 0 } |
| 27 | + } |
| 28 | + |
| 29 | + /// RMS of the first channel, mapped from the top 60 dB of headroom onto 0...1. |
| 30 | + private static func normalizedRMS(of buffer: AVAudioPCMBuffer) -> Float? { |
| 31 | + let frames = Int(buffer.frameLength) |
| 32 | + guard frames > 0 else { return nil } |
| 33 | + let stride = buffer.format.isInterleaved ? vDSP_Stride(buffer.format.channelCount) : 1 |
| 34 | + |
| 35 | + var rms: Float = 0 |
| 36 | + if let channel = buffer.floatChannelData?[0] { |
| 37 | + vDSP_rmsqv(channel, stride, &rms, vDSP_Length(frames)) |
| 38 | + } else if let channel = buffer.int16ChannelData?[0] { |
| 39 | + var samples = [Float](repeating: 0, count: frames) |
| 40 | + vDSP_vflt16(channel, stride, &samples, 1, vDSP_Length(frames)) |
| 41 | + var scale = 1 / Float(Int16.max) |
| 42 | + vDSP_vsmul(samples, 1, &scale, &samples, 1, vDSP_Length(frames)) |
| 43 | + vDSP_rmsqv(samples, 1, &rms, vDSP_Length(frames)) |
| 44 | + } else { |
| 45 | + return nil |
| 46 | + } |
| 47 | + |
| 48 | + let decibels = 20 * log10(max(rms, 0.000_001)) |
| 49 | + return min(max((decibels + 60) / 60, 0), 1) |
| 50 | + } |
| 51 | +} |
0 commit comments