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) ;
  skip ;
  st := folded_record.st ;
  acc := folded_record.acc ;

} 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 toto : (string * string) := ("foo","bar") ;

  var mylist : list(int) := list 1 ; 1 ; 1 end ;

  for x : int in list mylist
  begin
    toto.1 := "r";
    acc := acc + x ;
    st := st^"to" ;
  end

} with acc

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