source: src/Validate/EnumAndPointerDecay.cpp @ fbb930e

//
// 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.
//
// EnumAndPointerDecay.cpp -- Normalizes enumerations and types in functions.
//
// Author           : Andrew Beach
// Created On       : Tue Jun 28 15:50:00 2022
// Last Modified By : Andrew Beach
// Last Modified On : Tue Jul 12 14:45:00 2022
// Update Count     : 0
//

#include "EnumAndPointerDecay.hpp"

#include "AST/CVQualifiers.hpp"
#include "AST/Decl.hpp"
#include "AST/Pass.hpp"
#include "AST/Type.hpp"
#include "SymTab/FixFunction.h"

namespace Validate {

namespace {

struct EnumAndPointerDecayCore final : public ast::WithGuards {
        CodeLocation const * location = nullptr;
        void previsit( ast::ParseNode const * node );
        ast::EnumDecl const * previsit( ast::EnumDecl const * decl );
        ast::FunctionDecl const * previsit( ast::FunctionDecl const * decl );
        ast::FunctionType const * previsit( ast::FunctionType const * type );
};

void EnumAndPointerDecayCore::previsit( ast::ParseNode const * node ) {
        GuardValue( location ) = &node->location;
}

ast::EnumDecl const * EnumAndPointerDecayCore::previsit(
                ast::EnumDecl const * decl ) {
        if ( decl->members.empty() ) {
                return decl;
        }
        // Set the type of each member of the enumeration to be EnumContant.
        auto mut = ast::mutate( decl );
        for ( ast::ptr<ast::Decl> & member : mut->members ) {
                ast::ObjectDecl const * object = member.strict_as<ast::ObjectDecl>();
                member = ast::mutate_field( object, &ast::ObjectDecl::type,
                        new ast::EnumInstType( decl, ast::CV::Const ) );
        }
        GuardValue( location ) = &decl->location;
        return mut;
}

template<typename Member>
void fixFunctionList( CodeLocation const & location, bool isVarArgs,
                std::vector<ast::ptr<Member>> & list ) {
        bool hasVoid = false;
        for ( ast::ptr<Member> & member : list ) {
                member = SymTab::fixFunction( member, hasVoid );
        }

        // The remaining code only applies if void is present.
        if ( !hasVoid ) {
                return;
        }

        // So there is a void, which is only valid if it is the only argument.
        if ( 1 < list.size() || isVarArgs ) {
                SemanticError( location, "invalid type void in function type " );
        }

        // If a single "void" thing in the list to remove it.
        list.clear();
}

ast::FunctionDecl const * EnumAndPointerDecayCore::previsit(
                ast::FunctionDecl const * decl ) {
        auto mut = ast::mutate( decl );
        GuardValue( location ) = &decl->location;
        ast::ArgumentFlag isVarArgs = mut->type->isVarArgs;
        // It seems fixFunction (via fixFunctionList) does the pointer decay part.
        fixFunctionList( mut->location, isVarArgs, mut->params );
        fixFunctionList( mut->location, false, mut->returns );
        return mut;
}

ast::FunctionType const * EnumAndPointerDecayCore::previsit(
                ast::FunctionType const * type ) {
        assert( location );
        auto mut = ast::mutate( type );
        ast::ArgumentFlag isVarArgs = mut->isVarArgs;
        // It seems fixFunction (via fixFunctionList) does the pointer decay part.
        fixFunctionList( *location, isVarArgs, mut->params );
        fixFunctionList( *location, false, mut->returns );
        return mut;
}

} // namespace

void decayEnumsAndPointers( ast::TranslationUnit & translationUnit ) {
        ast::Pass<EnumAndPointerDecayCore>::run( translationUnit );
}

} // namespace Validate

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