Use bool instead of PartiolOrd in is_sorted_by

This commit is contained in:
EbbDrop 2023-12-10 23:29:11 +01:00
parent 7e452c123c
commit 606eeb84ad
6 changed files with 37 additions and 26 deletions

View File

@ -182,7 +182,7 @@ where
} }
// Ensure CGUs are sorted by name, so that we get deterministic results. // Ensure CGUs are sorted by name, so that we get deterministic results.
if !codegen_units.is_sorted_by(|a, b| Some(a.name().as_str().cmp(b.name().as_str()))) { if !codegen_units.is_sorted_by(|a, b| a.name().as_str() <= b.name().as_str()) {
let mut names = String::new(); let mut names = String::new();
for cgu in codegen_units.iter() { for cgu in codegen_units.iter() {
names += &format!("- {}\n", cgu.name()); names += &format!("- {}\n", cgu.name());
@ -311,7 +311,7 @@ fn merge_codegen_units<'tcx>(
assert!(cx.tcx.sess.codegen_units().as_usize() >= 1); assert!(cx.tcx.sess.codegen_units().as_usize() >= 1);
// A sorted order here ensures merging is deterministic. // A sorted order here ensures merging is deterministic.
assert!(codegen_units.is_sorted_by(|a, b| Some(a.name().as_str().cmp(b.name().as_str())))); assert!(codegen_units.is_sorted_by(|a, b| a.name().as_str() <= b.name().as_str()));
// This map keeps track of what got merged into what. // This map keeps track of what got merged into what.
let mut cgu_contents: FxHashMap<Symbol, Vec<Symbol>> = let mut cgu_contents: FxHashMap<Symbol, Vec<Symbol>> =

View File

@ -4026,42 +4026,42 @@ pub trait Iterator {
Self: Sized, Self: Sized,
Self::Item: PartialOrd, Self::Item: PartialOrd,
{ {
self.is_sorted_by(PartialOrd::partial_cmp) self.is_sorted_by(|a, b| a <= b)
} }
/// Checks if the elements of this iterator are sorted using the given comparator function. /// Checks if the elements of this iterator are sorted using the given comparator function.
/// ///
/// Instead of using `PartialOrd::partial_cmp`, this function uses the given `compare` /// Instead of using `PartialOrd::partial_cmp`, this function uses the given `compare`
/// function to determine the ordering of two elements. Apart from that, it's equivalent to /// function to determine whether two elements are to be considered in sorted order.
/// [`is_sorted`]; see its documentation for more information.
/// ///
/// # Examples /// # Examples
/// ///
/// ``` /// ```
/// #![feature(is_sorted)] /// #![feature(is_sorted)]
/// ///
/// assert!([1, 2, 2, 9].iter().is_sorted_by(|a, b| a.partial_cmp(b))); /// assert!([1, 2, 2, 9].iter().is_sorted_by(|a, b| a <= b));
/// assert!(![1, 3, 2, 4].iter().is_sorted_by(|a, b| a.partial_cmp(b))); /// assert!(![1, 2, 2, 9].iter().is_sorted_by(|a, b| a < b));
/// assert!([0].iter().is_sorted_by(|a, b| a.partial_cmp(b)));
/// assert!(std::iter::empty::<i32>().is_sorted_by(|a, b| a.partial_cmp(b)));
/// assert!(![0.0, 1.0, f32::NAN].iter().is_sorted_by(|a, b| a.partial_cmp(b)));
/// ```
/// ///
/// [`is_sorted`]: Iterator::is_sorted /// assert!([0].iter().is_sorted_by(|a, b| true));
/// assert!([0].iter().is_sorted_by(|a, b| false));
///
/// assert!(std::iter::empty::<i32>().is_sorted_by(|a, b| false));
/// assert!(std::iter::empty::<i32>().is_sorted_by(|a, b| true));
/// ```
#[unstable(feature = "is_sorted", reason = "new API", issue = "53485")] #[unstable(feature = "is_sorted", reason = "new API", issue = "53485")]
#[rustc_do_not_const_check] #[rustc_do_not_const_check]
fn is_sorted_by<F>(mut self, compare: F) -> bool fn is_sorted_by<F>(mut self, compare: F) -> bool
where where
Self: Sized, Self: Sized,
F: FnMut(&Self::Item, &Self::Item) -> Option<Ordering>, F: FnMut(&Self::Item, &Self::Item) -> bool,
{ {
#[inline] #[inline]
fn check<'a, T>( fn check<'a, T>(
last: &'a mut T, last: &'a mut T,
mut compare: impl FnMut(&T, &T) -> Option<Ordering> + 'a, mut compare: impl FnMut(&T, &T) -> bool + 'a,
) -> impl FnMut(T) -> bool + 'a { ) -> impl FnMut(T) -> bool + 'a {
move |curr| { move |curr| {
if let Some(Ordering::Greater) | None = compare(&last, &curr) { if !compare(&last, &curr) {
return false; return false;
} }
*last = curr; *last = curr;

View File

@ -4,7 +4,6 @@
mod macros; mod macros;
use crate::cmp; use crate::cmp;
use crate::cmp::Ordering;
use crate::fmt; use crate::fmt;
use crate::intrinsics::assume; use crate::intrinsics::assume;
use crate::iter::{ use crate::iter::{
@ -133,7 +132,7 @@ iterator! {struct Iter -> *const T, &'a T, const, {/* no mut */}, as_ref, {
fn is_sorted_by<F>(self, mut compare: F) -> bool fn is_sorted_by<F>(self, mut compare: F) -> bool
where where
Self: Sized, Self: Sized,
F: FnMut(&Self::Item, &Self::Item) -> Option<Ordering>, F: FnMut(&Self::Item, &Self::Item) -> bool,
{ {
self.as_slice().is_sorted_by(|a, b| compare(&a, &b)) self.as_slice().is_sorted_by(|a, b| compare(&a, &b))
} }

View File

@ -4103,23 +4103,36 @@ impl<T> [T] {
where where
T: PartialOrd, T: PartialOrd,
{ {
self.is_sorted_by(|a, b| a.partial_cmp(b)) self.is_sorted_by(|a, b| a <= b)
} }
/// Checks if the elements of this slice are sorted using the given comparator function. /// Checks if the elements of this slice are sorted using the given comparator function.
/// ///
/// Instead of using `PartialOrd::partial_cmp`, this function uses the given `compare` /// Instead of using `PartialOrd::partial_cmp`, this function uses the given `compare`
/// function to determine the ordering of two elements. Apart from that, it's equivalent to /// function to determine whether two elements are to be considered in sorted order.
/// [`is_sorted`]; see its documentation for more information.
/// ///
/// [`is_sorted`]: slice::is_sorted /// # Examples
///
/// ```
/// #![feature(is_sorted)]
///
/// assert!([1, 2, 2, 9].is_sorted_by(|a, b| a <= b));
/// assert!(![1, 2, 2, 9].is_sorted_by(|a, b| a < b));
///
/// assert!([0].is_sorted_by(|a, b| true));
/// assert!([0].is_sorted_by(|a, b| false));
///
/// let empty: [i32; 0] = [];
/// assert!(empty.is_sorted_by(|a, b| false));
/// assert!(empty.is_sorted_by(|a, b| true));
/// ```
#[unstable(feature = "is_sorted", reason = "new API", issue = "53485")] #[unstable(feature = "is_sorted", reason = "new API", issue = "53485")]
#[must_use] #[must_use]
pub fn is_sorted_by<'a, F>(&'a self, mut compare: F) -> bool pub fn is_sorted_by<'a, F>(&'a self, mut compare: F) -> bool
where where
F: FnMut(&'a T, &'a T) -> Option<Ordering>, F: FnMut(&'a T, &'a T) -> bool,
{ {
self.array_windows().all(|[a, b]| compare(a, b).map_or(false, Ordering::is_le)) self.array_windows().all(|[a, b]| compare(a, b))
} }
/// Checks if the elements of this slice are sorted using the given key extraction function. /// Checks if the elements of this slice are sorted using the given key extraction function.

View File

@ -1,4 +1,3 @@
use core::cmp::Ordering;
use core::num::NonZeroUsize; use core::num::NonZeroUsize;
/// A wrapper struct that implements `Eq` and `Ord` based on the wrapped /// A wrapper struct that implements `Eq` and `Ord` based on the wrapped
@ -402,7 +401,7 @@ fn test_is_sorted() {
// Tests for is_sorted_by // Tests for is_sorted_by
assert!(![6, 2, 8, 5, 1, -60, 1337].iter().is_sorted()); assert!(![6, 2, 8, 5, 1, -60, 1337].iter().is_sorted());
assert!([6, 2, 8, 5, 1, -60, 1337].iter().is_sorted_by(|_, _| Some(Ordering::Less))); assert!([6, 2, 8, 5, 1, -60, 1337].iter().is_sorted_by(|_, _| true));
// Tests for is_sorted_by_key // Tests for is_sorted_by_key
assert!([-2, -1, 0, 3].iter().is_sorted()); assert!([-2, -1, 0, 3].iter().is_sorted());

View File

@ -2307,7 +2307,7 @@ fn test_is_sorted() {
// Tests for is_sorted_by // Tests for is_sorted_by
assert!(![6, 2, 8, 5, 1, -60, 1337].is_sorted()); assert!(![6, 2, 8, 5, 1, -60, 1337].is_sorted());
assert!([6, 2, 8, 5, 1, -60, 1337].is_sorted_by(|_, _| Some(Ordering::Less))); assert!([6, 2, 8, 5, 1, -60, 1337].is_sorted_by(|_, _| true));
// Tests for is_sorted_by_key // Tests for is_sorted_by_key
assert!([-2, -1, 0, 3].is_sorted()); assert!([-2, -1, 0, 3].is_sorted());