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; 2513f74950SBarry Smith PetscInt *diag,*jj = a->j,i; 2649b5e25fSSatish Balay 2749b5e25fSSatish Balay PetscFunctionBegin; 28045c9aa0SHong Zhang ierr = MatMarkDiagonal_SeqSBAIJ(A);CHKERRQ(ierr); 2949b5e25fSSatish Balay diag = a->diag; 302af78befSBarry Smith *missing = PETSC_FALSE; 3149b5e25fSSatish Balay for (i=0; i<a->mbs; i++) { 322af78befSBarry Smith if (jj[diag[i]] != i) { 332af78befSBarry Smith *missing = PETSC_TRUE; 342af78befSBarry Smith if (dd) *dd = i; 352af78befSBarry Smith break; 362af78befSBarry Smith } 3749b5e25fSSatish Balay } 3849b5e25fSSatish Balay PetscFunctionReturn(0); 3949b5e25fSSatish Balay } 4049b5e25fSSatish Balay 414a2ae208SSatish Balay #undef __FUNCT__ 424a2ae208SSatish Balay #define __FUNCT__ "MatMarkDiagonal_SeqSBAIJ" 43dfbe8321SBarry Smith PetscErrorCode MatMarkDiagonal_SeqSBAIJ(Mat A) 4449b5e25fSSatish Balay { 45045c9aa0SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 466849ba73SBarry Smith PetscErrorCode ierr; 4709f38230SBarry Smith PetscInt i; 4849b5e25fSSatish Balay 4949b5e25fSSatish Balay PetscFunctionBegin; 5009f38230SBarry Smith if (!a->diag) { 5109f38230SBarry Smith ierr = PetscMalloc(a->mbs*sizeof(PetscInt),&a->diag);CHKERRQ(ierr); 52c760cd28SBarry Smith ierr = PetscLogObjectMemory(A,a->mbs*sizeof(PetscInt));CHKERRQ(ierr); 53c760cd28SBarry Smith a->free_diag = PETSC_TRUE; 5409f38230SBarry Smith } 5509f38230SBarry Smith for (i=0; i<a->mbs; i++) a->diag[i] = a->i[i]; 5649b5e25fSSatish Balay PetscFunctionReturn(0); 5749b5e25fSSatish Balay } 5849b5e25fSSatish Balay 594a2ae208SSatish Balay #undef __FUNCT__ 604a2ae208SSatish Balay #define __FUNCT__ "MatGetRowIJ_SeqSBAIJ" 61ace3abfcSBarry Smith static PetscErrorCode MatGetRowIJ_SeqSBAIJ(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *nn,PetscInt *ia[],PetscInt *ja[],PetscBool *done) 6249b5e25fSSatish Balay { 63a6ece127SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 64d0f46423SBarry Smith PetscInt i,j,n = a->mbs,nz = a->i[n],bs = A->rmap->bs; 658f7157efSSatish Balay PetscErrorCode ierr; 6649b5e25fSSatish Balay 6749b5e25fSSatish Balay PetscFunctionBegin; 68d3e5a4abSHong Zhang *nn = n; 69a1373b80SHong Zhang if (!ia) PetscFunctionReturn(0); 708f7157efSSatish Balay if (!blockcompressed) { 718f7157efSSatish Balay /* malloc & create the natural set of indices */ 72f1d0d59dSSatish Balay ierr = PetscMalloc2((n+1)*bs,PetscInt,ia,nz*bs,PetscInt,ja);CHKERRQ(ierr); 738f7157efSSatish Balay for (i=0; i<n+1; i++) { 748f7157efSSatish Balay for (j=0; j<bs; j++) { 758f7157efSSatish Balay *ia[i*bs+j] = a->i[i]*bs+j+oshift; 768f7157efSSatish Balay } 778f7157efSSatish Balay } 788f7157efSSatish Balay for (i=0; i<nz; i++) { 798f7157efSSatish Balay for (j=0; j<bs; j++) { 808f7157efSSatish Balay *ja[i*bs+j] = a->j[i]*bs+j+oshift; 818f7157efSSatish Balay } 828f7157efSSatish Balay } 838f7157efSSatish Balay } else { /* blockcompressed */ 84a6ece127SHong Zhang if (oshift == 1) { 85a6ece127SHong Zhang /* temporarily add 1 to i and j indices */ 866c6c5352SBarry Smith for (i=0; i<nz; i++) a->j[i]++; 87a1373b80SHong Zhang for (i=0; i<n+1; i++) a->i[i]++; 888f7157efSSatish Balay } 89a1373b80SHong Zhang *ia = a->i; *ja = a->j; 90a6ece127SHong Zhang } 918f7157efSSatish Balay 9249b5e25fSSatish Balay PetscFunctionReturn(0); 9349b5e25fSSatish Balay } 9449b5e25fSSatish Balay 954a2ae208SSatish Balay #undef __FUNCT__ 964a2ae208SSatish Balay #define __FUNCT__ "MatRestoreRowIJ_SeqSBAIJ" 97ace3abfcSBarry Smith static PetscErrorCode MatRestoreRowIJ_SeqSBAIJ(Mat A,PetscInt oshift,PetscBool symmetric,PetscBool blockcompressed,PetscInt *nn,PetscInt *ia[],PetscInt *ja[],PetscBool *done) 9849b5e25fSSatish Balay { 99b7aaefc3SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 1008f7157efSSatish Balay PetscInt i,n = a->mbs,nz = a->i[n]; 1018f7157efSSatish Balay PetscErrorCode ierr; 102a6ece127SHong Zhang 10349b5e25fSSatish Balay PetscFunctionBegin; 10449b5e25fSSatish Balay if (!ia) PetscFunctionReturn(0); 105a6ece127SHong Zhang 1068f7157efSSatish Balay if (!blockcompressed) { 1078f7157efSSatish Balay ierr = PetscFree2(*ia,*ja);CHKERRQ(ierr); 1088f7157efSSatish Balay } else if (oshift == 1) { /* blockcompressed */ 1096c6c5352SBarry Smith for (i=0; i<nz; i++) a->j[i]--; 110a6ece127SHong Zhang for (i=0; i<n+1; i++) a->i[i]--; 111a6ece127SHong Zhang } 1128f7157efSSatish Balay 113a6ece127SHong Zhang PetscFunctionReturn(0); 11449b5e25fSSatish Balay } 11549b5e25fSSatish Balay 1164a2ae208SSatish Balay #undef __FUNCT__ 1174a2ae208SSatish Balay #define __FUNCT__ "MatDestroy_SeqSBAIJ" 118dfbe8321SBarry Smith PetscErrorCode MatDestroy_SeqSBAIJ(Mat A) 11949b5e25fSSatish Balay { 12049b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 121dfbe8321SBarry Smith PetscErrorCode ierr; 12249b5e25fSSatish Balay 12349b5e25fSSatish Balay PetscFunctionBegin; 124a9f03627SSatish Balay #if defined(PETSC_USE_LOG) 125d0f46423SBarry Smith PetscLogObjectState((PetscObject)A,"Rows=%D, NZ=%D",A->rmap->N,a->nz); 126a9f03627SSatish Balay #endif 127e6b907acSBarry Smith ierr = MatSeqXAIJFreeAIJ(A,&a->a,&a->j,&a->i);CHKERRQ(ierr); 1287f53bb6cSHong Zhang if (a->free_diag){ierr = PetscFree(a->diag);CHKERRQ(ierr);} 1296bf464f9SBarry Smith ierr = ISDestroy(&a->row);CHKERRQ(ierr); 1306bf464f9SBarry Smith ierr = ISDestroy(&a->col);CHKERRQ(ierr); 1316bf464f9SBarry Smith ierr = ISDestroy(&a->icol);CHKERRQ(ierr); 132c31cb41cSBarry Smith ierr = PetscFree(a->idiag);CHKERRQ(ierr); 133c31cb41cSBarry Smith ierr = PetscFree(a->inode.size);CHKERRQ(ierr); 134c31cb41cSBarry Smith ierr = PetscFree(a->diag);CHKERRQ(ierr); 135c760cd28SBarry Smith if (a->free_imax_ilen) {ierr = PetscFree2(a->imax,a->ilen);CHKERRQ(ierr);} 13605b42c5fSBarry Smith ierr = PetscFree(a->solve_work);CHKERRQ(ierr); 13741f059aeSBarry Smith ierr = PetscFree(a->sor_work);CHKERRQ(ierr); 13805b42c5fSBarry Smith ierr = PetscFree(a->solves_work);CHKERRQ(ierr); 13905b42c5fSBarry Smith ierr = PetscFree(a->mult_work);CHKERRQ(ierr); 14005b42c5fSBarry Smith ierr = PetscFree(a->saved_values);CHKERRQ(ierr); 14105b42c5fSBarry Smith ierr = PetscFree(a->xtoy);CHKERRQ(ierr); 1424da8f245SBarry Smith if (a->free_jshort) {ierr = PetscFree(a->jshort);CHKERRQ(ierr);} 1431a3463dfSHong Zhang ierr = PetscFree(a->inew);CHKERRQ(ierr); 1446bf464f9SBarry Smith ierr = MatDestroy(&a->parent);CHKERRQ(ierr); 145bf0cc555SLisandro Dalcin ierr = PetscFree(A->data);CHKERRQ(ierr); 146901853e0SKris Buschelman 147dbd8c25aSHong Zhang ierr = PetscObjectChangeTypeName((PetscObject)A,0);CHKERRQ(ierr); 148901853e0SKris Buschelman ierr = PetscObjectComposeFunction((PetscObject)A,"MatStoreValues_C","",PETSC_NULL);CHKERRQ(ierr); 149901853e0SKris Buschelman ierr = PetscObjectComposeFunction((PetscObject)A,"MatRetrieveValues_C","",PETSC_NULL);CHKERRQ(ierr); 150901853e0SKris Buschelman ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetColumnIndices_C","",PETSC_NULL);CHKERRQ(ierr); 151901853e0SKris Buschelman ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqaij_C","",PETSC_NULL);CHKERRQ(ierr); 152901853e0SKris Buschelman ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqbaij_C","",PETSC_NULL);CHKERRQ(ierr); 153901853e0SKris Buschelman ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetPreallocation_C","",PETSC_NULL);CHKERRQ(ierr); 154aa5a9175SDahai Guo ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqsbstrm_C","",PETSC_NULL);CHKERRQ(ierr); 15549b5e25fSSatish Balay PetscFunctionReturn(0); 15649b5e25fSSatish Balay } 15749b5e25fSSatish Balay 1584a2ae208SSatish Balay #undef __FUNCT__ 1594a2ae208SSatish Balay #define __FUNCT__ "MatSetOption_SeqSBAIJ" 160ace3abfcSBarry Smith PetscErrorCode MatSetOption_SeqSBAIJ(Mat A,MatOption op,PetscBool flg) 16149b5e25fSSatish Balay { 162045c9aa0SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 16363ba0a88SBarry Smith PetscErrorCode ierr; 16449b5e25fSSatish Balay 16549b5e25fSSatish Balay PetscFunctionBegin; 1664d9d31abSKris Buschelman switch (op) { 1674d9d31abSKris Buschelman case MAT_ROW_ORIENTED: 1684e0d8c25SBarry Smith a->roworiented = flg; 1694d9d31abSKris Buschelman break; 170a9817697SBarry Smith case MAT_KEEP_NONZERO_PATTERN: 171a9817697SBarry Smith a->keepnonzeropattern = flg; 1724d9d31abSKris Buschelman break; 173512a5fc5SBarry Smith case MAT_NEW_NONZERO_LOCATIONS: 174512a5fc5SBarry Smith a->nonew = (flg ? 0 : 1); 1754d9d31abSKris Buschelman break; 1764d9d31abSKris Buschelman case MAT_NEW_NONZERO_LOCATION_ERR: 1774e0d8c25SBarry Smith a->nonew = (flg ? -1 : 0); 1784d9d31abSKris Buschelman break; 1794d9d31abSKris Buschelman case MAT_NEW_NONZERO_ALLOCATION_ERR: 1804e0d8c25SBarry Smith a->nonew = (flg ? -2 : 0); 1814d9d31abSKris Buschelman break; 18228b2fa4aSMatthew Knepley case MAT_UNUSED_NONZERO_LOCATION_ERR: 18328b2fa4aSMatthew Knepley a->nounused = (flg ? -1 : 0); 18428b2fa4aSMatthew Knepley break; 1854e0d8c25SBarry Smith case MAT_NEW_DIAGONALS: 1864d9d31abSKris Buschelman case MAT_IGNORE_OFF_PROC_ENTRIES: 1874d9d31abSKris Buschelman case MAT_USE_HASH_TABLE: 188290bbb0aSBarry Smith ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr); 1894d9d31abSKris Buschelman break; 1909a4540c5SBarry Smith case MAT_HERMITIAN: 191e32f2f54SBarry Smith if (!A->assembled) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Must call MatAssemblyEnd() first"); 1920bc54ff2SBarry Smith if (A->cmap->n < 65536 && A->cmap->bs == 1) { 193eeffb40dSHong Zhang A->ops->mult = MatMult_SeqSBAIJ_1_Hermitian_ushort; 1940bc54ff2SBarry Smith } else if (A->cmap->bs == 1) { 195eeffb40dSHong Zhang A->ops->mult = MatMult_SeqSBAIJ_1_Hermitian; 196e32f2f54SBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No support for Hermitian with block size greater than 1"); 197eeffb40dSHong Zhang break; 1983d472b54SHong Zhang case MAT_SPD: 1993d472b54SHong Zhang A->spd_set = PETSC_TRUE; 2003d472b54SHong Zhang A->spd = flg; 2013d472b54SHong Zhang if (flg) { 2023d472b54SHong Zhang A->symmetric = PETSC_TRUE; 2033d472b54SHong Zhang A->structurally_symmetric = PETSC_TRUE; 2043d472b54SHong Zhang A->symmetric_set = PETSC_TRUE; 2053d472b54SHong Zhang A->structurally_symmetric_set = PETSC_TRUE; 2063d472b54SHong Zhang } 2073d472b54SHong Zhang break; 20877e54ba9SKris Buschelman case MAT_SYMMETRIC: 20977e54ba9SKris Buschelman case MAT_STRUCTURALLY_SYMMETRIC: 2109a4540c5SBarry Smith case MAT_SYMMETRY_ETERNAL: 211e32f2f54SBarry Smith if (!flg) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix must be symmetric"); 212290bbb0aSBarry Smith ierr = PetscInfo1(A,"Option %s not relevent\n",MatOptions[op]);CHKERRQ(ierr); 213290bbb0aSBarry Smith break; 214941593c8SHong Zhang case MAT_IGNORE_LOWER_TRIANGULAR: 2154e0d8c25SBarry Smith a->ignore_ltriangular = flg; 216941593c8SHong Zhang break; 217941593c8SHong Zhang case MAT_ERROR_LOWER_TRIANGULAR: 2184e0d8c25SBarry Smith a->ignore_ltriangular = flg; 21977e54ba9SKris Buschelman break; 220f5edf698SHong Zhang case MAT_GETROW_UPPERTRIANGULAR: 2214e0d8c25SBarry Smith a->getrow_utriangular = flg; 222f5edf698SHong Zhang break; 2234d9d31abSKris Buschelman default: 224e32f2f54SBarry Smith SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"unknown option %d",op); 22549b5e25fSSatish Balay } 22649b5e25fSSatish Balay PetscFunctionReturn(0); 22749b5e25fSSatish Balay } 22849b5e25fSSatish Balay 2294a2ae208SSatish Balay #undef __FUNCT__ 2304a2ae208SSatish Balay #define __FUNCT__ "MatGetRow_SeqSBAIJ" 23113f74950SBarry Smith PetscErrorCode MatGetRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *ncols,PetscInt **cols,PetscScalar **v) 23249b5e25fSSatish Balay { 23349b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 2346849ba73SBarry Smith PetscErrorCode ierr; 23513f74950SBarry Smith PetscInt itmp,i,j,k,M,*ai,*aj,bs,bn,bp,*cols_i,bs2; 23649b5e25fSSatish Balay MatScalar *aa,*aa_i; 23787828ca2SBarry Smith PetscScalar *v_i; 23849b5e25fSSatish Balay 23949b5e25fSSatish Balay PetscFunctionBegin; 240e32f2f54SBarry 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()"); 241f5edf698SHong Zhang /* Get the upper triangular part of the row */ 242d0f46423SBarry Smith bs = A->rmap->bs; 24349b5e25fSSatish Balay ai = a->i; 24449b5e25fSSatish Balay aj = a->j; 24549b5e25fSSatish Balay aa = a->a; 24649b5e25fSSatish Balay bs2 = a->bs2; 24749b5e25fSSatish Balay 248e32f2f54SBarry Smith if (row < 0 || row >= A->rmap->N) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE, "Row %D out of range", row); 24949b5e25fSSatish Balay 25049b5e25fSSatish Balay bn = row/bs; /* Block number */ 25149b5e25fSSatish Balay bp = row % bs; /* Block position */ 25249b5e25fSSatish Balay M = ai[bn+1] - ai[bn]; 25349b5e25fSSatish Balay *ncols = bs*M; 25449b5e25fSSatish Balay 25549b5e25fSSatish Balay if (v) { 25649b5e25fSSatish Balay *v = 0; 25749b5e25fSSatish Balay if (*ncols) { 25887828ca2SBarry Smith ierr = PetscMalloc((*ncols+row)*sizeof(PetscScalar),v);CHKERRQ(ierr); 25949b5e25fSSatish Balay for (i=0; i<M; i++) { /* for each block in the block row */ 26049b5e25fSSatish Balay v_i = *v + i*bs; 26149b5e25fSSatish Balay aa_i = aa + bs2*(ai[bn] + i); 26249b5e25fSSatish Balay for (j=bp,k=0; j<bs2; j+=bs,k++) {v_i[k] = aa_i[j];} 26349b5e25fSSatish Balay } 26449b5e25fSSatish Balay } 26549b5e25fSSatish Balay } 26649b5e25fSSatish Balay 26749b5e25fSSatish Balay if (cols) { 26849b5e25fSSatish Balay *cols = 0; 26949b5e25fSSatish Balay if (*ncols) { 27013f74950SBarry Smith ierr = PetscMalloc((*ncols+row)*sizeof(PetscInt),cols);CHKERRQ(ierr); 27149b5e25fSSatish Balay for (i=0; i<M; i++) { /* for each block in the block row */ 27249b5e25fSSatish Balay cols_i = *cols + i*bs; 27349b5e25fSSatish Balay itmp = bs*aj[ai[bn] + i]; 27449b5e25fSSatish Balay for (j=0; j<bs; j++) {cols_i[j] = itmp++;} 27549b5e25fSSatish Balay } 27649b5e25fSSatish Balay } 27749b5e25fSSatish Balay } 27849b5e25fSSatish Balay 27949b5e25fSSatish Balay /*search column A(0:row-1,row) (=A(row,0:row-1)). Could be expensive! */ 2805ddb2528SHong Zhang /* this segment is currently removed, so only entries in the upper triangle are obtained */ 2815ddb2528SHong Zhang #ifdef column_search 28249b5e25fSSatish Balay v_i = *v + M*bs; 28349b5e25fSSatish Balay cols_i = *cols + M*bs; 28449b5e25fSSatish Balay for (i=0; i<bn; i++){ /* for each block row */ 28549b5e25fSSatish Balay M = ai[i+1] - ai[i]; 28649b5e25fSSatish Balay for (j=0; j<M; j++){ 28749b5e25fSSatish Balay itmp = aj[ai[i] + j]; /* block column value */ 28849b5e25fSSatish Balay if (itmp == bn){ 28949b5e25fSSatish Balay aa_i = aa + bs2*(ai[i] + j) + bs*bp; 29049b5e25fSSatish Balay for (k=0; k<bs; k++) { 29149b5e25fSSatish Balay *cols_i++ = i*bs+k; 29249b5e25fSSatish Balay *v_i++ = aa_i[k]; 29349b5e25fSSatish Balay } 29449b5e25fSSatish Balay *ncols += bs; 29549b5e25fSSatish Balay break; 29649b5e25fSSatish Balay } 29749b5e25fSSatish Balay } 29849b5e25fSSatish Balay } 2995ddb2528SHong Zhang #endif 30049b5e25fSSatish Balay PetscFunctionReturn(0); 30149b5e25fSSatish Balay } 30249b5e25fSSatish Balay 3034a2ae208SSatish Balay #undef __FUNCT__ 3044a2ae208SSatish Balay #define __FUNCT__ "MatRestoreRow_SeqSBAIJ" 30513f74950SBarry Smith PetscErrorCode MatRestoreRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v) 30649b5e25fSSatish Balay { 307dfbe8321SBarry Smith PetscErrorCode ierr; 30849b5e25fSSatish Balay 30949b5e25fSSatish Balay PetscFunctionBegin; 31005b42c5fSBarry Smith if (idx) {ierr = PetscFree(*idx);CHKERRQ(ierr);} 31105b42c5fSBarry Smith if (v) {ierr = PetscFree(*v);CHKERRQ(ierr);} 31249b5e25fSSatish Balay PetscFunctionReturn(0); 31349b5e25fSSatish Balay } 31449b5e25fSSatish Balay 3154a2ae208SSatish Balay #undef __FUNCT__ 316f5edf698SHong Zhang #define __FUNCT__ "MatGetRowUpperTriangular_SeqSBAIJ" 317f5edf698SHong Zhang PetscErrorCode MatGetRowUpperTriangular_SeqSBAIJ(Mat A) 318f5edf698SHong Zhang { 319f5edf698SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 320f5edf698SHong Zhang 321f5edf698SHong Zhang PetscFunctionBegin; 322f5edf698SHong Zhang a->getrow_utriangular = PETSC_TRUE; 323f5edf698SHong Zhang PetscFunctionReturn(0); 324f5edf698SHong Zhang } 325f5edf698SHong Zhang #undef __FUNCT__ 326f5edf698SHong Zhang #define __FUNCT__ "MatRestoreRowUpperTriangular_SeqSBAIJ" 327f5edf698SHong Zhang PetscErrorCode MatRestoreRowUpperTriangular_SeqSBAIJ(Mat A) 328f5edf698SHong Zhang { 329f5edf698SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 330f5edf698SHong Zhang 331f5edf698SHong Zhang PetscFunctionBegin; 332f5edf698SHong Zhang a->getrow_utriangular = PETSC_FALSE; 333f5edf698SHong Zhang PetscFunctionReturn(0); 334f5edf698SHong Zhang } 335f5edf698SHong Zhang 336f5edf698SHong Zhang #undef __FUNCT__ 3374a2ae208SSatish Balay #define __FUNCT__ "MatTranspose_SeqSBAIJ" 338fc4dec0aSBarry Smith PetscErrorCode MatTranspose_SeqSBAIJ(Mat A,MatReuse reuse,Mat *B) 33949b5e25fSSatish Balay { 340dfbe8321SBarry Smith PetscErrorCode ierr; 34149b5e25fSSatish Balay PetscFunctionBegin; 342815cbec1SBarry Smith if (reuse == MAT_INITIAL_MATRIX || *B != A) { 343999d9058SBarry Smith ierr = MatDuplicate(A,MAT_COPY_VALUES,B);CHKERRQ(ierr); 344fc4dec0aSBarry Smith } 3458115998fSBarry Smith PetscFunctionReturn(0); 34649b5e25fSSatish Balay } 34749b5e25fSSatish Balay 3484a2ae208SSatish Balay #undef __FUNCT__ 3494a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_ASCII" 3506849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_ASCII(Mat A,PetscViewer viewer) 35149b5e25fSSatish Balay { 35249b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 353dfbe8321SBarry Smith PetscErrorCode ierr; 354d0f46423SBarry Smith PetscInt i,j,bs = A->rmap->bs,k,l,bs2=a->bs2; 355f3ef73ceSBarry Smith PetscViewerFormat format; 356121deb67SSatish Balay PetscInt *diag; 35749b5e25fSSatish Balay 35849b5e25fSSatish Balay PetscFunctionBegin; 359b0a32e0cSBarry Smith ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 360456192e2SBarry Smith if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) { 36177431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," block size is %D\n",bs);CHKERRQ(ierr); 362fb9695e5SSatish Balay } else if (format == PETSC_VIEWER_ASCII_MATLAB) { 363d2507d54SMatthew Knepley Mat aij; 364d5f3da31SBarry Smith if (A->factortype && bs>1){ 36570d5e725SHong 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); 36670d5e725SHong Zhang PetscFunctionReturn(0); 36770d5e725SHong Zhang } 368c9f458caSMatthew Knepley ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&aij);CHKERRQ(ierr); 369c9f458caSMatthew Knepley ierr = MatView(aij,viewer);CHKERRQ(ierr); 3706bf464f9SBarry Smith ierr = MatDestroy(&aij);CHKERRQ(ierr); 371fb9695e5SSatish Balay } else if (format == PETSC_VIEWER_ASCII_COMMON) { 372d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 37349b5e25fSSatish Balay for (i=0; i<a->mbs; i++) { 37449b5e25fSSatish Balay for (j=0; j<bs; j++) { 37577431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr); 37649b5e25fSSatish Balay for (k=a->i[i]; k<a->i[i+1]; k++) { 37749b5e25fSSatish Balay for (l=0; l<bs; l++) { 37849b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX) 37949b5e25fSSatish Balay if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) { 380a83599f4SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %G + %G i) ",bs*a->j[k]+l, 38149b5e25fSSatish Balay PetscRealPart(a->a[bs2*k + l*bs + j]),PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 38249b5e25fSSatish Balay } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) { 383a83599f4SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %G - %G i) ",bs*a->j[k]+l, 38449b5e25fSSatish Balay PetscRealPart(a->a[bs2*k + l*bs + j]),-PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 38549b5e25fSSatish Balay } else if (PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) { 386a83599f4SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",bs*a->j[k]+l,PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 38749b5e25fSSatish Balay } 38849b5e25fSSatish Balay #else 38949b5e25fSSatish Balay if (a->a[bs2*k + l*bs + j] != 0.0) { 390a83599f4SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",bs*a->j[k]+l,a->a[bs2*k + l*bs + j]);CHKERRQ(ierr); 39149b5e25fSSatish Balay } 39249b5e25fSSatish Balay #endif 39349b5e25fSSatish Balay } 39449b5e25fSSatish Balay } 395b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 39649b5e25fSSatish Balay } 39749b5e25fSSatish Balay } 398d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 399c1490034SHong Zhang } else if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) { 400c1490034SHong Zhang PetscFunctionReturn(0); 40149b5e25fSSatish Balay } else { 402d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr); 4037566de4bSShri Abhyankar ierr = PetscObjectPrintClassNamePrefixType((PetscObject)A,viewer,"Matrix Object");CHKERRQ(ierr); 4042c990fa1SHong Zhang if (A->factortype){ /* for factored matrix */ 4052c990fa1SHong Zhang if (bs>1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"matrix is factored with bs>1. Not implemented yet"); 4062c990fa1SHong Zhang 407121deb67SSatish Balay diag=a->diag; 408121deb67SSatish Balay for (i=0; i<a->mbs; i++) { /* for row block i */ 4092c990fa1SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i);CHKERRQ(ierr); 4102c990fa1SHong Zhang /* diagonal entry */ 4112c990fa1SHong Zhang #if defined(PETSC_USE_COMPLEX) 4122c990fa1SHong Zhang if (PetscImaginaryPart(a->a[diag[i]]) > 0.0) { 413ca0704adSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %G + %G i) ",a->j[diag[i]],PetscRealPart(1.0/a->a[diag[i]]),PetscImaginaryPart(1.0/a->a[diag[i]]));CHKERRQ(ierr); 4142c990fa1SHong Zhang } else if (PetscImaginaryPart(a->a[diag[i]]) < 0.0) { 415ca0704adSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %G - %G i) ",a->j[diag[i]],PetscRealPart(1.0/a->a[diag[i]]),-PetscImaginaryPart(1.0/a->a[diag[i]]));CHKERRQ(ierr); 4162c990fa1SHong Zhang } else { 4172c990fa1SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",a->j[diag[i]],PetscRealPart(1.0/a->a[diag[i]]));CHKERRQ(ierr); 4182c990fa1SHong Zhang } 4192c990fa1SHong Zhang #else 4202c990fa1SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",a->j[diag[i]],1.0/a->a[diag[i]]);CHKERRQ(ierr); 4212c990fa1SHong Zhang #endif 4222c990fa1SHong Zhang /* off-diagonal entries */ 4232c990fa1SHong Zhang for (k=a->i[i]; k<a->i[i+1]-1; k++) { 4242c990fa1SHong Zhang #if defined(PETSC_USE_COMPLEX) 425ca0704adSBarry Smith if (PetscImaginaryPart(a->a[k]) > 0.0) { 426ca0704adSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %G + %G i) ",bs*a->j[k],PetscRealPart(a->a[k]),PetscImaginaryPart(a->a[k]));CHKERRQ(ierr); 427ca0704adSBarry Smith } else if (PetscImaginaryPart(a->a[k]) < 0.0) { 428ca0704adSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %G - %G i) ",bs*a->j[k],PetscRealPart(a->a[k]),-PetscImaginaryPart(a->a[k]));CHKERRQ(ierr); 4292c990fa1SHong Zhang } else { 430ca0704adSBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",bs*a->j[k],PetscRealPart(a->a[k]));CHKERRQ(ierr); 4312c990fa1SHong Zhang } 4322c990fa1SHong Zhang #else 4332c990fa1SHong Zhang ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",a->j[k],a->a[k]);CHKERRQ(ierr); 4342c990fa1SHong Zhang #endif 4352c990fa1SHong Zhang } 4362c990fa1SHong Zhang ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 4372c990fa1SHong Zhang } 4382c990fa1SHong Zhang 4392c990fa1SHong Zhang } else { /* for non-factored matrix */ 4400c74a584SJed Brown for (i=0; i<a->mbs; i++) { /* for row block i */ 4410c74a584SJed Brown for (j=0; j<bs; j++) { /* for row bs*i + j */ 44277431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr); 4430c74a584SJed Brown for (k=a->i[i]; k<a->i[i+1]; k++) { /* for column block */ 4440c74a584SJed Brown for (l=0; l<bs; l++) { /* for column */ 44549b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX) 44649b5e25fSSatish Balay if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0) { 447a83599f4SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %G + %G i) ",bs*a->j[k]+l, 44849b5e25fSSatish Balay PetscRealPart(a->a[bs2*k + l*bs + j]),PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 44949b5e25fSSatish Balay } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0) { 450a83599f4SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %G - %G i) ",bs*a->j[k]+l, 45149b5e25fSSatish Balay PetscRealPart(a->a[bs2*k + l*bs + j]),-PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 45249b5e25fSSatish Balay } else { 453a83599f4SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",bs*a->j[k]+l,PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 45449b5e25fSSatish Balay } 45549b5e25fSSatish Balay #else 456e9f7bc9eSHong Zhang ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",bs*a->j[k]+l,a->a[bs2*k + l*bs + j]);CHKERRQ(ierr); 45749b5e25fSSatish Balay #endif 45849b5e25fSSatish Balay } 45949b5e25fSSatish Balay } 460b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 46149b5e25fSSatish Balay } 46249b5e25fSSatish Balay } 4632c990fa1SHong Zhang } 464d00279f6SBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr); 46549b5e25fSSatish Balay } 466b0a32e0cSBarry Smith ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 46749b5e25fSSatish Balay PetscFunctionReturn(0); 46849b5e25fSSatish Balay } 46949b5e25fSSatish Balay 4704a2ae208SSatish Balay #undef __FUNCT__ 4714a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_Draw_Zoom" 4726849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw_Zoom(PetscDraw draw,void *Aa) 47349b5e25fSSatish Balay { 47449b5e25fSSatish Balay Mat A = (Mat) Aa; 47549b5e25fSSatish Balay Mat_SeqSBAIJ *a=(Mat_SeqSBAIJ*)A->data; 4766849ba73SBarry Smith PetscErrorCode ierr; 477d0f46423SBarry Smith PetscInt row,i,j,k,l,mbs=a->mbs,color,bs=A->rmap->bs,bs2=a->bs2; 47813f74950SBarry Smith PetscMPIInt rank; 47949b5e25fSSatish Balay PetscReal xl,yl,xr,yr,x_l,x_r,y_l,y_r; 48049b5e25fSSatish Balay MatScalar *aa; 48149b5e25fSSatish Balay MPI_Comm comm; 482b0a32e0cSBarry Smith PetscViewer viewer; 48349b5e25fSSatish Balay 48449b5e25fSSatish Balay PetscFunctionBegin; 48549b5e25fSSatish Balay /* 48649b5e25fSSatish Balay This is nasty. If this is called from an originally parallel matrix 48749b5e25fSSatish Balay then all processes call this,but only the first has the matrix so the 48849b5e25fSSatish Balay rest should return immediately. 48949b5e25fSSatish Balay */ 49049b5e25fSSatish Balay ierr = PetscObjectGetComm((PetscObject)draw,&comm);CHKERRQ(ierr); 49149b5e25fSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 49249b5e25fSSatish Balay if (rank) PetscFunctionReturn(0); 49349b5e25fSSatish Balay 49449b5e25fSSatish Balay ierr = PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer);CHKERRQ(ierr); 49549b5e25fSSatish Balay 496b0a32e0cSBarry Smith ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr); 497b0a32e0cSBarry Smith PetscDrawString(draw, .3*(xl+xr), .3*(yl+yr), PETSC_DRAW_BLACK, "symmetric"); 49849b5e25fSSatish Balay 49949b5e25fSSatish Balay /* loop over matrix elements drawing boxes */ 500b0a32e0cSBarry Smith color = PETSC_DRAW_BLUE; 50149b5e25fSSatish Balay for (i=0,row=0; i<mbs; i++,row+=bs) { 50249b5e25fSSatish Balay for (j=a->i[i]; j<a->i[i+1]; j++) { 503d0f46423SBarry Smith y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0; 50449b5e25fSSatish Balay x_l = a->j[j]*bs; x_r = x_l + 1.0; 50549b5e25fSSatish Balay aa = a->a + j*bs2; 50649b5e25fSSatish Balay for (k=0; k<bs; k++) { 50749b5e25fSSatish Balay for (l=0; l<bs; l++) { 50849b5e25fSSatish Balay if (PetscRealPart(*aa++) >= 0.) continue; 509b0a32e0cSBarry Smith ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr); 51049b5e25fSSatish Balay } 51149b5e25fSSatish Balay } 51249b5e25fSSatish Balay } 51349b5e25fSSatish Balay } 514b0a32e0cSBarry Smith color = PETSC_DRAW_CYAN; 51549b5e25fSSatish Balay for (i=0,row=0; i<mbs; i++,row+=bs) { 51649b5e25fSSatish Balay for (j=a->i[i]; j<a->i[i+1]; j++) { 517d0f46423SBarry Smith y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0; 51849b5e25fSSatish Balay x_l = a->j[j]*bs; x_r = x_l + 1.0; 51949b5e25fSSatish Balay aa = a->a + j*bs2; 52049b5e25fSSatish Balay for (k=0; k<bs; k++) { 52149b5e25fSSatish Balay for (l=0; l<bs; l++) { 52249b5e25fSSatish Balay if (PetscRealPart(*aa++) != 0.) continue; 523b0a32e0cSBarry Smith ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr); 52449b5e25fSSatish Balay } 52549b5e25fSSatish Balay } 52649b5e25fSSatish Balay } 52749b5e25fSSatish Balay } 52849b5e25fSSatish Balay 529b0a32e0cSBarry Smith color = PETSC_DRAW_RED; 53049b5e25fSSatish Balay for (i=0,row=0; i<mbs; i++,row+=bs) { 53149b5e25fSSatish Balay for (j=a->i[i]; j<a->i[i+1]; j++) { 532d0f46423SBarry Smith y_l = A->rmap->N - row - 1.0; y_r = y_l + 1.0; 53349b5e25fSSatish Balay x_l = a->j[j]*bs; x_r = x_l + 1.0; 53449b5e25fSSatish Balay aa = a->a + j*bs2; 53549b5e25fSSatish Balay for (k=0; k<bs; k++) { 53649b5e25fSSatish Balay for (l=0; l<bs; l++) { 53749b5e25fSSatish Balay if (PetscRealPart(*aa++) <= 0.) continue; 538b0a32e0cSBarry Smith ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr); 53949b5e25fSSatish Balay } 54049b5e25fSSatish Balay } 54149b5e25fSSatish Balay } 54249b5e25fSSatish Balay } 54349b5e25fSSatish Balay PetscFunctionReturn(0); 54449b5e25fSSatish Balay } 54549b5e25fSSatish Balay 5464a2ae208SSatish Balay #undef __FUNCT__ 5474a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_Draw" 5486849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw(Mat A,PetscViewer viewer) 54949b5e25fSSatish Balay { 550dfbe8321SBarry Smith PetscErrorCode ierr; 55149b5e25fSSatish Balay PetscReal xl,yl,xr,yr,w,h; 552b0a32e0cSBarry Smith PetscDraw draw; 553ace3abfcSBarry Smith PetscBool isnull; 55449b5e25fSSatish Balay 55549b5e25fSSatish Balay PetscFunctionBegin; 556b0a32e0cSBarry Smith ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr); 557b0a32e0cSBarry Smith ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr); if (isnull) PetscFunctionReturn(0); 55849b5e25fSSatish Balay 55949b5e25fSSatish Balay ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer);CHKERRQ(ierr); 560d0f46423SBarry Smith xr = A->rmap->N; yr = A->rmap->N; h = yr/10.0; w = xr/10.0; 56149b5e25fSSatish Balay xr += w; yr += h; xl = -w; yl = -h; 562b0a32e0cSBarry Smith ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr); 563b0a32e0cSBarry Smith ierr = PetscDrawZoom(draw,MatView_SeqSBAIJ_Draw_Zoom,A);CHKERRQ(ierr); 56449b5e25fSSatish Balay ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",PETSC_NULL);CHKERRQ(ierr); 56549b5e25fSSatish Balay PetscFunctionReturn(0); 56649b5e25fSSatish Balay } 56749b5e25fSSatish Balay 5684a2ae208SSatish Balay #undef __FUNCT__ 5694a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ" 570dfbe8321SBarry Smith PetscErrorCode MatView_SeqSBAIJ(Mat A,PetscViewer viewer) 57149b5e25fSSatish Balay { 572dfbe8321SBarry Smith PetscErrorCode ierr; 573ace3abfcSBarry Smith PetscBool iascii,isdraw; 57408917f38SBarry Smith FILE *file = 0; 57549b5e25fSSatish Balay 57649b5e25fSSatish Balay PetscFunctionBegin; 5772692d6eeSBarry Smith ierr = PetscTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr); 5782692d6eeSBarry Smith ierr = PetscTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr); 57932077d6dSBarry Smith if (iascii){ 58049b5e25fSSatish Balay ierr = MatView_SeqSBAIJ_ASCII(A,viewer);CHKERRQ(ierr); 58149b5e25fSSatish Balay } else if (isdraw) { 58249b5e25fSSatish Balay ierr = MatView_SeqSBAIJ_Draw(A,viewer);CHKERRQ(ierr); 58349b5e25fSSatish Balay } else { 584a5e6ed63SBarry Smith Mat B; 585ceb03754SKris Buschelman ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&B);CHKERRQ(ierr); 586a5e6ed63SBarry Smith ierr = MatView(B,viewer);CHKERRQ(ierr); 5876bf464f9SBarry Smith ierr = MatDestroy(&B);CHKERRQ(ierr); 58808917f38SBarry Smith ierr = PetscViewerBinaryGetInfoPointer(viewer,&file);CHKERRQ(ierr); 58908917f38SBarry Smith if (file) { 59008917f38SBarry Smith fprintf(file,"-matload_block_size %d\n",(int)A->rmap->bs); 59108917f38SBarry Smith } 59249b5e25fSSatish Balay } 59349b5e25fSSatish Balay PetscFunctionReturn(0); 59449b5e25fSSatish Balay } 59549b5e25fSSatish Balay 59649b5e25fSSatish Balay 5974a2ae208SSatish Balay #undef __FUNCT__ 5984a2ae208SSatish Balay #define __FUNCT__ "MatGetValues_SeqSBAIJ" 59913f74950SBarry Smith PetscErrorCode MatGetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],PetscScalar v[]) 60049b5e25fSSatish Balay { 601045c9aa0SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 60213f74950SBarry Smith PetscInt *rp,k,low,high,t,row,nrow,i,col,l,*aj = a->j; 60313f74950SBarry Smith PetscInt *ai = a->i,*ailen = a->ilen; 604d0f46423SBarry Smith PetscInt brow,bcol,ridx,cidx,bs=A->rmap->bs,bs2=a->bs2; 60597e567efSBarry Smith MatScalar *ap,*aa = a->a; 60649b5e25fSSatish Balay 60749b5e25fSSatish Balay PetscFunctionBegin; 60849b5e25fSSatish Balay for (k=0; k<m; k++) { /* loop over rows */ 60949b5e25fSSatish Balay row = im[k]; brow = row/bs; 610e32f2f54SBarry Smith if (row < 0) {v += n; continue;} /* SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative row: %D",row); */ 611e32f2f54SBarry 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); 61249b5e25fSSatish Balay rp = aj + ai[brow] ; ap = aa + bs2*ai[brow] ; 61349b5e25fSSatish Balay nrow = ailen[brow]; 61449b5e25fSSatish Balay for (l=0; l<n; l++) { /* loop over columns */ 615e32f2f54SBarry Smith if (in[l] < 0) {v++; continue;} /* SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column: %D",in[l]); */ 616e32f2f54SBarry 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); 61749b5e25fSSatish Balay col = in[l] ; 61849b5e25fSSatish Balay bcol = col/bs; 61949b5e25fSSatish Balay cidx = col%bs; 62049b5e25fSSatish Balay ridx = row%bs; 62149b5e25fSSatish Balay high = nrow; 62249b5e25fSSatish Balay low = 0; /* assume unsorted */ 62349b5e25fSSatish Balay while (high-low > 5) { 62449b5e25fSSatish Balay t = (low+high)/2; 62549b5e25fSSatish Balay if (rp[t] > bcol) high = t; 62649b5e25fSSatish Balay else low = t; 62749b5e25fSSatish Balay } 62849b5e25fSSatish Balay for (i=low; i<high; i++) { 62949b5e25fSSatish Balay if (rp[i] > bcol) break; 63049b5e25fSSatish Balay if (rp[i] == bcol) { 63149b5e25fSSatish Balay *v++ = ap[bs2*i+bs*cidx+ridx]; 63249b5e25fSSatish Balay goto finished; 63349b5e25fSSatish Balay } 63449b5e25fSSatish Balay } 63597e567efSBarry Smith *v++ = 0.0; 63649b5e25fSSatish Balay finished:; 63749b5e25fSSatish Balay } 63849b5e25fSSatish Balay } 63949b5e25fSSatish Balay PetscFunctionReturn(0); 64049b5e25fSSatish Balay } 64149b5e25fSSatish Balay 64249b5e25fSSatish Balay 6434a2ae208SSatish Balay #undef __FUNCT__ 6444a2ae208SSatish Balay #define __FUNCT__ "MatSetValuesBlocked_SeqSBAIJ" 64513f74950SBarry Smith PetscErrorCode MatSetValuesBlocked_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is) 64649b5e25fSSatish Balay { 6470880e062SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 6486849ba73SBarry Smith PetscErrorCode ierr; 649e2ee6c50SBarry Smith PetscInt *rp,k,low,high,t,ii,jj,row,nrow,i,col,l,rmax,N,lastcol = -1; 65013f74950SBarry Smith PetscInt *imax=a->imax,*ai=a->i,*ailen=a->ilen; 651d0f46423SBarry Smith PetscInt *aj=a->j,nonew=a->nonew,bs2=a->bs2,bs=A->rmap->bs,stepval; 652ace3abfcSBarry Smith PetscBool roworiented=a->roworiented; 653dd6ea824SBarry Smith const PetscScalar *value = v; 654f15d580aSBarry Smith MatScalar *ap,*aa = a->a,*bap; 6550880e062SHong Zhang 65649b5e25fSSatish Balay PetscFunctionBegin; 6570880e062SHong Zhang if (roworiented) { 6580880e062SHong Zhang stepval = (n-1)*bs; 6590880e062SHong Zhang } else { 6600880e062SHong Zhang stepval = (m-1)*bs; 6610880e062SHong Zhang } 6620880e062SHong Zhang for (k=0; k<m; k++) { /* loop over added rows */ 6630880e062SHong Zhang row = im[k]; 6640880e062SHong Zhang if (row < 0) continue; 6652515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 666e32f2f54SBarry Smith if (row >= a->mbs) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,a->mbs-1); 6670880e062SHong Zhang #endif 6680880e062SHong Zhang rp = aj + ai[row]; 6690880e062SHong Zhang ap = aa + bs2*ai[row]; 6700880e062SHong Zhang rmax = imax[row]; 6710880e062SHong Zhang nrow = ailen[row]; 6720880e062SHong Zhang low = 0; 673818f2c47SBarry Smith high = nrow; 6740880e062SHong Zhang for (l=0; l<n; l++) { /* loop over added columns */ 6750880e062SHong Zhang if (in[l] < 0) continue; 6760880e062SHong Zhang col = in[l]; 6772515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 678e32f2f54SBarry Smith if (col >= a->nbs) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",col,a->nbs-1); 679b1823623SSatish Balay #endif 680b98bf0e1SJed Brown if (col < row) { 681b98bf0e1SJed Brown if (a->ignore_ltriangular) { 682b98bf0e1SJed Brown continue; /* ignore lower triangular block */ 683b98bf0e1SJed Brown } else { 684e32f2f54SBarry Smith 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)"); 685b98bf0e1SJed Brown } 686b98bf0e1SJed Brown } 6870880e062SHong Zhang if (roworiented) { 6880880e062SHong Zhang value = v + k*(stepval+bs)*bs + l*bs; 6890880e062SHong Zhang } else { 6900880e062SHong Zhang value = v + l*(stepval+bs)*bs + k*bs; 6910880e062SHong Zhang } 6927cd84e04SBarry Smith if (col <= lastcol) low = 0; else high = nrow; 693e2ee6c50SBarry Smith lastcol = col; 6940880e062SHong Zhang while (high-low > 7) { 6950880e062SHong Zhang t = (low+high)/2; 6960880e062SHong Zhang if (rp[t] > col) high = t; 6970880e062SHong Zhang else low = t; 6980880e062SHong Zhang } 6990880e062SHong Zhang for (i=low; i<high; i++) { 7000880e062SHong Zhang if (rp[i] > col) break; 7010880e062SHong Zhang if (rp[i] == col) { 7020880e062SHong Zhang bap = ap + bs2*i; 7030880e062SHong Zhang if (roworiented) { 7040880e062SHong Zhang if (is == ADD_VALUES) { 7050880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 7060880e062SHong Zhang for (jj=ii; jj<bs2; jj+=bs) { 7070880e062SHong Zhang bap[jj] += *value++; 7080880e062SHong Zhang } 7090880e062SHong Zhang } 7100880e062SHong Zhang } else { 7110880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 7120880e062SHong Zhang for (jj=ii; jj<bs2; jj+=bs) { 7130880e062SHong Zhang bap[jj] = *value++; 7140880e062SHong Zhang } 7150880e062SHong Zhang } 7160880e062SHong Zhang } 7170880e062SHong Zhang } else { 7180880e062SHong Zhang if (is == ADD_VALUES) { 7190880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 7200880e062SHong Zhang for (jj=0; jj<bs; jj++) { 7210880e062SHong Zhang *bap++ += *value++; 7220880e062SHong Zhang } 7230880e062SHong Zhang } 7240880e062SHong Zhang } else { 7250880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 7260880e062SHong Zhang for (jj=0; jj<bs; jj++) { 7270880e062SHong Zhang *bap++ = *value++; 7280880e062SHong Zhang } 7290880e062SHong Zhang } 7300880e062SHong Zhang } 7310880e062SHong Zhang } 7320880e062SHong Zhang goto noinsert2; 7330880e062SHong Zhang } 7340880e062SHong Zhang } 7350880e062SHong Zhang if (nonew == 1) goto noinsert2; 736e32f2f54SBarry Smith if (nonew == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col); 737fef13f97SBarry Smith MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,row,col,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar); 738c03d1d03SSatish Balay N = nrow++ - 1; high++; 7390880e062SHong Zhang /* shift up all the later entries in this row */ 7400880e062SHong Zhang for (ii=N; ii>=i; ii--) { 7410880e062SHong Zhang rp[ii+1] = rp[ii]; 7420880e062SHong Zhang ierr = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr); 7430880e062SHong Zhang } 7440880e062SHong Zhang if (N >= i) { 7450880e062SHong Zhang ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr); 7460880e062SHong Zhang } 7470880e062SHong Zhang rp[i] = col; 7480880e062SHong Zhang bap = ap + bs2*i; 7490880e062SHong Zhang if (roworiented) { 7500880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 7510880e062SHong Zhang for (jj=ii; jj<bs2; jj+=bs) { 7520880e062SHong Zhang bap[jj] = *value++; 7530880e062SHong Zhang } 7540880e062SHong Zhang } 7550880e062SHong Zhang } else { 7560880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 7570880e062SHong Zhang for (jj=0; jj<bs; jj++) { 7580880e062SHong Zhang *bap++ = *value++; 7590880e062SHong Zhang } 7600880e062SHong Zhang } 7610880e062SHong Zhang } 7620880e062SHong Zhang noinsert2:; 7630880e062SHong Zhang low = i; 7640880e062SHong Zhang } 7650880e062SHong Zhang ailen[row] = nrow; 7660880e062SHong Zhang } 7670880e062SHong Zhang PetscFunctionReturn(0); 76849b5e25fSSatish Balay } 76949b5e25fSSatish Balay 77064831d72SBarry Smith /* 77164831d72SBarry Smith This is not yet used 77264831d72SBarry Smith */ 7734a2ae208SSatish Balay #undef __FUNCT__ 7744108e4d5SBarry Smith #define __FUNCT__ "MatAssemblyEnd_SeqSBAIJ_SeqAIJ_Inode" 7754108e4d5SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ_SeqAIJ_Inode(Mat A) 7760def2e27SBarry Smith { 7770def2e27SBarry Smith Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 7780def2e27SBarry Smith PetscErrorCode ierr; 7790def2e27SBarry Smith const PetscInt *ai = a->i, *aj = a->j,*cols; 7800def2e27SBarry Smith PetscInt i = 0,j,blk_size,m = A->rmap->n,node_count = 0,nzx,nzy,*ns,row,nz,cnt,cnt2,*counts; 781ace3abfcSBarry Smith PetscBool flag; 7820def2e27SBarry Smith 7830def2e27SBarry Smith PetscFunctionBegin; 7840def2e27SBarry Smith ierr = PetscMalloc(m*sizeof(PetscInt),&ns);CHKERRQ(ierr); 7850def2e27SBarry Smith while (i < m){ 7860def2e27SBarry Smith nzx = ai[i+1] - ai[i]; /* Number of nonzeros */ 7870def2e27SBarry Smith /* Limits the number of elements in a node to 'a->inode.limit' */ 7880def2e27SBarry Smith for (j=i+1,blk_size=1; j<m && blk_size <a->inode.limit; ++j,++blk_size) { 7890def2e27SBarry Smith nzy = ai[j+1] - ai[j]; 7900def2e27SBarry Smith if (nzy != (nzx - j + i)) break; 7910def2e27SBarry Smith ierr = PetscMemcmp(aj + ai[i] + j - i,aj + ai[j],nzy*sizeof(PetscInt),&flag);CHKERRQ(ierr); 7920def2e27SBarry Smith if (!flag) break; 7930def2e27SBarry Smith } 7940def2e27SBarry Smith ns[node_count++] = blk_size; 7950def2e27SBarry Smith i = j; 7960def2e27SBarry Smith } 7970def2e27SBarry Smith if (!a->inode.size && m && node_count > .9*m) { 7980def2e27SBarry Smith ierr = PetscFree(ns);CHKERRQ(ierr); 7990def2e27SBarry Smith ierr = PetscInfo2(A,"Found %D nodes out of %D rows. Not using Inode routines\n",node_count,m);CHKERRQ(ierr); 8000def2e27SBarry Smith } else { 8010def2e27SBarry Smith a->inode.node_count = node_count; 8020def2e27SBarry Smith ierr = PetscMalloc(node_count*sizeof(PetscInt),&a->inode.size);CHKERRQ(ierr); 803c760cd28SBarry Smith ierr = PetscLogObjectMemory(A,node_count*sizeof(PetscInt));CHKERRQ(ierr); 8040def2e27SBarry Smith ierr = PetscMemcpy(a->inode.size,ns,node_count*sizeof(PetscInt)); 8050def2e27SBarry Smith ierr = PetscFree(ns);CHKERRQ(ierr); 8060def2e27SBarry Smith ierr = PetscInfo3(A,"Found %D nodes of %D. Limit used: %D. Using Inode routines\n",node_count,m,a->inode.limit);CHKERRQ(ierr); 8070def2e27SBarry Smith 8080def2e27SBarry Smith /* count collections of adjacent columns in each inode */ 8090def2e27SBarry Smith row = 0; 8100def2e27SBarry Smith cnt = 0; 8110def2e27SBarry Smith for (i=0; i<node_count; i++) { 8120def2e27SBarry Smith cols = aj + ai[row] + a->inode.size[i]; 8130def2e27SBarry Smith nz = ai[row+1] - ai[row] - a->inode.size[i]; 8140def2e27SBarry Smith for (j=1; j<nz; j++) { 8150def2e27SBarry Smith if (cols[j] != cols[j-1]+1) { 8160def2e27SBarry Smith cnt++; 8170def2e27SBarry Smith } 8180def2e27SBarry Smith } 8190def2e27SBarry Smith cnt++; 8200def2e27SBarry Smith row += a->inode.size[i]; 8210def2e27SBarry Smith } 8220def2e27SBarry Smith ierr = PetscMalloc(2*cnt*sizeof(PetscInt),&counts);CHKERRQ(ierr); 8230def2e27SBarry Smith cnt = 0; 8240def2e27SBarry Smith row = 0; 8250def2e27SBarry Smith for (i=0; i<node_count; i++) { 8260def2e27SBarry Smith cols = aj + ai[row] + a->inode.size[i]; 8270def2e27SBarry Smith CHKMEMQ; 8280def2e27SBarry Smith counts[2*cnt] = cols[0]; 8290def2e27SBarry Smith CHKMEMQ; 8300def2e27SBarry Smith nz = ai[row+1] - ai[row] - a->inode.size[i]; 8310def2e27SBarry Smith cnt2 = 1; 8320def2e27SBarry Smith for (j=1; j<nz; j++) { 8330def2e27SBarry Smith if (cols[j] != cols[j-1]+1) { 8340def2e27SBarry Smith CHKMEMQ; 8350def2e27SBarry Smith counts[2*(cnt++)+1] = cnt2; 8360def2e27SBarry Smith counts[2*cnt] = cols[j]; 8370def2e27SBarry Smith CHKMEMQ; 8380def2e27SBarry Smith cnt2 = 1; 8390def2e27SBarry Smith } else cnt2++; 8400def2e27SBarry Smith } 8410def2e27SBarry Smith CHKMEMQ; 8420def2e27SBarry Smith counts[2*(cnt++)+1] = cnt2; 8430def2e27SBarry Smith CHKMEMQ; 8440def2e27SBarry Smith row += a->inode.size[i]; 8450def2e27SBarry Smith } 8460def2e27SBarry Smith ierr = PetscIntView(2*cnt,counts,0); 8470def2e27SBarry Smith } 84838702af4SBarry Smith PetscFunctionReturn(0); 84938702af4SBarry Smith } 85038702af4SBarry Smith 85138702af4SBarry Smith #undef __FUNCT__ 8524a2ae208SSatish Balay #define __FUNCT__ "MatAssemblyEnd_SeqSBAIJ" 853dfbe8321SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ(Mat A,MatAssemblyType mode) 85449b5e25fSSatish Balay { 85549b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 8566849ba73SBarry Smith PetscErrorCode ierr; 85713f74950SBarry Smith PetscInt fshift = 0,i,j,*ai = a->i,*aj = a->j,*imax = a->imax; 858d0f46423SBarry Smith PetscInt m = A->rmap->N,*ip,N,*ailen = a->ilen; 85913f74950SBarry Smith PetscInt mbs = a->mbs,bs2 = a->bs2,rmax = 0; 86049b5e25fSSatish Balay MatScalar *aa = a->a,*ap; 86149b5e25fSSatish Balay 86249b5e25fSSatish Balay PetscFunctionBegin; 86349b5e25fSSatish Balay if (mode == MAT_FLUSH_ASSEMBLY) PetscFunctionReturn(0); 86449b5e25fSSatish Balay 86549b5e25fSSatish Balay if (m) rmax = ailen[0]; 86649b5e25fSSatish Balay for (i=1; i<mbs; i++) { 86749b5e25fSSatish Balay /* move each row back by the amount of empty slots (fshift) before it*/ 86849b5e25fSSatish Balay fshift += imax[i-1] - ailen[i-1]; 86949b5e25fSSatish Balay rmax = PetscMax(rmax,ailen[i]); 87049b5e25fSSatish Balay if (fshift) { 87149b5e25fSSatish Balay ip = aj + ai[i]; ap = aa + bs2*ai[i]; 87249b5e25fSSatish Balay N = ailen[i]; 87349b5e25fSSatish Balay for (j=0; j<N; j++) { 87449b5e25fSSatish Balay ip[j-fshift] = ip[j]; 87549b5e25fSSatish Balay ierr = PetscMemcpy(ap+(j-fshift)*bs2,ap+j*bs2,bs2*sizeof(MatScalar));CHKERRQ(ierr); 87649b5e25fSSatish Balay } 87749b5e25fSSatish Balay } 87849b5e25fSSatish Balay ai[i] = ai[i-1] + ailen[i-1]; 87949b5e25fSSatish Balay } 88049b5e25fSSatish Balay if (mbs) { 88149b5e25fSSatish Balay fshift += imax[mbs-1] - ailen[mbs-1]; 88249b5e25fSSatish Balay ai[mbs] = ai[mbs-1] + ailen[mbs-1]; 88349b5e25fSSatish Balay } 88449b5e25fSSatish Balay /* reset ilen and imax for each row */ 88549b5e25fSSatish Balay for (i=0; i<mbs; i++) { 88649b5e25fSSatish Balay ailen[i] = imax[i] = ai[i+1] - ai[i]; 88749b5e25fSSatish Balay } 8886c6c5352SBarry Smith a->nz = ai[mbs]; 88949b5e25fSSatish Balay 890b424e231SHong Zhang /* diagonals may have moved, reset it */ 891b424e231SHong Zhang if (a->diag) { 8922ed38d0bSJed Brown ierr = PetscMemcpy(a->diag,ai,mbs*sizeof(PetscInt));CHKERRQ(ierr); 89349b5e25fSSatish Balay } 89428b2fa4aSMatthew Knepley if (fshift && a->nounused == -1) { 895e32f2f54SBarry Smith 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); 89628b2fa4aSMatthew Knepley } 897d0f46423SBarry 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); 898ae15b995SBarry Smith ierr = PetscInfo1(A,"Number of mallocs during MatSetValues is %D\n",a->reallocs);CHKERRQ(ierr); 899ae15b995SBarry Smith ierr = PetscInfo1(A,"Most nonzeros blocks in any row is %D\n",rmax);CHKERRQ(ierr); 9008e58a170SBarry Smith A->info.mallocs += a->reallocs; 90149b5e25fSSatish Balay a->reallocs = 0; 90249b5e25fSSatish Balay A->info.nz_unneeded = (PetscReal)fshift*bs2; 903061b2667SBarry Smith a->idiagvalid = PETSC_FALSE; 90438702af4SBarry Smith 90538702af4SBarry Smith if (A->cmap->n < 65536 && A->cmap->bs == 1) { 90617803ae8SHong Zhang if (a->jshort){ 90717803ae8SHong Zhang /* when matrix data structure is changed, previous jshort must be replaced */ 90817803ae8SHong Zhang ierr = PetscFree(a->jshort);CHKERRQ(ierr); 90917803ae8SHong Zhang } 91038702af4SBarry Smith ierr = PetscMalloc(a->i[A->rmap->n]*sizeof(unsigned short),&a->jshort);CHKERRQ(ierr); 911c760cd28SBarry Smith ierr = PetscLogObjectMemory(A,a->i[A->rmap->n]*sizeof(unsigned short));CHKERRQ(ierr); 91238702af4SBarry Smith for (i=0; i<a->i[A->rmap->n]; i++) a->jshort[i] = a->j[i]; 91338702af4SBarry Smith A->ops->mult = MatMult_SeqSBAIJ_1_ushort; 91441f059aeSBarry Smith A->ops->sor = MatSOR_SeqSBAIJ_ushort; 9154da8f245SBarry Smith a->free_jshort = PETSC_TRUE; 91638702af4SBarry Smith } 91749b5e25fSSatish Balay PetscFunctionReturn(0); 91849b5e25fSSatish Balay } 91949b5e25fSSatish Balay 92049b5e25fSSatish Balay /* 92149b5e25fSSatish Balay This function returns an array of flags which indicate the locations of contiguous 92249b5e25fSSatish Balay blocks that should be zeroed. for eg: if bs = 3 and is = [0,1,2,3,5,6,7,8,9] 92349b5e25fSSatish Balay then the resulting sizes = [3,1,1,3,1] correspondig to sets [(0,1,2),(3),(5),(6,7,8),(9)] 92449b5e25fSSatish Balay Assume: sizes should be long enough to hold all the values. 92549b5e25fSSatish Balay */ 9264a2ae208SSatish Balay #undef __FUNCT__ 9274a2ae208SSatish Balay #define __FUNCT__ "MatZeroRows_SeqSBAIJ_Check_Blocks" 92813f74950SBarry Smith PetscErrorCode MatZeroRows_SeqSBAIJ_Check_Blocks(PetscInt idx[],PetscInt n,PetscInt bs,PetscInt sizes[], PetscInt *bs_max) 92949b5e25fSSatish Balay { 93013f74950SBarry Smith PetscInt i,j,k,row; 931ace3abfcSBarry Smith PetscBool flg; 93249b5e25fSSatish Balay 93349b5e25fSSatish Balay PetscFunctionBegin; 93449b5e25fSSatish Balay for (i=0,j=0; i<n; j++) { 93549b5e25fSSatish Balay row = idx[i]; 93649b5e25fSSatish Balay if (row%bs!=0) { /* Not the begining of a block */ 93749b5e25fSSatish Balay sizes[j] = 1; 93849b5e25fSSatish Balay i++; 93949b5e25fSSatish Balay } else if (i+bs > n) { /* Beginning of a block, but complete block doesn't exist (at idx end) */ 94049b5e25fSSatish Balay sizes[j] = 1; /* Also makes sure atleast 'bs' values exist for next else */ 94149b5e25fSSatish Balay i++; 94249b5e25fSSatish Balay } else { /* Begining of the block, so check if the complete block exists */ 94349b5e25fSSatish Balay flg = PETSC_TRUE; 94449b5e25fSSatish Balay for (k=1; k<bs; k++) { 94549b5e25fSSatish Balay if (row+k != idx[i+k]) { /* break in the block */ 94649b5e25fSSatish Balay flg = PETSC_FALSE; 94749b5e25fSSatish Balay break; 94849b5e25fSSatish Balay } 94949b5e25fSSatish Balay } 950abc0a331SBarry Smith if (flg) { /* No break in the bs */ 95149b5e25fSSatish Balay sizes[j] = bs; 95249b5e25fSSatish Balay i+= bs; 95349b5e25fSSatish Balay } else { 95449b5e25fSSatish Balay sizes[j] = 1; 95549b5e25fSSatish Balay i++; 95649b5e25fSSatish Balay } 95749b5e25fSSatish Balay } 95849b5e25fSSatish Balay } 95949b5e25fSSatish Balay *bs_max = j; 96049b5e25fSSatish Balay PetscFunctionReturn(0); 96149b5e25fSSatish Balay } 96249b5e25fSSatish Balay 96349b5e25fSSatish Balay 96449b5e25fSSatish Balay /* Only add/insert a(i,j) with i<=j (blocks). 96549b5e25fSSatish Balay Any a(i,j) with i>j input by user is ingored. 96649b5e25fSSatish Balay */ 96749b5e25fSSatish Balay 9684a2ae208SSatish Balay #undef __FUNCT__ 9694a2ae208SSatish Balay #define __FUNCT__ "MatSetValues_SeqSBAIJ" 97013f74950SBarry Smith PetscErrorCode MatSetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is) 97149b5e25fSSatish Balay { 97249b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 9736849ba73SBarry Smith PetscErrorCode ierr; 974e2ee6c50SBarry Smith PetscInt *rp,k,low,high,t,ii,row,nrow,i,col,l,rmax,N,lastcol = -1; 97513f74950SBarry Smith PetscInt *imax=a->imax,*ai=a->i,*ailen=a->ilen,roworiented=a->roworiented; 976d0f46423SBarry Smith PetscInt *aj=a->j,nonew=a->nonew,bs=A->rmap->bs,brow,bcol; 97713f74950SBarry Smith PetscInt ridx,cidx,bs2=a->bs2; 97849b5e25fSSatish Balay MatScalar *ap,value,*aa=a->a,*bap; 97949b5e25fSSatish Balay 98049b5e25fSSatish Balay PetscFunctionBegin; 98171fd2e92SBarry Smith if (v) PetscValidScalarPointer(v,6); 98249b5e25fSSatish Balay for (k=0; k<m; k++) { /* loop over added rows */ 98349b5e25fSSatish Balay row = im[k]; /* row number */ 98449b5e25fSSatish Balay brow = row/bs; /* block row number */ 98549b5e25fSSatish Balay if (row < 0) continue; 9862515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 987e32f2f54SBarry 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); 98849b5e25fSSatish Balay #endif 98949b5e25fSSatish Balay rp = aj + ai[brow]; /*ptr to beginning of column value of the row block*/ 99049b5e25fSSatish Balay ap = aa + bs2*ai[brow]; /*ptr to beginning of element value of the row block*/ 99149b5e25fSSatish Balay rmax = imax[brow]; /* maximum space allocated for this row */ 99249b5e25fSSatish Balay nrow = ailen[brow]; /* actual length of this row */ 99349b5e25fSSatish Balay low = 0; 99449b5e25fSSatish Balay 99549b5e25fSSatish Balay for (l=0; l<n; l++) { /* loop over added columns */ 99649b5e25fSSatish Balay if (in[l] < 0) continue; 9972515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 998e32f2f54SBarry 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); 99949b5e25fSSatish Balay #endif 100049b5e25fSSatish Balay col = in[l]; 100149b5e25fSSatish Balay bcol = col/bs; /* block col number */ 100249b5e25fSSatish Balay 1003941593c8SHong Zhang if (brow > bcol) { 1004941593c8SHong Zhang if (a->ignore_ltriangular){ 1005941593c8SHong Zhang continue; /* ignore lower triangular values */ 1006941593c8SHong Zhang } else { 1007e32f2f54SBarry Smith 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)"); 1008941593c8SHong Zhang } 1009941593c8SHong Zhang } 1010f4989cb3SHong Zhang 101149b5e25fSSatish Balay ridx = row % bs; cidx = col % bs; /*row and col index inside the block */ 10128549e402SHong Zhang if ((brow==bcol && ridx<=cidx) || (brow<bcol)){ 101349b5e25fSSatish Balay /* element value a(k,l) */ 101449b5e25fSSatish Balay if (roworiented) { 101549b5e25fSSatish Balay value = v[l + k*n]; 101649b5e25fSSatish Balay } else { 101749b5e25fSSatish Balay value = v[k + l*m]; 101849b5e25fSSatish Balay } 101949b5e25fSSatish Balay 102049b5e25fSSatish Balay /* move pointer bap to a(k,l) quickly and add/insert value */ 10217cd84e04SBarry Smith if (col <= lastcol) low = 0; high = nrow; 1022e2ee6c50SBarry Smith lastcol = col; 102349b5e25fSSatish Balay while (high-low > 7) { 102449b5e25fSSatish Balay t = (low+high)/2; 102549b5e25fSSatish Balay if (rp[t] > bcol) high = t; 102649b5e25fSSatish Balay else low = t; 102749b5e25fSSatish Balay } 102849b5e25fSSatish Balay for (i=low; i<high; i++) { 102949b5e25fSSatish Balay if (rp[i] > bcol) break; 103049b5e25fSSatish Balay if (rp[i] == bcol) { 103149b5e25fSSatish Balay bap = ap + bs2*i + bs*cidx + ridx; 103249b5e25fSSatish Balay if (is == ADD_VALUES) *bap += value; 103349b5e25fSSatish Balay else *bap = value; 10348549e402SHong Zhang /* for diag block, add/insert its symmetric element a(cidx,ridx) */ 10358549e402SHong Zhang if (brow == bcol && ridx < cidx){ 10368549e402SHong Zhang bap = ap + bs2*i + bs*ridx + cidx; 10378549e402SHong Zhang if (is == ADD_VALUES) *bap += value; 10388549e402SHong Zhang else *bap = value; 10398549e402SHong Zhang } 104049b5e25fSSatish Balay goto noinsert1; 104149b5e25fSSatish Balay } 104249b5e25fSSatish Balay } 104349b5e25fSSatish Balay 104449b5e25fSSatish Balay if (nonew == 1) goto noinsert1; 1045e32f2f54SBarry Smith if (nonew == -1) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col); 1046fef13f97SBarry Smith MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,brow,bcol,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar); 104749b5e25fSSatish Balay 1048c03d1d03SSatish Balay N = nrow++ - 1; high++; 104949b5e25fSSatish Balay /* shift up all the later entries in this row */ 105049b5e25fSSatish Balay for (ii=N; ii>=i; ii--) { 105149b5e25fSSatish Balay rp[ii+1] = rp[ii]; 105249b5e25fSSatish Balay ierr = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr); 105349b5e25fSSatish Balay } 105449b5e25fSSatish Balay if (N>=i) { 105549b5e25fSSatish Balay ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr); 105649b5e25fSSatish Balay } 105749b5e25fSSatish Balay rp[i] = bcol; 105849b5e25fSSatish Balay ap[bs2*i + bs*cidx + ridx] = value; 105949b5e25fSSatish Balay noinsert1:; 106049b5e25fSSatish Balay low = i; 10618549e402SHong Zhang } 106249b5e25fSSatish Balay } /* end of loop over added columns */ 106349b5e25fSSatish Balay ailen[brow] = nrow; 106449b5e25fSSatish Balay } /* end of loop over added rows */ 106549b5e25fSSatish Balay PetscFunctionReturn(0); 106649b5e25fSSatish Balay } 106749b5e25fSSatish Balay 10684a2ae208SSatish Balay #undef __FUNCT__ 10694d101231SSatish Balay #define __FUNCT__ "MatICCFactor_SeqSBAIJ" 10700481f469SBarry Smith PetscErrorCode MatICCFactor_SeqSBAIJ(Mat inA,IS row,const MatFactorInfo *info) 107149b5e25fSSatish Balay { 10724ccecd49SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)inA->data; 107349b5e25fSSatish Balay Mat outA; 1074dfbe8321SBarry Smith PetscErrorCode ierr; 1075ace3abfcSBarry Smith PetscBool row_identity; 107649b5e25fSSatish Balay 107749b5e25fSSatish Balay PetscFunctionBegin; 1078e32f2f54SBarry Smith if (info->levels != 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only levels=0 is supported for in-place icc"); 1079c84f5b01SHong Zhang ierr = ISIdentity(row,&row_identity);CHKERRQ(ierr); 1080e32f2f54SBarry Smith if (!row_identity) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Matrix reordering is not supported"); 1081e32f2f54SBarry 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()! */ 1082c84f5b01SHong Zhang 108349b5e25fSSatish Balay outA = inA; 1084d5f3da31SBarry Smith inA->factortype = MAT_FACTOR_ICC; 108549b5e25fSSatish Balay 10861a3463dfSHong Zhang ierr = MatMarkDiagonal_SeqSBAIJ(inA);CHKERRQ(ierr); 1087d595f711SHong Zhang ierr = MatSeqSBAIJSetNumericFactorization_inplace(inA,row_identity);CHKERRQ(ierr); 108849b5e25fSSatish Balay 1089c3122656SLisandro Dalcin ierr = PetscObjectReference((PetscObject)row);CHKERRQ(ierr); 10906bf464f9SBarry Smith ierr = ISDestroy(&a->row);CHKERRQ(ierr); 1091c84f5b01SHong Zhang a->row = row; 1092c3122656SLisandro Dalcin ierr = PetscObjectReference((PetscObject)row);CHKERRQ(ierr); 10936bf464f9SBarry Smith ierr = ISDestroy(&a->col);CHKERRQ(ierr); 1094c84f5b01SHong Zhang a->col = row; 1095c84f5b01SHong Zhang 1096c84f5b01SHong Zhang /* Create the invert permutation so that it can be used in MatCholeskyFactorNumeric() */ 1097c84f5b01SHong Zhang if (a->icol) {ierr = ISInvertPermutation(row,PETSC_DECIDE, &a->icol);CHKERRQ(ierr);} 109852e6d16bSBarry Smith ierr = PetscLogObjectParent(inA,a->icol);CHKERRQ(ierr); 109949b5e25fSSatish Balay 110049b5e25fSSatish Balay if (!a->solve_work) { 1101d0f46423SBarry Smith ierr = PetscMalloc((inA->rmap->N+inA->rmap->bs)*sizeof(PetscScalar),&a->solve_work);CHKERRQ(ierr); 1102d0f46423SBarry Smith ierr = PetscLogObjectMemory(inA,(inA->rmap->N+inA->rmap->bs)*sizeof(PetscScalar));CHKERRQ(ierr); 110349b5e25fSSatish Balay } 110449b5e25fSSatish Balay 1105719d5645SBarry Smith ierr = MatCholeskyFactorNumeric(outA,inA,info);CHKERRQ(ierr); 110649b5e25fSSatish Balay PetscFunctionReturn(0); 110749b5e25fSSatish Balay } 1108950f1e5bSHong Zhang 110949b5e25fSSatish Balay EXTERN_C_BEGIN 11104a2ae208SSatish Balay #undef __FUNCT__ 11114a2ae208SSatish Balay #define __FUNCT__ "MatSeqSBAIJSetColumnIndices_SeqSBAIJ" 11127087cfbeSBarry Smith PetscErrorCode MatSeqSBAIJSetColumnIndices_SeqSBAIJ(Mat mat,PetscInt *indices) 111349b5e25fSSatish Balay { 1114045c9aa0SHong Zhang Mat_SeqSBAIJ *baij = (Mat_SeqSBAIJ *)mat->data; 111513f74950SBarry Smith PetscInt i,nz,n; 11167827cd58SJed Brown PetscErrorCode ierr; 111749b5e25fSSatish Balay 111849b5e25fSSatish Balay PetscFunctionBegin; 11196c6c5352SBarry Smith nz = baij->maxnz; 1120d0f46423SBarry Smith n = mat->cmap->n; 112149b5e25fSSatish Balay for (i=0; i<nz; i++) { 112249b5e25fSSatish Balay baij->j[i] = indices[i]; 112349b5e25fSSatish Balay } 11246c6c5352SBarry Smith baij->nz = nz; 112549b5e25fSSatish Balay for (i=0; i<n; i++) { 112649b5e25fSSatish Balay baij->ilen[i] = baij->imax[i]; 112749b5e25fSSatish Balay } 11287827cd58SJed Brown ierr = MatSetOption(mat,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr); 112949b5e25fSSatish Balay PetscFunctionReturn(0); 113049b5e25fSSatish Balay } 113149b5e25fSSatish Balay EXTERN_C_END 113249b5e25fSSatish Balay 11334a2ae208SSatish Balay #undef __FUNCT__ 11344a2ae208SSatish Balay #define __FUNCT__ "MatSeqSBAIJSetColumnIndices" 113549b5e25fSSatish Balay /*@ 113619585528SSatish Balay MatSeqSBAIJSetColumnIndices - Set the column indices for all the rows 113749b5e25fSSatish Balay in the matrix. 113849b5e25fSSatish Balay 113949b5e25fSSatish Balay Input Parameters: 114019585528SSatish Balay + mat - the SeqSBAIJ matrix 114149b5e25fSSatish Balay - indices - the column indices 114249b5e25fSSatish Balay 114349b5e25fSSatish Balay Level: advanced 114449b5e25fSSatish Balay 114549b5e25fSSatish Balay Notes: 114649b5e25fSSatish Balay This can be called if you have precomputed the nonzero structure of the 114749b5e25fSSatish Balay matrix and want to provide it to the matrix object to improve the performance 114849b5e25fSSatish Balay of the MatSetValues() operation. 114949b5e25fSSatish Balay 115049b5e25fSSatish Balay You MUST have set the correct numbers of nonzeros per row in the call to 1151d1be2dadSMatthew Knepley MatCreateSeqSBAIJ(), and the columns indices MUST be sorted. 115249b5e25fSSatish Balay 1153ab9f2c04SSatish Balay MUST be called before any calls to MatSetValues() 115449b5e25fSSatish Balay 1155ab9f2c04SSatish Balay .seealso: MatCreateSeqSBAIJ 115649b5e25fSSatish Balay @*/ 11577087cfbeSBarry Smith PetscErrorCode MatSeqSBAIJSetColumnIndices(Mat mat,PetscInt *indices) 115849b5e25fSSatish Balay { 11594ac538c5SBarry Smith PetscErrorCode ierr; 116049b5e25fSSatish Balay 116149b5e25fSSatish Balay PetscFunctionBegin; 11620700a824SBarry Smith PetscValidHeaderSpecific(mat,MAT_CLASSID,1); 11634482741eSBarry Smith PetscValidPointer(indices,2); 11644ac538c5SBarry Smith ierr = PetscUseMethod(mat,"MatSeqSBAIJSetColumnIndices_C",(Mat,PetscInt *),(mat,indices));CHKERRQ(ierr); 116549b5e25fSSatish Balay PetscFunctionReturn(0); 116649b5e25fSSatish Balay } 116749b5e25fSSatish Balay 11684a2ae208SSatish Balay #undef __FUNCT__ 11693c896bc6SHong Zhang #define __FUNCT__ "MatCopy_SeqSBAIJ" 11703c896bc6SHong Zhang PetscErrorCode MatCopy_SeqSBAIJ(Mat A,Mat B,MatStructure str) 11713c896bc6SHong Zhang { 11723c896bc6SHong Zhang PetscErrorCode ierr; 11733c896bc6SHong Zhang 11743c896bc6SHong Zhang PetscFunctionBegin; 11753c896bc6SHong Zhang /* If the two matrices have the same copy implementation, use fast copy. */ 11763c896bc6SHong Zhang if (str == SAME_NONZERO_PATTERN && (A->ops->copy == B->ops->copy)) { 11773c896bc6SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 11783c896bc6SHong Zhang Mat_SeqSBAIJ *b = (Mat_SeqSBAIJ*)B->data; 11793c896bc6SHong Zhang 1180e7e72b3dSBarry 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"); 1181d0f46423SBarry Smith ierr = PetscMemcpy(b->a,a->a,(a->i[A->rmap->N])*sizeof(PetscScalar));CHKERRQ(ierr); 11823c896bc6SHong Zhang } else { 1183f5edf698SHong Zhang ierr = MatGetRowUpperTriangular(A);CHKERRQ(ierr); 11843c896bc6SHong Zhang ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr); 1185f5edf698SHong Zhang ierr = MatRestoreRowUpperTriangular(A);CHKERRQ(ierr); 11863c896bc6SHong Zhang } 11873c896bc6SHong Zhang PetscFunctionReturn(0); 11883c896bc6SHong Zhang } 11893c896bc6SHong Zhang 11903c896bc6SHong Zhang #undef __FUNCT__ 11914994cf47SJed Brown #define __FUNCT__ "MatSetUp_SeqSBAIJ" 11924994cf47SJed Brown PetscErrorCode MatSetUp_SeqSBAIJ(Mat A) 1193273d9f13SBarry Smith { 1194dfbe8321SBarry Smith PetscErrorCode ierr; 1195273d9f13SBarry Smith 1196273d9f13SBarry Smith PetscFunctionBegin; 1197db4efbfdSBarry Smith ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(A,-PetscMax(A->rmap->bs,1),PETSC_DEFAULT,0);CHKERRQ(ierr); 1198273d9f13SBarry Smith PetscFunctionReturn(0); 1199273d9f13SBarry Smith } 1200273d9f13SBarry Smith 1201a6ece127SHong Zhang #undef __FUNCT__ 1202a6ece127SHong Zhang #define __FUNCT__ "MatGetArray_SeqSBAIJ" 1203dfbe8321SBarry Smith PetscErrorCode MatGetArray_SeqSBAIJ(Mat A,PetscScalar *array[]) 1204a6ece127SHong Zhang { 1205a6ece127SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 1206a6ece127SHong Zhang PetscFunctionBegin; 1207a6ece127SHong Zhang *array = a->a; 1208a6ece127SHong Zhang PetscFunctionReturn(0); 1209a6ece127SHong Zhang } 1210a6ece127SHong Zhang 1211a6ece127SHong Zhang #undef __FUNCT__ 1212a6ece127SHong Zhang #define __FUNCT__ "MatRestoreArray_SeqSBAIJ" 1213dfbe8321SBarry Smith PetscErrorCode MatRestoreArray_SeqSBAIJ(Mat A,PetscScalar *array[]) 1214a6ece127SHong Zhang { 1215a6ece127SHong Zhang PetscFunctionBegin; 1216a6ece127SHong Zhang PetscFunctionReturn(0); 1217a6ece127SHong Zhang } 1218a6ece127SHong Zhang 121942ee4b1aSHong Zhang #undef __FUNCT__ 122042ee4b1aSHong Zhang #define __FUNCT__ "MatAXPY_SeqSBAIJ" 1221f4df32b1SMatthew Knepley PetscErrorCode MatAXPY_SeqSBAIJ(Mat Y,PetscScalar a,Mat X,MatStructure str) 122242ee4b1aSHong Zhang { 122342ee4b1aSHong Zhang Mat_SeqSBAIJ *x=(Mat_SeqSBAIJ *)X->data, *y=(Mat_SeqSBAIJ *)Y->data; 1224dfbe8321SBarry Smith PetscErrorCode ierr; 1225e838b9e7SJed Brown PetscInt i,bs=Y->rmap->bs,bs2=bs*bs,j; 1226e838b9e7SJed Brown PetscBLASInt one = 1; 122742ee4b1aSHong Zhang 122842ee4b1aSHong Zhang PetscFunctionBegin; 122942ee4b1aSHong Zhang if (str == SAME_NONZERO_PATTERN) { 1230f4df32b1SMatthew Knepley PetscScalar alpha = a; 1231666a761fSJed Brown PetscBLASInt bnz = PetscBLASIntCast(x->nz*bs2); 1232f4df32b1SMatthew Knepley BLASaxpy_(&bnz,&alpha,x->a,&one,y->a,&one); 1233c537a176SHong Zhang } else if (str == SUBSET_NONZERO_PATTERN) { /* nonzeros of X is a subset of Y's */ 1234c4319e64SHong Zhang if (y->xtoy && y->XtoY != X) { 1235c4319e64SHong Zhang ierr = PetscFree(y->xtoy);CHKERRQ(ierr); 12366bf464f9SBarry Smith ierr = MatDestroy(&y->XtoY);CHKERRQ(ierr); 1237c537a176SHong Zhang } 1238c4319e64SHong Zhang if (!y->xtoy) { /* get xtoy */ 1239c4319e64SHong Zhang ierr = MatAXPYGetxtoy_Private(x->mbs,x->i,x->j,PETSC_NULL, y->i,y->j,PETSC_NULL, &y->xtoy);CHKERRQ(ierr); 1240c4319e64SHong Zhang y->XtoY = X; 1241c537a176SHong Zhang } 12426c6c5352SBarry Smith for (i=0; i<x->nz; i++) { 1243c4319e64SHong Zhang j = 0; 1244c4319e64SHong Zhang while (j < bs2){ 1245f4df32b1SMatthew Knepley y->a[bs2*y->xtoy[i]+j] += a*(x->a[bs2*i+j]); 1246c4319e64SHong Zhang j++; 1247c537a176SHong Zhang } 1248c4319e64SHong Zhang } 12491e2582c4SBarry Smith ierr = PetscInfo3(Y,"ratio of nnz_s(X)/nnz_s(Y): %D/%D = %G\n",bs2*x->nz,bs2*y->nz,(PetscReal)(bs2*x->nz)/(bs2*y->nz));CHKERRQ(ierr); 125042ee4b1aSHong Zhang } else { 1251f5edf698SHong Zhang ierr = MatGetRowUpperTriangular(X);CHKERRQ(ierr); 1252f4df32b1SMatthew Knepley ierr = MatAXPY_Basic(Y,a,X,str);CHKERRQ(ierr); 1253f5edf698SHong Zhang ierr = MatRestoreRowUpperTriangular(X);CHKERRQ(ierr); 125442ee4b1aSHong Zhang } 125542ee4b1aSHong Zhang PetscFunctionReturn(0); 125642ee4b1aSHong Zhang } 125742ee4b1aSHong Zhang 1258efcf0fc3SBarry Smith #undef __FUNCT__ 12596363de48SJed Brown #define __FUNCT__ "MatSetBlockSize_SeqSBAIJ" 12606363de48SJed Brown PetscErrorCode MatSetBlockSize_SeqSBAIJ(Mat A,PetscInt bs) 12616363de48SJed Brown { 12626363de48SJed Brown PetscInt rbs,cbs; 12636363de48SJed Brown PetscErrorCode ierr; 12646363de48SJed Brown 12656363de48SJed Brown PetscFunctionBegin; 12666363de48SJed Brown ierr = PetscLayoutGetBlockSize(A->rmap,&rbs);CHKERRQ(ierr); 12676363de48SJed Brown ierr = PetscLayoutGetBlockSize(A->cmap,&cbs);CHKERRQ(ierr); 1268e32f2f54SBarry Smith if (rbs != bs) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Attempt to set block size %d with SBAIJ %d",bs,rbs); 1269e32f2f54SBarry Smith if (cbs != bs) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Attempt to set block size %d with SBAIJ %d",bs,cbs); 12706363de48SJed Brown PetscFunctionReturn(0); 12716363de48SJed Brown } 12726363de48SJed Brown 12736363de48SJed Brown #undef __FUNCT__ 1274efcf0fc3SBarry Smith #define __FUNCT__ "MatIsSymmetric_SeqSBAIJ" 1275ace3abfcSBarry Smith PetscErrorCode MatIsSymmetric_SeqSBAIJ(Mat A,PetscReal tol,PetscBool *flg) 1276efcf0fc3SBarry Smith { 1277efcf0fc3SBarry Smith PetscFunctionBegin; 1278efcf0fc3SBarry Smith *flg = PETSC_TRUE; 1279efcf0fc3SBarry Smith PetscFunctionReturn(0); 1280efcf0fc3SBarry Smith } 1281efcf0fc3SBarry Smith 1282efcf0fc3SBarry Smith #undef __FUNCT__ 1283efcf0fc3SBarry Smith #define __FUNCT__ "MatIsStructurallySymmetric_SeqSBAIJ" 1284ace3abfcSBarry Smith PetscErrorCode MatIsStructurallySymmetric_SeqSBAIJ(Mat A,PetscBool *flg) 1285efcf0fc3SBarry Smith { 1286efcf0fc3SBarry Smith PetscFunctionBegin; 1287efcf0fc3SBarry Smith *flg = PETSC_TRUE; 1288efcf0fc3SBarry Smith PetscFunctionReturn(0); 1289efcf0fc3SBarry Smith } 1290efcf0fc3SBarry Smith 1291efcf0fc3SBarry Smith #undef __FUNCT__ 1292efcf0fc3SBarry Smith #define __FUNCT__ "MatIsHermitian_SeqSBAIJ" 1293ace3abfcSBarry Smith PetscErrorCode MatIsHermitian_SeqSBAIJ(Mat A,PetscReal tol,PetscBool *flg) 1294efcf0fc3SBarry Smith { 1295efcf0fc3SBarry Smith PetscFunctionBegin; 1296efcf0fc3SBarry Smith *flg = PETSC_FALSE; 1297efcf0fc3SBarry Smith PetscFunctionReturn(0); 1298efcf0fc3SBarry Smith } 1299efcf0fc3SBarry Smith 130099cafbc1SBarry Smith #undef __FUNCT__ 130199cafbc1SBarry Smith #define __FUNCT__ "MatRealPart_SeqSBAIJ" 130299cafbc1SBarry Smith PetscErrorCode MatRealPart_SeqSBAIJ(Mat A) 130399cafbc1SBarry Smith { 130499cafbc1SBarry Smith Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 130599cafbc1SBarry Smith PetscInt i,nz = a->bs2*a->i[a->mbs]; 1306dd6ea824SBarry Smith MatScalar *aa = a->a; 130799cafbc1SBarry Smith 130899cafbc1SBarry Smith PetscFunctionBegin; 130999cafbc1SBarry Smith for (i=0; i<nz; i++) aa[i] = PetscRealPart(aa[i]); 131099cafbc1SBarry Smith PetscFunctionReturn(0); 131199cafbc1SBarry Smith } 131299cafbc1SBarry Smith 131399cafbc1SBarry Smith #undef __FUNCT__ 131499cafbc1SBarry Smith #define __FUNCT__ "MatImaginaryPart_SeqSBAIJ" 131599cafbc1SBarry Smith PetscErrorCode MatImaginaryPart_SeqSBAIJ(Mat A) 131699cafbc1SBarry Smith { 131799cafbc1SBarry Smith Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 131899cafbc1SBarry Smith PetscInt i,nz = a->bs2*a->i[a->mbs]; 1319dd6ea824SBarry Smith MatScalar *aa = a->a; 132099cafbc1SBarry Smith 132199cafbc1SBarry Smith PetscFunctionBegin; 132299cafbc1SBarry Smith for (i=0; i<nz; i++) aa[i] = PetscImaginaryPart(aa[i]); 132399cafbc1SBarry Smith PetscFunctionReturn(0); 132499cafbc1SBarry Smith } 132599cafbc1SBarry Smith 13263bededecSBarry Smith #undef __FUNCT__ 13273bededecSBarry Smith #define __FUNCT__ "MatZeroRowsColumns_SeqSBAIJ" 13283bededecSBarry Smith PetscErrorCode MatZeroRowsColumns_SeqSBAIJ(Mat A,PetscInt is_n,const PetscInt is_idx[],PetscScalar diag,Vec x, Vec b) 13293bededecSBarry Smith { 13303bededecSBarry Smith Mat_SeqSBAIJ *baij=(Mat_SeqSBAIJ*)A->data; 13313bededecSBarry Smith PetscErrorCode ierr; 13323bededecSBarry Smith PetscInt i,j,k,count; 13333bededecSBarry Smith PetscInt bs=A->rmap->bs,bs2=baij->bs2,row,col; 13343bededecSBarry Smith PetscScalar zero = 0.0; 13353bededecSBarry Smith MatScalar *aa; 13363bededecSBarry Smith const PetscScalar *xx; 13373bededecSBarry Smith PetscScalar *bb; 133856777dd2SBarry Smith PetscBool *zeroed,vecs = PETSC_FALSE; 13393bededecSBarry Smith 13403bededecSBarry Smith PetscFunctionBegin; 13413bededecSBarry Smith /* fix right hand side if needed */ 13423bededecSBarry Smith if (x && b) { 13433bededecSBarry Smith ierr = VecGetArrayRead(x,&xx);CHKERRQ(ierr); 13443bededecSBarry Smith ierr = VecGetArray(b,&bb);CHKERRQ(ierr); 134556777dd2SBarry Smith vecs = PETSC_TRUE; 13463bededecSBarry Smith } 13473bededecSBarry Smith A->same_nonzero = PETSC_TRUE; 13483bededecSBarry Smith 13493bededecSBarry Smith /* zero the columns */ 13503bededecSBarry Smith ierr = PetscMalloc(A->rmap->n*sizeof(PetscBool),&zeroed);CHKERRQ(ierr); 13513bededecSBarry Smith ierr = PetscMemzero(zeroed,A->rmap->n*sizeof(PetscBool));CHKERRQ(ierr); 13523bededecSBarry Smith for (i=0; i<is_n; i++) { 13533bededecSBarry 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]); 13543bededecSBarry Smith zeroed[is_idx[i]] = PETSC_TRUE; 13553bededecSBarry Smith } 135656777dd2SBarry Smith if (vecs) { 135756777dd2SBarry Smith for (i=0; i<A->rmap->N; i++) { 135856777dd2SBarry Smith row = i/bs; 135956777dd2SBarry Smith for (j=baij->i[row]; j<baij->i[row+1]; j++) { 136056777dd2SBarry Smith for (k=0; k<bs; k++) { 136156777dd2SBarry Smith col = bs*baij->j[j] + k; 136256777dd2SBarry Smith if (col <= i) continue; 136356777dd2SBarry Smith aa = ((MatScalar*)(baij->a)) + j*bs2 + (i%bs) + bs*k; 136456777dd2SBarry Smith if (!zeroed[i] && zeroed[col]) { 136556777dd2SBarry Smith bb[i] -= aa[0]*xx[col]; 136656777dd2SBarry Smith } 136756777dd2SBarry Smith if (zeroed[i] && !zeroed[col]) { 136856777dd2SBarry Smith bb[col] -= aa[0]*xx[i]; 136956777dd2SBarry Smith } 137056777dd2SBarry Smith } 137156777dd2SBarry Smith } 137256777dd2SBarry Smith } 137356777dd2SBarry Smith for (i=0; i<is_n; i++) { 137456777dd2SBarry Smith bb[is_idx[i]] = diag*xx[is_idx[i]]; 137556777dd2SBarry Smith } 137656777dd2SBarry Smith } 137756777dd2SBarry Smith 13783bededecSBarry Smith for (i=0; i<A->rmap->N; i++) { 13793bededecSBarry Smith if (!zeroed[i]) { 13803bededecSBarry Smith row = i/bs; 13813bededecSBarry Smith for (j=baij->i[row]; j<baij->i[row+1]; j++) { 13823bededecSBarry Smith for (k=0; k<bs; k++) { 13833bededecSBarry Smith col = bs*baij->j[j] + k; 13843bededecSBarry Smith if (zeroed[col]) { 13853bededecSBarry Smith aa = ((MatScalar*)(baij->a)) + j*bs2 + (i%bs) + bs*k; 13863bededecSBarry Smith aa[0] = 0.0; 13873bededecSBarry Smith } 13883bededecSBarry Smith } 13893bededecSBarry Smith } 13903bededecSBarry Smith } 13913bededecSBarry Smith } 13923bededecSBarry Smith ierr = PetscFree(zeroed);CHKERRQ(ierr); 139356777dd2SBarry Smith if (vecs) { 139456777dd2SBarry Smith ierr = VecRestoreArrayRead(x,&xx);CHKERRQ(ierr); 139556777dd2SBarry Smith ierr = VecRestoreArray(b,&bb);CHKERRQ(ierr); 139656777dd2SBarry Smith } 13973bededecSBarry Smith 13983bededecSBarry Smith /* zero the rows */ 13993bededecSBarry Smith for (i=0; i<is_n; i++) { 14003bededecSBarry Smith row = is_idx[i]; 14013bededecSBarry Smith count = (baij->i[row/bs +1] - baij->i[row/bs])*bs; 14023bededecSBarry Smith aa = ((MatScalar*)(baij->a)) + baij->i[row/bs]*bs2 + (row%bs); 14033bededecSBarry Smith for (k=0; k<count; k++) { 14043bededecSBarry Smith aa[0] = zero; 14053bededecSBarry Smith aa += bs; 14063bededecSBarry Smith } 14073bededecSBarry Smith if (diag != 0.0) { 14083bededecSBarry Smith ierr = (*A->ops->setvalues)(A,1,&row,1,&row,&diag,INSERT_VALUES);CHKERRQ(ierr); 14093bededecSBarry Smith } 14103bededecSBarry Smith } 14113bededecSBarry Smith ierr = MatAssemblyEnd_SeqSBAIJ(A,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 14123bededecSBarry Smith PetscFunctionReturn(0); 14133bededecSBarry Smith } 14143bededecSBarry Smith 141549b5e25fSSatish Balay /* -------------------------------------------------------------------*/ 141649b5e25fSSatish Balay static struct _MatOps MatOps_Values = {MatSetValues_SeqSBAIJ, 141749b5e25fSSatish Balay MatGetRow_SeqSBAIJ, 141849b5e25fSSatish Balay MatRestoreRow_SeqSBAIJ, 141949b5e25fSSatish Balay MatMult_SeqSBAIJ_N, 142097304618SKris Buschelman /* 4*/ MatMultAdd_SeqSBAIJ_N, 1421431c96f7SBarry Smith MatMult_SeqSBAIJ_N, /* transpose versions are same as non-transpose versions */ 1422e005ede5SBarry Smith MatMultAdd_SeqSBAIJ_N, 1423db4efbfdSBarry Smith 0, 142449b5e25fSSatish Balay 0, 142549b5e25fSSatish Balay 0, 142697304618SKris Buschelman /*10*/ 0, 142749b5e25fSSatish Balay 0, 1428c078aec8SLisandro Dalcin MatCholeskyFactor_SeqSBAIJ, 142941f059aeSBarry Smith MatSOR_SeqSBAIJ, 143049b5e25fSSatish Balay MatTranspose_SeqSBAIJ, 143197304618SKris Buschelman /*15*/ MatGetInfo_SeqSBAIJ, 143249b5e25fSSatish Balay MatEqual_SeqSBAIJ, 143349b5e25fSSatish Balay MatGetDiagonal_SeqSBAIJ, 143449b5e25fSSatish Balay MatDiagonalScale_SeqSBAIJ, 143549b5e25fSSatish Balay MatNorm_SeqSBAIJ, 143697304618SKris Buschelman /*20*/ 0, 143749b5e25fSSatish Balay MatAssemblyEnd_SeqSBAIJ, 143849b5e25fSSatish Balay MatSetOption_SeqSBAIJ, 143949b5e25fSSatish Balay MatZeroEntries_SeqSBAIJ, 1440d519adbfSMatthew Knepley /*24*/ 0, 144149b5e25fSSatish Balay 0, 144249b5e25fSSatish Balay 0, 1443db4efbfdSBarry Smith 0, 1444db4efbfdSBarry Smith 0, 14454994cf47SJed Brown /*29*/ MatSetUp_SeqSBAIJ, 1446c464158bSHong Zhang 0, 1447db4efbfdSBarry Smith 0, 1448a6ece127SHong Zhang MatGetArray_SeqSBAIJ, 1449a6ece127SHong Zhang MatRestoreArray_SeqSBAIJ, 1450d519adbfSMatthew Knepley /*34*/ MatDuplicate_SeqSBAIJ, 1451719d5645SBarry Smith 0, 1452719d5645SBarry Smith 0, 145349b5e25fSSatish Balay 0, 1454c84f5b01SHong Zhang MatICCFactor_SeqSBAIJ, 1455d519adbfSMatthew Knepley /*39*/ MatAXPY_SeqSBAIJ, 145649b5e25fSSatish Balay MatGetSubMatrices_SeqSBAIJ, 145749b5e25fSSatish Balay MatIncreaseOverlap_SeqSBAIJ, 145849b5e25fSSatish Balay MatGetValues_SeqSBAIJ, 14593c896bc6SHong Zhang MatCopy_SeqSBAIJ, 1460d519adbfSMatthew Knepley /*44*/ 0, 146149b5e25fSSatish Balay MatScale_SeqSBAIJ, 146249b5e25fSSatish Balay 0, 146349b5e25fSSatish Balay 0, 14643bededecSBarry Smith MatZeroRowsColumns_SeqSBAIJ, 14656363de48SJed Brown /*49*/ MatSetBlockSize_SeqSBAIJ, 146649b5e25fSSatish Balay MatGetRowIJ_SeqSBAIJ, 146749b5e25fSSatish Balay MatRestoreRowIJ_SeqSBAIJ, 146849b5e25fSSatish Balay 0, 146949b5e25fSSatish Balay 0, 1470d519adbfSMatthew Knepley /*54*/ 0, 147149b5e25fSSatish Balay 0, 147249b5e25fSSatish Balay 0, 147349b5e25fSSatish Balay 0, 147449b5e25fSSatish Balay MatSetValuesBlocked_SeqSBAIJ, 1475d519adbfSMatthew Knepley /*59*/ MatGetSubMatrix_SeqSBAIJ, 147649b5e25fSSatish Balay 0, 147749b5e25fSSatish Balay 0, 1478357abbc8SBarry Smith 0, 1479d959ec07SHong Zhang 0, 1480d519adbfSMatthew Knepley /*64*/ 0, 1481d959ec07SHong Zhang 0, 1482d959ec07SHong Zhang 0, 1483d959ec07SHong Zhang 0, 1484d959ec07SHong Zhang 0, 1485d519adbfSMatthew Knepley /*69*/ MatGetRowMaxAbs_SeqSBAIJ, 14863e0d88b5SBarry Smith 0, 14873e0d88b5SBarry Smith 0, 14883e0d88b5SBarry Smith 0, 14893e0d88b5SBarry Smith 0, 1490d519adbfSMatthew Knepley /*74*/ 0, 14913e0d88b5SBarry Smith 0, 14923e0d88b5SBarry Smith 0, 14933e0d88b5SBarry Smith 0, 14943e0d88b5SBarry Smith 0, 1495d519adbfSMatthew Knepley /*79*/ 0, 14963e0d88b5SBarry Smith 0, 14973e0d88b5SBarry Smith 0, 149897304618SKris Buschelman MatGetInertia_SeqSBAIJ, 14995bba2384SShri Abhyankar MatLoad_SeqSBAIJ, 1500d519adbfSMatthew Knepley /*84*/ MatIsSymmetric_SeqSBAIJ, 1501865e5f61SKris Buschelman MatIsHermitian_SeqSBAIJ, 1502efcf0fc3SBarry Smith MatIsStructurallySymmetric_SeqSBAIJ, 1503865e5f61SKris Buschelman 0, 1504865e5f61SKris Buschelman 0, 1505d519adbfSMatthew Knepley /*89*/ 0, 1506865e5f61SKris Buschelman 0, 1507865e5f61SKris Buschelman 0, 1508865e5f61SKris Buschelman 0, 1509865e5f61SKris Buschelman 0, 1510d519adbfSMatthew Knepley /*94*/ 0, 1511865e5f61SKris Buschelman 0, 1512865e5f61SKris Buschelman 0, 151399cafbc1SBarry Smith 0, 151499cafbc1SBarry Smith 0, 1515d519adbfSMatthew Knepley /*99*/ 0, 151699cafbc1SBarry Smith 0, 151799cafbc1SBarry Smith 0, 151899cafbc1SBarry Smith 0, 151999cafbc1SBarry Smith 0, 1520d519adbfSMatthew Knepley /*104*/0, 152199cafbc1SBarry Smith MatRealPart_SeqSBAIJ, 1522f5edf698SHong Zhang MatImaginaryPart_SeqSBAIJ, 1523f5edf698SHong Zhang MatGetRowUpperTriangular_SeqSBAIJ, 15242af78befSBarry Smith MatRestoreRowUpperTriangular_SeqSBAIJ, 1525d519adbfSMatthew Knepley /*109*/0, 15262af78befSBarry Smith 0, 15272af78befSBarry Smith 0, 15282af78befSBarry Smith 0, 1529547795f9SHong Zhang MatMissingDiagonal_SeqSBAIJ, 1530547795f9SHong Zhang /*114*/0, 1531547795f9SHong Zhang 0, 1532547795f9SHong Zhang 0, 1533547795f9SHong Zhang 0, 1534547795f9SHong Zhang 0, 1535547795f9SHong Zhang /*119*/0, 1536547795f9SHong Zhang 0, 15372f480046SShri Abhyankar 0, 15385bba2384SShri Abhyankar 0 153999cafbc1SBarry Smith }; 1540be1d678aSKris Buschelman 154149b5e25fSSatish Balay EXTERN_C_BEGIN 15424a2ae208SSatish Balay #undef __FUNCT__ 15434a2ae208SSatish Balay #define __FUNCT__ "MatStoreValues_SeqSBAIJ" 15447087cfbeSBarry Smith PetscErrorCode MatStoreValues_SeqSBAIJ(Mat mat) 154549b5e25fSSatish Balay { 15464afc71dfSHong Zhang Mat_SeqSBAIJ *aij = (Mat_SeqSBAIJ *)mat->data; 1547d0f46423SBarry Smith PetscInt nz = aij->i[mat->rmap->N]*mat->rmap->bs*aij->bs2; 1548dfbe8321SBarry Smith PetscErrorCode ierr; 154949b5e25fSSatish Balay 155049b5e25fSSatish Balay PetscFunctionBegin; 1551e7e72b3dSBarry Smith if (aij->nonew != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first"); 155249b5e25fSSatish Balay 155349b5e25fSSatish Balay /* allocate space for values if not already there */ 155449b5e25fSSatish Balay if (!aij->saved_values) { 155587828ca2SBarry Smith ierr = PetscMalloc((nz+1)*sizeof(PetscScalar),&aij->saved_values);CHKERRQ(ierr); 155649b5e25fSSatish Balay } 155749b5e25fSSatish Balay 155849b5e25fSSatish Balay /* copy values over */ 155987828ca2SBarry Smith ierr = PetscMemcpy(aij->saved_values,aij->a,nz*sizeof(PetscScalar));CHKERRQ(ierr); 156049b5e25fSSatish Balay PetscFunctionReturn(0); 156149b5e25fSSatish Balay } 156249b5e25fSSatish Balay EXTERN_C_END 156349b5e25fSSatish Balay 156449b5e25fSSatish Balay EXTERN_C_BEGIN 15654a2ae208SSatish Balay #undef __FUNCT__ 15664a2ae208SSatish Balay #define __FUNCT__ "MatRetrieveValues_SeqSBAIJ" 15677087cfbeSBarry Smith PetscErrorCode MatRetrieveValues_SeqSBAIJ(Mat mat) 156849b5e25fSSatish Balay { 15694afc71dfSHong Zhang Mat_SeqSBAIJ *aij = (Mat_SeqSBAIJ *)mat->data; 15706849ba73SBarry Smith PetscErrorCode ierr; 1571d0f46423SBarry Smith PetscInt nz = aij->i[mat->rmap->N]*mat->rmap->bs*aij->bs2; 157249b5e25fSSatish Balay 157349b5e25fSSatish Balay PetscFunctionBegin; 1574e7e72b3dSBarry Smith if (aij->nonew != 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NEW_NONZERO_LOCATIONS,PETSC_FALSE);first"); 1575e7e72b3dSBarry Smith if (!aij->saved_values) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ORDER,"Must call MatStoreValues(A);first"); 157649b5e25fSSatish Balay 157749b5e25fSSatish Balay /* copy values over */ 157887828ca2SBarry Smith ierr = PetscMemcpy(aij->a,aij->saved_values,nz*sizeof(PetscScalar));CHKERRQ(ierr); 157949b5e25fSSatish Balay PetscFunctionReturn(0); 158049b5e25fSSatish Balay } 158149b5e25fSSatish Balay EXTERN_C_END 158249b5e25fSSatish Balay 15838549e402SHong Zhang EXTERN_C_BEGIN 15844a2ae208SSatish Balay #undef __FUNCT__ 1585a23d5eceSKris Buschelman #define __FUNCT__ "MatSeqSBAIJSetPreallocation_SeqSBAIJ" 15867087cfbeSBarry Smith PetscErrorCode MatSeqSBAIJSetPreallocation_SeqSBAIJ(Mat B,PetscInt bs,PetscInt nz,PetscInt *nnz) 158749b5e25fSSatish Balay { 1588c464158bSHong Zhang Mat_SeqSBAIJ *b = (Mat_SeqSBAIJ*)B->data; 15896849ba73SBarry Smith PetscErrorCode ierr; 1590db4efbfdSBarry Smith PetscInt i,mbs,bs2, newbs = PetscAbs(bs); 15912576faa2SJed Brown PetscBool skipallocation = PETSC_FALSE,flg = PETSC_FALSE,realalloc = PETSC_FALSE; 159249b5e25fSSatish Balay 159349b5e25fSSatish Balay PetscFunctionBegin; 15942576faa2SJed Brown if (nz >= 0 || nnz) realalloc = PETSC_TRUE; 1595273d9f13SBarry Smith B->preallocated = PETSC_TRUE; 1596db4efbfdSBarry Smith if (bs < 0) { 1597db4efbfdSBarry Smith ierr = PetscOptionsBegin(((PetscObject)B)->comm,((PetscObject)B)->prefix,"Options for MPISBAIJ matrix","Mat");CHKERRQ(ierr); 1598db4efbfdSBarry Smith ierr = PetscOptionsInt("-mat_block_size","Set the blocksize used to store the matrix","MatSeqSBAIJSetPreallocation",newbs,&newbs,PETSC_NULL);CHKERRQ(ierr); 1599db4efbfdSBarry Smith ierr = PetscOptionsEnd();CHKERRQ(ierr); 1600db4efbfdSBarry Smith bs = PetscAbs(bs); 1601db4efbfdSBarry Smith } 1602e7e72b3dSBarry Smith if (nnz && newbs != bs) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Cannot change blocksize from command line if setting nnz"); 1603db4efbfdSBarry Smith bs = newbs; 1604db4efbfdSBarry Smith 160526283091SBarry Smith ierr = PetscLayoutSetBlockSize(B->rmap,bs);CHKERRQ(ierr); 160626283091SBarry Smith ierr = PetscLayoutSetBlockSize(B->cmap,bs);CHKERRQ(ierr); 160726283091SBarry Smith ierr = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr); 160826283091SBarry Smith ierr = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr); 1609899cda47SBarry Smith 1610d0f46423SBarry Smith mbs = B->rmap->N/bs; 161149b5e25fSSatish Balay bs2 = bs*bs; 161249b5e25fSSatish Balay 1613e7e72b3dSBarry Smith if (mbs*bs != B->rmap->N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Number rows, cols must be divisible by blocksize"); 161449b5e25fSSatish Balay 1615ab93d7beSBarry Smith if (nz == MAT_SKIP_ALLOCATION) { 1616ab93d7beSBarry Smith skipallocation = PETSC_TRUE; 1617ab93d7beSBarry Smith nz = 0; 1618ab93d7beSBarry Smith } 1619ab93d7beSBarry Smith 1620435da068SBarry Smith if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 3; 1621e32f2f54SBarry Smith if (nz < 0) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"nz cannot be less than 0: value %D",nz); 162249b5e25fSSatish Balay if (nnz) { 162349b5e25fSSatish Balay for (i=0; i<mbs; i++) { 1624e32f2f54SBarry 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]); 1625e32f2f54SBarry 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); 162649b5e25fSSatish Balay } 162749b5e25fSSatish Balay } 162849b5e25fSSatish Balay 1629db4efbfdSBarry Smith B->ops->mult = MatMult_SeqSBAIJ_N; 1630db4efbfdSBarry Smith B->ops->multadd = MatMultAdd_SeqSBAIJ_N; 1631db4efbfdSBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_N; 1632db4efbfdSBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_N; 1633acfcf0e5SJed Brown ierr = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_no_unroll",&flg,PETSC_NULL);CHKERRQ(ierr); 163449b5e25fSSatish Balay if (!flg) { 163549b5e25fSSatish Balay switch (bs) { 163649b5e25fSSatish Balay case 1: 163749b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_1; 163849b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_1; 1639431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_1; 1640431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_1; 164149b5e25fSSatish Balay break; 164249b5e25fSSatish Balay case 2: 164349b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_2; 164449b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_2; 1645431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_2; 1646431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_2; 164749b5e25fSSatish Balay break; 164849b5e25fSSatish Balay case 3: 164949b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_3; 165049b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_3; 1651431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_3; 1652431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_3; 165349b5e25fSSatish Balay break; 165449b5e25fSSatish Balay case 4: 165549b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_4; 165649b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_4; 1657431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_4; 1658431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_4; 165949b5e25fSSatish Balay break; 166049b5e25fSSatish Balay case 5: 166149b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_5; 166249b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_5; 1663431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_5; 1664431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_5; 166549b5e25fSSatish Balay break; 166649b5e25fSSatish Balay case 6: 166749b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_6; 166849b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_6; 1669431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_6; 1670431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_6; 167149b5e25fSSatish Balay break; 167249b5e25fSSatish Balay case 7: 1673de53e5efSHong Zhang B->ops->mult = MatMult_SeqSBAIJ_7; 167449b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_7; 1675431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_7; 1676431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_7; 167749b5e25fSSatish Balay break; 167849b5e25fSSatish Balay } 167949b5e25fSSatish Balay } 168049b5e25fSSatish Balay 168149b5e25fSSatish Balay b->mbs = mbs; 16824afc71dfSHong Zhang b->nbs = mbs; 1683ab93d7beSBarry Smith if (!skipallocation) { 16842ee49352SLisandro Dalcin if (!b->imax) { 1685ab93d7beSBarry Smith ierr = PetscMalloc2(mbs,PetscInt,&b->imax,mbs,PetscInt,&b->ilen);CHKERRQ(ierr); 1686c760cd28SBarry Smith b->free_imax_ilen = PETSC_TRUE; 16872ee49352SLisandro Dalcin ierr = PetscLogObjectMemory(B,2*mbs*sizeof(PetscInt));CHKERRQ(ierr); 16882ee49352SLisandro Dalcin } 168949b5e25fSSatish Balay if (!nnz) { 1690435da068SBarry Smith if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 5; 169149b5e25fSSatish Balay else if (nz <= 0) nz = 1; 169249b5e25fSSatish Balay for (i=0; i<mbs; i++) { 16938cef66ccSHong Zhang b->imax[i] = nz; 169449b5e25fSSatish Balay } 1695153ea458SHong Zhang nz = nz*mbs; /* total nz */ 169649b5e25fSSatish Balay } else { 169749b5e25fSSatish Balay nz = 0; 16988cef66ccSHong Zhang for (i=0; i<mbs; i++) {b->imax[i] = nnz[i]; nz += nnz[i];} 169949b5e25fSSatish Balay } 17002ee49352SLisandro Dalcin /* b->ilen will count nonzeros in each block row so far. */ 17012ee49352SLisandro Dalcin for (i=0; i<mbs; i++) { b->ilen[i] = 0;} 17026c6c5352SBarry Smith /* nz=(nz+mbs)/2; */ /* total diagonal and superdiagonal nonzero blocks */ 170349b5e25fSSatish Balay 170449b5e25fSSatish Balay /* allocate the matrix space */ 17052ee49352SLisandro Dalcin ierr = MatSeqXAIJFreeAIJ(B,&b->a,&b->j,&b->i);CHKERRQ(ierr); 1706d0f46423SBarry Smith ierr = PetscMalloc3(bs2*nz,PetscScalar,&b->a,nz,PetscInt,&b->j,B->rmap->N+1,PetscInt,&b->i);CHKERRQ(ierr); 1707d0f46423SBarry Smith ierr = PetscLogObjectMemory(B,(B->rmap->N+1)*sizeof(PetscInt)+nz*(bs2*sizeof(PetscScalar)+sizeof(PetscInt)));CHKERRQ(ierr); 17086c6c5352SBarry Smith ierr = PetscMemzero(b->a,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr); 170913f74950SBarry Smith ierr = PetscMemzero(b->j,nz*sizeof(PetscInt));CHKERRQ(ierr); 171049b5e25fSSatish Balay b->singlemalloc = PETSC_TRUE; 171149b5e25fSSatish Balay 171249b5e25fSSatish Balay /* pointer to beginning of each row */ 1713e60cf9a0SBarry Smith b->i[0] = 0; 171449b5e25fSSatish Balay for (i=1; i<mbs+1; i++) { 171549b5e25fSSatish Balay b->i[i] = b->i[i-1] + b->imax[i-1]; 171649b5e25fSSatish Balay } 1717e6b907acSBarry Smith b->free_a = PETSC_TRUE; 1718e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 1719e811da20SHong Zhang } else { 1720e6b907acSBarry Smith b->free_a = PETSC_FALSE; 1721e6b907acSBarry Smith b->free_ij = PETSC_FALSE; 1722ab93d7beSBarry Smith } 172349b5e25fSSatish Balay 1724d0f46423SBarry Smith B->rmap->bs = bs; 172549b5e25fSSatish Balay b->bs2 = bs2; 17266c6c5352SBarry Smith b->nz = 0; 1727b32cb4a7SJed Brown b->maxnz = nz; 1728153ea458SHong Zhang 172916cdd363SHong Zhang b->inew = 0; 173016cdd363SHong Zhang b->jnew = 0; 173116cdd363SHong Zhang b->anew = 0; 173216cdd363SHong Zhang b->a2anew = 0; 17331a3463dfSHong Zhang b->permute = PETSC_FALSE; 17342576faa2SJed Brown if (realalloc) {ierr = MatSetOption(B,MAT_NEW_NONZERO_ALLOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr);} 1735c464158bSHong Zhang PetscFunctionReturn(0); 1736c464158bSHong Zhang } 1737a23d5eceSKris Buschelman EXTERN_C_END 1738153ea458SHong Zhang 1739db4efbfdSBarry Smith /* 1740db4efbfdSBarry Smith This is used to set the numeric factorization for both Cholesky and ICC symbolic factorization 1741db4efbfdSBarry Smith */ 17428b1456e3SHong Zhang #undef __FUNCT__ 1743d595f711SHong Zhang #define __FUNCT__ "MatSeqSBAIJSetNumericFactorization_inplace" 1744ace3abfcSBarry Smith PetscErrorCode MatSeqSBAIJSetNumericFactorization_inplace(Mat B,PetscBool natural) 1745db4efbfdSBarry Smith { 1746db4efbfdSBarry Smith PetscErrorCode ierr; 1747ace3abfcSBarry Smith PetscBool flg = PETSC_FALSE; 1748db4efbfdSBarry Smith PetscInt bs = B->rmap->bs; 1749db4efbfdSBarry Smith 1750db4efbfdSBarry Smith PetscFunctionBegin; 1751acfcf0e5SJed Brown ierr = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_no_unroll",&flg,PETSC_NULL);CHKERRQ(ierr); 1752db4efbfdSBarry Smith if (flg) bs = 8; 1753db4efbfdSBarry Smith 1754db4efbfdSBarry Smith if (!natural) { 1755db4efbfdSBarry Smith switch (bs) { 1756db4efbfdSBarry Smith case 1: 1757d595f711SHong Zhang B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_inplace; 1758db4efbfdSBarry Smith break; 1759db4efbfdSBarry Smith case 2: 1760db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2; 1761db4efbfdSBarry Smith break; 1762db4efbfdSBarry Smith case 3: 1763db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3; 1764db4efbfdSBarry Smith break; 1765db4efbfdSBarry Smith case 4: 1766db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4; 1767db4efbfdSBarry Smith break; 1768db4efbfdSBarry Smith case 5: 1769db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5; 1770db4efbfdSBarry Smith break; 1771db4efbfdSBarry Smith case 6: 1772db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6; 1773db4efbfdSBarry Smith break; 1774db4efbfdSBarry Smith case 7: 1775db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7; 1776db4efbfdSBarry Smith break; 1777db4efbfdSBarry Smith default: 1778db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N; 1779db4efbfdSBarry Smith break; 1780db4efbfdSBarry Smith } 1781db4efbfdSBarry Smith } else { 1782db4efbfdSBarry Smith switch (bs) { 1783db4efbfdSBarry Smith case 1: 1784d595f711SHong Zhang B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering_inplace; 1785db4efbfdSBarry Smith break; 1786db4efbfdSBarry Smith case 2: 1787db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering; 1788db4efbfdSBarry Smith break; 1789db4efbfdSBarry Smith case 3: 1790db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering; 1791db4efbfdSBarry Smith break; 1792db4efbfdSBarry Smith case 4: 1793db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering; 1794db4efbfdSBarry Smith break; 1795db4efbfdSBarry Smith case 5: 1796db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering; 1797db4efbfdSBarry Smith break; 1798db4efbfdSBarry Smith case 6: 1799db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering; 1800db4efbfdSBarry Smith break; 1801db4efbfdSBarry Smith case 7: 1802db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering; 1803db4efbfdSBarry Smith break; 1804db4efbfdSBarry Smith default: 1805db4efbfdSBarry Smith B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering; 1806db4efbfdSBarry Smith break; 1807db4efbfdSBarry Smith } 1808db4efbfdSBarry Smith } 1809db4efbfdSBarry Smith PetscFunctionReturn(0); 1810db4efbfdSBarry Smith } 1811db4efbfdSBarry Smith 1812d769727bSBarry Smith EXTERN_C_BEGIN 18137087cfbeSBarry Smith extern PetscErrorCode MatConvert_SeqSBAIJ_SeqAIJ(Mat, MatType,MatReuse,Mat*); 18147087cfbeSBarry Smith extern PetscErrorCode MatConvert_SeqSBAIJ_SeqBAIJ(Mat, MatType,MatReuse,Mat*); 1815d769727bSBarry Smith EXTERN_C_END 1816d769727bSBarry Smith 1817e631078cSBarry Smith 1818e631078cSBarry Smith EXTERN_C_BEGIN 18195c9eb25fSBarry Smith #undef __FUNCT__ 18205c9eb25fSBarry Smith #define __FUNCT__ "MatGetFactor_seqsbaij_petsc" 18215c9eb25fSBarry Smith PetscErrorCode MatGetFactor_seqsbaij_petsc(Mat A,MatFactorType ftype,Mat *B) 18225c9eb25fSBarry Smith { 1823d0f46423SBarry Smith PetscInt n = A->rmap->n; 18245c9eb25fSBarry Smith PetscErrorCode ierr; 18255c9eb25fSBarry Smith 18265c9eb25fSBarry Smith PetscFunctionBegin; 18275c9eb25fSBarry Smith ierr = MatCreate(((PetscObject)A)->comm,B);CHKERRQ(ierr); 18285c9eb25fSBarry Smith ierr = MatSetSizes(*B,n,n,n,n);CHKERRQ(ierr); 18295c9eb25fSBarry Smith if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) { 18305c9eb25fSBarry Smith ierr = MatSetType(*B,MATSEQSBAIJ);CHKERRQ(ierr); 18315c9eb25fSBarry Smith ierr = MatSeqSBAIJSetPreallocation(*B,1,MAT_SKIP_ALLOCATION,PETSC_NULL);CHKERRQ(ierr); 18327b056e98SHong Zhang (*B)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqSBAIJ; 1833c6d0d4f0SHong Zhang (*B)->ops->iccfactorsymbolic = MatICCFactorSymbolic_SeqSBAIJ; 1834e32f2f54SBarry Smith } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Factor type not supported"); 1835d5f3da31SBarry Smith (*B)->factortype = ftype; 18365c9eb25fSBarry Smith PetscFunctionReturn(0); 18375c9eb25fSBarry Smith } 1838e631078cSBarry Smith EXTERN_C_END 18395c9eb25fSBarry Smith 18405c9eb25fSBarry Smith EXTERN_C_BEGIN 1841db4efbfdSBarry Smith #undef __FUNCT__ 1842db4efbfdSBarry Smith #define __FUNCT__ "MatGetFactorAvailable_seqsbaij_petsc" 1843ace3abfcSBarry Smith PetscErrorCode MatGetFactorAvailable_seqsbaij_petsc(Mat A,MatFactorType ftype,PetscBool *flg) 1844db4efbfdSBarry Smith { 1845db4efbfdSBarry Smith PetscFunctionBegin; 1846db4efbfdSBarry Smith if (ftype == MAT_FACTOR_CHOLESKY || ftype == MAT_FACTOR_ICC) { 1847db4efbfdSBarry Smith *flg = PETSC_TRUE; 1848db4efbfdSBarry Smith } else { 1849db4efbfdSBarry Smith *flg = PETSC_FALSE; 1850db4efbfdSBarry Smith } 1851db4efbfdSBarry Smith PetscFunctionReturn(0); 1852db4efbfdSBarry Smith } 1853db4efbfdSBarry Smith EXTERN_C_END 1854db4efbfdSBarry Smith 1855db4efbfdSBarry Smith EXTERN_C_BEGIN 1856611f576cSBarry Smith #if defined(PETSC_HAVE_MUMPS) 1857bccb9932SShri Abhyankar extern PetscErrorCode MatGetFactor_sbaij_mumps(Mat,MatFactorType,Mat*); 1858611f576cSBarry Smith #endif 1859611f576cSBarry Smith #if defined(PETSC_HAVE_SPOOLES) 18605c9eb25fSBarry Smith extern PetscErrorCode MatGetFactor_seqsbaij_spooles(Mat,MatFactorType,Mat*); 1861611f576cSBarry Smith #endif 1862b5e56a35SBarry Smith #if defined(PETSC_HAVE_PASTIX) 1863b5e56a35SBarry Smith extern PetscErrorCode MatGetFactor_seqsbaij_pastix(Mat,MatFactorType,Mat*); 1864b5e56a35SBarry Smith #endif 186520db9a53SJed Brown #if defined(PETSC_HAVE_CHOLMOD) 186620db9a53SJed Brown extern PetscErrorCode MatGetFactor_seqsbaij_cholmod(Mat,MatFactorType,Mat*); 186720db9a53SJed Brown #endif 18682938c13dSDahai Guo extern PetscErrorCode MatGetFactor_seqsbaij_sbstrm(Mat,MatFactorType,Mat*); 1869b1f23a54SSatish Balay EXTERN_C_END 18705c9eb25fSBarry Smith 18710bad9183SKris Buschelman /*MC 1872fafad747SKris Buschelman MATSEQSBAIJ - MATSEQSBAIJ = "seqsbaij" - A matrix type to be used for sequential symmetric block sparse matrices, 18730bad9183SKris Buschelman based on block compressed sparse row format. Only the upper triangular portion of the matrix is stored. 18740bad9183SKris Buschelman 1875828413b8SBarry Smith For complex numbers by default this matrix is symmetric, NOT Hermitian symmetric. To make it Hermitian symmetric you 187671dad5bbSBarry Smith can call MatSetOption(Mat, MAT_HERMITIAN); after MatAssemblyEnd() 1877828413b8SBarry Smith 18780bad9183SKris Buschelman Options Database Keys: 18790bad9183SKris Buschelman . -mat_type seqsbaij - sets the matrix type to "seqsbaij" during a call to MatSetFromOptions() 18800bad9183SKris Buschelman 188171dad5bbSBarry Smith Notes: By default if you insert values into the lower triangular part of the matrix they are simply ignored (since they are not 188271dad5bbSBarry 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 188371dad5bbSBarry 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. 188471dad5bbSBarry Smith 188571dad5bbSBarry Smith 18860bad9183SKris Buschelman Level: beginner 18870bad9183SKris Buschelman 18880bad9183SKris Buschelman .seealso: MatCreateSeqSBAIJ 18890bad9183SKris Buschelman M*/ 18900bad9183SKris Buschelman 1891a23d5eceSKris Buschelman EXTERN_C_BEGIN 1892aa5a9175SDahai Guo extern PetscErrorCode MatConvert_SeqSBAIJ_SeqSBSTRM(Mat, MatType,MatReuse,Mat*); 1893aa5a9175SDahai Guo EXTERN_C_END 1894aa5a9175SDahai Guo 1895aa5a9175SDahai Guo 1896aa5a9175SDahai Guo EXTERN_C_BEGIN 1897a23d5eceSKris Buschelman #undef __FUNCT__ 1898a23d5eceSKris Buschelman #define __FUNCT__ "MatCreate_SeqSBAIJ" 18997087cfbeSBarry Smith PetscErrorCode MatCreate_SeqSBAIJ(Mat B) 1900a23d5eceSKris Buschelman { 1901a23d5eceSKris Buschelman Mat_SeqSBAIJ *b; 1902dfbe8321SBarry Smith PetscErrorCode ierr; 190313f74950SBarry Smith PetscMPIInt size; 1904ace3abfcSBarry Smith PetscBool no_unroll = PETSC_FALSE,no_inode = PETSC_FALSE; 1905a23d5eceSKris Buschelman 1906a23d5eceSKris Buschelman PetscFunctionBegin; 19077adad957SLisandro Dalcin ierr = MPI_Comm_size(((PetscObject)B)->comm,&size);CHKERRQ(ierr); 1908e32f2f54SBarry Smith if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Comm must be of size 1"); 1909a23d5eceSKris Buschelman 191038f2d2fdSLisandro Dalcin ierr = PetscNewLog(B,Mat_SeqSBAIJ,&b);CHKERRQ(ierr); 1911a23d5eceSKris Buschelman B->data = (void*)b; 1912a23d5eceSKris Buschelman ierr = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr); 1913a23d5eceSKris Buschelman B->ops->destroy = MatDestroy_SeqSBAIJ; 1914a23d5eceSKris Buschelman B->ops->view = MatView_SeqSBAIJ; 1915a23d5eceSKris Buschelman b->row = 0; 1916a23d5eceSKris Buschelman b->icol = 0; 1917a23d5eceSKris Buschelman b->reallocs = 0; 1918a23d5eceSKris Buschelman b->saved_values = 0; 19190def2e27SBarry Smith b->inode.limit = 5; 19200def2e27SBarry Smith b->inode.max_limit = 5; 1921a23d5eceSKris Buschelman 1922a23d5eceSKris Buschelman b->roworiented = PETSC_TRUE; 1923a23d5eceSKris Buschelman b->nonew = 0; 1924a23d5eceSKris Buschelman b->diag = 0; 1925a23d5eceSKris Buschelman b->solve_work = 0; 1926a23d5eceSKris Buschelman b->mult_work = 0; 1927a23d5eceSKris Buschelman B->spptr = 0; 1928f2cbd3d5SJed Brown B->info.nz_unneeded = (PetscReal)b->maxnz*b->bs2; 1929a9817697SBarry Smith b->keepnonzeropattern = PETSC_FALSE; 1930a23d5eceSKris Buschelman b->xtoy = 0; 1931a23d5eceSKris Buschelman b->XtoY = 0; 1932a23d5eceSKris Buschelman 1933a23d5eceSKris Buschelman b->inew = 0; 1934a23d5eceSKris Buschelman b->jnew = 0; 1935a23d5eceSKris Buschelman b->anew = 0; 1936a23d5eceSKris Buschelman b->a2anew = 0; 1937a23d5eceSKris Buschelman b->permute = PETSC_FALSE; 1938a23d5eceSKris Buschelman 193971dad5bbSBarry Smith b->ignore_ltriangular = PETSC_TRUE; 1940acfcf0e5SJed Brown ierr = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_ignore_lower_triangular",&b->ignore_ltriangular,PETSC_NULL);CHKERRQ(ierr); 1941941593c8SHong Zhang 1942f5edf698SHong Zhang b->getrow_utriangular = PETSC_FALSE; 1943acfcf0e5SJed Brown ierr = PetscOptionsGetBool(((PetscObject)B)->prefix,"-mat_getrow_uppertriangular",&b->getrow_utriangular,PETSC_NULL);CHKERRQ(ierr); 1944f5edf698SHong Zhang 1945b5e56a35SBarry Smith #if defined(PETSC_HAVE_PASTIX) 1946ec1065edSBarry Smith ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatGetFactor_pastix_C", 1947b5e56a35SBarry Smith "MatGetFactor_seqsbaij_pastix", 1948b5e56a35SBarry Smith MatGetFactor_seqsbaij_pastix);CHKERRQ(ierr); 1949b5e56a35SBarry Smith #endif 1950611f576cSBarry Smith #if defined(PETSC_HAVE_SPOOLES) 1951ec1065edSBarry Smith ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatGetFactor_spooles_C", 19525c9eb25fSBarry Smith "MatGetFactor_seqsbaij_spooles", 19535c9eb25fSBarry Smith MatGetFactor_seqsbaij_spooles);CHKERRQ(ierr); 1954611f576cSBarry Smith #endif 1955611f576cSBarry Smith #if defined(PETSC_HAVE_MUMPS) 1956ec1065edSBarry Smith ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatGetFactor_mumps_C", 1957bccb9932SShri Abhyankar "MatGetFactor_sbaij_mumps", 1958bccb9932SShri Abhyankar MatGetFactor_sbaij_mumps);CHKERRQ(ierr); 1959611f576cSBarry Smith #endif 196020db9a53SJed Brown #if defined(PETSC_HAVE_CHOLMOD) 196120db9a53SJed Brown ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatGetFactor_cholmod_C", 196220db9a53SJed Brown "MatGetFactor_seqsbaij_cholmod", 196320db9a53SJed Brown MatGetFactor_seqsbaij_cholmod);CHKERRQ(ierr); 196420db9a53SJed Brown #endif 1965ec1065edSBarry Smith ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatGetFactorAvailable_petsc_C", 1966db4efbfdSBarry Smith "MatGetFactorAvailable_seqsbaij_petsc", 1967db4efbfdSBarry Smith MatGetFactorAvailable_seqsbaij_petsc);CHKERRQ(ierr); 1968ec1065edSBarry Smith ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatGetFactor_petsc_C", 19695c9eb25fSBarry Smith "MatGetFactor_seqsbaij_petsc", 19705c9eb25fSBarry Smith MatGetFactor_seqsbaij_petsc);CHKERRQ(ierr); 19713edee7c7SSatish Balay ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatGetFactor_sbstrm_C", 19722938c13dSDahai Guo "MatGetFactor_seqsbaij_sbstrm", 19732938c13dSDahai Guo MatGetFactor_seqsbaij_sbstrm);CHKERRQ(ierr); 1974a23d5eceSKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatStoreValues_C", 1975a23d5eceSKris Buschelman "MatStoreValues_SeqSBAIJ", 1976a23d5eceSKris Buschelman MatStoreValues_SeqSBAIJ);CHKERRQ(ierr); 1977a23d5eceSKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatRetrieveValues_C", 1978a23d5eceSKris Buschelman "MatRetrieveValues_SeqSBAIJ", 1979a23d5eceSKris Buschelman (void*)MatRetrieveValues_SeqSBAIJ);CHKERRQ(ierr); 1980a23d5eceSKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatSeqSBAIJSetColumnIndices_C", 1981a23d5eceSKris Buschelman "MatSeqSBAIJSetColumnIndices_SeqSBAIJ", 1982a23d5eceSKris Buschelman MatSeqSBAIJSetColumnIndices_SeqSBAIJ);CHKERRQ(ierr); 19834e5e7fe4SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_seqsbaij_seqaij_C", 19844e5e7fe4SHong Zhang "MatConvert_SeqSBAIJ_SeqAIJ", 19854e5e7fe4SHong Zhang MatConvert_SeqSBAIJ_SeqAIJ);CHKERRQ(ierr); 1986a0e1a404SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_seqsbaij_seqbaij_C", 1987a0e1a404SHong Zhang "MatConvert_SeqSBAIJ_SeqBAIJ", 1988a0e1a404SHong Zhang MatConvert_SeqSBAIJ_SeqBAIJ);CHKERRQ(ierr); 1989a23d5eceSKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatSeqSBAIJSetPreallocation_C", 1990a23d5eceSKris Buschelman "MatSeqSBAIJSetPreallocation_SeqSBAIJ", 1991a23d5eceSKris Buschelman MatSeqSBAIJSetPreallocation_SeqSBAIJ);CHKERRQ(ierr); 1992aa5a9175SDahai Guo ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_seqsbaij_seqsbstrm_C", 1993aa5a9175SDahai Guo "MatConvert_SeqSBAIJ_SeqSBSTRM", 1994aa5a9175SDahai Guo MatConvert_SeqSBAIJ_SeqSBSTRM);CHKERRQ(ierr); 199523ce1328SBarry Smith 199623ce1328SBarry Smith B->symmetric = PETSC_TRUE; 199723ce1328SBarry Smith B->structurally_symmetric = PETSC_TRUE; 199823ce1328SBarry Smith B->symmetric_set = PETSC_TRUE; 199923ce1328SBarry Smith B->structurally_symmetric_set = PETSC_TRUE; 200017667f90SBarry Smith ierr = PetscObjectChangeTypeName((PetscObject)B,MATSEQSBAIJ);CHKERRQ(ierr); 20010def2e27SBarry Smith 20020def2e27SBarry Smith ierr = PetscOptionsBegin(((PetscObject)B)->comm,((PetscObject)B)->prefix,"Options for SEQSBAIJ matrix","Mat");CHKERRQ(ierr); 2003acfcf0e5SJed Brown ierr = PetscOptionsBool("-mat_no_unroll","Do not optimize for inodes (slower)",PETSC_NULL,no_unroll,&no_unroll,PETSC_NULL);CHKERRQ(ierr); 20040def2e27SBarry Smith if (no_unroll) {ierr = PetscInfo(B,"Not using Inode routines due to -mat_no_unroll\n");CHKERRQ(ierr);} 2005acfcf0e5SJed Brown ierr = PetscOptionsBool("-mat_no_inode","Do not optimize for inodes (slower)",PETSC_NULL,no_inode,&no_inode,PETSC_NULL);CHKERRQ(ierr); 20060def2e27SBarry Smith if (no_inode) {ierr = PetscInfo(B,"Not using Inode routines due to -mat_no_inode\n");CHKERRQ(ierr);} 20070def2e27SBarry Smith ierr = PetscOptionsInt("-mat_inode_limit","Do not use inodes larger then this value",PETSC_NULL,b->inode.limit,&b->inode.limit,PETSC_NULL);CHKERRQ(ierr); 20080def2e27SBarry Smith ierr = PetscOptionsEnd();CHKERRQ(ierr); 2009ace3abfcSBarry Smith b->inode.use = (PetscBool)(!(no_unroll || no_inode)); 20100def2e27SBarry Smith if (b->inode.limit > b->inode.max_limit) b->inode.limit = b->inode.max_limit; 20110def2e27SBarry Smith 2012a23d5eceSKris Buschelman PetscFunctionReturn(0); 2013a23d5eceSKris Buschelman } 2014a23d5eceSKris Buschelman EXTERN_C_END 2015a23d5eceSKris Buschelman 2016a23d5eceSKris Buschelman #undef __FUNCT__ 2017a23d5eceSKris Buschelman #define __FUNCT__ "MatSeqSBAIJSetPreallocation" 2018a23d5eceSKris Buschelman /*@C 2019a23d5eceSKris Buschelman MatSeqSBAIJSetPreallocation - Creates a sparse symmetric matrix in block AIJ (block 2020a23d5eceSKris Buschelman compressed row) format. For good matrix assembly performance the 2021a23d5eceSKris Buschelman user should preallocate the matrix storage by setting the parameter nz 2022a23d5eceSKris Buschelman (or the array nnz). By setting these parameters accurately, performance 2023a23d5eceSKris Buschelman during matrix assembly can be increased by more than a factor of 50. 2024a23d5eceSKris Buschelman 2025a23d5eceSKris Buschelman Collective on Mat 2026a23d5eceSKris Buschelman 2027a23d5eceSKris Buschelman Input Parameters: 2028a23d5eceSKris Buschelman + A - the symmetric matrix 2029a23d5eceSKris Buschelman . bs - size of block 2030a23d5eceSKris Buschelman . nz - number of block nonzeros per block row (same for all rows) 2031a23d5eceSKris Buschelman - nnz - array containing the number of block nonzeros in the upper triangular plus 2032a23d5eceSKris Buschelman diagonal portion of each block (possibly different for each block row) or PETSC_NULL 2033a23d5eceSKris Buschelman 2034a23d5eceSKris Buschelman Options Database Keys: 2035a23d5eceSKris Buschelman . -mat_no_unroll - uses code that does not unroll the loops in the 2036a23d5eceSKris Buschelman block calculations (much slower) 2037db4efbfdSBarry Smith . -mat_block_size - size of the blocks to use (only works if a negative bs is passed in 2038a23d5eceSKris Buschelman 2039a23d5eceSKris Buschelman Level: intermediate 2040a23d5eceSKris Buschelman 2041a23d5eceSKris Buschelman Notes: 2042a23d5eceSKris Buschelman Specify the preallocated storage with either nz or nnz (not both). 2043a23d5eceSKris Buschelman Set nz=PETSC_DEFAULT and nnz=PETSC_NULL for PETSc to control dynamic memory 20440598bfebSBarry Smith allocation. See the <a href="../../docs/manual.pdf#nameddest=ch_mat">Mat chapter of the users manual</a> for details. 2045a23d5eceSKris Buschelman 2046aa95bbe8SBarry Smith You can call MatGetInfo() to get information on how effective the preallocation was; 2047aa95bbe8SBarry Smith for example the fields mallocs,nz_allocated,nz_used,nz_unneeded; 2048aa95bbe8SBarry Smith You can also run with the option -info and look for messages with the string 2049aa95bbe8SBarry Smith malloc in them to see if additional memory allocation was needed. 2050aa95bbe8SBarry Smith 205149a6f317SBarry Smith If the nnz parameter is given then the nz parameter is ignored 205249a6f317SBarry Smith 205349a6f317SBarry Smith 2054*69b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateSBAIJ() 2055a23d5eceSKris Buschelman @*/ 20567087cfbeSBarry Smith PetscErrorCode MatSeqSBAIJSetPreallocation(Mat B,PetscInt bs,PetscInt nz,const PetscInt nnz[]) 205713f74950SBarry Smith { 20584ac538c5SBarry Smith PetscErrorCode ierr; 2059a23d5eceSKris Buschelman 2060a23d5eceSKris Buschelman PetscFunctionBegin; 20616ba663aaSJed Brown PetscValidHeaderSpecific(B,MAT_CLASSID,1); 20626ba663aaSJed Brown PetscValidType(B,1); 20636ba663aaSJed Brown PetscValidLogicalCollectiveInt(B,bs,2); 20644ac538c5SBarry Smith ierr = PetscTryMethod(B,"MatSeqSBAIJSetPreallocation_C",(Mat,PetscInt,PetscInt,const PetscInt[]),(B,bs,nz,nnz));CHKERRQ(ierr); 2065a23d5eceSKris Buschelman PetscFunctionReturn(0); 2066a23d5eceSKris Buschelman } 206749b5e25fSSatish Balay 20684a2ae208SSatish Balay #undef __FUNCT__ 20694a2ae208SSatish Balay #define __FUNCT__ "MatCreateSeqSBAIJ" 2070c464158bSHong Zhang /*@C 2071c464158bSHong Zhang MatCreateSeqSBAIJ - Creates a sparse symmetric matrix in block AIJ (block 2072c464158bSHong Zhang compressed row) format. For good matrix assembly performance the 2073c464158bSHong Zhang user should preallocate the matrix storage by setting the parameter nz 2074c464158bSHong Zhang (or the array nnz). By setting these parameters accurately, performance 2075c464158bSHong Zhang during matrix assembly can be increased by more than a factor of 50. 207649b5e25fSSatish Balay 2077c464158bSHong Zhang Collective on MPI_Comm 2078c464158bSHong Zhang 2079c464158bSHong Zhang Input Parameters: 2080c464158bSHong Zhang + comm - MPI communicator, set to PETSC_COMM_SELF 2081c464158bSHong Zhang . bs - size of block 2082c464158bSHong Zhang . m - number of rows, or number of columns 2083c464158bSHong Zhang . nz - number of block nonzeros per block row (same for all rows) 2084744e8345SSatish Balay - nnz - array containing the number of block nonzeros in the upper triangular plus 2085744e8345SSatish Balay diagonal portion of each block (possibly different for each block row) or PETSC_NULL 2086c464158bSHong Zhang 2087c464158bSHong Zhang Output Parameter: 2088c464158bSHong Zhang . A - the symmetric matrix 2089c464158bSHong Zhang 2090c464158bSHong Zhang Options Database Keys: 2091c464158bSHong Zhang . -mat_no_unroll - uses code that does not unroll the loops in the 2092c464158bSHong Zhang block calculations (much slower) 2093c464158bSHong Zhang . -mat_block_size - size of the blocks to use 2094c464158bSHong Zhang 2095c464158bSHong Zhang Level: intermediate 2096c464158bSHong Zhang 2097175b88e8SBarry Smith It is recommended that one use the MatCreate(), MatSetType() and/or MatSetFromOptions(), 2098ae1d86c5SBarry Smith MatXXXXSetPreallocation() paradgm instead of this routine directly. 2099175b88e8SBarry Smith [MatXXXXSetPreallocation() is, for example, MatSeqAIJSetPreallocation] 2100175b88e8SBarry Smith 2101c464158bSHong Zhang Notes: 21026d6d819aSHong Zhang The number of rows and columns must be divisible by blocksize. 21036d6d819aSHong Zhang This matrix type does not support complex Hermitian operation. 2104c464158bSHong Zhang 2105c464158bSHong Zhang Specify the preallocated storage with either nz or nnz (not both). 2106c464158bSHong Zhang Set nz=PETSC_DEFAULT and nnz=PETSC_NULL for PETSc to control dynamic memory 21070598bfebSBarry Smith allocation. See the <a href="../../docs/manual.pdf#nameddest=ch_mat">Mat chapter of the users manual</a> for details. 2108c464158bSHong Zhang 210949a6f317SBarry Smith If the nnz parameter is given then the nz parameter is ignored 211049a6f317SBarry Smith 2111*69b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateSBAIJ() 2112c464158bSHong Zhang @*/ 21137087cfbeSBarry Smith PetscErrorCode MatCreateSeqSBAIJ(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt nz,const PetscInt nnz[],Mat *A) 2114c464158bSHong Zhang { 2115dfbe8321SBarry Smith PetscErrorCode ierr; 2116c464158bSHong Zhang 2117c464158bSHong Zhang PetscFunctionBegin; 2118f69a0ea3SMatthew Knepley ierr = MatCreate(comm,A);CHKERRQ(ierr); 2119f69a0ea3SMatthew Knepley ierr = MatSetSizes(*A,m,n,m,n);CHKERRQ(ierr); 2120c464158bSHong Zhang ierr = MatSetType(*A,MATSEQSBAIJ);CHKERRQ(ierr); 2121ab93d7beSBarry Smith ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(*A,bs,nz,(PetscInt*)nnz);CHKERRQ(ierr); 212249b5e25fSSatish Balay PetscFunctionReturn(0); 212349b5e25fSSatish Balay } 212449b5e25fSSatish Balay 21254a2ae208SSatish Balay #undef __FUNCT__ 21264a2ae208SSatish Balay #define __FUNCT__ "MatDuplicate_SeqSBAIJ" 2127dfbe8321SBarry Smith PetscErrorCode MatDuplicate_SeqSBAIJ(Mat A,MatDuplicateOption cpvalues,Mat *B) 212849b5e25fSSatish Balay { 212949b5e25fSSatish Balay Mat C; 213049b5e25fSSatish Balay Mat_SeqSBAIJ *c,*a = (Mat_SeqSBAIJ*)A->data; 21316849ba73SBarry Smith PetscErrorCode ierr; 2132b40805acSSatish Balay PetscInt i,mbs = a->mbs,nz = a->nz,bs2 =a->bs2; 213349b5e25fSSatish Balay 213449b5e25fSSatish Balay PetscFunctionBegin; 2135e32f2f54SBarry Smith if (a->i[mbs] != nz) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Corrupt matrix"); 213649b5e25fSSatish Balay 213749b5e25fSSatish Balay *B = 0; 21387adad957SLisandro Dalcin ierr = MatCreate(((PetscObject)A)->comm,&C);CHKERRQ(ierr); 2139d0f46423SBarry Smith ierr = MatSetSizes(C,A->rmap->N,A->cmap->n,A->rmap->N,A->cmap->n);CHKERRQ(ierr); 21408e9a0fb8SHong Zhang ierr = MatSetType(C,MATSEQSBAIJ);CHKERRQ(ierr); 21411d5dac46SHong Zhang ierr = PetscMemcpy(C->ops,A->ops,sizeof(struct _MatOps));CHKERRQ(ierr); 2142692f9cbeSHong Zhang c = (Mat_SeqSBAIJ*)C->data; 2143692f9cbeSHong Zhang 2144273d9f13SBarry Smith C->preallocated = PETSC_TRUE; 2145d5f3da31SBarry Smith C->factortype = A->factortype; 214649b5e25fSSatish Balay c->row = 0; 214749b5e25fSSatish Balay c->icol = 0; 214849b5e25fSSatish Balay c->saved_values = 0; 2149a9817697SBarry Smith c->keepnonzeropattern = a->keepnonzeropattern; 215049b5e25fSSatish Balay C->assembled = PETSC_TRUE; 215149b5e25fSSatish Balay 21521e1e43feSBarry Smith ierr = PetscLayoutReference(A->rmap,&C->rmap);CHKERRQ(ierr); 21531e1e43feSBarry Smith ierr = PetscLayoutReference(A->cmap,&C->cmap);CHKERRQ(ierr); 215449b5e25fSSatish Balay c->bs2 = a->bs2; 215549b5e25fSSatish Balay c->mbs = a->mbs; 215649b5e25fSSatish Balay c->nbs = a->nbs; 215749b5e25fSSatish Balay 2158c760cd28SBarry Smith if (cpvalues == MAT_SHARE_NONZERO_PATTERN) { 2159c760cd28SBarry Smith c->imax = a->imax; 2160c760cd28SBarry Smith c->ilen = a->ilen; 2161c760cd28SBarry Smith c->free_imax_ilen = PETSC_FALSE; 2162c760cd28SBarry Smith } else { 21638777fc3fSSatish Balay ierr = PetscMalloc2((mbs+1),PetscInt,&c->imax,(mbs+1),PetscInt,&c->ilen);CHKERRQ(ierr); 2164c760cd28SBarry Smith ierr = PetscLogObjectMemory(C,2*(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr); 216549b5e25fSSatish Balay for (i=0; i<mbs; i++) { 216649b5e25fSSatish Balay c->imax[i] = a->imax[i]; 216749b5e25fSSatish Balay c->ilen[i] = a->ilen[i]; 216849b5e25fSSatish Balay } 2169c760cd28SBarry Smith c->free_imax_ilen = PETSC_TRUE; 2170c760cd28SBarry Smith } 217149b5e25fSSatish Balay 217249b5e25fSSatish Balay /* allocate the matrix space */ 21734da8f245SBarry Smith if (cpvalues == MAT_SHARE_NONZERO_PATTERN) { 21744da8f245SBarry Smith ierr = PetscMalloc(bs2*nz*sizeof(MatScalar),&c->a);CHKERRQ(ierr); 21754da8f245SBarry Smith ierr = PetscLogObjectMemory(C,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr); 21764da8f245SBarry Smith c->singlemalloc = PETSC_FALSE; 21774da8f245SBarry Smith c->free_ij = PETSC_FALSE; 21784da8f245SBarry Smith c->parent = A; 21794da8f245SBarry Smith ierr = PetscObjectReference((PetscObject)A);CHKERRQ(ierr); 21804da8f245SBarry Smith ierr = MatSetOption(A,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr); 21814da8f245SBarry Smith ierr = MatSetOption(C,MAT_NEW_NONZERO_LOCATION_ERR,PETSC_TRUE);CHKERRQ(ierr); 21824da8f245SBarry Smith } else { 2183b40805acSSatish Balay ierr = PetscMalloc3(bs2*nz,MatScalar,&c->a,nz,PetscInt,&c->j,mbs+1,PetscInt,&c->i);CHKERRQ(ierr); 218413f74950SBarry Smith ierr = PetscMemcpy(c->i,a->i,(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr); 2185b40805acSSatish Balay ierr = PetscLogObjectMemory(C,(mbs+1)*sizeof(PetscInt) + nz*(bs2*sizeof(MatScalar) + sizeof(PetscInt)));CHKERRQ(ierr); 21864da8f245SBarry Smith c->singlemalloc = PETSC_TRUE; 21874da8f245SBarry Smith c->free_ij = PETSC_TRUE; 21884da8f245SBarry Smith } 218949b5e25fSSatish Balay if (mbs > 0) { 21904da8f245SBarry Smith if (cpvalues != MAT_SHARE_NONZERO_PATTERN) { 219113f74950SBarry Smith ierr = PetscMemcpy(c->j,a->j,nz*sizeof(PetscInt));CHKERRQ(ierr); 21924da8f245SBarry Smith } 219349b5e25fSSatish Balay if (cpvalues == MAT_COPY_VALUES) { 219449b5e25fSSatish Balay ierr = PetscMemcpy(c->a,a->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr); 219549b5e25fSSatish Balay } else { 219649b5e25fSSatish Balay ierr = PetscMemzero(c->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr); 219749b5e25fSSatish Balay } 2198a1c3900fSBarry Smith if (a->jshort) { 21994da8f245SBarry Smith if (cpvalues == MAT_SHARE_NONZERO_PATTERN) { 22004da8f245SBarry Smith c->jshort = a->jshort; 22014da8f245SBarry Smith c->free_jshort = PETSC_FALSE; 22024da8f245SBarry Smith } else { 2203a1c3900fSBarry Smith ierr = PetscMalloc(nz*sizeof(unsigned short),&c->jshort);CHKERRQ(ierr); 2204c760cd28SBarry Smith ierr = PetscLogObjectMemory(C,nz*sizeof(unsigned short));CHKERRQ(ierr); 2205a1c3900fSBarry Smith ierr = PetscMemcpy(c->jshort,a->jshort,nz*sizeof(unsigned short));CHKERRQ(ierr); 22064da8f245SBarry Smith c->free_jshort = PETSC_TRUE; 22074da8f245SBarry Smith } 2208a1c3900fSBarry Smith } 220949b5e25fSSatish Balay } 221049b5e25fSSatish Balay 221149b5e25fSSatish Balay c->roworiented = a->roworiented; 221249b5e25fSSatish Balay c->nonew = a->nonew; 221349b5e25fSSatish Balay 221449b5e25fSSatish Balay if (a->diag) { 2215c760cd28SBarry Smith if (cpvalues == MAT_SHARE_NONZERO_PATTERN) { 2216c760cd28SBarry Smith c->diag = a->diag; 2217c760cd28SBarry Smith c->free_diag = PETSC_FALSE; 2218c760cd28SBarry Smith } else { 22192ed38d0bSJed Brown ierr = PetscMalloc(mbs*sizeof(PetscInt),&c->diag);CHKERRQ(ierr); 22202ed38d0bSJed Brown ierr = PetscLogObjectMemory(C,mbs*sizeof(PetscInt));CHKERRQ(ierr); 222149b5e25fSSatish Balay for (i=0; i<mbs; i++) { 222249b5e25fSSatish Balay c->diag[i] = a->diag[i]; 222349b5e25fSSatish Balay } 2224c760cd28SBarry Smith c->free_diag = PETSC_TRUE; 2225c760cd28SBarry Smith } 222649b5e25fSSatish Balay } else c->diag = 0; 22276c6c5352SBarry Smith c->nz = a->nz; 2228f2cbd3d5SJed Brown c->maxnz = a->nz; /* Since we allocate exactly the right amount */ 222949b5e25fSSatish Balay c->solve_work = 0; 223049b5e25fSSatish Balay c->mult_work = 0; 2231e6b907acSBarry Smith c->free_a = PETSC_TRUE; 223249b5e25fSSatish Balay *B = C; 22337adad957SLisandro Dalcin ierr = PetscFListDuplicate(((PetscObject)A)->qlist,&((PetscObject)C)->qlist);CHKERRQ(ierr); 223449b5e25fSSatish Balay PetscFunctionReturn(0); 223549b5e25fSSatish Balay } 223649b5e25fSSatish Balay 22374a2ae208SSatish Balay #undef __FUNCT__ 22385bba2384SShri Abhyankar #define __FUNCT__ "MatLoad_SeqSBAIJ" 2239112444f4SShri Abhyankar PetscErrorCode MatLoad_SeqSBAIJ(Mat newmat,PetscViewer viewer) 22402f480046SShri Abhyankar { 22412f480046SShri Abhyankar Mat_SeqSBAIJ *a; 22422f480046SShri Abhyankar PetscErrorCode ierr; 22432f480046SShri Abhyankar int fd; 22442f480046SShri Abhyankar PetscMPIInt size; 22452f480046SShri Abhyankar PetscInt i,nz,header[4],*rowlengths=0,M,N,bs=1; 22462f480046SShri Abhyankar PetscInt *mask,mbs,*jj,j,rowcount,nzcount,k,*s_browlengths,maskcount; 22472f480046SShri Abhyankar PetscInt kmax,jcount,block,idx,point,nzcountb,extra_rows,rows,cols; 22482f480046SShri Abhyankar PetscInt *masked,nmask,tmp,bs2,ishift; 22492f480046SShri Abhyankar PetscScalar *aa; 22502f480046SShri Abhyankar MPI_Comm comm = ((PetscObject)viewer)->comm; 22512f480046SShri Abhyankar 22522f480046SShri Abhyankar PetscFunctionBegin; 2253c55dd799SBarry Smith ierr = PetscOptionsGetInt(((PetscObject)newmat)->prefix,"-matload_block_size",&bs,PETSC_NULL);CHKERRQ(ierr); 22542f480046SShri Abhyankar bs2 = bs*bs; 22552f480046SShri Abhyankar 22562f480046SShri Abhyankar ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 22572f480046SShri Abhyankar if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"view must have one processor"); 22582f480046SShri Abhyankar ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr); 22592f480046SShri Abhyankar ierr = PetscBinaryRead(fd,header,4,PETSC_INT);CHKERRQ(ierr); 22602f480046SShri Abhyankar if (header[0] != MAT_FILE_CLASSID) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"not Mat object"); 22612f480046SShri Abhyankar M = header[1]; N = header[2]; nz = header[3]; 22622f480046SShri Abhyankar 22632f480046SShri Abhyankar if (header[3] < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Matrix stored in special format, cannot load as SeqSBAIJ"); 22642f480046SShri Abhyankar 22652f480046SShri Abhyankar if (M != N) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Can only do square matrices"); 22662f480046SShri Abhyankar 22672f480046SShri Abhyankar /* 22682f480046SShri Abhyankar This code adds extra rows to make sure the number of rows is 22692f480046SShri Abhyankar divisible by the blocksize 22702f480046SShri Abhyankar */ 22712f480046SShri Abhyankar mbs = M/bs; 22722f480046SShri Abhyankar extra_rows = bs - M + bs*(mbs); 22732f480046SShri Abhyankar if (extra_rows == bs) extra_rows = 0; 22742f480046SShri Abhyankar else mbs++; 22752f480046SShri Abhyankar if (extra_rows) { 22762f480046SShri Abhyankar ierr = PetscInfo(viewer,"Padding loaded matrix to match blocksize\n");CHKERRQ(ierr); 22772f480046SShri Abhyankar } 22782f480046SShri Abhyankar 22792f480046SShri Abhyankar /* Set global sizes if not already set */ 22802f480046SShri Abhyankar if (newmat->rmap->n < 0 && newmat->rmap->N < 0 && newmat->cmap->n < 0 && newmat->cmap->N < 0) { 22812f480046SShri Abhyankar ierr = MatSetSizes(newmat,PETSC_DECIDE,PETSC_DECIDE,M+extra_rows,N+extra_rows);CHKERRQ(ierr); 22822f480046SShri Abhyankar } else { /* Check if the matrix global sizes are correct */ 22832f480046SShri Abhyankar ierr = MatGetSize(newmat,&rows,&cols);CHKERRQ(ierr); 22842f480046SShri 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); 22852f480046SShri Abhyankar } 22862f480046SShri Abhyankar 22872f480046SShri Abhyankar /* read in row lengths */ 22882f480046SShri Abhyankar ierr = PetscMalloc((M+extra_rows)*sizeof(PetscInt),&rowlengths);CHKERRQ(ierr); 22892f480046SShri Abhyankar ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr); 22902f480046SShri Abhyankar for (i=0; i<extra_rows; i++) rowlengths[M+i] = 1; 22912f480046SShri Abhyankar 22922f480046SShri Abhyankar /* read in column indices */ 22932f480046SShri Abhyankar ierr = PetscMalloc((nz+extra_rows)*sizeof(PetscInt),&jj);CHKERRQ(ierr); 22942f480046SShri Abhyankar ierr = PetscBinaryRead(fd,jj,nz,PETSC_INT);CHKERRQ(ierr); 22952f480046SShri Abhyankar for (i=0; i<extra_rows; i++) jj[nz+i] = M+i; 22962f480046SShri Abhyankar 22972f480046SShri Abhyankar /* loop over row lengths determining block row lengths */ 22982f480046SShri Abhyankar ierr = PetscMalloc(mbs*sizeof(PetscInt),&s_browlengths);CHKERRQ(ierr); 22992f480046SShri Abhyankar ierr = PetscMemzero(s_browlengths,mbs*sizeof(PetscInt));CHKERRQ(ierr); 23002f480046SShri Abhyankar ierr = PetscMalloc2(mbs,PetscInt,&mask,mbs,PetscInt,&masked);CHKERRQ(ierr); 23012f480046SShri Abhyankar ierr = PetscMemzero(mask,mbs*sizeof(PetscInt));CHKERRQ(ierr); 23022f480046SShri Abhyankar rowcount = 0; 23032f480046SShri Abhyankar nzcount = 0; 23042f480046SShri Abhyankar for (i=0; i<mbs; i++) { 23052f480046SShri Abhyankar nmask = 0; 23062f480046SShri Abhyankar for (j=0; j<bs; j++) { 23072f480046SShri Abhyankar kmax = rowlengths[rowcount]; 23082f480046SShri Abhyankar for (k=0; k<kmax; k++) { 23092f480046SShri Abhyankar tmp = jj[nzcount++]/bs; /* block col. index */ 23102f480046SShri Abhyankar if (!mask[tmp] && tmp >= i) {masked[nmask++] = tmp; mask[tmp] = 1;} 23112f480046SShri Abhyankar } 23122f480046SShri Abhyankar rowcount++; 23132f480046SShri Abhyankar } 23142f480046SShri Abhyankar s_browlengths[i] += nmask; 23152f480046SShri Abhyankar 23162f480046SShri Abhyankar /* zero out the mask elements we set */ 23172f480046SShri Abhyankar for (j=0; j<nmask; j++) mask[masked[j]] = 0; 23182f480046SShri Abhyankar } 23192f480046SShri Abhyankar 23202f480046SShri Abhyankar /* Do preallocation */ 23212f480046SShri Abhyankar ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(newmat,bs,0,s_browlengths);CHKERRQ(ierr); 23222f480046SShri Abhyankar a = (Mat_SeqSBAIJ*)newmat->data; 23232f480046SShri Abhyankar 23242f480046SShri Abhyankar /* set matrix "i" values */ 23252f480046SShri Abhyankar a->i[0] = 0; 23262f480046SShri Abhyankar for (i=1; i<= mbs; i++) { 23272f480046SShri Abhyankar a->i[i] = a->i[i-1] + s_browlengths[i-1]; 23282f480046SShri Abhyankar a->ilen[i-1] = s_browlengths[i-1]; 23292f480046SShri Abhyankar } 23302f480046SShri Abhyankar a->nz = a->i[mbs]; 23312f480046SShri Abhyankar 23322f480046SShri Abhyankar /* read in nonzero values */ 23332f480046SShri Abhyankar ierr = PetscMalloc((nz+extra_rows)*sizeof(PetscScalar),&aa);CHKERRQ(ierr); 23342f480046SShri Abhyankar ierr = PetscBinaryRead(fd,aa,nz,PETSC_SCALAR);CHKERRQ(ierr); 23352f480046SShri Abhyankar for (i=0; i<extra_rows; i++) aa[nz+i] = 1.0; 23362f480046SShri Abhyankar 23372f480046SShri Abhyankar /* set "a" and "j" values into matrix */ 23382f480046SShri Abhyankar nzcount = 0; jcount = 0; 23392f480046SShri Abhyankar for (i=0; i<mbs; i++) { 23402f480046SShri Abhyankar nzcountb = nzcount; 23412f480046SShri Abhyankar nmask = 0; 23422f480046SShri Abhyankar for (j=0; j<bs; j++) { 23432f480046SShri Abhyankar kmax = rowlengths[i*bs+j]; 23442f480046SShri Abhyankar for (k=0; k<kmax; k++) { 23452f480046SShri Abhyankar tmp = jj[nzcount++]/bs; /* block col. index */ 23462f480046SShri Abhyankar if (!mask[tmp] && tmp >= i) { masked[nmask++] = tmp; mask[tmp] = 1;} 23472f480046SShri Abhyankar } 23482f480046SShri Abhyankar } 23492f480046SShri Abhyankar /* sort the masked values */ 23502f480046SShri Abhyankar ierr = PetscSortInt(nmask,masked);CHKERRQ(ierr); 23512f480046SShri Abhyankar 23522f480046SShri Abhyankar /* set "j" values into matrix */ 23532f480046SShri Abhyankar maskcount = 1; 23542f480046SShri Abhyankar for (j=0; j<nmask; j++) { 23552f480046SShri Abhyankar a->j[jcount++] = masked[j]; 23562f480046SShri Abhyankar mask[masked[j]] = maskcount++; 23572f480046SShri Abhyankar } 23582f480046SShri Abhyankar 23592f480046SShri Abhyankar /* set "a" values into matrix */ 23602f480046SShri Abhyankar ishift = bs2*a->i[i]; 23612f480046SShri Abhyankar for (j=0; j<bs; j++) { 23622f480046SShri Abhyankar kmax = rowlengths[i*bs+j]; 23632f480046SShri Abhyankar for (k=0; k<kmax; k++) { 23642f480046SShri Abhyankar tmp = jj[nzcountb]/bs ; /* block col. index */ 23652f480046SShri Abhyankar if (tmp >= i){ 23662f480046SShri Abhyankar block = mask[tmp] - 1; 23672f480046SShri Abhyankar point = jj[nzcountb] - bs*tmp; 23682f480046SShri Abhyankar idx = ishift + bs2*block + j + bs*point; 23692f480046SShri Abhyankar a->a[idx] = aa[nzcountb]; 23702f480046SShri Abhyankar } 23712f480046SShri Abhyankar nzcountb++; 23722f480046SShri Abhyankar } 23732f480046SShri Abhyankar } 23742f480046SShri Abhyankar /* zero out the mask elements we set */ 23752f480046SShri Abhyankar for (j=0; j<nmask; j++) mask[masked[j]] = 0; 23762f480046SShri Abhyankar } 23772f480046SShri Abhyankar if (jcount != a->nz) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Bad binary matrix"); 23782f480046SShri Abhyankar 23792f480046SShri Abhyankar ierr = PetscFree(rowlengths);CHKERRQ(ierr); 23802f480046SShri Abhyankar ierr = PetscFree(s_browlengths);CHKERRQ(ierr); 23812f480046SShri Abhyankar ierr = PetscFree(aa);CHKERRQ(ierr); 23822f480046SShri Abhyankar ierr = PetscFree(jj);CHKERRQ(ierr); 23832f480046SShri Abhyankar ierr = PetscFree2(mask,masked);CHKERRQ(ierr); 23842f480046SShri Abhyankar 23852f480046SShri Abhyankar ierr = MatAssemblyBegin(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 23862f480046SShri Abhyankar ierr = MatAssemblyEnd(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 23872f480046SShri Abhyankar ierr = MatView_Private(newmat);CHKERRQ(ierr); 23882f480046SShri Abhyankar PetscFunctionReturn(0); 23892f480046SShri Abhyankar } 23902f480046SShri Abhyankar 23912f480046SShri Abhyankar #undef __FUNCT__ 2392c75a6043SHong Zhang #define __FUNCT__ "MatCreateSeqSBAIJWithArrays" 2393c75a6043SHong Zhang /*@ 2394c75a6043SHong Zhang MatCreateSeqSBAIJWithArrays - Creates an sequential SBAIJ matrix using matrix elements 2395c75a6043SHong Zhang (upper triangular entries in CSR format) provided by the user. 2396c75a6043SHong Zhang 2397c75a6043SHong Zhang Collective on MPI_Comm 2398c75a6043SHong Zhang 2399c75a6043SHong Zhang Input Parameters: 2400c75a6043SHong Zhang + comm - must be an MPI communicator of size 1 2401c75a6043SHong Zhang . bs - size of block 2402c75a6043SHong Zhang . m - number of rows 2403c75a6043SHong Zhang . n - number of columns 2404c75a6043SHong Zhang . i - row indices 2405c75a6043SHong Zhang . j - column indices 2406c75a6043SHong Zhang - a - matrix values 2407c75a6043SHong Zhang 2408c75a6043SHong Zhang Output Parameter: 2409c75a6043SHong Zhang . mat - the matrix 2410c75a6043SHong Zhang 2411dfb205c3SBarry Smith Level: advanced 2412c75a6043SHong Zhang 2413c75a6043SHong Zhang Notes: 2414c75a6043SHong Zhang The i, j, and a arrays are not copied by this routine, the user must free these arrays 2415c75a6043SHong Zhang once the matrix is destroyed 2416c75a6043SHong Zhang 2417c75a6043SHong Zhang You cannot set new nonzero locations into this matrix, that will generate an error. 2418c75a6043SHong Zhang 2419c75a6043SHong Zhang The i and j indices are 0 based 2420c75a6043SHong Zhang 2421dfb205c3SBarry 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 2422dfb205c3SBarry Smith it is the regular CSR format excluding the lower triangular elements. 2423dfb205c3SBarry Smith 2424*69b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateSBAIJ(), MatCreateSeqSBAIJ() 2425c75a6043SHong Zhang 2426c75a6043SHong Zhang @*/ 24277087cfbeSBarry Smith PetscErrorCode MatCreateSeqSBAIJWithArrays(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt* i,PetscInt*j,PetscScalar *a,Mat *mat) 2428c75a6043SHong Zhang { 2429c75a6043SHong Zhang PetscErrorCode ierr; 2430c75a6043SHong Zhang PetscInt ii; 2431c75a6043SHong Zhang Mat_SeqSBAIJ *sbaij; 2432c75a6043SHong Zhang 2433c75a6043SHong Zhang PetscFunctionBegin; 2434e32f2f54SBarry Smith if (bs != 1) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"block size %D > 1 is not supported yet",bs); 2435e32f2f54SBarry Smith if (i[0]) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"i (row indices) must start with 0"); 2436c75a6043SHong Zhang 2437c75a6043SHong Zhang ierr = MatCreate(comm,mat);CHKERRQ(ierr); 2438c75a6043SHong Zhang ierr = MatSetSizes(*mat,m,n,m,n);CHKERRQ(ierr); 2439c75a6043SHong Zhang ierr = MatSetType(*mat,MATSEQSBAIJ);CHKERRQ(ierr); 2440c75a6043SHong Zhang ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(*mat,bs,MAT_SKIP_ALLOCATION,0);CHKERRQ(ierr); 2441c75a6043SHong Zhang sbaij = (Mat_SeqSBAIJ*)(*mat)->data; 2442c75a6043SHong Zhang ierr = PetscMalloc2(m,PetscInt,&sbaij->imax,m,PetscInt,&sbaij->ilen);CHKERRQ(ierr); 2443c760cd28SBarry Smith ierr = PetscLogObjectMemory(*mat,2*m*sizeof(PetscInt));CHKERRQ(ierr); 2444c75a6043SHong Zhang 2445c75a6043SHong Zhang sbaij->i = i; 2446c75a6043SHong Zhang sbaij->j = j; 2447c75a6043SHong Zhang sbaij->a = a; 2448c75a6043SHong Zhang sbaij->singlemalloc = PETSC_FALSE; 2449c75a6043SHong Zhang sbaij->nonew = -1; /*this indicates that inserting a new value in the matrix that generates a new nonzero is an error*/ 2450e6b907acSBarry Smith sbaij->free_a = PETSC_FALSE; 2451e6b907acSBarry Smith sbaij->free_ij = PETSC_FALSE; 2452c75a6043SHong Zhang 2453c75a6043SHong Zhang for (ii=0; ii<m; ii++) { 2454c75a6043SHong Zhang sbaij->ilen[ii] = sbaij->imax[ii] = i[ii+1] - i[ii]; 2455c75a6043SHong Zhang #if defined(PETSC_USE_DEBUG) 2456e32f2f54SBarry 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]); 2457c75a6043SHong Zhang #endif 2458c75a6043SHong Zhang } 2459c75a6043SHong Zhang #if defined(PETSC_USE_DEBUG) 2460c75a6043SHong Zhang for (ii=0; ii<sbaij->i[m]; ii++) { 2461e32f2f54SBarry Smith if (j[ii] < 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative column index at location = %d index = %d",ii,j[ii]); 2462e32f2f54SBarry 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]); 2463c75a6043SHong Zhang } 2464c75a6043SHong Zhang #endif 2465c75a6043SHong Zhang 2466c75a6043SHong Zhang ierr = MatAssemblyBegin(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2467c75a6043SHong Zhang ierr = MatAssemblyEnd(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2468c75a6043SHong Zhang PetscFunctionReturn(0); 2469c75a6043SHong Zhang } 2470d06b337dSHong Zhang 2471d06b337dSHong Zhang 2472d06b337dSHong Zhang 247349b5e25fSSatish Balay 247449b5e25fSSatish Balay 2475