xref: /petsc/src/mat/impls/sbaij/seq/sbaij.c (revision 31ce2d13fe61e17a5733e77466f3f6eef81b3347)
149b5e25fSSatish Balay 
249b5e25fSSatish Balay /*
3a1373b80SHong Zhang     Defines the basic matrix operations for the SBAIJ (compressed row)
449b5e25fSSatish Balay   matrix storage format.
549b5e25fSSatish Balay */
6c6db04a5SJed Brown #include <../src/mat/impls/baij/seq/baij.h>         /*I "petscmat.h" I*/
7c6db04a5SJed Brown #include <../src/mat/impls/sbaij/seq/sbaij.h>
8c6db04a5SJed Brown #include <petscblaslapack.h>
949b5e25fSSatish Balay 
10c6db04a5SJed Brown #include <../src/mat/impls/sbaij/seq/relax.h>
1170dcbbb9SBarry Smith #define USESHORT
12c6db04a5SJed Brown #include <../src/mat/impls/sbaij/seq/relax.h>
1370dcbbb9SBarry Smith 
14ace3abfcSBarry Smith extern PetscErrorCode MatSeqSBAIJSetNumericFactorization_inplace(Mat,PetscBool);
15b5b17502SBarry Smith 
1649b5e25fSSatish Balay /*
1749b5e25fSSatish Balay      Checks for missing diagonals
1849b5e25fSSatish Balay */
194a2ae208SSatish Balay #undef __FUNCT__
204a2ae208SSatish Balay #define __FUNCT__ "MatMissingDiagonal_SeqSBAIJ"
21ace3abfcSBarry Smith PetscErrorCode MatMissingDiagonal_SeqSBAIJ(Mat A,PetscBool  *missing,PetscInt *dd)
2249b5e25fSSatish Balay {
23045c9aa0SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
246849ba73SBarry Smith   PetscErrorCode ierr;
257734d3b5SMatthew G. Knepley   PetscInt       *diag,*ii = a->i,i;
2649b5e25fSSatish Balay 
2749b5e25fSSatish Balay   PetscFunctionBegin;
28045c9aa0SHong Zhang   ierr     = MatMarkDiagonal_SeqSBAIJ(A);CHKERRQ(ierr);
292af78befSBarry Smith   *missing = PETSC_FALSE;
307734d3b5SMatthew G. Knepley   if (A->rmap->n > 0 && !ii) {
31358d2f5dSShri Abhyankar     *missing = PETSC_TRUE;
32358d2f5dSShri Abhyankar     if (dd) *dd = 0;
33358d2f5dSShri Abhyankar     PetscInfo(A,"Matrix has no entries therefore is missing diagonal");
34358d2f5dSShri Abhyankar   } else {
35358d2f5dSShri Abhyankar     diag = a->diag;
3649b5e25fSSatish Balay     for (i=0; i<a->mbs; i++) {
377734d3b5SMatthew G. Knepley       if (diag[i] >= ii[i+1]) {
382af78befSBarry Smith         *missing = PETSC_TRUE;
392af78befSBarry Smith         if (dd) *dd = i;
402af78befSBarry Smith         break;
412af78befSBarry Smith       }
4249b5e25fSSatish Balay     }
43358d2f5dSShri Abhyankar   }
4449b5e25fSSatish Balay   PetscFunctionReturn(0);
4549b5e25fSSatish Balay }
4649b5e25fSSatish Balay 
474a2ae208SSatish Balay #undef __FUNCT__
484a2ae208SSatish Balay #define __FUNCT__ "MatMarkDiagonal_SeqSBAIJ"
49dfbe8321SBarry Smith PetscErrorCode MatMarkDiagonal_SeqSBAIJ(Mat A)
5049b5e25fSSatish Balay {
51045c9aa0SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
526849ba73SBarry Smith   PetscErrorCode ierr;
5348dd3d27SHong Zhang   PetscInt       i,j;
5449b5e25fSSatish Balay 
5549b5e25fSSatish Balay   PetscFunctionBegin;
5609f38230SBarry Smith   if (!a->diag) {
57785e854fSJed Brown     ierr         = PetscMalloc1(a->mbs,&a->diag);CHKERRQ(ierr);
583bb1ff40SBarry Smith     ierr         = PetscLogObjectMemory((PetscObject)A,a->mbs*sizeof(PetscInt));CHKERRQ(ierr);
59c760cd28SBarry Smith     a->free_diag = PETSC_TRUE;
6009f38230SBarry Smith   }
6148dd3d27SHong Zhang   for (i=0; i<a->mbs; i++) {
6248dd3d27SHong Zhang     a->diag[i] = a->i[i+1];
6348dd3d27SHong Zhang     for (j=a->i[i]; j<a->i[i+1]; j++) {
6448dd3d27SHong Zhang       if (a->j[j] == i) {
6548dd3d27SHong Zhang         a->diag[i] = j;
6648dd3d27SHong Zhang         break;
6748dd3d27SHong Zhang       }
6848dd3d27SHong Zhang     }
6948dd3d27SHong Zhang   }
7049b5e25fSSatish Balay   PetscFunctionReturn(0);
7149b5e25fSSatish Balay }
7249b5e25fSSatish Balay 
734a2ae208SSatish Balay #undef __FUNCT__
744a2ae208SSatish Balay #define __FUNCT__ "MatGetRowIJ_SeqSBAIJ"
751a83f524SJed Brown static PetscErrorCode MatGetRowIJ_SeqSBAIJ(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *nn,const PetscInt *inia[],const PetscInt *inja[],PetscBool  *done)
7649b5e25fSSatish Balay {
77a6ece127SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
78d0f46423SBarry Smith   PetscInt       i,j,n = a->mbs,nz = a->i[n],bs = A->rmap->bs;
791a83f524SJed Brown   PetscInt       **ia = (PetscInt**)inia,**ja = (PetscInt**)inja;
808f7157efSSatish Balay   PetscErrorCode ierr;
8149b5e25fSSatish Balay 
8249b5e25fSSatish Balay   PetscFunctionBegin;
83d3e5a4abSHong Zhang   *nn = n;
84a1373b80SHong Zhang   if (!ia) PetscFunctionReturn(0);
858f7157efSSatish Balay   if (!blockcompressed) {
868f7157efSSatish Balay     /* malloc & create the natural set of indices */
87dcca6d9dSJed Brown     ierr = PetscMalloc2((n+1)*bs,ia,nz*bs,ja);CHKERRQ(ierr);
888f7157efSSatish Balay     for (i=0; i<n+1; i++) {
898f7157efSSatish Balay       for (j=0; j<bs; j++) {
908f7157efSSatish Balay         *ia[i*bs+j] = a->i[i]*bs+j+oshift;
918f7157efSSatish Balay       }
928f7157efSSatish Balay     }
938f7157efSSatish Balay     for (i=0; i<nz; i++) {
948f7157efSSatish Balay       for (j=0; j<bs; j++) {
958f7157efSSatish Balay         *ja[i*bs+j] = a->j[i]*bs+j+oshift;
968f7157efSSatish Balay       }
978f7157efSSatish Balay     }
988f7157efSSatish Balay   } else { /* blockcompressed */
99a6ece127SHong Zhang     if (oshift == 1) {
100a6ece127SHong Zhang       /* temporarily add 1 to i and j indices */
1016c6c5352SBarry Smith       for (i=0; i<nz; i++) a->j[i]++;
102a1373b80SHong Zhang       for (i=0; i<n+1; i++) a->i[i]++;
1038f7157efSSatish Balay     }
104a1373b80SHong Zhang     *ia = a->i; *ja = a->j;
105a6ece127SHong Zhang   }
10649b5e25fSSatish Balay   PetscFunctionReturn(0);
10749b5e25fSSatish Balay }
10849b5e25fSSatish Balay 
1094a2ae208SSatish Balay #undef __FUNCT__
1104a2ae208SSatish Balay #define __FUNCT__ "MatRestoreRowIJ_SeqSBAIJ"
1111a83f524SJed Brown static PetscErrorCode MatRestoreRowIJ_SeqSBAIJ(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *nn,const PetscInt *ia[],const PetscInt *ja[],PetscBool  *done)
11249b5e25fSSatish Balay {
113b7aaefc3SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
1148f7157efSSatish Balay   PetscInt       i,n = a->mbs,nz = a->i[n];
1158f7157efSSatish Balay   PetscErrorCode ierr;
116a6ece127SHong Zhang 
11749b5e25fSSatish Balay   PetscFunctionBegin;
11849b5e25fSSatish Balay   if (!ia) PetscFunctionReturn(0);
119a6ece127SHong Zhang 
1208f7157efSSatish Balay   if (!blockcompressed) {
1218f7157efSSatish Balay     ierr = PetscFree2(*ia,*ja);CHKERRQ(ierr);
1228f7157efSSatish Balay   } else if (oshift == 1) { /* blockcompressed */
1236c6c5352SBarry Smith     for (i=0; i<nz; i++) a->j[i]--;
124a6ece127SHong Zhang     for (i=0; i<n+1; i++) a->i[i]--;
125a6ece127SHong Zhang   }
126a6ece127SHong Zhang   PetscFunctionReturn(0);
12749b5e25fSSatish Balay }
12849b5e25fSSatish Balay 
1294a2ae208SSatish Balay #undef __FUNCT__
1304a2ae208SSatish Balay #define __FUNCT__ "MatDestroy_SeqSBAIJ"
131dfbe8321SBarry Smith PetscErrorCode MatDestroy_SeqSBAIJ(Mat A)
13249b5e25fSSatish Balay {
13349b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
134dfbe8321SBarry Smith   PetscErrorCode ierr;
13549b5e25fSSatish Balay 
13649b5e25fSSatish Balay   PetscFunctionBegin;
137a9f03627SSatish Balay #if defined(PETSC_USE_LOG)
138d0f46423SBarry Smith   PetscLogObjectState((PetscObject)A,"Rows=%D, NZ=%D",A->rmap->N,a->nz);
139a9f03627SSatish Balay #endif
140e6b907acSBarry Smith   ierr = MatSeqXAIJFreeAIJ(A,&a->a,&a->j,&a->i);CHKERRQ(ierr);
1417f53bb6cSHong Zhang   if (a->free_diag) {ierr = PetscFree(a->diag);CHKERRQ(ierr);}
1426bf464f9SBarry Smith   ierr = ISDestroy(&a->row);CHKERRQ(ierr);
1436bf464f9SBarry Smith   ierr = ISDestroy(&a->col);CHKERRQ(ierr);
1446bf464f9SBarry Smith   ierr = ISDestroy(&a->icol);CHKERRQ(ierr);
145c31cb41cSBarry Smith   ierr = PetscFree(a->idiag);CHKERRQ(ierr);
146c31cb41cSBarry Smith   ierr = PetscFree(a->inode.size);CHKERRQ(ierr);
147c760cd28SBarry Smith   if (a->free_imax_ilen) {ierr = PetscFree2(a->imax,a->ilen);CHKERRQ(ierr);}
14805b42c5fSBarry Smith   ierr = PetscFree(a->solve_work);CHKERRQ(ierr);
14941f059aeSBarry Smith   ierr = PetscFree(a->sor_work);CHKERRQ(ierr);
15005b42c5fSBarry Smith   ierr = PetscFree(a->solves_work);CHKERRQ(ierr);
15105b42c5fSBarry Smith   ierr = PetscFree(a->mult_work);CHKERRQ(ierr);
15205b42c5fSBarry Smith   ierr = PetscFree(a->saved_values);CHKERRQ(ierr);
1534da8f245SBarry Smith   if (a->free_jshort) {ierr = PetscFree(a->jshort);CHKERRQ(ierr);}
1541a3463dfSHong Zhang   ierr = PetscFree(a->inew);CHKERRQ(ierr);
1556bf464f9SBarry Smith   ierr = MatDestroy(&a->parent);CHKERRQ(ierr);
156bf0cc555SLisandro Dalcin   ierr = PetscFree(A->data);CHKERRQ(ierr);
157901853e0SKris Buschelman 
158dbd8c25aSHong Zhang   ierr = PetscObjectChangeTypeName((PetscObject)A,0);CHKERRQ(ierr);
159bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatStoreValues_C",NULL);CHKERRQ(ierr);
160bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatRetrieveValues_C",NULL);CHKERRQ(ierr);
161bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetColumnIndices_C",NULL);CHKERRQ(ierr);
162bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqaij_C",NULL);CHKERRQ(ierr);
163bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqbaij_C",NULL);CHKERRQ(ierr);
164bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetPreallocation_C",NULL);CHKERRQ(ierr);
16538f409ebSLisandro Dalcin   ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetPreallocationCSR_C",NULL);CHKERRQ(ierr);
166bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqsbstrm_C",NULL);CHKERRQ(ierr);
16749b5e25fSSatish Balay   PetscFunctionReturn(0);
16849b5e25fSSatish Balay }
16949b5e25fSSatish Balay 
1704a2ae208SSatish Balay #undef __FUNCT__
1714a2ae208SSatish Balay #define __FUNCT__ "MatSetOption_SeqSBAIJ"
172ace3abfcSBarry Smith PetscErrorCode MatSetOption_SeqSBAIJ(Mat A,MatOption op,PetscBool flg)
17349b5e25fSSatish Balay {
174045c9aa0SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
17563ba0a88SBarry Smith   PetscErrorCode ierr;
17649b5e25fSSatish Balay 
17749b5e25fSSatish Balay   PetscFunctionBegin;
1784d9d31abSKris Buschelman   switch (op) {
1794d9d31abSKris Buschelman   case MAT_ROW_ORIENTED:
1804e0d8c25SBarry Smith     a->roworiented = flg;
1814d9d31abSKris Buschelman     break;
182a9817697SBarry Smith   case MAT_KEEP_NONZERO_PATTERN:
183a9817697SBarry Smith     a->keepnonzeropattern = flg;
1844d9d31abSKris Buschelman     break;
185512a5fc5SBarry Smith   case MAT_NEW_NONZERO_LOCATIONS:
186512a5fc5SBarry Smith     a->nonew = (flg ? 0 : 1);
1874d9d31abSKris Buschelman     break;
1884d9d31abSKris Buschelman   case MAT_NEW_NONZERO_LOCATION_ERR:
1894e0d8c25SBarry Smith     a->nonew = (flg ? -1 : 0);
1904d9d31abSKris Buschelman     break;
1914d9d31abSKris Buschelman   case MAT_NEW_NONZERO_ALLOCATION_ERR:
1924e0d8c25SBarry Smith     a->nonew = (flg ? -2 : 0);
1934d9d31abSKris Buschelman     break;
19428b2fa4aSMatthew Knepley   case MAT_UNUSED_NONZERO_LOCATION_ERR:
19528b2fa4aSMatthew Knepley     a->nounused = (flg ? -1 : 0);
19628b2fa4aSMatthew Knepley     break;
1974e0d8c25SBarry Smith   case MAT_NEW_DIAGONALS:
1984d9d31abSKris Buschelman   case MAT_IGNORE_OFF_PROC_ENTRIES:
1994d9d31abSKris Buschelman   case MAT_USE_HASH_TABLE:
200290bbb0aSBarry Smith     ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr);
2014d9d31abSKris Buschelman     break;
2029a4540c5SBarry Smith   case MAT_HERMITIAN:
203e32f2f54SBarry Smith     if (!A->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Must call MatAssemblyEnd() first");
2040bc54ff2SBarry Smith     if (A->cmap->n < 65536 && A->cmap->bs == 1) {
205eeffb40dSHong Zhang       A->ops->mult = MatMult_SeqSBAIJ_1_Hermitian_ushort;
2060bc54ff2SBarry Smith     } else if (A->cmap->bs == 1) {
207eeffb40dSHong Zhang       A->ops->mult = MatMult_SeqSBAIJ_1_Hermitian;
208e32f2f54SBarry Smith     } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No support for Hermitian with block size greater than 1");
209eeffb40dSHong Zhang     break;
2103d472b54SHong Zhang   case MAT_SPD:
2115021d80fSJed Brown     /* These options are handled directly by MatSetOption() */
2123d472b54SHong Zhang     break;
21377e54ba9SKris Buschelman   case MAT_SYMMETRIC:
21477e54ba9SKris Buschelman   case MAT_STRUCTURALLY_SYMMETRIC:
2159a4540c5SBarry Smith   case MAT_SYMMETRY_ETERNAL:
216e32f2f54SBarry Smith     if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix must be symmetric");
217290bbb0aSBarry Smith     ierr = PetscInfo1(A,"Option %s not relevent\n",MatOptions[op]);CHKERRQ(ierr);
218290bbb0aSBarry Smith     break;
219941593c8SHong Zhang   case MAT_IGNORE_LOWER_TRIANGULAR:
2204e0d8c25SBarry Smith     a->ignore_ltriangular = flg;
221941593c8SHong Zhang     break;
222941593c8SHong Zhang   case MAT_ERROR_LOWER_TRIANGULAR:
2234e0d8c25SBarry Smith     a->ignore_ltriangular = flg;
22477e54ba9SKris Buschelman     break;
225f5edf698SHong Zhang   case MAT_GETROW_UPPERTRIANGULAR:
2264e0d8c25SBarry Smith     a->getrow_utriangular = flg;
227f5edf698SHong Zhang     break;
2284d9d31abSKris Buschelman   default:
229e32f2f54SBarry Smith     SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"unknown option %d",op);
23049b5e25fSSatish Balay   }
23149b5e25fSSatish Balay   PetscFunctionReturn(0);
23249b5e25fSSatish Balay }
23349b5e25fSSatish Balay 
2344a2ae208SSatish Balay #undef __FUNCT__
2354a2ae208SSatish Balay #define __FUNCT__ "MatGetRow_SeqSBAIJ"
23652768537SHong Zhang PetscErrorCode MatGetRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
23749b5e25fSSatish Balay {
23849b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
2396849ba73SBarry Smith   PetscErrorCode ierr;
24049b5e25fSSatish Balay 
24149b5e25fSSatish Balay   PetscFunctionBegin;
242e32f2f54SBarry 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()");
24352768537SHong Zhang 
244f5edf698SHong Zhang   /* Get the upper triangular part of the row */
24552768537SHong Zhang   ierr = MatGetRow_SeqBAIJ_private(A,row,nz,idx,v,a->i,a->j,a->a);CHKERRQ(ierr);
24649b5e25fSSatish Balay   PetscFunctionReturn(0);
24749b5e25fSSatish Balay }
24849b5e25fSSatish Balay 
2494a2ae208SSatish Balay #undef __FUNCT__
2504a2ae208SSatish Balay #define __FUNCT__ "MatRestoreRow_SeqSBAIJ"
25113f74950SBarry Smith PetscErrorCode MatRestoreRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v)
25249b5e25fSSatish Balay {
253dfbe8321SBarry Smith   PetscErrorCode ierr;
25449b5e25fSSatish Balay 
25549b5e25fSSatish Balay   PetscFunctionBegin;
25605b42c5fSBarry Smith   if (idx) {ierr = PetscFree(*idx);CHKERRQ(ierr);}
25705b42c5fSBarry Smith   if (v)   {ierr = PetscFree(*v);CHKERRQ(ierr);}
25849b5e25fSSatish Balay   PetscFunctionReturn(0);
25949b5e25fSSatish Balay }
26049b5e25fSSatish Balay 
2614a2ae208SSatish Balay #undef __FUNCT__
262f5edf698SHong Zhang #define __FUNCT__ "MatGetRowUpperTriangular_SeqSBAIJ"
263f5edf698SHong Zhang PetscErrorCode MatGetRowUpperTriangular_SeqSBAIJ(Mat A)
264f5edf698SHong Zhang {
265f5edf698SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
266f5edf698SHong Zhang 
267f5edf698SHong Zhang   PetscFunctionBegin;
268f5edf698SHong Zhang   a->getrow_utriangular = PETSC_TRUE;
269f5edf698SHong Zhang   PetscFunctionReturn(0);
270f5edf698SHong Zhang }
271f5edf698SHong Zhang #undef __FUNCT__
272f5edf698SHong Zhang #define __FUNCT__ "MatRestoreRowUpperTriangular_SeqSBAIJ"
273f5edf698SHong Zhang PetscErrorCode MatRestoreRowUpperTriangular_SeqSBAIJ(Mat A)
274f5edf698SHong Zhang {
275f5edf698SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
276f5edf698SHong Zhang 
277f5edf698SHong Zhang   PetscFunctionBegin;
278f5edf698SHong Zhang   a->getrow_utriangular = PETSC_FALSE;
279f5edf698SHong Zhang   PetscFunctionReturn(0);
280f5edf698SHong Zhang }
281f5edf698SHong Zhang 
282f5edf698SHong Zhang #undef __FUNCT__
2834a2ae208SSatish Balay #define __FUNCT__ "MatTranspose_SeqSBAIJ"
284fc4dec0aSBarry Smith PetscErrorCode MatTranspose_SeqSBAIJ(Mat A,MatReuse reuse,Mat *B)
28549b5e25fSSatish Balay {
286dfbe8321SBarry Smith   PetscErrorCode ierr;
2875fd66863SKarl Rupp 
28849b5e25fSSatish Balay   PetscFunctionBegin;
289815cbec1SBarry Smith   if (reuse == MAT_INITIAL_MATRIX || *B != A) {
290999d9058SBarry Smith     ierr = MatDuplicate(A,MAT_COPY_VALUES,B);CHKERRQ(ierr);
291fc4dec0aSBarry Smith   }
2928115998fSBarry Smith   PetscFunctionReturn(0);
29349b5e25fSSatish Balay }
29449b5e25fSSatish Balay 
2954a2ae208SSatish Balay #undef __FUNCT__
2964a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_ASCII"
2977da1fb6eSBarry Smith PetscErrorCode MatView_SeqSBAIJ_ASCII(Mat A,PetscViewer viewer)
29849b5e25fSSatish Balay {
29949b5e25fSSatish Balay   Mat_SeqSBAIJ      *a = (Mat_SeqSBAIJ*)A->data;
300dfbe8321SBarry Smith   PetscErrorCode    ierr;
301d0f46423SBarry Smith   PetscInt          i,j,bs = A->rmap->bs,k,l,bs2=a->bs2;
302f3ef73ceSBarry Smith   PetscViewerFormat format;
303121deb67SSatish Balay   PetscInt          *diag;
30449b5e25fSSatish Balay 
30549b5e25fSSatish Balay   PetscFunctionBegin;
306b0a32e0cSBarry Smith   ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
307456192e2SBarry Smith   if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
30877431f27SBarry Smith     ierr = PetscViewerASCIIPrintf(viewer,"  block size is %D\n",bs);CHKERRQ(ierr);
309fb9695e5SSatish Balay   } else if (format == PETSC_VIEWER_ASCII_MATLAB) {
310d2507d54SMatthew Knepley     Mat aij;
311d5f3da31SBarry Smith     if (A->factortype && bs>1) {
31270d5e725SHong 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);
31370d5e725SHong Zhang       PetscFunctionReturn(0);
31470d5e725SHong Zhang     }
315c9f458caSMatthew Knepley     ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&aij);CHKERRQ(ierr);
316c9f458caSMatthew Knepley     ierr = MatView(aij,viewer);CHKERRQ(ierr);
3176bf464f9SBarry Smith     ierr = MatDestroy(&aij);CHKERRQ(ierr);
318fb9695e5SSatish Balay   } else if (format == PETSC_VIEWER_ASCII_COMMON) {
319d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
32049b5e25fSSatish Balay     for (i=0; i<a->mbs; i++) {
32149b5e25fSSatish Balay       for (j=0; j<bs; j++) {
32277431f27SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr);
32349b5e25fSSatish Balay         for (k=a->i[i]; k<a->i[i+1]; k++) {
32449b5e25fSSatish Balay           for (l=0; l<bs; l++) {
32549b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX)
32649b5e25fSSatish Balay             if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
32757622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k]+l,
32857622a8eSBarry Smith                                             (double)PetscRealPart(a->a[bs2*k + l*bs + j]),(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
32949b5e25fSSatish Balay             } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
33057622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k]+l,
33157622a8eSBarry Smith                                             (double)PetscRealPart(a->a[bs2*k + l*bs + j]),-(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
33249b5e25fSSatish Balay             } else if (PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) {
33357622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
33449b5e25fSSatish Balay             }
33549b5e25fSSatish Balay #else
33649b5e25fSSatish Balay             if (a->a[bs2*k + l*bs + j] != 0.0) {
33757622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)a->a[bs2*k + l*bs + j]);CHKERRQ(ierr);
33849b5e25fSSatish Balay             }
33949b5e25fSSatish Balay #endif
34049b5e25fSSatish Balay           }
34149b5e25fSSatish Balay         }
342b0a32e0cSBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
34349b5e25fSSatish Balay       }
34449b5e25fSSatish Balay     }
345d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
346c1490034SHong Zhang   } else if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) {
347c1490034SHong Zhang     PetscFunctionReturn(0);
34849b5e25fSSatish Balay   } else {
349d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
3502c990fa1SHong Zhang     if (A->factortype) { /* for factored matrix */
3512c990fa1SHong Zhang       if (bs>1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"matrix is factored with bs>1. Not implemented yet");
3522c990fa1SHong Zhang 
353121deb67SSatish Balay       diag=a->diag;
354121deb67SSatish Balay       for (i=0; i<a->mbs; i++) { /* for row block i */
3552c990fa1SHong Zhang         ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i);CHKERRQ(ierr);
3562c990fa1SHong Zhang         /* diagonal entry */
3572c990fa1SHong Zhang #if defined(PETSC_USE_COMPLEX)
3582c990fa1SHong Zhang         if (PetscImaginaryPart(a->a[diag[i]]) > 0.0) {
35957622a8eSBarry 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);
3602c990fa1SHong Zhang         } else if (PetscImaginaryPart(a->a[diag[i]]) < 0.0) {
36157622a8eSBarry 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);
3622c990fa1SHong Zhang         } else {
36357622a8eSBarry Smith           ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[diag[i]],(double)PetscRealPart(1.0/a->a[diag[i]]));CHKERRQ(ierr);
3642c990fa1SHong Zhang         }
3652c990fa1SHong Zhang #else
3666712e2f1SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[diag[i]],(double)(1.0/a->a[diag[i]]));CHKERRQ(ierr);
3672c990fa1SHong Zhang #endif
3682c990fa1SHong Zhang         /* off-diagonal entries */
3692c990fa1SHong Zhang         for (k=a->i[i]; k<a->i[i+1]-1; k++) {
3702c990fa1SHong Zhang #if defined(PETSC_USE_COMPLEX)
371ca0704adSBarry Smith           if (PetscImaginaryPart(a->a[k]) > 0.0) {
37257622a8eSBarry Smith             ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k],(double)PetscRealPart(a->a[k]),(double)PetscImaginaryPart(a->a[k]));CHKERRQ(ierr);
373ca0704adSBarry Smith           } else if (PetscImaginaryPart(a->a[k]) < 0.0) {
37457622a8eSBarry Smith             ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k],(double)PetscRealPart(a->a[k]),-(double)PetscImaginaryPart(a->a[k]));CHKERRQ(ierr);
3752c990fa1SHong Zhang           } else {
37657622a8eSBarry Smith             ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k],(double)PetscRealPart(a->a[k]));CHKERRQ(ierr);
3772c990fa1SHong Zhang           }
3782c990fa1SHong Zhang #else
37957622a8eSBarry Smith           ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",a->j[k],(double)a->a[k]);CHKERRQ(ierr);
3802c990fa1SHong Zhang #endif
3812c990fa1SHong Zhang         }
3822c990fa1SHong Zhang         ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
3832c990fa1SHong Zhang       }
3842c990fa1SHong Zhang 
3852c990fa1SHong Zhang     } else { /* for non-factored matrix */
3860c74a584SJed Brown       for (i=0; i<a->mbs; i++) { /* for row block i */
3870c74a584SJed Brown         for (j=0; j<bs; j++) {   /* for row bs*i + j */
38877431f27SBarry Smith           ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr);
3890c74a584SJed Brown           for (k=a->i[i]; k<a->i[i+1]; k++) { /* for column block */
3900c74a584SJed Brown             for (l=0; l<bs; l++) {            /* for column */
39149b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX)
39249b5e25fSSatish Balay               if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0) {
39357622a8eSBarry Smith                 ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k]+l,
39457622a8eSBarry Smith                                               (double)PetscRealPart(a->a[bs2*k + l*bs + j]),(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
39549b5e25fSSatish Balay               } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0) {
39657622a8eSBarry Smith                 ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k]+l,
39757622a8eSBarry Smith                                               (double)PetscRealPart(a->a[bs2*k + l*bs + j]),-(double)PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
39849b5e25fSSatish Balay               } else {
39957622a8eSBarry Smith                 ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr);
40049b5e25fSSatish Balay               }
40149b5e25fSSatish Balay #else
40257622a8eSBarry Smith               ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,(double)a->a[bs2*k + l*bs + j]);CHKERRQ(ierr);
40349b5e25fSSatish Balay #endif
40449b5e25fSSatish Balay             }
40549b5e25fSSatish Balay           }
406b0a32e0cSBarry Smith           ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
40749b5e25fSSatish Balay         }
40849b5e25fSSatish Balay       }
4092c990fa1SHong Zhang     }
410d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
41149b5e25fSSatish Balay   }
412b0a32e0cSBarry Smith   ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
41349b5e25fSSatish Balay   PetscFunctionReturn(0);
41449b5e25fSSatish Balay }
41549b5e25fSSatish Balay 
4169804daf3SBarry Smith #include <petscdraw.h>
4174a2ae208SSatish Balay #undef __FUNCT__
4184a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_Draw_Zoom"
4196849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw_Zoom(PetscDraw draw,void *Aa)
42049b5e25fSSatish Balay {
42149b5e25fSSatish Balay   Mat            A = (Mat) Aa;
42249b5e25fSSatish Balay   Mat_SeqSBAIJ   *a=(Mat_SeqSBAIJ*)A->data;
4236849ba73SBarry Smith   PetscErrorCode ierr;
424d0f46423SBarry Smith   PetscInt       row,i,j,k,l,mbs=a->mbs,color,bs=A->rmap->bs,bs2=a->bs2;
42513f74950SBarry Smith   PetscMPIInt    rank;
42649b5e25fSSatish Balay   PetscReal      xl,yl,xr,yr,x_l,x_r,y_l,y_r;
42749b5e25fSSatish Balay   MatScalar      *aa;
42849b5e25fSSatish Balay   MPI_Comm       comm;
429b0a32e0cSBarry Smith   PetscViewer    viewer;
43049b5e25fSSatish Balay 
43149b5e25fSSatish Balay   PetscFunctionBegin;
43249b5e25fSSatish Balay   /*
43349b5e25fSSatish Balay     This is nasty. If this is called from an originally parallel matrix
43449b5e25fSSatish Balay     then all processes call this,but only the first has the matrix so the
43549b5e25fSSatish Balay     rest should return immediately.
43649b5e25fSSatish Balay   */
43749b5e25fSSatish Balay   ierr = PetscObjectGetComm((PetscObject)draw,&comm);CHKERRQ(ierr);
43849b5e25fSSatish Balay   ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr);
43949b5e25fSSatish Balay   if (rank) PetscFunctionReturn(0);
44049b5e25fSSatish Balay 
44149b5e25fSSatish Balay   ierr = PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer);CHKERRQ(ierr);
44249b5e25fSSatish Balay 
443b0a32e0cSBarry Smith   ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr);
444b0a32e0cSBarry Smith   PetscDrawString(draw, .3*(xl+xr), .3*(yl+yr), PETSC_DRAW_BLACK, "symmetric");
44549b5e25fSSatish Balay 
44649b5e25fSSatish Balay   /* loop over matrix elements drawing boxes */
447b0a32e0cSBarry Smith   color = PETSC_DRAW_BLUE;
44849b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
44949b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
450d0f46423SBarry Smith       y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0;
45149b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
45249b5e25fSSatish Balay       aa  = a->a + j*bs2;
45349b5e25fSSatish Balay       for (k=0; k<bs; k++) {
45449b5e25fSSatish Balay         for (l=0; l<bs; l++) {
45549b5e25fSSatish Balay           if (PetscRealPart(*aa++) >=  0.) continue;
456b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
45749b5e25fSSatish Balay         }
45849b5e25fSSatish Balay       }
45949b5e25fSSatish Balay     }
46049b5e25fSSatish Balay   }
461b0a32e0cSBarry Smith   color = PETSC_DRAW_CYAN;
46249b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
46349b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
464d0f46423SBarry Smith       y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0;
46549b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
46649b5e25fSSatish Balay       aa = a->a + j*bs2;
46749b5e25fSSatish Balay       for (k=0; k<bs; k++) {
46849b5e25fSSatish Balay         for (l=0; l<bs; l++) {
46949b5e25fSSatish Balay           if (PetscRealPart(*aa++) != 0.) continue;
470b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
47149b5e25fSSatish Balay         }
47249b5e25fSSatish Balay       }
47349b5e25fSSatish Balay     }
47449b5e25fSSatish Balay   }
47549b5e25fSSatish Balay 
476b0a32e0cSBarry Smith   color = PETSC_DRAW_RED;
47749b5e25fSSatish Balay   for (i=0,row=0; i<mbs; i++,row+=bs) {
47849b5e25fSSatish Balay     for (j=a->i[i]; j<a->i[i+1]; j++) {
479d0f46423SBarry Smith       y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0;
48049b5e25fSSatish Balay       x_l = a->j[j]*bs; x_r = x_l + 1.0;
48149b5e25fSSatish Balay       aa = a->a + j*bs2;
48249b5e25fSSatish Balay       for (k=0; k<bs; k++) {
48349b5e25fSSatish Balay         for (l=0; l<bs; l++) {
48449b5e25fSSatish Balay           if (PetscRealPart(*aa++) <= 0.) continue;
485b0a32e0cSBarry Smith           ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr);
48649b5e25fSSatish Balay         }
48749b5e25fSSatish Balay       }
48849b5e25fSSatish Balay     }
48949b5e25fSSatish Balay   }
49049b5e25fSSatish Balay   PetscFunctionReturn(0);
49149b5e25fSSatish Balay }
49249b5e25fSSatish Balay 
4934a2ae208SSatish Balay #undef __FUNCT__
4944a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_Draw"
4956849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw(Mat A,PetscViewer viewer)
49649b5e25fSSatish Balay {
497dfbe8321SBarry Smith   PetscErrorCode ierr;
49849b5e25fSSatish Balay   PetscReal      xl,yl,xr,yr,w,h;
499b0a32e0cSBarry Smith   PetscDraw      draw;
500ace3abfcSBarry Smith   PetscBool      isnull;
50149b5e25fSSatish Balay 
50249b5e25fSSatish Balay   PetscFunctionBegin;
503b0a32e0cSBarry Smith   ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
504b0a32e0cSBarry Smith   ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr); if (isnull) PetscFunctionReturn(0);
50549b5e25fSSatish Balay 
50649b5e25fSSatish Balay   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer);CHKERRQ(ierr);
507d0f46423SBarry Smith   xr   = A->rmap->N; yr = A->rmap->N; h = yr/10.0; w = xr/10.0;
50849b5e25fSSatish Balay   xr  += w;    yr += h;  xl = -w;     yl = -h;
509b0a32e0cSBarry Smith   ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr);
510b0a32e0cSBarry Smith   ierr = PetscDrawZoom(draw,MatView_SeqSBAIJ_Draw_Zoom,A);CHKERRQ(ierr);
5110298fd71SBarry Smith   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",NULL);CHKERRQ(ierr);
51249b5e25fSSatish Balay   PetscFunctionReturn(0);
51349b5e25fSSatish Balay }
51449b5e25fSSatish Balay 
5154a2ae208SSatish Balay #undef __FUNCT__
5164a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ"
517dfbe8321SBarry Smith PetscErrorCode MatView_SeqSBAIJ(Mat A,PetscViewer viewer)
51849b5e25fSSatish Balay {
519dfbe8321SBarry Smith   PetscErrorCode ierr;
520ace3abfcSBarry Smith   PetscBool      iascii,isdraw;
52108917f38SBarry Smith   FILE           *file = 0;
52249b5e25fSSatish Balay 
52349b5e25fSSatish Balay   PetscFunctionBegin;
524251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
525251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr);
52632077d6dSBarry Smith   if (iascii) {
52749b5e25fSSatish Balay     ierr = MatView_SeqSBAIJ_ASCII(A,viewer);CHKERRQ(ierr);
52849b5e25fSSatish Balay   } else if (isdraw) {
52949b5e25fSSatish Balay     ierr = MatView_SeqSBAIJ_Draw(A,viewer);CHKERRQ(ierr);
53049b5e25fSSatish Balay   } else {
531a5e6ed63SBarry Smith     Mat B;
532ceb03754SKris Buschelman     ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&B);CHKERRQ(ierr);
533a5e6ed63SBarry Smith     ierr = MatView(B,viewer);CHKERRQ(ierr);
5346bf464f9SBarry Smith     ierr = MatDestroy(&B);CHKERRQ(ierr);
53508917f38SBarry Smith     ierr = PetscViewerBinaryGetInfoPointer(viewer,&file);CHKERRQ(ierr);
53608917f38SBarry Smith     if (file) {
53708917f38SBarry Smith       fprintf(file,"-matload_block_size %d\n",(int)A->rmap->bs);
53808917f38SBarry Smith     }
53949b5e25fSSatish Balay   }
54049b5e25fSSatish Balay   PetscFunctionReturn(0);
54149b5e25fSSatish Balay }
54249b5e25fSSatish Balay 
54349b5e25fSSatish Balay 
5444a2ae208SSatish Balay #undef __FUNCT__
5454a2ae208SSatish Balay #define __FUNCT__ "MatGetValues_SeqSBAIJ"
54613f74950SBarry Smith PetscErrorCode MatGetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],PetscScalar v[])
54749b5e25fSSatish Balay {
548045c9aa0SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
54913f74950SBarry Smith   PetscInt     *rp,k,low,high,t,row,nrow,i,col,l,*aj = a->j;
55013f74950SBarry Smith   PetscInt     *ai = a->i,*ailen = a->ilen;
551d0f46423SBarry Smith   PetscInt     brow,bcol,ridx,cidx,bs=A->rmap->bs,bs2=a->bs2;
55297e567efSBarry Smith   MatScalar    *ap,*aa = a->a;
55349b5e25fSSatish Balay 
55449b5e25fSSatish Balay   PetscFunctionBegin;
55549b5e25fSSatish Balay   for (k=0; k<m; k++) { /* loop over rows */
55649b5e25fSSatish Balay     row = im[k]; brow = row/bs;
557e32f2f54SBarry Smith     if (row < 0) {v += n; continue;} /* SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative row: %D",row); */
558e32f2f54SBarry 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);
55949b5e25fSSatish Balay     rp   = aj + ai[brow]; ap = aa + bs2*ai[brow];
56049b5e25fSSatish Balay     nrow = ailen[brow];
56149b5e25fSSatish Balay     for (l=0; l<n; l++) { /* loop over columns */
562e32f2f54SBarry Smith       if (in[l] < 0) {v++; continue;} /* SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column: %D",in[l]); */
563e32f2f54SBarry 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);
56449b5e25fSSatish Balay       col  = in[l];
56549b5e25fSSatish Balay       bcol = col/bs;
56649b5e25fSSatish Balay       cidx = col%bs;
56749b5e25fSSatish Balay       ridx = row%bs;
56849b5e25fSSatish Balay       high = nrow;
56949b5e25fSSatish Balay       low  = 0; /* assume unsorted */
57049b5e25fSSatish Balay       while (high-low > 5) {
57149b5e25fSSatish Balay         t = (low+high)/2;
57249b5e25fSSatish Balay         if (rp[t] > bcol) high = t;
57349b5e25fSSatish Balay         else              low  = t;
57449b5e25fSSatish Balay       }
57549b5e25fSSatish Balay       for (i=low; i<high; i++) {
57649b5e25fSSatish Balay         if (rp[i] > bcol) break;
57749b5e25fSSatish Balay         if (rp[i] == bcol) {
57849b5e25fSSatish Balay           *v++ = ap[bs2*i+bs*cidx+ridx];
57949b5e25fSSatish Balay           goto finished;
58049b5e25fSSatish Balay         }
58149b5e25fSSatish Balay       }
58297e567efSBarry Smith       *v++ = 0.0;
58349b5e25fSSatish Balay finished:;
58449b5e25fSSatish Balay     }
58549b5e25fSSatish Balay   }
58649b5e25fSSatish Balay   PetscFunctionReturn(0);
58749b5e25fSSatish Balay }
58849b5e25fSSatish Balay 
58949b5e25fSSatish Balay 
5904a2ae208SSatish Balay #undef __FUNCT__
5914a2ae208SSatish Balay #define __FUNCT__ "MatSetValuesBlocked_SeqSBAIJ"
59213f74950SBarry Smith PetscErrorCode MatSetValuesBlocked_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is)
59349b5e25fSSatish Balay {
5940880e062SHong Zhang   Mat_SeqSBAIJ      *a = (Mat_SeqSBAIJ*)A->data;
5956849ba73SBarry Smith   PetscErrorCode    ierr;
596e2ee6c50SBarry Smith   PetscInt          *rp,k,low,high,t,ii,jj,row,nrow,i,col,l,rmax,N,lastcol = -1;
59713f74950SBarry Smith   PetscInt          *imax      =a->imax,*ai=a->i,*ailen=a->ilen;
598d0f46423SBarry Smith   PetscInt          *aj        =a->j,nonew=a->nonew,bs2=a->bs2,bs=A->rmap->bs,stepval;
599ace3abfcSBarry Smith   PetscBool         roworiented=a->roworiented;
600dd6ea824SBarry Smith   const PetscScalar *value     = v;
601f15d580aSBarry Smith   MatScalar         *ap,*aa = a->a,*bap;
6020880e062SHong Zhang 
60349b5e25fSSatish Balay   PetscFunctionBegin;
60426fbe8dcSKarl Rupp   if (roworiented) stepval = (n-1)*bs;
60526fbe8dcSKarl Rupp   else stepval = (m-1)*bs;
60626fbe8dcSKarl Rupp 
6070880e062SHong Zhang   for (k=0; k<m; k++) { /* loop over added rows */
6080880e062SHong Zhang     row = im[k];
6090880e062SHong Zhang     if (row < 0) continue;
6102515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
611e32f2f54SBarry Smith     if (row >= a->mbs) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,a->mbs-1);
6120880e062SHong Zhang #endif
6130880e062SHong Zhang     rp   = aj + ai[row];
6140880e062SHong Zhang     ap   = aa + bs2*ai[row];
6150880e062SHong Zhang     rmax = imax[row];
6160880e062SHong Zhang     nrow = ailen[row];
6170880e062SHong Zhang     low  = 0;
618818f2c47SBarry Smith     high = nrow;
6190880e062SHong Zhang     for (l=0; l<n; l++) { /* loop over added columns */
6200880e062SHong Zhang       if (in[l] < 0) continue;
6210880e062SHong Zhang       col = in[l];
6222515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
623e32f2f54SBarry Smith       if (col >= a->nbs) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",col,a->nbs-1);
624b1823623SSatish Balay #endif
625b98bf0e1SJed Brown       if (col < row) {
62626fbe8dcSKarl Rupp         if (a->ignore_ltriangular) continue; /* ignore lower triangular block */
62726fbe8dcSKarl 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)");
628b98bf0e1SJed Brown       }
62926fbe8dcSKarl Rupp       if (roworiented) value = v + k*(stepval+bs)*bs + l*bs;
63026fbe8dcSKarl Rupp       else value = v + l*(stepval+bs)*bs + k*bs;
63126fbe8dcSKarl Rupp 
63226fbe8dcSKarl Rupp       if (col <= lastcol) low = 0;
63326fbe8dcSKarl Rupp       else high = nrow;
63426fbe8dcSKarl Rupp 
635e2ee6c50SBarry Smith       lastcol = col;
6360880e062SHong Zhang       while (high-low > 7) {
6370880e062SHong Zhang         t = (low+high)/2;
6380880e062SHong Zhang         if (rp[t] > col) high = t;
6390880e062SHong Zhang         else             low  = t;
6400880e062SHong Zhang       }
6410880e062SHong Zhang       for (i=low; i<high; i++) {
6420880e062SHong Zhang         if (rp[i] > col) break;
6430880e062SHong Zhang         if (rp[i] == col) {
6440880e062SHong Zhang           bap = ap +  bs2*i;
6450880e062SHong Zhang           if (roworiented) {
6460880e062SHong Zhang             if (is == ADD_VALUES) {
6470880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
6480880e062SHong Zhang                 for (jj=ii; jj<bs2; jj+=bs) {
6490880e062SHong Zhang                   bap[jj] += *value++;
6500880e062SHong Zhang                 }
6510880e062SHong Zhang               }
6520880e062SHong Zhang             } else {
6530880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
6540880e062SHong Zhang                 for (jj=ii; jj<bs2; jj+=bs) {
6550880e062SHong Zhang                   bap[jj] = *value++;
6560880e062SHong Zhang                 }
6570880e062SHong Zhang                }
6580880e062SHong Zhang             }
6590880e062SHong Zhang           } else {
6600880e062SHong Zhang             if (is == ADD_VALUES) {
6610880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
6620880e062SHong Zhang                 for (jj=0; jj<bs; jj++) {
6630880e062SHong Zhang                   *bap++ += *value++;
6640880e062SHong Zhang                 }
6650880e062SHong Zhang               }
6660880e062SHong Zhang             } else {
6670880e062SHong Zhang               for (ii=0; ii<bs; ii++,value+=stepval) {
6680880e062SHong Zhang                 for (jj=0; jj<bs; jj++) {
6690880e062SHong Zhang                   *bap++  = *value++;
6700880e062SHong Zhang                 }
6710880e062SHong Zhang               }
6720880e062SHong Zhang             }
6730880e062SHong Zhang           }
6740880e062SHong Zhang           goto noinsert2;
6750880e062SHong Zhang         }
6760880e062SHong Zhang       }
6770880e062SHong Zhang       if (nonew == 1) goto noinsert2;
678e32f2f54SBarry Smith       if (nonew == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col);
679fef13f97SBarry Smith       MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,row,col,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar);
680c03d1d03SSatish Balay       N = nrow++ - 1; high++;
6810880e062SHong Zhang       /* shift up all the later entries in this row */
6820880e062SHong Zhang       for (ii=N; ii>=i; ii--) {
6830880e062SHong Zhang         rp[ii+1] = rp[ii];
6840880e062SHong Zhang         ierr     = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr);
6850880e062SHong Zhang       }
6860880e062SHong Zhang       if (N >= i) {
6870880e062SHong Zhang         ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr);
6880880e062SHong Zhang       }
6890880e062SHong Zhang       rp[i] = col;
6900880e062SHong Zhang       bap   = ap +  bs2*i;
6910880e062SHong Zhang       if (roworiented) {
6920880e062SHong Zhang         for (ii=0; ii<bs; ii++,value+=stepval) {
6930880e062SHong Zhang           for (jj=ii; jj<bs2; jj+=bs) {
6940880e062SHong Zhang             bap[jj] = *value++;
6950880e062SHong Zhang           }
6960880e062SHong Zhang         }
6970880e062SHong Zhang       } else {
6980880e062SHong Zhang         for (ii=0; ii<bs; ii++,value+=stepval) {
6990880e062SHong Zhang           for (jj=0; jj<bs; jj++) {
7000880e062SHong Zhang             *bap++ = *value++;
7010880e062SHong Zhang           }
7020880e062SHong Zhang         }
7030880e062SHong Zhang        }
7040880e062SHong Zhang     noinsert2:;
7050880e062SHong Zhang       low = i;
7060880e062SHong Zhang     }
7070880e062SHong Zhang     ailen[row] = nrow;
7080880e062SHong Zhang   }
7090880e062SHong Zhang   PetscFunctionReturn(0);
71049b5e25fSSatish Balay }
71149b5e25fSSatish Balay 
71264831d72SBarry Smith /*
71364831d72SBarry Smith     This is not yet used
71464831d72SBarry Smith */
7154a2ae208SSatish Balay #undef __FUNCT__
7164108e4d5SBarry Smith #define __FUNCT__ "MatAssemblyEnd_SeqSBAIJ_SeqAIJ_Inode"
7174108e4d5SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ_SeqAIJ_Inode(Mat A)
7180def2e27SBarry Smith {
7190def2e27SBarry Smith   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
7200def2e27SBarry Smith   PetscErrorCode ierr;
7210def2e27SBarry Smith   const PetscInt *ai = a->i, *aj = a->j,*cols;
7220def2e27SBarry Smith   PetscInt       i   = 0,j,blk_size,m = A->rmap->n,node_count = 0,nzx,nzy,*ns,row,nz,cnt,cnt2,*counts;
723ace3abfcSBarry Smith   PetscBool      flag;
7240def2e27SBarry Smith 
7250def2e27SBarry Smith   PetscFunctionBegin;
726785e854fSJed Brown   ierr = PetscMalloc1(m,&ns);CHKERRQ(ierr);
7270def2e27SBarry Smith   while (i < m) {
7280def2e27SBarry Smith     nzx = ai[i+1] - ai[i];       /* Number of nonzeros */
7290def2e27SBarry Smith     /* Limits the number of elements in a node to 'a->inode.limit' */
7300def2e27SBarry Smith     for (j=i+1,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) {
7310def2e27SBarry Smith       nzy = ai[j+1] - ai[j];
7320def2e27SBarry Smith       if (nzy != (nzx - j + i)) break;
7330def2e27SBarry Smith       ierr = PetscMemcmp(aj + ai[i] + j - i,aj + ai[j],nzy*sizeof(PetscInt),&flag);CHKERRQ(ierr);
7340def2e27SBarry Smith       if (!flag) break;
7350def2e27SBarry Smith     }
7360def2e27SBarry Smith     ns[node_count++] = blk_size;
73726fbe8dcSKarl Rupp 
7380def2e27SBarry Smith     i = j;
7390def2e27SBarry Smith   }
7400def2e27SBarry Smith   if (!a->inode.size && m && node_count > .9*m) {
7410def2e27SBarry Smith     ierr = PetscFree(ns);CHKERRQ(ierr);
7420def2e27SBarry Smith     ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr);
7430def2e27SBarry Smith   } else {
7440def2e27SBarry Smith     a->inode.node_count = node_count;
74526fbe8dcSKarl Rupp 
746785e854fSJed Brown     ierr = PetscMalloc1(node_count,&a->inode.size);CHKERRQ(ierr);
7473bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)A,node_count*sizeof(PetscInt));CHKERRQ(ierr);
74822d28d08SBarry Smith     ierr = PetscMemcpy(a->inode.size,ns,node_count*sizeof(PetscInt));CHKERRQ(ierr);
7490def2e27SBarry Smith     ierr = PetscFree(ns);CHKERRQ(ierr);
7500def2e27SBarry Smith     ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr);
7510def2e27SBarry Smith 
7520def2e27SBarry Smith     /* count collections of adjacent columns in each inode */
7530def2e27SBarry Smith     row = 0;
7540def2e27SBarry Smith     cnt = 0;
7550def2e27SBarry Smith     for (i=0; i<node_count; i++) {
7560def2e27SBarry Smith       cols = aj + ai[row] + a->inode.size[i];
7570def2e27SBarry Smith       nz   = ai[row+1] - ai[row] - a->inode.size[i];
7580def2e27SBarry Smith       for (j=1; j<nz; j++) {
75926fbe8dcSKarl Rupp         if (cols[j] != cols[j-1]+1) cnt++;
7600def2e27SBarry Smith       }
7610def2e27SBarry Smith       cnt++;
7620def2e27SBarry Smith       row += a->inode.size[i];
7630def2e27SBarry Smith     }
764785e854fSJed Brown     ierr = PetscMalloc1(2*cnt,&counts);CHKERRQ(ierr);
7650def2e27SBarry Smith     cnt  = 0;
7660def2e27SBarry Smith     row  = 0;
7670def2e27SBarry Smith     for (i=0; i<node_count; i++) {
7680def2e27SBarry Smith       cols = aj + ai[row] + a->inode.size[i];
7690def2e27SBarry Smith       counts[2*cnt] = cols[0];
7700def2e27SBarry Smith       nz   = ai[row+1] - ai[row] - a->inode.size[i];
7710def2e27SBarry Smith       cnt2 = 1;
7720def2e27SBarry Smith       for (j=1; j<nz; j++) {
7730def2e27SBarry Smith         if (cols[j] != cols[j-1]+1) {
7740def2e27SBarry Smith           counts[2*(cnt++)+1] = cnt2;
7750def2e27SBarry Smith           counts[2*cnt]       = cols[j];
7760def2e27SBarry Smith           cnt2 = 1;
7770def2e27SBarry Smith         } else cnt2++;
7780def2e27SBarry Smith       }
7790def2e27SBarry Smith       counts[2*(cnt++)+1] = cnt2;
7800def2e27SBarry Smith       row += a->inode.size[i];
7810def2e27SBarry Smith     }
78222d28d08SBarry Smith     ierr = PetscIntView(2*cnt,counts,0);CHKERRQ(ierr);
7830def2e27SBarry Smith   }
78438702af4SBarry Smith   PetscFunctionReturn(0);
78538702af4SBarry Smith }
78638702af4SBarry Smith 
78738702af4SBarry Smith #undef __FUNCT__
7884a2ae208SSatish Balay #define __FUNCT__ "MatAssemblyEnd_SeqSBAIJ"
789dfbe8321SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ(Mat A,MatAssemblyType mode)
79049b5e25fSSatish Balay {
79149b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
7926849ba73SBarry Smith   PetscErrorCode ierr;
79313f74950SBarry Smith   PetscInt       fshift = 0,i,j,*ai = a->i,*aj = a->j,*imax = a->imax;
794d0f46423SBarry Smith   PetscInt       m      = A->rmap->N,*ip,N,*ailen = a->ilen;
79513f74950SBarry Smith   PetscInt       mbs    = a->mbs,bs2 = a->bs2,rmax = 0;
79649b5e25fSSatish Balay   MatScalar      *aa    = a->a,*ap;
79749b5e25fSSatish Balay 
79849b5e25fSSatish Balay   PetscFunctionBegin;
79949b5e25fSSatish Balay   if (mode == MAT_FLUSH_ASSEMBLY) PetscFunctionReturn(0);
80049b5e25fSSatish Balay 
80149b5e25fSSatish Balay   if (m) rmax = ailen[0];
80249b5e25fSSatish Balay   for (i=1; i<mbs; i++) {
80349b5e25fSSatish Balay     /* move each row back by the amount of empty slots (fshift) before it*/
80449b5e25fSSatish Balay     fshift += imax[i-1] - ailen[i-1];
80549b5e25fSSatish Balay     rmax    = PetscMax(rmax,ailen[i]);
80649b5e25fSSatish Balay     if (fshift) {
80749b5e25fSSatish Balay       ip = aj + ai[i]; ap = aa + bs2*ai[i];
80849b5e25fSSatish Balay       N  = ailen[i];
80949b5e25fSSatish Balay       for (j=0; j<N; j++) {
81049b5e25fSSatish Balay         ip[j-fshift] = ip[j];
81149b5e25fSSatish Balay         ierr         = PetscMemcpy(ap+(j-fshift)*bs2,ap+j*bs2,bs2*sizeof(MatScalar));CHKERRQ(ierr);
81249b5e25fSSatish Balay       }
81349b5e25fSSatish Balay     }
81449b5e25fSSatish Balay     ai[i] = ai[i-1] + ailen[i-1];
81549b5e25fSSatish Balay   }
81649b5e25fSSatish Balay   if (mbs) {
81749b5e25fSSatish Balay     fshift += imax[mbs-1] - ailen[mbs-1];
81849b5e25fSSatish Balay     ai[mbs] = ai[mbs-1] + ailen[mbs-1];
81949b5e25fSSatish Balay   }
82049b5e25fSSatish Balay   /* reset ilen and imax for each row */
82149b5e25fSSatish Balay   for (i=0; i<mbs; i++) {
82249b5e25fSSatish Balay     ailen[i] = imax[i] = ai[i+1] - ai[i];
82349b5e25fSSatish Balay   }
8246c6c5352SBarry Smith   a->nz = ai[mbs];
82549b5e25fSSatish Balay 
826b424e231SHong Zhang   /* diagonals may have moved, reset it */
827b424e231SHong Zhang   if (a->diag) {
8282ed38d0bSJed Brown     ierr = PetscMemcpy(a->diag,ai,mbs*sizeof(PetscInt));CHKERRQ(ierr);
82949b5e25fSSatish Balay   }
83026fbe8dcSKarl 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);
83126fbe8dcSKarl Rupp 
832d0f46423SBarry 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);
833ae15b995SBarry Smith   ierr = PetscInfo1(A,"Number of mallocs during MatSetValues is %D\n",a->reallocs);CHKERRQ(ierr);
834ae15b995SBarry Smith   ierr = PetscInfo1(A,"Most nonzeros blocks in any row is %D\n",rmax);CHKERRQ(ierr);
83526fbe8dcSKarl Rupp 
8368e58a170SBarry Smith   A->info.mallocs    += a->reallocs;
83749b5e25fSSatish Balay   a->reallocs         = 0;
83849b5e25fSSatish Balay   A->info.nz_unneeded = (PetscReal)fshift*bs2;
839061b2667SBarry Smith   a->idiagvalid       = PETSC_FALSE;
84038702af4SBarry Smith 
84138702af4SBarry Smith   if (A->cmap->n < 65536 && A->cmap->bs == 1) {
84244e1c64aSLisandro Dalcin     if (a->jshort && a->free_jshort) {
84317803ae8SHong Zhang       /* when matrix data structure is changed, previous jshort must be replaced */
84417803ae8SHong Zhang       ierr = PetscFree(a->jshort);CHKERRQ(ierr);
84517803ae8SHong Zhang     }
846785e854fSJed Brown     ierr = PetscMalloc1(a->i[A->rmap->n],&a->jshort);CHKERRQ(ierr);
8473bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)A,a->i[A->rmap->n]*sizeof(unsigned short));CHKERRQ(ierr);
84838702af4SBarry Smith     for (i=0; i<a->i[A->rmap->n]; i++) a->jshort[i] = a->j[i];
84938702af4SBarry Smith     A->ops->mult   = MatMult_SeqSBAIJ_1_ushort;
85041f059aeSBarry Smith     A->ops->sor    = MatSOR_SeqSBAIJ_ushort;
8514da8f245SBarry Smith     a->free_jshort = PETSC_TRUE;
85238702af4SBarry Smith   }
85349b5e25fSSatish Balay   PetscFunctionReturn(0);
85449b5e25fSSatish Balay }
85549b5e25fSSatish Balay 
85649b5e25fSSatish Balay /*
85749b5e25fSSatish Balay    This function returns an array of flags which indicate the locations of contiguous
85849b5e25fSSatish Balay    blocks that should be zeroed. for eg: if bs = 3  and is = [0,1,2,3,5,6,7,8,9]
85949b5e25fSSatish Balay    then the resulting sizes = [3,1,1,3,1] correspondig to sets [(0,1,2),(3),(5),(6,7,8),(9)]
86049b5e25fSSatish Balay    Assume: sizes should be long enough to hold all the values.
86149b5e25fSSatish Balay */
8624a2ae208SSatish Balay #undef __FUNCT__
8634a2ae208SSatish Balay #define __FUNCT__ "MatZeroRows_SeqSBAIJ_Check_Blocks"
86413f74950SBarry Smith PetscErrorCode MatZeroRows_SeqSBAIJ_Check_Blocks(PetscInt idx[],PetscInt n,PetscInt bs,PetscInt sizes[], PetscInt *bs_max)
86549b5e25fSSatish Balay {
86613f74950SBarry Smith   PetscInt  i,j,k,row;
867ace3abfcSBarry Smith   PetscBool flg;
86849b5e25fSSatish Balay 
86949b5e25fSSatish Balay   PetscFunctionBegin;
87049b5e25fSSatish Balay   for (i=0,j=0; i<n; j++) {
87149b5e25fSSatish Balay     row = idx[i];
87249b5e25fSSatish Balay     if (row%bs!=0) { /* Not the begining of a block */
87349b5e25fSSatish Balay       sizes[j] = 1;
87449b5e25fSSatish Balay       i++;
87549b5e25fSSatish Balay     } else if (i+bs > n) { /* Beginning of a block, but complete block doesn't exist (at idx end) */
87649b5e25fSSatish Balay       sizes[j] = 1;         /* Also makes sure atleast 'bs' values exist for next else */
87749b5e25fSSatish Balay       i++;
87849b5e25fSSatish Balay     } else { /* Begining of the block, so check if the complete block exists */
87949b5e25fSSatish Balay       flg = PETSC_TRUE;
88049b5e25fSSatish Balay       for (k=1; k<bs; k++) {
88149b5e25fSSatish Balay         if (row+k != idx[i+k]) { /* break in the block */
88249b5e25fSSatish Balay           flg = PETSC_FALSE;
88349b5e25fSSatish Balay           break;
88449b5e25fSSatish Balay         }
88549b5e25fSSatish Balay       }
886abc0a331SBarry Smith       if (flg) { /* No break in the bs */
88749b5e25fSSatish Balay         sizes[j] = bs;
88849b5e25fSSatish Balay         i       += bs;
88949b5e25fSSatish Balay       } else {
89049b5e25fSSatish Balay         sizes[j] = 1;
89149b5e25fSSatish Balay         i++;
89249b5e25fSSatish Balay       }
89349b5e25fSSatish Balay     }
89449b5e25fSSatish Balay   }
89549b5e25fSSatish Balay   *bs_max = j;
89649b5e25fSSatish Balay   PetscFunctionReturn(0);
89749b5e25fSSatish Balay }
89849b5e25fSSatish Balay 
89949b5e25fSSatish Balay 
90049b5e25fSSatish Balay /* Only add/insert a(i,j) with i<=j (blocks).
90149b5e25fSSatish Balay    Any a(i,j) with i>j input by user is ingored.
90249b5e25fSSatish Balay */
90349b5e25fSSatish Balay 
9044a2ae208SSatish Balay #undef __FUNCT__
9054a2ae208SSatish Balay #define __FUNCT__ "MatSetValues_SeqSBAIJ"
90613f74950SBarry Smith PetscErrorCode MatSetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is)
90749b5e25fSSatish Balay {
90849b5e25fSSatish Balay   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)A->data;
9096849ba73SBarry Smith   PetscErrorCode ierr;
910e2ee6c50SBarry Smith   PetscInt       *rp,k,low,high,t,ii,row,nrow,i,col,l,rmax,N,lastcol = -1;
91113f74950SBarry Smith   PetscInt       *imax=a->imax,*ai=a->i,*ailen=a->ilen,roworiented=a->roworiented;
912d0f46423SBarry Smith   PetscInt       *aj  =a->j,nonew=a->nonew,bs=A->rmap->bs,brow,bcol;
91313f74950SBarry Smith   PetscInt       ridx,cidx,bs2=a->bs2;
91449b5e25fSSatish Balay   MatScalar      *ap,value,*aa=a->a,*bap;
91549b5e25fSSatish Balay 
91649b5e25fSSatish Balay   PetscFunctionBegin;
91749b5e25fSSatish Balay   for (k=0; k<m; k++) { /* loop over added rows */
91849b5e25fSSatish Balay     row  = im[k];       /* row number */
91949b5e25fSSatish Balay     brow = row/bs;      /* block row number */
92049b5e25fSSatish Balay     if (row < 0) continue;
9212515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
922e32f2f54SBarry 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);
92349b5e25fSSatish Balay #endif
92449b5e25fSSatish Balay     rp   = aj + ai[brow]; /*ptr to beginning of column value of the row block*/
92549b5e25fSSatish Balay     ap   = aa + bs2*ai[brow]; /*ptr to beginning of element value of the row block*/
92649b5e25fSSatish Balay     rmax = imax[brow];  /* maximum space allocated for this row */
92749b5e25fSSatish Balay     nrow = ailen[brow]; /* actual length of this row */
92849b5e25fSSatish Balay     low  = 0;
92949b5e25fSSatish Balay 
93049b5e25fSSatish Balay     for (l=0; l<n; l++) { /* loop over added columns */
93149b5e25fSSatish Balay       if (in[l] < 0) continue;
9322515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
933e32f2f54SBarry 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);
93449b5e25fSSatish Balay #endif
93549b5e25fSSatish Balay       col  = in[l];
93649b5e25fSSatish Balay       bcol = col/bs;              /* block col number */
93749b5e25fSSatish Balay 
938941593c8SHong Zhang       if (brow > bcol) {
93926fbe8dcSKarl Rupp         if (a->ignore_ltriangular) continue; /* ignore lower triangular values */
94026fbe8dcSKarl 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)");
941941593c8SHong Zhang       }
942f4989cb3SHong Zhang 
94349b5e25fSSatish Balay       ridx = row % bs; cidx = col % bs; /*row and col index inside the block */
9448549e402SHong Zhang       if ((brow==bcol && ridx<=cidx) || (brow<bcol)) {
94549b5e25fSSatish Balay         /* element value a(k,l) */
94626fbe8dcSKarl Rupp         if (roworiented) value = v[l + k*n];
94726fbe8dcSKarl Rupp         else value = v[k + l*m];
94849b5e25fSSatish Balay 
94949b5e25fSSatish Balay         /* move pointer bap to a(k,l) quickly and add/insert value */
95026fbe8dcSKarl Rupp         if (col <= lastcol) low = 0;
95126fbe8dcSKarl Rupp         high = nrow;
952e2ee6c50SBarry Smith         lastcol = col;
95349b5e25fSSatish Balay         while (high-low > 7) {
95449b5e25fSSatish Balay           t = (low+high)/2;
95549b5e25fSSatish Balay           if (rp[t] > bcol) high = t;
95649b5e25fSSatish Balay           else              low  = t;
95749b5e25fSSatish Balay         }
95849b5e25fSSatish Balay         for (i=low; i<high; i++) {
95949b5e25fSSatish Balay           if (rp[i] > bcol) break;
96049b5e25fSSatish Balay           if (rp[i] == bcol) {
96149b5e25fSSatish Balay             bap = ap +  bs2*i + bs*cidx + ridx;
96249b5e25fSSatish Balay             if (is == ADD_VALUES) *bap += value;
96349b5e25fSSatish Balay             else                  *bap  = value;
9648549e402SHong Zhang             /* for diag block, add/insert its symmetric element a(cidx,ridx) */
9658549e402SHong Zhang             if (brow == bcol && ridx < cidx) {
9668549e402SHong Zhang               bap = ap +  bs2*i + bs*ridx + cidx;
9678549e402SHong Zhang               if (is == ADD_VALUES) *bap += value;
9688549e402SHong Zhang               else                  *bap  = value;
9698549e402SHong Zhang             }
97049b5e25fSSatish Balay             goto noinsert1;
97149b5e25fSSatish Balay           }
97249b5e25fSSatish Balay         }
97349b5e25fSSatish Balay 
97449b5e25fSSatish Balay         if (nonew == 1) goto noinsert1;
975e32f2f54SBarry Smith         if (nonew == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col);
976fef13f97SBarry Smith         MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,brow,bcol,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar);
97749b5e25fSSatish Balay 
978c03d1d03SSatish Balay         N = nrow++ - 1; high++;
97949b5e25fSSatish Balay         /* shift up all the later entries in this row */
98049b5e25fSSatish Balay         for (ii=N; ii>=i; ii--) {
98149b5e25fSSatish Balay           rp[ii+1] = rp[ii];
98249b5e25fSSatish Balay           ierr     = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr);
98349b5e25fSSatish Balay         }
98449b5e25fSSatish Balay         if (N>=i) {
98549b5e25fSSatish Balay           ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr);
98649b5e25fSSatish Balay         }
98749b5e25fSSatish Balay         rp[i]                      = bcol;
98849b5e25fSSatish Balay         ap[bs2*i + bs*cidx + ridx] = value;
989e56f5c9eSBarry Smith         A->nonzerostate++;
99049b5e25fSSatish Balay noinsert1:;
99149b5e25fSSatish Balay         low = i;
9928549e402SHong Zhang       }
99349b5e25fSSatish Balay     }   /* end of loop over added columns */
99449b5e25fSSatish Balay     ailen[brow] = nrow;
99549b5e25fSSatish Balay   }   /* end of loop over added rows */
99649b5e25fSSatish Balay   PetscFunctionReturn(0);
99749b5e25fSSatish Balay }
99849b5e25fSSatish Balay 
9994a2ae208SSatish Balay #undef __FUNCT__
10004d101231SSatish Balay #define __FUNCT__ "MatICCFactor_SeqSBAIJ"
10010481f469SBarry Smith PetscErrorCode MatICCFactor_SeqSBAIJ(Mat inA,IS row,const MatFactorInfo *info)
100249b5e25fSSatish Balay {
10034ccecd49SHong Zhang   Mat_SeqSBAIJ   *a = (Mat_SeqSBAIJ*)inA->data;
100449b5e25fSSatish Balay   Mat            outA;
1005dfbe8321SBarry Smith   PetscErrorCode ierr;
1006ace3abfcSBarry Smith   PetscBool      row_identity;
100749b5e25fSSatish Balay 
100849b5e25fSSatish Balay   PetscFunctionBegin;
1009e32f2f54SBarry Smith   if (info->levels != 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only levels=0 is supported for in-place icc");
1010c84f5b01SHong Zhang   ierr = ISIdentity(row,&row_identity);CHKERRQ(ierr);
1011e32f2f54SBarry Smith   if (!row_identity) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix reordering is not supported");
1012e32f2f54SBarry 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()! */
1013c84f5b01SHong Zhang 
101449b5e25fSSatish Balay   outA            = inA;
1015d5f3da31SBarry Smith   inA->factortype = MAT_FACTOR_ICC;
101649b5e25fSSatish Balay 
10171a3463dfSHong Zhang   ierr = MatMarkDiagonal_SeqSBAIJ(inA);CHKERRQ(ierr);
1018d595f711SHong Zhang   ierr = MatSeqSBAIJSetNumericFactorization_inplace(inA,row_identity);CHKERRQ(ierr);
101949b5e25fSSatish Balay 
1020c3122656SLisandro Dalcin   ierr   = PetscObjectReference((PetscObject)row);CHKERRQ(ierr);
10216bf464f9SBarry Smith   ierr   = ISDestroy(&a->row);CHKERRQ(ierr);
1022c84f5b01SHong Zhang   a->row = row;
1023c3122656SLisandro Dalcin   ierr   = PetscObjectReference((PetscObject)row);CHKERRQ(ierr);
10246bf464f9SBarry Smith   ierr   = ISDestroy(&a->col);CHKERRQ(ierr);
1025c84f5b01SHong Zhang   a->col = row;
1026c84f5b01SHong Zhang 
1027c84f5b01SHong Zhang   /* Create the invert permutation so that it can be used in MatCholeskyFactorNumeric() */
1028c84f5b01SHong Zhang   if (a->icol) {ierr = ISInvertPermutation(row,PETSC_DECIDE, &a->icol);CHKERRQ(ierr);}
10293bb1ff40SBarry Smith   ierr = PetscLogObjectParent((PetscObject)inA,(PetscObject)a->icol);CHKERRQ(ierr);
103049b5e25fSSatish Balay 
103149b5e25fSSatish Balay   if (!a->solve_work) {
1032785e854fSJed Brown     ierr = PetscMalloc1((inA->rmap->N+inA->rmap->bs),&a->solve_work);CHKERRQ(ierr);
10333bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)inA,(inA->rmap->N+inA->rmap->bs)*sizeof(PetscScalar));CHKERRQ(ierr);
103449b5e25fSSatish Balay   }
103549b5e25fSSatish Balay 
1036719d5645SBarry Smith   ierr = MatCholeskyFactorNumeric(outA,inA,info);CHKERRQ(ierr);
103749b5e25fSSatish Balay   PetscFunctionReturn(0);
103849b5e25fSSatish Balay }
1039950f1e5bSHong Zhang 
10404a2ae208SSatish Balay #undef __FUNCT__
10414a2ae208SSatish Balay #define __FUNCT__ "MatSeqSBAIJSetColumnIndices_SeqSBAIJ"
10427087cfbeSBarry Smith PetscErrorCode  MatSeqSBAIJSetColumnIndices_SeqSBAIJ(Mat mat,PetscInt *indices)
104349b5e25fSSatish Balay {
1044045c9aa0SHong Zhang   Mat_SeqSBAIJ   *baij = (Mat_SeqSBAIJ*)mat->data;
104513f74950SBarry Smith   PetscInt       i,nz,n;
10467827cd58SJed Brown   PetscErrorCode ierr;
104749b5e25fSSatish Balay 
104849b5e25fSSatish Balay   PetscFunctionBegin;
10496c6c5352SBarry Smith   nz = baij->maxnz;
1050d0f46423SBarry Smith   n  = mat->cmap->n;
105126fbe8dcSKarl Rupp   for (i=0; i<nz; i++) baij->j[i] = indices[i];
105226fbe8dcSKarl Rupp 
10536c6c5352SBarry Smith   baij->nz = nz;
105426fbe8dcSKarl Rupp   for (i=0; i<n; i++) baij->ilen[i] = baij->imax[i];
105526fbe8dcSKarl Rupp 
10567827cd58SJed Brown   ierr = MatSetOption(mat,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
105749b5e25fSSatish Balay   PetscFunctionReturn(0);
105849b5e25fSSatish Balay }
105949b5e25fSSatish Balay 
10604a2ae208SSatish Balay #undef __FUNCT__
10614a2ae208SSatish Balay #define __FUNCT__ "MatSeqSBAIJSetColumnIndices"
106249b5e25fSSatish Balay /*@
106319585528SSatish Balay   MatSeqSBAIJSetColumnIndices - Set the column indices for all the rows
106449b5e25fSSatish Balay   in the matrix.
106549b5e25fSSatish Balay 
106649b5e25fSSatish Balay   Input Parameters:
106719585528SSatish Balay   +  mat     - the SeqSBAIJ matrix
106849b5e25fSSatish Balay   -  indices - the column indices
106949b5e25fSSatish Balay 
107049b5e25fSSatish Balay   Level: advanced
107149b5e25fSSatish Balay 
107249b5e25fSSatish Balay   Notes:
107349b5e25fSSatish Balay   This can be called if you have precomputed the nonzero structure of the
107449b5e25fSSatish Balay   matrix and want to provide it to the matrix object to improve the performance
107549b5e25fSSatish Balay   of the MatSetValues() operation.
107649b5e25fSSatish Balay 
107749b5e25fSSatish Balay   You MUST have set the correct numbers of nonzeros per row in the call to
1078d1be2dadSMatthew Knepley   MatCreateSeqSBAIJ(), and the columns indices MUST be sorted.
107949b5e25fSSatish Balay 
1080ab9f2c04SSatish Balay   MUST be called before any calls to MatSetValues()
108149b5e25fSSatish Balay 
1082ab9f2c04SSatish Balay   .seealso: MatCreateSeqSBAIJ
108349b5e25fSSatish Balay @*/
10847087cfbeSBarry Smith PetscErrorCode  MatSeqSBAIJSetColumnIndices(Mat mat,PetscInt *indices)
108549b5e25fSSatish Balay {
10864ac538c5SBarry Smith   PetscErrorCode ierr;
108749b5e25fSSatish Balay 
108849b5e25fSSatish Balay   PetscFunctionBegin;
10890700a824SBarry Smith   PetscValidHeaderSpecific(mat,MAT_CLASSID,1);
10904482741eSBarry Smith   PetscValidPointer(indices,2);
10914ac538c5SBarry Smith   ierr = PetscUseMethod(mat,"MatSeqSBAIJSetColumnIndices_C",(Mat,PetscInt*),(mat,indices));CHKERRQ(ierr);
109249b5e25fSSatish Balay   PetscFunctionReturn(0);
109349b5e25fSSatish Balay }
109449b5e25fSSatish Balay 
10954a2ae208SSatish Balay #undef __FUNCT__
10963c896bc6SHong Zhang #define __FUNCT__ "MatCopy_SeqSBAIJ"
10973c896bc6SHong Zhang PetscErrorCode MatCopy_SeqSBAIJ(Mat A,Mat B,MatStructure str)
10983c896bc6SHong Zhang {
10993c896bc6SHong Zhang   PetscErrorCode ierr;
11003c896bc6SHong Zhang 
11013c896bc6SHong Zhang   PetscFunctionBegin;
11023c896bc6SHong Zhang   /* If the two matrices have the same copy implementation, use fast copy. */
11033c896bc6SHong Zhang   if (str == SAME_NONZERO_PATTERN && (A->ops->copy == B->ops->copy)) {
11043c896bc6SHong Zhang     Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
11053c896bc6SHong Zhang     Mat_SeqSBAIJ *b = (Mat_SeqSBAIJ*)B->data;
11063c896bc6SHong Zhang 
1107e7e72b3dSBarry 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");
1108d0f46423SBarry Smith     ierr = PetscMemcpy(b->a,a->a,(a->i[A->rmap->N])*sizeof(PetscScalar));CHKERRQ(ierr);
11093c896bc6SHong Zhang   } else {
1110f5edf698SHong Zhang     ierr = MatGetRowUpperTriangular(A);CHKERRQ(ierr);
11113c896bc6SHong Zhang     ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr);
1112f5edf698SHong Zhang     ierr = MatRestoreRowUpperTriangular(A);CHKERRQ(ierr);
11133c896bc6SHong Zhang   }
11143c896bc6SHong Zhang   PetscFunctionReturn(0);
11153c896bc6SHong Zhang }
11163c896bc6SHong Zhang 
11173c896bc6SHong Zhang #undef __FUNCT__
11184994cf47SJed Brown #define __FUNCT__ "MatSetUp_SeqSBAIJ"
11194994cf47SJed Brown PetscErrorCode MatSetUp_SeqSBAIJ(Mat A)
1120273d9f13SBarry Smith {
1121dfbe8321SBarry Smith   PetscErrorCode ierr;
1122273d9f13SBarry Smith 
1123273d9f13SBarry Smith   PetscFunctionBegin;
1124535b19f3SBarry Smith   ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(A,A->rmap->bs,PETSC_DEFAULT,0);CHKERRQ(ierr);
1125273d9f13SBarry Smith   PetscFunctionReturn(0);
1126273d9f13SBarry Smith }
1127273d9f13SBarry Smith 
1128a6ece127SHong Zhang #undef __FUNCT__
11298c778c55SBarry Smith #define __FUNCT__ "MatSeqSBAIJGetArray_SeqSBAIJ"
11308c778c55SBarry Smith PetscErrorCode MatSeqSBAIJGetArray_SeqSBAIJ(Mat A,PetscScalar *array[])
1131a6ece127SHong Zhang {
1132a6ece127SHong Zhang   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
11335fd66863SKarl Rupp 
1134a6ece127SHong Zhang   PetscFunctionBegin;
1135a6ece127SHong Zhang   *array = a->a;
1136a6ece127SHong Zhang   PetscFunctionReturn(0);
1137a6ece127SHong Zhang }
1138a6ece127SHong Zhang 
1139a6ece127SHong Zhang #undef __FUNCT__
11408c778c55SBarry Smith #define __FUNCT__ "MatSeqSBAIJRestoreArray_SeqSBAIJ"
11418c778c55SBarry Smith PetscErrorCode MatSeqSBAIJRestoreArray_SeqSBAIJ(Mat A,PetscScalar *array[])
1142a6ece127SHong Zhang {
1143a6ece127SHong Zhang   PetscFunctionBegin;
1144a6ece127SHong Zhang   PetscFunctionReturn(0);
1145a6ece127SHong Zhang }
1146a6ece127SHong Zhang 
114742ee4b1aSHong Zhang #undef __FUNCT__
114852768537SHong Zhang #define __FUNCT__ "MatAXPYGetPreallocation_SeqSBAIJ"
114952768537SHong Zhang PetscErrorCode MatAXPYGetPreallocation_SeqSBAIJ(Mat Y,Mat X,PetscInt *nnz)
115052768537SHong Zhang {
1151b264fe52SHong Zhang   PetscInt       bs = Y->rmap->bs,mbs = Y->rmap->N/bs;
115252768537SHong Zhang   Mat_SeqSBAIJ   *x = (Mat_SeqSBAIJ*)X->data;
115352768537SHong Zhang   Mat_SeqSBAIJ   *y = (Mat_SeqSBAIJ*)Y->data;
1154b264fe52SHong Zhang   PetscErrorCode ierr;
115552768537SHong Zhang 
115652768537SHong Zhang   PetscFunctionBegin;
115752768537SHong Zhang   /* Set the number of nonzeros in the new matrix */
1158b264fe52SHong Zhang   ierr = MatAXPYGetPreallocation_SeqX_private(mbs,x->i,x->j,y->i,y->j,nnz);CHKERRQ(ierr);
115952768537SHong Zhang   PetscFunctionReturn(0);
116052768537SHong Zhang }
116152768537SHong Zhang 
116252768537SHong Zhang #undef __FUNCT__
116342ee4b1aSHong Zhang #define __FUNCT__ "MatAXPY_SeqSBAIJ"
1164f4df32b1SMatthew Knepley PetscErrorCode MatAXPY_SeqSBAIJ(Mat Y,PetscScalar a,Mat X,MatStructure str)
116542ee4b1aSHong Zhang {
116642ee4b1aSHong Zhang   Mat_SeqSBAIJ   *x=(Mat_SeqSBAIJ*)X->data, *y=(Mat_SeqSBAIJ*)Y->data;
1167dfbe8321SBarry Smith   PetscErrorCode ierr;
1168*31ce2d13SHong Zhang   PetscInt       bs=Y->rmap->bs,bs2=bs*bs;
1169e838b9e7SJed Brown   PetscBLASInt   one = 1;
117042ee4b1aSHong Zhang 
117142ee4b1aSHong Zhang   PetscFunctionBegin;
117242ee4b1aSHong Zhang   if (str == SAME_NONZERO_PATTERN) {
1173f4df32b1SMatthew Knepley     PetscScalar  alpha = a;
1174c5df96a5SBarry Smith     PetscBLASInt bnz;
1175c5df96a5SBarry Smith     ierr = PetscBLASIntCast(x->nz*bs2,&bnz);CHKERRQ(ierr);
11768b83055fSJed Brown     PetscStackCallBLAS("BLASaxpy",BLASaxpy_(&bnz,&alpha,x->a,&one,y->a,&one));
1177a3fa217bSJose E. Roman     ierr = PetscObjectStateIncrease((PetscObject)Y);CHKERRQ(ierr);
117842ee4b1aSHong Zhang   } else {
117952768537SHong Zhang     Mat      B;
118052768537SHong Zhang     PetscInt *nnz;
118152768537SHong Zhang     if (bs != X->rmap->bs) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Matrices must have same block size");
1182f5edf698SHong Zhang     ierr = MatGetRowUpperTriangular(X);CHKERRQ(ierr);
118352768537SHong Zhang     ierr = MatGetRowUpperTriangular(Y);CHKERRQ(ierr);
118452768537SHong Zhang     ierr = PetscMalloc1(Y->rmap->N,&nnz);CHKERRQ(ierr);
118552768537SHong Zhang     ierr = MatCreate(PetscObjectComm((PetscObject)Y),&B);CHKERRQ(ierr);
118652768537SHong Zhang     ierr = PetscObjectSetName((PetscObject)B,((PetscObject)Y)->name);CHKERRQ(ierr);
118752768537SHong Zhang     ierr = MatSetSizes(B,Y->rmap->n,Y->cmap->n,Y->rmap->N,Y->cmap->N);CHKERRQ(ierr);
118852768537SHong Zhang     ierr = MatSetBlockSizesFromMats(B,Y,Y);CHKERRQ(ierr);
118952768537SHong Zhang     ierr = MatSetType(B,(MatType) ((PetscObject)Y)->type_name);CHKERRQ(ierr);
119052768537SHong Zhang     ierr = MatAXPYGetPreallocation_SeqSBAIJ(Y,X,nnz);CHKERRQ(ierr);
119152768537SHong Zhang     ierr = MatSeqSBAIJSetPreallocation(B,bs,0,nnz);CHKERRQ(ierr);
119252768537SHong Zhang 
119352768537SHong Zhang     ierr = MatAXPY_BasicWithPreallocation(B,Y,a,X,str);CHKERRQ(ierr);
119452768537SHong Zhang 
119552768537SHong Zhang     ierr = MatHeaderReplace(Y,B);CHKERRQ(ierr);
119652768537SHong Zhang     ierr = PetscFree(nnz);CHKERRQ(ierr);
1197f5edf698SHong Zhang     ierr = MatRestoreRowUpperTriangular(X);CHKERRQ(ierr);
119852768537SHong Zhang     ierr = MatRestoreRowUpperTriangular(Y);CHKERRQ(ierr);
119942ee4b1aSHong Zhang   }
120042ee4b1aSHong Zhang   PetscFunctionReturn(0);
120142ee4b1aSHong Zhang }
120242ee4b1aSHong Zhang 
1203efcf0fc3SBarry Smith #undef __FUNCT__
1204efcf0fc3SBarry Smith #define __FUNCT__ "MatIsSymmetric_SeqSBAIJ"
1205ace3abfcSBarry Smith PetscErrorCode MatIsSymmetric_SeqSBAIJ(Mat A,PetscReal tol,PetscBool  *flg)
1206efcf0fc3SBarry Smith {
1207efcf0fc3SBarry Smith   PetscFunctionBegin;
1208efcf0fc3SBarry Smith   *flg = PETSC_TRUE;
1209efcf0fc3SBarry Smith   PetscFunctionReturn(0);
1210efcf0fc3SBarry Smith }
1211efcf0fc3SBarry Smith 
1212efcf0fc3SBarry Smith #undef __FUNCT__
1213efcf0fc3SBarry Smith #define __FUNCT__ "MatIsStructurallySymmetric_SeqSBAIJ"
1214ace3abfcSBarry Smith PetscErrorCode MatIsStructurallySymmetric_SeqSBAIJ(Mat A,PetscBool  *flg)
1215efcf0fc3SBarry Smith {
1216efcf0fc3SBarry Smith   PetscFunctionBegin;
1217efcf0fc3SBarry Smith   *flg = PETSC_TRUE;
1218efcf0fc3SBarry Smith   PetscFunctionReturn(0);
1219efcf0fc3SBarry Smith }
1220efcf0fc3SBarry Smith 
1221efcf0fc3SBarry Smith #undef __FUNCT__
1222efcf0fc3SBarry Smith #define __FUNCT__ "MatIsHermitian_SeqSBAIJ"
1223ace3abfcSBarry Smith PetscErrorCode MatIsHermitian_SeqSBAIJ(Mat A,PetscReal tol,PetscBool  *flg)
1224efcf0fc3SBarry Smith {
1225efcf0fc3SBarry Smith   PetscFunctionBegin;
1226efcf0fc3SBarry Smith   *flg = PETSC_FALSE;
1227efcf0fc3SBarry Smith   PetscFunctionReturn(0);
1228efcf0fc3SBarry Smith }
1229efcf0fc3SBarry Smith 
123099cafbc1SBarry Smith #undef __FUNCT__
123199cafbc1SBarry Smith #define __FUNCT__ "MatRealPart_SeqSBAIJ"
123299cafbc1SBarry Smith PetscErrorCode MatRealPart_SeqSBAIJ(Mat A)
123399cafbc1SBarry Smith {
123499cafbc1SBarry Smith   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
123599cafbc1SBarry Smith   PetscInt     i,nz = a->bs2*a->i[a->mbs];
1236dd6ea824SBarry Smith   MatScalar    *aa = a->a;
123799cafbc1SBarry Smith 
123899cafbc1SBarry Smith   PetscFunctionBegin;
123999cafbc1SBarry Smith   for (i=0; i<nz; i++) aa[i] = PetscRealPart(aa[i]);
124099cafbc1SBarry Smith   PetscFunctionReturn(0);
124199cafbc1SBarry Smith }
124299cafbc1SBarry Smith 
124399cafbc1SBarry Smith #undef __FUNCT__
124499cafbc1SBarry Smith #define __FUNCT__ "MatImaginaryPart_SeqSBAIJ"
124599cafbc1SBarry Smith PetscErrorCode MatImaginaryPart_SeqSBAIJ(Mat A)
124699cafbc1SBarry Smith {
124799cafbc1SBarry Smith   Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data;
124899cafbc1SBarry Smith   PetscInt     i,nz = a->bs2*a->i[a->mbs];
1249dd6ea824SBarry Smith   MatScalar    *aa = a->a;
125099cafbc1SBarry Smith 
125199cafbc1SBarry Smith   PetscFunctionBegin;
125299cafbc1SBarry Smith   for (i=0; i<nz; i++) aa[i] = PetscImaginaryPart(aa[i]);
125399cafbc1SBarry Smith   PetscFunctionReturn(0);
125499cafbc1SBarry Smith }
125599cafbc1SBarry Smith 
12563bededecSBarry Smith #undef __FUNCT__
12573bededecSBarry Smith #define __FUNCT__ "MatZeroRowsColumns_SeqSBAIJ"
12583bededecSBarry Smith PetscErrorCode MatZeroRowsColumns_SeqSBAIJ(Mat A,PetscInt is_n,const PetscInt is_idx[],PetscScalar diag,Vec x, Vec b)
12593bededecSBarry Smith {
12603bededecSBarry Smith   Mat_SeqSBAIJ      *baij=(Mat_SeqSBAIJ*)A->data;
12613bededecSBarry Smith   PetscErrorCode    ierr;
12623bededecSBarry Smith   PetscInt          i,j,k,count;
12633bededecSBarry Smith   PetscInt          bs   =A->rmap->bs,bs2=baij->bs2,row,col;
12643bededecSBarry Smith   PetscScalar       zero = 0.0;
12653bededecSBarry Smith   MatScalar         *aa;
12663bededecSBarry Smith   const PetscScalar *xx;
12673bededecSBarry Smith   PetscScalar       *bb;
126856777dd2SBarry Smith   PetscBool         *zeroed,vecs = PETSC_FALSE;
12693bededecSBarry Smith 
12703bededecSBarry Smith   PetscFunctionBegin;
12713bededecSBarry Smith   /* fix right hand side if needed */
12723bededecSBarry Smith   if (x && b) {
12733bededecSBarry Smith     ierr = VecGetArrayRead(x,&xx);CHKERRQ(ierr);
12743bededecSBarry Smith     ierr = VecGetArray(b,&bb);CHKERRQ(ierr);
127556777dd2SBarry Smith     vecs = PETSC_TRUE;
12763bededecSBarry Smith   }
12773bededecSBarry Smith 
12783bededecSBarry Smith   /* zero the columns */
12791795a4d1SJed Brown   ierr = PetscCalloc1(A->rmap->n,&zeroed);CHKERRQ(ierr);
12803bededecSBarry Smith   for (i=0; i<is_n; i++) {
12813bededecSBarry 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]);
12823bededecSBarry Smith     zeroed[is_idx[i]] = PETSC_TRUE;
12833bededecSBarry Smith   }
128456777dd2SBarry Smith   if (vecs) {
128556777dd2SBarry Smith     for (i=0; i<A->rmap->N; i++) {
128656777dd2SBarry Smith       row = i/bs;
128756777dd2SBarry Smith       for (j=baij->i[row]; j<baij->i[row+1]; j++) {
128856777dd2SBarry Smith         for (k=0; k<bs; k++) {
128956777dd2SBarry Smith           col = bs*baij->j[j] + k;
129056777dd2SBarry Smith           if (col <= i) continue;
129156777dd2SBarry Smith           aa = ((MatScalar*)(baij->a)) + j*bs2 + (i%bs) + bs*k;
129226fbe8dcSKarl Rupp           if (!zeroed[i] && zeroed[col]) bb[i]   -= aa[0]*xx[col];
129326fbe8dcSKarl Rupp           if (zeroed[i] && !zeroed[col]) bb[col] -= aa[0]*xx[i];
129456777dd2SBarry Smith         }
129556777dd2SBarry Smith       }
129656777dd2SBarry Smith     }
129726fbe8dcSKarl Rupp     for (i=0; i<is_n; i++) bb[is_idx[i]] = diag*xx[is_idx[i]];
129856777dd2SBarry Smith   }
129956777dd2SBarry Smith 
13003bededecSBarry Smith   for (i=0; i<A->rmap->N; i++) {
13013bededecSBarry Smith     if (!zeroed[i]) {
13023bededecSBarry Smith       row = i/bs;
13033bededecSBarry Smith       for (j=baij->i[row]; j<baij->i[row+1]; j++) {
13043bededecSBarry Smith         for (k=0; k<bs; k++) {
13053bededecSBarry Smith           col = bs*baij->j[j] + k;
13063bededecSBarry Smith           if (zeroed[col]) {
13073bededecSBarry Smith             aa = ((MatScalar*)(baij->a)) + j*bs2 + (i%bs) + bs*k;
13083bededecSBarry Smith             aa[0] = 0.0;
13093bededecSBarry Smith           }
13103bededecSBarry Smith         }
13113bededecSBarry Smith       }
13123bededecSBarry Smith     }
13133bededecSBarry Smith   }
13143bededecSBarry Smith   ierr = PetscFree(zeroed);CHKERRQ(ierr);
131556777dd2SBarry Smith   if (vecs) {
131656777dd2SBarry Smith     ierr = VecRestoreArrayRead(x,&xx);CHKERRQ(ierr);
131756777dd2SBarry Smith     ierr = VecRestoreArray(b,&bb);CHKERRQ(ierr);
131856777dd2SBarry Smith   }
13193bededecSBarry Smith 
13203bededecSBarry Smith   /* zero the rows */
13213bededecSBarry Smith   for (i=0; i<is_n; i++) {
13223bededecSBarry Smith     row   = is_idx[i];
13233bededecSBarry Smith     count = (baij->i[row/bs +1] - baij->i[row/bs])*bs;
13243bededecSBarry Smith     aa    = ((MatScalar*)(baij->a)) + baij->i[row/bs]*bs2 + (row%bs);
13253bededecSBarry Smith     for (k=0; k<count; k++) {
13263bededecSBarry Smith       aa[0] =  zero;
13273bededecSBarry Smith       aa   += bs;
13283bededecSBarry Smith     }
13293bededecSBarry Smith     if (diag != 0.0) {
13303bededecSBarry Smith       ierr = (*A->ops->setvalues)(A,1,&row,1,&row,&diag,INSERT_VALUES);CHKERRQ(ierr);
13313bededecSBarry Smith     }
13323bededecSBarry Smith   }
13333bededecSBarry Smith   ierr = MatAssemblyEnd_SeqSBAIJ(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
13343bededecSBarry Smith   PetscFunctionReturn(0);
13353bededecSBarry Smith }
13363bededecSBarry Smith 
133749b5e25fSSatish Balay /* -------------------------------------------------------------------*/
13383964eb88SJed Brown static struct _MatOps MatOps_Values = {MatSetValues_SeqSBAIJ,
133949b5e25fSSatish Balay                                        MatGetRow_SeqSBAIJ,
134049b5e25fSSatish Balay                                        MatRestoreRow_SeqSBAIJ,
134149b5e25fSSatish Balay                                        MatMult_SeqSBAIJ_N,
134297304618SKris Buschelman                                /*  4*/ MatMultAdd_SeqSBAIJ_N,
1343431c96f7SBarry Smith                                        MatMult_SeqSBAIJ_N,       /* transpose versions are same as non-transpose versions */
1344e005ede5SBarry Smith                                        MatMultAdd_SeqSBAIJ_N,
1345db4efbfdSBarry Smith                                        0,
134649b5e25fSSatish Balay                                        0,
134749b5e25fSSatish Balay                                        0,
134897304618SKris Buschelman                                /* 10*/ 0,
134949b5e25fSSatish Balay                                        0,
1350c078aec8SLisandro Dalcin                                        MatCholeskyFactor_SeqSBAIJ,
135141f059aeSBarry Smith                                        MatSOR_SeqSBAIJ,
135249b5e25fSSatish Balay                                        MatTranspose_SeqSBAIJ,
135397304618SKris Buschelman                                /* 15*/ MatGetInfo_SeqSBAIJ,
135449b5e25fSSatish Balay                                        MatEqual_SeqSBAIJ,
135549b5e25fSSatish Balay                                        MatGetDiagonal_SeqSBAIJ,
135649b5e25fSSatish Balay                                        MatDiagonalScale_SeqSBAIJ,
135749b5e25fSSatish Balay                                        MatNorm_SeqSBAIJ,
135897304618SKris Buschelman                                /* 20*/ 0,
135949b5e25fSSatish Balay                                        MatAssemblyEnd_SeqSBAIJ,
136049b5e25fSSatish Balay                                        MatSetOption_SeqSBAIJ,
136149b5e25fSSatish Balay                                        MatZeroEntries_SeqSBAIJ,
1362d519adbfSMatthew Knepley                                /* 24*/ 0,
136349b5e25fSSatish Balay                                        0,
136449b5e25fSSatish Balay                                        0,
1365db4efbfdSBarry Smith                                        0,
1366db4efbfdSBarry Smith                                        0,
13674994cf47SJed Brown                                /* 29*/ MatSetUp_SeqSBAIJ,
1368c464158bSHong Zhang                                        0,
1369db4efbfdSBarry Smith                                        0,
13708c778c55SBarry Smith                                        0,
13718c778c55SBarry Smith                                        0,
1372d519adbfSMatthew Knepley                                /* 34*/ MatDuplicate_SeqSBAIJ,
1373719d5645SBarry Smith                                        0,
1374719d5645SBarry Smith                                        0,
137549b5e25fSSatish Balay                                        0,
1376c84f5b01SHong Zhang                                        MatICCFactor_SeqSBAIJ,
1377d519adbfSMatthew Knepley                                /* 39*/ MatAXPY_SeqSBAIJ,
137849b5e25fSSatish Balay                                        MatGetSubMatrices_SeqSBAIJ,
137949b5e25fSSatish Balay                                        MatIncreaseOverlap_SeqSBAIJ,
138049b5e25fSSatish Balay                                        MatGetValues_SeqSBAIJ,
13813c896bc6SHong Zhang                                        MatCopy_SeqSBAIJ,
1382d519adbfSMatthew Knepley                                /* 44*/ 0,
138349b5e25fSSatish Balay                                        MatScale_SeqSBAIJ,
138449b5e25fSSatish Balay                                        0,
138549b5e25fSSatish Balay                                        0,
13863bededecSBarry Smith                                        MatZeroRowsColumns_SeqSBAIJ,
1387f73d5cc4SBarry Smith                                /* 49*/ 0,
138849b5e25fSSatish Balay                                        MatGetRowIJ_SeqSBAIJ,
138949b5e25fSSatish Balay                                        MatRestoreRowIJ_SeqSBAIJ,
139049b5e25fSSatish Balay                                        0,
139149b5e25fSSatish Balay                                        0,
1392d519adbfSMatthew Knepley                                /* 54*/ 0,
139349b5e25fSSatish Balay                                        0,
139449b5e25fSSatish Balay                                        0,
139549b5e25fSSatish Balay                                        0,
139649b5e25fSSatish Balay                                        MatSetValuesBlocked_SeqSBAIJ,
1397d519adbfSMatthew Knepley                                /* 59*/ MatGetSubMatrix_SeqSBAIJ,
139849b5e25fSSatish Balay                                        0,
139949b5e25fSSatish Balay                                        0,
1400357abbc8SBarry Smith                                        0,
1401d959ec07SHong Zhang                                        0,
1402d519adbfSMatthew Knepley                                /* 64*/ 0,
1403d959ec07SHong Zhang                                        0,
1404d959ec07SHong Zhang                                        0,
1405d959ec07SHong Zhang                                        0,
1406d959ec07SHong Zhang                                        0,
1407d519adbfSMatthew Knepley                                /* 69*/ MatGetRowMaxAbs_SeqSBAIJ,
14083e0d88b5SBarry Smith                                        0,
14093e0d88b5SBarry Smith                                        0,
14103e0d88b5SBarry Smith                                        0,
14113e0d88b5SBarry Smith                                        0,
1412d519adbfSMatthew Knepley                                /* 74*/ 0,
14133e0d88b5SBarry Smith                                        0,
14143e0d88b5SBarry Smith                                        0,
14153e0d88b5SBarry Smith                                        0,
14163e0d88b5SBarry Smith                                        0,
1417d519adbfSMatthew Knepley                                /* 79*/ 0,
14183e0d88b5SBarry Smith                                        0,
14193e0d88b5SBarry Smith                                        0,
142097304618SKris Buschelman                                        MatGetInertia_SeqSBAIJ,
14215bba2384SShri Abhyankar                                        MatLoad_SeqSBAIJ,
1422d519adbfSMatthew Knepley                                /* 84*/ MatIsSymmetric_SeqSBAIJ,
1423865e5f61SKris Buschelman                                        MatIsHermitian_SeqSBAIJ,
1424efcf0fc3SBarry Smith                                        MatIsStructurallySymmetric_SeqSBAIJ,
1425865e5f61SKris Buschelman                                        0,
1426865e5f61SKris Buschelman                                        0,
1427d519adbfSMatthew Knepley                                /* 89*/ 0,
1428865e5f61SKris Buschelman                                        0,
1429865e5f61SKris Buschelman                                        0,
1430865e5f61SKris Buschelman                                        0,
1431865e5f61SKris Buschelman                                        0,
1432d519adbfSMatthew Knepley                                /* 94*/ 0,
1433865e5f61SKris Buschelman                                        0,
1434865e5f61SKris Buschelman                                        0,
143599cafbc1SBarry Smith                                        0,
143699cafbc1SBarry Smith                                        0,
1437d519adbfSMatthew Knepley                                /* 99*/ 0,
143899cafbc1SBarry Smith                                        0,
143999cafbc1SBarry Smith                                        0,
144099cafbc1SBarry Smith                                        0,
144199cafbc1SBarry Smith                                        0,
1442d519adbfSMatthew Knepley                                /*104*/ 0,
144399cafbc1SBarry Smith                                        MatRealPart_SeqSBAIJ,
1444f5edf698SHong Zhang                                        MatImaginaryPart_SeqSBAIJ,
1445f5edf698SHong Zhang                                        MatGetRowUpperTriangular_SeqSBAIJ,
14462af78befSBarry Smith                                        MatRestoreRowUpperTriangular_SeqSBAIJ,
1447d519adbfSMatthew Knepley                                /*109*/ 0,
14482af78befSBarry Smith                                        0,
14492af78befSBarry Smith                                        0,
14502af78befSBarry Smith                                        0,
1451547795f9SHong Zhang                                        MatMissingDiagonal_SeqSBAIJ,
1452547795f9SHong Zhang                                /*114*/ 0,
1453547795f9SHong Zhang                                        0,
1454547795f9SHong Zhang                                        0,
1455547795f9SHong Zhang                                        0,
1456547795f9SHong Zhang                                        0,
1457547795f9SHong Zhang                                /*119*/ 0,
1458547795f9SHong Zhang                                        0,
14592f480046SShri Abhyankar                                        0,
14603964eb88SJed Brown                                        0,
14613964eb88SJed Brown                                        0,
14623964eb88SJed Brown                                /*124*/ 0,
14633964eb88SJed Brown                                        0,
14643964eb88SJed Brown                                        0,
14653964eb88SJed Brown                                        0,
14663964eb88SJed Brown                                        0,
14673964eb88SJed Brown                                /*129*/ 0,
14683964eb88SJed Brown                                        0,
14693964eb88SJed Brown                                        0,
14703964eb88SJed Brown                                        0,
14713964eb88SJed Brown                                        0,
14723964eb88SJed Brown                                /*134*/ 0,
14733964eb88SJed Brown                                        0,
14743964eb88SJed Brown                                        0,
14753964eb88SJed Brown                                        0,
14763964eb88SJed Brown                                        0,
14773964eb88SJed Brown                                /*139*/ 0,
1478f9426fe0SMark Adams                                        0,
14795bba2384SShri Abhyankar                                        0
148099cafbc1SBarry Smith };
1481be1d678aSKris Buschelman 
14824a2ae208SSatish Balay #undef __FUNCT__
14834a2ae208SSatish Balay #define __FUNCT__ "MatStoreValues_SeqSBAIJ"
14847087cfbeSBarry Smith PetscErrorCode  MatStoreValues_SeqSBAIJ(Mat mat)
148549b5e25fSSatish Balay {
14864afc71dfSHong Zhang   Mat_SeqSBAIJ   *aij = (Mat_SeqSBAIJ*)mat->data;
1487d0f46423SBarry Smith   PetscInt       nz   = aij->i[mat->rmap->N]*mat->rmap->bs*aij->bs2;
1488dfbe8321SBarry Smith   PetscErrorCode ierr;
148949b5e25fSSatish Balay 
149049b5e25fSSatish Balay   PetscFunctionBegin;
1491e7e72b3dSBarry Smith   if (aij->nonew != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first");
149249b5e25fSSatish Balay 
149349b5e25fSSatish Balay   /* allocate space for values if not already there */
149449b5e25fSSatish Balay   if (!aij->saved_values) {
1495785e854fSJed Brown     ierr = PetscMalloc1((nz+1),&aij->saved_values);CHKERRQ(ierr);
149649b5e25fSSatish Balay   }
149749b5e25fSSatish Balay 
149849b5e25fSSatish Balay   /* copy values over */
149987828ca2SBarry Smith   ierr = PetscMemcpy(aij->saved_values,aij->a,nz*sizeof(PetscScalar));CHKERRQ(ierr);
150049b5e25fSSatish Balay   PetscFunctionReturn(0);
150149b5e25fSSatish Balay }
150249b5e25fSSatish Balay 
15034a2ae208SSatish Balay #undef __FUNCT__
15044a2ae208SSatish Balay #define __FUNCT__ "MatRetrieveValues_SeqSBAIJ"
15057087cfbeSBarry Smith PetscErrorCode  MatRetrieveValues_SeqSBAIJ(Mat mat)
150649b5e25fSSatish Balay {
15074afc71dfSHong Zhang   Mat_SeqSBAIJ   *aij = (Mat_SeqSBAIJ*)mat->data;
15086849ba73SBarry Smith   PetscErrorCode ierr;
1509d0f46423SBarry Smith   PetscInt       nz = aij->i[mat->rmap->N]*mat->rmap->bs*aij->bs2;
151049b5e25fSSatish Balay 
151149b5e25fSSatish Balay   PetscFunctionBegin;
1512e7e72b3dSBarry Smith   if (aij->nonew != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first");
1513e7e72b3dSBarry Smith   if (!aij->saved_values) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatStoreValues(A);first");
151449b5e25fSSatish Balay 
151549b5e25fSSatish Balay   /* copy values over */
151687828ca2SBarry Smith   ierr = PetscMemcpy(aij->a,aij->saved_values,nz*sizeof(PetscScalar));CHKERRQ(ierr);
151749b5e25fSSatish Balay   PetscFunctionReturn(0);
151849b5e25fSSatish Balay }
151949b5e25fSSatish Balay 
15204a2ae208SSatish Balay #undef __FUNCT__
1521a23d5eceSKris Buschelman #define __FUNCT__ "MatSeqSBAIJSetPreallocation_SeqSBAIJ"
15227087cfbeSBarry Smith PetscErrorCode  MatSeqSBAIJSetPreallocation_SeqSBAIJ(Mat B,PetscInt bs,PetscInt nz,PetscInt *nnz)
152349b5e25fSSatish Balay {
1524c464158bSHong Zhang   Mat_SeqSBAIJ   *b = (Mat_SeqSBAIJ*)B->data;
15256849ba73SBarry Smith   PetscErrorCode ierr;
1526535b19f3SBarry Smith   PetscInt       i,mbs,bs2;
15272576faa2SJed Brown   PetscBool      skipallocation = PETSC_FALSE,flg = PETSC_FALSE,realalloc = PETSC_FALSE;
152849b5e25fSSatish Balay 
152949b5e25fSSatish Balay   PetscFunctionBegin;
15302576faa2SJed Brown   if (nz >= 0 || nnz) realalloc = PETSC_TRUE;
1531273d9f13SBarry Smith   B->preallocated = PETSC_TRUE;
1532db4efbfdSBarry Smith 
153333d57670SJed Brown   ierr = MatSetBlockSize(B,PetscAbs(bs));CHKERRQ(ierr);
153426283091SBarry Smith   ierr = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr);
153526283091SBarry Smith   ierr = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr);
1536e02043d6SBarry Smith   ierr = PetscLayoutGetBlockSize(B->rmap,&bs);CHKERRQ(ierr);
1537899cda47SBarry Smith 
1538d0f46423SBarry Smith   mbs = B->rmap->N/bs;
153949b5e25fSSatish Balay   bs2 = bs*bs;
154049b5e25fSSatish Balay 
1541e7e72b3dSBarry Smith   if (mbs*bs != B->rmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Number rows, cols must be divisible by blocksize");
154249b5e25fSSatish Balay 
1543ab93d7beSBarry Smith   if (nz == MAT_SKIP_ALLOCATION) {
1544ab93d7beSBarry Smith     skipallocation = PETSC_TRUE;
1545ab93d7beSBarry Smith     nz             = 0;
1546ab93d7beSBarry Smith   }
1547ab93d7beSBarry Smith 
1548435da068SBarry Smith   if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 3;
1549e32f2f54SBarry Smith   if (nz < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"nz cannot be less than 0: value %D",nz);
155049b5e25fSSatish Balay   if (nnz) {
155149b5e25fSSatish Balay     for (i=0; i<mbs; i++) {
1552e32f2f54SBarry 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]);
1553e32f2f54SBarry 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);
155449b5e25fSSatish Balay     }
155549b5e25fSSatish Balay   }
155649b5e25fSSatish Balay 
1557db4efbfdSBarry Smith   B->ops->mult             = MatMult_SeqSBAIJ_N;
1558db4efbfdSBarry Smith   B->ops->multadd          = MatMultAdd_SeqSBAIJ_N;
1559db4efbfdSBarry Smith   B->ops->multtranspose    = MatMult_SeqSBAIJ_N;
1560db4efbfdSBarry Smith   B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_N;
156126fbe8dcSKarl Rupp 
15620298fd71SBarry Smith   ierr  = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_no_unroll",&flg,NULL);CHKERRQ(ierr);
156349b5e25fSSatish Balay   if (!flg) {
156449b5e25fSSatish Balay     switch (bs) {
156549b5e25fSSatish Balay     case 1:
156649b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_1;
156749b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_1;
1568431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_1;
1569431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_1;
157049b5e25fSSatish Balay       break;
157149b5e25fSSatish Balay     case 2:
157249b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_2;
157349b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_2;
1574431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_2;
1575431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_2;
157649b5e25fSSatish Balay       break;
157749b5e25fSSatish Balay     case 3:
157849b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_3;
157949b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_3;
1580431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_3;
1581431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_3;
158249b5e25fSSatish Balay       break;
158349b5e25fSSatish Balay     case 4:
158449b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_4;
158549b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_4;
1586431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_4;
1587431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_4;
158849b5e25fSSatish Balay       break;
158949b5e25fSSatish Balay     case 5:
159049b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_5;
159149b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_5;
1592431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_5;
1593431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_5;
159449b5e25fSSatish Balay       break;
159549b5e25fSSatish Balay     case 6:
159649b5e25fSSatish Balay       B->ops->mult             = MatMult_SeqSBAIJ_6;
159749b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_6;
1598431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_6;
1599431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_6;
160049b5e25fSSatish Balay       break;
160149b5e25fSSatish Balay     case 7:
1602de53e5efSHong Zhang       B->ops->mult             = MatMult_SeqSBAIJ_7;
160349b5e25fSSatish Balay       B->ops->multadd          = MatMultAdd_SeqSBAIJ_7;
1604431c96f7SBarry Smith       B->ops->multtranspose    = MatMult_SeqSBAIJ_7;
1605431c96f7SBarry Smith       B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_7;
160649b5e25fSSatish Balay       break;
160749b5e25fSSatish Balay     }
160849b5e25fSSatish Balay   }
160949b5e25fSSatish Balay 
161049b5e25fSSatish Balay   b->mbs = mbs;
16114afc71dfSHong Zhang   b->nbs = mbs;
1612ab93d7beSBarry Smith   if (!skipallocation) {
16132ee49352SLisandro Dalcin     if (!b->imax) {
1614dcca6d9dSJed Brown       ierr = PetscMalloc2(mbs,&b->imax,mbs,&b->ilen);CHKERRQ(ierr);
161526fbe8dcSKarl Rupp 
1616c760cd28SBarry Smith       b->free_imax_ilen = PETSC_TRUE;
161726fbe8dcSKarl Rupp 
16183bb1ff40SBarry Smith       ierr = PetscLogObjectMemory((PetscObject)B,2*mbs*sizeof(PetscInt));CHKERRQ(ierr);
16192ee49352SLisandro Dalcin     }
162049b5e25fSSatish Balay     if (!nnz) {
1621435da068SBarry Smith       if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 5;
162249b5e25fSSatish Balay       else if (nz <= 0) nz = 1;
162326fbe8dcSKarl Rupp       for (i=0; i<mbs; i++) b->imax[i] = nz;
1624153ea458SHong Zhang       nz = nz*mbs; /* total nz */
162549b5e25fSSatish Balay     } else {
162649b5e25fSSatish Balay       nz = 0;
16278cef66ccSHong Zhang       for (i=0; i<mbs; i++) {b->imax[i] = nnz[i]; nz += nnz[i];}
162849b5e25fSSatish Balay     }
16292ee49352SLisandro Dalcin     /* b->ilen will count nonzeros in each block row so far. */
163026fbe8dcSKarl Rupp     for (i=0; i<mbs; i++) b->ilen[i] = 0;
16316c6c5352SBarry Smith     /* nz=(nz+mbs)/2; */ /* total diagonal and superdiagonal nonzero blocks */
163249b5e25fSSatish Balay 
163349b5e25fSSatish Balay     /* allocate the matrix space */
16342ee49352SLisandro Dalcin     ierr = MatSeqXAIJFreeAIJ(B,&b->a,&b->j,&b->i);CHKERRQ(ierr);
1635dcca6d9dSJed Brown     ierr = PetscMalloc3(bs2*nz,&b->a,nz,&b->j,B->rmap->N+1,&b->i);CHKERRQ(ierr);
16363bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)B,(B->rmap->N+1)*sizeof(PetscInt)+nz*(bs2*sizeof(PetscScalar)+sizeof(PetscInt)));CHKERRQ(ierr);
16376c6c5352SBarry Smith     ierr = PetscMemzero(b->a,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr);
163813f74950SBarry Smith     ierr = PetscMemzero(b->j,nz*sizeof(PetscInt));CHKERRQ(ierr);
163926fbe8dcSKarl Rupp 
164049b5e25fSSatish Balay     b->singlemalloc = PETSC_TRUE;
164149b5e25fSSatish Balay 
164249b5e25fSSatish Balay     /* pointer to beginning of each row */
1643e60cf9a0SBarry Smith     b->i[0] = 0;
164426fbe8dcSKarl Rupp     for (i=1; i<mbs+1; i++) b->i[i] = b->i[i-1] + b->imax[i-1];
164526fbe8dcSKarl Rupp 
1646e6b907acSBarry Smith     b->free_a  = PETSC_TRUE;
1647e6b907acSBarry Smith     b->free_ij = PETSC_TRUE;
1648e811da20SHong Zhang   } else {
1649e6b907acSBarry Smith     b->free_a  = PETSC_FALSE;
1650e6b907acSBarry Smith     b->free_ij = PETSC_FALSE;
1651ab93d7beSBarry Smith   }
165249b5e25fSSatish Balay 
1653d0f46423SBarry Smith   B->rmap->bs = bs;
165449b5e25fSSatish Balay   b->bs2      = bs2;
16556c6c5352SBarry Smith   b->nz       = 0;
1656b32cb4a7SJed Brown   b->maxnz    = nz;
1657153ea458SHong Zhang 
165816cdd363SHong Zhang   b->inew    = 0;
165916cdd363SHong Zhang   b->jnew    = 0;
166016cdd363SHong Zhang   b->anew    = 0;
166116cdd363SHong Zhang   b->a2anew  = 0;
16621a3463dfSHong Zhang   b->permute = PETSC_FALSE;
16632576faa2SJed Brown   if (realalloc) {ierr = MatSetOption(B,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);}
1664c464158bSHong Zhang   PetscFunctionReturn(0);
1665c464158bSHong Zhang }
1666153ea458SHong Zhang 
166738f409ebSLisandro Dalcin #undef __FUNCT__
166838f409ebSLisandro Dalcin #define __FUNCT__ "MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ"
166938f409ebSLisandro Dalcin PetscErrorCode MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ(Mat B,PetscInt bs,const PetscInt ii[],const PetscInt jj[], const PetscScalar V[])
167038f409ebSLisandro Dalcin {
167138f409ebSLisandro Dalcin   PetscInt       i,j,m,nz,nz_max=0,*nnz;
167238f409ebSLisandro Dalcin   PetscScalar    *values=0;
167338f409ebSLisandro Dalcin   PetscBool      roworiented = ((Mat_SeqSBAIJ*)B->data)->roworiented;
167438f409ebSLisandro Dalcin   PetscErrorCode ierr;
167538f409ebSLisandro Dalcin   PetscFunctionBegin;
167638f409ebSLisandro Dalcin   if (bs < 1) SETERRQ1(PetscObjectComm((PetscObject)B),PETSC_ERR_ARG_OUTOFRANGE,"Invalid block size specified, must be positive but it is %D",bs);
167738f409ebSLisandro Dalcin   ierr   = PetscLayoutSetBlockSize(B->rmap,bs);CHKERRQ(ierr);
167838f409ebSLisandro Dalcin   ierr   = PetscLayoutSetBlockSize(B->cmap,bs);CHKERRQ(ierr);
167938f409ebSLisandro Dalcin   ierr   = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr);
168038f409ebSLisandro Dalcin   ierr   = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr);
168138f409ebSLisandro Dalcin   ierr   = PetscLayoutGetBlockSize(B->rmap,&bs);CHKERRQ(ierr);
168238f409ebSLisandro Dalcin   m      = B->rmap->n/bs;
168338f409ebSLisandro Dalcin 
168438f409ebSLisandro Dalcin   if (ii[0]) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"ii[0] must be 0 but it is %D",ii[0]);
1685785e854fSJed Brown   ierr = PetscMalloc1((m+1),&nnz);CHKERRQ(ierr);
168638f409ebSLisandro Dalcin   for (i=0; i<m; i++) {
168738f409ebSLisandro Dalcin     nz = ii[i+1] - ii[i];
168838f409ebSLisandro Dalcin     if (nz < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row %D has a negative number of columns %D",i,nz);
168938f409ebSLisandro Dalcin     nz_max = PetscMax(nz_max,nz);
169038f409ebSLisandro Dalcin     nnz[i] = nz;
169138f409ebSLisandro Dalcin   }
169238f409ebSLisandro Dalcin   ierr = MatSeqSBAIJSetPreallocation(B,bs,0,nnz);CHKERRQ(ierr);
169338f409ebSLisandro Dalcin   ierr = PetscFree(nnz);CHKERRQ(ierr);
169438f409ebSLisandro Dalcin 
169538f409ebSLisandro Dalcin   values = (PetscScalar*)V;
169638f409ebSLisandro Dalcin   if (!values) {
16971795a4d1SJed Brown     ierr = PetscCalloc1(bs*bs*nz_max,&values);CHKERRQ(ierr);
169838f409ebSLisandro Dalcin   }
169938f409ebSLisandro Dalcin   for (i=0; i<m; i++) {
170038f409ebSLisandro Dalcin     PetscInt          ncols  = ii[i+1] - ii[i];
170138f409ebSLisandro Dalcin     const PetscInt    *icols = jj + ii[i];
170238f409ebSLisandro Dalcin     if (!roworiented || bs == 1) {
170338f409ebSLisandro Dalcin       const PetscScalar *svals = values + (V ? (bs*bs*ii[i]) : 0);
170438f409ebSLisandro Dalcin       ierr = MatSetValuesBlocked_SeqSBAIJ(B,1,&i,ncols,icols,svals,INSERT_VALUES);CHKERRQ(ierr);
170538f409ebSLisandro Dalcin     } else {
170638f409ebSLisandro Dalcin       for (j=0; j<ncols; j++) {
170738f409ebSLisandro Dalcin         const PetscScalar *svals = values + (V ? (bs*bs*(ii[i]+j)) : 0);
170838f409ebSLisandro Dalcin         ierr = MatSetValuesBlocked_SeqSBAIJ(B,1,&i,1,&icols[j],svals,INSERT_VALUES);CHKERRQ(ierr);
170938f409ebSLisandro Dalcin       }
171038f409ebSLisandro Dalcin     }
171138f409ebSLisandro Dalcin   }
171238f409ebSLisandro Dalcin   if (!V) { ierr = PetscFree(values);CHKERRQ(ierr); }
171338f409ebSLisandro Dalcin   ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
171438f409ebSLisandro Dalcin   ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
171538f409ebSLisandro Dalcin   ierr = MatSetOption(B,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
171638f409ebSLisandro Dalcin   PetscFunctionReturn(0);
171738f409ebSLisandro Dalcin }
171838f409ebSLisandro Dalcin 
1719db4efbfdSBarry Smith /*
1720db4efbfdSBarry Smith    This is used to set the numeric factorization for both Cholesky and ICC symbolic factorization
1721db4efbfdSBarry Smith */
17228b1456e3SHong Zhang #undef __FUNCT__
1723d595f711SHong Zhang #define __FUNCT__ "MatSeqSBAIJSetNumericFactorization_inplace"
1724ace3abfcSBarry Smith PetscErrorCode MatSeqSBAIJSetNumericFactorization_inplace(Mat B,PetscBool natural)
1725db4efbfdSBarry Smith {
1726db4efbfdSBarry Smith   PetscErrorCode ierr;
1727ace3abfcSBarry Smith   PetscBool      flg = PETSC_FALSE;
1728db4efbfdSBarry Smith   PetscInt       bs  = B->rmap->bs;
1729db4efbfdSBarry Smith 
1730db4efbfdSBarry Smith   PetscFunctionBegin;
17310298fd71SBarry Smith   ierr = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_no_unroll",&flg,NULL);CHKERRQ(ierr);
1732db4efbfdSBarry Smith   if (flg) bs = 8;
1733db4efbfdSBarry Smith 
1734db4efbfdSBarry Smith   if (!natural) {
1735db4efbfdSBarry Smith     switch (bs) {
1736db4efbfdSBarry Smith     case 1:
1737d595f711SHong Zhang       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_inplace;
1738db4efbfdSBarry Smith       break;
1739db4efbfdSBarry Smith     case 2:
1740db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2;
1741db4efbfdSBarry Smith       break;
1742db4efbfdSBarry Smith     case 3:
1743db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3;
1744db4efbfdSBarry Smith       break;
1745db4efbfdSBarry Smith     case 4:
1746db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4;
1747db4efbfdSBarry Smith       break;
1748db4efbfdSBarry Smith     case 5:
1749db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5;
1750db4efbfdSBarry Smith       break;
1751db4efbfdSBarry Smith     case 6:
1752db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6;
1753db4efbfdSBarry Smith       break;
1754db4efbfdSBarry Smith     case 7:
1755db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7;
1756db4efbfdSBarry Smith       break;
1757db4efbfdSBarry Smith     default:
1758db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N;
1759db4efbfdSBarry Smith       break;
1760db4efbfdSBarry Smith     }
1761db4efbfdSBarry Smith   } else {
1762db4efbfdSBarry Smith     switch (bs) {
1763db4efbfdSBarry Smith     case 1:
1764d595f711SHong Zhang       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering_inplace;
1765db4efbfdSBarry Smith       break;
1766db4efbfdSBarry Smith     case 2:
1767db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering;
1768db4efbfdSBarry Smith       break;
1769db4efbfdSBarry Smith     case 3:
1770db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering;
1771db4efbfdSBarry Smith       break;
1772db4efbfdSBarry Smith     case 4:
1773db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering;
1774db4efbfdSBarry Smith       break;
1775db4efbfdSBarry Smith     case 5:
1776db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering;
1777db4efbfdSBarry Smith       break;
1778db4efbfdSBarry Smith     case 6:
1779db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering;
1780db4efbfdSBarry Smith       break;
1781db4efbfdSBarry Smith     case 7:
1782db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering;
1783db4efbfdSBarry Smith       break;
1784db4efbfdSBarry Smith     default:
1785db4efbfdSBarry Smith       B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering;
1786db4efbfdSBarry Smith       break;
1787db4efbfdSBarry Smith     }
1788db4efbfdSBarry Smith   }
1789db4efbfdSBarry Smith   PetscFunctionReturn(0);
1790db4efbfdSBarry Smith }
1791db4efbfdSBarry Smith 
17928cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqAIJ(Mat, MatType,MatReuse,Mat*);
17938cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqBAIJ(Mat, MatType,MatReuse,Mat*);
1794d769727bSBarry Smith 
17955c9eb25fSBarry Smith #undef __FUNCT__
17965c9eb25fSBarry Smith #define __FUNCT__ "MatGetFactor_seqsbaij_petsc"
17978cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_seqsbaij_petsc(Mat A,MatFactorType ftype,Mat *B)
17985c9eb25fSBarry Smith {
1799d0f46423SBarry Smith   PetscInt       n = A->rmap->n;
18005c9eb25fSBarry Smith   PetscErrorCode ierr;
18015c9eb25fSBarry Smith 
18025c9eb25fSBarry Smith   PetscFunctionBegin;
18030e92d65fSHong Zhang #if defined(PETSC_USE_COMPLEX)
18040e92d65fSHong Zhang   if (A->hermitian) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Hermitian Factor is not supported");
18050e92d65fSHong Zhang #endif
1806ce94432eSBarry Smith   ierr = MatCreate(PetscObjectComm((PetscObject)A),B);CHKERRQ(ierr);
18075c9eb25fSBarry Smith   ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr);
18085c9eb25fSBarry Smith   if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) {
18095c9eb25fSBarry Smith     ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr);
18100298fd71SBarry Smith     ierr = MatSeqSBAIJSetPreallocation(*B,A->rmap->bs,MAT_SKIP_ALLOCATION,NULL);CHKERRQ(ierr);
181126fbe8dcSKarl Rupp 
18127b056e98SHong Zhang     (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqSBAIJ;
1813c6d0d4f0SHong Zhang     (*B)->ops->iccfactorsymbolic      = MatICCFactorSymbolic_SeqSBAIJ;
1814e32f2f54SBarry Smith   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Factor type not supported");
1815d5f3da31SBarry Smith   (*B)->factortype = ftype;
18165c9eb25fSBarry Smith   PetscFunctionReturn(0);
18175c9eb25fSBarry Smith }
18185c9eb25fSBarry Smith 
1819db4efbfdSBarry Smith #undef __FUNCT__
1820db4efbfdSBarry Smith #define __FUNCT__ "MatGetFactorAvailable_seqsbaij_petsc"
1821ace3abfcSBarry Smith PetscErrorCode MatGetFactorAvailable_seqsbaij_petsc(Mat A,MatFactorType ftype,PetscBool  *flg)
1822db4efbfdSBarry Smith {
1823db4efbfdSBarry Smith   PetscFunctionBegin;
1824db4efbfdSBarry Smith   if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) {
1825db4efbfdSBarry Smith     *flg = PETSC_TRUE;
1826db4efbfdSBarry Smith   } else {
1827db4efbfdSBarry Smith     *flg = PETSC_FALSE;
1828db4efbfdSBarry Smith   }
1829db4efbfdSBarry Smith   PetscFunctionReturn(0);
1830db4efbfdSBarry Smith }
1831db4efbfdSBarry Smith 
1832611f576cSBarry Smith #if defined(PETSC_HAVE_MUMPS)
18338cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_sbaij_mumps(Mat,MatFactorType,Mat*);
1834611f576cSBarry Smith #endif
1835b5e56a35SBarry Smith #if defined(PETSC_HAVE_PASTIX)
18368cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_seqsbaij_pastix(Mat,MatFactorType,Mat*);
1837b5e56a35SBarry Smith #endif
1838913c410bSShri Abhyankar #if defined(PETSC_HAVE_SUITESPARSE)
18398cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_seqsbaij_cholmod(Mat,MatFactorType,Mat*);
184020db9a53SJed Brown #endif
18418cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_seqsbaij_sbstrm(Mat,MatFactorType,Mat*);
18425c9eb25fSBarry Smith 
18430bad9183SKris Buschelman /*MC
1844fafad747SKris Buschelman   MATSEQSBAIJ - MATSEQSBAIJ = "seqsbaij" - A matrix type to be used for sequential symmetric block sparse matrices,
18450bad9183SKris Buschelman   based on block compressed sparse row format.  Only the upper triangular portion of the matrix is stored.
18460bad9183SKris Buschelman 
1847828413b8SBarry Smith   For complex numbers by default this matrix is symmetric, NOT Hermitian symmetric. To make it Hermitian symmetric you
184871dad5bbSBarry Smith   can call MatSetOption(Mat, MAT_HERMITIAN); after MatAssemblyEnd()
1849828413b8SBarry Smith 
18500bad9183SKris Buschelman   Options Database Keys:
18510bad9183SKris Buschelman   . -mat_type seqsbaij - sets the matrix type to "seqsbaij" during a call to MatSetFromOptions()
18520bad9183SKris Buschelman 
185371dad5bbSBarry Smith   Notes: By default if you insert values into the lower triangular part of the matrix they are simply ignored (since they are not
185471dad5bbSBarry 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
185571dad5bbSBarry 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.
185671dad5bbSBarry Smith 
185771dad5bbSBarry Smith 
18580bad9183SKris Buschelman   Level: beginner
18590bad9183SKris Buschelman 
18600bad9183SKris Buschelman   .seealso: MatCreateSeqSBAIJ
18610bad9183SKris Buschelman M*/
18620bad9183SKris Buschelman 
18638cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqSBSTRM(Mat, MatType,MatReuse,Mat*);
1864aa5a9175SDahai Guo 
1865a23d5eceSKris Buschelman #undef __FUNCT__
1866a23d5eceSKris Buschelman #define __FUNCT__ "MatCreate_SeqSBAIJ"
18678cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatCreate_SeqSBAIJ(Mat B)
1868a23d5eceSKris Buschelman {
1869a23d5eceSKris Buschelman   Mat_SeqSBAIJ   *b;
1870dfbe8321SBarry Smith   PetscErrorCode ierr;
187113f74950SBarry Smith   PetscMPIInt    size;
1872ace3abfcSBarry Smith   PetscBool      no_unroll = PETSC_FALSE,no_inode = PETSC_FALSE;
1873a23d5eceSKris Buschelman 
1874a23d5eceSKris Buschelman   PetscFunctionBegin;
1875ce94432eSBarry Smith   ierr = MPI_Comm_size(PetscObjectComm((PetscObject)B),&size);CHKERRQ(ierr);
1876e32f2f54SBarry Smith   if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Comm must be of size 1");
1877a23d5eceSKris Buschelman 
1878b00a9115SJed Brown   ierr    = PetscNewLog(B,&b);CHKERRQ(ierr);
1879a23d5eceSKris Buschelman   B->data = (void*)b;
1880a23d5eceSKris Buschelman   ierr    = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr);
188126fbe8dcSKarl Rupp 
1882a23d5eceSKris Buschelman   B->ops->destroy    = MatDestroy_SeqSBAIJ;
1883a23d5eceSKris Buschelman   B->ops->view       = MatView_SeqSBAIJ;
1884a23d5eceSKris Buschelman   b->row             = 0;
1885a23d5eceSKris Buschelman   b->icol            = 0;
1886a23d5eceSKris Buschelman   b->reallocs        = 0;
1887a23d5eceSKris Buschelman   b->saved_values    = 0;
18880def2e27SBarry Smith   b->inode.limit     = 5;
18890def2e27SBarry Smith   b->inode.max_limit = 5;
1890a23d5eceSKris Buschelman 
1891a23d5eceSKris Buschelman   b->roworiented        = PETSC_TRUE;
1892a23d5eceSKris Buschelman   b->nonew              = 0;
1893a23d5eceSKris Buschelman   b->diag               = 0;
1894a23d5eceSKris Buschelman   b->solve_work         = 0;
1895a23d5eceSKris Buschelman   b->mult_work          = 0;
1896a23d5eceSKris Buschelman   B->spptr              = 0;
1897f2cbd3d5SJed Brown   B->info.nz_unneeded   = (PetscReal)b->maxnz*b->bs2;
1898a9817697SBarry Smith   b->keepnonzeropattern = PETSC_FALSE;
1899a23d5eceSKris Buschelman 
1900a23d5eceSKris Buschelman   b->inew    = 0;
1901a23d5eceSKris Buschelman   b->jnew    = 0;
1902a23d5eceSKris Buschelman   b->anew    = 0;
1903a23d5eceSKris Buschelman   b->a2anew  = 0;
1904a23d5eceSKris Buschelman   b->permute = PETSC_FALSE;
1905a23d5eceSKris Buschelman 
190671dad5bbSBarry Smith   b->ignore_ltriangular = PETSC_TRUE;
190726fbe8dcSKarl Rupp 
19080298fd71SBarry Smith   ierr = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_ignore_lower_triangular",&b->ignore_ltriangular,NULL);CHKERRQ(ierr);
1909941593c8SHong Zhang 
1910f5edf698SHong Zhang   b->getrow_utriangular = PETSC_FALSE;
191126fbe8dcSKarl Rupp 
19120298fd71SBarry Smith   ierr = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_getrow_uppertriangular",&b->getrow_utriangular,NULL);CHKERRQ(ierr);
1913f5edf698SHong Zhang 
1914b5e56a35SBarry Smith #if defined(PETSC_HAVE_PASTIX)
1915bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactor_pastix_C",MatGetFactor_seqsbaij_pastix);CHKERRQ(ierr);
1916b5e56a35SBarry Smith #endif
1917611f576cSBarry Smith #if defined(PETSC_HAVE_MUMPS)
1918bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactor_mumps_C",MatGetFactor_sbaij_mumps);CHKERRQ(ierr);
1919611f576cSBarry Smith #endif
1920913c410bSShri Abhyankar #if defined(PETSC_HAVE_SUITESPARSE)
1921bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactor_cholmod_C",MatGetFactor_seqsbaij_cholmod);CHKERRQ(ierr);
192220db9a53SJed Brown #endif
1923bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactorAvailable_petsc_C",MatGetFactorAvailable_seqsbaij_petsc);CHKERRQ(ierr);
1924bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactor_petsc_C",MatGetFactor_seqsbaij_petsc);CHKERRQ(ierr);
1925bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactor_sbstrm_C",MatGetFactor_seqsbaij_sbstrm);CHKERRQ(ierr);
1926bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatStoreValues_C",MatStoreValues_SeqSBAIJ);CHKERRQ(ierr);
1927bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatRetrieveValues_C",MatRetrieveValues_SeqSBAIJ);CHKERRQ(ierr);
1928bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetColumnIndices_C",MatSeqSBAIJSetColumnIndices_SeqSBAIJ);CHKERRQ(ierr);
1929bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqaij_C",MatConvert_SeqSBAIJ_SeqAIJ);CHKERRQ(ierr);
1930bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqbaij_C",MatConvert_SeqSBAIJ_SeqBAIJ);CHKERRQ(ierr);
1931bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetPreallocation_C",MatSeqSBAIJSetPreallocation_SeqSBAIJ);CHKERRQ(ierr);
193238f409ebSLisandro Dalcin   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqSBAIJSetPreallocationCSR_C",MatSeqSBAIJSetPreallocationCSR_SeqSBAIJ);CHKERRQ(ierr);
1933bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqsbaij_seqsbstrm_C",MatConvert_SeqSBAIJ_SeqSBSTRM);CHKERRQ(ierr);
193423ce1328SBarry Smith 
193523ce1328SBarry Smith   B->symmetric                  = PETSC_TRUE;
193623ce1328SBarry Smith   B->structurally_symmetric     = PETSC_TRUE;
193723ce1328SBarry Smith   B->symmetric_set              = PETSC_TRUE;
193823ce1328SBarry Smith   B->structurally_symmetric_set = PETSC_TRUE;
193926fbe8dcSKarl Rupp 
194017667f90SBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)B,MATSEQSBAIJ);CHKERRQ(ierr);
19410def2e27SBarry Smith 
1942ce94432eSBarry Smith   ierr = PetscOptionsBegin(PetscObjectComm((PetscObject)B),((PetscObject)B)->prefix,"Options for SEQSBAIJ matrix","Mat");CHKERRQ(ierr);
19430298fd71SBarry Smith   ierr = PetscOptionsBool("-mat_no_unroll","Do not optimize for inodes (slower)",NULL,no_unroll,&no_unroll,NULL);CHKERRQ(ierr);
194426fbe8dcSKarl Rupp   if (no_unroll) {
194526fbe8dcSKarl Rupp     ierr = PetscInfo(B,"Not using Inode routines due to -mat_no_unroll\n");CHKERRQ(ierr);
194626fbe8dcSKarl Rupp   }
19470298fd71SBarry Smith   ierr = PetscOptionsBool("-mat_no_inode","Do not optimize for inodes (slower)",NULL,no_inode,&no_inode,NULL);CHKERRQ(ierr);
194826fbe8dcSKarl Rupp   if (no_inode) {
194926fbe8dcSKarl Rupp     ierr = PetscInfo(B,"Not using Inode routines due to -mat_no_inode\n");CHKERRQ(ierr);
195026fbe8dcSKarl Rupp   }
19510298fd71SBarry Smith   ierr = PetscOptionsInt("-mat_inode_limit","Do not use inodes larger then this value",NULL,b->inode.limit,&b->inode.limit,NULL);CHKERRQ(ierr);
19520def2e27SBarry Smith   ierr = PetscOptionsEnd();CHKERRQ(ierr);
1953ace3abfcSBarry Smith   b->inode.use = (PetscBool)(!(no_unroll || no_inode));
19540def2e27SBarry Smith   if (b->inode.limit > b->inode.max_limit) b->inode.limit = b->inode.max_limit;
1955a23d5eceSKris Buschelman   PetscFunctionReturn(0);
1956a23d5eceSKris Buschelman }
1957a23d5eceSKris Buschelman 
1958a23d5eceSKris Buschelman #undef __FUNCT__
1959a23d5eceSKris Buschelman #define __FUNCT__ "MatSeqSBAIJSetPreallocation"
1960a23d5eceSKris Buschelman /*@C
1961a23d5eceSKris Buschelman    MatSeqSBAIJSetPreallocation - Creates a sparse symmetric matrix in block AIJ (block
1962a23d5eceSKris Buschelman    compressed row) format.  For good matrix assembly performance the
1963a23d5eceSKris Buschelman    user should preallocate the matrix storage by setting the parameter nz
1964a23d5eceSKris Buschelman    (or the array nnz).  By setting these parameters accurately, performance
1965a23d5eceSKris Buschelman    during matrix assembly can be increased by more than a factor of 50.
1966a23d5eceSKris Buschelman 
1967a23d5eceSKris Buschelman    Collective on Mat
1968a23d5eceSKris Buschelman 
1969a23d5eceSKris Buschelman    Input Parameters:
19701c4f3114SJed Brown +  B - the symmetric matrix
1971bb7ae925SBarry Smith .  bs - size of block, the blocks are ALWAYS square. One can use MatSetBlockSizes() to set a different row and column blocksize but the row
1972bb7ae925SBarry Smith           blocksize always defines the size of the blocks. The column blocksize sets the blocksize of the vectors obtained with MatCreateVecs()
1973a23d5eceSKris Buschelman .  nz - number of block nonzeros per block row (same for all rows)
1974a23d5eceSKris Buschelman -  nnz - array containing the number of block nonzeros in the upper triangular plus
19750298fd71SBarry Smith          diagonal portion of each block (possibly different for each block row) or NULL
1976a23d5eceSKris Buschelman 
1977a23d5eceSKris Buschelman    Options Database Keys:
1978a23d5eceSKris Buschelman .   -mat_no_unroll - uses code that does not unroll the loops in the
1979a23d5eceSKris Buschelman                      block calculations (much slower)
1980db4efbfdSBarry Smith .    -mat_block_size - size of the blocks to use (only works if a negative bs is passed in
1981a23d5eceSKris Buschelman 
1982a23d5eceSKris Buschelman    Level: intermediate
1983a23d5eceSKris Buschelman 
1984a23d5eceSKris Buschelman    Notes:
1985a23d5eceSKris Buschelman    Specify the preallocated storage with either nz or nnz (not both).
19860298fd71SBarry Smith    Set nz=PETSC_DEFAULT and nnz=NULL for PETSc to control dynamic memory
1987a7f22e61SSatish Balay    allocation.  See Users-Manual: ch_mat for details.
1988a23d5eceSKris Buschelman 
1989aa95bbe8SBarry Smith    You can call MatGetInfo() to get information on how effective the preallocation was;
1990aa95bbe8SBarry Smith    for example the fields mallocs,nz_allocated,nz_used,nz_unneeded;
1991aa95bbe8SBarry Smith    You can also run with the option -info and look for messages with the string
1992aa95bbe8SBarry Smith    malloc in them to see if additional memory allocation was needed.
1993aa95bbe8SBarry Smith 
199449a6f317SBarry Smith    If the nnz parameter is given then the nz parameter is ignored
199549a6f317SBarry Smith 
199649a6f317SBarry Smith 
199769b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateSBAIJ()
1998a23d5eceSKris Buschelman @*/
19997087cfbeSBarry Smith PetscErrorCode  MatSeqSBAIJSetPreallocation(Mat B,PetscInt bs,PetscInt nz,const PetscInt nnz[])
200013f74950SBarry Smith {
20014ac538c5SBarry Smith   PetscErrorCode ierr;
2002a23d5eceSKris Buschelman 
2003a23d5eceSKris Buschelman   PetscFunctionBegin;
20046ba663aaSJed Brown   PetscValidHeaderSpecific(B,MAT_CLASSID,1);
20056ba663aaSJed Brown   PetscValidType(B,1);
20066ba663aaSJed Brown   PetscValidLogicalCollectiveInt(B,bs,2);
20074ac538c5SBarry Smith   ierr = PetscTryMethod(B,"MatSeqSBAIJSetPreallocation_C",(Mat,PetscInt,PetscInt,const PetscInt[]),(B,bs,nz,nnz));CHKERRQ(ierr);
2008a23d5eceSKris Buschelman   PetscFunctionReturn(0);
2009a23d5eceSKris Buschelman }
201049b5e25fSSatish Balay 
20114a2ae208SSatish Balay #undef  __FUNCT__
201238f409ebSLisandro Dalcin #define __FUNCT__ "MatSeqSBAIJSetPreallocationCSR"
201338f409ebSLisandro Dalcin /*@C
201438f409ebSLisandro Dalcin    MatSeqSBAIJSetPreallocationCSR - Allocates memory for a sparse sequential matrix in symmetric block AIJ format.
201538f409ebSLisandro Dalcin 
201638f409ebSLisandro Dalcin    Input Parameters:
20171c4f3114SJed Brown +  B - the matrix
201838f409ebSLisandro Dalcin .  i - the indices into j for the start of each local row (starts with zero)
201938f409ebSLisandro Dalcin .  j - the column indices for each local row (starts with zero) these must be sorted for each row
202038f409ebSLisandro Dalcin -  v - optional values in the matrix
202138f409ebSLisandro Dalcin 
202238f409ebSLisandro Dalcin    Level: developer
202338f409ebSLisandro Dalcin 
202438f409ebSLisandro Dalcin    Notes:
202538f409ebSLisandro Dalcin    The order of the entries in values is specified by the MatOption MAT_ROW_ORIENTED.  For example, C programs
202638f409ebSLisandro 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
202738f409ebSLisandro Dalcin    over rows within a block and the last index is over columns within a block row.  Fortran programs will likely set
202838f409ebSLisandro 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
202938f409ebSLisandro Dalcin    block column and the second index is over columns within a block.
203038f409ebSLisandro Dalcin 
203138f409ebSLisandro Dalcin .keywords: matrix, block, aij, compressed row, sparse
203238f409ebSLisandro Dalcin 
203338f409ebSLisandro Dalcin .seealso: MatCreate(), MatCreateSeqSBAIJ(), MatSetValuesBlocked(), MatSeqSBAIJSetPreallocation(), MATSEQSBAIJ
203438f409ebSLisandro Dalcin @*/
203538f409ebSLisandro Dalcin PetscErrorCode MatSeqSBAIJSetPreallocationCSR(Mat B,PetscInt bs,const PetscInt i[],const PetscInt j[], const PetscScalar v[])
203638f409ebSLisandro Dalcin {
203738f409ebSLisandro Dalcin   PetscErrorCode ierr;
203838f409ebSLisandro Dalcin 
203938f409ebSLisandro Dalcin   PetscFunctionBegin;
204038f409ebSLisandro Dalcin   PetscValidHeaderSpecific(B,MAT_CLASSID,1);
204138f409ebSLisandro Dalcin   PetscValidType(B,1);
204238f409ebSLisandro Dalcin   PetscValidLogicalCollectiveInt(B,bs,2);
204338f409ebSLisandro Dalcin   ierr = PetscTryMethod(B,"MatSeqSBAIJSetPreallocationCSR_C",(Mat,PetscInt,const PetscInt[],const PetscInt[],const PetscScalar[]),(B,bs,i,j,v));CHKERRQ(ierr);
204438f409ebSLisandro Dalcin   PetscFunctionReturn(0);
204538f409ebSLisandro Dalcin }
204638f409ebSLisandro Dalcin 
204738f409ebSLisandro Dalcin #undef __FUNCT__
20484a2ae208SSatish Balay #define __FUNCT__ "MatCreateSeqSBAIJ"
2049c464158bSHong Zhang /*@C
2050c464158bSHong Zhang    MatCreateSeqSBAIJ - Creates a sparse symmetric matrix in block AIJ (block
2051c464158bSHong Zhang    compressed row) format.  For good matrix assembly performance the
2052c464158bSHong Zhang    user should preallocate the matrix storage by setting the parameter nz
2053c464158bSHong Zhang    (or the array nnz).  By setting these parameters accurately, performance
2054c464158bSHong Zhang    during matrix assembly can be increased by more than a factor of 50.
205549b5e25fSSatish Balay 
2056c464158bSHong Zhang    Collective on MPI_Comm
2057c464158bSHong Zhang 
2058c464158bSHong Zhang    Input Parameters:
2059c464158bSHong Zhang +  comm - MPI communicator, set to PETSC_COMM_SELF
2060bb7ae925SBarry Smith .  bs - size of block, the blocks are ALWAYS square. One can use MatSetBlockSizes() to set a different row and column blocksize but the row
2061bb7ae925SBarry Smith           blocksize always defines the size of the blocks. The column blocksize sets the blocksize of the vectors obtained with MatCreateVecs()
2062c464158bSHong Zhang .  m - number of rows, or number of columns
2063c464158bSHong Zhang .  nz - number of block nonzeros per block row (same for all rows)
2064744e8345SSatish Balay -  nnz - array containing the number of block nonzeros in the upper triangular plus
20650298fd71SBarry Smith          diagonal portion of each block (possibly different for each block row) or NULL
2066c464158bSHong Zhang 
2067c464158bSHong Zhang    Output Parameter:
2068c464158bSHong Zhang .  A - the symmetric matrix
2069c464158bSHong Zhang 
2070c464158bSHong Zhang    Options Database Keys:
2071c464158bSHong Zhang .   -mat_no_unroll - uses code that does not unroll the loops in the
2072c464158bSHong Zhang                      block calculations (much slower)
2073c464158bSHong Zhang .    -mat_block_size - size of the blocks to use
2074c464158bSHong Zhang 
2075c464158bSHong Zhang    Level: intermediate
2076c464158bSHong Zhang 
2077175b88e8SBarry Smith    It is recommended that one use the MatCreate(), MatSetType() and/or MatSetFromOptions(),
2078ae1d86c5SBarry Smith    MatXXXXSetPreallocation() paradgm instead of this routine directly.
2079175b88e8SBarry Smith    [MatXXXXSetPreallocation() is, for example, MatSeqAIJSetPreallocation]
2080175b88e8SBarry Smith 
2081c464158bSHong Zhang    Notes:
20826d6d819aSHong Zhang    The number of rows and columns must be divisible by blocksize.
20836d6d819aSHong Zhang    This matrix type does not support complex Hermitian operation.
2084c464158bSHong Zhang 
2085c464158bSHong Zhang    Specify the preallocated storage with either nz or nnz (not both).
20860298fd71SBarry Smith    Set nz=PETSC_DEFAULT and nnz=NULL for PETSc to control dynamic memory
2087a7f22e61SSatish Balay    allocation.  See Users-Manual: ch_mat for details.
2088c464158bSHong Zhang 
208949a6f317SBarry Smith    If the nnz parameter is given then the nz parameter is ignored
209049a6f317SBarry Smith 
209169b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateSBAIJ()
2092c464158bSHong Zhang @*/
20937087cfbeSBarry Smith PetscErrorCode  MatCreateSeqSBAIJ(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt nz,const PetscInt nnz[],Mat *A)
2094c464158bSHong Zhang {
2095dfbe8321SBarry Smith   PetscErrorCode ierr;
2096c464158bSHong Zhang 
2097c464158bSHong Zhang   PetscFunctionBegin;
2098f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,A);CHKERRQ(ierr);
2099f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*A,m,n,m,n);CHKERRQ(ierr);
2100c464158bSHong Zhang   ierr = MatSetType(*A,MATSEQSBAIJ);CHKERRQ(ierr);
2101ab93d7beSBarry Smith   ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(*A,bs,nz,(PetscInt*)nnz);CHKERRQ(ierr);
210249b5e25fSSatish Balay   PetscFunctionReturn(0);
210349b5e25fSSatish Balay }
210449b5e25fSSatish Balay 
21054a2ae208SSatish Balay #undef __FUNCT__
21064a2ae208SSatish Balay #define __FUNCT__ "MatDuplicate_SeqSBAIJ"
2107dfbe8321SBarry Smith PetscErrorCode MatDuplicate_SeqSBAIJ(Mat A,MatDuplicateOption cpvalues,Mat *B)
210849b5e25fSSatish Balay {
210949b5e25fSSatish Balay   Mat            C;
211049b5e25fSSatish Balay   Mat_SeqSBAIJ   *c,*a = (Mat_SeqSBAIJ*)A->data;
21116849ba73SBarry Smith   PetscErrorCode ierr;
2112b40805acSSatish Balay   PetscInt       i,mbs = a->mbs,nz = a->nz,bs2 =a->bs2;
211349b5e25fSSatish Balay 
211449b5e25fSSatish Balay   PetscFunctionBegin;
2115e32f2f54SBarry Smith   if (a->i[mbs] != nz) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Corrupt matrix");
211649b5e25fSSatish Balay 
211749b5e25fSSatish Balay   *B   = 0;
2118ce94432eSBarry Smith   ierr = MatCreate(PetscObjectComm((PetscObject)A),&C);CHKERRQ(ierr);
2119d0f46423SBarry Smith   ierr = MatSetSizes(C,A->rmap->N,A->cmap->n,A->rmap->N,A->cmap->n);CHKERRQ(ierr);
21208e9a0fb8SHong Zhang   ierr = MatSetType(C,MATSEQSBAIJ);CHKERRQ(ierr);
21211d5dac46SHong Zhang   ierr = PetscMemcpy(C->ops,A->ops,sizeof(struct _MatOps));CHKERRQ(ierr);
2122692f9cbeSHong Zhang   c    = (Mat_SeqSBAIJ*)C->data;
2123692f9cbeSHong Zhang 
2124273d9f13SBarry Smith   C->preallocated       = PETSC_TRUE;
2125d5f3da31SBarry Smith   C->factortype         = A->factortype;
212649b5e25fSSatish Balay   c->row                = 0;
212749b5e25fSSatish Balay   c->icol               = 0;
212849b5e25fSSatish Balay   c->saved_values       = 0;
2129a9817697SBarry Smith   c->keepnonzeropattern = a->keepnonzeropattern;
213049b5e25fSSatish Balay   C->assembled          = PETSC_TRUE;
213149b5e25fSSatish Balay 
21321e1e43feSBarry Smith   ierr   = PetscLayoutReference(A->rmap,&C->rmap);CHKERRQ(ierr);
21331e1e43feSBarry Smith   ierr   = PetscLayoutReference(A->cmap,&C->cmap);CHKERRQ(ierr);
213449b5e25fSSatish Balay   c->bs2 = a->bs2;
213549b5e25fSSatish Balay   c->mbs = a->mbs;
213649b5e25fSSatish Balay   c->nbs = a->nbs;
213749b5e25fSSatish Balay 
2138c760cd28SBarry Smith   if  (cpvalues == MAT_SHARE_NONZERO_PATTERN) {
2139c760cd28SBarry Smith     c->imax           = a->imax;
2140c760cd28SBarry Smith     c->ilen           = a->ilen;
2141c760cd28SBarry Smith     c->free_imax_ilen = PETSC_FALSE;
2142c760cd28SBarry Smith   } else {
2143dcca6d9dSJed Brown     ierr = PetscMalloc2((mbs+1),&c->imax,(mbs+1),&c->ilen);CHKERRQ(ierr);
21443bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)C,2*(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr);
214549b5e25fSSatish Balay     for (i=0; i<mbs; i++) {
214649b5e25fSSatish Balay       c->imax[i] = a->imax[i];
214749b5e25fSSatish Balay       c->ilen[i] = a->ilen[i];
214849b5e25fSSatish Balay     }
2149c760cd28SBarry Smith     c->free_imax_ilen = PETSC_TRUE;
2150c760cd28SBarry Smith   }
215149b5e25fSSatish Balay 
215249b5e25fSSatish Balay   /* allocate the matrix space */
21534da8f245SBarry Smith   if (cpvalues == MAT_SHARE_NONZERO_PATTERN) {
2154785e854fSJed Brown     ierr            = PetscMalloc1(bs2*nz,&c->a);CHKERRQ(ierr);
21553bb1ff40SBarry Smith     ierr            = PetscLogObjectMemory((PetscObject)C,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr);
215644e1c64aSLisandro Dalcin     c->i            = a->i;
215744e1c64aSLisandro Dalcin     c->j            = a->j;
21584da8f245SBarry Smith     c->singlemalloc = PETSC_FALSE;
215944e1c64aSLisandro Dalcin     c->free_a       = PETSC_TRUE;
21604da8f245SBarry Smith     c->free_ij      = PETSC_FALSE;
21614da8f245SBarry Smith     c->parent       = A;
21624da8f245SBarry Smith     ierr            = PetscObjectReference((PetscObject)A);CHKERRQ(ierr);
21634da8f245SBarry Smith     ierr            = MatSetOption(A,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
21644da8f245SBarry Smith     ierr            = MatSetOption(C,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);
21654da8f245SBarry Smith   } else {
2166dcca6d9dSJed Brown     ierr            = PetscMalloc3(bs2*nz,&c->a,nz,&c->j,mbs+1,&c->i);CHKERRQ(ierr);
216713f74950SBarry Smith     ierr            = PetscMemcpy(c->i,a->i,(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr);
21683bb1ff40SBarry Smith     ierr            = PetscLogObjectMemory((PetscObject)C,(mbs+1)*sizeof(PetscInt) + nz*(bs2*sizeof(MatScalar) + sizeof(PetscInt)));CHKERRQ(ierr);
21694da8f245SBarry Smith     c->singlemalloc = PETSC_TRUE;
217044e1c64aSLisandro Dalcin     c->free_a       = PETSC_TRUE;
21714da8f245SBarry Smith     c->free_ij      = PETSC_TRUE;
21724da8f245SBarry Smith   }
217349b5e25fSSatish Balay   if (mbs > 0) {
21744da8f245SBarry Smith     if (cpvalues != MAT_SHARE_NONZERO_PATTERN) {
217513f74950SBarry Smith       ierr = PetscMemcpy(c->j,a->j,nz*sizeof(PetscInt));CHKERRQ(ierr);
21764da8f245SBarry Smith     }
217749b5e25fSSatish Balay     if (cpvalues == MAT_COPY_VALUES) {
217849b5e25fSSatish Balay       ierr = PetscMemcpy(c->a,a->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr);
217949b5e25fSSatish Balay     } else {
218049b5e25fSSatish Balay       ierr = PetscMemzero(c->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr);
218149b5e25fSSatish Balay     }
2182a1c3900fSBarry Smith     if (a->jshort) {
218344e1c64aSLisandro Dalcin       /* cannot share jshort, it is reallocated in MatAssemblyEnd_SeqSBAIJ() */
218444e1c64aSLisandro Dalcin       /* if the parent matrix is reassembled, this child matrix will never notice */
2185785e854fSJed Brown       ierr = PetscMalloc1(nz,&c->jshort);CHKERRQ(ierr);
21863bb1ff40SBarry Smith       ierr = PetscLogObjectMemory((PetscObject)C,nz*sizeof(unsigned short));CHKERRQ(ierr);
2187a1c3900fSBarry Smith       ierr = PetscMemcpy(c->jshort,a->jshort,nz*sizeof(unsigned short));CHKERRQ(ierr);
218826fbe8dcSKarl Rupp 
21894da8f245SBarry Smith       c->free_jshort = PETSC_TRUE;
21904da8f245SBarry Smith     }
2191a1c3900fSBarry Smith   }
219249b5e25fSSatish Balay 
219349b5e25fSSatish Balay   c->roworiented = a->roworiented;
219449b5e25fSSatish Balay   c->nonew       = a->nonew;
219549b5e25fSSatish Balay 
219649b5e25fSSatish Balay   if (a->diag) {
2197c760cd28SBarry Smith     if (cpvalues == MAT_SHARE_NONZERO_PATTERN) {
2198c760cd28SBarry Smith       c->diag      = a->diag;
2199c760cd28SBarry Smith       c->free_diag = PETSC_FALSE;
2200c760cd28SBarry Smith     } else {
2201785e854fSJed Brown       ierr = PetscMalloc1(mbs,&c->diag);CHKERRQ(ierr);
22023bb1ff40SBarry Smith       ierr = PetscLogObjectMemory((PetscObject)C,mbs*sizeof(PetscInt));CHKERRQ(ierr);
220326fbe8dcSKarl Rupp       for (i=0; i<mbs; i++) c->diag[i] = a->diag[i];
2204c760cd28SBarry Smith       c->free_diag = PETSC_TRUE;
2205c760cd28SBarry Smith     }
220644e1c64aSLisandro Dalcin   }
22076c6c5352SBarry Smith   c->nz         = a->nz;
2208f2cbd3d5SJed Brown   c->maxnz      = a->nz; /* Since we allocate exactly the right amount */
220949b5e25fSSatish Balay   c->solve_work = 0;
221049b5e25fSSatish Balay   c->mult_work  = 0;
221126fbe8dcSKarl Rupp 
221249b5e25fSSatish Balay   *B   = C;
2213140e18c1SBarry Smith   ierr = PetscFunctionListDuplicate(((PetscObject)A)->qlist,&((PetscObject)C)->qlist);CHKERRQ(ierr);
221449b5e25fSSatish Balay   PetscFunctionReturn(0);
221549b5e25fSSatish Balay }
221649b5e25fSSatish Balay 
22174a2ae208SSatish Balay #undef __FUNCT__
22185bba2384SShri Abhyankar #define __FUNCT__ "MatLoad_SeqSBAIJ"
2219112444f4SShri Abhyankar PetscErrorCode MatLoad_SeqSBAIJ(Mat newmat,PetscViewer viewer)
22202f480046SShri Abhyankar {
22212f480046SShri Abhyankar   Mat_SeqSBAIJ   *a;
22222f480046SShri Abhyankar   PetscErrorCode ierr;
22232f480046SShri Abhyankar   int            fd;
22242f480046SShri Abhyankar   PetscMPIInt    size;
22252f480046SShri Abhyankar   PetscInt       i,nz,header[4],*rowlengths=0,M,N,bs=1;
22262f480046SShri Abhyankar   PetscInt       *mask,mbs,*jj,j,rowcount,nzcount,k,*s_browlengths,maskcount;
22272f480046SShri Abhyankar   PetscInt       kmax,jcount,block,idx,point,nzcountb,extra_rows,rows,cols;
22282f480046SShri Abhyankar   PetscInt       *masked,nmask,tmp,bs2,ishift;
22292f480046SShri Abhyankar   PetscScalar    *aa;
2230ce94432eSBarry Smith   MPI_Comm       comm;
22312f480046SShri Abhyankar 
22322f480046SShri Abhyankar   PetscFunctionBegin;
2233ce94432eSBarry Smith   ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr);
22340298fd71SBarry Smith   ierr = PetscOptionsGetInt(((PetscObject)newmat)->prefix,"-matload_block_size",&bs,NULL);CHKERRQ(ierr);
22352f480046SShri Abhyankar   bs2  = bs*bs;
22362f480046SShri Abhyankar 
22372f480046SShri Abhyankar   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
22382f480046SShri Abhyankar   if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"view must have one processor");
22392f480046SShri Abhyankar   ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr);
22402f480046SShri Abhyankar   ierr = PetscBinaryRead(fd,header,4,PETSC_INT);CHKERRQ(ierr);
22412f480046SShri Abhyankar   if (header[0] != MAT_FILE_CLASSID) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"not Mat object");
22422f480046SShri Abhyankar   M = header[1]; N = header[2]; nz = header[3];
22432f480046SShri Abhyankar 
22442f480046SShri Abhyankar   if (header[3] < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Matrix stored in special format, cannot load as SeqSBAIJ");
22452f480046SShri Abhyankar 
22462f480046SShri Abhyankar   if (M != N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Can only do square matrices");
22472f480046SShri Abhyankar 
22482f480046SShri Abhyankar   /*
22492f480046SShri Abhyankar      This code adds extra rows to make sure the number of rows is
22502f480046SShri Abhyankar     divisible by the blocksize
22512f480046SShri Abhyankar   */
22522f480046SShri Abhyankar   mbs        = M/bs;
22532f480046SShri Abhyankar   extra_rows = bs - M + bs*(mbs);
22542f480046SShri Abhyankar   if (extra_rows == bs) extra_rows = 0;
22552f480046SShri Abhyankar   else                  mbs++;
22562f480046SShri Abhyankar   if (extra_rows) {
22572f480046SShri Abhyankar     ierr = PetscInfo(viewer,"Padding loaded matrix to match blocksize\n");CHKERRQ(ierr);
22582f480046SShri Abhyankar   }
22592f480046SShri Abhyankar 
22602f480046SShri Abhyankar   /* Set global sizes if not already set */
22612f480046SShri Abhyankar   if (newmat->rmap->n < 0 && newmat->rmap->N < 0 && newmat->cmap->n < 0 && newmat->cmap->N < 0) {
22622f480046SShri Abhyankar     ierr = MatSetSizes(newmat,PETSC_DECIDE,PETSC_DECIDE,M+extra_rows,N+extra_rows);CHKERRQ(ierr);
22632f480046SShri Abhyankar   } else { /* Check if the matrix global sizes are correct */
22642f480046SShri Abhyankar     ierr = MatGetSize(newmat,&rows,&cols);CHKERRQ(ierr);
22652f480046SShri 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);
22662f480046SShri Abhyankar   }
22672f480046SShri Abhyankar 
22682f480046SShri Abhyankar   /* read in row lengths */
2269785e854fSJed Brown   ierr = PetscMalloc1((M+extra_rows),&rowlengths);CHKERRQ(ierr);
22702f480046SShri Abhyankar   ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr);
22712f480046SShri Abhyankar   for (i=0; i<extra_rows; i++) rowlengths[M+i] = 1;
22722f480046SShri Abhyankar 
22732f480046SShri Abhyankar   /* read in column indices */
2274785e854fSJed Brown   ierr = PetscMalloc1((nz+extra_rows),&jj);CHKERRQ(ierr);
22752f480046SShri Abhyankar   ierr = PetscBinaryRead(fd,jj,nz,PETSC_INT);CHKERRQ(ierr);
22762f480046SShri Abhyankar   for (i=0; i<extra_rows; i++) jj[nz+i] = M+i;
22772f480046SShri Abhyankar 
22782f480046SShri Abhyankar   /* loop over row lengths determining block row lengths */
22791795a4d1SJed Brown   ierr     = PetscCalloc1(mbs,&s_browlengths);CHKERRQ(ierr);
2280dcca6d9dSJed Brown   ierr     = PetscMalloc2(mbs,&mask,mbs,&masked);CHKERRQ(ierr);
22812f480046SShri Abhyankar   ierr     = PetscMemzero(mask,mbs*sizeof(PetscInt));CHKERRQ(ierr);
22822f480046SShri Abhyankar   rowcount = 0;
22832f480046SShri Abhyankar   nzcount  = 0;
22842f480046SShri Abhyankar   for (i=0; i<mbs; i++) {
22852f480046SShri Abhyankar     nmask = 0;
22862f480046SShri Abhyankar     for (j=0; j<bs; j++) {
22872f480046SShri Abhyankar       kmax = rowlengths[rowcount];
22882f480046SShri Abhyankar       for (k=0; k<kmax; k++) {
22892f480046SShri Abhyankar         tmp = jj[nzcount++]/bs;   /* block col. index */
22902f480046SShri Abhyankar         if (!mask[tmp] && tmp >= i) {masked[nmask++] = tmp; mask[tmp] = 1;}
22912f480046SShri Abhyankar       }
22922f480046SShri Abhyankar       rowcount++;
22932f480046SShri Abhyankar     }
22942f480046SShri Abhyankar     s_browlengths[i] += nmask;
22952f480046SShri Abhyankar 
22962f480046SShri Abhyankar     /* zero out the mask elements we set */
22972f480046SShri Abhyankar     for (j=0; j<nmask; j++) mask[masked[j]] = 0;
22982f480046SShri Abhyankar   }
22992f480046SShri Abhyankar 
23002f480046SShri Abhyankar   /* Do preallocation */
23012f480046SShri Abhyankar   ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(newmat,bs,0,s_browlengths);CHKERRQ(ierr);
23022f480046SShri Abhyankar   a    = (Mat_SeqSBAIJ*)newmat->data;
23032f480046SShri Abhyankar 
23042f480046SShri Abhyankar   /* set matrix "i" values */
23052f480046SShri Abhyankar   a->i[0] = 0;
23062f480046SShri Abhyankar   for (i=1; i<= mbs; i++) {
23072f480046SShri Abhyankar     a->i[i]      = a->i[i-1] + s_browlengths[i-1];
23082f480046SShri Abhyankar     a->ilen[i-1] = s_browlengths[i-1];
23092f480046SShri Abhyankar   }
23102f480046SShri Abhyankar   a->nz = a->i[mbs];
23112f480046SShri Abhyankar 
23122f480046SShri Abhyankar   /* read in nonzero values */
2313785e854fSJed Brown   ierr = PetscMalloc1((nz+extra_rows),&aa);CHKERRQ(ierr);
23142f480046SShri Abhyankar   ierr = PetscBinaryRead(fd,aa,nz,PETSC_SCALAR);CHKERRQ(ierr);
23152f480046SShri Abhyankar   for (i=0; i<extra_rows; i++) aa[nz+i] = 1.0;
23162f480046SShri Abhyankar 
23172f480046SShri Abhyankar   /* set "a" and "j" values into matrix */
23182f480046SShri Abhyankar   nzcount = 0; jcount = 0;
23192f480046SShri Abhyankar   for (i=0; i<mbs; i++) {
23202f480046SShri Abhyankar     nzcountb = nzcount;
23212f480046SShri Abhyankar     nmask    = 0;
23222f480046SShri Abhyankar     for (j=0; j<bs; j++) {
23232f480046SShri Abhyankar       kmax = rowlengths[i*bs+j];
23242f480046SShri Abhyankar       for (k=0; k<kmax; k++) {
23252f480046SShri Abhyankar         tmp = jj[nzcount++]/bs; /* block col. index */
23262f480046SShri Abhyankar         if (!mask[tmp] && tmp >= i) { masked[nmask++] = tmp; mask[tmp] = 1;}
23272f480046SShri Abhyankar       }
23282f480046SShri Abhyankar     }
23292f480046SShri Abhyankar     /* sort the masked values */
23302f480046SShri Abhyankar     ierr = PetscSortInt(nmask,masked);CHKERRQ(ierr);
23312f480046SShri Abhyankar 
23322f480046SShri Abhyankar     /* set "j" values into matrix */
23332f480046SShri Abhyankar     maskcount = 1;
23342f480046SShri Abhyankar     for (j=0; j<nmask; j++) {
23352f480046SShri Abhyankar       a->j[jcount++]  = masked[j];
23362f480046SShri Abhyankar       mask[masked[j]] = maskcount++;
23372f480046SShri Abhyankar     }
23382f480046SShri Abhyankar 
23392f480046SShri Abhyankar     /* set "a" values into matrix */
23402f480046SShri Abhyankar     ishift = bs2*a->i[i];
23412f480046SShri Abhyankar     for (j=0; j<bs; j++) {
23422f480046SShri Abhyankar       kmax = rowlengths[i*bs+j];
23432f480046SShri Abhyankar       for (k=0; k<kmax; k++) {
23442f480046SShri Abhyankar         tmp = jj[nzcountb]/bs;        /* block col. index */
23452f480046SShri Abhyankar         if (tmp >= i) {
23462f480046SShri Abhyankar           block     = mask[tmp] - 1;
23472f480046SShri Abhyankar           point     = jj[nzcountb] - bs*tmp;
23482f480046SShri Abhyankar           idx       = ishift + bs2*block + j + bs*point;
23492f480046SShri Abhyankar           a->a[idx] = aa[nzcountb];
23502f480046SShri Abhyankar         }
23512f480046SShri Abhyankar         nzcountb++;
23522f480046SShri Abhyankar       }
23532f480046SShri Abhyankar     }
23542f480046SShri Abhyankar     /* zero out the mask elements we set */
23552f480046SShri Abhyankar     for (j=0; j<nmask; j++) mask[masked[j]] = 0;
23562f480046SShri Abhyankar   }
23572f480046SShri Abhyankar   if (jcount != a->nz) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Bad binary matrix");
23582f480046SShri Abhyankar 
23592f480046SShri Abhyankar   ierr = PetscFree(rowlengths);CHKERRQ(ierr);
23602f480046SShri Abhyankar   ierr = PetscFree(s_browlengths);CHKERRQ(ierr);
23612f480046SShri Abhyankar   ierr = PetscFree(aa);CHKERRQ(ierr);
23622f480046SShri Abhyankar   ierr = PetscFree(jj);CHKERRQ(ierr);
23632f480046SShri Abhyankar   ierr = PetscFree2(mask,masked);CHKERRQ(ierr);
23642f480046SShri Abhyankar 
23652f480046SShri Abhyankar   ierr = MatAssemblyBegin(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
23662f480046SShri Abhyankar   ierr = MatAssemblyEnd(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
23672f480046SShri Abhyankar   PetscFunctionReturn(0);
23682f480046SShri Abhyankar }
23692f480046SShri Abhyankar 
23702f480046SShri Abhyankar #undef __FUNCT__
2371c75a6043SHong Zhang #define __FUNCT__ "MatCreateSeqSBAIJWithArrays"
2372c75a6043SHong Zhang /*@
2373c75a6043SHong Zhang      MatCreateSeqSBAIJWithArrays - Creates an sequential SBAIJ matrix using matrix elements
2374c75a6043SHong Zhang               (upper triangular entries in CSR format) provided by the user.
2375c75a6043SHong Zhang 
2376c75a6043SHong Zhang      Collective on MPI_Comm
2377c75a6043SHong Zhang 
2378c75a6043SHong Zhang    Input Parameters:
2379c75a6043SHong Zhang +  comm - must be an MPI communicator of size 1
2380c75a6043SHong Zhang .  bs - size of block
2381c75a6043SHong Zhang .  m - number of rows
2382c75a6043SHong Zhang .  n - number of columns
2383c75a6043SHong Zhang .  i - row indices
2384c75a6043SHong Zhang .  j - column indices
2385c75a6043SHong Zhang -  a - matrix values
2386c75a6043SHong Zhang 
2387c75a6043SHong Zhang    Output Parameter:
2388c75a6043SHong Zhang .  mat - the matrix
2389c75a6043SHong Zhang 
2390dfb205c3SBarry Smith    Level: advanced
2391c75a6043SHong Zhang 
2392c75a6043SHong Zhang    Notes:
2393c75a6043SHong Zhang        The i, j, and a arrays are not copied by this routine, the user must free these arrays
2394c75a6043SHong Zhang     once the matrix is destroyed
2395c75a6043SHong Zhang 
2396c75a6043SHong Zhang        You cannot set new nonzero locations into this matrix, that will generate an error.
2397c75a6043SHong Zhang 
2398c75a6043SHong Zhang        The i and j indices are 0 based
2399c75a6043SHong Zhang 
2400dfb205c3SBarry 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
2401dfb205c3SBarry Smith        it is the regular CSR format excluding the lower triangular elements.
2402dfb205c3SBarry Smith 
240369b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSBAIJ(), MatCreateSeqSBAIJ()
2404c75a6043SHong Zhang 
2405c75a6043SHong Zhang @*/
24067087cfbeSBarry Smith PetscErrorCode  MatCreateSeqSBAIJWithArrays(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt *i,PetscInt *j,PetscScalar *a,Mat *mat)
2407c75a6043SHong Zhang {
2408c75a6043SHong Zhang   PetscErrorCode ierr;
2409c75a6043SHong Zhang   PetscInt       ii;
2410c75a6043SHong Zhang   Mat_SeqSBAIJ   *sbaij;
2411c75a6043SHong Zhang 
2412c75a6043SHong Zhang   PetscFunctionBegin;
2413e32f2f54SBarry Smith   if (bs != 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"block size %D > 1 is not supported yet",bs);
2414e32f2f54SBarry Smith   if (i[0]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"i (row indices) must start with 0");
2415c75a6043SHong Zhang 
2416c75a6043SHong Zhang   ierr  = MatCreate(comm,mat);CHKERRQ(ierr);
2417c75a6043SHong Zhang   ierr  = MatSetSizes(*mat,m,n,m,n);CHKERRQ(ierr);
2418c75a6043SHong Zhang   ierr  = MatSetType(*mat,MATSEQSBAIJ);CHKERRQ(ierr);
2419c75a6043SHong Zhang   ierr  = MatSeqSBAIJSetPreallocation_SeqSBAIJ(*mat,bs,MAT_SKIP_ALLOCATION,0);CHKERRQ(ierr);
2420c75a6043SHong Zhang   sbaij = (Mat_SeqSBAIJ*)(*mat)->data;
2421dcca6d9dSJed Brown   ierr  = PetscMalloc2(m,&sbaij->imax,m,&sbaij->ilen);CHKERRQ(ierr);
24223bb1ff40SBarry Smith   ierr  = PetscLogObjectMemory((PetscObject)*mat,2*m*sizeof(PetscInt));CHKERRQ(ierr);
2423c75a6043SHong Zhang 
2424c75a6043SHong Zhang   sbaij->i = i;
2425c75a6043SHong Zhang   sbaij->j = j;
2426c75a6043SHong Zhang   sbaij->a = a;
242726fbe8dcSKarl Rupp 
2428c75a6043SHong Zhang   sbaij->singlemalloc = PETSC_FALSE;
2429c75a6043SHong Zhang   sbaij->nonew        = -1;             /*this indicates that inserting a new value in the matrix that generates a new nonzero is an error*/
2430e6b907acSBarry Smith   sbaij->free_a       = PETSC_FALSE;
2431e6b907acSBarry Smith   sbaij->free_ij      = PETSC_FALSE;
2432c75a6043SHong Zhang 
2433c75a6043SHong Zhang   for (ii=0; ii<m; ii++) {
2434c75a6043SHong Zhang     sbaij->ilen[ii] = sbaij->imax[ii] = i[ii+1] - i[ii];
2435c75a6043SHong Zhang #if defined(PETSC_USE_DEBUG)
2436e32f2f54SBarry 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]);
2437c75a6043SHong Zhang #endif
2438c75a6043SHong Zhang   }
2439c75a6043SHong Zhang #if defined(PETSC_USE_DEBUG)
2440c75a6043SHong Zhang   for (ii=0; ii<sbaij->i[m]; ii++) {
2441e32f2f54SBarry Smith     if (j[ii] < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column index at location = %d index = %d",ii,j[ii]);
2442e32f2f54SBarry 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]);
2443c75a6043SHong Zhang   }
2444c75a6043SHong Zhang #endif
2445c75a6043SHong Zhang 
2446c75a6043SHong Zhang   ierr = MatAssemblyBegin(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2447c75a6043SHong Zhang   ierr = MatAssemblyEnd(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
2448c75a6043SHong Zhang   PetscFunctionReturn(0);
2449c75a6043SHong Zhang }
2450d06b337dSHong Zhang 
2451d06b337dSHong Zhang 
2452d06b337dSHong Zhang 
245349b5e25fSSatish Balay 
245449b5e25fSSatish Balay 
2455