cppast/src/visitor.cpp
Jonathan Müller 664b5a6d4e Add cpp_variable
2017-01-22 13:06:18 +01:00

59 lines
1.8 KiB
C++

// Copyright (C) 2017 Jonathan Müller <jonathanmueller.dev@gmail.com>
// This file is subject to the license terms in the LICENSE file
// found in the top-level directory of this distribution.
#include <cppast/visitor.hpp>
#include <cppast/cpp_entity.hpp>
#include <cppast/cpp_entity_kind.hpp>
#include <cppast/cpp_enum.hpp>
#include <cppast/cpp_file.hpp>
#include <cppast/cpp_namespace.hpp>
using namespace cppast;
namespace
{
template <typename T>
bool handle_container(const cpp_entity& e, detail::visitor_callback_t cb, void* functor)
{
auto& container = static_cast<const T&>(e);
auto handle_children = cb(functor, container, visitor_info::container_entity_enter);
if (handle_children)
{
for (auto& child : container)
if (!detail::visit(child, cb, functor))
return false;
}
return cb(functor, container, visitor_info::container_entity_exit);
}
}
bool detail::visit(const cpp_entity& e, detail::visitor_callback_t cb, void* functor)
{
switch (e.kind())
{
case cpp_entity_kind::file_t:
return handle_container<cpp_file>(e, cb, functor);
case cpp_entity_kind::namespace_t:
return handle_container<cpp_namespace>(e, cb, functor);
case cpp_entity_kind::enum_t:
return handle_container<cpp_enum>(e, cb, functor);
case cpp_entity_kind::namespace_alias_t:
case cpp_entity_kind::using_directive_t:
case cpp_entity_kind::using_declaration_t:
case cpp_entity_kind::type_alias_t:
case cpp_entity_kind::enum_value_t:
case cpp_entity_kind::variable_t:
return cb(functor, e, visitor_info::leaf_entity);
case cpp_entity_kind::count:
break;
}
DEBUG_UNREACHABLE(detail::assert_handler{});
return true;
}