4 changed files with 575 additions and 53 deletions
@ -0,0 +1,243 @@ |
|||
package com.yunqiinnovation.ble_service |
|||
|
|||
import android.content.Context |
|||
import android.util.Log |
|||
import java.io.File |
|||
import java.io.FileOutputStream |
|||
import java.io.IOException |
|||
import java.io.RandomAccessFile |
|||
import java.nio.ByteBuffer |
|||
import java.nio.ByteOrder |
|||
import java.text.SimpleDateFormat |
|||
import java.util.Date |
|||
import java.util.Locale |
|||
import java.util.concurrent.LinkedBlockingQueue |
|||
import java.util.concurrent.atomic.AtomicBoolean |
|||
|
|||
/** |
|||
* WAV文件写入工具类 |
|||
* 用于将PCM音频数据保存为标准WAV格式文件 |
|||
*/ |
|||
class WavFileWriter(private val context: Context) { |
|||
|
|||
companion object { |
|||
private const val TAG = "WavFileWriter" |
|||
private const val WAV_HEADER_SIZE = 44 |
|||
} |
|||
|
|||
private var currentWavFile: File? = null |
|||
private var fos: FileOutputStream? = null |
|||
|
|||
// 音频参数 |
|||
private var sampleRate: Int = 16000 |
|||
private var channels: Int = 1 |
|||
private var bitsPerSample: Int = 16 |
|||
|
|||
// 异步写入相关 |
|||
private val writeQueue = LinkedBlockingQueue<ByteArray>() |
|||
private val isWriting = AtomicBoolean(false) |
|||
private var writeThread: Thread? = null |
|||
|
|||
// 数据统计 |
|||
private var totalDataSize = 0 |
|||
|
|||
/** |
|||
* 创建新的WAV文件 |
|||
* @param fileName 文件名前缀 |
|||
* @param sampleRate 采样率，默认16000Hz |
|||
* @param channels 声道数，默认1（单声道） |
|||
* @param bitsPerSample 位深度，默认16位 |
|||
*/ |
|||
fun createWavFile( |
|||
fileName: String, |
|||
sampleRate: Int = 16000, |
|||
channels: Int = 1, |
|||
bitsPerSample: Int = 16 |
|||
): Boolean { |
|||
try { |
|||
if (fos != null || currentWavFile != null) { |
|||
closeWavFile() |
|||
} |
|||
|
|||
this.sampleRate = sampleRate |
|||
this.channels = channels |
|||
this.bitsPerSample = bitsPerSample |
|||
|
|||
// 创建文件名 |
|||
val dateFormat = SimpleDateFormat("yyyyMMdd_HHmmss", Locale.getDefault()) |
|||
val timestamp = dateFormat.format(Date()) |
|||
val filePath = |
|||
File(context.getExternalFilesDir(null), "${fileName}_${timestamp}.wav").absolutePath |
|||
|
|||
currentWavFile = File(filePath) |
|||
currentWavFile?.createNewFile() |
|||
|
|||
fos = FileOutputStream(currentWavFile, false) |
|||
|
|||
// 写入WAV文件头（先写入占位符，稍后更新） |
|||
writeWavHeader(0) |
|||
|
|||
// 启动异步写入线程 |
|||
startWriteThread() |
|||
totalDataSize = 0 |
|||
writeQueue.clear() |
|||
Log.d(TAG, "创建WAV文件: ${currentWavFile?.absolutePath}") |
|||
Log.d( |
|||
TAG, |
|||
"音频参数 - 采样率: ${sampleRate}Hz, 声道: ${channels}, 位深: ${bitsPerSample}bit" |
|||
) |
|||
|
|||
return true |
|||
} catch (e: Exception) { |
|||
Log.e(TAG, "创建WAV文件失败: ${e.message}", e) |
|||
return false |
|||
} |
|||
} |
|||
|
|||
/** |
|||
* 写入PCM音频数据 |
|||
* @param pcmData PCM音频数据字节数组 |
|||
*/ |
|||
fun writePcmData(pcmData: ByteArray) { |
|||
if (fos == null || currentWavFile == null) { |
|||
Log.w(TAG, "WAV文件未创建，无法写入数据") |
|||
return |
|||
} |
|||
if (!isWriting.get()) { |
|||
Log.d(TAG, "写入线程停止运行，停止写入数据") |
|||
return |
|||
} |
|||
try { |
|||
// 将数据放入队列，由写入线程异步处理 |
|||
writeQueue.offer(pcmData.clone()) |
|||
totalDataSize += pcmData.size |
|||
|
|||
// Log.d(TAG, "添加PCM数据到写入队列: ${pcmData.size} 字节，总计: ${totalDataSize} 字节") |
|||
} catch (e: Exception) { |
|||
Log.e(TAG, "写入PCM数据失败: ${e.message}", e) |
|||
} |
|||
} |
|||
|
|||
/** |
|||
* 关闭WAV文件并更新文件头 |
|||
*/ |
|||
fun closeWavFile() { |
|||
try { |
|||
if (fos == null || currentWavFile == null) { |
|||
return |
|||
} |
|||
|
|||
// 停止写入线程 |
|||
isWriting.set(false) |
|||
writeThread?.join(1000) // 等待写入线程结束 |
|||
val finalDataSize = totalDataSize |
|||
fos?.close() |
|||
fos = null |
|||
writeQueue.clear() |
|||
// 更新WAV文件头 |
|||
currentWavFile?.let { file -> |
|||
if (file.length() <= WAV_HEADER_SIZE) { |
|||
// 如果文件只有头部没有数据，删除文件 |
|||
file.delete() |
|||
Log.d(TAG, "WAV文件无音频数据，已删除: ${file.absolutePath}") |
|||
} else { |
|||
// 更新文件头中的数据大小信息 |
|||
updateWavHeader(file, finalDataSize) |
|||
Log.d(TAG, "WAV文件保存完成: ${file.absolutePath}") |
|||
Log.d(TAG, "文件大小: ${file.length()} 字节，音频数据: ${finalDataSize} 字节") |
|||
} |
|||
} |
|||
|
|||
} catch (e: Exception) { |
|||
Log.e(TAG, "关闭WAV文件失败: ${e.message}", e) |
|||
} finally { |
|||
currentWavFile = null |
|||
totalDataSize = 0 |
|||
} |
|||
} |
|||
|
|||
/** |
|||
* 启动异步写入线程 |
|||
*/ |
|||
private fun startWriteThread() { |
|||
if (isWriting.get()) return |
|||
|
|||
isWriting.set(true) |
|||
writeThread = Thread { |
|||
try { |
|||
while (isWriting.get() || writeQueue.isNotEmpty()) { |
|||
val data = writeQueue.poll() ?: continue |
|||
fos?.write(data) |
|||
fos?.flush() |
|||
} |
|||
} catch (e: Exception) { |
|||
Log.e(TAG, "异步写入音频数据失败: ${e.message}", e) |
|||
} |
|||
} |
|||
writeThread?.start() |
|||
} |
|||
|
|||
/** |
|||
* 写入WAV文件头 |
|||
* @param dataSize 音频数据大小 |
|||
*/ |
|||
private fun writeWavHeader(dataSize: Int) { |
|||
val header = ByteBuffer.allocate(WAV_HEADER_SIZE) |
|||
header.order(ByteOrder.LITTLE_ENDIAN) |
|||
|
|||
// RIFF头 |
|||
header.put("RIFF".toByteArray()) |
|||
header.putInt(36 + dataSize) // 文件大小 - 8 |
|||
header.put("WAVE".toByteArray()) |
|||
|
|||
// fmt子块 |
|||
header.put("fmt ".toByteArray()) |
|||
header.putInt(16) // fmt子块大小 |
|||
header.putShort(1) // 音频格式（PCM） |
|||
header.putShort(channels.toShort()) // 声道数 |
|||
header.putInt(sampleRate) // 采样率 |
|||
header.putInt(sampleRate * channels * bitsPerSample / 8) // 字节率 |
|||
header.putShort((channels * bitsPerSample / 8).toShort()) // 块对齐 |
|||
header.putShort(bitsPerSample.toShort()) // 位深度 |
|||
|
|||
// data子块 |
|||
header.put("data".toByteArray()) |
|||
header.putInt(dataSize) // 音频数据大小 |
|||
|
|||
fos?.write(header.array()) |
|||
fos?.flush() |
|||
} |
|||
|
|||
/** |
|||
* 更新WAV文件头中的数据大小信息 |
|||
* @param file WAV文件 |
|||
* @param dataSize 实际音频数据大小 |
|||
*/ |
|||
private fun updateWavHeader(file: File, dataSize: Int) { |
|||
try { |
|||
RandomAccessFile(file, "rw").use { raf -> |
|||
// 更新文件大小（位置4） |
|||
raf.seek(4) |
|||
raf.write(intToLittleEndianBytes(36 + dataSize)) |
|||
|
|||
// 更新数据大小（位置40） |
|||
raf.seek(40) |
|||
raf.write(intToLittleEndianBytes(dataSize)) |
|||
} |
|||
} catch (e: IOException) { |
|||
Log.e(TAG, "更新WAV文件头失败: ${e.message}", e) |
|||
} |
|||
} |
|||
|
|||
/** |
|||
* 将整数转换为小端字节序字节数组 |
|||
*/ |
|||
private fun intToLittleEndianBytes(value: Int): ByteArray { |
|||
return byteArrayOf( |
|||
(value and 0xFF).toByte(), |
|||
((value shr 8) and 0xFF).toByte(), |
|||
((value shr 16) and 0xFF).toByte(), |
|||
((value shr 24) and 0xFF).toByte() |
|||
) |
|||
} |
|||
} |
|||
Loading…
Reference in new issue