xref: /petsc/src/mat/impls/sbaij/seq/sbaij.c (revision 7d3de750dec08ee2edc7d15bcef3046c0443ab7d)
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
17d24d4204SJose E. Roman #if defined(PETSC_HAVE_SCALAPACK)
18d24d4204SJose E. Roman PETSC_INTERN PetscErrorCode MatConvert_SBAIJ_ScaLAPACK(Mat,MatType,MatReuse,Mat*);
19d24d4204SJose E. Roman #endif
2028d58a37SPierre Jolivet PETSC_INTERN PetscErrorCode MatConvert_MPISBAIJ_Basic(Mat,MatType,MatReuse,Mat*);
21b5b17502SBarry Smith 
2249b5e25fSSatish Balay /*
2349b5e25fSSatish Balay      Checks for missing diagonals
2449b5e25fSSatish Balay */
25ace3abfcSBarry Smith PetscErrorCode MatMissingDiagonal_SeqSBAIJ(Mat A,PetscBool  *missing,PetscInt *dd)
2649b5e25fSSatish Balay {
27045c9aa0SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
286849ba73SBarry Smith   PetscErrorCode ierr;
297734d3b5SMatthew G. Knepley   PetscInt       *diag,*ii = a->i,i;
3049b5e25fSSatish Balay 
3149b5e25fSSatish Balay   PetscFunctionBegin;
32045c9aa0SHong Zhang   ierr     = MatMarkDiagonal_SeqSBAIJ(A);CHKERRQ(ierr);
332af78befSBarry Smith   *missing = PETSC_FALSE;
347734d3b5SMatthew G. Knepley   if (A->rmap->n > 0 && !ii) {
35358d2f5dSShri Abhyankar     *missing = PETSC_TRUE;
36358d2f5dSShri Abhyankar     if (dd) *dd = 0;
37955c1f14SBarry Smith     ierr = PetscInfo(A,"Matrix has no entries therefore is missing diagonal\n");CHKERRQ(ierr);
38358d2f5dSShri Abhyankar   } else {
39358d2f5dSShri Abhyankar     diag = a->diag;
4049b5e25fSSatish Balay     for (i=0; i<a->mbs; i++) {
417734d3b5SMatthew G. Knepley       if (diag[i] >= ii[i+1]) {
422af78befSBarry Smith         *missing = PETSC_TRUE;
432af78befSBarry Smith         if (dd) *dd = i;
442af78befSBarry Smith         break;
452af78befSBarry Smith       }
4649b5e25fSSatish Balay     }
47358d2f5dSShri Abhyankar   }
4849b5e25fSSatish Balay   PetscFunctionReturn(0);
4949b5e25fSSatish Balay }
5049b5e25fSSatish Balay 
51dfbe8321SBarry Smith PetscErrorCode MatMarkDiagonal_SeqSBAIJ(Mat A)
5249b5e25fSSatish Balay {
53045c9aa0SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
546849ba73SBarry Smith   PetscErrorCode ierr;
5548dd3d27SHong Zhang   PetscInt       i,j;
5649b5e25fSSatish Balay 
5749b5e25fSSatish Balay   PetscFunctionBegin;
5809f38230SBarry Smith   if (!a->diag) {
59785e854fSJed Brown     ierr         = PetscMalloc1(a->mbs,&a->diag);CHKERRQ(ierr);
603bb1ff40SBarry Smith     ierr         = PetscLogObjectMemory((PetscObject)A,a->mbs*sizeof(PetscInt));CHKERRQ(ierr);
61c760cd28SBarry Smith     a->free_diag = PETSC_TRUE;
6209f38230SBarry Smith   }
6348dd3d27SHong Zhang   for (i=0; i<a->mbs; i++) {
6448dd3d27SHong Zhang     a->diag[i] = a->i[i+1];
6548dd3d27SHong Zhang     for (j=a->i[i]; j<a->i[i+1]; j++) {
6648dd3d27SHong Zhang       if (a->j[j] == i) {
6748dd3d27SHong Zhang         a->diag[i] = j;
6848dd3d27SHong Zhang         break;
6948dd3d27SHong Zhang       }
7048dd3d27SHong Zhang     }
7148dd3d27SHong Zhang   }
7249b5e25fSSatish Balay   PetscFunctionReturn(0);
7349b5e25fSSatish Balay }
7449b5e25fSSatish Balay 
751a83f524SJed Brown static PetscErrorCode MatGetRowIJ_SeqSBAIJ(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *nn,const PetscInt *inia[],const PetscInt *inja[],PetscBool  *done)
7649b5e25fSSatish Balay {
77a6ece127SHong Zhang   Mat_SeqSBAIJ    *a = (Mat_SeqSBAIJ*)A->data;
788f7157efSSatish Balay   PetscErrorCode ierr;
792462f5fdSStefano Zampini   PetscInt       i,j,n = a->mbs,nz = a->i[n],*tia,*tja,bs = A->rmap->bs,k,l,cnt;
802462f5fdSStefano Zampini   PetscInt       **ia = (PetscInt**)inia,**ja = (PetscInt**)inja;
8149b5e25fSSatish Balay 
8249b5e25fSSatish Balay   PetscFunctionBegin;
83d3e5a4abSHong Zhang   *nn = n;
84a1373b80SHong Zhang   if (!ia) PetscFunctionReturn(0);
852462f5fdSStefano Zampini   if (symmetric) {
862462f5fdSStefano Zampini     ierr = MatToSymmetricIJ_SeqAIJ(n,a->i,a->j,PETSC_FALSE,0,0,&tia,&tja);CHKERRQ(ierr);
872462f5fdSStefano Zampini     nz   = tia[n];
882462f5fdSStefano Zampini   } else {
892462f5fdSStefano Zampini     tia = a->i; tja = a->j;
902462f5fdSStefano Zampini   }
912462f5fdSStefano Zampini 
922462f5fdSStefano Zampini   if (!blockcompressed && bs > 1) {
932462f5fdSStefano Zampini     (*nn) *= bs;
948f7157efSSatish Balay     /* malloc & create the natural set of indices */
952462f5fdSStefano Zampini     ierr = PetscMalloc1((n+1)*bs,ia);CHKERRQ(ierr);
962462f5fdSStefano Zampini     if (n) {
972462f5fdSStefano Zampini       (*ia)[0] = oshift;
982462f5fdSStefano Zampini       for (j=1; j<bs; j++) {
992462f5fdSStefano Zampini         (*ia)[j] = (tia[1]-tia[0])*bs+(*ia)[j-1];
1002462f5fdSStefano Zampini       }
1012462f5fdSStefano Zampini     }
1022462f5fdSStefano Zampini 
1032462f5fdSStefano Zampini     for (i=1; i<n; i++) {
1042462f5fdSStefano Zampini       (*ia)[i*bs] = (tia[i]-tia[i-1])*bs + (*ia)[i*bs-1];
1052462f5fdSStefano Zampini       for (j=1; j<bs; j++) {
1062462f5fdSStefano Zampini         (*ia)[i*bs+j] = (tia[i+1]-tia[i])*bs + (*ia)[i*bs+j-1];
1072462f5fdSStefano Zampini       }
1082462f5fdSStefano Zampini     }
1092462f5fdSStefano Zampini     if (n) {
1102462f5fdSStefano Zampini       (*ia)[n*bs] = (tia[n]-tia[n-1])*bs + (*ia)[n*bs-1];
1112462f5fdSStefano Zampini     }
1122462f5fdSStefano Zampini 
1132462f5fdSStefano Zampini     if (inja) {
1142462f5fdSStefano Zampini       ierr = PetscMalloc1(nz*bs*bs,ja);CHKERRQ(ierr);
1152462f5fdSStefano Zampini       cnt = 0;
1162462f5fdSStefano Zampini       for (i=0; i<n; i++) {
1178f7157efSSatish Balay         for (j=0; j<bs; j++) {
1182462f5fdSStefano Zampini           for (k=tia[i]; k<tia[i+1]; k++) {
1192462f5fdSStefano Zampini             for (l=0; l<bs; l++) {
1202462f5fdSStefano Zampini               (*ja)[cnt++] = bs*tja[k] + l;
1218f7157efSSatish Balay             }
1228f7157efSSatish Balay           }
1238f7157efSSatish Balay         }
1248f7157efSSatish Balay       }
1258f7157efSSatish Balay     }
1262462f5fdSStefano Zampini 
1272462f5fdSStefano Zampini     if (symmetric) { /* deallocate memory allocated in MatToSymmetricIJ_SeqAIJ() */
1282462f5fdSStefano Zampini       ierr = PetscFree(tia);CHKERRQ(ierr);
1292462f5fdSStefano Zampini       ierr = PetscFree(tja);CHKERRQ(ierr);
1302462f5fdSStefano Zampini     }
1312462f5fdSStefano Zampini   } else if (oshift == 1) {
1322462f5fdSStefano Zampini     if (symmetric) {
1332462f5fdSStefano Zampini       nz = tia[A->rmap->n/bs];
1342462f5fdSStefano Zampini       /*  add 1 to i and j indices */
1352462f5fdSStefano Zampini       for (i=0; i<A->rmap->n/bs+1; i++) tia[i] = tia[i] + 1;
1362462f5fdSStefano Zampini       *ia = tia;
1372462f5fdSStefano Zampini       if (ja) {
1382462f5fdSStefano Zampini         for (i=0; i<nz; i++) tja[i] = tja[i] + 1;
1392462f5fdSStefano Zampini         *ja = tja;
1402462f5fdSStefano Zampini       }
1412462f5fdSStefano Zampini     } else {
1422462f5fdSStefano Zampini       nz = a->i[A->rmap->n/bs];
1432462f5fdSStefano Zampini       /* malloc space and  add 1 to i and j indices */
1442462f5fdSStefano Zampini       ierr = PetscMalloc1(A->rmap->n/bs+1,ia);CHKERRQ(ierr);
1452462f5fdSStefano Zampini       for (i=0; i<A->rmap->n/bs+1; i++) (*ia)[i] = a->i[i] + 1;
1462462f5fdSStefano Zampini       if (ja) {
1472462f5fdSStefano Zampini         ierr = PetscMalloc1(nz,ja);CHKERRQ(ierr);
1482462f5fdSStefano Zampini         for (i=0; i<nz; i++) (*ja)[i] = a->j[i] + 1;
1492462f5fdSStefano Zampini       }
1502462f5fdSStefano Zampini     }
1512462f5fdSStefano Zampini   } else {
1522462f5fdSStefano Zampini     *ia = tia;
1532462f5fdSStefano Zampini     if (ja) *ja = tja;
154a6ece127SHong Zhang   }
15549b5e25fSSatish Balay   PetscFunctionReturn(0);
15649b5e25fSSatish Balay }
15749b5e25fSSatish Balay 
1581a83f524SJed Brown static PetscErrorCode MatRestoreRowIJ_SeqSBAIJ(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *nn,const PetscInt *ia[],const PetscInt *ja[],PetscBool  *done)
15949b5e25fSSatish Balay {
1608f7157efSSatish Balay   PetscErrorCode ierr;
161a6ece127SHong Zhang 
16249b5e25fSSatish Balay   PetscFunctionBegin;
16349b5e25fSSatish Balay   if (!ia) PetscFunctionReturn(0);
1642462f5fdSStefano Zampini   if ((!blockcompressed && A->rmap->bs > 1) || (symmetric || oshift == 1)) {
1652462f5fdSStefano Zampini     ierr = PetscFree(*ia);CHKERRQ(ierr);
1662462f5fdSStefano Zampini     if (ja) {ierr = PetscFree(*ja);CHKERRQ(ierr);}
167a6ece127SHong Zhang   }
168a6ece127SHong Zhang   PetscFunctionReturn(0);
16949b5e25fSSatish Balay }
17049b5e25fSSatish Balay 
171dfbe8321SBarry Smith PetscErrorCode MatDestroy_SeqSBAIJ(Mat A)
17249b5e25fSSatish Balay {
17349b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
174dfbe8321SBarry Smith   PetscErrorCode ierr;
17549b5e25fSSatish Balay 
17649b5e25fSSatish Balay   PetscFunctionBegin;
177a9f03627SSatish Balay #if defined(PETSC_USE_LOG)
178c0aa6a63SJacob Faibussowitsch   PetscLogObjectState((PetscObject)A,"Rows=%" PetscInt_FMT ", NZ=%" PetscInt_FMT,A->rmap->N,a->nz);
179a9f03627SSatish Balay #endif
180e6b907acSBarry Smith   ierr = MatSeqXAIJFreeAIJ(A,&a->a,&a->j,&a->i);CHKERRQ(ierr);
1817f53bb6cSHong Zhang   if (a->free_diag) {ierr = PetscFree(a->diag);CHKERRQ(ierr);}
1826bf464f9SBarry Smith   ierr = ISDestroy(&a->row);CHKERRQ(ierr);
1836bf464f9SBarry Smith   ierr = ISDestroy(&a->col);CHKERRQ(ierr);
1846bf464f9SBarry Smith   ierr = ISDestroy(&a->icol);CHKERRQ(ierr);
185c31cb41cSBarry Smith   ierr = PetscFree(a->idiag);CHKERRQ(ierr);
186c31cb41cSBarry Smith   ierr = PetscFree(a->inode.size);CHKERRQ(ierr);
187c760cd28SBarry Smith   if (a->free_imax_ilen) {ierr = PetscFree2(a->imax,a->ilen);CHKERRQ(ierr);}
18805b42c5fSBarry Smith   ierr = PetscFree(a->solve_work);CHKERRQ(ierr);
18941f059aeSBarry Smith   ierr = PetscFree(a->sor_work);CHKERRQ(ierr);
19005b42c5fSBarry Smith   ierr = PetscFree(a->solves_work);CHKERRQ(ierr);
19105b42c5fSBarry Smith   ierr = PetscFree(a->mult_work);CHKERRQ(ierr);
19205b42c5fSBarry Smith   ierr = PetscFree(a->saved_values);CHKERRQ(ierr);
1934da8f245SBarry Smith   if (a->free_jshort) {ierr = PetscFree(a->jshort);CHKERRQ(ierr);}
1941a3463dfSHong Zhang   ierr = PetscFree(a->inew);CHKERRQ(ierr);
1956bf464f9SBarry Smith   ierr = MatDestroy(&a->parent);CHKERRQ(ierr);
196bf0cc555SLisandro Dalcin   ierr = PetscFree(A->data);CHKERRQ(ierr);
197901853e0SKris Buschelman 
198f4259b30SLisandro Dalcin   ierr = PetscObjectChangeTypeName((PetscObject)A,NULL);CHKERRQ(ierr);
199bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatStoreValues_C",NULL);CHKERRQ(ierr);
200bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatRetrieveValues_C",NULL);CHKERRQ(ierr);
201bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetColumnIndices_C",NULL);CHKERRQ(ierr);
202bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqaij_C",NULL);CHKERRQ(ierr);
203bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqbaij_C",NULL);CHKERRQ(ierr);
204bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetPreallocation_C",NULL);CHKERRQ(ierr);
20538f409ebSLisandro Dalcin   ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetPreallocationCSR_C",NULL);CHKERRQ(ierr);
2066214f412SHong Zhang #if defined(PETSC_HAVE_ELEMENTAL)
2076214f412SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_elemental_C",NULL);CHKERRQ(ierr);
2086214f412SHong Zhang #endif
209d24d4204SJose E. Roman #if defined(PETSC_HAVE_SCALAPACK)
210d24d4204SJose E. Roman   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_scalapack_C",NULL);CHKERRQ(ierr);
211d24d4204SJose E. Roman #endif
21249b5e25fSSatish Balay   PetscFunctionReturn(0);
21349b5e25fSSatish Balay }
21449b5e25fSSatish Balay 
215ace3abfcSBarry Smith PetscErrorCode MatSetOption_SeqSBAIJ(Mat A,MatOption op,PetscBool flg)
21649b5e25fSSatish Balay {
217045c9aa0SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
218eb1ec7c1SStefano Zampini #if defined(PETSC_USE_COMPLEX)
219eb1ec7c1SStefano Zampini   PetscInt       bs;
220eb1ec7c1SStefano Zampini #endif
22163ba0a88SBarry Smith   PetscErrorCode ierr;
22249b5e25fSSatish Balay 
22349b5e25fSSatish Balay   PetscFunctionBegin;
224eb1ec7c1SStefano Zampini #if defined(PETSC_USE_COMPLEX)
225eb1ec7c1SStefano Zampini   ierr = MatGetBlockSize(A,&bs);CHKERRQ(ierr);
226eb1ec7c1SStefano Zampini #endif
2274d9d31abSKris Buschelman   switch (op) {
2284d9d31abSKris Buschelman   case MAT_ROW_ORIENTED:
2294e0d8c25SBarry Smith     a->roworiented = flg;
2304d9d31abSKris Buschelman     break;
231a9817697SBarry Smith   case MAT_KEEP_NONZERO_PATTERN:
232a9817697SBarry Smith     a->keepnonzeropattern = flg;
2334d9d31abSKris Buschelman     break;
234512a5fc5SBarry Smith   case MAT_NEW_NONZERO_LOCATIONS:
235512a5fc5SBarry Smith     a->nonew = (flg ? 0 : 1);
2364d9d31abSKris Buschelman     break;
2374d9d31abSKris Buschelman   case MAT_NEW_NONZERO_LOCATION_ERR:
2384e0d8c25SBarry Smith     a->nonew = (flg ? -1 : 0);
2394d9d31abSKris Buschelman     break;
2404d9d31abSKris Buschelman   case MAT_NEW_NONZERO_ALLOCATION_ERR:
2414e0d8c25SBarry Smith     a->nonew = (flg ? -2 : 0);
2424d9d31abSKris Buschelman     break;
24328b2fa4aSMatthew Knepley   case MAT_UNUSED_NONZERO_LOCATION_ERR:
24428b2fa4aSMatthew Knepley     a->nounused = (flg ? -1 : 0);
24528b2fa4aSMatthew Knepley     break;
2468c78258cSHong Zhang   case MAT_FORCE_DIAGONAL_ENTRIES:
2474d9d31abSKris Buschelman   case MAT_IGNORE_OFF_PROC_ENTRIES:
2484d9d31abSKris Buschelman   case MAT_USE_HASH_TABLE:
249071fcb05SBarry Smith   case MAT_SORTED_FULL:
250*7d3de750SJacob Faibussowitsch     ierr = PetscInfo(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr);
2514d9d31abSKris Buschelman     break;
2529a4540c5SBarry Smith   case MAT_HERMITIAN:
253eb1ec7c1SStefano Zampini #if defined(PETSC_USE_COMPLEX)
254eb1ec7c1SStefano Zampini     if (flg) { /* disable transpose ops */
255eb1ec7c1SStefano Zampini       if (bs > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No support for Hermitian with block size greater than 1");
256eb1ec7c1SStefano Zampini       A->ops->multtranspose    = NULL;
257eb1ec7c1SStefano Zampini       A->ops->multtransposeadd = NULL;
258eb1ec7c1SStefano Zampini       A->symmetric             = PETSC_FALSE;
259eb1ec7c1SStefano Zampini     }
2600f2140c7SStefano Zampini #endif
261eeffb40dSHong Zhang     break;
26277e54ba9SKris Buschelman   case MAT_SYMMETRIC:
263eb1ec7c1SStefano Zampini   case MAT_SPD:
264eb1ec7c1SStefano Zampini #if defined(PETSC_USE_COMPLEX)
265eb1ec7c1SStefano Zampini     if (flg) { /* An hermitian and symmetric matrix has zero imaginary part (restore back transpose ops) */
266eb1ec7c1SStefano Zampini       A->ops->multtranspose    = A->ops->mult;
267eb1ec7c1SStefano Zampini       A->ops->multtransposeadd = A->ops->multadd;
268eb1ec7c1SStefano Zampini     }
269eb1ec7c1SStefano Zampini #endif
270eb1ec7c1SStefano Zampini     break;
271eb1ec7c1SStefano Zampini     /* These options are handled directly by MatSetOption() */
27277e54ba9SKris Buschelman   case MAT_STRUCTURALLY_SYMMETRIC:
2739a4540c5SBarry Smith   case MAT_SYMMETRY_ETERNAL:
274672ba085SHong Zhang   case MAT_STRUCTURE_ONLY:
2754dcd73b1SHong Zhang     /* These options are handled directly by MatSetOption() */
276290bbb0aSBarry Smith     break;
277941593c8SHong Zhang   case MAT_IGNORE_LOWER_TRIANGULAR:
2784e0d8c25SBarry Smith     a->ignore_ltriangular = flg;
279941593c8SHong Zhang     break;
280941593c8SHong Zhang   case MAT_ERROR_LOWER_TRIANGULAR:
2814e0d8c25SBarry Smith     a->ignore_ltriangular = flg;
28277e54ba9SKris Buschelman     break;
283f5edf698SHong Zhang   case MAT_GETROW_UPPERTRIANGULAR:
2844e0d8c25SBarry Smith     a->getrow_utriangular = flg;
285f5edf698SHong Zhang     break;
286c10200c1SHong Zhang   case MAT_SUBMAT_SINGLEIS:
287c10200c1SHong Zhang     break;
2884d9d31abSKris Buschelman   default:
28998921bdaSJacob Faibussowitsch     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"unknown option %d",op);
29049b5e25fSSatish Balay   }
29149b5e25fSSatish Balay   PetscFunctionReturn(0);
29249b5e25fSSatish Balay }
29349b5e25fSSatish Balay 
29452768537SHong Zhang PetscErrorCode MatGetRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
29549b5e25fSSatish Balay {
29649b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
2976849ba73SBarry Smith   PetscErrorCode ierr;
29849b5e25fSSatish Balay 
29949b5e25fSSatish Balay   PetscFunctionBegin;
300e32f2f54SBarry 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()");
30152768537SHong Zhang 
302f5edf698SHong Zhang   /* Get the upper triangular part of the row */
30352768537SHong Zhang   ierr = MatGetRow_SeqBAIJ_private(A,row,nz,idx,v,a->i,a->j,a->a);CHKERRQ(ierr);
30449b5e25fSSatish Balay   PetscFunctionReturn(0);
30549b5e25fSSatish Balay }
30649b5e25fSSatish Balay 
30713f74950SBarry Smith PetscErrorCode MatRestoreRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
30849b5e25fSSatish Balay {
309dfbe8321SBarry Smith   PetscErrorCode ierr;
31049b5e25fSSatish Balay 
31149b5e25fSSatish Balay   PetscFunctionBegin;
312cb4a9cd9SHong Zhang   if (nz)  *nz = 0;
31305b42c5fSBarry Smith   if (idx) {ierr = PetscFree(*idx);CHKERRQ(ierr);}
31405b42c5fSBarry Smith   if (v)   {ierr = PetscFree(*v);CHKERRQ(ierr);}
31549b5e25fSSatish Balay   PetscFunctionReturn(0);
31649b5e25fSSatish Balay }
31749b5e25fSSatish Balay 
318f5edf698SHong Zhang PetscErrorCode MatGetRowUpperTriangular_SeqSBAIJ(Mat A)
319f5edf698SHong Zhang {
320f5edf698SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
321f5edf698SHong Zhang 
322f5edf698SHong Zhang   PetscFunctionBegin;
323f5edf698SHong Zhang   a->getrow_utriangular = PETSC_TRUE;
324f5edf698SHong Zhang   PetscFunctionReturn(0);
325f5edf698SHong Zhang }
326a323099bSStefano Zampini 
327f5edf698SHong Zhang PetscErrorCode MatRestoreRowUpperTriangular_SeqSBAIJ(Mat A)
328f5edf698SHong Zhang {
329f5edf698SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
330f5edf698SHong Zhang 
331f5edf698SHong Zhang   PetscFunctionBegin;
332f5edf698SHong Zhang   a->getrow_utriangular = PETSC_FALSE;
333f5edf698SHong Zhang   PetscFunctionReturn(0);
334f5edf698SHong Zhang }
335f5edf698SHong Zhang 
336fc4dec0aSBarry Smith PetscErrorCode MatTranspose_SeqSBAIJ(Mat A,MatReuse reuse,Mat *B)
33749b5e25fSSatish Balay {
338dfbe8321SBarry Smith   PetscErrorCode ierr;
3395fd66863SKarl Rupp 
34049b5e25fSSatish Balay   PetscFunctionBegin;
341cf37664fSBarry Smith   if (reuse == MAT_INITIAL_MATRIX) {
342999d9058SBarry Smith     ierr = MatDuplicate(A,MAT_COPY_VALUES,B);CHKERRQ(ierr);
343cf37664fSBarry Smith   } else if (reuse == MAT_REUSE_MATRIX) {
344cf37664fSBarry Smith     ierr = MatCopy(A,*B,SAME_NONZERO_PATTERN);CHKERRQ(ierr);
345fc4dec0aSBarry Smith   }
3468115998fSBarry Smith   PetscFunctionReturn(0);
34749b5e25fSSatish Balay }
34849b5e25fSSatish Balay 
3497da1fb6eSBarry Smith PetscErrorCode MatView_SeqSBAIJ_ASCII(Mat A,PetscViewer viewer)
35049b5e25fSSatish Balay {
35149b5e25fSSatish Balay   Mat_SeqSBAIJ      *a = (Mat_SeqSBAIJ*)A->data;
352dfbe8321SBarry Smith   PetscErrorCode    ierr;
353d0f46423SBarry Smith   PetscInt          i,j,bs = A->rmap->bs,k,l,bs2=a->bs2;
354f3ef73ceSBarry Smith   PetscViewerFormat format;
355121deb67SSatish Balay   PetscInt          *diag;
35649b5e25fSSatish Balay 
35749b5e25fSSatish Balay   PetscFunctionBegin;
358b0a32e0cSBarry Smith   ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
359456192e2SBarry Smith   if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
360c0aa6a63SJacob Faibussowitsch     ierr = PetscViewerASCIIPrintf(viewer,"  block size is %" PetscInt_FMT "\n",bs);CHKERRQ(ierr);
361fb9695e5SSatish Balay   } else if (format == PETSC_VIEWER_ASCII_MATLAB) {
362d2507d54SMatthew Knepley     Mat        aij;
363ade3a672SBarry Smith     const char *matname;
364ade3a672SBarry Smith 
365d5f3da31SBarry Smith     if (A->factortype && bs>1) {
36670d5e725SHong 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);
36770d5e725SHong Zhang       PetscFunctionReturn(0);
36870d5e725SHong Zhang     }
369c9f458caSMatthew Knepley     ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&aij);CHKERRQ(ierr);
370ade3a672SBarry Smith     ierr = PetscObjectGetName((PetscObject)A,&matname);CHKERRQ(ierr);
371ade3a672SBarry Smith     ierr = PetscObjectSetName((PetscObject)aij,matname);CHKERRQ(ierr);
372c9f458caSMatthew Knepley     ierr = MatView(aij,viewer);CHKERRQ(ierr);
3736bf464f9SBarry Smith     ierr = MatDestroy(&aij);CHKERRQ(ierr);
374fb9695e5SSatish Balay   } else if (format == PETSC_VIEWER_ASCII_COMMON) {
375d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
37649b5e25fSSatish Balay     for (i=0; i<a->mbs; i++) {
37749b5e25fSSatish Balay       for (j=0; j<bs; j++) {
378c0aa6a63SJacob Faibussowitsch         ierr = PetscViewerASCIIPrintf(viewer,"row %" PetscInt_FMT ":",i*bs+j);CHKERRQ(ierr);
37949b5e25fSSatish Balay         for (k=a->i[i]; k<a->i[i+1]; k++) {
38049b5e25fSSatish Balay           for (l=0; l<bs; l++) {
38149b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX)
38249b5e25fSSatish Balay             if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
383c0aa6a63SJacob Faibussowitsch               ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g + %g i) ",bs*a->j[k]+l,
38457622a8eSBarry Smith                                             (double)PetscRealPart(a->a[bs2*k + l*bs + j]),(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
38549b5e25fSSatish Balay             } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
386c0aa6a63SJacob Faibussowitsch               ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g - %g i) ",bs*a->j[k]+l,
38757622a8eSBarry Smith                                             (double)PetscRealPart(a->a[bs2*k + l*bs + j]),-(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
38849b5e25fSSatish Balay             } else if (PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
389c0aa6a63SJacob Faibussowitsch               ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",bs*a->j[k]+l,(double)PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
39049b5e25fSSatish Balay             }
39149b5e25fSSatish Balay #else
39249b5e25fSSatish Balay             if (a->a[bs2*k + l*bs + j] != 0.0) {
393c0aa6a63SJacob Faibussowitsch               ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",bs*a->j[k]+l,(double)a->a[bs2*k + l*bs + j]);CHKERRQ(ierr);
39449b5e25fSSatish Balay             }
39549b5e25fSSatish Balay #endif
39649b5e25fSSatish Balay           }
39749b5e25fSSatish Balay         }
398b0a32e0cSBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
39949b5e25fSSatish Balay       }
40049b5e25fSSatish Balay     }
401d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
402c1490034SHong Zhang   } else if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) {
403c1490034SHong Zhang     PetscFunctionReturn(0);
40449b5e25fSSatish Balay   } else {
405d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
4062c990fa1SHong Zhang     if (A->factortype) { /* for factored matrix */
4072c990fa1SHong Zhang       if (bs>1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"matrix is factored with bs>1. Not implemented yet");
4082c990fa1SHong Zhang 
409121deb67SSatish Balay       diag=a->diag;
410121deb67SSatish Balay       for (i=0; i<a->mbs; i++) { /* for row block i */
411c0aa6a63SJacob Faibussowitsch         ierr = PetscViewerASCIIPrintf(viewer,"row %" PetscInt_FMT ":",i);CHKERRQ(ierr);
4122c990fa1SHong Zhang         /* diagonal entry */
4132c990fa1SHong Zhang #if defined(PETSC_USE_COMPLEX)
4142c990fa1SHong Zhang         if (PetscImaginaryPart(a->a[diag[i]]) > 0.0) {
415c0aa6a63SJacob Faibussowitsch           ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %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);
4162c990fa1SHong Zhang         } else if (PetscImaginaryPart(a->a[diag[i]]) < 0.0) {
417c0aa6a63SJacob Faibussowitsch           ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %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);
4182c990fa1SHong Zhang         } else {
419c0aa6a63SJacob Faibussowitsch           ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->j[diag[i]],(double)PetscRealPart(1.0/a->a[diag[i]]));CHKERRQ(ierr);
4202c990fa1SHong Zhang         }
4212c990fa1SHong Zhang #else
422c0aa6a63SJacob Faibussowitsch         ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->j[diag[i]],(double)(1.0/a->a[diag[i]]));CHKERRQ(ierr);
4232c990fa1SHong Zhang #endif
4242c990fa1SHong Zhang         /* off-diagonal entries */
4252c990fa1SHong Zhang         for (k=a->i[i]; k<a->i[i+1]-1; k++) {
4262c990fa1SHong Zhang #if defined(PETSC_USE_COMPLEX)
427ca0704adSBarry Smith           if (PetscImaginaryPart(a->a[k]) > 0.0) {
428c0aa6a63SJacob Faibussowitsch             ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g + %g i) ",bs*a->j[k],(double)PetscRealPart(a->a[k]),(double)PetscImaginaryPart(a->a[k]));CHKERRQ(ierr);
429ca0704adSBarry Smith           } else if (PetscImaginaryPart(a->a[k]) < 0.0) {
430c0aa6a63SJacob Faibussowitsch             ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g - %g i) ",bs*a->j[k],(double)PetscRealPart(a->a[k]),-(double)PetscImaginaryPart(a->a[k]));CHKERRQ(ierr);
4312c990fa1SHong Zhang           } else {
432c0aa6a63SJacob Faibussowitsch             ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",bs*a->j[k],(double)PetscRealPart(a->a[k]));CHKERRQ(ierr);
4332c990fa1SHong Zhang           }
4342c990fa1SHong Zhang #else
435c0aa6a63SJacob Faibussowitsch           ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",a->j[k],(double)a->a[k]);CHKERRQ(ierr);
4362c990fa1SHong Zhang #endif
4372c990fa1SHong Zhang         }
4382c990fa1SHong Zhang         ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
4392c990fa1SHong Zhang       }
4402c990fa1SHong Zhang 
4412c990fa1SHong Zhang     } else { /* for non-factored matrix */
4420c74a584SJed Brown       for (i=0; i<a->mbs; i++) { /* for row block i */
4430c74a584SJed Brown         for (j=0; j<bs; j++) {   /* for row bs*i + j */
444c0aa6a63SJacob Faibussowitsch           ierr = PetscViewerASCIIPrintf(viewer,"row %" PetscInt_FMT ":",i*bs+j);CHKERRQ(ierr);
4450c74a584SJed Brown           for (k=a->i[i]; k<a->i[i+1]; k++) { /* for column block */
4460c74a584SJed Brown             for (l=0; l<bs; l++) {            /* for column */
44749b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX)
44849b5e25fSSatish Balay               if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0) {
449c0aa6a63SJacob Faibussowitsch                 ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g + %g i) ",bs*a->j[k]+l,
45057622a8eSBarry Smith                                               (double)PetscRealPart(a->a[bs2*k + l*bs + j]),(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
45149b5e25fSSatish Balay               } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0) {
452c0aa6a63SJacob Faibussowitsch                 ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g - %g i) ",bs*a->j[k]+l,
45357622a8eSBarry Smith                                               (double)PetscRealPart(a->a[bs2*k + l*bs + j]),-(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
45449b5e25fSSatish Balay               } else {
455c0aa6a63SJacob Faibussowitsch                 ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",bs*a->j[k]+l,(double)PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
45649b5e25fSSatish Balay               }
45749b5e25fSSatish Balay #else
458c0aa6a63SJacob Faibussowitsch               ierr = PetscViewerASCIIPrintf(viewer," (%" PetscInt_FMT ", %g) ",bs*a->j[k]+l,(double)a->a[bs2*k + l*bs + j]);CHKERRQ(ierr);
45949b5e25fSSatish Balay #endif
46049b5e25fSSatish Balay             }
46149b5e25fSSatish Balay           }
462b0a32e0cSBarry Smith           ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
46349b5e25fSSatish Balay         }
46449b5e25fSSatish Balay       }
4652c990fa1SHong Zhang     }
466d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
46749b5e25fSSatish Balay   }
468b0a32e0cSBarry Smith   ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
46949b5e25fSSatish Balay   PetscFunctionReturn(0);
47049b5e25fSSatish Balay }
47149b5e25fSSatish Balay 
4729804daf3SBarry Smith #include <petscdraw.h>
4736849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw_Zoom(PetscDraw draw,void *Aa)
47449b5e25fSSatish Balay {
47549b5e25fSSatish Balay   Mat            A = (Mat) Aa;
47649b5e25fSSatish Balay   Mat_SeqSBAIJ   *a=(Mat_SeqSBAIJ*)A->data;
4776849ba73SBarry Smith   PetscErrorCode ierr;
478d0f46423SBarry Smith   PetscInt       row,i,j,k,l,mbs=a->mbs,color,bs=A->rmap->bs,bs2=a->bs2;
47949b5e25fSSatish Balay   PetscReal      xl,yl,xr,yr,x_l,x_r,y_l,y_r;
48049b5e25fSSatish Balay   MatScalar      *aa;
481b0a32e0cSBarry Smith   PetscViewer    viewer;
48249b5e25fSSatish Balay 
48349b5e25fSSatish Balay   PetscFunctionBegin;
48449b5e25fSSatish Balay   ierr = PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer);CHKERRQ(ierr);
485b0a32e0cSBarry Smith   ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr);
48649b5e25fSSatish Balay 
48749b5e25fSSatish Balay   /* loop over matrix elements drawing boxes */
488383922c3SLisandro Dalcin 
489383922c3SLisandro Dalcin   ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr);
490383922c3SLisandro Dalcin   ierr = PetscDrawString(draw, .3*(xl+xr), .3*(yl+yr), PETSC_DRAW_BLACK, "symmetric");CHKERRQ(ierr);
491383922c3SLisandro Dalcin   /* Blue for negative, Cyan for zero and  Red for positive */
492b0a32e0cSBarry Smith   color = PETSC_DRAW_BLUE;
49349b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
49449b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
495d0f46423SBarry Smith       y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0;
49649b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
49749b5e25fSSatish Balay       aa  = a->a + j*bs2;
49849b5e25fSSatish Balay       for (k=0; k<bs; k++) {
49949b5e25fSSatish Balay         for (l=0; l<bs; l++) {
50049b5e25fSSatish Balay           if (PetscRealPart(*aa++) >=  0.) continue;
501b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
50249b5e25fSSatish Balay         }
50349b5e25fSSatish Balay       }
50449b5e25fSSatish Balay     }
50549b5e25fSSatish Balay   }
506b0a32e0cSBarry Smith   color = PETSC_DRAW_CYAN;
50749b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
50849b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
509d0f46423SBarry Smith       y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0;
51049b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
51149b5e25fSSatish Balay       aa = a->a + j*bs2;
51249b5e25fSSatish Balay       for (k=0; k<bs; k++) {
51349b5e25fSSatish Balay         for (l=0; l<bs; l++) {
51449b5e25fSSatish Balay           if (PetscRealPart(*aa++) != 0.) continue;
515b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
51649b5e25fSSatish Balay         }
51749b5e25fSSatish Balay       }
51849b5e25fSSatish Balay     }
51949b5e25fSSatish Balay   }
520b0a32e0cSBarry Smith   color = PETSC_DRAW_RED;
52149b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
52249b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
523d0f46423SBarry Smith       y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0;
52449b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
52549b5e25fSSatish Balay       aa = a->a + j*bs2;
52649b5e25fSSatish Balay       for (k=0; k<bs; k++) {
52749b5e25fSSatish Balay         for (l=0; l<bs; l++) {
52849b5e25fSSatish Balay           if (PetscRealPart(*aa++) <= 0.) continue;
529b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
53049b5e25fSSatish Balay         }
53149b5e25fSSatish Balay       }
53249b5e25fSSatish Balay     }
53349b5e25fSSatish Balay   }
534383922c3SLisandro Dalcin   ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr);
53549b5e25fSSatish Balay   PetscFunctionReturn(0);
53649b5e25fSSatish Balay }
53749b5e25fSSatish Balay 
5386849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw(Mat A,PetscViewer viewer)
53949b5e25fSSatish Balay {
540dfbe8321SBarry Smith   PetscErrorCode ierr;
54149b5e25fSSatish Balay   PetscReal      xl,yl,xr,yr,w,h;
542b0a32e0cSBarry Smith   PetscDraw      draw;
543ace3abfcSBarry Smith   PetscBool      isnull;
54449b5e25fSSatish Balay 
54549b5e25fSSatish Balay   PetscFunctionBegin;
546b0a32e0cSBarry Smith   ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
547383922c3SLisandro Dalcin   ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr);
548383922c3SLisandro Dalcin   if (isnull) PetscFunctionReturn(0);
54949b5e25fSSatish Balay 
550d0f46423SBarry Smith   xr   = A->rmap->N; yr = A->rmap->N; h = yr/10.0; w = xr/10.0;
55149b5e25fSSatish Balay   xr  += w;          yr += h;        xl = -w;     yl = -h;
552b0a32e0cSBarry Smith   ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr);
553832b7cebSLisandro Dalcin   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer);CHKERRQ(ierr);
554b0a32e0cSBarry Smith   ierr = PetscDrawZoom(draw,MatView_SeqSBAIJ_Draw_Zoom,A);CHKERRQ(ierr);
5550298fd71SBarry Smith   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",NULL);CHKERRQ(ierr);
556832b7cebSLisandro Dalcin   ierr = PetscDrawSave(draw);CHKERRQ(ierr);
55749b5e25fSSatish Balay   PetscFunctionReturn(0);
55849b5e25fSSatish Balay }
55949b5e25fSSatish Balay 
560618cc2edSLisandro Dalcin /* Used for both MPIBAIJ and MPISBAIJ matrices */
561618cc2edSLisandro Dalcin #define MatView_SeqSBAIJ_Binary MatView_SeqBAIJ_Binary
562618cc2edSLisandro Dalcin 
563dfbe8321SBarry Smith PetscErrorCode MatView_SeqSBAIJ(Mat A,PetscViewer viewer)
56449b5e25fSSatish Balay {
565dfbe8321SBarry Smith   PetscErrorCode ierr;
566618cc2edSLisandro Dalcin   PetscBool      iascii,isbinary,isdraw;
56749b5e25fSSatish Balay 
56849b5e25fSSatish Balay   PetscFunctionBegin;
569251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
570618cc2edSLisandro Dalcin   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr);
571251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr);
57232077d6dSBarry Smith   if (iascii) {
57349b5e25fSSatish Balay     ierr = MatView_SeqSBAIJ_ASCII(A,viewer);CHKERRQ(ierr);
574618cc2edSLisandro Dalcin   } else if (isbinary) {
575618cc2edSLisandro Dalcin     ierr = MatView_SeqSBAIJ_Binary(A,viewer);CHKERRQ(ierr);
57649b5e25fSSatish Balay   } else if (isdraw) {
57749b5e25fSSatish Balay     ierr = MatView_SeqSBAIJ_Draw(A,viewer);CHKERRQ(ierr);
57849b5e25fSSatish Balay   } else {
579a5e6ed63SBarry Smith     Mat        B;
580ade3a672SBarry Smith     const char *matname;
581ceb03754SKris Buschelman     ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&B);CHKERRQ(ierr);
582ade3a672SBarry Smith     ierr = PetscObjectGetName((PetscObject)A,&matname);CHKERRQ(ierr);
583ade3a672SBarry Smith     ierr = PetscObjectSetName((PetscObject)B,matname);CHKERRQ(ierr);
584a5e6ed63SBarry Smith     ierr = MatView(B,viewer);CHKERRQ(ierr);
5856bf464f9SBarry Smith     ierr = MatDestroy(&B);CHKERRQ(ierr);
58649b5e25fSSatish Balay   }
58749b5e25fSSatish Balay   PetscFunctionReturn(0);
58849b5e25fSSatish Balay }
58949b5e25fSSatish Balay 
59013f74950SBarry Smith PetscErrorCode MatGetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],PetscScalar v[])
59149b5e25fSSatish Balay {
592045c9aa0SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
59313f74950SBarry Smith   PetscInt     *rp,k,low,high,t,row,nrow,i,col,l,*aj = a->j;
59413f74950SBarry Smith   PetscInt     *ai = a->i,*ailen = a->ilen;
595d0f46423SBarry Smith   PetscInt     brow,bcol,ridx,cidx,bs=A->rmap->bs,bs2=a->bs2;
59697e567efSBarry Smith   MatScalar    *ap,*aa = a->a;
59749b5e25fSSatish Balay 
59849b5e25fSSatish Balay   PetscFunctionBegin;
59949b5e25fSSatish Balay   for (k=0; k<m; k++) { /* loop over rows */
60049b5e25fSSatish Balay     row = im[k]; brow = row/bs;
60198921bdaSJacob Faibussowitsch     if (row < 0) {v += n; continue;} /* SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative row: %" PetscInt_FMT,row); */
60298921bdaSJacob Faibussowitsch     if (row >= A->rmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %" PetscInt_FMT " max %" PetscInt_FMT,row,A->rmap->N-1);
60349b5e25fSSatish Balay     rp   = aj + ai[brow]; ap = aa + bs2*ai[brow];
60449b5e25fSSatish Balay     nrow = ailen[brow];
60549b5e25fSSatish Balay     for (l=0; l<n; l++) { /* loop over columns */
60698921bdaSJacob Faibussowitsch       if (in[l] < 0) {v++; continue;} /* SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column: %" PetscInt_FMT,in[l]); */
60798921bdaSJacob Faibussowitsch       if (in[l] >= A->cmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %" PetscInt_FMT " max %" PetscInt_FMT,in[l],A->cmap->n-1);
60849b5e25fSSatish Balay       col  = in[l];
60949b5e25fSSatish Balay       bcol = col/bs;
61049b5e25fSSatish Balay       cidx = col%bs;
61149b5e25fSSatish Balay       ridx = row%bs;
61249b5e25fSSatish Balay       high = nrow;
61349b5e25fSSatish Balay       low  = 0; /* assume unsorted */
61449b5e25fSSatish Balay       while (high-low > 5) {
61549b5e25fSSatish Balay         t = (low+high)/2;
61649b5e25fSSatish Balay         if (rp[t] > bcol) high = t;
61749b5e25fSSatish Balay         else              low  = t;
61849b5e25fSSatish Balay       }
61949b5e25fSSatish Balay       for (i=low; i<high; i++) {
62049b5e25fSSatish Balay         if (rp[i] > bcol) break;
62149b5e25fSSatish Balay         if (rp[i] == bcol) {
62249b5e25fSSatish Balay           *v++ = ap[bs2*i+bs*cidx+ridx];
62349b5e25fSSatish Balay           goto finished;
62449b5e25fSSatish Balay         }
62549b5e25fSSatish Balay       }
62697e567efSBarry Smith       *v++ = 0.0;
62749b5e25fSSatish Balay finished:;
62849b5e25fSSatish Balay     }
62949b5e25fSSatish Balay   }
63049b5e25fSSatish Balay   PetscFunctionReturn(0);
63149b5e25fSSatish Balay }
63249b5e25fSSatish Balay 
633dc29a518SPierre Jolivet PetscErrorCode MatPermute_SeqSBAIJ(Mat A,IS rowp,IS colp,Mat *B)
634dc29a518SPierre Jolivet {
635dc29a518SPierre Jolivet   Mat            C;
636dc29a518SPierre Jolivet   PetscErrorCode ierr;
637dc29a518SPierre Jolivet 
638dc29a518SPierre Jolivet   PetscFunctionBegin;
639dc29a518SPierre Jolivet   ierr = MatConvert(A,MATSEQBAIJ,MAT_INITIAL_MATRIX,&C);CHKERRQ(ierr);
640dc29a518SPierre Jolivet   ierr = MatPermute(C,rowp,colp,B);CHKERRQ(ierr);
641dc29a518SPierre Jolivet   ierr = MatDestroy(&C);CHKERRQ(ierr);
642dc29a518SPierre Jolivet   if (rowp == colp) {
643dc29a518SPierre Jolivet     ierr = MatConvert(*B,MATSEQSBAIJ,MAT_INPLACE_MATRIX,B);CHKERRQ(ierr);
644dc29a518SPierre Jolivet   }
645dc29a518SPierre Jolivet   PetscFunctionReturn(0);
646dc29a518SPierre Jolivet }
64749b5e25fSSatish Balay 
64813f74950SBarry Smith PetscErrorCode MatSetValuesBlocked_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is)
64949b5e25fSSatish Balay {
6500880e062SHong Zhang   Mat_SeqSBAIJ      *a = (Mat_SeqSBAIJ*)A->data;
6516849ba73SBarry Smith   PetscErrorCode    ierr;
652e2ee6c50SBarry Smith   PetscInt          *rp,k,low,high,t,ii,jj,row,nrow,i,col,l,rmax,N,lastcol = -1;
65313f74950SBarry Smith   PetscInt          *imax      =a->imax,*ai=a->i,*ailen=a->ilen;
654d0f46423SBarry Smith   PetscInt          *aj        =a->j,nonew=a->nonew,bs2=a->bs2,bs=A->rmap->bs,stepval;
655ace3abfcSBarry Smith   PetscBool         roworiented=a->roworiented;
656dd6ea824SBarry Smith   const PetscScalar *value     = v;
657f15d580aSBarry Smith   MatScalar         *ap,*aa = a->a,*bap;
6580880e062SHong Zhang 
65949b5e25fSSatish Balay   PetscFunctionBegin;
66026fbe8dcSKarl Rupp   if (roworiented) stepval = (n-1)*bs;
66126fbe8dcSKarl Rupp   else stepval = (m-1)*bs;
66226fbe8dcSKarl Rupp 
6630880e062SHong Zhang   for (k=0; k<m; k++) { /* loop over added rows */
6640880e062SHong Zhang     row = im[k];
6650880e062SHong Zhang     if (row < 0) continue;
66698921bdaSJacob Faibussowitsch     if (PetscUnlikelyDebug(row >= a->mbs)) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Block index row too large %" PetscInt_FMT " max %" PetscInt_FMT,row,a->mbs-1);
6670880e062SHong Zhang     rp   = aj + ai[row];
6680880e062SHong Zhang     ap   = aa + bs2*ai[row];
6690880e062SHong Zhang     rmax = imax[row];
6700880e062SHong Zhang     nrow = ailen[row];
6710880e062SHong Zhang     low  = 0;
672818f2c47SBarry Smith     high = nrow;
6730880e062SHong Zhang     for (l=0; l<n; l++) { /* loop over added columns */
6740880e062SHong Zhang       if (in[l] < 0) continue;
6750880e062SHong Zhang       col = in[l];
67698921bdaSJacob Faibussowitsch       if (PetscUnlikelyDebug(col >= a->nbs)) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Block index column too large %" PetscInt_FMT " max %" PetscInt_FMT,col,a->nbs-1);
677b98bf0e1SJed Brown       if (col < row) {
67826fbe8dcSKarl Rupp         if (a->ignore_ltriangular) continue; /* ignore lower triangular block */
67926fbe8dcSKarl 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)");
680b98bf0e1SJed Brown       }
68126fbe8dcSKarl Rupp       if (roworiented) value = v + k*(stepval+bs)*bs + l*bs;
68226fbe8dcSKarl Rupp       else value = v + l*(stepval+bs)*bs + k*bs;
68326fbe8dcSKarl Rupp 
68426fbe8dcSKarl Rupp       if (col <= lastcol) low = 0;
68526fbe8dcSKarl Rupp       else high = nrow;
68626fbe8dcSKarl Rupp 
687e2ee6c50SBarry Smith       lastcol = col;
6880880e062SHong Zhang       while (high-low > 7) {
6890880e062SHong Zhang         t = (low+high)/2;
6900880e062SHong Zhang         if (rp[t] > col) high = t;
6910880e062SHong Zhang         else             low  = t;
6920880e062SHong Zhang       }
6930880e062SHong Zhang       for (i=low; i<high; i++) {
6940880e062SHong Zhang         if (rp[i] > col) break;
6950880e062SHong Zhang         if (rp[i] == col) {
6960880e062SHong Zhang           bap = ap +  bs2*i;
6970880e062SHong Zhang           if (roworiented) {
6980880e062SHong Zhang             if (is == ADD_VALUES) {
6990880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
7000880e062SHong Zhang                 for (jj=ii; jj<bs2; jj+=bs) {
7010880e062SHong Zhang                   bap[jj] += *value++;
7020880e062SHong Zhang                 }
7030880e062SHong Zhang               }
7040880e062SHong Zhang             } else {
7050880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
7060880e062SHong Zhang                 for (jj=ii; jj<bs2; jj+=bs) {
7070880e062SHong Zhang                   bap[jj] = *value++;
7080880e062SHong Zhang                 }
7090880e062SHong Zhang                }
7100880e062SHong Zhang             }
7110880e062SHong Zhang           } else {
7120880e062SHong Zhang             if (is == ADD_VALUES) {
7130880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
7140880e062SHong Zhang                 for (jj=0; jj<bs; jj++) {
7150880e062SHong Zhang                   *bap++ += *value++;
7160880e062SHong Zhang                 }
7170880e062SHong Zhang               }
7180880e062SHong Zhang             } else {
7190880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
7200880e062SHong Zhang                 for (jj=0; jj<bs; jj++) {
7210880e062SHong Zhang                   *bap++  = *value++;
7220880e062SHong Zhang                 }
7230880e062SHong Zhang               }
7240880e062SHong Zhang             }
7250880e062SHong Zhang           }
7260880e062SHong Zhang           goto noinsert2;
7270880e062SHong Zhang         }
7280880e062SHong Zhang       }
7290880e062SHong Zhang       if (nonew == 1) goto noinsert2;
73098921bdaSJacob Faibussowitsch       if (nonew == -1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new block index nonzero block (%" PetscInt_FMT ", %" PetscInt_FMT ") in the matrix", row, col);
731fef13f97SBarry Smith       MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,row,col,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar);
732c03d1d03SSatish Balay       N = nrow++ - 1; high++;
7330880e062SHong Zhang       /* shift up all the later entries in this row */
734580bdb30SBarry Smith       ierr  = PetscArraymove(rp+i+1,rp+i,N-i+1);CHKERRQ(ierr);
735580bdb30SBarry Smith       ierr  = PetscArraymove(ap+bs2*(i+1),ap+bs2*i,bs2*(N-i+1));CHKERRQ(ierr);
736580bdb30SBarry Smith       ierr  = PetscArrayzero(ap+bs2*i,bs2);CHKERRQ(ierr);
7370880e062SHong Zhang       rp[i] = col;
7380880e062SHong Zhang       bap   = ap +  bs2*i;
7390880e062SHong Zhang       if (roworiented) {
7400880e062SHong Zhang         for (ii=0; ii<bs; ii++,value+=stepval) {
7410880e062SHong Zhang           for (jj=ii; jj<bs2; jj+=bs) {
7420880e062SHong Zhang             bap[jj] = *value++;
7430880e062SHong Zhang           }
7440880e062SHong Zhang         }
7450880e062SHong Zhang       } else {
7460880e062SHong Zhang         for (ii=0; ii<bs; ii++,value+=stepval) {
7470880e062SHong Zhang           for (jj=0; jj<bs; jj++) {
7480880e062SHong Zhang             *bap++ = *value++;
7490880e062SHong Zhang           }
7500880e062SHong Zhang         }
7510880e062SHong Zhang        }
7520880e062SHong Zhang     noinsert2:;
7530880e062SHong Zhang       low = i;
7540880e062SHong Zhang     }
7550880e062SHong Zhang     ailen[row] = nrow;
7560880e062SHong Zhang   }
7570880e062SHong Zhang   PetscFunctionReturn(0);
75849b5e25fSSatish Balay }
75949b5e25fSSatish Balay 
76064831d72SBarry Smith /*
76164831d72SBarry Smith     This is not yet used
76264831d72SBarry Smith */
7634108e4d5SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ_SeqAIJ_Inode(Mat A)
7640def2e27SBarry Smith {
7650def2e27SBarry Smith   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
7660def2e27SBarry Smith   PetscErrorCode ierr;
7670def2e27SBarry Smith   const PetscInt *ai = a->i, *aj = a->j,*cols;
7680def2e27SBarry Smith   PetscInt       i   = 0,j,blk_size,m = A->rmap->n,node_count = 0,nzx,nzy,*ns,row,nz,cnt,cnt2,*counts;
769ace3abfcSBarry Smith   PetscBool      flag;
7700def2e27SBarry Smith 
7710def2e27SBarry Smith   PetscFunctionBegin;
772785e854fSJed Brown   ierr = PetscMalloc1(m,&ns);CHKERRQ(ierr);
7730def2e27SBarry Smith   while (i < m) {
7740def2e27SBarry Smith     nzx = ai[i+1] - ai[i];       /* Number of nonzeros */
7750def2e27SBarry Smith     /* Limits the number of elements in a node to 'a->inode.limit' */
7760def2e27SBarry Smith     for (j=i+1,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) {
7770def2e27SBarry Smith       nzy = ai[j+1] - ai[j];
7780def2e27SBarry Smith       if (nzy != (nzx - j + i)) break;
779580bdb30SBarry Smith       ierr = PetscArraycmp(aj + ai[i] + j - i,aj + ai[j],nzy,&flag);CHKERRQ(ierr);
7800def2e27SBarry Smith       if (!flag) break;
7810def2e27SBarry Smith     }
7820def2e27SBarry Smith     ns[node_count++] = blk_size;
78326fbe8dcSKarl Rupp 
7840def2e27SBarry Smith     i = j;
7850def2e27SBarry Smith   }
7860def2e27SBarry Smith   if (!a->inode.size && m && node_count > .9*m) {
7870def2e27SBarry Smith     ierr = PetscFree(ns);CHKERRQ(ierr);
788*7d3de750SJacob Faibussowitsch     ierr = PetscInfo(A,"Found %" PetscInt_FMT " nodes out of %" PetscInt_FMT " rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr);
7890def2e27SBarry Smith   } else {
7900def2e27SBarry Smith     a->inode.node_count = node_count;
79126fbe8dcSKarl Rupp 
792785e854fSJed Brown     ierr = PetscMalloc1(node_count,&a->inode.size);CHKERRQ(ierr);
7933bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)A,node_count*sizeof(PetscInt));CHKERRQ(ierr);
794580bdb30SBarry Smith     ierr = PetscArraycpy(a->inode.size,ns,node_count);CHKERRQ(ierr);
7950def2e27SBarry Smith     ierr = PetscFree(ns);CHKERRQ(ierr);
796*7d3de750SJacob Faibussowitsch     ierr = PetscInfo(A,"Found %" PetscInt_FMT " nodes of %" PetscInt_FMT ". Limit used: %" PetscInt_FMT ". Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr);
7970def2e27SBarry Smith 
7980def2e27SBarry Smith     /* count collections of adjacent columns in each inode */
7990def2e27SBarry Smith     row = 0;
8000def2e27SBarry Smith     cnt = 0;
8010def2e27SBarry Smith     for (i=0; i<node_count; i++) {
8020def2e27SBarry Smith       cols = aj + ai[row] + a->inode.size[i];
8030def2e27SBarry Smith       nz   = ai[row+1] - ai[row] - a->inode.size[i];
8040def2e27SBarry Smith       for (j=1; j<nz; j++) {
80526fbe8dcSKarl Rupp         if (cols[j] != cols[j-1]+1) cnt++;
8060def2e27SBarry Smith       }
8070def2e27SBarry Smith       cnt++;
8080def2e27SBarry Smith       row += a->inode.size[i];
8090def2e27SBarry Smith     }
810785e854fSJed Brown     ierr = PetscMalloc1(2*cnt,&counts);CHKERRQ(ierr);
8110def2e27SBarry Smith     cnt  = 0;
8120def2e27SBarry Smith     row  = 0;
8130def2e27SBarry Smith     for (i=0; i<node_count; i++) {
8140def2e27SBarry Smith       cols = aj + ai[row] + a->inode.size[i];
8150def2e27SBarry Smith       counts[2*cnt] = cols[0];
8160def2e27SBarry Smith       nz   = ai[row+1] - ai[row] - a->inode.size[i];
8170def2e27SBarry Smith       cnt2 = 1;
8180def2e27SBarry Smith       for (j=1; j<nz; j++) {
8190def2e27SBarry Smith         if (cols[j] != cols[j-1]+1) {
8200def2e27SBarry Smith           counts[2*(cnt++)+1] = cnt2;
8210def2e27SBarry Smith           counts[2*cnt]       = cols[j];
8220def2e27SBarry Smith           cnt2 = 1;
8230def2e27SBarry Smith         } else cnt2++;
8240def2e27SBarry Smith       }
8250def2e27SBarry Smith       counts[2*(cnt++)+1] = cnt2;
8260def2e27SBarry Smith       row += a->inode.size[i];
8270def2e27SBarry Smith     }
828f4259b30SLisandro Dalcin     ierr = PetscIntView(2*cnt,counts,NULL);CHKERRQ(ierr);
8290def2e27SBarry Smith   }
83038702af4SBarry Smith   PetscFunctionReturn(0);
83138702af4SBarry Smith }
83238702af4SBarry Smith 
833dfbe8321SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ(Mat A,MatAssemblyType mode)
83449b5e25fSSatish Balay {
83549b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
8366849ba73SBarry Smith   PetscErrorCode ierr;
8378f8f2f0dSBarry Smith   PetscInt       fshift = 0,i,*ai = a->i,*aj = a->j,*imax = a->imax;
838d0f46423SBarry Smith   PetscInt       m      = A->rmap->N,*ip,N,*ailen = a->ilen;
83913f74950SBarry Smith   PetscInt       mbs    = a->mbs,bs2 = a->bs2,rmax = 0;
84049b5e25fSSatish Balay   MatScalar      *aa    = a->a,*ap;
84149b5e25fSSatish Balay 
84249b5e25fSSatish Balay   PetscFunctionBegin;
84349b5e25fSSatish Balay   if (mode == MAT_FLUSH_ASSEMBLY) PetscFunctionReturn(0);
84449b5e25fSSatish Balay 
84549b5e25fSSatish Balay   if (m) rmax = ailen[0];
84649b5e25fSSatish Balay   for (i=1; i<mbs; i++) {
84749b5e25fSSatish Balay     /* move each row back by the amount of empty slots (fshift) before it*/
84849b5e25fSSatish Balay     fshift += imax[i-1] - ailen[i-1];
84949b5e25fSSatish Balay     rmax    = PetscMax(rmax,ailen[i]);
85049b5e25fSSatish Balay     if (fshift) {
851580bdb30SBarry Smith       ip = aj + ai[i];
852580bdb30SBarry Smith       ap = aa + bs2*ai[i];
85349b5e25fSSatish Balay       N  = ailen[i];
854580bdb30SBarry Smith       ierr  = PetscArraymove(ip-fshift,ip,N);CHKERRQ(ierr);
855580bdb30SBarry Smith       ierr  = PetscArraymove(ap-bs2*fshift,ap,bs2*N);CHKERRQ(ierr);
85649b5e25fSSatish Balay     }
85749b5e25fSSatish Balay     ai[i] = ai[i-1] + ailen[i-1];
85849b5e25fSSatish Balay   }
85949b5e25fSSatish Balay   if (mbs) {
86049b5e25fSSatish Balay     fshift += imax[mbs-1] - ailen[mbs-1];
86149b5e25fSSatish Balay     ai[mbs] = ai[mbs-1] + ailen[mbs-1];
86249b5e25fSSatish Balay   }
86349b5e25fSSatish Balay   /* reset ilen and imax for each row */
86449b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
86549b5e25fSSatish Balay     ailen[i] = imax[i] = ai[i+1] - ai[i];
86649b5e25fSSatish Balay   }
8676c6c5352SBarry Smith   a->nz = ai[mbs];
86849b5e25fSSatish Balay 
869b424e231SHong Zhang   /* diagonals may have moved, reset it */
870b424e231SHong Zhang   if (a->diag) {
871580bdb30SBarry Smith     ierr = PetscArraycpy(a->diag,ai,mbs);CHKERRQ(ierr);
87249b5e25fSSatish Balay   }
87398921bdaSJacob Faibussowitsch   if (fshift && a->nounused == -1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB, "Unused space detected in matrix: %" PetscInt_FMT " X %" PetscInt_FMT " block size %" PetscInt_FMT ", %" PetscInt_FMT " unneeded", m, A->cmap->n, A->rmap->bs, fshift*bs2);
87426fbe8dcSKarl Rupp 
875*7d3de750SJacob Faibussowitsch   ierr = PetscInfo(A,"Matrix size: %" PetscInt_FMT " X %" PetscInt_FMT ", block size %" PetscInt_FMT "; storage space: %" PetscInt_FMT " unneeded, %" PetscInt_FMT " used\n",m,A->rmap->N,A->rmap->bs,fshift*bs2,a->nz*bs2);CHKERRQ(ierr);
876*7d3de750SJacob Faibussowitsch   ierr = PetscInfo(A,"Number of mallocs during MatSetValues is %" PetscInt_FMT "\n",a->reallocs);CHKERRQ(ierr);
877*7d3de750SJacob Faibussowitsch   ierr = PetscInfo(A,"Most nonzeros blocks in any row is %" PetscInt_FMT "\n",rmax);CHKERRQ(ierr);
87826fbe8dcSKarl Rupp 
8798e58a170SBarry Smith   A->info.mallocs    += a->reallocs;
88049b5e25fSSatish Balay   a->reallocs         = 0;
88149b5e25fSSatish Balay   A->info.nz_unneeded = (PetscReal)fshift*bs2;
882061b2667SBarry Smith   a->idiagvalid       = PETSC_FALSE;
8834dcd73b1SHong Zhang   a->rmax             = rmax;
88438702af4SBarry Smith 
88538702af4SBarry Smith   if (A->cmap->n < 65536 && A->cmap->bs == 1) {
88644e1c64aSLisandro Dalcin     if (a->jshort && a->free_jshort) {
88717803ae8SHong Zhang       /* when matrix data structure is changed, previous jshort must be replaced */
88817803ae8SHong Zhang       ierr = PetscFree(a->jshort);CHKERRQ(ierr);
88917803ae8SHong Zhang     }
890785e854fSJed Brown     ierr = PetscMalloc1(a->i[A->rmap->n],&a->jshort);CHKERRQ(ierr);
8913bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)A,a->i[A->rmap->n]*sizeof(unsigned short));CHKERRQ(ierr);
89238702af4SBarry Smith     for (i=0; i<a->i[A->rmap->n]; i++) a->jshort[i] = a->j[i];
89338702af4SBarry Smith     A->ops->mult   = MatMult_SeqSBAIJ_1_ushort;
89441f059aeSBarry Smith     A->ops->sor    = MatSOR_SeqSBAIJ_ushort;
8954da8f245SBarry Smith     a->free_jshort = PETSC_TRUE;
89638702af4SBarry Smith   }
89749b5e25fSSatish Balay   PetscFunctionReturn(0);
89849b5e25fSSatish Balay }
89949b5e25fSSatish Balay 
90049b5e25fSSatish Balay /*
90149b5e25fSSatish Balay    This function returns an array of flags which indicate the locations of contiguous
90249b5e25fSSatish Balay    blocks that should be zeroed. for eg: if bs = 3  and is = [0,1,2,3,5,6,7,8,9]
903a5b23f4aSJose E. Roman    then the resulting sizes = [3,1,1,3,1] corresponding to sets [(0,1,2),(3),(5),(6,7,8),(9)]
90449b5e25fSSatish Balay    Assume: sizes should be long enough to hold all the values.
90549b5e25fSSatish Balay */
90613f74950SBarry Smith PetscErrorCode MatZeroRows_SeqSBAIJ_Check_Blocks(PetscInt idx[],PetscInt n,PetscInt bs,PetscInt sizes[], PetscInt *bs_max)
90749b5e25fSSatish Balay {
90813f74950SBarry Smith   PetscInt  i,j,k,row;
909ace3abfcSBarry Smith   PetscBool flg;
91049b5e25fSSatish Balay 
91149b5e25fSSatish Balay   PetscFunctionBegin;
91249b5e25fSSatish Balay   for (i=0,j=0; i<n; j++) {
91349b5e25fSSatish Balay     row = idx[i];
914a5b23f4aSJose E. Roman     if (row%bs!=0) { /* Not the beginning of a block */
91549b5e25fSSatish Balay       sizes[j] = 1;
91649b5e25fSSatish Balay       i++;
91749b5e25fSSatish Balay     } else if (i+bs > n) { /* Beginning of a block, but complete block doesn't exist (at idx end) */
91849b5e25fSSatish Balay       sizes[j] = 1;         /* Also makes sure atleast 'bs' values exist for next else */
91949b5e25fSSatish Balay       i++;
92049b5e25fSSatish Balay     } else { /* Begining of the block, so check if the complete block exists */
92149b5e25fSSatish Balay       flg = PETSC_TRUE;
92249b5e25fSSatish Balay       for (k=1; k<bs; k++) {
92349b5e25fSSatish Balay         if (row+k != idx[i+k]) { /* break in the block */
92449b5e25fSSatish Balay           flg = PETSC_FALSE;
92549b5e25fSSatish Balay           break;
92649b5e25fSSatish Balay         }
92749b5e25fSSatish Balay       }
928abc0a331SBarry Smith       if (flg) { /* No break in the bs */
92949b5e25fSSatish Balay         sizes[j] = bs;
93049b5e25fSSatish Balay         i       += bs;
93149b5e25fSSatish Balay       } else {
93249b5e25fSSatish Balay         sizes[j] = 1;
93349b5e25fSSatish Balay         i++;
93449b5e25fSSatish Balay       }
93549b5e25fSSatish Balay     }
93649b5e25fSSatish Balay   }
93749b5e25fSSatish Balay   *bs_max = j;
93849b5e25fSSatish Balay   PetscFunctionReturn(0);
93949b5e25fSSatish Balay }
94049b5e25fSSatish Balay 
94149b5e25fSSatish Balay /* Only add/insert a(i,j) with i<=j (blocks).
94249b5e25fSSatish Balay    Any a(i,j) with i>j input by user is ingored.
94349b5e25fSSatish Balay */
94449b5e25fSSatish Balay 
94513f74950SBarry Smith PetscErrorCode MatSetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is)
94649b5e25fSSatish Balay {
94749b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
9486849ba73SBarry Smith   PetscErrorCode ierr;
949e2ee6c50SBarry Smith   PetscInt       *rp,k,low,high,t,ii,row,nrow,i,col,l,rmax,N,lastcol = -1;
95013f74950SBarry Smith   PetscInt       *imax=a->imax,*ai=a->i,*ailen=a->ilen,roworiented=a->roworiented;
951d0f46423SBarry Smith   PetscInt       *aj  =a->j,nonew=a->nonew,bs=A->rmap->bs,brow,bcol;
95213f74950SBarry Smith   PetscInt       ridx,cidx,bs2=a->bs2;
95349b5e25fSSatish Balay   MatScalar      *ap,value,*aa=a->a,*bap;
95449b5e25fSSatish Balay 
95549b5e25fSSatish Balay   PetscFunctionBegin;
95649b5e25fSSatish Balay   for (k=0; k<m; k++) { /* loop over added rows */
95749b5e25fSSatish Balay     row  = im[k];       /* row number */
95849b5e25fSSatish Balay     brow = row/bs;      /* block row number */
95949b5e25fSSatish Balay     if (row < 0) continue;
96098921bdaSJacob Faibussowitsch     if (PetscUnlikelyDebug(row >= A->rmap->N)) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %" PetscInt_FMT " max %" PetscInt_FMT,row,A->rmap->N-1);
96149b5e25fSSatish Balay     rp   = aj + ai[brow]; /*ptr to beginning of column value of the row block*/
96249b5e25fSSatish Balay     ap   = aa + bs2*ai[brow]; /*ptr to beginning of element value of the row block*/
96349b5e25fSSatish Balay     rmax = imax[brow];  /* maximum space allocated for this row */
96449b5e25fSSatish Balay     nrow = ailen[brow]; /* actual length of this row */
96549b5e25fSSatish Balay     low  = 0;
9668509e838SStefano Zampini     high = nrow;
96749b5e25fSSatish Balay     for (l=0; l<n; l++) { /* loop over added columns */
96849b5e25fSSatish Balay       if (in[l] < 0) continue;
96998921bdaSJacob Faibussowitsch       if (PetscUnlikelyDebug(in[l] >= A->cmap->N)) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %" PetscInt_FMT " max %" PetscInt_FMT,in[l],A->cmap->N-1);
97049b5e25fSSatish Balay       col  = in[l];
97149b5e25fSSatish Balay       bcol = col/bs;              /* block col number */
97249b5e25fSSatish Balay 
973941593c8SHong Zhang       if (brow > bcol) {
97426fbe8dcSKarl Rupp         if (a->ignore_ltriangular) continue; /* ignore lower triangular values */
97526fbe8dcSKarl 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)");
976941593c8SHong Zhang       }
977f4989cb3SHong Zhang 
97849b5e25fSSatish Balay       ridx = row % bs; cidx = col % bs; /*row and col index inside the block */
9798549e402SHong Zhang       if ((brow==bcol && ridx<=cidx) || (brow<bcol)) {
98049b5e25fSSatish Balay         /* element value a(k,l) */
98126fbe8dcSKarl Rupp         if (roworiented) value = v[l + k*n];
98226fbe8dcSKarl Rupp         else value = v[k + l*m];
98349b5e25fSSatish Balay 
98449b5e25fSSatish Balay         /* move pointer bap to a(k,l) quickly and add/insert value */
98526fbe8dcSKarl Rupp         if (col <= lastcol) low = 0;
9868509e838SStefano Zampini         else high = nrow;
9878509e838SStefano Zampini 
988e2ee6c50SBarry Smith         lastcol = col;
98949b5e25fSSatish Balay         while (high-low > 7) {
99049b5e25fSSatish Balay           t = (low+high)/2;
99149b5e25fSSatish Balay           if (rp[t] > bcol) high = t;
99249b5e25fSSatish Balay           else              low  = t;
99349b5e25fSSatish Balay         }
99449b5e25fSSatish Balay         for (i=low; i<high; i++) {
99549b5e25fSSatish Balay           if (rp[i] > bcol) break;
99649b5e25fSSatish Balay           if (rp[i] == bcol) {
99749b5e25fSSatish Balay             bap = ap +  bs2*i + bs*cidx + ridx;
99849b5e25fSSatish Balay             if (is == ADD_VALUES) *bap += value;
99949b5e25fSSatish Balay             else                  *bap  = value;
10008549e402SHong Zhang             /* for diag block, add/insert its symmetric element a(cidx,ridx) */
10018549e402SHong Zhang             if (brow == bcol && ridx < cidx) {
10028549e402SHong Zhang               bap = ap +  bs2*i + bs*ridx + cidx;
10038549e402SHong Zhang               if (is == ADD_VALUES) *bap += value;
10048549e402SHong Zhang               else                  *bap  = value;
10058549e402SHong Zhang             }
100649b5e25fSSatish Balay             goto noinsert1;
100749b5e25fSSatish Balay           }
100849b5e25fSSatish Balay         }
100949b5e25fSSatish Balay 
101049b5e25fSSatish Balay         if (nonew == 1) goto noinsert1;
101198921bdaSJacob Faibussowitsch         if (nonew == -1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%" PetscInt_FMT ", %" PetscInt_FMT ") in the matrix", row, col);
1012fef13f97SBarry Smith         MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,brow,bcol,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar);
101349b5e25fSSatish Balay 
1014c03d1d03SSatish Balay         N = nrow++ - 1; high++;
101549b5e25fSSatish Balay         /* shift up all the later entries in this row */
1016580bdb30SBarry Smith         ierr = PetscArraymove(rp+i+1,rp+i,N-i+1);CHKERRQ(ierr);
1017580bdb30SBarry Smith         ierr = PetscArraymove(ap+bs2*(i+1),ap+bs2*i,bs2*(N-i+1));CHKERRQ(ierr);
1018580bdb30SBarry Smith         ierr = PetscArrayzero(ap+bs2*i,bs2);CHKERRQ(ierr);
101949b5e25fSSatish Balay         rp[i]                      = bcol;
102049b5e25fSSatish Balay         ap[bs2*i + bs*cidx + ridx] = value;
10218509e838SStefano Zampini         /* for diag block, add/insert its symmetric element a(cidx,ridx) */
10228509e838SStefano Zampini         if (brow == bcol && ridx < cidx) {
10238509e838SStefano Zampini           ap[bs2*i + bs*ridx + cidx] = value;
10248509e838SStefano Zampini         }
1025e56f5c9eSBarry Smith         A->nonzerostate++;
102649b5e25fSSatish Balay noinsert1:;
102749b5e25fSSatish Balay         low = i;
10288549e402SHong Zhang       }
102949b5e25fSSatish Balay     }   /* end of loop over added columns */
103049b5e25fSSatish Balay     ailen[brow] = nrow;
103149b5e25fSSatish Balay   }   /* end of loop over added rows */
103249b5e25fSSatish Balay   PetscFunctionReturn(0);
103349b5e25fSSatish Balay }
103449b5e25fSSatish Balay 
10350481f469SBarry Smith PetscErrorCode MatICCFactor_SeqSBAIJ(Mat inA,IS row,const MatFactorInfo *info)
103649b5e25fSSatish Balay {
10374ccecd49SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)inA->data;
103849b5e25fSSatish Balay   Mat            outA;
1039dfbe8321SBarry Smith   PetscErrorCode ierr;
1040ace3abfcSBarry Smith   PetscBool      row_identity;
104149b5e25fSSatish Balay 
104249b5e25fSSatish Balay   PetscFunctionBegin;
1043e32f2f54SBarry Smith   if (info->levels != 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only levels=0 is supported for in-place icc");
1044c84f5b01SHong Zhang   ierr = ISIdentity(row,&row_identity);CHKERRQ(ierr);
1045e32f2f54SBarry Smith   if (!row_identity) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix reordering is not supported");
104698921bdaSJacob Faibussowitsch   if (inA->rmap->bs != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix block size %" PetscInt_FMT " is not supported",inA->rmap->bs); /* Need to replace MatCholeskyFactorSymbolic_SeqSBAIJ_MSR()! */
1047c84f5b01SHong Zhang 
104849b5e25fSSatish Balay   outA            = inA;
1049d5f3da31SBarry Smith   inA->factortype = MAT_FACTOR_ICC;
1050f6224b95SHong Zhang   ierr = PetscFree(inA->solvertype);CHKERRQ(ierr);
1051f6224b95SHong Zhang   ierr = PetscStrallocpy(MATSOLVERPETSC,&inA->solvertype);CHKERRQ(ierr);
105249b5e25fSSatish Balay 
10531a3463dfSHong Zhang   ierr = MatMarkDiagonal_SeqSBAIJ(inA);CHKERRQ(ierr);
1054d595f711SHong Zhang   ierr = MatSeqSBAIJSetNumericFactorization_inplace(inA,row_identity);CHKERRQ(ierr);
105549b5e25fSSatish Balay 
1056c3122656SLisandro Dalcin   ierr   = PetscObjectReference((PetscObject)row);CHKERRQ(ierr);
10576bf464f9SBarry Smith   ierr   = ISDestroy(&a->row);CHKERRQ(ierr);
1058c84f5b01SHong Zhang   a->row = row;
1059c3122656SLisandro Dalcin   ierr   = PetscObjectReference((PetscObject)row);CHKERRQ(ierr);
10606bf464f9SBarry Smith   ierr   = ISDestroy(&a->col);CHKERRQ(ierr);
1061c84f5b01SHong Zhang   a->col = row;
1062c84f5b01SHong Zhang 
1063c84f5b01SHong Zhang   /* Create the invert permutation so that it can be used in MatCholeskyFactorNumeric() */
1064c84f5b01SHong Zhang   if (a->icol) {ierr = ISInvertPermutation(row,PETSC_DECIDE, &a->icol);CHKERRQ(ierr);}
10653bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)inA,(PetscObject)a->icol);CHKERRQ(ierr);
106649b5e25fSSatish Balay 
106749b5e25fSSatish Balay   if (!a->solve_work) {
1068854ce69bSBarry Smith     ierr = PetscMalloc1(inA->rmap->N+inA->rmap->bs,&a->solve_work);CHKERRQ(ierr);
10693bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)inA,(inA->rmap->N+inA->rmap->bs)*sizeof(PetscScalar));CHKERRQ(ierr);
107049b5e25fSSatish Balay   }
107149b5e25fSSatish Balay 
1072719d5645SBarry Smith   ierr = MatCholeskyFactorNumeric(outA,inA,info);CHKERRQ(ierr);
107349b5e25fSSatish Balay   PetscFunctionReturn(0);
107449b5e25fSSatish Balay }
1075950f1e5bSHong Zhang 
10767087cfbeSBarry Smith PetscErrorCode  MatSeqSBAIJSetColumnIndices_SeqSBAIJ(Mat mat,PetscInt *indices)
107749b5e25fSSatish Balay {
1078045c9aa0SHong Zhang   Mat_SeqSBAIJ   *baij = (Mat_SeqSBAIJ*)mat->data;
107913f74950SBarry Smith   PetscInt       i,nz,n;
10807827cd58SJed Brown   PetscErrorCode ierr;
108149b5e25fSSatish Balay 
108249b5e25fSSatish Balay   PetscFunctionBegin;
10836c6c5352SBarry Smith   nz = baij->maxnz;
1084d0f46423SBarry Smith   n  = mat->cmap->n;
108526fbe8dcSKarl Rupp   for (i=0; i<nz; i++) baij->j[i] = indices[i];
108626fbe8dcSKarl Rupp 
10876c6c5352SBarry Smith   baij->nz = nz;
108826fbe8dcSKarl Rupp   for (i=0; i<n; i++) baij->ilen[i] = baij->imax[i];
108926fbe8dcSKarl Rupp 
10907827cd58SJed Brown   ierr = MatSetOption(mat,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
109149b5e25fSSatish Balay   PetscFunctionReturn(0);
109249b5e25fSSatish Balay }
109349b5e25fSSatish Balay 
109449b5e25fSSatish Balay /*@
109519585528SSatish Balay   MatSeqSBAIJSetColumnIndices - Set the column indices for all the rows
109649b5e25fSSatish Balay   in the matrix.
109749b5e25fSSatish Balay 
109849b5e25fSSatish Balay   Input Parameters:
109919585528SSatish Balay   +  mat     - the SeqSBAIJ matrix
110049b5e25fSSatish Balay   -  indices - the column indices
110149b5e25fSSatish Balay 
110249b5e25fSSatish Balay   Level: advanced
110349b5e25fSSatish Balay 
110449b5e25fSSatish Balay   Notes:
110549b5e25fSSatish Balay   This can be called if you have precomputed the nonzero structure of the
110649b5e25fSSatish Balay   matrix and want to provide it to the matrix object to improve the performance
110749b5e25fSSatish Balay   of the MatSetValues() operation.
110849b5e25fSSatish Balay 
110949b5e25fSSatish Balay   You MUST have set the correct numbers of nonzeros per row in the call to
1110d1be2dadSMatthew Knepley   MatCreateSeqSBAIJ(), and the columns indices MUST be sorted.
111149b5e25fSSatish Balay 
1112ab9f2c04SSatish Balay   MUST be called before any calls to MatSetValues()
111349b5e25fSSatish Balay 
1114ab9f2c04SSatish Balay   .seealso: MatCreateSeqSBAIJ
111549b5e25fSSatish Balay @*/
11167087cfbeSBarry Smith PetscErrorCode  MatSeqSBAIJSetColumnIndices(Mat mat,PetscInt *indices)
111749b5e25fSSatish Balay {
11184ac538c5SBarry Smith   PetscErrorCode ierr;
111949b5e25fSSatish Balay 
112049b5e25fSSatish Balay   PetscFunctionBegin;
11210700a824SBarry Smith   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
11224482741eSBarry Smith   PetscValidPointer(indices,2);
11234ac538c5SBarry Smith   ierr = PetscUseMethod(mat,"MatSeqSBAIJSetColumnIndices_C",(Mat,PetscInt*),(mat,indices));CHKERRQ(ierr);
112449b5e25fSSatish Balay   PetscFunctionReturn(0);
112549b5e25fSSatish Balay }
112649b5e25fSSatish Balay 
11273c896bc6SHong Zhang PetscErrorCode MatCopy_SeqSBAIJ(Mat A,Mat B,MatStructure str)
11283c896bc6SHong Zhang {
11293c896bc6SHong Zhang   PetscErrorCode ierr;
11304c7a3774SStefano Zampini   PetscBool      isbaij;
11313c896bc6SHong Zhang 
11323c896bc6SHong Zhang   PetscFunctionBegin;
11334c7a3774SStefano Zampini   ierr = PetscObjectTypeCompareAny((PetscObject)B,&isbaij,MATSEQSBAIJ,MATMPISBAIJ,"");CHKERRQ(ierr);
113498921bdaSJacob Faibussowitsch   if (!isbaij) SETERRQ(PetscObjectComm((PetscObject)B),PETSC_ERR_SUP,"Not for matrix type %s",((PetscObject)B)->type_name);
11354c7a3774SStefano Zampini   /* If the two matrices have the same copy implementation and nonzero pattern, use fast copy. */
11364c7a3774SStefano Zampini   if (str == SAME_NONZERO_PATTERN && A->ops->copy == B->ops->copy) {
11373c896bc6SHong Zhang     Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
11383c896bc6SHong Zhang     Mat_SeqSBAIJ *b = (Mat_SeqSBAIJ*)B->data;
11393c896bc6SHong Zhang 
11404c7a3774SStefano Zampini     if (a->i[a->mbs] != b->i[b->mbs]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Number of nonzeros in two matrices are different");
11414c7a3774SStefano Zampini     if (a->mbs != b->mbs) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Number of rows in two matrices are different");
11424c7a3774SStefano Zampini     if (a->bs2 != b->bs2) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Different block size");
1143580bdb30SBarry Smith     ierr = PetscArraycpy(b->a,a->a,a->bs2*a->i[a->mbs]);CHKERRQ(ierr);
1144cdc753b6SBarry Smith     ierr = PetscObjectStateIncrease((PetscObject)B);CHKERRQ(ierr);
11453c896bc6SHong Zhang   } else {
1146f5edf698SHong Zhang     ierr = MatGetRowUpperTriangular(A);CHKERRQ(ierr);
11473c896bc6SHong Zhang     ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr);
1148f5edf698SHong Zhang     ierr = MatRestoreRowUpperTriangular(A);CHKERRQ(ierr);
11493c896bc6SHong Zhang   }
11503c896bc6SHong Zhang   PetscFunctionReturn(0);
11513c896bc6SHong Zhang }
11523c896bc6SHong Zhang 
11534994cf47SJed Brown PetscErrorCode MatSetUp_SeqSBAIJ(Mat A)
1154273d9f13SBarry Smith {
1155dfbe8321SBarry Smith   PetscErrorCode ierr;
1156273d9f13SBarry Smith 
1157273d9f13SBarry Smith   PetscFunctionBegin;
1158f4259b30SLisandro Dalcin   ierr = MatSeqSBAIJSetPreallocation(A,A->rmap->bs,PETSC_DEFAULT,NULL);CHKERRQ(ierr);
1159273d9f13SBarry Smith   PetscFunctionReturn(0);
1160273d9f13SBarry Smith }
1161273d9f13SBarry Smith 
1162cda14afcSprj- static PetscErrorCode MatSeqSBAIJGetArray_SeqSBAIJ(Mat A,PetscScalar *array[])
1163a6ece127SHong Zhang {
1164a6ece127SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
11655fd66863SKarl Rupp 
1166a6ece127SHong Zhang   PetscFunctionBegin;
1167a6ece127SHong Zhang   *array = a->a;
1168a6ece127SHong Zhang   PetscFunctionReturn(0);
1169a6ece127SHong Zhang }
1170a6ece127SHong Zhang 
1171cda14afcSprj- static PetscErrorCode MatSeqSBAIJRestoreArray_SeqSBAIJ(Mat A,PetscScalar *array[])
1172a6ece127SHong Zhang {
1173a6ece127SHong Zhang   PetscFunctionBegin;
1174cda14afcSprj-   *array = NULL;
1175a6ece127SHong Zhang   PetscFunctionReturn(0);
1176a6ece127SHong Zhang }
1177a6ece127SHong Zhang 
117852768537SHong Zhang PetscErrorCode MatAXPYGetPreallocation_SeqSBAIJ(Mat Y,Mat X,PetscInt *nnz)
117952768537SHong Zhang {
1180b264fe52SHong Zhang   PetscInt       bs = Y->rmap->bs,mbs = Y->rmap->N/bs;
118152768537SHong Zhang   Mat_SeqSBAIJ   *x = (Mat_SeqSBAIJ*)X->data;
118252768537SHong Zhang   Mat_SeqSBAIJ   *y = (Mat_SeqSBAIJ*)Y->data;
1183b264fe52SHong Zhang   PetscErrorCode ierr;
118452768537SHong Zhang 
118552768537SHong Zhang   PetscFunctionBegin;
118652768537SHong Zhang   /* Set the number of nonzeros in the new matrix */
1187b264fe52SHong Zhang   ierr = MatAXPYGetPreallocation_SeqX_private(mbs,x->i,x->j,y->i,y->j,nnz);CHKERRQ(ierr);
118852768537SHong Zhang   PetscFunctionReturn(0);
118952768537SHong Zhang }
119052768537SHong Zhang 
1191f4df32b1SMatthew Knepley PetscErrorCode MatAXPY_SeqSBAIJ(Mat Y,PetscScalar a,Mat X,MatStructure str)
119242ee4b1aSHong Zhang {
119342ee4b1aSHong Zhang   Mat_SeqSBAIJ   *x=(Mat_SeqSBAIJ*)X->data, *y=(Mat_SeqSBAIJ*)Y->data;
1194dfbe8321SBarry Smith   PetscErrorCode ierr;
119531ce2d13SHong Zhang   PetscInt       bs=Y->rmap->bs,bs2=bs*bs;
1196e838b9e7SJed Brown   PetscBLASInt   one = 1;
119742ee4b1aSHong Zhang 
119842ee4b1aSHong Zhang   PetscFunctionBegin;
119942ee4b1aSHong Zhang   if (str == SAME_NONZERO_PATTERN) {
1200f4df32b1SMatthew Knepley     PetscScalar  alpha = a;
1201c5df96a5SBarry Smith     PetscBLASInt bnz;
1202c5df96a5SBarry Smith     ierr = PetscBLASIntCast(x->nz*bs2,&bnz);CHKERRQ(ierr);
12038b83055fSJed Brown     PetscStackCallBLAS("BLASaxpy",BLASaxpy_(&bnz,&alpha,x->a,&one,y->a,&one));
1204a3fa217bSJose E. Roman     ierr = PetscObjectStateIncrease((PetscObject)Y);CHKERRQ(ierr);
1205ab784542SHong Zhang   } else if (str == SUBSET_NONZERO_PATTERN) { /* nonzeros of X is a subset of Y's */
1206ab784542SHong Zhang     ierr = MatSetOption(X,MAT_GETROW_UPPERTRIANGULAR,PETSC_TRUE);CHKERRQ(ierr);
1207ab784542SHong Zhang     ierr = MatAXPY_Basic(Y,a,X,str);CHKERRQ(ierr);
1208ab784542SHong Zhang     ierr = MatSetOption(X,MAT_GETROW_UPPERTRIANGULAR,PETSC_FALSE);CHKERRQ(ierr);
120942ee4b1aSHong Zhang   } else {
121052768537SHong Zhang     Mat      B;
121152768537SHong Zhang     PetscInt *nnz;
121252768537SHong Zhang     if (bs != X->rmap->bs) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Matrices must have same block size");
1213f5edf698SHong Zhang     ierr = MatGetRowUpperTriangular(X);CHKERRQ(ierr);
121452768537SHong Zhang     ierr = MatGetRowUpperTriangular(Y);CHKERRQ(ierr);
121552768537SHong Zhang     ierr = PetscMalloc1(Y->rmap->N,&nnz);CHKERRQ(ierr);
121652768537SHong Zhang     ierr = MatCreate(PetscObjectComm((PetscObject)Y),&B);CHKERRQ(ierr);
121752768537SHong Zhang     ierr = PetscObjectSetName((PetscObject)B,((PetscObject)Y)->name);CHKERRQ(ierr);
121852768537SHong Zhang     ierr = MatSetSizes(B,Y->rmap->n,Y->cmap->n,Y->rmap->N,Y->cmap->N);CHKERRQ(ierr);
121952768537SHong Zhang     ierr = MatSetBlockSizesFromMats(B,Y,Y);CHKERRQ(ierr);
12204c7a3774SStefano Zampini     ierr = MatSetType(B,((PetscObject)Y)->type_name);CHKERRQ(ierr);
122152768537SHong Zhang     ierr = MatAXPYGetPreallocation_SeqSBAIJ(Y,X,nnz);CHKERRQ(ierr);
122252768537SHong Zhang     ierr = MatSeqSBAIJSetPreallocation(B,bs,0,nnz);CHKERRQ(ierr);
122352768537SHong Zhang 
122452768537SHong Zhang     ierr = MatAXPY_BasicWithPreallocation(B,Y,a,X,str);CHKERRQ(ierr);
122552768537SHong Zhang 
122679c2fd05SStefano Zampini     ierr = MatHeaderMerge(Y,&B);CHKERRQ(ierr);
122752768537SHong Zhang     ierr = PetscFree(nnz);CHKERRQ(ierr);
1228f5edf698SHong Zhang     ierr = MatRestoreRowUpperTriangular(X);CHKERRQ(ierr);
122952768537SHong Zhang     ierr = MatRestoreRowUpperTriangular(Y);CHKERRQ(ierr);
123042ee4b1aSHong Zhang   }
123142ee4b1aSHong Zhang   PetscFunctionReturn(0);
123242ee4b1aSHong Zhang }
123342ee4b1aSHong Zhang 
1234ace3abfcSBarry Smith PetscErrorCode MatIsSymmetric_SeqSBAIJ(Mat A,PetscReal tol,PetscBool  *flg)
1235efcf0fc3SBarry Smith {
1236efcf0fc3SBarry Smith   PetscFunctionBegin;
1237efcf0fc3SBarry Smith   *flg = PETSC_TRUE;
1238efcf0fc3SBarry Smith   PetscFunctionReturn(0);
1239efcf0fc3SBarry Smith }
1240efcf0fc3SBarry Smith 
1241ace3abfcSBarry Smith PetscErrorCode MatIsStructurallySymmetric_SeqSBAIJ(Mat A,PetscBool  *flg)
1242efcf0fc3SBarry Smith {
1243efcf0fc3SBarry Smith   PetscFunctionBegin;
1244efcf0fc3SBarry Smith   *flg = PETSC_TRUE;
1245efcf0fc3SBarry Smith   PetscFunctionReturn(0);
1246efcf0fc3SBarry Smith }
1247efcf0fc3SBarry Smith 
1248ace3abfcSBarry Smith PetscErrorCode MatIsHermitian_SeqSBAIJ(Mat A,PetscReal tol,PetscBool  *flg)
1249efcf0fc3SBarry Smith {
1250efcf0fc3SBarry Smith   PetscFunctionBegin;
1251efcf0fc3SBarry Smith   *flg = PETSC_FALSE;
1252efcf0fc3SBarry Smith   PetscFunctionReturn(0);
1253efcf0fc3SBarry Smith }
1254efcf0fc3SBarry Smith 
12552726fb6dSPierre Jolivet PetscErrorCode MatConjugate_SeqSBAIJ(Mat A)
12562726fb6dSPierre Jolivet {
12572726fb6dSPierre Jolivet #if defined(PETSC_USE_COMPLEX)
12582726fb6dSPierre Jolivet   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
12592726fb6dSPierre Jolivet   PetscInt     i,nz = a->bs2*a->i[a->mbs];
12602726fb6dSPierre Jolivet   MatScalar    *aa = a->a;
12612726fb6dSPierre Jolivet 
12622726fb6dSPierre Jolivet   PetscFunctionBegin;
12632726fb6dSPierre Jolivet   for (i=0; i<nz; i++) aa[i] = PetscConj(aa[i]);
12642726fb6dSPierre Jolivet #else
12652726fb6dSPierre Jolivet   PetscFunctionBegin;
12662726fb6dSPierre Jolivet #endif
12672726fb6dSPierre Jolivet   PetscFunctionReturn(0);
12682726fb6dSPierre Jolivet }
12692726fb6dSPierre Jolivet 
127099cafbc1SBarry Smith PetscErrorCode MatRealPart_SeqSBAIJ(Mat A)
127199cafbc1SBarry Smith {
127299cafbc1SBarry Smith   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
127399cafbc1SBarry Smith   PetscInt     i,nz = a->bs2*a->i[a->mbs];
1274dd6ea824SBarry Smith   MatScalar    *aa = a->a;
127599cafbc1SBarry Smith 
127699cafbc1SBarry Smith   PetscFunctionBegin;
127799cafbc1SBarry Smith   for (i=0; i<nz; i++) aa[i] = PetscRealPart(aa[i]);
127899cafbc1SBarry Smith   PetscFunctionReturn(0);
127999cafbc1SBarry Smith }
128099cafbc1SBarry Smith 
128199cafbc1SBarry Smith PetscErrorCode MatImaginaryPart_SeqSBAIJ(Mat A)
128299cafbc1SBarry Smith {
128399cafbc1SBarry Smith   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
128499cafbc1SBarry Smith   PetscInt     i,nz = a->bs2*a->i[a->mbs];
1285dd6ea824SBarry Smith   MatScalar    *aa = a->a;
128699cafbc1SBarry Smith 
128799cafbc1SBarry Smith   PetscFunctionBegin;
128899cafbc1SBarry Smith   for (i=0; i<nz; i++) aa[i] = PetscImaginaryPart(aa[i]);
128999cafbc1SBarry Smith   PetscFunctionReturn(0);
129099cafbc1SBarry Smith }
129199cafbc1SBarry Smith 
12923bededecSBarry Smith PetscErrorCode MatZeroRowsColumns_SeqSBAIJ(Mat A,PetscInt is_n,const PetscInt is_idx[],PetscScalar diag,Vec x, Vec b)
12933bededecSBarry Smith {
12943bededecSBarry Smith   Mat_SeqSBAIJ      *baij=(Mat_SeqSBAIJ*)A->data;
12953bededecSBarry Smith   PetscErrorCode    ierr;
12963bededecSBarry Smith   PetscInt          i,j,k,count;
12973bededecSBarry Smith   PetscInt          bs   =A->rmap->bs,bs2=baij->bs2,row,col;
12983bededecSBarry Smith   PetscScalar       zero = 0.0;
12993bededecSBarry Smith   MatScalar         *aa;
13003bededecSBarry Smith   const PetscScalar *xx;
13013bededecSBarry Smith   PetscScalar       *bb;
130256777dd2SBarry Smith   PetscBool         *zeroed,vecs = PETSC_FALSE;
13033bededecSBarry Smith 
13043bededecSBarry Smith   PetscFunctionBegin;
13053bededecSBarry Smith   /* fix right hand side if needed */
13063bededecSBarry Smith   if (x && b) {
13073bededecSBarry Smith     ierr = VecGetArrayRead(x,&xx);CHKERRQ(ierr);
13083bededecSBarry Smith     ierr = VecGetArray(b,&bb);CHKERRQ(ierr);
130956777dd2SBarry Smith     vecs = PETSC_TRUE;
13103bededecSBarry Smith   }
13113bededecSBarry Smith 
13123bededecSBarry Smith   /* zero the columns */
13131795a4d1SJed Brown   ierr = PetscCalloc1(A->rmap->n,&zeroed);CHKERRQ(ierr);
13143bededecSBarry Smith   for (i=0; i<is_n; i++) {
131598921bdaSJacob Faibussowitsch     if (is_idx[i] < 0 || is_idx[i] >= A->rmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"row %" PetscInt_FMT " out of range",is_idx[i]);
13163bededecSBarry Smith     zeroed[is_idx[i]] = PETSC_TRUE;
13173bededecSBarry Smith   }
131856777dd2SBarry Smith   if (vecs) {
131956777dd2SBarry Smith     for (i=0; i<A->rmap->N; i++) {
132056777dd2SBarry Smith       row = i/bs;
132156777dd2SBarry Smith       for (j=baij->i[row]; j<baij->i[row+1]; j++) {
132256777dd2SBarry Smith         for (k=0; k<bs; k++) {
132356777dd2SBarry Smith           col = bs*baij->j[j] + k;
132456777dd2SBarry Smith           if (col <= i) continue;
132556777dd2SBarry Smith           aa = ((MatScalar*)(baij->a)) + j*bs2 + (i%bs) + bs*k;
132626fbe8dcSKarl Rupp           if (!zeroed[i] && zeroed[col]) bb[i]   -= aa[0]*xx[col];
132726fbe8dcSKarl Rupp           if (zeroed[i] && !zeroed[col]) bb[col] -= aa[0]*xx[i];
132856777dd2SBarry Smith         }
132956777dd2SBarry Smith       }
133056777dd2SBarry Smith     }
133126fbe8dcSKarl Rupp     for (i=0; i<is_n; i++) bb[is_idx[i]] = diag*xx[is_idx[i]];
133256777dd2SBarry Smith   }
133356777dd2SBarry Smith 
13343bededecSBarry Smith   for (i=0; i<A->rmap->N; i++) {
13353bededecSBarry Smith     if (!zeroed[i]) {
13363bededecSBarry Smith       row = i/bs;
13373bededecSBarry Smith       for (j=baij->i[row]; j<baij->i[row+1]; j++) {
13383bededecSBarry Smith         for (k=0; k<bs; k++) {
13393bededecSBarry Smith           col = bs*baij->j[j] + k;
13403bededecSBarry Smith           if (zeroed[col]) {
13413bededecSBarry Smith             aa = ((MatScalar*)(baij->a)) + j*bs2 + (i%bs) + bs*k;
13423bededecSBarry Smith             aa[0] = 0.0;
13433bededecSBarry Smith           }
13443bededecSBarry Smith         }
13453bededecSBarry Smith       }
13463bededecSBarry Smith     }
13473bededecSBarry Smith   }
13483bededecSBarry Smith   ierr = PetscFree(zeroed);CHKERRQ(ierr);
134956777dd2SBarry Smith   if (vecs) {
135056777dd2SBarry Smith     ierr = VecRestoreArrayRead(x,&xx);CHKERRQ(ierr);
135156777dd2SBarry Smith     ierr = VecRestoreArray(b,&bb);CHKERRQ(ierr);
135256777dd2SBarry Smith   }
13533bededecSBarry Smith 
13543bededecSBarry Smith   /* zero the rows */
13553bededecSBarry Smith   for (i=0; i<is_n; i++) {
13563bededecSBarry Smith     row   = is_idx[i];
13573bededecSBarry Smith     count = (baij->i[row/bs +1] - baij->i[row/bs])*bs;
13583bededecSBarry Smith     aa    = ((MatScalar*)(baij->a)) + baij->i[row/bs]*bs2 + (row%bs);
13593bededecSBarry Smith     for (k=0; k<count; k++) {
13603bededecSBarry Smith       aa[0] =  zero;
13613bededecSBarry Smith       aa   += bs;
13623bededecSBarry Smith     }
13633bededecSBarry Smith     if (diag != 0.0) {
13643bededecSBarry Smith       ierr = (*A->ops->setvalues)(A,1,&row,1,&row,&diag,INSERT_VALUES);CHKERRQ(ierr);
13653bededecSBarry Smith     }
13663bededecSBarry Smith   }
13673bededecSBarry Smith   ierr = MatAssemblyEnd_SeqSBAIJ(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
13683bededecSBarry Smith   PetscFunctionReturn(0);
13693bededecSBarry Smith }
13703bededecSBarry Smith 
13717d68702bSBarry Smith PetscErrorCode MatShift_SeqSBAIJ(Mat Y,PetscScalar a)
13727d68702bSBarry Smith {
13737d68702bSBarry Smith   PetscErrorCode ierr;
13747d68702bSBarry Smith   Mat_SeqSBAIJ    *aij = (Mat_SeqSBAIJ*)Y->data;
13757d68702bSBarry Smith 
13767d68702bSBarry Smith   PetscFunctionBegin;
13776f33a894SBarry Smith   if (!Y->preallocated || !aij->nz) {
13787d68702bSBarry Smith     ierr = MatSeqSBAIJSetPreallocation(Y,Y->rmap->bs,1,NULL);CHKERRQ(ierr);
13797d68702bSBarry Smith   }
13807d68702bSBarry Smith   ierr = MatShift_Basic(Y,a);CHKERRQ(ierr);
13817d68702bSBarry Smith   PetscFunctionReturn(0);
13827d68702bSBarry Smith }
13837d68702bSBarry Smith 
138449b5e25fSSatish Balay /* -------------------------------------------------------------------*/
13853964eb88SJed Brown static struct _MatOps MatOps_Values = {MatSetValues_SeqSBAIJ,
138649b5e25fSSatish Balay                                        MatGetRow_SeqSBAIJ,
138749b5e25fSSatish Balay                                        MatRestoreRow_SeqSBAIJ,
138849b5e25fSSatish Balay                                        MatMult_SeqSBAIJ_N,
138997304618SKris Buschelman                                /*  4*/ MatMultAdd_SeqSBAIJ_N,
1390431c96f7SBarry Smith                                        MatMult_SeqSBAIJ_N,       /* transpose versions are same as non-transpose versions */
1391e005ede5SBarry Smith                                        MatMultAdd_SeqSBAIJ_N,
1392f4259b30SLisandro Dalcin                                        NULL,
1393f4259b30SLisandro Dalcin                                        NULL,
1394f4259b30SLisandro Dalcin                                        NULL,
1395f4259b30SLisandro Dalcin                                /* 10*/ NULL,
1396f4259b30SLisandro Dalcin                                        NULL,
1397c078aec8SLisandro Dalcin                                        MatCholeskyFactor_SeqSBAIJ,
139841f059aeSBarry Smith                                        MatSOR_SeqSBAIJ,
139949b5e25fSSatish Balay                                        MatTranspose_SeqSBAIJ,
140097304618SKris Buschelman                                /* 15*/ MatGetInfo_SeqSBAIJ,
140149b5e25fSSatish Balay                                        MatEqual_SeqSBAIJ,
140249b5e25fSSatish Balay                                        MatGetDiagonal_SeqSBAIJ,
140349b5e25fSSatish Balay                                        MatDiagonalScale_SeqSBAIJ,
140449b5e25fSSatish Balay                                        MatNorm_SeqSBAIJ,
1405f4259b30SLisandro Dalcin                                /* 20*/ NULL,
140649b5e25fSSatish Balay                                        MatAssemblyEnd_SeqSBAIJ,
140749b5e25fSSatish Balay                                        MatSetOption_SeqSBAIJ,
140849b5e25fSSatish Balay                                        MatZeroEntries_SeqSBAIJ,
1409f4259b30SLisandro Dalcin                                /* 24*/ NULL,
1410f4259b30SLisandro Dalcin                                        NULL,
1411f4259b30SLisandro Dalcin                                        NULL,
1412f4259b30SLisandro Dalcin                                        NULL,
1413f4259b30SLisandro Dalcin                                        NULL,
14144994cf47SJed Brown                                /* 29*/ MatSetUp_SeqSBAIJ,
1415f4259b30SLisandro Dalcin                                        NULL,
1416f4259b30SLisandro Dalcin                                        NULL,
1417f4259b30SLisandro Dalcin                                        NULL,
1418f4259b30SLisandro Dalcin                                        NULL,
1419d519adbfSMatthew Knepley                                /* 34*/ MatDuplicate_SeqSBAIJ,
1420f4259b30SLisandro Dalcin                                        NULL,
1421f4259b30SLisandro Dalcin                                        NULL,
1422f4259b30SLisandro Dalcin                                        NULL,
1423c84f5b01SHong Zhang                                        MatICCFactor_SeqSBAIJ,
1424d519adbfSMatthew Knepley                                /* 39*/ MatAXPY_SeqSBAIJ,
14257dae84e0SHong Zhang                                        MatCreateSubMatrices_SeqSBAIJ,
142649b5e25fSSatish Balay                                        MatIncreaseOverlap_SeqSBAIJ,
142749b5e25fSSatish Balay                                        MatGetValues_SeqSBAIJ,
14283c896bc6SHong Zhang                                        MatCopy_SeqSBAIJ,
1429f4259b30SLisandro Dalcin                                /* 44*/ NULL,
143049b5e25fSSatish Balay                                        MatScale_SeqSBAIJ,
14317d68702bSBarry Smith                                        MatShift_SeqSBAIJ,
1432f4259b30SLisandro Dalcin                                        NULL,
14333bededecSBarry Smith                                        MatZeroRowsColumns_SeqSBAIJ,
1434f4259b30SLisandro Dalcin                                /* 49*/ NULL,
143549b5e25fSSatish Balay                                        MatGetRowIJ_SeqSBAIJ,
143649b5e25fSSatish Balay                                        MatRestoreRowIJ_SeqSBAIJ,
1437f4259b30SLisandro Dalcin                                        NULL,
1438f4259b30SLisandro Dalcin                                        NULL,
1439f4259b30SLisandro Dalcin                                /* 54*/ NULL,
1440f4259b30SLisandro Dalcin                                        NULL,
1441f4259b30SLisandro Dalcin                                        NULL,
1442dc29a518SPierre Jolivet                                        MatPermute_SeqSBAIJ,
144349b5e25fSSatish Balay                                        MatSetValuesBlocked_SeqSBAIJ,
14447dae84e0SHong Zhang                                /* 59*/ MatCreateSubMatrix_SeqSBAIJ,
1445f4259b30SLisandro Dalcin                                        NULL,
1446f4259b30SLisandro Dalcin                                        NULL,
1447f4259b30SLisandro Dalcin                                        NULL,
1448f4259b30SLisandro Dalcin                                        NULL,
1449f4259b30SLisandro Dalcin                                /* 64*/ NULL,
1450f4259b30SLisandro Dalcin                                        NULL,
1451f4259b30SLisandro Dalcin                                        NULL,
1452f4259b30SLisandro Dalcin                                        NULL,
1453f4259b30SLisandro Dalcin                                        NULL,
1454d519adbfSMatthew Knepley                                /* 69*/ MatGetRowMaxAbs_SeqSBAIJ,
1455f4259b30SLisandro Dalcin                                        NULL,
145628d58a37SPierre Jolivet                                        MatConvert_MPISBAIJ_Basic,
1457f4259b30SLisandro Dalcin                                        NULL,
1458f4259b30SLisandro Dalcin                                        NULL,
1459f4259b30SLisandro Dalcin                                /* 74*/ NULL,
1460f4259b30SLisandro Dalcin                                        NULL,
1461f4259b30SLisandro Dalcin                                        NULL,
1462f4259b30SLisandro Dalcin                                        NULL,
1463f4259b30SLisandro Dalcin                                        NULL,
1464f4259b30SLisandro Dalcin                                /* 79*/ NULL,
1465f4259b30SLisandro Dalcin                                        NULL,
1466f4259b30SLisandro Dalcin                                        NULL,
146797304618SKris Buschelman                                        MatGetInertia_SeqSBAIJ,
14685bba2384SShri Abhyankar                                        MatLoad_SeqSBAIJ,
1469d519adbfSMatthew Knepley                                /* 84*/ MatIsSymmetric_SeqSBAIJ,
1470865e5f61SKris Buschelman                                        MatIsHermitian_SeqSBAIJ,
1471efcf0fc3SBarry Smith                                        MatIsStructurallySymmetric_SeqSBAIJ,
1472f4259b30SLisandro Dalcin                                        NULL,
1473f4259b30SLisandro Dalcin                                        NULL,
1474f4259b30SLisandro Dalcin                                /* 89*/ NULL,
1475f4259b30SLisandro Dalcin                                        NULL,
1476f4259b30SLisandro Dalcin                                        NULL,
1477f4259b30SLisandro Dalcin                                        NULL,
1478f4259b30SLisandro Dalcin                                        NULL,
1479f4259b30SLisandro Dalcin                                /* 94*/ NULL,
1480f4259b30SLisandro Dalcin                                        NULL,
1481f4259b30SLisandro Dalcin                                        NULL,
1482f4259b30SLisandro Dalcin                                        NULL,
1483f4259b30SLisandro Dalcin                                        NULL,
1484f4259b30SLisandro Dalcin                                /* 99*/ NULL,
1485f4259b30SLisandro Dalcin                                        NULL,
1486f4259b30SLisandro Dalcin                                        NULL,
14872726fb6dSPierre Jolivet                                        MatConjugate_SeqSBAIJ,
1488f4259b30SLisandro Dalcin                                        NULL,
1489f4259b30SLisandro Dalcin                                /*104*/ NULL,
149099cafbc1SBarry Smith                                        MatRealPart_SeqSBAIJ,
1491f5edf698SHong Zhang                                        MatImaginaryPart_SeqSBAIJ,
1492f5edf698SHong Zhang                                        MatGetRowUpperTriangular_SeqSBAIJ,
14932af78befSBarry Smith                                        MatRestoreRowUpperTriangular_SeqSBAIJ,
1494f4259b30SLisandro Dalcin                                /*109*/ NULL,
1495f4259b30SLisandro Dalcin                                        NULL,
1496f4259b30SLisandro Dalcin                                        NULL,
1497f4259b30SLisandro Dalcin                                        NULL,
1498547795f9SHong Zhang                                        MatMissingDiagonal_SeqSBAIJ,
1499f4259b30SLisandro Dalcin                                /*114*/ NULL,
1500f4259b30SLisandro Dalcin                                        NULL,
1501f4259b30SLisandro Dalcin                                        NULL,
1502f4259b30SLisandro Dalcin                                        NULL,
1503f4259b30SLisandro Dalcin                                        NULL,
1504f4259b30SLisandro Dalcin                                /*119*/ NULL,
1505f4259b30SLisandro Dalcin                                        NULL,
1506f4259b30SLisandro Dalcin                                        NULL,
1507f4259b30SLisandro Dalcin                                        NULL,
1508f4259b30SLisandro Dalcin                                        NULL,
1509f4259b30SLisandro Dalcin                                /*124*/ NULL,
1510f4259b30SLisandro Dalcin                                        NULL,
1511f4259b30SLisandro Dalcin                                        NULL,
1512f4259b30SLisandro Dalcin                                        NULL,
1513f4259b30SLisandro Dalcin                                        NULL,
1514f4259b30SLisandro Dalcin                                /*129*/ NULL,
1515f4259b30SLisandro Dalcin                                        NULL,
1516f4259b30SLisandro Dalcin                                        NULL,
1517f4259b30SLisandro Dalcin                                        NULL,
1518f4259b30SLisandro Dalcin                                        NULL,
1519f4259b30SLisandro Dalcin                                /*134*/ NULL,
1520f4259b30SLisandro Dalcin                                        NULL,
1521f4259b30SLisandro Dalcin                                        NULL,
1522f4259b30SLisandro Dalcin                                        NULL,
1523f4259b30SLisandro Dalcin                                        NULL,
152446533700Sstefano_zampini                                /*139*/ MatSetBlockSizes_Default,
1525f4259b30SLisandro Dalcin                                        NULL,
1526f4259b30SLisandro Dalcin                                        NULL,
1527f4259b30SLisandro Dalcin                                        NULL,
1528f4259b30SLisandro Dalcin                                        NULL,
152959f5e6ceSHong Zhang                                 /*144*/MatCreateMPIMatConcatenateSeqMat_SeqSBAIJ
153099cafbc1SBarry Smith };
1531be1d678aSKris Buschelman 
15327087cfbeSBarry Smith PetscErrorCode  MatStoreValues_SeqSBAIJ(Mat mat)
153349b5e25fSSatish Balay {
15344afc71dfSHong Zhang   Mat_SeqSBAIJ   *aij = (Mat_SeqSBAIJ*)mat->data;
1535d0f46423SBarry Smith   PetscInt       nz   = aij->i[mat->rmap->N]*mat->rmap->bs*aij->bs2;
1536dfbe8321SBarry Smith   PetscErrorCode ierr;
153749b5e25fSSatish Balay 
153849b5e25fSSatish Balay   PetscFunctionBegin;
1539e7e72b3dSBarry Smith   if (aij->nonew != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first");
154049b5e25fSSatish Balay 
154149b5e25fSSatish Balay   /* allocate space for values if not already there */
154249b5e25fSSatish Balay   if (!aij->saved_values) {
1543854ce69bSBarry Smith     ierr = PetscMalloc1(nz+1,&aij->saved_values);CHKERRQ(ierr);
154449b5e25fSSatish Balay   }
154549b5e25fSSatish Balay 
154649b5e25fSSatish Balay   /* copy values over */
1547580bdb30SBarry Smith   ierr = PetscArraycpy(aij->saved_values,aij->a,nz);CHKERRQ(ierr);
154849b5e25fSSatish Balay   PetscFunctionReturn(0);
154949b5e25fSSatish Balay }
155049b5e25fSSatish Balay 
15517087cfbeSBarry Smith PetscErrorCode  MatRetrieveValues_SeqSBAIJ(Mat mat)
155249b5e25fSSatish Balay {
15534afc71dfSHong Zhang   Mat_SeqSBAIJ   *aij = (Mat_SeqSBAIJ*)mat->data;
15546849ba73SBarry Smith   PetscErrorCode ierr;
1555d0f46423SBarry Smith   PetscInt       nz = aij->i[mat->rmap->N]*mat->rmap->bs*aij->bs2;
155649b5e25fSSatish Balay 
155749b5e25fSSatish Balay   PetscFunctionBegin;
1558e7e72b3dSBarry Smith   if (aij->nonew != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first");
1559e7e72b3dSBarry Smith   if (!aij->saved_values) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatStoreValues(A);first");
156049b5e25fSSatish Balay 
156149b5e25fSSatish Balay   /* copy values over */
1562580bdb30SBarry Smith   ierr = PetscArraycpy(aij->a,aij->saved_values,nz);CHKERRQ(ierr);
156349b5e25fSSatish Balay   PetscFunctionReturn(0);
156449b5e25fSSatish Balay }
156549b5e25fSSatish Balay 
1566367daffbSBarry Smith static PetscErrorCode  MatSeqSBAIJSetPreallocation_SeqSBAIJ(Mat B,PetscInt bs,PetscInt nz,PetscInt *nnz)
156749b5e25fSSatish Balay {
1568c464158bSHong Zhang   Mat_SeqSBAIJ   *b = (Mat_SeqSBAIJ*)B->data;
15696849ba73SBarry Smith   PetscErrorCode ierr;
15704dcd73b1SHong Zhang   PetscInt       i,mbs,nbs,bs2;
15712576faa2SJed Brown   PetscBool      skipallocation = PETSC_FALSE,flg = PETSC_FALSE,realalloc = PETSC_FALSE;
157249b5e25fSSatish Balay 
157349b5e25fSSatish Balay   PetscFunctionBegin;
15742576faa2SJed Brown   if (nz >= 0 || nnz) realalloc = PETSC_TRUE;
1575db4efbfdSBarry Smith 
157633d57670SJed Brown   ierr = MatSetBlockSize(B,PetscAbs(bs));CHKERRQ(ierr);
157726283091SBarry Smith   ierr = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr);
157826283091SBarry Smith   ierr = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr);
157998921bdaSJacob Faibussowitsch   if (B->rmap->N > B->cmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"SEQSBAIJ matrix cannot have more rows %" PetscInt_FMT " than columns %" PetscInt_FMT,B->rmap->N,B->cmap->N);
1580e02043d6SBarry Smith   ierr = PetscLayoutGetBlockSize(B->rmap,&bs);CHKERRQ(ierr);
1581899cda47SBarry Smith 
158221940c7eSstefano_zampini   B->preallocated = PETSC_TRUE;
158321940c7eSstefano_zampini 
1584d0f46423SBarry Smith   mbs = B->rmap->N/bs;
15854dcd73b1SHong Zhang   nbs = B->cmap->n/bs;
158649b5e25fSSatish Balay   bs2 = bs*bs;
158749b5e25fSSatish Balay 
15884dcd73b1SHong 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");
158949b5e25fSSatish Balay 
1590ab93d7beSBarry Smith   if (nz == MAT_SKIP_ALLOCATION) {
1591ab93d7beSBarry Smith     skipallocation = PETSC_TRUE;
1592ab93d7beSBarry Smith     nz             = 0;
1593ab93d7beSBarry Smith   }
1594ab93d7beSBarry Smith 
1595435da068SBarry Smith   if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 3;
159698921bdaSJacob Faibussowitsch   if (nz < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"nz cannot be less than 0: value %" PetscInt_FMT,nz);
159749b5e25fSSatish Balay   if (nnz) {
159849b5e25fSSatish Balay     for (i=0; i<mbs; i++) {
159998921bdaSJacob Faibussowitsch       if (nnz[i] < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"nnz cannot be less than 0: local row %" PetscInt_FMT " value %" PetscInt_FMT,i,nnz[i]);
160098921bdaSJacob Faibussowitsch       if (nnz[i] > nbs) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"nnz cannot be greater than block row length: local row %" PetscInt_FMT " value %" PetscInt_FMT " block rowlength %" PetscInt_FMT,i,nnz[i],nbs);
160149b5e25fSSatish Balay     }
160249b5e25fSSatish Balay   }
160349b5e25fSSatish Balay 
1604db4efbfdSBarry Smith   B->ops->mult             = MatMult_SeqSBAIJ_N;
1605db4efbfdSBarry Smith   B->ops->multadd          = MatMultAdd_SeqSBAIJ_N;
1606db4efbfdSBarry Smith   B->ops->multtranspose    = MatMult_SeqSBAIJ_N;
1607db4efbfdSBarry Smith   B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_N;
160826fbe8dcSKarl Rupp 
1609c5929fdfSBarry Smith   ierr  = PetscOptionsGetBool(((PetscObject)B)->options,((PetscObject)B)->prefix,"-mat_no_unroll",&flg,NULL);CHKERRQ(ierr);
161049b5e25fSSatish Balay   if (!flg) {
161149b5e25fSSatish Balay     switch (bs) {
161249b5e25fSSatish Balay     case 1:
161349b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_1;
161449b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_1;
1615431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_1;
1616431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_1;
161749b5e25fSSatish Balay       break;
161849b5e25fSSatish Balay     case 2:
161949b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_2;
162049b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_2;
1621431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_2;
1622431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_2;
162349b5e25fSSatish Balay       break;
162449b5e25fSSatish Balay     case 3:
162549b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_3;
162649b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_3;
1627431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_3;
1628431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_3;
162949b5e25fSSatish Balay       break;
163049b5e25fSSatish Balay     case 4:
163149b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_4;
163249b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_4;
1633431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_4;
1634431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_4;
163549b5e25fSSatish Balay       break;
163649b5e25fSSatish Balay     case 5:
163749b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_5;
163849b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_5;
1639431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_5;
1640431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_5;
164149b5e25fSSatish Balay       break;
164249b5e25fSSatish Balay     case 6:
164349b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_6;
164449b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_6;
1645431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_6;
1646431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_6;
164749b5e25fSSatish Balay       break;
164849b5e25fSSatish Balay     case 7:
1649de53e5efSHong Zhang       B->ops->mult             = MatMult_SeqSBAIJ_7;
165049b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_7;
1651431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_7;
1652431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_7;
165349b5e25fSSatish Balay       break;
165449b5e25fSSatish Balay     }
165549b5e25fSSatish Balay   }
165649b5e25fSSatish Balay 
165749b5e25fSSatish Balay   b->mbs = mbs;
16584dcd73b1SHong Zhang   b->nbs = nbs;
1659ab93d7beSBarry Smith   if (!skipallocation) {
16602ee49352SLisandro Dalcin     if (!b->imax) {
1661dcca6d9dSJed Brown       ierr = PetscMalloc2(mbs,&b->imax,mbs,&b->ilen);CHKERRQ(ierr);
166226fbe8dcSKarl Rupp 
1663c760cd28SBarry Smith       b->free_imax_ilen = PETSC_TRUE;
166426fbe8dcSKarl Rupp 
16653bb1ff40SBarry Smith       ierr = PetscLogObjectMemory((PetscObject)B,2*mbs*sizeof(PetscInt));CHKERRQ(ierr);
16662ee49352SLisandro Dalcin     }
166749b5e25fSSatish Balay     if (!nnz) {
1668435da068SBarry Smith       if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 5;
166949b5e25fSSatish Balay       else if (nz <= 0) nz = 1;
16705d2a9ed1SStefano Zampini       nz = PetscMin(nbs,nz);
167126fbe8dcSKarl Rupp       for (i=0; i<mbs; i++) b->imax[i] = nz;
167261778c46SBarry Smith       ierr = PetscIntMultError(nz,mbs,&nz);CHKERRQ(ierr);
167349b5e25fSSatish Balay     } else {
1674c73702f5SBarry Smith       PetscInt64 nz64 = 0;
1675c73702f5SBarry Smith       for (i=0; i<mbs; i++) {b->imax[i] = nnz[i]; nz64 += nnz[i];}
1676c73702f5SBarry Smith       ierr = PetscIntCast(nz64,&nz);CHKERRQ(ierr);
167749b5e25fSSatish Balay     }
16782ee49352SLisandro Dalcin     /* b->ilen will count nonzeros in each block row so far. */
167926fbe8dcSKarl Rupp     for (i=0; i<mbs; i++) b->ilen[i] = 0;
16806c6c5352SBarry Smith     /* nz=(nz+mbs)/2; */ /* total diagonal and superdiagonal nonzero blocks */
168149b5e25fSSatish Balay 
168249b5e25fSSatish Balay     /* allocate the matrix space */
16832ee49352SLisandro Dalcin     ierr = MatSeqXAIJFreeAIJ(B,&b->a,&b->j,&b->i);CHKERRQ(ierr);
1684dcca6d9dSJed Brown     ierr = PetscMalloc3(bs2*nz,&b->a,nz,&b->j,B->rmap->N+1,&b->i);CHKERRQ(ierr);
16853bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)B,(B->rmap->N+1)*sizeof(PetscInt)+nz*(bs2*sizeof(PetscScalar)+sizeof(PetscInt)));CHKERRQ(ierr);
1686580bdb30SBarry Smith     ierr = PetscArrayzero(b->a,nz*bs2);CHKERRQ(ierr);
1687580bdb30SBarry Smith     ierr = PetscArrayzero(b->j,nz);CHKERRQ(ierr);
168826fbe8dcSKarl Rupp 
168949b5e25fSSatish Balay     b->singlemalloc = PETSC_TRUE;
169049b5e25fSSatish Balay 
169149b5e25fSSatish Balay     /* pointer to beginning of each row */
1692e60cf9a0SBarry Smith     b->i[0] = 0;
169326fbe8dcSKarl Rupp     for (i=1; i<mbs+1; i++) b->i[i] = b->i[i-1] + b->imax[i-1];
169426fbe8dcSKarl Rupp 
1695e6b907acSBarry Smith     b->free_a  = PETSC_TRUE;
1696e6b907acSBarry Smith     b->free_ij = PETSC_TRUE;
1697e811da20SHong Zhang   } else {
1698e6b907acSBarry Smith     b->free_a  = PETSC_FALSE;
1699e6b907acSBarry Smith     b->free_ij = PETSC_FALSE;
1700ab93d7beSBarry Smith   }
170149b5e25fSSatish Balay 
170249b5e25fSSatish Balay   b->bs2     = bs2;
17036c6c5352SBarry Smith   b->nz      = 0;
1704b32cb4a7SJed Brown   b->maxnz   = nz;
1705f4259b30SLisandro Dalcin   b->inew    = NULL;
1706f4259b30SLisandro Dalcin   b->jnew    = NULL;
1707f4259b30SLisandro Dalcin   b->anew    = NULL;
1708f4259b30SLisandro Dalcin   b->a2anew  = NULL;
17091a3463dfSHong Zhang   b->permute = PETSC_FALSE;
1710cb7b82ddSBarry Smith 
1711cb7b82ddSBarry Smith   B->was_assembled = PETSC_FALSE;
1712cb7b82ddSBarry Smith   B->assembled     = PETSC_FALSE;
17132576faa2SJed Brown   if (realalloc) {ierr = MatSetOption(B,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);}
1714c464158bSHong Zhang   PetscFunctionReturn(0);
1715c464158bSHong Zhang }
1716153ea458SHong Zhang 
171738f409ebSLisandro Dalcin PetscErrorCode MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ(Mat B,PetscInt bs,const PetscInt ii[],const PetscInt jj[], const PetscScalar V[])
171838f409ebSLisandro Dalcin {
17190cd7f59aSBarry Smith   PetscInt       i,j,m,nz,anz, nz_max=0,*nnz;
1720f4259b30SLisandro Dalcin   PetscScalar    *values=NULL;
172138f409ebSLisandro Dalcin   PetscBool      roworiented = ((Mat_SeqSBAIJ*)B->data)->roworiented;
172238f409ebSLisandro Dalcin   PetscErrorCode ierr;
17230cd7f59aSBarry Smith 
172438f409ebSLisandro Dalcin   PetscFunctionBegin;
172598921bdaSJacob Faibussowitsch   if (bs < 1) SETERRQ(PetscObjectComm((PetscObject)B),PETSC_ERR_ARG_OUTOFRANGE,"Invalid block size specified, must be positive but it is %" PetscInt_FMT,bs);
172638f409ebSLisandro Dalcin   ierr   = PetscLayoutSetBlockSize(B->rmap,bs);CHKERRQ(ierr);
172738f409ebSLisandro Dalcin   ierr   = PetscLayoutSetBlockSize(B->cmap,bs);CHKERRQ(ierr);
172838f409ebSLisandro Dalcin   ierr   = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr);
172938f409ebSLisandro Dalcin   ierr   = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr);
173038f409ebSLisandro Dalcin   ierr   = PetscLayoutGetBlockSize(B->rmap,&bs);CHKERRQ(ierr);
173138f409ebSLisandro Dalcin   m      = B->rmap->n/bs;
173238f409ebSLisandro Dalcin 
173398921bdaSJacob Faibussowitsch   if (ii[0]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"ii[0] must be 0 but it is %" PetscInt_FMT,ii[0]);
1734854ce69bSBarry Smith   ierr = PetscMalloc1(m+1,&nnz);CHKERRQ(ierr);
173538f409ebSLisandro Dalcin   for (i=0; i<m; i++) {
173638f409ebSLisandro Dalcin     nz = ii[i+1] - ii[i];
173798921bdaSJacob Faibussowitsch     if (nz < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row %" PetscInt_FMT " has a negative number of columns %" PetscInt_FMT,i,nz);
17380cd7f59aSBarry Smith     anz = 0;
17390cd7f59aSBarry Smith     for (j=0; j<nz; j++) {
17400cd7f59aSBarry Smith       /* count only values on the diagonal or above */
17410cd7f59aSBarry Smith       if (jj[ii[i] + j] >= i) {
17420cd7f59aSBarry Smith         anz = nz - j;
17430cd7f59aSBarry Smith         break;
17440cd7f59aSBarry Smith       }
17450cd7f59aSBarry Smith     }
17460cd7f59aSBarry Smith     nz_max = PetscMax(nz_max,anz);
17470cd7f59aSBarry Smith     nnz[i] = anz;
174838f409ebSLisandro Dalcin   }
174938f409ebSLisandro Dalcin   ierr = MatSeqSBAIJSetPreallocation(B,bs,0,nnz);CHKERRQ(ierr);
175038f409ebSLisandro Dalcin   ierr = PetscFree(nnz);CHKERRQ(ierr);
175138f409ebSLisandro Dalcin 
175238f409ebSLisandro Dalcin   values = (PetscScalar*)V;
175338f409ebSLisandro Dalcin   if (!values) {
17541795a4d1SJed Brown     ierr = PetscCalloc1(bs*bs*nz_max,&values);CHKERRQ(ierr);
175538f409ebSLisandro Dalcin   }
175638f409ebSLisandro Dalcin   for (i=0; i<m; i++) {
175738f409ebSLisandro Dalcin     PetscInt          ncols  = ii[i+1] - ii[i];
175838f409ebSLisandro Dalcin     const PetscInt    *icols = jj + ii[i];
175938f409ebSLisandro Dalcin     if (!roworiented || bs == 1) {
176038f409ebSLisandro Dalcin       const PetscScalar *svals = values + (V ? (bs*bs*ii[i]) : 0);
176138f409ebSLisandro Dalcin       ierr = MatSetValuesBlocked_SeqSBAIJ(B,1,&i,ncols,icols,svals,INSERT_VALUES);CHKERRQ(ierr);
176238f409ebSLisandro Dalcin     } else {
176338f409ebSLisandro Dalcin       for (j=0; j<ncols; j++) {
176438f409ebSLisandro Dalcin         const PetscScalar *svals = values + (V ? (bs*bs*(ii[i]+j)) : 0);
176538f409ebSLisandro Dalcin         ierr = MatSetValuesBlocked_SeqSBAIJ(B,1,&i,1,&icols[j],svals,INSERT_VALUES);CHKERRQ(ierr);
176638f409ebSLisandro Dalcin       }
176738f409ebSLisandro Dalcin     }
176838f409ebSLisandro Dalcin   }
176938f409ebSLisandro Dalcin   if (!V) { ierr = PetscFree(values);CHKERRQ(ierr); }
177038f409ebSLisandro Dalcin   ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
177138f409ebSLisandro Dalcin   ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
177238f409ebSLisandro Dalcin   ierr = MatSetOption(B,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
177338f409ebSLisandro Dalcin   PetscFunctionReturn(0);
177438f409ebSLisandro Dalcin }
177538f409ebSLisandro Dalcin 
1776db4efbfdSBarry Smith /*
1777db4efbfdSBarry Smith    This is used to set the numeric factorization for both Cholesky and ICC symbolic factorization
1778db4efbfdSBarry Smith */
1779ace3abfcSBarry Smith PetscErrorCode MatSeqSBAIJSetNumericFactorization_inplace(Mat B,PetscBool natural)
1780db4efbfdSBarry Smith {
1781db4efbfdSBarry Smith   PetscErrorCode ierr;
1782ace3abfcSBarry Smith   PetscBool      flg = PETSC_FALSE;
1783db4efbfdSBarry Smith   PetscInt       bs  = B->rmap->bs;
1784db4efbfdSBarry Smith 
1785db4efbfdSBarry Smith   PetscFunctionBegin;
1786c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(((PetscObject)B)->options,((PetscObject)B)->prefix,"-mat_no_unroll",&flg,NULL);CHKERRQ(ierr);
1787db4efbfdSBarry Smith   if (flg) bs = 8;
1788db4efbfdSBarry Smith 
1789db4efbfdSBarry Smith   if (!natural) {
1790db4efbfdSBarry Smith     switch (bs) {
1791db4efbfdSBarry Smith     case 1:
1792d595f711SHong Zhang       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_inplace;
1793db4efbfdSBarry Smith       break;
1794db4efbfdSBarry Smith     case 2:
1795db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2;
1796db4efbfdSBarry Smith       break;
1797db4efbfdSBarry Smith     case 3:
1798db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3;
1799db4efbfdSBarry Smith       break;
1800db4efbfdSBarry Smith     case 4:
1801db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4;
1802db4efbfdSBarry Smith       break;
1803db4efbfdSBarry Smith     case 5:
1804db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5;
1805db4efbfdSBarry Smith       break;
1806db4efbfdSBarry Smith     case 6:
1807db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6;
1808db4efbfdSBarry Smith       break;
1809db4efbfdSBarry Smith     case 7:
1810db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7;
1811db4efbfdSBarry Smith       break;
1812db4efbfdSBarry Smith     default:
1813db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N;
1814db4efbfdSBarry Smith       break;
1815db4efbfdSBarry Smith     }
1816db4efbfdSBarry Smith   } else {
1817db4efbfdSBarry Smith     switch (bs) {
1818db4efbfdSBarry Smith     case 1:
1819d595f711SHong Zhang       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering_inplace;
1820db4efbfdSBarry Smith       break;
1821db4efbfdSBarry Smith     case 2:
1822db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering;
1823db4efbfdSBarry Smith       break;
1824db4efbfdSBarry Smith     case 3:
1825db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering;
1826db4efbfdSBarry Smith       break;
1827db4efbfdSBarry Smith     case 4:
1828db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering;
1829db4efbfdSBarry Smith       break;
1830db4efbfdSBarry Smith     case 5:
1831db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering;
1832db4efbfdSBarry Smith       break;
1833db4efbfdSBarry Smith     case 6:
1834db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering;
1835db4efbfdSBarry Smith       break;
1836db4efbfdSBarry Smith     case 7:
1837db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering;
1838db4efbfdSBarry Smith       break;
1839db4efbfdSBarry Smith     default:
1840db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering;
1841db4efbfdSBarry Smith       break;
1842db4efbfdSBarry Smith     }
1843db4efbfdSBarry Smith   }
1844db4efbfdSBarry Smith   PetscFunctionReturn(0);
1845db4efbfdSBarry Smith }
1846db4efbfdSBarry Smith 
1847cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqAIJ(Mat,MatType,MatReuse,Mat*);
1848cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqBAIJ(Mat,MatType,MatReuse,Mat*);
18494ac6704cSBarry Smith static PetscErrorCode MatFactorGetSolverType_petsc(Mat A,MatSolverType *type)
18504ac6704cSBarry Smith {
18514ac6704cSBarry Smith   PetscFunctionBegin;
18524ac6704cSBarry Smith   *type = MATSOLVERPETSC;
18534ac6704cSBarry Smith   PetscFunctionReturn(0);
18544ac6704cSBarry Smith }
1855d769727bSBarry Smith 
1856cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatGetFactor_seqsbaij_petsc(Mat A,MatFactorType ftype,Mat *B)
18575c9eb25fSBarry Smith {
1858d0f46423SBarry Smith   PetscInt       n = A->rmap->n;
18595c9eb25fSBarry Smith   PetscErrorCode ierr;
18605c9eb25fSBarry Smith 
18615c9eb25fSBarry Smith   PetscFunctionBegin;
18620e92d65fSHong Zhang #if defined(PETSC_USE_COMPLEX)
1863eb1ec7c1SStefano Zampini   if (A->hermitian && !A->symmetric && (ftype == MAT_FACTOR_CHOLESKY||ftype == MAT_FACTOR_ICC)) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Hermitian CHOLESKY or ICC Factor is not supported");
18640e92d65fSHong Zhang #endif
1865eb1ec7c1SStefano Zampini 
1866ce94432eSBarry Smith   ierr = MatCreate(PetscObjectComm((PetscObject)A),B);CHKERRQ(ierr);
18675c9eb25fSBarry Smith   ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr);
18685c9eb25fSBarry Smith   if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) {
18695c9eb25fSBarry Smith     ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr);
18700298fd71SBarry Smith     ierr = MatSeqSBAIJSetPreallocation(*B,A->rmap->bs,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr);
187126fbe8dcSKarl Rupp 
18727b056e98SHong Zhang     (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqSBAIJ;
1873c6d0d4f0SHong Zhang     (*B)->ops->iccfactorsymbolic      = MatICCFactorSymbolic_SeqSBAIJ;
18744ac6704cSBarry Smith     ierr = PetscStrallocpy(MATORDERINGNATURAL,(char**)&(*B)->preferredordering[MAT_FACTOR_CHOLESKY]);CHKERRQ(ierr);
18754ac6704cSBarry Smith     ierr = PetscStrallocpy(MATORDERINGNATURAL,(char**)&(*B)->preferredordering[MAT_FACTOR_ICC]);CHKERRQ(ierr);
1876e32f2f54SBarry Smith   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Factor type not supported");
187700c67f3bSHong Zhang 
1878d5f3da31SBarry Smith   (*B)->factortype = ftype;
1879f73b0415SBarry Smith   (*B)->canuseordering = PETSC_TRUE;
188000c67f3bSHong Zhang   ierr = PetscFree((*B)->solvertype);CHKERRQ(ierr);
188100c67f3bSHong Zhang   ierr = PetscStrallocpy(MATSOLVERPETSC,&(*B)->solvertype);CHKERRQ(ierr);
18824ac6704cSBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)*B,"MatFactorGetSolverType_C",MatFactorGetSolverType_petsc);CHKERRQ(ierr);
18835c9eb25fSBarry Smith   PetscFunctionReturn(0);
18845c9eb25fSBarry Smith }
18855c9eb25fSBarry Smith 
18868397e458SBarry Smith /*@C
18878397e458SBarry Smith    MatSeqSBAIJGetArray - gives access to the array where the data for a MATSEQSBAIJ matrix is stored
18888397e458SBarry Smith 
18898397e458SBarry Smith    Not Collective
18908397e458SBarry Smith 
18918397e458SBarry Smith    Input Parameter:
18928397e458SBarry Smith .  mat - a MATSEQSBAIJ matrix
18938397e458SBarry Smith 
18948397e458SBarry Smith    Output Parameter:
18958397e458SBarry Smith .   array - pointer to the data
18968397e458SBarry Smith 
18978397e458SBarry Smith    Level: intermediate
18988397e458SBarry Smith 
18998397e458SBarry Smith .seealso: MatSeqSBAIJRestoreArray(), MatSeqAIJGetArray(), MatSeqAIJRestoreArray()
19008397e458SBarry Smith @*/
19018397e458SBarry Smith PetscErrorCode  MatSeqSBAIJGetArray(Mat A,PetscScalar **array)
19028397e458SBarry Smith {
19038397e458SBarry Smith   PetscErrorCode ierr;
19048397e458SBarry Smith 
19058397e458SBarry Smith   PetscFunctionBegin;
19068397e458SBarry Smith   ierr = PetscUseMethod(A,"MatSeqSBAIJGetArray_C",(Mat,PetscScalar**),(A,array));CHKERRQ(ierr);
19078397e458SBarry Smith   PetscFunctionReturn(0);
19088397e458SBarry Smith }
19098397e458SBarry Smith 
19108397e458SBarry Smith /*@C
19118397e458SBarry Smith    MatSeqSBAIJRestoreArray - returns access to the array where the data for a MATSEQSBAIJ matrix is stored obtained by MatSeqSBAIJGetArray()
19128397e458SBarry Smith 
19138397e458SBarry Smith    Not Collective
19148397e458SBarry Smith 
19158397e458SBarry Smith    Input Parameters:
1916a2b725a8SWilliam Gropp +  mat - a MATSEQSBAIJ matrix
1917a2b725a8SWilliam Gropp -  array - pointer to the data
19188397e458SBarry Smith 
19198397e458SBarry Smith    Level: intermediate
19208397e458SBarry Smith 
19218397e458SBarry Smith .seealso: MatSeqSBAIJGetArray(), MatSeqAIJGetArray(), MatSeqAIJRestoreArray()
19228397e458SBarry Smith @*/
19238397e458SBarry Smith PetscErrorCode  MatSeqSBAIJRestoreArray(Mat A,PetscScalar **array)
19248397e458SBarry Smith {
19258397e458SBarry Smith   PetscErrorCode ierr;
19268397e458SBarry Smith 
19278397e458SBarry Smith   PetscFunctionBegin;
19288397e458SBarry Smith   ierr = PetscUseMethod(A,"MatSeqSBAIJRestoreArray_C",(Mat,PetscScalar**),(A,array));CHKERRQ(ierr);
19298397e458SBarry Smith   PetscFunctionReturn(0);
19308397e458SBarry Smith }
19318397e458SBarry Smith 
19320bad9183SKris Buschelman /*MC
1933fafad747SKris Buschelman   MATSEQSBAIJ - MATSEQSBAIJ = "seqsbaij" - A matrix type to be used for sequential symmetric block sparse matrices,
19340bad9183SKris Buschelman   based on block compressed sparse row format.  Only the upper triangular portion of the matrix is stored.
19350bad9183SKris Buschelman 
1936828413b8SBarry Smith   For complex numbers by default this matrix is symmetric, NOT Hermitian symmetric. To make it Hermitian symmetric you
1937eb1ec7c1SStefano Zampini   can call MatSetOption(Mat, MAT_HERMITIAN).
1938828413b8SBarry Smith 
19390bad9183SKris Buschelman   Options Database Keys:
19400bad9183SKris Buschelman   . -mat_type seqsbaij - sets the matrix type to "seqsbaij" during a call to MatSetFromOptions()
19410bad9183SKris Buschelman 
194295452b02SPatrick Sanan   Notes:
194395452b02SPatrick Sanan     By default if you insert values into the lower triangular part of the matrix they are simply ignored (since they are not
194471dad5bbSBarry 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
194571dad5bbSBarry 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.
194671dad5bbSBarry Smith 
1947476417e5SBarry Smith     The number of rows in the matrix must be less than or equal to the number of columns
194871dad5bbSBarry Smith 
19490bad9183SKris Buschelman   Level: beginner
19500bad9183SKris Buschelman 
1951476417e5SBarry Smith   .seealso: MatCreateSeqSBAIJ(), MatType, MATMPISBAIJ
19520bad9183SKris Buschelman M*/
19538cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatCreate_SeqSBAIJ(Mat B)
1954a23d5eceSKris Buschelman {
1955a23d5eceSKris Buschelman   Mat_SeqSBAIJ   *b;
1956dfbe8321SBarry Smith   PetscErrorCode ierr;
195713f74950SBarry Smith   PetscMPIInt    size;
1958ace3abfcSBarry Smith   PetscBool      no_unroll = PETSC_FALSE,no_inode = PETSC_FALSE;
1959a23d5eceSKris Buschelman 
1960a23d5eceSKris Buschelman   PetscFunctionBegin;
1961ffc4695bSBarry Smith   ierr = MPI_Comm_size(PetscObjectComm((PetscObject)B),&size);CHKERRMPI(ierr);
1962e32f2f54SBarry Smith   if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Comm must be of size 1");
1963a23d5eceSKris Buschelman 
1964b00a9115SJed Brown   ierr    = PetscNewLog(B,&b);CHKERRQ(ierr);
1965a23d5eceSKris Buschelman   B->data = (void*)b;
1966a23d5eceSKris Buschelman   ierr    = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr);
196726fbe8dcSKarl Rupp 
1968a23d5eceSKris Buschelman   B->ops->destroy    = MatDestroy_SeqSBAIJ;
1969a23d5eceSKris Buschelman   B->ops->view       = MatView_SeqSBAIJ;
1970f4259b30SLisandro Dalcin   b->row             = NULL;
1971f4259b30SLisandro Dalcin   b->icol            = NULL;
1972a23d5eceSKris Buschelman   b->reallocs        = 0;
1973f4259b30SLisandro Dalcin   b->saved_values    = NULL;
19740def2e27SBarry Smith   b->inode.limit     = 5;
19750def2e27SBarry Smith   b->inode.max_limit = 5;
1976a23d5eceSKris Buschelman 
1977a23d5eceSKris Buschelman   b->roworiented        = PETSC_TRUE;
1978a23d5eceSKris Buschelman   b->nonew              = 0;
1979f4259b30SLisandro Dalcin   b->diag               = NULL;
1980f4259b30SLisandro Dalcin   b->solve_work         = NULL;
1981f4259b30SLisandro Dalcin   b->mult_work          = NULL;
1982f4259b30SLisandro Dalcin   B->spptr              = NULL;
1983f2cbd3d5SJed Brown   B->info.nz_unneeded   = (PetscReal)b->maxnz*b->bs2;
1984a9817697SBarry Smith   b->keepnonzeropattern = PETSC_FALSE;
1985a23d5eceSKris Buschelman 
1986f4259b30SLisandro Dalcin   b->inew    = NULL;
1987f4259b30SLisandro Dalcin   b->jnew    = NULL;
1988f4259b30SLisandro Dalcin   b->anew    = NULL;
1989f4259b30SLisandro Dalcin   b->a2anew  = NULL;
1990a23d5eceSKris Buschelman   b->permute = PETSC_FALSE;
1991a23d5eceSKris Buschelman 
199271dad5bbSBarry Smith   b->ignore_ltriangular = PETSC_TRUE;
199326fbe8dcSKarl Rupp 
1994c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(((PetscObject)B)->options,((PetscObject)B)->prefix,"-mat_ignore_lower_triangular",&b->ignore_ltriangular,NULL);CHKERRQ(ierr);
1995941593c8SHong Zhang 
1996f5edf698SHong Zhang   b->getrow_utriangular = PETSC_FALSE;
199726fbe8dcSKarl Rupp 
1998c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(((PetscObject)B)->options,((PetscObject)B)->prefix,"-mat_getrow_uppertriangular",&b->getrow_utriangular,NULL);CHKERRQ(ierr);
1999f5edf698SHong Zhang 
20008397e458SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJGetArray_C",MatSeqSBAIJGetArray_SeqSBAIJ);CHKERRQ(ierr);
20018397e458SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJRestoreArray_C",MatSeqSBAIJRestoreArray_SeqSBAIJ);CHKERRQ(ierr);
2002bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatStoreValues_C",MatStoreValues_SeqSBAIJ);CHKERRQ(ierr);
2003bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatRetrieveValues_C",MatRetrieveValues_SeqSBAIJ);CHKERRQ(ierr);
2004bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetColumnIndices_C",MatSeqSBAIJSetColumnIndices_SeqSBAIJ);CHKERRQ(ierr);
2005bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqaij_C",MatConvert_SeqSBAIJ_SeqAIJ);CHKERRQ(ierr);
2006bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqbaij_C",MatConvert_SeqSBAIJ_SeqBAIJ);CHKERRQ(ierr);
2007bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetPreallocation_C",MatSeqSBAIJSetPreallocation_SeqSBAIJ);CHKERRQ(ierr);
200838f409ebSLisandro Dalcin   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetPreallocationCSR_C",MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ);CHKERRQ(ierr);
20096214f412SHong Zhang #if defined(PETSC_HAVE_ELEMENTAL)
20106214f412SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_elemental_C",MatConvert_SeqSBAIJ_Elemental);CHKERRQ(ierr);
20116214f412SHong Zhang #endif
2012d24d4204SJose E. Roman #if defined(PETSC_HAVE_SCALAPACK)
2013d24d4204SJose E. Roman   ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_scalapack_C",MatConvert_SBAIJ_ScaLAPACK);CHKERRQ(ierr);
2014d24d4204SJose E. Roman #endif
201523ce1328SBarry Smith 
201623ce1328SBarry Smith   B->symmetric                  = PETSC_TRUE;
201723ce1328SBarry Smith   B->structurally_symmetric     = PETSC_TRUE;
201823ce1328SBarry Smith   B->symmetric_set              = PETSC_TRUE;
201923ce1328SBarry Smith   B->structurally_symmetric_set = PETSC_TRUE;
20209899f194SHong Zhang   B->symmetric_eternal          = PETSC_TRUE;
2021eb1ec7c1SStefano Zampini #if defined(PETSC_USE_COMPLEX)
202213647f61SHong Zhang   B->hermitian                  = PETSC_FALSE;
202313647f61SHong Zhang   B->hermitian_set              = PETSC_FALSE;
2024eb1ec7c1SStefano Zampini #else
2025eb1ec7c1SStefano Zampini   B->hermitian                  = PETSC_TRUE;
2026eb1ec7c1SStefano Zampini   B->hermitian_set              = PETSC_TRUE;
2027eb1ec7c1SStefano Zampini #endif
202813647f61SHong Zhang 
202917667f90SBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)B,MATSEQSBAIJ);CHKERRQ(ierr);
20300def2e27SBarry Smith 
2031ce94432eSBarry Smith   ierr = PetscOptionsBegin(PetscObjectComm((PetscObject)B),((PetscObject)B)->prefix,"Options for SEQSBAIJ matrix","Mat");CHKERRQ(ierr);
20320298fd71SBarry Smith   ierr = PetscOptionsBool("-mat_no_unroll","Do not optimize for inodes (slower)",NULL,no_unroll,&no_unroll,NULL);CHKERRQ(ierr);
203326fbe8dcSKarl Rupp   if (no_unroll) {
203426fbe8dcSKarl Rupp     ierr = PetscInfo(B,"Not using Inode routines due to -mat_no_unroll\n");CHKERRQ(ierr);
203526fbe8dcSKarl Rupp   }
20360298fd71SBarry Smith   ierr = PetscOptionsBool("-mat_no_inode","Do not optimize for inodes (slower)",NULL,no_inode,&no_inode,NULL);CHKERRQ(ierr);
203726fbe8dcSKarl Rupp   if (no_inode) {
203826fbe8dcSKarl Rupp     ierr = PetscInfo(B,"Not using Inode routines due to -mat_no_inode\n");CHKERRQ(ierr);
203926fbe8dcSKarl Rupp   }
20400298fd71SBarry Smith   ierr = PetscOptionsInt("-mat_inode_limit","Do not use inodes larger then this value",NULL,b->inode.limit,&b->inode.limit,NULL);CHKERRQ(ierr);
20410def2e27SBarry Smith   ierr = PetscOptionsEnd();CHKERRQ(ierr);
2042ace3abfcSBarry Smith   b->inode.use = (PetscBool)(!(no_unroll || no_inode));
20430def2e27SBarry Smith   if (b->inode.limit > b->inode.max_limit) b->inode.limit = b->inode.max_limit;
2044a23d5eceSKris Buschelman   PetscFunctionReturn(0);
2045a23d5eceSKris Buschelman }
2046a23d5eceSKris Buschelman 
2047a23d5eceSKris Buschelman /*@C
2048a23d5eceSKris Buschelman    MatSeqSBAIJSetPreallocation - Creates a sparse symmetric matrix in block AIJ (block
2049a23d5eceSKris Buschelman    compressed row) format.  For good matrix assembly performance the
2050a23d5eceSKris Buschelman    user should preallocate the matrix storage by setting the parameter nz
2051a23d5eceSKris Buschelman    (or the array nnz).  By setting these parameters accurately, performance
2052a23d5eceSKris Buschelman    during matrix assembly can be increased by more than a factor of 50.
2053a23d5eceSKris Buschelman 
2054a23d5eceSKris Buschelman    Collective on Mat
2055a23d5eceSKris Buschelman 
2056a23d5eceSKris Buschelman    Input Parameters:
20571c4f3114SJed Brown +  B - the symmetric matrix
2058bb7ae925SBarry 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
2059bb7ae925SBarry Smith           blocksize always defines the size of the blocks. The column blocksize sets the blocksize of the vectors obtained with MatCreateVecs()
2060a23d5eceSKris Buschelman .  nz - number of block nonzeros per block row (same for all rows)
2061a23d5eceSKris Buschelman -  nnz - array containing the number of block nonzeros in the upper triangular plus
20620298fd71SBarry Smith          diagonal portion of each block (possibly different for each block row) or NULL
2063a23d5eceSKris Buschelman 
2064a23d5eceSKris Buschelman    Options Database Keys:
2065a2b725a8SWilliam Gropp +   -mat_no_unroll - uses code that does not unroll the loops in the
2066a23d5eceSKris Buschelman                      block calculations (much slower)
2067a2b725a8SWilliam Gropp -   -mat_block_size - size of the blocks to use (only works if a negative bs is passed in
2068a23d5eceSKris Buschelman 
2069a23d5eceSKris Buschelman    Level: intermediate
2070a23d5eceSKris Buschelman 
2071a23d5eceSKris Buschelman    Notes:
2072a23d5eceSKris Buschelman    Specify the preallocated storage with either nz or nnz (not both).
20730298fd71SBarry Smith    Set nz=PETSC_DEFAULT and nnz=NULL for PETSc to control dynamic memory
2074a7f22e61SSatish Balay    allocation.  See Users-Manual: ch_mat for details.
2075a23d5eceSKris Buschelman 
2076aa95bbe8SBarry Smith    You can call MatGetInfo() to get information on how effective the preallocation was;
2077aa95bbe8SBarry Smith    for example the fields mallocs,nz_allocated,nz_used,nz_unneeded;
2078aa95bbe8SBarry Smith    You can also run with the option -info and look for messages with the string
2079aa95bbe8SBarry Smith    malloc in them to see if additional memory allocation was needed.
2080aa95bbe8SBarry Smith 
208149a6f317SBarry Smith    If the nnz parameter is given then the nz parameter is ignored
208249a6f317SBarry Smith 
208369b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateSBAIJ()
2084a23d5eceSKris Buschelman @*/
20857087cfbeSBarry Smith PetscErrorCode  MatSeqSBAIJSetPreallocation(Mat B,PetscInt bs,PetscInt nz,const PetscInt nnz[])
208613f74950SBarry Smith {
20874ac538c5SBarry Smith   PetscErrorCode ierr;
2088a23d5eceSKris Buschelman 
2089a23d5eceSKris Buschelman   PetscFunctionBegin;
20906ba663aaSJed Brown   PetscValidHeaderSpecific(B,MAT_CLASSID,1);
20916ba663aaSJed Brown   PetscValidType(B,1);
20926ba663aaSJed Brown   PetscValidLogicalCollectiveInt(B,bs,2);
20934ac538c5SBarry Smith   ierr = PetscTryMethod(B,"MatSeqSBAIJSetPreallocation_C",(Mat,PetscInt,PetscInt,const PetscInt[]),(B,bs,nz,nnz));CHKERRQ(ierr);
2094a23d5eceSKris Buschelman   PetscFunctionReturn(0);
2095a23d5eceSKris Buschelman }
209649b5e25fSSatish Balay 
209738f409ebSLisandro Dalcin /*@C
2098664954b6SBarry Smith    MatSeqSBAIJSetPreallocationCSR - Creates a sparse parallel matrix in SBAIJ format using the given nonzero structure and (optional) numerical values
209938f409ebSLisandro Dalcin 
210038f409ebSLisandro Dalcin    Input Parameters:
21011c4f3114SJed Brown +  B - the matrix
2102eab78319SHong Zhang .  bs - size of block, the blocks are ALWAYS square.
210338f409ebSLisandro Dalcin .  i - the indices into j for the start of each local row (starts with zero)
210438f409ebSLisandro Dalcin .  j - the column indices for each local row (starts with zero) these must be sorted for each row
210538f409ebSLisandro Dalcin -  v - optional values in the matrix
210638f409ebSLisandro Dalcin 
2107664954b6SBarry Smith    Level: advanced
210838f409ebSLisandro Dalcin 
210938f409ebSLisandro Dalcin    Notes:
211038f409ebSLisandro Dalcin    The order of the entries in values is specified by the MatOption MAT_ROW_ORIENTED.  For example, C programs
211138f409ebSLisandro 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
211238f409ebSLisandro Dalcin    over rows within a block and the last index is over columns within a block row.  Fortran programs will likely set
211338f409ebSLisandro 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
211438f409ebSLisandro Dalcin    block column and the second index is over columns within a block.
211538f409ebSLisandro Dalcin 
211650c5228eSBarry Smith    Any entries below the diagonal are ignored
21170cd7f59aSBarry Smith 
21180cd7f59aSBarry Smith    Though this routine has Preallocation() in the name it also sets the exact nonzero locations of the matrix entries
21190cd7f59aSBarry Smith    and usually the numerical values as well
2120664954b6SBarry Smith 
212138f409ebSLisandro Dalcin .seealso: MatCreate(), MatCreateSeqSBAIJ(), MatSetValuesBlocked(), MatSeqSBAIJSetPreallocation(), MATSEQSBAIJ
212238f409ebSLisandro Dalcin @*/
212338f409ebSLisandro Dalcin PetscErrorCode MatSeqSBAIJSetPreallocationCSR(Mat B,PetscInt bs,const PetscInt i[],const PetscInt j[], const PetscScalar v[])
212438f409ebSLisandro Dalcin {
212538f409ebSLisandro Dalcin   PetscErrorCode ierr;
212638f409ebSLisandro Dalcin 
212738f409ebSLisandro Dalcin   PetscFunctionBegin;
212838f409ebSLisandro Dalcin   PetscValidHeaderSpecific(B,MAT_CLASSID,1);
212938f409ebSLisandro Dalcin   PetscValidType(B,1);
213038f409ebSLisandro Dalcin   PetscValidLogicalCollectiveInt(B,bs,2);
213138f409ebSLisandro Dalcin   ierr = PetscTryMethod(B,"MatSeqSBAIJSetPreallocationCSR_C",(Mat,PetscInt,const PetscInt[],const PetscInt[],const PetscScalar[]),(B,bs,i,j,v));CHKERRQ(ierr);
213238f409ebSLisandro Dalcin   PetscFunctionReturn(0);
213338f409ebSLisandro Dalcin }
213438f409ebSLisandro Dalcin 
2135c464158bSHong Zhang /*@C
2136c464158bSHong Zhang    MatCreateSeqSBAIJ - Creates a sparse symmetric matrix in block AIJ (block
2137c464158bSHong Zhang    compressed row) format.  For good matrix assembly performance the
2138c464158bSHong Zhang    user should preallocate the matrix storage by setting the parameter nz
2139c464158bSHong Zhang    (or the array nnz).  By setting these parameters accurately, performance
2140c464158bSHong Zhang    during matrix assembly can be increased by more than a factor of 50.
214149b5e25fSSatish Balay 
2142d083f849SBarry Smith    Collective
2143c464158bSHong Zhang 
2144c464158bSHong Zhang    Input Parameters:
2145c464158bSHong Zhang +  comm - MPI communicator, set to PETSC_COMM_SELF
2146bb7ae925SBarry 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
2147bb7ae925SBarry Smith           blocksize always defines the size of the blocks. The column blocksize sets the blocksize of the vectors obtained with MatCreateVecs()
2148c464158bSHong Zhang .  m - number of rows, or number of columns
2149c464158bSHong Zhang .  nz - number of block nonzeros per block row (same for all rows)
2150744e8345SSatish Balay -  nnz - array containing the number of block nonzeros in the upper triangular plus
21510298fd71SBarry Smith          diagonal portion of each block (possibly different for each block row) or NULL
2152c464158bSHong Zhang 
2153c464158bSHong Zhang    Output Parameter:
2154c464158bSHong Zhang .  A - the symmetric matrix
2155c464158bSHong Zhang 
2156c464158bSHong Zhang    Options Database Keys:
2157a2b725a8SWilliam Gropp +   -mat_no_unroll - uses code that does not unroll the loops in the
2158c464158bSHong Zhang                      block calculations (much slower)
2159a2b725a8SWilliam Gropp -   -mat_block_size - size of the blocks to use
2160c464158bSHong Zhang 
2161c464158bSHong Zhang    Level: intermediate
2162c464158bSHong Zhang 
2163175b88e8SBarry Smith    It is recommended that one use the MatCreate(), MatSetType() and/or MatSetFromOptions(),
2164f6f02116SRichard Tran Mills    MatXXXXSetPreallocation() paradigm instead of this routine directly.
2165175b88e8SBarry Smith    [MatXXXXSetPreallocation() is, for example, MatSeqAIJSetPreallocation]
2166175b88e8SBarry Smith 
2167c464158bSHong Zhang    Notes:
21686d6d819aSHong Zhang    The number of rows and columns must be divisible by blocksize.
21696d6d819aSHong Zhang    This matrix type does not support complex Hermitian operation.
2170c464158bSHong Zhang 
2171c464158bSHong Zhang    Specify the preallocated storage with either nz or nnz (not both).
21720298fd71SBarry Smith    Set nz=PETSC_DEFAULT and nnz=NULL for PETSc to control dynamic memory
2173a7f22e61SSatish Balay    allocation.  See Users-Manual: ch_mat for details.
2174c464158bSHong Zhang 
217549a6f317SBarry Smith    If the nnz parameter is given then the nz parameter is ignored
217649a6f317SBarry Smith 
217769b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateSBAIJ()
2178c464158bSHong Zhang @*/
21797087cfbeSBarry Smith PetscErrorCode  MatCreateSeqSBAIJ(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt nz,const PetscInt nnz[],Mat *A)
2180c464158bSHong Zhang {
2181dfbe8321SBarry Smith   PetscErrorCode ierr;
2182c464158bSHong Zhang 
2183c464158bSHong Zhang   PetscFunctionBegin;
2184f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,A);CHKERRQ(ierr);
2185f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*A,m,n,m,n);CHKERRQ(ierr);
2186c464158bSHong Zhang   ierr = MatSetType(*A,MATSEQSBAIJ);CHKERRQ(ierr);
2187367daffbSBarry Smith   ierr = MatSeqSBAIJSetPreallocation(*A,bs,nz,(PetscInt*)nnz);CHKERRQ(ierr);
218849b5e25fSSatish Balay   PetscFunctionReturn(0);
218949b5e25fSSatish Balay }
219049b5e25fSSatish Balay 
2191dfbe8321SBarry Smith PetscErrorCode MatDuplicate_SeqSBAIJ(Mat A,MatDuplicateOption cpvalues,Mat *B)
219249b5e25fSSatish Balay {
219349b5e25fSSatish Balay   Mat            C;
219449b5e25fSSatish Balay   Mat_SeqSBAIJ   *c,*a = (Mat_SeqSBAIJ*)A->data;
21956849ba73SBarry Smith   PetscErrorCode ierr;
2196b40805acSSatish Balay   PetscInt       i,mbs = a->mbs,nz = a->nz,bs2 =a->bs2;
219749b5e25fSSatish Balay 
219849b5e25fSSatish Balay   PetscFunctionBegin;
2199e32f2f54SBarry Smith   if (a->i[mbs] != nz) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Corrupt matrix");
220049b5e25fSSatish Balay 
2201f4259b30SLisandro Dalcin   *B   = NULL;
2202ce94432eSBarry Smith   ierr = MatCreate(PetscObjectComm((PetscObject)A),&C);CHKERRQ(ierr);
2203d0f46423SBarry Smith   ierr = MatSetSizes(C,A->rmap->N,A->cmap->n,A->rmap->N,A->cmap->n);CHKERRQ(ierr);
22044c7a3774SStefano Zampini   ierr = MatSetBlockSizesFromMats(C,A,A);CHKERRQ(ierr);
22058e9a0fb8SHong Zhang   ierr = MatSetType(C,MATSEQSBAIJ);CHKERRQ(ierr);
2206692f9cbeSHong Zhang   c    = (Mat_SeqSBAIJ*)C->data;
2207692f9cbeSHong Zhang 
2208273d9f13SBarry Smith   C->preallocated       = PETSC_TRUE;
2209d5f3da31SBarry Smith   C->factortype         = A->factortype;
2210f4259b30SLisandro Dalcin   c->row                = NULL;
2211f4259b30SLisandro Dalcin   c->icol               = NULL;
2212f4259b30SLisandro Dalcin   c->saved_values       = NULL;
2213a9817697SBarry Smith   c->keepnonzeropattern = a->keepnonzeropattern;
221449b5e25fSSatish Balay   C->assembled          = PETSC_TRUE;
221549b5e25fSSatish Balay 
22161e1e43feSBarry Smith   ierr   = PetscLayoutReference(A->rmap,&C->rmap);CHKERRQ(ierr);
22171e1e43feSBarry Smith   ierr   = PetscLayoutReference(A->cmap,&C->cmap);CHKERRQ(ierr);
221849b5e25fSSatish Balay   c->bs2 = a->bs2;
221949b5e25fSSatish Balay   c->mbs = a->mbs;
222049b5e25fSSatish Balay   c->nbs = a->nbs;
222149b5e25fSSatish Balay 
2222c760cd28SBarry Smith   if (cpvalues == MAT_SHARE_NONZERO_PATTERN) {
2223c760cd28SBarry Smith     c->imax           = a->imax;
2224c760cd28SBarry Smith     c->ilen           = a->ilen;
2225c760cd28SBarry Smith     c->free_imax_ilen = PETSC_FALSE;
2226c760cd28SBarry Smith   } else {
2227dcca6d9dSJed Brown     ierr = PetscMalloc2((mbs+1),&c->imax,(mbs+1),&c->ilen);CHKERRQ(ierr);
22283bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)C,2*(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr);
222949b5e25fSSatish Balay     for (i=0; i<mbs; i++) {
223049b5e25fSSatish Balay       c->imax[i] = a->imax[i];
223149b5e25fSSatish Balay       c->ilen[i] = a->ilen[i];
223249b5e25fSSatish Balay     }
2233c760cd28SBarry Smith     c->free_imax_ilen = PETSC_TRUE;
2234c760cd28SBarry Smith   }
223549b5e25fSSatish Balay 
223649b5e25fSSatish Balay   /* allocate the matrix space */
22374da8f245SBarry Smith   if (cpvalues == MAT_SHARE_NONZERO_PATTERN) {
2238785e854fSJed Brown     ierr            = PetscMalloc1(bs2*nz,&c->a);CHKERRQ(ierr);
22393bb1ff40SBarry Smith     ierr            = PetscLogObjectMemory((PetscObject)C,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr);
224044e1c64aSLisandro Dalcin     c->i            = a->i;
224144e1c64aSLisandro Dalcin     c->j            = a->j;
22424da8f245SBarry Smith     c->singlemalloc = PETSC_FALSE;
224344e1c64aSLisandro Dalcin     c->free_a       = PETSC_TRUE;
22444da8f245SBarry Smith     c->free_ij      = PETSC_FALSE;
22454da8f245SBarry Smith     c->parent       = A;
22464da8f245SBarry Smith     ierr            = PetscObjectReference((PetscObject)A);CHKERRQ(ierr);
22474da8f245SBarry Smith     ierr            = MatSetOption(A,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
22484da8f245SBarry Smith     ierr            = MatSetOption(C,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
22494da8f245SBarry Smith   } else {
2250dcca6d9dSJed Brown     ierr            = PetscMalloc3(bs2*nz,&c->a,nz,&c->j,mbs+1,&c->i);CHKERRQ(ierr);
2251580bdb30SBarry Smith     ierr            = PetscArraycpy(c->i,a->i,mbs+1);CHKERRQ(ierr);
22523bb1ff40SBarry Smith     ierr            = PetscLogObjectMemory((PetscObject)C,(mbs+1)*sizeof(PetscInt) + nz*(bs2*sizeof(MatScalar) + sizeof(PetscInt)));CHKERRQ(ierr);
22534da8f245SBarry Smith     c->singlemalloc = PETSC_TRUE;
225444e1c64aSLisandro Dalcin     c->free_a       = PETSC_TRUE;
22554da8f245SBarry Smith     c->free_ij      = PETSC_TRUE;
22564da8f245SBarry Smith   }
225749b5e25fSSatish Balay   if (mbs > 0) {
22584da8f245SBarry Smith     if (cpvalues != MAT_SHARE_NONZERO_PATTERN) {
2259580bdb30SBarry Smith       ierr = PetscArraycpy(c->j,a->j,nz);CHKERRQ(ierr);
22604da8f245SBarry Smith     }
226149b5e25fSSatish Balay     if (cpvalues == MAT_COPY_VALUES) {
2262580bdb30SBarry Smith       ierr = PetscArraycpy(c->a,a->a,bs2*nz);CHKERRQ(ierr);
226349b5e25fSSatish Balay     } else {
2264580bdb30SBarry Smith       ierr = PetscArrayzero(c->a,bs2*nz);CHKERRQ(ierr);
226549b5e25fSSatish Balay     }
2266a1c3900fSBarry Smith     if (a->jshort) {
226744e1c64aSLisandro Dalcin       /* cannot share jshort, it is reallocated in MatAssemblyEnd_SeqSBAIJ() */
226844e1c64aSLisandro Dalcin       /* if the parent matrix is reassembled, this child matrix will never notice */
2269785e854fSJed Brown       ierr = PetscMalloc1(nz,&c->jshort);CHKERRQ(ierr);
22703bb1ff40SBarry Smith       ierr = PetscLogObjectMemory((PetscObject)C,nz*sizeof(unsigned short));CHKERRQ(ierr);
2271580bdb30SBarry Smith       ierr = PetscArraycpy(c->jshort,a->jshort,nz);CHKERRQ(ierr);
227226fbe8dcSKarl Rupp 
22734da8f245SBarry Smith       c->free_jshort = PETSC_TRUE;
22744da8f245SBarry Smith     }
2275a1c3900fSBarry Smith   }
227649b5e25fSSatish Balay 
227749b5e25fSSatish Balay   c->roworiented = a->roworiented;
227849b5e25fSSatish Balay   c->nonew       = a->nonew;
227949b5e25fSSatish Balay 
228049b5e25fSSatish Balay   if (a->diag) {
2281c760cd28SBarry Smith     if (cpvalues == MAT_SHARE_NONZERO_PATTERN) {
2282c760cd28SBarry Smith       c->diag      = a->diag;
2283c760cd28SBarry Smith       c->free_diag = PETSC_FALSE;
2284c760cd28SBarry Smith     } else {
2285785e854fSJed Brown       ierr = PetscMalloc1(mbs,&c->diag);CHKERRQ(ierr);
22863bb1ff40SBarry Smith       ierr = PetscLogObjectMemory((PetscObject)C,mbs*sizeof(PetscInt));CHKERRQ(ierr);
228726fbe8dcSKarl Rupp       for (i=0; i<mbs; i++) c->diag[i] = a->diag[i];
2288c760cd28SBarry Smith       c->free_diag = PETSC_TRUE;
2289c760cd28SBarry Smith     }
229044e1c64aSLisandro Dalcin   }
22916c6c5352SBarry Smith   c->nz         = a->nz;
2292f2cbd3d5SJed Brown   c->maxnz      = a->nz; /* Since we allocate exactly the right amount */
2293f4259b30SLisandro Dalcin   c->solve_work = NULL;
2294f4259b30SLisandro Dalcin   c->mult_work  = NULL;
229526fbe8dcSKarl Rupp 
229649b5e25fSSatish Balay   *B   = C;
2297140e18c1SBarry Smith   ierr = PetscFunctionListDuplicate(((PetscObject)A)->qlist,&((PetscObject)C)->qlist);CHKERRQ(ierr);
229849b5e25fSSatish Balay   PetscFunctionReturn(0);
229949b5e25fSSatish Balay }
230049b5e25fSSatish Balay 
2301618cc2edSLisandro Dalcin /* Used for both SeqBAIJ and SeqSBAIJ matrices */
2302618cc2edSLisandro Dalcin #define MatLoad_SeqSBAIJ_Binary MatLoad_SeqBAIJ_Binary
2303618cc2edSLisandro Dalcin 
2304618cc2edSLisandro Dalcin PetscErrorCode MatLoad_SeqSBAIJ(Mat mat,PetscViewer viewer)
23052f480046SShri Abhyankar {
23062f480046SShri Abhyankar   PetscErrorCode ierr;
23077f489da9SVaclav Hapla   PetscBool      isbinary;
23082f480046SShri Abhyankar 
23092f480046SShri Abhyankar   PetscFunctionBegin;
23107f489da9SVaclav Hapla   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr);
231198921bdaSJacob Faibussowitsch   if (!isbinary) SETERRQ(PetscObjectComm((PetscObject)viewer),PETSC_ERR_SUP,"Viewer type %s not yet supported for reading %s matrices",((PetscObject)viewer)->type_name,((PetscObject)mat)->type_name);
2312618cc2edSLisandro Dalcin   ierr = MatLoad_SeqSBAIJ_Binary(mat,viewer);CHKERRQ(ierr);
23132f480046SShri Abhyankar   PetscFunctionReturn(0);
23142f480046SShri Abhyankar }
23152f480046SShri Abhyankar 
2316c75a6043SHong Zhang /*@
2317c75a6043SHong Zhang      MatCreateSeqSBAIJWithArrays - Creates an sequential SBAIJ matrix using matrix elements
2318c75a6043SHong Zhang               (upper triangular entries in CSR format) provided by the user.
2319c75a6043SHong Zhang 
2320d083f849SBarry Smith      Collective
2321c75a6043SHong Zhang 
2322c75a6043SHong Zhang    Input Parameters:
2323c75a6043SHong Zhang +  comm - must be an MPI communicator of size 1
2324c75a6043SHong Zhang .  bs - size of block
2325c75a6043SHong Zhang .  m - number of rows
2326c75a6043SHong Zhang .  n - number of columns
2327483a2f95SBarry Smith .  i - row indices; that is i[0] = 0, i[row] = i[row-1] + number of block elements in that row block row of the matrix
2328c75a6043SHong Zhang .  j - column indices
2329c75a6043SHong Zhang -  a - matrix values
2330c75a6043SHong Zhang 
2331c75a6043SHong Zhang    Output Parameter:
2332c75a6043SHong Zhang .  mat - the matrix
2333c75a6043SHong Zhang 
2334dfb205c3SBarry Smith    Level: advanced
2335c75a6043SHong Zhang 
2336c75a6043SHong Zhang    Notes:
2337c75a6043SHong Zhang        The i, j, and a arrays are not copied by this routine, the user must free these arrays
2338c75a6043SHong Zhang     once the matrix is destroyed
2339c75a6043SHong Zhang 
2340c75a6043SHong Zhang        You cannot set new nonzero locations into this matrix, that will generate an error.
2341c75a6043SHong Zhang 
2342c75a6043SHong Zhang        The i and j indices are 0 based
2343c75a6043SHong Zhang 
2344dfb205c3SBarry 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
2345dfb205c3SBarry Smith        it is the regular CSR format excluding the lower triangular elements.
2346dfb205c3SBarry Smith 
234769b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSBAIJ(), MatCreateSeqSBAIJ()
2348c75a6043SHong Zhang 
2349c75a6043SHong Zhang @*/
2350c3c607ccSBarry Smith PetscErrorCode  MatCreateSeqSBAIJWithArrays(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt i[],PetscInt j[],PetscScalar a[],Mat *mat)
2351c75a6043SHong Zhang {
2352c75a6043SHong Zhang   PetscErrorCode ierr;
2353c75a6043SHong Zhang   PetscInt       ii;
2354c75a6043SHong Zhang   Mat_SeqSBAIJ   *sbaij;
2355c75a6043SHong Zhang 
2356c75a6043SHong Zhang   PetscFunctionBegin;
235798921bdaSJacob Faibussowitsch   if (bs != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"block size %" PetscInt_FMT " > 1 is not supported yet",bs);
235841096f02SStefano Zampini   if (m > 0 && i[0]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"i (row indices) must start with 0");
2359c75a6043SHong Zhang 
2360c75a6043SHong Zhang   ierr  = MatCreate(comm,mat);CHKERRQ(ierr);
2361c75a6043SHong Zhang   ierr  = MatSetSizes(*mat,m,n,m,n);CHKERRQ(ierr);
2362c75a6043SHong Zhang   ierr  = MatSetType(*mat,MATSEQSBAIJ);CHKERRQ(ierr);
2363f4259b30SLisandro Dalcin   ierr  = MatSeqSBAIJSetPreallocation(*mat,bs,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr);
2364c75a6043SHong Zhang   sbaij = (Mat_SeqSBAIJ*)(*mat)->data;
2365dcca6d9dSJed Brown   ierr  = PetscMalloc2(m,&sbaij->imax,m,&sbaij->ilen);CHKERRQ(ierr);
23663bb1ff40SBarry Smith   ierr  = PetscLogObjectMemory((PetscObject)*mat,2*m*sizeof(PetscInt));CHKERRQ(ierr);
2367c75a6043SHong Zhang 
2368c75a6043SHong Zhang   sbaij->i = i;
2369c75a6043SHong Zhang   sbaij->j = j;
2370c75a6043SHong Zhang   sbaij->a = a;
237126fbe8dcSKarl Rupp 
2372c75a6043SHong Zhang   sbaij->singlemalloc   = PETSC_FALSE;
2373c75a6043SHong Zhang   sbaij->nonew          = -1;             /*this indicates that inserting a new value in the matrix that generates a new nonzero is an error*/
2374e6b907acSBarry Smith   sbaij->free_a         = PETSC_FALSE;
2375e6b907acSBarry Smith   sbaij->free_ij        = PETSC_FALSE;
2376ddf7884eSMatthew Knepley   sbaij->free_imax_ilen = PETSC_TRUE;
2377c75a6043SHong Zhang 
2378c75a6043SHong Zhang   for (ii=0; ii<m; ii++) {
2379c75a6043SHong Zhang     sbaij->ilen[ii] = sbaij->imax[ii] = i[ii+1] - i[ii];
238098921bdaSJacob Faibussowitsch     if (PetscUnlikelyDebug(i[ii+1] - i[ii] < 0)) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative row length in i (row indices) row = %" PetscInt_FMT " length = %" PetscInt_FMT,ii,i[ii+1] - i[ii]);
2381c75a6043SHong Zhang   }
238276bd3646SJed Brown   if (PetscDefined(USE_DEBUG)) {
2383c75a6043SHong Zhang     for (ii=0; ii<sbaij->i[m]; ii++) {
238498921bdaSJacob Faibussowitsch       if (j[ii] < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column index at location = %" PetscInt_FMT " index = %" PetscInt_FMT,ii,j[ii]);
238598921bdaSJacob Faibussowitsch       if (j[ii] > n - 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column index to large at location = %" PetscInt_FMT " index = %" PetscInt_FMT,ii,j[ii]);
2386c75a6043SHong Zhang     }
238776bd3646SJed Brown   }
2388c75a6043SHong Zhang 
2389c75a6043SHong Zhang   ierr = MatAssemblyBegin(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2390c75a6043SHong Zhang   ierr = MatAssemblyEnd(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2391c75a6043SHong Zhang   PetscFunctionReturn(0);
2392c75a6043SHong Zhang }
2393d06b337dSHong Zhang 
239459f5e6ceSHong Zhang PetscErrorCode MatCreateMPIMatConcatenateSeqMat_SeqSBAIJ(MPI_Comm comm,Mat inmat,PetscInt n,MatReuse scall,Mat *outmat)
239559f5e6ceSHong Zhang {
239659f5e6ceSHong Zhang   PetscErrorCode ierr;
23978761c3d6SHong Zhang   PetscMPIInt    size;
239859f5e6ceSHong Zhang 
239959f5e6ceSHong Zhang   PetscFunctionBegin;
2400ffc4695bSBarry Smith   ierr = MPI_Comm_size(comm,&size);CHKERRMPI(ierr);
24018761c3d6SHong Zhang   if (size == 1 && scall == MAT_REUSE_MATRIX) {
24028761c3d6SHong Zhang     ierr = MatCopy(inmat,*outmat,SAME_NONZERO_PATTERN);CHKERRQ(ierr);
24038761c3d6SHong Zhang   } else {
240459f5e6ceSHong Zhang     ierr = MatCreateMPIMatConcatenateSeqMat_MPISBAIJ(comm,inmat,n,scall,outmat);CHKERRQ(ierr);
24058761c3d6SHong Zhang   }
240659f5e6ceSHong Zhang   PetscFunctionReturn(0);
240759f5e6ceSHong Zhang }
2408