//
// 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.
//
// Visitor.cc --
//
// Author           : Richard C. Bilson
// Created On       : Mon May 18 07:44:20 2015
// Last Modified By : Peter A. Buhr
// Last Modified On : Thu Aug 17 15:39:38 2017
// Update Count     : 29
//

#include <cassert>        // for assert
#include <list>           // for list

#include "Attribute.h"    // for Attribute
#include "Constant.h"     // for Constant
#include "Declaration.h"  // for DeclarationWithType, ObjectDecl, Declaration
#include "Expression.h"   // for Expression, ConstantExpr, ImplicitCopyCtorExpr
#include "Initializer.h"  // for Initializer, Designation, ConstructorInit
#include "Statement.h"    // for Statement, CatchStmt, AsmStmt, CompoundStmt
#include "Type.h"         // for Type, Type::ForallList, AttrType, FunctionType
#include "Visitor.h"

class Subrange;

Visitor::Visitor() {}

Visitor::~Visitor() {}

void Visitor::visit( ObjectDecl *objectDecl ) {
	maybeAccept( objectDecl->get_type(), *this );
	maybeAccept( objectDecl->get_init(), *this );
	maybeAccept( objectDecl->get_bitfieldWidth(), *this );
	acceptAll( objectDecl->attributes, *this );
}

void Visitor::visit( FunctionDecl *functionDecl ) {
	maybeAccept( functionDecl->get_functionType(), *this );
	maybeAccept( functionDecl->get_statements(), *this );
	acceptAll( functionDecl->attributes, *this );
}

void Visitor::handleAggregateDecl( AggregateDecl *aggregateDecl ) {
	acceptAll( aggregateDecl->get_parameters(), *this );
	acceptAll( aggregateDecl->get_members(), *this );
}

void Visitor::visit( StructDecl *aggregateDecl ) {
	handleAggregateDecl( static_cast< AggregateDecl* >( aggregateDecl ) );
}

void Visitor::visit( UnionDecl *aggregateDecl ) {
	handleAggregateDecl( static_cast< AggregateDecl* >( aggregateDecl ) );
}

void Visitor::visit( EnumDecl *aggregateDecl ) {
	handleAggregateDecl( static_cast< AggregateDecl* >( aggregateDecl ) );
}

void Visitor::visit( TraitDecl *aggregateDecl ) {
	handleAggregateDecl( static_cast< AggregateDecl* >( aggregateDecl ) );
}

void Visitor::handleNamedTypeDecl( NamedTypeDecl *typeDecl ) {
	acceptAll( typeDecl->get_parameters(), *this );
	acceptAll( typeDecl->get_assertions(), *this );
	maybeAccept( typeDecl->get_base(), *this );
}

void Visitor::visit( TypeDecl *typeDecl ) {
	handleNamedTypeDecl( static_cast< NamedTypeDecl* >( typeDecl ) );
	maybeAccept( typeDecl->get_init(), *this );
}

void Visitor::visit( TypedefDecl *typeDecl ) {
	handleNamedTypeDecl( static_cast< NamedTypeDecl* >( typeDecl ) );
}

void Visitor::visit( AsmDecl *asmDecl ) {
	maybeAccept( asmDecl->get_stmt(), *this );
}


void Visitor::visit( CompoundStmt *compoundStmt ) {
	acceptAll( compoundStmt->get_kids(), *this );
}

void Visitor::visit( ExprStmt *exprStmt ) {
	maybeAccept( exprStmt->get_expr(), *this );
}

void Visitor::visit( AsmStmt *asmStmt ) {
	maybeAccept( asmStmt->get_instruction(), *this );
	acceptAll( asmStmt->get_output(), *this );
	acceptAll( asmStmt->get_input(), *this );
	acceptAll( asmStmt->get_clobber(), *this );
}

void Visitor::visit( IfStmt *ifStmt ) {
	acceptAll( ifStmt->get_initialization(), *this );
	maybeAccept( ifStmt->get_condition(), *this );
	maybeAccept( ifStmt->get_thenPart(), *this );
	maybeAccept( ifStmt->get_elsePart(), *this );
}

void Visitor::visit( WhileStmt *whileStmt ) {
	maybeAccept( whileStmt->get_condition(), *this );
	maybeAccept( whileStmt->get_body(), *this );
}

void Visitor::visit( ForStmt *forStmt ) {
	acceptAll( forStmt->get_initialization(), *this );
	maybeAccept( forStmt->get_condition(), *this );
	maybeAccept( forStmt->get_increment(), *this );
	maybeAccept( forStmt->get_body(), *this );
}

void Visitor::visit( SwitchStmt *switchStmt ) {
	maybeAccept( switchStmt->get_condition(), *this );
	acceptAll( switchStmt->get_statements(), *this );
}

