Actual source code: ex61.c
1: /*$Id: ex61.c,v 1.13 2001/08/07 03:03:07 balay Exp $*/
3: static char help[] = "Tests MatSeq(B)AIJSetColumnIndices().nn";
5: #include petscmat.h
7: /*
8: Generate the following matrix:
10: 1 0 3
11: 1 2 3
12: 0 0 3
13: */
14: int main(int argc,char **args)
15: {
16: Mat A;
17: PetscScalar v;
18: int ierr,i,j,rowlens[] = {2,3,1},cols[] = {0,2,0,1,2,2};
19: PetscTruth flg;
21: PetscInitialize(&argc,&args,(char *)0,help);
23: PetscOptionsHasName(PETSC_NULL,"-baij",&flg);
24: if (flg) {
25: MatCreateSeqBAIJ(PETSC_COMM_WORLD,1,3,3,PETSC_NULL,rowlens,&A);
26: MatSeqBAIJSetColumnIndices(A,cols);
27: } else {
28: MatCreateSeqAIJ(PETSC_COMM_WORLD,3,3,PETSC_NULL,rowlens,&A);
29: MatSeqAIJSetColumnIndices(A,cols);
30: }
32: i = 0; j = 0; v = 1.0;
33: MatSetValues(A,1,&i,1,&j,&v,INSERT_VALUES);
34: i = 0; j = 2; v = 3.0;
35: MatSetValues(A,1,&i,1,&j,&v,INSERT_VALUES);
37: i = 1; j = 0; v = 1.0;
38: MatSetValues(A,1,&i,1,&j,&v,INSERT_VALUES);
39: i = 1; j = 1; v = 2.0;
40: MatSetValues(A,1,&i,1,&j,&v,INSERT_VALUES);
41: i = 1; j = 2; v = 3.0;
42: MatSetValues(A,1,&i,1,&j,&v,INSERT_VALUES);
44: i = 2; j = 2; v = 3.0;
45: MatSetValues(A,1,&i,1,&j,&v,INSERT_VALUES);
47: MatAssemblyBegin(A,MAT_FINAL_ASSEMBLY);
48: MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY);
49: MatView(A,PETSC_VIEWER_STDOUT_WORLD);
51: MatDestroy(A);
52: PetscFinalize();
53: return 0;
54: }