xref: /petsc/src/mat/impls/sbaij/seq/sbaij.c (revision c6eed6cf04a35cf2a4dee2c7f140142355706926)
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++) {
90*c6eed6cfSHong Zhang         (*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++) {
95*c6eed6cfSHong Zhang         (*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;
312ade3a672SBarry Smith     const char *matname;
313ade3a672SBarry Smith 
314d5f3da31SBarry Smith     if (A->factortype && bs>1) {
31570d5e725SHong 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);
31670d5e725SHong Zhang       PetscFunctionReturn(0);
31770d5e725SHong Zhang     }
318c9f458caSMatthew Knepley     ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&aij);CHKERRQ(ierr);
319ade3a672SBarry Smith     ierr = PetscObjectGetName((PetscObject)A,&matname);CHKERRQ(ierr);
320ade3a672SBarry Smith     ierr = PetscObjectSetName((PetscObject)aij,matname);CHKERRQ(ierr);
321c9f458caSMatthew Knepley     ierr = MatView(aij,viewer);CHKERRQ(ierr);
3226bf464f9SBarry Smith     ierr = MatDestroy(&aij);CHKERRQ(ierr);
323fb9695e5SSatish Balay   } else if (format == PETSC_VIEWER_ASCII_COMMON) {
324d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
32549b5e25fSSatish Balay     for (i=0; i<a->mbs; i++) {
32649b5e25fSSatish Balay       for (j=0; j<bs; j++) {
32777431f27SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr);
32849b5e25fSSatish Balay         for (k=a->i[i]; k<a->i[i+1]; k++) {
32949b5e25fSSatish Balay           for (l=0; l<bs; l++) {
33049b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX)
33149b5e25fSSatish Balay             if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
33257622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k]+l,
33357622a8eSBarry Smith                                             (double)PetscRealPart(a->a[bs2*k + l*bs + j]),(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
33449b5e25fSSatish Balay             } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
33557622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k]+l,
33657622a8eSBarry Smith                                             (double)PetscRealPart(a->a[bs2*k + l*bs + j]),-(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
33749b5e25fSSatish Balay             } else if (PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
33857622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
33949b5e25fSSatish Balay             }
34049b5e25fSSatish Balay #else
34149b5e25fSSatish Balay             if (a->a[bs2*k + l*bs + j] != 0.0) {
34257622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)a->a[bs2*k + l*bs + j]);CHKERRQ(ierr);
34349b5e25fSSatish Balay             }
34449b5e25fSSatish Balay #endif
34549b5e25fSSatish Balay           }
34649b5e25fSSatish Balay         }
347b0a32e0cSBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
34849b5e25fSSatish Balay       }
34949b5e25fSSatish Balay     }
350d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
351c1490034SHong Zhang   } else if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) {
352c1490034SHong Zhang     PetscFunctionReturn(0);
35349b5e25fSSatish Balay   } else {
354d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
3552c990fa1SHong Zhang     if (A->factortype) { /* for factored matrix */
3562c990fa1SHong Zhang       if (bs>1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"matrix is factored with bs>1. Not implemented yet");
3572c990fa1SHong Zhang 
358121deb67SSatish Balay       diag=a->diag;
359121deb67SSatish Balay       for (i=0; i<a->mbs; i++) { /* for row block i */
3602c990fa1SHong Zhang         ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i);CHKERRQ(ierr);
3612c990fa1SHong Zhang         /* diagonal entry */
3622c990fa1SHong Zhang #if defined(PETSC_USE_COMPLEX)
3632c990fa1SHong Zhang         if (PetscImaginaryPart(a->a[diag[i]]) > 0.0) {
36457622a8eSBarry 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);
3652c990fa1SHong Zhang         } else if (PetscImaginaryPart(a->a[diag[i]]) < 0.0) {
36657622a8eSBarry 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);
3672c990fa1SHong Zhang         } else {
36857622a8eSBarry Smith           ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[diag[i]],(double)PetscRealPart(1.0/a->a[diag[i]]));CHKERRQ(ierr);
3692c990fa1SHong Zhang         }
3702c990fa1SHong Zhang #else
3716712e2f1SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[diag[i]],(double)(1.0/a->a[diag[i]]));CHKERRQ(ierr);
3722c990fa1SHong Zhang #endif
3732c990fa1SHong Zhang         /* off-diagonal entries */
3742c990fa1SHong Zhang         for (k=a->i[i]; k<a->i[i+1]-1; k++) {
3752c990fa1SHong Zhang #if defined(PETSC_USE_COMPLEX)
376ca0704adSBarry Smith           if (PetscImaginaryPart(a->a[k]) > 0.0) {
37757622a8eSBarry Smith             ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k],(double)PetscRealPart(a->a[k]),(double)PetscImaginaryPart(a->a[k]));CHKERRQ(ierr);
378ca0704adSBarry Smith           } else if (PetscImaginaryPart(a->a[k]) < 0.0) {
37957622a8eSBarry Smith             ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k],(double)PetscRealPart(a->a[k]),-(double)PetscImaginaryPart(a->a[k]));CHKERRQ(ierr);
3802c990fa1SHong Zhang           } else {
38157622a8eSBarry Smith             ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k],(double)PetscRealPart(a->a[k]));CHKERRQ(ierr);
3822c990fa1SHong Zhang           }
3832c990fa1SHong Zhang #else
38457622a8eSBarry Smith           ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[k],(double)a->a[k]);CHKERRQ(ierr);
3852c990fa1SHong Zhang #endif
3862c990fa1SHong Zhang         }
3872c990fa1SHong Zhang         ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
3882c990fa1SHong Zhang       }
3892c990fa1SHong Zhang 
3902c990fa1SHong Zhang     } else { /* for non-factored matrix */
3910c74a584SJed Brown       for (i=0; i<a->mbs; i++) { /* for row block i */
3920c74a584SJed Brown         for (j=0; j<bs; j++) {   /* for row bs*i + j */
39377431f27SBarry Smith           ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr);
3940c74a584SJed Brown           for (k=a->i[i]; k<a->i[i+1]; k++) { /* for column block */
3950c74a584SJed Brown             for (l=0; l<bs; l++) {            /* for column */
39649b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX)
39749b5e25fSSatish Balay               if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0) {
39857622a8eSBarry Smith                 ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k]+l,
39957622a8eSBarry Smith                                               (double)PetscRealPart(a->a[bs2*k + l*bs + j]),(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
40049b5e25fSSatish Balay               } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0) {
40157622a8eSBarry Smith                 ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k]+l,
40257622a8eSBarry Smith                                               (double)PetscRealPart(a->a[bs2*k + l*bs + j]),-(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
40349b5e25fSSatish Balay               } else {
40457622a8eSBarry Smith                 ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
40549b5e25fSSatish Balay               }
40649b5e25fSSatish Balay #else
40757622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)a->a[bs2*k + l*bs + j]);CHKERRQ(ierr);
40849b5e25fSSatish Balay #endif
40949b5e25fSSatish Balay             }
41049b5e25fSSatish Balay           }
411b0a32e0cSBarry Smith           ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
41249b5e25fSSatish Balay         }
41349b5e25fSSatish Balay       }
4142c990fa1SHong Zhang     }
415d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
41649b5e25fSSatish Balay   }
417b0a32e0cSBarry Smith   ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
41849b5e25fSSatish Balay   PetscFunctionReturn(0);
41949b5e25fSSatish Balay }
42049b5e25fSSatish Balay 
4219804daf3SBarry Smith #include <petscdraw.h>
4224a2ae208SSatish Balay #undef __FUNCT__
4234a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_Draw_Zoom"
4246849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw_Zoom(PetscDraw draw,void *Aa)
42549b5e25fSSatish Balay {
42649b5e25fSSatish Balay   Mat            A = (Mat) Aa;
42749b5e25fSSatish Balay   Mat_SeqSBAIJ   *a=(Mat_SeqSBAIJ*)A->data;
4286849ba73SBarry Smith   PetscErrorCode ierr;
429d0f46423SBarry Smith   PetscInt       row,i,j,k,l,mbs=a->mbs,color,bs=A->rmap->bs,bs2=a->bs2;
43013f74950SBarry Smith   PetscMPIInt    rank;
43149b5e25fSSatish Balay   PetscReal      xl,yl,xr,yr,x_l,x_r,y_l,y_r;
43249b5e25fSSatish Balay   MatScalar      *aa;
43349b5e25fSSatish Balay   MPI_Comm       comm;
434b0a32e0cSBarry Smith   PetscViewer    viewer;
43549b5e25fSSatish Balay 
43649b5e25fSSatish Balay   PetscFunctionBegin;
43749b5e25fSSatish Balay   /*
43849b5e25fSSatish Balay     This is nasty. If this is called from an originally parallel matrix
43949b5e25fSSatish Balay     then all processes call this,but only the first has the matrix so the
44049b5e25fSSatish Balay     rest should return immediately.
44149b5e25fSSatish Balay   */
44249b5e25fSSatish Balay   ierr = PetscObjectGetComm((PetscObject)draw,&comm);CHKERRQ(ierr);
44349b5e25fSSatish Balay   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
44449b5e25fSSatish Balay   if (rank) PetscFunctionReturn(0);
44549b5e25fSSatish Balay 
44649b5e25fSSatish Balay   ierr = PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer);CHKERRQ(ierr);
44749b5e25fSSatish Balay 
448b0a32e0cSBarry Smith   ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr);
449b0a32e0cSBarry Smith   PetscDrawString(draw, .3*(xl+xr), .3*(yl+yr), PETSC_DRAW_BLACK, "symmetric");
45049b5e25fSSatish Balay 
45149b5e25fSSatish Balay   /* loop over matrix elements drawing boxes */
452b0a32e0cSBarry Smith   color = PETSC_DRAW_BLUE;
45349b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
45449b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
455d0f46423SBarry Smith       y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0;
45649b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
45749b5e25fSSatish Balay       aa  = a->a + j*bs2;
45849b5e25fSSatish Balay       for (k=0; k<bs; k++) {
45949b5e25fSSatish Balay         for (l=0; l<bs; l++) {
46049b5e25fSSatish Balay           if (PetscRealPart(*aa++) >=  0.) continue;
461b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
46249b5e25fSSatish Balay         }
46349b5e25fSSatish Balay       }
46449b5e25fSSatish Balay     }
46549b5e25fSSatish Balay   }
466b0a32e0cSBarry Smith   color = PETSC_DRAW_CYAN;
46749b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
46849b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
469d0f46423SBarry Smith       y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0;
47049b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
47149b5e25fSSatish Balay       aa = a->a + j*bs2;
47249b5e25fSSatish Balay       for (k=0; k<bs; k++) {
47349b5e25fSSatish Balay         for (l=0; l<bs; l++) {
47449b5e25fSSatish Balay           if (PetscRealPart(*aa++) != 0.) continue;
475b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
47649b5e25fSSatish Balay         }
47749b5e25fSSatish Balay       }
47849b5e25fSSatish Balay     }
47949b5e25fSSatish Balay   }
48049b5e25fSSatish Balay 
481b0a32e0cSBarry Smith   color = PETSC_DRAW_RED;
48249b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
48349b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
484d0f46423SBarry Smith       y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0;
48549b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
48649b5e25fSSatish Balay       aa = a->a + j*bs2;
48749b5e25fSSatish Balay       for (k=0; k<bs; k++) {
48849b5e25fSSatish Balay         for (l=0; l<bs; l++) {
48949b5e25fSSatish Balay           if (PetscRealPart(*aa++) <= 0.) continue;
490b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
49149b5e25fSSatish Balay         }
49249b5e25fSSatish Balay       }
49349b5e25fSSatish Balay     }
49449b5e25fSSatish Balay   }
49549b5e25fSSatish Balay   PetscFunctionReturn(0);
49649b5e25fSSatish Balay }
49749b5e25fSSatish Balay 
4984a2ae208SSatish Balay #undef __FUNCT__
4994a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_Draw"
5006849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw(Mat A,PetscViewer viewer)
50149b5e25fSSatish Balay {
502dfbe8321SBarry Smith   PetscErrorCode ierr;
50349b5e25fSSatish Balay   PetscReal      xl,yl,xr,yr,w,h;
504b0a32e0cSBarry Smith   PetscDraw      draw;
505ace3abfcSBarry Smith   PetscBool      isnull;
50649b5e25fSSatish Balay 
50749b5e25fSSatish Balay   PetscFunctionBegin;
508b0a32e0cSBarry Smith   ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
509b0a32e0cSBarry Smith   ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr); if (isnull) PetscFunctionReturn(0);
51049b5e25fSSatish Balay 
51149b5e25fSSatish Balay   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer);CHKERRQ(ierr);
512d0f46423SBarry Smith   xr   = A->rmap->N; yr = A->rmap->N; h = yr/10.0; w = xr/10.0;
51349b5e25fSSatish Balay   xr  += w;    yr += h;  xl = -w;     yl = -h;
514b0a32e0cSBarry Smith   ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr);
515b0a32e0cSBarry Smith   ierr = PetscDrawZoom(draw,MatView_SeqSBAIJ_Draw_Zoom,A);CHKERRQ(ierr);
5160298fd71SBarry Smith   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",NULL);CHKERRQ(ierr);
51749b5e25fSSatish Balay   PetscFunctionReturn(0);
51849b5e25fSSatish Balay }
51949b5e25fSSatish Balay 
5204a2ae208SSatish Balay #undef __FUNCT__
5214a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ"
522dfbe8321SBarry Smith PetscErrorCode MatView_SeqSBAIJ(Mat A,PetscViewer viewer)
52349b5e25fSSatish Balay {
524dfbe8321SBarry Smith   PetscErrorCode ierr;
525ace3abfcSBarry Smith   PetscBool      iascii,isdraw;
52608917f38SBarry Smith   FILE           *file = 0;
52749b5e25fSSatish Balay 
52849b5e25fSSatish Balay   PetscFunctionBegin;
529251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
530251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr);
53132077d6dSBarry Smith   if (iascii) {
53249b5e25fSSatish Balay     ierr = MatView_SeqSBAIJ_ASCII(A,viewer);CHKERRQ(ierr);
53349b5e25fSSatish Balay   } else if (isdraw) {
53449b5e25fSSatish Balay     ierr = MatView_SeqSBAIJ_Draw(A,viewer);CHKERRQ(ierr);
53549b5e25fSSatish Balay   } else {
536a5e6ed63SBarry Smith     Mat        B;
537ade3a672SBarry Smith     const char *matname;
538ceb03754SKris Buschelman     ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&B);CHKERRQ(ierr);
539ade3a672SBarry Smith     ierr = PetscObjectGetName((PetscObject)A,&matname);CHKERRQ(ierr);
540ade3a672SBarry Smith     ierr = PetscObjectSetName((PetscObject)B,matname);CHKERRQ(ierr);
541a5e6ed63SBarry Smith     ierr = MatView(B,viewer);CHKERRQ(ierr);
5426bf464f9SBarry Smith     ierr = MatDestroy(&B);CHKERRQ(ierr);
54308917f38SBarry Smith     ierr = PetscViewerBinaryGetInfoPointer(viewer,&file);CHKERRQ(ierr);
54408917f38SBarry Smith     if (file) {
54508917f38SBarry Smith       fprintf(file,"-matload_block_size %d\n",(int)A->rmap->bs);
54608917f38SBarry Smith     }
54749b5e25fSSatish Balay   }
54849b5e25fSSatish Balay   PetscFunctionReturn(0);
54949b5e25fSSatish Balay }
55049b5e25fSSatish Balay 
55149b5e25fSSatish Balay 
5524a2ae208SSatish Balay #undef __FUNCT__
5534a2ae208SSatish Balay #define __FUNCT__ "MatGetValues_SeqSBAIJ"
55413f74950SBarry Smith PetscErrorCode MatGetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],PetscScalar v[])
55549b5e25fSSatish Balay {
556045c9aa0SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
55713f74950SBarry Smith   PetscInt     *rp,k,low,high,t,row,nrow,i,col,l,*aj = a->j;
55813f74950SBarry Smith   PetscInt     *ai = a->i,*ailen = a->ilen;
559d0f46423SBarry Smith   PetscInt     brow,bcol,ridx,cidx,bs=A->rmap->bs,bs2=a->bs2;
56097e567efSBarry Smith   MatScalar    *ap,*aa = a->a;
56149b5e25fSSatish Balay 
56249b5e25fSSatish Balay   PetscFunctionBegin;
56349b5e25fSSatish Balay   for (k=0; k<m; k++) { /* loop over rows */
56449b5e25fSSatish Balay     row = im[k]; brow = row/bs;
565e32f2f54SBarry Smith     if (row < 0) {v += n; continue;} /* SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative row: %D",row); */
566e32f2f54SBarry 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);
56749b5e25fSSatish Balay     rp   = aj + ai[brow]; ap = aa + bs2*ai[brow];
56849b5e25fSSatish Balay     nrow = ailen[brow];
56949b5e25fSSatish Balay     for (l=0; l<n; l++) { /* loop over columns */
570e32f2f54SBarry Smith       if (in[l] < 0) {v++; continue;} /* SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column: %D",in[l]); */
571e32f2f54SBarry 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);
57249b5e25fSSatish Balay       col  = in[l];
57349b5e25fSSatish Balay       bcol = col/bs;
57449b5e25fSSatish Balay       cidx = col%bs;
57549b5e25fSSatish Balay       ridx = row%bs;
57649b5e25fSSatish Balay       high = nrow;
57749b5e25fSSatish Balay       low  = 0; /* assume unsorted */
57849b5e25fSSatish Balay       while (high-low > 5) {
57949b5e25fSSatish Balay         t = (low+high)/2;
58049b5e25fSSatish Balay         if (rp[t] > bcol) high = t;
58149b5e25fSSatish Balay         else              low  = t;
58249b5e25fSSatish Balay       }
58349b5e25fSSatish Balay       for (i=low; i<high; i++) {
58449b5e25fSSatish Balay         if (rp[i] > bcol) break;
58549b5e25fSSatish Balay         if (rp[i] == bcol) {
58649b5e25fSSatish Balay           *v++ = ap[bs2*i+bs*cidx+ridx];
58749b5e25fSSatish Balay           goto finished;
58849b5e25fSSatish Balay         }
58949b5e25fSSatish Balay       }
59097e567efSBarry Smith       *v++ = 0.0;
59149b5e25fSSatish Balay finished:;
59249b5e25fSSatish Balay     }
59349b5e25fSSatish Balay   }
59449b5e25fSSatish Balay   PetscFunctionReturn(0);
59549b5e25fSSatish Balay }
59649b5e25fSSatish Balay 
59749b5e25fSSatish Balay 
5984a2ae208SSatish Balay #undef __FUNCT__
5994a2ae208SSatish Balay #define __FUNCT__ "MatSetValuesBlocked_SeqSBAIJ"
60013f74950SBarry Smith PetscErrorCode MatSetValuesBlocked_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is)
60149b5e25fSSatish Balay {
6020880e062SHong Zhang   Mat_SeqSBAIJ      *a = (Mat_SeqSBAIJ*)A->data;
6036849ba73SBarry Smith   PetscErrorCode    ierr;
604e2ee6c50SBarry Smith   PetscInt          *rp,k,low,high,t,ii,jj,row,nrow,i,col,l,rmax,N,lastcol = -1;
60513f74950SBarry Smith   PetscInt          *imax      =a->imax,*ai=a->i,*ailen=a->ilen;
606d0f46423SBarry Smith   PetscInt          *aj        =a->j,nonew=a->nonew,bs2=a->bs2,bs=A->rmap->bs,stepval;
607ace3abfcSBarry Smith   PetscBool         roworiented=a->roworiented;
608dd6ea824SBarry Smith   const PetscScalar *value     = v;
609f15d580aSBarry Smith   MatScalar         *ap,*aa = a->a,*bap;
6100880e062SHong Zhang 
61149b5e25fSSatish Balay   PetscFunctionBegin;
61226fbe8dcSKarl Rupp   if (roworiented) stepval = (n-1)*bs;
61326fbe8dcSKarl Rupp   else stepval = (m-1)*bs;
61426fbe8dcSKarl Rupp 
6150880e062SHong Zhang   for (k=0; k<m; k++) { /* loop over added rows */
6160880e062SHong Zhang     row = im[k];
6170880e062SHong Zhang     if (row < 0) continue;
6182515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
619e32f2f54SBarry Smith     if (row >= a->mbs) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,a->mbs-1);
6200880e062SHong Zhang #endif
6210880e062SHong Zhang     rp   = aj + ai[row];
6220880e062SHong Zhang     ap   = aa + bs2*ai[row];
6230880e062SHong Zhang     rmax = imax[row];
6240880e062SHong Zhang     nrow = ailen[row];
6250880e062SHong Zhang     low  = 0;
626818f2c47SBarry Smith     high = nrow;
6270880e062SHong Zhang     for (l=0; l<n; l++) { /* loop over added columns */
6280880e062SHong Zhang       if (in[l] < 0) continue;
6290880e062SHong Zhang       col = in[l];
6302515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
631e32f2f54SBarry Smith       if (col >= a->nbs) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",col,a->nbs-1);
632b1823623SSatish Balay #endif
633b98bf0e1SJed Brown       if (col < row) {
63426fbe8dcSKarl Rupp         if (a->ignore_ltriangular) continue; /* ignore lower triangular block */
63526fbe8dcSKarl 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)");
636b98bf0e1SJed Brown       }
63726fbe8dcSKarl Rupp       if (roworiented) value = v + k*(stepval+bs)*bs + l*bs;
63826fbe8dcSKarl Rupp       else value = v + l*(stepval+bs)*bs + k*bs;
63926fbe8dcSKarl Rupp 
64026fbe8dcSKarl Rupp       if (col <= lastcol) low = 0;
64126fbe8dcSKarl Rupp       else high = nrow;
64226fbe8dcSKarl Rupp 
643e2ee6c50SBarry Smith       lastcol = col;
6440880e062SHong Zhang       while (high-low > 7) {
6450880e062SHong Zhang         t = (low+high)/2;
6460880e062SHong Zhang         if (rp[t] > col) high = t;
6470880e062SHong Zhang         else             low  = t;
6480880e062SHong Zhang       }
6490880e062SHong Zhang       for (i=low; i<high; i++) {
6500880e062SHong Zhang         if (rp[i] > col) break;
6510880e062SHong Zhang         if (rp[i] == col) {
6520880e062SHong Zhang           bap = ap +  bs2*i;
6530880e062SHong Zhang           if (roworiented) {
6540880e062SHong Zhang             if (is == ADD_VALUES) {
6550880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
6560880e062SHong Zhang                 for (jj=ii; jj<bs2; jj+=bs) {
6570880e062SHong Zhang                   bap[jj] += *value++;
6580880e062SHong Zhang                 }
6590880e062SHong Zhang               }
6600880e062SHong Zhang             } else {
6610880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
6620880e062SHong Zhang                 for (jj=ii; jj<bs2; jj+=bs) {
6630880e062SHong Zhang                   bap[jj] = *value++;
6640880e062SHong Zhang                 }
6650880e062SHong Zhang                }
6660880e062SHong Zhang             }
6670880e062SHong Zhang           } else {
6680880e062SHong Zhang             if (is == ADD_VALUES) {
6690880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
6700880e062SHong Zhang                 for (jj=0; jj<bs; jj++) {
6710880e062SHong Zhang                   *bap++ += *value++;
6720880e062SHong Zhang                 }
6730880e062SHong Zhang               }
6740880e062SHong Zhang             } else {
6750880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
6760880e062SHong Zhang                 for (jj=0; jj<bs; jj++) {
6770880e062SHong Zhang                   *bap++  = *value++;
6780880e062SHong Zhang                 }
6790880e062SHong Zhang               }
6800880e062SHong Zhang             }
6810880e062SHong Zhang           }
6820880e062SHong Zhang           goto noinsert2;
6830880e062SHong Zhang         }
6840880e062SHong Zhang       }
6850880e062SHong Zhang       if (nonew == 1) goto noinsert2;
686e32f2f54SBarry Smith       if (nonew == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col);
687fef13f97SBarry Smith       MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,row,col,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar);
688c03d1d03SSatish Balay       N = nrow++ - 1; high++;
6890880e062SHong Zhang       /* shift up all the later entries in this row */
6900880e062SHong Zhang       for (ii=N; ii>=i; ii--) {
6910880e062SHong Zhang         rp[ii+1] = rp[ii];
6920880e062SHong Zhang         ierr     = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr);
6930880e062SHong Zhang       }
6940880e062SHong Zhang       if (N >= i) {
6950880e062SHong Zhang         ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr);
6960880e062SHong Zhang       }
6970880e062SHong Zhang       rp[i] = col;
6980880e062SHong Zhang       bap   = ap +  bs2*i;
6990880e062SHong Zhang       if (roworiented) {
7000880e062SHong Zhang         for (ii=0; ii<bs; ii++,value+=stepval) {
7010880e062SHong Zhang           for (jj=ii; jj<bs2; jj+=bs) {
7020880e062SHong Zhang             bap[jj] = *value++;
7030880e062SHong Zhang           }
7040880e062SHong Zhang         }
7050880e062SHong Zhang       } else {
7060880e062SHong Zhang         for (ii=0; ii<bs; ii++,value+=stepval) {
7070880e062SHong Zhang           for (jj=0; jj<bs; jj++) {
7080880e062SHong Zhang             *bap++ = *value++;
7090880e062SHong Zhang           }
7100880e062SHong Zhang         }
7110880e062SHong Zhang        }
7120880e062SHong Zhang     noinsert2:;
7130880e062SHong Zhang       low = i;
7140880e062SHong Zhang     }
7150880e062SHong Zhang     ailen[row] = nrow;
7160880e062SHong Zhang   }
7170880e062SHong Zhang   PetscFunctionReturn(0);
71849b5e25fSSatish Balay }
71949b5e25fSSatish Balay 
72064831d72SBarry Smith /*
72164831d72SBarry Smith     This is not yet used
72264831d72SBarry Smith */
7234a2ae208SSatish Balay #undef __FUNCT__
7244108e4d5SBarry Smith #define __FUNCT__ "MatAssemblyEnd_SeqSBAIJ_SeqAIJ_Inode"
7254108e4d5SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ_SeqAIJ_Inode(Mat A)
7260def2e27SBarry Smith {
7270def2e27SBarry Smith   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
7280def2e27SBarry Smith   PetscErrorCode ierr;
7290def2e27SBarry Smith   const PetscInt *ai = a->i, *aj = a->j,*cols;
7300def2e27SBarry Smith   PetscInt       i   = 0,j,blk_size,m = A->rmap->n,node_count = 0,nzx,nzy,*ns,row,nz,cnt,cnt2,*counts;
731ace3abfcSBarry Smith   PetscBool      flag;
7320def2e27SBarry Smith 
7330def2e27SBarry Smith   PetscFunctionBegin;
734785e854fSJed Brown   ierr = PetscMalloc1(m,&ns);CHKERRQ(ierr);
7350def2e27SBarry Smith   while (i < m) {
7360def2e27SBarry Smith     nzx = ai[i+1] - ai[i];       /* Number of nonzeros */
7370def2e27SBarry Smith     /* Limits the number of elements in a node to 'a->inode.limit' */
7380def2e27SBarry Smith     for (j=i+1,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) {
7390def2e27SBarry Smith       nzy = ai[j+1] - ai[j];
7400def2e27SBarry Smith       if (nzy != (nzx - j + i)) break;
7410def2e27SBarry Smith       ierr = PetscMemcmp(aj + ai[i] + j - i,aj + ai[j],nzy*sizeof(PetscInt),&flag);CHKERRQ(ierr);
7420def2e27SBarry Smith       if (!flag) break;
7430def2e27SBarry Smith     }
7440def2e27SBarry Smith     ns[node_count++] = blk_size;
74526fbe8dcSKarl Rupp 
7460def2e27SBarry Smith     i = j;
7470def2e27SBarry Smith   }
7480def2e27SBarry Smith   if (!a->inode.size && m && node_count > .9*m) {
7490def2e27SBarry Smith     ierr = PetscFree(ns);CHKERRQ(ierr);
7500def2e27SBarry Smith     ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr);
7510def2e27SBarry Smith   } else {
7520def2e27SBarry Smith     a->inode.node_count = node_count;
75326fbe8dcSKarl Rupp 
754785e854fSJed Brown     ierr = PetscMalloc1(node_count,&a->inode.size);CHKERRQ(ierr);
7553bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)A,node_count*sizeof(PetscInt));CHKERRQ(ierr);
75622d28d08SBarry Smith     ierr = PetscMemcpy(a->inode.size,ns,node_count*sizeof(PetscInt));CHKERRQ(ierr);
7570def2e27SBarry Smith     ierr = PetscFree(ns);CHKERRQ(ierr);
7580def2e27SBarry Smith     ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr);
7590def2e27SBarry Smith 
7600def2e27SBarry Smith     /* count collections of adjacent columns in each inode */
7610def2e27SBarry Smith     row = 0;
7620def2e27SBarry Smith     cnt = 0;
7630def2e27SBarry Smith     for (i=0; i<node_count; i++) {
7640def2e27SBarry Smith       cols = aj + ai[row] + a->inode.size[i];
7650def2e27SBarry Smith       nz   = ai[row+1] - ai[row] - a->inode.size[i];
7660def2e27SBarry Smith       for (j=1; j<nz; j++) {
76726fbe8dcSKarl Rupp         if (cols[j] != cols[j-1]+1) cnt++;
7680def2e27SBarry Smith       }
7690def2e27SBarry Smith       cnt++;
7700def2e27SBarry Smith       row += a->inode.size[i];
7710def2e27SBarry Smith     }
772785e854fSJed Brown     ierr = PetscMalloc1(2*cnt,&counts);CHKERRQ(ierr);
7730def2e27SBarry Smith     cnt  = 0;
7740def2e27SBarry Smith     row  = 0;
7750def2e27SBarry Smith     for (i=0; i<node_count; i++) {
7760def2e27SBarry Smith       cols = aj + ai[row] + a->inode.size[i];
7770def2e27SBarry Smith       counts[2*cnt] = cols[0];
7780def2e27SBarry Smith       nz   = ai[row+1] - ai[row] - a->inode.size[i];
7790def2e27SBarry Smith       cnt2 = 1;
7800def2e27SBarry Smith       for (j=1; j<nz; j++) {
7810def2e27SBarry Smith         if (cols[j] != cols[j-1]+1) {
7820def2e27SBarry Smith           counts[2*(cnt++)+1] = cnt2;
7830def2e27SBarry Smith           counts[2*cnt]       = cols[j];
7840def2e27SBarry Smith           cnt2 = 1;
7850def2e27SBarry Smith         } else cnt2++;
7860def2e27SBarry Smith       }
7870def2e27SBarry Smith       counts[2*(cnt++)+1] = cnt2;
7880def2e27SBarry Smith       row += a->inode.size[i];
7890def2e27SBarry Smith     }
79022d28d08SBarry Smith     ierr = PetscIntView(2*cnt,counts,0);CHKERRQ(ierr);
7910def2e27SBarry Smith   }
79238702af4SBarry Smith   PetscFunctionReturn(0);
79338702af4SBarry Smith }
79438702af4SBarry Smith 
79538702af4SBarry Smith #undef __FUNCT__
7964a2ae208SSatish Balay #define __FUNCT__ "MatAssemblyEnd_SeqSBAIJ"
797dfbe8321SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ(Mat A,MatAssemblyType mode)
79849b5e25fSSatish Balay {
79949b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
8006849ba73SBarry Smith   PetscErrorCode ierr;
80113f74950SBarry Smith   PetscInt       fshift = 0,i,j,*ai = a->i,*aj = a->j,*imax = a->imax;
802d0f46423SBarry Smith   PetscInt       m      = A->rmap->N,*ip,N,*ailen = a->ilen;
80313f74950SBarry Smith   PetscInt       mbs    = a->mbs,bs2 = a->bs2,rmax = 0;
80449b5e25fSSatish Balay   MatScalar      *aa    = a->a,*ap;
80549b5e25fSSatish Balay 
80649b5e25fSSatish Balay   PetscFunctionBegin;
80749b5e25fSSatish Balay   if (mode == MAT_FLUSH_ASSEMBLY) PetscFunctionReturn(0);
80849b5e25fSSatish Balay 
80949b5e25fSSatish Balay   if (m) rmax = ailen[0];
81049b5e25fSSatish Balay   for (i=1; i<mbs; i++) {
81149b5e25fSSatish Balay     /* move each row back by the amount of empty slots (fshift) before it*/
81249b5e25fSSatish Balay     fshift += imax[i-1] - ailen[i-1];
81349b5e25fSSatish Balay     rmax    = PetscMax(rmax,ailen[i]);
81449b5e25fSSatish Balay     if (fshift) {
81549b5e25fSSatish Balay       ip = aj + ai[i]; ap = aa + bs2*ai[i];
81649b5e25fSSatish Balay       N  = ailen[i];
81749b5e25fSSatish Balay       for (j=0; j<N; j++) {
81849b5e25fSSatish Balay         ip[j-fshift] = ip[j];
81949b5e25fSSatish Balay         ierr         = PetscMemcpy(ap+(j-fshift)*bs2,ap+j*bs2,bs2*sizeof(MatScalar));CHKERRQ(ierr);
82049b5e25fSSatish Balay       }
82149b5e25fSSatish Balay     }
82249b5e25fSSatish Balay     ai[i] = ai[i-1] + ailen[i-1];
82349b5e25fSSatish Balay   }
82449b5e25fSSatish Balay   if (mbs) {
82549b5e25fSSatish Balay     fshift += imax[mbs-1] - ailen[mbs-1];
82649b5e25fSSatish Balay     ai[mbs] = ai[mbs-1] + ailen[mbs-1];
82749b5e25fSSatish Balay   }
82849b5e25fSSatish Balay   /* reset ilen and imax for each row */
82949b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
83049b5e25fSSatish Balay     ailen[i] = imax[i] = ai[i+1] - ai[i];
83149b5e25fSSatish Balay   }
8326c6c5352SBarry Smith   a->nz = ai[mbs];
83349b5e25fSSatish Balay 
834b424e231SHong Zhang   /* diagonals may have moved, reset it */
835b424e231SHong Zhang   if (a->diag) {
8362ed38d0bSJed Brown     ierr = PetscMemcpy(a->diag,ai,mbs*sizeof(PetscInt));CHKERRQ(ierr);
83749b5e25fSSatish Balay   }
83826fbe8dcSKarl 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);
83926fbe8dcSKarl Rupp 
840d0f46423SBarry 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);
841ae15b995SBarry Smith   ierr = PetscInfo1(A,"Number of mallocs during MatSetValues is %D\n",a->reallocs);CHKERRQ(ierr);
842ae15b995SBarry Smith   ierr = PetscInfo1(A,"Most nonzeros blocks in any row is %D\n",rmax);CHKERRQ(ierr);
84326fbe8dcSKarl Rupp 
8448e58a170SBarry Smith   A->info.mallocs    += a->reallocs;
84549b5e25fSSatish Balay   a->reallocs         = 0;
84649b5e25fSSatish Balay   A->info.nz_unneeded = (PetscReal)fshift*bs2;
847061b2667SBarry Smith   a->idiagvalid       = PETSC_FALSE;
84838702af4SBarry Smith 
84938702af4SBarry Smith   if (A->cmap->n < 65536 && A->cmap->bs == 1) {
85044e1c64aSLisandro Dalcin     if (a->jshort && a->free_jshort) {
85117803ae8SHong Zhang       /* when matrix data structure is changed, previous jshort must be replaced */
85217803ae8SHong Zhang       ierr = PetscFree(a->jshort);CHKERRQ(ierr);
85317803ae8SHong Zhang     }
854785e854fSJed Brown     ierr = PetscMalloc1(a->i[A->rmap->n],&a->jshort);CHKERRQ(ierr);
8553bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)A,a->i[A->rmap->n]*sizeof(unsigned short));CHKERRQ(ierr);
85638702af4SBarry Smith     for (i=0; i<a->i[A->rmap->n]; i++) a->jshort[i] = a->j[i];
85738702af4SBarry Smith     A->ops->mult   = MatMult_SeqSBAIJ_1_ushort;
85841f059aeSBarry Smith     A->ops->sor    = MatSOR_SeqSBAIJ_ushort;
8594da8f245SBarry Smith     a->free_jshort = PETSC_TRUE;
86038702af4SBarry Smith   }
86149b5e25fSSatish Balay   PetscFunctionReturn(0);
86249b5e25fSSatish Balay }
86349b5e25fSSatish Balay 
86449b5e25fSSatish Balay /*
86549b5e25fSSatish Balay    This function returns an array of flags which indicate the locations of contiguous
86649b5e25fSSatish Balay    blocks that should be zeroed. for eg: if bs = 3  and is = [0,1,2,3,5,6,7,8,9]
86749b5e25fSSatish Balay    then the resulting sizes = [3,1,1,3,1] correspondig to sets [(0,1,2),(3),(5),(6,7,8),(9)]
86849b5e25fSSatish Balay    Assume: sizes should be long enough to hold all the values.
86949b5e25fSSatish Balay */
8704a2ae208SSatish Balay #undef __FUNCT__
8714a2ae208SSatish Balay #define __FUNCT__ "MatZeroRows_SeqSBAIJ_Check_Blocks"
87213f74950SBarry Smith PetscErrorCode MatZeroRows_SeqSBAIJ_Check_Blocks(PetscInt idx[],PetscInt n,PetscInt bs,PetscInt sizes[], PetscInt *bs_max)
87349b5e25fSSatish Balay {
87413f74950SBarry Smith   PetscInt  i,j,k,row;
875ace3abfcSBarry Smith   PetscBool flg;
87649b5e25fSSatish Balay 
87749b5e25fSSatish Balay   PetscFunctionBegin;
87849b5e25fSSatish Balay   for (i=0,j=0; i<n; j++) {
87949b5e25fSSatish Balay     row = idx[i];
88049b5e25fSSatish Balay     if (row%bs!=0) { /* Not the begining of a block */
88149b5e25fSSatish Balay       sizes[j] = 1;
88249b5e25fSSatish Balay       i++;
88349b5e25fSSatish Balay     } else if (i+bs > n) { /* Beginning of a block, but complete block doesn't exist (at idx end) */
88449b5e25fSSatish Balay       sizes[j] = 1;         /* Also makes sure atleast 'bs' values exist for next else */
88549b5e25fSSatish Balay       i++;
88649b5e25fSSatish Balay     } else { /* Begining of the block, so check if the complete block exists */
88749b5e25fSSatish Balay       flg = PETSC_TRUE;
88849b5e25fSSatish Balay       for (k=1; k<bs; k++) {
88949b5e25fSSatish Balay         if (row+k != idx[i+k]) { /* break in the block */
89049b5e25fSSatish Balay           flg = PETSC_FALSE;
89149b5e25fSSatish Balay           break;
89249b5e25fSSatish Balay         }
89349b5e25fSSatish Balay       }
894abc0a331SBarry Smith       if (flg) { /* No break in the bs */
89549b5e25fSSatish Balay         sizes[j] = bs;
89649b5e25fSSatish Balay         i       += bs;
89749b5e25fSSatish Balay       } else {
89849b5e25fSSatish Balay         sizes[j] = 1;
89949b5e25fSSatish Balay         i++;
90049b5e25fSSatish Balay       }
90149b5e25fSSatish Balay     }
90249b5e25fSSatish Balay   }
90349b5e25fSSatish Balay   *bs_max = j;
90449b5e25fSSatish Balay   PetscFunctionReturn(0);
90549b5e25fSSatish Balay }
90649b5e25fSSatish Balay 
90749b5e25fSSatish Balay 
90849b5e25fSSatish Balay /* Only add/insert a(i,j) with i<=j (blocks).
90949b5e25fSSatish Balay    Any a(i,j) with i>j input by user is ingored.
91049b5e25fSSatish Balay */
91149b5e25fSSatish Balay 
9124a2ae208SSatish Balay #undef __FUNCT__
9134a2ae208SSatish Balay #define __FUNCT__ "MatSetValues_SeqSBAIJ"
91413f74950SBarry Smith PetscErrorCode MatSetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is)
91549b5e25fSSatish Balay {
91649b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
9176849ba73SBarry Smith   PetscErrorCode ierr;
918e2ee6c50SBarry Smith   PetscInt       *rp,k,low,high,t,ii,row,nrow,i,col,l,rmax,N,lastcol = -1;
91913f74950SBarry Smith   PetscInt       *imax=a->imax,*ai=a->i,*ailen=a->ilen,roworiented=a->roworiented;
920d0f46423SBarry Smith   PetscInt       *aj  =a->j,nonew=a->nonew,bs=A->rmap->bs,brow,bcol;
92113f74950SBarry Smith   PetscInt       ridx,cidx,bs2=a->bs2;
92249b5e25fSSatish Balay   MatScalar      *ap,value,*aa=a->a,*bap;
92349b5e25fSSatish Balay 
92449b5e25fSSatish Balay   PetscFunctionBegin;
92549b5e25fSSatish Balay   for (k=0; k<m; k++) { /* loop over added rows */
92649b5e25fSSatish Balay     row  = im[k];       /* row number */
92749b5e25fSSatish Balay     brow = row/bs;      /* block row number */
92849b5e25fSSatish Balay     if (row < 0) continue;
9292515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
930e32f2f54SBarry 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);
93149b5e25fSSatish Balay #endif
93249b5e25fSSatish Balay     rp   = aj + ai[brow]; /*ptr to beginning of column value of the row block*/
93349b5e25fSSatish Balay     ap   = aa + bs2*ai[brow]; /*ptr to beginning of element value of the row block*/
93449b5e25fSSatish Balay     rmax = imax[brow];  /* maximum space allocated for this row */
93549b5e25fSSatish Balay     nrow = ailen[brow]; /* actual length of this row */
93649b5e25fSSatish Balay     low  = 0;
93749b5e25fSSatish Balay 
93849b5e25fSSatish Balay     for (l=0; l<n; l++) { /* loop over added columns */
93949b5e25fSSatish Balay       if (in[l] < 0) continue;
9402515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
941e32f2f54SBarry 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);
94249b5e25fSSatish Balay #endif
94349b5e25fSSatish Balay       col  = in[l];
94449b5e25fSSatish Balay       bcol = col/bs;              /* block col number */
94549b5e25fSSatish Balay 
946941593c8SHong Zhang       if (brow > bcol) {
94726fbe8dcSKarl Rupp         if (a->ignore_ltriangular) continue; /* ignore lower triangular values */
94826fbe8dcSKarl 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)");
949941593c8SHong Zhang       }
950f4989cb3SHong Zhang 
95149b5e25fSSatish Balay       ridx = row % bs; cidx = col % bs; /*row and col index inside the block */
9528549e402SHong Zhang       if ((brow==bcol && ridx<=cidx) || (brow<bcol)) {
95349b5e25fSSatish Balay         /* element value a(k,l) */
95426fbe8dcSKarl Rupp         if (roworiented) value = v[l + k*n];
95526fbe8dcSKarl Rupp         else value = v[k + l*m];
95649b5e25fSSatish Balay 
95749b5e25fSSatish Balay         /* move pointer bap to a(k,l) quickly and add/insert value */
95826fbe8dcSKarl Rupp         if (col <= lastcol) low = 0;
95926fbe8dcSKarl Rupp         high = nrow;
960e2ee6c50SBarry Smith         lastcol = col;
96149b5e25fSSatish Balay         while (high-low > 7) {
96249b5e25fSSatish Balay           t = (low+high)/2;
96349b5e25fSSatish Balay           if (rp[t] > bcol) high = t;
96449b5e25fSSatish Balay           else              low  = t;
96549b5e25fSSatish Balay         }
96649b5e25fSSatish Balay         for (i=low; i<high; i++) {
96749b5e25fSSatish Balay           if (rp[i] > bcol) break;
96849b5e25fSSatish Balay           if (rp[i] == bcol) {
96949b5e25fSSatish Balay             bap = ap +  bs2*i + bs*cidx + ridx;
97049b5e25fSSatish Balay             if (is == ADD_VALUES) *bap += value;
97149b5e25fSSatish Balay             else                  *bap  = value;
9728549e402SHong Zhang             /* for diag block, add/insert its symmetric element a(cidx,ridx) */
9738549e402SHong Zhang             if (brow == bcol && ridx < cidx) {
9748549e402SHong Zhang               bap = ap +  bs2*i + bs*ridx + cidx;
9758549e402SHong Zhang               if (is == ADD_VALUES) *bap += value;
9768549e402SHong Zhang               else                  *bap  = value;
9778549e402SHong Zhang             }
97849b5e25fSSatish Balay             goto noinsert1;
97949b5e25fSSatish Balay           }
98049b5e25fSSatish Balay         }
98149b5e25fSSatish Balay 
98249b5e25fSSatish Balay         if (nonew == 1) goto noinsert1;
983e32f2f54SBarry Smith         if (nonew == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col);
984fef13f97SBarry Smith         MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,brow,bcol,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar);
98549b5e25fSSatish Balay 
986c03d1d03SSatish Balay         N = nrow++ - 1; high++;
98749b5e25fSSatish Balay         /* shift up all the later entries in this row */
98849b5e25fSSatish Balay         for (ii=N; ii>=i; ii--) {
98949b5e25fSSatish Balay           rp[ii+1] = rp[ii];
99049b5e25fSSatish Balay           ierr     = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr);
99149b5e25fSSatish Balay         }
99249b5e25fSSatish Balay         if (N>=i) {
99349b5e25fSSatish Balay           ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr);
99449b5e25fSSatish Balay         }
99549b5e25fSSatish Balay         rp[i]                      = bcol;
99649b5e25fSSatish Balay         ap[bs2*i + bs*cidx + ridx] = value;
997e56f5c9eSBarry Smith         A->nonzerostate++;
99849b5e25fSSatish Balay noinsert1:;
99949b5e25fSSatish Balay         low = i;
10008549e402SHong Zhang       }
100149b5e25fSSatish Balay     }   /* end of loop over added columns */
100249b5e25fSSatish Balay     ailen[brow] = nrow;
100349b5e25fSSatish Balay   }   /* end of loop over added rows */
100449b5e25fSSatish Balay   PetscFunctionReturn(0);
100549b5e25fSSatish Balay }
100649b5e25fSSatish Balay 
10074a2ae208SSatish Balay #undef __FUNCT__
10084d101231SSatish Balay #define __FUNCT__ "MatICCFactor_SeqSBAIJ"
10090481f469SBarry Smith PetscErrorCode MatICCFactor_SeqSBAIJ(Mat inA,IS row,const MatFactorInfo *info)
101049b5e25fSSatish Balay {
10114ccecd49SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)inA->data;
101249b5e25fSSatish Balay   Mat            outA;
1013dfbe8321SBarry Smith   PetscErrorCode ierr;
1014ace3abfcSBarry Smith   PetscBool      row_identity;
101549b5e25fSSatish Balay 
101649b5e25fSSatish Balay   PetscFunctionBegin;
1017e32f2f54SBarry Smith   if (info->levels != 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only levels=0 is supported for in-place icc");
1018c84f5b01SHong Zhang   ierr = ISIdentity(row,&row_identity);CHKERRQ(ierr);
1019e32f2f54SBarry Smith   if (!row_identity) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix reordering is not supported");
1020e32f2f54SBarry 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()! */
1021c84f5b01SHong Zhang 
102249b5e25fSSatish Balay   outA            = inA;
1023d5f3da31SBarry Smith   inA->factortype = MAT_FACTOR_ICC;
102449b5e25fSSatish Balay 
10251a3463dfSHong Zhang   ierr = MatMarkDiagonal_SeqSBAIJ(inA);CHKERRQ(ierr);
1026d595f711SHong Zhang   ierr = MatSeqSBAIJSetNumericFactorization_inplace(inA,row_identity);CHKERRQ(ierr);
102749b5e25fSSatish Balay 
1028c3122656SLisandro Dalcin   ierr   = PetscObjectReference((PetscObject)row);CHKERRQ(ierr);
10296bf464f9SBarry Smith   ierr   = ISDestroy(&a->row);CHKERRQ(ierr);
1030c84f5b01SHong Zhang   a->row = row;
1031c3122656SLisandro Dalcin   ierr   = PetscObjectReference((PetscObject)row);CHKERRQ(ierr);
10326bf464f9SBarry Smith   ierr   = ISDestroy(&a->col);CHKERRQ(ierr);
1033c84f5b01SHong Zhang   a->col = row;
1034c84f5b01SHong Zhang 
1035c84f5b01SHong Zhang   /* Create the invert permutation so that it can be used in MatCholeskyFactorNumeric() */
1036c84f5b01SHong Zhang   if (a->icol) {ierr = ISInvertPermutation(row,PETSC_DECIDE, &a->icol);CHKERRQ(ierr);}
10373bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)inA,(PetscObject)a->icol);CHKERRQ(ierr);
103849b5e25fSSatish Balay 
103949b5e25fSSatish Balay   if (!a->solve_work) {
1040785e854fSJed Brown     ierr = PetscMalloc1((inA->rmap->N+inA->rmap->bs),&a->solve_work);CHKERRQ(ierr);
10413bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)inA,(inA->rmap->N+inA->rmap->bs)*sizeof(PetscScalar));CHKERRQ(ierr);
104249b5e25fSSatish Balay   }
104349b5e25fSSatish Balay 
1044719d5645SBarry Smith   ierr = MatCholeskyFactorNumeric(outA,inA,info);CHKERRQ(ierr);
104549b5e25fSSatish Balay   PetscFunctionReturn(0);
104649b5e25fSSatish Balay }
1047950f1e5bSHong Zhang 
10484a2ae208SSatish Balay #undef __FUNCT__
10494a2ae208SSatish Balay #define __FUNCT__ "MatSeqSBAIJSetColumnIndices_SeqSBAIJ"
10507087cfbeSBarry Smith PetscErrorCode  MatSeqSBAIJSetColumnIndices_SeqSBAIJ(Mat mat,PetscInt *indices)
105149b5e25fSSatish Balay {
1052045c9aa0SHong Zhang   Mat_SeqSBAIJ   *baij = (Mat_SeqSBAIJ*)mat->data;
105313f74950SBarry Smith   PetscInt       i,nz,n;
10547827cd58SJed Brown   PetscErrorCode ierr;
105549b5e25fSSatish Balay 
105649b5e25fSSatish Balay   PetscFunctionBegin;
10576c6c5352SBarry Smith   nz = baij->maxnz;
1058d0f46423SBarry Smith   n  = mat->cmap->n;
105926fbe8dcSKarl Rupp   for (i=0; i<nz; i++) baij->j[i] = indices[i];
106026fbe8dcSKarl Rupp 
10616c6c5352SBarry Smith   baij->nz = nz;
106226fbe8dcSKarl Rupp   for (i=0; i<n; i++) baij->ilen[i] = baij->imax[i];
106326fbe8dcSKarl Rupp 
10647827cd58SJed Brown   ierr = MatSetOption(mat,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
106549b5e25fSSatish Balay   PetscFunctionReturn(0);
106649b5e25fSSatish Balay }
106749b5e25fSSatish Balay 
10684a2ae208SSatish Balay #undef __FUNCT__
10694a2ae208SSatish Balay #define __FUNCT__ "MatSeqSBAIJSetColumnIndices"
107049b5e25fSSatish Balay /*@
107119585528SSatish Balay   MatSeqSBAIJSetColumnIndices - Set the column indices for all the rows
107249b5e25fSSatish Balay   in the matrix.
107349b5e25fSSatish Balay 
107449b5e25fSSatish Balay   Input Parameters:
107519585528SSatish Balay   +  mat     - the SeqSBAIJ matrix
107649b5e25fSSatish Balay   -  indices - the column indices
107749b5e25fSSatish Balay 
107849b5e25fSSatish Balay   Level: advanced
107949b5e25fSSatish Balay 
108049b5e25fSSatish Balay   Notes:
108149b5e25fSSatish Balay   This can be called if you have precomputed the nonzero structure of the
108249b5e25fSSatish Balay   matrix and want to provide it to the matrix object to improve the performance
108349b5e25fSSatish Balay   of the MatSetValues() operation.
108449b5e25fSSatish Balay 
108549b5e25fSSatish Balay   You MUST have set the correct numbers of nonzeros per row in the call to
1086d1be2dadSMatthew Knepley   MatCreateSeqSBAIJ(), and the columns indices MUST be sorted.
108749b5e25fSSatish Balay 
1088ab9f2c04SSatish Balay   MUST be called before any calls to MatSetValues()
108949b5e25fSSatish Balay 
1090ab9f2c04SSatish Balay   .seealso: MatCreateSeqSBAIJ
109149b5e25fSSatish Balay @*/
10927087cfbeSBarry Smith PetscErrorCode  MatSeqSBAIJSetColumnIndices(Mat mat,PetscInt *indices)
109349b5e25fSSatish Balay {
10944ac538c5SBarry Smith   PetscErrorCode ierr;
109549b5e25fSSatish Balay 
109649b5e25fSSatish Balay   PetscFunctionBegin;
10970700a824SBarry Smith   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
10984482741eSBarry Smith   PetscValidPointer(indices,2);
10994ac538c5SBarry Smith   ierr = PetscUseMethod(mat,"MatSeqSBAIJSetColumnIndices_C",(Mat,PetscInt*),(mat,indices));CHKERRQ(ierr);
110049b5e25fSSatish Balay   PetscFunctionReturn(0);
110149b5e25fSSatish Balay }
110249b5e25fSSatish Balay 
11034a2ae208SSatish Balay #undef __FUNCT__
11043c896bc6SHong Zhang #define __FUNCT__ "MatCopy_SeqSBAIJ"
11053c896bc6SHong Zhang PetscErrorCode MatCopy_SeqSBAIJ(Mat A,Mat B,MatStructure str)
11063c896bc6SHong Zhang {
11073c896bc6SHong Zhang   PetscErrorCode ierr;
11083c896bc6SHong Zhang 
11093c896bc6SHong Zhang   PetscFunctionBegin;
11103c896bc6SHong Zhang   /* If the two matrices have the same copy implementation, use fast copy. */
11113c896bc6SHong Zhang   if (str == SAME_NONZERO_PATTERN && (A->ops->copy == B->ops->copy)) {
11123c896bc6SHong Zhang     Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
11133c896bc6SHong Zhang     Mat_SeqSBAIJ *b = (Mat_SeqSBAIJ*)B->data;
11143c896bc6SHong Zhang 
1115e7e72b3dSBarry 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");
1116d0f46423SBarry Smith     ierr = PetscMemcpy(b->a,a->a,(a->i[A->rmap->N])*sizeof(PetscScalar));CHKERRQ(ierr);
11173c896bc6SHong Zhang   } else {
1118f5edf698SHong Zhang     ierr = MatGetRowUpperTriangular(A);CHKERRQ(ierr);
11193c896bc6SHong Zhang     ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr);
1120f5edf698SHong Zhang     ierr = MatRestoreRowUpperTriangular(A);CHKERRQ(ierr);
11213c896bc6SHong Zhang   }
11223c896bc6SHong Zhang   PetscFunctionReturn(0);
11233c896bc6SHong Zhang }
11243c896bc6SHong Zhang 
11253c896bc6SHong Zhang #undef __FUNCT__
11264994cf47SJed Brown #define __FUNCT__ "MatSetUp_SeqSBAIJ"
11274994cf47SJed Brown PetscErrorCode MatSetUp_SeqSBAIJ(Mat A)
1128273d9f13SBarry Smith {
1129dfbe8321SBarry Smith   PetscErrorCode ierr;
1130273d9f13SBarry Smith 
1131273d9f13SBarry Smith   PetscFunctionBegin;
1132535b19f3SBarry Smith   ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(A,A->rmap->bs,PETSC_DEFAULT,0);CHKERRQ(ierr);
1133273d9f13SBarry Smith   PetscFunctionReturn(0);
1134273d9f13SBarry Smith }
1135273d9f13SBarry Smith 
1136a6ece127SHong Zhang #undef __FUNCT__
11378c778c55SBarry Smith #define __FUNCT__ "MatSeqSBAIJGetArray_SeqSBAIJ"
11388c778c55SBarry Smith PetscErrorCode MatSeqSBAIJGetArray_SeqSBAIJ(Mat A,PetscScalar *array[])
1139a6ece127SHong Zhang {
1140a6ece127SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
11415fd66863SKarl Rupp 
1142a6ece127SHong Zhang   PetscFunctionBegin;
1143a6ece127SHong Zhang   *array = a->a;
1144a6ece127SHong Zhang   PetscFunctionReturn(0);
1145a6ece127SHong Zhang }
1146a6ece127SHong Zhang 
1147a6ece127SHong Zhang #undef __FUNCT__
11488c778c55SBarry Smith #define __FUNCT__ "MatSeqSBAIJRestoreArray_SeqSBAIJ"
11498c778c55SBarry Smith PetscErrorCode MatSeqSBAIJRestoreArray_SeqSBAIJ(Mat A,PetscScalar *array[])
1150a6ece127SHong Zhang {
1151a6ece127SHong Zhang   PetscFunctionBegin;
1152a6ece127SHong Zhang   PetscFunctionReturn(0);
1153a6ece127SHong Zhang }
1154a6ece127SHong Zhang 
115542ee4b1aSHong Zhang #undef __FUNCT__
115652768537SHong Zhang #define __FUNCT__ "MatAXPYGetPreallocation_SeqSBAIJ"
115752768537SHong Zhang PetscErrorCode MatAXPYGetPreallocation_SeqSBAIJ(Mat Y,Mat X,PetscInt *nnz)
115852768537SHong Zhang {
1159b264fe52SHong Zhang   PetscInt       bs = Y->rmap->bs,mbs = Y->rmap->N/bs;
116052768537SHong Zhang   Mat_SeqSBAIJ   *x = (Mat_SeqSBAIJ*)X->data;
116152768537SHong Zhang   Mat_SeqSBAIJ   *y = (Mat_SeqSBAIJ*)Y->data;
1162b264fe52SHong Zhang   PetscErrorCode ierr;
116352768537SHong Zhang 
116452768537SHong Zhang   PetscFunctionBegin;
116552768537SHong Zhang   /* Set the number of nonzeros in the new matrix */
1166b264fe52SHong Zhang   ierr = MatAXPYGetPreallocation_SeqX_private(mbs,x->i,x->j,y->i,y->j,nnz);CHKERRQ(ierr);
116752768537SHong Zhang   PetscFunctionReturn(0);
116852768537SHong Zhang }
116952768537SHong Zhang 
117052768537SHong Zhang #undef __FUNCT__
117142ee4b1aSHong Zhang #define __FUNCT__ "MatAXPY_SeqSBAIJ"
1172f4df32b1SMatthew Knepley PetscErrorCode MatAXPY_SeqSBAIJ(Mat Y,PetscScalar a,Mat X,MatStructure str)
117342ee4b1aSHong Zhang {
117442ee4b1aSHong Zhang   Mat_SeqSBAIJ   *x=(Mat_SeqSBAIJ*)X->data, *y=(Mat_SeqSBAIJ*)Y->data;
1175dfbe8321SBarry Smith   PetscErrorCode ierr;
11767b2ba04bSHong Zhang   PetscInt       bs=Y->rmap->bs,bs2=bs*bs;
1177e838b9e7SJed Brown   PetscBLASInt   one = 1;
117842ee4b1aSHong Zhang 
117942ee4b1aSHong Zhang   PetscFunctionBegin;
118042ee4b1aSHong Zhang   if (str == SAME_NONZERO_PATTERN) {
1181f4df32b1SMatthew Knepley     PetscScalar  alpha = a;
1182c5df96a5SBarry Smith     PetscBLASInt bnz;
1183c5df96a5SBarry Smith     ierr = PetscBLASIntCast(x->nz*bs2,&bnz);CHKERRQ(ierr);
11848b83055fSJed Brown     PetscStackCallBLAS("BLASaxpy",BLASaxpy_(&bnz,&alpha,x->a,&one,y->a,&one));
1185a3fa217bSJose E. Roman     ierr = PetscObjectStateIncrease((PetscObject)Y);CHKERRQ(ierr);
1186c537a176SHong Zhang   } else if (str == SUBSET_NONZERO_PATTERN) { /* nonzeros of X is a subset of Y's */
11877b2ba04bSHong Zhang     ierr = MatSetOption(X,MAT_GETROW_UPPERTRIANGULAR,PETSC_TRUE);CHKERRQ(ierr);
11887b2ba04bSHong Zhang     ierr = MatAXPY_Basic(Y,a,X,str);CHKERRQ(ierr);
11897b2ba04bSHong Zhang     ierr = MatSetOption(X,MAT_GETROW_UPPERTRIANGULAR,PETSC_FALSE);CHKERRQ(ierr);
119042ee4b1aSHong Zhang   } else {
119152768537SHong Zhang     Mat      B;
119252768537SHong Zhang     PetscInt *nnz;
119352768537SHong Zhang     if (bs != X->rmap->bs) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Matrices must have same block size");
1194f5edf698SHong Zhang     ierr = MatGetRowUpperTriangular(X);CHKERRQ(ierr);
119552768537SHong Zhang     ierr = MatGetRowUpperTriangular(Y);CHKERRQ(ierr);
119652768537SHong Zhang     ierr = PetscMalloc1(Y->rmap->N,&nnz);CHKERRQ(ierr);
119752768537SHong Zhang     ierr = MatCreate(PetscObjectComm((PetscObject)Y),&B);CHKERRQ(ierr);
119852768537SHong Zhang     ierr = PetscObjectSetName((PetscObject)B,((PetscObject)Y)->name);CHKERRQ(ierr);
119952768537SHong Zhang     ierr = MatSetSizes(B,Y->rmap->n,Y->cmap->n,Y->rmap->N,Y->cmap->N);CHKERRQ(ierr);
120052768537SHong Zhang     ierr = MatSetBlockSizesFromMats(B,Y,Y);CHKERRQ(ierr);
120152768537SHong Zhang     ierr = MatSetType(B,(MatType) ((PetscObject)Y)->type_name);CHKERRQ(ierr);
120252768537SHong Zhang     ierr = MatAXPYGetPreallocation_SeqSBAIJ(Y,X,nnz);CHKERRQ(ierr);
120352768537SHong Zhang     ierr = MatSeqSBAIJSetPreallocation(B,bs,0,nnz);CHKERRQ(ierr);
120452768537SHong Zhang 
120552768537SHong Zhang     ierr = MatAXPY_BasicWithPreallocation(B,Y,a,X,str);CHKERRQ(ierr);
120652768537SHong Zhang 
120752768537SHong Zhang     ierr = MatHeaderReplace(Y,B);CHKERRQ(ierr);
120852768537SHong Zhang     ierr = PetscFree(nnz);CHKERRQ(ierr);
1209f5edf698SHong Zhang     ierr = MatRestoreRowUpperTriangular(X);CHKERRQ(ierr);
121052768537SHong Zhang     ierr = MatRestoreRowUpperTriangular(Y);CHKERRQ(ierr);
121142ee4b1aSHong Zhang   }
121242ee4b1aSHong Zhang   PetscFunctionReturn(0);
121342ee4b1aSHong Zhang }
121442ee4b1aSHong Zhang 
1215efcf0fc3SBarry Smith #undef __FUNCT__
1216efcf0fc3SBarry Smith #define __FUNCT__ "MatIsSymmetric_SeqSBAIJ"
1217ace3abfcSBarry Smith PetscErrorCode MatIsSymmetric_SeqSBAIJ(Mat A,PetscReal tol,PetscBool  *flg)
1218efcf0fc3SBarry Smith {
1219efcf0fc3SBarry Smith   PetscFunctionBegin;
1220efcf0fc3SBarry Smith   *flg = PETSC_TRUE;
1221efcf0fc3SBarry Smith   PetscFunctionReturn(0);
1222efcf0fc3SBarry Smith }
1223efcf0fc3SBarry Smith 
1224efcf0fc3SBarry Smith #undef __FUNCT__
1225efcf0fc3SBarry Smith #define __FUNCT__ "MatIsStructurallySymmetric_SeqSBAIJ"
1226ace3abfcSBarry Smith PetscErrorCode MatIsStructurallySymmetric_SeqSBAIJ(Mat A,PetscBool  *flg)
1227efcf0fc3SBarry Smith {
1228efcf0fc3SBarry Smith   PetscFunctionBegin;
1229efcf0fc3SBarry Smith   *flg = PETSC_TRUE;
1230efcf0fc3SBarry Smith   PetscFunctionReturn(0);
1231efcf0fc3SBarry Smith }
1232efcf0fc3SBarry Smith 
1233efcf0fc3SBarry Smith #undef __FUNCT__
1234efcf0fc3SBarry Smith #define __FUNCT__ "MatIsHermitian_SeqSBAIJ"
1235ace3abfcSBarry Smith PetscErrorCode MatIsHermitian_SeqSBAIJ(Mat A,PetscReal tol,PetscBool  *flg)
1236efcf0fc3SBarry Smith {
1237efcf0fc3SBarry Smith   PetscFunctionBegin;
1238efcf0fc3SBarry Smith   *flg = PETSC_FALSE;
1239efcf0fc3SBarry Smith   PetscFunctionReturn(0);
1240efcf0fc3SBarry Smith }
1241efcf0fc3SBarry Smith 
124299cafbc1SBarry Smith #undef __FUNCT__
124399cafbc1SBarry Smith #define __FUNCT__ "MatRealPart_SeqSBAIJ"
124499cafbc1SBarry Smith PetscErrorCode MatRealPart_SeqSBAIJ(Mat A)
124599cafbc1SBarry Smith {
124699cafbc1SBarry Smith   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
124799cafbc1SBarry Smith   PetscInt     i,nz = a->bs2*a->i[a->mbs];
1248dd6ea824SBarry Smith   MatScalar    *aa = a->a;
124999cafbc1SBarry Smith 
125099cafbc1SBarry Smith   PetscFunctionBegin;
125199cafbc1SBarry Smith   for (i=0; i<nz; i++) aa[i] = PetscRealPart(aa[i]);
125299cafbc1SBarry Smith   PetscFunctionReturn(0);
125399cafbc1SBarry Smith }
125499cafbc1SBarry Smith 
125599cafbc1SBarry Smith #undef __FUNCT__
125699cafbc1SBarry Smith #define __FUNCT__ "MatImaginaryPart_SeqSBAIJ"
125799cafbc1SBarry Smith PetscErrorCode MatImaginaryPart_SeqSBAIJ(Mat A)
125899cafbc1SBarry Smith {
125999cafbc1SBarry Smith   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
126099cafbc1SBarry Smith   PetscInt     i,nz = a->bs2*a->i[a->mbs];
1261dd6ea824SBarry Smith   MatScalar    *aa = a->a;
126299cafbc1SBarry Smith 
126399cafbc1SBarry Smith   PetscFunctionBegin;
126499cafbc1SBarry Smith   for (i=0; i<nz; i++) aa[i] = PetscImaginaryPart(aa[i]);
126599cafbc1SBarry Smith   PetscFunctionReturn(0);
126699cafbc1SBarry Smith }
126799cafbc1SBarry Smith 
12683bededecSBarry Smith #undef __FUNCT__
12693bededecSBarry Smith #define __FUNCT__ "MatZeroRowsColumns_SeqSBAIJ"
12703bededecSBarry Smith PetscErrorCode MatZeroRowsColumns_SeqSBAIJ(Mat A,PetscInt is_n,const PetscInt is_idx[],PetscScalar diag,Vec x, Vec b)
12713bededecSBarry Smith {
12723bededecSBarry Smith   Mat_SeqSBAIJ      *baij=(Mat_SeqSBAIJ*)A->data;
12733bededecSBarry Smith   PetscErrorCode    ierr;
12743bededecSBarry Smith   PetscInt          i,j,k,count;
12753bededecSBarry Smith   PetscInt          bs   =A->rmap->bs,bs2=baij->bs2,row,col;
12763bededecSBarry Smith   PetscScalar       zero = 0.0;
12773bededecSBarry Smith   MatScalar         *aa;
12783bededecSBarry Smith   const PetscScalar *xx;
12793bededecSBarry Smith   PetscScalar       *bb;
128056777dd2SBarry Smith   PetscBool         *zeroed,vecs = PETSC_FALSE;
12813bededecSBarry Smith 
12823bededecSBarry Smith   PetscFunctionBegin;
12833bededecSBarry Smith   /* fix right hand side if needed */
12843bededecSBarry Smith   if (x && b) {
12853bededecSBarry Smith     ierr = VecGetArrayRead(x,&xx);CHKERRQ(ierr);
12863bededecSBarry Smith     ierr = VecGetArray(b,&bb);CHKERRQ(ierr);
128756777dd2SBarry Smith     vecs = PETSC_TRUE;
12883bededecSBarry Smith   }
12893bededecSBarry Smith 
12903bededecSBarry Smith   /* zero the columns */
12911795a4d1SJed Brown   ierr = PetscCalloc1(A->rmap->n,&zeroed);CHKERRQ(ierr);
12923bededecSBarry Smith   for (i=0; i<is_n; i++) {
12933bededecSBarry 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]);
12943bededecSBarry Smith     zeroed[is_idx[i]] = PETSC_TRUE;
12953bededecSBarry Smith   }
129656777dd2SBarry Smith   if (vecs) {
129756777dd2SBarry Smith     for (i=0; i<A->rmap->N; i++) {
129856777dd2SBarry Smith       row = i/bs;
129956777dd2SBarry Smith       for (j=baij->i[row]; j<baij->i[row+1]; j++) {
130056777dd2SBarry Smith         for (k=0; k<bs; k++) {
130156777dd2SBarry Smith           col = bs*baij->j[j] + k;
130256777dd2SBarry Smith           if (col <= i) continue;
130356777dd2SBarry Smith           aa = ((MatScalar*)(baij->a)) + j*bs2 + (i%bs) + bs*k;
130426fbe8dcSKarl Rupp           if (!zeroed[i] && zeroed[col]) bb[i]   -= aa[0]*xx[col];
130526fbe8dcSKarl Rupp           if (zeroed[i] && !zeroed[col]) bb[col] -= aa[0]*xx[i];
130656777dd2SBarry Smith         }
130756777dd2SBarry Smith       }
130856777dd2SBarry Smith     }
130926fbe8dcSKarl Rupp     for (i=0; i<is_n; i++) bb[is_idx[i]] = diag*xx[is_idx[i]];
131056777dd2SBarry Smith   }
131156777dd2SBarry Smith 
13123bededecSBarry Smith   for (i=0; i<A->rmap->N; i++) {
13133bededecSBarry Smith     if (!zeroed[i]) {
13143bededecSBarry Smith       row = i/bs;
13153bededecSBarry Smith       for (j=baij->i[row]; j<baij->i[row+1]; j++) {
13163bededecSBarry Smith         for (k=0; k<bs; k++) {
13173bededecSBarry Smith           col = bs*baij->j[j] + k;
13183bededecSBarry Smith           if (zeroed[col]) {
13193bededecSBarry Smith             aa = ((MatScalar*)(baij->a)) + j*bs2 + (i%bs) + bs*k;
13203bededecSBarry Smith             aa[0] = 0.0;
13213bededecSBarry Smith           }
13223bededecSBarry Smith         }
13233bededecSBarry Smith       }
13243bededecSBarry Smith     }
13253bededecSBarry Smith   }
13263bededecSBarry Smith   ierr = PetscFree(zeroed);CHKERRQ(ierr);
132756777dd2SBarry Smith   if (vecs) {
132856777dd2SBarry Smith     ierr = VecRestoreArrayRead(x,&xx);CHKERRQ(ierr);
132956777dd2SBarry Smith     ierr = VecRestoreArray(b,&bb);CHKERRQ(ierr);
133056777dd2SBarry Smith   }
13313bededecSBarry Smith 
13323bededecSBarry Smith   /* zero the rows */
13333bededecSBarry Smith   for (i=0; i<is_n; i++) {
13343bededecSBarry Smith     row   = is_idx[i];
13353bededecSBarry Smith     count = (baij->i[row/bs +1] - baij->i[row/bs])*bs;
13363bededecSBarry Smith     aa    = ((MatScalar*)(baij->a)) + baij->i[row/bs]*bs2 + (row%bs);
13373bededecSBarry Smith     for (k=0; k<count; k++) {
13383bededecSBarry Smith       aa[0] =  zero;
13393bededecSBarry Smith       aa   += bs;
13403bededecSBarry Smith     }
13413bededecSBarry Smith     if (diag != 0.0) {
13423bededecSBarry Smith       ierr = (*A->ops->setvalues)(A,1,&row,1,&row,&diag,INSERT_VALUES);CHKERRQ(ierr);
13433bededecSBarry Smith     }
13443bededecSBarry Smith   }
13453bededecSBarry Smith   ierr = MatAssemblyEnd_SeqSBAIJ(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
13463bededecSBarry Smith   PetscFunctionReturn(0);
13473bededecSBarry Smith }
13483bededecSBarry Smith 
134949b5e25fSSatish Balay /* -------------------------------------------------------------------*/
13503964eb88SJed Brown static struct _MatOps MatOps_Values = {MatSetValues_SeqSBAIJ,
135149b5e25fSSatish Balay                                        MatGetRow_SeqSBAIJ,
135249b5e25fSSatish Balay                                        MatRestoreRow_SeqSBAIJ,
135349b5e25fSSatish Balay                                        MatMult_SeqSBAIJ_N,
135497304618SKris Buschelman                                /*  4*/ MatMultAdd_SeqSBAIJ_N,
1355431c96f7SBarry Smith                                        MatMult_SeqSBAIJ_N,       /* transpose versions are same as non-transpose versions */
1356e005ede5SBarry Smith                                        MatMultAdd_SeqSBAIJ_N,
1357db4efbfdSBarry Smith                                        0,
135849b5e25fSSatish Balay                                        0,
135949b5e25fSSatish Balay                                        0,
136097304618SKris Buschelman                                /* 10*/ 0,
136149b5e25fSSatish Balay                                        0,
1362c078aec8SLisandro Dalcin                                        MatCholeskyFactor_SeqSBAIJ,
136341f059aeSBarry Smith                                        MatSOR_SeqSBAIJ,
136449b5e25fSSatish Balay                                        MatTranspose_SeqSBAIJ,
136597304618SKris Buschelman                                /* 15*/ MatGetInfo_SeqSBAIJ,
136649b5e25fSSatish Balay                                        MatEqual_SeqSBAIJ,
136749b5e25fSSatish Balay                                        MatGetDiagonal_SeqSBAIJ,
136849b5e25fSSatish Balay                                        MatDiagonalScale_SeqSBAIJ,
136949b5e25fSSatish Balay                                        MatNorm_SeqSBAIJ,
137097304618SKris Buschelman                                /* 20*/ 0,
137149b5e25fSSatish Balay                                        MatAssemblyEnd_SeqSBAIJ,
137249b5e25fSSatish Balay                                        MatSetOption_SeqSBAIJ,
137349b5e25fSSatish Balay                                        MatZeroEntries_SeqSBAIJ,
1374d519adbfSMatthew Knepley                                /* 24*/ 0,
137549b5e25fSSatish Balay                                        0,
137649b5e25fSSatish Balay                                        0,
1377db4efbfdSBarry Smith                                        0,
1378db4efbfdSBarry Smith                                        0,
13794994cf47SJed Brown                                /* 29*/ MatSetUp_SeqSBAIJ,
1380c464158bSHong Zhang                                        0,
1381db4efbfdSBarry Smith                                        0,
13828c778c55SBarry Smith                                        0,
13838c778c55SBarry Smith                                        0,
1384d519adbfSMatthew Knepley                                /* 34*/ MatDuplicate_SeqSBAIJ,
1385719d5645SBarry Smith                                        0,
1386719d5645SBarry Smith                                        0,
138749b5e25fSSatish Balay                                        0,
1388c84f5b01SHong Zhang                                        MatICCFactor_SeqSBAIJ,
1389d519adbfSMatthew Knepley                                /* 39*/ MatAXPY_SeqSBAIJ,
139049b5e25fSSatish Balay                                        MatGetSubMatrices_SeqSBAIJ,
139149b5e25fSSatish Balay                                        MatIncreaseOverlap_SeqSBAIJ,
139249b5e25fSSatish Balay                                        MatGetValues_SeqSBAIJ,
13933c896bc6SHong Zhang                                        MatCopy_SeqSBAIJ,
1394d519adbfSMatthew Knepley                                /* 44*/ 0,
139549b5e25fSSatish Balay                                        MatScale_SeqSBAIJ,
139649b5e25fSSatish Balay                                        0,
139749b5e25fSSatish Balay                                        0,
13983bededecSBarry Smith                                        MatZeroRowsColumns_SeqSBAIJ,
1399f73d5cc4SBarry Smith                                /* 49*/ 0,
140049b5e25fSSatish Balay                                        MatGetRowIJ_SeqSBAIJ,
140149b5e25fSSatish Balay                                        MatRestoreRowIJ_SeqSBAIJ,
140249b5e25fSSatish Balay                                        0,
140349b5e25fSSatish Balay                                        0,
1404d519adbfSMatthew Knepley                                /* 54*/ 0,
140549b5e25fSSatish Balay                                        0,
140649b5e25fSSatish Balay                                        0,
140749b5e25fSSatish Balay                                        0,
140849b5e25fSSatish Balay                                        MatSetValuesBlocked_SeqSBAIJ,
1409d519adbfSMatthew Knepley                                /* 59*/ MatGetSubMatrix_SeqSBAIJ,
141049b5e25fSSatish Balay                                        0,
141149b5e25fSSatish Balay                                        0,
1412357abbc8SBarry Smith                                        0,
1413d959ec07SHong Zhang                                        0,
1414d519adbfSMatthew Knepley                                /* 64*/ 0,
1415d959ec07SHong Zhang                                        0,
1416d959ec07SHong Zhang                                        0,
1417d959ec07SHong Zhang                                        0,
1418d959ec07SHong Zhang                                        0,
1419d519adbfSMatthew Knepley                                /* 69*/ MatGetRowMaxAbs_SeqSBAIJ,
14203e0d88b5SBarry Smith                                        0,
14213e0d88b5SBarry Smith                                        0,
14223e0d88b5SBarry Smith                                        0,
14233e0d88b5SBarry Smith                                        0,
1424d519adbfSMatthew Knepley                                /* 74*/ 0,
14253e0d88b5SBarry Smith                                        0,
14263e0d88b5SBarry Smith                                        0,
14273e0d88b5SBarry Smith                                        0,
14283e0d88b5SBarry Smith                                        0,
1429d519adbfSMatthew Knepley                                /* 79*/ 0,
14303e0d88b5SBarry Smith                                        0,
14313e0d88b5SBarry Smith                                        0,
143297304618SKris Buschelman                                        MatGetInertia_SeqSBAIJ,
14335bba2384SShri Abhyankar                                        MatLoad_SeqSBAIJ,
1434d519adbfSMatthew Knepley                                /* 84*/ MatIsSymmetric_SeqSBAIJ,
1435865e5f61SKris Buschelman                                        MatIsHermitian_SeqSBAIJ,
1436efcf0fc3SBarry Smith                                        MatIsStructurallySymmetric_SeqSBAIJ,
1437865e5f61SKris Buschelman                                        0,
1438865e5f61SKris Buschelman                                        0,
1439d519adbfSMatthew Knepley                                /* 89*/ 0,
1440865e5f61SKris Buschelman                                        0,
1441865e5f61SKris Buschelman                                        0,
1442865e5f61SKris Buschelman                                        0,
1443865e5f61SKris Buschelman                                        0,
1444d519adbfSMatthew Knepley                                /* 94*/ 0,
1445865e5f61SKris Buschelman                                        0,
1446865e5f61SKris Buschelman                                        0,
144799cafbc1SBarry Smith                                        0,
144899cafbc1SBarry Smith                                        0,
1449d519adbfSMatthew Knepley                                /* 99*/ 0,
145099cafbc1SBarry Smith                                        0,
145199cafbc1SBarry Smith                                        0,
145299cafbc1SBarry Smith                                        0,
145399cafbc1SBarry Smith                                        0,
1454d519adbfSMatthew Knepley                                /*104*/ 0,
145599cafbc1SBarry Smith                                        MatRealPart_SeqSBAIJ,
1456f5edf698SHong Zhang                                        MatImaginaryPart_SeqSBAIJ,
1457f5edf698SHong Zhang                                        MatGetRowUpperTriangular_SeqSBAIJ,
14582af78befSBarry Smith                                        MatRestoreRowUpperTriangular_SeqSBAIJ,
1459d519adbfSMatthew Knepley                                /*109*/ 0,
14602af78befSBarry Smith                                        0,
14612af78befSBarry Smith                                        0,
14622af78befSBarry Smith                                        0,
1463547795f9SHong Zhang                                        MatMissingDiagonal_SeqSBAIJ,
1464547795f9SHong Zhang                                /*114*/ 0,
1465547795f9SHong Zhang                                        0,
1466547795f9SHong Zhang                                        0,
1467547795f9SHong Zhang                                        0,
1468547795f9SHong Zhang                                        0,
1469547795f9SHong Zhang                                /*119*/ 0,
1470547795f9SHong Zhang                                        0,
14712f480046SShri Abhyankar                                        0,
14723964eb88SJed Brown                                        0,
14733964eb88SJed Brown                                        0,
14743964eb88SJed Brown                                /*124*/ 0,
14753964eb88SJed Brown                                        0,
14763964eb88SJed Brown                                        0,
14773964eb88SJed Brown                                        0,
14783964eb88SJed Brown                                        0,
14793964eb88SJed Brown                                /*129*/ 0,
14803964eb88SJed Brown                                        0,
14813964eb88SJed Brown                                        0,
14823964eb88SJed Brown                                        0,
14833964eb88SJed Brown                                        0,
14843964eb88SJed Brown                                /*134*/ 0,
14853964eb88SJed Brown                                        0,
14863964eb88SJed Brown                                        0,
14873964eb88SJed Brown                                        0,
14883964eb88SJed Brown                                        0,
14893964eb88SJed Brown                                /*139*/ 0,
1490f9426fe0SMark Adams                                        0,
14915bba2384SShri Abhyankar                                        0
149299cafbc1SBarry Smith };
1493be1d678aSKris Buschelman 
14944a2ae208SSatish Balay #undef __FUNCT__
14954a2ae208SSatish Balay #define __FUNCT__ "MatStoreValues_SeqSBAIJ"
14967087cfbeSBarry Smith PetscErrorCode  MatStoreValues_SeqSBAIJ(Mat mat)
149749b5e25fSSatish Balay {
14984afc71dfSHong Zhang   Mat_SeqSBAIJ   *aij = (Mat_SeqSBAIJ*)mat->data;
1499d0f46423SBarry Smith   PetscInt       nz   = aij->i[mat->rmap->N]*mat->rmap->bs*aij->bs2;
1500dfbe8321SBarry Smith   PetscErrorCode ierr;
150149b5e25fSSatish Balay 
150249b5e25fSSatish Balay   PetscFunctionBegin;
1503e7e72b3dSBarry Smith   if (aij->nonew != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first");
150449b5e25fSSatish Balay 
150549b5e25fSSatish Balay   /* allocate space for values if not already there */
150649b5e25fSSatish Balay   if (!aij->saved_values) {
1507785e854fSJed Brown     ierr = PetscMalloc1((nz+1),&aij->saved_values);CHKERRQ(ierr);
150849b5e25fSSatish Balay   }
150949b5e25fSSatish Balay 
151049b5e25fSSatish Balay   /* copy values over */
151187828ca2SBarry Smith   ierr = PetscMemcpy(aij->saved_values,aij->a,nz*sizeof(PetscScalar));CHKERRQ(ierr);
151249b5e25fSSatish Balay   PetscFunctionReturn(0);
151349b5e25fSSatish Balay }
151449b5e25fSSatish Balay 
15154a2ae208SSatish Balay #undef __FUNCT__
15164a2ae208SSatish Balay #define __FUNCT__ "MatRetrieveValues_SeqSBAIJ"
15177087cfbeSBarry Smith PetscErrorCode  MatRetrieveValues_SeqSBAIJ(Mat mat)
151849b5e25fSSatish Balay {
15194afc71dfSHong Zhang   Mat_SeqSBAIJ   *aij = (Mat_SeqSBAIJ*)mat->data;
15206849ba73SBarry Smith   PetscErrorCode ierr;
1521d0f46423SBarry Smith   PetscInt       nz = aij->i[mat->rmap->N]*mat->rmap->bs*aij->bs2;
152249b5e25fSSatish Balay 
152349b5e25fSSatish Balay   PetscFunctionBegin;
1524e7e72b3dSBarry Smith   if (aij->nonew != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first");
1525e7e72b3dSBarry Smith   if (!aij->saved_values) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatStoreValues(A);first");
152649b5e25fSSatish Balay 
152749b5e25fSSatish Balay   /* copy values over */
152887828ca2SBarry Smith   ierr = PetscMemcpy(aij->a,aij->saved_values,nz*sizeof(PetscScalar));CHKERRQ(ierr);
152949b5e25fSSatish Balay   PetscFunctionReturn(0);
153049b5e25fSSatish Balay }
153149b5e25fSSatish Balay 
15324a2ae208SSatish Balay #undef __FUNCT__
1533a23d5eceSKris Buschelman #define __FUNCT__ "MatSeqSBAIJSetPreallocation_SeqSBAIJ"
15347087cfbeSBarry Smith PetscErrorCode  MatSeqSBAIJSetPreallocation_SeqSBAIJ(Mat B,PetscInt bs,PetscInt nz,PetscInt *nnz)
153549b5e25fSSatish Balay {
1536c464158bSHong Zhang   Mat_SeqSBAIJ   *b = (Mat_SeqSBAIJ*)B->data;
15376849ba73SBarry Smith   PetscErrorCode ierr;
1538535b19f3SBarry Smith   PetscInt       i,mbs,bs2;
15392576faa2SJed Brown   PetscBool      skipallocation = PETSC_FALSE,flg = PETSC_FALSE,realalloc = PETSC_FALSE;
154049b5e25fSSatish Balay 
154149b5e25fSSatish Balay   PetscFunctionBegin;
15422576faa2SJed Brown   if (nz >= 0 || nnz) realalloc = PETSC_TRUE;
1543273d9f13SBarry Smith   B->preallocated = PETSC_TRUE;
1544db4efbfdSBarry Smith 
154533d57670SJed Brown   ierr = MatSetBlockSize(B,PetscAbs(bs));CHKERRQ(ierr);
154626283091SBarry Smith   ierr = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr);
154726283091SBarry Smith   ierr = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr);
1548e02043d6SBarry Smith   ierr = PetscLayoutGetBlockSize(B->rmap,&bs);CHKERRQ(ierr);
1549899cda47SBarry Smith 
1550d0f46423SBarry Smith   mbs = B->rmap->N/bs;
155149b5e25fSSatish Balay   bs2 = bs*bs;
155249b5e25fSSatish Balay 
1553e7e72b3dSBarry Smith   if (mbs*bs != B->rmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Number rows, cols must be divisible by blocksize");
155449b5e25fSSatish Balay 
1555ab93d7beSBarry Smith   if (nz == MAT_SKIP_ALLOCATION) {
1556ab93d7beSBarry Smith     skipallocation = PETSC_TRUE;
1557ab93d7beSBarry Smith     nz             = 0;
1558ab93d7beSBarry Smith   }
1559ab93d7beSBarry Smith 
1560435da068SBarry Smith   if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 3;
1561e32f2f54SBarry Smith   if (nz < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"nz cannot be less than 0: value %D",nz);
156249b5e25fSSatish Balay   if (nnz) {
156349b5e25fSSatish Balay     for (i=0; i<mbs; i++) {
1564e32f2f54SBarry 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]);
1565e32f2f54SBarry 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);
156649b5e25fSSatish Balay     }
156749b5e25fSSatish Balay   }
156849b5e25fSSatish Balay 
1569db4efbfdSBarry Smith   B->ops->mult             = MatMult_SeqSBAIJ_N;
1570db4efbfdSBarry Smith   B->ops->multadd          = MatMultAdd_SeqSBAIJ_N;
1571db4efbfdSBarry Smith   B->ops->multtranspose    = MatMult_SeqSBAIJ_N;
1572db4efbfdSBarry Smith   B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_N;
157326fbe8dcSKarl Rupp 
15740298fd71SBarry Smith   ierr  = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_no_unroll",&flg,NULL);CHKERRQ(ierr);
157549b5e25fSSatish Balay   if (!flg) {
157649b5e25fSSatish Balay     switch (bs) {
157749b5e25fSSatish Balay     case 1:
157849b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_1;
157949b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_1;
1580431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_1;
1581431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_1;
158249b5e25fSSatish Balay       break;
158349b5e25fSSatish Balay     case 2:
158449b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_2;
158549b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_2;
1586431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_2;
1587431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_2;
158849b5e25fSSatish Balay       break;
158949b5e25fSSatish Balay     case 3:
159049b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_3;
159149b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_3;
1592431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_3;
1593431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_3;
159449b5e25fSSatish Balay       break;
159549b5e25fSSatish Balay     case 4:
159649b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_4;
159749b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_4;
1598431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_4;
1599431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_4;
160049b5e25fSSatish Balay       break;
160149b5e25fSSatish Balay     case 5:
160249b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_5;
160349b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_5;
1604431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_5;
1605431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_5;
160649b5e25fSSatish Balay       break;
160749b5e25fSSatish Balay     case 6:
160849b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_6;
160949b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_6;
1610431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_6;
1611431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_6;
161249b5e25fSSatish Balay       break;
161349b5e25fSSatish Balay     case 7:
1614de53e5efSHong Zhang       B->ops->mult             = MatMult_SeqSBAIJ_7;
161549b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_7;
1616431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_7;
1617431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_7;
161849b5e25fSSatish Balay       break;
161949b5e25fSSatish Balay     }
162049b5e25fSSatish Balay   }
162149b5e25fSSatish Balay 
162249b5e25fSSatish Balay   b->mbs = mbs;
16234afc71dfSHong Zhang   b->nbs = mbs;
1624ab93d7beSBarry Smith   if (!skipallocation) {
16252ee49352SLisandro Dalcin     if (!b->imax) {
1626dcca6d9dSJed Brown       ierr = PetscMalloc2(mbs,&b->imax,mbs,&b->ilen);CHKERRQ(ierr);
162726fbe8dcSKarl Rupp 
1628c760cd28SBarry Smith       b->free_imax_ilen = PETSC_TRUE;
162926fbe8dcSKarl Rupp 
16303bb1ff40SBarry Smith       ierr = PetscLogObjectMemory((PetscObject)B,2*mbs*sizeof(PetscInt));CHKERRQ(ierr);
16312ee49352SLisandro Dalcin     }
163249b5e25fSSatish Balay     if (!nnz) {
1633435da068SBarry Smith       if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 5;
163449b5e25fSSatish Balay       else if (nz <= 0) nz = 1;
163526fbe8dcSKarl Rupp       for (i=0; i<mbs; i++) b->imax[i] = nz;
1636153ea458SHong Zhang       nz = nz*mbs; /* total nz */
163749b5e25fSSatish Balay     } else {
163849b5e25fSSatish Balay       nz = 0;
16398cef66ccSHong Zhang       for (i=0; i<mbs; i++) {b->imax[i] = nnz[i]; nz += nnz[i];}
164049b5e25fSSatish Balay     }
16412ee49352SLisandro Dalcin     /* b->ilen will count nonzeros in each block row so far. */
164226fbe8dcSKarl Rupp     for (i=0; i<mbs; i++) b->ilen[i] = 0;
16436c6c5352SBarry Smith     /* nz=(nz+mbs)/2; */ /* total diagonal and superdiagonal nonzero blocks */
164449b5e25fSSatish Balay 
164549b5e25fSSatish Balay     /* allocate the matrix space */
16462ee49352SLisandro Dalcin     ierr = MatSeqXAIJFreeAIJ(B,&b->a,&b->j,&b->i);CHKERRQ(ierr);
1647dcca6d9dSJed Brown     ierr = PetscMalloc3(bs2*nz,&b->a,nz,&b->j,B->rmap->N+1,&b->i);CHKERRQ(ierr);
16483bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)B,(B->rmap->N+1)*sizeof(PetscInt)+nz*(bs2*sizeof(PetscScalar)+sizeof(PetscInt)));CHKERRQ(ierr);
16496c6c5352SBarry Smith     ierr = PetscMemzero(b->a,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr);
165013f74950SBarry Smith     ierr = PetscMemzero(b->j,nz*sizeof(PetscInt));CHKERRQ(ierr);
165126fbe8dcSKarl Rupp 
165249b5e25fSSatish Balay     b->singlemalloc = PETSC_TRUE;
165349b5e25fSSatish Balay 
165449b5e25fSSatish Balay     /* pointer to beginning of each row */
1655e60cf9a0SBarry Smith     b->i[0] = 0;
165626fbe8dcSKarl Rupp     for (i=1; i<mbs+1; i++) b->i[i] = b->i[i-1] + b->imax[i-1];
165726fbe8dcSKarl Rupp 
1658e6b907acSBarry Smith     b->free_a  = PETSC_TRUE;
1659e6b907acSBarry Smith     b->free_ij = PETSC_TRUE;
1660e811da20SHong Zhang   } else {
1661e6b907acSBarry Smith     b->free_a  = PETSC_FALSE;
1662e6b907acSBarry Smith     b->free_ij = PETSC_FALSE;
1663ab93d7beSBarry Smith   }
166449b5e25fSSatish Balay 
1665d0f46423SBarry Smith   B->rmap->bs = bs;
166649b5e25fSSatish Balay   b->bs2      = bs2;
16676c6c5352SBarry Smith   b->nz       = 0;
1668b32cb4a7SJed Brown   b->maxnz    = nz;
1669153ea458SHong Zhang 
167016cdd363SHong Zhang   b->inew    = 0;
167116cdd363SHong Zhang   b->jnew    = 0;
167216cdd363SHong Zhang   b->anew    = 0;
167316cdd363SHong Zhang   b->a2anew  = 0;
16741a3463dfSHong Zhang   b->permute = PETSC_FALSE;
16752576faa2SJed Brown   if (realalloc) {ierr = MatSetOption(B,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);}
1676c464158bSHong Zhang   PetscFunctionReturn(0);
1677c464158bSHong Zhang }
1678153ea458SHong Zhang 
167938f409ebSLisandro Dalcin #undef __FUNCT__
168038f409ebSLisandro Dalcin #define __FUNCT__ "MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ"
168138f409ebSLisandro Dalcin PetscErrorCode MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ(Mat B,PetscInt bs,const PetscInt ii[],const PetscInt jj[], const PetscScalar V[])
168238f409ebSLisandro Dalcin {
168338f409ebSLisandro Dalcin   PetscInt       i,j,m,nz,nz_max=0,*nnz;
168438f409ebSLisandro Dalcin   PetscScalar    *values=0;
168538f409ebSLisandro Dalcin   PetscBool      roworiented = ((Mat_SeqSBAIJ*)B->data)->roworiented;
168638f409ebSLisandro Dalcin   PetscErrorCode ierr;
168738f409ebSLisandro Dalcin   PetscFunctionBegin;
168838f409ebSLisandro Dalcin   if (bs < 1) SETERRQ1(PetscObjectComm((PetscObject)B),PETSC_ERR_ARG_OUTOFRANGE,"Invalid block size specified, must be positive but it is %D",bs);
168938f409ebSLisandro Dalcin   ierr   = PetscLayoutSetBlockSize(B->rmap,bs);CHKERRQ(ierr);
169038f409ebSLisandro Dalcin   ierr   = PetscLayoutSetBlockSize(B->cmap,bs);CHKERRQ(ierr);
169138f409ebSLisandro Dalcin   ierr   = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr);
169238f409ebSLisandro Dalcin   ierr   = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr);
169338f409ebSLisandro Dalcin   ierr   = PetscLayoutGetBlockSize(B->rmap,&bs);CHKERRQ(ierr);
169438f409ebSLisandro Dalcin   m      = B->rmap->n/bs;
169538f409ebSLisandro Dalcin 
169638f409ebSLisandro Dalcin   if (ii[0]) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"ii[0] must be 0 but it is %D",ii[0]);
1697785e854fSJed Brown   ierr = PetscMalloc1((m+1),&nnz);CHKERRQ(ierr);
169838f409ebSLisandro Dalcin   for (i=0; i<m; i++) {
169938f409ebSLisandro Dalcin     nz = ii[i+1] - ii[i];
170038f409ebSLisandro Dalcin     if (nz < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row %D has a negative number of columns %D",i,nz);
170138f409ebSLisandro Dalcin     nz_max = PetscMax(nz_max,nz);
170238f409ebSLisandro Dalcin     nnz[i] = nz;
170338f409ebSLisandro Dalcin   }
170438f409ebSLisandro Dalcin   ierr = MatSeqSBAIJSetPreallocation(B,bs,0,nnz);CHKERRQ(ierr);
170538f409ebSLisandro Dalcin   ierr = PetscFree(nnz);CHKERRQ(ierr);
170638f409ebSLisandro Dalcin 
170738f409ebSLisandro Dalcin   values = (PetscScalar*)V;
170838f409ebSLisandro Dalcin   if (!values) {
17091795a4d1SJed Brown     ierr = PetscCalloc1(bs*bs*nz_max,&values);CHKERRQ(ierr);
171038f409ebSLisandro Dalcin   }
171138f409ebSLisandro Dalcin   for (i=0; i<m; i++) {
171238f409ebSLisandro Dalcin     PetscInt          ncols  = ii[i+1] - ii[i];
171338f409ebSLisandro Dalcin     const PetscInt    *icols = jj + ii[i];
171438f409ebSLisandro Dalcin     if (!roworiented || bs == 1) {
171538f409ebSLisandro Dalcin       const PetscScalar *svals = values + (V ? (bs*bs*ii[i]) : 0);
171638f409ebSLisandro Dalcin       ierr = MatSetValuesBlocked_SeqSBAIJ(B,1,&i,ncols,icols,svals,INSERT_VALUES);CHKERRQ(ierr);
171738f409ebSLisandro Dalcin     } else {
171838f409ebSLisandro Dalcin       for (j=0; j<ncols; j++) {
171938f409ebSLisandro Dalcin         const PetscScalar *svals = values + (V ? (bs*bs*(ii[i]+j)) : 0);
172038f409ebSLisandro Dalcin         ierr = MatSetValuesBlocked_SeqSBAIJ(B,1,&i,1,&icols[j],svals,INSERT_VALUES);CHKERRQ(ierr);
172138f409ebSLisandro Dalcin       }
172238f409ebSLisandro Dalcin     }
172338f409ebSLisandro Dalcin   }
172438f409ebSLisandro Dalcin   if (!V) { ierr = PetscFree(values);CHKERRQ(ierr); }
172538f409ebSLisandro Dalcin   ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
172638f409ebSLisandro Dalcin   ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
172738f409ebSLisandro Dalcin   ierr = MatSetOption(B,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
172838f409ebSLisandro Dalcin   PetscFunctionReturn(0);
172938f409ebSLisandro Dalcin }
173038f409ebSLisandro Dalcin 
1731db4efbfdSBarry Smith /*
1732db4efbfdSBarry Smith    This is used to set the numeric factorization for both Cholesky and ICC symbolic factorization
1733db4efbfdSBarry Smith */
17348b1456e3SHong Zhang #undef __FUNCT__
1735d595f711SHong Zhang #define __FUNCT__ "MatSeqSBAIJSetNumericFactorization_inplace"
1736ace3abfcSBarry Smith PetscErrorCode MatSeqSBAIJSetNumericFactorization_inplace(Mat B,PetscBool natural)
1737db4efbfdSBarry Smith {
1738db4efbfdSBarry Smith   PetscErrorCode ierr;
1739ace3abfcSBarry Smith   PetscBool      flg = PETSC_FALSE;
1740db4efbfdSBarry Smith   PetscInt       bs  = B->rmap->bs;
1741db4efbfdSBarry Smith 
1742db4efbfdSBarry Smith   PetscFunctionBegin;
17430298fd71SBarry Smith   ierr = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_no_unroll",&flg,NULL);CHKERRQ(ierr);
1744db4efbfdSBarry Smith   if (flg) bs = 8;
1745db4efbfdSBarry Smith 
1746db4efbfdSBarry Smith   if (!natural) {
1747db4efbfdSBarry Smith     switch (bs) {
1748db4efbfdSBarry Smith     case 1:
1749d595f711SHong Zhang       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_inplace;
1750db4efbfdSBarry Smith       break;
1751db4efbfdSBarry Smith     case 2:
1752db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2;
1753db4efbfdSBarry Smith       break;
1754db4efbfdSBarry Smith     case 3:
1755db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3;
1756db4efbfdSBarry Smith       break;
1757db4efbfdSBarry Smith     case 4:
1758db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4;
1759db4efbfdSBarry Smith       break;
1760db4efbfdSBarry Smith     case 5:
1761db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5;
1762db4efbfdSBarry Smith       break;
1763db4efbfdSBarry Smith     case 6:
1764db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6;
1765db4efbfdSBarry Smith       break;
1766db4efbfdSBarry Smith     case 7:
1767db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7;
1768db4efbfdSBarry Smith       break;
1769db4efbfdSBarry Smith     default:
1770db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N;
1771db4efbfdSBarry Smith       break;
1772db4efbfdSBarry Smith     }
1773db4efbfdSBarry Smith   } else {
1774db4efbfdSBarry Smith     switch (bs) {
1775db4efbfdSBarry Smith     case 1:
1776d595f711SHong Zhang       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering_inplace;
1777db4efbfdSBarry Smith       break;
1778db4efbfdSBarry Smith     case 2:
1779db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering;
1780db4efbfdSBarry Smith       break;
1781db4efbfdSBarry Smith     case 3:
1782db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering;
1783db4efbfdSBarry Smith       break;
1784db4efbfdSBarry Smith     case 4:
1785db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering;
1786db4efbfdSBarry Smith       break;
1787db4efbfdSBarry Smith     case 5:
1788db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering;
1789db4efbfdSBarry Smith       break;
1790db4efbfdSBarry Smith     case 6:
1791db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering;
1792db4efbfdSBarry Smith       break;
1793db4efbfdSBarry Smith     case 7:
1794db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering;
1795db4efbfdSBarry Smith       break;
1796db4efbfdSBarry Smith     default:
1797db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering;
1798db4efbfdSBarry Smith       break;
1799db4efbfdSBarry Smith     }
1800db4efbfdSBarry Smith   }
1801db4efbfdSBarry Smith   PetscFunctionReturn(0);
1802db4efbfdSBarry Smith }
1803db4efbfdSBarry Smith 
18048cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqAIJ(Mat, MatType,MatReuse,Mat*);
18058cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqBAIJ(Mat, MatType,MatReuse,Mat*);
1806d769727bSBarry Smith 
18075c9eb25fSBarry Smith #undef __FUNCT__
18085c9eb25fSBarry Smith #define __FUNCT__ "MatGetFactor_seqsbaij_petsc"
18098cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_seqsbaij_petsc(Mat A,MatFactorType ftype,Mat *B)
18105c9eb25fSBarry Smith {
1811d0f46423SBarry Smith   PetscInt       n = A->rmap->n;
18125c9eb25fSBarry Smith   PetscErrorCode ierr;
18135c9eb25fSBarry Smith 
18145c9eb25fSBarry Smith   PetscFunctionBegin;
18150e92d65fSHong Zhang #if defined(PETSC_USE_COMPLEX)
18160e92d65fSHong Zhang   if (A->hermitian) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Hermitian Factor is not supported");
18170e92d65fSHong Zhang #endif
1818ce94432eSBarry Smith   ierr = MatCreate(PetscObjectComm((PetscObject)A),B);CHKERRQ(ierr);
18195c9eb25fSBarry Smith   ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr);
18205c9eb25fSBarry Smith   if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) {
18215c9eb25fSBarry Smith     ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr);
18220298fd71SBarry Smith     ierr = MatSeqSBAIJSetPreallocation(*B,A->rmap->bs,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr);
182326fbe8dcSKarl Rupp 
18247b056e98SHong Zhang     (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqSBAIJ;
1825c6d0d4f0SHong Zhang     (*B)->ops->iccfactorsymbolic      = MatICCFactorSymbolic_SeqSBAIJ;
1826e32f2f54SBarry Smith   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Factor type not supported");
1827d5f3da31SBarry Smith   (*B)->factortype = ftype;
18285c9eb25fSBarry Smith   PetscFunctionReturn(0);
18295c9eb25fSBarry Smith }
18305c9eb25fSBarry Smith 
1831db4efbfdSBarry Smith #undef __FUNCT__
1832db4efbfdSBarry Smith #define __FUNCT__ "MatGetFactorAvailable_seqsbaij_petsc"
1833ace3abfcSBarry Smith PetscErrorCode MatGetFactorAvailable_seqsbaij_petsc(Mat A,MatFactorType ftype,PetscBool  *flg)
1834db4efbfdSBarry Smith {
1835db4efbfdSBarry Smith   PetscFunctionBegin;
1836db4efbfdSBarry Smith   if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) {
1837db4efbfdSBarry Smith     *flg = PETSC_TRUE;
1838db4efbfdSBarry Smith   } else {
1839db4efbfdSBarry Smith     *flg = PETSC_FALSE;
1840db4efbfdSBarry Smith   }
1841db4efbfdSBarry Smith   PetscFunctionReturn(0);
1842db4efbfdSBarry Smith }
1843db4efbfdSBarry Smith 
1844611f576cSBarry Smith #if defined(PETSC_HAVE_MUMPS)
18458cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_sbaij_mumps(Mat,MatFactorType,Mat*);
1846611f576cSBarry Smith #endif
1847b5e56a35SBarry Smith #if defined(PETSC_HAVE_PASTIX)
18488cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_seqsbaij_pastix(Mat,MatFactorType,Mat*);
1849b5e56a35SBarry Smith #endif
1850913c410bSShri Abhyankar #if defined(PETSC_HAVE_SUITESPARSE)
18518cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_seqsbaij_cholmod(Mat,MatFactorType,Mat*);
185220db9a53SJed Brown #endif
18538cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_seqsbaij_sbstrm(Mat,MatFactorType,Mat*);
18545c9eb25fSBarry Smith 
18550bad9183SKris Buschelman /*MC
1856fafad747SKris Buschelman   MATSEQSBAIJ - MATSEQSBAIJ = "seqsbaij" - A matrix type to be used for sequential symmetric block sparse matrices,
18570bad9183SKris Buschelman   based on block compressed sparse row format.  Only the upper triangular portion of the matrix is stored.
18580bad9183SKris Buschelman 
1859828413b8SBarry Smith   For complex numbers by default this matrix is symmetric, NOT Hermitian symmetric. To make it Hermitian symmetric you
186071dad5bbSBarry Smith   can call MatSetOption(Mat, MAT_HERMITIAN); after MatAssemblyEnd()
1861828413b8SBarry Smith 
18620bad9183SKris Buschelman   Options Database Keys:
18630bad9183SKris Buschelman   . -mat_type seqsbaij - sets the matrix type to "seqsbaij" during a call to MatSetFromOptions()
18640bad9183SKris Buschelman 
186571dad5bbSBarry Smith   Notes: By default if you insert values into the lower triangular part of the matrix they are simply ignored (since they are not
186671dad5bbSBarry 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
186771dad5bbSBarry 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.
186871dad5bbSBarry Smith 
186971dad5bbSBarry Smith 
18700bad9183SKris Buschelman   Level: beginner
18710bad9183SKris Buschelman 
18720bad9183SKris Buschelman   .seealso: MatCreateSeqSBAIJ
18730bad9183SKris Buschelman M*/
18740bad9183SKris Buschelman 
18758cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqSBSTRM(Mat, MatType,MatReuse,Mat*);
1876aa5a9175SDahai Guo 
1877a23d5eceSKris Buschelman #undef __FUNCT__
1878a23d5eceSKris Buschelman #define __FUNCT__ "MatCreate_SeqSBAIJ"
18798cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatCreate_SeqSBAIJ(Mat B)
1880a23d5eceSKris Buschelman {
1881a23d5eceSKris Buschelman   Mat_SeqSBAIJ   *b;
1882dfbe8321SBarry Smith   PetscErrorCode ierr;
188313f74950SBarry Smith   PetscMPIInt    size;
1884ace3abfcSBarry Smith   PetscBool      no_unroll = PETSC_FALSE,no_inode = PETSC_FALSE;
1885a23d5eceSKris Buschelman 
1886a23d5eceSKris Buschelman   PetscFunctionBegin;
1887ce94432eSBarry Smith   ierr = MPI_Comm_size(PetscObjectComm((PetscObject)B),&size);CHKERRQ(ierr);
1888e32f2f54SBarry Smith   if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Comm must be of size 1");
1889a23d5eceSKris Buschelman 
1890b00a9115SJed Brown   ierr    = PetscNewLog(B,&b);CHKERRQ(ierr);
1891a23d5eceSKris Buschelman   B->data = (void*)b;
1892a23d5eceSKris Buschelman   ierr    = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr);
189326fbe8dcSKarl Rupp 
1894a23d5eceSKris Buschelman   B->ops->destroy    = MatDestroy_SeqSBAIJ;
1895a23d5eceSKris Buschelman   B->ops->view       = MatView_SeqSBAIJ;
1896a23d5eceSKris Buschelman   b->row             = 0;
1897a23d5eceSKris Buschelman   b->icol            = 0;
1898a23d5eceSKris Buschelman   b->reallocs        = 0;
1899a23d5eceSKris Buschelman   b->saved_values    = 0;
19000def2e27SBarry Smith   b->inode.limit     = 5;
19010def2e27SBarry Smith   b->inode.max_limit = 5;
1902a23d5eceSKris Buschelman 
1903a23d5eceSKris Buschelman   b->roworiented        = PETSC_TRUE;
1904a23d5eceSKris Buschelman   b->nonew              = 0;
1905a23d5eceSKris Buschelman   b->diag               = 0;
1906a23d5eceSKris Buschelman   b->solve_work         = 0;
1907a23d5eceSKris Buschelman   b->mult_work          = 0;
1908a23d5eceSKris Buschelman   B->spptr              = 0;
1909f2cbd3d5SJed Brown   B->info.nz_unneeded   = (PetscReal)b->maxnz*b->bs2;
1910a9817697SBarry Smith   b->keepnonzeropattern = PETSC_FALSE;
1911a23d5eceSKris Buschelman   b->xtoy               = 0;
1912a23d5eceSKris Buschelman   b->XtoY               = 0;
1913a23d5eceSKris Buschelman 
1914a23d5eceSKris Buschelman   b->inew    = 0;
1915a23d5eceSKris Buschelman   b->jnew    = 0;
1916a23d5eceSKris Buschelman   b->anew    = 0;
1917a23d5eceSKris Buschelman   b->a2anew  = 0;
1918a23d5eceSKris Buschelman   b->permute = PETSC_FALSE;
1919a23d5eceSKris Buschelman 
192071dad5bbSBarry Smith   b->ignore_ltriangular = PETSC_TRUE;
192126fbe8dcSKarl Rupp 
19220298fd71SBarry Smith   ierr = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_ignore_lower_triangular",&b->ignore_ltriangular,NULL);CHKERRQ(ierr);
1923941593c8SHong Zhang 
1924f5edf698SHong Zhang   b->getrow_utriangular = PETSC_FALSE;
192526fbe8dcSKarl Rupp 
19260298fd71SBarry Smith   ierr = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_getrow_uppertriangular",&b->getrow_utriangular,NULL);CHKERRQ(ierr);
1927f5edf698SHong Zhang 
1928b5e56a35SBarry Smith #if defined(PETSC_HAVE_PASTIX)
1929bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactor_pastix_C",MatGetFactor_seqsbaij_pastix);CHKERRQ(ierr);
1930b5e56a35SBarry Smith #endif
1931611f576cSBarry Smith #if defined(PETSC_HAVE_MUMPS)
1932bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactor_mumps_C",MatGetFactor_sbaij_mumps);CHKERRQ(ierr);
1933611f576cSBarry Smith #endif
1934913c410bSShri Abhyankar #if defined(PETSC_HAVE_SUITESPARSE)
1935bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactor_cholmod_C",MatGetFactor_seqsbaij_cholmod);CHKERRQ(ierr);
193620db9a53SJed Brown #endif
1937bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactorAvailable_petsc_C",MatGetFactorAvailable_seqsbaij_petsc);CHKERRQ(ierr);
1938bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactor_petsc_C",MatGetFactor_seqsbaij_petsc);CHKERRQ(ierr);
1939bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactor_sbstrm_C",MatGetFactor_seqsbaij_sbstrm);CHKERRQ(ierr);
1940bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatStoreValues_C",MatStoreValues_SeqSBAIJ);CHKERRQ(ierr);
1941bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatRetrieveValues_C",MatRetrieveValues_SeqSBAIJ);CHKERRQ(ierr);
1942bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetColumnIndices_C",MatSeqSBAIJSetColumnIndices_SeqSBAIJ);CHKERRQ(ierr);
1943bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqaij_C",MatConvert_SeqSBAIJ_SeqAIJ);CHKERRQ(ierr);
1944bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqbaij_C",MatConvert_SeqSBAIJ_SeqBAIJ);CHKERRQ(ierr);
1945bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetPreallocation_C",MatSeqSBAIJSetPreallocation_SeqSBAIJ);CHKERRQ(ierr);
194638f409ebSLisandro Dalcin   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetPreallocationCSR_C",MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ);CHKERRQ(ierr);
1947bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqsbstrm_C",MatConvert_SeqSBAIJ_SeqSBSTRM);CHKERRQ(ierr);
194823ce1328SBarry Smith 
194923ce1328SBarry Smith   B->symmetric                  = PETSC_TRUE;
195023ce1328SBarry Smith   B->structurally_symmetric     = PETSC_TRUE;
195123ce1328SBarry Smith   B->symmetric_set              = PETSC_TRUE;
195223ce1328SBarry Smith   B->structurally_symmetric_set = PETSC_TRUE;
195326fbe8dcSKarl Rupp 
195417667f90SBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)B,MATSEQSBAIJ);CHKERRQ(ierr);
19550def2e27SBarry Smith 
1956ce94432eSBarry Smith   ierr = PetscOptionsBegin(PetscObjectComm((PetscObject)B),((PetscObject)B)->prefix,"Options for SEQSBAIJ matrix","Mat");CHKERRQ(ierr);
19570298fd71SBarry Smith   ierr = PetscOptionsBool("-mat_no_unroll","Do not optimize for inodes (slower)",NULL,no_unroll,&no_unroll,NULL);CHKERRQ(ierr);
195826fbe8dcSKarl Rupp   if (no_unroll) {
195926fbe8dcSKarl Rupp     ierr = PetscInfo(B,"Not using Inode routines due to -mat_no_unroll\n");CHKERRQ(ierr);
196026fbe8dcSKarl Rupp   }
19610298fd71SBarry Smith   ierr = PetscOptionsBool("-mat_no_inode","Do not optimize for inodes (slower)",NULL,no_inode,&no_inode,NULL);CHKERRQ(ierr);
196226fbe8dcSKarl Rupp   if (no_inode) {
196326fbe8dcSKarl Rupp     ierr = PetscInfo(B,"Not using Inode routines due to -mat_no_inode\n");CHKERRQ(ierr);
196426fbe8dcSKarl Rupp   }
19650298fd71SBarry Smith   ierr = PetscOptionsInt("-mat_inode_limit","Do not use inodes larger then this value",NULL,b->inode.limit,&b->inode.limit,NULL);CHKERRQ(ierr);
19660def2e27SBarry Smith   ierr = PetscOptionsEnd();CHKERRQ(ierr);
1967ace3abfcSBarry Smith   b->inode.use = (PetscBool)(!(no_unroll || no_inode));
19680def2e27SBarry Smith   if (b->inode.limit > b->inode.max_limit) b->inode.limit = b->inode.max_limit;
1969a23d5eceSKris Buschelman   PetscFunctionReturn(0);
1970a23d5eceSKris Buschelman }
1971a23d5eceSKris Buschelman 
1972a23d5eceSKris Buschelman #undef __FUNCT__
1973a23d5eceSKris Buschelman #define __FUNCT__ "MatSeqSBAIJSetPreallocation"
1974a23d5eceSKris Buschelman /*@C
1975a23d5eceSKris Buschelman    MatSeqSBAIJSetPreallocation - Creates a sparse symmetric matrix in block AIJ (block
1976a23d5eceSKris Buschelman    compressed row) format.  For good matrix assembly performance the
1977a23d5eceSKris Buschelman    user should preallocate the matrix storage by setting the parameter nz
1978a23d5eceSKris Buschelman    (or the array nnz).  By setting these parameters accurately, performance
1979a23d5eceSKris Buschelman    during matrix assembly can be increased by more than a factor of 50.
1980a23d5eceSKris Buschelman 
1981a23d5eceSKris Buschelman    Collective on Mat
1982a23d5eceSKris Buschelman 
1983a23d5eceSKris Buschelman    Input Parameters:
19841c4f3114SJed Brown +  B - the symmetric matrix
1985a23d5eceSKris Buschelman .  bs - size of block
1986a23d5eceSKris Buschelman .  nz - number of block nonzeros per block row (same for all rows)
1987a23d5eceSKris Buschelman -  nnz - array containing the number of block nonzeros in the upper triangular plus
19880298fd71SBarry Smith          diagonal portion of each block (possibly different for each block row) or NULL
1989a23d5eceSKris Buschelman 
1990a23d5eceSKris Buschelman    Options Database Keys:
1991a23d5eceSKris Buschelman .   -mat_no_unroll - uses code that does not unroll the loops in the
1992a23d5eceSKris Buschelman                      block calculations (much slower)
1993db4efbfdSBarry Smith .    -mat_block_size - size of the blocks to use (only works if a negative bs is passed in
1994a23d5eceSKris Buschelman 
1995a23d5eceSKris Buschelman    Level: intermediate
1996a23d5eceSKris Buschelman 
1997a23d5eceSKris Buschelman    Notes:
1998a23d5eceSKris Buschelman    Specify the preallocated storage with either nz or nnz (not both).
19990298fd71SBarry Smith    Set nz=PETSC_DEFAULT and nnz=NULL for PETSc to control dynamic memory
2000a7f22e61SSatish Balay    allocation.  See Users-Manual: ch_mat for details.
2001a23d5eceSKris Buschelman 
2002aa95bbe8SBarry Smith    You can call MatGetInfo() to get information on how effective the preallocation was;
2003aa95bbe8SBarry Smith    for example the fields mallocs,nz_allocated,nz_used,nz_unneeded;
2004aa95bbe8SBarry Smith    You can also run with the option -info and look for messages with the string
2005aa95bbe8SBarry Smith    malloc in them to see if additional memory allocation was needed.
2006aa95bbe8SBarry Smith 
200749a6f317SBarry Smith    If the nnz parameter is given then the nz parameter is ignored
200849a6f317SBarry Smith 
200949a6f317SBarry Smith 
201069b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateSBAIJ()
2011a23d5eceSKris Buschelman @*/
20127087cfbeSBarry Smith PetscErrorCode  MatSeqSBAIJSetPreallocation(Mat B,PetscInt bs,PetscInt nz,const PetscInt nnz[])
201313f74950SBarry Smith {
20144ac538c5SBarry Smith   PetscErrorCode ierr;
2015a23d5eceSKris Buschelman 
2016a23d5eceSKris Buschelman   PetscFunctionBegin;
20176ba663aaSJed Brown   PetscValidHeaderSpecific(B,MAT_CLASSID,1);
20186ba663aaSJed Brown   PetscValidType(B,1);
20196ba663aaSJed Brown   PetscValidLogicalCollectiveInt(B,bs,2);
20204ac538c5SBarry Smith   ierr = PetscTryMethod(B,"MatSeqSBAIJSetPreallocation_C",(Mat,PetscInt,PetscInt,const PetscInt[]),(B,bs,nz,nnz));CHKERRQ(ierr);
2021a23d5eceSKris Buschelman   PetscFunctionReturn(0);
2022a23d5eceSKris Buschelman }
202349b5e25fSSatish Balay 
20244a2ae208SSatish Balay #undef  __FUNCT__
202538f409ebSLisandro Dalcin #define __FUNCT__ "MatSeqSBAIJSetPreallocationCSR"
202638f409ebSLisandro Dalcin /*@C
202738f409ebSLisandro Dalcin    MatSeqSBAIJSetPreallocationCSR - Allocates memory for a sparse sequential matrix in symmetric block AIJ format.
202838f409ebSLisandro Dalcin 
202938f409ebSLisandro Dalcin    Input Parameters:
20301c4f3114SJed Brown +  B - the matrix
203138f409ebSLisandro Dalcin .  i - the indices into j for the start of each local row (starts with zero)
203238f409ebSLisandro Dalcin .  j - the column indices for each local row (starts with zero) these must be sorted for each row
203338f409ebSLisandro Dalcin -  v - optional values in the matrix
203438f409ebSLisandro Dalcin 
203538f409ebSLisandro Dalcin    Level: developer
203638f409ebSLisandro Dalcin 
203738f409ebSLisandro Dalcin    Notes:
203838f409ebSLisandro Dalcin    The order of the entries in values is specified by the MatOption MAT_ROW_ORIENTED.  For example, C programs
203938f409ebSLisandro 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
204038f409ebSLisandro Dalcin    over rows within a block and the last index is over columns within a block row.  Fortran programs will likely set
204138f409ebSLisandro 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
204238f409ebSLisandro Dalcin    block column and the second index is over columns within a block.
204338f409ebSLisandro Dalcin 
204438f409ebSLisandro Dalcin .keywords: matrix, block, aij, compressed row, sparse
204538f409ebSLisandro Dalcin 
204638f409ebSLisandro Dalcin .seealso: MatCreate(), MatCreateSeqSBAIJ(), MatSetValuesBlocked(), MatSeqSBAIJSetPreallocation(), MATSEQSBAIJ
204738f409ebSLisandro Dalcin @*/
204838f409ebSLisandro Dalcin PetscErrorCode MatSeqSBAIJSetPreallocationCSR(Mat B,PetscInt bs,const PetscInt i[],const PetscInt j[], const PetscScalar v[])
204938f409ebSLisandro Dalcin {
205038f409ebSLisandro Dalcin   PetscErrorCode ierr;
205138f409ebSLisandro Dalcin 
205238f409ebSLisandro Dalcin   PetscFunctionBegin;
205338f409ebSLisandro Dalcin   PetscValidHeaderSpecific(B,MAT_CLASSID,1);
205438f409ebSLisandro Dalcin   PetscValidType(B,1);
205538f409ebSLisandro Dalcin   PetscValidLogicalCollectiveInt(B,bs,2);
205638f409ebSLisandro Dalcin   ierr = PetscTryMethod(B,"MatSeqSBAIJSetPreallocationCSR_C",(Mat,PetscInt,const PetscInt[],const PetscInt[],const PetscScalar[]),(B,bs,i,j,v));CHKERRQ(ierr);
205738f409ebSLisandro Dalcin   PetscFunctionReturn(0);
205838f409ebSLisandro Dalcin }
205938f409ebSLisandro Dalcin 
206038f409ebSLisandro Dalcin #undef __FUNCT__
20614a2ae208SSatish Balay #define __FUNCT__ "MatCreateSeqSBAIJ"
2062c464158bSHong Zhang /*@C
2063c464158bSHong Zhang    MatCreateSeqSBAIJ - Creates a sparse symmetric matrix in block AIJ (block
2064c464158bSHong Zhang    compressed row) format.  For good matrix assembly performance the
2065c464158bSHong Zhang    user should preallocate the matrix storage by setting the parameter nz
2066c464158bSHong Zhang    (or the array nnz).  By setting these parameters accurately, performance
2067c464158bSHong Zhang    during matrix assembly can be increased by more than a factor of 50.
206849b5e25fSSatish Balay 
2069c464158bSHong Zhang    Collective on MPI_Comm
2070c464158bSHong Zhang 
2071c464158bSHong Zhang    Input Parameters:
2072c464158bSHong Zhang +  comm - MPI communicator, set to PETSC_COMM_SELF
2073c464158bSHong Zhang .  bs - size of block
2074c464158bSHong Zhang .  m - number of rows, or number of columns
2075c464158bSHong Zhang .  nz - number of block nonzeros per block row (same for all rows)
2076744e8345SSatish Balay -  nnz - array containing the number of block nonzeros in the upper triangular plus
20770298fd71SBarry Smith          diagonal portion of each block (possibly different for each block row) or NULL
2078c464158bSHong Zhang 
2079c464158bSHong Zhang    Output Parameter:
2080c464158bSHong Zhang .  A - the symmetric matrix
2081c464158bSHong Zhang 
2082c464158bSHong Zhang    Options Database Keys:
2083c464158bSHong Zhang .   -mat_no_unroll - uses code that does not unroll the loops in the
2084c464158bSHong Zhang                      block calculations (much slower)
2085c464158bSHong Zhang .    -mat_block_size - size of the blocks to use
2086c464158bSHong Zhang 
2087c464158bSHong Zhang    Level: intermediate
2088c464158bSHong Zhang 
2089175b88e8SBarry Smith    It is recommended that one use the MatCreate(), MatSetType() and/or MatSetFromOptions(),
2090ae1d86c5SBarry Smith    MatXXXXSetPreallocation() paradgm instead of this routine directly.
2091175b88e8SBarry Smith    [MatXXXXSetPreallocation() is, for example, MatSeqAIJSetPreallocation]
2092175b88e8SBarry Smith 
2093c464158bSHong Zhang    Notes:
20946d6d819aSHong Zhang    The number of rows and columns must be divisible by blocksize.
20956d6d819aSHong Zhang    This matrix type does not support complex Hermitian operation.
2096c464158bSHong Zhang 
2097c464158bSHong Zhang    Specify the preallocated storage with either nz or nnz (not both).
20980298fd71SBarry Smith    Set nz=PETSC_DEFAULT and nnz=NULL for PETSc to control dynamic memory
2099a7f22e61SSatish Balay    allocation.  See Users-Manual: ch_mat for details.
2100c464158bSHong Zhang 
210149a6f317SBarry Smith    If the nnz parameter is given then the nz parameter is ignored
210249a6f317SBarry Smith 
210369b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateSBAIJ()
2104c464158bSHong Zhang @*/
21057087cfbeSBarry Smith PetscErrorCode  MatCreateSeqSBAIJ(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt nz,const PetscInt nnz[],Mat *A)
2106c464158bSHong Zhang {
2107dfbe8321SBarry Smith   PetscErrorCode ierr;
2108c464158bSHong Zhang 
2109c464158bSHong Zhang   PetscFunctionBegin;
2110f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,A);CHKERRQ(ierr);
2111f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*A,m,n,m,n);CHKERRQ(ierr);
2112c464158bSHong Zhang   ierr = MatSetType(*A,MATSEQSBAIJ);CHKERRQ(ierr);
2113ab93d7beSBarry Smith   ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(*A,bs,nz,(PetscInt*)nnz);CHKERRQ(ierr);
211449b5e25fSSatish Balay   PetscFunctionReturn(0);
211549b5e25fSSatish Balay }
211649b5e25fSSatish Balay 
21174a2ae208SSatish Balay #undef __FUNCT__
21184a2ae208SSatish Balay #define __FUNCT__ "MatDuplicate_SeqSBAIJ"
2119dfbe8321SBarry Smith PetscErrorCode MatDuplicate_SeqSBAIJ(Mat A,MatDuplicateOption cpvalues,Mat *B)
212049b5e25fSSatish Balay {
212149b5e25fSSatish Balay   Mat            C;
212249b5e25fSSatish Balay   Mat_SeqSBAIJ   *c,*a = (Mat_SeqSBAIJ*)A->data;
21236849ba73SBarry Smith   PetscErrorCode ierr;
2124b40805acSSatish Balay   PetscInt       i,mbs = a->mbs,nz = a->nz,bs2 =a->bs2;
212549b5e25fSSatish Balay 
212649b5e25fSSatish Balay   PetscFunctionBegin;
2127e32f2f54SBarry Smith   if (a->i[mbs] != nz) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Corrupt matrix");
212849b5e25fSSatish Balay 
212949b5e25fSSatish Balay   *B   = 0;
2130ce94432eSBarry Smith   ierr = MatCreate(PetscObjectComm((PetscObject)A),&C);CHKERRQ(ierr);
2131d0f46423SBarry Smith   ierr = MatSetSizes(C,A->rmap->N,A->cmap->n,A->rmap->N,A->cmap->n);CHKERRQ(ierr);
21328e9a0fb8SHong Zhang   ierr = MatSetType(C,MATSEQSBAIJ);CHKERRQ(ierr);
21331d5dac46SHong Zhang   ierr = PetscMemcpy(C->ops,A->ops,sizeof(struct _MatOps));CHKERRQ(ierr);
2134692f9cbeSHong Zhang   c    = (Mat_SeqSBAIJ*)C->data;
2135692f9cbeSHong Zhang 
2136273d9f13SBarry Smith   C->preallocated       = PETSC_TRUE;
2137d5f3da31SBarry Smith   C->factortype         = A->factortype;
213849b5e25fSSatish Balay   c->row                = 0;
213949b5e25fSSatish Balay   c->icol               = 0;
214049b5e25fSSatish Balay   c->saved_values       = 0;
2141a9817697SBarry Smith   c->keepnonzeropattern = a->keepnonzeropattern;
214249b5e25fSSatish Balay   C->assembled          = PETSC_TRUE;
214349b5e25fSSatish Balay 
21441e1e43feSBarry Smith   ierr   = PetscLayoutReference(A->rmap,&C->rmap);CHKERRQ(ierr);
21451e1e43feSBarry Smith   ierr   = PetscLayoutReference(A->cmap,&C->cmap);CHKERRQ(ierr);
214649b5e25fSSatish Balay   c->bs2 = a->bs2;
214749b5e25fSSatish Balay   c->mbs = a->mbs;
214849b5e25fSSatish Balay   c->nbs = a->nbs;
214949b5e25fSSatish Balay 
2150c760cd28SBarry Smith   if  (cpvalues == MAT_SHARE_NONZERO_PATTERN) {
2151c760cd28SBarry Smith     c->imax           = a->imax;
2152c760cd28SBarry Smith     c->ilen           = a->ilen;
2153c760cd28SBarry Smith     c->free_imax_ilen = PETSC_FALSE;
2154c760cd28SBarry Smith   } else {
2155dcca6d9dSJed Brown     ierr = PetscMalloc2((mbs+1),&c->imax,(mbs+1),&c->ilen);CHKERRQ(ierr);
21563bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)C,2*(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr);
215749b5e25fSSatish Balay     for (i=0; i<mbs; i++) {
215849b5e25fSSatish Balay       c->imax[i] = a->imax[i];
215949b5e25fSSatish Balay       c->ilen[i] = a->ilen[i];
216049b5e25fSSatish Balay     }
2161c760cd28SBarry Smith     c->free_imax_ilen = PETSC_TRUE;
2162c760cd28SBarry Smith   }
216349b5e25fSSatish Balay 
216449b5e25fSSatish Balay   /* allocate the matrix space */
21654da8f245SBarry Smith   if (cpvalues == MAT_SHARE_NONZERO_PATTERN) {
2166785e854fSJed Brown     ierr            = PetscMalloc1(bs2*nz,&c->a);CHKERRQ(ierr);
21673bb1ff40SBarry Smith     ierr            = PetscLogObjectMemory((PetscObject)C,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr);
216844e1c64aSLisandro Dalcin     c->i            = a->i;
216944e1c64aSLisandro Dalcin     c->j            = a->j;
21704da8f245SBarry Smith     c->singlemalloc = PETSC_FALSE;
217144e1c64aSLisandro Dalcin     c->free_a       = PETSC_TRUE;
21724da8f245SBarry Smith     c->free_ij      = PETSC_FALSE;
21734da8f245SBarry Smith     c->parent       = A;
21744da8f245SBarry Smith     ierr            = PetscObjectReference((PetscObject)A);CHKERRQ(ierr);
21754da8f245SBarry Smith     ierr            = MatSetOption(A,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
21764da8f245SBarry Smith     ierr            = MatSetOption(C,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
21774da8f245SBarry Smith   } else {
2178dcca6d9dSJed Brown     ierr            = PetscMalloc3(bs2*nz,&c->a,nz,&c->j,mbs+1,&c->i);CHKERRQ(ierr);
217913f74950SBarry Smith     ierr            = PetscMemcpy(c->i,a->i,(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr);
21803bb1ff40SBarry Smith     ierr            = PetscLogObjectMemory((PetscObject)C,(mbs+1)*sizeof(PetscInt) + nz*(bs2*sizeof(MatScalar) + sizeof(PetscInt)));CHKERRQ(ierr);
21814da8f245SBarry Smith     c->singlemalloc = PETSC_TRUE;
218244e1c64aSLisandro Dalcin     c->free_a       = PETSC_TRUE;
21834da8f245SBarry Smith     c->free_ij      = PETSC_TRUE;
21844da8f245SBarry Smith   }
218549b5e25fSSatish Balay   if (mbs > 0) {
21864da8f245SBarry Smith     if (cpvalues != MAT_SHARE_NONZERO_PATTERN) {
218713f74950SBarry Smith       ierr = PetscMemcpy(c->j,a->j,nz*sizeof(PetscInt));CHKERRQ(ierr);
21884da8f245SBarry Smith     }
218949b5e25fSSatish Balay     if (cpvalues == MAT_COPY_VALUES) {
219049b5e25fSSatish Balay       ierr = PetscMemcpy(c->a,a->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr);
219149b5e25fSSatish Balay     } else {
219249b5e25fSSatish Balay       ierr = PetscMemzero(c->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr);
219349b5e25fSSatish Balay     }
2194a1c3900fSBarry Smith     if (a->jshort) {
219544e1c64aSLisandro Dalcin       /* cannot share jshort, it is reallocated in MatAssemblyEnd_SeqSBAIJ() */
219644e1c64aSLisandro Dalcin       /* if the parent matrix is reassembled, this child matrix will never notice */
2197785e854fSJed Brown       ierr = PetscMalloc1(nz,&c->jshort);CHKERRQ(ierr);
21983bb1ff40SBarry Smith       ierr = PetscLogObjectMemory((PetscObject)C,nz*sizeof(unsigned short));CHKERRQ(ierr);
2199a1c3900fSBarry Smith       ierr = PetscMemcpy(c->jshort,a->jshort,nz*sizeof(unsigned short));CHKERRQ(ierr);
220026fbe8dcSKarl Rupp 
22014da8f245SBarry Smith       c->free_jshort = PETSC_TRUE;
22024da8f245SBarry Smith     }
2203a1c3900fSBarry Smith   }
220449b5e25fSSatish Balay 
220549b5e25fSSatish Balay   c->roworiented = a->roworiented;
220649b5e25fSSatish Balay   c->nonew       = a->nonew;
220749b5e25fSSatish Balay 
220849b5e25fSSatish Balay   if (a->diag) {
2209c760cd28SBarry Smith     if (cpvalues == MAT_SHARE_NONZERO_PATTERN) {
2210c760cd28SBarry Smith       c->diag      = a->diag;
2211c760cd28SBarry Smith       c->free_diag = PETSC_FALSE;
2212c760cd28SBarry Smith     } else {
2213785e854fSJed Brown       ierr = PetscMalloc1(mbs,&c->diag);CHKERRQ(ierr);
22143bb1ff40SBarry Smith       ierr = PetscLogObjectMemory((PetscObject)C,mbs*sizeof(PetscInt));CHKERRQ(ierr);
221526fbe8dcSKarl Rupp       for (i=0; i<mbs; i++) c->diag[i] = a->diag[i];
2216c760cd28SBarry Smith       c->free_diag = PETSC_TRUE;
2217c760cd28SBarry Smith     }
221844e1c64aSLisandro Dalcin   }
22196c6c5352SBarry Smith   c->nz         = a->nz;
2220f2cbd3d5SJed Brown   c->maxnz      = a->nz; /* Since we allocate exactly the right amount */
222149b5e25fSSatish Balay   c->solve_work = 0;
222249b5e25fSSatish Balay   c->mult_work  = 0;
222326fbe8dcSKarl Rupp 
222449b5e25fSSatish Balay   *B   = C;
2225140e18c1SBarry Smith   ierr = PetscFunctionListDuplicate(((PetscObject)A)->qlist,&((PetscObject)C)->qlist);CHKERRQ(ierr);
222649b5e25fSSatish Balay   PetscFunctionReturn(0);
222749b5e25fSSatish Balay }
222849b5e25fSSatish Balay 
22294a2ae208SSatish Balay #undef __FUNCT__
22305bba2384SShri Abhyankar #define __FUNCT__ "MatLoad_SeqSBAIJ"
2231112444f4SShri Abhyankar PetscErrorCode MatLoad_SeqSBAIJ(Mat newmat,PetscViewer viewer)
22322f480046SShri Abhyankar {
22332f480046SShri Abhyankar   Mat_SeqSBAIJ   *a;
22342f480046SShri Abhyankar   PetscErrorCode ierr;
22352f480046SShri Abhyankar   int            fd;
22362f480046SShri Abhyankar   PetscMPIInt    size;
22373059b6faSBarry Smith   PetscInt       i,nz,header[4],*rowlengths=0,M,N,bs = newmat->rmap->bs;
22382f480046SShri Abhyankar   PetscInt       *mask,mbs,*jj,j,rowcount,nzcount,k,*s_browlengths,maskcount;
22392f480046SShri Abhyankar   PetscInt       kmax,jcount,block,idx,point,nzcountb,extra_rows,rows,cols;
22402f480046SShri Abhyankar   PetscInt       *masked,nmask,tmp,bs2,ishift;
22412f480046SShri Abhyankar   PetscScalar    *aa;
2242ce94432eSBarry Smith   MPI_Comm       comm;
22432f480046SShri Abhyankar 
22442f480046SShri Abhyankar   PetscFunctionBegin;
2245ce94432eSBarry Smith   ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr);
22460298fd71SBarry Smith   ierr = PetscOptionsGetInt(((PetscObject)newmat)->prefix,"-matload_block_size",&bs,NULL);CHKERRQ(ierr);
22473059b6faSBarry Smith   if (bs < 0) bs = 1;
22482f480046SShri Abhyankar   bs2  = bs*bs;
22492f480046SShri Abhyankar 
22502f480046SShri Abhyankar   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
22512f480046SShri Abhyankar   if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"view must have one processor");
22522f480046SShri Abhyankar   ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr);
22532f480046SShri Abhyankar   ierr = PetscBinaryRead(fd,header,4,PETSC_INT);CHKERRQ(ierr);
22542f480046SShri Abhyankar   if (header[0] != MAT_FILE_CLASSID) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"not Mat object");
22552f480046SShri Abhyankar   M = header[1]; N = header[2]; nz = header[3];
22562f480046SShri Abhyankar 
22572f480046SShri Abhyankar   if (header[3] < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Matrix stored in special format, cannot load as SeqSBAIJ");
22582f480046SShri Abhyankar 
22592f480046SShri Abhyankar   if (M != N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Can only do square matrices");
22602f480046SShri Abhyankar 
22612f480046SShri Abhyankar   /*
22622f480046SShri Abhyankar      This code adds extra rows to make sure the number of rows is
22632f480046SShri Abhyankar     divisible by the blocksize
22642f480046SShri Abhyankar   */
22652f480046SShri Abhyankar   mbs        = M/bs;
22662f480046SShri Abhyankar   extra_rows = bs - M + bs*(mbs);
22672f480046SShri Abhyankar   if (extra_rows == bs) extra_rows = 0;
22682f480046SShri Abhyankar   else                  mbs++;
22692f480046SShri Abhyankar   if (extra_rows) {
22702f480046SShri Abhyankar     ierr = PetscInfo(viewer,"Padding loaded matrix to match blocksize\n");CHKERRQ(ierr);
22712f480046SShri Abhyankar   }
22722f480046SShri Abhyankar 
22732f480046SShri Abhyankar   /* Set global sizes if not already set */
22742f480046SShri Abhyankar   if (newmat->rmap->n < 0 && newmat->rmap->N < 0 && newmat->cmap->n < 0 && newmat->cmap->N < 0) {
22752f480046SShri Abhyankar     ierr = MatSetSizes(newmat,PETSC_DECIDE,PETSC_DECIDE,M+extra_rows,N+extra_rows);CHKERRQ(ierr);
22762f480046SShri Abhyankar   } else { /* Check if the matrix global sizes are correct */
22772f480046SShri Abhyankar     ierr = MatGetSize(newmat,&rows,&cols);CHKERRQ(ierr);
22782f480046SShri 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);
22792f480046SShri Abhyankar   }
22802f480046SShri Abhyankar 
22812f480046SShri Abhyankar   /* read in row lengths */
2282785e854fSJed Brown   ierr = PetscMalloc1((M+extra_rows),&rowlengths);CHKERRQ(ierr);
22832f480046SShri Abhyankar   ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr);
22842f480046SShri Abhyankar   for (i=0; i<extra_rows; i++) rowlengths[M+i] = 1;
22852f480046SShri Abhyankar 
22862f480046SShri Abhyankar   /* read in column indices */
2287785e854fSJed Brown   ierr = PetscMalloc1((nz+extra_rows),&jj);CHKERRQ(ierr);
22882f480046SShri Abhyankar   ierr = PetscBinaryRead(fd,jj,nz,PETSC_INT);CHKERRQ(ierr);
22892f480046SShri Abhyankar   for (i=0; i<extra_rows; i++) jj[nz+i] = M+i;
22902f480046SShri Abhyankar 
22912f480046SShri Abhyankar   /* loop over row lengths determining block row lengths */
22921795a4d1SJed Brown   ierr     = PetscCalloc1(mbs,&s_browlengths);CHKERRQ(ierr);
2293dcca6d9dSJed Brown   ierr     = PetscMalloc2(mbs,&mask,mbs,&masked);CHKERRQ(ierr);
22942f480046SShri Abhyankar   ierr     = PetscMemzero(mask,mbs*sizeof(PetscInt));CHKERRQ(ierr);
22952f480046SShri Abhyankar   rowcount = 0;
22962f480046SShri Abhyankar   nzcount  = 0;
22972f480046SShri Abhyankar   for (i=0; i<mbs; i++) {
22982f480046SShri Abhyankar     nmask = 0;
22992f480046SShri Abhyankar     for (j=0; j<bs; j++) {
23002f480046SShri Abhyankar       kmax = rowlengths[rowcount];
23012f480046SShri Abhyankar       for (k=0; k<kmax; k++) {
23022f480046SShri Abhyankar         tmp = jj[nzcount++]/bs;   /* block col. index */
23032f480046SShri Abhyankar         if (!mask[tmp] && tmp >= i) {masked[nmask++] = tmp; mask[tmp] = 1;}
23042f480046SShri Abhyankar       }
23052f480046SShri Abhyankar       rowcount++;
23062f480046SShri Abhyankar     }
23072f480046SShri Abhyankar     s_browlengths[i] += nmask;
23082f480046SShri Abhyankar 
23092f480046SShri Abhyankar     /* zero out the mask elements we set */
23102f480046SShri Abhyankar     for (j=0; j<nmask; j++) mask[masked[j]] = 0;
23112f480046SShri Abhyankar   }
23122f480046SShri Abhyankar 
23132f480046SShri Abhyankar   /* Do preallocation */
23142f480046SShri Abhyankar   ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(newmat,bs,0,s_browlengths);CHKERRQ(ierr);
23152f480046SShri Abhyankar   a    = (Mat_SeqSBAIJ*)newmat->data;
23162f480046SShri Abhyankar 
23172f480046SShri Abhyankar   /* set matrix "i" values */
23182f480046SShri Abhyankar   a->i[0] = 0;
23192f480046SShri Abhyankar   for (i=1; i<= mbs; i++) {
23202f480046SShri Abhyankar     a->i[i]      = a->i[i-1] + s_browlengths[i-1];
23212f480046SShri Abhyankar     a->ilen[i-1] = s_browlengths[i-1];
23222f480046SShri Abhyankar   }
23232f480046SShri Abhyankar   a->nz = a->i[mbs];
23242f480046SShri Abhyankar 
23252f480046SShri Abhyankar   /* read in nonzero values */
2326785e854fSJed Brown   ierr = PetscMalloc1((nz+extra_rows),&aa);CHKERRQ(ierr);
23272f480046SShri Abhyankar   ierr = PetscBinaryRead(fd,aa,nz,PETSC_SCALAR);CHKERRQ(ierr);
23282f480046SShri Abhyankar   for (i=0; i<extra_rows; i++) aa[nz+i] = 1.0;
23292f480046SShri Abhyankar 
23302f480046SShri Abhyankar   /* set "a" and "j" values into matrix */
23312f480046SShri Abhyankar   nzcount = 0; jcount = 0;
23322f480046SShri Abhyankar   for (i=0; i<mbs; i++) {
23332f480046SShri Abhyankar     nzcountb = nzcount;
23342f480046SShri Abhyankar     nmask    = 0;
23352f480046SShri Abhyankar     for (j=0; j<bs; j++) {
23362f480046SShri Abhyankar       kmax = rowlengths[i*bs+j];
23372f480046SShri Abhyankar       for (k=0; k<kmax; k++) {
23382f480046SShri Abhyankar         tmp = jj[nzcount++]/bs; /* block col. index */
23392f480046SShri Abhyankar         if (!mask[tmp] && tmp >= i) { masked[nmask++] = tmp; mask[tmp] = 1;}
23402f480046SShri Abhyankar       }
23412f480046SShri Abhyankar     }
23422f480046SShri Abhyankar     /* sort the masked values */
23432f480046SShri Abhyankar     ierr = PetscSortInt(nmask,masked);CHKERRQ(ierr);
23442f480046SShri Abhyankar 
23452f480046SShri Abhyankar     /* set "j" values into matrix */
23462f480046SShri Abhyankar     maskcount = 1;
23472f480046SShri Abhyankar     for (j=0; j<nmask; j++) {
23482f480046SShri Abhyankar       a->j[jcount++]  = masked[j];
23492f480046SShri Abhyankar       mask[masked[j]] = maskcount++;
23502f480046SShri Abhyankar     }
23512f480046SShri Abhyankar 
23522f480046SShri Abhyankar     /* set "a" values into matrix */
23532f480046SShri Abhyankar     ishift = bs2*a->i[i];
23542f480046SShri Abhyankar     for (j=0; j<bs; j++) {
23552f480046SShri Abhyankar       kmax = rowlengths[i*bs+j];
23562f480046SShri Abhyankar       for (k=0; k<kmax; k++) {
23572f480046SShri Abhyankar         tmp = jj[nzcountb]/bs;        /* block col. index */
23582f480046SShri Abhyankar         if (tmp >= i) {
23592f480046SShri Abhyankar           block     = mask[tmp] - 1;
23602f480046SShri Abhyankar           point     = jj[nzcountb] - bs*tmp;
23612f480046SShri Abhyankar           idx       = ishift + bs2*block + j + bs*point;
23622f480046SShri Abhyankar           a->a[idx] = aa[nzcountb];
23632f480046SShri Abhyankar         }
23642f480046SShri Abhyankar         nzcountb++;
23652f480046SShri Abhyankar       }
23662f480046SShri Abhyankar     }
23672f480046SShri Abhyankar     /* zero out the mask elements we set */
23682f480046SShri Abhyankar     for (j=0; j<nmask; j++) mask[masked[j]] = 0;
23692f480046SShri Abhyankar   }
23702f480046SShri Abhyankar   if (jcount != a->nz) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Bad binary matrix");
23712f480046SShri Abhyankar 
23722f480046SShri Abhyankar   ierr = PetscFree(rowlengths);CHKERRQ(ierr);
23732f480046SShri Abhyankar   ierr = PetscFree(s_browlengths);CHKERRQ(ierr);
23742f480046SShri Abhyankar   ierr = PetscFree(aa);CHKERRQ(ierr);
23752f480046SShri Abhyankar   ierr = PetscFree(jj);CHKERRQ(ierr);
23762f480046SShri Abhyankar   ierr = PetscFree2(mask,masked);CHKERRQ(ierr);
23772f480046SShri Abhyankar 
23782f480046SShri Abhyankar   ierr = MatAssemblyBegin(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
23792f480046SShri Abhyankar   ierr = MatAssemblyEnd(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
23802f480046SShri Abhyankar   PetscFunctionReturn(0);
23812f480046SShri Abhyankar }
23822f480046SShri Abhyankar 
23832f480046SShri Abhyankar #undef __FUNCT__
2384c75a6043SHong Zhang #define __FUNCT__ "MatCreateSeqSBAIJWithArrays"
2385c75a6043SHong Zhang /*@
2386c75a6043SHong Zhang      MatCreateSeqSBAIJWithArrays - Creates an sequential SBAIJ matrix using matrix elements
2387c75a6043SHong Zhang               (upper triangular entries in CSR format) provided by the user.
2388c75a6043SHong Zhang 
2389c75a6043SHong Zhang      Collective on MPI_Comm
2390c75a6043SHong Zhang 
2391c75a6043SHong Zhang    Input Parameters:
2392c75a6043SHong Zhang +  comm - must be an MPI communicator of size 1
2393c75a6043SHong Zhang .  bs - size of block
2394c75a6043SHong Zhang .  m - number of rows
2395c75a6043SHong Zhang .  n - number of columns
2396c75a6043SHong Zhang .  i - row indices
2397c75a6043SHong Zhang .  j - column indices
2398c75a6043SHong Zhang -  a - matrix values
2399c75a6043SHong Zhang 
2400c75a6043SHong Zhang    Output Parameter:
2401c75a6043SHong Zhang .  mat - the matrix
2402c75a6043SHong Zhang 
2403dfb205c3SBarry Smith    Level: advanced
2404c75a6043SHong Zhang 
2405c75a6043SHong Zhang    Notes:
2406c75a6043SHong Zhang        The i, j, and a arrays are not copied by this routine, the user must free these arrays
2407c75a6043SHong Zhang     once the matrix is destroyed
2408c75a6043SHong Zhang 
2409c75a6043SHong Zhang        You cannot set new nonzero locations into this matrix, that will generate an error.
2410c75a6043SHong Zhang 
2411c75a6043SHong Zhang        The i and j indices are 0 based
2412c75a6043SHong Zhang 
2413dfb205c3SBarry 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
2414dfb205c3SBarry Smith        it is the regular CSR format excluding the lower triangular elements.
2415dfb205c3SBarry Smith 
241669b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSBAIJ(), MatCreateSeqSBAIJ()
2417c75a6043SHong Zhang 
2418c75a6043SHong Zhang @*/
24197087cfbeSBarry Smith PetscErrorCode  MatCreateSeqSBAIJWithArrays(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt *i,PetscInt *j,PetscScalar *a,Mat *mat)
2420c75a6043SHong Zhang {
2421c75a6043SHong Zhang   PetscErrorCode ierr;
2422c75a6043SHong Zhang   PetscInt       ii;
2423c75a6043SHong Zhang   Mat_SeqSBAIJ   *sbaij;
2424c75a6043SHong Zhang 
2425c75a6043SHong Zhang   PetscFunctionBegin;
2426e32f2f54SBarry Smith   if (bs != 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"block size %D > 1 is not supported yet",bs);
2427e32f2f54SBarry Smith   if (i[0]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"i (row indices) must start with 0");
2428c75a6043SHong Zhang 
2429c75a6043SHong Zhang   ierr  = MatCreate(comm,mat);CHKERRQ(ierr);
2430c75a6043SHong Zhang   ierr  = MatSetSizes(*mat,m,n,m,n);CHKERRQ(ierr);
2431c75a6043SHong Zhang   ierr  = MatSetType(*mat,MATSEQSBAIJ);CHKERRQ(ierr);
2432c75a6043SHong Zhang   ierr  = MatSeqSBAIJSetPreallocation_SeqSBAIJ(*mat,bs,MAT_SKIP_ALLOCATION,0);CHKERRQ(ierr);
2433c75a6043SHong Zhang   sbaij = (Mat_SeqSBAIJ*)(*mat)->data;
2434dcca6d9dSJed Brown   ierr  = PetscMalloc2(m,&sbaij->imax,m,&sbaij->ilen);CHKERRQ(ierr);
24353bb1ff40SBarry Smith   ierr  = PetscLogObjectMemory((PetscObject)*mat,2*m*sizeof(PetscInt));CHKERRQ(ierr);
2436c75a6043SHong Zhang 
2437c75a6043SHong Zhang   sbaij->i = i;
2438c75a6043SHong Zhang   sbaij->j = j;
2439c75a6043SHong Zhang   sbaij->a = a;
244026fbe8dcSKarl Rupp 
2441c75a6043SHong Zhang   sbaij->singlemalloc = PETSC_FALSE;
2442c75a6043SHong Zhang   sbaij->nonew        = -1;             /*this indicates that inserting a new value in the matrix that generates a new nonzero is an error*/
2443e6b907acSBarry Smith   sbaij->free_a       = PETSC_FALSE;
2444e6b907acSBarry Smith   sbaij->free_ij      = PETSC_FALSE;
2445c75a6043SHong Zhang 
2446c75a6043SHong Zhang   for (ii=0; ii<m; ii++) {
2447c75a6043SHong Zhang     sbaij->ilen[ii] = sbaij->imax[ii] = i[ii+1] - i[ii];
2448c75a6043SHong Zhang #if defined(PETSC_USE_DEBUG)
2449e32f2f54SBarry 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]);
2450c75a6043SHong Zhang #endif
2451c75a6043SHong Zhang   }
2452c75a6043SHong Zhang #if defined(PETSC_USE_DEBUG)
2453c75a6043SHong Zhang   for (ii=0; ii<sbaij->i[m]; ii++) {
2454e32f2f54SBarry Smith     if (j[ii] < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column index at location = %d index = %d",ii,j[ii]);
2455e32f2f54SBarry 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]);
2456c75a6043SHong Zhang   }
2457c75a6043SHong Zhang #endif
2458c75a6043SHong Zhang 
2459c75a6043SHong Zhang   ierr = MatAssemblyBegin(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2460c75a6043SHong Zhang   ierr = MatAssemblyEnd(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2461c75a6043SHong Zhang   PetscFunctionReturn(0);
2462c75a6043SHong Zhang }
2463d06b337dSHong Zhang 
2464d06b337dSHong Zhang 
2465d06b337dSHong Zhang 
246649b5e25fSSatish Balay 
246749b5e25fSSatish Balay 
2468