使用第三方框架 go-astiav 綁定的ffmpeg 5.1

ffmpeg -version 

ffmpeg version 5.1.3 Copyright (c) 2000-2022 the FFmpeg developers
built with Apple clang version 14.0.0 (clang-1400.0.29.201)
configuration: --prefix=/usr/local/ffmpeg --enable-gpl --enable-version3 --enable-nonfree --enable-postproc --disable-libass --enable-libfdk-aac --enable-libfreetype --enable-libopenjpeg --enable-libopus --enable-libspeex --enable-libvorbis --enable-libvpx --enable-libx264 --enable-static --enable-shared --enable-debug=3 --enable-ffplay --enable-libvpx --enable-encoder=libvpx_vp8 --enable-encoder=libvpx_vp9 --enable-decoder=vp8 --enable-decoder=vp9 --enable-parser=vp8 --enable-parser=vp9
libavutil      57. 28.100 / 57. 28.100
libavcodec     59. 37.100 / 59. 37.100
libavformat    59. 27.100 / 59. 27.100
libavdevice    59.  7.100 / 59.  7.100
libavfilter     8. 44.100 /  8. 44.100
libswscale      6.  7.100 /  6.  7.100
libswresample   4.  7.100 /  4.  7.100
libpostproc    56.  6.100 / 56.  6.100

詳細代碼如下：

package main

import (
    "errors"
    "fmt"
    "log"
    "strings"

    "github.com/asticode/go-astiav"
)

var (
    //定義一個字符串數(shù)組
    inputs = []string{
        "/Users/tianwen/GolandProjects/immediate-video-deal/asserts/475/475_2_1682325633772.webm",
        "/Users/tianwen/GolandProjects/immediate-video-deal/asserts/475/475_2_1682325643877.webm",
        "/Users/tianwen/GolandProjects/immediate-video-deal/asserts/475/475_2_1682325684043.webm",
    }
    mergeOutput         = "/Users/tianwen/GolandProjects/immediate-video-deal/asserts/a.webm"
    inputFormatContexts []*astiav.FormatContext
    recodes             []Recode
)

