Changes in src/AST/Decl.cpp [37273c8:c36a419]

File:

• src/AST/Decl.cpp (modified) (1 diff)

src/AST/Decl.cpp

@@ -41 +41 @@
 
 FunctionDecl::FunctionDecl( const CodeLocation & loc, const std::string & name,
+        std::vector<ptr<TypeDecl>>&& forall,
         std::vector<ptr<DeclWithType>>&& params, std::vector<ptr<DeclWithType>>&& returns,
         CompoundStmt * stmts, Storage::Classes storage, Linkage::Spec linkage,
         std::vector<ptr<Attribute>>&& attrs, Function::Specs fs, ArgumentFlag isVarArgs )
-: FunctionDecl( loc, name, {}, {}, std::move(params), std::move(returns),
-                stmts, storage, linkage, std::move(attrs), fs, isVarArgs ) {
+: DeclWithType( loc, name, storage, linkage, std::move(attrs), fs ),
+        type_params(std::move(forall)), assertions(),
+        params(std::move(params)), returns(std::move(returns)), stmts( stmts ) {
+        FunctionType * ftype = new FunctionType( isVarArgs );
+        for (auto & param : this->params) {
+                ftype->params.emplace_back(param->get_type());
+        }
+        for (auto & ret : this->returns) {
+                ftype->returns.emplace_back(ret->get_type());
+        }
+        for (auto & tp : this->type_params) {
+                ftype->forall.emplace_back(new TypeInstType(tp));
+                for (auto & ap: tp->assertions) {
+                        ftype->assertions.emplace_back(new VariableExpr(loc, ap));
+                }
+        }
+        this->type = ftype;
 }