void Visitor::visit( CaseStmt *caseStmt ) {
	maybeAccept( caseStmt->get_condition(), *this );
	acceptAll( caseStmt->get_statements(), *this );
}

void Visitor::visit( __attribute__((unused)) BranchStmt *branchStmt ) {
}

void Visitor::visit( ReturnStmt *returnStmt ) {
	maybeAccept( returnStmt->get_expr(), *this );
}

void Visitor::visit( ThrowStmt * throwStmt ) {
	maybeAccept( throwStmt->get_expr(), *this );
	maybeAccept( throwStmt->get_target(), *this );
}

void Visitor::visit( TryStmt *tryStmt ) {
	maybeAccept( tryStmt->get_block(), *this );
	acceptAll( tryStmt->get_catchers(), *this );
	maybeAccept( tryStmt->get_finally(), *this );
}

void Visitor::visit( CatchStmt *catchStmt ) {
	maybeAccept( catchStmt->get_decl(), *this );
	maybeAccept( catchStmt->get_cond(), *this );
	maybeAccept( catchStmt->get_body(), *this );
}

void Visitor::visit( FinallyStmt *finalStmt ) {
	maybeAccept( finalStmt->get_block(), *this );
}

void Visitor::visit( WaitForStmt *waitforStmt ) {
	for( auto & clause : waitforStmt->clauses ) {
		maybeAccept( clause.target.function, *this );
		acceptAll( clause.target.arguments, *this );

		maybeAccept( clause.statement, *this );
		maybeAccept( clause.condition, *this );
	}

	maybeAccept( waitforStmt->timeout.time, *this );
	maybeAccept( waitforStmt->timeout.statement, *this );
	maybeAccept( waitforStmt->timeout.condition, *this );
	maybeAccept( waitforStmt->orelse.statement, *this );
	maybeAccept( waitforStmt->orelse.condition, *this );
}

void Visitor::visit( WithStmt * withStmt ) {
	acceptAll( withStmt->exprs, *this );
	maybeAccept( withStmt->stmt, *this );
}

void Visitor::visit( NullStmt * ) {
}

void Visitor::visit( DeclStmt *declStmt ) {
	maybeAccept( declStmt->get_decl(), *this );
}

void Visitor::visit( ImplicitCtorDtorStmt *impCtorDtorStmt ) {
	maybeAccept( impCtorDtorStmt->get_callStmt(), *this );
}


void Visitor::visit( ApplicationExpr *applicationExpr ) {
	maybeAccept( applicationExpr->get_result(), *this );
	maybeAccept( applicationExpr->get_function(), *this );
	acceptAll( applicationExpr->get_args(), *this );
}

void Visitor::visit( UntypedExpr *untypedExpr ) {
	maybeAccept( untypedExpr->get_result(), *this );
	acceptAll( untypedExpr->get_args(), *this );
}

void Visitor::visit( NameExpr *nameExpr ) {
	maybeAccept( nameExpr->get_result(), *this );
}

void Visitor::visit( AddressExpr *addressExpr ) {
	maybeAccept( addressExpr->get_result(), *this );
	maybeAccept( addressExpr->get_arg(), *this );
}

void Visitor::visit( LabelAddressExpr *labAddressExpr ) {
	maybeAccept( labAddressExpr->get_result(), *this );
}

void Visitor::visit( CastExpr *castExpr ) {
	maybeAccept( castExpr->get_result(), *this );
	maybeAccept( castExpr->get_arg(), *this );
}

void Visitor::visit( VirtualCastExpr *castExpr ) {
	maybeAccept( castExpr->get_result(), *this );
	maybeAccept( castExpr->get_arg(), *this );
}

void Visitor::visit( UntypedMemberExpr *memberExpr ) {
	maybeAccept( memberExpr->get_result(), *this );
	maybeAccept( memberExpr->get_aggregate(), *this );
	maybeAccept( memberExpr->get_member(), *this );
}

void Visitor::visit( MemberExpr *memberExpr ) {
	maybeAccept( memberExpr->get_result(), *this );
	maybeAccept( memberExpr->get_aggregate(), *this );
}

void Visitor::visit( VariableExpr *variableExpr ) {
	maybeAccept( variableExpr->get_result(), *this );
}

void Visitor::visit( ConstantExpr *constantExpr ) {
	maybeAccept( constantExpr->get_result(), *this );
	maybeAccept( constantExpr->get_constant(), *this );
}

void Visitor::visit( SizeofExpr *sizeofExpr ) {
	maybeAccept( sizeofExpr->get_result(), *this );
	if ( sizeofExpr->get_isType() ) {
		maybeAccept( sizeofExpr->get_type(), *this );
	} else {
		maybeAccept( sizeofExpr->get_expr(), *this );
	}
}

