add_binding.h

Go to the documentation of this file.

// Author(s): Wieger Wesselink
// Copyright: see the accompanying file COPYING or copy at
// https://github.com/mCRL2org/mCRL2/blob/master/COPYING
//
// 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)
//
 
#ifndef MCRL2_CORE_ADD_BINDING_H
#define MCRL2_CORE_ADD_BINDING_H
 
#include "mcrl2/atermpp/type_traits.h"
 
namespace mcrl2
{
 
namespace core
{
 
template <template <class> class TraverserOrBuilder, typename Derived, typename Variable>
class add_binding : public TraverserOrBuilder<Derived>
{
  public:
    typedef TraverserOrBuilder<Derived> super;
    typedef Variable variable_type;
 
  protected:
    std::multiset<variable_type> m_bound_variables;
 
    void increase_bind_count(const variable_type& var)
    {
      m_bound_variables.insert(var);
    }
 
    template <typename Container>
    void increase_bind_count(const Container& variables, typename atermpp::enable_if_container<Container, variable_type>::type* = nullptr)
    {
      for (typename Container::const_iterator i = variables.begin(); i != variables.end(); ++i)
      {
        m_bound_variables.insert(*i);
      }
    }
 
    void decrease_bind_count(const variable_type& var)
    {
      m_bound_variables.erase(m_bound_variables.find(var));
    }
 
    template <typename Container>
    void decrease_bind_count(const Container& variables, typename atermpp::enable_if_container<Container, variable_type>::type* = nullptr)
    {
      for (typename Container::const_iterator i = variables.begin(); i != variables.end(); ++i)
      {
        m_bound_variables.erase(m_bound_variables.find(*i));
      }
    }
 
  public:
    bool is_bound(variable_type const& v) const
    {
      return m_bound_variables.find(v) != m_bound_variables.end();
    }
 
    const std::multiset<variable_type>& bound_variables() const
    {
      return m_bound_variables;
    }
 
    typename std::multiset<variable_type>::size_type bind_count(const variable_type& v)
    {
      return m_bound_variables.count(v);
    }
};
 
} // namespace core
 
} // namespace mcrl2
 
#endif // MCRL2_CORE_ADD_BINDING_H