xref: /petsc/src/mat/impls/sbaij/seq/sbaij.c (revision 7b2ba04b74f263a5e21a714469f897bc60247435)
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;
5348dd3d27SHong 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   }
6148dd3d27SHong Zhang   for (i=0; i<a->mbs; i++) {
6248dd3d27SHong Zhang     a->diag[i] = a->i[i+1];
6348dd3d27SHong Zhang     for (j=a->i[i]; j<a->i[i+1]; j++) {
6448dd3d27SHong Zhang       if (a->j[j] == i) {
6548dd3d27SHong Zhang         a->diag[i] = j;
6648dd3d27SHong Zhang         break;
6748dd3d27SHong Zhang       }
6848dd3d27SHong Zhang     }
6948dd3d27SHong 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"
23752768537SHong Zhang PetscErrorCode MatGetRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
23849b5e25fSSatish Balay {
23949b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
2406849ba73SBarry Smith   PetscErrorCode ierr;
24149b5e25fSSatish Balay 
24249b5e25fSSatish Balay   PetscFunctionBegin;
243e32f2f54SBarry 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()");
24452768537SHong Zhang 
245f5edf698SHong Zhang   /* Get the upper triangular part of the row */
24652768537SHong Zhang   ierr = MatGetRow_SeqBAIJ_private(A,row,nz,idx,v,a->i,a->j,a->a);CHKERRQ(ierr);
24749b5e25fSSatish Balay   PetscFunctionReturn(0);
24849b5e25fSSatish Balay }
24949b5e25fSSatish Balay 
2504a2ae208SSatish Balay #undef __FUNCT__
2514a2ae208SSatish Balay #define __FUNCT__ "MatRestoreRow_SeqSBAIJ"
25213f74950SBarry Smith PetscErrorCode MatRestoreRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
25349b5e25fSSatish Balay {
254dfbe8321SBarry Smith   PetscErrorCode ierr;
25549b5e25fSSatish Balay 
25649b5e25fSSatish Balay   PetscFunctionBegin;
25705b42c5fSBarry Smith   if (idx) {ierr = PetscFree(*idx);CHKERRQ(ierr);}
25805b42c5fSBarry Smith   if (v)   {ierr = PetscFree(*v);CHKERRQ(ierr);}
25949b5e25fSSatish Balay   PetscFunctionReturn(0);
26049b5e25fSSatish Balay }
26149b5e25fSSatish Balay 
2624a2ae208SSatish Balay #undef __FUNCT__
263f5edf698SHong Zhang #define __FUNCT__ "MatGetRowUpperTriangular_SeqSBAIJ"
264f5edf698SHong Zhang PetscErrorCode MatGetRowUpperTriangular_SeqSBAIJ(Mat A)
265f5edf698SHong Zhang {
266f5edf698SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
267f5edf698SHong Zhang 
268f5edf698SHong Zhang   PetscFunctionBegin;
269f5edf698SHong Zhang   a->getrow_utriangular = PETSC_TRUE;
270f5edf698SHong Zhang   PetscFunctionReturn(0);
271f5edf698SHong Zhang }
272f5edf698SHong Zhang #undef __FUNCT__
273f5edf698SHong Zhang #define __FUNCT__ "MatRestoreRowUpperTriangular_SeqSBAIJ"
274f5edf698SHong Zhang PetscErrorCode MatRestoreRowUpperTriangular_SeqSBAIJ(Mat A)
275f5edf698SHong Zhang {
276f5edf698SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
277f5edf698SHong Zhang 
278f5edf698SHong Zhang   PetscFunctionBegin;
279f5edf698SHong Zhang   a->getrow_utriangular = PETSC_FALSE;
280f5edf698SHong Zhang   PetscFunctionReturn(0);
281f5edf698SHong Zhang }
282f5edf698SHong Zhang 
283f5edf698SHong Zhang #undef __FUNCT__
2844a2ae208SSatish Balay #define __FUNCT__ "MatTranspose_SeqSBAIJ"
285fc4dec0aSBarry Smith PetscErrorCode MatTranspose_SeqSBAIJ(Mat A,MatReuse reuse,Mat *B)
28649b5e25fSSatish Balay {
287dfbe8321SBarry Smith   PetscErrorCode ierr;
2885fd66863SKarl Rupp 
28949b5e25fSSatish Balay   PetscFunctionBegin;
290815cbec1SBarry Smith   if (reuse == MAT_INITIAL_MATRIX || *B != A) {
291999d9058SBarry Smith     ierr = MatDuplicate(A,MAT_COPY_VALUES,B);CHKERRQ(ierr);
292fc4dec0aSBarry Smith   }
2938115998fSBarry Smith   PetscFunctionReturn(0);
29449b5e25fSSatish Balay }
29549b5e25fSSatish Balay 
2964a2ae208SSatish Balay #undef __FUNCT__
2974a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_ASCII"
2987da1fb6eSBarry Smith PetscErrorCode MatView_SeqSBAIJ_ASCII(Mat A,PetscViewer viewer)
29949b5e25fSSatish Balay {
30049b5e25fSSatish Balay   Mat_SeqSBAIJ      *a = (Mat_SeqSBAIJ*)A->data;
301dfbe8321SBarry Smith   PetscErrorCode    ierr;
302d0f46423SBarry Smith   PetscInt          i,j,bs = A->rmap->bs,k,l,bs2=a->bs2;
303f3ef73ceSBarry Smith   PetscViewerFormat format;
304121deb67SSatish Balay   PetscInt          *diag;
30549b5e25fSSatish Balay 
30649b5e25fSSatish Balay   PetscFunctionBegin;
307b0a32e0cSBarry Smith   ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
308456192e2SBarry Smith   if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
30977431f27SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  block size is %D\n",bs);CHKERRQ(ierr);
310fb9695e5SSatish Balay   } else if (format == PETSC_VIEWER_ASCII_MATLAB) {
311d2507d54SMatthew Knepley     Mat aij;
312d5f3da31SBarry Smith     if (A->factortype && bs>1) {
31370d5e725SHong 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);
31470d5e725SHong Zhang       PetscFunctionReturn(0);
31570d5e725SHong Zhang     }
316c9f458caSMatthew Knepley     ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&aij);CHKERRQ(ierr);
317c9f458caSMatthew Knepley     ierr = MatView(aij,viewer);CHKERRQ(ierr);
3186bf464f9SBarry Smith     ierr = MatDestroy(&aij);CHKERRQ(ierr);
319fb9695e5SSatish Balay   } else if (format == PETSC_VIEWER_ASCII_COMMON) {
320d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
32149b5e25fSSatish Balay     for (i=0; i<a->mbs; i++) {
32249b5e25fSSatish Balay       for (j=0; j<bs; j++) {
32377431f27SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr);
32449b5e25fSSatish Balay         for (k=a->i[i]; k<a->i[i+1]; k++) {
32549b5e25fSSatish Balay           for (l=0; l<bs; l++) {
32649b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX)
32749b5e25fSSatish Balay             if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
32857622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k]+l,
32957622a8eSBarry Smith                                             (double)PetscRealPart(a->a[bs2*k + l*bs + j]),(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
33049b5e25fSSatish Balay             } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
33157622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k]+l,
33257622a8eSBarry Smith                                             (double)PetscRealPart(a->a[bs2*k + l*bs + j]),-(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
33349b5e25fSSatish Balay             } else if (PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
33457622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
33549b5e25fSSatish Balay             }
33649b5e25fSSatish Balay #else
33749b5e25fSSatish Balay             if (a->a[bs2*k + l*bs + j] != 0.0) {
33857622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)a->a[bs2*k + l*bs + j]);CHKERRQ(ierr);
33949b5e25fSSatish Balay             }
34049b5e25fSSatish Balay #endif
34149b5e25fSSatish Balay           }
34249b5e25fSSatish Balay         }
343b0a32e0cSBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
34449b5e25fSSatish Balay       }
34549b5e25fSSatish Balay     }
346d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
347c1490034SHong Zhang   } else if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) {
348c1490034SHong Zhang     PetscFunctionReturn(0);
34949b5e25fSSatish Balay   } else {
350d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
3512c990fa1SHong Zhang     if (A->factortype) { /* for factored matrix */
3522c990fa1SHong Zhang       if (bs>1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"matrix is factored with bs>1. Not implemented yet");
3532c990fa1SHong Zhang 
354121deb67SSatish Balay       diag=a->diag;
355121deb67SSatish Balay       for (i=0; i<a->mbs; i++) { /* for row block i */
3562c990fa1SHong Zhang         ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i);CHKERRQ(ierr);
3572c990fa1SHong Zhang         /* diagonal entry */
3582c990fa1SHong Zhang #if defined(PETSC_USE_COMPLEX)
3592c990fa1SHong Zhang         if (PetscImaginaryPart(a->a[diag[i]]) > 0.0) {
36057622a8eSBarry 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);
3612c990fa1SHong Zhang         } else if (PetscImaginaryPart(a->a[diag[i]]) < 0.0) {
36257622a8eSBarry 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);
3632c990fa1SHong Zhang         } else {
36457622a8eSBarry Smith           ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[diag[i]],(double)PetscRealPart(1.0/a->a[diag[i]]));CHKERRQ(ierr);
3652c990fa1SHong Zhang         }
3662c990fa1SHong Zhang #else
3676712e2f1SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[diag[i]],(double)(1.0/a->a[diag[i]]));CHKERRQ(ierr);
3682c990fa1SHong Zhang #endif
3692c990fa1SHong Zhang         /* off-diagonal entries */
3702c990fa1SHong Zhang         for (k=a->i[i]; k<a->i[i+1]-1; k++) {
3712c990fa1SHong Zhang #if defined(PETSC_USE_COMPLEX)
372ca0704adSBarry Smith           if (PetscImaginaryPart(a->a[k]) > 0.0) {
37357622a8eSBarry Smith             ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k],(double)PetscRealPart(a->a[k]),(double)PetscImaginaryPart(a->a[k]));CHKERRQ(ierr);
374ca0704adSBarry Smith           } else if (PetscImaginaryPart(a->a[k]) < 0.0) {
37557622a8eSBarry Smith             ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k],(double)PetscRealPart(a->a[k]),-(double)PetscImaginaryPart(a->a[k]));CHKERRQ(ierr);
3762c990fa1SHong Zhang           } else {
37757622a8eSBarry Smith             ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k],(double)PetscRealPart(a->a[k]));CHKERRQ(ierr);
3782c990fa1SHong Zhang           }
3792c990fa1SHong Zhang #else
38057622a8eSBarry Smith           ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[k],(double)a->a[k]);CHKERRQ(ierr);
3812c990fa1SHong Zhang #endif
3822c990fa1SHong Zhang         }
3832c990fa1SHong Zhang         ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
3842c990fa1SHong Zhang       }
3852c990fa1SHong Zhang 
3862c990fa1SHong Zhang     } else { /* for non-factored matrix */
3870c74a584SJed Brown       for (i=0; i<a->mbs; i++) { /* for row block i */
3880c74a584SJed Brown         for (j=0; j<bs; j++) {   /* for row bs*i + j */
38977431f27SBarry Smith           ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr);
3900c74a584SJed Brown           for (k=a->i[i]; k<a->i[i+1]; k++) { /* for column block */
3910c74a584SJed Brown             for (l=0; l<bs; l++) {            /* for column */
39249b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX)
39349b5e25fSSatish Balay               if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0) {
39457622a8eSBarry Smith                 ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k]+l,
39557622a8eSBarry Smith                                               (double)PetscRealPart(a->a[bs2*k + l*bs + j]),(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
39649b5e25fSSatish Balay               } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0) {
39757622a8eSBarry Smith                 ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k]+l,
39857622a8eSBarry Smith                                               (double)PetscRealPart(a->a[bs2*k + l*bs + j]),-(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
39949b5e25fSSatish Balay               } else {
40057622a8eSBarry Smith                 ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
40149b5e25fSSatish Balay               }
40249b5e25fSSatish Balay #else
40357622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)a->a[bs2*k + l*bs + j]);CHKERRQ(ierr);
40449b5e25fSSatish Balay #endif
40549b5e25fSSatish Balay             }
40649b5e25fSSatish Balay           }
407b0a32e0cSBarry Smith           ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
40849b5e25fSSatish Balay         }
40949b5e25fSSatish Balay       }
4102c990fa1SHong Zhang     }
411d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
41249b5e25fSSatish Balay   }
413b0a32e0cSBarry Smith   ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
41449b5e25fSSatish Balay   PetscFunctionReturn(0);
41549b5e25fSSatish Balay }
41649b5e25fSSatish Balay 
4179804daf3SBarry Smith #include <petscdraw.h>
4184a2ae208SSatish Balay #undef __FUNCT__
4194a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_Draw_Zoom"
4206849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw_Zoom(PetscDraw draw,void *Aa)
42149b5e25fSSatish Balay {
42249b5e25fSSatish Balay   Mat            A = (Mat) Aa;
42349b5e25fSSatish Balay   Mat_SeqSBAIJ   *a=(Mat_SeqSBAIJ*)A->data;
4246849ba73SBarry Smith   PetscErrorCode ierr;
425d0f46423SBarry Smith   PetscInt       row,i,j,k,l,mbs=a->mbs,color,bs=A->rmap->bs,bs2=a->bs2;
42613f74950SBarry Smith   PetscMPIInt    rank;
42749b5e25fSSatish Balay   PetscReal      xl,yl,xr,yr,x_l,x_r,y_l,y_r;
42849b5e25fSSatish Balay   MatScalar      *aa;
42949b5e25fSSatish Balay   MPI_Comm       comm;
430b0a32e0cSBarry Smith   PetscViewer    viewer;
43149b5e25fSSatish Balay 
43249b5e25fSSatish Balay   PetscFunctionBegin;
43349b5e25fSSatish Balay   /*
43449b5e25fSSatish Balay     This is nasty. If this is called from an originally parallel matrix
43549b5e25fSSatish Balay     then all processes call this,but only the first has the matrix so the
43649b5e25fSSatish Balay     rest should return immediately.
43749b5e25fSSatish Balay   */
43849b5e25fSSatish Balay   ierr = PetscObjectGetComm((PetscObject)draw,&comm);CHKERRQ(ierr);
43949b5e25fSSatish Balay   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
44049b5e25fSSatish Balay   if (rank) PetscFunctionReturn(0);
44149b5e25fSSatish Balay 
44249b5e25fSSatish Balay   ierr = PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer);CHKERRQ(ierr);
44349b5e25fSSatish Balay 
444b0a32e0cSBarry Smith   ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr);
445b0a32e0cSBarry Smith   PetscDrawString(draw, .3*(xl+xr), .3*(yl+yr), PETSC_DRAW_BLACK, "symmetric");
44649b5e25fSSatish Balay 
44749b5e25fSSatish Balay   /* loop over matrix elements drawing boxes */
448b0a32e0cSBarry Smith   color = PETSC_DRAW_BLUE;
44949b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
45049b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
451d0f46423SBarry Smith       y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0;
45249b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
45349b5e25fSSatish Balay       aa  = a->a + j*bs2;
45449b5e25fSSatish Balay       for (k=0; k<bs; k++) {
45549b5e25fSSatish Balay         for (l=0; l<bs; l++) {
45649b5e25fSSatish Balay           if (PetscRealPart(*aa++) >=  0.) continue;
457b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
45849b5e25fSSatish Balay         }
45949b5e25fSSatish Balay       }
46049b5e25fSSatish Balay     }
46149b5e25fSSatish Balay   }
462b0a32e0cSBarry Smith   color = PETSC_DRAW_CYAN;
46349b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
46449b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
465d0f46423SBarry Smith       y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0;
46649b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
46749b5e25fSSatish Balay       aa = a->a + j*bs2;
46849b5e25fSSatish Balay       for (k=0; k<bs; k++) {
46949b5e25fSSatish Balay         for (l=0; l<bs; l++) {
47049b5e25fSSatish Balay           if (PetscRealPart(*aa++) != 0.) continue;
471b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
47249b5e25fSSatish Balay         }
47349b5e25fSSatish Balay       }
47449b5e25fSSatish Balay     }
47549b5e25fSSatish Balay   }
47649b5e25fSSatish Balay 
477b0a32e0cSBarry Smith   color = PETSC_DRAW_RED;
47849b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
47949b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
480d0f46423SBarry Smith       y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0;
48149b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
48249b5e25fSSatish Balay       aa = a->a + j*bs2;
48349b5e25fSSatish Balay       for (k=0; k<bs; k++) {
48449b5e25fSSatish Balay         for (l=0; l<bs; l++) {
48549b5e25fSSatish Balay           if (PetscRealPart(*aa++) <= 0.) continue;
486b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
48749b5e25fSSatish Balay         }
48849b5e25fSSatish Balay       }
48949b5e25fSSatish Balay     }
49049b5e25fSSatish Balay   }
49149b5e25fSSatish Balay   PetscFunctionReturn(0);
49249b5e25fSSatish Balay }
49349b5e25fSSatish Balay 
4944a2ae208SSatish Balay #undef __FUNCT__
4954a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_Draw"
4966849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw(Mat A,PetscViewer viewer)
49749b5e25fSSatish Balay {
498dfbe8321SBarry Smith   PetscErrorCode ierr;
49949b5e25fSSatish Balay   PetscReal      xl,yl,xr,yr,w,h;
500b0a32e0cSBarry Smith   PetscDraw      draw;
501ace3abfcSBarry Smith   PetscBool      isnull;
50249b5e25fSSatish Balay 
50349b5e25fSSatish Balay   PetscFunctionBegin;
504b0a32e0cSBarry Smith   ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
505b0a32e0cSBarry Smith   ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr); if (isnull) PetscFunctionReturn(0);
50649b5e25fSSatish Balay 
50749b5e25fSSatish Balay   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer);CHKERRQ(ierr);
508d0f46423SBarry Smith   xr   = A->rmap->N; yr = A->rmap->N; h = yr/10.0; w = xr/10.0;
50949b5e25fSSatish Balay   xr  += w;    yr += h;  xl = -w;     yl = -h;
510b0a32e0cSBarry Smith   ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr);
511b0a32e0cSBarry Smith   ierr = PetscDrawZoom(draw,MatView_SeqSBAIJ_Draw_Zoom,A);CHKERRQ(ierr);
5120298fd71SBarry Smith   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",NULL);CHKERRQ(ierr);
51349b5e25fSSatish Balay   PetscFunctionReturn(0);
51449b5e25fSSatish Balay }
51549b5e25fSSatish Balay 
5164a2ae208SSatish Balay #undef __FUNCT__
5174a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ"
518dfbe8321SBarry Smith PetscErrorCode MatView_SeqSBAIJ(Mat A,PetscViewer viewer)
51949b5e25fSSatish Balay {
520dfbe8321SBarry Smith   PetscErrorCode ierr;
521ace3abfcSBarry Smith   PetscBool      iascii,isdraw;
52208917f38SBarry Smith   FILE           *file = 0;
52349b5e25fSSatish Balay 
52449b5e25fSSatish Balay   PetscFunctionBegin;
525251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
526251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr);
52732077d6dSBarry Smith   if (iascii) {
52849b5e25fSSatish Balay     ierr = MatView_SeqSBAIJ_ASCII(A,viewer);CHKERRQ(ierr);
52949b5e25fSSatish Balay   } else if (isdraw) {
53049b5e25fSSatish Balay     ierr = MatView_SeqSBAIJ_Draw(A,viewer);CHKERRQ(ierr);
53149b5e25fSSatish Balay   } else {
532a5e6ed63SBarry Smith     Mat B;
533ceb03754SKris Buschelman     ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&B);CHKERRQ(ierr);
534a5e6ed63SBarry Smith     ierr = MatView(B,viewer);CHKERRQ(ierr);
5356bf464f9SBarry Smith     ierr = MatDestroy(&B);CHKERRQ(ierr);
53608917f38SBarry Smith     ierr = PetscViewerBinaryGetInfoPointer(viewer,&file);CHKERRQ(ierr);
53708917f38SBarry Smith     if (file) {
53808917f38SBarry Smith       fprintf(file,"-matload_block_size %d\n",(int)A->rmap->bs);
53908917f38SBarry Smith     }
54049b5e25fSSatish Balay   }
54149b5e25fSSatish Balay   PetscFunctionReturn(0);
54249b5e25fSSatish Balay }
54349b5e25fSSatish Balay 
54449b5e25fSSatish Balay 
5454a2ae208SSatish Balay #undef __FUNCT__
5464a2ae208SSatish Balay #define __FUNCT__ "MatGetValues_SeqSBAIJ"
54713f74950SBarry Smith PetscErrorCode MatGetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],PetscScalar v[])
54849b5e25fSSatish Balay {
549045c9aa0SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
55013f74950SBarry Smith   PetscInt     *rp,k,low,high,t,row,nrow,i,col,l,*aj = a->j;
55113f74950SBarry Smith   PetscInt     *ai = a->i,*ailen = a->ilen;
552d0f46423SBarry Smith   PetscInt     brow,bcol,ridx,cidx,bs=A->rmap->bs,bs2=a->bs2;
55397e567efSBarry Smith   MatScalar    *ap,*aa = a->a;
55449b5e25fSSatish Balay 
55549b5e25fSSatish Balay   PetscFunctionBegin;
55649b5e25fSSatish Balay   for (k=0; k<m; k++) { /* loop over rows */
55749b5e25fSSatish Balay     row = im[k]; brow = row/bs;
558e32f2f54SBarry Smith     if (row < 0) {v += n; continue;} /* SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative row: %D",row); */
559e32f2f54SBarry 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);
56049b5e25fSSatish Balay     rp   = aj + ai[brow]; ap = aa + bs2*ai[brow];
56149b5e25fSSatish Balay     nrow = ailen[brow];
56249b5e25fSSatish Balay     for (l=0; l<n; l++) { /* loop over columns */
563e32f2f54SBarry Smith       if (in[l] < 0) {v++; continue;} /* SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column: %D",in[l]); */
564e32f2f54SBarry 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);
56549b5e25fSSatish Balay       col  = in[l];
56649b5e25fSSatish Balay       bcol = col/bs;
56749b5e25fSSatish Balay       cidx = col%bs;
56849b5e25fSSatish Balay       ridx = row%bs;
56949b5e25fSSatish Balay       high = nrow;
57049b5e25fSSatish Balay       low  = 0; /* assume unsorted */
57149b5e25fSSatish Balay       while (high-low > 5) {
57249b5e25fSSatish Balay         t = (low+high)/2;
57349b5e25fSSatish Balay         if (rp[t] > bcol) high = t;
57449b5e25fSSatish Balay         else              low  = t;
57549b5e25fSSatish Balay       }
57649b5e25fSSatish Balay       for (i=low; i<high; i++) {
57749b5e25fSSatish Balay         if (rp[i] > bcol) break;
57849b5e25fSSatish Balay         if (rp[i] == bcol) {
57949b5e25fSSatish Balay           *v++ = ap[bs2*i+bs*cidx+ridx];
58049b5e25fSSatish Balay           goto finished;
58149b5e25fSSatish Balay         }
58249b5e25fSSatish Balay       }
58397e567efSBarry Smith       *v++ = 0.0;
58449b5e25fSSatish Balay finished:;
58549b5e25fSSatish Balay     }
58649b5e25fSSatish Balay   }
58749b5e25fSSatish Balay   PetscFunctionReturn(0);
58849b5e25fSSatish Balay }
58949b5e25fSSatish Balay 
59049b5e25fSSatish Balay 
5914a2ae208SSatish Balay #undef __FUNCT__
5924a2ae208SSatish Balay #define __FUNCT__ "MatSetValuesBlocked_SeqSBAIJ"
59313f74950SBarry Smith PetscErrorCode MatSetValuesBlocked_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is)
59449b5e25fSSatish Balay {
5950880e062SHong Zhang   Mat_SeqSBAIJ      *a = (Mat_SeqSBAIJ*)A->data;
5966849ba73SBarry Smith   PetscErrorCode    ierr;
597e2ee6c50SBarry Smith   PetscInt          *rp,k,low,high,t,ii,jj,row,nrow,i,col,l,rmax,N,lastcol = -1;
59813f74950SBarry Smith   PetscInt          *imax      =a->imax,*ai=a->i,*ailen=a->ilen;
599d0f46423SBarry Smith   PetscInt          *aj        =a->j,nonew=a->nonew,bs2=a->bs2,bs=A->rmap->bs,stepval;
600ace3abfcSBarry Smith   PetscBool         roworiented=a->roworiented;
601dd6ea824SBarry Smith   const PetscScalar *value     = v;
602f15d580aSBarry Smith   MatScalar         *ap,*aa = a->a,*bap;
6030880e062SHong Zhang 
60449b5e25fSSatish Balay   PetscFunctionBegin;
60526fbe8dcSKarl Rupp   if (roworiented) stepval = (n-1)*bs;
60626fbe8dcSKarl Rupp   else stepval = (m-1)*bs;
60726fbe8dcSKarl Rupp 
6080880e062SHong Zhang   for (k=0; k<m; k++) { /* loop over added rows */
6090880e062SHong Zhang     row = im[k];
6100880e062SHong Zhang     if (row < 0) continue;
6112515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
612e32f2f54SBarry Smith     if (row >= a->mbs) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,a->mbs-1);
6130880e062SHong Zhang #endif
6140880e062SHong Zhang     rp   = aj + ai[row];
6150880e062SHong Zhang     ap   = aa + bs2*ai[row];
6160880e062SHong Zhang     rmax = imax[row];
6170880e062SHong Zhang     nrow = ailen[row];
6180880e062SHong Zhang     low  = 0;
619818f2c47SBarry Smith     high = nrow;
6200880e062SHong Zhang     for (l=0; l<n; l++) { /* loop over added columns */
6210880e062SHong Zhang       if (in[l] < 0) continue;
6220880e062SHong Zhang       col = in[l];
6232515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
624e32f2f54SBarry Smith       if (col >= a->nbs) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",col,a->nbs-1);
625b1823623SSatish Balay #endif
626b98bf0e1SJed Brown       if (col < row) {
62726fbe8dcSKarl Rupp         if (a->ignore_ltriangular) continue; /* ignore lower triangular block */
62826fbe8dcSKarl 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)");
629b98bf0e1SJed Brown       }
63026fbe8dcSKarl Rupp       if (roworiented) value = v + k*(stepval+bs)*bs + l*bs;
63126fbe8dcSKarl Rupp       else value = v + l*(stepval+bs)*bs + k*bs;
63226fbe8dcSKarl Rupp 
63326fbe8dcSKarl Rupp       if (col <= lastcol) low = 0;
63426fbe8dcSKarl Rupp       else high = nrow;
63526fbe8dcSKarl Rupp 
636e2ee6c50SBarry Smith       lastcol = col;
6370880e062SHong Zhang       while (high-low > 7) {
6380880e062SHong Zhang         t = (low+high)/2;
6390880e062SHong Zhang         if (rp[t] > col) high = t;
6400880e062SHong Zhang         else             low  = t;
6410880e062SHong Zhang       }
6420880e062SHong Zhang       for (i=low; i<high; i++) {
6430880e062SHong Zhang         if (rp[i] > col) break;
6440880e062SHong Zhang         if (rp[i] == col) {
6450880e062SHong Zhang           bap = ap +  bs2*i;
6460880e062SHong Zhang           if (roworiented) {
6470880e062SHong Zhang             if (is == ADD_VALUES) {
6480880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
6490880e062SHong Zhang                 for (jj=ii; jj<bs2; jj+=bs) {
6500880e062SHong Zhang                   bap[jj] += *value++;
6510880e062SHong Zhang                 }
6520880e062SHong Zhang               }
6530880e062SHong Zhang             } else {
6540880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
6550880e062SHong Zhang                 for (jj=ii; jj<bs2; jj+=bs) {
6560880e062SHong Zhang                   bap[jj] = *value++;
6570880e062SHong Zhang                 }
6580880e062SHong Zhang                }
6590880e062SHong Zhang             }
6600880e062SHong Zhang           } else {
6610880e062SHong Zhang             if (is == ADD_VALUES) {
6620880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
6630880e062SHong Zhang                 for (jj=0; jj<bs; jj++) {
6640880e062SHong Zhang                   *bap++ += *value++;
6650880e062SHong Zhang                 }
6660880e062SHong Zhang               }
6670880e062SHong Zhang             } else {
6680880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
6690880e062SHong Zhang                 for (jj=0; jj<bs; jj++) {
6700880e062SHong Zhang                   *bap++  = *value++;
6710880e062SHong Zhang                 }
6720880e062SHong Zhang               }
6730880e062SHong Zhang             }
6740880e062SHong Zhang           }
6750880e062SHong Zhang           goto noinsert2;
6760880e062SHong Zhang         }
6770880e062SHong Zhang       }
6780880e062SHong Zhang       if (nonew == 1) goto noinsert2;
679e32f2f54SBarry Smith       if (nonew == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col);
680fef13f97SBarry Smith       MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,row,col,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar);
681c03d1d03SSatish Balay       N = nrow++ - 1; high++;
6820880e062SHong Zhang       /* shift up all the later entries in this row */
6830880e062SHong Zhang       for (ii=N; ii>=i; ii--) {
6840880e062SHong Zhang         rp[ii+1] = rp[ii];
6850880e062SHong Zhang         ierr     = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr);
6860880e062SHong Zhang       }
6870880e062SHong Zhang       if (N >= i) {
6880880e062SHong Zhang         ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr);
6890880e062SHong Zhang       }
6900880e062SHong Zhang       rp[i] = col;
6910880e062SHong Zhang       bap   = ap +  bs2*i;
6920880e062SHong Zhang       if (roworiented) {
6930880e062SHong Zhang         for (ii=0; ii<bs; ii++,value+=stepval) {
6940880e062SHong Zhang           for (jj=ii; jj<bs2; jj+=bs) {
6950880e062SHong Zhang             bap[jj] = *value++;
6960880e062SHong Zhang           }
6970880e062SHong Zhang         }
6980880e062SHong Zhang       } else {
6990880e062SHong Zhang         for (ii=0; ii<bs; ii++,value+=stepval) {
7000880e062SHong Zhang           for (jj=0; jj<bs; jj++) {
7010880e062SHong Zhang             *bap++ = *value++;
7020880e062SHong Zhang           }
7030880e062SHong Zhang         }
7040880e062SHong Zhang        }
7050880e062SHong Zhang     noinsert2:;
7060880e062SHong Zhang       low = i;
7070880e062SHong Zhang     }
7080880e062SHong Zhang     ailen[row] = nrow;
7090880e062SHong Zhang   }
7100880e062SHong Zhang   PetscFunctionReturn(0);
71149b5e25fSSatish Balay }
71249b5e25fSSatish Balay 
71364831d72SBarry Smith /*
71464831d72SBarry Smith     This is not yet used
71564831d72SBarry Smith */
7164a2ae208SSatish Balay #undef __FUNCT__
7174108e4d5SBarry Smith #define __FUNCT__ "MatAssemblyEnd_SeqSBAIJ_SeqAIJ_Inode"
7184108e4d5SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ_SeqAIJ_Inode(Mat A)
7190def2e27SBarry Smith {
7200def2e27SBarry Smith   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
7210def2e27SBarry Smith   PetscErrorCode ierr;
7220def2e27SBarry Smith   const PetscInt *ai = a->i, *aj = a->j,*cols;
7230def2e27SBarry Smith   PetscInt       i   = 0,j,blk_size,m = A->rmap->n,node_count = 0,nzx,nzy,*ns,row,nz,cnt,cnt2,*counts;
724ace3abfcSBarry Smith   PetscBool      flag;
7250def2e27SBarry Smith 
7260def2e27SBarry Smith   PetscFunctionBegin;
727785e854fSJed Brown   ierr = PetscMalloc1(m,&ns);CHKERRQ(ierr);
7280def2e27SBarry Smith   while (i < m) {
7290def2e27SBarry Smith     nzx = ai[i+1] - ai[i];       /* Number of nonzeros */
7300def2e27SBarry Smith     /* Limits the number of elements in a node to 'a->inode.limit' */
7310def2e27SBarry Smith     for (j=i+1,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) {
7320def2e27SBarry Smith       nzy = ai[j+1] - ai[j];
7330def2e27SBarry Smith       if (nzy != (nzx - j + i)) break;
7340def2e27SBarry Smith       ierr = PetscMemcmp(aj + ai[i] + j - i,aj + ai[j],nzy*sizeof(PetscInt),&flag);CHKERRQ(ierr);
7350def2e27SBarry Smith       if (!flag) break;
7360def2e27SBarry Smith     }
7370def2e27SBarry Smith     ns[node_count++] = blk_size;
73826fbe8dcSKarl Rupp 
7390def2e27SBarry Smith     i = j;
7400def2e27SBarry Smith   }
7410def2e27SBarry Smith   if (!a->inode.size && m && node_count > .9*m) {
7420def2e27SBarry Smith     ierr = PetscFree(ns);CHKERRQ(ierr);
7430def2e27SBarry Smith     ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr);
7440def2e27SBarry Smith   } else {
7450def2e27SBarry Smith     a->inode.node_count = node_count;
74626fbe8dcSKarl Rupp 
747785e854fSJed Brown     ierr = PetscMalloc1(node_count,&a->inode.size);CHKERRQ(ierr);
7483bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)A,node_count*sizeof(PetscInt));CHKERRQ(ierr);
74922d28d08SBarry Smith     ierr = PetscMemcpy(a->inode.size,ns,node_count*sizeof(PetscInt));CHKERRQ(ierr);
7500def2e27SBarry Smith     ierr = PetscFree(ns);CHKERRQ(ierr);
7510def2e27SBarry Smith     ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr);
7520def2e27SBarry Smith 
7530def2e27SBarry Smith     /* count collections of adjacent columns in each inode */
7540def2e27SBarry Smith     row = 0;
7550def2e27SBarry Smith     cnt = 0;
7560def2e27SBarry Smith     for (i=0; i<node_count; i++) {
7570def2e27SBarry Smith       cols = aj + ai[row] + a->inode.size[i];
7580def2e27SBarry Smith       nz   = ai[row+1] - ai[row] - a->inode.size[i];
7590def2e27SBarry Smith       for (j=1; j<nz; j++) {
76026fbe8dcSKarl Rupp         if (cols[j] != cols[j-1]+1) cnt++;
7610def2e27SBarry Smith       }
7620def2e27SBarry Smith       cnt++;
7630def2e27SBarry Smith       row += a->inode.size[i];
7640def2e27SBarry Smith     }
765785e854fSJed Brown     ierr = PetscMalloc1(2*cnt,&counts);CHKERRQ(ierr);
7660def2e27SBarry Smith     cnt  = 0;
7670def2e27SBarry Smith     row  = 0;
7680def2e27SBarry Smith     for (i=0; i<node_count; i++) {
7690def2e27SBarry Smith       cols = aj + ai[row] + a->inode.size[i];
7700def2e27SBarry Smith       counts[2*cnt] = cols[0];
7710def2e27SBarry Smith       nz   = ai[row+1] - ai[row] - a->inode.size[i];
7720def2e27SBarry Smith       cnt2 = 1;
7730def2e27SBarry Smith       for (j=1; j<nz; j++) {
7740def2e27SBarry Smith         if (cols[j] != cols[j-1]+1) {
7750def2e27SBarry Smith           counts[2*(cnt++)+1] = cnt2;
7760def2e27SBarry Smith           counts[2*cnt]       = cols[j];
7770def2e27SBarry Smith           cnt2 = 1;
7780def2e27SBarry Smith         } else cnt2++;
7790def2e27SBarry Smith       }
7800def2e27SBarry Smith       counts[2*(cnt++)+1] = cnt2;
7810def2e27SBarry Smith       row += a->inode.size[i];
7820def2e27SBarry Smith     }
78322d28d08SBarry Smith     ierr = PetscIntView(2*cnt,counts,0);CHKERRQ(ierr);
7840def2e27SBarry Smith   }
78538702af4SBarry Smith   PetscFunctionReturn(0);
78638702af4SBarry Smith }
78738702af4SBarry Smith 
78838702af4SBarry Smith #undef __FUNCT__
7894a2ae208SSatish Balay #define __FUNCT__ "MatAssemblyEnd_SeqSBAIJ"
790dfbe8321SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ(Mat A,MatAssemblyType mode)
79149b5e25fSSatish Balay {
79249b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
7936849ba73SBarry Smith   PetscErrorCode ierr;
79413f74950SBarry Smith   PetscInt       fshift = 0,i,j,*ai = a->i,*aj = a->j,*imax = a->imax;
795d0f46423SBarry Smith   PetscInt       m      = A->rmap->N,*ip,N,*ailen = a->ilen;
79613f74950SBarry Smith   PetscInt       mbs    = a->mbs,bs2 = a->bs2,rmax = 0;
79749b5e25fSSatish Balay   MatScalar      *aa    = a->a,*ap;
79849b5e25fSSatish Balay 
79949b5e25fSSatish Balay   PetscFunctionBegin;
80049b5e25fSSatish Balay   if (mode == MAT_FLUSH_ASSEMBLY) PetscFunctionReturn(0);
80149b5e25fSSatish Balay 
80249b5e25fSSatish Balay   if (m) rmax = ailen[0];
80349b5e25fSSatish Balay   for (i=1; i<mbs; i++) {
80449b5e25fSSatish Balay     /* move each row back by the amount of empty slots (fshift) before it*/
80549b5e25fSSatish Balay     fshift += imax[i-1] - ailen[i-1];
80649b5e25fSSatish Balay     rmax    = PetscMax(rmax,ailen[i]);
80749b5e25fSSatish Balay     if (fshift) {
80849b5e25fSSatish Balay       ip = aj + ai[i]; ap = aa + bs2*ai[i];
80949b5e25fSSatish Balay       N  = ailen[i];
81049b5e25fSSatish Balay       for (j=0; j<N; j++) {
81149b5e25fSSatish Balay         ip[j-fshift] = ip[j];
81249b5e25fSSatish Balay         ierr         = PetscMemcpy(ap+(j-fshift)*bs2,ap+j*bs2,bs2*sizeof(MatScalar));CHKERRQ(ierr);
81349b5e25fSSatish Balay       }
81449b5e25fSSatish Balay     }
81549b5e25fSSatish Balay     ai[i] = ai[i-1] + ailen[i-1];
81649b5e25fSSatish Balay   }
81749b5e25fSSatish Balay   if (mbs) {
81849b5e25fSSatish Balay     fshift += imax[mbs-1] - ailen[mbs-1];
81949b5e25fSSatish Balay     ai[mbs] = ai[mbs-1] + ailen[mbs-1];
82049b5e25fSSatish Balay   }
82149b5e25fSSatish Balay   /* reset ilen and imax for each row */
82249b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
82349b5e25fSSatish Balay     ailen[i] = imax[i] = ai[i+1] - ai[i];
82449b5e25fSSatish Balay   }
8256c6c5352SBarry Smith   a->nz = ai[mbs];
82649b5e25fSSatish Balay 
827b424e231SHong Zhang   /* diagonals may have moved, reset it */
828b424e231SHong Zhang   if (a->diag) {
8292ed38d0bSJed Brown     ierr = PetscMemcpy(a->diag,ai,mbs*sizeof(PetscInt));CHKERRQ(ierr);
83049b5e25fSSatish Balay   }
83126fbe8dcSKarl 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);
83226fbe8dcSKarl Rupp 
833d0f46423SBarry 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);
834ae15b995SBarry Smith   ierr = PetscInfo1(A,"Number of mallocs during MatSetValues is %D\n",a->reallocs);CHKERRQ(ierr);
835ae15b995SBarry Smith   ierr = PetscInfo1(A,"Most nonzeros blocks in any row is %D\n",rmax);CHKERRQ(ierr);
83626fbe8dcSKarl Rupp 
8378e58a170SBarry Smith   A->info.mallocs    += a->reallocs;
83849b5e25fSSatish Balay   a->reallocs         = 0;
83949b5e25fSSatish Balay   A->info.nz_unneeded = (PetscReal)fshift*bs2;
840061b2667SBarry Smith   a->idiagvalid       = PETSC_FALSE;
84138702af4SBarry Smith 
84238702af4SBarry Smith   if (A->cmap->n < 65536 && A->cmap->bs == 1) {
84344e1c64aSLisandro Dalcin     if (a->jshort && a->free_jshort) {
84417803ae8SHong Zhang       /* when matrix data structure is changed, previous jshort must be replaced */
84517803ae8SHong Zhang       ierr = PetscFree(a->jshort);CHKERRQ(ierr);
84617803ae8SHong Zhang     }
847785e854fSJed Brown     ierr = PetscMalloc1(a->i[A->rmap->n],&a->jshort);CHKERRQ(ierr);
8483bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)A,a->i[A->rmap->n]*sizeof(unsigned short));CHKERRQ(ierr);
84938702af4SBarry Smith     for (i=0; i<a->i[A->rmap->n]; i++) a->jshort[i] = a->j[i];
85038702af4SBarry Smith     A->ops->mult   = MatMult_SeqSBAIJ_1_ushort;
85141f059aeSBarry Smith     A->ops->sor    = MatSOR_SeqSBAIJ_ushort;
8524da8f245SBarry Smith     a->free_jshort = PETSC_TRUE;
85338702af4SBarry Smith   }
85449b5e25fSSatish Balay   PetscFunctionReturn(0);
85549b5e25fSSatish Balay }
85649b5e25fSSatish Balay 
85749b5e25fSSatish Balay /*
85849b5e25fSSatish Balay    This function returns an array of flags which indicate the locations of contiguous
85949b5e25fSSatish Balay    blocks that should be zeroed. for eg: if bs = 3  and is = [0,1,2,3,5,6,7,8,9]
86049b5e25fSSatish Balay    then the resulting sizes = [3,1,1,3,1] correspondig to sets [(0,1,2),(3),(5),(6,7,8),(9)]
86149b5e25fSSatish Balay    Assume: sizes should be long enough to hold all the values.
86249b5e25fSSatish Balay */
8634a2ae208SSatish Balay #undef __FUNCT__
8644a2ae208SSatish Balay #define __FUNCT__ "MatZeroRows_SeqSBAIJ_Check_Blocks"
86513f74950SBarry Smith PetscErrorCode MatZeroRows_SeqSBAIJ_Check_Blocks(PetscInt idx[],PetscInt n,PetscInt bs,PetscInt sizes[], PetscInt *bs_max)
86649b5e25fSSatish Balay {
86713f74950SBarry Smith   PetscInt  i,j,k,row;
868ace3abfcSBarry Smith   PetscBool flg;
86949b5e25fSSatish Balay 
87049b5e25fSSatish Balay   PetscFunctionBegin;
87149b5e25fSSatish Balay   for (i=0,j=0; i<n; j++) {
87249b5e25fSSatish Balay     row = idx[i];
87349b5e25fSSatish Balay     if (row%bs!=0) { /* Not the begining of a block */
87449b5e25fSSatish Balay       sizes[j] = 1;
87549b5e25fSSatish Balay       i++;
87649b5e25fSSatish Balay     } else if (i+bs > n) { /* Beginning of a block, but complete block doesn't exist (at idx end) */
87749b5e25fSSatish Balay       sizes[j] = 1;         /* Also makes sure atleast 'bs' values exist for next else */
87849b5e25fSSatish Balay       i++;
87949b5e25fSSatish Balay     } else { /* Begining of the block, so check if the complete block exists */
88049b5e25fSSatish Balay       flg = PETSC_TRUE;
88149b5e25fSSatish Balay       for (k=1; k<bs; k++) {
88249b5e25fSSatish Balay         if (row+k != idx[i+k]) { /* break in the block */
88349b5e25fSSatish Balay           flg = PETSC_FALSE;
88449b5e25fSSatish Balay           break;
88549b5e25fSSatish Balay         }
88649b5e25fSSatish Balay       }
887abc0a331SBarry Smith       if (flg) { /* No break in the bs */
88849b5e25fSSatish Balay         sizes[j] = bs;
88949b5e25fSSatish Balay         i       += bs;
89049b5e25fSSatish Balay       } else {
89149b5e25fSSatish Balay         sizes[j] = 1;
89249b5e25fSSatish Balay         i++;
89349b5e25fSSatish Balay       }
89449b5e25fSSatish Balay     }
89549b5e25fSSatish Balay   }
89649b5e25fSSatish Balay   *bs_max = j;
89749b5e25fSSatish Balay   PetscFunctionReturn(0);
89849b5e25fSSatish Balay }
89949b5e25fSSatish Balay 
90049b5e25fSSatish Balay 
90149b5e25fSSatish Balay /* Only add/insert a(i,j) with i<=j (blocks).
90249b5e25fSSatish Balay    Any a(i,j) with i>j input by user is ingored.
90349b5e25fSSatish Balay */
90449b5e25fSSatish Balay 
9054a2ae208SSatish Balay #undef __FUNCT__
9064a2ae208SSatish Balay #define __FUNCT__ "MatSetValues_SeqSBAIJ"
90713f74950SBarry Smith PetscErrorCode MatSetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is)
90849b5e25fSSatish Balay {
90949b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
9106849ba73SBarry Smith   PetscErrorCode ierr;
911e2ee6c50SBarry Smith   PetscInt       *rp,k,low,high,t,ii,row,nrow,i,col,l,rmax,N,lastcol = -1;
91213f74950SBarry Smith   PetscInt       *imax=a->imax,*ai=a->i,*ailen=a->ilen,roworiented=a->roworiented;
913d0f46423SBarry Smith   PetscInt       *aj  =a->j,nonew=a->nonew,bs=A->rmap->bs,brow,bcol;
91413f74950SBarry Smith   PetscInt       ridx,cidx,bs2=a->bs2;
91549b5e25fSSatish Balay   MatScalar      *ap,value,*aa=a->a,*bap;
91649b5e25fSSatish Balay 
91749b5e25fSSatish Balay   PetscFunctionBegin;
91849b5e25fSSatish Balay   for (k=0; k<m; k++) { /* loop over added rows */
91949b5e25fSSatish Balay     row  = im[k];       /* row number */
92049b5e25fSSatish Balay     brow = row/bs;      /* block row number */
92149b5e25fSSatish Balay     if (row < 0) continue;
9222515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
923e32f2f54SBarry 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);
92449b5e25fSSatish Balay #endif
92549b5e25fSSatish Balay     rp   = aj + ai[brow]; /*ptr to beginning of column value of the row block*/
92649b5e25fSSatish Balay     ap   = aa + bs2*ai[brow]; /*ptr to beginning of element value of the row block*/
92749b5e25fSSatish Balay     rmax = imax[brow];  /* maximum space allocated for this row */
92849b5e25fSSatish Balay     nrow = ailen[brow]; /* actual length of this row */
92949b5e25fSSatish Balay     low  = 0;
93049b5e25fSSatish Balay 
93149b5e25fSSatish Balay     for (l=0; l<n; l++) { /* loop over added columns */
93249b5e25fSSatish Balay       if (in[l] < 0) continue;
9332515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
934e32f2f54SBarry 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);
93549b5e25fSSatish Balay #endif
93649b5e25fSSatish Balay       col  = in[l];
93749b5e25fSSatish Balay       bcol = col/bs;              /* block col number */
93849b5e25fSSatish Balay 
939941593c8SHong Zhang       if (brow > bcol) {
94026fbe8dcSKarl Rupp         if (a->ignore_ltriangular) continue; /* ignore lower triangular values */
94126fbe8dcSKarl 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)");
942941593c8SHong Zhang       }
943f4989cb3SHong Zhang 
94449b5e25fSSatish Balay       ridx = row % bs; cidx = col % bs; /*row and col index inside the block */
9458549e402SHong Zhang       if ((brow==bcol && ridx<=cidx) || (brow<bcol)) {
94649b5e25fSSatish Balay         /* element value a(k,l) */
94726fbe8dcSKarl Rupp         if (roworiented) value = v[l + k*n];
94826fbe8dcSKarl Rupp         else value = v[k + l*m];
94949b5e25fSSatish Balay 
95049b5e25fSSatish Balay         /* move pointer bap to a(k,l) quickly and add/insert value */
95126fbe8dcSKarl Rupp         if (col <= lastcol) low = 0;
95226fbe8dcSKarl Rupp         high = nrow;
953e2ee6c50SBarry Smith         lastcol = col;
95449b5e25fSSatish Balay         while (high-low > 7) {
95549b5e25fSSatish Balay           t = (low+high)/2;
95649b5e25fSSatish Balay           if (rp[t] > bcol) high = t;
95749b5e25fSSatish Balay           else              low  = t;
95849b5e25fSSatish Balay         }
95949b5e25fSSatish Balay         for (i=low; i<high; i++) {
96049b5e25fSSatish Balay           if (rp[i] > bcol) break;
96149b5e25fSSatish Balay           if (rp[i] == bcol) {
96249b5e25fSSatish Balay             bap = ap +  bs2*i + bs*cidx + ridx;
96349b5e25fSSatish Balay             if (is == ADD_VALUES) *bap += value;
96449b5e25fSSatish Balay             else                  *bap  = value;
9658549e402SHong Zhang             /* for diag block, add/insert its symmetric element a(cidx,ridx) */
9668549e402SHong Zhang             if (brow == bcol && ridx < cidx) {
9678549e402SHong Zhang               bap = ap +  bs2*i + bs*ridx + cidx;
9688549e402SHong Zhang               if (is == ADD_VALUES) *bap += value;
9698549e402SHong Zhang               else                  *bap  = value;
9708549e402SHong Zhang             }
97149b5e25fSSatish Balay             goto noinsert1;
97249b5e25fSSatish Balay           }
97349b5e25fSSatish Balay         }
97449b5e25fSSatish Balay 
97549b5e25fSSatish Balay         if (nonew == 1) goto noinsert1;
976e32f2f54SBarry Smith         if (nonew == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col);
977fef13f97SBarry Smith         MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,brow,bcol,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar);
97849b5e25fSSatish Balay 
979c03d1d03SSatish Balay         N = nrow++ - 1; high++;
98049b5e25fSSatish Balay         /* shift up all the later entries in this row */
98149b5e25fSSatish Balay         for (ii=N; ii>=i; ii--) {
98249b5e25fSSatish Balay           rp[ii+1] = rp[ii];
98349b5e25fSSatish Balay           ierr     = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr);
98449b5e25fSSatish Balay         }
98549b5e25fSSatish Balay         if (N>=i) {
98649b5e25fSSatish Balay           ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr);
98749b5e25fSSatish Balay         }
98849b5e25fSSatish Balay         rp[i]                      = bcol;
98949b5e25fSSatish Balay         ap[bs2*i + bs*cidx + ridx] = value;
990e56f5c9eSBarry Smith         A->nonzerostate++;
99149b5e25fSSatish Balay noinsert1:;
99249b5e25fSSatish Balay         low = i;
9938549e402SHong Zhang       }
99449b5e25fSSatish Balay     }   /* end of loop over added columns */
99549b5e25fSSatish Balay     ailen[brow] = nrow;
99649b5e25fSSatish Balay   }   /* end of loop over added rows */
99749b5e25fSSatish Balay   PetscFunctionReturn(0);
99849b5e25fSSatish Balay }
99949b5e25fSSatish Balay 
10004a2ae208SSatish Balay #undef __FUNCT__
10014d101231SSatish Balay #define __FUNCT__ "MatICCFactor_SeqSBAIJ"
10020481f469SBarry Smith PetscErrorCode MatICCFactor_SeqSBAIJ(Mat inA,IS row,const MatFactorInfo *info)
100349b5e25fSSatish Balay {
10044ccecd49SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)inA->data;
100549b5e25fSSatish Balay   Mat            outA;
1006dfbe8321SBarry Smith   PetscErrorCode ierr;
1007ace3abfcSBarry Smith   PetscBool      row_identity;
100849b5e25fSSatish Balay 
100949b5e25fSSatish Balay   PetscFunctionBegin;
1010e32f2f54SBarry Smith   if (info->levels != 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only levels=0 is supported for in-place icc");
1011c84f5b01SHong Zhang   ierr = ISIdentity(row,&row_identity);CHKERRQ(ierr);
1012e32f2f54SBarry Smith   if (!row_identity) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix reordering is not supported");
1013e32f2f54SBarry 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()! */
1014c84f5b01SHong Zhang 
101549b5e25fSSatish Balay   outA            = inA;
1016d5f3da31SBarry Smith   inA->factortype = MAT_FACTOR_ICC;
101749b5e25fSSatish Balay 
10181a3463dfSHong Zhang   ierr = MatMarkDiagonal_SeqSBAIJ(inA);CHKERRQ(ierr);
1019d595f711SHong Zhang   ierr = MatSeqSBAIJSetNumericFactorization_inplace(inA,row_identity);CHKERRQ(ierr);
102049b5e25fSSatish Balay 
1021c3122656SLisandro Dalcin   ierr   = PetscObjectReference((PetscObject)row);CHKERRQ(ierr);
10226bf464f9SBarry Smith   ierr   = ISDestroy(&a->row);CHKERRQ(ierr);
1023c84f5b01SHong Zhang   a->row = row;
1024c3122656SLisandro Dalcin   ierr   = PetscObjectReference((PetscObject)row);CHKERRQ(ierr);
10256bf464f9SBarry Smith   ierr   = ISDestroy(&a->col);CHKERRQ(ierr);
1026c84f5b01SHong Zhang   a->col = row;
1027c84f5b01SHong Zhang 
1028c84f5b01SHong Zhang   /* Create the invert permutation so that it can be used in MatCholeskyFactorNumeric() */
1029c84f5b01SHong Zhang   if (a->icol) {ierr = ISInvertPermutation(row,PETSC_DECIDE, &a->icol);CHKERRQ(ierr);}
10303bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)inA,(PetscObject)a->icol);CHKERRQ(ierr);
103149b5e25fSSatish Balay 
103249b5e25fSSatish Balay   if (!a->solve_work) {
1033785e854fSJed Brown     ierr = PetscMalloc1((inA->rmap->N+inA->rmap->bs),&a->solve_work);CHKERRQ(ierr);
10343bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)inA,(inA->rmap->N+inA->rmap->bs)*sizeof(PetscScalar));CHKERRQ(ierr);
103549b5e25fSSatish Balay   }
103649b5e25fSSatish Balay 
1037719d5645SBarry Smith   ierr = MatCholeskyFactorNumeric(outA,inA,info);CHKERRQ(ierr);
103849b5e25fSSatish Balay   PetscFunctionReturn(0);
103949b5e25fSSatish Balay }
1040950f1e5bSHong Zhang 
10414a2ae208SSatish Balay #undef __FUNCT__
10424a2ae208SSatish Balay #define __FUNCT__ "MatSeqSBAIJSetColumnIndices_SeqSBAIJ"
10437087cfbeSBarry Smith PetscErrorCode  MatSeqSBAIJSetColumnIndices_SeqSBAIJ(Mat mat,PetscInt *indices)
104449b5e25fSSatish Balay {
1045045c9aa0SHong Zhang   Mat_SeqSBAIJ   *baij = (Mat_SeqSBAIJ*)mat->data;
104613f74950SBarry Smith   PetscInt       i,nz,n;
10477827cd58SJed Brown   PetscErrorCode ierr;
104849b5e25fSSatish Balay 
104949b5e25fSSatish Balay   PetscFunctionBegin;
10506c6c5352SBarry Smith   nz = baij->maxnz;
1051d0f46423SBarry Smith   n  = mat->cmap->n;
105226fbe8dcSKarl Rupp   for (i=0; i<nz; i++) baij->j[i] = indices[i];
105326fbe8dcSKarl Rupp 
10546c6c5352SBarry Smith   baij->nz = nz;
105526fbe8dcSKarl Rupp   for (i=0; i<n; i++) baij->ilen[i] = baij->imax[i];
105626fbe8dcSKarl Rupp 
10577827cd58SJed Brown   ierr = MatSetOption(mat,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
105849b5e25fSSatish Balay   PetscFunctionReturn(0);
105949b5e25fSSatish Balay }
106049b5e25fSSatish Balay 
10614a2ae208SSatish Balay #undef __FUNCT__
10624a2ae208SSatish Balay #define __FUNCT__ "MatSeqSBAIJSetColumnIndices"
106349b5e25fSSatish Balay /*@
106419585528SSatish Balay   MatSeqSBAIJSetColumnIndices - Set the column indices for all the rows
106549b5e25fSSatish Balay   in the matrix.
106649b5e25fSSatish Balay 
106749b5e25fSSatish Balay   Input Parameters:
106819585528SSatish Balay   +  mat     - the SeqSBAIJ matrix
106949b5e25fSSatish Balay   -  indices - the column indices
107049b5e25fSSatish Balay 
107149b5e25fSSatish Balay   Level: advanced
107249b5e25fSSatish Balay 
107349b5e25fSSatish Balay   Notes:
107449b5e25fSSatish Balay   This can be called if you have precomputed the nonzero structure of the
107549b5e25fSSatish Balay   matrix and want to provide it to the matrix object to improve the performance
107649b5e25fSSatish Balay   of the MatSetValues() operation.
107749b5e25fSSatish Balay 
107849b5e25fSSatish Balay   You MUST have set the correct numbers of nonzeros per row in the call to
1079d1be2dadSMatthew Knepley   MatCreateSeqSBAIJ(), and the columns indices MUST be sorted.
108049b5e25fSSatish Balay 
1081ab9f2c04SSatish Balay   MUST be called before any calls to MatSetValues()
108249b5e25fSSatish Balay 
1083ab9f2c04SSatish Balay   .seealso: MatCreateSeqSBAIJ
108449b5e25fSSatish Balay @*/
10857087cfbeSBarry Smith PetscErrorCode  MatSeqSBAIJSetColumnIndices(Mat mat,PetscInt *indices)
108649b5e25fSSatish Balay {
10874ac538c5SBarry Smith   PetscErrorCode ierr;
108849b5e25fSSatish Balay 
108949b5e25fSSatish Balay   PetscFunctionBegin;
10900700a824SBarry Smith   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
10914482741eSBarry Smith   PetscValidPointer(indices,2);
10924ac538c5SBarry Smith   ierr = PetscUseMethod(mat,"MatSeqSBAIJSetColumnIndices_C",(Mat,PetscInt*),(mat,indices));CHKERRQ(ierr);
109349b5e25fSSatish Balay   PetscFunctionReturn(0);
109449b5e25fSSatish Balay }
109549b5e25fSSatish Balay 
10964a2ae208SSatish Balay #undef __FUNCT__
10973c896bc6SHong Zhang #define __FUNCT__ "MatCopy_SeqSBAIJ"
10983c896bc6SHong Zhang PetscErrorCode MatCopy_SeqSBAIJ(Mat A,Mat B,MatStructure str)
10993c896bc6SHong Zhang {
11003c896bc6SHong Zhang   PetscErrorCode ierr;
11013c896bc6SHong Zhang 
11023c896bc6SHong Zhang   PetscFunctionBegin;
11033c896bc6SHong Zhang   /* If the two matrices have the same copy implementation, use fast copy. */
11043c896bc6SHong Zhang   if (str == SAME_NONZERO_PATTERN && (A->ops->copy == B->ops->copy)) {
11053c896bc6SHong Zhang     Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
11063c896bc6SHong Zhang     Mat_SeqSBAIJ *b = (Mat_SeqSBAIJ*)B->data;
11073c896bc6SHong Zhang 
1108e7e72b3dSBarry 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");
1109d0f46423SBarry Smith     ierr = PetscMemcpy(b->a,a->a,(a->i[A->rmap->N])*sizeof(PetscScalar));CHKERRQ(ierr);
11103c896bc6SHong Zhang   } else {
1111f5edf698SHong Zhang     ierr = MatGetRowUpperTriangular(A);CHKERRQ(ierr);
11123c896bc6SHong Zhang     ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr);
1113f5edf698SHong Zhang     ierr = MatRestoreRowUpperTriangular(A);CHKERRQ(ierr);
11143c896bc6SHong Zhang   }
11153c896bc6SHong Zhang   PetscFunctionReturn(0);
11163c896bc6SHong Zhang }
11173c896bc6SHong Zhang 
11183c896bc6SHong Zhang #undef __FUNCT__
11194994cf47SJed Brown #define __FUNCT__ "MatSetUp_SeqSBAIJ"
11204994cf47SJed Brown PetscErrorCode MatSetUp_SeqSBAIJ(Mat A)
1121273d9f13SBarry Smith {
1122dfbe8321SBarry Smith   PetscErrorCode ierr;
1123273d9f13SBarry Smith 
1124273d9f13SBarry Smith   PetscFunctionBegin;
1125535b19f3SBarry Smith   ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(A,A->rmap->bs,PETSC_DEFAULT,0);CHKERRQ(ierr);
1126273d9f13SBarry Smith   PetscFunctionReturn(0);
1127273d9f13SBarry Smith }
1128273d9f13SBarry Smith 
1129a6ece127SHong Zhang #undef __FUNCT__
11308c778c55SBarry Smith #define __FUNCT__ "MatSeqSBAIJGetArray_SeqSBAIJ"
11318c778c55SBarry Smith PetscErrorCode MatSeqSBAIJGetArray_SeqSBAIJ(Mat A,PetscScalar *array[])
1132a6ece127SHong Zhang {
1133a6ece127SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
11345fd66863SKarl Rupp 
1135a6ece127SHong Zhang   PetscFunctionBegin;
1136a6ece127SHong Zhang   *array = a->a;
1137a6ece127SHong Zhang   PetscFunctionReturn(0);
1138a6ece127SHong Zhang }
1139a6ece127SHong Zhang 
1140a6ece127SHong Zhang #undef __FUNCT__
11418c778c55SBarry Smith #define __FUNCT__ "MatSeqSBAIJRestoreArray_SeqSBAIJ"
11428c778c55SBarry Smith PetscErrorCode MatSeqSBAIJRestoreArray_SeqSBAIJ(Mat A,PetscScalar *array[])
1143a6ece127SHong Zhang {
1144a6ece127SHong Zhang   PetscFunctionBegin;
1145a6ece127SHong Zhang   PetscFunctionReturn(0);
1146a6ece127SHong Zhang }
1147a6ece127SHong Zhang 
114842ee4b1aSHong Zhang #undef __FUNCT__
114952768537SHong Zhang #define __FUNCT__ "MatAXPYGetPreallocation_SeqSBAIJ"
115052768537SHong Zhang PetscErrorCode MatAXPYGetPreallocation_SeqSBAIJ(Mat Y,Mat X,PetscInt *nnz)
115152768537SHong Zhang {
1152b264fe52SHong Zhang   PetscInt       bs = Y->rmap->bs,mbs = Y->rmap->N/bs;
115352768537SHong Zhang   Mat_SeqSBAIJ   *x = (Mat_SeqSBAIJ*)X->data;
115452768537SHong Zhang   Mat_SeqSBAIJ   *y = (Mat_SeqSBAIJ*)Y->data;
1155b264fe52SHong Zhang   PetscErrorCode ierr;
115652768537SHong Zhang 
115752768537SHong Zhang   PetscFunctionBegin;
115852768537SHong Zhang   /* Set the number of nonzeros in the new matrix */
1159b264fe52SHong Zhang   ierr = MatAXPYGetPreallocation_SeqX_private(mbs,x->i,x->j,y->i,y->j,nnz);CHKERRQ(ierr);
116052768537SHong Zhang   PetscFunctionReturn(0);
116152768537SHong Zhang }
116252768537SHong Zhang 
116352768537SHong Zhang #undef __FUNCT__
116442ee4b1aSHong Zhang #define __FUNCT__ "MatAXPY_SeqSBAIJ"
1165f4df32b1SMatthew Knepley PetscErrorCode MatAXPY_SeqSBAIJ(Mat Y,PetscScalar a,Mat X,MatStructure str)
116642ee4b1aSHong Zhang {
116742ee4b1aSHong Zhang   Mat_SeqSBAIJ   *x=(Mat_SeqSBAIJ*)X->data, *y=(Mat_SeqSBAIJ*)Y->data;
1168dfbe8321SBarry Smith   PetscErrorCode ierr;
1169*7b2ba04bSHong Zhang   PetscInt       bs=Y->rmap->bs,bs2=bs*bs;
1170e838b9e7SJed Brown   PetscBLASInt   one = 1;
117142ee4b1aSHong Zhang 
117242ee4b1aSHong Zhang   PetscFunctionBegin;
117342ee4b1aSHong Zhang   if (str == SAME_NONZERO_PATTERN) {
1174f4df32b1SMatthew Knepley     PetscScalar  alpha = a;
1175c5df96a5SBarry Smith     PetscBLASInt bnz;
1176c5df96a5SBarry Smith     ierr = PetscBLASIntCast(x->nz*bs2,&bnz);CHKERRQ(ierr);
11778b83055fSJed Brown     PetscStackCallBLAS("BLASaxpy",BLASaxpy_(&bnz,&alpha,x->a,&one,y->a,&one));
1178a3fa217bSJose E. Roman     ierr = PetscObjectStateIncrease((PetscObject)Y);CHKERRQ(ierr);
1179c537a176SHong Zhang   } else if (str == SUBSET_NONZERO_PATTERN) { /* nonzeros of X is a subset of Y's */
1180*7b2ba04bSHong Zhang     ierr = MatSetOption(X,MAT_GETROW_UPPERTRIANGULAR,PETSC_TRUE);CHKERRQ(ierr);
1181*7b2ba04bSHong Zhang     ierr = MatAXPY_Basic(Y,a,X,str);CHKERRQ(ierr);
1182*7b2ba04bSHong Zhang     ierr = MatSetOption(X,MAT_GETROW_UPPERTRIANGULAR,PETSC_FALSE);CHKERRQ(ierr);
118342ee4b1aSHong Zhang   } else {
118452768537SHong Zhang     Mat      B;
118552768537SHong Zhang     PetscInt *nnz;
118652768537SHong Zhang     if (bs != X->rmap->bs) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Matrices must have same block size");
1187f5edf698SHong Zhang     ierr = MatGetRowUpperTriangular(X);CHKERRQ(ierr);
118852768537SHong Zhang     ierr = MatGetRowUpperTriangular(Y);CHKERRQ(ierr);
118952768537SHong Zhang     ierr = PetscMalloc1(Y->rmap->N,&nnz);CHKERRQ(ierr);
119052768537SHong Zhang     ierr = MatCreate(PetscObjectComm((PetscObject)Y),&B);CHKERRQ(ierr);
119152768537SHong Zhang     ierr = PetscObjectSetName((PetscObject)B,((PetscObject)Y)->name);CHKERRQ(ierr);
119252768537SHong Zhang     ierr = MatSetSizes(B,Y->rmap->n,Y->cmap->n,Y->rmap->N,Y->cmap->N);CHKERRQ(ierr);
119352768537SHong Zhang     ierr = MatSetBlockSizesFromMats(B,Y,Y);CHKERRQ(ierr);
119452768537SHong Zhang     ierr = MatSetType(B,(MatType) ((PetscObject)Y)->type_name);CHKERRQ(ierr);
119552768537SHong Zhang     ierr = MatAXPYGetPreallocation_SeqSBAIJ(Y,X,nnz);CHKERRQ(ierr);
119652768537SHong Zhang     ierr = MatSeqSBAIJSetPreallocation(B,bs,0,nnz);CHKERRQ(ierr);
119752768537SHong Zhang 
119852768537SHong Zhang     ierr = MatAXPY_BasicWithPreallocation(B,Y,a,X,str);CHKERRQ(ierr);
119952768537SHong Zhang 
120052768537SHong Zhang     ierr = MatHeaderReplace(Y,B);CHKERRQ(ierr);
120152768537SHong Zhang     ierr = PetscFree(nnz);CHKERRQ(ierr);
1202f5edf698SHong Zhang     ierr = MatRestoreRowUpperTriangular(X);CHKERRQ(ierr);
120352768537SHong Zhang     ierr = MatRestoreRowUpperTriangular(Y);CHKERRQ(ierr);
120442ee4b1aSHong Zhang   }
120542ee4b1aSHong Zhang   PetscFunctionReturn(0);
120642ee4b1aSHong Zhang }
120742ee4b1aSHong Zhang 
1208efcf0fc3SBarry Smith #undef __FUNCT__
1209efcf0fc3SBarry Smith #define __FUNCT__ "MatIsSymmetric_SeqSBAIJ"
1210ace3abfcSBarry Smith PetscErrorCode MatIsSymmetric_SeqSBAIJ(Mat A,PetscReal tol,PetscBool  *flg)
1211efcf0fc3SBarry Smith {
1212efcf0fc3SBarry Smith   PetscFunctionBegin;
1213efcf0fc3SBarry Smith   *flg = PETSC_TRUE;
1214efcf0fc3SBarry Smith   PetscFunctionReturn(0);
1215efcf0fc3SBarry Smith }
1216efcf0fc3SBarry Smith 
1217efcf0fc3SBarry Smith #undef __FUNCT__
1218efcf0fc3SBarry Smith #define __FUNCT__ "MatIsStructurallySymmetric_SeqSBAIJ"
1219ace3abfcSBarry Smith PetscErrorCode MatIsStructurallySymmetric_SeqSBAIJ(Mat A,PetscBool  *flg)
1220efcf0fc3SBarry Smith {
1221efcf0fc3SBarry Smith   PetscFunctionBegin;
1222efcf0fc3SBarry Smith   *flg = PETSC_TRUE;
1223efcf0fc3SBarry Smith   PetscFunctionReturn(0);
1224efcf0fc3SBarry Smith }
1225efcf0fc3SBarry Smith 
1226efcf0fc3SBarry Smith #undef __FUNCT__
1227efcf0fc3SBarry Smith #define __FUNCT__ "MatIsHermitian_SeqSBAIJ"
1228ace3abfcSBarry Smith PetscErrorCode MatIsHermitian_SeqSBAIJ(Mat A,PetscReal tol,PetscBool  *flg)
1229efcf0fc3SBarry Smith {
1230efcf0fc3SBarry Smith   PetscFunctionBegin;
1231efcf0fc3SBarry Smith   *flg = PETSC_FALSE;
1232efcf0fc3SBarry Smith   PetscFunctionReturn(0);
1233efcf0fc3SBarry Smith }
1234efcf0fc3SBarry Smith 
123599cafbc1SBarry Smith #undef __FUNCT__
123699cafbc1SBarry Smith #define __FUNCT__ "MatRealPart_SeqSBAIJ"
123799cafbc1SBarry Smith PetscErrorCode MatRealPart_SeqSBAIJ(Mat A)
123899cafbc1SBarry Smith {
123999cafbc1SBarry Smith   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
124099cafbc1SBarry Smith   PetscInt     i,nz = a->bs2*a->i[a->mbs];
1241dd6ea824SBarry Smith   MatScalar    *aa = a->a;
124299cafbc1SBarry Smith 
124399cafbc1SBarry Smith   PetscFunctionBegin;
124499cafbc1SBarry Smith   for (i=0; i<nz; i++) aa[i] = PetscRealPart(aa[i]);
124599cafbc1SBarry Smith   PetscFunctionReturn(0);
124699cafbc1SBarry Smith }
124799cafbc1SBarry Smith 
124899cafbc1SBarry Smith #undef __FUNCT__
124999cafbc1SBarry Smith #define __FUNCT__ "MatImaginaryPart_SeqSBAIJ"
125099cafbc1SBarry Smith PetscErrorCode MatImaginaryPart_SeqSBAIJ(Mat A)
125199cafbc1SBarry Smith {
125299cafbc1SBarry Smith   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
125399cafbc1SBarry Smith   PetscInt     i,nz = a->bs2*a->i[a->mbs];
1254dd6ea824SBarry Smith   MatScalar    *aa = a->a;
125599cafbc1SBarry Smith 
125699cafbc1SBarry Smith   PetscFunctionBegin;
125799cafbc1SBarry Smith   for (i=0; i<nz; i++) aa[i] = PetscImaginaryPart(aa[i]);
125899cafbc1SBarry Smith   PetscFunctionReturn(0);
125999cafbc1SBarry Smith }
126099cafbc1SBarry Smith 
12613bededecSBarry Smith #undef __FUNCT__
12623bededecSBarry Smith #define __FUNCT__ "MatZeroRowsColumns_SeqSBAIJ"
12633bededecSBarry Smith PetscErrorCode MatZeroRowsColumns_SeqSBAIJ(Mat A,PetscInt is_n,const PetscInt is_idx[],PetscScalar diag,Vec x, Vec b)
12643bededecSBarry Smith {
12653bededecSBarry Smith   Mat_SeqSBAIJ      *baij=(Mat_SeqSBAIJ*)A->data;
12663bededecSBarry Smith   PetscErrorCode    ierr;
12673bededecSBarry Smith   PetscInt          i,j,k,count;
12683bededecSBarry Smith   PetscInt          bs   =A->rmap->bs,bs2=baij->bs2,row,col;
12693bededecSBarry Smith   PetscScalar       zero = 0.0;
12703bededecSBarry Smith   MatScalar         *aa;
12713bededecSBarry Smith   const PetscScalar *xx;
12723bededecSBarry Smith   PetscScalar       *bb;
127356777dd2SBarry Smith   PetscBool         *zeroed,vecs = PETSC_FALSE;
12743bededecSBarry Smith 
12753bededecSBarry Smith   PetscFunctionBegin;
12763bededecSBarry Smith   /* fix right hand side if needed */
12773bededecSBarry Smith   if (x && b) {
12783bededecSBarry Smith     ierr = VecGetArrayRead(x,&xx);CHKERRQ(ierr);
12793bededecSBarry Smith     ierr = VecGetArray(b,&bb);CHKERRQ(ierr);
128056777dd2SBarry Smith     vecs = PETSC_TRUE;
12813bededecSBarry Smith   }
12823bededecSBarry Smith 
12833bededecSBarry Smith   /* zero the columns */
12841795a4d1SJed Brown   ierr = PetscCalloc1(A->rmap->n,&zeroed);CHKERRQ(ierr);
12853bededecSBarry Smith   for (i=0; i<is_n; i++) {
12863bededecSBarry 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]);
12873bededecSBarry Smith     zeroed[is_idx[i]] = PETSC_TRUE;
12883bededecSBarry Smith   }
128956777dd2SBarry Smith   if (vecs) {
129056777dd2SBarry Smith     for (i=0; i<A->rmap->N; i++) {
129156777dd2SBarry Smith       row = i/bs;
129256777dd2SBarry Smith       for (j=baij->i[row]; j<baij->i[row+1]; j++) {
129356777dd2SBarry Smith         for (k=0; k<bs; k++) {
129456777dd2SBarry Smith           col = bs*baij->j[j] + k;
129556777dd2SBarry Smith           if (col <= i) continue;
129656777dd2SBarry Smith           aa = ((MatScalar*)(baij->a)) + j*bs2 + (i%bs) + bs*k;
129726fbe8dcSKarl Rupp           if (!zeroed[i] && zeroed[col]) bb[i]   -= aa[0]*xx[col];
129826fbe8dcSKarl Rupp           if (zeroed[i] && !zeroed[col]) bb[col] -= aa[0]*xx[i];
129956777dd2SBarry Smith         }
130056777dd2SBarry Smith       }
130156777dd2SBarry Smith     }
130226fbe8dcSKarl Rupp     for (i=0; i<is_n; i++) bb[is_idx[i]] = diag*xx[is_idx[i]];
130356777dd2SBarry Smith   }
130456777dd2SBarry Smith 
13053bededecSBarry Smith   for (i=0; i<A->rmap->N; i++) {
13063bededecSBarry Smith     if (!zeroed[i]) {
13073bededecSBarry Smith       row = i/bs;
13083bededecSBarry Smith       for (j=baij->i[row]; j<baij->i[row+1]; j++) {
13093bededecSBarry Smith         for (k=0; k<bs; k++) {
13103bededecSBarry Smith           col = bs*baij->j[j] + k;
13113bededecSBarry Smith           if (zeroed[col]) {
13123bededecSBarry Smith             aa = ((MatScalar*)(baij->a)) + j*bs2 + (i%bs) + bs*k;
13133bededecSBarry Smith             aa[0] = 0.0;
13143bededecSBarry Smith           }
13153bededecSBarry Smith         }
13163bededecSBarry Smith       }
13173bededecSBarry Smith     }
13183bededecSBarry Smith   }
13193bededecSBarry Smith   ierr = PetscFree(zeroed);CHKERRQ(ierr);
132056777dd2SBarry Smith   if (vecs) {
132156777dd2SBarry Smith     ierr = VecRestoreArrayRead(x,&xx);CHKERRQ(ierr);
132256777dd2SBarry Smith     ierr = VecRestoreArray(b,&bb);CHKERRQ(ierr);
132356777dd2SBarry Smith   }
13243bededecSBarry Smith 
13253bededecSBarry Smith   /* zero the rows */
13263bededecSBarry Smith   for (i=0; i<is_n; i++) {
13273bededecSBarry Smith     row   = is_idx[i];
13283bededecSBarry Smith     count = (baij->i[row/bs +1] - baij->i[row/bs])*bs;
13293bededecSBarry Smith     aa    = ((MatScalar*)(baij->a)) + baij->i[row/bs]*bs2 + (row%bs);
13303bededecSBarry Smith     for (k=0; k<count; k++) {
13313bededecSBarry Smith       aa[0] =  zero;
13323bededecSBarry Smith       aa   += bs;
13333bededecSBarry Smith     }
13343bededecSBarry Smith     if (diag != 0.0) {
13353bededecSBarry Smith       ierr = (*A->ops->setvalues)(A,1,&row,1,&row,&diag,INSERT_VALUES);CHKERRQ(ierr);
13363bededecSBarry Smith     }
13373bededecSBarry Smith   }
13383bededecSBarry Smith   ierr = MatAssemblyEnd_SeqSBAIJ(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
13393bededecSBarry Smith   PetscFunctionReturn(0);
13403bededecSBarry Smith }
13413bededecSBarry Smith 
134249b5e25fSSatish Balay /* -------------------------------------------------------------------*/
13433964eb88SJed Brown static struct _MatOps MatOps_Values = {MatSetValues_SeqSBAIJ,
134449b5e25fSSatish Balay                                        MatGetRow_SeqSBAIJ,
134549b5e25fSSatish Balay                                        MatRestoreRow_SeqSBAIJ,
134649b5e25fSSatish Balay                                        MatMult_SeqSBAIJ_N,
134797304618SKris Buschelman                                /*  4*/ MatMultAdd_SeqSBAIJ_N,
1348431c96f7SBarry Smith                                        MatMult_SeqSBAIJ_N,       /* transpose versions are same as non-transpose versions */
1349e005ede5SBarry Smith                                        MatMultAdd_SeqSBAIJ_N,
1350db4efbfdSBarry Smith                                        0,
135149b5e25fSSatish Balay                                        0,
135249b5e25fSSatish Balay                                        0,
135397304618SKris Buschelman                                /* 10*/ 0,
135449b5e25fSSatish Balay                                        0,
1355c078aec8SLisandro Dalcin                                        MatCholeskyFactor_SeqSBAIJ,
135641f059aeSBarry Smith                                        MatSOR_SeqSBAIJ,
135749b5e25fSSatish Balay                                        MatTranspose_SeqSBAIJ,
135897304618SKris Buschelman                                /* 15*/ MatGetInfo_SeqSBAIJ,
135949b5e25fSSatish Balay                                        MatEqual_SeqSBAIJ,
136049b5e25fSSatish Balay                                        MatGetDiagonal_SeqSBAIJ,
136149b5e25fSSatish Balay                                        MatDiagonalScale_SeqSBAIJ,
136249b5e25fSSatish Balay                                        MatNorm_SeqSBAIJ,
136397304618SKris Buschelman                                /* 20*/ 0,
136449b5e25fSSatish Balay                                        MatAssemblyEnd_SeqSBAIJ,
136549b5e25fSSatish Balay                                        MatSetOption_SeqSBAIJ,
136649b5e25fSSatish Balay                                        MatZeroEntries_SeqSBAIJ,
1367d519adbfSMatthew Knepley                                /* 24*/ 0,
136849b5e25fSSatish Balay                                        0,
136949b5e25fSSatish Balay                                        0,
1370db4efbfdSBarry Smith                                        0,
1371db4efbfdSBarry Smith                                        0,
13724994cf47SJed Brown                                /* 29*/ MatSetUp_SeqSBAIJ,
1373c464158bSHong Zhang                                        0,
1374db4efbfdSBarry Smith                                        0,
13758c778c55SBarry Smith                                        0,
13768c778c55SBarry Smith                                        0,
1377d519adbfSMatthew Knepley                                /* 34*/ MatDuplicate_SeqSBAIJ,
1378719d5645SBarry Smith                                        0,
1379719d5645SBarry Smith                                        0,
138049b5e25fSSatish Balay                                        0,
1381c84f5b01SHong Zhang                                        MatICCFactor_SeqSBAIJ,
1382d519adbfSMatthew Knepley                                /* 39*/ MatAXPY_SeqSBAIJ,
138349b5e25fSSatish Balay                                        MatGetSubMatrices_SeqSBAIJ,
138449b5e25fSSatish Balay                                        MatIncreaseOverlap_SeqSBAIJ,
138549b5e25fSSatish Balay                                        MatGetValues_SeqSBAIJ,
13863c896bc6SHong Zhang                                        MatCopy_SeqSBAIJ,
1387d519adbfSMatthew Knepley                                /* 44*/ 0,
138849b5e25fSSatish Balay                                        MatScale_SeqSBAIJ,
138949b5e25fSSatish Balay                                        0,
139049b5e25fSSatish Balay                                        0,
13913bededecSBarry Smith                                        MatZeroRowsColumns_SeqSBAIJ,
1392f73d5cc4SBarry Smith                                /* 49*/ 0,
139349b5e25fSSatish Balay                                        MatGetRowIJ_SeqSBAIJ,
139449b5e25fSSatish Balay                                        MatRestoreRowIJ_SeqSBAIJ,
139549b5e25fSSatish Balay                                        0,
139649b5e25fSSatish Balay                                        0,
1397d519adbfSMatthew Knepley                                /* 54*/ 0,
139849b5e25fSSatish Balay                                        0,
139949b5e25fSSatish Balay                                        0,
140049b5e25fSSatish Balay                                        0,
140149b5e25fSSatish Balay                                        MatSetValuesBlocked_SeqSBAIJ,
1402d519adbfSMatthew Knepley                                /* 59*/ MatGetSubMatrix_SeqSBAIJ,
140349b5e25fSSatish Balay                                        0,
140449b5e25fSSatish Balay                                        0,
1405357abbc8SBarry Smith                                        0,
1406d959ec07SHong Zhang                                        0,
1407d519adbfSMatthew Knepley                                /* 64*/ 0,
1408d959ec07SHong Zhang                                        0,
1409d959ec07SHong Zhang                                        0,
1410d959ec07SHong Zhang                                        0,
1411d959ec07SHong Zhang                                        0,
1412d519adbfSMatthew Knepley                                /* 69*/ MatGetRowMaxAbs_SeqSBAIJ,
14133e0d88b5SBarry Smith                                        0,
14143e0d88b5SBarry Smith                                        0,
14153e0d88b5SBarry Smith                                        0,
14163e0d88b5SBarry Smith                                        0,
1417d519adbfSMatthew Knepley                                /* 74*/ 0,
14183e0d88b5SBarry Smith                                        0,
14193e0d88b5SBarry Smith                                        0,
14203e0d88b5SBarry Smith                                        0,
14213e0d88b5SBarry Smith                                        0,
1422d519adbfSMatthew Knepley                                /* 79*/ 0,
14233e0d88b5SBarry Smith                                        0,
14243e0d88b5SBarry Smith                                        0,
142597304618SKris Buschelman                                        MatGetInertia_SeqSBAIJ,
14265bba2384SShri Abhyankar                                        MatLoad_SeqSBAIJ,
1427d519adbfSMatthew Knepley                                /* 84*/ MatIsSymmetric_SeqSBAIJ,
1428865e5f61SKris Buschelman                                        MatIsHermitian_SeqSBAIJ,
1429efcf0fc3SBarry Smith                                        MatIsStructurallySymmetric_SeqSBAIJ,
1430865e5f61SKris Buschelman                                        0,
1431865e5f61SKris Buschelman                                        0,
1432d519adbfSMatthew Knepley                                /* 89*/ 0,
1433865e5f61SKris Buschelman                                        0,
1434865e5f61SKris Buschelman                                        0,
1435865e5f61SKris Buschelman                                        0,
1436865e5f61SKris Buschelman                                        0,
1437d519adbfSMatthew Knepley                                /* 94*/ 0,
1438865e5f61SKris Buschelman                                        0,
1439865e5f61SKris Buschelman                                        0,
144099cafbc1SBarry Smith                                        0,
144199cafbc1SBarry Smith                                        0,
1442d519adbfSMatthew Knepley                                /* 99*/ 0,
144399cafbc1SBarry Smith                                        0,
144499cafbc1SBarry Smith                                        0,
144599cafbc1SBarry Smith                                        0,
144699cafbc1SBarry Smith                                        0,
1447d519adbfSMatthew Knepley                                /*104*/ 0,
144899cafbc1SBarry Smith                                        MatRealPart_SeqSBAIJ,
1449f5edf698SHong Zhang                                        MatImaginaryPart_SeqSBAIJ,
1450f5edf698SHong Zhang                                        MatGetRowUpperTriangular_SeqSBAIJ,
14512af78befSBarry Smith                                        MatRestoreRowUpperTriangular_SeqSBAIJ,
1452d519adbfSMatthew Knepley                                /*109*/ 0,
14532af78befSBarry Smith                                        0,
14542af78befSBarry Smith                                        0,
14552af78befSBarry Smith                                        0,
1456547795f9SHong Zhang                                        MatMissingDiagonal_SeqSBAIJ,
1457547795f9SHong Zhang                                /*114*/ 0,
1458547795f9SHong Zhang                                        0,
1459547795f9SHong Zhang                                        0,
1460547795f9SHong Zhang                                        0,
1461547795f9SHong Zhang                                        0,
1462547795f9SHong Zhang                                /*119*/ 0,
1463547795f9SHong Zhang                                        0,
14642f480046SShri Abhyankar                                        0,
14653964eb88SJed Brown                                        0,
14663964eb88SJed Brown                                        0,
14673964eb88SJed Brown                                /*124*/ 0,
14683964eb88SJed Brown                                        0,
14693964eb88SJed Brown                                        0,
14703964eb88SJed Brown                                        0,
14713964eb88SJed Brown                                        0,
14723964eb88SJed Brown                                /*129*/ 0,
14733964eb88SJed Brown                                        0,
14743964eb88SJed Brown                                        0,
14753964eb88SJed Brown                                        0,
14763964eb88SJed Brown                                        0,
14773964eb88SJed Brown                                /*134*/ 0,
14783964eb88SJed Brown                                        0,
14793964eb88SJed Brown                                        0,
14803964eb88SJed Brown                                        0,
14813964eb88SJed Brown                                        0,
14823964eb88SJed Brown                                /*139*/ 0,
1483f9426fe0SMark Adams                                        0,
14845bba2384SShri Abhyankar                                        0
148599cafbc1SBarry Smith };
1486be1d678aSKris Buschelman 
14874a2ae208SSatish Balay #undef __FUNCT__
14884a2ae208SSatish Balay #define __FUNCT__ "MatStoreValues_SeqSBAIJ"
14897087cfbeSBarry Smith PetscErrorCode  MatStoreValues_SeqSBAIJ(Mat mat)
149049b5e25fSSatish Balay {
14914afc71dfSHong Zhang   Mat_SeqSBAIJ   *aij = (Mat_SeqSBAIJ*)mat->data;
1492d0f46423SBarry Smith   PetscInt       nz   = aij->i[mat->rmap->N]*mat->rmap->bs*aij->bs2;
1493dfbe8321SBarry Smith   PetscErrorCode ierr;
149449b5e25fSSatish Balay 
149549b5e25fSSatish Balay   PetscFunctionBegin;
1496e7e72b3dSBarry Smith   if (aij->nonew != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first");
149749b5e25fSSatish Balay 
149849b5e25fSSatish Balay   /* allocate space for values if not already there */
149949b5e25fSSatish Balay   if (!aij->saved_values) {
1500785e854fSJed Brown     ierr = PetscMalloc1((nz+1),&aij->saved_values);CHKERRQ(ierr);
150149b5e25fSSatish Balay   }
150249b5e25fSSatish Balay 
150349b5e25fSSatish Balay   /* copy values over */
150487828ca2SBarry Smith   ierr = PetscMemcpy(aij->saved_values,aij->a,nz*sizeof(PetscScalar));CHKERRQ(ierr);
150549b5e25fSSatish Balay   PetscFunctionReturn(0);
150649b5e25fSSatish Balay }
150749b5e25fSSatish Balay 
15084a2ae208SSatish Balay #undef __FUNCT__
15094a2ae208SSatish Balay #define __FUNCT__ "MatRetrieveValues_SeqSBAIJ"
15107087cfbeSBarry Smith PetscErrorCode  MatRetrieveValues_SeqSBAIJ(Mat mat)
151149b5e25fSSatish Balay {
15124afc71dfSHong Zhang   Mat_SeqSBAIJ   *aij = (Mat_SeqSBAIJ*)mat->data;
15136849ba73SBarry Smith   PetscErrorCode ierr;
1514d0f46423SBarry Smith   PetscInt       nz = aij->i[mat->rmap->N]*mat->rmap->bs*aij->bs2;
151549b5e25fSSatish Balay 
151649b5e25fSSatish Balay   PetscFunctionBegin;
1517e7e72b3dSBarry Smith   if (aij->nonew != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first");
1518e7e72b3dSBarry Smith   if (!aij->saved_values) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatStoreValues(A);first");
151949b5e25fSSatish Balay 
152049b5e25fSSatish Balay   /* copy values over */
152187828ca2SBarry Smith   ierr = PetscMemcpy(aij->a,aij->saved_values,nz*sizeof(PetscScalar));CHKERRQ(ierr);
152249b5e25fSSatish Balay   PetscFunctionReturn(0);
152349b5e25fSSatish Balay }
152449b5e25fSSatish Balay 
15254a2ae208SSatish Balay #undef __FUNCT__
1526a23d5eceSKris Buschelman #define __FUNCT__ "MatSeqSBAIJSetPreallocation_SeqSBAIJ"
15277087cfbeSBarry Smith PetscErrorCode  MatSeqSBAIJSetPreallocation_SeqSBAIJ(Mat B,PetscInt bs,PetscInt nz,PetscInt *nnz)
152849b5e25fSSatish Balay {
1529c464158bSHong Zhang   Mat_SeqSBAIJ   *b = (Mat_SeqSBAIJ*)B->data;
15306849ba73SBarry Smith   PetscErrorCode ierr;
1531535b19f3SBarry Smith   PetscInt       i,mbs,bs2;
15322576faa2SJed Brown   PetscBool      skipallocation = PETSC_FALSE,flg = PETSC_FALSE,realalloc = PETSC_FALSE;
153349b5e25fSSatish Balay 
153449b5e25fSSatish Balay   PetscFunctionBegin;
15352576faa2SJed Brown   if (nz >= 0 || nnz) realalloc = PETSC_TRUE;
1536273d9f13SBarry Smith   B->preallocated = PETSC_TRUE;
1537db4efbfdSBarry Smith 
153833d57670SJed Brown   ierr = MatSetBlockSize(B,PetscAbs(bs));CHKERRQ(ierr);
153926283091SBarry Smith   ierr = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr);
154026283091SBarry Smith   ierr = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr);
1541e02043d6SBarry Smith   ierr = PetscLayoutGetBlockSize(B->rmap,&bs);CHKERRQ(ierr);
1542899cda47SBarry Smith 
1543d0f46423SBarry Smith   mbs = B->rmap->N/bs;
154449b5e25fSSatish Balay   bs2 = bs*bs;
154549b5e25fSSatish Balay 
1546e7e72b3dSBarry Smith   if (mbs*bs != B->rmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Number rows, cols must be divisible by blocksize");
154749b5e25fSSatish Balay 
1548ab93d7beSBarry Smith   if (nz == MAT_SKIP_ALLOCATION) {
1549ab93d7beSBarry Smith     skipallocation = PETSC_TRUE;
1550ab93d7beSBarry Smith     nz             = 0;
1551ab93d7beSBarry Smith   }
1552ab93d7beSBarry Smith 
1553435da068SBarry Smith   if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 3;
1554e32f2f54SBarry Smith   if (nz < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"nz cannot be less than 0: value %D",nz);
155549b5e25fSSatish Balay   if (nnz) {
155649b5e25fSSatish Balay     for (i=0; i<mbs; i++) {
1557e32f2f54SBarry 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]);
1558e32f2f54SBarry 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);
155949b5e25fSSatish Balay     }
156049b5e25fSSatish Balay   }
156149b5e25fSSatish Balay 
1562db4efbfdSBarry Smith   B->ops->mult             = MatMult_SeqSBAIJ_N;
1563db4efbfdSBarry Smith   B->ops->multadd          = MatMultAdd_SeqSBAIJ_N;
1564db4efbfdSBarry Smith   B->ops->multtranspose    = MatMult_SeqSBAIJ_N;
1565db4efbfdSBarry Smith   B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_N;
156626fbe8dcSKarl Rupp 
15670298fd71SBarry Smith   ierr  = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_no_unroll",&flg,NULL);CHKERRQ(ierr);
156849b5e25fSSatish Balay   if (!flg) {
156949b5e25fSSatish Balay     switch (bs) {
157049b5e25fSSatish Balay     case 1:
157149b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_1;
157249b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_1;
1573431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_1;
1574431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_1;
157549b5e25fSSatish Balay       break;
157649b5e25fSSatish Balay     case 2:
157749b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_2;
157849b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_2;
1579431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_2;
1580431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_2;
158149b5e25fSSatish Balay       break;
158249b5e25fSSatish Balay     case 3:
158349b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_3;
158449b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_3;
1585431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_3;
1586431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_3;
158749b5e25fSSatish Balay       break;
158849b5e25fSSatish Balay     case 4:
158949b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_4;
159049b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_4;
1591431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_4;
1592431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_4;
159349b5e25fSSatish Balay       break;
159449b5e25fSSatish Balay     case 5:
159549b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_5;
159649b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_5;
1597431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_5;
1598431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_5;
159949b5e25fSSatish Balay       break;
160049b5e25fSSatish Balay     case 6:
160149b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_6;
160249b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_6;
1603431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_6;
1604431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_6;
160549b5e25fSSatish Balay       break;
160649b5e25fSSatish Balay     case 7:
1607de53e5efSHong Zhang       B->ops->mult             = MatMult_SeqSBAIJ_7;
160849b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_7;
1609431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_7;
1610431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_7;
161149b5e25fSSatish Balay       break;
161249b5e25fSSatish Balay     }
161349b5e25fSSatish Balay   }
161449b5e25fSSatish Balay 
161549b5e25fSSatish Balay   b->mbs = mbs;
16164afc71dfSHong Zhang   b->nbs = mbs;
1617ab93d7beSBarry Smith   if (!skipallocation) {
16182ee49352SLisandro Dalcin     if (!b->imax) {
1619dcca6d9dSJed Brown       ierr = PetscMalloc2(mbs,&b->imax,mbs,&b->ilen);CHKERRQ(ierr);
162026fbe8dcSKarl Rupp 
1621c760cd28SBarry Smith       b->free_imax_ilen = PETSC_TRUE;
162226fbe8dcSKarl Rupp 
16233bb1ff40SBarry Smith       ierr = PetscLogObjectMemory((PetscObject)B,2*mbs*sizeof(PetscInt));CHKERRQ(ierr);
16242ee49352SLisandro Dalcin     }
162549b5e25fSSatish Balay     if (!nnz) {
1626435da068SBarry Smith       if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 5;
162749b5e25fSSatish Balay       else if (nz <= 0) nz = 1;
162826fbe8dcSKarl Rupp       for (i=0; i<mbs; i++) b->imax[i] = nz;
1629153ea458SHong Zhang       nz = nz*mbs; /* total nz */
163049b5e25fSSatish Balay     } else {
163149b5e25fSSatish Balay       nz = 0;
16328cef66ccSHong Zhang       for (i=0; i<mbs; i++) {b->imax[i] = nnz[i]; nz += nnz[i];}
163349b5e25fSSatish Balay     }
16342ee49352SLisandro Dalcin     /* b->ilen will count nonzeros in each block row so far. */
163526fbe8dcSKarl Rupp     for (i=0; i<mbs; i++) b->ilen[i] = 0;
16366c6c5352SBarry Smith     /* nz=(nz+mbs)/2; */ /* total diagonal and superdiagonal nonzero blocks */
163749b5e25fSSatish Balay 
163849b5e25fSSatish Balay     /* allocate the matrix space */
16392ee49352SLisandro Dalcin     ierr = MatSeqXAIJFreeAIJ(B,&b->a,&b->j,&b->i);CHKERRQ(ierr);
1640dcca6d9dSJed Brown     ierr = PetscMalloc3(bs2*nz,&b->a,nz,&b->j,B->rmap->N+1,&b->i);CHKERRQ(ierr);
16413bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)B,(B->rmap->N+1)*sizeof(PetscInt)+nz*(bs2*sizeof(PetscScalar)+sizeof(PetscInt)));CHKERRQ(ierr);
16426c6c5352SBarry Smith     ierr = PetscMemzero(b->a,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr);
164313f74950SBarry Smith     ierr = PetscMemzero(b->j,nz*sizeof(PetscInt));CHKERRQ(ierr);
164426fbe8dcSKarl Rupp 
164549b5e25fSSatish Balay     b->singlemalloc = PETSC_TRUE;
164649b5e25fSSatish Balay 
164749b5e25fSSatish Balay     /* pointer to beginning of each row */
1648e60cf9a0SBarry Smith     b->i[0] = 0;
164926fbe8dcSKarl Rupp     for (i=1; i<mbs+1; i++) b->i[i] = b->i[i-1] + b->imax[i-1];
165026fbe8dcSKarl Rupp 
1651e6b907acSBarry Smith     b->free_a  = PETSC_TRUE;
1652e6b907acSBarry Smith     b->free_ij = PETSC_TRUE;
1653e811da20SHong Zhang   } else {
1654e6b907acSBarry Smith     b->free_a  = PETSC_FALSE;
1655e6b907acSBarry Smith     b->free_ij = PETSC_FALSE;
1656ab93d7beSBarry Smith   }
165749b5e25fSSatish Balay 
1658d0f46423SBarry Smith   B->rmap->bs = bs;
165949b5e25fSSatish Balay   b->bs2      = bs2;
16606c6c5352SBarry Smith   b->nz       = 0;
1661b32cb4a7SJed Brown   b->maxnz    = nz;
1662153ea458SHong Zhang 
166316cdd363SHong Zhang   b->inew    = 0;
166416cdd363SHong Zhang   b->jnew    = 0;
166516cdd363SHong Zhang   b->anew    = 0;
166616cdd363SHong Zhang   b->a2anew  = 0;
16671a3463dfSHong Zhang   b->permute = PETSC_FALSE;
16682576faa2SJed Brown   if (realalloc) {ierr = MatSetOption(B,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);}
1669c464158bSHong Zhang   PetscFunctionReturn(0);
1670c464158bSHong Zhang }
1671153ea458SHong Zhang 
167238f409ebSLisandro Dalcin #undef __FUNCT__
167338f409ebSLisandro Dalcin #define __FUNCT__ "MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ"
167438f409ebSLisandro Dalcin PetscErrorCode MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ(Mat B,PetscInt bs,const PetscInt ii[],const PetscInt jj[], const PetscScalar V[])
167538f409ebSLisandro Dalcin {
167638f409ebSLisandro Dalcin   PetscInt       i,j,m,nz,nz_max=0,*nnz;
167738f409ebSLisandro Dalcin   PetscScalar    *values=0;
167838f409ebSLisandro Dalcin   PetscBool      roworiented = ((Mat_SeqSBAIJ*)B->data)->roworiented;
167938f409ebSLisandro Dalcin   PetscErrorCode ierr;
168038f409ebSLisandro Dalcin   PetscFunctionBegin;
168138f409ebSLisandro Dalcin   if (bs < 1) SETERRQ1(PetscObjectComm((PetscObject)B),PETSC_ERR_ARG_OUTOFRANGE,"Invalid block size specified, must be positive but it is %D",bs);
168238f409ebSLisandro Dalcin   ierr   = PetscLayoutSetBlockSize(B->rmap,bs);CHKERRQ(ierr);
168338f409ebSLisandro Dalcin   ierr   = PetscLayoutSetBlockSize(B->cmap,bs);CHKERRQ(ierr);
168438f409ebSLisandro Dalcin   ierr   = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr);
168538f409ebSLisandro Dalcin   ierr   = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr);
168638f409ebSLisandro Dalcin   ierr   = PetscLayoutGetBlockSize(B->rmap,&bs);CHKERRQ(ierr);
168738f409ebSLisandro Dalcin   m      = B->rmap->n/bs;
168838f409ebSLisandro Dalcin 
168938f409ebSLisandro Dalcin   if (ii[0]) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"ii[0] must be 0 but it is %D",ii[0]);
1690785e854fSJed Brown   ierr = PetscMalloc1((m+1),&nnz);CHKERRQ(ierr);
169138f409ebSLisandro Dalcin   for (i=0; i<m; i++) {
169238f409ebSLisandro Dalcin     nz = ii[i+1] - ii[i];
169338f409ebSLisandro Dalcin     if (nz < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row %D has a negative number of columns %D",i,nz);
169438f409ebSLisandro Dalcin     nz_max = PetscMax(nz_max,nz);
169538f409ebSLisandro Dalcin     nnz[i] = nz;
169638f409ebSLisandro Dalcin   }
169738f409ebSLisandro Dalcin   ierr = MatSeqSBAIJSetPreallocation(B,bs,0,nnz);CHKERRQ(ierr);
169838f409ebSLisandro Dalcin   ierr = PetscFree(nnz);CHKERRQ(ierr);
169938f409ebSLisandro Dalcin 
170038f409ebSLisandro Dalcin   values = (PetscScalar*)V;
170138f409ebSLisandro Dalcin   if (!values) {
17021795a4d1SJed Brown     ierr = PetscCalloc1(bs*bs*nz_max,&values);CHKERRQ(ierr);
170338f409ebSLisandro Dalcin   }
170438f409ebSLisandro Dalcin   for (i=0; i<m; i++) {
170538f409ebSLisandro Dalcin     PetscInt          ncols  = ii[i+1] - ii[i];
170638f409ebSLisandro Dalcin     const PetscInt    *icols = jj + ii[i];
170738f409ebSLisandro Dalcin     if (!roworiented || bs == 1) {
170838f409ebSLisandro Dalcin       const PetscScalar *svals = values + (V ? (bs*bs*ii[i]) : 0);
170938f409ebSLisandro Dalcin       ierr = MatSetValuesBlocked_SeqSBAIJ(B,1,&i,ncols,icols,svals,INSERT_VALUES);CHKERRQ(ierr);
171038f409ebSLisandro Dalcin     } else {
171138f409ebSLisandro Dalcin       for (j=0; j<ncols; j++) {
171238f409ebSLisandro Dalcin         const PetscScalar *svals = values + (V ? (bs*bs*(ii[i]+j)) : 0);
171338f409ebSLisandro Dalcin         ierr = MatSetValuesBlocked_SeqSBAIJ(B,1,&i,1,&icols[j],svals,INSERT_VALUES);CHKERRQ(ierr);
171438f409ebSLisandro Dalcin       }
171538f409ebSLisandro Dalcin     }
171638f409ebSLisandro Dalcin   }
171738f409ebSLisandro Dalcin   if (!V) { ierr = PetscFree(values);CHKERRQ(ierr); }
171838f409ebSLisandro Dalcin   ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
171938f409ebSLisandro Dalcin   ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
172038f409ebSLisandro Dalcin   ierr = MatSetOption(B,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
172138f409ebSLisandro Dalcin   PetscFunctionReturn(0);
172238f409ebSLisandro Dalcin }
172338f409ebSLisandro Dalcin 
1724db4efbfdSBarry Smith /*
1725db4efbfdSBarry Smith    This is used to set the numeric factorization for both Cholesky and ICC symbolic factorization
1726db4efbfdSBarry Smith */
17278b1456e3SHong Zhang #undef __FUNCT__
1728d595f711SHong Zhang #define __FUNCT__ "MatSeqSBAIJSetNumericFactorization_inplace"
1729ace3abfcSBarry Smith PetscErrorCode MatSeqSBAIJSetNumericFactorization_inplace(Mat B,PetscBool natural)
1730db4efbfdSBarry Smith {
1731db4efbfdSBarry Smith   PetscErrorCode ierr;
1732ace3abfcSBarry Smith   PetscBool      flg = PETSC_FALSE;
1733db4efbfdSBarry Smith   PetscInt       bs  = B->rmap->bs;
1734db4efbfdSBarry Smith 
1735db4efbfdSBarry Smith   PetscFunctionBegin;
17360298fd71SBarry Smith   ierr = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_no_unroll",&flg,NULL);CHKERRQ(ierr);
1737db4efbfdSBarry Smith   if (flg) bs = 8;
1738db4efbfdSBarry Smith 
1739db4efbfdSBarry Smith   if (!natural) {
1740db4efbfdSBarry Smith     switch (bs) {
1741db4efbfdSBarry Smith     case 1:
1742d595f711SHong Zhang       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_inplace;
1743db4efbfdSBarry Smith       break;
1744db4efbfdSBarry Smith     case 2:
1745db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2;
1746db4efbfdSBarry Smith       break;
1747db4efbfdSBarry Smith     case 3:
1748db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3;
1749db4efbfdSBarry Smith       break;
1750db4efbfdSBarry Smith     case 4:
1751db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4;
1752db4efbfdSBarry Smith       break;
1753db4efbfdSBarry Smith     case 5:
1754db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5;
1755db4efbfdSBarry Smith       break;
1756db4efbfdSBarry Smith     case 6:
1757db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6;
1758db4efbfdSBarry Smith       break;
1759db4efbfdSBarry Smith     case 7:
1760db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7;
1761db4efbfdSBarry Smith       break;
1762db4efbfdSBarry Smith     default:
1763db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N;
1764db4efbfdSBarry Smith       break;
1765db4efbfdSBarry Smith     }
1766db4efbfdSBarry Smith   } else {
1767db4efbfdSBarry Smith     switch (bs) {
1768db4efbfdSBarry Smith     case 1:
1769d595f711SHong Zhang       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering_inplace;
1770db4efbfdSBarry Smith       break;
1771db4efbfdSBarry Smith     case 2:
1772db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering;
1773db4efbfdSBarry Smith       break;
1774db4efbfdSBarry Smith     case 3:
1775db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering;
1776db4efbfdSBarry Smith       break;
1777db4efbfdSBarry Smith     case 4:
1778db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering;
1779db4efbfdSBarry Smith       break;
1780db4efbfdSBarry Smith     case 5:
1781db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering;
1782db4efbfdSBarry Smith       break;
1783db4efbfdSBarry Smith     case 6:
1784db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering;
1785db4efbfdSBarry Smith       break;
1786db4efbfdSBarry Smith     case 7:
1787db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering;
1788db4efbfdSBarry Smith       break;
1789db4efbfdSBarry Smith     default:
1790db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering;
1791db4efbfdSBarry Smith       break;
1792db4efbfdSBarry Smith     }
1793db4efbfdSBarry Smith   }
1794db4efbfdSBarry Smith   PetscFunctionReturn(0);
1795db4efbfdSBarry Smith }
1796db4efbfdSBarry Smith 
17978cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqAIJ(Mat, MatType,MatReuse,Mat*);
17988cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqBAIJ(Mat, MatType,MatReuse,Mat*);
1799d769727bSBarry Smith 
18005c9eb25fSBarry Smith #undef __FUNCT__
18015c9eb25fSBarry Smith #define __FUNCT__ "MatGetFactor_seqsbaij_petsc"
18028cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_seqsbaij_petsc(Mat A,MatFactorType ftype,Mat *B)
18035c9eb25fSBarry Smith {
1804d0f46423SBarry Smith   PetscInt       n = A->rmap->n;
18055c9eb25fSBarry Smith   PetscErrorCode ierr;
18065c9eb25fSBarry Smith 
18075c9eb25fSBarry Smith   PetscFunctionBegin;
18080e92d65fSHong Zhang #if defined(PETSC_USE_COMPLEX)
18090e92d65fSHong Zhang   if (A->hermitian) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Hermitian Factor is not supported");
18100e92d65fSHong Zhang #endif
1811ce94432eSBarry Smith   ierr = MatCreate(PetscObjectComm((PetscObject)A),B);CHKERRQ(ierr);
18125c9eb25fSBarry Smith   ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr);
18135c9eb25fSBarry Smith   if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) {
18145c9eb25fSBarry Smith     ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr);
18150298fd71SBarry Smith     ierr = MatSeqSBAIJSetPreallocation(*B,A->rmap->bs,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr);
181626fbe8dcSKarl Rupp 
18177b056e98SHong Zhang     (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqSBAIJ;
1818c6d0d4f0SHong Zhang     (*B)->ops->iccfactorsymbolic      = MatICCFactorSymbolic_SeqSBAIJ;
1819e32f2f54SBarry Smith   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Factor type not supported");
1820d5f3da31SBarry Smith   (*B)->factortype = ftype;
18215c9eb25fSBarry Smith   PetscFunctionReturn(0);
18225c9eb25fSBarry Smith }
18235c9eb25fSBarry Smith 
1824db4efbfdSBarry Smith #undef __FUNCT__
1825db4efbfdSBarry Smith #define __FUNCT__ "MatGetFactorAvailable_seqsbaij_petsc"
1826ace3abfcSBarry Smith PetscErrorCode MatGetFactorAvailable_seqsbaij_petsc(Mat A,MatFactorType ftype,PetscBool  *flg)
1827db4efbfdSBarry Smith {
1828db4efbfdSBarry Smith   PetscFunctionBegin;
1829db4efbfdSBarry Smith   if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) {
1830db4efbfdSBarry Smith     *flg = PETSC_TRUE;
1831db4efbfdSBarry Smith   } else {
1832db4efbfdSBarry Smith     *flg = PETSC_FALSE;
1833db4efbfdSBarry Smith   }
1834db4efbfdSBarry Smith   PetscFunctionReturn(0);
1835db4efbfdSBarry Smith }
1836db4efbfdSBarry Smith 
1837611f576cSBarry Smith #if defined(PETSC_HAVE_MUMPS)
18388cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_sbaij_mumps(Mat,MatFactorType,Mat*);
1839611f576cSBarry Smith #endif
1840b5e56a35SBarry Smith #if defined(PETSC_HAVE_PASTIX)
18418cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_seqsbaij_pastix(Mat,MatFactorType,Mat*);
1842b5e56a35SBarry Smith #endif
1843913c410bSShri Abhyankar #if defined(PETSC_HAVE_SUITESPARSE)
18448cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_seqsbaij_cholmod(Mat,MatFactorType,Mat*);
184520db9a53SJed Brown #endif
18468cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_seqsbaij_sbstrm(Mat,MatFactorType,Mat*);
18475c9eb25fSBarry Smith 
18480bad9183SKris Buschelman /*MC
1849fafad747SKris Buschelman   MATSEQSBAIJ - MATSEQSBAIJ = "seqsbaij" - A matrix type to be used for sequential symmetric block sparse matrices,
18500bad9183SKris Buschelman   based on block compressed sparse row format.  Only the upper triangular portion of the matrix is stored.
18510bad9183SKris Buschelman 
1852828413b8SBarry Smith   For complex numbers by default this matrix is symmetric, NOT Hermitian symmetric. To make it Hermitian symmetric you
185371dad5bbSBarry Smith   can call MatSetOption(Mat, MAT_HERMITIAN); after MatAssemblyEnd()
1854828413b8SBarry Smith 
18550bad9183SKris Buschelman   Options Database Keys:
18560bad9183SKris Buschelman   . -mat_type seqsbaij - sets the matrix type to "seqsbaij" during a call to MatSetFromOptions()
18570bad9183SKris Buschelman 
185871dad5bbSBarry Smith   Notes: By default if you insert values into the lower triangular part of the matrix they are simply ignored (since they are not
185971dad5bbSBarry 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
186071dad5bbSBarry 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.
186171dad5bbSBarry Smith 
186271dad5bbSBarry Smith 
18630bad9183SKris Buschelman   Level: beginner
18640bad9183SKris Buschelman 
18650bad9183SKris Buschelman   .seealso: MatCreateSeqSBAIJ
18660bad9183SKris Buschelman M*/
18670bad9183SKris Buschelman 
18688cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqSBSTRM(Mat, MatType,MatReuse,Mat*);
1869aa5a9175SDahai Guo 
1870a23d5eceSKris Buschelman #undef __FUNCT__
1871a23d5eceSKris Buschelman #define __FUNCT__ "MatCreate_SeqSBAIJ"
18728cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatCreate_SeqSBAIJ(Mat B)
1873a23d5eceSKris Buschelman {
1874a23d5eceSKris Buschelman   Mat_SeqSBAIJ   *b;
1875dfbe8321SBarry Smith   PetscErrorCode ierr;
187613f74950SBarry Smith   PetscMPIInt    size;
1877ace3abfcSBarry Smith   PetscBool      no_unroll = PETSC_FALSE,no_inode = PETSC_FALSE;
1878a23d5eceSKris Buschelman 
1879a23d5eceSKris Buschelman   PetscFunctionBegin;
1880ce94432eSBarry Smith   ierr = MPI_Comm_size(PetscObjectComm((PetscObject)B),&size);CHKERRQ(ierr);
1881e32f2f54SBarry Smith   if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Comm must be of size 1");
1882a23d5eceSKris Buschelman 
1883b00a9115SJed Brown   ierr    = PetscNewLog(B,&b);CHKERRQ(ierr);
1884a23d5eceSKris Buschelman   B->data = (void*)b;
1885a23d5eceSKris Buschelman   ierr    = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr);
188626fbe8dcSKarl Rupp 
1887a23d5eceSKris Buschelman   B->ops->destroy    = MatDestroy_SeqSBAIJ;
1888a23d5eceSKris Buschelman   B->ops->view       = MatView_SeqSBAIJ;
1889a23d5eceSKris Buschelman   b->row             = 0;
1890a23d5eceSKris Buschelman   b->icol            = 0;
1891a23d5eceSKris Buschelman   b->reallocs        = 0;
1892a23d5eceSKris Buschelman   b->saved_values    = 0;
18930def2e27SBarry Smith   b->inode.limit     = 5;
18940def2e27SBarry Smith   b->inode.max_limit = 5;
1895a23d5eceSKris Buschelman 
1896a23d5eceSKris Buschelman   b->roworiented        = PETSC_TRUE;
1897a23d5eceSKris Buschelman   b->nonew              = 0;
1898a23d5eceSKris Buschelman   b->diag               = 0;
1899a23d5eceSKris Buschelman   b->solve_work         = 0;
1900a23d5eceSKris Buschelman   b->mult_work          = 0;
1901a23d5eceSKris Buschelman   B->spptr              = 0;
1902f2cbd3d5SJed Brown   B->info.nz_unneeded   = (PetscReal)b->maxnz*b->bs2;
1903a9817697SBarry Smith   b->keepnonzeropattern = PETSC_FALSE;
1904a23d5eceSKris Buschelman   b->xtoy               = 0;
1905a23d5eceSKris Buschelman   b->XtoY               = 0;
1906a23d5eceSKris Buschelman 
1907a23d5eceSKris Buschelman   b->inew    = 0;
1908a23d5eceSKris Buschelman   b->jnew    = 0;
1909a23d5eceSKris Buschelman   b->anew    = 0;
1910a23d5eceSKris Buschelman   b->a2anew  = 0;
1911a23d5eceSKris Buschelman   b->permute = PETSC_FALSE;
1912a23d5eceSKris Buschelman 
191371dad5bbSBarry Smith   b->ignore_ltriangular = PETSC_TRUE;
191426fbe8dcSKarl Rupp 
19150298fd71SBarry Smith   ierr = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_ignore_lower_triangular",&b->ignore_ltriangular,NULL);CHKERRQ(ierr);
1916941593c8SHong Zhang 
1917f5edf698SHong Zhang   b->getrow_utriangular = PETSC_FALSE;
191826fbe8dcSKarl Rupp 
19190298fd71SBarry Smith   ierr = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_getrow_uppertriangular",&b->getrow_utriangular,NULL);CHKERRQ(ierr);
1920f5edf698SHong Zhang 
1921b5e56a35SBarry Smith #if defined(PETSC_HAVE_PASTIX)
1922bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactor_pastix_C",MatGetFactor_seqsbaij_pastix);CHKERRQ(ierr);
1923b5e56a35SBarry Smith #endif
1924611f576cSBarry Smith #if defined(PETSC_HAVE_MUMPS)
1925bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactor_mumps_C",MatGetFactor_sbaij_mumps);CHKERRQ(ierr);
1926611f576cSBarry Smith #endif
1927913c410bSShri Abhyankar #if defined(PETSC_HAVE_SUITESPARSE)
1928bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactor_cholmod_C",MatGetFactor_seqsbaij_cholmod);CHKERRQ(ierr);
192920db9a53SJed Brown #endif
1930bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactorAvailable_petsc_C",MatGetFactorAvailable_seqsbaij_petsc);CHKERRQ(ierr);
1931bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactor_petsc_C",MatGetFactor_seqsbaij_petsc);CHKERRQ(ierr);
1932bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactor_sbstrm_C",MatGetFactor_seqsbaij_sbstrm);CHKERRQ(ierr);
1933bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatStoreValues_C",MatStoreValues_SeqSBAIJ);CHKERRQ(ierr);
1934bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatRetrieveValues_C",MatRetrieveValues_SeqSBAIJ);CHKERRQ(ierr);
1935bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetColumnIndices_C",MatSeqSBAIJSetColumnIndices_SeqSBAIJ);CHKERRQ(ierr);
1936bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqaij_C",MatConvert_SeqSBAIJ_SeqAIJ);CHKERRQ(ierr);
1937bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqbaij_C",MatConvert_SeqSBAIJ_SeqBAIJ);CHKERRQ(ierr);
1938bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetPreallocation_C",MatSeqSBAIJSetPreallocation_SeqSBAIJ);CHKERRQ(ierr);
193938f409ebSLisandro Dalcin   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetPreallocationCSR_C",MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ);CHKERRQ(ierr);
1940bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqsbstrm_C",MatConvert_SeqSBAIJ_SeqSBSTRM);CHKERRQ(ierr);
194123ce1328SBarry Smith 
194223ce1328SBarry Smith   B->symmetric                  = PETSC_TRUE;
194323ce1328SBarry Smith   B->structurally_symmetric     = PETSC_TRUE;
194423ce1328SBarry Smith   B->symmetric_set              = PETSC_TRUE;
194523ce1328SBarry Smith   B->structurally_symmetric_set = PETSC_TRUE;
194626fbe8dcSKarl Rupp 
194717667f90SBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)B,MATSEQSBAIJ);CHKERRQ(ierr);
19480def2e27SBarry Smith 
1949ce94432eSBarry Smith   ierr = PetscOptionsBegin(PetscObjectComm((PetscObject)B),((PetscObject)B)->prefix,"Options for SEQSBAIJ matrix","Mat");CHKERRQ(ierr);
19500298fd71SBarry Smith   ierr = PetscOptionsBool("-mat_no_unroll","Do not optimize for inodes (slower)",NULL,no_unroll,&no_unroll,NULL);CHKERRQ(ierr);
195126fbe8dcSKarl Rupp   if (no_unroll) {
195226fbe8dcSKarl Rupp     ierr = PetscInfo(B,"Not using Inode routines due to -mat_no_unroll\n");CHKERRQ(ierr);
195326fbe8dcSKarl Rupp   }
19540298fd71SBarry Smith   ierr = PetscOptionsBool("-mat_no_inode","Do not optimize for inodes (slower)",NULL,no_inode,&no_inode,NULL);CHKERRQ(ierr);
195526fbe8dcSKarl Rupp   if (no_inode) {
195626fbe8dcSKarl Rupp     ierr = PetscInfo(B,"Not using Inode routines due to -mat_no_inode\n");CHKERRQ(ierr);
195726fbe8dcSKarl Rupp   }
19580298fd71SBarry Smith   ierr = PetscOptionsInt("-mat_inode_limit","Do not use inodes larger then this value",NULL,b->inode.limit,&b->inode.limit,NULL);CHKERRQ(ierr);
19590def2e27SBarry Smith   ierr = PetscOptionsEnd();CHKERRQ(ierr);
1960ace3abfcSBarry Smith   b->inode.use = (PetscBool)(!(no_unroll || no_inode));
19610def2e27SBarry Smith   if (b->inode.limit > b->inode.max_limit) b->inode.limit = b->inode.max_limit;
1962a23d5eceSKris Buschelman   PetscFunctionReturn(0);
1963a23d5eceSKris Buschelman }
1964a23d5eceSKris Buschelman 
1965a23d5eceSKris Buschelman #undef __FUNCT__
1966a23d5eceSKris Buschelman #define __FUNCT__ "MatSeqSBAIJSetPreallocation"
1967a23d5eceSKris Buschelman /*@C
1968a23d5eceSKris Buschelman    MatSeqSBAIJSetPreallocation - Creates a sparse symmetric matrix in block AIJ (block
1969a23d5eceSKris Buschelman    compressed row) format.  For good matrix assembly performance the
1970a23d5eceSKris Buschelman    user should preallocate the matrix storage by setting the parameter nz
1971a23d5eceSKris Buschelman    (or the array nnz).  By setting these parameters accurately, performance
1972a23d5eceSKris Buschelman    during matrix assembly can be increased by more than a factor of 50.
1973a23d5eceSKris Buschelman 
1974a23d5eceSKris Buschelman    Collective on Mat
1975a23d5eceSKris Buschelman 
1976a23d5eceSKris Buschelman    Input Parameters:
19771c4f3114SJed Brown +  B - the symmetric matrix
1978a23d5eceSKris Buschelman .  bs - size of block
1979a23d5eceSKris Buschelman .  nz - number of block nonzeros per block row (same for all rows)
1980a23d5eceSKris Buschelman -  nnz - array containing the number of block nonzeros in the upper triangular plus
19810298fd71SBarry Smith          diagonal portion of each block (possibly different for each block row) or NULL
1982a23d5eceSKris Buschelman 
1983a23d5eceSKris Buschelman    Options Database Keys:
1984a23d5eceSKris Buschelman .   -mat_no_unroll - uses code that does not unroll the loops in the
1985a23d5eceSKris Buschelman                      block calculations (much slower)
1986db4efbfdSBarry Smith .    -mat_block_size - size of the blocks to use (only works if a negative bs is passed in
1987a23d5eceSKris Buschelman 
1988a23d5eceSKris Buschelman    Level: intermediate
1989a23d5eceSKris Buschelman 
1990a23d5eceSKris Buschelman    Notes:
1991a23d5eceSKris Buschelman    Specify the preallocated storage with either nz or nnz (not both).
19920298fd71SBarry Smith    Set nz=PETSC_DEFAULT and nnz=NULL for PETSc to control dynamic memory
1993a7f22e61SSatish Balay    allocation.  See Users-Manual: ch_mat for details.
1994a23d5eceSKris Buschelman 
1995aa95bbe8SBarry Smith    You can call MatGetInfo() to get information on how effective the preallocation was;
1996aa95bbe8SBarry Smith    for example the fields mallocs,nz_allocated,nz_used,nz_unneeded;
1997aa95bbe8SBarry Smith    You can also run with the option -info and look for messages with the string
1998aa95bbe8SBarry Smith    malloc in them to see if additional memory allocation was needed.
1999aa95bbe8SBarry Smith 
200049a6f317SBarry Smith    If the nnz parameter is given then the nz parameter is ignored
200149a6f317SBarry Smith 
200249a6f317SBarry Smith 
200369b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateSBAIJ()
2004a23d5eceSKris Buschelman @*/
20057087cfbeSBarry Smith PetscErrorCode  MatSeqSBAIJSetPreallocation(Mat B,PetscInt bs,PetscInt nz,const PetscInt nnz[])
200613f74950SBarry Smith {
20074ac538c5SBarry Smith   PetscErrorCode ierr;
2008a23d5eceSKris Buschelman 
2009a23d5eceSKris Buschelman   PetscFunctionBegin;
20106ba663aaSJed Brown   PetscValidHeaderSpecific(B,MAT_CLASSID,1);
20116ba663aaSJed Brown   PetscValidType(B,1);
20126ba663aaSJed Brown   PetscValidLogicalCollectiveInt(B,bs,2);
20134ac538c5SBarry Smith   ierr = PetscTryMethod(B,"MatSeqSBAIJSetPreallocation_C",(Mat,PetscInt,PetscInt,const PetscInt[]),(B,bs,nz,nnz));CHKERRQ(ierr);
2014a23d5eceSKris Buschelman   PetscFunctionReturn(0);
2015a23d5eceSKris Buschelman }
201649b5e25fSSatish Balay 
20174a2ae208SSatish Balay #undef  __FUNCT__
201838f409ebSLisandro Dalcin #define __FUNCT__ "MatSeqSBAIJSetPreallocationCSR"
201938f409ebSLisandro Dalcin /*@C
202038f409ebSLisandro Dalcin    MatSeqSBAIJSetPreallocationCSR - Allocates memory for a sparse sequential matrix in symmetric block AIJ format.
202138f409ebSLisandro Dalcin 
202238f409ebSLisandro Dalcin    Input Parameters:
20231c4f3114SJed Brown +  B - the matrix
202438f409ebSLisandro Dalcin .  i - the indices into j for the start of each local row (starts with zero)
202538f409ebSLisandro Dalcin .  j - the column indices for each local row (starts with zero) these must be sorted for each row
202638f409ebSLisandro Dalcin -  v - optional values in the matrix
202738f409ebSLisandro Dalcin 
202838f409ebSLisandro Dalcin    Level: developer
202938f409ebSLisandro Dalcin 
203038f409ebSLisandro Dalcin    Notes:
203138f409ebSLisandro Dalcin    The order of the entries in values is specified by the MatOption MAT_ROW_ORIENTED.  For example, C programs
203238f409ebSLisandro 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
203338f409ebSLisandro Dalcin    over rows within a block and the last index is over columns within a block row.  Fortran programs will likely set
203438f409ebSLisandro 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
203538f409ebSLisandro Dalcin    block column and the second index is over columns within a block.
203638f409ebSLisandro Dalcin 
203738f409ebSLisandro Dalcin .keywords: matrix, block, aij, compressed row, sparse
203838f409ebSLisandro Dalcin 
203938f409ebSLisandro Dalcin .seealso: MatCreate(), MatCreateSeqSBAIJ(), MatSetValuesBlocked(), MatSeqSBAIJSetPreallocation(), MATSEQSBAIJ
204038f409ebSLisandro Dalcin @*/
204138f409ebSLisandro Dalcin PetscErrorCode MatSeqSBAIJSetPreallocationCSR(Mat B,PetscInt bs,const PetscInt i[],const PetscInt j[], const PetscScalar v[])
204238f409ebSLisandro Dalcin {
204338f409ebSLisandro Dalcin   PetscErrorCode ierr;
204438f409ebSLisandro Dalcin 
204538f409ebSLisandro Dalcin   PetscFunctionBegin;
204638f409ebSLisandro Dalcin   PetscValidHeaderSpecific(B,MAT_CLASSID,1);
204738f409ebSLisandro Dalcin   PetscValidType(B,1);
204838f409ebSLisandro Dalcin   PetscValidLogicalCollectiveInt(B,bs,2);
204938f409ebSLisandro Dalcin   ierr = PetscTryMethod(B,"MatSeqSBAIJSetPreallocationCSR_C",(Mat,PetscInt,const PetscInt[],const PetscInt[],const PetscScalar[]),(B,bs,i,j,v));CHKERRQ(ierr);
205038f409ebSLisandro Dalcin   PetscFunctionReturn(0);
205138f409ebSLisandro Dalcin }
205238f409ebSLisandro Dalcin 
205338f409ebSLisandro Dalcin #undef __FUNCT__
20544a2ae208SSatish Balay #define __FUNCT__ "MatCreateSeqSBAIJ"
2055c464158bSHong Zhang /*@C
2056c464158bSHong Zhang    MatCreateSeqSBAIJ - Creates a sparse symmetric matrix in block AIJ (block
2057c464158bSHong Zhang    compressed row) format.  For good matrix assembly performance the
2058c464158bSHong Zhang    user should preallocate the matrix storage by setting the parameter nz
2059c464158bSHong Zhang    (or the array nnz).  By setting these parameters accurately, performance
2060c464158bSHong Zhang    during matrix assembly can be increased by more than a factor of 50.
206149b5e25fSSatish Balay 
2062c464158bSHong Zhang    Collective on MPI_Comm
2063c464158bSHong Zhang 
2064c464158bSHong Zhang    Input Parameters:
2065c464158bSHong Zhang +  comm - MPI communicator, set to PETSC_COMM_SELF
2066c464158bSHong Zhang .  bs - size of block
2067c464158bSHong Zhang .  m - number of rows, or number of columns
2068c464158bSHong Zhang .  nz - number of block nonzeros per block row (same for all rows)
2069744e8345SSatish Balay -  nnz - array containing the number of block nonzeros in the upper triangular plus
20700298fd71SBarry Smith          diagonal portion of each block (possibly different for each block row) or NULL
2071c464158bSHong Zhang 
2072c464158bSHong Zhang    Output Parameter:
2073c464158bSHong Zhang .  A - the symmetric matrix
2074c464158bSHong Zhang 
2075c464158bSHong Zhang    Options Database Keys:
2076c464158bSHong Zhang .   -mat_no_unroll - uses code that does not unroll the loops in the
2077c464158bSHong Zhang                      block calculations (much slower)
2078c464158bSHong Zhang .    -mat_block_size - size of the blocks to use
2079c464158bSHong Zhang 
2080c464158bSHong Zhang    Level: intermediate
2081c464158bSHong Zhang 
2082175b88e8SBarry Smith    It is recommended that one use the MatCreate(), MatSetType() and/or MatSetFromOptions(),
2083ae1d86c5SBarry Smith    MatXXXXSetPreallocation() paradgm instead of this routine directly.
2084175b88e8SBarry Smith    [MatXXXXSetPreallocation() is, for example, MatSeqAIJSetPreallocation]
2085175b88e8SBarry Smith 
2086c464158bSHong Zhang    Notes:
20876d6d819aSHong Zhang    The number of rows and columns must be divisible by blocksize.
20886d6d819aSHong Zhang    This matrix type does not support complex Hermitian operation.
2089c464158bSHong Zhang 
2090c464158bSHong Zhang    Specify the preallocated storage with either nz or nnz (not both).
20910298fd71SBarry Smith    Set nz=PETSC_DEFAULT and nnz=NULL for PETSc to control dynamic memory
2092a7f22e61SSatish Balay    allocation.  See Users-Manual: ch_mat for details.
2093c464158bSHong Zhang 
209449a6f317SBarry Smith    If the nnz parameter is given then the nz parameter is ignored
209549a6f317SBarry Smith 
209669b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateSBAIJ()
2097c464158bSHong Zhang @*/
20987087cfbeSBarry Smith PetscErrorCode  MatCreateSeqSBAIJ(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt nz,const PetscInt nnz[],Mat *A)
2099c464158bSHong Zhang {
2100dfbe8321SBarry Smith   PetscErrorCode ierr;
2101c464158bSHong Zhang 
2102c464158bSHong Zhang   PetscFunctionBegin;
2103f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,A);CHKERRQ(ierr);
2104f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*A,m,n,m,n);CHKERRQ(ierr);
2105c464158bSHong Zhang   ierr = MatSetType(*A,MATSEQSBAIJ);CHKERRQ(ierr);
2106ab93d7beSBarry Smith   ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(*A,bs,nz,(PetscInt*)nnz);CHKERRQ(ierr);
210749b5e25fSSatish Balay   PetscFunctionReturn(0);
210849b5e25fSSatish Balay }
210949b5e25fSSatish Balay 
21104a2ae208SSatish Balay #undef __FUNCT__
21114a2ae208SSatish Balay #define __FUNCT__ "MatDuplicate_SeqSBAIJ"
2112dfbe8321SBarry Smith PetscErrorCode MatDuplicate_SeqSBAIJ(Mat A,MatDuplicateOption cpvalues,Mat *B)
211349b5e25fSSatish Balay {
211449b5e25fSSatish Balay   Mat            C;
211549b5e25fSSatish Balay   Mat_SeqSBAIJ   *c,*a = (Mat_SeqSBAIJ*)A->data;
21166849ba73SBarry Smith   PetscErrorCode ierr;
2117b40805acSSatish Balay   PetscInt       i,mbs = a->mbs,nz = a->nz,bs2 =a->bs2;
211849b5e25fSSatish Balay 
211949b5e25fSSatish Balay   PetscFunctionBegin;
2120e32f2f54SBarry Smith   if (a->i[mbs] != nz) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Corrupt matrix");
212149b5e25fSSatish Balay 
212249b5e25fSSatish Balay   *B   = 0;
2123ce94432eSBarry Smith   ierr = MatCreate(PetscObjectComm((PetscObject)A),&C);CHKERRQ(ierr);
2124d0f46423SBarry Smith   ierr = MatSetSizes(C,A->rmap->N,A->cmap->n,A->rmap->N,A->cmap->n);CHKERRQ(ierr);
21258e9a0fb8SHong Zhang   ierr = MatSetType(C,MATSEQSBAIJ);CHKERRQ(ierr);
21261d5dac46SHong Zhang   ierr = PetscMemcpy(C->ops,A->ops,sizeof(struct _MatOps));CHKERRQ(ierr);
2127692f9cbeSHong Zhang   c    = (Mat_SeqSBAIJ*)C->data;
2128692f9cbeSHong Zhang 
2129273d9f13SBarry Smith   C->preallocated       = PETSC_TRUE;
2130d5f3da31SBarry Smith   C->factortype         = A->factortype;
213149b5e25fSSatish Balay   c->row                = 0;
213249b5e25fSSatish Balay   c->icol               = 0;
213349b5e25fSSatish Balay   c->saved_values       = 0;
2134a9817697SBarry Smith   c->keepnonzeropattern = a->keepnonzeropattern;
213549b5e25fSSatish Balay   C->assembled          = PETSC_TRUE;
213649b5e25fSSatish Balay 
21371e1e43feSBarry Smith   ierr   = PetscLayoutReference(A->rmap,&C->rmap);CHKERRQ(ierr);
21381e1e43feSBarry Smith   ierr   = PetscLayoutReference(A->cmap,&C->cmap);CHKERRQ(ierr);
213949b5e25fSSatish Balay   c->bs2 = a->bs2;
214049b5e25fSSatish Balay   c->mbs = a->mbs;
214149b5e25fSSatish Balay   c->nbs = a->nbs;
214249b5e25fSSatish Balay 
2143c760cd28SBarry Smith   if  (cpvalues == MAT_SHARE_NONZERO_PATTERN) {
2144c760cd28SBarry Smith     c->imax           = a->imax;
2145c760cd28SBarry Smith     c->ilen           = a->ilen;
2146c760cd28SBarry Smith     c->free_imax_ilen = PETSC_FALSE;
2147c760cd28SBarry Smith   } else {
2148dcca6d9dSJed Brown     ierr = PetscMalloc2((mbs+1),&c->imax,(mbs+1),&c->ilen);CHKERRQ(ierr);
21493bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)C,2*(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr);
215049b5e25fSSatish Balay     for (i=0; i<mbs; i++) {
215149b5e25fSSatish Balay       c->imax[i] = a->imax[i];
215249b5e25fSSatish Balay       c->ilen[i] = a->ilen[i];
215349b5e25fSSatish Balay     }
2154c760cd28SBarry Smith     c->free_imax_ilen = PETSC_TRUE;
2155c760cd28SBarry Smith   }
215649b5e25fSSatish Balay 
215749b5e25fSSatish Balay   /* allocate the matrix space */
21584da8f245SBarry Smith   if (cpvalues == MAT_SHARE_NONZERO_PATTERN) {
2159785e854fSJed Brown     ierr            = PetscMalloc1(bs2*nz,&c->a);CHKERRQ(ierr);
21603bb1ff40SBarry Smith     ierr            = PetscLogObjectMemory((PetscObject)C,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr);
216144e1c64aSLisandro Dalcin     c->i            = a->i;
216244e1c64aSLisandro Dalcin     c->j            = a->j;
21634da8f245SBarry Smith     c->singlemalloc = PETSC_FALSE;
216444e1c64aSLisandro Dalcin     c->free_a       = PETSC_TRUE;
21654da8f245SBarry Smith     c->free_ij      = PETSC_FALSE;
21664da8f245SBarry Smith     c->parent       = A;
21674da8f245SBarry Smith     ierr            = PetscObjectReference((PetscObject)A);CHKERRQ(ierr);
21684da8f245SBarry Smith     ierr            = MatSetOption(A,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
21694da8f245SBarry Smith     ierr            = MatSetOption(C,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
21704da8f245SBarry Smith   } else {
2171dcca6d9dSJed Brown     ierr            = PetscMalloc3(bs2*nz,&c->a,nz,&c->j,mbs+1,&c->i);CHKERRQ(ierr);
217213f74950SBarry Smith     ierr            = PetscMemcpy(c->i,a->i,(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr);
21733bb1ff40SBarry Smith     ierr            = PetscLogObjectMemory((PetscObject)C,(mbs+1)*sizeof(PetscInt) + nz*(bs2*sizeof(MatScalar) + sizeof(PetscInt)));CHKERRQ(ierr);
21744da8f245SBarry Smith     c->singlemalloc = PETSC_TRUE;
217544e1c64aSLisandro Dalcin     c->free_a       = PETSC_TRUE;
21764da8f245SBarry Smith     c->free_ij      = PETSC_TRUE;
21774da8f245SBarry Smith   }
217849b5e25fSSatish Balay   if (mbs > 0) {
21794da8f245SBarry Smith     if (cpvalues != MAT_SHARE_NONZERO_PATTERN) {
218013f74950SBarry Smith       ierr = PetscMemcpy(c->j,a->j,nz*sizeof(PetscInt));CHKERRQ(ierr);
21814da8f245SBarry Smith     }
218249b5e25fSSatish Balay     if (cpvalues == MAT_COPY_VALUES) {
218349b5e25fSSatish Balay       ierr = PetscMemcpy(c->a,a->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr);
218449b5e25fSSatish Balay     } else {
218549b5e25fSSatish Balay       ierr = PetscMemzero(c->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr);
218649b5e25fSSatish Balay     }
2187a1c3900fSBarry Smith     if (a->jshort) {
218844e1c64aSLisandro Dalcin       /* cannot share jshort, it is reallocated in MatAssemblyEnd_SeqSBAIJ() */
218944e1c64aSLisandro Dalcin       /* if the parent matrix is reassembled, this child matrix will never notice */
2190785e854fSJed Brown       ierr = PetscMalloc1(nz,&c->jshort);CHKERRQ(ierr);
21913bb1ff40SBarry Smith       ierr = PetscLogObjectMemory((PetscObject)C,nz*sizeof(unsigned short));CHKERRQ(ierr);
2192a1c3900fSBarry Smith       ierr = PetscMemcpy(c->jshort,a->jshort,nz*sizeof(unsigned short));CHKERRQ(ierr);
219326fbe8dcSKarl Rupp 
21944da8f245SBarry Smith       c->free_jshort = PETSC_TRUE;
21954da8f245SBarry Smith     }
2196a1c3900fSBarry Smith   }
219749b5e25fSSatish Balay 
219849b5e25fSSatish Balay   c->roworiented = a->roworiented;
219949b5e25fSSatish Balay   c->nonew       = a->nonew;
220049b5e25fSSatish Balay 
220149b5e25fSSatish Balay   if (a->diag) {
2202c760cd28SBarry Smith     if (cpvalues == MAT_SHARE_NONZERO_PATTERN) {
2203c760cd28SBarry Smith       c->diag      = a->diag;
2204c760cd28SBarry Smith       c->free_diag = PETSC_FALSE;
2205c760cd28SBarry Smith     } else {
2206785e854fSJed Brown       ierr = PetscMalloc1(mbs,&c->diag);CHKERRQ(ierr);
22073bb1ff40SBarry Smith       ierr = PetscLogObjectMemory((PetscObject)C,mbs*sizeof(PetscInt));CHKERRQ(ierr);
220826fbe8dcSKarl Rupp       for (i=0; i<mbs; i++) c->diag[i] = a->diag[i];
2209c760cd28SBarry Smith       c->free_diag = PETSC_TRUE;
2210c760cd28SBarry Smith     }
221144e1c64aSLisandro Dalcin   }
22126c6c5352SBarry Smith   c->nz         = a->nz;
2213f2cbd3d5SJed Brown   c->maxnz      = a->nz; /* Since we allocate exactly the right amount */
221449b5e25fSSatish Balay   c->solve_work = 0;
221549b5e25fSSatish Balay   c->mult_work  = 0;
221626fbe8dcSKarl Rupp 
221749b5e25fSSatish Balay   *B   = C;
2218140e18c1SBarry Smith   ierr = PetscFunctionListDuplicate(((PetscObject)A)->qlist,&((PetscObject)C)->qlist);CHKERRQ(ierr);
221949b5e25fSSatish Balay   PetscFunctionReturn(0);
222049b5e25fSSatish Balay }
222149b5e25fSSatish Balay 
22224a2ae208SSatish Balay #undef __FUNCT__
22235bba2384SShri Abhyankar #define __FUNCT__ "MatLoad_SeqSBAIJ"
2224112444f4SShri Abhyankar PetscErrorCode MatLoad_SeqSBAIJ(Mat newmat,PetscViewer viewer)
22252f480046SShri Abhyankar {
22262f480046SShri Abhyankar   Mat_SeqSBAIJ   *a;
22272f480046SShri Abhyankar   PetscErrorCode ierr;
22282f480046SShri Abhyankar   int            fd;
22292f480046SShri Abhyankar   PetscMPIInt    size;
22302f480046SShri Abhyankar   PetscInt       i,nz,header[4],*rowlengths=0,M,N,bs=1;
22312f480046SShri Abhyankar   PetscInt       *mask,mbs,*jj,j,rowcount,nzcount,k,*s_browlengths,maskcount;
22322f480046SShri Abhyankar   PetscInt       kmax,jcount,block,idx,point,nzcountb,extra_rows,rows,cols;
22332f480046SShri Abhyankar   PetscInt       *masked,nmask,tmp,bs2,ishift;
22342f480046SShri Abhyankar   PetscScalar    *aa;
2235ce94432eSBarry Smith   MPI_Comm       comm;
22362f480046SShri Abhyankar 
22372f480046SShri Abhyankar   PetscFunctionBegin;
2238ce94432eSBarry Smith   ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr);
22390298fd71SBarry Smith   ierr = PetscOptionsGetInt(((PetscObject)newmat)->prefix,"-matload_block_size",&bs,NULL);CHKERRQ(ierr);
22402f480046SShri Abhyankar   bs2  = bs*bs;
22412f480046SShri Abhyankar 
22422f480046SShri Abhyankar   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
22432f480046SShri Abhyankar   if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"view must have one processor");
22442f480046SShri Abhyankar   ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr);
22452f480046SShri Abhyankar   ierr = PetscBinaryRead(fd,header,4,PETSC_INT);CHKERRQ(ierr);
22462f480046SShri Abhyankar   if (header[0] != MAT_FILE_CLASSID) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"not Mat object");
22472f480046SShri Abhyankar   M = header[1]; N = header[2]; nz = header[3];
22482f480046SShri Abhyankar 
22492f480046SShri Abhyankar   if (header[3] < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Matrix stored in special format, cannot load as SeqSBAIJ");
22502f480046SShri Abhyankar 
22512f480046SShri Abhyankar   if (M != N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Can only do square matrices");
22522f480046SShri Abhyankar 
22532f480046SShri Abhyankar   /*
22542f480046SShri Abhyankar      This code adds extra rows to make sure the number of rows is
22552f480046SShri Abhyankar     divisible by the blocksize
22562f480046SShri Abhyankar   */
22572f480046SShri Abhyankar   mbs        = M/bs;
22582f480046SShri Abhyankar   extra_rows = bs - M + bs*(mbs);
22592f480046SShri Abhyankar   if (extra_rows == bs) extra_rows = 0;
22602f480046SShri Abhyankar   else                  mbs++;
22612f480046SShri Abhyankar   if (extra_rows) {
22622f480046SShri Abhyankar     ierr = PetscInfo(viewer,"Padding loaded matrix to match blocksize\n");CHKERRQ(ierr);
22632f480046SShri Abhyankar   }
22642f480046SShri Abhyankar 
22652f480046SShri Abhyankar   /* Set global sizes if not already set */
22662f480046SShri Abhyankar   if (newmat->rmap->n < 0 && newmat->rmap->N < 0 && newmat->cmap->n < 0 && newmat->cmap->N < 0) {
22672f480046SShri Abhyankar     ierr = MatSetSizes(newmat,PETSC_DECIDE,PETSC_DECIDE,M+extra_rows,N+extra_rows);CHKERRQ(ierr);
22682f480046SShri Abhyankar   } else { /* Check if the matrix global sizes are correct */
22692f480046SShri Abhyankar     ierr = MatGetSize(newmat,&rows,&cols);CHKERRQ(ierr);
22702f480046SShri Abhyankar     if (M != rows ||  N != cols) SETERRQ4(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Matrix in file of different length (%d, %d) than the input matrix (%d, %d)",M,N,rows,cols);
22712f480046SShri Abhyankar   }
22722f480046SShri Abhyankar 
22732f480046SShri Abhyankar   /* read in row lengths */
2274785e854fSJed Brown   ierr = PetscMalloc1((M+extra_rows),&rowlengths);CHKERRQ(ierr);
22752f480046SShri Abhyankar   ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr);
22762f480046SShri Abhyankar   for (i=0; i<extra_rows; i++) rowlengths[M+i] = 1;
22772f480046SShri Abhyankar 
22782f480046SShri Abhyankar   /* read in column indices */
2279785e854fSJed Brown   ierr = PetscMalloc1((nz+extra_rows),&jj);CHKERRQ(ierr);
22802f480046SShri Abhyankar   ierr = PetscBinaryRead(fd,jj,nz,PETSC_INT);CHKERRQ(ierr);
22812f480046SShri Abhyankar   for (i=0; i<extra_rows; i++) jj[nz+i] = M+i;
22822f480046SShri Abhyankar 
22832f480046SShri Abhyankar   /* loop over row lengths determining block row lengths */
22841795a4d1SJed Brown   ierr     = PetscCalloc1(mbs,&s_browlengths);CHKERRQ(ierr);
2285dcca6d9dSJed Brown   ierr     = PetscMalloc2(mbs,&mask,mbs,&masked);CHKERRQ(ierr);
22862f480046SShri Abhyankar   ierr     = PetscMemzero(mask,mbs*sizeof(PetscInt));CHKERRQ(ierr);
22872f480046SShri Abhyankar   rowcount = 0;
22882f480046SShri Abhyankar   nzcount  = 0;
22892f480046SShri Abhyankar   for (i=0; i<mbs; i++) {
22902f480046SShri Abhyankar     nmask = 0;
22912f480046SShri Abhyankar     for (j=0; j<bs; j++) {
22922f480046SShri Abhyankar       kmax = rowlengths[rowcount];
22932f480046SShri Abhyankar       for (k=0; k<kmax; k++) {
22942f480046SShri Abhyankar         tmp = jj[nzcount++]/bs;   /* block col. index */
22952f480046SShri Abhyankar         if (!mask[tmp] && tmp >= i) {masked[nmask++] = tmp; mask[tmp] = 1;}
22962f480046SShri Abhyankar       }
22972f480046SShri Abhyankar       rowcount++;
22982f480046SShri Abhyankar     }
22992f480046SShri Abhyankar     s_browlengths[i] += nmask;
23002f480046SShri Abhyankar 
23012f480046SShri Abhyankar     /* zero out the mask elements we set */
23022f480046SShri Abhyankar     for (j=0; j<nmask; j++) mask[masked[j]] = 0;
23032f480046SShri Abhyankar   }
23042f480046SShri Abhyankar 
23052f480046SShri Abhyankar   /* Do preallocation */
23062f480046SShri Abhyankar   ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(newmat,bs,0,s_browlengths);CHKERRQ(ierr);
23072f480046SShri Abhyankar   a    = (Mat_SeqSBAIJ*)newmat->data;
23082f480046SShri Abhyankar 
23092f480046SShri Abhyankar   /* set matrix "i" values */
23102f480046SShri Abhyankar   a->i[0] = 0;
23112f480046SShri Abhyankar   for (i=1; i<= mbs; i++) {
23122f480046SShri Abhyankar     a->i[i]      = a->i[i-1] + s_browlengths[i-1];
23132f480046SShri Abhyankar     a->ilen[i-1] = s_browlengths[i-1];
23142f480046SShri Abhyankar   }
23152f480046SShri Abhyankar   a->nz = a->i[mbs];
23162f480046SShri Abhyankar 
23172f480046SShri Abhyankar   /* read in nonzero values */
2318785e854fSJed Brown   ierr = PetscMalloc1((nz+extra_rows),&aa);CHKERRQ(ierr);
23192f480046SShri Abhyankar   ierr = PetscBinaryRead(fd,aa,nz,PETSC_SCALAR);CHKERRQ(ierr);
23202f480046SShri Abhyankar   for (i=0; i<extra_rows; i++) aa[nz+i] = 1.0;
23212f480046SShri Abhyankar 
23222f480046SShri Abhyankar   /* set "a" and "j" values into matrix */
23232f480046SShri Abhyankar   nzcount = 0; jcount = 0;
23242f480046SShri Abhyankar   for (i=0; i<mbs; i++) {
23252f480046SShri Abhyankar     nzcountb = nzcount;
23262f480046SShri Abhyankar     nmask    = 0;
23272f480046SShri Abhyankar     for (j=0; j<bs; j++) {
23282f480046SShri Abhyankar       kmax = rowlengths[i*bs+j];
23292f480046SShri Abhyankar       for (k=0; k<kmax; k++) {
23302f480046SShri Abhyankar         tmp = jj[nzcount++]/bs; /* block col. index */
23312f480046SShri Abhyankar         if (!mask[tmp] && tmp >= i) { masked[nmask++] = tmp; mask[tmp] = 1;}
23322f480046SShri Abhyankar       }
23332f480046SShri Abhyankar     }
23342f480046SShri Abhyankar     /* sort the masked values */
23352f480046SShri Abhyankar     ierr = PetscSortInt(nmask,masked);CHKERRQ(ierr);
23362f480046SShri Abhyankar 
23372f480046SShri Abhyankar     /* set "j" values into matrix */
23382f480046SShri Abhyankar     maskcount = 1;
23392f480046SShri Abhyankar     for (j=0; j<nmask; j++) {
23402f480046SShri Abhyankar       a->j[jcount++]  = masked[j];
23412f480046SShri Abhyankar       mask[masked[j]] = maskcount++;
23422f480046SShri Abhyankar     }
23432f480046SShri Abhyankar 
23442f480046SShri Abhyankar     /* set "a" values into matrix */
23452f480046SShri Abhyankar     ishift = bs2*a->i[i];
23462f480046SShri Abhyankar     for (j=0; j<bs; j++) {
23472f480046SShri Abhyankar       kmax = rowlengths[i*bs+j];
23482f480046SShri Abhyankar       for (k=0; k<kmax; k++) {
23492f480046SShri Abhyankar         tmp = jj[nzcountb]/bs;        /* block col. index */
23502f480046SShri Abhyankar         if (tmp >= i) {
23512f480046SShri Abhyankar           block     = mask[tmp] - 1;
23522f480046SShri Abhyankar           point     = jj[nzcountb] - bs*tmp;
23532f480046SShri Abhyankar           idx       = ishift + bs2*block + j + bs*point;
23542f480046SShri Abhyankar           a->a[idx] = aa[nzcountb];
23552f480046SShri Abhyankar         }
23562f480046SShri Abhyankar         nzcountb++;
23572f480046SShri Abhyankar       }
23582f480046SShri Abhyankar     }
23592f480046SShri Abhyankar     /* zero out the mask elements we set */
23602f480046SShri Abhyankar     for (j=0; j<nmask; j++) mask[masked[j]] = 0;
23612f480046SShri Abhyankar   }
23622f480046SShri Abhyankar   if (jcount != a->nz) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Bad binary matrix");
23632f480046SShri Abhyankar 
23642f480046SShri Abhyankar   ierr = PetscFree(rowlengths);CHKERRQ(ierr);
23652f480046SShri Abhyankar   ierr = PetscFree(s_browlengths);CHKERRQ(ierr);
23662f480046SShri Abhyankar   ierr = PetscFree(aa);CHKERRQ(ierr);
23672f480046SShri Abhyankar   ierr = PetscFree(jj);CHKERRQ(ierr);
23682f480046SShri Abhyankar   ierr = PetscFree2(mask,masked);CHKERRQ(ierr);
23692f480046SShri Abhyankar 
23702f480046SShri Abhyankar   ierr = MatAssemblyBegin(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
23712f480046SShri Abhyankar   ierr = MatAssemblyEnd(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
23722f480046SShri Abhyankar   PetscFunctionReturn(0);
23732f480046SShri Abhyankar }
23742f480046SShri Abhyankar 
23752f480046SShri Abhyankar #undef __FUNCT__
2376c75a6043SHong Zhang #define __FUNCT__ "MatCreateSeqSBAIJWithArrays"
2377c75a6043SHong Zhang /*@
2378c75a6043SHong Zhang      MatCreateSeqSBAIJWithArrays - Creates an sequential SBAIJ matrix using matrix elements
2379c75a6043SHong Zhang               (upper triangular entries in CSR format) provided by the user.
2380c75a6043SHong Zhang 
2381c75a6043SHong Zhang      Collective on MPI_Comm
2382c75a6043SHong Zhang 
2383c75a6043SHong Zhang    Input Parameters:
2384c75a6043SHong Zhang +  comm - must be an MPI communicator of size 1
2385c75a6043SHong Zhang .  bs - size of block
2386c75a6043SHong Zhang .  m - number of rows
2387c75a6043SHong Zhang .  n - number of columns
2388c75a6043SHong Zhang .  i - row indices
2389c75a6043SHong Zhang .  j - column indices
2390c75a6043SHong Zhang -  a - matrix values
2391c75a6043SHong Zhang 
2392c75a6043SHong Zhang    Output Parameter:
2393c75a6043SHong Zhang .  mat - the matrix
2394c75a6043SHong Zhang 
2395dfb205c3SBarry Smith    Level: advanced
2396c75a6043SHong Zhang 
2397c75a6043SHong Zhang    Notes:
2398c75a6043SHong Zhang        The i, j, and a arrays are not copied by this routine, the user must free these arrays
2399c75a6043SHong Zhang     once the matrix is destroyed
2400c75a6043SHong Zhang 
2401c75a6043SHong Zhang        You cannot set new nonzero locations into this matrix, that will generate an error.
2402c75a6043SHong Zhang 
2403c75a6043SHong Zhang        The i and j indices are 0 based
2404c75a6043SHong Zhang 
2405dfb205c3SBarry 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
2406dfb205c3SBarry Smith        it is the regular CSR format excluding the lower triangular elements.
2407dfb205c3SBarry Smith 
240869b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSBAIJ(), MatCreateSeqSBAIJ()
2409c75a6043SHong Zhang 
2410c75a6043SHong Zhang @*/
24117087cfbeSBarry Smith PetscErrorCode  MatCreateSeqSBAIJWithArrays(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt *i,PetscInt *j,PetscScalar *a,Mat *mat)
2412c75a6043SHong Zhang {
2413c75a6043SHong Zhang   PetscErrorCode ierr;
2414c75a6043SHong Zhang   PetscInt       ii;
2415c75a6043SHong Zhang   Mat_SeqSBAIJ   *sbaij;
2416c75a6043SHong Zhang 
2417c75a6043SHong Zhang   PetscFunctionBegin;
2418e32f2f54SBarry Smith   if (bs != 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"block size %D > 1 is not supported yet",bs);
2419e32f2f54SBarry Smith   if (i[0]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"i (row indices) must start with 0");
2420c75a6043SHong Zhang 
2421c75a6043SHong Zhang   ierr  = MatCreate(comm,mat);CHKERRQ(ierr);
2422c75a6043SHong Zhang   ierr  = MatSetSizes(*mat,m,n,m,n);CHKERRQ(ierr);
2423c75a6043SHong Zhang   ierr  = MatSetType(*mat,MATSEQSBAIJ);CHKERRQ(ierr);
2424c75a6043SHong Zhang   ierr  = MatSeqSBAIJSetPreallocation_SeqSBAIJ(*mat,bs,MAT_SKIP_ALLOCATION,0);CHKERRQ(ierr);
2425c75a6043SHong Zhang   sbaij = (Mat_SeqSBAIJ*)(*mat)->data;
2426dcca6d9dSJed Brown   ierr  = PetscMalloc2(m,&sbaij->imax,m,&sbaij->ilen);CHKERRQ(ierr);
24273bb1ff40SBarry Smith   ierr  = PetscLogObjectMemory((PetscObject)*mat,2*m*sizeof(PetscInt));CHKERRQ(ierr);
2428c75a6043SHong Zhang 
2429c75a6043SHong Zhang   sbaij->i = i;
2430c75a6043SHong Zhang   sbaij->j = j;
2431c75a6043SHong Zhang   sbaij->a = a;
243226fbe8dcSKarl Rupp 
2433c75a6043SHong Zhang   sbaij->singlemalloc = PETSC_FALSE;
2434c75a6043SHong Zhang   sbaij->nonew        = -1;             /*this indicates that inserting a new value in the matrix that generates a new nonzero is an error*/
2435e6b907acSBarry Smith   sbaij->free_a       = PETSC_FALSE;
2436e6b907acSBarry Smith   sbaij->free_ij      = PETSC_FALSE;
2437c75a6043SHong Zhang 
2438c75a6043SHong Zhang   for (ii=0; ii<m; ii++) {
2439c75a6043SHong Zhang     sbaij->ilen[ii] = sbaij->imax[ii] = i[ii+1] - i[ii];
2440c75a6043SHong Zhang #if defined(PETSC_USE_DEBUG)
2441e32f2f54SBarry 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]);
2442c75a6043SHong Zhang #endif
2443c75a6043SHong Zhang   }
2444c75a6043SHong Zhang #if defined(PETSC_USE_DEBUG)
2445c75a6043SHong Zhang   for (ii=0; ii<sbaij->i[m]; ii++) {
2446e32f2f54SBarry Smith     if (j[ii] < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column index at location = %d index = %d",ii,j[ii]);
2447e32f2f54SBarry 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]);
2448c75a6043SHong Zhang   }
2449c75a6043SHong Zhang #endif
2450c75a6043SHong Zhang 
2451c75a6043SHong Zhang   ierr = MatAssemblyBegin(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2452c75a6043SHong Zhang   ierr = MatAssemblyEnd(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2453c75a6043SHong Zhang   PetscFunctionReturn(0);
2454c75a6043SHong Zhang }
2455d06b337dSHong Zhang 
2456d06b337dSHong Zhang 
2457d06b337dSHong Zhang 
245849b5e25fSSatish Balay 
245949b5e25fSSatish Balay 
2460