void Visitor::visit( AlignofExpr *alignofExpr ) {
	maybeAccept( alignofExpr->get_result(), *this );
	if ( alignofExpr->get_isType() ) {
		maybeAccept( alignofExpr->get_type(), *this );
	} else {
		maybeAccept( alignofExpr->get_expr(), *this );
	}
}

void Visitor::visit( UntypedOffsetofExpr *offsetofExpr ) {
	maybeAccept( offsetofExpr->get_result(), *this );
	maybeAccept( offsetofExpr->get_type(), *this );
}

void Visitor::visit( OffsetofExpr *offsetofExpr ) {
	maybeAccept( offsetofExpr->get_result(), *this );
	maybeAccept( offsetofExpr->get_type(), *this );
	maybeAccept( offsetofExpr->get_member(), *this );
}

void Visitor::visit( OffsetPackExpr *offsetPackExpr ) {
	maybeAccept( offsetPackExpr->get_result(), *this );
	maybeAccept( offsetPackExpr->get_type(), *this );
}

void Visitor::visit( AttrExpr *attrExpr ) {
	maybeAccept( attrExpr->get_result(), *this );
	if ( attrExpr->get_isType() ) {
		maybeAccept( attrExpr->get_type(), *this );
	} else {
		maybeAccept( attrExpr->get_expr(), *this );
	}
}

void Visitor::visit( LogicalExpr *logicalExpr ) {
	maybeAccept( logicalExpr->get_result(), *this );
	maybeAccept( logicalExpr->get_arg1(), *this );
	maybeAccept( logicalExpr->get_arg2(), *this );
}

void Visitor::visit( ConditionalExpr *conditionalExpr ) {
	maybeAccept( conditionalExpr->get_result(), *this );
	maybeAccept( conditionalExpr->get_arg1(), *this );
	maybeAccept( conditionalExpr->get_arg2(), *this );
	maybeAccept( conditionalExpr->get_arg3(), *this );
}

void Visitor::visit( CommaExpr *commaExpr ) {
	maybeAccept( commaExpr->get_result(), *this );
	maybeAccept( commaExpr->get_arg1(), *this );
	maybeAccept( commaExpr->get_arg2(), *this );
}

void Visitor::visit( TypeExpr *typeExpr ) {
	maybeAccept( typeExpr->get_result(), *this );
	maybeAccept( typeExpr->get_type(), *this );
}

void Visitor::visit( AsmExpr *asmExpr ) {
	maybeAccept( asmExpr->get_inout(), *this );
	maybeAccept( asmExpr->get_constraint(), *this );
	maybeAccept( asmExpr->get_operand(), *this );
}

void Visitor::visit( ImplicitCopyCtorExpr *impCpCtorExpr ) {
	maybeAccept( impCpCtorExpr->get_result(), *this );
	maybeAccept( impCpCtorExpr->get_callExpr(), *this );
	acceptAll( impCpCtorExpr->get_tempDecls(), *this );
	acceptAll( impCpCtorExpr->get_returnDecls(), *this );
	acceptAll( impCpCtorExpr->get_dtors(), *this );
}

void Visitor::visit( ConstructorExpr * ctorExpr ) {
	maybeAccept( ctorExpr->get_result(), *this );
	maybeAccept( ctorExpr->get_callExpr(), *this );
}

void Visitor::visit( CompoundLiteralExpr *compLitExpr ) {
	maybeAccept( compLitExpr->get_result(), *this );
	maybeAccept( compLitExpr->get_initializer(), *this );
}

void Visitor::visit( RangeExpr *rangeExpr ) {
	maybeAccept( rangeExpr->get_low(), *this );
	maybeAccept( rangeExpr->get_high(), *this );
}

void Visitor::visit( UntypedTupleExpr *tupleExpr ) {
	maybeAccept( tupleExpr->get_result(), *this );
	acceptAll( tupleExpr->get_exprs(), *this );
}

void Visitor::visit( TupleExpr *tupleExpr ) {
	maybeAccept( tupleExpr->get_result(), *this );
	acceptAll( tupleExpr->get_exprs(), *this );
}

void Visitor::visit( TupleIndexExpr *tupleExpr ) {
	maybeAccept( tupleExpr->get_result(), *this );
	maybeAccept( tupleExpr->get_tuple(), *this );
}

void Visitor::visit( TupleAssignExpr *assignExpr ) {
	maybeAccept( assignExpr->get_result(), *this );
	maybeAccept( assignExpr->get_stmtExpr(), *this );
}

void Visitor::visit( StmtExpr *stmtExpr ) {
	maybeAccept( stmtExpr->get_result(), *this );
	maybeAccept( stmtExpr->get_statements(), *this );
	acceptAll( stmtExpr->get_returnDecls(), *this );
	acceptAll( stmtExpr->get_dtors(), *this );
}

