xref: /petsc/src/mat/impls/sbaij/seq/sbaij.c (revision 0f2140c79afca2ec0056695d230f077dadf805c7)
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 
146214f412SHong Zhang #if defined(PETSC_HAVE_ELEMENTAL)
15cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatConvert_SeqSBAIJ_Elemental(Mat,MatType,MatReuse,Mat*);
166214f412SHong Zhang #endif
17b5b17502SBarry Smith 
1849b5e25fSSatish Balay /*
1949b5e25fSSatish Balay      Checks for missing diagonals
2049b5e25fSSatish Balay */
21ace3abfcSBarry Smith PetscErrorCode MatMissingDiagonal_SeqSBAIJ(Mat A,PetscBool  *missing,PetscInt *dd)
2249b5e25fSSatish Balay {
23045c9aa0SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
246849ba73SBarry Smith   PetscErrorCode ierr;
257734d3b5SMatthew G. Knepley   PetscInt       *diag,*ii = a->i,i;
2649b5e25fSSatish Balay 
2749b5e25fSSatish Balay   PetscFunctionBegin;
28045c9aa0SHong Zhang   ierr     = MatMarkDiagonal_SeqSBAIJ(A);CHKERRQ(ierr);
292af78befSBarry Smith   *missing = PETSC_FALSE;
307734d3b5SMatthew G. Knepley   if (A->rmap->n > 0 && !ii) {
31358d2f5dSShri Abhyankar     *missing = PETSC_TRUE;
32358d2f5dSShri Abhyankar     if (dd) *dd = 0;
33955c1f14SBarry Smith     ierr = PetscInfo(A,"Matrix has no entries therefore is missing diagonal\n");CHKERRQ(ierr);
34358d2f5dSShri Abhyankar   } else {
35358d2f5dSShri Abhyankar     diag = a->diag;
3649b5e25fSSatish Balay     for (i=0; i<a->mbs; i++) {
377734d3b5SMatthew G. Knepley       if (diag[i] >= ii[i+1]) {
382af78befSBarry Smith         *missing = PETSC_TRUE;
392af78befSBarry Smith         if (dd) *dd = i;
402af78befSBarry Smith         break;
412af78befSBarry Smith       }
4249b5e25fSSatish Balay     }
43358d2f5dSShri Abhyankar   }
4449b5e25fSSatish Balay   PetscFunctionReturn(0);
4549b5e25fSSatish Balay }
4649b5e25fSSatish Balay 
47dfbe8321SBarry Smith PetscErrorCode MatMarkDiagonal_SeqSBAIJ(Mat A)
4849b5e25fSSatish Balay {
49045c9aa0SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
506849ba73SBarry Smith   PetscErrorCode ierr;
5148dd3d27SHong Zhang   PetscInt       i,j;
5249b5e25fSSatish Balay 
5349b5e25fSSatish Balay   PetscFunctionBegin;
5409f38230SBarry Smith   if (!a->diag) {
55785e854fSJed Brown     ierr         = PetscMalloc1(a->mbs,&a->diag);CHKERRQ(ierr);
563bb1ff40SBarry Smith     ierr         = PetscLogObjectMemory((PetscObject)A,a->mbs*sizeof(PetscInt));CHKERRQ(ierr);
57c760cd28SBarry Smith     a->free_diag = PETSC_TRUE;
5809f38230SBarry Smith   }
5948dd3d27SHong Zhang   for (i=0; i<a->mbs; i++) {
6048dd3d27SHong Zhang     a->diag[i] = a->i[i+1];
6148dd3d27SHong Zhang     for (j=a->i[i]; j<a->i[i+1]; j++) {
6248dd3d27SHong Zhang       if (a->j[j] == i) {
6348dd3d27SHong Zhang         a->diag[i] = j;
6448dd3d27SHong Zhang         break;
6548dd3d27SHong Zhang       }
6648dd3d27SHong Zhang     }
6748dd3d27SHong Zhang   }
6849b5e25fSSatish Balay   PetscFunctionReturn(0);
6949b5e25fSSatish Balay }
7049b5e25fSSatish Balay 
711a83f524SJed Brown static PetscErrorCode MatGetRowIJ_SeqSBAIJ(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *nn,const PetscInt *inia[],const PetscInt *inja[],PetscBool  *done)
7249b5e25fSSatish Balay {
73a6ece127SHong Zhang   Mat_SeqSBAIJ    *a = (Mat_SeqSBAIJ*)A->data;
748f7157efSSatish Balay   PetscErrorCode ierr;
752462f5fdSStefano Zampini   PetscInt       i,j,n = a->mbs,nz = a->i[n],*tia,*tja,bs = A->rmap->bs,k,l,cnt;
762462f5fdSStefano Zampini   PetscInt       **ia = (PetscInt**)inia,**ja = (PetscInt**)inja;
7749b5e25fSSatish Balay 
7849b5e25fSSatish Balay   PetscFunctionBegin;
79d3e5a4abSHong Zhang   *nn = n;
80a1373b80SHong Zhang   if (!ia) PetscFunctionReturn(0);
812462f5fdSStefano Zampini   if (symmetric) {
822462f5fdSStefano Zampini     ierr = MatToSymmetricIJ_SeqAIJ(n,a->i,a->j,PETSC_FALSE,0,0,&tia,&tja);CHKERRQ(ierr);
832462f5fdSStefano Zampini     nz   = tia[n];
842462f5fdSStefano Zampini   } else {
852462f5fdSStefano Zampini     tia = a->i; tja = a->j;
862462f5fdSStefano Zampini   }
872462f5fdSStefano Zampini 
882462f5fdSStefano Zampini   if (!blockcompressed && bs > 1) {
892462f5fdSStefano Zampini     (*nn) *= bs;
908f7157efSSatish Balay     /* malloc & create the natural set of indices */
912462f5fdSStefano Zampini     ierr = PetscMalloc1((n+1)*bs,ia);CHKERRQ(ierr);
922462f5fdSStefano Zampini     if (n) {
932462f5fdSStefano Zampini       (*ia)[0] = oshift;
942462f5fdSStefano Zampini       for (j=1; j<bs; j++) {
952462f5fdSStefano Zampini         (*ia)[j] = (tia[1]-tia[0])*bs+(*ia)[j-1];
962462f5fdSStefano Zampini       }
972462f5fdSStefano Zampini     }
982462f5fdSStefano Zampini 
992462f5fdSStefano Zampini     for (i=1; i<n; i++) {
1002462f5fdSStefano Zampini       (*ia)[i*bs] = (tia[i]-tia[i-1])*bs + (*ia)[i*bs-1];
1012462f5fdSStefano Zampini       for (j=1; j<bs; j++) {
1022462f5fdSStefano Zampini         (*ia)[i*bs+j] = (tia[i+1]-tia[i])*bs + (*ia)[i*bs+j-1];
1032462f5fdSStefano Zampini       }
1042462f5fdSStefano Zampini     }
1052462f5fdSStefano Zampini     if (n) {
1062462f5fdSStefano Zampini       (*ia)[n*bs] = (tia[n]-tia[n-1])*bs + (*ia)[n*bs-1];
1072462f5fdSStefano Zampini     }
1082462f5fdSStefano Zampini 
1092462f5fdSStefano Zampini     if (inja) {
1102462f5fdSStefano Zampini       ierr = PetscMalloc1(nz*bs*bs,ja);CHKERRQ(ierr);
1112462f5fdSStefano Zampini       cnt = 0;
1122462f5fdSStefano Zampini       for (i=0; i<n; i++) {
1138f7157efSSatish Balay         for (j=0; j<bs; j++) {
1142462f5fdSStefano Zampini           for (k=tia[i]; k<tia[i+1]; k++) {
1152462f5fdSStefano Zampini             for (l=0; l<bs; l++) {
1162462f5fdSStefano Zampini               (*ja)[cnt++] = bs*tja[k] + l;
1178f7157efSSatish Balay             }
1188f7157efSSatish Balay           }
1198f7157efSSatish Balay         }
1208f7157efSSatish Balay       }
1218f7157efSSatish Balay     }
1222462f5fdSStefano Zampini 
1232462f5fdSStefano Zampini     if (symmetric) { /* deallocate memory allocated in MatToSymmetricIJ_SeqAIJ() */
1242462f5fdSStefano Zampini       ierr = PetscFree(tia);CHKERRQ(ierr);
1252462f5fdSStefano Zampini       ierr = PetscFree(tja);CHKERRQ(ierr);
1262462f5fdSStefano Zampini     }
1272462f5fdSStefano Zampini   } else if (oshift == 1) {
1282462f5fdSStefano Zampini     if (symmetric) {
1292462f5fdSStefano Zampini       nz = tia[A->rmap->n/bs];
1302462f5fdSStefano Zampini       /*  add 1 to i and j indices */
1312462f5fdSStefano Zampini       for (i=0; i<A->rmap->n/bs+1; i++) tia[i] = tia[i] + 1;
1322462f5fdSStefano Zampini       *ia = tia;
1332462f5fdSStefano Zampini       if (ja) {
1342462f5fdSStefano Zampini         for (i=0; i<nz; i++) tja[i] = tja[i] + 1;
1352462f5fdSStefano Zampini         *ja = tja;
1362462f5fdSStefano Zampini       }
1372462f5fdSStefano Zampini     } else {
1382462f5fdSStefano Zampini       nz = a->i[A->rmap->n/bs];
1392462f5fdSStefano Zampini       /* malloc space and  add 1 to i and j indices */
1402462f5fdSStefano Zampini       ierr = PetscMalloc1(A->rmap->n/bs+1,ia);CHKERRQ(ierr);
1412462f5fdSStefano Zampini       for (i=0; i<A->rmap->n/bs+1; i++) (*ia)[i] = a->i[i] + 1;
1422462f5fdSStefano Zampini       if (ja) {
1432462f5fdSStefano Zampini         ierr = PetscMalloc1(nz,ja);CHKERRQ(ierr);
1442462f5fdSStefano Zampini         for (i=0; i<nz; i++) (*ja)[i] = a->j[i] + 1;
1452462f5fdSStefano Zampini       }
1462462f5fdSStefano Zampini     }
1472462f5fdSStefano Zampini   } else {
1482462f5fdSStefano Zampini     *ia = tia;
1492462f5fdSStefano Zampini     if (ja) *ja = tja;
150a6ece127SHong Zhang   }
15149b5e25fSSatish Balay   PetscFunctionReturn(0);
15249b5e25fSSatish Balay }
15349b5e25fSSatish Balay 
1541a83f524SJed Brown static PetscErrorCode MatRestoreRowIJ_SeqSBAIJ(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *nn,const PetscInt *ia[],const PetscInt *ja[],PetscBool  *done)
15549b5e25fSSatish Balay {
1568f7157efSSatish Balay   PetscErrorCode ierr;
157a6ece127SHong Zhang 
15849b5e25fSSatish Balay   PetscFunctionBegin;
15949b5e25fSSatish Balay   if (!ia) PetscFunctionReturn(0);
1602462f5fdSStefano Zampini   if ((!blockcompressed && A->rmap->bs > 1) || (symmetric || oshift == 1)) {
1612462f5fdSStefano Zampini     ierr = PetscFree(*ia);CHKERRQ(ierr);
1622462f5fdSStefano Zampini     if (ja) {ierr = PetscFree(*ja);CHKERRQ(ierr);}
163a6ece127SHong Zhang   }
164a6ece127SHong Zhang   PetscFunctionReturn(0);
16549b5e25fSSatish Balay }
16649b5e25fSSatish Balay 
167dfbe8321SBarry Smith PetscErrorCode MatDestroy_SeqSBAIJ(Mat A)
16849b5e25fSSatish Balay {
16949b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
170dfbe8321SBarry Smith   PetscErrorCode ierr;
17149b5e25fSSatish Balay 
17249b5e25fSSatish Balay   PetscFunctionBegin;
173a9f03627SSatish Balay #if defined(PETSC_USE_LOG)
174d0f46423SBarry Smith   PetscLogObjectState((PetscObject)A,"Rows=%D, NZ=%D",A->rmap->N,a->nz);
175a9f03627SSatish Balay #endif
176e6b907acSBarry Smith   ierr = MatSeqXAIJFreeAIJ(A,&a->a,&a->j,&a->i);CHKERRQ(ierr);
1777f53bb6cSHong Zhang   if (a->free_diag) {ierr = PetscFree(a->diag);CHKERRQ(ierr);}
1786bf464f9SBarry Smith   ierr = ISDestroy(&a->row);CHKERRQ(ierr);
1796bf464f9SBarry Smith   ierr = ISDestroy(&a->col);CHKERRQ(ierr);
1806bf464f9SBarry Smith   ierr = ISDestroy(&a->icol);CHKERRQ(ierr);
181c31cb41cSBarry Smith   ierr = PetscFree(a->idiag);CHKERRQ(ierr);
182c31cb41cSBarry Smith   ierr = PetscFree(a->inode.size);CHKERRQ(ierr);
183c760cd28SBarry Smith   if (a->free_imax_ilen) {ierr = PetscFree2(a->imax,a->ilen);CHKERRQ(ierr);}
18405b42c5fSBarry Smith   ierr = PetscFree(a->solve_work);CHKERRQ(ierr);
18541f059aeSBarry Smith   ierr = PetscFree(a->sor_work);CHKERRQ(ierr);
18605b42c5fSBarry Smith   ierr = PetscFree(a->solves_work);CHKERRQ(ierr);
18705b42c5fSBarry Smith   ierr = PetscFree(a->mult_work);CHKERRQ(ierr);
18805b42c5fSBarry Smith   ierr = PetscFree(a->saved_values);CHKERRQ(ierr);
1894da8f245SBarry Smith   if (a->free_jshort) {ierr = PetscFree(a->jshort);CHKERRQ(ierr);}
1901a3463dfSHong Zhang   ierr = PetscFree(a->inew);CHKERRQ(ierr);
1916bf464f9SBarry Smith   ierr = MatDestroy(&a->parent);CHKERRQ(ierr);
192bf0cc555SLisandro Dalcin   ierr = PetscFree(A->data);CHKERRQ(ierr);
193901853e0SKris Buschelman 
194dbd8c25aSHong Zhang   ierr = PetscObjectChangeTypeName((PetscObject)A,0);CHKERRQ(ierr);
195bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatStoreValues_C",NULL);CHKERRQ(ierr);
196bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatRetrieveValues_C",NULL);CHKERRQ(ierr);
197bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetColumnIndices_C",NULL);CHKERRQ(ierr);
198bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqaij_C",NULL);CHKERRQ(ierr);
199bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqbaij_C",NULL);CHKERRQ(ierr);
200bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetPreallocation_C",NULL);CHKERRQ(ierr);
20138f409ebSLisandro Dalcin   ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetPreallocationCSR_C",NULL);CHKERRQ(ierr);
202bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqsbstrm_C",NULL);CHKERRQ(ierr);
2036214f412SHong Zhang #if defined(PETSC_HAVE_ELEMENTAL)
2046214f412SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_elemental_C",NULL);CHKERRQ(ierr);
2056214f412SHong Zhang #endif
20649b5e25fSSatish Balay   PetscFunctionReturn(0);
20749b5e25fSSatish Balay }
20849b5e25fSSatish Balay 
209ace3abfcSBarry Smith PetscErrorCode MatSetOption_SeqSBAIJ(Mat A,MatOption op,PetscBool flg)
21049b5e25fSSatish Balay {
211045c9aa0SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
21263ba0a88SBarry Smith   PetscErrorCode ierr;
21349b5e25fSSatish Balay 
21449b5e25fSSatish Balay   PetscFunctionBegin;
2154d9d31abSKris Buschelman   switch (op) {
2164d9d31abSKris Buschelman   case MAT_ROW_ORIENTED:
2174e0d8c25SBarry Smith     a->roworiented = flg;
2184d9d31abSKris Buschelman     break;
219a9817697SBarry Smith   case MAT_KEEP_NONZERO_PATTERN:
220a9817697SBarry Smith     a->keepnonzeropattern = flg;
2214d9d31abSKris Buschelman     break;
222512a5fc5SBarry Smith   case MAT_NEW_NONZERO_LOCATIONS:
223512a5fc5SBarry Smith     a->nonew = (flg ? 0 : 1);
2244d9d31abSKris Buschelman     break;
2254d9d31abSKris Buschelman   case MAT_NEW_NONZERO_LOCATION_ERR:
2264e0d8c25SBarry Smith     a->nonew = (flg ? -1 : 0);
2274d9d31abSKris Buschelman     break;
2284d9d31abSKris Buschelman   case MAT_NEW_NONZERO_ALLOCATION_ERR:
2294e0d8c25SBarry Smith     a->nonew = (flg ? -2 : 0);
2304d9d31abSKris Buschelman     break;
23128b2fa4aSMatthew Knepley   case MAT_UNUSED_NONZERO_LOCATION_ERR:
23228b2fa4aSMatthew Knepley     a->nounused = (flg ? -1 : 0);
23328b2fa4aSMatthew Knepley     break;
2344e0d8c25SBarry Smith   case MAT_NEW_DIAGONALS:
2354d9d31abSKris Buschelman   case MAT_IGNORE_OFF_PROC_ENTRIES:
2364d9d31abSKris Buschelman   case MAT_USE_HASH_TABLE:
237290bbb0aSBarry Smith     ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr);
2384d9d31abSKris Buschelman     break;
2399a4540c5SBarry Smith   case MAT_HERMITIAN:
240*0f2140c7SStefano Zampini #if defined(PETSC_USE_COMPLEX) /* MAT_HERMITIAN is a synonym for MAT_SYMMETRIC with reals */
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");
247*0f2140c7SStefano Zampini #endif
248eeffb40dSHong Zhang     break;
2493d472b54SHong Zhang   case MAT_SPD:
2505021d80fSJed Brown     /* These options are handled directly by MatSetOption() */
2513d472b54SHong Zhang     break;
25277e54ba9SKris Buschelman   case MAT_SYMMETRIC:
25377e54ba9SKris Buschelman   case MAT_STRUCTURALLY_SYMMETRIC:
2549a4540c5SBarry Smith   case MAT_SYMMETRY_ETERNAL:
2554dcd73b1SHong Zhang     /* These options are handled directly by MatSetOption() */
256290bbb0aSBarry Smith     break;
257941593c8SHong Zhang   case MAT_IGNORE_LOWER_TRIANGULAR:
2584e0d8c25SBarry Smith     a->ignore_ltriangular = flg;
259941593c8SHong Zhang     break;
260941593c8SHong Zhang   case MAT_ERROR_LOWER_TRIANGULAR:
2614e0d8c25SBarry Smith     a->ignore_ltriangular = flg;
26277e54ba9SKris Buschelman     break;
263f5edf698SHong Zhang   case MAT_GETROW_UPPERTRIANGULAR:
2644e0d8c25SBarry Smith     a->getrow_utriangular = flg;
265f5edf698SHong Zhang     break;
266c10200c1SHong Zhang   case MAT_SUBMAT_SINGLEIS:
267c10200c1SHong Zhang     break;
2684d9d31abSKris Buschelman   default:
269e32f2f54SBarry Smith     SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"unknown option %d",op);
27049b5e25fSSatish Balay   }
27149b5e25fSSatish Balay   PetscFunctionReturn(0);
27249b5e25fSSatish Balay }
27349b5e25fSSatish Balay 
27452768537SHong Zhang PetscErrorCode MatGetRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
27549b5e25fSSatish Balay {
27649b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
2776849ba73SBarry Smith   PetscErrorCode ierr;
27849b5e25fSSatish Balay 
27949b5e25fSSatish Balay   PetscFunctionBegin;
280e32f2f54SBarry 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()");
28152768537SHong Zhang 
282f5edf698SHong Zhang   /* Get the upper triangular part of the row */
28352768537SHong Zhang   ierr = MatGetRow_SeqBAIJ_private(A,row,nz,idx,v,a->i,a->j,a->a);CHKERRQ(ierr);
28449b5e25fSSatish Balay   PetscFunctionReturn(0);
28549b5e25fSSatish Balay }
28649b5e25fSSatish Balay 
28713f74950SBarry Smith PetscErrorCode MatRestoreRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
28849b5e25fSSatish Balay {
289dfbe8321SBarry Smith   PetscErrorCode ierr;
29049b5e25fSSatish Balay 
29149b5e25fSSatish Balay   PetscFunctionBegin;
29205b42c5fSBarry Smith   if (idx) {ierr = PetscFree(*idx);CHKERRQ(ierr);}
29305b42c5fSBarry Smith   if (v)   {ierr = PetscFree(*v);CHKERRQ(ierr);}
29449b5e25fSSatish Balay   PetscFunctionReturn(0);
29549b5e25fSSatish Balay }
29649b5e25fSSatish Balay 
297f5edf698SHong Zhang PetscErrorCode MatGetRowUpperTriangular_SeqSBAIJ(Mat A)
298f5edf698SHong Zhang {
299f5edf698SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
300f5edf698SHong Zhang 
301f5edf698SHong Zhang   PetscFunctionBegin;
302f5edf698SHong Zhang   a->getrow_utriangular = PETSC_TRUE;
303f5edf698SHong Zhang   PetscFunctionReturn(0);
304f5edf698SHong Zhang }
305f5edf698SHong Zhang PetscErrorCode MatRestoreRowUpperTriangular_SeqSBAIJ(Mat A)
306f5edf698SHong Zhang {
307f5edf698SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
308f5edf698SHong Zhang 
309f5edf698SHong Zhang   PetscFunctionBegin;
310f5edf698SHong Zhang   a->getrow_utriangular = PETSC_FALSE;
311f5edf698SHong Zhang   PetscFunctionReturn(0);
312f5edf698SHong Zhang }
313f5edf698SHong Zhang 
314fc4dec0aSBarry Smith PetscErrorCode MatTranspose_SeqSBAIJ(Mat A,MatReuse reuse,Mat *B)
31549b5e25fSSatish Balay {
316dfbe8321SBarry Smith   PetscErrorCode ierr;
3175fd66863SKarl Rupp 
31849b5e25fSSatish Balay   PetscFunctionBegin;
319cf37664fSBarry Smith   if (reuse == MAT_INITIAL_MATRIX) {
320999d9058SBarry Smith     ierr = MatDuplicate(A,MAT_COPY_VALUES,B);CHKERRQ(ierr);
321cf37664fSBarry Smith   } else if (reuse == MAT_REUSE_MATRIX) {
322cf37664fSBarry Smith     ierr = MatCopy(A,*B,SAME_NONZERO_PATTERN);CHKERRQ(ierr);
323fc4dec0aSBarry Smith   }
3248115998fSBarry Smith   PetscFunctionReturn(0);
32549b5e25fSSatish Balay }
32649b5e25fSSatish Balay 
3277da1fb6eSBarry Smith PetscErrorCode MatView_SeqSBAIJ_ASCII(Mat A,PetscViewer viewer)
32849b5e25fSSatish Balay {
32949b5e25fSSatish Balay   Mat_SeqSBAIJ      *a = (Mat_SeqSBAIJ*)A->data;
330dfbe8321SBarry Smith   PetscErrorCode    ierr;
331d0f46423SBarry Smith   PetscInt          i,j,bs = A->rmap->bs,k,l,bs2=a->bs2;
332f3ef73ceSBarry Smith   PetscViewerFormat format;
333121deb67SSatish Balay   PetscInt          *diag;
33449b5e25fSSatish Balay 
33549b5e25fSSatish Balay   PetscFunctionBegin;
336b0a32e0cSBarry Smith   ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
337456192e2SBarry Smith   if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
33877431f27SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  block size is %D\n",bs);CHKERRQ(ierr);
339fb9695e5SSatish Balay   } else if (format == PETSC_VIEWER_ASCII_MATLAB) {
340d2507d54SMatthew Knepley     Mat        aij;
341ade3a672SBarry Smith     const char *matname;
342ade3a672SBarry Smith 
343d5f3da31SBarry Smith     if (A->factortype && bs>1) {
34470d5e725SHong 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);
34570d5e725SHong Zhang       PetscFunctionReturn(0);
34670d5e725SHong Zhang     }
347c9f458caSMatthew Knepley     ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&aij);CHKERRQ(ierr);
348ade3a672SBarry Smith     ierr = PetscObjectGetName((PetscObject)A,&matname);CHKERRQ(ierr);
349ade3a672SBarry Smith     ierr = PetscObjectSetName((PetscObject)aij,matname);CHKERRQ(ierr);
350c9f458caSMatthew Knepley     ierr = MatView(aij,viewer);CHKERRQ(ierr);
3516bf464f9SBarry Smith     ierr = MatDestroy(&aij);CHKERRQ(ierr);
352fb9695e5SSatish Balay   } else if (format == PETSC_VIEWER_ASCII_COMMON) {
353d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
35449b5e25fSSatish Balay     for (i=0; i<a->mbs; i++) {
35549b5e25fSSatish Balay       for (j=0; j<bs; j++) {
35677431f27SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr);
35749b5e25fSSatish Balay         for (k=a->i[i]; k<a->i[i+1]; k++) {
35849b5e25fSSatish Balay           for (l=0; l<bs; l++) {
35949b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX)
36049b5e25fSSatish Balay             if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
36157622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k]+l,
36257622a8eSBarry Smith                                             (double)PetscRealPart(a->a[bs2*k + l*bs + j]),(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
36349b5e25fSSatish Balay             } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
36457622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k]+l,
36557622a8eSBarry Smith                                             (double)PetscRealPart(a->a[bs2*k + l*bs + j]),-(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
36649b5e25fSSatish Balay             } else if (PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
36757622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
36849b5e25fSSatish Balay             }
36949b5e25fSSatish Balay #else
37049b5e25fSSatish Balay             if (a->a[bs2*k + l*bs + j] != 0.0) {
37157622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)a->a[bs2*k + l*bs + j]);CHKERRQ(ierr);
37249b5e25fSSatish Balay             }
37349b5e25fSSatish Balay #endif
37449b5e25fSSatish Balay           }
37549b5e25fSSatish Balay         }
376b0a32e0cSBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
37749b5e25fSSatish Balay       }
37849b5e25fSSatish Balay     }
379d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
380c1490034SHong Zhang   } else if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) {
381c1490034SHong Zhang     PetscFunctionReturn(0);
38249b5e25fSSatish Balay   } else {
383d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
3842c990fa1SHong Zhang     if (A->factortype) { /* for factored matrix */
3852c990fa1SHong Zhang       if (bs>1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"matrix is factored with bs>1. Not implemented yet");
3862c990fa1SHong Zhang 
387121deb67SSatish Balay       diag=a->diag;
388121deb67SSatish Balay       for (i=0; i<a->mbs; i++) { /* for row block i */
3892c990fa1SHong Zhang         ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i);CHKERRQ(ierr);
3902c990fa1SHong Zhang         /* diagonal entry */
3912c990fa1SHong Zhang #if defined(PETSC_USE_COMPLEX)
3922c990fa1SHong Zhang         if (PetscImaginaryPart(a->a[diag[i]]) > 0.0) {
39357622a8eSBarry 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);
3942c990fa1SHong Zhang         } else if (PetscImaginaryPart(a->a[diag[i]]) < 0.0) {
39557622a8eSBarry 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);
3962c990fa1SHong Zhang         } else {
39757622a8eSBarry Smith           ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[diag[i]],(double)PetscRealPart(1.0/a->a[diag[i]]));CHKERRQ(ierr);
3982c990fa1SHong Zhang         }
3992c990fa1SHong Zhang #else
4006712e2f1SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[diag[i]],(double)(1.0/a->a[diag[i]]));CHKERRQ(ierr);
4012c990fa1SHong Zhang #endif
4022c990fa1SHong Zhang         /* off-diagonal entries */
4032c990fa1SHong Zhang         for (k=a->i[i]; k<a->i[i+1]-1; k++) {
4042c990fa1SHong Zhang #if defined(PETSC_USE_COMPLEX)
405ca0704adSBarry Smith           if (PetscImaginaryPart(a->a[k]) > 0.0) {
40657622a8eSBarry Smith             ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k],(double)PetscRealPart(a->a[k]),(double)PetscImaginaryPart(a->a[k]));CHKERRQ(ierr);
407ca0704adSBarry Smith           } else if (PetscImaginaryPart(a->a[k]) < 0.0) {
40857622a8eSBarry Smith             ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k],(double)PetscRealPart(a->a[k]),-(double)PetscImaginaryPart(a->a[k]));CHKERRQ(ierr);
4092c990fa1SHong Zhang           } else {
41057622a8eSBarry Smith             ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k],(double)PetscRealPart(a->a[k]));CHKERRQ(ierr);
4112c990fa1SHong Zhang           }
4122c990fa1SHong Zhang #else
41357622a8eSBarry Smith           ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[k],(double)a->a[k]);CHKERRQ(ierr);
4142c990fa1SHong Zhang #endif
4152c990fa1SHong Zhang         }
4162c990fa1SHong Zhang         ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
4172c990fa1SHong Zhang       }
4182c990fa1SHong Zhang 
4192c990fa1SHong Zhang     } else { /* for non-factored matrix */
4200c74a584SJed Brown       for (i=0; i<a->mbs; i++) { /* for row block i */
4210c74a584SJed Brown         for (j=0; j<bs; j++) {   /* for row bs*i + j */
42277431f27SBarry Smith           ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr);
4230c74a584SJed Brown           for (k=a->i[i]; k<a->i[i+1]; k++) { /* for column block */
4240c74a584SJed Brown             for (l=0; l<bs; l++) {            /* for column */
42549b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX)
42649b5e25fSSatish Balay               if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0) {
42757622a8eSBarry Smith                 ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k]+l,
42857622a8eSBarry Smith                                               (double)PetscRealPart(a->a[bs2*k + l*bs + j]),(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
42949b5e25fSSatish Balay               } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0) {
43057622a8eSBarry Smith                 ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k]+l,
43157622a8eSBarry Smith                                               (double)PetscRealPart(a->a[bs2*k + l*bs + j]),-(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
43249b5e25fSSatish Balay               } else {
43357622a8eSBarry Smith                 ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
43449b5e25fSSatish Balay               }
43549b5e25fSSatish Balay #else
43657622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)a->a[bs2*k + l*bs + j]);CHKERRQ(ierr);
43749b5e25fSSatish Balay #endif
43849b5e25fSSatish Balay             }
43949b5e25fSSatish Balay           }
440b0a32e0cSBarry Smith           ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
44149b5e25fSSatish Balay         }
44249b5e25fSSatish Balay       }
4432c990fa1SHong Zhang     }
444d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
44549b5e25fSSatish Balay   }
446b0a32e0cSBarry Smith   ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
44749b5e25fSSatish Balay   PetscFunctionReturn(0);
44849b5e25fSSatish Balay }
44949b5e25fSSatish Balay 
4509804daf3SBarry Smith #include <petscdraw.h>
4516849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw_Zoom(PetscDraw draw,void *Aa)
45249b5e25fSSatish Balay {
45349b5e25fSSatish Balay   Mat            A = (Mat) Aa;
45449b5e25fSSatish Balay   Mat_SeqSBAIJ   *a=(Mat_SeqSBAIJ*)A->data;
4556849ba73SBarry Smith   PetscErrorCode ierr;
456d0f46423SBarry Smith   PetscInt       row,i,j,k,l,mbs=a->mbs,color,bs=A->rmap->bs,bs2=a->bs2;
45749b5e25fSSatish Balay   PetscReal      xl,yl,xr,yr,x_l,x_r,y_l,y_r;
45849b5e25fSSatish Balay   MatScalar      *aa;
459b0a32e0cSBarry Smith   PetscViewer    viewer;
46049b5e25fSSatish Balay 
46149b5e25fSSatish Balay   PetscFunctionBegin;
46249b5e25fSSatish Balay   ierr = PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer);CHKERRQ(ierr);
463b0a32e0cSBarry Smith   ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr);
46449b5e25fSSatish Balay 
46549b5e25fSSatish Balay   /* loop over matrix elements drawing boxes */
466383922c3SLisandro Dalcin 
467383922c3SLisandro Dalcin   ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr);
468383922c3SLisandro Dalcin   ierr = PetscDrawString(draw, .3*(xl+xr), .3*(yl+yr), PETSC_DRAW_BLACK, "symmetric");CHKERRQ(ierr);
469383922c3SLisandro Dalcin   /* Blue for negative, Cyan for zero and  Red for positive */
470b0a32e0cSBarry Smith   color = PETSC_DRAW_BLUE;
47149b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
47249b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
473d0f46423SBarry Smith       y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0;
47449b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
47549b5e25fSSatish Balay       aa  = a->a + j*bs2;
47649b5e25fSSatish Balay       for (k=0; k<bs; k++) {
47749b5e25fSSatish Balay         for (l=0; l<bs; l++) {
47849b5e25fSSatish Balay           if (PetscRealPart(*aa++) >=  0.) continue;
479b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
48049b5e25fSSatish Balay         }
48149b5e25fSSatish Balay       }
48249b5e25fSSatish Balay     }
48349b5e25fSSatish Balay   }
484b0a32e0cSBarry Smith   color = PETSC_DRAW_CYAN;
48549b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
48649b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
487d0f46423SBarry Smith       y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0;
48849b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
48949b5e25fSSatish Balay       aa = a->a + j*bs2;
49049b5e25fSSatish Balay       for (k=0; k<bs; k++) {
49149b5e25fSSatish Balay         for (l=0; l<bs; l++) {
49249b5e25fSSatish Balay           if (PetscRealPart(*aa++) != 0.) continue;
493b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
49449b5e25fSSatish Balay         }
49549b5e25fSSatish Balay       }
49649b5e25fSSatish Balay     }
49749b5e25fSSatish Balay   }
498b0a32e0cSBarry Smith   color = PETSC_DRAW_RED;
49949b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
50049b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
501d0f46423SBarry Smith       y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0;
50249b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
50349b5e25fSSatish Balay       aa = a->a + j*bs2;
50449b5e25fSSatish Balay       for (k=0; k<bs; k++) {
50549b5e25fSSatish Balay         for (l=0; l<bs; l++) {
50649b5e25fSSatish Balay           if (PetscRealPart(*aa++) <= 0.) continue;
507b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
50849b5e25fSSatish Balay         }
50949b5e25fSSatish Balay       }
51049b5e25fSSatish Balay     }
51149b5e25fSSatish Balay   }
512383922c3SLisandro Dalcin   ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr);
51349b5e25fSSatish Balay   PetscFunctionReturn(0);
51449b5e25fSSatish Balay }
51549b5e25fSSatish Balay 
5166849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw(Mat A,PetscViewer viewer)
51749b5e25fSSatish Balay {
518dfbe8321SBarry Smith   PetscErrorCode ierr;
51949b5e25fSSatish Balay   PetscReal      xl,yl,xr,yr,w,h;
520b0a32e0cSBarry Smith   PetscDraw      draw;
521ace3abfcSBarry Smith   PetscBool      isnull;
52249b5e25fSSatish Balay 
52349b5e25fSSatish Balay   PetscFunctionBegin;
524b0a32e0cSBarry Smith   ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
525383922c3SLisandro Dalcin   ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr);
526383922c3SLisandro Dalcin   if (isnull) PetscFunctionReturn(0);
52749b5e25fSSatish Balay 
528d0f46423SBarry Smith   xr   = A->rmap->N; yr = A->rmap->N; h = yr/10.0; w = xr/10.0;
52949b5e25fSSatish Balay   xr  += w;          yr += h;        xl = -w;     yl = -h;
530b0a32e0cSBarry Smith   ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr);
531832b7cebSLisandro Dalcin   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer);CHKERRQ(ierr);
532b0a32e0cSBarry Smith   ierr = PetscDrawZoom(draw,MatView_SeqSBAIJ_Draw_Zoom,A);CHKERRQ(ierr);
5330298fd71SBarry Smith   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",NULL);CHKERRQ(ierr);
534832b7cebSLisandro Dalcin   ierr = PetscDrawSave(draw);CHKERRQ(ierr);
53549b5e25fSSatish Balay   PetscFunctionReturn(0);
53649b5e25fSSatish Balay }
53749b5e25fSSatish Balay 
538dfbe8321SBarry Smith PetscErrorCode MatView_SeqSBAIJ(Mat A,PetscViewer viewer)
53949b5e25fSSatish Balay {
540dfbe8321SBarry Smith   PetscErrorCode ierr;
541ace3abfcSBarry Smith   PetscBool      iascii,isdraw;
54208917f38SBarry Smith   FILE           *file = 0;
54349b5e25fSSatish Balay 
54449b5e25fSSatish Balay   PetscFunctionBegin;
545251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
546251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr);
54732077d6dSBarry Smith   if (iascii) {
54849b5e25fSSatish Balay     ierr = MatView_SeqSBAIJ_ASCII(A,viewer);CHKERRQ(ierr);
54949b5e25fSSatish Balay   } else if (isdraw) {
55049b5e25fSSatish Balay     ierr = MatView_SeqSBAIJ_Draw(A,viewer);CHKERRQ(ierr);
55149b5e25fSSatish Balay   } else {
552a5e6ed63SBarry Smith     Mat        B;
553ade3a672SBarry Smith     const char *matname;
554ceb03754SKris Buschelman     ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&B);CHKERRQ(ierr);
555ade3a672SBarry Smith     ierr = PetscObjectGetName((PetscObject)A,&matname);CHKERRQ(ierr);
556ade3a672SBarry Smith     ierr = PetscObjectSetName((PetscObject)B,matname);CHKERRQ(ierr);
557a5e6ed63SBarry Smith     ierr = MatView(B,viewer);CHKERRQ(ierr);
5586bf464f9SBarry Smith     ierr = MatDestroy(&B);CHKERRQ(ierr);
55908917f38SBarry Smith     ierr = PetscViewerBinaryGetInfoPointer(viewer,&file);CHKERRQ(ierr);
56008917f38SBarry Smith     if (file) {
56108917f38SBarry Smith       fprintf(file,"-matload_block_size %d\n",(int)A->rmap->bs);
56208917f38SBarry Smith     }
56349b5e25fSSatish Balay   }
56449b5e25fSSatish Balay   PetscFunctionReturn(0);
56549b5e25fSSatish Balay }
56649b5e25fSSatish Balay 
56749b5e25fSSatish Balay 
56813f74950SBarry Smith PetscErrorCode MatGetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],PetscScalar v[])
56949b5e25fSSatish Balay {
570045c9aa0SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
57113f74950SBarry Smith   PetscInt     *rp,k,low,high,t,row,nrow,i,col,l,*aj = a->j;
57213f74950SBarry Smith   PetscInt     *ai = a->i,*ailen = a->ilen;
573d0f46423SBarry Smith   PetscInt     brow,bcol,ridx,cidx,bs=A->rmap->bs,bs2=a->bs2;
57497e567efSBarry Smith   MatScalar    *ap,*aa = a->a;
57549b5e25fSSatish Balay 
57649b5e25fSSatish Balay   PetscFunctionBegin;
57749b5e25fSSatish Balay   for (k=0; k<m; k++) { /* loop over rows */
57849b5e25fSSatish Balay     row = im[k]; brow = row/bs;
579e32f2f54SBarry Smith     if (row < 0) {v += n; continue;} /* SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative row: %D",row); */
580e32f2f54SBarry 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);
58149b5e25fSSatish Balay     rp   = aj + ai[brow]; ap = aa + bs2*ai[brow];
58249b5e25fSSatish Balay     nrow = ailen[brow];
58349b5e25fSSatish Balay     for (l=0; l<n; l++) { /* loop over columns */
584e32f2f54SBarry Smith       if (in[l] < 0) {v++; continue;} /* SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column: %D",in[l]); */
585e32f2f54SBarry 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);
58649b5e25fSSatish Balay       col  = in[l];
58749b5e25fSSatish Balay       bcol = col/bs;
58849b5e25fSSatish Balay       cidx = col%bs;
58949b5e25fSSatish Balay       ridx = row%bs;
59049b5e25fSSatish Balay       high = nrow;
59149b5e25fSSatish Balay       low  = 0; /* assume unsorted */
59249b5e25fSSatish Balay       while (high-low > 5) {
59349b5e25fSSatish Balay         t = (low+high)/2;
59449b5e25fSSatish Balay         if (rp[t] > bcol) high = t;
59549b5e25fSSatish Balay         else              low  = t;
59649b5e25fSSatish Balay       }
59749b5e25fSSatish Balay       for (i=low; i<high; i++) {
59849b5e25fSSatish Balay         if (rp[i] > bcol) break;
59949b5e25fSSatish Balay         if (rp[i] == bcol) {
60049b5e25fSSatish Balay           *v++ = ap[bs2*i+bs*cidx+ridx];
60149b5e25fSSatish Balay           goto finished;
60249b5e25fSSatish Balay         }
60349b5e25fSSatish Balay       }
60497e567efSBarry Smith       *v++ = 0.0;
60549b5e25fSSatish Balay finished:;
60649b5e25fSSatish Balay     }
60749b5e25fSSatish Balay   }
60849b5e25fSSatish Balay   PetscFunctionReturn(0);
60949b5e25fSSatish Balay }
61049b5e25fSSatish Balay 
61149b5e25fSSatish Balay 
61213f74950SBarry Smith PetscErrorCode MatSetValuesBlocked_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is)
61349b5e25fSSatish Balay {
6140880e062SHong Zhang   Mat_SeqSBAIJ      *a = (Mat_SeqSBAIJ*)A->data;
6156849ba73SBarry Smith   PetscErrorCode    ierr;
616e2ee6c50SBarry Smith   PetscInt          *rp,k,low,high,t,ii,jj,row,nrow,i,col,l,rmax,N,lastcol = -1;
61713f74950SBarry Smith   PetscInt          *imax      =a->imax,*ai=a->i,*ailen=a->ilen;
618d0f46423SBarry Smith   PetscInt          *aj        =a->j,nonew=a->nonew,bs2=a->bs2,bs=A->rmap->bs,stepval;
619ace3abfcSBarry Smith   PetscBool         roworiented=a->roworiented;
620dd6ea824SBarry Smith   const PetscScalar *value     = v;
621f15d580aSBarry Smith   MatScalar         *ap,*aa = a->a,*bap;
6220880e062SHong Zhang 
62349b5e25fSSatish Balay   PetscFunctionBegin;
62426fbe8dcSKarl Rupp   if (roworiented) stepval = (n-1)*bs;
62526fbe8dcSKarl Rupp   else stepval = (m-1)*bs;
62626fbe8dcSKarl Rupp 
6270880e062SHong Zhang   for (k=0; k<m; k++) { /* loop over added rows */
6280880e062SHong Zhang     row = im[k];
6290880e062SHong Zhang     if (row < 0) continue;
6302515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
6312f7d4af7SBarry 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);
6320880e062SHong Zhang #endif
6330880e062SHong Zhang     rp   = aj + ai[row];
6340880e062SHong Zhang     ap   = aa + bs2*ai[row];
6350880e062SHong Zhang     rmax = imax[row];
6360880e062SHong Zhang     nrow = ailen[row];
6370880e062SHong Zhang     low  = 0;
638818f2c47SBarry Smith     high = nrow;
6390880e062SHong Zhang     for (l=0; l<n; l++) { /* loop over added columns */
6400880e062SHong Zhang       if (in[l] < 0) continue;
6410880e062SHong Zhang       col = in[l];
6422515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
6432f7d4af7SBarry 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);
644b1823623SSatish Balay #endif
645b98bf0e1SJed Brown       if (col < row) {
64626fbe8dcSKarl Rupp         if (a->ignore_ltriangular) continue; /* ignore lower triangular block */
64726fbe8dcSKarl 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)");
648b98bf0e1SJed Brown       }
64926fbe8dcSKarl Rupp       if (roworiented) value = v + k*(stepval+bs)*bs + l*bs;
65026fbe8dcSKarl Rupp       else value = v + l*(stepval+bs)*bs + k*bs;
65126fbe8dcSKarl Rupp 
65226fbe8dcSKarl Rupp       if (col <= lastcol) low = 0;
65326fbe8dcSKarl Rupp       else high = nrow;
65426fbe8dcSKarl Rupp 
655e2ee6c50SBarry Smith       lastcol = col;
6560880e062SHong Zhang       while (high-low > 7) {
6570880e062SHong Zhang         t = (low+high)/2;
6580880e062SHong Zhang         if (rp[t] > col) high = t;
6590880e062SHong Zhang         else             low  = t;
6600880e062SHong Zhang       }
6610880e062SHong Zhang       for (i=low; i<high; i++) {
6620880e062SHong Zhang         if (rp[i] > col) break;
6630880e062SHong Zhang         if (rp[i] == col) {
6640880e062SHong Zhang           bap = ap +  bs2*i;
6650880e062SHong Zhang           if (roworiented) {
6660880e062SHong Zhang             if (is == ADD_VALUES) {
6670880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
6680880e062SHong Zhang                 for (jj=ii; jj<bs2; jj+=bs) {
6690880e062SHong Zhang                   bap[jj] += *value++;
6700880e062SHong Zhang                 }
6710880e062SHong Zhang               }
6720880e062SHong Zhang             } else {
6730880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
6740880e062SHong Zhang                 for (jj=ii; jj<bs2; jj+=bs) {
6750880e062SHong Zhang                   bap[jj] = *value++;
6760880e062SHong Zhang                 }
6770880e062SHong Zhang                }
6780880e062SHong Zhang             }
6790880e062SHong Zhang           } else {
6800880e062SHong Zhang             if (is == ADD_VALUES) {
6810880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
6820880e062SHong Zhang                 for (jj=0; jj<bs; jj++) {
6830880e062SHong Zhang                   *bap++ += *value++;
6840880e062SHong Zhang                 }
6850880e062SHong Zhang               }
6860880e062SHong Zhang             } else {
6870880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
6880880e062SHong Zhang                 for (jj=0; jj<bs; jj++) {
6890880e062SHong Zhang                   *bap++  = *value++;
6900880e062SHong Zhang                 }
6910880e062SHong Zhang               }
6920880e062SHong Zhang             }
6930880e062SHong Zhang           }
6940880e062SHong Zhang           goto noinsert2;
6950880e062SHong Zhang         }
6960880e062SHong Zhang       }
6970880e062SHong Zhang       if (nonew == 1) goto noinsert2;
6982f7d4af7SBarry 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);
699fef13f97SBarry Smith       MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,row,col,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar);
700c03d1d03SSatish Balay       N = nrow++ - 1; high++;
7010880e062SHong Zhang       /* shift up all the later entries in this row */
7020880e062SHong Zhang       for (ii=N; ii>=i; ii--) {
7030880e062SHong Zhang         rp[ii+1] = rp[ii];
7040880e062SHong Zhang         ierr     = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr);
7050880e062SHong Zhang       }
7060880e062SHong Zhang       if (N >= i) {
7070880e062SHong Zhang         ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr);
7080880e062SHong Zhang       }
7090880e062SHong Zhang       rp[i] = col;
7100880e062SHong Zhang       bap   = ap +  bs2*i;
7110880e062SHong Zhang       if (roworiented) {
7120880e062SHong Zhang         for (ii=0; ii<bs; ii++,value+=stepval) {
7130880e062SHong Zhang           for (jj=ii; jj<bs2; jj+=bs) {
7140880e062SHong Zhang             bap[jj] = *value++;
7150880e062SHong Zhang           }
7160880e062SHong Zhang         }
7170880e062SHong Zhang       } else {
7180880e062SHong Zhang         for (ii=0; ii<bs; ii++,value+=stepval) {
7190880e062SHong Zhang           for (jj=0; jj<bs; jj++) {
7200880e062SHong Zhang             *bap++ = *value++;
7210880e062SHong Zhang           }
7220880e062SHong Zhang         }
7230880e062SHong Zhang        }
7240880e062SHong Zhang     noinsert2:;
7250880e062SHong Zhang       low = i;
7260880e062SHong Zhang     }
7270880e062SHong Zhang     ailen[row] = nrow;
7280880e062SHong Zhang   }
7290880e062SHong Zhang   PetscFunctionReturn(0);
73049b5e25fSSatish Balay }
73149b5e25fSSatish Balay 
73264831d72SBarry Smith /*
73364831d72SBarry Smith     This is not yet used
73464831d72SBarry Smith */
7354108e4d5SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ_SeqAIJ_Inode(Mat A)
7360def2e27SBarry Smith {
7370def2e27SBarry Smith   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
7380def2e27SBarry Smith   PetscErrorCode ierr;
7390def2e27SBarry Smith   const PetscInt *ai = a->i, *aj = a->j,*cols;
7400def2e27SBarry Smith   PetscInt       i   = 0,j,blk_size,m = A->rmap->n,node_count = 0,nzx,nzy,*ns,row,nz,cnt,cnt2,*counts;
741ace3abfcSBarry Smith   PetscBool      flag;
7420def2e27SBarry Smith 
7430def2e27SBarry Smith   PetscFunctionBegin;
744785e854fSJed Brown   ierr = PetscMalloc1(m,&ns);CHKERRQ(ierr);
7450def2e27SBarry Smith   while (i < m) {
7460def2e27SBarry Smith     nzx = ai[i+1] - ai[i];       /* Number of nonzeros */
7470def2e27SBarry Smith     /* Limits the number of elements in a node to 'a->inode.limit' */
7480def2e27SBarry Smith     for (j=i+1,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) {
7490def2e27SBarry Smith       nzy = ai[j+1] - ai[j];
7500def2e27SBarry Smith       if (nzy != (nzx - j + i)) break;
7510def2e27SBarry Smith       ierr = PetscMemcmp(aj + ai[i] + j - i,aj + ai[j],nzy*sizeof(PetscInt),&flag);CHKERRQ(ierr);
7520def2e27SBarry Smith       if (!flag) break;
7530def2e27SBarry Smith     }
7540def2e27SBarry Smith     ns[node_count++] = blk_size;
75526fbe8dcSKarl Rupp 
7560def2e27SBarry Smith     i = j;
7570def2e27SBarry Smith   }
7580def2e27SBarry Smith   if (!a->inode.size && m && node_count > .9*m) {
7590def2e27SBarry Smith     ierr = PetscFree(ns);CHKERRQ(ierr);
7600def2e27SBarry Smith     ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr);
7610def2e27SBarry Smith   } else {
7620def2e27SBarry Smith     a->inode.node_count = node_count;
76326fbe8dcSKarl Rupp 
764785e854fSJed Brown     ierr = PetscMalloc1(node_count,&a->inode.size);CHKERRQ(ierr);
7653bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)A,node_count*sizeof(PetscInt));CHKERRQ(ierr);
76622d28d08SBarry Smith     ierr = PetscMemcpy(a->inode.size,ns,node_count*sizeof(PetscInt));CHKERRQ(ierr);
7670def2e27SBarry Smith     ierr = PetscFree(ns);CHKERRQ(ierr);
7680def2e27SBarry Smith     ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr);
7690def2e27SBarry Smith 
7700def2e27SBarry Smith     /* count collections of adjacent columns in each inode */
7710def2e27SBarry Smith     row = 0;
7720def2e27SBarry Smith     cnt = 0;
7730def2e27SBarry Smith     for (i=0; i<node_count; i++) {
7740def2e27SBarry Smith       cols = aj + ai[row] + a->inode.size[i];
7750def2e27SBarry Smith       nz   = ai[row+1] - ai[row] - a->inode.size[i];
7760def2e27SBarry Smith       for (j=1; j<nz; j++) {
77726fbe8dcSKarl Rupp         if (cols[j] != cols[j-1]+1) cnt++;
7780def2e27SBarry Smith       }
7790def2e27SBarry Smith       cnt++;
7800def2e27SBarry Smith       row += a->inode.size[i];
7810def2e27SBarry Smith     }
782785e854fSJed Brown     ierr = PetscMalloc1(2*cnt,&counts);CHKERRQ(ierr);
7830def2e27SBarry Smith     cnt  = 0;
7840def2e27SBarry Smith     row  = 0;
7850def2e27SBarry Smith     for (i=0; i<node_count; i++) {
7860def2e27SBarry Smith       cols = aj + ai[row] + a->inode.size[i];
7870def2e27SBarry Smith       counts[2*cnt] = cols[0];
7880def2e27SBarry Smith       nz   = ai[row+1] - ai[row] - a->inode.size[i];
7890def2e27SBarry Smith       cnt2 = 1;
7900def2e27SBarry Smith       for (j=1; j<nz; j++) {
7910def2e27SBarry Smith         if (cols[j] != cols[j-1]+1) {
7920def2e27SBarry Smith           counts[2*(cnt++)+1] = cnt2;
7930def2e27SBarry Smith           counts[2*cnt]       = cols[j];
7940def2e27SBarry Smith           cnt2 = 1;
7950def2e27SBarry Smith         } else cnt2++;
7960def2e27SBarry Smith       }
7970def2e27SBarry Smith       counts[2*(cnt++)+1] = cnt2;
7980def2e27SBarry Smith       row += a->inode.size[i];
7990def2e27SBarry Smith     }
80022d28d08SBarry Smith     ierr = PetscIntView(2*cnt,counts,0);CHKERRQ(ierr);
8010def2e27SBarry Smith   }
80238702af4SBarry Smith   PetscFunctionReturn(0);
80338702af4SBarry Smith }
80438702af4SBarry Smith 
805dfbe8321SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ(Mat A,MatAssemblyType mode)
80649b5e25fSSatish Balay {
80749b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
8086849ba73SBarry Smith   PetscErrorCode ierr;
80913f74950SBarry Smith   PetscInt       fshift = 0,i,j,*ai = a->i,*aj = a->j,*imax = a->imax;
810d0f46423SBarry Smith   PetscInt       m      = A->rmap->N,*ip,N,*ailen = a->ilen;
81113f74950SBarry Smith   PetscInt       mbs    = a->mbs,bs2 = a->bs2,rmax = 0;
81249b5e25fSSatish Balay   MatScalar      *aa    = a->a,*ap;
81349b5e25fSSatish Balay 
81449b5e25fSSatish Balay   PetscFunctionBegin;
81549b5e25fSSatish Balay   if (mode == MAT_FLUSH_ASSEMBLY) PetscFunctionReturn(0);
81649b5e25fSSatish Balay 
81749b5e25fSSatish Balay   if (m) rmax = ailen[0];
81849b5e25fSSatish Balay   for (i=1; i<mbs; i++) {
81949b5e25fSSatish Balay     /* move each row back by the amount of empty slots (fshift) before it*/
82049b5e25fSSatish Balay     fshift += imax[i-1] - ailen[i-1];
82149b5e25fSSatish Balay     rmax    = PetscMax(rmax,ailen[i]);
82249b5e25fSSatish Balay     if (fshift) {
82349b5e25fSSatish Balay       ip = aj + ai[i]; ap = aa + bs2*ai[i];
82449b5e25fSSatish Balay       N  = ailen[i];
82549b5e25fSSatish Balay       for (j=0; j<N; j++) {
82649b5e25fSSatish Balay         ip[j-fshift] = ip[j];
82749b5e25fSSatish Balay         ierr         = PetscMemcpy(ap+(j-fshift)*bs2,ap+j*bs2,bs2*sizeof(MatScalar));CHKERRQ(ierr);
82849b5e25fSSatish Balay       }
82949b5e25fSSatish Balay     }
83049b5e25fSSatish Balay     ai[i] = ai[i-1] + ailen[i-1];
83149b5e25fSSatish Balay   }
83249b5e25fSSatish Balay   if (mbs) {
83349b5e25fSSatish Balay     fshift += imax[mbs-1] - ailen[mbs-1];
83449b5e25fSSatish Balay     ai[mbs] = ai[mbs-1] + ailen[mbs-1];
83549b5e25fSSatish Balay   }
83649b5e25fSSatish Balay   /* reset ilen and imax for each row */
83749b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
83849b5e25fSSatish Balay     ailen[i] = imax[i] = ai[i+1] - ai[i];
83949b5e25fSSatish Balay   }
8406c6c5352SBarry Smith   a->nz = ai[mbs];
84149b5e25fSSatish Balay 
842b424e231SHong Zhang   /* diagonals may have moved, reset it */
843b424e231SHong Zhang   if (a->diag) {
8442ed38d0bSJed Brown     ierr = PetscMemcpy(a->diag,ai,mbs*sizeof(PetscInt));CHKERRQ(ierr);
84549b5e25fSSatish Balay   }
84626fbe8dcSKarl 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);
84726fbe8dcSKarl Rupp 
848d0f46423SBarry 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);
849ae15b995SBarry Smith   ierr = PetscInfo1(A,"Number of mallocs during MatSetValues is %D\n",a->reallocs);CHKERRQ(ierr);
850ae15b995SBarry Smith   ierr = PetscInfo1(A,"Most nonzeros blocks in any row is %D\n",rmax);CHKERRQ(ierr);
85126fbe8dcSKarl Rupp 
8528e58a170SBarry Smith   A->info.mallocs    += a->reallocs;
85349b5e25fSSatish Balay   a->reallocs         = 0;
85449b5e25fSSatish Balay   A->info.nz_unneeded = (PetscReal)fshift*bs2;
855061b2667SBarry Smith   a->idiagvalid       = PETSC_FALSE;
8564dcd73b1SHong Zhang   a->rmax             = rmax;
85738702af4SBarry Smith 
85838702af4SBarry Smith   if (A->cmap->n < 65536 && A->cmap->bs == 1) {
85944e1c64aSLisandro Dalcin     if (a->jshort && a->free_jshort) {
86017803ae8SHong Zhang       /* when matrix data structure is changed, previous jshort must be replaced */
86117803ae8SHong Zhang       ierr = PetscFree(a->jshort);CHKERRQ(ierr);
86217803ae8SHong Zhang     }
863785e854fSJed Brown     ierr = PetscMalloc1(a->i[A->rmap->n],&a->jshort);CHKERRQ(ierr);
8643bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)A,a->i[A->rmap->n]*sizeof(unsigned short));CHKERRQ(ierr);
86538702af4SBarry Smith     for (i=0; i<a->i[A->rmap->n]; i++) a->jshort[i] = a->j[i];
86638702af4SBarry Smith     A->ops->mult   = MatMult_SeqSBAIJ_1_ushort;
86741f059aeSBarry Smith     A->ops->sor    = MatSOR_SeqSBAIJ_ushort;
8684da8f245SBarry Smith     a->free_jshort = PETSC_TRUE;
86938702af4SBarry Smith   }
87049b5e25fSSatish Balay   PetscFunctionReturn(0);
87149b5e25fSSatish Balay }
87249b5e25fSSatish Balay 
87349b5e25fSSatish Balay /*
87449b5e25fSSatish Balay    This function returns an array of flags which indicate the locations of contiguous
87549b5e25fSSatish Balay    blocks that should be zeroed. for eg: if bs = 3  and is = [0,1,2,3,5,6,7,8,9]
87649b5e25fSSatish Balay    then the resulting sizes = [3,1,1,3,1] correspondig to sets [(0,1,2),(3),(5),(6,7,8),(9)]
87749b5e25fSSatish Balay    Assume: sizes should be long enough to hold all the values.
87849b5e25fSSatish Balay */
87913f74950SBarry Smith PetscErrorCode MatZeroRows_SeqSBAIJ_Check_Blocks(PetscInt idx[],PetscInt n,PetscInt bs,PetscInt sizes[], PetscInt *bs_max)
88049b5e25fSSatish Balay {
88113f74950SBarry Smith   PetscInt  i,j,k,row;
882ace3abfcSBarry Smith   PetscBool flg;
88349b5e25fSSatish Balay 
88449b5e25fSSatish Balay   PetscFunctionBegin;
88549b5e25fSSatish Balay   for (i=0,j=0; i<n; j++) {
88649b5e25fSSatish Balay     row = idx[i];
88749b5e25fSSatish Balay     if (row%bs!=0) { /* Not the begining of a block */
88849b5e25fSSatish Balay       sizes[j] = 1;
88949b5e25fSSatish Balay       i++;
89049b5e25fSSatish Balay     } else if (i+bs > n) { /* Beginning of a block, but complete block doesn't exist (at idx end) */
89149b5e25fSSatish Balay       sizes[j] = 1;         /* Also makes sure atleast 'bs' values exist for next else */
89249b5e25fSSatish Balay       i++;
89349b5e25fSSatish Balay     } else { /* Begining of the block, so check if the complete block exists */
89449b5e25fSSatish Balay       flg = PETSC_TRUE;
89549b5e25fSSatish Balay       for (k=1; k<bs; k++) {
89649b5e25fSSatish Balay         if (row+k != idx[i+k]) { /* break in the block */
89749b5e25fSSatish Balay           flg = PETSC_FALSE;
89849b5e25fSSatish Balay           break;
89949b5e25fSSatish Balay         }
90049b5e25fSSatish Balay       }
901abc0a331SBarry Smith       if (flg) { /* No break in the bs */
90249b5e25fSSatish Balay         sizes[j] = bs;
90349b5e25fSSatish Balay         i       += bs;
90449b5e25fSSatish Balay       } else {
90549b5e25fSSatish Balay         sizes[j] = 1;
90649b5e25fSSatish Balay         i++;
90749b5e25fSSatish Balay       }
90849b5e25fSSatish Balay     }
90949b5e25fSSatish Balay   }
91049b5e25fSSatish Balay   *bs_max = j;
91149b5e25fSSatish Balay   PetscFunctionReturn(0);
91249b5e25fSSatish Balay }
91349b5e25fSSatish Balay 
91449b5e25fSSatish Balay 
91549b5e25fSSatish Balay /* Only add/insert a(i,j) with i<=j (blocks).
91649b5e25fSSatish Balay    Any a(i,j) with i>j input by user is ingored.
91749b5e25fSSatish Balay */
91849b5e25fSSatish Balay 
91913f74950SBarry Smith PetscErrorCode MatSetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is)
92049b5e25fSSatish Balay {
92149b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
9226849ba73SBarry Smith   PetscErrorCode ierr;
923e2ee6c50SBarry Smith   PetscInt       *rp,k,low,high,t,ii,row,nrow,i,col,l,rmax,N,lastcol = -1;
92413f74950SBarry Smith   PetscInt       *imax=a->imax,*ai=a->i,*ailen=a->ilen,roworiented=a->roworiented;
925d0f46423SBarry Smith   PetscInt       *aj  =a->j,nonew=a->nonew,bs=A->rmap->bs,brow,bcol;
92613f74950SBarry Smith   PetscInt       ridx,cidx,bs2=a->bs2;
92749b5e25fSSatish Balay   MatScalar      *ap,value,*aa=a->a,*bap;
92849b5e25fSSatish Balay 
92949b5e25fSSatish Balay   PetscFunctionBegin;
93049b5e25fSSatish Balay   for (k=0; k<m; k++) { /* loop over added rows */
93149b5e25fSSatish Balay     row  = im[k];       /* row number */
93249b5e25fSSatish Balay     brow = row/bs;      /* block row number */
93349b5e25fSSatish Balay     if (row < 0) continue;
9342515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
935e32f2f54SBarry 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);
93649b5e25fSSatish Balay #endif
93749b5e25fSSatish Balay     rp   = aj + ai[brow]; /*ptr to beginning of column value of the row block*/
93849b5e25fSSatish Balay     ap   = aa + bs2*ai[brow]; /*ptr to beginning of element value of the row block*/
93949b5e25fSSatish Balay     rmax = imax[brow];  /* maximum space allocated for this row */
94049b5e25fSSatish Balay     nrow = ailen[brow]; /* actual length of this row */
94149b5e25fSSatish Balay     low  = 0;
94249b5e25fSSatish Balay 
94349b5e25fSSatish Balay     for (l=0; l<n; l++) { /* loop over added columns */
94449b5e25fSSatish Balay       if (in[l] < 0) continue;
9452515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
946e32f2f54SBarry 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);
94749b5e25fSSatish Balay #endif
94849b5e25fSSatish Balay       col  = in[l];
94949b5e25fSSatish Balay       bcol = col/bs;              /* block col number */
95049b5e25fSSatish Balay 
951941593c8SHong Zhang       if (brow > bcol) {
95226fbe8dcSKarl Rupp         if (a->ignore_ltriangular) continue; /* ignore lower triangular values */
95326fbe8dcSKarl 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)");
954941593c8SHong Zhang       }
955f4989cb3SHong Zhang 
95649b5e25fSSatish Balay       ridx = row % bs; cidx = col % bs; /*row and col index inside the block */
9578549e402SHong Zhang       if ((brow==bcol && ridx<=cidx) || (brow<bcol)) {
95849b5e25fSSatish Balay         /* element value a(k,l) */
95926fbe8dcSKarl Rupp         if (roworiented) value = v[l + k*n];
96026fbe8dcSKarl Rupp         else value = v[k + l*m];
96149b5e25fSSatish Balay 
96249b5e25fSSatish Balay         /* move pointer bap to a(k,l) quickly and add/insert value */
96326fbe8dcSKarl Rupp         if (col <= lastcol) low = 0;
96426fbe8dcSKarl Rupp         high = nrow;
965e2ee6c50SBarry Smith         lastcol = col;
96649b5e25fSSatish Balay         while (high-low > 7) {
96749b5e25fSSatish Balay           t = (low+high)/2;
96849b5e25fSSatish Balay           if (rp[t] > bcol) high = t;
96949b5e25fSSatish Balay           else              low  = t;
97049b5e25fSSatish Balay         }
97149b5e25fSSatish Balay         for (i=low; i<high; i++) {
97249b5e25fSSatish Balay           if (rp[i] > bcol) break;
97349b5e25fSSatish Balay           if (rp[i] == bcol) {
97449b5e25fSSatish Balay             bap = ap +  bs2*i + bs*cidx + ridx;
97549b5e25fSSatish Balay             if (is == ADD_VALUES) *bap += value;
97649b5e25fSSatish Balay             else                  *bap  = value;
9778549e402SHong Zhang             /* for diag block, add/insert its symmetric element a(cidx,ridx) */
9788549e402SHong Zhang             if (brow == bcol && ridx < cidx) {
9798549e402SHong Zhang               bap = ap +  bs2*i + bs*ridx + cidx;
9808549e402SHong Zhang               if (is == ADD_VALUES) *bap += value;
9818549e402SHong Zhang               else                  *bap  = value;
9828549e402SHong Zhang             }
98349b5e25fSSatish Balay             goto noinsert1;
98449b5e25fSSatish Balay           }
98549b5e25fSSatish Balay         }
98649b5e25fSSatish Balay 
98749b5e25fSSatish Balay         if (nonew == 1) goto noinsert1;
988e32f2f54SBarry Smith         if (nonew == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col);
989fef13f97SBarry Smith         MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,brow,bcol,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar);
99049b5e25fSSatish Balay 
991c03d1d03SSatish Balay         N = nrow++ - 1; high++;
99249b5e25fSSatish Balay         /* shift up all the later entries in this row */
99349b5e25fSSatish Balay         for (ii=N; ii>=i; ii--) {
99449b5e25fSSatish Balay           rp[ii+1] = rp[ii];
99549b5e25fSSatish Balay           ierr     = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr);
99649b5e25fSSatish Balay         }
99749b5e25fSSatish Balay         if (N>=i) {
99849b5e25fSSatish Balay           ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr);
99949b5e25fSSatish Balay         }
100049b5e25fSSatish Balay         rp[i]                      = bcol;
100149b5e25fSSatish Balay         ap[bs2*i + bs*cidx + ridx] = value;
1002e56f5c9eSBarry Smith         A->nonzerostate++;
100349b5e25fSSatish Balay noinsert1:;
100449b5e25fSSatish Balay         low = i;
10058549e402SHong Zhang       }
100649b5e25fSSatish Balay     }   /* end of loop over added columns */
100749b5e25fSSatish Balay     ailen[brow] = nrow;
100849b5e25fSSatish Balay   }   /* end of loop over added rows */
100949b5e25fSSatish Balay   PetscFunctionReturn(0);
101049b5e25fSSatish Balay }
101149b5e25fSSatish Balay 
10120481f469SBarry Smith PetscErrorCode MatICCFactor_SeqSBAIJ(Mat inA,IS row,const MatFactorInfo *info)
101349b5e25fSSatish Balay {
10144ccecd49SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)inA->data;
101549b5e25fSSatish Balay   Mat            outA;
1016dfbe8321SBarry Smith   PetscErrorCode ierr;
1017ace3abfcSBarry Smith   PetscBool      row_identity;
101849b5e25fSSatish Balay 
101949b5e25fSSatish Balay   PetscFunctionBegin;
1020e32f2f54SBarry Smith   if (info->levels != 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only levels=0 is supported for in-place icc");
1021c84f5b01SHong Zhang   ierr = ISIdentity(row,&row_identity);CHKERRQ(ierr);
1022e32f2f54SBarry Smith   if (!row_identity) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix reordering is not supported");
1023e32f2f54SBarry 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()! */
1024c84f5b01SHong Zhang 
102549b5e25fSSatish Balay   outA            = inA;
1026d5f3da31SBarry Smith   inA->factortype = MAT_FACTOR_ICC;
1027f6224b95SHong Zhang   ierr = PetscFree(inA->solvertype);CHKERRQ(ierr);
1028f6224b95SHong Zhang   ierr = PetscStrallocpy(MATSOLVERPETSC,&inA->solvertype);CHKERRQ(ierr);
102949b5e25fSSatish Balay 
10301a3463dfSHong Zhang   ierr = MatMarkDiagonal_SeqSBAIJ(inA);CHKERRQ(ierr);
1031d595f711SHong Zhang   ierr = MatSeqSBAIJSetNumericFactorization_inplace(inA,row_identity);CHKERRQ(ierr);
103249b5e25fSSatish Balay 
1033c3122656SLisandro Dalcin   ierr   = PetscObjectReference((PetscObject)row);CHKERRQ(ierr);
10346bf464f9SBarry Smith   ierr   = ISDestroy(&a->row);CHKERRQ(ierr);
1035c84f5b01SHong Zhang   a->row = row;
1036c3122656SLisandro Dalcin   ierr   = PetscObjectReference((PetscObject)row);CHKERRQ(ierr);
10376bf464f9SBarry Smith   ierr   = ISDestroy(&a->col);CHKERRQ(ierr);
1038c84f5b01SHong Zhang   a->col = row;
1039c84f5b01SHong Zhang 
1040c84f5b01SHong Zhang   /* Create the invert permutation so that it can be used in MatCholeskyFactorNumeric() */
1041c84f5b01SHong Zhang   if (a->icol) {ierr = ISInvertPermutation(row,PETSC_DECIDE, &a->icol);CHKERRQ(ierr);}
10423bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)inA,(PetscObject)a->icol);CHKERRQ(ierr);
104349b5e25fSSatish Balay 
104449b5e25fSSatish Balay   if (!a->solve_work) {
1045854ce69bSBarry Smith     ierr = PetscMalloc1(inA->rmap->N+inA->rmap->bs,&a->solve_work);CHKERRQ(ierr);
10463bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)inA,(inA->rmap->N+inA->rmap->bs)*sizeof(PetscScalar));CHKERRQ(ierr);
104749b5e25fSSatish Balay   }
104849b5e25fSSatish Balay 
1049719d5645SBarry Smith   ierr = MatCholeskyFactorNumeric(outA,inA,info);CHKERRQ(ierr);
105049b5e25fSSatish Balay   PetscFunctionReturn(0);
105149b5e25fSSatish Balay }
1052950f1e5bSHong Zhang 
10537087cfbeSBarry Smith PetscErrorCode  MatSeqSBAIJSetColumnIndices_SeqSBAIJ(Mat mat,PetscInt *indices)
105449b5e25fSSatish Balay {
1055045c9aa0SHong Zhang   Mat_SeqSBAIJ   *baij = (Mat_SeqSBAIJ*)mat->data;
105613f74950SBarry Smith   PetscInt       i,nz,n;
10577827cd58SJed Brown   PetscErrorCode ierr;
105849b5e25fSSatish Balay 
105949b5e25fSSatish Balay   PetscFunctionBegin;
10606c6c5352SBarry Smith   nz = baij->maxnz;
1061d0f46423SBarry Smith   n  = mat->cmap->n;
106226fbe8dcSKarl Rupp   for (i=0; i<nz; i++) baij->j[i] = indices[i];
106326fbe8dcSKarl Rupp 
10646c6c5352SBarry Smith   baij->nz = nz;
106526fbe8dcSKarl Rupp   for (i=0; i<n; i++) baij->ilen[i] = baij->imax[i];
106626fbe8dcSKarl Rupp 
10677827cd58SJed Brown   ierr = MatSetOption(mat,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
106849b5e25fSSatish Balay   PetscFunctionReturn(0);
106949b5e25fSSatish Balay }
107049b5e25fSSatish Balay 
107149b5e25fSSatish Balay /*@
107219585528SSatish Balay   MatSeqSBAIJSetColumnIndices - Set the column indices for all the rows
107349b5e25fSSatish Balay   in the matrix.
107449b5e25fSSatish Balay 
107549b5e25fSSatish Balay   Input Parameters:
107619585528SSatish Balay   +  mat     - the SeqSBAIJ matrix
107749b5e25fSSatish Balay   -  indices - the column indices
107849b5e25fSSatish Balay 
107949b5e25fSSatish Balay   Level: advanced
108049b5e25fSSatish Balay 
108149b5e25fSSatish Balay   Notes:
108249b5e25fSSatish Balay   This can be called if you have precomputed the nonzero structure of the
108349b5e25fSSatish Balay   matrix and want to provide it to the matrix object to improve the performance
108449b5e25fSSatish Balay   of the MatSetValues() operation.
108549b5e25fSSatish Balay 
108649b5e25fSSatish Balay   You MUST have set the correct numbers of nonzeros per row in the call to
1087d1be2dadSMatthew Knepley   MatCreateSeqSBAIJ(), and the columns indices MUST be sorted.
108849b5e25fSSatish Balay 
1089ab9f2c04SSatish Balay   MUST be called before any calls to MatSetValues()
109049b5e25fSSatish Balay 
1091ab9f2c04SSatish Balay   .seealso: MatCreateSeqSBAIJ
109249b5e25fSSatish Balay @*/
10937087cfbeSBarry Smith PetscErrorCode  MatSeqSBAIJSetColumnIndices(Mat mat,PetscInt *indices)
109449b5e25fSSatish Balay {
10954ac538c5SBarry Smith   PetscErrorCode ierr;
109649b5e25fSSatish Balay 
109749b5e25fSSatish Balay   PetscFunctionBegin;
10980700a824SBarry Smith   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
10994482741eSBarry Smith   PetscValidPointer(indices,2);
11004ac538c5SBarry Smith   ierr = PetscUseMethod(mat,"MatSeqSBAIJSetColumnIndices_C",(Mat,PetscInt*),(mat,indices));CHKERRQ(ierr);
110149b5e25fSSatish Balay   PetscFunctionReturn(0);
110249b5e25fSSatish Balay }
110349b5e25fSSatish Balay 
11043c896bc6SHong Zhang PetscErrorCode MatCopy_SeqSBAIJ(Mat A,Mat B,MatStructure str)
11053c896bc6SHong Zhang {
11063c896bc6SHong Zhang   PetscErrorCode ierr;
11073c896bc6SHong Zhang 
11083c896bc6SHong Zhang   PetscFunctionBegin;
11093c896bc6SHong Zhang   /* If the two matrices have the same copy implementation, use fast copy. */
11103c896bc6SHong Zhang   if (str == SAME_NONZERO_PATTERN && (A->ops->copy == B->ops->copy)) {
11113c896bc6SHong Zhang     Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
11123c896bc6SHong Zhang     Mat_SeqSBAIJ *b = (Mat_SeqSBAIJ*)B->data;
11133c896bc6SHong Zhang 
1114e7e72b3dSBarry 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");
1115d0f46423SBarry Smith     ierr = PetscMemcpy(b->a,a->a,(a->i[A->rmap->N])*sizeof(PetscScalar));CHKERRQ(ierr);
11163c896bc6SHong Zhang   } else {
1117f5edf698SHong Zhang     ierr = MatGetRowUpperTriangular(A);CHKERRQ(ierr);
11183c896bc6SHong Zhang     ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr);
1119f5edf698SHong Zhang     ierr = MatRestoreRowUpperTriangular(A);CHKERRQ(ierr);
11203c896bc6SHong Zhang   }
11213c896bc6SHong Zhang   PetscFunctionReturn(0);
11223c896bc6SHong Zhang }
11233c896bc6SHong Zhang 
11244994cf47SJed Brown PetscErrorCode MatSetUp_SeqSBAIJ(Mat A)
1125273d9f13SBarry Smith {
1126dfbe8321SBarry Smith   PetscErrorCode ierr;
1127273d9f13SBarry Smith 
1128273d9f13SBarry Smith   PetscFunctionBegin;
1129367daffbSBarry Smith   ierr = MatSeqSBAIJSetPreallocation(A,A->rmap->bs,PETSC_DEFAULT,0);CHKERRQ(ierr);
1130273d9f13SBarry Smith   PetscFunctionReturn(0);
1131273d9f13SBarry Smith }
1132273d9f13SBarry Smith 
11338c778c55SBarry Smith PetscErrorCode MatSeqSBAIJGetArray_SeqSBAIJ(Mat A,PetscScalar *array[])
1134a6ece127SHong Zhang {
1135a6ece127SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
11365fd66863SKarl Rupp 
1137a6ece127SHong Zhang   PetscFunctionBegin;
1138a6ece127SHong Zhang   *array = a->a;
1139a6ece127SHong Zhang   PetscFunctionReturn(0);
1140a6ece127SHong Zhang }
1141a6ece127SHong Zhang 
11428c778c55SBarry Smith PetscErrorCode MatSeqSBAIJRestoreArray_SeqSBAIJ(Mat A,PetscScalar *array[])
1143a6ece127SHong Zhang {
1144a6ece127SHong Zhang   PetscFunctionBegin;
1145a6ece127SHong Zhang   PetscFunctionReturn(0);
1146a6ece127SHong Zhang }
1147a6ece127SHong Zhang 
114852768537SHong Zhang PetscErrorCode MatAXPYGetPreallocation_SeqSBAIJ(Mat Y,Mat X,PetscInt *nnz)
114952768537SHong Zhang {
1150b264fe52SHong Zhang   PetscInt       bs = Y->rmap->bs,mbs = Y->rmap->N/bs;
115152768537SHong Zhang   Mat_SeqSBAIJ   *x = (Mat_SeqSBAIJ*)X->data;
115252768537SHong Zhang   Mat_SeqSBAIJ   *y = (Mat_SeqSBAIJ*)Y->data;
1153b264fe52SHong Zhang   PetscErrorCode ierr;
115452768537SHong Zhang 
115552768537SHong Zhang   PetscFunctionBegin;
115652768537SHong Zhang   /* Set the number of nonzeros in the new matrix */
1157b264fe52SHong Zhang   ierr = MatAXPYGetPreallocation_SeqX_private(mbs,x->i,x->j,y->i,y->j,nnz);CHKERRQ(ierr);
115852768537SHong Zhang   PetscFunctionReturn(0);
115952768537SHong Zhang }
116052768537SHong Zhang 
1161f4df32b1SMatthew Knepley PetscErrorCode MatAXPY_SeqSBAIJ(Mat Y,PetscScalar a,Mat X,MatStructure str)
116242ee4b1aSHong Zhang {
116342ee4b1aSHong Zhang   Mat_SeqSBAIJ   *x=(Mat_SeqSBAIJ*)X->data, *y=(Mat_SeqSBAIJ*)Y->data;
1164dfbe8321SBarry Smith   PetscErrorCode ierr;
116531ce2d13SHong Zhang   PetscInt       bs=Y->rmap->bs,bs2=bs*bs;
1166e838b9e7SJed Brown   PetscBLASInt   one = 1;
116742ee4b1aSHong Zhang 
116842ee4b1aSHong Zhang   PetscFunctionBegin;
116942ee4b1aSHong Zhang   if (str == SAME_NONZERO_PATTERN) {
1170f4df32b1SMatthew Knepley     PetscScalar  alpha = a;
1171c5df96a5SBarry Smith     PetscBLASInt bnz;
1172c5df96a5SBarry Smith     ierr = PetscBLASIntCast(x->nz*bs2,&bnz);CHKERRQ(ierr);
11738b83055fSJed Brown     PetscStackCallBLAS("BLASaxpy",BLASaxpy_(&bnz,&alpha,x->a,&one,y->a,&one));
1174a3fa217bSJose E. Roman     ierr = PetscObjectStateIncrease((PetscObject)Y);CHKERRQ(ierr);
1175ab784542SHong Zhang   } else if (str == SUBSET_NONZERO_PATTERN) { /* nonzeros of X is a subset of Y's */
1176ab784542SHong Zhang     ierr = MatSetOption(X,MAT_GETROW_UPPERTRIANGULAR,PETSC_TRUE);CHKERRQ(ierr);
1177ab784542SHong Zhang     ierr = MatAXPY_Basic(Y,a,X,str);CHKERRQ(ierr);
1178ab784542SHong Zhang     ierr = MatSetOption(X,MAT_GETROW_UPPERTRIANGULAR,PETSC_FALSE);CHKERRQ(ierr);
117942ee4b1aSHong Zhang   } else {
118052768537SHong Zhang     Mat      B;
118152768537SHong Zhang     PetscInt *nnz;
118252768537SHong Zhang     if (bs != X->rmap->bs) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Matrices must have same block size");
1183f5edf698SHong Zhang     ierr = MatGetRowUpperTriangular(X);CHKERRQ(ierr);
118452768537SHong Zhang     ierr = MatGetRowUpperTriangular(Y);CHKERRQ(ierr);
118552768537SHong Zhang     ierr = PetscMalloc1(Y->rmap->N,&nnz);CHKERRQ(ierr);
118652768537SHong Zhang     ierr = MatCreate(PetscObjectComm((PetscObject)Y),&B);CHKERRQ(ierr);
118752768537SHong Zhang     ierr = PetscObjectSetName((PetscObject)B,((PetscObject)Y)->name);CHKERRQ(ierr);
118852768537SHong Zhang     ierr = MatSetSizes(B,Y->rmap->n,Y->cmap->n,Y->rmap->N,Y->cmap->N);CHKERRQ(ierr);
118952768537SHong Zhang     ierr = MatSetBlockSizesFromMats(B,Y,Y);CHKERRQ(ierr);
119052768537SHong Zhang     ierr = MatSetType(B,(MatType) ((PetscObject)Y)->type_name);CHKERRQ(ierr);
119152768537SHong Zhang     ierr = MatAXPYGetPreallocation_SeqSBAIJ(Y,X,nnz);CHKERRQ(ierr);
119252768537SHong Zhang     ierr = MatSeqSBAIJSetPreallocation(B,bs,0,nnz);CHKERRQ(ierr);
119352768537SHong Zhang 
119452768537SHong Zhang     ierr = MatAXPY_BasicWithPreallocation(B,Y,a,X,str);CHKERRQ(ierr);
119552768537SHong Zhang 
119628be2f97SBarry Smith     ierr = MatHeaderReplace(Y,&B);CHKERRQ(ierr);
119752768537SHong Zhang     ierr = PetscFree(nnz);CHKERRQ(ierr);
1198f5edf698SHong Zhang     ierr = MatRestoreRowUpperTriangular(X);CHKERRQ(ierr);
119952768537SHong Zhang     ierr = MatRestoreRowUpperTriangular(Y);CHKERRQ(ierr);
120042ee4b1aSHong Zhang   }
120142ee4b1aSHong Zhang   PetscFunctionReturn(0);
120242ee4b1aSHong Zhang }
120342ee4b1aSHong Zhang 
1204ace3abfcSBarry Smith PetscErrorCode MatIsSymmetric_SeqSBAIJ(Mat A,PetscReal tol,PetscBool  *flg)
1205efcf0fc3SBarry Smith {
1206efcf0fc3SBarry Smith   PetscFunctionBegin;
1207efcf0fc3SBarry Smith   *flg = PETSC_TRUE;
1208efcf0fc3SBarry Smith   PetscFunctionReturn(0);
1209efcf0fc3SBarry Smith }
1210efcf0fc3SBarry Smith 
1211ace3abfcSBarry Smith PetscErrorCode MatIsStructurallySymmetric_SeqSBAIJ(Mat A,PetscBool  *flg)
1212efcf0fc3SBarry Smith {
1213efcf0fc3SBarry Smith   PetscFunctionBegin;
1214efcf0fc3SBarry Smith   *flg = PETSC_TRUE;
1215efcf0fc3SBarry Smith   PetscFunctionReturn(0);
1216efcf0fc3SBarry Smith }
1217efcf0fc3SBarry Smith 
1218ace3abfcSBarry Smith PetscErrorCode MatIsHermitian_SeqSBAIJ(Mat A,PetscReal tol,PetscBool  *flg)
1219efcf0fc3SBarry Smith {
1220efcf0fc3SBarry Smith   PetscFunctionBegin;
1221efcf0fc3SBarry Smith   *flg = PETSC_FALSE;
1222efcf0fc3SBarry Smith   PetscFunctionReturn(0);
1223efcf0fc3SBarry Smith }
1224efcf0fc3SBarry Smith 
122599cafbc1SBarry Smith PetscErrorCode MatRealPart_SeqSBAIJ(Mat A)
122699cafbc1SBarry Smith {
122799cafbc1SBarry Smith   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
122899cafbc1SBarry Smith   PetscInt     i,nz = a->bs2*a->i[a->mbs];
1229dd6ea824SBarry Smith   MatScalar    *aa = a->a;
123099cafbc1SBarry Smith 
123199cafbc1SBarry Smith   PetscFunctionBegin;
123299cafbc1SBarry Smith   for (i=0; i<nz; i++) aa[i] = PetscRealPart(aa[i]);
123399cafbc1SBarry Smith   PetscFunctionReturn(0);
123499cafbc1SBarry Smith }
123599cafbc1SBarry Smith 
123699cafbc1SBarry Smith PetscErrorCode MatImaginaryPart_SeqSBAIJ(Mat A)
123799cafbc1SBarry Smith {
123899cafbc1SBarry Smith   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
123999cafbc1SBarry Smith   PetscInt     i,nz = a->bs2*a->i[a->mbs];
1240dd6ea824SBarry Smith   MatScalar    *aa = a->a;
124199cafbc1SBarry Smith 
124299cafbc1SBarry Smith   PetscFunctionBegin;
124399cafbc1SBarry Smith   for (i=0; i<nz; i++) aa[i] = PetscImaginaryPart(aa[i]);
124499cafbc1SBarry Smith   PetscFunctionReturn(0);
124599cafbc1SBarry Smith }
124699cafbc1SBarry Smith 
12473bededecSBarry Smith PetscErrorCode MatZeroRowsColumns_SeqSBAIJ(Mat A,PetscInt is_n,const PetscInt is_idx[],PetscScalar diag,Vec x, Vec b)
12483bededecSBarry Smith {
12493bededecSBarry Smith   Mat_SeqSBAIJ      *baij=(Mat_SeqSBAIJ*)A->data;
12503bededecSBarry Smith   PetscErrorCode    ierr;
12513bededecSBarry Smith   PetscInt          i,j,k,count;
12523bededecSBarry Smith   PetscInt          bs   =A->rmap->bs,bs2=baij->bs2,row,col;
12533bededecSBarry Smith   PetscScalar       zero = 0.0;
12543bededecSBarry Smith   MatScalar         *aa;
12553bededecSBarry Smith   const PetscScalar *xx;
12563bededecSBarry Smith   PetscScalar       *bb;
125756777dd2SBarry Smith   PetscBool         *zeroed,vecs = PETSC_FALSE;
12583bededecSBarry Smith 
12593bededecSBarry Smith   PetscFunctionBegin;
12603bededecSBarry Smith   /* fix right hand side if needed */
12613bededecSBarry Smith   if (x && b) {
12623bededecSBarry Smith     ierr = VecGetArrayRead(x,&xx);CHKERRQ(ierr);
12633bededecSBarry Smith     ierr = VecGetArray(b,&bb);CHKERRQ(ierr);
126456777dd2SBarry Smith     vecs = PETSC_TRUE;
12653bededecSBarry Smith   }
12663bededecSBarry Smith 
12673bededecSBarry Smith   /* zero the columns */
12681795a4d1SJed Brown   ierr = PetscCalloc1(A->rmap->n,&zeroed);CHKERRQ(ierr);
12693bededecSBarry Smith   for (i=0; i<is_n; i++) {
12703bededecSBarry 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]);
12713bededecSBarry Smith     zeroed[is_idx[i]] = PETSC_TRUE;
12723bededecSBarry Smith   }
127356777dd2SBarry Smith   if (vecs) {
127456777dd2SBarry Smith     for (i=0; i<A->rmap->N; i++) {
127556777dd2SBarry Smith       row = i/bs;
127656777dd2SBarry Smith       for (j=baij->i[row]; j<baij->i[row+1]; j++) {
127756777dd2SBarry Smith         for (k=0; k<bs; k++) {
127856777dd2SBarry Smith           col = bs*baij->j[j] + k;
127956777dd2SBarry Smith           if (col <= i) continue;
128056777dd2SBarry Smith           aa = ((MatScalar*)(baij->a)) + j*bs2 + (i%bs) + bs*k;
128126fbe8dcSKarl Rupp           if (!zeroed[i] && zeroed[col]) bb[i]   -= aa[0]*xx[col];
128226fbe8dcSKarl Rupp           if (zeroed[i] && !zeroed[col]) bb[col] -= aa[0]*xx[i];
128356777dd2SBarry Smith         }
128456777dd2SBarry Smith       }
128556777dd2SBarry Smith     }
128626fbe8dcSKarl Rupp     for (i=0; i<is_n; i++) bb[is_idx[i]] = diag*xx[is_idx[i]];
128756777dd2SBarry Smith   }
128856777dd2SBarry Smith 
12893bededecSBarry Smith   for (i=0; i<A->rmap->N; i++) {
12903bededecSBarry Smith     if (!zeroed[i]) {
12913bededecSBarry Smith       row = i/bs;
12923bededecSBarry Smith       for (j=baij->i[row]; j<baij->i[row+1]; j++) {
12933bededecSBarry Smith         for (k=0; k<bs; k++) {
12943bededecSBarry Smith           col = bs*baij->j[j] + k;
12953bededecSBarry Smith           if (zeroed[col]) {
12963bededecSBarry Smith             aa = ((MatScalar*)(baij->a)) + j*bs2 + (i%bs) + bs*k;
12973bededecSBarry Smith             aa[0] = 0.0;
12983bededecSBarry Smith           }
12993bededecSBarry Smith         }
13003bededecSBarry Smith       }
13013bededecSBarry Smith     }
13023bededecSBarry Smith   }
13033bededecSBarry Smith   ierr = PetscFree(zeroed);CHKERRQ(ierr);
130456777dd2SBarry Smith   if (vecs) {
130556777dd2SBarry Smith     ierr = VecRestoreArrayRead(x,&xx);CHKERRQ(ierr);
130656777dd2SBarry Smith     ierr = VecRestoreArray(b,&bb);CHKERRQ(ierr);
130756777dd2SBarry Smith   }
13083bededecSBarry Smith 
13093bededecSBarry Smith   /* zero the rows */
13103bededecSBarry Smith   for (i=0; i<is_n; i++) {
13113bededecSBarry Smith     row   = is_idx[i];
13123bededecSBarry Smith     count = (baij->i[row/bs +1] - baij->i[row/bs])*bs;
13133bededecSBarry Smith     aa    = ((MatScalar*)(baij->a)) + baij->i[row/bs]*bs2 + (row%bs);
13143bededecSBarry Smith     for (k=0; k<count; k++) {
13153bededecSBarry Smith       aa[0] =  zero;
13163bededecSBarry Smith       aa   += bs;
13173bededecSBarry Smith     }
13183bededecSBarry Smith     if (diag != 0.0) {
13193bededecSBarry Smith       ierr = (*A->ops->setvalues)(A,1,&row,1,&row,&diag,INSERT_VALUES);CHKERRQ(ierr);
13203bededecSBarry Smith     }
13213bededecSBarry Smith   }
13223bededecSBarry Smith   ierr = MatAssemblyEnd_SeqSBAIJ(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
13233bededecSBarry Smith   PetscFunctionReturn(0);
13243bededecSBarry Smith }
13253bededecSBarry Smith 
13267d68702bSBarry Smith PetscErrorCode MatShift_SeqSBAIJ(Mat Y,PetscScalar a)
13277d68702bSBarry Smith {
13287d68702bSBarry Smith   PetscErrorCode ierr;
13297d68702bSBarry Smith   Mat_SeqSBAIJ    *aij = (Mat_SeqSBAIJ*)Y->data;
13307d68702bSBarry Smith 
13317d68702bSBarry Smith   PetscFunctionBegin;
13326f33a894SBarry Smith   if (!Y->preallocated || !aij->nz) {
13337d68702bSBarry Smith     ierr = MatSeqSBAIJSetPreallocation(Y,Y->rmap->bs,1,NULL);CHKERRQ(ierr);
13347d68702bSBarry Smith   }
13357d68702bSBarry Smith   ierr = MatShift_Basic(Y,a);CHKERRQ(ierr);
13367d68702bSBarry Smith   PetscFunctionReturn(0);
13377d68702bSBarry Smith }
13387d68702bSBarry Smith 
133949b5e25fSSatish Balay /* -------------------------------------------------------------------*/
13403964eb88SJed Brown static struct _MatOps MatOps_Values = {MatSetValues_SeqSBAIJ,
134149b5e25fSSatish Balay                                        MatGetRow_SeqSBAIJ,
134249b5e25fSSatish Balay                                        MatRestoreRow_SeqSBAIJ,
134349b5e25fSSatish Balay                                        MatMult_SeqSBAIJ_N,
134497304618SKris Buschelman                                /*  4*/ MatMultAdd_SeqSBAIJ_N,
1345431c96f7SBarry Smith                                        MatMult_SeqSBAIJ_N,       /* transpose versions are same as non-transpose versions */
1346e005ede5SBarry Smith                                        MatMultAdd_SeqSBAIJ_N,
1347db4efbfdSBarry Smith                                        0,
134849b5e25fSSatish Balay                                        0,
134949b5e25fSSatish Balay                                        0,
135097304618SKris Buschelman                                /* 10*/ 0,
135149b5e25fSSatish Balay                                        0,
1352c078aec8SLisandro Dalcin                                        MatCholeskyFactor_SeqSBAIJ,
135341f059aeSBarry Smith                                        MatSOR_SeqSBAIJ,
135449b5e25fSSatish Balay                                        MatTranspose_SeqSBAIJ,
135597304618SKris Buschelman                                /* 15*/ MatGetInfo_SeqSBAIJ,
135649b5e25fSSatish Balay                                        MatEqual_SeqSBAIJ,
135749b5e25fSSatish Balay                                        MatGetDiagonal_SeqSBAIJ,
135849b5e25fSSatish Balay                                        MatDiagonalScale_SeqSBAIJ,
135949b5e25fSSatish Balay                                        MatNorm_SeqSBAIJ,
136097304618SKris Buschelman                                /* 20*/ 0,
136149b5e25fSSatish Balay                                        MatAssemblyEnd_SeqSBAIJ,
136249b5e25fSSatish Balay                                        MatSetOption_SeqSBAIJ,
136349b5e25fSSatish Balay                                        MatZeroEntries_SeqSBAIJ,
1364d519adbfSMatthew Knepley                                /* 24*/ 0,
136549b5e25fSSatish Balay                                        0,
136649b5e25fSSatish Balay                                        0,
1367db4efbfdSBarry Smith                                        0,
1368db4efbfdSBarry Smith                                        0,
13694994cf47SJed Brown                                /* 29*/ MatSetUp_SeqSBAIJ,
1370c464158bSHong Zhang                                        0,
1371db4efbfdSBarry Smith                                        0,
13728c778c55SBarry Smith                                        0,
13738c778c55SBarry Smith                                        0,
1374d519adbfSMatthew Knepley                                /* 34*/ MatDuplicate_SeqSBAIJ,
1375719d5645SBarry Smith                                        0,
1376719d5645SBarry Smith                                        0,
137749b5e25fSSatish Balay                                        0,
1378c84f5b01SHong Zhang                                        MatICCFactor_SeqSBAIJ,
1379d519adbfSMatthew Knepley                                /* 39*/ MatAXPY_SeqSBAIJ,
13807dae84e0SHong Zhang                                        MatCreateSubMatrices_SeqSBAIJ,
138149b5e25fSSatish Balay                                        MatIncreaseOverlap_SeqSBAIJ,
138249b5e25fSSatish Balay                                        MatGetValues_SeqSBAIJ,
13833c896bc6SHong Zhang                                        MatCopy_SeqSBAIJ,
1384d519adbfSMatthew Knepley                                /* 44*/ 0,
138549b5e25fSSatish Balay                                        MatScale_SeqSBAIJ,
13867d68702bSBarry Smith                                        MatShift_SeqSBAIJ,
138749b5e25fSSatish Balay                                        0,
13883bededecSBarry Smith                                        MatZeroRowsColumns_SeqSBAIJ,
1389f73d5cc4SBarry Smith                                /* 49*/ 0,
139049b5e25fSSatish Balay                                        MatGetRowIJ_SeqSBAIJ,
139149b5e25fSSatish Balay                                        MatRestoreRowIJ_SeqSBAIJ,
139249b5e25fSSatish Balay                                        0,
139349b5e25fSSatish Balay                                        0,
1394d519adbfSMatthew Knepley                                /* 54*/ 0,
139549b5e25fSSatish Balay                                        0,
139649b5e25fSSatish Balay                                        0,
139749b5e25fSSatish Balay                                        0,
139849b5e25fSSatish Balay                                        MatSetValuesBlocked_SeqSBAIJ,
13997dae84e0SHong Zhang                                /* 59*/ MatCreateSubMatrix_SeqSBAIJ,
140049b5e25fSSatish Balay                                        0,
140149b5e25fSSatish Balay                                        0,
1402357abbc8SBarry Smith                                        0,
1403d959ec07SHong Zhang                                        0,
1404d519adbfSMatthew Knepley                                /* 64*/ 0,
1405d959ec07SHong Zhang                                        0,
1406d959ec07SHong Zhang                                        0,
1407d959ec07SHong Zhang                                        0,
1408d959ec07SHong Zhang                                        0,
1409d519adbfSMatthew Knepley                                /* 69*/ MatGetRowMaxAbs_SeqSBAIJ,
14103e0d88b5SBarry Smith                                        0,
14113e0d88b5SBarry Smith                                        0,
14123e0d88b5SBarry Smith                                        0,
14133e0d88b5SBarry Smith                                        0,
1414d519adbfSMatthew Knepley                                /* 74*/ 0,
14153e0d88b5SBarry Smith                                        0,
14163e0d88b5SBarry Smith                                        0,
14173e0d88b5SBarry Smith                                        0,
14183e0d88b5SBarry Smith                                        0,
1419d519adbfSMatthew Knepley                                /* 79*/ 0,
14203e0d88b5SBarry Smith                                        0,
14213e0d88b5SBarry Smith                                        0,
142297304618SKris Buschelman                                        MatGetInertia_SeqSBAIJ,
14235bba2384SShri Abhyankar                                        MatLoad_SeqSBAIJ,
1424d519adbfSMatthew Knepley                                /* 84*/ MatIsSymmetric_SeqSBAIJ,
1425865e5f61SKris Buschelman                                        MatIsHermitian_SeqSBAIJ,
1426efcf0fc3SBarry Smith                                        MatIsStructurallySymmetric_SeqSBAIJ,
1427865e5f61SKris Buschelman                                        0,
1428865e5f61SKris Buschelman                                        0,
1429d519adbfSMatthew Knepley                                /* 89*/ 0,
1430865e5f61SKris Buschelman                                        0,
1431865e5f61SKris Buschelman                                        0,
1432865e5f61SKris Buschelman                                        0,
1433865e5f61SKris Buschelman                                        0,
1434d519adbfSMatthew Knepley                                /* 94*/ 0,
1435865e5f61SKris Buschelman                                        0,
1436865e5f61SKris Buschelman                                        0,
143799cafbc1SBarry Smith                                        0,
143899cafbc1SBarry Smith                                        0,
1439d519adbfSMatthew Knepley                                /* 99*/ 0,
144099cafbc1SBarry Smith                                        0,
144199cafbc1SBarry Smith                                        0,
144299cafbc1SBarry Smith                                        0,
144399cafbc1SBarry Smith                                        0,
1444d519adbfSMatthew Knepley                                /*104*/ 0,
144599cafbc1SBarry Smith                                        MatRealPart_SeqSBAIJ,
1446f5edf698SHong Zhang                                        MatImaginaryPart_SeqSBAIJ,
1447f5edf698SHong Zhang                                        MatGetRowUpperTriangular_SeqSBAIJ,
14482af78befSBarry Smith                                        MatRestoreRowUpperTriangular_SeqSBAIJ,
1449d519adbfSMatthew Knepley                                /*109*/ 0,
14502af78befSBarry Smith                                        0,
14512af78befSBarry Smith                                        0,
14522af78befSBarry Smith                                        0,
1453547795f9SHong Zhang                                        MatMissingDiagonal_SeqSBAIJ,
1454547795f9SHong Zhang                                /*114*/ 0,
1455547795f9SHong Zhang                                        0,
1456547795f9SHong Zhang                                        0,
1457547795f9SHong Zhang                                        0,
1458547795f9SHong Zhang                                        0,
1459547795f9SHong Zhang                                /*119*/ 0,
1460547795f9SHong Zhang                                        0,
14612f480046SShri Abhyankar                                        0,
14623964eb88SJed Brown                                        0,
14633964eb88SJed Brown                                        0,
14643964eb88SJed Brown                                /*124*/ 0,
14653964eb88SJed Brown                                        0,
14663964eb88SJed Brown                                        0,
14673964eb88SJed Brown                                        0,
14683964eb88SJed Brown                                        0,
14693964eb88SJed Brown                                /*129*/ 0,
14703964eb88SJed Brown                                        0,
14713964eb88SJed Brown                                        0,
14723964eb88SJed Brown                                        0,
14733964eb88SJed Brown                                        0,
14743964eb88SJed Brown                                /*134*/ 0,
14753964eb88SJed Brown                                        0,
14763964eb88SJed Brown                                        0,
14773964eb88SJed Brown                                        0,
14783964eb88SJed Brown                                        0,
147946533700Sstefano_zampini                                /*139*/ MatSetBlockSizes_Default,
1480f9426fe0SMark Adams                                        0,
148159f5e6ceSHong Zhang                                        0,
148259f5e6ceSHong Zhang                                        0,
148359f5e6ceSHong Zhang                                        0,
148459f5e6ceSHong Zhang                                 /*144*/MatCreateMPIMatConcatenateSeqMat_SeqSBAIJ
148599cafbc1SBarry Smith };
1486be1d678aSKris Buschelman 
14877087cfbeSBarry Smith PetscErrorCode  MatStoreValues_SeqSBAIJ(Mat mat)
148849b5e25fSSatish Balay {
14894afc71dfSHong Zhang   Mat_SeqSBAIJ   *aij = (Mat_SeqSBAIJ*)mat->data;
1490d0f46423SBarry Smith   PetscInt       nz   = aij->i[mat->rmap->N]*mat->rmap->bs*aij->bs2;
1491dfbe8321SBarry Smith   PetscErrorCode ierr;
149249b5e25fSSatish Balay 
149349b5e25fSSatish Balay   PetscFunctionBegin;
1494e7e72b3dSBarry Smith   if (aij->nonew != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first");
149549b5e25fSSatish Balay 
149649b5e25fSSatish Balay   /* allocate space for values if not already there */
149749b5e25fSSatish Balay   if (!aij->saved_values) {
1498854ce69bSBarry Smith     ierr = PetscMalloc1(nz+1,&aij->saved_values);CHKERRQ(ierr);
149949b5e25fSSatish Balay   }
150049b5e25fSSatish Balay 
150149b5e25fSSatish Balay   /* copy values over */
150287828ca2SBarry Smith   ierr = PetscMemcpy(aij->saved_values,aij->a,nz*sizeof(PetscScalar));CHKERRQ(ierr);
150349b5e25fSSatish Balay   PetscFunctionReturn(0);
150449b5e25fSSatish Balay }
150549b5e25fSSatish Balay 
15067087cfbeSBarry Smith PetscErrorCode  MatRetrieveValues_SeqSBAIJ(Mat mat)
150749b5e25fSSatish Balay {
15084afc71dfSHong Zhang   Mat_SeqSBAIJ   *aij = (Mat_SeqSBAIJ*)mat->data;
15096849ba73SBarry Smith   PetscErrorCode ierr;
1510d0f46423SBarry Smith   PetscInt       nz = aij->i[mat->rmap->N]*mat->rmap->bs*aij->bs2;
151149b5e25fSSatish Balay 
151249b5e25fSSatish Balay   PetscFunctionBegin;
1513e7e72b3dSBarry Smith   if (aij->nonew != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first");
1514e7e72b3dSBarry Smith   if (!aij->saved_values) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatStoreValues(A);first");
151549b5e25fSSatish Balay 
151649b5e25fSSatish Balay   /* copy values over */
151787828ca2SBarry Smith   ierr = PetscMemcpy(aij->a,aij->saved_values,nz*sizeof(PetscScalar));CHKERRQ(ierr);
151849b5e25fSSatish Balay   PetscFunctionReturn(0);
151949b5e25fSSatish Balay }
152049b5e25fSSatish Balay 
1521367daffbSBarry Smith static PetscErrorCode  MatSeqSBAIJSetPreallocation_SeqSBAIJ(Mat B,PetscInt bs,PetscInt nz,PetscInt *nnz)
152249b5e25fSSatish Balay {
1523c464158bSHong Zhang   Mat_SeqSBAIJ   *b = (Mat_SeqSBAIJ*)B->data;
15246849ba73SBarry Smith   PetscErrorCode ierr;
15254dcd73b1SHong Zhang   PetscInt       i,mbs,nbs,bs2;
15262576faa2SJed Brown   PetscBool      skipallocation = PETSC_FALSE,flg = PETSC_FALSE,realalloc = PETSC_FALSE;
152749b5e25fSSatish Balay 
152849b5e25fSSatish Balay   PetscFunctionBegin;
15292576faa2SJed Brown   if (nz >= 0 || nnz) realalloc = PETSC_TRUE;
1530db4efbfdSBarry Smith 
153133d57670SJed Brown   ierr = MatSetBlockSize(B,PetscAbs(bs));CHKERRQ(ierr);
153226283091SBarry Smith   ierr = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr);
153326283091SBarry Smith   ierr = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr);
1534e02043d6SBarry Smith   ierr = PetscLayoutGetBlockSize(B->rmap,&bs);CHKERRQ(ierr);
1535899cda47SBarry Smith 
153621940c7eSstefano_zampini   B->preallocated = PETSC_TRUE;
153721940c7eSstefano_zampini 
1538d0f46423SBarry Smith   mbs = B->rmap->N/bs;
15394dcd73b1SHong Zhang   nbs = B->cmap->n/bs;
154049b5e25fSSatish Balay   bs2 = bs*bs;
154149b5e25fSSatish Balay 
15424dcd73b1SHong 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");
154349b5e25fSSatish Balay 
1544ab93d7beSBarry Smith   if (nz == MAT_SKIP_ALLOCATION) {
1545ab93d7beSBarry Smith     skipallocation = PETSC_TRUE;
1546ab93d7beSBarry Smith     nz             = 0;
1547ab93d7beSBarry Smith   }
1548ab93d7beSBarry Smith 
1549435da068SBarry Smith   if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 3;
1550e32f2f54SBarry Smith   if (nz < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"nz cannot be less than 0: value %D",nz);
155149b5e25fSSatish Balay   if (nnz) {
155249b5e25fSSatish Balay     for (i=0; i<mbs; i++) {
1553e32f2f54SBarry 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]);
1554de64b629SHong 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);
155549b5e25fSSatish Balay     }
155649b5e25fSSatish Balay   }
155749b5e25fSSatish Balay 
1558db4efbfdSBarry Smith   B->ops->mult             = MatMult_SeqSBAIJ_N;
1559db4efbfdSBarry Smith   B->ops->multadd          = MatMultAdd_SeqSBAIJ_N;
1560db4efbfdSBarry Smith   B->ops->multtranspose    = MatMult_SeqSBAIJ_N;
1561db4efbfdSBarry Smith   B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_N;
156226fbe8dcSKarl Rupp 
1563c5929fdfSBarry Smith   ierr  = PetscOptionsGetBool(((PetscObject)B)->options,((PetscObject)B)->prefix,"-mat_no_unroll",&flg,NULL);CHKERRQ(ierr);
156449b5e25fSSatish Balay   if (!flg) {
156549b5e25fSSatish Balay     switch (bs) {
156649b5e25fSSatish Balay     case 1:
156749b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_1;
156849b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_1;
1569431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_1;
1570431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_1;
157149b5e25fSSatish Balay       break;
157249b5e25fSSatish Balay     case 2:
157349b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_2;
157449b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_2;
1575431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_2;
1576431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_2;
157749b5e25fSSatish Balay       break;
157849b5e25fSSatish Balay     case 3:
157949b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_3;
158049b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_3;
1581431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_3;
1582431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_3;
158349b5e25fSSatish Balay       break;
158449b5e25fSSatish Balay     case 4:
158549b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_4;
158649b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_4;
1587431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_4;
1588431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_4;
158949b5e25fSSatish Balay       break;
159049b5e25fSSatish Balay     case 5:
159149b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_5;
159249b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_5;
1593431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_5;
1594431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_5;
159549b5e25fSSatish Balay       break;
159649b5e25fSSatish Balay     case 6:
159749b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_6;
159849b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_6;
1599431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_6;
1600431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_6;
160149b5e25fSSatish Balay       break;
160249b5e25fSSatish Balay     case 7:
1603de53e5efSHong Zhang       B->ops->mult             = MatMult_SeqSBAIJ_7;
160449b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_7;
1605431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_7;
1606431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_7;
160749b5e25fSSatish Balay       break;
160849b5e25fSSatish Balay     }
160949b5e25fSSatish Balay   }
161049b5e25fSSatish Balay 
161149b5e25fSSatish Balay   b->mbs = mbs;
16124dcd73b1SHong Zhang   b->nbs = nbs;
1613ab93d7beSBarry Smith   if (!skipallocation) {
16142ee49352SLisandro Dalcin     if (!b->imax) {
1615dcca6d9dSJed Brown       ierr = PetscMalloc2(mbs,&b->imax,mbs,&b->ilen);CHKERRQ(ierr);
161626fbe8dcSKarl Rupp 
1617c760cd28SBarry Smith       b->free_imax_ilen = PETSC_TRUE;
161826fbe8dcSKarl Rupp 
16193bb1ff40SBarry Smith       ierr = PetscLogObjectMemory((PetscObject)B,2*mbs*sizeof(PetscInt));CHKERRQ(ierr);
16202ee49352SLisandro Dalcin     }
162149b5e25fSSatish Balay     if (!nnz) {
1622435da068SBarry Smith       if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 5;
162349b5e25fSSatish Balay       else if (nz <= 0) nz = 1;
162426fbe8dcSKarl Rupp       for (i=0; i<mbs; i++) b->imax[i] = nz;
1625153ea458SHong Zhang       nz = nz*mbs; /* total nz */
162649b5e25fSSatish Balay     } else {
162749b5e25fSSatish Balay       nz = 0;
16288cef66ccSHong Zhang       for (i=0; i<mbs; i++) {b->imax[i] = nnz[i]; nz += nnz[i];}
162949b5e25fSSatish Balay     }
16302ee49352SLisandro Dalcin     /* b->ilen will count nonzeros in each block row so far. */
163126fbe8dcSKarl Rupp     for (i=0; i<mbs; i++) b->ilen[i] = 0;
16326c6c5352SBarry Smith     /* nz=(nz+mbs)/2; */ /* total diagonal and superdiagonal nonzero blocks */
163349b5e25fSSatish Balay 
163449b5e25fSSatish Balay     /* allocate the matrix space */
16352ee49352SLisandro Dalcin     ierr = MatSeqXAIJFreeAIJ(B,&b->a,&b->j,&b->i);CHKERRQ(ierr);
1636dcca6d9dSJed Brown     ierr = PetscMalloc3(bs2*nz,&b->a,nz,&b->j,B->rmap->N+1,&b->i);CHKERRQ(ierr);
16373bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)B,(B->rmap->N+1)*sizeof(PetscInt)+nz*(bs2*sizeof(PetscScalar)+sizeof(PetscInt)));CHKERRQ(ierr);
16386c6c5352SBarry Smith     ierr = PetscMemzero(b->a,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr);
163913f74950SBarry Smith     ierr = PetscMemzero(b->j,nz*sizeof(PetscInt));CHKERRQ(ierr);
164026fbe8dcSKarl Rupp 
164149b5e25fSSatish Balay     b->singlemalloc = PETSC_TRUE;
164249b5e25fSSatish Balay 
164349b5e25fSSatish Balay     /* pointer to beginning of each row */
1644e60cf9a0SBarry Smith     b->i[0] = 0;
164526fbe8dcSKarl Rupp     for (i=1; i<mbs+1; i++) b->i[i] = b->i[i-1] + b->imax[i-1];
164626fbe8dcSKarl Rupp 
1647e6b907acSBarry Smith     b->free_a  = PETSC_TRUE;
1648e6b907acSBarry Smith     b->free_ij = PETSC_TRUE;
1649e811da20SHong Zhang   } else {
1650e6b907acSBarry Smith     b->free_a  = PETSC_FALSE;
1651e6b907acSBarry Smith     b->free_ij = PETSC_FALSE;
1652ab93d7beSBarry Smith   }
165349b5e25fSSatish Balay 
165449b5e25fSSatish Balay   b->bs2     = bs2;
16556c6c5352SBarry Smith   b->nz      = 0;
1656b32cb4a7SJed Brown   b->maxnz   = nz;
165716cdd363SHong Zhang   b->inew    = 0;
165816cdd363SHong Zhang   b->jnew    = 0;
165916cdd363SHong Zhang   b->anew    = 0;
166016cdd363SHong Zhang   b->a2anew  = 0;
16611a3463dfSHong Zhang   b->permute = PETSC_FALSE;
1662cb7b82ddSBarry Smith 
1663cb7b82ddSBarry Smith   B->was_assembled = PETSC_FALSE;
1664cb7b82ddSBarry Smith   B->assembled     = PETSC_FALSE;
16652576faa2SJed Brown   if (realalloc) {ierr = MatSetOption(B,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);}
1666c464158bSHong Zhang   PetscFunctionReturn(0);
1667c464158bSHong Zhang }
1668153ea458SHong Zhang 
166938f409ebSLisandro Dalcin PetscErrorCode MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ(Mat B,PetscInt bs,const PetscInt ii[],const PetscInt jj[], const PetscScalar V[])
167038f409ebSLisandro Dalcin {
167138f409ebSLisandro Dalcin   PetscInt       i,j,m,nz,nz_max=0,*nnz;
167238f409ebSLisandro Dalcin   PetscScalar    *values=0;
167338f409ebSLisandro Dalcin   PetscBool      roworiented = ((Mat_SeqSBAIJ*)B->data)->roworiented;
167438f409ebSLisandro Dalcin   PetscErrorCode ierr;
167538f409ebSLisandro Dalcin   PetscFunctionBegin;
167638f409ebSLisandro Dalcin   if (bs < 1) SETERRQ1(PetscObjectComm((PetscObject)B),PETSC_ERR_ARG_OUTOFRANGE,"Invalid block size specified, must be positive but it is %D",bs);
167738f409ebSLisandro Dalcin   ierr   = PetscLayoutSetBlockSize(B->rmap,bs);CHKERRQ(ierr);
167838f409ebSLisandro Dalcin   ierr   = PetscLayoutSetBlockSize(B->cmap,bs);CHKERRQ(ierr);
167938f409ebSLisandro Dalcin   ierr   = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr);
168038f409ebSLisandro Dalcin   ierr   = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr);
168138f409ebSLisandro Dalcin   ierr   = PetscLayoutGetBlockSize(B->rmap,&bs);CHKERRQ(ierr);
168238f409ebSLisandro Dalcin   m      = B->rmap->n/bs;
168338f409ebSLisandro Dalcin 
168438f409ebSLisandro Dalcin   if (ii[0]) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"ii[0] must be 0 but it is %D",ii[0]);
1685854ce69bSBarry Smith   ierr = PetscMalloc1(m+1,&nnz);CHKERRQ(ierr);
168638f409ebSLisandro Dalcin   for (i=0; i<m; i++) {
168738f409ebSLisandro Dalcin     nz = ii[i+1] - ii[i];
168838f409ebSLisandro Dalcin     if (nz < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row %D has a negative number of columns %D",i,nz);
168938f409ebSLisandro Dalcin     nz_max = PetscMax(nz_max,nz);
169038f409ebSLisandro Dalcin     nnz[i] = nz;
169138f409ebSLisandro Dalcin   }
169238f409ebSLisandro Dalcin   ierr = MatSeqSBAIJSetPreallocation(B,bs,0,nnz);CHKERRQ(ierr);
169338f409ebSLisandro Dalcin   ierr = PetscFree(nnz);CHKERRQ(ierr);
169438f409ebSLisandro Dalcin 
169538f409ebSLisandro Dalcin   values = (PetscScalar*)V;
169638f409ebSLisandro Dalcin   if (!values) {
16971795a4d1SJed Brown     ierr = PetscCalloc1(bs*bs*nz_max,&values);CHKERRQ(ierr);
169838f409ebSLisandro Dalcin   }
169938f409ebSLisandro Dalcin   for (i=0; i<m; i++) {
170038f409ebSLisandro Dalcin     PetscInt          ncols  = ii[i+1] - ii[i];
170138f409ebSLisandro Dalcin     const PetscInt    *icols = jj + ii[i];
170238f409ebSLisandro Dalcin     if (!roworiented || bs == 1) {
170338f409ebSLisandro Dalcin       const PetscScalar *svals = values + (V ? (bs*bs*ii[i]) : 0);
170438f409ebSLisandro Dalcin       ierr = MatSetValuesBlocked_SeqSBAIJ(B,1,&i,ncols,icols,svals,INSERT_VALUES);CHKERRQ(ierr);
170538f409ebSLisandro Dalcin     } else {
170638f409ebSLisandro Dalcin       for (j=0; j<ncols; j++) {
170738f409ebSLisandro Dalcin         const PetscScalar *svals = values + (V ? (bs*bs*(ii[i]+j)) : 0);
170838f409ebSLisandro Dalcin         ierr = MatSetValuesBlocked_SeqSBAIJ(B,1,&i,1,&icols[j],svals,INSERT_VALUES);CHKERRQ(ierr);
170938f409ebSLisandro Dalcin       }
171038f409ebSLisandro Dalcin     }
171138f409ebSLisandro Dalcin   }
171238f409ebSLisandro Dalcin   if (!V) { ierr = PetscFree(values);CHKERRQ(ierr); }
171338f409ebSLisandro Dalcin   ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
171438f409ebSLisandro Dalcin   ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
171538f409ebSLisandro Dalcin   ierr = MatSetOption(B,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
171638f409ebSLisandro Dalcin   PetscFunctionReturn(0);
171738f409ebSLisandro Dalcin }
171838f409ebSLisandro Dalcin 
1719db4efbfdSBarry Smith /*
1720db4efbfdSBarry Smith    This is used to set the numeric factorization for both Cholesky and ICC symbolic factorization
1721db4efbfdSBarry Smith */
1722ace3abfcSBarry Smith PetscErrorCode MatSeqSBAIJSetNumericFactorization_inplace(Mat B,PetscBool natural)
1723db4efbfdSBarry Smith {
1724db4efbfdSBarry Smith   PetscErrorCode ierr;
1725ace3abfcSBarry Smith   PetscBool      flg = PETSC_FALSE;
1726db4efbfdSBarry Smith   PetscInt       bs  = B->rmap->bs;
1727db4efbfdSBarry Smith 
1728db4efbfdSBarry Smith   PetscFunctionBegin;
1729c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(((PetscObject)B)->options,((PetscObject)B)->prefix,"-mat_no_unroll",&flg,NULL);CHKERRQ(ierr);
1730db4efbfdSBarry Smith   if (flg) bs = 8;
1731db4efbfdSBarry Smith 
1732db4efbfdSBarry Smith   if (!natural) {
1733db4efbfdSBarry Smith     switch (bs) {
1734db4efbfdSBarry Smith     case 1:
1735d595f711SHong Zhang       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_inplace;
1736db4efbfdSBarry Smith       break;
1737db4efbfdSBarry Smith     case 2:
1738db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2;
1739db4efbfdSBarry Smith       break;
1740db4efbfdSBarry Smith     case 3:
1741db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3;
1742db4efbfdSBarry Smith       break;
1743db4efbfdSBarry Smith     case 4:
1744db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4;
1745db4efbfdSBarry Smith       break;
1746db4efbfdSBarry Smith     case 5:
1747db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5;
1748db4efbfdSBarry Smith       break;
1749db4efbfdSBarry Smith     case 6:
1750db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6;
1751db4efbfdSBarry Smith       break;
1752db4efbfdSBarry Smith     case 7:
1753db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7;
1754db4efbfdSBarry Smith       break;
1755db4efbfdSBarry Smith     default:
1756db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N;
1757db4efbfdSBarry Smith       break;
1758db4efbfdSBarry Smith     }
1759db4efbfdSBarry Smith   } else {
1760db4efbfdSBarry Smith     switch (bs) {
1761db4efbfdSBarry Smith     case 1:
1762d595f711SHong Zhang       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering_inplace;
1763db4efbfdSBarry Smith       break;
1764db4efbfdSBarry Smith     case 2:
1765db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering;
1766db4efbfdSBarry Smith       break;
1767db4efbfdSBarry Smith     case 3:
1768db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering;
1769db4efbfdSBarry Smith       break;
1770db4efbfdSBarry Smith     case 4:
1771db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering;
1772db4efbfdSBarry Smith       break;
1773db4efbfdSBarry Smith     case 5:
1774db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering;
1775db4efbfdSBarry Smith       break;
1776db4efbfdSBarry Smith     case 6:
1777db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering;
1778db4efbfdSBarry Smith       break;
1779db4efbfdSBarry Smith     case 7:
1780db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering;
1781db4efbfdSBarry Smith       break;
1782db4efbfdSBarry Smith     default:
1783db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering;
1784db4efbfdSBarry Smith       break;
1785db4efbfdSBarry Smith     }
1786db4efbfdSBarry Smith   }
1787db4efbfdSBarry Smith   PetscFunctionReturn(0);
1788db4efbfdSBarry Smith }
1789db4efbfdSBarry Smith 
1790cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqAIJ(Mat, MatType,MatReuse,Mat*);
1791cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqBAIJ(Mat, MatType,MatReuse,Mat*);
1792d769727bSBarry Smith 
1793cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatGetFactor_seqsbaij_petsc(Mat A,MatFactorType ftype,Mat *B)
17945c9eb25fSBarry Smith {
1795d0f46423SBarry Smith   PetscInt       n = A->rmap->n;
17965c9eb25fSBarry Smith   PetscErrorCode ierr;
17975c9eb25fSBarry Smith 
17985c9eb25fSBarry Smith   PetscFunctionBegin;
17990e92d65fSHong Zhang #if defined(PETSC_USE_COMPLEX)
18000e92d65fSHong Zhang   if (A->hermitian) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Hermitian Factor is not supported");
18010e92d65fSHong Zhang #endif
1802ce94432eSBarry Smith   ierr = MatCreate(PetscObjectComm((PetscObject)A),B);CHKERRQ(ierr);
18035c9eb25fSBarry Smith   ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr);
18045c9eb25fSBarry Smith   if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) {
18055c9eb25fSBarry Smith     ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr);
18060298fd71SBarry Smith     ierr = MatSeqSBAIJSetPreallocation(*B,A->rmap->bs,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr);
180726fbe8dcSKarl Rupp 
18087b056e98SHong Zhang     (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqSBAIJ;
1809c6d0d4f0SHong Zhang     (*B)->ops->iccfactorsymbolic      = MatICCFactorSymbolic_SeqSBAIJ;
1810e32f2f54SBarry Smith   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Factor type not supported");
181100c67f3bSHong Zhang 
1812d5f3da31SBarry Smith   (*B)->factortype = ftype;
181300c67f3bSHong Zhang   ierr = PetscFree((*B)->solvertype);CHKERRQ(ierr);
181400c67f3bSHong Zhang   ierr = PetscStrallocpy(MATSOLVERPETSC,&(*B)->solvertype);CHKERRQ(ierr);
18155c9eb25fSBarry Smith   PetscFunctionReturn(0);
18165c9eb25fSBarry Smith }
18175c9eb25fSBarry Smith 
18188397e458SBarry Smith /*@C
18198397e458SBarry Smith    MatSeqSBAIJGetArray - gives access to the array where the data for a MATSEQSBAIJ matrix is stored
18208397e458SBarry Smith 
18218397e458SBarry Smith    Not Collective
18228397e458SBarry Smith 
18238397e458SBarry Smith    Input Parameter:
18248397e458SBarry Smith .  mat - a MATSEQSBAIJ matrix
18258397e458SBarry Smith 
18268397e458SBarry Smith    Output Parameter:
18278397e458SBarry Smith .   array - pointer to the data
18288397e458SBarry Smith 
18298397e458SBarry Smith    Level: intermediate
18308397e458SBarry Smith 
18318397e458SBarry Smith .seealso: MatSeqSBAIJRestoreArray(), MatSeqAIJGetArray(), MatSeqAIJRestoreArray()
18328397e458SBarry Smith @*/
18338397e458SBarry Smith PetscErrorCode  MatSeqSBAIJGetArray(Mat A,PetscScalar **array)
18348397e458SBarry Smith {
18358397e458SBarry Smith   PetscErrorCode ierr;
18368397e458SBarry Smith 
18378397e458SBarry Smith   PetscFunctionBegin;
18388397e458SBarry Smith   ierr = PetscUseMethod(A,"MatSeqSBAIJGetArray_C",(Mat,PetscScalar**),(A,array));CHKERRQ(ierr);
18398397e458SBarry Smith   PetscFunctionReturn(0);
18408397e458SBarry Smith }
18418397e458SBarry Smith 
18428397e458SBarry Smith /*@C
18438397e458SBarry Smith    MatSeqSBAIJRestoreArray - returns access to the array where the data for a MATSEQSBAIJ matrix is stored obtained by MatSeqSBAIJGetArray()
18448397e458SBarry Smith 
18458397e458SBarry Smith    Not Collective
18468397e458SBarry Smith 
18478397e458SBarry Smith    Input Parameters:
18488397e458SBarry Smith .  mat - a MATSEQSBAIJ matrix
18498397e458SBarry Smith .  array - pointer to the data
18508397e458SBarry Smith 
18518397e458SBarry Smith    Level: intermediate
18528397e458SBarry Smith 
18538397e458SBarry Smith .seealso: MatSeqSBAIJGetArray(), MatSeqAIJGetArray(), MatSeqAIJRestoreArray()
18548397e458SBarry Smith @*/
18558397e458SBarry Smith PetscErrorCode  MatSeqSBAIJRestoreArray(Mat A,PetscScalar **array)
18568397e458SBarry Smith {
18578397e458SBarry Smith   PetscErrorCode ierr;
18588397e458SBarry Smith 
18598397e458SBarry Smith   PetscFunctionBegin;
18608397e458SBarry Smith   ierr = PetscUseMethod(A,"MatSeqSBAIJRestoreArray_C",(Mat,PetscScalar**),(A,array));CHKERRQ(ierr);
18618397e458SBarry Smith   PetscFunctionReturn(0);
18628397e458SBarry Smith }
18638397e458SBarry Smith 
18640bad9183SKris Buschelman /*MC
1865fafad747SKris Buschelman   MATSEQSBAIJ - MATSEQSBAIJ = "seqsbaij" - A matrix type to be used for sequential symmetric block sparse matrices,
18660bad9183SKris Buschelman   based on block compressed sparse row format.  Only the upper triangular portion of the matrix is stored.
18670bad9183SKris Buschelman 
1868828413b8SBarry Smith   For complex numbers by default this matrix is symmetric, NOT Hermitian symmetric. To make it Hermitian symmetric you
186971dad5bbSBarry Smith   can call MatSetOption(Mat, MAT_HERMITIAN); after MatAssemblyEnd()
1870828413b8SBarry Smith 
18710bad9183SKris Buschelman   Options Database Keys:
18720bad9183SKris Buschelman   . -mat_type seqsbaij - sets the matrix type to "seqsbaij" during a call to MatSetFromOptions()
18730bad9183SKris Buschelman 
187471dad5bbSBarry Smith   Notes: By default if you insert values into the lower triangular part of the matrix they are simply ignored (since they are not
187571dad5bbSBarry 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
187671dad5bbSBarry 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.
187771dad5bbSBarry Smith 
187871dad5bbSBarry Smith 
18790bad9183SKris Buschelman   Level: beginner
18800bad9183SKris Buschelman 
18810bad9183SKris Buschelman   .seealso: MatCreateSeqSBAIJ
18820bad9183SKris Buschelman M*/
18830bad9183SKris Buschelman 
1884cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqSBSTRM(Mat, MatType,MatReuse,Mat*);
1885aa5a9175SDahai Guo 
18868cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatCreate_SeqSBAIJ(Mat B)
1887a23d5eceSKris Buschelman {
1888a23d5eceSKris Buschelman   Mat_SeqSBAIJ   *b;
1889dfbe8321SBarry Smith   PetscErrorCode ierr;
189013f74950SBarry Smith   PetscMPIInt    size;
1891ace3abfcSBarry Smith   PetscBool      no_unroll = PETSC_FALSE,no_inode = PETSC_FALSE;
1892a23d5eceSKris Buschelman 
1893a23d5eceSKris Buschelman   PetscFunctionBegin;
1894ce94432eSBarry Smith   ierr = MPI_Comm_size(PetscObjectComm((PetscObject)B),&size);CHKERRQ(ierr);
1895e32f2f54SBarry Smith   if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Comm must be of size 1");
1896a23d5eceSKris Buschelman 
1897b00a9115SJed Brown   ierr    = PetscNewLog(B,&b);CHKERRQ(ierr);
1898a23d5eceSKris Buschelman   B->data = (void*)b;
1899a23d5eceSKris Buschelman   ierr    = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr);
190026fbe8dcSKarl Rupp 
1901a23d5eceSKris Buschelman   B->ops->destroy    = MatDestroy_SeqSBAIJ;
1902a23d5eceSKris Buschelman   B->ops->view       = MatView_SeqSBAIJ;
1903a23d5eceSKris Buschelman   b->row             = 0;
1904a23d5eceSKris Buschelman   b->icol            = 0;
1905a23d5eceSKris Buschelman   b->reallocs        = 0;
1906a23d5eceSKris Buschelman   b->saved_values    = 0;
19070def2e27SBarry Smith   b->inode.limit     = 5;
19080def2e27SBarry Smith   b->inode.max_limit = 5;
1909a23d5eceSKris Buschelman 
1910a23d5eceSKris Buschelman   b->roworiented        = PETSC_TRUE;
1911a23d5eceSKris Buschelman   b->nonew              = 0;
1912a23d5eceSKris Buschelman   b->diag               = 0;
1913a23d5eceSKris Buschelman   b->solve_work         = 0;
1914a23d5eceSKris Buschelman   b->mult_work          = 0;
1915a23d5eceSKris Buschelman   B->spptr              = 0;
1916f2cbd3d5SJed Brown   B->info.nz_unneeded   = (PetscReal)b->maxnz*b->bs2;
1917a9817697SBarry Smith   b->keepnonzeropattern = PETSC_FALSE;
1918a23d5eceSKris Buschelman 
1919a23d5eceSKris Buschelman   b->inew    = 0;
1920a23d5eceSKris Buschelman   b->jnew    = 0;
1921a23d5eceSKris Buschelman   b->anew    = 0;
1922a23d5eceSKris Buschelman   b->a2anew  = 0;
1923a23d5eceSKris Buschelman   b->permute = PETSC_FALSE;
1924a23d5eceSKris Buschelman 
192571dad5bbSBarry Smith   b->ignore_ltriangular = PETSC_TRUE;
192626fbe8dcSKarl Rupp 
1927c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(((PetscObject)B)->options,((PetscObject)B)->prefix,"-mat_ignore_lower_triangular",&b->ignore_ltriangular,NULL);CHKERRQ(ierr);
1928941593c8SHong Zhang 
1929f5edf698SHong Zhang   b->getrow_utriangular = PETSC_FALSE;
193026fbe8dcSKarl Rupp 
1931c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(((PetscObject)B)->options,((PetscObject)B)->prefix,"-mat_getrow_uppertriangular",&b->getrow_utriangular,NULL);CHKERRQ(ierr);
1932f5edf698SHong Zhang 
19338397e458SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJGetArray_C",MatSeqSBAIJGetArray_SeqSBAIJ);CHKERRQ(ierr);
19348397e458SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJRestoreArray_C",MatSeqSBAIJRestoreArray_SeqSBAIJ);CHKERRQ(ierr);
1935bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatStoreValues_C",MatStoreValues_SeqSBAIJ);CHKERRQ(ierr);
1936bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatRetrieveValues_C",MatRetrieveValues_SeqSBAIJ);CHKERRQ(ierr);
1937bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetColumnIndices_C",MatSeqSBAIJSetColumnIndices_SeqSBAIJ);CHKERRQ(ierr);
1938bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqaij_C",MatConvert_SeqSBAIJ_SeqAIJ);CHKERRQ(ierr);
1939bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqbaij_C",MatConvert_SeqSBAIJ_SeqBAIJ);CHKERRQ(ierr);
1940bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetPreallocation_C",MatSeqSBAIJSetPreallocation_SeqSBAIJ);CHKERRQ(ierr);
194138f409ebSLisandro Dalcin   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetPreallocationCSR_C",MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ);CHKERRQ(ierr);
19426214f412SHong Zhang #if defined(PETSC_HAVE_ELEMENTAL)
19436214f412SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_elemental_C",MatConvert_SeqSBAIJ_Elemental);CHKERRQ(ierr);
19446214f412SHong Zhang #endif
194523ce1328SBarry Smith 
194623ce1328SBarry Smith   B->symmetric                  = PETSC_TRUE;
194723ce1328SBarry Smith   B->structurally_symmetric     = PETSC_TRUE;
194823ce1328SBarry Smith   B->symmetric_set              = PETSC_TRUE;
194923ce1328SBarry Smith   B->structurally_symmetric_set = PETSC_TRUE;
195026fbe8dcSKarl Rupp 
195117667f90SBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)B,MATSEQSBAIJ);CHKERRQ(ierr);
19520def2e27SBarry Smith 
1953ce94432eSBarry Smith   ierr = PetscOptionsBegin(PetscObjectComm((PetscObject)B),((PetscObject)B)->prefix,"Options for SEQSBAIJ matrix","Mat");CHKERRQ(ierr);
19540298fd71SBarry Smith   ierr = PetscOptionsBool("-mat_no_unroll","Do not optimize for inodes (slower)",NULL,no_unroll,&no_unroll,NULL);CHKERRQ(ierr);
195526fbe8dcSKarl Rupp   if (no_unroll) {
195626fbe8dcSKarl Rupp     ierr = PetscInfo(B,"Not using Inode routines due to -mat_no_unroll\n");CHKERRQ(ierr);
195726fbe8dcSKarl Rupp   }
19580298fd71SBarry Smith   ierr = PetscOptionsBool("-mat_no_inode","Do not optimize for inodes (slower)",NULL,no_inode,&no_inode,NULL);CHKERRQ(ierr);
195926fbe8dcSKarl Rupp   if (no_inode) {
196026fbe8dcSKarl Rupp     ierr = PetscInfo(B,"Not using Inode routines due to -mat_no_inode\n");CHKERRQ(ierr);
196126fbe8dcSKarl Rupp   }
19620298fd71SBarry Smith   ierr = PetscOptionsInt("-mat_inode_limit","Do not use inodes larger then this value",NULL,b->inode.limit,&b->inode.limit,NULL);CHKERRQ(ierr);
19630def2e27SBarry Smith   ierr = PetscOptionsEnd();CHKERRQ(ierr);
1964ace3abfcSBarry Smith   b->inode.use = (PetscBool)(!(no_unroll || no_inode));
19650def2e27SBarry Smith   if (b->inode.limit > b->inode.max_limit) b->inode.limit = b->inode.max_limit;
1966a23d5eceSKris Buschelman   PetscFunctionReturn(0);
1967a23d5eceSKris Buschelman }
1968a23d5eceSKris Buschelman 
1969a23d5eceSKris Buschelman /*@C
1970a23d5eceSKris Buschelman    MatSeqSBAIJSetPreallocation - Creates a sparse symmetric matrix in block AIJ (block
1971a23d5eceSKris Buschelman    compressed row) format.  For good matrix assembly performance the
1972a23d5eceSKris Buschelman    user should preallocate the matrix storage by setting the parameter nz
1973a23d5eceSKris Buschelman    (or the array nnz).  By setting these parameters accurately, performance
1974a23d5eceSKris Buschelman    during matrix assembly can be increased by more than a factor of 50.
1975a23d5eceSKris Buschelman 
1976a23d5eceSKris Buschelman    Collective on Mat
1977a23d5eceSKris Buschelman 
1978a23d5eceSKris Buschelman    Input Parameters:
19791c4f3114SJed Brown +  B - the symmetric matrix
1980bb7ae925SBarry 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
1981bb7ae925SBarry Smith           blocksize always defines the size of the blocks. The column blocksize sets the blocksize of the vectors obtained with MatCreateVecs()
1982a23d5eceSKris Buschelman .  nz - number of block nonzeros per block row (same for all rows)
1983a23d5eceSKris Buschelman -  nnz - array containing the number of block nonzeros in the upper triangular plus
19840298fd71SBarry Smith          diagonal portion of each block (possibly different for each block row) or NULL
1985a23d5eceSKris Buschelman 
1986a23d5eceSKris Buschelman    Options Database Keys:
1987a23d5eceSKris Buschelman .   -mat_no_unroll - uses code that does not unroll the loops in the
1988a23d5eceSKris Buschelman                      block calculations (much slower)
1989db4efbfdSBarry Smith .   -mat_block_size - size of the blocks to use (only works if a negative bs is passed in
1990a23d5eceSKris Buschelman 
1991a23d5eceSKris Buschelman    Level: intermediate
1992a23d5eceSKris Buschelman 
1993a23d5eceSKris Buschelman    Notes:
1994a23d5eceSKris Buschelman    Specify the preallocated storage with either nz or nnz (not both).
19950298fd71SBarry Smith    Set nz=PETSC_DEFAULT and nnz=NULL for PETSc to control dynamic memory
1996a7f22e61SSatish Balay    allocation.  See Users-Manual: ch_mat for details.
1997a23d5eceSKris Buschelman 
1998aa95bbe8SBarry Smith    You can call MatGetInfo() to get information on how effective the preallocation was;
1999aa95bbe8SBarry Smith    for example the fields mallocs,nz_allocated,nz_used,nz_unneeded;
2000aa95bbe8SBarry Smith    You can also run with the option -info and look for messages with the string
2001aa95bbe8SBarry Smith    malloc in them to see if additional memory allocation was needed.
2002aa95bbe8SBarry Smith 
200349a6f317SBarry Smith    If the nnz parameter is given then the nz parameter is ignored
200449a6f317SBarry Smith 
200549a6f317SBarry Smith 
200669b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateSBAIJ()
2007a23d5eceSKris Buschelman @*/
20087087cfbeSBarry Smith PetscErrorCode  MatSeqSBAIJSetPreallocation(Mat B,PetscInt bs,PetscInt nz,const PetscInt nnz[])
200913f74950SBarry Smith {
20104ac538c5SBarry Smith   PetscErrorCode ierr;
2011a23d5eceSKris Buschelman 
2012a23d5eceSKris Buschelman   PetscFunctionBegin;
20136ba663aaSJed Brown   PetscValidHeaderSpecific(B,MAT_CLASSID,1);
20146ba663aaSJed Brown   PetscValidType(B,1);
20156ba663aaSJed Brown   PetscValidLogicalCollectiveInt(B,bs,2);
20164ac538c5SBarry Smith   ierr = PetscTryMethod(B,"MatSeqSBAIJSetPreallocation_C",(Mat,PetscInt,PetscInt,const PetscInt[]),(B,bs,nz,nnz));CHKERRQ(ierr);
2017a23d5eceSKris Buschelman   PetscFunctionReturn(0);
2018a23d5eceSKris Buschelman }
201949b5e25fSSatish Balay 
202038f409ebSLisandro Dalcin /*@C
202138f409ebSLisandro Dalcin    MatSeqSBAIJSetPreallocationCSR - Allocates memory for a sparse sequential matrix in symmetric block AIJ format.
202238f409ebSLisandro Dalcin 
202338f409ebSLisandro Dalcin    Input Parameters:
20241c4f3114SJed Brown +  B - the matrix
2025eab78319SHong Zhang .  bs - size of block, the blocks are ALWAYS square.
202638f409ebSLisandro Dalcin .  i - the indices into j for the start of each local row (starts with zero)
202738f409ebSLisandro Dalcin .  j - the column indices for each local row (starts with zero) these must be sorted for each row
202838f409ebSLisandro Dalcin -  v - optional values in the matrix
202938f409ebSLisandro Dalcin 
203038f409ebSLisandro Dalcin    Level: developer
203138f409ebSLisandro Dalcin 
203238f409ebSLisandro Dalcin    Notes:
203338f409ebSLisandro Dalcin    The order of the entries in values is specified by the MatOption MAT_ROW_ORIENTED.  For example, C programs
203438f409ebSLisandro 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
203538f409ebSLisandro Dalcin    over rows within a block and the last index is over columns within a block row.  Fortran programs will likely set
203638f409ebSLisandro 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
203738f409ebSLisandro Dalcin    block column and the second index is over columns within a block.
203838f409ebSLisandro Dalcin 
203938f409ebSLisandro Dalcin .keywords: matrix, block, aij, compressed row, sparse
204038f409ebSLisandro Dalcin 
204138f409ebSLisandro Dalcin .seealso: MatCreate(), MatCreateSeqSBAIJ(), MatSetValuesBlocked(), MatSeqSBAIJSetPreallocation(), MATSEQSBAIJ
204238f409ebSLisandro Dalcin @*/
204338f409ebSLisandro Dalcin PetscErrorCode MatSeqSBAIJSetPreallocationCSR(Mat B,PetscInt bs,const PetscInt i[],const PetscInt j[], const PetscScalar v[])
204438f409ebSLisandro Dalcin {
204538f409ebSLisandro Dalcin   PetscErrorCode ierr;
204638f409ebSLisandro Dalcin 
204738f409ebSLisandro Dalcin   PetscFunctionBegin;
204838f409ebSLisandro Dalcin   PetscValidHeaderSpecific(B,MAT_CLASSID,1);
204938f409ebSLisandro Dalcin   PetscValidType(B,1);
205038f409ebSLisandro Dalcin   PetscValidLogicalCollectiveInt(B,bs,2);
205138f409ebSLisandro Dalcin   ierr = PetscTryMethod(B,"MatSeqSBAIJSetPreallocationCSR_C",(Mat,PetscInt,const PetscInt[],const PetscInt[],const PetscScalar[]),(B,bs,i,j,v));CHKERRQ(ierr);
205238f409ebSLisandro Dalcin   PetscFunctionReturn(0);
205338f409ebSLisandro Dalcin }
205438f409ebSLisandro Dalcin 
2055c464158bSHong Zhang /*@C
2056c464158bSHong Zhang    MatCreateSeqSBAIJ - Creates a sparse symmetric matrix in block AIJ (block
2057c464158bSHong Zhang    compressed row) format.  For good matrix assembly performance the
2058c464158bSHong Zhang    user should preallocate the matrix storage by setting the parameter nz
2059c464158bSHong Zhang    (or the array nnz).  By setting these parameters accurately, performance
2060c464158bSHong Zhang    during matrix assembly can be increased by more than a factor of 50.
206149b5e25fSSatish Balay 
2062c464158bSHong Zhang    Collective on MPI_Comm
2063c464158bSHong Zhang 
2064c464158bSHong Zhang    Input Parameters:
2065c464158bSHong Zhang +  comm - MPI communicator, set to PETSC_COMM_SELF
2066bb7ae925SBarry 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
2067bb7ae925SBarry Smith           blocksize always defines the size of the blocks. The column blocksize sets the blocksize of the vectors obtained with MatCreateVecs()
2068c464158bSHong Zhang .  m - number of rows, or number of columns
2069c464158bSHong Zhang .  nz - number of block nonzeros per block row (same for all rows)
2070744e8345SSatish Balay -  nnz - array containing the number of block nonzeros in the upper triangular plus
20710298fd71SBarry Smith          diagonal portion of each block (possibly different for each block row) or NULL
2072c464158bSHong Zhang 
2073c464158bSHong Zhang    Output Parameter:
2074c464158bSHong Zhang .  A - the symmetric matrix
2075c464158bSHong Zhang 
2076c464158bSHong Zhang    Options Database Keys:
2077c464158bSHong Zhang .   -mat_no_unroll - uses code that does not unroll the loops in the
2078c464158bSHong Zhang                      block calculations (much slower)
2079c464158bSHong Zhang .    -mat_block_size - size of the blocks to use
2080c464158bSHong Zhang 
2081c464158bSHong Zhang    Level: intermediate
2082c464158bSHong Zhang 
2083175b88e8SBarry Smith    It is recommended that one use the MatCreate(), MatSetType() and/or MatSetFromOptions(),
2084ae1d86c5SBarry Smith    MatXXXXSetPreallocation() paradgm instead of this routine directly.
2085175b88e8SBarry Smith    [MatXXXXSetPreallocation() is, for example, MatSeqAIJSetPreallocation]
2086175b88e8SBarry Smith 
2087c464158bSHong Zhang    Notes:
20886d6d819aSHong Zhang    The number of rows and columns must be divisible by blocksize.
20896d6d819aSHong Zhang    This matrix type does not support complex Hermitian operation.
2090c464158bSHong Zhang 
2091c464158bSHong Zhang    Specify the preallocated storage with either nz or nnz (not both).
20920298fd71SBarry Smith    Set nz=PETSC_DEFAULT and nnz=NULL for PETSc to control dynamic memory
2093a7f22e61SSatish Balay    allocation.  See Users-Manual: ch_mat for details.
2094c464158bSHong Zhang 
209549a6f317SBarry Smith    If the nnz parameter is given then the nz parameter is ignored
209649a6f317SBarry Smith 
209769b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateSBAIJ()
2098c464158bSHong Zhang @*/
20997087cfbeSBarry Smith PetscErrorCode  MatCreateSeqSBAIJ(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt nz,const PetscInt nnz[],Mat *A)
2100c464158bSHong Zhang {
2101dfbe8321SBarry Smith   PetscErrorCode ierr;
2102c464158bSHong Zhang 
2103c464158bSHong Zhang   PetscFunctionBegin;
2104f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,A);CHKERRQ(ierr);
2105f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*A,m,n,m,n);CHKERRQ(ierr);
2106c464158bSHong Zhang   ierr = MatSetType(*A,MATSEQSBAIJ);CHKERRQ(ierr);
2107367daffbSBarry Smith   ierr = MatSeqSBAIJSetPreallocation(*A,bs,nz,(PetscInt*)nnz);CHKERRQ(ierr);
210849b5e25fSSatish Balay   PetscFunctionReturn(0);
210949b5e25fSSatish Balay }
211049b5e25fSSatish Balay 
2111dfbe8321SBarry Smith PetscErrorCode MatDuplicate_SeqSBAIJ(Mat A,MatDuplicateOption cpvalues,Mat *B)
211249b5e25fSSatish Balay {
211349b5e25fSSatish Balay   Mat            C;
211449b5e25fSSatish Balay   Mat_SeqSBAIJ   *c,*a = (Mat_SeqSBAIJ*)A->data;
21156849ba73SBarry Smith   PetscErrorCode ierr;
2116b40805acSSatish Balay   PetscInt       i,mbs = a->mbs,nz = a->nz,bs2 =a->bs2;
211749b5e25fSSatish Balay 
211849b5e25fSSatish Balay   PetscFunctionBegin;
2119e32f2f54SBarry Smith   if (a->i[mbs] != nz) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Corrupt matrix");
212049b5e25fSSatish Balay 
212149b5e25fSSatish Balay   *B   = 0;
2122ce94432eSBarry Smith   ierr = MatCreate(PetscObjectComm((PetscObject)A),&C);CHKERRQ(ierr);
2123d0f46423SBarry Smith   ierr = MatSetSizes(C,A->rmap->N,A->cmap->n,A->rmap->N,A->cmap->n);CHKERRQ(ierr);
21248e9a0fb8SHong Zhang   ierr = MatSetType(C,MATSEQSBAIJ);CHKERRQ(ierr);
21251d5dac46SHong Zhang   ierr = PetscMemcpy(C->ops,A->ops,sizeof(struct _MatOps));CHKERRQ(ierr);
2126692f9cbeSHong Zhang   c    = (Mat_SeqSBAIJ*)C->data;
2127692f9cbeSHong Zhang 
2128273d9f13SBarry Smith   C->preallocated       = PETSC_TRUE;
2129d5f3da31SBarry Smith   C->factortype         = A->factortype;
213049b5e25fSSatish Balay   c->row                = 0;
213149b5e25fSSatish Balay   c->icol               = 0;
213249b5e25fSSatish Balay   c->saved_values       = 0;
2133a9817697SBarry Smith   c->keepnonzeropattern = a->keepnonzeropattern;
213449b5e25fSSatish Balay   C->assembled          = PETSC_TRUE;
213549b5e25fSSatish Balay 
21361e1e43feSBarry Smith   ierr   = PetscLayoutReference(A->rmap,&C->rmap);CHKERRQ(ierr);
21371e1e43feSBarry Smith   ierr   = PetscLayoutReference(A->cmap,&C->cmap);CHKERRQ(ierr);
213849b5e25fSSatish Balay   c->bs2 = a->bs2;
213949b5e25fSSatish Balay   c->mbs = a->mbs;
214049b5e25fSSatish Balay   c->nbs = a->nbs;
214149b5e25fSSatish Balay 
2142c760cd28SBarry Smith   if  (cpvalues == MAT_SHARE_NONZERO_PATTERN) {
2143c760cd28SBarry Smith     c->imax           = a->imax;
2144c760cd28SBarry Smith     c->ilen           = a->ilen;
2145c760cd28SBarry Smith     c->free_imax_ilen = PETSC_FALSE;
2146c760cd28SBarry Smith   } else {
2147dcca6d9dSJed Brown     ierr = PetscMalloc2((mbs+1),&c->imax,(mbs+1),&c->ilen);CHKERRQ(ierr);
21483bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)C,2*(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr);
214949b5e25fSSatish Balay     for (i=0; i<mbs; i++) {
215049b5e25fSSatish Balay       c->imax[i] = a->imax[i];
215149b5e25fSSatish Balay       c->ilen[i] = a->ilen[i];
215249b5e25fSSatish Balay     }
2153c760cd28SBarry Smith     c->free_imax_ilen = PETSC_TRUE;
2154c760cd28SBarry Smith   }
215549b5e25fSSatish Balay 
215649b5e25fSSatish Balay   /* allocate the matrix space */
21574da8f245SBarry Smith   if (cpvalues == MAT_SHARE_NONZERO_PATTERN) {
2158785e854fSJed Brown     ierr            = PetscMalloc1(bs2*nz,&c->a);CHKERRQ(ierr);
21593bb1ff40SBarry Smith     ierr            = PetscLogObjectMemory((PetscObject)C,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr);
216044e1c64aSLisandro Dalcin     c->i            = a->i;
216144e1c64aSLisandro Dalcin     c->j            = a->j;
21624da8f245SBarry Smith     c->singlemalloc = PETSC_FALSE;
216344e1c64aSLisandro Dalcin     c->free_a       = PETSC_TRUE;
21644da8f245SBarry Smith     c->free_ij      = PETSC_FALSE;
21654da8f245SBarry Smith     c->parent       = A;
21664da8f245SBarry Smith     ierr            = PetscObjectReference((PetscObject)A);CHKERRQ(ierr);
21674da8f245SBarry Smith     ierr            = MatSetOption(A,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
21684da8f245SBarry Smith     ierr            = MatSetOption(C,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
21694da8f245SBarry Smith   } else {
2170dcca6d9dSJed Brown     ierr            = PetscMalloc3(bs2*nz,&c->a,nz,&c->j,mbs+1,&c->i);CHKERRQ(ierr);
217113f74950SBarry Smith     ierr            = PetscMemcpy(c->i,a->i,(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr);
21723bb1ff40SBarry Smith     ierr            = PetscLogObjectMemory((PetscObject)C,(mbs+1)*sizeof(PetscInt) + nz*(bs2*sizeof(MatScalar) + sizeof(PetscInt)));CHKERRQ(ierr);
21734da8f245SBarry Smith     c->singlemalloc = PETSC_TRUE;
217444e1c64aSLisandro Dalcin     c->free_a       = PETSC_TRUE;
21754da8f245SBarry Smith     c->free_ij      = PETSC_TRUE;
21764da8f245SBarry Smith   }
217749b5e25fSSatish Balay   if (mbs > 0) {
21784da8f245SBarry Smith     if (cpvalues != MAT_SHARE_NONZERO_PATTERN) {
217913f74950SBarry Smith       ierr = PetscMemcpy(c->j,a->j,nz*sizeof(PetscInt));CHKERRQ(ierr);
21804da8f245SBarry Smith     }
218149b5e25fSSatish Balay     if (cpvalues == MAT_COPY_VALUES) {
218249b5e25fSSatish Balay       ierr = PetscMemcpy(c->a,a->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr);
218349b5e25fSSatish Balay     } else {
218449b5e25fSSatish Balay       ierr = PetscMemzero(c->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr);
218549b5e25fSSatish Balay     }
2186a1c3900fSBarry Smith     if (a->jshort) {
218744e1c64aSLisandro Dalcin       /* cannot share jshort, it is reallocated in MatAssemblyEnd_SeqSBAIJ() */
218844e1c64aSLisandro Dalcin       /* if the parent matrix is reassembled, this child matrix will never notice */
2189785e854fSJed Brown       ierr = PetscMalloc1(nz,&c->jshort);CHKERRQ(ierr);
21903bb1ff40SBarry Smith       ierr = PetscLogObjectMemory((PetscObject)C,nz*sizeof(unsigned short));CHKERRQ(ierr);
2191a1c3900fSBarry Smith       ierr = PetscMemcpy(c->jshort,a->jshort,nz*sizeof(unsigned short));CHKERRQ(ierr);
219226fbe8dcSKarl Rupp 
21934da8f245SBarry Smith       c->free_jshort = PETSC_TRUE;
21944da8f245SBarry Smith     }
2195a1c3900fSBarry Smith   }
219649b5e25fSSatish Balay 
219749b5e25fSSatish Balay   c->roworiented = a->roworiented;
219849b5e25fSSatish Balay   c->nonew       = a->nonew;
219949b5e25fSSatish Balay 
220049b5e25fSSatish Balay   if (a->diag) {
2201c760cd28SBarry Smith     if (cpvalues == MAT_SHARE_NONZERO_PATTERN) {
2202c760cd28SBarry Smith       c->diag      = a->diag;
2203c760cd28SBarry Smith       c->free_diag = PETSC_FALSE;
2204c760cd28SBarry Smith     } else {
2205785e854fSJed Brown       ierr = PetscMalloc1(mbs,&c->diag);CHKERRQ(ierr);
22063bb1ff40SBarry Smith       ierr = PetscLogObjectMemory((PetscObject)C,mbs*sizeof(PetscInt));CHKERRQ(ierr);
220726fbe8dcSKarl Rupp       for (i=0; i<mbs; i++) c->diag[i] = a->diag[i];
2208c760cd28SBarry Smith       c->free_diag = PETSC_TRUE;
2209c760cd28SBarry Smith     }
221044e1c64aSLisandro Dalcin   }
22116c6c5352SBarry Smith   c->nz         = a->nz;
2212f2cbd3d5SJed Brown   c->maxnz      = a->nz; /* Since we allocate exactly the right amount */
221349b5e25fSSatish Balay   c->solve_work = 0;
221449b5e25fSSatish Balay   c->mult_work  = 0;
221526fbe8dcSKarl Rupp 
221649b5e25fSSatish Balay   *B   = C;
2217140e18c1SBarry Smith   ierr = PetscFunctionListDuplicate(((PetscObject)A)->qlist,&((PetscObject)C)->qlist);CHKERRQ(ierr);
221849b5e25fSSatish Balay   PetscFunctionReturn(0);
221949b5e25fSSatish Balay }
222049b5e25fSSatish Balay 
2221112444f4SShri Abhyankar PetscErrorCode MatLoad_SeqSBAIJ(Mat newmat,PetscViewer viewer)
22222f480046SShri Abhyankar {
22232f480046SShri Abhyankar   Mat_SeqSBAIJ   *a;
22242f480046SShri Abhyankar   PetscErrorCode ierr;
22252f480046SShri Abhyankar   int            fd;
22262f480046SShri Abhyankar   PetscMPIInt    size;
22273059b6faSBarry Smith   PetscInt       i,nz,header[4],*rowlengths=0,M,N,bs = newmat->rmap->bs;
22282f480046SShri Abhyankar   PetscInt       *mask,mbs,*jj,j,rowcount,nzcount,k,*s_browlengths,maskcount;
22292f480046SShri Abhyankar   PetscInt       kmax,jcount,block,idx,point,nzcountb,extra_rows,rows,cols;
22302f480046SShri Abhyankar   PetscInt       *masked,nmask,tmp,bs2,ishift;
22312f480046SShri Abhyankar   PetscScalar    *aa;
2232ce94432eSBarry Smith   MPI_Comm       comm;
22332f480046SShri Abhyankar 
22342f480046SShri Abhyankar   PetscFunctionBegin;
2235c98fd787SBarry Smith   /* force binary viewer to load .info file if it has not yet done so */
2236c98fd787SBarry Smith   ierr = PetscViewerSetUp(viewer);CHKERRQ(ierr);
2237ce94432eSBarry Smith   ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr);
2238c5929fdfSBarry Smith   ierr = PetscOptionsGetInt(((PetscObject)newmat)->options,((PetscObject)newmat)->prefix,"-matload_block_size",&bs,NULL);CHKERRQ(ierr);
22393059b6faSBarry Smith   if (bs < 0) bs = 1;
22402f480046SShri Abhyankar   bs2  = bs*bs;
22412f480046SShri Abhyankar 
22422f480046SShri Abhyankar   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
22432f480046SShri Abhyankar   if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"view must have one processor");
22442f480046SShri Abhyankar   ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr);
22452f480046SShri Abhyankar   ierr = PetscBinaryRead(fd,header,4,PETSC_INT);CHKERRQ(ierr);
22462f480046SShri Abhyankar   if (header[0] != MAT_FILE_CLASSID) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"not Mat object");
22472f480046SShri Abhyankar   M = header[1]; N = header[2]; nz = header[3];
22482f480046SShri Abhyankar 
22492f480046SShri Abhyankar   if (header[3] < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Matrix stored in special format, cannot load as SeqSBAIJ");
22502f480046SShri Abhyankar 
22512f480046SShri Abhyankar   if (M != N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Can only do square matrices");
22522f480046SShri Abhyankar 
22532f480046SShri Abhyankar   /*
22542f480046SShri Abhyankar      This code adds extra rows to make sure the number of rows is
22552f480046SShri Abhyankar     divisible by the blocksize
22562f480046SShri Abhyankar   */
22572f480046SShri Abhyankar   mbs        = M/bs;
22582f480046SShri Abhyankar   extra_rows = bs - M + bs*(mbs);
22592f480046SShri Abhyankar   if (extra_rows == bs) extra_rows = 0;
22602f480046SShri Abhyankar   else                  mbs++;
22612f480046SShri Abhyankar   if (extra_rows) {
22622f480046SShri Abhyankar     ierr = PetscInfo(viewer,"Padding loaded matrix to match blocksize\n");CHKERRQ(ierr);
22632f480046SShri Abhyankar   }
22642f480046SShri Abhyankar 
22652f480046SShri Abhyankar   /* Set global sizes if not already set */
22662f480046SShri Abhyankar   if (newmat->rmap->n < 0 && newmat->rmap->N < 0 && newmat->cmap->n < 0 && newmat->cmap->N < 0) {
22672f480046SShri Abhyankar     ierr = MatSetSizes(newmat,PETSC_DECIDE,PETSC_DECIDE,M+extra_rows,N+extra_rows);CHKERRQ(ierr);
22682f480046SShri Abhyankar   } else { /* Check if the matrix global sizes are correct */
22692f480046SShri Abhyankar     ierr = MatGetSize(newmat,&rows,&cols);CHKERRQ(ierr);
22702f480046SShri 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);
22712f480046SShri Abhyankar   }
22722f480046SShri Abhyankar 
22732f480046SShri Abhyankar   /* read in row lengths */
2274854ce69bSBarry Smith   ierr = PetscMalloc1(M+extra_rows,&rowlengths);CHKERRQ(ierr);
22752f480046SShri Abhyankar   ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr);
22762f480046SShri Abhyankar   for (i=0; i<extra_rows; i++) rowlengths[M+i] = 1;
22772f480046SShri Abhyankar 
22782f480046SShri Abhyankar   /* read in column indices */
2279854ce69bSBarry Smith   ierr = PetscMalloc1(nz+extra_rows,&jj);CHKERRQ(ierr);
22802f480046SShri Abhyankar   ierr = PetscBinaryRead(fd,jj,nz,PETSC_INT);CHKERRQ(ierr);
22812f480046SShri Abhyankar   for (i=0; i<extra_rows; i++) jj[nz+i] = M+i;
22822f480046SShri Abhyankar 
22832f480046SShri Abhyankar   /* loop over row lengths determining block row lengths */
22841795a4d1SJed Brown   ierr     = PetscCalloc1(mbs,&s_browlengths);CHKERRQ(ierr);
2285dcca6d9dSJed Brown   ierr     = PetscMalloc2(mbs,&mask,mbs,&masked);CHKERRQ(ierr);
22862f480046SShri Abhyankar   ierr     = PetscMemzero(mask,mbs*sizeof(PetscInt));CHKERRQ(ierr);
22872f480046SShri Abhyankar   rowcount = 0;
22882f480046SShri Abhyankar   nzcount  = 0;
22892f480046SShri Abhyankar   for (i=0; i<mbs; i++) {
22902f480046SShri Abhyankar     nmask = 0;
22912f480046SShri Abhyankar     for (j=0; j<bs; j++) {
22922f480046SShri Abhyankar       kmax = rowlengths[rowcount];
22932f480046SShri Abhyankar       for (k=0; k<kmax; k++) {
22942f480046SShri Abhyankar         tmp = jj[nzcount++]/bs;   /* block col. index */
22952f480046SShri Abhyankar         if (!mask[tmp] && tmp >= i) {masked[nmask++] = tmp; mask[tmp] = 1;}
22962f480046SShri Abhyankar       }
22972f480046SShri Abhyankar       rowcount++;
22982f480046SShri Abhyankar     }
22992f480046SShri Abhyankar     s_browlengths[i] += nmask;
23002f480046SShri Abhyankar 
23012f480046SShri Abhyankar     /* zero out the mask elements we set */
23022f480046SShri Abhyankar     for (j=0; j<nmask; j++) mask[masked[j]] = 0;
23032f480046SShri Abhyankar   }
23042f480046SShri Abhyankar 
23052f480046SShri Abhyankar   /* Do preallocation */
2306367daffbSBarry Smith   ierr = MatSeqSBAIJSetPreallocation(newmat,bs,0,s_browlengths);CHKERRQ(ierr);
23072f480046SShri Abhyankar   a    = (Mat_SeqSBAIJ*)newmat->data;
23082f480046SShri Abhyankar 
23092f480046SShri Abhyankar   /* set matrix "i" values */
23102f480046SShri Abhyankar   a->i[0] = 0;
23112f480046SShri Abhyankar   for (i=1; i<= mbs; i++) {
23122f480046SShri Abhyankar     a->i[i]      = a->i[i-1] + s_browlengths[i-1];
23132f480046SShri Abhyankar     a->ilen[i-1] = s_browlengths[i-1];
23142f480046SShri Abhyankar   }
23152f480046SShri Abhyankar   a->nz = a->i[mbs];
23162f480046SShri Abhyankar 
23172f480046SShri Abhyankar   /* read in nonzero values */
2318854ce69bSBarry Smith   ierr = PetscMalloc1(nz+extra_rows,&aa);CHKERRQ(ierr);
23192f480046SShri Abhyankar   ierr = PetscBinaryRead(fd,aa,nz,PETSC_SCALAR);CHKERRQ(ierr);
23202f480046SShri Abhyankar   for (i=0; i<extra_rows; i++) aa[nz+i] = 1.0;
23212f480046SShri Abhyankar 
23222f480046SShri Abhyankar   /* set "a" and "j" values into matrix */
23232f480046SShri Abhyankar   nzcount = 0; jcount = 0;
23242f480046SShri Abhyankar   for (i=0; i<mbs; i++) {
23252f480046SShri Abhyankar     nzcountb = nzcount;
23262f480046SShri Abhyankar     nmask    = 0;
23272f480046SShri Abhyankar     for (j=0; j<bs; j++) {
23282f480046SShri Abhyankar       kmax = rowlengths[i*bs+j];
23292f480046SShri Abhyankar       for (k=0; k<kmax; k++) {
23302f480046SShri Abhyankar         tmp = jj[nzcount++]/bs; /* block col. index */
23312f480046SShri Abhyankar         if (!mask[tmp] && tmp >= i) { masked[nmask++] = tmp; mask[tmp] = 1;}
23322f480046SShri Abhyankar       }
23332f480046SShri Abhyankar     }
23342f480046SShri Abhyankar     /* sort the masked values */
23352f480046SShri Abhyankar     ierr = PetscSortInt(nmask,masked);CHKERRQ(ierr);
23362f480046SShri Abhyankar 
23372f480046SShri Abhyankar     /* set "j" values into matrix */
23382f480046SShri Abhyankar     maskcount = 1;
23392f480046SShri Abhyankar     for (j=0; j<nmask; j++) {
23402f480046SShri Abhyankar       a->j[jcount++]  = masked[j];
23412f480046SShri Abhyankar       mask[masked[j]] = maskcount++;
23422f480046SShri Abhyankar     }
23432f480046SShri Abhyankar 
23442f480046SShri Abhyankar     /* set "a" values into matrix */
23452f480046SShri Abhyankar     ishift = bs2*a->i[i];
23462f480046SShri Abhyankar     for (j=0; j<bs; j++) {
23472f480046SShri Abhyankar       kmax = rowlengths[i*bs+j];
23482f480046SShri Abhyankar       for (k=0; k<kmax; k++) {
23492f480046SShri Abhyankar         tmp = jj[nzcountb]/bs;        /* block col. index */
23502f480046SShri Abhyankar         if (tmp >= i) {
23512f480046SShri Abhyankar           block     = mask[tmp] - 1;
23522f480046SShri Abhyankar           point     = jj[nzcountb] - bs*tmp;
23532f480046SShri Abhyankar           idx       = ishift + bs2*block + j + bs*point;
23542f480046SShri Abhyankar           a->a[idx] = aa[nzcountb];
23552f480046SShri Abhyankar         }
23562f480046SShri Abhyankar         nzcountb++;
23572f480046SShri Abhyankar       }
23582f480046SShri Abhyankar     }
23592f480046SShri Abhyankar     /* zero out the mask elements we set */
23602f480046SShri Abhyankar     for (j=0; j<nmask; j++) mask[masked[j]] = 0;
23612f480046SShri Abhyankar   }
23622f480046SShri Abhyankar   if (jcount != a->nz) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Bad binary matrix");
23632f480046SShri Abhyankar 
23642f480046SShri Abhyankar   ierr = PetscFree(rowlengths);CHKERRQ(ierr);
23652f480046SShri Abhyankar   ierr = PetscFree(s_browlengths);CHKERRQ(ierr);
23662f480046SShri Abhyankar   ierr = PetscFree(aa);CHKERRQ(ierr);
23672f480046SShri Abhyankar   ierr = PetscFree(jj);CHKERRQ(ierr);
23682f480046SShri Abhyankar   ierr = PetscFree2(mask,masked);CHKERRQ(ierr);
23692f480046SShri Abhyankar 
23702f480046SShri Abhyankar   ierr = MatAssemblyBegin(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
23712f480046SShri Abhyankar   ierr = MatAssemblyEnd(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
23722f480046SShri Abhyankar   PetscFunctionReturn(0);
23732f480046SShri Abhyankar }
23742f480046SShri Abhyankar 
2375c75a6043SHong Zhang /*@
2376c75a6043SHong Zhang      MatCreateSeqSBAIJWithArrays - Creates an sequential SBAIJ matrix using matrix elements
2377c75a6043SHong Zhang               (upper triangular entries in CSR format) provided by the user.
2378c75a6043SHong Zhang 
2379c75a6043SHong Zhang      Collective on MPI_Comm
2380c75a6043SHong Zhang 
2381c75a6043SHong Zhang    Input Parameters:
2382c75a6043SHong Zhang +  comm - must be an MPI communicator of size 1
2383c75a6043SHong Zhang .  bs - size of block
2384c75a6043SHong Zhang .  m - number of rows
2385c75a6043SHong Zhang .  n - number of columns
2386c75a6043SHong Zhang .  i - row indices
2387c75a6043SHong Zhang .  j - column indices
2388c75a6043SHong Zhang -  a - matrix values
2389c75a6043SHong Zhang 
2390c75a6043SHong Zhang    Output Parameter:
2391c75a6043SHong Zhang .  mat - the matrix
2392c75a6043SHong Zhang 
2393dfb205c3SBarry Smith    Level: advanced
2394c75a6043SHong Zhang 
2395c75a6043SHong Zhang    Notes:
2396c75a6043SHong Zhang        The i, j, and a arrays are not copied by this routine, the user must free these arrays
2397c75a6043SHong Zhang     once the matrix is destroyed
2398c75a6043SHong Zhang 
2399c75a6043SHong Zhang        You cannot set new nonzero locations into this matrix, that will generate an error.
2400c75a6043SHong Zhang 
2401c75a6043SHong Zhang        The i and j indices are 0 based
2402c75a6043SHong Zhang 
2403dfb205c3SBarry 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
2404dfb205c3SBarry Smith        it is the regular CSR format excluding the lower triangular elements.
2405dfb205c3SBarry Smith 
240669b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSBAIJ(), MatCreateSeqSBAIJ()
2407c75a6043SHong Zhang 
2408c75a6043SHong Zhang @*/
2409c3c607ccSBarry Smith PetscErrorCode  MatCreateSeqSBAIJWithArrays(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt i[],PetscInt j[],PetscScalar a[],Mat *mat)
2410c75a6043SHong Zhang {
2411c75a6043SHong Zhang   PetscErrorCode ierr;
2412c75a6043SHong Zhang   PetscInt       ii;
2413c75a6043SHong Zhang   Mat_SeqSBAIJ   *sbaij;
2414c75a6043SHong Zhang 
2415c75a6043SHong Zhang   PetscFunctionBegin;
2416e32f2f54SBarry Smith   if (bs != 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"block size %D > 1 is not supported yet",bs);
241741096f02SStefano Zampini   if (m > 0 && i[0]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"i (row indices) must start with 0");
2418c75a6043SHong Zhang 
2419c75a6043SHong Zhang   ierr  = MatCreate(comm,mat);CHKERRQ(ierr);
2420c75a6043SHong Zhang   ierr  = MatSetSizes(*mat,m,n,m,n);CHKERRQ(ierr);
2421c75a6043SHong Zhang   ierr  = MatSetType(*mat,MATSEQSBAIJ);CHKERRQ(ierr);
2422367daffbSBarry Smith   ierr  = MatSeqSBAIJSetPreallocation(*mat,bs,MAT_SKIP_ALLOCATION,0);CHKERRQ(ierr);
2423c75a6043SHong Zhang   sbaij = (Mat_SeqSBAIJ*)(*mat)->data;
2424dcca6d9dSJed Brown   ierr  = PetscMalloc2(m,&sbaij->imax,m,&sbaij->ilen);CHKERRQ(ierr);
24253bb1ff40SBarry Smith   ierr  = PetscLogObjectMemory((PetscObject)*mat,2*m*sizeof(PetscInt));CHKERRQ(ierr);
2426c75a6043SHong Zhang 
2427c75a6043SHong Zhang   sbaij->i = i;
2428c75a6043SHong Zhang   sbaij->j = j;
2429c75a6043SHong Zhang   sbaij->a = a;
243026fbe8dcSKarl Rupp 
2431c75a6043SHong Zhang   sbaij->singlemalloc = PETSC_FALSE;
2432c75a6043SHong Zhang   sbaij->nonew        = -1;             /*this indicates that inserting a new value in the matrix that generates a new nonzero is an error*/
2433e6b907acSBarry Smith   sbaij->free_a       = PETSC_FALSE;
2434e6b907acSBarry Smith   sbaij->free_ij      = PETSC_FALSE;
2435c75a6043SHong Zhang 
2436c75a6043SHong Zhang   for (ii=0; ii<m; ii++) {
2437c75a6043SHong Zhang     sbaij->ilen[ii] = sbaij->imax[ii] = i[ii+1] - i[ii];
2438c75a6043SHong Zhang #if defined(PETSC_USE_DEBUG)
2439e32f2f54SBarry 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]);
2440c75a6043SHong Zhang #endif
2441c75a6043SHong Zhang   }
2442c75a6043SHong Zhang #if defined(PETSC_USE_DEBUG)
2443c75a6043SHong Zhang   for (ii=0; ii<sbaij->i[m]; ii++) {
2444e32f2f54SBarry Smith     if (j[ii] < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column index at location = %d index = %d",ii,j[ii]);
2445e32f2f54SBarry 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]);
2446c75a6043SHong Zhang   }
2447c75a6043SHong Zhang #endif
2448c75a6043SHong Zhang 
2449c75a6043SHong Zhang   ierr = MatAssemblyBegin(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2450c75a6043SHong Zhang   ierr = MatAssemblyEnd(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2451c75a6043SHong Zhang   PetscFunctionReturn(0);
2452c75a6043SHong Zhang }
2453d06b337dSHong Zhang 
245459f5e6ceSHong Zhang PetscErrorCode MatCreateMPIMatConcatenateSeqMat_SeqSBAIJ(MPI_Comm comm,Mat inmat,PetscInt n,MatReuse scall,Mat *outmat)
245559f5e6ceSHong Zhang {
245659f5e6ceSHong Zhang   PetscErrorCode ierr;
24578761c3d6SHong Zhang   PetscMPIInt    size;
245859f5e6ceSHong Zhang 
245959f5e6ceSHong Zhang   PetscFunctionBegin;
24608761c3d6SHong Zhang   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
24618761c3d6SHong Zhang   if (size == 1 && scall == MAT_REUSE_MATRIX) {
24628761c3d6SHong Zhang     ierr = MatCopy(inmat,*outmat,SAME_NONZERO_PATTERN);CHKERRQ(ierr);
24638761c3d6SHong Zhang   } else {
246459f5e6ceSHong Zhang     ierr = MatCreateMPIMatConcatenateSeqMat_MPISBAIJ(comm,inmat,n,scall,outmat);CHKERRQ(ierr);
24658761c3d6SHong Zhang   }
246659f5e6ceSHong Zhang   PetscFunctionReturn(0);
246759f5e6ceSHong Zhang }
2468d06b337dSHong Zhang 
2469d06b337dSHong Zhang 
247049b5e25fSSatish Balay 
247149b5e25fSSatish Balay 
2472