15#ifndef dealii_index_set_h
16#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
135# ifdef DEAL_II_TRILINOS_WITH_TPETRA
140 Teuchos::RCP<
const Tpetra::Map<int, types::signed_global_dof_index>> map);
146 explicit IndexSet(
const Epetra_BlockMap &map);
219 template <
typename ForwardIterator>
448 std::vector<IndexSet>
450 const std::vector<types::global_dof_index> &n_indices_per_block)
const;
501 std::vector<size_type>
526 template <
typename VectorType>
544 template <
typename StreamType>
546 print(StreamType &out)
const;
553 write(std::ostream &out)
const;
560 read(std::istream &in);
576#ifdef DEAL_II_WITH_TRILINOS
609# ifdef DEAL_II_TRILINOS_WITH_TPETRA
610 Tpetra::Map<int, types::signed_global_dof_index>
614 Teuchos::RCP<Tpetra::Map<int, types::signed_global_dof_index>>
620#ifdef DEAL_II_WITH_PETSC
635 <<
"The global index " << arg1
636 <<
" is not an element of this set.");
643 template <
class Archive>
645 serialize(Archive &ar,
const unsigned int version);
1066 return sizeof(
Range);
1074 template <
class Archive>
1076 serialize(Archive &ar,
const unsigned int version);
1168 boost::container::small_vector<std::pair<size_type, size_type>, 200>
1170 const bool ranges_are_sorted);
1210 , range_idx(range_idx)
1220 , range_idx(
numbers::invalid_dof_index)
1227 : index_set(other.index_set)
1228 , range_idx(other.range_idx)
1240 return index_set->ranges[range_idx].end - index_set->ranges[range_idx].begin;
1248 return index_set !=
nullptr && range_idx < index_set->n_intervals();
1257 return {index_set, range_idx, index_set->ranges[range_idx].begin};
1268 if (range_idx < index_set->
ranges.size() - 1)
1269 return {index_set, range_idx + 1, index_set->ranges[range_idx + 1].begin};
1271 return index_set->end();
1281 return index_set->ranges[range_idx].end - 1;
1304 "Can not compare accessors pointing to different IndexSets"));
1316 "Can not compare accessors pointing to different IndexSets"));
1328 "Impossible to advance an IndexSet::IntervalIterator that is invalid"));
1332 if (range_idx >= index_set->ranges.size())
1342 : accessor(idxset, range_idx)
1407 return !(*
this == other);
1427 "Can not compare iterators belonging to different IndexSets"));
1430 accessor.index_set->ranges.size() :
1434 accessor.index_set->ranges.size() :
1438 return static_cast<int>(lhs - rhs);
1440 return -
static_cast<int>(rhs - lhs);
1452 , range_idx(range_idx)
1457 "Invalid range index for IndexSet::ElementIterator constructor."));
1462 "Invalid index argument for IndexSet::ElementIterator constructor."));
1469 , range_idx(
numbers::invalid_dof_index)
1470 , idx(
numbers::invalid_dof_index)
1480 (range_idx < index_set->
ranges.size() &&
1481 idx < index_set->
ranges[range_idx].end),
1484 return (range_idx < index_set->
ranges.size() &&
1485 idx < index_set->
ranges[range_idx].end);
1496 "Impossible to dereference an IndexSet::ElementIterator that is invalid"));
1508 "Can not compare iterators belonging to different IndexSets"));
1509 return range_idx == other.
range_idx && idx == other.
idx;
1520 "Impossible to advance an IndexSet::ElementIterator that is invalid"));
1521 if (idx < index_set->
ranges[range_idx].
end)
1524 if (idx == index_set->ranges[range_idx].end)
1527 if (range_idx < index_set->
ranges.size() - 1)
1530 idx = index_set->ranges[range_idx].begin;
1566 return !(*
this == other);
1577 "Can not compare iterators belonging to different IndexSets"));
1584inline std::ptrdiff_t
1590 "Can not compare iterators belonging to different IndexSets"));
1593 if (!(*
this < other))
1594 return -(other - *
this);
1603 std::ptrdiff_t c = index_set->ranges[range_idx].end - idx;
1607 range < index_set->ranges.size() && range <= other.
range_idx;
1609 c += index_set->ranges[range].end - index_set->ranges[range].begin;
1612 other.
range_idx < index_set->ranges.size() ||
1615 "Inconsistent iterator state. Did you invalidate iterators by modifying the IndexSet?"));
1620 c -= index_set->ranges[other.
range_idx].end - other.
idx;
1655 : is_compressed(true)
1656 , index_space_size(size)
1657 , largest_range(
numbers::invalid_unsigned_int)
1663 : ranges(std::move(is.ranges))
1664 , is_compressed(is.is_compressed)
1665 , index_space_size(is.index_space_size)
1666 , largest_range(is.largest_range)
1669 is.is_compressed =
true;
1670 is.index_space_size = 0;
1681 ranges = std::move(is.ranges);
1682 is_compressed = is.is_compressed;
1683 index_space_size = is.index_space_size;
1684 largest_range = is.largest_range;
1687 is.is_compressed =
true;
1688 is.index_space_size = 0;
1703 return {
this, 0,
ranges[0].begin};
1756 ExcMessage(
"This function can only be called if the current "
1757 "object does not yet contain any elements."));
1818template <
typename ForwardIterator>
1830 boost::container::small_vector<std::pair<size_type, size_type>, 200>
1832 bool ranges_are_sorted =
true;
1833 for (ForwardIterator p =
begin; p !=
end;)
1845 ForwardIterator q = p;
1847 while ((q !=
end) && (*q == *p))
1855 while ((q !=
end) && (
static_cast<size_type>(*q) == end_index))
1858 while ((q !=
end) && (
static_cast<size_type>(*q) == end_index))
1865 tmp_ranges.emplace_back(begin_index, end_index);
1873 if ((p !=
end) && (
static_cast<size_type>(*p) < end_index))
1874 ranges_are_sorted =
false;
1885 if (
ranges.empty() ==
false)
1894 else if (
ranges.size() > 1)
1909 return (
ranges.size() <= 1);
1937 for (
const auto &range :
ranges)
1938 s += (range.end - range.begin);
1962 const std::vector<Range>::const_iterator main_range =
1965 return main_range->nth_index_in_set;
1980 if (n >= main_range->nth_index_in_set &&
1982 return main_range->begin + (n - main_range->nth_index_in_set);
2007 else if (
ranges.size() > 1)
2021 return (is.
size() == 0);
2023 return (
size() == 0);
2042 return (is.
size() != 0);
2044 return (
size() != 0);
2057template <
typename Vector>
2065 std::fill(vector.
begin(), vector.
end(), 0);
2069 for (
const auto &range :
ranges)
2070 for (
size_type i = range.begin; i < range.end; ++i)
2076template <
typename StreamType>
2082 std::vector<Range>::const_iterator p;
2085 if (p->end - p->begin == 1)
2088 out <<
'[' << p->begin <<
',' << p->end - 1 <<
']';
2093 out <<
'}' << std::endl;
2098template <
class Archive>
2107template <
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=(const IntervalIterator &other)=default
IntervalIterator & operator++()
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
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
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
IndexSet & operator=(const IndexSet &)=default
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
Teuchos::RCP< Tpetra::Map< int, types::signed_global_dof_index > > make_tpetra_map_rcp(const MPI_Comm communicator=MPI_COMM_WORLD, const bool overlapping=false) 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
#define Assert(cond, exc)
static ::ExceptionBase & ExcIndexNotPresent(size_type arg1)
#define AssertIndexRange(index, range)
static ::ExceptionBase & ExcInternalError()
static ::ExceptionBase & ExcDimensionMismatch(std::size_t arg1, std::size_t arg2)
#define DeclException1(Exception1, type1, outsequence)
static ::ExceptionBase & ExcMessage(std::string arg1)
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)