source: src/SynTree/FunctionDecl.cc @ 8a78dd3

ADTaaron-thesisarm-ehast-experimentalcleanup-dtorsdeferred_resndemanglerenumforall-pointer-decayjacob/cs343-translationjenkins-sandboxnew-astnew-ast-unique-exprnew-envno_listpersistent-indexerpthread-emulationqualifiedEnumresolv-newwith_gc
Last change on this file since 8a78dd3 was 50377a4, checked in by Rob Schluntz <rschlunt@…>, 7 years ago

Refactor tree print code to use Indenter

  • Property mode set to 100644
File size: 3.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// FunctionDecl.cc --
8//
9// Author           : Richard C. Bilson
10// Created On       : Mon May 18 07:44:20 2015
11// Last Modified By : Peter A. Buhr
12// Last Modified On : Thu Mar 16 08:33:41 2017
13// Update Count     : 74
14//
15
16#include <cassert>               // for assert
17#include <list>                  // for list
18#include <ostream>               // for operator<<, ostream, basic_ostream
19#include <string>                // for operator<<, string, char_traits, ope...
20
21#include "Attribute.h"           // for Attribute
22#include "CodeGen/FixMain.h"     // for FixMain
23#include "Common/utility.h"      // for maybeClone, printAll
24#include "Declaration.h"         // for FunctionDecl, FunctionDecl::Parent
25#include "Parser/LinkageSpec.h"  // for Spec, linkageName, Cforall
26#include "Statement.h"           // for CompoundStmt
27#include "Type.h"                // for Type, FunctionType, Type::FuncSpecif...
28#include "VarExprReplacer.h"
29
30extern bool translation_unit_nomain;
31
32FunctionDecl::FunctionDecl( const std::string &name, Type::StorageClasses scs, LinkageSpec::Spec linkage, FunctionType *type, CompoundStmt *statements, std::list< Attribute * > attributes, Type::FuncSpecifiers fs )
33        : Parent( name, scs, linkage, attributes, fs ), type( type ), statements( statements ) {
34        // hack forcing the function "main" to have Cforall linkage to replace main even if it is inside an extern
35        if ( name == "main" ) {
36                set_linkage( CodeGen::FixMain::mainLinkage() );
37        } // if
38}
39
40FunctionDecl::FunctionDecl( const FunctionDecl &other )
41                : Parent( other ), type( maybeClone( other.type ) ), statements( maybeClone( other.statements ) ) {
42
43        VarExprReplacer::DeclMap declMap;
44        for ( auto p : group_iterate( other.type->parameters, type->parameters ) ) {
45                declMap[ std::get<0>(p) ] = std::get<1>(p);
46        }
47        for ( auto p : group_iterate( other.type->returnVals, type->returnVals ) ) {
48                declMap[ std::get<0>(p) ] = std::get<1>(p);
49        }
50        if ( ! declMap.empty() ) {
51                VarExprReplacer replacer( declMap );
52                accept( replacer );
53        }
54}
55
56FunctionDecl::~FunctionDecl() {
57        delete type;
58        delete statements;
59}
60
61FunctionDecl * FunctionDecl::newFunction( const std::string & name, FunctionType * type, CompoundStmt * statements ) {
62        return new FunctionDecl( name, Type::StorageClasses(), LinkageSpec::C, type, statements );
63}
64
65void FunctionDecl::print( std::ostream &os, Indenter indent ) const {
66        using std::endl;
67        using std::string;
68
69        if ( name != "" ) {
70                os << name << ": ";
71        } // if
72        if ( linkage != LinkageSpec::Cforall ) {
73                os << LinkageSpec::linkageName( linkage ) << " ";
74        } // if
75
76        printAll( attributes, os, indent );
77
78        get_storageClasses().print( os );
79        get_funcSpec().print( os );
80
81        if ( type ) {
82                type->print( os, indent );
83        } else {
84                os << "untyped entity ";
85        } // if
86
87        if ( statements ) {
88                os << indent << "... with body " << endl << indent+1;
89                statements->print( os, indent+1 );
90        } // if
91}
92
93void FunctionDecl::printShort( std::ostream &os, Indenter indent ) const {
94        using std::endl;
95        using std::string;
96
97        if ( name != "" ) {
98                os << name << ": ";
99        } // if
100
101        get_storageClasses().print( os );
102        get_funcSpec().print( os );
103
104        if ( type ) {
105                type->print( os, indent );
106        } else {
107                os << "untyped entity ";
108        } // if
109}
110
111// Local Variables: //
112// tab-width: 4 //
113// mode: c++ //
114// compile-command: "make install" //
115// End: //
Note: See TracBrowser for help on using the repository browser.