source: src/ResolvExpr/EraseWith.cpp @ 3f37f5b

//
// 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.
//
// EraseWith.cpp -- After resolution, erase all with constructs.
//
// Author           : Andrew Beach
// Created On       : Sun Oct  8  9:42:00 2023
// Last Modified By : Andrew Beach
// Last Modified On : Sun Oct  8 10:03:00 2023
// Update Count     : 0
//

#include "EraseWith.hpp"

#include "AST/Decl.hpp"
#include "AST/Pass.hpp"
#include "AST/Stmt.hpp"

namespace ResolvExpr {

namespace {

struct WithEraser {
        ast::FunctionDecl const * postvisit( ast::FunctionDecl const * decl ) {
                if ( decl->withExprs.empty() ) return decl;
                auto mutDecl = mutate( decl );
                mutDecl->withExprs.clear();
                return mutDecl;
        }

        ast::Stmt const * postvisit( ast::DeclStmt const * stmt ) {
                if ( auto decl = stmt->decl.as<ast::WithStmt>() ) return decl->stmt;
                return stmt;
        }
};

} // namespace

void eraseWith( ast::TranslationUnit & translationUnit ) {
        ast::Pass<WithEraser>::run( translationUnit );
}

} // namespace ResolvExpr