18 #ifdef DEAL_II_WITH_PETSC 36 VectorReference::operator PetscScalar()
const 50 VecGetOwnershipRange(vector.vector, &begin, &end);
53 Vec locally_stored_elements = PETSC_NULL;
54 ierr = VecGhostGetLocalForm(vector.vector, &locally_stored_elements);
58 ierr = VecGetSize(locally_stored_elements, &lsize);
62 ierr = VecGetArray(locally_stored_elements, &ptr);
67 if (index >= static_cast<size_type>(begin) &&
68 index < static_cast<size_type>(end))
71 value = *(ptr + index -
begin);
77 vector.ghost_indices.index_within_set(index);
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);
120 , obtained_ownership(true)
123 ExcMessage(
"PETSc does not support multi-threaded access, set " 124 "the thread limit to 1 in MPI_InitFinalize()."));
137 ExcMessage(
"PETSc does not support multi-threaded access, set " 138 "the thread limit to 1 in MPI_InitFinalize()."));
157 ExcMessage(
"PETSc does not support multi-threaded access, set " 158 "the thread limit to 1 in MPI_InitFinalize()."));
167 const PetscErrorCode ierr = VecDestroy(&
vector);
180 const PetscErrorCode ierr = VecDestroy(&
vector);
199 PetscErrorCode ierr = VecSet(
vector, s);
204 Vec ghost = PETSC_NULL;
205 ierr = VecGhostGetLocalForm(
vector, &ghost);
208 ierr = VecSet(ghost, s);
211 ierr = VecGhostRestoreLocalForm(
vector, &ghost);
226 const PetscErrorCode ierr = VecEqual(
vector, v.
vector, &flag);
229 return (flag == PETSC_TRUE);
240 const PetscErrorCode ierr = VecEqual(
vector, v.
vector, &flag);
243 return (flag == PETSC_FALSE);
252 const PetscErrorCode ierr = VecGetSize(
vector, &sz);
264 const PetscErrorCode ierr = VecGetLocalSize(
vector, &sz);
272 std::pair<VectorBase::size_type, VectorBase::size_type>
276 const PetscErrorCode ierr =
277 VecGetOwnershipRange(static_cast<const Vec &>(
vector), &begin, &end);
280 return std::make_pair(begin, end);
287 const std::vector<PetscScalar> &values)
289 Assert(indices.size() == values.size(),
290 ExcMessage(
"Function called with arguments of different sizes"));
298 const std::vector<PetscScalar> &values)
300 Assert(indices.size() == values.size(),
301 ExcMessage(
"Function called with arguments of different sizes"));
309 const ::Vector<PetscScalar> &values)
311 Assert(indices.size() == values.size(),
312 ExcMessage(
"Function called with arguments of different sizes"));
321 const PetscScalar *values)
341 const PetscErrorCode ierr = VecDot(vec.
vector,
vector, &result);
365 # ifdef DEAL_II_WITH_MPI 369 int all_int_last_action;
371 const int ierr = MPI_Allreduce(&my_int_last_action,
372 &all_int_last_action,
381 ExcMessage(
"Error: not all processors agree on the last " 382 "VectorOperation before this compress() call."));
391 "Missing compress() or calling with wrong VectorOperation argument."));
407 PetscErrorCode ierr = VecAssemblyBegin(
vector);
409 ierr = VecAssemblyEnd(
vector);
434 if (dynamic_cast<const PETScWrappers::MPI::Vector *>(
this) !=
nullptr)
437 const PetscErrorCode ierr = VecSum(
vector, &sum);
439 return sum /
static_cast<PetscReal
>(
size());
444 PetscScalar * start_ptr;
445 PetscErrorCode ierr = VecGetArray(
vector, &start_ptr);
448 PetscScalar
mean = 0;
450 PetscScalar sum0 = 0, sum1 = 0, sum2 = 0, sum3 = 0;
455 const PetscScalar *ptr = start_ptr;
456 const PetscScalar *eptr = ptr + (
size() / 4) * 4;
465 while (ptr != start_ptr +
size())
468 mean = (sum0 + sum1 + sum2 + sum3) / static_cast<PetscReal>(
size());
473 ierr = VecRestoreArray(
vector, &start_ptr);
485 const PetscErrorCode ierr = VecNorm(
vector, NORM_1, &d);
498 const PetscErrorCode ierr = VecNorm(
vector, NORM_2, &d);
511 PetscScalar * start_ptr;
512 PetscErrorCode ierr = VecGetArray(
vector, &start_ptr);
517 real_type sum0 = 0, sum1 = 0, sum2 = 0, sum3 = 0;
522 const PetscScalar *ptr = start_ptr;
523 const PetscScalar *eptr = ptr + (
size() / 4) * 4;
532 while (ptr != start_ptr +
size())
535 norm = std::pow(sum0 + sum1 + sum2 + sum3, 1. / p);
540 ierr = VecRestoreArray(
vector, &start_ptr);
553 const PetscErrorCode ierr = VecNorm(
vector, NORM_INFINITY, &d);
567 const PetscErrorCode ierr = VecMin(
vector, &p, &d);
580 const PetscErrorCode ierr = VecMax(
vector, &p, &d);
593 PetscScalar * start_ptr;
594 PetscErrorCode ierr = VecGetArray(
vector, &start_ptr);
597 const PetscScalar *ptr = start_ptr, *eptr = start_ptr +
local_size();
611 ierr = VecRestoreArray(
vector, &start_ptr);
620 template <
typename T>
629 template <
typename T>
635 "whether it is non-negative."))
return true;
646 PetscScalar * start_ptr;
647 PetscErrorCode ierr = VecGetArray(
vector, &start_ptr);
650 const PetscScalar *ptr = start_ptr, *eptr = start_ptr +
local_size();
664 ierr = VecRestoreArray(
vector, &start_ptr);
678 const PetscErrorCode ierr = VecScale(
vector, a);
692 const PetscScalar factor = 1. / a;
695 const PetscErrorCode ierr = VecScale(
vector, factor);
707 const PetscErrorCode ierr = VecAXPY(
vector, 1, v);
719 const PetscErrorCode ierr = VecAXPY(
vector, -1, v);
733 const PetscErrorCode ierr = VecShift(
vector, s);
745 const PetscErrorCode ierr = VecAXPY(
vector, a, v);
761 const PetscScalar weights[2] = {a, b};
764 const PetscErrorCode ierr = VecMAXPY(
vector, 2, weights, addends);
776 const PetscErrorCode ierr = VecAYPX(
vector, s, v);
804 const PetscErrorCode ierr = VecPointwiseMult(
vector, factors,
vector);
835 PetscErrorCode ierr =
836 PetscViewerSetFormat(PETSC_VIEWER_STDOUT_WORLD, format);
840 ierr = VecView(
vector, PETSC_VIEWER_STDOUT_WORLD);
848 const unsigned int precision,
849 const bool scientific,
850 const bool across)
const 857 PetscErrorCode ierr = VecGetArray(
vector, &val);
862 const std::ios::fmtflags old_flags = out.flags();
863 const unsigned int old_precision = out.precision(precision);
865 out.precision(precision);
867 out.setf(std::ios::scientific, std::ios::floatfield);
869 out.setf(std::ios::fixed, std::ios::floatfield);
873 out << val[i] <<
' ';
876 out << val[i] << std::endl;
880 out.flags(old_flags);
881 out.precision(old_precision);
885 ierr = VecRestoreArray(
vector, &val);
902 VectorBase::operator
const Vec &()
const 911 std::size_t mem =
sizeof(Vec) +
sizeof(
last_action) +
933 const PetscScalar *values,
934 const bool add_values)
941 internal::VectorReference::ExcWrongMode(action,
last_action));
951 const PetscInt *petsc_indices =
952 reinterpret_cast<const PetscInt *
>(indices);
954 const InsertMode mode = (add_values ? ADD_VALUES : INSERT_VALUES);
955 const PetscErrorCode ierr =
956 VecSetValues(
vector, n_elements, petsc_indices, values, mode);
969 #endif // DEAL_II_WITH_PETSC
#define AssertNothrow(cond, exc)
VectorBase & operator-=(const VectorBase &V)
types::global_dof_index size_type
static ::ExceptionBase & ExcIO()
void sadd(const PetscScalar s, const VectorBase &V)
real_type l2_norm() const
#define AssertIndexRange(index, range)
bool is_non_negative() const
double norm(const FEValuesBase< dim > &fe, const ArrayView< const std::vector< Tensor< 1, dim >>> &Du)
VectorBase & operator/=(const PetscScalar factor)
#define AssertThrow(cond, exc)
real_type norm_sqr() const
void scale(const VectorBase &scaling_factors)
real_type linfty_norm() const
VectorBase & operator=(const VectorBase &)=delete
std::pair< size_type, size_type > local_range() const
real_type lp_norm(const real_type p) const
bool is_non_negative(const T &t)
static ::ExceptionBase & ExcMessage(std::string arg1)
size_type local_size() const
void compress(const VectorOperation::values operation)
T sum(const T &t, const MPI_Comm &mpi_communicator)
void do_set_add_operation(const size_type n_elements, const size_type *indices, const PetscScalar *values, const bool add_values)
VectorOperation::values last_action
#define Assert(cond, exc)
static ::ExceptionBase & ExcDimensionMismatch(std::size_t arg1, std::size_t arg2)
static bool is_running_single_threaded()
#define DEAL_II_NAMESPACE_CLOSE
PetscScalar add_and_dot(const PetscScalar a, const VectorBase &V, const VectorBase &W)
VectorType::value_type * end(VectorType &V)
std::size_t memory_consumption() const
void equ(const PetscScalar a, const VectorBase &V)
SymmetricTensor< 2, dim, Number > d(const Tensor< 2, dim, Number > &F, const Tensor< 2, dim, Number > &dF_dt)
VectorBase & operator*=(const PetscScalar factor)
real_type l1_norm() const
SymmetricTensor< 2, dim, Number > b(const Tensor< 2, dim, Number > &F)
static ::ExceptionBase & ExcGhostsPresent()
Tensor< 2, dim, Number > w(const Tensor< 2, dim, Number > &F, const Tensor< 2, dim, Number > &dF_dt)
VectorBase & operator+=(const VectorBase &V)
void print(std::ostream &out, const unsigned int precision=3, const bool scientific=true, const bool across=true) const
#define AssertThrowMPI(error_code)
PetscScalar operator*(const VectorBase &vec) const
bool has_ghost_elements() const
PetscScalar mean_value() const
#define DEAL_II_NAMESPACE_OPEN
VectorType::value_type * begin(VectorType &V)
bool operator==(const VectorBase &v) const
void add(const std::vector< size_type > &indices, const std::vector< PetscScalar > &values)
void write_ascii(const PetscViewerFormat format=PETSC_VIEWER_DEFAULT)
virtual const MPI_Comm & get_mpi_communicator() const
virtual ~VectorBase() override
size_type n_elements() const
#define AssertIsFinite(number)
std::enable_if< std::is_fundamental< T >::value, std::size_t >::type memory_consumption(const T &t)
bool operator!=(const VectorBase &v) const
void set(const std::vector< size_type > &indices, const std::vector< PetscScalar > &values)