source: src/CodeGen/CodeGenerator.hpp@ 01afd8d

Last change on this file since 01afd8d was c92bdcc, checked in by Andrew Beach <ajbeach@…>, 18 months ago

Updated the rest of the names in src/ (except for the generated files).
  • Property mode set to 100644
File size: 6.4 KB
Line 
1//
2// Cforall Version 1.0.0 Copyright (C) 2015 University of Waterloo
3//
4// The contents of this file are covered under the licence agreement in the
5// file "LICENCE" distributed with Cforall.
6//
7// CodeGenerator.hpp --
8//
9// Author : Andrew Beach
10// Created On : Tue Oct 17 15:54:00 2023
11// Last Modified By : Andrew Beach
12// Last Modified On : Wed Oct 25 17:56:00 2023
13// Update Count : 0
14//
15
16#pragma once
17
18#include <ostream> // for ostream, operator<<
19
20#include "AST/Fwd.hpp"
21#include "AST/Pass.hpp" // for WithGuards, WithShortCircuiting, ...
22#include "CodeGen/Options.hpp" // for Options
23#include "Common/Indenter.hpp" // for Indenter
24
25
26namespace CodeGen {
27
28struct CodeGenerator final :
29 public ast::WithGuards,
30 public ast::WithShortCircuiting,
31 public ast::WithVisitorRef<CodeGenerator> {
32 CodeGenerator( std::ostream & out, Options const & options );
33
34 // Turn off visit_children for all nodes.
35 void previsit( ast::Node const * );
36 void previsit( ast::ParseNode const * );
37
38 // Error for unhandled node types.
39 void postvisit( ast::Node const * );
40
41 // Print type for all expressions.
42 void previsit( ast::Expr const * );
43
44 void postvisit( ast::StructDecl const * );
45 void postvisit( ast::FunctionDecl const * );
46 // Yes, there is one visit that does modify the ast.
47 ast::ObjectDecl const * postvisit( ast::ObjectDecl const * );
48 void postvisit( ast::UnionDecl const * );
49 void postvisit( ast::EnumDecl const * );
50 void postvisit( ast::TraitDecl const * );
51 void postvisit( ast::TypedefDecl const * );
52 void postvisit( ast::TypeDecl const * );
53 void postvisit( ast::StaticAssertDecl const * );
54
55 void postvisit( ast::Designation const * );
56 void postvisit( ast::SingleInit const * );
57 void postvisit( ast::ListInit const * );
58 void postvisit( ast::ConstructorInit const * );
59
60 void postvisit( ast::ApplicationExpr const * );
61 void postvisit( ast::UntypedExpr const * );
62 void postvisit( ast::RangeExpr const * );
63 void postvisit( ast::NameExpr const * );
64 void postvisit( ast::AddressExpr const * );
65 void postvisit( ast::LabelAddressExpr const * );
66 void postvisit( ast::CastExpr const * );
67 void postvisit( ast::KeywordCastExpr const * );
68 void postvisit( ast::VirtualCastExpr const * );
69 void postvisit( ast::UntypedMemberExpr const * );
70 void postvisit( ast::MemberExpr const * );
71 void postvisit( ast::VariableExpr const * );
72 void postvisit( ast::ConstantExpr const * );
73 void postvisit( ast::SizeofExpr const * );
74 void postvisit( ast::AlignofExpr const * );
75 void postvisit( ast::UntypedOffsetofExpr const * );
76 void postvisit( ast::OffsetofExpr const * );
77 void postvisit( ast::OffsetPackExpr const * );
78 void postvisit( ast::LogicalExpr const * );
79 void postvisit( ast::ConditionalExpr const * );
80 void postvisit( ast::CommaExpr const * );
81 void postvisit( ast::CompoundLiteralExpr const * );
82 void postvisit( ast::UniqueExpr const * );
83 void postvisit( ast::TupleAssignExpr const * );
84 void postvisit( ast::UntypedTupleExpr const * );
85 void postvisit( ast::TupleExpr const * );
86 void postvisit( ast::TupleIndexExpr const * );
87 void postvisit( ast::TypeExpr const * );
88 void postvisit( ast::DimensionExpr const * );
89 void postvisit( ast::AsmExpr const * );
90 void postvisit( ast::StmtExpr const * );
91 void postvisit( ast::ConstructorExpr const * );
92 void postvisit( ast::DeletedExpr const * );
93 void postvisit( ast::DefaultArgExpr const * );
94 void postvisit( ast::GenericExpr const * );
95
96 void postvisit( ast::CompoundStmt const * );
97 void postvisit( ast::ExprStmt const * );
98 void postvisit( ast::AsmStmt const * );
99 void postvisit( ast::DirectiveStmt const* );
100 void postvisit( ast::AsmDecl const * );
101 void postvisit( ast::DirectiveDecl const * );
102 void postvisit( ast::IfStmt const * );
103 void postvisit( ast::SwitchStmt const * );
104 void postvisit( ast::CaseClause const * );
105 void postvisit( ast::BranchStmt const * );
106 void postvisit( ast::ReturnStmt const * );
107 void postvisit( ast::ThrowStmt const * );
108 void postvisit( ast::CatchClause const * );
109 void postvisit( ast::WaitForStmt const * );
110 void postvisit( ast::WithStmt const * );
111 void postvisit( ast::WhileDoStmt const * );
112 void postvisit( ast::ForStmt const * );
113 void postvisit( ast::NullStmt const * );
114 void postvisit( ast::DeclStmt const * );
115 void postvisit( ast::ImplicitCtorDtorStmt const * );
116 void postvisit( ast::MutexStmt const * stmt );
117
118private:
119 /// Custom local implementation of endl that updates print location.
120 struct LineEnder {
121 CodeGenerator & cg;
122 LineEnder( CodeGenerator & cg ) : cg( cg ) {}
123 std::ostream & operator()( std::ostream & ) const;
124 };
125 friend std::ostream & operator<<( std::ostream & os, const LineEnder & e ) {
126 return e( os );
127 }
128
129 /// Wrapper class to help print vectors of Labels.
130 struct LabelPrinter {
131 LabelPrinter( CodeGenerator & cg ) : cg( cg ), labels( nullptr ) {}
132 LabelPrinter & operator()( std::vector<ast::Label> const & l );
133 std::ostream & operator()( std::ostream & ) const;
134 CodeGenerator & cg;
135 std::vector<ast::Label> const * labels;
136 };
137 friend std::ostream & operator<<( std::ostream & os, const LabelPrinter & p ) {
138 return p( os );
139 }
140
141 static int tabsize;
142
143 Indenter indent;
144 std::ostream & output;
145 Options options;
146 LabelPrinter printLabels;
147
148 CodeLocation currentLocation;
149 void updateLocation( CodeLocation const & to );
150
151 template<typename Iterator>
152 void genCommaList( Iterator begin, Iterator end ) {
153 if ( begin == end ) return;
154 while (true) {
155 (*begin++)->accept( *visitor );
156 if ( begin == end ) break;
157 output << ", ";
158 }
159 }
160
161public:
162 LineEnder endl;
163 void updateLocation( ast::ParseNode const * to );
164
165 template<typename T>
166 void genCommaList( std::vector<ast::ptr<T>> const & range ) {
167 genCommaList( range.begin(), range.end() );
168 }
169
170 void genAttributes( std::vector<ast::ptr<ast::Attribute>> const & );
171
172private:
173 void asmName( ast::DeclWithType const * decl );
174 void extension( ast::Decl const * );
175 void extension( ast::Expr const * );
176
177 void handleStorageClass( ast::DeclWithType const * decl );
178 void handleAggregate( ast::AggregateDecl const *, const std::string & );
179 void handleTypedef( ast::NamedTypeDecl const * type );
180 std::string mangleName( ast::DeclWithType const * decl );
181};
182
183inline bool doSemicolon( ast::Decl const * decl ) {
184 if ( auto func = dynamic_cast<ast::FunctionDecl const *>( decl ) ) {
185 return !func->stmts;
186 }
187 return true;
188}
189
190/// Returns the C-compatible name of the declaration.
191std::string genName( ast::DeclWithType const * decl );
192
193} // namespace CodeGen
Note: See TracBrowser for help on using the repository browser.