mirror of
https://github.com/uutils/coreutils
synced 2024-12-18 17:14:42 +00:00
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:
parent
8632e19b44
commit
76abc7e51d
1 changed files with 30 additions and 28 deletions
|
@ -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.
|
||||||
|
|
Loading…
Reference in a new issue