caa0eacea8
This PR adds range support to`BitVec` and the `UInt*` types. This means that it is now possible to write, for example, `for i in (1 : UInt8)...5 do`, in order to loop over the values 1, 2, 3 and 4 of type `UInt8`.
124 lines
3 KiB
Text
124 lines
3 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
|
|
|
|
end Int
|
|
|
|
section UInt
|
|
|
|
example : ((1 : UInt8)...3).toList = [1, 2] := by native_decide
|
|
example : ((-1 : UInt8)...3).toList = [] := by native_decide -- 255 ≤ x < 3 is impossible
|
|
example : ((1 : UInt8)...=3).toList = [1, 2, 3] := by native_decide
|
|
example : ((250 : UInt8)...-1).toList = [250, 251, 252, 253, 254] := by native_decide
|
|
|
|
example : ((1 : UInt16)...3).toList = [1, 2] := by native_decide
|
|
example : ((-1 : UInt16)...3).toList = [] := by native_decide
|
|
example : ((1 : UInt16)...=3).toList = [1, 2, 3] := by native_decide
|
|
example : ((-4 : UInt16)...-1).toList = [-4, -3, -2] := by native_decide
|
|
|
|
example : ((1 : UInt32)...3).toList = [1, 2] := by native_decide
|
|
example : ((-1 : UInt32)...3).toList = [] := by native_decide
|
|
example : ((1 : UInt32)...=3).toList = [1, 2, 3] := by native_decide
|
|
example : ((-4 : UInt32)...-1).toList = [-4, -3, -2] := by native_decide
|
|
|
|
example : ((1 : UInt64)...3).toList = [1, 2] := by native_decide
|
|
example : ((-1 : UInt64)...3).toList = [] := by native_decide
|
|
example : ((1 : UInt64)...=3).toList = [1, 2, 3] := by native_decide
|
|
example : ((-4 : UInt64)...-1).toList = [-4, -3, -2] := by native_decide
|
|
|
|
example : ((1 : USize)...3).toList = [1, 2] := by native_decide
|
|
example : ((-1 : USize)...3).toList = [] := by native_decide
|
|
example : ((1 : USize)...=3).toList = [1, 2, 3] := by native_decide
|
|
example : ((-4 : USize)...-1).toList = [-4, -3, -2] := by native_decide
|
|
|
|
end UInt
|