|
|
@ -1,4 +1,5 @@ |
|
|
package com.yunqiinnovation.azure_speech.tools |
|
|
package com.yunqiinnovation.azure_speech.tools |
|
|
|
|
|
|
|
|
import java.io.File |
|
|
import java.io.File |
|
|
import java.io.FileOutputStream |
|
|
import java.io.FileOutputStream |
|
|
import java.io.RandomAccessFile |
|
|
import java.io.RandomAccessFile |
|
|
@ -8,41 +9,41 @@ import java.io.FileInputStream |
|
|
import java.util.concurrent.LinkedBlockingQueue |
|
|
import java.util.concurrent.LinkedBlockingQueue |
|
|
import java.util.concurrent.atomic.AtomicBoolean |
|
|
import java.util.concurrent.atomic.AtomicBoolean |
|
|
|
|
|
|
|
|
object RecordFile { |
|
|
object RecordFile { |
|
|
private var currentAudioFile: File? = null |
|
|
private var currentAudioFile: File? = null |
|
|
private var fos: FileOutputStream? = null |
|
|
private var fos: FileOutputStream? = null |
|
|
|
|
|
|
|
|
// 新增：用于异步写入的队列和线程 |
|
|
// 新增：用于异步写入的队列和线程 |
|
|
private val writeQueue = LinkedBlockingQueue<ByteArray>() |
|
|
private val writeQueue = LinkedBlockingQueue<ByteArray>() |
|
|
private val isWriting = AtomicBoolean(false) |
|
|
private val isWriting = AtomicBoolean(false) |
|
|
private var writeThread: Thread? = null |
|
|
private var writeThread: Thread? = null |
|
|
|
|
|
|
|
|
// 用于存储音频数据的缓冲区 |
|
|
// 用于存储音频数据的缓冲区 |
|
|
private val dataBuffer = mutableListOf<ByteArray>() |
|
|
private val dataBuffer = mutableListOf<ByteArray>() |
|
|
private var totalBytesWritten = 0 |
|
|
private var totalBytesWritten = 0 |
|
|
|
|
|
|
|
|
// 音频配置 |
|
|
// 音频配置 |
|
|
private val sampleRate = 16000 |
|
|
private val sampleRate = 16000 |
|
|
|
|
|
|
|
|
// 新增：用于存储最后成功保存的文件 |
|
|
// 新增：用于存储最后成功保存的文件 |
|
|
private var lastSavedFile: File? = null |
|
|
private var lastSavedFile: File? = null |
|
|
private var lastFileTimestamp: String = "" |
|
|
private var lastFileTimestamp: String = "" |
|
|
var isPause = false |
|
|
var isPause = false |
|
|
var fileName = "" |
|
|
var fileName = "" |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// 修改：添加 filePath 参数 |
|
|
// 修改：添加 filePath 参数 |
|
|
internal fun creatingFiles(filePath: String) { |
|
|
internal fun creatingFiles(filePath: String) { |
|
|
if (fos != null || currentAudioFile != null) { |
|
|
if (fos != null || currentAudioFile != null) { |
|
|
return |
|
|
return |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
val file = File(filePath) |
|
|
val file = File(filePath) |
|
|
val parentDir = file.parentFile |
|
|
val parentDir = file.parentFile |
|
|
if (!parentDir.exists()) { |
|
|
if (!parentDir.exists()) { |
|
|
parentDir.mkdirs() |
|
|
parentDir.mkdirs() |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
currentAudioFile = file |
|
|
currentAudioFile = file |
|
|
currentAudioFile?.createNewFile() |
|
|
currentAudioFile?.createNewFile() |
|
|
|
|
|
|
|
|
@ -56,59 +57,59 @@ import java.util.concurrent.atomic.AtomicBoolean |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// 新增：启动写入线程 |
|
|
// 新增：启动写入线程 |
|
|
private fun startWriteThread() { |
|
|
private fun startWriteThread() { |
|
|
if (isWriting.get()) return |
|
|
if (isWriting.get()) return |
|
|
isWriting.set(true) |
|
|
isWriting.set(true) |
|
|
writeThread = Thread { |
|
|
writeThread = Thread { |
|
|
try { |
|
|
try { |
|
|
while (isWriting.get() || writeQueue.isNotEmpty()) { |
|
|
while (isWriting.get() || writeQueue.isNotEmpty()) { |
|
|
val data = writeQueue.poll() ?: continue |
|
|
val data = writeQueue.poll() ?: continue |
|
|
fos?.write(data) |
|
|
fos?.write(data) |
|
|
} |
|
|
|
|
|
} catch (e: Exception) { |
|
|
|
|
|
Log.e("tag", "异步写入音频数据失败: ${e.message}") |
|
|
|
|
|
} |
|
|
} |
|
|
|
|
|
} catch (e: Exception) { |
|
|
|
|
|
Log.e("tag", "异步写入音频数据失败: ${e.message}") |
|
|
} |
|
|
} |
|
|
writeThread?.start() |
|
|
|
|
|
} |
|
|
} |
|
|
|
|
|
writeThread?.start() |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
/** |
|
|
/** |
|
|
* 保存音频数据到 WAV 文件（异步） |
|
|
* 保存音频数据到 WAV 文件（异步） |
|
|
*/ |
|
|
*/ |
|
|
internal fun saveAudioDataToWav(buffer: ByteArray) { |
|
|
internal fun saveAudioDataToWav(buffer: ByteArray) { |
|
|
if (fos == null || currentAudioFile == null) return |
|
|
if (fos == null || currentAudioFile == null) return |
|
|
// 放入队列，由写线程写入 |
|
|
// 放入队列，由写线程写入 |
|
|
writeQueue.offer(buffer.copyOf()) |
|
|
writeQueue.offer(buffer.copyOf()) |
|
|
totalBytesWritten += buffer.size |
|
|
totalBytesWritten += buffer.size |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// 更新文件头生成（修正RIFF长度计算） |
|
|
// 更新文件头生成（修正RIFF长度计算） |
|
|
private fun generateWavHeader(dataLength: Int): ByteArray { |
|
|
private fun generateWavHeader(dataLength: Int): ByteArray { |
|
|
val totalLength = 36 + dataLength // RIFF块总长度 = 头部36字节 + 音频数据 |
|
|
val totalLength = 36 + dataLength // RIFF块总长度 = 头部36字节 + 音频数据 |
|
|
val byteRate = sampleRate * 2 * 1 // 采样率 * 字节/样本 * 通道数 |
|
|
val byteRate = sampleRate * 2 * 1 // 采样率 * 字节/样本 * 通道数 |
|
|
|
|
|
|
|
|
return byteArrayOf( |
|
|
return byteArrayOf( |
|
|
'R'.code.toByte(), 'I'.code.toByte(), 'F'.code.toByte(), 'F'.code.toByte(), |
|
|
'R'.code.toByte(), 'I'.code.toByte(), 'F'.code.toByte(), 'F'.code.toByte(), |
|
|
(totalLength and 0xFF).toByte(), ((totalLength shr 8) and 0xFF).toByte(), |
|
|
(totalLength and 0xFF).toByte(), ((totalLength shr 8) and 0xFF).toByte(), |
|
|
((totalLength shr 16) and 0xFF).toByte(), ((totalLength shr 24) and 0xFF).toByte(), |
|
|
((totalLength shr 16) and 0xFF).toByte(), ((totalLength shr 24) and 0xFF).toByte(), |
|
|
'W'.code.toByte(), 'A'.code.toByte(), 'V'.code.toByte(), 'E'.code.toByte(), |
|
|
'W'.code.toByte(), 'A'.code.toByte(), 'V'.code.toByte(), 'E'.code.toByte(), |
|
|
'f'.code.toByte(), 'm'.code.toByte(), 't'.code.toByte(), ' '.code.toByte(), |
|
|
'f'.code.toByte(), 'm'.code.toByte(), 't'.code.toByte(), ' '.code.toByte(), |
|
|
16, 0, 0, 0, // PCM头长度 |
|
|
16, 0, 0, 0, // PCM头长度 |
|
|
1, 0, // PCM格式 |
|
|
1, 0, // PCM格式 |
|
|
1, 0, // 单声道 |
|
|
1, 0, // 单声道 |
|
|
(sampleRate and 0xFF).toByte(), ((sampleRate shr 8) and 0xFF).toByte(), 0, 0, // 采样率 |
|
|
(sampleRate and 0xFF).toByte(), ((sampleRate shr 8) and 0xFF).toByte(), 0, 0, // 采样率 |
|
|
(byteRate and 0xFF).toByte(), ((byteRate shr 8) and 0xFF).toByte(), 0, 0, // 字节率 |
|
|
(byteRate and 0xFF).toByte(), ((byteRate shr 8) and 0xFF).toByte(), 0, 0, // 字节率 |
|
|
2, 0, // 块对齐 (通道数 * 样本位数/8) |
|
|
2, 0, // 块对齐 (通道数 * 样本位数/8) |
|
|
16, 0, // 样本位数 |
|
|
16, 0, // 样本位数 |
|
|
'd'.code.toByte(), 'a'.code.toByte(), 't'.code.toByte(), 'a'.code.toByte(), |
|
|
'd'.code.toByte(), 'a'.code.toByte(), 't'.code.toByte(), 'a'.code.toByte(), |
|
|
(dataLength and 0xFF).toByte(), ((dataLength shr 8) and 0xFF).toByte(), |
|
|
(dataLength and 0xFF).toByte(), ((dataLength shr 8) and 0xFF).toByte(), |
|
|
((dataLength shr 16) and 0xFF).toByte(), ((dataLength shr 24) and 0xFF).toByte() |
|
|
((dataLength shr 16) and 0xFF).toByte(), ((dataLength shr 24) and 0xFF).toByte() |
|
|
) |
|
|
) |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
/** |
|
|
/** |
|
|
* 重命名指定路径的音频文件 |
|
|
* 重命名指定路径的音频文件 |
|
|
* @param filePath 要重命名的文件完整路径 |
|
|
* @param filePath 要重命名的文件完整路径 |
|
|
* @param newName 新的文件名（不含扩展名） |
|
|
* @param newName 新的文件名（不含扩展名） |
|
|
@ -116,28 +117,28 @@ import java.util.concurrent.atomic.AtomicBoolean |
|
|
*/ |
|
|
*/ |
|
|
internal fun renameFile(filePath: String, newName: String): Boolean { |
|
|
internal fun renameFile(filePath: String, newName: String): Boolean { |
|
|
val sourceFile = File(filePath) |
|
|
val sourceFile = File(filePath) |
|
|
|
|
|
|
|
|
// 验证源文件存在 |
|
|
// 验证源文件存在 |
|
|
if (!sourceFile.exists()) { |
|
|
if (!sourceFile.exists()) { |
|
|
Log.w("tag", "重命名失败: 文件不存在 $filePath") |
|
|
Log.w("tag", "重命名失败: 文件不存在 $filePath") |
|
|
return false |
|
|
return false |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
// 防止重命名正在录制的文件 |
|
|
// 防止重命名正在录制的文件 |
|
|
if (sourceFile == currentAudioFile) { |
|
|
if (sourceFile == currentAudioFile) { |
|
|
Log.w("tag", "重命名失败: 不能重命名正在录制的文件") |
|
|
Log.w("tag", "重命名失败: 不能重命名正在录制的文件") |
|
|
return false |
|
|
return false |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
val parentDir = sourceFile.parentFile |
|
|
val parentDir = sourceFile.parentFile |
|
|
val newFile = File(parentDir, "$newName.wav") // 保持.wav扩展名 |
|
|
val newFile = File(parentDir, "$newName.wav") // 保持.wav扩展名 |
|
|
|
|
|
|
|
|
// 处理目标文件已存在的情况 |
|
|
// 处理目标文件已存在的情况 |
|
|
if (newFile.exists()) { |
|
|
if (newFile.exists()) { |
|
|
Log.w("tag", "重命名失败: 目标文件已存在 ${newFile.absolutePath}") |
|
|
Log.w("tag", "重命名失败: 目标文件已存在 ${newFile.absolutePath}") |
|
|
return false |
|
|
return false |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
// 执行重命名 |
|
|
// 执行重命名 |
|
|
return try { |
|
|
return try { |
|
|
val success = sourceFile.renameTo(newFile) |
|
|
val success = sourceFile.renameTo(newFile) |
|
|
@ -156,7 +157,8 @@ import java.util.concurrent.atomic.AtomicBoolean |
|
|
false |
|
|
false |
|
|
} |
|
|
} |
|
|
} |
|
|
} |
|
|
/** |
|
|
|
|
|
|
|
|
/** |
|
|
* 移动文件到新路径 |
|
|
* 移动文件到新路径 |
|
|
* @param sourcePath 原始文件完整路径 |
|
|
* @param sourcePath 原始文件完整路径 |
|
|
* @param destPath 目标文件完整路径 |
|
|
* @param destPath 目标文件完整路径 |
|
|
@ -165,25 +167,25 @@ import java.util.concurrent.atomic.AtomicBoolean |
|
|
internal fun moveFile(sourcePath: String, destPath: String): Boolean { |
|
|
internal fun moveFile(sourcePath: String, destPath: String): Boolean { |
|
|
val sourceFile = File(sourcePath) |
|
|
val sourceFile = File(sourcePath) |
|
|
val destFile = File(destPath) |
|
|
val destFile = File(destPath) |
|
|
|
|
|
|
|
|
if (!sourceFile.exists()) { |
|
|
if (!sourceFile.exists()) { |
|
|
Log.w("tag", "移动失败: 源文件不存在 ${sourceFile.absolutePath}") |
|
|
Log.w("tag", "移动失败: 源文件不存在 ${sourceFile.absolutePath}") |
|
|
return false |
|
|
return false |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
// 确保目标目录存在 |
|
|
// 确保目标目录存在 |
|
|
val destDir = destFile.parentFile |
|
|
val destDir = destFile.parentFile |
|
|
if (!destDir.exists() && !destDir.mkdirs()) { |
|
|
if (!destDir.exists() && !destDir.mkdirs()) { |
|
|
Log.w("tag", "移动失败: 无法创建目标目录 ${destDir.absolutePath}") |
|
|
Log.w("tag", "移动失败: 无法创建目标目录 ${destDir.absolutePath}") |
|
|
return false |
|
|
return false |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
// 删除已存在的目标文件 |
|
|
// 删除已存在的目标文件 |
|
|
if (destFile.exists() && !destFile.delete()) { |
|
|
if (destFile.exists() && !destFile.delete()) { |
|
|
Log.w("tag", "移动失败: 无法覆盖已存在文件 ${destFile.absolutePath}") |
|
|
Log.w("tag", "移动失败: 无法覆盖已存在文件 ${destFile.absolutePath}") |
|
|
return false |
|
|
return false |
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
return try { |
|
|
return try { |
|
|
// 使用renameTo尝试移动（同分区最快） |
|
|
// 使用renameTo尝试移动（同分区最快） |
|
|
if (sourceFile.renameTo(destFile)) { |
|
|
if (sourceFile.renameTo(destFile)) { |
|
|
@ -213,39 +215,40 @@ import java.util.concurrent.atomic.AtomicBoolean |
|
|
false |
|
|
false |
|
|
} |
|
|
} |
|
|
} |
|
|
} |
|
|
internal fun closeFile( isSave:Boolean): Boolean { |
|
|
|
|
|
try { |
|
|
internal fun closeFile(isSave: Boolean): Boolean { |
|
|
if (fos == null || currentAudioFile == null) { |
|
|
try { |
|
|
|
|
|
if (fos == null || currentAudioFile == null) { |
|
|
|
|
|
return false |
|
|
|
|
|
} |
|
|
|
|
|
isWriting.set(false) |
|
|
|
|
|
writeThread?.join(500) |
|
|
|
|
|
fos?.close() |
|
|
|
|
|
|
|
|
|
|
|
currentAudioFile?.let { file -> |
|
|
|
|
|
if (file.length() <= 44 || isSave == false) { |
|
|
|
|
|
file.delete() |
|
|
|
|
|
Log.d("tag", "音频文件过小已删除: ${file.absolutePath}") |
|
|
return false |
|
|
return false |
|
|
} |
|
|
} else { |
|
|
isWriting.set(false) |
|
|
RandomAccessFile(file, "rw").use { raf -> |
|
|
writeThread?.join(500) |
|
|
raf.seek(0) |
|
|
fos?.close() |
|
|
raf.write(generateWavHeader(totalBytesWritten)) |
|
|
|
|
|
|
|
|
currentAudioFile?.let { file -> |
|
|
|
|
|
if (file.length() <= 44||isSave==false) { |
|
|
|
|
|
file.delete() |
|
|
|
|
|
Log.d("tag", "音频文件过小已删除: ${file.absolutePath}") |
|
|
|
|
|
return false |
|
|
|
|
|
} else { |
|
|
|
|
|
RandomAccessFile(file, "rw").use { raf -> |
|
|
|
|
|
raf.seek(0) |
|
|
|
|
|
raf.write(generateWavHeader(totalBytesWritten)) |
|
|
|
|
|
} |
|
|
|
|
|
Log.d("tag", "音频文件保存完成: ${file.absolutePath}") |
|
|
|
|
|
|
|
|
|
|
|
// 保存成功文件引用 |
|
|
|
|
|
lastSavedFile = file |
|
|
|
|
|
return true |
|
|
|
|
|
} |
|
|
} |
|
|
|
|
|
Log.d("tag", "音频文件保存完成: ${file.absolutePath}") |
|
|
|
|
|
|
|
|
|
|
|
// 保存成功文件引用 |
|
|
|
|
|
lastSavedFile = file |
|
|
|
|
|
return true |
|
|
} |
|
|
} |
|
|
} catch (e: Exception) { |
|
|
|
|
|
Log.e("tag", "更新WAV文件头失败: ${e.message}") |
|
|
|
|
|
return false |
|
|
|
|
|
} finally { |
|
|
|
|
|
fos = null |
|
|
|
|
|
currentAudioFile = null |
|
|
|
|
|
} |
|
|
} |
|
|
|
|
|
} catch (e: Exception) { |
|
|
|
|
|
Log.e("tag", "更新WAV文件头失败: ${e.message}") |
|
|
return false |
|
|
return false |
|
|
|
|
|
} finally { |
|
|
|
|
|
fos = null |
|
|
|
|
|
currentAudioFile = null |
|
|
} |
|
|
} |
|
|
} |
|
|
return false |
|
|
|
|
|
} |
|
|
|
|
|
} |
fdp
1 year ago