xref: /petsc/src/mat/impls/sbaij/seq/sbaij.c (revision 48dd3d27d4f900b58bc41f8c260a0584bcdb3fe2)
149b5e25fSSatish Balay 
249b5e25fSSatish Balay /*
3a1373b80SHong Zhang     Defines the basic matrix operations for the SBAIJ (compressed row)
449b5e25fSSatish Balay   matrix storage format.
549b5e25fSSatish Balay */
6c6db04a5SJed Brown #include <../src/mat/impls/baij/seq/baij.h>         /*I "petscmat.h" I*/
7c6db04a5SJed Brown #include <../src/mat/impls/sbaij/seq/sbaij.h>
8c6db04a5SJed Brown #include <petscblaslapack.h>
949b5e25fSSatish Balay 
10c6db04a5SJed Brown #include <../src/mat/impls/sbaij/seq/relax.h>
1170dcbbb9SBarry Smith #define USESHORT
12c6db04a5SJed Brown #include <../src/mat/impls/sbaij/seq/relax.h>
1370dcbbb9SBarry Smith 
14ace3abfcSBarry Smith extern PetscErrorCode MatSeqSBAIJSetNumericFactorization_inplace(Mat,PetscBool);
15b5b17502SBarry Smith 
1649b5e25fSSatish Balay /*
1749b5e25fSSatish Balay      Checks for missing diagonals
1849b5e25fSSatish Balay */
194a2ae208SSatish Balay #undef __FUNCT__
204a2ae208SSatish Balay #define __FUNCT__ "MatMissingDiagonal_SeqSBAIJ"
21ace3abfcSBarry Smith PetscErrorCode MatMissingDiagonal_SeqSBAIJ(Mat A,PetscBool  *missing,PetscInt *dd)
2249b5e25fSSatish Balay {
23045c9aa0SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
246849ba73SBarry Smith   PetscErrorCode ierr;
257734d3b5SMatthew G. Knepley   PetscInt       *diag,*ii = a->i,i;
2649b5e25fSSatish Balay 
2749b5e25fSSatish Balay   PetscFunctionBegin;
28045c9aa0SHong Zhang   ierr     = MatMarkDiagonal_SeqSBAIJ(A);CHKERRQ(ierr);
292af78befSBarry Smith   *missing = PETSC_FALSE;
307734d3b5SMatthew G. Knepley   if (A->rmap->n > 0 && !ii) {
31358d2f5dSShri Abhyankar     *missing = PETSC_TRUE;
32358d2f5dSShri Abhyankar     if (dd) *dd = 0;
33358d2f5dSShri Abhyankar     PetscInfo(A,"Matrix has no entries therefore is missing diagonal");
34358d2f5dSShri Abhyankar   } else {
35358d2f5dSShri Abhyankar     diag = a->diag;
3649b5e25fSSatish Balay     for (i=0; i<a->mbs; i++) {
377734d3b5SMatthew G. Knepley       if (diag[i] >= ii[i+1]) {
382af78befSBarry Smith         *missing = PETSC_TRUE;
392af78befSBarry Smith         if (dd) *dd = i;
402af78befSBarry Smith         break;
412af78befSBarry Smith       }
4249b5e25fSSatish Balay     }
43358d2f5dSShri Abhyankar   }
4449b5e25fSSatish Balay   PetscFunctionReturn(0);
4549b5e25fSSatish Balay }
4649b5e25fSSatish Balay 
474a2ae208SSatish Balay #undef __FUNCT__
484a2ae208SSatish Balay #define __FUNCT__ "MatMarkDiagonal_SeqSBAIJ"
49dfbe8321SBarry Smith PetscErrorCode MatMarkDiagonal_SeqSBAIJ(Mat A)
5049b5e25fSSatish Balay {
51045c9aa0SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
526849ba73SBarry Smith   PetscErrorCode ierr;
53*48dd3d27SHong Zhang   PetscInt       i,j;
5449b5e25fSSatish Balay 
5549b5e25fSSatish Balay   PetscFunctionBegin;
5609f38230SBarry Smith   if (!a->diag) {
57785e854fSJed Brown     ierr         = PetscMalloc1(a->mbs,&a->diag);CHKERRQ(ierr);
583bb1ff40SBarry Smith     ierr         = PetscLogObjectMemory((PetscObject)A,a->mbs*sizeof(PetscInt));CHKERRQ(ierr);
59c760cd28SBarry Smith     a->free_diag = PETSC_TRUE;
6009f38230SBarry Smith   }
61*48dd3d27SHong Zhang   for (i=0; i<a->mbs; i++) {
62*48dd3d27SHong Zhang     a->diag[i] = a->i[i+1];
63*48dd3d27SHong Zhang     for (j=a->i[i]; j<a->i[i+1]; j++) {
64*48dd3d27SHong Zhang       if (a->j[j] == i) {
65*48dd3d27SHong Zhang         a->diag[i] = j;
66*48dd3d27SHong Zhang         break;
67*48dd3d27SHong Zhang       }
68*48dd3d27SHong Zhang     }
69*48dd3d27SHong Zhang   }
7049b5e25fSSatish Balay   PetscFunctionReturn(0);
7149b5e25fSSatish Balay }
7249b5e25fSSatish Balay 
734a2ae208SSatish Balay #undef __FUNCT__
744a2ae208SSatish Balay #define __FUNCT__ "MatGetRowIJ_SeqSBAIJ"
751a83f524SJed Brown static PetscErrorCode MatGetRowIJ_SeqSBAIJ(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *nn,const PetscInt *inia[],const PetscInt *inja[],PetscBool  *done)
7649b5e25fSSatish Balay {
77a6ece127SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
78d0f46423SBarry Smith   PetscInt       i,j,n = a->mbs,nz = a->i[n],bs = A->rmap->bs;
791a83f524SJed Brown   PetscInt       **ia = (PetscInt**)inia,**ja = (PetscInt**)inja;
808f7157efSSatish Balay   PetscErrorCode ierr;
8149b5e25fSSatish Balay 
8249b5e25fSSatish Balay   PetscFunctionBegin;
83d3e5a4abSHong Zhang   *nn = n;
84a1373b80SHong Zhang   if (!ia) PetscFunctionReturn(0);
858f7157efSSatish Balay   if (!blockcompressed) {
868f7157efSSatish Balay     /* malloc & create the natural set of indices */
87dcca6d9dSJed Brown     ierr = PetscMalloc2((n+1)*bs,ia,nz*bs,ja);CHKERRQ(ierr);
888f7157efSSatish Balay     for (i=0; i<n+1; i++) {
898f7157efSSatish Balay       for (j=0; j<bs; j++) {
908f7157efSSatish Balay         *ia[i*bs+j] = a->i[i]*bs+j+oshift;
918f7157efSSatish Balay       }
928f7157efSSatish Balay     }
938f7157efSSatish Balay     for (i=0; i<nz; i++) {
948f7157efSSatish Balay       for (j=0; j<bs; j++) {
958f7157efSSatish Balay         *ja[i*bs+j] = a->j[i]*bs+j+oshift;
968f7157efSSatish Balay       }
978f7157efSSatish Balay     }
988f7157efSSatish Balay   } else { /* blockcompressed */
99a6ece127SHong Zhang     if (oshift == 1) {
100a6ece127SHong Zhang       /* temporarily add 1 to i and j indices */
1016c6c5352SBarry Smith       for (i=0; i<nz; i++) a->j[i]++;
102a1373b80SHong Zhang       for (i=0; i<n+1; i++) a->i[i]++;
1038f7157efSSatish Balay     }
104a1373b80SHong Zhang     *ia = a->i; *ja = a->j;
105a6ece127SHong Zhang   }
10649b5e25fSSatish Balay   PetscFunctionReturn(0);
10749b5e25fSSatish Balay }
10849b5e25fSSatish Balay 
1094a2ae208SSatish Balay #undef __FUNCT__
1104a2ae208SSatish Balay #define __FUNCT__ "MatRestoreRowIJ_SeqSBAIJ"
1111a83f524SJed Brown static PetscErrorCode MatRestoreRowIJ_SeqSBAIJ(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *nn,const PetscInt *ia[],const PetscInt *ja[],PetscBool  *done)
11249b5e25fSSatish Balay {
113b7aaefc3SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
1148f7157efSSatish Balay   PetscInt       i,n = a->mbs,nz = a->i[n];
1158f7157efSSatish Balay   PetscErrorCode ierr;
116a6ece127SHong Zhang 
11749b5e25fSSatish Balay   PetscFunctionBegin;
11849b5e25fSSatish Balay   if (!ia) PetscFunctionReturn(0);
119a6ece127SHong Zhang 
1208f7157efSSatish Balay   if (!blockcompressed) {
1218f7157efSSatish Balay     ierr = PetscFree2(*ia,*ja);CHKERRQ(ierr);
1228f7157efSSatish Balay   } else if (oshift == 1) { /* blockcompressed */
1236c6c5352SBarry Smith     for (i=0; i<nz; i++) a->j[i]--;
124a6ece127SHong Zhang     for (i=0; i<n+1; i++) a->i[i]--;
125a6ece127SHong Zhang   }
126a6ece127SHong Zhang   PetscFunctionReturn(0);
12749b5e25fSSatish Balay }
12849b5e25fSSatish Balay 
1294a2ae208SSatish Balay #undef __FUNCT__
1304a2ae208SSatish Balay #define __FUNCT__ "MatDestroy_SeqSBAIJ"
131dfbe8321SBarry Smith PetscErrorCode MatDestroy_SeqSBAIJ(Mat A)
13249b5e25fSSatish Balay {
13349b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
134dfbe8321SBarry Smith   PetscErrorCode ierr;
13549b5e25fSSatish Balay 
13649b5e25fSSatish Balay   PetscFunctionBegin;
137a9f03627SSatish Balay #if defined(PETSC_USE_LOG)
138d0f46423SBarry Smith   PetscLogObjectState((PetscObject)A,"Rows=%D, NZ=%D",A->rmap->N,a->nz);
139a9f03627SSatish Balay #endif
140e6b907acSBarry Smith   ierr = MatSeqXAIJFreeAIJ(A,&a->a,&a->j,&a->i);CHKERRQ(ierr);
1417f53bb6cSHong Zhang   if (a->free_diag) {ierr = PetscFree(a->diag);CHKERRQ(ierr);}
1426bf464f9SBarry Smith   ierr = ISDestroy(&a->row);CHKERRQ(ierr);
1436bf464f9SBarry Smith   ierr = ISDestroy(&a->col);CHKERRQ(ierr);
1446bf464f9SBarry Smith   ierr = ISDestroy(&a->icol);CHKERRQ(ierr);
145c31cb41cSBarry Smith   ierr = PetscFree(a->idiag);CHKERRQ(ierr);
146c31cb41cSBarry Smith   ierr = PetscFree(a->inode.size);CHKERRQ(ierr);
147c760cd28SBarry Smith   if (a->free_imax_ilen) {ierr = PetscFree2(a->imax,a->ilen);CHKERRQ(ierr);}
14805b42c5fSBarry Smith   ierr = PetscFree(a->solve_work);CHKERRQ(ierr);
14941f059aeSBarry Smith   ierr = PetscFree(a->sor_work);CHKERRQ(ierr);
15005b42c5fSBarry Smith   ierr = PetscFree(a->solves_work);CHKERRQ(ierr);
15105b42c5fSBarry Smith   ierr = PetscFree(a->mult_work);CHKERRQ(ierr);
15205b42c5fSBarry Smith   ierr = PetscFree(a->saved_values);CHKERRQ(ierr);
15305b42c5fSBarry Smith   ierr = PetscFree(a->xtoy);CHKERRQ(ierr);
1544da8f245SBarry Smith   if (a->free_jshort) {ierr = PetscFree(a->jshort);CHKERRQ(ierr);}
1551a3463dfSHong Zhang   ierr = PetscFree(a->inew);CHKERRQ(ierr);
1566bf464f9SBarry Smith   ierr = MatDestroy(&a->parent);CHKERRQ(ierr);
157bf0cc555SLisandro Dalcin   ierr = PetscFree(A->data);CHKERRQ(ierr);
158901853e0SKris Buschelman 
159dbd8c25aSHong Zhang   ierr = PetscObjectChangeTypeName((PetscObject)A,0);CHKERRQ(ierr);
160bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatStoreValues_C",NULL);CHKERRQ(ierr);
161bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatRetrieveValues_C",NULL);CHKERRQ(ierr);
162bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetColumnIndices_C",NULL);CHKERRQ(ierr);
163bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqaij_C",NULL);CHKERRQ(ierr);
164bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqbaij_C",NULL);CHKERRQ(ierr);
165bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetPreallocation_C",NULL);CHKERRQ(ierr);
16638f409ebSLisandro Dalcin   ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetPreallocationCSR_C",NULL);CHKERRQ(ierr);
167bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqsbstrm_C",NULL);CHKERRQ(ierr);
16849b5e25fSSatish Balay   PetscFunctionReturn(0);
16949b5e25fSSatish Balay }
17049b5e25fSSatish Balay 
1714a2ae208SSatish Balay #undef __FUNCT__
1724a2ae208SSatish Balay #define __FUNCT__ "MatSetOption_SeqSBAIJ"
173ace3abfcSBarry Smith PetscErrorCode MatSetOption_SeqSBAIJ(Mat A,MatOption op,PetscBool flg)
17449b5e25fSSatish Balay {
175045c9aa0SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
17663ba0a88SBarry Smith   PetscErrorCode ierr;
17749b5e25fSSatish Balay 
17849b5e25fSSatish Balay   PetscFunctionBegin;
1794d9d31abSKris Buschelman   switch (op) {
1804d9d31abSKris Buschelman   case MAT_ROW_ORIENTED:
1814e0d8c25SBarry Smith     a->roworiented = flg;
1824d9d31abSKris Buschelman     break;
183a9817697SBarry Smith   case MAT_KEEP_NONZERO_PATTERN:
184a9817697SBarry Smith     a->keepnonzeropattern = flg;
1854d9d31abSKris Buschelman     break;
186512a5fc5SBarry Smith   case MAT_NEW_NONZERO_LOCATIONS:
187512a5fc5SBarry Smith     a->nonew = (flg ? 0 : 1);
1884d9d31abSKris Buschelman     break;
1894d9d31abSKris Buschelman   case MAT_NEW_NONZERO_LOCATION_ERR:
1904e0d8c25SBarry Smith     a->nonew = (flg ? -1 : 0);
1914d9d31abSKris Buschelman     break;
1924d9d31abSKris Buschelman   case MAT_NEW_NONZERO_ALLOCATION_ERR:
1934e0d8c25SBarry Smith     a->nonew = (flg ? -2 : 0);
1944d9d31abSKris Buschelman     break;
19528b2fa4aSMatthew Knepley   case MAT_UNUSED_NONZERO_LOCATION_ERR:
19628b2fa4aSMatthew Knepley     a->nounused = (flg ? -1 : 0);
19728b2fa4aSMatthew Knepley     break;
1984e0d8c25SBarry Smith   case MAT_NEW_DIAGONALS:
1994d9d31abSKris Buschelman   case MAT_IGNORE_OFF_PROC_ENTRIES:
2004d9d31abSKris Buschelman   case MAT_USE_HASH_TABLE:
201290bbb0aSBarry Smith     ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr);
2024d9d31abSKris Buschelman     break;
2039a4540c5SBarry Smith   case MAT_HERMITIAN:
204e32f2f54SBarry Smith     if (!A->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Must call MatAssemblyEnd() first");
2050bc54ff2SBarry Smith     if (A->cmap->n < 65536 && A->cmap->bs == 1) {
206eeffb40dSHong Zhang       A->ops->mult = MatMult_SeqSBAIJ_1_Hermitian_ushort;
2070bc54ff2SBarry Smith     } else if (A->cmap->bs == 1) {
208eeffb40dSHong Zhang       A->ops->mult = MatMult_SeqSBAIJ_1_Hermitian;
209e32f2f54SBarry Smith     } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No support for Hermitian with block size greater than 1");
210eeffb40dSHong Zhang     break;
2113d472b54SHong Zhang   case MAT_SPD:
2125021d80fSJed Brown     /* These options are handled directly by MatSetOption() */
2133d472b54SHong Zhang     break;
21477e54ba9SKris Buschelman   case MAT_SYMMETRIC:
21577e54ba9SKris Buschelman   case MAT_STRUCTURALLY_SYMMETRIC:
2169a4540c5SBarry Smith   case MAT_SYMMETRY_ETERNAL:
217e32f2f54SBarry Smith     if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix must be symmetric");
218290bbb0aSBarry Smith     ierr = PetscInfo1(A,"Option %s not relevent\n",MatOptions[op]);CHKERRQ(ierr);
219290bbb0aSBarry Smith     break;
220941593c8SHong Zhang   case MAT_IGNORE_LOWER_TRIANGULAR:
2214e0d8c25SBarry Smith     a->ignore_ltriangular = flg;
222941593c8SHong Zhang     break;
223941593c8SHong Zhang   case MAT_ERROR_LOWER_TRIANGULAR:
2244e0d8c25SBarry Smith     a->ignore_ltriangular = flg;
22577e54ba9SKris Buschelman     break;
226f5edf698SHong Zhang   case MAT_GETROW_UPPERTRIANGULAR:
2274e0d8c25SBarry Smith     a->getrow_utriangular = flg;
228f5edf698SHong Zhang     break;
2294d9d31abSKris Buschelman   default:
230e32f2f54SBarry Smith     SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"unknown option %d",op);
23149b5e25fSSatish Balay   }
23249b5e25fSSatish Balay   PetscFunctionReturn(0);
23349b5e25fSSatish Balay }
23449b5e25fSSatish Balay 
2354a2ae208SSatish Balay #undef __FUNCT__
2364a2ae208SSatish Balay #define __FUNCT__ "MatGetRow_SeqSBAIJ"
23713f74950SBarry Smith PetscErrorCode MatGetRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *ncols,PetscInt **cols,PetscScalar **v)
23849b5e25fSSatish Balay {
23949b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
2406849ba73SBarry Smith   PetscErrorCode ierr;
24113f74950SBarry Smith   PetscInt       itmp,i,j,k,M,*ai,*aj,bs,bn,bp,*cols_i,bs2;
24249b5e25fSSatish Balay   MatScalar      *aa,*aa_i;
24387828ca2SBarry Smith   PetscScalar    *v_i;
24449b5e25fSSatish Balay 
24549b5e25fSSatish Balay   PetscFunctionBegin;
246e32f2f54SBarry 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()");
247f5edf698SHong Zhang   /* Get the upper triangular part of the row */
248d0f46423SBarry Smith   bs  = A->rmap->bs;
24949b5e25fSSatish Balay   ai  = a->i;
25049b5e25fSSatish Balay   aj  = a->j;
25149b5e25fSSatish Balay   aa  = a->a;
25249b5e25fSSatish Balay   bs2 = a->bs2;
25349b5e25fSSatish Balay 
254e32f2f54SBarry Smith   if (row < 0 || row >= A->rmap->N) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE, "Row %D out of range", row);
25549b5e25fSSatish Balay 
25649b5e25fSSatish Balay   bn     = row/bs; /* Block number */
25749b5e25fSSatish Balay   bp     = row % bs; /* Block position */
25849b5e25fSSatish Balay   M      = ai[bn+1] - ai[bn];
25949b5e25fSSatish Balay   *ncols = bs*M;
26049b5e25fSSatish Balay 
26149b5e25fSSatish Balay   if (v) {
26249b5e25fSSatish Balay     *v = 0;
26349b5e25fSSatish Balay     if (*ncols) {
264785e854fSJed Brown       ierr = PetscMalloc1((*ncols+row),v);CHKERRQ(ierr);
26549b5e25fSSatish Balay       for (i=0; i<M; i++) { /* for each block in the block row */
26649b5e25fSSatish Balay         v_i  = *v + i*bs;
26749b5e25fSSatish Balay         aa_i = aa + bs2*(ai[bn] + i);
26826fbe8dcSKarl Rupp         for (j=bp,k=0; j<bs2; j+=bs,k++) v_i[k] = aa_i[j];
26949b5e25fSSatish Balay       }
27049b5e25fSSatish Balay     }
27149b5e25fSSatish Balay   }
27249b5e25fSSatish Balay 
27349b5e25fSSatish Balay   if (cols) {
27449b5e25fSSatish Balay     *cols = 0;
27549b5e25fSSatish Balay     if (*ncols) {
276785e854fSJed Brown       ierr = PetscMalloc1((*ncols+row),cols);CHKERRQ(ierr);
27749b5e25fSSatish Balay       for (i=0; i<M; i++) { /* for each block in the block row */
27849b5e25fSSatish Balay         cols_i = *cols + i*bs;
27949b5e25fSSatish Balay         itmp   = bs*aj[ai[bn] + i];
28026fbe8dcSKarl Rupp         for (j=0; j<bs; j++) cols_i[j] = itmp++;
28149b5e25fSSatish Balay       }
28249b5e25fSSatish Balay     }
28349b5e25fSSatish Balay   }
28449b5e25fSSatish Balay 
28549b5e25fSSatish Balay   /*search column A(0:row-1,row) (=A(row,0:row-1)). Could be expensive! */
2865ddb2528SHong Zhang   /* this segment is currently removed, so only entries in the upper triangle are obtained */
287519f805aSKarl Rupp #if defined(column_search)
28849b5e25fSSatish Balay   v_i    = *v    + M*bs;
28949b5e25fSSatish Balay   cols_i = *cols + M*bs;
29049b5e25fSSatish Balay   for (i=0; i<bn; i++) { /* for each block row */
29149b5e25fSSatish Balay     M = ai[i+1] - ai[i];
29249b5e25fSSatish Balay     for (j=0; j<M; j++) {
29349b5e25fSSatish Balay       itmp = aj[ai[i] + j];    /* block column value */
29449b5e25fSSatish Balay       if (itmp == bn) {
29549b5e25fSSatish Balay         aa_i = aa + bs2*(ai[i] + j) + bs*bp;
29649b5e25fSSatish Balay         for (k=0; k<bs; k++) {
29749b5e25fSSatish Balay           *cols_i++ = i*bs+k;
29849b5e25fSSatish Balay           *v_i++    = aa_i[k];
29949b5e25fSSatish Balay         }
30049b5e25fSSatish Balay         *ncols += bs;
30149b5e25fSSatish Balay         break;
30249b5e25fSSatish Balay       }
30349b5e25fSSatish Balay     }
30449b5e25fSSatish Balay   }
3055ddb2528SHong Zhang #endif
30649b5e25fSSatish Balay   PetscFunctionReturn(0);
30749b5e25fSSatish Balay }
30849b5e25fSSatish Balay 
3094a2ae208SSatish Balay #undef __FUNCT__
3104a2ae208SSatish Balay #define __FUNCT__ "MatRestoreRow_SeqSBAIJ"
31113f74950SBarry Smith PetscErrorCode MatRestoreRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
31249b5e25fSSatish Balay {
313dfbe8321SBarry Smith   PetscErrorCode ierr;
31449b5e25fSSatish Balay 
31549b5e25fSSatish Balay   PetscFunctionBegin;
31605b42c5fSBarry Smith   if (idx) {ierr = PetscFree(*idx);CHKERRQ(ierr);}
31705b42c5fSBarry Smith   if (v)   {ierr = PetscFree(*v);CHKERRQ(ierr);}
31849b5e25fSSatish Balay   PetscFunctionReturn(0);
31949b5e25fSSatish Balay }
32049b5e25fSSatish Balay 
3214a2ae208SSatish Balay #undef __FUNCT__
322f5edf698SHong Zhang #define __FUNCT__ "MatGetRowUpperTriangular_SeqSBAIJ"
323f5edf698SHong Zhang PetscErrorCode MatGetRowUpperTriangular_SeqSBAIJ(Mat A)
324f5edf698SHong Zhang {
325f5edf698SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
326f5edf698SHong Zhang 
327f5edf698SHong Zhang   PetscFunctionBegin;
328f5edf698SHong Zhang   a->getrow_utriangular = PETSC_TRUE;
329f5edf698SHong Zhang   PetscFunctionReturn(0);
330f5edf698SHong Zhang }
331f5edf698SHong Zhang #undef __FUNCT__
332f5edf698SHong Zhang #define __FUNCT__ "MatRestoreRowUpperTriangular_SeqSBAIJ"
333f5edf698SHong Zhang PetscErrorCode MatRestoreRowUpperTriangular_SeqSBAIJ(Mat A)
334f5edf698SHong Zhang {
335f5edf698SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
336f5edf698SHong Zhang 
337f5edf698SHong Zhang   PetscFunctionBegin;
338f5edf698SHong Zhang   a->getrow_utriangular = PETSC_FALSE;
339f5edf698SHong Zhang   PetscFunctionReturn(0);
340f5edf698SHong Zhang }
341f5edf698SHong Zhang 
342f5edf698SHong Zhang #undef __FUNCT__
3434a2ae208SSatish Balay #define __FUNCT__ "MatTranspose_SeqSBAIJ"
344fc4dec0aSBarry Smith PetscErrorCode MatTranspose_SeqSBAIJ(Mat A,MatReuse reuse,Mat *B)
34549b5e25fSSatish Balay {
346dfbe8321SBarry Smith   PetscErrorCode ierr;
3475fd66863SKarl Rupp 
34849b5e25fSSatish Balay   PetscFunctionBegin;
349815cbec1SBarry Smith   if (reuse == MAT_INITIAL_MATRIX || *B != A) {
350999d9058SBarry Smith     ierr = MatDuplicate(A,MAT_COPY_VALUES,B);CHKERRQ(ierr);
351fc4dec0aSBarry Smith   }
3528115998fSBarry Smith   PetscFunctionReturn(0);
35349b5e25fSSatish Balay }
35449b5e25fSSatish Balay 
3554a2ae208SSatish Balay #undef __FUNCT__
3564a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_ASCII"
3577da1fb6eSBarry Smith PetscErrorCode MatView_SeqSBAIJ_ASCII(Mat A,PetscViewer viewer)
35849b5e25fSSatish Balay {
35949b5e25fSSatish Balay   Mat_SeqSBAIJ      *a = (Mat_SeqSBAIJ*)A->data;
360dfbe8321SBarry Smith   PetscErrorCode    ierr;
361d0f46423SBarry Smith   PetscInt          i,j,bs = A->rmap->bs,k,l,bs2=a->bs2;
362f3ef73ceSBarry Smith   PetscViewerFormat format;
363121deb67SSatish Balay   PetscInt          *diag;
36449b5e25fSSatish Balay 
36549b5e25fSSatish Balay   PetscFunctionBegin;
366b0a32e0cSBarry Smith   ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
367456192e2SBarry Smith   if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
36877431f27SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  block size is %D\n",bs);CHKERRQ(ierr);
369fb9695e5SSatish Balay   } else if (format == PETSC_VIEWER_ASCII_MATLAB) {
370d2507d54SMatthew Knepley     Mat aij;
371d5f3da31SBarry Smith     if (A->factortype && bs>1) {
37270d5e725SHong 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);
37370d5e725SHong Zhang       PetscFunctionReturn(0);
37470d5e725SHong Zhang     }
375c9f458caSMatthew Knepley     ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&aij);CHKERRQ(ierr);
376c9f458caSMatthew Knepley     ierr = MatView(aij,viewer);CHKERRQ(ierr);
3776bf464f9SBarry Smith     ierr = MatDestroy(&aij);CHKERRQ(ierr);
378fb9695e5SSatish Balay   } else if (format == PETSC_VIEWER_ASCII_COMMON) {
379d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
38049b5e25fSSatish Balay     for (i=0; i<a->mbs; i++) {
38149b5e25fSSatish Balay       for (j=0; j<bs; j++) {
38277431f27SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr);
38349b5e25fSSatish Balay         for (k=a->i[i]; k<a->i[i+1]; k++) {
38449b5e25fSSatish Balay           for (l=0; l<bs; l++) {
38549b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX)
38649b5e25fSSatish Balay             if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
38757622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k]+l,
38857622a8eSBarry Smith                                             (double)PetscRealPart(a->a[bs2*k + l*bs + j]),(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
38949b5e25fSSatish Balay             } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
39057622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k]+l,
39157622a8eSBarry Smith                                             (double)PetscRealPart(a->a[bs2*k + l*bs + j]),-(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
39249b5e25fSSatish Balay             } else if (PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
39357622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
39449b5e25fSSatish Balay             }
39549b5e25fSSatish Balay #else
39649b5e25fSSatish Balay             if (a->a[bs2*k + l*bs + j] != 0.0) {
39757622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)a->a[bs2*k + l*bs + j]);CHKERRQ(ierr);
39849b5e25fSSatish Balay             }
39949b5e25fSSatish Balay #endif
40049b5e25fSSatish Balay           }
40149b5e25fSSatish Balay         }
402b0a32e0cSBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
40349b5e25fSSatish Balay       }
40449b5e25fSSatish Balay     }
405d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
406c1490034SHong Zhang   } else if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) {
407c1490034SHong Zhang     PetscFunctionReturn(0);
40849b5e25fSSatish Balay   } else {
409d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
4102c990fa1SHong Zhang     if (A->factortype) { /* for factored matrix */
4112c990fa1SHong Zhang       if (bs>1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"matrix is factored with bs>1. Not implemented yet");
4122c990fa1SHong Zhang 
413121deb67SSatish Balay       diag=a->diag;
414121deb67SSatish Balay       for (i=0; i<a->mbs; i++) { /* for row block i */
4152c990fa1SHong Zhang         ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i);CHKERRQ(ierr);
4162c990fa1SHong Zhang         /* diagonal entry */
4172c990fa1SHong Zhang #if defined(PETSC_USE_COMPLEX)
4182c990fa1SHong Zhang         if (PetscImaginaryPart(a->a[diag[i]]) > 0.0) {
41957622a8eSBarry 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);
4202c990fa1SHong Zhang         } else if (PetscImaginaryPart(a->a[diag[i]]) < 0.0) {
42157622a8eSBarry 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);
4222c990fa1SHong Zhang         } else {
42357622a8eSBarry Smith           ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[diag[i]],(double)PetscRealPart(1.0/a->a[diag[i]]));CHKERRQ(ierr);
4242c990fa1SHong Zhang         }
4252c990fa1SHong Zhang #else
4266712e2f1SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[diag[i]],(double)(1.0/a->a[diag[i]]));CHKERRQ(ierr);
4272c990fa1SHong Zhang #endif
4282c990fa1SHong Zhang         /* off-diagonal entries */
4292c990fa1SHong Zhang         for (k=a->i[i]; k<a->i[i+1]-1; k++) {
4302c990fa1SHong Zhang #if defined(PETSC_USE_COMPLEX)
431ca0704adSBarry Smith           if (PetscImaginaryPart(a->a[k]) > 0.0) {
43257622a8eSBarry Smith             ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k],(double)PetscRealPart(a->a[k]),(double)PetscImaginaryPart(a->a[k]));CHKERRQ(ierr);
433ca0704adSBarry Smith           } else if (PetscImaginaryPart(a->a[k]) < 0.0) {
43457622a8eSBarry Smith             ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k],(double)PetscRealPart(a->a[k]),-(double)PetscImaginaryPart(a->a[k]));CHKERRQ(ierr);
4352c990fa1SHong Zhang           } else {
43657622a8eSBarry Smith             ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k],(double)PetscRealPart(a->a[k]));CHKERRQ(ierr);
4372c990fa1SHong Zhang           }
4382c990fa1SHong Zhang #else
43957622a8eSBarry Smith           ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[k],(double)a->a[k]);CHKERRQ(ierr);
4402c990fa1SHong Zhang #endif
4412c990fa1SHong Zhang         }
4422c990fa1SHong Zhang         ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
4432c990fa1SHong Zhang       }
4442c990fa1SHong Zhang 
4452c990fa1SHong Zhang     } else { /* for non-factored matrix */
4460c74a584SJed Brown       for (i=0; i<a->mbs; i++) { /* for row block i */
4470c74a584SJed Brown         for (j=0; j<bs; j++) {   /* for row bs*i + j */
44877431f27SBarry Smith           ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr);
4490c74a584SJed Brown           for (k=a->i[i]; k<a->i[i+1]; k++) { /* for column block */
4500c74a584SJed Brown             for (l=0; l<bs; l++) {            /* for column */
45149b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX)
45249b5e25fSSatish Balay               if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0) {
45357622a8eSBarry Smith                 ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k]+l,
45457622a8eSBarry Smith                                               (double)PetscRealPart(a->a[bs2*k + l*bs + j]),(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
45549b5e25fSSatish Balay               } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0) {
45657622a8eSBarry Smith                 ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k]+l,
45757622a8eSBarry Smith                                               (double)PetscRealPart(a->a[bs2*k + l*bs + j]),-(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
45849b5e25fSSatish Balay               } else {
45957622a8eSBarry Smith                 ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
46049b5e25fSSatish Balay               }
46149b5e25fSSatish Balay #else
46257622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)a->a[bs2*k + l*bs + j]);CHKERRQ(ierr);
46349b5e25fSSatish Balay #endif
46449b5e25fSSatish Balay             }
46549b5e25fSSatish Balay           }
466b0a32e0cSBarry Smith           ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
46749b5e25fSSatish Balay         }
46849b5e25fSSatish Balay       }
4692c990fa1SHong Zhang     }
470d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
47149b5e25fSSatish Balay   }
472b0a32e0cSBarry Smith   ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
47349b5e25fSSatish Balay   PetscFunctionReturn(0);
47449b5e25fSSatish Balay }
47549b5e25fSSatish Balay 
4769804daf3SBarry Smith #include <petscdraw.h>
4774a2ae208SSatish Balay #undef __FUNCT__
4784a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_Draw_Zoom"
4796849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw_Zoom(PetscDraw draw,void *Aa)
48049b5e25fSSatish Balay {
48149b5e25fSSatish Balay   Mat            A = (Mat) Aa;
48249b5e25fSSatish Balay   Mat_SeqSBAIJ   *a=(Mat_SeqSBAIJ*)A->data;
4836849ba73SBarry Smith   PetscErrorCode ierr;
484d0f46423SBarry Smith   PetscInt       row,i,j,k,l,mbs=a->mbs,color,bs=A->rmap->bs,bs2=a->bs2;
48513f74950SBarry Smith   PetscMPIInt    rank;
48649b5e25fSSatish Balay   PetscReal      xl,yl,xr,yr,x_l,x_r,y_l,y_r;
48749b5e25fSSatish Balay   MatScalar      *aa;
48849b5e25fSSatish Balay   MPI_Comm       comm;
489b0a32e0cSBarry Smith   PetscViewer    viewer;
49049b5e25fSSatish Balay 
49149b5e25fSSatish Balay   PetscFunctionBegin;
49249b5e25fSSatish Balay   /*
49349b5e25fSSatish Balay     This is nasty. If this is called from an originally parallel matrix
49449b5e25fSSatish Balay     then all processes call this,but only the first has the matrix so the
49549b5e25fSSatish Balay     rest should return immediately.
49649b5e25fSSatish Balay   */
49749b5e25fSSatish Balay   ierr = PetscObjectGetComm((PetscObject)draw,&comm);CHKERRQ(ierr);
49849b5e25fSSatish Balay   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
49949b5e25fSSatish Balay   if (rank) PetscFunctionReturn(0);
50049b5e25fSSatish Balay 
50149b5e25fSSatish Balay   ierr = PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer);CHKERRQ(ierr);
50249b5e25fSSatish Balay 
503b0a32e0cSBarry Smith   ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr);
504b0a32e0cSBarry Smith   PetscDrawString(draw, .3*(xl+xr), .3*(yl+yr), PETSC_DRAW_BLACK, "symmetric");
50549b5e25fSSatish Balay 
50649b5e25fSSatish Balay   /* loop over matrix elements drawing boxes */
507b0a32e0cSBarry Smith   color = PETSC_DRAW_BLUE;
50849b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
50949b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
510d0f46423SBarry Smith       y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0;
51149b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
51249b5e25fSSatish Balay       aa  = a->a + j*bs2;
51349b5e25fSSatish Balay       for (k=0; k<bs; k++) {
51449b5e25fSSatish Balay         for (l=0; l<bs; l++) {
51549b5e25fSSatish Balay           if (PetscRealPart(*aa++) >=  0.) continue;
516b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
51749b5e25fSSatish Balay         }
51849b5e25fSSatish Balay       }
51949b5e25fSSatish Balay     }
52049b5e25fSSatish Balay   }
521b0a32e0cSBarry Smith   color = PETSC_DRAW_CYAN;
52249b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
52349b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
524d0f46423SBarry Smith       y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0;
52549b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
52649b5e25fSSatish Balay       aa = a->a + j*bs2;
52749b5e25fSSatish Balay       for (k=0; k<bs; k++) {
52849b5e25fSSatish Balay         for (l=0; l<bs; l++) {
52949b5e25fSSatish Balay           if (PetscRealPart(*aa++) != 0.) continue;
530b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
53149b5e25fSSatish Balay         }
53249b5e25fSSatish Balay       }
53349b5e25fSSatish Balay     }
53449b5e25fSSatish Balay   }
53549b5e25fSSatish Balay 
536b0a32e0cSBarry Smith   color = PETSC_DRAW_RED;
53749b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
53849b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
539d0f46423SBarry Smith       y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0;
54049b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
54149b5e25fSSatish Balay       aa = a->a + j*bs2;
54249b5e25fSSatish Balay       for (k=0; k<bs; k++) {
54349b5e25fSSatish Balay         for (l=0; l<bs; l++) {
54449b5e25fSSatish Balay           if (PetscRealPart(*aa++) <= 0.) continue;
545b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
54649b5e25fSSatish Balay         }
54749b5e25fSSatish Balay       }
54849b5e25fSSatish Balay     }
54949b5e25fSSatish Balay   }
55049b5e25fSSatish Balay   PetscFunctionReturn(0);
55149b5e25fSSatish Balay }
55249b5e25fSSatish Balay 
5534a2ae208SSatish Balay #undef __FUNCT__
5544a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_Draw"
5556849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw(Mat A,PetscViewer viewer)
55649b5e25fSSatish Balay {
557dfbe8321SBarry Smith   PetscErrorCode ierr;
55849b5e25fSSatish Balay   PetscReal      xl,yl,xr,yr,w,h;
559b0a32e0cSBarry Smith   PetscDraw      draw;
560ace3abfcSBarry Smith   PetscBool      isnull;
56149b5e25fSSatish Balay 
56249b5e25fSSatish Balay   PetscFunctionBegin;
563b0a32e0cSBarry Smith   ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
564b0a32e0cSBarry Smith   ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr); if (isnull) PetscFunctionReturn(0);
56549b5e25fSSatish Balay 
56649b5e25fSSatish Balay   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer);CHKERRQ(ierr);
567d0f46423SBarry Smith   xr   = A->rmap->N; yr = A->rmap->N; h = yr/10.0; w = xr/10.0;
56849b5e25fSSatish Balay   xr  += w;    yr += h;  xl = -w;     yl = -h;
569b0a32e0cSBarry Smith   ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr);
570b0a32e0cSBarry Smith   ierr = PetscDrawZoom(draw,MatView_SeqSBAIJ_Draw_Zoom,A);CHKERRQ(ierr);
5710298fd71SBarry Smith   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",NULL);CHKERRQ(ierr);
57249b5e25fSSatish Balay   PetscFunctionReturn(0);
57349b5e25fSSatish Balay }
57449b5e25fSSatish Balay 
5754a2ae208SSatish Balay #undef __FUNCT__
5764a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ"
577dfbe8321SBarry Smith PetscErrorCode MatView_SeqSBAIJ(Mat A,PetscViewer viewer)
57849b5e25fSSatish Balay {
579dfbe8321SBarry Smith   PetscErrorCode ierr;
580ace3abfcSBarry Smith   PetscBool      iascii,isdraw;
58108917f38SBarry Smith   FILE           *file = 0;
58249b5e25fSSatish Balay 
58349b5e25fSSatish Balay   PetscFunctionBegin;
584251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
585251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr);
58632077d6dSBarry Smith   if (iascii) {
58749b5e25fSSatish Balay     ierr = MatView_SeqSBAIJ_ASCII(A,viewer);CHKERRQ(ierr);
58849b5e25fSSatish Balay   } else if (isdraw) {
58949b5e25fSSatish Balay     ierr = MatView_SeqSBAIJ_Draw(A,viewer);CHKERRQ(ierr);
59049b5e25fSSatish Balay   } else {
591a5e6ed63SBarry Smith     Mat B;
592ceb03754SKris Buschelman     ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&B);CHKERRQ(ierr);
593a5e6ed63SBarry Smith     ierr = MatView(B,viewer);CHKERRQ(ierr);
5946bf464f9SBarry Smith     ierr = MatDestroy(&B);CHKERRQ(ierr);
59508917f38SBarry Smith     ierr = PetscViewerBinaryGetInfoPointer(viewer,&file);CHKERRQ(ierr);
59608917f38SBarry Smith     if (file) {
59708917f38SBarry Smith       fprintf(file,"-matload_block_size %d\n",(int)A->rmap->bs);
59808917f38SBarry Smith     }
59949b5e25fSSatish Balay   }
60049b5e25fSSatish Balay   PetscFunctionReturn(0);
60149b5e25fSSatish Balay }
60249b5e25fSSatish Balay 
60349b5e25fSSatish Balay 
6044a2ae208SSatish Balay #undef __FUNCT__
6054a2ae208SSatish Balay #define __FUNCT__ "MatGetValues_SeqSBAIJ"
60613f74950SBarry Smith PetscErrorCode MatGetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],PetscScalar v[])
60749b5e25fSSatish Balay {
608045c9aa0SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
60913f74950SBarry Smith   PetscInt     *rp,k,low,high,t,row,nrow,i,col,l,*aj = a->j;
61013f74950SBarry Smith   PetscInt     *ai = a->i,*ailen = a->ilen;
611d0f46423SBarry Smith   PetscInt     brow,bcol,ridx,cidx,bs=A->rmap->bs,bs2=a->bs2;
61297e567efSBarry Smith   MatScalar    *ap,*aa = a->a;
61349b5e25fSSatish Balay 
61449b5e25fSSatish Balay   PetscFunctionBegin;
61549b5e25fSSatish Balay   for (k=0; k<m; k++) { /* loop over rows */
61649b5e25fSSatish Balay     row = im[k]; brow = row/bs;
617e32f2f54SBarry Smith     if (row < 0) {v += n; continue;} /* SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative row: %D",row); */
618e32f2f54SBarry 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);
61949b5e25fSSatish Balay     rp   = aj + ai[brow]; ap = aa + bs2*ai[brow];
62049b5e25fSSatish Balay     nrow = ailen[brow];
62149b5e25fSSatish Balay     for (l=0; l<n; l++) { /* loop over columns */
622e32f2f54SBarry Smith       if (in[l] < 0) {v++; continue;} /* SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column: %D",in[l]); */
623e32f2f54SBarry 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);
62449b5e25fSSatish Balay       col  = in[l];
62549b5e25fSSatish Balay       bcol = col/bs;
62649b5e25fSSatish Balay       cidx = col%bs;
62749b5e25fSSatish Balay       ridx = row%bs;
62849b5e25fSSatish Balay       high = nrow;
62949b5e25fSSatish Balay       low  = 0; /* assume unsorted */
63049b5e25fSSatish Balay       while (high-low > 5) {
63149b5e25fSSatish Balay         t = (low+high)/2;
63249b5e25fSSatish Balay         if (rp[t] > bcol) high = t;
63349b5e25fSSatish Balay         else              low  = t;
63449b5e25fSSatish Balay       }
63549b5e25fSSatish Balay       for (i=low; i<high; i++) {
63649b5e25fSSatish Balay         if (rp[i] > bcol) break;
63749b5e25fSSatish Balay         if (rp[i] == bcol) {
63849b5e25fSSatish Balay           *v++ = ap[bs2*i+bs*cidx+ridx];
63949b5e25fSSatish Balay           goto finished;
64049b5e25fSSatish Balay         }
64149b5e25fSSatish Balay       }
64297e567efSBarry Smith       *v++ = 0.0;
64349b5e25fSSatish Balay finished:;
64449b5e25fSSatish Balay     }
64549b5e25fSSatish Balay   }
64649b5e25fSSatish Balay   PetscFunctionReturn(0);
64749b5e25fSSatish Balay }
64849b5e25fSSatish Balay 
64949b5e25fSSatish Balay 
6504a2ae208SSatish Balay #undef __FUNCT__
6514a2ae208SSatish Balay #define __FUNCT__ "MatSetValuesBlocked_SeqSBAIJ"
65213f74950SBarry Smith PetscErrorCode MatSetValuesBlocked_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is)
65349b5e25fSSatish Balay {
6540880e062SHong Zhang   Mat_SeqSBAIJ      *a = (Mat_SeqSBAIJ*)A->data;
6556849ba73SBarry Smith   PetscErrorCode    ierr;
656e2ee6c50SBarry Smith   PetscInt          *rp,k,low,high,t,ii,jj,row,nrow,i,col,l,rmax,N,lastcol = -1;
65713f74950SBarry Smith   PetscInt          *imax      =a->imax,*ai=a->i,*ailen=a->ilen;
658d0f46423SBarry Smith   PetscInt          *aj        =a->j,nonew=a->nonew,bs2=a->bs2,bs=A->rmap->bs,stepval;
659ace3abfcSBarry Smith   PetscBool         roworiented=a->roworiented;
660dd6ea824SBarry Smith   const PetscScalar *value     = v;
661f15d580aSBarry Smith   MatScalar         *ap,*aa = a->a,*bap;
6620880e062SHong Zhang 
66349b5e25fSSatish Balay   PetscFunctionBegin;
66426fbe8dcSKarl Rupp   if (roworiented) stepval = (n-1)*bs;
66526fbe8dcSKarl Rupp   else stepval = (m-1)*bs;
66626fbe8dcSKarl Rupp 
6670880e062SHong Zhang   for (k=0; k<m; k++) { /* loop over added rows */
6680880e062SHong Zhang     row = im[k];
6690880e062SHong Zhang     if (row < 0) continue;
6702515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
671e32f2f54SBarry Smith     if (row >= a->mbs) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,a->mbs-1);
6720880e062SHong Zhang #endif
6730880e062SHong Zhang     rp   = aj + ai[row];
6740880e062SHong Zhang     ap   = aa + bs2*ai[row];
6750880e062SHong Zhang     rmax = imax[row];
6760880e062SHong Zhang     nrow = ailen[row];
6770880e062SHong Zhang     low  = 0;
678818f2c47SBarry Smith     high = nrow;
6790880e062SHong Zhang     for (l=0; l<n; l++) { /* loop over added columns */
6800880e062SHong Zhang       if (in[l] < 0) continue;
6810880e062SHong Zhang       col = in[l];
6822515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
683e32f2f54SBarry Smith       if (col >= a->nbs) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",col,a->nbs-1);
684b1823623SSatish Balay #endif
685b98bf0e1SJed Brown       if (col < row) {
68626fbe8dcSKarl Rupp         if (a->ignore_ltriangular) continue; /* ignore lower triangular block */
68726fbe8dcSKarl 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)");
688b98bf0e1SJed Brown       }
68926fbe8dcSKarl Rupp       if (roworiented) value = v + k*(stepval+bs)*bs + l*bs;
69026fbe8dcSKarl Rupp       else value = v + l*(stepval+bs)*bs + k*bs;
69126fbe8dcSKarl Rupp 
69226fbe8dcSKarl Rupp       if (col <= lastcol) low = 0;
69326fbe8dcSKarl Rupp       else high = nrow;
69426fbe8dcSKarl Rupp 
695e2ee6c50SBarry Smith       lastcol = col;
6960880e062SHong Zhang       while (high-low > 7) {
6970880e062SHong Zhang         t = (low+high)/2;
6980880e062SHong Zhang         if (rp[t] > col) high = t;
6990880e062SHong Zhang         else             low  = t;
7000880e062SHong Zhang       }
7010880e062SHong Zhang       for (i=low; i<high; i++) {
7020880e062SHong Zhang         if (rp[i] > col) break;
7030880e062SHong Zhang         if (rp[i] == col) {
7040880e062SHong Zhang           bap = ap +  bs2*i;
7050880e062SHong Zhang           if (roworiented) {
7060880e062SHong Zhang             if (is == ADD_VALUES) {
7070880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
7080880e062SHong Zhang                 for (jj=ii; jj<bs2; jj+=bs) {
7090880e062SHong Zhang                   bap[jj] += *value++;
7100880e062SHong Zhang                 }
7110880e062SHong Zhang               }
7120880e062SHong Zhang             } else {
7130880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
7140880e062SHong Zhang                 for (jj=ii; jj<bs2; jj+=bs) {
7150880e062SHong Zhang                   bap[jj] = *value++;
7160880e062SHong Zhang                 }
7170880e062SHong Zhang                }
7180880e062SHong Zhang             }
7190880e062SHong Zhang           } else {
7200880e062SHong Zhang             if (is == ADD_VALUES) {
7210880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
7220880e062SHong Zhang                 for (jj=0; jj<bs; jj++) {
7230880e062SHong Zhang                   *bap++ += *value++;
7240880e062SHong Zhang                 }
7250880e062SHong Zhang               }
7260880e062SHong Zhang             } else {
7270880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
7280880e062SHong Zhang                 for (jj=0; jj<bs; jj++) {
7290880e062SHong Zhang                   *bap++  = *value++;
7300880e062SHong Zhang                 }
7310880e062SHong Zhang               }
7320880e062SHong Zhang             }
7330880e062SHong Zhang           }
7340880e062SHong Zhang           goto noinsert2;
7350880e062SHong Zhang         }
7360880e062SHong Zhang       }
7370880e062SHong Zhang       if (nonew == 1) goto noinsert2;
738e32f2f54SBarry Smith       if (nonew == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col);
739fef13f97SBarry Smith       MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,row,col,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar);
740c03d1d03SSatish Balay       N = nrow++ - 1; high++;
7410880e062SHong Zhang       /* shift up all the later entries in this row */
7420880e062SHong Zhang       for (ii=N; ii>=i; ii--) {
7430880e062SHong Zhang         rp[ii+1] = rp[ii];
7440880e062SHong Zhang         ierr     = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr);
7450880e062SHong Zhang       }
7460880e062SHong Zhang       if (N >= i) {
7470880e062SHong Zhang         ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr);
7480880e062SHong Zhang       }
7490880e062SHong Zhang       rp[i] = col;
7500880e062SHong Zhang       bap   = ap +  bs2*i;
7510880e062SHong Zhang       if (roworiented) {
7520880e062SHong Zhang         for (ii=0; ii<bs; ii++,value+=stepval) {
7530880e062SHong Zhang           for (jj=ii; jj<bs2; jj+=bs) {
7540880e062SHong Zhang             bap[jj] = *value++;
7550880e062SHong Zhang           }
7560880e062SHong Zhang         }
7570880e062SHong Zhang       } else {
7580880e062SHong Zhang         for (ii=0; ii<bs; ii++,value+=stepval) {
7590880e062SHong Zhang           for (jj=0; jj<bs; jj++) {
7600880e062SHong Zhang             *bap++ = *value++;
7610880e062SHong Zhang           }
7620880e062SHong Zhang         }
7630880e062SHong Zhang        }
7640880e062SHong Zhang     noinsert2:;
7650880e062SHong Zhang       low = i;
7660880e062SHong Zhang     }
7670880e062SHong Zhang     ailen[row] = nrow;
7680880e062SHong Zhang   }
7690880e062SHong Zhang   PetscFunctionReturn(0);
77049b5e25fSSatish Balay }
77149b5e25fSSatish Balay 
77264831d72SBarry Smith /*
77364831d72SBarry Smith     This is not yet used
77464831d72SBarry Smith */
7754a2ae208SSatish Balay #undef __FUNCT__
7764108e4d5SBarry Smith #define __FUNCT__ "MatAssemblyEnd_SeqSBAIJ_SeqAIJ_Inode"
7774108e4d5SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ_SeqAIJ_Inode(Mat A)
7780def2e27SBarry Smith {
7790def2e27SBarry Smith   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
7800def2e27SBarry Smith   PetscErrorCode ierr;
7810def2e27SBarry Smith   const PetscInt *ai = a->i, *aj = a->j,*cols;
7820def2e27SBarry Smith   PetscInt       i   = 0,j,blk_size,m = A->rmap->n,node_count = 0,nzx,nzy,*ns,row,nz,cnt,cnt2,*counts;
783ace3abfcSBarry Smith   PetscBool      flag;
7840def2e27SBarry Smith 
7850def2e27SBarry Smith   PetscFunctionBegin;
786785e854fSJed Brown   ierr = PetscMalloc1(m,&ns);CHKERRQ(ierr);
7870def2e27SBarry Smith   while (i < m) {
7880def2e27SBarry Smith     nzx = ai[i+1] - ai[i];       /* Number of nonzeros */
7890def2e27SBarry Smith     /* Limits the number of elements in a node to 'a->inode.limit' */
7900def2e27SBarry Smith     for (j=i+1,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) {
7910def2e27SBarry Smith       nzy = ai[j+1] - ai[j];
7920def2e27SBarry Smith       if (nzy != (nzx - j + i)) break;
7930def2e27SBarry Smith       ierr = PetscMemcmp(aj + ai[i] + j - i,aj + ai[j],nzy*sizeof(PetscInt),&flag);CHKERRQ(ierr);
7940def2e27SBarry Smith       if (!flag) break;
7950def2e27SBarry Smith     }
7960def2e27SBarry Smith     ns[node_count++] = blk_size;
79726fbe8dcSKarl Rupp 
7980def2e27SBarry Smith     i = j;
7990def2e27SBarry Smith   }
8000def2e27SBarry Smith   if (!a->inode.size && m && node_count > .9*m) {
8010def2e27SBarry Smith     ierr = PetscFree(ns);CHKERRQ(ierr);
8020def2e27SBarry Smith     ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr);
8030def2e27SBarry Smith   } else {
8040def2e27SBarry Smith     a->inode.node_count = node_count;
80526fbe8dcSKarl Rupp 
806785e854fSJed Brown     ierr = PetscMalloc1(node_count,&a->inode.size);CHKERRQ(ierr);
8073bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)A,node_count*sizeof(PetscInt));CHKERRQ(ierr);
80822d28d08SBarry Smith     ierr = PetscMemcpy(a->inode.size,ns,node_count*sizeof(PetscInt));CHKERRQ(ierr);
8090def2e27SBarry Smith     ierr = PetscFree(ns);CHKERRQ(ierr);
8100def2e27SBarry Smith     ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr);
8110def2e27SBarry Smith 
8120def2e27SBarry Smith     /* count collections of adjacent columns in each inode */
8130def2e27SBarry Smith     row = 0;
8140def2e27SBarry Smith     cnt = 0;
8150def2e27SBarry Smith     for (i=0; i<node_count; i++) {
8160def2e27SBarry Smith       cols = aj + ai[row] + a->inode.size[i];
8170def2e27SBarry Smith       nz   = ai[row+1] - ai[row] - a->inode.size[i];
8180def2e27SBarry Smith       for (j=1; j<nz; j++) {
81926fbe8dcSKarl Rupp         if (cols[j] != cols[j-1]+1) cnt++;
8200def2e27SBarry Smith       }
8210def2e27SBarry Smith       cnt++;
8220def2e27SBarry Smith       row += a->inode.size[i];
8230def2e27SBarry Smith     }
824785e854fSJed Brown     ierr = PetscMalloc1(2*cnt,&counts);CHKERRQ(ierr);
8250def2e27SBarry Smith     cnt  = 0;
8260def2e27SBarry Smith     row  = 0;
8270def2e27SBarry Smith     for (i=0; i<node_count; i++) {
8280def2e27SBarry Smith       cols = aj + ai[row] + a->inode.size[i];
8290def2e27SBarry Smith       counts[2*cnt] = cols[0];
8300def2e27SBarry Smith       nz   = ai[row+1] - ai[row] - a->inode.size[i];
8310def2e27SBarry Smith       cnt2 = 1;
8320def2e27SBarry Smith       for (j=1; j<nz; j++) {
8330def2e27SBarry Smith         if (cols[j] != cols[j-1]+1) {
8340def2e27SBarry Smith           counts[2*(cnt++)+1] = cnt2;
8350def2e27SBarry Smith           counts[2*cnt]       = cols[j];
8360def2e27SBarry Smith           cnt2 = 1;
8370def2e27SBarry Smith         } else cnt2++;
8380def2e27SBarry Smith       }
8390def2e27SBarry Smith       counts[2*(cnt++)+1] = cnt2;
8400def2e27SBarry Smith       row += a->inode.size[i];
8410def2e27SBarry Smith     }
84222d28d08SBarry Smith     ierr = PetscIntView(2*cnt,counts,0);CHKERRQ(ierr);
8430def2e27SBarry Smith   }
84438702af4SBarry Smith   PetscFunctionReturn(0);
84538702af4SBarry Smith }
84638702af4SBarry Smith 
84738702af4SBarry Smith #undef __FUNCT__
8484a2ae208SSatish Balay #define __FUNCT__ "MatAssemblyEnd_SeqSBAIJ"
849dfbe8321SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ(Mat A,MatAssemblyType mode)
85049b5e25fSSatish Balay {
85149b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
8526849ba73SBarry Smith   PetscErrorCode ierr;
85313f74950SBarry Smith   PetscInt       fshift = 0,i,j,*ai = a->i,*aj = a->j,*imax = a->imax;
854d0f46423SBarry Smith   PetscInt       m      = A->rmap->N,*ip,N,*ailen = a->ilen;
85513f74950SBarry Smith   PetscInt       mbs    = a->mbs,bs2 = a->bs2,rmax = 0;
85649b5e25fSSatish Balay   MatScalar      *aa    = a->a,*ap;
85749b5e25fSSatish Balay 
85849b5e25fSSatish Balay   PetscFunctionBegin;
85949b5e25fSSatish Balay   if (mode == MAT_FLUSH_ASSEMBLY) PetscFunctionReturn(0);
86049b5e25fSSatish Balay 
86149b5e25fSSatish Balay   if (m) rmax = ailen[0];
86249b5e25fSSatish Balay   for (i=1; i<mbs; i++) {
86349b5e25fSSatish Balay     /* move each row back by the amount of empty slots (fshift) before it*/
86449b5e25fSSatish Balay     fshift += imax[i-1] - ailen[i-1];
86549b5e25fSSatish Balay     rmax    = PetscMax(rmax,ailen[i]);
86649b5e25fSSatish Balay     if (fshift) {
86749b5e25fSSatish Balay       ip = aj + ai[i]; ap = aa + bs2*ai[i];
86849b5e25fSSatish Balay       N  = ailen[i];
86949b5e25fSSatish Balay       for (j=0; j<N; j++) {
87049b5e25fSSatish Balay         ip[j-fshift] = ip[j];
87149b5e25fSSatish Balay         ierr         = PetscMemcpy(ap+(j-fshift)*bs2,ap+j*bs2,bs2*sizeof(MatScalar));CHKERRQ(ierr);
87249b5e25fSSatish Balay       }
87349b5e25fSSatish Balay     }
87449b5e25fSSatish Balay     ai[i] = ai[i-1] + ailen[i-1];
87549b5e25fSSatish Balay   }
87649b5e25fSSatish Balay   if (mbs) {
87749b5e25fSSatish Balay     fshift += imax[mbs-1] - ailen[mbs-1];
87849b5e25fSSatish Balay     ai[mbs] = ai[mbs-1] + ailen[mbs-1];
87949b5e25fSSatish Balay   }
88049b5e25fSSatish Balay   /* reset ilen and imax for each row */
88149b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
88249b5e25fSSatish Balay     ailen[i] = imax[i] = ai[i+1] - ai[i];
88349b5e25fSSatish Balay   }
8846c6c5352SBarry Smith   a->nz = ai[mbs];
88549b5e25fSSatish Balay 
886b424e231SHong Zhang   /* diagonals may have moved, reset it */
887b424e231SHong Zhang   if (a->diag) {
8882ed38d0bSJed Brown     ierr = PetscMemcpy(a->diag,ai,mbs*sizeof(PetscInt));CHKERRQ(ierr);
88949b5e25fSSatish Balay   }
89026fbe8dcSKarl 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);
89126fbe8dcSKarl Rupp 
892d0f46423SBarry 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);
893ae15b995SBarry Smith   ierr = PetscInfo1(A,"Number of mallocs during MatSetValues is %D\n",a->reallocs);CHKERRQ(ierr);
894ae15b995SBarry Smith   ierr = PetscInfo1(A,"Most nonzeros blocks in any row is %D\n",rmax);CHKERRQ(ierr);
89526fbe8dcSKarl Rupp 
8968e58a170SBarry Smith   A->info.mallocs    += a->reallocs;
89749b5e25fSSatish Balay   a->reallocs         = 0;
89849b5e25fSSatish Balay   A->info.nz_unneeded = (PetscReal)fshift*bs2;
899061b2667SBarry Smith   a->idiagvalid       = PETSC_FALSE;
90038702af4SBarry Smith 
90138702af4SBarry Smith   if (A->cmap->n < 65536 && A->cmap->bs == 1) {
90244e1c64aSLisandro Dalcin     if (a->jshort && a->free_jshort) {
90317803ae8SHong Zhang       /* when matrix data structure is changed, previous jshort must be replaced */
90417803ae8SHong Zhang       ierr = PetscFree(a->jshort);CHKERRQ(ierr);
90517803ae8SHong Zhang     }
906785e854fSJed Brown     ierr = PetscMalloc1(a->i[A->rmap->n],&a->jshort);CHKERRQ(ierr);
9073bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)A,a->i[A->rmap->n]*sizeof(unsigned short));CHKERRQ(ierr);
90838702af4SBarry Smith     for (i=0; i<a->i[A->rmap->n]; i++) a->jshort[i] = a->j[i];
90938702af4SBarry Smith     A->ops->mult   = MatMult_SeqSBAIJ_1_ushort;
91041f059aeSBarry Smith     A->ops->sor    = MatSOR_SeqSBAIJ_ushort;
9114da8f245SBarry Smith     a->free_jshort = PETSC_TRUE;
91238702af4SBarry Smith   }
91349b5e25fSSatish Balay   PetscFunctionReturn(0);
91449b5e25fSSatish Balay }
91549b5e25fSSatish Balay 
91649b5e25fSSatish Balay /*
91749b5e25fSSatish Balay    This function returns an array of flags which indicate the locations of contiguous
91849b5e25fSSatish Balay    blocks that should be zeroed. for eg: if bs = 3  and is = [0,1,2,3,5,6,7,8,9]
91949b5e25fSSatish Balay    then the resulting sizes = [3,1,1,3,1] correspondig to sets [(0,1,2),(3),(5),(6,7,8),(9)]
92049b5e25fSSatish Balay    Assume: sizes should be long enough to hold all the values.
92149b5e25fSSatish Balay */
9224a2ae208SSatish Balay #undef __FUNCT__
9234a2ae208SSatish Balay #define __FUNCT__ "MatZeroRows_SeqSBAIJ_Check_Blocks"
92413f74950SBarry Smith PetscErrorCode MatZeroRows_SeqSBAIJ_Check_Blocks(PetscInt idx[],PetscInt n,PetscInt bs,PetscInt sizes[], PetscInt *bs_max)
92549b5e25fSSatish Balay {
92613f74950SBarry Smith   PetscInt  i,j,k,row;
927ace3abfcSBarry Smith   PetscBool flg;
92849b5e25fSSatish Balay 
92949b5e25fSSatish Balay   PetscFunctionBegin;
93049b5e25fSSatish Balay   for (i=0,j=0; i<n; j++) {
93149b5e25fSSatish Balay     row = idx[i];
93249b5e25fSSatish Balay     if (row%bs!=0) { /* Not the begining of a block */
93349b5e25fSSatish Balay       sizes[j] = 1;
93449b5e25fSSatish Balay       i++;
93549b5e25fSSatish Balay     } else if (i+bs > n) { /* Beginning of a block, but complete block doesn't exist (at idx end) */
93649b5e25fSSatish Balay       sizes[j] = 1;         /* Also makes sure atleast 'bs' values exist for next else */
93749b5e25fSSatish Balay       i++;
93849b5e25fSSatish Balay     } else { /* Begining of the block, so check if the complete block exists */
93949b5e25fSSatish Balay       flg = PETSC_TRUE;
94049b5e25fSSatish Balay       for (k=1; k<bs; k++) {
94149b5e25fSSatish Balay         if (row+k != idx[i+k]) { /* break in the block */
94249b5e25fSSatish Balay           flg = PETSC_FALSE;
94349b5e25fSSatish Balay           break;
94449b5e25fSSatish Balay         }
94549b5e25fSSatish Balay       }
946abc0a331SBarry Smith       if (flg) { /* No break in the bs */
94749b5e25fSSatish Balay         sizes[j] = bs;
94849b5e25fSSatish Balay         i       += bs;
94949b5e25fSSatish Balay       } else {
95049b5e25fSSatish Balay         sizes[j] = 1;
95149b5e25fSSatish Balay         i++;
95249b5e25fSSatish Balay       }
95349b5e25fSSatish Balay     }
95449b5e25fSSatish Balay   }
95549b5e25fSSatish Balay   *bs_max = j;
95649b5e25fSSatish Balay   PetscFunctionReturn(0);
95749b5e25fSSatish Balay }
95849b5e25fSSatish Balay 
95949b5e25fSSatish Balay 
96049b5e25fSSatish Balay /* Only add/insert a(i,j) with i<=j (blocks).
96149b5e25fSSatish Balay    Any a(i,j) with i>j input by user is ingored.
96249b5e25fSSatish Balay */
96349b5e25fSSatish Balay 
9644a2ae208SSatish Balay #undef __FUNCT__
9654a2ae208SSatish Balay #define __FUNCT__ "MatSetValues_SeqSBAIJ"
96613f74950SBarry Smith PetscErrorCode MatSetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is)
96749b5e25fSSatish Balay {
96849b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
9696849ba73SBarry Smith   PetscErrorCode ierr;
970e2ee6c50SBarry Smith   PetscInt       *rp,k,low,high,t,ii,row,nrow,i,col,l,rmax,N,lastcol = -1;
97113f74950SBarry Smith   PetscInt       *imax=a->imax,*ai=a->i,*ailen=a->ilen,roworiented=a->roworiented;
972d0f46423SBarry Smith   PetscInt       *aj  =a->j,nonew=a->nonew,bs=A->rmap->bs,brow,bcol;
97313f74950SBarry Smith   PetscInt       ridx,cidx,bs2=a->bs2;
97449b5e25fSSatish Balay   MatScalar      *ap,value,*aa=a->a,*bap;
97549b5e25fSSatish Balay 
97649b5e25fSSatish Balay   PetscFunctionBegin;
97749b5e25fSSatish Balay   for (k=0; k<m; k++) { /* loop over added rows */
97849b5e25fSSatish Balay     row  = im[k];       /* row number */
97949b5e25fSSatish Balay     brow = row/bs;      /* block row number */
98049b5e25fSSatish Balay     if (row < 0) continue;
9812515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
982e32f2f54SBarry 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);
98349b5e25fSSatish Balay #endif
98449b5e25fSSatish Balay     rp   = aj + ai[brow]; /*ptr to beginning of column value of the row block*/
98549b5e25fSSatish Balay     ap   = aa + bs2*ai[brow]; /*ptr to beginning of element value of the row block*/
98649b5e25fSSatish Balay     rmax = imax[brow];  /* maximum space allocated for this row */
98749b5e25fSSatish Balay     nrow = ailen[brow]; /* actual length of this row */
98849b5e25fSSatish Balay     low  = 0;
98949b5e25fSSatish Balay 
99049b5e25fSSatish Balay     for (l=0; l<n; l++) { /* loop over added columns */
99149b5e25fSSatish Balay       if (in[l] < 0) continue;
9922515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
993e32f2f54SBarry 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);
99449b5e25fSSatish Balay #endif
99549b5e25fSSatish Balay       col  = in[l];
99649b5e25fSSatish Balay       bcol = col/bs;              /* block col number */
99749b5e25fSSatish Balay 
998941593c8SHong Zhang       if (brow > bcol) {
99926fbe8dcSKarl Rupp         if (a->ignore_ltriangular) continue; /* ignore lower triangular values */
100026fbe8dcSKarl 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)");
1001941593c8SHong Zhang       }
1002f4989cb3SHong Zhang 
100349b5e25fSSatish Balay       ridx = row % bs; cidx = col % bs; /*row and col index inside the block */
10048549e402SHong Zhang       if ((brow==bcol && ridx<=cidx) || (brow<bcol)) {
100549b5e25fSSatish Balay         /* element value a(k,l) */
100626fbe8dcSKarl Rupp         if (roworiented) value = v[l + k*n];
100726fbe8dcSKarl Rupp         else value = v[k + l*m];
100849b5e25fSSatish Balay 
100949b5e25fSSatish Balay         /* move pointer bap to a(k,l) quickly and add/insert value */
101026fbe8dcSKarl Rupp         if (col <= lastcol) low = 0;
101126fbe8dcSKarl Rupp         high = nrow;
1012e2ee6c50SBarry Smith         lastcol = col;
101349b5e25fSSatish Balay         while (high-low > 7) {
101449b5e25fSSatish Balay           t = (low+high)/2;
101549b5e25fSSatish Balay           if (rp[t] > bcol) high = t;
101649b5e25fSSatish Balay           else              low  = t;
101749b5e25fSSatish Balay         }
101849b5e25fSSatish Balay         for (i=low; i<high; i++) {
101949b5e25fSSatish Balay           if (rp[i] > bcol) break;
102049b5e25fSSatish Balay           if (rp[i] == bcol) {
102149b5e25fSSatish Balay             bap = ap +  bs2*i + bs*cidx + ridx;
102249b5e25fSSatish Balay             if (is == ADD_VALUES) *bap += value;
102349b5e25fSSatish Balay             else                  *bap  = value;
10248549e402SHong Zhang             /* for diag block, add/insert its symmetric element a(cidx,ridx) */
10258549e402SHong Zhang             if (brow == bcol && ridx < cidx) {
10268549e402SHong Zhang               bap = ap +  bs2*i + bs*ridx + cidx;
10278549e402SHong Zhang               if (is == ADD_VALUES) *bap += value;
10288549e402SHong Zhang               else                  *bap  = value;
10298549e402SHong Zhang             }
103049b5e25fSSatish Balay             goto noinsert1;
103149b5e25fSSatish Balay           }
103249b5e25fSSatish Balay         }
103349b5e25fSSatish Balay 
103449b5e25fSSatish Balay         if (nonew == 1) goto noinsert1;
1035e32f2f54SBarry Smith         if (nonew == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col);
1036fef13f97SBarry Smith         MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,brow,bcol,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar);
103749b5e25fSSatish Balay 
1038c03d1d03SSatish Balay         N = nrow++ - 1; high++;
103949b5e25fSSatish Balay         /* shift up all the later entries in this row */
104049b5e25fSSatish Balay         for (ii=N; ii>=i; ii--) {
104149b5e25fSSatish Balay           rp[ii+1] = rp[ii];
104249b5e25fSSatish Balay           ierr     = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr);
104349b5e25fSSatish Balay         }
104449b5e25fSSatish Balay         if (N>=i) {
104549b5e25fSSatish Balay           ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr);
104649b5e25fSSatish Balay         }
104749b5e25fSSatish Balay         rp[i]                      = bcol;
104849b5e25fSSatish Balay         ap[bs2*i + bs*cidx + ridx] = value;
1049e56f5c9eSBarry Smith         A->nonzerostate++;
105049b5e25fSSatish Balay noinsert1:;
105149b5e25fSSatish Balay         low = i;
10528549e402SHong Zhang       }
105349b5e25fSSatish Balay     }   /* end of loop over added columns */
105449b5e25fSSatish Balay     ailen[brow] = nrow;
105549b5e25fSSatish Balay   }   /* end of loop over added rows */
105649b5e25fSSatish Balay   PetscFunctionReturn(0);
105749b5e25fSSatish Balay }
105849b5e25fSSatish Balay 
10594a2ae208SSatish Balay #undef __FUNCT__
10604d101231SSatish Balay #define __FUNCT__ "MatICCFactor_SeqSBAIJ"
10610481f469SBarry Smith PetscErrorCode MatICCFactor_SeqSBAIJ(Mat inA,IS row,const MatFactorInfo *info)
106249b5e25fSSatish Balay {
10634ccecd49SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)inA->data;
106449b5e25fSSatish Balay   Mat            outA;
1065dfbe8321SBarry Smith   PetscErrorCode ierr;
1066ace3abfcSBarry Smith   PetscBool      row_identity;
106749b5e25fSSatish Balay 
106849b5e25fSSatish Balay   PetscFunctionBegin;
1069e32f2f54SBarry Smith   if (info->levels != 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only levels=0 is supported for in-place icc");
1070c84f5b01SHong Zhang   ierr = ISIdentity(row,&row_identity);CHKERRQ(ierr);
1071e32f2f54SBarry Smith   if (!row_identity) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix reordering is not supported");
1072e32f2f54SBarry 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()! */
1073c84f5b01SHong Zhang 
107449b5e25fSSatish Balay   outA            = inA;
1075d5f3da31SBarry Smith   inA->factortype = MAT_FACTOR_ICC;
107649b5e25fSSatish Balay 
10771a3463dfSHong Zhang   ierr = MatMarkDiagonal_SeqSBAIJ(inA);CHKERRQ(ierr);
1078d595f711SHong Zhang   ierr = MatSeqSBAIJSetNumericFactorization_inplace(inA,row_identity);CHKERRQ(ierr);
107949b5e25fSSatish Balay 
1080c3122656SLisandro Dalcin   ierr   = PetscObjectReference((PetscObject)row);CHKERRQ(ierr);
10816bf464f9SBarry Smith   ierr   = ISDestroy(&a->row);CHKERRQ(ierr);
1082c84f5b01SHong Zhang   a->row = row;
1083c3122656SLisandro Dalcin   ierr   = PetscObjectReference((PetscObject)row);CHKERRQ(ierr);
10846bf464f9SBarry Smith   ierr   = ISDestroy(&a->col);CHKERRQ(ierr);
1085c84f5b01SHong Zhang   a->col = row;
1086c84f5b01SHong Zhang 
1087c84f5b01SHong Zhang   /* Create the invert permutation so that it can be used in MatCholeskyFactorNumeric() */
1088c84f5b01SHong Zhang   if (a->icol) {ierr = ISInvertPermutation(row,PETSC_DECIDE, &a->icol);CHKERRQ(ierr);}
10893bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)inA,(PetscObject)a->icol);CHKERRQ(ierr);
109049b5e25fSSatish Balay 
109149b5e25fSSatish Balay   if (!a->solve_work) {
1092785e854fSJed Brown     ierr = PetscMalloc1((inA->rmap->N+inA->rmap->bs),&a->solve_work);CHKERRQ(ierr);
10933bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)inA,(inA->rmap->N+inA->rmap->bs)*sizeof(PetscScalar));CHKERRQ(ierr);
109449b5e25fSSatish Balay   }
109549b5e25fSSatish Balay 
1096719d5645SBarry Smith   ierr = MatCholeskyFactorNumeric(outA,inA,info);CHKERRQ(ierr);
109749b5e25fSSatish Balay   PetscFunctionReturn(0);
109849b5e25fSSatish Balay }
1099950f1e5bSHong Zhang 
11004a2ae208SSatish Balay #undef __FUNCT__
11014a2ae208SSatish Balay #define __FUNCT__ "MatSeqSBAIJSetColumnIndices_SeqSBAIJ"
11027087cfbeSBarry Smith PetscErrorCode  MatSeqSBAIJSetColumnIndices_SeqSBAIJ(Mat mat,PetscInt *indices)
110349b5e25fSSatish Balay {
1104045c9aa0SHong Zhang   Mat_SeqSBAIJ   *baij = (Mat_SeqSBAIJ*)mat->data;
110513f74950SBarry Smith   PetscInt       i,nz,n;
11067827cd58SJed Brown   PetscErrorCode ierr;
110749b5e25fSSatish Balay 
110849b5e25fSSatish Balay   PetscFunctionBegin;
11096c6c5352SBarry Smith   nz = baij->maxnz;
1110d0f46423SBarry Smith   n  = mat->cmap->n;
111126fbe8dcSKarl Rupp   for (i=0; i<nz; i++) baij->j[i] = indices[i];
111226fbe8dcSKarl Rupp 
11136c6c5352SBarry Smith   baij->nz = nz;
111426fbe8dcSKarl Rupp   for (i=0; i<n; i++) baij->ilen[i] = baij->imax[i];
111526fbe8dcSKarl Rupp 
11167827cd58SJed Brown   ierr = MatSetOption(mat,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
111749b5e25fSSatish Balay   PetscFunctionReturn(0);
111849b5e25fSSatish Balay }
111949b5e25fSSatish Balay 
11204a2ae208SSatish Balay #undef __FUNCT__
11214a2ae208SSatish Balay #define __FUNCT__ "MatSeqSBAIJSetColumnIndices"
112249b5e25fSSatish Balay /*@
112319585528SSatish Balay   MatSeqSBAIJSetColumnIndices - Set the column indices for all the rows
112449b5e25fSSatish Balay   in the matrix.
112549b5e25fSSatish Balay 
112649b5e25fSSatish Balay   Input Parameters:
112719585528SSatish Balay   +  mat     - the SeqSBAIJ matrix
112849b5e25fSSatish Balay   -  indices - the column indices
112949b5e25fSSatish Balay 
113049b5e25fSSatish Balay   Level: advanced
113149b5e25fSSatish Balay 
113249b5e25fSSatish Balay   Notes:
113349b5e25fSSatish Balay   This can be called if you have precomputed the nonzero structure of the
113449b5e25fSSatish Balay   matrix and want to provide it to the matrix object to improve the performance
113549b5e25fSSatish Balay   of the MatSetValues() operation.
113649b5e25fSSatish Balay 
113749b5e25fSSatish Balay   You MUST have set the correct numbers of nonzeros per row in the call to
1138d1be2dadSMatthew Knepley   MatCreateSeqSBAIJ(), and the columns indices MUST be sorted.
113949b5e25fSSatish Balay 
1140ab9f2c04SSatish Balay   MUST be called before any calls to MatSetValues()
114149b5e25fSSatish Balay 
1142ab9f2c04SSatish Balay   .seealso: MatCreateSeqSBAIJ
114349b5e25fSSatish Balay @*/
11447087cfbeSBarry Smith PetscErrorCode  MatSeqSBAIJSetColumnIndices(Mat mat,PetscInt *indices)
114549b5e25fSSatish Balay {
11464ac538c5SBarry Smith   PetscErrorCode ierr;
114749b5e25fSSatish Balay 
114849b5e25fSSatish Balay   PetscFunctionBegin;
11490700a824SBarry Smith   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
11504482741eSBarry Smith   PetscValidPointer(indices,2);
11514ac538c5SBarry Smith   ierr = PetscUseMethod(mat,"MatSeqSBAIJSetColumnIndices_C",(Mat,PetscInt*),(mat,indices));CHKERRQ(ierr);
115249b5e25fSSatish Balay   PetscFunctionReturn(0);
115349b5e25fSSatish Balay }
115449b5e25fSSatish Balay 
11554a2ae208SSatish Balay #undef __FUNCT__
11563c896bc6SHong Zhang #define __FUNCT__ "MatCopy_SeqSBAIJ"
11573c896bc6SHong Zhang PetscErrorCode MatCopy_SeqSBAIJ(Mat A,Mat B,MatStructure str)
11583c896bc6SHong Zhang {
11593c896bc6SHong Zhang   PetscErrorCode ierr;
11603c896bc6SHong Zhang 
11613c896bc6SHong Zhang   PetscFunctionBegin;
11623c896bc6SHong Zhang   /* If the two matrices have the same copy implementation, use fast copy. */
11633c896bc6SHong Zhang   if (str == SAME_NONZERO_PATTERN && (A->ops->copy == B->ops->copy)) {
11643c896bc6SHong Zhang     Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
11653c896bc6SHong Zhang     Mat_SeqSBAIJ *b = (Mat_SeqSBAIJ*)B->data;
11663c896bc6SHong Zhang 
1167e7e72b3dSBarry 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");
1168d0f46423SBarry Smith     ierr = PetscMemcpy(b->a,a->a,(a->i[A->rmap->N])*sizeof(PetscScalar));CHKERRQ(ierr);
11693c896bc6SHong Zhang   } else {
1170f5edf698SHong Zhang     ierr = MatGetRowUpperTriangular(A);CHKERRQ(ierr);
11713c896bc6SHong Zhang     ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr);
1172f5edf698SHong Zhang     ierr = MatRestoreRowUpperTriangular(A);CHKERRQ(ierr);
11733c896bc6SHong Zhang   }
11743c896bc6SHong Zhang   PetscFunctionReturn(0);
11753c896bc6SHong Zhang }
11763c896bc6SHong Zhang 
11773c896bc6SHong Zhang #undef __FUNCT__
11784994cf47SJed Brown #define __FUNCT__ "MatSetUp_SeqSBAIJ"
11794994cf47SJed Brown PetscErrorCode MatSetUp_SeqSBAIJ(Mat A)
1180273d9f13SBarry Smith {
1181dfbe8321SBarry Smith   PetscErrorCode ierr;
1182273d9f13SBarry Smith 
1183273d9f13SBarry Smith   PetscFunctionBegin;
1184535b19f3SBarry Smith   ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(A,A->rmap->bs,PETSC_DEFAULT,0);CHKERRQ(ierr);
1185273d9f13SBarry Smith   PetscFunctionReturn(0);
1186273d9f13SBarry Smith }
1187273d9f13SBarry Smith 
1188a6ece127SHong Zhang #undef __FUNCT__
11898c778c55SBarry Smith #define __FUNCT__ "MatSeqSBAIJGetArray_SeqSBAIJ"
11908c778c55SBarry Smith PetscErrorCode MatSeqSBAIJGetArray_SeqSBAIJ(Mat A,PetscScalar *array[])
1191a6ece127SHong Zhang {
1192a6ece127SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
11935fd66863SKarl Rupp 
1194a6ece127SHong Zhang   PetscFunctionBegin;
1195a6ece127SHong Zhang   *array = a->a;
1196a6ece127SHong Zhang   PetscFunctionReturn(0);
1197a6ece127SHong Zhang }
1198a6ece127SHong Zhang 
1199a6ece127SHong Zhang #undef __FUNCT__
12008c778c55SBarry Smith #define __FUNCT__ "MatSeqSBAIJRestoreArray_SeqSBAIJ"
12018c778c55SBarry Smith PetscErrorCode MatSeqSBAIJRestoreArray_SeqSBAIJ(Mat A,PetscScalar *array[])
1202a6ece127SHong Zhang {
1203a6ece127SHong Zhang   PetscFunctionBegin;
1204a6ece127SHong Zhang   PetscFunctionReturn(0);
1205a6ece127SHong Zhang }
1206a6ece127SHong Zhang 
120742ee4b1aSHong Zhang #undef __FUNCT__
120842ee4b1aSHong Zhang #define __FUNCT__ "MatAXPY_SeqSBAIJ"
1209f4df32b1SMatthew Knepley PetscErrorCode MatAXPY_SeqSBAIJ(Mat Y,PetscScalar a,Mat X,MatStructure str)
121042ee4b1aSHong Zhang {
121142ee4b1aSHong Zhang   Mat_SeqSBAIJ   *x=(Mat_SeqSBAIJ*)X->data, *y=(Mat_SeqSBAIJ*)Y->data;
1212dfbe8321SBarry Smith   PetscErrorCode ierr;
1213e838b9e7SJed Brown   PetscInt       i,bs=Y->rmap->bs,bs2=bs*bs,j;
1214e838b9e7SJed Brown   PetscBLASInt   one = 1;
121542ee4b1aSHong Zhang 
121642ee4b1aSHong Zhang   PetscFunctionBegin;
121742ee4b1aSHong Zhang   if (str == SAME_NONZERO_PATTERN) {
1218f4df32b1SMatthew Knepley     PetscScalar  alpha = a;
1219c5df96a5SBarry Smith     PetscBLASInt bnz;
1220c5df96a5SBarry Smith     ierr = PetscBLASIntCast(x->nz*bs2,&bnz);CHKERRQ(ierr);
12218b83055fSJed Brown     PetscStackCallBLAS("BLASaxpy",BLASaxpy_(&bnz,&alpha,x->a,&one,y->a,&one));
1222a3fa217bSJose E. Roman     ierr = PetscObjectStateIncrease((PetscObject)Y);CHKERRQ(ierr);
1223c537a176SHong Zhang   } else if (str == SUBSET_NONZERO_PATTERN) { /* nonzeros of X is a subset of Y's */
1224c4319e64SHong Zhang     if (y->xtoy && y->XtoY != X) {
1225c4319e64SHong Zhang       ierr = PetscFree(y->xtoy);CHKERRQ(ierr);
12266bf464f9SBarry Smith       ierr = MatDestroy(&y->XtoY);CHKERRQ(ierr);
1227c537a176SHong Zhang     }
1228c4319e64SHong Zhang     if (!y->xtoy) { /* get xtoy */
12290298fd71SBarry Smith       ierr    = MatAXPYGetxtoy_Private(x->mbs,x->i,x->j,NULL, y->i,y->j,NULL, &y->xtoy);CHKERRQ(ierr);
1230c4319e64SHong Zhang       y->XtoY = X;
1231c537a176SHong Zhang     }
12326c6c5352SBarry Smith     for (i=0; i<x->nz; i++) {
1233c4319e64SHong Zhang       j = 0;
1234c4319e64SHong Zhang       while (j < bs2) {
1235f4df32b1SMatthew Knepley         y->a[bs2*y->xtoy[i]+j] += a*(x->a[bs2*i+j]);
1236c4319e64SHong Zhang         j++;
1237c537a176SHong Zhang       }
1238c4319e64SHong Zhang     }
1239a3fa217bSJose E. Roman     ierr = PetscObjectStateIncrease((PetscObject)Y);CHKERRQ(ierr);
12406712e2f1SBarry Smith     ierr = PetscInfo3(Y,"ratio of nnz_s(X)/nnz_s(Y): %D/%D = %g\n",bs2*x->nz,bs2*y->nz,(double)((PetscReal)(bs2*x->nz)/(PetscReal)(bs2*y->nz)));CHKERRQ(ierr);
124142ee4b1aSHong Zhang   } else {
1242f5edf698SHong Zhang     ierr = MatGetRowUpperTriangular(X);CHKERRQ(ierr);
1243f4df32b1SMatthew Knepley     ierr = MatAXPY_Basic(Y,a,X,str);CHKERRQ(ierr);
1244f5edf698SHong Zhang     ierr = MatRestoreRowUpperTriangular(X);CHKERRQ(ierr);
124542ee4b1aSHong Zhang   }
124642ee4b1aSHong Zhang   PetscFunctionReturn(0);
124742ee4b1aSHong Zhang }
124842ee4b1aSHong Zhang 
1249efcf0fc3SBarry Smith #undef __FUNCT__
1250efcf0fc3SBarry Smith #define __FUNCT__ "MatIsSymmetric_SeqSBAIJ"
1251ace3abfcSBarry Smith PetscErrorCode MatIsSymmetric_SeqSBAIJ(Mat A,PetscReal tol,PetscBool  *flg)
1252efcf0fc3SBarry Smith {
1253efcf0fc3SBarry Smith   PetscFunctionBegin;
1254efcf0fc3SBarry Smith   *flg = PETSC_TRUE;
1255efcf0fc3SBarry Smith   PetscFunctionReturn(0);
1256efcf0fc3SBarry Smith }
1257efcf0fc3SBarry Smith 
1258efcf0fc3SBarry Smith #undef __FUNCT__
1259efcf0fc3SBarry Smith #define __FUNCT__ "MatIsStructurallySymmetric_SeqSBAIJ"
1260ace3abfcSBarry Smith PetscErrorCode MatIsStructurallySymmetric_SeqSBAIJ(Mat A,PetscBool  *flg)
1261efcf0fc3SBarry Smith {
1262efcf0fc3SBarry Smith   PetscFunctionBegin;
1263efcf0fc3SBarry Smith   *flg = PETSC_TRUE;
1264efcf0fc3SBarry Smith   PetscFunctionReturn(0);
1265efcf0fc3SBarry Smith }
1266efcf0fc3SBarry Smith 
1267efcf0fc3SBarry Smith #undef __FUNCT__
1268efcf0fc3SBarry Smith #define __FUNCT__ "MatIsHermitian_SeqSBAIJ"
1269ace3abfcSBarry Smith PetscErrorCode MatIsHermitian_SeqSBAIJ(Mat A,PetscReal tol,PetscBool  *flg)
1270efcf0fc3SBarry Smith {
1271efcf0fc3SBarry Smith   PetscFunctionBegin;
1272efcf0fc3SBarry Smith   *flg = PETSC_FALSE;
1273efcf0fc3SBarry Smith   PetscFunctionReturn(0);
1274efcf0fc3SBarry Smith }
1275efcf0fc3SBarry Smith 
127699cafbc1SBarry Smith #undef __FUNCT__
127799cafbc1SBarry Smith #define __FUNCT__ "MatRealPart_SeqSBAIJ"
127899cafbc1SBarry Smith PetscErrorCode MatRealPart_SeqSBAIJ(Mat A)
127999cafbc1SBarry Smith {
128099cafbc1SBarry Smith   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
128199cafbc1SBarry Smith   PetscInt     i,nz = a->bs2*a->i[a->mbs];
1282dd6ea824SBarry Smith   MatScalar    *aa = a->a;
128399cafbc1SBarry Smith 
128499cafbc1SBarry Smith   PetscFunctionBegin;
128599cafbc1SBarry Smith   for (i=0; i<nz; i++) aa[i] = PetscRealPart(aa[i]);
128699cafbc1SBarry Smith   PetscFunctionReturn(0);
128799cafbc1SBarry Smith }
128899cafbc1SBarry Smith 
128999cafbc1SBarry Smith #undef __FUNCT__
129099cafbc1SBarry Smith #define __FUNCT__ "MatImaginaryPart_SeqSBAIJ"
129199cafbc1SBarry Smith PetscErrorCode MatImaginaryPart_SeqSBAIJ(Mat A)
129299cafbc1SBarry Smith {
129399cafbc1SBarry Smith   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
129499cafbc1SBarry Smith   PetscInt     i,nz = a->bs2*a->i[a->mbs];
1295dd6ea824SBarry Smith   MatScalar    *aa = a->a;
129699cafbc1SBarry Smith 
129799cafbc1SBarry Smith   PetscFunctionBegin;
129899cafbc1SBarry Smith   for (i=0; i<nz; i++) aa[i] = PetscImaginaryPart(aa[i]);
129999cafbc1SBarry Smith   PetscFunctionReturn(0);
130099cafbc1SBarry Smith }
130199cafbc1SBarry Smith 
13023bededecSBarry Smith #undef __FUNCT__
13033bededecSBarry Smith #define __FUNCT__ "MatZeroRowsColumns_SeqSBAIJ"
13043bededecSBarry Smith PetscErrorCode MatZeroRowsColumns_SeqSBAIJ(Mat A,PetscInt is_n,const PetscInt is_idx[],PetscScalar diag,Vec x, Vec b)
13053bededecSBarry Smith {
13063bededecSBarry Smith   Mat_SeqSBAIJ      *baij=(Mat_SeqSBAIJ*)A->data;
13073bededecSBarry Smith   PetscErrorCode    ierr;
13083bededecSBarry Smith   PetscInt          i,j,k,count;
13093bededecSBarry Smith   PetscInt          bs   =A->rmap->bs,bs2=baij->bs2,row,col;
13103bededecSBarry Smith   PetscScalar       zero = 0.0;
13113bededecSBarry Smith   MatScalar         *aa;
13123bededecSBarry Smith   const PetscScalar *xx;
13133bededecSBarry Smith   PetscScalar       *bb;
131456777dd2SBarry Smith   PetscBool         *zeroed,vecs = PETSC_FALSE;
13153bededecSBarry Smith 
13163bededecSBarry Smith   PetscFunctionBegin;
13173bededecSBarry Smith   /* fix right hand side if needed */
13183bededecSBarry Smith   if (x && b) {
13193bededecSBarry Smith     ierr = VecGetArrayRead(x,&xx);CHKERRQ(ierr);
13203bededecSBarry Smith     ierr = VecGetArray(b,&bb);CHKERRQ(ierr);
132156777dd2SBarry Smith     vecs = PETSC_TRUE;
13223bededecSBarry Smith   }
13233bededecSBarry Smith 
13243bededecSBarry Smith   /* zero the columns */
13251795a4d1SJed Brown   ierr = PetscCalloc1(A->rmap->n,&zeroed);CHKERRQ(ierr);
13263bededecSBarry Smith   for (i=0; i<is_n; i++) {
13273bededecSBarry 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]);
13283bededecSBarry Smith     zeroed[is_idx[i]] = PETSC_TRUE;
13293bededecSBarry Smith   }
133056777dd2SBarry Smith   if (vecs) {
133156777dd2SBarry Smith     for (i=0; i<A->rmap->N; i++) {
133256777dd2SBarry Smith       row = i/bs;
133356777dd2SBarry Smith       for (j=baij->i[row]; j<baij->i[row+1]; j++) {
133456777dd2SBarry Smith         for (k=0; k<bs; k++) {
133556777dd2SBarry Smith           col = bs*baij->j[j] + k;
133656777dd2SBarry Smith           if (col <= i) continue;
133756777dd2SBarry Smith           aa = ((MatScalar*)(baij->a)) + j*bs2 + (i%bs) + bs*k;
133826fbe8dcSKarl Rupp           if (!zeroed[i] && zeroed[col]) bb[i]   -= aa[0]*xx[col];
133926fbe8dcSKarl Rupp           if (zeroed[i] && !zeroed[col]) bb[col] -= aa[0]*xx[i];
134056777dd2SBarry Smith         }
134156777dd2SBarry Smith       }
134256777dd2SBarry Smith     }
134326fbe8dcSKarl Rupp     for (i=0; i<is_n; i++) bb[is_idx[i]] = diag*xx[is_idx[i]];
134456777dd2SBarry Smith   }
134556777dd2SBarry Smith 
13463bededecSBarry Smith   for (i=0; i<A->rmap->N; i++) {
13473bededecSBarry Smith     if (!zeroed[i]) {
13483bededecSBarry Smith       row = i/bs;
13493bededecSBarry Smith       for (j=baij->i[row]; j<baij->i[row+1]; j++) {
13503bededecSBarry Smith         for (k=0; k<bs; k++) {
13513bededecSBarry Smith           col = bs*baij->j[j] + k;
13523bededecSBarry Smith           if (zeroed[col]) {
13533bededecSBarry Smith             aa = ((MatScalar*)(baij->a)) + j*bs2 + (i%bs) + bs*k;
13543bededecSBarry Smith             aa[0] = 0.0;
13553bededecSBarry Smith           }
13563bededecSBarry Smith         }
13573bededecSBarry Smith       }
13583bededecSBarry Smith     }
13593bededecSBarry Smith   }
13603bededecSBarry Smith   ierr = PetscFree(zeroed);CHKERRQ(ierr);
136156777dd2SBarry Smith   if (vecs) {
136256777dd2SBarry Smith     ierr = VecRestoreArrayRead(x,&xx);CHKERRQ(ierr);
136356777dd2SBarry Smith     ierr = VecRestoreArray(b,&bb);CHKERRQ(ierr);
136456777dd2SBarry Smith   }
13653bededecSBarry Smith 
13663bededecSBarry Smith   /* zero the rows */
13673bededecSBarry Smith   for (i=0; i<is_n; i++) {
13683bededecSBarry Smith     row   = is_idx[i];
13693bededecSBarry Smith     count = (baij->i[row/bs +1] - baij->i[row/bs])*bs;
13703bededecSBarry Smith     aa    = ((MatScalar*)(baij->a)) + baij->i[row/bs]*bs2 + (row%bs);
13713bededecSBarry Smith     for (k=0; k<count; k++) {
13723bededecSBarry Smith       aa[0] =  zero;
13733bededecSBarry Smith       aa   += bs;
13743bededecSBarry Smith     }
13753bededecSBarry Smith     if (diag != 0.0) {
13763bededecSBarry Smith       ierr = (*A->ops->setvalues)(A,1,&row,1,&row,&diag,INSERT_VALUES);CHKERRQ(ierr);
13773bededecSBarry Smith     }
13783bededecSBarry Smith   }
13793bededecSBarry Smith   ierr = MatAssemblyEnd_SeqSBAIJ(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
13803bededecSBarry Smith   PetscFunctionReturn(0);
13813bededecSBarry Smith }
13823bededecSBarry Smith 
138349b5e25fSSatish Balay /* -------------------------------------------------------------------*/
13843964eb88SJed Brown static struct _MatOps MatOps_Values = {MatSetValues_SeqSBAIJ,
138549b5e25fSSatish Balay                                        MatGetRow_SeqSBAIJ,
138649b5e25fSSatish Balay                                        MatRestoreRow_SeqSBAIJ,
138749b5e25fSSatish Balay                                        MatMult_SeqSBAIJ_N,
138897304618SKris Buschelman                                /*  4*/ MatMultAdd_SeqSBAIJ_N,
1389431c96f7SBarry Smith                                        MatMult_SeqSBAIJ_N,       /* transpose versions are same as non-transpose versions */
1390e005ede5SBarry Smith                                        MatMultAdd_SeqSBAIJ_N,
1391db4efbfdSBarry Smith                                        0,
139249b5e25fSSatish Balay                                        0,
139349b5e25fSSatish Balay                                        0,
139497304618SKris Buschelman                                /* 10*/ 0,
139549b5e25fSSatish Balay                                        0,
1396c078aec8SLisandro Dalcin                                        MatCholeskyFactor_SeqSBAIJ,
139741f059aeSBarry Smith                                        MatSOR_SeqSBAIJ,
139849b5e25fSSatish Balay                                        MatTranspose_SeqSBAIJ,
139997304618SKris Buschelman                                /* 15*/ MatGetInfo_SeqSBAIJ,
140049b5e25fSSatish Balay                                        MatEqual_SeqSBAIJ,
140149b5e25fSSatish Balay                                        MatGetDiagonal_SeqSBAIJ,
140249b5e25fSSatish Balay                                        MatDiagonalScale_SeqSBAIJ,
140349b5e25fSSatish Balay                                        MatNorm_SeqSBAIJ,
140497304618SKris Buschelman                                /* 20*/ 0,
140549b5e25fSSatish Balay                                        MatAssemblyEnd_SeqSBAIJ,
140649b5e25fSSatish Balay                                        MatSetOption_SeqSBAIJ,
140749b5e25fSSatish Balay                                        MatZeroEntries_SeqSBAIJ,
1408d519adbfSMatthew Knepley                                /* 24*/ 0,
140949b5e25fSSatish Balay                                        0,
141049b5e25fSSatish Balay                                        0,
1411db4efbfdSBarry Smith                                        0,
1412db4efbfdSBarry Smith                                        0,
14134994cf47SJed Brown                                /* 29*/ MatSetUp_SeqSBAIJ,
1414c464158bSHong Zhang                                        0,
1415db4efbfdSBarry Smith                                        0,
14168c778c55SBarry Smith                                        0,
14178c778c55SBarry Smith                                        0,
1418d519adbfSMatthew Knepley                                /* 34*/ MatDuplicate_SeqSBAIJ,
1419719d5645SBarry Smith                                        0,
1420719d5645SBarry Smith                                        0,
142149b5e25fSSatish Balay                                        0,
1422c84f5b01SHong Zhang                                        MatICCFactor_SeqSBAIJ,
1423d519adbfSMatthew Knepley                                /* 39*/ MatAXPY_SeqSBAIJ,
142449b5e25fSSatish Balay                                        MatGetSubMatrices_SeqSBAIJ,
142549b5e25fSSatish Balay                                        MatIncreaseOverlap_SeqSBAIJ,
142649b5e25fSSatish Balay                                        MatGetValues_SeqSBAIJ,
14273c896bc6SHong Zhang                                        MatCopy_SeqSBAIJ,
1428d519adbfSMatthew Knepley                                /* 44*/ 0,
142949b5e25fSSatish Balay                                        MatScale_SeqSBAIJ,
143049b5e25fSSatish Balay                                        0,
143149b5e25fSSatish Balay                                        0,
14323bededecSBarry Smith                                        MatZeroRowsColumns_SeqSBAIJ,
1433f73d5cc4SBarry Smith                                /* 49*/ 0,
143449b5e25fSSatish Balay                                        MatGetRowIJ_SeqSBAIJ,
143549b5e25fSSatish Balay                                        MatRestoreRowIJ_SeqSBAIJ,
143649b5e25fSSatish Balay                                        0,
143749b5e25fSSatish Balay                                        0,
1438d519adbfSMatthew Knepley                                /* 54*/ 0,
143949b5e25fSSatish Balay                                        0,
144049b5e25fSSatish Balay                                        0,
144149b5e25fSSatish Balay                                        0,
144249b5e25fSSatish Balay                                        MatSetValuesBlocked_SeqSBAIJ,
1443d519adbfSMatthew Knepley                                /* 59*/ MatGetSubMatrix_SeqSBAIJ,
144449b5e25fSSatish Balay                                        0,
144549b5e25fSSatish Balay                                        0,
1446357abbc8SBarry Smith                                        0,
1447d959ec07SHong Zhang                                        0,
1448d519adbfSMatthew Knepley                                /* 64*/ 0,
1449d959ec07SHong Zhang                                        0,
1450d959ec07SHong Zhang                                        0,
1451d959ec07SHong Zhang                                        0,
1452d959ec07SHong Zhang                                        0,
1453d519adbfSMatthew Knepley                                /* 69*/ MatGetRowMaxAbs_SeqSBAIJ,
14543e0d88b5SBarry Smith                                        0,
14553e0d88b5SBarry Smith                                        0,
14563e0d88b5SBarry Smith                                        0,
14573e0d88b5SBarry Smith                                        0,
1458d519adbfSMatthew Knepley                                /* 74*/ 0,
14593e0d88b5SBarry Smith                                        0,
14603e0d88b5SBarry Smith                                        0,
14613e0d88b5SBarry Smith                                        0,
14623e0d88b5SBarry Smith                                        0,
1463d519adbfSMatthew Knepley                                /* 79*/ 0,
14643e0d88b5SBarry Smith                                        0,
14653e0d88b5SBarry Smith                                        0,
146697304618SKris Buschelman                                        MatGetInertia_SeqSBAIJ,
14675bba2384SShri Abhyankar                                        MatLoad_SeqSBAIJ,
1468d519adbfSMatthew Knepley                                /* 84*/ MatIsSymmetric_SeqSBAIJ,
1469865e5f61SKris Buschelman                                        MatIsHermitian_SeqSBAIJ,
1470efcf0fc3SBarry Smith                                        MatIsStructurallySymmetric_SeqSBAIJ,
1471865e5f61SKris Buschelman                                        0,
1472865e5f61SKris Buschelman                                        0,
1473d519adbfSMatthew Knepley                                /* 89*/ 0,
1474865e5f61SKris Buschelman                                        0,
1475865e5f61SKris Buschelman                                        0,
1476865e5f61SKris Buschelman                                        0,
1477865e5f61SKris Buschelman                                        0,
1478d519adbfSMatthew Knepley                                /* 94*/ 0,
1479865e5f61SKris Buschelman                                        0,
1480865e5f61SKris Buschelman                                        0,
148199cafbc1SBarry Smith                                        0,
148299cafbc1SBarry Smith                                        0,
1483d519adbfSMatthew Knepley                                /* 99*/ 0,
148499cafbc1SBarry Smith                                        0,
148599cafbc1SBarry Smith                                        0,
148699cafbc1SBarry Smith                                        0,
148799cafbc1SBarry Smith                                        0,
1488d519adbfSMatthew Knepley                                /*104*/ 0,
148999cafbc1SBarry Smith                                        MatRealPart_SeqSBAIJ,
1490f5edf698SHong Zhang                                        MatImaginaryPart_SeqSBAIJ,
1491f5edf698SHong Zhang                                        MatGetRowUpperTriangular_SeqSBAIJ,
14922af78befSBarry Smith                                        MatRestoreRowUpperTriangular_SeqSBAIJ,
1493d519adbfSMatthew Knepley                                /*109*/ 0,
14942af78befSBarry Smith                                        0,
14952af78befSBarry Smith                                        0,
14962af78befSBarry Smith                                        0,
1497547795f9SHong Zhang                                        MatMissingDiagonal_SeqSBAIJ,
1498547795f9SHong Zhang                                /*114*/ 0,
1499547795f9SHong Zhang                                        0,
1500547795f9SHong Zhang                                        0,
1501547795f9SHong Zhang                                        0,
1502547795f9SHong Zhang                                        0,
1503547795f9SHong Zhang                                /*119*/ 0,
1504547795f9SHong Zhang                                        0,
15052f480046SShri Abhyankar                                        0,
15063964eb88SJed Brown                                        0,
15073964eb88SJed Brown                                        0,
15083964eb88SJed Brown                                /*124*/ 0,
15093964eb88SJed Brown                                        0,
15103964eb88SJed Brown                                        0,
15113964eb88SJed Brown                                        0,
15123964eb88SJed Brown                                        0,
15133964eb88SJed Brown                                /*129*/ 0,
15143964eb88SJed Brown                                        0,
15153964eb88SJed Brown                                        0,
15163964eb88SJed Brown                                        0,
15173964eb88SJed Brown                                        0,
15183964eb88SJed Brown                                /*134*/ 0,
15193964eb88SJed Brown                                        0,
15203964eb88SJed Brown                                        0,
15213964eb88SJed Brown                                        0,
15223964eb88SJed Brown                                        0,
15233964eb88SJed Brown                                /*139*/ 0,
1524f9426fe0SMark Adams                                        0,
15255bba2384SShri Abhyankar                                        0
152699cafbc1SBarry Smith };
1527be1d678aSKris Buschelman 
15284a2ae208SSatish Balay #undef __FUNCT__
15294a2ae208SSatish Balay #define __FUNCT__ "MatStoreValues_SeqSBAIJ"
15307087cfbeSBarry Smith PetscErrorCode  MatStoreValues_SeqSBAIJ(Mat mat)
153149b5e25fSSatish Balay {
15324afc71dfSHong Zhang   Mat_SeqSBAIJ   *aij = (Mat_SeqSBAIJ*)mat->data;
1533d0f46423SBarry Smith   PetscInt       nz   = aij->i[mat->rmap->N]*mat->rmap->bs*aij->bs2;
1534dfbe8321SBarry Smith   PetscErrorCode ierr;
153549b5e25fSSatish Balay 
153649b5e25fSSatish Balay   PetscFunctionBegin;
1537e7e72b3dSBarry Smith   if (aij->nonew != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first");
153849b5e25fSSatish Balay 
153949b5e25fSSatish Balay   /* allocate space for values if not already there */
154049b5e25fSSatish Balay   if (!aij->saved_values) {
1541785e854fSJed Brown     ierr = PetscMalloc1((nz+1),&aij->saved_values);CHKERRQ(ierr);
154249b5e25fSSatish Balay   }
154349b5e25fSSatish Balay 
154449b5e25fSSatish Balay   /* copy values over */
154587828ca2SBarry Smith   ierr = PetscMemcpy(aij->saved_values,aij->a,nz*sizeof(PetscScalar));CHKERRQ(ierr);
154649b5e25fSSatish Balay   PetscFunctionReturn(0);
154749b5e25fSSatish Balay }
154849b5e25fSSatish Balay 
15494a2ae208SSatish Balay #undef __FUNCT__
15504a2ae208SSatish Balay #define __FUNCT__ "MatRetrieveValues_SeqSBAIJ"
15517087cfbeSBarry Smith PetscErrorCode  MatRetrieveValues_SeqSBAIJ(Mat mat)
155249b5e25fSSatish Balay {
15534afc71dfSHong Zhang   Mat_SeqSBAIJ   *aij = (Mat_SeqSBAIJ*)mat->data;
15546849ba73SBarry Smith   PetscErrorCode ierr;
1555d0f46423SBarry Smith   PetscInt       nz = aij->i[mat->rmap->N]*mat->rmap->bs*aij->bs2;
155649b5e25fSSatish Balay 
155749b5e25fSSatish Balay   PetscFunctionBegin;
1558e7e72b3dSBarry Smith   if (aij->nonew != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first");
1559e7e72b3dSBarry Smith   if (!aij->saved_values) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatStoreValues(A);first");
156049b5e25fSSatish Balay 
156149b5e25fSSatish Balay   /* copy values over */
156287828ca2SBarry Smith   ierr = PetscMemcpy(aij->a,aij->saved_values,nz*sizeof(PetscScalar));CHKERRQ(ierr);
156349b5e25fSSatish Balay   PetscFunctionReturn(0);
156449b5e25fSSatish Balay }
156549b5e25fSSatish Balay 
15664a2ae208SSatish Balay #undef __FUNCT__
1567a23d5eceSKris Buschelman #define __FUNCT__ "MatSeqSBAIJSetPreallocation_SeqSBAIJ"
15687087cfbeSBarry Smith PetscErrorCode  MatSeqSBAIJSetPreallocation_SeqSBAIJ(Mat B,PetscInt bs,PetscInt nz,PetscInt *nnz)
156949b5e25fSSatish Balay {
1570c464158bSHong Zhang   Mat_SeqSBAIJ   *b = (Mat_SeqSBAIJ*)B->data;
15716849ba73SBarry Smith   PetscErrorCode ierr;
1572535b19f3SBarry Smith   PetscInt       i,mbs,bs2;
15732576faa2SJed Brown   PetscBool      skipallocation = PETSC_FALSE,flg = PETSC_FALSE,realalloc = PETSC_FALSE;
157449b5e25fSSatish Balay 
157549b5e25fSSatish Balay   PetscFunctionBegin;
15762576faa2SJed Brown   if (nz >= 0 || nnz) realalloc = PETSC_TRUE;
1577273d9f13SBarry Smith   B->preallocated = PETSC_TRUE;
1578db4efbfdSBarry Smith 
157933d57670SJed Brown   ierr = MatSetBlockSize(B,PetscAbs(bs));CHKERRQ(ierr);
158026283091SBarry Smith   ierr = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr);
158126283091SBarry Smith   ierr = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr);
1582e02043d6SBarry Smith   ierr = PetscLayoutGetBlockSize(B->rmap,&bs);CHKERRQ(ierr);
1583899cda47SBarry Smith 
1584d0f46423SBarry Smith   mbs = B->rmap->N/bs;
158549b5e25fSSatish Balay   bs2 = bs*bs;
158649b5e25fSSatish Balay 
1587e7e72b3dSBarry Smith   if (mbs*bs != B->rmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Number rows, cols must be divisible by blocksize");
158849b5e25fSSatish Balay 
1589ab93d7beSBarry Smith   if (nz == MAT_SKIP_ALLOCATION) {
1590ab93d7beSBarry Smith     skipallocation = PETSC_TRUE;
1591ab93d7beSBarry Smith     nz             = 0;
1592ab93d7beSBarry Smith   }
1593ab93d7beSBarry Smith 
1594435da068SBarry Smith   if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 3;
1595e32f2f54SBarry Smith   if (nz < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"nz cannot be less than 0: value %D",nz);
159649b5e25fSSatish Balay   if (nnz) {
159749b5e25fSSatish Balay     for (i=0; i<mbs; i++) {
1598e32f2f54SBarry 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]);
1599e32f2f54SBarry Smith       if (nnz[i] > mbs) SETERRQ3(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"nnz cannot be greater than block row length: local row %D value %D rowlength %D",i,nnz[i],mbs);
160049b5e25fSSatish Balay     }
160149b5e25fSSatish Balay   }
160249b5e25fSSatish Balay 
1603db4efbfdSBarry Smith   B->ops->mult             = MatMult_SeqSBAIJ_N;
1604db4efbfdSBarry Smith   B->ops->multadd          = MatMultAdd_SeqSBAIJ_N;
1605db4efbfdSBarry Smith   B->ops->multtranspose    = MatMult_SeqSBAIJ_N;
1606db4efbfdSBarry Smith   B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_N;
160726fbe8dcSKarl Rupp 
16080298fd71SBarry Smith   ierr  = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_no_unroll",&flg,NULL);CHKERRQ(ierr);
160949b5e25fSSatish Balay   if (!flg) {
161049b5e25fSSatish Balay     switch (bs) {
161149b5e25fSSatish Balay     case 1:
161249b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_1;
161349b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_1;
1614431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_1;
1615431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_1;
161649b5e25fSSatish Balay       break;
161749b5e25fSSatish Balay     case 2:
161849b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_2;
161949b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_2;
1620431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_2;
1621431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_2;
162249b5e25fSSatish Balay       break;
162349b5e25fSSatish Balay     case 3:
162449b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_3;
162549b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_3;
1626431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_3;
1627431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_3;
162849b5e25fSSatish Balay       break;
162949b5e25fSSatish Balay     case 4:
163049b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_4;
163149b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_4;
1632431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_4;
1633431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_4;
163449b5e25fSSatish Balay       break;
163549b5e25fSSatish Balay     case 5:
163649b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_5;
163749b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_5;
1638431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_5;
1639431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_5;
164049b5e25fSSatish Balay       break;
164149b5e25fSSatish Balay     case 6:
164249b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_6;
164349b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_6;
1644431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_6;
1645431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_6;
164649b5e25fSSatish Balay       break;
164749b5e25fSSatish Balay     case 7:
1648de53e5efSHong Zhang       B->ops->mult             = MatMult_SeqSBAIJ_7;
164949b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_7;
1650431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_7;
1651431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_7;
165249b5e25fSSatish Balay       break;
165349b5e25fSSatish Balay     }
165449b5e25fSSatish Balay   }
165549b5e25fSSatish Balay 
165649b5e25fSSatish Balay   b->mbs = mbs;
16574afc71dfSHong Zhang   b->nbs = mbs;
1658ab93d7beSBarry Smith   if (!skipallocation) {
16592ee49352SLisandro Dalcin     if (!b->imax) {
1660dcca6d9dSJed Brown       ierr = PetscMalloc2(mbs,&b->imax,mbs,&b->ilen);CHKERRQ(ierr);
166126fbe8dcSKarl Rupp 
1662c760cd28SBarry Smith       b->free_imax_ilen = PETSC_TRUE;
166326fbe8dcSKarl Rupp 
16643bb1ff40SBarry Smith       ierr = PetscLogObjectMemory((PetscObject)B,2*mbs*sizeof(PetscInt));CHKERRQ(ierr);
16652ee49352SLisandro Dalcin     }
166649b5e25fSSatish Balay     if (!nnz) {
1667435da068SBarry Smith       if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 5;
166849b5e25fSSatish Balay       else if (nz <= 0) nz = 1;
166926fbe8dcSKarl Rupp       for (i=0; i<mbs; i++) b->imax[i] = nz;
1670153ea458SHong Zhang       nz = nz*mbs; /* total nz */
167149b5e25fSSatish Balay     } else {
167249b5e25fSSatish Balay       nz = 0;
16738cef66ccSHong Zhang       for (i=0; i<mbs; i++) {b->imax[i] = nnz[i]; nz += nnz[i];}
167449b5e25fSSatish Balay     }
16752ee49352SLisandro Dalcin     /* b->ilen will count nonzeros in each block row so far. */
167626fbe8dcSKarl Rupp     for (i=0; i<mbs; i++) b->ilen[i] = 0;
16776c6c5352SBarry Smith     /* nz=(nz+mbs)/2; */ /* total diagonal and superdiagonal nonzero blocks */
167849b5e25fSSatish Balay 
167949b5e25fSSatish Balay     /* allocate the matrix space */
16802ee49352SLisandro Dalcin     ierr = MatSeqXAIJFreeAIJ(B,&b->a,&b->j,&b->i);CHKERRQ(ierr);
1681dcca6d9dSJed Brown     ierr = PetscMalloc3(bs2*nz,&b->a,nz,&b->j,B->rmap->N+1,&b->i);CHKERRQ(ierr);
16823bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)B,(B->rmap->N+1)*sizeof(PetscInt)+nz*(bs2*sizeof(PetscScalar)+sizeof(PetscInt)));CHKERRQ(ierr);
16836c6c5352SBarry Smith     ierr = PetscMemzero(b->a,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr);
168413f74950SBarry Smith     ierr = PetscMemzero(b->j,nz*sizeof(PetscInt));CHKERRQ(ierr);
168526fbe8dcSKarl Rupp 
168649b5e25fSSatish Balay     b->singlemalloc = PETSC_TRUE;
168749b5e25fSSatish Balay 
168849b5e25fSSatish Balay     /* pointer to beginning of each row */
1689e60cf9a0SBarry Smith     b->i[0] = 0;
169026fbe8dcSKarl Rupp     for (i=1; i<mbs+1; i++) b->i[i] = b->i[i-1] + b->imax[i-1];
169126fbe8dcSKarl Rupp 
1692e6b907acSBarry Smith     b->free_a  = PETSC_TRUE;
1693e6b907acSBarry Smith     b->free_ij = PETSC_TRUE;
1694e811da20SHong Zhang   } else {
1695e6b907acSBarry Smith     b->free_a  = PETSC_FALSE;
1696e6b907acSBarry Smith     b->free_ij = PETSC_FALSE;
1697ab93d7beSBarry Smith   }
169849b5e25fSSatish Balay 
1699d0f46423SBarry Smith   B->rmap->bs = bs;
170049b5e25fSSatish Balay   b->bs2      = bs2;
17016c6c5352SBarry Smith   b->nz       = 0;
1702b32cb4a7SJed Brown   b->maxnz    = nz;
1703153ea458SHong Zhang 
170416cdd363SHong Zhang   b->inew    = 0;
170516cdd363SHong Zhang   b->jnew    = 0;
170616cdd363SHong Zhang   b->anew    = 0;
170716cdd363SHong Zhang   b->a2anew  = 0;
17081a3463dfSHong Zhang   b->permute = PETSC_FALSE;
17092576faa2SJed Brown   if (realalloc) {ierr = MatSetOption(B,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);}
1710c464158bSHong Zhang   PetscFunctionReturn(0);
1711c464158bSHong Zhang }
1712153ea458SHong Zhang 
171338f409ebSLisandro Dalcin #undef __FUNCT__
171438f409ebSLisandro Dalcin #define __FUNCT__ "MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ"
171538f409ebSLisandro Dalcin PetscErrorCode MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ(Mat B,PetscInt bs,const PetscInt ii[],const PetscInt jj[], const PetscScalar V[])
171638f409ebSLisandro Dalcin {
171738f409ebSLisandro Dalcin   PetscInt       i,j,m,nz,nz_max=0,*nnz;
171838f409ebSLisandro Dalcin   PetscScalar    *values=0;
171938f409ebSLisandro Dalcin   PetscBool      roworiented = ((Mat_SeqSBAIJ*)B->data)->roworiented;
172038f409ebSLisandro Dalcin   PetscErrorCode ierr;
172138f409ebSLisandro Dalcin   PetscFunctionBegin;
172238f409ebSLisandro Dalcin   if (bs < 1) SETERRQ1(PetscObjectComm((PetscObject)B),PETSC_ERR_ARG_OUTOFRANGE,"Invalid block size specified, must be positive but it is %D",bs);
172338f409ebSLisandro Dalcin   ierr   = PetscLayoutSetBlockSize(B->rmap,bs);CHKERRQ(ierr);
172438f409ebSLisandro Dalcin   ierr   = PetscLayoutSetBlockSize(B->cmap,bs);CHKERRQ(ierr);
172538f409ebSLisandro Dalcin   ierr   = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr);
172638f409ebSLisandro Dalcin   ierr   = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr);
172738f409ebSLisandro Dalcin   ierr   = PetscLayoutGetBlockSize(B->rmap,&bs);CHKERRQ(ierr);
172838f409ebSLisandro Dalcin   m      = B->rmap->n/bs;
172938f409ebSLisandro Dalcin 
173038f409ebSLisandro Dalcin   if (ii[0]) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"ii[0] must be 0 but it is %D",ii[0]);
1731785e854fSJed Brown   ierr = PetscMalloc1((m+1),&nnz);CHKERRQ(ierr);
173238f409ebSLisandro Dalcin   for (i=0; i<m; i++) {
173338f409ebSLisandro Dalcin     nz = ii[i+1] - ii[i];
173438f409ebSLisandro Dalcin     if (nz < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row %D has a negative number of columns %D",i,nz);
173538f409ebSLisandro Dalcin     nz_max = PetscMax(nz_max,nz);
173638f409ebSLisandro Dalcin     nnz[i] = nz;
173738f409ebSLisandro Dalcin   }
173838f409ebSLisandro Dalcin   ierr = MatSeqSBAIJSetPreallocation(B,bs,0,nnz);CHKERRQ(ierr);
173938f409ebSLisandro Dalcin   ierr = PetscFree(nnz);CHKERRQ(ierr);
174038f409ebSLisandro Dalcin 
174138f409ebSLisandro Dalcin   values = (PetscScalar*)V;
174238f409ebSLisandro Dalcin   if (!values) {
17431795a4d1SJed Brown     ierr = PetscCalloc1(bs*bs*nz_max,&values);CHKERRQ(ierr);
174438f409ebSLisandro Dalcin   }
174538f409ebSLisandro Dalcin   for (i=0; i<m; i++) {
174638f409ebSLisandro Dalcin     PetscInt          ncols  = ii[i+1] - ii[i];
174738f409ebSLisandro Dalcin     const PetscInt    *icols = jj + ii[i];
174838f409ebSLisandro Dalcin     if (!roworiented || bs == 1) {
174938f409ebSLisandro Dalcin       const PetscScalar *svals = values + (V ? (bs*bs*ii[i]) : 0);
175038f409ebSLisandro Dalcin       ierr = MatSetValuesBlocked_SeqSBAIJ(B,1,&i,ncols,icols,svals,INSERT_VALUES);CHKERRQ(ierr);
175138f409ebSLisandro Dalcin     } else {
175238f409ebSLisandro Dalcin       for (j=0; j<ncols; j++) {
175338f409ebSLisandro Dalcin         const PetscScalar *svals = values + (V ? (bs*bs*(ii[i]+j)) : 0);
175438f409ebSLisandro Dalcin         ierr = MatSetValuesBlocked_SeqSBAIJ(B,1,&i,1,&icols[j],svals,INSERT_VALUES);CHKERRQ(ierr);
175538f409ebSLisandro Dalcin       }
175638f409ebSLisandro Dalcin     }
175738f409ebSLisandro Dalcin   }
175838f409ebSLisandro Dalcin   if (!V) { ierr = PetscFree(values);CHKERRQ(ierr); }
175938f409ebSLisandro Dalcin   ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
176038f409ebSLisandro Dalcin   ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
176138f409ebSLisandro Dalcin   ierr = MatSetOption(B,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
176238f409ebSLisandro Dalcin   PetscFunctionReturn(0);
176338f409ebSLisandro Dalcin }
176438f409ebSLisandro Dalcin 
1765db4efbfdSBarry Smith /*
1766db4efbfdSBarry Smith    This is used to set the numeric factorization for both Cholesky and ICC symbolic factorization
1767db4efbfdSBarry Smith */
17688b1456e3SHong Zhang #undef __FUNCT__
1769d595f711SHong Zhang #define __FUNCT__ "MatSeqSBAIJSetNumericFactorization_inplace"
1770ace3abfcSBarry Smith PetscErrorCode MatSeqSBAIJSetNumericFactorization_inplace(Mat B,PetscBool natural)
1771db4efbfdSBarry Smith {
1772db4efbfdSBarry Smith   PetscErrorCode ierr;
1773ace3abfcSBarry Smith   PetscBool      flg = PETSC_FALSE;
1774db4efbfdSBarry Smith   PetscInt       bs  = B->rmap->bs;
1775db4efbfdSBarry Smith 
1776db4efbfdSBarry Smith   PetscFunctionBegin;
17770298fd71SBarry Smith   ierr = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_no_unroll",&flg,NULL);CHKERRQ(ierr);
1778db4efbfdSBarry Smith   if (flg) bs = 8;
1779db4efbfdSBarry Smith 
1780db4efbfdSBarry Smith   if (!natural) {
1781db4efbfdSBarry Smith     switch (bs) {
1782db4efbfdSBarry Smith     case 1:
1783d595f711SHong Zhang       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_inplace;
1784db4efbfdSBarry Smith       break;
1785db4efbfdSBarry Smith     case 2:
1786db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2;
1787db4efbfdSBarry Smith       break;
1788db4efbfdSBarry Smith     case 3:
1789db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3;
1790db4efbfdSBarry Smith       break;
1791db4efbfdSBarry Smith     case 4:
1792db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4;
1793db4efbfdSBarry Smith       break;
1794db4efbfdSBarry Smith     case 5:
1795db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5;
1796db4efbfdSBarry Smith       break;
1797db4efbfdSBarry Smith     case 6:
1798db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6;
1799db4efbfdSBarry Smith       break;
1800db4efbfdSBarry Smith     case 7:
1801db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7;
1802db4efbfdSBarry Smith       break;
1803db4efbfdSBarry Smith     default:
1804db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N;
1805db4efbfdSBarry Smith       break;
1806db4efbfdSBarry Smith     }
1807db4efbfdSBarry Smith   } else {
1808db4efbfdSBarry Smith     switch (bs) {
1809db4efbfdSBarry Smith     case 1:
1810d595f711SHong Zhang       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering_inplace;
1811db4efbfdSBarry Smith       break;
1812db4efbfdSBarry Smith     case 2:
1813db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering;
1814db4efbfdSBarry Smith       break;
1815db4efbfdSBarry Smith     case 3:
1816db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering;
1817db4efbfdSBarry Smith       break;
1818db4efbfdSBarry Smith     case 4:
1819db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering;
1820db4efbfdSBarry Smith       break;
1821db4efbfdSBarry Smith     case 5:
1822db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering;
1823db4efbfdSBarry Smith       break;
1824db4efbfdSBarry Smith     case 6:
1825db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering;
1826db4efbfdSBarry Smith       break;
1827db4efbfdSBarry Smith     case 7:
1828db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering;
1829db4efbfdSBarry Smith       break;
1830db4efbfdSBarry Smith     default:
1831db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering;
1832db4efbfdSBarry Smith       break;
1833db4efbfdSBarry Smith     }
1834db4efbfdSBarry Smith   }
1835db4efbfdSBarry Smith   PetscFunctionReturn(0);
1836db4efbfdSBarry Smith }
1837db4efbfdSBarry Smith 
18388cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqAIJ(Mat, MatType,MatReuse,Mat*);
18398cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqBAIJ(Mat, MatType,MatReuse,Mat*);
1840d769727bSBarry Smith 
18415c9eb25fSBarry Smith #undef __FUNCT__
18425c9eb25fSBarry Smith #define __FUNCT__ "MatGetFactor_seqsbaij_petsc"
18438cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_seqsbaij_petsc(Mat A,MatFactorType ftype,Mat *B)
18445c9eb25fSBarry Smith {
1845d0f46423SBarry Smith   PetscInt       n = A->rmap->n;
18465c9eb25fSBarry Smith   PetscErrorCode ierr;
18475c9eb25fSBarry Smith 
18485c9eb25fSBarry Smith   PetscFunctionBegin;
18490e92d65fSHong Zhang #if defined(PETSC_USE_COMPLEX)
18500e92d65fSHong Zhang   if (A->hermitian) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Hermitian Factor is not supported");
18510e92d65fSHong Zhang #endif
1852ce94432eSBarry Smith   ierr = MatCreate(PetscObjectComm((PetscObject)A),B);CHKERRQ(ierr);
18535c9eb25fSBarry Smith   ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr);
18545c9eb25fSBarry Smith   if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) {
18555c9eb25fSBarry Smith     ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr);
18560298fd71SBarry Smith     ierr = MatSeqSBAIJSetPreallocation(*B,A->rmap->bs,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr);
185726fbe8dcSKarl Rupp 
18587b056e98SHong Zhang     (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqSBAIJ;
1859c6d0d4f0SHong Zhang     (*B)->ops->iccfactorsymbolic      = MatICCFactorSymbolic_SeqSBAIJ;
1860e32f2f54SBarry Smith   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Factor type not supported");
1861d5f3da31SBarry Smith   (*B)->factortype = ftype;
18625c9eb25fSBarry Smith   PetscFunctionReturn(0);
18635c9eb25fSBarry Smith }
18645c9eb25fSBarry Smith 
1865db4efbfdSBarry Smith #undef __FUNCT__
1866db4efbfdSBarry Smith #define __FUNCT__ "MatGetFactorAvailable_seqsbaij_petsc"
1867ace3abfcSBarry Smith PetscErrorCode MatGetFactorAvailable_seqsbaij_petsc(Mat A,MatFactorType ftype,PetscBool  *flg)
1868db4efbfdSBarry Smith {
1869db4efbfdSBarry Smith   PetscFunctionBegin;
1870db4efbfdSBarry Smith   if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) {
1871db4efbfdSBarry Smith     *flg = PETSC_TRUE;
1872db4efbfdSBarry Smith   } else {
1873db4efbfdSBarry Smith     *flg = PETSC_FALSE;
1874db4efbfdSBarry Smith   }
1875db4efbfdSBarry Smith   PetscFunctionReturn(0);
1876db4efbfdSBarry Smith }
1877db4efbfdSBarry Smith 
1878611f576cSBarry Smith #if defined(PETSC_HAVE_MUMPS)
18798cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_sbaij_mumps(Mat,MatFactorType,Mat*);
1880611f576cSBarry Smith #endif
1881b5e56a35SBarry Smith #if defined(PETSC_HAVE_PASTIX)
18828cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_seqsbaij_pastix(Mat,MatFactorType,Mat*);
1883b5e56a35SBarry Smith #endif
1884913c410bSShri Abhyankar #if defined(PETSC_HAVE_SUITESPARSE)
18858cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_seqsbaij_cholmod(Mat,MatFactorType,Mat*);
188620db9a53SJed Brown #endif
18878cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_seqsbaij_sbstrm(Mat,MatFactorType,Mat*);
18885c9eb25fSBarry Smith 
18890bad9183SKris Buschelman /*MC
1890fafad747SKris Buschelman   MATSEQSBAIJ - MATSEQSBAIJ = "seqsbaij" - A matrix type to be used for sequential symmetric block sparse matrices,
18910bad9183SKris Buschelman   based on block compressed sparse row format.  Only the upper triangular portion of the matrix is stored.
18920bad9183SKris Buschelman 
1893828413b8SBarry Smith   For complex numbers by default this matrix is symmetric, NOT Hermitian symmetric. To make it Hermitian symmetric you
189471dad5bbSBarry Smith   can call MatSetOption(Mat, MAT_HERMITIAN); after MatAssemblyEnd()
1895828413b8SBarry Smith 
18960bad9183SKris Buschelman   Options Database Keys:
18970bad9183SKris Buschelman   . -mat_type seqsbaij - sets the matrix type to "seqsbaij" during a call to MatSetFromOptions()
18980bad9183SKris Buschelman 
189971dad5bbSBarry Smith   Notes: By default if you insert values into the lower triangular part of the matrix they are simply ignored (since they are not
190071dad5bbSBarry 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
190171dad5bbSBarry 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.
190271dad5bbSBarry Smith 
190371dad5bbSBarry Smith 
19040bad9183SKris Buschelman   Level: beginner
19050bad9183SKris Buschelman 
19060bad9183SKris Buschelman   .seealso: MatCreateSeqSBAIJ
19070bad9183SKris Buschelman M*/
19080bad9183SKris Buschelman 
19098cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqSBSTRM(Mat, MatType,MatReuse,Mat*);
1910aa5a9175SDahai Guo 
1911a23d5eceSKris Buschelman #undef __FUNCT__
1912a23d5eceSKris Buschelman #define __FUNCT__ "MatCreate_SeqSBAIJ"
19138cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatCreate_SeqSBAIJ(Mat B)
1914a23d5eceSKris Buschelman {
1915a23d5eceSKris Buschelman   Mat_SeqSBAIJ   *b;
1916dfbe8321SBarry Smith   PetscErrorCode ierr;
191713f74950SBarry Smith   PetscMPIInt    size;
1918ace3abfcSBarry Smith   PetscBool      no_unroll = PETSC_FALSE,no_inode = PETSC_FALSE;
1919a23d5eceSKris Buschelman 
1920a23d5eceSKris Buschelman   PetscFunctionBegin;
1921ce94432eSBarry Smith   ierr = MPI_Comm_size(PetscObjectComm((PetscObject)B),&size);CHKERRQ(ierr);
1922e32f2f54SBarry Smith   if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Comm must be of size 1");
1923a23d5eceSKris Buschelman 
1924b00a9115SJed Brown   ierr    = PetscNewLog(B,&b);CHKERRQ(ierr);
1925a23d5eceSKris Buschelman   B->data = (void*)b;
1926a23d5eceSKris Buschelman   ierr    = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr);
192726fbe8dcSKarl Rupp 
1928a23d5eceSKris Buschelman   B->ops->destroy    = MatDestroy_SeqSBAIJ;
1929a23d5eceSKris Buschelman   B->ops->view       = MatView_SeqSBAIJ;
1930a23d5eceSKris Buschelman   b->row             = 0;
1931a23d5eceSKris Buschelman   b->icol            = 0;
1932a23d5eceSKris Buschelman   b->reallocs        = 0;
1933a23d5eceSKris Buschelman   b->saved_values    = 0;
19340def2e27SBarry Smith   b->inode.limit     = 5;
19350def2e27SBarry Smith   b->inode.max_limit = 5;
1936a23d5eceSKris Buschelman 
1937a23d5eceSKris Buschelman   b->roworiented        = PETSC_TRUE;
1938a23d5eceSKris Buschelman   b->nonew              = 0;
1939a23d5eceSKris Buschelman   b->diag               = 0;
1940a23d5eceSKris Buschelman   b->solve_work         = 0;
1941a23d5eceSKris Buschelman   b->mult_work          = 0;
1942a23d5eceSKris Buschelman   B->spptr              = 0;
1943f2cbd3d5SJed Brown   B->info.nz_unneeded   = (PetscReal)b->maxnz*b->bs2;
1944a9817697SBarry Smith   b->keepnonzeropattern = PETSC_FALSE;
1945a23d5eceSKris Buschelman   b->xtoy               = 0;
1946a23d5eceSKris Buschelman   b->XtoY               = 0;
1947a23d5eceSKris Buschelman 
1948a23d5eceSKris Buschelman   b->inew    = 0;
1949a23d5eceSKris Buschelman   b->jnew    = 0;
1950a23d5eceSKris Buschelman   b->anew    = 0;
1951a23d5eceSKris Buschelman   b->a2anew  = 0;
1952a23d5eceSKris Buschelman   b->permute = PETSC_FALSE;
1953a23d5eceSKris Buschelman 
195471dad5bbSBarry Smith   b->ignore_ltriangular = PETSC_TRUE;
195526fbe8dcSKarl Rupp 
19560298fd71SBarry Smith   ierr = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_ignore_lower_triangular",&b->ignore_ltriangular,NULL);CHKERRQ(ierr);
1957941593c8SHong Zhang 
1958f5edf698SHong Zhang   b->getrow_utriangular = PETSC_FALSE;
195926fbe8dcSKarl Rupp 
19600298fd71SBarry Smith   ierr = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_getrow_uppertriangular",&b->getrow_utriangular,NULL);CHKERRQ(ierr);
1961f5edf698SHong Zhang 
1962b5e56a35SBarry Smith #if defined(PETSC_HAVE_PASTIX)
1963bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactor_pastix_C",MatGetFactor_seqsbaij_pastix);CHKERRQ(ierr);
1964b5e56a35SBarry Smith #endif
1965611f576cSBarry Smith #if defined(PETSC_HAVE_MUMPS)
1966bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactor_mumps_C",MatGetFactor_sbaij_mumps);CHKERRQ(ierr);
1967611f576cSBarry Smith #endif
1968913c410bSShri Abhyankar #if defined(PETSC_HAVE_SUITESPARSE)
1969bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactor_cholmod_C",MatGetFactor_seqsbaij_cholmod);CHKERRQ(ierr);
197020db9a53SJed Brown #endif
1971bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactorAvailable_petsc_C",MatGetFactorAvailable_seqsbaij_petsc);CHKERRQ(ierr);
1972bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactor_petsc_C",MatGetFactor_seqsbaij_petsc);CHKERRQ(ierr);
1973bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactor_sbstrm_C",MatGetFactor_seqsbaij_sbstrm);CHKERRQ(ierr);
1974bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatStoreValues_C",MatStoreValues_SeqSBAIJ);CHKERRQ(ierr);
1975bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatRetrieveValues_C",MatRetrieveValues_SeqSBAIJ);CHKERRQ(ierr);
1976bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetColumnIndices_C",MatSeqSBAIJSetColumnIndices_SeqSBAIJ);CHKERRQ(ierr);
1977bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqaij_C",MatConvert_SeqSBAIJ_SeqAIJ);CHKERRQ(ierr);
1978bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqbaij_C",MatConvert_SeqSBAIJ_SeqBAIJ);CHKERRQ(ierr);
1979bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetPreallocation_C",MatSeqSBAIJSetPreallocation_SeqSBAIJ);CHKERRQ(ierr);
198038f409ebSLisandro Dalcin   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetPreallocationCSR_C",MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ);CHKERRQ(ierr);
1981bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqsbstrm_C",MatConvert_SeqSBAIJ_SeqSBSTRM);CHKERRQ(ierr);
198223ce1328SBarry Smith 
198323ce1328SBarry Smith   B->symmetric                  = PETSC_TRUE;
198423ce1328SBarry Smith   B->structurally_symmetric     = PETSC_TRUE;
198523ce1328SBarry Smith   B->symmetric_set              = PETSC_TRUE;
198623ce1328SBarry Smith   B->structurally_symmetric_set = PETSC_TRUE;
198726fbe8dcSKarl Rupp 
198817667f90SBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)B,MATSEQSBAIJ);CHKERRQ(ierr);
19890def2e27SBarry Smith 
1990ce94432eSBarry Smith   ierr = PetscOptionsBegin(PetscObjectComm((PetscObject)B),((PetscObject)B)->prefix,"Options for SEQSBAIJ matrix","Mat");CHKERRQ(ierr);
19910298fd71SBarry Smith   ierr = PetscOptionsBool("-mat_no_unroll","Do not optimize for inodes (slower)",NULL,no_unroll,&no_unroll,NULL);CHKERRQ(ierr);
199226fbe8dcSKarl Rupp   if (no_unroll) {
199326fbe8dcSKarl Rupp     ierr = PetscInfo(B,"Not using Inode routines due to -mat_no_unroll\n");CHKERRQ(ierr);
199426fbe8dcSKarl Rupp   }
19950298fd71SBarry Smith   ierr = PetscOptionsBool("-mat_no_inode","Do not optimize for inodes (slower)",NULL,no_inode,&no_inode,NULL);CHKERRQ(ierr);
199626fbe8dcSKarl Rupp   if (no_inode) {
199726fbe8dcSKarl Rupp     ierr = PetscInfo(B,"Not using Inode routines due to -mat_no_inode\n");CHKERRQ(ierr);
199826fbe8dcSKarl Rupp   }
19990298fd71SBarry Smith   ierr = PetscOptionsInt("-mat_inode_limit","Do not use inodes larger then this value",NULL,b->inode.limit,&b->inode.limit,NULL);CHKERRQ(ierr);
20000def2e27SBarry Smith   ierr = PetscOptionsEnd();CHKERRQ(ierr);
2001ace3abfcSBarry Smith   b->inode.use = (PetscBool)(!(no_unroll || no_inode));
20020def2e27SBarry Smith   if (b->inode.limit > b->inode.max_limit) b->inode.limit = b->inode.max_limit;
2003a23d5eceSKris Buschelman   PetscFunctionReturn(0);
2004a23d5eceSKris Buschelman }
2005a23d5eceSKris Buschelman 
2006a23d5eceSKris Buschelman #undef __FUNCT__
2007a23d5eceSKris Buschelman #define __FUNCT__ "MatSeqSBAIJSetPreallocation"
2008a23d5eceSKris Buschelman /*@C
2009a23d5eceSKris Buschelman    MatSeqSBAIJSetPreallocation - Creates a sparse symmetric matrix in block AIJ (block
2010a23d5eceSKris Buschelman    compressed row) format.  For good matrix assembly performance the
2011a23d5eceSKris Buschelman    user should preallocate the matrix storage by setting the parameter nz
2012a23d5eceSKris Buschelman    (or the array nnz).  By setting these parameters accurately, performance
2013a23d5eceSKris Buschelman    during matrix assembly can be increased by more than a factor of 50.
2014a23d5eceSKris Buschelman 
2015a23d5eceSKris Buschelman    Collective on Mat
2016a23d5eceSKris Buschelman 
2017a23d5eceSKris Buschelman    Input Parameters:
2018a23d5eceSKris Buschelman +  A - the symmetric matrix
2019a23d5eceSKris Buschelman .  bs - size of block
2020a23d5eceSKris Buschelman .  nz - number of block nonzeros per block row (same for all rows)
2021a23d5eceSKris Buschelman -  nnz - array containing the number of block nonzeros in the upper triangular plus
20220298fd71SBarry Smith          diagonal portion of each block (possibly different for each block row) or NULL
2023a23d5eceSKris Buschelman 
2024a23d5eceSKris Buschelman    Options Database Keys:
2025a23d5eceSKris Buschelman .   -mat_no_unroll - uses code that does not unroll the loops in the
2026a23d5eceSKris Buschelman                      block calculations (much slower)
2027db4efbfdSBarry Smith .    -mat_block_size - size of the blocks to use (only works if a negative bs is passed in
2028a23d5eceSKris Buschelman 
2029a23d5eceSKris Buschelman    Level: intermediate
2030a23d5eceSKris Buschelman 
2031a23d5eceSKris Buschelman    Notes:
2032a23d5eceSKris Buschelman    Specify the preallocated storage with either nz or nnz (not both).
20330298fd71SBarry Smith    Set nz=PETSC_DEFAULT and nnz=NULL for PETSc to control dynamic memory
2034a7f22e61SSatish Balay    allocation.  See Users-Manual: ch_mat for details.
2035a23d5eceSKris Buschelman 
2036aa95bbe8SBarry Smith    You can call MatGetInfo() to get information on how effective the preallocation was;
2037aa95bbe8SBarry Smith    for example the fields mallocs,nz_allocated,nz_used,nz_unneeded;
2038aa95bbe8SBarry Smith    You can also run with the option -info and look for messages with the string
2039aa95bbe8SBarry Smith    malloc in them to see if additional memory allocation was needed.
2040aa95bbe8SBarry Smith 
204149a6f317SBarry Smith    If the nnz parameter is given then the nz parameter is ignored
204249a6f317SBarry Smith 
204349a6f317SBarry Smith 
204469b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateSBAIJ()
2045a23d5eceSKris Buschelman @*/
20467087cfbeSBarry Smith PetscErrorCode  MatSeqSBAIJSetPreallocation(Mat B,PetscInt bs,PetscInt nz,const PetscInt nnz[])
204713f74950SBarry Smith {
20484ac538c5SBarry Smith   PetscErrorCode ierr;
2049a23d5eceSKris Buschelman 
2050a23d5eceSKris Buschelman   PetscFunctionBegin;
20516ba663aaSJed Brown   PetscValidHeaderSpecific(B,MAT_CLASSID,1);
20526ba663aaSJed Brown   PetscValidType(B,1);
20536ba663aaSJed Brown   PetscValidLogicalCollectiveInt(B,bs,2);
20544ac538c5SBarry Smith   ierr = PetscTryMethod(B,"MatSeqSBAIJSetPreallocation_C",(Mat,PetscInt,PetscInt,const PetscInt[]),(B,bs,nz,nnz));CHKERRQ(ierr);
2055a23d5eceSKris Buschelman   PetscFunctionReturn(0);
2056a23d5eceSKris Buschelman }
205749b5e25fSSatish Balay 
20584a2ae208SSatish Balay #undef  __FUNCT__
205938f409ebSLisandro Dalcin #define __FUNCT__ "MatSeqSBAIJSetPreallocationCSR"
206038f409ebSLisandro Dalcin /*@C
206138f409ebSLisandro Dalcin    MatSeqSBAIJSetPreallocationCSR - Allocates memory for a sparse sequential matrix in symmetric block AIJ format.
206238f409ebSLisandro Dalcin 
206338f409ebSLisandro Dalcin    Input Parameters:
206438f409ebSLisandro Dalcin +  A - the matrix
206538f409ebSLisandro Dalcin .  i - the indices into j for the start of each local row (starts with zero)
206638f409ebSLisandro Dalcin .  j - the column indices for each local row (starts with zero) these must be sorted for each row
206738f409ebSLisandro Dalcin -  v - optional values in the matrix
206838f409ebSLisandro Dalcin 
206938f409ebSLisandro Dalcin    Level: developer
207038f409ebSLisandro Dalcin 
207138f409ebSLisandro Dalcin    Notes:
207238f409ebSLisandro Dalcin    The order of the entries in values is specified by the MatOption MAT_ROW_ORIENTED.  For example, C programs
207338f409ebSLisandro 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
207438f409ebSLisandro Dalcin    over rows within a block and the last index is over columns within a block row.  Fortran programs will likely set
207538f409ebSLisandro 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
207638f409ebSLisandro Dalcin    block column and the second index is over columns within a block.
207738f409ebSLisandro Dalcin 
207838f409ebSLisandro Dalcin .keywords: matrix, block, aij, compressed row, sparse
207938f409ebSLisandro Dalcin 
208038f409ebSLisandro Dalcin .seealso: MatCreate(), MatCreateSeqSBAIJ(), MatSetValuesBlocked(), MatSeqSBAIJSetPreallocation(), MATSEQSBAIJ
208138f409ebSLisandro Dalcin @*/
208238f409ebSLisandro Dalcin PetscErrorCode MatSeqSBAIJSetPreallocationCSR(Mat B,PetscInt bs,const PetscInt i[],const PetscInt j[], const PetscScalar v[])
208338f409ebSLisandro Dalcin {
208438f409ebSLisandro Dalcin   PetscErrorCode ierr;
208538f409ebSLisandro Dalcin 
208638f409ebSLisandro Dalcin   PetscFunctionBegin;
208738f409ebSLisandro Dalcin   PetscValidHeaderSpecific(B,MAT_CLASSID,1);
208838f409ebSLisandro Dalcin   PetscValidType(B,1);
208938f409ebSLisandro Dalcin   PetscValidLogicalCollectiveInt(B,bs,2);
209038f409ebSLisandro Dalcin   ierr = PetscTryMethod(B,"MatSeqSBAIJSetPreallocationCSR_C",(Mat,PetscInt,const PetscInt[],const PetscInt[],const PetscScalar[]),(B,bs,i,j,v));CHKERRQ(ierr);
209138f409ebSLisandro Dalcin   PetscFunctionReturn(0);
209238f409ebSLisandro Dalcin }
209338f409ebSLisandro Dalcin 
209438f409ebSLisandro Dalcin #undef __FUNCT__
20954a2ae208SSatish Balay #define __FUNCT__ "MatCreateSeqSBAIJ"
2096c464158bSHong Zhang /*@C
2097c464158bSHong Zhang    MatCreateSeqSBAIJ - Creates a sparse symmetric matrix in block AIJ (block
2098c464158bSHong Zhang    compressed row) format.  For good matrix assembly performance the
2099c464158bSHong Zhang    user should preallocate the matrix storage by setting the parameter nz
2100c464158bSHong Zhang    (or the array nnz).  By setting these parameters accurately, performance
2101c464158bSHong Zhang    during matrix assembly can be increased by more than a factor of 50.
210249b5e25fSSatish Balay 
2103c464158bSHong Zhang    Collective on MPI_Comm
2104c464158bSHong Zhang 
2105c464158bSHong Zhang    Input Parameters:
2106c464158bSHong Zhang +  comm - MPI communicator, set to PETSC_COMM_SELF
2107c464158bSHong Zhang .  bs - size of block
2108c464158bSHong Zhang .  m - number of rows, or number of columns
2109c464158bSHong Zhang .  nz - number of block nonzeros per block row (same for all rows)
2110744e8345SSatish Balay -  nnz - array containing the number of block nonzeros in the upper triangular plus
21110298fd71SBarry Smith          diagonal portion of each block (possibly different for each block row) or NULL
2112c464158bSHong Zhang 
2113c464158bSHong Zhang    Output Parameter:
2114c464158bSHong Zhang .  A - the symmetric matrix
2115c464158bSHong Zhang 
2116c464158bSHong Zhang    Options Database Keys:
2117c464158bSHong Zhang .   -mat_no_unroll - uses code that does not unroll the loops in the
2118c464158bSHong Zhang                      block calculations (much slower)
2119c464158bSHong Zhang .    -mat_block_size - size of the blocks to use
2120c464158bSHong Zhang 
2121c464158bSHong Zhang    Level: intermediate
2122c464158bSHong Zhang 
2123175b88e8SBarry Smith    It is recommended that one use the MatCreate(), MatSetType() and/or MatSetFromOptions(),
2124ae1d86c5SBarry Smith    MatXXXXSetPreallocation() paradgm instead of this routine directly.
2125175b88e8SBarry Smith    [MatXXXXSetPreallocation() is, for example, MatSeqAIJSetPreallocation]
2126175b88e8SBarry Smith 
2127c464158bSHong Zhang    Notes:
21286d6d819aSHong Zhang    The number of rows and columns must be divisible by blocksize.
21296d6d819aSHong Zhang    This matrix type does not support complex Hermitian operation.
2130c464158bSHong Zhang 
2131c464158bSHong Zhang    Specify the preallocated storage with either nz or nnz (not both).
21320298fd71SBarry Smith    Set nz=PETSC_DEFAULT and nnz=NULL for PETSc to control dynamic memory
2133a7f22e61SSatish Balay    allocation.  See Users-Manual: ch_mat for details.
2134c464158bSHong Zhang 
213549a6f317SBarry Smith    If the nnz parameter is given then the nz parameter is ignored
213649a6f317SBarry Smith 
213769b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateSBAIJ()
2138c464158bSHong Zhang @*/
21397087cfbeSBarry Smith PetscErrorCode  MatCreateSeqSBAIJ(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt nz,const PetscInt nnz[],Mat *A)
2140c464158bSHong Zhang {
2141dfbe8321SBarry Smith   PetscErrorCode ierr;
2142c464158bSHong Zhang 
2143c464158bSHong Zhang   PetscFunctionBegin;
2144f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,A);CHKERRQ(ierr);
2145f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*A,m,n,m,n);CHKERRQ(ierr);
2146c464158bSHong Zhang   ierr = MatSetType(*A,MATSEQSBAIJ);CHKERRQ(ierr);
2147ab93d7beSBarry Smith   ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(*A,bs,nz,(PetscInt*)nnz);CHKERRQ(ierr);
214849b5e25fSSatish Balay   PetscFunctionReturn(0);
214949b5e25fSSatish Balay }
215049b5e25fSSatish Balay 
21514a2ae208SSatish Balay #undef __FUNCT__
21524a2ae208SSatish Balay #define __FUNCT__ "MatDuplicate_SeqSBAIJ"
2153dfbe8321SBarry Smith PetscErrorCode MatDuplicate_SeqSBAIJ(Mat A,MatDuplicateOption cpvalues,Mat *B)
215449b5e25fSSatish Balay {
215549b5e25fSSatish Balay   Mat            C;
215649b5e25fSSatish Balay   Mat_SeqSBAIJ   *c,*a = (Mat_SeqSBAIJ*)A->data;
21576849ba73SBarry Smith   PetscErrorCode ierr;
2158b40805acSSatish Balay   PetscInt       i,mbs = a->mbs,nz = a->nz,bs2 =a->bs2;
215949b5e25fSSatish Balay 
216049b5e25fSSatish Balay   PetscFunctionBegin;
2161e32f2f54SBarry Smith   if (a->i[mbs] != nz) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Corrupt matrix");
216249b5e25fSSatish Balay 
216349b5e25fSSatish Balay   *B   = 0;
2164ce94432eSBarry Smith   ierr = MatCreate(PetscObjectComm((PetscObject)A),&C);CHKERRQ(ierr);
2165d0f46423SBarry Smith   ierr = MatSetSizes(C,A->rmap->N,A->cmap->n,A->rmap->N,A->cmap->n);CHKERRQ(ierr);
21668e9a0fb8SHong Zhang   ierr = MatSetType(C,MATSEQSBAIJ);CHKERRQ(ierr);
21671d5dac46SHong Zhang   ierr = PetscMemcpy(C->ops,A->ops,sizeof(struct _MatOps));CHKERRQ(ierr);
2168692f9cbeSHong Zhang   c    = (Mat_SeqSBAIJ*)C->data;
2169692f9cbeSHong Zhang 
2170273d9f13SBarry Smith   C->preallocated       = PETSC_TRUE;
2171d5f3da31SBarry Smith   C->factortype         = A->factortype;
217249b5e25fSSatish Balay   c->row                = 0;
217349b5e25fSSatish Balay   c->icol               = 0;
217449b5e25fSSatish Balay   c->saved_values       = 0;
2175a9817697SBarry Smith   c->keepnonzeropattern = a->keepnonzeropattern;
217649b5e25fSSatish Balay   C->assembled          = PETSC_TRUE;
217749b5e25fSSatish Balay 
21781e1e43feSBarry Smith   ierr   = PetscLayoutReference(A->rmap,&C->rmap);CHKERRQ(ierr);
21791e1e43feSBarry Smith   ierr   = PetscLayoutReference(A->cmap,&C->cmap);CHKERRQ(ierr);
218049b5e25fSSatish Balay   c->bs2 = a->bs2;
218149b5e25fSSatish Balay   c->mbs = a->mbs;
218249b5e25fSSatish Balay   c->nbs = a->nbs;
218349b5e25fSSatish Balay 
2184c760cd28SBarry Smith   if  (cpvalues == MAT_SHARE_NONZERO_PATTERN) {
2185c760cd28SBarry Smith     c->imax           = a->imax;
2186c760cd28SBarry Smith     c->ilen           = a->ilen;
2187c760cd28SBarry Smith     c->free_imax_ilen = PETSC_FALSE;
2188c760cd28SBarry Smith   } else {
2189dcca6d9dSJed Brown     ierr = PetscMalloc2((mbs+1),&c->imax,(mbs+1),&c->ilen);CHKERRQ(ierr);
21903bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)C,2*(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr);
219149b5e25fSSatish Balay     for (i=0; i<mbs; i++) {
219249b5e25fSSatish Balay       c->imax[i] = a->imax[i];
219349b5e25fSSatish Balay       c->ilen[i] = a->ilen[i];
219449b5e25fSSatish Balay     }
2195c760cd28SBarry Smith     c->free_imax_ilen = PETSC_TRUE;
2196c760cd28SBarry Smith   }
219749b5e25fSSatish Balay 
219849b5e25fSSatish Balay   /* allocate the matrix space */
21994da8f245SBarry Smith   if (cpvalues == MAT_SHARE_NONZERO_PATTERN) {
2200785e854fSJed Brown     ierr            = PetscMalloc1(bs2*nz,&c->a);CHKERRQ(ierr);
22013bb1ff40SBarry Smith     ierr            = PetscLogObjectMemory((PetscObject)C,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr);
220244e1c64aSLisandro Dalcin     c->i            = a->i;
220344e1c64aSLisandro Dalcin     c->j            = a->j;
22044da8f245SBarry Smith     c->singlemalloc = PETSC_FALSE;
220544e1c64aSLisandro Dalcin     c->free_a       = PETSC_TRUE;
22064da8f245SBarry Smith     c->free_ij      = PETSC_FALSE;
22074da8f245SBarry Smith     c->parent       = A;
22084da8f245SBarry Smith     ierr            = PetscObjectReference((PetscObject)A);CHKERRQ(ierr);
22094da8f245SBarry Smith     ierr            = MatSetOption(A,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
22104da8f245SBarry Smith     ierr            = MatSetOption(C,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
22114da8f245SBarry Smith   } else {
2212dcca6d9dSJed Brown     ierr            = PetscMalloc3(bs2*nz,&c->a,nz,&c->j,mbs+1,&c->i);CHKERRQ(ierr);
221313f74950SBarry Smith     ierr            = PetscMemcpy(c->i,a->i,(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr);
22143bb1ff40SBarry Smith     ierr            = PetscLogObjectMemory((PetscObject)C,(mbs+1)*sizeof(PetscInt) + nz*(bs2*sizeof(MatScalar) + sizeof(PetscInt)));CHKERRQ(ierr);
22154da8f245SBarry Smith     c->singlemalloc = PETSC_TRUE;
221644e1c64aSLisandro Dalcin     c->free_a       = PETSC_TRUE;
22174da8f245SBarry Smith     c->free_ij      = PETSC_TRUE;
22184da8f245SBarry Smith   }
221949b5e25fSSatish Balay   if (mbs > 0) {
22204da8f245SBarry Smith     if (cpvalues != MAT_SHARE_NONZERO_PATTERN) {
222113f74950SBarry Smith       ierr = PetscMemcpy(c->j,a->j,nz*sizeof(PetscInt));CHKERRQ(ierr);
22224da8f245SBarry Smith     }
222349b5e25fSSatish Balay     if (cpvalues == MAT_COPY_VALUES) {
222449b5e25fSSatish Balay       ierr = PetscMemcpy(c->a,a->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr);
222549b5e25fSSatish Balay     } else {
222649b5e25fSSatish Balay       ierr = PetscMemzero(c->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr);
222749b5e25fSSatish Balay     }
2228a1c3900fSBarry Smith     if (a->jshort) {
222944e1c64aSLisandro Dalcin       /* cannot share jshort, it is reallocated in MatAssemblyEnd_SeqSBAIJ() */
223044e1c64aSLisandro Dalcin       /* if the parent matrix is reassembled, this child matrix will never notice */
2231785e854fSJed Brown       ierr = PetscMalloc1(nz,&c->jshort);CHKERRQ(ierr);
22323bb1ff40SBarry Smith       ierr = PetscLogObjectMemory((PetscObject)C,nz*sizeof(unsigned short));CHKERRQ(ierr);
2233a1c3900fSBarry Smith       ierr = PetscMemcpy(c->jshort,a->jshort,nz*sizeof(unsigned short));CHKERRQ(ierr);
223426fbe8dcSKarl Rupp 
22354da8f245SBarry Smith       c->free_jshort = PETSC_TRUE;
22364da8f245SBarry Smith     }
2237a1c3900fSBarry Smith   }
223849b5e25fSSatish Balay 
223949b5e25fSSatish Balay   c->roworiented = a->roworiented;
224049b5e25fSSatish Balay   c->nonew       = a->nonew;
224149b5e25fSSatish Balay 
224249b5e25fSSatish Balay   if (a->diag) {
2243c760cd28SBarry Smith     if (cpvalues == MAT_SHARE_NONZERO_PATTERN) {
2244c760cd28SBarry Smith       c->diag      = a->diag;
2245c760cd28SBarry Smith       c->free_diag = PETSC_FALSE;
2246c760cd28SBarry Smith     } else {
2247785e854fSJed Brown       ierr = PetscMalloc1(mbs,&c->diag);CHKERRQ(ierr);
22483bb1ff40SBarry Smith       ierr = PetscLogObjectMemory((PetscObject)C,mbs*sizeof(PetscInt));CHKERRQ(ierr);
224926fbe8dcSKarl Rupp       for (i=0; i<mbs; i++) c->diag[i] = a->diag[i];
2250c760cd28SBarry Smith       c->free_diag = PETSC_TRUE;
2251c760cd28SBarry Smith     }
225244e1c64aSLisandro Dalcin   }
22536c6c5352SBarry Smith   c->nz         = a->nz;
2254f2cbd3d5SJed Brown   c->maxnz      = a->nz; /* Since we allocate exactly the right amount */
225549b5e25fSSatish Balay   c->solve_work = 0;
225649b5e25fSSatish Balay   c->mult_work  = 0;
225726fbe8dcSKarl Rupp 
225849b5e25fSSatish Balay   *B   = C;
2259140e18c1SBarry Smith   ierr = PetscFunctionListDuplicate(((PetscObject)A)->qlist,&((PetscObject)C)->qlist);CHKERRQ(ierr);
226049b5e25fSSatish Balay   PetscFunctionReturn(0);
226149b5e25fSSatish Balay }
226249b5e25fSSatish Balay 
22634a2ae208SSatish Balay #undef __FUNCT__
22645bba2384SShri Abhyankar #define __FUNCT__ "MatLoad_SeqSBAIJ"
2265112444f4SShri Abhyankar PetscErrorCode MatLoad_SeqSBAIJ(Mat newmat,PetscViewer viewer)
22662f480046SShri Abhyankar {
22672f480046SShri Abhyankar   Mat_SeqSBAIJ   *a;
22682f480046SShri Abhyankar   PetscErrorCode ierr;
22692f480046SShri Abhyankar   int            fd;
22702f480046SShri Abhyankar   PetscMPIInt    size;
22712f480046SShri Abhyankar   PetscInt       i,nz,header[4],*rowlengths=0,M,N,bs=1;
22722f480046SShri Abhyankar   PetscInt       *mask,mbs,*jj,j,rowcount,nzcount,k,*s_browlengths,maskcount;
22732f480046SShri Abhyankar   PetscInt       kmax,jcount,block,idx,point,nzcountb,extra_rows,rows,cols;
22742f480046SShri Abhyankar   PetscInt       *masked,nmask,tmp,bs2,ishift;
22752f480046SShri Abhyankar   PetscScalar    *aa;
2276ce94432eSBarry Smith   MPI_Comm       comm;
22772f480046SShri Abhyankar 
22782f480046SShri Abhyankar   PetscFunctionBegin;
2279ce94432eSBarry Smith   ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr);
22800298fd71SBarry Smith   ierr = PetscOptionsGetInt(((PetscObject)newmat)->prefix,"-matload_block_size",&bs,NULL);CHKERRQ(ierr);
22812f480046SShri Abhyankar   bs2  = bs*bs;
22822f480046SShri Abhyankar 
22832f480046SShri Abhyankar   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
22842f480046SShri Abhyankar   if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"view must have one processor");
22852f480046SShri Abhyankar   ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr);
22862f480046SShri Abhyankar   ierr = PetscBinaryRead(fd,header,4,PETSC_INT);CHKERRQ(ierr);
22872f480046SShri Abhyankar   if (header[0] != MAT_FILE_CLASSID) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"not Mat object");
22882f480046SShri Abhyankar   M = header[1]; N = header[2]; nz = header[3];
22892f480046SShri Abhyankar 
22902f480046SShri Abhyankar   if (header[3] < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Matrix stored in special format, cannot load as SeqSBAIJ");
22912f480046SShri Abhyankar 
22922f480046SShri Abhyankar   if (M != N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Can only do square matrices");
22932f480046SShri Abhyankar 
22942f480046SShri Abhyankar   /*
22952f480046SShri Abhyankar      This code adds extra rows to make sure the number of rows is
22962f480046SShri Abhyankar     divisible by the blocksize
22972f480046SShri Abhyankar   */
22982f480046SShri Abhyankar   mbs        = M/bs;
22992f480046SShri Abhyankar   extra_rows = bs - M + bs*(mbs);
23002f480046SShri Abhyankar   if (extra_rows == bs) extra_rows = 0;
23012f480046SShri Abhyankar   else                  mbs++;
23022f480046SShri Abhyankar   if (extra_rows) {
23032f480046SShri Abhyankar     ierr = PetscInfo(viewer,"Padding loaded matrix to match blocksize\n");CHKERRQ(ierr);
23042f480046SShri Abhyankar   }
23052f480046SShri Abhyankar 
23062f480046SShri Abhyankar   /* Set global sizes if not already set */
23072f480046SShri Abhyankar   if (newmat->rmap->n < 0 && newmat->rmap->N < 0 && newmat->cmap->n < 0 && newmat->cmap->N < 0) {
23082f480046SShri Abhyankar     ierr = MatSetSizes(newmat,PETSC_DECIDE,PETSC_DECIDE,M+extra_rows,N+extra_rows);CHKERRQ(ierr);
23092f480046SShri Abhyankar   } else { /* Check if the matrix global sizes are correct */
23102f480046SShri Abhyankar     ierr = MatGetSize(newmat,&rows,&cols);CHKERRQ(ierr);
23112f480046SShri 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);
23122f480046SShri Abhyankar   }
23132f480046SShri Abhyankar 
23142f480046SShri Abhyankar   /* read in row lengths */
2315785e854fSJed Brown   ierr = PetscMalloc1((M+extra_rows),&rowlengths);CHKERRQ(ierr);
23162f480046SShri Abhyankar   ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr);
23172f480046SShri Abhyankar   for (i=0; i<extra_rows; i++) rowlengths[M+i] = 1;
23182f480046SShri Abhyankar 
23192f480046SShri Abhyankar   /* read in column indices */
2320785e854fSJed Brown   ierr = PetscMalloc1((nz+extra_rows),&jj);CHKERRQ(ierr);
23212f480046SShri Abhyankar   ierr = PetscBinaryRead(fd,jj,nz,PETSC_INT);CHKERRQ(ierr);
23222f480046SShri Abhyankar   for (i=0; i<extra_rows; i++) jj[nz+i] = M+i;
23232f480046SShri Abhyankar 
23242f480046SShri Abhyankar   /* loop over row lengths determining block row lengths */
23251795a4d1SJed Brown   ierr     = PetscCalloc1(mbs,&s_browlengths);CHKERRQ(ierr);
2326dcca6d9dSJed Brown   ierr     = PetscMalloc2(mbs,&mask,mbs,&masked);CHKERRQ(ierr);
23272f480046SShri Abhyankar   ierr     = PetscMemzero(mask,mbs*sizeof(PetscInt));CHKERRQ(ierr);
23282f480046SShri Abhyankar   rowcount = 0;
23292f480046SShri Abhyankar   nzcount  = 0;
23302f480046SShri Abhyankar   for (i=0; i<mbs; i++) {
23312f480046SShri Abhyankar     nmask = 0;
23322f480046SShri Abhyankar     for (j=0; j<bs; j++) {
23332f480046SShri Abhyankar       kmax = rowlengths[rowcount];
23342f480046SShri Abhyankar       for (k=0; k<kmax; k++) {
23352f480046SShri Abhyankar         tmp = jj[nzcount++]/bs;   /* block col. index */
23362f480046SShri Abhyankar         if (!mask[tmp] && tmp >= i) {masked[nmask++] = tmp; mask[tmp] = 1;}
23372f480046SShri Abhyankar       }
23382f480046SShri Abhyankar       rowcount++;
23392f480046SShri Abhyankar     }
23402f480046SShri Abhyankar     s_browlengths[i] += nmask;
23412f480046SShri Abhyankar 
23422f480046SShri Abhyankar     /* zero out the mask elements we set */
23432f480046SShri Abhyankar     for (j=0; j<nmask; j++) mask[masked[j]] = 0;
23442f480046SShri Abhyankar   }
23452f480046SShri Abhyankar 
23462f480046SShri Abhyankar   /* Do preallocation */
23472f480046SShri Abhyankar   ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(newmat,bs,0,s_browlengths);CHKERRQ(ierr);
23482f480046SShri Abhyankar   a    = (Mat_SeqSBAIJ*)newmat->data;
23492f480046SShri Abhyankar 
23502f480046SShri Abhyankar   /* set matrix "i" values */
23512f480046SShri Abhyankar   a->i[0] = 0;
23522f480046SShri Abhyankar   for (i=1; i<= mbs; i++) {
23532f480046SShri Abhyankar     a->i[i]      = a->i[i-1] + s_browlengths[i-1];
23542f480046SShri Abhyankar     a->ilen[i-1] = s_browlengths[i-1];
23552f480046SShri Abhyankar   }
23562f480046SShri Abhyankar   a->nz = a->i[mbs];
23572f480046SShri Abhyankar 
23582f480046SShri Abhyankar   /* read in nonzero values */
2359785e854fSJed Brown   ierr = PetscMalloc1((nz+extra_rows),&aa);CHKERRQ(ierr);
23602f480046SShri Abhyankar   ierr = PetscBinaryRead(fd,aa,nz,PETSC_SCALAR);CHKERRQ(ierr);
23612f480046SShri Abhyankar   for (i=0; i<extra_rows; i++) aa[nz+i] = 1.0;
23622f480046SShri Abhyankar 
23632f480046SShri Abhyankar   /* set "a" and "j" values into matrix */
23642f480046SShri Abhyankar   nzcount = 0; jcount = 0;
23652f480046SShri Abhyankar   for (i=0; i<mbs; i++) {
23662f480046SShri Abhyankar     nzcountb = nzcount;
23672f480046SShri Abhyankar     nmask    = 0;
23682f480046SShri Abhyankar     for (j=0; j<bs; j++) {
23692f480046SShri Abhyankar       kmax = rowlengths[i*bs+j];
23702f480046SShri Abhyankar       for (k=0; k<kmax; k++) {
23712f480046SShri Abhyankar         tmp = jj[nzcount++]/bs; /* block col. index */
23722f480046SShri Abhyankar         if (!mask[tmp] && tmp >= i) { masked[nmask++] = tmp; mask[tmp] = 1;}
23732f480046SShri Abhyankar       }
23742f480046SShri Abhyankar     }
23752f480046SShri Abhyankar     /* sort the masked values */
23762f480046SShri Abhyankar     ierr = PetscSortInt(nmask,masked);CHKERRQ(ierr);
23772f480046SShri Abhyankar 
23782f480046SShri Abhyankar     /* set "j" values into matrix */
23792f480046SShri Abhyankar     maskcount = 1;
23802f480046SShri Abhyankar     for (j=0; j<nmask; j++) {
23812f480046SShri Abhyankar       a->j[jcount++]  = masked[j];
23822f480046SShri Abhyankar       mask[masked[j]] = maskcount++;
23832f480046SShri Abhyankar     }
23842f480046SShri Abhyankar 
23852f480046SShri Abhyankar     /* set "a" values into matrix */
23862f480046SShri Abhyankar     ishift = bs2*a->i[i];
23872f480046SShri Abhyankar     for (j=0; j<bs; j++) {
23882f480046SShri Abhyankar       kmax = rowlengths[i*bs+j];
23892f480046SShri Abhyankar       for (k=0; k<kmax; k++) {
23902f480046SShri Abhyankar         tmp = jj[nzcountb]/bs;        /* block col. index */
23912f480046SShri Abhyankar         if (tmp >= i) {
23922f480046SShri Abhyankar           block     = mask[tmp] - 1;
23932f480046SShri Abhyankar           point     = jj[nzcountb] - bs*tmp;
23942f480046SShri Abhyankar           idx       = ishift + bs2*block + j + bs*point;
23952f480046SShri Abhyankar           a->a[idx] = aa[nzcountb];
23962f480046SShri Abhyankar         }
23972f480046SShri Abhyankar         nzcountb++;
23982f480046SShri Abhyankar       }
23992f480046SShri Abhyankar     }
24002f480046SShri Abhyankar     /* zero out the mask elements we set */
24012f480046SShri Abhyankar     for (j=0; j<nmask; j++) mask[masked[j]] = 0;
24022f480046SShri Abhyankar   }
24032f480046SShri Abhyankar   if (jcount != a->nz) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Bad binary matrix");
24042f480046SShri Abhyankar 
24052f480046SShri Abhyankar   ierr = PetscFree(rowlengths);CHKERRQ(ierr);
24062f480046SShri Abhyankar   ierr = PetscFree(s_browlengths);CHKERRQ(ierr);
24072f480046SShri Abhyankar   ierr = PetscFree(aa);CHKERRQ(ierr);
24082f480046SShri Abhyankar   ierr = PetscFree(jj);CHKERRQ(ierr);
24092f480046SShri Abhyankar   ierr = PetscFree2(mask,masked);CHKERRQ(ierr);
24102f480046SShri Abhyankar 
24112f480046SShri Abhyankar   ierr = MatAssemblyBegin(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
24122f480046SShri Abhyankar   ierr = MatAssemblyEnd(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
24132f480046SShri Abhyankar   PetscFunctionReturn(0);
24142f480046SShri Abhyankar }
24152f480046SShri Abhyankar 
24162f480046SShri Abhyankar #undef __FUNCT__
2417c75a6043SHong Zhang #define __FUNCT__ "MatCreateSeqSBAIJWithArrays"
2418c75a6043SHong Zhang /*@
2419c75a6043SHong Zhang      MatCreateSeqSBAIJWithArrays - Creates an sequential SBAIJ matrix using matrix elements
2420c75a6043SHong Zhang               (upper triangular entries in CSR format) provided by the user.
2421c75a6043SHong Zhang 
2422c75a6043SHong Zhang      Collective on MPI_Comm
2423c75a6043SHong Zhang 
2424c75a6043SHong Zhang    Input Parameters:
2425c75a6043SHong Zhang +  comm - must be an MPI communicator of size 1
2426c75a6043SHong Zhang .  bs - size of block
2427c75a6043SHong Zhang .  m - number of rows
2428c75a6043SHong Zhang .  n - number of columns
2429c75a6043SHong Zhang .  i - row indices
2430c75a6043SHong Zhang .  j - column indices
2431c75a6043SHong Zhang -  a - matrix values
2432c75a6043SHong Zhang 
2433c75a6043SHong Zhang    Output Parameter:
2434c75a6043SHong Zhang .  mat - the matrix
2435c75a6043SHong Zhang 
2436dfb205c3SBarry Smith    Level: advanced
2437c75a6043SHong Zhang 
2438c75a6043SHong Zhang    Notes:
2439c75a6043SHong Zhang        The i, j, and a arrays are not copied by this routine, the user must free these arrays
2440c75a6043SHong Zhang     once the matrix is destroyed
2441c75a6043SHong Zhang 
2442c75a6043SHong Zhang        You cannot set new nonzero locations into this matrix, that will generate an error.
2443c75a6043SHong Zhang 
2444c75a6043SHong Zhang        The i and j indices are 0 based
2445c75a6043SHong Zhang 
2446dfb205c3SBarry 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
2447dfb205c3SBarry Smith        it is the regular CSR format excluding the lower triangular elements.
2448dfb205c3SBarry Smith 
244969b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSBAIJ(), MatCreateSeqSBAIJ()
2450c75a6043SHong Zhang 
2451c75a6043SHong Zhang @*/
24527087cfbeSBarry Smith PetscErrorCode  MatCreateSeqSBAIJWithArrays(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt *i,PetscInt *j,PetscScalar *a,Mat *mat)
2453c75a6043SHong Zhang {
2454c75a6043SHong Zhang   PetscErrorCode ierr;
2455c75a6043SHong Zhang   PetscInt       ii;
2456c75a6043SHong Zhang   Mat_SeqSBAIJ   *sbaij;
2457c75a6043SHong Zhang 
2458c75a6043SHong Zhang   PetscFunctionBegin;
2459e32f2f54SBarry Smith   if (bs != 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"block size %D > 1 is not supported yet",bs);
2460e32f2f54SBarry Smith   if (i[0]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"i (row indices) must start with 0");
2461c75a6043SHong Zhang 
2462c75a6043SHong Zhang   ierr  = MatCreate(comm,mat);CHKERRQ(ierr);
2463c75a6043SHong Zhang   ierr  = MatSetSizes(*mat,m,n,m,n);CHKERRQ(ierr);
2464c75a6043SHong Zhang   ierr  = MatSetType(*mat,MATSEQSBAIJ);CHKERRQ(ierr);
2465c75a6043SHong Zhang   ierr  = MatSeqSBAIJSetPreallocation_SeqSBAIJ(*mat,bs,MAT_SKIP_ALLOCATION,0);CHKERRQ(ierr);
2466c75a6043SHong Zhang   sbaij = (Mat_SeqSBAIJ*)(*mat)->data;
2467dcca6d9dSJed Brown   ierr  = PetscMalloc2(m,&sbaij->imax,m,&sbaij->ilen);CHKERRQ(ierr);
24683bb1ff40SBarry Smith   ierr  = PetscLogObjectMemory((PetscObject)*mat,2*m*sizeof(PetscInt));CHKERRQ(ierr);
2469c75a6043SHong Zhang 
2470c75a6043SHong Zhang   sbaij->i = i;
2471c75a6043SHong Zhang   sbaij->j = j;
2472c75a6043SHong Zhang   sbaij->a = a;
247326fbe8dcSKarl Rupp 
2474c75a6043SHong Zhang   sbaij->singlemalloc = PETSC_FALSE;
2475c75a6043SHong Zhang   sbaij->nonew        = -1;             /*this indicates that inserting a new value in the matrix that generates a new nonzero is an error*/
2476e6b907acSBarry Smith   sbaij->free_a       = PETSC_FALSE;
2477e6b907acSBarry Smith   sbaij->free_ij      = PETSC_FALSE;
2478c75a6043SHong Zhang 
2479c75a6043SHong Zhang   for (ii=0; ii<m; ii++) {
2480c75a6043SHong Zhang     sbaij->ilen[ii] = sbaij->imax[ii] = i[ii+1] - i[ii];
2481c75a6043SHong Zhang #if defined(PETSC_USE_DEBUG)
2482e32f2f54SBarry 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]);
2483c75a6043SHong Zhang #endif
2484c75a6043SHong Zhang   }
2485c75a6043SHong Zhang #if defined(PETSC_USE_DEBUG)
2486c75a6043SHong Zhang   for (ii=0; ii<sbaij->i[m]; ii++) {
2487e32f2f54SBarry Smith     if (j[ii] < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column index at location = %d index = %d",ii,j[ii]);
2488e32f2f54SBarry 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]);
2489c75a6043SHong Zhang   }
2490c75a6043SHong Zhang #endif
2491c75a6043SHong Zhang 
2492c75a6043SHong Zhang   ierr = MatAssemblyBegin(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2493c75a6043SHong Zhang   ierr = MatAssemblyEnd(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2494c75a6043SHong Zhang   PetscFunctionReturn(0);
2495c75a6043SHong Zhang }
2496d06b337dSHong Zhang 
2497d06b337dSHong Zhang 
2498d06b337dSHong Zhang 
249949b5e25fSSatish Balay 
250049b5e25fSSatish Balay 
2501