//
// Cforall Version 1.0.0 Copyright (C) 2015 University of Waterloo
//
// The contents of this file are covered under the licence agreement in the
// file "LICENCE" distributed with Cforall.
//
// GcTracer.h --
//
// Author           : Aaron B. Moss
// Created On       : Thu Mar 15 14:47:00 2018
// Last Modified By : Aaron B. Moss
// Last Modified On : Thu Mar 15 14:47:00 2018
// Update Count     : 1
//

#pragma once

#include <list>

#include "BaseSyntaxNode.h"
#include "Expression.h"

#include "Common/GC.h"
#include "Common/PassVisitor.h"

class Expression;

/// Implements `trace` method for syntax nodes
class GcTracer final : public WithShortCircuiting, public WithVisitorRef<GcTracer> {
	const GC& gc;

public:
	GcTracer( const GC& gc ) : gc(gc) {}

	void previsit( BaseSyntaxNode * node ) {
		// skip tree if already seen
		if ( node->mark == gc.mark ) {
			visit_children = false;
			return;
		}

		// mark node
		node->mark = gc.mark;
	}

	void postvisit( Expression* expr ) {
		maybeAccept( expr->env, *visitor );
	}
};

/// Traces entire translation unit recursively
static inline const GC& operator<< ( const GC& gc, const std::list<Declaration*>& translationUnit ) {
	PassVisitor<GcTracer> tracer{ gc };
	acceptAll( translationUnit, tracer );
	return gc;
}

// Local Variables: //
// tab-width: 4 //
// mode: c++ //
// compile-command: "make install" //
// End: //