Trans - Coming along

1 files changed, 159 insertions, 13 deletions

diff --git a/src/trans/codegen_c.cpp b/src/trans/codegen_c.cpp
index 6d331e65..d3c14ef5 100644
--- a/src/trans/codegen_c.cpp
+++ b/src/trans/codegen_c.cpp
@@ -10,18 +10,34 @@
 #include <fstream>
 #include <hir/hir.hpp>
 #include <mir/mir.hpp>
+#include <hir_typeck/static.hpp>
 
 namespace {
     class CodeGenerator_C:
         public CodeGenerator
     {
         const ::HIR::Crate& m_crate;
+        ::StaticTraitResolve    m_resolve;
         ::std::ofstream m_of;
     public:
         CodeGenerator_C(const ::HIR::Crate& crate, const ::std::string& outfile):
             m_crate(crate),
+            m_resolve(crate),
             m_of(outfile)
         {
+            m_of
+                << "/*\n"
+                << " * AUTOGENERATED by mrustc\n"
+                << " */\n"
+                << "#include <stddef.h>\n"
+                << "#include <stdint.h>\n"
+                << "#include <stdbool.h>\n"
+                << "typedef uint32_t CHAR;\n"
+                << "typedef struct { } tUNIT;\n"
+                << "typedef struct { char* PTR; size_t META; } STR_PTR;\n"
+                << "typedef struct { void* PTR; size_t META; } SLICE_PTR;\n"
+                << "typedef struct { void* PTR; void* META; } TRAITOBJ_PTR;\n"
+                << "\n";
         }
         
         ~CodeGenerator_C() {}
@@ -30,13 +46,143 @@ namespace {
         {
         }
         
-        void emit_struct(const ::HIR::GenericPath& p, const ::HIR::Struct& item) override
+        void emit_type(const ::HIR::TypeRef& ty) override
         {
+            TU_IFLET( ::HIR::TypeRef::Data, ty.m_data, Tuple, te,
+                if( te.size() > 0 )
+                {
+                    m_of << "typedef struct {\n";
+                    for(unsigned int i = 0; i < te.size(); i++)
+                    {
+                        m_of << "\t";
+                        emit_ctype(te[i]);
+                        m_of << " _" << i << ";\n";
+                    }
+                    // TODO: Fields.
+                    m_of << "} "; emit_ctype(ty); m_of << ";\n";
+                }
+            )
+            else TU_IFLET( ::HIR::TypeRef::Data, ty.m_data, Function, te,
+                m_of << "typedef ";
+                // TODO: ABI marker, need an ABI enum?
+                emit_ctype(*te.m_rettype); m_of << " (*"; emit_ctype(ty); m_of << ")(";
+                if( te.m_arg_types.size() == 0 )
+                {
+                    m_of << "void)";
+                }
+                else
+                {
+                    for(unsigned int i = 0; i < te.m_arg_types.size(); i ++)
+                    {
+                        if(i != 0)  m_of << ",";
+                        m_of << " ";
+                        emit_ctype(te.m_arg_types[i]);
+                    }
+                    m_of << " )";
+                }
+                m_of << ";\n";
+            )
+            else {
+            }
+        }
+        
+        void emit_struct(const Span& sp, const ::HIR::GenericPath& p, const ::HIR::Struct& item) override
+        {
+            ::HIR::TypeRef  tmp;
+            auto monomorph = [&](const auto& x)->const auto& {
+                if( monomorphise_type_needed(x) ) {
+                    tmp = monomorphise_type(sp, item.m_params, p.m_params, x);
+                    m_resolve.expand_associated_types(sp, tmp);
+                    return tmp;
+                }
+                else {
+                    return x;
+                }
+                };
+            auto emit_struct_fld_ty = [&](const auto& ty) {
+                TU_IFLET(::HIR::TypeRef::Data, ty.m_data, Slice, te,
+                    emit_ctype( monomorph(*te.inner) );
+                    m_of << "[]";
+                )
+                else {
+                    emit_ctype( monomorph(ty) );
+                }
+                };
             m_of << "struct s_" << Trans_Mangle(p) << " {\n";
-            m_of << "}\n";
+            TU_MATCHA( (item.m_data), (e),
+            (Unit,
+                ),
+            (Tuple,
+                for(unsigned int i = 0; i < e.size(); i ++)
+                {
+                    const auto& fld = e[i];
+                    m_of << "\t";
+                    emit_struct_fld_ty(fld.ent);
+                    m_of << " _" << i << ";\n";
+                }
+                ),
+            (Named,
+                for(unsigned int i = 0; i < e.size(); i ++)
+                {
+                    const auto& fld = e[i].second;
+                    m_of << "\t";
+                    emit_struct_fld_ty( fld.ent );
+                    m_of << " _" << i << ";\n";
+                }
+                )
+            )
+            m_of << "};\n";
         }
         //virtual void emit_union(const ::HIR::GenericPath& p, const ::HIR::Union& item);
-        //virtual void emit_enum(const ::HIR::GenericPath& p, const ::HIR::Enum& item);
+        void emit_enum(const Span& sp, const ::HIR::GenericPath& p, const ::HIR::Enum& item) override
+        {
+            ::HIR::TypeRef  tmp;
+            auto monomorph = [&](const auto& x)->const auto& {
+                if( monomorphise_type_needed(x) ) {
+                    tmp = monomorphise_type(sp, item.m_params, p.m_params, x);
+                    m_resolve.expand_associated_types(sp, tmp);
+                    return tmp;
+                }
+                else {
+                    return x;
+                }
+                };
+            m_of << "struct e_" << Trans_Mangle(p) << " {\n";
+            m_of << "\tunsigned int TAG;\n";
+            m_of << "\tunion {\n";
+            for(unsigned int i = 0; i < item.m_variants.size(); i ++)
+            {
+                m_of << "\t\tstruct {\n";
+                TU_MATCHA( (item.m_variants[i].second), (e),
+                (Unit,
+                    ),
+                (Value,
+                    // TODO: omit
+                    ),
+                (Tuple,
+                    for(unsigned int i = 0; i < e.size(); i ++)
+                    {
+                        const auto& fld = e[i];
+                        m_of << "\t\t\t";
+                        emit_ctype( monomorph(fld.ent) );
+                        m_of << " _" << i << ";\n";
+                    }
+                    ),
+                (Struct,
+                    for(unsigned int i = 0; i < e.size(); i ++)
+                    {
+                        const auto& fld = e[i];
+                        m_of << "\t\t\t";
+                        emit_ctype( monomorph(fld.second.ent) );
+                        m_of << " _" << i << ";\n";
+                    }
+                    )
+                )
+                m_of << "\t\t} var_" << i << ";\n";
+            }
+            m_of << "\t} DATA;\n";
+            m_of << "};\n";
+        }
         
         //virtual void emit_static_ext(const ::HIR::Path& p);
         //virtual void emit_static_local(const ::HIR::Path& p, const ::HIR::Static& item, const Trans_Params& params);
@@ -257,7 +403,7 @@ namespace {
                         if(j != 0)  m_of << ",";
                         m_of << " "; emit_lvalue(e.args[j]);
                     }
-                    m_of << " )\n";
+                    m_of << " );\n";
                     m_of << "\tgoto bb" << e.ret_block << ";\n";
                     ),
                 (CallPath,
@@ -266,7 +412,7 @@ namespace {
                         if(j != 0)  m_of << ",";
                         m_of << " "; emit_lvalue(e.args[j]);
                     }
-                    m_of << " )\n";
+                    m_of << " );\n";
                     m_of << "\tgoto bb" << e.ret_block << ";\n";
                     )
                 )
@@ -318,8 +464,9 @@ namespace {
                 m_of << "._" << e.field_index;
                 ),
             (Deref,
-                m_of << "*";
+                m_of << "(*";
                 emit_lvalue(*e.val);
+                m_of << ")";
                 ),
             (Index,
                 m_of << "(";
@@ -330,7 +477,7 @@ namespace {
                 ),
             (Downcast,
                 emit_lvalue(*e.val);
-                m_of << ".var_" << e.variant_index;
+                m_of << ".DATA.var_" << e.variant_index;
                 )
             )
         }
