xref: /petsc/src/mat/impls/sbaij/seq/sbaij.c (revision 759322aff80acb813c83cd009d678c91f6222bd3)
1be1d678aSKris Buschelman #define PETSCMAT_DLL
249b5e25fSSatish Balay 
349b5e25fSSatish Balay /*
4a1373b80SHong Zhang     Defines the basic matrix operations for the SBAIJ (compressed row)
549b5e25fSSatish Balay   matrix storage format.
649b5e25fSSatish Balay */
79e070d67SMatthew Knepley #include "src/mat/impls/baij/seq/baij.h"         /*I "petscmat.h" I*/
849b5e25fSSatish Balay #include "src/inline/spops.h"
9aec82049SSatish Balay #include "src/mat/impls/sbaij/seq/sbaij.h"
1049b5e25fSSatish Balay 
1149b5e25fSSatish Balay #define CHUNKSIZE  10
1249b5e25fSSatish Balay 
1349b5e25fSSatish Balay /*
1449b5e25fSSatish Balay      Checks for missing diagonals
1549b5e25fSSatish Balay */
164a2ae208SSatish Balay #undef __FUNCT__
174a2ae208SSatish Balay #define __FUNCT__ "MatMissingDiagonal_SeqSBAIJ"
18dfbe8321SBarry Smith PetscErrorCode MatMissingDiagonal_SeqSBAIJ(Mat A)
1949b5e25fSSatish Balay {
20045c9aa0SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
216849ba73SBarry Smith   PetscErrorCode ierr;
2213f74950SBarry Smith   PetscInt       *diag,*jj = a->j,i;
2349b5e25fSSatish Balay 
2449b5e25fSSatish Balay   PetscFunctionBegin;
25045c9aa0SHong Zhang   ierr = MatMarkDiagonal_SeqSBAIJ(A);CHKERRQ(ierr);
2649b5e25fSSatish Balay   diag = a->diag;
2749b5e25fSSatish Balay   for (i=0; i<a->mbs; i++) {
2877431f27SBarry Smith     if (jj[diag[i]] != i) SETERRQ1(PETSC_ERR_ARG_CORRUPT,"Matrix is missing diagonal number %D",i);
2949b5e25fSSatish Balay   }
3049b5e25fSSatish Balay   PetscFunctionReturn(0);
3149b5e25fSSatish Balay }
3249b5e25fSSatish Balay 
334a2ae208SSatish Balay #undef __FUNCT__
344a2ae208SSatish Balay #define __FUNCT__ "MatMarkDiagonal_SeqSBAIJ"
35dfbe8321SBarry Smith PetscErrorCode MatMarkDiagonal_SeqSBAIJ(Mat A)
3649b5e25fSSatish Balay {
37045c9aa0SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
386849ba73SBarry Smith   PetscErrorCode ierr;
3909f38230SBarry Smith   PetscInt       i;
4049b5e25fSSatish Balay 
4149b5e25fSSatish Balay   PetscFunctionBegin;
4209f38230SBarry Smith   if (!a->diag) {
4309f38230SBarry Smith     ierr = PetscMalloc(a->mbs*sizeof(PetscInt),&a->diag);CHKERRQ(ierr);
4409f38230SBarry Smith   }
4509f38230SBarry Smith   for (i=0; i<a->mbs; i++) a->diag[i] = a->i[i];
4649b5e25fSSatish Balay   PetscFunctionReturn(0);
4749b5e25fSSatish Balay }
4849b5e25fSSatish Balay 
494a2ae208SSatish Balay #undef __FUNCT__
504a2ae208SSatish Balay #define __FUNCT__ "MatGetRowIJ_SeqSBAIJ"
5113f74950SBarry Smith static PetscErrorCode MatGetRowIJ_SeqSBAIJ(Mat A,PetscInt oshift,PetscTruth symmetric,PetscInt *nn,PetscInt *ia[],PetscInt *ja[],PetscTruth *done)
5249b5e25fSSatish Balay {
53a6ece127SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
5413f74950SBarry Smith   PetscInt     n = a->mbs,i;
5549b5e25fSSatish Balay 
5649b5e25fSSatish Balay   PetscFunctionBegin;
57d3e5a4abSHong Zhang   *nn = n;
58a1373b80SHong Zhang   if (!ia) PetscFunctionReturn(0);
59a1373b80SHong Zhang 
60a6ece127SHong Zhang   if (oshift == 1) {
61a6ece127SHong Zhang     /* temporarily add 1 to i and j indices */
6213f74950SBarry Smith     PetscInt nz = a->i[n];
636c6c5352SBarry Smith     for (i=0; i<nz; i++) a->j[i]++;
64a1373b80SHong Zhang     for (i=0; i<n+1; i++) a->i[i]++;
65a1373b80SHong Zhang     *ia = a->i; *ja = a->j;
66a1373b80SHong Zhang   } else {
67a1373b80SHong Zhang     *ia = a->i; *ja = a->j;
68a6ece127SHong Zhang   }
6949b5e25fSSatish Balay   PetscFunctionReturn(0);
7049b5e25fSSatish Balay }
7149b5e25fSSatish Balay 
724a2ae208SSatish Balay #undef __FUNCT__
734a2ae208SSatish Balay #define __FUNCT__ "MatRestoreRowIJ_SeqSBAIJ"
7413f74950SBarry Smith static PetscErrorCode MatRestoreRowIJ_SeqSBAIJ(Mat A,PetscInt oshift,PetscTruth symmetric,PetscInt *nn,PetscInt *ia[],PetscInt *ja[],PetscTruth *done)
7549b5e25fSSatish Balay {
76b7aaefc3SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
7713f74950SBarry Smith   PetscInt     i,n = a->mbs;
78a6ece127SHong Zhang 
7949b5e25fSSatish Balay   PetscFunctionBegin;
8049b5e25fSSatish Balay   if (!ia) PetscFunctionReturn(0);
81a6ece127SHong Zhang 
82a6ece127SHong Zhang   if (oshift == 1) {
8313f74950SBarry Smith     PetscInt nz = a->i[n]-1;
846c6c5352SBarry Smith     for (i=0; i<nz; i++) a->j[i]--;
85a6ece127SHong Zhang     for (i=0; i<n+1; i++) a->i[i]--;
86a6ece127SHong Zhang   }
87a6ece127SHong Zhang   PetscFunctionReturn(0);
8849b5e25fSSatish Balay }
8949b5e25fSSatish Balay 
904a2ae208SSatish Balay #undef __FUNCT__
914a2ae208SSatish Balay #define __FUNCT__ "MatDestroy_SeqSBAIJ"
92dfbe8321SBarry Smith PetscErrorCode MatDestroy_SeqSBAIJ(Mat A)
9349b5e25fSSatish Balay {
9449b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
95dfbe8321SBarry Smith   PetscErrorCode ierr;
9649b5e25fSSatish Balay 
9749b5e25fSSatish Balay   PetscFunctionBegin;
98a9f03627SSatish Balay #if defined(PETSC_USE_LOG)
99899cda47SBarry Smith   PetscLogObjectState((PetscObject)A,"Rows=%D, NZ=%D",A->rmap.N,a->nz);
100a9f03627SSatish Balay #endif
101e6b907acSBarry Smith   ierr = MatSeqXAIJFreeAIJ(A,&a->a,&a->j,&a->i);CHKERRQ(ierr);
1029bfd6278SHong Zhang   if (a->row) {ierr = ISDestroy(a->row);CHKERRQ(ierr);}
1039bfd6278SHong Zhang   if (a->col){ierr = ISDestroy(a->col);CHKERRQ(ierr);}
1049bfd6278SHong Zhang   if (a->icol) {ierr = ISDestroy(a->icol);CHKERRQ(ierr);}
10505b42c5fSBarry Smith   ierr = PetscFree(a->diag);CHKERRQ(ierr);
10605b42c5fSBarry Smith   ierr = PetscFree2(a->imax,a->ilen);CHKERRQ(ierr);
10705b42c5fSBarry Smith   ierr = PetscFree(a->solve_work);CHKERRQ(ierr);
108e2ee2a47SBarry Smith   ierr = PetscFree(a->relax_work);CHKERRQ(ierr);
10905b42c5fSBarry Smith   ierr = PetscFree(a->solves_work);CHKERRQ(ierr);
11005b42c5fSBarry Smith   ierr = PetscFree(a->mult_work);CHKERRQ(ierr);
11105b42c5fSBarry Smith   ierr = PetscFree(a->saved_values);CHKERRQ(ierr);
11205b42c5fSBarry Smith   ierr = PetscFree(a->xtoy);CHKERRQ(ierr);
1131a3463dfSHong Zhang 
1141a3463dfSHong Zhang   ierr = PetscFree(a->inew);CHKERRQ(ierr);
11549b5e25fSSatish Balay   ierr = PetscFree(a);CHKERRQ(ierr);
116901853e0SKris Buschelman 
117dbd8c25aSHong Zhang   ierr = PetscObjectChangeTypeName((PetscObject)A,0);CHKERRQ(ierr);
118901853e0SKris Buschelman   ierr = PetscObjectComposeFunction((PetscObject)A,"MatStoreValues_C","",PETSC_NULL);CHKERRQ(ierr);
119901853e0SKris Buschelman   ierr = PetscObjectComposeFunction((PetscObject)A,"MatRetrieveValues_C","",PETSC_NULL);CHKERRQ(ierr);
120901853e0SKris Buschelman   ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetColumnIndices_C","",PETSC_NULL);CHKERRQ(ierr);
121901853e0SKris Buschelman   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqaij_C","",PETSC_NULL);CHKERRQ(ierr);
122901853e0SKris Buschelman   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqbaij_C","",PETSC_NULL);CHKERRQ(ierr);
123901853e0SKris Buschelman   ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetPreallocation_C","",PETSC_NULL);CHKERRQ(ierr);
12449b5e25fSSatish Balay   PetscFunctionReturn(0);
12549b5e25fSSatish Balay }
12649b5e25fSSatish Balay 
1274a2ae208SSatish Balay #undef __FUNCT__
1284a2ae208SSatish Balay #define __FUNCT__ "MatSetOption_SeqSBAIJ"
129dfbe8321SBarry Smith PetscErrorCode MatSetOption_SeqSBAIJ(Mat A,MatOption op)
13049b5e25fSSatish Balay {
131045c9aa0SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
13263ba0a88SBarry Smith   PetscErrorCode ierr;
13349b5e25fSSatish Balay 
13449b5e25fSSatish Balay   PetscFunctionBegin;
1354d9d31abSKris Buschelman   switch (op) {
1364d9d31abSKris Buschelman   case MAT_ROW_ORIENTED:
1374d9d31abSKris Buschelman     a->roworiented = PETSC_TRUE;
1384d9d31abSKris Buschelman     break;
1394d9d31abSKris Buschelman   case MAT_COLUMN_ORIENTED:
1404d9d31abSKris Buschelman     a->roworiented = PETSC_FALSE;
1414d9d31abSKris Buschelman     break;
1424d9d31abSKris Buschelman   case MAT_COLUMNS_SORTED:
1434d9d31abSKris Buschelman     a->sorted = PETSC_TRUE;
1444d9d31abSKris Buschelman     break;
1454d9d31abSKris Buschelman   case MAT_COLUMNS_UNSORTED:
1464d9d31abSKris Buschelman     a->sorted = PETSC_FALSE;
1474d9d31abSKris Buschelman     break;
1484d9d31abSKris Buschelman   case MAT_KEEP_ZEROED_ROWS:
1494d9d31abSKris Buschelman     a->keepzeroedrows = PETSC_TRUE;
1504d9d31abSKris Buschelman     break;
1514d9d31abSKris Buschelman   case MAT_NO_NEW_NONZERO_LOCATIONS:
1524d9d31abSKris Buschelman     a->nonew = 1;
1534d9d31abSKris Buschelman     break;
1544d9d31abSKris Buschelman   case MAT_NEW_NONZERO_LOCATION_ERR:
1554d9d31abSKris Buschelman     a->nonew = -1;
1564d9d31abSKris Buschelman     break;
1574d9d31abSKris Buschelman   case MAT_NEW_NONZERO_ALLOCATION_ERR:
1584d9d31abSKris Buschelman     a->nonew = -2;
1594d9d31abSKris Buschelman     break;
1604d9d31abSKris Buschelman   case MAT_YES_NEW_NONZERO_LOCATIONS:
1614d9d31abSKris Buschelman     a->nonew = 0;
1624d9d31abSKris Buschelman     break;
1634d9d31abSKris Buschelman   case MAT_ROWS_SORTED:
1644d9d31abSKris Buschelman   case MAT_ROWS_UNSORTED:
1654d9d31abSKris Buschelman   case MAT_YES_NEW_DIAGONALS:
1664d9d31abSKris Buschelman   case MAT_IGNORE_OFF_PROC_ENTRIES:
1674d9d31abSKris Buschelman   case MAT_USE_HASH_TABLE:
168290bbb0aSBarry Smith     ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr);
1694d9d31abSKris Buschelman     break;
1704d9d31abSKris Buschelman   case MAT_NO_NEW_DIAGONALS:
17129bbc08cSBarry Smith     SETERRQ(PETSC_ERR_SUP,"MAT_NO_NEW_DIAGONALS");
1729a4540c5SBarry Smith   case MAT_NOT_SYMMETRIC:
1739a4540c5SBarry Smith   case MAT_NOT_STRUCTURALLY_SYMMETRIC:
1749a4540c5SBarry Smith   case MAT_HERMITIAN:
1759a4540c5SBarry Smith     SETERRQ(PETSC_ERR_SUP,"Matrix must be symmetric");
17677e54ba9SKris Buschelman   case MAT_SYMMETRIC:
17777e54ba9SKris Buschelman   case MAT_STRUCTURALLY_SYMMETRIC:
1789a4540c5SBarry Smith   case MAT_NOT_HERMITIAN:
1799a4540c5SBarry Smith   case MAT_SYMMETRY_ETERNAL:
1809a4540c5SBarry Smith   case MAT_NOT_SYMMETRY_ETERNAL:
181290bbb0aSBarry Smith     ierr = PetscInfo1(A,"Option %s not relevent\n",MatOptions[op]);CHKERRQ(ierr);
182290bbb0aSBarry Smith     break;
183941593c8SHong Zhang   case MAT_IGNORE_LOWER_TRIANGULAR:
184941593c8SHong Zhang     a->ignore_ltriangular = PETSC_TRUE;
185941593c8SHong Zhang     break;
186941593c8SHong Zhang   case MAT_ERROR_LOWER_TRIANGULAR:
187941593c8SHong Zhang     a->ignore_ltriangular = PETSC_FALSE;
18877e54ba9SKris Buschelman     break;
189f5edf698SHong Zhang   case MAT_GETROW_UPPERTRIANGULAR:
190f5edf698SHong Zhang     a->getrow_utriangular = PETSC_TRUE;
191f5edf698SHong Zhang     break;
1924d9d31abSKris Buschelman   default:
193ad86a440SBarry Smith     SETERRQ1(PETSC_ERR_SUP,"unknown option %d",op);
19449b5e25fSSatish Balay   }
19549b5e25fSSatish Balay   PetscFunctionReturn(0);
19649b5e25fSSatish Balay }
19749b5e25fSSatish Balay 
1984a2ae208SSatish Balay #undef __FUNCT__
1994a2ae208SSatish Balay #define __FUNCT__ "MatGetRow_SeqSBAIJ"
20013f74950SBarry Smith PetscErrorCode MatGetRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *ncols,PetscInt **cols,PetscScalar **v)
20149b5e25fSSatish Balay {
20249b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
2036849ba73SBarry Smith   PetscErrorCode ierr;
20413f74950SBarry Smith   PetscInt       itmp,i,j,k,M,*ai,*aj,bs,bn,bp,*cols_i,bs2;
20549b5e25fSSatish Balay   MatScalar      *aa,*aa_i;
20687828ca2SBarry Smith   PetscScalar    *v_i;
20749b5e25fSSatish Balay 
20849b5e25fSSatish Balay   PetscFunctionBegin;
209f5edf698SHong Zhang   if (A && !a->getrow_utriangular) SETERRQ(PETSC_ERR_SUP,"MatGetRow is not supported for SBAIJ matrix format. Getting the upper triangular part of row, run with -mat_getrow_uppertriangular, call MatSetOption(mat,MAT_GETROW_UPPERTRIANGULAR) or MatGetRowUpperTriangular()");
210f5edf698SHong Zhang   /* Get the upper triangular part of the row */
211899cda47SBarry Smith   bs  = A->rmap.bs;
21249b5e25fSSatish Balay   ai  = a->i;
21349b5e25fSSatish Balay   aj  = a->j;
21449b5e25fSSatish Balay   aa  = a->a;
21549b5e25fSSatish Balay   bs2 = a->bs2;
21649b5e25fSSatish Balay 
217899cda47SBarry Smith   if (row < 0 || row >= A->rmap.N) SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE, "Row %D out of range", row);
21849b5e25fSSatish Balay 
21949b5e25fSSatish Balay   bn  = row/bs;   /* Block number */
22049b5e25fSSatish Balay   bp  = row % bs; /* Block position */
22149b5e25fSSatish Balay   M   = ai[bn+1] - ai[bn];
22249b5e25fSSatish Balay   *ncols = bs*M;
22349b5e25fSSatish Balay 
22449b5e25fSSatish Balay   if (v) {
22549b5e25fSSatish Balay     *v = 0;
22649b5e25fSSatish Balay     if (*ncols) {
22787828ca2SBarry Smith       ierr = PetscMalloc((*ncols+row)*sizeof(PetscScalar),v);CHKERRQ(ierr);
22849b5e25fSSatish Balay       for (i=0; i<M; i++) { /* for each block in the block row */
22949b5e25fSSatish Balay         v_i  = *v + i*bs;
23049b5e25fSSatish Balay         aa_i = aa + bs2*(ai[bn] + i);
23149b5e25fSSatish Balay         for (j=bp,k=0; j<bs2; j+=bs,k++) {v_i[k] = aa_i[j];}
23249b5e25fSSatish Balay       }
23349b5e25fSSatish Balay     }
23449b5e25fSSatish Balay   }
23549b5e25fSSatish Balay 
23649b5e25fSSatish Balay   if (cols) {
23749b5e25fSSatish Balay     *cols = 0;
23849b5e25fSSatish Balay     if (*ncols) {
23913f74950SBarry Smith       ierr = PetscMalloc((*ncols+row)*sizeof(PetscInt),cols);CHKERRQ(ierr);
24049b5e25fSSatish Balay       for (i=0; i<M; i++) { /* for each block in the block row */
24149b5e25fSSatish Balay         cols_i = *cols + i*bs;
24249b5e25fSSatish Balay         itmp  = bs*aj[ai[bn] + i];
24349b5e25fSSatish Balay         for (j=0; j<bs; j++) {cols_i[j] = itmp++;}
24449b5e25fSSatish Balay       }
24549b5e25fSSatish Balay     }
24649b5e25fSSatish Balay   }
24749b5e25fSSatish Balay 
24849b5e25fSSatish Balay   /*search column A(0:row-1,row) (=A(row,0:row-1)). Could be expensive! */
2495ddb2528SHong Zhang   /* this segment is currently removed, so only entries in the upper triangle are obtained */
2505ddb2528SHong Zhang #ifdef column_search
25149b5e25fSSatish Balay   v_i    = *v    + M*bs;
25249b5e25fSSatish Balay   cols_i = *cols + M*bs;
25349b5e25fSSatish Balay   for (i=0; i<bn; i++){ /* for each block row */
25449b5e25fSSatish Balay     M = ai[i+1] - ai[i];
25549b5e25fSSatish Balay     for (j=0; j<M; j++){
25649b5e25fSSatish Balay       itmp = aj[ai[i] + j];    /* block column value */
25749b5e25fSSatish Balay       if (itmp == bn){
25849b5e25fSSatish Balay         aa_i   = aa    + bs2*(ai[i] + j) + bs*bp;
25949b5e25fSSatish Balay         for (k=0; k<bs; k++) {
26049b5e25fSSatish Balay           *cols_i++ = i*bs+k;
26149b5e25fSSatish Balay           *v_i++    = aa_i[k];
26249b5e25fSSatish Balay         }
26349b5e25fSSatish Balay         *ncols += bs;
26449b5e25fSSatish Balay         break;
26549b5e25fSSatish Balay       }
26649b5e25fSSatish Balay     }
26749b5e25fSSatish Balay   }
2685ddb2528SHong Zhang #endif
26949b5e25fSSatish Balay   PetscFunctionReturn(0);
27049b5e25fSSatish Balay }
27149b5e25fSSatish Balay 
2724a2ae208SSatish Balay #undef __FUNCT__
2734a2ae208SSatish Balay #define __FUNCT__ "MatRestoreRow_SeqSBAIJ"
27413f74950SBarry Smith PetscErrorCode MatRestoreRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
27549b5e25fSSatish Balay {
276dfbe8321SBarry Smith   PetscErrorCode ierr;
27749b5e25fSSatish Balay 
27849b5e25fSSatish Balay   PetscFunctionBegin;
27905b42c5fSBarry Smith   if (idx) {ierr = PetscFree(*idx);CHKERRQ(ierr);}
28005b42c5fSBarry Smith   if (v)   {ierr = PetscFree(*v);CHKERRQ(ierr);}
28149b5e25fSSatish Balay   PetscFunctionReturn(0);
28249b5e25fSSatish Balay }
28349b5e25fSSatish Balay 
2844a2ae208SSatish Balay #undef __FUNCT__
285f5edf698SHong Zhang #define __FUNCT__ "MatGetRowUpperTriangular_SeqSBAIJ"
286f5edf698SHong Zhang PetscErrorCode MatGetRowUpperTriangular_SeqSBAIJ(Mat A)
287f5edf698SHong Zhang {
288f5edf698SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
289f5edf698SHong Zhang 
290f5edf698SHong Zhang   PetscFunctionBegin;
291f5edf698SHong Zhang   a->getrow_utriangular = PETSC_TRUE;
292f5edf698SHong Zhang   PetscFunctionReturn(0);
293f5edf698SHong Zhang }
294f5edf698SHong Zhang #undef __FUNCT__
295f5edf698SHong Zhang #define __FUNCT__ "MatRestoreRowUpperTriangular_SeqSBAIJ"
296f5edf698SHong Zhang PetscErrorCode MatRestoreRowUpperTriangular_SeqSBAIJ(Mat A)
297f5edf698SHong Zhang {
298f5edf698SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
299f5edf698SHong Zhang 
300f5edf698SHong Zhang   PetscFunctionBegin;
301f5edf698SHong Zhang   a->getrow_utriangular = PETSC_FALSE;
302f5edf698SHong Zhang   PetscFunctionReturn(0);
303f5edf698SHong Zhang }
304f5edf698SHong Zhang 
305f5edf698SHong Zhang #undef __FUNCT__
3064a2ae208SSatish Balay #define __FUNCT__ "MatTranspose_SeqSBAIJ"
307dfbe8321SBarry Smith PetscErrorCode MatTranspose_SeqSBAIJ(Mat A,Mat *B)
30849b5e25fSSatish Balay {
309dfbe8321SBarry Smith   PetscErrorCode ierr;
31049b5e25fSSatish Balay   PetscFunctionBegin;
311999d9058SBarry Smith   ierr = MatDuplicate(A,MAT_COPY_VALUES,B);CHKERRQ(ierr);
3128115998fSBarry Smith   PetscFunctionReturn(0);
31349b5e25fSSatish Balay }
31449b5e25fSSatish Balay 
3154a2ae208SSatish Balay #undef __FUNCT__
3164a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_ASCII"
3176849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_ASCII(Mat A,PetscViewer viewer)
31849b5e25fSSatish Balay {
31949b5e25fSSatish Balay   Mat_SeqSBAIJ      *a = (Mat_SeqSBAIJ*)A->data;
320dfbe8321SBarry Smith   PetscErrorCode    ierr;
321899cda47SBarry Smith   PetscInt          i,j,bs = A->rmap.bs,k,l,bs2=a->bs2;
3222dcb1b2aSMatthew Knepley   const char        *name;
323f3ef73ceSBarry Smith   PetscViewerFormat format;
32449b5e25fSSatish Balay 
32549b5e25fSSatish Balay   PetscFunctionBegin;
32680fe4e49SBarry Smith   ierr = PetscObjectGetName((PetscObject)A,&name);CHKERRQ(ierr);
327b0a32e0cSBarry Smith   ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
328456192e2SBarry Smith   if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
32977431f27SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  block size is %D\n",bs);CHKERRQ(ierr);
330fb9695e5SSatish Balay   } else if (format == PETSC_VIEWER_ASCII_MATLAB) {
33129bbc08cSBarry Smith     SETERRQ(PETSC_ERR_SUP,"Matlab format not supported");
332fb9695e5SSatish Balay   } else if (format == PETSC_VIEWER_ASCII_COMMON) {
333b0a32e0cSBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_NO);CHKERRQ(ierr);
33449b5e25fSSatish Balay     for (i=0; i<a->mbs; i++) {
33549b5e25fSSatish Balay       for (j=0; j<bs; j++) {
33677431f27SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr);
33749b5e25fSSatish Balay         for (k=a->i[i]; k<a->i[i+1]; k++) {
33849b5e25fSSatish Balay           for (l=0; l<bs; l++) {
33949b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX)
34049b5e25fSSatish Balay             if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
341a83599f4SBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %G + %G i) ",bs*a->j[k]+l,
34249b5e25fSSatish Balay                                             PetscRealPart(a->a[bs2*k + l*bs + j]),PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
34349b5e25fSSatish Balay             } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
344a83599f4SBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %G - %G i) ",bs*a->j[k]+l,
34549b5e25fSSatish Balay                                             PetscRealPart(a->a[bs2*k + l*bs + j]),-PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
34649b5e25fSSatish Balay             } else if (PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
347a83599f4SBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",bs*a->j[k]+l,PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
34849b5e25fSSatish Balay             }
34949b5e25fSSatish Balay #else
35049b5e25fSSatish Balay             if (a->a[bs2*k + l*bs + j] != 0.0) {
351a83599f4SBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",bs*a->j[k]+l,a->a[bs2*k + l*bs + j]);CHKERRQ(ierr);
35249b5e25fSSatish Balay             }
35349b5e25fSSatish Balay #endif
35449b5e25fSSatish Balay           }
35549b5e25fSSatish Balay         }
356b0a32e0cSBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
35749b5e25fSSatish Balay       }
35849b5e25fSSatish Balay     }
359b0a32e0cSBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_YES);CHKERRQ(ierr);
360c1490034SHong Zhang   } else if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) {
361c1490034SHong Zhang      PetscFunctionReturn(0);
36249b5e25fSSatish Balay   } else {
363b0a32e0cSBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_NO);CHKERRQ(ierr);
36449b5e25fSSatish Balay     for (i=0; i<a->mbs; i++) {
36549b5e25fSSatish Balay       for (j=0; j<bs; j++) {
36677431f27SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr);
36749b5e25fSSatish Balay         for (k=a->i[i]; k<a->i[i+1]; k++) {
36849b5e25fSSatish Balay           for (l=0; l<bs; l++) {
36949b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX)
37049b5e25fSSatish Balay             if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0) {
371a83599f4SBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %G + %G i) ",bs*a->j[k]+l,
37249b5e25fSSatish Balay                                             PetscRealPart(a->a[bs2*k + l*bs + j]),PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
37349b5e25fSSatish Balay             } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0) {
374a83599f4SBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %G - %G i) ",bs*a->j[k]+l,
37549b5e25fSSatish Balay                                             PetscRealPart(a->a[bs2*k + l*bs + j]),-PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
37649b5e25fSSatish Balay             } else {
377a83599f4SBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",bs*a->j[k]+l,PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
37849b5e25fSSatish Balay             }
37949b5e25fSSatish Balay #else
380e9f7bc9eSHong Zhang             ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",bs*a->j[k]+l,a->a[bs2*k + l*bs + j]);CHKERRQ(ierr);
38149b5e25fSSatish Balay #endif
38249b5e25fSSatish Balay           }
38349b5e25fSSatish Balay         }
384b0a32e0cSBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
38549b5e25fSSatish Balay       }
38649b5e25fSSatish Balay     }
387b0a32e0cSBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_YES);CHKERRQ(ierr);
38849b5e25fSSatish Balay   }
389b0a32e0cSBarry Smith   ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
39049b5e25fSSatish Balay   PetscFunctionReturn(0);
39149b5e25fSSatish Balay }
39249b5e25fSSatish Balay 
3934a2ae208SSatish Balay #undef __FUNCT__
3944a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_Draw_Zoom"
3956849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw_Zoom(PetscDraw draw,void *Aa)
39649b5e25fSSatish Balay {
39749b5e25fSSatish Balay   Mat            A = (Mat) Aa;
39849b5e25fSSatish Balay   Mat_SeqSBAIJ   *a=(Mat_SeqSBAIJ*)A->data;
3996849ba73SBarry Smith   PetscErrorCode ierr;
400899cda47SBarry Smith   PetscInt       row,i,j,k,l,mbs=a->mbs,color,bs=A->rmap.bs,bs2=a->bs2;
40113f74950SBarry Smith   PetscMPIInt    rank;
40249b5e25fSSatish Balay   PetscReal      xl,yl,xr,yr,x_l,x_r,y_l,y_r;
40349b5e25fSSatish Balay   MatScalar      *aa;
40449b5e25fSSatish Balay   MPI_Comm       comm;
405b0a32e0cSBarry Smith   PetscViewer    viewer;
40649b5e25fSSatish Balay 
40749b5e25fSSatish Balay   PetscFunctionBegin;
40849b5e25fSSatish Balay   /*
40949b5e25fSSatish Balay     This is nasty. If this is called from an originally parallel matrix
41049b5e25fSSatish Balay     then all processes call this,but only the first has the matrix so the
41149b5e25fSSatish Balay     rest should return immediately.
41249b5e25fSSatish Balay   */
41349b5e25fSSatish Balay   ierr = PetscObjectGetComm((PetscObject)draw,&comm);CHKERRQ(ierr);
41449b5e25fSSatish Balay   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
41549b5e25fSSatish Balay   if (rank) PetscFunctionReturn(0);
41649b5e25fSSatish Balay 
41749b5e25fSSatish Balay   ierr = PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer);CHKERRQ(ierr);
41849b5e25fSSatish Balay 
419b0a32e0cSBarry Smith   ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr);
420b0a32e0cSBarry Smith   PetscDrawString(draw, .3*(xl+xr), .3*(yl+yr), PETSC_DRAW_BLACK, "symmetric");
42149b5e25fSSatish Balay 
42249b5e25fSSatish Balay   /* loop over matrix elements drawing boxes */
423b0a32e0cSBarry Smith   color = PETSC_DRAW_BLUE;
42449b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
42549b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
426899cda47SBarry Smith       y_l = A->rmap.N - row - 1.0; y_r = y_l + 1.0;
42749b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
42849b5e25fSSatish Balay       aa = a->a + j*bs2;
42949b5e25fSSatish Balay       for (k=0; k<bs; k++) {
43049b5e25fSSatish Balay         for (l=0; l<bs; l++) {
43149b5e25fSSatish Balay           if (PetscRealPart(*aa++) >=  0.) continue;
432b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
43349b5e25fSSatish Balay         }
43449b5e25fSSatish Balay       }
43549b5e25fSSatish Balay     }
43649b5e25fSSatish Balay   }
437b0a32e0cSBarry Smith   color = PETSC_DRAW_CYAN;
43849b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
43949b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
440899cda47SBarry Smith       y_l = A->rmap.N - row - 1.0; y_r = y_l + 1.0;
44149b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
44249b5e25fSSatish Balay       aa = a->a + j*bs2;
44349b5e25fSSatish Balay       for (k=0; k<bs; k++) {
44449b5e25fSSatish Balay         for (l=0; l<bs; l++) {
44549b5e25fSSatish Balay           if (PetscRealPart(*aa++) != 0.) continue;
446b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
44749b5e25fSSatish Balay         }
44849b5e25fSSatish Balay       }
44949b5e25fSSatish Balay     }
45049b5e25fSSatish Balay   }
45149b5e25fSSatish Balay 
452b0a32e0cSBarry Smith   color = PETSC_DRAW_RED;
45349b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
45449b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
455899cda47SBarry Smith       y_l = A->rmap.N - row - 1.0; y_r = y_l + 1.0;
45649b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
45749b5e25fSSatish Balay       aa = a->a + j*bs2;
45849b5e25fSSatish Balay       for (k=0; k<bs; k++) {
45949b5e25fSSatish Balay         for (l=0; l<bs; l++) {
46049b5e25fSSatish Balay           if (PetscRealPart(*aa++) <= 0.) continue;
461b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
46249b5e25fSSatish Balay         }
46349b5e25fSSatish Balay       }
46449b5e25fSSatish Balay     }
46549b5e25fSSatish Balay   }
46649b5e25fSSatish Balay   PetscFunctionReturn(0);
46749b5e25fSSatish Balay }
46849b5e25fSSatish Balay 
4694a2ae208SSatish Balay #undef __FUNCT__
4704a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_Draw"
4716849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw(Mat A,PetscViewer viewer)
47249b5e25fSSatish Balay {
473dfbe8321SBarry Smith   PetscErrorCode ierr;
47449b5e25fSSatish Balay   PetscReal      xl,yl,xr,yr,w,h;
475b0a32e0cSBarry Smith   PetscDraw      draw;
47649b5e25fSSatish Balay   PetscTruth     isnull;
47749b5e25fSSatish Balay 
47849b5e25fSSatish Balay   PetscFunctionBegin;
47949b5e25fSSatish Balay 
480b0a32e0cSBarry Smith   ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
481b0a32e0cSBarry Smith   ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr); if (isnull) PetscFunctionReturn(0);
48249b5e25fSSatish Balay 
48349b5e25fSSatish Balay   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer);CHKERRQ(ierr);
484899cda47SBarry Smith   xr  = A->rmap.N; yr = A->rmap.N; h = yr/10.0; w = xr/10.0;
48549b5e25fSSatish Balay   xr += w;    yr += h;  xl = -w;     yl = -h;
486b0a32e0cSBarry Smith   ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr);
487b0a32e0cSBarry Smith   ierr = PetscDrawZoom(draw,MatView_SeqSBAIJ_Draw_Zoom,A);CHKERRQ(ierr);
48849b5e25fSSatish Balay   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",PETSC_NULL);CHKERRQ(ierr);
48949b5e25fSSatish Balay   PetscFunctionReturn(0);
49049b5e25fSSatish Balay }
49149b5e25fSSatish Balay 
4924a2ae208SSatish Balay #undef __FUNCT__
4934a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ"
494dfbe8321SBarry Smith PetscErrorCode MatView_SeqSBAIJ(Mat A,PetscViewer viewer)
49549b5e25fSSatish Balay {
496dfbe8321SBarry Smith   PetscErrorCode ierr;
49732077d6dSBarry Smith   PetscTruth     iascii,isdraw;
49849b5e25fSSatish Balay 
49949b5e25fSSatish Balay   PetscFunctionBegin;
50032077d6dSBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr);
501fb9695e5SSatish Balay   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_DRAW,&isdraw);CHKERRQ(ierr);
50232077d6dSBarry Smith   if (iascii){
50349b5e25fSSatish Balay     ierr = MatView_SeqSBAIJ_ASCII(A,viewer);CHKERRQ(ierr);
50449b5e25fSSatish Balay   } else if (isdraw) {
50549b5e25fSSatish Balay     ierr = MatView_SeqSBAIJ_Draw(A,viewer);CHKERRQ(ierr);
50649b5e25fSSatish Balay   } else {
507a5e6ed63SBarry Smith     Mat B;
508ceb03754SKris Buschelman     ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&B);CHKERRQ(ierr);
509a5e6ed63SBarry Smith     ierr = MatView(B,viewer);CHKERRQ(ierr);
510a5e6ed63SBarry Smith     ierr = MatDestroy(B);CHKERRQ(ierr);
51149b5e25fSSatish Balay   }
51249b5e25fSSatish Balay   PetscFunctionReturn(0);
51349b5e25fSSatish Balay }
51449b5e25fSSatish Balay 
51549b5e25fSSatish Balay 
5164a2ae208SSatish Balay #undef __FUNCT__
5174a2ae208SSatish Balay #define __FUNCT__ "MatGetValues_SeqSBAIJ"
51813f74950SBarry Smith PetscErrorCode MatGetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],PetscScalar v[])
51949b5e25fSSatish Balay {
520045c9aa0SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
52113f74950SBarry Smith   PetscInt     *rp,k,low,high,t,row,nrow,i,col,l,*aj = a->j;
52213f74950SBarry Smith   PetscInt     *ai = a->i,*ailen = a->ilen;
523899cda47SBarry Smith   PetscInt     brow,bcol,ridx,cidx,bs=A->rmap.bs,bs2=a->bs2;
52449b5e25fSSatish Balay   MatScalar    *ap,*aa = a->a,zero = 0.0;
52549b5e25fSSatish Balay 
52649b5e25fSSatish Balay   PetscFunctionBegin;
52749b5e25fSSatish Balay   for (k=0; k<m; k++) { /* loop over rows */
52849b5e25fSSatish Balay     row  = im[k]; brow = row/bs;
52977431f27SBarry Smith     if (row < 0) SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"Negative row: %D",row);
530899cda47SBarry Smith     if (row >= A->rmap.N) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,A->rmap.N-1);
53149b5e25fSSatish Balay     rp   = aj + ai[brow] ; ap = aa + bs2*ai[brow] ;
53249b5e25fSSatish Balay     nrow = ailen[brow];
53349b5e25fSSatish Balay     for (l=0; l<n; l++) { /* loop over columns */
53477431f27SBarry Smith       if (in[l] < 0) SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"Negative column: %D",in[l]);
535899cda47SBarry Smith       if (in[l] >= A->cmap.n) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",in[l],A->cmap.n-1);
53649b5e25fSSatish Balay       col  = in[l] ;
53749b5e25fSSatish Balay       bcol = col/bs;
53849b5e25fSSatish Balay       cidx = col%bs;
53949b5e25fSSatish Balay       ridx = row%bs;
54049b5e25fSSatish Balay       high = nrow;
54149b5e25fSSatish Balay       low  = 0; /* assume unsorted */
54249b5e25fSSatish Balay       while (high-low > 5) {
54349b5e25fSSatish Balay         t = (low+high)/2;
54449b5e25fSSatish Balay         if (rp[t] > bcol) high = t;
54549b5e25fSSatish Balay         else             low  = t;
54649b5e25fSSatish Balay       }
54749b5e25fSSatish Balay       for (i=low; i<high; i++) {
54849b5e25fSSatish Balay         if (rp[i] > bcol) break;
54949b5e25fSSatish Balay         if (rp[i] == bcol) {
55049b5e25fSSatish Balay           *v++ = ap[bs2*i+bs*cidx+ridx];
55149b5e25fSSatish Balay           goto finished;
55249b5e25fSSatish Balay         }
55349b5e25fSSatish Balay       }
55449b5e25fSSatish Balay       *v++ = zero;
55549b5e25fSSatish Balay        finished:;
55649b5e25fSSatish Balay     }
55749b5e25fSSatish Balay   }
55849b5e25fSSatish Balay   PetscFunctionReturn(0);
55949b5e25fSSatish Balay }
56049b5e25fSSatish Balay 
56149b5e25fSSatish Balay 
5624a2ae208SSatish Balay #undef __FUNCT__
5634a2ae208SSatish Balay #define __FUNCT__ "MatSetValuesBlocked_SeqSBAIJ"
56413f74950SBarry Smith PetscErrorCode MatSetValuesBlocked_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is)
56549b5e25fSSatish Balay {
5660880e062SHong Zhang   Mat_SeqSBAIJ    *a = (Mat_SeqSBAIJ*)A->data;
5676849ba73SBarry Smith   PetscErrorCode  ierr;
568e2ee6c50SBarry Smith   PetscInt        *rp,k,low,high,t,ii,jj,row,nrow,i,col,l,rmax,N,lastcol = -1;
56913f74950SBarry Smith   PetscInt        *imax=a->imax,*ai=a->i,*ailen=a->ilen;
570899cda47SBarry Smith   PetscInt        *aj=a->j,nonew=a->nonew,bs2=a->bs2,bs=A->rmap.bs,stepval;
5710880e062SHong Zhang   PetscTruth      roworiented=a->roworiented;
572f15d580aSBarry Smith   const MatScalar *value = v;
573f15d580aSBarry Smith   MatScalar       *ap,*aa = a->a,*bap;
5740880e062SHong Zhang 
57549b5e25fSSatish Balay   PetscFunctionBegin;
5760880e062SHong Zhang   if (roworiented) {
5770880e062SHong Zhang     stepval = (n-1)*bs;
5780880e062SHong Zhang   } else {
5790880e062SHong Zhang     stepval = (m-1)*bs;
5800880e062SHong Zhang   }
5810880e062SHong Zhang   for (k=0; k<m; k++) { /* loop over added rows */
5820880e062SHong Zhang     row  = im[k];
5830880e062SHong Zhang     if (row < 0) continue;
5842515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
58577431f27SBarry Smith     if (row >= a->mbs) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,a->mbs-1);
5860880e062SHong Zhang #endif
5870880e062SHong Zhang     rp   = aj + ai[row];
5880880e062SHong Zhang     ap   = aa + bs2*ai[row];
5890880e062SHong Zhang     rmax = imax[row];
5900880e062SHong Zhang     nrow = ailen[row];
5910880e062SHong Zhang     low  = 0;
592818f2c47SBarry Smith     high = nrow;
5930880e062SHong Zhang     for (l=0; l<n; l++) { /* loop over added columns */
5940880e062SHong Zhang       if (in[l] < 0) continue;
5950880e062SHong Zhang       col = in[l];
5962515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
59777431f27SBarry Smith       if (col >= a->nbs) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",col,a->nbs-1);
598b1823623SSatish Balay #endif
5990880e062SHong Zhang       if (col < row) continue; /* ignore lower triangular block */
6000880e062SHong Zhang       if (roworiented) {
6010880e062SHong Zhang         value = v + k*(stepval+bs)*bs + l*bs;
6020880e062SHong Zhang       } else {
6030880e062SHong Zhang         value = v + l*(stepval+bs)*bs + k*bs;
6040880e062SHong Zhang       }
6057cd84e04SBarry Smith       if (col <= lastcol) low = 0; else high = nrow;
606e2ee6c50SBarry Smith       lastcol = col;
6070880e062SHong Zhang       while (high-low > 7) {
6080880e062SHong Zhang         t = (low+high)/2;
6090880e062SHong Zhang         if (rp[t] > col) high = t;
6100880e062SHong Zhang         else             low  = t;
6110880e062SHong Zhang       }
6120880e062SHong Zhang       for (i=low; i<high; i++) {
6130880e062SHong Zhang         if (rp[i] > col) break;
6140880e062SHong Zhang         if (rp[i] == col) {
6150880e062SHong Zhang           bap  = ap +  bs2*i;
6160880e062SHong Zhang           if (roworiented) {
6170880e062SHong Zhang             if (is == ADD_VALUES) {
6180880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
6190880e062SHong Zhang                 for (jj=ii; jj<bs2; jj+=bs) {
6200880e062SHong Zhang                   bap[jj] += *value++;
6210880e062SHong Zhang                 }
6220880e062SHong Zhang               }
6230880e062SHong Zhang             } else {
6240880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
6250880e062SHong Zhang                 for (jj=ii; jj<bs2; jj+=bs) {
6260880e062SHong Zhang                   bap[jj] = *value++;
6270880e062SHong Zhang                 }
6280880e062SHong Zhang                }
6290880e062SHong Zhang             }
6300880e062SHong Zhang           } else {
6310880e062SHong Zhang             if (is == ADD_VALUES) {
6320880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
6330880e062SHong Zhang                 for (jj=0; jj<bs; jj++) {
6340880e062SHong Zhang                   *bap++ += *value++;
6350880e062SHong Zhang                 }
6360880e062SHong Zhang               }
6370880e062SHong Zhang             } else {
6380880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
6390880e062SHong Zhang                 for (jj=0; jj<bs; jj++) {
6400880e062SHong Zhang                   *bap++  = *value++;
6410880e062SHong Zhang                 }
6420880e062SHong Zhang               }
6430880e062SHong Zhang             }
6440880e062SHong Zhang           }
6450880e062SHong Zhang           goto noinsert2;
6460880e062SHong Zhang         }
6470880e062SHong Zhang       }
6480880e062SHong Zhang       if (nonew == 1) goto noinsert2;
649085a36d4SBarry Smith       if (nonew == -1) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col);
650421e10b8SBarry Smith       MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,row,col,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar);
651c03d1d03SSatish Balay       N = nrow++ - 1; high++;
6520880e062SHong Zhang       /* shift up all the later entries in this row */
6530880e062SHong Zhang       for (ii=N; ii>=i; ii--) {
6540880e062SHong Zhang         rp[ii+1] = rp[ii];
6550880e062SHong Zhang         ierr = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr);
6560880e062SHong Zhang       }
6570880e062SHong Zhang       if (N >= i) {
6580880e062SHong Zhang         ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr);
6590880e062SHong Zhang       }
6600880e062SHong Zhang       rp[i] = col;
6610880e062SHong Zhang       bap   = ap +  bs2*i;
6620880e062SHong Zhang       if (roworiented) {
6630880e062SHong Zhang         for (ii=0; ii<bs; ii++,value+=stepval) {
6640880e062SHong Zhang           for (jj=ii; jj<bs2; jj+=bs) {
6650880e062SHong Zhang             bap[jj] = *value++;
6660880e062SHong Zhang           }
6670880e062SHong Zhang         }
6680880e062SHong Zhang       } else {
6690880e062SHong Zhang         for (ii=0; ii<bs; ii++,value+=stepval) {
6700880e062SHong Zhang           for (jj=0; jj<bs; jj++) {
6710880e062SHong Zhang             *bap++  = *value++;
6720880e062SHong Zhang           }
6730880e062SHong Zhang         }
6740880e062SHong Zhang        }
6750880e062SHong Zhang     noinsert2:;
6760880e062SHong Zhang       low = i;
6770880e062SHong Zhang     }
6780880e062SHong Zhang     ailen[row] = nrow;
6790880e062SHong Zhang   }
6800880e062SHong Zhang    PetscFunctionReturn(0);
68149b5e25fSSatish Balay }
68249b5e25fSSatish Balay 
6834a2ae208SSatish Balay #undef __FUNCT__
6844a2ae208SSatish Balay #define __FUNCT__ "MatAssemblyEnd_SeqSBAIJ"
685dfbe8321SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ(Mat A,MatAssemblyType mode)
68649b5e25fSSatish Balay {
68749b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
6886849ba73SBarry Smith   PetscErrorCode ierr;
68913f74950SBarry Smith   PetscInt       fshift = 0,i,j,*ai = a->i,*aj = a->j,*imax = a->imax;
690899cda47SBarry Smith   PetscInt       m = A->rmap.N,*ip,N,*ailen = a->ilen;
69113f74950SBarry Smith   PetscInt       mbs = a->mbs,bs2 = a->bs2,rmax = 0;
69249b5e25fSSatish Balay   MatScalar      *aa = a->a,*ap;
69349b5e25fSSatish Balay 
69449b5e25fSSatish Balay   PetscFunctionBegin;
69549b5e25fSSatish Balay   if (mode == MAT_FLUSH_ASSEMBLY) PetscFunctionReturn(0);
69649b5e25fSSatish Balay 
69749b5e25fSSatish Balay   if (m) rmax = ailen[0];
69849b5e25fSSatish Balay   for (i=1; i<mbs; i++) {
69949b5e25fSSatish Balay     /* move each row back by the amount of empty slots (fshift) before it*/
70049b5e25fSSatish Balay     fshift += imax[i-1] - ailen[i-1];
70149b5e25fSSatish Balay      rmax   = PetscMax(rmax,ailen[i]);
70249b5e25fSSatish Balay      if (fshift) {
70349b5e25fSSatish Balay        ip = aj + ai[i]; ap = aa + bs2*ai[i];
70449b5e25fSSatish Balay        N = ailen[i];
70549b5e25fSSatish Balay        for (j=0; j<N; j++) {
70649b5e25fSSatish Balay          ip[j-fshift] = ip[j];
70749b5e25fSSatish Balay          ierr = PetscMemcpy(ap+(j-fshift)*bs2,ap+j*bs2,bs2*sizeof(MatScalar));CHKERRQ(ierr);
70849b5e25fSSatish Balay        }
70949b5e25fSSatish Balay      }
71049b5e25fSSatish Balay      ai[i] = ai[i-1] + ailen[i-1];
71149b5e25fSSatish Balay   }
71249b5e25fSSatish Balay   if (mbs) {
71349b5e25fSSatish Balay     fshift += imax[mbs-1] - ailen[mbs-1];
71449b5e25fSSatish Balay      ai[mbs] = ai[mbs-1] + ailen[mbs-1];
71549b5e25fSSatish Balay   }
71649b5e25fSSatish Balay   /* reset ilen and imax for each row */
71749b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
71849b5e25fSSatish Balay     ailen[i] = imax[i] = ai[i+1] - ai[i];
71949b5e25fSSatish Balay   }
7206c6c5352SBarry Smith   a->nz = ai[mbs];
72149b5e25fSSatish Balay 
722b424e231SHong Zhang   /* diagonals may have moved, reset it */
723b424e231SHong Zhang   if (a->diag) {
72413f74950SBarry Smith     ierr = PetscMemcpy(a->diag,ai,(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr);
72549b5e25fSSatish Balay   }
726899cda47SBarry Smith   ierr = PetscInfo5(A,"Matrix size: %D X %D, block size %D; storage space: %D unneeded, %D used\n",m,A->rmap.N,A->rmap.bs,fshift*bs2,a->nz*bs2);CHKERRQ(ierr);
727ae15b995SBarry Smith   ierr = PetscInfo1(A,"Number of mallocs during MatSetValues is %D\n",a->reallocs);CHKERRQ(ierr);
728ae15b995SBarry Smith   ierr = PetscInfo1(A,"Most nonzeros blocks in any row is %D\n",rmax);CHKERRQ(ierr);
72949b5e25fSSatish Balay   a->reallocs          = 0;
73049b5e25fSSatish Balay   A->info.nz_unneeded  = (PetscReal)fshift*bs2;
73149b5e25fSSatish Balay   PetscFunctionReturn(0);
73249b5e25fSSatish Balay }
73349b5e25fSSatish Balay 
73449b5e25fSSatish Balay /*
73549b5e25fSSatish Balay    This function returns an array of flags which indicate the locations of contiguous
73649b5e25fSSatish Balay    blocks that should be zeroed. for eg: if bs = 3  and is = [0,1,2,3,5,6,7,8,9]
73749b5e25fSSatish Balay    then the resulting sizes = [3,1,1,3,1] correspondig to sets [(0,1,2),(3),(5),(6,7,8),(9)]
73849b5e25fSSatish Balay    Assume: sizes should be long enough to hold all the values.
73949b5e25fSSatish Balay */
7404a2ae208SSatish Balay #undef __FUNCT__
7414a2ae208SSatish Balay #define __FUNCT__ "MatZeroRows_SeqSBAIJ_Check_Blocks"
74213f74950SBarry Smith PetscErrorCode MatZeroRows_SeqSBAIJ_Check_Blocks(PetscInt idx[],PetscInt n,PetscInt bs,PetscInt sizes[], PetscInt *bs_max)
74349b5e25fSSatish Balay {
74413f74950SBarry Smith   PetscInt   i,j,k,row;
74549b5e25fSSatish Balay   PetscTruth flg;
74649b5e25fSSatish Balay 
74749b5e25fSSatish Balay   PetscFunctionBegin;
74849b5e25fSSatish Balay    for (i=0,j=0; i<n; j++) {
74949b5e25fSSatish Balay      row = idx[i];
75049b5e25fSSatish Balay      if (row%bs!=0) { /* Not the begining of a block */
75149b5e25fSSatish Balay        sizes[j] = 1;
75249b5e25fSSatish Balay        i++;
75349b5e25fSSatish Balay      } else if (i+bs > n) { /* Beginning of a block, but complete block doesn't exist (at idx end) */
75449b5e25fSSatish Balay        sizes[j] = 1;         /* Also makes sure atleast 'bs' values exist for next else */
75549b5e25fSSatish Balay        i++;
75649b5e25fSSatish Balay      } else { /* Begining of the block, so check if the complete block exists */
75749b5e25fSSatish Balay        flg = PETSC_TRUE;
75849b5e25fSSatish Balay        for (k=1; k<bs; k++) {
75949b5e25fSSatish Balay          if (row+k != idx[i+k]) { /* break in the block */
76049b5e25fSSatish Balay            flg = PETSC_FALSE;
76149b5e25fSSatish Balay            break;
76249b5e25fSSatish Balay          }
76349b5e25fSSatish Balay        }
764abc0a331SBarry Smith        if (flg) { /* No break in the bs */
76549b5e25fSSatish Balay          sizes[j] = bs;
76649b5e25fSSatish Balay          i+= bs;
76749b5e25fSSatish Balay        } else {
76849b5e25fSSatish Balay          sizes[j] = 1;
76949b5e25fSSatish Balay          i++;
77049b5e25fSSatish Balay        }
77149b5e25fSSatish Balay      }
77249b5e25fSSatish Balay    }
77349b5e25fSSatish Balay    *bs_max = j;
77449b5e25fSSatish Balay    PetscFunctionReturn(0);
77549b5e25fSSatish Balay }
77649b5e25fSSatish Balay 
77749b5e25fSSatish Balay 
77849b5e25fSSatish Balay /* Only add/insert a(i,j) with i<=j (blocks).
77949b5e25fSSatish Balay    Any a(i,j) with i>j input by user is ingored.
78049b5e25fSSatish Balay */
78149b5e25fSSatish Balay 
7824a2ae208SSatish Balay #undef __FUNCT__
7834a2ae208SSatish Balay #define __FUNCT__ "MatSetValues_SeqSBAIJ"
78413f74950SBarry Smith PetscErrorCode MatSetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is)
78549b5e25fSSatish Balay {
78649b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
7876849ba73SBarry Smith   PetscErrorCode ierr;
788e2ee6c50SBarry Smith   PetscInt       *rp,k,low,high,t,ii,row,nrow,i,col,l,rmax,N,lastcol = -1;
78913f74950SBarry Smith   PetscInt       *imax=a->imax,*ai=a->i,*ailen=a->ilen,roworiented=a->roworiented;
790899cda47SBarry Smith   PetscInt       *aj=a->j,nonew=a->nonew,bs=A->rmap.bs,brow,bcol;
79113f74950SBarry Smith   PetscInt       ridx,cidx,bs2=a->bs2;
79249b5e25fSSatish Balay   MatScalar      *ap,value,*aa=a->a,*bap;
79349b5e25fSSatish Balay 
79449b5e25fSSatish Balay   PetscFunctionBegin;
79549b5e25fSSatish Balay   for (k=0; k<m; k++) { /* loop over added rows */
79649b5e25fSSatish Balay     row  = im[k];       /* row number */
79749b5e25fSSatish Balay     brow = row/bs;      /* block row number */
79849b5e25fSSatish Balay     if (row < 0) continue;
7992515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
800899cda47SBarry Smith     if (row >= A->rmap.N) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,A->rmap.N-1);
80149b5e25fSSatish Balay #endif
80249b5e25fSSatish Balay     rp   = aj + ai[brow]; /*ptr to beginning of column value of the row block*/
80349b5e25fSSatish Balay     ap   = aa + bs2*ai[brow]; /*ptr to beginning of element value of the row block*/
80449b5e25fSSatish Balay     rmax = imax[brow];  /* maximum space allocated for this row */
80549b5e25fSSatish Balay     nrow = ailen[brow]; /* actual length of this row */
80649b5e25fSSatish Balay     low  = 0;
80749b5e25fSSatish Balay 
80849b5e25fSSatish Balay     for (l=0; l<n; l++) { /* loop over added columns */
80949b5e25fSSatish Balay       if (in[l] < 0) continue;
8102515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
811899cda47SBarry Smith       if (in[l] >= A->rmap.N) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",in[l],A->rmap.N-1);
81249b5e25fSSatish Balay #endif
81349b5e25fSSatish Balay       col = in[l];
81449b5e25fSSatish Balay       bcol = col/bs;              /* block col number */
81549b5e25fSSatish Balay 
816941593c8SHong Zhang       if (brow > bcol) {
817941593c8SHong Zhang         if (a->ignore_ltriangular){
818941593c8SHong Zhang           continue; /* ignore lower triangular values */
819941593c8SHong Zhang         } else {
820941593c8SHong Zhang           SETERRQ(PETSC_ERR_USER,"Lower triangular value cannot be set for sbaij format. Ignoring these values, run with -mat_ignore_lower_triangular or call MatSetOption(mat,MAT_IGNORE_LOWER_TRIANGULAR)");
821941593c8SHong Zhang         }
822941593c8SHong Zhang       }
823f4989cb3SHong Zhang 
82449b5e25fSSatish Balay       ridx = row % bs; cidx = col % bs; /*row and col index inside the block */
8258549e402SHong Zhang       if ((brow==bcol && ridx<=cidx) || (brow<bcol)){
82649b5e25fSSatish Balay         /* element value a(k,l) */
82749b5e25fSSatish Balay         if (roworiented) {
82849b5e25fSSatish Balay           value = v[l + k*n];
82949b5e25fSSatish Balay         } else {
83049b5e25fSSatish Balay           value = v[k + l*m];
83149b5e25fSSatish Balay         }
83249b5e25fSSatish Balay 
83349b5e25fSSatish Balay         /* move pointer bap to a(k,l) quickly and add/insert value */
8347cd84e04SBarry Smith         if (col <= lastcol) low = 0; high = nrow;
835e2ee6c50SBarry Smith         lastcol = col;
83649b5e25fSSatish Balay         while (high-low > 7) {
83749b5e25fSSatish Balay           t = (low+high)/2;
83849b5e25fSSatish Balay           if (rp[t] > bcol) high = t;
83949b5e25fSSatish Balay           else              low  = t;
84049b5e25fSSatish Balay         }
84149b5e25fSSatish Balay         for (i=low; i<high; i++) {
84249b5e25fSSatish Balay           if (rp[i] > bcol) break;
84349b5e25fSSatish Balay           if (rp[i] == bcol) {
84449b5e25fSSatish Balay             bap  = ap +  bs2*i + bs*cidx + ridx;
84549b5e25fSSatish Balay             if (is == ADD_VALUES) *bap += value;
84649b5e25fSSatish Balay             else                  *bap  = value;
8478549e402SHong Zhang             /* for diag block, add/insert its symmetric element a(cidx,ridx) */
8488549e402SHong Zhang             if (brow == bcol && ridx < cidx){
8498549e402SHong Zhang               bap  = ap +  bs2*i + bs*ridx + cidx;
8508549e402SHong Zhang               if (is == ADD_VALUES) *bap += value;
8518549e402SHong Zhang               else                  *bap  = value;
8528549e402SHong Zhang             }
85349b5e25fSSatish Balay             goto noinsert1;
85449b5e25fSSatish Balay           }
85549b5e25fSSatish Balay         }
85649b5e25fSSatish Balay 
85749b5e25fSSatish Balay         if (nonew == 1) goto noinsert1;
858085a36d4SBarry Smith         if (nonew == -1) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col);
859421e10b8SBarry Smith         MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,brow,bcol,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar);
86049b5e25fSSatish Balay 
861c03d1d03SSatish Balay         N = nrow++ - 1; high++;
86249b5e25fSSatish Balay         /* shift up all the later entries in this row */
86349b5e25fSSatish Balay         for (ii=N; ii>=i; ii--) {
86449b5e25fSSatish Balay           rp[ii+1] = rp[ii];
86549b5e25fSSatish Balay           ierr     = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr);
86649b5e25fSSatish Balay         }
86749b5e25fSSatish Balay         if (N>=i) {
86849b5e25fSSatish Balay           ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr);
86949b5e25fSSatish Balay         }
87049b5e25fSSatish Balay         rp[i]                      = bcol;
87149b5e25fSSatish Balay         ap[bs2*i + bs*cidx + ridx] = value;
87249b5e25fSSatish Balay       noinsert1:;
87349b5e25fSSatish Balay         low = i;
8748549e402SHong Zhang       }
87549b5e25fSSatish Balay     }   /* end of loop over added columns */
87649b5e25fSSatish Balay     ailen[brow] = nrow;
87749b5e25fSSatish Balay   }   /* end of loop over added rows */
87849b5e25fSSatish Balay   PetscFunctionReturn(0);
87949b5e25fSSatish Balay }
88049b5e25fSSatish Balay 
8816849ba73SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorSymbolic_SeqSBAIJ(Mat,IS,MatFactorInfo*,Mat*);
8826849ba73SBarry Smith EXTERN PetscErrorCode MatCholeskyFactor_SeqSBAIJ(Mat,IS,MatFactorInfo*);
88313f74950SBarry Smith EXTERN PetscErrorCode MatIncreaseOverlap_SeqSBAIJ(Mat,PetscInt,IS[],PetscInt);
88413f74950SBarry Smith EXTERN PetscErrorCode MatGetSubMatrix_SeqSBAIJ(Mat,IS,IS,PetscInt,MatReuse,Mat*);
88513f74950SBarry Smith EXTERN PetscErrorCode MatGetSubMatrices_SeqSBAIJ(Mat,PetscInt,const IS[],const IS[],MatReuse,Mat*[]);
886f4df32b1SMatthew Knepley EXTERN PetscErrorCode MatScale_SeqSBAIJ(Mat,PetscScalar);
8876849ba73SBarry Smith EXTERN PetscErrorCode MatNorm_SeqSBAIJ(Mat,NormType,PetscReal *);
8886849ba73SBarry Smith EXTERN PetscErrorCode MatEqual_SeqSBAIJ(Mat,Mat,PetscTruth*);
8896849ba73SBarry Smith EXTERN PetscErrorCode MatGetDiagonal_SeqSBAIJ(Mat,Vec);
8906849ba73SBarry Smith EXTERN PetscErrorCode MatDiagonalScale_SeqSBAIJ(Mat,Vec,Vec);
8916849ba73SBarry Smith EXTERN PetscErrorCode MatGetInfo_SeqSBAIJ(Mat,MatInfoType,MatInfo *);
8926849ba73SBarry Smith EXTERN PetscErrorCode MatZeroEntries_SeqSBAIJ(Mat);
8936849ba73SBarry Smith EXTERN PetscErrorCode MatGetRowMax_SeqSBAIJ(Mat,Vec);
89449b5e25fSSatish Balay 
8956849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_N(Mat,Vec,Vec);
8966849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_1(Mat,Vec,Vec);
8976849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_2(Mat,Vec,Vec);
8986849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_3(Mat,Vec,Vec);
8996849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_4(Mat,Vec,Vec);
9006849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_5(Mat,Vec,Vec);
9016849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_6(Mat,Vec,Vec);
9026849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_7(Mat,Vec,Vec);
90349b5e25fSSatish Balay 
9046849ba73SBarry Smith EXTERN PetscErrorCode MatSolves_SeqSBAIJ_1(Mat,Vecs,Vecs);
905d59c15a7SBarry Smith 
9066849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_1_NaturalOrdering(Mat,Vec,Vec);
9076849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_2_NaturalOrdering(Mat,Vec,Vec);
9086849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_3_NaturalOrdering(Mat,Vec,Vec);
9096849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_4_NaturalOrdering(Mat,Vec,Vec);
9106849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_5_NaturalOrdering(Mat,Vec,Vec);
9116849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_6_NaturalOrdering(Mat,Vec,Vec);
9126849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_7_NaturalOrdering(Mat,Vec,Vec);
9136849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_N_NaturalOrdering(Mat,Vec,Vec);
91407f98182SSatish Balay 
915af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_N(Mat,MatFactorInfo*,Mat*);
916af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_1(Mat,MatFactorInfo*,Mat*);
917af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_2(Mat,MatFactorInfo*,Mat*);
918af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_3(Mat,MatFactorInfo*,Mat*);
919af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_4(Mat,MatFactorInfo*,Mat*);
920af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_5(Mat,MatFactorInfo*,Mat*);
921af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_6(Mat,MatFactorInfo*,Mat*);
922af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_7(Mat,MatFactorInfo*,Mat*);
92313f74950SBarry Smith EXTERN PetscErrorCode MatGetInertia_SeqSBAIJ(Mat,PetscInt*,PetscInt*,PetscInt*);
92449b5e25fSSatish Balay 
9256849ba73SBarry Smith EXTERN PetscErrorCode MatMult_SeqSBAIJ_1(Mat,Vec,Vec);
9266849ba73SBarry Smith EXTERN PetscErrorCode MatMult_SeqSBAIJ_2(Mat,Vec,Vec);
9276849ba73SBarry Smith EXTERN PetscErrorCode MatMult_SeqSBAIJ_3(Mat,Vec,Vec);
9286849ba73SBarry Smith EXTERN PetscErrorCode MatMult_SeqSBAIJ_4(Mat,Vec,Vec);
9296849ba73SBarry Smith EXTERN PetscErrorCode MatMult_SeqSBAIJ_5(Mat,Vec,Vec);
9306849ba73SBarry Smith EXTERN PetscErrorCode MatMult_SeqSBAIJ_6(Mat,Vec,Vec);
9316849ba73SBarry Smith EXTERN PetscErrorCode MatMult_SeqSBAIJ_7(Mat,Vec,Vec);
9326849ba73SBarry Smith EXTERN PetscErrorCode MatMult_SeqSBAIJ_N(Mat,Vec,Vec);
93349b5e25fSSatish Balay 
9346849ba73SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_1(Mat,Vec,Vec,Vec);
9356849ba73SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_2(Mat,Vec,Vec,Vec);
9366849ba73SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_3(Mat,Vec,Vec,Vec);
9376849ba73SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_4(Mat,Vec,Vec,Vec);
9386849ba73SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_5(Mat,Vec,Vec,Vec);
9396849ba73SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_6(Mat,Vec,Vec,Vec);
9406849ba73SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_7(Mat,Vec,Vec,Vec);
9416849ba73SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_N(Mat,Vec,Vec,Vec);
94249b5e25fSSatish Balay 
9434a2ae208SSatish Balay #undef __FUNCT__
9444d101231SSatish Balay #define __FUNCT__ "MatICCFactor_SeqSBAIJ"
945dfbe8321SBarry Smith PetscErrorCode MatICCFactor_SeqSBAIJ(Mat inA,IS row,MatFactorInfo *info)
94649b5e25fSSatish Balay {
9474ccecd49SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)inA->data;
94849b5e25fSSatish Balay   Mat            outA;
949dfbe8321SBarry Smith   PetscErrorCode ierr;
950c84f5b01SHong Zhang   PetscTruth     row_identity;
95149b5e25fSSatish Balay 
95249b5e25fSSatish Balay   PetscFunctionBegin;
953c84f5b01SHong Zhang   if (info->levels != 0) SETERRQ(PETSC_ERR_SUP,"Only levels=0 is supported for in-place icc");
954c84f5b01SHong Zhang   ierr = ISIdentity(row,&row_identity);CHKERRQ(ierr);
955c84f5b01SHong Zhang   if (!row_identity) SETERRQ(PETSC_ERR_SUP,"Matrix reordering is not supported");
956c84f5b01SHong Zhang   if (inA->rmap.bs != 1) SETERRQ1(PETSC_ERR_SUP,"Matrix block size %D is not supported",inA->rmap.bs); /* Need to replace MatCholeskyFactorSymbolic_SeqSBAIJ_MSR()! */
957c84f5b01SHong Zhang 
95849b5e25fSSatish Balay   outA        = inA;
9591260e1f8SHong Zhang   inA->factor = FACTOR_CHOLESKY;
96049b5e25fSSatish Balay 
9611a3463dfSHong Zhang   ierr = MatMarkDiagonal_SeqSBAIJ(inA);CHKERRQ(ierr);
96249b5e25fSSatish Balay   /*
963c84f5b01SHong Zhang     Blocksize < 8 have a special faster factorization/solver
964c84f5b01SHong Zhang     for ICC(0) factorization with natural ordering
96549b5e25fSSatish Balay   */
966c84f5b01SHong Zhang   switch (inA->rmap.bs){ /* Note: row_identity = PETSC_TRUE! */
96749b5e25fSSatish Balay   case 1:
968c84f5b01SHong Zhang     inA->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering;
9690fe9e5beSBarry Smith     inA->ops->solve            = MatSolve_SeqSBAIJ_1_NaturalOrdering;
97007f98182SSatish Balay     inA->ops->solvetranspose   = MatSolve_SeqSBAIJ_1_NaturalOrdering;
971d59c15a7SBarry Smith     inA->ops->solves           = MatSolves_SeqSBAIJ_1;
972*759322afSHong Zhang     inA->ops->forwardsolve     = MatForwardSolve_SeqSBAIJ_1_NaturalOrdering;
973*759322afSHong Zhang     inA->ops->backwardsolve    = MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering;
974c84f5b01SHong Zhang     ierr = PetscInfo(inA,"Using special in-place natural ordering solvetrans BS=1\n");CHKERRQ(ierr);
975c84f5b01SHong Zhang     break;
97649b5e25fSSatish Balay   case 2:
977c84f5b01SHong Zhang     inA->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering;
9781a3463dfSHong Zhang     inA->ops->solve           = MatSolve_SeqSBAIJ_2_NaturalOrdering;
9791a3463dfSHong Zhang     inA->ops->solvetranspose  = MatSolve_SeqSBAIJ_2_NaturalOrdering;
980*759322afSHong Zhang     inA->ops->forwardsolve     = MatForwardSolve_SeqSBAIJ_2_NaturalOrdering;
981*759322afSHong Zhang     inA->ops->backwardsolve    = MatBackwardSolve_SeqSBAIJ_2_NaturalOrdering;
982ae15b995SBarry Smith     ierr = PetscInfo(inA,"Using special in-place natural ordering factor and solve BS=2\n");CHKERRQ(ierr);
98349b5e25fSSatish Balay     break;
98449b5e25fSSatish Balay   case 3:
985c84f5b01SHong Zhang      inA->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering;
9861a3463dfSHong Zhang      inA->ops->solve           = MatSolve_SeqSBAIJ_3_NaturalOrdering;
9871a3463dfSHong Zhang      inA->ops->solvetranspose  = MatSolve_SeqSBAIJ_3_NaturalOrdering;
988*759322afSHong Zhang      inA->ops->forwardsolve    = MatForwardSolve_SeqSBAIJ_3_NaturalOrdering;
989*759322afSHong Zhang      inA->ops->backwardsolve   = MatBackwardSolve_SeqSBAIJ_3_NaturalOrdering;
990ae15b995SBarry Smith      ierr = PetscInfo(inA,"Using special in-place natural ordering factor and solve BS=3\n");CHKERRQ(ierr);
99149b5e25fSSatish Balay      break;
99249b5e25fSSatish Balay   case 4:
993c84f5b01SHong Zhang     inA->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering;
9941a3463dfSHong Zhang     inA->ops->solve           = MatSolve_SeqSBAIJ_4_NaturalOrdering;
9951a3463dfSHong Zhang     inA->ops->solvetranspose  = MatSolve_SeqSBAIJ_4_NaturalOrdering;
996*759322afSHong Zhang     inA->ops->forwardsolve    = MatForwardSolve_SeqSBAIJ_4_NaturalOrdering;
997*759322afSHong Zhang     inA->ops->backwardsolve   = MatBackwardSolve_SeqSBAIJ_4_NaturalOrdering;
998ae15b995SBarry Smith     ierr = PetscInfo(inA,"Using special in-place natural ordering factor and solve BS=4\n");CHKERRQ(ierr);
99949b5e25fSSatish Balay     break;
100049b5e25fSSatish Balay   case 5:
1001c84f5b01SHong Zhang     inA->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering;
10021a3463dfSHong Zhang     inA->ops->solve           = MatSolve_SeqSBAIJ_5_NaturalOrdering;
10031a3463dfSHong Zhang     inA->ops->solvetranspose  = MatSolve_SeqSBAIJ_5_NaturalOrdering;
1004*759322afSHong Zhang     inA->ops->forwardsolve    = MatForwardSolve_SeqSBAIJ_5_NaturalOrdering;
1005*759322afSHong Zhang     inA->ops->backwardsolve   = MatBackwardSolve_SeqSBAIJ_5_NaturalOrdering;
1006ae15b995SBarry Smith     ierr = PetscInfo(inA,"Using special in-place natural ordering factor and solve BS=5\n");CHKERRQ(ierr);
100749b5e25fSSatish Balay     break;
100849b5e25fSSatish Balay   case 6:
1009c84f5b01SHong Zhang     inA->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering;
10101a3463dfSHong Zhang     inA->ops->solve           = MatSolve_SeqSBAIJ_6_NaturalOrdering;
10111a3463dfSHong Zhang     inA->ops->solvetranspose  = MatSolve_SeqSBAIJ_6_NaturalOrdering;
1012*759322afSHong Zhang     inA->ops->forwardsolve    = MatForwardSolve_SeqSBAIJ_6_NaturalOrdering;
1013*759322afSHong Zhang     inA->ops->backwardsolve   = MatBackwardSolve_SeqSBAIJ_6_NaturalOrdering;
1014ae15b995SBarry Smith     ierr = PetscInfo(inA,"Using special in-place natural ordering factor and solve BS=6\n");CHKERRQ(ierr);
101549b5e25fSSatish Balay     break;
101649b5e25fSSatish Balay   case 7:
1017c84f5b01SHong Zhang     inA->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering;
10181a3463dfSHong Zhang     inA->ops->solvetranspose  = MatSolve_SeqSBAIJ_7_NaturalOrdering;
10191a3463dfSHong Zhang     inA->ops->solve           = MatSolve_SeqSBAIJ_7_NaturalOrdering;
1020*759322afSHong Zhang     inA->ops->forwardsolve    = MatForwardSolve_SeqSBAIJ_7_NaturalOrdering;
1021*759322afSHong Zhang     inA->ops->backwardsolve   = MatBackwardSolve_SeqSBAIJ_7_NaturalOrdering;
1022ae15b995SBarry Smith     ierr = PetscInfo(inA,"Using special in-place natural ordering factor and solve BS=7\n");CHKERRQ(ierr);
102349b5e25fSSatish Balay     break;
102449b5e25fSSatish Balay   default:
1025c84f5b01SHong Zhang     inA->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering;
1026c84f5b01SHong Zhang     inA->ops->solvetranspose  = MatSolve_SeqSBAIJ_N_NaturalOrdering;
1027c84f5b01SHong Zhang     inA->ops->solve           = MatSolve_SeqSBAIJ_N_NaturalOrdering;
1028*759322afSHong Zhang     inA->ops->forwardsolve    = MatForwardSolve_SeqSBAIJ_N_NaturalOrdering;
1029*759322afSHong Zhang     inA->ops->backwardsolve   = MatBackwardSolve_SeqSBAIJ_N_NaturalOrdering;
1030c84f5b01SHong Zhang     break;
1031c84f5b01SHong Zhang   }
103249b5e25fSSatish Balay 
1033c84f5b01SHong Zhang   a->row = row;
1034c84f5b01SHong Zhang   a->col = row;
1035c84f5b01SHong Zhang   ierr   = PetscObjectReference((PetscObject)row);CHKERRQ(ierr);
1036c84f5b01SHong Zhang   ierr   = PetscObjectReference((PetscObject)row);CHKERRQ(ierr);
1037c84f5b01SHong Zhang 
1038c84f5b01SHong Zhang   /* Create the invert permutation so that it can be used in MatCholeskyFactorNumeric() */
1039c84f5b01SHong Zhang   if (a->icol) {ierr = ISInvertPermutation(row,PETSC_DECIDE, &a->icol);CHKERRQ(ierr);}
104052e6d16bSBarry Smith   ierr = PetscLogObjectParent(inA,a->icol);CHKERRQ(ierr);
104149b5e25fSSatish Balay 
104249b5e25fSSatish Balay   if (!a->solve_work) {
1043c84f5b01SHong Zhang     ierr = PetscMalloc((inA->rmap.N+inA->rmap.bs)*sizeof(PetscScalar),&a->solve_work);CHKERRQ(ierr);
1044c84f5b01SHong Zhang     ierr = PetscLogObjectMemory(inA,(inA->rmap.N+inA->rmap.bs)*sizeof(PetscScalar));CHKERRQ(ierr);
104549b5e25fSSatish Balay   }
104649b5e25fSSatish Balay 
1047af281ebdSHong Zhang   ierr = MatCholeskyFactorNumeric(inA,info,&outA);CHKERRQ(ierr);
104849b5e25fSSatish Balay   PetscFunctionReturn(0);
104949b5e25fSSatish Balay }
1050950f1e5bSHong Zhang 
105149b5e25fSSatish Balay EXTERN_C_BEGIN
10524a2ae208SSatish Balay #undef __FUNCT__
10534a2ae208SSatish Balay #define __FUNCT__ "MatSeqSBAIJSetColumnIndices_SeqSBAIJ"
1054be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatSeqSBAIJSetColumnIndices_SeqSBAIJ(Mat mat,PetscInt *indices)
105549b5e25fSSatish Balay {
1056045c9aa0SHong Zhang   Mat_SeqSBAIJ *baij = (Mat_SeqSBAIJ *)mat->data;
105713f74950SBarry Smith   PetscInt     i,nz,n;
105849b5e25fSSatish Balay 
105949b5e25fSSatish Balay   PetscFunctionBegin;
10606c6c5352SBarry Smith   nz = baij->maxnz;
1061899cda47SBarry Smith   n  = mat->cmap.n;
106249b5e25fSSatish Balay   for (i=0; i<nz; i++) {
106349b5e25fSSatish Balay     baij->j[i] = indices[i];
106449b5e25fSSatish Balay   }
10656c6c5352SBarry Smith    baij->nz = nz;
106649b5e25fSSatish Balay    for (i=0; i<n; i++) {
106749b5e25fSSatish Balay      baij->ilen[i] = baij->imax[i];
106849b5e25fSSatish Balay    }
106949b5e25fSSatish Balay    PetscFunctionReturn(0);
107049b5e25fSSatish Balay }
107149b5e25fSSatish Balay EXTERN_C_END
107249b5e25fSSatish Balay 
10734a2ae208SSatish Balay #undef __FUNCT__
10744a2ae208SSatish Balay #define __FUNCT__ "MatSeqSBAIJSetColumnIndices"
107549b5e25fSSatish Balay /*@
107619585528SSatish Balay   MatSeqSBAIJSetColumnIndices - Set the column indices for all the rows
107749b5e25fSSatish Balay   in the matrix.
107849b5e25fSSatish Balay 
107949b5e25fSSatish Balay   Input Parameters:
108019585528SSatish Balay   +  mat     - the SeqSBAIJ matrix
108149b5e25fSSatish Balay   -  indices - the column indices
108249b5e25fSSatish Balay 
108349b5e25fSSatish Balay   Level: advanced
108449b5e25fSSatish Balay 
108549b5e25fSSatish Balay   Notes:
108649b5e25fSSatish Balay   This can be called if you have precomputed the nonzero structure of the
108749b5e25fSSatish Balay   matrix and want to provide it to the matrix object to improve the performance
108849b5e25fSSatish Balay   of the MatSetValues() operation.
108949b5e25fSSatish Balay 
109049b5e25fSSatish Balay   You MUST have set the correct numbers of nonzeros per row in the call to
1091d1be2dadSMatthew Knepley   MatCreateSeqSBAIJ(), and the columns indices MUST be sorted.
109249b5e25fSSatish Balay 
1093ab9f2c04SSatish Balay   MUST be called before any calls to MatSetValues()
109449b5e25fSSatish Balay 
1095ab9f2c04SSatish Balay   .seealso: MatCreateSeqSBAIJ
109649b5e25fSSatish Balay @*/
1097be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatSeqSBAIJSetColumnIndices(Mat mat,PetscInt *indices)
109849b5e25fSSatish Balay {
109913f74950SBarry Smith   PetscErrorCode ierr,(*f)(Mat,PetscInt *);
110049b5e25fSSatish Balay 
110149b5e25fSSatish Balay   PetscFunctionBegin;
11024482741eSBarry Smith   PetscValidHeaderSpecific(mat,MAT_COOKIE,1);
11034482741eSBarry Smith   PetscValidPointer(indices,2);
1104c134de8dSSatish Balay   ierr = PetscObjectQueryFunction((PetscObject)mat,"MatSeqSBAIJSetColumnIndices_C",(void (**)(void))&f);CHKERRQ(ierr);
110549b5e25fSSatish Balay   if (f) {
110649b5e25fSSatish Balay     ierr = (*f)(mat,indices);CHKERRQ(ierr);
110749b5e25fSSatish Balay   } else {
1108e005ede5SBarry Smith     SETERRQ(PETSC_ERR_SUP,"Wrong type of matrix to set column indices");
110949b5e25fSSatish Balay   }
111049b5e25fSSatish Balay   PetscFunctionReturn(0);
111149b5e25fSSatish Balay }
111249b5e25fSSatish Balay 
11134a2ae208SSatish Balay #undef __FUNCT__
11143c896bc6SHong Zhang #define __FUNCT__ "MatCopy_SeqSBAIJ"
11153c896bc6SHong Zhang PetscErrorCode MatCopy_SeqSBAIJ(Mat A,Mat B,MatStructure str)
11163c896bc6SHong Zhang {
11173c896bc6SHong Zhang   PetscErrorCode ierr;
11183c896bc6SHong Zhang 
11193c896bc6SHong Zhang   PetscFunctionBegin;
11203c896bc6SHong Zhang   /* If the two matrices have the same copy implementation, use fast copy. */
11213c896bc6SHong Zhang   if (str == SAME_NONZERO_PATTERN && (A->ops->copy == B->ops->copy)) {
11223c896bc6SHong Zhang     Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
11233c896bc6SHong Zhang     Mat_SeqSBAIJ *b = (Mat_SeqSBAIJ*)B->data;
11243c896bc6SHong Zhang 
1125899cda47SBarry Smith     if (a->i[A->rmap.N] != b->i[B->rmap.N]) {
11263c896bc6SHong Zhang       SETERRQ(PETSC_ERR_ARG_INCOMP,"Number of nonzeros in two matrices are different");
11273c896bc6SHong Zhang     }
1128899cda47SBarry Smith     ierr = PetscMemcpy(b->a,a->a,(a->i[A->rmap.N])*sizeof(PetscScalar));CHKERRQ(ierr);
11293c896bc6SHong Zhang   } else {
1130f5edf698SHong Zhang     ierr = MatGetRowUpperTriangular(A);CHKERRQ(ierr);
11313c896bc6SHong Zhang     ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr);
1132f5edf698SHong Zhang     ierr = MatRestoreRowUpperTriangular(A);CHKERRQ(ierr);
11333c896bc6SHong Zhang   }
11343c896bc6SHong Zhang   PetscFunctionReturn(0);
11353c896bc6SHong Zhang }
11363c896bc6SHong Zhang 
11373c896bc6SHong Zhang #undef __FUNCT__
11384a2ae208SSatish Balay #define __FUNCT__ "MatSetUpPreallocation_SeqSBAIJ"
1139dfbe8321SBarry Smith PetscErrorCode MatSetUpPreallocation_SeqSBAIJ(Mat A)
1140273d9f13SBarry Smith {
1141dfbe8321SBarry Smith   PetscErrorCode ierr;
1142273d9f13SBarry Smith 
1143273d9f13SBarry Smith   PetscFunctionBegin;
11447edd0491SSatish Balay   ierr =  MatSeqSBAIJSetPreallocation_SeqSBAIJ(A,PetscMax(A->rmap.bs,1),PETSC_DEFAULT,0);CHKERRQ(ierr);
1145273d9f13SBarry Smith   PetscFunctionReturn(0);
1146273d9f13SBarry Smith }
1147273d9f13SBarry Smith 
1148a6ece127SHong Zhang #undef __FUNCT__
1149a6ece127SHong Zhang #define __FUNCT__ "MatGetArray_SeqSBAIJ"
1150dfbe8321SBarry Smith PetscErrorCode MatGetArray_SeqSBAIJ(Mat A,PetscScalar *array[])
1151a6ece127SHong Zhang {
1152a6ece127SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
1153a6ece127SHong Zhang   PetscFunctionBegin;
1154a6ece127SHong Zhang   *array = a->a;
1155a6ece127SHong Zhang   PetscFunctionReturn(0);
1156a6ece127SHong Zhang }
1157a6ece127SHong Zhang 
1158a6ece127SHong Zhang #undef __FUNCT__
1159a6ece127SHong Zhang #define __FUNCT__ "MatRestoreArray_SeqSBAIJ"
1160dfbe8321SBarry Smith PetscErrorCode MatRestoreArray_SeqSBAIJ(Mat A,PetscScalar *array[])
1161a6ece127SHong Zhang {
1162a6ece127SHong Zhang   PetscFunctionBegin;
1163a6ece127SHong Zhang   PetscFunctionReturn(0);
1164a6ece127SHong Zhang  }
1165a6ece127SHong Zhang 
116642ee4b1aSHong Zhang #include "petscblaslapack.h"
116742ee4b1aSHong Zhang #undef __FUNCT__
116842ee4b1aSHong Zhang #define __FUNCT__ "MatAXPY_SeqSBAIJ"
1169f4df32b1SMatthew Knepley PetscErrorCode MatAXPY_SeqSBAIJ(Mat Y,PetscScalar a,Mat X,MatStructure str)
117042ee4b1aSHong Zhang {
117142ee4b1aSHong Zhang   Mat_SeqSBAIJ   *x=(Mat_SeqSBAIJ *)X->data, *y=(Mat_SeqSBAIJ *)Y->data;
1172dfbe8321SBarry Smith   PetscErrorCode ierr;
1173899cda47SBarry Smith   PetscInt       i,bs=Y->rmap.bs,bs2,j;
11744ce68768SBarry Smith   PetscBLASInt   bnz = (PetscBLASInt)x->nz,one = 1;
117542ee4b1aSHong Zhang 
117642ee4b1aSHong Zhang   PetscFunctionBegin;
117742ee4b1aSHong Zhang   if (str == SAME_NONZERO_PATTERN) {
1178f4df32b1SMatthew Knepley     PetscScalar alpha = a;
1179f4df32b1SMatthew Knepley     BLASaxpy_(&bnz,&alpha,x->a,&one,y->a,&one);
1180c537a176SHong Zhang   } else if (str == SUBSET_NONZERO_PATTERN) { /* nonzeros of X is a subset of Y's */
1181c4319e64SHong Zhang     if (y->xtoy && y->XtoY != X) {
1182c4319e64SHong Zhang       ierr = PetscFree(y->xtoy);CHKERRQ(ierr);
1183c4319e64SHong Zhang       ierr = MatDestroy(y->XtoY);CHKERRQ(ierr);
1184c537a176SHong Zhang     }
1185c4319e64SHong Zhang     if (!y->xtoy) { /* get xtoy */
1186c4319e64SHong Zhang       ierr = MatAXPYGetxtoy_Private(x->mbs,x->i,x->j,PETSC_NULL, y->i,y->j,PETSC_NULL, &y->xtoy);CHKERRQ(ierr);
1187c4319e64SHong Zhang       y->XtoY = X;
1188c537a176SHong Zhang     }
1189c4319e64SHong Zhang     bs2 = bs*bs;
11906c6c5352SBarry Smith     for (i=0; i<x->nz; i++) {
1191c4319e64SHong Zhang       j = 0;
1192c4319e64SHong Zhang       while (j < bs2){
1193f4df32b1SMatthew Knepley         y->a[bs2*y->xtoy[i]+j] += a*(x->a[bs2*i+j]);
1194c4319e64SHong Zhang         j++;
1195c537a176SHong Zhang       }
1196c4319e64SHong Zhang     }
1197ae15b995SBarry Smith     ierr = PetscInfo3(0,"ratio of nnz_s(X)/nnz_s(Y): %D/%D = %G\n",bs2*x->nz,bs2*y->nz,(PetscReal)(bs2*x->nz)/(bs2*y->nz));CHKERRQ(ierr);
119842ee4b1aSHong Zhang   } else {
1199f5edf698SHong Zhang     ierr = MatGetRowUpperTriangular(X);CHKERRQ(ierr);
1200f4df32b1SMatthew Knepley     ierr = MatAXPY_Basic(Y,a,X,str);CHKERRQ(ierr);
1201f5edf698SHong Zhang     ierr = MatRestoreRowUpperTriangular(X);CHKERRQ(ierr);
120242ee4b1aSHong Zhang   }
120342ee4b1aSHong Zhang   PetscFunctionReturn(0);
120442ee4b1aSHong Zhang }
120542ee4b1aSHong Zhang 
1206efcf0fc3SBarry Smith #undef __FUNCT__
1207efcf0fc3SBarry Smith #define __FUNCT__ "MatIsSymmetric_SeqSBAIJ"
1208dfbe8321SBarry Smith PetscErrorCode MatIsSymmetric_SeqSBAIJ(Mat A,PetscReal tol,PetscTruth *flg)
1209efcf0fc3SBarry Smith {
1210efcf0fc3SBarry Smith   PetscFunctionBegin;
1211efcf0fc3SBarry Smith   *flg = PETSC_TRUE;
1212efcf0fc3SBarry Smith   PetscFunctionReturn(0);
1213efcf0fc3SBarry Smith }
1214efcf0fc3SBarry Smith 
1215efcf0fc3SBarry Smith #undef __FUNCT__
1216efcf0fc3SBarry Smith #define __FUNCT__ "MatIsStructurallySymmetric_SeqSBAIJ"
1217dfbe8321SBarry Smith PetscErrorCode MatIsStructurallySymmetric_SeqSBAIJ(Mat A,PetscTruth *flg)
1218efcf0fc3SBarry Smith {
1219efcf0fc3SBarry Smith    PetscFunctionBegin;
1220efcf0fc3SBarry Smith    *flg = PETSC_TRUE;
1221efcf0fc3SBarry Smith    PetscFunctionReturn(0);
1222efcf0fc3SBarry Smith }
1223efcf0fc3SBarry Smith 
1224efcf0fc3SBarry Smith #undef __FUNCT__
1225efcf0fc3SBarry Smith #define __FUNCT__ "MatIsHermitian_SeqSBAIJ"
1226dfbe8321SBarry Smith PetscErrorCode MatIsHermitian_SeqSBAIJ(Mat A,PetscTruth *flg)
1227efcf0fc3SBarry Smith  {
1228efcf0fc3SBarry Smith    PetscFunctionBegin;
1229efcf0fc3SBarry Smith    *flg = PETSC_FALSE;
1230efcf0fc3SBarry Smith    PetscFunctionReturn(0);
1231efcf0fc3SBarry Smith  }
1232efcf0fc3SBarry Smith 
123399cafbc1SBarry Smith #undef __FUNCT__
123499cafbc1SBarry Smith #define __FUNCT__ "MatRealPart_SeqSBAIJ"
123599cafbc1SBarry Smith PetscErrorCode MatRealPart_SeqSBAIJ(Mat A)
123699cafbc1SBarry Smith {
123799cafbc1SBarry Smith   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
123899cafbc1SBarry Smith   PetscInt       i,nz = a->bs2*a->i[a->mbs];
123999cafbc1SBarry Smith   PetscScalar    *aa = a->a;
124099cafbc1SBarry Smith 
124199cafbc1SBarry Smith   PetscFunctionBegin;
124299cafbc1SBarry Smith   for (i=0; i<nz; i++) aa[i] = PetscRealPart(aa[i]);
124399cafbc1SBarry Smith   PetscFunctionReturn(0);
124499cafbc1SBarry Smith }
124599cafbc1SBarry Smith 
124699cafbc1SBarry Smith #undef __FUNCT__
124799cafbc1SBarry Smith #define __FUNCT__ "MatImaginaryPart_SeqSBAIJ"
124899cafbc1SBarry Smith PetscErrorCode MatImaginaryPart_SeqSBAIJ(Mat A)
124999cafbc1SBarry Smith {
125099cafbc1SBarry Smith   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
125199cafbc1SBarry Smith   PetscInt       i,nz = a->bs2*a->i[a->mbs];
125299cafbc1SBarry Smith   PetscScalar    *aa = a->a;
125399cafbc1SBarry Smith 
125499cafbc1SBarry Smith   PetscFunctionBegin;
125599cafbc1SBarry Smith   for (i=0; i<nz; i++) aa[i] = PetscImaginaryPart(aa[i]);
125699cafbc1SBarry Smith   PetscFunctionReturn(0);
125799cafbc1SBarry Smith }
125899cafbc1SBarry Smith 
125949b5e25fSSatish Balay /* -------------------------------------------------------------------*/
126049b5e25fSSatish Balay static struct _MatOps MatOps_Values = {MatSetValues_SeqSBAIJ,
126149b5e25fSSatish Balay        MatGetRow_SeqSBAIJ,
126249b5e25fSSatish Balay        MatRestoreRow_SeqSBAIJ,
126349b5e25fSSatish Balay        MatMult_SeqSBAIJ_N,
126497304618SKris Buschelman /* 4*/ MatMultAdd_SeqSBAIJ_N,
1265431c96f7SBarry Smith        MatMult_SeqSBAIJ_N,       /* transpose versions are same as non-transpose versions */
1266e005ede5SBarry Smith        MatMultAdd_SeqSBAIJ_N,
126749b5e25fSSatish Balay        MatSolve_SeqSBAIJ_N,
126849b5e25fSSatish Balay        0,
126949b5e25fSSatish Balay        0,
127097304618SKris Buschelman /*10*/ 0,
127149b5e25fSSatish Balay        0,
12725f4ef2deSHong Zhang        MatCholeskyFactor_SeqSBAIJ,
1273d06b337dSHong Zhang        MatRelax_SeqSBAIJ,
127449b5e25fSSatish Balay        MatTranspose_SeqSBAIJ,
127597304618SKris Buschelman /*15*/ MatGetInfo_SeqSBAIJ,
127649b5e25fSSatish Balay        MatEqual_SeqSBAIJ,
127749b5e25fSSatish Balay        MatGetDiagonal_SeqSBAIJ,
127849b5e25fSSatish Balay        MatDiagonalScale_SeqSBAIJ,
127949b5e25fSSatish Balay        MatNorm_SeqSBAIJ,
128097304618SKris Buschelman /*20*/ 0,
128149b5e25fSSatish Balay        MatAssemblyEnd_SeqSBAIJ,
128249b5e25fSSatish Balay        0,
128349b5e25fSSatish Balay        MatSetOption_SeqSBAIJ,
128449b5e25fSSatish Balay        MatZeroEntries_SeqSBAIJ,
1285dcf5cc72SBarry Smith /*25*/ 0,
128649b5e25fSSatish Balay        0,
128749b5e25fSSatish Balay        0,
12885f4ef2deSHong Zhang        MatCholeskyFactorSymbolic_SeqSBAIJ,
12895f4ef2deSHong Zhang        MatCholeskyFactorNumeric_SeqSBAIJ_N,
129097304618SKris Buschelman /*30*/ MatSetUpPreallocation_SeqSBAIJ,
1291c464158bSHong Zhang        0,
12924d101231SSatish Balay        MatICCFactorSymbolic_SeqSBAIJ,
1293a6ece127SHong Zhang        MatGetArray_SeqSBAIJ,
1294a6ece127SHong Zhang        MatRestoreArray_SeqSBAIJ,
129597304618SKris Buschelman /*35*/ MatDuplicate_SeqSBAIJ,
1296*759322afSHong Zhang        MatForwardSolve_SeqSBAIJ_N_NaturalOrdering,
1297*759322afSHong Zhang        MatBackwardSolve_SeqSBAIJ_N_NaturalOrdering,
129849b5e25fSSatish Balay        0,
1299c84f5b01SHong Zhang        MatICCFactor_SeqSBAIJ,
130097304618SKris Buschelman /*40*/ MatAXPY_SeqSBAIJ,
130149b5e25fSSatish Balay        MatGetSubMatrices_SeqSBAIJ,
130249b5e25fSSatish Balay        MatIncreaseOverlap_SeqSBAIJ,
130349b5e25fSSatish Balay        MatGetValues_SeqSBAIJ,
13043c896bc6SHong Zhang        MatCopy_SeqSBAIJ,
13058c07d4e3SBarry Smith /*45*/ 0,
130649b5e25fSSatish Balay        MatScale_SeqSBAIJ,
130749b5e25fSSatish Balay        0,
130849b5e25fSSatish Balay        0,
130949b5e25fSSatish Balay        0,
1310521d7252SBarry Smith /*50*/ 0,
131149b5e25fSSatish Balay        MatGetRowIJ_SeqSBAIJ,
131249b5e25fSSatish Balay        MatRestoreRowIJ_SeqSBAIJ,
131349b5e25fSSatish Balay        0,
131449b5e25fSSatish Balay        0,
131597304618SKris Buschelman /*55*/ 0,
131649b5e25fSSatish Balay        0,
131749b5e25fSSatish Balay        0,
131849b5e25fSSatish Balay        0,
131949b5e25fSSatish Balay        MatSetValuesBlocked_SeqSBAIJ,
132097304618SKris Buschelman /*60*/ MatGetSubMatrix_SeqSBAIJ,
132149b5e25fSSatish Balay        0,
132249b5e25fSSatish Balay        0,
1323357abbc8SBarry Smith        0,
1324d959ec07SHong Zhang        0,
132597304618SKris Buschelman /*65*/ 0,
1326d959ec07SHong Zhang        0,
1327d959ec07SHong Zhang        0,
1328d959ec07SHong Zhang        0,
1329d959ec07SHong Zhang        0,
133097304618SKris Buschelman /*70*/ MatGetRowMax_SeqSBAIJ,
13313e0d88b5SBarry Smith        0,
13323e0d88b5SBarry Smith        0,
13333e0d88b5SBarry Smith        0,
13343e0d88b5SBarry Smith        0,
133597304618SKris Buschelman /*75*/ 0,
13363e0d88b5SBarry Smith        0,
13373e0d88b5SBarry Smith        0,
13383e0d88b5SBarry Smith        0,
13393e0d88b5SBarry Smith        0,
134097304618SKris Buschelman /*80*/ 0,
13413e0d88b5SBarry Smith        0,
13423e0d88b5SBarry Smith        0,
13433e0d88b5SBarry Smith #if !defined(PETSC_USE_COMPLEX)
134497304618SKris Buschelman        MatGetInertia_SeqSBAIJ,
13453e0d88b5SBarry Smith #else
134697304618SKris Buschelman        0,
13473e0d88b5SBarry Smith #endif
1348865e5f61SKris Buschelman        MatLoad_SeqSBAIJ,
1349865e5f61SKris Buschelman /*85*/ MatIsSymmetric_SeqSBAIJ,
1350865e5f61SKris Buschelman        MatIsHermitian_SeqSBAIJ,
1351efcf0fc3SBarry Smith        MatIsStructurallySymmetric_SeqSBAIJ,
1352865e5f61SKris Buschelman        0,
1353865e5f61SKris Buschelman        0,
1354865e5f61SKris Buschelman /*90*/ 0,
1355865e5f61SKris Buschelman        0,
1356865e5f61SKris Buschelman        0,
1357865e5f61SKris Buschelman        0,
1358865e5f61SKris Buschelman        0,
1359865e5f61SKris Buschelman /*95*/ 0,
1360865e5f61SKris Buschelman        0,
1361865e5f61SKris Buschelman        0,
136299cafbc1SBarry Smith        0,
136399cafbc1SBarry Smith        0,
136499cafbc1SBarry Smith /*100*/0,
136599cafbc1SBarry Smith        0,
136699cafbc1SBarry Smith        0,
136799cafbc1SBarry Smith        0,
136899cafbc1SBarry Smith        0,
136999cafbc1SBarry Smith /*105*/0,
137099cafbc1SBarry Smith        MatRealPart_SeqSBAIJ,
1371f5edf698SHong Zhang        MatImaginaryPart_SeqSBAIJ,
1372f5edf698SHong Zhang        MatGetRowUpperTriangular_SeqSBAIJ,
1373f5edf698SHong Zhang        MatRestoreRowUpperTriangular_SeqSBAIJ
137499cafbc1SBarry Smith };
1375be1d678aSKris Buschelman 
137649b5e25fSSatish Balay EXTERN_C_BEGIN
13774a2ae208SSatish Balay #undef __FUNCT__
13784a2ae208SSatish Balay #define __FUNCT__ "MatStoreValues_SeqSBAIJ"
1379be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatStoreValues_SeqSBAIJ(Mat mat)
138049b5e25fSSatish Balay {
13814afc71dfSHong Zhang   Mat_SeqSBAIJ   *aij = (Mat_SeqSBAIJ *)mat->data;
1382899cda47SBarry Smith   PetscInt       nz = aij->i[mat->rmap.N]*mat->rmap.bs*aij->bs2;
1383dfbe8321SBarry Smith   PetscErrorCode ierr;
138449b5e25fSSatish Balay 
138549b5e25fSSatish Balay   PetscFunctionBegin;
138649b5e25fSSatish Balay   if (aij->nonew != 1) {
1387e005ede5SBarry Smith     SETERRQ(PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NO_NEW_NONZERO_LOCATIONS);first");
138849b5e25fSSatish Balay   }
138949b5e25fSSatish Balay 
139049b5e25fSSatish Balay   /* allocate space for values if not already there */
139149b5e25fSSatish Balay   if (!aij->saved_values) {
139287828ca2SBarry Smith     ierr = PetscMalloc((nz+1)*sizeof(PetscScalar),&aij->saved_values);CHKERRQ(ierr);
139349b5e25fSSatish Balay   }
139449b5e25fSSatish Balay 
139549b5e25fSSatish Balay   /* copy values over */
139687828ca2SBarry Smith   ierr = PetscMemcpy(aij->saved_values,aij->a,nz*sizeof(PetscScalar));CHKERRQ(ierr);
139749b5e25fSSatish Balay   PetscFunctionReturn(0);
139849b5e25fSSatish Balay }
139949b5e25fSSatish Balay EXTERN_C_END
140049b5e25fSSatish Balay 
140149b5e25fSSatish Balay EXTERN_C_BEGIN
14024a2ae208SSatish Balay #undef __FUNCT__
14034a2ae208SSatish Balay #define __FUNCT__ "MatRetrieveValues_SeqSBAIJ"
1404be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatRetrieveValues_SeqSBAIJ(Mat mat)
140549b5e25fSSatish Balay {
14064afc71dfSHong Zhang   Mat_SeqSBAIJ   *aij = (Mat_SeqSBAIJ *)mat->data;
14076849ba73SBarry Smith   PetscErrorCode ierr;
1408899cda47SBarry Smith   PetscInt       nz = aij->i[mat->rmap.N]*mat->rmap.bs*aij->bs2;
140949b5e25fSSatish Balay 
141049b5e25fSSatish Balay   PetscFunctionBegin;
141149b5e25fSSatish Balay   if (aij->nonew != 1) {
1412e005ede5SBarry Smith     SETERRQ(PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NO_NEW_NONZERO_LOCATIONS);first");
141349b5e25fSSatish Balay   }
141449b5e25fSSatish Balay   if (!aij->saved_values) {
1415e005ede5SBarry Smith     SETERRQ(PETSC_ERR_ORDER,"Must call MatStoreValues(A);first");
141649b5e25fSSatish Balay   }
141749b5e25fSSatish Balay 
141849b5e25fSSatish Balay   /* copy values over */
141987828ca2SBarry Smith   ierr = PetscMemcpy(aij->a,aij->saved_values,nz*sizeof(PetscScalar));CHKERRQ(ierr);
142049b5e25fSSatish Balay   PetscFunctionReturn(0);
142149b5e25fSSatish Balay }
142249b5e25fSSatish Balay EXTERN_C_END
142349b5e25fSSatish Balay 
14248549e402SHong Zhang EXTERN_C_BEGIN
14254a2ae208SSatish Balay #undef __FUNCT__
1426a23d5eceSKris Buschelman #define __FUNCT__ "MatSeqSBAIJSetPreallocation_SeqSBAIJ"
1427be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatSeqSBAIJSetPreallocation_SeqSBAIJ(Mat B,PetscInt bs,PetscInt nz,PetscInt *nnz)
142849b5e25fSSatish Balay {
1429c464158bSHong Zhang   Mat_SeqSBAIJ   *b = (Mat_SeqSBAIJ*)B->data;
14306849ba73SBarry Smith   PetscErrorCode ierr;
1431ab93d7beSBarry Smith   PetscInt       i,mbs,bs2;
1432ab93d7beSBarry Smith   PetscTruth     skipallocation = PETSC_FALSE,flg;
143349b5e25fSSatish Balay 
143449b5e25fSSatish Balay   PetscFunctionBegin;
1435273d9f13SBarry Smith   B->preallocated = PETSC_TRUE;
1436e82a3eeeSBarry Smith   ierr = PetscOptionsGetInt(B->prefix,"-mat_block_size",&bs,PETSC_NULL);CHKERRQ(ierr);
1437899cda47SBarry Smith   B->rmap.bs = B->cmap.bs = bs;
1438899cda47SBarry Smith   ierr = PetscMapInitialize(B->comm,&B->rmap);CHKERRQ(ierr);
1439899cda47SBarry Smith   ierr = PetscMapInitialize(B->comm,&B->cmap);CHKERRQ(ierr);
1440899cda47SBarry Smith 
1441899cda47SBarry Smith   mbs  = B->rmap.N/bs;
144249b5e25fSSatish Balay   bs2  = bs*bs;
144349b5e25fSSatish Balay 
1444899cda47SBarry Smith   if (mbs*bs != B->rmap.N) {
144529bbc08cSBarry Smith     SETERRQ(PETSC_ERR_ARG_SIZ,"Number rows, cols must be divisible by blocksize");
144649b5e25fSSatish Balay   }
144749b5e25fSSatish Balay 
1448ab93d7beSBarry Smith   if (nz == MAT_SKIP_ALLOCATION) {
1449ab93d7beSBarry Smith     skipallocation = PETSC_TRUE;
1450ab93d7beSBarry Smith     nz             = 0;
1451ab93d7beSBarry Smith   }
1452ab93d7beSBarry Smith 
1453435da068SBarry Smith   if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 3;
145477431f27SBarry Smith   if (nz < 0) SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"nz cannot be less than 0: value %D",nz);
145549b5e25fSSatish Balay   if (nnz) {
145649b5e25fSSatish Balay     for (i=0; i<mbs; i++) {
145777431f27SBarry Smith       if (nnz[i] < 0) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"nnz cannot be less than 0: local row %D value %D",i,nnz[i]);
145877431f27SBarry Smith       if (nnz[i] > mbs) SETERRQ3(PETSC_ERR_ARG_OUTOFRANGE,"nnz cannot be greater than block row length: local row %D value %D rowlength %D",i,nnz[i],mbs);
145949b5e25fSSatish Balay     }
146049b5e25fSSatish Balay   }
146149b5e25fSSatish Balay 
1462e82a3eeeSBarry Smith   ierr    = PetscOptionsHasName(B->prefix,"-mat_no_unroll",&flg);CHKERRQ(ierr);
146349b5e25fSSatish Balay   if (!flg) {
146449b5e25fSSatish Balay     switch (bs) {
146549b5e25fSSatish Balay     case 1:
14664ccecd49SHong Zhang       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1;
146749b5e25fSSatish Balay       B->ops->solve            = MatSolve_SeqSBAIJ_1;
1468d59c15a7SBarry Smith       B->ops->solves           = MatSolves_SeqSBAIJ_1;
1469e005ede5SBarry Smith       B->ops->solvetranspose   = MatSolve_SeqSBAIJ_1;
147049b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_1;
147149b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_1;
1472431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_1;
1473431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_1;
147449b5e25fSSatish Balay       break;
147549b5e25fSSatish Balay     case 2:
14764ccecd49SHong Zhang       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2;
147749b5e25fSSatish Balay       B->ops->solve            = MatSolve_SeqSBAIJ_2;
1478e005ede5SBarry Smith       B->ops->solvetranspose   = MatSolve_SeqSBAIJ_2;
147949b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_2;
148049b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_2;
1481431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_2;
1482431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_2;
148349b5e25fSSatish Balay       break;
148449b5e25fSSatish Balay     case 3:
14855f4ef2deSHong Zhang       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3;
148649b5e25fSSatish Balay       B->ops->solve            = MatSolve_SeqSBAIJ_3;
1487e005ede5SBarry Smith       B->ops->solvetranspose   = MatSolve_SeqSBAIJ_3;
148849b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_3;
148949b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_3;
1490431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_3;
1491431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_3;
149249b5e25fSSatish Balay       break;
149349b5e25fSSatish Balay     case 4:
14945f4ef2deSHong Zhang       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4;
149549b5e25fSSatish Balay       B->ops->solve            = MatSolve_SeqSBAIJ_4;
1496e005ede5SBarry Smith       B->ops->solvetranspose   = MatSolve_SeqSBAIJ_4;
149749b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_4;
149849b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_4;
1499431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_4;
1500431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_4;
150149b5e25fSSatish Balay       break;
150249b5e25fSSatish Balay     case 5:
15035f4ef2deSHong Zhang       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5;
150449b5e25fSSatish Balay       B->ops->solve            = MatSolve_SeqSBAIJ_5;
1505e005ede5SBarry Smith       B->ops->solvetranspose   = MatSolve_SeqSBAIJ_5;
150649b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_5;
150749b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_5;
1508431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_5;
1509431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_5;
151049b5e25fSSatish Balay       break;
151149b5e25fSSatish Balay     case 6:
15125f4ef2deSHong Zhang       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6;
151349b5e25fSSatish Balay       B->ops->solve            = MatSolve_SeqSBAIJ_6;
1514e005ede5SBarry Smith       B->ops->solvetranspose   = MatSolve_SeqSBAIJ_6;
151549b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_6;
151649b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_6;
1517431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_6;
1518431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_6;
151949b5e25fSSatish Balay       break;
152049b5e25fSSatish Balay     case 7:
1521de53e5efSHong Zhang       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7;
152249b5e25fSSatish Balay       B->ops->solve            = MatSolve_SeqSBAIJ_7;
1523e005ede5SBarry Smith       B->ops->solvetranspose   = MatSolve_SeqSBAIJ_7;
1524de53e5efSHong Zhang       B->ops->mult             = MatMult_SeqSBAIJ_7;
152549b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_7;
1526431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_7;
1527431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_7;
152849b5e25fSSatish Balay       break;
152949b5e25fSSatish Balay     }
153049b5e25fSSatish Balay   }
153149b5e25fSSatish Balay 
153249b5e25fSSatish Balay   b->mbs = mbs;
15334afc71dfSHong Zhang   b->nbs = mbs;
1534ab93d7beSBarry Smith   if (!skipallocation) {
1535ab93d7beSBarry Smith     /* b->ilen will count nonzeros in each block row so far. */
1536ab93d7beSBarry Smith     ierr   = PetscMalloc2(mbs,PetscInt,&b->imax,mbs,PetscInt,&b->ilen);CHKERRQ(ierr);
1537ab93d7beSBarry Smith     for (i=0; i<mbs; i++) { b->ilen[i] = 0;}
153849b5e25fSSatish Balay     if (!nnz) {
1539435da068SBarry Smith       if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 5;
154049b5e25fSSatish Balay       else if (nz <= 0)        nz = 1;
154149b5e25fSSatish Balay       for (i=0; i<mbs; i++) {
15428cef66ccSHong Zhang         b->imax[i] = nz;
154349b5e25fSSatish Balay       }
1544153ea458SHong Zhang       nz = nz*mbs; /* total nz */
154549b5e25fSSatish Balay     } else {
154649b5e25fSSatish Balay       nz = 0;
15478cef66ccSHong Zhang       for (i=0; i<mbs; i++) {b->imax[i] = nnz[i]; nz += nnz[i];}
154849b5e25fSSatish Balay     }
15496c6c5352SBarry Smith     /* nz=(nz+mbs)/2; */ /* total diagonal and superdiagonal nonzero blocks */
155049b5e25fSSatish Balay 
155149b5e25fSSatish Balay     /* allocate the matrix space */
1552899cda47SBarry Smith     ierr = PetscMalloc3(bs2*nz,PetscScalar,&b->a,nz,PetscInt,&b->j,B->rmap.N+1,PetscInt,&b->i);CHKERRQ(ierr);
15536c6c5352SBarry Smith     ierr = PetscMemzero(b->a,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr);
155413f74950SBarry Smith     ierr = PetscMemzero(b->j,nz*sizeof(PetscInt));CHKERRQ(ierr);
155549b5e25fSSatish Balay     b->singlemalloc = PETSC_TRUE;
155649b5e25fSSatish Balay 
155749b5e25fSSatish Balay     /* pointer to beginning of each row */
155849b5e25fSSatish Balay     b->i[0] = 0;
155949b5e25fSSatish Balay     for (i=1; i<mbs+1; i++) {
156049b5e25fSSatish Balay       b->i[i] = b->i[i-1] + b->imax[i-1];
156149b5e25fSSatish Balay     }
1562e6b907acSBarry Smith     b->free_a     = PETSC_TRUE;
1563e6b907acSBarry Smith     b->free_ij    = PETSC_TRUE;
1564e811da20SHong Zhang   } else {
1565e6b907acSBarry Smith     b->free_a     = PETSC_FALSE;
1566e6b907acSBarry Smith     b->free_ij    = PETSC_FALSE;
1567ab93d7beSBarry Smith   }
156849b5e25fSSatish Balay 
1569899cda47SBarry Smith   B->rmap.bs               = bs;
157049b5e25fSSatish Balay   b->bs2              = bs2;
15716c6c5352SBarry Smith   b->nz             = 0;
15726c6c5352SBarry Smith   b->maxnz          = nz*bs2;
1573153ea458SHong Zhang 
157416cdd363SHong Zhang   b->inew             = 0;
157516cdd363SHong Zhang   b->jnew             = 0;
157616cdd363SHong Zhang   b->anew             = 0;
157716cdd363SHong Zhang   b->a2anew           = 0;
15781a3463dfSHong Zhang   b->permute          = PETSC_FALSE;
1579c464158bSHong Zhang   PetscFunctionReturn(0);
1580c464158bSHong Zhang }
1581a23d5eceSKris Buschelman EXTERN_C_END
1582153ea458SHong Zhang 
1583d769727bSBarry Smith EXTERN_C_BEGIN
1584f69a0ea3SMatthew Knepley EXTERN PetscErrorCode PETSCMAT_DLLEXPORT MatConvert_SeqSBAIJ_SeqAIJ(Mat, MatType,MatReuse,Mat*);
1585f69a0ea3SMatthew Knepley EXTERN PetscErrorCode PETSCMAT_DLLEXPORT MatConvert_SeqSBAIJ_SeqBAIJ(Mat, MatType,MatReuse,Mat*);
1586d769727bSBarry Smith EXTERN_C_END
1587d769727bSBarry Smith 
15880bad9183SKris Buschelman /*MC
1589fafad747SKris Buschelman   MATSEQSBAIJ - MATSEQSBAIJ = "seqsbaij" - A matrix type to be used for sequential symmetric block sparse matrices,
15900bad9183SKris Buschelman   based on block compressed sparse row format.  Only the upper triangular portion of the matrix is stored.
15910bad9183SKris Buschelman 
15920bad9183SKris Buschelman   Options Database Keys:
15930bad9183SKris Buschelman   . -mat_type seqsbaij - sets the matrix type to "seqsbaij" during a call to MatSetFromOptions()
15940bad9183SKris Buschelman 
15950bad9183SKris Buschelman   Level: beginner
15960bad9183SKris Buschelman 
15970bad9183SKris Buschelman   .seealso: MatCreateSeqSBAIJ
15980bad9183SKris Buschelman M*/
15990bad9183SKris Buschelman 
1600a23d5eceSKris Buschelman EXTERN_C_BEGIN
1601a23d5eceSKris Buschelman #undef __FUNCT__
1602a23d5eceSKris Buschelman #define __FUNCT__ "MatCreate_SeqSBAIJ"
1603be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreate_SeqSBAIJ(Mat B)
1604a23d5eceSKris Buschelman {
1605a23d5eceSKris Buschelman   Mat_SeqSBAIJ   *b;
1606dfbe8321SBarry Smith   PetscErrorCode ierr;
160713f74950SBarry Smith   PetscMPIInt    size;
1608941593c8SHong Zhang   PetscTruth     flg;
1609a23d5eceSKris Buschelman 
1610a23d5eceSKris Buschelman   PetscFunctionBegin;
1611a23d5eceSKris Buschelman   ierr = MPI_Comm_size(B->comm,&size);CHKERRQ(ierr);
1612a23d5eceSKris Buschelman   if (size > 1) SETERRQ(PETSC_ERR_ARG_WRONG,"Comm must be of size 1");
1613a23d5eceSKris Buschelman 
1614a23d5eceSKris Buschelman   ierr    = PetscNew(Mat_SeqSBAIJ,&b);CHKERRQ(ierr);
1615a23d5eceSKris Buschelman   B->data = (void*)b;
1616a23d5eceSKris Buschelman   ierr    = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr);
1617a23d5eceSKris Buschelman   B->ops->destroy     = MatDestroy_SeqSBAIJ;
1618a23d5eceSKris Buschelman   B->ops->view        = MatView_SeqSBAIJ;
1619a23d5eceSKris Buschelman   B->factor           = 0;
1620a23d5eceSKris Buschelman   B->mapping          = 0;
1621a23d5eceSKris Buschelman   b->row              = 0;
1622a23d5eceSKris Buschelman   b->icol             = 0;
1623a23d5eceSKris Buschelman   b->reallocs         = 0;
1624a23d5eceSKris Buschelman   b->saved_values     = 0;
1625a23d5eceSKris Buschelman 
1626a23d5eceSKris Buschelman 
1627a23d5eceSKris Buschelman   b->sorted           = PETSC_FALSE;
1628a23d5eceSKris Buschelman   b->roworiented      = PETSC_TRUE;
1629a23d5eceSKris Buschelman   b->nonew            = 0;
1630a23d5eceSKris Buschelman   b->diag             = 0;
1631a23d5eceSKris Buschelman   b->solve_work       = 0;
1632a23d5eceSKris Buschelman   b->mult_work        = 0;
1633a23d5eceSKris Buschelman   B->spptr            = 0;
1634a23d5eceSKris Buschelman   b->keepzeroedrows   = PETSC_FALSE;
1635a23d5eceSKris Buschelman   b->xtoy             = 0;
1636a23d5eceSKris Buschelman   b->XtoY             = 0;
1637a23d5eceSKris Buschelman 
1638a23d5eceSKris Buschelman   b->inew             = 0;
1639a23d5eceSKris Buschelman   b->jnew             = 0;
1640a23d5eceSKris Buschelman   b->anew             = 0;
1641a23d5eceSKris Buschelman   b->a2anew           = 0;
1642a23d5eceSKris Buschelman   b->permute          = PETSC_FALSE;
1643a23d5eceSKris Buschelman 
1644941593c8SHong Zhang   b->ignore_ltriangular = PETSC_FALSE;
1645941593c8SHong Zhang   ierr = PetscOptionsHasName(PETSC_NULL,"-mat_ignore_lower_triangular",&flg);CHKERRQ(ierr);
1646941593c8SHong Zhang   if (flg) b->ignore_ltriangular = PETSC_TRUE;
1647941593c8SHong Zhang 
1648f5edf698SHong Zhang   b->getrow_utriangular = PETSC_FALSE;
1649f5edf698SHong Zhang   ierr = PetscOptionsHasName(PETSC_NULL,"-mat_getrow_uppertriangular",&flg);CHKERRQ(ierr);
1650f5edf698SHong Zhang   if (flg) b->getrow_utriangular = PETSC_TRUE;
1651f5edf698SHong Zhang 
1652a23d5eceSKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatStoreValues_C",
1653a23d5eceSKris Buschelman                                      "MatStoreValues_SeqSBAIJ",
1654a23d5eceSKris Buschelman                                      MatStoreValues_SeqSBAIJ);CHKERRQ(ierr);
1655a23d5eceSKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatRetrieveValues_C",
1656a23d5eceSKris Buschelman                                      "MatRetrieveValues_SeqSBAIJ",
1657a23d5eceSKris Buschelman                                      (void*)MatRetrieveValues_SeqSBAIJ);CHKERRQ(ierr);
1658a23d5eceSKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatSeqSBAIJSetColumnIndices_C",
1659a23d5eceSKris Buschelman                                      "MatSeqSBAIJSetColumnIndices_SeqSBAIJ",
1660a23d5eceSKris Buschelman                                      MatSeqSBAIJSetColumnIndices_SeqSBAIJ);CHKERRQ(ierr);
16614e5e7fe4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_seqsbaij_seqaij_C",
16624e5e7fe4SHong Zhang                                      "MatConvert_SeqSBAIJ_SeqAIJ",
16634e5e7fe4SHong Zhang                                       MatConvert_SeqSBAIJ_SeqAIJ);CHKERRQ(ierr);
1664a0e1a404SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_seqsbaij_seqbaij_C",
1665a0e1a404SHong Zhang                                      "MatConvert_SeqSBAIJ_SeqBAIJ",
1666a0e1a404SHong Zhang                                       MatConvert_SeqSBAIJ_SeqBAIJ);CHKERRQ(ierr);
1667a23d5eceSKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatSeqSBAIJSetPreallocation_C",
1668a23d5eceSKris Buschelman                                      "MatSeqSBAIJSetPreallocation_SeqSBAIJ",
1669a23d5eceSKris Buschelman                                      MatSeqSBAIJSetPreallocation_SeqSBAIJ);CHKERRQ(ierr);
167023ce1328SBarry Smith 
167123ce1328SBarry Smith   B->symmetric                  = PETSC_TRUE;
167223ce1328SBarry Smith   B->structurally_symmetric     = PETSC_TRUE;
167323ce1328SBarry Smith   B->symmetric_set              = PETSC_TRUE;
167423ce1328SBarry Smith   B->structurally_symmetric_set = PETSC_TRUE;
167517667f90SBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)B,MATSEQSBAIJ);CHKERRQ(ierr);
1676a23d5eceSKris Buschelman   PetscFunctionReturn(0);
1677a23d5eceSKris Buschelman }
1678a23d5eceSKris Buschelman EXTERN_C_END
1679a23d5eceSKris Buschelman 
1680a23d5eceSKris Buschelman #undef __FUNCT__
1681a23d5eceSKris Buschelman #define __FUNCT__ "MatSeqSBAIJSetPreallocation"
1682a23d5eceSKris Buschelman /*@C
1683a23d5eceSKris Buschelman    MatSeqSBAIJSetPreallocation - Creates a sparse symmetric matrix in block AIJ (block
1684a23d5eceSKris Buschelman    compressed row) format.  For good matrix assembly performance the
1685a23d5eceSKris Buschelman    user should preallocate the matrix storage by setting the parameter nz
1686a23d5eceSKris Buschelman    (or the array nnz).  By setting these parameters accurately, performance
1687a23d5eceSKris Buschelman    during matrix assembly can be increased by more than a factor of 50.
1688a23d5eceSKris Buschelman 
1689a23d5eceSKris Buschelman    Collective on Mat
1690a23d5eceSKris Buschelman 
1691a23d5eceSKris Buschelman    Input Parameters:
1692a23d5eceSKris Buschelman +  A - the symmetric matrix
1693a23d5eceSKris Buschelman .  bs - size of block
1694a23d5eceSKris Buschelman .  nz - number of block nonzeros per block row (same for all rows)
1695a23d5eceSKris Buschelman -  nnz - array containing the number of block nonzeros in the upper triangular plus
1696a23d5eceSKris Buschelman          diagonal portion of each block (possibly different for each block row) or PETSC_NULL
1697a23d5eceSKris Buschelman 
1698a23d5eceSKris Buschelman    Options Database Keys:
1699a23d5eceSKris Buschelman .   -mat_no_unroll - uses code that does not unroll the loops in the
1700a23d5eceSKris Buschelman                      block calculations (much slower)
1701a23d5eceSKris Buschelman .    -mat_block_size - size of the blocks to use
1702a23d5eceSKris Buschelman 
1703a23d5eceSKris Buschelman    Level: intermediate
1704a23d5eceSKris Buschelman 
1705a23d5eceSKris Buschelman    Notes:
1706a23d5eceSKris Buschelman    Specify the preallocated storage with either nz or nnz (not both).
1707a23d5eceSKris Buschelman    Set nz=PETSC_DEFAULT and nnz=PETSC_NULL for PETSc to control dynamic memory
1708a23d5eceSKris Buschelman    allocation.  For additional details, see the users manual chapter on
1709a23d5eceSKris Buschelman    matrices.
1710a23d5eceSKris Buschelman 
171149a6f317SBarry Smith    If the nnz parameter is given then the nz parameter is ignored
171249a6f317SBarry Smith 
171349a6f317SBarry Smith 
1714a23d5eceSKris Buschelman .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateMPISBAIJ()
1715a23d5eceSKris Buschelman @*/
1716be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatSeqSBAIJSetPreallocation(Mat B,PetscInt bs,PetscInt nz,const PetscInt nnz[])
171713f74950SBarry Smith {
171813f74950SBarry Smith   PetscErrorCode ierr,(*f)(Mat,PetscInt,PetscInt,const PetscInt[]);
1719a23d5eceSKris Buschelman 
1720a23d5eceSKris Buschelman   PetscFunctionBegin;
1721a23d5eceSKris Buschelman   ierr = PetscObjectQueryFunction((PetscObject)B,"MatSeqSBAIJSetPreallocation_C",(void (**)(void))&f);CHKERRQ(ierr);
1722a23d5eceSKris Buschelman   if (f) {
1723a23d5eceSKris Buschelman     ierr = (*f)(B,bs,nz,nnz);CHKERRQ(ierr);
1724a23d5eceSKris Buschelman   }
1725a23d5eceSKris Buschelman   PetscFunctionReturn(0);
1726a23d5eceSKris Buschelman }
172749b5e25fSSatish Balay 
17284a2ae208SSatish Balay #undef __FUNCT__
17294a2ae208SSatish Balay #define __FUNCT__ "MatCreateSeqSBAIJ"
1730c464158bSHong Zhang /*@C
1731c464158bSHong Zhang    MatCreateSeqSBAIJ - Creates a sparse symmetric matrix in block AIJ (block
1732c464158bSHong Zhang    compressed row) format.  For good matrix assembly performance the
1733c464158bSHong Zhang    user should preallocate the matrix storage by setting the parameter nz
1734c464158bSHong Zhang    (or the array nnz).  By setting these parameters accurately, performance
1735c464158bSHong Zhang    during matrix assembly can be increased by more than a factor of 50.
173649b5e25fSSatish Balay 
1737c464158bSHong Zhang    Collective on MPI_Comm
1738c464158bSHong Zhang 
1739c464158bSHong Zhang    Input Parameters:
1740c464158bSHong Zhang +  comm - MPI communicator, set to PETSC_COMM_SELF
1741c464158bSHong Zhang .  bs - size of block
1742c464158bSHong Zhang .  m - number of rows, or number of columns
1743c464158bSHong Zhang .  nz - number of block nonzeros per block row (same for all rows)
1744744e8345SSatish Balay -  nnz - array containing the number of block nonzeros in the upper triangular plus
1745744e8345SSatish Balay          diagonal portion of each block (possibly different for each block row) or PETSC_NULL
1746c464158bSHong Zhang 
1747c464158bSHong Zhang    Output Parameter:
1748c464158bSHong Zhang .  A - the symmetric matrix
1749c464158bSHong Zhang 
1750c464158bSHong Zhang    Options Database Keys:
1751c464158bSHong Zhang .   -mat_no_unroll - uses code that does not unroll the loops in the
1752c464158bSHong Zhang                      block calculations (much slower)
1753c464158bSHong Zhang .    -mat_block_size - size of the blocks to use
1754c464158bSHong Zhang 
1755c464158bSHong Zhang    Level: intermediate
1756c464158bSHong Zhang 
1757c464158bSHong Zhang    Notes:
1758c464158bSHong Zhang 
1759c464158bSHong Zhang    Specify the preallocated storage with either nz or nnz (not both).
1760c464158bSHong Zhang    Set nz=PETSC_DEFAULT and nnz=PETSC_NULL for PETSc to control dynamic memory
1761c464158bSHong Zhang    allocation.  For additional details, see the users manual chapter on
1762c464158bSHong Zhang    matrices.
1763c464158bSHong Zhang 
176449a6f317SBarry Smith    If the nnz parameter is given then the nz parameter is ignored
176549a6f317SBarry Smith 
1766c464158bSHong Zhang .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateMPISBAIJ()
1767c464158bSHong Zhang @*/
1768be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreateSeqSBAIJ(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt nz,const PetscInt nnz[],Mat *A)
1769c464158bSHong Zhang {
1770dfbe8321SBarry Smith   PetscErrorCode ierr;
1771c464158bSHong Zhang 
1772c464158bSHong Zhang   PetscFunctionBegin;
1773f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,A);CHKERRQ(ierr);
1774f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*A,m,n,m,n);CHKERRQ(ierr);
1775c464158bSHong Zhang   ierr = MatSetType(*A,MATSEQSBAIJ);CHKERRQ(ierr);
1776ab93d7beSBarry Smith   ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(*A,bs,nz,(PetscInt*)nnz);CHKERRQ(ierr);
177749b5e25fSSatish Balay   PetscFunctionReturn(0);
177849b5e25fSSatish Balay }
177949b5e25fSSatish Balay 
17804a2ae208SSatish Balay #undef __FUNCT__
17814a2ae208SSatish Balay #define __FUNCT__ "MatDuplicate_SeqSBAIJ"
1782dfbe8321SBarry Smith PetscErrorCode MatDuplicate_SeqSBAIJ(Mat A,MatDuplicateOption cpvalues,Mat *B)
178349b5e25fSSatish Balay {
178449b5e25fSSatish Balay   Mat            C;
178549b5e25fSSatish Balay   Mat_SeqSBAIJ   *c,*a = (Mat_SeqSBAIJ*)A->data;
17866849ba73SBarry Smith   PetscErrorCode ierr;
1787b40805acSSatish Balay   PetscInt       i,mbs = a->mbs,nz = a->nz,bs2 =a->bs2;
178849b5e25fSSatish Balay 
178949b5e25fSSatish Balay   PetscFunctionBegin;
179029bbc08cSBarry Smith   if (a->i[mbs] != nz) SETERRQ(PETSC_ERR_PLIB,"Corrupt matrix");
179149b5e25fSSatish Balay 
179249b5e25fSSatish Balay   *B = 0;
1793f69a0ea3SMatthew Knepley   ierr = MatCreate(A->comm,&C);CHKERRQ(ierr);
1794899cda47SBarry Smith   ierr = MatSetSizes(C,A->rmap.N,A->cmap.n,A->rmap.N,A->cmap.n);CHKERRQ(ierr);
1795be5d1d56SKris Buschelman   ierr = MatSetType(C,A->type_name);CHKERRQ(ierr);
17961d5dac46SHong Zhang   ierr = PetscMemcpy(C->ops,A->ops,sizeof(struct _MatOps));CHKERRQ(ierr);
1797692f9cbeSHong Zhang   c    = (Mat_SeqSBAIJ*)C->data;
1798692f9cbeSHong Zhang 
1799273d9f13SBarry Smith   C->preallocated   = PETSC_TRUE;
180049b5e25fSSatish Balay   C->factor         = A->factor;
180149b5e25fSSatish Balay   c->row            = 0;
180249b5e25fSSatish Balay   c->icol           = 0;
180349b5e25fSSatish Balay   c->saved_values   = 0;
180449b5e25fSSatish Balay   c->keepzeroedrows = a->keepzeroedrows;
180549b5e25fSSatish Balay   C->assembled      = PETSC_TRUE;
180649b5e25fSSatish Balay 
1807899cda47SBarry Smith   ierr = PetscMapCopy(A->comm,&A->rmap,&C->rmap);CHKERRQ(ierr);
1808899cda47SBarry Smith   ierr = PetscMapCopy(A->comm,&A->cmap,&C->cmap);CHKERRQ(ierr);
180949b5e25fSSatish Balay   c->bs2  = a->bs2;
181049b5e25fSSatish Balay   c->mbs  = a->mbs;
181149b5e25fSSatish Balay   c->nbs  = a->nbs;
181249b5e25fSSatish Balay 
18138777fc3fSSatish Balay   ierr = PetscMalloc2((mbs+1),PetscInt,&c->imax,(mbs+1),PetscInt,&c->ilen);CHKERRQ(ierr);
181449b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
181549b5e25fSSatish Balay     c->imax[i] = a->imax[i];
181649b5e25fSSatish Balay     c->ilen[i] = a->ilen[i];
181749b5e25fSSatish Balay   }
181849b5e25fSSatish Balay 
181949b5e25fSSatish Balay   /* allocate the matrix space */
1820b40805acSSatish Balay   ierr = PetscMalloc3(bs2*nz,MatScalar,&c->a,nz,PetscInt,&c->j,mbs+1,PetscInt,&c->i);CHKERRQ(ierr);
182149b5e25fSSatish Balay   c->singlemalloc = PETSC_TRUE;
182213f74950SBarry Smith   ierr = PetscMemcpy(c->i,a->i,(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr);
1823b40805acSSatish Balay   ierr = PetscLogObjectMemory(C,(mbs+1)*sizeof(PetscInt) + nz*(bs2*sizeof(MatScalar) + sizeof(PetscInt)));CHKERRQ(ierr);
182449b5e25fSSatish Balay   if (mbs > 0) {
182513f74950SBarry Smith     ierr = PetscMemcpy(c->j,a->j,nz*sizeof(PetscInt));CHKERRQ(ierr);
182649b5e25fSSatish Balay     if (cpvalues == MAT_COPY_VALUES) {
182749b5e25fSSatish Balay       ierr = PetscMemcpy(c->a,a->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr);
182849b5e25fSSatish Balay     } else {
182949b5e25fSSatish Balay       ierr = PetscMemzero(c->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr);
183049b5e25fSSatish Balay     }
183149b5e25fSSatish Balay   }
183249b5e25fSSatish Balay 
183349b5e25fSSatish Balay   c->sorted      = a->sorted;
183449b5e25fSSatish Balay   c->roworiented = a->roworiented;
183549b5e25fSSatish Balay   c->nonew       = a->nonew;
183649b5e25fSSatish Balay 
183749b5e25fSSatish Balay   if (a->diag) {
183813f74950SBarry Smith     ierr = PetscMalloc((mbs+1)*sizeof(PetscInt),&c->diag);CHKERRQ(ierr);
183952e6d16bSBarry Smith     ierr = PetscLogObjectMemory(C,(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr);
184049b5e25fSSatish Balay     for (i=0; i<mbs; i++) {
184149b5e25fSSatish Balay       c->diag[i] = a->diag[i];
184249b5e25fSSatish Balay     }
184349b5e25fSSatish Balay   } else c->diag  = 0;
18446c6c5352SBarry Smith   c->nz           = a->nz;
18456c6c5352SBarry Smith   c->maxnz        = a->maxnz;
184649b5e25fSSatish Balay   c->solve_work   = 0;
184749b5e25fSSatish Balay   c->mult_work    = 0;
1848e6b907acSBarry Smith   c->free_a       = PETSC_TRUE;
1849e6b907acSBarry Smith   c->free_ij      = PETSC_TRUE;
185049b5e25fSSatish Balay   *B = C;
1851b0a32e0cSBarry Smith   ierr = PetscFListDuplicate(A->qlist,&C->qlist);CHKERRQ(ierr);
185249b5e25fSSatish Balay   PetscFunctionReturn(0);
185349b5e25fSSatish Balay }
185449b5e25fSSatish Balay 
18554a2ae208SSatish Balay #undef __FUNCT__
18564a2ae208SSatish Balay #define __FUNCT__ "MatLoad_SeqSBAIJ"
1857f69a0ea3SMatthew Knepley PetscErrorCode MatLoad_SeqSBAIJ(PetscViewer viewer, MatType type,Mat *A)
185849b5e25fSSatish Balay {
185949b5e25fSSatish Balay   Mat_SeqSBAIJ   *a;
186049b5e25fSSatish Balay   Mat            B;
18616849ba73SBarry Smith   PetscErrorCode ierr;
186213f74950SBarry Smith   int            fd;
186313f74950SBarry Smith   PetscMPIInt    size;
186413f74950SBarry Smith   PetscInt       i,nz,header[4],*rowlengths=0,M,N,bs=1;
186513f74950SBarry Smith   PetscInt       *mask,mbs,*jj,j,rowcount,nzcount,k,*s_browlengths,maskcount;
186613f74950SBarry Smith   PetscInt       kmax,jcount,block,idx,point,nzcountb,extra_rows;
186713f74950SBarry Smith   PetscInt       *masked,nmask,tmp,bs2,ishift;
186887828ca2SBarry Smith   PetscScalar    *aa;
186949b5e25fSSatish Balay   MPI_Comm       comm = ((PetscObject)viewer)->comm;
187049b5e25fSSatish Balay 
187149b5e25fSSatish Balay   PetscFunctionBegin;
1872b0a32e0cSBarry Smith   ierr = PetscOptionsGetInt(PETSC_NULL,"-matload_block_size",&bs,PETSC_NULL);CHKERRQ(ierr);
187349b5e25fSSatish Balay   bs2  = bs*bs;
187449b5e25fSSatish Balay 
187549b5e25fSSatish Balay   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
187629bbc08cSBarry Smith   if (size > 1) SETERRQ(PETSC_ERR_ARG_WRONG,"view must have one processor");
1877b0a32e0cSBarry Smith   ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr);
187849b5e25fSSatish Balay   ierr = PetscBinaryRead(fd,header,4,PETSC_INT);CHKERRQ(ierr);
1879552e946dSBarry Smith   if (header[0] != MAT_FILE_COOKIE) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"not Mat object");
188049b5e25fSSatish Balay   M = header[1]; N = header[2]; nz = header[3];
188149b5e25fSSatish Balay 
188249b5e25fSSatish Balay   if (header[3] < 0) {
188329bbc08cSBarry Smith     SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"Matrix stored in special format, cannot load as SeqSBAIJ");
188449b5e25fSSatish Balay   }
188549b5e25fSSatish Balay 
188629bbc08cSBarry Smith   if (M != N) SETERRQ(PETSC_ERR_SUP,"Can only do square matrices");
188749b5e25fSSatish Balay 
188849b5e25fSSatish Balay   /*
188949b5e25fSSatish Balay      This code adds extra rows to make sure the number of rows is
189049b5e25fSSatish Balay     divisible by the blocksize
189149b5e25fSSatish Balay   */
189249b5e25fSSatish Balay   mbs        = M/bs;
189349b5e25fSSatish Balay   extra_rows = bs - M + bs*(mbs);
189449b5e25fSSatish Balay   if (extra_rows == bs) extra_rows = 0;
189549b5e25fSSatish Balay   else                  mbs++;
189649b5e25fSSatish Balay   if (extra_rows) {
1897ae15b995SBarry Smith     ierr = PetscInfo(0,"Padding loaded matrix to match blocksize\n");CHKERRQ(ierr);
189849b5e25fSSatish Balay   }
189949b5e25fSSatish Balay 
190049b5e25fSSatish Balay   /* read in row lengths */
190113f74950SBarry Smith   ierr = PetscMalloc((M+extra_rows)*sizeof(PetscInt),&rowlengths);CHKERRQ(ierr);
190249b5e25fSSatish Balay   ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr);
190349b5e25fSSatish Balay   for (i=0; i<extra_rows; i++) rowlengths[M+i] = 1;
190449b5e25fSSatish Balay 
190549b5e25fSSatish Balay   /* read in column indices */
190613f74950SBarry Smith   ierr = PetscMalloc((nz+extra_rows)*sizeof(PetscInt),&jj);CHKERRQ(ierr);
190749b5e25fSSatish Balay   ierr = PetscBinaryRead(fd,jj,nz,PETSC_INT);CHKERRQ(ierr);
190849b5e25fSSatish Balay   for (i=0; i<extra_rows; i++) jj[nz+i] = M+i;
190949b5e25fSSatish Balay 
191049b5e25fSSatish Balay   /* loop over row lengths determining block row lengths */
191113f74950SBarry Smith   ierr     = PetscMalloc(mbs*sizeof(PetscInt),&s_browlengths);CHKERRQ(ierr);
191213f74950SBarry Smith   ierr     = PetscMemzero(s_browlengths,mbs*sizeof(PetscInt));CHKERRQ(ierr);
191313f74950SBarry Smith   ierr     = PetscMalloc(2*mbs*sizeof(PetscInt),&mask);CHKERRQ(ierr);
191413f74950SBarry Smith   ierr     = PetscMemzero(mask,mbs*sizeof(PetscInt));CHKERRQ(ierr);
191549b5e25fSSatish Balay   masked   = mask + mbs;
191649b5e25fSSatish Balay   rowcount = 0; nzcount = 0;
191749b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
191849b5e25fSSatish Balay     nmask = 0;
191949b5e25fSSatish Balay     for (j=0; j<bs; j++) {
192049b5e25fSSatish Balay       kmax = rowlengths[rowcount];
192149b5e25fSSatish Balay       for (k=0; k<kmax; k++) {
19222d703238SHong Zhang         tmp = jj[nzcount++]/bs;   /* block col. index */
192303630b6eSHong Zhang         if (!mask[tmp] && tmp >= i) {masked[nmask++] = tmp; mask[tmp] = 1;}
192449b5e25fSSatish Balay       }
192549b5e25fSSatish Balay       rowcount++;
192649b5e25fSSatish Balay     }
1927574b2666SHong Zhang     s_browlengths[i] += nmask;
1928574b2666SHong Zhang 
192949b5e25fSSatish Balay     /* zero out the mask elements we set */
193049b5e25fSSatish Balay     for (j=0; j<nmask; j++) mask[masked[j]] = 0;
193149b5e25fSSatish Balay   }
193249b5e25fSSatish Balay 
193349b5e25fSSatish Balay   /* create our matrix */
1934f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,&B);CHKERRQ(ierr);
1935f69a0ea3SMatthew Knepley   ierr = MatSetSizes(B,M+extra_rows,N+extra_rows,M+extra_rows,N+extra_rows);CHKERRQ(ierr);
19369abb65ffSKris Buschelman   ierr = MatSetType(B,type);CHKERRQ(ierr);
1937ab93d7beSBarry Smith   ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(B,bs,0,s_browlengths);CHKERRQ(ierr);
193849b5e25fSSatish Balay   a = (Mat_SeqSBAIJ*)B->data;
193949b5e25fSSatish Balay 
194049b5e25fSSatish Balay   /* set matrix "i" values */
194149b5e25fSSatish Balay   a->i[0] = 0;
194249b5e25fSSatish Balay   for (i=1; i<= mbs; i++) {
1943574b2666SHong Zhang     a->i[i]      = a->i[i-1] + s_browlengths[i-1];
1944574b2666SHong Zhang     a->ilen[i-1] = s_browlengths[i-1];
194549b5e25fSSatish Balay   }
19466c6c5352SBarry Smith   a->nz = a->i[mbs];
194749b5e25fSSatish Balay 
194849b5e25fSSatish Balay   /* read in nonzero values */
194987828ca2SBarry Smith   ierr = PetscMalloc((nz+extra_rows)*sizeof(PetscScalar),&aa);CHKERRQ(ierr);
195049b5e25fSSatish Balay   ierr = PetscBinaryRead(fd,aa,nz,PETSC_SCALAR);CHKERRQ(ierr);
195149b5e25fSSatish Balay   for (i=0; i<extra_rows; i++) aa[nz+i] = 1.0;
195249b5e25fSSatish Balay 
195349b5e25fSSatish Balay   /* set "a" and "j" values into matrix */
195449b5e25fSSatish Balay   nzcount = 0; jcount = 0;
195549b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
195649b5e25fSSatish Balay     nzcountb = nzcount;
195749b5e25fSSatish Balay     nmask    = 0;
195849b5e25fSSatish Balay     for (j=0; j<bs; j++) {
195949b5e25fSSatish Balay       kmax = rowlengths[i*bs+j];
196049b5e25fSSatish Balay       for (k=0; k<kmax; k++) {
19612d703238SHong Zhang         tmp = jj[nzcount++]/bs; /* block col. index */
196203630b6eSHong Zhang         if (!mask[tmp] && tmp >= i) { masked[nmask++] = tmp; mask[tmp] = 1;}
19632d703238SHong Zhang       }
19642d703238SHong Zhang     }
19652d703238SHong Zhang     /* sort the masked values */
19662d703238SHong Zhang     ierr = PetscSortInt(nmask,masked);CHKERRQ(ierr);
19672d703238SHong Zhang 
19682d703238SHong Zhang     /* set "j" values into matrix */
19692d703238SHong Zhang     maskcount = 1;
19702d703238SHong Zhang     for (j=0; j<nmask; j++) {
197149b5e25fSSatish Balay       a->j[jcount++]  = masked[j];
197249b5e25fSSatish Balay       mask[masked[j]] = maskcount++;
197349b5e25fSSatish Balay     }
1974574b2666SHong Zhang 
197549b5e25fSSatish Balay     /* set "a" values into matrix */
197649b5e25fSSatish Balay     ishift = bs2*a->i[i];
197749b5e25fSSatish Balay     for (j=0; j<bs; j++) {
197849b5e25fSSatish Balay       kmax = rowlengths[i*bs+j];
197949b5e25fSSatish Balay       for (k=0; k<kmax; k++) {
1980574b2666SHong Zhang         tmp       = jj[nzcountb]/bs ; /* block col. index */
1981574b2666SHong Zhang         if (tmp >= i){
198249b5e25fSSatish Balay           block     = mask[tmp] - 1;
198349b5e25fSSatish Balay           point     = jj[nzcountb] - bs*tmp;
198449b5e25fSSatish Balay           idx       = ishift + bs2*block + j + bs*point;
1985574b2666SHong Zhang           a->a[idx] = aa[nzcountb];
1986574b2666SHong Zhang         }
1987574b2666SHong Zhang         nzcountb++;
198849b5e25fSSatish Balay       }
198949b5e25fSSatish Balay     }
199049b5e25fSSatish Balay     /* zero out the mask elements we set */
199149b5e25fSSatish Balay     for (j=0; j<nmask; j++) mask[masked[j]] = 0;
199249b5e25fSSatish Balay   }
19936c6c5352SBarry Smith   if (jcount != a->nz) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"Bad binary matrix");
199449b5e25fSSatish Balay 
199549b5e25fSSatish Balay   ierr = PetscFree(rowlengths);CHKERRQ(ierr);
1996574b2666SHong Zhang   ierr = PetscFree(s_browlengths);CHKERRQ(ierr);
199749b5e25fSSatish Balay   ierr = PetscFree(aa);CHKERRQ(ierr);
199849b5e25fSSatish Balay   ierr = PetscFree(jj);CHKERRQ(ierr);
199949b5e25fSSatish Balay   ierr = PetscFree(mask);CHKERRQ(ierr);
200049b5e25fSSatish Balay 
20019abb65ffSKris Buschelman   ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
20029abb65ffSKris Buschelman   ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
200349b5e25fSSatish Balay   ierr = MatView_Private(B);CHKERRQ(ierr);
20049abb65ffSKris Buschelman   *A = B;
200549b5e25fSSatish Balay   PetscFunctionReturn(0);
200649b5e25fSSatish Balay }
2007574b2666SHong Zhang 
2008d06b337dSHong Zhang #undef __FUNCT__
2009d06b337dSHong Zhang #define __FUNCT__ "MatRelax_SeqSBAIJ"
201013f74950SBarry Smith PetscErrorCode MatRelax_SeqSBAIJ(Mat A,Vec bb,PetscReal omega,MatSORType flag,PetscReal fshift,PetscInt its,PetscInt lits,Vec xx)
2011d06b337dSHong Zhang {
2012d06b337dSHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
2013d06b337dSHong Zhang   MatScalar      *aa=a->a,*v,*v1;
2014d06b337dSHong Zhang   PetscScalar    *x,*b,*t,sum,d;
20156849ba73SBarry Smith   PetscErrorCode ierr;
2016899cda47SBarry Smith   PetscInt       m=a->mbs,bs=A->rmap.bs,*ai=a->i,*aj=a->j;
2017521d7252SBarry Smith   PetscInt       nz,nz1,*vj,*vj1,i;
2018d06b337dSHong Zhang 
2019d06b337dSHong Zhang   PetscFunctionBegin;
202051f519a2SBarry Smith   if (flag & SOR_EISENSTAT) SETERRQ(PETSC_ERR_SUP,"No support yet for Eisenstat");
202151f519a2SBarry Smith 
2022b965ef7fSBarry Smith   its = its*lits;
202377431f27SBarry Smith   if (its <= 0) SETERRQ2(PETSC_ERR_ARG_WRONG,"Relaxation requires global its %D and local its %D both positive",its,lits);
2024d06b337dSHong Zhang 
2025d06b337dSHong Zhang   if (bs > 1)
2026d06b337dSHong Zhang     SETERRQ(PETSC_ERR_SUP,"SSOR for block size > 1 is not yet implemented");
2027d06b337dSHong Zhang 
20281ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
2029d06b337dSHong Zhang   if (xx != bb) {
20301ebc52fbSHong Zhang     ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
2031d06b337dSHong Zhang   } else {
2032d06b337dSHong Zhang     b = x;
2033d06b337dSHong Zhang   }
2034d06b337dSHong Zhang 
2035e2ee2a47SBarry Smith   if (!a->relax_work) {
2036e2ee2a47SBarry Smith     ierr = PetscMalloc(m*sizeof(PetscScalar),&a->relax_work);CHKERRQ(ierr);
2037e2ee2a47SBarry Smith   }
2038e2ee2a47SBarry Smith   t = a->relax_work;
2039d06b337dSHong Zhang 
2040d06b337dSHong Zhang   if (flag & SOR_ZERO_INITIAL_GUESS) {
20412798e883SHong Zhang     if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){
2042290bbb0aSBarry Smith       ierr = PetscMemcpy(t,b,m*sizeof(PetscScalar));CHKERRQ(ierr);
2043d06b337dSHong Zhang 
2044d06b337dSHong Zhang       for (i=0; i<m; i++){
204544706875SHong Zhang         d  = *(aa + ai[i]);  /* diag[i] */
2046d06b337dSHong Zhang         v  = aa + ai[i] + 1;
2047d06b337dSHong Zhang         vj = aj + ai[i] + 1;
2048d06b337dSHong Zhang         nz = ai[i+1] - ai[i] - 1;
2049e6b907acSBarry Smith         ierr = PetscLogFlops(2*nz-1);CHKERRQ(ierr);
2050d06b337dSHong Zhang         x[i] = omega*t[i]/d;
2051d06b337dSHong Zhang         while (nz--) t[*vj++] -= x[i]*(*v++); /* update rhs */
2052d06b337dSHong Zhang       }
2053d06b337dSHong Zhang     }
2054d06b337dSHong Zhang 
20552798e883SHong Zhang     if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){
205695d750ceSBarry Smith       if (!(flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP)){
205795d750ceSBarry Smith         t = b;
2058d06b337dSHong Zhang       }
205995d750ceSBarry Smith 
2060d06b337dSHong Zhang       for (i=m-1; i>=0; i--){
2061d06b337dSHong Zhang         d  = *(aa + ai[i]);
2062d06b337dSHong Zhang         v  = aa + ai[i] + 1;
2063d06b337dSHong Zhang         vj = aj + ai[i] + 1;
2064d06b337dSHong Zhang         nz = ai[i+1] - ai[i] - 1;
2065e6b907acSBarry Smith         ierr = PetscLogFlops(2*nz-1);CHKERRQ(ierr);
2066d06b337dSHong Zhang         sum = t[i];
2067d06b337dSHong Zhang         while (nz--) sum -= x[*vj++]*(*v++);
2068d06b337dSHong Zhang         x[i] =   (1-omega)*x[i] + omega*sum/d;
2069d06b337dSHong Zhang       }
207095d750ceSBarry Smith       t = a->relax_work;
2071d06b337dSHong Zhang     }
2072d06b337dSHong Zhang     its--;
2073d06b337dSHong Zhang   }
2074d06b337dSHong Zhang 
2075d06b337dSHong Zhang   while (its--) {
207644706875SHong Zhang     /*
207744706875SHong Zhang        forward sweep:
207844706875SHong Zhang        for i=0,...,m-1:
207944706875SHong Zhang          sum[i] = (b[i] - U(i,:)x )/d[i];
208044706875SHong Zhang          x[i]   = (1-omega)x[i] + omega*sum[i];
208144706875SHong Zhang          b      = b - x[i]*U^T(i,:);
2082d06b337dSHong Zhang 
208344706875SHong Zhang     */
20842798e883SHong Zhang     if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){
2085290bbb0aSBarry Smith       ierr = PetscMemcpy(t,b,m*sizeof(PetscScalar));CHKERRQ(ierr);
2086d06b337dSHong Zhang 
2087d06b337dSHong Zhang       for (i=0; i<m; i++){
208844706875SHong Zhang         d  = *(aa + ai[i]);  /* diag[i] */
2089d06b337dSHong Zhang         v  = aa + ai[i] + 1; v1=v;
2090d06b337dSHong Zhang         vj = aj + ai[i] + 1; vj1=vj;
2091d06b337dSHong Zhang         nz = ai[i+1] - ai[i] - 1; nz1=nz;
2092d06b337dSHong Zhang         sum = t[i];
2093e6b907acSBarry Smith         ierr = PetscLogFlops(4*nz-2);CHKERRQ(ierr);
2094d06b337dSHong Zhang         while (nz1--) sum -= (*v1++)*x[*vj1++];
2095d06b337dSHong Zhang         x[i] = (1-omega)*x[i] + omega*sum/d;
2096d06b337dSHong Zhang         while (nz--) t[*vj++] -= x[i]*(*v++);
2097d06b337dSHong Zhang       }
2098d06b337dSHong Zhang     }
2099d06b337dSHong Zhang 
21002798e883SHong Zhang     if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){
210144706875SHong Zhang       /*
210244706875SHong Zhang        backward sweep:
210344706875SHong Zhang        b = b - x[i]*U^T(i,:), i=0,...,n-2
210444706875SHong Zhang        for i=m-1,...,0:
210544706875SHong Zhang          sum[i] = (b[i] - U(i,:)x )/d[i];
210644706875SHong Zhang          x[i]   = (1-omega)x[i] + omega*sum[i];
210744706875SHong Zhang       */
210895d750ceSBarry Smith       /* if there was a forward sweep done above then I thing the next two for loops are not needed */
2109290bbb0aSBarry Smith       ierr = PetscMemcpy(t,b,m*sizeof(PetscScalar));CHKERRQ(ierr);
2110d06b337dSHong Zhang 
2111d06b337dSHong Zhang       for (i=0; i<m-1; i++){  /* update rhs */
2112d06b337dSHong Zhang         v  = aa + ai[i] + 1;
2113d06b337dSHong Zhang         vj = aj + ai[i] + 1;
2114d06b337dSHong Zhang         nz = ai[i+1] - ai[i] - 1;
2115efee365bSSatish Balay         ierr = PetscLogFlops(2*nz-1);CHKERRQ(ierr);
2116e6b907acSBarry Smith         while (nz--) t[*vj++] -= x[i]*(*v++);
2117d06b337dSHong Zhang       }
2118d06b337dSHong Zhang       for (i=m-1; i>=0; i--){
2119d06b337dSHong Zhang         d  = *(aa + ai[i]);
2120d06b337dSHong Zhang         v  = aa + ai[i] + 1;
2121d06b337dSHong Zhang         vj = aj + ai[i] + 1;
2122d06b337dSHong Zhang         nz = ai[i+1] - ai[i] - 1;
2123e6b907acSBarry Smith         ierr = PetscLogFlops(2*nz-1);CHKERRQ(ierr);
2124d06b337dSHong Zhang         sum = t[i];
2125d06b337dSHong Zhang         while (nz--) sum -= x[*vj++]*(*v++);
2126d06b337dSHong Zhang         x[i] =   (1-omega)*x[i] + omega*sum/d;
2127d06b337dSHong Zhang       }
2128d06b337dSHong Zhang     }
2129d06b337dSHong Zhang   }
2130d06b337dSHong Zhang 
21311ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
2132d06b337dSHong Zhang   if (bb != xx) {
21331ebc52fbSHong Zhang     ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
2134d06b337dSHong Zhang   }
2135d06b337dSHong Zhang   PetscFunctionReturn(0);
2136d06b337dSHong Zhang }
2137d06b337dSHong Zhang 
2138c75a6043SHong Zhang #undef __FUNCT__
2139c75a6043SHong Zhang #define __FUNCT__ "MatCreateSeqSBAIJWithArrays"
2140c75a6043SHong Zhang /*@
2141c75a6043SHong Zhang      MatCreateSeqSBAIJWithArrays - Creates an sequential SBAIJ matrix using matrix elements
2142c75a6043SHong Zhang               (upper triangular entries in CSR format) provided by the user.
2143c75a6043SHong Zhang 
2144c75a6043SHong Zhang      Collective on MPI_Comm
2145c75a6043SHong Zhang 
2146c75a6043SHong Zhang    Input Parameters:
2147c75a6043SHong Zhang +  comm - must be an MPI communicator of size 1
2148c75a6043SHong Zhang .  bs - size of block
2149c75a6043SHong Zhang .  m - number of rows
2150c75a6043SHong Zhang .  n - number of columns
2151c75a6043SHong Zhang .  i - row indices
2152c75a6043SHong Zhang .  j - column indices
2153c75a6043SHong Zhang -  a - matrix values
2154c75a6043SHong Zhang 
2155c75a6043SHong Zhang    Output Parameter:
2156c75a6043SHong Zhang .  mat - the matrix
2157c75a6043SHong Zhang 
2158c75a6043SHong Zhang    Level: intermediate
2159c75a6043SHong Zhang 
2160c75a6043SHong Zhang    Notes:
2161c75a6043SHong Zhang        The i, j, and a arrays are not copied by this routine, the user must free these arrays
2162c75a6043SHong Zhang     once the matrix is destroyed
2163c75a6043SHong Zhang 
2164c75a6043SHong Zhang        You cannot set new nonzero locations into this matrix, that will generate an error.
2165c75a6043SHong Zhang 
2166c75a6043SHong Zhang        The i and j indices are 0 based
2167c75a6043SHong Zhang 
2168c75a6043SHong Zhang .seealso: MatCreate(), MatCreateMPISBAIJ(), MatCreateSeqSBAIJ()
2169c75a6043SHong Zhang 
2170c75a6043SHong Zhang @*/
2171c75a6043SHong Zhang PetscErrorCode PETSCMAT_DLLEXPORT MatCreateSeqSBAIJWithArrays(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt* i,PetscInt*j,PetscScalar *a,Mat *mat)
2172c75a6043SHong Zhang {
2173c75a6043SHong Zhang   PetscErrorCode ierr;
2174c75a6043SHong Zhang   PetscInt       ii;
2175c75a6043SHong Zhang   Mat_SeqSBAIJ   *sbaij;
2176c75a6043SHong Zhang 
2177c75a6043SHong Zhang   PetscFunctionBegin;
2178c75a6043SHong Zhang   if (bs != 1) SETERRQ1(PETSC_ERR_SUP,"block size %D > 1 is not supported yet",bs);
2179c75a6043SHong Zhang   if (i[0]) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"i (row indices) must start with 0");
2180c75a6043SHong Zhang 
2181c75a6043SHong Zhang   ierr = MatCreate(comm,mat);CHKERRQ(ierr);
2182c75a6043SHong Zhang   ierr = MatSetSizes(*mat,m,n,m,n);CHKERRQ(ierr);
2183c75a6043SHong Zhang   ierr = MatSetType(*mat,MATSEQSBAIJ);CHKERRQ(ierr);
2184c75a6043SHong Zhang   ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(*mat,bs,MAT_SKIP_ALLOCATION,0);CHKERRQ(ierr);
2185c75a6043SHong Zhang   sbaij = (Mat_SeqSBAIJ*)(*mat)->data;
2186c75a6043SHong Zhang   ierr = PetscMalloc2(m,PetscInt,&sbaij->imax,m,PetscInt,&sbaij->ilen);CHKERRQ(ierr);
2187c75a6043SHong Zhang 
2188c75a6043SHong Zhang   sbaij->i = i;
2189c75a6043SHong Zhang   sbaij->j = j;
2190c75a6043SHong Zhang   sbaij->a = a;
2191c75a6043SHong Zhang   sbaij->singlemalloc = PETSC_FALSE;
2192c75a6043SHong Zhang   sbaij->nonew        = -1;             /*this indicates that inserting a new value in the matrix that generates a new nonzero is an error*/
2193e6b907acSBarry Smith   sbaij->free_a       = PETSC_FALSE;
2194e6b907acSBarry Smith   sbaij->free_ij      = PETSC_FALSE;
2195c75a6043SHong Zhang 
2196c75a6043SHong Zhang   for (ii=0; ii<m; ii++) {
2197c75a6043SHong Zhang     sbaij->ilen[ii] = sbaij->imax[ii] = i[ii+1] - i[ii];
2198c75a6043SHong Zhang #if defined(PETSC_USE_DEBUG)
2199c75a6043SHong Zhang     if (i[ii+1] - i[ii] < 0) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Negative row length in i (row indices) row = %d length = %d",ii,i[ii+1] - i[ii]);
2200c75a6043SHong Zhang #endif
2201c75a6043SHong Zhang   }
2202c75a6043SHong Zhang #if defined(PETSC_USE_DEBUG)
2203c75a6043SHong Zhang   for (ii=0; ii<sbaij->i[m]; ii++) {
2204c75a6043SHong Zhang     if (j[ii] < 0) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Negative column index at location = %d index = %d",ii,j[ii]);
2205c75a6043SHong Zhang     if (j[ii] > n - 1) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Column index to large at location = %d index = %d",ii,j[ii]);
2206c75a6043SHong Zhang   }
2207c75a6043SHong Zhang #endif
2208c75a6043SHong Zhang 
2209c75a6043SHong Zhang   ierr = MatAssemblyBegin(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2210c75a6043SHong Zhang   ierr = MatAssemblyEnd(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2211c75a6043SHong Zhang   PetscFunctionReturn(0);
2212c75a6043SHong Zhang }
2213d06b337dSHong Zhang 
2214d06b337dSHong Zhang 
2215d06b337dSHong Zhang 
221649b5e25fSSatish Balay 
221749b5e25fSSatish Balay 
2218