source: src/SynTree/GcTracer.h @ 5af7306

new-envwith_gc
Last change on this file since 5af7306 was 5af7306, checked in by Aaron Moss <a3moss@…>, 4 years ago

Assorted bug fixes

  • Property mode set to 100644
File size: 2.5 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// GcTracer.h --
8//
9// Author           : Aaron B. Moss
10// Created On       : Thu Mar 15 14:47:00 2018
11// Last Modified By : Aaron B. Moss
12// Last Modified On : Thu Mar 15 14:47:00 2018
13// Update Count     : 1
14//
15
16#pragma once
17
18#include <list>
19
20#include "BaseSyntaxNode.h"
21#include "Expression.h"
22#include "Type.h"
23
24#include "Common/GC.h"
25#include "Common/PassVisitor.h"
26
27class Expression;
28
29/// Implements `trace` method for syntax nodes
30class GcTracer final : public WithShortCircuiting, public WithVisitorRef<GcTracer> {
31        const GC& gc;
32
33public:
34        GcTracer( const GC& gc ) : gc(gc) {}
35
36        // mark node and children
37
38        void previsit( BaseSyntaxNode * node ) {
39                // skip tree if already seen
40                if ( node->mark == gc.mark ) {
41                        visit_children = false;
42                        return;
43                }
44
45                // mark node
46                node->mark = gc.mark;
47        }
48
49        // add visits left out by PassVisitor
50
51        void postvisit( Constant* con ) {
52                maybeAccept( con->get_type(), *visitor );
53        }
54
55        void postvisit( DeclarationWithType* decl ) {
56                maybeAccept( decl->asmName, *visitor );
57        }
58
59private:
60        void visit( InferredParams& inferParams ) {
61                for ( auto& entry : inferParams ) {
62                        maybeAccept( entry.second.actualType, *visitor );
63                        maybeAccept( entry.second.formalType, *visitor );
64                        maybeAccept( entry.second.expr, *visitor );
65                        visit( *entry.second.inferParams );
66                }
67        }
68
69public:
70        void postvisit( Expression* expr ) {
71                maybeAccept( expr->env, *visitor );
72                visit( expr->inferParams );
73        }
74
75        void postvisit( UntypedExpr* expr ) {
76                postvisit( static_cast<Expression*>(expr) );
77                maybeAccept( expr->function, *visitor );
78        }
79
80        void postvisit( VariableExpr* expr ) {
81                postvisit( static_cast<Expression*>(expr) );
82                maybeAccept( expr->var, *visitor );  // not in PassVisitor because it causes cycle
83        }
84
85        void postvisit( Type* type ) {
86                acceptAll( type->attributes, *visitor );
87        }
88
89        void postvisit( PointerType* type ) {
90                postvisit( static_cast<Type*>(type) );
91                maybeAccept( type->dimension, *visitor );
92        }
93};
94
95/// Traces entire translation unit recursively
96static inline const GC& operator<< ( const GC& gc, const std::list<Declaration*>& translationUnit ) {
97        PassVisitor<GcTracer> tracer{ gc };
98        acceptAll( const_cast<std::list<Declaration*>&>( translationUnit ), tracer );
99        return gc;
100}
101
102// Local Variables: //
103// tab-width: 4 //
104// mode: c++ //
105// compile-command: "make install" //
106// End: //
Note: See TracBrowser for help on using the repository browser.