使用Android系统的MediaMuxer写一个音视频合并的方法。

使用Android系统的MediaMuxer写一个音视频合并的方法。

import android.media.MediaExtractor
import android.media.MediaFormat
import android.media.MediaMuxer
import java.io.IOException

fun mergeAudioVideo(audioFile: String, videoFile: String, outputFile: String) {
    // 创建一个MediaExtractor对象，用于从音频文件中提取音频轨道
    val audioExtractor = MediaExtractor()
    try {
        audioExtractor.setDataSource(audioFile)
    } catch (e: IOException) {
        e.printStackTrace()
        return
    }
    // 检查音频文件是否有音频轨道，并获取其格式
    val audioTrackCount = audioExtractor.trackCount
    if (audioTrackCount <= 0) {
        println("No audio track found in $audioFile")
        return
    }
    audioExtractor.selectTrack(0) // 选择第一个音频轨道
    val audioFormat = audioExtractor.getTrackFormat(0)

    // 创建一个MediaExtractor对象，用于从视频文件中提取视频轨道
    val videoExtractor = MediaExtractor()
    try {
        videoExtractor.setDataSource(videoFile)
    } catch (e: IOException) {
        e.printStackTrace()
        return
    }
    // 检查视频文件是否有视频轨道，并获取其格式
    val videoTrackCount = videoExtractor.trackCount
    if (videoTrackCount <= 0) {
        println("No video track found in $videoFile")
        return
    }
    videoExtractor.selectTrack(0) // 选择第一个视频轨道
    val videoFormat = videoExtractor.getTrackFormat(0)

    // 创建一个MediaMuxer对象，用于将音频和视频轨道复用到输出文件中
    val muxer = MediaMuxer(outputFile, MediaMuxer.OutputFormat.MUXER_OUTPUT_MPEG_4)
    // 添加音频和视频轨道，并获取它们的索引
    val audioTrackIndex = muxer.addTrack(audioFormat)
    val videoTrackIndex = muxer.addTrack(videoFormat)
    // 开始复用
    muxer.start()

    // 定义一些变量，用于读取和写入音频和视频数据
    val bufferInfo = MediaCodec.BufferInfo() // 用于存储媒体数据的元信息，如时间戳，大小，偏移量等
    val bufferSize = 256 * 1024 // 设置缓冲区的大小，根据需要调整
    val buffer = ByteBuffer.allocate(bufferSize) // 创建一个字节缓冲区，用于存储媒体数据
    var isAudioDone = false // 标记音频是否读取完毕
    var isVideoDone = false // 标记视频是否读取完毕

    // 循环读取和写入音频和视频数据，直到其中一个结束
    while (!isAudioDone || !isVideoDone) {
        // 读取音频数据
        if (!isAudioDone) {
            // 从音频轨道中读取一帧数据到缓冲区中
            val audioSize = audioExtractor.readSampleData(buffer, 0)
            if (audioSize < 0) {
                // 如果读取到了文件末尾，标记音频结束
                isAudioDone = true
            } else {
                // 如果读取到了有效的数据，设置缓冲区信息
                bufferInfo.size = audioSize
                bufferInfo.offset = 0
                bufferInfo.presentationTimeUs = audioExtractor.sampleTime // 获取当前帧的时间戳
                bufferInfo.flags = audioExtractor.sampleFlags // 获取当前帧的标志位，如是否为关键帧等
                // 将缓冲区中的数据写入到输出文件的音频轨道中
                muxer.writeSampleData(audioTrackIndex, buffer, bufferInfo)
                // 移动到下一帧
                audioExtractor.advance()
            }
        }

        // 读取视频数据
        if (!isVideoDone) {
            // 从视频轨道中读取一帧数据到缓冲区中
            val videoSize = videoExtractor.readSampleData(buffer, 0)
            if (videoSize < 0) {
                // 如果读取到了文件末尾，标记视频结束
                isVideoDone = true
            } else {
                // 如果读取到了有效的数据，设置缓冲区信息
                bufferInfo.size = videoSize
                bufferInfo.offset = 0
                bufferInfo.presentationTimeUs = videoExtractor.sampleTime // 获取当前帧的时间戳
                bufferInfo.flags = videoExtractor.sampleFlags // 获取当前帧的标志位，如是否为关键帧等
                // 将缓冲区中的数据写入到输出文件的视频轨道中
                muxer.writeSampleData(videoTrackIndex, buffer, bufferInfo)
                // 移动到下一帧
                videoExtractor.advance()
            }
        }
    }

    // 结束复用并释放资源
    muxer.stop()
    muxer.release()
    audioExtractor.release()
    videoExtractor.release()
}