f12177d01e
This PR implements the necessary typeclasses so that range notation works for integers. For example, `((-2)...3).toList = [-2, -1, 0, 1, 2] : List Int`.
98 lines
1.8 KiB
Text
98 lines
1.8 KiB
Text
import Std.Data.Iterators
|
|
|
|
open Std.Iterators
|
|
|
|
/-- info: true -/
|
|
#guard_msgs in
|
|
#eval "b" ∈ ("a"...="c")
|
|
|
|
/-- info: [1, 2, 3, 4] -/
|
|
#guard_msgs in
|
|
#eval (1...=4).toList
|
|
|
|
/-- info: [2, 3] -/
|
|
#guard_msgs in
|
|
#eval (1<...4).toList
|
|
|
|
/-- info: [1, 2, 3] -/
|
|
#guard_msgs in
|
|
#eval (1...4).toList
|
|
|
|
/-- info: [1, 2, 3] -/
|
|
#guard_msgs in
|
|
#eval (1...<4).toList
|
|
|
|
/-- info: [0, 1, 2, 3, 4] -/
|
|
#guard_msgs in
|
|
#eval (*...=4).toList
|
|
|
|
/-- info: 2 -/
|
|
#guard_msgs in
|
|
#eval (1<...4).size
|
|
|
|
/-- info: true -/
|
|
#guard_msgs in
|
|
#eval (1...<1).isEmpty
|
|
|
|
/-- info: [3, 5, 7, 9, 11, 13] -/
|
|
#guard_msgs in
|
|
#eval (2<...<15).iter.stepSize 2 |>.toList
|
|
|
|
/-- info: true -/
|
|
#guard_msgs in
|
|
#eval 1 ∈ (1...=5)
|
|
|
|
def g (xs : Array Nat) : Nat := Id.run do
|
|
let mut sum := 0
|
|
for h : i in *...<xs.size do
|
|
sum := sum + xs[i]
|
|
return sum
|
|
|
|
/-- info: 6 -/
|
|
#guard_msgs in
|
|
#eval g #[1, 2, 3]
|
|
|
|
def h (n : Nat) : IO Unit := do
|
|
for i in *...n, j in 2...* do
|
|
IO.println s!"i={i}, j={j}"
|
|
|
|
example (xs : Vector Nat 16) : Id Unit := do
|
|
let mut x := 0
|
|
for _h : i in *...<(12 : Nat) do
|
|
x := x + xs[i]
|
|
|
|
example (xs : List Nat) (h : xs.length = 16) : Id Unit := do
|
|
let mut x := 0
|
|
for _h : i in *...<(12 : Nat) do
|
|
x := x + xs[i]
|
|
|
|
example (xs : Array Nat) : Id Unit := do
|
|
let mut x := 0
|
|
for _h : i in *...<(xs.size - 5 : Nat) do
|
|
x := x + xs[i]
|
|
|
|
example (xs : Array Nat) : Id Unit := do
|
|
let mut x := 0
|
|
for _h : i in *...=xs.size do
|
|
for _h' : i' in *...<i do
|
|
x := x + xs[i']
|
|
|
|
/--
|
|
info: i=0, j=2
|
|
i=1, j=3
|
|
i=2, j=4
|
|
i=3, j=5
|
|
i=4, j=6
|
|
-/
|
|
#guard_msgs in
|
|
#eval h 5
|
|
|
|
section Int
|
|
|
|
example : ((-2)...3).toList = [-2, -1, 0, 1, 2] := by native_decide
|
|
example : ((-2)...=3).toList = [-2, -1, 0, 1, 2, 3] := by native_decide
|
|
|
|
example : Std.PRange.LawfulRangeSize .closed Int := inferInstance
|
|
example : Std.PRange.LawfulRangeSize .open Int := inferInstance
|
|
|
|
end Int
|