xref: /petsc/src/mat/impls/sbaij/seq/sbaij.c (revision 21940c7e5bf7eaf8e23251c5f0156531fa9cce27)
149b5e25fSSatish Balay 
249b5e25fSSatish Balay /*
3a1373b80SHong Zhang     Defines the basic matrix operations for the SBAIJ (compressed row)
449b5e25fSSatish Balay   matrix storage format.
549b5e25fSSatish Balay */
6c6db04a5SJed Brown #include <../src/mat/impls/baij/seq/baij.h>         /*I "petscmat.h" I*/
7c6db04a5SJed Brown #include <../src/mat/impls/sbaij/seq/sbaij.h>
8c6db04a5SJed Brown #include <petscblaslapack.h>
949b5e25fSSatish Balay 
10c6db04a5SJed Brown #include <../src/mat/impls/sbaij/seq/relax.h>
1170dcbbb9SBarry Smith #define USESHORT
12c6db04a5SJed Brown #include <../src/mat/impls/sbaij/seq/relax.h>
1370dcbbb9SBarry Smith 
14ace3abfcSBarry Smith extern PetscErrorCode MatSeqSBAIJSetNumericFactorization_inplace(Mat,PetscBool);
156214f412SHong Zhang #if defined(PETSC_HAVE_ELEMENTAL)
16cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatConvert_SeqSBAIJ_Elemental(Mat,MatType,MatReuse,Mat*);
176214f412SHong Zhang #endif
18b5b17502SBarry Smith 
1949b5e25fSSatish Balay /*
2049b5e25fSSatish Balay      Checks for missing diagonals
2149b5e25fSSatish Balay */
224a2ae208SSatish Balay #undef __FUNCT__
234a2ae208SSatish Balay #define __FUNCT__ "MatMissingDiagonal_SeqSBAIJ"
24ace3abfcSBarry Smith PetscErrorCode MatMissingDiagonal_SeqSBAIJ(Mat A,PetscBool  *missing,PetscInt *dd)
2549b5e25fSSatish Balay {
26045c9aa0SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
276849ba73SBarry Smith   PetscErrorCode ierr;
287734d3b5SMatthew G. Knepley   PetscInt       *diag,*ii = a->i,i;
2949b5e25fSSatish Balay 
3049b5e25fSSatish Balay   PetscFunctionBegin;
31045c9aa0SHong Zhang   ierr     = MatMarkDiagonal_SeqSBAIJ(A);CHKERRQ(ierr);
322af78befSBarry Smith   *missing = PETSC_FALSE;
337734d3b5SMatthew G. Knepley   if (A->rmap->n > 0 && !ii) {
34358d2f5dSShri Abhyankar     *missing = PETSC_TRUE;
35358d2f5dSShri Abhyankar     if (dd) *dd = 0;
36955c1f14SBarry Smith     ierr = PetscInfo(A,"Matrix has no entries therefore is missing diagonal\n");CHKERRQ(ierr);
37358d2f5dSShri Abhyankar   } else {
38358d2f5dSShri Abhyankar     diag = a->diag;
3949b5e25fSSatish Balay     for (i=0; i<a->mbs; i++) {
407734d3b5SMatthew G. Knepley       if (diag[i] >= ii[i+1]) {
412af78befSBarry Smith         *missing = PETSC_TRUE;
422af78befSBarry Smith         if (dd) *dd = i;
432af78befSBarry Smith         break;
442af78befSBarry Smith       }
4549b5e25fSSatish Balay     }
46358d2f5dSShri Abhyankar   }
4749b5e25fSSatish Balay   PetscFunctionReturn(0);
4849b5e25fSSatish Balay }
4949b5e25fSSatish Balay 
504a2ae208SSatish Balay #undef __FUNCT__
514a2ae208SSatish Balay #define __FUNCT__ "MatMarkDiagonal_SeqSBAIJ"
52dfbe8321SBarry Smith PetscErrorCode MatMarkDiagonal_SeqSBAIJ(Mat A)
5349b5e25fSSatish Balay {
54045c9aa0SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
556849ba73SBarry Smith   PetscErrorCode ierr;
5648dd3d27SHong Zhang   PetscInt       i,j;
5749b5e25fSSatish Balay 
5849b5e25fSSatish Balay   PetscFunctionBegin;
5909f38230SBarry Smith   if (!a->diag) {
60785e854fSJed Brown     ierr         = PetscMalloc1(a->mbs,&a->diag);CHKERRQ(ierr);
613bb1ff40SBarry Smith     ierr         = PetscLogObjectMemory((PetscObject)A,a->mbs*sizeof(PetscInt));CHKERRQ(ierr);
62c760cd28SBarry Smith     a->free_diag = PETSC_TRUE;
6309f38230SBarry Smith   }
6448dd3d27SHong Zhang   for (i=0; i<a->mbs; i++) {
6548dd3d27SHong Zhang     a->diag[i] = a->i[i+1];
6648dd3d27SHong Zhang     for (j=a->i[i]; j<a->i[i+1]; j++) {
6748dd3d27SHong Zhang       if (a->j[j] == i) {
6848dd3d27SHong Zhang         a->diag[i] = j;
6948dd3d27SHong Zhang         break;
7048dd3d27SHong Zhang       }
7148dd3d27SHong Zhang     }
7248dd3d27SHong Zhang   }
7349b5e25fSSatish Balay   PetscFunctionReturn(0);
7449b5e25fSSatish Balay }
7549b5e25fSSatish Balay 
764a2ae208SSatish Balay #undef __FUNCT__
774a2ae208SSatish Balay #define __FUNCT__ "MatGetRowIJ_SeqSBAIJ"
781a83f524SJed Brown static PetscErrorCode MatGetRowIJ_SeqSBAIJ(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *nn,const PetscInt *inia[],const PetscInt *inja[],PetscBool  *done)
7949b5e25fSSatish Balay {
80a6ece127SHong Zhang   Mat_SeqSBAIJ    *a = (Mat_SeqSBAIJ*)A->data;
818f7157efSSatish Balay   PetscErrorCode ierr;
822462f5fdSStefano Zampini   PetscInt       i,j,n = a->mbs,nz = a->i[n],*tia,*tja,bs = A->rmap->bs,k,l,cnt;
832462f5fdSStefano Zampini   PetscInt       **ia = (PetscInt**)inia,**ja = (PetscInt**)inja;
8449b5e25fSSatish Balay 
8549b5e25fSSatish Balay   PetscFunctionBegin;
86d3e5a4abSHong Zhang   *nn = n;
87a1373b80SHong Zhang   if (!ia) PetscFunctionReturn(0);
882462f5fdSStefano Zampini   if (symmetric) {
892462f5fdSStefano Zampini     ierr = MatToSymmetricIJ_SeqAIJ(n,a->i,a->j,PETSC_FALSE,0,0,&tia,&tja);CHKERRQ(ierr);
902462f5fdSStefano Zampini     nz   = tia[n];
912462f5fdSStefano Zampini   } else {
922462f5fdSStefano Zampini     tia = a->i; tja = a->j;
932462f5fdSStefano Zampini   }
942462f5fdSStefano Zampini 
952462f5fdSStefano Zampini   if (!blockcompressed && bs > 1) {
962462f5fdSStefano Zampini     (*nn) *= bs;
978f7157efSSatish Balay     /* malloc & create the natural set of indices */
982462f5fdSStefano Zampini     ierr = PetscMalloc1((n+1)*bs,ia);CHKERRQ(ierr);
992462f5fdSStefano Zampini     if (n) {
1002462f5fdSStefano Zampini       (*ia)[0] = oshift;
1012462f5fdSStefano Zampini       for (j=1; j<bs; j++) {
1022462f5fdSStefano Zampini         (*ia)[j] = (tia[1]-tia[0])*bs+(*ia)[j-1];
1032462f5fdSStefano Zampini       }
1042462f5fdSStefano Zampini     }
1052462f5fdSStefano Zampini 
1062462f5fdSStefano Zampini     for (i=1; i<n; i++) {
1072462f5fdSStefano Zampini       (*ia)[i*bs] = (tia[i]-tia[i-1])*bs + (*ia)[i*bs-1];
1082462f5fdSStefano Zampini       for (j=1; j<bs; j++) {
1092462f5fdSStefano Zampini         (*ia)[i*bs+j] = (tia[i+1]-tia[i])*bs + (*ia)[i*bs+j-1];
1102462f5fdSStefano Zampini       }
1112462f5fdSStefano Zampini     }
1122462f5fdSStefano Zampini     if (n) {
1132462f5fdSStefano Zampini       (*ia)[n*bs] = (tia[n]-tia[n-1])*bs + (*ia)[n*bs-1];
1142462f5fdSStefano Zampini     }
1152462f5fdSStefano Zampini 
1162462f5fdSStefano Zampini     if (inja) {
1172462f5fdSStefano Zampini       ierr = PetscMalloc1(nz*bs*bs,ja);CHKERRQ(ierr);
1182462f5fdSStefano Zampini       cnt = 0;
1192462f5fdSStefano Zampini       for (i=0; i<n; i++) {
1208f7157efSSatish Balay         for (j=0; j<bs; j++) {
1212462f5fdSStefano Zampini           for (k=tia[i]; k<tia[i+1]; k++) {
1222462f5fdSStefano Zampini             for (l=0; l<bs; l++) {
1232462f5fdSStefano Zampini               (*ja)[cnt++] = bs*tja[k] + l;
1248f7157efSSatish Balay             }
1258f7157efSSatish Balay           }
1268f7157efSSatish Balay         }
1278f7157efSSatish Balay       }
1288f7157efSSatish Balay     }
1292462f5fdSStefano Zampini 
1302462f5fdSStefano Zampini     if (symmetric) { /* deallocate memory allocated in MatToSymmetricIJ_SeqAIJ() */
1312462f5fdSStefano Zampini       ierr = PetscFree(tia);CHKERRQ(ierr);
1322462f5fdSStefano Zampini       ierr = PetscFree(tja);CHKERRQ(ierr);
1332462f5fdSStefano Zampini     }
1342462f5fdSStefano Zampini   } else if (oshift == 1) {
1352462f5fdSStefano Zampini     if (symmetric) {
1362462f5fdSStefano Zampini       nz = tia[A->rmap->n/bs];
1372462f5fdSStefano Zampini       /*  add 1 to i and j indices */
1382462f5fdSStefano Zampini       for (i=0; i<A->rmap->n/bs+1; i++) tia[i] = tia[i] + 1;
1392462f5fdSStefano Zampini       *ia = tia;
1402462f5fdSStefano Zampini       if (ja) {
1412462f5fdSStefano Zampini         for (i=0; i<nz; i++) tja[i] = tja[i] + 1;
1422462f5fdSStefano Zampini         *ja = tja;
1432462f5fdSStefano Zampini       }
1442462f5fdSStefano Zampini     } else {
1452462f5fdSStefano Zampini       nz = a->i[A->rmap->n/bs];
1462462f5fdSStefano Zampini       /* malloc space and  add 1 to i and j indices */
1472462f5fdSStefano Zampini       ierr = PetscMalloc1(A->rmap->n/bs+1,ia);CHKERRQ(ierr);
1482462f5fdSStefano Zampini       for (i=0; i<A->rmap->n/bs+1; i++) (*ia)[i] = a->i[i] + 1;
1492462f5fdSStefano Zampini       if (ja) {
1502462f5fdSStefano Zampini         ierr = PetscMalloc1(nz,ja);CHKERRQ(ierr);
1512462f5fdSStefano Zampini         for (i=0; i<nz; i++) (*ja)[i] = a->j[i] + 1;
1522462f5fdSStefano Zampini       }
1532462f5fdSStefano Zampini     }
1542462f5fdSStefano Zampini   } else {
1552462f5fdSStefano Zampini     *ia = tia;
1562462f5fdSStefano Zampini     if (ja) *ja = tja;
157a6ece127SHong Zhang   }
15849b5e25fSSatish Balay   PetscFunctionReturn(0);
15949b5e25fSSatish Balay }
16049b5e25fSSatish Balay 
1614a2ae208SSatish Balay #undef __FUNCT__
1624a2ae208SSatish Balay #define __FUNCT__ "MatRestoreRowIJ_SeqSBAIJ"
1631a83f524SJed Brown static PetscErrorCode MatRestoreRowIJ_SeqSBAIJ(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *nn,const PetscInt *ia[],const PetscInt *ja[],PetscBool  *done)
16449b5e25fSSatish Balay {
1658f7157efSSatish Balay   PetscErrorCode ierr;
166a6ece127SHong Zhang 
16749b5e25fSSatish Balay   PetscFunctionBegin;
16849b5e25fSSatish Balay   if (!ia) PetscFunctionReturn(0);
1692462f5fdSStefano Zampini   if ((!blockcompressed && A->rmap->bs > 1) || (symmetric || oshift == 1)) {
1702462f5fdSStefano Zampini     ierr = PetscFree(*ia);CHKERRQ(ierr);
1712462f5fdSStefano Zampini     if (ja) {ierr = PetscFree(*ja);CHKERRQ(ierr);}
172a6ece127SHong Zhang   }
173a6ece127SHong Zhang   PetscFunctionReturn(0);
17449b5e25fSSatish Balay }
17549b5e25fSSatish Balay 
1764a2ae208SSatish Balay #undef __FUNCT__
1774a2ae208SSatish Balay #define __FUNCT__ "MatDestroy_SeqSBAIJ"
178dfbe8321SBarry Smith PetscErrorCode MatDestroy_SeqSBAIJ(Mat A)
17949b5e25fSSatish Balay {
18049b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
181dfbe8321SBarry Smith   PetscErrorCode ierr;
18249b5e25fSSatish Balay 
18349b5e25fSSatish Balay   PetscFunctionBegin;
184a9f03627SSatish Balay #if defined(PETSC_USE_LOG)
185d0f46423SBarry Smith   PetscLogObjectState((PetscObject)A,"Rows=%D, NZ=%D",A->rmap->N,a->nz);
186a9f03627SSatish Balay #endif
187e6b907acSBarry Smith   ierr = MatSeqXAIJFreeAIJ(A,&a->a,&a->j,&a->i);CHKERRQ(ierr);
1887f53bb6cSHong Zhang   if (a->free_diag) {ierr = PetscFree(a->diag);CHKERRQ(ierr);}
1896bf464f9SBarry Smith   ierr = ISDestroy(&a->row);CHKERRQ(ierr);
1906bf464f9SBarry Smith   ierr = ISDestroy(&a->col);CHKERRQ(ierr);
1916bf464f9SBarry Smith   ierr = ISDestroy(&a->icol);CHKERRQ(ierr);
192c31cb41cSBarry Smith   ierr = PetscFree(a->idiag);CHKERRQ(ierr);
193c31cb41cSBarry Smith   ierr = PetscFree(a->inode.size);CHKERRQ(ierr);
194c760cd28SBarry Smith   if (a->free_imax_ilen) {ierr = PetscFree2(a->imax,a->ilen);CHKERRQ(ierr);}
19505b42c5fSBarry Smith   ierr = PetscFree(a->solve_work);CHKERRQ(ierr);
19641f059aeSBarry Smith   ierr = PetscFree(a->sor_work);CHKERRQ(ierr);
19705b42c5fSBarry Smith   ierr = PetscFree(a->solves_work);CHKERRQ(ierr);
19805b42c5fSBarry Smith   ierr = PetscFree(a->mult_work);CHKERRQ(ierr);
19905b42c5fSBarry Smith   ierr = PetscFree(a->saved_values);CHKERRQ(ierr);
2004da8f245SBarry Smith   if (a->free_jshort) {ierr = PetscFree(a->jshort);CHKERRQ(ierr);}
2011a3463dfSHong Zhang   ierr = PetscFree(a->inew);CHKERRQ(ierr);
2026bf464f9SBarry Smith   ierr = MatDestroy(&a->parent);CHKERRQ(ierr);
203bf0cc555SLisandro Dalcin   ierr = PetscFree(A->data);CHKERRQ(ierr);
204901853e0SKris Buschelman 
205dbd8c25aSHong Zhang   ierr = PetscObjectChangeTypeName((PetscObject)A,0);CHKERRQ(ierr);
206bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatStoreValues_C",NULL);CHKERRQ(ierr);
207bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatRetrieveValues_C",NULL);CHKERRQ(ierr);
208bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetColumnIndices_C",NULL);CHKERRQ(ierr);
209bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqaij_C",NULL);CHKERRQ(ierr);
210bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqbaij_C",NULL);CHKERRQ(ierr);
211bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetPreallocation_C",NULL);CHKERRQ(ierr);
21238f409ebSLisandro Dalcin   ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetPreallocationCSR_C",NULL);CHKERRQ(ierr);
213bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqsbstrm_C",NULL);CHKERRQ(ierr);
2146214f412SHong Zhang #if defined(PETSC_HAVE_ELEMENTAL)
2156214f412SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_elemental_C",NULL);CHKERRQ(ierr);
2166214f412SHong Zhang #endif
21749b5e25fSSatish Balay   PetscFunctionReturn(0);
21849b5e25fSSatish Balay }
21949b5e25fSSatish Balay 
2204a2ae208SSatish Balay #undef __FUNCT__
2214a2ae208SSatish Balay #define __FUNCT__ "MatSetOption_SeqSBAIJ"
222ace3abfcSBarry Smith PetscErrorCode MatSetOption_SeqSBAIJ(Mat A,MatOption op,PetscBool flg)
22349b5e25fSSatish Balay {
224045c9aa0SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
22563ba0a88SBarry Smith   PetscErrorCode ierr;
22649b5e25fSSatish Balay 
22749b5e25fSSatish Balay   PetscFunctionBegin;
2284d9d31abSKris Buschelman   switch (op) {
2294d9d31abSKris Buschelman   case MAT_ROW_ORIENTED:
2304e0d8c25SBarry Smith     a->roworiented = flg;
2314d9d31abSKris Buschelman     break;
232a9817697SBarry Smith   case MAT_KEEP_NONZERO_PATTERN:
233a9817697SBarry Smith     a->keepnonzeropattern = flg;
2344d9d31abSKris Buschelman     break;
235512a5fc5SBarry Smith   case MAT_NEW_NONZERO_LOCATIONS:
236512a5fc5SBarry Smith     a->nonew = (flg ? 0 : 1);
2374d9d31abSKris Buschelman     break;
2384d9d31abSKris Buschelman   case MAT_NEW_NONZERO_LOCATION_ERR:
2394e0d8c25SBarry Smith     a->nonew = (flg ? -1 : 0);
2404d9d31abSKris Buschelman     break;
2414d9d31abSKris Buschelman   case MAT_NEW_NONZERO_ALLOCATION_ERR:
2424e0d8c25SBarry Smith     a->nonew = (flg ? -2 : 0);
2434d9d31abSKris Buschelman     break;
24428b2fa4aSMatthew Knepley   case MAT_UNUSED_NONZERO_LOCATION_ERR:
24528b2fa4aSMatthew Knepley     a->nounused = (flg ? -1 : 0);
24628b2fa4aSMatthew Knepley     break;
2474e0d8c25SBarry Smith   case MAT_NEW_DIAGONALS:
2484d9d31abSKris Buschelman   case MAT_IGNORE_OFF_PROC_ENTRIES:
2494d9d31abSKris Buschelman   case MAT_USE_HASH_TABLE:
250290bbb0aSBarry Smith     ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr);
2514d9d31abSKris Buschelman     break;
2529a4540c5SBarry Smith   case MAT_HERMITIAN:
253e32f2f54SBarry Smith     if (!A->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Must call MatAssemblyEnd() first");
2540bc54ff2SBarry Smith     if (A->cmap->n < 65536 && A->cmap->bs == 1) {
255eeffb40dSHong Zhang       A->ops->mult = MatMult_SeqSBAIJ_1_Hermitian_ushort;
2560bc54ff2SBarry Smith     } else if (A->cmap->bs == 1) {
257eeffb40dSHong Zhang       A->ops->mult = MatMult_SeqSBAIJ_1_Hermitian;
258e32f2f54SBarry Smith     } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No support for Hermitian with block size greater than 1");
259eeffb40dSHong Zhang     break;
2603d472b54SHong Zhang   case MAT_SPD:
2615021d80fSJed Brown     /* These options are handled directly by MatSetOption() */
2623d472b54SHong Zhang     break;
26377e54ba9SKris Buschelman   case MAT_SYMMETRIC:
26477e54ba9SKris Buschelman   case MAT_STRUCTURALLY_SYMMETRIC:
2659a4540c5SBarry Smith   case MAT_SYMMETRY_ETERNAL:
2664dcd73b1SHong Zhang     /* These options are handled directly by MatSetOption() */
267290bbb0aSBarry Smith     break;
268941593c8SHong Zhang   case MAT_IGNORE_LOWER_TRIANGULAR:
2694e0d8c25SBarry Smith     a->ignore_ltriangular = flg;
270941593c8SHong Zhang     break;
271941593c8SHong Zhang   case MAT_ERROR_LOWER_TRIANGULAR:
2724e0d8c25SBarry Smith     a->ignore_ltriangular = flg;
27377e54ba9SKris Buschelman     break;
274f5edf698SHong Zhang   case MAT_GETROW_UPPERTRIANGULAR:
2754e0d8c25SBarry Smith     a->getrow_utriangular = flg;
276f5edf698SHong Zhang     break;
2774d9d31abSKris Buschelman   default:
278e32f2f54SBarry Smith     SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"unknown option %d",op);
27949b5e25fSSatish Balay   }
28049b5e25fSSatish Balay   PetscFunctionReturn(0);
28149b5e25fSSatish Balay }
28249b5e25fSSatish Balay 
2834a2ae208SSatish Balay #undef __FUNCT__
2844a2ae208SSatish Balay #define __FUNCT__ "MatGetRow_SeqSBAIJ"
28552768537SHong Zhang PetscErrorCode MatGetRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
28649b5e25fSSatish Balay {
28749b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
2886849ba73SBarry Smith   PetscErrorCode ierr;
28949b5e25fSSatish Balay 
29049b5e25fSSatish Balay   PetscFunctionBegin;
291e32f2f54SBarry 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()");
29252768537SHong Zhang 
293f5edf698SHong Zhang   /* Get the upper triangular part of the row */
29452768537SHong Zhang   ierr = MatGetRow_SeqBAIJ_private(A,row,nz,idx,v,a->i,a->j,a->a);CHKERRQ(ierr);
29549b5e25fSSatish Balay   PetscFunctionReturn(0);
29649b5e25fSSatish Balay }
29749b5e25fSSatish Balay 
2984a2ae208SSatish Balay #undef __FUNCT__
2994a2ae208SSatish Balay #define __FUNCT__ "MatRestoreRow_SeqSBAIJ"
30013f74950SBarry Smith PetscErrorCode MatRestoreRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
30149b5e25fSSatish Balay {
302dfbe8321SBarry Smith   PetscErrorCode ierr;
30349b5e25fSSatish Balay 
30449b5e25fSSatish Balay   PetscFunctionBegin;
30505b42c5fSBarry Smith   if (idx) {ierr = PetscFree(*idx);CHKERRQ(ierr);}
30605b42c5fSBarry Smith   if (v)   {ierr = PetscFree(*v);CHKERRQ(ierr);}
30749b5e25fSSatish Balay   PetscFunctionReturn(0);
30849b5e25fSSatish Balay }
30949b5e25fSSatish Balay 
3104a2ae208SSatish Balay #undef __FUNCT__
311f5edf698SHong Zhang #define __FUNCT__ "MatGetRowUpperTriangular_SeqSBAIJ"
312f5edf698SHong Zhang PetscErrorCode MatGetRowUpperTriangular_SeqSBAIJ(Mat A)
313f5edf698SHong Zhang {
314f5edf698SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
315f5edf698SHong Zhang 
316f5edf698SHong Zhang   PetscFunctionBegin;
317f5edf698SHong Zhang   a->getrow_utriangular = PETSC_TRUE;
318f5edf698SHong Zhang   PetscFunctionReturn(0);
319f5edf698SHong Zhang }
320f5edf698SHong Zhang #undef __FUNCT__
321f5edf698SHong Zhang #define __FUNCT__ "MatRestoreRowUpperTriangular_SeqSBAIJ"
322f5edf698SHong Zhang PetscErrorCode MatRestoreRowUpperTriangular_SeqSBAIJ(Mat A)
323f5edf698SHong Zhang {
324f5edf698SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
325f5edf698SHong Zhang 
326f5edf698SHong Zhang   PetscFunctionBegin;
327f5edf698SHong Zhang   a->getrow_utriangular = PETSC_FALSE;
328f5edf698SHong Zhang   PetscFunctionReturn(0);
329f5edf698SHong Zhang }
330f5edf698SHong Zhang 
331f5edf698SHong Zhang #undef __FUNCT__
3324a2ae208SSatish Balay #define __FUNCT__ "MatTranspose_SeqSBAIJ"
333fc4dec0aSBarry Smith PetscErrorCode MatTranspose_SeqSBAIJ(Mat A,MatReuse reuse,Mat *B)
33449b5e25fSSatish Balay {
335dfbe8321SBarry Smith   PetscErrorCode ierr;
3365fd66863SKarl Rupp 
33749b5e25fSSatish Balay   PetscFunctionBegin;
338815cbec1SBarry Smith   if (reuse == MAT_INITIAL_MATRIX || *B != A) {
339999d9058SBarry Smith     ierr = MatDuplicate(A,MAT_COPY_VALUES,B);CHKERRQ(ierr);
340fc4dec0aSBarry Smith   }
3418115998fSBarry Smith   PetscFunctionReturn(0);
34249b5e25fSSatish Balay }
34349b5e25fSSatish Balay 
3444a2ae208SSatish Balay #undef __FUNCT__
3454a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_ASCII"
3467da1fb6eSBarry Smith PetscErrorCode MatView_SeqSBAIJ_ASCII(Mat A,PetscViewer viewer)
34749b5e25fSSatish Balay {
34849b5e25fSSatish Balay   Mat_SeqSBAIJ      *a = (Mat_SeqSBAIJ*)A->data;
349dfbe8321SBarry Smith   PetscErrorCode    ierr;
350d0f46423SBarry Smith   PetscInt          i,j,bs = A->rmap->bs,k,l,bs2=a->bs2;
351f3ef73ceSBarry Smith   PetscViewerFormat format;
352121deb67SSatish Balay   PetscInt          *diag;
35349b5e25fSSatish Balay 
35449b5e25fSSatish Balay   PetscFunctionBegin;
355b0a32e0cSBarry Smith   ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
356456192e2SBarry Smith   if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
35777431f27SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  block size is %D\n",bs);CHKERRQ(ierr);
358fb9695e5SSatish Balay   } else if (format == PETSC_VIEWER_ASCII_MATLAB) {
359d2507d54SMatthew Knepley     Mat        aij;
360ade3a672SBarry Smith     const char *matname;
361ade3a672SBarry Smith 
362d5f3da31SBarry Smith     if (A->factortype && bs>1) {
36370d5e725SHong 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);
36470d5e725SHong Zhang       PetscFunctionReturn(0);
36570d5e725SHong Zhang     }
366c9f458caSMatthew Knepley     ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&aij);CHKERRQ(ierr);
367ade3a672SBarry Smith     ierr = PetscObjectGetName((PetscObject)A,&matname);CHKERRQ(ierr);
368ade3a672SBarry Smith     ierr = PetscObjectSetName((PetscObject)aij,matname);CHKERRQ(ierr);
369c9f458caSMatthew Knepley     ierr = MatView(aij,viewer);CHKERRQ(ierr);
3706bf464f9SBarry Smith     ierr = MatDestroy(&aij);CHKERRQ(ierr);
371fb9695e5SSatish Balay   } else if (format == PETSC_VIEWER_ASCII_COMMON) {
372d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
37349b5e25fSSatish Balay     for (i=0; i<a->mbs; i++) {
37449b5e25fSSatish Balay       for (j=0; j<bs; j++) {
37577431f27SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr);
37649b5e25fSSatish Balay         for (k=a->i[i]; k<a->i[i+1]; k++) {
37749b5e25fSSatish Balay           for (l=0; l<bs; l++) {
37849b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX)
37949b5e25fSSatish Balay             if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
38057622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k]+l,
38157622a8eSBarry Smith                                             (double)PetscRealPart(a->a[bs2*k + l*bs + j]),(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
38249b5e25fSSatish Balay             } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
38357622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %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 (PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
38657622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
38749b5e25fSSatish Balay             }
38849b5e25fSSatish Balay #else
38949b5e25fSSatish Balay             if (a->a[bs2*k + l*bs + j] != 0.0) {
39057622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)a->a[bs2*k + l*bs + j]);CHKERRQ(ierr);
39149b5e25fSSatish Balay             }
39249b5e25fSSatish Balay #endif
39349b5e25fSSatish Balay           }
39449b5e25fSSatish Balay         }
395b0a32e0cSBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
39649b5e25fSSatish Balay       }
39749b5e25fSSatish Balay     }
398d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
399c1490034SHong Zhang   } else if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) {
400c1490034SHong Zhang     PetscFunctionReturn(0);
40149b5e25fSSatish Balay   } else {
402d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
4032c990fa1SHong Zhang     if (A->factortype) { /* for factored matrix */
4042c990fa1SHong Zhang       if (bs>1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"matrix is factored with bs>1. Not implemented yet");
4052c990fa1SHong Zhang 
406121deb67SSatish Balay       diag=a->diag;
407121deb67SSatish Balay       for (i=0; i<a->mbs; i++) { /* for row block i */
4082c990fa1SHong Zhang         ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i);CHKERRQ(ierr);
4092c990fa1SHong Zhang         /* diagonal entry */
4102c990fa1SHong Zhang #if defined(PETSC_USE_COMPLEX)
4112c990fa1SHong Zhang         if (PetscImaginaryPart(a->a[diag[i]]) > 0.0) {
41257622a8eSBarry Smith           ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",a->j[diag[i]],(double)PetscRealPart(1.0/a->a[diag[i]]),(double)PetscImaginaryPart(1.0/a->a[diag[i]]));CHKERRQ(ierr);
4132c990fa1SHong Zhang         } else if (PetscImaginaryPart(a->a[diag[i]]) < 0.0) {
41457622a8eSBarry Smith           ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",a->j[diag[i]],(double)PetscRealPart(1.0/a->a[diag[i]]),-(double)PetscImaginaryPart(1.0/a->a[diag[i]]));CHKERRQ(ierr);
4152c990fa1SHong Zhang         } else {
41657622a8eSBarry Smith           ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[diag[i]],(double)PetscRealPart(1.0/a->a[diag[i]]));CHKERRQ(ierr);
4172c990fa1SHong Zhang         }
4182c990fa1SHong Zhang #else
4196712e2f1SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[diag[i]],(double)(1.0/a->a[diag[i]]));CHKERRQ(ierr);
4202c990fa1SHong Zhang #endif
4212c990fa1SHong Zhang         /* off-diagonal entries */
4222c990fa1SHong Zhang         for (k=a->i[i]; k<a->i[i+1]-1; k++) {
4232c990fa1SHong Zhang #if defined(PETSC_USE_COMPLEX)
424ca0704adSBarry Smith           if (PetscImaginaryPart(a->a[k]) > 0.0) {
42557622a8eSBarry Smith             ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k],(double)PetscRealPart(a->a[k]),(double)PetscImaginaryPart(a->a[k]));CHKERRQ(ierr);
426ca0704adSBarry Smith           } else if (PetscImaginaryPart(a->a[k]) < 0.0) {
42757622a8eSBarry Smith             ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k],(double)PetscRealPart(a->a[k]),-(double)PetscImaginaryPart(a->a[k]));CHKERRQ(ierr);
4282c990fa1SHong Zhang           } else {
42957622a8eSBarry Smith             ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k],(double)PetscRealPart(a->a[k]));CHKERRQ(ierr);
4302c990fa1SHong Zhang           }
4312c990fa1SHong Zhang #else
43257622a8eSBarry Smith           ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[k],(double)a->a[k]);CHKERRQ(ierr);
4332c990fa1SHong Zhang #endif
4342c990fa1SHong Zhang         }
4352c990fa1SHong Zhang         ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
4362c990fa1SHong Zhang       }
4372c990fa1SHong Zhang 
4382c990fa1SHong Zhang     } else { /* for non-factored matrix */
4390c74a584SJed Brown       for (i=0; i<a->mbs; i++) { /* for row block i */
4400c74a584SJed Brown         for (j=0; j<bs; j++) {   /* for row bs*i + j */
44177431f27SBarry Smith           ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr);
4420c74a584SJed Brown           for (k=a->i[i]; k<a->i[i+1]; k++) { /* for column block */
4430c74a584SJed Brown             for (l=0; l<bs; l++) {            /* for column */
44449b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX)
44549b5e25fSSatish Balay               if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0) {
44657622a8eSBarry Smith                 ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k]+l,
44757622a8eSBarry Smith                                               (double)PetscRealPart(a->a[bs2*k + l*bs + j]),(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
44849b5e25fSSatish Balay               } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0) {
44957622a8eSBarry Smith                 ierr = PetscViewerASCIIPrintf(viewer," (%D, %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 {
45257622a8eSBarry Smith                 ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
45349b5e25fSSatish Balay               }
45449b5e25fSSatish Balay #else
45557622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)a->a[bs2*k + l*bs + j]);CHKERRQ(ierr);
45649b5e25fSSatish Balay #endif
45749b5e25fSSatish Balay             }
45849b5e25fSSatish Balay           }
459b0a32e0cSBarry Smith           ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
46049b5e25fSSatish Balay         }
46149b5e25fSSatish Balay       }
4622c990fa1SHong Zhang     }
463d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
46449b5e25fSSatish Balay   }
465b0a32e0cSBarry Smith   ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
46649b5e25fSSatish Balay   PetscFunctionReturn(0);
46749b5e25fSSatish Balay }
46849b5e25fSSatish Balay 
4699804daf3SBarry Smith #include <petscdraw.h>
4704a2ae208SSatish Balay #undef __FUNCT__
4714a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_Draw_Zoom"
4726849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw_Zoom(PetscDraw draw,void *Aa)
47349b5e25fSSatish Balay {
47449b5e25fSSatish Balay   Mat            A = (Mat) Aa;
47549b5e25fSSatish Balay   Mat_SeqSBAIJ   *a=(Mat_SeqSBAIJ*)A->data;
4766849ba73SBarry Smith   PetscErrorCode ierr;
477d0f46423SBarry Smith   PetscInt       row,i,j,k,l,mbs=a->mbs,color,bs=A->rmap->bs,bs2=a->bs2;
47849b5e25fSSatish Balay   PetscReal      xl,yl,xr,yr,x_l,x_r,y_l,y_r;
47949b5e25fSSatish Balay   MatScalar      *aa;
480b0a32e0cSBarry Smith   PetscViewer    viewer;
48149b5e25fSSatish Balay 
48249b5e25fSSatish Balay   PetscFunctionBegin;
48349b5e25fSSatish Balay   ierr = PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer);CHKERRQ(ierr);
484b0a32e0cSBarry Smith   ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr);
48549b5e25fSSatish Balay 
48649b5e25fSSatish Balay   /* loop over matrix elements drawing boxes */
487383922c3SLisandro Dalcin 
488383922c3SLisandro Dalcin   ierr = PetscDrawCollectiveBegin(draw);CHKERRQ(ierr);
489383922c3SLisandro Dalcin   ierr = PetscDrawString(draw, .3*(xl+xr), .3*(yl+yr), PETSC_DRAW_BLACK, "symmetric");CHKERRQ(ierr);
490383922c3SLisandro Dalcin   /* Blue for negative, Cyan for zero and  Red for positive */
491b0a32e0cSBarry Smith   color = PETSC_DRAW_BLUE;
49249b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
49349b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
494d0f46423SBarry Smith       y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0;
49549b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
49649b5e25fSSatish Balay       aa  = a->a + j*bs2;
49749b5e25fSSatish Balay       for (k=0; k<bs; k++) {
49849b5e25fSSatish Balay         for (l=0; l<bs; l++) {
49949b5e25fSSatish Balay           if (PetscRealPart(*aa++) >=  0.) continue;
500b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
50149b5e25fSSatish Balay         }
50249b5e25fSSatish Balay       }
50349b5e25fSSatish Balay     }
50449b5e25fSSatish Balay   }
505b0a32e0cSBarry Smith   color = PETSC_DRAW_CYAN;
50649b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
50749b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
508d0f46423SBarry Smith       y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0;
50949b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
51049b5e25fSSatish Balay       aa = a->a + j*bs2;
51149b5e25fSSatish Balay       for (k=0; k<bs; k++) {
51249b5e25fSSatish Balay         for (l=0; l<bs; l++) {
51349b5e25fSSatish Balay           if (PetscRealPart(*aa++) != 0.) continue;
514b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
51549b5e25fSSatish Balay         }
51649b5e25fSSatish Balay       }
51749b5e25fSSatish Balay     }
51849b5e25fSSatish Balay   }
519b0a32e0cSBarry Smith   color = PETSC_DRAW_RED;
52049b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
52149b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
522d0f46423SBarry Smith       y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0;
52349b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
52449b5e25fSSatish Balay       aa = a->a + j*bs2;
52549b5e25fSSatish Balay       for (k=0; k<bs; k++) {
52649b5e25fSSatish Balay         for (l=0; l<bs; l++) {
52749b5e25fSSatish Balay           if (PetscRealPart(*aa++) <= 0.) continue;
528b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
52949b5e25fSSatish Balay         }
53049b5e25fSSatish Balay       }
53149b5e25fSSatish Balay     }
53249b5e25fSSatish Balay   }
533383922c3SLisandro Dalcin   ierr = PetscDrawCollectiveEnd(draw);CHKERRQ(ierr);
53449b5e25fSSatish Balay   PetscFunctionReturn(0);
53549b5e25fSSatish Balay }
53649b5e25fSSatish Balay 
5374a2ae208SSatish Balay #undef __FUNCT__
5384a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_Draw"
5396849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw(Mat A,PetscViewer viewer)
54049b5e25fSSatish Balay {
541dfbe8321SBarry Smith   PetscErrorCode ierr;
54249b5e25fSSatish Balay   PetscReal      xl,yl,xr,yr,w,h;
543b0a32e0cSBarry Smith   PetscDraw      draw;
544ace3abfcSBarry Smith   PetscBool      isnull;
54549b5e25fSSatish Balay 
54649b5e25fSSatish Balay   PetscFunctionBegin;
547b0a32e0cSBarry Smith   ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
548383922c3SLisandro Dalcin   ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr);
549383922c3SLisandro Dalcin   if (isnull) PetscFunctionReturn(0);
55049b5e25fSSatish Balay 
551d0f46423SBarry Smith   xr   = A->rmap->N; yr = A->rmap->N; h = yr/10.0; w = xr/10.0;
55249b5e25fSSatish Balay   xr  += w;          yr += h;        xl = -w;     yl = -h;
553b0a32e0cSBarry Smith   ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr);
554832b7cebSLisandro Dalcin   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer);CHKERRQ(ierr);
555b0a32e0cSBarry Smith   ierr = PetscDrawZoom(draw,MatView_SeqSBAIJ_Draw_Zoom,A);CHKERRQ(ierr);
5560298fd71SBarry Smith   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",NULL);CHKERRQ(ierr);
557832b7cebSLisandro Dalcin   ierr = PetscDrawSave(draw);CHKERRQ(ierr);
55849b5e25fSSatish Balay   PetscFunctionReturn(0);
55949b5e25fSSatish Balay }
56049b5e25fSSatish Balay 
5614a2ae208SSatish Balay #undef __FUNCT__
5624a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ"
563dfbe8321SBarry Smith PetscErrorCode MatView_SeqSBAIJ(Mat A,PetscViewer viewer)
56449b5e25fSSatish Balay {
565dfbe8321SBarry Smith   PetscErrorCode ierr;
566ace3abfcSBarry Smith   PetscBool      iascii,isdraw;
56708917f38SBarry Smith   FILE           *file = 0;
56849b5e25fSSatish Balay 
56949b5e25fSSatish Balay   PetscFunctionBegin;
570251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);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);
57449b5e25fSSatish Balay   } else if (isdraw) {
57549b5e25fSSatish Balay     ierr = MatView_SeqSBAIJ_Draw(A,viewer);CHKERRQ(ierr);
57649b5e25fSSatish Balay   } else {
577a5e6ed63SBarry Smith     Mat        B;
578ade3a672SBarry Smith     const char *matname;
579ceb03754SKris Buschelman     ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&B);CHKERRQ(ierr);
580ade3a672SBarry Smith     ierr = PetscObjectGetName((PetscObject)A,&matname);CHKERRQ(ierr);
581ade3a672SBarry Smith     ierr = PetscObjectSetName((PetscObject)B,matname);CHKERRQ(ierr);
582a5e6ed63SBarry Smith     ierr = MatView(B,viewer);CHKERRQ(ierr);
5836bf464f9SBarry Smith     ierr = MatDestroy(&B);CHKERRQ(ierr);
58408917f38SBarry Smith     ierr = PetscViewerBinaryGetInfoPointer(viewer,&file);CHKERRQ(ierr);
58508917f38SBarry Smith     if (file) {
58608917f38SBarry Smith       fprintf(file,"-matload_block_size %d\n",(int)A->rmap->bs);
58708917f38SBarry Smith     }
58849b5e25fSSatish Balay   }
58949b5e25fSSatish Balay   PetscFunctionReturn(0);
59049b5e25fSSatish Balay }
59149b5e25fSSatish Balay 
59249b5e25fSSatish Balay 
5934a2ae208SSatish Balay #undef __FUNCT__
5944a2ae208SSatish Balay #define __FUNCT__ "MatGetValues_SeqSBAIJ"
59513f74950SBarry Smith PetscErrorCode MatGetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],PetscScalar v[])
59649b5e25fSSatish Balay {
597045c9aa0SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
59813f74950SBarry Smith   PetscInt     *rp,k,low,high,t,row,nrow,i,col,l,*aj = a->j;
59913f74950SBarry Smith   PetscInt     *ai = a->i,*ailen = a->ilen;
600d0f46423SBarry Smith   PetscInt     brow,bcol,ridx,cidx,bs=A->rmap->bs,bs2=a->bs2;
60197e567efSBarry Smith   MatScalar    *ap,*aa = a->a;
60249b5e25fSSatish Balay 
60349b5e25fSSatish Balay   PetscFunctionBegin;
60449b5e25fSSatish Balay   for (k=0; k<m; k++) { /* loop over rows */
60549b5e25fSSatish Balay     row = im[k]; brow = row/bs;
606e32f2f54SBarry Smith     if (row < 0) {v += n; continue;} /* SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative row: %D",row); */
607e32f2f54SBarry Smith     if (row >= A->rmap->N) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,A->rmap->N-1);
60849b5e25fSSatish Balay     rp   = aj + ai[brow]; ap = aa + bs2*ai[brow];
60949b5e25fSSatish Balay     nrow = ailen[brow];
61049b5e25fSSatish Balay     for (l=0; l<n; l++) { /* loop over columns */
611e32f2f54SBarry Smith       if (in[l] < 0) {v++; continue;} /* SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column: %D",in[l]); */
612e32f2f54SBarry Smith       if (in[l] >= A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",in[l],A->cmap->n-1);
61349b5e25fSSatish Balay       col  = in[l];
61449b5e25fSSatish Balay       bcol = col/bs;
61549b5e25fSSatish Balay       cidx = col%bs;
61649b5e25fSSatish Balay       ridx = row%bs;
61749b5e25fSSatish Balay       high = nrow;
61849b5e25fSSatish Balay       low  = 0; /* assume unsorted */
61949b5e25fSSatish Balay       while (high-low > 5) {
62049b5e25fSSatish Balay         t = (low+high)/2;
62149b5e25fSSatish Balay         if (rp[t] > bcol) high = t;
62249b5e25fSSatish Balay         else              low  = t;
62349b5e25fSSatish Balay       }
62449b5e25fSSatish Balay       for (i=low; i<high; i++) {
62549b5e25fSSatish Balay         if (rp[i] > bcol) break;
62649b5e25fSSatish Balay         if (rp[i] == bcol) {
62749b5e25fSSatish Balay           *v++ = ap[bs2*i+bs*cidx+ridx];
62849b5e25fSSatish Balay           goto finished;
62949b5e25fSSatish Balay         }
63049b5e25fSSatish Balay       }
63197e567efSBarry Smith       *v++ = 0.0;
63249b5e25fSSatish Balay finished:;
63349b5e25fSSatish Balay     }
63449b5e25fSSatish Balay   }
63549b5e25fSSatish Balay   PetscFunctionReturn(0);
63649b5e25fSSatish Balay }
63749b5e25fSSatish Balay 
63849b5e25fSSatish Balay 
6394a2ae208SSatish Balay #undef __FUNCT__
6404a2ae208SSatish Balay #define __FUNCT__ "MatSetValuesBlocked_SeqSBAIJ"
64113f74950SBarry Smith PetscErrorCode MatSetValuesBlocked_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is)
64249b5e25fSSatish Balay {
6430880e062SHong Zhang   Mat_SeqSBAIJ      *a = (Mat_SeqSBAIJ*)A->data;
6446849ba73SBarry Smith   PetscErrorCode    ierr;
645e2ee6c50SBarry Smith   PetscInt          *rp,k,low,high,t,ii,jj,row,nrow,i,col,l,rmax,N,lastcol = -1;
64613f74950SBarry Smith   PetscInt          *imax      =a->imax,*ai=a->i,*ailen=a->ilen;
647d0f46423SBarry Smith   PetscInt          *aj        =a->j,nonew=a->nonew,bs2=a->bs2,bs=A->rmap->bs,stepval;
648ace3abfcSBarry Smith   PetscBool         roworiented=a->roworiented;
649dd6ea824SBarry Smith   const PetscScalar *value     = v;
650f15d580aSBarry Smith   MatScalar         *ap,*aa = a->a,*bap;
6510880e062SHong Zhang 
65249b5e25fSSatish Balay   PetscFunctionBegin;
65326fbe8dcSKarl Rupp   if (roworiented) stepval = (n-1)*bs;
65426fbe8dcSKarl Rupp   else stepval = (m-1)*bs;
65526fbe8dcSKarl Rupp 
6560880e062SHong Zhang   for (k=0; k<m; k++) { /* loop over added rows */
6570880e062SHong Zhang     row = im[k];
6580880e062SHong Zhang     if (row < 0) continue;
6592515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
6602f7d4af7SBarry Smith     if (row >= a->mbs) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Block index row too large %D max %D",row,a->mbs-1);
6610880e062SHong Zhang #endif
6620880e062SHong Zhang     rp   = aj + ai[row];
6630880e062SHong Zhang     ap   = aa + bs2*ai[row];
6640880e062SHong Zhang     rmax = imax[row];
6650880e062SHong Zhang     nrow = ailen[row];
6660880e062SHong Zhang     low  = 0;
667818f2c47SBarry Smith     high = nrow;
6680880e062SHong Zhang     for (l=0; l<n; l++) { /* loop over added columns */
6690880e062SHong Zhang       if (in[l] < 0) continue;
6700880e062SHong Zhang       col = in[l];
6712515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
6722f7d4af7SBarry Smith       if (col >= a->nbs) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Block index column too large %D max %D",col,a->nbs-1);
673b1823623SSatish Balay #endif
674b98bf0e1SJed Brown       if (col < row) {
67526fbe8dcSKarl Rupp         if (a->ignore_ltriangular) continue; /* ignore lower triangular block */
67626fbe8dcSKarl 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)");
677b98bf0e1SJed Brown       }
67826fbe8dcSKarl Rupp       if (roworiented) value = v + k*(stepval+bs)*bs + l*bs;
67926fbe8dcSKarl Rupp       else value = v + l*(stepval+bs)*bs + k*bs;
68026fbe8dcSKarl Rupp 
68126fbe8dcSKarl Rupp       if (col <= lastcol) low = 0;
68226fbe8dcSKarl Rupp       else high = nrow;
68326fbe8dcSKarl Rupp 
684e2ee6c50SBarry Smith       lastcol = col;
6850880e062SHong Zhang       while (high-low > 7) {
6860880e062SHong Zhang         t = (low+high)/2;
6870880e062SHong Zhang         if (rp[t] > col) high = t;
6880880e062SHong Zhang         else             low  = t;
6890880e062SHong Zhang       }
6900880e062SHong Zhang       for (i=low; i<high; i++) {
6910880e062SHong Zhang         if (rp[i] > col) break;
6920880e062SHong Zhang         if (rp[i] == col) {
6930880e062SHong Zhang           bap = ap +  bs2*i;
6940880e062SHong Zhang           if (roworiented) {
6950880e062SHong Zhang             if (is == ADD_VALUES) {
6960880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
6970880e062SHong Zhang                 for (jj=ii; jj<bs2; jj+=bs) {
6980880e062SHong Zhang                   bap[jj] += *value++;
6990880e062SHong Zhang                 }
7000880e062SHong Zhang               }
7010880e062SHong Zhang             } else {
7020880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
7030880e062SHong Zhang                 for (jj=ii; jj<bs2; jj+=bs) {
7040880e062SHong Zhang                   bap[jj] = *value++;
7050880e062SHong Zhang                 }
7060880e062SHong Zhang                }
7070880e062SHong Zhang             }
7080880e062SHong Zhang           } else {
7090880e062SHong Zhang             if (is == ADD_VALUES) {
7100880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
7110880e062SHong Zhang                 for (jj=0; jj<bs; jj++) {
7120880e062SHong Zhang                   *bap++ += *value++;
7130880e062SHong Zhang                 }
7140880e062SHong Zhang               }
7150880e062SHong Zhang             } else {
7160880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
7170880e062SHong Zhang                 for (jj=0; jj<bs; jj++) {
7180880e062SHong Zhang                   *bap++  = *value++;
7190880e062SHong Zhang                 }
7200880e062SHong Zhang               }
7210880e062SHong Zhang             }
7220880e062SHong Zhang           }
7230880e062SHong Zhang           goto noinsert2;
7240880e062SHong Zhang         }
7250880e062SHong Zhang       }
7260880e062SHong Zhang       if (nonew == 1) goto noinsert2;
7272f7d4af7SBarry Smith       if (nonew == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new block index nonzero block (%D, %D) in the matrix", row, col);
728fef13f97SBarry Smith       MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,row,col,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar);
729c03d1d03SSatish Balay       N = nrow++ - 1; high++;
7300880e062SHong Zhang       /* shift up all the later entries in this row */
7310880e062SHong Zhang       for (ii=N; ii>=i; ii--) {
7320880e062SHong Zhang         rp[ii+1] = rp[ii];
7330880e062SHong Zhang         ierr     = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr);
7340880e062SHong Zhang       }
7350880e062SHong Zhang       if (N >= i) {
7360880e062SHong Zhang         ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr);
7370880e062SHong Zhang       }
7380880e062SHong Zhang       rp[i] = col;
7390880e062SHong Zhang       bap   = ap +  bs2*i;
7400880e062SHong Zhang       if (roworiented) {
7410880e062SHong Zhang         for (ii=0; ii<bs; ii++,value+=stepval) {
7420880e062SHong Zhang           for (jj=ii; jj<bs2; jj+=bs) {
7430880e062SHong Zhang             bap[jj] = *value++;
7440880e062SHong Zhang           }
7450880e062SHong Zhang         }
7460880e062SHong Zhang       } else {
7470880e062SHong Zhang         for (ii=0; ii<bs; ii++,value+=stepval) {
7480880e062SHong Zhang           for (jj=0; jj<bs; jj++) {
7490880e062SHong Zhang             *bap++ = *value++;
7500880e062SHong Zhang           }
7510880e062SHong Zhang         }
7520880e062SHong Zhang        }
7530880e062SHong Zhang     noinsert2:;
7540880e062SHong Zhang       low = i;
7550880e062SHong Zhang     }
7560880e062SHong Zhang     ailen[row] = nrow;
7570880e062SHong Zhang   }
7580880e062SHong Zhang   PetscFunctionReturn(0);
75949b5e25fSSatish Balay }
76049b5e25fSSatish Balay 
76164831d72SBarry Smith /*
76264831d72SBarry Smith     This is not yet used
76364831d72SBarry Smith */
7644a2ae208SSatish Balay #undef __FUNCT__
7654108e4d5SBarry Smith #define __FUNCT__ "MatAssemblyEnd_SeqSBAIJ_SeqAIJ_Inode"
7664108e4d5SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ_SeqAIJ_Inode(Mat A)
7670def2e27SBarry Smith {
7680def2e27SBarry Smith   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
7690def2e27SBarry Smith   PetscErrorCode ierr;
7700def2e27SBarry Smith   const PetscInt *ai = a->i, *aj = a->j,*cols;
7710def2e27SBarry Smith   PetscInt       i   = 0,j,blk_size,m = A->rmap->n,node_count = 0,nzx,nzy,*ns,row,nz,cnt,cnt2,*counts;
772ace3abfcSBarry Smith   PetscBool      flag;
7730def2e27SBarry Smith 
7740def2e27SBarry Smith   PetscFunctionBegin;
775785e854fSJed Brown   ierr = PetscMalloc1(m,&ns);CHKERRQ(ierr);
7760def2e27SBarry Smith   while (i < m) {
7770def2e27SBarry Smith     nzx = ai[i+1] - ai[i];       /* Number of nonzeros */
7780def2e27SBarry Smith     /* Limits the number of elements in a node to 'a->inode.limit' */
7790def2e27SBarry Smith     for (j=i+1,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) {
7800def2e27SBarry Smith       nzy = ai[j+1] - ai[j];
7810def2e27SBarry Smith       if (nzy != (nzx - j + i)) break;
7820def2e27SBarry Smith       ierr = PetscMemcmp(aj + ai[i] + j - i,aj + ai[j],nzy*sizeof(PetscInt),&flag);CHKERRQ(ierr);
7830def2e27SBarry Smith       if (!flag) break;
7840def2e27SBarry Smith     }
7850def2e27SBarry Smith     ns[node_count++] = blk_size;
78626fbe8dcSKarl Rupp 
7870def2e27SBarry Smith     i = j;
7880def2e27SBarry Smith   }
7890def2e27SBarry Smith   if (!a->inode.size && m && node_count > .9*m) {
7900def2e27SBarry Smith     ierr = PetscFree(ns);CHKERRQ(ierr);
7910def2e27SBarry Smith     ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr);
7920def2e27SBarry Smith   } else {
7930def2e27SBarry Smith     a->inode.node_count = node_count;
79426fbe8dcSKarl Rupp 
795785e854fSJed Brown     ierr = PetscMalloc1(node_count,&a->inode.size);CHKERRQ(ierr);
7963bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)A,node_count*sizeof(PetscInt));CHKERRQ(ierr);
79722d28d08SBarry Smith     ierr = PetscMemcpy(a->inode.size,ns,node_count*sizeof(PetscInt));CHKERRQ(ierr);
7980def2e27SBarry Smith     ierr = PetscFree(ns);CHKERRQ(ierr);
7990def2e27SBarry Smith     ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr);
8000def2e27SBarry Smith 
8010def2e27SBarry Smith     /* count collections of adjacent columns in each inode */
8020def2e27SBarry Smith     row = 0;
8030def2e27SBarry Smith     cnt = 0;
8040def2e27SBarry Smith     for (i=0; i<node_count; i++) {
8050def2e27SBarry Smith       cols = aj + ai[row] + a->inode.size[i];
8060def2e27SBarry Smith       nz   = ai[row+1] - ai[row] - a->inode.size[i];
8070def2e27SBarry Smith       for (j=1; j<nz; j++) {
80826fbe8dcSKarl Rupp         if (cols[j] != cols[j-1]+1) cnt++;
8090def2e27SBarry Smith       }
8100def2e27SBarry Smith       cnt++;
8110def2e27SBarry Smith       row += a->inode.size[i];
8120def2e27SBarry Smith     }
813785e854fSJed Brown     ierr = PetscMalloc1(2*cnt,&counts);CHKERRQ(ierr);
8140def2e27SBarry Smith     cnt  = 0;
8150def2e27SBarry Smith     row  = 0;
8160def2e27SBarry Smith     for (i=0; i<node_count; i++) {
8170def2e27SBarry Smith       cols = aj + ai[row] + a->inode.size[i];
8180def2e27SBarry Smith       counts[2*cnt] = cols[0];
8190def2e27SBarry Smith       nz   = ai[row+1] - ai[row] - a->inode.size[i];
8200def2e27SBarry Smith       cnt2 = 1;
8210def2e27SBarry Smith       for (j=1; j<nz; j++) {
8220def2e27SBarry Smith         if (cols[j] != cols[j-1]+1) {
8230def2e27SBarry Smith           counts[2*(cnt++)+1] = cnt2;
8240def2e27SBarry Smith           counts[2*cnt]       = cols[j];
8250def2e27SBarry Smith           cnt2 = 1;
8260def2e27SBarry Smith         } else cnt2++;
8270def2e27SBarry Smith       }
8280def2e27SBarry Smith       counts[2*(cnt++)+1] = cnt2;
8290def2e27SBarry Smith       row += a->inode.size[i];
8300def2e27SBarry Smith     }
83122d28d08SBarry Smith     ierr = PetscIntView(2*cnt,counts,0);CHKERRQ(ierr);
8320def2e27SBarry Smith   }
83338702af4SBarry Smith   PetscFunctionReturn(0);
83438702af4SBarry Smith }
83538702af4SBarry Smith 
83638702af4SBarry Smith #undef __FUNCT__
8374a2ae208SSatish Balay #define __FUNCT__ "MatAssemblyEnd_SeqSBAIJ"
838dfbe8321SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ(Mat A,MatAssemblyType mode)
83949b5e25fSSatish Balay {
84049b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
8416849ba73SBarry Smith   PetscErrorCode ierr;
84213f74950SBarry Smith   PetscInt       fshift = 0,i,j,*ai = a->i,*aj = a->j,*imax = a->imax;
843d0f46423SBarry Smith   PetscInt       m      = A->rmap->N,*ip,N,*ailen = a->ilen;
84413f74950SBarry Smith   PetscInt       mbs    = a->mbs,bs2 = a->bs2,rmax = 0;
84549b5e25fSSatish Balay   MatScalar      *aa    = a->a,*ap;
84649b5e25fSSatish Balay 
84749b5e25fSSatish Balay   PetscFunctionBegin;
84849b5e25fSSatish Balay   if (mode == MAT_FLUSH_ASSEMBLY) PetscFunctionReturn(0);
84949b5e25fSSatish Balay 
85049b5e25fSSatish Balay   if (m) rmax = ailen[0];
85149b5e25fSSatish Balay   for (i=1; i<mbs; i++) {
85249b5e25fSSatish Balay     /* move each row back by the amount of empty slots (fshift) before it*/
85349b5e25fSSatish Balay     fshift += imax[i-1] - ailen[i-1];
85449b5e25fSSatish Balay     rmax    = PetscMax(rmax,ailen[i]);
85549b5e25fSSatish Balay     if (fshift) {
85649b5e25fSSatish Balay       ip = aj + ai[i]; ap = aa + bs2*ai[i];
85749b5e25fSSatish Balay       N  = ailen[i];
85849b5e25fSSatish Balay       for (j=0; j<N; j++) {
85949b5e25fSSatish Balay         ip[j-fshift] = ip[j];
86049b5e25fSSatish Balay         ierr         = PetscMemcpy(ap+(j-fshift)*bs2,ap+j*bs2,bs2*sizeof(MatScalar));CHKERRQ(ierr);
86149b5e25fSSatish Balay       }
86249b5e25fSSatish Balay     }
86349b5e25fSSatish Balay     ai[i] = ai[i-1] + ailen[i-1];
86449b5e25fSSatish Balay   }
86549b5e25fSSatish Balay   if (mbs) {
86649b5e25fSSatish Balay     fshift += imax[mbs-1] - ailen[mbs-1];
86749b5e25fSSatish Balay     ai[mbs] = ai[mbs-1] + ailen[mbs-1];
86849b5e25fSSatish Balay   }
86949b5e25fSSatish Balay   /* reset ilen and imax for each row */
87049b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
87149b5e25fSSatish Balay     ailen[i] = imax[i] = ai[i+1] - ai[i];
87249b5e25fSSatish Balay   }
8736c6c5352SBarry Smith   a->nz = ai[mbs];
87449b5e25fSSatish Balay 
875b424e231SHong Zhang   /* diagonals may have moved, reset it */
876b424e231SHong Zhang   if (a->diag) {
8772ed38d0bSJed Brown     ierr = PetscMemcpy(a->diag,ai,mbs*sizeof(PetscInt));CHKERRQ(ierr);
87849b5e25fSSatish Balay   }
87926fbe8dcSKarl Rupp   if (fshift && a->nounused == -1) SETERRQ4(PETSC_COMM_SELF,PETSC_ERR_PLIB, "Unused space detected in matrix: %D X %D block size %D, %D unneeded", m, A->cmap->n, A->rmap->bs, fshift*bs2);
88026fbe8dcSKarl Rupp 
881d0f46423SBarry Smith   ierr = PetscInfo5(A,"Matrix size: %D X %D, block size %D; storage space: %D unneeded, %D used\n",m,A->rmap->N,A->rmap->bs,fshift*bs2,a->nz*bs2);CHKERRQ(ierr);
882ae15b995SBarry Smith   ierr = PetscInfo1(A,"Number of mallocs during MatSetValues is %D\n",a->reallocs);CHKERRQ(ierr);
883ae15b995SBarry Smith   ierr = PetscInfo1(A,"Most nonzeros blocks in any row is %D\n",rmax);CHKERRQ(ierr);
88426fbe8dcSKarl Rupp 
8858e58a170SBarry Smith   A->info.mallocs    += a->reallocs;
88649b5e25fSSatish Balay   a->reallocs         = 0;
88749b5e25fSSatish Balay   A->info.nz_unneeded = (PetscReal)fshift*bs2;
888061b2667SBarry Smith   a->idiagvalid       = PETSC_FALSE;
8894dcd73b1SHong Zhang   a->rmax             = rmax;
89038702af4SBarry Smith 
89138702af4SBarry Smith   if (A->cmap->n < 65536 && A->cmap->bs == 1) {
89244e1c64aSLisandro Dalcin     if (a->jshort && a->free_jshort) {
89317803ae8SHong Zhang       /* when matrix data structure is changed, previous jshort must be replaced */
89417803ae8SHong Zhang       ierr = PetscFree(a->jshort);CHKERRQ(ierr);
89517803ae8SHong Zhang     }
896785e854fSJed Brown     ierr = PetscMalloc1(a->i[A->rmap->n],&a->jshort);CHKERRQ(ierr);
8973bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)A,a->i[A->rmap->n]*sizeof(unsigned short));CHKERRQ(ierr);
89838702af4SBarry Smith     for (i=0; i<a->i[A->rmap->n]; i++) a->jshort[i] = a->j[i];
89938702af4SBarry Smith     A->ops->mult   = MatMult_SeqSBAIJ_1_ushort;
90041f059aeSBarry Smith     A->ops->sor    = MatSOR_SeqSBAIJ_ushort;
9014da8f245SBarry Smith     a->free_jshort = PETSC_TRUE;
90238702af4SBarry Smith   }
90349b5e25fSSatish Balay   PetscFunctionReturn(0);
90449b5e25fSSatish Balay }
90549b5e25fSSatish Balay 
90649b5e25fSSatish Balay /*
90749b5e25fSSatish Balay    This function returns an array of flags which indicate the locations of contiguous
90849b5e25fSSatish Balay    blocks that should be zeroed. for eg: if bs = 3  and is = [0,1,2,3,5,6,7,8,9]
90949b5e25fSSatish Balay    then the resulting sizes = [3,1,1,3,1] correspondig to sets [(0,1,2),(3),(5),(6,7,8),(9)]
91049b5e25fSSatish Balay    Assume: sizes should be long enough to hold all the values.
91149b5e25fSSatish Balay */
9124a2ae208SSatish Balay #undef __FUNCT__
9134a2ae208SSatish Balay #define __FUNCT__ "MatZeroRows_SeqSBAIJ_Check_Blocks"
91413f74950SBarry Smith PetscErrorCode MatZeroRows_SeqSBAIJ_Check_Blocks(PetscInt idx[],PetscInt n,PetscInt bs,PetscInt sizes[], PetscInt *bs_max)
91549b5e25fSSatish Balay {
91613f74950SBarry Smith   PetscInt  i,j,k,row;
917ace3abfcSBarry Smith   PetscBool flg;
91849b5e25fSSatish Balay 
91949b5e25fSSatish Balay   PetscFunctionBegin;
92049b5e25fSSatish Balay   for (i=0,j=0; i<n; j++) {
92149b5e25fSSatish Balay     row = idx[i];
92249b5e25fSSatish Balay     if (row%bs!=0) { /* Not the begining of a block */
92349b5e25fSSatish Balay       sizes[j] = 1;
92449b5e25fSSatish Balay       i++;
92549b5e25fSSatish Balay     } else if (i+bs > n) { /* Beginning of a block, but complete block doesn't exist (at idx end) */
92649b5e25fSSatish Balay       sizes[j] = 1;         /* Also makes sure atleast 'bs' values exist for next else */
92749b5e25fSSatish Balay       i++;
92849b5e25fSSatish Balay     } else { /* Begining of the block, so check if the complete block exists */
92949b5e25fSSatish Balay       flg = PETSC_TRUE;
93049b5e25fSSatish Balay       for (k=1; k<bs; k++) {
93149b5e25fSSatish Balay         if (row+k != idx[i+k]) { /* break in the block */
93249b5e25fSSatish Balay           flg = PETSC_FALSE;
93349b5e25fSSatish Balay           break;
93449b5e25fSSatish Balay         }
93549b5e25fSSatish Balay       }
936abc0a331SBarry Smith       if (flg) { /* No break in the bs */
93749b5e25fSSatish Balay         sizes[j] = bs;
93849b5e25fSSatish Balay         i       += bs;
93949b5e25fSSatish Balay       } else {
94049b5e25fSSatish Balay         sizes[j] = 1;
94149b5e25fSSatish Balay         i++;
94249b5e25fSSatish Balay       }
94349b5e25fSSatish Balay     }
94449b5e25fSSatish Balay   }
94549b5e25fSSatish Balay   *bs_max = j;
94649b5e25fSSatish Balay   PetscFunctionReturn(0);
94749b5e25fSSatish Balay }
94849b5e25fSSatish Balay 
94949b5e25fSSatish Balay 
95049b5e25fSSatish Balay /* Only add/insert a(i,j) with i<=j (blocks).
95149b5e25fSSatish Balay    Any a(i,j) with i>j input by user is ingored.
95249b5e25fSSatish Balay */
95349b5e25fSSatish Balay 
9544a2ae208SSatish Balay #undef __FUNCT__
9554a2ae208SSatish Balay #define __FUNCT__ "MatSetValues_SeqSBAIJ"
95613f74950SBarry Smith PetscErrorCode MatSetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is)
95749b5e25fSSatish Balay {
95849b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
9596849ba73SBarry Smith   PetscErrorCode ierr;
960e2ee6c50SBarry Smith   PetscInt       *rp,k,low,high,t,ii,row,nrow,i,col,l,rmax,N,lastcol = -1;
96113f74950SBarry Smith   PetscInt       *imax=a->imax,*ai=a->i,*ailen=a->ilen,roworiented=a->roworiented;
962d0f46423SBarry Smith   PetscInt       *aj  =a->j,nonew=a->nonew,bs=A->rmap->bs,brow,bcol;
96313f74950SBarry Smith   PetscInt       ridx,cidx,bs2=a->bs2;
96449b5e25fSSatish Balay   MatScalar      *ap,value,*aa=a->a,*bap;
96549b5e25fSSatish Balay 
96649b5e25fSSatish Balay   PetscFunctionBegin;
96749b5e25fSSatish Balay   for (k=0; k<m; k++) { /* loop over added rows */
96849b5e25fSSatish Balay     row  = im[k];       /* row number */
96949b5e25fSSatish Balay     brow = row/bs;      /* block row number */
97049b5e25fSSatish Balay     if (row < 0) continue;
9712515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
972e32f2f54SBarry Smith     if (row >= A->rmap->N) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,A->rmap->N-1);
97349b5e25fSSatish Balay #endif
97449b5e25fSSatish Balay     rp   = aj + ai[brow]; /*ptr to beginning of column value of the row block*/
97549b5e25fSSatish Balay     ap   = aa + bs2*ai[brow]; /*ptr to beginning of element value of the row block*/
97649b5e25fSSatish Balay     rmax = imax[brow];  /* maximum space allocated for this row */
97749b5e25fSSatish Balay     nrow = ailen[brow]; /* actual length of this row */
97849b5e25fSSatish Balay     low  = 0;
97949b5e25fSSatish Balay 
98049b5e25fSSatish Balay     for (l=0; l<n; l++) { /* loop over added columns */
98149b5e25fSSatish Balay       if (in[l] < 0) continue;
9822515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
983e32f2f54SBarry Smith       if (in[l] >= A->rmap->N) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",in[l],A->rmap->N-1);
98449b5e25fSSatish Balay #endif
98549b5e25fSSatish Balay       col  = in[l];
98649b5e25fSSatish Balay       bcol = col/bs;              /* block col number */
98749b5e25fSSatish Balay 
988941593c8SHong Zhang       if (brow > bcol) {
98926fbe8dcSKarl Rupp         if (a->ignore_ltriangular) continue; /* ignore lower triangular values */
99026fbe8dcSKarl 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)");
991941593c8SHong Zhang       }
992f4989cb3SHong Zhang 
99349b5e25fSSatish Balay       ridx = row % bs; cidx = col % bs; /*row and col index inside the block */
9948549e402SHong Zhang       if ((brow==bcol && ridx<=cidx) || (brow<bcol)) {
99549b5e25fSSatish Balay         /* element value a(k,l) */
99626fbe8dcSKarl Rupp         if (roworiented) value = v[l + k*n];
99726fbe8dcSKarl Rupp         else value = v[k + l*m];
99849b5e25fSSatish Balay 
99949b5e25fSSatish Balay         /* move pointer bap to a(k,l) quickly and add/insert value */
100026fbe8dcSKarl Rupp         if (col <= lastcol) low = 0;
100126fbe8dcSKarl Rupp         high = nrow;
1002e2ee6c50SBarry Smith         lastcol = col;
100349b5e25fSSatish Balay         while (high-low > 7) {
100449b5e25fSSatish Balay           t = (low+high)/2;
100549b5e25fSSatish Balay           if (rp[t] > bcol) high = t;
100649b5e25fSSatish Balay           else              low  = t;
100749b5e25fSSatish Balay         }
100849b5e25fSSatish Balay         for (i=low; i<high; i++) {
100949b5e25fSSatish Balay           if (rp[i] > bcol) break;
101049b5e25fSSatish Balay           if (rp[i] == bcol) {
101149b5e25fSSatish Balay             bap = ap +  bs2*i + bs*cidx + ridx;
101249b5e25fSSatish Balay             if (is == ADD_VALUES) *bap += value;
101349b5e25fSSatish Balay             else                  *bap  = value;
10148549e402SHong Zhang             /* for diag block, add/insert its symmetric element a(cidx,ridx) */
10158549e402SHong Zhang             if (brow == bcol && ridx < cidx) {
10168549e402SHong Zhang               bap = ap +  bs2*i + bs*ridx + cidx;
10178549e402SHong Zhang               if (is == ADD_VALUES) *bap += value;
10188549e402SHong Zhang               else                  *bap  = value;
10198549e402SHong Zhang             }
102049b5e25fSSatish Balay             goto noinsert1;
102149b5e25fSSatish Balay           }
102249b5e25fSSatish Balay         }
102349b5e25fSSatish Balay 
102449b5e25fSSatish Balay         if (nonew == 1) goto noinsert1;
1025e32f2f54SBarry Smith         if (nonew == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col);
1026fef13f97SBarry Smith         MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,brow,bcol,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar);
102749b5e25fSSatish Balay 
1028c03d1d03SSatish Balay         N = nrow++ - 1; high++;
102949b5e25fSSatish Balay         /* shift up all the later entries in this row */
103049b5e25fSSatish Balay         for (ii=N; ii>=i; ii--) {
103149b5e25fSSatish Balay           rp[ii+1] = rp[ii];
103249b5e25fSSatish Balay           ierr     = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr);
103349b5e25fSSatish Balay         }
103449b5e25fSSatish Balay         if (N>=i) {
103549b5e25fSSatish Balay           ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr);
103649b5e25fSSatish Balay         }
103749b5e25fSSatish Balay         rp[i]                      = bcol;
103849b5e25fSSatish Balay         ap[bs2*i + bs*cidx + ridx] = value;
1039e56f5c9eSBarry Smith         A->nonzerostate++;
104049b5e25fSSatish Balay noinsert1:;
104149b5e25fSSatish Balay         low = i;
10428549e402SHong Zhang       }
104349b5e25fSSatish Balay     }   /* end of loop over added columns */
104449b5e25fSSatish Balay     ailen[brow] = nrow;
104549b5e25fSSatish Balay   }   /* end of loop over added rows */
104649b5e25fSSatish Balay   PetscFunctionReturn(0);
104749b5e25fSSatish Balay }
104849b5e25fSSatish Balay 
10494a2ae208SSatish Balay #undef __FUNCT__
10504d101231SSatish Balay #define __FUNCT__ "MatICCFactor_SeqSBAIJ"
10510481f469SBarry Smith PetscErrorCode MatICCFactor_SeqSBAIJ(Mat inA,IS row,const MatFactorInfo *info)
105249b5e25fSSatish Balay {
10534ccecd49SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)inA->data;
105449b5e25fSSatish Balay   Mat            outA;
1055dfbe8321SBarry Smith   PetscErrorCode ierr;
1056ace3abfcSBarry Smith   PetscBool      row_identity;
105749b5e25fSSatish Balay 
105849b5e25fSSatish Balay   PetscFunctionBegin;
1059e32f2f54SBarry Smith   if (info->levels != 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only levels=0 is supported for in-place icc");
1060c84f5b01SHong Zhang   ierr = ISIdentity(row,&row_identity);CHKERRQ(ierr);
1061e32f2f54SBarry Smith   if (!row_identity) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix reordering is not supported");
1062e32f2f54SBarry Smith   if (inA->rmap->bs != 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix block size %D is not supported",inA->rmap->bs); /* Need to replace MatCholeskyFactorSymbolic_SeqSBAIJ_MSR()! */
1063c84f5b01SHong Zhang 
106449b5e25fSSatish Balay   outA            = inA;
1065d5f3da31SBarry Smith   inA->factortype = MAT_FACTOR_ICC;
1066f6224b95SHong Zhang   ierr = PetscFree(inA->solvertype);CHKERRQ(ierr);
1067f6224b95SHong Zhang   ierr = PetscStrallocpy(MATSOLVERPETSC,&inA->solvertype);CHKERRQ(ierr);
106849b5e25fSSatish Balay 
10691a3463dfSHong Zhang   ierr = MatMarkDiagonal_SeqSBAIJ(inA);CHKERRQ(ierr);
1070d595f711SHong Zhang   ierr = MatSeqSBAIJSetNumericFactorization_inplace(inA,row_identity);CHKERRQ(ierr);
107149b5e25fSSatish Balay 
1072c3122656SLisandro Dalcin   ierr   = PetscObjectReference((PetscObject)row);CHKERRQ(ierr);
10736bf464f9SBarry Smith   ierr   = ISDestroy(&a->row);CHKERRQ(ierr);
1074c84f5b01SHong Zhang   a->row = row;
1075c3122656SLisandro Dalcin   ierr   = PetscObjectReference((PetscObject)row);CHKERRQ(ierr);
10766bf464f9SBarry Smith   ierr   = ISDestroy(&a->col);CHKERRQ(ierr);
1077c84f5b01SHong Zhang   a->col = row;
1078c84f5b01SHong Zhang 
1079c84f5b01SHong Zhang   /* Create the invert permutation so that it can be used in MatCholeskyFactorNumeric() */
1080c84f5b01SHong Zhang   if (a->icol) {ierr = ISInvertPermutation(row,PETSC_DECIDE, &a->icol);CHKERRQ(ierr);}
10813bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)inA,(PetscObject)a->icol);CHKERRQ(ierr);
108249b5e25fSSatish Balay 
108349b5e25fSSatish Balay   if (!a->solve_work) {
1084854ce69bSBarry Smith     ierr = PetscMalloc1(inA->rmap->N+inA->rmap->bs,&a->solve_work);CHKERRQ(ierr);
10853bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)inA,(inA->rmap->N+inA->rmap->bs)*sizeof(PetscScalar));CHKERRQ(ierr);
108649b5e25fSSatish Balay   }
108749b5e25fSSatish Balay 
1088719d5645SBarry Smith   ierr = MatCholeskyFactorNumeric(outA,inA,info);CHKERRQ(ierr);
108949b5e25fSSatish Balay   PetscFunctionReturn(0);
109049b5e25fSSatish Balay }
1091950f1e5bSHong Zhang 
10924a2ae208SSatish Balay #undef __FUNCT__
10934a2ae208SSatish Balay #define __FUNCT__ "MatSeqSBAIJSetColumnIndices_SeqSBAIJ"
10947087cfbeSBarry Smith PetscErrorCode  MatSeqSBAIJSetColumnIndices_SeqSBAIJ(Mat mat,PetscInt *indices)
109549b5e25fSSatish Balay {
1096045c9aa0SHong Zhang   Mat_SeqSBAIJ   *baij = (Mat_SeqSBAIJ*)mat->data;
109713f74950SBarry Smith   PetscInt       i,nz,n;
10987827cd58SJed Brown   PetscErrorCode ierr;
109949b5e25fSSatish Balay 
110049b5e25fSSatish Balay   PetscFunctionBegin;
11016c6c5352SBarry Smith   nz = baij->maxnz;
1102d0f46423SBarry Smith   n  = mat->cmap->n;
110326fbe8dcSKarl Rupp   for (i=0; i<nz; i++) baij->j[i] = indices[i];
110426fbe8dcSKarl Rupp 
11056c6c5352SBarry Smith   baij->nz = nz;
110626fbe8dcSKarl Rupp   for (i=0; i<n; i++) baij->ilen[i] = baij->imax[i];
110726fbe8dcSKarl Rupp 
11087827cd58SJed Brown   ierr = MatSetOption(mat,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
110949b5e25fSSatish Balay   PetscFunctionReturn(0);
111049b5e25fSSatish Balay }
111149b5e25fSSatish Balay 
11124a2ae208SSatish Balay #undef __FUNCT__
11134a2ae208SSatish Balay #define __FUNCT__ "MatSeqSBAIJSetColumnIndices"
111449b5e25fSSatish Balay /*@
111519585528SSatish Balay   MatSeqSBAIJSetColumnIndices - Set the column indices for all the rows
111649b5e25fSSatish Balay   in the matrix.
111749b5e25fSSatish Balay 
111849b5e25fSSatish Balay   Input Parameters:
111919585528SSatish Balay   +  mat     - the SeqSBAIJ matrix
112049b5e25fSSatish Balay   -  indices - the column indices
112149b5e25fSSatish Balay 
112249b5e25fSSatish Balay   Level: advanced
112349b5e25fSSatish Balay 
112449b5e25fSSatish Balay   Notes:
112549b5e25fSSatish Balay   This can be called if you have precomputed the nonzero structure of the
112649b5e25fSSatish Balay   matrix and want to provide it to the matrix object to improve the performance
112749b5e25fSSatish Balay   of the MatSetValues() operation.
112849b5e25fSSatish Balay 
112949b5e25fSSatish Balay   You MUST have set the correct numbers of nonzeros per row in the call to
1130d1be2dadSMatthew Knepley   MatCreateSeqSBAIJ(), and the columns indices MUST be sorted.
113149b5e25fSSatish Balay 
1132ab9f2c04SSatish Balay   MUST be called before any calls to MatSetValues()
113349b5e25fSSatish Balay 
1134ab9f2c04SSatish Balay   .seealso: MatCreateSeqSBAIJ
113549b5e25fSSatish Balay @*/
11367087cfbeSBarry Smith PetscErrorCode  MatSeqSBAIJSetColumnIndices(Mat mat,PetscInt *indices)
113749b5e25fSSatish Balay {
11384ac538c5SBarry Smith   PetscErrorCode ierr;
113949b5e25fSSatish Balay 
114049b5e25fSSatish Balay   PetscFunctionBegin;
11410700a824SBarry Smith   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
11424482741eSBarry Smith   PetscValidPointer(indices,2);
11434ac538c5SBarry Smith   ierr = PetscUseMethod(mat,"MatSeqSBAIJSetColumnIndices_C",(Mat,PetscInt*),(mat,indices));CHKERRQ(ierr);
114449b5e25fSSatish Balay   PetscFunctionReturn(0);
114549b5e25fSSatish Balay }
114649b5e25fSSatish Balay 
11474a2ae208SSatish Balay #undef __FUNCT__
11483c896bc6SHong Zhang #define __FUNCT__ "MatCopy_SeqSBAIJ"
11493c896bc6SHong Zhang PetscErrorCode MatCopy_SeqSBAIJ(Mat A,Mat B,MatStructure str)
11503c896bc6SHong Zhang {
11513c896bc6SHong Zhang   PetscErrorCode ierr;
11523c896bc6SHong Zhang 
11533c896bc6SHong Zhang   PetscFunctionBegin;
11543c896bc6SHong Zhang   /* If the two matrices have the same copy implementation, use fast copy. */
11553c896bc6SHong Zhang   if (str == SAME_NONZERO_PATTERN && (A->ops->copy == B->ops->copy)) {
11563c896bc6SHong Zhang     Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
11573c896bc6SHong Zhang     Mat_SeqSBAIJ *b = (Mat_SeqSBAIJ*)B->data;
11583c896bc6SHong Zhang 
1159e7e72b3dSBarry Smith     if (a->i[A->rmap->N] != b->i[B->rmap->N]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_INCOMP,"Number of nonzeros in two matrices are different");
1160d0f46423SBarry Smith     ierr = PetscMemcpy(b->a,a->a,(a->i[A->rmap->N])*sizeof(PetscScalar));CHKERRQ(ierr);
11613c896bc6SHong Zhang   } else {
1162f5edf698SHong Zhang     ierr = MatGetRowUpperTriangular(A);CHKERRQ(ierr);
11633c896bc6SHong Zhang     ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr);
1164f5edf698SHong Zhang     ierr = MatRestoreRowUpperTriangular(A);CHKERRQ(ierr);
11653c896bc6SHong Zhang   }
11663c896bc6SHong Zhang   PetscFunctionReturn(0);
11673c896bc6SHong Zhang }
11683c896bc6SHong Zhang 
11693c896bc6SHong Zhang #undef __FUNCT__
11704994cf47SJed Brown #define __FUNCT__ "MatSetUp_SeqSBAIJ"
11714994cf47SJed Brown PetscErrorCode MatSetUp_SeqSBAIJ(Mat A)
1172273d9f13SBarry Smith {
1173dfbe8321SBarry Smith   PetscErrorCode ierr;
1174273d9f13SBarry Smith 
1175273d9f13SBarry Smith   PetscFunctionBegin;
1176367daffbSBarry Smith   ierr = MatSeqSBAIJSetPreallocation(A,A->rmap->bs,PETSC_DEFAULT,0);CHKERRQ(ierr);
1177273d9f13SBarry Smith   PetscFunctionReturn(0);
1178273d9f13SBarry Smith }
1179273d9f13SBarry Smith 
1180a6ece127SHong Zhang #undef __FUNCT__
11818c778c55SBarry Smith #define __FUNCT__ "MatSeqSBAIJGetArray_SeqSBAIJ"
11828c778c55SBarry Smith PetscErrorCode MatSeqSBAIJGetArray_SeqSBAIJ(Mat A,PetscScalar *array[])
1183a6ece127SHong Zhang {
1184a6ece127SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
11855fd66863SKarl Rupp 
1186a6ece127SHong Zhang   PetscFunctionBegin;
1187a6ece127SHong Zhang   *array = a->a;
1188a6ece127SHong Zhang   PetscFunctionReturn(0);
1189a6ece127SHong Zhang }
1190a6ece127SHong Zhang 
1191a6ece127SHong Zhang #undef __FUNCT__
11928c778c55SBarry Smith #define __FUNCT__ "MatSeqSBAIJRestoreArray_SeqSBAIJ"
11938c778c55SBarry Smith PetscErrorCode MatSeqSBAIJRestoreArray_SeqSBAIJ(Mat A,PetscScalar *array[])
1194a6ece127SHong Zhang {
1195a6ece127SHong Zhang   PetscFunctionBegin;
1196a6ece127SHong Zhang   PetscFunctionReturn(0);
1197a6ece127SHong Zhang }
1198a6ece127SHong Zhang 
119942ee4b1aSHong Zhang #undef __FUNCT__
120052768537SHong Zhang #define __FUNCT__ "MatAXPYGetPreallocation_SeqSBAIJ"
120152768537SHong Zhang PetscErrorCode MatAXPYGetPreallocation_SeqSBAIJ(Mat Y,Mat X,PetscInt *nnz)
120252768537SHong Zhang {
1203b264fe52SHong Zhang   PetscInt       bs = Y->rmap->bs,mbs = Y->rmap->N/bs;
120452768537SHong Zhang   Mat_SeqSBAIJ   *x = (Mat_SeqSBAIJ*)X->data;
120552768537SHong Zhang   Mat_SeqSBAIJ   *y = (Mat_SeqSBAIJ*)Y->data;
1206b264fe52SHong Zhang   PetscErrorCode ierr;
120752768537SHong Zhang 
120852768537SHong Zhang   PetscFunctionBegin;
120952768537SHong Zhang   /* Set the number of nonzeros in the new matrix */
1210b264fe52SHong Zhang   ierr = MatAXPYGetPreallocation_SeqX_private(mbs,x->i,x->j,y->i,y->j,nnz);CHKERRQ(ierr);
121152768537SHong Zhang   PetscFunctionReturn(0);
121252768537SHong Zhang }
121352768537SHong Zhang 
121452768537SHong Zhang #undef __FUNCT__
121542ee4b1aSHong Zhang #define __FUNCT__ "MatAXPY_SeqSBAIJ"
1216f4df32b1SMatthew Knepley PetscErrorCode MatAXPY_SeqSBAIJ(Mat Y,PetscScalar a,Mat X,MatStructure str)
121742ee4b1aSHong Zhang {
121842ee4b1aSHong Zhang   Mat_SeqSBAIJ   *x=(Mat_SeqSBAIJ*)X->data, *y=(Mat_SeqSBAIJ*)Y->data;
1219dfbe8321SBarry Smith   PetscErrorCode ierr;
122031ce2d13SHong Zhang   PetscInt       bs=Y->rmap->bs,bs2=bs*bs;
1221e838b9e7SJed Brown   PetscBLASInt   one = 1;
122242ee4b1aSHong Zhang 
122342ee4b1aSHong Zhang   PetscFunctionBegin;
122442ee4b1aSHong Zhang   if (str == SAME_NONZERO_PATTERN) {
1225f4df32b1SMatthew Knepley     PetscScalar  alpha = a;
1226c5df96a5SBarry Smith     PetscBLASInt bnz;
1227c5df96a5SBarry Smith     ierr = PetscBLASIntCast(x->nz*bs2,&bnz);CHKERRQ(ierr);
12288b83055fSJed Brown     PetscStackCallBLAS("BLASaxpy",BLASaxpy_(&bnz,&alpha,x->a,&one,y->a,&one));
1229a3fa217bSJose E. Roman     ierr = PetscObjectStateIncrease((PetscObject)Y);CHKERRQ(ierr);
1230ab784542SHong Zhang   } else if (str == SUBSET_NONZERO_PATTERN) { /* nonzeros of X is a subset of Y's */
1231ab784542SHong Zhang     ierr = MatSetOption(X,MAT_GETROW_UPPERTRIANGULAR,PETSC_TRUE);CHKERRQ(ierr);
1232ab784542SHong Zhang     ierr = MatAXPY_Basic(Y,a,X,str);CHKERRQ(ierr);
1233ab784542SHong Zhang     ierr = MatSetOption(X,MAT_GETROW_UPPERTRIANGULAR,PETSC_FALSE);CHKERRQ(ierr);
123442ee4b1aSHong Zhang   } else {
123552768537SHong Zhang     Mat      B;
123652768537SHong Zhang     PetscInt *nnz;
123752768537SHong Zhang     if (bs != X->rmap->bs) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Matrices must have same block size");
1238f5edf698SHong Zhang     ierr = MatGetRowUpperTriangular(X);CHKERRQ(ierr);
123952768537SHong Zhang     ierr = MatGetRowUpperTriangular(Y);CHKERRQ(ierr);
124052768537SHong Zhang     ierr = PetscMalloc1(Y->rmap->N,&nnz);CHKERRQ(ierr);
124152768537SHong Zhang     ierr = MatCreate(PetscObjectComm((PetscObject)Y),&B);CHKERRQ(ierr);
124252768537SHong Zhang     ierr = PetscObjectSetName((PetscObject)B,((PetscObject)Y)->name);CHKERRQ(ierr);
124352768537SHong Zhang     ierr = MatSetSizes(B,Y->rmap->n,Y->cmap->n,Y->rmap->N,Y->cmap->N);CHKERRQ(ierr);
124452768537SHong Zhang     ierr = MatSetBlockSizesFromMats(B,Y,Y);CHKERRQ(ierr);
124552768537SHong Zhang     ierr = MatSetType(B,(MatType) ((PetscObject)Y)->type_name);CHKERRQ(ierr);
124652768537SHong Zhang     ierr = MatAXPYGetPreallocation_SeqSBAIJ(Y,X,nnz);CHKERRQ(ierr);
124752768537SHong Zhang     ierr = MatSeqSBAIJSetPreallocation(B,bs,0,nnz);CHKERRQ(ierr);
124852768537SHong Zhang 
124952768537SHong Zhang     ierr = MatAXPY_BasicWithPreallocation(B,Y,a,X,str);CHKERRQ(ierr);
125052768537SHong Zhang 
125128be2f97SBarry Smith     ierr = MatHeaderReplace(Y,&B);CHKERRQ(ierr);
125252768537SHong Zhang     ierr = PetscFree(nnz);CHKERRQ(ierr);
1253f5edf698SHong Zhang     ierr = MatRestoreRowUpperTriangular(X);CHKERRQ(ierr);
125452768537SHong Zhang     ierr = MatRestoreRowUpperTriangular(Y);CHKERRQ(ierr);
125542ee4b1aSHong Zhang   }
125642ee4b1aSHong Zhang   PetscFunctionReturn(0);
125742ee4b1aSHong Zhang }
125842ee4b1aSHong Zhang 
1259efcf0fc3SBarry Smith #undef __FUNCT__
1260efcf0fc3SBarry Smith #define __FUNCT__ "MatIsSymmetric_SeqSBAIJ"
1261ace3abfcSBarry Smith PetscErrorCode MatIsSymmetric_SeqSBAIJ(Mat A,PetscReal tol,PetscBool  *flg)
1262efcf0fc3SBarry Smith {
1263efcf0fc3SBarry Smith   PetscFunctionBegin;
1264efcf0fc3SBarry Smith   *flg = PETSC_TRUE;
1265efcf0fc3SBarry Smith   PetscFunctionReturn(0);
1266efcf0fc3SBarry Smith }
1267efcf0fc3SBarry Smith 
1268efcf0fc3SBarry Smith #undef __FUNCT__
1269efcf0fc3SBarry Smith #define __FUNCT__ "MatIsStructurallySymmetric_SeqSBAIJ"
1270ace3abfcSBarry Smith PetscErrorCode MatIsStructurallySymmetric_SeqSBAIJ(Mat A,PetscBool  *flg)
1271efcf0fc3SBarry Smith {
1272efcf0fc3SBarry Smith   PetscFunctionBegin;
1273efcf0fc3SBarry Smith   *flg = PETSC_TRUE;
1274efcf0fc3SBarry Smith   PetscFunctionReturn(0);
1275efcf0fc3SBarry Smith }
1276efcf0fc3SBarry Smith 
1277efcf0fc3SBarry Smith #undef __FUNCT__
1278efcf0fc3SBarry Smith #define __FUNCT__ "MatIsHermitian_SeqSBAIJ"
1279ace3abfcSBarry Smith PetscErrorCode MatIsHermitian_SeqSBAIJ(Mat A,PetscReal tol,PetscBool  *flg)
1280efcf0fc3SBarry Smith {
1281efcf0fc3SBarry Smith   PetscFunctionBegin;
1282efcf0fc3SBarry Smith   *flg = PETSC_FALSE;
1283efcf0fc3SBarry Smith   PetscFunctionReturn(0);
1284efcf0fc3SBarry Smith }
1285efcf0fc3SBarry Smith 
128699cafbc1SBarry Smith #undef __FUNCT__
128799cafbc1SBarry Smith #define __FUNCT__ "MatRealPart_SeqSBAIJ"
128899cafbc1SBarry Smith PetscErrorCode MatRealPart_SeqSBAIJ(Mat A)
128999cafbc1SBarry Smith {
129099cafbc1SBarry Smith   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
129199cafbc1SBarry Smith   PetscInt     i,nz = a->bs2*a->i[a->mbs];
1292dd6ea824SBarry Smith   MatScalar    *aa = a->a;
129399cafbc1SBarry Smith 
129499cafbc1SBarry Smith   PetscFunctionBegin;
129599cafbc1SBarry Smith   for (i=0; i<nz; i++) aa[i] = PetscRealPart(aa[i]);
129699cafbc1SBarry Smith   PetscFunctionReturn(0);
129799cafbc1SBarry Smith }
129899cafbc1SBarry Smith 
129999cafbc1SBarry Smith #undef __FUNCT__
130099cafbc1SBarry Smith #define __FUNCT__ "MatImaginaryPart_SeqSBAIJ"
130199cafbc1SBarry Smith PetscErrorCode MatImaginaryPart_SeqSBAIJ(Mat A)
130299cafbc1SBarry Smith {
130399cafbc1SBarry Smith   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
130499cafbc1SBarry Smith   PetscInt     i,nz = a->bs2*a->i[a->mbs];
1305dd6ea824SBarry Smith   MatScalar    *aa = a->a;
130699cafbc1SBarry Smith 
130799cafbc1SBarry Smith   PetscFunctionBegin;
130899cafbc1SBarry Smith   for (i=0; i<nz; i++) aa[i] = PetscImaginaryPart(aa[i]);
130999cafbc1SBarry Smith   PetscFunctionReturn(0);
131099cafbc1SBarry Smith }
131199cafbc1SBarry Smith 
13123bededecSBarry Smith #undef __FUNCT__
13133bededecSBarry Smith #define __FUNCT__ "MatZeroRowsColumns_SeqSBAIJ"
13143bededecSBarry Smith PetscErrorCode MatZeroRowsColumns_SeqSBAIJ(Mat A,PetscInt is_n,const PetscInt is_idx[],PetscScalar diag,Vec x, Vec b)
13153bededecSBarry Smith {
13163bededecSBarry Smith   Mat_SeqSBAIJ      *baij=(Mat_SeqSBAIJ*)A->data;
13173bededecSBarry Smith   PetscErrorCode    ierr;
13183bededecSBarry Smith   PetscInt          i,j,k,count;
13193bededecSBarry Smith   PetscInt          bs   =A->rmap->bs,bs2=baij->bs2,row,col;
13203bededecSBarry Smith   PetscScalar       zero = 0.0;
13213bededecSBarry Smith   MatScalar         *aa;
13223bededecSBarry Smith   const PetscScalar *xx;
13233bededecSBarry Smith   PetscScalar       *bb;
132456777dd2SBarry Smith   PetscBool         *zeroed,vecs = PETSC_FALSE;
13253bededecSBarry Smith 
13263bededecSBarry Smith   PetscFunctionBegin;
13273bededecSBarry Smith   /* fix right hand side if needed */
13283bededecSBarry Smith   if (x && b) {
13293bededecSBarry Smith     ierr = VecGetArrayRead(x,&xx);CHKERRQ(ierr);
13303bededecSBarry Smith     ierr = VecGetArray(b,&bb);CHKERRQ(ierr);
133156777dd2SBarry Smith     vecs = PETSC_TRUE;
13323bededecSBarry Smith   }
13333bededecSBarry Smith 
13343bededecSBarry Smith   /* zero the columns */
13351795a4d1SJed Brown   ierr = PetscCalloc1(A->rmap->n,&zeroed);CHKERRQ(ierr);
13363bededecSBarry Smith   for (i=0; i<is_n; i++) {
13373bededecSBarry Smith     if (is_idx[i] < 0 || is_idx[i] >= A->rmap->N) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"row %D out of range",is_idx[i]);
13383bededecSBarry Smith     zeroed[is_idx[i]] = PETSC_TRUE;
13393bededecSBarry Smith   }
134056777dd2SBarry Smith   if (vecs) {
134156777dd2SBarry Smith     for (i=0; i<A->rmap->N; i++) {
134256777dd2SBarry Smith       row = i/bs;
134356777dd2SBarry Smith       for (j=baij->i[row]; j<baij->i[row+1]; j++) {
134456777dd2SBarry Smith         for (k=0; k<bs; k++) {
134556777dd2SBarry Smith           col = bs*baij->j[j] + k;
134656777dd2SBarry Smith           if (col <= i) continue;
134756777dd2SBarry Smith           aa = ((MatScalar*)(baij->a)) + j*bs2 + (i%bs) + bs*k;
134826fbe8dcSKarl Rupp           if (!zeroed[i] && zeroed[col]) bb[i]   -= aa[0]*xx[col];
134926fbe8dcSKarl Rupp           if (zeroed[i] && !zeroed[col]) bb[col] -= aa[0]*xx[i];
135056777dd2SBarry Smith         }
135156777dd2SBarry Smith       }
135256777dd2SBarry Smith     }
135326fbe8dcSKarl Rupp     for (i=0; i<is_n; i++) bb[is_idx[i]] = diag*xx[is_idx[i]];
135456777dd2SBarry Smith   }
135556777dd2SBarry Smith 
13563bededecSBarry Smith   for (i=0; i<A->rmap->N; i++) {
13573bededecSBarry Smith     if (!zeroed[i]) {
13583bededecSBarry Smith       row = i/bs;
13593bededecSBarry Smith       for (j=baij->i[row]; j<baij->i[row+1]; j++) {
13603bededecSBarry Smith         for (k=0; k<bs; k++) {
13613bededecSBarry Smith           col = bs*baij->j[j] + k;
13623bededecSBarry Smith           if (zeroed[col]) {
13633bededecSBarry Smith             aa = ((MatScalar*)(baij->a)) + j*bs2 + (i%bs) + bs*k;
13643bededecSBarry Smith             aa[0] = 0.0;
13653bededecSBarry Smith           }
13663bededecSBarry Smith         }
13673bededecSBarry Smith       }
13683bededecSBarry Smith     }
13693bededecSBarry Smith   }
13703bededecSBarry Smith   ierr = PetscFree(zeroed);CHKERRQ(ierr);
137156777dd2SBarry Smith   if (vecs) {
137256777dd2SBarry Smith     ierr = VecRestoreArrayRead(x,&xx);CHKERRQ(ierr);
137356777dd2SBarry Smith     ierr = VecRestoreArray(b,&bb);CHKERRQ(ierr);
137456777dd2SBarry Smith   }
13753bededecSBarry Smith 
13763bededecSBarry Smith   /* zero the rows */
13773bededecSBarry Smith   for (i=0; i<is_n; i++) {
13783bededecSBarry Smith     row   = is_idx[i];
13793bededecSBarry Smith     count = (baij->i[row/bs +1] - baij->i[row/bs])*bs;
13803bededecSBarry Smith     aa    = ((MatScalar*)(baij->a)) + baij->i[row/bs]*bs2 + (row%bs);
13813bededecSBarry Smith     for (k=0; k<count; k++) {
13823bededecSBarry Smith       aa[0] =  zero;
13833bededecSBarry Smith       aa   += bs;
13843bededecSBarry Smith     }
13853bededecSBarry Smith     if (diag != 0.0) {
13863bededecSBarry Smith       ierr = (*A->ops->setvalues)(A,1,&row,1,&row,&diag,INSERT_VALUES);CHKERRQ(ierr);
13873bededecSBarry Smith     }
13883bededecSBarry Smith   }
13893bededecSBarry Smith   ierr = MatAssemblyEnd_SeqSBAIJ(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
13903bededecSBarry Smith   PetscFunctionReturn(0);
13913bededecSBarry Smith }
13923bededecSBarry Smith 
13937d68702bSBarry Smith #undef __FUNCT__
13947d68702bSBarry Smith #define __FUNCT__ "MatShift_SeqSBAIJ"
13957d68702bSBarry Smith PetscErrorCode MatShift_SeqSBAIJ(Mat Y,PetscScalar a)
13967d68702bSBarry Smith {
13977d68702bSBarry Smith   PetscErrorCode ierr;
13987d68702bSBarry Smith   Mat_SeqSBAIJ    *aij = (Mat_SeqSBAIJ*)Y->data;
13997d68702bSBarry Smith 
14007d68702bSBarry Smith   PetscFunctionBegin;
14016f33a894SBarry Smith   if (!Y->preallocated || !aij->nz) {
14027d68702bSBarry Smith     ierr = MatSeqSBAIJSetPreallocation(Y,Y->rmap->bs,1,NULL);CHKERRQ(ierr);
14037d68702bSBarry Smith   }
14047d68702bSBarry Smith   ierr = MatShift_Basic(Y,a);CHKERRQ(ierr);
14057d68702bSBarry Smith   PetscFunctionReturn(0);
14067d68702bSBarry Smith }
14077d68702bSBarry Smith 
140849b5e25fSSatish Balay /* -------------------------------------------------------------------*/
14093964eb88SJed Brown static struct _MatOps MatOps_Values = {MatSetValues_SeqSBAIJ,
141049b5e25fSSatish Balay                                        MatGetRow_SeqSBAIJ,
141149b5e25fSSatish Balay                                        MatRestoreRow_SeqSBAIJ,
141249b5e25fSSatish Balay                                        MatMult_SeqSBAIJ_N,
141397304618SKris Buschelman                                /*  4*/ MatMultAdd_SeqSBAIJ_N,
1414431c96f7SBarry Smith                                        MatMult_SeqSBAIJ_N,       /* transpose versions are same as non-transpose versions */
1415e005ede5SBarry Smith                                        MatMultAdd_SeqSBAIJ_N,
1416db4efbfdSBarry Smith                                        0,
141749b5e25fSSatish Balay                                        0,
141849b5e25fSSatish Balay                                        0,
141997304618SKris Buschelman                                /* 10*/ 0,
142049b5e25fSSatish Balay                                        0,
1421c078aec8SLisandro Dalcin                                        MatCholeskyFactor_SeqSBAIJ,
142241f059aeSBarry Smith                                        MatSOR_SeqSBAIJ,
142349b5e25fSSatish Balay                                        MatTranspose_SeqSBAIJ,
142497304618SKris Buschelman                                /* 15*/ MatGetInfo_SeqSBAIJ,
142549b5e25fSSatish Balay                                        MatEqual_SeqSBAIJ,
142649b5e25fSSatish Balay                                        MatGetDiagonal_SeqSBAIJ,
142749b5e25fSSatish Balay                                        MatDiagonalScale_SeqSBAIJ,
142849b5e25fSSatish Balay                                        MatNorm_SeqSBAIJ,
142997304618SKris Buschelman                                /* 20*/ 0,
143049b5e25fSSatish Balay                                        MatAssemblyEnd_SeqSBAIJ,
143149b5e25fSSatish Balay                                        MatSetOption_SeqSBAIJ,
143249b5e25fSSatish Balay                                        MatZeroEntries_SeqSBAIJ,
1433d519adbfSMatthew Knepley                                /* 24*/ 0,
143449b5e25fSSatish Balay                                        0,
143549b5e25fSSatish Balay                                        0,
1436db4efbfdSBarry Smith                                        0,
1437db4efbfdSBarry Smith                                        0,
14384994cf47SJed Brown                                /* 29*/ MatSetUp_SeqSBAIJ,
1439c464158bSHong Zhang                                        0,
1440db4efbfdSBarry Smith                                        0,
14418c778c55SBarry Smith                                        0,
14428c778c55SBarry Smith                                        0,
1443d519adbfSMatthew Knepley                                /* 34*/ MatDuplicate_SeqSBAIJ,
1444719d5645SBarry Smith                                        0,
1445719d5645SBarry Smith                                        0,
144649b5e25fSSatish Balay                                        0,
1447c84f5b01SHong Zhang                                        MatICCFactor_SeqSBAIJ,
1448d519adbfSMatthew Knepley                                /* 39*/ MatAXPY_SeqSBAIJ,
144949b5e25fSSatish Balay                                        MatGetSubMatrices_SeqSBAIJ,
145049b5e25fSSatish Balay                                        MatIncreaseOverlap_SeqSBAIJ,
145149b5e25fSSatish Balay                                        MatGetValues_SeqSBAIJ,
14523c896bc6SHong Zhang                                        MatCopy_SeqSBAIJ,
1453d519adbfSMatthew Knepley                                /* 44*/ 0,
145449b5e25fSSatish Balay                                        MatScale_SeqSBAIJ,
14557d68702bSBarry Smith                                        MatShift_SeqSBAIJ,
145649b5e25fSSatish Balay                                        0,
14573bededecSBarry Smith                                        MatZeroRowsColumns_SeqSBAIJ,
1458f73d5cc4SBarry Smith                                /* 49*/ 0,
145949b5e25fSSatish Balay                                        MatGetRowIJ_SeqSBAIJ,
146049b5e25fSSatish Balay                                        MatRestoreRowIJ_SeqSBAIJ,
146149b5e25fSSatish Balay                                        0,
146249b5e25fSSatish Balay                                        0,
1463d519adbfSMatthew Knepley                                /* 54*/ 0,
146449b5e25fSSatish Balay                                        0,
146549b5e25fSSatish Balay                                        0,
146649b5e25fSSatish Balay                                        0,
146749b5e25fSSatish Balay                                        MatSetValuesBlocked_SeqSBAIJ,
1468d519adbfSMatthew Knepley                                /* 59*/ MatGetSubMatrix_SeqSBAIJ,
146949b5e25fSSatish Balay                                        0,
147049b5e25fSSatish Balay                                        0,
1471357abbc8SBarry Smith                                        0,
1472d959ec07SHong Zhang                                        0,
1473d519adbfSMatthew Knepley                                /* 64*/ 0,
1474d959ec07SHong Zhang                                        0,
1475d959ec07SHong Zhang                                        0,
1476d959ec07SHong Zhang                                        0,
1477d959ec07SHong Zhang                                        0,
1478d519adbfSMatthew Knepley                                /* 69*/ MatGetRowMaxAbs_SeqSBAIJ,
14793e0d88b5SBarry Smith                                        0,
14803e0d88b5SBarry Smith                                        0,
14813e0d88b5SBarry Smith                                        0,
14823e0d88b5SBarry Smith                                        0,
1483d519adbfSMatthew Knepley                                /* 74*/ 0,
14843e0d88b5SBarry Smith                                        0,
14853e0d88b5SBarry Smith                                        0,
14863e0d88b5SBarry Smith                                        0,
14873e0d88b5SBarry Smith                                        0,
1488d519adbfSMatthew Knepley                                /* 79*/ 0,
14893e0d88b5SBarry Smith                                        0,
14903e0d88b5SBarry Smith                                        0,
149197304618SKris Buschelman                                        MatGetInertia_SeqSBAIJ,
14925bba2384SShri Abhyankar                                        MatLoad_SeqSBAIJ,
1493d519adbfSMatthew Knepley                                /* 84*/ MatIsSymmetric_SeqSBAIJ,
1494865e5f61SKris Buschelman                                        MatIsHermitian_SeqSBAIJ,
1495efcf0fc3SBarry Smith                                        MatIsStructurallySymmetric_SeqSBAIJ,
1496865e5f61SKris Buschelman                                        0,
1497865e5f61SKris Buschelman                                        0,
1498d519adbfSMatthew Knepley                                /* 89*/ 0,
1499865e5f61SKris Buschelman                                        0,
1500865e5f61SKris Buschelman                                        0,
1501865e5f61SKris Buschelman                                        0,
1502865e5f61SKris Buschelman                                        0,
1503d519adbfSMatthew Knepley                                /* 94*/ 0,
1504865e5f61SKris Buschelman                                        0,
1505865e5f61SKris Buschelman                                        0,
150699cafbc1SBarry Smith                                        0,
150799cafbc1SBarry Smith                                        0,
1508d519adbfSMatthew Knepley                                /* 99*/ 0,
150999cafbc1SBarry Smith                                        0,
151099cafbc1SBarry Smith                                        0,
151199cafbc1SBarry Smith                                        0,
151299cafbc1SBarry Smith                                        0,
1513d519adbfSMatthew Knepley                                /*104*/ 0,
151499cafbc1SBarry Smith                                        MatRealPart_SeqSBAIJ,
1515f5edf698SHong Zhang                                        MatImaginaryPart_SeqSBAIJ,
1516f5edf698SHong Zhang                                        MatGetRowUpperTriangular_SeqSBAIJ,
15172af78befSBarry Smith                                        MatRestoreRowUpperTriangular_SeqSBAIJ,
1518d519adbfSMatthew Knepley                                /*109*/ 0,
15192af78befSBarry Smith                                        0,
15202af78befSBarry Smith                                        0,
15212af78befSBarry Smith                                        0,
1522547795f9SHong Zhang                                        MatMissingDiagonal_SeqSBAIJ,
1523547795f9SHong Zhang                                /*114*/ 0,
1524547795f9SHong Zhang                                        0,
1525547795f9SHong Zhang                                        0,
1526547795f9SHong Zhang                                        0,
1527547795f9SHong Zhang                                        0,
1528547795f9SHong Zhang                                /*119*/ 0,
1529547795f9SHong Zhang                                        0,
15302f480046SShri Abhyankar                                        0,
15313964eb88SJed Brown                                        0,
15323964eb88SJed Brown                                        0,
15333964eb88SJed Brown                                /*124*/ 0,
15343964eb88SJed Brown                                        0,
15353964eb88SJed Brown                                        0,
15363964eb88SJed Brown                                        0,
15373964eb88SJed Brown                                        0,
15383964eb88SJed Brown                                /*129*/ 0,
15393964eb88SJed Brown                                        0,
15403964eb88SJed Brown                                        0,
15413964eb88SJed Brown                                        0,
15423964eb88SJed Brown                                        0,
15433964eb88SJed Brown                                /*134*/ 0,
15443964eb88SJed Brown                                        0,
15453964eb88SJed Brown                                        0,
15463964eb88SJed Brown                                        0,
15473964eb88SJed Brown                                        0,
154846533700Sstefano_zampini                                /*139*/ MatSetBlockSizes_Default,
1549f9426fe0SMark Adams                                        0,
155059f5e6ceSHong Zhang                                        0,
155159f5e6ceSHong Zhang                                        0,
155259f5e6ceSHong Zhang                                        0,
155359f5e6ceSHong Zhang                                 /*144*/MatCreateMPIMatConcatenateSeqMat_SeqSBAIJ
155499cafbc1SBarry Smith };
1555be1d678aSKris Buschelman 
15564a2ae208SSatish Balay #undef __FUNCT__
15574a2ae208SSatish Balay #define __FUNCT__ "MatStoreValues_SeqSBAIJ"
15587087cfbeSBarry Smith PetscErrorCode  MatStoreValues_SeqSBAIJ(Mat mat)
155949b5e25fSSatish Balay {
15604afc71dfSHong Zhang   Mat_SeqSBAIJ   *aij = (Mat_SeqSBAIJ*)mat->data;
1561d0f46423SBarry Smith   PetscInt       nz   = aij->i[mat->rmap->N]*mat->rmap->bs*aij->bs2;
1562dfbe8321SBarry Smith   PetscErrorCode ierr;
156349b5e25fSSatish Balay 
156449b5e25fSSatish Balay   PetscFunctionBegin;
1565e7e72b3dSBarry Smith   if (aij->nonew != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first");
156649b5e25fSSatish Balay 
156749b5e25fSSatish Balay   /* allocate space for values if not already there */
156849b5e25fSSatish Balay   if (!aij->saved_values) {
1569854ce69bSBarry Smith     ierr = PetscMalloc1(nz+1,&aij->saved_values);CHKERRQ(ierr);
157049b5e25fSSatish Balay   }
157149b5e25fSSatish Balay 
157249b5e25fSSatish Balay   /* copy values over */
157387828ca2SBarry Smith   ierr = PetscMemcpy(aij->saved_values,aij->a,nz*sizeof(PetscScalar));CHKERRQ(ierr);
157449b5e25fSSatish Balay   PetscFunctionReturn(0);
157549b5e25fSSatish Balay }
157649b5e25fSSatish Balay 
15774a2ae208SSatish Balay #undef __FUNCT__
15784a2ae208SSatish Balay #define __FUNCT__ "MatRetrieveValues_SeqSBAIJ"
15797087cfbeSBarry Smith PetscErrorCode  MatRetrieveValues_SeqSBAIJ(Mat mat)
158049b5e25fSSatish Balay {
15814afc71dfSHong Zhang   Mat_SeqSBAIJ   *aij = (Mat_SeqSBAIJ*)mat->data;
15826849ba73SBarry Smith   PetscErrorCode ierr;
1583d0f46423SBarry Smith   PetscInt       nz = aij->i[mat->rmap->N]*mat->rmap->bs*aij->bs2;
158449b5e25fSSatish Balay 
158549b5e25fSSatish Balay   PetscFunctionBegin;
1586e7e72b3dSBarry Smith   if (aij->nonew != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first");
1587e7e72b3dSBarry Smith   if (!aij->saved_values) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatStoreValues(A);first");
158849b5e25fSSatish Balay 
158949b5e25fSSatish Balay   /* copy values over */
159087828ca2SBarry Smith   ierr = PetscMemcpy(aij->a,aij->saved_values,nz*sizeof(PetscScalar));CHKERRQ(ierr);
159149b5e25fSSatish Balay   PetscFunctionReturn(0);
159249b5e25fSSatish Balay }
159349b5e25fSSatish Balay 
15944a2ae208SSatish Balay #undef __FUNCT__
1595a23d5eceSKris Buschelman #define __FUNCT__ "MatSeqSBAIJSetPreallocation_SeqSBAIJ"
1596367daffbSBarry Smith static PetscErrorCode  MatSeqSBAIJSetPreallocation_SeqSBAIJ(Mat B,PetscInt bs,PetscInt nz,PetscInt *nnz)
159749b5e25fSSatish Balay {
1598c464158bSHong Zhang   Mat_SeqSBAIJ   *b = (Mat_SeqSBAIJ*)B->data;
15996849ba73SBarry Smith   PetscErrorCode ierr;
16004dcd73b1SHong Zhang   PetscInt       i,mbs,nbs,bs2;
16012576faa2SJed Brown   PetscBool      skipallocation = PETSC_FALSE,flg = PETSC_FALSE,realalloc = PETSC_FALSE;
160249b5e25fSSatish Balay 
160349b5e25fSSatish Balay   PetscFunctionBegin;
16042576faa2SJed Brown   if (nz >= 0 || nnz) realalloc = PETSC_TRUE;
1605db4efbfdSBarry Smith 
160633d57670SJed Brown   ierr = MatSetBlockSize(B,PetscAbs(bs));CHKERRQ(ierr);
160726283091SBarry Smith   ierr = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr);
160826283091SBarry Smith   ierr = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr);
1609e02043d6SBarry Smith   ierr = PetscLayoutGetBlockSize(B->rmap,&bs);CHKERRQ(ierr);
1610899cda47SBarry Smith 
1611*21940c7eSstefano_zampini   B->preallocated = PETSC_TRUE;
1612*21940c7eSstefano_zampini 
1613d0f46423SBarry Smith   mbs = B->rmap->N/bs;
16144dcd73b1SHong Zhang   nbs = B->cmap->n/bs;
161549b5e25fSSatish Balay   bs2 = bs*bs;
161649b5e25fSSatish Balay 
16174dcd73b1SHong 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");
161849b5e25fSSatish Balay 
1619ab93d7beSBarry Smith   if (nz == MAT_SKIP_ALLOCATION) {
1620ab93d7beSBarry Smith     skipallocation = PETSC_TRUE;
1621ab93d7beSBarry Smith     nz             = 0;
1622ab93d7beSBarry Smith   }
1623ab93d7beSBarry Smith 
1624435da068SBarry Smith   if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 3;
1625e32f2f54SBarry Smith   if (nz < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"nz cannot be less than 0: value %D",nz);
162649b5e25fSSatish Balay   if (nnz) {
162749b5e25fSSatish Balay     for (i=0; i<mbs; i++) {
1628e32f2f54SBarry Smith       if (nnz[i] < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"nnz cannot be less than 0: local row %D value %D",i,nnz[i]);
1629de64b629SHong Zhang       if (nnz[i] > nbs) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"nnz cannot be greater than block row length: local row %D value %D block rowlength %D",i,nnz[i],nbs);
163049b5e25fSSatish Balay     }
163149b5e25fSSatish Balay   }
163249b5e25fSSatish Balay 
1633db4efbfdSBarry Smith   B->ops->mult             = MatMult_SeqSBAIJ_N;
1634db4efbfdSBarry Smith   B->ops->multadd          = MatMultAdd_SeqSBAIJ_N;
1635db4efbfdSBarry Smith   B->ops->multtranspose    = MatMult_SeqSBAIJ_N;
1636db4efbfdSBarry Smith   B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_N;
163726fbe8dcSKarl Rupp 
1638c5929fdfSBarry Smith   ierr  = PetscOptionsGetBool(((PetscObject)B)->options,((PetscObject)B)->prefix,"-mat_no_unroll",&flg,NULL);CHKERRQ(ierr);
163949b5e25fSSatish Balay   if (!flg) {
164049b5e25fSSatish Balay     switch (bs) {
164149b5e25fSSatish Balay     case 1:
164249b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_1;
164349b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_1;
1644431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_1;
1645431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_1;
164649b5e25fSSatish Balay       break;
164749b5e25fSSatish Balay     case 2:
164849b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_2;
164949b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_2;
1650431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_2;
1651431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_2;
165249b5e25fSSatish Balay       break;
165349b5e25fSSatish Balay     case 3:
165449b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_3;
165549b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_3;
1656431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_3;
1657431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_3;
165849b5e25fSSatish Balay       break;
165949b5e25fSSatish Balay     case 4:
166049b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_4;
166149b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_4;
1662431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_4;
1663431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_4;
166449b5e25fSSatish Balay       break;
166549b5e25fSSatish Balay     case 5:
166649b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_5;
166749b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_5;
1668431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_5;
1669431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_5;
167049b5e25fSSatish Balay       break;
167149b5e25fSSatish Balay     case 6:
167249b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_6;
167349b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_6;
1674431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_6;
1675431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_6;
167649b5e25fSSatish Balay       break;
167749b5e25fSSatish Balay     case 7:
1678de53e5efSHong Zhang       B->ops->mult             = MatMult_SeqSBAIJ_7;
167949b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_7;
1680431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_7;
1681431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_7;
168249b5e25fSSatish Balay       break;
168349b5e25fSSatish Balay     }
168449b5e25fSSatish Balay   }
168549b5e25fSSatish Balay 
168649b5e25fSSatish Balay   b->mbs = mbs;
16874dcd73b1SHong Zhang   b->nbs = nbs;
1688ab93d7beSBarry Smith   if (!skipallocation) {
16892ee49352SLisandro Dalcin     if (!b->imax) {
1690dcca6d9dSJed Brown       ierr = PetscMalloc2(mbs,&b->imax,mbs,&b->ilen);CHKERRQ(ierr);
169126fbe8dcSKarl Rupp 
1692c760cd28SBarry Smith       b->free_imax_ilen = PETSC_TRUE;
169326fbe8dcSKarl Rupp 
16943bb1ff40SBarry Smith       ierr = PetscLogObjectMemory((PetscObject)B,2*mbs*sizeof(PetscInt));CHKERRQ(ierr);
16952ee49352SLisandro Dalcin     }
169649b5e25fSSatish Balay     if (!nnz) {
1697435da068SBarry Smith       if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 5;
169849b5e25fSSatish Balay       else if (nz <= 0) nz = 1;
169926fbe8dcSKarl Rupp       for (i=0; i<mbs; i++) b->imax[i] = nz;
1700153ea458SHong Zhang       nz = nz*mbs; /* total nz */
170149b5e25fSSatish Balay     } else {
170249b5e25fSSatish Balay       nz = 0;
17038cef66ccSHong Zhang       for (i=0; i<mbs; i++) {b->imax[i] = nnz[i]; nz += nnz[i];}
170449b5e25fSSatish Balay     }
17052ee49352SLisandro Dalcin     /* b->ilen will count nonzeros in each block row so far. */
170626fbe8dcSKarl Rupp     for (i=0; i<mbs; i++) b->ilen[i] = 0;
17076c6c5352SBarry Smith     /* nz=(nz+mbs)/2; */ /* total diagonal and superdiagonal nonzero blocks */
170849b5e25fSSatish Balay 
170949b5e25fSSatish Balay     /* allocate the matrix space */
17102ee49352SLisandro Dalcin     ierr = MatSeqXAIJFreeAIJ(B,&b->a,&b->j,&b->i);CHKERRQ(ierr);
1711dcca6d9dSJed Brown     ierr = PetscMalloc3(bs2*nz,&b->a,nz,&b->j,B->rmap->N+1,&b->i);CHKERRQ(ierr);
17123bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)B,(B->rmap->N+1)*sizeof(PetscInt)+nz*(bs2*sizeof(PetscScalar)+sizeof(PetscInt)));CHKERRQ(ierr);
17136c6c5352SBarry Smith     ierr = PetscMemzero(b->a,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr);
171413f74950SBarry Smith     ierr = PetscMemzero(b->j,nz*sizeof(PetscInt));CHKERRQ(ierr);
171526fbe8dcSKarl Rupp 
171649b5e25fSSatish Balay     b->singlemalloc = PETSC_TRUE;
171749b5e25fSSatish Balay 
171849b5e25fSSatish Balay     /* pointer to beginning of each row */
1719e60cf9a0SBarry Smith     b->i[0] = 0;
172026fbe8dcSKarl Rupp     for (i=1; i<mbs+1; i++) b->i[i] = b->i[i-1] + b->imax[i-1];
172126fbe8dcSKarl Rupp 
1722e6b907acSBarry Smith     b->free_a  = PETSC_TRUE;
1723e6b907acSBarry Smith     b->free_ij = PETSC_TRUE;
1724e811da20SHong Zhang   } else {
1725e6b907acSBarry Smith     b->free_a  = PETSC_FALSE;
1726e6b907acSBarry Smith     b->free_ij = PETSC_FALSE;
1727ab93d7beSBarry Smith   }
172849b5e25fSSatish Balay 
172949b5e25fSSatish Balay   b->bs2     = bs2;
17306c6c5352SBarry Smith   b->nz      = 0;
1731b32cb4a7SJed Brown   b->maxnz   = nz;
173216cdd363SHong Zhang   b->inew    = 0;
173316cdd363SHong Zhang   b->jnew    = 0;
173416cdd363SHong Zhang   b->anew    = 0;
173516cdd363SHong Zhang   b->a2anew  = 0;
17361a3463dfSHong Zhang   b->permute = PETSC_FALSE;
17372576faa2SJed Brown   if (realalloc) {ierr = MatSetOption(B,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);}
1738c464158bSHong Zhang   PetscFunctionReturn(0);
1739c464158bSHong Zhang }
1740153ea458SHong Zhang 
174138f409ebSLisandro Dalcin #undef __FUNCT__
174238f409ebSLisandro Dalcin #define __FUNCT__ "MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ"
174338f409ebSLisandro Dalcin PetscErrorCode MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ(Mat B,PetscInt bs,const PetscInt ii[],const PetscInt jj[], const PetscScalar V[])
174438f409ebSLisandro Dalcin {
174538f409ebSLisandro Dalcin   PetscInt       i,j,m,nz,nz_max=0,*nnz;
174638f409ebSLisandro Dalcin   PetscScalar    *values=0;
174738f409ebSLisandro Dalcin   PetscBool      roworiented = ((Mat_SeqSBAIJ*)B->data)->roworiented;
174838f409ebSLisandro Dalcin   PetscErrorCode ierr;
174938f409ebSLisandro Dalcin   PetscFunctionBegin;
175038f409ebSLisandro Dalcin   if (bs < 1) SETERRQ1(PetscObjectComm((PetscObject)B),PETSC_ERR_ARG_OUTOFRANGE,"Invalid block size specified, must be positive but it is %D",bs);
175138f409ebSLisandro Dalcin   ierr   = PetscLayoutSetBlockSize(B->rmap,bs);CHKERRQ(ierr);
175238f409ebSLisandro Dalcin   ierr   = PetscLayoutSetBlockSize(B->cmap,bs);CHKERRQ(ierr);
175338f409ebSLisandro Dalcin   ierr   = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr);
175438f409ebSLisandro Dalcin   ierr   = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr);
175538f409ebSLisandro Dalcin   ierr   = PetscLayoutGetBlockSize(B->rmap,&bs);CHKERRQ(ierr);
175638f409ebSLisandro Dalcin   m      = B->rmap->n/bs;
175738f409ebSLisandro Dalcin 
175838f409ebSLisandro Dalcin   if (ii[0]) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"ii[0] must be 0 but it is %D",ii[0]);
1759854ce69bSBarry Smith   ierr = PetscMalloc1(m+1,&nnz);CHKERRQ(ierr);
176038f409ebSLisandro Dalcin   for (i=0; i<m; i++) {
176138f409ebSLisandro Dalcin     nz = ii[i+1] - ii[i];
176238f409ebSLisandro Dalcin     if (nz < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row %D has a negative number of columns %D",i,nz);
176338f409ebSLisandro Dalcin     nz_max = PetscMax(nz_max,nz);
176438f409ebSLisandro Dalcin     nnz[i] = nz;
176538f409ebSLisandro Dalcin   }
176638f409ebSLisandro Dalcin   ierr = MatSeqSBAIJSetPreallocation(B,bs,0,nnz);CHKERRQ(ierr);
176738f409ebSLisandro Dalcin   ierr = PetscFree(nnz);CHKERRQ(ierr);
176838f409ebSLisandro Dalcin 
176938f409ebSLisandro Dalcin   values = (PetscScalar*)V;
177038f409ebSLisandro Dalcin   if (!values) {
17711795a4d1SJed Brown     ierr = PetscCalloc1(bs*bs*nz_max,&values);CHKERRQ(ierr);
177238f409ebSLisandro Dalcin   }
177338f409ebSLisandro Dalcin   for (i=0; i<m; i++) {
177438f409ebSLisandro Dalcin     PetscInt          ncols  = ii[i+1] - ii[i];
177538f409ebSLisandro Dalcin     const PetscInt    *icols = jj + ii[i];
177638f409ebSLisandro Dalcin     if (!roworiented || bs == 1) {
177738f409ebSLisandro Dalcin       const PetscScalar *svals = values + (V ? (bs*bs*ii[i]) : 0);
177838f409ebSLisandro Dalcin       ierr = MatSetValuesBlocked_SeqSBAIJ(B,1,&i,ncols,icols,svals,INSERT_VALUES);CHKERRQ(ierr);
177938f409ebSLisandro Dalcin     } else {
178038f409ebSLisandro Dalcin       for (j=0; j<ncols; j++) {
178138f409ebSLisandro Dalcin         const PetscScalar *svals = values + (V ? (bs*bs*(ii[i]+j)) : 0);
178238f409ebSLisandro Dalcin         ierr = MatSetValuesBlocked_SeqSBAIJ(B,1,&i,1,&icols[j],svals,INSERT_VALUES);CHKERRQ(ierr);
178338f409ebSLisandro Dalcin       }
178438f409ebSLisandro Dalcin     }
178538f409ebSLisandro Dalcin   }
178638f409ebSLisandro Dalcin   if (!V) { ierr = PetscFree(values);CHKERRQ(ierr); }
178738f409ebSLisandro Dalcin   ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
178838f409ebSLisandro Dalcin   ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
178938f409ebSLisandro Dalcin   ierr = MatSetOption(B,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
179038f409ebSLisandro Dalcin   PetscFunctionReturn(0);
179138f409ebSLisandro Dalcin }
179238f409ebSLisandro Dalcin 
1793db4efbfdSBarry Smith /*
1794db4efbfdSBarry Smith    This is used to set the numeric factorization for both Cholesky and ICC symbolic factorization
1795db4efbfdSBarry Smith */
17968b1456e3SHong Zhang #undef __FUNCT__
1797d595f711SHong Zhang #define __FUNCT__ "MatSeqSBAIJSetNumericFactorization_inplace"
1798ace3abfcSBarry Smith PetscErrorCode MatSeqSBAIJSetNumericFactorization_inplace(Mat B,PetscBool natural)
1799db4efbfdSBarry Smith {
1800db4efbfdSBarry Smith   PetscErrorCode ierr;
1801ace3abfcSBarry Smith   PetscBool      flg = PETSC_FALSE;
1802db4efbfdSBarry Smith   PetscInt       bs  = B->rmap->bs;
1803db4efbfdSBarry Smith 
1804db4efbfdSBarry Smith   PetscFunctionBegin;
1805c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(((PetscObject)B)->options,((PetscObject)B)->prefix,"-mat_no_unroll",&flg,NULL);CHKERRQ(ierr);
1806db4efbfdSBarry Smith   if (flg) bs = 8;
1807db4efbfdSBarry Smith 
1808db4efbfdSBarry Smith   if (!natural) {
1809db4efbfdSBarry Smith     switch (bs) {
1810db4efbfdSBarry Smith     case 1:
1811d595f711SHong Zhang       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_inplace;
1812db4efbfdSBarry Smith       break;
1813db4efbfdSBarry Smith     case 2:
1814db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2;
1815db4efbfdSBarry Smith       break;
1816db4efbfdSBarry Smith     case 3:
1817db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3;
1818db4efbfdSBarry Smith       break;
1819db4efbfdSBarry Smith     case 4:
1820db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4;
1821db4efbfdSBarry Smith       break;
1822db4efbfdSBarry Smith     case 5:
1823db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5;
1824db4efbfdSBarry Smith       break;
1825db4efbfdSBarry Smith     case 6:
1826db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6;
1827db4efbfdSBarry Smith       break;
1828db4efbfdSBarry Smith     case 7:
1829db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7;
1830db4efbfdSBarry Smith       break;
1831db4efbfdSBarry Smith     default:
1832db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N;
1833db4efbfdSBarry Smith       break;
1834db4efbfdSBarry Smith     }
1835db4efbfdSBarry Smith   } else {
1836db4efbfdSBarry Smith     switch (bs) {
1837db4efbfdSBarry Smith     case 1:
1838d595f711SHong Zhang       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering_inplace;
1839db4efbfdSBarry Smith       break;
1840db4efbfdSBarry Smith     case 2:
1841db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering;
1842db4efbfdSBarry Smith       break;
1843db4efbfdSBarry Smith     case 3:
1844db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering;
1845db4efbfdSBarry Smith       break;
1846db4efbfdSBarry Smith     case 4:
1847db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering;
1848db4efbfdSBarry Smith       break;
1849db4efbfdSBarry Smith     case 5:
1850db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering;
1851db4efbfdSBarry Smith       break;
1852db4efbfdSBarry Smith     case 6:
1853db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering;
1854db4efbfdSBarry Smith       break;
1855db4efbfdSBarry Smith     case 7:
1856db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering;
1857db4efbfdSBarry Smith       break;
1858db4efbfdSBarry Smith     default:
1859db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering;
1860db4efbfdSBarry Smith       break;
1861db4efbfdSBarry Smith     }
1862db4efbfdSBarry Smith   }
1863db4efbfdSBarry Smith   PetscFunctionReturn(0);
1864db4efbfdSBarry Smith }
1865db4efbfdSBarry Smith 
1866cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqAIJ(Mat, MatType,MatReuse,Mat*);
1867cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqBAIJ(Mat, MatType,MatReuse,Mat*);
1868d769727bSBarry Smith 
18695c9eb25fSBarry Smith #undef __FUNCT__
18705c9eb25fSBarry Smith #define __FUNCT__ "MatGetFactor_seqsbaij_petsc"
1871cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatGetFactor_seqsbaij_petsc(Mat A,MatFactorType ftype,Mat *B)
18725c9eb25fSBarry Smith {
1873d0f46423SBarry Smith   PetscInt       n = A->rmap->n;
18745c9eb25fSBarry Smith   PetscErrorCode ierr;
18755c9eb25fSBarry Smith 
18765c9eb25fSBarry Smith   PetscFunctionBegin;
18770e92d65fSHong Zhang #if defined(PETSC_USE_COMPLEX)
18780e92d65fSHong Zhang   if (A->hermitian) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Hermitian Factor is not supported");
18790e92d65fSHong Zhang #endif
1880ce94432eSBarry Smith   ierr = MatCreate(PetscObjectComm((PetscObject)A),B);CHKERRQ(ierr);
18815c9eb25fSBarry Smith   ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr);
18825c9eb25fSBarry Smith   if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) {
18835c9eb25fSBarry Smith     ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr);
18840298fd71SBarry Smith     ierr = MatSeqSBAIJSetPreallocation(*B,A->rmap->bs,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr);
188526fbe8dcSKarl Rupp 
18867b056e98SHong Zhang     (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqSBAIJ;
1887c6d0d4f0SHong Zhang     (*B)->ops->iccfactorsymbolic      = MatICCFactorSymbolic_SeqSBAIJ;
1888e32f2f54SBarry Smith   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Factor type not supported");
188900c67f3bSHong Zhang 
1890d5f3da31SBarry Smith   (*B)->factortype = ftype;
189100c67f3bSHong Zhang   ierr = PetscFree((*B)->solvertype);CHKERRQ(ierr);
189200c67f3bSHong Zhang   ierr = PetscStrallocpy(MATSOLVERPETSC,&(*B)->solvertype);CHKERRQ(ierr);
18935c9eb25fSBarry Smith   PetscFunctionReturn(0);
18945c9eb25fSBarry Smith }
18955c9eb25fSBarry Smith 
18960bad9183SKris Buschelman /*MC
1897fafad747SKris Buschelman   MATSEQSBAIJ - MATSEQSBAIJ = "seqsbaij" - A matrix type to be used for sequential symmetric block sparse matrices,
18980bad9183SKris Buschelman   based on block compressed sparse row format.  Only the upper triangular portion of the matrix is stored.
18990bad9183SKris Buschelman 
1900828413b8SBarry Smith   For complex numbers by default this matrix is symmetric, NOT Hermitian symmetric. To make it Hermitian symmetric you
190171dad5bbSBarry Smith   can call MatSetOption(Mat, MAT_HERMITIAN); after MatAssemblyEnd()
1902828413b8SBarry Smith 
19030bad9183SKris Buschelman   Options Database Keys:
19040bad9183SKris Buschelman   . -mat_type seqsbaij - sets the matrix type to "seqsbaij" during a call to MatSetFromOptions()
19050bad9183SKris Buschelman 
190671dad5bbSBarry Smith   Notes: By default if you insert values into the lower triangular part of the matrix they are simply ignored (since they are not
190771dad5bbSBarry 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
190871dad5bbSBarry 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.
190971dad5bbSBarry Smith 
191071dad5bbSBarry Smith 
19110bad9183SKris Buschelman   Level: beginner
19120bad9183SKris Buschelman 
19130bad9183SKris Buschelman   .seealso: MatCreateSeqSBAIJ
19140bad9183SKris Buschelman M*/
19150bad9183SKris Buschelman 
1916cc2e6a90SBarry Smith PETSC_INTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqSBSTRM(Mat, MatType,MatReuse,Mat*);
1917aa5a9175SDahai Guo 
1918a23d5eceSKris Buschelman #undef __FUNCT__
1919a23d5eceSKris Buschelman #define __FUNCT__ "MatCreate_SeqSBAIJ"
19208cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatCreate_SeqSBAIJ(Mat B)
1921a23d5eceSKris Buschelman {
1922a23d5eceSKris Buschelman   Mat_SeqSBAIJ   *b;
1923dfbe8321SBarry Smith   PetscErrorCode ierr;
192413f74950SBarry Smith   PetscMPIInt    size;
1925ace3abfcSBarry Smith   PetscBool      no_unroll = PETSC_FALSE,no_inode = PETSC_FALSE;
1926a23d5eceSKris Buschelman 
1927a23d5eceSKris Buschelman   PetscFunctionBegin;
1928ce94432eSBarry Smith   ierr = MPI_Comm_size(PetscObjectComm((PetscObject)B),&size);CHKERRQ(ierr);
1929e32f2f54SBarry Smith   if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Comm must be of size 1");
1930a23d5eceSKris Buschelman 
1931b00a9115SJed Brown   ierr    = PetscNewLog(B,&b);CHKERRQ(ierr);
1932a23d5eceSKris Buschelman   B->data = (void*)b;
1933a23d5eceSKris Buschelman   ierr    = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr);
193426fbe8dcSKarl Rupp 
1935a23d5eceSKris Buschelman   B->ops->destroy    = MatDestroy_SeqSBAIJ;
1936a23d5eceSKris Buschelman   B->ops->view       = MatView_SeqSBAIJ;
1937a23d5eceSKris Buschelman   b->row             = 0;
1938a23d5eceSKris Buschelman   b->icol            = 0;
1939a23d5eceSKris Buschelman   b->reallocs        = 0;
1940a23d5eceSKris Buschelman   b->saved_values    = 0;
19410def2e27SBarry Smith   b->inode.limit     = 5;
19420def2e27SBarry Smith   b->inode.max_limit = 5;
1943a23d5eceSKris Buschelman 
1944a23d5eceSKris Buschelman   b->roworiented        = PETSC_TRUE;
1945a23d5eceSKris Buschelman   b->nonew              = 0;
1946a23d5eceSKris Buschelman   b->diag               = 0;
1947a23d5eceSKris Buschelman   b->solve_work         = 0;
1948a23d5eceSKris Buschelman   b->mult_work          = 0;
1949a23d5eceSKris Buschelman   B->spptr              = 0;
1950f2cbd3d5SJed Brown   B->info.nz_unneeded   = (PetscReal)b->maxnz*b->bs2;
1951a9817697SBarry Smith   b->keepnonzeropattern = PETSC_FALSE;
1952a23d5eceSKris Buschelman 
1953a23d5eceSKris Buschelman   b->inew    = 0;
1954a23d5eceSKris Buschelman   b->jnew    = 0;
1955a23d5eceSKris Buschelman   b->anew    = 0;
1956a23d5eceSKris Buschelman   b->a2anew  = 0;
1957a23d5eceSKris Buschelman   b->permute = PETSC_FALSE;
1958a23d5eceSKris Buschelman 
195971dad5bbSBarry Smith   b->ignore_ltriangular = PETSC_TRUE;
196026fbe8dcSKarl Rupp 
1961c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(((PetscObject)B)->options,((PetscObject)B)->prefix,"-mat_ignore_lower_triangular",&b->ignore_ltriangular,NULL);CHKERRQ(ierr);
1962941593c8SHong Zhang 
1963f5edf698SHong Zhang   b->getrow_utriangular = PETSC_FALSE;
196426fbe8dcSKarl Rupp 
1965c5929fdfSBarry Smith   ierr = PetscOptionsGetBool(((PetscObject)B)->options,((PetscObject)B)->prefix,"-mat_getrow_uppertriangular",&b->getrow_utriangular,NULL);CHKERRQ(ierr);
1966f5edf698SHong Zhang 
1967bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatStoreValues_C",MatStoreValues_SeqSBAIJ);CHKERRQ(ierr);
1968bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatRetrieveValues_C",MatRetrieveValues_SeqSBAIJ);CHKERRQ(ierr);
1969bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetColumnIndices_C",MatSeqSBAIJSetColumnIndices_SeqSBAIJ);CHKERRQ(ierr);
1970bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqaij_C",MatConvert_SeqSBAIJ_SeqAIJ);CHKERRQ(ierr);
1971bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqbaij_C",MatConvert_SeqSBAIJ_SeqBAIJ);CHKERRQ(ierr);
1972bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetPreallocation_C",MatSeqSBAIJSetPreallocation_SeqSBAIJ);CHKERRQ(ierr);
197338f409ebSLisandro Dalcin   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetPreallocationCSR_C",MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ);CHKERRQ(ierr);
19746214f412SHong Zhang #if defined(PETSC_HAVE_ELEMENTAL)
19756214f412SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_elemental_C",MatConvert_SeqSBAIJ_Elemental);CHKERRQ(ierr);
19766214f412SHong Zhang #endif
197723ce1328SBarry Smith 
197823ce1328SBarry Smith   B->symmetric                  = PETSC_TRUE;
197923ce1328SBarry Smith   B->structurally_symmetric     = PETSC_TRUE;
198023ce1328SBarry Smith   B->symmetric_set              = PETSC_TRUE;
198123ce1328SBarry Smith   B->structurally_symmetric_set = PETSC_TRUE;
198226fbe8dcSKarl Rupp 
198317667f90SBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)B,MATSEQSBAIJ);CHKERRQ(ierr);
19840def2e27SBarry Smith 
1985ce94432eSBarry Smith   ierr = PetscOptionsBegin(PetscObjectComm((PetscObject)B),((PetscObject)B)->prefix,"Options for SEQSBAIJ matrix","Mat");CHKERRQ(ierr);
19860298fd71SBarry Smith   ierr = PetscOptionsBool("-mat_no_unroll","Do not optimize for inodes (slower)",NULL,no_unroll,&no_unroll,NULL);CHKERRQ(ierr);
198726fbe8dcSKarl Rupp   if (no_unroll) {
198826fbe8dcSKarl Rupp     ierr = PetscInfo(B,"Not using Inode routines due to -mat_no_unroll\n");CHKERRQ(ierr);
198926fbe8dcSKarl Rupp   }
19900298fd71SBarry Smith   ierr = PetscOptionsBool("-mat_no_inode","Do not optimize for inodes (slower)",NULL,no_inode,&no_inode,NULL);CHKERRQ(ierr);
199126fbe8dcSKarl Rupp   if (no_inode) {
199226fbe8dcSKarl Rupp     ierr = PetscInfo(B,"Not using Inode routines due to -mat_no_inode\n");CHKERRQ(ierr);
199326fbe8dcSKarl Rupp   }
19940298fd71SBarry Smith   ierr = PetscOptionsInt("-mat_inode_limit","Do not use inodes larger then this value",NULL,b->inode.limit,&b->inode.limit,NULL);CHKERRQ(ierr);
19950def2e27SBarry Smith   ierr = PetscOptionsEnd();CHKERRQ(ierr);
1996ace3abfcSBarry Smith   b->inode.use = (PetscBool)(!(no_unroll || no_inode));
19970def2e27SBarry Smith   if (b->inode.limit > b->inode.max_limit) b->inode.limit = b->inode.max_limit;
1998a23d5eceSKris Buschelman   PetscFunctionReturn(0);
1999a23d5eceSKris Buschelman }
2000a23d5eceSKris Buschelman 
2001a23d5eceSKris Buschelman #undef __FUNCT__
2002a23d5eceSKris Buschelman #define __FUNCT__ "MatSeqSBAIJSetPreallocation"
2003a23d5eceSKris Buschelman /*@C
2004a23d5eceSKris Buschelman    MatSeqSBAIJSetPreallocation - Creates a sparse symmetric matrix in block AIJ (block
2005a23d5eceSKris Buschelman    compressed row) format.  For good matrix assembly performance the
2006a23d5eceSKris Buschelman    user should preallocate the matrix storage by setting the parameter nz
2007a23d5eceSKris Buschelman    (or the array nnz).  By setting these parameters accurately, performance
2008a23d5eceSKris Buschelman    during matrix assembly can be increased by more than a factor of 50.
2009a23d5eceSKris Buschelman 
2010a23d5eceSKris Buschelman    Collective on Mat
2011a23d5eceSKris Buschelman 
2012a23d5eceSKris Buschelman    Input Parameters:
20131c4f3114SJed Brown +  B - the symmetric matrix
2014bb7ae925SBarry 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
2015bb7ae925SBarry Smith           blocksize always defines the size of the blocks. The column blocksize sets the blocksize of the vectors obtained with MatCreateVecs()
2016a23d5eceSKris Buschelman .  nz - number of block nonzeros per block row (same for all rows)
2017a23d5eceSKris Buschelman -  nnz - array containing the number of block nonzeros in the upper triangular plus
20180298fd71SBarry Smith          diagonal portion of each block (possibly different for each block row) or NULL
2019a23d5eceSKris Buschelman 
2020a23d5eceSKris Buschelman    Options Database Keys:
2021a23d5eceSKris Buschelman .   -mat_no_unroll - uses code that does not unroll the loops in the
2022a23d5eceSKris Buschelman                      block calculations (much slower)
2023db4efbfdSBarry Smith .   -mat_block_size - size of the blocks to use (only works if a negative bs is passed in
2024a23d5eceSKris Buschelman 
2025a23d5eceSKris Buschelman    Level: intermediate
2026a23d5eceSKris Buschelman 
2027a23d5eceSKris Buschelman    Notes:
2028a23d5eceSKris Buschelman    Specify the preallocated storage with either nz or nnz (not both).
20290298fd71SBarry Smith    Set nz=PETSC_DEFAULT and nnz=NULL for PETSc to control dynamic memory
2030a7f22e61SSatish Balay    allocation.  See Users-Manual: ch_mat for details.
2031a23d5eceSKris Buschelman 
2032aa95bbe8SBarry Smith    You can call MatGetInfo() to get information on how effective the preallocation was;
2033aa95bbe8SBarry Smith    for example the fields mallocs,nz_allocated,nz_used,nz_unneeded;
2034aa95bbe8SBarry Smith    You can also run with the option -info and look for messages with the string
2035aa95bbe8SBarry Smith    malloc in them to see if additional memory allocation was needed.
2036aa95bbe8SBarry Smith 
203749a6f317SBarry Smith    If the nnz parameter is given then the nz parameter is ignored
203849a6f317SBarry Smith 
203949a6f317SBarry Smith 
204069b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateSBAIJ()
2041a23d5eceSKris Buschelman @*/
20427087cfbeSBarry Smith PetscErrorCode  MatSeqSBAIJSetPreallocation(Mat B,PetscInt bs,PetscInt nz,const PetscInt nnz[])
204313f74950SBarry Smith {
20444ac538c5SBarry Smith   PetscErrorCode ierr;
2045a23d5eceSKris Buschelman 
2046a23d5eceSKris Buschelman   PetscFunctionBegin;
20476ba663aaSJed Brown   PetscValidHeaderSpecific(B,MAT_CLASSID,1);
20486ba663aaSJed Brown   PetscValidType(B,1);
20496ba663aaSJed Brown   PetscValidLogicalCollectiveInt(B,bs,2);
20504ac538c5SBarry Smith   ierr = PetscTryMethod(B,"MatSeqSBAIJSetPreallocation_C",(Mat,PetscInt,PetscInt,const PetscInt[]),(B,bs,nz,nnz));CHKERRQ(ierr);
2051a23d5eceSKris Buschelman   PetscFunctionReturn(0);
2052a23d5eceSKris Buschelman }
205349b5e25fSSatish Balay 
20544a2ae208SSatish Balay #undef  __FUNCT__
205538f409ebSLisandro Dalcin #define __FUNCT__ "MatSeqSBAIJSetPreallocationCSR"
205638f409ebSLisandro Dalcin /*@C
205738f409ebSLisandro Dalcin    MatSeqSBAIJSetPreallocationCSR - Allocates memory for a sparse sequential matrix in symmetric block AIJ format.
205838f409ebSLisandro Dalcin 
205938f409ebSLisandro Dalcin    Input Parameters:
20601c4f3114SJed Brown +  B - the matrix
2061eab78319SHong Zhang .  bs - size of block, the blocks are ALWAYS square.
206238f409ebSLisandro Dalcin .  i - the indices into j for the start of each local row (starts with zero)
206338f409ebSLisandro Dalcin .  j - the column indices for each local row (starts with zero) these must be sorted for each row
206438f409ebSLisandro Dalcin -  v - optional values in the matrix
206538f409ebSLisandro Dalcin 
206638f409ebSLisandro Dalcin    Level: developer
206738f409ebSLisandro Dalcin 
206838f409ebSLisandro Dalcin    Notes:
206938f409ebSLisandro Dalcin    The order of the entries in values is specified by the MatOption MAT_ROW_ORIENTED.  For example, C programs
207038f409ebSLisandro 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
207138f409ebSLisandro Dalcin    over rows within a block and the last index is over columns within a block row.  Fortran programs will likely set
207238f409ebSLisandro 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
207338f409ebSLisandro Dalcin    block column and the second index is over columns within a block.
207438f409ebSLisandro Dalcin 
207538f409ebSLisandro Dalcin .keywords: matrix, block, aij, compressed row, sparse
207638f409ebSLisandro Dalcin 
207738f409ebSLisandro Dalcin .seealso: MatCreate(), MatCreateSeqSBAIJ(), MatSetValuesBlocked(), MatSeqSBAIJSetPreallocation(), MATSEQSBAIJ
207838f409ebSLisandro Dalcin @*/
207938f409ebSLisandro Dalcin PetscErrorCode MatSeqSBAIJSetPreallocationCSR(Mat B,PetscInt bs,const PetscInt i[],const PetscInt j[], const PetscScalar v[])
208038f409ebSLisandro Dalcin {
208138f409ebSLisandro Dalcin   PetscErrorCode ierr;
208238f409ebSLisandro Dalcin 
208338f409ebSLisandro Dalcin   PetscFunctionBegin;
208438f409ebSLisandro Dalcin   PetscValidHeaderSpecific(B,MAT_CLASSID,1);
208538f409ebSLisandro Dalcin   PetscValidType(B,1);
208638f409ebSLisandro Dalcin   PetscValidLogicalCollectiveInt(B,bs,2);
208738f409ebSLisandro Dalcin   ierr = PetscTryMethod(B,"MatSeqSBAIJSetPreallocationCSR_C",(Mat,PetscInt,const PetscInt[],const PetscInt[],const PetscScalar[]),(B,bs,i,j,v));CHKERRQ(ierr);
208838f409ebSLisandro Dalcin   PetscFunctionReturn(0);
208938f409ebSLisandro Dalcin }
209038f409ebSLisandro Dalcin 
209138f409ebSLisandro Dalcin #undef __FUNCT__
20924a2ae208SSatish Balay #define __FUNCT__ "MatCreateSeqSBAIJ"
2093c464158bSHong Zhang /*@C
2094c464158bSHong Zhang    MatCreateSeqSBAIJ - Creates a sparse symmetric matrix in block AIJ (block
2095c464158bSHong Zhang    compressed row) format.  For good matrix assembly performance the
2096c464158bSHong Zhang    user should preallocate the matrix storage by setting the parameter nz
2097c464158bSHong Zhang    (or the array nnz).  By setting these parameters accurately, performance
2098c464158bSHong Zhang    during matrix assembly can be increased by more than a factor of 50.
209949b5e25fSSatish Balay 
2100c464158bSHong Zhang    Collective on MPI_Comm
2101c464158bSHong Zhang 
2102c464158bSHong Zhang    Input Parameters:
2103c464158bSHong Zhang +  comm - MPI communicator, set to PETSC_COMM_SELF
2104bb7ae925SBarry 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
2105bb7ae925SBarry Smith           blocksize always defines the size of the blocks. The column blocksize sets the blocksize of the vectors obtained with MatCreateVecs()
2106c464158bSHong Zhang .  m - number of rows, or number of columns
2107c464158bSHong Zhang .  nz - number of block nonzeros per block row (same for all rows)
2108744e8345SSatish Balay -  nnz - array containing the number of block nonzeros in the upper triangular plus
21090298fd71SBarry Smith          diagonal portion of each block (possibly different for each block row) or NULL
2110c464158bSHong Zhang 
2111c464158bSHong Zhang    Output Parameter:
2112c464158bSHong Zhang .  A - the symmetric matrix
2113c464158bSHong Zhang 
2114c464158bSHong Zhang    Options Database Keys:
2115c464158bSHong Zhang .   -mat_no_unroll - uses code that does not unroll the loops in the
2116c464158bSHong Zhang                      block calculations (much slower)
2117c464158bSHong Zhang .    -mat_block_size - size of the blocks to use
2118c464158bSHong Zhang 
2119c464158bSHong Zhang    Level: intermediate
2120c464158bSHong Zhang 
2121175b88e8SBarry Smith    It is recommended that one use the MatCreate(), MatSetType() and/or MatSetFromOptions(),
2122ae1d86c5SBarry Smith    MatXXXXSetPreallocation() paradgm instead of this routine directly.
2123175b88e8SBarry Smith    [MatXXXXSetPreallocation() is, for example, MatSeqAIJSetPreallocation]
2124175b88e8SBarry Smith 
2125c464158bSHong Zhang    Notes:
21266d6d819aSHong Zhang    The number of rows and columns must be divisible by blocksize.
21276d6d819aSHong Zhang    This matrix type does not support complex Hermitian operation.
2128c464158bSHong Zhang 
2129c464158bSHong Zhang    Specify the preallocated storage with either nz or nnz (not both).
21300298fd71SBarry Smith    Set nz=PETSC_DEFAULT and nnz=NULL for PETSc to control dynamic memory
2131a7f22e61SSatish Balay    allocation.  See Users-Manual: ch_mat for details.
2132c464158bSHong Zhang 
213349a6f317SBarry Smith    If the nnz parameter is given then the nz parameter is ignored
213449a6f317SBarry Smith 
213569b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateSBAIJ()
2136c464158bSHong Zhang @*/
21377087cfbeSBarry Smith PetscErrorCode  MatCreateSeqSBAIJ(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt nz,const PetscInt nnz[],Mat *A)
2138c464158bSHong Zhang {
2139dfbe8321SBarry Smith   PetscErrorCode ierr;
2140c464158bSHong Zhang 
2141c464158bSHong Zhang   PetscFunctionBegin;
2142f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,A);CHKERRQ(ierr);
2143f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*A,m,n,m,n);CHKERRQ(ierr);
2144c464158bSHong Zhang   ierr = MatSetType(*A,MATSEQSBAIJ);CHKERRQ(ierr);
2145367daffbSBarry Smith   ierr = MatSeqSBAIJSetPreallocation(*A,bs,nz,(PetscInt*)nnz);CHKERRQ(ierr);
214649b5e25fSSatish Balay   PetscFunctionReturn(0);
214749b5e25fSSatish Balay }
214849b5e25fSSatish Balay 
21494a2ae208SSatish Balay #undef __FUNCT__
21504a2ae208SSatish Balay #define __FUNCT__ "MatDuplicate_SeqSBAIJ"
2151dfbe8321SBarry Smith PetscErrorCode MatDuplicate_SeqSBAIJ(Mat A,MatDuplicateOption cpvalues,Mat *B)
215249b5e25fSSatish Balay {
215349b5e25fSSatish Balay   Mat            C;
215449b5e25fSSatish Balay   Mat_SeqSBAIJ   *c,*a = (Mat_SeqSBAIJ*)A->data;
21556849ba73SBarry Smith   PetscErrorCode ierr;
2156b40805acSSatish Balay   PetscInt       i,mbs = a->mbs,nz = a->nz,bs2 =a->bs2;
215749b5e25fSSatish Balay 
215849b5e25fSSatish Balay   PetscFunctionBegin;
2159e32f2f54SBarry Smith   if (a->i[mbs] != nz) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Corrupt matrix");
216049b5e25fSSatish Balay 
216149b5e25fSSatish Balay   *B   = 0;
2162ce94432eSBarry Smith   ierr = MatCreate(PetscObjectComm((PetscObject)A),&C);CHKERRQ(ierr);
2163d0f46423SBarry Smith   ierr = MatSetSizes(C,A->rmap->N,A->cmap->n,A->rmap->N,A->cmap->n);CHKERRQ(ierr);
21648e9a0fb8SHong Zhang   ierr = MatSetType(C,MATSEQSBAIJ);CHKERRQ(ierr);
21651d5dac46SHong Zhang   ierr = PetscMemcpy(C->ops,A->ops,sizeof(struct _MatOps));CHKERRQ(ierr);
2166692f9cbeSHong Zhang   c    = (Mat_SeqSBAIJ*)C->data;
2167692f9cbeSHong Zhang 
2168273d9f13SBarry Smith   C->preallocated       = PETSC_TRUE;
2169d5f3da31SBarry Smith   C->factortype         = A->factortype;
217049b5e25fSSatish Balay   c->row                = 0;
217149b5e25fSSatish Balay   c->icol               = 0;
217249b5e25fSSatish Balay   c->saved_values       = 0;
2173a9817697SBarry Smith   c->keepnonzeropattern = a->keepnonzeropattern;
217449b5e25fSSatish Balay   C->assembled          = PETSC_TRUE;
217549b5e25fSSatish Balay 
21761e1e43feSBarry Smith   ierr   = PetscLayoutReference(A->rmap,&C->rmap);CHKERRQ(ierr);
21771e1e43feSBarry Smith   ierr   = PetscLayoutReference(A->cmap,&C->cmap);CHKERRQ(ierr);
217849b5e25fSSatish Balay   c->bs2 = a->bs2;
217949b5e25fSSatish Balay   c->mbs = a->mbs;
218049b5e25fSSatish Balay   c->nbs = a->nbs;
218149b5e25fSSatish Balay 
2182c760cd28SBarry Smith   if  (cpvalues == MAT_SHARE_NONZERO_PATTERN) {
2183c760cd28SBarry Smith     c->imax           = a->imax;
2184c760cd28SBarry Smith     c->ilen           = a->ilen;
2185c760cd28SBarry Smith     c->free_imax_ilen = PETSC_FALSE;
2186c760cd28SBarry Smith   } else {
2187dcca6d9dSJed Brown     ierr = PetscMalloc2((mbs+1),&c->imax,(mbs+1),&c->ilen);CHKERRQ(ierr);
21883bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)C,2*(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr);
218949b5e25fSSatish Balay     for (i=0; i<mbs; i++) {
219049b5e25fSSatish Balay       c->imax[i] = a->imax[i];
219149b5e25fSSatish Balay       c->ilen[i] = a->ilen[i];
219249b5e25fSSatish Balay     }
2193c760cd28SBarry Smith     c->free_imax_ilen = PETSC_TRUE;
2194c760cd28SBarry Smith   }
219549b5e25fSSatish Balay 
219649b5e25fSSatish Balay   /* allocate the matrix space */
21974da8f245SBarry Smith   if (cpvalues == MAT_SHARE_NONZERO_PATTERN) {
2198785e854fSJed Brown     ierr            = PetscMalloc1(bs2*nz,&c->a);CHKERRQ(ierr);
21993bb1ff40SBarry Smith     ierr            = PetscLogObjectMemory((PetscObject)C,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr);
220044e1c64aSLisandro Dalcin     c->i            = a->i;
220144e1c64aSLisandro Dalcin     c->j            = a->j;
22024da8f245SBarry Smith     c->singlemalloc = PETSC_FALSE;
220344e1c64aSLisandro Dalcin     c->free_a       = PETSC_TRUE;
22044da8f245SBarry Smith     c->free_ij      = PETSC_FALSE;
22054da8f245SBarry Smith     c->parent       = A;
22064da8f245SBarry Smith     ierr            = PetscObjectReference((PetscObject)A);CHKERRQ(ierr);
22074da8f245SBarry Smith     ierr            = MatSetOption(A,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
22084da8f245SBarry Smith     ierr            = MatSetOption(C,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
22094da8f245SBarry Smith   } else {
2210dcca6d9dSJed Brown     ierr            = PetscMalloc3(bs2*nz,&c->a,nz,&c->j,mbs+1,&c->i);CHKERRQ(ierr);
221113f74950SBarry Smith     ierr            = PetscMemcpy(c->i,a->i,(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr);
22123bb1ff40SBarry Smith     ierr            = PetscLogObjectMemory((PetscObject)C,(mbs+1)*sizeof(PetscInt) + nz*(bs2*sizeof(MatScalar) + sizeof(PetscInt)));CHKERRQ(ierr);
22134da8f245SBarry Smith     c->singlemalloc = PETSC_TRUE;
221444e1c64aSLisandro Dalcin     c->free_a       = PETSC_TRUE;
22154da8f245SBarry Smith     c->free_ij      = PETSC_TRUE;
22164da8f245SBarry Smith   }
221749b5e25fSSatish Balay   if (mbs > 0) {
22184da8f245SBarry Smith     if (cpvalues != MAT_SHARE_NONZERO_PATTERN) {
221913f74950SBarry Smith       ierr = PetscMemcpy(c->j,a->j,nz*sizeof(PetscInt));CHKERRQ(ierr);
22204da8f245SBarry Smith     }
222149b5e25fSSatish Balay     if (cpvalues == MAT_COPY_VALUES) {
222249b5e25fSSatish Balay       ierr = PetscMemcpy(c->a,a->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr);
222349b5e25fSSatish Balay     } else {
222449b5e25fSSatish Balay       ierr = PetscMemzero(c->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr);
222549b5e25fSSatish Balay     }
2226a1c3900fSBarry Smith     if (a->jshort) {
222744e1c64aSLisandro Dalcin       /* cannot share jshort, it is reallocated in MatAssemblyEnd_SeqSBAIJ() */
222844e1c64aSLisandro Dalcin       /* if the parent matrix is reassembled, this child matrix will never notice */
2229785e854fSJed Brown       ierr = PetscMalloc1(nz,&c->jshort);CHKERRQ(ierr);
22303bb1ff40SBarry Smith       ierr = PetscLogObjectMemory((PetscObject)C,nz*sizeof(unsigned short));CHKERRQ(ierr);
2231a1c3900fSBarry Smith       ierr = PetscMemcpy(c->jshort,a->jshort,nz*sizeof(unsigned short));CHKERRQ(ierr);
223226fbe8dcSKarl Rupp 
22334da8f245SBarry Smith       c->free_jshort = PETSC_TRUE;
22344da8f245SBarry Smith     }
2235a1c3900fSBarry Smith   }
223649b5e25fSSatish Balay 
223749b5e25fSSatish Balay   c->roworiented = a->roworiented;
223849b5e25fSSatish Balay   c->nonew       = a->nonew;
223949b5e25fSSatish Balay 
224049b5e25fSSatish Balay   if (a->diag) {
2241c760cd28SBarry Smith     if (cpvalues == MAT_SHARE_NONZERO_PATTERN) {
2242c760cd28SBarry Smith       c->diag      = a->diag;
2243c760cd28SBarry Smith       c->free_diag = PETSC_FALSE;
2244c760cd28SBarry Smith     } else {
2245785e854fSJed Brown       ierr = PetscMalloc1(mbs,&c->diag);CHKERRQ(ierr);
22463bb1ff40SBarry Smith       ierr = PetscLogObjectMemory((PetscObject)C,mbs*sizeof(PetscInt));CHKERRQ(ierr);
224726fbe8dcSKarl Rupp       for (i=0; i<mbs; i++) c->diag[i] = a->diag[i];
2248c760cd28SBarry Smith       c->free_diag = PETSC_TRUE;
2249c760cd28SBarry Smith     }
225044e1c64aSLisandro Dalcin   }
22516c6c5352SBarry Smith   c->nz         = a->nz;
2252f2cbd3d5SJed Brown   c->maxnz      = a->nz; /* Since we allocate exactly the right amount */
225349b5e25fSSatish Balay   c->solve_work = 0;
225449b5e25fSSatish Balay   c->mult_work  = 0;
225526fbe8dcSKarl Rupp 
225649b5e25fSSatish Balay   *B   = C;
2257140e18c1SBarry Smith   ierr = PetscFunctionListDuplicate(((PetscObject)A)->qlist,&((PetscObject)C)->qlist);CHKERRQ(ierr);
225849b5e25fSSatish Balay   PetscFunctionReturn(0);
225949b5e25fSSatish Balay }
226049b5e25fSSatish Balay 
22614a2ae208SSatish Balay #undef __FUNCT__
22625bba2384SShri Abhyankar #define __FUNCT__ "MatLoad_SeqSBAIJ"
2263112444f4SShri Abhyankar PetscErrorCode MatLoad_SeqSBAIJ(Mat newmat,PetscViewer viewer)
22642f480046SShri Abhyankar {
22652f480046SShri Abhyankar   Mat_SeqSBAIJ   *a;
22662f480046SShri Abhyankar   PetscErrorCode ierr;
22672f480046SShri Abhyankar   int            fd;
22682f480046SShri Abhyankar   PetscMPIInt    size;
22693059b6faSBarry Smith   PetscInt       i,nz,header[4],*rowlengths=0,M,N,bs = newmat->rmap->bs;
22702f480046SShri Abhyankar   PetscInt       *mask,mbs,*jj,j,rowcount,nzcount,k,*s_browlengths,maskcount;
22712f480046SShri Abhyankar   PetscInt       kmax,jcount,block,idx,point,nzcountb,extra_rows,rows,cols;
22722f480046SShri Abhyankar   PetscInt       *masked,nmask,tmp,bs2,ishift;
22732f480046SShri Abhyankar   PetscScalar    *aa;
2274ce94432eSBarry Smith   MPI_Comm       comm;
22752f480046SShri Abhyankar 
22762f480046SShri Abhyankar   PetscFunctionBegin;
2277c98fd787SBarry Smith   /* force binary viewer to load .info file if it has not yet done so */
2278c98fd787SBarry Smith   ierr = PetscViewerSetUp(viewer);CHKERRQ(ierr);
2279ce94432eSBarry Smith   ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr);
2280c5929fdfSBarry Smith   ierr = PetscOptionsGetInt(((PetscObject)newmat)->options,((PetscObject)newmat)->prefix,"-matload_block_size",&bs,NULL);CHKERRQ(ierr);
22813059b6faSBarry Smith   if (bs < 0) bs = 1;
22822f480046SShri Abhyankar   bs2  = bs*bs;
22832f480046SShri Abhyankar 
22842f480046SShri Abhyankar   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
22852f480046SShri Abhyankar   if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"view must have one processor");
22862f480046SShri Abhyankar   ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr);
22872f480046SShri Abhyankar   ierr = PetscBinaryRead(fd,header,4,PETSC_INT);CHKERRQ(ierr);
22882f480046SShri Abhyankar   if (header[0] != MAT_FILE_CLASSID) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"not Mat object");
22892f480046SShri Abhyankar   M = header[1]; N = header[2]; nz = header[3];
22902f480046SShri Abhyankar 
22912f480046SShri Abhyankar   if (header[3] < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Matrix stored in special format, cannot load as SeqSBAIJ");
22922f480046SShri Abhyankar 
22932f480046SShri Abhyankar   if (M != N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Can only do square matrices");
22942f480046SShri Abhyankar 
22952f480046SShri Abhyankar   /*
22962f480046SShri Abhyankar      This code adds extra rows to make sure the number of rows is
22972f480046SShri Abhyankar     divisible by the blocksize
22982f480046SShri Abhyankar   */
22992f480046SShri Abhyankar   mbs        = M/bs;
23002f480046SShri Abhyankar   extra_rows = bs - M + bs*(mbs);
23012f480046SShri Abhyankar   if (extra_rows == bs) extra_rows = 0;
23022f480046SShri Abhyankar   else                  mbs++;
23032f480046SShri Abhyankar   if (extra_rows) {
23042f480046SShri Abhyankar     ierr = PetscInfo(viewer,"Padding loaded matrix to match blocksize\n");CHKERRQ(ierr);
23052f480046SShri Abhyankar   }
23062f480046SShri Abhyankar 
23072f480046SShri Abhyankar   /* Set global sizes if not already set */
23082f480046SShri Abhyankar   if (newmat->rmap->n < 0 && newmat->rmap->N < 0 && newmat->cmap->n < 0 && newmat->cmap->N < 0) {
23092f480046SShri Abhyankar     ierr = MatSetSizes(newmat,PETSC_DECIDE,PETSC_DECIDE,M+extra_rows,N+extra_rows);CHKERRQ(ierr);
23102f480046SShri Abhyankar   } else { /* Check if the matrix global sizes are correct */
23112f480046SShri Abhyankar     ierr = MatGetSize(newmat,&rows,&cols);CHKERRQ(ierr);
23122f480046SShri Abhyankar     if (M != rows ||  N != cols) SETERRQ4(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Matrix in file of different length (%d, %d) than the input matrix (%d, %d)",M,N,rows,cols);
23132f480046SShri Abhyankar   }
23142f480046SShri Abhyankar 
23152f480046SShri Abhyankar   /* read in row lengths */
2316854ce69bSBarry Smith   ierr = PetscMalloc1(M+extra_rows,&rowlengths);CHKERRQ(ierr);
23172f480046SShri Abhyankar   ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr);
23182f480046SShri Abhyankar   for (i=0; i<extra_rows; i++) rowlengths[M+i] = 1;
23192f480046SShri Abhyankar 
23202f480046SShri Abhyankar   /* read in column indices */
2321854ce69bSBarry Smith   ierr = PetscMalloc1(nz+extra_rows,&jj);CHKERRQ(ierr);
23222f480046SShri Abhyankar   ierr = PetscBinaryRead(fd,jj,nz,PETSC_INT);CHKERRQ(ierr);
23232f480046SShri Abhyankar   for (i=0; i<extra_rows; i++) jj[nz+i] = M+i;
23242f480046SShri Abhyankar 
23252f480046SShri Abhyankar   /* loop over row lengths determining block row lengths */
23261795a4d1SJed Brown   ierr     = PetscCalloc1(mbs,&s_browlengths);CHKERRQ(ierr);
2327dcca6d9dSJed Brown   ierr     = PetscMalloc2(mbs,&mask,mbs,&masked);CHKERRQ(ierr);
23282f480046SShri Abhyankar   ierr     = PetscMemzero(mask,mbs*sizeof(PetscInt));CHKERRQ(ierr);
23292f480046SShri Abhyankar   rowcount = 0;
23302f480046SShri Abhyankar   nzcount  = 0;
23312f480046SShri Abhyankar   for (i=0; i<mbs; i++) {
23322f480046SShri Abhyankar     nmask = 0;
23332f480046SShri Abhyankar     for (j=0; j<bs; j++) {
23342f480046SShri Abhyankar       kmax = rowlengths[rowcount];
23352f480046SShri Abhyankar       for (k=0; k<kmax; k++) {
23362f480046SShri Abhyankar         tmp = jj[nzcount++]/bs;   /* block col. index */
23372f480046SShri Abhyankar         if (!mask[tmp] && tmp >= i) {masked[nmask++] = tmp; mask[tmp] = 1;}
23382f480046SShri Abhyankar       }
23392f480046SShri Abhyankar       rowcount++;
23402f480046SShri Abhyankar     }
23412f480046SShri Abhyankar     s_browlengths[i] += nmask;
23422f480046SShri Abhyankar 
23432f480046SShri Abhyankar     /* zero out the mask elements we set */
23442f480046SShri Abhyankar     for (j=0; j<nmask; j++) mask[masked[j]] = 0;
23452f480046SShri Abhyankar   }
23462f480046SShri Abhyankar 
23472f480046SShri Abhyankar   /* Do preallocation */
2348367daffbSBarry Smith   ierr = MatSeqSBAIJSetPreallocation(newmat,bs,0,s_browlengths);CHKERRQ(ierr);
23492f480046SShri Abhyankar   a    = (Mat_SeqSBAIJ*)newmat->data;
23502f480046SShri Abhyankar 
23512f480046SShri Abhyankar   /* set matrix "i" values */
23522f480046SShri Abhyankar   a->i[0] = 0;
23532f480046SShri Abhyankar   for (i=1; i<= mbs; i++) {
23542f480046SShri Abhyankar     a->i[i]      = a->i[i-1] + s_browlengths[i-1];
23552f480046SShri Abhyankar     a->ilen[i-1] = s_browlengths[i-1];
23562f480046SShri Abhyankar   }
23572f480046SShri Abhyankar   a->nz = a->i[mbs];
23582f480046SShri Abhyankar 
23592f480046SShri Abhyankar   /* read in nonzero values */
2360854ce69bSBarry Smith   ierr = PetscMalloc1(nz+extra_rows,&aa);CHKERRQ(ierr);
23612f480046SShri Abhyankar   ierr = PetscBinaryRead(fd,aa,nz,PETSC_SCALAR);CHKERRQ(ierr);
23622f480046SShri Abhyankar   for (i=0; i<extra_rows; i++) aa[nz+i] = 1.0;
23632f480046SShri Abhyankar 
23642f480046SShri Abhyankar   /* set "a" and "j" values into matrix */
23652f480046SShri Abhyankar   nzcount = 0; jcount = 0;
23662f480046SShri Abhyankar   for (i=0; i<mbs; i++) {
23672f480046SShri Abhyankar     nzcountb = nzcount;
23682f480046SShri Abhyankar     nmask    = 0;
23692f480046SShri Abhyankar     for (j=0; j<bs; j++) {
23702f480046SShri Abhyankar       kmax = rowlengths[i*bs+j];
23712f480046SShri Abhyankar       for (k=0; k<kmax; k++) {
23722f480046SShri Abhyankar         tmp = jj[nzcount++]/bs; /* block col. index */
23732f480046SShri Abhyankar         if (!mask[tmp] && tmp >= i) { masked[nmask++] = tmp; mask[tmp] = 1;}
23742f480046SShri Abhyankar       }
23752f480046SShri Abhyankar     }
23762f480046SShri Abhyankar     /* sort the masked values */
23772f480046SShri Abhyankar     ierr = PetscSortInt(nmask,masked);CHKERRQ(ierr);
23782f480046SShri Abhyankar 
23792f480046SShri Abhyankar     /* set "j" values into matrix */
23802f480046SShri Abhyankar     maskcount = 1;
23812f480046SShri Abhyankar     for (j=0; j<nmask; j++) {
23822f480046SShri Abhyankar       a->j[jcount++]  = masked[j];
23832f480046SShri Abhyankar       mask[masked[j]] = maskcount++;
23842f480046SShri Abhyankar     }
23852f480046SShri Abhyankar 
23862f480046SShri Abhyankar     /* set "a" values into matrix */
23872f480046SShri Abhyankar     ishift = bs2*a->i[i];
23882f480046SShri Abhyankar     for (j=0; j<bs; j++) {
23892f480046SShri Abhyankar       kmax = rowlengths[i*bs+j];
23902f480046SShri Abhyankar       for (k=0; k<kmax; k++) {
23912f480046SShri Abhyankar         tmp = jj[nzcountb]/bs;        /* block col. index */
23922f480046SShri Abhyankar         if (tmp >= i) {
23932f480046SShri Abhyankar           block     = mask[tmp] - 1;
23942f480046SShri Abhyankar           point     = jj[nzcountb] - bs*tmp;
23952f480046SShri Abhyankar           idx       = ishift + bs2*block + j + bs*point;
23962f480046SShri Abhyankar           a->a[idx] = aa[nzcountb];
23972f480046SShri Abhyankar         }
23982f480046SShri Abhyankar         nzcountb++;
23992f480046SShri Abhyankar       }
24002f480046SShri Abhyankar     }
24012f480046SShri Abhyankar     /* zero out the mask elements we set */
24022f480046SShri Abhyankar     for (j=0; j<nmask; j++) mask[masked[j]] = 0;
24032f480046SShri Abhyankar   }
24042f480046SShri Abhyankar   if (jcount != a->nz) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Bad binary matrix");
24052f480046SShri Abhyankar 
24062f480046SShri Abhyankar   ierr = PetscFree(rowlengths);CHKERRQ(ierr);
24072f480046SShri Abhyankar   ierr = PetscFree(s_browlengths);CHKERRQ(ierr);
24082f480046SShri Abhyankar   ierr = PetscFree(aa);CHKERRQ(ierr);
24092f480046SShri Abhyankar   ierr = PetscFree(jj);CHKERRQ(ierr);
24102f480046SShri Abhyankar   ierr = PetscFree2(mask,masked);CHKERRQ(ierr);
24112f480046SShri Abhyankar 
24122f480046SShri Abhyankar   ierr = MatAssemblyBegin(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
24132f480046SShri Abhyankar   ierr = MatAssemblyEnd(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
24142f480046SShri Abhyankar   PetscFunctionReturn(0);
24152f480046SShri Abhyankar }
24162f480046SShri Abhyankar 
24172f480046SShri Abhyankar #undef __FUNCT__
2418c75a6043SHong Zhang #define __FUNCT__ "MatCreateSeqSBAIJWithArrays"
2419c75a6043SHong Zhang /*@
2420c75a6043SHong Zhang      MatCreateSeqSBAIJWithArrays - Creates an sequential SBAIJ matrix using matrix elements
2421c75a6043SHong Zhang               (upper triangular entries in CSR format) provided by the user.
2422c75a6043SHong Zhang 
2423c75a6043SHong Zhang      Collective on MPI_Comm
2424c75a6043SHong Zhang 
2425c75a6043SHong Zhang    Input Parameters:
2426c75a6043SHong Zhang +  comm - must be an MPI communicator of size 1
2427c75a6043SHong Zhang .  bs - size of block
2428c75a6043SHong Zhang .  m - number of rows
2429c75a6043SHong Zhang .  n - number of columns
2430c75a6043SHong Zhang .  i - row indices
2431c75a6043SHong Zhang .  j - column indices
2432c75a6043SHong Zhang -  a - matrix values
2433c75a6043SHong Zhang 
2434c75a6043SHong Zhang    Output Parameter:
2435c75a6043SHong Zhang .  mat - the matrix
2436c75a6043SHong Zhang 
2437dfb205c3SBarry Smith    Level: advanced
2438c75a6043SHong Zhang 
2439c75a6043SHong Zhang    Notes:
2440c75a6043SHong Zhang        The i, j, and a arrays are not copied by this routine, the user must free these arrays
2441c75a6043SHong Zhang     once the matrix is destroyed
2442c75a6043SHong Zhang 
2443c75a6043SHong Zhang        You cannot set new nonzero locations into this matrix, that will generate an error.
2444c75a6043SHong Zhang 
2445c75a6043SHong Zhang        The i and j indices are 0 based
2446c75a6043SHong Zhang 
2447dfb205c3SBarry 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
2448dfb205c3SBarry Smith        it is the regular CSR format excluding the lower triangular elements.
2449dfb205c3SBarry Smith 
245069b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSBAIJ(), MatCreateSeqSBAIJ()
2451c75a6043SHong Zhang 
2452c75a6043SHong Zhang @*/
24537087cfbeSBarry Smith PetscErrorCode  MatCreateSeqSBAIJWithArrays(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt *i,PetscInt *j,PetscScalar *a,Mat *mat)
2454c75a6043SHong Zhang {
2455c75a6043SHong Zhang   PetscErrorCode ierr;
2456c75a6043SHong Zhang   PetscInt       ii;
2457c75a6043SHong Zhang   Mat_SeqSBAIJ   *sbaij;
2458c75a6043SHong Zhang 
2459c75a6043SHong Zhang   PetscFunctionBegin;
2460e32f2f54SBarry Smith   if (bs != 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"block size %D > 1 is not supported yet",bs);
246141096f02SStefano Zampini   if (m > 0 && i[0]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"i (row indices) must start with 0");
2462c75a6043SHong Zhang 
2463c75a6043SHong Zhang   ierr  = MatCreate(comm,mat);CHKERRQ(ierr);
2464c75a6043SHong Zhang   ierr  = MatSetSizes(*mat,m,n,m,n);CHKERRQ(ierr);
2465c75a6043SHong Zhang   ierr  = MatSetType(*mat,MATSEQSBAIJ);CHKERRQ(ierr);
2466367daffbSBarry Smith   ierr  = MatSeqSBAIJSetPreallocation(*mat,bs,MAT_SKIP_ALLOCATION,0);CHKERRQ(ierr);
2467c75a6043SHong Zhang   sbaij = (Mat_SeqSBAIJ*)(*mat)->data;
2468dcca6d9dSJed Brown   ierr  = PetscMalloc2(m,&sbaij->imax,m,&sbaij->ilen);CHKERRQ(ierr);
24693bb1ff40SBarry Smith   ierr  = PetscLogObjectMemory((PetscObject)*mat,2*m*sizeof(PetscInt));CHKERRQ(ierr);
2470c75a6043SHong Zhang 
2471c75a6043SHong Zhang   sbaij->i = i;
2472c75a6043SHong Zhang   sbaij->j = j;
2473c75a6043SHong Zhang   sbaij->a = a;
247426fbe8dcSKarl Rupp 
2475c75a6043SHong Zhang   sbaij->singlemalloc = PETSC_FALSE;
2476c75a6043SHong Zhang   sbaij->nonew        = -1;             /*this indicates that inserting a new value in the matrix that generates a new nonzero is an error*/
2477e6b907acSBarry Smith   sbaij->free_a       = PETSC_FALSE;
2478e6b907acSBarry Smith   sbaij->free_ij      = PETSC_FALSE;
2479c75a6043SHong Zhang 
2480c75a6043SHong Zhang   for (ii=0; ii<m; ii++) {
2481c75a6043SHong Zhang     sbaij->ilen[ii] = sbaij->imax[ii] = i[ii+1] - i[ii];
2482c75a6043SHong Zhang #if defined(PETSC_USE_DEBUG)
2483e32f2f54SBarry Smith     if (i[ii+1] - i[ii] < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative row length in i (row indices) row = %d length = %d",ii,i[ii+1] - i[ii]);
2484c75a6043SHong Zhang #endif
2485c75a6043SHong Zhang   }
2486c75a6043SHong Zhang #if defined(PETSC_USE_DEBUG)
2487c75a6043SHong Zhang   for (ii=0; ii<sbaij->i[m]; ii++) {
2488e32f2f54SBarry Smith     if (j[ii] < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column index at location = %d index = %d",ii,j[ii]);
2489e32f2f54SBarry Smith     if (j[ii] > n - 1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column index to large at location = %d index = %d",ii,j[ii]);
2490c75a6043SHong Zhang   }
2491c75a6043SHong Zhang #endif
2492c75a6043SHong Zhang 
2493c75a6043SHong Zhang   ierr = MatAssemblyBegin(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2494c75a6043SHong Zhang   ierr = MatAssemblyEnd(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2495c75a6043SHong Zhang   PetscFunctionReturn(0);
2496c75a6043SHong Zhang }
2497d06b337dSHong Zhang 
249859f5e6ceSHong Zhang #undef __FUNCT__
249959f5e6ceSHong Zhang #define __FUNCT__ "MatCreateMPIMatConcatenateSeqMat_SeqSBAIJ"
250059f5e6ceSHong Zhang PetscErrorCode MatCreateMPIMatConcatenateSeqMat_SeqSBAIJ(MPI_Comm comm,Mat inmat,PetscInt n,MatReuse scall,Mat *outmat)
250159f5e6ceSHong Zhang {
250259f5e6ceSHong Zhang   PetscErrorCode ierr;
250359f5e6ceSHong Zhang 
250459f5e6ceSHong Zhang   PetscFunctionBegin;
250559f5e6ceSHong Zhang   ierr = MatCreateMPIMatConcatenateSeqMat_MPISBAIJ(comm,inmat,n,scall,outmat);CHKERRQ(ierr);
250659f5e6ceSHong Zhang   PetscFunctionReturn(0);
250759f5e6ceSHong Zhang }
2508d06b337dSHong Zhang 
2509d06b337dSHong Zhang 
251049b5e25fSSatish Balay 
251149b5e25fSSatish Balay 
2512