lama_byterun/stdlib/Array.expr

import List;

public fun initArray (n, f) {
  local a = makeArray (n), i;

  for i := 0, i < n, i := i + 1 do
    a [i] := f (i)
  od;

  a
}

public fun mapArray (f, a) {
  initArray (a.length, fun (i) {a[i]})
}

public fun arrayList (a) {
  local i = 0, res = [0, {}], curr = res;

  for skip, i < a.length, i := i + 1 do
    curr [1] := a [i] : curr [1];
    curr := curr [1]
  od;

  res [1]
}

public fun listArray (l) {
  local a = makeArray (l.size);

  fun inner (i, l) {
    case l of {} -> skip | h : t -> a[i] := h; inner (i+1, t) esac   
  }

  inner (0, l);
  a
}

public fun foldlArray (f, acc, a) {
  local i = 0;
  
  for skip, i < a.length, i := i+1 do
    acc := f (acc, a[i])
  od;

  acc
}

public fun foldrArray (f, acc, a) {
  local i = a.length - 1;
 
  for skip, i >= 0, i := i-1 do
    acc := f (acc, a[i])
  od;

  acc
}

public fun iterArray (f, a) {
  local i = 0;
  
  for skip, i < a.length, i := i+1 do
    f (a [i])
  od
}

public fun iteriArray (f, a) {
  local i = 0;
  
  for skip, i < a.length, i := i+1 do
    f (i, a [i])
  od
}