xref: /petsc/src/mat/impls/sbaij/seq/sbaij.c (revision 290bbb0a1dcfb34dbf94efcfcc44171581b0efea)
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 
117901853e0SKris Buschelman   ierr = PetscObjectComposeFunction((PetscObject)A,"MatStoreValues_C","",PETSC_NULL);CHKERRQ(ierr);
118901853e0SKris Buschelman   ierr = PetscObjectComposeFunction((PetscObject)A,"MatRetrieveValues_C","",PETSC_NULL);CHKERRQ(ierr);
119901853e0SKris Buschelman   ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetColumnIndices_C","",PETSC_NULL);CHKERRQ(ierr);
120901853e0SKris Buschelman   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqaij_C","",PETSC_NULL);CHKERRQ(ierr);
121901853e0SKris Buschelman   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqbaij_C","",PETSC_NULL);CHKERRQ(ierr);
122901853e0SKris Buschelman   ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetPreallocation_C","",PETSC_NULL);CHKERRQ(ierr);
12349b5e25fSSatish Balay   PetscFunctionReturn(0);
12449b5e25fSSatish Balay }
12549b5e25fSSatish Balay 
1264a2ae208SSatish Balay #undef __FUNCT__
1274a2ae208SSatish Balay #define __FUNCT__ "MatSetOption_SeqSBAIJ"
128dfbe8321SBarry Smith PetscErrorCode MatSetOption_SeqSBAIJ(Mat A,MatOption op)
12949b5e25fSSatish Balay {
130045c9aa0SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
13163ba0a88SBarry Smith   PetscErrorCode ierr;
13249b5e25fSSatish Balay 
13349b5e25fSSatish Balay   PetscFunctionBegin;
1344d9d31abSKris Buschelman   switch (op) {
1354d9d31abSKris Buschelman   case MAT_ROW_ORIENTED:
1364d9d31abSKris Buschelman     a->roworiented = PETSC_TRUE;
1374d9d31abSKris Buschelman     break;
1384d9d31abSKris Buschelman   case MAT_COLUMN_ORIENTED:
1394d9d31abSKris Buschelman     a->roworiented = PETSC_FALSE;
1404d9d31abSKris Buschelman     break;
1414d9d31abSKris Buschelman   case MAT_COLUMNS_SORTED:
1424d9d31abSKris Buschelman     a->sorted = PETSC_TRUE;
1434d9d31abSKris Buschelman     break;
1444d9d31abSKris Buschelman   case MAT_COLUMNS_UNSORTED:
1454d9d31abSKris Buschelman     a->sorted = PETSC_FALSE;
1464d9d31abSKris Buschelman     break;
1474d9d31abSKris Buschelman   case MAT_KEEP_ZEROED_ROWS:
1484d9d31abSKris Buschelman     a->keepzeroedrows = PETSC_TRUE;
1494d9d31abSKris Buschelman     break;
1504d9d31abSKris Buschelman   case MAT_NO_NEW_NONZERO_LOCATIONS:
1514d9d31abSKris Buschelman     a->nonew = 1;
1524d9d31abSKris Buschelman     break;
1534d9d31abSKris Buschelman   case MAT_NEW_NONZERO_LOCATION_ERR:
1544d9d31abSKris Buschelman     a->nonew = -1;
1554d9d31abSKris Buschelman     break;
1564d9d31abSKris Buschelman   case MAT_NEW_NONZERO_ALLOCATION_ERR:
1574d9d31abSKris Buschelman     a->nonew = -2;
1584d9d31abSKris Buschelman     break;
1594d9d31abSKris Buschelman   case MAT_YES_NEW_NONZERO_LOCATIONS:
1604d9d31abSKris Buschelman     a->nonew = 0;
1614d9d31abSKris Buschelman     break;
1624d9d31abSKris Buschelman   case MAT_ROWS_SORTED:
1634d9d31abSKris Buschelman   case MAT_ROWS_UNSORTED:
1644d9d31abSKris Buschelman   case MAT_YES_NEW_DIAGONALS:
1654d9d31abSKris Buschelman   case MAT_IGNORE_OFF_PROC_ENTRIES:
1664d9d31abSKris Buschelman   case MAT_USE_HASH_TABLE:
167*290bbb0aSBarry Smith     ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr);
1684d9d31abSKris Buschelman     break;
1694d9d31abSKris Buschelman   case MAT_NO_NEW_DIAGONALS:
17029bbc08cSBarry Smith     SETERRQ(PETSC_ERR_SUP,"MAT_NO_NEW_DIAGONALS");
1719a4540c5SBarry Smith   case MAT_NOT_SYMMETRIC:
1729a4540c5SBarry Smith   case MAT_NOT_STRUCTURALLY_SYMMETRIC:
1739a4540c5SBarry Smith   case MAT_HERMITIAN:
1749a4540c5SBarry Smith     SETERRQ(PETSC_ERR_SUP,"Matrix must be symmetric");
17577e54ba9SKris Buschelman   case MAT_SYMMETRIC:
17677e54ba9SKris Buschelman   case MAT_STRUCTURALLY_SYMMETRIC:
1779a4540c5SBarry Smith   case MAT_NOT_HERMITIAN:
1789a4540c5SBarry Smith   case MAT_SYMMETRY_ETERNAL:
1799a4540c5SBarry Smith   case MAT_NOT_SYMMETRY_ETERNAL:
180*290bbb0aSBarry Smith     ierr = PetscInfo1(A,"Option %s not relevent\n",MatOptions[op]);CHKERRQ(ierr);
181*290bbb0aSBarry Smith     break;
182941593c8SHong Zhang   case MAT_IGNORE_LOWER_TRIANGULAR:
183941593c8SHong Zhang     a->ignore_ltriangular = PETSC_TRUE;
184941593c8SHong Zhang     break;
185941593c8SHong Zhang   case MAT_ERROR_LOWER_TRIANGULAR:
186941593c8SHong Zhang     a->ignore_ltriangular = PETSC_FALSE;
18777e54ba9SKris Buschelman     break;
188f5edf698SHong Zhang   case MAT_GETROW_UPPERTRIANGULAR:
189f5edf698SHong Zhang     a->getrow_utriangular = PETSC_TRUE;
190f5edf698SHong Zhang     break;
1914d9d31abSKris Buschelman   default:
192ad86a440SBarry Smith     SETERRQ1(PETSC_ERR_SUP,"unknown option %d",op);
19349b5e25fSSatish Balay   }
19449b5e25fSSatish Balay   PetscFunctionReturn(0);
19549b5e25fSSatish Balay }
19649b5e25fSSatish Balay 
1974a2ae208SSatish Balay #undef __FUNCT__
1984a2ae208SSatish Balay #define __FUNCT__ "MatGetRow_SeqSBAIJ"
19913f74950SBarry Smith PetscErrorCode MatGetRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *ncols,PetscInt **cols,PetscScalar **v)
20049b5e25fSSatish Balay {
20149b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
2026849ba73SBarry Smith   PetscErrorCode ierr;
20313f74950SBarry Smith   PetscInt       itmp,i,j,k,M,*ai,*aj,bs,bn,bp,*cols_i,bs2;
20449b5e25fSSatish Balay   MatScalar      *aa,*aa_i;
20587828ca2SBarry Smith   PetscScalar    *v_i;
20649b5e25fSSatish Balay 
20749b5e25fSSatish Balay   PetscFunctionBegin;
208f5edf698SHong 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()");
209f5edf698SHong Zhang   /* Get the upper triangular part of the row */
210899cda47SBarry Smith   bs  = A->rmap.bs;
21149b5e25fSSatish Balay   ai  = a->i;
21249b5e25fSSatish Balay   aj  = a->j;
21349b5e25fSSatish Balay   aa  = a->a;
21449b5e25fSSatish Balay   bs2 = a->bs2;
21549b5e25fSSatish Balay 
216899cda47SBarry Smith   if (row < 0 || row >= A->rmap.N) SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE, "Row %D out of range", row);
21749b5e25fSSatish Balay 
21849b5e25fSSatish Balay   bn  = row/bs;   /* Block number */
21949b5e25fSSatish Balay   bp  = row % bs; /* Block position */
22049b5e25fSSatish Balay   M   = ai[bn+1] - ai[bn];
22149b5e25fSSatish Balay   *ncols = bs*M;
22249b5e25fSSatish Balay 
22349b5e25fSSatish Balay   if (v) {
22449b5e25fSSatish Balay     *v = 0;
22549b5e25fSSatish Balay     if (*ncols) {
22687828ca2SBarry Smith       ierr = PetscMalloc((*ncols+row)*sizeof(PetscScalar),v);CHKERRQ(ierr);
22749b5e25fSSatish Balay       for (i=0; i<M; i++) { /* for each block in the block row */
22849b5e25fSSatish Balay         v_i  = *v + i*bs;
22949b5e25fSSatish Balay         aa_i = aa + bs2*(ai[bn] + i);
23049b5e25fSSatish Balay         for (j=bp,k=0; j<bs2; j+=bs,k++) {v_i[k] = aa_i[j];}
23149b5e25fSSatish Balay       }
23249b5e25fSSatish Balay     }
23349b5e25fSSatish Balay   }
23449b5e25fSSatish Balay 
23549b5e25fSSatish Balay   if (cols) {
23649b5e25fSSatish Balay     *cols = 0;
23749b5e25fSSatish Balay     if (*ncols) {
23813f74950SBarry Smith       ierr = PetscMalloc((*ncols+row)*sizeof(PetscInt),cols);CHKERRQ(ierr);
23949b5e25fSSatish Balay       for (i=0; i<M; i++) { /* for each block in the block row */
24049b5e25fSSatish Balay         cols_i = *cols + i*bs;
24149b5e25fSSatish Balay         itmp  = bs*aj[ai[bn] + i];
24249b5e25fSSatish Balay         for (j=0; j<bs; j++) {cols_i[j] = itmp++;}
24349b5e25fSSatish Balay       }
24449b5e25fSSatish Balay     }
24549b5e25fSSatish Balay   }
24649b5e25fSSatish Balay 
24749b5e25fSSatish Balay   /*search column A(0:row-1,row) (=A(row,0:row-1)). Could be expensive! */
2485ddb2528SHong Zhang   /* this segment is currently removed, so only entries in the upper triangle are obtained */
2495ddb2528SHong Zhang #ifdef column_search
25049b5e25fSSatish Balay   v_i    = *v    + M*bs;
25149b5e25fSSatish Balay   cols_i = *cols + M*bs;
25249b5e25fSSatish Balay   for (i=0; i<bn; i++){ /* for each block row */
25349b5e25fSSatish Balay     M = ai[i+1] - ai[i];
25449b5e25fSSatish Balay     for (j=0; j<M; j++){
25549b5e25fSSatish Balay       itmp = aj[ai[i] + j];    /* block column value */
25649b5e25fSSatish Balay       if (itmp == bn){
25749b5e25fSSatish Balay         aa_i   = aa    + bs2*(ai[i] + j) + bs*bp;
25849b5e25fSSatish Balay         for (k=0; k<bs; k++) {
25949b5e25fSSatish Balay           *cols_i++ = i*bs+k;
26049b5e25fSSatish Balay           *v_i++    = aa_i[k];
26149b5e25fSSatish Balay         }
26249b5e25fSSatish Balay         *ncols += bs;
26349b5e25fSSatish Balay         break;
26449b5e25fSSatish Balay       }
26549b5e25fSSatish Balay     }
26649b5e25fSSatish Balay   }
2675ddb2528SHong Zhang #endif
26849b5e25fSSatish Balay   PetscFunctionReturn(0);
26949b5e25fSSatish Balay }
27049b5e25fSSatish Balay 
2714a2ae208SSatish Balay #undef __FUNCT__
2724a2ae208SSatish Balay #define __FUNCT__ "MatRestoreRow_SeqSBAIJ"
27313f74950SBarry Smith PetscErrorCode MatRestoreRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
27449b5e25fSSatish Balay {
275dfbe8321SBarry Smith   PetscErrorCode ierr;
27649b5e25fSSatish Balay 
27749b5e25fSSatish Balay   PetscFunctionBegin;
27805b42c5fSBarry Smith   if (idx) {ierr = PetscFree(*idx);CHKERRQ(ierr);}
27905b42c5fSBarry Smith   if (v)   {ierr = PetscFree(*v);CHKERRQ(ierr);}
28049b5e25fSSatish Balay   PetscFunctionReturn(0);
28149b5e25fSSatish Balay }
28249b5e25fSSatish Balay 
2834a2ae208SSatish Balay #undef __FUNCT__
284f5edf698SHong Zhang #define __FUNCT__ "MatGetRowUpperTriangular_SeqSBAIJ"
285f5edf698SHong Zhang PetscErrorCode MatGetRowUpperTriangular_SeqSBAIJ(Mat A)
286f5edf698SHong Zhang {
287f5edf698SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
288f5edf698SHong Zhang 
289f5edf698SHong Zhang   PetscFunctionBegin;
290f5edf698SHong Zhang   a->getrow_utriangular = PETSC_TRUE;
291f5edf698SHong Zhang   PetscFunctionReturn(0);
292f5edf698SHong Zhang }
293f5edf698SHong Zhang #undef __FUNCT__
294f5edf698SHong Zhang #define __FUNCT__ "MatRestoreRowUpperTriangular_SeqSBAIJ"
295f5edf698SHong Zhang PetscErrorCode MatRestoreRowUpperTriangular_SeqSBAIJ(Mat A)
296f5edf698SHong Zhang {
297f5edf698SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
298f5edf698SHong Zhang 
299f5edf698SHong Zhang   PetscFunctionBegin;
300f5edf698SHong Zhang   a->getrow_utriangular = PETSC_FALSE;
301f5edf698SHong Zhang   PetscFunctionReturn(0);
302f5edf698SHong Zhang }
303f5edf698SHong Zhang 
304f5edf698SHong Zhang #undef __FUNCT__
3054a2ae208SSatish Balay #define __FUNCT__ "MatTranspose_SeqSBAIJ"
306dfbe8321SBarry Smith PetscErrorCode MatTranspose_SeqSBAIJ(Mat A,Mat *B)
30749b5e25fSSatish Balay {
308dfbe8321SBarry Smith   PetscErrorCode ierr;
30949b5e25fSSatish Balay   PetscFunctionBegin;
310999d9058SBarry Smith   ierr = MatDuplicate(A,MAT_COPY_VALUES,B);CHKERRQ(ierr);
3118115998fSBarry Smith   PetscFunctionReturn(0);
31249b5e25fSSatish Balay }
31349b5e25fSSatish Balay 
3144a2ae208SSatish Balay #undef __FUNCT__
3154a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_ASCII"
3166849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_ASCII(Mat A,PetscViewer viewer)
31749b5e25fSSatish Balay {
31849b5e25fSSatish Balay   Mat_SeqSBAIJ      *a = (Mat_SeqSBAIJ*)A->data;
319dfbe8321SBarry Smith   PetscErrorCode    ierr;
320899cda47SBarry Smith   PetscInt          i,j,bs = A->rmap.bs,k,l,bs2=a->bs2;
3212dcb1b2aSMatthew Knepley   const char        *name;
322f3ef73ceSBarry Smith   PetscViewerFormat format;
32349b5e25fSSatish Balay 
32449b5e25fSSatish Balay   PetscFunctionBegin;
32580fe4e49SBarry Smith   ierr = PetscObjectGetName((PetscObject)A,&name);CHKERRQ(ierr);
326b0a32e0cSBarry Smith   ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
327456192e2SBarry Smith   if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
32877431f27SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  block size is %D\n",bs);CHKERRQ(ierr);
329fb9695e5SSatish Balay   } else if (format == PETSC_VIEWER_ASCII_MATLAB) {
33029bbc08cSBarry Smith     SETERRQ(PETSC_ERR_SUP,"Matlab format not supported");
331fb9695e5SSatish Balay   } else if (format == PETSC_VIEWER_ASCII_COMMON) {
332b0a32e0cSBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_NO);CHKERRQ(ierr);
33349b5e25fSSatish Balay     for (i=0; i<a->mbs; i++) {
33449b5e25fSSatish Balay       for (j=0; j<bs; j++) {
33577431f27SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr);
33649b5e25fSSatish Balay         for (k=a->i[i]; k<a->i[i+1]; k++) {
33749b5e25fSSatish Balay           for (l=0; l<bs; l++) {
33849b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX)
33949b5e25fSSatish Balay             if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
340a83599f4SBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %G + %G i) ",bs*a->j[k]+l,
34149b5e25fSSatish Balay                                             PetscRealPart(a->a[bs2*k + l*bs + j]),PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
34249b5e25fSSatish Balay             } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
343a83599f4SBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %G - %G i) ",bs*a->j[k]+l,
34449b5e25fSSatish Balay                                             PetscRealPart(a->a[bs2*k + l*bs + j]),-PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
34549b5e25fSSatish Balay             } else if (PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
346a83599f4SBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",bs*a->j[k]+l,PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
34749b5e25fSSatish Balay             }
34849b5e25fSSatish Balay #else
34949b5e25fSSatish Balay             if (a->a[bs2*k + l*bs + j] != 0.0) {
350a83599f4SBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",bs*a->j[k]+l,a->a[bs2*k + l*bs + j]);CHKERRQ(ierr);
35149b5e25fSSatish Balay             }
35249b5e25fSSatish Balay #endif
35349b5e25fSSatish Balay           }
35449b5e25fSSatish Balay         }
355b0a32e0cSBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
35649b5e25fSSatish Balay       }
35749b5e25fSSatish Balay     }
358b0a32e0cSBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_YES);CHKERRQ(ierr);
359c1490034SHong Zhang   } else if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) {
360c1490034SHong Zhang      PetscFunctionReturn(0);
36149b5e25fSSatish Balay   } else {
362b0a32e0cSBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_NO);CHKERRQ(ierr);
36349b5e25fSSatish Balay     for (i=0; i<a->mbs; i++) {
36449b5e25fSSatish Balay       for (j=0; j<bs; j++) {
36577431f27SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr);
36649b5e25fSSatish Balay         for (k=a->i[i]; k<a->i[i+1]; k++) {
36749b5e25fSSatish Balay           for (l=0; l<bs; l++) {
36849b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX)
36949b5e25fSSatish Balay             if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0) {
370a83599f4SBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %G + %G i) ",bs*a->j[k]+l,
37149b5e25fSSatish Balay                                             PetscRealPart(a->a[bs2*k + l*bs + j]),PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
37249b5e25fSSatish Balay             } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0) {
373a83599f4SBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %G - %G i) ",bs*a->j[k]+l,
37449b5e25fSSatish Balay                                             PetscRealPart(a->a[bs2*k + l*bs + j]),-PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
37549b5e25fSSatish Balay             } else {
376a83599f4SBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",bs*a->j[k]+l,PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
37749b5e25fSSatish Balay             }
37849b5e25fSSatish Balay #else
379e9f7bc9eSHong Zhang             ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",bs*a->j[k]+l,a->a[bs2*k + l*bs + j]);CHKERRQ(ierr);
38049b5e25fSSatish Balay #endif
38149b5e25fSSatish Balay           }
38249b5e25fSSatish Balay         }
383b0a32e0cSBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
38449b5e25fSSatish Balay       }
38549b5e25fSSatish Balay     }
386b0a32e0cSBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_YES);CHKERRQ(ierr);
38749b5e25fSSatish Balay   }
388b0a32e0cSBarry Smith   ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
38949b5e25fSSatish Balay   PetscFunctionReturn(0);
39049b5e25fSSatish Balay }
39149b5e25fSSatish Balay 
3924a2ae208SSatish Balay #undef __FUNCT__
3934a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_Draw_Zoom"
3946849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw_Zoom(PetscDraw draw,void *Aa)
39549b5e25fSSatish Balay {
39649b5e25fSSatish Balay   Mat            A = (Mat) Aa;
39749b5e25fSSatish Balay   Mat_SeqSBAIJ   *a=(Mat_SeqSBAIJ*)A->data;
3986849ba73SBarry Smith   PetscErrorCode ierr;
399899cda47SBarry Smith   PetscInt       row,i,j,k,l,mbs=a->mbs,color,bs=A->rmap.bs,bs2=a->bs2;
40013f74950SBarry Smith   PetscMPIInt    rank;
40149b5e25fSSatish Balay   PetscReal      xl,yl,xr,yr,x_l,x_r,y_l,y_r;
40249b5e25fSSatish Balay   MatScalar      *aa;
40349b5e25fSSatish Balay   MPI_Comm       comm;
404b0a32e0cSBarry Smith   PetscViewer    viewer;
40549b5e25fSSatish Balay 
40649b5e25fSSatish Balay   PetscFunctionBegin;
40749b5e25fSSatish Balay   /*
40849b5e25fSSatish Balay     This is nasty. If this is called from an originally parallel matrix
40949b5e25fSSatish Balay     then all processes call this,but only the first has the matrix so the
41049b5e25fSSatish Balay     rest should return immediately.
41149b5e25fSSatish Balay   */
41249b5e25fSSatish Balay   ierr = PetscObjectGetComm((PetscObject)draw,&comm);CHKERRQ(ierr);
41349b5e25fSSatish Balay   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
41449b5e25fSSatish Balay   if (rank) PetscFunctionReturn(0);
41549b5e25fSSatish Balay 
41649b5e25fSSatish Balay   ierr = PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer);CHKERRQ(ierr);
41749b5e25fSSatish Balay 
418b0a32e0cSBarry Smith   ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr);
419b0a32e0cSBarry Smith   PetscDrawString(draw, .3*(xl+xr), .3*(yl+yr), PETSC_DRAW_BLACK, "symmetric");
42049b5e25fSSatish Balay 
42149b5e25fSSatish Balay   /* loop over matrix elements drawing boxes */
422b0a32e0cSBarry Smith   color = PETSC_DRAW_BLUE;
42349b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
42449b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
425899cda47SBarry Smith       y_l = A->rmap.N - row - 1.0; y_r = y_l + 1.0;
42649b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
42749b5e25fSSatish Balay       aa = a->a + j*bs2;
42849b5e25fSSatish Balay       for (k=0; k<bs; k++) {
42949b5e25fSSatish Balay         for (l=0; l<bs; l++) {
43049b5e25fSSatish Balay           if (PetscRealPart(*aa++) >=  0.) continue;
431b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
43249b5e25fSSatish Balay         }
43349b5e25fSSatish Balay       }
43449b5e25fSSatish Balay     }
43549b5e25fSSatish Balay   }
436b0a32e0cSBarry Smith   color = PETSC_DRAW_CYAN;
43749b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
43849b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
439899cda47SBarry Smith       y_l = A->rmap.N - row - 1.0; y_r = y_l + 1.0;
44049b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
44149b5e25fSSatish Balay       aa = a->a + j*bs2;
44249b5e25fSSatish Balay       for (k=0; k<bs; k++) {
44349b5e25fSSatish Balay         for (l=0; l<bs; l++) {
44449b5e25fSSatish Balay           if (PetscRealPart(*aa++) != 0.) continue;
445b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
44649b5e25fSSatish Balay         }
44749b5e25fSSatish Balay       }
44849b5e25fSSatish Balay     }
44949b5e25fSSatish Balay   }
45049b5e25fSSatish Balay 
451b0a32e0cSBarry Smith   color = PETSC_DRAW_RED;
45249b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
45349b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
454899cda47SBarry Smith       y_l = A->rmap.N - row - 1.0; y_r = y_l + 1.0;
45549b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
45649b5e25fSSatish Balay       aa = a->a + j*bs2;
45749b5e25fSSatish Balay       for (k=0; k<bs; k++) {
45849b5e25fSSatish Balay         for (l=0; l<bs; l++) {
45949b5e25fSSatish Balay           if (PetscRealPart(*aa++) <= 0.) continue;
460b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
46149b5e25fSSatish Balay         }
46249b5e25fSSatish Balay       }
46349b5e25fSSatish Balay     }
46449b5e25fSSatish Balay   }
46549b5e25fSSatish Balay   PetscFunctionReturn(0);
46649b5e25fSSatish Balay }
46749b5e25fSSatish Balay 
4684a2ae208SSatish Balay #undef __FUNCT__
4694a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_Draw"
4706849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw(Mat A,PetscViewer viewer)
47149b5e25fSSatish Balay {
472dfbe8321SBarry Smith   PetscErrorCode ierr;
47349b5e25fSSatish Balay   PetscReal      xl,yl,xr,yr,w,h;
474b0a32e0cSBarry Smith   PetscDraw      draw;
47549b5e25fSSatish Balay   PetscTruth     isnull;
47649b5e25fSSatish Balay 
47749b5e25fSSatish Balay   PetscFunctionBegin;
47849b5e25fSSatish Balay 
479b0a32e0cSBarry Smith   ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
480b0a32e0cSBarry Smith   ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr); if (isnull) PetscFunctionReturn(0);
48149b5e25fSSatish Balay 
48249b5e25fSSatish Balay   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer);CHKERRQ(ierr);
483899cda47SBarry Smith   xr  = A->rmap.N; yr = A->rmap.N; h = yr/10.0; w = xr/10.0;
48449b5e25fSSatish Balay   xr += w;    yr += h;  xl = -w;     yl = -h;
485b0a32e0cSBarry Smith   ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr);
486b0a32e0cSBarry Smith   ierr = PetscDrawZoom(draw,MatView_SeqSBAIJ_Draw_Zoom,A);CHKERRQ(ierr);
48749b5e25fSSatish Balay   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",PETSC_NULL);CHKERRQ(ierr);
48849b5e25fSSatish Balay   PetscFunctionReturn(0);
48949b5e25fSSatish Balay }
49049b5e25fSSatish Balay 
4914a2ae208SSatish Balay #undef __FUNCT__
4924a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ"
493dfbe8321SBarry Smith PetscErrorCode MatView_SeqSBAIJ(Mat A,PetscViewer viewer)
49449b5e25fSSatish Balay {
495dfbe8321SBarry Smith   PetscErrorCode ierr;
49632077d6dSBarry Smith   PetscTruth     iascii,isdraw;
49749b5e25fSSatish Balay 
49849b5e25fSSatish Balay   PetscFunctionBegin;
49932077d6dSBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr);
500fb9695e5SSatish Balay   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_DRAW,&isdraw);CHKERRQ(ierr);
50132077d6dSBarry Smith   if (iascii){
50249b5e25fSSatish Balay     ierr = MatView_SeqSBAIJ_ASCII(A,viewer);CHKERRQ(ierr);
50349b5e25fSSatish Balay   } else if (isdraw) {
50449b5e25fSSatish Balay     ierr = MatView_SeqSBAIJ_Draw(A,viewer);CHKERRQ(ierr);
50549b5e25fSSatish Balay   } else {
506a5e6ed63SBarry Smith     Mat B;
507ceb03754SKris Buschelman     ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&B);CHKERRQ(ierr);
508a5e6ed63SBarry Smith     ierr = MatView(B,viewer);CHKERRQ(ierr);
509a5e6ed63SBarry Smith     ierr = MatDestroy(B);CHKERRQ(ierr);
51049b5e25fSSatish Balay   }
51149b5e25fSSatish Balay   PetscFunctionReturn(0);
51249b5e25fSSatish Balay }
51349b5e25fSSatish Balay 
51449b5e25fSSatish Balay 
5154a2ae208SSatish Balay #undef __FUNCT__
5164a2ae208SSatish Balay #define __FUNCT__ "MatGetValues_SeqSBAIJ"
51713f74950SBarry Smith PetscErrorCode MatGetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],PetscScalar v[])
51849b5e25fSSatish Balay {
519045c9aa0SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
52013f74950SBarry Smith   PetscInt     *rp,k,low,high,t,row,nrow,i,col,l,*aj = a->j;
52113f74950SBarry Smith   PetscInt     *ai = a->i,*ailen = a->ilen;
522899cda47SBarry Smith   PetscInt     brow,bcol,ridx,cidx,bs=A->rmap.bs,bs2=a->bs2;
52349b5e25fSSatish Balay   MatScalar    *ap,*aa = a->a,zero = 0.0;
52449b5e25fSSatish Balay 
52549b5e25fSSatish Balay   PetscFunctionBegin;
52649b5e25fSSatish Balay   for (k=0; k<m; k++) { /* loop over rows */
52749b5e25fSSatish Balay     row  = im[k]; brow = row/bs;
52877431f27SBarry Smith     if (row < 0) SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"Negative row: %D",row);
529899cda47SBarry Smith     if (row >= A->rmap.N) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,A->rmap.N-1);
53049b5e25fSSatish Balay     rp   = aj + ai[brow] ; ap = aa + bs2*ai[brow] ;
53149b5e25fSSatish Balay     nrow = ailen[brow];
53249b5e25fSSatish Balay     for (l=0; l<n; l++) { /* loop over columns */
53377431f27SBarry Smith       if (in[l] < 0) SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"Negative column: %D",in[l]);
534899cda47SBarry 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);
53549b5e25fSSatish Balay       col  = in[l] ;
53649b5e25fSSatish Balay       bcol = col/bs;
53749b5e25fSSatish Balay       cidx = col%bs;
53849b5e25fSSatish Balay       ridx = row%bs;
53949b5e25fSSatish Balay       high = nrow;
54049b5e25fSSatish Balay       low  = 0; /* assume unsorted */
54149b5e25fSSatish Balay       while (high-low > 5) {
54249b5e25fSSatish Balay         t = (low+high)/2;
54349b5e25fSSatish Balay         if (rp[t] > bcol) high = t;
54449b5e25fSSatish Balay         else             low  = t;
54549b5e25fSSatish Balay       }
54649b5e25fSSatish Balay       for (i=low; i<high; i++) {
54749b5e25fSSatish Balay         if (rp[i] > bcol) break;
54849b5e25fSSatish Balay         if (rp[i] == bcol) {
54949b5e25fSSatish Balay           *v++ = ap[bs2*i+bs*cidx+ridx];
55049b5e25fSSatish Balay           goto finished;
55149b5e25fSSatish Balay         }
55249b5e25fSSatish Balay       }
55349b5e25fSSatish Balay       *v++ = zero;
55449b5e25fSSatish Balay        finished:;
55549b5e25fSSatish Balay     }
55649b5e25fSSatish Balay   }
55749b5e25fSSatish Balay   PetscFunctionReturn(0);
55849b5e25fSSatish Balay }
55949b5e25fSSatish Balay 
56049b5e25fSSatish Balay 
5614a2ae208SSatish Balay #undef __FUNCT__
5624a2ae208SSatish Balay #define __FUNCT__ "MatSetValuesBlocked_SeqSBAIJ"
56313f74950SBarry Smith PetscErrorCode MatSetValuesBlocked_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is)
56449b5e25fSSatish Balay {
5650880e062SHong Zhang   Mat_SeqSBAIJ    *a = (Mat_SeqSBAIJ*)A->data;
5666849ba73SBarry Smith   PetscErrorCode  ierr;
567e2ee6c50SBarry Smith   PetscInt        *rp,k,low,high,t,ii,jj,row,nrow,i,col,l,rmax,N,lastcol = -1;
56813f74950SBarry Smith   PetscInt        *imax=a->imax,*ai=a->i,*ailen=a->ilen;
569899cda47SBarry Smith   PetscInt        *aj=a->j,nonew=a->nonew,bs2=a->bs2,bs=A->rmap.bs,stepval;
5700880e062SHong Zhang   PetscTruth      roworiented=a->roworiented;
571f15d580aSBarry Smith   const MatScalar *value = v;
572f15d580aSBarry Smith   MatScalar       *ap,*aa = a->a,*bap;
5730880e062SHong Zhang 
57449b5e25fSSatish Balay   PetscFunctionBegin;
5750880e062SHong Zhang   if (roworiented) {
5760880e062SHong Zhang     stepval = (n-1)*bs;
5770880e062SHong Zhang   } else {
5780880e062SHong Zhang     stepval = (m-1)*bs;
5790880e062SHong Zhang   }
5800880e062SHong Zhang   for (k=0; k<m; k++) { /* loop over added rows */
5810880e062SHong Zhang     row  = im[k];
5820880e062SHong Zhang     if (row < 0) continue;
5832515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
58477431f27SBarry Smith     if (row >= a->mbs) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,a->mbs-1);
5850880e062SHong Zhang #endif
5860880e062SHong Zhang     rp   = aj + ai[row];
5870880e062SHong Zhang     ap   = aa + bs2*ai[row];
5880880e062SHong Zhang     rmax = imax[row];
5890880e062SHong Zhang     nrow = ailen[row];
5900880e062SHong Zhang     low  = 0;
591818f2c47SBarry Smith     high = nrow;
5920880e062SHong Zhang     for (l=0; l<n; l++) { /* loop over added columns */
5930880e062SHong Zhang       if (in[l] < 0) continue;
5940880e062SHong Zhang       col = in[l];
5952515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
59677431f27SBarry Smith       if (col >= a->nbs) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",col,a->nbs-1);
597b1823623SSatish Balay #endif
5980880e062SHong Zhang       if (col < row) continue; /* ignore lower triangular block */
5990880e062SHong Zhang       if (roworiented) {
6000880e062SHong Zhang         value = v + k*(stepval+bs)*bs + l*bs;
6010880e062SHong Zhang       } else {
6020880e062SHong Zhang         value = v + l*(stepval+bs)*bs + k*bs;
6030880e062SHong Zhang       }
6047cd84e04SBarry Smith       if (col <= lastcol) low = 0; else high = nrow;
605e2ee6c50SBarry Smith       lastcol = col;
6060880e062SHong Zhang       while (high-low > 7) {
6070880e062SHong Zhang         t = (low+high)/2;
6080880e062SHong Zhang         if (rp[t] > col) high = t;
6090880e062SHong Zhang         else             low  = t;
6100880e062SHong Zhang       }
6110880e062SHong Zhang       for (i=low; i<high; i++) {
6120880e062SHong Zhang         if (rp[i] > col) break;
6130880e062SHong Zhang         if (rp[i] == col) {
6140880e062SHong Zhang           bap  = ap +  bs2*i;
6150880e062SHong Zhang           if (roworiented) {
6160880e062SHong Zhang             if (is == ADD_VALUES) {
6170880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
6180880e062SHong Zhang                 for (jj=ii; jj<bs2; jj+=bs) {
6190880e062SHong Zhang                   bap[jj] += *value++;
6200880e062SHong Zhang                 }
6210880e062SHong Zhang               }
6220880e062SHong Zhang             } else {
6230880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
6240880e062SHong Zhang                 for (jj=ii; jj<bs2; jj+=bs) {
6250880e062SHong Zhang                   bap[jj] = *value++;
6260880e062SHong Zhang                 }
6270880e062SHong Zhang                }
6280880e062SHong Zhang             }
6290880e062SHong Zhang           } else {
6300880e062SHong Zhang             if (is == ADD_VALUES) {
6310880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
6320880e062SHong Zhang                 for (jj=0; jj<bs; jj++) {
6330880e062SHong Zhang                   *bap++ += *value++;
6340880e062SHong Zhang                 }
6350880e062SHong Zhang               }
6360880e062SHong Zhang             } else {
6370880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
6380880e062SHong Zhang                 for (jj=0; jj<bs; jj++) {
6390880e062SHong Zhang                   *bap++  = *value++;
6400880e062SHong Zhang                 }
6410880e062SHong Zhang               }
6420880e062SHong Zhang             }
6430880e062SHong Zhang           }
6440880e062SHong Zhang           goto noinsert2;
6450880e062SHong Zhang         }
6460880e062SHong Zhang       }
6470880e062SHong Zhang       if (nonew == 1) goto noinsert2;
648085a36d4SBarry Smith       if (nonew == -1) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col);
649421e10b8SBarry Smith       MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,row,col,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar);
650c03d1d03SSatish Balay       N = nrow++ - 1; high++;
6510880e062SHong Zhang       /* shift up all the later entries in this row */
6520880e062SHong Zhang       for (ii=N; ii>=i; ii--) {
6530880e062SHong Zhang         rp[ii+1] = rp[ii];
6540880e062SHong Zhang         ierr = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr);
6550880e062SHong Zhang       }
6560880e062SHong Zhang       if (N >= i) {
6570880e062SHong Zhang         ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr);
6580880e062SHong Zhang       }
6590880e062SHong Zhang       rp[i] = col;
6600880e062SHong Zhang       bap   = ap +  bs2*i;
6610880e062SHong Zhang       if (roworiented) {
6620880e062SHong Zhang         for (ii=0; ii<bs; ii++,value+=stepval) {
6630880e062SHong Zhang           for (jj=ii; jj<bs2; jj+=bs) {
6640880e062SHong Zhang             bap[jj] = *value++;
6650880e062SHong Zhang           }
6660880e062SHong Zhang         }
6670880e062SHong Zhang       } else {
6680880e062SHong Zhang         for (ii=0; ii<bs; ii++,value+=stepval) {
6690880e062SHong Zhang           for (jj=0; jj<bs; jj++) {
6700880e062SHong Zhang             *bap++  = *value++;
6710880e062SHong Zhang           }
6720880e062SHong Zhang         }
6730880e062SHong Zhang        }
6740880e062SHong Zhang     noinsert2:;
6750880e062SHong Zhang       low = i;
6760880e062SHong Zhang     }
6770880e062SHong Zhang     ailen[row] = nrow;
6780880e062SHong Zhang   }
6790880e062SHong Zhang    PetscFunctionReturn(0);
68049b5e25fSSatish Balay }
68149b5e25fSSatish Balay 
6824a2ae208SSatish Balay #undef __FUNCT__
6834a2ae208SSatish Balay #define __FUNCT__ "MatAssemblyEnd_SeqSBAIJ"
684dfbe8321SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ(Mat A,MatAssemblyType mode)
68549b5e25fSSatish Balay {
68649b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
6876849ba73SBarry Smith   PetscErrorCode ierr;
68813f74950SBarry Smith   PetscInt       fshift = 0,i,j,*ai = a->i,*aj = a->j,*imax = a->imax;
689899cda47SBarry Smith   PetscInt       m = A->rmap.N,*ip,N,*ailen = a->ilen;
69013f74950SBarry Smith   PetscInt       mbs = a->mbs,bs2 = a->bs2,rmax = 0;
69149b5e25fSSatish Balay   MatScalar      *aa = a->a,*ap;
69249b5e25fSSatish Balay 
69349b5e25fSSatish Balay   PetscFunctionBegin;
69449b5e25fSSatish Balay   if (mode == MAT_FLUSH_ASSEMBLY) PetscFunctionReturn(0);
69549b5e25fSSatish Balay 
69649b5e25fSSatish Balay   if (m) rmax = ailen[0];
69749b5e25fSSatish Balay   for (i=1; i<mbs; i++) {
69849b5e25fSSatish Balay     /* move each row back by the amount of empty slots (fshift) before it*/
69949b5e25fSSatish Balay     fshift += imax[i-1] - ailen[i-1];
70049b5e25fSSatish Balay      rmax   = PetscMax(rmax,ailen[i]);
70149b5e25fSSatish Balay      if (fshift) {
70249b5e25fSSatish Balay        ip = aj + ai[i]; ap = aa + bs2*ai[i];
70349b5e25fSSatish Balay        N = ailen[i];
70449b5e25fSSatish Balay        for (j=0; j<N; j++) {
70549b5e25fSSatish Balay          ip[j-fshift] = ip[j];
70649b5e25fSSatish Balay          ierr = PetscMemcpy(ap+(j-fshift)*bs2,ap+j*bs2,bs2*sizeof(MatScalar));CHKERRQ(ierr);
70749b5e25fSSatish Balay        }
70849b5e25fSSatish Balay      }
70949b5e25fSSatish Balay      ai[i] = ai[i-1] + ailen[i-1];
71049b5e25fSSatish Balay   }
71149b5e25fSSatish Balay   if (mbs) {
71249b5e25fSSatish Balay     fshift += imax[mbs-1] - ailen[mbs-1];
71349b5e25fSSatish Balay      ai[mbs] = ai[mbs-1] + ailen[mbs-1];
71449b5e25fSSatish Balay   }
71549b5e25fSSatish Balay   /* reset ilen and imax for each row */
71649b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
71749b5e25fSSatish Balay     ailen[i] = imax[i] = ai[i+1] - ai[i];
71849b5e25fSSatish Balay   }
7196c6c5352SBarry Smith   a->nz = ai[mbs];
72049b5e25fSSatish Balay 
721b424e231SHong Zhang   /* diagonals may have moved, reset it */
722b424e231SHong Zhang   if (a->diag) {
72313f74950SBarry Smith     ierr = PetscMemcpy(a->diag,ai,(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr);
72449b5e25fSSatish Balay   }
725899cda47SBarry 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);
726ae15b995SBarry Smith   ierr = PetscInfo1(A,"Number of mallocs during MatSetValues is %D\n",a->reallocs);CHKERRQ(ierr);
727ae15b995SBarry Smith   ierr = PetscInfo1(A,"Most nonzeros blocks in any row is %D\n",rmax);CHKERRQ(ierr);
72849b5e25fSSatish Balay   a->reallocs          = 0;
72949b5e25fSSatish Balay   A->info.nz_unneeded  = (PetscReal)fshift*bs2;
73049b5e25fSSatish Balay   PetscFunctionReturn(0);
73149b5e25fSSatish Balay }
73249b5e25fSSatish Balay 
73349b5e25fSSatish Balay /*
73449b5e25fSSatish Balay    This function returns an array of flags which indicate the locations of contiguous
73549b5e25fSSatish Balay    blocks that should be zeroed. for eg: if bs = 3  and is = [0,1,2,3,5,6,7,8,9]
73649b5e25fSSatish Balay    then the resulting sizes = [3,1,1,3,1] correspondig to sets [(0,1,2),(3),(5),(6,7,8),(9)]
73749b5e25fSSatish Balay    Assume: sizes should be long enough to hold all the values.
73849b5e25fSSatish Balay */
7394a2ae208SSatish Balay #undef __FUNCT__
7404a2ae208SSatish Balay #define __FUNCT__ "MatZeroRows_SeqSBAIJ_Check_Blocks"
74113f74950SBarry Smith PetscErrorCode MatZeroRows_SeqSBAIJ_Check_Blocks(PetscInt idx[],PetscInt n,PetscInt bs,PetscInt sizes[], PetscInt *bs_max)
74249b5e25fSSatish Balay {
74313f74950SBarry Smith   PetscInt   i,j,k,row;
74449b5e25fSSatish Balay   PetscTruth flg;
74549b5e25fSSatish Balay 
74649b5e25fSSatish Balay   PetscFunctionBegin;
74749b5e25fSSatish Balay    for (i=0,j=0; i<n; j++) {
74849b5e25fSSatish Balay      row = idx[i];
74949b5e25fSSatish Balay      if (row%bs!=0) { /* Not the begining of a block */
75049b5e25fSSatish Balay        sizes[j] = 1;
75149b5e25fSSatish Balay        i++;
75249b5e25fSSatish Balay      } else if (i+bs > n) { /* Beginning of a block, but complete block doesn't exist (at idx end) */
75349b5e25fSSatish Balay        sizes[j] = 1;         /* Also makes sure atleast 'bs' values exist for next else */
75449b5e25fSSatish Balay        i++;
75549b5e25fSSatish Balay      } else { /* Begining of the block, so check if the complete block exists */
75649b5e25fSSatish Balay        flg = PETSC_TRUE;
75749b5e25fSSatish Balay        for (k=1; k<bs; k++) {
75849b5e25fSSatish Balay          if (row+k != idx[i+k]) { /* break in the block */
75949b5e25fSSatish Balay            flg = PETSC_FALSE;
76049b5e25fSSatish Balay            break;
76149b5e25fSSatish Balay          }
76249b5e25fSSatish Balay        }
763abc0a331SBarry Smith        if (flg) { /* No break in the bs */
76449b5e25fSSatish Balay          sizes[j] = bs;
76549b5e25fSSatish Balay          i+= bs;
76649b5e25fSSatish Balay        } else {
76749b5e25fSSatish Balay          sizes[j] = 1;
76849b5e25fSSatish Balay          i++;
76949b5e25fSSatish Balay        }
77049b5e25fSSatish Balay      }
77149b5e25fSSatish Balay    }
77249b5e25fSSatish Balay    *bs_max = j;
77349b5e25fSSatish Balay    PetscFunctionReturn(0);
77449b5e25fSSatish Balay }
77549b5e25fSSatish Balay 
77649b5e25fSSatish Balay 
77749b5e25fSSatish Balay /* Only add/insert a(i,j) with i<=j (blocks).
77849b5e25fSSatish Balay    Any a(i,j) with i>j input by user is ingored.
77949b5e25fSSatish Balay */
78049b5e25fSSatish Balay 
7814a2ae208SSatish Balay #undef __FUNCT__
7824a2ae208SSatish Balay #define __FUNCT__ "MatSetValues_SeqSBAIJ"
78313f74950SBarry Smith PetscErrorCode MatSetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is)
78449b5e25fSSatish Balay {
78549b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
7866849ba73SBarry Smith   PetscErrorCode ierr;
787e2ee6c50SBarry Smith   PetscInt       *rp,k,low,high,t,ii,row,nrow,i,col,l,rmax,N,lastcol = -1;
78813f74950SBarry Smith   PetscInt       *imax=a->imax,*ai=a->i,*ailen=a->ilen,roworiented=a->roworiented;
789899cda47SBarry Smith   PetscInt       *aj=a->j,nonew=a->nonew,bs=A->rmap.bs,brow,bcol;
79013f74950SBarry Smith   PetscInt       ridx,cidx,bs2=a->bs2;
79149b5e25fSSatish Balay   MatScalar      *ap,value,*aa=a->a,*bap;
79249b5e25fSSatish Balay 
79349b5e25fSSatish Balay   PetscFunctionBegin;
79449b5e25fSSatish Balay   for (k=0; k<m; k++) { /* loop over added rows */
79549b5e25fSSatish Balay     row  = im[k];       /* row number */
79649b5e25fSSatish Balay     brow = row/bs;      /* block row number */
79749b5e25fSSatish Balay     if (row < 0) continue;
7982515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
799899cda47SBarry Smith     if (row >= A->rmap.N) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,A->rmap.N-1);
80049b5e25fSSatish Balay #endif
80149b5e25fSSatish Balay     rp   = aj + ai[brow]; /*ptr to beginning of column value of the row block*/
80249b5e25fSSatish Balay     ap   = aa + bs2*ai[brow]; /*ptr to beginning of element value of the row block*/
80349b5e25fSSatish Balay     rmax = imax[brow];  /* maximum space allocated for this row */
80449b5e25fSSatish Balay     nrow = ailen[brow]; /* actual length of this row */
80549b5e25fSSatish Balay     low  = 0;
80649b5e25fSSatish Balay 
80749b5e25fSSatish Balay     for (l=0; l<n; l++) { /* loop over added columns */
80849b5e25fSSatish Balay       if (in[l] < 0) continue;
8092515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
810899cda47SBarry 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);
81149b5e25fSSatish Balay #endif
81249b5e25fSSatish Balay       col = in[l];
81349b5e25fSSatish Balay       bcol = col/bs;              /* block col number */
81449b5e25fSSatish Balay 
815941593c8SHong Zhang       if (brow > bcol) {
816941593c8SHong Zhang         if (a->ignore_ltriangular){
817941593c8SHong Zhang           continue; /* ignore lower triangular values */
818941593c8SHong Zhang         } else {
819941593c8SHong 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)");
820941593c8SHong Zhang         }
821941593c8SHong Zhang       }
822f4989cb3SHong Zhang 
82349b5e25fSSatish Balay       ridx = row % bs; cidx = col % bs; /*row and col index inside the block */
8248549e402SHong Zhang       if ((brow==bcol && ridx<=cidx) || (brow<bcol)){
82549b5e25fSSatish Balay         /* element value a(k,l) */
82649b5e25fSSatish Balay         if (roworiented) {
82749b5e25fSSatish Balay           value = v[l + k*n];
82849b5e25fSSatish Balay         } else {
82949b5e25fSSatish Balay           value = v[k + l*m];
83049b5e25fSSatish Balay         }
83149b5e25fSSatish Balay 
83249b5e25fSSatish Balay         /* move pointer bap to a(k,l) quickly and add/insert value */
8337cd84e04SBarry Smith         if (col <= lastcol) low = 0; high = nrow;
834e2ee6c50SBarry Smith         lastcol = col;
83549b5e25fSSatish Balay         while (high-low > 7) {
83649b5e25fSSatish Balay           t = (low+high)/2;
83749b5e25fSSatish Balay           if (rp[t] > bcol) high = t;
83849b5e25fSSatish Balay           else              low  = t;
83949b5e25fSSatish Balay         }
84049b5e25fSSatish Balay         for (i=low; i<high; i++) {
84149b5e25fSSatish Balay           if (rp[i] > bcol) break;
84249b5e25fSSatish Balay           if (rp[i] == bcol) {
84349b5e25fSSatish Balay             bap  = ap +  bs2*i + bs*cidx + ridx;
84449b5e25fSSatish Balay             if (is == ADD_VALUES) *bap += value;
84549b5e25fSSatish Balay             else                  *bap  = value;
8468549e402SHong Zhang             /* for diag block, add/insert its symmetric element a(cidx,ridx) */
8478549e402SHong Zhang             if (brow == bcol && ridx < cidx){
8488549e402SHong Zhang               bap  = ap +  bs2*i + bs*ridx + cidx;
8498549e402SHong Zhang               if (is == ADD_VALUES) *bap += value;
8508549e402SHong Zhang               else                  *bap  = value;
8518549e402SHong Zhang             }
85249b5e25fSSatish Balay             goto noinsert1;
85349b5e25fSSatish Balay           }
85449b5e25fSSatish Balay         }
85549b5e25fSSatish Balay 
85649b5e25fSSatish Balay         if (nonew == 1) goto noinsert1;
857085a36d4SBarry Smith         if (nonew == -1) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col);
858421e10b8SBarry Smith         MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,brow,bcol,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar);
85949b5e25fSSatish Balay 
860c03d1d03SSatish Balay         N = nrow++ - 1; high++;
86149b5e25fSSatish Balay         /* shift up all the later entries in this row */
86249b5e25fSSatish Balay         for (ii=N; ii>=i; ii--) {
86349b5e25fSSatish Balay           rp[ii+1] = rp[ii];
86449b5e25fSSatish Balay           ierr     = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr);
86549b5e25fSSatish Balay         }
86649b5e25fSSatish Balay         if (N>=i) {
86749b5e25fSSatish Balay           ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr);
86849b5e25fSSatish Balay         }
86949b5e25fSSatish Balay         rp[i]                      = bcol;
87049b5e25fSSatish Balay         ap[bs2*i + bs*cidx + ridx] = value;
87149b5e25fSSatish Balay       noinsert1:;
87249b5e25fSSatish Balay         low = i;
8738549e402SHong Zhang       }
87449b5e25fSSatish Balay     }   /* end of loop over added columns */
87549b5e25fSSatish Balay     ailen[brow] = nrow;
87649b5e25fSSatish Balay   }   /* end of loop over added rows */
87749b5e25fSSatish Balay   PetscFunctionReturn(0);
87849b5e25fSSatish Balay }
87949b5e25fSSatish Balay 
8806849ba73SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorSymbolic_SeqSBAIJ(Mat,IS,MatFactorInfo*,Mat*);
8816849ba73SBarry Smith EXTERN PetscErrorCode MatCholeskyFactor_SeqSBAIJ(Mat,IS,MatFactorInfo*);
88213f74950SBarry Smith EXTERN PetscErrorCode MatIncreaseOverlap_SeqSBAIJ(Mat,PetscInt,IS[],PetscInt);
88313f74950SBarry Smith EXTERN PetscErrorCode MatGetSubMatrix_SeqSBAIJ(Mat,IS,IS,PetscInt,MatReuse,Mat*);
88413f74950SBarry Smith EXTERN PetscErrorCode MatGetSubMatrices_SeqSBAIJ(Mat,PetscInt,const IS[],const IS[],MatReuse,Mat*[]);
885f4df32b1SMatthew Knepley EXTERN PetscErrorCode MatScale_SeqSBAIJ(Mat,PetscScalar);
8866849ba73SBarry Smith EXTERN PetscErrorCode MatNorm_SeqSBAIJ(Mat,NormType,PetscReal *);
8876849ba73SBarry Smith EXTERN PetscErrorCode MatEqual_SeqSBAIJ(Mat,Mat,PetscTruth*);
8886849ba73SBarry Smith EXTERN PetscErrorCode MatGetDiagonal_SeqSBAIJ(Mat,Vec);
8896849ba73SBarry Smith EXTERN PetscErrorCode MatDiagonalScale_SeqSBAIJ(Mat,Vec,Vec);
8906849ba73SBarry Smith EXTERN PetscErrorCode MatGetInfo_SeqSBAIJ(Mat,MatInfoType,MatInfo *);
8916849ba73SBarry Smith EXTERN PetscErrorCode MatZeroEntries_SeqSBAIJ(Mat);
8926849ba73SBarry Smith EXTERN PetscErrorCode MatGetRowMax_SeqSBAIJ(Mat,Vec);
89349b5e25fSSatish Balay 
8946849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_N(Mat,Vec,Vec);
8956849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_1(Mat,Vec,Vec);
8966849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_2(Mat,Vec,Vec);
8976849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_3(Mat,Vec,Vec);
8986849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_4(Mat,Vec,Vec);
8996849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_5(Mat,Vec,Vec);
9006849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_6(Mat,Vec,Vec);
9016849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_7(Mat,Vec,Vec);
90249b5e25fSSatish Balay 
9036849ba73SBarry Smith EXTERN PetscErrorCode MatSolves_SeqSBAIJ_1(Mat,Vecs,Vecs);
904d59c15a7SBarry Smith 
9056849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_1_NaturalOrdering(Mat,Vec,Vec);
9066849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_2_NaturalOrdering(Mat,Vec,Vec);
9076849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_3_NaturalOrdering(Mat,Vec,Vec);
9086849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_4_NaturalOrdering(Mat,Vec,Vec);
9096849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_5_NaturalOrdering(Mat,Vec,Vec);
9106849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_6_NaturalOrdering(Mat,Vec,Vec);
9116849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_7_NaturalOrdering(Mat,Vec,Vec);
9126849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_N_NaturalOrdering(Mat,Vec,Vec);
91307f98182SSatish Balay 
914af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_N(Mat,MatFactorInfo*,Mat*);
915af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_1(Mat,MatFactorInfo*,Mat*);
916af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_2(Mat,MatFactorInfo*,Mat*);
917af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_3(Mat,MatFactorInfo*,Mat*);
918af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_4(Mat,MatFactorInfo*,Mat*);
919af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_5(Mat,MatFactorInfo*,Mat*);
920af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_6(Mat,MatFactorInfo*,Mat*);
921af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_7(Mat,MatFactorInfo*,Mat*);
92213f74950SBarry Smith EXTERN PetscErrorCode MatGetInertia_SeqSBAIJ(Mat,PetscInt*,PetscInt*,PetscInt*);
92349b5e25fSSatish Balay 
9246849ba73SBarry Smith EXTERN PetscErrorCode MatMult_SeqSBAIJ_1(Mat,Vec,Vec);
9256849ba73SBarry Smith EXTERN PetscErrorCode MatMult_SeqSBAIJ_2(Mat,Vec,Vec);
9266849ba73SBarry Smith EXTERN PetscErrorCode MatMult_SeqSBAIJ_3(Mat,Vec,Vec);
9276849ba73SBarry Smith EXTERN PetscErrorCode MatMult_SeqSBAIJ_4(Mat,Vec,Vec);
9286849ba73SBarry Smith EXTERN PetscErrorCode MatMult_SeqSBAIJ_5(Mat,Vec,Vec);
9296849ba73SBarry Smith EXTERN PetscErrorCode MatMult_SeqSBAIJ_6(Mat,Vec,Vec);
9306849ba73SBarry Smith EXTERN PetscErrorCode MatMult_SeqSBAIJ_7(Mat,Vec,Vec);
9316849ba73SBarry Smith EXTERN PetscErrorCode MatMult_SeqSBAIJ_N(Mat,Vec,Vec);
93249b5e25fSSatish Balay 
9336849ba73SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_1(Mat,Vec,Vec,Vec);
9346849ba73SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_2(Mat,Vec,Vec,Vec);
9356849ba73SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_3(Mat,Vec,Vec,Vec);
9366849ba73SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_4(Mat,Vec,Vec,Vec);
9376849ba73SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_5(Mat,Vec,Vec,Vec);
9386849ba73SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_6(Mat,Vec,Vec,Vec);
9396849ba73SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_7(Mat,Vec,Vec,Vec);
9406849ba73SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_N(Mat,Vec,Vec,Vec);
94149b5e25fSSatish Balay 
9424a2ae208SSatish Balay #undef __FUNCT__
9434d101231SSatish Balay #define __FUNCT__ "MatICCFactor_SeqSBAIJ"
944dfbe8321SBarry Smith PetscErrorCode MatICCFactor_SeqSBAIJ(Mat inA,IS row,MatFactorInfo *info)
94549b5e25fSSatish Balay {
9464ccecd49SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)inA->data;
94749b5e25fSSatish Balay   Mat            outA;
948dfbe8321SBarry Smith   PetscErrorCode ierr;
949c84f5b01SHong Zhang   PetscTruth     row_identity;
95049b5e25fSSatish Balay 
95149b5e25fSSatish Balay   PetscFunctionBegin;
952c84f5b01SHong Zhang   if (info->levels != 0) SETERRQ(PETSC_ERR_SUP,"Only levels=0 is supported for in-place icc");
953c84f5b01SHong Zhang   ierr = ISIdentity(row,&row_identity);CHKERRQ(ierr);
954c84f5b01SHong Zhang   if (!row_identity) SETERRQ(PETSC_ERR_SUP,"Matrix reordering is not supported");
955c84f5b01SHong 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()! */
956c84f5b01SHong Zhang 
95749b5e25fSSatish Balay   outA        = inA;
9581260e1f8SHong Zhang   inA->factor = FACTOR_CHOLESKY;
95949b5e25fSSatish Balay 
9601a3463dfSHong Zhang   ierr = MatMarkDiagonal_SeqSBAIJ(inA);CHKERRQ(ierr);
96149b5e25fSSatish Balay   /*
962c84f5b01SHong Zhang     Blocksize < 8 have a special faster factorization/solver
963c84f5b01SHong Zhang     for ICC(0) factorization with natural ordering
96449b5e25fSSatish Balay   */
965c84f5b01SHong Zhang   switch (inA->rmap.bs){ /* Note: row_identity = PETSC_TRUE! */
96649b5e25fSSatish Balay   case 1:
967c84f5b01SHong Zhang     inA->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering;
9680fe9e5beSBarry Smith     inA->ops->solve            = MatSolve_SeqSBAIJ_1_NaturalOrdering;
96907f98182SSatish Balay     inA->ops->solvetranspose   = MatSolve_SeqSBAIJ_1_NaturalOrdering;
970d59c15a7SBarry Smith     inA->ops->solves           = MatSolves_SeqSBAIJ_1;
971c84f5b01SHong Zhang     ierr = PetscInfo(inA,"Using special in-place natural ordering solvetrans BS=1\n");CHKERRQ(ierr);
972c84f5b01SHong Zhang     break;
97349b5e25fSSatish Balay   case 2:
974c84f5b01SHong Zhang     inA->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering;
9751a3463dfSHong Zhang     inA->ops->solve           = MatSolve_SeqSBAIJ_2_NaturalOrdering;
9761a3463dfSHong Zhang     inA->ops->solvetranspose  = MatSolve_SeqSBAIJ_2_NaturalOrdering;
977ae15b995SBarry Smith     ierr = PetscInfo(inA,"Using special in-place natural ordering factor and solve BS=2\n");CHKERRQ(ierr);
97849b5e25fSSatish Balay     break;
97949b5e25fSSatish Balay   case 3:
980c84f5b01SHong Zhang      inA->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering;
9811a3463dfSHong Zhang      inA->ops->solve           = MatSolve_SeqSBAIJ_3_NaturalOrdering;
9821a3463dfSHong Zhang      inA->ops->solvetranspose  = MatSolve_SeqSBAIJ_3_NaturalOrdering;
983ae15b995SBarry Smith      ierr = PetscInfo(inA,"Using special in-place natural ordering factor and solve BS=3\n");CHKERRQ(ierr);
98449b5e25fSSatish Balay      break;
98549b5e25fSSatish Balay   case 4:
986c84f5b01SHong Zhang     inA->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering;
9871a3463dfSHong Zhang     inA->ops->solve           = MatSolve_SeqSBAIJ_4_NaturalOrdering;
9881a3463dfSHong Zhang     inA->ops->solvetranspose  = MatSolve_SeqSBAIJ_4_NaturalOrdering;
989ae15b995SBarry Smith     ierr = PetscInfo(inA,"Using special in-place natural ordering factor and solve BS=4\n");CHKERRQ(ierr);
99049b5e25fSSatish Balay     break;
99149b5e25fSSatish Balay   case 5:
992c84f5b01SHong Zhang     inA->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering;
9931a3463dfSHong Zhang     inA->ops->solve           = MatSolve_SeqSBAIJ_5_NaturalOrdering;
9941a3463dfSHong Zhang     inA->ops->solvetranspose  = MatSolve_SeqSBAIJ_5_NaturalOrdering;
995ae15b995SBarry Smith     ierr = PetscInfo(inA,"Using special in-place natural ordering factor and solve BS=5\n");CHKERRQ(ierr);
99649b5e25fSSatish Balay     break;
99749b5e25fSSatish Balay   case 6:
998c84f5b01SHong Zhang     inA->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering;
9991a3463dfSHong Zhang     inA->ops->solve           = MatSolve_SeqSBAIJ_6_NaturalOrdering;
10001a3463dfSHong Zhang     inA->ops->solvetranspose  = MatSolve_SeqSBAIJ_6_NaturalOrdering;
1001ae15b995SBarry Smith     ierr = PetscInfo(inA,"Using special in-place natural ordering factor and solve BS=6\n");CHKERRQ(ierr);
100249b5e25fSSatish Balay     break;
100349b5e25fSSatish Balay   case 7:
1004c84f5b01SHong Zhang     inA->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering;
10051a3463dfSHong Zhang     inA->ops->solvetranspose  = MatSolve_SeqSBAIJ_7_NaturalOrdering;
10061a3463dfSHong Zhang     inA->ops->solve           = MatSolve_SeqSBAIJ_7_NaturalOrdering;
1007ae15b995SBarry Smith     ierr = PetscInfo(inA,"Using special in-place natural ordering factor and solve BS=7\n");CHKERRQ(ierr);
100849b5e25fSSatish Balay     break;
100949b5e25fSSatish Balay   default:
1010c84f5b01SHong Zhang     inA->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering;
1011c84f5b01SHong Zhang     inA->ops->solvetranspose  = MatSolve_SeqSBAIJ_N_NaturalOrdering;
1012c84f5b01SHong Zhang     inA->ops->solve           = MatSolve_SeqSBAIJ_N_NaturalOrdering;
1013c84f5b01SHong Zhang     break;
1014c84f5b01SHong Zhang   }
101549b5e25fSSatish Balay 
1016c84f5b01SHong Zhang   a->row = row;
1017c84f5b01SHong Zhang   a->col = row;
1018c84f5b01SHong Zhang   ierr   = PetscObjectReference((PetscObject)row);CHKERRQ(ierr);
1019c84f5b01SHong Zhang   ierr   = PetscObjectReference((PetscObject)row);CHKERRQ(ierr);
1020c84f5b01SHong Zhang 
1021c84f5b01SHong Zhang   /* Create the invert permutation so that it can be used in MatCholeskyFactorNumeric() */
1022c84f5b01SHong Zhang   if (a->icol) {ierr = ISInvertPermutation(row,PETSC_DECIDE, &a->icol);CHKERRQ(ierr);}
102352e6d16bSBarry Smith   ierr = PetscLogObjectParent(inA,a->icol);CHKERRQ(ierr);
102449b5e25fSSatish Balay 
102549b5e25fSSatish Balay   if (!a->solve_work) {
1026c84f5b01SHong Zhang     ierr = PetscMalloc((inA->rmap.N+inA->rmap.bs)*sizeof(PetscScalar),&a->solve_work);CHKERRQ(ierr);
1027c84f5b01SHong Zhang     ierr = PetscLogObjectMemory(inA,(inA->rmap.N+inA->rmap.bs)*sizeof(PetscScalar));CHKERRQ(ierr);
102849b5e25fSSatish Balay   }
102949b5e25fSSatish Balay 
1030af281ebdSHong Zhang   ierr = MatCholeskyFactorNumeric(inA,info,&outA);CHKERRQ(ierr);
103149b5e25fSSatish Balay   PetscFunctionReturn(0);
103249b5e25fSSatish Balay }
1033950f1e5bSHong Zhang 
103449b5e25fSSatish Balay EXTERN_C_BEGIN
10354a2ae208SSatish Balay #undef __FUNCT__
10364a2ae208SSatish Balay #define __FUNCT__ "MatSeqSBAIJSetColumnIndices_SeqSBAIJ"
1037be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatSeqSBAIJSetColumnIndices_SeqSBAIJ(Mat mat,PetscInt *indices)
103849b5e25fSSatish Balay {
1039045c9aa0SHong Zhang   Mat_SeqSBAIJ *baij = (Mat_SeqSBAIJ *)mat->data;
104013f74950SBarry Smith   PetscInt     i,nz,n;
104149b5e25fSSatish Balay 
104249b5e25fSSatish Balay   PetscFunctionBegin;
10436c6c5352SBarry Smith   nz = baij->maxnz;
1044899cda47SBarry Smith   n  = mat->cmap.n;
104549b5e25fSSatish Balay   for (i=0; i<nz; i++) {
104649b5e25fSSatish Balay     baij->j[i] = indices[i];
104749b5e25fSSatish Balay   }
10486c6c5352SBarry Smith    baij->nz = nz;
104949b5e25fSSatish Balay    for (i=0; i<n; i++) {
105049b5e25fSSatish Balay      baij->ilen[i] = baij->imax[i];
105149b5e25fSSatish Balay    }
105249b5e25fSSatish Balay    PetscFunctionReturn(0);
105349b5e25fSSatish Balay }
105449b5e25fSSatish Balay EXTERN_C_END
105549b5e25fSSatish Balay 
10564a2ae208SSatish Balay #undef __FUNCT__
10574a2ae208SSatish Balay #define __FUNCT__ "MatSeqSBAIJSetColumnIndices"
105849b5e25fSSatish Balay /*@
105919585528SSatish Balay   MatSeqSBAIJSetColumnIndices - Set the column indices for all the rows
106049b5e25fSSatish Balay   in the matrix.
106149b5e25fSSatish Balay 
106249b5e25fSSatish Balay   Input Parameters:
106319585528SSatish Balay   +  mat     - the SeqSBAIJ matrix
106449b5e25fSSatish Balay   -  indices - the column indices
106549b5e25fSSatish Balay 
106649b5e25fSSatish Balay   Level: advanced
106749b5e25fSSatish Balay 
106849b5e25fSSatish Balay   Notes:
106949b5e25fSSatish Balay   This can be called if you have precomputed the nonzero structure of the
107049b5e25fSSatish Balay   matrix and want to provide it to the matrix object to improve the performance
107149b5e25fSSatish Balay   of the MatSetValues() operation.
107249b5e25fSSatish Balay 
107349b5e25fSSatish Balay   You MUST have set the correct numbers of nonzeros per row in the call to
1074d1be2dadSMatthew Knepley   MatCreateSeqSBAIJ(), and the columns indices MUST be sorted.
107549b5e25fSSatish Balay 
1076ab9f2c04SSatish Balay   MUST be called before any calls to MatSetValues()
107749b5e25fSSatish Balay 
1078ab9f2c04SSatish Balay   .seealso: MatCreateSeqSBAIJ
107949b5e25fSSatish Balay @*/
1080be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatSeqSBAIJSetColumnIndices(Mat mat,PetscInt *indices)
108149b5e25fSSatish Balay {
108213f74950SBarry Smith   PetscErrorCode ierr,(*f)(Mat,PetscInt *);
108349b5e25fSSatish Balay 
108449b5e25fSSatish Balay   PetscFunctionBegin;
10854482741eSBarry Smith   PetscValidHeaderSpecific(mat,MAT_COOKIE,1);
10864482741eSBarry Smith   PetscValidPointer(indices,2);
1087c134de8dSSatish Balay   ierr = PetscObjectQueryFunction((PetscObject)mat,"MatSeqSBAIJSetColumnIndices_C",(void (**)(void))&f);CHKERRQ(ierr);
108849b5e25fSSatish Balay   if (f) {
108949b5e25fSSatish Balay     ierr = (*f)(mat,indices);CHKERRQ(ierr);
109049b5e25fSSatish Balay   } else {
1091e005ede5SBarry Smith     SETERRQ(PETSC_ERR_SUP,"Wrong type of matrix to set column indices");
109249b5e25fSSatish Balay   }
109349b5e25fSSatish Balay   PetscFunctionReturn(0);
109449b5e25fSSatish Balay }
109549b5e25fSSatish Balay 
10964a2ae208SSatish Balay #undef __FUNCT__
10973c896bc6SHong Zhang #define __FUNCT__ "MatCopy_SeqSBAIJ"
10983c896bc6SHong Zhang PetscErrorCode MatCopy_SeqSBAIJ(Mat A,Mat B,MatStructure str)
10993c896bc6SHong Zhang {
11003c896bc6SHong Zhang   PetscErrorCode ierr;
11013c896bc6SHong Zhang 
11023c896bc6SHong Zhang   PetscFunctionBegin;
11033c896bc6SHong Zhang   /* If the two matrices have the same copy implementation, use fast copy. */
11043c896bc6SHong Zhang   if (str == SAME_NONZERO_PATTERN && (A->ops->copy == B->ops->copy)) {
11053c896bc6SHong Zhang     Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
11063c896bc6SHong Zhang     Mat_SeqSBAIJ *b = (Mat_SeqSBAIJ*)B->data;
11073c896bc6SHong Zhang 
1108899cda47SBarry Smith     if (a->i[A->rmap.N] != b->i[B->rmap.N]) {
11093c896bc6SHong Zhang       SETERRQ(PETSC_ERR_ARG_INCOMP,"Number of nonzeros in two matrices are different");
11103c896bc6SHong Zhang     }
1111899cda47SBarry Smith     ierr = PetscMemcpy(b->a,a->a,(a->i[A->rmap.N])*sizeof(PetscScalar));CHKERRQ(ierr);
11123c896bc6SHong Zhang   } else {
1113f5edf698SHong Zhang     ierr = MatGetRowUpperTriangular(A);CHKERRQ(ierr);
11143c896bc6SHong Zhang     ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr);
1115f5edf698SHong Zhang     ierr = MatRestoreRowUpperTriangular(A);CHKERRQ(ierr);
11163c896bc6SHong Zhang   }
11173c896bc6SHong Zhang   PetscFunctionReturn(0);
11183c896bc6SHong Zhang }
11193c896bc6SHong Zhang 
11203c896bc6SHong Zhang #undef __FUNCT__
11214a2ae208SSatish Balay #define __FUNCT__ "MatSetUpPreallocation_SeqSBAIJ"
1122dfbe8321SBarry Smith PetscErrorCode MatSetUpPreallocation_SeqSBAIJ(Mat A)
1123273d9f13SBarry Smith {
1124dfbe8321SBarry Smith   PetscErrorCode ierr;
1125273d9f13SBarry Smith 
1126273d9f13SBarry Smith   PetscFunctionBegin;
11277edd0491SSatish Balay   ierr =  MatSeqSBAIJSetPreallocation_SeqSBAIJ(A,PetscMax(A->rmap.bs,1),PETSC_DEFAULT,0);CHKERRQ(ierr);
1128273d9f13SBarry Smith   PetscFunctionReturn(0);
1129273d9f13SBarry Smith }
1130273d9f13SBarry Smith 
1131a6ece127SHong Zhang #undef __FUNCT__
1132a6ece127SHong Zhang #define __FUNCT__ "MatGetArray_SeqSBAIJ"
1133dfbe8321SBarry Smith PetscErrorCode MatGetArray_SeqSBAIJ(Mat A,PetscScalar *array[])
1134a6ece127SHong Zhang {
1135a6ece127SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
1136a6ece127SHong Zhang   PetscFunctionBegin;
1137a6ece127SHong Zhang   *array = a->a;
1138a6ece127SHong Zhang   PetscFunctionReturn(0);
1139a6ece127SHong Zhang }
1140a6ece127SHong Zhang 
1141a6ece127SHong Zhang #undef __FUNCT__
1142a6ece127SHong Zhang #define __FUNCT__ "MatRestoreArray_SeqSBAIJ"
1143dfbe8321SBarry Smith PetscErrorCode MatRestoreArray_SeqSBAIJ(Mat A,PetscScalar *array[])
1144a6ece127SHong Zhang {
1145a6ece127SHong Zhang   PetscFunctionBegin;
1146a6ece127SHong Zhang   PetscFunctionReturn(0);
1147a6ece127SHong Zhang  }
1148a6ece127SHong Zhang 
114942ee4b1aSHong Zhang #include "petscblaslapack.h"
115042ee4b1aSHong Zhang #undef __FUNCT__
115142ee4b1aSHong Zhang #define __FUNCT__ "MatAXPY_SeqSBAIJ"
1152f4df32b1SMatthew Knepley PetscErrorCode MatAXPY_SeqSBAIJ(Mat Y,PetscScalar a,Mat X,MatStructure str)
115342ee4b1aSHong Zhang {
115442ee4b1aSHong Zhang   Mat_SeqSBAIJ   *x=(Mat_SeqSBAIJ *)X->data, *y=(Mat_SeqSBAIJ *)Y->data;
1155dfbe8321SBarry Smith   PetscErrorCode ierr;
1156899cda47SBarry Smith   PetscInt       i,bs=Y->rmap.bs,bs2,j;
11574ce68768SBarry Smith   PetscBLASInt   bnz = (PetscBLASInt)x->nz,one = 1;
115842ee4b1aSHong Zhang 
115942ee4b1aSHong Zhang   PetscFunctionBegin;
116042ee4b1aSHong Zhang   if (str == SAME_NONZERO_PATTERN) {
1161f4df32b1SMatthew Knepley     PetscScalar alpha = a;
1162f4df32b1SMatthew Knepley     BLASaxpy_(&bnz,&alpha,x->a,&one,y->a,&one);
1163c537a176SHong Zhang   } else if (str == SUBSET_NONZERO_PATTERN) { /* nonzeros of X is a subset of Y's */
1164c4319e64SHong Zhang     if (y->xtoy && y->XtoY != X) {
1165c4319e64SHong Zhang       ierr = PetscFree(y->xtoy);CHKERRQ(ierr);
1166c4319e64SHong Zhang       ierr = MatDestroy(y->XtoY);CHKERRQ(ierr);
1167c537a176SHong Zhang     }
1168c4319e64SHong Zhang     if (!y->xtoy) { /* get xtoy */
1169c4319e64SHong Zhang       ierr = MatAXPYGetxtoy_Private(x->mbs,x->i,x->j,PETSC_NULL, y->i,y->j,PETSC_NULL, &y->xtoy);CHKERRQ(ierr);
1170c4319e64SHong Zhang       y->XtoY = X;
1171c537a176SHong Zhang     }
1172c4319e64SHong Zhang     bs2 = bs*bs;
11736c6c5352SBarry Smith     for (i=0; i<x->nz; i++) {
1174c4319e64SHong Zhang       j = 0;
1175c4319e64SHong Zhang       while (j < bs2){
1176f4df32b1SMatthew Knepley         y->a[bs2*y->xtoy[i]+j] += a*(x->a[bs2*i+j]);
1177c4319e64SHong Zhang         j++;
1178c537a176SHong Zhang       }
1179c4319e64SHong Zhang     }
1180ae15b995SBarry 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);
118142ee4b1aSHong Zhang   } else {
1182f5edf698SHong Zhang     ierr = MatGetRowUpperTriangular(X);CHKERRQ(ierr);
1183f4df32b1SMatthew Knepley     ierr = MatAXPY_Basic(Y,a,X,str);CHKERRQ(ierr);
1184f5edf698SHong Zhang     ierr = MatRestoreRowUpperTriangular(X);CHKERRQ(ierr);
118542ee4b1aSHong Zhang   }
118642ee4b1aSHong Zhang   PetscFunctionReturn(0);
118742ee4b1aSHong Zhang }
118842ee4b1aSHong Zhang 
1189efcf0fc3SBarry Smith #undef __FUNCT__
1190efcf0fc3SBarry Smith #define __FUNCT__ "MatIsSymmetric_SeqSBAIJ"
1191dfbe8321SBarry Smith PetscErrorCode MatIsSymmetric_SeqSBAIJ(Mat A,PetscReal tol,PetscTruth *flg)
1192efcf0fc3SBarry Smith {
1193efcf0fc3SBarry Smith   PetscFunctionBegin;
1194efcf0fc3SBarry Smith   *flg = PETSC_TRUE;
1195efcf0fc3SBarry Smith   PetscFunctionReturn(0);
1196efcf0fc3SBarry Smith }
1197efcf0fc3SBarry Smith 
1198efcf0fc3SBarry Smith #undef __FUNCT__
1199efcf0fc3SBarry Smith #define __FUNCT__ "MatIsStructurallySymmetric_SeqSBAIJ"
1200dfbe8321SBarry Smith PetscErrorCode MatIsStructurallySymmetric_SeqSBAIJ(Mat A,PetscTruth *flg)
1201efcf0fc3SBarry Smith {
1202efcf0fc3SBarry Smith    PetscFunctionBegin;
1203efcf0fc3SBarry Smith    *flg = PETSC_TRUE;
1204efcf0fc3SBarry Smith    PetscFunctionReturn(0);
1205efcf0fc3SBarry Smith }
1206efcf0fc3SBarry Smith 
1207efcf0fc3SBarry Smith #undef __FUNCT__
1208efcf0fc3SBarry Smith #define __FUNCT__ "MatIsHermitian_SeqSBAIJ"
1209dfbe8321SBarry Smith PetscErrorCode MatIsHermitian_SeqSBAIJ(Mat A,PetscTruth *flg)
1210efcf0fc3SBarry Smith  {
1211efcf0fc3SBarry Smith    PetscFunctionBegin;
1212efcf0fc3SBarry Smith    *flg = PETSC_FALSE;
1213efcf0fc3SBarry Smith    PetscFunctionReturn(0);
1214efcf0fc3SBarry Smith  }
1215efcf0fc3SBarry Smith 
121699cafbc1SBarry Smith #undef __FUNCT__
121799cafbc1SBarry Smith #define __FUNCT__ "MatRealPart_SeqSBAIJ"
121899cafbc1SBarry Smith PetscErrorCode MatRealPart_SeqSBAIJ(Mat A)
121999cafbc1SBarry Smith {
122099cafbc1SBarry Smith   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
122199cafbc1SBarry Smith   PetscInt       i,nz = a->bs2*a->i[a->mbs];
122299cafbc1SBarry Smith   PetscScalar    *aa = a->a;
122399cafbc1SBarry Smith 
122499cafbc1SBarry Smith   PetscFunctionBegin;
122599cafbc1SBarry Smith   for (i=0; i<nz; i++) aa[i] = PetscRealPart(aa[i]);
122699cafbc1SBarry Smith   PetscFunctionReturn(0);
122799cafbc1SBarry Smith }
122899cafbc1SBarry Smith 
122999cafbc1SBarry Smith #undef __FUNCT__
123099cafbc1SBarry Smith #define __FUNCT__ "MatImaginaryPart_SeqSBAIJ"
123199cafbc1SBarry Smith PetscErrorCode MatImaginaryPart_SeqSBAIJ(Mat A)
123299cafbc1SBarry Smith {
123399cafbc1SBarry Smith   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
123499cafbc1SBarry Smith   PetscInt       i,nz = a->bs2*a->i[a->mbs];
123599cafbc1SBarry Smith   PetscScalar    *aa = a->a;
123699cafbc1SBarry Smith 
123799cafbc1SBarry Smith   PetscFunctionBegin;
123899cafbc1SBarry Smith   for (i=0; i<nz; i++) aa[i] = PetscImaginaryPart(aa[i]);
123999cafbc1SBarry Smith   PetscFunctionReturn(0);
124099cafbc1SBarry Smith }
124199cafbc1SBarry Smith 
124249b5e25fSSatish Balay /* -------------------------------------------------------------------*/
124349b5e25fSSatish Balay static struct _MatOps MatOps_Values = {MatSetValues_SeqSBAIJ,
124449b5e25fSSatish Balay        MatGetRow_SeqSBAIJ,
124549b5e25fSSatish Balay        MatRestoreRow_SeqSBAIJ,
124649b5e25fSSatish Balay        MatMult_SeqSBAIJ_N,
124797304618SKris Buschelman /* 4*/ MatMultAdd_SeqSBAIJ_N,
1248431c96f7SBarry Smith        MatMult_SeqSBAIJ_N,       /* transpose versions are same as non-transpose versions */
1249e005ede5SBarry Smith        MatMultAdd_SeqSBAIJ_N,
125049b5e25fSSatish Balay        MatSolve_SeqSBAIJ_N,
125149b5e25fSSatish Balay        0,
125249b5e25fSSatish Balay        0,
125397304618SKris Buschelman /*10*/ 0,
125449b5e25fSSatish Balay        0,
12555f4ef2deSHong Zhang        MatCholeskyFactor_SeqSBAIJ,
1256d06b337dSHong Zhang        MatRelax_SeqSBAIJ,
125749b5e25fSSatish Balay        MatTranspose_SeqSBAIJ,
125897304618SKris Buschelman /*15*/ MatGetInfo_SeqSBAIJ,
125949b5e25fSSatish Balay        MatEqual_SeqSBAIJ,
126049b5e25fSSatish Balay        MatGetDiagonal_SeqSBAIJ,
126149b5e25fSSatish Balay        MatDiagonalScale_SeqSBAIJ,
126249b5e25fSSatish Balay        MatNorm_SeqSBAIJ,
126397304618SKris Buschelman /*20*/ 0,
126449b5e25fSSatish Balay        MatAssemblyEnd_SeqSBAIJ,
126549b5e25fSSatish Balay        0,
126649b5e25fSSatish Balay        MatSetOption_SeqSBAIJ,
126749b5e25fSSatish Balay        MatZeroEntries_SeqSBAIJ,
1268dcf5cc72SBarry Smith /*25*/ 0,
126949b5e25fSSatish Balay        0,
127049b5e25fSSatish Balay        0,
12715f4ef2deSHong Zhang        MatCholeskyFactorSymbolic_SeqSBAIJ,
12725f4ef2deSHong Zhang        MatCholeskyFactorNumeric_SeqSBAIJ_N,
127397304618SKris Buschelman /*30*/ MatSetUpPreallocation_SeqSBAIJ,
1274c464158bSHong Zhang        0,
12754d101231SSatish Balay        MatICCFactorSymbolic_SeqSBAIJ,
1276a6ece127SHong Zhang        MatGetArray_SeqSBAIJ,
1277a6ece127SHong Zhang        MatRestoreArray_SeqSBAIJ,
127897304618SKris Buschelman /*35*/ MatDuplicate_SeqSBAIJ,
127949b5e25fSSatish Balay        0,
128049b5e25fSSatish Balay        0,
128149b5e25fSSatish Balay        0,
1282c84f5b01SHong Zhang        MatICCFactor_SeqSBAIJ,
128397304618SKris Buschelman /*40*/ MatAXPY_SeqSBAIJ,
128449b5e25fSSatish Balay        MatGetSubMatrices_SeqSBAIJ,
128549b5e25fSSatish Balay        MatIncreaseOverlap_SeqSBAIJ,
128649b5e25fSSatish Balay        MatGetValues_SeqSBAIJ,
12873c896bc6SHong Zhang        MatCopy_SeqSBAIJ,
12888c07d4e3SBarry Smith /*45*/ 0,
128949b5e25fSSatish Balay        MatScale_SeqSBAIJ,
129049b5e25fSSatish Balay        0,
129149b5e25fSSatish Balay        0,
129249b5e25fSSatish Balay        0,
1293521d7252SBarry Smith /*50*/ 0,
129449b5e25fSSatish Balay        MatGetRowIJ_SeqSBAIJ,
129549b5e25fSSatish Balay        MatRestoreRowIJ_SeqSBAIJ,
129649b5e25fSSatish Balay        0,
129749b5e25fSSatish Balay        0,
129897304618SKris Buschelman /*55*/ 0,
129949b5e25fSSatish Balay        0,
130049b5e25fSSatish Balay        0,
130149b5e25fSSatish Balay        0,
130249b5e25fSSatish Balay        MatSetValuesBlocked_SeqSBAIJ,
130397304618SKris Buschelman /*60*/ MatGetSubMatrix_SeqSBAIJ,
130449b5e25fSSatish Balay        0,
130549b5e25fSSatish Balay        0,
1306357abbc8SBarry Smith        0,
1307d959ec07SHong Zhang        0,
130897304618SKris Buschelman /*65*/ 0,
1309d959ec07SHong Zhang        0,
1310d959ec07SHong Zhang        0,
1311d959ec07SHong Zhang        0,
1312d959ec07SHong Zhang        0,
131397304618SKris Buschelman /*70*/ MatGetRowMax_SeqSBAIJ,
13143e0d88b5SBarry Smith        0,
13153e0d88b5SBarry Smith        0,
13163e0d88b5SBarry Smith        0,
13173e0d88b5SBarry Smith        0,
131897304618SKris Buschelman /*75*/ 0,
13193e0d88b5SBarry Smith        0,
13203e0d88b5SBarry Smith        0,
13213e0d88b5SBarry Smith        0,
13223e0d88b5SBarry Smith        0,
132397304618SKris Buschelman /*80*/ 0,
13243e0d88b5SBarry Smith        0,
13253e0d88b5SBarry Smith        0,
13263e0d88b5SBarry Smith #if !defined(PETSC_USE_COMPLEX)
132797304618SKris Buschelman        MatGetInertia_SeqSBAIJ,
13283e0d88b5SBarry Smith #else
132997304618SKris Buschelman        0,
13303e0d88b5SBarry Smith #endif
1331865e5f61SKris Buschelman        MatLoad_SeqSBAIJ,
1332865e5f61SKris Buschelman /*85*/ MatIsSymmetric_SeqSBAIJ,
1333865e5f61SKris Buschelman        MatIsHermitian_SeqSBAIJ,
1334efcf0fc3SBarry Smith        MatIsStructurallySymmetric_SeqSBAIJ,
1335865e5f61SKris Buschelman        0,
1336865e5f61SKris Buschelman        0,
1337865e5f61SKris Buschelman /*90*/ 0,
1338865e5f61SKris Buschelman        0,
1339865e5f61SKris Buschelman        0,
1340865e5f61SKris Buschelman        0,
1341865e5f61SKris Buschelman        0,
1342865e5f61SKris Buschelman /*95*/ 0,
1343865e5f61SKris Buschelman        0,
1344865e5f61SKris Buschelman        0,
134599cafbc1SBarry Smith        0,
134699cafbc1SBarry Smith        0,
134799cafbc1SBarry Smith /*100*/0,
134899cafbc1SBarry Smith        0,
134999cafbc1SBarry Smith        0,
135099cafbc1SBarry Smith        0,
135199cafbc1SBarry Smith        0,
135299cafbc1SBarry Smith /*105*/0,
135399cafbc1SBarry Smith        MatRealPart_SeqSBAIJ,
1354f5edf698SHong Zhang        MatImaginaryPart_SeqSBAIJ,
1355f5edf698SHong Zhang        MatGetRowUpperTriangular_SeqSBAIJ,
1356f5edf698SHong Zhang        MatRestoreRowUpperTriangular_SeqSBAIJ
135799cafbc1SBarry Smith };
1358be1d678aSKris Buschelman 
135949b5e25fSSatish Balay EXTERN_C_BEGIN
13604a2ae208SSatish Balay #undef __FUNCT__
13614a2ae208SSatish Balay #define __FUNCT__ "MatStoreValues_SeqSBAIJ"
1362be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatStoreValues_SeqSBAIJ(Mat mat)
136349b5e25fSSatish Balay {
13644afc71dfSHong Zhang   Mat_SeqSBAIJ   *aij = (Mat_SeqSBAIJ *)mat->data;
1365899cda47SBarry Smith   PetscInt       nz = aij->i[mat->rmap.N]*mat->rmap.bs*aij->bs2;
1366dfbe8321SBarry Smith   PetscErrorCode ierr;
136749b5e25fSSatish Balay 
136849b5e25fSSatish Balay   PetscFunctionBegin;
136949b5e25fSSatish Balay   if (aij->nonew != 1) {
1370e005ede5SBarry Smith     SETERRQ(PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NO_NEW_NONZERO_LOCATIONS);first");
137149b5e25fSSatish Balay   }
137249b5e25fSSatish Balay 
137349b5e25fSSatish Balay   /* allocate space for values if not already there */
137449b5e25fSSatish Balay   if (!aij->saved_values) {
137587828ca2SBarry Smith     ierr = PetscMalloc((nz+1)*sizeof(PetscScalar),&aij->saved_values);CHKERRQ(ierr);
137649b5e25fSSatish Balay   }
137749b5e25fSSatish Balay 
137849b5e25fSSatish Balay   /* copy values over */
137987828ca2SBarry Smith   ierr = PetscMemcpy(aij->saved_values,aij->a,nz*sizeof(PetscScalar));CHKERRQ(ierr);
138049b5e25fSSatish Balay   PetscFunctionReturn(0);
138149b5e25fSSatish Balay }
138249b5e25fSSatish Balay EXTERN_C_END
138349b5e25fSSatish Balay 
138449b5e25fSSatish Balay EXTERN_C_BEGIN
13854a2ae208SSatish Balay #undef __FUNCT__
13864a2ae208SSatish Balay #define __FUNCT__ "MatRetrieveValues_SeqSBAIJ"
1387be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatRetrieveValues_SeqSBAIJ(Mat mat)
138849b5e25fSSatish Balay {
13894afc71dfSHong Zhang   Mat_SeqSBAIJ   *aij = (Mat_SeqSBAIJ *)mat->data;
13906849ba73SBarry Smith   PetscErrorCode ierr;
1391899cda47SBarry Smith   PetscInt       nz = aij->i[mat->rmap.N]*mat->rmap.bs*aij->bs2;
139249b5e25fSSatish Balay 
139349b5e25fSSatish Balay   PetscFunctionBegin;
139449b5e25fSSatish Balay   if (aij->nonew != 1) {
1395e005ede5SBarry Smith     SETERRQ(PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NO_NEW_NONZERO_LOCATIONS);first");
139649b5e25fSSatish Balay   }
139749b5e25fSSatish Balay   if (!aij->saved_values) {
1398e005ede5SBarry Smith     SETERRQ(PETSC_ERR_ORDER,"Must call MatStoreValues(A);first");
139949b5e25fSSatish Balay   }
140049b5e25fSSatish Balay 
140149b5e25fSSatish Balay   /* copy values over */
140287828ca2SBarry Smith   ierr = PetscMemcpy(aij->a,aij->saved_values,nz*sizeof(PetscScalar));CHKERRQ(ierr);
140349b5e25fSSatish Balay   PetscFunctionReturn(0);
140449b5e25fSSatish Balay }
140549b5e25fSSatish Balay EXTERN_C_END
140649b5e25fSSatish Balay 
14078549e402SHong Zhang EXTERN_C_BEGIN
14084a2ae208SSatish Balay #undef __FUNCT__
1409a23d5eceSKris Buschelman #define __FUNCT__ "MatSeqSBAIJSetPreallocation_SeqSBAIJ"
1410be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatSeqSBAIJSetPreallocation_SeqSBAIJ(Mat B,PetscInt bs,PetscInt nz,PetscInt *nnz)
141149b5e25fSSatish Balay {
1412c464158bSHong Zhang   Mat_SeqSBAIJ   *b = (Mat_SeqSBAIJ*)B->data;
14136849ba73SBarry Smith   PetscErrorCode ierr;
1414ab93d7beSBarry Smith   PetscInt       i,mbs,bs2;
1415ab93d7beSBarry Smith   PetscTruth     skipallocation = PETSC_FALSE,flg;
141649b5e25fSSatish Balay 
141749b5e25fSSatish Balay   PetscFunctionBegin;
1418273d9f13SBarry Smith   B->preallocated = PETSC_TRUE;
1419e82a3eeeSBarry Smith   ierr = PetscOptionsGetInt(B->prefix,"-mat_block_size",&bs,PETSC_NULL);CHKERRQ(ierr);
1420899cda47SBarry Smith   B->rmap.bs = B->cmap.bs = bs;
1421899cda47SBarry Smith   ierr = PetscMapInitialize(B->comm,&B->rmap);CHKERRQ(ierr);
1422899cda47SBarry Smith   ierr = PetscMapInitialize(B->comm,&B->cmap);CHKERRQ(ierr);
1423899cda47SBarry Smith 
1424899cda47SBarry Smith   mbs  = B->rmap.N/bs;
142549b5e25fSSatish Balay   bs2  = bs*bs;
142649b5e25fSSatish Balay 
1427899cda47SBarry Smith   if (mbs*bs != B->rmap.N) {
142829bbc08cSBarry Smith     SETERRQ(PETSC_ERR_ARG_SIZ,"Number rows, cols must be divisible by blocksize");
142949b5e25fSSatish Balay   }
143049b5e25fSSatish Balay 
1431ab93d7beSBarry Smith   if (nz == MAT_SKIP_ALLOCATION) {
1432ab93d7beSBarry Smith     skipallocation = PETSC_TRUE;
1433ab93d7beSBarry Smith     nz             = 0;
1434ab93d7beSBarry Smith   }
1435ab93d7beSBarry Smith 
1436435da068SBarry Smith   if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 3;
143777431f27SBarry Smith   if (nz < 0) SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"nz cannot be less than 0: value %D",nz);
143849b5e25fSSatish Balay   if (nnz) {
143949b5e25fSSatish Balay     for (i=0; i<mbs; i++) {
144077431f27SBarry Smith       if (nnz[i] < 0) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"nnz cannot be less than 0: local row %D value %D",i,nnz[i]);
144177431f27SBarry 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);
144249b5e25fSSatish Balay     }
144349b5e25fSSatish Balay   }
144449b5e25fSSatish Balay 
1445e82a3eeeSBarry Smith   ierr    = PetscOptionsHasName(B->prefix,"-mat_no_unroll",&flg);CHKERRQ(ierr);
144649b5e25fSSatish Balay   if (!flg) {
144749b5e25fSSatish Balay     switch (bs) {
144849b5e25fSSatish Balay     case 1:
14494ccecd49SHong Zhang       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1;
145049b5e25fSSatish Balay       B->ops->solve            = MatSolve_SeqSBAIJ_1;
1451d59c15a7SBarry Smith       B->ops->solves           = MatSolves_SeqSBAIJ_1;
1452e005ede5SBarry Smith       B->ops->solvetranspose   = MatSolve_SeqSBAIJ_1;
145349b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_1;
145449b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_1;
1455431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_1;
1456431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_1;
145749b5e25fSSatish Balay       break;
145849b5e25fSSatish Balay     case 2:
14594ccecd49SHong Zhang       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2;
146049b5e25fSSatish Balay       B->ops->solve            = MatSolve_SeqSBAIJ_2;
1461e005ede5SBarry Smith       B->ops->solvetranspose   = MatSolve_SeqSBAIJ_2;
146249b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_2;
146349b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_2;
1464431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_2;
1465431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_2;
146649b5e25fSSatish Balay       break;
146749b5e25fSSatish Balay     case 3:
14685f4ef2deSHong Zhang       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3;
146949b5e25fSSatish Balay       B->ops->solve            = MatSolve_SeqSBAIJ_3;
1470e005ede5SBarry Smith       B->ops->solvetranspose   = MatSolve_SeqSBAIJ_3;
147149b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_3;
147249b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_3;
1473431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_3;
1474431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_3;
147549b5e25fSSatish Balay       break;
147649b5e25fSSatish Balay     case 4:
14775f4ef2deSHong Zhang       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4;
147849b5e25fSSatish Balay       B->ops->solve            = MatSolve_SeqSBAIJ_4;
1479e005ede5SBarry Smith       B->ops->solvetranspose   = MatSolve_SeqSBAIJ_4;
148049b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_4;
148149b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_4;
1482431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_4;
1483431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_4;
148449b5e25fSSatish Balay       break;
148549b5e25fSSatish Balay     case 5:
14865f4ef2deSHong Zhang       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5;
148749b5e25fSSatish Balay       B->ops->solve            = MatSolve_SeqSBAIJ_5;
1488e005ede5SBarry Smith       B->ops->solvetranspose   = MatSolve_SeqSBAIJ_5;
148949b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_5;
149049b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_5;
1491431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_5;
1492431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_5;
149349b5e25fSSatish Balay       break;
149449b5e25fSSatish Balay     case 6:
14955f4ef2deSHong Zhang       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6;
149649b5e25fSSatish Balay       B->ops->solve            = MatSolve_SeqSBAIJ_6;
1497e005ede5SBarry Smith       B->ops->solvetranspose   = MatSolve_SeqSBAIJ_6;
149849b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_6;
149949b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_6;
1500431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_6;
1501431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_6;
150249b5e25fSSatish Balay       break;
150349b5e25fSSatish Balay     case 7:
1504de53e5efSHong Zhang       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7;
150549b5e25fSSatish Balay       B->ops->solve            = MatSolve_SeqSBAIJ_7;
1506e005ede5SBarry Smith       B->ops->solvetranspose   = MatSolve_SeqSBAIJ_7;
1507de53e5efSHong Zhang       B->ops->mult             = MatMult_SeqSBAIJ_7;
150849b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_7;
1509431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_7;
1510431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_7;
151149b5e25fSSatish Balay       break;
151249b5e25fSSatish Balay     }
151349b5e25fSSatish Balay   }
151449b5e25fSSatish Balay 
151549b5e25fSSatish Balay   b->mbs = mbs;
15164afc71dfSHong Zhang   b->nbs = mbs;
1517ab93d7beSBarry Smith   if (!skipallocation) {
1518ab93d7beSBarry Smith     /* b->ilen will count nonzeros in each block row so far. */
1519ab93d7beSBarry Smith     ierr   = PetscMalloc2(mbs,PetscInt,&b->imax,mbs,PetscInt,&b->ilen);CHKERRQ(ierr);
1520ab93d7beSBarry Smith     for (i=0; i<mbs; i++) { b->ilen[i] = 0;}
152149b5e25fSSatish Balay     if (!nnz) {
1522435da068SBarry Smith       if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 5;
152349b5e25fSSatish Balay       else if (nz <= 0)        nz = 1;
152449b5e25fSSatish Balay       for (i=0; i<mbs; i++) {
15258cef66ccSHong Zhang         b->imax[i] = nz;
152649b5e25fSSatish Balay       }
1527153ea458SHong Zhang       nz = nz*mbs; /* total nz */
152849b5e25fSSatish Balay     } else {
152949b5e25fSSatish Balay       nz = 0;
15308cef66ccSHong Zhang       for (i=0; i<mbs; i++) {b->imax[i] = nnz[i]; nz += nnz[i];}
153149b5e25fSSatish Balay     }
15326c6c5352SBarry Smith     /* nz=(nz+mbs)/2; */ /* total diagonal and superdiagonal nonzero blocks */
153349b5e25fSSatish Balay 
153449b5e25fSSatish Balay     /* allocate the matrix space */
1535899cda47SBarry Smith     ierr = PetscMalloc3(bs2*nz,PetscScalar,&b->a,nz,PetscInt,&b->j,B->rmap.N+1,PetscInt,&b->i);CHKERRQ(ierr);
15366c6c5352SBarry Smith     ierr = PetscMemzero(b->a,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr);
153713f74950SBarry Smith     ierr = PetscMemzero(b->j,nz*sizeof(PetscInt));CHKERRQ(ierr);
153849b5e25fSSatish Balay     b->singlemalloc = PETSC_TRUE;
153949b5e25fSSatish Balay 
154049b5e25fSSatish Balay     /* pointer to beginning of each row */
154149b5e25fSSatish Balay     b->i[0] = 0;
154249b5e25fSSatish Balay     for (i=1; i<mbs+1; i++) {
154349b5e25fSSatish Balay       b->i[i] = b->i[i-1] + b->imax[i-1];
154449b5e25fSSatish Balay     }
1545e6b907acSBarry Smith     b->free_a     = PETSC_TRUE;
1546e6b907acSBarry Smith     b->free_ij    = PETSC_TRUE;
1547e811da20SHong Zhang   } else {
1548e6b907acSBarry Smith     b->free_a     = PETSC_FALSE;
1549e6b907acSBarry Smith     b->free_ij    = PETSC_FALSE;
1550ab93d7beSBarry Smith   }
155149b5e25fSSatish Balay 
1552899cda47SBarry Smith   B->rmap.bs               = bs;
155349b5e25fSSatish Balay   b->bs2              = bs2;
15546c6c5352SBarry Smith   b->nz             = 0;
15556c6c5352SBarry Smith   b->maxnz          = nz*bs2;
1556153ea458SHong Zhang 
155716cdd363SHong Zhang   b->inew             = 0;
155816cdd363SHong Zhang   b->jnew             = 0;
155916cdd363SHong Zhang   b->anew             = 0;
156016cdd363SHong Zhang   b->a2anew           = 0;
15611a3463dfSHong Zhang   b->permute          = PETSC_FALSE;
1562c464158bSHong Zhang   PetscFunctionReturn(0);
1563c464158bSHong Zhang }
1564a23d5eceSKris Buschelman EXTERN_C_END
1565153ea458SHong Zhang 
1566d769727bSBarry Smith EXTERN_C_BEGIN
1567f69a0ea3SMatthew Knepley EXTERN PetscErrorCode PETSCMAT_DLLEXPORT MatConvert_SeqSBAIJ_SeqAIJ(Mat, MatType,MatReuse,Mat*);
1568f69a0ea3SMatthew Knepley EXTERN PetscErrorCode PETSCMAT_DLLEXPORT MatConvert_SeqSBAIJ_SeqBAIJ(Mat, MatType,MatReuse,Mat*);
1569d769727bSBarry Smith EXTERN_C_END
1570d769727bSBarry Smith 
15710bad9183SKris Buschelman /*MC
1572fafad747SKris Buschelman   MATSEQSBAIJ - MATSEQSBAIJ = "seqsbaij" - A matrix type to be used for sequential symmetric block sparse matrices,
15730bad9183SKris Buschelman   based on block compressed sparse row format.  Only the upper triangular portion of the matrix is stored.
15740bad9183SKris Buschelman 
15750bad9183SKris Buschelman   Options Database Keys:
15760bad9183SKris Buschelman   . -mat_type seqsbaij - sets the matrix type to "seqsbaij" during a call to MatSetFromOptions()
15770bad9183SKris Buschelman 
15780bad9183SKris Buschelman   Level: beginner
15790bad9183SKris Buschelman 
15800bad9183SKris Buschelman   .seealso: MatCreateSeqSBAIJ
15810bad9183SKris Buschelman M*/
15820bad9183SKris Buschelman 
1583a23d5eceSKris Buschelman EXTERN_C_BEGIN
1584a23d5eceSKris Buschelman #undef __FUNCT__
1585a23d5eceSKris Buschelman #define __FUNCT__ "MatCreate_SeqSBAIJ"
1586be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreate_SeqSBAIJ(Mat B)
1587a23d5eceSKris Buschelman {
1588a23d5eceSKris Buschelman   Mat_SeqSBAIJ   *b;
1589dfbe8321SBarry Smith   PetscErrorCode ierr;
159013f74950SBarry Smith   PetscMPIInt    size;
1591941593c8SHong Zhang   PetscTruth     flg;
1592a23d5eceSKris Buschelman 
1593a23d5eceSKris Buschelman   PetscFunctionBegin;
1594a23d5eceSKris Buschelman   ierr = MPI_Comm_size(B->comm,&size);CHKERRQ(ierr);
1595a23d5eceSKris Buschelman   if (size > 1) SETERRQ(PETSC_ERR_ARG_WRONG,"Comm must be of size 1");
1596a23d5eceSKris Buschelman 
1597a23d5eceSKris Buschelman   ierr    = PetscNew(Mat_SeqSBAIJ,&b);CHKERRQ(ierr);
1598a23d5eceSKris Buschelman   B->data = (void*)b;
1599a23d5eceSKris Buschelman   ierr    = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr);
1600a23d5eceSKris Buschelman   B->ops->destroy     = MatDestroy_SeqSBAIJ;
1601a23d5eceSKris Buschelman   B->ops->view        = MatView_SeqSBAIJ;
1602a23d5eceSKris Buschelman   B->factor           = 0;
1603a23d5eceSKris Buschelman   B->mapping          = 0;
1604a23d5eceSKris Buschelman   b->row              = 0;
1605a23d5eceSKris Buschelman   b->icol             = 0;
1606a23d5eceSKris Buschelman   b->reallocs         = 0;
1607a23d5eceSKris Buschelman   b->saved_values     = 0;
1608a23d5eceSKris Buschelman 
1609a23d5eceSKris Buschelman 
1610a23d5eceSKris Buschelman   b->sorted           = PETSC_FALSE;
1611a23d5eceSKris Buschelman   b->roworiented      = PETSC_TRUE;
1612a23d5eceSKris Buschelman   b->nonew            = 0;
1613a23d5eceSKris Buschelman   b->diag             = 0;
1614a23d5eceSKris Buschelman   b->solve_work       = 0;
1615a23d5eceSKris Buschelman   b->mult_work        = 0;
1616a23d5eceSKris Buschelman   B->spptr            = 0;
1617a23d5eceSKris Buschelman   b->keepzeroedrows   = PETSC_FALSE;
1618a23d5eceSKris Buschelman   b->xtoy             = 0;
1619a23d5eceSKris Buschelman   b->XtoY             = 0;
1620a23d5eceSKris Buschelman 
1621a23d5eceSKris Buschelman   b->inew             = 0;
1622a23d5eceSKris Buschelman   b->jnew             = 0;
1623a23d5eceSKris Buschelman   b->anew             = 0;
1624a23d5eceSKris Buschelman   b->a2anew           = 0;
1625a23d5eceSKris Buschelman   b->permute          = PETSC_FALSE;
1626a23d5eceSKris Buschelman 
1627941593c8SHong Zhang   b->ignore_ltriangular = PETSC_FALSE;
1628941593c8SHong Zhang   ierr = PetscOptionsHasName(PETSC_NULL,"-mat_ignore_lower_triangular",&flg);CHKERRQ(ierr);
1629941593c8SHong Zhang   if (flg) b->ignore_ltriangular = PETSC_TRUE;
1630941593c8SHong Zhang 
1631f5edf698SHong Zhang   b->getrow_utriangular = PETSC_FALSE;
1632f5edf698SHong Zhang   ierr = PetscOptionsHasName(PETSC_NULL,"-mat_getrow_uppertriangular",&flg);CHKERRQ(ierr);
1633f5edf698SHong Zhang   if (flg) b->getrow_utriangular = PETSC_TRUE;
1634f5edf698SHong Zhang 
1635a23d5eceSKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatStoreValues_C",
1636a23d5eceSKris Buschelman                                      "MatStoreValues_SeqSBAIJ",
1637a23d5eceSKris Buschelman                                      MatStoreValues_SeqSBAIJ);CHKERRQ(ierr);
1638a23d5eceSKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatRetrieveValues_C",
1639a23d5eceSKris Buschelman                                      "MatRetrieveValues_SeqSBAIJ",
1640a23d5eceSKris Buschelman                                      (void*)MatRetrieveValues_SeqSBAIJ);CHKERRQ(ierr);
1641a23d5eceSKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatSeqSBAIJSetColumnIndices_C",
1642a23d5eceSKris Buschelman                                      "MatSeqSBAIJSetColumnIndices_SeqSBAIJ",
1643a23d5eceSKris Buschelman                                      MatSeqSBAIJSetColumnIndices_SeqSBAIJ);CHKERRQ(ierr);
16444e5e7fe4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_seqsbaij_seqaij_C",
16454e5e7fe4SHong Zhang                                      "MatConvert_SeqSBAIJ_SeqAIJ",
16464e5e7fe4SHong Zhang                                       MatConvert_SeqSBAIJ_SeqAIJ);CHKERRQ(ierr);
1647a0e1a404SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_seqsbaij_seqbaij_C",
1648a0e1a404SHong Zhang                                      "MatConvert_SeqSBAIJ_SeqBAIJ",
1649a0e1a404SHong Zhang                                       MatConvert_SeqSBAIJ_SeqBAIJ);CHKERRQ(ierr);
1650a23d5eceSKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatSeqSBAIJSetPreallocation_C",
1651a23d5eceSKris Buschelman                                      "MatSeqSBAIJSetPreallocation_SeqSBAIJ",
1652a23d5eceSKris Buschelman                                      MatSeqSBAIJSetPreallocation_SeqSBAIJ);CHKERRQ(ierr);
165323ce1328SBarry Smith 
165423ce1328SBarry Smith   B->symmetric                  = PETSC_TRUE;
165523ce1328SBarry Smith   B->structurally_symmetric     = PETSC_TRUE;
165623ce1328SBarry Smith   B->symmetric_set              = PETSC_TRUE;
165723ce1328SBarry Smith   B->structurally_symmetric_set = PETSC_TRUE;
165817667f90SBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)B,MATSEQSBAIJ);CHKERRQ(ierr);
1659a23d5eceSKris Buschelman   PetscFunctionReturn(0);
1660a23d5eceSKris Buschelman }
1661a23d5eceSKris Buschelman EXTERN_C_END
1662a23d5eceSKris Buschelman 
1663a23d5eceSKris Buschelman #undef __FUNCT__
1664a23d5eceSKris Buschelman #define __FUNCT__ "MatSeqSBAIJSetPreallocation"
1665a23d5eceSKris Buschelman /*@C
1666a23d5eceSKris Buschelman    MatSeqSBAIJSetPreallocation - Creates a sparse symmetric matrix in block AIJ (block
1667a23d5eceSKris Buschelman    compressed row) format.  For good matrix assembly performance the
1668a23d5eceSKris Buschelman    user should preallocate the matrix storage by setting the parameter nz
1669a23d5eceSKris Buschelman    (or the array nnz).  By setting these parameters accurately, performance
1670a23d5eceSKris Buschelman    during matrix assembly can be increased by more than a factor of 50.
1671a23d5eceSKris Buschelman 
1672a23d5eceSKris Buschelman    Collective on Mat
1673a23d5eceSKris Buschelman 
1674a23d5eceSKris Buschelman    Input Parameters:
1675a23d5eceSKris Buschelman +  A - the symmetric matrix
1676a23d5eceSKris Buschelman .  bs - size of block
1677a23d5eceSKris Buschelman .  nz - number of block nonzeros per block row (same for all rows)
1678a23d5eceSKris Buschelman -  nnz - array containing the number of block nonzeros in the upper triangular plus
1679a23d5eceSKris Buschelman          diagonal portion of each block (possibly different for each block row) or PETSC_NULL
1680a23d5eceSKris Buschelman 
1681a23d5eceSKris Buschelman    Options Database Keys:
1682a23d5eceSKris Buschelman .   -mat_no_unroll - uses code that does not unroll the loops in the
1683a23d5eceSKris Buschelman                      block calculations (much slower)
1684a23d5eceSKris Buschelman .    -mat_block_size - size of the blocks to use
1685a23d5eceSKris Buschelman 
1686a23d5eceSKris Buschelman    Level: intermediate
1687a23d5eceSKris Buschelman 
1688a23d5eceSKris Buschelman    Notes:
1689a23d5eceSKris Buschelman    Specify the preallocated storage with either nz or nnz (not both).
1690a23d5eceSKris Buschelman    Set nz=PETSC_DEFAULT and nnz=PETSC_NULL for PETSc to control dynamic memory
1691a23d5eceSKris Buschelman    allocation.  For additional details, see the users manual chapter on
1692a23d5eceSKris Buschelman    matrices.
1693a23d5eceSKris Buschelman 
169449a6f317SBarry Smith    If the nnz parameter is given then the nz parameter is ignored
169549a6f317SBarry Smith 
169649a6f317SBarry Smith 
1697a23d5eceSKris Buschelman .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateMPISBAIJ()
1698a23d5eceSKris Buschelman @*/
1699be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatSeqSBAIJSetPreallocation(Mat B,PetscInt bs,PetscInt nz,const PetscInt nnz[])
170013f74950SBarry Smith {
170113f74950SBarry Smith   PetscErrorCode ierr,(*f)(Mat,PetscInt,PetscInt,const PetscInt[]);
1702a23d5eceSKris Buschelman 
1703a23d5eceSKris Buschelman   PetscFunctionBegin;
1704a23d5eceSKris Buschelman   ierr = PetscObjectQueryFunction((PetscObject)B,"MatSeqSBAIJSetPreallocation_C",(void (**)(void))&f);CHKERRQ(ierr);
1705a23d5eceSKris Buschelman   if (f) {
1706a23d5eceSKris Buschelman     ierr = (*f)(B,bs,nz,nnz);CHKERRQ(ierr);
1707a23d5eceSKris Buschelman   }
1708a23d5eceSKris Buschelman   PetscFunctionReturn(0);
1709a23d5eceSKris Buschelman }
171049b5e25fSSatish Balay 
17114a2ae208SSatish Balay #undef __FUNCT__
17124a2ae208SSatish Balay #define __FUNCT__ "MatCreateSeqSBAIJ"
1713c464158bSHong Zhang /*@C
1714c464158bSHong Zhang    MatCreateSeqSBAIJ - Creates a sparse symmetric matrix in block AIJ (block
1715c464158bSHong Zhang    compressed row) format.  For good matrix assembly performance the
1716c464158bSHong Zhang    user should preallocate the matrix storage by setting the parameter nz
1717c464158bSHong Zhang    (or the array nnz).  By setting these parameters accurately, performance
1718c464158bSHong Zhang    during matrix assembly can be increased by more than a factor of 50.
171949b5e25fSSatish Balay 
1720c464158bSHong Zhang    Collective on MPI_Comm
1721c464158bSHong Zhang 
1722c464158bSHong Zhang    Input Parameters:
1723c464158bSHong Zhang +  comm - MPI communicator, set to PETSC_COMM_SELF
1724c464158bSHong Zhang .  bs - size of block
1725c464158bSHong Zhang .  m - number of rows, or number of columns
1726c464158bSHong Zhang .  nz - number of block nonzeros per block row (same for all rows)
1727744e8345SSatish Balay -  nnz - array containing the number of block nonzeros in the upper triangular plus
1728744e8345SSatish Balay          diagonal portion of each block (possibly different for each block row) or PETSC_NULL
1729c464158bSHong Zhang 
1730c464158bSHong Zhang    Output Parameter:
1731c464158bSHong Zhang .  A - the symmetric matrix
1732c464158bSHong Zhang 
1733c464158bSHong Zhang    Options Database Keys:
1734c464158bSHong Zhang .   -mat_no_unroll - uses code that does not unroll the loops in the
1735c464158bSHong Zhang                      block calculations (much slower)
1736c464158bSHong Zhang .    -mat_block_size - size of the blocks to use
1737c464158bSHong Zhang 
1738c464158bSHong Zhang    Level: intermediate
1739c464158bSHong Zhang 
1740c464158bSHong Zhang    Notes:
1741c464158bSHong Zhang 
1742c464158bSHong Zhang    Specify the preallocated storage with either nz or nnz (not both).
1743c464158bSHong Zhang    Set nz=PETSC_DEFAULT and nnz=PETSC_NULL for PETSc to control dynamic memory
1744c464158bSHong Zhang    allocation.  For additional details, see the users manual chapter on
1745c464158bSHong Zhang    matrices.
1746c464158bSHong Zhang 
174749a6f317SBarry Smith    If the nnz parameter is given then the nz parameter is ignored
174849a6f317SBarry Smith 
1749c464158bSHong Zhang .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateMPISBAIJ()
1750c464158bSHong Zhang @*/
1751be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreateSeqSBAIJ(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt nz,const PetscInt nnz[],Mat *A)
1752c464158bSHong Zhang {
1753dfbe8321SBarry Smith   PetscErrorCode ierr;
1754c464158bSHong Zhang 
1755c464158bSHong Zhang   PetscFunctionBegin;
1756f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,A);CHKERRQ(ierr);
1757f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*A,m,n,m,n);CHKERRQ(ierr);
1758c464158bSHong Zhang   ierr = MatSetType(*A,MATSEQSBAIJ);CHKERRQ(ierr);
1759ab93d7beSBarry Smith   ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(*A,bs,nz,(PetscInt*)nnz);CHKERRQ(ierr);
176049b5e25fSSatish Balay   PetscFunctionReturn(0);
176149b5e25fSSatish Balay }
176249b5e25fSSatish Balay 
17634a2ae208SSatish Balay #undef __FUNCT__
17644a2ae208SSatish Balay #define __FUNCT__ "MatDuplicate_SeqSBAIJ"
1765dfbe8321SBarry Smith PetscErrorCode MatDuplicate_SeqSBAIJ(Mat A,MatDuplicateOption cpvalues,Mat *B)
176649b5e25fSSatish Balay {
176749b5e25fSSatish Balay   Mat            C;
176849b5e25fSSatish Balay   Mat_SeqSBAIJ   *c,*a = (Mat_SeqSBAIJ*)A->data;
17696849ba73SBarry Smith   PetscErrorCode ierr;
1770b40805acSSatish Balay   PetscInt       i,mbs = a->mbs,nz = a->nz,bs2 =a->bs2;
177149b5e25fSSatish Balay 
177249b5e25fSSatish Balay   PetscFunctionBegin;
177329bbc08cSBarry Smith   if (a->i[mbs] != nz) SETERRQ(PETSC_ERR_PLIB,"Corrupt matrix");
177449b5e25fSSatish Balay 
177549b5e25fSSatish Balay   *B = 0;
1776f69a0ea3SMatthew Knepley   ierr = MatCreate(A->comm,&C);CHKERRQ(ierr);
1777899cda47SBarry Smith   ierr = MatSetSizes(C,A->rmap.N,A->cmap.n,A->rmap.N,A->cmap.n);CHKERRQ(ierr);
1778be5d1d56SKris Buschelman   ierr = MatSetType(C,A->type_name);CHKERRQ(ierr);
17791d5dac46SHong Zhang   ierr = PetscMemcpy(C->ops,A->ops,sizeof(struct _MatOps));CHKERRQ(ierr);
1780692f9cbeSHong Zhang   c    = (Mat_SeqSBAIJ*)C->data;
1781692f9cbeSHong Zhang 
1782273d9f13SBarry Smith   C->preallocated   = PETSC_TRUE;
178349b5e25fSSatish Balay   C->factor         = A->factor;
178449b5e25fSSatish Balay   c->row            = 0;
178549b5e25fSSatish Balay   c->icol           = 0;
178649b5e25fSSatish Balay   c->saved_values   = 0;
178749b5e25fSSatish Balay   c->keepzeroedrows = a->keepzeroedrows;
178849b5e25fSSatish Balay   C->assembled      = PETSC_TRUE;
178949b5e25fSSatish Balay 
1790899cda47SBarry Smith   ierr = PetscMapCopy(A->comm,&A->rmap,&C->rmap);CHKERRQ(ierr);
1791899cda47SBarry Smith   ierr = PetscMapCopy(A->comm,&A->cmap,&C->cmap);CHKERRQ(ierr);
179249b5e25fSSatish Balay   c->bs2  = a->bs2;
179349b5e25fSSatish Balay   c->mbs  = a->mbs;
179449b5e25fSSatish Balay   c->nbs  = a->nbs;
179549b5e25fSSatish Balay 
17968777fc3fSSatish Balay   ierr = PetscMalloc2((mbs+1),PetscInt,&c->imax,(mbs+1),PetscInt,&c->ilen);CHKERRQ(ierr);
179749b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
179849b5e25fSSatish Balay     c->imax[i] = a->imax[i];
179949b5e25fSSatish Balay     c->ilen[i] = a->ilen[i];
180049b5e25fSSatish Balay   }
180149b5e25fSSatish Balay 
180249b5e25fSSatish Balay   /* allocate the matrix space */
1803b40805acSSatish Balay   ierr = PetscMalloc3(bs2*nz,MatScalar,&c->a,nz,PetscInt,&c->j,mbs+1,PetscInt,&c->i);CHKERRQ(ierr);
180449b5e25fSSatish Balay   c->singlemalloc = PETSC_TRUE;
180513f74950SBarry Smith   ierr = PetscMemcpy(c->i,a->i,(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr);
1806b40805acSSatish Balay   ierr = PetscLogObjectMemory(C,(mbs+1)*sizeof(PetscInt) + nz*(bs2*sizeof(MatScalar) + sizeof(PetscInt)));CHKERRQ(ierr);
180749b5e25fSSatish Balay   if (mbs > 0) {
180813f74950SBarry Smith     ierr = PetscMemcpy(c->j,a->j,nz*sizeof(PetscInt));CHKERRQ(ierr);
180949b5e25fSSatish Balay     if (cpvalues == MAT_COPY_VALUES) {
181049b5e25fSSatish Balay       ierr = PetscMemcpy(c->a,a->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr);
181149b5e25fSSatish Balay     } else {
181249b5e25fSSatish Balay       ierr = PetscMemzero(c->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr);
181349b5e25fSSatish Balay     }
181449b5e25fSSatish Balay   }
181549b5e25fSSatish Balay 
181649b5e25fSSatish Balay   c->sorted      = a->sorted;
181749b5e25fSSatish Balay   c->roworiented = a->roworiented;
181849b5e25fSSatish Balay   c->nonew       = a->nonew;
181949b5e25fSSatish Balay 
182049b5e25fSSatish Balay   if (a->diag) {
182113f74950SBarry Smith     ierr = PetscMalloc((mbs+1)*sizeof(PetscInt),&c->diag);CHKERRQ(ierr);
182252e6d16bSBarry Smith     ierr = PetscLogObjectMemory(C,(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr);
182349b5e25fSSatish Balay     for (i=0; i<mbs; i++) {
182449b5e25fSSatish Balay       c->diag[i] = a->diag[i];
182549b5e25fSSatish Balay     }
182649b5e25fSSatish Balay   } else c->diag  = 0;
18276c6c5352SBarry Smith   c->nz           = a->nz;
18286c6c5352SBarry Smith   c->maxnz        = a->maxnz;
182949b5e25fSSatish Balay   c->solve_work   = 0;
183049b5e25fSSatish Balay   c->mult_work    = 0;
1831e6b907acSBarry Smith   c->free_a       = PETSC_TRUE;
1832e6b907acSBarry Smith   c->free_ij      = PETSC_TRUE;
183349b5e25fSSatish Balay   *B = C;
1834b0a32e0cSBarry Smith   ierr = PetscFListDuplicate(A->qlist,&C->qlist);CHKERRQ(ierr);
183549b5e25fSSatish Balay   PetscFunctionReturn(0);
183649b5e25fSSatish Balay }
183749b5e25fSSatish Balay 
18384a2ae208SSatish Balay #undef __FUNCT__
18394a2ae208SSatish Balay #define __FUNCT__ "MatLoad_SeqSBAIJ"
1840f69a0ea3SMatthew Knepley PetscErrorCode MatLoad_SeqSBAIJ(PetscViewer viewer, MatType type,Mat *A)
184149b5e25fSSatish Balay {
184249b5e25fSSatish Balay   Mat_SeqSBAIJ   *a;
184349b5e25fSSatish Balay   Mat            B;
18446849ba73SBarry Smith   PetscErrorCode ierr;
184513f74950SBarry Smith   int            fd;
184613f74950SBarry Smith   PetscMPIInt    size;
184713f74950SBarry Smith   PetscInt       i,nz,header[4],*rowlengths=0,M,N,bs=1;
184813f74950SBarry Smith   PetscInt       *mask,mbs,*jj,j,rowcount,nzcount,k,*s_browlengths,maskcount;
184913f74950SBarry Smith   PetscInt       kmax,jcount,block,idx,point,nzcountb,extra_rows;
185013f74950SBarry Smith   PetscInt       *masked,nmask,tmp,bs2,ishift;
185187828ca2SBarry Smith   PetscScalar    *aa;
185249b5e25fSSatish Balay   MPI_Comm       comm = ((PetscObject)viewer)->comm;
185349b5e25fSSatish Balay 
185449b5e25fSSatish Balay   PetscFunctionBegin;
1855b0a32e0cSBarry Smith   ierr = PetscOptionsGetInt(PETSC_NULL,"-matload_block_size",&bs,PETSC_NULL);CHKERRQ(ierr);
185649b5e25fSSatish Balay   bs2  = bs*bs;
185749b5e25fSSatish Balay 
185849b5e25fSSatish Balay   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
185929bbc08cSBarry Smith   if (size > 1) SETERRQ(PETSC_ERR_ARG_WRONG,"view must have one processor");
1860b0a32e0cSBarry Smith   ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr);
186149b5e25fSSatish Balay   ierr = PetscBinaryRead(fd,header,4,PETSC_INT);CHKERRQ(ierr);
1862552e946dSBarry Smith   if (header[0] != MAT_FILE_COOKIE) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"not Mat object");
186349b5e25fSSatish Balay   M = header[1]; N = header[2]; nz = header[3];
186449b5e25fSSatish Balay 
186549b5e25fSSatish Balay   if (header[3] < 0) {
186629bbc08cSBarry Smith     SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"Matrix stored in special format, cannot load as SeqSBAIJ");
186749b5e25fSSatish Balay   }
186849b5e25fSSatish Balay 
186929bbc08cSBarry Smith   if (M != N) SETERRQ(PETSC_ERR_SUP,"Can only do square matrices");
187049b5e25fSSatish Balay 
187149b5e25fSSatish Balay   /*
187249b5e25fSSatish Balay      This code adds extra rows to make sure the number of rows is
187349b5e25fSSatish Balay     divisible by the blocksize
187449b5e25fSSatish Balay   */
187549b5e25fSSatish Balay   mbs        = M/bs;
187649b5e25fSSatish Balay   extra_rows = bs - M + bs*(mbs);
187749b5e25fSSatish Balay   if (extra_rows == bs) extra_rows = 0;
187849b5e25fSSatish Balay   else                  mbs++;
187949b5e25fSSatish Balay   if (extra_rows) {
1880ae15b995SBarry Smith     ierr = PetscInfo(0,"Padding loaded matrix to match blocksize\n");CHKERRQ(ierr);
188149b5e25fSSatish Balay   }
188249b5e25fSSatish Balay 
188349b5e25fSSatish Balay   /* read in row lengths */
188413f74950SBarry Smith   ierr = PetscMalloc((M+extra_rows)*sizeof(PetscInt),&rowlengths);CHKERRQ(ierr);
188549b5e25fSSatish Balay   ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr);
188649b5e25fSSatish Balay   for (i=0; i<extra_rows; i++) rowlengths[M+i] = 1;
188749b5e25fSSatish Balay 
188849b5e25fSSatish Balay   /* read in column indices */
188913f74950SBarry Smith   ierr = PetscMalloc((nz+extra_rows)*sizeof(PetscInt),&jj);CHKERRQ(ierr);
189049b5e25fSSatish Balay   ierr = PetscBinaryRead(fd,jj,nz,PETSC_INT);CHKERRQ(ierr);
189149b5e25fSSatish Balay   for (i=0; i<extra_rows; i++) jj[nz+i] = M+i;
189249b5e25fSSatish Balay 
189349b5e25fSSatish Balay   /* loop over row lengths determining block row lengths */
189413f74950SBarry Smith   ierr     = PetscMalloc(mbs*sizeof(PetscInt),&s_browlengths);CHKERRQ(ierr);
189513f74950SBarry Smith   ierr     = PetscMemzero(s_browlengths,mbs*sizeof(PetscInt));CHKERRQ(ierr);
189613f74950SBarry Smith   ierr     = PetscMalloc(2*mbs*sizeof(PetscInt),&mask);CHKERRQ(ierr);
189713f74950SBarry Smith   ierr     = PetscMemzero(mask,mbs*sizeof(PetscInt));CHKERRQ(ierr);
189849b5e25fSSatish Balay   masked   = mask + mbs;
189949b5e25fSSatish Balay   rowcount = 0; nzcount = 0;
190049b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
190149b5e25fSSatish Balay     nmask = 0;
190249b5e25fSSatish Balay     for (j=0; j<bs; j++) {
190349b5e25fSSatish Balay       kmax = rowlengths[rowcount];
190449b5e25fSSatish Balay       for (k=0; k<kmax; k++) {
19052d703238SHong Zhang         tmp = jj[nzcount++]/bs;   /* block col. index */
190603630b6eSHong Zhang         if (!mask[tmp] && tmp >= i) {masked[nmask++] = tmp; mask[tmp] = 1;}
190749b5e25fSSatish Balay       }
190849b5e25fSSatish Balay       rowcount++;
190949b5e25fSSatish Balay     }
1910574b2666SHong Zhang     s_browlengths[i] += nmask;
1911574b2666SHong Zhang 
191249b5e25fSSatish Balay     /* zero out the mask elements we set */
191349b5e25fSSatish Balay     for (j=0; j<nmask; j++) mask[masked[j]] = 0;
191449b5e25fSSatish Balay   }
191549b5e25fSSatish Balay 
191649b5e25fSSatish Balay   /* create our matrix */
1917f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,&B);CHKERRQ(ierr);
1918f69a0ea3SMatthew Knepley   ierr = MatSetSizes(B,M+extra_rows,N+extra_rows,M+extra_rows,N+extra_rows);CHKERRQ(ierr);
19199abb65ffSKris Buschelman   ierr = MatSetType(B,type);CHKERRQ(ierr);
1920ab93d7beSBarry Smith   ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(B,bs,0,s_browlengths);CHKERRQ(ierr);
192149b5e25fSSatish Balay   a = (Mat_SeqSBAIJ*)B->data;
192249b5e25fSSatish Balay 
192349b5e25fSSatish Balay   /* set matrix "i" values */
192449b5e25fSSatish Balay   a->i[0] = 0;
192549b5e25fSSatish Balay   for (i=1; i<= mbs; i++) {
1926574b2666SHong Zhang     a->i[i]      = a->i[i-1] + s_browlengths[i-1];
1927574b2666SHong Zhang     a->ilen[i-1] = s_browlengths[i-1];
192849b5e25fSSatish Balay   }
19296c6c5352SBarry Smith   a->nz = a->i[mbs];
193049b5e25fSSatish Balay 
193149b5e25fSSatish Balay   /* read in nonzero values */
193287828ca2SBarry Smith   ierr = PetscMalloc((nz+extra_rows)*sizeof(PetscScalar),&aa);CHKERRQ(ierr);
193349b5e25fSSatish Balay   ierr = PetscBinaryRead(fd,aa,nz,PETSC_SCALAR);CHKERRQ(ierr);
193449b5e25fSSatish Balay   for (i=0; i<extra_rows; i++) aa[nz+i] = 1.0;
193549b5e25fSSatish Balay 
193649b5e25fSSatish Balay   /* set "a" and "j" values into matrix */
193749b5e25fSSatish Balay   nzcount = 0; jcount = 0;
193849b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
193949b5e25fSSatish Balay     nzcountb = nzcount;
194049b5e25fSSatish Balay     nmask    = 0;
194149b5e25fSSatish Balay     for (j=0; j<bs; j++) {
194249b5e25fSSatish Balay       kmax = rowlengths[i*bs+j];
194349b5e25fSSatish Balay       for (k=0; k<kmax; k++) {
19442d703238SHong Zhang         tmp = jj[nzcount++]/bs; /* block col. index */
194503630b6eSHong Zhang         if (!mask[tmp] && tmp >= i) { masked[nmask++] = tmp; mask[tmp] = 1;}
19462d703238SHong Zhang       }
19472d703238SHong Zhang     }
19482d703238SHong Zhang     /* sort the masked values */
19492d703238SHong Zhang     ierr = PetscSortInt(nmask,masked);CHKERRQ(ierr);
19502d703238SHong Zhang 
19512d703238SHong Zhang     /* set "j" values into matrix */
19522d703238SHong Zhang     maskcount = 1;
19532d703238SHong Zhang     for (j=0; j<nmask; j++) {
195449b5e25fSSatish Balay       a->j[jcount++]  = masked[j];
195549b5e25fSSatish Balay       mask[masked[j]] = maskcount++;
195649b5e25fSSatish Balay     }
1957574b2666SHong Zhang 
195849b5e25fSSatish Balay     /* set "a" values into matrix */
195949b5e25fSSatish Balay     ishift = bs2*a->i[i];
196049b5e25fSSatish Balay     for (j=0; j<bs; j++) {
196149b5e25fSSatish Balay       kmax = rowlengths[i*bs+j];
196249b5e25fSSatish Balay       for (k=0; k<kmax; k++) {
1963574b2666SHong Zhang         tmp       = jj[nzcountb]/bs ; /* block col. index */
1964574b2666SHong Zhang         if (tmp >= i){
196549b5e25fSSatish Balay           block     = mask[tmp] - 1;
196649b5e25fSSatish Balay           point     = jj[nzcountb] - bs*tmp;
196749b5e25fSSatish Balay           idx       = ishift + bs2*block + j + bs*point;
1968574b2666SHong Zhang           a->a[idx] = aa[nzcountb];
1969574b2666SHong Zhang         }
1970574b2666SHong Zhang         nzcountb++;
197149b5e25fSSatish Balay       }
197249b5e25fSSatish Balay     }
197349b5e25fSSatish Balay     /* zero out the mask elements we set */
197449b5e25fSSatish Balay     for (j=0; j<nmask; j++) mask[masked[j]] = 0;
197549b5e25fSSatish Balay   }
19766c6c5352SBarry Smith   if (jcount != a->nz) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"Bad binary matrix");
197749b5e25fSSatish Balay 
197849b5e25fSSatish Balay   ierr = PetscFree(rowlengths);CHKERRQ(ierr);
1979574b2666SHong Zhang   ierr = PetscFree(s_browlengths);CHKERRQ(ierr);
198049b5e25fSSatish Balay   ierr = PetscFree(aa);CHKERRQ(ierr);
198149b5e25fSSatish Balay   ierr = PetscFree(jj);CHKERRQ(ierr);
198249b5e25fSSatish Balay   ierr = PetscFree(mask);CHKERRQ(ierr);
198349b5e25fSSatish Balay 
19849abb65ffSKris Buschelman   ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
19859abb65ffSKris Buschelman   ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
198649b5e25fSSatish Balay   ierr = MatView_Private(B);CHKERRQ(ierr);
19879abb65ffSKris Buschelman   *A = B;
198849b5e25fSSatish Balay   PetscFunctionReturn(0);
198949b5e25fSSatish Balay }
1990574b2666SHong Zhang 
1991d06b337dSHong Zhang #undef __FUNCT__
1992d06b337dSHong Zhang #define __FUNCT__ "MatRelax_SeqSBAIJ"
199313f74950SBarry Smith PetscErrorCode MatRelax_SeqSBAIJ(Mat A,Vec bb,PetscReal omega,MatSORType flag,PetscReal fshift,PetscInt its,PetscInt lits,Vec xx)
1994d06b337dSHong Zhang {
1995d06b337dSHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
1996d06b337dSHong Zhang   MatScalar      *aa=a->a,*v,*v1;
1997d06b337dSHong Zhang   PetscScalar    *x,*b,*t,sum,d;
19986849ba73SBarry Smith   PetscErrorCode ierr;
1999899cda47SBarry Smith   PetscInt       m=a->mbs,bs=A->rmap.bs,*ai=a->i,*aj=a->j;
2000521d7252SBarry Smith   PetscInt       nz,nz1,*vj,*vj1,i;
2001d06b337dSHong Zhang 
2002d06b337dSHong Zhang   PetscFunctionBegin;
200351f519a2SBarry Smith   if (flag & SOR_EISENSTAT) SETERRQ(PETSC_ERR_SUP,"No support yet for Eisenstat");
200451f519a2SBarry Smith 
2005b965ef7fSBarry Smith   its = its*lits;
200677431f27SBarry Smith   if (its <= 0) SETERRQ2(PETSC_ERR_ARG_WRONG,"Relaxation requires global its %D and local its %D both positive",its,lits);
2007d06b337dSHong Zhang 
2008d06b337dSHong Zhang   if (bs > 1)
2009d06b337dSHong Zhang     SETERRQ(PETSC_ERR_SUP,"SSOR for block size > 1 is not yet implemented");
2010d06b337dSHong Zhang 
20111ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
2012d06b337dSHong Zhang   if (xx != bb) {
20131ebc52fbSHong Zhang     ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
2014d06b337dSHong Zhang   } else {
2015d06b337dSHong Zhang     b = x;
2016d06b337dSHong Zhang   }
2017d06b337dSHong Zhang 
2018e2ee2a47SBarry Smith   if (!a->relax_work) {
2019e2ee2a47SBarry Smith     ierr = PetscMalloc(m*sizeof(PetscScalar),&a->relax_work);CHKERRQ(ierr);
2020e2ee2a47SBarry Smith   }
2021e2ee2a47SBarry Smith   t = a->relax_work;
2022d06b337dSHong Zhang 
2023d06b337dSHong Zhang   if (flag & SOR_ZERO_INITIAL_GUESS) {
20242798e883SHong Zhang     if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){
2025*290bbb0aSBarry Smith       ierr = PetscMemcpy(t,b,m*sizeof(PetscScalar));CHKERRQ(ierr);
2026d06b337dSHong Zhang 
2027d06b337dSHong Zhang       for (i=0; i<m; i++){
202844706875SHong Zhang         d  = *(aa + ai[i]);  /* diag[i] */
2029d06b337dSHong Zhang         v  = aa + ai[i] + 1;
2030d06b337dSHong Zhang         vj = aj + ai[i] + 1;
2031d06b337dSHong Zhang         nz = ai[i+1] - ai[i] - 1;
2032e6b907acSBarry Smith         ierr = PetscLogFlops(2*nz-1);CHKERRQ(ierr);
2033d06b337dSHong Zhang         x[i] = omega*t[i]/d;
2034d06b337dSHong Zhang         while (nz--) t[*vj++] -= x[i]*(*v++); /* update rhs */
2035d06b337dSHong Zhang       }
2036d06b337dSHong Zhang     }
2037d06b337dSHong Zhang 
20382798e883SHong Zhang     if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){
203995d750ceSBarry Smith       if (!(flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP)){
204095d750ceSBarry Smith         t = b;
2041d06b337dSHong Zhang       }
204295d750ceSBarry Smith 
2043d06b337dSHong Zhang       for (i=m-1; i>=0; i--){
2044d06b337dSHong Zhang         d  = *(aa + ai[i]);
2045d06b337dSHong Zhang         v  = aa + ai[i] + 1;
2046d06b337dSHong Zhang         vj = aj + ai[i] + 1;
2047d06b337dSHong Zhang         nz = ai[i+1] - ai[i] - 1;
2048e6b907acSBarry Smith         ierr = PetscLogFlops(2*nz-1);CHKERRQ(ierr);
2049d06b337dSHong Zhang         sum = t[i];
2050d06b337dSHong Zhang         while (nz--) sum -= x[*vj++]*(*v++);
2051d06b337dSHong Zhang         x[i] =   (1-omega)*x[i] + omega*sum/d;
2052d06b337dSHong Zhang       }
205395d750ceSBarry Smith       t = a->relax_work;
2054d06b337dSHong Zhang     }
2055d06b337dSHong Zhang     its--;
2056d06b337dSHong Zhang   }
2057d06b337dSHong Zhang 
2058d06b337dSHong Zhang   while (its--) {
205944706875SHong Zhang     /*
206044706875SHong Zhang        forward sweep:
206144706875SHong Zhang        for i=0,...,m-1:
206244706875SHong Zhang          sum[i] = (b[i] - U(i,:)x )/d[i];
206344706875SHong Zhang          x[i]   = (1-omega)x[i] + omega*sum[i];
206444706875SHong Zhang          b      = b - x[i]*U^T(i,:);
2065d06b337dSHong Zhang 
206644706875SHong Zhang     */
20672798e883SHong Zhang     if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){
2068*290bbb0aSBarry Smith       ierr = PetscMemcpy(t,b,m*sizeof(PetscScalar));CHKERRQ(ierr);
2069d06b337dSHong Zhang 
2070d06b337dSHong Zhang       for (i=0; i<m; i++){
207144706875SHong Zhang         d  = *(aa + ai[i]);  /* diag[i] */
2072d06b337dSHong Zhang         v  = aa + ai[i] + 1; v1=v;
2073d06b337dSHong Zhang         vj = aj + ai[i] + 1; vj1=vj;
2074d06b337dSHong Zhang         nz = ai[i+1] - ai[i] - 1; nz1=nz;
2075d06b337dSHong Zhang         sum = t[i];
2076e6b907acSBarry Smith         ierr = PetscLogFlops(4*nz-2);CHKERRQ(ierr);
2077d06b337dSHong Zhang         while (nz1--) sum -= (*v1++)*x[*vj1++];
2078d06b337dSHong Zhang         x[i] = (1-omega)*x[i] + omega*sum/d;
2079d06b337dSHong Zhang         while (nz--) t[*vj++] -= x[i]*(*v++);
2080d06b337dSHong Zhang       }
2081d06b337dSHong Zhang     }
2082d06b337dSHong Zhang 
20832798e883SHong Zhang     if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){
208444706875SHong Zhang       /*
208544706875SHong Zhang        backward sweep:
208644706875SHong Zhang        b = b - x[i]*U^T(i,:), i=0,...,n-2
208744706875SHong Zhang        for i=m-1,...,0:
208844706875SHong Zhang          sum[i] = (b[i] - U(i,:)x )/d[i];
208944706875SHong Zhang          x[i]   = (1-omega)x[i] + omega*sum[i];
209044706875SHong Zhang       */
209195d750ceSBarry Smith       /* if there was a forward sweep done above then I thing the next two for loops are not needed */
2092*290bbb0aSBarry Smith       ierr = PetscMemcpy(t,b,m*sizeof(PetscScalar));CHKERRQ(ierr);
2093d06b337dSHong Zhang 
2094d06b337dSHong Zhang       for (i=0; i<m-1; i++){  /* update rhs */
2095d06b337dSHong Zhang         v  = aa + ai[i] + 1;
2096d06b337dSHong Zhang         vj = aj + ai[i] + 1;
2097d06b337dSHong Zhang         nz = ai[i+1] - ai[i] - 1;
2098efee365bSSatish Balay         ierr = PetscLogFlops(2*nz-1);CHKERRQ(ierr);
2099e6b907acSBarry Smith         while (nz--) t[*vj++] -= x[i]*(*v++);
2100d06b337dSHong Zhang       }
2101d06b337dSHong Zhang       for (i=m-1; i>=0; i--){
2102d06b337dSHong Zhang         d  = *(aa + ai[i]);
2103d06b337dSHong Zhang         v  = aa + ai[i] + 1;
2104d06b337dSHong Zhang         vj = aj + ai[i] + 1;
2105d06b337dSHong Zhang         nz = ai[i+1] - ai[i] - 1;
2106e6b907acSBarry Smith         ierr = PetscLogFlops(2*nz-1);CHKERRQ(ierr);
2107d06b337dSHong Zhang         sum = t[i];
2108d06b337dSHong Zhang         while (nz--) sum -= x[*vj++]*(*v++);
2109d06b337dSHong Zhang         x[i] =   (1-omega)*x[i] + omega*sum/d;
2110d06b337dSHong Zhang       }
2111d06b337dSHong Zhang     }
2112d06b337dSHong Zhang   }
2113d06b337dSHong Zhang 
21141ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
2115d06b337dSHong Zhang   if (bb != xx) {
21161ebc52fbSHong Zhang     ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
2117d06b337dSHong Zhang   }
2118d06b337dSHong Zhang   PetscFunctionReturn(0);
2119d06b337dSHong Zhang }
2120d06b337dSHong Zhang 
2121c75a6043SHong Zhang #undef __FUNCT__
2122c75a6043SHong Zhang #define __FUNCT__ "MatCreateSeqSBAIJWithArrays"
2123c75a6043SHong Zhang /*@
2124c75a6043SHong Zhang      MatCreateSeqSBAIJWithArrays - Creates an sequential SBAIJ matrix using matrix elements
2125c75a6043SHong Zhang               (upper triangular entries in CSR format) provided by the user.
2126c75a6043SHong Zhang 
2127c75a6043SHong Zhang      Collective on MPI_Comm
2128c75a6043SHong Zhang 
2129c75a6043SHong Zhang    Input Parameters:
2130c75a6043SHong Zhang +  comm - must be an MPI communicator of size 1
2131c75a6043SHong Zhang .  bs - size of block
2132c75a6043SHong Zhang .  m - number of rows
2133c75a6043SHong Zhang .  n - number of columns
2134c75a6043SHong Zhang .  i - row indices
2135c75a6043SHong Zhang .  j - column indices
2136c75a6043SHong Zhang -  a - matrix values
2137c75a6043SHong Zhang 
2138c75a6043SHong Zhang    Output Parameter:
2139c75a6043SHong Zhang .  mat - the matrix
2140c75a6043SHong Zhang 
2141c75a6043SHong Zhang    Level: intermediate
2142c75a6043SHong Zhang 
2143c75a6043SHong Zhang    Notes:
2144c75a6043SHong Zhang        The i, j, and a arrays are not copied by this routine, the user must free these arrays
2145c75a6043SHong Zhang     once the matrix is destroyed
2146c75a6043SHong Zhang 
2147c75a6043SHong Zhang        You cannot set new nonzero locations into this matrix, that will generate an error.
2148c75a6043SHong Zhang 
2149c75a6043SHong Zhang        The i and j indices are 0 based
2150c75a6043SHong Zhang 
2151c75a6043SHong Zhang .seealso: MatCreate(), MatCreateMPISBAIJ(), MatCreateSeqSBAIJ()
2152c75a6043SHong Zhang 
2153c75a6043SHong Zhang @*/
2154c75a6043SHong Zhang PetscErrorCode PETSCMAT_DLLEXPORT MatCreateSeqSBAIJWithArrays(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt* i,PetscInt*j,PetscScalar *a,Mat *mat)
2155c75a6043SHong Zhang {
2156c75a6043SHong Zhang   PetscErrorCode ierr;
2157c75a6043SHong Zhang   PetscInt       ii;
2158c75a6043SHong Zhang   Mat_SeqSBAIJ   *sbaij;
2159c75a6043SHong Zhang 
2160c75a6043SHong Zhang   PetscFunctionBegin;
2161c75a6043SHong Zhang   if (bs != 1) SETERRQ1(PETSC_ERR_SUP,"block size %D > 1 is not supported yet",bs);
2162c75a6043SHong Zhang   if (i[0]) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"i (row indices) must start with 0");
2163c75a6043SHong Zhang 
2164c75a6043SHong Zhang   ierr = MatCreate(comm,mat);CHKERRQ(ierr);
2165c75a6043SHong Zhang   ierr = MatSetSizes(*mat,m,n,m,n);CHKERRQ(ierr);
2166c75a6043SHong Zhang   ierr = MatSetType(*mat,MATSEQSBAIJ);CHKERRQ(ierr);
2167c75a6043SHong Zhang   ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(*mat,bs,MAT_SKIP_ALLOCATION,0);CHKERRQ(ierr);
2168c75a6043SHong Zhang   sbaij = (Mat_SeqSBAIJ*)(*mat)->data;
2169c75a6043SHong Zhang   ierr = PetscMalloc2(m,PetscInt,&sbaij->imax,m,PetscInt,&sbaij->ilen);CHKERRQ(ierr);
2170c75a6043SHong Zhang 
2171c75a6043SHong Zhang   sbaij->i = i;
2172c75a6043SHong Zhang   sbaij->j = j;
2173c75a6043SHong Zhang   sbaij->a = a;
2174c75a6043SHong Zhang   sbaij->singlemalloc = PETSC_FALSE;
2175c75a6043SHong Zhang   sbaij->nonew        = -1;             /*this indicates that inserting a new value in the matrix that generates a new nonzero is an error*/
2176e6b907acSBarry Smith   sbaij->free_a       = PETSC_FALSE;
2177e6b907acSBarry Smith   sbaij->free_ij      = PETSC_FALSE;
2178c75a6043SHong Zhang 
2179c75a6043SHong Zhang   for (ii=0; ii<m; ii++) {
2180c75a6043SHong Zhang     sbaij->ilen[ii] = sbaij->imax[ii] = i[ii+1] - i[ii];
2181c75a6043SHong Zhang #if defined(PETSC_USE_DEBUG)
2182c75a6043SHong 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]);
2183c75a6043SHong Zhang #endif
2184c75a6043SHong Zhang   }
2185c75a6043SHong Zhang #if defined(PETSC_USE_DEBUG)
2186c75a6043SHong Zhang   for (ii=0; ii<sbaij->i[m]; ii++) {
2187c75a6043SHong Zhang     if (j[ii] < 0) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Negative column index at location = %d index = %d",ii,j[ii]);
2188c75a6043SHong Zhang     if (j[ii] > n - 1) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Column index to large at location = %d index = %d",ii,j[ii]);
2189c75a6043SHong Zhang   }
2190c75a6043SHong Zhang #endif
2191c75a6043SHong Zhang 
2192c75a6043SHong Zhang   ierr = MatAssemblyBegin(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2193c75a6043SHong Zhang   ierr = MatAssemblyEnd(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2194c75a6043SHong Zhang   PetscFunctionReturn(0);
2195c75a6043SHong Zhang }
2196d06b337dSHong Zhang 
2197d06b337dSHong Zhang 
2198d06b337dSHong Zhang 
219949b5e25fSSatish Balay 
220049b5e25fSSatish Balay 
2201