ChineseSubFinder/internal/types/subparser/fileinfo.go

package subparser

import (
	"github.com/allanpk716/ChineseSubFinder/internal/common"
	"github.com/allanpk716/ChineseSubFinder/internal/pkg/my_util"
	"github.com/allanpk716/ChineseSubFinder/internal/types/language"
	"math"
	"sort"
	"time"
)

type FileInfo struct {
	Content           string              // 字幕的内容
	FromWhereSite     string              // 从那个网站下载的
	Name              string              // 字幕的名称，注意，这里需要额外的赋值，不会自动检测
	Ext               string              // 字幕的后缀名
	Lang              language.MyLanguage // 识别出来的语言
	FileFullPath      string              // 字幕文件的全路径
	Data              []byte              // 字幕的二进制文件内容
	Dialogues         []OneDialogue       // 整个字幕文件的所有对话，如果是做时间轴匹配，就使用原始的
	DialoguesFilter   []OneDialogue       // 整个字幕文件的所有对话，过滤掉特殊字符的对白
	DialoguesFilterEx []OneDialogueEx     // 整个字幕文件的所有对话，过滤掉特殊字符的对白，这里会把一句话中支持的 中、英、韩、日 四国语言给分离出来
	CHLines           []string            // 抽取出所有的中文对话
	OtherLines        []string            // 抽取出所有的第二语言对话，可能是英文、韩文、日文
}

// SortDialogues 排序对话，时间递减
func (f *FileInfo) SortDialogues() {
	sort.Sort(OneDialogueByStartTime(f.Dialogues))
	sort.Sort(OneDialogueByStartTime(f.DialoguesFilter))
	sort.Sort(OneDialogueByStartTimeEx(f.DialoguesFilterEx))
}

// GetTimeFormat 获取时间轴的格式化格式
func (f FileInfo) GetTimeFormat() string {
	if f.Ext == common.SubExtASS || f.Ext == common.SubExtSSA {
		return common.TimeFormatPoint2
	} else {
		return common.TimeFormatPoint3
	}
}

// GetDialogueExContent 获取当前字幕文件语言对应索引的对白内容
// 凡是带有 Eng 的返回 Eng，其他的就与对应语言相关
func (f FileInfo) GetDialogueExContent(index int) string {

	switch f.Lang {
	case language.ChineseSimple, language.ChineseTraditional,
		language.ChineseSimpleJapanese, language.ChineseSimpleKorean,
		language.ChineseTraditionalJapanese, language.ChineseTraditionalKorean:
		// 带有中文的，但是又不是中英的
		return f.DialoguesFilterEx[index].ChLine
	case language.English, language.ChineseSimpleEnglish, language.ChineseTraditionalEnglish:
		return f.DialoguesFilterEx[index].EnLine
	case language.Japanese:
		return f.DialoguesFilterEx[index].JpLine
	case language.Korean:
		return f.DialoguesFilterEx[index].KrLine
	default:
		return f.DialoguesFilterEx[index].EnLine
	}
}

// ChangeDialoguesTimeByFramerateRatio 根据帧数比率调整时间轴 对应 ffsubsync -- SubtitleScaler
func (f *FileInfo) ChangeDialoguesTimeByFramerateRatio(framerateRatio float64) error {

	timeFormat := f.GetTimeFormat()
	f.changeOneDialoguesFramerateRatio(f.Dialogues, framerateRatio, timeFormat)
	f.changeOneDialoguesFramerateRatio(f.DialoguesFilter, framerateRatio, timeFormat)
	f.changeOneDialogueExsFramerateRatio(f.DialoguesFilterEx, framerateRatio, timeFormat)

	return nil
}

func (f *FileInfo) changeOneDialoguesFramerateRatio(oneDialogues []OneDialogue, framerateRatio float64, timeFormat string) {
	for i := 0; i < len(oneDialogues); i++ {

		timeStart := oneDialogues[i].GetStartTime()
		timeEnd := oneDialogues[i].GetEndTime()
		timeStartNumber := my_util.Time2SecondNumber(timeStart)
		timeEndNumber := my_util.Time2SecondNumber(timeEnd)

		scaleTimeStart := my_util.TimeNumber2Time(timeStartNumber * framerateRatio)
		scaleTimeEnd := my_util.TimeNumber2Time(timeEndNumber * framerateRatio)

		oneDialogues[i].StartTime = my_util.Time2SubTimeString(scaleTimeStart, timeFormat)
		oneDialogues[i].EndTime = my_util.Time2SubTimeString(scaleTimeEnd, timeFormat)
	}
}

