17#ifdef DEAL_II_WITH_TRILINOS
27# include <boost/io/ios_state.hpp>
29# include <Epetra_Export.h>
30# include <Epetra_Import.h>
31# include <Epetra_Vector.h>
53 vector.vector->Map().LID(
55# ifndef DEAL_II_WITH_64BIT_INDICES
59 vector.vector->Map().NumMyElements(),
60 vector.vector->Map().MinMyGID(),
61 vector.vector->Map().MaxMyGID()));
66 vector.vector->Map().NumMyElements(),
67 vector.vector->Map().MinMyGID64(),
68 vector.vector->Map().MaxMyGID64()));
72 return (*(vector.vector))[0][local_index];
84 , vector(new Epetra_FEVector(
85 Epetra_Map(0, 0, 0,
Utilities::Trilinos::comm_self())))
94 reinit(parallel_partitioning, communicator);
131 vector = std::make_unique<Epetra_FEVector>(
143 reinit(local, ghost, communicator,
false);
153 Epetra_Map map(0, 0, Epetra_MpiComm(MPI_COMM_SELF));
156 vector = std::make_unique<Epetra_FEVector>(map);
175 vector = std::make_unique<Epetra_FEVector>(map);
205 const bool omit_zeroing_entries,
206 const bool allow_different_maps)
213 if (allow_different_maps ==
false)
219 const Epetra_MpiComm *my_comm =
220 dynamic_cast<const Epetra_MpiComm *
>(&
vector->Comm());
221 const Epetra_MpiComm *v_comm =
222 dynamic_cast<const Epetra_MpiComm *
>(&v.
vector->Comm());
223 const bool same_communicators =
224 my_comm !=
nullptr && v_comm !=
nullptr &&
225 my_comm->DataPtr() == v_comm->DataPtr();
226 if (!same_communicators ||
229 vector = std::make_unique<Epetra_FEVector>(v.
vector->Map());
234 else if (omit_zeroing_entries ==
false)
243 ierr =
vector->PutScalar(0.0);
256 Assert(omit_zeroing_entries ==
false,
258 "It is not possible to exchange data with the "
259 "option 'omit_zeroing_entries' set, which would not write "
265 Epetra_Import data_exchange(
vector->Map(), v.
vector->Map());
267 const int ierr =
vector->Import(*v.
vector, data_exchange, Insert);
273 const Epetra_MpiComm *comm_ptr =
274 dynamic_cast<const Epetra_MpiComm *
>(&(v.
vector->Comm()));
299 size_type n_elements = 0, added_elements = 0, block_offset = 0;
301 n_elements += v.
block(block).vector->Map().NumMyElements();
302 std::vector<TrilinosWrappers::types::int_type> global_ids(n_elements, -1);
307 v.
block(block).trilinos_partitioner());
308 size_type vector_size = v.
block(block).vector->Map().NumMyElements();
309 for (
size_type i = 0; i < vector_size; ++i)
310 global_ids[added_elements++] = glob_elements[i] + block_offset;
313 block_offset += v.
block(block).size();
317 Epetra_Map new_map(v.
size(),
321 v.
block(0).trilinos_partitioner().Comm());
323 auto actual_vec = std::make_unique<Epetra_FEVector>(new_map);
328 v.
block(block).trilinos_vector().ExtractCopy(entries, 0);
329 entries += v.
block(block).vector->Map().NumMyElements();
332 if (import_data ==
true)
335 *actual_vec)) == v.
size(),
340 Epetra_Import data_exchange(
vector->Map(), actual_vec->Map());
342 const int ierr =
vector->Import(*actual_vec, data_exchange, Insert);
348 vector = std::move(actual_vec);
350 const Epetra_MpiComm *comm_ptr =
351 dynamic_cast<const Epetra_MpiComm *
>(&(
vector->Comm()));
366 const bool vector_writable)
370 if (vector_writable ==
false)
372 IndexSet parallel_partitioner = locally_owned_entries;
376 vector = std::make_unique<Epetra_FEVector>(map);
383 ExcMessage(
"A writable vector must not have ghost entries in "
384 "its parallel partitioning"));
386 if (
vector->Map().SameAs(map) ==
false)
387 vector = std::make_unique<Epetra_FEVector>(map);
390 const int ierr =
vector->PutScalar(0.);
395 IndexSet nonlocal_entries(ghost_entries);
399 Epetra_Map nonlocal_map =
402 std::make_unique<Epetra_MultiVector>(nonlocal_map, 1);
422 const std::shared_ptr<const Utilities::MPI::Partitioner> &partitioner,
423 const bool make_ghosted,
424 const bool vector_writable)
428 Assert(partitioner->ghost_indices_initialized(),
429 ExcMessage(
"You asked to create a ghosted vector, but the "
430 "partitioner does not provide ghost indices."));
432 this->
reinit(partitioner->locally_owned_range(),
433 partitioner->ghost_indices(),
434 partitioner->get_mpi_communicator(),
439 this->
reinit(partitioner->locally_owned_range(),
440 partitioner->get_mpi_communicator());
450 ExcMessage(
"Vector is not constructed properly."));
454 const Epetra_MpiComm *my_comm =
455 dynamic_cast<const Epetra_MpiComm *
>(&
vector->Comm());
456 const Epetra_MpiComm *v_comm =
457 dynamic_cast<const Epetra_MpiComm *
>(&v.
vector->Comm());
458 const bool same_communicators = my_comm !=
nullptr && v_comm !=
nullptr &&
459 my_comm->DataPtr() == v_comm->DataPtr();
486 if (same_communicators && v.
vector->Map().SameAs(
vector->Map()))
499 (v.
vector->Map().UniqueGIDs() ||
vector->Map().UniqueGIDs()))
532 template <
typename number>
558 "Cannot find exchange information!"));
560 ExcMessage(
"The input vector has overlapping data, "
561 "which is not allowed."));
567 Epetra_Import data_exchange(
vector->Map(), v.
vector->Map());
568 const int ierr =
vector->Import(*v.
vector, data_exchange, Insert);
583 "Both vectors need to have the same size for import_elements() to work!"));
593 (*
this)[idx] = rwv[idx];
598 (*
this)[idx] += rwv[idx];
612 "Calling compress() is only useful if a vector "
613 "has been written into, but this is a vector with ghost "
614 "elements and consequently is read-only. It does "
615 "not make sense to call compress() for such "
634 "compress() can only be called with VectorOperation add, insert, or unknown"));
644 "The last operation on the Vector and the given last action in the compress() call do not agree!"));
652 const double double_mode = mode;
653 const Epetra_MpiComm *comm_ptr =
661 "Not all processors agree whether the last operation on "
662 "this vector was an addition or a set operation. This will "
663 "prevent the compress() operation from succeeding."));
670 const auto ierr =
vector->GlobalAssemble(mode);
700 if (trilinos_i == -1)
702# ifndef DEAL_II_WITH_64BIT_INDICES
705 vector->Map().NumMyElements(),
707 vector->Map().MaxMyGID()));
711 vector->Map().NumMyElements(),
712 vector->Map().MinMyGID64(),
713 vector->Map().MaxMyGID64()));
717 value = (*vector)[0][trilinos_i];
727 if (allow_different_maps ==
false)
735# if DEAL_II_TRILINOS_VERSION_GTE(11, 11, 0)
736 Epetra_Import data_exchange(
vector->Map(), v.
vector->Map());
738 vector->Import(*v.
vector, data_exchange, Epetra_AddLocalAlso);
745 Epetra_MultiVector dummy(
vector->Map(), 1,
false);
746 Epetra_Import data_exchange(dummy.Map(), v.
vector->Map());
748 int ierr = dummy.Import(*v.
vector, data_exchange, Insert);
751 ierr =
vector->Update(1.0, dummy, 1.0);
763 if (
vector->Map().NumMyElements() != v.
vector->Map().NumMyElements())
767 for (
size_type i = 0; i < vector_size; ++i)
781 return (!(*
this == v));
793 *eptr = start_ptr +
vector->Map().NumMyElements();
794 unsigned int flag = 0;
807 const Epetra_MpiComm *mpi_comm =
808 dynamic_cast<const Epetra_MpiComm *
>(&
vector->Map().Comm());
811 return num_nonzero == 0;
823 *eptr = start_ptr +
vector->Map().NumMyElements();
824 unsigned int flag = 0;
837 const auto max_n_negative =
839 return max_n_negative == 0;
846 const unsigned int precision,
847 const bool scientific,
848 const bool across)
const
851 boost::io::ios_flags_saver restore_flags(out);
854 out.precision(precision);
856 out.setf(std::ios::scientific, std::ios::floatfield);
858 out.setf(std::ios::fixed, std::ios::floatfield);
861 if (
size() != vector_size)
863 auto global_id = [&](
const size_type index) {
866 out <<
"size:" <<
size()
867 <<
" locally_owned_size:" <<
vector->Map().NumMyElements() <<
" :"
869 for (
size_type i = 0; i < vector_size; ++i)
870 out <<
"[" << global_id(i) <<
"]: " << (*(
vector))[0][i]
876 int leading_dimension;
877 int ierr =
vector->ExtractView(&val, &leading_dimension);
882 out <<
static_cast<double>(val[i]) <<
' ';
885 out <<
static_cast<double>(val[i]) << std::endl;
897 std::swap(last_action, v.last_action);
898 std::swap(compressed, v.compressed);
899 std::swap(has_ghosts, v.has_ghosts);
900 std::swap(vector, v.vector);
901 std::swap(nonlocal_vector, v.nonlocal_vector);
902 std::swap(owned_elements, v.owned_elements);
915 return sizeof(*this) +
916 this->
vector->Map().NumMyElements() *
922# include "trilinos_vector.inst"
virtual size_type size() const override
unsigned int n_blocks() const
BlockType & block(const unsigned int i)
bool is_ascending_and_one_to_one(const MPI_Comm communicator) const
size_type n_elements() const
void set_size(const size_type size)
Epetra_Map make_trilinos_map(const MPI_Comm communicator=MPI_COMM_WORLD, const bool overlapping=false) const
void subtract_set(const IndexSet &other)
void add_indices(const ForwardIterator &begin, const ForwardIterator &end)
size_type size() const override
const IndexSet & get_stored_elements() const
void compress(VectorOperation::values operation)
void add(const std::vector< size_type > &indices, const std::vector< TrilinosScalar > &values)
std::unique_ptr< Epetra_MultiVector > nonlocal_vector
void import_elements(const LinearAlgebra::ReadWriteVector< double > &rwv, const VectorOperation::values operation)
MPI_Comm get_mpi_communicator() const
void swap(Vector &v) noexcept
const Epetra_BlockMap & trilinos_partitioner() const
reference operator()(const size_type index)
void import_nonlocal_data_for_fe(const ::TrilinosWrappers::SparseMatrix &matrix, const Vector &vector)
std::unique_ptr< Epetra_FEVector > vector
size_type size() const override
void print(std::ostream &out, const unsigned int precision=3, const bool scientific=true, const bool across=true) const
IndexSet locally_owned_elements() const
void reinit(const Vector &v, const bool omit_zeroing_entries=false, const bool allow_different_maps=false)
bool has_ghost_elements() const
std::pair< size_type, size_type > local_range() const
bool operator!=(const Vector &v) const
bool operator==(const Vector &v) const
Epetra_CombineMode last_action
Vector & operator=(const TrilinosScalar s)
bool is_non_negative() const
std::size_t memory_consumption() const
const Epetra_CrsMatrix & trilinos_matrix() const
#define DEAL_II_NAMESPACE_OPEN
#define DEAL_II_NAMESPACE_CLOSE
static ::ExceptionBase & ExcIO()
static ::ExceptionBase & ExcGhostsPresent()
static ::ExceptionBase & ExcNotImplemented()
static ::ExceptionBase & ExcAccessToNonLocalElement(size_type arg1, size_type arg2, size_type arg3, size_type arg4)
#define Assert(cond, exc)
#define AssertIndexRange(index, range)
static ::ExceptionBase & ExcTrilinosError(int arg1)
static ::ExceptionBase & ExcInternalError()
static ::ExceptionBase & ExcDimensionMismatch(std::size_t arg1, std::size_t arg2)
static ::ExceptionBase & ExcMessage(std::string arg1)
#define AssertThrow(cond, exc)
void swap(BlockVector &u, BlockVector &v) noexcept
TrilinosWrappers::types::int_type global_length(const Epetra_MultiVector &vector)
int gid(const Epetra_BlockMap &map, int i)
TrilinosWrappers::types::int_type * my_global_elements(const Epetra_BlockMap &map)
TrilinosWrappers::types::int64_type n_global_elements(const Epetra_BlockMap &map)
T sum(const T &t, const MPI_Comm mpi_communicator)
unsigned int n_mpi_processes(const MPI_Comm mpi_communicator)
T max(const T &t, const MPI_Comm mpi_communicator)
MinMaxAvg min_max_avg(const double my_value, const MPI_Comm mpi_communicator)