xref: /petsc/src/mat/tests/ex133.c (revision c4762a1b19cd2af06abeed90e8f9d34fb975dd94)
1*c4762a1bSJed Brown 
2*c4762a1bSJed Brown static char help[] = "Test saving SeqSBAIJ matrix that is missing diagonal entries.";
3*c4762a1bSJed Brown 
4*c4762a1bSJed Brown #include <petscmat.h>
5*c4762a1bSJed Brown 
6*c4762a1bSJed Brown int main(int argc,char **args)
7*c4762a1bSJed Brown {
8*c4762a1bSJed Brown   Mat            A;
9*c4762a1bSJed Brown   PetscInt       bs=3,m=4,i,j,val = 10,row[2],col[3],rstart;
10*c4762a1bSJed Brown   PetscErrorCode ierr;
11*c4762a1bSJed Brown   PetscMPIInt    size;
12*c4762a1bSJed Brown   PetscScalar    x[6][9];
13*c4762a1bSJed Brown 
14*c4762a1bSJed Brown   ierr = PetscInitialize(&argc,&args,(char*)0,help);if (ierr) return ierr;
15*c4762a1bSJed Brown   ierr = MPI_Comm_size(PETSC_COMM_WORLD,&size);CHKERRQ(ierr);
16*c4762a1bSJed Brown   if (size > 1) SETERRQ(PETSC_COMM_WORLD,PETSC_ERR_SUP,"Test is only for seqeuntial");
17*c4762a1bSJed Brown   ierr   = MatCreateSeqSBAIJ(PETSC_COMM_SELF,bs,m*bs,m*bs,1,NULL,&A);CHKERRQ(ierr);
18*c4762a1bSJed Brown   ierr   = MatSetOption(A,MAT_IGNORE_LOWER_TRIANGULAR,PETSC_TRUE);CHKERRQ(ierr);
19*c4762a1bSJed Brown   rstart = 0;
20*c4762a1bSJed Brown 
21*c4762a1bSJed Brown   row[0] =rstart+0;  row[1] =rstart+2;
22*c4762a1bSJed Brown   col[0] =rstart+0;  col[1] =rstart+1;  col[2] =rstart+3;
23*c4762a1bSJed Brown   for (i=0; i<6; i++) {
24*c4762a1bSJed Brown     for (j =0; j< 9; j++) x[i][j] = (PetscScalar)val++;
25*c4762a1bSJed Brown   }
26*c4762a1bSJed Brown   ierr = MatSetValuesBlocked(A,2,row,3,col,&x[0][0],INSERT_VALUES);CHKERRQ(ierr);
27*c4762a1bSJed Brown   ierr = MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
28*c4762a1bSJed Brown   ierr = MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
29*c4762a1bSJed Brown   ierr = MatView(A,PETSC_VIEWER_BINARY_WORLD);CHKERRQ(ierr);
30*c4762a1bSJed Brown 
31*c4762a1bSJed Brown   ierr = MatDestroy(&A);CHKERRQ(ierr);
32*c4762a1bSJed Brown   ierr = PetscFinalize();
33*c4762a1bSJed Brown   return ierr;
34*c4762a1bSJed Brown }
35