Consider:
val example = "1234567"
fn digit(c: char): int =
case- c of
| '0' => 0 | '1' => 1 | '2' => 2 | '3' => 3 | '4' => 4
| '5' => 5 | '6' => 6 | '7' => 7 | '8' => 8 | '9' => 9
fn f(): int = loop(0, 0) where {
fun loop(i: int, acc: int): int =
if example[i] = '\000' then acc else
loop(i + 1, acc + digit(example[i]))
}
implement main0() = () where {
val () = println!("f: ", f())
}
This (tries to) loop over the indices of a string, summing the characters of the string as digits. I've solved several similar problems with .foldleft
and with streamize_string_char
, but the actual task requires math on the indexes themselves (i.e., instead of using every char, it should only use a char if the char at i+10 is as an even-numbered digit).
Actually the math is relevant, because it seems to force $UNSAFE.cast2int
, since there's no division operator for the result of strlen(input)
:
fn day2(): uint = loop(input, 0, 0) where {
val len = $UNSAFE.cast2int(strlen(input))
fn nextindex(i: int): int = (i + len/2) mod len
fn get(i: int): char = input[i] // <-- also broken at this point
// this next line is just me slowly going mad
fun loop{n:int}{i:nat | i <= n}(s: string(n), i: size_t(i), acc: uint): uint =
if i >= len then acc else
if s[i] = s[nextindex(i)] then loop(i+1, acc + digit(s[i])) else
loop(i+1, acc)
}
How should f()
be written above? Please give me an example of a function that loops over the indexes of a string and fetches chars by index from the string. Again I don't need a solution like
typedef charint = (char, int)
fn day1(): int = sum where {
val lastchar = input[strlen(input)-1]
val res = input.foldleft(TYPE{charint})((lastchar, 0), (lam((last, sum): charint, c: char) =>
if last = c then (c, sum + digit(c)) else (c, sum)))
val sum = res.1
}
because I need to test properties based on the indices.
EDIT:
Well I finally came up with some kind of solution, but look at how absurd it is. There must be a right and proper ATS way to do this.
#include "share/atspre_staload.hats"
val example = "1234567"
fn digit(c: char): int =
case- c of
| '0' => 0 | '1' => 1 | '2' => 2 | '3' => 3 | '4' => 4
| '5' => 5 | '6' => 6 | '7' => 7 | '8' => 8 | '9' => 9
fn f(): int = loop(0, 0) where {
fn get(i: int): char = loop(i, string2ptr(example)) where {
fun loop(i: int, s: ptr): char =
if i > 0 then loop(i-1, ptr0_succ<char>(s)) else
$UNSAFE.ptr0_get<char>(s)
}
fun loop(i: int, acc: int): int =
if get(i) = '\000' then acc else
loop(i + 1, acc + digit(get(i)))
}
implement main0() = () where {
val () = println!("f: ", f())
}
Outputs:
f: 28
EDIT2:
less absurd:
...
val p = string2ptr(example)
fn get(i: int): char = $UNSAFE.ptr0_get<char>(add_ptr_bsz(p, g0int2uint(i) * sizeof<char>))
...
EDIT3:
I can use string[i]
again with
overload + with add_ptr_bsz
fn string_get_at(str, i) = $UNSAFE.ptr0_get<charNZ>(string2ptr(str)+g0int2uint(i))
overload [] with string_get_at
which is almost identical to what I see in prelude/DATS/string.dats ... what's the problem?
Okay, the following implementation of day2
is safe:
fn
day2
(input: string): uint = let
val
[n:int]
input = g1ofg0(input)
val n0 = strlen(input)
val n0 = sz2i(n0) // int(n)
fun
nextindex
(
i: natLt(n)
) : natLt(n) = nmod(i + n0/2, n0)
fun
loop(i: natLte(n), acc: uint): uint =
if i >= n0 then acc else
(
if input[i] = input[nextindex(i)]
then loop(i+1, acc + digit2uint(input[i]))
else loop(i+1, acc)
)
in
loop(0, 0u)
end // end of [day2]