xref: /petsc/src/mat/impls/sbaij/seq/sbaij.c (revision 121deb677e391b9f6da12621925c77122ab08559)
149b5e25fSSatish Balay 
249b5e25fSSatish Balay /*
3a1373b80SHong Zhang     Defines the basic matrix operations for the SBAIJ (compressed row)
449b5e25fSSatish Balay   matrix storage format.
549b5e25fSSatish Balay */
6c6db04a5SJed Brown #include <../src/mat/impls/baij/seq/baij.h>         /*I "petscmat.h" I*/
7c6db04a5SJed Brown #include <../src/mat/impls/sbaij/seq/sbaij.h>
8c6db04a5SJed Brown #include <petscblaslapack.h>
949b5e25fSSatish Balay 
10c6db04a5SJed Brown #include <../src/mat/impls/sbaij/seq/relax.h>
1170dcbbb9SBarry Smith #define USESHORT
12c6db04a5SJed Brown #include <../src/mat/impls/sbaij/seq/relax.h>
1370dcbbb9SBarry Smith 
14ace3abfcSBarry Smith extern PetscErrorCode MatSeqSBAIJSetNumericFactorization_inplace(Mat,PetscBool );
15b5b17502SBarry Smith 
1649b5e25fSSatish Balay /*
1749b5e25fSSatish Balay      Checks for missing diagonals
1849b5e25fSSatish Balay */
194a2ae208SSatish Balay #undef __FUNCT__
204a2ae208SSatish Balay #define __FUNCT__ "MatMissingDiagonal_SeqSBAIJ"
21ace3abfcSBarry Smith PetscErrorCode MatMissingDiagonal_SeqSBAIJ(Mat A,PetscBool  *missing,PetscInt *dd)
2249b5e25fSSatish Balay {
23045c9aa0SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
246849ba73SBarry Smith   PetscErrorCode ierr;
2513f74950SBarry Smith   PetscInt       *diag,*jj = a->j,i;
2649b5e25fSSatish Balay 
2749b5e25fSSatish Balay   PetscFunctionBegin;
28045c9aa0SHong Zhang   ierr = MatMarkDiagonal_SeqSBAIJ(A);CHKERRQ(ierr);
2949b5e25fSSatish Balay   diag = a->diag;
302af78befSBarry Smith   *missing = PETSC_FALSE;
3149b5e25fSSatish Balay   for (i=0; i<a->mbs; i++) {
322af78befSBarry Smith     if (jj[diag[i]] != i) {
332af78befSBarry Smith       *missing    = PETSC_TRUE;
342af78befSBarry Smith       if (dd) *dd = i;
352af78befSBarry Smith       break;
362af78befSBarry Smith     }
3749b5e25fSSatish Balay   }
3849b5e25fSSatish Balay   PetscFunctionReturn(0);
3949b5e25fSSatish Balay }
4049b5e25fSSatish Balay 
414a2ae208SSatish Balay #undef __FUNCT__
424a2ae208SSatish Balay #define __FUNCT__ "MatMarkDiagonal_SeqSBAIJ"
43dfbe8321SBarry Smith PetscErrorCode MatMarkDiagonal_SeqSBAIJ(Mat A)
4449b5e25fSSatish Balay {
45045c9aa0SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
466849ba73SBarry Smith   PetscErrorCode ierr;
4709f38230SBarry Smith   PetscInt       i;
4849b5e25fSSatish Balay 
4949b5e25fSSatish Balay   PetscFunctionBegin;
5009f38230SBarry Smith   if (!a->diag) {
5109f38230SBarry Smith     ierr = PetscMalloc(a->mbs*sizeof(PetscInt),&a->diag);CHKERRQ(ierr);
52c760cd28SBarry Smith     ierr = PetscLogObjectMemory(A,a->mbs*sizeof(PetscInt));CHKERRQ(ierr);
53c760cd28SBarry Smith     a->free_diag = PETSC_TRUE;
5409f38230SBarry Smith   }
5509f38230SBarry Smith   for (i=0; i<a->mbs; i++) a->diag[i] = a->i[i];
5649b5e25fSSatish Balay   PetscFunctionReturn(0);
5749b5e25fSSatish Balay }
5849b5e25fSSatish Balay 
594a2ae208SSatish Balay #undef __FUNCT__
604a2ae208SSatish Balay #define __FUNCT__ "MatGetRowIJ_SeqSBAIJ"
61ace3abfcSBarry Smith static PetscErrorCode MatGetRowIJ_SeqSBAIJ(Mat A,PetscInt oshift,PetscBool  symmetric,PetscBool  blockcompressed,PetscInt *nn,PetscInt *ia[],PetscInt *ja[],PetscBool  *done)
6249b5e25fSSatish Balay {
63a6ece127SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
64d0f46423SBarry Smith   PetscInt     i,j,n = a->mbs,nz = a->i[n],bs = A->rmap->bs;
658f7157efSSatish Balay   PetscErrorCode ierr;
6649b5e25fSSatish Balay 
6749b5e25fSSatish Balay   PetscFunctionBegin;
68d3e5a4abSHong Zhang   *nn = n;
69a1373b80SHong Zhang   if (!ia) PetscFunctionReturn(0);
708f7157efSSatish Balay   if (!blockcompressed) {
718f7157efSSatish Balay     /* malloc & create the natural set of indices */
72f1d0d59dSSatish Balay     ierr = PetscMalloc2((n+1)*bs,PetscInt,ia,nz*bs,PetscInt,ja);CHKERRQ(ierr);
738f7157efSSatish Balay     for (i=0; i<n+1; i++) {
748f7157efSSatish Balay       for (j=0; j<bs; j++) {
758f7157efSSatish Balay         *ia[i*bs+j] = a->i[i]*bs+j+oshift;
768f7157efSSatish Balay       }
778f7157efSSatish Balay     }
788f7157efSSatish Balay     for (i=0; i<nz; i++) {
798f7157efSSatish Balay       for (j=0; j<bs; j++) {
808f7157efSSatish Balay         *ja[i*bs+j] = a->j[i]*bs+j+oshift;
818f7157efSSatish Balay       }
828f7157efSSatish Balay     }
838f7157efSSatish Balay   } else { /* blockcompressed */
84a6ece127SHong Zhang     if (oshift == 1) {
85a6ece127SHong Zhang       /* temporarily add 1 to i and j indices */
866c6c5352SBarry Smith       for (i=0; i<nz; i++) a->j[i]++;
87a1373b80SHong Zhang       for (i=0; i<n+1; i++) a->i[i]++;
888f7157efSSatish Balay     }
89a1373b80SHong Zhang     *ia = a->i; *ja = a->j;
90a6ece127SHong Zhang   }
918f7157efSSatish Balay 
9249b5e25fSSatish Balay   PetscFunctionReturn(0);
9349b5e25fSSatish Balay }
9449b5e25fSSatish Balay 
954a2ae208SSatish Balay #undef __FUNCT__
964a2ae208SSatish Balay #define __FUNCT__ "MatRestoreRowIJ_SeqSBAIJ"
97ace3abfcSBarry Smith static PetscErrorCode MatRestoreRowIJ_SeqSBAIJ(Mat A,PetscInt oshift,PetscBool  symmetric,PetscBool  blockcompressed,PetscInt *nn,PetscInt *ia[],PetscInt *ja[],PetscBool  *done)
9849b5e25fSSatish Balay {
99b7aaefc3SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
1008f7157efSSatish Balay   PetscInt     i,n = a->mbs,nz = a->i[n];
1018f7157efSSatish Balay   PetscErrorCode ierr;
102a6ece127SHong Zhang 
10349b5e25fSSatish Balay   PetscFunctionBegin;
10449b5e25fSSatish Balay   if (!ia) PetscFunctionReturn(0);
105a6ece127SHong Zhang 
1068f7157efSSatish Balay   if (!blockcompressed) {
1078f7157efSSatish Balay     ierr = PetscFree2(*ia,*ja);CHKERRQ(ierr);
1088f7157efSSatish Balay   } else if (oshift == 1) { /* blockcompressed */
1096c6c5352SBarry Smith     for (i=0; i<nz; i++) a->j[i]--;
110a6ece127SHong Zhang     for (i=0; i<n+1; i++) a->i[i]--;
111a6ece127SHong Zhang   }
1128f7157efSSatish Balay 
113a6ece127SHong Zhang   PetscFunctionReturn(0);
11449b5e25fSSatish Balay }
11549b5e25fSSatish Balay 
1164a2ae208SSatish Balay #undef __FUNCT__
1174a2ae208SSatish Balay #define __FUNCT__ "MatDestroy_SeqSBAIJ"
118dfbe8321SBarry Smith PetscErrorCode MatDestroy_SeqSBAIJ(Mat A)
11949b5e25fSSatish Balay {
12049b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
121dfbe8321SBarry Smith   PetscErrorCode ierr;
12249b5e25fSSatish Balay 
12349b5e25fSSatish Balay   PetscFunctionBegin;
124a9f03627SSatish Balay #if defined(PETSC_USE_LOG)
125d0f46423SBarry Smith   PetscLogObjectState((PetscObject)A,"Rows=%D, NZ=%D",A->rmap->N,a->nz);
126a9f03627SSatish Balay #endif
127e6b907acSBarry Smith   ierr = MatSeqXAIJFreeAIJ(A,&a->a,&a->j,&a->i);CHKERRQ(ierr);
1287f53bb6cSHong Zhang   if (a->free_diag){ierr = PetscFree(a->diag);CHKERRQ(ierr);}
1296bf464f9SBarry Smith   ierr = ISDestroy(&a->row);CHKERRQ(ierr);
1306bf464f9SBarry Smith   ierr = ISDestroy(&a->col);CHKERRQ(ierr);
1316bf464f9SBarry Smith   ierr = ISDestroy(&a->icol);CHKERRQ(ierr);
132c31cb41cSBarry Smith   ierr = PetscFree(a->idiag);CHKERRQ(ierr);
133c31cb41cSBarry Smith   ierr = PetscFree(a->inode.size);CHKERRQ(ierr);
134c31cb41cSBarry Smith   ierr = PetscFree(a->diag);CHKERRQ(ierr);
135c760cd28SBarry Smith   if (a->free_imax_ilen) {ierr = PetscFree2(a->imax,a->ilen);CHKERRQ(ierr);}
13605b42c5fSBarry Smith   ierr = PetscFree(a->solve_work);CHKERRQ(ierr);
13741f059aeSBarry Smith   ierr = PetscFree(a->sor_work);CHKERRQ(ierr);
13805b42c5fSBarry Smith   ierr = PetscFree(a->solves_work);CHKERRQ(ierr);
13905b42c5fSBarry Smith   ierr = PetscFree(a->mult_work);CHKERRQ(ierr);
14005b42c5fSBarry Smith   ierr = PetscFree(a->saved_values);CHKERRQ(ierr);
14105b42c5fSBarry Smith   ierr = PetscFree(a->xtoy);CHKERRQ(ierr);
1424da8f245SBarry Smith   if (a->free_jshort) {ierr = PetscFree(a->jshort);CHKERRQ(ierr);}
1431a3463dfSHong Zhang   ierr = PetscFree(a->inew);CHKERRQ(ierr);
1446bf464f9SBarry Smith   ierr = MatDestroy(&a->parent);CHKERRQ(ierr);
145bf0cc555SLisandro Dalcin   ierr = PetscFree(A->data);CHKERRQ(ierr);
146901853e0SKris Buschelman 
147dbd8c25aSHong Zhang   ierr = PetscObjectChangeTypeName((PetscObject)A,0);CHKERRQ(ierr);
148901853e0SKris Buschelman   ierr = PetscObjectComposeFunction((PetscObject)A,"MatStoreValues_C","",PETSC_NULL);CHKERRQ(ierr);
149901853e0SKris Buschelman   ierr = PetscObjectComposeFunction((PetscObject)A,"MatRetrieveValues_C","",PETSC_NULL);CHKERRQ(ierr);
150901853e0SKris Buschelman   ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetColumnIndices_C","",PETSC_NULL);CHKERRQ(ierr);
151901853e0SKris Buschelman   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqaij_C","",PETSC_NULL);CHKERRQ(ierr);
152901853e0SKris Buschelman   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqbaij_C","",PETSC_NULL);CHKERRQ(ierr);
153901853e0SKris Buschelman   ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetPreallocation_C","",PETSC_NULL);CHKERRQ(ierr);
154aa5a9175SDahai Guo   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqsbstrm_C","",PETSC_NULL);CHKERRQ(ierr);
15549b5e25fSSatish Balay   PetscFunctionReturn(0);
15649b5e25fSSatish Balay }
15749b5e25fSSatish Balay 
1584a2ae208SSatish Balay #undef __FUNCT__
1594a2ae208SSatish Balay #define __FUNCT__ "MatSetOption_SeqSBAIJ"
160ace3abfcSBarry Smith PetscErrorCode MatSetOption_SeqSBAIJ(Mat A,MatOption op,PetscBool  flg)
16149b5e25fSSatish Balay {
162045c9aa0SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
16363ba0a88SBarry Smith   PetscErrorCode ierr;
16449b5e25fSSatish Balay 
16549b5e25fSSatish Balay   PetscFunctionBegin;
1664d9d31abSKris Buschelman   switch (op) {
1674d9d31abSKris Buschelman   case MAT_ROW_ORIENTED:
1684e0d8c25SBarry Smith     a->roworiented = flg;
1694d9d31abSKris Buschelman     break;
170a9817697SBarry Smith   case MAT_KEEP_NONZERO_PATTERN:
171a9817697SBarry Smith     a->keepnonzeropattern = flg;
1724d9d31abSKris Buschelman     break;
173512a5fc5SBarry Smith   case MAT_NEW_NONZERO_LOCATIONS:
174512a5fc5SBarry Smith     a->nonew = (flg ? 0 : 1);
1754d9d31abSKris Buschelman     break;
1764d9d31abSKris Buschelman   case MAT_NEW_NONZERO_LOCATION_ERR:
1774e0d8c25SBarry Smith     a->nonew = (flg ? -1 : 0);
1784d9d31abSKris Buschelman     break;
1794d9d31abSKris Buschelman   case MAT_NEW_NONZERO_ALLOCATION_ERR:
1804e0d8c25SBarry Smith     a->nonew = (flg ? -2 : 0);
1814d9d31abSKris Buschelman     break;
18228b2fa4aSMatthew Knepley   case MAT_UNUSED_NONZERO_LOCATION_ERR:
18328b2fa4aSMatthew Knepley     a->nounused = (flg ? -1 : 0);
18428b2fa4aSMatthew Knepley     break;
1854e0d8c25SBarry Smith   case MAT_NEW_DIAGONALS:
1864d9d31abSKris Buschelman   case MAT_IGNORE_OFF_PROC_ENTRIES:
1874d9d31abSKris Buschelman   case MAT_USE_HASH_TABLE:
188290bbb0aSBarry Smith     ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr);
1894d9d31abSKris Buschelman     break;
1909a4540c5SBarry Smith   case MAT_HERMITIAN:
191e32f2f54SBarry Smith     if (!A->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Must call MatAssemblyEnd() first");
1920bc54ff2SBarry Smith     if (A->cmap->n < 65536 && A->cmap->bs == 1) {
193eeffb40dSHong Zhang       A->ops->mult = MatMult_SeqSBAIJ_1_Hermitian_ushort;
1940bc54ff2SBarry Smith     } else if (A->cmap->bs == 1) {
195eeffb40dSHong Zhang       A->ops->mult = MatMult_SeqSBAIJ_1_Hermitian;
196e32f2f54SBarry Smith     } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No support for Hermitian with block size greater than 1");
197eeffb40dSHong Zhang     break;
1983d472b54SHong Zhang   case MAT_SPD:
1993d472b54SHong Zhang     A->spd_set                         = PETSC_TRUE;
2003d472b54SHong Zhang     A->spd                             = flg;
2013d472b54SHong Zhang     if (flg) {
2023d472b54SHong Zhang       A->symmetric                     = PETSC_TRUE;
2033d472b54SHong Zhang       A->structurally_symmetric        = PETSC_TRUE;
2043d472b54SHong Zhang       A->symmetric_set                 = PETSC_TRUE;
2053d472b54SHong Zhang       A->structurally_symmetric_set    = PETSC_TRUE;
2063d472b54SHong Zhang     }
2073d472b54SHong Zhang     break;
20877e54ba9SKris Buschelman   case MAT_SYMMETRIC:
20977e54ba9SKris Buschelman   case MAT_STRUCTURALLY_SYMMETRIC:
2109a4540c5SBarry Smith   case MAT_SYMMETRY_ETERNAL:
211e32f2f54SBarry Smith     if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix must be symmetric");
212290bbb0aSBarry Smith     ierr = PetscInfo1(A,"Option %s not relevent\n",MatOptions[op]);CHKERRQ(ierr);
213290bbb0aSBarry Smith     break;
214941593c8SHong Zhang   case MAT_IGNORE_LOWER_TRIANGULAR:
2154e0d8c25SBarry Smith     a->ignore_ltriangular = flg;
216941593c8SHong Zhang     break;
217941593c8SHong Zhang   case MAT_ERROR_LOWER_TRIANGULAR:
2184e0d8c25SBarry Smith     a->ignore_ltriangular = flg;
21977e54ba9SKris Buschelman     break;
220f5edf698SHong Zhang   case MAT_GETROW_UPPERTRIANGULAR:
2214e0d8c25SBarry Smith     a->getrow_utriangular = flg;
222f5edf698SHong Zhang     break;
2234d9d31abSKris Buschelman   default:
224e32f2f54SBarry Smith     SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"unknown option %d",op);
22549b5e25fSSatish Balay   }
22649b5e25fSSatish Balay   PetscFunctionReturn(0);
22749b5e25fSSatish Balay }
22849b5e25fSSatish Balay 
2294a2ae208SSatish Balay #undef __FUNCT__
2304a2ae208SSatish Balay #define __FUNCT__ "MatGetRow_SeqSBAIJ"
23113f74950SBarry Smith PetscErrorCode MatGetRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *ncols,PetscInt **cols,PetscScalar **v)
23249b5e25fSSatish Balay {
23349b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
2346849ba73SBarry Smith   PetscErrorCode ierr;
23513f74950SBarry Smith   PetscInt       itmp,i,j,k,M,*ai,*aj,bs,bn,bp,*cols_i,bs2;
23649b5e25fSSatish Balay   MatScalar      *aa,*aa_i;
23787828ca2SBarry Smith   PetscScalar    *v_i;
23849b5e25fSSatish Balay 
23949b5e25fSSatish Balay   PetscFunctionBegin;
240e32f2f54SBarry Smith   if (A && !a->getrow_utriangular) SETERRQ(PETSC_COMM_SELF,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,PETSC_TRUE) or MatGetRowUpperTriangular()");
241f5edf698SHong Zhang   /* Get the upper triangular part of the row */
242d0f46423SBarry Smith   bs  = A->rmap->bs;
24349b5e25fSSatish Balay   ai  = a->i;
24449b5e25fSSatish Balay   aj  = a->j;
24549b5e25fSSatish Balay   aa  = a->a;
24649b5e25fSSatish Balay   bs2 = a->bs2;
24749b5e25fSSatish Balay 
248e32f2f54SBarry Smith   if (row < 0 || row >= A->rmap->N) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE, "Row %D out of range", row);
24949b5e25fSSatish Balay 
25049b5e25fSSatish Balay   bn  = row/bs;   /* Block number */
25149b5e25fSSatish Balay   bp  = row % bs; /* Block position */
25249b5e25fSSatish Balay   M   = ai[bn+1] - ai[bn];
25349b5e25fSSatish Balay   *ncols = bs*M;
25449b5e25fSSatish Balay 
25549b5e25fSSatish Balay   if (v) {
25649b5e25fSSatish Balay     *v = 0;
25749b5e25fSSatish Balay     if (*ncols) {
25887828ca2SBarry Smith       ierr = PetscMalloc((*ncols+row)*sizeof(PetscScalar),v);CHKERRQ(ierr);
25949b5e25fSSatish Balay       for (i=0; i<M; i++) { /* for each block in the block row */
26049b5e25fSSatish Balay         v_i  = *v + i*bs;
26149b5e25fSSatish Balay         aa_i = aa + bs2*(ai[bn] + i);
26249b5e25fSSatish Balay         for (j=bp,k=0; j<bs2; j+=bs,k++) {v_i[k] = aa_i[j];}
26349b5e25fSSatish Balay       }
26449b5e25fSSatish Balay     }
26549b5e25fSSatish Balay   }
26649b5e25fSSatish Balay 
26749b5e25fSSatish Balay   if (cols) {
26849b5e25fSSatish Balay     *cols = 0;
26949b5e25fSSatish Balay     if (*ncols) {
27013f74950SBarry Smith       ierr = PetscMalloc((*ncols+row)*sizeof(PetscInt),cols);CHKERRQ(ierr);
27149b5e25fSSatish Balay       for (i=0; i<M; i++) { /* for each block in the block row */
27249b5e25fSSatish Balay         cols_i = *cols + i*bs;
27349b5e25fSSatish Balay         itmp  = bs*aj[ai[bn] + i];
27449b5e25fSSatish Balay         for (j=0; j<bs; j++) {cols_i[j] = itmp++;}
27549b5e25fSSatish Balay       }
27649b5e25fSSatish Balay     }
27749b5e25fSSatish Balay   }
27849b5e25fSSatish Balay 
27949b5e25fSSatish Balay   /*search column A(0:row-1,row) (=A(row,0:row-1)). Could be expensive! */
2805ddb2528SHong Zhang   /* this segment is currently removed, so only entries in the upper triangle are obtained */
2815ddb2528SHong Zhang #ifdef column_search
28249b5e25fSSatish Balay   v_i    = *v    + M*bs;
28349b5e25fSSatish Balay   cols_i = *cols + M*bs;
28449b5e25fSSatish Balay   for (i=0; i<bn; i++){ /* for each block row */
28549b5e25fSSatish Balay     M = ai[i+1] - ai[i];
28649b5e25fSSatish Balay     for (j=0; j<M; j++){
28749b5e25fSSatish Balay       itmp = aj[ai[i] + j];    /* block column value */
28849b5e25fSSatish Balay       if (itmp == bn){
28949b5e25fSSatish Balay         aa_i   = aa    + bs2*(ai[i] + j) + bs*bp;
29049b5e25fSSatish Balay         for (k=0; k<bs; k++) {
29149b5e25fSSatish Balay           *cols_i++ = i*bs+k;
29249b5e25fSSatish Balay           *v_i++    = aa_i[k];
29349b5e25fSSatish Balay         }
29449b5e25fSSatish Balay         *ncols += bs;
29549b5e25fSSatish Balay         break;
29649b5e25fSSatish Balay       }
29749b5e25fSSatish Balay     }
29849b5e25fSSatish Balay   }
2995ddb2528SHong Zhang #endif
30049b5e25fSSatish Balay   PetscFunctionReturn(0);
30149b5e25fSSatish Balay }
30249b5e25fSSatish Balay 
3034a2ae208SSatish Balay #undef __FUNCT__
3044a2ae208SSatish Balay #define __FUNCT__ "MatRestoreRow_SeqSBAIJ"
30513f74950SBarry Smith PetscErrorCode MatRestoreRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
30649b5e25fSSatish Balay {
307dfbe8321SBarry Smith   PetscErrorCode ierr;
30849b5e25fSSatish Balay 
30949b5e25fSSatish Balay   PetscFunctionBegin;
31005b42c5fSBarry Smith   if (idx) {ierr = PetscFree(*idx);CHKERRQ(ierr);}
31105b42c5fSBarry Smith   if (v)   {ierr = PetscFree(*v);CHKERRQ(ierr);}
31249b5e25fSSatish Balay   PetscFunctionReturn(0);
31349b5e25fSSatish Balay }
31449b5e25fSSatish Balay 
3154a2ae208SSatish Balay #undef __FUNCT__
316f5edf698SHong Zhang #define __FUNCT__ "MatGetRowUpperTriangular_SeqSBAIJ"
317f5edf698SHong Zhang PetscErrorCode MatGetRowUpperTriangular_SeqSBAIJ(Mat A)
318f5edf698SHong Zhang {
319f5edf698SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
320f5edf698SHong Zhang 
321f5edf698SHong Zhang   PetscFunctionBegin;
322f5edf698SHong Zhang   a->getrow_utriangular = PETSC_TRUE;
323f5edf698SHong Zhang   PetscFunctionReturn(0);
324f5edf698SHong Zhang }
325f5edf698SHong Zhang #undef __FUNCT__
326f5edf698SHong Zhang #define __FUNCT__ "MatRestoreRowUpperTriangular_SeqSBAIJ"
327f5edf698SHong Zhang PetscErrorCode MatRestoreRowUpperTriangular_SeqSBAIJ(Mat A)
328f5edf698SHong Zhang {
329f5edf698SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
330f5edf698SHong Zhang 
331f5edf698SHong Zhang   PetscFunctionBegin;
332f5edf698SHong Zhang   a->getrow_utriangular = PETSC_FALSE;
333f5edf698SHong Zhang   PetscFunctionReturn(0);
334f5edf698SHong Zhang }
335f5edf698SHong Zhang 
336f5edf698SHong Zhang #undef __FUNCT__
3374a2ae208SSatish Balay #define __FUNCT__ "MatTranspose_SeqSBAIJ"
338fc4dec0aSBarry Smith PetscErrorCode MatTranspose_SeqSBAIJ(Mat A,MatReuse reuse,Mat *B)
33949b5e25fSSatish Balay {
340dfbe8321SBarry Smith   PetscErrorCode ierr;
34149b5e25fSSatish Balay   PetscFunctionBegin;
342815cbec1SBarry Smith   if (reuse == MAT_INITIAL_MATRIX || *B != A) {
343999d9058SBarry Smith     ierr = MatDuplicate(A,MAT_COPY_VALUES,B);CHKERRQ(ierr);
344fc4dec0aSBarry Smith   }
3458115998fSBarry Smith   PetscFunctionReturn(0);
34649b5e25fSSatish Balay }
34749b5e25fSSatish Balay 
3484a2ae208SSatish Balay #undef __FUNCT__
3494a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_ASCII"
3506849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_ASCII(Mat A,PetscViewer viewer)
35149b5e25fSSatish Balay {
35249b5e25fSSatish Balay   Mat_SeqSBAIJ      *a = (Mat_SeqSBAIJ*)A->data;
353dfbe8321SBarry Smith   PetscErrorCode    ierr;
354d0f46423SBarry Smith   PetscInt          i,j,bs = A->rmap->bs,k,l,bs2=a->bs2;
355f3ef73ceSBarry Smith   PetscViewerFormat format;
356*121deb67SSatish Balay   PetscInt          *diag;
35749b5e25fSSatish Balay 
35849b5e25fSSatish Balay   PetscFunctionBegin;
359b0a32e0cSBarry Smith   ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
360456192e2SBarry Smith   if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
36177431f27SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  block size is %D\n",bs);CHKERRQ(ierr);
362fb9695e5SSatish Balay   } else if (format == PETSC_VIEWER_ASCII_MATLAB) {
363d2507d54SMatthew Knepley     Mat aij;
364d5f3da31SBarry Smith     if (A->factortype && bs>1){
36570d5e725SHong Zhang       ierr = PetscPrintf(PETSC_COMM_SELF,"Warning: matrix is factored with bs>1. MatView() with PETSC_VIEWER_ASCII_MATLAB is not supported and ignored!\n");CHKERRQ(ierr);
36670d5e725SHong Zhang       PetscFunctionReturn(0);
36770d5e725SHong Zhang     }
368c9f458caSMatthew Knepley     ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&aij);CHKERRQ(ierr);
369c9f458caSMatthew Knepley     ierr = MatView(aij,viewer);CHKERRQ(ierr);
3706bf464f9SBarry Smith     ierr = MatDestroy(&aij);CHKERRQ(ierr);
371fb9695e5SSatish Balay   } else if (format == PETSC_VIEWER_ASCII_COMMON) {
372d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
37349b5e25fSSatish Balay     for (i=0; i<a->mbs; i++) {
37449b5e25fSSatish Balay       for (j=0; j<bs; j++) {
37577431f27SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr);
37649b5e25fSSatish Balay         for (k=a->i[i]; k<a->i[i+1]; k++) {
37749b5e25fSSatish Balay           for (l=0; l<bs; l++) {
37849b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX)
37949b5e25fSSatish Balay             if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
380a83599f4SBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %G + %G i) ",bs*a->j[k]+l,
38149b5e25fSSatish Balay                                             PetscRealPart(a->a[bs2*k + l*bs + j]),PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
38249b5e25fSSatish Balay             } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
383a83599f4SBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %G - %G i) ",bs*a->j[k]+l,
38449b5e25fSSatish Balay                                             PetscRealPart(a->a[bs2*k + l*bs + j]),-PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
38549b5e25fSSatish Balay             } else if (PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
386a83599f4SBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",bs*a->j[k]+l,PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
38749b5e25fSSatish Balay             }
38849b5e25fSSatish Balay #else
38949b5e25fSSatish Balay             if (a->a[bs2*k + l*bs + j] != 0.0) {
390a83599f4SBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",bs*a->j[k]+l,a->a[bs2*k + l*bs + j]);CHKERRQ(ierr);
39149b5e25fSSatish Balay             }
39249b5e25fSSatish Balay #endif
39349b5e25fSSatish Balay           }
39449b5e25fSSatish Balay         }
395b0a32e0cSBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
39649b5e25fSSatish Balay       }
39749b5e25fSSatish Balay     }
398d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
399c1490034SHong Zhang   } else if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) {
400c1490034SHong Zhang      PetscFunctionReturn(0);
40149b5e25fSSatish Balay   } else {
402d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
4037566de4bSShri Abhyankar     ierr = PetscObjectPrintClassNamePrefixType((PetscObject)A,viewer,"Matrix Object");CHKERRQ(ierr);
4042c990fa1SHong Zhang     if (A->factortype){ /* for factored matrix */
4052c990fa1SHong Zhang       if (bs>1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"matrix is factored with bs>1. Not implemented yet");
4062c990fa1SHong Zhang 
407*121deb67SSatish Balay       diag=a->diag;
408*121deb67SSatish Balay       for (i=0; i<a->mbs; i++) { /* for row block i */
4092c990fa1SHong Zhang         ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i);CHKERRQ(ierr);
4102c990fa1SHong Zhang         /* diagonal entry */
4112c990fa1SHong Zhang #if defined(PETSC_USE_COMPLEX)
4122c990fa1SHong Zhang         if (PetscImaginaryPart(a->a[diag[i]]) > 0.0) {
4132c990fa1SHong Zhang           ierr = PetscViewerASCIIPrintf(viewer," (%D, %G + %G i) ",a->j[diag[i]],
4142c990fa1SHong Zhang                                         PetscRealPart(1.0/a->a[diag[i]]),PetscImaginaryPart(1.0/a->a[diag[i]]));CHKERRQ(ierr);
4152c990fa1SHong Zhang         } else if (PetscImaginaryPart(a->a[diag[i]]) < 0.0) {
4162c990fa1SHong Zhang           ierr = PetscViewerASCIIPrintf(viewer," (%D, %G - %G i) ",a->j[diag[i]],
4172c990fa1SHong Zhang                                         PetscRealPart(1.0/a->a[diag[i]]),-PetscImaginaryPart(1.0/a->a[diag[i]]));CHKERRQ(ierr);
4182c990fa1SHong Zhang         } else {
4192c990fa1SHong Zhang           ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",a->j[diag[i]],PetscRealPart(1.0/a->a[diag[i]]));CHKERRQ(ierr);
4202c990fa1SHong Zhang         }
4212c990fa1SHong Zhang #else
4222c990fa1SHong Zhang         ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",a->j[diag[i]],1.0/a->a[diag[i]]);CHKERRQ(ierr);
4232c990fa1SHong Zhang #endif
4242c990fa1SHong Zhang         /* off-diagonal entries */
4252c990fa1SHong Zhang         for (k=a->i[i]; k<a->i[i+1]-1; k++) {
4262c990fa1SHong Zhang #if defined(PETSC_USE_COMPLEX)
4272c990fa1SHong Zhang           if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0) {
4282c990fa1SHong Zhang             ierr = PetscViewerASCIIPrintf(viewer," (%D, %G + %G i) ",bs*a->j[k]+l,
4292c990fa1SHong Zhang                                           PetscRealPart(a->a[bs2*k + l*bs + j]),PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
4302c990fa1SHong Zhang           } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0) {
4312c990fa1SHong Zhang             ierr = PetscViewerASCIIPrintf(viewer," (%D, %G - %G i) ",bs*a->j[k]+l,
4322c990fa1SHong Zhang                                           PetscRealPart(a->a[bs2*k + l*bs + j]),-PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
4332c990fa1SHong Zhang           } else {
4342c990fa1SHong Zhang             ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",bs*a->j[k]+l,PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
4352c990fa1SHong Zhang           }
4362c990fa1SHong Zhang #else
4372c990fa1SHong Zhang           ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",a->j[k],a->a[k]);CHKERRQ(ierr);
4382c990fa1SHong Zhang #endif
4392c990fa1SHong Zhang         }
4402c990fa1SHong Zhang         ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
4412c990fa1SHong Zhang       }
4422c990fa1SHong Zhang 
4432c990fa1SHong Zhang     } else { /* for non-factored matrix */
4442c990fa1SHong Zhang       for (i=0; i<a->mbs; i++) { // for row block i
4452c990fa1SHong Zhang         for (j=0; j<bs; j++) {   // for row bs*i + j
44677431f27SBarry Smith           ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr);
4472c990fa1SHong Zhang           for (k=a->i[i]; k<a->i[i+1]; k++) { // for column block
4482c990fa1SHong Zhang             for (l=0; l<bs; l++) {            // for column
44949b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX)
45049b5e25fSSatish Balay               if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0) {
451a83599f4SBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %G + %G i) ",bs*a->j[k]+l,
45249b5e25fSSatish Balay                                             PetscRealPart(a->a[bs2*k + l*bs + j]),PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
45349b5e25fSSatish Balay               } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0) {
454a83599f4SBarry Smith                 ierr = PetscViewerASCIIPrintf(viewer," (%D, %G - %G i) ",bs*a->j[k]+l,
45549b5e25fSSatish Balay                                             PetscRealPart(a->a[bs2*k + l*bs + j]),-PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
45649b5e25fSSatish Balay               } else {
457a83599f4SBarry Smith                 ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",bs*a->j[k]+l,PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
45849b5e25fSSatish Balay               }
45949b5e25fSSatish Balay #else
460e9f7bc9eSHong Zhang               ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",bs*a->j[k]+l,a->a[bs2*k + l*bs + j]);CHKERRQ(ierr);
46149b5e25fSSatish Balay #endif
46249b5e25fSSatish Balay             }
46349b5e25fSSatish Balay           }
464b0a32e0cSBarry Smith           ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
46549b5e25fSSatish Balay         }
46649b5e25fSSatish Balay       }
4672c990fa1SHong Zhang     }
468d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
46949b5e25fSSatish Balay   }
470b0a32e0cSBarry Smith   ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
47149b5e25fSSatish Balay   PetscFunctionReturn(0);
47249b5e25fSSatish Balay }
47349b5e25fSSatish Balay 
4744a2ae208SSatish Balay #undef __FUNCT__
4754a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_Draw_Zoom"
4766849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw_Zoom(PetscDraw draw,void *Aa)
47749b5e25fSSatish Balay {
47849b5e25fSSatish Balay   Mat            A = (Mat) Aa;
47949b5e25fSSatish Balay   Mat_SeqSBAIJ   *a=(Mat_SeqSBAIJ*)A->data;
4806849ba73SBarry Smith   PetscErrorCode ierr;
481d0f46423SBarry Smith   PetscInt       row,i,j,k,l,mbs=a->mbs,color,bs=A->rmap->bs,bs2=a->bs2;
48213f74950SBarry Smith   PetscMPIInt    rank;
48349b5e25fSSatish Balay   PetscReal      xl,yl,xr,yr,x_l,x_r,y_l,y_r;
48449b5e25fSSatish Balay   MatScalar      *aa;
48549b5e25fSSatish Balay   MPI_Comm       comm;
486b0a32e0cSBarry Smith   PetscViewer    viewer;
48749b5e25fSSatish Balay 
48849b5e25fSSatish Balay   PetscFunctionBegin;
48949b5e25fSSatish Balay   /*
49049b5e25fSSatish Balay     This is nasty. If this is called from an originally parallel matrix
49149b5e25fSSatish Balay     then all processes call this,but only the first has the matrix so the
49249b5e25fSSatish Balay     rest should return immediately.
49349b5e25fSSatish Balay   */
49449b5e25fSSatish Balay   ierr = PetscObjectGetComm((PetscObject)draw,&comm);CHKERRQ(ierr);
49549b5e25fSSatish Balay   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
49649b5e25fSSatish Balay   if (rank) PetscFunctionReturn(0);
49749b5e25fSSatish Balay 
49849b5e25fSSatish Balay   ierr = PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer);CHKERRQ(ierr);
49949b5e25fSSatish Balay 
500b0a32e0cSBarry Smith   ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr);
501b0a32e0cSBarry Smith   PetscDrawString(draw, .3*(xl+xr), .3*(yl+yr), PETSC_DRAW_BLACK, "symmetric");
50249b5e25fSSatish Balay 
50349b5e25fSSatish Balay   /* loop over matrix elements drawing boxes */
504b0a32e0cSBarry Smith   color = PETSC_DRAW_BLUE;
50549b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
50649b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
507d0f46423SBarry Smith       y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0;
50849b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
50949b5e25fSSatish Balay       aa = a->a + j*bs2;
51049b5e25fSSatish Balay       for (k=0; k<bs; k++) {
51149b5e25fSSatish Balay         for (l=0; l<bs; l++) {
51249b5e25fSSatish Balay           if (PetscRealPart(*aa++) >=  0.) continue;
513b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
51449b5e25fSSatish Balay         }
51549b5e25fSSatish Balay       }
51649b5e25fSSatish Balay     }
51749b5e25fSSatish Balay   }
518b0a32e0cSBarry Smith   color = PETSC_DRAW_CYAN;
51949b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
52049b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
521d0f46423SBarry Smith       y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0;
52249b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
52349b5e25fSSatish Balay       aa = a->a + j*bs2;
52449b5e25fSSatish Balay       for (k=0; k<bs; k++) {
52549b5e25fSSatish Balay         for (l=0; l<bs; l++) {
52649b5e25fSSatish Balay           if (PetscRealPart(*aa++) != 0.) continue;
527b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
52849b5e25fSSatish Balay         }
52949b5e25fSSatish Balay       }
53049b5e25fSSatish Balay     }
53149b5e25fSSatish Balay   }
53249b5e25fSSatish Balay 
533b0a32e0cSBarry Smith   color = PETSC_DRAW_RED;
53449b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
53549b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
536d0f46423SBarry Smith       y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0;
53749b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
53849b5e25fSSatish Balay       aa = a->a + j*bs2;
53949b5e25fSSatish Balay       for (k=0; k<bs; k++) {
54049b5e25fSSatish Balay         for (l=0; l<bs; l++) {
54149b5e25fSSatish Balay           if (PetscRealPart(*aa++) <= 0.) continue;
542b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
54349b5e25fSSatish Balay         }
54449b5e25fSSatish Balay       }
54549b5e25fSSatish Balay     }
54649b5e25fSSatish Balay   }
54749b5e25fSSatish Balay   PetscFunctionReturn(0);
54849b5e25fSSatish Balay }
54949b5e25fSSatish Balay 
5504a2ae208SSatish Balay #undef __FUNCT__
5514a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_Draw"
5526849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw(Mat A,PetscViewer viewer)
55349b5e25fSSatish Balay {
554dfbe8321SBarry Smith   PetscErrorCode ierr;
55549b5e25fSSatish Balay   PetscReal      xl,yl,xr,yr,w,h;
556b0a32e0cSBarry Smith   PetscDraw      draw;
557ace3abfcSBarry Smith   PetscBool      isnull;
55849b5e25fSSatish Balay 
55949b5e25fSSatish Balay   PetscFunctionBegin;
560b0a32e0cSBarry Smith   ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
561b0a32e0cSBarry Smith   ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr); if (isnull) PetscFunctionReturn(0);
56249b5e25fSSatish Balay 
56349b5e25fSSatish Balay   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer);CHKERRQ(ierr);
564d0f46423SBarry Smith   xr  = A->rmap->N; yr = A->rmap->N; h = yr/10.0; w = xr/10.0;
56549b5e25fSSatish Balay   xr += w;    yr += h;  xl = -w;     yl = -h;
566b0a32e0cSBarry Smith   ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr);
567b0a32e0cSBarry Smith   ierr = PetscDrawZoom(draw,MatView_SeqSBAIJ_Draw_Zoom,A);CHKERRQ(ierr);
56849b5e25fSSatish Balay   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",PETSC_NULL);CHKERRQ(ierr);
56949b5e25fSSatish Balay   PetscFunctionReturn(0);
57049b5e25fSSatish Balay }
57149b5e25fSSatish Balay 
5724a2ae208SSatish Balay #undef __FUNCT__
5734a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ"
574dfbe8321SBarry Smith PetscErrorCode MatView_SeqSBAIJ(Mat A,PetscViewer viewer)
57549b5e25fSSatish Balay {
576dfbe8321SBarry Smith   PetscErrorCode ierr;
577ace3abfcSBarry Smith   PetscBool      iascii,isdraw;
57808917f38SBarry Smith   FILE           *file = 0;
57949b5e25fSSatish Balay 
58049b5e25fSSatish Balay   PetscFunctionBegin;
5812692d6eeSBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
5822692d6eeSBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr);
58332077d6dSBarry Smith   if (iascii){
58449b5e25fSSatish Balay     ierr = MatView_SeqSBAIJ_ASCII(A,viewer);CHKERRQ(ierr);
58549b5e25fSSatish Balay   } else if (isdraw) {
58649b5e25fSSatish Balay     ierr = MatView_SeqSBAIJ_Draw(A,viewer);CHKERRQ(ierr);
58749b5e25fSSatish Balay   } else {
588a5e6ed63SBarry Smith     Mat B;
589ceb03754SKris Buschelman     ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&B);CHKERRQ(ierr);
590a5e6ed63SBarry Smith     ierr = MatView(B,viewer);CHKERRQ(ierr);
5916bf464f9SBarry Smith     ierr = MatDestroy(&B);CHKERRQ(ierr);
59208917f38SBarry Smith     ierr = PetscViewerBinaryGetInfoPointer(viewer,&file);CHKERRQ(ierr);
59308917f38SBarry Smith     if (file) {
59408917f38SBarry Smith       fprintf(file,"-matload_block_size %d\n",(int)A->rmap->bs);
59508917f38SBarry Smith     }
59649b5e25fSSatish Balay   }
59749b5e25fSSatish Balay   PetscFunctionReturn(0);
59849b5e25fSSatish Balay }
59949b5e25fSSatish Balay 
60049b5e25fSSatish Balay 
6014a2ae208SSatish Balay #undef __FUNCT__
6024a2ae208SSatish Balay #define __FUNCT__ "MatGetValues_SeqSBAIJ"
60313f74950SBarry Smith PetscErrorCode MatGetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],PetscScalar v[])
60449b5e25fSSatish Balay {
605045c9aa0SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
60613f74950SBarry Smith   PetscInt     *rp,k,low,high,t,row,nrow,i,col,l,*aj = a->j;
60713f74950SBarry Smith   PetscInt     *ai = a->i,*ailen = a->ilen;
608d0f46423SBarry Smith   PetscInt     brow,bcol,ridx,cidx,bs=A->rmap->bs,bs2=a->bs2;
60997e567efSBarry Smith   MatScalar    *ap,*aa = a->a;
61049b5e25fSSatish Balay 
61149b5e25fSSatish Balay   PetscFunctionBegin;
61249b5e25fSSatish Balay   for (k=0; k<m; k++) { /* loop over rows */
61349b5e25fSSatish Balay     row  = im[k]; brow = row/bs;
614e32f2f54SBarry Smith     if (row < 0) {v += n; continue;} /* SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative row: %D",row); */
615e32f2f54SBarry Smith     if (row >= A->rmap->N) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,A->rmap->N-1);
61649b5e25fSSatish Balay     rp   = aj + ai[brow] ; ap = aa + bs2*ai[brow] ;
61749b5e25fSSatish Balay     nrow = ailen[brow];
61849b5e25fSSatish Balay     for (l=0; l<n; l++) { /* loop over columns */
619e32f2f54SBarry Smith       if (in[l] < 0) {v++; continue;} /* SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column: %D",in[l]); */
620e32f2f54SBarry Smith       if (in[l] >= A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",in[l],A->cmap->n-1);
62149b5e25fSSatish Balay       col  = in[l] ;
62249b5e25fSSatish Balay       bcol = col/bs;
62349b5e25fSSatish Balay       cidx = col%bs;
62449b5e25fSSatish Balay       ridx = row%bs;
62549b5e25fSSatish Balay       high = nrow;
62649b5e25fSSatish Balay       low  = 0; /* assume unsorted */
62749b5e25fSSatish Balay       while (high-low > 5) {
62849b5e25fSSatish Balay         t = (low+high)/2;
62949b5e25fSSatish Balay         if (rp[t] > bcol) high = t;
63049b5e25fSSatish Balay         else             low  = t;
63149b5e25fSSatish Balay       }
63249b5e25fSSatish Balay       for (i=low; i<high; i++) {
63349b5e25fSSatish Balay         if (rp[i] > bcol) break;
63449b5e25fSSatish Balay         if (rp[i] == bcol) {
63549b5e25fSSatish Balay           *v++ = ap[bs2*i+bs*cidx+ridx];
63649b5e25fSSatish Balay           goto finished;
63749b5e25fSSatish Balay         }
63849b5e25fSSatish Balay       }
63997e567efSBarry Smith       *v++ = 0.0;
64049b5e25fSSatish Balay       finished:;
64149b5e25fSSatish Balay     }
64249b5e25fSSatish Balay   }
64349b5e25fSSatish Balay   PetscFunctionReturn(0);
64449b5e25fSSatish Balay }
64549b5e25fSSatish Balay 
64649b5e25fSSatish Balay 
6474a2ae208SSatish Balay #undef __FUNCT__
6484a2ae208SSatish Balay #define __FUNCT__ "MatSetValuesBlocked_SeqSBAIJ"
64913f74950SBarry Smith PetscErrorCode MatSetValuesBlocked_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is)
65049b5e25fSSatish Balay {
6510880e062SHong Zhang   Mat_SeqSBAIJ      *a = (Mat_SeqSBAIJ*)A->data;
6526849ba73SBarry Smith   PetscErrorCode    ierr;
653e2ee6c50SBarry Smith   PetscInt          *rp,k,low,high,t,ii,jj,row,nrow,i,col,l,rmax,N,lastcol = -1;
65413f74950SBarry Smith   PetscInt          *imax=a->imax,*ai=a->i,*ailen=a->ilen;
655d0f46423SBarry Smith   PetscInt          *aj=a->j,nonew=a->nonew,bs2=a->bs2,bs=A->rmap->bs,stepval;
656ace3abfcSBarry Smith   PetscBool         roworiented=a->roworiented;
657dd6ea824SBarry Smith   const PetscScalar *value = v;
658f15d580aSBarry Smith   MatScalar         *ap,*aa = a->a,*bap;
6590880e062SHong Zhang 
66049b5e25fSSatish Balay   PetscFunctionBegin;
6610880e062SHong Zhang   if (roworiented) {
6620880e062SHong Zhang     stepval = (n-1)*bs;
6630880e062SHong Zhang   } else {
6640880e062SHong Zhang     stepval = (m-1)*bs;
6650880e062SHong Zhang   }
6660880e062SHong Zhang   for (k=0; k<m; k++) { /* loop over added rows */
6670880e062SHong Zhang     row  = im[k];
6680880e062SHong Zhang     if (row < 0) continue;
6692515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
670e32f2f54SBarry Smith     if (row >= a->mbs) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,a->mbs-1);
6710880e062SHong Zhang #endif
6720880e062SHong Zhang     rp   = aj + ai[row];
6730880e062SHong Zhang     ap   = aa + bs2*ai[row];
6740880e062SHong Zhang     rmax = imax[row];
6750880e062SHong Zhang     nrow = ailen[row];
6760880e062SHong Zhang     low  = 0;
677818f2c47SBarry Smith     high = nrow;
6780880e062SHong Zhang     for (l=0; l<n; l++) { /* loop over added columns */
6790880e062SHong Zhang       if (in[l] < 0) continue;
6800880e062SHong Zhang       col = in[l];
6812515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
682e32f2f54SBarry Smith       if (col >= a->nbs) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",col,a->nbs-1);
683b1823623SSatish Balay #endif
684b98bf0e1SJed Brown       if (col < row) {
685b98bf0e1SJed Brown         if (a->ignore_ltriangular) {
686b98bf0e1SJed Brown           continue; /* ignore lower triangular block */
687b98bf0e1SJed Brown         } else {
688e32f2f54SBarry Smith           SETERRQ(PETSC_COMM_SELF,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,PETSC_TRUE)");
689b98bf0e1SJed Brown         }
690b98bf0e1SJed Brown       }
6910880e062SHong Zhang       if (roworiented) {
6920880e062SHong Zhang         value = v + k*(stepval+bs)*bs + l*bs;
6930880e062SHong Zhang       } else {
6940880e062SHong Zhang         value = v + l*(stepval+bs)*bs + k*bs;
6950880e062SHong Zhang       }
6967cd84e04SBarry Smith       if (col <= lastcol) low = 0; else high = nrow;
697e2ee6c50SBarry Smith       lastcol = col;
6980880e062SHong Zhang       while (high-low > 7) {
6990880e062SHong Zhang         t = (low+high)/2;
7000880e062SHong Zhang         if (rp[t] > col) high = t;
7010880e062SHong Zhang         else             low  = t;
7020880e062SHong Zhang       }
7030880e062SHong Zhang       for (i=low; i<high; i++) {
7040880e062SHong Zhang         if (rp[i] > col) break;
7050880e062SHong Zhang         if (rp[i] == col) {
7060880e062SHong Zhang           bap  = ap +  bs2*i;
7070880e062SHong Zhang           if (roworiented) {
7080880e062SHong Zhang             if (is == ADD_VALUES) {
7090880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
7100880e062SHong Zhang                 for (jj=ii; jj<bs2; jj+=bs) {
7110880e062SHong Zhang                   bap[jj] += *value++;
7120880e062SHong Zhang                 }
7130880e062SHong Zhang               }
7140880e062SHong Zhang             } else {
7150880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
7160880e062SHong Zhang                 for (jj=ii; jj<bs2; jj+=bs) {
7170880e062SHong Zhang                   bap[jj] = *value++;
7180880e062SHong Zhang                 }
7190880e062SHong Zhang                }
7200880e062SHong Zhang             }
7210880e062SHong Zhang           } else {
7220880e062SHong Zhang             if (is == ADD_VALUES) {
7230880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
7240880e062SHong Zhang                 for (jj=0; jj<bs; jj++) {
7250880e062SHong Zhang                   *bap++ += *value++;
7260880e062SHong Zhang                 }
7270880e062SHong Zhang               }
7280880e062SHong Zhang             } else {
7290880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
7300880e062SHong Zhang                 for (jj=0; jj<bs; jj++) {
7310880e062SHong Zhang                   *bap++  = *value++;
7320880e062SHong Zhang                 }
7330880e062SHong Zhang               }
7340880e062SHong Zhang             }
7350880e062SHong Zhang           }
7360880e062SHong Zhang           goto noinsert2;
7370880e062SHong Zhang         }
7380880e062SHong Zhang       }
7390880e062SHong Zhang       if (nonew == 1) goto noinsert2;
740e32f2f54SBarry Smith       if (nonew == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col);
741fef13f97SBarry Smith       MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,row,col,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar);
742c03d1d03SSatish Balay       N = nrow++ - 1; high++;
7430880e062SHong Zhang       /* shift up all the later entries in this row */
7440880e062SHong Zhang       for (ii=N; ii>=i; ii--) {
7450880e062SHong Zhang         rp[ii+1] = rp[ii];
7460880e062SHong Zhang         ierr = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr);
7470880e062SHong Zhang       }
7480880e062SHong Zhang       if (N >= i) {
7490880e062SHong Zhang         ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr);
7500880e062SHong Zhang       }
7510880e062SHong Zhang       rp[i] = col;
7520880e062SHong Zhang       bap   = ap +  bs2*i;
7530880e062SHong Zhang       if (roworiented) {
7540880e062SHong Zhang         for (ii=0; ii<bs; ii++,value+=stepval) {
7550880e062SHong Zhang           for (jj=ii; jj<bs2; jj+=bs) {
7560880e062SHong Zhang             bap[jj] = *value++;
7570880e062SHong Zhang           }
7580880e062SHong Zhang         }
7590880e062SHong Zhang       } else {
7600880e062SHong Zhang         for (ii=0; ii<bs; ii++,value+=stepval) {
7610880e062SHong Zhang           for (jj=0; jj<bs; jj++) {
7620880e062SHong Zhang             *bap++  = *value++;
7630880e062SHong Zhang           }
7640880e062SHong Zhang         }
7650880e062SHong Zhang        }
7660880e062SHong Zhang     noinsert2:;
7670880e062SHong Zhang       low = i;
7680880e062SHong Zhang     }
7690880e062SHong Zhang     ailen[row] = nrow;
7700880e062SHong Zhang   }
7710880e062SHong Zhang    PetscFunctionReturn(0);
77249b5e25fSSatish Balay }
77349b5e25fSSatish Balay 
77464831d72SBarry Smith /*
77564831d72SBarry Smith     This is not yet used
77664831d72SBarry Smith */
7774a2ae208SSatish Balay #undef __FUNCT__
7784108e4d5SBarry Smith #define __FUNCT__ "MatAssemblyEnd_SeqSBAIJ_SeqAIJ_Inode"
7794108e4d5SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ_SeqAIJ_Inode(Mat A)
7800def2e27SBarry Smith {
7810def2e27SBarry Smith   Mat_SeqSBAIJ    *a = (Mat_SeqSBAIJ*)A->data;
7820def2e27SBarry Smith   PetscErrorCode  ierr;
7830def2e27SBarry Smith   const PetscInt  *ai = a->i, *aj = a->j,*cols;
7840def2e27SBarry Smith   PetscInt        i = 0,j,blk_size,m = A->rmap->n,node_count = 0,nzx,nzy,*ns,row,nz,cnt,cnt2,*counts;
785ace3abfcSBarry Smith   PetscBool       flag;
7860def2e27SBarry Smith 
7870def2e27SBarry Smith   PetscFunctionBegin;
7880def2e27SBarry Smith   ierr = PetscMalloc(m*sizeof(PetscInt),&ns);CHKERRQ(ierr);
7890def2e27SBarry Smith   while (i < m){
7900def2e27SBarry Smith     nzx = ai[i+1] - ai[i];       /* Number of nonzeros */
7910def2e27SBarry Smith     /* Limits the number of elements in a node to 'a->inode.limit' */
7920def2e27SBarry Smith     for (j=i+1,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) {
7930def2e27SBarry Smith       nzy  = ai[j+1] - ai[j];
7940def2e27SBarry Smith       if (nzy != (nzx - j + i)) break;
7950def2e27SBarry Smith       ierr = PetscMemcmp(aj + ai[i] + j - i,aj + ai[j],nzy*sizeof(PetscInt),&flag);CHKERRQ(ierr);
7960def2e27SBarry Smith       if (!flag) break;
7970def2e27SBarry Smith     }
7980def2e27SBarry Smith     ns[node_count++] = blk_size;
7990def2e27SBarry Smith     i = j;
8000def2e27SBarry Smith   }
8010def2e27SBarry Smith   if (!a->inode.size && m && node_count > .9*m) {
8020def2e27SBarry Smith     ierr = PetscFree(ns);CHKERRQ(ierr);
8030def2e27SBarry Smith     ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr);
8040def2e27SBarry Smith   } else {
8050def2e27SBarry Smith     a->inode.node_count = node_count;
8060def2e27SBarry Smith     ierr = PetscMalloc(node_count*sizeof(PetscInt),&a->inode.size);CHKERRQ(ierr);
807c760cd28SBarry Smith     ierr = PetscLogObjectMemory(A,node_count*sizeof(PetscInt));CHKERRQ(ierr);
8080def2e27SBarry Smith     ierr = PetscMemcpy(a->inode.size,ns,node_count*sizeof(PetscInt));
8090def2e27SBarry Smith     ierr = PetscFree(ns);CHKERRQ(ierr);
8100def2e27SBarry Smith     ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr);
8110def2e27SBarry Smith 
8120def2e27SBarry Smith     /* count collections of adjacent columns in each inode */
8130def2e27SBarry Smith     row = 0;
8140def2e27SBarry Smith     cnt = 0;
8150def2e27SBarry Smith     for (i=0; i<node_count; i++) {
8160def2e27SBarry Smith       cols = aj + ai[row] + a->inode.size[i];
8170def2e27SBarry Smith       nz   = ai[row+1] - ai[row] - a->inode.size[i];
8180def2e27SBarry Smith       for (j=1; j<nz; j++) {
8190def2e27SBarry Smith         if (cols[j] != cols[j-1]+1) {
8200def2e27SBarry Smith           cnt++;
8210def2e27SBarry Smith         }
8220def2e27SBarry Smith       }
8230def2e27SBarry Smith       cnt++;
8240def2e27SBarry Smith       row += a->inode.size[i];
8250def2e27SBarry Smith     }
8260def2e27SBarry Smith     ierr = PetscMalloc(2*cnt*sizeof(PetscInt),&counts);CHKERRQ(ierr);
8270def2e27SBarry Smith     cnt = 0;
8280def2e27SBarry Smith     row = 0;
8290def2e27SBarry Smith     for (i=0; i<node_count; i++) {
8300def2e27SBarry Smith       cols          = aj + ai[row] + a->inode.size[i];
8310def2e27SBarry Smith 	  CHKMEMQ;
8320def2e27SBarry Smith       counts[2*cnt] = cols[0];
8330def2e27SBarry Smith 	  CHKMEMQ;
8340def2e27SBarry Smith       nz            = ai[row+1] - ai[row] - a->inode.size[i];
8350def2e27SBarry Smith       cnt2          = 1;
8360def2e27SBarry Smith       for (j=1; j<nz; j++) {
8370def2e27SBarry Smith         if (cols[j] != cols[j-1]+1) {
8380def2e27SBarry Smith 	  CHKMEMQ;
8390def2e27SBarry Smith           counts[2*(cnt++)+1] = cnt2;
8400def2e27SBarry Smith           counts[2*cnt]       = cols[j];
8410def2e27SBarry Smith 	  CHKMEMQ;
8420def2e27SBarry Smith           cnt2                = 1;
8430def2e27SBarry Smith         } else cnt2++;
8440def2e27SBarry Smith       }
8450def2e27SBarry Smith 	  CHKMEMQ;
8460def2e27SBarry Smith       counts[2*(cnt++)+1] = cnt2;
8470def2e27SBarry Smith 	  CHKMEMQ;
8480def2e27SBarry Smith       row += a->inode.size[i];
8490def2e27SBarry Smith     }
8500def2e27SBarry Smith     ierr = PetscIntView(2*cnt,counts,0);
8510def2e27SBarry Smith   }
85238702af4SBarry Smith   PetscFunctionReturn(0);
85338702af4SBarry Smith }
85438702af4SBarry Smith 
85538702af4SBarry Smith #undef __FUNCT__
8564a2ae208SSatish Balay #define __FUNCT__ "MatAssemblyEnd_SeqSBAIJ"
857dfbe8321SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ(Mat A,MatAssemblyType mode)
85849b5e25fSSatish Balay {
85949b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
8606849ba73SBarry Smith   PetscErrorCode ierr;
86113f74950SBarry Smith   PetscInt       fshift = 0,i,j,*ai = a->i,*aj = a->j,*imax = a->imax;
862d0f46423SBarry Smith   PetscInt       m = A->rmap->N,*ip,N,*ailen = a->ilen;
86313f74950SBarry Smith   PetscInt       mbs = a->mbs,bs2 = a->bs2,rmax = 0;
86449b5e25fSSatish Balay   MatScalar      *aa = a->a,*ap;
86549b5e25fSSatish Balay 
86649b5e25fSSatish Balay   PetscFunctionBegin;
86749b5e25fSSatish Balay   if (mode == MAT_FLUSH_ASSEMBLY) PetscFunctionReturn(0);
86849b5e25fSSatish Balay 
86949b5e25fSSatish Balay   if (m) rmax = ailen[0];
87049b5e25fSSatish Balay   for (i=1; i<mbs; i++) {
87149b5e25fSSatish Balay     /* move each row back by the amount of empty slots (fshift) before it*/
87249b5e25fSSatish Balay     fshift += imax[i-1] - ailen[i-1];
87349b5e25fSSatish Balay      rmax   = PetscMax(rmax,ailen[i]);
87449b5e25fSSatish Balay      if (fshift) {
87549b5e25fSSatish Balay        ip = aj + ai[i]; ap = aa + bs2*ai[i];
87649b5e25fSSatish Balay        N = ailen[i];
87749b5e25fSSatish Balay        for (j=0; j<N; j++) {
87849b5e25fSSatish Balay          ip[j-fshift] = ip[j];
87949b5e25fSSatish Balay          ierr = PetscMemcpy(ap+(j-fshift)*bs2,ap+j*bs2,bs2*sizeof(MatScalar));CHKERRQ(ierr);
88049b5e25fSSatish Balay        }
88149b5e25fSSatish Balay      }
88249b5e25fSSatish Balay      ai[i] = ai[i-1] + ailen[i-1];
88349b5e25fSSatish Balay   }
88449b5e25fSSatish Balay   if (mbs) {
88549b5e25fSSatish Balay     fshift += imax[mbs-1] - ailen[mbs-1];
88649b5e25fSSatish Balay      ai[mbs] = ai[mbs-1] + ailen[mbs-1];
88749b5e25fSSatish Balay   }
88849b5e25fSSatish Balay   /* reset ilen and imax for each row */
88949b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
89049b5e25fSSatish Balay     ailen[i] = imax[i] = ai[i+1] - ai[i];
89149b5e25fSSatish Balay   }
8926c6c5352SBarry Smith   a->nz = ai[mbs];
89349b5e25fSSatish Balay 
894b424e231SHong Zhang   /* diagonals may have moved, reset it */
895b424e231SHong Zhang   if (a->diag) {
8962ed38d0bSJed Brown     ierr = PetscMemcpy(a->diag,ai,mbs*sizeof(PetscInt));CHKERRQ(ierr);
89749b5e25fSSatish Balay   }
89828b2fa4aSMatthew Knepley   if (fshift && a->nounused == -1) {
899e32f2f54SBarry Smith     SETERRQ4(PETSC_COMM_SELF,PETSC_ERR_PLIB, "Unused space detected in matrix: %D X %D block size %D, %D unneeded", m, A->cmap->n, A->rmap->bs, fshift*bs2);
90028b2fa4aSMatthew Knepley   }
901d0f46423SBarry 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);
902ae15b995SBarry Smith   ierr = PetscInfo1(A,"Number of mallocs during MatSetValues is %D\n",a->reallocs);CHKERRQ(ierr);
903ae15b995SBarry Smith   ierr = PetscInfo1(A,"Most nonzeros blocks in any row is %D\n",rmax);CHKERRQ(ierr);
9048e58a170SBarry Smith   A->info.mallocs     += a->reallocs;
90549b5e25fSSatish Balay   a->reallocs          = 0;
90649b5e25fSSatish Balay   A->info.nz_unneeded  = (PetscReal)fshift*bs2;
907061b2667SBarry Smith   a->idiagvalid = PETSC_FALSE;
90838702af4SBarry Smith 
90938702af4SBarry Smith   if (A->cmap->n < 65536 && A->cmap->bs == 1) {
91038702af4SBarry Smith     if (!a->jshort) {
91138702af4SBarry Smith       ierr = PetscMalloc(a->i[A->rmap->n]*sizeof(unsigned short),&a->jshort);CHKERRQ(ierr);
912c760cd28SBarry Smith       ierr = PetscLogObjectMemory(A,a->i[A->rmap->n]*sizeof(unsigned short));CHKERRQ(ierr);
91338702af4SBarry Smith       for (i=0; i<a->i[A->rmap->n]; i++) a->jshort[i] = a->j[i];
91438702af4SBarry Smith       A->ops->mult  = MatMult_SeqSBAIJ_1_ushort;
91541f059aeSBarry Smith       A->ops->sor = MatSOR_SeqSBAIJ_ushort;
9164da8f245SBarry Smith       a->free_jshort = PETSC_TRUE;
91738702af4SBarry Smith     }
91838702af4SBarry Smith   }
91949b5e25fSSatish Balay   PetscFunctionReturn(0);
92049b5e25fSSatish Balay }
92149b5e25fSSatish Balay 
92249b5e25fSSatish Balay /*
92349b5e25fSSatish Balay    This function returns an array of flags which indicate the locations of contiguous
92449b5e25fSSatish Balay    blocks that should be zeroed. for eg: if bs = 3  and is = [0,1,2,3,5,6,7,8,9]
92549b5e25fSSatish Balay    then the resulting sizes = [3,1,1,3,1] correspondig to sets [(0,1,2),(3),(5),(6,7,8),(9)]
92649b5e25fSSatish Balay    Assume: sizes should be long enough to hold all the values.
92749b5e25fSSatish Balay */
9284a2ae208SSatish Balay #undef __FUNCT__
9294a2ae208SSatish Balay #define __FUNCT__ "MatZeroRows_SeqSBAIJ_Check_Blocks"
93013f74950SBarry Smith PetscErrorCode MatZeroRows_SeqSBAIJ_Check_Blocks(PetscInt idx[],PetscInt n,PetscInt bs,PetscInt sizes[], PetscInt *bs_max)
93149b5e25fSSatish Balay {
93213f74950SBarry Smith   PetscInt   i,j,k,row;
933ace3abfcSBarry Smith   PetscBool  flg;
93449b5e25fSSatish Balay 
93549b5e25fSSatish Balay   PetscFunctionBegin;
93649b5e25fSSatish Balay    for (i=0,j=0; i<n; j++) {
93749b5e25fSSatish Balay      row = idx[i];
93849b5e25fSSatish Balay      if (row%bs!=0) { /* Not the begining of a block */
93949b5e25fSSatish Balay        sizes[j] = 1;
94049b5e25fSSatish Balay        i++;
94149b5e25fSSatish Balay      } else if (i+bs > n) { /* Beginning of a block, but complete block doesn't exist (at idx end) */
94249b5e25fSSatish Balay        sizes[j] = 1;         /* Also makes sure atleast 'bs' values exist for next else */
94349b5e25fSSatish Balay        i++;
94449b5e25fSSatish Balay      } else { /* Begining of the block, so check if the complete block exists */
94549b5e25fSSatish Balay        flg = PETSC_TRUE;
94649b5e25fSSatish Balay        for (k=1; k<bs; k++) {
94749b5e25fSSatish Balay          if (row+k != idx[i+k]) { /* break in the block */
94849b5e25fSSatish Balay            flg = PETSC_FALSE;
94949b5e25fSSatish Balay            break;
95049b5e25fSSatish Balay          }
95149b5e25fSSatish Balay        }
952abc0a331SBarry Smith        if (flg) { /* No break in the bs */
95349b5e25fSSatish Balay          sizes[j] = bs;
95449b5e25fSSatish Balay          i+= bs;
95549b5e25fSSatish Balay        } else {
95649b5e25fSSatish Balay          sizes[j] = 1;
95749b5e25fSSatish Balay          i++;
95849b5e25fSSatish Balay        }
95949b5e25fSSatish Balay      }
96049b5e25fSSatish Balay    }
96149b5e25fSSatish Balay    *bs_max = j;
96249b5e25fSSatish Balay    PetscFunctionReturn(0);
96349b5e25fSSatish Balay }
96449b5e25fSSatish Balay 
96549b5e25fSSatish Balay 
96649b5e25fSSatish Balay /* Only add/insert a(i,j) with i<=j (blocks).
96749b5e25fSSatish Balay    Any a(i,j) with i>j input by user is ingored.
96849b5e25fSSatish Balay */
96949b5e25fSSatish Balay 
9704a2ae208SSatish Balay #undef __FUNCT__
9714a2ae208SSatish Balay #define __FUNCT__ "MatSetValues_SeqSBAIJ"
97213f74950SBarry Smith PetscErrorCode MatSetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is)
97349b5e25fSSatish Balay {
97449b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
9756849ba73SBarry Smith   PetscErrorCode ierr;
976e2ee6c50SBarry Smith   PetscInt       *rp,k,low,high,t,ii,row,nrow,i,col,l,rmax,N,lastcol = -1;
97713f74950SBarry Smith   PetscInt       *imax=a->imax,*ai=a->i,*ailen=a->ilen,roworiented=a->roworiented;
978d0f46423SBarry Smith   PetscInt       *aj=a->j,nonew=a->nonew,bs=A->rmap->bs,brow,bcol;
97913f74950SBarry Smith   PetscInt       ridx,cidx,bs2=a->bs2;
98049b5e25fSSatish Balay   MatScalar      *ap,value,*aa=a->a,*bap;
98149b5e25fSSatish Balay 
98249b5e25fSSatish Balay   PetscFunctionBegin;
98371fd2e92SBarry Smith   if (v) PetscValidScalarPointer(v,6);
98449b5e25fSSatish Balay   for (k=0; k<m; k++) { /* loop over added rows */
98549b5e25fSSatish Balay     row  = im[k];       /* row number */
98649b5e25fSSatish Balay     brow = row/bs;      /* block row number */
98749b5e25fSSatish Balay     if (row < 0) continue;
9882515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
989e32f2f54SBarry Smith     if (row >= A->rmap->N) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,A->rmap->N-1);
99049b5e25fSSatish Balay #endif
99149b5e25fSSatish Balay     rp   = aj + ai[brow]; /*ptr to beginning of column value of the row block*/
99249b5e25fSSatish Balay     ap   = aa + bs2*ai[brow]; /*ptr to beginning of element value of the row block*/
99349b5e25fSSatish Balay     rmax = imax[brow];  /* maximum space allocated for this row */
99449b5e25fSSatish Balay     nrow = ailen[brow]; /* actual length of this row */
99549b5e25fSSatish Balay     low  = 0;
99649b5e25fSSatish Balay 
99749b5e25fSSatish Balay     for (l=0; l<n; l++) { /* loop over added columns */
99849b5e25fSSatish Balay       if (in[l] < 0) continue;
9992515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
1000e32f2f54SBarry Smith       if (in[l] >= A->rmap->N) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",in[l],A->rmap->N-1);
100149b5e25fSSatish Balay #endif
100249b5e25fSSatish Balay       col = in[l];
100349b5e25fSSatish Balay       bcol = col/bs;              /* block col number */
100449b5e25fSSatish Balay 
1005941593c8SHong Zhang       if (brow > bcol) {
1006941593c8SHong Zhang         if (a->ignore_ltriangular){
1007941593c8SHong Zhang           continue; /* ignore lower triangular values */
1008941593c8SHong Zhang         } else {
1009e32f2f54SBarry Smith           SETERRQ(PETSC_COMM_SELF,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,PETSC_TRUE)");
1010941593c8SHong Zhang         }
1011941593c8SHong Zhang       }
1012f4989cb3SHong Zhang 
101349b5e25fSSatish Balay       ridx = row % bs; cidx = col % bs; /*row and col index inside the block */
10148549e402SHong Zhang       if ((brow==bcol && ridx<=cidx) || (brow<bcol)){
101549b5e25fSSatish Balay         /* element value a(k,l) */
101649b5e25fSSatish Balay         if (roworiented) {
101749b5e25fSSatish Balay           value = v[l + k*n];
101849b5e25fSSatish Balay         } else {
101949b5e25fSSatish Balay           value = v[k + l*m];
102049b5e25fSSatish Balay         }
102149b5e25fSSatish Balay 
102249b5e25fSSatish Balay         /* move pointer bap to a(k,l) quickly and add/insert value */
10237cd84e04SBarry Smith         if (col <= lastcol) low = 0; high = nrow;
1024e2ee6c50SBarry Smith         lastcol = col;
102549b5e25fSSatish Balay         while (high-low > 7) {
102649b5e25fSSatish Balay           t = (low+high)/2;
102749b5e25fSSatish Balay           if (rp[t] > bcol) high = t;
102849b5e25fSSatish Balay           else              low  = t;
102949b5e25fSSatish Balay         }
103049b5e25fSSatish Balay         for (i=low; i<high; i++) {
103149b5e25fSSatish Balay           if (rp[i] > bcol) break;
103249b5e25fSSatish Balay           if (rp[i] == bcol) {
103349b5e25fSSatish Balay             bap  = ap +  bs2*i + bs*cidx + ridx;
103449b5e25fSSatish Balay             if (is == ADD_VALUES) *bap += value;
103549b5e25fSSatish Balay             else                  *bap  = value;
10368549e402SHong Zhang             /* for diag block, add/insert its symmetric element a(cidx,ridx) */
10378549e402SHong Zhang             if (brow == bcol && ridx < cidx){
10388549e402SHong Zhang               bap  = ap +  bs2*i + bs*ridx + cidx;
10398549e402SHong Zhang               if (is == ADD_VALUES) *bap += value;
10408549e402SHong Zhang               else                  *bap  = value;
10418549e402SHong Zhang             }
104249b5e25fSSatish Balay             goto noinsert1;
104349b5e25fSSatish Balay           }
104449b5e25fSSatish Balay         }
104549b5e25fSSatish Balay 
104649b5e25fSSatish Balay         if (nonew == 1) goto noinsert1;
1047e32f2f54SBarry Smith         if (nonew == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col);
1048fef13f97SBarry Smith         MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,brow,bcol,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar);
104949b5e25fSSatish Balay 
1050c03d1d03SSatish Balay         N = nrow++ - 1; high++;
105149b5e25fSSatish Balay         /* shift up all the later entries in this row */
105249b5e25fSSatish Balay         for (ii=N; ii>=i; ii--) {
105349b5e25fSSatish Balay           rp[ii+1] = rp[ii];
105449b5e25fSSatish Balay           ierr     = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr);
105549b5e25fSSatish Balay         }
105649b5e25fSSatish Balay         if (N>=i) {
105749b5e25fSSatish Balay           ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr);
105849b5e25fSSatish Balay         }
105949b5e25fSSatish Balay         rp[i]                      = bcol;
106049b5e25fSSatish Balay         ap[bs2*i + bs*cidx + ridx] = value;
106149b5e25fSSatish Balay       noinsert1:;
106249b5e25fSSatish Balay         low = i;
10638549e402SHong Zhang       }
106449b5e25fSSatish Balay     }   /* end of loop over added columns */
106549b5e25fSSatish Balay     ailen[brow] = nrow;
106649b5e25fSSatish Balay   }   /* end of loop over added rows */
106749b5e25fSSatish Balay   PetscFunctionReturn(0);
106849b5e25fSSatish Balay }
106949b5e25fSSatish Balay 
10704a2ae208SSatish Balay #undef __FUNCT__
10714d101231SSatish Balay #define __FUNCT__ "MatICCFactor_SeqSBAIJ"
10720481f469SBarry Smith PetscErrorCode MatICCFactor_SeqSBAIJ(Mat inA,IS row,const MatFactorInfo *info)
107349b5e25fSSatish Balay {
10744ccecd49SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)inA->data;
107549b5e25fSSatish Balay   Mat            outA;
1076dfbe8321SBarry Smith   PetscErrorCode ierr;
1077ace3abfcSBarry Smith   PetscBool      row_identity;
107849b5e25fSSatish Balay 
107949b5e25fSSatish Balay   PetscFunctionBegin;
1080e32f2f54SBarry Smith   if (info->levels != 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only levels=0 is supported for in-place icc");
1081c84f5b01SHong Zhang   ierr = ISIdentity(row,&row_identity);CHKERRQ(ierr);
1082e32f2f54SBarry Smith   if (!row_identity) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix reordering is not supported");
1083e32f2f54SBarry Smith   if (inA->rmap->bs != 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix block size %D is not supported",inA->rmap->bs); /* Need to replace MatCholeskyFactorSymbolic_SeqSBAIJ_MSR()! */
1084c84f5b01SHong Zhang 
108549b5e25fSSatish Balay   outA            = inA;
1086d5f3da31SBarry Smith   inA->factortype = MAT_FACTOR_ICC;
108749b5e25fSSatish Balay 
10881a3463dfSHong Zhang   ierr = MatMarkDiagonal_SeqSBAIJ(inA);CHKERRQ(ierr);
1089d595f711SHong Zhang   ierr = MatSeqSBAIJSetNumericFactorization_inplace(inA,row_identity);CHKERRQ(ierr);
109049b5e25fSSatish Balay 
1091c3122656SLisandro Dalcin   ierr   = PetscObjectReference((PetscObject)row);CHKERRQ(ierr);
10926bf464f9SBarry Smith   ierr = ISDestroy(&a->row);CHKERRQ(ierr);
1093c84f5b01SHong Zhang   a->row = row;
1094c3122656SLisandro Dalcin   ierr   = PetscObjectReference((PetscObject)row);CHKERRQ(ierr);
10956bf464f9SBarry Smith   ierr = ISDestroy(&a->col);CHKERRQ(ierr);
1096c84f5b01SHong Zhang   a->col = row;
1097c84f5b01SHong Zhang 
1098c84f5b01SHong Zhang   /* Create the invert permutation so that it can be used in MatCholeskyFactorNumeric() */
1099c84f5b01SHong Zhang   if (a->icol) {ierr = ISInvertPermutation(row,PETSC_DECIDE, &a->icol);CHKERRQ(ierr);}
110052e6d16bSBarry Smith   ierr = PetscLogObjectParent(inA,a->icol);CHKERRQ(ierr);
110149b5e25fSSatish Balay 
110249b5e25fSSatish Balay   if (!a->solve_work) {
1103d0f46423SBarry Smith     ierr = PetscMalloc((inA->rmap->N+inA->rmap->bs)*sizeof(PetscScalar),&a->solve_work);CHKERRQ(ierr);
1104d0f46423SBarry Smith     ierr = PetscLogObjectMemory(inA,(inA->rmap->N+inA->rmap->bs)*sizeof(PetscScalar));CHKERRQ(ierr);
110549b5e25fSSatish Balay   }
110649b5e25fSSatish Balay 
1107719d5645SBarry Smith   ierr = MatCholeskyFactorNumeric(outA,inA,info);CHKERRQ(ierr);
110849b5e25fSSatish Balay   PetscFunctionReturn(0);
110949b5e25fSSatish Balay }
1110950f1e5bSHong Zhang 
111149b5e25fSSatish Balay EXTERN_C_BEGIN
11124a2ae208SSatish Balay #undef __FUNCT__
11134a2ae208SSatish Balay #define __FUNCT__ "MatSeqSBAIJSetColumnIndices_SeqSBAIJ"
11147087cfbeSBarry Smith PetscErrorCode  MatSeqSBAIJSetColumnIndices_SeqSBAIJ(Mat mat,PetscInt *indices)
111549b5e25fSSatish Balay {
1116045c9aa0SHong Zhang   Mat_SeqSBAIJ *baij = (Mat_SeqSBAIJ *)mat->data;
111713f74950SBarry Smith   PetscInt     i,nz,n;
111849b5e25fSSatish Balay 
111949b5e25fSSatish Balay   PetscFunctionBegin;
11206c6c5352SBarry Smith   nz = baij->maxnz;
1121d0f46423SBarry Smith   n  = mat->cmap->n;
112249b5e25fSSatish Balay   for (i=0; i<nz; i++) {
112349b5e25fSSatish Balay     baij->j[i] = indices[i];
112449b5e25fSSatish Balay   }
11256c6c5352SBarry Smith    baij->nz = nz;
112649b5e25fSSatish Balay    for (i=0; i<n; i++) {
112749b5e25fSSatish Balay      baij->ilen[i] = baij->imax[i];
112849b5e25fSSatish Balay    }
112949b5e25fSSatish Balay    PetscFunctionReturn(0);
113049b5e25fSSatish Balay }
113149b5e25fSSatish Balay EXTERN_C_END
113249b5e25fSSatish Balay 
11334a2ae208SSatish Balay #undef __FUNCT__
11344a2ae208SSatish Balay #define __FUNCT__ "MatSeqSBAIJSetColumnIndices"
113549b5e25fSSatish Balay /*@
113619585528SSatish Balay   MatSeqSBAIJSetColumnIndices - Set the column indices for all the rows
113749b5e25fSSatish Balay   in the matrix.
113849b5e25fSSatish Balay 
113949b5e25fSSatish Balay   Input Parameters:
114019585528SSatish Balay   +  mat     - the SeqSBAIJ matrix
114149b5e25fSSatish Balay   -  indices - the column indices
114249b5e25fSSatish Balay 
114349b5e25fSSatish Balay   Level: advanced
114449b5e25fSSatish Balay 
114549b5e25fSSatish Balay   Notes:
114649b5e25fSSatish Balay   This can be called if you have precomputed the nonzero structure of the
114749b5e25fSSatish Balay   matrix and want to provide it to the matrix object to improve the performance
114849b5e25fSSatish Balay   of the MatSetValues() operation.
114949b5e25fSSatish Balay 
115049b5e25fSSatish Balay   You MUST have set the correct numbers of nonzeros per row in the call to
1151d1be2dadSMatthew Knepley   MatCreateSeqSBAIJ(), and the columns indices MUST be sorted.
115249b5e25fSSatish Balay 
1153ab9f2c04SSatish Balay   MUST be called before any calls to MatSetValues()
115449b5e25fSSatish Balay 
1155ab9f2c04SSatish Balay   .seealso: MatCreateSeqSBAIJ
115649b5e25fSSatish Balay @*/
11577087cfbeSBarry Smith PetscErrorCode  MatSeqSBAIJSetColumnIndices(Mat mat,PetscInt *indices)
115849b5e25fSSatish Balay {
11594ac538c5SBarry Smith   PetscErrorCode ierr;
116049b5e25fSSatish Balay 
116149b5e25fSSatish Balay   PetscFunctionBegin;
11620700a824SBarry Smith   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
11634482741eSBarry Smith   PetscValidPointer(indices,2);
11644ac538c5SBarry Smith   ierr = PetscUseMethod(mat,"MatSeqSBAIJSetColumnIndices_C",(Mat,PetscInt *),(mat,indices));CHKERRQ(ierr);
116549b5e25fSSatish Balay   PetscFunctionReturn(0);
116649b5e25fSSatish Balay }
116749b5e25fSSatish Balay 
11684a2ae208SSatish Balay #undef __FUNCT__
11693c896bc6SHong Zhang #define __FUNCT__ "MatCopy_SeqSBAIJ"
11703c896bc6SHong Zhang PetscErrorCode MatCopy_SeqSBAIJ(Mat A,Mat B,MatStructure str)
11713c896bc6SHong Zhang {
11723c896bc6SHong Zhang   PetscErrorCode ierr;
11733c896bc6SHong Zhang 
11743c896bc6SHong Zhang   PetscFunctionBegin;
11753c896bc6SHong Zhang   /* If the two matrices have the same copy implementation, use fast copy. */
11763c896bc6SHong Zhang   if (str == SAME_NONZERO_PATTERN && (A->ops->copy == B->ops->copy)) {
11773c896bc6SHong Zhang     Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
11783c896bc6SHong Zhang     Mat_SeqSBAIJ *b = (Mat_SeqSBAIJ*)B->data;
11793c896bc6SHong Zhang 
1180e7e72b3dSBarry Smith     if (a->i[A->rmap->N] != b->i[B->rmap->N]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Number of nonzeros in two matrices are different");
1181d0f46423SBarry Smith     ierr = PetscMemcpy(b->a,a->a,(a->i[A->rmap->N])*sizeof(PetscScalar));CHKERRQ(ierr);
11823c896bc6SHong Zhang   } else {
1183f5edf698SHong Zhang     ierr = MatGetRowUpperTriangular(A);CHKERRQ(ierr);
11843c896bc6SHong Zhang     ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr);
1185f5edf698SHong Zhang     ierr = MatRestoreRowUpperTriangular(A);CHKERRQ(ierr);
11863c896bc6SHong Zhang   }
11873c896bc6SHong Zhang   PetscFunctionReturn(0);
11883c896bc6SHong Zhang }
11893c896bc6SHong Zhang 
11903c896bc6SHong Zhang #undef __FUNCT__
11914a2ae208SSatish Balay #define __FUNCT__ "MatSetUpPreallocation_SeqSBAIJ"
1192dfbe8321SBarry Smith PetscErrorCode MatSetUpPreallocation_SeqSBAIJ(Mat A)
1193273d9f13SBarry Smith {
1194dfbe8321SBarry Smith   PetscErrorCode ierr;
1195273d9f13SBarry Smith 
1196273d9f13SBarry Smith   PetscFunctionBegin;
1197db4efbfdSBarry Smith   ierr =  MatSeqSBAIJSetPreallocation_SeqSBAIJ(A,-PetscMax(A->rmap->bs,1),PETSC_DEFAULT,0);CHKERRQ(ierr);
1198273d9f13SBarry Smith   PetscFunctionReturn(0);
1199273d9f13SBarry Smith }
1200273d9f13SBarry Smith 
1201a6ece127SHong Zhang #undef __FUNCT__
1202a6ece127SHong Zhang #define __FUNCT__ "MatGetArray_SeqSBAIJ"
1203dfbe8321SBarry Smith PetscErrorCode MatGetArray_SeqSBAIJ(Mat A,PetscScalar *array[])
1204a6ece127SHong Zhang {
1205a6ece127SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
1206a6ece127SHong Zhang   PetscFunctionBegin;
1207a6ece127SHong Zhang   *array = a->a;
1208a6ece127SHong Zhang   PetscFunctionReturn(0);
1209a6ece127SHong Zhang }
1210a6ece127SHong Zhang 
1211a6ece127SHong Zhang #undef __FUNCT__
1212a6ece127SHong Zhang #define __FUNCT__ "MatRestoreArray_SeqSBAIJ"
1213dfbe8321SBarry Smith PetscErrorCode MatRestoreArray_SeqSBAIJ(Mat A,PetscScalar *array[])
1214a6ece127SHong Zhang {
1215a6ece127SHong Zhang   PetscFunctionBegin;
1216a6ece127SHong Zhang   PetscFunctionReturn(0);
1217a6ece127SHong Zhang  }
1218a6ece127SHong Zhang 
121942ee4b1aSHong Zhang #undef __FUNCT__
122042ee4b1aSHong Zhang #define __FUNCT__ "MatAXPY_SeqSBAIJ"
1221f4df32b1SMatthew Knepley PetscErrorCode MatAXPY_SeqSBAIJ(Mat Y,PetscScalar a,Mat X,MatStructure str)
122242ee4b1aSHong Zhang {
122342ee4b1aSHong Zhang   Mat_SeqSBAIJ   *x=(Mat_SeqSBAIJ *)X->data, *y=(Mat_SeqSBAIJ *)Y->data;
1224dfbe8321SBarry Smith   PetscErrorCode ierr;
1225e838b9e7SJed Brown   PetscInt       i,bs=Y->rmap->bs,bs2=bs*bs,j;
1226e838b9e7SJed Brown   PetscBLASInt   one = 1;
122742ee4b1aSHong Zhang 
122842ee4b1aSHong Zhang   PetscFunctionBegin;
122942ee4b1aSHong Zhang   if (str == SAME_NONZERO_PATTERN) {
1230f4df32b1SMatthew Knepley     PetscScalar alpha = a;
1231666a761fSJed Brown     PetscBLASInt bnz = PetscBLASIntCast(x->nz*bs2);
1232f4df32b1SMatthew Knepley     BLASaxpy_(&bnz,&alpha,x->a,&one,y->a,&one);
1233c537a176SHong Zhang   } else if (str == SUBSET_NONZERO_PATTERN) { /* nonzeros of X is a subset of Y's */
1234c4319e64SHong Zhang     if (y->xtoy && y->XtoY != X) {
1235c4319e64SHong Zhang       ierr = PetscFree(y->xtoy);CHKERRQ(ierr);
12366bf464f9SBarry Smith       ierr = MatDestroy(&y->XtoY);CHKERRQ(ierr);
1237c537a176SHong Zhang     }
1238c4319e64SHong Zhang     if (!y->xtoy) { /* get xtoy */
1239c4319e64SHong Zhang       ierr = MatAXPYGetxtoy_Private(x->mbs,x->i,x->j,PETSC_NULL, y->i,y->j,PETSC_NULL, &y->xtoy);CHKERRQ(ierr);
1240c4319e64SHong Zhang       y->XtoY = X;
1241c537a176SHong Zhang     }
12426c6c5352SBarry Smith     for (i=0; i<x->nz; i++) {
1243c4319e64SHong Zhang       j = 0;
1244c4319e64SHong Zhang       while (j < bs2){
1245f4df32b1SMatthew Knepley         y->a[bs2*y->xtoy[i]+j] += a*(x->a[bs2*i+j]);
1246c4319e64SHong Zhang         j++;
1247c537a176SHong Zhang       }
1248c4319e64SHong Zhang     }
12491e2582c4SBarry Smith     ierr = PetscInfo3(Y,"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);
125042ee4b1aSHong Zhang   } else {
1251f5edf698SHong Zhang     ierr = MatGetRowUpperTriangular(X);CHKERRQ(ierr);
1252f4df32b1SMatthew Knepley     ierr = MatAXPY_Basic(Y,a,X,str);CHKERRQ(ierr);
1253f5edf698SHong Zhang     ierr = MatRestoreRowUpperTriangular(X);CHKERRQ(ierr);
125442ee4b1aSHong Zhang   }
125542ee4b1aSHong Zhang   PetscFunctionReturn(0);
125642ee4b1aSHong Zhang }
125742ee4b1aSHong Zhang 
1258efcf0fc3SBarry Smith #undef __FUNCT__
12596363de48SJed Brown #define __FUNCT__ "MatSetBlockSize_SeqSBAIJ"
12606363de48SJed Brown PetscErrorCode MatSetBlockSize_SeqSBAIJ(Mat A,PetscInt bs)
12616363de48SJed Brown {
12626363de48SJed Brown   PetscInt rbs,cbs;
12636363de48SJed Brown   PetscErrorCode ierr;
12646363de48SJed Brown 
12656363de48SJed Brown   PetscFunctionBegin;
12666363de48SJed Brown   ierr = PetscLayoutGetBlockSize(A->rmap,&rbs);CHKERRQ(ierr);
12676363de48SJed Brown   ierr = PetscLayoutGetBlockSize(A->cmap,&cbs);CHKERRQ(ierr);
1268e32f2f54SBarry Smith   if (rbs != bs) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Attempt to set block size %d with SBAIJ %d",bs,rbs);
1269e32f2f54SBarry Smith   if (cbs != bs) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Attempt to set block size %d with SBAIJ %d",bs,cbs);
12706363de48SJed Brown   PetscFunctionReturn(0);
12716363de48SJed Brown }
12726363de48SJed Brown 
12736363de48SJed Brown #undef __FUNCT__
1274efcf0fc3SBarry Smith #define __FUNCT__ "MatIsSymmetric_SeqSBAIJ"
1275ace3abfcSBarry Smith PetscErrorCode MatIsSymmetric_SeqSBAIJ(Mat A,PetscReal tol,PetscBool  *flg)
1276efcf0fc3SBarry Smith {
1277efcf0fc3SBarry Smith   PetscFunctionBegin;
1278efcf0fc3SBarry Smith   *flg = PETSC_TRUE;
1279efcf0fc3SBarry Smith   PetscFunctionReturn(0);
1280efcf0fc3SBarry Smith }
1281efcf0fc3SBarry Smith 
1282efcf0fc3SBarry Smith #undef __FUNCT__
1283efcf0fc3SBarry Smith #define __FUNCT__ "MatIsStructurallySymmetric_SeqSBAIJ"
1284ace3abfcSBarry Smith PetscErrorCode MatIsStructurallySymmetric_SeqSBAIJ(Mat A,PetscBool  *flg)
1285efcf0fc3SBarry Smith {
1286efcf0fc3SBarry Smith    PetscFunctionBegin;
1287efcf0fc3SBarry Smith    *flg = PETSC_TRUE;
1288efcf0fc3SBarry Smith    PetscFunctionReturn(0);
1289efcf0fc3SBarry Smith }
1290efcf0fc3SBarry Smith 
1291efcf0fc3SBarry Smith #undef __FUNCT__
1292efcf0fc3SBarry Smith #define __FUNCT__ "MatIsHermitian_SeqSBAIJ"
1293ace3abfcSBarry Smith PetscErrorCode MatIsHermitian_SeqSBAIJ(Mat A,PetscReal tol,PetscBool  *flg)
1294efcf0fc3SBarry Smith  {
1295efcf0fc3SBarry Smith    PetscFunctionBegin;
1296efcf0fc3SBarry Smith    *flg = PETSC_FALSE;
1297efcf0fc3SBarry Smith    PetscFunctionReturn(0);
1298efcf0fc3SBarry Smith  }
1299efcf0fc3SBarry Smith 
130099cafbc1SBarry Smith #undef __FUNCT__
130199cafbc1SBarry Smith #define __FUNCT__ "MatRealPart_SeqSBAIJ"
130299cafbc1SBarry Smith PetscErrorCode MatRealPart_SeqSBAIJ(Mat A)
130399cafbc1SBarry Smith {
130499cafbc1SBarry Smith   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
130599cafbc1SBarry Smith   PetscInt       i,nz = a->bs2*a->i[a->mbs];
1306dd6ea824SBarry Smith   MatScalar      *aa = a->a;
130799cafbc1SBarry Smith 
130899cafbc1SBarry Smith   PetscFunctionBegin;
130999cafbc1SBarry Smith   for (i=0; i<nz; i++) aa[i] = PetscRealPart(aa[i]);
131099cafbc1SBarry Smith   PetscFunctionReturn(0);
131199cafbc1SBarry Smith }
131299cafbc1SBarry Smith 
131399cafbc1SBarry Smith #undef __FUNCT__
131499cafbc1SBarry Smith #define __FUNCT__ "MatImaginaryPart_SeqSBAIJ"
131599cafbc1SBarry Smith PetscErrorCode MatImaginaryPart_SeqSBAIJ(Mat A)
131699cafbc1SBarry Smith {
131799cafbc1SBarry Smith   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
131899cafbc1SBarry Smith   PetscInt       i,nz = a->bs2*a->i[a->mbs];
1319dd6ea824SBarry Smith   MatScalar      *aa = a->a;
132099cafbc1SBarry Smith 
132199cafbc1SBarry Smith   PetscFunctionBegin;
132299cafbc1SBarry Smith   for (i=0; i<nz; i++) aa[i] = PetscImaginaryPart(aa[i]);
132399cafbc1SBarry Smith   PetscFunctionReturn(0);
132499cafbc1SBarry Smith }
132599cafbc1SBarry Smith 
13263bededecSBarry Smith #undef __FUNCT__
13273bededecSBarry Smith #define __FUNCT__ "MatZeroRowsColumns_SeqSBAIJ"
13283bededecSBarry Smith PetscErrorCode MatZeroRowsColumns_SeqSBAIJ(Mat A,PetscInt is_n,const PetscInt is_idx[],PetscScalar diag,Vec x, Vec b)
13293bededecSBarry Smith {
13303bededecSBarry Smith   Mat_SeqSBAIJ      *baij=(Mat_SeqSBAIJ*)A->data;
13313bededecSBarry Smith   PetscErrorCode    ierr;
13323bededecSBarry Smith   PetscInt          i,j,k,count;
13333bededecSBarry Smith   PetscInt          bs=A->rmap->bs,bs2=baij->bs2,row,col;
13343bededecSBarry Smith   PetscScalar       zero = 0.0;
13353bededecSBarry Smith   MatScalar         *aa;
13363bededecSBarry Smith   const PetscScalar *xx;
13373bededecSBarry Smith   PetscScalar       *bb;
133856777dd2SBarry Smith   PetscBool         *zeroed,vecs = PETSC_FALSE;
13393bededecSBarry Smith 
13403bededecSBarry Smith   PetscFunctionBegin;
13413bededecSBarry Smith   /* fix right hand side if needed */
13423bededecSBarry Smith   if (x && b) {
13433bededecSBarry Smith     ierr = VecGetArrayRead(x,&xx);CHKERRQ(ierr);
13443bededecSBarry Smith     ierr = VecGetArray(b,&bb);CHKERRQ(ierr);
134556777dd2SBarry Smith     vecs = PETSC_TRUE;
13463bededecSBarry Smith   }
13473bededecSBarry Smith   A->same_nonzero = PETSC_TRUE;
13483bededecSBarry Smith 
13493bededecSBarry Smith   /* zero the columns */
13503bededecSBarry Smith   ierr = PetscMalloc(A->rmap->n*sizeof(PetscBool),&zeroed);CHKERRQ(ierr);
13513bededecSBarry Smith   ierr = PetscMemzero(zeroed,A->rmap->n*sizeof(PetscBool));CHKERRQ(ierr);
13523bededecSBarry Smith   for (i=0; i<is_n; i++) {
13533bededecSBarry Smith     if (is_idx[i] < 0 || is_idx[i] >= A->rmap->N) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"row %D out of range",is_idx[i]);
13543bededecSBarry Smith     zeroed[is_idx[i]] = PETSC_TRUE;
13553bededecSBarry Smith   }
135656777dd2SBarry Smith   if (vecs) {
135756777dd2SBarry Smith     for (i=0; i<A->rmap->N; i++) {
135856777dd2SBarry Smith       row = i/bs;
135956777dd2SBarry Smith       for (j=baij->i[row]; j<baij->i[row+1]; j++) {
136056777dd2SBarry Smith 	for (k=0; k<bs; k++) {
136156777dd2SBarry Smith 	  col = bs*baij->j[j] + k;
136256777dd2SBarry Smith           if (col <= i) continue;
136356777dd2SBarry Smith 	  aa = ((MatScalar*)(baij->a)) + j*bs2 + (i%bs) + bs*k;
136456777dd2SBarry Smith 	  if (!zeroed[i] && zeroed[col]) {
136556777dd2SBarry Smith 	    bb[i] -= aa[0]*xx[col];
136656777dd2SBarry Smith 	  }
136756777dd2SBarry Smith 	  if (zeroed[i] && !zeroed[col]) {
136856777dd2SBarry Smith 	    bb[col] -= aa[0]*xx[i];
136956777dd2SBarry Smith 	  }
137056777dd2SBarry Smith 	}
137156777dd2SBarry Smith       }
137256777dd2SBarry Smith     }
137356777dd2SBarry Smith     for (i=0; i<is_n; i++) {
137456777dd2SBarry Smith       bb[is_idx[i]] = diag*xx[is_idx[i]];
137556777dd2SBarry Smith     }
137656777dd2SBarry Smith   }
137756777dd2SBarry Smith 
13783bededecSBarry Smith   for (i=0; i<A->rmap->N; i++) {
13793bededecSBarry Smith     if (!zeroed[i]) {
13803bededecSBarry Smith       row = i/bs;
13813bededecSBarry Smith       for (j=baij->i[row]; j<baij->i[row+1]; j++) {
13823bededecSBarry Smith         for (k=0; k<bs; k++) {
13833bededecSBarry Smith           col = bs*baij->j[j] + k;
13843bededecSBarry Smith 	  if (zeroed[col]) {
13853bededecSBarry Smith 	    aa = ((MatScalar*)(baij->a)) + j*bs2 + (i%bs) + bs*k;
13863bededecSBarry Smith             aa[0] = 0.0;
13873bededecSBarry Smith           }
13883bededecSBarry Smith         }
13893bededecSBarry Smith       }
13903bededecSBarry Smith     }
13913bededecSBarry Smith   }
13923bededecSBarry Smith   ierr = PetscFree(zeroed);CHKERRQ(ierr);
139356777dd2SBarry Smith   if (vecs) {
139456777dd2SBarry Smith     ierr = VecRestoreArrayRead(x,&xx);CHKERRQ(ierr);
139556777dd2SBarry Smith     ierr = VecRestoreArray(b,&bb);CHKERRQ(ierr);
139656777dd2SBarry Smith   }
13973bededecSBarry Smith 
13983bededecSBarry Smith   /* zero the rows */
13993bededecSBarry Smith   for (i=0; i<is_n; i++) {
14003bededecSBarry Smith     row   = is_idx[i];
14013bededecSBarry Smith     count = (baij->i[row/bs +1] - baij->i[row/bs])*bs;
14023bededecSBarry Smith     aa    = ((MatScalar*)(baij->a)) + baij->i[row/bs]*bs2 + (row%bs);
14033bededecSBarry Smith     for (k=0; k<count; k++) {
14043bededecSBarry Smith       aa[0] =  zero;
14053bededecSBarry Smith       aa    += bs;
14063bededecSBarry Smith     }
14073bededecSBarry Smith     if (diag != 0.0) {
14083bededecSBarry Smith       ierr = (*A->ops->setvalues)(A,1,&row,1,&row,&diag,INSERT_VALUES);CHKERRQ(ierr);
14093bededecSBarry Smith     }
14103bededecSBarry Smith   }
14113bededecSBarry Smith   ierr = MatAssemblyEnd_SeqSBAIJ(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
14123bededecSBarry Smith   PetscFunctionReturn(0);
14133bededecSBarry Smith }
14143bededecSBarry Smith 
141549b5e25fSSatish Balay /* -------------------------------------------------------------------*/
141649b5e25fSSatish Balay static struct _MatOps MatOps_Values = {MatSetValues_SeqSBAIJ,
141749b5e25fSSatish Balay        MatGetRow_SeqSBAIJ,
141849b5e25fSSatish Balay        MatRestoreRow_SeqSBAIJ,
141949b5e25fSSatish Balay        MatMult_SeqSBAIJ_N,
142097304618SKris Buschelman /* 4*/ MatMultAdd_SeqSBAIJ_N,
1421431c96f7SBarry Smith        MatMult_SeqSBAIJ_N,       /* transpose versions are same as non-transpose versions */
1422e005ede5SBarry Smith        MatMultAdd_SeqSBAIJ_N,
1423db4efbfdSBarry Smith        0,
142449b5e25fSSatish Balay        0,
142549b5e25fSSatish Balay        0,
142697304618SKris Buschelman /*10*/ 0,
142749b5e25fSSatish Balay        0,
1428c078aec8SLisandro Dalcin        MatCholeskyFactor_SeqSBAIJ,
142941f059aeSBarry Smith        MatSOR_SeqSBAIJ,
143049b5e25fSSatish Balay        MatTranspose_SeqSBAIJ,
143197304618SKris Buschelman /*15*/ MatGetInfo_SeqSBAIJ,
143249b5e25fSSatish Balay        MatEqual_SeqSBAIJ,
143349b5e25fSSatish Balay        MatGetDiagonal_SeqSBAIJ,
143449b5e25fSSatish Balay        MatDiagonalScale_SeqSBAIJ,
143549b5e25fSSatish Balay        MatNorm_SeqSBAIJ,
143697304618SKris Buschelman /*20*/ 0,
143749b5e25fSSatish Balay        MatAssemblyEnd_SeqSBAIJ,
143849b5e25fSSatish Balay        MatSetOption_SeqSBAIJ,
143949b5e25fSSatish Balay        MatZeroEntries_SeqSBAIJ,
1440d519adbfSMatthew Knepley /*24*/ 0,
144149b5e25fSSatish Balay        0,
144249b5e25fSSatish Balay        0,
1443db4efbfdSBarry Smith        0,
1444db4efbfdSBarry Smith        0,
1445d519adbfSMatthew Knepley /*29*/ MatSetUpPreallocation_SeqSBAIJ,
1446c464158bSHong Zhang        0,
1447db4efbfdSBarry Smith        0,
1448a6ece127SHong Zhang        MatGetArray_SeqSBAIJ,
1449a6ece127SHong Zhang        MatRestoreArray_SeqSBAIJ,
1450d519adbfSMatthew Knepley /*34*/ MatDuplicate_SeqSBAIJ,
1451719d5645SBarry Smith        0,
1452719d5645SBarry Smith        0,
145349b5e25fSSatish Balay        0,
1454c84f5b01SHong Zhang        MatICCFactor_SeqSBAIJ,
1455d519adbfSMatthew Knepley /*39*/ MatAXPY_SeqSBAIJ,
145649b5e25fSSatish Balay        MatGetSubMatrices_SeqSBAIJ,
145749b5e25fSSatish Balay        MatIncreaseOverlap_SeqSBAIJ,
145849b5e25fSSatish Balay        MatGetValues_SeqSBAIJ,
14593c896bc6SHong Zhang        MatCopy_SeqSBAIJ,
1460d519adbfSMatthew Knepley /*44*/ 0,
146149b5e25fSSatish Balay        MatScale_SeqSBAIJ,
146249b5e25fSSatish Balay        0,
146349b5e25fSSatish Balay        0,
14643bededecSBarry Smith        MatZeroRowsColumns_SeqSBAIJ,
14656363de48SJed Brown /*49*/ MatSetBlockSize_SeqSBAIJ,
146649b5e25fSSatish Balay        MatGetRowIJ_SeqSBAIJ,
146749b5e25fSSatish Balay        MatRestoreRowIJ_SeqSBAIJ,
146849b5e25fSSatish Balay        0,
146949b5e25fSSatish Balay        0,
1470d519adbfSMatthew Knepley /*54*/ 0,
147149b5e25fSSatish Balay        0,
147249b5e25fSSatish Balay        0,
147349b5e25fSSatish Balay        0,
147449b5e25fSSatish Balay        MatSetValuesBlocked_SeqSBAIJ,
1475d519adbfSMatthew Knepley /*59*/ MatGetSubMatrix_SeqSBAIJ,
147649b5e25fSSatish Balay        0,
147749b5e25fSSatish Balay        0,
1478357abbc8SBarry Smith        0,
1479d959ec07SHong Zhang        0,
1480d519adbfSMatthew Knepley /*64*/ 0,
1481d959ec07SHong Zhang        0,
1482d959ec07SHong Zhang        0,
1483d959ec07SHong Zhang        0,
1484d959ec07SHong Zhang        0,
1485d519adbfSMatthew Knepley /*69*/ MatGetRowMaxAbs_SeqSBAIJ,
14863e0d88b5SBarry Smith        0,
14873e0d88b5SBarry Smith        0,
14883e0d88b5SBarry Smith        0,
14893e0d88b5SBarry Smith        0,
1490d519adbfSMatthew Knepley /*74*/ 0,
14913e0d88b5SBarry Smith        0,
14923e0d88b5SBarry Smith        0,
14933e0d88b5SBarry Smith        0,
14943e0d88b5SBarry Smith        0,
1495d519adbfSMatthew Knepley /*79*/ 0,
14963e0d88b5SBarry Smith        0,
14973e0d88b5SBarry Smith        0,
149897304618SKris Buschelman        MatGetInertia_SeqSBAIJ,
14995bba2384SShri Abhyankar        MatLoad_SeqSBAIJ,
1500d519adbfSMatthew Knepley /*84*/ MatIsSymmetric_SeqSBAIJ,
1501865e5f61SKris Buschelman        MatIsHermitian_SeqSBAIJ,
1502efcf0fc3SBarry Smith        MatIsStructurallySymmetric_SeqSBAIJ,
1503865e5f61SKris Buschelman        0,
1504865e5f61SKris Buschelman        0,
1505d519adbfSMatthew Knepley /*89*/ 0,
1506865e5f61SKris Buschelman        0,
1507865e5f61SKris Buschelman        0,
1508865e5f61SKris Buschelman        0,
1509865e5f61SKris Buschelman        0,
1510d519adbfSMatthew Knepley /*94*/ 0,
1511865e5f61SKris Buschelman        0,
1512865e5f61SKris Buschelman        0,
151399cafbc1SBarry Smith        0,
151499cafbc1SBarry Smith        0,
1515d519adbfSMatthew Knepley /*99*/ 0,
151699cafbc1SBarry Smith        0,
151799cafbc1SBarry Smith        0,
151899cafbc1SBarry Smith        0,
151999cafbc1SBarry Smith        0,
1520d519adbfSMatthew Knepley /*104*/0,
152199cafbc1SBarry Smith        MatRealPart_SeqSBAIJ,
1522f5edf698SHong Zhang        MatImaginaryPart_SeqSBAIJ,
1523f5edf698SHong Zhang        MatGetRowUpperTriangular_SeqSBAIJ,
15242af78befSBarry Smith        MatRestoreRowUpperTriangular_SeqSBAIJ,
1525d519adbfSMatthew Knepley /*109*/0,
15262af78befSBarry Smith        0,
15272af78befSBarry Smith        0,
15282af78befSBarry Smith        0,
1529547795f9SHong Zhang        MatMissingDiagonal_SeqSBAIJ,
1530547795f9SHong Zhang /*114*/0,
1531547795f9SHong Zhang        0,
1532547795f9SHong Zhang        0,
1533547795f9SHong Zhang        0,
1534547795f9SHong Zhang        0,
1535547795f9SHong Zhang /*119*/0,
1536547795f9SHong Zhang        0,
15372f480046SShri Abhyankar        0,
15385bba2384SShri Abhyankar        0
153999cafbc1SBarry Smith };
1540be1d678aSKris Buschelman 
154149b5e25fSSatish Balay EXTERN_C_BEGIN
15424a2ae208SSatish Balay #undef __FUNCT__
15434a2ae208SSatish Balay #define __FUNCT__ "MatStoreValues_SeqSBAIJ"
15447087cfbeSBarry Smith PetscErrorCode  MatStoreValues_SeqSBAIJ(Mat mat)
154549b5e25fSSatish Balay {
15464afc71dfSHong Zhang   Mat_SeqSBAIJ   *aij = (Mat_SeqSBAIJ *)mat->data;
1547d0f46423SBarry Smith   PetscInt       nz = aij->i[mat->rmap->N]*mat->rmap->bs*aij->bs2;
1548dfbe8321SBarry Smith   PetscErrorCode ierr;
154949b5e25fSSatish Balay 
155049b5e25fSSatish Balay   PetscFunctionBegin;
1551e7e72b3dSBarry Smith   if (aij->nonew != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first");
155249b5e25fSSatish Balay 
155349b5e25fSSatish Balay   /* allocate space for values if not already there */
155449b5e25fSSatish Balay   if (!aij->saved_values) {
155587828ca2SBarry Smith     ierr = PetscMalloc((nz+1)*sizeof(PetscScalar),&aij->saved_values);CHKERRQ(ierr);
155649b5e25fSSatish Balay   }
155749b5e25fSSatish Balay 
155849b5e25fSSatish Balay   /* copy values over */
155987828ca2SBarry Smith   ierr = PetscMemcpy(aij->saved_values,aij->a,nz*sizeof(PetscScalar));CHKERRQ(ierr);
156049b5e25fSSatish Balay   PetscFunctionReturn(0);
156149b5e25fSSatish Balay }
156249b5e25fSSatish Balay EXTERN_C_END
156349b5e25fSSatish Balay 
156449b5e25fSSatish Balay EXTERN_C_BEGIN
15654a2ae208SSatish Balay #undef __FUNCT__
15664a2ae208SSatish Balay #define __FUNCT__ "MatRetrieveValues_SeqSBAIJ"
15677087cfbeSBarry Smith PetscErrorCode  MatRetrieveValues_SeqSBAIJ(Mat mat)
156849b5e25fSSatish Balay {
15694afc71dfSHong Zhang   Mat_SeqSBAIJ   *aij = (Mat_SeqSBAIJ *)mat->data;
15706849ba73SBarry Smith   PetscErrorCode ierr;
1571d0f46423SBarry Smith   PetscInt       nz = aij->i[mat->rmap->N]*mat->rmap->bs*aij->bs2;
157249b5e25fSSatish Balay 
157349b5e25fSSatish Balay   PetscFunctionBegin;
1574e7e72b3dSBarry Smith   if (aij->nonew != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first");
1575e7e72b3dSBarry Smith   if (!aij->saved_values) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatStoreValues(A);first");
157649b5e25fSSatish Balay 
157749b5e25fSSatish Balay   /* copy values over */
157887828ca2SBarry Smith   ierr = PetscMemcpy(aij->a,aij->saved_values,nz*sizeof(PetscScalar));CHKERRQ(ierr);
157949b5e25fSSatish Balay   PetscFunctionReturn(0);
158049b5e25fSSatish Balay }
158149b5e25fSSatish Balay EXTERN_C_END
158249b5e25fSSatish Balay 
15838549e402SHong Zhang EXTERN_C_BEGIN
15844a2ae208SSatish Balay #undef __FUNCT__
1585a23d5eceSKris Buschelman #define __FUNCT__ "MatSeqSBAIJSetPreallocation_SeqSBAIJ"
15867087cfbeSBarry Smith PetscErrorCode  MatSeqSBAIJSetPreallocation_SeqSBAIJ(Mat B,PetscInt bs,PetscInt nz,PetscInt *nnz)
158749b5e25fSSatish Balay {
1588c464158bSHong Zhang   Mat_SeqSBAIJ   *b = (Mat_SeqSBAIJ*)B->data;
15896849ba73SBarry Smith   PetscErrorCode ierr;
1590db4efbfdSBarry Smith   PetscInt       i,mbs,bs2, newbs = PetscAbs(bs);
1591ace3abfcSBarry Smith   PetscBool      skipallocation = PETSC_FALSE,flg = PETSC_FALSE;
159249b5e25fSSatish Balay 
159349b5e25fSSatish Balay   PetscFunctionBegin;
1594273d9f13SBarry Smith   B->preallocated = PETSC_TRUE;
1595db4efbfdSBarry Smith   if (bs < 0) {
1596db4efbfdSBarry Smith     ierr = PetscOptionsBegin(((PetscObject)B)->comm,((PetscObject)B)->prefix,"Options for MPISBAIJ matrix","Mat");CHKERRQ(ierr);
1597db4efbfdSBarry Smith       ierr = PetscOptionsInt("-mat_block_size","Set the blocksize used to store the matrix","MatSeqSBAIJSetPreallocation",newbs,&newbs,PETSC_NULL);CHKERRQ(ierr);
1598db4efbfdSBarry Smith     ierr = PetscOptionsEnd();CHKERRQ(ierr);
1599db4efbfdSBarry Smith     bs   = PetscAbs(bs);
1600db4efbfdSBarry Smith   }
1601e7e72b3dSBarry Smith   if (nnz && newbs != bs) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Cannot change blocksize from command line if setting nnz");
1602db4efbfdSBarry Smith   bs = newbs;
1603db4efbfdSBarry Smith 
160426283091SBarry Smith   ierr = PetscLayoutSetBlockSize(B->rmap,bs);CHKERRQ(ierr);
160526283091SBarry Smith   ierr = PetscLayoutSetBlockSize(B->cmap,bs);CHKERRQ(ierr);
160626283091SBarry Smith   ierr = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr);
160726283091SBarry Smith   ierr = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr);
1608899cda47SBarry Smith 
1609d0f46423SBarry Smith   mbs  = B->rmap->N/bs;
161049b5e25fSSatish Balay   bs2  = bs*bs;
161149b5e25fSSatish Balay 
1612e7e72b3dSBarry Smith   if (mbs*bs != B->rmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Number rows, cols must be divisible by blocksize");
161349b5e25fSSatish Balay 
1614ab93d7beSBarry Smith   if (nz == MAT_SKIP_ALLOCATION) {
1615ab93d7beSBarry Smith     skipallocation = PETSC_TRUE;
1616ab93d7beSBarry Smith     nz             = 0;
1617ab93d7beSBarry Smith   }
1618ab93d7beSBarry Smith 
1619435da068SBarry Smith   if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 3;
1620e32f2f54SBarry Smith   if (nz < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"nz cannot be less than 0: value %D",nz);
162149b5e25fSSatish Balay   if (nnz) {
162249b5e25fSSatish Balay     for (i=0; i<mbs; i++) {
1623e32f2f54SBarry Smith       if (nnz[i] < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"nnz cannot be less than 0: local row %D value %D",i,nnz[i]);
1624e32f2f54SBarry Smith       if (nnz[i] > mbs) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"nnz cannot be greater than block row length: local row %D value %D rowlength %D",i,nnz[i],mbs);
162549b5e25fSSatish Balay     }
162649b5e25fSSatish Balay   }
162749b5e25fSSatish Balay 
1628db4efbfdSBarry Smith   B->ops->mult             = MatMult_SeqSBAIJ_N;
1629db4efbfdSBarry Smith   B->ops->multadd          = MatMultAdd_SeqSBAIJ_N;
1630db4efbfdSBarry Smith   B->ops->multtranspose    = MatMult_SeqSBAIJ_N;
1631db4efbfdSBarry Smith   B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_N;
1632acfcf0e5SJed Brown   ierr  = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_no_unroll",&flg,PETSC_NULL);CHKERRQ(ierr);
163349b5e25fSSatish Balay   if (!flg) {
163449b5e25fSSatish Balay     switch (bs) {
163549b5e25fSSatish Balay     case 1:
163649b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_1;
163749b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_1;
1638431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_1;
1639431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_1;
164049b5e25fSSatish Balay       break;
164149b5e25fSSatish Balay     case 2:
164249b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_2;
164349b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_2;
1644431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_2;
1645431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_2;
164649b5e25fSSatish Balay       break;
164749b5e25fSSatish Balay     case 3:
164849b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_3;
164949b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_3;
1650431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_3;
1651431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_3;
165249b5e25fSSatish Balay       break;
165349b5e25fSSatish Balay     case 4:
165449b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_4;
165549b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_4;
1656431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_4;
1657431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_4;
165849b5e25fSSatish Balay       break;
165949b5e25fSSatish Balay     case 5:
166049b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_5;
166149b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_5;
1662431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_5;
1663431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_5;
166449b5e25fSSatish Balay       break;
166549b5e25fSSatish Balay     case 6:
166649b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_6;
166749b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_6;
1668431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_6;
1669431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_6;
167049b5e25fSSatish Balay       break;
167149b5e25fSSatish Balay     case 7:
1672de53e5efSHong Zhang       B->ops->mult             = MatMult_SeqSBAIJ_7;
167349b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_7;
1674431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_7;
1675431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_7;
167649b5e25fSSatish Balay       break;
167749b5e25fSSatish Balay     }
167849b5e25fSSatish Balay   }
167949b5e25fSSatish Balay 
168049b5e25fSSatish Balay   b->mbs = mbs;
16814afc71dfSHong Zhang   b->nbs = mbs;
1682ab93d7beSBarry Smith   if (!skipallocation) {
16832ee49352SLisandro Dalcin     if (!b->imax) {
1684ab93d7beSBarry Smith       ierr = PetscMalloc2(mbs,PetscInt,&b->imax,mbs,PetscInt,&b->ilen);CHKERRQ(ierr);
1685c760cd28SBarry Smith       b->free_imax_ilen = PETSC_TRUE;
16862ee49352SLisandro Dalcin       ierr = PetscLogObjectMemory(B,2*mbs*sizeof(PetscInt));CHKERRQ(ierr);
16872ee49352SLisandro Dalcin     }
168849b5e25fSSatish Balay     if (!nnz) {
1689435da068SBarry Smith       if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 5;
169049b5e25fSSatish Balay       else if (nz <= 0)        nz = 1;
169149b5e25fSSatish Balay       for (i=0; i<mbs; i++) {
16928cef66ccSHong Zhang         b->imax[i] = nz;
169349b5e25fSSatish Balay       }
1694153ea458SHong Zhang       nz = nz*mbs; /* total nz */
169549b5e25fSSatish Balay     } else {
169649b5e25fSSatish Balay       nz = 0;
16978cef66ccSHong Zhang       for (i=0; i<mbs; i++) {b->imax[i] = nnz[i]; nz += nnz[i];}
169849b5e25fSSatish Balay     }
16992ee49352SLisandro Dalcin     /* b->ilen will count nonzeros in each block row so far. */
17002ee49352SLisandro Dalcin     for (i=0; i<mbs; i++) { b->ilen[i] = 0;}
17016c6c5352SBarry Smith     /* nz=(nz+mbs)/2; */ /* total diagonal and superdiagonal nonzero blocks */
170249b5e25fSSatish Balay 
170349b5e25fSSatish Balay     /* allocate the matrix space */
17042ee49352SLisandro Dalcin     ierr = MatSeqXAIJFreeAIJ(B,&b->a,&b->j,&b->i);CHKERRQ(ierr);
1705d0f46423SBarry Smith     ierr = PetscMalloc3(bs2*nz,PetscScalar,&b->a,nz,PetscInt,&b->j,B->rmap->N+1,PetscInt,&b->i);CHKERRQ(ierr);
1706d0f46423SBarry Smith     ierr = PetscLogObjectMemory(B,(B->rmap->N+1)*sizeof(PetscInt)+nz*(bs2*sizeof(PetscScalar)+sizeof(PetscInt)));CHKERRQ(ierr);
17076c6c5352SBarry Smith     ierr = PetscMemzero(b->a,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr);
170813f74950SBarry Smith     ierr = PetscMemzero(b->j,nz*sizeof(PetscInt));CHKERRQ(ierr);
170949b5e25fSSatish Balay     b->singlemalloc = PETSC_TRUE;
171049b5e25fSSatish Balay 
171149b5e25fSSatish Balay     /* pointer to beginning of each row */
1712e60cf9a0SBarry Smith     b->i[0] = 0;
171349b5e25fSSatish Balay     for (i=1; i<mbs+1; i++) {
171449b5e25fSSatish Balay       b->i[i] = b->i[i-1] + b->imax[i-1];
171549b5e25fSSatish Balay     }
1716e6b907acSBarry Smith     b->free_a     = PETSC_TRUE;
1717e6b907acSBarry Smith     b->free_ij    = PETSC_TRUE;
1718e811da20SHong Zhang   } else {
1719e6b907acSBarry Smith     b->free_a     = PETSC_FALSE;
1720e6b907acSBarry Smith     b->free_ij    = PETSC_FALSE;
1721ab93d7beSBarry Smith   }
172249b5e25fSSatish Balay 
1723d0f46423SBarry Smith   B->rmap->bs     = bs;
172449b5e25fSSatish Balay   b->bs2          = bs2;
17256c6c5352SBarry Smith   b->nz           = 0;
1726b32cb4a7SJed Brown   b->maxnz        = nz;
1727153ea458SHong Zhang 
172816cdd363SHong Zhang   b->inew             = 0;
172916cdd363SHong Zhang   b->jnew             = 0;
173016cdd363SHong Zhang   b->anew             = 0;
173116cdd363SHong Zhang   b->a2anew           = 0;
17321a3463dfSHong Zhang   b->permute          = PETSC_FALSE;
1733c464158bSHong Zhang   PetscFunctionReturn(0);
1734c464158bSHong Zhang }
1735a23d5eceSKris Buschelman EXTERN_C_END
1736153ea458SHong Zhang 
1737db4efbfdSBarry Smith /*
1738db4efbfdSBarry Smith    This is used to set the numeric factorization for both Cholesky and ICC symbolic factorization
1739db4efbfdSBarry Smith */
17408b1456e3SHong Zhang #undef __FUNCT__
1741d595f711SHong Zhang #define __FUNCT__ "MatSeqSBAIJSetNumericFactorization_inplace"
1742ace3abfcSBarry Smith PetscErrorCode MatSeqSBAIJSetNumericFactorization_inplace(Mat B,PetscBool  natural)
1743db4efbfdSBarry Smith {
1744db4efbfdSBarry Smith   PetscErrorCode ierr;
1745ace3abfcSBarry Smith   PetscBool      flg = PETSC_FALSE;
1746db4efbfdSBarry Smith   PetscInt       bs = B->rmap->bs;
1747db4efbfdSBarry Smith 
1748db4efbfdSBarry Smith   PetscFunctionBegin;
1749acfcf0e5SJed Brown   ierr    = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_no_unroll",&flg,PETSC_NULL);CHKERRQ(ierr);
1750db4efbfdSBarry Smith   if (flg) bs = 8;
1751db4efbfdSBarry Smith 
1752db4efbfdSBarry Smith   if (!natural) {
1753db4efbfdSBarry Smith     switch (bs) {
1754db4efbfdSBarry Smith     case 1:
1755d595f711SHong Zhang       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_inplace;
1756db4efbfdSBarry Smith       break;
1757db4efbfdSBarry Smith     case 2:
1758db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2;
1759db4efbfdSBarry Smith       break;
1760db4efbfdSBarry Smith     case 3:
1761db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3;
1762db4efbfdSBarry Smith       break;
1763db4efbfdSBarry Smith     case 4:
1764db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4;
1765db4efbfdSBarry Smith       break;
1766db4efbfdSBarry Smith     case 5:
1767db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5;
1768db4efbfdSBarry Smith       break;
1769db4efbfdSBarry Smith     case 6:
1770db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6;
1771db4efbfdSBarry Smith       break;
1772db4efbfdSBarry Smith     case 7:
1773db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7;
1774db4efbfdSBarry Smith       break;
1775db4efbfdSBarry Smith     default:
1776db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N;
1777db4efbfdSBarry Smith       break;
1778db4efbfdSBarry Smith     }
1779db4efbfdSBarry Smith   } else {
1780db4efbfdSBarry Smith     switch (bs) {
1781db4efbfdSBarry Smith     case 1:
1782d595f711SHong Zhang       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering_inplace;
1783db4efbfdSBarry Smith       break;
1784db4efbfdSBarry Smith     case 2:
1785db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering;
1786db4efbfdSBarry Smith       break;
1787db4efbfdSBarry Smith     case 3:
1788db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering;
1789db4efbfdSBarry Smith       break;
1790db4efbfdSBarry Smith     case 4:
1791db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering;
1792db4efbfdSBarry Smith       break;
1793db4efbfdSBarry Smith     case 5:
1794db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering;
1795db4efbfdSBarry Smith       break;
1796db4efbfdSBarry Smith     case 6:
1797db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering;
1798db4efbfdSBarry Smith       break;
1799db4efbfdSBarry Smith     case 7:
1800db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering;
1801db4efbfdSBarry Smith       break;
1802db4efbfdSBarry Smith     default:
1803db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering;
1804db4efbfdSBarry Smith       break;
1805db4efbfdSBarry Smith     }
1806db4efbfdSBarry Smith   }
1807db4efbfdSBarry Smith   PetscFunctionReturn(0);
1808db4efbfdSBarry Smith }
1809db4efbfdSBarry Smith 
1810d769727bSBarry Smith EXTERN_C_BEGIN
18117087cfbeSBarry Smith extern PetscErrorCode  MatConvert_SeqSBAIJ_SeqAIJ(Mat, MatType,MatReuse,Mat*);
18127087cfbeSBarry Smith extern PetscErrorCode  MatConvert_SeqSBAIJ_SeqBAIJ(Mat, MatType,MatReuse,Mat*);
1813d769727bSBarry Smith EXTERN_C_END
1814d769727bSBarry Smith 
1815e631078cSBarry Smith 
1816e631078cSBarry Smith EXTERN_C_BEGIN
18175c9eb25fSBarry Smith #undef __FUNCT__
18185c9eb25fSBarry Smith #define __FUNCT__ "MatGetFactor_seqsbaij_petsc"
18195c9eb25fSBarry Smith PetscErrorCode MatGetFactor_seqsbaij_petsc(Mat A,MatFactorType ftype,Mat *B)
18205c9eb25fSBarry Smith {
1821d0f46423SBarry Smith   PetscInt           n = A->rmap->n;
18225c9eb25fSBarry Smith   PetscErrorCode     ierr;
18235c9eb25fSBarry Smith 
18245c9eb25fSBarry Smith   PetscFunctionBegin;
18255c9eb25fSBarry Smith   ierr = MatCreate(((PetscObject)A)->comm,B);CHKERRQ(ierr);
18265c9eb25fSBarry Smith   ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr);
18275c9eb25fSBarry Smith   if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) {
18285c9eb25fSBarry Smith     ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr);
18295c9eb25fSBarry Smith     ierr = MatSeqSBAIJSetPreallocation(*B,1,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr);
18307b056e98SHong Zhang     (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqSBAIJ;
1831c6d0d4f0SHong Zhang     (*B)->ops->iccfactorsymbolic      = MatICCFactorSymbolic_SeqSBAIJ;
1832e32f2f54SBarry Smith   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Factor type not supported");
1833d5f3da31SBarry Smith   (*B)->factortype = ftype;
18345c9eb25fSBarry Smith   PetscFunctionReturn(0);
18355c9eb25fSBarry Smith }
1836e631078cSBarry Smith EXTERN_C_END
18375c9eb25fSBarry Smith 
18385c9eb25fSBarry Smith EXTERN_C_BEGIN
1839db4efbfdSBarry Smith #undef __FUNCT__
1840db4efbfdSBarry Smith #define __FUNCT__ "MatGetFactorAvailable_seqsbaij_petsc"
1841ace3abfcSBarry Smith PetscErrorCode MatGetFactorAvailable_seqsbaij_petsc(Mat A,MatFactorType ftype,PetscBool  *flg)
1842db4efbfdSBarry Smith {
1843db4efbfdSBarry Smith   PetscFunctionBegin;
1844db4efbfdSBarry Smith   if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) {
1845db4efbfdSBarry Smith     *flg = PETSC_TRUE;
1846db4efbfdSBarry Smith   } else {
1847db4efbfdSBarry Smith     *flg = PETSC_FALSE;
1848db4efbfdSBarry Smith   }
1849db4efbfdSBarry Smith   PetscFunctionReturn(0);
1850db4efbfdSBarry Smith }
1851db4efbfdSBarry Smith EXTERN_C_END
1852db4efbfdSBarry Smith 
1853db4efbfdSBarry Smith EXTERN_C_BEGIN
1854611f576cSBarry Smith #if defined(PETSC_HAVE_MUMPS)
1855bccb9932SShri Abhyankar extern PetscErrorCode MatGetFactor_sbaij_mumps(Mat,MatFactorType,Mat*);
1856611f576cSBarry Smith #endif
1857611f576cSBarry Smith #if defined(PETSC_HAVE_SPOOLES)
18585c9eb25fSBarry Smith extern PetscErrorCode MatGetFactor_seqsbaij_spooles(Mat,MatFactorType,Mat*);
1859611f576cSBarry Smith #endif
1860b5e56a35SBarry Smith #if defined(PETSC_HAVE_PASTIX)
1861b5e56a35SBarry Smith extern PetscErrorCode MatGetFactor_seqsbaij_pastix(Mat,MatFactorType,Mat*);
1862b5e56a35SBarry Smith #endif
186320db9a53SJed Brown #if defined(PETSC_HAVE_CHOLMOD)
186420db9a53SJed Brown extern PetscErrorCode MatGetFactor_seqsbaij_cholmod(Mat,MatFactorType,Mat*);
186520db9a53SJed Brown #endif
18662938c13dSDahai Guo extern PetscErrorCode MatGetFactor_seqsbaij_sbstrm(Mat,MatFactorType,Mat*);
1867b1f23a54SSatish Balay EXTERN_C_END
18685c9eb25fSBarry Smith 
18690bad9183SKris Buschelman /*MC
1870fafad747SKris Buschelman   MATSEQSBAIJ - MATSEQSBAIJ = "seqsbaij" - A matrix type to be used for sequential symmetric block sparse matrices,
18710bad9183SKris Buschelman   based on block compressed sparse row format.  Only the upper triangular portion of the matrix is stored.
18720bad9183SKris Buschelman 
1873828413b8SBarry Smith   For complex numbers by default this matrix is symmetric, NOT Hermitian symmetric. To make it Hermitian symmetric you
187471dad5bbSBarry Smith   can call MatSetOption(Mat, MAT_HERMITIAN); after MatAssemblyEnd()
1875828413b8SBarry Smith 
18760bad9183SKris Buschelman   Options Database Keys:
18770bad9183SKris Buschelman   . -mat_type seqsbaij - sets the matrix type to "seqsbaij" during a call to MatSetFromOptions()
18780bad9183SKris Buschelman 
187971dad5bbSBarry Smith   Notes: By default if you insert values into the lower triangular part of the matrix they are simply ignored (since they are not
188071dad5bbSBarry Smith      stored and it is assumed they symmetric to the upper triangular). If you call MatSetOption(Mat,MAT_IGNORE_LOWER_TRIANGULAR,PETSC_FALSE) or use
188171dad5bbSBarry Smith      the options database -mat_ignore_lower_triangular false it will generate an error if you try to set a value in the lower triangular portion.
188271dad5bbSBarry Smith 
188371dad5bbSBarry Smith 
18840bad9183SKris Buschelman   Level: beginner
18850bad9183SKris Buschelman 
18860bad9183SKris Buschelman   .seealso: MatCreateSeqSBAIJ
18870bad9183SKris Buschelman M*/
18880bad9183SKris Buschelman 
1889a23d5eceSKris Buschelman EXTERN_C_BEGIN
1890aa5a9175SDahai Guo extern PetscErrorCode  MatConvert_SeqSBAIJ_SeqSBSTRM(Mat, MatType,MatReuse,Mat*);
1891aa5a9175SDahai Guo EXTERN_C_END
1892aa5a9175SDahai Guo 
1893aa5a9175SDahai Guo 
1894aa5a9175SDahai Guo EXTERN_C_BEGIN
1895a23d5eceSKris Buschelman #undef __FUNCT__
1896a23d5eceSKris Buschelman #define __FUNCT__ "MatCreate_SeqSBAIJ"
18977087cfbeSBarry Smith PetscErrorCode  MatCreate_SeqSBAIJ(Mat B)
1898a23d5eceSKris Buschelman {
1899a23d5eceSKris Buschelman   Mat_SeqSBAIJ   *b;
1900dfbe8321SBarry Smith   PetscErrorCode ierr;
190113f74950SBarry Smith   PetscMPIInt    size;
1902ace3abfcSBarry Smith   PetscBool      no_unroll = PETSC_FALSE,no_inode = PETSC_FALSE;
1903a23d5eceSKris Buschelman 
1904a23d5eceSKris Buschelman   PetscFunctionBegin;
19057adad957SLisandro Dalcin   ierr = MPI_Comm_size(((PetscObject)B)->comm,&size);CHKERRQ(ierr);
1906e32f2f54SBarry Smith   if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Comm must be of size 1");
1907a23d5eceSKris Buschelman 
190838f2d2fdSLisandro Dalcin   ierr    = PetscNewLog(B,Mat_SeqSBAIJ,&b);CHKERRQ(ierr);
1909a23d5eceSKris Buschelman   B->data = (void*)b;
1910a23d5eceSKris Buschelman   ierr    = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr);
1911a23d5eceSKris Buschelman   B->ops->destroy     = MatDestroy_SeqSBAIJ;
1912a23d5eceSKris Buschelman   B->ops->view        = MatView_SeqSBAIJ;
1913a23d5eceSKris Buschelman   b->row              = 0;
1914a23d5eceSKris Buschelman   b->icol             = 0;
1915a23d5eceSKris Buschelman   b->reallocs         = 0;
1916a23d5eceSKris Buschelman   b->saved_values     = 0;
19170def2e27SBarry Smith   b->inode.limit      = 5;
19180def2e27SBarry Smith   b->inode.max_limit  = 5;
1919a23d5eceSKris Buschelman 
1920a23d5eceSKris Buschelman   b->roworiented      = PETSC_TRUE;
1921a23d5eceSKris Buschelman   b->nonew            = 0;
1922a23d5eceSKris Buschelman   b->diag             = 0;
1923a23d5eceSKris Buschelman   b->solve_work       = 0;
1924a23d5eceSKris Buschelman   b->mult_work        = 0;
1925a23d5eceSKris Buschelman   B->spptr            = 0;
1926f2cbd3d5SJed Brown   B->info.nz_unneeded = (PetscReal)b->maxnz*b->bs2;
1927a9817697SBarry Smith   b->keepnonzeropattern   = PETSC_FALSE;
1928a23d5eceSKris Buschelman   b->xtoy             = 0;
1929a23d5eceSKris Buschelman   b->XtoY             = 0;
1930a23d5eceSKris Buschelman 
1931a23d5eceSKris Buschelman   b->inew             = 0;
1932a23d5eceSKris Buschelman   b->jnew             = 0;
1933a23d5eceSKris Buschelman   b->anew             = 0;
1934a23d5eceSKris Buschelman   b->a2anew           = 0;
1935a23d5eceSKris Buschelman   b->permute          = PETSC_FALSE;
1936a23d5eceSKris Buschelman 
193771dad5bbSBarry Smith   b->ignore_ltriangular = PETSC_TRUE;
1938acfcf0e5SJed Brown   ierr = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_ignore_lower_triangular",&b->ignore_ltriangular,PETSC_NULL);CHKERRQ(ierr);
1939941593c8SHong Zhang 
1940f5edf698SHong Zhang   b->getrow_utriangular = PETSC_FALSE;
1941acfcf0e5SJed Brown   ierr = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_getrow_uppertriangular",&b->getrow_utriangular,PETSC_NULL);CHKERRQ(ierr);
1942f5edf698SHong Zhang 
1943b5e56a35SBarry Smith #if defined(PETSC_HAVE_PASTIX)
1944ec1065edSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatGetFactor_pastix_C",
1945b5e56a35SBarry Smith 					   "MatGetFactor_seqsbaij_pastix",
1946b5e56a35SBarry Smith 					   MatGetFactor_seqsbaij_pastix);CHKERRQ(ierr);
1947b5e56a35SBarry Smith #endif
1948611f576cSBarry Smith #if defined(PETSC_HAVE_SPOOLES)
1949ec1065edSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatGetFactor_spooles_C",
19505c9eb25fSBarry Smith                                      "MatGetFactor_seqsbaij_spooles",
19515c9eb25fSBarry Smith                                      MatGetFactor_seqsbaij_spooles);CHKERRQ(ierr);
1952611f576cSBarry Smith #endif
1953611f576cSBarry Smith #if defined(PETSC_HAVE_MUMPS)
1954ec1065edSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatGetFactor_mumps_C",
1955bccb9932SShri Abhyankar                                      "MatGetFactor_sbaij_mumps",
1956bccb9932SShri Abhyankar                                      MatGetFactor_sbaij_mumps);CHKERRQ(ierr);
1957611f576cSBarry Smith #endif
195820db9a53SJed Brown #if defined(PETSC_HAVE_CHOLMOD)
195920db9a53SJed Brown   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatGetFactor_cholmod_C",
196020db9a53SJed Brown                                      "MatGetFactor_seqsbaij_cholmod",
196120db9a53SJed Brown                                      MatGetFactor_seqsbaij_cholmod);CHKERRQ(ierr);
196220db9a53SJed Brown #endif
1963ec1065edSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatGetFactorAvailable_petsc_C",
1964db4efbfdSBarry Smith                                      "MatGetFactorAvailable_seqsbaij_petsc",
1965db4efbfdSBarry Smith                                      MatGetFactorAvailable_seqsbaij_petsc);CHKERRQ(ierr);
1966ec1065edSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatGetFactor_petsc_C",
19675c9eb25fSBarry Smith                                      "MatGetFactor_seqsbaij_petsc",
19685c9eb25fSBarry Smith                                      MatGetFactor_seqsbaij_petsc);CHKERRQ(ierr);
19693edee7c7SSatish Balay   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatGetFactor_sbstrm_C",
19702938c13dSDahai Guo                                      "MatGetFactor_seqsbaij_sbstrm",
19712938c13dSDahai Guo                                      MatGetFactor_seqsbaij_sbstrm);CHKERRQ(ierr);
1972a23d5eceSKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatStoreValues_C",
1973a23d5eceSKris Buschelman                                      "MatStoreValues_SeqSBAIJ",
1974a23d5eceSKris Buschelman                                      MatStoreValues_SeqSBAIJ);CHKERRQ(ierr);
1975a23d5eceSKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatRetrieveValues_C",
1976a23d5eceSKris Buschelman                                      "MatRetrieveValues_SeqSBAIJ",
1977a23d5eceSKris Buschelman                                      (void*)MatRetrieveValues_SeqSBAIJ);CHKERRQ(ierr);
1978a23d5eceSKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatSeqSBAIJSetColumnIndices_C",
1979a23d5eceSKris Buschelman                                      "MatSeqSBAIJSetColumnIndices_SeqSBAIJ",
1980a23d5eceSKris Buschelman                                      MatSeqSBAIJSetColumnIndices_SeqSBAIJ);CHKERRQ(ierr);
19814e5e7fe4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_seqsbaij_seqaij_C",
19824e5e7fe4SHong Zhang                                      "MatConvert_SeqSBAIJ_SeqAIJ",
19834e5e7fe4SHong Zhang                                       MatConvert_SeqSBAIJ_SeqAIJ);CHKERRQ(ierr);
1984a0e1a404SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_seqsbaij_seqbaij_C",
1985a0e1a404SHong Zhang                                      "MatConvert_SeqSBAIJ_SeqBAIJ",
1986a0e1a404SHong Zhang                                       MatConvert_SeqSBAIJ_SeqBAIJ);CHKERRQ(ierr);
1987a23d5eceSKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatSeqSBAIJSetPreallocation_C",
1988a23d5eceSKris Buschelman                                      "MatSeqSBAIJSetPreallocation_SeqSBAIJ",
1989a23d5eceSKris Buschelman                                      MatSeqSBAIJSetPreallocation_SeqSBAIJ);CHKERRQ(ierr);
1990aa5a9175SDahai Guo   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_seqsbaij_seqsbstrm_C",
1991aa5a9175SDahai Guo                                      "MatConvert_SeqSBAIJ_SeqSBSTRM",
1992aa5a9175SDahai Guo                                       MatConvert_SeqSBAIJ_SeqSBSTRM);CHKERRQ(ierr);
199323ce1328SBarry Smith 
199423ce1328SBarry Smith   B->symmetric                  = PETSC_TRUE;
199523ce1328SBarry Smith   B->structurally_symmetric     = PETSC_TRUE;
199623ce1328SBarry Smith   B->symmetric_set              = PETSC_TRUE;
199723ce1328SBarry Smith   B->structurally_symmetric_set = PETSC_TRUE;
199817667f90SBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)B,MATSEQSBAIJ);CHKERRQ(ierr);
19990def2e27SBarry Smith 
20000def2e27SBarry Smith   ierr = PetscOptionsBegin(((PetscObject)B)->comm,((PetscObject)B)->prefix,"Options for SEQSBAIJ matrix","Mat");CHKERRQ(ierr);
2001acfcf0e5SJed Brown     ierr = PetscOptionsBool("-mat_no_unroll","Do not optimize for inodes (slower)",PETSC_NULL,no_unroll,&no_unroll,PETSC_NULL);CHKERRQ(ierr);
20020def2e27SBarry Smith     if (no_unroll) {ierr = PetscInfo(B,"Not using Inode routines due to -mat_no_unroll\n");CHKERRQ(ierr);}
2003acfcf0e5SJed Brown     ierr = PetscOptionsBool("-mat_no_inode","Do not optimize for inodes (slower)",PETSC_NULL,no_inode,&no_inode,PETSC_NULL);CHKERRQ(ierr);
20040def2e27SBarry Smith     if (no_inode) {ierr = PetscInfo(B,"Not using Inode routines due to -mat_no_inode\n");CHKERRQ(ierr);}
20050def2e27SBarry Smith     ierr = PetscOptionsInt("-mat_inode_limit","Do not use inodes larger then this value",PETSC_NULL,b->inode.limit,&b->inode.limit,PETSC_NULL);CHKERRQ(ierr);
20060def2e27SBarry Smith   ierr = PetscOptionsEnd();CHKERRQ(ierr);
2007ace3abfcSBarry Smith   b->inode.use = (PetscBool)(!(no_unroll || no_inode));
20080def2e27SBarry Smith   if (b->inode.limit > b->inode.max_limit) b->inode.limit = b->inode.max_limit;
20090def2e27SBarry Smith 
2010a23d5eceSKris Buschelman   PetscFunctionReturn(0);
2011a23d5eceSKris Buschelman }
2012a23d5eceSKris Buschelman EXTERN_C_END
2013a23d5eceSKris Buschelman 
2014a23d5eceSKris Buschelman #undef __FUNCT__
2015a23d5eceSKris Buschelman #define __FUNCT__ "MatSeqSBAIJSetPreallocation"
2016a23d5eceSKris Buschelman /*@C
2017a23d5eceSKris Buschelman    MatSeqSBAIJSetPreallocation - Creates a sparse symmetric matrix in block AIJ (block
2018a23d5eceSKris Buschelman    compressed row) format.  For good matrix assembly performance the
2019a23d5eceSKris Buschelman    user should preallocate the matrix storage by setting the parameter nz
2020a23d5eceSKris Buschelman    (or the array nnz).  By setting these parameters accurately, performance
2021a23d5eceSKris Buschelman    during matrix assembly can be increased by more than a factor of 50.
2022a23d5eceSKris Buschelman 
2023a23d5eceSKris Buschelman    Collective on Mat
2024a23d5eceSKris Buschelman 
2025a23d5eceSKris Buschelman    Input Parameters:
2026a23d5eceSKris Buschelman +  A - the symmetric matrix
2027a23d5eceSKris Buschelman .  bs - size of block
2028a23d5eceSKris Buschelman .  nz - number of block nonzeros per block row (same for all rows)
2029a23d5eceSKris Buschelman -  nnz - array containing the number of block nonzeros in the upper triangular plus
2030a23d5eceSKris Buschelman          diagonal portion of each block (possibly different for each block row) or PETSC_NULL
2031a23d5eceSKris Buschelman 
2032a23d5eceSKris Buschelman    Options Database Keys:
2033a23d5eceSKris Buschelman .   -mat_no_unroll - uses code that does not unroll the loops in the
2034a23d5eceSKris Buschelman                      block calculations (much slower)
2035db4efbfdSBarry Smith .    -mat_block_size - size of the blocks to use (only works if a negative bs is passed in
2036a23d5eceSKris Buschelman 
2037a23d5eceSKris Buschelman    Level: intermediate
2038a23d5eceSKris Buschelman 
2039a23d5eceSKris Buschelman    Notes:
2040a23d5eceSKris Buschelman    Specify the preallocated storage with either nz or nnz (not both).
2041a23d5eceSKris Buschelman    Set nz=PETSC_DEFAULT and nnz=PETSC_NULL for PETSc to control dynamic memory
20420598bfebSBarry Smith    allocation.  See the <a href="../../docs/manual.pdf#nameddest=ch_mat">Mat chapter of the users manual</a> for details.
2043a23d5eceSKris Buschelman 
2044aa95bbe8SBarry Smith    You can call MatGetInfo() to get information on how effective the preallocation was;
2045aa95bbe8SBarry Smith    for example the fields mallocs,nz_allocated,nz_used,nz_unneeded;
2046aa95bbe8SBarry Smith    You can also run with the option -info and look for messages with the string
2047aa95bbe8SBarry Smith    malloc in them to see if additional memory allocation was needed.
2048aa95bbe8SBarry Smith 
204949a6f317SBarry Smith    If the nnz parameter is given then the nz parameter is ignored
205049a6f317SBarry Smith 
205149a6f317SBarry Smith 
2052a23d5eceSKris Buschelman .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateMPISBAIJ()
2053a23d5eceSKris Buschelman @*/
20547087cfbeSBarry Smith PetscErrorCode  MatSeqSBAIJSetPreallocation(Mat B,PetscInt bs,PetscInt nz,const PetscInt nnz[])
205513f74950SBarry Smith {
20564ac538c5SBarry Smith   PetscErrorCode ierr;
2057a23d5eceSKris Buschelman 
2058a23d5eceSKris Buschelman   PetscFunctionBegin;
20594ac538c5SBarry Smith   ierr = PetscTryMethod(B,"MatSeqSBAIJSetPreallocation_C",(Mat,PetscInt,PetscInt,const PetscInt[]),(B,bs,nz,nnz));CHKERRQ(ierr);
2060a23d5eceSKris Buschelman   PetscFunctionReturn(0);
2061a23d5eceSKris Buschelman }
206249b5e25fSSatish Balay 
20634a2ae208SSatish Balay #undef __FUNCT__
20644a2ae208SSatish Balay #define __FUNCT__ "MatCreateSeqSBAIJ"
2065c464158bSHong Zhang /*@C
2066c464158bSHong Zhang    MatCreateSeqSBAIJ - Creates a sparse symmetric matrix in block AIJ (block
2067c464158bSHong Zhang    compressed row) format.  For good matrix assembly performance the
2068c464158bSHong Zhang    user should preallocate the matrix storage by setting the parameter nz
2069c464158bSHong Zhang    (or the array nnz).  By setting these parameters accurately, performance
2070c464158bSHong Zhang    during matrix assembly can be increased by more than a factor of 50.
207149b5e25fSSatish Balay 
2072c464158bSHong Zhang    Collective on MPI_Comm
2073c464158bSHong Zhang 
2074c464158bSHong Zhang    Input Parameters:
2075c464158bSHong Zhang +  comm - MPI communicator, set to PETSC_COMM_SELF
2076c464158bSHong Zhang .  bs - size of block
2077c464158bSHong Zhang .  m - number of rows, or number of columns
2078c464158bSHong Zhang .  nz - number of block nonzeros per block row (same for all rows)
2079744e8345SSatish Balay -  nnz - array containing the number of block nonzeros in the upper triangular plus
2080744e8345SSatish Balay          diagonal portion of each block (possibly different for each block row) or PETSC_NULL
2081c464158bSHong Zhang 
2082c464158bSHong Zhang    Output Parameter:
2083c464158bSHong Zhang .  A - the symmetric matrix
2084c464158bSHong Zhang 
2085c464158bSHong Zhang    Options Database Keys:
2086c464158bSHong Zhang .   -mat_no_unroll - uses code that does not unroll the loops in the
2087c464158bSHong Zhang                      block calculations (much slower)
2088c464158bSHong Zhang .    -mat_block_size - size of the blocks to use
2089c464158bSHong Zhang 
2090c464158bSHong Zhang    Level: intermediate
2091c464158bSHong Zhang 
2092175b88e8SBarry Smith    It is recommended that one use the MatCreate(), MatSetType() and/or MatSetFromOptions(),
2093ae1d86c5SBarry Smith    MatXXXXSetPreallocation() paradgm instead of this routine directly.
2094175b88e8SBarry Smith    [MatXXXXSetPreallocation() is, for example, MatSeqAIJSetPreallocation]
2095175b88e8SBarry Smith 
2096c464158bSHong Zhang    Notes:
20976d6d819aSHong Zhang    The number of rows and columns must be divisible by blocksize.
20986d6d819aSHong Zhang    This matrix type does not support complex Hermitian operation.
2099c464158bSHong Zhang 
2100c464158bSHong Zhang    Specify the preallocated storage with either nz or nnz (not both).
2101c464158bSHong Zhang    Set nz=PETSC_DEFAULT and nnz=PETSC_NULL for PETSc to control dynamic memory
21020598bfebSBarry Smith    allocation.  See the <a href="../../docs/manual.pdf#nameddest=ch_mat">Mat chapter of the users manual</a> for details.
2103c464158bSHong Zhang 
210449a6f317SBarry Smith    If the nnz parameter is given then the nz parameter is ignored
210549a6f317SBarry Smith 
2106c464158bSHong Zhang .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateMPISBAIJ()
2107c464158bSHong Zhang @*/
21087087cfbeSBarry Smith PetscErrorCode  MatCreateSeqSBAIJ(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt nz,const PetscInt nnz[],Mat *A)
2109c464158bSHong Zhang {
2110dfbe8321SBarry Smith   PetscErrorCode ierr;
2111c464158bSHong Zhang 
2112c464158bSHong Zhang   PetscFunctionBegin;
2113f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,A);CHKERRQ(ierr);
2114f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*A,m,n,m,n);CHKERRQ(ierr);
2115c464158bSHong Zhang   ierr = MatSetType(*A,MATSEQSBAIJ);CHKERRQ(ierr);
2116ab93d7beSBarry Smith   ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(*A,bs,nz,(PetscInt*)nnz);CHKERRQ(ierr);
211749b5e25fSSatish Balay   PetscFunctionReturn(0);
211849b5e25fSSatish Balay }
211949b5e25fSSatish Balay 
21204a2ae208SSatish Balay #undef __FUNCT__
21214a2ae208SSatish Balay #define __FUNCT__ "MatDuplicate_SeqSBAIJ"
2122dfbe8321SBarry Smith PetscErrorCode MatDuplicate_SeqSBAIJ(Mat A,MatDuplicateOption cpvalues,Mat *B)
212349b5e25fSSatish Balay {
212449b5e25fSSatish Balay   Mat            C;
212549b5e25fSSatish Balay   Mat_SeqSBAIJ   *c,*a = (Mat_SeqSBAIJ*)A->data;
21266849ba73SBarry Smith   PetscErrorCode ierr;
2127b40805acSSatish Balay   PetscInt       i,mbs = a->mbs,nz = a->nz,bs2 =a->bs2;
212849b5e25fSSatish Balay 
212949b5e25fSSatish Balay   PetscFunctionBegin;
2130e32f2f54SBarry Smith   if (a->i[mbs] != nz) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Corrupt matrix");
213149b5e25fSSatish Balay 
213249b5e25fSSatish Balay   *B = 0;
21337adad957SLisandro Dalcin   ierr = MatCreate(((PetscObject)A)->comm,&C);CHKERRQ(ierr);
2134d0f46423SBarry Smith   ierr = MatSetSizes(C,A->rmap->N,A->cmap->n,A->rmap->N,A->cmap->n);CHKERRQ(ierr);
21358e9a0fb8SHong Zhang   ierr = MatSetType(C,MATSEQSBAIJ);CHKERRQ(ierr);
21361d5dac46SHong Zhang   ierr = PetscMemcpy(C->ops,A->ops,sizeof(struct _MatOps));CHKERRQ(ierr);
2137692f9cbeSHong Zhang   c    = (Mat_SeqSBAIJ*)C->data;
2138692f9cbeSHong Zhang 
2139273d9f13SBarry Smith   C->preallocated       = PETSC_TRUE;
2140d5f3da31SBarry Smith   C->factortype         = A->factortype;
214149b5e25fSSatish Balay   c->row                = 0;
214249b5e25fSSatish Balay   c->icol               = 0;
214349b5e25fSSatish Balay   c->saved_values       = 0;
2144a9817697SBarry Smith   c->keepnonzeropattern = a->keepnonzeropattern;
214549b5e25fSSatish Balay   C->assembled          = PETSC_TRUE;
214649b5e25fSSatish Balay 
21471e1e43feSBarry Smith   ierr = PetscLayoutReference(A->rmap,&C->rmap);CHKERRQ(ierr);
21481e1e43feSBarry Smith   ierr = PetscLayoutReference(A->cmap,&C->cmap);CHKERRQ(ierr);
214949b5e25fSSatish Balay   c->bs2  = a->bs2;
215049b5e25fSSatish Balay   c->mbs  = a->mbs;
215149b5e25fSSatish Balay   c->nbs  = a->nbs;
215249b5e25fSSatish Balay 
2153c760cd28SBarry Smith   if  (cpvalues == MAT_SHARE_NONZERO_PATTERN) {
2154c760cd28SBarry Smith     c->imax           = a->imax;
2155c760cd28SBarry Smith     c->ilen           = a->ilen;
2156c760cd28SBarry Smith     c->free_imax_ilen = PETSC_FALSE;
2157c760cd28SBarry Smith   } else {
21588777fc3fSSatish Balay     ierr = PetscMalloc2((mbs+1),PetscInt,&c->imax,(mbs+1),PetscInt,&c->ilen);CHKERRQ(ierr);
2159c760cd28SBarry Smith     ierr = PetscLogObjectMemory(C,2*(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr);
216049b5e25fSSatish Balay     for (i=0; i<mbs; i++) {
216149b5e25fSSatish Balay       c->imax[i] = a->imax[i];
216249b5e25fSSatish Balay       c->ilen[i] = a->ilen[i];
216349b5e25fSSatish Balay     }
2164c760cd28SBarry Smith     c->free_imax_ilen = PETSC_TRUE;
2165c760cd28SBarry Smith   }
216649b5e25fSSatish Balay 
216749b5e25fSSatish Balay   /* allocate the matrix space */
21684da8f245SBarry Smith   if (cpvalues == MAT_SHARE_NONZERO_PATTERN) {
21694da8f245SBarry Smith     ierr = PetscMalloc(bs2*nz*sizeof(MatScalar),&c->a);CHKERRQ(ierr);
21704da8f245SBarry Smith     ierr = PetscLogObjectMemory(C,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr);
21714da8f245SBarry Smith     c->singlemalloc = PETSC_FALSE;
21724da8f245SBarry Smith     c->free_ij      = PETSC_FALSE;
21734da8f245SBarry Smith     c->parent       = A;
21744da8f245SBarry Smith     ierr            = PetscObjectReference((PetscObject)A);CHKERRQ(ierr);
21754da8f245SBarry Smith     ierr            = MatSetOption(A,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
21764da8f245SBarry Smith     ierr            = MatSetOption(C,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
21774da8f245SBarry Smith   } else {
2178b40805acSSatish Balay     ierr = PetscMalloc3(bs2*nz,MatScalar,&c->a,nz,PetscInt,&c->j,mbs+1,PetscInt,&c->i);CHKERRQ(ierr);
217913f74950SBarry Smith     ierr = PetscMemcpy(c->i,a->i,(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr);
2180b40805acSSatish Balay     ierr = PetscLogObjectMemory(C,(mbs+1)*sizeof(PetscInt) + nz*(bs2*sizeof(MatScalar) + sizeof(PetscInt)));CHKERRQ(ierr);
21814da8f245SBarry Smith     c->singlemalloc = PETSC_TRUE;
21824da8f245SBarry Smith     c->free_ij      = PETSC_TRUE;
21834da8f245SBarry Smith   }
218449b5e25fSSatish Balay   if (mbs > 0) {
21854da8f245SBarry Smith     if (cpvalues != MAT_SHARE_NONZERO_PATTERN) {
218613f74950SBarry Smith       ierr = PetscMemcpy(c->j,a->j,nz*sizeof(PetscInt));CHKERRQ(ierr);
21874da8f245SBarry Smith     }
218849b5e25fSSatish Balay     if (cpvalues == MAT_COPY_VALUES) {
218949b5e25fSSatish Balay       ierr = PetscMemcpy(c->a,a->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr);
219049b5e25fSSatish Balay     } else {
219149b5e25fSSatish Balay       ierr = PetscMemzero(c->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr);
219249b5e25fSSatish Balay     }
2193a1c3900fSBarry Smith     if (a->jshort) {
21944da8f245SBarry Smith       if (cpvalues == MAT_SHARE_NONZERO_PATTERN) {
21954da8f245SBarry Smith         c->jshort      = a->jshort;
21964da8f245SBarry Smith         c->free_jshort = PETSC_FALSE;
21974da8f245SBarry Smith       } else {
2198a1c3900fSBarry Smith         ierr = PetscMalloc(nz*sizeof(unsigned short),&c->jshort);CHKERRQ(ierr);
2199c760cd28SBarry Smith         ierr = PetscLogObjectMemory(C,nz*sizeof(unsigned short));CHKERRQ(ierr);
2200a1c3900fSBarry Smith         ierr = PetscMemcpy(c->jshort,a->jshort,nz*sizeof(unsigned short));CHKERRQ(ierr);
22014da8f245SBarry Smith         c->free_jshort = PETSC_TRUE;
22024da8f245SBarry Smith       }
2203a1c3900fSBarry Smith     }
220449b5e25fSSatish Balay   }
220549b5e25fSSatish Balay 
220649b5e25fSSatish Balay   c->roworiented = a->roworiented;
220749b5e25fSSatish Balay   c->nonew       = a->nonew;
220849b5e25fSSatish Balay 
220949b5e25fSSatish Balay   if (a->diag) {
2210c760cd28SBarry Smith     if (cpvalues == MAT_SHARE_NONZERO_PATTERN) {
2211c760cd28SBarry Smith       c->diag      = a->diag;
2212c760cd28SBarry Smith       c->free_diag = PETSC_FALSE;
2213c760cd28SBarry Smith     } else {
22142ed38d0bSJed Brown       ierr = PetscMalloc(mbs*sizeof(PetscInt),&c->diag);CHKERRQ(ierr);
22152ed38d0bSJed Brown       ierr = PetscLogObjectMemory(C,mbs*sizeof(PetscInt));CHKERRQ(ierr);
221649b5e25fSSatish Balay       for (i=0; i<mbs; i++) {
221749b5e25fSSatish Balay 	c->diag[i] = a->diag[i];
221849b5e25fSSatish Balay       }
2219c760cd28SBarry Smith       c->free_diag = PETSC_TRUE;
2220c760cd28SBarry Smith     }
222149b5e25fSSatish Balay   } else c->diag  = 0;
22226c6c5352SBarry Smith   c->nz           = a->nz;
2223f2cbd3d5SJed Brown   c->maxnz        = a->nz; /* Since we allocate exactly the right amount */
222449b5e25fSSatish Balay   c->solve_work   = 0;
222549b5e25fSSatish Balay   c->mult_work    = 0;
2226e6b907acSBarry Smith   c->free_a       = PETSC_TRUE;
222749b5e25fSSatish Balay   *B = C;
22287adad957SLisandro Dalcin   ierr = PetscFListDuplicate(((PetscObject)A)->qlist,&((PetscObject)C)->qlist);CHKERRQ(ierr);
222949b5e25fSSatish Balay   PetscFunctionReturn(0);
223049b5e25fSSatish Balay }
223149b5e25fSSatish Balay 
22324a2ae208SSatish Balay #undef __FUNCT__
22335bba2384SShri Abhyankar #define __FUNCT__ "MatLoad_SeqSBAIJ"
2234112444f4SShri Abhyankar PetscErrorCode MatLoad_SeqSBAIJ(Mat newmat,PetscViewer viewer)
22352f480046SShri Abhyankar {
22362f480046SShri Abhyankar   Mat_SeqSBAIJ   *a;
22372f480046SShri Abhyankar   PetscErrorCode ierr;
22382f480046SShri Abhyankar   int            fd;
22392f480046SShri Abhyankar   PetscMPIInt    size;
22402f480046SShri Abhyankar   PetscInt       i,nz,header[4],*rowlengths=0,M,N,bs=1;
22412f480046SShri Abhyankar   PetscInt       *mask,mbs,*jj,j,rowcount,nzcount,k,*s_browlengths,maskcount;
22422f480046SShri Abhyankar   PetscInt       kmax,jcount,block,idx,point,nzcountb,extra_rows,rows,cols;
22432f480046SShri Abhyankar   PetscInt       *masked,nmask,tmp,bs2,ishift;
22442f480046SShri Abhyankar   PetscScalar    *aa;
22452f480046SShri Abhyankar   MPI_Comm       comm = ((PetscObject)viewer)->comm;
22462f480046SShri Abhyankar 
22472f480046SShri Abhyankar   PetscFunctionBegin;
22482f480046SShri Abhyankar   ierr = PetscOptionsGetInt(PETSC_NULL,"-matload_block_size",&bs,PETSC_NULL);CHKERRQ(ierr);
22492f480046SShri Abhyankar   bs2  = bs*bs;
22502f480046SShri Abhyankar 
22512f480046SShri Abhyankar   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
22522f480046SShri Abhyankar   if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"view must have one processor");
22532f480046SShri Abhyankar   ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr);
22542f480046SShri Abhyankar   ierr = PetscBinaryRead(fd,header,4,PETSC_INT);CHKERRQ(ierr);
22552f480046SShri Abhyankar   if (header[0] != MAT_FILE_CLASSID) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"not Mat object");
22562f480046SShri Abhyankar   M = header[1]; N = header[2]; nz = header[3];
22572f480046SShri Abhyankar 
22582f480046SShri Abhyankar   if (header[3] < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Matrix stored in special format, cannot load as SeqSBAIJ");
22592f480046SShri Abhyankar 
22602f480046SShri Abhyankar   if (M != N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Can only do square matrices");
22612f480046SShri Abhyankar 
22622f480046SShri Abhyankar   /*
22632f480046SShri Abhyankar      This code adds extra rows to make sure the number of rows is
22642f480046SShri Abhyankar     divisible by the blocksize
22652f480046SShri Abhyankar   */
22662f480046SShri Abhyankar   mbs        = M/bs;
22672f480046SShri Abhyankar   extra_rows = bs - M + bs*(mbs);
22682f480046SShri Abhyankar   if (extra_rows == bs) extra_rows = 0;
22692f480046SShri Abhyankar   else                  mbs++;
22702f480046SShri Abhyankar   if (extra_rows) {
22712f480046SShri Abhyankar     ierr = PetscInfo(viewer,"Padding loaded matrix to match blocksize\n");CHKERRQ(ierr);
22722f480046SShri Abhyankar   }
22732f480046SShri Abhyankar 
22742f480046SShri Abhyankar   /* Set global sizes if not already set */
22752f480046SShri Abhyankar   if (newmat->rmap->n < 0 && newmat->rmap->N < 0 && newmat->cmap->n < 0 && newmat->cmap->N < 0) {
22762f480046SShri Abhyankar     ierr = MatSetSizes(newmat,PETSC_DECIDE,PETSC_DECIDE,M+extra_rows,N+extra_rows);CHKERRQ(ierr);
22772f480046SShri Abhyankar   } else { /* Check if the matrix global sizes are correct */
22782f480046SShri Abhyankar     ierr = MatGetSize(newmat,&rows,&cols);CHKERRQ(ierr);
22792f480046SShri Abhyankar     if (M != rows ||  N != cols) SETERRQ4(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Matrix in file of different length (%d, %d) than the input matrix (%d, %d)",M,N,rows,cols);
22802f480046SShri Abhyankar   }
22812f480046SShri Abhyankar 
22822f480046SShri Abhyankar   /* read in row lengths */
22832f480046SShri Abhyankar   ierr = PetscMalloc((M+extra_rows)*sizeof(PetscInt),&rowlengths);CHKERRQ(ierr);
22842f480046SShri Abhyankar   ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr);
22852f480046SShri Abhyankar   for (i=0; i<extra_rows; i++) rowlengths[M+i] = 1;
22862f480046SShri Abhyankar 
22872f480046SShri Abhyankar   /* read in column indices */
22882f480046SShri Abhyankar   ierr = PetscMalloc((nz+extra_rows)*sizeof(PetscInt),&jj);CHKERRQ(ierr);
22892f480046SShri Abhyankar   ierr = PetscBinaryRead(fd,jj,nz,PETSC_INT);CHKERRQ(ierr);
22902f480046SShri Abhyankar   for (i=0; i<extra_rows; i++) jj[nz+i] = M+i;
22912f480046SShri Abhyankar 
22922f480046SShri Abhyankar   /* loop over row lengths determining block row lengths */
22932f480046SShri Abhyankar   ierr     = PetscMalloc(mbs*sizeof(PetscInt),&s_browlengths);CHKERRQ(ierr);
22942f480046SShri Abhyankar   ierr     = PetscMemzero(s_browlengths,mbs*sizeof(PetscInt));CHKERRQ(ierr);
22952f480046SShri Abhyankar   ierr     = PetscMalloc2(mbs,PetscInt,&mask,mbs,PetscInt,&masked);CHKERRQ(ierr);
22962f480046SShri Abhyankar   ierr     = PetscMemzero(mask,mbs*sizeof(PetscInt));CHKERRQ(ierr);
22972f480046SShri Abhyankar   rowcount = 0;
22982f480046SShri Abhyankar   nzcount  = 0;
22992f480046SShri Abhyankar   for (i=0; i<mbs; i++) {
23002f480046SShri Abhyankar     nmask = 0;
23012f480046SShri Abhyankar     for (j=0; j<bs; j++) {
23022f480046SShri Abhyankar       kmax = rowlengths[rowcount];
23032f480046SShri Abhyankar       for (k=0; k<kmax; k++) {
23042f480046SShri Abhyankar         tmp = jj[nzcount++]/bs;   /* block col. index */
23052f480046SShri Abhyankar         if (!mask[tmp] && tmp >= i) {masked[nmask++] = tmp; mask[tmp] = 1;}
23062f480046SShri Abhyankar       }
23072f480046SShri Abhyankar       rowcount++;
23082f480046SShri Abhyankar     }
23092f480046SShri Abhyankar     s_browlengths[i] += nmask;
23102f480046SShri Abhyankar 
23112f480046SShri Abhyankar     /* zero out the mask elements we set */
23122f480046SShri Abhyankar     for (j=0; j<nmask; j++) mask[masked[j]] = 0;
23132f480046SShri Abhyankar   }
23142f480046SShri Abhyankar 
23152f480046SShri Abhyankar   /* Do preallocation */
23162f480046SShri Abhyankar   ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(newmat,bs,0,s_browlengths);CHKERRQ(ierr);
23172f480046SShri Abhyankar   a = (Mat_SeqSBAIJ*)newmat->data;
23182f480046SShri Abhyankar 
23192f480046SShri Abhyankar   /* set matrix "i" values */
23202f480046SShri Abhyankar   a->i[0] = 0;
23212f480046SShri Abhyankar   for (i=1; i<= mbs; i++) {
23222f480046SShri Abhyankar     a->i[i]      = a->i[i-1] + s_browlengths[i-1];
23232f480046SShri Abhyankar     a->ilen[i-1] = s_browlengths[i-1];
23242f480046SShri Abhyankar   }
23252f480046SShri Abhyankar   a->nz = a->i[mbs];
23262f480046SShri Abhyankar 
23272f480046SShri Abhyankar   /* read in nonzero values */
23282f480046SShri Abhyankar   ierr = PetscMalloc((nz+extra_rows)*sizeof(PetscScalar),&aa);CHKERRQ(ierr);
23292f480046SShri Abhyankar   ierr = PetscBinaryRead(fd,aa,nz,PETSC_SCALAR);CHKERRQ(ierr);
23302f480046SShri Abhyankar   for (i=0; i<extra_rows; i++) aa[nz+i] = 1.0;
23312f480046SShri Abhyankar 
23322f480046SShri Abhyankar   /* set "a" and "j" values into matrix */
23332f480046SShri Abhyankar   nzcount = 0; jcount = 0;
23342f480046SShri Abhyankar   for (i=0; i<mbs; i++) {
23352f480046SShri Abhyankar     nzcountb = nzcount;
23362f480046SShri Abhyankar     nmask    = 0;
23372f480046SShri Abhyankar     for (j=0; j<bs; j++) {
23382f480046SShri Abhyankar       kmax = rowlengths[i*bs+j];
23392f480046SShri Abhyankar       for (k=0; k<kmax; k++) {
23402f480046SShri Abhyankar         tmp = jj[nzcount++]/bs; /* block col. index */
23412f480046SShri Abhyankar         if (!mask[tmp] && tmp >= i) { masked[nmask++] = tmp; mask[tmp] = 1;}
23422f480046SShri Abhyankar       }
23432f480046SShri Abhyankar     }
23442f480046SShri Abhyankar     /* sort the masked values */
23452f480046SShri Abhyankar     ierr = PetscSortInt(nmask,masked);CHKERRQ(ierr);
23462f480046SShri Abhyankar 
23472f480046SShri Abhyankar     /* set "j" values into matrix */
23482f480046SShri Abhyankar     maskcount = 1;
23492f480046SShri Abhyankar     for (j=0; j<nmask; j++) {
23502f480046SShri Abhyankar       a->j[jcount++]  = masked[j];
23512f480046SShri Abhyankar       mask[masked[j]] = maskcount++;
23522f480046SShri Abhyankar     }
23532f480046SShri Abhyankar 
23542f480046SShri Abhyankar     /* set "a" values into matrix */
23552f480046SShri Abhyankar     ishift = bs2*a->i[i];
23562f480046SShri Abhyankar     for (j=0; j<bs; j++) {
23572f480046SShri Abhyankar       kmax = rowlengths[i*bs+j];
23582f480046SShri Abhyankar       for (k=0; k<kmax; k++) {
23592f480046SShri Abhyankar         tmp       = jj[nzcountb]/bs ; /* block col. index */
23602f480046SShri Abhyankar         if (tmp >= i){
23612f480046SShri Abhyankar           block     = mask[tmp] - 1;
23622f480046SShri Abhyankar           point     = jj[nzcountb] - bs*tmp;
23632f480046SShri Abhyankar           idx       = ishift + bs2*block + j + bs*point;
23642f480046SShri Abhyankar           a->a[idx] = aa[nzcountb];
23652f480046SShri Abhyankar         }
23662f480046SShri Abhyankar         nzcountb++;
23672f480046SShri Abhyankar       }
23682f480046SShri Abhyankar     }
23692f480046SShri Abhyankar     /* zero out the mask elements we set */
23702f480046SShri Abhyankar     for (j=0; j<nmask; j++) mask[masked[j]] = 0;
23712f480046SShri Abhyankar   }
23722f480046SShri Abhyankar   if (jcount != a->nz) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Bad binary matrix");
23732f480046SShri Abhyankar 
23742f480046SShri Abhyankar   ierr = PetscFree(rowlengths);CHKERRQ(ierr);
23752f480046SShri Abhyankar   ierr = PetscFree(s_browlengths);CHKERRQ(ierr);
23762f480046SShri Abhyankar   ierr = PetscFree(aa);CHKERRQ(ierr);
23772f480046SShri Abhyankar   ierr = PetscFree(jj);CHKERRQ(ierr);
23782f480046SShri Abhyankar   ierr = PetscFree2(mask,masked);CHKERRQ(ierr);
23792f480046SShri Abhyankar 
23802f480046SShri Abhyankar   ierr = MatAssemblyBegin(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
23812f480046SShri Abhyankar   ierr = MatAssemblyEnd(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
23822f480046SShri Abhyankar   ierr = MatView_Private(newmat);CHKERRQ(ierr);
23832f480046SShri Abhyankar   PetscFunctionReturn(0);
23842f480046SShri Abhyankar }
23852f480046SShri Abhyankar 
23862f480046SShri Abhyankar #undef __FUNCT__
2387c75a6043SHong Zhang #define __FUNCT__ "MatCreateSeqSBAIJWithArrays"
2388c75a6043SHong Zhang /*@
2389c75a6043SHong Zhang      MatCreateSeqSBAIJWithArrays - Creates an sequential SBAIJ matrix using matrix elements
2390c75a6043SHong Zhang               (upper triangular entries in CSR format) provided by the user.
2391c75a6043SHong Zhang 
2392c75a6043SHong Zhang      Collective on MPI_Comm
2393c75a6043SHong Zhang 
2394c75a6043SHong Zhang    Input Parameters:
2395c75a6043SHong Zhang +  comm - must be an MPI communicator of size 1
2396c75a6043SHong Zhang .  bs - size of block
2397c75a6043SHong Zhang .  m - number of rows
2398c75a6043SHong Zhang .  n - number of columns
2399c75a6043SHong Zhang .  i - row indices
2400c75a6043SHong Zhang .  j - column indices
2401c75a6043SHong Zhang -  a - matrix values
2402c75a6043SHong Zhang 
2403c75a6043SHong Zhang    Output Parameter:
2404c75a6043SHong Zhang .  mat - the matrix
2405c75a6043SHong Zhang 
2406dfb205c3SBarry Smith    Level: advanced
2407c75a6043SHong Zhang 
2408c75a6043SHong Zhang    Notes:
2409c75a6043SHong Zhang        The i, j, and a arrays are not copied by this routine, the user must free these arrays
2410c75a6043SHong Zhang     once the matrix is destroyed
2411c75a6043SHong Zhang 
2412c75a6043SHong Zhang        You cannot set new nonzero locations into this matrix, that will generate an error.
2413c75a6043SHong Zhang 
2414c75a6043SHong Zhang        The i and j indices are 0 based
2415c75a6043SHong Zhang 
2416dfb205c3SBarry Smith        When block size is greater than 1 the matrix values must be stored using the SBAIJ storage format (see the SBAIJ code to determine this). For block size of 1
2417dfb205c3SBarry Smith        it is the regular CSR format excluding the lower triangular elements.
2418dfb205c3SBarry Smith 
2419c75a6043SHong Zhang .seealso: MatCreate(), MatCreateMPISBAIJ(), MatCreateSeqSBAIJ()
2420c75a6043SHong Zhang 
2421c75a6043SHong Zhang @*/
24227087cfbeSBarry Smith PetscErrorCode  MatCreateSeqSBAIJWithArrays(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt* i,PetscInt*j,PetscScalar *a,Mat *mat)
2423c75a6043SHong Zhang {
2424c75a6043SHong Zhang   PetscErrorCode ierr;
2425c75a6043SHong Zhang   PetscInt       ii;
2426c75a6043SHong Zhang   Mat_SeqSBAIJ   *sbaij;
2427c75a6043SHong Zhang 
2428c75a6043SHong Zhang   PetscFunctionBegin;
2429e32f2f54SBarry Smith   if (bs != 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"block size %D > 1 is not supported yet",bs);
2430e32f2f54SBarry Smith   if (i[0]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"i (row indices) must start with 0");
2431c75a6043SHong Zhang 
2432c75a6043SHong Zhang   ierr = MatCreate(comm,mat);CHKERRQ(ierr);
2433c75a6043SHong Zhang   ierr = MatSetSizes(*mat,m,n,m,n);CHKERRQ(ierr);
2434c75a6043SHong Zhang   ierr = MatSetType(*mat,MATSEQSBAIJ);CHKERRQ(ierr);
2435c75a6043SHong Zhang   ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(*mat,bs,MAT_SKIP_ALLOCATION,0);CHKERRQ(ierr);
2436c75a6043SHong Zhang   sbaij = (Mat_SeqSBAIJ*)(*mat)->data;
2437c75a6043SHong Zhang   ierr = PetscMalloc2(m,PetscInt,&sbaij->imax,m,PetscInt,&sbaij->ilen);CHKERRQ(ierr);
2438c760cd28SBarry Smith   ierr = PetscLogObjectMemory(*mat,2*m*sizeof(PetscInt));CHKERRQ(ierr);
2439c75a6043SHong Zhang 
2440c75a6043SHong Zhang   sbaij->i = i;
2441c75a6043SHong Zhang   sbaij->j = j;
2442c75a6043SHong Zhang   sbaij->a = a;
2443c75a6043SHong Zhang   sbaij->singlemalloc = PETSC_FALSE;
2444c75a6043SHong Zhang   sbaij->nonew        = -1;             /*this indicates that inserting a new value in the matrix that generates a new nonzero is an error*/
2445e6b907acSBarry Smith   sbaij->free_a       = PETSC_FALSE;
2446e6b907acSBarry Smith   sbaij->free_ij      = PETSC_FALSE;
2447c75a6043SHong Zhang 
2448c75a6043SHong Zhang   for (ii=0; ii<m; ii++) {
2449c75a6043SHong Zhang     sbaij->ilen[ii] = sbaij->imax[ii] = i[ii+1] - i[ii];
2450c75a6043SHong Zhang #if defined(PETSC_USE_DEBUG)
2451e32f2f54SBarry Smith     if (i[ii+1] - i[ii] < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative row length in i (row indices) row = %d length = %d",ii,i[ii+1] - i[ii]);
2452c75a6043SHong Zhang #endif
2453c75a6043SHong Zhang   }
2454c75a6043SHong Zhang #if defined(PETSC_USE_DEBUG)
2455c75a6043SHong Zhang   for (ii=0; ii<sbaij->i[m]; ii++) {
2456e32f2f54SBarry Smith     if (j[ii] < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column index at location = %d index = %d",ii,j[ii]);
2457e32f2f54SBarry Smith     if (j[ii] > n - 1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column index to large at location = %d index = %d",ii,j[ii]);
2458c75a6043SHong Zhang   }
2459c75a6043SHong Zhang #endif
2460c75a6043SHong Zhang 
2461c75a6043SHong Zhang   ierr = MatAssemblyBegin(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2462c75a6043SHong Zhang   ierr = MatAssemblyEnd(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2463c75a6043SHong Zhang   PetscFunctionReturn(0);
2464c75a6043SHong Zhang }
2465d06b337dSHong Zhang 
2466d06b337dSHong Zhang 
2467d06b337dSHong Zhang 
246849b5e25fSSatish Balay 
246949b5e25fSSatish Balay 
2470