@@ -345,8 +492,8 @@ namespace {
             (Primitive,
                 switch(te)
                 {
-                case ::HIR::CoreType::Usize:    m_of << "size_t";   break;
-                case ::HIR::CoreType::Isize:    m_of << "ssize_t";  break;
+                case ::HIR::CoreType::Usize:    m_of << "uintptr_t";   break;
+                case ::HIR::CoreType::Isize:    m_of << "intptr_t";  break;
                 case ::HIR::CoreType::U8:  m_of << "uint8_t"; break;
                 case ::HIR::CoreType::I8:  m_of << "int8_t"; break;
                 case ::HIR::CoreType::U16: m_of << "uint16_t"; break;
@@ -360,7 +507,7 @@ namespace {
                 case ::HIR::CoreType::F64: m_of << "double"; break;
                 
                 case ::HIR::CoreType::Bool: m_of << "bool"; break;
-                case ::HIR::CoreType::Char: m_of << "uin32_t";  break;
+                case ::HIR::CoreType::Char: m_of << "CHAR";  break;
                 case ::HIR::CoreType::Str:
                     BUG(Span(), "Raw str");
                 }
@@ -394,7 +541,7 @@ namespace {
                 BUG(Span(), "ErasedType in trans - " << ty);
                 ),
             (Array,
-                m_of << "("; emit_ctype(*te.inner); m_of << ")[" << te.size_val << "]";
+                emit_ctype(*te.inner); m_of << "[" << te.size_val << "]";
                 ),
             (Slice,
                 BUG(Span(), "Raw slice object - " << ty);
@@ -439,8 +586,7 @@ namespace {
                 }
                 ),
             (Function,
-                m_of << "void(void)*";
-                //TODO(Span(), "Emit function pointer type - " << ty);
+                m_of << "t_" << Trans_Mangle(ty);
                 ),
             (Closure,
                 BUG(Span(), "Closure during trans - " << ty);