123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103 |
- // Copyright (C) 2005-2010 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
- // One bit per color property map (gray and black are the same, green is not
- // supported)
- #ifndef BOOST_ONE_BIT_COLOR_MAP_HPP
- #define BOOST_ONE_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 one_bit_color_type {
- one_bit_white = 0,
- one_bit_not_white = 1
- };
- template <>
- struct color_traits<one_bit_color_type>
- {
- static one_bit_color_type white() { return one_bit_white; }
- static one_bit_color_type gray() { return one_bit_not_white; }
- static one_bit_color_type black() { return one_bit_not_white; }
- };
- template<typename IndexMap = identity_property_map>
- struct one_bit_color_map
- {
- BOOST_STATIC_CONSTANT(int, bits_per_char = std::numeric_limits<unsigned char>::digits);
- std::size_t n;
- IndexMap index;
- shared_array<unsigned char> data;
- typedef typename property_traits<IndexMap>::key_type key_type;
- typedef one_bit_color_type value_type;
- typedef void reference;
- typedef read_write_property_map_tag category;
- explicit one_bit_color_map(std::size_t n, const IndexMap& index = IndexMap())
- : n(n), index(index), data(new unsigned char[(n + bits_per_char - 1) / bits_per_char])
- {
- // Fill to white
- std::fill(data.get(), data.get() + (n + bits_per_char - 1) / bits_per_char, 0);
- }
- };
- template<typename IndexMap>
- inline one_bit_color_type
- get(const one_bit_color_map<IndexMap>& pm,
- typename property_traits<IndexMap>::key_type key)
- {
- BOOST_STATIC_CONSTANT(int, bits_per_char = one_bit_color_map<IndexMap>::bits_per_char);
- typename property_traits<IndexMap>::value_type i = get(pm.index, key);
- BOOST_ASSERT ((std::size_t)i < pm.n);
- return one_bit_color_type((pm.data.get()[i / bits_per_char] >> (i % bits_per_char)) & 1);
- }
- template<typename IndexMap>
- inline void
- put(const one_bit_color_map<IndexMap>& pm,
- typename property_traits<IndexMap>::key_type key,
- one_bit_color_type value)
- {
- BOOST_STATIC_CONSTANT(int, bits_per_char = one_bit_color_map<IndexMap>::bits_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 < 2);
- std::size_t byte_num = i / bits_per_char;
- std::size_t bit_position = (i % bits_per_char);
- pm.data.get()[byte_num] =
- (unsigned char)
- ((pm.data.get()[byte_num] & ~(1 << bit_position))
- | (value << bit_position));
- }
- template<typename IndexMap>
- inline one_bit_color_map<IndexMap>
- make_one_bit_color_map(std::size_t n, const IndexMap& index_map)
- {
- return one_bit_color_map<IndexMap>(n, index_map);
- }
- } // end namespace boost
- #include BOOST_GRAPH_MPI_INCLUDE(<boost/graph/distributed/one_bit_color_map.hpp>)
- #endif // BOOST_ONE_BIT_COLOR_MAP_HPP
|