使用mp4ff库处理MP4文件元数据

mp4ff是一个用于解析和操作MP4文件格式的Go语言库。它可以帮助你读取和修改MP4容器中的视频、音频、字幕轨道以及元数据信息。

安装

go get -u github.com/edgeware/mp4ff

基本用法

1. 解析MP4文件

package main

import (
	"fmt"
	"os"
	"log"
	"github.com/edgeware/mp4ff/mp4"
)

func main() {
	// 打开MP4文件
	file, err := os.Open("example.mp4")
	if err != nil {
		log.Fatal(err)
	}
	defer file.Close()

	// 解析MP4文件
	parsedMp4, err := mp4.DecodeFile(file)
	if err != nil {
		log.Fatal(err)
	}

	// 打印基本信息
	fmt.Printf("File type: %s\n", parsedMp4.Ftyp.Name)
	fmt.Printf("Major brand: %s\n", parsedMp4.Ftyp.MajorBrand)
	fmt.Printf("Timescale: %d\n", parsedMp4.Moov.Mvhd.Timescale)
}

2. 获取视频和音频轨道信息

func printTracksInfo(parsedMp4 *mp4.File) {
	// 遍历所有轨道
	for _, trak := range parsedMp4.Moov.Traks {
		// 获取轨道类型
		handlerType := trak.Mdia.Hdlr.HandlerType
		fmt.Printf("\nTrack ID: %d, Type: %s\n", trak.Tkhd.TrackID, handlerType)

		// 视频轨道
		if handlerType == "vide" {
			width := float64(trak.Tkhd.Width) / 65536
			height := float64(trak.Tkhd.Height) / 65536
			fmt.Printf("  Video dimensions: %.2fx%.2f\n", width, height)
			
			// 获取视频编解码器信息
			if trak.Mdia.Minf.Stbl.Stsd.AvcX != nil {
				fmt.Printf("  Codec: AVC (H.264)\n")
			} else if trak.Mdia.Minf.Stbl.Stsd.HvcX != nil {
				fmt.Printf("  Codec: HEVC (H.265)\n")
			}
		}

		// 音频轨道
		if handlerType == "soun" {
			sampleRate := float64(trak.Mdia.Mdhd.Timescale)
			fmt.Printf("  Audio sample rate: %.0f Hz\n", sampleRate)
			
			// 获取音频编解码器信息
			if trak.Mdia.Minf.Stbl.Stsd.Mp4a != nil {
				fmt.Printf("  Codec: MPEG-4 Audio\n")
			}
		}

		// 字幕轨道
		if handlerType == "subt" {
			fmt.Println("  Subtitle track")
		}
	}
}

3. 读取元数据

func printMetadata(parsedMp4 *mp4.File) {
	// 检查是否有元数据盒子(udta)
	if parsedMp4.Moov.Udta != nil {
		fmt.Println("\nMetadata:")
		
		// 遍历所有元数据条目
		for _, box := range parsedMp4.Moov.Udta.Children {
			switch b := box.(type) {
			case *mp4.MetaBox:
				for _, metaChild := range b.Children {
					if ilst, ok := metaChild.(*mp4.IlstBox); ok {
						for _, entry := range ilst.Entries {
							fmt.Printf("  %s: %s\n", entry.Name, entry.ValueString())
						}
					}
				}
			}
		}
	} else {
		fmt.Println("No metadata found")
	}
}

4. 修改元数据

func addMetadata(parsedMp4 *mp4.File, title, artist string) {
	// 创建或获取udta盒子
	if parsedMp4.Moov.Udta == nil {
		parsedMp4.Moov.Udta = &mp4.UdtaBox{}
	}

	// 创建meta盒子
	meta := &mp4.MetaBox{}
	hdlr := &mp4.HdlrBox{HandlerType: "mdir", Name: "MetadataHandler"}
	meta.AddChild(hdlr)

	// 创建ilst盒子并添加元数据
	ilst := &mp4.IlstBox{}
	
	// 添加标题
	if title != "" {
		ilst.AddEntry(&mp4.MetadataEntry{
			Type: mp4.MetadataEntryString,
			Name: "©nam",
			Data: []byte(title),
		})
	}
	
	// 添加艺术家
	if artist != "" {
		ilst.AddEntry(&mp4.MetadataEntry{
			Type: mp4.MetadataEntryString,
			Name: "©ART",
			Data: []byte(artist),
		})
	}
	
	meta.AddChild(ilst)
	parsedMp4.Moov.Udta.AddChild(meta)
}

5. 保存修改后的文件

func saveModifiedFile(parsedMp4 *mp4.File, outputPath string) error {
	// 创建输出文件
	outFile, err := os.Create(outputPath)
	if err != nil {
		return err
	}
	defer outFile.Close()

	// 写入修改后的MP4
	err = parsedMp4.Encode(outFile)
	if err != nil {
		return err
	}

	return nil
}

完整示例

func main() {
	// 1. 解析MP4文件
	file, err := os.Open("input.mp4")
	if err != nil {
		log.Fatal(err)
	}
	defer file.Close()

	parsedMp4, err := mp4.DecodeFile(file)
	if err != nil {
		log.Fatal(err)
	}

	// 2. 打印轨道信息
	printTracksInfo(parsedMp4)

	// 3. 打印现有元数据
	printMetadata(parsedMp4)

	// 4. 添加新元数据
	addMetadata(parsedMp4, "My Video", "John Doe")

	// 5. 保存修改后的文件
	err = saveModifiedFile(parsedMp4, "output.mp4")
	if err != nil {
		log.Fatal(err)
	}
	fmt.Println("File saved successfully")
}

高级功能

提取视频关键帧时间戳

func printKeyFrameTimes(parsedMp4 *mp4.File) {
	for _, trak := range parsedMp4.Moov.Traks {
		if trak.Mdia.Hdlr.HandlerType == "vide" {
			stss := trak.Mdia.Minf.Stbl.Stss
			stts := trak.Mdia.Minf.Stbl.Stts
			ctts := trak.Mdia.Minf.Stbl.Ctts
			stsc := trak.Mdia.Minf.Stbl.Stsc
			stsz := trak.Mdia.Minf.Stbl.Stsz
			stco := trak.Mdia.Minf.Stbl.Stco

			if stss == nil {
				fmt.Println("No keyframe information available")
				return
			}

			timescale := trak.Mdia.Mdhd.Timescale
			fmt.Printf("Key frames (timescale: %d):\n", timescale)

			for _, sampleNum := range stss.SampleNumber {
				time, err := mp4.GetSampleTime(sampleNum, stts, ctts)
				if err != nil {
					log.Printf("Error getting time for sample %d: %v", sampleNum, err)
					continue
				}
				
				// 转换为秒
				seconds := float64(time) / float64(timescale)
				fmt.Printf("  Sample %d: %.3f seconds\n", sampleNum, seconds)
			}
		}
	}
}

mp4ff库提供了强大的功能来处理MP4文件的各个方面，包括视频、音频、字幕轨道以及元数据。你可以根据需要进一步探索其API文档来实现更复杂的功能。