1 | #include "TranslateEnumRange.hpp" |
---|
2 | |
---|
3 | #include "AST/Pass.hpp" |
---|
4 | #include "AST/TranslationUnit.hpp" |
---|
5 | |
---|
6 | namespace ControlStruct { |
---|
7 | |
---|
8 | struct addInitType { |
---|
9 | const ast::Stmt * postvisit( const ast::ForStmt * stmt ); |
---|
10 | }; |
---|
11 | |
---|
12 | struct addInit { |
---|
13 | const ast::Stmt * postvisit( const ast::ForStmt * stmt ); |
---|
14 | }; |
---|
15 | |
---|
16 | struct translateEnumRangeCore { |
---|
17 | const ast::Stmt * postvisit( const ast::ForStmt * stmt ); |
---|
18 | }; |
---|
19 | |
---|
20 | const ast::Stmt* addInitType::postvisit( const ast::ForStmt * stmt ) { |
---|
21 | if ( stmt->range_over ) { |
---|
22 | auto typeExpr = stmt->range_over.strict_as<ast::TypeExpr>(); |
---|
23 | auto type = typeExpr->type; |
---|
24 | |
---|
25 | auto objInit = stmt->inits.front(); |
---|
26 | assert( stmt->inits.size() == 1 ); |
---|
27 | |
---|
28 | if (auto declStmt = objInit.as<ast::DeclStmt>()) { |
---|
29 | auto decl = declStmt->decl; |
---|
30 | if ( auto objDecl = decl.as<ast::ObjectDecl>()) { |
---|
31 | if ( !objDecl->type && type ) { |
---|
32 | auto objDeclWithType = ast::mutate_field( objDecl, &ast::ObjectDecl::type, type ); |
---|
33 | auto declWithType = ast::mutate_field( declStmt, &ast::DeclStmt::decl, objDeclWithType ); |
---|
34 | stmt = ast::mutate_field_index( stmt, &ast::ForStmt::inits, 0, declWithType ); |
---|
35 | } |
---|
36 | } |
---|
37 | } |
---|
38 | } |
---|
39 | return stmt; |
---|
40 | } |
---|
41 | |
---|
42 | const ast::Stmt* addInit::postvisit( const ast::ForStmt * stmt ) { |
---|
43 | if ( stmt->range_over ) { |
---|
44 | auto inits = stmt->inits; |
---|
45 | auto init = stmt->inits.front(); |
---|
46 | |
---|
47 | if (auto declStmt = init.as<ast::DeclStmt>()) { |
---|
48 | auto decl = declStmt->decl; |
---|
49 | if ( auto objDecl = decl.as<ast::ObjectDecl>()) { |
---|
50 | if ( !objDecl->init ) { |
---|
51 | auto location = stmt->location; |
---|
52 | ast::SingleInit * newInit = new ast::SingleInit( location, |
---|
53 | stmt->is_inc? |
---|
54 | ast::UntypedExpr::createCall( location, "lowerBound", {} ): |
---|
55 | ast::UntypedExpr::createCall( location, "upperBound", {} ), |
---|
56 | ast::ConstructFlag::MaybeConstruct |
---|
57 | ); |
---|
58 | auto objDeclWithInit = ast::mutate_field( objDecl, &ast::ObjectDecl::init, newInit ); |
---|
59 | auto declWithInit = ast::mutate_field( declStmt, &ast::DeclStmt::decl, objDeclWithInit ); |
---|
60 | stmt = ast::mutate_field_index( stmt, &ast::ForStmt::inits, 0, declWithInit ); |
---|
61 | } |
---|
62 | } |
---|
63 | } |
---|
64 | } |
---|
65 | return stmt; |
---|
66 | } |
---|
67 | |
---|
68 | const ast::Stmt* translateEnumRangeCore::postvisit( const ast::ForStmt * stmt ) { |
---|
69 | if ( !stmt->range_over ) return stmt; |
---|
70 | auto location = stmt->location; |
---|
71 | auto declStmt = stmt->inits.front().strict_as<ast::DeclStmt>(); |
---|
72 | auto initDecl = declStmt->decl.strict_as<ast::ObjectDecl>(); |
---|
73 | auto indexName = initDecl->name; |
---|
74 | |
---|
75 | // Both inc and dec check if the current posn less than the number of enumerator |
---|
76 | // for dec, it keeps call pred until it passes 0 (the first enumerator) and underflow, |
---|
77 | // it wraps around and become unsigned max |
---|
78 | ast::UntypedExpr * condition = ast::UntypedExpr::createCall( location, |
---|
79 | "?<=?", |
---|
80 | { new ast::NameExpr( location, indexName ), |
---|
81 | ast::UntypedExpr::createCall( location, "upperBound", {} ) }); |
---|
82 | auto increment = ast::UntypedExpr::createCall( location, |
---|
83 | stmt->is_inc? "succ": "pred", |
---|
84 | { new ast::NameExpr( location, indexName ) }); |
---|
85 | auto assig = ast::UntypedExpr::createAssign( location, new ast::NameExpr( location, indexName ), increment ); |
---|
86 | auto mut = ast::mutate_field( stmt, &ast::ForStmt::cond, condition ); |
---|
87 | mut = ast::mutate_field(stmt, &ast::ForStmt::inc, assig ); |
---|
88 | return mut; |
---|
89 | } |
---|
90 | |
---|
91 | void translateEnumRange( ast::TranslationUnit & translationUnit ) { |
---|
92 | ast::Pass<addInit>::run( translationUnit ); |
---|
93 | ast::Pass<translateEnumRangeCore>::run( translationUnit ); |
---|
94 | } |
---|
95 | } |
---|