//
// Cforall Version 1.0.0 Copyright (C) 2018 University of Waterloo
//
// The contents of this file are covered under the licence agreement in the
// file "LICENCE" distributed with Cforall.
//
// EliminateTypedef.cpp -- Removes TypedefDecl nodes from the AST.
//
// Author           : Andrew Beach
// Created On       : Wed Apr 20 16:37:00 2022
// Last Modified By : Andrew Beach
// Last Modified On : Mon Jul 11 16:30:00 2022
// Update Count     : 1
//

#include "Validate/EliminateTypedef.hpp"

#include <algorithm>

#include "AST/Decl.hpp"
#include "AST/Pass.hpp"
#include "AST/Stmt.hpp"
#include "Common/utility.h"

namespace Validate {

namespace {

struct EliminateTypedefCore {
	// Remove typedefs from inside aggregates.
	ast::StructDecl const * previsit( ast::StructDecl const * decl );
	ast::UnionDecl const * previsit( ast::UnionDecl const * decl );
	// Remove typedefs from statement lists.
	ast::CompoundStmt const * previsit( ast::CompoundStmt const * stmt );
	// Remove typedefs from control structure initializers.
	ast::IfStmt const * previsit( ast::IfStmt const * stmt );
	ast::ForStmt const * previsit( ast::ForStmt const * stmt );
	ast::WhileDoStmt const * previsit( ast::WhileDoStmt const * stmt );
};

bool isTypedef( ast::ptr<ast::Decl> const & decl ) {
	return (nullptr != decl.as<ast::TypedefDecl>());
}

bool isTypedefStmt( ast::ptr<ast::Stmt> const & stmt ) {
	if ( auto declStmt = stmt.as<ast::DeclStmt>() ) {
		return isTypedef( declStmt->decl );
	}
	return false;
}

template<typename node_t, typename super_t, typename field_t, typename pred_t>
node_t const * field_erase_if( node_t const * node, field_t super_t::*field, pred_t && pred) {
	node_t * mut = ast::mutate( node );
	erase_if( mut->*field, pred );
	return mut;
}

ast::StructDecl const * EliminateTypedefCore::previsit( ast::StructDecl const * decl ) {
	return field_erase_if( decl, &ast::StructDecl::members, isTypedef );
}

ast::UnionDecl const * EliminateTypedefCore::previsit( ast::UnionDecl const * decl ) {
	return field_erase_if( decl, &ast::UnionDecl::members, isTypedef );
}

ast::CompoundStmt const * EliminateTypedefCore::previsit( ast::CompoundStmt const * stmt ) {
	return field_erase_if( stmt, &ast::CompoundStmt::kids, isTypedefStmt );
}

ast::IfStmt const * EliminateTypedefCore::previsit( ast::IfStmt const * stmt ) {
	return field_erase_if( stmt, &ast::IfStmt::inits, isTypedefStmt );
}

ast::ForStmt const * EliminateTypedefCore::previsit( ast::ForStmt const * stmt ) {
	return field_erase_if( stmt, &ast::ForStmt::inits, isTypedefStmt );
}

ast::WhileDoStmt const * EliminateTypedefCore::previsit( ast::WhileDoStmt const * stmt ) {
	return field_erase_if( stmt, &ast::WhileDoStmt::inits, isTypedefStmt );
}

} // namespace

/// Removes TypedefDecl nodes from the AST.
void eliminateTypedef( ast::TranslationUnit & translationUnit ) {
	ast::Pass<EliminateTypedefCore>::run( translationUnit );
	erase_if( translationUnit.decls, isTypedef );
}

} // namespace Validate

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