sort: refactor compare_fns into Settings

Also split out a compare_by(a, b, settings) helper function,
which may be used by --merge, later.
This commit is contained in:
David Laban 2016-08-03 20:43:36 +01:00
parent 8632e19b44
commit 76abc7e51d

View file

@ -46,6 +46,7 @@ struct Settings {
stable: bool, stable: bool,
unique: bool, unique: bool,
check: bool, check: bool,
compare_fns: Vec<fn(&String, &String) -> Ordering>,
} }
impl Default for Settings { impl Default for Settings {
@ -57,6 +58,7 @@ impl Default for Settings {
stable: false, stable: false,
unique: false, unique: false,
check: false, check: false,
compare_fns: Vec::new(),
} }
} }
} }
@ -125,6 +127,21 @@ With no FILE, or when FILE is -, read standard input.", NAME, VERSION);
files.push("-".to_owned()); files.push("-".to_owned());
} }
settings.compare_fns.push(match settings.mode {
SortMode::Numeric => numeric_compare,
SortMode::HumanNumeric => human_numeric_size_compare,
SortMode::Month => month_compare,
SortMode::Version => version_compare,
SortMode::Default => String::cmp
});
if !settings.stable {
match settings.mode {
SortMode::Default => {}
_ => settings.compare_fns.push(String::cmp)
}
}
exec(files, &settings) exec(files, &settings)
} }
@ -150,24 +167,7 @@ fn exec(files: Vec<String>, settings: &Settings) -> i32 {
let original_lines = lines.to_vec(); let original_lines = lines.to_vec();
let mut compare_fns = Vec::new(); sort_by(&mut lines, &settings);
compare_fns.push(match settings.mode {
SortMode::Numeric => numeric_compare,
SortMode::HumanNumeric => human_numeric_size_compare,
SortMode::Month => month_compare,
SortMode::Version => version_compare,
SortMode::Default => String::cmp
});
if !settings.stable {
match settings.mode {
SortMode::Default => {}
_ => compare_fns.push(String::cmp)
}
}
sort_by(&mut lines, compare_fns);
if settings.unique { if settings.unique {
lines.dedup() lines.dedup()
@ -193,20 +193,22 @@ fn exec(files: Vec<String>, settings: &Settings) -> i32 {
} }
fn sort_by<F>(lines: &mut Vec<String>, compare_fns: Vec<F>) fn sort_by(lines: &mut Vec<String>, settings: &Settings) {
where F: Fn( &String, &String ) -> Ordering
{
lines.sort_by(|a, b| { lines.sort_by(|a, b| {
for compare_fn in &compare_fns { compare_by(a, b, &settings)
let cmp = compare_fn(a, b);
if cmp != Ordering::Equal {
return cmp;
}
}
return Ordering::Equal;
}) })
} }
fn compare_by(a: &String, b: &String, settings: &Settings) -> Ordering {
for compare_fn in &settings.compare_fns {
let cmp = compare_fn(a, b);
if cmp != Ordering::Equal {
return cmp;
}
}
return Ordering::Equal;
}
/// Parse the beginning string into an f64, returning -inf instead of NaN on errors. /// Parse the beginning string into an f64, returning -inf instead of NaN on errors.
fn permissive_f64_parse(a: &str) -> f64 { fn permissive_f64_parse(a: &str) -> f64 {
// Maybe should be split on non-digit, but then 10e100 won't parse properly. // Maybe should be split on non-digit, but then 10e100 won't parse properly.