Cutlery/Kyoo
1
0
Fork 0
mirror of https://github.com/zoriya/Kyoo.git synced 2025-06-01 04:34:50 -04:00
Code Issues Packages Projects Releases Wiki Activity

Transcoder: Make Chapter parsing simpler

Browse Source
This commit is contained in:
Arthur Jamet 2024-05-13 21:43:35 +02:00 committed by Arthur Jamet
parent 8a2343e76b
commit 324afc8022
1 changed files with 23 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
transcoder/src/info.go
View File

@ -153,15 +153,6 @@ func Map[T, U any](ts []T, f func(T, int) U) []U {
return us return us
} }
func Filter[T any](ss []T, test func(T, int) bool) (ret []T) {
for i, s := range ss {
if test(s, i) {
ret = append(ret, s)
}
}
return
}
func OrNull(str string) *string { func OrNull(str string) *string {
if str == "" { if str == "" {
return nil return nil
@ -349,30 +340,32 @@ func chapterTimeIsValid(chapterTime string) bool {
func getChapters(chapters_begin uint32, chapters_end uint32, mi *mediainfo.File, duration float32) []Chapter { func getChapters(chapters_begin uint32, chapters_end uint32, mi *mediainfo.File, duration float32) []Chapter {
chapterCount := max(chapters_end-chapters_begin, 1) chapterCount := max(chapters_end-chapters_begin, 1)
chapterIterationCount := chapterCount chapterIterationCount := chapterCount
chapters := make([]Chapter, chapterCount)
chapterIndex := 0
for i := uint32(0); i < chapterIterationCount; i++ { for i := 0; i < int(chapterIterationCount); i++ {
startTime := mi.GetI(mediainfo.StreamMenu, 0, int(chapters_begin+i), mediainfo.InfoName) rawStartTime := mi.GetI(mediainfo.StreamMenu, 0, int(chapters_begin)+i, mediainfo.InfoName)
if !chapterTimeIsValid(startTime) {
chapterIterationCount = chapterIterationCount + 1
}
}
return Filter(Map(make([]Chapter, chapterIterationCount), func(_ Chapter, i int) Chapter {
startTime := mi.GetI(mediainfo.StreamMenu, 0, int(chapters_begin)+i, mediainfo.InfoName)
if !chapterTimeIsValid(startTime) {
return Chapter{}
}
// +1 is safe, the value at chapters_end contains the right duration
rawEndTime := mi.GetI(mediainfo.StreamMenu, 0, int(chapters_begin)+i+1, mediainfo.InfoName) rawEndTime := mi.GetI(mediainfo.StreamMenu, 0, int(chapters_begin)+i+1, mediainfo.InfoName)
parsedEndTime := duration // If true, this "chapter" is invalid. We skip it
if chapterTimeIsValid(rawEndTime) { if !chapterTimeIsValid(rawStartTime) {
parsedEndTime = ParseTime(rawEndTime) chapterIterationCount = chapterIterationCount + 1
continue
} }
return Chapter{ var endTime float32
StartTime: ParseTime(startTime), // If this fails, we probably are at the end of the video
EndTime: parsedEndTime, // Since there would be no following chapter,
// we defacto set the end time to the end of the video (i.e. its duration)
if chapterTimeIsValid(rawEndTime) {
endTime = ParseTime(rawEndTime)
} else {
endTime = duration
}
chapters[chapterIndex] = Chapter{
StartTime: ParseTime(rawStartTime),
EndTime: endTime,
Name: mi.GetI(mediainfo.StreamMenu, 0, int(chapters_begin)+i, mediainfo.InfoText), Name: mi.GetI(mediainfo.StreamMenu, 0, int(chapters_begin)+i, mediainfo.InfoText),
} }
}), func(chapter Chapter, i int) bool { chapterIndex++
return chapter != Chapter{} }
}) return chapters
} }