MightyPork
/
srtune
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fixed and improved

Browse Source
automove
Ondřej Hruška 5 years ago
parent 5285b61e9f
commit b904965e6a
Signed by: MightyPork
GPG Key ID: 2C5FD5035250423D
1 changed files with 385 additions and 279 deletions
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Show Stats Download Patch File Download Diff File
  1. 664
      src/main.rs

664
src/main.rs
Unescape View File

@ -5,7 +5,7 @@ extern crate lazy_static;
use regex::Regex;
use std::fs::{File, OpenOptions};
use std::io::{BufReader, Write};
use std::io::{BufReader, Write as ioWrite, StdoutLock, StdinLock};
use std::io;
use std::str::FromStr;
use std::convert::TryFrom;
@ -13,6 +13,7 @@ use std::io::BufRead;
use std::ops::{Add, Mul, MulAssign, AddAssign, SubAssign, Sub};
use std::fmt::{self, Display};
use serde::export::fmt::Debug;
use core::fmt::Write as fmtWrite;
const LOG_LEVELS: [&str; 5] = ["error", "warn", "info", "debug", "trace"];
const SPAMMY_LIBS: [&str; 5] = ["tokio_reactor", "hyper", "reqwest", "mio", "want"];
@ -24,14 +25,21 @@ fn main() {
.about("Modify a .srt file to match a video. Input and output can be a file or stream, \
so you pipe multiple invocations to create more complex operations. However, a single \
invocation should suffice in most cases.\n\
Times can be specified in any format: \
seconds (400, 14.52), hours:minutes:seconds (14:00, 15:51.12, 1:30:00). Decimal point \
can be period or comma; Times copied directly from the .srt file will also work.\n\
When a command allows both time and index as a value, index must be prefixed with '@'.\
The tool should be used iteratively, adjusting the invocation until the generated \
subtitle file meets expectations. As such, times and indices accepted by its parameters \
are, by default, the ones seen in the output file. Prefix a time or index with '^' \
to use the original value from the input instead (i.e. original index 14 is '^@14')
\n\
Times are specified with colons and always include seconds (HH:MM:SS, MM:SS, 0:SS). \
Decimal point can be either period or comma, so times can be copied directly from the \
.srt file. Numbers without colons are assumed to be subtitle indices.\n\
\n\
The tool can be used iteratively, adjusting the invocation until the generated \
subtitle file matches the audio track. As such, times accepted by its parameters \
are, by default, the ones seen in the output file (after shifts and moving), while \
indices are those from the input file.\n\
\n\
Indices are normally not renumbered, so the output file can be used as a reference \
for both times and indices. The flag '--renumber' will give each output entry a new \
sequential number. Please note that, once renumbered, the indices in the output file \
should no longer be used in the command invocation, as there can be a mismatch. They \
can be used when piped into a new process, of course.
")
.arg(clap::Arg::with_name("input")
.value_name("INFILE")
@ -43,43 +51,46 @@ fn main() {
.value_name("OUTFILE")
.help("Output file, defaults to stdout"),
)
.arg(clap::Arg::with_name("from")
.short("f")
.long("from-time")
.value_name("TIME")
.help("Time of the first affected entry, or its index. Defaults to 0/^@0."),
)
// .arg(clap::Arg::with_name("drop")
// .short("d")
// .long("drop")
// .value_name("ITEMS")
// .help("Drop one or multiple entries (separated by comma). Entries can be TIME \
// or INDEX. Two entries can be joined by '..' to specify a range. Use '..ENTRY'\
// or 'ENTRY..' to drop all in one direction. Ranges are inclusive."),
// )
.arg(clap::Arg::with_name("move")
.short("m")
.long("move")
.value_name("TIME")
.help("Move subtitles in time. Use seconds, M:S or H:M:S, decimals and minus \
are supported. Starts at the point specified by '--from'"),
.value_name("OFFSET")
.help("Move all subtitles in time (e.g 12:00.15 or -0:44)"),
)
.arg(clap::Arg::with_name("automove")
.short("M")
.long("automove")
.value_name("SUBTIME=VIDEOTIME")
.value_name("ENTRY=VIDEOTIME")
.multiple(true)
.help("Move subtitles following a given time or index to match video times. \
This automatically sets '--from' if not given explicitly."),
.help("Move subtitles starting at a given time or index to align with \
a matching audio track time. This argument can be given multiple times. \
Some subtitles may be dropped if they fall outside the timeline after \
the move."),
)
.arg(clap::Arg::with_name("scale")
.short("s")
.long("scale")
.value_name("RATIO")
.help("Scale subtitle times and durations to compensate for bitrate \
.help("Scale all subtitle times and durations to compensate for bitrate \
differences. 1 means identity, 1.1 makes all times 10% longer. Scaling is \
relative to the first emitted subtitle with positive time (after shifting). \
Has no effect if '--autoscale' is used."),
relative to the first emitted subtitle; align it with '--move'. This option \
has no effect if '--autoscale' is used."),
)
.arg(clap::Arg::with_name("autoscale")
.short("S")
.long("autoscale")
.value_name("SUBTIME=VIDEOTIME")
.help("Calculate scaling based on a perceived difference. The scaling is \
related to the first emitted subtitle, so ensure it is aligned properly \
with '--move'."),
related to the first emitted subtitle; align it with '--move'. \
This overrides '--scale'."),
)
.arg(clap::Arg::with_name("durscale")
.short("d")
@ -92,10 +103,10 @@ fn main() {
.arg(clap::Arg::with_name("renumber")
.short("r")
.long("renumber")
.help("Change all numbers to be sequential starting with 1"),
.help("Renumber all emitted entries with sequential 1-based numbers."),
)
.arg(clap::Arg::with_name("v").short("v").multiple(true).help(
"Sets the level of verbosity (adds to the default - info)",
"Increase the logging verbosity; can be used multiple times",
))
.get_matches();
@ -128,48 +139,25 @@ fn main() {
}
builder.init();
let from = match argv.value_of("from") {
Some(mut s) => {
if s.starts_with('^') {
s = &s[1..];
if s.starts_with('@') {
let index = &s[1..].parse().expect("Bad --from format");
FromTag::ByIndexOrig(*index)
} else {
// this is always the orig time
FromTag::ByTime(SubDuration::try_from(s).expect("Bad --from format").as_instant())
}
} else {
if s.starts_with('@') {
let index = &s[1..].parse().expect("Bad --from format");
FromTag::ByIndex(*index)
} else{
FromTag::ByTime(SubDuration::try_from(s).expect("Bad --from format").as_instant())
}
}
}
None => FromTag::ByIndex(0)
};
let shift = match argv.value_of("move") {
Some(s) => {
SubDuration::try_from(s).expect("Bad --move format")
}
None => SubDuration(0f32)
None => SubDuration(0f64)
};
let scale = match argv.value_of("scale") {
Some(s) => {
s.parse().expect("Bad --scale format")
}
None => 1f32
None => 1f64
};
let durscale = match argv.value_of("durscale") {
Some(s) => {
s.parse().expect("Bad --durscale format")
}
None => 1f32
None => 1f64
}; // always also shrink durations
let autoscale = match argv.value_of("autoscale") {
@ -179,11 +167,11 @@ fn main() {
panic!("Bad --autoscale format, should be SUBTIME=VIDEOTIME")
}
let (first, second) = (halves[0], halves[1]);
if first.starts_with('^') {
panic!("'--autoscale' always uses original times");
if !first.contains(':') || !second.contains(':') {
panic!("'--autoscale' requires two times");
}
let subtime = SubDuration::try_from(first).expect("Bad --autoscale format").as_instant();
let vidtime = SubDuration::try_from(second).expect("Bad --autoscale format").as_instant();
let subtime = SubDuration::try_from(first).expect("Bad --autoscale time format").as_instant();
let vidtime = SubDuration::try_from(second).expect("Bad --autoscale time format").as_instant();
Some((subtime, vidtime))
}
@ -191,57 +179,68 @@ fn main() {
};
let mut automove = Vec::<AutoMoveTag>::new();
let mut automove_indices = vec![];
match argv.values_of("automove") {
Some(ss) => {
for s in ss {
let halves: Vec<&str> = s.split("=").collect();
if halves.len() != 2 {
panic!("Bad --automove format, should be SUBTIME=VIDEOTIME")
panic!("Bad --automove format, should be ENTRY=VIDEOTIME")
}
let (first, second) = (halves[0], halves[1]);
if !second.contains(':') {
panic!("'--automove' requires time after '='");
}
let (mut first, second) = (halves[0], halves[1]);
let vidtime = SubDuration::try_from(second).expect("Bad --automove format").as_instant();
if first.starts_with('^') {
first = &first[1..];
if s.starts_with('@') {
let index = &first[1..].parse().expect("Bad --automove format");
automove.push(AutoMoveTag::ByIndexOrig(*index, vidtime));
} else {
let subtime = SubDuration::try_from(first).expect("Bad --automove format").as_instant();
automove.push(AutoMoveTag::ByTimeOrig(subtime, vidtime));
}
if first.contains(':') {
let subtime = SubDuration::try_from(first).expect("Bad --automove format").as_instant();
automove.push(AutoMoveTag::ByTime(subtime, vidtime));
} else {
if s.starts_with('@') {
let index = &first[1..].parse().expect("Bad --automove format");
automove.push(AutoMoveTag::ByIndex(*index, vidtime));
} else{
let subtime = SubDuration::try_from(first).expect("Bad --automove format").as_instant();
automove.push(AutoMoveTag::ByTime(subtime, vidtime));
let index : u32 = first.parse().expect("Bad --automove format");
if automove_indices.contains(&index) {
panic!("Index {} already used in automove.", index);
}
automove_indices.push(index);
automove.push(AutoMoveTag::ByIndex(index, vidtime));
}
}
}
None => (/* no automoves */)
}
debug!("Automove: {:?}", automove);
let inf = argv.value_of("input");
let outf = argv.value_of("output");
let stdin = io::stdin();
let stdout = io::stdout();
let mut lines_iterator: Box<dyn Iterator<Item=Result<String, io::Error>>> = match inf {
let renumber = argv.is_present("renumber");
let opts = TransformOpts {
renumber,
autoscale,
durscale,
scale,
shift,
automove,
};
let lines_iterator: Box<dyn SubsInput> = match inf {
None => {
Box::new(stdin.lock().lines())
Box::new(StdinSubsInput::new(stdin.lock()))
}
Some(f) => {
let file = File::open(f).expect(&format!("Could not open file: {:?}", f));
Box::new(BufReader::new(file).lines())
Box::new(FileSubsInput::new(file))
}
};
let mut outfile: Box<dyn Write> = match outf {
let outfile: Box<dyn SubsOutput> = match outf {
None => {
Box::new(stdout.lock())
Box::new(StdoutSubsOutput::new(stdout.lock()))
}
Some(f) => {
let file = OpenOptions::new()
@ -250,65 +249,130 @@ fn main() {
.write(true)
.open(f)
.expect(&format!("Could not open file: {:?}", f));
Box::new(file)
Box::new(FileSubsOutput::new(file))
}
};
let renumber = argv.is_present("renumber");
transform_subtitles(lines_iterator, outfile, opts);
}
transform_subtitles(&mut lines_iterator, &mut outfile, TransformOpts {
renumber,
autoscale,
durscale,
scale,
shift,
from,
automove,
});
//region SubInput
trait SubsInput : Iterator<Item=String> {}
struct StdinSubsInput<'a> {
inner : io::Lines<StdinLock<'a>>
}
impl<'a> Iterator for StdinSubsInput<'a> {
type Item = String;
fn next(&mut self) -> Option<Self::Item> {
self.inner.next().map_or(None, Result::ok)
}
}
impl<'a> StdinSubsInput<'a> {
pub fn new(lock : StdinLock<'a>) -> Self {
Self {
inner : lock.lines()
}
}
}
impl<'a> SubsInput for StdinSubsInput<'a> {}
struct FileSubsInput {
inner : io::Lines<BufReader<File>>
}
impl FileSubsInput {
pub fn new(file : File) -> Self {
Self {
inner : BufReader::new(file).lines()
}
}
}
impl Iterator for FileSubsInput {
type Item = String;
fn next(&mut self) -> Option<Self::Item> {
self.inner.next().map_or(None, Result::ok)
}
}
impl SubsInput for FileSubsInput {}
//endregion
//region SubOutput
trait SubsOutput {
fn emit(&mut self, subtitle : Subtitle);
}
struct StdoutSubsOutput<'a> {
inner : StdoutLock<'a>
}
impl<'a> StdoutSubsOutput<'a> {
pub fn new(inner : StdoutLock<'a>) -> Self {
Self { inner }
}
}
impl<'a> SubsOutput for StdoutSubsOutput<'a> {
fn emit(&mut self, subtitle: Subtitle) {
self.inner.write(subtitle.to_string().as_bytes()).expect("failed to write");
}
}
struct FileSubsOutput {
inner : File
}
impl FileSubsOutput {
pub fn new(inner : File) -> Self {
Self { inner }
}
}
impl SubsOutput for FileSubsOutput {
fn emit(&mut self, subtitle: Subtitle) {
self.inner.write(subtitle.to_string().as_bytes()).expect("failed to write");
}
}
//endregion
#[derive(Debug)]
struct TransformOpts {
renumber: bool,
autoscale: Option<(SubInstant, SubInstant)>,
durscale: f32,
scale: f32,
durscale: f64,
scale: f64,
shift: SubDuration,
automove: Vec<AutoMoveTag>,
from: FromTag,
}
#[derive(Debug)]
enum AutoMoveTag {
ByTime(SubInstant, SubInstant),
ByTimeOrig(SubInstant, SubInstant),
ByIndex(u32, SubInstant),
ByIndexOrig(u32, SubInstant),
ByIndexRelative(u32, SubDuration),
}
#[derive(Debug)]
enum FromTag {
ByTime(SubInstant),
ByIndex(u32),
ByIndexOrig(u32)
ByIndex(u32, SubInstant)
}
#[derive(Debug,Default,Clone,Copy)]
struct IterState {
start_time : Option<SubInstant>,
renumber_i : u32,
timeline_head : SubInstant,
}
fn transform_subtitles<'a>(lines : &mut Box<dyn Iterator<Item=Result<String, io::Error>> + 'a>,
outfile : &mut Box<dyn Write + 'a>,
mut opts : TransformOpts) {
fn transform_subtitles<'a>(mut lines : Box<dyn SubsInput + 'a>, mut outfile : Box<dyn SubsOutput + 'a>, mut opts : TransformOpts) {
debug!("Opts: {:#?}", opts);
let mut istate = IterState::default();
let mut linebuf = vec![];
while let Some(Ok(x)) = lines.next() {
let mut linebuf : Vec<String> = vec![];
'lines: while let Some(x) = lines.next() {
let mut x = x.trim();
if x.starts_with('\u{feff}') {
debug!("Stripping BOM mark");
@ -319,16 +383,14 @@ fn transform_subtitles<'a>(lines : &mut Box<dyn Iterator<Item=Result<String, io:
continue;
}
let istate_backup = istate;
// 236
// 00:18:01,755 --> 00:18:03,774
// （掃除機の音）
// う～ん…。
match u32::from_str(x) {
Ok(num_orig) => {
Ok(num) => {
// println!("Entry {}", num);
let datesrow = lines.next().unwrap().unwrap();
let datesrow = lines.next().unwrap();
if datesrow.contains(" --> ") {
let mut halves = datesrow.split(" --> ");
let (first, second) = (halves.next().unwrap(), halves.next().unwrap());
@ -336,132 +398,111 @@ fn transform_subtitles<'a>(lines : &mut Box<dyn Iterator<Item=Result<String, io:
let sub_end = SubInstant::try_from(second).unwrap();
linebuf.clear();
while let Some(Ok(x)) = lines.next() {
'text: while let Some(x) = lines.next() {
if x.is_empty() {
break; // space between the entries
break 'text; // space between the entries
}
linebuf.push(x);
}
let num_new = if opts.renumber {
istate.renumber_i += 1;
istate.renumber_i
} else {
num_orig
};
// advance numbering only for the really emitted entries
let mut subtitle = Subtitle {
num : num_new,
num,
start: sub_start,
dur: sub_end - sub_start,
text: linebuf.join("\n"),
};
if match opts.from {
FromTag::ByTime(ins) => ins <= sub_start,
FromTag::ByIndex(idx) => idx <= num_new,
FromTag::ByIndexOrig(idx) => idx <= num_orig,
} {
if istate.start_time.is_none() {
debug!("Scaling anchored at {} (#{}), editing starts", sub_start, num_orig);
debug!("Shifting by: {}", opts.shift);
istate.start_time = Some(sub_start);
if let Some((mut subt, mut vidt)) = opts.autoscale {
debug!("Autoscale: VT {} -> ST {}", vidt, subt);
subt -= sub_start;
vidt -= sub_start + opts.shift;
if subt.0 <= 0f32 {
panic!("Error in autoscale, start time is negative or zero.");
}
if vidt.0 <= 0f32 {
panic!("Error in autoscale, end time is negative or zero.");
}
debug!(" relative to #{}, after \"move\": VT {} -> ST {}", num_orig, vidt, subt);
opts.scale = vidt.0 / subt.0;
debug!("Resolved scale as {}", opts.scale);
}
if istate.start_time.is_none() {
debug!("Scaling anchored at {} (#{})", sub_start, num);
istate.start_time = Some(sub_start);
opts.durscale *= opts.scale;
debug!("Duration scaling is {}", opts.durscale);
if let Some((mut subt, mut vidt)) = opts.autoscale {
debug!("Autoscale: VT {} -> ST {}", vidt, subt);
subt -= sub_start;
vidt -= sub_start + opts.shift;
if subt.0 <= 0f64 {
panic!("Error in autoscale, start time is negative or zero.");
}
if vidt.0 <= 0f64 {
panic!("Error in autoscale, end time is negative or zero.");
}
debug!(" relative to #{}, after \"move\": VT {} -> ST {}", num, vidt, subt);
opts.scale = vidt.0 / subt.0;
debug!("Resolved scale as {}", opts.scale);
}
if opts.scale != 1f32 {
subtitle.start = subtitle.start.scale(istate.start_time.unwrap(), opts.scale);
}
opts.durscale *= opts.scale;
debug!("Duration scaling is {}", opts.durscale);
}
subtitle.dur *= opts.durscale;
subtitle.start += opts.shift;
for amove in opts.automove.iter_mut() {
match amove {
AutoMoveTag::ByIndex(idx, ref vidt) => {
if num_new >= *idx {
debug!("Move by new index starts, reached {}", idx);
let vidt = *vidt;
let dif = vidt - subtitle.start;
subtitle.start = vidt;
std::mem::replace(amove, AutoMoveTag::ByIndexRelative(num_new, dif));
} else if *vidt < subtitle.start && *idx > num_new {
// istate = istate_backup;
warn!("Skip overlapped #{} (by index)", num_orig);
continue;
}
}
AutoMoveTag::ByIndexOrig(idx, ref vidt) => {
if num_orig >= *idx {
debug!("Move by orig index starts, reached {}", idx);
let vidt = *vidt;
let dif = vidt - subtitle.start;
subtitle.start = vidt;
std::mem::replace(amove, AutoMoveTag::ByIndexRelative(num_new, dif));
} else if *vidt < sub_start && *idx > num_orig {
// istate = istate_backup;
warn!("Skip overlapped #{} (by orig index)", num_orig);
continue;
}
if opts.scale != 1f64 {
let scaled = subtitle.start.scale(istate.start_time.unwrap(), opts.scale);
trace!("Scale #{} ({}) -> {}", num, subtitle.start, scaled);
subtitle.start = scaled;
}
subtitle.dur *= opts.durscale;
// TODO prevent durations overlap (will need to buffer one entry)
let would_be_shifted_start = subtitle.start + opts.shift;
// TODO use drain_filter when stable
let mut to_drop = vec![];
for (i, amove) in opts.automove.iter().enumerate() {
match amove {
AutoMoveTag::ByIndex(idx, ref vidt) => {
if num >= *idx {
let dif = *vidt - would_be_shifted_start;
debug!("Move by index #{} starts at #{} ({}) -> {}, diff {}", *idx, num, subtitle.start, *vidt, dif);
opts.shift += dif;
to_drop.push(i);
} else if would_be_shifted_start > *vidt {
warn!("Discarding out-of-order entry #{} @ {} (timeline head is {})", num, subtitle.start, istate.timeline_head);
continue 'lines;
}
AutoMoveTag::ByTime(ref subt, ref vidt) => {
if subtitle.start >= *vidt {
// TODO verify
subtitle.start += *vidt - *subt;
} else if *vidt < subtitle.start && *subt > subtitle.start {
// istate = istate_backup;
warn!("Skip overlapped #{} (by time)", num_orig);
continue;
}
}
AutoMoveTag::ByTime(ref subt, ref vidt) => {
if would_be_shifted_start >= *subt {
let dif = *vidt - *subt;
debug!("Move by time {} starts at #{} ({}) -> {}, diff {}", subt, num, subtitle.start, *vidt, dif);
opts.shift += dif;
to_drop.push(i);
}
AutoMoveTag::ByTimeOrig(ref subt, ref vidt) => {
if sub_start >= *subt {
// TODO verify
subtitle.start += *vidt - *subt;
} else if *vidt < subtitle.start && *subt > sub_start {
// istate = istate_backup;
warn!("Skip overlapped #{} (by orig time)", num_orig);
continue;
}
},
// this is used internally
AutoMoveTag::ByIndexRelative(ref idx, ref dif) => {
if num_new >= *idx {
subtitle.start += *dif;
}
},
}
}
}
for i in &to_drop {
let x = opts.automove.swap_remove(*i);
trace!("Clean up: {:?}", x);
}
if !to_drop.is_empty() {
debug!("New shift: {}, Timeline head: {}", opts.shift, istate.timeline_head);
}
if opts.shift.0 != 0f64 {
let shifted = subtitle.start + opts.shift;
trace!("Shift #{} ({}) by {} -> {}", num, subtitle.start, opts.shift, shifted);
subtitle.start = shifted;
}
if subtitle.start.0 < 0f32 {
warn!("Discarding negative time entry #{} @ {:.3}s", subtitle.num, subtitle.start.0);
istate = istate_backup;
continue;
if subtitle.start <= istate.timeline_head {
warn!("Discarding out-of-order entry #{} @ {} (timeline head is {})", num, subtitle.start, istate.timeline_head);
continue 'lines;
}
outfile.write(subtitle.to_string().as_bytes()).expect("failed to write");
if subtitle.start.is_negative() {
warn!("Discarding negative time entry #{} @ {:.3}s", num, sub_start);
continue 'lines;
}
istate.timeline_head = subtitle.start;
if opts.renumber {
istate.renumber_i += 1;
subtitle.num = istate.renumber_i;
}
outfile.emit(subtitle);
}
}
Err(e) => {
@ -473,12 +514,11 @@ fn transform_subtitles<'a>(lines : &mut Box<dyn Iterator<Item=Result<String, io:
}
}
}
outfile.flush().unwrap();
}
#[derive(Copy, Clone, PartialEq, PartialOrd)]
struct SubInstant(f32);
//region Time types
#[derive(Default, Copy, Clone, PartialEq, PartialOrd)]
struct SubInstant(f64);
impl Debug for SubInstant {
fn fmt(&self, f: &mut fmt::Formatter) -> Result<(), fmt::Error> {
@ -486,8 +526,14 @@ impl Debug for SubInstant {
}
}
#[derive(Copy, Clone, PartialEq, PartialOrd)]
struct SubDuration(f32);
impl SubInstant {
fn is_negative(&self) -> bool {
self.0 < 0f64
}
}
#[derive(Default, Copy, Clone, PartialEq, PartialOrd)]
struct SubDuration(f64);
impl Debug for SubDuration {
fn fmt(&self, f: &mut fmt::Formatter) -> Result<(), fmt::Error> {
@ -497,7 +543,28 @@ impl Debug for SubDuration {
impl Display for SubDuration {
fn fmt(&self, f: &mut fmt::Formatter) -> Result<(), fmt::Error> {
write!(f, "{}", SubInstant(self.0))
let sign = self.0.signum();
let mut secs = self.0.abs();
let hours = (secs / 3600f64).floor();
secs -= hours * 3600f64;
let mins = (secs / 60f64).floor();
secs -= mins * 60f64;
let msecs = ((secs % 1f64) * 1000f64).round();
secs = secs.floor();
if sign.is_sign_negative() {
f.write_char('-')?;
}
if hours > 0f64 {
write!(f, "{:02}:", hours)?;
}
if hours > 0f64 || mins > 0f64 {
write!(f, "{:02}:{:02},{:03}", mins, secs, msecs)
} else {
write!(f, "{},{:03}", secs, msecs)
}
}
}
@ -525,35 +592,35 @@ impl Sub<SubInstant> for SubInstant {
}
}
impl Mul<f32> for SubDuration {
impl Mul<f64> for SubDuration {
type Output = SubDuration;
fn mul(self, rhs: f32) -> Self::Output {
fn mul(self, rhs: f64) -> Self::Output {
SubDuration(self.0 * rhs)
}
}
impl MulAssign<f32> for SubDuration {
fn mul_assign(&mut self, rhs: f32) {
impl MulAssign<f64> for SubDuration {
fn mul_assign(&mut self, rhs: f64) {
self.0 *= rhs;
}
}
impl AddAssign<f32> for SubDuration {
fn add_assign(&mut self, rhs: f32) {
impl AddAssign<f64> for SubDuration {
fn add_assign(&mut self, rhs: f64) {
self.0 += rhs;
}
}
impl SubInstant {
/// Scale by a factor with a custom start time
pub fn scale(&self, start: SubInstant, factor: f32) -> SubInstant {
pub fn scale(&self, start: SubInstant, factor: f64) -> SubInstant {
SubInstant(start.0 + (self.0 - start.0) * factor)
}
}
impl AddAssign<f32> for SubInstant {
fn add_assign(&mut self, rhs: f32) {
impl AddAssign<f64> for SubInstant {
fn add_assign(&mut self, rhs: f64) {
self.0 += rhs;
}
}
@ -576,6 +643,18 @@ impl SubAssign<SubDuration> for SubInstant {
}
}
impl SubAssign<SubDuration> for SubDuration {
fn sub_assign(&mut self, rhs: SubDuration) {
self.0 -= rhs.0;
}
}
impl AddAssign<SubDuration> for SubDuration {
fn add_assign(&mut self, rhs: SubDuration) {
self.0 += rhs.0;
}
}
impl AddAssign<SubDuration> for SubInstant {
fn add_assign(&mut self, rhs: SubDuration) {
self.0 += rhs.0;
@ -584,19 +663,19 @@ impl AddAssign<SubDuration> for SubInstant {
impl Display for SubInstant {
fn fmt(&self, f: &mut fmt::Formatter) -> Result<(), fmt::Error> {
// TODO optimize this
let sign = self.0.signum();
let mut secs = self.0.abs();
let hours = (secs / 3600f32).floor();
secs -= hours * 3600f32;
let minutes = (secs / 60f32).floor();
secs -= minutes * 60f32;
let msecs = ((secs % 1f32) * 1000f32).round();
let hours = (secs / 3600f64).floor();
secs -= hours * 3600f64;
let minutes = (secs / 60f64).floor();
secs -= minutes * 60f64;
let msecs = ((secs % 1f64) * 1000f64).round();
write!(f, "{}{:02}:{:02}:{:02},{:03}",
if sign.is_sign_negative() { "-" } else { "" },
hours, minutes, secs.floor(), msecs)
}
}
//endregion
#[derive(Clone, Debug)]
struct Subtitle {
@ -627,15 +706,15 @@ impl TryFrom<&str> for SubInstant {
match DATE_RE.captures(value) {
Some(caps) => {
let minus = if caps.get(1).is_some() { -1f32 } else { 1f32 };
let minus = if caps.get(1).is_some() { -1f64 } else { 1f64 };
let h = &caps["h"];
let m = &caps["m"];
let s = caps["s"].replace(",", ".");
Ok(SubInstant(minus * (f32::from_str(h).unwrap() * 3600f32 +
f32::from_str(m).unwrap() * 60f32 +
f32::from_str(&s).unwrap())))
Ok(SubInstant(minus * (f64::from_str(h).unwrap() * 3600f64 +
f64::from_str(m).unwrap() * 60f64 +
f64::from_str(&s).unwrap())))
}
None => Err(failure::format_err!("Error parsing time: {}", value))
}
@ -651,21 +730,29 @@ impl SubDuration {
impl TryFrom<&str> for SubDuration {
type Error = failure::Error;
fn try_from(value: &str) -> Result<Self, Self::Error> {
fn try_from(mut value: &str) -> Result<Self, Self::Error> {
lazy_static! {
static ref TIME_RE: Regex = Regex::new(r"^(?U)(?P<n>-)?(?:(?P<h>\d+):)?(?:(?P<m>\d+):)?(?P<s>\d+(?:[.,]\d+)?)$").unwrap();
static ref TIME_RE: Regex = Regex::new(r"^(?U)(?:(?P<h>\d+):)?(?:(?P<m>\d+):)?(?P<s>\d+(?:[.,]\d+)?)$").unwrap();
}
let negative = value.starts_with('-');
if negative {
value = &value[1..];
}
if value.starts_with(':') {
// prefixed colon when someone is lazy to type 0:
value = &value[1..];
}
match TIME_RE.captures(value) {
Some(caps) => {
let minus = if caps.name("n").is_some() { -1f32 } else { 1f32 };
let h = caps.name("h").map_or(0f32, |m| f32::from_str(m.as_str()).unwrap());
let m = caps.name("m").map_or(0f32, |m| f32::from_str(m.as_str()).unwrap());
let s = caps.name("s").map_or(0f32, |m| f32::from_str(&m.as_str().replace(",", ".")).unwrap());
Ok(SubDuration(minus * (h * 3600f32 +
m * 60f32 +
s)))
let minus = if negative { -1f64 } else { 1f64 };
let h = caps.name("h").map_or(0f64, |m| f64::from_str(m.as_str()).unwrap());
let m = caps.name("m").map_or(0f64, |m| f64::from_str(m.as_str()).unwrap());
let s = caps.name("s").map_or(0f64, |m| f64::from_str(&m.as_str().replace(",", ".")).unwrap());
Ok(SubDuration(minus * (h * 3600f64 + m * 60f64 + s)))
}
None => {
Err(failure::format_err!("Error parsing time: {}", value))
@ -677,38 +764,57 @@ impl TryFrom<&str> for SubDuration {
#[test]
fn test_parse_duration() {
// this is used for user input on the command line
let bad = SubDuration(-1f32);
assert_eq!(SubDuration(45678f32), SubDuration::try_from("45678").unwrap_or(bad), "integer secs");
assert_eq!(SubDuration(1.23f32), SubDuration::try_from("1.23").unwrap_or(bad), "float secs with period");
assert_eq!(SubDuration(-1.23f32), SubDuration::try_from("-1.23").unwrap_or(bad), "MINUS float secs with period");
assert_eq!(SubDuration(1.23f32), SubDuration::try_from("1,23").unwrap_or(bad), "float secs with comma");
assert_eq!(SubDuration(121.15f32), SubDuration::try_from("2:1.15").unwrap_or(bad), "m:s.frac");
assert_eq!(SubDuration(121.15f32), SubDuration::try_from("2:01.15").unwrap_or(bad), "m:0s.frac");
assert_eq!(SubDuration(121.15f32), SubDuration::try_from("02:01.15").unwrap_or(bad), "0m:0s.frac");
assert_eq!(SubDuration(121.15f32), SubDuration::try_from("02:01,15").unwrap_or(bad), "0m:0s,frac");
assert_eq!(SubDuration(3721.15f32), SubDuration::try_from("1:02:01,15").unwrap_or(bad), "h:0m:0s,frac");
assert_eq!(SubDuration(3721.15f32), SubDuration::try_from("1:02:01,15").unwrap_or(bad), "h:0m:0s.frac");
assert_eq!(SubDuration(3721.15f32), SubDuration::try_from("01:02:01,15").unwrap_or(bad), "0h:0m:0s,frac");
assert_eq!(SubDuration(-3721.15f32), SubDuration::try_from("-01:02:01,15").unwrap_or(bad), "-0h:0m:0s,frac");
let bad = SubDuration(-1f64);
assert_eq!(SubDuration::try_from(":45678").unwrap_or(bad), SubDuration(45678f64), "integer secs with colon prefix");
assert_eq!(SubDuration::try_from("-:45678").unwrap_or(bad), SubDuration(-45678f64), "neg integer secs with colon prefix");
assert_eq!(SubDuration::try_from("45678").unwrap_or(bad), SubDuration(45678f64), "integer secs");
assert_eq!(SubDuration::try_from("1.23").unwrap_or(bad), SubDuration(1.23f64), "float secs with period");
assert_eq!(SubDuration::try_from("-1.23").unwrap_or(bad), SubDuration(-1.23f64), "MINUS float secs with period");
assert_eq!(SubDuration::try_from("1,23").unwrap_or(bad), SubDuration(1.23f64), "float secs with comma");
assert_eq!(SubDuration::try_from("2:1.15").unwrap_or(bad), SubDuration(121.15f64), "m:s.frac");
assert_eq!(SubDuration::try_from("2:01.15").unwrap_or(bad), SubDuration(121.15f64), "m:0s.frac");
assert_eq!(SubDuration::try_from("02:01.15").unwrap_or(bad), SubDuration(121.15f64), "0m:0s.frac");
assert_eq!(SubDuration::try_from("02:01,15").unwrap_or(bad), SubDuration(121.15f64), "0m:0s,frac");
assert_eq!(SubDuration::try_from("1:02:01,15").unwrap_or(bad), SubDuration(3721.15f64), "h:0m:0s,frac");
assert_eq!(SubDuration::try_from("1:02:01,15").unwrap_or(bad), SubDuration(3721.15f64), "h:0m:0s.frac");
assert_eq!(SubDuration::try_from("01:02:01,15").unwrap_or(bad), SubDuration(3721.15f64), "0h:0m:0s,frac");
assert_eq!(SubDuration::try_from("-01:02:01,15").unwrap_or(bad), SubDuration(-3721.15f64), "-0h:0m:0s,frac");
assert_eq!(SubDuration::try_from("18:01,755").unwrap_or(bad), SubDuration(1081.755f64));
assert_eq!(SubDuration::try_from("-18:01,755").unwrap_or(bad), SubDuration(-1081.755f64));
assert_eq!(SubDuration::try_from("-18:01.7").unwrap_or(bad), SubDuration(-1081.7f64));
assert_eq!(SubDuration::try_from("-12:18:01.7").unwrap_or(bad), SubDuration(-44281.7f64));
assert_eq!(SubDuration::try_from("0,000").unwrap_or(bad), SubDuration(0f64));
assert_eq!(SubDuration::try_from("-0,000").unwrap_or(bad), SubDuration(-0f64));
}
#[test]
fn test_parse_instant() {
let bad = SubInstant(-1f32);
assert_eq!(SubInstant(1081.755f32), SubInstant::try_from("00:18:01,755").unwrap_or(bad));
assert_eq!(SubInstant(1081.755f32), SubInstant::try_from("00:18:01.755").unwrap_or(bad));
assert_eq!(SubInstant(1081.7f32), SubInstant::try_from("00:18:01.7").unwrap_or(bad));
assert_eq!(SubInstant(1081.7f32), SubInstant::try_from("0:18:1.7").unwrap_or(bad));
assert_eq!(SubInstant(0f32), SubInstant::try_from("00:00:00,000").unwrap_or(bad));
assert_eq!(SubInstant(-3600f32), SubInstant::try_from("-01:00:00,000").unwrap_or(bad));
let bad = SubInstant(-1f64);
assert_eq!(SubInstant::try_from("00:18:01,755").unwrap_or(bad), SubInstant(1081.755f64));
assert_eq!(SubInstant::try_from("00:18:01.755").unwrap_or(bad), SubInstant(1081.755f64));
assert_eq!(SubInstant::try_from("00:18:01.7").unwrap_or(bad), SubInstant(1081.7f64));
assert_eq!(SubInstant::try_from("0:18:1.7").unwrap_or(bad), SubInstant(1081.7f64));
assert_eq!(SubInstant::try_from("00:00:00,000").unwrap_or(bad), SubInstant(0f64));
assert_eq!(SubInstant::try_from("-01:00:00,000").unwrap_or(bad), SubInstant(-3600f64));
}
#[test]
fn test_stringify_instant() {
assert_eq!("00:18:01,755", SubInstant::try_from("00:18:01,755").unwrap().to_string());
assert_eq!("-00:18:01,755", SubInstant::try_from("-00:18:01,755").unwrap().to_string());
assert_eq!("-00:18:01,700", SubInstant::try_from("-00:18:01.7").unwrap().to_string());
assert_eq!("00:00:00,000", SubInstant::try_from("00:00:00,000").unwrap().to_string());
assert_eq!("-00:00:00,000", SubInstant::try_from("-00:00:00,000").unwrap().to_string());
assert_eq!(SubInstant::try_from("00:18:01,755").unwrap().to_string(), "00:18:01,755");
assert_eq!(SubInstant::try_from("-00:18:01,755").unwrap().to_string(), "-00:18:01,755");
assert_eq!(SubInstant::try_from("-00:18:01.7").unwrap().to_string(), "-00:18:01,700");
assert_eq!(SubInstant::try_from("00:00:00,000").unwrap().to_string(), "00:00:00,000");
assert_eq!(SubInstant::try_from("-00:00:00,000").unwrap().to_string(), "-00:00:00,000");
}
#[test]
fn test_stringify_duration() {
assert_eq!(SubDuration::try_from("18:01,755").unwrap().to_string(), "18:01,755");
assert_eq!(SubDuration::try_from("-18:01,755").unwrap().to_string(), "-18:01,755");
assert_eq!(SubDuration::try_from("-18:01.7").unwrap().to_string(), "-18:01,700");
assert_eq!(SubDuration::try_from("-12:18:01.7").unwrap().to_string(), "-12:18:01,700");
assert_eq!(SubDuration::try_from("0,000").unwrap().to_string(), "0,000");
assert_eq!(SubDuration::try_from("-0,000").unwrap().to_string(), "-0,000");
}

Write Preview
Loading…
Cancel
Save