void Visitor::visit( UniqueExpr *uniqueExpr ) {
	maybeAccept( uniqueExpr->get_result(), *this );
	maybeAccept( uniqueExpr->get_expr(), *this );
}

void Visitor::visit( UntypedInitExpr * initExpr ) {
	maybeAccept( initExpr->get_result(), *this );
	maybeAccept( initExpr->get_expr(), *this );
	// not currently visiting initAlts, but this doesn't matter since this node is only used in the resolver.
}

void Visitor::visit( InitExpr * initExpr ) {
	maybeAccept( initExpr->get_result(), *this );
	maybeAccept( initExpr->get_expr(), *this );
	maybeAccept( initExpr->get_designation(), *this );
}


void Visitor::visit( VoidType *voidType ) {
	acceptAll( voidType->get_forall(), *this );
}

void Visitor::visit( BasicType *basicType ) {
	acceptAll( basicType->get_forall(), *this );
}

void Visitor::visit( PointerType *pointerType ) {
	acceptAll( pointerType->get_forall(), *this );
	// xxx - should PointerType visit/mutate dimension?
	maybeAccept( pointerType->get_base(), *this );
}

void Visitor::visit( ArrayType *arrayType ) {
	acceptAll( arrayType->get_forall(), *this );
	maybeAccept( arrayType->get_dimension(), *this );
	maybeAccept( arrayType->get_base(), *this );
}

void Visitor::visit( ReferenceType *refType ) {
	acceptAll( refType->get_forall(), *this );
	maybeAccept( refType->get_base(), *this );
}

void Visitor::visit( FunctionType *functionType ) {
	acceptAll( functionType->get_forall(), *this );
	acceptAll( functionType->get_returnVals(), *this );
	acceptAll( functionType->get_parameters(), *this );
}

void Visitor::handleReferenceToType( ReferenceToType *aggregateUseType ) {
	acceptAll( aggregateUseType->get_forall(), *this );
	acceptAll( aggregateUseType->get_parameters(), *this );
}

void Visitor::visit( StructInstType *aggregateUseType ) {
	handleReferenceToType( static_cast< ReferenceToType * >( aggregateUseType ) );
}

void Visitor::visit( UnionInstType *aggregateUseType ) {
	handleReferenceToType( static_cast< ReferenceToType * >( aggregateUseType ) );
}

void Visitor::visit( EnumInstType *aggregateUseType ) {
	handleReferenceToType( static_cast< ReferenceToType * >( aggregateUseType ) );
}

void Visitor::visit( TraitInstType *aggregateUseType ) {
	handleReferenceToType( static_cast< ReferenceToType * >( aggregateUseType ) );
}

void Visitor::visit( TypeInstType *aggregateUseType ) {
	handleReferenceToType( static_cast< ReferenceToType * >( aggregateUseType ) );
}

void Visitor::visit( TupleType *tupleType ) {
	acceptAll( tupleType->get_forall(), *this );
	acceptAll( tupleType->get_types(), *this );
	acceptAll( tupleType->get_members(), *this );
}

void Visitor::visit( TypeofType *typeofType ) {
	assert( typeofType->get_expr() );
	typeofType->get_expr()->accept( *this );
}

void Visitor::visit( AttrType *attrType ) {
	if ( attrType->get_isType() ) {
		assert( attrType->get_type() );
		attrType->get_type()->accept( *this );
	} else {
		assert( attrType->get_expr() );
		attrType->get_expr()->accept( *this );
	} // if
}

void Visitor::visit( VarArgsType *varArgsType ) {
	acceptAll( varArgsType->get_forall(), *this );
}

void Visitor::visit( ZeroType *zeroType ) {
	acceptAll( zeroType->get_forall(), *this );
}

void Visitor::visit( OneType *oneType ) {
	acceptAll( oneType->get_forall(), *this );
}

void Visitor::visit( Designation * designation ) {
	acceptAll( designation->get_designators(), *this );
}

void Visitor::visit( SingleInit *singleInit ) {
	singleInit->get_value()->accept( *this );
}

void Visitor::visit( ListInit *listInit ) {
	acceptAll( listInit->get_designations(), *this );
	acceptAll( listInit->get_initializers(), *this );
}

void Visitor::visit( ConstructorInit *ctorInit ) {
	maybeAccept( ctorInit->get_ctor(), *this );
	maybeAccept( ctorInit->get_dtor(), *this );
	maybeAccept( ctorInit->get_init(), *this );
}


void Visitor::visit( Subrange * ) {}


void Visitor::visit( Constant * ) {}

void Visitor::visit( Attribute * attribute ) {
	acceptAll( attribute->parameters, *this );
}

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