16 #include <deal.II/lac/petsc_vector_base.h> 18 #ifdef DEAL_II_WITH_PETSC 20 # include <deal.II/base/memory_consumption.h> 21 # include <deal.II/base/multithread_info.h> 23 # include <deal.II/lac/exceptions.h> 24 # include <deal.II/lac/petsc_compatibility.h> 25 # include <deal.II/lac/petsc_vector.h> 29 DEAL_II_NAMESPACE_OPEN
35 VectorReference::operator PetscScalar()
const 49 VecGetOwnershipRange(vector.vector, &begin, &end);
52 Vec locally_stored_elements = PETSC_NULL;
53 ierr = VecGhostGetLocalForm(vector.vector, &locally_stored_elements);
57 ierr = VecGetSize(locally_stored_elements, &lsize);
61 ierr = VecGetArray(locally_stored_elements, &ptr);
66 if (index >= static_cast<size_type>(begin) &&
67 index < static_cast<size_type>(end))
70 value = *(ptr + index - begin);
76 vector.ghost_indices.index_within_set(index);
78 Assert(ghostidx + end - begin < (size_type)lsize,
80 value = *(ptr + ghostidx + end - begin);
83 ierr = VecRestoreArray(locally_stored_elements, &ptr);
87 VecGhostRestoreLocalForm(vector.vector, &locally_stored_elements);
99 PetscErrorCode ierr = VecGetOwnershipRange(vector.vector, &begin, &end);
102 AssertThrow((index >= static_cast<size_type>(begin)) &&
103 (index < static_cast<size_type>(end)),
104 ExcAccessToNonlocalElement(index, begin, end - 1));
106 PetscInt idx = index;
108 ierr = VecGetValues(vector.vector, 1, &idx, &value);
119 , obtained_ownership(true)
122 ExcMessage(
"PETSc does not support multi-threaded access, set " 123 "the thread limit to 1 in MPI_InitFinalize()."));
136 ExcMessage(
"PETSc does not support multi-threaded access, set " 137 "the thread limit to 1 in MPI_InitFinalize()."));
156 ExcMessage(
"PETSc does not support multi-threaded access, set " 157 "the thread limit to 1 in MPI_InitFinalize()."));
166 const PetscErrorCode ierr = VecDestroy(&
vector);
179 const PetscErrorCode ierr = VecDestroy(&
vector);
198 PetscErrorCode ierr = VecSet(
vector, s);
203 Vec ghost = PETSC_NULL;
204 ierr = VecGhostGetLocalForm(
vector, &ghost);
207 ierr = VecSet(ghost, s);
210 ierr = VecGhostRestoreLocalForm(
vector, &ghost);
225 const PetscErrorCode ierr = VecEqual(
vector, v.
vector, &flag);
228 return (flag == PETSC_TRUE);
239 const PetscErrorCode ierr = VecEqual(
vector, v.
vector, &flag);
242 return (flag == PETSC_FALSE);
247 VectorBase::size_type
251 const PetscErrorCode ierr = VecGetSize(
vector, &sz);
259 VectorBase::size_type
263 const PetscErrorCode ierr = VecGetLocalSize(
vector, &sz);
271 std::pair<VectorBase::size_type, VectorBase::size_type>
275 const PetscErrorCode ierr =
276 VecGetOwnershipRange(static_cast<const Vec &>(
vector), &begin, &end);
279 return std::make_pair(begin, end);
286 const std::vector<PetscScalar> &values)
288 Assert(indices.size() == values.size(),
289 ExcMessage(
"Function called with arguments of different sizes"));
297 const std::vector<PetscScalar> &values)
299 Assert(indices.size() == values.size(),
300 ExcMessage(
"Function called with arguments of different sizes"));
308 const ::Vector<PetscScalar> &values)
310 Assert(indices.size() == values.size(),
311 ExcMessage(
"Function called with arguments of different sizes"));
319 const size_type * indices,
320 const PetscScalar *values)
340 const PetscErrorCode ierr = VecDot(vec.
vector,
vector, &result);
364 # ifdef DEAL_II_WITH_MPI 368 int all_int_last_action;
370 const int ierr = MPI_Allreduce(&my_int_last_action,
371 &all_int_last_action,
380 ExcMessage(
"Error: not all processors agree on the last " 381 "VectorOperation before this compress() call."));
390 "Missing compress() or calling with wrong VectorOperation argument."));
406 PetscErrorCode ierr = VecAssemblyBegin(
vector);
408 ierr = VecAssemblyEnd(
vector);
419 VectorBase::real_type
433 if (dynamic_cast<const PETScWrappers::MPI::Vector *>(
this) !=
nullptr)
436 const PetscErrorCode ierr = VecSum(
vector, &sum);
438 return sum /
static_cast<PetscReal
>(
size());
443 PetscScalar * start_ptr;
444 PetscErrorCode ierr = VecGetArray(
vector, &start_ptr);
447 PetscScalar mean = 0;
449 PetscScalar sum0 = 0, sum1 = 0, sum2 = 0, sum3 = 0;
454 const PetscScalar *ptr = start_ptr;
455 const PetscScalar *eptr = ptr + (
size() / 4) * 4;
464 while (ptr != start_ptr +
size())
467 mean = (sum0 + sum1 + sum2 + sum3) / static_cast<PetscReal>(
size());
472 ierr = VecRestoreArray(
vector, &start_ptr);
479 VectorBase::real_type
484 const PetscErrorCode ierr = VecNorm(
vector, NORM_1, &d);
492 VectorBase::real_type
497 const PetscErrorCode ierr = VecNorm(
vector, NORM_2, &d);
505 VectorBase::real_type
510 PetscScalar * start_ptr;
511 PetscErrorCode ierr = VecGetArray(
vector, &start_ptr);
516 real_type sum0 = 0, sum1 = 0, sum2 = 0, sum3 = 0;
521 const PetscScalar *ptr = start_ptr;
522 const PetscScalar *eptr = ptr + (
size() / 4) * 4;
531 while (ptr != start_ptr +
size())
534 norm = std::pow(sum0 + sum1 + sum2 + sum3, 1. / p);
539 ierr = VecRestoreArray(
vector, &start_ptr);
547 VectorBase::real_type
552 const PetscErrorCode ierr = VecNorm(
vector, NORM_INFINITY, &d);
560 VectorBase::real_type
566 const PetscErrorCode ierr = VecMin(
vector, &p, &d);
573 VectorBase::real_type
579 const PetscErrorCode ierr = VecMax(
vector, &p, &d);
592 PetscScalar * start_ptr;
593 PetscErrorCode ierr = VecGetArray(
vector, &start_ptr);
596 const PetscScalar *ptr = start_ptr, *eptr = start_ptr +
local_size();
610 ierr = VecRestoreArray(
vector, &start_ptr);
619 template <
typename T>
628 template <
typename T>
634 "whether it is non-negative."))
return true;
645 PetscScalar * start_ptr;
646 PetscErrorCode ierr = VecGetArray(
vector, &start_ptr);
649 const PetscScalar *ptr = start_ptr, *eptr = start_ptr +
local_size();
653 if (!internal::is_non_negative(*ptr))
663 ierr = VecRestoreArray(
vector, &start_ptr);
677 const PetscErrorCode ierr = VecScale(
vector, a);
691 const PetscScalar factor = 1. / a;
694 const PetscErrorCode ierr = VecScale(
vector, factor);
706 const PetscErrorCode ierr = VecAXPY(
vector, 1, v);
718 const PetscErrorCode ierr = VecAXPY(
vector, -1, v);
732 const PetscErrorCode ierr = VecShift(
vector, s);
744 const PetscErrorCode ierr = VecAXPY(
vector, a, v);
760 const PetscScalar weights[2] = {a, b};
763 const PetscErrorCode ierr = VecMAXPY(
vector, 2, weights, addends);
775 const PetscErrorCode ierr = VecAYPX(
vector, s, v);
803 const PetscErrorCode ierr = VecPointwiseMult(
vector, factors,
vector);
832 const PetscErrorCode ierr = VecPointwiseDivide(
vector, a, b);
844 PetscErrorCode ierr =
845 PetscViewerSetFormat(PETSC_VIEWER_STDOUT_WORLD, format);
849 ierr = VecView(
vector, PETSC_VIEWER_STDOUT_WORLD);
857 const unsigned int precision,
858 const bool scientific,
859 const bool across)
const 866 PetscErrorCode ierr = VecGetArray(
vector, &val);
871 const std::ios::fmtflags old_flags = out.flags();
872 const unsigned int old_precision = out.precision(precision);
874 out.precision(precision);
876 out.setf(std::ios::scientific, std::ios::floatfield);
878 out.setf(std::ios::fixed, std::ios::floatfield);
882 out << val[i] <<
' ';
885 out << val[i] << std::endl;
889 out.flags(old_flags);
890 out.precision(old_precision);
894 ierr = VecRestoreArray(
vector, &val);
911 VectorBase::operator
const Vec &()
const 920 std::size_t mem =
sizeof(Vec) +
sizeof(
last_action) +
941 const size_type * indices,
942 const PetscScalar *values,
943 const bool add_values)
950 internal::VectorReference::ExcWrongMode(action,
last_action));
960 const PetscInt *petsc_indices =
961 reinterpret_cast<const PetscInt *
>(indices);
963 const InsertMode mode = (add_values ? ADD_VALUES : INSERT_VALUES);
964 const PetscErrorCode ierr =
965 VecSetValues(
vector, n_elements, petsc_indices, values, mode);
976 DEAL_II_NAMESPACE_CLOSE
978 #endif // DEAL_II_WITH_PETSC real_type lp_norm(const real_type p) const
real_type l1_norm() const
#define AssertNothrow(cond, exc)
VectorBase & operator-=(const VectorBase &V)
static::ExceptionBase & ExcIO()
void sadd(const PetscScalar s, const VectorBase &V)
real_type norm_sqr() const
void ratio(const VectorBase &a, const VectorBase &b)
size_type n_elements() const
bool has_ghost_elements() const
VectorBase & operator/=(const PetscScalar factor)
#define AssertThrow(cond, exc)
VectorBase & operator=(const PetscScalar s)
static::ExceptionBase & ExcIndexRange(int arg1, int arg2, int arg3)
void scale(const VectorBase &scaling_factors)
real_type l2_norm() const
static::ExceptionBase & ExcMessage(std::string arg1)
real_type linfty_norm() const
void compress(const VectorOperation::values operation)
virtual const MPI_Comm & get_mpi_communicator() const
void do_set_add_operation(const size_type n_elements, const size_type *indices, const PetscScalar *values, const bool add_values)
bool operator!=(const VectorBase &v) const
VectorOperation::values last_action
#define Assert(cond, exc)
static::ExceptionBase & ExcDimensionMismatch(std::size_t arg1, std::size_t arg2)
PetscScalar mean_value() const
static bool is_running_single_threaded()
bool operator==(const VectorBase &v) const
PetscScalar add_and_dot(const PetscScalar a, const VectorBase &V, const VectorBase &W)
std::pair< size_type, size_type > local_range() const
void print(std::ostream &out, const unsigned int precision=3, const bool scientific=true, const bool across=true) const
void equ(const PetscScalar a, const VectorBase &V)
PetscScalar operator*(const VectorBase &vec) const
VectorBase & operator*=(const PetscScalar factor)
types::global_dof_index size_type
static::ExceptionBase & ExcGhostsPresent()
size_type local_size() const
VectorBase & operator+=(const VectorBase &V)
#define AssertThrowMPI(error_code)
bool is_non_negative() const
std::size_t memory_consumption() const
void add(const std::vector< size_type > &indices, const std::vector< PetscScalar > &values)
void write_ascii(const PetscViewerFormat format=PETSC_VIEWER_DEFAULT)
virtual ~VectorBase() override
#define AssertIsFinite(number)
std::enable_if< std::is_fundamental< T >::value, std::size_t >::type memory_consumption(const T &t)
static::ExceptionBase & ExcInternalError()
void set(const std::vector< size_type > &indices, const std::vector< PetscScalar > &values)