Website : rimsha.abasa.com
backdoor
Home
Console
Upload
information
Create File
Create Folder
About
Tools
:
/
usr
/
include
/
boost
/
graph
/
Filename :
two_bit_color_map.hpp
back
Copy
// Copyright (C) 2005-2006 The Trustees of Indiana University. // 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) // Authors: Jeremiah Willcock // Douglas Gregor // Andrew Lumsdaine // Two bit per color property map #ifndef BOOST_TWO_BIT_COLOR_MAP_HPP #define BOOST_TWO_BIT_COLOR_MAP_HPP #include <boost/property_map/property_map.hpp> #include <boost/graph/properties.hpp> #include <boost/graph/detail/mpi_include.hpp> #include <boost/shared_array.hpp> #include <boost/config.hpp> #include <boost/assert.hpp> #include <algorithm> #include <limits> namespace boost { enum two_bit_color_type { two_bit_white = 0, two_bit_gray = 1, two_bit_green = 2, two_bit_black = 3 }; template <> struct color_traits< two_bit_color_type > { static two_bit_color_type white() { return two_bit_white; } static two_bit_color_type gray() { return two_bit_gray; } static two_bit_color_type green() { return two_bit_green; } static two_bit_color_type black() { return two_bit_black; } }; template < typename IndexMap = identity_property_map > struct two_bit_color_map { std::size_t n; IndexMap index; shared_array< unsigned char > data; BOOST_STATIC_CONSTANT( int, bits_per_char = std::numeric_limits< unsigned char >::digits); BOOST_STATIC_CONSTANT(int, elements_per_char = bits_per_char / 2); typedef typename property_traits< IndexMap >::key_type key_type; typedef two_bit_color_type value_type; typedef void reference; typedef read_write_property_map_tag category; explicit two_bit_color_map( std::size_t n, const IndexMap& index = IndexMap()) : n(n) , index(index) , data(new unsigned char[(n + elements_per_char - 1) / elements_per_char]()) { } }; template < typename IndexMap > inline two_bit_color_type get(const two_bit_color_map< IndexMap >& pm, typename property_traits< IndexMap >::key_type key) { BOOST_STATIC_CONSTANT(int, elements_per_char = two_bit_color_map< IndexMap >::elements_per_char); typename property_traits< IndexMap >::value_type i = get(pm.index, key); BOOST_ASSERT((std::size_t)i < pm.n); std::size_t byte_num = i / elements_per_char; std::size_t bit_position = ((i % elements_per_char) * 2); return two_bit_color_type((pm.data.get()[byte_num] >> bit_position) & 3); } template < typename IndexMap > inline void put(const two_bit_color_map< IndexMap >& pm, typename property_traits< IndexMap >::key_type key, two_bit_color_type value) { BOOST_STATIC_CONSTANT(int, elements_per_char = two_bit_color_map< IndexMap >::elements_per_char); typename property_traits< IndexMap >::value_type i = get(pm.index, key); BOOST_ASSERT((std::size_t)i < pm.n); BOOST_ASSERT(value >= 0 && value < 4); std::size_t byte_num = i / elements_per_char; std::size_t bit_position = ((i % elements_per_char) * 2); pm.data.get()[byte_num] = (unsigned char)((pm.data.get()[byte_num] & ~(3 << bit_position)) | (value << bit_position)); } template < typename IndexMap > inline two_bit_color_map< IndexMap > make_two_bit_color_map( std::size_t n, const IndexMap& index_map) { return two_bit_color_map< IndexMap >(n, index_map); } } // end namespace boost #include BOOST_GRAPH_MPI_INCLUDE(<boost/graph/distributed/two_bit_color_map.hpp>) #endif // BOOST_TWO_BIT_COLOR_MAP_HPP