1: ! Test MatCreateNest() with NULL index sets

  3: program main
  4: #include <petsc/finclude/petscmat.h>
  5: use petscmat
  6: implicit none

  8:   Mat                   :: A,D,Id,Acopy
  9:   Mat,dimension(4)      :: mats
 10:   Vec                   :: v,w
 11:   PetscInt              :: i,rstart,rend
 12:   PetscInt,parameter    :: n=6,nb=2
 13:   PetscScalar,parameter :: one=1.0, two=2.0
 14:   PetscErrorCode        :: ierr

 16:   PetscCallA(PetscInitialize(ierr))

 18:   PetscCallA(MatCreateConstantDiagonal(PETSC_COMM_WORLD,PETSC_DETERMINE,PETSC_DETERMINE,n,n,one,Id,ierr))
 19:   PetscCallA(MatCreateVecs(Id,v,w,ierr))
 20:   PetscCallA(VecGetOwnershipRange(v,rstart,rend,ierr))
 21:   do i=rstart,rend-1
 22:     PetscCallA(VecSetValue(v,i,two/(i+1),INSERT_VALUES,ierr))
 23:   end do
 24:   PetscCallA(VecAssemblyBegin(v,ierr))
 25:   PetscCallA(VecAssemblyEnd(v,ierr))
 26:   PetscCallA(MatCreateDiagonal(v,D,ierr))

 28:   mats = [PETSC_NULL_MAT, D, Id, PETSC_NULL_MAT]
 29:   PetscCallA(MatCreateNest(PETSC_COMM_WORLD,nb,PETSC_NULL_IS_ARRAY,nb,PETSC_NULL_IS_ARRAY,mats,A,ierr))
 30:   PetscCallA(MatView(A,PETSC_VIEWER_STDOUT_WORLD,ierr))
 31:   PetscCallA(MatNestSetSubMats(A,nb,PETSC_NULL_IS_ARRAY,nb,PETSC_NULL_IS_ARRAY,mats,ierr))
 32:   PetscCallA(MatView(A,PETSC_VIEWER_STDOUT_WORLD,ierr))

 34:   ! test MatCopy()
 35:   PetscCallA(MatDuplicate(A,MAT_DO_NOT_COPY_VALUES,Acopy,ierr))
 36:   PetscCallA(MatCopy(A,Acopy,DIFFERENT_NONZERO_PATTERN,ierr))

 38:   PetscCallA(MatDestroy(Acopy,ierr))
 39:   PetscCallA(MatDestroy(Id,ierr))
 40:   PetscCallA(VecDestroy(v,ierr))
 41:   PetscCallA(VecDestroy(w,ierr))
 42:   PetscCallA(MatDestroy(D,ierr))
 43:   PetscCallA(MatDestroy(A,ierr))
 44:   PetscCallA(PetscFinalize(ierr))

 46: end program

 48: !/*TEST
 49: !
 50: !   test:
 51: !      nsize: 2
 52: !
 53: !TEST*/