149b5e25fSSatish Balay 249b5e25fSSatish Balay /* 3a1373b80SHong Zhang Defines the basic matrix operations for the SBAIJ (compressed row) 449b5e25fSSatish Balay matrix storage format. 549b5e25fSSatish Balay */ 69e070d67SMatthew Knepley #include "src/mat/impls/baij/seq/baij.h" /*I "petscmat.h" I*/ 749b5e25fSSatish Balay #include "src/inline/spops.h" 8aec82049SSatish Balay #include "src/mat/impls/sbaij/seq/sbaij.h" 949b5e25fSSatish Balay 1049b5e25fSSatish Balay #define CHUNKSIZE 10 1149b5e25fSSatish Balay 1249b5e25fSSatish Balay /* 1349b5e25fSSatish Balay Checks for missing diagonals 1449b5e25fSSatish Balay */ 154a2ae208SSatish Balay #undef __FUNCT__ 164a2ae208SSatish Balay #define __FUNCT__ "MatMissingDiagonal_SeqSBAIJ" 17dfbe8321SBarry Smith PetscErrorCode MatMissingDiagonal_SeqSBAIJ(Mat A) 1849b5e25fSSatish Balay { 19045c9aa0SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 206849ba73SBarry Smith PetscErrorCode ierr; 2113f74950SBarry Smith PetscInt *diag,*jj = a->j,i; 2249b5e25fSSatish Balay 2349b5e25fSSatish Balay PetscFunctionBegin; 24045c9aa0SHong Zhang ierr = MatMarkDiagonal_SeqSBAIJ(A);CHKERRQ(ierr); 2549b5e25fSSatish Balay diag = a->diag; 2649b5e25fSSatish Balay for (i=0; i<a->mbs; i++) { 2777431f27SBarry Smith if (jj[diag[i]] != i) SETERRQ1(PETSC_ERR_ARG_CORRUPT,"Matrix is missing diagonal number %D",i); 2849b5e25fSSatish Balay } 2949b5e25fSSatish Balay PetscFunctionReturn(0); 3049b5e25fSSatish Balay } 3149b5e25fSSatish Balay 324a2ae208SSatish Balay #undef __FUNCT__ 334a2ae208SSatish Balay #define __FUNCT__ "MatMarkDiagonal_SeqSBAIJ" 34dfbe8321SBarry Smith PetscErrorCode MatMarkDiagonal_SeqSBAIJ(Mat A) 3549b5e25fSSatish Balay { 36045c9aa0SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 376849ba73SBarry Smith PetscErrorCode ierr; 3813f74950SBarry Smith PetscInt i,mbs = a->mbs; 3949b5e25fSSatish Balay 4049b5e25fSSatish Balay PetscFunctionBegin; 4149b5e25fSSatish Balay if (a->diag) PetscFunctionReturn(0); 4249b5e25fSSatish Balay 4313f74950SBarry Smith ierr = PetscMalloc((mbs+1)*sizeof(PetscInt),&a->diag);CHKERRQ(ierr); 4452e6d16bSBarry Smith ierr = PetscLogObjectMemory(A,(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr); 45b424e231SHong Zhang for (i=0; i<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; 9877431f27SBarry Smith PetscLogObjectState((PetscObject)A,"Rows=%D, NZ=%D",A->m,a->nz); 9949b5e25fSSatish Balay ierr = PetscFree(a->a);CHKERRQ(ierr); 10049b5e25fSSatish Balay if (!a->singlemalloc) { 10149b5e25fSSatish Balay ierr = PetscFree(a->i);CHKERRQ(ierr); 10263c8bf9fSHong Zhang ierr = PetscFree(a->j);CHKERRQ(ierr); 10349b5e25fSSatish Balay } 10449b5e25fSSatish Balay if (a->row) { 10549b5e25fSSatish Balay ierr = ISDestroy(a->row);CHKERRQ(ierr); 10649b5e25fSSatish Balay } 10749b5e25fSSatish Balay if (a->diag) {ierr = PetscFree(a->diag);CHKERRQ(ierr);} 10849b5e25fSSatish Balay if (a->ilen) {ierr = PetscFree(a->ilen);CHKERRQ(ierr);} 10949b5e25fSSatish Balay if (a->imax) {ierr = PetscFree(a->imax);CHKERRQ(ierr);} 11049b5e25fSSatish Balay if (a->icol) {ierr = ISDestroy(a->icol);CHKERRQ(ierr);} 111d59c15a7SBarry Smith if (a->solve_work) {ierr = PetscFree(a->solve_work);CHKERRQ(ierr);} 112d59c15a7SBarry Smith if (a->solves_work) {ierr = PetscFree(a->solves_work);CHKERRQ(ierr);} 113d59c15a7SBarry Smith if (a->mult_work) {ierr = PetscFree(a->mult_work);CHKERRQ(ierr);} 11449b5e25fSSatish Balay if (a->saved_values) {ierr = PetscFree(a->saved_values);CHKERRQ(ierr);} 115c4319e64SHong Zhang if (a->xtoy) {ierr = PetscFree(a->xtoy);CHKERRQ(ierr);} 1161a3463dfSHong Zhang 1171a3463dfSHong Zhang if (a->inew){ 1181a3463dfSHong Zhang ierr = PetscFree(a->inew);CHKERRQ(ierr); 1191a3463dfSHong Zhang a->inew = 0; 1201a3463dfSHong Zhang } 12149b5e25fSSatish Balay ierr = PetscFree(a);CHKERRQ(ierr); 122901853e0SKris Buschelman 123901853e0SKris Buschelman ierr = PetscObjectComposeFunction((PetscObject)A,"MatStoreValues_C","",PETSC_NULL);CHKERRQ(ierr); 124901853e0SKris Buschelman ierr = PetscObjectComposeFunction((PetscObject)A,"MatRetrieveValues_C","",PETSC_NULL);CHKERRQ(ierr); 125901853e0SKris Buschelman ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetColumnIndices_C","",PETSC_NULL);CHKERRQ(ierr); 126901853e0SKris Buschelman ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqaij_C","",PETSC_NULL);CHKERRQ(ierr); 127901853e0SKris Buschelman ierr = PetscObjectComposeFunction((PetscObject)A,"MatConvert_seqsbaij_seqbaij_C","",PETSC_NULL);CHKERRQ(ierr); 128901853e0SKris Buschelman ierr = PetscObjectComposeFunction((PetscObject)A,"MatSeqSBAIJSetPreallocation_C","",PETSC_NULL);CHKERRQ(ierr); 12949b5e25fSSatish Balay PetscFunctionReturn(0); 13049b5e25fSSatish Balay } 13149b5e25fSSatish Balay 1324a2ae208SSatish Balay #undef __FUNCT__ 1334a2ae208SSatish Balay #define __FUNCT__ "MatSetOption_SeqSBAIJ" 134dfbe8321SBarry Smith PetscErrorCode MatSetOption_SeqSBAIJ(Mat A,MatOption op) 13549b5e25fSSatish Balay { 136045c9aa0SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 13749b5e25fSSatish Balay 13849b5e25fSSatish Balay PetscFunctionBegin; 1394d9d31abSKris Buschelman switch (op) { 1404d9d31abSKris Buschelman case MAT_ROW_ORIENTED: 1414d9d31abSKris Buschelman a->roworiented = PETSC_TRUE; 1424d9d31abSKris Buschelman break; 1434d9d31abSKris Buschelman case MAT_COLUMN_ORIENTED: 1444d9d31abSKris Buschelman a->roworiented = PETSC_FALSE; 1454d9d31abSKris Buschelman break; 1464d9d31abSKris Buschelman case MAT_COLUMNS_SORTED: 1474d9d31abSKris Buschelman a->sorted = PETSC_TRUE; 1484d9d31abSKris Buschelman break; 1494d9d31abSKris Buschelman case MAT_COLUMNS_UNSORTED: 1504d9d31abSKris Buschelman a->sorted = PETSC_FALSE; 1514d9d31abSKris Buschelman break; 1524d9d31abSKris Buschelman case MAT_KEEP_ZEROED_ROWS: 1534d9d31abSKris Buschelman a->keepzeroedrows = PETSC_TRUE; 1544d9d31abSKris Buschelman break; 1554d9d31abSKris Buschelman case MAT_NO_NEW_NONZERO_LOCATIONS: 1564d9d31abSKris Buschelman a->nonew = 1; 1574d9d31abSKris Buschelman break; 1584d9d31abSKris Buschelman case MAT_NEW_NONZERO_LOCATION_ERR: 1594d9d31abSKris Buschelman a->nonew = -1; 1604d9d31abSKris Buschelman break; 1614d9d31abSKris Buschelman case MAT_NEW_NONZERO_ALLOCATION_ERR: 1624d9d31abSKris Buschelman a->nonew = -2; 1634d9d31abSKris Buschelman break; 1644d9d31abSKris Buschelman case MAT_YES_NEW_NONZERO_LOCATIONS: 1654d9d31abSKris Buschelman a->nonew = 0; 1664d9d31abSKris Buschelman break; 1674d9d31abSKris Buschelman case MAT_ROWS_SORTED: 1684d9d31abSKris Buschelman case MAT_ROWS_UNSORTED: 1694d9d31abSKris Buschelman case MAT_YES_NEW_DIAGONALS: 1704d9d31abSKris Buschelman case MAT_IGNORE_OFF_PROC_ENTRIES: 1714d9d31abSKris Buschelman case MAT_USE_HASH_TABLE: 172b0a32e0cSBarry Smith PetscLogInfo(A,"MatSetOption_SeqSBAIJ:Option ignored\n"); 1734d9d31abSKris Buschelman break; 1744d9d31abSKris Buschelman case MAT_NO_NEW_DIAGONALS: 17529bbc08cSBarry Smith SETERRQ(PETSC_ERR_SUP,"MAT_NO_NEW_DIAGONALS"); 1769a4540c5SBarry Smith case MAT_NOT_SYMMETRIC: 1779a4540c5SBarry Smith case MAT_NOT_STRUCTURALLY_SYMMETRIC: 1789a4540c5SBarry Smith case MAT_HERMITIAN: 1799a4540c5SBarry Smith SETERRQ(PETSC_ERR_SUP,"Matrix must be symmetric"); 18077e54ba9SKris Buschelman case MAT_SYMMETRIC: 18177e54ba9SKris Buschelman case MAT_STRUCTURALLY_SYMMETRIC: 1829a4540c5SBarry Smith case MAT_NOT_HERMITIAN: 1839a4540c5SBarry Smith case MAT_SYMMETRY_ETERNAL: 1849a4540c5SBarry Smith case MAT_NOT_SYMMETRY_ETERNAL: 185*941593c8SHong Zhang case MAT_IGNORE_LOWER_TRIANGULAR: 186*941593c8SHong Zhang a->ignore_ltriangular = PETSC_TRUE; 187*941593c8SHong Zhang break; 188*941593c8SHong Zhang case MAT_ERROR_LOWER_TRIANGULAR: 189*941593c8SHong Zhang a->ignore_ltriangular = PETSC_FALSE; 19077e54ba9SKris Buschelman break; 1914d9d31abSKris Buschelman default: 19229bbc08cSBarry Smith SETERRQ(PETSC_ERR_SUP,"unknown option"); 19349b5e25fSSatish Balay } 19449b5e25fSSatish Balay PetscFunctionReturn(0); 19549b5e25fSSatish Balay } 19649b5e25fSSatish Balay 1974a2ae208SSatish Balay #undef __FUNCT__ 1984a2ae208SSatish Balay #define __FUNCT__ "MatGetRow_SeqSBAIJ" 19913f74950SBarry Smith PetscErrorCode MatGetRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *ncols,PetscInt **cols,PetscScalar **v) 20049b5e25fSSatish Balay { 20149b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 2026849ba73SBarry Smith PetscErrorCode ierr; 20313f74950SBarry Smith PetscInt itmp,i,j,k,M,*ai,*aj,bs,bn,bp,*cols_i,bs2; 20449b5e25fSSatish Balay MatScalar *aa,*aa_i; 20587828ca2SBarry Smith PetscScalar *v_i; 20649b5e25fSSatish Balay 20749b5e25fSSatish Balay PetscFunctionBegin; 208521d7252SBarry Smith bs = A->bs; 20949b5e25fSSatish Balay ai = a->i; 21049b5e25fSSatish Balay aj = a->j; 21149b5e25fSSatish Balay aa = a->a; 21249b5e25fSSatish Balay bs2 = a->bs2; 21349b5e25fSSatish Balay 21477431f27SBarry Smith if (row < 0 || row >= A->m) SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE, "Row %D out of range", row); 21549b5e25fSSatish Balay 21649b5e25fSSatish Balay bn = row/bs; /* Block number */ 21749b5e25fSSatish Balay bp = row % bs; /* Block position */ 21849b5e25fSSatish Balay M = ai[bn+1] - ai[bn]; 21949b5e25fSSatish Balay *ncols = bs*M; 22049b5e25fSSatish Balay 22149b5e25fSSatish Balay if (v) { 22249b5e25fSSatish Balay *v = 0; 22349b5e25fSSatish Balay if (*ncols) { 22487828ca2SBarry Smith ierr = PetscMalloc((*ncols+row)*sizeof(PetscScalar),v);CHKERRQ(ierr); 22549b5e25fSSatish Balay for (i=0; i<M; i++) { /* for each block in the block row */ 22649b5e25fSSatish Balay v_i = *v + i*bs; 22749b5e25fSSatish Balay aa_i = aa + bs2*(ai[bn] + i); 22849b5e25fSSatish Balay for (j=bp,k=0; j<bs2; j+=bs,k++) {v_i[k] = aa_i[j];} 22949b5e25fSSatish Balay } 23049b5e25fSSatish Balay } 23149b5e25fSSatish Balay } 23249b5e25fSSatish Balay 23349b5e25fSSatish Balay if (cols) { 23449b5e25fSSatish Balay *cols = 0; 23549b5e25fSSatish Balay if (*ncols) { 23613f74950SBarry Smith ierr = PetscMalloc((*ncols+row)*sizeof(PetscInt),cols);CHKERRQ(ierr); 23749b5e25fSSatish Balay for (i=0; i<M; i++) { /* for each block in the block row */ 23849b5e25fSSatish Balay cols_i = *cols + i*bs; 23949b5e25fSSatish Balay itmp = bs*aj[ai[bn] + i]; 24049b5e25fSSatish Balay for (j=0; j<bs; j++) {cols_i[j] = itmp++;} 24149b5e25fSSatish Balay } 24249b5e25fSSatish Balay } 24349b5e25fSSatish Balay } 24449b5e25fSSatish Balay 24549b5e25fSSatish Balay /*search column A(0:row-1,row) (=A(row,0:row-1)). Could be expensive! */ 2465ddb2528SHong Zhang /* this segment is currently removed, so only entries in the upper triangle are obtained */ 2475ddb2528SHong Zhang #ifdef column_search 24849b5e25fSSatish Balay v_i = *v + M*bs; 24949b5e25fSSatish Balay cols_i = *cols + M*bs; 25049b5e25fSSatish Balay for (i=0; i<bn; i++){ /* for each block row */ 25149b5e25fSSatish Balay M = ai[i+1] - ai[i]; 25249b5e25fSSatish Balay for (j=0; j<M; j++){ 25349b5e25fSSatish Balay itmp = aj[ai[i] + j]; /* block column value */ 25449b5e25fSSatish Balay if (itmp == bn){ 25549b5e25fSSatish Balay aa_i = aa + bs2*(ai[i] + j) + bs*bp; 25649b5e25fSSatish Balay for (k=0; k<bs; k++) { 25749b5e25fSSatish Balay *cols_i++ = i*bs+k; 25849b5e25fSSatish Balay *v_i++ = aa_i[k]; 25949b5e25fSSatish Balay } 26049b5e25fSSatish Balay *ncols += bs; 26149b5e25fSSatish Balay break; 26249b5e25fSSatish Balay } 26349b5e25fSSatish Balay } 26449b5e25fSSatish Balay } 2655ddb2528SHong Zhang #endif 26649b5e25fSSatish Balay PetscFunctionReturn(0); 26749b5e25fSSatish Balay } 26849b5e25fSSatish Balay 2694a2ae208SSatish Balay #undef __FUNCT__ 2704a2ae208SSatish Balay #define __FUNCT__ "MatRestoreRow_SeqSBAIJ" 27113f74950SBarry Smith PetscErrorCode MatRestoreRow_SeqSBAIJ(Mat A,PetscInt row,PetscInt *nz,PetscInt **idx,PetscScalar **v) 27249b5e25fSSatish Balay { 273dfbe8321SBarry Smith PetscErrorCode ierr; 27449b5e25fSSatish Balay 27549b5e25fSSatish Balay PetscFunctionBegin; 27649b5e25fSSatish Balay if (idx) {if (*idx) {ierr = PetscFree(*idx);CHKERRQ(ierr);}} 27749b5e25fSSatish Balay if (v) {if (*v) {ierr = PetscFree(*v);CHKERRQ(ierr);}} 27849b5e25fSSatish Balay PetscFunctionReturn(0); 27949b5e25fSSatish Balay } 28049b5e25fSSatish Balay 2814a2ae208SSatish Balay #undef __FUNCT__ 2824a2ae208SSatish Balay #define __FUNCT__ "MatTranspose_SeqSBAIJ" 283dfbe8321SBarry Smith PetscErrorCode MatTranspose_SeqSBAIJ(Mat A,Mat *B) 28449b5e25fSSatish Balay { 285dfbe8321SBarry Smith PetscErrorCode ierr; 28649b5e25fSSatish Balay PetscFunctionBegin; 287999d9058SBarry Smith ierr = MatDuplicate(A,MAT_COPY_VALUES,B);CHKERRQ(ierr); 2888115998fSBarry Smith PetscFunctionReturn(0); 28949b5e25fSSatish Balay } 29049b5e25fSSatish Balay 2914a2ae208SSatish Balay #undef __FUNCT__ 2924a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_ASCII" 2936849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_ASCII(Mat A,PetscViewer viewer) 29449b5e25fSSatish Balay { 29549b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 296dfbe8321SBarry Smith PetscErrorCode ierr; 297521d7252SBarry Smith PetscInt i,j,bs = A->bs,k,l,bs2=a->bs2; 298fb9695e5SSatish Balay char *name; 299f3ef73ceSBarry Smith PetscViewerFormat format; 30049b5e25fSSatish Balay 30149b5e25fSSatish Balay PetscFunctionBegin; 30280fe4e49SBarry Smith ierr = PetscObjectGetName((PetscObject)A,&name);CHKERRQ(ierr); 303b0a32e0cSBarry Smith ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr); 304456192e2SBarry Smith if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) { 30577431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," block size is %D\n",bs);CHKERRQ(ierr); 306fb9695e5SSatish Balay } else if (format == PETSC_VIEWER_ASCII_MATLAB) { 30729bbc08cSBarry Smith SETERRQ(PETSC_ERR_SUP,"Matlab format not supported"); 308fb9695e5SSatish Balay } else if (format == PETSC_VIEWER_ASCII_COMMON) { 309b0a32e0cSBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_NO);CHKERRQ(ierr); 31049b5e25fSSatish Balay for (i=0; i<a->mbs; i++) { 31149b5e25fSSatish Balay for (j=0; j<bs; j++) { 31277431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr); 31349b5e25fSSatish Balay for (k=a->i[i]; k<a->i[i+1]; k++) { 31449b5e25fSSatish Balay for (l=0; l<bs; l++) { 31549b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX) 31649b5e25fSSatish Balay if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) { 31777431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k]+l, 31849b5e25fSSatish Balay PetscRealPart(a->a[bs2*k + l*bs + j]),PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 31949b5e25fSSatish Balay } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0 && PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) { 32077431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k]+l, 32149b5e25fSSatish Balay PetscRealPart(a->a[bs2*k + l*bs + j]),-PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 32249b5e25fSSatish Balay } else if (PetscRealPart(a->a[bs2*k + l*bs + j]) != 0.0) { 32377431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 32449b5e25fSSatish Balay } 32549b5e25fSSatish Balay #else 32649b5e25fSSatish Balay if (a->a[bs2*k + l*bs + j] != 0.0) { 32777431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,a->a[bs2*k + l*bs + j]);CHKERRQ(ierr); 32849b5e25fSSatish Balay } 32949b5e25fSSatish Balay #endif 33049b5e25fSSatish Balay } 33149b5e25fSSatish Balay } 332b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 33349b5e25fSSatish Balay } 33449b5e25fSSatish Balay } 335b0a32e0cSBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_YES);CHKERRQ(ierr); 33649b5e25fSSatish Balay } else { 337b0a32e0cSBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_NO);CHKERRQ(ierr); 33849b5e25fSSatish Balay for (i=0; i<a->mbs; i++) { 33949b5e25fSSatish Balay for (j=0; j<bs; j++) { 34077431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i*bs+j);CHKERRQ(ierr); 34149b5e25fSSatish Balay for (k=a->i[i]; k<a->i[i+1]; k++) { 34249b5e25fSSatish Balay for (l=0; l<bs; l++) { 34349b5e25fSSatish Balay #if defined(PETSC_USE_COMPLEX) 34449b5e25fSSatish Balay if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) > 0.0) { 34577431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",bs*a->j[k]+l, 34649b5e25fSSatish Balay PetscRealPart(a->a[bs2*k + l*bs + j]),PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 34749b5e25fSSatish Balay } else if (PetscImaginaryPart(a->a[bs2*k + l*bs + j]) < 0.0) { 34877431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g - %g i) ",bs*a->j[k]+l, 34949b5e25fSSatish Balay PetscRealPart(a->a[bs2*k + l*bs + j]),-PetscImaginaryPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 35049b5e25fSSatish Balay } else { 35177431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",bs*a->j[k]+l,PetscRealPart(a->a[bs2*k + l*bs + j]));CHKERRQ(ierr); 35249b5e25fSSatish Balay } 35349b5e25fSSatish Balay #else 35477431f27SBarry Smith ierr = PetscViewerASCIIPrintf(viewer," %D %g ",bs*a->j[k]+l,a->a[bs2*k + l*bs + j]);CHKERRQ(ierr); 35549b5e25fSSatish Balay #endif 35649b5e25fSSatish Balay } 35749b5e25fSSatish Balay } 358b0a32e0cSBarry Smith ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr); 35949b5e25fSSatish Balay } 36049b5e25fSSatish Balay } 361b0a32e0cSBarry Smith ierr = PetscViewerASCIIUseTabs(viewer,PETSC_YES);CHKERRQ(ierr); 36249b5e25fSSatish Balay } 363b0a32e0cSBarry Smith ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 36449b5e25fSSatish Balay PetscFunctionReturn(0); 36549b5e25fSSatish Balay } 36649b5e25fSSatish Balay 3674a2ae208SSatish Balay #undef __FUNCT__ 3684a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_Draw_Zoom" 3696849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw_Zoom(PetscDraw draw,void *Aa) 37049b5e25fSSatish Balay { 37149b5e25fSSatish Balay Mat A = (Mat) Aa; 37249b5e25fSSatish Balay Mat_SeqSBAIJ *a=(Mat_SeqSBAIJ*)A->data; 3736849ba73SBarry Smith PetscErrorCode ierr; 37413f74950SBarry Smith PetscInt row,i,j,k,l,mbs=a->mbs,color,bs=A->bs,bs2=a->bs2; 37513f74950SBarry Smith PetscMPIInt rank; 37649b5e25fSSatish Balay PetscReal xl,yl,xr,yr,x_l,x_r,y_l,y_r; 37749b5e25fSSatish Balay MatScalar *aa; 37849b5e25fSSatish Balay MPI_Comm comm; 379b0a32e0cSBarry Smith PetscViewer viewer; 38049b5e25fSSatish Balay 38149b5e25fSSatish Balay PetscFunctionBegin; 38249b5e25fSSatish Balay /* 38349b5e25fSSatish Balay This is nasty. If this is called from an originally parallel matrix 38449b5e25fSSatish Balay then all processes call this,but only the first has the matrix so the 38549b5e25fSSatish Balay rest should return immediately. 38649b5e25fSSatish Balay */ 38749b5e25fSSatish Balay ierr = PetscObjectGetComm((PetscObject)draw,&comm);CHKERRQ(ierr); 38849b5e25fSSatish Balay ierr = MPI_Comm_rank(comm,&rank);CHKERRQ(ierr); 38949b5e25fSSatish Balay if (rank) PetscFunctionReturn(0); 39049b5e25fSSatish Balay 39149b5e25fSSatish Balay ierr = PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer);CHKERRQ(ierr); 39249b5e25fSSatish Balay 393b0a32e0cSBarry Smith ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr); 394b0a32e0cSBarry Smith PetscDrawString(draw, .3*(xl+xr), .3*(yl+yr), PETSC_DRAW_BLACK, "symmetric"); 39549b5e25fSSatish Balay 39649b5e25fSSatish Balay /* loop over matrix elements drawing boxes */ 397b0a32e0cSBarry Smith color = PETSC_DRAW_BLUE; 39849b5e25fSSatish Balay for (i=0,row=0; i<mbs; i++,row+=bs) { 39949b5e25fSSatish Balay for (j=a->i[i]; j<a->i[i+1]; j++) { 400c464158bSHong Zhang y_l = A->m - row - 1.0; y_r = y_l + 1.0; 40149b5e25fSSatish Balay x_l = a->j[j]*bs; x_r = x_l + 1.0; 40249b5e25fSSatish Balay aa = a->a + j*bs2; 40349b5e25fSSatish Balay for (k=0; k<bs; k++) { 40449b5e25fSSatish Balay for (l=0; l<bs; l++) { 40549b5e25fSSatish Balay if (PetscRealPart(*aa++) >= 0.) continue; 406b0a32e0cSBarry Smith ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr); 40749b5e25fSSatish Balay } 40849b5e25fSSatish Balay } 40949b5e25fSSatish Balay } 41049b5e25fSSatish Balay } 411b0a32e0cSBarry Smith color = PETSC_DRAW_CYAN; 41249b5e25fSSatish Balay for (i=0,row=0; i<mbs; i++,row+=bs) { 41349b5e25fSSatish Balay for (j=a->i[i]; j<a->i[i+1]; j++) { 414c464158bSHong Zhang y_l = A->m - row - 1.0; y_r = y_l + 1.0; 41549b5e25fSSatish Balay x_l = a->j[j]*bs; x_r = x_l + 1.0; 41649b5e25fSSatish Balay aa = a->a + j*bs2; 41749b5e25fSSatish Balay for (k=0; k<bs; k++) { 41849b5e25fSSatish Balay for (l=0; l<bs; l++) { 41949b5e25fSSatish Balay if (PetscRealPart(*aa++) != 0.) continue; 420b0a32e0cSBarry Smith ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr); 42149b5e25fSSatish Balay } 42249b5e25fSSatish Balay } 42349b5e25fSSatish Balay } 42449b5e25fSSatish Balay } 42549b5e25fSSatish Balay 426b0a32e0cSBarry Smith color = PETSC_DRAW_RED; 42749b5e25fSSatish Balay for (i=0,row=0; i<mbs; i++,row+=bs) { 42849b5e25fSSatish Balay for (j=a->i[i]; j<a->i[i+1]; j++) { 429c464158bSHong Zhang y_l = A->m - row - 1.0; y_r = y_l + 1.0; 43049b5e25fSSatish Balay x_l = a->j[j]*bs; x_r = x_l + 1.0; 43149b5e25fSSatish Balay aa = a->a + j*bs2; 43249b5e25fSSatish Balay for (k=0; k<bs; k++) { 43349b5e25fSSatish Balay for (l=0; l<bs; l++) { 43449b5e25fSSatish Balay if (PetscRealPart(*aa++) <= 0.) continue; 435b0a32e0cSBarry Smith ierr = PetscDrawRectangle(draw,x_l+k,y_l-l,x_r+k,y_r-l,color,color,color,color);CHKERRQ(ierr); 43649b5e25fSSatish Balay } 43749b5e25fSSatish Balay } 43849b5e25fSSatish Balay } 43949b5e25fSSatish Balay } 44049b5e25fSSatish Balay PetscFunctionReturn(0); 44149b5e25fSSatish Balay } 44249b5e25fSSatish Balay 4434a2ae208SSatish Balay #undef __FUNCT__ 4444a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ_Draw" 4456849ba73SBarry Smith static PetscErrorCode MatView_SeqSBAIJ_Draw(Mat A,PetscViewer viewer) 44649b5e25fSSatish Balay { 447dfbe8321SBarry Smith PetscErrorCode ierr; 44849b5e25fSSatish Balay PetscReal xl,yl,xr,yr,w,h; 449b0a32e0cSBarry Smith PetscDraw draw; 45049b5e25fSSatish Balay PetscTruth isnull; 45149b5e25fSSatish Balay 45249b5e25fSSatish Balay PetscFunctionBegin; 45349b5e25fSSatish Balay 454b0a32e0cSBarry Smith ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr); 455b0a32e0cSBarry Smith ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr); if (isnull) PetscFunctionReturn(0); 45649b5e25fSSatish Balay 45749b5e25fSSatish Balay ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer);CHKERRQ(ierr); 458c464158bSHong Zhang xr = A->m; yr = A->m; h = yr/10.0; w = xr/10.0; 45949b5e25fSSatish Balay xr += w; yr += h; xl = -w; yl = -h; 460b0a32e0cSBarry Smith ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr); 461b0a32e0cSBarry Smith ierr = PetscDrawZoom(draw,MatView_SeqSBAIJ_Draw_Zoom,A);CHKERRQ(ierr); 46249b5e25fSSatish Balay ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",PETSC_NULL);CHKERRQ(ierr); 46349b5e25fSSatish Balay PetscFunctionReturn(0); 46449b5e25fSSatish Balay } 46549b5e25fSSatish Balay 4664a2ae208SSatish Balay #undef __FUNCT__ 4674a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqSBAIJ" 468dfbe8321SBarry Smith PetscErrorCode MatView_SeqSBAIJ(Mat A,PetscViewer viewer) 46949b5e25fSSatish Balay { 470dfbe8321SBarry Smith PetscErrorCode ierr; 47132077d6dSBarry Smith PetscTruth iascii,isdraw; 47249b5e25fSSatish Balay 47349b5e25fSSatish Balay PetscFunctionBegin; 47432077d6dSBarry Smith ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr); 475fb9695e5SSatish Balay ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_DRAW,&isdraw);CHKERRQ(ierr); 47632077d6dSBarry Smith if (iascii){ 47749b5e25fSSatish Balay ierr = MatView_SeqSBAIJ_ASCII(A,viewer);CHKERRQ(ierr); 47849b5e25fSSatish Balay } else if (isdraw) { 47949b5e25fSSatish Balay ierr = MatView_SeqSBAIJ_Draw(A,viewer);CHKERRQ(ierr); 48049b5e25fSSatish Balay } else { 481a5e6ed63SBarry Smith Mat B; 482ceb03754SKris Buschelman ierr = MatConvert(A,MATSEQAIJ,MAT_INITIAL_MATRIX,&B);CHKERRQ(ierr); 483a5e6ed63SBarry Smith ierr = MatView(B,viewer);CHKERRQ(ierr); 484a5e6ed63SBarry Smith ierr = MatDestroy(B);CHKERRQ(ierr); 48549b5e25fSSatish Balay } 48649b5e25fSSatish Balay PetscFunctionReturn(0); 48749b5e25fSSatish Balay } 48849b5e25fSSatish Balay 48949b5e25fSSatish Balay 4904a2ae208SSatish Balay #undef __FUNCT__ 4914a2ae208SSatish Balay #define __FUNCT__ "MatGetValues_SeqSBAIJ" 49213f74950SBarry Smith PetscErrorCode MatGetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],PetscScalar v[]) 49349b5e25fSSatish Balay { 494045c9aa0SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 49513f74950SBarry Smith PetscInt *rp,k,low,high,t,row,nrow,i,col,l,*aj = a->j; 49613f74950SBarry Smith PetscInt *ai = a->i,*ailen = a->ilen; 49713f74950SBarry Smith PetscInt brow,bcol,ridx,cidx,bs=A->bs,bs2=a->bs2; 49849b5e25fSSatish Balay MatScalar *ap,*aa = a->a,zero = 0.0; 49949b5e25fSSatish Balay 50049b5e25fSSatish Balay PetscFunctionBegin; 50149b5e25fSSatish Balay for (k=0; k<m; k++) { /* loop over rows */ 50249b5e25fSSatish Balay row = im[k]; brow = row/bs; 50377431f27SBarry Smith if (row < 0) SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"Negative row: %D",row); 50477431f27SBarry Smith if (row >= A->m) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,A->m-1); 50549b5e25fSSatish Balay rp = aj + ai[brow] ; ap = aa + bs2*ai[brow] ; 50649b5e25fSSatish Balay nrow = ailen[brow]; 50749b5e25fSSatish Balay for (l=0; l<n; l++) { /* loop over columns */ 50877431f27SBarry Smith if (in[l] < 0) SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"Negative column: %D",in[l]); 50977431f27SBarry Smith if (in[l] >= A->n) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",in[l],A->n-1); 51049b5e25fSSatish Balay col = in[l] ; 51149b5e25fSSatish Balay bcol = col/bs; 51249b5e25fSSatish Balay cidx = col%bs; 51349b5e25fSSatish Balay ridx = row%bs; 51449b5e25fSSatish Balay high = nrow; 51549b5e25fSSatish Balay low = 0; /* assume unsorted */ 51649b5e25fSSatish Balay while (high-low > 5) { 51749b5e25fSSatish Balay t = (low+high)/2; 51849b5e25fSSatish Balay if (rp[t] > bcol) high = t; 51949b5e25fSSatish Balay else low = t; 52049b5e25fSSatish Balay } 52149b5e25fSSatish Balay for (i=low; i<high; i++) { 52249b5e25fSSatish Balay if (rp[i] > bcol) break; 52349b5e25fSSatish Balay if (rp[i] == bcol) { 52449b5e25fSSatish Balay *v++ = ap[bs2*i+bs*cidx+ridx]; 52549b5e25fSSatish Balay goto finished; 52649b5e25fSSatish Balay } 52749b5e25fSSatish Balay } 52849b5e25fSSatish Balay *v++ = zero; 52949b5e25fSSatish Balay finished:; 53049b5e25fSSatish Balay } 53149b5e25fSSatish Balay } 53249b5e25fSSatish Balay PetscFunctionReturn(0); 53349b5e25fSSatish Balay } 53449b5e25fSSatish Balay 53549b5e25fSSatish Balay 5364a2ae208SSatish Balay #undef __FUNCT__ 5374a2ae208SSatish Balay #define __FUNCT__ "MatSetValuesBlocked_SeqSBAIJ" 53813f74950SBarry Smith PetscErrorCode MatSetValuesBlocked_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is) 53949b5e25fSSatish Balay { 5400880e062SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 5416849ba73SBarry Smith PetscErrorCode ierr; 542e2ee6c50SBarry Smith PetscInt *rp,k,low,high,t,ii,jj,row,nrow,i,col,l,rmax,N,lastcol = -1; 54313f74950SBarry Smith PetscInt *imax=a->imax,*ai=a->i,*ailen=a->ilen; 54413f74950SBarry Smith PetscInt *aj=a->j,nonew=a->nonew,bs2=a->bs2,bs=A->bs,stepval; 5450880e062SHong Zhang PetscTruth roworiented=a->roworiented; 546f15d580aSBarry Smith const MatScalar *value = v; 547f15d580aSBarry Smith MatScalar *ap,*aa = a->a,*bap; 5480880e062SHong Zhang 54949b5e25fSSatish Balay PetscFunctionBegin; 5500880e062SHong Zhang if (roworiented) { 5510880e062SHong Zhang stepval = (n-1)*bs; 5520880e062SHong Zhang } else { 5530880e062SHong Zhang stepval = (m-1)*bs; 5540880e062SHong Zhang } 5550880e062SHong Zhang for (k=0; k<m; k++) { /* loop over added rows */ 5560880e062SHong Zhang row = im[k]; 5570880e062SHong Zhang if (row < 0) continue; 5582515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 55977431f27SBarry Smith if (row >= a->mbs) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,a->mbs-1); 5600880e062SHong Zhang #endif 5610880e062SHong Zhang rp = aj + ai[row]; 5620880e062SHong Zhang ap = aa + bs2*ai[row]; 5630880e062SHong Zhang rmax = imax[row]; 5640880e062SHong Zhang nrow = ailen[row]; 5650880e062SHong Zhang low = 0; 566818f2c47SBarry Smith high = nrow; 5670880e062SHong Zhang for (l=0; l<n; l++) { /* loop over added columns */ 5680880e062SHong Zhang if (in[l] < 0) continue; 5690880e062SHong Zhang col = in[l]; 5702515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 57177431f27SBarry Smith if (col >= a->nbs) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",col,a->nbs-1); 572b1823623SSatish Balay #endif 5730880e062SHong Zhang if (col < row) continue; /* ignore lower triangular block */ 5740880e062SHong Zhang if (roworiented) { 5750880e062SHong Zhang value = v + k*(stepval+bs)*bs + l*bs; 5760880e062SHong Zhang } else { 5770880e062SHong Zhang value = v + l*(stepval+bs)*bs + k*bs; 5780880e062SHong Zhang } 579818f2c47SBarry Smith if (col < lastcol) low = 0; else high = nrow; 580e2ee6c50SBarry Smith lastcol = col; 5810880e062SHong Zhang while (high-low > 7) { 5820880e062SHong Zhang t = (low+high)/2; 5830880e062SHong Zhang if (rp[t] > col) high = t; 5840880e062SHong Zhang else low = t; 5850880e062SHong Zhang } 5860880e062SHong Zhang for (i=low; i<high; i++) { 5870880e062SHong Zhang if (rp[i] > col) break; 5880880e062SHong Zhang if (rp[i] == col) { 5890880e062SHong Zhang bap = ap + bs2*i; 5900880e062SHong Zhang if (roworiented) { 5910880e062SHong Zhang if (is == ADD_VALUES) { 5920880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 5930880e062SHong Zhang for (jj=ii; jj<bs2; jj+=bs) { 5940880e062SHong Zhang bap[jj] += *value++; 5950880e062SHong Zhang } 5960880e062SHong Zhang } 5970880e062SHong Zhang } else { 5980880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 5990880e062SHong Zhang for (jj=ii; jj<bs2; jj+=bs) { 6000880e062SHong Zhang bap[jj] = *value++; 6010880e062SHong Zhang } 6020880e062SHong Zhang } 6030880e062SHong Zhang } 6040880e062SHong Zhang } else { 6050880e062SHong Zhang if (is == ADD_VALUES) { 6060880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 6070880e062SHong Zhang for (jj=0; jj<bs; jj++) { 6080880e062SHong Zhang *bap++ += *value++; 6090880e062SHong Zhang } 6100880e062SHong Zhang } 6110880e062SHong Zhang } else { 6120880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 6130880e062SHong Zhang for (jj=0; jj<bs; jj++) { 6140880e062SHong Zhang *bap++ = *value++; 6150880e062SHong Zhang } 6160880e062SHong Zhang } 6170880e062SHong Zhang } 6180880e062SHong Zhang } 6190880e062SHong Zhang goto noinsert2; 6200880e062SHong Zhang } 6210880e062SHong Zhang } 6220880e062SHong Zhang if (nonew == 1) goto noinsert2; 62377431f27SBarry Smith else if (nonew == -1) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col); 6240880e062SHong Zhang if (nrow >= rmax) { 6250880e062SHong Zhang /* there is no extra room in row, therefore enlarge */ 62613f74950SBarry Smith PetscInt new_nz = ai[a->mbs] + CHUNKSIZE,len,*new_i,*new_j; 6270880e062SHong Zhang MatScalar *new_a; 6280880e062SHong Zhang 62977431f27SBarry Smith if (nonew == -2) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col); 6300880e062SHong Zhang 6310880e062SHong Zhang /* malloc new storage space */ 63213f74950SBarry Smith len = new_nz*(sizeof(PetscInt)+bs2*sizeof(MatScalar))+(a->mbs+1)*sizeof(PetscInt); 6330880e062SHong Zhang ierr = PetscMalloc(len,&new_a);CHKERRQ(ierr); 63413f74950SBarry Smith new_j = (PetscInt*)(new_a + bs2*new_nz); 6350880e062SHong Zhang new_i = new_j + new_nz; 6360880e062SHong Zhang 6370880e062SHong Zhang /* copy over old data into new slots */ 6380880e062SHong Zhang for (ii=0; ii<row+1; ii++) {new_i[ii] = ai[ii];} 6390880e062SHong Zhang for (ii=row+1; ii<a->mbs+1; ii++) {new_i[ii] = ai[ii]+CHUNKSIZE;} 64013f74950SBarry Smith ierr = PetscMemcpy(new_j,aj,(ai[row]+nrow)*sizeof(PetscInt));CHKERRQ(ierr); 6410880e062SHong Zhang len = (new_nz - CHUNKSIZE - ai[row] - nrow); 64213f74950SBarry Smith ierr = PetscMemcpy(new_j+ai[row]+nrow+CHUNKSIZE,aj+ai[row]+nrow,len*sizeof(PetscInt));CHKERRQ(ierr); 6430880e062SHong Zhang ierr = PetscMemcpy(new_a,aa,(ai[row]+nrow)*bs2*sizeof(MatScalar));CHKERRQ(ierr); 6440880e062SHong Zhang ierr = PetscMemzero(new_a+bs2*(ai[row]+nrow),bs2*CHUNKSIZE*sizeof(MatScalar));CHKERRQ(ierr); 6450880e062SHong Zhang ierr = PetscMemcpy(new_a+bs2*(ai[row]+nrow+CHUNKSIZE),aa+bs2*(ai[row]+nrow),bs2*len*sizeof(MatScalar));CHKERRQ(ierr); 6460880e062SHong Zhang /* free up old matrix storage */ 6470880e062SHong Zhang ierr = PetscFree(a->a);CHKERRQ(ierr); 6480880e062SHong Zhang if (!a->singlemalloc) { 6490880e062SHong Zhang ierr = PetscFree(a->i);CHKERRQ(ierr); 6500880e062SHong Zhang ierr = PetscFree(a->j);CHKERRQ(ierr); 6510880e062SHong Zhang } 6520880e062SHong Zhang aa = a->a = new_a; ai = a->i = new_i; aj = a->j = new_j; 6530880e062SHong Zhang a->singlemalloc = PETSC_TRUE; 6540880e062SHong Zhang 6550880e062SHong Zhang rp = aj + ai[row]; ap = aa + bs2*ai[row]; 6560880e062SHong Zhang rmax = imax[row] = imax[row] + CHUNKSIZE; 65752e6d16bSBarry Smith ierr = PetscLogObjectMemory(A,CHUNKSIZE*(sizeof(PetscInt) + bs2*sizeof(MatScalar)));CHKERRQ(ierr); 6586c6c5352SBarry Smith a->maxnz += bs2*CHUNKSIZE; 6590880e062SHong Zhang a->reallocs++; 6606c6c5352SBarry Smith a->nz++; 6610880e062SHong Zhang } 6620880e062SHong Zhang N = nrow++ - 1; 6630880e062SHong Zhang /* shift up all the later entries in this row */ 6640880e062SHong Zhang for (ii=N; ii>=i; ii--) { 6650880e062SHong Zhang rp[ii+1] = rp[ii]; 6660880e062SHong Zhang ierr = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr); 6670880e062SHong Zhang } 6680880e062SHong Zhang if (N >= i) { 6690880e062SHong Zhang ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr); 6700880e062SHong Zhang } 6710880e062SHong Zhang rp[i] = col; 6720880e062SHong Zhang bap = ap + bs2*i; 6730880e062SHong Zhang if (roworiented) { 6740880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 6750880e062SHong Zhang for (jj=ii; jj<bs2; jj+=bs) { 6760880e062SHong Zhang bap[jj] = *value++; 6770880e062SHong Zhang } 6780880e062SHong Zhang } 6790880e062SHong Zhang } else { 6800880e062SHong Zhang for (ii=0; ii<bs; ii++,value+=stepval) { 6810880e062SHong Zhang for (jj=0; jj<bs; jj++) { 6820880e062SHong Zhang *bap++ = *value++; 6830880e062SHong Zhang } 6840880e062SHong Zhang } 6850880e062SHong Zhang } 6860880e062SHong Zhang noinsert2:; 6870880e062SHong Zhang low = i; 6880880e062SHong Zhang } 6890880e062SHong Zhang ailen[row] = nrow; 6900880e062SHong Zhang } 6910880e062SHong Zhang PetscFunctionReturn(0); 69249b5e25fSSatish Balay } 69349b5e25fSSatish Balay 6944a2ae208SSatish Balay #undef __FUNCT__ 6954a2ae208SSatish Balay #define __FUNCT__ "MatAssemblyEnd_SeqSBAIJ" 696dfbe8321SBarry Smith PetscErrorCode MatAssemblyEnd_SeqSBAIJ(Mat A,MatAssemblyType mode) 69749b5e25fSSatish Balay { 69849b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 6996849ba73SBarry Smith PetscErrorCode ierr; 70013f74950SBarry Smith PetscInt fshift = 0,i,j,*ai = a->i,*aj = a->j,*imax = a->imax; 70113f74950SBarry Smith PetscInt m = A->m,*ip,N,*ailen = a->ilen; 70213f74950SBarry Smith PetscInt mbs = a->mbs,bs2 = a->bs2,rmax = 0; 70349b5e25fSSatish Balay MatScalar *aa = a->a,*ap; 70449b5e25fSSatish Balay 70549b5e25fSSatish Balay PetscFunctionBegin; 70649b5e25fSSatish Balay if (mode == MAT_FLUSH_ASSEMBLY) PetscFunctionReturn(0); 70749b5e25fSSatish Balay 70849b5e25fSSatish Balay if (m) rmax = ailen[0]; 70949b5e25fSSatish Balay for (i=1; i<mbs; i++) { 71049b5e25fSSatish Balay /* move each row back by the amount of empty slots (fshift) before it*/ 71149b5e25fSSatish Balay fshift += imax[i-1] - ailen[i-1]; 71249b5e25fSSatish Balay rmax = PetscMax(rmax,ailen[i]); 71349b5e25fSSatish Balay if (fshift) { 71449b5e25fSSatish Balay ip = aj + ai[i]; ap = aa + bs2*ai[i]; 71549b5e25fSSatish Balay N = ailen[i]; 71649b5e25fSSatish Balay for (j=0; j<N; j++) { 71749b5e25fSSatish Balay ip[j-fshift] = ip[j]; 71849b5e25fSSatish Balay ierr = PetscMemcpy(ap+(j-fshift)*bs2,ap+j*bs2,bs2*sizeof(MatScalar));CHKERRQ(ierr); 71949b5e25fSSatish Balay } 72049b5e25fSSatish Balay } 72149b5e25fSSatish Balay ai[i] = ai[i-1] + ailen[i-1]; 72249b5e25fSSatish Balay } 72349b5e25fSSatish Balay if (mbs) { 72449b5e25fSSatish Balay fshift += imax[mbs-1] - ailen[mbs-1]; 72549b5e25fSSatish Balay ai[mbs] = ai[mbs-1] + ailen[mbs-1]; 72649b5e25fSSatish Balay } 72749b5e25fSSatish Balay /* reset ilen and imax for each row */ 72849b5e25fSSatish Balay for (i=0; i<mbs; i++) { 72949b5e25fSSatish Balay ailen[i] = imax[i] = ai[i+1] - ai[i]; 73049b5e25fSSatish Balay } 7316c6c5352SBarry Smith a->nz = ai[mbs]; 73249b5e25fSSatish Balay 733b424e231SHong Zhang /* diagonals may have moved, reset it */ 734b424e231SHong Zhang if (a->diag) { 73513f74950SBarry Smith ierr = PetscMemcpy(a->diag,ai,(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr); 73649b5e25fSSatish Balay } 73777431f27SBarry Smith PetscLogInfo(A,"MatAssemblyEnd_SeqSBAIJ:Matrix size: %D X %D, block size %D; storage space: %D unneeded, %D used\n", 738521d7252SBarry Smith m,A->m,A->bs,fshift*bs2,a->nz*bs2); 73977431f27SBarry Smith PetscLogInfo(A,"MatAssemblyEnd_SeqSBAIJ:Number of mallocs during MatSetValues is %D\n", 74049b5e25fSSatish Balay a->reallocs); 74177431f27SBarry Smith PetscLogInfo(A,"MatAssemblyEnd_SeqSBAIJ:Most nonzeros blocks in any row is %D\n",rmax); 74249b5e25fSSatish Balay a->reallocs = 0; 74349b5e25fSSatish Balay A->info.nz_unneeded = (PetscReal)fshift*bs2; 74449b5e25fSSatish Balay PetscFunctionReturn(0); 74549b5e25fSSatish Balay } 74649b5e25fSSatish Balay 74749b5e25fSSatish Balay /* 74849b5e25fSSatish Balay This function returns an array of flags which indicate the locations of contiguous 74949b5e25fSSatish Balay blocks that should be zeroed. for eg: if bs = 3 and is = [0,1,2,3,5,6,7,8,9] 75049b5e25fSSatish Balay then the resulting sizes = [3,1,1,3,1] correspondig to sets [(0,1,2),(3),(5),(6,7,8),(9)] 75149b5e25fSSatish Balay Assume: sizes should be long enough to hold all the values. 75249b5e25fSSatish Balay */ 7534a2ae208SSatish Balay #undef __FUNCT__ 7544a2ae208SSatish Balay #define __FUNCT__ "MatZeroRows_SeqSBAIJ_Check_Blocks" 75513f74950SBarry Smith PetscErrorCode MatZeroRows_SeqSBAIJ_Check_Blocks(PetscInt idx[],PetscInt n,PetscInt bs,PetscInt sizes[], PetscInt *bs_max) 75649b5e25fSSatish Balay { 75713f74950SBarry Smith PetscInt i,j,k,row; 75849b5e25fSSatish Balay PetscTruth flg; 75949b5e25fSSatish Balay 76049b5e25fSSatish Balay PetscFunctionBegin; 76149b5e25fSSatish Balay for (i=0,j=0; i<n; j++) { 76249b5e25fSSatish Balay row = idx[i]; 76349b5e25fSSatish Balay if (row%bs!=0) { /* Not the begining of a block */ 76449b5e25fSSatish Balay sizes[j] = 1; 76549b5e25fSSatish Balay i++; 76649b5e25fSSatish Balay } else if (i+bs > n) { /* Beginning of a block, but complete block doesn't exist (at idx end) */ 76749b5e25fSSatish Balay sizes[j] = 1; /* Also makes sure atleast 'bs' values exist for next else */ 76849b5e25fSSatish Balay i++; 76949b5e25fSSatish Balay } else { /* Begining of the block, so check if the complete block exists */ 77049b5e25fSSatish Balay flg = PETSC_TRUE; 77149b5e25fSSatish Balay for (k=1; k<bs; k++) { 77249b5e25fSSatish Balay if (row+k != idx[i+k]) { /* break in the block */ 77349b5e25fSSatish Balay flg = PETSC_FALSE; 77449b5e25fSSatish Balay break; 77549b5e25fSSatish Balay } 77649b5e25fSSatish Balay } 777abc0a331SBarry Smith if (flg) { /* No break in the bs */ 77849b5e25fSSatish Balay sizes[j] = bs; 77949b5e25fSSatish Balay i+= bs; 78049b5e25fSSatish Balay } else { 78149b5e25fSSatish Balay sizes[j] = 1; 78249b5e25fSSatish Balay i++; 78349b5e25fSSatish Balay } 78449b5e25fSSatish Balay } 78549b5e25fSSatish Balay } 78649b5e25fSSatish Balay *bs_max = j; 78749b5e25fSSatish Balay PetscFunctionReturn(0); 78849b5e25fSSatish Balay } 78949b5e25fSSatish Balay 79049b5e25fSSatish Balay 79149b5e25fSSatish Balay /* Only add/insert a(i,j) with i<=j (blocks). 79249b5e25fSSatish Balay Any a(i,j) with i>j input by user is ingored. 79349b5e25fSSatish Balay */ 79449b5e25fSSatish Balay 7954a2ae208SSatish Balay #undef __FUNCT__ 7964a2ae208SSatish Balay #define __FUNCT__ "MatSetValues_SeqSBAIJ" 79713f74950SBarry Smith PetscErrorCode MatSetValues_SeqSBAIJ(Mat A,PetscInt m,const PetscInt im[],PetscInt n,const PetscInt in[],const PetscScalar v[],InsertMode is) 79849b5e25fSSatish Balay { 79949b5e25fSSatish Balay Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 8006849ba73SBarry Smith PetscErrorCode ierr; 801e2ee6c50SBarry Smith PetscInt *rp,k,low,high,t,ii,row,nrow,i,col,l,rmax,N,lastcol = -1; 80213f74950SBarry Smith PetscInt *imax=a->imax,*ai=a->i,*ailen=a->ilen,roworiented=a->roworiented; 80313f74950SBarry Smith PetscInt *aj=a->j,nonew=a->nonew,bs=A->bs,brow,bcol; 80413f74950SBarry Smith PetscInt ridx,cidx,bs2=a->bs2; 80549b5e25fSSatish Balay MatScalar *ap,value,*aa=a->a,*bap; 80649b5e25fSSatish Balay 80749b5e25fSSatish Balay PetscFunctionBegin; 80849b5e25fSSatish Balay for (k=0; k<m; k++) { /* loop over added rows */ 80949b5e25fSSatish Balay row = im[k]; /* row number */ 81049b5e25fSSatish Balay brow = row/bs; /* block row number */ 81149b5e25fSSatish Balay if (row < 0) continue; 8122515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 81377431f27SBarry Smith if (row >= A->m) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",row,A->m-1); 81449b5e25fSSatish Balay #endif 81549b5e25fSSatish Balay rp = aj + ai[brow]; /*ptr to beginning of column value of the row block*/ 81649b5e25fSSatish Balay ap = aa + bs2*ai[brow]; /*ptr to beginning of element value of the row block*/ 81749b5e25fSSatish Balay rmax = imax[brow]; /* maximum space allocated for this row */ 81849b5e25fSSatish Balay nrow = ailen[brow]; /* actual length of this row */ 81949b5e25fSSatish Balay low = 0; 82049b5e25fSSatish Balay 82149b5e25fSSatish Balay for (l=0; l<n; l++) { /* loop over added columns */ 82249b5e25fSSatish Balay if (in[l] < 0) continue; 8232515c552SBarry Smith #if defined(PETSC_USE_DEBUG) 82477431f27SBarry Smith if (in[l] >= A->m) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",in[l],A->m-1); 82549b5e25fSSatish Balay #endif 82649b5e25fSSatish Balay col = in[l]; 82749b5e25fSSatish Balay bcol = col/bs; /* block col number */ 82849b5e25fSSatish Balay 829*941593c8SHong Zhang if (brow > bcol) { 830*941593c8SHong Zhang if (a->ignore_ltriangular){ 831*941593c8SHong Zhang continue; /* ignore lower triangular values */ 832*941593c8SHong Zhang } else { 833*941593c8SHong 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)"); 834*941593c8SHong Zhang } 835*941593c8SHong Zhang } 836f4989cb3SHong Zhang 83749b5e25fSSatish Balay ridx = row % bs; cidx = col % bs; /*row and col index inside the block */ 8388549e402SHong Zhang if ((brow==bcol && ridx<=cidx) || (brow<bcol)){ 83949b5e25fSSatish Balay /* element value a(k,l) */ 84049b5e25fSSatish Balay if (roworiented) { 84149b5e25fSSatish Balay value = v[l + k*n]; 84249b5e25fSSatish Balay } else { 84349b5e25fSSatish Balay value = v[k + l*m]; 84449b5e25fSSatish Balay } 84549b5e25fSSatish Balay 84649b5e25fSSatish Balay /* move pointer bap to a(k,l) quickly and add/insert value */ 847e2ee6c50SBarry Smith if (col < lastcol) low = 0; high = nrow; 848e2ee6c50SBarry Smith lastcol = col; 84949b5e25fSSatish Balay while (high-low > 7) { 85049b5e25fSSatish Balay t = (low+high)/2; 85149b5e25fSSatish Balay if (rp[t] > bcol) high = t; 85249b5e25fSSatish Balay else low = t; 85349b5e25fSSatish Balay } 85449b5e25fSSatish Balay for (i=low; i<high; i++) { 85549b5e25fSSatish Balay if (rp[i] > bcol) break; 85649b5e25fSSatish Balay if (rp[i] == bcol) { 85749b5e25fSSatish Balay bap = ap + bs2*i + bs*cidx + ridx; 85849b5e25fSSatish Balay if (is == ADD_VALUES) *bap += value; 85949b5e25fSSatish Balay else *bap = value; 8608549e402SHong Zhang /* for diag block, add/insert its symmetric element a(cidx,ridx) */ 8618549e402SHong Zhang if (brow == bcol && ridx < cidx){ 8628549e402SHong Zhang bap = ap + bs2*i + bs*ridx + cidx; 8638549e402SHong Zhang if (is == ADD_VALUES) *bap += value; 8648549e402SHong Zhang else *bap = value; 8658549e402SHong Zhang } 86649b5e25fSSatish Balay goto noinsert1; 86749b5e25fSSatish Balay } 86849b5e25fSSatish Balay } 86949b5e25fSSatish Balay 87049b5e25fSSatish Balay if (nonew == 1) goto noinsert1; 87177431f27SBarry Smith else if (nonew == -1) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col); 87249b5e25fSSatish Balay if (nrow >= rmax) { 87349b5e25fSSatish Balay /* there is no extra room in row, therefore enlarge */ 87413f74950SBarry Smith PetscInt new_nz = ai[a->mbs] + CHUNKSIZE,len,*new_i,*new_j; 87549b5e25fSSatish Balay MatScalar *new_a; 87649b5e25fSSatish Balay 87777431f27SBarry Smith if (nonew == -2) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Inserting a new nonzero (%D, %D) in the matrix", row, col); 87849b5e25fSSatish Balay 87949b5e25fSSatish Balay /* Malloc new storage space */ 88013f74950SBarry Smith len = new_nz*(sizeof(PetscInt)+bs2*sizeof(MatScalar))+(a->mbs+1)*sizeof(PetscInt); 88182502324SSatish Balay ierr = PetscMalloc(len,&new_a);CHKERRQ(ierr); 88213f74950SBarry Smith new_j = (PetscInt*)(new_a + bs2*new_nz); 88349b5e25fSSatish Balay new_i = new_j + new_nz; 88449b5e25fSSatish Balay 88549b5e25fSSatish Balay /* copy over old data into new slots */ 88649b5e25fSSatish Balay for (ii=0; ii<brow+1; ii++) {new_i[ii] = ai[ii];} 88749b5e25fSSatish Balay for (ii=brow+1; ii<a->mbs+1; ii++) {new_i[ii] = ai[ii]+CHUNKSIZE;} 88813f74950SBarry Smith ierr = PetscMemcpy(new_j,aj,(ai[brow]+nrow)*sizeof(PetscInt));CHKERRQ(ierr); 88949b5e25fSSatish Balay len = (new_nz - CHUNKSIZE - ai[brow] - nrow); 89013f74950SBarry Smith ierr = PetscMemcpy(new_j+ai[brow]+nrow+CHUNKSIZE,aj+ai[brow]+nrow,len*sizeof(PetscInt));CHKERRQ(ierr); 89149b5e25fSSatish Balay ierr = PetscMemcpy(new_a,aa,(ai[brow]+nrow)*bs2*sizeof(MatScalar));CHKERRQ(ierr); 89249b5e25fSSatish Balay ierr = PetscMemzero(new_a+bs2*(ai[brow]+nrow),bs2*CHUNKSIZE*sizeof(MatScalar));CHKERRQ(ierr); 89349b5e25fSSatish Balay ierr = PetscMemcpy(new_a+bs2*(ai[brow]+nrow+CHUNKSIZE),aa+bs2*(ai[brow]+nrow),bs2*len*sizeof(MatScalar));CHKERRQ(ierr); 89449b5e25fSSatish Balay /* free up old matrix storage */ 89549b5e25fSSatish Balay ierr = PetscFree(a->a);CHKERRQ(ierr); 89649b5e25fSSatish Balay if (!a->singlemalloc) { 89749b5e25fSSatish Balay ierr = PetscFree(a->i);CHKERRQ(ierr); 89849b5e25fSSatish Balay ierr = PetscFree(a->j);CHKERRQ(ierr); 89949b5e25fSSatish Balay } 90049b5e25fSSatish Balay aa = a->a = new_a; ai = a->i = new_i; aj = a->j = new_j; 90149b5e25fSSatish Balay a->singlemalloc = PETSC_TRUE; 90249b5e25fSSatish Balay 90349b5e25fSSatish Balay rp = aj + ai[brow]; ap = aa + bs2*ai[brow]; 90449b5e25fSSatish Balay rmax = imax[brow] = imax[brow] + CHUNKSIZE; 90552e6d16bSBarry Smith ierr = PetscLogObjectMemory(A,CHUNKSIZE*(sizeof(PetscInt) + bs2*sizeof(MatScalar)));CHKERRQ(ierr); 9066c6c5352SBarry Smith a->maxnz += bs2*CHUNKSIZE; 90749b5e25fSSatish Balay a->reallocs++; 9086c6c5352SBarry Smith a->nz++; 90949b5e25fSSatish Balay } 91049b5e25fSSatish Balay 91149b5e25fSSatish Balay N = nrow++ - 1; 91249b5e25fSSatish Balay /* shift up all the later entries in this row */ 91349b5e25fSSatish Balay for (ii=N; ii>=i; ii--) { 91449b5e25fSSatish Balay rp[ii+1] = rp[ii]; 91549b5e25fSSatish Balay ierr = PetscMemcpy(ap+bs2*(ii+1),ap+bs2*(ii),bs2*sizeof(MatScalar));CHKERRQ(ierr); 91649b5e25fSSatish Balay } 91749b5e25fSSatish Balay if (N>=i) { 91849b5e25fSSatish Balay ierr = PetscMemzero(ap+bs2*i,bs2*sizeof(MatScalar));CHKERRQ(ierr); 91949b5e25fSSatish Balay } 92049b5e25fSSatish Balay rp[i] = bcol; 92149b5e25fSSatish Balay ap[bs2*i + bs*cidx + ridx] = value; 92249b5e25fSSatish Balay noinsert1:; 92349b5e25fSSatish Balay low = i; 9248549e402SHong Zhang } 92549b5e25fSSatish Balay } /* end of loop over added columns */ 92649b5e25fSSatish Balay ailen[brow] = nrow; 92749b5e25fSSatish Balay } /* end of loop over added rows */ 92849b5e25fSSatish Balay PetscFunctionReturn(0); 92949b5e25fSSatish Balay } 93049b5e25fSSatish Balay 9316849ba73SBarry Smith EXTERN PetscErrorCode MatCholeskyFactorSymbolic_SeqSBAIJ(Mat,IS,MatFactorInfo*,Mat*); 9326849ba73SBarry Smith EXTERN PetscErrorCode MatCholeskyFactor_SeqSBAIJ(Mat,IS,MatFactorInfo*); 93313f74950SBarry Smith EXTERN PetscErrorCode MatIncreaseOverlap_SeqSBAIJ(Mat,PetscInt,IS[],PetscInt); 93413f74950SBarry Smith EXTERN PetscErrorCode MatGetSubMatrix_SeqSBAIJ(Mat,IS,IS,PetscInt,MatReuse,Mat*); 93513f74950SBarry Smith EXTERN PetscErrorCode MatGetSubMatrices_SeqSBAIJ(Mat,PetscInt,const IS[],const IS[],MatReuse,Mat*[]); 9366849ba73SBarry Smith EXTERN PetscErrorCode MatScale_SeqSBAIJ(const PetscScalar*,Mat); 9376849ba73SBarry Smith EXTERN PetscErrorCode MatNorm_SeqSBAIJ(Mat,NormType,PetscReal *); 9386849ba73SBarry Smith EXTERN PetscErrorCode MatEqual_SeqSBAIJ(Mat,Mat,PetscTruth*); 9396849ba73SBarry Smith EXTERN PetscErrorCode MatGetDiagonal_SeqSBAIJ(Mat,Vec); 9406849ba73SBarry Smith EXTERN PetscErrorCode MatDiagonalScale_SeqSBAIJ(Mat,Vec,Vec); 9416849ba73SBarry Smith EXTERN PetscErrorCode MatGetInfo_SeqSBAIJ(Mat,MatInfoType,MatInfo *); 9426849ba73SBarry Smith EXTERN PetscErrorCode MatZeroEntries_SeqSBAIJ(Mat); 9436849ba73SBarry Smith EXTERN PetscErrorCode MatGetRowMax_SeqSBAIJ(Mat,Vec); 94449b5e25fSSatish Balay 9456849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_N(Mat,Vec,Vec); 9466849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_1(Mat,Vec,Vec); 9476849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_2(Mat,Vec,Vec); 9486849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_3(Mat,Vec,Vec); 9496849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_4(Mat,Vec,Vec); 9506849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_5(Mat,Vec,Vec); 9516849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_6(Mat,Vec,Vec); 9526849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_7(Mat,Vec,Vec); 95349b5e25fSSatish Balay 9546849ba73SBarry Smith EXTERN PetscErrorCode MatSolves_SeqSBAIJ_1(Mat,Vecs,Vecs); 955d59c15a7SBarry Smith 9566849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_1_NaturalOrdering(Mat,Vec,Vec); 9576849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_2_NaturalOrdering(Mat,Vec,Vec); 9586849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_3_NaturalOrdering(Mat,Vec,Vec); 9596849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_4_NaturalOrdering(Mat,Vec,Vec); 9606849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_5_NaturalOrdering(Mat,Vec,Vec); 9616849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_6_NaturalOrdering(Mat,Vec,Vec); 9626849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_7_NaturalOrdering(Mat,Vec,Vec); 9636849ba73SBarry Smith EXTERN PetscErrorCode MatSolve_SeqSBAIJ_N_NaturalOrdering(Mat,Vec,Vec); 96407f98182SSatish Balay 965af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_N(Mat,MatFactorInfo*,Mat*); 966af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_1(Mat,MatFactorInfo*,Mat*); 967af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_2(Mat,MatFactorInfo*,Mat*); 968af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_3(Mat,MatFactorInfo*,Mat*); 969af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_4(Mat,MatFactorInfo*,Mat*); 970af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_5(Mat,MatFactorInfo*,Mat*); 971af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_6(Mat,MatFactorInfo*,Mat*); 972af281ebdSHong Zhang EXTERN PetscErrorCode MatCholeskyFactorNumeric_SeqSBAIJ_7(Mat,MatFactorInfo*,Mat*); 97313f74950SBarry Smith EXTERN PetscErrorCode MatGetInertia_SeqSBAIJ(Mat,PetscInt*,PetscInt*,PetscInt*); 97449b5e25fSSatish Balay 9756849ba73SBarry Smith EXTERN PetscErrorCode MatMult_SeqSBAIJ_1(Mat,Vec,Vec); 9766849ba73SBarry Smith EXTERN PetscErrorCode MatMult_SeqSBAIJ_2(Mat,Vec,Vec); 9776849ba73SBarry Smith EXTERN PetscErrorCode MatMult_SeqSBAIJ_3(Mat,Vec,Vec); 9786849ba73SBarry Smith EXTERN PetscErrorCode MatMult_SeqSBAIJ_4(Mat,Vec,Vec); 9796849ba73SBarry Smith EXTERN PetscErrorCode MatMult_SeqSBAIJ_5(Mat,Vec,Vec); 9806849ba73SBarry Smith EXTERN PetscErrorCode MatMult_SeqSBAIJ_6(Mat,Vec,Vec); 9816849ba73SBarry Smith EXTERN PetscErrorCode MatMult_SeqSBAIJ_7(Mat,Vec,Vec); 9826849ba73SBarry Smith EXTERN PetscErrorCode MatMult_SeqSBAIJ_N(Mat,Vec,Vec); 98349b5e25fSSatish Balay 9846849ba73SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_1(Mat,Vec,Vec,Vec); 9856849ba73SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_2(Mat,Vec,Vec,Vec); 9866849ba73SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_3(Mat,Vec,Vec,Vec); 9876849ba73SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_4(Mat,Vec,Vec,Vec); 9886849ba73SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_5(Mat,Vec,Vec,Vec); 9896849ba73SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_6(Mat,Vec,Vec,Vec); 9906849ba73SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_7(Mat,Vec,Vec,Vec); 9916849ba73SBarry Smith EXTERN PetscErrorCode MatMultAdd_SeqSBAIJ_N(Mat,Vec,Vec,Vec); 99249b5e25fSSatish Balay 9934d101231SSatish Balay #ifdef HAVE_MatICCFactor 9946849ba73SBarry Smith /* modified from MatILUFactor_SeqSBAIJ, needs further work! */ 9954a2ae208SSatish Balay #undef __FUNCT__ 9964d101231SSatish Balay #define __FUNCT__ "MatICCFactor_SeqSBAIJ" 997dfbe8321SBarry Smith PetscErrorCode MatICCFactor_SeqSBAIJ(Mat inA,IS row,MatFactorInfo *info) 99849b5e25fSSatish Balay { 9994ccecd49SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)inA->data; 100049b5e25fSSatish Balay Mat outA; 1001dfbe8321SBarry Smith PetscErrorCode ierr; 100249b5e25fSSatish Balay PetscTruth row_identity,col_identity; 100349b5e25fSSatish Balay 100449b5e25fSSatish Balay PetscFunctionBegin; 100549b5e25fSSatish Balay outA = inA; 10061260e1f8SHong Zhang inA->factor = FACTOR_CHOLESKY; 100749b5e25fSSatish Balay 100849b5e25fSSatish Balay if (!a->diag) { 10091a3463dfSHong Zhang ierr = MatMarkDiagonal_SeqSBAIJ(inA);CHKERRQ(ierr); 101049b5e25fSSatish Balay } 101149b5e25fSSatish Balay /* 101249b5e25fSSatish Balay Blocksize 2, 3, 4, 5, 6 and 7 have a special faster factorization/solver 101349b5e25fSSatish Balay for ILU(0) factorization with natural ordering 101449b5e25fSSatish Balay */ 101549b5e25fSSatish Balay switch (a->bs) { 101649b5e25fSSatish Balay case 1: 10170fe9e5beSBarry Smith inA->ops->solve = MatSolve_SeqSBAIJ_1_NaturalOrdering; 101807f98182SSatish Balay inA->ops->solvetranspose = MatSolve_SeqSBAIJ_1_NaturalOrdering; 1019d59c15a7SBarry Smith inA->ops->solves = MatSolves_SeqSBAIJ_1; 10204d101231SSatish Balay PetscLoginfo(inA,"MatICCFactor_SeqSBAIJ:Using special in-place natural ordering solvetrans BS=1\n"); 102149b5e25fSSatish Balay case 2: 10221a3463dfSHong Zhang inA->ops->lufactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2_NaturalOrdering; 10231a3463dfSHong Zhang inA->ops->solve = MatSolve_SeqSBAIJ_2_NaturalOrdering; 10241a3463dfSHong Zhang inA->ops->solvetranspose = MatSolve_SeqSBAIJ_2_NaturalOrdering; 10254d101231SSatish Balay PetscLogInfo(inA,"MatICCFactor_SeqSBAIJ:Using special in-place natural ordering factor and solve BS=2\n"); 102649b5e25fSSatish Balay break; 102749b5e25fSSatish Balay case 3: 10281a3463dfSHong Zhang inA->ops->lufactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3_NaturalOrdering; 10291a3463dfSHong Zhang inA->ops->solve = MatSolve_SeqSBAIJ_3_NaturalOrdering; 10301a3463dfSHong Zhang inA->ops->solvetranspose = MatSolve_SeqSBAIJ_3_NaturalOrdering; 10314d101231SSatish Balay PetscLogInfo(inA,"MatICCFactor_SeqSBAIJ:Using special in-place natural ordering factor and solve BS=3\n"); 103249b5e25fSSatish Balay break; 103349b5e25fSSatish Balay case 4: 10341a3463dfSHong Zhang inA->ops->lufactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4_NaturalOrdering; 10351a3463dfSHong Zhang inA->ops->solve = MatSolve_SeqSBAIJ_4_NaturalOrdering; 10361a3463dfSHong Zhang inA->ops->solvetranspose = MatSolve_SeqSBAIJ_4_NaturalOrdering; 10374d101231SSatish Balay PetscLogInfo(inA,"MatICCFactor_SeqSBAIJ:Using special in-place natural ordering factor and solve BS=4\n"); 103849b5e25fSSatish Balay break; 103949b5e25fSSatish Balay case 5: 10401a3463dfSHong Zhang inA->ops->lufactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5_NaturalOrdering; 10411a3463dfSHong Zhang inA->ops->solve = MatSolve_SeqSBAIJ_5_NaturalOrdering; 10421a3463dfSHong Zhang inA->ops->solvetranspose = MatSolve_SeqSBAIJ_5_NaturalOrdering; 10434d101231SSatish Balay PetscLogInfo(inA,"MatICCFactor_SeqSBAIJ:Using special in-place natural ordering factor and solve BS=5\n"); 104449b5e25fSSatish Balay break; 104549b5e25fSSatish Balay case 6: 10461a3463dfSHong Zhang inA->ops->lufactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6_NaturalOrdering; 10471a3463dfSHong Zhang inA->ops->solve = MatSolve_SeqSBAIJ_6_NaturalOrdering; 10481a3463dfSHong Zhang inA->ops->solvetranspose = MatSolve_SeqSBAIJ_6_NaturalOrdering; 10494d101231SSatish Balay PetscLogInfo(inA,"MatICCFactor_SeqSBAIJ:Using special in-place natural ordering factor and solve BS=6\n"); 105049b5e25fSSatish Balay break; 105149b5e25fSSatish Balay case 7: 10521a3463dfSHong Zhang inA->ops->lufactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7_NaturalOrdering; 10531a3463dfSHong Zhang inA->ops->solvetranspose = MatSolve_SeqSBAIJ_7_NaturalOrdering; 10541a3463dfSHong Zhang inA->ops->solve = MatSolve_SeqSBAIJ_7_NaturalOrdering; 10554d101231SSatish Balay PetscLogInfo(inA,"MatICCFactor_SeqSBAIJ:Using special in-place natural ordering factor and solve BS=7\n"); 105649b5e25fSSatish Balay break; 105749b5e25fSSatish Balay default: 105849b5e25fSSatish Balay a->row = row; 10591a3463dfSHong Zhang a->icol = col; 106049b5e25fSSatish Balay ierr = PetscObjectReference((PetscObject)row);CHKERRQ(ierr); 106149b5e25fSSatish Balay ierr = PetscObjectReference((PetscObject)col);CHKERRQ(ierr); 106249b5e25fSSatish Balay 106349b5e25fSSatish Balay /* Create the invert permutation so that it can be used in MatLUFactorNumeric() */ 106449b5e25fSSatish Balay ierr = ISInvertPermutation(col,PETSC_DECIDE, &(a->icol));CHKERRQ(ierr); 106552e6d16bSBarry Smith ierr = PetscLogObjectParent(inA,a->icol);CHKERRQ(ierr); 106649b5e25fSSatish Balay 106749b5e25fSSatish Balay if (!a->solve_work) { 106887828ca2SBarry Smith ierr = PetscMalloc((A->m+a->bs)*sizeof(PetscScalar),&a->solve_work);CHKERRQ(ierr); 106952e6d16bSBarry Smith ierr = PetscLogObjectMemory(inA,(A->m+a->bs)*sizeof(PetscScalar));CHKERRQ(ierr); 107049b5e25fSSatish Balay } 107149b5e25fSSatish Balay } 107249b5e25fSSatish Balay 1073af281ebdSHong Zhang ierr = MatCholeskyFactorNumeric(inA,info,&outA);CHKERRQ(ierr); 107449b5e25fSSatish Balay PetscFunctionReturn(0); 107549b5e25fSSatish Balay } 1076950f1e5bSHong Zhang #endif 1077950f1e5bSHong Zhang 10784a2ae208SSatish Balay #undef __FUNCT__ 10794a2ae208SSatish Balay #define __FUNCT__ "MatPrintHelp_SeqSBAIJ" 1080dfbe8321SBarry Smith PetscErrorCode MatPrintHelp_SeqSBAIJ(Mat A) 108149b5e25fSSatish Balay { 108249b5e25fSSatish Balay static PetscTruth called = PETSC_FALSE; 108349b5e25fSSatish Balay MPI_Comm comm = A->comm; 1084dfbe8321SBarry Smith PetscErrorCode ierr; 108549b5e25fSSatish Balay 108649b5e25fSSatish Balay PetscFunctionBegin; 108749b5e25fSSatish Balay if (called) {PetscFunctionReturn(0);} else called = PETSC_TRUE; 108849b5e25fSSatish Balay ierr = (*PetscHelpPrintf)(comm," Options for MATSEQSBAIJ and MATMPISBAIJ matrix formats (the defaults):\n");CHKERRQ(ierr); 108949b5e25fSSatish Balay ierr = (*PetscHelpPrintf)(comm," -mat_block_size <block_size>\n");CHKERRQ(ierr); 1090*941593c8SHong Zhang ierr = (*PetscHelpPrintf)(comm," -mat_ignore_ltriangular: Ignore lower triangular values set by user\n");CHKERRQ(ierr); 109149b5e25fSSatish Balay PetscFunctionReturn(0); 109249b5e25fSSatish Balay } 109349b5e25fSSatish Balay 109449b5e25fSSatish Balay EXTERN_C_BEGIN 10954a2ae208SSatish Balay #undef __FUNCT__ 10964a2ae208SSatish Balay #define __FUNCT__ "MatSeqSBAIJSetColumnIndices_SeqSBAIJ" 109713f74950SBarry Smith PetscErrorCode MatSeqSBAIJSetColumnIndices_SeqSBAIJ(Mat mat,PetscInt *indices) 109849b5e25fSSatish Balay { 1099045c9aa0SHong Zhang Mat_SeqSBAIJ *baij = (Mat_SeqSBAIJ *)mat->data; 110013f74950SBarry Smith PetscInt i,nz,n; 110149b5e25fSSatish Balay 110249b5e25fSSatish Balay PetscFunctionBegin; 11036c6c5352SBarry Smith nz = baij->maxnz; 1104c464158bSHong Zhang n = mat->n; 110549b5e25fSSatish Balay for (i=0; i<nz; i++) { 110649b5e25fSSatish Balay baij->j[i] = indices[i]; 110749b5e25fSSatish Balay } 11086c6c5352SBarry Smith baij->nz = nz; 110949b5e25fSSatish Balay for (i=0; i<n; i++) { 111049b5e25fSSatish Balay baij->ilen[i] = baij->imax[i]; 111149b5e25fSSatish Balay } 111249b5e25fSSatish Balay PetscFunctionReturn(0); 111349b5e25fSSatish Balay } 111449b5e25fSSatish Balay EXTERN_C_END 111549b5e25fSSatish Balay 11164a2ae208SSatish Balay #undef __FUNCT__ 11174a2ae208SSatish Balay #define __FUNCT__ "MatSeqSBAIJSetColumnIndices" 111849b5e25fSSatish Balay /*@ 111919585528SSatish Balay MatSeqSBAIJSetColumnIndices - Set the column indices for all the rows 112049b5e25fSSatish Balay in the matrix. 112149b5e25fSSatish Balay 112249b5e25fSSatish Balay Input Parameters: 112319585528SSatish Balay + mat - the SeqSBAIJ matrix 112449b5e25fSSatish Balay - indices - the column indices 112549b5e25fSSatish Balay 112649b5e25fSSatish Balay Level: advanced 112749b5e25fSSatish Balay 112849b5e25fSSatish Balay Notes: 112949b5e25fSSatish Balay This can be called if you have precomputed the nonzero structure of the 113049b5e25fSSatish Balay matrix and want to provide it to the matrix object to improve the performance 113149b5e25fSSatish Balay of the MatSetValues() operation. 113249b5e25fSSatish Balay 113349b5e25fSSatish Balay You MUST have set the correct numbers of nonzeros per row in the call to 1134d1be2dadSMatthew Knepley MatCreateSeqSBAIJ(), and the columns indices MUST be sorted. 113549b5e25fSSatish Balay 1136ab9f2c04SSatish Balay MUST be called before any calls to MatSetValues() 113749b5e25fSSatish Balay 1138ab9f2c04SSatish Balay .seealso: MatCreateSeqSBAIJ 113949b5e25fSSatish Balay @*/ 114013f74950SBarry Smith PetscErrorCode MatSeqSBAIJSetColumnIndices(Mat mat,PetscInt *indices) 114149b5e25fSSatish Balay { 114213f74950SBarry Smith PetscErrorCode ierr,(*f)(Mat,PetscInt *); 114349b5e25fSSatish Balay 114449b5e25fSSatish Balay PetscFunctionBegin; 11454482741eSBarry Smith PetscValidHeaderSpecific(mat,MAT_COOKIE,1); 11464482741eSBarry Smith PetscValidPointer(indices,2); 1147c134de8dSSatish Balay ierr = PetscObjectQueryFunction((PetscObject)mat,"MatSeqSBAIJSetColumnIndices_C",(void (**)(void))&f);CHKERRQ(ierr); 114849b5e25fSSatish Balay if (f) { 114949b5e25fSSatish Balay ierr = (*f)(mat,indices);CHKERRQ(ierr); 115049b5e25fSSatish Balay } else { 1151e005ede5SBarry Smith SETERRQ(PETSC_ERR_SUP,"Wrong type of matrix to set column indices"); 115249b5e25fSSatish Balay } 115349b5e25fSSatish Balay PetscFunctionReturn(0); 115449b5e25fSSatish Balay } 115549b5e25fSSatish Balay 11564a2ae208SSatish Balay #undef __FUNCT__ 11574a2ae208SSatish Balay #define __FUNCT__ "MatSetUpPreallocation_SeqSBAIJ" 1158dfbe8321SBarry Smith PetscErrorCode MatSetUpPreallocation_SeqSBAIJ(Mat A) 1159273d9f13SBarry Smith { 1160dfbe8321SBarry Smith PetscErrorCode ierr; 1161273d9f13SBarry Smith 1162273d9f13SBarry Smith PetscFunctionBegin; 1163273d9f13SBarry Smith ierr = MatSeqSBAIJSetPreallocation(A,1,PETSC_DEFAULT,0);CHKERRQ(ierr); 1164273d9f13SBarry Smith PetscFunctionReturn(0); 1165273d9f13SBarry Smith } 1166273d9f13SBarry Smith 1167a6ece127SHong Zhang #undef __FUNCT__ 1168a6ece127SHong Zhang #define __FUNCT__ "MatGetArray_SeqSBAIJ" 1169dfbe8321SBarry Smith PetscErrorCode MatGetArray_SeqSBAIJ(Mat A,PetscScalar *array[]) 1170a6ece127SHong Zhang { 1171a6ece127SHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 1172a6ece127SHong Zhang PetscFunctionBegin; 1173a6ece127SHong Zhang *array = a->a; 1174a6ece127SHong Zhang PetscFunctionReturn(0); 1175a6ece127SHong Zhang } 1176a6ece127SHong Zhang 1177a6ece127SHong Zhang #undef __FUNCT__ 1178a6ece127SHong Zhang #define __FUNCT__ "MatRestoreArray_SeqSBAIJ" 1179dfbe8321SBarry Smith PetscErrorCode MatRestoreArray_SeqSBAIJ(Mat A,PetscScalar *array[]) 1180a6ece127SHong Zhang { 1181a6ece127SHong Zhang PetscFunctionBegin; 1182a6ece127SHong Zhang PetscFunctionReturn(0); 1183a6ece127SHong Zhang } 1184a6ece127SHong Zhang 118542ee4b1aSHong Zhang #include "petscblaslapack.h" 118642ee4b1aSHong Zhang #undef __FUNCT__ 118742ee4b1aSHong Zhang #define __FUNCT__ "MatAXPY_SeqSBAIJ" 1188dfbe8321SBarry Smith PetscErrorCode MatAXPY_SeqSBAIJ(const PetscScalar *a,Mat X,Mat Y,MatStructure str) 118942ee4b1aSHong Zhang { 119042ee4b1aSHong Zhang Mat_SeqSBAIJ *x=(Mat_SeqSBAIJ *)X->data, *y=(Mat_SeqSBAIJ *)Y->data; 1191dfbe8321SBarry Smith PetscErrorCode ierr; 1192521d7252SBarry Smith PetscInt i,bs=Y->bs,bs2,j; 11934ce68768SBarry Smith PetscBLASInt bnz = (PetscBLASInt)x->nz,one = 1; 119442ee4b1aSHong Zhang 119542ee4b1aSHong Zhang PetscFunctionBegin; 119642ee4b1aSHong Zhang if (str == SAME_NONZERO_PATTERN) { 119771044d3cSBarry Smith BLASaxpy_(&bnz,(PetscScalar*)a,x->a,&one,y->a,&one); 1198c537a176SHong Zhang } else if (str == SUBSET_NONZERO_PATTERN) { /* nonzeros of X is a subset of Y's */ 1199c4319e64SHong Zhang if (y->xtoy && y->XtoY != X) { 1200c4319e64SHong Zhang ierr = PetscFree(y->xtoy);CHKERRQ(ierr); 1201c4319e64SHong Zhang ierr = MatDestroy(y->XtoY);CHKERRQ(ierr); 1202c537a176SHong Zhang } 1203c4319e64SHong Zhang if (!y->xtoy) { /* get xtoy */ 1204c4319e64SHong Zhang ierr = MatAXPYGetxtoy_Private(x->mbs,x->i,x->j,PETSC_NULL, y->i,y->j,PETSC_NULL, &y->xtoy);CHKERRQ(ierr); 1205c4319e64SHong Zhang y->XtoY = X; 1206c537a176SHong Zhang } 1207c4319e64SHong Zhang bs2 = bs*bs; 12086c6c5352SBarry Smith for (i=0; i<x->nz; i++) { 1209c4319e64SHong Zhang j = 0; 1210c4319e64SHong Zhang while (j < bs2){ 12116550863bSHong Zhang y->a[bs2*y->xtoy[i]+j] += (*a)*(x->a[bs2*i+j]); 1212c4319e64SHong Zhang j++; 1213c537a176SHong Zhang } 1214c4319e64SHong Zhang } 121577431f27SBarry Smith PetscLogInfo(0,"MatAXPY_SeqSBAIJ: 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)); 121642ee4b1aSHong Zhang } else { 121742ee4b1aSHong Zhang ierr = MatAXPY_Basic(a,X,Y,str);CHKERRQ(ierr); 121842ee4b1aSHong Zhang } 121942ee4b1aSHong Zhang PetscFunctionReturn(0); 122042ee4b1aSHong Zhang } 122142ee4b1aSHong Zhang 1222efcf0fc3SBarry Smith #undef __FUNCT__ 1223efcf0fc3SBarry Smith #define __FUNCT__ "MatIsSymmetric_SeqSBAIJ" 1224dfbe8321SBarry Smith PetscErrorCode MatIsSymmetric_SeqSBAIJ(Mat A,PetscReal tol,PetscTruth *flg) 1225efcf0fc3SBarry Smith { 1226efcf0fc3SBarry Smith PetscFunctionBegin; 1227efcf0fc3SBarry Smith *flg = PETSC_TRUE; 1228efcf0fc3SBarry Smith PetscFunctionReturn(0); 1229efcf0fc3SBarry Smith } 1230efcf0fc3SBarry Smith 1231efcf0fc3SBarry Smith #undef __FUNCT__ 1232efcf0fc3SBarry Smith #define __FUNCT__ "MatIsStructurallySymmetric_SeqSBAIJ" 1233dfbe8321SBarry Smith PetscErrorCode MatIsStructurallySymmetric_SeqSBAIJ(Mat A,PetscTruth *flg) 1234efcf0fc3SBarry Smith { 1235efcf0fc3SBarry Smith PetscFunctionBegin; 1236efcf0fc3SBarry Smith *flg = PETSC_TRUE; 1237efcf0fc3SBarry Smith PetscFunctionReturn(0); 1238efcf0fc3SBarry Smith } 1239efcf0fc3SBarry Smith 1240efcf0fc3SBarry Smith #undef __FUNCT__ 1241efcf0fc3SBarry Smith #define __FUNCT__ "MatIsHermitian_SeqSBAIJ" 1242dfbe8321SBarry Smith PetscErrorCode MatIsHermitian_SeqSBAIJ(Mat A,PetscTruth *flg) 1243efcf0fc3SBarry Smith { 1244efcf0fc3SBarry Smith PetscFunctionBegin; 1245efcf0fc3SBarry Smith *flg = PETSC_FALSE; 1246efcf0fc3SBarry Smith PetscFunctionReturn(0); 1247efcf0fc3SBarry Smith } 1248efcf0fc3SBarry Smith 124949b5e25fSSatish Balay /* -------------------------------------------------------------------*/ 125049b5e25fSSatish Balay static struct _MatOps MatOps_Values = {MatSetValues_SeqSBAIJ, 125149b5e25fSSatish Balay MatGetRow_SeqSBAIJ, 125249b5e25fSSatish Balay MatRestoreRow_SeqSBAIJ, 125349b5e25fSSatish Balay MatMult_SeqSBAIJ_N, 125497304618SKris Buschelman /* 4*/ MatMultAdd_SeqSBAIJ_N, 1255e005ede5SBarry Smith MatMult_SeqSBAIJ_N, 1256e005ede5SBarry Smith MatMultAdd_SeqSBAIJ_N, 125749b5e25fSSatish Balay MatSolve_SeqSBAIJ_N, 125849b5e25fSSatish Balay 0, 125949b5e25fSSatish Balay 0, 126097304618SKris Buschelman /*10*/ 0, 126149b5e25fSSatish Balay 0, 12625f4ef2deSHong Zhang MatCholeskyFactor_SeqSBAIJ, 1263d06b337dSHong Zhang MatRelax_SeqSBAIJ, 126449b5e25fSSatish Balay MatTranspose_SeqSBAIJ, 126597304618SKris Buschelman /*15*/ MatGetInfo_SeqSBAIJ, 126649b5e25fSSatish Balay MatEqual_SeqSBAIJ, 126749b5e25fSSatish Balay MatGetDiagonal_SeqSBAIJ, 126849b5e25fSSatish Balay MatDiagonalScale_SeqSBAIJ, 126949b5e25fSSatish Balay MatNorm_SeqSBAIJ, 127097304618SKris Buschelman /*20*/ 0, 127149b5e25fSSatish Balay MatAssemblyEnd_SeqSBAIJ, 127249b5e25fSSatish Balay 0, 127349b5e25fSSatish Balay MatSetOption_SeqSBAIJ, 127449b5e25fSSatish Balay MatZeroEntries_SeqSBAIJ, 1275dcf5cc72SBarry Smith /*25*/ 0, 127649b5e25fSSatish Balay 0, 127749b5e25fSSatish Balay 0, 12785f4ef2deSHong Zhang MatCholeskyFactorSymbolic_SeqSBAIJ, 12795f4ef2deSHong Zhang MatCholeskyFactorNumeric_SeqSBAIJ_N, 128097304618SKris Buschelman /*30*/ MatSetUpPreallocation_SeqSBAIJ, 1281c464158bSHong Zhang 0, 12824d101231SSatish Balay MatICCFactorSymbolic_SeqSBAIJ, 1283a6ece127SHong Zhang MatGetArray_SeqSBAIJ, 1284a6ece127SHong Zhang MatRestoreArray_SeqSBAIJ, 128597304618SKris Buschelman /*35*/ MatDuplicate_SeqSBAIJ, 128649b5e25fSSatish Balay 0, 128749b5e25fSSatish Balay 0, 128849b5e25fSSatish Balay 0, 1289950f1e5bSHong Zhang 0, 129097304618SKris Buschelman /*40*/ MatAXPY_SeqSBAIJ, 129149b5e25fSSatish Balay MatGetSubMatrices_SeqSBAIJ, 129249b5e25fSSatish Balay MatIncreaseOverlap_SeqSBAIJ, 129349b5e25fSSatish Balay MatGetValues_SeqSBAIJ, 129449b5e25fSSatish Balay 0, 129597304618SKris Buschelman /*45*/ MatPrintHelp_SeqSBAIJ, 129649b5e25fSSatish Balay MatScale_SeqSBAIJ, 129749b5e25fSSatish Balay 0, 129849b5e25fSSatish Balay 0, 129949b5e25fSSatish Balay 0, 1300521d7252SBarry Smith /*50*/ 0, 130149b5e25fSSatish Balay MatGetRowIJ_SeqSBAIJ, 130249b5e25fSSatish Balay MatRestoreRowIJ_SeqSBAIJ, 130349b5e25fSSatish Balay 0, 130449b5e25fSSatish Balay 0, 130597304618SKris Buschelman /*55*/ 0, 130649b5e25fSSatish Balay 0, 130749b5e25fSSatish Balay 0, 130849b5e25fSSatish Balay 0, 130949b5e25fSSatish Balay MatSetValuesBlocked_SeqSBAIJ, 131097304618SKris Buschelman /*60*/ MatGetSubMatrix_SeqSBAIJ, 131149b5e25fSSatish Balay 0, 131249b5e25fSSatish Balay 0, 13138a124369SBarry Smith MatGetPetscMaps_Petsc, 1314d959ec07SHong Zhang 0, 131597304618SKris Buschelman /*65*/ 0, 1316d959ec07SHong Zhang 0, 1317d959ec07SHong Zhang 0, 1318d959ec07SHong Zhang 0, 1319d959ec07SHong Zhang 0, 132097304618SKris Buschelman /*70*/ MatGetRowMax_SeqSBAIJ, 13213e0d88b5SBarry Smith 0, 13223e0d88b5SBarry Smith 0, 13233e0d88b5SBarry Smith 0, 13243e0d88b5SBarry Smith 0, 132597304618SKris Buschelman /*75*/ 0, 13263e0d88b5SBarry Smith 0, 13273e0d88b5SBarry Smith 0, 13283e0d88b5SBarry Smith 0, 13293e0d88b5SBarry Smith 0, 133097304618SKris Buschelman /*80*/ 0, 13313e0d88b5SBarry Smith 0, 13323e0d88b5SBarry Smith 0, 13333e0d88b5SBarry Smith #if !defined(PETSC_USE_COMPLEX) 133497304618SKris Buschelman MatGetInertia_SeqSBAIJ, 13353e0d88b5SBarry Smith #else 133697304618SKris Buschelman 0, 13373e0d88b5SBarry Smith #endif 1338865e5f61SKris Buschelman MatLoad_SeqSBAIJ, 1339865e5f61SKris Buschelman /*85*/ MatIsSymmetric_SeqSBAIJ, 1340865e5f61SKris Buschelman MatIsHermitian_SeqSBAIJ, 1341efcf0fc3SBarry Smith MatIsStructurallySymmetric_SeqSBAIJ, 1342865e5f61SKris Buschelman 0, 1343865e5f61SKris Buschelman 0, 1344865e5f61SKris Buschelman /*90*/ 0, 1345865e5f61SKris Buschelman 0, 1346865e5f61SKris Buschelman 0, 1347865e5f61SKris Buschelman 0, 1348865e5f61SKris Buschelman 0, 1349865e5f61SKris Buschelman /*95*/ 0, 1350865e5f61SKris Buschelman 0, 1351865e5f61SKris Buschelman 0, 1352865e5f61SKris Buschelman 0}; 135349b5e25fSSatish Balay 135449b5e25fSSatish Balay EXTERN_C_BEGIN 13554a2ae208SSatish Balay #undef __FUNCT__ 13564a2ae208SSatish Balay #define __FUNCT__ "MatStoreValues_SeqSBAIJ" 1357dfbe8321SBarry Smith PetscErrorCode MatStoreValues_SeqSBAIJ(Mat mat) 135849b5e25fSSatish Balay { 13594afc71dfSHong Zhang Mat_SeqSBAIJ *aij = (Mat_SeqSBAIJ *)mat->data; 1360521d7252SBarry Smith PetscInt nz = aij->i[mat->m]*mat->bs*aij->bs2; 1361dfbe8321SBarry Smith PetscErrorCode ierr; 136249b5e25fSSatish Balay 136349b5e25fSSatish Balay PetscFunctionBegin; 136449b5e25fSSatish Balay if (aij->nonew != 1) { 1365e005ede5SBarry Smith SETERRQ(PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NO_NEW_NONZERO_LOCATIONS);first"); 136649b5e25fSSatish Balay } 136749b5e25fSSatish Balay 136849b5e25fSSatish Balay /* allocate space for values if not already there */ 136949b5e25fSSatish Balay if (!aij->saved_values) { 137087828ca2SBarry Smith ierr = PetscMalloc((nz+1)*sizeof(PetscScalar),&aij->saved_values);CHKERRQ(ierr); 137149b5e25fSSatish Balay } 137249b5e25fSSatish Balay 137349b5e25fSSatish Balay /* copy values over */ 137487828ca2SBarry Smith ierr = PetscMemcpy(aij->saved_values,aij->a,nz*sizeof(PetscScalar));CHKERRQ(ierr); 137549b5e25fSSatish Balay PetscFunctionReturn(0); 137649b5e25fSSatish Balay } 137749b5e25fSSatish Balay EXTERN_C_END 137849b5e25fSSatish Balay 137949b5e25fSSatish Balay EXTERN_C_BEGIN 13804a2ae208SSatish Balay #undef __FUNCT__ 13814a2ae208SSatish Balay #define __FUNCT__ "MatRetrieveValues_SeqSBAIJ" 1382dfbe8321SBarry Smith PetscErrorCode MatRetrieveValues_SeqSBAIJ(Mat mat) 138349b5e25fSSatish Balay { 13844afc71dfSHong Zhang Mat_SeqSBAIJ *aij = (Mat_SeqSBAIJ *)mat->data; 13856849ba73SBarry Smith PetscErrorCode ierr; 1386521d7252SBarry Smith PetscInt nz = aij->i[mat->m]*mat->bs*aij->bs2; 138749b5e25fSSatish Balay 138849b5e25fSSatish Balay PetscFunctionBegin; 138949b5e25fSSatish Balay if (aij->nonew != 1) { 1390e005ede5SBarry Smith SETERRQ(PETSC_ERR_ORDER,"Must call MatSetOption(A,MAT_NO_NEW_NONZERO_LOCATIONS);first"); 139149b5e25fSSatish Balay } 139249b5e25fSSatish Balay if (!aij->saved_values) { 1393e005ede5SBarry Smith SETERRQ(PETSC_ERR_ORDER,"Must call MatStoreValues(A);first"); 139449b5e25fSSatish Balay } 139549b5e25fSSatish Balay 139649b5e25fSSatish Balay /* copy values over */ 139787828ca2SBarry Smith ierr = PetscMemcpy(aij->a,aij->saved_values,nz*sizeof(PetscScalar));CHKERRQ(ierr); 139849b5e25fSSatish Balay PetscFunctionReturn(0); 139949b5e25fSSatish Balay } 140049b5e25fSSatish Balay EXTERN_C_END 140149b5e25fSSatish Balay 14028549e402SHong Zhang EXTERN_C_BEGIN 14034a2ae208SSatish Balay #undef __FUNCT__ 1404a23d5eceSKris Buschelman #define __FUNCT__ "MatSeqSBAIJSetPreallocation_SeqSBAIJ" 140513f74950SBarry Smith PetscErrorCode MatSeqSBAIJSetPreallocation_SeqSBAIJ(Mat B,PetscInt bs,PetscInt nz,PetscInt *nnz) 140649b5e25fSSatish Balay { 1407c464158bSHong Zhang Mat_SeqSBAIJ *b = (Mat_SeqSBAIJ*)B->data; 14086849ba73SBarry Smith PetscErrorCode ierr; 140913f74950SBarry Smith PetscInt i,len,mbs,bs2; 141049b5e25fSSatish Balay PetscTruth flg; 141149b5e25fSSatish Balay 141249b5e25fSSatish Balay PetscFunctionBegin; 1413273d9f13SBarry Smith B->preallocated = PETSC_TRUE; 1414e82a3eeeSBarry Smith ierr = PetscOptionsGetInt(B->prefix,"-mat_block_size",&bs,PETSC_NULL);CHKERRQ(ierr); 1415c464158bSHong Zhang mbs = B->m/bs; 141649b5e25fSSatish Balay bs2 = bs*bs; 141749b5e25fSSatish Balay 1418c464158bSHong Zhang if (mbs*bs != B->m) { 141929bbc08cSBarry Smith SETERRQ(PETSC_ERR_ARG_SIZ,"Number rows, cols must be divisible by blocksize"); 142049b5e25fSSatish Balay } 142149b5e25fSSatish Balay 1422435da068SBarry Smith if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 3; 142377431f27SBarry Smith if (nz < 0) SETERRQ1(PETSC_ERR_ARG_OUTOFRANGE,"nz cannot be less than 0: value %D",nz); 142449b5e25fSSatish Balay if (nnz) { 142549b5e25fSSatish Balay for (i=0; i<mbs; i++) { 142677431f27SBarry Smith if (nnz[i] < 0) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"nnz cannot be less than 0: local row %D value %D",i,nnz[i]); 142777431f27SBarry 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); 142849b5e25fSSatish Balay } 142949b5e25fSSatish Balay } 143049b5e25fSSatish Balay 1431e82a3eeeSBarry Smith ierr = PetscOptionsHasName(B->prefix,"-mat_no_unroll",&flg);CHKERRQ(ierr); 143249b5e25fSSatish Balay if (!flg) { 143349b5e25fSSatish Balay switch (bs) { 143449b5e25fSSatish Balay case 1: 14354ccecd49SHong Zhang B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_1; 143649b5e25fSSatish Balay B->ops->solve = MatSolve_SeqSBAIJ_1; 1437d59c15a7SBarry Smith B->ops->solves = MatSolves_SeqSBAIJ_1; 1438e005ede5SBarry Smith B->ops->solvetranspose = MatSolve_SeqSBAIJ_1; 143949b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_1; 144049b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_1; 144149b5e25fSSatish Balay break; 144249b5e25fSSatish Balay case 2: 14434ccecd49SHong Zhang B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_2; 144449b5e25fSSatish Balay B->ops->solve = MatSolve_SeqSBAIJ_2; 1445e005ede5SBarry Smith B->ops->solvetranspose = MatSolve_SeqSBAIJ_2; 144649b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_2; 144749b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_2; 144849b5e25fSSatish Balay break; 144949b5e25fSSatish Balay case 3: 14505f4ef2deSHong Zhang B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_3; 145149b5e25fSSatish Balay B->ops->solve = MatSolve_SeqSBAIJ_3; 1452e005ede5SBarry Smith B->ops->solvetranspose = MatSolve_SeqSBAIJ_3; 145349b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_3; 145449b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_3; 145549b5e25fSSatish Balay break; 145649b5e25fSSatish Balay case 4: 14575f4ef2deSHong Zhang B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_4; 145849b5e25fSSatish Balay B->ops->solve = MatSolve_SeqSBAIJ_4; 1459e005ede5SBarry Smith B->ops->solvetranspose = MatSolve_SeqSBAIJ_4; 146049b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_4; 146149b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_4; 146249b5e25fSSatish Balay break; 146349b5e25fSSatish Balay case 5: 14645f4ef2deSHong Zhang B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_5; 146549b5e25fSSatish Balay B->ops->solve = MatSolve_SeqSBAIJ_5; 1466e005ede5SBarry Smith B->ops->solvetranspose = MatSolve_SeqSBAIJ_5; 146749b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_5; 146849b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_5; 146949b5e25fSSatish Balay break; 147049b5e25fSSatish Balay case 6: 14715f4ef2deSHong Zhang B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_6; 147249b5e25fSSatish Balay B->ops->solve = MatSolve_SeqSBAIJ_6; 1473e005ede5SBarry Smith B->ops->solvetranspose = MatSolve_SeqSBAIJ_6; 147449b5e25fSSatish Balay B->ops->mult = MatMult_SeqSBAIJ_6; 147549b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_6; 147649b5e25fSSatish Balay break; 147749b5e25fSSatish Balay case 7: 1478de53e5efSHong Zhang B->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqSBAIJ_7; 147949b5e25fSSatish Balay B->ops->solve = MatSolve_SeqSBAIJ_7; 1480e005ede5SBarry Smith B->ops->solvetranspose = MatSolve_SeqSBAIJ_7; 1481de53e5efSHong Zhang B->ops->mult = MatMult_SeqSBAIJ_7; 148249b5e25fSSatish Balay B->ops->multadd = MatMultAdd_SeqSBAIJ_7; 148349b5e25fSSatish Balay break; 148449b5e25fSSatish Balay } 148549b5e25fSSatish Balay } 148649b5e25fSSatish Balay 148749b5e25fSSatish Balay b->mbs = mbs; 14884afc71dfSHong Zhang b->nbs = mbs; 148913f74950SBarry Smith ierr = PetscMalloc((mbs+1)*sizeof(PetscInt),&b->imax);CHKERRQ(ierr); 149049b5e25fSSatish Balay if (!nnz) { 1491435da068SBarry Smith if (nz == PETSC_DEFAULT || nz == PETSC_DECIDE) nz = 5; 149249b5e25fSSatish Balay else if (nz <= 0) nz = 1; 149349b5e25fSSatish Balay for (i=0; i<mbs; i++) { 14948cef66ccSHong Zhang b->imax[i] = nz; 149549b5e25fSSatish Balay } 1496153ea458SHong Zhang nz = nz*mbs; /* total nz */ 149749b5e25fSSatish Balay } else { 149849b5e25fSSatish Balay nz = 0; 14998cef66ccSHong Zhang for (i=0; i<mbs; i++) {b->imax[i] = nnz[i]; nz += nnz[i];} 150049b5e25fSSatish Balay } 15016c6c5352SBarry Smith /* nz=(nz+mbs)/2; */ /* total diagonal and superdiagonal nonzero blocks */ 150249b5e25fSSatish Balay 150349b5e25fSSatish Balay /* allocate the matrix space */ 150413f74950SBarry Smith len = nz*sizeof(PetscInt) + nz*bs2*sizeof(MatScalar) + (B->m+1)*sizeof(PetscInt); 150582502324SSatish Balay ierr = PetscMalloc(len,&b->a);CHKERRQ(ierr); 15066c6c5352SBarry Smith ierr = PetscMemzero(b->a,nz*bs2*sizeof(MatScalar));CHKERRQ(ierr); 150713f74950SBarry Smith b->j = (PetscInt*)(b->a + nz*bs2); 150813f74950SBarry Smith ierr = PetscMemzero(b->j,nz*sizeof(PetscInt));CHKERRQ(ierr); 15096c6c5352SBarry Smith b->i = b->j + nz; 151049b5e25fSSatish Balay b->singlemalloc = PETSC_TRUE; 151149b5e25fSSatish Balay 151249b5e25fSSatish Balay /* pointer to beginning of each row */ 151349b5e25fSSatish Balay b->i[0] = 0; 151449b5e25fSSatish Balay for (i=1; i<mbs+1; i++) { 151549b5e25fSSatish Balay b->i[i] = b->i[i-1] + b->imax[i-1]; 151649b5e25fSSatish Balay } 151749b5e25fSSatish Balay 151849b5e25fSSatish Balay /* b->ilen will count nonzeros in each block row so far. */ 151913f74950SBarry Smith ierr = PetscMalloc((mbs+1)*sizeof(PetscInt),&b->ilen);CHKERRQ(ierr); 152052e6d16bSBarry Smith ierr = PetscLogObjectMemory(B,len+2*(mbs+1)*sizeof(PetscInt)+sizeof(struct _p_Mat)+sizeof(Mat_SeqSBAIJ));CHKERRQ(ierr); 152149b5e25fSSatish Balay for (i=0; i<mbs; i++) { b->ilen[i] = 0;} 152249b5e25fSSatish Balay 1523521d7252SBarry Smith B->bs = bs; 152449b5e25fSSatish Balay b->bs2 = bs2; 15256c6c5352SBarry Smith b->nz = 0; 15266c6c5352SBarry Smith b->maxnz = nz*bs2; 1527153ea458SHong Zhang 152816cdd363SHong Zhang b->inew = 0; 152916cdd363SHong Zhang b->jnew = 0; 153016cdd363SHong Zhang b->anew = 0; 153116cdd363SHong Zhang b->a2anew = 0; 15321a3463dfSHong Zhang b->permute = PETSC_FALSE; 1533c464158bSHong Zhang PetscFunctionReturn(0); 1534c464158bSHong Zhang } 1535a23d5eceSKris Buschelman EXTERN_C_END 1536153ea458SHong Zhang 1537d769727bSBarry Smith EXTERN_C_BEGIN 1538ceb03754SKris Buschelman EXTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqAIJ(Mat,const MatType,MatReuse,Mat*); 1539ceb03754SKris Buschelman EXTERN PetscErrorCode MatConvert_SeqSBAIJ_SeqBAIJ(Mat,const MatType,MatReuse,Mat*); 1540d769727bSBarry Smith EXTERN_C_END 1541d769727bSBarry Smith 15420bad9183SKris Buschelman /*MC 1543fafad747SKris Buschelman MATSEQSBAIJ - MATSEQSBAIJ = "seqsbaij" - A matrix type to be used for sequential symmetric block sparse matrices, 15440bad9183SKris Buschelman based on block compressed sparse row format. Only the upper triangular portion of the matrix is stored. 15450bad9183SKris Buschelman 15460bad9183SKris Buschelman Options Database Keys: 15470bad9183SKris Buschelman . -mat_type seqsbaij - sets the matrix type to "seqsbaij" during a call to MatSetFromOptions() 15480bad9183SKris Buschelman 15490bad9183SKris Buschelman Level: beginner 15500bad9183SKris Buschelman 15510bad9183SKris Buschelman .seealso: MatCreateSeqSBAIJ 15520bad9183SKris Buschelman M*/ 15530bad9183SKris Buschelman 1554a23d5eceSKris Buschelman EXTERN_C_BEGIN 1555a23d5eceSKris Buschelman #undef __FUNCT__ 1556a23d5eceSKris Buschelman #define __FUNCT__ "MatCreate_SeqSBAIJ" 1557dfbe8321SBarry Smith PetscErrorCode MatCreate_SeqSBAIJ(Mat B) 1558a23d5eceSKris Buschelman { 1559a23d5eceSKris Buschelman Mat_SeqSBAIJ *b; 1560dfbe8321SBarry Smith PetscErrorCode ierr; 156113f74950SBarry Smith PetscMPIInt size; 1562*941593c8SHong Zhang PetscTruth flg; 1563a23d5eceSKris Buschelman 1564a23d5eceSKris Buschelman PetscFunctionBegin; 1565a23d5eceSKris Buschelman ierr = MPI_Comm_size(B->comm,&size);CHKERRQ(ierr); 1566a23d5eceSKris Buschelman if (size > 1) SETERRQ(PETSC_ERR_ARG_WRONG,"Comm must be of size 1"); 1567a23d5eceSKris Buschelman B->m = B->M = PetscMax(B->m,B->M); 1568a23d5eceSKris Buschelman B->n = B->N = PetscMax(B->n,B->N); 1569a23d5eceSKris Buschelman 1570a23d5eceSKris Buschelman ierr = PetscNew(Mat_SeqSBAIJ,&b);CHKERRQ(ierr); 1571a23d5eceSKris Buschelman B->data = (void*)b; 1572a23d5eceSKris Buschelman ierr = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr); 1573a23d5eceSKris Buschelman B->ops->destroy = MatDestroy_SeqSBAIJ; 1574a23d5eceSKris Buschelman B->ops->view = MatView_SeqSBAIJ; 1575a23d5eceSKris Buschelman B->factor = 0; 1576a23d5eceSKris Buschelman B->lupivotthreshold = 1.0; 1577a23d5eceSKris Buschelman B->mapping = 0; 1578a23d5eceSKris Buschelman b->row = 0; 1579a23d5eceSKris Buschelman b->icol = 0; 1580a23d5eceSKris Buschelman b->reallocs = 0; 1581a23d5eceSKris Buschelman b->saved_values = 0; 1582a23d5eceSKris Buschelman 1583a23d5eceSKris Buschelman ierr = PetscMapCreateMPI(B->comm,B->m,B->M,&B->rmap);CHKERRQ(ierr); 1584a23d5eceSKris Buschelman ierr = PetscMapCreateMPI(B->comm,B->n,B->N,&B->cmap);CHKERRQ(ierr); 1585a23d5eceSKris Buschelman 1586a23d5eceSKris Buschelman b->sorted = PETSC_FALSE; 1587a23d5eceSKris Buschelman b->roworiented = PETSC_TRUE; 1588a23d5eceSKris Buschelman b->nonew = 0; 1589a23d5eceSKris Buschelman b->diag = 0; 1590a23d5eceSKris Buschelman b->solve_work = 0; 1591a23d5eceSKris Buschelman b->mult_work = 0; 1592a23d5eceSKris Buschelman B->spptr = 0; 1593a23d5eceSKris Buschelman b->keepzeroedrows = PETSC_FALSE; 1594a23d5eceSKris Buschelman b->xtoy = 0; 1595a23d5eceSKris Buschelman b->XtoY = 0; 1596a23d5eceSKris Buschelman 1597a23d5eceSKris Buschelman b->inew = 0; 1598a23d5eceSKris Buschelman b->jnew = 0; 1599a23d5eceSKris Buschelman b->anew = 0; 1600a23d5eceSKris Buschelman b->a2anew = 0; 1601a23d5eceSKris Buschelman b->permute = PETSC_FALSE; 1602a23d5eceSKris Buschelman 1603*941593c8SHong Zhang b->ignore_ltriangular = PETSC_FALSE; 1604*941593c8SHong Zhang ierr = PetscOptionsHasName(PETSC_NULL,"-mat_ignore_lower_triangular",&flg);CHKERRQ(ierr); 1605*941593c8SHong Zhang if (flg) b->ignore_ltriangular = PETSC_TRUE; 1606*941593c8SHong Zhang 1607a23d5eceSKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatStoreValues_C", 1608a23d5eceSKris Buschelman "MatStoreValues_SeqSBAIJ", 1609a23d5eceSKris Buschelman MatStoreValues_SeqSBAIJ);CHKERRQ(ierr); 1610a23d5eceSKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatRetrieveValues_C", 1611a23d5eceSKris Buschelman "MatRetrieveValues_SeqSBAIJ", 1612a23d5eceSKris Buschelman (void*)MatRetrieveValues_SeqSBAIJ);CHKERRQ(ierr); 1613a23d5eceSKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatSeqSBAIJSetColumnIndices_C", 1614a23d5eceSKris Buschelman "MatSeqSBAIJSetColumnIndices_SeqSBAIJ", 1615a23d5eceSKris Buschelman MatSeqSBAIJSetColumnIndices_SeqSBAIJ);CHKERRQ(ierr); 16164e5e7fe4SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_seqsbaij_seqaij_C", 16174e5e7fe4SHong Zhang "MatConvert_SeqSBAIJ_SeqAIJ", 16184e5e7fe4SHong Zhang MatConvert_SeqSBAIJ_SeqAIJ);CHKERRQ(ierr); 1619a0e1a404SHong Zhang ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatConvert_seqsbaij_seqbaij_C", 1620a0e1a404SHong Zhang "MatConvert_SeqSBAIJ_SeqBAIJ", 1621a0e1a404SHong Zhang MatConvert_SeqSBAIJ_SeqBAIJ);CHKERRQ(ierr); 1622a23d5eceSKris Buschelman ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatSeqSBAIJSetPreallocation_C", 1623a23d5eceSKris Buschelman "MatSeqSBAIJSetPreallocation_SeqSBAIJ", 1624a23d5eceSKris Buschelman MatSeqSBAIJSetPreallocation_SeqSBAIJ);CHKERRQ(ierr); 162523ce1328SBarry Smith 162623ce1328SBarry Smith B->symmetric = PETSC_TRUE; 162723ce1328SBarry Smith B->structurally_symmetric = PETSC_TRUE; 162823ce1328SBarry Smith B->symmetric_set = PETSC_TRUE; 162923ce1328SBarry Smith B->structurally_symmetric_set = PETSC_TRUE; 1630a23d5eceSKris Buschelman PetscFunctionReturn(0); 1631a23d5eceSKris Buschelman } 1632a23d5eceSKris Buschelman EXTERN_C_END 1633a23d5eceSKris Buschelman 1634a23d5eceSKris Buschelman #undef __FUNCT__ 1635a23d5eceSKris Buschelman #define __FUNCT__ "MatSeqSBAIJSetPreallocation" 1636a23d5eceSKris Buschelman /*@C 1637a23d5eceSKris Buschelman MatSeqSBAIJSetPreallocation - Creates a sparse symmetric matrix in block AIJ (block 1638a23d5eceSKris Buschelman compressed row) format. For good matrix assembly performance the 1639a23d5eceSKris Buschelman user should preallocate the matrix storage by setting the parameter nz 1640a23d5eceSKris Buschelman (or the array nnz). By setting these parameters accurately, performance 1641a23d5eceSKris Buschelman during matrix assembly can be increased by more than a factor of 50. 1642a23d5eceSKris Buschelman 1643a23d5eceSKris Buschelman Collective on Mat 1644a23d5eceSKris Buschelman 1645a23d5eceSKris Buschelman Input Parameters: 1646a23d5eceSKris Buschelman + A - the symmetric matrix 1647a23d5eceSKris Buschelman . bs - size of block 1648a23d5eceSKris Buschelman . nz - number of block nonzeros per block row (same for all rows) 1649a23d5eceSKris Buschelman - nnz - array containing the number of block nonzeros in the upper triangular plus 1650a23d5eceSKris Buschelman diagonal portion of each block (possibly different for each block row) or PETSC_NULL 1651a23d5eceSKris Buschelman 1652a23d5eceSKris Buschelman Options Database Keys: 1653a23d5eceSKris Buschelman . -mat_no_unroll - uses code that does not unroll the loops in the 1654a23d5eceSKris Buschelman block calculations (much slower) 1655a23d5eceSKris Buschelman . -mat_block_size - size of the blocks to use 1656a23d5eceSKris Buschelman 1657a23d5eceSKris Buschelman Level: intermediate 1658a23d5eceSKris Buschelman 1659a23d5eceSKris Buschelman Notes: 1660a23d5eceSKris Buschelman Specify the preallocated storage with either nz or nnz (not both). 1661a23d5eceSKris Buschelman Set nz=PETSC_DEFAULT and nnz=PETSC_NULL for PETSc to control dynamic memory 1662a23d5eceSKris Buschelman allocation. For additional details, see the users manual chapter on 1663a23d5eceSKris Buschelman matrices. 1664a23d5eceSKris Buschelman 166549a6f317SBarry Smith If the nnz parameter is given then the nz parameter is ignored 166649a6f317SBarry Smith 166749a6f317SBarry Smith 1668a23d5eceSKris Buschelman .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateMPISBAIJ() 1669a23d5eceSKris Buschelman @*/ 167013f74950SBarry Smith PetscErrorCode MatSeqSBAIJSetPreallocation(Mat B,PetscInt bs,PetscInt nz,const PetscInt nnz[]) 167113f74950SBarry Smith { 167213f74950SBarry Smith PetscErrorCode ierr,(*f)(Mat,PetscInt,PetscInt,const PetscInt[]); 1673a23d5eceSKris Buschelman 1674a23d5eceSKris Buschelman PetscFunctionBegin; 1675a23d5eceSKris Buschelman ierr = PetscObjectQueryFunction((PetscObject)B,"MatSeqSBAIJSetPreallocation_C",(void (**)(void))&f);CHKERRQ(ierr); 1676a23d5eceSKris Buschelman if (f) { 1677a23d5eceSKris Buschelman ierr = (*f)(B,bs,nz,nnz);CHKERRQ(ierr); 1678a23d5eceSKris Buschelman } 1679a23d5eceSKris Buschelman PetscFunctionReturn(0); 1680a23d5eceSKris Buschelman } 168149b5e25fSSatish Balay 16824a2ae208SSatish Balay #undef __FUNCT__ 16834a2ae208SSatish Balay #define __FUNCT__ "MatCreateSeqSBAIJ" 1684c464158bSHong Zhang /*@C 1685c464158bSHong Zhang MatCreateSeqSBAIJ - Creates a sparse symmetric matrix in block AIJ (block 1686c464158bSHong Zhang compressed row) format. For good matrix assembly performance the 1687c464158bSHong Zhang user should preallocate the matrix storage by setting the parameter nz 1688c464158bSHong Zhang (or the array nnz). By setting these parameters accurately, performance 1689c464158bSHong Zhang during matrix assembly can be increased by more than a factor of 50. 169049b5e25fSSatish Balay 1691c464158bSHong Zhang Collective on MPI_Comm 1692c464158bSHong Zhang 1693c464158bSHong Zhang Input Parameters: 1694c464158bSHong Zhang + comm - MPI communicator, set to PETSC_COMM_SELF 1695c464158bSHong Zhang . bs - size of block 1696c464158bSHong Zhang . m - number of rows, or number of columns 1697c464158bSHong Zhang . nz - number of block nonzeros per block row (same for all rows) 1698744e8345SSatish Balay - nnz - array containing the number of block nonzeros in the upper triangular plus 1699744e8345SSatish Balay diagonal portion of each block (possibly different for each block row) or PETSC_NULL 1700c464158bSHong Zhang 1701c464158bSHong Zhang Output Parameter: 1702c464158bSHong Zhang . A - the symmetric matrix 1703c464158bSHong Zhang 1704c464158bSHong Zhang Options Database Keys: 1705c464158bSHong Zhang . -mat_no_unroll - uses code that does not unroll the loops in the 1706c464158bSHong Zhang block calculations (much slower) 1707c464158bSHong Zhang . -mat_block_size - size of the blocks to use 1708c464158bSHong Zhang 1709c464158bSHong Zhang Level: intermediate 1710c464158bSHong Zhang 1711c464158bSHong Zhang Notes: 1712c464158bSHong Zhang 1713c464158bSHong Zhang Specify the preallocated storage with either nz or nnz (not both). 1714c464158bSHong Zhang Set nz=PETSC_DEFAULT and nnz=PETSC_NULL for PETSc to control dynamic memory 1715c464158bSHong Zhang allocation. For additional details, see the users manual chapter on 1716c464158bSHong Zhang matrices. 1717c464158bSHong Zhang 171849a6f317SBarry Smith If the nnz parameter is given then the nz parameter is ignored 171949a6f317SBarry Smith 1720c464158bSHong Zhang .seealso: MatCreate(), MatCreateSeqAIJ(), MatSetValues(), MatCreateMPISBAIJ() 1721c464158bSHong Zhang @*/ 172213f74950SBarry Smith PetscErrorCode MatCreateSeqSBAIJ(MPI_Comm comm,PetscInt bs,PetscInt m,PetscInt n,PetscInt nz,const PetscInt nnz[],Mat *A) 1723c464158bSHong Zhang { 1724dfbe8321SBarry Smith PetscErrorCode ierr; 1725c464158bSHong Zhang 1726c464158bSHong Zhang PetscFunctionBegin; 1727c464158bSHong Zhang ierr = MatCreate(comm,m,n,m,n,A);CHKERRQ(ierr); 1728c464158bSHong Zhang ierr = MatSetType(*A,MATSEQSBAIJ);CHKERRQ(ierr); 1729273d9f13SBarry Smith ierr = MatSeqSBAIJSetPreallocation(*A,bs,nz,nnz);CHKERRQ(ierr); 173049b5e25fSSatish Balay PetscFunctionReturn(0); 173149b5e25fSSatish Balay } 173249b5e25fSSatish Balay 17334a2ae208SSatish Balay #undef __FUNCT__ 17344a2ae208SSatish Balay #define __FUNCT__ "MatDuplicate_SeqSBAIJ" 1735dfbe8321SBarry Smith PetscErrorCode MatDuplicate_SeqSBAIJ(Mat A,MatDuplicateOption cpvalues,Mat *B) 173649b5e25fSSatish Balay { 173749b5e25fSSatish Balay Mat C; 173849b5e25fSSatish Balay Mat_SeqSBAIJ *c,*a = (Mat_SeqSBAIJ*)A->data; 17396849ba73SBarry Smith PetscErrorCode ierr; 174013f74950SBarry Smith PetscInt i,len,mbs = a->mbs,nz = a->nz,bs2 =a->bs2; 174149b5e25fSSatish Balay 174249b5e25fSSatish Balay PetscFunctionBegin; 174329bbc08cSBarry Smith if (a->i[mbs] != nz) SETERRQ(PETSC_ERR_PLIB,"Corrupt matrix"); 174449b5e25fSSatish Balay 174549b5e25fSSatish Balay *B = 0; 1746692f9cbeSHong Zhang ierr = MatCreate(A->comm,A->m,A->n,A->m,A->n,&C);CHKERRQ(ierr); 1747be5d1d56SKris Buschelman ierr = MatSetType(C,A->type_name);CHKERRQ(ierr); 17481d5dac46SHong Zhang ierr = PetscMemcpy(C->ops,A->ops,sizeof(struct _MatOps));CHKERRQ(ierr); 1749692f9cbeSHong Zhang c = (Mat_SeqSBAIJ*)C->data; 1750692f9cbeSHong Zhang 1751273d9f13SBarry Smith C->preallocated = PETSC_TRUE; 175249b5e25fSSatish Balay C->factor = A->factor; 175349b5e25fSSatish Balay c->row = 0; 175449b5e25fSSatish Balay c->icol = 0; 175549b5e25fSSatish Balay c->saved_values = 0; 175649b5e25fSSatish Balay c->keepzeroedrows = a->keepzeroedrows; 175749b5e25fSSatish Balay C->assembled = PETSC_TRUE; 175849b5e25fSSatish Balay 175982327fa8SHong Zhang C->M = A->M; 176082327fa8SHong Zhang C->N = A->N; 1761521d7252SBarry Smith C->bs = A->bs; 176249b5e25fSSatish Balay c->bs2 = a->bs2; 176349b5e25fSSatish Balay c->mbs = a->mbs; 176449b5e25fSSatish Balay c->nbs = a->nbs; 176549b5e25fSSatish Balay 176613f74950SBarry Smith ierr = PetscMalloc((mbs+1)*sizeof(PetscInt),&c->imax);CHKERRQ(ierr); 176713f74950SBarry Smith ierr = PetscMalloc((mbs+1)*sizeof(PetscInt),&c->ilen);CHKERRQ(ierr); 176849b5e25fSSatish Balay for (i=0; i<mbs; i++) { 176949b5e25fSSatish Balay c->imax[i] = a->imax[i]; 177049b5e25fSSatish Balay c->ilen[i] = a->ilen[i]; 177149b5e25fSSatish Balay } 177249b5e25fSSatish Balay 177349b5e25fSSatish Balay /* allocate the matrix space */ 177449b5e25fSSatish Balay c->singlemalloc = PETSC_TRUE; 177513f74950SBarry Smith len = (mbs+1)*sizeof(PetscInt) + nz*(bs2*sizeof(MatScalar) + sizeof(PetscInt)); 177682502324SSatish Balay ierr = PetscMalloc(len,&c->a);CHKERRQ(ierr); 177713f74950SBarry Smith c->j = (PetscInt*)(c->a + nz*bs2); 177849b5e25fSSatish Balay c->i = c->j + nz; 177913f74950SBarry Smith ierr = PetscMemcpy(c->i,a->i,(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr); 178049b5e25fSSatish Balay if (mbs > 0) { 178113f74950SBarry Smith ierr = PetscMemcpy(c->j,a->j,nz*sizeof(PetscInt));CHKERRQ(ierr); 178249b5e25fSSatish Balay if (cpvalues == MAT_COPY_VALUES) { 178349b5e25fSSatish Balay ierr = PetscMemcpy(c->a,a->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr); 178449b5e25fSSatish Balay } else { 178549b5e25fSSatish Balay ierr = PetscMemzero(c->a,bs2*nz*sizeof(MatScalar));CHKERRQ(ierr); 178649b5e25fSSatish Balay } 178749b5e25fSSatish Balay } 178849b5e25fSSatish Balay 178952e6d16bSBarry Smith ierr = PetscLogObjectMemory(C,len+2*(mbs+1)*sizeof(PetscInt)+sizeof(struct _p_Mat)+sizeof(Mat_SeqSBAIJ));CHKERRQ(ierr); 179049b5e25fSSatish Balay c->sorted = a->sorted; 179149b5e25fSSatish Balay c->roworiented = a->roworiented; 179249b5e25fSSatish Balay c->nonew = a->nonew; 179349b5e25fSSatish Balay 179449b5e25fSSatish Balay if (a->diag) { 179513f74950SBarry Smith ierr = PetscMalloc((mbs+1)*sizeof(PetscInt),&c->diag);CHKERRQ(ierr); 179652e6d16bSBarry Smith ierr = PetscLogObjectMemory(C,(mbs+1)*sizeof(PetscInt));CHKERRQ(ierr); 179749b5e25fSSatish Balay for (i=0; i<mbs; i++) { 179849b5e25fSSatish Balay c->diag[i] = a->diag[i]; 179949b5e25fSSatish Balay } 180049b5e25fSSatish Balay } else c->diag = 0; 18016c6c5352SBarry Smith c->nz = a->nz; 18026c6c5352SBarry Smith c->maxnz = a->maxnz; 180349b5e25fSSatish Balay c->solve_work = 0; 180449b5e25fSSatish Balay c->mult_work = 0; 180549b5e25fSSatish Balay *B = C; 1806b0a32e0cSBarry Smith ierr = PetscFListDuplicate(A->qlist,&C->qlist);CHKERRQ(ierr); 180749b5e25fSSatish Balay PetscFunctionReturn(0); 180849b5e25fSSatish Balay } 180949b5e25fSSatish Balay 18104a2ae208SSatish Balay #undef __FUNCT__ 18114a2ae208SSatish Balay #define __FUNCT__ "MatLoad_SeqSBAIJ" 1812dfbe8321SBarry Smith PetscErrorCode MatLoad_SeqSBAIJ(PetscViewer viewer,const MatType type,Mat *A) 181349b5e25fSSatish Balay { 181449b5e25fSSatish Balay Mat_SeqSBAIJ *a; 181549b5e25fSSatish Balay Mat B; 18166849ba73SBarry Smith PetscErrorCode ierr; 181713f74950SBarry Smith int fd; 181813f74950SBarry Smith PetscMPIInt size; 181913f74950SBarry Smith PetscInt i,nz,header[4],*rowlengths=0,M,N,bs=1; 182013f74950SBarry Smith PetscInt *mask,mbs,*jj,j,rowcount,nzcount,k,*s_browlengths,maskcount; 182113f74950SBarry Smith PetscInt kmax,jcount,block,idx,point,nzcountb,extra_rows; 182213f74950SBarry Smith PetscInt *masked,nmask,tmp,bs2,ishift; 182387828ca2SBarry Smith PetscScalar *aa; 182449b5e25fSSatish Balay MPI_Comm comm = ((PetscObject)viewer)->comm; 182549b5e25fSSatish Balay 182649b5e25fSSatish Balay PetscFunctionBegin; 1827b0a32e0cSBarry Smith ierr = PetscOptionsGetInt(PETSC_NULL,"-matload_block_size",&bs,PETSC_NULL);CHKERRQ(ierr); 182849b5e25fSSatish Balay bs2 = bs*bs; 182949b5e25fSSatish Balay 183049b5e25fSSatish Balay ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 183129bbc08cSBarry Smith if (size > 1) SETERRQ(PETSC_ERR_ARG_WRONG,"view must have one processor"); 1832b0a32e0cSBarry Smith ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr); 183349b5e25fSSatish Balay ierr = PetscBinaryRead(fd,header,4,PETSC_INT);CHKERRQ(ierr); 1834552e946dSBarry Smith if (header[0] != MAT_FILE_COOKIE) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"not Mat object"); 183549b5e25fSSatish Balay M = header[1]; N = header[2]; nz = header[3]; 183649b5e25fSSatish Balay 183749b5e25fSSatish Balay if (header[3] < 0) { 183829bbc08cSBarry Smith SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"Matrix stored in special format, cannot load as SeqSBAIJ"); 183949b5e25fSSatish Balay } 184049b5e25fSSatish Balay 184129bbc08cSBarry Smith if (M != N) SETERRQ(PETSC_ERR_SUP,"Can only do square matrices"); 184249b5e25fSSatish Balay 184349b5e25fSSatish Balay /* 184449b5e25fSSatish Balay This code adds extra rows to make sure the number of rows is 184549b5e25fSSatish Balay divisible by the blocksize 184649b5e25fSSatish Balay */ 184749b5e25fSSatish Balay mbs = M/bs; 184849b5e25fSSatish Balay extra_rows = bs - M + bs*(mbs); 184949b5e25fSSatish Balay if (extra_rows == bs) extra_rows = 0; 185049b5e25fSSatish Balay else mbs++; 185149b5e25fSSatish Balay if (extra_rows) { 1852b0a32e0cSBarry Smith PetscLogInfo(0,"MatLoad_SeqSBAIJ:Padding loaded matrix to match blocksize\n"); 185349b5e25fSSatish Balay } 185449b5e25fSSatish Balay 185549b5e25fSSatish Balay /* read in row lengths */ 185613f74950SBarry Smith ierr = PetscMalloc((M+extra_rows)*sizeof(PetscInt),&rowlengths);CHKERRQ(ierr); 185749b5e25fSSatish Balay ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr); 185849b5e25fSSatish Balay for (i=0; i<extra_rows; i++) rowlengths[M+i] = 1; 185949b5e25fSSatish Balay 186049b5e25fSSatish Balay /* read in column indices */ 186113f74950SBarry Smith ierr = PetscMalloc((nz+extra_rows)*sizeof(PetscInt),&jj);CHKERRQ(ierr); 186249b5e25fSSatish Balay ierr = PetscBinaryRead(fd,jj,nz,PETSC_INT);CHKERRQ(ierr); 186349b5e25fSSatish Balay for (i=0; i<extra_rows; i++) jj[nz+i] = M+i; 186449b5e25fSSatish Balay 186549b5e25fSSatish Balay /* loop over row lengths determining block row lengths */ 186613f74950SBarry Smith ierr = PetscMalloc(mbs*sizeof(PetscInt),&s_browlengths);CHKERRQ(ierr); 186713f74950SBarry Smith ierr = PetscMemzero(s_browlengths,mbs*sizeof(PetscInt));CHKERRQ(ierr); 186813f74950SBarry Smith ierr = PetscMalloc(2*mbs*sizeof(PetscInt),&mask);CHKERRQ(ierr); 186913f74950SBarry Smith ierr = PetscMemzero(mask,mbs*sizeof(PetscInt));CHKERRQ(ierr); 187049b5e25fSSatish Balay masked = mask + mbs; 187149b5e25fSSatish Balay rowcount = 0; nzcount = 0; 187249b5e25fSSatish Balay for (i=0; i<mbs; i++) { 187349b5e25fSSatish Balay nmask = 0; 187449b5e25fSSatish Balay for (j=0; j<bs; j++) { 187549b5e25fSSatish Balay kmax = rowlengths[rowcount]; 187649b5e25fSSatish Balay for (k=0; k<kmax; k++) { 18772d703238SHong Zhang tmp = jj[nzcount++]/bs; /* block col. index */ 187803630b6eSHong Zhang if (!mask[tmp] && tmp >= i) {masked[nmask++] = tmp; mask[tmp] = 1;} 187949b5e25fSSatish Balay } 188049b5e25fSSatish Balay rowcount++; 188149b5e25fSSatish Balay } 1882574b2666SHong Zhang s_browlengths[i] += nmask; 1883574b2666SHong Zhang 188449b5e25fSSatish Balay /* zero out the mask elements we set */ 188549b5e25fSSatish Balay for (j=0; j<nmask; j++) mask[masked[j]] = 0; 188649b5e25fSSatish Balay } 188749b5e25fSSatish Balay 188849b5e25fSSatish Balay /* create our matrix */ 18899abb65ffSKris Buschelman ierr = MatCreate(comm,M+extra_rows,N+extra_rows,M+extra_rows,N+extra_rows,&B);CHKERRQ(ierr); 18909abb65ffSKris Buschelman ierr = MatSetType(B,type);CHKERRQ(ierr); 189178473fd7SKris Buschelman ierr = MatSeqSBAIJSetPreallocation(B,bs,0,s_browlengths);CHKERRQ(ierr); 189249b5e25fSSatish Balay a = (Mat_SeqSBAIJ*)B->data; 189349b5e25fSSatish Balay 189449b5e25fSSatish Balay /* set matrix "i" values */ 189549b5e25fSSatish Balay a->i[0] = 0; 189649b5e25fSSatish Balay for (i=1; i<= mbs; i++) { 1897574b2666SHong Zhang a->i[i] = a->i[i-1] + s_browlengths[i-1]; 1898574b2666SHong Zhang a->ilen[i-1] = s_browlengths[i-1]; 189949b5e25fSSatish Balay } 19006c6c5352SBarry Smith a->nz = a->i[mbs]; 190149b5e25fSSatish Balay 190249b5e25fSSatish Balay /* read in nonzero values */ 190387828ca2SBarry Smith ierr = PetscMalloc((nz+extra_rows)*sizeof(PetscScalar),&aa);CHKERRQ(ierr); 190449b5e25fSSatish Balay ierr = PetscBinaryRead(fd,aa,nz,PETSC_SCALAR);CHKERRQ(ierr); 190549b5e25fSSatish Balay for (i=0; i<extra_rows; i++) aa[nz+i] = 1.0; 190649b5e25fSSatish Balay 190749b5e25fSSatish Balay /* set "a" and "j" values into matrix */ 190849b5e25fSSatish Balay nzcount = 0; jcount = 0; 190949b5e25fSSatish Balay for (i=0; i<mbs; i++) { 191049b5e25fSSatish Balay nzcountb = nzcount; 191149b5e25fSSatish Balay nmask = 0; 191249b5e25fSSatish Balay for (j=0; j<bs; j++) { 191349b5e25fSSatish Balay kmax = rowlengths[i*bs+j]; 191449b5e25fSSatish Balay for (k=0; k<kmax; k++) { 19152d703238SHong Zhang tmp = jj[nzcount++]/bs; /* block col. index */ 191603630b6eSHong Zhang if (!mask[tmp] && tmp >= i) { masked[nmask++] = tmp; mask[tmp] = 1;} 19172d703238SHong Zhang } 19182d703238SHong Zhang } 19192d703238SHong Zhang /* sort the masked values */ 19202d703238SHong Zhang ierr = PetscSortInt(nmask,masked);CHKERRQ(ierr); 19212d703238SHong Zhang 19222d703238SHong Zhang /* set "j" values into matrix */ 19232d703238SHong Zhang maskcount = 1; 19242d703238SHong Zhang for (j=0; j<nmask; j++) { 192549b5e25fSSatish Balay a->j[jcount++] = masked[j]; 192649b5e25fSSatish Balay mask[masked[j]] = maskcount++; 192749b5e25fSSatish Balay } 1928574b2666SHong Zhang 192949b5e25fSSatish Balay /* set "a" values into matrix */ 193049b5e25fSSatish Balay ishift = bs2*a->i[i]; 193149b5e25fSSatish Balay for (j=0; j<bs; j++) { 193249b5e25fSSatish Balay kmax = rowlengths[i*bs+j]; 193349b5e25fSSatish Balay for (k=0; k<kmax; k++) { 1934574b2666SHong Zhang tmp = jj[nzcountb]/bs ; /* block col. index */ 1935574b2666SHong Zhang if (tmp >= i){ 193649b5e25fSSatish Balay block = mask[tmp] - 1; 193749b5e25fSSatish Balay point = jj[nzcountb] - bs*tmp; 193849b5e25fSSatish Balay idx = ishift + bs2*block + j + bs*point; 1939574b2666SHong Zhang a->a[idx] = aa[nzcountb]; 1940574b2666SHong Zhang } 1941574b2666SHong Zhang nzcountb++; 194249b5e25fSSatish Balay } 194349b5e25fSSatish Balay } 194449b5e25fSSatish Balay /* zero out the mask elements we set */ 194549b5e25fSSatish Balay for (j=0; j<nmask; j++) mask[masked[j]] = 0; 194649b5e25fSSatish Balay } 19476c6c5352SBarry Smith if (jcount != a->nz) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"Bad binary matrix"); 194849b5e25fSSatish Balay 194949b5e25fSSatish Balay ierr = PetscFree(rowlengths);CHKERRQ(ierr); 1950574b2666SHong Zhang ierr = PetscFree(s_browlengths);CHKERRQ(ierr); 195149b5e25fSSatish Balay ierr = PetscFree(aa);CHKERRQ(ierr); 195249b5e25fSSatish Balay ierr = PetscFree(jj);CHKERRQ(ierr); 195349b5e25fSSatish Balay ierr = PetscFree(mask);CHKERRQ(ierr); 195449b5e25fSSatish Balay 19559abb65ffSKris Buschelman ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 19569abb65ffSKris Buschelman ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 195749b5e25fSSatish Balay ierr = MatView_Private(B);CHKERRQ(ierr); 19589abb65ffSKris Buschelman *A = B; 195949b5e25fSSatish Balay PetscFunctionReturn(0); 196049b5e25fSSatish Balay } 1961574b2666SHong Zhang 1962d06b337dSHong Zhang #undef __FUNCT__ 1963d06b337dSHong Zhang #define __FUNCT__ "MatRelax_SeqSBAIJ" 196413f74950SBarry Smith PetscErrorCode MatRelax_SeqSBAIJ(Mat A,Vec bb,PetscReal omega,MatSORType flag,PetscReal fshift,PetscInt its,PetscInt lits,Vec xx) 1965d06b337dSHong Zhang { 1966d06b337dSHong Zhang Mat_SeqSBAIJ *a = (Mat_SeqSBAIJ*)A->data; 1967d06b337dSHong Zhang MatScalar *aa=a->a,*v,*v1; 1968d06b337dSHong Zhang PetscScalar *x,*b,*t,sum,d; 19696849ba73SBarry Smith PetscErrorCode ierr; 1970521d7252SBarry Smith PetscInt m=a->mbs,bs=A->bs,*ai=a->i,*aj=a->j; 1971521d7252SBarry Smith PetscInt nz,nz1,*vj,*vj1,i; 1972d06b337dSHong Zhang 1973d06b337dSHong Zhang PetscFunctionBegin; 1974b965ef7fSBarry Smith its = its*lits; 197577431f27SBarry Smith if (its <= 0) SETERRQ2(PETSC_ERR_ARG_WRONG,"Relaxation requires global its %D and local its %D both positive",its,lits); 1976d06b337dSHong Zhang 1977d06b337dSHong Zhang if (bs > 1) 1978d06b337dSHong Zhang SETERRQ(PETSC_ERR_SUP,"SSOR for block size > 1 is not yet implemented"); 1979d06b337dSHong Zhang 19801ebc52fbSHong Zhang ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1981d06b337dSHong Zhang if (xx != bb) { 19821ebc52fbSHong Zhang ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 1983d06b337dSHong Zhang } else { 1984d06b337dSHong Zhang b = x; 1985d06b337dSHong Zhang } 1986d06b337dSHong Zhang 1987d06b337dSHong Zhang ierr = PetscMalloc(m*sizeof(PetscScalar),&t);CHKERRQ(ierr); 1988d06b337dSHong Zhang 1989d06b337dSHong Zhang if (flag & SOR_ZERO_INITIAL_GUESS) { 19902798e883SHong Zhang if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){ 1991d06b337dSHong Zhang for (i=0; i<m; i++) 1992d06b337dSHong Zhang t[i] = b[i]; 1993d06b337dSHong Zhang 1994d06b337dSHong Zhang for (i=0; i<m; i++){ 199544706875SHong Zhang d = *(aa + ai[i]); /* diag[i] */ 1996d06b337dSHong Zhang v = aa + ai[i] + 1; 1997d06b337dSHong Zhang vj = aj + ai[i] + 1; 1998d06b337dSHong Zhang nz = ai[i+1] - ai[i] - 1; 1999d06b337dSHong Zhang x[i] = omega*t[i]/d; 2000d06b337dSHong Zhang while (nz--) t[*vj++] -= x[i]*(*v++); /* update rhs */ 200144706875SHong Zhang PetscLogFlops(2*nz-1); 2002d06b337dSHong Zhang } 2003d06b337dSHong Zhang } 2004d06b337dSHong Zhang 20052798e883SHong Zhang if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){ 2006d06b337dSHong Zhang for (i=0; i<m; i++) 2007d06b337dSHong Zhang t[i] = b[i]; 2008d06b337dSHong Zhang 2009d06b337dSHong Zhang for (i=0; i<m-1; i++){ /* update rhs */ 2010d06b337dSHong Zhang v = aa + ai[i] + 1; 2011d06b337dSHong Zhang vj = aj + ai[i] + 1; 2012d06b337dSHong Zhang nz = ai[i+1] - ai[i] - 1; 2013d06b337dSHong Zhang while (nz--) t[*vj++] -= x[i]*(*v++); 201444706875SHong Zhang PetscLogFlops(2*nz-1); 2015d06b337dSHong Zhang } 2016d06b337dSHong Zhang for (i=m-1; i>=0; i--){ 2017d06b337dSHong Zhang d = *(aa + ai[i]); 2018d06b337dSHong Zhang v = aa + ai[i] + 1; 2019d06b337dSHong Zhang vj = aj + ai[i] + 1; 2020d06b337dSHong Zhang nz = ai[i+1] - ai[i] - 1; 2021d06b337dSHong Zhang sum = t[i]; 2022d06b337dSHong Zhang while (nz--) sum -= x[*vj++]*(*v++); 202344706875SHong Zhang PetscLogFlops(2*nz-1); 2024d06b337dSHong Zhang x[i] = (1-omega)*x[i] + omega*sum/d; 2025d06b337dSHong Zhang } 2026d06b337dSHong Zhang } 2027d06b337dSHong Zhang its--; 2028d06b337dSHong Zhang } 2029d06b337dSHong Zhang 2030d06b337dSHong Zhang while (its--) { 203144706875SHong Zhang /* 203244706875SHong Zhang forward sweep: 203344706875SHong Zhang for i=0,...,m-1: 203444706875SHong Zhang sum[i] = (b[i] - U(i,:)x )/d[i]; 203544706875SHong Zhang x[i] = (1-omega)x[i] + omega*sum[i]; 203644706875SHong Zhang b = b - x[i]*U^T(i,:); 2037d06b337dSHong Zhang 203844706875SHong Zhang */ 20392798e883SHong Zhang if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){ 2040d06b337dSHong Zhang for (i=0; i<m; i++) 2041d06b337dSHong Zhang t[i] = b[i]; 2042d06b337dSHong Zhang 2043d06b337dSHong Zhang for (i=0; i<m; i++){ 204444706875SHong Zhang d = *(aa + ai[i]); /* diag[i] */ 2045d06b337dSHong Zhang v = aa + ai[i] + 1; v1=v; 2046d06b337dSHong Zhang vj = aj + ai[i] + 1; vj1=vj; 2047d06b337dSHong Zhang nz = ai[i+1] - ai[i] - 1; nz1=nz; 2048d06b337dSHong Zhang sum = t[i]; 2049d06b337dSHong Zhang while (nz1--) sum -= (*v1++)*x[*vj1++]; 2050d06b337dSHong Zhang x[i] = (1-omega)*x[i] + omega*sum/d; 2051d06b337dSHong Zhang while (nz--) t[*vj++] -= x[i]*(*v++); 205244706875SHong Zhang PetscLogFlops(4*nz-2); 2053d06b337dSHong Zhang } 2054d06b337dSHong Zhang } 2055d06b337dSHong Zhang 20562798e883SHong Zhang if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){ 205744706875SHong Zhang /* 205844706875SHong Zhang backward sweep: 205944706875SHong Zhang b = b - x[i]*U^T(i,:), i=0,...,n-2 206044706875SHong Zhang for i=m-1,...,0: 206144706875SHong Zhang sum[i] = (b[i] - U(i,:)x )/d[i]; 206244706875SHong Zhang x[i] = (1-omega)x[i] + omega*sum[i]; 206344706875SHong Zhang */ 2064d06b337dSHong Zhang for (i=0; i<m; i++) 2065d06b337dSHong Zhang t[i] = b[i]; 2066d06b337dSHong Zhang 2067d06b337dSHong Zhang for (i=0; i<m-1; i++){ /* update rhs */ 2068d06b337dSHong Zhang v = aa + ai[i] + 1; 2069d06b337dSHong Zhang vj = aj + ai[i] + 1; 2070d06b337dSHong Zhang nz = ai[i+1] - ai[i] - 1; 2071d06b337dSHong Zhang while (nz--) t[*vj++] -= x[i]*(*v++); 207244706875SHong Zhang PetscLogFlops(2*nz-1); 2073d06b337dSHong Zhang } 2074d06b337dSHong Zhang for (i=m-1; i>=0; i--){ 2075d06b337dSHong Zhang d = *(aa + ai[i]); 2076d06b337dSHong Zhang v = aa + ai[i] + 1; 2077d06b337dSHong Zhang vj = aj + ai[i] + 1; 2078d06b337dSHong Zhang nz = ai[i+1] - ai[i] - 1; 2079d06b337dSHong Zhang sum = t[i]; 2080d06b337dSHong Zhang while (nz--) sum -= x[*vj++]*(*v++); 208144706875SHong Zhang PetscLogFlops(2*nz-1); 2082d06b337dSHong Zhang x[i] = (1-omega)*x[i] + omega*sum/d; 2083d06b337dSHong Zhang } 2084d06b337dSHong Zhang } 2085d06b337dSHong Zhang } 2086d06b337dSHong Zhang 2087d06b337dSHong Zhang ierr = PetscFree(t);CHKERRQ(ierr); 20881ebc52fbSHong Zhang ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 2089d06b337dSHong Zhang if (bb != xx) { 20901ebc52fbSHong Zhang ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr); 2091d06b337dSHong Zhang } 2092d06b337dSHong Zhang PetscFunctionReturn(0); 2093d06b337dSHong Zhang } 2094d06b337dSHong Zhang 2095d06b337dSHong Zhang 2096d06b337dSHong Zhang 2097d06b337dSHong Zhang 209849b5e25fSSatish Balay 209949b5e25fSSatish Balay 2100