authorgravatar for andrew@ziglang.orgAndrew Kelley <andrew@ziglang.org> 2017-11-25 00:57:48-05:00
committergravatar for andrew@ziglang.orgAndrew Kelley <andrew@ziglang.org> 2017-11-25 00:57:48-05:00
logbf20b260ce1b76b3b3d2c8a1bc1819eb3871bb00
tree754b9f655c409405a1d39a9cb1c02dd0ac6b7e07
parent18eb3c5f90900627b0582b7e341f9db9177fcc09

translate-c supports for loops

2 files changed, 60 insertions(+), 6 deletions(-)

src/translate_c.cpp+47-6
......@@ -2119,9 +2119,6 @@ static AstNode *trans_unary_expr_or_type_trait_expr(Context *c, AstNode *block,
21192119}
21202120
21212121static AstNode *trans_do_loop(Context *c, AstNode *block, DoStmt *stmt) {
2122 stmt->getBody();
2123 stmt->getCond();
2124
21252122 AstNode *while_node = trans_create_node(c, NodeTypeWhileExpr);
21262123
21272124 AstNode *true_node = trans_create_node(c, NodeTypeBoolLiteral);
......@@ -2171,6 +2168,51 @@ static AstNode *trans_do_loop(Context *c, AstNode *block, DoStmt *stmt) {
21712168 return while_node;
21722169}
21732170
2171static AstNode *trans_for_loop(Context *c, AstNode *block, ForStmt *stmt) {
2172 AstNode *loop_block_node;
2173 AstNode *while_node = trans_create_node(c, NodeTypeWhileExpr);
2174 Stmt *init_stmt = stmt->getInit();
2175 if (init_stmt == nullptr) {
2176 loop_block_node = while_node;
2177 } else {
2178 loop_block_node = trans_create_node(c, NodeTypeBlock);
2179
2180 AstNode *vars_node = trans_stmt(c, false, loop_block_node, init_stmt, TransRValue);
2181 if (vars_node == nullptr)
2182 return nullptr;
2183 if (vars_node != skip_add_to_block_node)
2184 loop_block_node->data.block.statements.append(vars_node);
2185
2186 loop_block_node->data.block.statements.append(while_node);
2187 }
2188
2189 Stmt *cond_stmt = stmt->getCond();
2190 if (cond_stmt == nullptr) {
2191 AstNode *true_node = trans_create_node(c, NodeTypeBoolLiteral);
2192 true_node->data.bool_literal.value = true;
2193 while_node->data.while_expr.condition = true_node;
2194 } else {
2195 while_node->data.while_expr.condition = trans_stmt(c, false, loop_block_node, cond_stmt, TransRValue);
2196 if (while_node->data.while_expr.condition == nullptr)
2197 return nullptr;
2198 }
2199
2200 Stmt *inc_stmt = stmt->getInc();
2201 if (inc_stmt != nullptr) {
2202 AstNode *inc_node = trans_stmt(c, false, loop_block_node, inc_stmt, TransRValue);
2203 if (inc_node == nullptr)
2204 return nullptr;
2205 while_node->data.while_expr.continue_expr = inc_node;
2206 }
2207
2208 AstNode *child_statement = trans_stmt(c, false, loop_block_node, stmt->getBody(), TransRValue);
2209 if (child_statement == nullptr)
2210 return nullptr;
2211 while_node->data.while_expr.body = child_statement;
2212
2213 return loop_block_node;
2214}
2215
21742216static AstNode *trans_string_literal(Context *c, AstNode *block, StringLiteral *stmt) {
21752217 switch (stmt->getKind()) {
21762218 case StringLiteral::Ascii:
......@@ -2230,6 +2272,8 @@ static AstNode *trans_stmt(Context *c, bool result_used, AstNode *block, Stmt *s
22302272 return trans_unary_expr_or_type_trait_expr(c, block, (UnaryExprOrTypeTraitExpr *)stmt);
22312273 case Stmt::DoStmtClass:
22322274 return trans_do_loop(c, block, (DoStmt *)stmt);
2275 case Stmt::ForStmtClass:
2276 return trans_for_loop(c, block, (ForStmt *)stmt);
22332277 case Stmt::StringLiteralClass:
22342278 return trans_string_literal(c, block, (StringLiteral *)stmt);
22352279 case Stmt::CaseStmtClass:
......@@ -2568,9 +2612,6 @@ static AstNode *trans_stmt(Context *c, bool result_used, AstNode *block, Stmt *s
25682612 case Stmt::VAArgExprClass:
25692613 emit_warning(c, stmt->getLocStart(), "TODO handle C VAArgExprClass");
25702614 return nullptr;
2571 case Stmt::ForStmtClass:
2572 emit_warning(c, stmt->getLocStart(), "TODO handle C ForStmtClass");
2573 return nullptr;
25742615 case Stmt::GotoStmtClass:
25752616 emit_warning(c, stmt->getLocStart(), "TODO handle C GotoStmtClass");
25762617 return nullptr;
test/translate_c.zig+13
......@@ -949,6 +949,19 @@ pub fn addCases(cases: &tests.TranslateCContext) {
949949 \\ return;
950950 \\}
951951 );
952
953 cases.add("for loop",
954 \\void foo(void) {
955 \\ for (int i = 0; i < 10; i += 1) { }
956 \\}
957 ,
958 \\pub fn foo() {
959 \\ {
960 \\ var i: c_int = 0;
961 \\ while (i < 10) : (i += 1) {};
962 \\ };
963 \\}
964 );
952965}
953966
954967