func (f *FileInfo) changeOneDialogueExsFramerateRatio(oneDialogues []OneDialogueEx, framerateRatio float64, timeFormat string) {
	for i := 0; i < len(oneDialogues); i++ {

		timeStart := oneDialogues[i].GetStartTime()
		timeEnd := oneDialogues[i].GetEndTime()
		timeStartNumber := my_util.Time2SecondNumber(timeStart)
		timeEndNumber := my_util.Time2SecondNumber(timeEnd)

		scaleTimeStart := my_util.TimeNumber2Time(timeStartNumber * framerateRatio)
		scaleTimeEnd := my_util.TimeNumber2Time(timeEndNumber * framerateRatio)

		oneDialogues[i].StartTime = my_util.Time2SubTimeString(scaleTimeStart, timeFormat)
		oneDialogues[i].EndTime = my_util.Time2SubTimeString(scaleTimeEnd, timeFormat)
	}
}

// GetStartTime 获取的是从 Dialogues 得到的
func (f FileInfo) GetStartTime() time.Time {
	startTime := math.MaxFloat64
	for i := 0; i < len(f.Dialogues); i++ {
		// 找到最小的开始时间
		tmpNowStartTimeNumber := my_util.Time2SecondNumber(f.Dialogues[i].GetStartTime())
		startTime = math.Min(startTime, tmpNowStartTimeNumber)
	}
	return my_util.TimeNumber2Time(startTime)
}

// GetEndTime 获取的是从 Dialogues 得到的
func (f FileInfo) GetEndTime() time.Time {
	endTime := -math.MaxFloat64
	for i := 0; i < len(f.Dialogues); i++ {
		// 找到最大的结束时间
		tmpNowEndTimeNumber := my_util.Time2SecondNumber(f.Dialogues[i].GetEndTime())
		endTime = math.Max(endTime, tmpNowEndTimeNumber)
	}
	return my_util.TimeNumber2Time(endTime)
}

// GetNumFrames 获取这个字幕的时间 Frame 数量
func (f FileInfo) GetNumFrames() int {

	return int(math.Abs((my_util.Time2SecondNumber(f.GetEndTime()) - my_util.Time2SecondNumber(f.GetStartTime())) * 100))
}

// OneDialogue 一句对话
type OneDialogue struct {
	StartTime string   // 开始时间
	EndTime   string   // 结束时间
	StyleName string   // StyleName
	Lines     []string // 台词
}

func (o OneDialogue) GetStartTime() time.Time {
	srcTimeStartNow, err := my_util.ParseTime(o.StartTime)
	if err != nil {
		return time.Time{}
	}
	return srcTimeStartNow
}

func (o OneDialogue) GetEndTime() time.Time {
	srcTimeEndNow, err := my_util.ParseTime(o.EndTime)
	if err != nil {
		return time.Time{}
	}
	return srcTimeEndNow
}

type OneDialogueByStartTime []OneDialogue

func (d OneDialogueByStartTime) Len() int {
	return len(d)
}

func (d OneDialogueByStartTime) Swap(i, j int) {
	d[i], d[j] = d[j], d[i]
}

func (d OneDialogueByStartTime) Less(i, j int) bool {

	subStartTimeI, err := my_util.ParseTime(d[i].StartTime)
	if err != nil {
		return false
	}
	subStartTimeJ, err := my_util.ParseTime(d[j].StartTime)
	if err != nil {
		return false
	}
	return my_util.Time2SecondNumber(subStartTimeI) < my_util.Time2SecondNumber(subStartTimeJ)
}

// OneDialogueEx 一句对话，这里会把一句话中支持的 中、英、韩、日 四国语言给分离出来
type OneDialogueEx struct {
	StartTime string // 开始时间
	EndTime   string // 结束时间
	ChLine    string
	EnLine    string
	KrLine    string
	JpLine    string
}

func (o OneDialogueEx) GetStartTime() time.Time {
	srcTimeStartNow, err := my_util.ParseTime(o.StartTime)
	if err != nil {
		return time.Time{}
	}
	return srcTimeStartNow
}

func (o OneDialogueEx) GetEndTime() time.Time {
	srcTimeEndNow, err := my_util.ParseTime(o.EndTime)
	if err != nil {
		return time.Time{}
	}
	return srcTimeEndNow
}

type OneDialogueByStartTimeEx []OneDialogueEx

func (d OneDialogueByStartTimeEx) Len() int {
	return len(d)
}

func (d OneDialogueByStartTimeEx) Swap(i, j int) {
	d[i], d[j] = d[j], d[i]
}

func (d OneDialogueByStartTimeEx) Less(i, j int) bool {

	subStartTimeI, err := my_util.ParseTime(d[i].StartTime)
	if err != nil {
		return false
	}
	subStartTimeJ, err := my_util.ParseTime(d[j].StartTime)
	if err != nil {
		return false
	}
	return my_util.Time2SecondNumber(subStartTimeI) < my_util.Time2SecondNumber(subStartTimeJ)
}

const (
	Sub_Ext_Mark_Default = ".default" // 指定这个字幕是默认的
	Sub_Ext_Mark_Forced  = ".forced"  // 指定这个字幕是强制的
)