16 #ifndef dealii_index_set_h
17 #define dealii_index_set_h
25 #include <boost/container/small_vector.hpp>
31 #ifdef DEAL_II_WITH_TRILINOS
32 # include <Epetra_Map.h>
33 # ifdef DEAL_II_TRILINOS_WITH_TPETRA
34 # include <Tpetra_Map.hpp>
38 #ifdef DEAL_II_WITH_PETSC
133 #ifdef DEAL_II_WITH_TRILINOS
137 explicit IndexSet(
const Epetra_BlockMap &map);
210 template <
typename ForwardIterator>
383 std::vector<IndexSet>
385 const std::vector<types::global_dof_index> &n_indices_per_block)
const;
436 std::vector<size_type>
461 template <
typename VectorType>
479 template <
typename StreamType>
481 print(StreamType &out)
const;
488 write(std::ostream &out)
const;
495 read(std::istream &in);
511 #ifdef DEAL_II_WITH_TRILINOS
542 const bool overlapping =
false)
const;
544 # ifdef DEAL_II_TRILINOS_WITH_TPETRA
545 Tpetra::Map<int, types::signed_global_dof_index>
547 const bool overlapping =
false)
const;
551 #ifdef DEAL_II_WITH_PETSC
553 make_petsc_is(
const MPI_Comm communicator = MPI_COMM_WORLD)
const;
566 <<
"The global index " << arg1
567 <<
" is not an element of this set.");
574 template <
class Archive>
576 serialize(Archive &ar,
const unsigned int version);
997 return sizeof(
Range);
1005 template <
class Archive>
1007 serialize(Archive &ar,
const unsigned int version);
1099 boost::container::small_vector<std::pair<size_type, size_type>, 200>
1101 const bool ranges_are_sorted);
1141 , range_idx(range_idx)
1158 : index_set(other.index_set)
1159 , range_idx(other.range_idx)
1171 return index_set->ranges[range_idx].end - index_set->ranges[range_idx].begin;
1179 return index_set !=
nullptr && range_idx < index_set->n_intervals();
1188 return {index_set, range_idx, index_set->ranges[range_idx].begin};
1199 if (range_idx < index_set->
ranges.size() - 1)
1200 return {index_set, range_idx + 1, index_set->ranges[range_idx + 1].begin};
1202 return index_set->end();
1212 return index_set->ranges[range_idx].end - 1;
1235 "Can not compare accessors pointing to different IndexSets"));
1247 "Can not compare accessors pointing to different IndexSets"));
1259 "Impossible to advance an IndexSet::IntervalIterator that is invalid"));
1263 if (range_idx >= index_set->ranges.size())
1273 : accessor(idxset, range_idx)
1338 return !(*
this == other);
1358 "Can not compare iterators belonging to different IndexSets"));
1361 accessor.index_set->ranges.size() :
1365 accessor.index_set->ranges.size() :
1369 return static_cast<int>(lhs - rhs);
1371 return -
static_cast<int>(rhs - lhs);
1383 , range_idx(range_idx)
1388 "Invalid range index for IndexSet::ElementIterator constructor."));
1393 "Invalid index argument for IndexSet::ElementIterator constructor."));
1411 (range_idx < index_set->
ranges.size() &&
1412 idx < index_set->
ranges[range_idx].end),
1415 return (range_idx < index_set->
ranges.size() &&
1416 idx < index_set->
ranges[range_idx].end);
1427 "Impossible to dereference an IndexSet::ElementIterator that is invalid"));
1439 "Can not compare iterators belonging to different IndexSets"));
1440 return range_idx == other.
range_idx && idx == other.
idx;
1451 "Impossible to advance an IndexSet::ElementIterator that is invalid"));
1452 if (idx < index_set->
ranges[range_idx].
end)
1455 if (idx == index_set->ranges[range_idx].end)
1458 if (range_idx < index_set->
ranges.size() - 1)
1461 idx = index_set->ranges[range_idx].begin;
1497 return !(*
this == other);
1508 "Can not compare iterators belonging to different IndexSets"));
1515 inline std::ptrdiff_t
1521 "Can not compare iterators belonging to different IndexSets"));
1524 if (!(*
this < other))
1525 return -(other - *
this);
1534 std::ptrdiff_t c = index_set->ranges[range_idx].end - idx;
1538 range < index_set->ranges.size() && range <= other.
range_idx;
1540 c += index_set->ranges[range].end - index_set->ranges[range].begin;
1543 other.
range_idx < index_set->ranges.size() ||
1546 "Inconsistent iterator state. Did you invalidate iterators by modifying the IndexSet?"));
1551 c -= index_set->ranges[other.
range_idx].end - other.
idx;
1586 : is_compressed(true)
1587 , index_space_size(size)
1594 : ranges(std::move(is.ranges))
1595 , is_compressed(is.is_compressed)
1596 , index_space_size(is.index_space_size)
1597 , largest_range(is.largest_range)
1600 is.is_compressed =
true;
1601 is.index_space_size = 0;
1612 ranges = std::move(is.ranges);
1613 is_compressed = is.is_compressed;
1614 index_space_size = is.index_space_size;
1615 largest_range = is.largest_range;
1618 is.is_compressed =
true;
1619 is.index_space_size = 0;
1634 return {
this, 0,
ranges[0].begin};
1687 ExcMessage(
"This function can only be called if the current "
1688 "object does not yet contain any elements."));
1749 template <
typename ForwardIterator>
1761 boost::container::small_vector<std::pair<size_type, size_type>, 200>
1763 bool ranges_are_sorted =
true;
1764 for (ForwardIterator p =
begin; p !=
end;)
1776 ForwardIterator q = p;
1778 while ((q !=
end) && (*q == *p))
1786 while ((q !=
end) && (
static_cast<size_type>(*q) == end_index))
1789 while ((q !=
end) && (
static_cast<size_type>(*q) == end_index))
1796 tmp_ranges.emplace_back(begin_index, end_index);
1804 if ((p !=
end) && (
static_cast<size_type>(*p) < end_index))
1805 ranges_are_sorted =
false;
1816 if (
ranges.empty() ==
false)
1825 else if (
ranges.size() > 1)
1840 return (
ranges.size() <= 1);
1868 for (
const auto &range :
ranges)
1869 s += (range.end - range.begin);
1893 const std::vector<Range>::const_iterator main_range =
1896 return main_range->nth_index_in_set;
1911 if (n >= main_range->nth_index_in_set &&
1913 return main_range->begin + (n - main_range->nth_index_in_set);
1938 else if (
ranges.size() > 1)
1952 return (is.
size() == 0);
1954 return (
size() == 0);
1973 return (is.
size() != 0);
1975 return (
size() != 0);
1988 template <
typename Vector>
1996 std::fill(vector.
begin(), vector.
end(), 0);
2000 for (
const auto &range :
ranges)
2001 for (
size_type i = range.begin; i < range.end; ++i)
2007 template <
typename StreamType>
2013 std::vector<Range>::const_iterator p;
2016 if (p->end - p->begin == 1)
2019 out <<
'[' << p->begin <<
',' << p->end - 1 <<
']';
2024 out <<
'}' << std::endl;
2029 template <
class Archive>
2038 template <
class Archive>
size_type operator*() const
bool operator==(const ElementIterator &) const
ElementIterator(const IndexSet *idxset, const size_type range_idx, const size_type index)
std::ptrdiff_t difference_type
ElementIterator & operator++()
bool operator<(const ElementIterator &) const
bool operator!=(const ElementIterator &) const
std::ptrdiff_t operator-(const ElementIterator &p) const
std::forward_iterator_tag iterator_category
const IndexSet * index_set
ElementIterator end() const
size_type n_elements() const
bool operator==(const IntervalAccessor &other) const
IntervalAccessor & operator=(const IntervalAccessor &other)
bool operator<(const IntervalAccessor &other) const
IntervalAccessor(const IndexSet *idxset, const size_type range_idx)
const IndexSet * index_set
ElementIterator begin() const
std::ptrdiff_t difference_type
const IntervalAccessor & operator*() const
IntervalIterator(const IntervalIterator &other)=default
bool operator==(const IntervalIterator &) const
int operator-(const IntervalIterator &p) const
bool operator<(const IntervalIterator &) const
std::forward_iterator_tag iterator_category
IntervalIterator & operator++()
IntervalIterator & operator=(const IntervalIterator &other)=default
const IntervalAccessor * operator->() const
IntervalAccessor accessor
bool operator!=(const IntervalIterator &) const
bool is_subset_of(const IndexSet &other) const
bool is_element_binary_search(const size_type local_index) const
size_type index_within_set_binary_search(const size_type global_index) const
IS make_petsc_is(const MPI_Comm communicator=MPI_COMM_WORLD) const
bool is_ascending_and_one_to_one(const MPI_Comm communicator) const
bool is_contiguous() const
unsigned int n_intervals() const
IndexSet(const IndexSet &)=default
IndexSet & operator=(const IndexSet &)=default
IntervalIterator end_intervals() const
ElementIterator at(const size_type global_index) const
void fill_index_vector(std::vector< size_type > &indices) const
size_type index_within_set(const size_type global_index) const
std::vector< IndexSet > split_by_block(const std::vector< types::global_dof_index > &n_indices_per_block) const
size_type n_elements() const
bool operator==(const IndexSet &is) const
void add_range_lower_bound(const Range &range)
bool is_element(const size_type index) const
void serialize(Archive &ar, const unsigned int version)
ElementIterator begin() const
void set_size(const size_type size)
bool operator!=(const IndexSet &is) const
void read(std::istream &in)
Epetra_Map make_trilinos_map(const MPI_Comm communicator=MPI_COMM_WORLD, const bool overlapping=false) const
size_type largest_range_starting_index() const
IndexSet tensor_product(const IndexSet &other) const
void add_index(const size_type index)
void write(std::ostream &out) const
void block_read(std::istream &in)
void add_ranges_internal(boost::container::small_vector< std::pair< size_type, size_type >, 200 > &tmp_ranges, const bool ranges_are_sorted)
IntervalIterator begin_intervals() const
Tpetra::Map< int, types::signed_global_dof_index > make_tpetra_map(const MPI_Comm communicator=MPI_COMM_WORLD, const bool overlapping=false) const
void fill_binary_vector(VectorType &vector) const
std::vector< Range > ranges
void subtract_set(const IndexSet &other)
ElementIterator end() const
Threads::Mutex compress_mutex
IndexSet complete_index_set(const IndexSet::size_type N)
size_type index_space_size
void block_write(std::ostream &out) const
IndexSet get_view(const size_type begin, const size_type end) const
void add_range(const size_type begin, const size_type end)
size_type nth_index_in_set(const size_type local_index) const
std::size_t memory_consumption() const
void print(StreamType &out) const
size_type nth_index_in_set_binary_search(const size_type local_index) const
std::vector< size_type > get_index_vector() const
types::global_dof_index size_type
void add_indices(const ForwardIterator &begin, const ForwardIterator &end)
IndexSet operator&(const IndexSet &is) const
virtual size_type size() const override
#define DEAL_II_DEPRECATED
#define DEAL_II_NAMESPACE_OPEN
#define DEAL_II_NAMESPACE_CLOSE
static ::ExceptionBase & ExcIndexNotPresent(size_type arg1)
static ::ExceptionBase & ExcInternalError()
static ::ExceptionBase & ExcDimensionMismatch(std::size_t arg1, std::size_t arg2)
#define Assert(cond, exc)
#define AssertIndexRange(index, range)
#define DeclException1(Exception1, type1, outsequence)
static ::ExceptionBase & ExcMessage(std::string arg1)
VectorType::value_type * begin(VectorType &V)
VectorType::value_type * end(VectorType &V)
TrilinosWrappers::types::int_type global_index(const Epetra_BlockMap &map, const ::types::global_dof_index i)
std::string compress(const std::string &input)
static const unsigned int invalid_unsigned_int
const types::global_dof_index invalid_dof_index
unsigned int global_dof_index
static bool end_compare(const IndexSet::Range &x, const IndexSet::Range &y)
static bool nth_index_compare(const IndexSet::Range &x, const IndexSet::Range &y)
friend bool operator==(const Range &range_1, const Range &range_2)
size_type nth_index_in_set
static std::size_t memory_consumption()
void serialize(Archive &ar, const unsigned int version)
friend bool operator<(const Range &range_1, const Range &range_2)
void advance(std::tuple< I1, I2 > &t, const unsigned int n)