xref: /petsc/src/mat/impls/sbaij/seq/sbaij.c (revision c3c607cc91279636665fbf935e3f85aadf0f476e)
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);
156214f412SHong Zhang #if defined(PETSC_HAVE_ELEMENTAL)
16cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatConvert_SeqSBAIJ_Elemental(Mat,MatType,MatReuse,Mat*);
176214f412SHong Zhang #endif
18b5b17502SBarry Smith 
1949b5e25fSSatish Balay /*
2049b5e25fSSatish Balay      Checks for missing diagonals
2149b5e25fSSatish Balay */
22ace3abfcSBarry Smith PetscErrorCode MatMissingDiagonal_SeqSBAIJ(Mat A,PetscBool  *missing,PetscInt *dd)
2349b5e25fSSatish Balay {
24045c9aa0SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
256849ba73SBarry Smith   PetscErrorCode ierr;
267734d3b5SMatthew G. Knepley   PetscInt       *diag,*ii = a->i,i;
2749b5e25fSSatish Balay 
2849b5e25fSSatish Balay   PetscFunctionBegin;
29045c9aa0SHong Zhang   ierr     = MatMarkDiagonal_SeqSBAIJ(A);CHKERRQ(ierr);
302af78befSBarry Smith   *missing = PETSC_FALSE;
317734d3b5SMatthew G. Knepley   if (A->rmap->n > 0 && !ii) {
32358d2f5dSShri Abhyankar     *missing = PETSC_TRUE;
33358d2f5dSShri Abhyankar     if (dd) *dd = 0;
34955c1f14SBarry Smith     ierr = PetscInfo(A,"Matrix has no entries therefore is missing diagonal\n");CHKERRQ(ierr);
35358d2f5dSShri Abhyankar   } else {
36358d2f5dSShri Abhyankar     diag = a->diag;
3749b5e25fSSatish Balay     for (i=0; i<a->mbs; i++) {
387734d3b5SMatthew G. Knepley       if (diag[i] >= ii[i+1]) {
392af78befSBarry Smith         *missing = PETSC_TRUE;
402af78befSBarry Smith         if (dd) *dd = i;
412af78befSBarry Smith         break;
422af78befSBarry Smith       }
4349b5e25fSSatish Balay     }
44358d2f5dSShri Abhyankar   }
4549b5e25fSSatish Balay   PetscFunctionReturn(0);
4649b5e25fSSatish Balay }
4749b5e25fSSatish Balay 
48dfbe8321SBarry Smith PetscErrorCode MatMarkDiagonal_SeqSBAIJ(Mat A)
4949b5e25fSSatish Balay {
50045c9aa0SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
516849ba73SBarry Smith   PetscErrorCode ierr;
5248dd3d27SHong Zhang   PetscInt       i,j;
5349b5e25fSSatish Balay 
5449b5e25fSSatish Balay   PetscFunctionBegin;
5509f38230SBarry Smith   if (!a->diag) {
56785e854fSJed Brown     ierr         = PetscMalloc1(a->mbs,&a->diag);CHKERRQ(ierr);
573bb1ff40SBarry Smith     ierr         = PetscLogObjectMemory((PetscObject)A,a->mbs*sizeof(PetscInt));CHKERRQ(ierr);
58c760cd28SBarry Smith     a->free_diag = PETSC_TRUE;
5909f38230SBarry Smith   }
6048dd3d27SHong Zhang   for (i=0; i<a->mbs; i++) {
6148dd3d27SHong Zhang     a->diag[i] = a->i[i+1];
6248dd3d27SHong Zhang     for (j=a->i[i]; j<a->i[i+1]; j++) {
6348dd3d27SHong Zhang       if (a->j[j] == i) {
6448dd3d27SHong Zhang         a->diag[i] = j;
6548dd3d27SHong Zhang         break;
6648dd3d27SHong Zhang       }
6748dd3d27SHong Zhang     }
6848dd3d27SHong Zhang   }
6949b5e25fSSatish Balay   PetscFunctionReturn(0);
7049b5e25fSSatish Balay }
7149b5e25fSSatish Balay 
721a83f524SJed Brown static PetscErrorCode MatGetRowIJ_SeqSBAIJ(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *nn,const PetscInt *inia[],const PetscInt *inja[],PetscBool  *done)
7349b5e25fSSatish Balay {
74a6ece127SHong Zhang   Mat_SeqSBAIJ    *a = (Mat_SeqSBAIJ*)A->data;
758f7157efSSatish Balay   PetscErrorCode ierr;
762462f5fdSStefano Zampini   PetscInt       i,j,n = a->mbs,nz = a->i[n],*tia,*tja,bs = A->rmap->bs,k,l,cnt;
772462f5fdSStefano Zampini   PetscInt       **ia = (PetscInt**)inia,**ja = (PetscInt**)inja;
7849b5e25fSSatish Balay 
7949b5e25fSSatish Balay   PetscFunctionBegin;
80d3e5a4abSHong Zhang   *nn = n;
81a1373b80SHong Zhang   if (!ia) PetscFunctionReturn(0);
822462f5fdSStefano Zampini   if (symmetric) {
832462f5fdSStefano Zampini     ierr = MatToSymmetricIJ_SeqAIJ(n,a->i,a->j,PETSC_FALSE,0,0,&tia,&tja);CHKERRQ(ierr);
842462f5fdSStefano Zampini     nz   = tia[n];
852462f5fdSStefano Zampini   } else {
862462f5fdSStefano Zampini     tia = a->i; tja = a->j;
872462f5fdSStefano Zampini   }
882462f5fdSStefano Zampini 
892462f5fdSStefano Zampini   if (!blockcompressed && bs > 1) {
902462f5fdSStefano Zampini     (*nn) *= bs;
918f7157efSSatish Balay     /* malloc & create the natural set of indices */
922462f5fdSStefano Zampini     ierr = PetscMalloc1((n+1)*bs,ia);CHKERRQ(ierr);
932462f5fdSStefano Zampini     if (n) {
942462f5fdSStefano Zampini       (*ia)[0] = oshift;
952462f5fdSStefano Zampini       for (j=1; j<bs; j++) {
962462f5fdSStefano Zampini         (*ia)[j] = (tia[1]-tia[0])*bs+(*ia)[j-1];
972462f5fdSStefano Zampini       }
982462f5fdSStefano Zampini     }
992462f5fdSStefano Zampini 
1002462f5fdSStefano Zampini     for (i=1; i<n; i++) {
1012462f5fdSStefano Zampini       (*ia)[i*bs] = (tia[i]-tia[i-1])*bs + (*ia)[i*bs-1];
1022462f5fdSStefano Zampini       for (j=1; j<bs; j++) {
1032462f5fdSStefano Zampini         (*ia)[i*bs+j] = (tia[i+1]-tia[i])*bs + (*ia)[i*bs+j-1];
1042462f5fdSStefano Zampini       }
1052462f5fdSStefano Zampini     }
1062462f5fdSStefano Zampini     if (n) {
1072462f5fdSStefano Zampini       (*ia)[n*bs] = (tia[n]-tia[n-1])*bs + (*ia)[n*bs-1];
1082462f5fdSStefano Zampini     }
1092462f5fdSStefano Zampini 
1102462f5fdSStefano Zampini     if (inja) {
1112462f5fdSStefano Zampini       ierr = PetscMalloc1(nz*bs*bs,ja);CHKERRQ(ierr);
1122462f5fdSStefano Zampini       cnt = 0;
1132462f5fdSStefano Zampini       for (i=0; i<n; i++) {
1148f7157efSSatish Balay         for (j=0; j<bs; j++) {
1152462f5fdSStefano Zampini           for (k=tia[i]; k<tia[i+1]; k++) {
1162462f5fdSStefano Zampini             for (l=0; l<bs; l++) {
1172462f5fdSStefano Zampini               (*ja)[cnt++] = bs*tja[k] + l;
1188f7157efSSatish Balay             }
1198f7157efSSatish Balay           }
1208f7157efSSatish Balay         }
1218f7157efSSatish Balay       }
1228f7157efSSatish Balay     }
1232462f5fdSStefano Zampini 
1242462f5fdSStefano Zampini     if (symmetric) { /* deallocate memory allocated in MatToSymmetricIJ_SeqAIJ() */
1252462f5fdSStefano Zampini       ierr = PetscFree(tia);CHKERRQ(ierr);
1262462f5fdSStefano Zampini       ierr = PetscFree(tja);CHKERRQ(ierr);
1272462f5fdSStefano Zampini     }
1282462f5fdSStefano Zampini   } else if (oshift == 1) {
1292462f5fdSStefano Zampini     if (symmetric) {
1302462f5fdSStefano Zampini       nz = tia[A->rmap->n/bs];
1312462f5fdSStefano Zampini       /*  add 1 to i and j indices */
1322462f5fdSStefano Zampini       for (i=0; i<A->rmap->n/bs+1; i++) tia[i] = tia[i] + 1;
1332462f5fdSStefano Zampini       *ia = tia;
1342462f5fdSStefano Zampini       if (ja) {
1352462f5fdSStefano Zampini         for (i=0; i<nz; i++) tja[i] = tja[i] + 1;
1362462f5fdSStefano Zampini         *ja = tja;
1372462f5fdSStefano Zampini       }
1382462f5fdSStefano Zampini     } else {
1392462f5fdSStefano Zampini       nz = a->i[A->rmap->n/bs];
1402462f5fdSStefano Zampini       /* malloc space and  add 1 to i and j indices */
1412462f5fdSStefano Zampini       ierr = PetscMalloc1(A->rmap->n/bs+1,ia);CHKERRQ(ierr);
1422462f5fdSStefano Zampini       for (i=0; i<A->rmap->n/bs+1; i++) (*ia)[i] = a->i[i] + 1;
1432462f5fdSStefano Zampini       if (ja) {
1442462f5fdSStefano Zampini         ierr = PetscMalloc1(nz,ja);CHKERRQ(ierr);
1452462f5fdSStefano Zampini         for (i=0; i<nz; i++) (*ja)[i] = a->j[i] + 1;
1462462f5fdSStefano Zampini       }
1472462f5fdSStefano Zampini     }
1482462f5fdSStefano Zampini   } else {
1492462f5fdSStefano Zampini     *ia = tia;
1502462f5fdSStefano Zampini     if (ja) *ja = tja;
151a6ece127SHong Zhang   }
15249b5e25fSSatish Balay   PetscFunctionReturn(0);
15349b5e25fSSatish Balay }
15449b5e25fSSatish Balay 
1551a83f524SJed Brown static PetscErrorCode MatRestoreRowIJ_SeqSBAIJ(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *nn,const PetscInt *ia[],const PetscInt *ja[],PetscBool  *done)
15649b5e25fSSatish Balay {
1578f7157efSSatish Balay   PetscErrorCode ierr;
158a6ece127SHong Zhang 
15949b5e25fSSatish Balay   PetscFunctionBegin;
16049b5e25fSSatish Balay   if (!ia) PetscFunctionReturn(0);
1612462f5fdSStefano Zampini   if ((!blockcompressed && A->rmap->bs > 1) || (symmetric || oshift == 1)) {
1622462f5fdSStefano Zampini     ierr = PetscFree(*ia);CHKERRQ(ierr);
1632462f5fdSStefano Zampini     if (ja) {ierr = PetscFree(*ja);CHKERRQ(ierr);}
164a6ece127SHong Zhang   }
165a6ece127SHong Zhang   PetscFunctionReturn(0);
16649b5e25fSSatish Balay }
16749b5e25fSSatish Balay 
168dfbe8321SBarry Smith PetscErrorCode MatDestroy_SeqSBAIJ(Mat A)
16949b5e25fSSatish Balay {
17049b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
171dfbe8321SBarry Smith   PetscErrorCode ierr;
17249b5e25fSSatish Balay 
17349b5e25fSSatish Balay   PetscFunctionBegin;
174a9f03627SSatish Balay #if defined(PETSC_USE_LOG)
175d0f46423SBarry Smith   PetscLogObjectState((PetscObject)A,"Rows=%D, NZ=%D",A->rmap->N,a->nz);
176a9f03627SSatish Balay #endif
177e6b907acSBarry Smith   ierr = MatSeqXAIJFreeAIJ(A,&a->a,&a->j,&a->i);CHKERRQ(ierr);
1787f53bb6cSHong Zhang   if (a->free_diag) {ierr = PetscFree(a->diag);CHKERRQ(ierr);}
1796bf464f9SBarry Smith   ierr = ISDestroy(&a->row);CHKERRQ(ierr);
1806bf464f9SBarry Smith   ierr = ISDestroy(&a->col);CHKERRQ(ierr);
1816bf464f9SBarry Smith   ierr = ISDestroy(&a->icol);CHKERRQ(ierr);
182c31cb41cSBarry Smith   ierr = PetscFree(a->idiag);CHKERRQ(ierr);
183c31cb41cSBarry Smith   ierr = PetscFree(a->inode.size);CHKERRQ(ierr);
184c760cd28SBarry Smith   if (a->free_imax_ilen) {ierr = PetscFree2(a->imax,a->ilen);CHKERRQ(ierr);}
18505b42c5fSBarry Smith   ierr = PetscFree(a->solve_work);CHKERRQ(ierr);
18641f059aeSBarry Smith   ierr = PetscFree(a->sor_work);CHKERRQ(ierr);
18705b42c5fSBarry Smith   ierr = PetscFree(a->solves_work);CHKERRQ(ierr);
18805b42c5fSBarry Smith   ierr = PetscFree(a->mult_work);CHKERRQ(ierr);
18905b42c5fSBarry Smith   ierr = PetscFree(a->saved_values);CHKERRQ(ierr);
1904da8f245SBarry Smith   if (a->free_jshort) {ierr = PetscFree(a->jshort);CHKERRQ(ierr);}
1911a3463dfSHong Zhang   ierr = PetscFree(a->inew);CHKERRQ(ierr);
1926bf464f9SBarry Smith   ierr = MatDestroy(&a->parent);CHKERRQ(ierr);
193bf0cc555SLisandro Dalcin   ierr = PetscFree(A->data);CHKERRQ(ierr);
194901853e0SKris Buschelman 
195dbd8c25aSHong Zhang   ierr = PetscObjectChangeTypeName((PetscObject)A,0);CHKERRQ(ierr);
196bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatStoreValues_C",NULL);CHKERRQ(ierr);
197bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatRetrieveValues_C",NULL);CHKERRQ(ierr);
198bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetColumnIndices_C",NULL);CHKERRQ(ierr);
199bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqaij_C",NULL);CHKERRQ(ierr);
200bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqbaij_C",NULL);CHKERRQ(ierr);
201bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetPreallocation_C",NULL);CHKERRQ(ierr);
20238f409ebSLisandro Dalcin   ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetPreallocationCSR_C",NULL);CHKERRQ(ierr);
203bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqsbstrm_C",NULL);CHKERRQ(ierr);
2046214f412SHong Zhang #if defined(PETSC_HAVE_ELEMENTAL)
2056214f412SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_elemental_C",NULL);CHKERRQ(ierr);
2066214f412SHong Zhang #endif
20749b5e25fSSatish Balay   PetscFunctionReturn(0);
20849b5e25fSSatish Balay }
20949b5e25fSSatish Balay 
210ace3abfcSBarry Smith PetscErrorCode MatSetOption_SeqSBAIJ(Mat A,MatOption op,PetscBool flg)
21149b5e25fSSatish Balay {
212045c9aa0SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
21363ba0a88SBarry Smith   PetscErrorCode ierr;
21449b5e25fSSatish Balay 
21549b5e25fSSatish Balay   PetscFunctionBegin;
2164d9d31abSKris Buschelman   switch (op) {
2174d9d31abSKris Buschelman   case MAT_ROW_ORIENTED:
2184e0d8c25SBarry Smith     a->roworiented = flg;
2194d9d31abSKris Buschelman     break;
220a9817697SBarry Smith   case MAT_KEEP_NONZERO_PATTERN:
221a9817697SBarry Smith     a->keepnonzeropattern = flg;
2224d9d31abSKris Buschelman     break;
223512a5fc5SBarry Smith   case MAT_NEW_NONZERO_LOCATIONS:
224512a5fc5SBarry Smith     a->nonew = (flg ? 0 : 1);
2254d9d31abSKris Buschelman     break;
2264d9d31abSKris Buschelman   case MAT_NEW_NONZERO_LOCATION_ERR:
2274e0d8c25SBarry Smith     a->nonew = (flg ? -1 : 0);
2284d9d31abSKris Buschelman     break;
2294d9d31abSKris Buschelman   case MAT_NEW_NONZERO_ALLOCATION_ERR:
2304e0d8c25SBarry Smith     a->nonew = (flg ? -2 : 0);
2314d9d31abSKris Buschelman     break;
23228b2fa4aSMatthew Knepley   case MAT_UNUSED_NONZERO_LOCATION_ERR:
23328b2fa4aSMatthew Knepley     a->nounused = (flg ? -1 : 0);
23428b2fa4aSMatthew Knepley     break;
2354e0d8c25SBarry Smith   case MAT_NEW_DIAGONALS:
2364d9d31abSKris Buschelman   case MAT_IGNORE_OFF_PROC_ENTRIES:
2374d9d31abSKris Buschelman   case MAT_USE_HASH_TABLE:
238290bbb0aSBarry Smith     ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr);
2394d9d31abSKris Buschelman     break;
2409a4540c5SBarry Smith   case MAT_HERMITIAN:
241e32f2f54SBarry Smith     if (!A->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Must call MatAssemblyEnd() first");
2420bc54ff2SBarry Smith     if (A->cmap->n < 65536 && A->cmap->bs == 1) {
243eeffb40dSHong Zhang       A->ops->mult = MatMult_SeqSBAIJ_1_Hermitian_ushort;
2440bc54ff2SBarry Smith     } else if (A->cmap->bs == 1) {
245eeffb40dSHong Zhang       A->ops->mult = MatMult_SeqSBAIJ_1_Hermitian;
246e32f2f54SBarry Smith     } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No support for Hermitian with block size greater than 1");
247eeffb40dSHong Zhang     break;
2483d472b54SHong Zhang   case MAT_SPD:
2495021d80fSJed Brown     /* These options are handled directly by MatSetOption() */
2503d472b54SHong Zhang     break;
25177e54ba9SKris Buschelman   case MAT_SYMMETRIC:
25277e54ba9SKris Buschelman   case MAT_STRUCTURALLY_SYMMETRIC:
2539a4540c5SBarry Smith   case MAT_SYMMETRY_ETERNAL:
2544dcd73b1SHong Zhang     /* These options are handled directly by MatSetOption() */
255290bbb0aSBarry Smith     break;
256941593c8SHong Zhang   case MAT_IGNORE_LOWER_TRIANGULAR:
2574e0d8c25SBarry Smith     a->ignore_ltriangular = flg;
258941593c8SHong Zhang     break;
259941593c8SHong Zhang   case MAT_ERROR_LOWER_TRIANGULAR:
2604e0d8c25SBarry Smith     a->ignore_ltriangular = flg;
26177e54ba9SKris Buschelman     break;
262f5edf698SHong Zhang   case MAT_GETROW_UPPERTRIANGULAR:
2634e0d8c25SBarry Smith     a->getrow_utriangular = flg;
264f5edf698SHong Zhang     break;
265c10200c1SHong Zhang   case MAT_SUBMAT_SINGLEIS:
266c10200c1SHong Zhang     break;
2674d9d31abSKris Buschelman   default:
268e32f2f54SBarry Smith     SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"unknown option %d",op);
26949b5e25fSSatish Balay   }
27049b5e25fSSatish Balay   PetscFunctionReturn(0);
27149b5e25fSSatish Balay }
27249b5e25fSSatish Balay 
27352768537SHong Zhang PetscErrorCode MatGetRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
27449b5e25fSSatish Balay {
27549b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
2766849ba73SBarry Smith   PetscErrorCode ierr;
27749b5e25fSSatish Balay 
27849b5e25fSSatish Balay   PetscFunctionBegin;
279e32f2f54SBarry 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()");
28052768537SHong Zhang 
281f5edf698SHong Zhang   /* Get the upper triangular part of the row */
28252768537SHong Zhang   ierr = MatGetRow_SeqBAIJ_private(A,row,nz,idx,v,a->i,a->j,a->a);CHKERRQ(ierr);
28349b5e25fSSatish Balay   PetscFunctionReturn(0);
28449b5e25fSSatish Balay }
28549b5e25fSSatish Balay 
28613f74950SBarry Smith PetscErrorCode MatRestoreRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
28749b5e25fSSatish Balay {
288dfbe8321SBarry Smith   PetscErrorCode ierr;
28949b5e25fSSatish Balay 
29049b5e25fSSatish Balay   PetscFunctionBegin;
29105b42c5fSBarry Smith   if (idx) {ierr = PetscFree(*idx);CHKERRQ(ierr);}
29205b42c5fSBarry Smith   if (v)   {ierr = PetscFree(*v);CHKERRQ(ierr);}
29349b5e25fSSatish Balay   PetscFunctionReturn(0);
29449b5e25fSSatish Balay }
29549b5e25fSSatish Balay 
296f5edf698SHong Zhang PetscErrorCode MatGetRowUpperTriangular_SeqSBAIJ(Mat A)
297f5edf698SHong Zhang {
298f5edf698SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
299f5edf698SHong Zhang 
300f5edf698SHong Zhang   PetscFunctionBegin;
301f5edf698SHong Zhang   a->getrow_utriangular = PETSC_TRUE;
302f5edf698SHong Zhang   PetscFunctionReturn(0);
303f5edf698SHong Zhang }
304f5edf698SHong Zhang PetscErrorCode MatRestoreRowUpperTriangular_SeqSBAIJ(Mat A)
305f5edf698SHong Zhang {
306f5edf698SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
307f5edf698SHong Zhang 
308f5edf698SHong Zhang   PetscFunctionBegin;
309f5edf698SHong Zhang   a->getrow_utriangular = PETSC_FALSE;
310f5edf698SHong Zhang   PetscFunctionReturn(0);
311f5edf698SHong Zhang }
312f5edf698SHong Zhang 
313fc4dec0aSBarry Smith PetscErrorCode MatTranspose_SeqSBAIJ(Mat A,MatReuse reuse,Mat *B)
31449b5e25fSSatish Balay {
315dfbe8321SBarry Smith   PetscErrorCode ierr;
3165fd66863SKarl Rupp 
31749b5e25fSSatish Balay   PetscFunctionBegin;
318cf37664fSBarry Smith   if (reuse == MAT_INITIAL_MATRIX) {
319999d9058SBarry Smith     ierr = MatDuplicate(A,MAT_COPY_VALUES,B);CHKERRQ(ierr);
320cf37664fSBarry Smith   } else if (reuse == MAT_REUSE_MATRIX) {
321cf37664fSBarry Smith     ierr = MatCopy(A,*B,SAME_NONZERO_PATTERN);CHKERRQ(ierr);
322fc4dec0aSBarry Smith   }
3238115998fSBarry Smith   PetscFunctionReturn(0);
32449b5e25fSSatish Balay }
32549b5e25fSSatish Balay 
3267da1fb6eSBarry Smith PetscErrorCode MatView_SeqSBAIJ_ASCII(Mat A,PetscViewer viewer)
32749b5e25fSSatish Balay {
32849b5e25fSSatish Balay   Mat_SeqSBAIJ      *a = (Mat_SeqSBAIJ*)A->data;
329dfbe8321SBarry Smith   PetscErrorCode    ierr;
330d0f46423SBarry Smith   PetscInt          i,j,bs = A->rmap->bs,k,l,bs2=a->bs2;
331f3ef73ceSBarry Smith   PetscViewerFormat format;
332121deb67SSatish Balay   PetscInt          *diag;
33349b5e25fSSatish Balay 
33449b5e25fSSatish Balay   PetscFunctionBegin;
335b0a32e0cSBarry Smith   ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
336456192e2SBarry Smith   if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
33777431f27SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  block size is %D\n",bs);CHKERRQ(ierr);
338fb9695e5SSatish Balay   } else if (format == PETSC_VIEWER_ASCII_MATLAB) {
339d2507d54SMatthew Knepley     Mat        aij;
340ade3a672SBarry Smith     const char *matname;
341ade3a672SBarry Smith 
342d5f3da31SBarry Smith     if (A->factortype && bs>1) {
34370d5e725SHong 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);
34470d5e725SHong Zhang       PetscFunctionReturn(0);
34570d5e725SHong Zhang     }
346c9f458caSMatthew Knepley     ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&aij);CHKERRQ(ierr);
347ade3a672SBarry Smith     ierr = PetscObjectGetName((PetscObject)A,&matname);CHKERRQ(ierr);
348ade3a672SBarry Smith     ierr = PetscObjectSetName((PetscObject)aij,matname);CHKERRQ(ierr);
349c9f458caSMatthew Knepley     ierr = MatView(aij,viewer);CHKERRQ(ierr);
3506bf464f9SBarry Smith     ierr = MatDestroy(&aij);CHKERRQ(ierr);
351fb9695e5SSatish Balay   } else if (format == PETSC_VIEWER_ASCII_COMMON) {
352d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
35349b5e25fSSatish Balay     for (i=0; i<a->mbs; i++) {
35449b5e25fSSatish Balay       for (j=0; j<bs; j++) {
35577431f27SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr);
35649b5e25fSSatish Balay         for (k=a->i[i]; k<a->i[i+1]; k++) {
35749b5e25fSSatish Balay           for (l=0; l<bs; l++) {
35849b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX)
35949b5e25fSSatish Balay             if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
36057622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k]+l,
36157622a8eSBarry Smith                                             (double)PetscRealPart(a->a[bs2*k + l*bs + j]),(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
36249b5e25fSSatish Balay             } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
36357622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k]+l,
36457622a8eSBarry Smith                                             (double)PetscRealPart(a->a[bs2*k + l*bs + j]),-(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
36549b5e25fSSatish Balay             } else if (PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
36657622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
36749b5e25fSSatish Balay             }
36849b5e25fSSatish Balay #else
36949b5e25fSSatish Balay             if (a->a[bs2*k + l*bs + j] != 0.0) {
37057622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)a->a[bs2*k + l*bs + j]);CHKERRQ(ierr);
37149b5e25fSSatish Balay             }
37249b5e25fSSatish Balay #endif
37349b5e25fSSatish Balay           }
37449b5e25fSSatish Balay         }
375b0a32e0cSBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
37649b5e25fSSatish Balay       }
37749b5e25fSSatish Balay     }
378d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
379c1490034SHong Zhang   } else if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) {
380c1490034SHong Zhang     PetscFunctionReturn(0);
38149b5e25fSSatish Balay   } else {
382d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
3832c990fa1SHong Zhang     if (A->factortype) { /* for factored matrix */
3842c990fa1SHong Zhang       if (bs>1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"matrix is factored with bs>1. Not implemented yet");
3852c990fa1SHong Zhang 
386121deb67SSatish Balay       diag=a->diag;
387121deb67SSatish Balay       for (i=0; i<a->mbs; i++) { /* for row block i */
3882c990fa1SHong Zhang         ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i);CHKERRQ(ierr);
3892c990fa1SHong Zhang         /* diagonal entry */
3902c990fa1SHong Zhang #if defined(PETSC_USE_COMPLEX)
3912c990fa1SHong Zhang         if (PetscImaginaryPart(a->a[diag[i]]) > 0.0) {
39257622a8eSBarry Smith           ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",a->j[diag[i]],(double)PetscRealPart(1.0/a->a[diag[i]]),(double)PetscImaginaryPart(1.0/a->a[diag[i]]));CHKERRQ(ierr);
3932c990fa1SHong Zhang         } else if (PetscImaginaryPart(a->a[diag[i]]) < 0.0) {
39457622a8eSBarry Smith           ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",a->j[diag[i]],(double)PetscRealPart(1.0/a->a[diag[i]]),-(double)PetscImaginaryPart(1.0/a->a[diag[i]]));CHKERRQ(ierr);
3952c990fa1SHong Zhang         } else {
39657622a8eSBarry Smith           ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[diag[i]],(double)PetscRealPart(1.0/a->a[diag[i]]));CHKERRQ(ierr);
3972c990fa1SHong Zhang         }
3982c990fa1SHong Zhang #else
3996712e2f1SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[diag[i]],(double)(1.0/a->a[diag[i]]));CHKERRQ(ierr);
4002c990fa1SHong Zhang #endif
4012c990fa1SHong Zhang         /* off-diagonal entries */
4022c990fa1SHong Zhang         for (k=a->i[i]; k<a->i[i+1]-1; k++) {
4032c990fa1SHong Zhang #if defined(PETSC_USE_COMPLEX)
404ca0704adSBarry Smith           if (PetscImaginaryPart(a->a[k]) > 0.0) {
40557622a8eSBarry Smith             ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k],(double)PetscRealPart(a->a[k]),(double)PetscImaginaryPart(a->a[k]));CHKERRQ(ierr);
406ca0704adSBarry Smith           } else if (PetscImaginaryPart(a->a[k]) < 0.0) {
40757622a8eSBarry Smith             ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k],(double)PetscRealPart(a->a[k]),-(double)PetscImaginaryPart(a->a[k]));CHKERRQ(ierr);
4082c990fa1SHong Zhang           } else {
40957622a8eSBarry Smith             ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k],(double)PetscRealPart(a->a[k]));CHKERRQ(ierr);
4102c990fa1SHong Zhang           }
4112c990fa1SHong Zhang #else
41257622a8eSBarry Smith           ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[k],(double)a->a[k]);CHKERRQ(ierr);
4132c990fa1SHong Zhang #endif
4142c990fa1SHong Zhang         }
4152c990fa1SHong Zhang         ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
4162c990fa1SHong Zhang       }
4172c990fa1SHong Zhang 
4182c990fa1SHong Zhang     } else { /* for non-factored matrix */
4190c74a584SJed Brown       for (i=0; i<a->mbs; i++) { /* for row block i */
4200c74a584SJed Brown         for (j=0; j<bs; j++) {   /* for row bs*i + j */
42177431f27SBarry Smith           ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr);
4220c74a584SJed Brown           for (k=a->i[i]; k<a->i[i+1]; k++) { /* for column block */
4230c74a584SJed Brown             for (l=0; l<bs; l++) {            /* for column */
42449b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX)
42549b5e25fSSatish Balay               if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0) {
42657622a8eSBarry Smith                 ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k]+l,
42757622a8eSBarry Smith                                               (double)PetscRealPart(a->a[bs2*k + l*bs + j]),(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
42849b5e25fSSatish Balay               } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0) {
42957622a8eSBarry Smith                 ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k]+l,
43057622a8eSBarry Smith                                               (double)PetscRealPart(a->a[bs2*k + l*bs + j]),-(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
43149b5e25fSSatish Balay               } else {
43257622a8eSBarry Smith                 ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
43349b5e25fSSatish Balay               }
43449b5e25fSSatish Balay #else
43557622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)a->a[bs2*k + l*bs + j]);CHKERRQ(ierr);
43649b5e25fSSatish Balay #endif
43749b5e25fSSatish Balay             }
43849b5e25fSSatish Balay           }
439b0a32e0cSBarry Smith           ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
44049b5e25fSSatish Balay         }
44149b5e25fSSatish Balay       }
4422c990fa1SHong Zhang     }
443d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
44449b5e25fSSatish Balay   }
445b0a32e0cSBarry Smith   ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
44649b5e25fSSatish Balay   PetscFunctionReturn(0);
44749b5e25fSSatish Balay }
44849b5e25fSSatish Balay 
4499804daf3SBarry Smith #include <petscdraw.h>
4506849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw_Zoom(PetscDraw draw,void *Aa)
45149b5e25fSSatish Balay {
45249b5e25fSSatish Balay   Mat            A = (Mat) Aa;
45349b5e25fSSatish Balay   Mat_SeqSBAIJ   *a=(Mat_SeqSBAIJ*)A->data;
4546849ba73SBarry Smith   PetscErrorCode ierr;
455d0f46423SBarry Smith   PetscInt       row,i,j,k,l,mbs=a->mbs,color,bs=A->rmap->bs,bs2=a->bs2;
45649b5e25fSSatish Balay   PetscReal      xl,yl,xr,yr,x_l,x_r,y_l,y_r;
45749b5e25fSSatish Balay   MatScalar      *aa;
458b0a32e0cSBarry Smith   PetscViewer    viewer;
45949b5e25fSSatish Balay 
46049b5e25fSSatish Balay   PetscFunctionBegin;
46149b5e25fSSatish Balay   ierr = PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer);CHKERRQ(ierr);
462b0a32e0cSBarry Smith   ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr);
46349b5e25fSSatish Balay 
46449b5e25fSSatish Balay   /* loop over matrix elements drawing boxes */
465383922c3SLisandro Dalcin 
466383922c3SLisandro Dalcin   ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr);
467383922c3SLisandro Dalcin   ierr = PetscDrawString(draw, .3*(xl+xr), .3*(yl+yr), PETSC_DRAW_BLACK, "symmetric");CHKERRQ(ierr);
468383922c3SLisandro Dalcin   /* Blue for negative, Cyan for zero and  Red for positive */
469b0a32e0cSBarry Smith   color = PETSC_DRAW_BLUE;
47049b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
47149b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
472d0f46423SBarry Smith       y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0;
47349b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
47449b5e25fSSatish Balay       aa  = a->a + j*bs2;
47549b5e25fSSatish Balay       for (k=0; k<bs; k++) {
47649b5e25fSSatish Balay         for (l=0; l<bs; l++) {
47749b5e25fSSatish Balay           if (PetscRealPart(*aa++) >=  0.) continue;
478b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
47949b5e25fSSatish Balay         }
48049b5e25fSSatish Balay       }
48149b5e25fSSatish Balay     }
48249b5e25fSSatish Balay   }
483b0a32e0cSBarry Smith   color = PETSC_DRAW_CYAN;
48449b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
48549b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
486d0f46423SBarry Smith       y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0;
48749b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
48849b5e25fSSatish Balay       aa = a->a + j*bs2;
48949b5e25fSSatish Balay       for (k=0; k<bs; k++) {
49049b5e25fSSatish Balay         for (l=0; l<bs; l++) {
49149b5e25fSSatish Balay           if (PetscRealPart(*aa++) != 0.) continue;
492b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
49349b5e25fSSatish Balay         }
49449b5e25fSSatish Balay       }
49549b5e25fSSatish Balay     }
49649b5e25fSSatish Balay   }
497b0a32e0cSBarry Smith   color = PETSC_DRAW_RED;
49849b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
49949b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
500d0f46423SBarry Smith       y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0;
50149b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
50249b5e25fSSatish Balay       aa = a->a + j*bs2;
50349b5e25fSSatish Balay       for (k=0; k<bs; k++) {
50449b5e25fSSatish Balay         for (l=0; l<bs; l++) {
50549b5e25fSSatish Balay           if (PetscRealPart(*aa++) <= 0.) continue;
506b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
50749b5e25fSSatish Balay         }
50849b5e25fSSatish Balay       }
50949b5e25fSSatish Balay     }
51049b5e25fSSatish Balay   }
511383922c3SLisandro Dalcin   ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr);
51249b5e25fSSatish Balay   PetscFunctionReturn(0);
51349b5e25fSSatish Balay }
51449b5e25fSSatish Balay 
5156849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw(Mat A,PetscViewer viewer)
51649b5e25fSSatish Balay {
517dfbe8321SBarry Smith   PetscErrorCode ierr;
51849b5e25fSSatish Balay   PetscReal      xl,yl,xr,yr,w,h;
519b0a32e0cSBarry Smith   PetscDraw      draw;
520ace3abfcSBarry Smith   PetscBool      isnull;
52149b5e25fSSatish Balay 
52249b5e25fSSatish Balay   PetscFunctionBegin;
523b0a32e0cSBarry Smith   ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
524383922c3SLisandro Dalcin   ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr);
525383922c3SLisandro Dalcin   if (isnull) PetscFunctionReturn(0);
52649b5e25fSSatish Balay 
527d0f46423SBarry Smith   xr   = A->rmap->N; yr = A->rmap->N; h = yr/10.0; w = xr/10.0;
52849b5e25fSSatish Balay   xr  += w;          yr += h;        xl = -w;     yl = -h;
529b0a32e0cSBarry Smith   ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr);
530832b7cebSLisandro Dalcin   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer);CHKERRQ(ierr);
531b0a32e0cSBarry Smith   ierr = PetscDrawZoom(draw,MatView_SeqSBAIJ_Draw_Zoom,A);CHKERRQ(ierr);
5320298fd71SBarry Smith   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",NULL);CHKERRQ(ierr);
533832b7cebSLisandro Dalcin   ierr = PetscDrawSave(draw);CHKERRQ(ierr);
53449b5e25fSSatish Balay   PetscFunctionReturn(0);
53549b5e25fSSatish Balay }
53649b5e25fSSatish Balay 
537dfbe8321SBarry Smith PetscErrorCode MatView_SeqSBAIJ(Mat A,PetscViewer viewer)
53849b5e25fSSatish Balay {
539dfbe8321SBarry Smith   PetscErrorCode ierr;
540ace3abfcSBarry Smith   PetscBool      iascii,isdraw;
54108917f38SBarry Smith   FILE           *file = 0;
54249b5e25fSSatish Balay 
54349b5e25fSSatish Balay   PetscFunctionBegin;
544251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
545251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr);
54632077d6dSBarry Smith   if (iascii) {
54749b5e25fSSatish Balay     ierr = MatView_SeqSBAIJ_ASCII(A,viewer);CHKERRQ(ierr);
54849b5e25fSSatish Balay   } else if (isdraw) {
54949b5e25fSSatish Balay     ierr = MatView_SeqSBAIJ_Draw(A,viewer);CHKERRQ(ierr);
55049b5e25fSSatish Balay   } else {
551a5e6ed63SBarry Smith     Mat        B;
552ade3a672SBarry Smith     const char *matname;
553ceb03754SKris Buschelman     ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&B);CHKERRQ(ierr);
554ade3a672SBarry Smith     ierr = PetscObjectGetName((PetscObject)A,&matname);CHKERRQ(ierr);
555ade3a672SBarry Smith     ierr = PetscObjectSetName((PetscObject)B,matname);CHKERRQ(ierr);
556a5e6ed63SBarry Smith     ierr = MatView(B,viewer);CHKERRQ(ierr);
5576bf464f9SBarry Smith     ierr = MatDestroy(&B);CHKERRQ(ierr);
55808917f38SBarry Smith     ierr = PetscViewerBinaryGetInfoPointer(viewer,&file);CHKERRQ(ierr);
55908917f38SBarry Smith     if (file) {
56008917f38SBarry Smith       fprintf(file,"-matload_block_size %d\n",(int)A->rmap->bs);
56108917f38SBarry Smith     }
56249b5e25fSSatish Balay   }
56349b5e25fSSatish Balay   PetscFunctionReturn(0);
56449b5e25fSSatish Balay }
56549b5e25fSSatish Balay 
56649b5e25fSSatish Balay 
56713f74950SBarry Smith PetscErrorCode MatGetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],PetscScalar v[])
56849b5e25fSSatish Balay {
569045c9aa0SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
57013f74950SBarry Smith   PetscInt     *rp,k,low,high,t,row,nrow,i,col,l,*aj = a->j;
57113f74950SBarry Smith   PetscInt     *ai = a->i,*ailen = a->ilen;
572d0f46423SBarry Smith   PetscInt     brow,bcol,ridx,cidx,bs=A->rmap->bs,bs2=a->bs2;
57397e567efSBarry Smith   MatScalar    *ap,*aa = a->a;
57449b5e25fSSatish Balay 
57549b5e25fSSatish Balay   PetscFunctionBegin;
57649b5e25fSSatish Balay   for (k=0; k<m; k++) { /* loop over rows */
57749b5e25fSSatish Balay     row = im[k]; brow = row/bs;
578e32f2f54SBarry Smith     if (row < 0) {v += n; continue;} /* SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative row: %D",row); */
579e32f2f54SBarry 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);
58049b5e25fSSatish Balay     rp   = aj + ai[brow]; ap = aa + bs2*ai[brow];
58149b5e25fSSatish Balay     nrow = ailen[brow];
58249b5e25fSSatish Balay     for (l=0; l<n; l++) { /* loop over columns */
583e32f2f54SBarry Smith       if (in[l] < 0) {v++; continue;} /* SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column: %D",in[l]); */
584e32f2f54SBarry 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);
58549b5e25fSSatish Balay       col  = in[l];
58649b5e25fSSatish Balay       bcol = col/bs;
58749b5e25fSSatish Balay       cidx = col%bs;
58849b5e25fSSatish Balay       ridx = row%bs;
58949b5e25fSSatish Balay       high = nrow;
59049b5e25fSSatish Balay       low  = 0; /* assume unsorted */
59149b5e25fSSatish Balay       while (high-low > 5) {
59249b5e25fSSatish Balay         t = (low+high)/2;
59349b5e25fSSatish Balay         if (rp[t] > bcol) high = t;
59449b5e25fSSatish Balay         else              low  = t;
59549b5e25fSSatish Balay       }
59649b5e25fSSatish Balay       for (i=low; i<high; i++) {
59749b5e25fSSatish Balay         if (rp[i] > bcol) break;
59849b5e25fSSatish Balay         if (rp[i] == bcol) {
59949b5e25fSSatish Balay           *v++ = ap[bs2*i+bs*cidx+ridx];
60049b5e25fSSatish Balay           goto finished;
60149b5e25fSSatish Balay         }
60249b5e25fSSatish Balay       }
60397e567efSBarry Smith       *v++ = 0.0;
60449b5e25fSSatish Balay finished:;
60549b5e25fSSatish Balay     }
60649b5e25fSSatish Balay   }
60749b5e25fSSatish Balay   PetscFunctionReturn(0);
60849b5e25fSSatish Balay }
60949b5e25fSSatish Balay 
61049b5e25fSSatish Balay 
61113f74950SBarry Smith PetscErrorCode MatSetValuesBlocked_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is)
61249b5e25fSSatish Balay {
6130880e062SHong Zhang   Mat_SeqSBAIJ      *a = (Mat_SeqSBAIJ*)A->data;
6146849ba73SBarry Smith   PetscErrorCode    ierr;
615e2ee6c50SBarry Smith   PetscInt          *rp,k,low,high,t,ii,jj,row,nrow,i,col,l,rmax,N,lastcol = -1;
61613f74950SBarry Smith   PetscInt          *imax      =a->imax,*ai=a->i,*ailen=a->ilen;
617d0f46423SBarry Smith   PetscInt          *aj        =a->j,nonew=a->nonew,bs2=a->bs2,bs=A->rmap->bs,stepval;
618ace3abfcSBarry Smith   PetscBool         roworiented=a->roworiented;
619dd6ea824SBarry Smith   const PetscScalar *value     = v;
620f15d580aSBarry Smith   MatScalar         *ap,*aa = a->a,*bap;
6210880e062SHong Zhang 
62249b5e25fSSatish Balay   PetscFunctionBegin;
62326fbe8dcSKarl Rupp   if (roworiented) stepval = (n-1)*bs;
62426fbe8dcSKarl Rupp   else stepval = (m-1)*bs;
62526fbe8dcSKarl Rupp 
6260880e062SHong Zhang   for (k=0; k<m; k++) { /* loop over added rows */
6270880e062SHong Zhang     row = im[k];
6280880e062SHong Zhang     if (row < 0) continue;
6292515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
6302f7d4af7SBarry Smith     if (row >= a->mbs) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Block index row too large %D max %D",row,a->mbs-1);
6310880e062SHong Zhang #endif
6320880e062SHong Zhang     rp   = aj + ai[row];
6330880e062SHong Zhang     ap   = aa + bs2*ai[row];
6340880e062SHong Zhang     rmax = imax[row];
6350880e062SHong Zhang     nrow = ailen[row];
6360880e062SHong Zhang     low  = 0;
637818f2c47SBarry Smith     high = nrow;
6380880e062SHong Zhang     for (l=0; l<n; l++) { /* loop over added columns */
6390880e062SHong Zhang       if (in[l] < 0) continue;
6400880e062SHong Zhang       col = in[l];
6412515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
6422f7d4af7SBarry Smith       if (col >= a->nbs) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Block index column too large %D max %D",col,a->nbs-1);
643b1823623SSatish Balay #endif
644b98bf0e1SJed Brown       if (col < row) {
64526fbe8dcSKarl Rupp         if (a->ignore_ltriangular) continue; /* ignore lower triangular block */
64626fbe8dcSKarl Rupp         else 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)");
647b98bf0e1SJed Brown       }
64826fbe8dcSKarl Rupp       if (roworiented) value = v + k*(stepval+bs)*bs + l*bs;
64926fbe8dcSKarl Rupp       else value = v + l*(stepval+bs)*bs + k*bs;
65026fbe8dcSKarl Rupp 
65126fbe8dcSKarl Rupp       if (col <= lastcol) low = 0;
65226fbe8dcSKarl Rupp       else high = nrow;
65326fbe8dcSKarl Rupp 
654e2ee6c50SBarry Smith       lastcol = col;
6550880e062SHong Zhang       while (high-low > 7) {
6560880e062SHong Zhang         t = (low+high)/2;
6570880e062SHong Zhang         if (rp[t] > col) high = t;
6580880e062SHong Zhang         else             low  = t;
6590880e062SHong Zhang       }
6600880e062SHong Zhang       for (i=low; i<high; i++) {
6610880e062SHong Zhang         if (rp[i] > col) break;
6620880e062SHong Zhang         if (rp[i] == col) {
6630880e062SHong Zhang           bap = ap +  bs2*i;
6640880e062SHong Zhang           if (roworiented) {
6650880e062SHong Zhang             if (is == ADD_VALUES) {
6660880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
6670880e062SHong Zhang                 for (jj=ii; jj<bs2; jj+=bs) {
6680880e062SHong Zhang                   bap[jj] += *value++;
6690880e062SHong Zhang                 }
6700880e062SHong Zhang               }
6710880e062SHong Zhang             } else {
6720880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
6730880e062SHong Zhang                 for (jj=ii; jj<bs2; jj+=bs) {
6740880e062SHong Zhang                   bap[jj] = *value++;
6750880e062SHong Zhang                 }
6760880e062SHong Zhang                }
6770880e062SHong Zhang             }
6780880e062SHong Zhang           } else {
6790880e062SHong Zhang             if (is == ADD_VALUES) {
6800880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
6810880e062SHong Zhang                 for (jj=0; jj<bs; jj++) {
6820880e062SHong Zhang                   *bap++ += *value++;
6830880e062SHong Zhang                 }
6840880e062SHong Zhang               }
6850880e062SHong Zhang             } else {
6860880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
6870880e062SHong Zhang                 for (jj=0; jj<bs; jj++) {
6880880e062SHong Zhang                   *bap++  = *value++;
6890880e062SHong Zhang                 }
6900880e062SHong Zhang               }
6910880e062SHong Zhang             }
6920880e062SHong Zhang           }
6930880e062SHong Zhang           goto noinsert2;
6940880e062SHong Zhang         }
6950880e062SHong Zhang       }
6960880e062SHong Zhang       if (nonew == 1) goto noinsert2;
6972f7d4af7SBarry Smith       if (nonew == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new block index nonzero block (%D, %D) in the matrix", row, col);
698fef13f97SBarry Smith       MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,row,col,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar);
699c03d1d03SSatish Balay       N = nrow++ - 1; high++;
7000880e062SHong Zhang       /* shift up all the later entries in this row */
7010880e062SHong Zhang       for (ii=N; ii>=i; ii--) {
7020880e062SHong Zhang         rp[ii+1] = rp[ii];
7030880e062SHong Zhang         ierr     = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr);
7040880e062SHong Zhang       }
7050880e062SHong Zhang       if (N >= i) {
7060880e062SHong Zhang         ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr);
7070880e062SHong Zhang       }
7080880e062SHong Zhang       rp[i] = col;
7090880e062SHong Zhang       bap   = ap +  bs2*i;
7100880e062SHong Zhang       if (roworiented) {
7110880e062SHong Zhang         for (ii=0; ii<bs; ii++,value+=stepval) {
7120880e062SHong Zhang           for (jj=ii; jj<bs2; jj+=bs) {
7130880e062SHong Zhang             bap[jj] = *value++;
7140880e062SHong Zhang           }
7150880e062SHong Zhang         }
7160880e062SHong Zhang       } else {
7170880e062SHong Zhang         for (ii=0; ii<bs; ii++,value+=stepval) {
7180880e062SHong Zhang           for (jj=0; jj<bs; jj++) {
7190880e062SHong Zhang             *bap++ = *value++;
7200880e062SHong Zhang           }
7210880e062SHong Zhang         }
7220880e062SHong Zhang        }
7230880e062SHong Zhang     noinsert2:;
7240880e062SHong Zhang       low = i;
7250880e062SHong Zhang     }
7260880e062SHong Zhang     ailen[row] = nrow;
7270880e062SHong Zhang   }
7280880e062SHong Zhang   PetscFunctionReturn(0);
72949b5e25fSSatish Balay }
73049b5e25fSSatish Balay 
73164831d72SBarry Smith /*
73264831d72SBarry Smith     This is not yet used
73364831d72SBarry Smith */
7344108e4d5SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ_SeqAIJ_Inode(Mat A)
7350def2e27SBarry Smith {
7360def2e27SBarry Smith   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
7370def2e27SBarry Smith   PetscErrorCode ierr;
7380def2e27SBarry Smith   const PetscInt *ai = a->i, *aj = a->j,*cols;
7390def2e27SBarry Smith   PetscInt       i   = 0,j,blk_size,m = A->rmap->n,node_count = 0,nzx,nzy,*ns,row,nz,cnt,cnt2,*counts;
740ace3abfcSBarry Smith   PetscBool      flag;
7410def2e27SBarry Smith 
7420def2e27SBarry Smith   PetscFunctionBegin;
743785e854fSJed Brown   ierr = PetscMalloc1(m,&ns);CHKERRQ(ierr);
7440def2e27SBarry Smith   while (i < m) {
7450def2e27SBarry Smith     nzx = ai[i+1] - ai[i];       /* Number of nonzeros */
7460def2e27SBarry Smith     /* Limits the number of elements in a node to 'a->inode.limit' */
7470def2e27SBarry Smith     for (j=i+1,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) {
7480def2e27SBarry Smith       nzy = ai[j+1] - ai[j];
7490def2e27SBarry Smith       if (nzy != (nzx - j + i)) break;
7500def2e27SBarry Smith       ierr = PetscMemcmp(aj + ai[i] + j - i,aj + ai[j],nzy*sizeof(PetscInt),&flag);CHKERRQ(ierr);
7510def2e27SBarry Smith       if (!flag) break;
7520def2e27SBarry Smith     }
7530def2e27SBarry Smith     ns[node_count++] = blk_size;
75426fbe8dcSKarl Rupp 
7550def2e27SBarry Smith     i = j;
7560def2e27SBarry Smith   }
7570def2e27SBarry Smith   if (!a->inode.size && m && node_count > .9*m) {
7580def2e27SBarry Smith     ierr = PetscFree(ns);CHKERRQ(ierr);
7590def2e27SBarry Smith     ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr);
7600def2e27SBarry Smith   } else {
7610def2e27SBarry Smith     a->inode.node_count = node_count;
76226fbe8dcSKarl Rupp 
763785e854fSJed Brown     ierr = PetscMalloc1(node_count,&a->inode.size);CHKERRQ(ierr);
7643bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)A,node_count*sizeof(PetscInt));CHKERRQ(ierr);
76522d28d08SBarry Smith     ierr = PetscMemcpy(a->inode.size,ns,node_count*sizeof(PetscInt));CHKERRQ(ierr);
7660def2e27SBarry Smith     ierr = PetscFree(ns);CHKERRQ(ierr);
7670def2e27SBarry Smith     ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr);
7680def2e27SBarry Smith 
7690def2e27SBarry Smith     /* count collections of adjacent columns in each inode */
7700def2e27SBarry Smith     row = 0;
7710def2e27SBarry Smith     cnt = 0;
7720def2e27SBarry Smith     for (i=0; i<node_count; i++) {
7730def2e27SBarry Smith       cols = aj + ai[row] + a->inode.size[i];
7740def2e27SBarry Smith       nz   = ai[row+1] - ai[row] - a->inode.size[i];
7750def2e27SBarry Smith       for (j=1; j<nz; j++) {
77626fbe8dcSKarl Rupp         if (cols[j] != cols[j-1]+1) cnt++;
7770def2e27SBarry Smith       }
7780def2e27SBarry Smith       cnt++;
7790def2e27SBarry Smith       row += a->inode.size[i];
7800def2e27SBarry Smith     }
781785e854fSJed Brown     ierr = PetscMalloc1(2*cnt,&counts);CHKERRQ(ierr);
7820def2e27SBarry Smith     cnt  = 0;
7830def2e27SBarry Smith     row  = 0;
7840def2e27SBarry Smith     for (i=0; i<node_count; i++) {
7850def2e27SBarry Smith       cols = aj + ai[row] + a->inode.size[i];
7860def2e27SBarry Smith       counts[2*cnt] = cols[0];
7870def2e27SBarry Smith       nz   = ai[row+1] - ai[row] - a->inode.size[i];
7880def2e27SBarry Smith       cnt2 = 1;
7890def2e27SBarry Smith       for (j=1; j<nz; j++) {
7900def2e27SBarry Smith         if (cols[j] != cols[j-1]+1) {
7910def2e27SBarry Smith           counts[2*(cnt++)+1] = cnt2;
7920def2e27SBarry Smith           counts[2*cnt]       = cols[j];
7930def2e27SBarry Smith           cnt2 = 1;
7940def2e27SBarry Smith         } else cnt2++;
7950def2e27SBarry Smith       }
7960def2e27SBarry Smith       counts[2*(cnt++)+1] = cnt2;
7970def2e27SBarry Smith       row += a->inode.size[i];
7980def2e27SBarry Smith     }
79922d28d08SBarry Smith     ierr = PetscIntView(2*cnt,counts,0);CHKERRQ(ierr);
8000def2e27SBarry Smith   }
80138702af4SBarry Smith   PetscFunctionReturn(0);
80238702af4SBarry Smith }
80338702af4SBarry Smith 
804dfbe8321SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ(Mat A,MatAssemblyType mode)
80549b5e25fSSatish Balay {
80649b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
8076849ba73SBarry Smith   PetscErrorCode ierr;
80813f74950SBarry Smith   PetscInt       fshift = 0,i,j,*ai = a->i,*aj = a->j,*imax = a->imax;
809d0f46423SBarry Smith   PetscInt       m      = A->rmap->N,*ip,N,*ailen = a->ilen;
81013f74950SBarry Smith   PetscInt       mbs    = a->mbs,bs2 = a->bs2,rmax = 0;
81149b5e25fSSatish Balay   MatScalar      *aa    = a->a,*ap;
81249b5e25fSSatish Balay 
81349b5e25fSSatish Balay   PetscFunctionBegin;
81449b5e25fSSatish Balay   if (mode == MAT_FLUSH_ASSEMBLY) PetscFunctionReturn(0);
81549b5e25fSSatish Balay 
81649b5e25fSSatish Balay   if (m) rmax = ailen[0];
81749b5e25fSSatish Balay   for (i=1; i<mbs; i++) {
81849b5e25fSSatish Balay     /* move each row back by the amount of empty slots (fshift) before it*/
81949b5e25fSSatish Balay     fshift += imax[i-1] - ailen[i-1];
82049b5e25fSSatish Balay     rmax    = PetscMax(rmax,ailen[i]);
82149b5e25fSSatish Balay     if (fshift) {
82249b5e25fSSatish Balay       ip = aj + ai[i]; ap = aa + bs2*ai[i];
82349b5e25fSSatish Balay       N  = ailen[i];
82449b5e25fSSatish Balay       for (j=0; j<N; j++) {
82549b5e25fSSatish Balay         ip[j-fshift] = ip[j];
82649b5e25fSSatish Balay         ierr         = PetscMemcpy(ap+(j-fshift)*bs2,ap+j*bs2,bs2*sizeof(MatScalar));CHKERRQ(ierr);
82749b5e25fSSatish Balay       }
82849b5e25fSSatish Balay     }
82949b5e25fSSatish Balay     ai[i] = ai[i-1] + ailen[i-1];
83049b5e25fSSatish Balay   }
83149b5e25fSSatish Balay   if (mbs) {
83249b5e25fSSatish Balay     fshift += imax[mbs-1] - ailen[mbs-1];
83349b5e25fSSatish Balay     ai[mbs] = ai[mbs-1] + ailen[mbs-1];
83449b5e25fSSatish Balay   }
83549b5e25fSSatish Balay   /* reset ilen and imax for each row */
83649b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
83749b5e25fSSatish Balay     ailen[i] = imax[i] = ai[i+1] - ai[i];
83849b5e25fSSatish Balay   }
8396c6c5352SBarry Smith   a->nz = ai[mbs];
84049b5e25fSSatish Balay 
841b424e231SHong Zhang   /* diagonals may have moved, reset it */
842b424e231SHong Zhang   if (a->diag) {
8432ed38d0bSJed Brown     ierr = PetscMemcpy(a->diag,ai,mbs*sizeof(PetscInt));CHKERRQ(ierr);
84449b5e25fSSatish Balay   }
84526fbe8dcSKarl Rupp   if (fshift && a->nounused == -1) 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);
84626fbe8dcSKarl Rupp 
847d0f46423SBarry 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);
848ae15b995SBarry Smith   ierr = PetscInfo1(A,"Number of mallocs during MatSetValues is %D\n",a->reallocs);CHKERRQ(ierr);
849ae15b995SBarry Smith   ierr = PetscInfo1(A,"Most nonzeros blocks in any row is %D\n",rmax);CHKERRQ(ierr);
85026fbe8dcSKarl Rupp 
8518e58a170SBarry Smith   A->info.mallocs    += a->reallocs;
85249b5e25fSSatish Balay   a->reallocs         = 0;
85349b5e25fSSatish Balay   A->info.nz_unneeded = (PetscReal)fshift*bs2;
854061b2667SBarry Smith   a->idiagvalid       = PETSC_FALSE;
8554dcd73b1SHong Zhang   a->rmax             = rmax;
85638702af4SBarry Smith 
85738702af4SBarry Smith   if (A->cmap->n < 65536 && A->cmap->bs == 1) {
85844e1c64aSLisandro Dalcin     if (a->jshort && a->free_jshort) {
85917803ae8SHong Zhang       /* when matrix data structure is changed, previous jshort must be replaced */
86017803ae8SHong Zhang       ierr = PetscFree(a->jshort);CHKERRQ(ierr);
86117803ae8SHong Zhang     }
862785e854fSJed Brown     ierr = PetscMalloc1(a->i[A->rmap->n],&a->jshort);CHKERRQ(ierr);
8633bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)A,a->i[A->rmap->n]*sizeof(unsigned short));CHKERRQ(ierr);
86438702af4SBarry Smith     for (i=0; i<a->i[A->rmap->n]; i++) a->jshort[i] = a->j[i];
86538702af4SBarry Smith     A->ops->mult   = MatMult_SeqSBAIJ_1_ushort;
86641f059aeSBarry Smith     A->ops->sor    = MatSOR_SeqSBAIJ_ushort;
8674da8f245SBarry Smith     a->free_jshort = PETSC_TRUE;
86838702af4SBarry Smith   }
86949b5e25fSSatish Balay   PetscFunctionReturn(0);
87049b5e25fSSatish Balay }
87149b5e25fSSatish Balay 
87249b5e25fSSatish Balay /*
87349b5e25fSSatish Balay    This function returns an array of flags which indicate the locations of contiguous
87449b5e25fSSatish Balay    blocks that should be zeroed. for eg: if bs = 3  and is = [0,1,2,3,5,6,7,8,9]
87549b5e25fSSatish Balay    then the resulting sizes = [3,1,1,3,1] correspondig to sets [(0,1,2),(3),(5),(6,7,8),(9)]
87649b5e25fSSatish Balay    Assume: sizes should be long enough to hold all the values.
87749b5e25fSSatish Balay */
87813f74950SBarry Smith PetscErrorCode MatZeroRows_SeqSBAIJ_Check_Blocks(PetscInt idx[],PetscInt n,PetscInt bs,PetscInt sizes[], PetscInt *bs_max)
87949b5e25fSSatish Balay {
88013f74950SBarry Smith   PetscInt  i,j,k,row;
881ace3abfcSBarry Smith   PetscBool flg;
88249b5e25fSSatish Balay 
88349b5e25fSSatish Balay   PetscFunctionBegin;
88449b5e25fSSatish Balay   for (i=0,j=0; i<n; j++) {
88549b5e25fSSatish Balay     row = idx[i];
88649b5e25fSSatish Balay     if (row%bs!=0) { /* Not the begining of a block */
88749b5e25fSSatish Balay       sizes[j] = 1;
88849b5e25fSSatish Balay       i++;
88949b5e25fSSatish Balay     } else if (i+bs > n) { /* Beginning of a block, but complete block doesn't exist (at idx end) */
89049b5e25fSSatish Balay       sizes[j] = 1;         /* Also makes sure atleast 'bs' values exist for next else */
89149b5e25fSSatish Balay       i++;
89249b5e25fSSatish Balay     } else { /* Begining of the block, so check if the complete block exists */
89349b5e25fSSatish Balay       flg = PETSC_TRUE;
89449b5e25fSSatish Balay       for (k=1; k<bs; k++) {
89549b5e25fSSatish Balay         if (row+k != idx[i+k]) { /* break in the block */
89649b5e25fSSatish Balay           flg = PETSC_FALSE;
89749b5e25fSSatish Balay           break;
89849b5e25fSSatish Balay         }
89949b5e25fSSatish Balay       }
900abc0a331SBarry Smith       if (flg) { /* No break in the bs */
90149b5e25fSSatish Balay         sizes[j] = bs;
90249b5e25fSSatish Balay         i       += bs;
90349b5e25fSSatish Balay       } else {
90449b5e25fSSatish Balay         sizes[j] = 1;
90549b5e25fSSatish Balay         i++;
90649b5e25fSSatish Balay       }
90749b5e25fSSatish Balay     }
90849b5e25fSSatish Balay   }
90949b5e25fSSatish Balay   *bs_max = j;
91049b5e25fSSatish Balay   PetscFunctionReturn(0);
91149b5e25fSSatish Balay }
91249b5e25fSSatish Balay 
91349b5e25fSSatish Balay 
91449b5e25fSSatish Balay /* Only add/insert a(i,j) with i<=j (blocks).
91549b5e25fSSatish Balay    Any a(i,j) with i>j input by user is ingored.
91649b5e25fSSatish Balay */
91749b5e25fSSatish Balay 
91813f74950SBarry Smith PetscErrorCode MatSetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is)
91949b5e25fSSatish Balay {
92049b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
9216849ba73SBarry Smith   PetscErrorCode ierr;
922e2ee6c50SBarry Smith   PetscInt       *rp,k,low,high,t,ii,row,nrow,i,col,l,rmax,N,lastcol = -1;
92313f74950SBarry Smith   PetscInt       *imax=a->imax,*ai=a->i,*ailen=a->ilen,roworiented=a->roworiented;
924d0f46423SBarry Smith   PetscInt       *aj  =a->j,nonew=a->nonew,bs=A->rmap->bs,brow,bcol;
92513f74950SBarry Smith   PetscInt       ridx,cidx,bs2=a->bs2;
92649b5e25fSSatish Balay   MatScalar      *ap,value,*aa=a->a,*bap;
92749b5e25fSSatish Balay 
92849b5e25fSSatish Balay   PetscFunctionBegin;
92949b5e25fSSatish Balay   for (k=0; k<m; k++) { /* loop over added rows */
93049b5e25fSSatish Balay     row  = im[k];       /* row number */
93149b5e25fSSatish Balay     brow = row/bs;      /* block row number */
93249b5e25fSSatish Balay     if (row < 0) continue;
9332515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
934e32f2f54SBarry 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);
93549b5e25fSSatish Balay #endif
93649b5e25fSSatish Balay     rp   = aj + ai[brow]; /*ptr to beginning of column value of the row block*/
93749b5e25fSSatish Balay     ap   = aa + bs2*ai[brow]; /*ptr to beginning of element value of the row block*/
93849b5e25fSSatish Balay     rmax = imax[brow];  /* maximum space allocated for this row */
93949b5e25fSSatish Balay     nrow = ailen[brow]; /* actual length of this row */
94049b5e25fSSatish Balay     low  = 0;
94149b5e25fSSatish Balay 
94249b5e25fSSatish Balay     for (l=0; l<n; l++) { /* loop over added columns */
94349b5e25fSSatish Balay       if (in[l] < 0) continue;
9442515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
945e32f2f54SBarry 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);
94649b5e25fSSatish Balay #endif
94749b5e25fSSatish Balay       col  = in[l];
94849b5e25fSSatish Balay       bcol = col/bs;              /* block col number */
94949b5e25fSSatish Balay 
950941593c8SHong Zhang       if (brow > bcol) {
95126fbe8dcSKarl Rupp         if (a->ignore_ltriangular) continue; /* ignore lower triangular values */
95226fbe8dcSKarl Rupp         else 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)");
953941593c8SHong Zhang       }
954f4989cb3SHong Zhang 
95549b5e25fSSatish Balay       ridx = row % bs; cidx = col % bs; /*row and col index inside the block */
9568549e402SHong Zhang       if ((brow==bcol && ridx<=cidx) || (brow<bcol)) {
95749b5e25fSSatish Balay         /* element value a(k,l) */
95826fbe8dcSKarl Rupp         if (roworiented) value = v[l + k*n];
95926fbe8dcSKarl Rupp         else value = v[k + l*m];
96049b5e25fSSatish Balay 
96149b5e25fSSatish Balay         /* move pointer bap to a(k,l) quickly and add/insert value */
96226fbe8dcSKarl Rupp         if (col <= lastcol) low = 0;
96326fbe8dcSKarl Rupp         high = nrow;
964e2ee6c50SBarry Smith         lastcol = col;
96549b5e25fSSatish Balay         while (high-low > 7) {
96649b5e25fSSatish Balay           t = (low+high)/2;
96749b5e25fSSatish Balay           if (rp[t] > bcol) high = t;
96849b5e25fSSatish Balay           else              low  = t;
96949b5e25fSSatish Balay         }
97049b5e25fSSatish Balay         for (i=low; i<high; i++) {
97149b5e25fSSatish Balay           if (rp[i] > bcol) break;
97249b5e25fSSatish Balay           if (rp[i] == bcol) {
97349b5e25fSSatish Balay             bap = ap +  bs2*i + bs*cidx + ridx;
97449b5e25fSSatish Balay             if (is == ADD_VALUES) *bap += value;
97549b5e25fSSatish Balay             else                  *bap  = value;
9768549e402SHong Zhang             /* for diag block, add/insert its symmetric element a(cidx,ridx) */
9778549e402SHong Zhang             if (brow == bcol && ridx < cidx) {
9788549e402SHong Zhang               bap = ap +  bs2*i + bs*ridx + cidx;
9798549e402SHong Zhang               if (is == ADD_VALUES) *bap += value;
9808549e402SHong Zhang               else                  *bap  = value;
9818549e402SHong Zhang             }
98249b5e25fSSatish Balay             goto noinsert1;
98349b5e25fSSatish Balay           }
98449b5e25fSSatish Balay         }
98549b5e25fSSatish Balay 
98649b5e25fSSatish Balay         if (nonew == 1) goto noinsert1;
987e32f2f54SBarry Smith         if (nonew == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col);
988fef13f97SBarry Smith         MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,brow,bcol,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar);
98949b5e25fSSatish Balay 
990c03d1d03SSatish Balay         N = nrow++ - 1; high++;
99149b5e25fSSatish Balay         /* shift up all the later entries in this row */
99249b5e25fSSatish Balay         for (ii=N; ii>=i; ii--) {
99349b5e25fSSatish Balay           rp[ii+1] = rp[ii];
99449b5e25fSSatish Balay           ierr     = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr);
99549b5e25fSSatish Balay         }
99649b5e25fSSatish Balay         if (N>=i) {
99749b5e25fSSatish Balay           ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr);
99849b5e25fSSatish Balay         }
99949b5e25fSSatish Balay         rp[i]                      = bcol;
100049b5e25fSSatish Balay         ap[bs2*i + bs*cidx + ridx] = value;
1001e56f5c9eSBarry Smith         A->nonzerostate++;
100249b5e25fSSatish Balay noinsert1:;
100349b5e25fSSatish Balay         low = i;
10048549e402SHong Zhang       }
100549b5e25fSSatish Balay     }   /* end of loop over added columns */
100649b5e25fSSatish Balay     ailen[brow] = nrow;
100749b5e25fSSatish Balay   }   /* end of loop over added rows */
100849b5e25fSSatish Balay   PetscFunctionReturn(0);
100949b5e25fSSatish Balay }
101049b5e25fSSatish Balay 
10110481f469SBarry Smith PetscErrorCode MatICCFactor_SeqSBAIJ(Mat inA,IS row,const MatFactorInfo *info)
101249b5e25fSSatish Balay {
10134ccecd49SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)inA->data;
101449b5e25fSSatish Balay   Mat            outA;
1015dfbe8321SBarry Smith   PetscErrorCode ierr;
1016ace3abfcSBarry Smith   PetscBool      row_identity;
101749b5e25fSSatish Balay 
101849b5e25fSSatish Balay   PetscFunctionBegin;
1019e32f2f54SBarry Smith   if (info->levels != 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only levels=0 is supported for in-place icc");
1020c84f5b01SHong Zhang   ierr = ISIdentity(row,&row_identity);CHKERRQ(ierr);
1021e32f2f54SBarry Smith   if (!row_identity) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix reordering is not supported");
1022e32f2f54SBarry 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()! */
1023c84f5b01SHong Zhang 
102449b5e25fSSatish Balay   outA            = inA;
1025d5f3da31SBarry Smith   inA->factortype = MAT_FACTOR_ICC;
1026f6224b95SHong Zhang   ierr = PetscFree(inA->solvertype);CHKERRQ(ierr);
1027f6224b95SHong Zhang   ierr = PetscStrallocpy(MATSOLVERPETSC,&inA->solvertype);CHKERRQ(ierr);
102849b5e25fSSatish Balay 
10291a3463dfSHong Zhang   ierr = MatMarkDiagonal_SeqSBAIJ(inA);CHKERRQ(ierr);
1030d595f711SHong Zhang   ierr = MatSeqSBAIJSetNumericFactorization_inplace(inA,row_identity);CHKERRQ(ierr);
103149b5e25fSSatish Balay 
1032c3122656SLisandro Dalcin   ierr   = PetscObjectReference((PetscObject)row);CHKERRQ(ierr);
10336bf464f9SBarry Smith   ierr   = ISDestroy(&a->row);CHKERRQ(ierr);
1034c84f5b01SHong Zhang   a->row = row;
1035c3122656SLisandro Dalcin   ierr   = PetscObjectReference((PetscObject)row);CHKERRQ(ierr);
10366bf464f9SBarry Smith   ierr   = ISDestroy(&a->col);CHKERRQ(ierr);
1037c84f5b01SHong Zhang   a->col = row;
1038c84f5b01SHong Zhang 
1039c84f5b01SHong Zhang   /* Create the invert permutation so that it can be used in MatCholeskyFactorNumeric() */
1040c84f5b01SHong Zhang   if (a->icol) {ierr = ISInvertPermutation(row,PETSC_DECIDE, &a->icol);CHKERRQ(ierr);}
10413bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)inA,(PetscObject)a->icol);CHKERRQ(ierr);
104249b5e25fSSatish Balay 
104349b5e25fSSatish Balay   if (!a->solve_work) {
1044854ce69bSBarry Smith     ierr = PetscMalloc1(inA->rmap->N+inA->rmap->bs,&a->solve_work);CHKERRQ(ierr);
10453bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)inA,(inA->rmap->N+inA->rmap->bs)*sizeof(PetscScalar));CHKERRQ(ierr);
104649b5e25fSSatish Balay   }
104749b5e25fSSatish Balay 
1048719d5645SBarry Smith   ierr = MatCholeskyFactorNumeric(outA,inA,info);CHKERRQ(ierr);
104949b5e25fSSatish Balay   PetscFunctionReturn(0);
105049b5e25fSSatish Balay }
1051950f1e5bSHong Zhang 
10527087cfbeSBarry Smith PetscErrorCode  MatSeqSBAIJSetColumnIndices_SeqSBAIJ(Mat mat,PetscInt *indices)
105349b5e25fSSatish Balay {
1054045c9aa0SHong Zhang   Mat_SeqSBAIJ   *baij = (Mat_SeqSBAIJ*)mat->data;
105513f74950SBarry Smith   PetscInt       i,nz,n;
10567827cd58SJed Brown   PetscErrorCode ierr;
105749b5e25fSSatish Balay 
105849b5e25fSSatish Balay   PetscFunctionBegin;
10596c6c5352SBarry Smith   nz = baij->maxnz;
1060d0f46423SBarry Smith   n  = mat->cmap->n;
106126fbe8dcSKarl Rupp   for (i=0; i<nz; i++) baij->j[i] = indices[i];
106226fbe8dcSKarl Rupp 
10636c6c5352SBarry Smith   baij->nz = nz;
106426fbe8dcSKarl Rupp   for (i=0; i<n; i++) baij->ilen[i] = baij->imax[i];
106526fbe8dcSKarl Rupp 
10667827cd58SJed Brown   ierr = MatSetOption(mat,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
106749b5e25fSSatish Balay   PetscFunctionReturn(0);
106849b5e25fSSatish Balay }
106949b5e25fSSatish Balay 
107049b5e25fSSatish Balay /*@
107119585528SSatish Balay   MatSeqSBAIJSetColumnIndices - Set the column indices for all the rows
107249b5e25fSSatish Balay   in the matrix.
107349b5e25fSSatish Balay 
107449b5e25fSSatish Balay   Input Parameters:
107519585528SSatish Balay   +  mat     - the SeqSBAIJ matrix
107649b5e25fSSatish Balay   -  indices - the column indices
107749b5e25fSSatish Balay 
107849b5e25fSSatish Balay   Level: advanced
107949b5e25fSSatish Balay 
108049b5e25fSSatish Balay   Notes:
108149b5e25fSSatish Balay   This can be called if you have precomputed the nonzero structure of the
108249b5e25fSSatish Balay   matrix and want to provide it to the matrix object to improve the performance
108349b5e25fSSatish Balay   of the MatSetValues() operation.
108449b5e25fSSatish Balay 
108549b5e25fSSatish Balay   You MUST have set the correct numbers of nonzeros per row in the call to
1086d1be2dadSMatthew Knepley   MatCreateSeqSBAIJ(), and the columns indices MUST be sorted.
108749b5e25fSSatish Balay 
1088ab9f2c04SSatish Balay   MUST be called before any calls to MatSetValues()
108949b5e25fSSatish Balay 
1090ab9f2c04SSatish Balay   .seealso: MatCreateSeqSBAIJ
109149b5e25fSSatish Balay @*/
10927087cfbeSBarry Smith PetscErrorCode  MatSeqSBAIJSetColumnIndices(Mat mat,PetscInt *indices)
109349b5e25fSSatish Balay {
10944ac538c5SBarry Smith   PetscErrorCode ierr;
109549b5e25fSSatish Balay 
109649b5e25fSSatish Balay   PetscFunctionBegin;
10970700a824SBarry Smith   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
10984482741eSBarry Smith   PetscValidPointer(indices,2);
10994ac538c5SBarry Smith   ierr = PetscUseMethod(mat,"MatSeqSBAIJSetColumnIndices_C",(Mat,PetscInt*),(mat,indices));CHKERRQ(ierr);
110049b5e25fSSatish Balay   PetscFunctionReturn(0);
110149b5e25fSSatish Balay }
110249b5e25fSSatish Balay 
11033c896bc6SHong Zhang PetscErrorCode MatCopy_SeqSBAIJ(Mat A,Mat B,MatStructure str)
11043c896bc6SHong Zhang {
11053c896bc6SHong Zhang   PetscErrorCode ierr;
11063c896bc6SHong Zhang 
11073c896bc6SHong Zhang   PetscFunctionBegin;
11083c896bc6SHong Zhang   /* If the two matrices have the same copy implementation, use fast copy. */
11093c896bc6SHong Zhang   if (str == SAME_NONZERO_PATTERN && (A->ops->copy == B->ops->copy)) {
11103c896bc6SHong Zhang     Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
11113c896bc6SHong Zhang     Mat_SeqSBAIJ *b = (Mat_SeqSBAIJ*)B->data;
11123c896bc6SHong Zhang 
1113e7e72b3dSBarry 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");
1114d0f46423SBarry Smith     ierr = PetscMemcpy(b->a,a->a,(a->i[A->rmap->N])*sizeof(PetscScalar));CHKERRQ(ierr);
11153c896bc6SHong Zhang   } else {
1116f5edf698SHong Zhang     ierr = MatGetRowUpperTriangular(A);CHKERRQ(ierr);
11173c896bc6SHong Zhang     ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr);
1118f5edf698SHong Zhang     ierr = MatRestoreRowUpperTriangular(A);CHKERRQ(ierr);
11193c896bc6SHong Zhang   }
11203c896bc6SHong Zhang   PetscFunctionReturn(0);
11213c896bc6SHong Zhang }
11223c896bc6SHong Zhang 
11234994cf47SJed Brown PetscErrorCode MatSetUp_SeqSBAIJ(Mat A)
1124273d9f13SBarry Smith {
1125dfbe8321SBarry Smith   PetscErrorCode ierr;
1126273d9f13SBarry Smith 
1127273d9f13SBarry Smith   PetscFunctionBegin;
1128367daffbSBarry Smith   ierr = MatSeqSBAIJSetPreallocation(A,A->rmap->bs,PETSC_DEFAULT,0);CHKERRQ(ierr);
1129273d9f13SBarry Smith   PetscFunctionReturn(0);
1130273d9f13SBarry Smith }
1131273d9f13SBarry Smith 
11328c778c55SBarry Smith PetscErrorCode MatSeqSBAIJGetArray_SeqSBAIJ(Mat A,PetscScalar *array[])
1133a6ece127SHong Zhang {
1134a6ece127SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
11355fd66863SKarl Rupp 
1136a6ece127SHong Zhang   PetscFunctionBegin;
1137a6ece127SHong Zhang   *array = a->a;
1138a6ece127SHong Zhang   PetscFunctionReturn(0);
1139a6ece127SHong Zhang }
1140a6ece127SHong Zhang 
11418c778c55SBarry Smith PetscErrorCode MatSeqSBAIJRestoreArray_SeqSBAIJ(Mat A,PetscScalar *array[])
1142a6ece127SHong Zhang {
1143a6ece127SHong Zhang   PetscFunctionBegin;
1144a6ece127SHong Zhang   PetscFunctionReturn(0);
1145a6ece127SHong Zhang }
1146a6ece127SHong Zhang 
114752768537SHong Zhang PetscErrorCode MatAXPYGetPreallocation_SeqSBAIJ(Mat Y,Mat X,PetscInt *nnz)
114852768537SHong Zhang {
1149b264fe52SHong Zhang   PetscInt       bs = Y->rmap->bs,mbs = Y->rmap->N/bs;
115052768537SHong Zhang   Mat_SeqSBAIJ   *x = (Mat_SeqSBAIJ*)X->data;
115152768537SHong Zhang   Mat_SeqSBAIJ   *y = (Mat_SeqSBAIJ*)Y->data;
1152b264fe52SHong Zhang   PetscErrorCode ierr;
115352768537SHong Zhang 
115452768537SHong Zhang   PetscFunctionBegin;
115552768537SHong Zhang   /* Set the number of nonzeros in the new matrix */
1156b264fe52SHong Zhang   ierr = MatAXPYGetPreallocation_SeqX_private(mbs,x->i,x->j,y->i,y->j,nnz);CHKERRQ(ierr);
115752768537SHong Zhang   PetscFunctionReturn(0);
115852768537SHong Zhang }
115952768537SHong Zhang 
1160f4df32b1SMatthew Knepley PetscErrorCode MatAXPY_SeqSBAIJ(Mat Y,PetscScalar a,Mat X,MatStructure str)
116142ee4b1aSHong Zhang {
116242ee4b1aSHong Zhang   Mat_SeqSBAIJ   *x=(Mat_SeqSBAIJ*)X->data, *y=(Mat_SeqSBAIJ*)Y->data;
1163dfbe8321SBarry Smith   PetscErrorCode ierr;
116431ce2d13SHong Zhang   PetscInt       bs=Y->rmap->bs,bs2=bs*bs;
1165e838b9e7SJed Brown   PetscBLASInt   one = 1;
116642ee4b1aSHong Zhang 
116742ee4b1aSHong Zhang   PetscFunctionBegin;
116842ee4b1aSHong Zhang   if (str == SAME_NONZERO_PATTERN) {
1169f4df32b1SMatthew Knepley     PetscScalar  alpha = a;
1170c5df96a5SBarry Smith     PetscBLASInt bnz;
1171c5df96a5SBarry Smith     ierr = PetscBLASIntCast(x->nz*bs2,&bnz);CHKERRQ(ierr);
11728b83055fSJed Brown     PetscStackCallBLAS("BLASaxpy",BLASaxpy_(&bnz,&alpha,x->a,&one,y->a,&one));
1173a3fa217bSJose E. Roman     ierr = PetscObjectStateIncrease((PetscObject)Y);CHKERRQ(ierr);
1174ab784542SHong Zhang   } else if (str == SUBSET_NONZERO_PATTERN) { /* nonzeros of X is a subset of Y's */
1175ab784542SHong Zhang     ierr = MatSetOption(X,MAT_GETROW_UPPERTRIANGULAR,PETSC_TRUE);CHKERRQ(ierr);
1176ab784542SHong Zhang     ierr = MatAXPY_Basic(Y,a,X,str);CHKERRQ(ierr);
1177ab784542SHong Zhang     ierr = MatSetOption(X,MAT_GETROW_UPPERTRIANGULAR,PETSC_FALSE);CHKERRQ(ierr);
117842ee4b1aSHong Zhang   } else {
117952768537SHong Zhang     Mat      B;
118052768537SHong Zhang     PetscInt *nnz;
118152768537SHong Zhang     if (bs != X->rmap->bs) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Matrices must have same block size");
1182f5edf698SHong Zhang     ierr = MatGetRowUpperTriangular(X);CHKERRQ(ierr);
118352768537SHong Zhang     ierr = MatGetRowUpperTriangular(Y);CHKERRQ(ierr);
118452768537SHong Zhang     ierr = PetscMalloc1(Y->rmap->N,&nnz);CHKERRQ(ierr);
118552768537SHong Zhang     ierr = MatCreate(PetscObjectComm((PetscObject)Y),&B);CHKERRQ(ierr);
118652768537SHong Zhang     ierr = PetscObjectSetName((PetscObject)B,((PetscObject)Y)->name);CHKERRQ(ierr);
118752768537SHong Zhang     ierr = MatSetSizes(B,Y->rmap->n,Y->cmap->n,Y->rmap->N,Y->cmap->N);CHKERRQ(ierr);
118852768537SHong Zhang     ierr = MatSetBlockSizesFromMats(B,Y,Y);CHKERRQ(ierr);
118952768537SHong Zhang     ierr = MatSetType(B,(MatType) ((PetscObject)Y)->type_name);CHKERRQ(ierr);
119052768537SHong Zhang     ierr = MatAXPYGetPreallocation_SeqSBAIJ(Y,X,nnz);CHKERRQ(ierr);
119152768537SHong Zhang     ierr = MatSeqSBAIJSetPreallocation(B,bs,0,nnz);CHKERRQ(ierr);
119252768537SHong Zhang 
119352768537SHong Zhang     ierr = MatAXPY_BasicWithPreallocation(B,Y,a,X,str);CHKERRQ(ierr);
119452768537SHong Zhang 
119528be2f97SBarry Smith     ierr = MatHeaderReplace(Y,&B);CHKERRQ(ierr);
119652768537SHong Zhang     ierr = PetscFree(nnz);CHKERRQ(ierr);
1197f5edf698SHong Zhang     ierr = MatRestoreRowUpperTriangular(X);CHKERRQ(ierr);
119852768537SHong Zhang     ierr = MatRestoreRowUpperTriangular(Y);CHKERRQ(ierr);
119942ee4b1aSHong Zhang   }
120042ee4b1aSHong Zhang   PetscFunctionReturn(0);
120142ee4b1aSHong Zhang }
120242ee4b1aSHong Zhang 
1203ace3abfcSBarry Smith PetscErrorCode MatIsSymmetric_SeqSBAIJ(Mat A,PetscReal tol,PetscBool  *flg)
1204efcf0fc3SBarry Smith {
1205efcf0fc3SBarry Smith   PetscFunctionBegin;
1206efcf0fc3SBarry Smith   *flg = PETSC_TRUE;
1207efcf0fc3SBarry Smith   PetscFunctionReturn(0);
1208efcf0fc3SBarry Smith }
1209efcf0fc3SBarry Smith 
1210ace3abfcSBarry Smith PetscErrorCode MatIsStructurallySymmetric_SeqSBAIJ(Mat A,PetscBool  *flg)
1211efcf0fc3SBarry Smith {
1212efcf0fc3SBarry Smith   PetscFunctionBegin;
1213efcf0fc3SBarry Smith   *flg = PETSC_TRUE;
1214efcf0fc3SBarry Smith   PetscFunctionReturn(0);
1215efcf0fc3SBarry Smith }
1216efcf0fc3SBarry Smith 
1217ace3abfcSBarry Smith PetscErrorCode MatIsHermitian_SeqSBAIJ(Mat A,PetscReal tol,PetscBool  *flg)
1218efcf0fc3SBarry Smith {
1219efcf0fc3SBarry Smith   PetscFunctionBegin;
1220efcf0fc3SBarry Smith   *flg = PETSC_FALSE;
1221efcf0fc3SBarry Smith   PetscFunctionReturn(0);
1222efcf0fc3SBarry Smith }
1223efcf0fc3SBarry Smith 
122499cafbc1SBarry Smith PetscErrorCode MatRealPart_SeqSBAIJ(Mat A)
122599cafbc1SBarry Smith {
122699cafbc1SBarry Smith   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
122799cafbc1SBarry Smith   PetscInt     i,nz = a->bs2*a->i[a->mbs];
1228dd6ea824SBarry Smith   MatScalar    *aa = a->a;
122999cafbc1SBarry Smith 
123099cafbc1SBarry Smith   PetscFunctionBegin;
123199cafbc1SBarry Smith   for (i=0; i<nz; i++) aa[i] = PetscRealPart(aa[i]);
123299cafbc1SBarry Smith   PetscFunctionReturn(0);
123399cafbc1SBarry Smith }
123499cafbc1SBarry Smith 
123599cafbc1SBarry Smith PetscErrorCode MatImaginaryPart_SeqSBAIJ(Mat A)
123699cafbc1SBarry Smith {
123799cafbc1SBarry Smith   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
123899cafbc1SBarry Smith   PetscInt     i,nz = a->bs2*a->i[a->mbs];
1239dd6ea824SBarry Smith   MatScalar    *aa = a->a;
124099cafbc1SBarry Smith 
124199cafbc1SBarry Smith   PetscFunctionBegin;
124299cafbc1SBarry Smith   for (i=0; i<nz; i++) aa[i] = PetscImaginaryPart(aa[i]);
124399cafbc1SBarry Smith   PetscFunctionReturn(0);
124499cafbc1SBarry Smith }
124599cafbc1SBarry Smith 
12463bededecSBarry Smith PetscErrorCode MatZeroRowsColumns_SeqSBAIJ(Mat A,PetscInt is_n,const PetscInt is_idx[],PetscScalar diag,Vec x, Vec b)
12473bededecSBarry Smith {
12483bededecSBarry Smith   Mat_SeqSBAIJ      *baij=(Mat_SeqSBAIJ*)A->data;
12493bededecSBarry Smith   PetscErrorCode    ierr;
12503bededecSBarry Smith   PetscInt          i,j,k,count;
12513bededecSBarry Smith   PetscInt          bs   =A->rmap->bs,bs2=baij->bs2,row,col;
12523bededecSBarry Smith   PetscScalar       zero = 0.0;
12533bededecSBarry Smith   MatScalar         *aa;
12543bededecSBarry Smith   const PetscScalar *xx;
12553bededecSBarry Smith   PetscScalar       *bb;
125656777dd2SBarry Smith   PetscBool         *zeroed,vecs = PETSC_FALSE;
12573bededecSBarry Smith 
12583bededecSBarry Smith   PetscFunctionBegin;
12593bededecSBarry Smith   /* fix right hand side if needed */
12603bededecSBarry Smith   if (x && b) {
12613bededecSBarry Smith     ierr = VecGetArrayRead(x,&xx);CHKERRQ(ierr);
12623bededecSBarry Smith     ierr = VecGetArray(b,&bb);CHKERRQ(ierr);
126356777dd2SBarry Smith     vecs = PETSC_TRUE;
12643bededecSBarry Smith   }
12653bededecSBarry Smith 
12663bededecSBarry Smith   /* zero the columns */
12671795a4d1SJed Brown   ierr = PetscCalloc1(A->rmap->n,&zeroed);CHKERRQ(ierr);
12683bededecSBarry Smith   for (i=0; i<is_n; i++) {
12693bededecSBarry 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]);
12703bededecSBarry Smith     zeroed[is_idx[i]] = PETSC_TRUE;
12713bededecSBarry Smith   }
127256777dd2SBarry Smith   if (vecs) {
127356777dd2SBarry Smith     for (i=0; i<A->rmap->N; i++) {
127456777dd2SBarry Smith       row = i/bs;
127556777dd2SBarry Smith       for (j=baij->i[row]; j<baij->i[row+1]; j++) {
127656777dd2SBarry Smith         for (k=0; k<bs; k++) {
127756777dd2SBarry Smith           col = bs*baij->j[j] + k;
127856777dd2SBarry Smith           if (col <= i) continue;
127956777dd2SBarry Smith           aa = ((MatScalar*)(baij->a)) + j*bs2 + (i%bs) + bs*k;
128026fbe8dcSKarl Rupp           if (!zeroed[i] && zeroed[col]) bb[i]   -= aa[0]*xx[col];
128126fbe8dcSKarl Rupp           if (zeroed[i] && !zeroed[col]) bb[col] -= aa[0]*xx[i];
128256777dd2SBarry Smith         }
128356777dd2SBarry Smith       }
128456777dd2SBarry Smith     }
128526fbe8dcSKarl Rupp     for (i=0; i<is_n; i++) bb[is_idx[i]] = diag*xx[is_idx[i]];
128656777dd2SBarry Smith   }
128756777dd2SBarry Smith 
12883bededecSBarry Smith   for (i=0; i<A->rmap->N; i++) {
12893bededecSBarry Smith     if (!zeroed[i]) {
12903bededecSBarry Smith       row = i/bs;
12913bededecSBarry Smith       for (j=baij->i[row]; j<baij->i[row+1]; j++) {
12923bededecSBarry Smith         for (k=0; k<bs; k++) {
12933bededecSBarry Smith           col = bs*baij->j[j] + k;
12943bededecSBarry Smith           if (zeroed[col]) {
12953bededecSBarry Smith             aa = ((MatScalar*)(baij->a)) + j*bs2 + (i%bs) + bs*k;
12963bededecSBarry Smith             aa[0] = 0.0;
12973bededecSBarry Smith           }
12983bededecSBarry Smith         }
12993bededecSBarry Smith       }
13003bededecSBarry Smith     }
13013bededecSBarry Smith   }
13023bededecSBarry Smith   ierr = PetscFree(zeroed);CHKERRQ(ierr);
130356777dd2SBarry Smith   if (vecs) {
130456777dd2SBarry Smith     ierr = VecRestoreArrayRead(x,&xx);CHKERRQ(ierr);
130556777dd2SBarry Smith     ierr = VecRestoreArray(b,&bb);CHKERRQ(ierr);
130656777dd2SBarry Smith   }
13073bededecSBarry Smith 
13083bededecSBarry Smith   /* zero the rows */
13093bededecSBarry Smith   for (i=0; i<is_n; i++) {
13103bededecSBarry Smith     row   = is_idx[i];
13113bededecSBarry Smith     count = (baij->i[row/bs +1] - baij->i[row/bs])*bs;
13123bededecSBarry Smith     aa    = ((MatScalar*)(baij->a)) + baij->i[row/bs]*bs2 + (row%bs);
13133bededecSBarry Smith     for (k=0; k<count; k++) {
13143bededecSBarry Smith       aa[0] =  zero;
13153bededecSBarry Smith       aa   += bs;
13163bededecSBarry Smith     }
13173bededecSBarry Smith     if (diag != 0.0) {
13183bededecSBarry Smith       ierr = (*A->ops->setvalues)(A,1,&row,1,&row,&diag,INSERT_VALUES);CHKERRQ(ierr);
13193bededecSBarry Smith     }
13203bededecSBarry Smith   }
13213bededecSBarry Smith   ierr = MatAssemblyEnd_SeqSBAIJ(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
13223bededecSBarry Smith   PetscFunctionReturn(0);
13233bededecSBarry Smith }
13243bededecSBarry Smith 
13257d68702bSBarry Smith PetscErrorCode MatShift_SeqSBAIJ(Mat Y,PetscScalar a)
13267d68702bSBarry Smith {
13277d68702bSBarry Smith   PetscErrorCode ierr;
13287d68702bSBarry Smith   Mat_SeqSBAIJ    *aij = (Mat_SeqSBAIJ*)Y->data;
13297d68702bSBarry Smith 
13307d68702bSBarry Smith   PetscFunctionBegin;
13316f33a894SBarry Smith   if (!Y->preallocated || !aij->nz) {
13327d68702bSBarry Smith     ierr = MatSeqSBAIJSetPreallocation(Y,Y->rmap->bs,1,NULL);CHKERRQ(ierr);
13337d68702bSBarry Smith   }
13347d68702bSBarry Smith   ierr = MatShift_Basic(Y,a);CHKERRQ(ierr);
13357d68702bSBarry Smith   PetscFunctionReturn(0);
13367d68702bSBarry Smith }
13377d68702bSBarry Smith 
133849b5e25fSSatish Balay /* -------------------------------------------------------------------*/
13393964eb88SJed Brown static struct _MatOps MatOps_Values = {MatSetValues_SeqSBAIJ,
134049b5e25fSSatish Balay                                        MatGetRow_SeqSBAIJ,
134149b5e25fSSatish Balay                                        MatRestoreRow_SeqSBAIJ,
134249b5e25fSSatish Balay                                        MatMult_SeqSBAIJ_N,
134397304618SKris Buschelman                                /*  4*/ MatMultAdd_SeqSBAIJ_N,
1344431c96f7SBarry Smith                                        MatMult_SeqSBAIJ_N,       /* transpose versions are same as non-transpose versions */
1345e005ede5SBarry Smith                                        MatMultAdd_SeqSBAIJ_N,
1346db4efbfdSBarry Smith                                        0,
134749b5e25fSSatish Balay                                        0,
134849b5e25fSSatish Balay                                        0,
134997304618SKris Buschelman                                /* 10*/ 0,
135049b5e25fSSatish Balay                                        0,
1351c078aec8SLisandro Dalcin                                        MatCholeskyFactor_SeqSBAIJ,
135241f059aeSBarry Smith                                        MatSOR_SeqSBAIJ,
135349b5e25fSSatish Balay                                        MatTranspose_SeqSBAIJ,
135497304618SKris Buschelman                                /* 15*/ MatGetInfo_SeqSBAIJ,
135549b5e25fSSatish Balay                                        MatEqual_SeqSBAIJ,
135649b5e25fSSatish Balay                                        MatGetDiagonal_SeqSBAIJ,
135749b5e25fSSatish Balay                                        MatDiagonalScale_SeqSBAIJ,
135849b5e25fSSatish Balay                                        MatNorm_SeqSBAIJ,
135997304618SKris Buschelman                                /* 20*/ 0,
136049b5e25fSSatish Balay                                        MatAssemblyEnd_SeqSBAIJ,
136149b5e25fSSatish Balay                                        MatSetOption_SeqSBAIJ,
136249b5e25fSSatish Balay                                        MatZeroEntries_SeqSBAIJ,
1363d519adbfSMatthew Knepley                                /* 24*/ 0,
136449b5e25fSSatish Balay                                        0,
136549b5e25fSSatish Balay                                        0,
1366db4efbfdSBarry Smith                                        0,
1367db4efbfdSBarry Smith                                        0,
13684994cf47SJed Brown                                /* 29*/ MatSetUp_SeqSBAIJ,
1369c464158bSHong Zhang                                        0,
1370db4efbfdSBarry Smith                                        0,
13718c778c55SBarry Smith                                        0,
13728c778c55SBarry Smith                                        0,
1373d519adbfSMatthew Knepley                                /* 34*/ MatDuplicate_SeqSBAIJ,
1374719d5645SBarry Smith                                        0,
1375719d5645SBarry Smith                                        0,
137649b5e25fSSatish Balay                                        0,
1377c84f5b01SHong Zhang                                        MatICCFactor_SeqSBAIJ,
1378d519adbfSMatthew Knepley                                /* 39*/ MatAXPY_SeqSBAIJ,
137949b5e25fSSatish Balay                                        MatGetSubMatrices_SeqSBAIJ,
138049b5e25fSSatish Balay                                        MatIncreaseOverlap_SeqSBAIJ,
138149b5e25fSSatish Balay                                        MatGetValues_SeqSBAIJ,
13823c896bc6SHong Zhang                                        MatCopy_SeqSBAIJ,
1383d519adbfSMatthew Knepley                                /* 44*/ 0,
138449b5e25fSSatish Balay                                        MatScale_SeqSBAIJ,
13857d68702bSBarry Smith                                        MatShift_SeqSBAIJ,
138649b5e25fSSatish Balay                                        0,
13873bededecSBarry Smith                                        MatZeroRowsColumns_SeqSBAIJ,
1388f73d5cc4SBarry Smith                                /* 49*/ 0,
138949b5e25fSSatish Balay                                        MatGetRowIJ_SeqSBAIJ,
139049b5e25fSSatish Balay                                        MatRestoreRowIJ_SeqSBAIJ,
139149b5e25fSSatish Balay                                        0,
139249b5e25fSSatish Balay                                        0,
1393d519adbfSMatthew Knepley                                /* 54*/ 0,
139449b5e25fSSatish Balay                                        0,
139549b5e25fSSatish Balay                                        0,
139649b5e25fSSatish Balay                                        0,
139749b5e25fSSatish Balay                                        MatSetValuesBlocked_SeqSBAIJ,
1398d519adbfSMatthew Knepley                                /* 59*/ MatGetSubMatrix_SeqSBAIJ,
139949b5e25fSSatish Balay                                        0,
140049b5e25fSSatish Balay                                        0,
1401357abbc8SBarry Smith                                        0,
1402d959ec07SHong Zhang                                        0,
1403d519adbfSMatthew Knepley                                /* 64*/ 0,
1404d959ec07SHong Zhang                                        0,
1405d959ec07SHong Zhang                                        0,
1406d959ec07SHong Zhang                                        0,
1407d959ec07SHong Zhang                                        0,
1408d519adbfSMatthew Knepley                                /* 69*/ MatGetRowMaxAbs_SeqSBAIJ,
14093e0d88b5SBarry Smith                                        0,
14103e0d88b5SBarry Smith                                        0,
14113e0d88b5SBarry Smith                                        0,
14123e0d88b5SBarry Smith                                        0,
1413d519adbfSMatthew Knepley                                /* 74*/ 0,
14143e0d88b5SBarry Smith                                        0,
14153e0d88b5SBarry Smith                                        0,
14163e0d88b5SBarry Smith                                        0,
14173e0d88b5SBarry Smith                                        0,
1418d519adbfSMatthew Knepley                                /* 79*/ 0,
14193e0d88b5SBarry Smith                                        0,
14203e0d88b5SBarry Smith                                        0,
142197304618SKris Buschelman                                        MatGetInertia_SeqSBAIJ,
14225bba2384SShri Abhyankar                                        MatLoad_SeqSBAIJ,
1423d519adbfSMatthew Knepley                                /* 84*/ MatIsSymmetric_SeqSBAIJ,
1424865e5f61SKris Buschelman                                        MatIsHermitian_SeqSBAIJ,
1425efcf0fc3SBarry Smith                                        MatIsStructurallySymmetric_SeqSBAIJ,
1426865e5f61SKris Buschelman                                        0,
1427865e5f61SKris Buschelman                                        0,
1428d519adbfSMatthew Knepley                                /* 89*/ 0,
1429865e5f61SKris Buschelman                                        0,
1430865e5f61SKris Buschelman                                        0,
1431865e5f61SKris Buschelman                                        0,
1432865e5f61SKris Buschelman                                        0,
1433d519adbfSMatthew Knepley                                /* 94*/ 0,
1434865e5f61SKris Buschelman                                        0,
1435865e5f61SKris Buschelman                                        0,
143699cafbc1SBarry Smith                                        0,
143799cafbc1SBarry Smith                                        0,
1438d519adbfSMatthew Knepley                                /* 99*/ 0,
143999cafbc1SBarry Smith                                        0,
144099cafbc1SBarry Smith                                        0,
144199cafbc1SBarry Smith                                        0,
144299cafbc1SBarry Smith                                        0,
1443d519adbfSMatthew Knepley                                /*104*/ 0,
144499cafbc1SBarry Smith                                        MatRealPart_SeqSBAIJ,
1445f5edf698SHong Zhang                                        MatImaginaryPart_SeqSBAIJ,
1446f5edf698SHong Zhang                                        MatGetRowUpperTriangular_SeqSBAIJ,
14472af78befSBarry Smith                                        MatRestoreRowUpperTriangular_SeqSBAIJ,
1448d519adbfSMatthew Knepley                                /*109*/ 0,
14492af78befSBarry Smith                                        0,
14502af78befSBarry Smith                                        0,
14512af78befSBarry Smith                                        0,
1452547795f9SHong Zhang                                        MatMissingDiagonal_SeqSBAIJ,
1453547795f9SHong Zhang                                /*114*/ 0,
1454547795f9SHong Zhang                                        0,
1455547795f9SHong Zhang                                        0,
1456547795f9SHong Zhang                                        0,
1457547795f9SHong Zhang                                        0,
1458547795f9SHong Zhang                                /*119*/ 0,
1459547795f9SHong Zhang                                        0,
14602f480046SShri Abhyankar                                        0,
14613964eb88SJed Brown                                        0,
14623964eb88SJed Brown                                        0,
14633964eb88SJed Brown                                /*124*/ 0,
14643964eb88SJed Brown                                        0,
14653964eb88SJed Brown                                        0,
14663964eb88SJed Brown                                        0,
14673964eb88SJed Brown                                        0,
14683964eb88SJed Brown                                /*129*/ 0,
14693964eb88SJed Brown                                        0,
14703964eb88SJed Brown                                        0,
14713964eb88SJed Brown                                        0,
14723964eb88SJed Brown                                        0,
14733964eb88SJed Brown                                /*134*/ 0,
14743964eb88SJed Brown                                        0,
14753964eb88SJed Brown                                        0,
14763964eb88SJed Brown                                        0,
14773964eb88SJed Brown                                        0,
147846533700Sstefano_zampini                                /*139*/ MatSetBlockSizes_Default,
1479f9426fe0SMark Adams                                        0,
148059f5e6ceSHong Zhang                                        0,
148159f5e6ceSHong Zhang                                        0,
148259f5e6ceSHong Zhang                                        0,
148359f5e6ceSHong Zhang                                 /*144*/MatCreateMPIMatConcatenateSeqMat_SeqSBAIJ
148499cafbc1SBarry Smith };
1485be1d678aSKris Buschelman 
14867087cfbeSBarry Smith PetscErrorCode  MatStoreValues_SeqSBAIJ(Mat mat)
148749b5e25fSSatish Balay {
14884afc71dfSHong Zhang   Mat_SeqSBAIJ   *aij = (Mat_SeqSBAIJ*)mat->data;
1489d0f46423SBarry Smith   PetscInt       nz   = aij->i[mat->rmap->N]*mat->rmap->bs*aij->bs2;
1490dfbe8321SBarry Smith   PetscErrorCode ierr;
149149b5e25fSSatish Balay 
149249b5e25fSSatish Balay   PetscFunctionBegin;
1493e7e72b3dSBarry Smith   if (aij->nonew != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first");
149449b5e25fSSatish Balay 
149549b5e25fSSatish Balay   /* allocate space for values if not already there */
149649b5e25fSSatish Balay   if (!aij->saved_values) {
1497854ce69bSBarry Smith     ierr = PetscMalloc1(nz+1,&aij->saved_values);CHKERRQ(ierr);
149849b5e25fSSatish Balay   }
149949b5e25fSSatish Balay 
150049b5e25fSSatish Balay   /* copy values over */
150187828ca2SBarry Smith   ierr = PetscMemcpy(aij->saved_values,aij->a,nz*sizeof(PetscScalar));CHKERRQ(ierr);
150249b5e25fSSatish Balay   PetscFunctionReturn(0);
150349b5e25fSSatish Balay }
150449b5e25fSSatish Balay 
15057087cfbeSBarry Smith PetscErrorCode  MatRetrieveValues_SeqSBAIJ(Mat mat)
150649b5e25fSSatish Balay {
15074afc71dfSHong Zhang   Mat_SeqSBAIJ   *aij = (Mat_SeqSBAIJ*)mat->data;
15086849ba73SBarry Smith   PetscErrorCode ierr;
1509d0f46423SBarry Smith   PetscInt       nz = aij->i[mat->rmap->N]*mat->rmap->bs*aij->bs2;
151049b5e25fSSatish Balay 
151149b5e25fSSatish Balay   PetscFunctionBegin;
1512e7e72b3dSBarry Smith   if (aij->nonew != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first");
1513e7e72b3dSBarry Smith   if (!aij->saved_values) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatStoreValues(A);first");
151449b5e25fSSatish Balay 
151549b5e25fSSatish Balay   /* copy values over */
151687828ca2SBarry Smith   ierr = PetscMemcpy(aij->a,aij->saved_values,nz*sizeof(PetscScalar));CHKERRQ(ierr);
151749b5e25fSSatish Balay   PetscFunctionReturn(0);
151849b5e25fSSatish Balay }
151949b5e25fSSatish Balay 
1520367daffbSBarry Smith static PetscErrorCode  MatSeqSBAIJSetPreallocation_SeqSBAIJ(Mat B,PetscInt bs,PetscInt nz,PetscInt *nnz)
152149b5e25fSSatish Balay {
1522c464158bSHong Zhang   Mat_SeqSBAIJ   *b = (Mat_SeqSBAIJ*)B->data;
15236849ba73SBarry Smith   PetscErrorCode ierr;
15244dcd73b1SHong Zhang   PetscInt       i,mbs,nbs,bs2;
15252576faa2SJed Brown   PetscBool      skipallocation = PETSC_FALSE,flg = PETSC_FALSE,realalloc = PETSC_FALSE;
152649b5e25fSSatish Balay 
152749b5e25fSSatish Balay   PetscFunctionBegin;
15282576faa2SJed Brown   if (nz >= 0 || nnz) realalloc = PETSC_TRUE;
1529db4efbfdSBarry Smith 
153033d57670SJed Brown   ierr = MatSetBlockSize(B,PetscAbs(bs));CHKERRQ(ierr);
153126283091SBarry Smith   ierr = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr);
153226283091SBarry Smith   ierr = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr);
1533e02043d6SBarry Smith   ierr = PetscLayoutGetBlockSize(B->rmap,&bs);CHKERRQ(ierr);
1534899cda47SBarry Smith 
153521940c7eSstefano_zampini   B->preallocated = PETSC_TRUE;
153621940c7eSstefano_zampini 
1537d0f46423SBarry Smith   mbs = B->rmap->N/bs;
15384dcd73b1SHong Zhang   nbs = B->cmap->n/bs;
153949b5e25fSSatish Balay   bs2 = bs*bs;
154049b5e25fSSatish Balay 
15414dcd73b1SHong Zhang   if (mbs*bs != B->rmap->N || nbs*bs!=B->cmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Number rows, cols must be divisible by blocksize");
154249b5e25fSSatish Balay 
1543ab93d7beSBarry Smith   if (nz == MAT_SKIP_ALLOCATION) {
1544ab93d7beSBarry Smith     skipallocation = PETSC_TRUE;
1545ab93d7beSBarry Smith     nz             = 0;
1546ab93d7beSBarry Smith   }
1547ab93d7beSBarry Smith 
1548435da068SBarry Smith   if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 3;
1549e32f2f54SBarry Smith   if (nz < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"nz cannot be less than 0: value %D",nz);
155049b5e25fSSatish Balay   if (nnz) {
155149b5e25fSSatish Balay     for (i=0; i<mbs; i++) {
1552e32f2f54SBarry 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]);
1553de64b629SHong Zhang       if (nnz[i] > nbs) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"nnz cannot be greater than block row length: local row %D value %D block rowlength %D",i,nnz[i],nbs);
155449b5e25fSSatish Balay     }
155549b5e25fSSatish Balay   }
155649b5e25fSSatish Balay 
1557db4efbfdSBarry Smith   B->ops->mult             = MatMult_SeqSBAIJ_N;
1558db4efbfdSBarry Smith   B->ops->multadd          = MatMultAdd_SeqSBAIJ_N;
1559db4efbfdSBarry Smith   B->ops->multtranspose    = MatMult_SeqSBAIJ_N;
1560db4efbfdSBarry Smith   B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_N;
156126fbe8dcSKarl Rupp 
1562c5929fdfSBarry Smith   ierr  = PetscOptionsGetBool(((PetscObject)B)->options,((PetscObject)B)->prefix,"-mat_no_unroll",&flg,NULL);CHKERRQ(ierr);
156349b5e25fSSatish Balay   if (!flg) {
156449b5e25fSSatish Balay     switch (bs) {
156549b5e25fSSatish Balay     case 1:
156649b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_1;
156749b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_1;
1568431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_1;
1569431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_1;
157049b5e25fSSatish Balay       break;
157149b5e25fSSatish Balay     case 2:
157249b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_2;
157349b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_2;
1574431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_2;
1575431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_2;
157649b5e25fSSatish Balay       break;
157749b5e25fSSatish Balay     case 3:
157849b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_3;
157949b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_3;
1580431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_3;
1581431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_3;
158249b5e25fSSatish Balay       break;
158349b5e25fSSatish Balay     case 4:
158449b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_4;
158549b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_4;
1586431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_4;
1587431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_4;
158849b5e25fSSatish Balay       break;
158949b5e25fSSatish Balay     case 5:
159049b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_5;
159149b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_5;
1592431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_5;
1593431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_5;
159449b5e25fSSatish Balay       break;
159549b5e25fSSatish Balay     case 6:
159649b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_6;
159749b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_6;
1598431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_6;
1599431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_6;
160049b5e25fSSatish Balay       break;
160149b5e25fSSatish Balay     case 7:
1602de53e5efSHong Zhang       B->ops->mult             = MatMult_SeqSBAIJ_7;
160349b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_7;
1604431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_7;
1605431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_7;
160649b5e25fSSatish Balay       break;
160749b5e25fSSatish Balay     }
160849b5e25fSSatish Balay   }
160949b5e25fSSatish Balay 
161049b5e25fSSatish Balay   b->mbs = mbs;
16114dcd73b1SHong Zhang   b->nbs = nbs;
1612ab93d7beSBarry Smith   if (!skipallocation) {
16132ee49352SLisandro Dalcin     if (!b->imax) {
1614dcca6d9dSJed Brown       ierr = PetscMalloc2(mbs,&b->imax,mbs,&b->ilen);CHKERRQ(ierr);
161526fbe8dcSKarl Rupp 
1616c760cd28SBarry Smith       b->free_imax_ilen = PETSC_TRUE;
161726fbe8dcSKarl Rupp 
16183bb1ff40SBarry Smith       ierr = PetscLogObjectMemory((PetscObject)B,2*mbs*sizeof(PetscInt));CHKERRQ(ierr);
16192ee49352SLisandro Dalcin     }
162049b5e25fSSatish Balay     if (!nnz) {
1621435da068SBarry Smith       if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 5;
162249b5e25fSSatish Balay       else if (nz <= 0) nz = 1;
162326fbe8dcSKarl Rupp       for (i=0; i<mbs; i++) b->imax[i] = nz;
1624153ea458SHong Zhang       nz = nz*mbs; /* total nz */
162549b5e25fSSatish Balay     } else {
162649b5e25fSSatish Balay       nz = 0;
16278cef66ccSHong Zhang       for (i=0; i<mbs; i++) {b->imax[i] = nnz[i]; nz += nnz[i];}
162849b5e25fSSatish Balay     }
16292ee49352SLisandro Dalcin     /* b->ilen will count nonzeros in each block row so far. */
163026fbe8dcSKarl Rupp     for (i=0; i<mbs; i++) b->ilen[i] = 0;
16316c6c5352SBarry Smith     /* nz=(nz+mbs)/2; */ /* total diagonal and superdiagonal nonzero blocks */
163249b5e25fSSatish Balay 
163349b5e25fSSatish Balay     /* allocate the matrix space */
16342ee49352SLisandro Dalcin     ierr = MatSeqXAIJFreeAIJ(B,&b->a,&b->j,&b->i);CHKERRQ(ierr);
1635dcca6d9dSJed Brown     ierr = PetscMalloc3(bs2*nz,&b->a,nz,&b->j,B->rmap->N+1,&b->i);CHKERRQ(ierr);
16363bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)B,(B->rmap->N+1)*sizeof(PetscInt)+nz*(bs2*sizeof(PetscScalar)+sizeof(PetscInt)));CHKERRQ(ierr);
16376c6c5352SBarry Smith     ierr = PetscMemzero(b->a,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr);
163813f74950SBarry Smith     ierr = PetscMemzero(b->j,nz*sizeof(PetscInt));CHKERRQ(ierr);
163926fbe8dcSKarl Rupp 
164049b5e25fSSatish Balay     b->singlemalloc = PETSC_TRUE;
164149b5e25fSSatish Balay 
164249b5e25fSSatish Balay     /* pointer to beginning of each row */
1643e60cf9a0SBarry Smith     b->i[0] = 0;
164426fbe8dcSKarl Rupp     for (i=1; i<mbs+1; i++) b->i[i] = b->i[i-1] + b->imax[i-1];
164526fbe8dcSKarl Rupp 
1646e6b907acSBarry Smith     b->free_a  = PETSC_TRUE;
1647e6b907acSBarry Smith     b->free_ij = PETSC_TRUE;
1648e811da20SHong Zhang   } else {
1649e6b907acSBarry Smith     b->free_a  = PETSC_FALSE;
1650e6b907acSBarry Smith     b->free_ij = PETSC_FALSE;
1651ab93d7beSBarry Smith   }
165249b5e25fSSatish Balay 
165349b5e25fSSatish Balay   b->bs2     = bs2;
16546c6c5352SBarry Smith   b->nz      = 0;
1655b32cb4a7SJed Brown   b->maxnz   = nz;
165616cdd363SHong Zhang   b->inew    = 0;
165716cdd363SHong Zhang   b->jnew    = 0;
165816cdd363SHong Zhang   b->anew    = 0;
165916cdd363SHong Zhang   b->a2anew  = 0;
16601a3463dfSHong Zhang   b->permute = PETSC_FALSE;
1661cb7b82ddSBarry Smith 
1662cb7b82ddSBarry Smith   B->was_assembled = PETSC_FALSE;
1663cb7b82ddSBarry Smith   B->assembled     = PETSC_FALSE;
16642576faa2SJed Brown   if (realalloc) {ierr = MatSetOption(B,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);}
1665c464158bSHong Zhang   PetscFunctionReturn(0);
1666c464158bSHong Zhang }
1667153ea458SHong Zhang 
166838f409ebSLisandro Dalcin PetscErrorCode MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ(Mat B,PetscInt bs,const PetscInt ii[],const PetscInt jj[], const PetscScalar V[])
166938f409ebSLisandro Dalcin {
167038f409ebSLisandro Dalcin   PetscInt       i,j,m,nz,nz_max=0,*nnz;
167138f409ebSLisandro Dalcin   PetscScalar    *values=0;
167238f409ebSLisandro Dalcin   PetscBool      roworiented = ((Mat_SeqSBAIJ*)B->data)->roworiented;
167338f409ebSLisandro Dalcin   PetscErrorCode ierr;
167438f409ebSLisandro Dalcin   PetscFunctionBegin;
167538f409ebSLisandro Dalcin   if (bs < 1) SETERRQ1(PetscObjectComm((PetscObject)B),PETSC_ERR_ARG_OUTOFRANGE,"Invalid block size specified, must be positive but it is %D",bs);
167638f409ebSLisandro Dalcin   ierr   = PetscLayoutSetBlockSize(B->rmap,bs);CHKERRQ(ierr);
167738f409ebSLisandro Dalcin   ierr   = PetscLayoutSetBlockSize(B->cmap,bs);CHKERRQ(ierr);
167838f409ebSLisandro Dalcin   ierr   = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr);
167938f409ebSLisandro Dalcin   ierr   = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr);
168038f409ebSLisandro Dalcin   ierr   = PetscLayoutGetBlockSize(B->rmap,&bs);CHKERRQ(ierr);
168138f409ebSLisandro Dalcin   m      = B->rmap->n/bs;
168238f409ebSLisandro Dalcin 
168338f409ebSLisandro Dalcin   if (ii[0]) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"ii[0] must be 0 but it is %D",ii[0]);
1684854ce69bSBarry Smith   ierr = PetscMalloc1(m+1,&nnz);CHKERRQ(ierr);
168538f409ebSLisandro Dalcin   for (i=0; i<m; i++) {
168638f409ebSLisandro Dalcin     nz = ii[i+1] - ii[i];
168738f409ebSLisandro Dalcin     if (nz < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row %D has a negative number of columns %D",i,nz);
168838f409ebSLisandro Dalcin     nz_max = PetscMax(nz_max,nz);
168938f409ebSLisandro Dalcin     nnz[i] = nz;
169038f409ebSLisandro Dalcin   }
169138f409ebSLisandro Dalcin   ierr = MatSeqSBAIJSetPreallocation(B,bs,0,nnz);CHKERRQ(ierr);
169238f409ebSLisandro Dalcin   ierr = PetscFree(nnz);CHKERRQ(ierr);
169338f409ebSLisandro Dalcin 
169438f409ebSLisandro Dalcin   values = (PetscScalar*)V;
169538f409ebSLisandro Dalcin   if (!values) {
16961795a4d1SJed Brown     ierr = PetscCalloc1(bs*bs*nz_max,&values);CHKERRQ(ierr);
169738f409ebSLisandro Dalcin   }
169838f409ebSLisandro Dalcin   for (i=0; i<m; i++) {
169938f409ebSLisandro Dalcin     PetscInt          ncols  = ii[i+1] - ii[i];
170038f409ebSLisandro Dalcin     const PetscInt    *icols = jj + ii[i];
170138f409ebSLisandro Dalcin     if (!roworiented || bs == 1) {
170238f409ebSLisandro Dalcin       const PetscScalar *svals = values + (V ? (bs*bs*ii[i]) : 0);
170338f409ebSLisandro Dalcin       ierr = MatSetValuesBlocked_SeqSBAIJ(B,1,&i,ncols,icols,svals,INSERT_VALUES);CHKERRQ(ierr);
170438f409ebSLisandro Dalcin     } else {
170538f409ebSLisandro Dalcin       for (j=0; j<ncols; j++) {
170638f409ebSLisandro Dalcin         const PetscScalar *svals = values + (V ? (bs*bs*(ii[i]+j)) : 0);
170738f409ebSLisandro Dalcin         ierr = MatSetValuesBlocked_SeqSBAIJ(B,1,&i,1,&icols[j],svals,INSERT_VALUES);CHKERRQ(ierr);
170838f409ebSLisandro Dalcin       }
170938f409ebSLisandro Dalcin     }
171038f409ebSLisandro Dalcin   }
171138f409ebSLisandro Dalcin   if (!V) { ierr = PetscFree(values);CHKERRQ(ierr); }
171238f409ebSLisandro Dalcin   ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
171338f409ebSLisandro Dalcin   ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
171438f409ebSLisandro Dalcin   ierr = MatSetOption(B,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
171538f409ebSLisandro Dalcin   PetscFunctionReturn(0);
171638f409ebSLisandro Dalcin }
171738f409ebSLisandro Dalcin 
1718db4efbfdSBarry Smith /*
1719db4efbfdSBarry Smith    This is used to set the numeric factorization for both Cholesky and ICC symbolic factorization
1720db4efbfdSBarry Smith */
1721ace3abfcSBarry Smith PetscErrorCode MatSeqSBAIJSetNumericFactorization_inplace(Mat B,PetscBool natural)
1722db4efbfdSBarry Smith {
1723db4efbfdSBarry Smith   PetscErrorCode ierr;
1724ace3abfcSBarry Smith   PetscBool      flg = PETSC_FALSE;
1725db4efbfdSBarry Smith   PetscInt       bs  = B->rmap->bs;
1726db4efbfdSBarry Smith 
1727db4efbfdSBarry Smith   PetscFunctionBegin;
1728c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(((PetscObject)B)->options,((PetscObject)B)->prefix,"-mat_no_unroll",&flg,NULL);CHKERRQ(ierr);
1729db4efbfdSBarry Smith   if (flg) bs = 8;
1730db4efbfdSBarry Smith 
1731db4efbfdSBarry Smith   if (!natural) {
1732db4efbfdSBarry Smith     switch (bs) {
1733db4efbfdSBarry Smith     case 1:
1734d595f711SHong Zhang       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_inplace;
1735db4efbfdSBarry Smith       break;
1736db4efbfdSBarry Smith     case 2:
1737db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2;
1738db4efbfdSBarry Smith       break;
1739db4efbfdSBarry Smith     case 3:
1740db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3;
1741db4efbfdSBarry Smith       break;
1742db4efbfdSBarry Smith     case 4:
1743db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4;
1744db4efbfdSBarry Smith       break;
1745db4efbfdSBarry Smith     case 5:
1746db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5;
1747db4efbfdSBarry Smith       break;
1748db4efbfdSBarry Smith     case 6:
1749db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6;
1750db4efbfdSBarry Smith       break;
1751db4efbfdSBarry Smith     case 7:
1752db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7;
1753db4efbfdSBarry Smith       break;
1754db4efbfdSBarry Smith     default:
1755db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N;
1756db4efbfdSBarry Smith       break;
1757db4efbfdSBarry Smith     }
1758db4efbfdSBarry Smith   } else {
1759db4efbfdSBarry Smith     switch (bs) {
1760db4efbfdSBarry Smith     case 1:
1761d595f711SHong Zhang       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering_inplace;
1762db4efbfdSBarry Smith       break;
1763db4efbfdSBarry Smith     case 2:
1764db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering;
1765db4efbfdSBarry Smith       break;
1766db4efbfdSBarry Smith     case 3:
1767db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering;
1768db4efbfdSBarry Smith       break;
1769db4efbfdSBarry Smith     case 4:
1770db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering;
1771db4efbfdSBarry Smith       break;
1772db4efbfdSBarry Smith     case 5:
1773db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering;
1774db4efbfdSBarry Smith       break;
1775db4efbfdSBarry Smith     case 6:
1776db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering;
1777db4efbfdSBarry Smith       break;
1778db4efbfdSBarry Smith     case 7:
1779db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering;
1780db4efbfdSBarry Smith       break;
1781db4efbfdSBarry Smith     default:
1782db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering;
1783db4efbfdSBarry Smith       break;
1784db4efbfdSBarry Smith     }
1785db4efbfdSBarry Smith   }
1786db4efbfdSBarry Smith   PetscFunctionReturn(0);
1787db4efbfdSBarry Smith }
1788db4efbfdSBarry Smith 
1789cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqAIJ(Mat, MatType,MatReuse,Mat*);
1790cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqBAIJ(Mat, MatType,MatReuse,Mat*);
1791d769727bSBarry Smith 
1792cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatGetFactor_seqsbaij_petsc(Mat A,MatFactorType ftype,Mat *B)
17935c9eb25fSBarry Smith {
1794d0f46423SBarry Smith   PetscInt       n = A->rmap->n;
17955c9eb25fSBarry Smith   PetscErrorCode ierr;
17965c9eb25fSBarry Smith 
17975c9eb25fSBarry Smith   PetscFunctionBegin;
17980e92d65fSHong Zhang #if defined(PETSC_USE_COMPLEX)
17990e92d65fSHong Zhang   if (A->hermitian) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Hermitian Factor is not supported");
18000e92d65fSHong Zhang #endif
1801ce94432eSBarry Smith   ierr = MatCreate(PetscObjectComm((PetscObject)A),B);CHKERRQ(ierr);
18025c9eb25fSBarry Smith   ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr);
18035c9eb25fSBarry Smith   if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) {
18045c9eb25fSBarry Smith     ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr);
18050298fd71SBarry Smith     ierr = MatSeqSBAIJSetPreallocation(*B,A->rmap->bs,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr);
180626fbe8dcSKarl Rupp 
18077b056e98SHong Zhang     (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqSBAIJ;
1808c6d0d4f0SHong Zhang     (*B)->ops->iccfactorsymbolic      = MatICCFactorSymbolic_SeqSBAIJ;
1809e32f2f54SBarry Smith   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Factor type not supported");
181000c67f3bSHong Zhang 
1811d5f3da31SBarry Smith   (*B)->factortype = ftype;
181200c67f3bSHong Zhang   ierr = PetscFree((*B)->solvertype);CHKERRQ(ierr);
181300c67f3bSHong Zhang   ierr = PetscStrallocpy(MATSOLVERPETSC,&(*B)->solvertype);CHKERRQ(ierr);
18145c9eb25fSBarry Smith   PetscFunctionReturn(0);
18155c9eb25fSBarry Smith }
18165c9eb25fSBarry Smith 
18178397e458SBarry Smith /*@C
18188397e458SBarry Smith    MatSeqSBAIJGetArray - gives access to the array where the data for a MATSEQSBAIJ matrix is stored
18198397e458SBarry Smith 
18208397e458SBarry Smith    Not Collective
18218397e458SBarry Smith 
18228397e458SBarry Smith    Input Parameter:
18238397e458SBarry Smith .  mat - a MATSEQSBAIJ matrix
18248397e458SBarry Smith 
18258397e458SBarry Smith    Output Parameter:
18268397e458SBarry Smith .   array - pointer to the data
18278397e458SBarry Smith 
18288397e458SBarry Smith    Level: intermediate
18298397e458SBarry Smith 
18308397e458SBarry Smith .seealso: MatSeqSBAIJRestoreArray(), MatSeqAIJGetArray(), MatSeqAIJRestoreArray()
18318397e458SBarry Smith @*/
18328397e458SBarry Smith PetscErrorCode  MatSeqSBAIJGetArray(Mat A,PetscScalar **array)
18338397e458SBarry Smith {
18348397e458SBarry Smith   PetscErrorCode ierr;
18358397e458SBarry Smith 
18368397e458SBarry Smith   PetscFunctionBegin;
18378397e458SBarry Smith   ierr = PetscUseMethod(A,"MatSeqSBAIJGetArray_C",(Mat,PetscScalar**),(A,array));CHKERRQ(ierr);
18388397e458SBarry Smith   PetscFunctionReturn(0);
18398397e458SBarry Smith }
18408397e458SBarry Smith 
18418397e458SBarry Smith /*@C
18428397e458SBarry Smith    MatSeqSBAIJRestoreArray - returns access to the array where the data for a MATSEQSBAIJ matrix is stored obtained by MatSeqSBAIJGetArray()
18438397e458SBarry Smith 
18448397e458SBarry Smith    Not Collective
18458397e458SBarry Smith 
18468397e458SBarry Smith    Input Parameters:
18478397e458SBarry Smith .  mat - a MATSEQSBAIJ matrix
18488397e458SBarry Smith .  array - pointer to the data
18498397e458SBarry Smith 
18508397e458SBarry Smith    Level: intermediate
18518397e458SBarry Smith 
18528397e458SBarry Smith .seealso: MatSeqSBAIJGetArray(), MatSeqAIJGetArray(), MatSeqAIJRestoreArray()
18538397e458SBarry Smith @*/
18548397e458SBarry Smith PetscErrorCode  MatSeqSBAIJRestoreArray(Mat A,PetscScalar **array)
18558397e458SBarry Smith {
18568397e458SBarry Smith   PetscErrorCode ierr;
18578397e458SBarry Smith 
18588397e458SBarry Smith   PetscFunctionBegin;
18598397e458SBarry Smith   ierr = PetscUseMethod(A,"MatSeqSBAIJRestoreArray_C",(Mat,PetscScalar**),(A,array));CHKERRQ(ierr);
18608397e458SBarry Smith   PetscFunctionReturn(0);
18618397e458SBarry Smith }
18628397e458SBarry Smith 
18630bad9183SKris Buschelman /*MC
1864fafad747SKris Buschelman   MATSEQSBAIJ - MATSEQSBAIJ = "seqsbaij" - A matrix type to be used for sequential symmetric block sparse matrices,
18650bad9183SKris Buschelman   based on block compressed sparse row format.  Only the upper triangular portion of the matrix is stored.
18660bad9183SKris Buschelman 
1867828413b8SBarry Smith   For complex numbers by default this matrix is symmetric, NOT Hermitian symmetric. To make it Hermitian symmetric you
186871dad5bbSBarry Smith   can call MatSetOption(Mat, MAT_HERMITIAN); after MatAssemblyEnd()
1869828413b8SBarry Smith 
18700bad9183SKris Buschelman   Options Database Keys:
18710bad9183SKris Buschelman   . -mat_type seqsbaij - sets the matrix type to "seqsbaij" during a call to MatSetFromOptions()
18720bad9183SKris Buschelman 
187371dad5bbSBarry Smith   Notes: By default if you insert values into the lower triangular part of the matrix they are simply ignored (since they are not
187471dad5bbSBarry 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
187571dad5bbSBarry 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.
187671dad5bbSBarry Smith 
187771dad5bbSBarry Smith 
18780bad9183SKris Buschelman   Level: beginner
18790bad9183SKris Buschelman 
18800bad9183SKris Buschelman   .seealso: MatCreateSeqSBAIJ
18810bad9183SKris Buschelman M*/
18820bad9183SKris Buschelman 
1883cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqSBSTRM(Mat, MatType,MatReuse,Mat*);
1884aa5a9175SDahai Guo 
18858cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatCreate_SeqSBAIJ(Mat B)
1886a23d5eceSKris Buschelman {
1887a23d5eceSKris Buschelman   Mat_SeqSBAIJ   *b;
1888dfbe8321SBarry Smith   PetscErrorCode ierr;
188913f74950SBarry Smith   PetscMPIInt    size;
1890ace3abfcSBarry Smith   PetscBool      no_unroll = PETSC_FALSE,no_inode = PETSC_FALSE;
1891a23d5eceSKris Buschelman 
1892a23d5eceSKris Buschelman   PetscFunctionBegin;
1893ce94432eSBarry Smith   ierr = MPI_Comm_size(PetscObjectComm((PetscObject)B),&size);CHKERRQ(ierr);
1894e32f2f54SBarry Smith   if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Comm must be of size 1");
1895a23d5eceSKris Buschelman 
1896b00a9115SJed Brown   ierr    = PetscNewLog(B,&b);CHKERRQ(ierr);
1897a23d5eceSKris Buschelman   B->data = (void*)b;
1898a23d5eceSKris Buschelman   ierr    = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr);
189926fbe8dcSKarl Rupp 
1900a23d5eceSKris Buschelman   B->ops->destroy    = MatDestroy_SeqSBAIJ;
1901a23d5eceSKris Buschelman   B->ops->view       = MatView_SeqSBAIJ;
1902a23d5eceSKris Buschelman   b->row             = 0;
1903a23d5eceSKris Buschelman   b->icol            = 0;
1904a23d5eceSKris Buschelman   b->reallocs        = 0;
1905a23d5eceSKris Buschelman   b->saved_values    = 0;
19060def2e27SBarry Smith   b->inode.limit     = 5;
19070def2e27SBarry Smith   b->inode.max_limit = 5;
1908a23d5eceSKris Buschelman 
1909a23d5eceSKris Buschelman   b->roworiented        = PETSC_TRUE;
1910a23d5eceSKris Buschelman   b->nonew              = 0;
1911a23d5eceSKris Buschelman   b->diag               = 0;
1912a23d5eceSKris Buschelman   b->solve_work         = 0;
1913a23d5eceSKris Buschelman   b->mult_work          = 0;
1914a23d5eceSKris Buschelman   B->spptr              = 0;
1915f2cbd3d5SJed Brown   B->info.nz_unneeded   = (PetscReal)b->maxnz*b->bs2;
1916a9817697SBarry Smith   b->keepnonzeropattern = PETSC_FALSE;
1917a23d5eceSKris Buschelman 
1918a23d5eceSKris Buschelman   b->inew    = 0;
1919a23d5eceSKris Buschelman   b->jnew    = 0;
1920a23d5eceSKris Buschelman   b->anew    = 0;
1921a23d5eceSKris Buschelman   b->a2anew  = 0;
1922a23d5eceSKris Buschelman   b->permute = PETSC_FALSE;
1923a23d5eceSKris Buschelman 
192471dad5bbSBarry Smith   b->ignore_ltriangular = PETSC_TRUE;
192526fbe8dcSKarl Rupp 
1926c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(((PetscObject)B)->options,((PetscObject)B)->prefix,"-mat_ignore_lower_triangular",&b->ignore_ltriangular,NULL);CHKERRQ(ierr);
1927941593c8SHong Zhang 
1928f5edf698SHong Zhang   b->getrow_utriangular = PETSC_FALSE;
192926fbe8dcSKarl Rupp 
1930c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(((PetscObject)B)->options,((PetscObject)B)->prefix,"-mat_getrow_uppertriangular",&b->getrow_utriangular,NULL);CHKERRQ(ierr);
1931f5edf698SHong Zhang 
19328397e458SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJGetArray_C",MatSeqSBAIJGetArray_SeqSBAIJ);CHKERRQ(ierr);
19338397e458SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJRestoreArray_C",MatSeqSBAIJRestoreArray_SeqSBAIJ);CHKERRQ(ierr);
1934bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatStoreValues_C",MatStoreValues_SeqSBAIJ);CHKERRQ(ierr);
1935bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatRetrieveValues_C",MatRetrieveValues_SeqSBAIJ);CHKERRQ(ierr);
1936bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetColumnIndices_C",MatSeqSBAIJSetColumnIndices_SeqSBAIJ);CHKERRQ(ierr);
1937bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqaij_C",MatConvert_SeqSBAIJ_SeqAIJ);CHKERRQ(ierr);
1938bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqbaij_C",MatConvert_SeqSBAIJ_SeqBAIJ);CHKERRQ(ierr);
1939bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetPreallocation_C",MatSeqSBAIJSetPreallocation_SeqSBAIJ);CHKERRQ(ierr);
194038f409ebSLisandro Dalcin   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetPreallocationCSR_C",MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ);CHKERRQ(ierr);
19416214f412SHong Zhang #if defined(PETSC_HAVE_ELEMENTAL)
19426214f412SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_elemental_C",MatConvert_SeqSBAIJ_Elemental);CHKERRQ(ierr);
19436214f412SHong Zhang #endif
194423ce1328SBarry Smith 
194523ce1328SBarry Smith   B->symmetric                  = PETSC_TRUE;
194623ce1328SBarry Smith   B->structurally_symmetric     = PETSC_TRUE;
194723ce1328SBarry Smith   B->symmetric_set              = PETSC_TRUE;
194823ce1328SBarry Smith   B->structurally_symmetric_set = PETSC_TRUE;
194926fbe8dcSKarl Rupp 
195017667f90SBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)B,MATSEQSBAIJ);CHKERRQ(ierr);
19510def2e27SBarry Smith 
1952ce94432eSBarry Smith   ierr = PetscOptionsBegin(PetscObjectComm((PetscObject)B),((PetscObject)B)->prefix,"Options for SEQSBAIJ matrix","Mat");CHKERRQ(ierr);
19530298fd71SBarry Smith   ierr = PetscOptionsBool("-mat_no_unroll","Do not optimize for inodes (slower)",NULL,no_unroll,&no_unroll,NULL);CHKERRQ(ierr);
195426fbe8dcSKarl Rupp   if (no_unroll) {
195526fbe8dcSKarl Rupp     ierr = PetscInfo(B,"Not using Inode routines due to -mat_no_unroll\n");CHKERRQ(ierr);
195626fbe8dcSKarl Rupp   }
19570298fd71SBarry Smith   ierr = PetscOptionsBool("-mat_no_inode","Do not optimize for inodes (slower)",NULL,no_inode,&no_inode,NULL);CHKERRQ(ierr);
195826fbe8dcSKarl Rupp   if (no_inode) {
195926fbe8dcSKarl Rupp     ierr = PetscInfo(B,"Not using Inode routines due to -mat_no_inode\n");CHKERRQ(ierr);
196026fbe8dcSKarl Rupp   }
19610298fd71SBarry Smith   ierr = PetscOptionsInt("-mat_inode_limit","Do not use inodes larger then this value",NULL,b->inode.limit,&b->inode.limit,NULL);CHKERRQ(ierr);
19620def2e27SBarry Smith   ierr = PetscOptionsEnd();CHKERRQ(ierr);
1963ace3abfcSBarry Smith   b->inode.use = (PetscBool)(!(no_unroll || no_inode));
19640def2e27SBarry Smith   if (b->inode.limit > b->inode.max_limit) b->inode.limit = b->inode.max_limit;
1965a23d5eceSKris Buschelman   PetscFunctionReturn(0);
1966a23d5eceSKris Buschelman }
1967a23d5eceSKris Buschelman 
1968a23d5eceSKris Buschelman /*@C
1969a23d5eceSKris Buschelman    MatSeqSBAIJSetPreallocation - Creates a sparse symmetric matrix in block AIJ (block
1970a23d5eceSKris Buschelman    compressed row) format.  For good matrix assembly performance the
1971a23d5eceSKris Buschelman    user should preallocate the matrix storage by setting the parameter nz
1972a23d5eceSKris Buschelman    (or the array nnz).  By setting these parameters accurately, performance
1973a23d5eceSKris Buschelman    during matrix assembly can be increased by more than a factor of 50.
1974a23d5eceSKris Buschelman 
1975a23d5eceSKris Buschelman    Collective on Mat
1976a23d5eceSKris Buschelman 
1977a23d5eceSKris Buschelman    Input Parameters:
19781c4f3114SJed Brown +  B - the symmetric matrix
1979bb7ae925SBarry Smith .  bs - size of block, the blocks are ALWAYS square. One can use MatSetBlockSizes() to set a different row and column blocksize but the row
1980bb7ae925SBarry Smith           blocksize always defines the size of the blocks. The column blocksize sets the blocksize of the vectors obtained with MatCreateVecs()
1981a23d5eceSKris Buschelman .  nz - number of block nonzeros per block row (same for all rows)
1982a23d5eceSKris Buschelman -  nnz - array containing the number of block nonzeros in the upper triangular plus
19830298fd71SBarry Smith          diagonal portion of each block (possibly different for each block row) or NULL
1984a23d5eceSKris Buschelman 
1985a23d5eceSKris Buschelman    Options Database Keys:
1986a23d5eceSKris Buschelman .   -mat_no_unroll - uses code that does not unroll the loops in the
1987a23d5eceSKris Buschelman                      block calculations (much slower)
1988db4efbfdSBarry Smith .   -mat_block_size - size of the blocks to use (only works if a negative bs is passed in
1989a23d5eceSKris Buschelman 
1990a23d5eceSKris Buschelman    Level: intermediate
1991a23d5eceSKris Buschelman 
1992a23d5eceSKris Buschelman    Notes:
1993a23d5eceSKris Buschelman    Specify the preallocated storage with either nz or nnz (not both).
19940298fd71SBarry Smith    Set nz=PETSC_DEFAULT and nnz=NULL for PETSc to control dynamic memory
1995a7f22e61SSatish Balay    allocation.  See Users-Manual: ch_mat for details.
1996a23d5eceSKris Buschelman 
1997aa95bbe8SBarry Smith    You can call MatGetInfo() to get information on how effective the preallocation was;
1998aa95bbe8SBarry Smith    for example the fields mallocs,nz_allocated,nz_used,nz_unneeded;
1999aa95bbe8SBarry Smith    You can also run with the option -info and look for messages with the string
2000aa95bbe8SBarry Smith    malloc in them to see if additional memory allocation was needed.
2001aa95bbe8SBarry Smith 
200249a6f317SBarry Smith    If the nnz parameter is given then the nz parameter is ignored
200349a6f317SBarry Smith 
200449a6f317SBarry Smith 
200569b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateSBAIJ()
2006a23d5eceSKris Buschelman @*/
20077087cfbeSBarry Smith PetscErrorCode  MatSeqSBAIJSetPreallocation(Mat B,PetscInt bs,PetscInt nz,const PetscInt nnz[])
200813f74950SBarry Smith {
20094ac538c5SBarry Smith   PetscErrorCode ierr;
2010a23d5eceSKris Buschelman 
2011a23d5eceSKris Buschelman   PetscFunctionBegin;
20126ba663aaSJed Brown   PetscValidHeaderSpecific(B,MAT_CLASSID,1);
20136ba663aaSJed Brown   PetscValidType(B,1);
20146ba663aaSJed Brown   PetscValidLogicalCollectiveInt(B,bs,2);
20154ac538c5SBarry Smith   ierr = PetscTryMethod(B,"MatSeqSBAIJSetPreallocation_C",(Mat,PetscInt,PetscInt,const PetscInt[]),(B,bs,nz,nnz));CHKERRQ(ierr);
2016a23d5eceSKris Buschelman   PetscFunctionReturn(0);
2017a23d5eceSKris Buschelman }
201849b5e25fSSatish Balay 
201938f409ebSLisandro Dalcin /*@C
202038f409ebSLisandro Dalcin    MatSeqSBAIJSetPreallocationCSR - Allocates memory for a sparse sequential matrix in symmetric block AIJ format.
202138f409ebSLisandro Dalcin 
202238f409ebSLisandro Dalcin    Input Parameters:
20231c4f3114SJed Brown +  B - the matrix
2024eab78319SHong Zhang .  bs - size of block, the blocks are ALWAYS square.
202538f409ebSLisandro Dalcin .  i - the indices into j for the start of each local row (starts with zero)
202638f409ebSLisandro Dalcin .  j - the column indices for each local row (starts with zero) these must be sorted for each row
202738f409ebSLisandro Dalcin -  v - optional values in the matrix
202838f409ebSLisandro Dalcin 
202938f409ebSLisandro Dalcin    Level: developer
203038f409ebSLisandro Dalcin 
203138f409ebSLisandro Dalcin    Notes:
203238f409ebSLisandro Dalcin    The order of the entries in values is specified by the MatOption MAT_ROW_ORIENTED.  For example, C programs
203338f409ebSLisandro Dalcin    may want to use the default MAT_ROW_ORIENTED=PETSC_TRUE and use an array v[nnz][bs][bs] where the second index is
203438f409ebSLisandro Dalcin    over rows within a block and the last index is over columns within a block row.  Fortran programs will likely set
203538f409ebSLisandro Dalcin    MAT_ROW_ORIENTED=PETSC_FALSE and use a Fortran array v(bs,bs,nnz) in which the first index is over rows within a
203638f409ebSLisandro Dalcin    block column and the second index is over columns within a block.
203738f409ebSLisandro Dalcin 
203838f409ebSLisandro Dalcin .keywords: matrix, block, aij, compressed row, sparse
203938f409ebSLisandro Dalcin 
204038f409ebSLisandro Dalcin .seealso: MatCreate(), MatCreateSeqSBAIJ(), MatSetValuesBlocked(), MatSeqSBAIJSetPreallocation(), MATSEQSBAIJ
204138f409ebSLisandro Dalcin @*/
204238f409ebSLisandro Dalcin PetscErrorCode MatSeqSBAIJSetPreallocationCSR(Mat B,PetscInt bs,const PetscInt i[],const PetscInt j[], const PetscScalar v[])
204338f409ebSLisandro Dalcin {
204438f409ebSLisandro Dalcin   PetscErrorCode ierr;
204538f409ebSLisandro Dalcin 
204638f409ebSLisandro Dalcin   PetscFunctionBegin;
204738f409ebSLisandro Dalcin   PetscValidHeaderSpecific(B,MAT_CLASSID,1);
204838f409ebSLisandro Dalcin   PetscValidType(B,1);
204938f409ebSLisandro Dalcin   PetscValidLogicalCollectiveInt(B,bs,2);
205038f409ebSLisandro Dalcin   ierr = PetscTryMethod(B,"MatSeqSBAIJSetPreallocationCSR_C",(Mat,PetscInt,const PetscInt[],const PetscInt[],const PetscScalar[]),(B,bs,i,j,v));CHKERRQ(ierr);
205138f409ebSLisandro Dalcin   PetscFunctionReturn(0);
205238f409ebSLisandro Dalcin }
205338f409ebSLisandro Dalcin 
2054c464158bSHong Zhang /*@C
2055c464158bSHong Zhang    MatCreateSeqSBAIJ - Creates a sparse symmetric matrix in block AIJ (block
2056c464158bSHong Zhang    compressed row) format.  For good matrix assembly performance the
2057c464158bSHong Zhang    user should preallocate the matrix storage by setting the parameter nz
2058c464158bSHong Zhang    (or the array nnz).  By setting these parameters accurately, performance
2059c464158bSHong Zhang    during matrix assembly can be increased by more than a factor of 50.
206049b5e25fSSatish Balay 
2061c464158bSHong Zhang    Collective on MPI_Comm
2062c464158bSHong Zhang 
2063c464158bSHong Zhang    Input Parameters:
2064c464158bSHong Zhang +  comm - MPI communicator, set to PETSC_COMM_SELF
2065bb7ae925SBarry Smith .  bs - size of block, the blocks are ALWAYS square. One can use MatSetBlockSizes() to set a different row and column blocksize but the row
2066bb7ae925SBarry Smith           blocksize always defines the size of the blocks. The column blocksize sets the blocksize of the vectors obtained with MatCreateVecs()
2067c464158bSHong Zhang .  m - number of rows, or number of columns
2068c464158bSHong Zhang .  nz - number of block nonzeros per block row (same for all rows)
2069744e8345SSatish Balay -  nnz - array containing the number of block nonzeros in the upper triangular plus
20700298fd71SBarry Smith          diagonal portion of each block (possibly different for each block row) or NULL
2071c464158bSHong Zhang 
2072c464158bSHong Zhang    Output Parameter:
2073c464158bSHong Zhang .  A - the symmetric matrix
2074c464158bSHong Zhang 
2075c464158bSHong Zhang    Options Database Keys:
2076c464158bSHong Zhang .   -mat_no_unroll - uses code that does not unroll the loops in the
2077c464158bSHong Zhang                      block calculations (much slower)
2078c464158bSHong Zhang .    -mat_block_size - size of the blocks to use
2079c464158bSHong Zhang 
2080c464158bSHong Zhang    Level: intermediate
2081c464158bSHong Zhang 
2082175b88e8SBarry Smith    It is recommended that one use the MatCreate(), MatSetType() and/or MatSetFromOptions(),
2083ae1d86c5SBarry Smith    MatXXXXSetPreallocation() paradgm instead of this routine directly.
2084175b88e8SBarry Smith    [MatXXXXSetPreallocation() is, for example, MatSeqAIJSetPreallocation]
2085175b88e8SBarry Smith 
2086c464158bSHong Zhang    Notes:
20876d6d819aSHong Zhang    The number of rows and columns must be divisible by blocksize.
20886d6d819aSHong Zhang    This matrix type does not support complex Hermitian operation.
2089c464158bSHong Zhang 
2090c464158bSHong Zhang    Specify the preallocated storage with either nz or nnz (not both).
20910298fd71SBarry Smith    Set nz=PETSC_DEFAULT and nnz=NULL for PETSc to control dynamic memory
2092a7f22e61SSatish Balay    allocation.  See Users-Manual: ch_mat for details.
2093c464158bSHong Zhang 
209449a6f317SBarry Smith    If the nnz parameter is given then the nz parameter is ignored
209549a6f317SBarry Smith 
209669b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateSBAIJ()
2097c464158bSHong Zhang @*/
20987087cfbeSBarry Smith PetscErrorCode  MatCreateSeqSBAIJ(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt nz,const PetscInt nnz[],Mat *A)
2099c464158bSHong Zhang {
2100dfbe8321SBarry Smith   PetscErrorCode ierr;
2101c464158bSHong Zhang 
2102c464158bSHong Zhang   PetscFunctionBegin;
2103f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,A);CHKERRQ(ierr);
2104f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*A,m,n,m,n);CHKERRQ(ierr);
2105c464158bSHong Zhang   ierr = MatSetType(*A,MATSEQSBAIJ);CHKERRQ(ierr);
2106367daffbSBarry Smith   ierr = MatSeqSBAIJSetPreallocation(*A,bs,nz,(PetscInt*)nnz);CHKERRQ(ierr);
210749b5e25fSSatish Balay   PetscFunctionReturn(0);
210849b5e25fSSatish Balay }
210949b5e25fSSatish Balay 
2110dfbe8321SBarry Smith PetscErrorCode MatDuplicate_SeqSBAIJ(Mat A,MatDuplicateOption cpvalues,Mat *B)
211149b5e25fSSatish Balay {
211249b5e25fSSatish Balay   Mat            C;
211349b5e25fSSatish Balay   Mat_SeqSBAIJ   *c,*a = (Mat_SeqSBAIJ*)A->data;
21146849ba73SBarry Smith   PetscErrorCode ierr;
2115b40805acSSatish Balay   PetscInt       i,mbs = a->mbs,nz = a->nz,bs2 =a->bs2;
211649b5e25fSSatish Balay 
211749b5e25fSSatish Balay   PetscFunctionBegin;
2118e32f2f54SBarry Smith   if (a->i[mbs] != nz) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Corrupt matrix");
211949b5e25fSSatish Balay 
212049b5e25fSSatish Balay   *B   = 0;
2121ce94432eSBarry Smith   ierr = MatCreate(PetscObjectComm((PetscObject)A),&C);CHKERRQ(ierr);
2122d0f46423SBarry Smith   ierr = MatSetSizes(C,A->rmap->N,A->cmap->n,A->rmap->N,A->cmap->n);CHKERRQ(ierr);
21238e9a0fb8SHong Zhang   ierr = MatSetType(C,MATSEQSBAIJ);CHKERRQ(ierr);
21241d5dac46SHong Zhang   ierr = PetscMemcpy(C->ops,A->ops,sizeof(struct _MatOps));CHKERRQ(ierr);
2125692f9cbeSHong Zhang   c    = (Mat_SeqSBAIJ*)C->data;
2126692f9cbeSHong Zhang 
2127273d9f13SBarry Smith   C->preallocated       = PETSC_TRUE;
2128d5f3da31SBarry Smith   C->factortype         = A->factortype;
212949b5e25fSSatish Balay   c->row                = 0;
213049b5e25fSSatish Balay   c->icol               = 0;
213149b5e25fSSatish Balay   c->saved_values       = 0;
2132a9817697SBarry Smith   c->keepnonzeropattern = a->keepnonzeropattern;
213349b5e25fSSatish Balay   C->assembled          = PETSC_TRUE;
213449b5e25fSSatish Balay 
21351e1e43feSBarry Smith   ierr   = PetscLayoutReference(A->rmap,&C->rmap);CHKERRQ(ierr);
21361e1e43feSBarry Smith   ierr   = PetscLayoutReference(A->cmap,&C->cmap);CHKERRQ(ierr);
213749b5e25fSSatish Balay   c->bs2 = a->bs2;
213849b5e25fSSatish Balay   c->mbs = a->mbs;
213949b5e25fSSatish Balay   c->nbs = a->nbs;
214049b5e25fSSatish Balay 
2141c760cd28SBarry Smith   if  (cpvalues == MAT_SHARE_NONZERO_PATTERN) {
2142c760cd28SBarry Smith     c->imax           = a->imax;
2143c760cd28SBarry Smith     c->ilen           = a->ilen;
2144c760cd28SBarry Smith     c->free_imax_ilen = PETSC_FALSE;
2145c760cd28SBarry Smith   } else {
2146dcca6d9dSJed Brown     ierr = PetscMalloc2((mbs+1),&c->imax,(mbs+1),&c->ilen);CHKERRQ(ierr);
21473bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)C,2*(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr);
214849b5e25fSSatish Balay     for (i=0; i<mbs; i++) {
214949b5e25fSSatish Balay       c->imax[i] = a->imax[i];
215049b5e25fSSatish Balay       c->ilen[i] = a->ilen[i];
215149b5e25fSSatish Balay     }
2152c760cd28SBarry Smith     c->free_imax_ilen = PETSC_TRUE;
2153c760cd28SBarry Smith   }
215449b5e25fSSatish Balay 
215549b5e25fSSatish Balay   /* allocate the matrix space */
21564da8f245SBarry Smith   if (cpvalues == MAT_SHARE_NONZERO_PATTERN) {
2157785e854fSJed Brown     ierr            = PetscMalloc1(bs2*nz,&c->a);CHKERRQ(ierr);
21583bb1ff40SBarry Smith     ierr            = PetscLogObjectMemory((PetscObject)C,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr);
215944e1c64aSLisandro Dalcin     c->i            = a->i;
216044e1c64aSLisandro Dalcin     c->j            = a->j;
21614da8f245SBarry Smith     c->singlemalloc = PETSC_FALSE;
216244e1c64aSLisandro Dalcin     c->free_a       = PETSC_TRUE;
21634da8f245SBarry Smith     c->free_ij      = PETSC_FALSE;
21644da8f245SBarry Smith     c->parent       = A;
21654da8f245SBarry Smith     ierr            = PetscObjectReference((PetscObject)A);CHKERRQ(ierr);
21664da8f245SBarry Smith     ierr            = MatSetOption(A,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
21674da8f245SBarry Smith     ierr            = MatSetOption(C,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
21684da8f245SBarry Smith   } else {
2169dcca6d9dSJed Brown     ierr            = PetscMalloc3(bs2*nz,&c->a,nz,&c->j,mbs+1,&c->i);CHKERRQ(ierr);
217013f74950SBarry Smith     ierr            = PetscMemcpy(c->i,a->i,(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr);
21713bb1ff40SBarry Smith     ierr            = PetscLogObjectMemory((PetscObject)C,(mbs+1)*sizeof(PetscInt) + nz*(bs2*sizeof(MatScalar) + sizeof(PetscInt)));CHKERRQ(ierr);
21724da8f245SBarry Smith     c->singlemalloc = PETSC_TRUE;
217344e1c64aSLisandro Dalcin     c->free_a       = PETSC_TRUE;
21744da8f245SBarry Smith     c->free_ij      = PETSC_TRUE;
21754da8f245SBarry Smith   }
217649b5e25fSSatish Balay   if (mbs > 0) {
21774da8f245SBarry Smith     if (cpvalues != MAT_SHARE_NONZERO_PATTERN) {
217813f74950SBarry Smith       ierr = PetscMemcpy(c->j,a->j,nz*sizeof(PetscInt));CHKERRQ(ierr);
21794da8f245SBarry Smith     }
218049b5e25fSSatish Balay     if (cpvalues == MAT_COPY_VALUES) {
218149b5e25fSSatish Balay       ierr = PetscMemcpy(c->a,a->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr);
218249b5e25fSSatish Balay     } else {
218349b5e25fSSatish Balay       ierr = PetscMemzero(c->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr);
218449b5e25fSSatish Balay     }
2185a1c3900fSBarry Smith     if (a->jshort) {
218644e1c64aSLisandro Dalcin       /* cannot share jshort, it is reallocated in MatAssemblyEnd_SeqSBAIJ() */
218744e1c64aSLisandro Dalcin       /* if the parent matrix is reassembled, this child matrix will never notice */
2188785e854fSJed Brown       ierr = PetscMalloc1(nz,&c->jshort);CHKERRQ(ierr);
21893bb1ff40SBarry Smith       ierr = PetscLogObjectMemory((PetscObject)C,nz*sizeof(unsigned short));CHKERRQ(ierr);
2190a1c3900fSBarry Smith       ierr = PetscMemcpy(c->jshort,a->jshort,nz*sizeof(unsigned short));CHKERRQ(ierr);
219126fbe8dcSKarl Rupp 
21924da8f245SBarry Smith       c->free_jshort = PETSC_TRUE;
21934da8f245SBarry Smith     }
2194a1c3900fSBarry Smith   }
219549b5e25fSSatish Balay 
219649b5e25fSSatish Balay   c->roworiented = a->roworiented;
219749b5e25fSSatish Balay   c->nonew       = a->nonew;
219849b5e25fSSatish Balay 
219949b5e25fSSatish Balay   if (a->diag) {
2200c760cd28SBarry Smith     if (cpvalues == MAT_SHARE_NONZERO_PATTERN) {
2201c760cd28SBarry Smith       c->diag      = a->diag;
2202c760cd28SBarry Smith       c->free_diag = PETSC_FALSE;
2203c760cd28SBarry Smith     } else {
2204785e854fSJed Brown       ierr = PetscMalloc1(mbs,&c->diag);CHKERRQ(ierr);
22053bb1ff40SBarry Smith       ierr = PetscLogObjectMemory((PetscObject)C,mbs*sizeof(PetscInt));CHKERRQ(ierr);
220626fbe8dcSKarl Rupp       for (i=0; i<mbs; i++) c->diag[i] = a->diag[i];
2207c760cd28SBarry Smith       c->free_diag = PETSC_TRUE;
2208c760cd28SBarry Smith     }
220944e1c64aSLisandro Dalcin   }
22106c6c5352SBarry Smith   c->nz         = a->nz;
2211f2cbd3d5SJed Brown   c->maxnz      = a->nz; /* Since we allocate exactly the right amount */
221249b5e25fSSatish Balay   c->solve_work = 0;
221349b5e25fSSatish Balay   c->mult_work  = 0;
221426fbe8dcSKarl Rupp 
221549b5e25fSSatish Balay   *B   = C;
2216140e18c1SBarry Smith   ierr = PetscFunctionListDuplicate(((PetscObject)A)->qlist,&((PetscObject)C)->qlist);CHKERRQ(ierr);
221749b5e25fSSatish Balay   PetscFunctionReturn(0);
221849b5e25fSSatish Balay }
221949b5e25fSSatish Balay 
2220112444f4SShri Abhyankar PetscErrorCode MatLoad_SeqSBAIJ(Mat newmat,PetscViewer viewer)
22212f480046SShri Abhyankar {
22222f480046SShri Abhyankar   Mat_SeqSBAIJ   *a;
22232f480046SShri Abhyankar   PetscErrorCode ierr;
22242f480046SShri Abhyankar   int            fd;
22252f480046SShri Abhyankar   PetscMPIInt    size;
22263059b6faSBarry Smith   PetscInt       i,nz,header[4],*rowlengths=0,M,N,bs = newmat->rmap->bs;
22272f480046SShri Abhyankar   PetscInt       *mask,mbs,*jj,j,rowcount,nzcount,k,*s_browlengths,maskcount;
22282f480046SShri Abhyankar   PetscInt       kmax,jcount,block,idx,point,nzcountb,extra_rows,rows,cols;
22292f480046SShri Abhyankar   PetscInt       *masked,nmask,tmp,bs2,ishift;
22302f480046SShri Abhyankar   PetscScalar    *aa;
2231ce94432eSBarry Smith   MPI_Comm       comm;
22322f480046SShri Abhyankar 
22332f480046SShri Abhyankar   PetscFunctionBegin;
2234c98fd787SBarry Smith   /* force binary viewer to load .info file if it has not yet done so */
2235c98fd787SBarry Smith   ierr = PetscViewerSetUp(viewer);CHKERRQ(ierr);
2236ce94432eSBarry Smith   ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr);
2237c5929fdfSBarry Smith   ierr = PetscOptionsGetInt(((PetscObject)newmat)->options,((PetscObject)newmat)->prefix,"-matload_block_size",&bs,NULL);CHKERRQ(ierr);
22383059b6faSBarry Smith   if (bs < 0) bs = 1;
22392f480046SShri Abhyankar   bs2  = bs*bs;
22402f480046SShri Abhyankar 
22412f480046SShri Abhyankar   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
22422f480046SShri Abhyankar   if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"view must have one processor");
22432f480046SShri Abhyankar   ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr);
22442f480046SShri Abhyankar   ierr = PetscBinaryRead(fd,header,4,PETSC_INT);CHKERRQ(ierr);
22452f480046SShri Abhyankar   if (header[0] != MAT_FILE_CLASSID) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"not Mat object");
22462f480046SShri Abhyankar   M = header[1]; N = header[2]; nz = header[3];
22472f480046SShri Abhyankar 
22482f480046SShri Abhyankar   if (header[3] < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Matrix stored in special format, cannot load as SeqSBAIJ");
22492f480046SShri Abhyankar 
22502f480046SShri Abhyankar   if (M != N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Can only do square matrices");
22512f480046SShri Abhyankar 
22522f480046SShri Abhyankar   /*
22532f480046SShri Abhyankar      This code adds extra rows to make sure the number of rows is
22542f480046SShri Abhyankar     divisible by the blocksize
22552f480046SShri Abhyankar   */
22562f480046SShri Abhyankar   mbs        = M/bs;
22572f480046SShri Abhyankar   extra_rows = bs - M + bs*(mbs);
22582f480046SShri Abhyankar   if (extra_rows == bs) extra_rows = 0;
22592f480046SShri Abhyankar   else                  mbs++;
22602f480046SShri Abhyankar   if (extra_rows) {
22612f480046SShri Abhyankar     ierr = PetscInfo(viewer,"Padding loaded matrix to match blocksize\n");CHKERRQ(ierr);
22622f480046SShri Abhyankar   }
22632f480046SShri Abhyankar 
22642f480046SShri Abhyankar   /* Set global sizes if not already set */
22652f480046SShri Abhyankar   if (newmat->rmap->n < 0 && newmat->rmap->N < 0 && newmat->cmap->n < 0 && newmat->cmap->N < 0) {
22662f480046SShri Abhyankar     ierr = MatSetSizes(newmat,PETSC_DECIDE,PETSC_DECIDE,M+extra_rows,N+extra_rows);CHKERRQ(ierr);
22672f480046SShri Abhyankar   } else { /* Check if the matrix global sizes are correct */
22682f480046SShri Abhyankar     ierr = MatGetSize(newmat,&rows,&cols);CHKERRQ(ierr);
22692f480046SShri 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);
22702f480046SShri Abhyankar   }
22712f480046SShri Abhyankar 
22722f480046SShri Abhyankar   /* read in row lengths */
2273854ce69bSBarry Smith   ierr = PetscMalloc1(M+extra_rows,&rowlengths);CHKERRQ(ierr);
22742f480046SShri Abhyankar   ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr);
22752f480046SShri Abhyankar   for (i=0; i<extra_rows; i++) rowlengths[M+i] = 1;
22762f480046SShri Abhyankar 
22772f480046SShri Abhyankar   /* read in column indices */
2278854ce69bSBarry Smith   ierr = PetscMalloc1(nz+extra_rows,&jj);CHKERRQ(ierr);
22792f480046SShri Abhyankar   ierr = PetscBinaryRead(fd,jj,nz,PETSC_INT);CHKERRQ(ierr);
22802f480046SShri Abhyankar   for (i=0; i<extra_rows; i++) jj[nz+i] = M+i;
22812f480046SShri Abhyankar 
22822f480046SShri Abhyankar   /* loop over row lengths determining block row lengths */
22831795a4d1SJed Brown   ierr     = PetscCalloc1(mbs,&s_browlengths);CHKERRQ(ierr);
2284dcca6d9dSJed Brown   ierr     = PetscMalloc2(mbs,&mask,mbs,&masked);CHKERRQ(ierr);
22852f480046SShri Abhyankar   ierr     = PetscMemzero(mask,mbs*sizeof(PetscInt));CHKERRQ(ierr);
22862f480046SShri Abhyankar   rowcount = 0;
22872f480046SShri Abhyankar   nzcount  = 0;
22882f480046SShri Abhyankar   for (i=0; i<mbs; i++) {
22892f480046SShri Abhyankar     nmask = 0;
22902f480046SShri Abhyankar     for (j=0; j<bs; j++) {
22912f480046SShri Abhyankar       kmax = rowlengths[rowcount];
22922f480046SShri Abhyankar       for (k=0; k<kmax; k++) {
22932f480046SShri Abhyankar         tmp = jj[nzcount++]/bs;   /* block col. index */
22942f480046SShri Abhyankar         if (!mask[tmp] && tmp >= i) {masked[nmask++] = tmp; mask[tmp] = 1;}
22952f480046SShri Abhyankar       }
22962f480046SShri Abhyankar       rowcount++;
22972f480046SShri Abhyankar     }
22982f480046SShri Abhyankar     s_browlengths[i] += nmask;
22992f480046SShri Abhyankar 
23002f480046SShri Abhyankar     /* zero out the mask elements we set */
23012f480046SShri Abhyankar     for (j=0; j<nmask; j++) mask[masked[j]] = 0;
23022f480046SShri Abhyankar   }
23032f480046SShri Abhyankar 
23042f480046SShri Abhyankar   /* Do preallocation */
2305367daffbSBarry Smith   ierr = MatSeqSBAIJSetPreallocation(newmat,bs,0,s_browlengths);CHKERRQ(ierr);
23062f480046SShri Abhyankar   a    = (Mat_SeqSBAIJ*)newmat->data;
23072f480046SShri Abhyankar 
23082f480046SShri Abhyankar   /* set matrix "i" values */
23092f480046SShri Abhyankar   a->i[0] = 0;
23102f480046SShri Abhyankar   for (i=1; i<= mbs; i++) {
23112f480046SShri Abhyankar     a->i[i]      = a->i[i-1] + s_browlengths[i-1];
23122f480046SShri Abhyankar     a->ilen[i-1] = s_browlengths[i-1];
23132f480046SShri Abhyankar   }
23142f480046SShri Abhyankar   a->nz = a->i[mbs];
23152f480046SShri Abhyankar 
23162f480046SShri Abhyankar   /* read in nonzero values */
2317854ce69bSBarry Smith   ierr = PetscMalloc1(nz+extra_rows,&aa);CHKERRQ(ierr);
23182f480046SShri Abhyankar   ierr = PetscBinaryRead(fd,aa,nz,PETSC_SCALAR);CHKERRQ(ierr);
23192f480046SShri Abhyankar   for (i=0; i<extra_rows; i++) aa[nz+i] = 1.0;
23202f480046SShri Abhyankar 
23212f480046SShri Abhyankar   /* set "a" and "j" values into matrix */
23222f480046SShri Abhyankar   nzcount = 0; jcount = 0;
23232f480046SShri Abhyankar   for (i=0; i<mbs; i++) {
23242f480046SShri Abhyankar     nzcountb = nzcount;
23252f480046SShri Abhyankar     nmask    = 0;
23262f480046SShri Abhyankar     for (j=0; j<bs; j++) {
23272f480046SShri Abhyankar       kmax = rowlengths[i*bs+j];
23282f480046SShri Abhyankar       for (k=0; k<kmax; k++) {
23292f480046SShri Abhyankar         tmp = jj[nzcount++]/bs; /* block col. index */
23302f480046SShri Abhyankar         if (!mask[tmp] && tmp >= i) { masked[nmask++] = tmp; mask[tmp] = 1;}
23312f480046SShri Abhyankar       }
23322f480046SShri Abhyankar     }
23332f480046SShri Abhyankar     /* sort the masked values */
23342f480046SShri Abhyankar     ierr = PetscSortInt(nmask,masked);CHKERRQ(ierr);
23352f480046SShri Abhyankar 
23362f480046SShri Abhyankar     /* set "j" values into matrix */
23372f480046SShri Abhyankar     maskcount = 1;
23382f480046SShri Abhyankar     for (j=0; j<nmask; j++) {
23392f480046SShri Abhyankar       a->j[jcount++]  = masked[j];
23402f480046SShri Abhyankar       mask[masked[j]] = maskcount++;
23412f480046SShri Abhyankar     }
23422f480046SShri Abhyankar 
23432f480046SShri Abhyankar     /* set "a" values into matrix */
23442f480046SShri Abhyankar     ishift = bs2*a->i[i];
23452f480046SShri Abhyankar     for (j=0; j<bs; j++) {
23462f480046SShri Abhyankar       kmax = rowlengths[i*bs+j];
23472f480046SShri Abhyankar       for (k=0; k<kmax; k++) {
23482f480046SShri Abhyankar         tmp = jj[nzcountb]/bs;        /* block col. index */
23492f480046SShri Abhyankar         if (tmp >= i) {
23502f480046SShri Abhyankar           block     = mask[tmp] - 1;
23512f480046SShri Abhyankar           point     = jj[nzcountb] - bs*tmp;
23522f480046SShri Abhyankar           idx       = ishift + bs2*block + j + bs*point;
23532f480046SShri Abhyankar           a->a[idx] = aa[nzcountb];
23542f480046SShri Abhyankar         }
23552f480046SShri Abhyankar         nzcountb++;
23562f480046SShri Abhyankar       }
23572f480046SShri Abhyankar     }
23582f480046SShri Abhyankar     /* zero out the mask elements we set */
23592f480046SShri Abhyankar     for (j=0; j<nmask; j++) mask[masked[j]] = 0;
23602f480046SShri Abhyankar   }
23612f480046SShri Abhyankar   if (jcount != a->nz) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Bad binary matrix");
23622f480046SShri Abhyankar 
23632f480046SShri Abhyankar   ierr = PetscFree(rowlengths);CHKERRQ(ierr);
23642f480046SShri Abhyankar   ierr = PetscFree(s_browlengths);CHKERRQ(ierr);
23652f480046SShri Abhyankar   ierr = PetscFree(aa);CHKERRQ(ierr);
23662f480046SShri Abhyankar   ierr = PetscFree(jj);CHKERRQ(ierr);
23672f480046SShri Abhyankar   ierr = PetscFree2(mask,masked);CHKERRQ(ierr);
23682f480046SShri Abhyankar 
23692f480046SShri Abhyankar   ierr = MatAssemblyBegin(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
23702f480046SShri Abhyankar   ierr = MatAssemblyEnd(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
23712f480046SShri Abhyankar   PetscFunctionReturn(0);
23722f480046SShri Abhyankar }
23732f480046SShri Abhyankar 
2374c75a6043SHong Zhang /*@
2375c75a6043SHong Zhang      MatCreateSeqSBAIJWithArrays - Creates an sequential SBAIJ matrix using matrix elements
2376c75a6043SHong Zhang               (upper triangular entries in CSR format) provided by the user.
2377c75a6043SHong Zhang 
2378c75a6043SHong Zhang      Collective on MPI_Comm
2379c75a6043SHong Zhang 
2380c75a6043SHong Zhang    Input Parameters:
2381c75a6043SHong Zhang +  comm - must be an MPI communicator of size 1
2382c75a6043SHong Zhang .  bs - size of block
2383c75a6043SHong Zhang .  m - number of rows
2384c75a6043SHong Zhang .  n - number of columns
2385c75a6043SHong Zhang .  i - row indices
2386c75a6043SHong Zhang .  j - column indices
2387c75a6043SHong Zhang -  a - matrix values
2388c75a6043SHong Zhang 
2389c75a6043SHong Zhang    Output Parameter:
2390c75a6043SHong Zhang .  mat - the matrix
2391c75a6043SHong Zhang 
2392dfb205c3SBarry Smith    Level: advanced
2393c75a6043SHong Zhang 
2394c75a6043SHong Zhang    Notes:
2395c75a6043SHong Zhang        The i, j, and a arrays are not copied by this routine, the user must free these arrays
2396c75a6043SHong Zhang     once the matrix is destroyed
2397c75a6043SHong Zhang 
2398c75a6043SHong Zhang        You cannot set new nonzero locations into this matrix, that will generate an error.
2399c75a6043SHong Zhang 
2400c75a6043SHong Zhang        The i and j indices are 0 based
2401c75a6043SHong Zhang 
2402dfb205c3SBarry 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
2403dfb205c3SBarry Smith        it is the regular CSR format excluding the lower triangular elements.
2404dfb205c3SBarry Smith 
240569b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSBAIJ(), MatCreateSeqSBAIJ()
2406c75a6043SHong Zhang 
2407c75a6043SHong Zhang @*/
2408*c3c607ccSBarry Smith PetscErrorCode  MatCreateSeqSBAIJWithArrays(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt i[],PetscInt j[],PetscScalar a[],Mat *mat)
2409c75a6043SHong Zhang {
2410c75a6043SHong Zhang   PetscErrorCode ierr;
2411c75a6043SHong Zhang   PetscInt       ii;
2412c75a6043SHong Zhang   Mat_SeqSBAIJ   *sbaij;
2413c75a6043SHong Zhang 
2414c75a6043SHong Zhang   PetscFunctionBegin;
2415e32f2f54SBarry Smith   if (bs != 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"block size %D > 1 is not supported yet",bs);
241641096f02SStefano Zampini   if (m > 0 && i[0]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"i (row indices) must start with 0");
2417c75a6043SHong Zhang 
2418c75a6043SHong Zhang   ierr  = MatCreate(comm,mat);CHKERRQ(ierr);
2419c75a6043SHong Zhang   ierr  = MatSetSizes(*mat,m,n,m,n);CHKERRQ(ierr);
2420c75a6043SHong Zhang   ierr  = MatSetType(*mat,MATSEQSBAIJ);CHKERRQ(ierr);
2421367daffbSBarry Smith   ierr  = MatSeqSBAIJSetPreallocation(*mat,bs,MAT_SKIP_ALLOCATION,0);CHKERRQ(ierr);
2422c75a6043SHong Zhang   sbaij = (Mat_SeqSBAIJ*)(*mat)->data;
2423dcca6d9dSJed Brown   ierr  = PetscMalloc2(m,&sbaij->imax,m,&sbaij->ilen);CHKERRQ(ierr);
24243bb1ff40SBarry Smith   ierr  = PetscLogObjectMemory((PetscObject)*mat,2*m*sizeof(PetscInt));CHKERRQ(ierr);
2425c75a6043SHong Zhang 
2426c75a6043SHong Zhang   sbaij->i = i;
2427c75a6043SHong Zhang   sbaij->j = j;
2428c75a6043SHong Zhang   sbaij->a = a;
242926fbe8dcSKarl Rupp 
2430c75a6043SHong Zhang   sbaij->singlemalloc = PETSC_FALSE;
2431c75a6043SHong Zhang   sbaij->nonew        = -1;             /*this indicates that inserting a new value in the matrix that generates a new nonzero is an error*/
2432e6b907acSBarry Smith   sbaij->free_a       = PETSC_FALSE;
2433e6b907acSBarry Smith   sbaij->free_ij      = PETSC_FALSE;
2434c75a6043SHong Zhang 
2435c75a6043SHong Zhang   for (ii=0; ii<m; ii++) {
2436c75a6043SHong Zhang     sbaij->ilen[ii] = sbaij->imax[ii] = i[ii+1] - i[ii];
2437c75a6043SHong Zhang #if defined(PETSC_USE_DEBUG)
2438e32f2f54SBarry 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]);
2439c75a6043SHong Zhang #endif
2440c75a6043SHong Zhang   }
2441c75a6043SHong Zhang #if defined(PETSC_USE_DEBUG)
2442c75a6043SHong Zhang   for (ii=0; ii<sbaij->i[m]; ii++) {
2443e32f2f54SBarry Smith     if (j[ii] < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column index at location = %d index = %d",ii,j[ii]);
2444e32f2f54SBarry 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]);
2445c75a6043SHong Zhang   }
2446c75a6043SHong Zhang #endif
2447c75a6043SHong Zhang 
2448c75a6043SHong Zhang   ierr = MatAssemblyBegin(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2449c75a6043SHong Zhang   ierr = MatAssemblyEnd(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2450c75a6043SHong Zhang   PetscFunctionReturn(0);
2451c75a6043SHong Zhang }
2452d06b337dSHong Zhang 
245359f5e6ceSHong Zhang PetscErrorCode MatCreateMPIMatConcatenateSeqMat_SeqSBAIJ(MPI_Comm comm,Mat inmat,PetscInt n,MatReuse scall,Mat *outmat)
245459f5e6ceSHong Zhang {
245559f5e6ceSHong Zhang   PetscErrorCode ierr;
245659f5e6ceSHong Zhang 
245759f5e6ceSHong Zhang   PetscFunctionBegin;
245859f5e6ceSHong Zhang   ierr = MatCreateMPIMatConcatenateSeqMat_MPISBAIJ(comm,inmat,n,scall,outmat);CHKERRQ(ierr);
245959f5e6ceSHong Zhang   PetscFunctionReturn(0);
246059f5e6ceSHong Zhang }
2461d06b337dSHong Zhang 
2462d06b337dSHong Zhang 
246349b5e25fSSatish Balay 
246449b5e25fSSatish Balay 
2465