// 請調(diào)整下方的代碼，使得可以將多個視頻文件合并成一個視頻文件
// 要求：
func main() {
    // Handle ffmpeg logs
    astiav.SetLogLevel(astiav.LogLevelDebug)
    astiav.SetLogCallback(func(l astiav.LogLevel, fmt, msg, parent string) {
        log.Printf("ffmpeg log: %s (level: %d)\n", strings.TrimSpace(msg), l)
    })

    // Alloc output format context
    outputFormatContext, err := astiav.AllocOutputFormatContext(nil, "", output)
    if err != nil {
        log.Fatal(fmt.Errorf("main: allocating output format context failed: %w", err))
    }
    if outputFormatContext == nil {
        log.Fatal(errors.New("main: output format context is nil"))
    }
    defer func() {
        outputFormatContext.Free()
        if len(inputFormatContexts) > 0 {
            for _, v := range inputFormatContexts {
                v.Free()
            }
        }
    }()

    // If this is a file, we need to use an io context
    if !outputFormatContext.OutputFormat().Flags().Has(astiav.IOFormatFlagNofile) {
        // Create io context
        ioContext := astiav.NewIOContext()

        // Open io context
        if err = ioContext.Open(mergeOutput, astiav.NewIOContextFlags(astiav.IOContextFlagWrite)); err != nil {
            log.Fatal(fmt.Errorf("main: opening io context failed: %w", err))
        }
        defer ioContext.Closep() //nolint:errcheck

        // Update output format context
        outputFormatContext.SetPb(ioContext)

    }

    // Loop through streams
    outputStreams := make(map[int]*astiav.Stream) // Indexed by input stream index

    for _, input := range inputs {
        // Alloc input format context
        inputFormatContext := astiav.AllocFormatContext()
        if inputFormatContext == nil {
            log.Fatal(errors.New("main: input format context is nil"))
        }

        // Open input
        if err := inputFormatContext.OpenInput(input, nil, nil); err != nil {
            log.Fatal(fmt.Errorf("main: opening input failed: %w", err))
        }

        // Find stream info
        if err := inputFormatContext.FindStreamInfo(nil); err != nil {
            log.Fatal(fmt.Errorf("main: finding stream info failed: %w", err))
        }

        for index, is := range inputFormatContext.Streams() {
            fmt.Println(is.TimeBase())
            // Only process audio or video
            if is.CodecParameters().MediaType() != astiav.MediaTypeAudio &&
                is.CodecParameters().MediaType() != astiav.MediaTypeVideo {
                continue
            }
            if nil == outputStreams[index] {
                // Add stream to output format context
                os := outputFormatContext.NewStream(nil)
                if os == nil {
                    log.Fatal(errors.New("main: output stream is nil"))
                }

                // Copy codec parameters
                if err = is.CodecParameters().Copy(os.CodecParameters()); err != nil {
                    log.Fatal(fmt.Errorf("main: copying codec parameters failed: %w", err))
                }

                // Reset codec tag
                os.CodecParameters().SetCodecTag(0)

                // Add output stream
                outputStreams[index] = os
            }
        }
        inputFormatContexts = append(inputFormatContexts, inputFormatContext)
        recodes = append(recodes, Recode{0, 0, 0, 0})
    }

    // Write header
    if err = outputFormatContext.WriteHeader(nil); err != nil {
        log.Fatal(fmt.Errorf("main: writing header failed: %w", err))
    }

    // Alloc packet
    pkt := astiav.AllocPacket()
    defer pkt.Free()

    for i, inputFormatContext := range inputFormatContexts {
        // Loop through packets
        for {
            // Read frame
            if err = inputFormatContext.ReadFrame(pkt); err != nil {
                if errors.Is(err, astiav.ErrEof) {
                    break
                }
                log.Fatal(fmt.Errorf("main: reading frame failed: %w", err))
            }

            if pkt.Dts() < 0 && pkt.Pts() < 0 {
                continue
            }

            // Get input stream
            inputStream := inputFormatContext.Streams()[pkt.StreamIndex()]

            // Get output stream
            outputStream := outputStreams[pkt.StreamIndex()]

            // Update packet
            pkt.SetStreamIndex(outputStream.Index())

            if pkt.StreamIndex() == 0 { // video stream
                recodes[i].VideoDts = pkt.Dts()
                recodes[i].VideoPts = pkt.Pts()
                pkt.SetDts(getTs(i, 1) + pkt.Dts())
                pkt.SetPts(getTs(i, 2) + pkt.Pts())
            } else if pkt.StreamIndex() == 1 { // audio stream
                recodes[i].VoiceDts = pkt.Dts()
                recodes[i].VoicePts = pkt.Pts()
                pkt.SetDts(getTs(i, 3) + pkt.Dts())
                pkt.SetPts(getTs(i, 4) + pkt.Pts())
            }
            pkt.RescaleTs(inputStream.TimeBase(), outputStream.TimeBase())
            pkt.SetPos(-1)

            // Write frame
            if err = outputFormatContext.WriteInterleavedFrame(pkt); err != nil {
                log.Fatal(fmt.Errorf("main: writing interleaved frame failed: %w", err))
            }
        }
    }

    // Write trailer
    if err = outputFormatContext.WriteTrailer(); err != nil {
        log.Fatal(fmt.Errorf("main: writing trailer failed: %w", err))
    }

    // Success
    log.Println("success")
}

func getTs(index, kind int) int64 {
    var ts int64 = 0
    for i, recode := range recodes {
        if i == index {
            break
        }
        switch kind {
        case 1:
            ts += recode.VideoDts
        case 2:
            ts += recode.VideoPts
        case 3:
            ts += recode.VoiceDts
        case 4:
            ts += recode.VoicePts
        }
    }
    return ts
}

type Recode struct {
    VideoDts int64 `json:"videoDts"` // 視頻最后編碼時間
    VideoPts int64 `json:"videoPts"` // 視頻最后播放時間
    VoiceDts int64 `json:"voiceDts"` // 聲音最后編碼時間
    VoicePts int64 `json:"voicePts"` // 聲音最后播放時間
}