.ass 时间基准和视频对不上时按第一条弹幕自动对齐,并去掉重复的还原日志

1. 整场弹幕都被判「超出视频时长」,合并直接取消

   上游原版录出来的 .ass 时间不是相对录制起点,而是相对弹幕客户端连上房间的时刻。
   程序先开着、直播中途才开始录的场次,这份 .ass 的零点会比视频起点早一大截
   (用户实测那份早了 7849.92 秒:955 条弹幕、跨度 5436.7 秒,视频只有 5456.9 秒)。
   一律按「视频时间 = 弹幕时间 + 偏移」算的话,这 955 条全都落在 5456.9 秒之外,
   于是「有 955 条弹幕超出了录播视频的时长,已跳过」→「弹幕文件里没有弹幕,取消手动合并」,
   一分半的录播一条弹幕都合不出来。

   这类偏移是能救的:跨度(末 - 首)5436.7 秒装得进 5456.9 秒,整体前移 7849.92 秒之后
   覆盖 0 ~ 5436.7 秒、尾部正好空 20.2 秒 —— 和正常录播的尾部空隙(实测 6~20 秒)吻合,
   是时间基准偏移,不是文件配错。

   新增 fitDanmuToVideo,在探到视频时长之后、裁剪之前介入,触发条件收得很紧:
   只有「按现偏移一条弹幕都用不上」**且**「弹幕跨度 < 视频时长」才整体平移,把第一条
   对齐到视频起点,日志写明平移量、对齐后的覆盖范围和尾部空隙,并提示整体偏早或偏晚
   可以在「偏移」里微调。其余情况完全保持原行为:只是一部分超出 → 照旧只丢尾部;
   跨度本身就装不下 → 一条不留。这样不可能影响任何现在能正常合并的场次。

2. 「已从 xxx.ass 还原出 N 条弹幕」会打印两遍

   页面的「检查」和真正合并会各解析一遍 .ass,而这条日志写在被复用的 parseAssFile 里,
   于是同一句话出现两次,第二遍后面还跟着「已把 xxx.ass 还原成 xxx.jsonl」,看不出谁是谁。
   把这两条日志从 parseAssFile 里摘出去(它改成返回坏行数 bad,由调用方决定要不要告警),
   交给真正落盘那次的调用方 prepareDanmuSource(path, true) 打印,现在还会写明写成了
   哪个文件;坏行警告统一走新增的 logAssBadLines,不再重复。

   顺手给「最后一条弹幕超出视频总长」那条警告补了指引,说明合成时会自动对齐。

3. 测试里后台任务漏到用例结束之后

   safeGo 原来直接把 goroutine 扔出去,用例结束它还在读全局配置,下一个用例开头一改配置,
   -race 就判成数据竞争(而且报在后面的用例上,很难看出真凶)。把「怎么起后台任务」抽成
   var runBackground,生产行为不变,测试里换成可等待的版本,清理时等它收工。

   注意:这个竞争单独 -run 一个用例复现不出来(-race -count=3 也过),必须整包跑
   go test -race . 才会在后续用例上报 race detected。

验证

- gofmt 无输出 / go vet 通过 / go test(182 个,18.5s)全过 / go test -race 整包跑 20 轮
  (13 轮顺序 + 6 轮 -shuffle=on)全过 / staticcheck 无输出 / GOOS=linux go build 通过。
  另:有一轮 -race 曾报过 FAIL、随后 20 轮都没能复现,怀疑是 TestDownloadAvatarCapsResponseBody
  自带的抖动(它的断言依赖「服务端写不完 10MB」,受内核缓冲大小影响),与本批改动无关,
  该文件本批也没动过;待单独确认。
- 反向验证从 32 项扩到 37 项全过(新增:自动对齐 / 只丢尾部不平移 / 跨度装不进不强平移 /
  还原日志只打一次 / 后台任务可等待);脚本新增单用例附加参数 "args" 与不带 -run 的整包用例。
