From 4f479ac68028a89c8e545f72335642c42a5c4efe Mon Sep 17 00:00:00 2001
From: lemon <lsof@mailbox.org>
Date: Wed, 24 Aug 2022 19:52:42 +0200
Subject: slice and stuff

---
 src/llvm.cff | 52 +++++++++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 51 insertions(+), 1 deletion(-)

(limited to 'src/llvm.cff')

diff --git a/src/llvm.cff b/src/llvm.cff
index 5772f76..e64d2e3 100644
--- a/src/llvm.cff
+++ b/src/llvm.cff
@@ -232,16 +232,29 @@ fn genaddr(f *Fn, ex *Expr) Value {
    case Dot dot;
       let lhs = dot.lhs.ty->is(:Ptr) ? genexpr(f, dot.lhs) : genaddr(f, dot.lhs);
       assert(dot.fld.off % dot.fld.ty.align == 0, "field align");
-      let idx int = dot.fld - ex.ty.u.Agg.flds.#ptr;
+      let idx int #?;
+      if dot.lhs.ty->is(:Ptr) {
+         idx = dot.fld - dot.lhs.ty.u.Ptr.u.Agg.flds.#ptr;
+      } else {
+         idx = dot.fld - dot.lhs.ty.u.Agg.flds.#ptr;
+      }
       let addr = mktmp(mkptrtype(ex.ty));
       switch lhs.ty.u.Ptr.u.Agg.kind {
       case :Struct;
          gen("\t%v = getelementptr %t, %t %v, i32 0, i32 %d\n", addr, lhs.ty.u.Ptr, lhs.ty, lhs, idx);
       case :Union;
          gen("\t%v = bitcast %t %v to %v", addr, lhs.ty, lhs, addr.ty);
+      case else
+         assert(#f, "nyi eunion access");
       }
       return addr;
 
+   case EUTag eex;
+      let lhs = eex.ty->is(:Ptr) ? genexpr(f, eex) : genaddr(f, eex);
+      let addr = mktmp(mkptrtype(ex.ty));
+      gen("\t%v = bitcast %t %v to %v", addr, lhs.ty, lhs, addr);
+      return addr;
+
    case AggIni ini;
       let tmp = mktmp(mkptrtype(ex.ty));
       gen("\t%v = alloca %t\n", tmp, ex.ty);
@@ -642,6 +655,13 @@ fn genexpr(f *Fn, ex *Expr) Value {
       gen("\t%v = load %t, %t %v\n", tmp, ex.ty, addr.ty, addr);
       return tmp;
 
+   case EUTag;
+      let tmp = mktmp(ex.ty);
+      let addr = genaddr(f, ex);
+      gen("\t%v = load %t, %t %v\n", tmp, ex.ty, addr.ty, addr);
+      return tmp;
+
+
    case Cast it;
       return convert(f, ex.ty, it);
 
@@ -655,6 +675,36 @@ fn genexpr(f *Fn, ex *Expr) Value {
          gen("\t%v = load %t, %t %v\n", tmp, ex.ty, addr.ty, addr);
          return tmp;
       }
+
+   case Slice sl;
+      if sl.lhs.ty.u.#tag == :Arr {
+         assert(#f,"");
+      } else if sl.lhs.ty.u.#tag == :Ptr {
+         let addr = genexpr(f, sl.lhs);
+         let begin = convert(f, ty_usize, sl.begin);
+         let end = convert(f, ty_usize, sl.end);
+         let len = mktmp(ty_usize);
+         gen("\t%v = sub %t %v, %v\n", len, ty_usize, end, begin);
+         let tmp = mktmp(ex.ty);
+         gen("\t%v = insertvalue %t undef, %t %v, 0\n", tmp, ex.ty, addr.ty, addr);
+         let res = mktmp(ex.ty);
+         gen("\t%v = insertvalue %t %v, %t %v, 1\n", res, ex.ty, tmp, len.ty, len);
+         return res;
+      } else if sl.lhs.ty.u.#tag == :Slice {
+         let slice = genexpr(f, sl.lhs);
+         let begin = convert(f, ty_usize, sl.begin);
+         let end = convert(f, ty_usize, sl.end);
+         let addr = mktmp(mkptrtype(slice.ty.u.Slice));
+         gen("\t%v = extractvalue %t %v, 0", addr, ex.ty, slice);
+         let len = mktmp(ty_usize);
+         gen("\t%v = sub %t %v, %v\n", len, ty_usize, end, begin);
+         let tmp = mktmp(ex.ty);
+         gen("\t%v = insertvalue %t undef, %t %v, 0\n", tmp, ex.ty, addr.ty, addr);
+         let res = mktmp(ex.ty);
+         gen("\t%v = insertvalue %t %v, %t %v, 1\n", res, ex.ty, tmp, len.ty, len);
+         return res;
+      }
+
    case Call call;
       let lhs = genexpr(f, call.lhs);
       let fnty = &(lhs.ty->is(:Ptr) ? lhs.ty.u.Ptr : lhs.ty).u.Fn;
-- 
cgit v1.2.3