lang_2023/tests/test_code.lang

basic (Float : #Ord #Div #Str)
basic (Int : #Ord #IDiv #Str)
basic (String : #Ord #Str #CharContainer #Copy)
basic (Char : #Ord #Str #Copy)
basic (Bool : #Ord #Str #Copy)
basic (Unit : #Str #Copy)

//

decl not : Bool -> Bool
def not : x =
  (match x with
     | true -> false
     | false -> true)

decl ( && ) : Bool -> Bool -> Bool
def ( && ) : x y =
  match x with
    | true -> (
      match y with
        | true -> true
        | false -> false
    )
    | false -> false

decl ( || ) : Bool -> Bool -> Bool
def ( || ) : x y =
  match x with
    | true -> true
    | false -> (
      match y with
        | true -> true
        | false -> false
    )

//

typeclass CharContainer =
  & var size : -> Int
  & var at : Int -> Char

//

typeclass Move =
  & var ( <- ) : Move -> Unit // TODO

typeclass Copy =
  & var ( = ) : Copy -> Unit

//

typeclass (Sum : #Copy) =
  & var ( += ) : Sum -> Unit
  & var ( -= ) : Sum -> Unit
  & var ( + ) : Sum -> Sum
  & var ( - ) : Sum -> Sum
  & zero : -> Sum

namespace var Sum {
  def ( + ) : x = {
    var ans = self
    ; ans += x
    return ans
  }

  def ( - ) : x = {
    var ans = self
    ; ans -= x
    return ans
  }
}

typeclass (Mult : #Sum) =
  & var ( *= ) : Mult -> Unit
  & var ( * ) : Mult -> Mult

namespace var Mult {
  def ( * ) : x = {
    var ans = self
    ; ans *= x
    return ans
  }
}

typeclass (IDiv : #Mult) =
  & var div : IDiv -> IDiv
  & var mod : IDiv -> IDiv

namespace var IDiv {
  def mod : x = self -. x * self.div: x
}

typeclass (Div : #Mult) =
  & var ( /= ) : Div -> Unit
  & var ( / ) : Div -> Div

namespace var Div {
  def ( / ) : x = {
    var ans = self
    ; ans /= x
    return ans
  }
}

//

typeclass Eq =
  & var ( == ) : Eq -> Bool
  & var ( != ) : Eq -> Bool

namespace var Eq {
  def ( != ) : x = not: (self == x)
}

//

struct Order =
  | EQ
  | LT
  | GT

typeclass (Ord : #Eq) =
  & var compare : Ord -> Order
  & var ( < ) : Ord -> Bool
  & var ( >= ) : Ord -> Bool
  & var ( > ) : Ord -> Bool
  & var ( <= ) : Ord -> Bool

decl min ('A : #Ord) : 'A -> 'A -> 'A
def min : x y = if x < y then x else y

decl max ('A : #Ord) : 'A -> 'A -> 'A
def max : x y = if x < y then y else x

namespace var Ord {
  def compare : x =
    if self == x then $EQ
    elif self < x then $LT
    else $GT

  def ( >= ) : x = not: (self < x)
  def ( > ) : x = x < self
  def ( <= ) : x = not: (x < self)
}

//

typeclass Show =
  & var show : -> String

typeclass Read =
  & read : String -> Read

typeclass (Str : #Show #Read)

// typeclass DebugShow = // TODO
//   & debug_show : -> String

//

typeclass Default =
  & default : -> Default

//

typeclass Bounded =
  & min_bound : -> Bounded
  & max_bound : -> Bounded
  & var is_max_bound : -> Bool
  & var is_min_bound : -> Bool

//

typeclass Enum =
  & var succ : -> (Optional Enum)
  & var pred : -> (Optional Enum)
  & to_enum : Int -> Enum
  & var from_enum : -> Int

//

namespace IO {
  decl print : String -> Unit
  decl scan : -> String
  decl random : -> Int // TODO
}

//

// // bad
// typeclass Functor 'A =
//   & fmap 'B ('F : (#Functor 'B)) : ('A -> 'B) -> Functor -> 'F

// typeclass (Iterator : #Eq) =
//   & next : -> Unit
//   & prev : -> Unit
//
// typeclass Iterable ('Iter : #Iterable) =
//   & begin : -> 'Iter
//   & end : -> 'Iter

//

decl ret_one : -> Int
def ret_one = 1

decl ( -- ) : Int -> Int -> Int_0
def ( -- ) : begin end = {
  var current = begin
  return (while current < end do {
    ; current += 1
    bring current - 1
  })
}

// decl func : String -> Int -> Int
// def func : s i = {
//   ; print: s
//   var x = s
//   ; print: (i.show:)
//   return 5
// }
//
// exec main {
//   for i in 0--19 do func: "a" (i * 2 +. 3)
//   ; print: ({
//     if true then bring scan: else { ; print: "aaa" }
//     bring "nothing"
//     ; print: "aaa"
//   })
// }

decl scan_int : -> Int
def scan_int = Int.read: (IO.scan:)

decl print_int : Int -> Unit
def print_int : x = IO.print: (x.show:)

decl scan_anything ('A : #Read) : -> 'A
def scan_anything = 'A.read: (IO.scan:)

decl print_anything ('A : #Show) : 'A -> Unit
def print_anything : x = IO.print: (x.show:)

/*
struct Array 'A = & data : 'A_0

namespace Array {
  decl construct : 'A_0 -> Array
  def construct: x = $(Array 'A) & data = x
} // TODO: construct decl + default def -> segfault
*/

exec main {
  var n = scan_anything Int:
  var a = for _ in 0--n do scan_int:
  ; print_anything Int: n
}