- 用用户那份真实素材(955 条,7849.92 ~ 13286.59 秒,视频 5456.883 秒)端到端跑通:
  不填偏移 + 5460 秒视频 → 日志「弹幕时间基准和视频对不上:955 条弹幕全部落在视频时长
  (5460.0 秒)之外,最早那条在 7849.9 秒;已按「第一条弹幕对齐视频起点」整体前移
  7849.9 秒(偏移 0.0 → -7849.9)」「对齐后弹幕覆盖 0.0 ~ 5436.7 秒,视频总长 5460.0 秒
  (尾部空 23.3 秒)」,分 5 块渲染,exit=0,成片 5462.31 秒 / 23.3 MB。
  再按 jsonl + 偏移反推「哪些时刻该有弹幕」,随机取 40 个应有时刻 + 20 个任意弹幕窗口都
  不覆盖的空屏时刻,抽帧数白像素(白字黑底):60/60 全对,跨块边界(1116/2435/3393/4408)
  与最尾部 t=5435.7 都对得上。
- 60 秒小片新旧对照(真实 9 条弹幕、原始时间戳 7849.92 ~ 7908.7):旧版「有 9 条弹幕超出了
  录播视频的时长(60.1 秒),已跳过」+「jsonl 里没有弹幕」、无成片;新版自动前移 7849.9 秒、
  9 条全画上、成片 34 MB。抽帧看轨迹仍是匀速 10 秒穿屏(danmuShowSeconds),
  头像跟着文字同步走,没有回归。
