42e98bd3c9
This PR changes the signature of `Array.swap`, so it takes `Nat` arguments with tactic provided bounds checking. It also renames `Array.swap!` to `Array.swapIfInBounds`.
64 lines
1.7 KiB
Text
64 lines
1.7 KiB
Text
/-
|
||
Copyright (c) 2024 Lean FRO, LLC. All rights reserved.
|
||
Released under Apache 2.0 license as described in the file LICENSE.
|
||
Authors: David Thrane Christiansen
|
||
-/
|
||
|
||
prelude
|
||
import Init.Data.Array.Basic
|
||
import Init.Data.Array.Subarray
|
||
import Init.Omega
|
||
|
||
/-
|
||
This module contains splitting operations on subarrays that crucially rely on `omega` for proof
|
||
automation. Placing them in another module breaks an import cycle, because `omega` itself uses the
|
||
array library.
|
||
-/
|
||
|
||
namespace Subarray
|
||
/--
|
||
Splits a subarray into two parts.
|
||
-/
|
||
def split (s : Subarray α) (i : Fin s.size.succ) : (Subarray α × Subarray α) :=
|
||
let ⟨i', isLt⟩ := i
|
||
have := s.start_le_stop
|
||
have := s.stop_le_array_size
|
||
have : s.start + i' ≤ s.stop := by
|
||
simp only [size] at isLt
|
||
omega
|
||
let pre := {s with
|
||
stop := s.start + i',
|
||
start_le_stop := by omega,
|
||
stop_le_array_size := by omega
|
||
}
|
||
let post := {s with
|
||
start := s.start + i'
|
||
start_le_stop := by assumption
|
||
}
|
||
(pre, post)
|
||
|
||
/--
|
||
Removes the first `i` elements of the subarray. If there are `i` or fewer elements, the resulting
|
||
subarray is empty.
|
||
-/
|
||
def drop (arr : Subarray α) (i : Nat) : Subarray α where
|
||
array := arr.array
|
||
start := min (arr.start + i) arr.stop
|
||
stop := arr.stop
|
||
start_le_stop := by omega
|
||
stop_le_array_size := arr.stop_le_array_size
|
||
|
||
/--
|
||
Keeps only the first `i` elements of the subarray. If there are `i` or fewer elements, the resulting
|
||
subarray is empty.
|
||
-/
|
||
def take (arr : Subarray α) (i : Nat) : Subarray α where
|
||
array := arr.array
|
||
start := arr.start
|
||
stop := min (arr.start + i) arr.stop
|
||
start_le_stop := by
|
||
have := arr.start_le_stop
|
||
omega
|
||
stop_le_array_size := by
|
||
have := arr.stop_le_array_size
|
||
omega
|