structs and functions in ctx

[toc] / src / typeInfo.h

diff --git a/src/typeInfo.h b/src/typeInfo.h
index 39148473ffec798787384177a4677267b0312b41..89fff2a1800d7ceebaed464a9c6c6cebc6314d0d 100644 (file)
--- a/src/typeInfo.h
+++ b/src/typeInfo.h
@@ -15,7 +15,8 @@ TypeInfo typeType(const Program & p, Type t)
   TypeInfo result;\r
   result.isStruct = true;\r
   if (t.name == "int" || t.name == "float" || t.name == "double" ||\r
-      t.name == "char" || t.name == "long" || t.name == "short" || t.name == "bool")\r
+      t.name == "char" || t.name == "long" || t.name == "short" || t.name == "bool" ||\r
+      t.name == "void")\r
   {\r
     result.isStruct = false;\r
   }\r
@@ -23,7 +24,7 @@ TypeInfo typeType(const Program & p, Type t)
   return result;\r
 }\r
 \r
-TypeInfo typeExpr(const Program & p, const std::vector<string> & globalNamespace, Expr e)\r
+TypeInfo typeExpr(const Program & p, const std::vector<std::string> & globalNamespace, std::shared_ptr<Context> globalCtx, Expr e)\r
 {\r
   TypeInfo result;\r
 \r
@@ -35,7 +36,7 @@ TypeInfo typeExpr(const Program & p, const std::vector<string> & globalNamespace
     namespacePrefixes.insert(namespacePrefixes.end(),\r
       e._func.namespacePrefixes.begin(),\r
       e._func.namespacePrefixes.end());\r
-    auto f = findFunction(p, e._func.functionName, namespacePrefixes);\r
+    auto f = findFunction(e._func.functionName, e._func.namespacePrefixes, globalCtx);\r
     if (!f.has_value())\r
       throw "Unknown function";\r
     result = typeType(p, f.value().returnType);\r
@@ -43,11 +44,16 @@ TypeInfo typeExpr(const Program & p, const std::vector<string> & globalNamespace
   }\r
   case ExprType::Method:\r
   {\r
-    TypeInfo tiCaller = typeExpr(p, globalNamespace, *e._method.expr);\r
-    auto m = findStructMethod(p, e._method.methodName, tiCaller);\r
+    TypeInfo tiCaller = typeExpr(p, globalNamespace, globalCtx, *e._method.expr);\r
+    if (!tiCaller.isStruct)\r
+      throw "Calling method on non-struct";\r
+    auto s = findStruct(tiCaller.type.name, tiCaller.type.namespacePrefixes, globalCtx);\r
+    if (!s.has_value())\r
+      throw "Calling method on unknown struct";\r
+    auto m = findStructMethod(e._method.methodName, s.value());\r
     if (!m.has_value())\r
       throw "Unknown method";\r
-    result = typeType(p, m.value().returnType);\r
+    result = typeType(p, m.value().t.returnType);\r
     break;\r
   }\r
   case ExprType::Lit:\r
@@ -61,32 +67,37 @@ TypeInfo typeExpr(const Program & p, const std::vector<string> & globalNamespace
     }\r
     break;\r
   case ExprType::Paren:\r
-    result = typeExpr(p, globalNamespace, *e._paren.expr);\r
+    result = typeExpr(p, globalNamespace, globalCtx, *e._paren.expr);\r
     break;\r
   case ExprType::Dot:\r
   {\r
-    auto sm = findStructMember(p,\r
-        typeExpr(p, globalNamespace, *e._dot.expr), e._dot.identifier);\r
+    auto tiCaller = typeExpr(p, globalNamespace, globalCtx, *e._dot.expr);\r
+    if (!tiCaller.isStruct)\r
+      throw "Accessing member of non-struct";\r
+    auto s = findStruct(tiCaller.type.name, tiCaller.type.namespacePrefixes, globalCtx);\r
+    if (!s.has_value())\r
+      throw "Calling method on unknown struct";\r
+    auto sm = findStructMember(e._dot.identifier, s.value());\r
     if (!sm.has_value())\r
       throw "Unknown struct member";\r
-    result = typeType(p, sm.value().type);\r
+    result = typeType(p, sm.value().t.type);\r
     break;\r
   }\r
   case ExprType::PrefixOp:\r
-    result = typeExpr(p, globalNamespace, *e._prefixOp.expr);\r
+    result = typeExpr(p, globalNamespace, globalCtx, *e._prefixOp.expr);\r
     break;\r
   case ExprType::PostfixOp:\r
-    result = typeExpr(p, globalNamespace, *e._postfixOp.expr);\r
+    result = typeExpr(p, globalNamespace, globalCtx, *e._postfixOp.expr);\r
     break;\r
   case ExprType::BinaryOp:\r
-    result = typeExpr(p, globalNamespace, *e._binaryOp.lexpr);\r
+    result = typeExpr(p, globalNamespace, globalCtx, *e._binaryOp.lexpr);\r
     break;\r
   case ExprType::TernaryOp:\r
-    result = typeExpr(p, globalNamespace, *e._ternaryOp.rexprTrue);\r
+    result = typeExpr(p, globalNamespace, globalCtx, *e._ternaryOp.rexprTrue);\r
     break;\r
   case ExprType::Bracket:\r
   {\r
-    TypeInfo ti = typeExpr(p, globalNamespace, *e._brackets.lexpr);\r
+    TypeInfo ti = typeExpr(p, globalNamespace, globalCtx, *e._brackets.lexpr);\r
     if (!ti.type.modifiers.empty())\r
     {\r
       result = ti;\r
@@ -103,7 +114,7 @@ TypeInfo typeExpr(const Program & p, const std::vector<string> & globalNamespace
     namespacePrefixes.insert(namespacePrefixes.end(),\r
       e._identifier.namespacePrefixes.begin(),\r
       e._identifier.namespacePrefixes.end());\r
-    auto v = findVariable(p, e._identifier.identifier, namespacePrefixes);\r
+    auto v = findVariable(e._identifier.identifier, namespacePrefixes, globalCtx);\r
     if (!v.has_value())\r
       throw "Unknown variable";\r
     result = typeType(p, v.value().type);\r