This commit is contained in:
tamakyi
2026-09-26 17:51:43 +08:00
parent 6f3c8096b4
commit 3e16e78003
8 changed files with 353 additions and 38 deletions
+10 -14
View File
@@ -139,12 +139,14 @@ func speakerText(name string, uid int64, content string) string {
}
// 从 ASS 文件还原弹幕记录,支持原格式(\move)和昵称头像版(\an7\move)
func parseAssFile(path string) (danmuMeta, []danmuRecord, error) {
// 第二个返回值是没能解析出来的行数,日志交给调用方打:页面的「检查」和真正的合并
// 会各解析一遍同一个 .ass,日志只该出现一次
func parseAssFile(path string) (danmuMeta, []danmuRecord, int, error) {
var meta danmuMeta
f, err := os.Open(path)
if err != nil {
return meta, nil, err
return meta, nil, 0, err
}
defer f.Close()
@@ -275,33 +277,27 @@ func parseAssFile(path string) (danmuMeta, []danmuRecord, error) {
}
}
if err := scanner.Err(); err != nil {
return meta, nil, err
return meta, nil, bad, err
}
if meta.PlayResX == 0 || meta.PlayResY == 0 {
return meta, nil, errors.New("ASS 文件里没有 PlayResX / PlayResY,可能不是弹幕文件")
return meta, nil, bad, errors.New("ASS 文件里没有 PlayResX / PlayResY,可能不是弹幕文件")
}
if meta.FontSize == 0 {
return meta, nil, errors.New("ASS 文件里没找到字幕字号(Style: Danmu)")
return meta, nil, bad, errors.New("ASS 文件里没找到字幕字号(Style: Danmu)")
}
if len(records) == 0 {
if bad > 0 {
return meta, nil, fmt.Errorf("ASS 里的 %d 行弹幕都没能解析出来", bad)
return meta, nil, bad, fmt.Errorf("ASS 里的 %d 行弹幕都没能解析出来", bad)
}
return meta, nil, errNoDanmu
return meta, nil, bad, errNoDanmu
}
// 分块渲染要求记录按时间递增,这里统一排一遍
sort.SliceStable(records, func(i, j int) bool { return records[i].Time < records[j].Time })
// 文件名前缀就是录制开始的时刻,用它算偏移最稳
meta.StartTime = nameStartNs(path)
if bad > 0 {
lPrintWarnf("%s 里有 %d 行弹幕没能解析出来,已跳过", filepath.Base(path), bad)
}
lPrintf("已从 %s 还原出 %d 条弹幕(%dx%d,字号 %d,弹幕到 %s)",
filepath.Base(path), len(records), meta.PlayResX, meta.PlayResY, meta.FontSize,
formatSeconds(records[len(records)-1].Time))
return meta, records, nil
return meta, records, bad, nil
}
// 按本程序的 jsonl 格式写文件(首行是元数据)
+61 -11
View File
@@ -287,6 +287,64 @@ func runDanmuburn(danmuFile, videoFile, outFile string, offset float64) {
dropTempDanmu(src)
}
// 按偏移算出视频时间,落在视频时长之外的丢掉;返回留下的弹幕和被丢掉的条数
func clipDanmuToVideo(records []danmuRecord, offset, videoDuration float64) ([]danmuRecord, int) {
kept := make([]danmuRecord, 0, len(records))
for _, r := range records {
if r.Time+offset < videoDuration {
kept = append(kept, r)
}
}
return kept, len(records) - len(kept)
}
// 把弹幕对齐到视频的时间轴。
// 正常情况就是「超出视频时长的部分丢掉」;但 upstream 录出来的 .ass 时间不一定相对录制起点
// (程序先开着、直播中途才开始录的场次,弹幕时间基准会比视频晚一大截),
// 那种场次会被整体判成「超出时长」然后一条都合不出来,这里按「第一条弹幕对齐视频起点」平移救回来。
// 只在「一条都用不上」且「弹幕跨度装得进视频」时才平移,本来就能合成的场次完全不受影响。
// 返回调整后的弹幕和偏移(后续渲染都用这个偏移)。
func fitDanmuToVideo(records []danmuRecord, offset, videoDuration float64) ([]danmuRecord, float64) {
kept, dropped := clipDanmuToVideo(records, offset, videoDuration)
if dropped == 0 {
return records, offset
}
if len(kept) > 0 {
// 只是一部分超出(弹幕确实录得比视频久),保持原行为:丢掉超出的那部分
lPrintWarnf("有 %d 条弹幕超出了录播视频的时长(%.1f 秒),已跳过", dropped, videoDuration)
return kept, offset
}
first, last := records[0].Time, records[0].Time
for _, r := range records[1:] {
if r.Time < first {
first = r.Time
}
if r.Time > last {
last = r.Time
}
}
late := first + offset // 最早的那条比视频起点晚多少秒(走到这里必然 ≥ 视频时长)
if last-first >= videoDuration {
// 弹幕跨度本身就装不进视频:平移也救不回来
lPrintWarnf("有 %d 条弹幕超出了录播视频的时长(%.1f 秒),已跳过", dropped, videoDuration)
return nil, offset
}
newOffset := -first
lPrintWarnf("弹幕时间基准和视频对不上:%d 条弹幕全部落在视频时长(%.1f 秒)之外,最早那条在 %.1f 秒;"+
"已按「第一条弹幕对齐视频起点」整体前移 %.1f 秒(偏移 %.1f → %.1f)",
len(records), videoDuration, late, late, offset, newOffset)
// 跨度装得进视频,所以平移之后不会再丢弹幕
kept, _ = clipDanmuToVideo(records, newOffset, videoDuration)
start, end := kept[0].Time+newOffset, kept[len(kept)-1].Time+newOffset
lPrintf("对齐后弹幕覆盖 %.1f ~ %.1f 秒,视频总长 %.1f 秒(尾部空 %.1f 秒);"+
"如果弹幕整体偏早或偏晚,在「偏移」里微调后重新合并",
start, end, videoDuration, videoDuration-end)
return kept, newOffset
}
// 合成一个 jsonl(和可选的视频)为带头像的弹幕层
// videoFile 为空时输出透明头像层(ProRes 4444),否则把弹幕层压进视频
func burnDanmu(jsonlFile, videoFile, outFile string, offset float64) error {
@@ -362,21 +420,13 @@ func burnDanmu(jsonlFile, videoFile, outFile string, offset float64) error {
return errors.New("找不到 ffmpeg,请先安装并加入 PATH,或把 ffmpeg 放在本程序所在文件夹,或在设置里填写 ffmpeg 路径")
}
// 弹幕可能录得比视频久(视频中途中断过),超出视频时长的部分没法渲染,直接丢掉
// 弹幕可能录得比视频久(视频中途中断过),超出视频时长的部分没法渲染,直接丢掉;
// 时间基准整体偏移的 .ass(见 fitDanmuToVideo)在这一步救回来
videoDuration := 0.0
if burnMode {
videoDuration = probeDuration(ffmpegFile, videoFile)
if videoDuration > 0 {
kept := records[:0]
for _, r := range records {
if r.Time+offset < videoDuration {
kept = append(kept, r)
}
}
if dropped := len(records) - len(kept); dropped > 0 {
lPrintWarnf("有 %d 条弹幕超出了录播视频的时长(%.1f 秒),已跳过", dropped, videoDuration)
}
records = kept
records, offset = fitDanmuToVideo(records, offset, videoDuration)
if len(records) == 0 {
return errNoDanmu
}
+21 -3
View File
@@ -278,7 +278,7 @@ func prepareDanmuSource(path string, write bool) (danmuSource, error) {
}
}
meta, records, err := parseAssFile(path)
meta, records, bad, err := parseAssFile(path)
if err != nil {
return danmuSource{}, err
}
@@ -289,6 +289,12 @@ func prepareDanmuSource(path string, write bool) (danmuSource, error) {
return src, nil
}
// 还原日志只在真正落盘这次打:页面的「检查」也会解析一遍同一个 .ass,
// 两处都打日志的话同一次还原会在日志里出现两遍
summary := fmt.Sprintf("已从 %s 还原出 %d 条弹幕(%dx%d,字号 %d,弹幕到 %s)",
filepath.Base(path), len(records), meta.PlayResX, meta.PlayResY, meta.FontSize,
formatSeconds(records[len(records)-1].Time))
if err := writeJsonlFile(jsonlPath, meta, records); err != nil {
// 目录只读等情况退到临时目录,不影响这次合并
tmp := filepath.Join(os.TempDir(), chunkTempPrefix+"converted.jsonl")
@@ -298,12 +304,22 @@ func prepareDanmuSource(path string, write bool) (danmuSource, error) {
lPrintWarnf("无法把还原结果写入 %s(%v),改用临时文件 %s", jsonlPath, err, tmp)
src.jsonlPath = tmp
src.tempJsonl = true
lPrintf("%s,写成 %s", summary, filepath.Base(tmp))
logAssBadLines(path, bad)
return src, nil
}
lPrintf("已把 %s 还原成 %s", filepath.Base(path), filepath.Base(jsonlPath))
lPrintf("%s,写成 %s", summary, filepath.Base(jsonlPath))
logAssBadLines(path, bad)
return src, nil
}
// .ass 里有行没能解析出来时提醒一次(只在真正落盘那次调,避免重复)
func logAssBadLines(path string, bad int) {
if bad > 0 {
lPrintWarnf("%s 里有 %d 行弹幕没能解析出来,已跳过", filepath.Base(path), bad)
}
}
// .ass 还原结果退到系统临时目录时,用完就删。只删标记过的临时文件:
// 正常情况 jsonlPath 是用户录播目录里的同名 jsonl,删了就是丢数据
func dropTempDanmu(src danmuSource) {
@@ -460,7 +476,9 @@ func buildMergePlan(req mergeRequest) (mergePlan, error) {
if info.Duration > 0 && src.last+info.Offset >= info.Duration {
info.Warnings = append(info.Warnings, fmt.Sprintf(
"最后一条弹幕在 %.1f 秒,加上偏移后已超出视频总长 %.1f 秒,超出的弹幕会被丢掉;"+
"如果这个视频只是整场的一段,请把同场的其它分段一起加进来",
"如果这个视频只是整场的一段,请把同场的其它分段一起加进来;"+
"如果弹幕整体都落在时长之外,通常是这个弹幕文件的时间基准和视频不一致"+
"(合成时会按第一条弹幕自动对齐,日志里有平移量),也可以自己填「偏移」",
src.last+info.Offset, info.Duration))
}
if info.Duration == 0 {
+4 -4
View File
@@ -163,7 +163,7 @@ func TestParseAssFilePlain(t *testing.T) {
dir := setupMergeTest(t)
path := writeTestFile(t, filepath.Join(dir, "2026-09-25 01-00-00 测试主播 标题.ass"), plainAssFixture)
meta, records, err := parseAssFile(path)
meta, records, _, err := parseAssFile(path)
if err != nil {
t.Fatalf("解析失败:%v", err)
}
@@ -203,7 +203,7 @@ func TestParseAssFileNick(t *testing.T) {
dir := setupMergeTest(t)
path := writeTestFile(t, filepath.Join(dir, "2026-09-25 01-00-00 测试主播 标题.nick.ass"), nickAssFixture)
meta, records, err := parseAssFile(path)
meta, records, _, err := parseAssFile(path)
if err != nil {
t.Fatalf("解析失败:%v", err)
}
@@ -238,7 +238,7 @@ func TestParseAssFileWithoutFormatLine(t *testing.T) {
dir := setupMergeTest(t)
path := writeTestFile(t, filepath.Join(dir, "noformat.ass"), noFormatAssFixture)
meta, records, err := parseAssFile(path)
meta, records, _, err := parseAssFile(path)
if err != nil {
t.Fatalf("解析失败:%v", err)
}
@@ -267,7 +267,7 @@ func TestParseAssFileRejectsBadFile(t *testing.T) {
}
for name, content := range cases {
path := writeTestFile(t, filepath.Join(dir, name), content)
if _, _, err := parseAssFile(path); err == nil {
if _, _, _, err := parseAssFile(path); err == nil {
t.Fatalf("%s 应该报错", name)
}
}
+1 -1
View File
@@ -175,7 +175,7 @@ func TestParseAssFileAcceptsBOM(t *testing.T) {
t.Fatal(err)
}
meta, records, err := parseAssFile(p)
meta, records, _, err := parseAssFile(p)
if err != nil {
t.Fatalf("带 BOM 的 ASS 应当能正常解析,实际报错:%v", err)
}
+30 -3
View File
@@ -8,12 +8,28 @@ import (
"os"
"path/filepath"
"strings"
"sync"
"testing"
"time"
"github.com/orzogc/acfundanmu"
)
// 等替换过的后台任务全部收工。用例里触发了 fire-and-forget 的合并时,必须在用例结束前等它们跑完
func waitBackground(t *testing.T, bg *sync.WaitGroup, timeout time.Duration) {
t.Helper()
done := make(chan struct{})
go func() {
bg.Wait()
close(done)
}()
select {
case <-done:
case <-time.After(timeout):
t.Error("等后台任务收工超时,它可能还在读全局配置")
}
}
// 用真正的录制写入器(danmuWriter)产生 jsonl,再按合并的流程重排一遍,
// 验证「本程序自己录下来的弹幕」合出来和以前一模一样(弹道、宽度都不变)。
// 这也是手动合并与自动合并的分水岭:录制侧写文件完全不经过 relayoutLanes,
@@ -183,12 +199,23 @@ func TestDanmuWriteNoFalseAlarm(t *testing.T) {
// 弹幕文件写坏之后必须跳过自动合并:只是打一条日志的话,
// 照样会合成一个「缺了一大截弹幕」的成片,而且是静默的
func TestFinishPartSkipsMergeWhenDanmuBroken(t *testing.T) {
// 这个用例会真的触发自动合并(safeGo 起的 goroutine)。它跑到用例结束之后还会读全局配置,
// 而下一个用例开头就改配置 → -race 会判成数据竞争。所以这里把后台任务换成可等待的版本,
// 清理时先等它收工再还原全局变量
var bg sync.WaitGroup
oldRun := runBackground
runBackground = func(f func()) {
bg.Add(1)
go func() {
defer bg.Done()
f()
}()
}
oldAuto, oldCtx := config.AutoMerge, mainCtx
t.Cleanup(func() {
waitBackground(t, &bg, 60*time.Second)
config.AutoMerge, mainCtx = oldAuto, oldCtx
// 等后台那两个(可能启动了的)合并跑完,别把任务状态留给后面的用例
takeMergeSlot()
releaseMergeSlot()
runBackground = oldRun
})
config.AutoMerge = true
mainCtx = context.Background()
+7 -2
View File
@@ -107,17 +107,22 @@ func checkErr(err error) {
}
}
// 后台任务真正怎么起。单独抽成一个变量是为了让测试能换成「可等待」的版本:
// fire-and-forget 的 goroutine 漏到测试结束之后,会在下一个测试改全局配置时读它们,
// -race 会判成数据竞争
var runBackground = func(f func()) { go f() }
// 后台任务的统一入口:兜底 recover,避免单个任务的 panic 终止整个进程
// name 用于日志里指出是哪个任务出错,f 里可以继续做自己的 recover
func safeGo(name string, f func()) {
go func() {
runBackground(func() {
defer func() {
if err := recover(); err != nil {
lPrintErrf("%s 发生错误并已恢复,其它录制任务不受影响:%v", name, err)
}
}()
f()
}()
})
}
// 不需要重试的错误:再试一遍也只是重复同样的失败(比如头像本身就超出体积/尺寸上限),
+219
View File
@@ -0,0 +1,219 @@
package main
import (
"bytes"
"fmt"
"os"
"os/exec"
"path/filepath"
"strings"
"testing"
)
// 造 n 条弹幕:从 start 秒开始,每条间隔 step
func mkDanmuForFit(start, step float64, n int) []danmuRecord {
out := make([]danmuRecord, 0, n)
for i := 0; i < n; i++ {
out = append(out, danmuRecord{
Time: start + float64(i)*step, UID: int64(100 + i), Name: "甲",
Content: fmt.Sprintf("第 %d 条", i), Width: 100,
})
}
return out
}
// 正常场次:时间轴一点都不该动
func TestFitDanmuToVideoLeavesNormalAlone(t *testing.T) {
records := mkDanmuForFit(1, 1, 10) // 1 ~ 10 秒
got, off := fitDanmuToVideo(records, 0, 60)
if len(got) != 10 || off != 0 {
t.Fatalf("正常场次不该动:留 %d 条、偏移 %v", len(got), off)
}
}
// 弹幕确实录得比视频久(视频中途断过):只丢尾部超出的,时间轴不动
func TestFitDanmuToVideoDropsTailOnly(t *testing.T) {
records := mkDanmuForFit(1, 5, 20) // 1 ~ 96 秒,视频只有 50 秒
got, off := fitDanmuToVideo(records, 0, 50)
if off != 0 {
t.Fatalf("只该丢尾部,偏移应保持 0,实际 %v", off)
}
if len(got) == 0 || len(got) == len(records) {
t.Fatalf("应该留一部分、丢一部分,实际留 %d 条", len(got))
}
if last := got[len(got)-1].Time; last >= 50 {
t.Fatalf("留下的最后一条应在视频内,实际 %v", last)
}
}
// 时间基准整体偏移的 .ass(时间相对程序启动、不是录制开始):按第一条弹幕对齐视频起点
func TestFitDanmuToVideoAlignsShiftedTimeBase(t *testing.T) {
// 现实素材的形状:7849.92 ~ 13286.59 秒 / 视频 5456.883 秒
records := mkDanmuForFit(7849.92, 5.7, 955)
got, off := fitDanmuToVideo(records, 0, 5456.883)
if off != -7849.92 {
t.Fatalf("偏移应为 -7849.92,实际 %v", off)
}
if len(got) != len(records) {
t.Fatalf("跨度装得进视频,不该丢弹幕,实际留 %d / %d 条", len(got), len(records))
}
if first := got[0].Time + off; first != 0 {
t.Fatalf("第一条应落在视频起点,实际 %v", first)
}
if end := got[len(got)-1].Time + off; end >= 5456.883 {
t.Fatalf("最后一条应在视频内,实际 %v", end)
}
}
// 边界:跨度刚好比视频短一点点,对齐后一条都不该丢
func TestFitDanmuToVideoAlignJustFits(t *testing.T) {
records := mkDanmuForFit(1000, 1, 1000) // 1000 ~ 1999 秒,跨度 999 秒
got, off := fitDanmuToVideo(records, 0, 1000)
if off != -1000 {
t.Fatalf("偏移应为 -1000,实际 %v", off)
}
if len(got) != len(records) {
t.Fatalf("跨度装得进视频,不该丢弹幕,实际留 %d / %d 条", len(got), len(records))
}
if end := got[len(got)-1].Time + off; end >= 1000 {
t.Fatalf("最后一条应在视频内,实际 %v", end)
}
}
// 弹幕跨度本身就装不进视频:平移救不回来,只能按原行为一条不留
func TestFitDanmuToVideoGivesUpWhenSpanTooLong(t *testing.T) {
records := mkDanmuForFit(7849.92, 10, 100) // 跨度 990 秒 > 视频 500 秒
got, off := fitDanmuToVideo(records, 0, 500)
if len(got) != 0 {
t.Fatalf("跨度装不下时应一条不留,实际 %d 条", len(got))
}
if off != 0 {
t.Fatalf("放弃时不该改偏移,实际 %v", off)
}
}
// 偏移把弹幕整体推到视频之前时保持原行为(这些条目不算「超出时长」,由渲染那边处理)
func TestFitDanmuToVideoKeepsAllBeforeVideoBehaviour(t *testing.T) {
records := mkDanmuForFit(100, 1, 10)
got, off := fitDanmuToVideo(records, -1000, 500)
if len(got) != len(records) || off != -1000 {
t.Fatalf("整体在视频之前时不该动:留 %d 条、偏移 %v", len(got), off)
}
}
// 用户实测的那种场次:.ass 时间基准比视频晚一大截,合并要能成功并且真的画上弹幕
func TestManualMergeFromShiftedAssEndToEnd(t *testing.T) {
dir := setupMergeTest(t)
ffmpeg := getFFmpeg()
if ffmpeg == "" {
t.Skip("没有 ffmpeg")
}
name := "2026-09-25 02-00-00 测试主播 时间基准偏移"
video := filepath.Join(recordSaveDir(), name+".mp4")
cmd := exec.Command(ffmpeg, "-v", "error", "-y",
"-f", "lavfi", "-i", "color=c=black:size=640x360:rate=25", "-t", "8",
"-c:v", "libx264", "-preset", "ultrafast", "-pix_fmt", "yuv420p", video)
if out, err := cmd.CombinedOutput(); err != nil {
t.Fatalf("生成测试视频失败:%v\n%s", err, out)
}
// 弹幕时间整体比录制起点晚 2 小时(模拟「程序先开着、直播中途才开始录」)
ass := fmt.Sprintf(`[Script Info]
Title: %s
ScriptType: v4.00+
Collisions: Normal
PlayResX: 640
PlayResY: 360
[V4+ Styles]
Format: Name, Fontname, Fontsize, PrimaryColour, SecondaryColour, OutlineColour, BackColour, Bold, Italic, Underline, StrikeOut, ScaleX, ScaleY, Spacing, Angle, BorderStyle, Outline, Shadow, Alignment, MarginL, MarginR, MarginV, Encoding
Style: Danmu,Microsoft YaHei,40,&H00FFFFFF,&H00FFFFFF,&H00000000,&H00000000,0,0,0,0,100,100,0,0,1,1,0,2,20,20,2,0
[Events]
Format: Layer, Start, End, Style, Name, MarginL, MarginR, MarginV, Effect, Text
Dialogue: 0,2:00:01.00,2:00:11.00,Danmu,甲(101),20,20,2,,{\move(400,40,-40,40)}第一条弹幕
Dialogue: 0,2:00:03.00,2:00:13.00,Danmu,乙(202),20,20,2,,{\move(400,80,-40,80)}第二条弹幕
Dialogue: 0,2:00:05.00,2:00:15.00,Danmu,甲(101),20,20,2,,{\move(400,120,-40,120)}第三条弹幕
`, name)
assPath := writeTestFile(t, filepath.Join(recordSaveDir(), name+".ass"), ass)
if _, err := prepareDanmuSource(assPath, true); err != nil {
t.Fatalf("还原弹幕失败:%v", err)
}
plan, err := buildMergePlan(mergeRequest{Videos: []string{video}, Danmu: assPath})
if err != nil {
t.Fatalf("检查失败:%v", err)
}
var buf bytes.Buffer
oldOut := logger.Writer()
logger.SetOutput(&buf)
err = execManualMerge(plan)
logger.SetOutput(oldOut)
if err != nil {
t.Fatalf("时间基准偏移的场次应当能合出来,实际失败:%v\n日志:\n%s", err, buf.String())
}
log := buf.String()
if !strings.Contains(log, "按「第一条弹幕对齐视频起点」") {
t.Fatalf("日志里应写明自动对齐:\n%s", log)
}
if !strings.Contains(log, "对齐后弹幕覆盖") {
t.Fatalf("日志里应写明对齐后的覆盖范围:\n%s", log)
}
if _, err := os.Stat(plan.out); err != nil {
t.Fatalf("没有生成成片:%v", err)
}
// 成片里确实画了弹幕:同一时刻抽帧,有字的帧 PNG 一定比纯黑源片大
shot := func(src, tag string) int {
out := filepath.Join(dir, "shot_"+tag+".png")
c := exec.Command(ffmpeg, "-v", "error", "-y", "-ss", "3", "-i", src,
"-frames:v", "1", "-update", "1", out)
if o, err := c.CombinedOutput(); err != nil {
t.Fatalf("抽帧失败:%v\n%s", err, o)
}
fi, err := os.Stat(out)
if err != nil {
t.Fatal(err)
}
return int(fi.Size())
}
srcSize, outSize := shot(video, "src"), shot(plan.out, "out")
if outSize <= srcSize {
t.Fatalf("成片的帧没有比纯黑源片大,弹幕很可能没画上:源 %d 字节、成片 %d 字节", srcSize, outSize)
}
t.Logf("抽帧大小:源 %d 字节、成片 %d 字节", srcSize, outSize)
}
// 同一次 .ass 还原只该在日志里出现一条(页面的「检查」和合并都会解析一遍)
func TestPrepareDanmuSourceLogsRestoreOnce(t *testing.T) {
setupMergeTest(t)
path := writeTestFile(t,
filepath.Join(recordSaveDir(), "2026-09-25 03-00-00 测试主播 标题.ass"), plainAssFixture)
var buf bytes.Buffer
oldOut := logger.Writer()
logger.SetOutput(&buf)
defer logger.SetOutput(oldOut)
// 「检查」先解析一遍(不落盘),合并再解析一遍并落盘
if _, err := prepareDanmuSource(path, false); err != nil {
t.Fatalf("检查失败:%v", err)
}
if _, err := prepareDanmuSource(path, true); err != nil {
t.Fatalf("还原失败:%v", err)
}
log := buf.String()
if n := strings.Count(log, "还原出"); n != 1 {
t.Fatalf("「还原出」应只出现 1 次,实际 %d 次:\n%s", n, log)
}
if n := strings.Count(log, "还原成"); n != 0 {
t.Fatalf("不该再有「还原成」这条日志,实际 %d 次:\n%s", n, log)
}
if !strings.Contains(log, filepath.Base(path)) {
t.Fatalf("日志里应写明是哪个文件:\n%s", log)
}
}