sub_helper.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596
  1. package sub_helper
  2. import (
  3. "errors"
  4. "github.com/allanpk716/ChineseSubFinder/internal/common"
  5. "github.com/allanpk716/ChineseSubFinder/internal/pkg/archive_helper"
  6. "github.com/allanpk716/ChineseSubFinder/internal/pkg/decode"
  7. "github.com/allanpk716/ChineseSubFinder/internal/pkg/language"
  8. "github.com/allanpk716/ChineseSubFinder/internal/pkg/log_helper"
  9. "github.com/allanpk716/ChineseSubFinder/internal/pkg/my_util"
  10. "github.com/allanpk716/ChineseSubFinder/internal/pkg/regex_things"
  11. "github.com/allanpk716/ChineseSubFinder/internal/pkg/sub_parser_hub"
  12. "github.com/allanpk716/ChineseSubFinder/internal/pkg/vad"
  13. "github.com/allanpk716/ChineseSubFinder/internal/types/subparser"
  14. "github.com/allanpk716/ChineseSubFinder/internal/types/supplier"
  15. "math"
  16. "os"
  17. "path/filepath"
  18. "strconv"
  19. "strings"
  20. "time"
  21. )
  22. // OrganizeDlSubFiles 需要从汇总来是网站字幕中,解压对应的压缩包中的字幕出来
  23. func OrganizeDlSubFiles(tmpFolderName string, subInfos []supplier.SubInfo) (map[string][]string, error) {
  24. // 缓存列表,整理后的字幕列表
  25. // SxEx - []string 字幕的路径
  26. var siteSubInfoDict = make(map[string][]string)
  27. tmpFolderFullPath, err := my_util.GetTmpFolderByName(tmpFolderName)
  28. if err != nil {
  29. return nil, err
  30. }
  31. // 把后缀名给改好
  32. ChangeVideoExt2SubExt(subInfos)
  33. // 第三方的解压库,首先不支持 io.Reader 的操作,也就是得缓存到本地硬盘再读取解压
  34. // 且使用 walk 会无法解压 rar,得指定具体的实例,太麻烦了,直接用通用的接口得了,就是得都缓存下来再判断
  35. // 基于以上两点,写了一堆啰嗦的逻辑···
  36. for i := range subInfos {
  37. // 先存下来,保存是时候需要前缀,前缀就是从那个网站下载来的
  38. nowFileSaveFullPath := filepath.Join(tmpFolderFullPath, GetFrontNameAndOrgName(&subInfos[i]))
  39. err = my_util.WriteFile(nowFileSaveFullPath, subInfos[i].Data)
  40. if err != nil {
  41. log_helper.GetLogger().Errorln("getFrontNameAndOrgName - WriteFile", subInfos[i].FromWhere, subInfos[i].Name, subInfos[i].TopN, err)
  42. continue
  43. }
  44. nowExt := strings.ToLower(subInfos[i].Ext)
  45. epsKey := my_util.GetEpisodeKeyName(subInfos[i].Season, subInfos[i].Episode)
  46. _, ok := siteSubInfoDict[epsKey]
  47. if ok == false {
  48. // 不存在则实例化
  49. siteSubInfoDict[epsKey] = make([]string, 0)
  50. }
  51. if nowExt != ".zip" && nowExt != ".tar" && nowExt != ".rar" && nowExt != ".7z" {
  52. // 是否是受支持的字幕类型
  53. if sub_parser_hub.IsSubExtWanted(nowExt) == false {
  54. log_helper.GetLogger().Debugln("OrganizeDlSubFiles -> IsSubExtWanted == false", "Name:", subInfos[i].Name, "FileUrl:", subInfos[i].FileUrl)
  55. continue
  56. }
  57. // 加入缓存列表
  58. siteSubInfoDict[epsKey] = append(siteSubInfoDict[epsKey], nowFileSaveFullPath)
  59. } else {
  60. // 那么就是需要解压的文件了
  61. // 解压,给一个单独的文件夹
  62. unzipTmpFolder := filepath.Join(tmpFolderFullPath, subInfos[i].FromWhere)
  63. err = os.MkdirAll(unzipTmpFolder, os.ModePerm)
  64. if err != nil {
  65. return nil, err
  66. }
  67. err = archive_helper.UnArchiveFile(nowFileSaveFullPath, unzipTmpFolder)
  68. // 解压完成后,遍历受支持的字幕列表,加入缓存列表
  69. if err != nil {
  70. log_helper.GetLogger().Errorln("archiver.UnArchive", subInfos[i].FromWhere, subInfos[i].Name, subInfos[i].TopN, err)
  71. continue
  72. }
  73. // 搜索这个目录下的所有符合字幕格式的文件
  74. subFileFullPaths, err := SearchMatchedSubFileByDir(unzipTmpFolder)
  75. if err != nil {
  76. log_helper.GetLogger().Errorln("searchMatchedSubFile", subInfos[i].FromWhere, subInfos[i].Name, subInfos[i].TopN, err)
  77. continue
  78. }
  79. // 这里需要给这些下载到的文件进行改名,加是从那个网站来的前缀,后续好查找
  80. for _, fileFullPath := range subFileFullPaths {
  81. newSubName := AddFrontName(subInfos[i], filepath.Base(fileFullPath))
  82. newSubNameFullPath := filepath.Join(tmpFolderFullPath, newSubName)
  83. // 改名
  84. err = os.Rename(fileFullPath, newSubNameFullPath)
  85. if err != nil {
  86. log_helper.GetLogger().Errorln("os.Rename", subInfos[i].FromWhere, subInfos[i].Name, subInfos[i].TopN, err)
  87. continue
  88. }
  89. // 加入缓存列表
  90. siteSubInfoDict[epsKey] = append(siteSubInfoDict[epsKey], newSubNameFullPath)
  91. }
  92. }
  93. }
  94. return siteSubInfoDict, nil
  95. }
  96. // ChangeVideoExt2SubExt 检测 Name,如果是视频的后缀名就改为字幕的后缀名
  97. func ChangeVideoExt2SubExt(subInfos []supplier.SubInfo) {
  98. for x, info := range subInfos {
  99. tmpSubFileName := info.Name
  100. // 如果后缀名是下载字幕目标的后缀名 或者 是压缩包格式的,则跳过
  101. if strings.Contains(tmpSubFileName, info.Ext) == true || archive_helper.IsWantedArchiveExtName(tmpSubFileName) == true {
  102. } else {
  103. subInfos[x].Name = tmpSubFileName + info.Ext
  104. }
  105. }
  106. }
  107. // SelectChineseBestBilingualSubtitle 找到合适的双语中文字幕,简体->繁体,以及 字幕类型的优先级选择
  108. func SelectChineseBestBilingualSubtitle(subs []subparser.FileInfo, subTypePriority int) *subparser.FileInfo {
  109. // 先傻一点实现优先双语的,之前的写法有 bug
  110. for _, info := range subs {
  111. // 找到了中文字幕
  112. if language.HasChineseLang(info.Lang) == true {
  113. // 字幕的优先级 0 - 原样, 1 - srt , 2 - ass/ssa
  114. if subTypePriority == 1 {
  115. // 1 - srt
  116. if strings.ToLower(info.Ext) == common.SubExtSRT {
  117. // 优先双语
  118. if language.IsBilingualSubtitle(info.Lang) == true {
  119. return &info
  120. }
  121. }
  122. } else if subTypePriority == 2 {
  123. // 2 - ass/ssa
  124. if strings.ToLower(info.Ext) == common.SubExtASS || strings.ToLower(info.Ext) == common.SubExtSSA {
  125. // 优先双语
  126. if language.IsBilingualSubtitle(info.Lang) == true {
  127. return &info
  128. }
  129. }
  130. } else {
  131. // 优先双语
  132. if language.IsBilingualSubtitle(info.Lang) == true {
  133. return &info
  134. }
  135. }
  136. }
  137. }
  138. return nil
  139. }
  140. // SelectChineseBestSubtitle 找到合适的中文字幕,简体->繁体,以及 字幕类型的优先级选择
  141. func SelectChineseBestSubtitle(subs []subparser.FileInfo, subTypePriority int) *subparser.FileInfo {
  142. // 先傻一点实现优先双语的,之前的写法有 bug
  143. for _, info := range subs {
  144. // 找到了中文字幕
  145. if language.HasChineseLang(info.Lang) == true {
  146. // 字幕的优先级 0 - 原样, 1 - srt , 2 - ass/ssa
  147. if subTypePriority == 1 {
  148. // 1 - srt
  149. if strings.ToLower(info.Ext) == common.SubExtSRT {
  150. return &info
  151. }
  152. } else if subTypePriority == 2 {
  153. // 2 - ass/ssa
  154. if strings.ToLower(info.Ext) == common.SubExtASS || strings.ToLower(info.Ext) == common.SubExtSSA {
  155. return &info
  156. }
  157. } else {
  158. return &info
  159. }
  160. }
  161. }
  162. return nil
  163. }
  164. // GetFrontNameAndOrgName 返回的名称包含,那个网站下载的,这个网站中排名第几,文件名
  165. func GetFrontNameAndOrgName(info *supplier.SubInfo) string {
  166. infoName := ""
  167. fileName, err := decode.GetVideoInfoFromFileName(info.Name)
  168. if err != nil {
  169. log_helper.GetLogger().Warnln("", err)
  170. infoName = info.Name
  171. } else {
  172. infoName = fileName.Title + "_S" + strconv.Itoa(fileName.Season) + "E" + strconv.Itoa(fileName.Episode) + filepath.Ext(info.Name)
  173. }
  174. info.Name = infoName
  175. return "[" + info.FromWhere + "]_" + strconv.FormatInt(info.TopN, 10) + "_" + infoName
  176. }
  177. // AddFrontName 添加文件的前缀
  178. func AddFrontName(info supplier.SubInfo, orgName string) string {
  179. return "[" + info.FromWhere + "]_" + strconv.FormatInt(info.TopN, 10) + "_" + orgName
  180. }
  181. // SearchMatchedSubFileByDir 搜索符合后缀名的视频文件,排除 Sub_SxE0 这样的文件夹中的文件
  182. func SearchMatchedSubFileByDir(dir string) ([]string, error) {
  183. // 这里有个梗,会出现 __MACOSX 这类文件夹,那么里面会有一样的文件,需要用文件大小排除一下,至少大于 1 kb 吧
  184. var fileFullPathList = make([]string, 0)
  185. pathSep := string(os.PathSeparator)
  186. files, err := os.ReadDir(dir)
  187. if err != nil {
  188. return nil, err
  189. }
  190. for _, curFile := range files {
  191. fullPath := dir + pathSep + curFile.Name()
  192. if my_util.IsDir(fullPath) == true {
  193. // 需要排除 Sub_S1E0、Sub_S2E0 这样的整季的字幕文件夹,这里仅仅是缓存,不会被加载的
  194. matched := regex_things.RegOneSeasonSubFolderNameMatch.FindAllStringSubmatch(curFile.Name(), -1)
  195. if matched != nil && len(matched) > 0 {
  196. continue
  197. }
  198. // 内层的错误就无视了
  199. oneList, _ := SearchMatchedSubFileByDir(fullPath)
  200. if oneList != nil {
  201. fileFullPathList = append(fileFullPathList, oneList...)
  202. }
  203. } else {
  204. // 这里就是文件了
  205. info, err := curFile.Info()
  206. if err != nil {
  207. return nil, err
  208. }
  209. if info.Size() < 1000 {
  210. continue
  211. }
  212. if sub_parser_hub.IsSubExtWanted(filepath.Ext(curFile.Name())) == true {
  213. fileFullPathList = append(fileFullPathList, fullPath)
  214. }
  215. }
  216. }
  217. return fileFullPathList, nil
  218. }
  219. // SearchMatchedSubFileByOneVideo 搜索这个视频当前目录下匹配的字幕
  220. func SearchMatchedSubFileByOneVideo(oneVideoFullPath string) ([]string, error) {
  221. dir := filepath.Dir(oneVideoFullPath)
  222. fileName := filepath.Base(oneVideoFullPath)
  223. fileName = strings.ToLower(fileName)
  224. fileName = strings.ReplaceAll(fileName, filepath.Ext(fileName), "")
  225. pathSep := string(os.PathSeparator)
  226. files, err := os.ReadDir(dir)
  227. if err != nil {
  228. return nil, err
  229. }
  230. var matchedSubs = make([]string, 0)
  231. for _, curFile := range files {
  232. if curFile.IsDir() {
  233. continue
  234. }
  235. // 这里就是文件了
  236. info, err := curFile.Info()
  237. if err != nil {
  238. return nil, err
  239. }
  240. if info.Size() < 1000 {
  241. continue
  242. }
  243. // 判断的时候用小写的,后续重命名的时候用原有的名称
  244. nowFileName := strings.ToLower(curFile.Name())
  245. // 后缀名得对
  246. if sub_parser_hub.IsSubExtWanted(filepath.Ext(nowFileName)) == false {
  247. continue
  248. }
  249. // 字幕文件名应该包含 视频文件名(无后缀)
  250. if strings.Contains(nowFileName, fileName) == false {
  251. continue
  252. }
  253. oldPath := dir + pathSep + curFile.Name()
  254. matchedSubs = append(matchedSubs, oldPath)
  255. }
  256. return matchedSubs, nil
  257. }
  258. // SearchVideoMatchSubFileAndRemoveExtMark 找到找个视频目录下相匹配的字幕,同时去除这些字幕中 .default 或者 .forced 的标记。注意这两个标记不应该同时出现,否则无法正确去除
  259. func SearchVideoMatchSubFileAndRemoveExtMark(oneVideoFullPath string) error {
  260. dir := filepath.Dir(oneVideoFullPath)
  261. fileName := filepath.Base(oneVideoFullPath)
  262. fileName = strings.ToLower(fileName)
  263. fileName = strings.ReplaceAll(fileName, filepath.Ext(fileName), "")
  264. pathSep := string(os.PathSeparator)
  265. files, err := os.ReadDir(dir)
  266. if err != nil {
  267. return err
  268. }
  269. for _, curFile := range files {
  270. if curFile.IsDir() {
  271. continue
  272. } else {
  273. // 这里就是文件了
  274. info, err := curFile.Info()
  275. if err != nil {
  276. return err
  277. }
  278. if info.Size() < 1000 {
  279. continue
  280. }
  281. // 判断的时候用小写的,后续重命名的时候用原有的名称
  282. nowFileName := strings.ToLower(curFile.Name())
  283. // 后缀名得对
  284. if sub_parser_hub.IsSubExtWanted(filepath.Ext(nowFileName)) == false {
  285. continue
  286. }
  287. // 字幕文件名应该包含 视频文件名(无后缀)
  288. if strings.Contains(nowFileName, fileName) == false {
  289. continue
  290. }
  291. // 得包含 .default. 找个关键词
  292. if strings.Contains(nowFileName, subparser.Sub_Ext_Mark_Default+".") == true {
  293. oldPath := dir + pathSep + curFile.Name()
  294. newPath := dir + pathSep + strings.ReplaceAll(curFile.Name(), subparser.Sub_Ext_Mark_Default+".", ".")
  295. err = os.Rename(oldPath, newPath)
  296. if err != nil {
  297. return err
  298. }
  299. } else if strings.Contains(nowFileName, subparser.Sub_Ext_Mark_Forced+".") == true {
  300. // 得包含 .forced. 找个关键词
  301. oldPath := dir + pathSep + curFile.Name()
  302. newPath := dir + pathSep + strings.ReplaceAll(curFile.Name(), subparser.Sub_Ext_Mark_Forced+".", ".")
  303. err = os.Rename(oldPath, newPath)
  304. if err != nil {
  305. return err
  306. }
  307. } else {
  308. continue
  309. }
  310. }
  311. }
  312. return nil
  313. }
  314. // DeleteOneSeasonSubCacheFolder 删除一个连续剧中的所有一季字幕的缓存文件夹
  315. func DeleteOneSeasonSubCacheFolder(seriesDir string) error {
  316. debugFolderByName, err := my_util.GetDebugFolderByName([]string{filepath.Base(seriesDir)})
  317. if err != nil {
  318. return err
  319. }
  320. files, err := os.ReadDir(debugFolderByName)
  321. if err != nil {
  322. return err
  323. }
  324. pathSep := string(os.PathSeparator)
  325. for _, curFile := range files {
  326. if curFile.IsDir() == true {
  327. matched := regex_things.RegOneSeasonSubFolderNameMatch.FindAllStringSubmatch(curFile.Name(), -1)
  328. if matched == nil || len(matched) < 1 {
  329. continue
  330. }
  331. fullPath := debugFolderByName + pathSep + curFile.Name()
  332. err = os.RemoveAll(fullPath)
  333. if err != nil {
  334. return err
  335. }
  336. }
  337. }
  338. return nil
  339. }
  340. /*
  341. 只针对英文字幕进行合并分散的 DialoguesFilter
  342. 会遇到这样的字幕,如下0
  343. 2line-The Card Counter (2021) WEBDL-1080p.chinese(inside).ass
  344. 它的对白一句话分了两个 dialogue 去做。这样做后续字幕时间轴校正就会遇到问题,因为只有一半,匹配占比会很低
  345. (每一个 Dialogue 的首字母需要分析,大写和小写的占比是多少,统计一下,正常的,和上述特殊的)
  346. 那么,就需要额外的逻辑去对 DialoguesFilterEx 进行额外的推断
  347. 暂时考虑的方案是,英文对白每一句的开头应该是英文大写字幕,如果是小写字幕,就应该与上语句合并,且每一句的字符长度有大于一定才触发
  348. */
  349. func MergeMultiDialogue4EngSubtitle(inSubParser *subparser.FileInfo) {
  350. merger := NewDialogueMerger()
  351. for _, dialogueEx := range inSubParser.DialoguesFilterEx {
  352. merger.Add(dialogueEx)
  353. }
  354. inSubParser.DialoguesFilterEx = merger.Get()
  355. }
  356. // GetVADInfoFeatureFromSub 跟下面的 GetVADInfoFeatureFromSubNeedOffsetTimeWillInsert 函数功能一致
  357. func GetVADInfoFeatureFromSub(fileInfo *subparser.FileInfo, frontAndEndPer float64, subUnitMaxCount int, insert bool) ([]SubUnit, error) {
  358. return GetVADInfoFeatureFromSubNeedOffsetTimeWillInsert(fileInfo, frontAndEndPer, subUnitMaxCount, 0, insert)
  359. }
  360. /*
  361. GetVADInfoFeatureFromSubNeedOffsetTimeWillInsert 只不过这里可以加一个每一句话固定的偏移时间
  362. 这里的字幕要求是完整的一个字幕
  363. 1. 抽取字幕的时间片段的时候,暂定,前 15% 和后 15% 要避开,前奏、主题曲、结尾曲
  364. 2. 将整个字幕,抽取连续 5 句对话为一个单元,提取时间片段信息
  365. 3. 这里抽取的是特征,也就有额外的逻辑去找这个特征(本程序内会描述为“钥匙”)
  366. */
  367. func GetVADInfoFeatureFromSubNeedOffsetTimeWillInsert(fileInfo *subparser.FileInfo, SkipFrontAndEndPer float64, subUnitMaxCount int, offsetTime float64, insert bool) ([]SubUnit, error) {
  368. if subUnitMaxCount < 0 {
  369. subUnitMaxCount = 0
  370. }
  371. nowDialogue := fileInfo.Dialogues
  372. srcSubUnitList := make([]SubUnit, 0)
  373. srcSubDialogueList := make([]subparser.OneDialogue, 0)
  374. srcOneSubUnit := NewSubUnit()
  375. // 最后一个对话的结束时间
  376. lastDialogueExTimeEnd, err := my_util.ParseTime(nowDialogue[len(nowDialogue)-1].EndTime)
  377. if err != nil {
  378. return nil, err
  379. }
  380. // 相当于总时长
  381. fullDuration := my_util.Time2SecondNumber(lastDialogueExTimeEnd)
  382. // 最低的起始时间,因为可能需要裁剪范围
  383. startRangeTimeMin := fullDuration * SkipFrontAndEndPer
  384. endRangeTimeMax := fullDuration * (1.0 - SkipFrontAndEndPer)
  385. println(startRangeTimeMin)
  386. println(endRangeTimeMax)
  387. for i := 0; i < len(nowDialogue); i++ {
  388. oneDialogueExTimeStart, err := my_util.ParseTime(nowDialogue[i].StartTime)
  389. if err != nil {
  390. return nil, err
  391. }
  392. oneDialogueExTimeEnd, err := my_util.ParseTime(nowDialogue[i].EndTime)
  393. if err != nil {
  394. return nil, err
  395. }
  396. oneStart := my_util.Time2SecondNumber(oneDialogueExTimeStart)
  397. if SkipFrontAndEndPer > 0 {
  398. if fullDuration*SkipFrontAndEndPer > oneStart || fullDuration*(1.0-SkipFrontAndEndPer) < oneStart {
  399. continue
  400. }
  401. }
  402. if nowDialogue[i].Lines == nil || len(nowDialogue[i].Lines) == 0 {
  403. continue
  404. }
  405. // 如果当前的这一句话,为空,或者进过正则表达式剔除特殊字符后为空,则跳过
  406. if my_util.ReplaceSpecString(nowDialogue[i].Lines[0], "") == "" {
  407. continue
  408. }
  409. // 如果当前的这一句话,为空,或者进过正则表达式剔除特殊字符后为空,则跳过
  410. if my_util.ReplaceSpecString(fileInfo.GetDialogueExContent(i), "") == "" {
  411. continue
  412. }
  413. // 低于 5句对白,则添加
  414. if srcOneSubUnit.GetDialogueCount() < subUnitMaxCount {
  415. // 算上偏移
  416. offsetTimeDuration := time.Duration(offsetTime * math.Pow10(9))
  417. oneDialogueExTimeStart = oneDialogueExTimeStart.Add(offsetTimeDuration)
  418. oneDialogueExTimeEnd = oneDialogueExTimeEnd.Add(offsetTimeDuration)
  419. // 如果没有偏移就是 0
  420. if insert == true {
  421. srcOneSubUnit.AddAndInsert(oneDialogueExTimeStart, oneDialogueExTimeEnd)
  422. } else {
  423. srcOneSubUnit.Add(oneDialogueExTimeStart, oneDialogueExTimeEnd)
  424. }
  425. // 这一个单元的 Dialogue 需要合并起来,才能判断是否符合“钥匙”的要求
  426. srcSubDialogueList = append(srcSubDialogueList, nowDialogue[i])
  427. } else {
  428. // 用完清空
  429. srcSubDialogueList = make([]subparser.OneDialogue, 0)
  430. // 将拼凑起来的对话组成一个单元进行存储起来
  431. srcSubUnitList = append(srcSubUnitList, *srcOneSubUnit)
  432. // 然后重置
  433. srcOneSubUnit = NewSubUnit()
  434. }
  435. }
  436. if srcOneSubUnit.GetDialogueCount() > 0 {
  437. srcSubUnitList = append(srcSubUnitList, *srcOneSubUnit)
  438. }
  439. return srcSubUnitList, nil
  440. }
  441. /*
  442. GetVADInfoFeatureFromSubNew 将 Sub 文件转换为 VAD List 信息
  443. */
  444. func GetVADInfoFeatureFromSubNew(fileInfo *subparser.FileInfo, SkipFrontAndEndPer float64) (*SubUnit, error) {
  445. outSubUnits := NewSubUnit()
  446. if len(fileInfo.Dialogues) <= 0 {
  447. return nil, errors.New("GetVADInfoFeatureFromSubNew fileInfo Dialogue Length is 0")
  448. }
  449. /*
  450. 先拼凑出完整的一个 VAD List
  451. 因为 VAD 的窗口是 10ms,那么需要多每一句话按 10 ms 的单位进行取整
  452. 每一句话开始、结束的时间,需要向下取整
  453. */
  454. subStartTimeFloor := my_util.MakeFloor10msMultipleFromFloat(my_util.Time2SecondNumber(fileInfo.GetStartTime()))
  455. subEndTimeFloor := my_util.MakeFloor10msMultipleFromFloat(my_util.Time2SecondNumber(fileInfo.GetEndTime()))
  456. // 如果想要从 0 时间点开始算,那么 subStartTimeFloor 这个值就需要重置到0
  457. subStartTimeFloor = 0
  458. subFullSecondTimeFloor := subEndTimeFloor - subStartTimeFloor
  459. // 根据这个时长就能够得到一个完整的 VAD List,然后再通过每一句对白进行 VAD 值的调整即可,这样就能够保证
  460. // 相同的一个字幕因为使用 ffmpeg 导出 srt 和 ass 后的,可能存在总体时间轴不一致的问题
  461. // 123.450 - > 12345
  462. vadLen := int(subFullSecondTimeFloor*100) + 2
  463. subVADs := make([]vad.VADInfo, vadLen)
  464. subStartTimeFloor10ms := subStartTimeFloor * 100
  465. for i := 0; i < vadLen; i++ {
  466. subVADs[i] = *vad.NewVADInfoBase(false, time.Duration((subStartTimeFloor10ms+float64(i))*math.Pow10(7)))
  467. }
  468. // 计算出需要截取的片段,起始和结束
  469. skipLen := int(float64(vadLen) * SkipFrontAndEndPer)
  470. skipStartIndex := skipLen
  471. skipEndIndex := vadLen - skipLen
  472. // 现在需要从 fileInfo 的每一句对白也就对应一段连续的 VAD active = true 来进行改写,记得向下取整
  473. lastDialogueIndex := 0
  474. for _, dialogue := range fileInfo.Dialogues {
  475. if dialogue.Lines == nil || len(dialogue.Lines) == 0 {
  476. continue
  477. }
  478. // 如果当前的这一句话,为空,或者进过正则表达式剔除特殊字符后为空,则跳过
  479. if my_util.ReplaceSpecString(dialogue.Lines[0], "") == "" {
  480. continue
  481. }
  482. // 字幕的开始时间
  483. oneDialogueStartTime, err := my_util.ParseTime(dialogue.StartTime)
  484. if err != nil {
  485. return nil, err
  486. }
  487. // 字幕的结束时间
  488. oneDialogueEndTime, err := my_util.ParseTime(dialogue.EndTime)
  489. if err != nil {
  490. return nil, err
  491. }
  492. // 字幕的时长,对时间进行向下取整
  493. oneDialogueStartTimeFloor := my_util.MakeCeil10msMultipleFromFloat(my_util.Time2SecondNumber(oneDialogueStartTime))
  494. oneDialogueEndTimeFloor := my_util.MakeFloor10msMultipleFromFloat(my_util.Time2SecondNumber(oneDialogueEndTime))
  495. // 得到一句对白的时长
  496. changeVADStartIndex := int(oneDialogueStartTimeFloor * 100)
  497. changeVADEndIndex := int(oneDialogueEndTimeFloor * 100)
  498. // 不能超过 最后一句话的时常
  499. if changeVADStartIndex > int(subEndTimeFloor*100) {
  500. continue
  501. }
  502. // 也不能比起始的第一句话时间轴更低
  503. if changeVADStartIndex < int(subStartTimeFloor10ms) {
  504. continue
  505. }
  506. // 当前这句话的开始和结束信息
  507. changerStartIndex := changeVADStartIndex - int(subStartTimeFloor10ms)
  508. if changerStartIndex < 0 {
  509. continue
  510. }
  511. changerEndIndex := changeVADEndIndex - int(subStartTimeFloor10ms)
  512. if changerEndIndex < 0 {
  513. continue
  514. }
  515. // 如果上一个对白的最后一个 OffsetIndex 连接着当前这一句的索引的 VAD 信息 active 是 true 就设置为 false
  516. if lastDialogueIndex == changerStartIndex {
  517. for i := 1; i <= 2; i++ {
  518. if lastDialogueIndex-i >= 0 && subVADs[lastDialogueIndex-i].Active == true {
  519. subVADs[lastDialogueIndex-i].Active = false
  520. }
  521. }
  522. }
  523. // 开始根据当前这句话进行 VAD 信息的设置
  524. // 调整之前做好的整体 VAD 的信息,符合 VAD active = true
  525. if changerEndIndex >= vadLen {
  526. changerEndIndex = vadLen - 1
  527. }
  528. for i := changerStartIndex; i <= changerEndIndex; i++ {
  529. subVADs[i].Active = true
  530. }
  531. lastDialogueIndex = changerEndIndex
  532. }
  533. // 截取出来当前这一段
  534. tmpVADList := subVADs[skipStartIndex:skipEndIndex]
  535. outSubUnits.VADList = tmpVADList
  536. tmpStartTime := time.Time{}
  537. tmpStartTime = tmpStartTime.Add(tmpVADList[0].Time)
  538. tmpEndTime := time.Time{}
  539. tmpEndTime = tmpEndTime.Add(tmpVADList[len(tmpVADList)-1].Time)
  540. outSubUnits.SetBaseTime(tmpStartTime)
  541. outSubUnits.SetOffsetStartTime(tmpStartTime)
  542. outSubUnits.SetOffsetEndTime(tmpEndTime)
  543. return outSubUnits, nil
  544. }