/*@C
ISGetNonlocalIndices - Retrieve an array of indices from remote processors
in this communicator.
Collective on IS
Input Parameter:
. is - the index set
Output Parameter:
. indices - indices with rank 0 indices first, and so on, omitting
the current rank. Total number of indices is the difference
total and local, obtained with ISGetSize() and ISGetLocalSize(),
respectively.
Level: intermediate
Notes: restore the indices using ISRestoreNonlocalIndices().
The same scalability considerations as those for ISGetTotalIndices
apply here.
Concepts: index sets^getting nonlocal indices
.seealso: ISGetTotalIndices(), ISRestoreNonlocalIndices(), ISGetSize(), ISGetLocalSize().
@*/
PetscErrorCode ISGetNonlocalIndices(IS is, const PetscInt *indices[])
{
PetscErrorCode ierr;
PetscMPIInt size;
PetscInt n, N;
PetscFunctionBegin;
PetscValidHeaderSpecific(is,IS_CLASSID,1);
PetscValidPointer(indices,2);
ierr = MPI_Comm_size(PetscObjectComm((PetscObject)is), &size);
CHKERRQ(ierr);
if (size == 1) *indices = NULL;
else {
if (!is->total) {
ierr = ISGatherTotal_Private(is);
CHKERRQ(ierr);
}
ierr = ISGetLocalSize(is,&n);
CHKERRQ(ierr);
ierr = ISGetSize(is,&N);
CHKERRQ(ierr);
ierr = PetscMalloc(sizeof(PetscInt)*(N-n), &(is->nonlocal));
CHKERRQ(ierr);
ierr = PetscMemcpy(is->nonlocal, is->total, sizeof(PetscInt)*is->local_offset);
CHKERRQ(ierr);
ierr = PetscMemcpy(is->nonlocal+is->local_offset, is->total+is->local_offset+n, sizeof(PetscInt)*(N - is->local_offset - n));
CHKERRQ(ierr);
*indices = is->nonlocal;
}
PetscFunctionReturn(0);
}
/*@C
ISGetTotalIndices - Retrieve an array containing all indices across the communicator.
Collective on IS
Input Parameter:
. is - the index set
Output Parameter:
. indices - total indices with rank 0 indices first, and so on; total array size is
the same as returned with ISGetSize().
Level: intermediate
Notes: this is potentially nonscalable, but depends on the size of the total index set
and the size of the communicator. This may be feasible for index sets defined on
subcommunicators, such that the set size does not grow with PETSC_WORLD_COMM.
Note also that there is no way to tell where the local part of the indices starts
(use ISGetIndices() and ISGetNonlocalIndices() to retrieve just the local and just
the nonlocal part (complement), respectively).
Concepts: index sets^getting nonlocal indices
.seealso: ISRestoreTotalIndices(), ISGetNonlocalIndices(), ISGetSize()
@*/
PetscErrorCode ISGetTotalIndices(IS is, const PetscInt *indices[])
{
PetscErrorCode ierr;
PetscMPIInt size;
PetscFunctionBegin;
PetscValidHeaderSpecific(is,IS_CLASSID,1);
PetscValidPointer(indices,2);
ierr = MPI_Comm_size(PetscObjectComm((PetscObject)is), &size);CHKERRQ(ierr);
if (size == 1) {
ierr = (*is->ops->getindices)(is,indices);CHKERRQ(ierr);
} else {
if (!is->total) {
ierr = ISGatherTotal_Private(is);CHKERRQ(ierr);
}
*indices = is->total;
}
PetscFunctionReturn(0);
}