3 changed files with 245 additions and 2 deletions
@ -0,0 +1,188 @@ |
|||||
|
package com.threecloud.dataserviceyy.util; |
||||
|
|
||||
|
import org.bytedeco.javacpp.Loader; |
||||
|
import org.slf4j.Logger; |
||||
|
import org.slf4j.LoggerFactory; |
||||
|
import org.springframework.beans.factory.annotation.Value; |
||||
|
import org.springframework.stereotype.Component; |
||||
|
|
||||
|
import java.io.BufferedReader; |
||||
|
import java.io.File; |
||||
|
import java.io.IOException; |
||||
|
import java.io.InputStreamReader; |
||||
|
import java.io.RandomAccessFile; |
||||
|
import java.nio.charset.StandardCharsets; |
||||
|
import java.util.ArrayList; |
||||
|
import java.util.List; |
||||
|
import java.util.concurrent.TimeUnit; |
||||
|
|
||||
|
/** |
||||
|
* G.729 录音转码工具 |
||||
|
* |
||||
|
* 【背景】录音盒导出的 WAV 部分为 G.729 压缩编码（wFormatTag=0x729a）， |
||||
|
* 浏览器无法直接播放，需转码为标准 PCM WAV。 |
||||
|
* 【方式】调用系统 ffmpeg 命令完成转码（ffmpeg 内置 G.729 解码器，无需额外库）。 |
||||
|
* 【降级】ffmpeg 不存在或转码失败时抛出异常，由调用方决定是否回退原文件。 |
||||
|
*/ |
||||
|
@Component |
||||
|
public class G729Converter { |
||||
|
|
||||
|
private static final Logger logger = LoggerFactory.getLogger(G729Converter.class); |
||||
|
|
||||
|
/** G.729 的 WAV 格式标签（FFmpeg TwoCC，0x729a） */ |
||||
|
private static final int WAVE_FORMAT_G729 = 0x729a; |
||||
|
|
||||
|
/** 转码超时时间（秒），防止异常文件卡死 */ |
||||
|
private static final long TRANSCODE_TIMEOUT_SECONDS = 600; |
||||
|
|
||||
|
/** 转码方式：system=系统ffmpeg（默认），javacv=bytedeco内置ffmpeg */ |
||||
|
@Value("${ffmpeg.mode:javacv}") |
||||
|
private String ffmpegMode; |
||||
|
|
||||
|
/** 系统 ffmpeg 可执行文件路径（mode=system 时生效），默认从 PATH 中查找 */ |
||||
|
@Value("${ffmpeg.path:ffmpeg}") |
||||
|
private String ffmpegPath; |
||||
|
|
||||
|
/** |
||||
|
* 判断文件是否为 G.729 编码的 WAV |
||||
|
* |
||||
|
* @param file 音频文件 |
||||
|
* @return true 表示为 G.729（wFormatTag=0x729a），需要转码 |
||||
|
*/ |
||||
|
public boolean isG729(File file) { |
||||
|
if (file == null || !file.isFile()) { |
||||
|
return false; |
||||
|
} |
||||
|
try { |
||||
|
Integer format = readWavAudioFormat(file); |
||||
|
return format != null && format == WAVE_FORMAT_G729; |
||||
|
} catch (Exception e) { |
||||
|
logger.warn("判断 WAV 编码失败: {}, 原因={}", file.getName(), e.getMessage()); |
||||
|
return false; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 将 G.729 WAV 转码为 PCM WAV（16bit 单声道 8kHz） |
||||
|
* |
||||
|
* @param src 源 G.729 文件 |
||||
|
* @return 转码后的 PCM 文件（生成在源文件同目录，文件名为源文件名 + ".pcm.wav"） |
||||
|
* @throws IOException IO 异常或 ffmpeg 转码失败 |
||||
|
* @throws InterruptedException 等待被中断 |
||||
|
*/ |
||||
|
public File convertToPcm(File src) throws IOException, InterruptedException { |
||||
|
if (src == null || !src.isFile()) { |
||||
|
throw new IOException("待转码文件不存在: " + (src == null ? "null" : src.getPath())); |
||||
|
} |
||||
|
File dest = new File(src.getParentFile(), src.getName() + ".pcm.wav"); |
||||
|
|
||||
|
List<String> command = new ArrayList<>(); |
||||
|
command.add(resolveFfmpegPath()); |
||||
|
command.add("-y"); |
||||
|
command.add("-i"); |
||||
|
command.add(src.getAbsolutePath()); |
||||
|
command.add("-ar"); |
||||
|
command.add("8000"); |
||||
|
command.add("-ac"); |
||||
|
command.add("1"); |
||||
|
command.add("-c:a"); |
||||
|
command.add("pcm_s16le"); |
||||
|
command.add(dest.getAbsolutePath()); |
||||
|
|
||||
|
logger.info("执行 ffmpeg 转码: {}", String.join(" ", command)); |
||||
|
ProcessBuilder pb = new ProcessBuilder(command); |
||||
|
pb.redirectErrorStream(true); |
||||
|
Process process = pb.start(); |
||||
|
|
||||
|
boolean finished = process.waitFor(TRANSCODE_TIMEOUT_SECONDS, TimeUnit.SECONDS); |
||||
|
if (!finished) { |
||||
|
process.destroyForcibly(); |
||||
|
throw new IOException("ffmpeg 转码超时（>" + TRANSCODE_TIMEOUT_SECONDS + "s）: " + src.getName()); |
||||
|
} |
||||
|
|
||||
|
// 进程已结束，读取输出用于日志与错误排查
|
||||
|
StringBuilder output = new StringBuilder(); |
||||
|
try (BufferedReader reader = new BufferedReader( |
||||
|
new InputStreamReader(process.getInputStream(), StandardCharsets.UTF_8))) { |
||||
|
String line; |
||||
|
while ((line = reader.readLine()) != null) { |
||||
|
output.append(line).append('\n'); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
int exitCode = process.exitValue(); |
||||
|
if (exitCode != 0) { |
||||
|
String err = output.length() > 500 ? output.substring(output.length() - 500) : output.toString(); |
||||
|
throw new IOException("ffmpeg 转码失败(exit=" + exitCode + "): " + src.getName() + " -> " + err); |
||||
|
} |
||||
|
if (!dest.isFile() || dest.length() == 0) { |
||||
|
throw new IOException("ffmpeg 转码未生成有效文件: " + dest.getAbsolutePath()); |
||||
|
} |
||||
|
return dest; |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 根据配置解析 ffmpeg 可执行文件路径 |
||||
|
* |
||||
|
* mode=system：使用系统安装的 ffmpeg（ffmpeg.path 配置） |
||||
|
* mode=javacv：使用 bytedeco 打包在依赖中的 ffmpeg（无需服务器安装） |
||||
|
* |
||||
|
* @return ffmpeg 可执行文件路径 |
||||
|
*/ |
||||
|
private String resolveFfmpegPath() { |
||||
|
if ("javacv".equalsIgnoreCase(ffmpegMode)) { |
||||
|
return Loader.load(org.bytedeco.ffmpeg.ffmpeg.class); |
||||
|
} |
||||
|
return ffmpegPath; |
||||
|
} |
||||
|
|
||||
|
/** |
||||
|
* 读取 WAV 文件的音频格式标签（wFormatTag） |
||||
|
* |
||||
|
* 遍历 RIFF chunk，定位 fmt chunk 后读取前 2 字节（小端）。 |
||||
|
* |
||||
|
* @param file WAV 文件 |
||||
|
* @return wFormatTag；非 WAV 或解析失败返回 null |
||||
|
*/ |
||||
|
private Integer readWavAudioFormat(File file) throws IOException { |
||||
|
try (RandomAccessFile raf = new RandomAccessFile(file, "r")) { |
||||
|
if (raf.length() < 12) { |
||||
|
return null; |
||||
|
} |
||||
|
byte[] riff = new byte[12]; |
||||
|
raf.readFully(riff); |
||||
|
// RIFF....WAVE
|
||||
|
if (!(riff[0] == 'R' && riff[1] == 'I' && riff[2] == 'F' && riff[3] == 'F' |
||||
|
&& riff[8] == 'W' && riff[9] == 'A' && riff[10] == 'V' && riff[11] == 'E')) { |
||||
|
return null; |
||||
|
} |
||||
|
|
||||
|
long offset = 12; |
||||
|
while (offset + 8 <= raf.length()) { |
||||
|
raf.seek(offset); |
||||
|
byte[] chunkHeader = new byte[8]; |
||||
|
raf.readFully(chunkHeader); |
||||
|
String chunkId = new String(chunkHeader, 0, 4, StandardCharsets.US_ASCII); |
||||
|
long chunkSize = (chunkHeader[4] & 0xFFL) |
||||
|
| ((chunkHeader[5] & 0xFFL) << 8) |
||||
|
| ((chunkHeader[6] & 0xFFL) << 16) |
||||
|
| ((chunkHeader[7] & 0xFFL) << 24); |
||||
|
if ("fmt ".equals(chunkId)) { |
||||
|
if (chunkSize < 2) { |
||||
|
return null; |
||||
|
} |
||||
|
raf.seek(offset + 8); |
||||
|
int b0 = raf.readUnsignedByte(); |
||||
|
int b1 = raf.readUnsignedByte(); |
||||
|
return b0 | (b1 << 8); |
||||
|
} |
||||
|
offset += 8 + chunkSize; |
||||
|
// RIFF chunk 按 2 字节对齐
|
||||
|
if (chunkSize % 2 == 1) { |
||||
|
offset += 1; |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
return null; |
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue