ligo/src/test/contracts/loop.ligo

// Test while loops in PascaLIGO

function counter (var n : nat) : nat is block {
  var i : nat := 0n ;
  while (i < n) block {
    i := i + 1n ;
  }
} with i

function while_sum (var n : nat) : nat is block {
  var i : nat := 0n ;
  var r : nat := 0n ;
  while (i < n) block {
    i := i + 1n ;
    r := r + i ;
  }
} with r

function for_sum (var n : nat) : int is block {
  var acc : int := 0 ;
  for i := 1 to int(n)
    begin 
      acc := acc + i ;
    end
} with acc 


function lamby (var accu : (record st : string ; acc : int end) ; var i : int )
  : (record st : string ; acc : int end) is block {
    accu.acc := accu.acc + i ;
    accu.st := accu.st ^ "to" ;
} with accu

function for_collection (var nee : unit; var nuu : unit) : (int * string) is block {
  var acc : int := 0 ;
  var st : string := "to" ;
  var mylist : list(int) := list 1 ; 1 ; 1 end ;

  var init_record : (record st : string; acc : int end ) :=
    record st = st; acc = acc; end;
  var folded_record : (record st : string; acc : int end ) :=
    list_fold(mylist , init_record , lamby) ;
} with (folded_record.acc , folded_record.st)

// function for_collection_ (var nee : unit; var nuu : unit) : (int * string) is block {
//   var acc : int := 0 ;
//   var st : string := "to" ;
//   var mylist : list(int) := list 1 ; 1 ; 1 end ;

//   for x : int in list mylist
//   begin
//     acc := acc + x ;
//     st := st^"to" ;
//   end

// } with acc

function dummy (const n : nat) : nat is block {
  while (False) block { skip }
} with n