lib/ast/StmtVisitor.cpp

Go to the documentation of this file.

//===-- ast/StmtVisitor.cpp ----------------------------------- -*- C++ -*-===//
//
// This file is distributed under the MIT license. See LICENSE.txt for details.
//
// Copyright (C) 2009-2010, Stephen Wilson
//
//===----------------------------------------------------------------------===//

#include "comma/ast/StmtVisitor.h"
#include "comma/ast/Stmt.h"

using namespace comma;

using llvm::dyn_cast;
using llvm::cast;
using llvm::isa;

void StmtVisitor::visitAst(Ast *node)
{
    if (Stmt *stmt = dyn_cast<Stmt>(node))
        visitStmt(stmt);
}

#define DISPATCH(TYPE, NODE)         \
    Ast::AST_ ## TYPE:               \
    visit ## TYPE(cast<TYPE>(NODE)); \
    break

void StmtVisitor::visitStmt(Stmt *node)
{
    switch (node->getKind()) {
    default:
        assert(false && "Cannot visit this kind of node!");
        break;

    case DISPATCH(StmtSequence, node);
    case DISPATCH(BlockStmt, node);
    case DISPATCH(ProcedureCallStmt, node);
    case DISPATCH(ReturnStmt, node);
    case DISPATCH(AssignmentStmt, node);
    case DISPATCH(IfStmt, node);
    case DISPATCH(WhileStmt, node);
    case DISPATCH(ForStmt, node);
    case DISPATCH(LoopStmt, node);
    case DISPATCH(RaiseStmt, node);
    case DISPATCH(PragmaStmt, node);
    case DISPATCH(NullStmt, node);
    }
}

void StmtVisitor::visitStmtSequence(StmtSequence *node)
{
    if (BlockStmt *block = dyn_cast<BlockStmt>(node))
        visitBlockStmt(block);
}

void StmtVisitor::visitBlockStmt(BlockStmt *node) { }
void StmtVisitor::visitProcedureCallStmt(ProcedureCallStmt *node) { }
void StmtVisitor::visitReturnStmt(ReturnStmt *node) { }
void StmtVisitor::visitAssignmentStmt(AssignmentStmt *node) { }
void StmtVisitor::visitIfStmt(IfStmt *node) { }
void StmtVisitor::visitWhileStmt(WhileStmt *node) { }
void StmtVisitor::visitForStmt(ForStmt *node) { }
void StmtVisitor::visitLoopStmt(LoopStmt *node) { }
void StmtVisitor::visitRaiseStmt(RaiseStmt *node) { }
void StmtVisitor::visitPragmaStmt(PragmaStmt *node) { }
void StmtVisitor::visitNullStmt(NullStmt *node) { }

---