mirrors/codon
1
0
Fork 0
mirror of https://github.com/exaloop/codon.git synced 2025-06-03 15:03:52 +08:00
Code Issues Projects Releases Wiki Activity

stdlib/internal/internal.codon

Browse Source
This commit is contained in:
Ishak Numanagić 2022-01-24 08:04:39 +01:00
parent 25fa9a5da5
commit eee893b32a
1 changed files with 48 additions and 35 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

83
stdlib/internal/internal.codon
View File

@ -1,10 +1,18 @@
# (c) 2022 Exaloop Inc. All rights reserved.
from internal.gc import free
@pure
@C
def seq_check_errno() -> str: pass
def seq_check_errno() -> str:
pass
from C import seq_print(str)
from C import exit(int)
@extend
class __internal__:
@pure
@ -14,7 +22,7 @@ class __internal__:
@pure
@llvm
def _tuple_getitem_llvm[T, E](t: T, idx: int) -> E:
def _tuple_getitem_llvm(t: T, idx: int, T: type, E: type) -> E:
%x = alloca {=T}
store {=T} %t, {=T}* %x
%p0 = bitcast {=T}* %x to {=E}*
@ -26,21 +34,23 @@ class __internal__:
if idx < 0:
idx += len
if idx < 0 or idx >= len:
raise IndexError("tuple index " + str(idx) + ' out of range 0..' + str(len))
raise IndexError("tuple index " + str(idx) + " out of range 0.." + str(len))
return idx
def tuple_getitem[T, E](t: T, idx: int) -> E:
return __internal__._tuple_getitem_llvm(t, __internal__.tuple_fix_index(idx, staticlen(t)), T, E)
def tuple_getitem(t: T, idx: int, T: type, E: type) -> E:
return __internal__._tuple_getitem_llvm(
t, __internal__.tuple_fix_index(idx, staticlen(t)), T, E
)
@pure
@llvm
def fn_new[T](ptr: Ptr[byte]) -> T:
def fn_new(ptr: Ptr[byte], T: type) -> T:
%0 = bitcast i8* %ptr to {=T}
ret {=T} %0
@pure
@llvm
def fn_raw[T](fn: T) -> Ptr[byte]:
def fn_raw(fn: T, T: type) -> Ptr[byte]:
%0 = bitcast {=T} %fn to i8*
ret i8* %0
@ -63,76 +73,77 @@ class __internal__:
ret i{=T} %0
def seq_assert(file: str, line: int, msg: str) -> AssertionError:
s = 'Assert failed'
if msg: s += ': ' + msg
s += ' (' + file + ':' + line.__repr__() + ')'
s = "Assert failed"
if msg:
s += ": " + msg
s += " (" + file + ":" + line.__repr__() + ")"
return AssertionError(s)
def seq_assert_test(file: str, line: int, msg: str):
def seq_assert_test(file: str, line: int, msg: str) -> void:
s = "\033[1;31mTEST FAILED:\033[0m " + file + " (line " + str(line) + ")"
if msg:
s += ": " + msg
seq_print(s + "\n")
def check_errno(prefix: str):
def check_errno(prefix: str) -> void:
msg = seq_check_errno()
if msg:
raise OSError(prefix + msg)
@pure
@llvm
def opt_tuple_new[T]() -> Optional[T]:
def opt_tuple_new(T: type) -> Optional[T]:
ret { i1, {=T} } { i1 false, {=T} undef }
@pure
@llvm
def opt_ref_new[T]() -> Optional[T]:
def opt_ref_new(T: type) -> Optional[T]:
ret i8* null
@pure
@llvm
def opt_tuple_new_arg[T](what: T) -> Optional[T]:
def opt_tuple_new_arg(what: T, T: type) -> Optional[T]:
%0 = insertvalue { i1, {=T} } { i1 true, {=T} undef }, {=T} %what, 1
ret { i1, {=T} } %0
@pure
@llvm
def opt_ref_new_arg[T](what: T) -> Optional[T]:
def opt_ref_new_arg(what: T, T: type) -> Optional[T]:
ret i8* %what
@pure
@llvm
def opt_tuple_bool[T](what: Optional[T]) -> bool:
def opt_tuple_bool(what: Optional[T], T: type) -> bool:
%0 = extractvalue { i1, {=T} } %what, 0
%1 = zext i1 %0 to i8
ret i8 %1
@pure
@llvm
def opt_ref_bool[T](what: Optional[T]) -> bool:
def opt_ref_bool(what: Optional[T], T: type) -> bool:
%0 = icmp ne i8* %what, null
%1 = zext i1 %0 to i8
ret i8 %1
@pure
@llvm
def opt_tuple_invert[T](what: Optional[T]) -> T:
def opt_tuple_invert(what: Optional[T], T: type) -> T:
%0 = extractvalue { i1, {=T} } %what, 1
ret {=T} %0
@pure
@llvm
def opt_ref_invert[T](what: Optional[T]) -> T:
def opt_ref_invert(what: Optional[T], T: type) -> T:
ret i8* %what
@pure
@llvm
def to_class_ptr[T](ptr: Ptr[byte]) -> T:
def to_class_ptr(ptr: Ptr[byte], T: type) -> T:
%0 = bitcast i8* %ptr to {=T}
ret {=T} %0
@pure
def _tuple_offsetof(x, field: Static[int]):
def _tuple_offsetof(x, field: Static[int]) -> int:
@llvm
def _llvm_offsetof(T: type, idx: Static[int], TE: type) -> int:
%a = alloca {=T}
@ -141,6 +152,7 @@ class __internal__:
%elem = ptrtoint {=TE}* %b to i64
%offset = sub i64 %elem, %base
ret i64 %offset
return _llvm_offsetof(type(x), field, type(x[field]))
def raw_type_str(p: Ptr[byte], name: str) -> str:
@ -149,21 +161,21 @@ class __internal__:
total = 1 + name.len + 4 + pstr.len + 1
buf = Ptr[byte](total)
where = 0
buf[where] = byte(60) # '<'
buf[where] = byte(60) # '<'
where += 1
str.memcpy(buf + where, name.ptr, name.len)
where += name.len
buf[where] = byte(32) # ' '
buf[where] = byte(32) # ' '
where += 1
buf[where] = byte(97) # 'a'
buf[where] = byte(97) # 'a'
where += 1
buf[where] = byte(116) # 't'
buf[where] = byte(116) # 't'
where += 1
buf[where] = byte(32) # ' '
buf[where] = byte(32) # ' '
where += 1
str.memcpy(buf + where, pstr.ptr, pstr.len)
where += pstr.len
buf[where] = byte(62) # '>'
buf[where] = byte(62) # '>'
free(pstr.ptr)
return str(buf, total)
@ -179,7 +191,7 @@ class __internal__:
i += 1
buf = Ptr[byte](total)
where = 0
buf[where] = byte(40) # '('
buf[where] = byte(40) # '('
where += 1
i = 0
while i < n:
@ -188,21 +200,21 @@ class __internal__:
if l:
str.memcpy(buf + where, s.ptr, l)
where += l
buf[where] = byte(58) # ':'
buf[where] = byte(58) # ':'
where += 1
buf[where] = byte(32) # ' '
buf[where] = byte(32) # ' '
where += 1
s = strs[i]
l = s.len
str.memcpy(buf + where, s.ptr, l)
where += l
if i < n - 1:
buf[where] = byte(44) # ','
buf[where] = byte(44) # ','
where += 1
buf[where] = byte(32) # ' '
buf[where] = byte(32) # ' '
where += 1
i += 1
buf[where] = byte(41) # ')'
buf[where] = byte(41) # ')'
return str(buf, total)
@ -210,5 +222,6 @@ class __internal__:
class Import:
name: str
file: str
def __repr__(self):
def __repr__(self) -> str:
return f"<module '{self.name}' from '{self.file}'>"