TeamPiped-mirror/sponsorblock-mirror
1
0
Fork 0
mirror of https://github.com/TeamPiped/sponsorblock-mirror.git synced 2024-08-14 23:57:05 +00:00
Code Issues Projects Releases Packages Wiki Activity

Improve segment overlap checking for chapters.

Browse source
This commit is contained in:
Kavin 2022-10-25 13:13:06 +01:00
parent f4d8e01841
commit 525cded71e
No known key found for this signature in database
GPG key ID: 49451E4482CC5BCD
1 changed files with 12 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
src/routes.rs
View file

@ -88,7 +88,7 @@ pub async fn skip_segments(
let mut found_similar = false; let mut found_similar = false;
for seg in &sponsor.segments { for seg in &sponsor.segments {
if is_overlap(&segment, &seg.category, seg.segment[0], seg.segment[1]) { if is_overlap(&segment, &seg.category, &seg.action_type, seg.segment[0], seg.segment[1]) {
found_similar = true; found_similar = true;
break; break;
} }
@ -140,7 +140,7 @@ fn similar_segments(segment: &Segment, hash: &str, segments: &Vec<SponsorTime>)
continue; continue;
} }
let is_similar = is_overlap(segment, &seg.category, seg.start_time, seg.end_time); let is_similar = is_overlap(segment, &seg.category, &seg.action_type, seg.start_time, seg.end_time);
if is_similar { if is_similar {
similar_segments.push(Segment { similar_segments.push(Segment {
@ -160,7 +160,7 @@ fn similar_segments(segment: &Segment, hash: &str, segments: &Vec<SponsorTime>)
similar_segments similar_segments
} }
fn is_overlap(seg: &Segment, cat: &str, start: f32, end: f32) -> bool { fn is_overlap(seg: &Segment, cat: &str, action_type: &str, start: f32, end: f32) -> bool {
if seg.category != cat { if seg.category != cat {
return false; return false;
} }
@ -170,7 +170,15 @@ fn is_overlap(seg: &Segment, cat: &str, start: f32, end: f32) -> bool {
} }
let overlap = f32::min(seg.segment[1], end) - f32::max(seg.segment[0], start); let overlap = f32::min(seg.segment[1], end) - f32::max(seg.segment[0], start);
let duration = f32::max(seg.segment[1], end) - f32::min(seg.segment[0], start); let duration = f32::max(seg.segment[1], end) - f32::min(seg.segment[0], start);
overlap / duration > 0.1 overlap / duration > {
if cat == "chapter" {
0.8
} else if seg.action_type == action_type {
0.6
} else {
0.1
}
}
} }
fn best_segment(segments: &Vec<Segment>) -> Segment { fn best_segment(segments: &Vec<Segment>) -> Segment {