source: src/Validate/FixQualifiedTypes.cpp @ a1f0cb6

//
// 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.
//
// FixQualifiedTypes.cpp -- Replace the qualified type with a direct type.
//
// Author           : Andrew Beach
// Created On       : Thr Apr 21 11:13:00 2022
// Last Modified By : Andrew Beach
// Last Modified On : Tue Sep 20 16:15:00 2022
// Update Count     : 1
//

#include "Validate/FixQualifiedTypes.hpp"

#include "AST/Copy.hpp"
#include "AST/LinkageSpec.hpp"             // for Linkage
#include "AST/Pass.hpp"
#include "AST/TranslationUnit.hpp"
#include "Common/ToString.hpp"             // for toString
#include "SymTab/Mangler.h"                // for Mangler
#include "Validate/NoIdSymbolTable.hpp"

namespace Validate {

namespace {

struct FixQualifiedTypesCore :
                public WithNoIdSymbolTable,
                public ast::WithCodeLocation {
        ast::Type const * postvisit( ast::QualifiedType const * type ) {
                assert( location );

                ast::ptr<ast::Type> const & parent = type->parent;
                ast::ptr<ast::Type> const & child = type->child;
                if ( parent.as<ast::GlobalScopeType>() ) {
                        // .T => lookup T at global scope.
                        if ( auto inst = child.as<ast::TypeInstType>() ) {
                                auto td = symtab.globalLookupType( inst->name );
                                if ( !td ) {
                                        SemanticError( *location, toString("Use of undefined global type ", inst->name) );
                                }
                                auto base = td->base;
                                assert( base );
                                ast::Type * ret = ast::deepCopy( base );
                                ret->qualifiers = type->qualifiers;
                                return ret;
                        } else {
                                // .T => T is not a type name.
                                assertf( false, "unhandled global qualified child type: %s", toCString(child) );
                        }
                } else {
                        // S.T => S must be an aggregate type, find the declaration for T in S.
                        ast::AggregateDecl const * aggr = nullptr;
                        ast::BaseInstType const * instp = nullptr;
                        if ( auto inst = parent.as<ast::StructInstType>() ) {
                                aggr = inst->base;
                                instp = inst;
                        } else if ( auto inst = parent.as<ast::UnionInstType>() ) {
                                aggr = inst->base;
                                instp = inst;
                        } else {
                                SemanticError( *location, toString("Qualified type requires an aggregate on the left, but has: ", parent) );
                        }
                        // TODO: Need to handle forward declarations.
                        assert( aggr );
                        for ( ast::ptr<ast::Decl> const & member : aggr->members ) {
                                if ( auto inst = child.as<ast::TypeInstType>() ) {
                                        if ( auto decl = member.as<ast::NamedTypeDecl>() ) {
                                                if ( decl->name == inst->name ) {
                                                        assert( decl->base );
                                                        ast::Type * ret = ast::deepCopy( decl->base );
                                                        ret->qualifiers = type->qualifiers;
                                                        ast::TypeSubstitution sub( aggr->params, instp->params );
                                                        auto result = sub.apply(ret);
                                                        return result.node.release();
                                                }
                                        }
                                } else {
                                        // S.T - S is not an aggregate => error.
                                        assertf( false, "unhandled qualified child type: %s", toCString(type) );
                                }
                        }
                        // failed to find a satisfying definition of type
                        SemanticError( *location, toString("Undefined type in qualified type: ", type) );
                }
        }

        ast::Expr const * postvisit( ast::QualifiedNameExpr const * t) {
                assert( location );
                if ( t->type_decl ) {
                auto enumName = t->type_decl->name;
                const ast::EnumDecl * enumDecl = symtab.lookupEnum( enumName );
                        for ( ast::ptr<ast::Decl> const & member : enumDecl->members ) {
                                if ( auto memberAsObj = member.as<ast::ObjectDecl>() ) {
                                        if ( memberAsObj->name == t->name ) {
                                                return new ast::VariableExpr( t->location, memberAsObj );
                                        }
                                } else {
                                        assertf( false, "unhandled qualified child type");
                                }
                        }

                auto var = new ast::ObjectDecl( t->location, t->name,
                        new ast::EnumInstType(enumDecl, ast::CV::Const), nullptr, {}, ast::Linkage::Cforall );
                        var->mangleName = Mangle::mangle( var );
                        return new ast::VariableExpr( t->location, var );
        }

                return t;
        }

};

} // namespace

void fixQualifiedTypes( ast::TranslationUnit & translationUnit ) {
        ast::Pass<FixQualifiedTypesCore>::run( translationUnit );
}

} // namespace Validate

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