Actual source code: ex133.c

petsc-3.4.5 2014-06-29
  2: static char help[] = "Test saving SeqSBAIJ matrix that is missing diagonal entries.";

  4: #include <petscmat.h>

  8: int main(int argc,char **args)
  9: {
 10:   Mat            A;
 11:   PetscInt       bs=3,m=4,i,j,val = 10,row[2],col[3],rstart;
 13:   PetscMPIInt    size;
 14:   PetscScalar    x[6][9];

 16:   PetscInitialize(&argc,&args,(char*)0,help);

 18:   MPI_Comm_size(PETSC_COMM_WORLD,&size);
 19:   if (size > 1) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_SUP,"Test is only for seqeuntial");
 20:   MatCreateSeqSBAIJ(PETSC_COMM_SELF,bs,m*bs,m*bs,1,NULL,&A);
 21:   MatSetOption(A,MAT_IGNORE_LOWER_TRIANGULAR,PETSC_TRUE);
 22:   rstart = 0;

 24:   row[0] =rstart+0;  row[1] =rstart+2;
 25:   col[0] =rstart+0;  col[1] =rstart+1;  col[2] =rstart+3;
 26:   for (i=0; i<6; i++) {
 27:     for (j =0; j< 9; j++) x[i][j] = (PetscScalar)val++;
 28:   }
 29:   MatSetValuesBlocked(A,2,row,3,col,&x[0][0],INSERT_VALUES);
 30:   MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);
 31:   MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);
 32:   MatView(A,PETSC_VIEWER_BINARY_WORLD);

 34:   MatDestroy(&A);
 35:   PetscFinalize();
 36:   return 0;
 37: }