/*=============================================================================
    Copyright (c) 2001-2011 Joel de Guzman

    Distributed under the Boost Software License, Version 1.0. (See accompanying
    file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
=============================================================================*/
#if !defined(BOOST_SPIRIT_CONJURE_ANNOTATION_HPP)
#define BOOST_SPIRIT_CONJURE_ANNOTATION_HPP

#include <map>
#include <boost/variant/apply_visitor.hpp>
#include <boost/type_traits/is_base_of.hpp>
#include <boost/mpl/bool.hpp>
#include "ast.hpp"

namespace client
{
    ///////////////////////////////////////////////////////////////////////////////
    //  The annotation handler links the AST to a map of iterator positions
    //  for the purpose of subsequent semantic error handling when the
    //  program is being compiled.
    ///////////////////////////////////////////////////////////////////////////////
    struct set_annotation_id
    {
        typedef void result_type;

        int id;
        set_annotation_id(int id) : id(id) {}

        void operator()(ast::function_call& x) const
        {
            x.function_name.id = id;
        }

        template <typename T>
        void dispatch(T& x, boost::mpl::true_) const
        {
            x.id = id;
        }

        template <typename T>
        void dispatch(T& x, boost::mpl::false_) const
        {
            // no-op
        }

        template <typename T>
        void operator()(T& x) const
        {
            typename boost::is_base_of<ast::tagged, T> is_tagged;
            dispatch(x, is_tagged);
        }
    };

    struct get_annotation_id
    {
        typedef int result_type;

        int operator()(ast::function_call& x) const
        {
            return x.function_name.id;
        }

        template <typename T>
        int dispatch(T& x, boost::mpl::true_) const
        {
            return x.id;
        }

        template <typename T>
        int dispatch(T& x, boost::mpl::false_) const
        {
            return -1;
        }

        template <typename T>
        int operator()(T& x) const
        {
            typename boost::is_base_of<ast::tagged, T> is_tagged;
            return dispatch(x, is_tagged);
        }
    };

    template <typename Iterator>
    struct annotation
    {
        template <typename, typename>
        struct result { typedef void type; };

        std::vector<Iterator>& iters;
        annotation(std::vector<Iterator>& iters)
          : iters(iters) {}

        void operator()(ast::operand& ast, Iterator pos) const
        {
            int id = iters.size();
            iters.push_back(pos);
            boost::apply_visitor(set_annotation_id(id), ast);
            ast.id = id;
        }

        void operator()(ast::primary_expr& ast, Iterator pos) const
        {
            int id = iters.size();
            iters.push_back(pos);
            boost::apply_visitor(set_annotation_id(id), ast);
            ast.id = id;
        }

        void operator()(ast::variable_declaration& ast, Iterator pos) const
        {
            int id = iters.size();
            iters.push_back(pos);
            ast.lhs.id = id;
        }

        void operator()(ast::assignment& ast, Iterator pos) const
        {
            int id = iters.size();
            iters.push_back(pos);
            ast.lhs.id = id;
        }

        void operator()(ast::return_statement& ast, Iterator pos) const
        {
            int id = iters.size();
            iters.push_back(pos);
            ast.id = id;
        }

        void operator()(ast::identifier& ast, Iterator pos) const
        {
            int id = iters.size();
            iters.push_back(pos);
            ast.id = id;
        }
    };
}

#endif