fdp
1 year ago
13 changed files with 833 additions and 246 deletions
@ -0,0 +1,251 @@ |
|||||
|
package com.yunqiinnovation.azure_speech.tools |
||||
|
import java.io.File |
||||
|
import java.io.FileOutputStream |
||||
|
import java.io.RandomAccessFile |
||||
|
|
||||
|
import android.util.Log |
||||
|
import java.io.FileInputStream |
||||
|
import java.util.concurrent.LinkedBlockingQueue |
||||
|
import java.util.concurrent.atomic.AtomicBoolean |
||||
|
|
||||
|
object RecordFile { |
||||
|
private var currentAudioFile: File? = null |
||||
|
private var fos: FileOutputStream? = null |
||||
|
|
||||
|
// 新增：用于异步写入的队列和线程 |
||||
|
private val writeQueue = LinkedBlockingQueue<ByteArray>() |
||||
|
private val isWriting = AtomicBoolean(false) |
||||
|
private var writeThread: Thread? = null |
||||
|
|
||||
|
// 用于存储音频数据的缓冲区 |
||||
|
private val dataBuffer = mutableListOf<ByteArray>() |
||||
|
private var totalBytesWritten = 0 |
||||
|
|
||||
|
// 音频配置 |
||||
|
private val sampleRate = 16000 |
||||
|
|
||||
|
// 新增：用于存储最后成功保存的文件 |
||||
|
private var lastSavedFile: File? = null |
||||
|
private var lastFileTimestamp: String = "" |
||||
|
var isPause = false |
||||
|
var fileName = "" |
||||
|
|
||||
|
|
||||
|
// 修改：添加 filePath 参数 |
||||
|
internal fun creatingFiles(filePath: String) { |
||||
|
if (fos != null || currentAudioFile != null) { |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
val file = File(filePath) |
||||
|
val parentDir = file.parentFile |
||||
|
if (!parentDir.exists()) { |
||||
|
parentDir.mkdirs() |
||||
|
} |
||||
|
|
||||
|
currentAudioFile = file |
||||
|
currentAudioFile?.createNewFile() |
||||
|
|
||||
|
fos = FileOutputStream(currentAudioFile).apply { |
||||
|
write(generateWavHeader(0)) |
||||
|
close() |
||||
|
} |
||||
|
|
||||
|
fos = FileOutputStream(currentAudioFile, true) |
||||
|
startWriteThread() |
||||
|
} |
||||
|
|
||||
|
|
||||
|
// 新增：启动写入线程 |
||||
|
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) |
||||
|
} |
||||
|
} catch (e: Exception) { |
||||
|
Log.e("tag", "异步写入音频数据失败: ${e.message}") |
||||
|
} |
||||
|
} |
||||
|
writeThread?.start() |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 保存音频数据到 WAV 文件（异步） |
||||
|
*/ |
||||
|
internal fun saveAudioDataToWav(buffer: ByteArray) { |
||||
|
if (fos == null || currentAudioFile == null) return |
||||
|
// 放入队列，由写线程写入 |
||||
|
writeQueue.offer(buffer.copyOf()) |
||||
|
totalBytesWritten += buffer.size |
||||
|
} |
||||
|
|
||||
|
|
||||
|
// 更新文件头生成（修正RIFF长度计算） |
||||
|
private fun generateWavHeader(dataLength: Int): ByteArray { |
||||
|
val totalLength = 36 + dataLength // RIFF块总长度 = 头部36字节 + 音频数据 |
||||
|
val byteRate = sampleRate * 2 * 1 // 采样率 * 字节/样本 * 通道数 |
||||
|
|
||||
|
return byteArrayOf( |
||||
|
'R'.code.toByte(), 'I'.code.toByte(), 'F'.code.toByte(), 'F'.code.toByte(), |
||||
|
(totalLength and 0xFF).toByte(), ((totalLength shr 8) 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(), |
||||
|
'f'.code.toByte(), 'm'.code.toByte(), 't'.code.toByte(), ' '.code.toByte(), |
||||
|
16, 0, 0, 0, // PCM头长度 |
||||
|
1, 0, // PCM格式 |
||||
|
1, 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, // 字节率 |
||||
|
2, 0, // 块对齐 (通道数 * 样本位数/8) |
||||
|
16, 0, // 样本位数 |
||||
|
'd'.code.toByte(), 'a'.code.toByte(), 't'.code.toByte(), 'a'.code.toByte(), |
||||
|
(dataLength and 0xFF).toByte(), ((dataLength shr 8) and 0xFF).toByte(), |
||||
|
((dataLength shr 16) and 0xFF).toByte(), ((dataLength shr 24) and 0xFF).toByte() |
||||
|
) |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 重命名指定路径的音频文件 |
||||
|
* @param filePath 要重命名的文件完整路径 |
||||
|
* @param newName 新的文件名（不含扩展名） |
||||
|
* @return true 重命名成功，false 失败 |
||||
|
*/ |
||||
|
internal fun renameFile(filePath: String, newName: String): Boolean { |
||||
|
val sourceFile = File(filePath) |
||||
|
|
||||
|
// 验证源文件存在 |
||||
|
if (!sourceFile.exists()) { |
||||
|
Log.w("tag", "重命名失败: 文件不存在 $filePath") |
||||
|
return false |
||||
|
} |
||||
|
|
||||
|
// 防止重命名正在录制的文件 |
||||
|
if (sourceFile == currentAudioFile) { |
||||
|
Log.w("tag", "重命名失败: 不能重命名正在录制的文件") |
||||
|
return false |
||||
|
} |
||||
|
|
||||
|
val parentDir = sourceFile.parentFile |
||||
|
val newFile = File(parentDir, "$newName.wav") // 保持.wav扩展名 |
||||
|
|
||||
|
// 处理目标文件已存在的情况 |
||||
|
if (newFile.exists()) { |
||||
|
Log.w("tag", "重命名失败: 目标文件已存在 ${newFile.absolutePath}") |
||||
|
return false |
||||
|
} |
||||
|
|
||||
|
// 执行重命名 |
||||
|
return try { |
||||
|
val success = sourceFile.renameTo(newFile) |
||||
|
if (success) { |
||||
|
Log.d("tag", "文件重命名成功: ${sourceFile.name} -> ${newFile.name}") |
||||
|
// 更新最后保存的文件引用（如果是同一个文件） |
||||
|
if (lastSavedFile == sourceFile) { |
||||
|
lastSavedFile = newFile |
||||
|
} |
||||
|
} else { |
||||
|
Log.w("tag", "文件重命名操作失败") |
||||
|
} |
||||
|
success |
||||
|
} catch (e: Exception) { |
||||
|
Log.e("tag", "重命名异常: ${e.message}") |
||||
|
false |
||||
|
} |
||||
|
} |
||||
|
/** |
||||
|
* 移动文件到新路径 |
||||
|
* @param sourcePath 原始文件完整路径 |
||||
|
* @param destPath 目标文件完整路径 |
||||
|
* @return true 移动成功，false 移动失败 |
||||
|
*/ |
||||
|
internal fun moveFile(sourcePath: String, destPath: String): Boolean { |
||||
|
val sourceFile = File(sourcePath) |
||||
|
val destFile = File(destPath) |
||||
|
|
||||
|
if (!sourceFile.exists()) { |
||||
|
Log.w("tag", "移动失败: 源文件不存在 ${sourceFile.absolutePath}") |
||||
|
return false |
||||
|
} |
||||
|
|
||||
|
// 确保目标目录存在 |
||||
|
val destDir = destFile.parentFile |
||||
|
if (!destDir.exists() && !destDir.mkdirs()) { |
||||
|
Log.w("tag", "移动失败: 无法创建目标目录 ${destDir.absolutePath}") |
||||
|
return false |
||||
|
} |
||||
|
|
||||
|
// 删除已存在的目标文件 |
||||
|
if (destFile.exists() && !destFile.delete()) { |
||||
|
Log.w("tag", "移动失败: 无法覆盖已存在文件 ${destFile.absolutePath}") |
||||
|
return false |
||||
|
} |
||||
|
|
||||
|
return try { |
||||
|
// 使用renameTo尝试移动（同分区最快） |
||||
|
if (sourceFile.renameTo(destFile)) { |
||||
|
true |
||||
|
} else { |
||||
|
// 跨分区移动：复制+删除 |
||||
|
FileInputStream(sourceFile).use { input -> |
||||
|
FileOutputStream(destFile).use { output -> |
||||
|
input.copyTo(output) |
||||
|
} |
||||
|
} |
||||
|
if (!sourceFile.delete()) { |
||||
|
Log.w("tag", "移动警告: 源文件删除失败 ${sourceFile.absolutePath}") |
||||
|
} |
||||
|
true |
||||
|
}.also { |
||||
|
if (it) { |
||||
|
Log.d("tag", "文件移动成功: ${sourcePath} -> ${destPath}") |
||||
|
// 更新最后保存的文件引用 |
||||
|
if (lastSavedFile?.absolutePath == sourcePath) { |
||||
|
lastSavedFile = destFile |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
} catch (e: Exception) { |
||||
|
Log.e("tag", "文件移动异常: ${e.message}") |
||||
|
false |
||||
|
} |
||||
|
} |
||||
|
internal fun closeFile( isSave:Boolean): Boolean { |
||||
|
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 |
||||
|
} else { |
||||
|
RandomAccessFile(file, "rw").use { raf -> |
||||
|
raf.seek(0) |
||||
|
raf.write(generateWavHeader(totalBytesWritten)) |
||||
|
} |
||||
|
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 |
||||
|
} |
||||
|
return false |
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue