1be1d678aSKris Buschelman #define PETSCMAT_DLL 249b5e25fSSatish Balay 349b5e25fSSatish Balay /* 4a1373b80SHong Zhang Defines the basic matrix operations for the SBAIJ (compressed row) 549b5e25fSSatish Balay matrix storage format. 649b5e25fSSatish Balay */ 79e070d67SMatthew Knepley #include "src/mat/impls/baij/seq/baij.h" /*I "petscmat.h" I*/ 849b5e25fSSatish Balay #include "src/inline/spops.h" 9aec82049SSatish Balay #include "src/mat/impls/sbaij/seq/sbaij.h" 1049b5e25fSSatish Balay 1149b5e25fSSatish Balay #define CHUNKSIZE 10 1249b5e25fSSatish Balay 1349b5e25fSSatish Balay /* 1449b5e25fSSatish Balay Checks for missing diagonals 1549b5e25fSSatish Balay */ 164a2ae208SSatish Balay #undef __FUNCT__ 174a2ae208SSatish Balay #define __FUNCT__ "MatMissingDiagonal_SeqSBAIJ" 18dfbe8321SBarry Smith PetscErrorCode MatMissingDiagonal_SeqSBAIJ(Mat A) 1949b5e25fSSatish Balay { 20045c9aa0SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 216849ba73SBarry Smith PetscErrorCode ierr; 2213f74950SBarry Smith PetscInt *diag,*jj = a->j,i; 2349b5e25fSSatish Balay 2449b5e25fSSatish Balay PetscFunctionBegin; 25045c9aa0SHong Zhang ierr = MatMarkDiagonal_SeqSBAIJ(A);CHKERRQ(ierr); 2649b5e25fSSatish Balay diag = a->diag; 2749b5e25fSSatish Balay for (i=0; i<a->mbs; i++) { 2877431f27SBarry Smith if (jj[diag[i]] != i) SETERRQ1(PETSC_ERR_ARG_CORRUPT,"Matrix is missing diagonal number %D",i); 2949b5e25fSSatish Balay } 3049b5e25fSSatish Balay PetscFunctionReturn(0); 3149b5e25fSSatish Balay } 3249b5e25fSSatish Balay 334a2ae208SSatish Balay #undef __FUNCT__ 344a2ae208SSatish Balay #define __FUNCT__ "MatMarkDiagonal_SeqSBAIJ" 35dfbe8321SBarry Smith PetscErrorCode MatMarkDiagonal_SeqSBAIJ(Mat A) 3649b5e25fSSatish Balay { 37045c9aa0SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 386849ba73SBarry Smith PetscErrorCode ierr; 3909f38230SBarry Smith PetscInt i; 4049b5e25fSSatish Balay 4149b5e25fSSatish Balay PetscFunctionBegin; 4209f38230SBarry Smith if (!a->diag) { 4309f38230SBarry Smith ierr = PetscMalloc(a->mbs*sizeof(PetscInt),&a->diag);CHKERRQ(ierr); 4409f38230SBarry Smith } 4509f38230SBarry Smith for (i=0; i<a->mbs; i++) a->diag[i] = a->i[i]; 4649b5e25fSSatish Balay PetscFunctionReturn(0); 4749b5e25fSSatish Balay } 4849b5e25fSSatish Balay 494a2ae208SSatish Balay #undef __FUNCT__ 504a2ae208SSatish Balay #define __FUNCT__ "MatGetRowIJ_SeqSBAIJ" 5113f74950SBarry Smith static PetscErrorCode MatGetRowIJ_SeqSBAIJ(Mat A,PetscInt oshift,PetscTruth symmetric,PetscInt *nn,PetscInt *ia[],PetscInt *ja[],PetscTruth *done) 5249b5e25fSSatish Balay { 53a6ece127SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 5413f74950SBarry Smith PetscInt n = a->mbs,i; 5549b5e25fSSatish Balay 5649b5e25fSSatish Balay PetscFunctionBegin; 57d3e5a4abSHong Zhang *nn = n; 58a1373b80SHong Zhang if (!ia) PetscFunctionReturn(0); 59a1373b80SHong Zhang 60a6ece127SHong Zhang if (oshift == 1) { 61a6ece127SHong Zhang /* temporarily add 1 to i and j indices */ 6213f74950SBarry Smith PetscInt nz = a->i[n]; 636c6c5352SBarry Smith for (i=0; i<nz; i++) a->j[i]++; 64a1373b80SHong Zhang for (i=0; i<n+1; i++) a->i[i]++; 65a1373b80SHong Zhang *ia = a->i; *ja = a->j; 66a1373b80SHong Zhang } else { 67a1373b80SHong Zhang *ia = a->i; *ja = a->j; 68a6ece127SHong Zhang } 6949b5e25fSSatish Balay PetscFunctionReturn(0); 7049b5e25fSSatish Balay } 7149b5e25fSSatish Balay 724a2ae208SSatish Balay #undef __FUNCT__ 734a2ae208SSatish Balay #define __FUNCT__ "MatRestoreRowIJ_SeqSBAIJ" 7413f74950SBarry Smith static PetscErrorCode MatRestoreRowIJ_SeqSBAIJ(Mat A,PetscInt oshift,PetscTruth symmetric,PetscInt *nn,PetscInt *ia[],PetscInt *ja[],PetscTruth *done) 7549b5e25fSSatish Balay { 76b7aaefc3SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 7713f74950SBarry Smith PetscInt i,n = a->mbs; 78a6ece127SHong Zhang 7949b5e25fSSatish Balay PetscFunctionBegin; 8049b5e25fSSatish Balay if (!ia) PetscFunctionReturn(0); 81a6ece127SHong Zhang 82a6ece127SHong Zhang if (oshift == 1) { 8313f74950SBarry Smith PetscInt nz = a->i[n]-1; 846c6c5352SBarry Smith for (i=0; i<nz; i++) a->j[i]--; 85a6ece127SHong Zhang for (i=0; i<n+1; i++) a->i[i]--; 86a6ece127SHong Zhang } 87a6ece127SHong Zhang PetscFunctionReturn(0); 8849b5e25fSSatish Balay } 8949b5e25fSSatish Balay 904a2ae208SSatish Balay #undef __FUNCT__ 914a2ae208SSatish Balay #define __FUNCT__ "MatDestroy_SeqSBAIJ" 92dfbe8321SBarry Smith PetscErrorCode MatDestroy_SeqSBAIJ(Mat A) 9349b5e25fSSatish Balay { 9449b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 95dfbe8321SBarry Smith PetscErrorCode ierr; 9649b5e25fSSatish Balay 9749b5e25fSSatish Balay PetscFunctionBegin; 98a9f03627SSatish Balay #if defined(PETSC_USE_LOG) 99899cda47SBarry Smith PetscLogObjectState((PetscObject)A,"Rows=%D, NZ=%D",A->rmap.N,a->nz); 100a9f03627SSatish Balay #endif 101e6b907acSBarry Smith ierr = MatSeqXAIJFreeAIJ(A,&a->a,&a->j,&a->i);CHKERRQ(ierr); 1029bfd6278SHong Zhang if (a->row) {ierr = ISDestroy(a->row);CHKERRQ(ierr);} 1039bfd6278SHong Zhang if (a->col){ierr = ISDestroy(a->col);CHKERRQ(ierr);} 1049bfd6278SHong Zhang if (a->icol) {ierr = ISDestroy(a->icol);CHKERRQ(ierr);} 10505b42c5fSBarry Smith ierr = PetscFree(a->diag);CHKERRQ(ierr); 10605b42c5fSBarry Smith ierr = PetscFree2(a->imax,a->ilen);CHKERRQ(ierr); 10705b42c5fSBarry Smith ierr = PetscFree(a->solve_work);CHKERRQ(ierr); 108e2ee2a47SBarry Smith ierr = PetscFree(a->relax_work);CHKERRQ(ierr); 10905b42c5fSBarry Smith ierr = PetscFree(a->solves_work);CHKERRQ(ierr); 11005b42c5fSBarry Smith ierr = PetscFree(a->mult_work);CHKERRQ(ierr); 11105b42c5fSBarry Smith ierr = PetscFree(a->saved_values);CHKERRQ(ierr); 11205b42c5fSBarry Smith ierr = PetscFree(a->xtoy);CHKERRQ(ierr); 1131a3463dfSHong Zhang 1141a3463dfSHong Zhang ierr = PetscFree(a->inew);CHKERRQ(ierr); 11549b5e25fSSatish Balay ierr = PetscFree(a);CHKERRQ(ierr); 116901853e0SKris Buschelman 117*dbd8c25aSHong Zhang ierr = PetscObjectChangeTypeName((PetscObject)A,0);CHKERRQ(ierr); 118901853e0SKris Buschelman ierr = PetscObjectComposeFunction((PetscObject)A,"MatStoreValues_C","",PETSC_NULL);CHKERRQ(ierr); 119901853e0SKris Buschelman ierr = PetscObjectComposeFunction((PetscObject)A,"MatRetrieveValues_C","",PETSC_NULL);CHKERRQ(ierr); 120901853e0SKris Buschelman ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetColumnIndices_C","",PETSC_NULL);CHKERRQ(ierr); 121901853e0SKris Buschelman ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqaij_C","",PETSC_NULL);CHKERRQ(ierr); 122901853e0SKris Buschelman ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqbaij_C","",PETSC_NULL);CHKERRQ(ierr); 123901853e0SKris Buschelman ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetPreallocation_C","",PETSC_NULL);CHKERRQ(ierr); 12449b5e25fSSatish Balay PetscFunctionReturn(0); 12549b5e25fSSatish Balay } 12649b5e25fSSatish Balay 1274a2ae208SSatish Balay #undef __FUNCT__ 1284a2ae208SSatish Balay #define __FUNCT__ "MatSetOption_SeqSBAIJ" 129dfbe8321SBarry Smith PetscErrorCode MatSetOption_SeqSBAIJ(Mat A,MatOption op) 13049b5e25fSSatish Balay { 131045c9aa0SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 13263ba0a88SBarry Smith PetscErrorCode ierr; 13349b5e25fSSatish Balay 13449b5e25fSSatish Balay PetscFunctionBegin; 1354d9d31abSKris Buschelman switch (op) { 1364d9d31abSKris Buschelman case MAT_ROW_ORIENTED: 1374d9d31abSKris Buschelman a->roworiented = PETSC_TRUE; 1384d9d31abSKris Buschelman break; 1394d9d31abSKris Buschelman case MAT_COLUMN_ORIENTED: 1404d9d31abSKris Buschelman a->roworiented = PETSC_FALSE; 1414d9d31abSKris Buschelman break; 1424d9d31abSKris Buschelman case MAT_COLUMNS_SORTED: 1434d9d31abSKris Buschelman a->sorted = PETSC_TRUE; 1444d9d31abSKris Buschelman break; 1454d9d31abSKris Buschelman case MAT_COLUMNS_UNSORTED: 1464d9d31abSKris Buschelman a->sorted = PETSC_FALSE; 1474d9d31abSKris Buschelman break; 1484d9d31abSKris Buschelman case MAT_KEEP_ZEROED_ROWS: 1494d9d31abSKris Buschelman a->keepzeroedrows = PETSC_TRUE; 1504d9d31abSKris Buschelman break; 1514d9d31abSKris Buschelman case MAT_NO_NEW_NONZERO_LOCATIONS: 1524d9d31abSKris Buschelman a->nonew = 1; 1534d9d31abSKris Buschelman break; 1544d9d31abSKris Buschelman case MAT_NEW_NONZERO_LOCATION_ERR: 1554d9d31abSKris Buschelman a->nonew = -1; 1564d9d31abSKris Buschelman break; 1574d9d31abSKris Buschelman case MAT_NEW_NONZERO_ALLOCATION_ERR: 1584d9d31abSKris Buschelman a->nonew = -2; 1594d9d31abSKris Buschelman break; 1604d9d31abSKris Buschelman case MAT_YES_NEW_NONZERO_LOCATIONS: 1614d9d31abSKris Buschelman a->nonew = 0; 1624d9d31abSKris Buschelman break; 1634d9d31abSKris Buschelman case MAT_ROWS_SORTED: 1644d9d31abSKris Buschelman case MAT_ROWS_UNSORTED: 1654d9d31abSKris Buschelman case MAT_YES_NEW_DIAGONALS: 1664d9d31abSKris Buschelman case MAT_IGNORE_OFF_PROC_ENTRIES: 1674d9d31abSKris Buschelman case MAT_USE_HASH_TABLE: 168290bbb0aSBarry Smith ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr); 1694d9d31abSKris Buschelman break; 1704d9d31abSKris Buschelman case MAT_NO_NEW_DIAGONALS: 17129bbc08cSBarry Smith SETERRQ(PETSC_ERR_SUP,"MAT_NO_NEW_DIAGONALS"); 1729a4540c5SBarry Smith case MAT_NOT_SYMMETRIC: 1739a4540c5SBarry Smith case MAT_NOT_STRUCTURALLY_SYMMETRIC: 1749a4540c5SBarry Smith case MAT_HERMITIAN: 1759a4540c5SBarry Smith SETERRQ(PETSC_ERR_SUP,"Matrix must be symmetric"); 17677e54ba9SKris Buschelman case MAT_SYMMETRIC: 17777e54ba9SKris Buschelman case MAT_STRUCTURALLY_SYMMETRIC: 1789a4540c5SBarry Smith case MAT_NOT_HERMITIAN: 1799a4540c5SBarry Smith case MAT_SYMMETRY_ETERNAL: 1809a4540c5SBarry Smith case MAT_NOT_SYMMETRY_ETERNAL: 181290bbb0aSBarry Smith ierr = PetscInfo1(A,"Option %s not relevent\n",MatOptions[op]);CHKERRQ(ierr); 182290bbb0aSBarry Smith break; 183941593c8SHong Zhang case MAT_IGNORE_LOWER_TRIANGULAR: 184941593c8SHong Zhang a->ignore_ltriangular = PETSC_TRUE; 185941593c8SHong Zhang break; 186941593c8SHong Zhang case MAT_ERROR_LOWER_TRIANGULAR: 187941593c8SHong Zhang a->ignore_ltriangular = PETSC_FALSE; 18877e54ba9SKris Buschelman break; 189f5edf698SHong Zhang case MAT_GETROW_UPPERTRIANGULAR: 190f5edf698SHong Zhang a->getrow_utriangular = PETSC_TRUE; 191f5edf698SHong Zhang break; 1924d9d31abSKris Buschelman default: 193ad86a440SBarry Smith SETERRQ1(PETSC_ERR_SUP,"unknown option %d",op); 19449b5e25fSSatish Balay } 19549b5e25fSSatish Balay PetscFunctionReturn(0); 19649b5e25fSSatish Balay } 19749b5e25fSSatish Balay 1984a2ae208SSatish Balay #undef __FUNCT__ 1994a2ae208SSatish Balay #define __FUNCT__ "MatGetRow_SeqSBAIJ" 20013f74950SBarry Smith PetscErrorCode MatGetRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *ncols,PetscInt **cols,PetscScalar **v) 20149b5e25fSSatish Balay { 20249b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 2036849ba73SBarry Smith PetscErrorCode ierr; 20413f74950SBarry Smith PetscInt itmp,i,j,k,M,*ai,*aj,bs,bn,bp,*cols_i,bs2; 20549b5e25fSSatish Balay MatScalar *aa,*aa_i; 20687828ca2SBarry Smith PetscScalar *v_i; 20749b5e25fSSatish Balay 20849b5e25fSSatish Balay PetscFunctionBegin; 209f5edf698SHong Zhang if (A && !a->getrow_utriangular) SETERRQ(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) or MatGetRowUpperTriangular()"); 210f5edf698SHong Zhang /* Get the upper triangular part of the row */ 211899cda47SBarry Smith bs = A->rmap.bs; 21249b5e25fSSatish Balay ai = a->i; 21349b5e25fSSatish Balay aj = a->j; 21449b5e25fSSatish Balay aa = a->a; 21549b5e25fSSatish Balay bs2 = a->bs2; 21649b5e25fSSatish Balay 217899cda47SBarry Smith if (row < 0 || row >= A->rmap.N) SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE, "Row %D out of range", row); 21849b5e25fSSatish Balay 21949b5e25fSSatish Balay bn = row/bs; /* Block number */ 22049b5e25fSSatish Balay bp = row % bs; /* Block position */ 22149b5e25fSSatish Balay M = ai[bn+1] - ai[bn]; 22249b5e25fSSatish Balay *ncols = bs*M; 22349b5e25fSSatish Balay 22449b5e25fSSatish Balay if (v) { 22549b5e25fSSatish Balay *v = 0; 22649b5e25fSSatish Balay if (*ncols) { 22787828ca2SBarry Smith ierr = PetscMalloc((*ncols+row)*sizeof(PetscScalar),v);CHKERRQ(ierr); 22849b5e25fSSatish Balay for (i=0; i<M; i++) { /* for each block in the block row */ 22949b5e25fSSatish Balay v_i = *v + i*bs; 23049b5e25fSSatish Balay aa_i = aa + bs2*(ai[bn] + i); 23149b5e25fSSatish Balay for (j=bp,k=0; j<bs2; j+=bs,k++) {v_i[k] = aa_i[j];} 23249b5e25fSSatish Balay } 23349b5e25fSSatish Balay } 23449b5e25fSSatish Balay } 23549b5e25fSSatish Balay 23649b5e25fSSatish Balay if (cols) { 23749b5e25fSSatish Balay *cols = 0; 23849b5e25fSSatish Balay if (*ncols) { 23913f74950SBarry Smith ierr = PetscMalloc((*ncols+row)*sizeof(PetscInt),cols);CHKERRQ(ierr); 24049b5e25fSSatish Balay for (i=0; i<M; i++) { /* for each block in the block row */ 24149b5e25fSSatish Balay cols_i = *cols + i*bs; 24249b5e25fSSatish Balay itmp = bs*aj[ai[bn] + i]; 24349b5e25fSSatish Balay for (j=0; j<bs; j++) {cols_i[j] = itmp++;} 24449b5e25fSSatish Balay } 24549b5e25fSSatish Balay } 24649b5e25fSSatish Balay } 24749b5e25fSSatish Balay 24849b5e25fSSatish Balay /*search column A(0:row-1,row) (=A(row,0:row-1)). Could be expensive! */ 2495ddb2528SHong Zhang /* this segment is currently removed, so only entries in the upper triangle are obtained */ 2505ddb2528SHong Zhang #ifdef column_search 25149b5e25fSSatish Balay v_i = *v + M*bs; 25249b5e25fSSatish Balay cols_i = *cols + M*bs; 25349b5e25fSSatish Balay for (i=0; i<bn; i++){ /* for each block row */ 25449b5e25fSSatish Balay M = ai[i+1] - ai[i]; 25549b5e25fSSatish Balay for (j=0; j<M; j++){ 25649b5e25fSSatish Balay itmp = aj[ai[i] + j]; /* block column value */ 25749b5e25fSSatish Balay if (itmp == bn){ 25849b5e25fSSatish Balay aa_i = aa + bs2*(ai[i] + j) + bs*bp; 25949b5e25fSSatish Balay for (k=0; k<bs; k++) { 26049b5e25fSSatish Balay *cols_i++ = i*bs+k; 26149b5e25fSSatish Balay *v_i++ = aa_i[k]; 26249b5e25fSSatish Balay } 26349b5e25fSSatish Balay *ncols += bs; 26449b5e25fSSatish Balay break; 26549b5e25fSSatish Balay } 26649b5e25fSSatish Balay } 26749b5e25fSSatish Balay } 2685ddb2528SHong Zhang #endif 26949b5e25fSSatish Balay PetscFunctionReturn(0); 27049b5e25fSSatish Balay } 27149b5e25fSSatish Balay 2724a2ae208SSatish Balay #undef __FUNCT__ 2734a2ae208SSatish Balay #define __FUNCT__ "MatRestoreRow_SeqSBAIJ" 27413f74950SBarry Smith PetscErrorCode MatRestoreRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v) 27549b5e25fSSatish Balay { 276dfbe8321SBarry Smith PetscErrorCode ierr; 27749b5e25fSSatish Balay 27849b5e25fSSatish Balay PetscFunctionBegin; 27905b42c5fSBarry Smith if (idx) {ierr = PetscFree(*idx);CHKERRQ(ierr);} 28005b42c5fSBarry Smith if (v) {ierr = PetscFree(*v);CHKERRQ(ierr);} 28149b5e25fSSatish Balay PetscFunctionReturn(0); 28249b5e25fSSatish Balay } 28349b5e25fSSatish Balay 2844a2ae208SSatish Balay #undef __FUNCT__ 285f5edf698SHong Zhang #define __FUNCT__ "MatGetRowUpperTriangular_SeqSBAIJ" 286f5edf698SHong Zhang PetscErrorCode MatGetRowUpperTriangular_SeqSBAIJ(Mat A) 287f5edf698SHong Zhang { 288f5edf698SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 289f5edf698SHong Zhang 290f5edf698SHong Zhang PetscFunctionBegin; 291f5edf698SHong Zhang a->getrow_utriangular = PETSC_TRUE; 292f5edf698SHong Zhang PetscFunctionReturn(0); 293f5edf698SHong Zhang } 294f5edf698SHong Zhang #undef __FUNCT__ 295f5edf698SHong Zhang #define __FUNCT__ "MatRestoreRowUpperTriangular_SeqSBAIJ" 296f5edf698SHong Zhang PetscErrorCode MatRestoreRowUpperTriangular_SeqSBAIJ(Mat A) 297f5edf698SHong Zhang { 298f5edf698SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 299f5edf698SHong Zhang 300f5edf698SHong Zhang PetscFunctionBegin; 301f5edf698SHong Zhang a->getrow_utriangular = PETSC_FALSE; 302f5edf698SHong Zhang PetscFunctionReturn(0); 303f5edf698SHong Zhang } 304f5edf698SHong Zhang 305f5edf698SHong Zhang #undef __FUNCT__ 3064a2ae208SSatish Balay #define __FUNCT__ "MatTranspose_SeqSBAIJ" 307dfbe8321SBarry Smith PetscErrorCode MatTranspose_SeqSBAIJ(Mat A,Mat *B) 30849b5e25fSSatish Balay { 309dfbe8321SBarry Smith PetscErrorCode ierr; 31049b5e25fSSatish Balay PetscFunctionBegin; 311999d9058SBarry Smith ierr = MatDuplicate(A,MAT_COPY_VALUES,B);CHKERRQ(ierr); 3128115998fSBarry Smith PetscFunctionReturn(0); 31349b5e25fSSatish Balay } 31449b5e25fSSatish Balay 3154a2ae208SSatish Balay #undef __FUNCT__ 3164a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_ASCII" 3176849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_ASCII(Mat A,PetscViewer viewer) 31849b5e25fSSatish Balay { 31949b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 320dfbe8321SBarry Smith PetscErrorCode ierr; 321899cda47SBarry Smith PetscInt i,j,bs = A->rmap.bs,k,l,bs2=a->bs2; 3222dcb1b2aSMatthew Knepley const char *name; 323f3ef73ceSBarry Smith PetscViewerFormat format; 32449b5e25fSSatish Balay 32549b5e25fSSatish Balay PetscFunctionBegin; 32680fe4e49SBarry Smith ierr = PetscObjectGetName((PetscObject)A,&name);CHKERRQ(ierr); 327b0a32e0cSBarry Smith ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 328456192e2SBarry Smith if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) { 32977431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," block size is %D\n",bs);CHKERRQ(ierr); 330fb9695e5SSatish Balay } else if (format == PETSC_VIEWER_ASCII_MATLAB) { 33129bbc08cSBarry Smith SETERRQ(PETSC_ERR_SUP,"Matlab format not supported"); 332fb9695e5SSatish Balay } else if (format == PETSC_VIEWER_ASCII_COMMON) { 333b0a32e0cSBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_NO);CHKERRQ(ierr); 33449b5e25fSSatish Balay for (i=0; i<a->mbs; i++) { 33549b5e25fSSatish Balay for (j=0; j<bs; j++) { 33677431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr); 33749b5e25fSSatish Balay for (k=a->i[i]; k<a->i[i+1]; k++) { 33849b5e25fSSatish Balay for (l=0; l<bs; l++) { 33949b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX) 34049b5e25fSSatish Balay if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) { 341a83599f4SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %G + %G i) ",bs*a->j[k]+l, 34249b5e25fSSatish Balay PetscRealPart(a->a[bs2*k + l*bs + j]),PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 34349b5e25fSSatish Balay } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) { 344a83599f4SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %G - %G i) ",bs*a->j[k]+l, 34549b5e25fSSatish Balay PetscRealPart(a->a[bs2*k + l*bs + j]),-PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 34649b5e25fSSatish Balay } else if (PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) { 347a83599f4SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",bs*a->j[k]+l,PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 34849b5e25fSSatish Balay } 34949b5e25fSSatish Balay #else 35049b5e25fSSatish Balay if (a->a[bs2*k + l*bs + j] != 0.0) { 351a83599f4SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",bs*a->j[k]+l,a->a[bs2*k + l*bs + j]);CHKERRQ(ierr); 35249b5e25fSSatish Balay } 35349b5e25fSSatish Balay #endif 35449b5e25fSSatish Balay } 35549b5e25fSSatish Balay } 356b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 35749b5e25fSSatish Balay } 35849b5e25fSSatish Balay } 359b0a32e0cSBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_YES);CHKERRQ(ierr); 360c1490034SHong Zhang } else if (format == PETSC_VIEWER_ASCII_FACTOR_INFO) { 361c1490034SHong Zhang PetscFunctionReturn(0); 36249b5e25fSSatish Balay } else { 363b0a32e0cSBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_NO);CHKERRQ(ierr); 36449b5e25fSSatish Balay for (i=0; i<a->mbs; i++) { 36549b5e25fSSatish Balay for (j=0; j<bs; j++) { 36677431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr); 36749b5e25fSSatish Balay for (k=a->i[i]; k<a->i[i+1]; k++) { 36849b5e25fSSatish Balay for (l=0; l<bs; l++) { 36949b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX) 37049b5e25fSSatish Balay if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0) { 371a83599f4SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %G + %G i) ",bs*a->j[k]+l, 37249b5e25fSSatish Balay PetscRealPart(a->a[bs2*k + l*bs + j]),PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 37349b5e25fSSatish Balay } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0) { 374a83599f4SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %G - %G i) ",bs*a->j[k]+l, 37549b5e25fSSatish Balay PetscRealPart(a->a[bs2*k + l*bs + j]),-PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 37649b5e25fSSatish Balay } else { 377a83599f4SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",bs*a->j[k]+l,PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 37849b5e25fSSatish Balay } 37949b5e25fSSatish Balay #else 380e9f7bc9eSHong Zhang ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",bs*a->j[k]+l,a->a[bs2*k + l*bs + j]);CHKERRQ(ierr); 38149b5e25fSSatish Balay #endif 38249b5e25fSSatish Balay } 38349b5e25fSSatish Balay } 384b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 38549b5e25fSSatish Balay } 38649b5e25fSSatish Balay } 387b0a32e0cSBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_YES);CHKERRQ(ierr); 38849b5e25fSSatish Balay } 389b0a32e0cSBarry Smith ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 39049b5e25fSSatish Balay PetscFunctionReturn(0); 39149b5e25fSSatish Balay } 39249b5e25fSSatish Balay 3934a2ae208SSatish Balay #undef __FUNCT__ 3944a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_Draw_Zoom" 3956849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw_Zoom(PetscDraw draw,void *Aa) 39649b5e25fSSatish Balay { 39749b5e25fSSatish Balay Mat A = (Mat) Aa; 39849b5e25fSSatish Balay Mat_SeqSBAIJ *a=(Mat_SeqSBAIJ*)A->data; 3996849ba73SBarry Smith PetscErrorCode ierr; 400899cda47SBarry Smith PetscInt row,i,j,k,l,mbs=a->mbs,color,bs=A->rmap.bs,bs2=a->bs2; 40113f74950SBarry Smith PetscMPIInt rank; 40249b5e25fSSatish Balay PetscReal xl,yl,xr,yr,x_l,x_r,y_l,y_r; 40349b5e25fSSatish Balay MatScalar *aa; 40449b5e25fSSatish Balay MPI_Comm comm; 405b0a32e0cSBarry Smith PetscViewer viewer; 40649b5e25fSSatish Balay 40749b5e25fSSatish Balay PetscFunctionBegin; 40849b5e25fSSatish Balay /* 40949b5e25fSSatish Balay This is nasty. If this is called from an originally parallel matrix 41049b5e25fSSatish Balay then all processes call this,but only the first has the matrix so the 41149b5e25fSSatish Balay rest should return immediately. 41249b5e25fSSatish Balay */ 41349b5e25fSSatish Balay ierr = PetscObjectGetComm((PetscObject)draw,&comm);CHKERRQ(ierr); 41449b5e25fSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 41549b5e25fSSatish Balay if (rank) PetscFunctionReturn(0); 41649b5e25fSSatish Balay 41749b5e25fSSatish Balay ierr = PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer);CHKERRQ(ierr); 41849b5e25fSSatish Balay 419b0a32e0cSBarry Smith ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr); 420b0a32e0cSBarry Smith PetscDrawString(draw, .3*(xl+xr), .3*(yl+yr), PETSC_DRAW_BLACK, "symmetric"); 42149b5e25fSSatish Balay 42249b5e25fSSatish Balay /* loop over matrix elements drawing boxes */ 423b0a32e0cSBarry Smith color = PETSC_DRAW_BLUE; 42449b5e25fSSatish Balay for (i=0,row=0; i<mbs; i++,row+=bs) { 42549b5e25fSSatish Balay for (j=a->i[i]; j<a->i[i+1]; j++) { 426899cda47SBarry Smith y_l = A->rmap.N - row - 1.0; y_r = y_l + 1.0; 42749b5e25fSSatish Balay x_l = a->j[j]*bs; x_r = x_l + 1.0; 42849b5e25fSSatish Balay aa = a->a + j*bs2; 42949b5e25fSSatish Balay for (k=0; k<bs; k++) { 43049b5e25fSSatish Balay for (l=0; l<bs; l++) { 43149b5e25fSSatish Balay if (PetscRealPart(*aa++) >= 0.) continue; 432b0a32e0cSBarry Smith ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr); 43349b5e25fSSatish Balay } 43449b5e25fSSatish Balay } 43549b5e25fSSatish Balay } 43649b5e25fSSatish Balay } 437b0a32e0cSBarry Smith color = PETSC_DRAW_CYAN; 43849b5e25fSSatish Balay for (i=0,row=0; i<mbs; i++,row+=bs) { 43949b5e25fSSatish Balay for (j=a->i[i]; j<a->i[i+1]; j++) { 440899cda47SBarry Smith y_l = A->rmap.N - row - 1.0; y_r = y_l + 1.0; 44149b5e25fSSatish Balay x_l = a->j[j]*bs; x_r = x_l + 1.0; 44249b5e25fSSatish Balay aa = a->a + j*bs2; 44349b5e25fSSatish Balay for (k=0; k<bs; k++) { 44449b5e25fSSatish Balay for (l=0; l<bs; l++) { 44549b5e25fSSatish Balay if (PetscRealPart(*aa++) != 0.) continue; 446b0a32e0cSBarry Smith ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr); 44749b5e25fSSatish Balay } 44849b5e25fSSatish Balay } 44949b5e25fSSatish Balay } 45049b5e25fSSatish Balay } 45149b5e25fSSatish Balay 452b0a32e0cSBarry Smith color = PETSC_DRAW_RED; 45349b5e25fSSatish Balay for (i=0,row=0; i<mbs; i++,row+=bs) { 45449b5e25fSSatish Balay for (j=a->i[i]; j<a->i[i+1]; j++) { 455899cda47SBarry Smith y_l = A->rmap.N - row - 1.0; y_r = y_l + 1.0; 45649b5e25fSSatish Balay x_l = a->j[j]*bs; x_r = x_l + 1.0; 45749b5e25fSSatish Balay aa = a->a + j*bs2; 45849b5e25fSSatish Balay for (k=0; k<bs; k++) { 45949b5e25fSSatish Balay for (l=0; l<bs; l++) { 46049b5e25fSSatish Balay if (PetscRealPart(*aa++) <= 0.) continue; 461b0a32e0cSBarry Smith ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr); 46249b5e25fSSatish Balay } 46349b5e25fSSatish Balay } 46449b5e25fSSatish Balay } 46549b5e25fSSatish Balay } 46649b5e25fSSatish Balay PetscFunctionReturn(0); 46749b5e25fSSatish Balay } 46849b5e25fSSatish Balay 4694a2ae208SSatish Balay #undef __FUNCT__ 4704a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_Draw" 4716849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw(Mat A,PetscViewer viewer) 47249b5e25fSSatish Balay { 473dfbe8321SBarry Smith PetscErrorCode ierr; 47449b5e25fSSatish Balay PetscReal xl,yl,xr,yr,w,h; 475b0a32e0cSBarry Smith PetscDraw draw; 47649b5e25fSSatish Balay PetscTruth isnull; 47749b5e25fSSatish Balay 47849b5e25fSSatish Balay PetscFunctionBegin; 47949b5e25fSSatish Balay 480b0a32e0cSBarry Smith ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr); 481b0a32e0cSBarry Smith ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr); if (isnull) PetscFunctionReturn(0); 48249b5e25fSSatish Balay 48349b5e25fSSatish Balay ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer);CHKERRQ(ierr); 484899cda47SBarry Smith xr = A->rmap.N; yr = A->rmap.N; h = yr/10.0; w = xr/10.0; 48549b5e25fSSatish Balay xr += w; yr += h; xl = -w; yl = -h; 486b0a32e0cSBarry Smith ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr); 487b0a32e0cSBarry Smith ierr = PetscDrawZoom(draw,MatView_SeqSBAIJ_Draw_Zoom,A);CHKERRQ(ierr); 48849b5e25fSSatish Balay ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",PETSC_NULL);CHKERRQ(ierr); 48949b5e25fSSatish Balay PetscFunctionReturn(0); 49049b5e25fSSatish Balay } 49149b5e25fSSatish Balay 4924a2ae208SSatish Balay #undef __FUNCT__ 4934a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ" 494dfbe8321SBarry Smith PetscErrorCode MatView_SeqSBAIJ(Mat A,PetscViewer viewer) 49549b5e25fSSatish Balay { 496dfbe8321SBarry Smith PetscErrorCode ierr; 49732077d6dSBarry Smith PetscTruth iascii,isdraw; 49849b5e25fSSatish Balay 49949b5e25fSSatish Balay PetscFunctionBegin; 50032077d6dSBarry Smith ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr); 501fb9695e5SSatish Balay ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_DRAW,&isdraw);CHKERRQ(ierr); 50232077d6dSBarry Smith if (iascii){ 50349b5e25fSSatish Balay ierr = MatView_SeqSBAIJ_ASCII(A,viewer);CHKERRQ(ierr); 50449b5e25fSSatish Balay } else if (isdraw) { 50549b5e25fSSatish Balay ierr = MatView_SeqSBAIJ_Draw(A,viewer);CHKERRQ(ierr); 50649b5e25fSSatish Balay } else { 507a5e6ed63SBarry Smith Mat B; 508ceb03754SKris Buschelman ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&B);CHKERRQ(ierr); 509a5e6ed63SBarry Smith ierr = MatView(B,viewer);CHKERRQ(ierr); 510a5e6ed63SBarry Smith ierr = MatDestroy(B);CHKERRQ(ierr); 51149b5e25fSSatish Balay } 51249b5e25fSSatish Balay PetscFunctionReturn(0); 51349b5e25fSSatish Balay } 51449b5e25fSSatish Balay 51549b5e25fSSatish Balay 5164a2ae208SSatish Balay #undef __FUNCT__ 5174a2ae208SSatish Balay #define __FUNCT__ "MatGetValues_SeqSBAIJ" 51813f74950SBarry Smith PetscErrorCode MatGetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],PetscScalar v[]) 51949b5e25fSSatish Balay { 520045c9aa0SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 52113f74950SBarry Smith PetscInt *rp,k,low,high,t,row,nrow,i,col,l,*aj = a->j; 52213f74950SBarry Smith PetscInt *ai = a->i,*ailen = a->ilen; 523899cda47SBarry Smith PetscInt brow,bcol,ridx,cidx,bs=A->rmap.bs,bs2=a->bs2; 52449b5e25fSSatish Balay MatScalar *ap,*aa = a->a,zero = 0.0; 52549b5e25fSSatish Balay 52649b5e25fSSatish Balay PetscFunctionBegin; 52749b5e25fSSatish Balay for (k=0; k<m; k++) { /* loop over rows */ 52849b5e25fSSatish Balay row = im[k]; brow = row/bs; 52977431f27SBarry Smith if (row < 0) SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"Negative row: %D",row); 530899cda47SBarry Smith if (row >= A->rmap.N) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,A->rmap.N-1); 53149b5e25fSSatish Balay rp = aj + ai[brow] ; ap = aa + bs2*ai[brow] ; 53249b5e25fSSatish Balay nrow = ailen[brow]; 53349b5e25fSSatish Balay for (l=0; l<n; l++) { /* loop over columns */ 53477431f27SBarry Smith if (in[l] < 0) SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"Negative column: %D",in[l]); 535899cda47SBarry Smith if (in[l] >= A->cmap.n) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",in[l],A->cmap.n-1); 53649b5e25fSSatish Balay col = in[l] ; 53749b5e25fSSatish Balay bcol = col/bs; 53849b5e25fSSatish Balay cidx = col%bs; 53949b5e25fSSatish Balay ridx = row%bs; 54049b5e25fSSatish Balay high = nrow; 54149b5e25fSSatish Balay low = 0; /* assume unsorted */ 54249b5e25fSSatish Balay while (high-low > 5) { 54349b5e25fSSatish Balay t = (low+high)/2; 54449b5e25fSSatish Balay if (rp[t] > bcol) high = t; 54549b5e25fSSatish Balay else low = t; 54649b5e25fSSatish Balay } 54749b5e25fSSatish Balay for (i=low; i<high; i++) { 54849b5e25fSSatish Balay if (rp[i] > bcol) break; 54949b5e25fSSatish Balay if (rp[i] == bcol) { 55049b5e25fSSatish Balay *v++ = ap[bs2*i+bs*cidx+ridx]; 55149b5e25fSSatish Balay goto finished; 55249b5e25fSSatish Balay } 55349b5e25fSSatish Balay } 55449b5e25fSSatish Balay *v++ = zero; 55549b5e25fSSatish Balay finished:; 55649b5e25fSSatish Balay } 55749b5e25fSSatish Balay } 55849b5e25fSSatish Balay PetscFunctionReturn(0); 55949b5e25fSSatish Balay } 56049b5e25fSSatish Balay 56149b5e25fSSatish Balay 5624a2ae208SSatish Balay #undef __FUNCT__ 5634a2ae208SSatish Balay #define __FUNCT__ "MatSetValuesBlocked_SeqSBAIJ" 56413f74950SBarry Smith PetscErrorCode MatSetValuesBlocked_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is) 56549b5e25fSSatish Balay { 5660880e062SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 5676849ba73SBarry Smith PetscErrorCode ierr; 568e2ee6c50SBarry Smith PetscInt *rp,k,low,high,t,ii,jj,row,nrow,i,col,l,rmax,N,lastcol = -1; 56913f74950SBarry Smith PetscInt *imax=a->imax,*ai=a->i,*ailen=a->ilen; 570899cda47SBarry Smith PetscInt *aj=a->j,nonew=a->nonew,bs2=a->bs2,bs=A->rmap.bs,stepval; 5710880e062SHong Zhang PetscTruth roworiented=a->roworiented; 572f15d580aSBarry Smith const MatScalar *value = v; 573f15d580aSBarry Smith MatScalar *ap,*aa = a->a,*bap; 5740880e062SHong Zhang 57549b5e25fSSatish Balay PetscFunctionBegin; 5760880e062SHong Zhang if (roworiented) { 5770880e062SHong Zhang stepval = (n-1)*bs; 5780880e062SHong Zhang } else { 5790880e062SHong Zhang stepval = (m-1)*bs; 5800880e062SHong Zhang } 5810880e062SHong Zhang for (k=0; k<m; k++) { /* loop over added rows */ 5820880e062SHong Zhang row = im[k]; 5830880e062SHong Zhang if (row < 0) continue; 5842515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 58577431f27SBarry Smith if (row >= a->mbs) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,a->mbs-1); 5860880e062SHong Zhang #endif 5870880e062SHong Zhang rp = aj + ai[row]; 5880880e062SHong Zhang ap = aa + bs2*ai[row]; 5890880e062SHong Zhang rmax = imax[row]; 5900880e062SHong Zhang nrow = ailen[row]; 5910880e062SHong Zhang low = 0; 592818f2c47SBarry Smith high = nrow; 5930880e062SHong Zhang for (l=0; l<n; l++) { /* loop over added columns */ 5940880e062SHong Zhang if (in[l] < 0) continue; 5950880e062SHong Zhang col = in[l]; 5962515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 59777431f27SBarry Smith if (col >= a->nbs) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",col,a->nbs-1); 598b1823623SSatish Balay #endif 5990880e062SHong Zhang if (col < row) continue; /* ignore lower triangular block */ 6000880e062SHong Zhang if (roworiented) { 6010880e062SHong Zhang value = v + k*(stepval+bs)*bs + l*bs; 6020880e062SHong Zhang } else { 6030880e062SHong Zhang value = v + l*(stepval+bs)*bs + k*bs; 6040880e062SHong Zhang } 6057cd84e04SBarry Smith if (col <= lastcol) low = 0; else high = nrow; 606e2ee6c50SBarry Smith lastcol = col; 6070880e062SHong Zhang while (high-low > 7) { 6080880e062SHong Zhang t = (low+high)/2; 6090880e062SHong Zhang if (rp[t] > col) high = t; 6100880e062SHong Zhang else low = t; 6110880e062SHong Zhang } 6120880e062SHong Zhang for (i=low; i<high; i++) { 6130880e062SHong Zhang if (rp[i] > col) break; 6140880e062SHong Zhang if (rp[i] == col) { 6150880e062SHong Zhang bap = ap + bs2*i; 6160880e062SHong Zhang if (roworiented) { 6170880e062SHong Zhang if (is == ADD_VALUES) { 6180880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 6190880e062SHong Zhang for (jj=ii; jj<bs2; jj+=bs) { 6200880e062SHong Zhang bap[jj] += *value++; 6210880e062SHong Zhang } 6220880e062SHong Zhang } 6230880e062SHong Zhang } else { 6240880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 6250880e062SHong Zhang for (jj=ii; jj<bs2; jj+=bs) { 6260880e062SHong Zhang bap[jj] = *value++; 6270880e062SHong Zhang } 6280880e062SHong Zhang } 6290880e062SHong Zhang } 6300880e062SHong Zhang } else { 6310880e062SHong Zhang if (is == ADD_VALUES) { 6320880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 6330880e062SHong Zhang for (jj=0; jj<bs; jj++) { 6340880e062SHong Zhang *bap++ += *value++; 6350880e062SHong Zhang } 6360880e062SHong Zhang } 6370880e062SHong Zhang } else { 6380880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 6390880e062SHong Zhang for (jj=0; jj<bs; jj++) { 6400880e062SHong Zhang *bap++ = *value++; 6410880e062SHong Zhang } 6420880e062SHong Zhang } 6430880e062SHong Zhang } 6440880e062SHong Zhang } 6450880e062SHong Zhang goto noinsert2; 6460880e062SHong Zhang } 6470880e062SHong Zhang } 6480880e062SHong Zhang if (nonew == 1) goto noinsert2; 649085a36d4SBarry Smith if (nonew == -1) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col); 650421e10b8SBarry Smith MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,row,col,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar); 651c03d1d03SSatish Balay N = nrow++ - 1; high++; 6520880e062SHong Zhang /* shift up all the later entries in this row */ 6530880e062SHong Zhang for (ii=N; ii>=i; ii--) { 6540880e062SHong Zhang rp[ii+1] = rp[ii]; 6550880e062SHong Zhang ierr = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr); 6560880e062SHong Zhang } 6570880e062SHong Zhang if (N >= i) { 6580880e062SHong Zhang ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr); 6590880e062SHong Zhang } 6600880e062SHong Zhang rp[i] = col; 6610880e062SHong Zhang bap = ap + bs2*i; 6620880e062SHong Zhang if (roworiented) { 6630880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 6640880e062SHong Zhang for (jj=ii; jj<bs2; jj+=bs) { 6650880e062SHong Zhang bap[jj] = *value++; 6660880e062SHong Zhang } 6670880e062SHong Zhang } 6680880e062SHong Zhang } else { 6690880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 6700880e062SHong Zhang for (jj=0; jj<bs; jj++) { 6710880e062SHong Zhang *bap++ = *value++; 6720880e062SHong Zhang } 6730880e062SHong Zhang } 6740880e062SHong Zhang } 6750880e062SHong Zhang noinsert2:; 6760880e062SHong Zhang low = i; 6770880e062SHong Zhang } 6780880e062SHong Zhang ailen[row] = nrow; 6790880e062SHong Zhang } 6800880e062SHong Zhang PetscFunctionReturn(0); 68149b5e25fSSatish Balay } 68249b5e25fSSatish Balay 6834a2ae208SSatish Balay #undef __FUNCT__ 6844a2ae208SSatish Balay #define __FUNCT__ "MatAssemblyEnd_SeqSBAIJ" 685dfbe8321SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ(Mat A,MatAssemblyType mode) 68649b5e25fSSatish Balay { 68749b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 6886849ba73SBarry Smith PetscErrorCode ierr; 68913f74950SBarry Smith PetscInt fshift = 0,i,j,*ai = a->i,*aj = a->j,*imax = a->imax; 690899cda47SBarry Smith PetscInt m = A->rmap.N,*ip,N,*ailen = a->ilen; 69113f74950SBarry Smith PetscInt mbs = a->mbs,bs2 = a->bs2,rmax = 0; 69249b5e25fSSatish Balay MatScalar *aa = a->a,*ap; 69349b5e25fSSatish Balay 69449b5e25fSSatish Balay PetscFunctionBegin; 69549b5e25fSSatish Balay if (mode == MAT_FLUSH_ASSEMBLY) PetscFunctionReturn(0); 69649b5e25fSSatish Balay 69749b5e25fSSatish Balay if (m) rmax = ailen[0]; 69849b5e25fSSatish Balay for (i=1; i<mbs; i++) { 69949b5e25fSSatish Balay /* move each row back by the amount of empty slots (fshift) before it*/ 70049b5e25fSSatish Balay fshift += imax[i-1] - ailen[i-1]; 70149b5e25fSSatish Balay rmax = PetscMax(rmax,ailen[i]); 70249b5e25fSSatish Balay if (fshift) { 70349b5e25fSSatish Balay ip = aj + ai[i]; ap = aa + bs2*ai[i]; 70449b5e25fSSatish Balay N = ailen[i]; 70549b5e25fSSatish Balay for (j=0; j<N; j++) { 70649b5e25fSSatish Balay ip[j-fshift] = ip[j]; 70749b5e25fSSatish Balay ierr = PetscMemcpy(ap+(j-fshift)*bs2,ap+j*bs2,bs2*sizeof(MatScalar));CHKERRQ(ierr); 70849b5e25fSSatish Balay } 70949b5e25fSSatish Balay } 71049b5e25fSSatish Balay ai[i] = ai[i-1] + ailen[i-1]; 71149b5e25fSSatish Balay } 71249b5e25fSSatish Balay if (mbs) { 71349b5e25fSSatish Balay fshift += imax[mbs-1] - ailen[mbs-1]; 71449b5e25fSSatish Balay ai[mbs] = ai[mbs-1] + ailen[mbs-1]; 71549b5e25fSSatish Balay } 71649b5e25fSSatish Balay /* reset ilen and imax for each row */ 71749b5e25fSSatish Balay for (i=0; i<mbs; i++) { 71849b5e25fSSatish Balay ailen[i] = imax[i] = ai[i+1] - ai[i]; 71949b5e25fSSatish Balay } 7206c6c5352SBarry Smith a->nz = ai[mbs]; 72149b5e25fSSatish Balay 722b424e231SHong Zhang /* diagonals may have moved, reset it */ 723b424e231SHong Zhang if (a->diag) { 72413f74950SBarry Smith ierr = PetscMemcpy(a->diag,ai,(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr); 72549b5e25fSSatish Balay } 726899cda47SBarry 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); 727ae15b995SBarry Smith ierr = PetscInfo1(A,"Number of mallocs during MatSetValues is %D\n",a->reallocs);CHKERRQ(ierr); 728ae15b995SBarry Smith ierr = PetscInfo1(A,"Most nonzeros blocks in any row is %D\n",rmax);CHKERRQ(ierr); 72949b5e25fSSatish Balay a->reallocs = 0; 73049b5e25fSSatish Balay A->info.nz_unneeded = (PetscReal)fshift*bs2; 73149b5e25fSSatish Balay PetscFunctionReturn(0); 73249b5e25fSSatish Balay } 73349b5e25fSSatish Balay 73449b5e25fSSatish Balay /* 73549b5e25fSSatish Balay This function returns an array of flags which indicate the locations of contiguous 73649b5e25fSSatish Balay blocks that should be zeroed. for eg: if bs = 3 and is = [0,1,2,3,5,6,7,8,9] 73749b5e25fSSatish Balay then the resulting sizes = [3,1,1,3,1] correspondig to sets [(0,1,2),(3),(5),(6,7,8),(9)] 73849b5e25fSSatish Balay Assume: sizes should be long enough to hold all the values. 73949b5e25fSSatish Balay */ 7404a2ae208SSatish Balay #undef __FUNCT__ 7414a2ae208SSatish Balay #define __FUNCT__ "MatZeroRows_SeqSBAIJ_Check_Blocks" 74213f74950SBarry Smith PetscErrorCode MatZeroRows_SeqSBAIJ_Check_Blocks(PetscInt idx[],PetscInt n,PetscInt bs,PetscInt sizes[], PetscInt *bs_max) 74349b5e25fSSatish Balay { 74413f74950SBarry Smith PetscInt i,j,k,row; 74549b5e25fSSatish Balay PetscTruth flg; 74649b5e25fSSatish Balay 74749b5e25fSSatish Balay PetscFunctionBegin; 74849b5e25fSSatish Balay for (i=0,j=0; i<n; j++) { 74949b5e25fSSatish Balay row = idx[i]; 75049b5e25fSSatish Balay if (row%bs!=0) { /* Not the begining of a block */ 75149b5e25fSSatish Balay sizes[j] = 1; 75249b5e25fSSatish Balay i++; 75349b5e25fSSatish Balay } else if (i+bs > n) { /* Beginning of a block, but complete block doesn't exist (at idx end) */ 75449b5e25fSSatish Balay sizes[j] = 1; /* Also makes sure atleast 'bs' values exist for next else */ 75549b5e25fSSatish Balay i++; 75649b5e25fSSatish Balay } else { /* Begining of the block, so check if the complete block exists */ 75749b5e25fSSatish Balay flg = PETSC_TRUE; 75849b5e25fSSatish Balay for (k=1; k<bs; k++) { 75949b5e25fSSatish Balay if (row+k != idx[i+k]) { /* break in the block */ 76049b5e25fSSatish Balay flg = PETSC_FALSE; 76149b5e25fSSatish Balay break; 76249b5e25fSSatish Balay } 76349b5e25fSSatish Balay } 764abc0a331SBarry Smith if (flg) { /* No break in the bs */ 76549b5e25fSSatish Balay sizes[j] = bs; 76649b5e25fSSatish Balay i+= bs; 76749b5e25fSSatish Balay } else { 76849b5e25fSSatish Balay sizes[j] = 1; 76949b5e25fSSatish Balay i++; 77049b5e25fSSatish Balay } 77149b5e25fSSatish Balay } 77249b5e25fSSatish Balay } 77349b5e25fSSatish Balay *bs_max = j; 77449b5e25fSSatish Balay PetscFunctionReturn(0); 77549b5e25fSSatish Balay } 77649b5e25fSSatish Balay 77749b5e25fSSatish Balay 77849b5e25fSSatish Balay /* Only add/insert a(i,j) with i<=j (blocks). 77949b5e25fSSatish Balay Any a(i,j) with i>j input by user is ingored. 78049b5e25fSSatish Balay */ 78149b5e25fSSatish Balay 7824a2ae208SSatish Balay #undef __FUNCT__ 7834a2ae208SSatish Balay #define __FUNCT__ "MatSetValues_SeqSBAIJ" 78413f74950SBarry Smith PetscErrorCode MatSetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is) 78549b5e25fSSatish Balay { 78649b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 7876849ba73SBarry Smith PetscErrorCode ierr; 788e2ee6c50SBarry Smith PetscInt *rp,k,low,high,t,ii,row,nrow,i,col,l,rmax,N,lastcol = -1; 78913f74950SBarry Smith PetscInt *imax=a->imax,*ai=a->i,*ailen=a->ilen,roworiented=a->roworiented; 790899cda47SBarry Smith PetscInt *aj=a->j,nonew=a->nonew,bs=A->rmap.bs,brow,bcol; 79113f74950SBarry Smith PetscInt ridx,cidx,bs2=a->bs2; 79249b5e25fSSatish Balay MatScalar *ap,value,*aa=a->a,*bap; 79349b5e25fSSatish Balay 79449b5e25fSSatish Balay PetscFunctionBegin; 79549b5e25fSSatish Balay for (k=0; k<m; k++) { /* loop over added rows */ 79649b5e25fSSatish Balay row = im[k]; /* row number */ 79749b5e25fSSatish Balay brow = row/bs; /* block row number */ 79849b5e25fSSatish Balay if (row < 0) continue; 7992515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 800899cda47SBarry Smith if (row >= A->rmap.N) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,A->rmap.N-1); 80149b5e25fSSatish Balay #endif 80249b5e25fSSatish Balay rp = aj + ai[brow]; /*ptr to beginning of column value of the row block*/ 80349b5e25fSSatish Balay ap = aa + bs2*ai[brow]; /*ptr to beginning of element value of the row block*/ 80449b5e25fSSatish Balay rmax = imax[brow]; /* maximum space allocated for this row */ 80549b5e25fSSatish Balay nrow = ailen[brow]; /* actual length of this row */ 80649b5e25fSSatish Balay low = 0; 80749b5e25fSSatish Balay 80849b5e25fSSatish Balay for (l=0; l<n; l++) { /* loop over added columns */ 80949b5e25fSSatish Balay if (in[l] < 0) continue; 8102515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 811899cda47SBarry Smith if (in[l] >= A->rmap.N) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",in[l],A->rmap.N-1); 81249b5e25fSSatish Balay #endif 81349b5e25fSSatish Balay col = in[l]; 81449b5e25fSSatish Balay bcol = col/bs; /* block col number */ 81549b5e25fSSatish Balay 816941593c8SHong Zhang if (brow > bcol) { 817941593c8SHong Zhang if (a->ignore_ltriangular){ 818941593c8SHong Zhang continue; /* ignore lower triangular values */ 819941593c8SHong Zhang } else { 820941593c8SHong Zhang SETERRQ(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)"); 821941593c8SHong Zhang } 822941593c8SHong Zhang } 823f4989cb3SHong Zhang 82449b5e25fSSatish Balay ridx = row % bs; cidx = col % bs; /*row and col index inside the block */ 8258549e402SHong Zhang if ((brow==bcol && ridx<=cidx) || (brow<bcol)){ 82649b5e25fSSatish Balay /* element value a(k,l) */ 82749b5e25fSSatish Balay if (roworiented) { 82849b5e25fSSatish Balay value = v[l + k*n]; 82949b5e25fSSatish Balay } else { 83049b5e25fSSatish Balay value = v[k + l*m]; 83149b5e25fSSatish Balay } 83249b5e25fSSatish Balay 83349b5e25fSSatish Balay /* move pointer bap to a(k,l) quickly and add/insert value */ 8347cd84e04SBarry Smith if (col <= lastcol) low = 0; high = nrow; 835e2ee6c50SBarry Smith lastcol = col; 83649b5e25fSSatish Balay while (high-low > 7) { 83749b5e25fSSatish Balay t = (low+high)/2; 83849b5e25fSSatish Balay if (rp[t] > bcol) high = t; 83949b5e25fSSatish Balay else low = t; 84049b5e25fSSatish Balay } 84149b5e25fSSatish Balay for (i=low; i<high; i++) { 84249b5e25fSSatish Balay if (rp[i] > bcol) break; 84349b5e25fSSatish Balay if (rp[i] == bcol) { 84449b5e25fSSatish Balay bap = ap + bs2*i + bs*cidx + ridx; 84549b5e25fSSatish Balay if (is == ADD_VALUES) *bap += value; 84649b5e25fSSatish Balay else *bap = value; 8478549e402SHong Zhang /* for diag block, add/insert its symmetric element a(cidx,ridx) */ 8488549e402SHong Zhang if (brow == bcol && ridx < cidx){ 8498549e402SHong Zhang bap = ap + bs2*i + bs*ridx + cidx; 8508549e402SHong Zhang if (is == ADD_VALUES) *bap += value; 8518549e402SHong Zhang else *bap = value; 8528549e402SHong Zhang } 85349b5e25fSSatish Balay goto noinsert1; 85449b5e25fSSatish Balay } 85549b5e25fSSatish Balay } 85649b5e25fSSatish Balay 85749b5e25fSSatish Balay if (nonew == 1) goto noinsert1; 858085a36d4SBarry Smith if (nonew == -1) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col); 859421e10b8SBarry Smith MatSeqXAIJReallocateAIJ(A,a->mbs,bs2,nrow,brow,bcol,rmax,aa,ai,aj,rp,ap,imax,nonew,MatScalar); 86049b5e25fSSatish Balay 861c03d1d03SSatish Balay N = nrow++ - 1; high++; 86249b5e25fSSatish Balay /* shift up all the later entries in this row */ 86349b5e25fSSatish Balay for (ii=N; ii>=i; ii--) { 86449b5e25fSSatish Balay rp[ii+1] = rp[ii]; 86549b5e25fSSatish Balay ierr = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr); 86649b5e25fSSatish Balay } 86749b5e25fSSatish Balay if (N>=i) { 86849b5e25fSSatish Balay ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr); 86949b5e25fSSatish Balay } 87049b5e25fSSatish Balay rp[i] = bcol; 87149b5e25fSSatish Balay ap[bs2*i + bs*cidx + ridx] = value; 87249b5e25fSSatish Balay noinsert1:; 87349b5e25fSSatish Balay low = i; 8748549e402SHong Zhang } 87549b5e25fSSatish Balay } /* end of loop over added columns */ 87649b5e25fSSatish Balay ailen[brow] = nrow; 87749b5e25fSSatish Balay } /* end of loop over added rows */ 87849b5e25fSSatish Balay PetscFunctionReturn(0); 87949b5e25fSSatish Balay } 88049b5e25fSSatish Balay 8816849ba73SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorSymbolic_SeqSBAIJ(Mat,IS,MatFactorInfo*,Mat*); 8826849ba73SBarry Smith EXTERN PetscErrorCode MatCholeskyFactor_SeqSBAIJ(Mat,IS,MatFactorInfo*); 88313f74950SBarry Smith EXTERN PetscErrorCode MatIncreaseOverlap_SeqSBAIJ(Mat,PetscInt,IS[],PetscInt); 88413f74950SBarry Smith EXTERN PetscErrorCode MatGetSubMatrix_SeqSBAIJ(Mat,IS,IS,PetscInt,MatReuse,Mat*); 88513f74950SBarry Smith EXTERN PetscErrorCode MatGetSubMatrices_SeqSBAIJ(Mat,PetscInt,const IS[],const IS[],MatReuse,Mat*[]); 886f4df32b1SMatthew Knepley EXTERN PetscErrorCode MatScale_SeqSBAIJ(Mat,PetscScalar); 8876849ba73SBarry Smith EXTERN PetscErrorCode MatNorm_SeqSBAIJ(Mat,NormType,PetscReal *); 8886849ba73SBarry Smith EXTERN PetscErrorCode MatEqual_SeqSBAIJ(Mat,Mat,PetscTruth*); 8896849ba73SBarry Smith EXTERN PetscErrorCode MatGetDiagonal_SeqSBAIJ(Mat,Vec); 8906849ba73SBarry Smith EXTERN PetscErrorCode MatDiagonalScale_SeqSBAIJ(Mat,Vec,Vec); 8916849ba73SBarry Smith EXTERN PetscErrorCode MatGetInfo_SeqSBAIJ(Mat,MatInfoType,MatInfo *); 8926849ba73SBarry Smith EXTERN PetscErrorCode MatZeroEntries_SeqSBAIJ(Mat); 8936849ba73SBarry Smith EXTERN PetscErrorCode MatGetRowMax_SeqSBAIJ(Mat,Vec); 89449b5e25fSSatish Balay 8956849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_N(Mat,Vec,Vec); 8966849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_1(Mat,Vec,Vec); 8976849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_2(Mat,Vec,Vec); 8986849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_3(Mat,Vec,Vec); 8996849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_4(Mat,Vec,Vec); 9006849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_5(Mat,Vec,Vec); 9016849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_6(Mat,Vec,Vec); 9026849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_7(Mat,Vec,Vec); 90349b5e25fSSatish Balay 9046849ba73SBarry Smith EXTERN PetscErrorCode MatSolves_SeqSBAIJ_1(Mat,Vecs,Vecs); 905d59c15a7SBarry Smith 9066849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_1_NaturalOrdering(Mat,Vec,Vec); 9076849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_2_NaturalOrdering(Mat,Vec,Vec); 9086849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_3_NaturalOrdering(Mat,Vec,Vec); 9096849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_4_NaturalOrdering(Mat,Vec,Vec); 9106849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_5_NaturalOrdering(Mat,Vec,Vec); 9116849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_6_NaturalOrdering(Mat,Vec,Vec); 9126849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_7_NaturalOrdering(Mat,Vec,Vec); 9136849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_N_NaturalOrdering(Mat,Vec,Vec); 91407f98182SSatish Balay 915af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_N(Mat,MatFactorInfo*,Mat*); 916af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_1(Mat,MatFactorInfo*,Mat*); 917af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_2(Mat,MatFactorInfo*,Mat*); 918af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_3(Mat,MatFactorInfo*,Mat*); 919af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_4(Mat,MatFactorInfo*,Mat*); 920af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_5(Mat,MatFactorInfo*,Mat*); 921af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_6(Mat,MatFactorInfo*,Mat*); 922af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_7(Mat,MatFactorInfo*,Mat*); 92313f74950SBarry Smith EXTERN PetscErrorCode MatGetInertia_SeqSBAIJ(Mat,PetscInt*,PetscInt*,PetscInt*); 92449b5e25fSSatish Balay 9256849ba73SBarry Smith EXTERN PetscErrorCode MatMult_SeqSBAIJ_1(Mat,Vec,Vec); 9266849ba73SBarry Smith EXTERN PetscErrorCode MatMult_SeqSBAIJ_2(Mat,Vec,Vec); 9276849ba73SBarry Smith EXTERN PetscErrorCode MatMult_SeqSBAIJ_3(Mat,Vec,Vec); 9286849ba73SBarry Smith EXTERN PetscErrorCode MatMult_SeqSBAIJ_4(Mat,Vec,Vec); 9296849ba73SBarry Smith EXTERN PetscErrorCode MatMult_SeqSBAIJ_5(Mat,Vec,Vec); 9306849ba73SBarry Smith EXTERN PetscErrorCode MatMult_SeqSBAIJ_6(Mat,Vec,Vec); 9316849ba73SBarry Smith EXTERN PetscErrorCode MatMult_SeqSBAIJ_7(Mat,Vec,Vec); 9326849ba73SBarry Smith EXTERN PetscErrorCode MatMult_SeqSBAIJ_N(Mat,Vec,Vec); 93349b5e25fSSatish Balay 9346849ba73SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_1(Mat,Vec,Vec,Vec); 9356849ba73SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_2(Mat,Vec,Vec,Vec); 9366849ba73SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_3(Mat,Vec,Vec,Vec); 9376849ba73SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_4(Mat,Vec,Vec,Vec); 9386849ba73SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_5(Mat,Vec,Vec,Vec); 9396849ba73SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_6(Mat,Vec,Vec,Vec); 9406849ba73SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_7(Mat,Vec,Vec,Vec); 9416849ba73SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_N(Mat,Vec,Vec,Vec); 94249b5e25fSSatish Balay 9434a2ae208SSatish Balay #undef __FUNCT__ 9444d101231SSatish Balay #define __FUNCT__ "MatICCFactor_SeqSBAIJ" 945dfbe8321SBarry Smith PetscErrorCode MatICCFactor_SeqSBAIJ(Mat inA,IS row,MatFactorInfo *info) 94649b5e25fSSatish Balay { 9474ccecd49SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)inA->data; 94849b5e25fSSatish Balay Mat outA; 949dfbe8321SBarry Smith PetscErrorCode ierr; 950c84f5b01SHong Zhang PetscTruth row_identity; 95149b5e25fSSatish Balay 95249b5e25fSSatish Balay PetscFunctionBegin; 953c84f5b01SHong Zhang if (info->levels != 0) SETERRQ(PETSC_ERR_SUP,"Only levels=0 is supported for in-place icc"); 954c84f5b01SHong Zhang ierr = ISIdentity(row,&row_identity);CHKERRQ(ierr); 955c84f5b01SHong Zhang if (!row_identity) SETERRQ(PETSC_ERR_SUP,"Matrix reordering is not supported"); 956c84f5b01SHong Zhang if (inA->rmap.bs != 1) SETERRQ1(PETSC_ERR_SUP,"Matrix block size %D is not supported",inA->rmap.bs); /* Need to replace MatCholeskyFactorSymbolic_SeqSBAIJ_MSR()! */ 957c84f5b01SHong Zhang 95849b5e25fSSatish Balay outA = inA; 9591260e1f8SHong Zhang inA->factor = FACTOR_CHOLESKY; 96049b5e25fSSatish Balay 9611a3463dfSHong Zhang ierr = MatMarkDiagonal_SeqSBAIJ(inA);CHKERRQ(ierr); 96249b5e25fSSatish Balay /* 963c84f5b01SHong Zhang Blocksize < 8 have a special faster factorization/solver 964c84f5b01SHong Zhang for ICC(0) factorization with natural ordering 96549b5e25fSSatish Balay */ 966c84f5b01SHong Zhang switch (inA->rmap.bs){ /* Note: row_identity = PETSC_TRUE! */ 96749b5e25fSSatish Balay case 1: 968c84f5b01SHong Zhang inA->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1_NaturalOrdering; 9690fe9e5beSBarry Smith inA->ops->solve = MatSolve_SeqSBAIJ_1_NaturalOrdering; 97007f98182SSatish Balay inA->ops->solvetranspose = MatSolve_SeqSBAIJ_1_NaturalOrdering; 971d59c15a7SBarry Smith inA->ops->solves = MatSolves_SeqSBAIJ_1; 972c84f5b01SHong Zhang ierr = PetscInfo(inA,"Using special in-place natural ordering solvetrans BS=1\n");CHKERRQ(ierr); 973c84f5b01SHong Zhang break; 97449b5e25fSSatish Balay case 2: 975c84f5b01SHong Zhang inA->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering; 9761a3463dfSHong Zhang inA->ops->solve = MatSolve_SeqSBAIJ_2_NaturalOrdering; 9771a3463dfSHong Zhang inA->ops->solvetranspose = MatSolve_SeqSBAIJ_2_NaturalOrdering; 978ae15b995SBarry Smith ierr = PetscInfo(inA,"Using special in-place natural ordering factor and solve BS=2\n");CHKERRQ(ierr); 97949b5e25fSSatish Balay break; 98049b5e25fSSatish Balay case 3: 981c84f5b01SHong Zhang inA->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering; 9821a3463dfSHong Zhang inA->ops->solve = MatSolve_SeqSBAIJ_3_NaturalOrdering; 9831a3463dfSHong Zhang inA->ops->solvetranspose = MatSolve_SeqSBAIJ_3_NaturalOrdering; 984ae15b995SBarry Smith ierr = PetscInfo(inA,"Using special in-place natural ordering factor and solve BS=3\n");CHKERRQ(ierr); 98549b5e25fSSatish Balay break; 98649b5e25fSSatish Balay case 4: 987c84f5b01SHong Zhang inA->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering; 9881a3463dfSHong Zhang inA->ops->solve = MatSolve_SeqSBAIJ_4_NaturalOrdering; 9891a3463dfSHong Zhang inA->ops->solvetranspose = MatSolve_SeqSBAIJ_4_NaturalOrdering; 990ae15b995SBarry Smith ierr = PetscInfo(inA,"Using special in-place natural ordering factor and solve BS=4\n");CHKERRQ(ierr); 99149b5e25fSSatish Balay break; 99249b5e25fSSatish Balay case 5: 993c84f5b01SHong Zhang inA->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering; 9941a3463dfSHong Zhang inA->ops->solve = MatSolve_SeqSBAIJ_5_NaturalOrdering; 9951a3463dfSHong Zhang inA->ops->solvetranspose = MatSolve_SeqSBAIJ_5_NaturalOrdering; 996ae15b995SBarry Smith ierr = PetscInfo(inA,"Using special in-place natural ordering factor and solve BS=5\n");CHKERRQ(ierr); 99749b5e25fSSatish Balay break; 99849b5e25fSSatish Balay case 6: 999c84f5b01SHong Zhang inA->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering; 10001a3463dfSHong Zhang inA->ops->solve = MatSolve_SeqSBAIJ_6_NaturalOrdering; 10011a3463dfSHong Zhang inA->ops->solvetranspose = MatSolve_SeqSBAIJ_6_NaturalOrdering; 1002ae15b995SBarry Smith ierr = PetscInfo(inA,"Using special in-place natural ordering factor and solve BS=6\n");CHKERRQ(ierr); 100349b5e25fSSatish Balay break; 100449b5e25fSSatish Balay case 7: 1005c84f5b01SHong Zhang inA->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering; 10061a3463dfSHong Zhang inA->ops->solvetranspose = MatSolve_SeqSBAIJ_7_NaturalOrdering; 10071a3463dfSHong Zhang inA->ops->solve = MatSolve_SeqSBAIJ_7_NaturalOrdering; 1008ae15b995SBarry Smith ierr = PetscInfo(inA,"Using special in-place natural ordering factor and solve BS=7\n");CHKERRQ(ierr); 100949b5e25fSSatish Balay break; 101049b5e25fSSatish Balay default: 1011c84f5b01SHong Zhang inA->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering; 1012c84f5b01SHong Zhang inA->ops->solvetranspose = MatSolve_SeqSBAIJ_N_NaturalOrdering; 1013c84f5b01SHong Zhang inA->ops->solve = MatSolve_SeqSBAIJ_N_NaturalOrdering; 1014c84f5b01SHong Zhang break; 1015c84f5b01SHong Zhang } 101649b5e25fSSatish Balay 1017c84f5b01SHong Zhang a->row = row; 1018c84f5b01SHong Zhang a->col = row; 1019c84f5b01SHong Zhang ierr = PetscObjectReference((PetscObject)row);CHKERRQ(ierr); 1020c84f5b01SHong Zhang ierr = PetscObjectReference((PetscObject)row);CHKERRQ(ierr); 1021c84f5b01SHong Zhang 1022c84f5b01SHong Zhang /* Create the invert permutation so that it can be used in MatCholeskyFactorNumeric() */ 1023c84f5b01SHong Zhang if (a->icol) {ierr = ISInvertPermutation(row,PETSC_DECIDE, &a->icol);CHKERRQ(ierr);} 102452e6d16bSBarry Smith ierr = PetscLogObjectParent(inA,a->icol);CHKERRQ(ierr); 102549b5e25fSSatish Balay 102649b5e25fSSatish Balay if (!a->solve_work) { 1027c84f5b01SHong Zhang ierr = PetscMalloc((inA->rmap.N+inA->rmap.bs)*sizeof(PetscScalar),&a->solve_work);CHKERRQ(ierr); 1028c84f5b01SHong Zhang ierr = PetscLogObjectMemory(inA,(inA->rmap.N+inA->rmap.bs)*sizeof(PetscScalar));CHKERRQ(ierr); 102949b5e25fSSatish Balay } 103049b5e25fSSatish Balay 1031af281ebdSHong Zhang ierr = MatCholeskyFactorNumeric(inA,info,&outA);CHKERRQ(ierr); 103249b5e25fSSatish Balay PetscFunctionReturn(0); 103349b5e25fSSatish Balay } 1034950f1e5bSHong Zhang 103549b5e25fSSatish Balay EXTERN_C_BEGIN 10364a2ae208SSatish Balay #undef __FUNCT__ 10374a2ae208SSatish Balay #define __FUNCT__ "MatSeqSBAIJSetColumnIndices_SeqSBAIJ" 1038be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatSeqSBAIJSetColumnIndices_SeqSBAIJ(Mat mat,PetscInt *indices) 103949b5e25fSSatish Balay { 1040045c9aa0SHong Zhang Mat_SeqSBAIJ *baij = (Mat_SeqSBAIJ *)mat->data; 104113f74950SBarry Smith PetscInt i,nz,n; 104249b5e25fSSatish Balay 104349b5e25fSSatish Balay PetscFunctionBegin; 10446c6c5352SBarry Smith nz = baij->maxnz; 1045899cda47SBarry Smith n = mat->cmap.n; 104649b5e25fSSatish Balay for (i=0; i<nz; i++) { 104749b5e25fSSatish Balay baij->j[i] = indices[i]; 104849b5e25fSSatish Balay } 10496c6c5352SBarry Smith baij->nz = nz; 105049b5e25fSSatish Balay for (i=0; i<n; i++) { 105149b5e25fSSatish Balay baij->ilen[i] = baij->imax[i]; 105249b5e25fSSatish Balay } 105349b5e25fSSatish Balay PetscFunctionReturn(0); 105449b5e25fSSatish Balay } 105549b5e25fSSatish Balay EXTERN_C_END 105649b5e25fSSatish Balay 10574a2ae208SSatish Balay #undef __FUNCT__ 10584a2ae208SSatish Balay #define __FUNCT__ "MatSeqSBAIJSetColumnIndices" 105949b5e25fSSatish Balay /*@ 106019585528SSatish Balay MatSeqSBAIJSetColumnIndices - Set the column indices for all the rows 106149b5e25fSSatish Balay in the matrix. 106249b5e25fSSatish Balay 106349b5e25fSSatish Balay Input Parameters: 106419585528SSatish Balay + mat - the SeqSBAIJ matrix 106549b5e25fSSatish Balay - indices - the column indices 106649b5e25fSSatish Balay 106749b5e25fSSatish Balay Level: advanced 106849b5e25fSSatish Balay 106949b5e25fSSatish Balay Notes: 107049b5e25fSSatish Balay This can be called if you have precomputed the nonzero structure of the 107149b5e25fSSatish Balay matrix and want to provide it to the matrix object to improve the performance 107249b5e25fSSatish Balay of the MatSetValues() operation. 107349b5e25fSSatish Balay 107449b5e25fSSatish Balay You MUST have set the correct numbers of nonzeros per row in the call to 1075d1be2dadSMatthew Knepley MatCreateSeqSBAIJ(), and the columns indices MUST be sorted. 107649b5e25fSSatish Balay 1077ab9f2c04SSatish Balay MUST be called before any calls to MatSetValues() 107849b5e25fSSatish Balay 1079ab9f2c04SSatish Balay .seealso: MatCreateSeqSBAIJ 108049b5e25fSSatish Balay @*/ 1081be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatSeqSBAIJSetColumnIndices(Mat mat,PetscInt *indices) 108249b5e25fSSatish Balay { 108313f74950SBarry Smith PetscErrorCode ierr,(*f)(Mat,PetscInt *); 108449b5e25fSSatish Balay 108549b5e25fSSatish Balay PetscFunctionBegin; 10864482741eSBarry Smith PetscValidHeaderSpecific(mat,MAT_COOKIE,1); 10874482741eSBarry Smith PetscValidPointer(indices,2); 1088c134de8dSSatish Balay ierr = PetscObjectQueryFunction((PetscObject)mat,"MatSeqSBAIJSetColumnIndices_C",(void (**)(void))&f);CHKERRQ(ierr); 108949b5e25fSSatish Balay if (f) { 109049b5e25fSSatish Balay ierr = (*f)(mat,indices);CHKERRQ(ierr); 109149b5e25fSSatish Balay } else { 1092e005ede5SBarry Smith SETERRQ(PETSC_ERR_SUP,"Wrong type of matrix to set column indices"); 109349b5e25fSSatish Balay } 109449b5e25fSSatish Balay PetscFunctionReturn(0); 109549b5e25fSSatish Balay } 109649b5e25fSSatish Balay 10974a2ae208SSatish Balay #undef __FUNCT__ 10983c896bc6SHong Zhang #define __FUNCT__ "MatCopy_SeqSBAIJ" 10993c896bc6SHong Zhang PetscErrorCode MatCopy_SeqSBAIJ(Mat A,Mat B,MatStructure str) 11003c896bc6SHong Zhang { 11013c896bc6SHong Zhang PetscErrorCode ierr; 11023c896bc6SHong Zhang 11033c896bc6SHong Zhang PetscFunctionBegin; 11043c896bc6SHong Zhang /* If the two matrices have the same copy implementation, use fast copy. */ 11053c896bc6SHong Zhang if (str == SAME_NONZERO_PATTERN && (A->ops->copy == B->ops->copy)) { 11063c896bc6SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 11073c896bc6SHong Zhang Mat_SeqSBAIJ *b = (Mat_SeqSBAIJ*)B->data; 11083c896bc6SHong Zhang 1109899cda47SBarry Smith if (a->i[A->rmap.N] != b->i[B->rmap.N]) { 11103c896bc6SHong Zhang SETERRQ(PETSC_ERR_ARG_INCOMP,"Number of nonzeros in two matrices are different"); 11113c896bc6SHong Zhang } 1112899cda47SBarry Smith ierr = PetscMemcpy(b->a,a->a,(a->i[A->rmap.N])*sizeof(PetscScalar));CHKERRQ(ierr); 11133c896bc6SHong Zhang } else { 1114f5edf698SHong Zhang ierr = MatGetRowUpperTriangular(A);CHKERRQ(ierr); 11153c896bc6SHong Zhang ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr); 1116f5edf698SHong Zhang ierr = MatRestoreRowUpperTriangular(A);CHKERRQ(ierr); 11173c896bc6SHong Zhang } 11183c896bc6SHong Zhang PetscFunctionReturn(0); 11193c896bc6SHong Zhang } 11203c896bc6SHong Zhang 11213c896bc6SHong Zhang #undef __FUNCT__ 11224a2ae208SSatish Balay #define __FUNCT__ "MatSetUpPreallocation_SeqSBAIJ" 1123dfbe8321SBarry Smith PetscErrorCode MatSetUpPreallocation_SeqSBAIJ(Mat A) 1124273d9f13SBarry Smith { 1125dfbe8321SBarry Smith PetscErrorCode ierr; 1126273d9f13SBarry Smith 1127273d9f13SBarry Smith PetscFunctionBegin; 11287edd0491SSatish Balay ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(A,PetscMax(A->rmap.bs,1),PETSC_DEFAULT,0);CHKERRQ(ierr); 1129273d9f13SBarry Smith PetscFunctionReturn(0); 1130273d9f13SBarry Smith } 1131273d9f13SBarry Smith 1132a6ece127SHong Zhang #undef __FUNCT__ 1133a6ece127SHong Zhang #define __FUNCT__ "MatGetArray_SeqSBAIJ" 1134dfbe8321SBarry Smith PetscErrorCode MatGetArray_SeqSBAIJ(Mat A,PetscScalar *array[]) 1135a6ece127SHong Zhang { 1136a6ece127SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 1137a6ece127SHong Zhang PetscFunctionBegin; 1138a6ece127SHong Zhang *array = a->a; 1139a6ece127SHong Zhang PetscFunctionReturn(0); 1140a6ece127SHong Zhang } 1141a6ece127SHong Zhang 1142a6ece127SHong Zhang #undef __FUNCT__ 1143a6ece127SHong Zhang #define __FUNCT__ "MatRestoreArray_SeqSBAIJ" 1144dfbe8321SBarry Smith PetscErrorCode MatRestoreArray_SeqSBAIJ(Mat A,PetscScalar *array[]) 1145a6ece127SHong Zhang { 1146a6ece127SHong Zhang PetscFunctionBegin; 1147a6ece127SHong Zhang PetscFunctionReturn(0); 1148a6ece127SHong Zhang } 1149a6ece127SHong Zhang 115042ee4b1aSHong Zhang #include "petscblaslapack.h" 115142ee4b1aSHong Zhang #undef __FUNCT__ 115242ee4b1aSHong Zhang #define __FUNCT__ "MatAXPY_SeqSBAIJ" 1153f4df32b1SMatthew Knepley PetscErrorCode MatAXPY_SeqSBAIJ(Mat Y,PetscScalar a,Mat X,MatStructure str) 115442ee4b1aSHong Zhang { 115542ee4b1aSHong Zhang Mat_SeqSBAIJ *x=(Mat_SeqSBAIJ *)X->data, *y=(Mat_SeqSBAIJ *)Y->data; 1156dfbe8321SBarry Smith PetscErrorCode ierr; 1157899cda47SBarry Smith PetscInt i,bs=Y->rmap.bs,bs2,j; 11584ce68768SBarry Smith PetscBLASInt bnz = (PetscBLASInt)x->nz,one = 1; 115942ee4b1aSHong Zhang 116042ee4b1aSHong Zhang PetscFunctionBegin; 116142ee4b1aSHong Zhang if (str == SAME_NONZERO_PATTERN) { 1162f4df32b1SMatthew Knepley PetscScalar alpha = a; 1163f4df32b1SMatthew Knepley BLASaxpy_(&bnz,&alpha,x->a,&one,y->a,&one); 1164c537a176SHong Zhang } else if (str == SUBSET_NONZERO_PATTERN) { /* nonzeros of X is a subset of Y's */ 1165c4319e64SHong Zhang if (y->xtoy && y->XtoY != X) { 1166c4319e64SHong Zhang ierr = PetscFree(y->xtoy);CHKERRQ(ierr); 1167c4319e64SHong Zhang ierr = MatDestroy(y->XtoY);CHKERRQ(ierr); 1168c537a176SHong Zhang } 1169c4319e64SHong Zhang if (!y->xtoy) { /* get xtoy */ 1170c4319e64SHong Zhang ierr = MatAXPYGetxtoy_Private(x->mbs,x->i,x->j,PETSC_NULL, y->i,y->j,PETSC_NULL, &y->xtoy);CHKERRQ(ierr); 1171c4319e64SHong Zhang y->XtoY = X; 1172c537a176SHong Zhang } 1173c4319e64SHong Zhang bs2 = bs*bs; 11746c6c5352SBarry Smith for (i=0; i<x->nz; i++) { 1175c4319e64SHong Zhang j = 0; 1176c4319e64SHong Zhang while (j < bs2){ 1177f4df32b1SMatthew Knepley y->a[bs2*y->xtoy[i]+j] += a*(x->a[bs2*i+j]); 1178c4319e64SHong Zhang j++; 1179c537a176SHong Zhang } 1180c4319e64SHong Zhang } 1181ae15b995SBarry Smith ierr = PetscInfo3(0,"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); 118242ee4b1aSHong Zhang } else { 1183f5edf698SHong Zhang ierr = MatGetRowUpperTriangular(X);CHKERRQ(ierr); 1184f4df32b1SMatthew Knepley ierr = MatAXPY_Basic(Y,a,X,str);CHKERRQ(ierr); 1185f5edf698SHong Zhang ierr = MatRestoreRowUpperTriangular(X);CHKERRQ(ierr); 118642ee4b1aSHong Zhang } 118742ee4b1aSHong Zhang PetscFunctionReturn(0); 118842ee4b1aSHong Zhang } 118942ee4b1aSHong Zhang 1190efcf0fc3SBarry Smith #undef __FUNCT__ 1191efcf0fc3SBarry Smith #define __FUNCT__ "MatIsSymmetric_SeqSBAIJ" 1192dfbe8321SBarry Smith PetscErrorCode MatIsSymmetric_SeqSBAIJ(Mat A,PetscReal tol,PetscTruth *flg) 1193efcf0fc3SBarry Smith { 1194efcf0fc3SBarry Smith PetscFunctionBegin; 1195efcf0fc3SBarry Smith *flg = PETSC_TRUE; 1196efcf0fc3SBarry Smith PetscFunctionReturn(0); 1197efcf0fc3SBarry Smith } 1198efcf0fc3SBarry Smith 1199efcf0fc3SBarry Smith #undef __FUNCT__ 1200efcf0fc3SBarry Smith #define __FUNCT__ "MatIsStructurallySymmetric_SeqSBAIJ" 1201dfbe8321SBarry Smith PetscErrorCode MatIsStructurallySymmetric_SeqSBAIJ(Mat A,PetscTruth *flg) 1202efcf0fc3SBarry Smith { 1203efcf0fc3SBarry Smith PetscFunctionBegin; 1204efcf0fc3SBarry Smith *flg = PETSC_TRUE; 1205efcf0fc3SBarry Smith PetscFunctionReturn(0); 1206efcf0fc3SBarry Smith } 1207efcf0fc3SBarry Smith 1208efcf0fc3SBarry Smith #undef __FUNCT__ 1209efcf0fc3SBarry Smith #define __FUNCT__ "MatIsHermitian_SeqSBAIJ" 1210dfbe8321SBarry Smith PetscErrorCode MatIsHermitian_SeqSBAIJ(Mat A,PetscTruth *flg) 1211efcf0fc3SBarry Smith { 1212efcf0fc3SBarry Smith PetscFunctionBegin; 1213efcf0fc3SBarry Smith *flg = PETSC_FALSE; 1214efcf0fc3SBarry Smith PetscFunctionReturn(0); 1215efcf0fc3SBarry Smith } 1216efcf0fc3SBarry Smith 121799cafbc1SBarry Smith #undef __FUNCT__ 121899cafbc1SBarry Smith #define __FUNCT__ "MatRealPart_SeqSBAIJ" 121999cafbc1SBarry Smith PetscErrorCode MatRealPart_SeqSBAIJ(Mat A) 122099cafbc1SBarry Smith { 122199cafbc1SBarry Smith Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 122299cafbc1SBarry Smith PetscInt i,nz = a->bs2*a->i[a->mbs]; 122399cafbc1SBarry Smith PetscScalar *aa = a->a; 122499cafbc1SBarry Smith 122599cafbc1SBarry Smith PetscFunctionBegin; 122699cafbc1SBarry Smith for (i=0; i<nz; i++) aa[i] = PetscRealPart(aa[i]); 122799cafbc1SBarry Smith PetscFunctionReturn(0); 122899cafbc1SBarry Smith } 122999cafbc1SBarry Smith 123099cafbc1SBarry Smith #undef __FUNCT__ 123199cafbc1SBarry Smith #define __FUNCT__ "MatImaginaryPart_SeqSBAIJ" 123299cafbc1SBarry Smith PetscErrorCode MatImaginaryPart_SeqSBAIJ(Mat A) 123399cafbc1SBarry Smith { 123499cafbc1SBarry Smith Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 123599cafbc1SBarry Smith PetscInt i,nz = a->bs2*a->i[a->mbs]; 123699cafbc1SBarry Smith PetscScalar *aa = a->a; 123799cafbc1SBarry Smith 123899cafbc1SBarry Smith PetscFunctionBegin; 123999cafbc1SBarry Smith for (i=0; i<nz; i++) aa[i] = PetscImaginaryPart(aa[i]); 124099cafbc1SBarry Smith PetscFunctionReturn(0); 124199cafbc1SBarry Smith } 124299cafbc1SBarry Smith 124349b5e25fSSatish Balay /* -------------------------------------------------------------------*/ 124449b5e25fSSatish Balay static struct _MatOps MatOps_Values = {MatSetValues_SeqSBAIJ, 124549b5e25fSSatish Balay MatGetRow_SeqSBAIJ, 124649b5e25fSSatish Balay MatRestoreRow_SeqSBAIJ, 124749b5e25fSSatish Balay MatMult_SeqSBAIJ_N, 124897304618SKris Buschelman /* 4*/ MatMultAdd_SeqSBAIJ_N, 1249431c96f7SBarry Smith MatMult_SeqSBAIJ_N, /* transpose versions are same as non-transpose versions */ 1250e005ede5SBarry Smith MatMultAdd_SeqSBAIJ_N, 125149b5e25fSSatish Balay MatSolve_SeqSBAIJ_N, 125249b5e25fSSatish Balay 0, 125349b5e25fSSatish Balay 0, 125497304618SKris Buschelman /*10*/ 0, 125549b5e25fSSatish Balay 0, 12565f4ef2deSHong Zhang MatCholeskyFactor_SeqSBAIJ, 1257d06b337dSHong Zhang MatRelax_SeqSBAIJ, 125849b5e25fSSatish Balay MatTranspose_SeqSBAIJ, 125997304618SKris Buschelman /*15*/ MatGetInfo_SeqSBAIJ, 126049b5e25fSSatish Balay MatEqual_SeqSBAIJ, 126149b5e25fSSatish Balay MatGetDiagonal_SeqSBAIJ, 126249b5e25fSSatish Balay MatDiagonalScale_SeqSBAIJ, 126349b5e25fSSatish Balay MatNorm_SeqSBAIJ, 126497304618SKris Buschelman /*20*/ 0, 126549b5e25fSSatish Balay MatAssemblyEnd_SeqSBAIJ, 126649b5e25fSSatish Balay 0, 126749b5e25fSSatish Balay MatSetOption_SeqSBAIJ, 126849b5e25fSSatish Balay MatZeroEntries_SeqSBAIJ, 1269dcf5cc72SBarry Smith /*25*/ 0, 127049b5e25fSSatish Balay 0, 127149b5e25fSSatish Balay 0, 12725f4ef2deSHong Zhang MatCholeskyFactorSymbolic_SeqSBAIJ, 12735f4ef2deSHong Zhang MatCholeskyFactorNumeric_SeqSBAIJ_N, 127497304618SKris Buschelman /*30*/ MatSetUpPreallocation_SeqSBAIJ, 1275c464158bSHong Zhang 0, 12764d101231SSatish Balay MatICCFactorSymbolic_SeqSBAIJ, 1277a6ece127SHong Zhang MatGetArray_SeqSBAIJ, 1278a6ece127SHong Zhang MatRestoreArray_SeqSBAIJ, 127997304618SKris Buschelman /*35*/ MatDuplicate_SeqSBAIJ, 128049b5e25fSSatish Balay 0, 128149b5e25fSSatish Balay 0, 128249b5e25fSSatish Balay 0, 1283c84f5b01SHong Zhang MatICCFactor_SeqSBAIJ, 128497304618SKris Buschelman /*40*/ MatAXPY_SeqSBAIJ, 128549b5e25fSSatish Balay MatGetSubMatrices_SeqSBAIJ, 128649b5e25fSSatish Balay MatIncreaseOverlap_SeqSBAIJ, 128749b5e25fSSatish Balay MatGetValues_SeqSBAIJ, 12883c896bc6SHong Zhang MatCopy_SeqSBAIJ, 12898c07d4e3SBarry Smith /*45*/ 0, 129049b5e25fSSatish Balay MatScale_SeqSBAIJ, 129149b5e25fSSatish Balay 0, 129249b5e25fSSatish Balay 0, 129349b5e25fSSatish Balay 0, 1294521d7252SBarry Smith /*50*/ 0, 129549b5e25fSSatish Balay MatGetRowIJ_SeqSBAIJ, 129649b5e25fSSatish Balay MatRestoreRowIJ_SeqSBAIJ, 129749b5e25fSSatish Balay 0, 129849b5e25fSSatish Balay 0, 129997304618SKris Buschelman /*55*/ 0, 130049b5e25fSSatish Balay 0, 130149b5e25fSSatish Balay 0, 130249b5e25fSSatish Balay 0, 130349b5e25fSSatish Balay MatSetValuesBlocked_SeqSBAIJ, 130497304618SKris Buschelman /*60*/ MatGetSubMatrix_SeqSBAIJ, 130549b5e25fSSatish Balay 0, 130649b5e25fSSatish Balay 0, 1307357abbc8SBarry Smith 0, 1308d959ec07SHong Zhang 0, 130997304618SKris Buschelman /*65*/ 0, 1310d959ec07SHong Zhang 0, 1311d959ec07SHong Zhang 0, 1312d959ec07SHong Zhang 0, 1313d959ec07SHong Zhang 0, 131497304618SKris Buschelman /*70*/ MatGetRowMax_SeqSBAIJ, 13153e0d88b5SBarry Smith 0, 13163e0d88b5SBarry Smith 0, 13173e0d88b5SBarry Smith 0, 13183e0d88b5SBarry Smith 0, 131997304618SKris Buschelman /*75*/ 0, 13203e0d88b5SBarry Smith 0, 13213e0d88b5SBarry Smith 0, 13223e0d88b5SBarry Smith 0, 13233e0d88b5SBarry Smith 0, 132497304618SKris Buschelman /*80*/ 0, 13253e0d88b5SBarry Smith 0, 13263e0d88b5SBarry Smith 0, 13273e0d88b5SBarry Smith #if !defined(PETSC_USE_COMPLEX) 132897304618SKris Buschelman MatGetInertia_SeqSBAIJ, 13293e0d88b5SBarry Smith #else 133097304618SKris Buschelman 0, 13313e0d88b5SBarry Smith #endif 1332865e5f61SKris Buschelman MatLoad_SeqSBAIJ, 1333865e5f61SKris Buschelman /*85*/ MatIsSymmetric_SeqSBAIJ, 1334865e5f61SKris Buschelman MatIsHermitian_SeqSBAIJ, 1335efcf0fc3SBarry Smith MatIsStructurallySymmetric_SeqSBAIJ, 1336865e5f61SKris Buschelman 0, 1337865e5f61SKris Buschelman 0, 1338865e5f61SKris Buschelman /*90*/ 0, 1339865e5f61SKris Buschelman 0, 1340865e5f61SKris Buschelman 0, 1341865e5f61SKris Buschelman 0, 1342865e5f61SKris Buschelman 0, 1343865e5f61SKris Buschelman /*95*/ 0, 1344865e5f61SKris Buschelman 0, 1345865e5f61SKris Buschelman 0, 134699cafbc1SBarry Smith 0, 134799cafbc1SBarry Smith 0, 134899cafbc1SBarry Smith /*100*/0, 134999cafbc1SBarry Smith 0, 135099cafbc1SBarry Smith 0, 135199cafbc1SBarry Smith 0, 135299cafbc1SBarry Smith 0, 135399cafbc1SBarry Smith /*105*/0, 135499cafbc1SBarry Smith MatRealPart_SeqSBAIJ, 1355f5edf698SHong Zhang MatImaginaryPart_SeqSBAIJ, 1356f5edf698SHong Zhang MatGetRowUpperTriangular_SeqSBAIJ, 1357f5edf698SHong Zhang MatRestoreRowUpperTriangular_SeqSBAIJ 135899cafbc1SBarry Smith }; 1359be1d678aSKris Buschelman 136049b5e25fSSatish Balay EXTERN_C_BEGIN 13614a2ae208SSatish Balay #undef __FUNCT__ 13624a2ae208SSatish Balay #define __FUNCT__ "MatStoreValues_SeqSBAIJ" 1363be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatStoreValues_SeqSBAIJ(Mat mat) 136449b5e25fSSatish Balay { 13654afc71dfSHong Zhang Mat_SeqSBAIJ *aij = (Mat_SeqSBAIJ *)mat->data; 1366899cda47SBarry Smith PetscInt nz = aij->i[mat->rmap.N]*mat->rmap.bs*aij->bs2; 1367dfbe8321SBarry Smith PetscErrorCode ierr; 136849b5e25fSSatish Balay 136949b5e25fSSatish Balay PetscFunctionBegin; 137049b5e25fSSatish Balay if (aij->nonew != 1) { 1371e005ede5SBarry Smith SETERRQ(PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NO_NEW_NONZERO_LOCATIONS);first"); 137249b5e25fSSatish Balay } 137349b5e25fSSatish Balay 137449b5e25fSSatish Balay /* allocate space for values if not already there */ 137549b5e25fSSatish Balay if (!aij->saved_values) { 137687828ca2SBarry Smith ierr = PetscMalloc((nz+1)*sizeof(PetscScalar),&aij->saved_values);CHKERRQ(ierr); 137749b5e25fSSatish Balay } 137849b5e25fSSatish Balay 137949b5e25fSSatish Balay /* copy values over */ 138087828ca2SBarry Smith ierr = PetscMemcpy(aij->saved_values,aij->a,nz*sizeof(PetscScalar));CHKERRQ(ierr); 138149b5e25fSSatish Balay PetscFunctionReturn(0); 138249b5e25fSSatish Balay } 138349b5e25fSSatish Balay EXTERN_C_END 138449b5e25fSSatish Balay 138549b5e25fSSatish Balay EXTERN_C_BEGIN 13864a2ae208SSatish Balay #undef __FUNCT__ 13874a2ae208SSatish Balay #define __FUNCT__ "MatRetrieveValues_SeqSBAIJ" 1388be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatRetrieveValues_SeqSBAIJ(Mat mat) 138949b5e25fSSatish Balay { 13904afc71dfSHong Zhang Mat_SeqSBAIJ *aij = (Mat_SeqSBAIJ *)mat->data; 13916849ba73SBarry Smith PetscErrorCode ierr; 1392899cda47SBarry Smith PetscInt nz = aij->i[mat->rmap.N]*mat->rmap.bs*aij->bs2; 139349b5e25fSSatish Balay 139449b5e25fSSatish Balay PetscFunctionBegin; 139549b5e25fSSatish Balay if (aij->nonew != 1) { 1396e005ede5SBarry Smith SETERRQ(PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NO_NEW_NONZERO_LOCATIONS);first"); 139749b5e25fSSatish Balay } 139849b5e25fSSatish Balay if (!aij->saved_values) { 1399e005ede5SBarry Smith SETERRQ(PETSC_ERR_ORDER,"Must call MatStoreValues(A);first"); 140049b5e25fSSatish Balay } 140149b5e25fSSatish Balay 140249b5e25fSSatish Balay /* copy values over */ 140387828ca2SBarry Smith ierr = PetscMemcpy(aij->a,aij->saved_values,nz*sizeof(PetscScalar));CHKERRQ(ierr); 140449b5e25fSSatish Balay PetscFunctionReturn(0); 140549b5e25fSSatish Balay } 140649b5e25fSSatish Balay EXTERN_C_END 140749b5e25fSSatish Balay 14088549e402SHong Zhang EXTERN_C_BEGIN 14094a2ae208SSatish Balay #undef __FUNCT__ 1410a23d5eceSKris Buschelman #define __FUNCT__ "MatSeqSBAIJSetPreallocation_SeqSBAIJ" 1411be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatSeqSBAIJSetPreallocation_SeqSBAIJ(Mat B,PetscInt bs,PetscInt nz,PetscInt *nnz) 141249b5e25fSSatish Balay { 1413c464158bSHong Zhang Mat_SeqSBAIJ *b = (Mat_SeqSBAIJ*)B->data; 14146849ba73SBarry Smith PetscErrorCode ierr; 1415ab93d7beSBarry Smith PetscInt i,mbs,bs2; 1416ab93d7beSBarry Smith PetscTruth skipallocation = PETSC_FALSE,flg; 141749b5e25fSSatish Balay 141849b5e25fSSatish Balay PetscFunctionBegin; 1419273d9f13SBarry Smith B->preallocated = PETSC_TRUE; 1420e82a3eeeSBarry Smith ierr = PetscOptionsGetInt(B->prefix,"-mat_block_size",&bs,PETSC_NULL);CHKERRQ(ierr); 1421899cda47SBarry Smith B->rmap.bs = B->cmap.bs = bs; 1422899cda47SBarry Smith ierr = PetscMapInitialize(B->comm,&B->rmap);CHKERRQ(ierr); 1423899cda47SBarry Smith ierr = PetscMapInitialize(B->comm,&B->cmap);CHKERRQ(ierr); 1424899cda47SBarry Smith 1425899cda47SBarry Smith mbs = B->rmap.N/bs; 142649b5e25fSSatish Balay bs2 = bs*bs; 142749b5e25fSSatish Balay 1428899cda47SBarry Smith if (mbs*bs != B->rmap.N) { 142929bbc08cSBarry Smith SETERRQ(PETSC_ERR_ARG_SIZ,"Number rows, cols must be divisible by blocksize"); 143049b5e25fSSatish Balay } 143149b5e25fSSatish Balay 1432ab93d7beSBarry Smith if (nz == MAT_SKIP_ALLOCATION) { 1433ab93d7beSBarry Smith skipallocation = PETSC_TRUE; 1434ab93d7beSBarry Smith nz = 0; 1435ab93d7beSBarry Smith } 1436ab93d7beSBarry Smith 1437435da068SBarry Smith if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 3; 143877431f27SBarry Smith if (nz < 0) SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"nz cannot be less than 0: value %D",nz); 143949b5e25fSSatish Balay if (nnz) { 144049b5e25fSSatish Balay for (i=0; i<mbs; i++) { 144177431f27SBarry Smith if (nnz[i] < 0) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"nnz cannot be less than 0: local row %D value %D",i,nnz[i]); 144277431f27SBarry Smith if (nnz[i] > mbs) SETERRQ3(PETSC_ERR_ARG_OUTOFRANGE,"nnz cannot be greater than block row length: local row %D value %D rowlength %D",i,nnz[i],mbs); 144349b5e25fSSatish Balay } 144449b5e25fSSatish Balay } 144549b5e25fSSatish Balay 1446e82a3eeeSBarry Smith ierr = PetscOptionsHasName(B->prefix,"-mat_no_unroll",&flg);CHKERRQ(ierr); 144749b5e25fSSatish Balay if (!flg) { 144849b5e25fSSatish Balay switch (bs) { 144949b5e25fSSatish Balay case 1: 14504ccecd49SHong Zhang B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1; 145149b5e25fSSatish Balay B->ops->solve = MatSolve_SeqSBAIJ_1; 1452d59c15a7SBarry Smith B->ops->solves = MatSolves_SeqSBAIJ_1; 1453e005ede5SBarry Smith B->ops->solvetranspose = MatSolve_SeqSBAIJ_1; 145449b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_1; 145549b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_1; 1456431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_1; 1457431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_1; 145849b5e25fSSatish Balay break; 145949b5e25fSSatish Balay case 2: 14604ccecd49SHong Zhang B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2; 146149b5e25fSSatish Balay B->ops->solve = MatSolve_SeqSBAIJ_2; 1462e005ede5SBarry Smith B->ops->solvetranspose = MatSolve_SeqSBAIJ_2; 146349b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_2; 146449b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_2; 1465431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_2; 1466431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_2; 146749b5e25fSSatish Balay break; 146849b5e25fSSatish Balay case 3: 14695f4ef2deSHong Zhang B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3; 147049b5e25fSSatish Balay B->ops->solve = MatSolve_SeqSBAIJ_3; 1471e005ede5SBarry Smith B->ops->solvetranspose = MatSolve_SeqSBAIJ_3; 147249b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_3; 147349b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_3; 1474431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_3; 1475431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_3; 147649b5e25fSSatish Balay break; 147749b5e25fSSatish Balay case 4: 14785f4ef2deSHong Zhang B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4; 147949b5e25fSSatish Balay B->ops->solve = MatSolve_SeqSBAIJ_4; 1480e005ede5SBarry Smith B->ops->solvetranspose = MatSolve_SeqSBAIJ_4; 148149b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_4; 148249b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_4; 1483431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_4; 1484431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_4; 148549b5e25fSSatish Balay break; 148649b5e25fSSatish Balay case 5: 14875f4ef2deSHong Zhang B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5; 148849b5e25fSSatish Balay B->ops->solve = MatSolve_SeqSBAIJ_5; 1489e005ede5SBarry Smith B->ops->solvetranspose = MatSolve_SeqSBAIJ_5; 149049b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_5; 149149b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_5; 1492431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_5; 1493431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_5; 149449b5e25fSSatish Balay break; 149549b5e25fSSatish Balay case 6: 14965f4ef2deSHong Zhang B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6; 149749b5e25fSSatish Balay B->ops->solve = MatSolve_SeqSBAIJ_6; 1498e005ede5SBarry Smith B->ops->solvetranspose = MatSolve_SeqSBAIJ_6; 149949b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_6; 150049b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_6; 1501431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_6; 1502431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_6; 150349b5e25fSSatish Balay break; 150449b5e25fSSatish Balay case 7: 1505de53e5efSHong Zhang B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7; 150649b5e25fSSatish Balay B->ops->solve = MatSolve_SeqSBAIJ_7; 1507e005ede5SBarry Smith B->ops->solvetranspose = MatSolve_SeqSBAIJ_7; 1508de53e5efSHong Zhang B->ops->mult = MatMult_SeqSBAIJ_7; 150949b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_7; 1510431c96f7SBarry Smith B->ops->multtranspose = MatMult_SeqSBAIJ_7; 1511431c96f7SBarry Smith B->ops->multtransposeadd = MatMultAdd_SeqSBAIJ_7; 151249b5e25fSSatish Balay break; 151349b5e25fSSatish Balay } 151449b5e25fSSatish Balay } 151549b5e25fSSatish Balay 151649b5e25fSSatish Balay b->mbs = mbs; 15174afc71dfSHong Zhang b->nbs = mbs; 1518ab93d7beSBarry Smith if (!skipallocation) { 1519ab93d7beSBarry Smith /* b->ilen will count nonzeros in each block row so far. */ 1520ab93d7beSBarry Smith ierr = PetscMalloc2(mbs,PetscInt,&b->imax,mbs,PetscInt,&b->ilen);CHKERRQ(ierr); 1521ab93d7beSBarry Smith for (i=0; i<mbs; i++) { b->ilen[i] = 0;} 152249b5e25fSSatish Balay if (!nnz) { 1523435da068SBarry Smith if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 5; 152449b5e25fSSatish Balay else if (nz <= 0) nz = 1; 152549b5e25fSSatish Balay for (i=0; i<mbs; i++) { 15268cef66ccSHong Zhang b->imax[i] = nz; 152749b5e25fSSatish Balay } 1528153ea458SHong Zhang nz = nz*mbs; /* total nz */ 152949b5e25fSSatish Balay } else { 153049b5e25fSSatish Balay nz = 0; 15318cef66ccSHong Zhang for (i=0; i<mbs; i++) {b->imax[i] = nnz[i]; nz += nnz[i];} 153249b5e25fSSatish Balay } 15336c6c5352SBarry Smith /* nz=(nz+mbs)/2; */ /* total diagonal and superdiagonal nonzero blocks */ 153449b5e25fSSatish Balay 153549b5e25fSSatish Balay /* allocate the matrix space */ 1536899cda47SBarry Smith ierr = PetscMalloc3(bs2*nz,PetscScalar,&b->a,nz,PetscInt,&b->j,B->rmap.N+1,PetscInt,&b->i);CHKERRQ(ierr); 15376c6c5352SBarry Smith ierr = PetscMemzero(b->a,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr); 153813f74950SBarry Smith ierr = PetscMemzero(b->j,nz*sizeof(PetscInt));CHKERRQ(ierr); 153949b5e25fSSatish Balay b->singlemalloc = PETSC_TRUE; 154049b5e25fSSatish Balay 154149b5e25fSSatish Balay /* pointer to beginning of each row */ 154249b5e25fSSatish Balay b->i[0] = 0; 154349b5e25fSSatish Balay for (i=1; i<mbs+1; i++) { 154449b5e25fSSatish Balay b->i[i] = b->i[i-1] + b->imax[i-1]; 154549b5e25fSSatish Balay } 1546e6b907acSBarry Smith b->free_a = PETSC_TRUE; 1547e6b907acSBarry Smith b->free_ij = PETSC_TRUE; 1548e811da20SHong Zhang } else { 1549e6b907acSBarry Smith b->free_a = PETSC_FALSE; 1550e6b907acSBarry Smith b->free_ij = PETSC_FALSE; 1551ab93d7beSBarry Smith } 155249b5e25fSSatish Balay 1553899cda47SBarry Smith B->rmap.bs = bs; 155449b5e25fSSatish Balay b->bs2 = bs2; 15556c6c5352SBarry Smith b->nz = 0; 15566c6c5352SBarry Smith b->maxnz = nz*bs2; 1557153ea458SHong Zhang 155816cdd363SHong Zhang b->inew = 0; 155916cdd363SHong Zhang b->jnew = 0; 156016cdd363SHong Zhang b->anew = 0; 156116cdd363SHong Zhang b->a2anew = 0; 15621a3463dfSHong Zhang b->permute = PETSC_FALSE; 1563c464158bSHong Zhang PetscFunctionReturn(0); 1564c464158bSHong Zhang } 1565a23d5eceSKris Buschelman EXTERN_C_END 1566153ea458SHong Zhang 1567d769727bSBarry Smith EXTERN_C_BEGIN 1568f69a0ea3SMatthew Knepley EXTERN PetscErrorCode PETSCMAT_DLLEXPORT MatConvert_SeqSBAIJ_SeqAIJ(Mat, MatType,MatReuse,Mat*); 1569f69a0ea3SMatthew Knepley EXTERN PetscErrorCode PETSCMAT_DLLEXPORT MatConvert_SeqSBAIJ_SeqBAIJ(Mat, MatType,MatReuse,Mat*); 1570d769727bSBarry Smith EXTERN_C_END 1571d769727bSBarry Smith 15720bad9183SKris Buschelman /*MC 1573fafad747SKris Buschelman MATSEQSBAIJ - MATSEQSBAIJ = "seqsbaij" - A matrix type to be used for sequential symmetric block sparse matrices, 15740bad9183SKris Buschelman based on block compressed sparse row format. Only the upper triangular portion of the matrix is stored. 15750bad9183SKris Buschelman 15760bad9183SKris Buschelman Options Database Keys: 15770bad9183SKris Buschelman . -mat_type seqsbaij - sets the matrix type to "seqsbaij" during a call to MatSetFromOptions() 15780bad9183SKris Buschelman 15790bad9183SKris Buschelman Level: beginner 15800bad9183SKris Buschelman 15810bad9183SKris Buschelman .seealso: MatCreateSeqSBAIJ 15820bad9183SKris Buschelman M*/ 15830bad9183SKris Buschelman 1584a23d5eceSKris Buschelman EXTERN_C_BEGIN 1585a23d5eceSKris Buschelman #undef __FUNCT__ 1586a23d5eceSKris Buschelman #define __FUNCT__ "MatCreate_SeqSBAIJ" 1587be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreate_SeqSBAIJ(Mat B) 1588a23d5eceSKris Buschelman { 1589a23d5eceSKris Buschelman Mat_SeqSBAIJ *b; 1590dfbe8321SBarry Smith PetscErrorCode ierr; 159113f74950SBarry Smith PetscMPIInt size; 1592941593c8SHong Zhang PetscTruth flg; 1593a23d5eceSKris Buschelman 1594a23d5eceSKris Buschelman PetscFunctionBegin; 1595a23d5eceSKris Buschelman ierr = MPI_Comm_size(B->comm,&size);CHKERRQ(ierr); 1596a23d5eceSKris Buschelman if (size > 1) SETERRQ(PETSC_ERR_ARG_WRONG,"Comm must be of size 1"); 1597a23d5eceSKris Buschelman 1598a23d5eceSKris Buschelman ierr = PetscNew(Mat_SeqSBAIJ,&b);CHKERRQ(ierr); 1599a23d5eceSKris Buschelman B->data = (void*)b; 1600a23d5eceSKris Buschelman ierr = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr); 1601a23d5eceSKris Buschelman B->ops->destroy = MatDestroy_SeqSBAIJ; 1602a23d5eceSKris Buschelman B->ops->view = MatView_SeqSBAIJ; 1603a23d5eceSKris Buschelman B->factor = 0; 1604a23d5eceSKris Buschelman B->mapping = 0; 1605a23d5eceSKris Buschelman b->row = 0; 1606a23d5eceSKris Buschelman b->icol = 0; 1607a23d5eceSKris Buschelman b->reallocs = 0; 1608a23d5eceSKris Buschelman b->saved_values = 0; 1609a23d5eceSKris Buschelman 1610a23d5eceSKris Buschelman 1611a23d5eceSKris Buschelman b->sorted = PETSC_FALSE; 1612a23d5eceSKris Buschelman b->roworiented = PETSC_TRUE; 1613a23d5eceSKris Buschelman b->nonew = 0; 1614a23d5eceSKris Buschelman b->diag = 0; 1615a23d5eceSKris Buschelman b->solve_work = 0; 1616a23d5eceSKris Buschelman b->mult_work = 0; 1617a23d5eceSKris Buschelman B->spptr = 0; 1618a23d5eceSKris Buschelman b->keepzeroedrows = PETSC_FALSE; 1619a23d5eceSKris Buschelman b->xtoy = 0; 1620a23d5eceSKris Buschelman b->XtoY = 0; 1621a23d5eceSKris Buschelman 1622a23d5eceSKris Buschelman b->inew = 0; 1623a23d5eceSKris Buschelman b->jnew = 0; 1624a23d5eceSKris Buschelman b->anew = 0; 1625a23d5eceSKris Buschelman b->a2anew = 0; 1626a23d5eceSKris Buschelman b->permute = PETSC_FALSE; 1627a23d5eceSKris Buschelman 1628941593c8SHong Zhang b->ignore_ltriangular = PETSC_FALSE; 1629941593c8SHong Zhang ierr = PetscOptionsHasName(PETSC_NULL,"-mat_ignore_lower_triangular",&flg);CHKERRQ(ierr); 1630941593c8SHong Zhang if (flg) b->ignore_ltriangular = PETSC_TRUE; 1631941593c8SHong Zhang 1632f5edf698SHong Zhang b->getrow_utriangular = PETSC_FALSE; 1633f5edf698SHong Zhang ierr = PetscOptionsHasName(PETSC_NULL,"-mat_getrow_uppertriangular",&flg);CHKERRQ(ierr); 1634f5edf698SHong Zhang if (flg) b->getrow_utriangular = PETSC_TRUE; 1635f5edf698SHong Zhang 1636a23d5eceSKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatStoreValues_C", 1637a23d5eceSKris Buschelman "MatStoreValues_SeqSBAIJ", 1638a23d5eceSKris Buschelman MatStoreValues_SeqSBAIJ);CHKERRQ(ierr); 1639a23d5eceSKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatRetrieveValues_C", 1640a23d5eceSKris Buschelman "MatRetrieveValues_SeqSBAIJ", 1641a23d5eceSKris Buschelman (void*)MatRetrieveValues_SeqSBAIJ);CHKERRQ(ierr); 1642a23d5eceSKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatSeqSBAIJSetColumnIndices_C", 1643a23d5eceSKris Buschelman "MatSeqSBAIJSetColumnIndices_SeqSBAIJ", 1644a23d5eceSKris Buschelman MatSeqSBAIJSetColumnIndices_SeqSBAIJ);CHKERRQ(ierr); 16454e5e7fe4SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_seqsbaij_seqaij_C", 16464e5e7fe4SHong Zhang "MatConvert_SeqSBAIJ_SeqAIJ", 16474e5e7fe4SHong Zhang MatConvert_SeqSBAIJ_SeqAIJ);CHKERRQ(ierr); 1648a0e1a404SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_seqsbaij_seqbaij_C", 1649a0e1a404SHong Zhang "MatConvert_SeqSBAIJ_SeqBAIJ", 1650a0e1a404SHong Zhang MatConvert_SeqSBAIJ_SeqBAIJ);CHKERRQ(ierr); 1651a23d5eceSKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatSeqSBAIJSetPreallocation_C", 1652a23d5eceSKris Buschelman "MatSeqSBAIJSetPreallocation_SeqSBAIJ", 1653a23d5eceSKris Buschelman MatSeqSBAIJSetPreallocation_SeqSBAIJ);CHKERRQ(ierr); 165423ce1328SBarry Smith 165523ce1328SBarry Smith B->symmetric = PETSC_TRUE; 165623ce1328SBarry Smith B->structurally_symmetric = PETSC_TRUE; 165723ce1328SBarry Smith B->symmetric_set = PETSC_TRUE; 165823ce1328SBarry Smith B->structurally_symmetric_set = PETSC_TRUE; 165917667f90SBarry Smith ierr = PetscObjectChangeTypeName((PetscObject)B,MATSEQSBAIJ);CHKERRQ(ierr); 1660a23d5eceSKris Buschelman PetscFunctionReturn(0); 1661a23d5eceSKris Buschelman } 1662a23d5eceSKris Buschelman EXTERN_C_END 1663a23d5eceSKris Buschelman 1664a23d5eceSKris Buschelman #undef __FUNCT__ 1665a23d5eceSKris Buschelman #define __FUNCT__ "MatSeqSBAIJSetPreallocation" 1666a23d5eceSKris Buschelman /*@C 1667a23d5eceSKris Buschelman MatSeqSBAIJSetPreallocation - Creates a sparse symmetric matrix in block AIJ (block 1668a23d5eceSKris Buschelman compressed row) format. For good matrix assembly performance the 1669a23d5eceSKris Buschelman user should preallocate the matrix storage by setting the parameter nz 1670a23d5eceSKris Buschelman (or the array nnz). By setting these parameters accurately, performance 1671a23d5eceSKris Buschelman during matrix assembly can be increased by more than a factor of 50. 1672a23d5eceSKris Buschelman 1673a23d5eceSKris Buschelman Collective on Mat 1674a23d5eceSKris Buschelman 1675a23d5eceSKris Buschelman Input Parameters: 1676a23d5eceSKris Buschelman + A - the symmetric matrix 1677a23d5eceSKris Buschelman . bs - size of block 1678a23d5eceSKris Buschelman . nz - number of block nonzeros per block row (same for all rows) 1679a23d5eceSKris Buschelman - nnz - array containing the number of block nonzeros in the upper triangular plus 1680a23d5eceSKris Buschelman diagonal portion of each block (possibly different for each block row) or PETSC_NULL 1681a23d5eceSKris Buschelman 1682a23d5eceSKris Buschelman Options Database Keys: 1683a23d5eceSKris Buschelman . -mat_no_unroll - uses code that does not unroll the loops in the 1684a23d5eceSKris Buschelman block calculations (much slower) 1685a23d5eceSKris Buschelman . -mat_block_size - size of the blocks to use 1686a23d5eceSKris Buschelman 1687a23d5eceSKris Buschelman Level: intermediate 1688a23d5eceSKris Buschelman 1689a23d5eceSKris Buschelman Notes: 1690a23d5eceSKris Buschelman Specify the preallocated storage with either nz or nnz (not both). 1691a23d5eceSKris Buschelman Set nz=PETSC_DEFAULT and nnz=PETSC_NULL for PETSc to control dynamic memory 1692a23d5eceSKris Buschelman allocation. For additional details, see the users manual chapter on 1693a23d5eceSKris Buschelman matrices. 1694a23d5eceSKris Buschelman 169549a6f317SBarry Smith If the nnz parameter is given then the nz parameter is ignored 169649a6f317SBarry Smith 169749a6f317SBarry Smith 1698a23d5eceSKris Buschelman .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateMPISBAIJ() 1699a23d5eceSKris Buschelman @*/ 1700be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatSeqSBAIJSetPreallocation(Mat B,PetscInt bs,PetscInt nz,const PetscInt nnz[]) 170113f74950SBarry Smith { 170213f74950SBarry Smith PetscErrorCode ierr,(*f)(Mat,PetscInt,PetscInt,const PetscInt[]); 1703a23d5eceSKris Buschelman 1704a23d5eceSKris Buschelman PetscFunctionBegin; 1705a23d5eceSKris Buschelman ierr = PetscObjectQueryFunction((PetscObject)B,"MatSeqSBAIJSetPreallocation_C",(void (**)(void))&f);CHKERRQ(ierr); 1706a23d5eceSKris Buschelman if (f) { 1707a23d5eceSKris Buschelman ierr = (*f)(B,bs,nz,nnz);CHKERRQ(ierr); 1708a23d5eceSKris Buschelman } 1709a23d5eceSKris Buschelman PetscFunctionReturn(0); 1710a23d5eceSKris Buschelman } 171149b5e25fSSatish Balay 17124a2ae208SSatish Balay #undef __FUNCT__ 17134a2ae208SSatish Balay #define __FUNCT__ "MatCreateSeqSBAIJ" 1714c464158bSHong Zhang /*@C 1715c464158bSHong Zhang MatCreateSeqSBAIJ - Creates a sparse symmetric matrix in block AIJ (block 1716c464158bSHong Zhang compressed row) format. For good matrix assembly performance the 1717c464158bSHong Zhang user should preallocate the matrix storage by setting the parameter nz 1718c464158bSHong Zhang (or the array nnz). By setting these parameters accurately, performance 1719c464158bSHong Zhang during matrix assembly can be increased by more than a factor of 50. 172049b5e25fSSatish Balay 1721c464158bSHong Zhang Collective on MPI_Comm 1722c464158bSHong Zhang 1723c464158bSHong Zhang Input Parameters: 1724c464158bSHong Zhang + comm - MPI communicator, set to PETSC_COMM_SELF 1725c464158bSHong Zhang . bs - size of block 1726c464158bSHong Zhang . m - number of rows, or number of columns 1727c464158bSHong Zhang . nz - number of block nonzeros per block row (same for all rows) 1728744e8345SSatish Balay - nnz - array containing the number of block nonzeros in the upper triangular plus 1729744e8345SSatish Balay diagonal portion of each block (possibly different for each block row) or PETSC_NULL 1730c464158bSHong Zhang 1731c464158bSHong Zhang Output Parameter: 1732c464158bSHong Zhang . A - the symmetric matrix 1733c464158bSHong Zhang 1734c464158bSHong Zhang Options Database Keys: 1735c464158bSHong Zhang . -mat_no_unroll - uses code that does not unroll the loops in the 1736c464158bSHong Zhang block calculations (much slower) 1737c464158bSHong Zhang . -mat_block_size - size of the blocks to use 1738c464158bSHong Zhang 1739c464158bSHong Zhang Level: intermediate 1740c464158bSHong Zhang 1741c464158bSHong Zhang Notes: 1742c464158bSHong Zhang 1743c464158bSHong Zhang Specify the preallocated storage with either nz or nnz (not both). 1744c464158bSHong Zhang Set nz=PETSC_DEFAULT and nnz=PETSC_NULL for PETSc to control dynamic memory 1745c464158bSHong Zhang allocation. For additional details, see the users manual chapter on 1746c464158bSHong Zhang matrices. 1747c464158bSHong Zhang 174849a6f317SBarry Smith If the nnz parameter is given then the nz parameter is ignored 174949a6f317SBarry Smith 1750c464158bSHong Zhang .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateMPISBAIJ() 1751c464158bSHong Zhang @*/ 1752be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreateSeqSBAIJ(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt nz,const PetscInt nnz[],Mat *A) 1753c464158bSHong Zhang { 1754dfbe8321SBarry Smith PetscErrorCode ierr; 1755c464158bSHong Zhang 1756c464158bSHong Zhang PetscFunctionBegin; 1757f69a0ea3SMatthew Knepley ierr = MatCreate(comm,A);CHKERRQ(ierr); 1758f69a0ea3SMatthew Knepley ierr = MatSetSizes(*A,m,n,m,n);CHKERRQ(ierr); 1759c464158bSHong Zhang ierr = MatSetType(*A,MATSEQSBAIJ);CHKERRQ(ierr); 1760ab93d7beSBarry Smith ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(*A,bs,nz,(PetscInt*)nnz);CHKERRQ(ierr); 176149b5e25fSSatish Balay PetscFunctionReturn(0); 176249b5e25fSSatish Balay } 176349b5e25fSSatish Balay 17644a2ae208SSatish Balay #undef __FUNCT__ 17654a2ae208SSatish Balay #define __FUNCT__ "MatDuplicate_SeqSBAIJ" 1766dfbe8321SBarry Smith PetscErrorCode MatDuplicate_SeqSBAIJ(Mat A,MatDuplicateOption cpvalues,Mat *B) 176749b5e25fSSatish Balay { 176849b5e25fSSatish Balay Mat C; 176949b5e25fSSatish Balay Mat_SeqSBAIJ *c,*a = (Mat_SeqSBAIJ*)A->data; 17706849ba73SBarry Smith PetscErrorCode ierr; 1771b40805acSSatish Balay PetscInt i,mbs = a->mbs,nz = a->nz,bs2 =a->bs2; 177249b5e25fSSatish Balay 177349b5e25fSSatish Balay PetscFunctionBegin; 177429bbc08cSBarry Smith if (a->i[mbs] != nz) SETERRQ(PETSC_ERR_PLIB,"Corrupt matrix"); 177549b5e25fSSatish Balay 177649b5e25fSSatish Balay *B = 0; 1777f69a0ea3SMatthew Knepley ierr = MatCreate(A->comm,&C);CHKERRQ(ierr); 1778899cda47SBarry Smith ierr = MatSetSizes(C,A->rmap.N,A->cmap.n,A->rmap.N,A->cmap.n);CHKERRQ(ierr); 1779be5d1d56SKris Buschelman ierr = MatSetType(C,A->type_name);CHKERRQ(ierr); 17801d5dac46SHong Zhang ierr = PetscMemcpy(C->ops,A->ops,sizeof(struct _MatOps));CHKERRQ(ierr); 1781692f9cbeSHong Zhang c = (Mat_SeqSBAIJ*)C->data; 1782692f9cbeSHong Zhang 1783273d9f13SBarry Smith C->preallocated = PETSC_TRUE; 178449b5e25fSSatish Balay C->factor = A->factor; 178549b5e25fSSatish Balay c->row = 0; 178649b5e25fSSatish Balay c->icol = 0; 178749b5e25fSSatish Balay c->saved_values = 0; 178849b5e25fSSatish Balay c->keepzeroedrows = a->keepzeroedrows; 178949b5e25fSSatish Balay C->assembled = PETSC_TRUE; 179049b5e25fSSatish Balay 1791899cda47SBarry Smith ierr = PetscMapCopy(A->comm,&A->rmap,&C->rmap);CHKERRQ(ierr); 1792899cda47SBarry Smith ierr = PetscMapCopy(A->comm,&A->cmap,&C->cmap);CHKERRQ(ierr); 179349b5e25fSSatish Balay c->bs2 = a->bs2; 179449b5e25fSSatish Balay c->mbs = a->mbs; 179549b5e25fSSatish Balay c->nbs = a->nbs; 179649b5e25fSSatish Balay 17978777fc3fSSatish Balay ierr = PetscMalloc2((mbs+1),PetscInt,&c->imax,(mbs+1),PetscInt,&c->ilen);CHKERRQ(ierr); 179849b5e25fSSatish Balay for (i=0; i<mbs; i++) { 179949b5e25fSSatish Balay c->imax[i] = a->imax[i]; 180049b5e25fSSatish Balay c->ilen[i] = a->ilen[i]; 180149b5e25fSSatish Balay } 180249b5e25fSSatish Balay 180349b5e25fSSatish Balay /* allocate the matrix space */ 1804b40805acSSatish Balay ierr = PetscMalloc3(bs2*nz,MatScalar,&c->a,nz,PetscInt,&c->j,mbs+1,PetscInt,&c->i);CHKERRQ(ierr); 180549b5e25fSSatish Balay c->singlemalloc = PETSC_TRUE; 180613f74950SBarry Smith ierr = PetscMemcpy(c->i,a->i,(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr); 1807b40805acSSatish Balay ierr = PetscLogObjectMemory(C,(mbs+1)*sizeof(PetscInt) + nz*(bs2*sizeof(MatScalar) + sizeof(PetscInt)));CHKERRQ(ierr); 180849b5e25fSSatish Balay if (mbs > 0) { 180913f74950SBarry Smith ierr = PetscMemcpy(c->j,a->j,nz*sizeof(PetscInt));CHKERRQ(ierr); 181049b5e25fSSatish Balay if (cpvalues == MAT_COPY_VALUES) { 181149b5e25fSSatish Balay ierr = PetscMemcpy(c->a,a->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr); 181249b5e25fSSatish Balay } else { 181349b5e25fSSatish Balay ierr = PetscMemzero(c->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr); 181449b5e25fSSatish Balay } 181549b5e25fSSatish Balay } 181649b5e25fSSatish Balay 181749b5e25fSSatish Balay c->sorted = a->sorted; 181849b5e25fSSatish Balay c->roworiented = a->roworiented; 181949b5e25fSSatish Balay c->nonew = a->nonew; 182049b5e25fSSatish Balay 182149b5e25fSSatish Balay if (a->diag) { 182213f74950SBarry Smith ierr = PetscMalloc((mbs+1)*sizeof(PetscInt),&c->diag);CHKERRQ(ierr); 182352e6d16bSBarry Smith ierr = PetscLogObjectMemory(C,(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr); 182449b5e25fSSatish Balay for (i=0; i<mbs; i++) { 182549b5e25fSSatish Balay c->diag[i] = a->diag[i]; 182649b5e25fSSatish Balay } 182749b5e25fSSatish Balay } else c->diag = 0; 18286c6c5352SBarry Smith c->nz = a->nz; 18296c6c5352SBarry Smith c->maxnz = a->maxnz; 183049b5e25fSSatish Balay c->solve_work = 0; 183149b5e25fSSatish Balay c->mult_work = 0; 1832e6b907acSBarry Smith c->free_a = PETSC_TRUE; 1833e6b907acSBarry Smith c->free_ij = PETSC_TRUE; 183449b5e25fSSatish Balay *B = C; 1835b0a32e0cSBarry Smith ierr = PetscFListDuplicate(A->qlist,&C->qlist);CHKERRQ(ierr); 183649b5e25fSSatish Balay PetscFunctionReturn(0); 183749b5e25fSSatish Balay } 183849b5e25fSSatish Balay 18394a2ae208SSatish Balay #undef __FUNCT__ 18404a2ae208SSatish Balay #define __FUNCT__ "MatLoad_SeqSBAIJ" 1841f69a0ea3SMatthew Knepley PetscErrorCode MatLoad_SeqSBAIJ(PetscViewer viewer, MatType type,Mat *A) 184249b5e25fSSatish Balay { 184349b5e25fSSatish Balay Mat_SeqSBAIJ *a; 184449b5e25fSSatish Balay Mat B; 18456849ba73SBarry Smith PetscErrorCode ierr; 184613f74950SBarry Smith int fd; 184713f74950SBarry Smith PetscMPIInt size; 184813f74950SBarry Smith PetscInt i,nz,header[4],*rowlengths=0,M,N,bs=1; 184913f74950SBarry Smith PetscInt *mask,mbs,*jj,j,rowcount,nzcount,k,*s_browlengths,maskcount; 185013f74950SBarry Smith PetscInt kmax,jcount,block,idx,point,nzcountb,extra_rows; 185113f74950SBarry Smith PetscInt *masked,nmask,tmp,bs2,ishift; 185287828ca2SBarry Smith PetscScalar *aa; 185349b5e25fSSatish Balay MPI_Comm comm = ((PetscObject)viewer)->comm; 185449b5e25fSSatish Balay 185549b5e25fSSatish Balay PetscFunctionBegin; 1856b0a32e0cSBarry Smith ierr = PetscOptionsGetInt(PETSC_NULL,"-matload_block_size",&bs,PETSC_NULL);CHKERRQ(ierr); 185749b5e25fSSatish Balay bs2 = bs*bs; 185849b5e25fSSatish Balay 185949b5e25fSSatish Balay ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 186029bbc08cSBarry Smith if (size > 1) SETERRQ(PETSC_ERR_ARG_WRONG,"view must have one processor"); 1861b0a32e0cSBarry Smith ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr); 186249b5e25fSSatish Balay ierr = PetscBinaryRead(fd,header,4,PETSC_INT);CHKERRQ(ierr); 1863552e946dSBarry Smith if (header[0] != MAT_FILE_COOKIE) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"not Mat object"); 186449b5e25fSSatish Balay M = header[1]; N = header[2]; nz = header[3]; 186549b5e25fSSatish Balay 186649b5e25fSSatish Balay if (header[3] < 0) { 186729bbc08cSBarry Smith SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"Matrix stored in special format, cannot load as SeqSBAIJ"); 186849b5e25fSSatish Balay } 186949b5e25fSSatish Balay 187029bbc08cSBarry Smith if (M != N) SETERRQ(PETSC_ERR_SUP,"Can only do square matrices"); 187149b5e25fSSatish Balay 187249b5e25fSSatish Balay /* 187349b5e25fSSatish Balay This code adds extra rows to make sure the number of rows is 187449b5e25fSSatish Balay divisible by the blocksize 187549b5e25fSSatish Balay */ 187649b5e25fSSatish Balay mbs = M/bs; 187749b5e25fSSatish Balay extra_rows = bs - M + bs*(mbs); 187849b5e25fSSatish Balay if (extra_rows == bs) extra_rows = 0; 187949b5e25fSSatish Balay else mbs++; 188049b5e25fSSatish Balay if (extra_rows) { 1881ae15b995SBarry Smith ierr = PetscInfo(0,"Padding loaded matrix to match blocksize\n");CHKERRQ(ierr); 188249b5e25fSSatish Balay } 188349b5e25fSSatish Balay 188449b5e25fSSatish Balay /* read in row lengths */ 188513f74950SBarry Smith ierr = PetscMalloc((M+extra_rows)*sizeof(PetscInt),&rowlengths);CHKERRQ(ierr); 188649b5e25fSSatish Balay ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr); 188749b5e25fSSatish Balay for (i=0; i<extra_rows; i++) rowlengths[M+i] = 1; 188849b5e25fSSatish Balay 188949b5e25fSSatish Balay /* read in column indices */ 189013f74950SBarry Smith ierr = PetscMalloc((nz+extra_rows)*sizeof(PetscInt),&jj);CHKERRQ(ierr); 189149b5e25fSSatish Balay ierr = PetscBinaryRead(fd,jj,nz,PETSC_INT);CHKERRQ(ierr); 189249b5e25fSSatish Balay for (i=0; i<extra_rows; i++) jj[nz+i] = M+i; 189349b5e25fSSatish Balay 189449b5e25fSSatish Balay /* loop over row lengths determining block row lengths */ 189513f74950SBarry Smith ierr = PetscMalloc(mbs*sizeof(PetscInt),&s_browlengths);CHKERRQ(ierr); 189613f74950SBarry Smith ierr = PetscMemzero(s_browlengths,mbs*sizeof(PetscInt));CHKERRQ(ierr); 189713f74950SBarry Smith ierr = PetscMalloc(2*mbs*sizeof(PetscInt),&mask);CHKERRQ(ierr); 189813f74950SBarry Smith ierr = PetscMemzero(mask,mbs*sizeof(PetscInt));CHKERRQ(ierr); 189949b5e25fSSatish Balay masked = mask + mbs; 190049b5e25fSSatish Balay rowcount = 0; nzcount = 0; 190149b5e25fSSatish Balay for (i=0; i<mbs; i++) { 190249b5e25fSSatish Balay nmask = 0; 190349b5e25fSSatish Balay for (j=0; j<bs; j++) { 190449b5e25fSSatish Balay kmax = rowlengths[rowcount]; 190549b5e25fSSatish Balay for (k=0; k<kmax; k++) { 19062d703238SHong Zhang tmp = jj[nzcount++]/bs; /* block col. index */ 190703630b6eSHong Zhang if (!mask[tmp] && tmp >= i) {masked[nmask++] = tmp; mask[tmp] = 1;} 190849b5e25fSSatish Balay } 190949b5e25fSSatish Balay rowcount++; 191049b5e25fSSatish Balay } 1911574b2666SHong Zhang s_browlengths[i] += nmask; 1912574b2666SHong Zhang 191349b5e25fSSatish Balay /* zero out the mask elements we set */ 191449b5e25fSSatish Balay for (j=0; j<nmask; j++) mask[masked[j]] = 0; 191549b5e25fSSatish Balay } 191649b5e25fSSatish Balay 191749b5e25fSSatish Balay /* create our matrix */ 1918f69a0ea3SMatthew Knepley ierr = MatCreate(comm,&B);CHKERRQ(ierr); 1919f69a0ea3SMatthew Knepley ierr = MatSetSizes(B,M+extra_rows,N+extra_rows,M+extra_rows,N+extra_rows);CHKERRQ(ierr); 19209abb65ffSKris Buschelman ierr = MatSetType(B,type);CHKERRQ(ierr); 1921ab93d7beSBarry Smith ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(B,bs,0,s_browlengths);CHKERRQ(ierr); 192249b5e25fSSatish Balay a = (Mat_SeqSBAIJ*)B->data; 192349b5e25fSSatish Balay 192449b5e25fSSatish Balay /* set matrix "i" values */ 192549b5e25fSSatish Balay a->i[0] = 0; 192649b5e25fSSatish Balay for (i=1; i<= mbs; i++) { 1927574b2666SHong Zhang a->i[i] = a->i[i-1] + s_browlengths[i-1]; 1928574b2666SHong Zhang a->ilen[i-1] = s_browlengths[i-1]; 192949b5e25fSSatish Balay } 19306c6c5352SBarry Smith a->nz = a->i[mbs]; 193149b5e25fSSatish Balay 193249b5e25fSSatish Balay /* read in nonzero values */ 193387828ca2SBarry Smith ierr = PetscMalloc((nz+extra_rows)*sizeof(PetscScalar),&aa);CHKERRQ(ierr); 193449b5e25fSSatish Balay ierr = PetscBinaryRead(fd,aa,nz,PETSC_SCALAR);CHKERRQ(ierr); 193549b5e25fSSatish Balay for (i=0; i<extra_rows; i++) aa[nz+i] = 1.0; 193649b5e25fSSatish Balay 193749b5e25fSSatish Balay /* set "a" and "j" values into matrix */ 193849b5e25fSSatish Balay nzcount = 0; jcount = 0; 193949b5e25fSSatish Balay for (i=0; i<mbs; i++) { 194049b5e25fSSatish Balay nzcountb = nzcount; 194149b5e25fSSatish Balay nmask = 0; 194249b5e25fSSatish Balay for (j=0; j<bs; j++) { 194349b5e25fSSatish Balay kmax = rowlengths[i*bs+j]; 194449b5e25fSSatish Balay for (k=0; k<kmax; k++) { 19452d703238SHong Zhang tmp = jj[nzcount++]/bs; /* block col. index */ 194603630b6eSHong Zhang if (!mask[tmp] && tmp >= i) { masked[nmask++] = tmp; mask[tmp] = 1;} 19472d703238SHong Zhang } 19482d703238SHong Zhang } 19492d703238SHong Zhang /* sort the masked values */ 19502d703238SHong Zhang ierr = PetscSortInt(nmask,masked);CHKERRQ(ierr); 19512d703238SHong Zhang 19522d703238SHong Zhang /* set "j" values into matrix */ 19532d703238SHong Zhang maskcount = 1; 19542d703238SHong Zhang for (j=0; j<nmask; j++) { 195549b5e25fSSatish Balay a->j[jcount++] = masked[j]; 195649b5e25fSSatish Balay mask[masked[j]] = maskcount++; 195749b5e25fSSatish Balay } 1958574b2666SHong Zhang 195949b5e25fSSatish Balay /* set "a" values into matrix */ 196049b5e25fSSatish Balay ishift = bs2*a->i[i]; 196149b5e25fSSatish Balay for (j=0; j<bs; j++) { 196249b5e25fSSatish Balay kmax = rowlengths[i*bs+j]; 196349b5e25fSSatish Balay for (k=0; k<kmax; k++) { 1964574b2666SHong Zhang tmp = jj[nzcountb]/bs ; /* block col. index */ 1965574b2666SHong Zhang if (tmp >= i){ 196649b5e25fSSatish Balay block = mask[tmp] - 1; 196749b5e25fSSatish Balay point = jj[nzcountb] - bs*tmp; 196849b5e25fSSatish Balay idx = ishift + bs2*block + j + bs*point; 1969574b2666SHong Zhang a->a[idx] = aa[nzcountb]; 1970574b2666SHong Zhang } 1971574b2666SHong Zhang nzcountb++; 197249b5e25fSSatish Balay } 197349b5e25fSSatish Balay } 197449b5e25fSSatish Balay /* zero out the mask elements we set */ 197549b5e25fSSatish Balay for (j=0; j<nmask; j++) mask[masked[j]] = 0; 197649b5e25fSSatish Balay } 19776c6c5352SBarry Smith if (jcount != a->nz) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"Bad binary matrix"); 197849b5e25fSSatish Balay 197949b5e25fSSatish Balay ierr = PetscFree(rowlengths);CHKERRQ(ierr); 1980574b2666SHong Zhang ierr = PetscFree(s_browlengths);CHKERRQ(ierr); 198149b5e25fSSatish Balay ierr = PetscFree(aa);CHKERRQ(ierr); 198249b5e25fSSatish Balay ierr = PetscFree(jj);CHKERRQ(ierr); 198349b5e25fSSatish Balay ierr = PetscFree(mask);CHKERRQ(ierr); 198449b5e25fSSatish Balay 19859abb65ffSKris Buschelman ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 19869abb65ffSKris Buschelman ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 198749b5e25fSSatish Balay ierr = MatView_Private(B);CHKERRQ(ierr); 19889abb65ffSKris Buschelman *A = B; 198949b5e25fSSatish Balay PetscFunctionReturn(0); 199049b5e25fSSatish Balay } 1991574b2666SHong Zhang 1992d06b337dSHong Zhang #undef __FUNCT__ 1993d06b337dSHong Zhang #define __FUNCT__ "MatRelax_SeqSBAIJ" 199413f74950SBarry Smith PetscErrorCode MatRelax_SeqSBAIJ(Mat A,Vec bb,PetscReal omega,MatSORType flag,PetscReal fshift,PetscInt its,PetscInt lits,Vec xx) 1995d06b337dSHong Zhang { 1996d06b337dSHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 1997d06b337dSHong Zhang MatScalar *aa=a->a,*v,*v1; 1998d06b337dSHong Zhang PetscScalar *x,*b,*t,sum,d; 19996849ba73SBarry Smith PetscErrorCode ierr; 2000899cda47SBarry Smith PetscInt m=a->mbs,bs=A->rmap.bs,*ai=a->i,*aj=a->j; 2001521d7252SBarry Smith PetscInt nz,nz1,*vj,*vj1,i; 2002d06b337dSHong Zhang 2003d06b337dSHong Zhang PetscFunctionBegin; 200451f519a2SBarry Smith if (flag & SOR_EISENSTAT) SETERRQ(PETSC_ERR_SUP,"No support yet for Eisenstat"); 200551f519a2SBarry Smith 2006b965ef7fSBarry Smith its = its*lits; 200777431f27SBarry Smith if (its <= 0) SETERRQ2(PETSC_ERR_ARG_WRONG,"Relaxation requires global its %D and local its %D both positive",its,lits); 2008d06b337dSHong Zhang 2009d06b337dSHong Zhang if (bs > 1) 2010d06b337dSHong Zhang SETERRQ(PETSC_ERR_SUP,"SSOR for block size > 1 is not yet implemented"); 2011d06b337dSHong Zhang 20121ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 2013d06b337dSHong Zhang if (xx != bb) { 20141ebc52fbSHong Zhang ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 2015d06b337dSHong Zhang } else { 2016d06b337dSHong Zhang b = x; 2017d06b337dSHong Zhang } 2018d06b337dSHong Zhang 2019e2ee2a47SBarry Smith if (!a->relax_work) { 2020e2ee2a47SBarry Smith ierr = PetscMalloc(m*sizeof(PetscScalar),&a->relax_work);CHKERRQ(ierr); 2021e2ee2a47SBarry Smith } 2022e2ee2a47SBarry Smith t = a->relax_work; 2023d06b337dSHong Zhang 2024d06b337dSHong Zhang if (flag & SOR_ZERO_INITIAL_GUESS) { 20252798e883SHong Zhang if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){ 2026290bbb0aSBarry Smith ierr = PetscMemcpy(t,b,m*sizeof(PetscScalar));CHKERRQ(ierr); 2027d06b337dSHong Zhang 2028d06b337dSHong Zhang for (i=0; i<m; i++){ 202944706875SHong Zhang d = *(aa + ai[i]); /* diag[i] */ 2030d06b337dSHong Zhang v = aa + ai[i] + 1; 2031d06b337dSHong Zhang vj = aj + ai[i] + 1; 2032d06b337dSHong Zhang nz = ai[i+1] - ai[i] - 1; 2033e6b907acSBarry Smith ierr = PetscLogFlops(2*nz-1);CHKERRQ(ierr); 2034d06b337dSHong Zhang x[i] = omega*t[i]/d; 2035d06b337dSHong Zhang while (nz--) t[*vj++] -= x[i]*(*v++); /* update rhs */ 2036d06b337dSHong Zhang } 2037d06b337dSHong Zhang } 2038d06b337dSHong Zhang 20392798e883SHong Zhang if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){ 204095d750ceSBarry Smith if (!(flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP)){ 204195d750ceSBarry Smith t = b; 2042d06b337dSHong Zhang } 204395d750ceSBarry Smith 2044d06b337dSHong Zhang for (i=m-1; i>=0; i--){ 2045d06b337dSHong Zhang d = *(aa + ai[i]); 2046d06b337dSHong Zhang v = aa + ai[i] + 1; 2047d06b337dSHong Zhang vj = aj + ai[i] + 1; 2048d06b337dSHong Zhang nz = ai[i+1] - ai[i] - 1; 2049e6b907acSBarry Smith ierr = PetscLogFlops(2*nz-1);CHKERRQ(ierr); 2050d06b337dSHong Zhang sum = t[i]; 2051d06b337dSHong Zhang while (nz--) sum -= x[*vj++]*(*v++); 2052d06b337dSHong Zhang x[i] = (1-omega)*x[i] + omega*sum/d; 2053d06b337dSHong Zhang } 205495d750ceSBarry Smith t = a->relax_work; 2055d06b337dSHong Zhang } 2056d06b337dSHong Zhang its--; 2057d06b337dSHong Zhang } 2058d06b337dSHong Zhang 2059d06b337dSHong Zhang while (its--) { 206044706875SHong Zhang /* 206144706875SHong Zhang forward sweep: 206244706875SHong Zhang for i=0,...,m-1: 206344706875SHong Zhang sum[i] = (b[i] - U(i,:)x )/d[i]; 206444706875SHong Zhang x[i] = (1-omega)x[i] + omega*sum[i]; 206544706875SHong Zhang b = b - x[i]*U^T(i,:); 2066d06b337dSHong Zhang 206744706875SHong Zhang */ 20682798e883SHong Zhang if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){ 2069290bbb0aSBarry Smith ierr = PetscMemcpy(t,b,m*sizeof(PetscScalar));CHKERRQ(ierr); 2070d06b337dSHong Zhang 2071d06b337dSHong Zhang for (i=0; i<m; i++){ 207244706875SHong Zhang d = *(aa + ai[i]); /* diag[i] */ 2073d06b337dSHong Zhang v = aa + ai[i] + 1; v1=v; 2074d06b337dSHong Zhang vj = aj + ai[i] + 1; vj1=vj; 2075d06b337dSHong Zhang nz = ai[i+1] - ai[i] - 1; nz1=nz; 2076d06b337dSHong Zhang sum = t[i]; 2077e6b907acSBarry Smith ierr = PetscLogFlops(4*nz-2);CHKERRQ(ierr); 2078d06b337dSHong Zhang while (nz1--) sum -= (*v1++)*x[*vj1++]; 2079d06b337dSHong Zhang x[i] = (1-omega)*x[i] + omega*sum/d; 2080d06b337dSHong Zhang while (nz--) t[*vj++] -= x[i]*(*v++); 2081d06b337dSHong Zhang } 2082d06b337dSHong Zhang } 2083d06b337dSHong Zhang 20842798e883SHong Zhang if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){ 208544706875SHong Zhang /* 208644706875SHong Zhang backward sweep: 208744706875SHong Zhang b = b - x[i]*U^T(i,:), i=0,...,n-2 208844706875SHong Zhang for i=m-1,...,0: 208944706875SHong Zhang sum[i] = (b[i] - U(i,:)x )/d[i]; 209044706875SHong Zhang x[i] = (1-omega)x[i] + omega*sum[i]; 209144706875SHong Zhang */ 209295d750ceSBarry Smith /* if there was a forward sweep done above then I thing the next two for loops are not needed */ 2093290bbb0aSBarry Smith ierr = PetscMemcpy(t,b,m*sizeof(PetscScalar));CHKERRQ(ierr); 2094d06b337dSHong Zhang 2095d06b337dSHong Zhang for (i=0; i<m-1; i++){ /* update rhs */ 2096d06b337dSHong Zhang v = aa + ai[i] + 1; 2097d06b337dSHong Zhang vj = aj + ai[i] + 1; 2098d06b337dSHong Zhang nz = ai[i+1] - ai[i] - 1; 2099efee365bSSatish Balay ierr = PetscLogFlops(2*nz-1);CHKERRQ(ierr); 2100e6b907acSBarry Smith while (nz--) t[*vj++] -= x[i]*(*v++); 2101d06b337dSHong Zhang } 2102d06b337dSHong Zhang for (i=m-1; i>=0; i--){ 2103d06b337dSHong Zhang d = *(aa + ai[i]); 2104d06b337dSHong Zhang v = aa + ai[i] + 1; 2105d06b337dSHong Zhang vj = aj + ai[i] + 1; 2106d06b337dSHong Zhang nz = ai[i+1] - ai[i] - 1; 2107e6b907acSBarry Smith ierr = PetscLogFlops(2*nz-1);CHKERRQ(ierr); 2108d06b337dSHong Zhang sum = t[i]; 2109d06b337dSHong Zhang while (nz--) sum -= x[*vj++]*(*v++); 2110d06b337dSHong Zhang x[i] = (1-omega)*x[i] + omega*sum/d; 2111d06b337dSHong Zhang } 2112d06b337dSHong Zhang } 2113d06b337dSHong Zhang } 2114d06b337dSHong Zhang 21151ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 2116d06b337dSHong Zhang if (bb != xx) { 21171ebc52fbSHong Zhang ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr); 2118d06b337dSHong Zhang } 2119d06b337dSHong Zhang PetscFunctionReturn(0); 2120d06b337dSHong Zhang } 2121d06b337dSHong Zhang 2122c75a6043SHong Zhang #undef __FUNCT__ 2123c75a6043SHong Zhang #define __FUNCT__ "MatCreateSeqSBAIJWithArrays" 2124c75a6043SHong Zhang /*@ 2125c75a6043SHong Zhang MatCreateSeqSBAIJWithArrays - Creates an sequential SBAIJ matrix using matrix elements 2126c75a6043SHong Zhang (upper triangular entries in CSR format) provided by the user. 2127c75a6043SHong Zhang 2128c75a6043SHong Zhang Collective on MPI_Comm 2129c75a6043SHong Zhang 2130c75a6043SHong Zhang Input Parameters: 2131c75a6043SHong Zhang + comm - must be an MPI communicator of size 1 2132c75a6043SHong Zhang . bs - size of block 2133c75a6043SHong Zhang . m - number of rows 2134c75a6043SHong Zhang . n - number of columns 2135c75a6043SHong Zhang . i - row indices 2136c75a6043SHong Zhang . j - column indices 2137c75a6043SHong Zhang - a - matrix values 2138c75a6043SHong Zhang 2139c75a6043SHong Zhang Output Parameter: 2140c75a6043SHong Zhang . mat - the matrix 2141c75a6043SHong Zhang 2142c75a6043SHong Zhang Level: intermediate 2143c75a6043SHong Zhang 2144c75a6043SHong Zhang Notes: 2145c75a6043SHong Zhang The i, j, and a arrays are not copied by this routine, the user must free these arrays 2146c75a6043SHong Zhang once the matrix is destroyed 2147c75a6043SHong Zhang 2148c75a6043SHong Zhang You cannot set new nonzero locations into this matrix, that will generate an error. 2149c75a6043SHong Zhang 2150c75a6043SHong Zhang The i and j indices are 0 based 2151c75a6043SHong Zhang 2152c75a6043SHong Zhang .seealso: MatCreate(), MatCreateMPISBAIJ(), MatCreateSeqSBAIJ() 2153c75a6043SHong Zhang 2154c75a6043SHong Zhang @*/ 2155c75a6043SHong Zhang PetscErrorCode PETSCMAT_DLLEXPORT MatCreateSeqSBAIJWithArrays(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt* i,PetscInt*j,PetscScalar *a,Mat *mat) 2156c75a6043SHong Zhang { 2157c75a6043SHong Zhang PetscErrorCode ierr; 2158c75a6043SHong Zhang PetscInt ii; 2159c75a6043SHong Zhang Mat_SeqSBAIJ *sbaij; 2160c75a6043SHong Zhang 2161c75a6043SHong Zhang PetscFunctionBegin; 2162c75a6043SHong Zhang if (bs != 1) SETERRQ1(PETSC_ERR_SUP,"block size %D > 1 is not supported yet",bs); 2163c75a6043SHong Zhang if (i[0]) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,"i (row indices) must start with 0"); 2164c75a6043SHong Zhang 2165c75a6043SHong Zhang ierr = MatCreate(comm,mat);CHKERRQ(ierr); 2166c75a6043SHong Zhang ierr = MatSetSizes(*mat,m,n,m,n);CHKERRQ(ierr); 2167c75a6043SHong Zhang ierr = MatSetType(*mat,MATSEQSBAIJ);CHKERRQ(ierr); 2168c75a6043SHong Zhang ierr = MatSeqSBAIJSetPreallocation_SeqSBAIJ(*mat,bs,MAT_SKIP_ALLOCATION,0);CHKERRQ(ierr); 2169c75a6043SHong Zhang sbaij = (Mat_SeqSBAIJ*)(*mat)->data; 2170c75a6043SHong Zhang ierr = PetscMalloc2(m,PetscInt,&sbaij->imax,m,PetscInt,&sbaij->ilen);CHKERRQ(ierr); 2171c75a6043SHong Zhang 2172c75a6043SHong Zhang sbaij->i = i; 2173c75a6043SHong Zhang sbaij->j = j; 2174c75a6043SHong Zhang sbaij->a = a; 2175c75a6043SHong Zhang sbaij->singlemalloc = PETSC_FALSE; 2176c75a6043SHong Zhang sbaij->nonew = -1; /*this indicates that inserting a new value in the matrix that generates a new nonzero is an error*/ 2177e6b907acSBarry Smith sbaij->free_a = PETSC_FALSE; 2178e6b907acSBarry Smith sbaij->free_ij = PETSC_FALSE; 2179c75a6043SHong Zhang 2180c75a6043SHong Zhang for (ii=0; ii<m; ii++) { 2181c75a6043SHong Zhang sbaij->ilen[ii] = sbaij->imax[ii] = i[ii+1] - i[ii]; 2182c75a6043SHong Zhang #if defined(PETSC_USE_DEBUG) 2183c75a6043SHong Zhang if (i[ii+1] - i[ii] < 0) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Negative row length in i (row indices) row = %d length = %d",ii,i[ii+1] - i[ii]); 2184c75a6043SHong Zhang #endif 2185c75a6043SHong Zhang } 2186c75a6043SHong Zhang #if defined(PETSC_USE_DEBUG) 2187c75a6043SHong Zhang for (ii=0; ii<sbaij->i[m]; ii++) { 2188c75a6043SHong Zhang if (j[ii] < 0) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Negative column index at location = %d index = %d",ii,j[ii]); 2189c75a6043SHong Zhang if (j[ii] > n - 1) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Column index to large at location = %d index = %d",ii,j[ii]); 2190c75a6043SHong Zhang } 2191c75a6043SHong Zhang #endif 2192c75a6043SHong Zhang 2193c75a6043SHong Zhang ierr = MatAssemblyBegin(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2194c75a6043SHong Zhang ierr = MatAssemblyEnd(*mat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 2195c75a6043SHong Zhang PetscFunctionReturn(0); 2196c75a6043SHong Zhang } 2197d06b337dSHong Zhang 2198d06b337dSHong Zhang 2199d06b337dSHong Zhang 220049b5e25fSSatish Balay 220149b5e25fSSatish Balay 2202