1a5eb4965SSatish Balay #ifdef PETSC_RCS_HEADER 2*0f5bd95cSBarry Smith static char vcid[] = "$Id: dense.c,v 1.174 1999/10/01 21:21:11 bsmith Exp bsmith $"; 3cb512458SBarry Smith #endif 467e560aaSBarry Smith /* 567e560aaSBarry Smith Defines the basic matrix operations for sequential dense. 667e560aaSBarry Smith */ 7289bc588SBarry Smith 870f55243SBarry Smith #include "src/mat/impls/dense/seq/dense.h" 9eadb2fb4SBarry Smith #include "pinclude/blaslapack.h" 10289bc588SBarry Smith 115615d1e5SSatish Balay #undef __FUNC__ 125615d1e5SSatish Balay #define __FUNC__ "MatAXPY_SeqDense" 131987afe7SBarry Smith int MatAXPY_SeqDense(Scalar *alpha,Mat X,Mat Y) 141987afe7SBarry Smith { 151987afe7SBarry Smith Mat_SeqDense *x = (Mat_SeqDense*) X->data,*y = (Mat_SeqDense*) Y->data; 161987afe7SBarry Smith int N = x->m*x->n, one = 1; 173a40ed3dSBarry Smith 183a40ed3dSBarry Smith PetscFunctionBegin; 191987afe7SBarry Smith BLaxpy_( &N, alpha, x->v, &one, y->v, &one ); 201987afe7SBarry Smith PLogFlops(2*N-1); 213a40ed3dSBarry Smith PetscFunctionReturn(0); 221987afe7SBarry Smith } 231987afe7SBarry Smith 245615d1e5SSatish Balay #undef __FUNC__ 255615d1e5SSatish Balay #define __FUNC__ "MatGetInfo_SeqDense" 268f6be9afSLois Curfman McInnes int MatGetInfo_SeqDense(Mat A,MatInfoType flag,MatInfo *info) 27289bc588SBarry Smith { 284e220ebcSLois Curfman McInnes Mat_SeqDense *a = (Mat_SeqDense *) A->data; 294e220ebcSLois Curfman McInnes int i,N = a->m*a->n,count = 0; 304e220ebcSLois Curfman McInnes Scalar *v = a->v; 313a40ed3dSBarry Smith 323a40ed3dSBarry Smith PetscFunctionBegin; 33289bc588SBarry Smith for ( i=0; i<N; i++ ) {if (*v != 0.0) count++; v++;} 344e220ebcSLois Curfman McInnes 354e220ebcSLois Curfman McInnes info->rows_global = (double)a->m; 364e220ebcSLois Curfman McInnes info->columns_global = (double)a->n; 374e220ebcSLois Curfman McInnes info->rows_local = (double)a->m; 384e220ebcSLois Curfman McInnes info->columns_local = (double)a->n; 394e220ebcSLois Curfman McInnes info->block_size = 1.0; 404e220ebcSLois Curfman McInnes info->nz_allocated = (double)N; 414e220ebcSLois Curfman McInnes info->nz_used = (double)count; 424e220ebcSLois Curfman McInnes info->nz_unneeded = (double)(N-count); 434e220ebcSLois Curfman McInnes info->assemblies = (double)A->num_ass; 444e220ebcSLois Curfman McInnes info->mallocs = 0; 454e220ebcSLois Curfman McInnes info->memory = A->mem; 464e220ebcSLois Curfman McInnes info->fill_ratio_given = 0; 474e220ebcSLois Curfman McInnes info->fill_ratio_needed = 0; 484e220ebcSLois Curfman McInnes info->factor_mallocs = 0; 494e220ebcSLois Curfman McInnes 503a40ed3dSBarry Smith PetscFunctionReturn(0); 51289bc588SBarry Smith } 52289bc588SBarry Smith 535615d1e5SSatish Balay #undef __FUNC__ 545615d1e5SSatish Balay #define __FUNC__ "MatScale_SeqDense" 558f6be9afSLois Curfman McInnes int MatScale_SeqDense(Scalar *alpha,Mat inA) 5680cd9d93SLois Curfman McInnes { 5780cd9d93SLois Curfman McInnes Mat_SeqDense *a = (Mat_SeqDense *) inA->data; 5880cd9d93SLois Curfman McInnes int one = 1, nz; 5980cd9d93SLois Curfman McInnes 603a40ed3dSBarry Smith PetscFunctionBegin; 6180cd9d93SLois Curfman McInnes nz = a->m*a->n; 6280cd9d93SLois Curfman McInnes BLscal_( &nz, alpha, a->v, &one ); 6380cd9d93SLois Curfman McInnes PLogFlops(nz); 643a40ed3dSBarry Smith PetscFunctionReturn(0); 6580cd9d93SLois Curfman McInnes } 6680cd9d93SLois Curfman McInnes 67289bc588SBarry Smith /* ---------------------------------------------------------------*/ 68289bc588SBarry Smith /* COMMENT: I have chosen to hide column permutation in the pivots, 69289bc588SBarry Smith rather than put it in the Mat->col slot.*/ 705615d1e5SSatish Balay #undef __FUNC__ 715615d1e5SSatish Balay #define __FUNC__ "MatLUFactor_SeqDense" 728f6be9afSLois Curfman McInnes int MatLUFactor_SeqDense(Mat A,IS row,IS col,double f) 73289bc588SBarry Smith { 74c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 756abc6512SBarry Smith int info; 7655659b69SBarry Smith 773a40ed3dSBarry Smith PetscFunctionBegin; 78289bc588SBarry Smith if (!mat->pivots) { 7945d48df9SBarry Smith mat->pivots = (int *) PetscMalloc((mat->m+1)*sizeof(int));CHKPTRQ(mat->pivots); 80c0bbcb79SLois Curfman McInnes PLogObjectMemory(A,mat->m*sizeof(int)); 81289bc588SBarry Smith } 827a97a34bSBarry Smith if (!mat->m || !mat->n) PetscFunctionReturn(0); 83289bc588SBarry Smith LAgetrf_(&mat->m,&mat->n,mat->v,&mat->m,mat->pivots,&info); 84a8c6a408SBarry Smith if (info<0) SETERRQ(PETSC_ERR_LIB,0,"Bad argument to LU factorization"); 85e3372554SBarry Smith if (info>0) SETERRQ(PETSC_ERR_MAT_LU_ZRPVT,0,"Bad LU factorization"); 86c0bbcb79SLois Curfman McInnes A->factor = FACTOR_LU; 8755659b69SBarry Smith PLogFlops((2*mat->n*mat->n*mat->n)/3); 883a40ed3dSBarry Smith PetscFunctionReturn(0); 89289bc588SBarry Smith } 906ee01492SSatish Balay 915615d1e5SSatish Balay #undef __FUNC__ 925609ef8eSBarry Smith #define __FUNC__ "MatDuplicate_SeqDense" 935609ef8eSBarry Smith int MatDuplicate_SeqDense(Mat A,MatDuplicateOption cpvalues,Mat *newmat) 9402cad45dSBarry Smith { 9502cad45dSBarry Smith Mat_SeqDense *mat = (Mat_SeqDense *) A->data, *l; 9602cad45dSBarry Smith int ierr; 9702cad45dSBarry Smith Mat newi; 9802cad45dSBarry Smith 993a40ed3dSBarry Smith PetscFunctionBegin; 10002cad45dSBarry Smith ierr = MatCreateSeqDense(A->comm,mat->m,mat->n,PETSC_NULL,&newi);CHKERRQ(ierr); 10102cad45dSBarry Smith l = (Mat_SeqDense *) newi->data; 1025609ef8eSBarry Smith if (cpvalues == MAT_COPY_VALUES) { 103549d3d68SSatish Balay ierr = PetscMemcpy(l->v,mat->v,mat->m*mat->n*sizeof(Scalar));CHKERRQ(ierr); 10402cad45dSBarry Smith } 10502cad45dSBarry Smith newi->assembled = PETSC_TRUE; 10602cad45dSBarry Smith *newmat = newi; 1073a40ed3dSBarry Smith PetscFunctionReturn(0); 10802cad45dSBarry Smith } 10902cad45dSBarry Smith 1105615d1e5SSatish Balay #undef __FUNC__ 1115615d1e5SSatish Balay #define __FUNC__ "MatLUFactorSymbolic_SeqDense" 1128f6be9afSLois Curfman McInnes int MatLUFactorSymbolic_SeqDense(Mat A,IS row,IS col,double f,Mat *fact) 113289bc588SBarry Smith { 1143a40ed3dSBarry Smith int ierr; 1153a40ed3dSBarry Smith 1163a40ed3dSBarry Smith PetscFunctionBegin; 1175609ef8eSBarry Smith ierr = MatDuplicate_SeqDense(A,MAT_DO_NOT_COPY_VALUES,fact);CHKERRQ(ierr); 1183a40ed3dSBarry Smith PetscFunctionReturn(0); 119289bc588SBarry Smith } 1206ee01492SSatish Balay 1215615d1e5SSatish Balay #undef __FUNC__ 1225615d1e5SSatish Balay #define __FUNC__ "MatLUFactorNumeric_SeqDense" 1238f6be9afSLois Curfman McInnes int MatLUFactorNumeric_SeqDense(Mat A,Mat *fact) 124289bc588SBarry Smith { 12502cad45dSBarry Smith Mat_SeqDense *mat = (Mat_SeqDense*) A->data, *l = (Mat_SeqDense*) (*fact)->data; 1263a40ed3dSBarry Smith int ierr; 1273a40ed3dSBarry Smith 1283a40ed3dSBarry Smith PetscFunctionBegin; 12902cad45dSBarry Smith /* copy the numerical values */ 130549d3d68SSatish Balay ierr = PetscMemcpy(l->v,mat->v,mat->m*mat->n*sizeof(Scalar));CHKERRQ(ierr); 13102cad45dSBarry Smith (*fact)->factor = 0; 1323a40ed3dSBarry Smith ierr = MatLUFactor(*fact,0,0,1.0);CHKERRQ(ierr); 1333a40ed3dSBarry Smith PetscFunctionReturn(0); 134289bc588SBarry Smith } 1356ee01492SSatish Balay 1365615d1e5SSatish Balay #undef __FUNC__ 1375615d1e5SSatish Balay #define __FUNC__ "MatCholeskyFactorSymbolic_SeqDense" 1388f6be9afSLois Curfman McInnes int MatCholeskyFactorSymbolic_SeqDense(Mat A,IS row,double f,Mat *fact) 139289bc588SBarry Smith { 1403a40ed3dSBarry Smith int ierr; 1413a40ed3dSBarry Smith 1423a40ed3dSBarry Smith PetscFunctionBegin; 1433a40ed3dSBarry Smith ierr = MatConvert(A,MATSAME,fact);CHKERRQ(ierr); 1443a40ed3dSBarry Smith PetscFunctionReturn(0); 145289bc588SBarry Smith } 1466ee01492SSatish Balay 1475615d1e5SSatish Balay #undef __FUNC__ 1485615d1e5SSatish Balay #define __FUNC__ "MatCholeskyFactorNumeric_SeqDense" 1498f6be9afSLois Curfman McInnes int MatCholeskyFactorNumeric_SeqDense(Mat A,Mat *fact) 150289bc588SBarry Smith { 1513a40ed3dSBarry Smith int ierr; 1523a40ed3dSBarry Smith 1533a40ed3dSBarry Smith PetscFunctionBegin; 1543a40ed3dSBarry Smith ierr = MatCholeskyFactor(*fact,0,1.0);CHKERRQ(ierr); 1553a40ed3dSBarry Smith PetscFunctionReturn(0); 156289bc588SBarry Smith } 1576ee01492SSatish Balay 1585615d1e5SSatish Balay #undef __FUNC__ 1595615d1e5SSatish Balay #define __FUNC__ "MatCholeskyFactor_SeqDense" 1608f6be9afSLois Curfman McInnes int MatCholeskyFactor_SeqDense(Mat A,IS perm,double f) 161289bc588SBarry Smith { 162c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 163606d414cSSatish Balay int info,ierr; 16455659b69SBarry Smith 1653a40ed3dSBarry Smith PetscFunctionBegin; 166752f5567SLois Curfman McInnes if (mat->pivots) { 167606d414cSSatish Balay ierr = PetscFree(mat->pivots);CHKERRQ(ierr); 168c0bbcb79SLois Curfman McInnes PLogObjectMemory(A,-mat->m*sizeof(int)); 169752f5567SLois Curfman McInnes mat->pivots = 0; 170752f5567SLois Curfman McInnes } 1717a97a34bSBarry Smith if (!mat->m || !mat->n) PetscFunctionReturn(0); 172289bc588SBarry Smith LApotrf_("L",&mat->n,mat->v,&mat->m,&info); 173e3372554SBarry Smith if (info) SETERRQ(PETSC_ERR_MAT_CH_ZRPVT,0,"Bad factorization"); 174c0bbcb79SLois Curfman McInnes A->factor = FACTOR_CHOLESKY; 17555659b69SBarry Smith PLogFlops((mat->n*mat->n*mat->n)/3); 1763a40ed3dSBarry Smith PetscFunctionReturn(0); 177289bc588SBarry Smith } 178289bc588SBarry Smith 1795615d1e5SSatish Balay #undef __FUNC__ 1805615d1e5SSatish Balay #define __FUNC__ "MatSolve_SeqDense" 1818f6be9afSLois Curfman McInnes int MatSolve_SeqDense(Mat A,Vec xx,Vec yy) 182289bc588SBarry Smith { 183c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 184a57ad546SLois Curfman McInnes int one = 1, info, ierr; 1856abc6512SBarry Smith Scalar *x, *y; 18667e560aaSBarry Smith 1873a40ed3dSBarry Smith PetscFunctionBegin; 1887a97a34bSBarry Smith if (!mat->m || !mat->n) PetscFunctionReturn(0); 1897a97a34bSBarry Smith ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 1907a97a34bSBarry Smith ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 191549d3d68SSatish Balay ierr = PetscMemcpy(y,x,mat->m*sizeof(Scalar));CHKERRQ(ierr); 192c0bbcb79SLois Curfman McInnes if (A->factor == FACTOR_LU) { 19348d91487SBarry Smith LAgetrs_( "N", &mat->m, &one, mat->v, &mat->m, mat->pivots,y, &mat->m, &info ); 1947a97a34bSBarry Smith } else if (A->factor == FACTOR_CHOLESKY){ 19548d91487SBarry Smith LApotrs_( "L", &mat->m, &one, mat->v, &mat->m,y, &mat->m, &info ); 196289bc588SBarry Smith } 197a8c6a408SBarry Smith else SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Matrix must be factored to solve"); 198a8c6a408SBarry Smith if (info) SETERRQ(PETSC_ERR_LIB,0,"MBad solve"); 1997a97a34bSBarry Smith ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 2007a97a34bSBarry Smith ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 20155659b69SBarry Smith PLogFlops(mat->n*mat->n - mat->n); 2023a40ed3dSBarry Smith PetscFunctionReturn(0); 203289bc588SBarry Smith } 2046ee01492SSatish Balay 2055615d1e5SSatish Balay #undef __FUNC__ 2065615d1e5SSatish Balay #define __FUNC__ "MatSolveTrans_SeqDense" 2078f6be9afSLois Curfman McInnes int MatSolveTrans_SeqDense(Mat A,Vec xx,Vec yy) 208da3a660dSBarry Smith { 209c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 2107a97a34bSBarry Smith int ierr,one = 1, info; 2116abc6512SBarry Smith Scalar *x, *y; 21267e560aaSBarry Smith 2133a40ed3dSBarry Smith PetscFunctionBegin; 2147a97a34bSBarry Smith if (!mat->m || !mat->n) PetscFunctionReturn(0); 2157a97a34bSBarry Smith ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 2167a97a34bSBarry Smith ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 217549d3d68SSatish Balay ierr = PetscMemcpy(y,x,mat->m*sizeof(Scalar));CHKERRQ(ierr); 218752f5567SLois Curfman McInnes /* assume if pivots exist then use LU; else Cholesky */ 219da3a660dSBarry Smith if (mat->pivots) { 22048d91487SBarry Smith LAgetrs_( "T", &mat->m, &one, mat->v, &mat->m, mat->pivots,y, &mat->m, &info ); 2217a97a34bSBarry Smith } else { 22248d91487SBarry Smith LApotrs_( "L", &mat->m, &one, mat->v, &mat->m,y, &mat->m, &info ); 223da3a660dSBarry Smith } 224a8c6a408SBarry Smith if (info) SETERRQ(PETSC_ERR_LIB,0,"Bad solve"); 2257a97a34bSBarry Smith ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 2267a97a34bSBarry Smith ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 22755659b69SBarry Smith PLogFlops(mat->n*mat->n - mat->n); 2283a40ed3dSBarry Smith PetscFunctionReturn(0); 229da3a660dSBarry Smith } 2306ee01492SSatish Balay 2315615d1e5SSatish Balay #undef __FUNC__ 2325615d1e5SSatish Balay #define __FUNC__ "MatSolveAdd_SeqDense" 2338f6be9afSLois Curfman McInnes int MatSolveAdd_SeqDense(Mat A,Vec xx,Vec zz,Vec yy) 234da3a660dSBarry Smith { 235c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 2367a97a34bSBarry Smith int ierr,one = 1, info; 2376abc6512SBarry Smith Scalar *x, *y, sone = 1.0; 238da3a660dSBarry Smith Vec tmp = 0; 23967e560aaSBarry Smith 2403a40ed3dSBarry Smith PetscFunctionBegin; 2417a97a34bSBarry Smith ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 2427a97a34bSBarry Smith ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 2437a97a34bSBarry Smith if (!mat->m || !mat->n) PetscFunctionReturn(0); 244da3a660dSBarry Smith if (yy == zz) { 24578b31e54SBarry Smith ierr = VecDuplicate(yy,&tmp);CHKERRQ(ierr); 246c0bbcb79SLois Curfman McInnes PLogObjectParent(A,tmp); 24778b31e54SBarry Smith ierr = VecCopy(yy,tmp);CHKERRQ(ierr); 248da3a660dSBarry Smith } 249549d3d68SSatish Balay ierr = PetscMemcpy(y,x,mat->m*sizeof(Scalar));CHKERRQ(ierr); 250752f5567SLois Curfman McInnes /* assume if pivots exist then use LU; else Cholesky */ 251da3a660dSBarry Smith if (mat->pivots) { 25248d91487SBarry Smith LAgetrs_( "N", &mat->m, &one, mat->v, &mat->m, mat->pivots,y, &mat->m, &info ); 253a8c6a408SBarry Smith } else { 25448d91487SBarry Smith LApotrs_( "L", &mat->m, &one, mat->v, &mat->m,y, &mat->m, &info ); 255da3a660dSBarry Smith } 256a8c6a408SBarry Smith if (info) SETERRQ(PETSC_ERR_LIB,0,"Bad solve"); 257a8c6a408SBarry Smith if (tmp) {ierr = VecAXPY(&sone,tmp,yy);CHKERRQ(ierr); ierr = VecDestroy(tmp);CHKERRQ(ierr);} 258a8c6a408SBarry Smith else {ierr = VecAXPY(&sone,zz,yy);CHKERRQ(ierr);} 2597a97a34bSBarry Smith ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 2607a97a34bSBarry Smith ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 26155659b69SBarry Smith PLogFlops(mat->n*mat->n - mat->n); 2623a40ed3dSBarry Smith PetscFunctionReturn(0); 263da3a660dSBarry Smith } 26467e560aaSBarry Smith 2655615d1e5SSatish Balay #undef __FUNC__ 2665615d1e5SSatish Balay #define __FUNC__ "MatSolveTransAdd_SeqDense" 2678f6be9afSLois Curfman McInnes int MatSolveTransAdd_SeqDense(Mat A,Vec xx,Vec zz, Vec yy) 268da3a660dSBarry Smith { 269c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 2706abc6512SBarry Smith int one = 1, info,ierr; 2716abc6512SBarry Smith Scalar *x, *y, sone = 1.0; 272da3a660dSBarry Smith Vec tmp; 27367e560aaSBarry Smith 2743a40ed3dSBarry Smith PetscFunctionBegin; 2757a97a34bSBarry Smith if (!mat->m || !mat->n) PetscFunctionReturn(0); 2767a97a34bSBarry Smith ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 2777a97a34bSBarry Smith ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 278da3a660dSBarry Smith if (yy == zz) { 27978b31e54SBarry Smith ierr = VecDuplicate(yy,&tmp);CHKERRQ(ierr); 280c0bbcb79SLois Curfman McInnes PLogObjectParent(A,tmp); 28178b31e54SBarry Smith ierr = VecCopy(yy,tmp);CHKERRQ(ierr); 282da3a660dSBarry Smith } 283549d3d68SSatish Balay ierr = PetscMemcpy(y,x,mat->m*sizeof(Scalar));CHKERRQ(ierr); 284752f5567SLois Curfman McInnes /* assume if pivots exist then use LU; else Cholesky */ 285da3a660dSBarry Smith if (mat->pivots) { 28648d91487SBarry Smith LAgetrs_( "T", &mat->m, &one, mat->v, &mat->m, mat->pivots,y, &mat->m, &info ); 2873a40ed3dSBarry Smith } else { 28848d91487SBarry Smith LApotrs_( "L", &mat->m, &one, mat->v, &mat->m,y, &mat->m, &info ); 289da3a660dSBarry Smith } 290a8c6a408SBarry Smith if (info) SETERRQ(PETSC_ERR_LIB,0,"Bad solve"); 29190f02eecSBarry Smith if (tmp) { 29290f02eecSBarry Smith ierr = VecAXPY(&sone,tmp,yy);CHKERRQ(ierr); 29390f02eecSBarry Smith ierr = VecDestroy(tmp);CHKERRQ(ierr); 2943a40ed3dSBarry Smith } else { 29590f02eecSBarry Smith ierr = VecAXPY(&sone,zz,yy);CHKERRQ(ierr); 29690f02eecSBarry Smith } 2977a97a34bSBarry Smith ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 2987a97a34bSBarry Smith ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 29955659b69SBarry Smith PLogFlops(mat->n*mat->n - mat->n); 3003a40ed3dSBarry Smith PetscFunctionReturn(0); 301da3a660dSBarry Smith } 302289bc588SBarry Smith /* ------------------------------------------------------------------*/ 3035615d1e5SSatish Balay #undef __FUNC__ 3045615d1e5SSatish Balay #define __FUNC__ "MatRelax_SeqDense" 3058f6be9afSLois Curfman McInnes int MatRelax_SeqDense(Mat A,Vec bb,double omega,MatSORType flag, 30620e1a0d4SLois Curfman McInnes double shift,int its,Vec xx) 307289bc588SBarry Smith { 308c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 309289bc588SBarry Smith Scalar *x, *b, *v = mat->v, zero = 0.0, xt; 310bc1b551cSSatish Balay int ierr, m = mat->m, i; 311aa482453SBarry Smith #if !defined(PETSC_USE_COMPLEX) 312bc1b551cSSatish Balay int o = 1; 313bc1b551cSSatish Balay #endif 314289bc588SBarry Smith 3153a40ed3dSBarry Smith PetscFunctionBegin; 316289bc588SBarry Smith if (flag & SOR_ZERO_INITIAL_GUESS) { 3173a40ed3dSBarry Smith /* this is a hack fix, should have another version without the second BLdot */ 318bbb6d6a8SBarry Smith ierr = VecSet(&zero,xx);CHKERRQ(ierr); 319289bc588SBarry Smith } 3207a97a34bSBarry Smith ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 3217a97a34bSBarry Smith ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 322289bc588SBarry Smith while (its--) { 323289bc588SBarry Smith if (flag & SOR_FORWARD_SWEEP){ 324289bc588SBarry Smith for ( i=0; i<m; i++ ) { 325aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX) 326f1747703SBarry Smith /* cannot use BLAS dot for complex because compiler/linker is 327f1747703SBarry Smith not happy about returning a double complex */ 328f1747703SBarry Smith int _i; 329f1747703SBarry Smith Scalar sum = b[i]; 330f1747703SBarry Smith for ( _i=0; _i<m; _i++ ) { 3313f6de6efSSatish Balay sum -= PetscConj(v[i+_i*m])*x[_i]; 332f1747703SBarry Smith } 333f1747703SBarry Smith xt = sum; 334f1747703SBarry Smith #else 335289bc588SBarry Smith xt = b[i]-BLdot_(&m,v+i,&m,x,&o); 336f1747703SBarry Smith #endif 33755a1b374SBarry Smith x[i] = (1. - omega)*x[i] + omega*(xt+v[i + i*m]*x[i])/(v[i + i*m]+shift); 338289bc588SBarry Smith } 339289bc588SBarry Smith } 340289bc588SBarry Smith if (flag & SOR_BACKWARD_SWEEP) { 341289bc588SBarry Smith for ( i=m-1; i>=0; i-- ) { 342aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX) 343f1747703SBarry Smith /* cannot use BLAS dot for complex because compiler/linker is 344f1747703SBarry Smith not happy about returning a double complex */ 345f1747703SBarry Smith int _i; 346f1747703SBarry Smith Scalar sum = b[i]; 347f1747703SBarry Smith for ( _i=0; _i<m; _i++ ) { 3483f6de6efSSatish Balay sum -= PetscConj(v[i+_i*m])*x[_i]; 349f1747703SBarry Smith } 350f1747703SBarry Smith xt = sum; 351f1747703SBarry Smith #else 352289bc588SBarry Smith xt = b[i]-BLdot_(&m,v+i,&m,x,&o); 353f1747703SBarry Smith #endif 35455a1b374SBarry Smith x[i] = (1. - omega)*x[i] + omega*(xt+v[i + i*m]*x[i])/(v[i + i*m]+shift); 355289bc588SBarry Smith } 356289bc588SBarry Smith } 357289bc588SBarry Smith } 358600d6d0bSBarry Smith ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr); 3597a97a34bSBarry Smith ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 3603a40ed3dSBarry Smith PetscFunctionReturn(0); 361289bc588SBarry Smith } 362289bc588SBarry Smith 363289bc588SBarry Smith /* -----------------------------------------------------------------*/ 3645615d1e5SSatish Balay #undef __FUNC__ 3655615d1e5SSatish Balay #define __FUNC__ "MatMultTrans_SeqDense" 36644cd7ae7SLois Curfman McInnes int MatMultTrans_SeqDense(Mat A,Vec xx,Vec yy) 367289bc588SBarry Smith { 368c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 369289bc588SBarry Smith Scalar *v = mat->v, *x, *y; 3707a97a34bSBarry Smith int ierr,_One=1;Scalar _DOne=1.0, _DZero=0.0; 3713a40ed3dSBarry Smith 3723a40ed3dSBarry Smith PetscFunctionBegin; 3737a97a34bSBarry Smith if (!mat->m || !mat->n) PetscFunctionReturn(0); 3747a97a34bSBarry Smith ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 3757a97a34bSBarry Smith ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 37648d91487SBarry Smith LAgemv_("T",&(mat->m),&(mat->n),&_DOne,v,&(mat->m),x,&_One,&_DZero,y,&_One); 3777a97a34bSBarry Smith ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 3787a97a34bSBarry Smith ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 37955659b69SBarry Smith PLogFlops(2*mat->m*mat->n - mat->n); 3803a40ed3dSBarry Smith PetscFunctionReturn(0); 381289bc588SBarry Smith } 3826ee01492SSatish Balay 3835615d1e5SSatish Balay #undef __FUNC__ 3845615d1e5SSatish Balay #define __FUNC__ "MatMult_SeqDense" 38544cd7ae7SLois Curfman McInnes int MatMult_SeqDense(Mat A,Vec xx,Vec yy) 386289bc588SBarry Smith { 387c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 388289bc588SBarry Smith Scalar *v = mat->v, *x, *y; 3897a97a34bSBarry Smith int ierr,_One=1;Scalar _DOne=1.0, _DZero=0.0; 3903a40ed3dSBarry Smith 3913a40ed3dSBarry Smith PetscFunctionBegin; 3927a97a34bSBarry Smith if (!mat->m || !mat->n) PetscFunctionReturn(0); 3937a97a34bSBarry Smith ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 3947a97a34bSBarry Smith ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 39548d91487SBarry Smith LAgemv_( "N", &(mat->m), &(mat->n), &_DOne, v, &(mat->m),x,&_One,&_DZero,y,&_One); 3967a97a34bSBarry Smith ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 3977a97a34bSBarry Smith ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 39855659b69SBarry Smith PLogFlops(2*mat->m*mat->n - mat->m); 3993a40ed3dSBarry Smith PetscFunctionReturn(0); 400289bc588SBarry Smith } 4016ee01492SSatish Balay 4025615d1e5SSatish Balay #undef __FUNC__ 4035615d1e5SSatish Balay #define __FUNC__ "MatMultAdd_SeqDense" 40444cd7ae7SLois Curfman McInnes int MatMultAdd_SeqDense(Mat A,Vec xx,Vec zz,Vec yy) 405289bc588SBarry Smith { 406c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 407600d6d0bSBarry Smith Scalar *v = mat->v, *x, *y; 4087a97a34bSBarry Smith int ierr,_One=1; Scalar _DOne=1.0; 4093a40ed3dSBarry Smith 4103a40ed3dSBarry Smith PetscFunctionBegin; 4117a97a34bSBarry Smith if (!mat->m || !mat->n) PetscFunctionReturn(0); 412600d6d0bSBarry Smith if (zz != yy) {ierr = VecCopy(zz,yy);CHKERRQ(ierr);} 4137a97a34bSBarry Smith ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 4147a97a34bSBarry Smith ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 41548d91487SBarry Smith LAgemv_( "N", &(mat->m), &(mat->n),&_DOne,v,&(mat->m),x,&_One,&_DOne,y,&_One); 4167a97a34bSBarry Smith ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 4177a97a34bSBarry Smith ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 41855659b69SBarry Smith PLogFlops(2*mat->m*mat->n); 4193a40ed3dSBarry Smith PetscFunctionReturn(0); 420289bc588SBarry Smith } 4216ee01492SSatish Balay 4225615d1e5SSatish Balay #undef __FUNC__ 4235615d1e5SSatish Balay #define __FUNC__ "MatMultTransAdd_SeqDense" 42444cd7ae7SLois Curfman McInnes int MatMultTransAdd_SeqDense(Mat A,Vec xx,Vec zz,Vec yy) 425289bc588SBarry Smith { 426c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 427600d6d0bSBarry Smith Scalar *v = mat->v, *x, *y; 4287a97a34bSBarry Smith int ierr,_One=1; 4296abc6512SBarry Smith Scalar _DOne=1.0; 4303a40ed3dSBarry Smith 4313a40ed3dSBarry Smith PetscFunctionBegin; 4327a97a34bSBarry Smith if (!mat->m || !mat->n) PetscFunctionReturn(0); 433600d6d0bSBarry Smith if (zz != yy) {ierr = VecCopy(zz,yy);CHKERRQ(ierr);} 4347a97a34bSBarry Smith ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 4357a97a34bSBarry Smith ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 43648d91487SBarry Smith LAgemv_( "T", &(mat->m), &(mat->n), &_DOne, v, &(mat->m),x,&_One,&_DOne,y,&_One); 4377a97a34bSBarry Smith ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr); 4387a97a34bSBarry Smith ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr); 43955659b69SBarry Smith PLogFlops(2*mat->m*mat->n); 4403a40ed3dSBarry Smith PetscFunctionReturn(0); 441289bc588SBarry Smith } 442289bc588SBarry Smith 443289bc588SBarry Smith /* -----------------------------------------------------------------*/ 4445615d1e5SSatish Balay #undef __FUNC__ 4455615d1e5SSatish Balay #define __FUNC__ "MatGetRow_SeqDense" 4468f6be9afSLois Curfman McInnes int MatGetRow_SeqDense(Mat A,int row,int *ncols,int **cols,Scalar **vals) 447289bc588SBarry Smith { 448c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 449289bc588SBarry Smith Scalar *v; 450289bc588SBarry Smith int i; 45167e560aaSBarry Smith 4523a40ed3dSBarry Smith PetscFunctionBegin; 453289bc588SBarry Smith *ncols = mat->n; 454289bc588SBarry Smith if (cols) { 45545d48df9SBarry Smith *cols = (int *) PetscMalloc((mat->n+1)*sizeof(int));CHKPTRQ(*cols); 45673c3ce57SBarry Smith for ( i=0; i<mat->n; i++ ) (*cols)[i] = i; 457289bc588SBarry Smith } 458289bc588SBarry Smith if (vals) { 45945d48df9SBarry Smith *vals = (Scalar *) PetscMalloc((mat->n+1)*sizeof(Scalar));CHKPTRQ(*vals); 460289bc588SBarry Smith v = mat->v + row; 46173c3ce57SBarry Smith for ( i=0; i<mat->n; i++ ) {(*vals)[i] = *v; v += mat->m;} 462289bc588SBarry Smith } 4633a40ed3dSBarry Smith PetscFunctionReturn(0); 464289bc588SBarry Smith } 4656ee01492SSatish Balay 4665615d1e5SSatish Balay #undef __FUNC__ 4675615d1e5SSatish Balay #define __FUNC__ "MatRestoreRow_SeqDense" 4688f6be9afSLois Curfman McInnes int MatRestoreRow_SeqDense(Mat A,int row,int *ncols,int **cols,Scalar **vals) 469289bc588SBarry Smith { 470606d414cSSatish Balay int ierr; 471606d414cSSatish Balay PetscFunctionBegin; 472606d414cSSatish Balay if (cols) {ierr = PetscFree(*cols);CHKERRQ(ierr);} 473606d414cSSatish Balay if (vals) {ierr = PetscFree(*vals);CHKERRQ(ierr); } 4743a40ed3dSBarry Smith PetscFunctionReturn(0); 475289bc588SBarry Smith } 476289bc588SBarry Smith /* ----------------------------------------------------------------*/ 4775615d1e5SSatish Balay #undef __FUNC__ 4785615d1e5SSatish Balay #define __FUNC__ "MatSetValues_SeqDense" 4798f6be9afSLois Curfman McInnes int MatSetValues_SeqDense(Mat A,int m,int *indexm,int n, 480e8d4e0b9SBarry Smith int *indexn,Scalar *v,InsertMode addv) 481289bc588SBarry Smith { 482c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 483289bc588SBarry Smith int i,j; 484d6dfbf8fSBarry Smith 4853a40ed3dSBarry Smith PetscFunctionBegin; 486289bc588SBarry Smith if (!mat->roworiented) { 487dbb450caSBarry Smith if (addv == INSERT_VALUES) { 488289bc588SBarry Smith for ( j=0; j<n; j++ ) { 4895ef9f2a5SBarry Smith if (indexn[j] < 0) {v += m; continue;} 490aa482453SBarry Smith #if defined(PETSC_USE_BOPT_g) 491a8c6a408SBarry Smith if (indexn[j] >= A->n) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Column too large"); 49258804f6dSBarry Smith #endif 493289bc588SBarry Smith for ( i=0; i<m; i++ ) { 4945ef9f2a5SBarry Smith if (indexm[i] < 0) {v++; continue;} 495aa482453SBarry Smith #if defined(PETSC_USE_BOPT_g) 496a8c6a408SBarry Smith if (indexm[i] >= A->m) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Row too large"); 49758804f6dSBarry Smith #endif 498289bc588SBarry Smith mat->v[indexn[j]*mat->m + indexm[i]] = *v++; 499289bc588SBarry Smith } 500289bc588SBarry Smith } 5013a40ed3dSBarry Smith } else { 502289bc588SBarry Smith for ( j=0; j<n; j++ ) { 5035ef9f2a5SBarry Smith if (indexn[j] < 0) {v += m; continue;} 504aa482453SBarry Smith #if defined(PETSC_USE_BOPT_g) 505a8c6a408SBarry Smith if (indexn[j] >= A->n) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Column too large"); 50658804f6dSBarry Smith #endif 507289bc588SBarry Smith for ( i=0; i<m; i++ ) { 5085ef9f2a5SBarry Smith if (indexm[i] < 0) {v++; continue;} 509aa482453SBarry Smith #if defined(PETSC_USE_BOPT_g) 510a8c6a408SBarry Smith if (indexm[i] >= A->m) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Row too large"); 51158804f6dSBarry Smith #endif 512289bc588SBarry Smith mat->v[indexn[j]*mat->m + indexm[i]] += *v++; 513289bc588SBarry Smith } 514289bc588SBarry Smith } 515289bc588SBarry Smith } 5163a40ed3dSBarry Smith } else { 517dbb450caSBarry Smith if (addv == INSERT_VALUES) { 518e8d4e0b9SBarry Smith for ( i=0; i<m; i++ ) { 5195ef9f2a5SBarry Smith if (indexm[i] < 0) { v += n; continue;} 520aa482453SBarry Smith #if defined(PETSC_USE_BOPT_g) 521a8c6a408SBarry Smith if (indexm[i] >= A->m) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Row too large"); 52258804f6dSBarry Smith #endif 523e8d4e0b9SBarry Smith for ( j=0; j<n; j++ ) { 5245ef9f2a5SBarry Smith if (indexn[j] < 0) { v++; continue;} 525aa482453SBarry Smith #if defined(PETSC_USE_BOPT_g) 526a8c6a408SBarry Smith if (indexn[j] >= A->n) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Column too large"); 52758804f6dSBarry Smith #endif 528e8d4e0b9SBarry Smith mat->v[indexn[j]*mat->m + indexm[i]] = *v++; 529e8d4e0b9SBarry Smith } 530e8d4e0b9SBarry Smith } 5313a40ed3dSBarry Smith } else { 532289bc588SBarry Smith for ( i=0; i<m; i++ ) { 5335ef9f2a5SBarry Smith if (indexm[i] < 0) { v += n; continue;} 534aa482453SBarry Smith #if defined(PETSC_USE_BOPT_g) 535a8c6a408SBarry Smith if (indexm[i] >= A->m) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Row too large"); 53658804f6dSBarry Smith #endif 537289bc588SBarry Smith for ( j=0; j<n; j++ ) { 5385ef9f2a5SBarry Smith if (indexn[j] < 0) { v++; continue;} 539aa482453SBarry Smith #if defined(PETSC_USE_BOPT_g) 540a8c6a408SBarry Smith if (indexn[j] >= A->n) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Column too large"); 54158804f6dSBarry Smith #endif 542289bc588SBarry Smith mat->v[indexn[j]*mat->m + indexm[i]] += *v++; 543289bc588SBarry Smith } 544289bc588SBarry Smith } 545289bc588SBarry Smith } 546e8d4e0b9SBarry Smith } 5473a40ed3dSBarry Smith PetscFunctionReturn(0); 548289bc588SBarry Smith } 549e8d4e0b9SBarry Smith 5505615d1e5SSatish Balay #undef __FUNC__ 5515615d1e5SSatish Balay #define __FUNC__ "MatGetValues_SeqDense" 5528f6be9afSLois Curfman McInnes int MatGetValues_SeqDense(Mat A,int m,int *indexm,int n,int *indexn,Scalar *v) 553ae80bb75SLois Curfman McInnes { 554ae80bb75SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 555ae80bb75SLois Curfman McInnes int i, j; 556ae80bb75SLois Curfman McInnes Scalar *vpt = v; 557ae80bb75SLois Curfman McInnes 5583a40ed3dSBarry Smith PetscFunctionBegin; 559ae80bb75SLois Curfman McInnes /* row-oriented output */ 560ae80bb75SLois Curfman McInnes for ( i=0; i<m; i++ ) { 561ae80bb75SLois Curfman McInnes for ( j=0; j<n; j++ ) { 562ae80bb75SLois Curfman McInnes *vpt++ = mat->v[indexn[j]*mat->m + indexm[i]]; 563ae80bb75SLois Curfman McInnes } 564ae80bb75SLois Curfman McInnes } 5653a40ed3dSBarry Smith PetscFunctionReturn(0); 566ae80bb75SLois Curfman McInnes } 567ae80bb75SLois Curfman McInnes 568289bc588SBarry Smith /* -----------------------------------------------------------------*/ 569289bc588SBarry Smith 57077c4ece6SBarry Smith #include "sys.h" 57188e32edaSLois Curfman McInnes 5725615d1e5SSatish Balay #undef __FUNC__ 5735615d1e5SSatish Balay #define __FUNC__ "MatLoad_SeqDense" 57419bcc07fSBarry Smith int MatLoad_SeqDense(Viewer viewer,MatType type,Mat *A) 57588e32edaSLois Curfman McInnes { 57688e32edaSLois Curfman McInnes Mat_SeqDense *a; 57788e32edaSLois Curfman McInnes Mat B; 57888e32edaSLois Curfman McInnes int *scols, i, j, nz, ierr, fd, header[4], size; 57988e32edaSLois Curfman McInnes int *rowlengths = 0, M, N, *cols; 5804a41a4d3SSatish Balay Scalar *vals, *svals, *v,*w; 58119bcc07fSBarry Smith MPI_Comm comm = ((PetscObject)viewer)->comm; 58288e32edaSLois Curfman McInnes 5833a40ed3dSBarry Smith PetscFunctionBegin; 584d132466eSBarry Smith ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 585a8c6a408SBarry Smith if (size > 1) SETERRQ(PETSC_ERR_ARG_WRONG,0,"view must have one processor"); 58690ace30eSBarry Smith ierr = ViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr); 5870752156aSBarry Smith ierr = PetscBinaryRead(fd,header,4,PETSC_INT);CHKERRQ(ierr); 588a8c6a408SBarry Smith if (header[0] != MAT_COOKIE) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,0,"Not matrix object"); 58988e32edaSLois Curfman McInnes M = header[1]; N = header[2]; nz = header[3]; 59088e32edaSLois Curfman McInnes 59190ace30eSBarry Smith if (nz == MATRIX_BINARY_FORMAT_DENSE) { /* matrix in file is dense */ 59290ace30eSBarry Smith ierr = MatCreateSeqDense(comm,M,N,PETSC_NULL,A);CHKERRQ(ierr); 59390ace30eSBarry Smith B = *A; 59490ace30eSBarry Smith a = (Mat_SeqDense *) B->data; 5954a41a4d3SSatish Balay v = a->v; 5964a41a4d3SSatish Balay /* Allocate some temp space to read in the values and then flip them 5974a41a4d3SSatish Balay from row major to column major */ 5984a41a4d3SSatish Balay w = (Scalar *) PetscMalloc((M*N+1)*sizeof(Scalar));CHKPTRQ(w); 59990ace30eSBarry Smith /* read in nonzero values */ 6004a41a4d3SSatish Balay ierr = PetscBinaryRead(fd,w,M*N,PETSC_SCALAR);CHKERRQ(ierr); 6014a41a4d3SSatish Balay /* now flip the values and store them in the matrix*/ 6024a41a4d3SSatish Balay for ( j=0; j<N; j++ ) { 6034a41a4d3SSatish Balay for ( i=0; i<M; i++ ) { 6044a41a4d3SSatish Balay *v++ =w[i*N+j]; 6054a41a4d3SSatish Balay } 6064a41a4d3SSatish Balay } 607606d414cSSatish Balay ierr = PetscFree(w);CHKERRQ(ierr); 6086d4a8577SBarry Smith ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 6096d4a8577SBarry Smith ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 61090ace30eSBarry Smith } else { 61188e32edaSLois Curfman McInnes /* read row lengths */ 61245d48df9SBarry Smith rowlengths = (int*) PetscMalloc( (M+1)*sizeof(int) );CHKPTRQ(rowlengths); 6130752156aSBarry Smith ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr); 61488e32edaSLois Curfman McInnes 61588e32edaSLois Curfman McInnes /* create our matrix */ 616b4fd4287SBarry Smith ierr = MatCreateSeqDense(comm,M,N,PETSC_NULL,A);CHKERRQ(ierr); 61788e32edaSLois Curfman McInnes B = *A; 61888e32edaSLois Curfman McInnes a = (Mat_SeqDense *) B->data; 61988e32edaSLois Curfman McInnes v = a->v; 62088e32edaSLois Curfman McInnes 62188e32edaSLois Curfman McInnes /* read column indices and nonzeros */ 62245d48df9SBarry Smith cols = scols = (int *) PetscMalloc( (nz+1)*sizeof(int) );CHKPTRQ(cols); 6230752156aSBarry Smith ierr = PetscBinaryRead(fd,cols,nz,PETSC_INT);CHKERRQ(ierr); 62445d48df9SBarry Smith vals = svals = (Scalar *) PetscMalloc( (nz+1)*sizeof(Scalar) );CHKPTRQ(vals); 6250752156aSBarry Smith ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR);CHKERRQ(ierr); 62688e32edaSLois Curfman McInnes 62788e32edaSLois Curfman McInnes /* insert into matrix */ 62888e32edaSLois Curfman McInnes for ( i=0; i<M; i++ ) { 62988e32edaSLois Curfman McInnes for ( j=0; j<rowlengths[i]; j++ ) v[i+M*scols[j]] = svals[j]; 63088e32edaSLois Curfman McInnes svals += rowlengths[i]; scols += rowlengths[i]; 63188e32edaSLois Curfman McInnes } 632606d414cSSatish Balay ierr = PetscFree(vals);CHKERRQ(ierr); 633606d414cSSatish Balay ierr = PetscFree(cols);CHKERRQ(ierr); 634606d414cSSatish Balay ierr = PetscFree(rowlengths);CHKERRQ(ierr); 63588e32edaSLois Curfman McInnes 6366d4a8577SBarry Smith ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 6376d4a8577SBarry Smith ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 63890ace30eSBarry Smith } 6393a40ed3dSBarry Smith PetscFunctionReturn(0); 64088e32edaSLois Curfman McInnes } 64188e32edaSLois Curfman McInnes 64277c4ece6SBarry Smith #include "sys.h" 643932b0c3eSLois Curfman McInnes 6445615d1e5SSatish Balay #undef __FUNC__ 6455615d1e5SSatish Balay #define __FUNC__ "MatView_SeqDense_ASCII" 646932b0c3eSLois Curfman McInnes static int MatView_SeqDense_ASCII(Mat A,Viewer viewer) 647289bc588SBarry Smith { 648932b0c3eSLois Curfman McInnes Mat_SeqDense *a = (Mat_SeqDense *) A->data; 649932b0c3eSLois Curfman McInnes int ierr, i, j, format; 650d636dbe3SBarry Smith FILE *fd; 651932b0c3eSLois Curfman McInnes char *outputname; 652932b0c3eSLois Curfman McInnes Scalar *v; 653932b0c3eSLois Curfman McInnes 6543a40ed3dSBarry Smith PetscFunctionBegin; 65590ace30eSBarry Smith ierr = ViewerASCIIGetPointer(viewer,&fd);CHKERRQ(ierr); 65677ed5343SBarry Smith ierr = ViewerGetOutputname(viewer,&outputname);CHKERRQ(ierr); 657888f2ed8SSatish Balay ierr = ViewerGetFormat(viewer,&format);CHKERRQ(ierr); 658639f9d9dSBarry Smith if (format == VIEWER_FORMAT_ASCII_INFO || format == VIEWER_FORMAT_ASCII_INFO_LONG) { 6593a40ed3dSBarry Smith PetscFunctionReturn(0); /* do nothing for now */ 660f72585f2SLois Curfman McInnes } 66102cad45dSBarry Smith else if (format == VIEWER_FORMAT_ASCII_COMMON) { 66280cd9d93SLois Curfman McInnes for ( i=0; i<a->m; i++ ) { 66344cd7ae7SLois Curfman McInnes v = a->v + i; 66480cd9d93SLois Curfman McInnes fprintf(fd,"row %d:",i); 66580cd9d93SLois Curfman McInnes for ( j=0; j<a->n; j++ ) { 666aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX) 667e20fef11SSatish Balay if (PetscReal(*v) != 0.0 && PetscImaginary(*v) != 0.0) fprintf(fd," %d %g + %g i",j,PetscReal(*v),PetscImaginary(*v)); 6683f6de6efSSatish Balay else if (PetscReal(*v)) fprintf(fd," %d %g ",j,PetscReal(*v)); 66980cd9d93SLois Curfman McInnes #else 67080cd9d93SLois Curfman McInnes if (*v) fprintf(fd," %d %g ",j,*v); 67180cd9d93SLois Curfman McInnes #endif 672d64ed03dSBarry Smith v += a->m; 67380cd9d93SLois Curfman McInnes } 67480cd9d93SLois Curfman McInnes fprintf(fd,"\n"); 67580cd9d93SLois Curfman McInnes } 6763a40ed3dSBarry Smith } else { 677aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX) 67847989497SBarry Smith int allreal = 1; 67947989497SBarry Smith /* determine if matrix has all real values */ 68047989497SBarry Smith v = a->v; 68147989497SBarry Smith for ( i=0; i<a->m*a->n; i++ ) { 682e20fef11SSatish Balay if (PetscImaginary(v[i])) { allreal = 0; break ;} 68347989497SBarry Smith } 68447989497SBarry Smith #endif 685932b0c3eSLois Curfman McInnes for ( i=0; i<a->m; i++ ) { 686932b0c3eSLois Curfman McInnes v = a->v + i; 687932b0c3eSLois Curfman McInnes for ( j=0; j<a->n; j++ ) { 688aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX) 68947989497SBarry Smith if (allreal) { 6903f6de6efSSatish Balay fprintf(fd,"%6.4e ",PetscReal(*v)); v += a->m; 69147989497SBarry Smith } else { 692e20fef11SSatish Balay fprintf(fd,"%6.4e + %6.4e i ",PetscReal(*v),PetscImaginary(*v)); v += a->m; 69347989497SBarry Smith } 694289bc588SBarry Smith #else 695932b0c3eSLois Curfman McInnes fprintf(fd,"%6.4e ",*v); v += a->m; 696289bc588SBarry Smith #endif 697289bc588SBarry Smith } 6988e37dc09SBarry Smith fprintf(fd,"\n"); 699289bc588SBarry Smith } 700da3a660dSBarry Smith } 701932b0c3eSLois Curfman McInnes fflush(fd); 7023a40ed3dSBarry Smith PetscFunctionReturn(0); 703289bc588SBarry Smith } 704289bc588SBarry Smith 7055615d1e5SSatish Balay #undef __FUNC__ 7065615d1e5SSatish Balay #define __FUNC__ "MatView_SeqDense_Binary" 707932b0c3eSLois Curfman McInnes static int MatView_SeqDense_Binary(Mat A,Viewer viewer) 708932b0c3eSLois Curfman McInnes { 709932b0c3eSLois Curfman McInnes Mat_SeqDense *a = (Mat_SeqDense *) A->data; 710932b0c3eSLois Curfman McInnes int ict, j, n = a->n, m = a->m, i, fd, *col_lens, ierr, nz = m*n; 71190ace30eSBarry Smith int format; 71290ace30eSBarry Smith Scalar *v, *anonz,*vals; 713932b0c3eSLois Curfman McInnes 7143a40ed3dSBarry Smith PetscFunctionBegin; 71590ace30eSBarry Smith ierr = ViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr); 71690ace30eSBarry Smith 71790ace30eSBarry Smith ierr = ViewerGetFormat(viewer,&format);CHKERRQ(ierr); 71802cad45dSBarry Smith if (format == VIEWER_FORMAT_BINARY_NATIVE) { 71990ace30eSBarry Smith /* store the matrix as a dense matrix */ 72090ace30eSBarry Smith col_lens = (int *) PetscMalloc( 4*sizeof(int) );CHKPTRQ(col_lens); 72190ace30eSBarry Smith col_lens[0] = MAT_COOKIE; 72290ace30eSBarry Smith col_lens[1] = m; 72390ace30eSBarry Smith col_lens[2] = n; 72490ace30eSBarry Smith col_lens[3] = MATRIX_BINARY_FORMAT_DENSE; 7250752156aSBarry Smith ierr = PetscBinaryWrite(fd,col_lens,4,PETSC_INT,1);CHKERRQ(ierr); 726606d414cSSatish Balay ierr = PetscFree(col_lens);CHKERRQ(ierr); 72790ace30eSBarry Smith 72890ace30eSBarry Smith /* write out matrix, by rows */ 72945d48df9SBarry Smith vals = (Scalar *) PetscMalloc((m*n+1)*sizeof(Scalar));CHKPTRQ(vals); 73090ace30eSBarry Smith v = a->v; 73190ace30eSBarry Smith for ( i=0; i<m; i++ ) { 73290ace30eSBarry Smith for ( j=0; j<n; j++ ) { 73390ace30eSBarry Smith vals[i + j*m] = *v++; 73490ace30eSBarry Smith } 73590ace30eSBarry Smith } 7360752156aSBarry Smith ierr = PetscBinaryWrite(fd,vals,n*m,PETSC_SCALAR,0);CHKERRQ(ierr); 737606d414cSSatish Balay ierr = PetscFree(vals);CHKERRQ(ierr); 73890ace30eSBarry Smith } else { 7390452661fSBarry Smith col_lens = (int *) PetscMalloc( (4+nz)*sizeof(int) );CHKPTRQ(col_lens); 740932b0c3eSLois Curfman McInnes col_lens[0] = MAT_COOKIE; 741932b0c3eSLois Curfman McInnes col_lens[1] = m; 742932b0c3eSLois Curfman McInnes col_lens[2] = n; 743932b0c3eSLois Curfman McInnes col_lens[3] = nz; 744932b0c3eSLois Curfman McInnes 745932b0c3eSLois Curfman McInnes /* store lengths of each row and write (including header) to file */ 746932b0c3eSLois Curfman McInnes for ( i=0; i<m; i++ ) col_lens[4+i] = n; 7470752156aSBarry Smith ierr = PetscBinaryWrite(fd,col_lens,4+m,PETSC_INT,1);CHKERRQ(ierr); 748932b0c3eSLois Curfman McInnes 749932b0c3eSLois Curfman McInnes /* Possibly should write in smaller increments, not whole matrix at once? */ 750932b0c3eSLois Curfman McInnes /* store column indices (zero start index) */ 751932b0c3eSLois Curfman McInnes ict = 0; 752932b0c3eSLois Curfman McInnes for ( i=0; i<m; i++ ) { 753932b0c3eSLois Curfman McInnes for ( j=0; j<n; j++ ) col_lens[ict++] = j; 754932b0c3eSLois Curfman McInnes } 7550752156aSBarry Smith ierr = PetscBinaryWrite(fd,col_lens,nz,PETSC_INT,0);CHKERRQ(ierr); 756606d414cSSatish Balay ierr = PetscFree(col_lens);CHKERRQ(ierr); 757932b0c3eSLois Curfman McInnes 758932b0c3eSLois Curfman McInnes /* store nonzero values */ 75945d48df9SBarry Smith anonz = (Scalar *) PetscMalloc((nz+1)*sizeof(Scalar));CHKPTRQ(anonz); 760932b0c3eSLois Curfman McInnes ict = 0; 761932b0c3eSLois Curfman McInnes for ( i=0; i<m; i++ ) { 762932b0c3eSLois Curfman McInnes v = a->v + i; 763932b0c3eSLois Curfman McInnes for ( j=0; j<n; j++ ) { 764932b0c3eSLois Curfman McInnes anonz[ict++] = *v; v += a->m; 765932b0c3eSLois Curfman McInnes } 766932b0c3eSLois Curfman McInnes } 7670752156aSBarry Smith ierr = PetscBinaryWrite(fd,anonz,nz,PETSC_SCALAR,0);CHKERRQ(ierr); 768606d414cSSatish Balay ierr = PetscFree(anonz);CHKERRQ(ierr); 76990ace30eSBarry Smith } 7703a40ed3dSBarry Smith PetscFunctionReturn(0); 771932b0c3eSLois Curfman McInnes } 772932b0c3eSLois Curfman McInnes 7735615d1e5SSatish Balay #undef __FUNC__ 7745615d1e5SSatish Balay #define __FUNC__ "MatView_SeqDense" 775c6e7dd08SBarry Smith int MatView_SeqDense(Mat A,Viewer viewer) 776932b0c3eSLois Curfman McInnes { 777932b0c3eSLois Curfman McInnes Mat_SeqDense *a = (Mat_SeqDense*) A->data; 778bcd2baecSBarry Smith int ierr; 779*0f5bd95cSBarry Smith int issocket,isascii,isbinary; 780932b0c3eSLois Curfman McInnes 7813a40ed3dSBarry Smith PetscFunctionBegin; 782*0f5bd95cSBarry Smith issocket = PetscTypeCompare(viewer,SOCKET_VIEWER); 783*0f5bd95cSBarry Smith isascii = PetscTypeCompare(viewer,ASCII_VIEWER); 784*0f5bd95cSBarry Smith isbinary = PetscTypeCompare(viewer,BINARY_VIEWER); 785*0f5bd95cSBarry Smith 786*0f5bd95cSBarry Smith if (issocket) { 7877b2a1423SBarry Smith ierr = ViewerSocketPutScalar_Private(viewer,a->m,a->n,a->v);CHKERRQ(ierr); 788*0f5bd95cSBarry Smith } else if (isascii) { 7893a40ed3dSBarry Smith ierr = MatView_SeqDense_ASCII(A,viewer);CHKERRQ(ierr); 790*0f5bd95cSBarry Smith } else if (isbinary) { 7913a40ed3dSBarry Smith ierr = MatView_SeqDense_Binary(A,viewer);CHKERRQ(ierr); 7925cd90555SBarry Smith } else { 793*0f5bd95cSBarry Smith SETERRQ1(1,1,"Viewer type %s not supported by dense matrix",((PetscObject)viewer)->type_name); 794932b0c3eSLois Curfman McInnes } 7953a40ed3dSBarry Smith PetscFunctionReturn(0); 796932b0c3eSLois Curfman McInnes } 797289bc588SBarry Smith 7985615d1e5SSatish Balay #undef __FUNC__ 7995615d1e5SSatish Balay #define __FUNC__ "MatDestroy_SeqDense" 800c6e7dd08SBarry Smith int MatDestroy_SeqDense(Mat mat) 801289bc588SBarry Smith { 802ec8511deSBarry Smith Mat_SeqDense *l = (Mat_SeqDense *) mat->data; 80390f02eecSBarry Smith int ierr; 80490f02eecSBarry Smith 8053a40ed3dSBarry Smith PetscFunctionBegin; 80694d884c6SBarry Smith 80794d884c6SBarry Smith if (mat->mapping) { 80894d884c6SBarry Smith ierr = ISLocalToGlobalMappingDestroy(mat->mapping);CHKERRQ(ierr); 80994d884c6SBarry Smith } 81094d884c6SBarry Smith if (mat->bmapping) { 81194d884c6SBarry Smith ierr = ISLocalToGlobalMappingDestroy(mat->bmapping);CHKERRQ(ierr); 81294d884c6SBarry Smith } 813aa482453SBarry Smith #if defined(PETSC_USE_LOG) 814c6e7dd08SBarry Smith PLogObjectState((PetscObject)mat,"Rows %d Cols %d",l->m,l->n); 815a5a9c739SBarry Smith #endif 816606d414cSSatish Balay if (l->pivots) {ierr = PetscFree(l->pivots);CHKERRQ(ierr);} 817606d414cSSatish Balay if (!l->user_alloc) {ierr = PetscFree(l->v);CHKERRQ(ierr);} 818606d414cSSatish Balay ierr = PetscFree(l);CHKERRQ(ierr); 819a5ae1ecdSBarry Smith if (mat->rmap) { 820a5ae1ecdSBarry Smith ierr = MapDestroy(mat->rmap);CHKERRQ(ierr); 821a5ae1ecdSBarry Smith } 822a5ae1ecdSBarry Smith if (mat->cmap) { 823a5ae1ecdSBarry Smith ierr = MapDestroy(mat->cmap);CHKERRQ(ierr); 82490f02eecSBarry Smith } 825a5a9c739SBarry Smith PLogObjectDestroy(mat); 8260452661fSBarry Smith PetscHeaderDestroy(mat); 8273a40ed3dSBarry Smith PetscFunctionReturn(0); 828289bc588SBarry Smith } 829289bc588SBarry Smith 8305615d1e5SSatish Balay #undef __FUNC__ 8315615d1e5SSatish Balay #define __FUNC__ "MatTranspose_SeqDense" 8328f6be9afSLois Curfman McInnes int MatTranspose_SeqDense(Mat A,Mat *matout) 833289bc588SBarry Smith { 834c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 835549d3d68SSatish Balay int k, j, m, n, ierr; 836d3e5ee88SLois Curfman McInnes Scalar *v, tmp; 83748b35521SBarry Smith 8383a40ed3dSBarry Smith PetscFunctionBegin; 839d3e5ee88SLois Curfman McInnes v = mat->v; m = mat->m; n = mat->n; 8403638b69dSLois Curfman McInnes if (matout == PETSC_NULL) { /* in place transpose */ 841d64ed03dSBarry Smith if (m != n) { /* malloc temp to hold transpose */ 842d64ed03dSBarry Smith Scalar *w = (Scalar *) PetscMalloc((m+1)*(n+1)*sizeof(Scalar));CHKPTRQ(w); 843d64ed03dSBarry Smith for ( j=0; j<m; j++ ) { 844d64ed03dSBarry Smith for ( k=0; k<n; k++ ) { 845d64ed03dSBarry Smith w[k + j*n] = v[j + k*m]; 846d64ed03dSBarry Smith } 847d64ed03dSBarry Smith } 848549d3d68SSatish Balay ierr = PetscMemcpy(v,w,m*n*sizeof(Scalar));CHKERRQ(ierr); 849606d414cSSatish Balay ierr = PetscFree(w);CHKERRQ(ierr); 850d64ed03dSBarry Smith } else { 851d3e5ee88SLois Curfman McInnes for ( j=0; j<m; j++ ) { 852289bc588SBarry Smith for ( k=0; k<j; k++ ) { 853d3e5ee88SLois Curfman McInnes tmp = v[j + k*n]; 854d3e5ee88SLois Curfman McInnes v[j + k*n] = v[k + j*n]; 855d3e5ee88SLois Curfman McInnes v[k + j*n] = tmp; 856289bc588SBarry Smith } 857289bc588SBarry Smith } 858d64ed03dSBarry Smith } 8593a40ed3dSBarry Smith } else { /* out-of-place transpose */ 860d3e5ee88SLois Curfman McInnes Mat tmat; 861ec8511deSBarry Smith Mat_SeqDense *tmatd; 862d3e5ee88SLois Curfman McInnes Scalar *v2; 863b4fd4287SBarry Smith ierr = MatCreateSeqDense(A->comm,mat->n,mat->m,PETSC_NULL,&tmat);CHKERRQ(ierr); 864ec8511deSBarry Smith tmatd = (Mat_SeqDense *) tmat->data; 8650de55854SLois Curfman McInnes v = mat->v; v2 = tmatd->v; 866d3e5ee88SLois Curfman McInnes for ( j=0; j<n; j++ ) { 8670de55854SLois Curfman McInnes for ( k=0; k<m; k++ ) v2[j + k*n] = v[k + j*m]; 868d3e5ee88SLois Curfman McInnes } 8696d4a8577SBarry Smith ierr = MatAssemblyBegin(tmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 8706d4a8577SBarry Smith ierr = MatAssemblyEnd(tmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 871d3e5ee88SLois Curfman McInnes *matout = tmat; 87248b35521SBarry Smith } 8733a40ed3dSBarry Smith PetscFunctionReturn(0); 874289bc588SBarry Smith } 875289bc588SBarry Smith 8765615d1e5SSatish Balay #undef __FUNC__ 8775615d1e5SSatish Balay #define __FUNC__ "MatEqual_SeqDense" 8788f6be9afSLois Curfman McInnes int MatEqual_SeqDense(Mat A1,Mat A2, PetscTruth *flg) 879289bc588SBarry Smith { 880c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat1 = (Mat_SeqDense *) A1->data; 881c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat2 = (Mat_SeqDense *) A2->data; 882289bc588SBarry Smith int i; 883289bc588SBarry Smith Scalar *v1 = mat1->v, *v2 = mat2->v; 8849ea5d5aeSSatish Balay 8853a40ed3dSBarry Smith PetscFunctionBegin; 8863a40ed3dSBarry Smith if (A2->type != MATSEQDENSE) SETERRQ(PETSC_ERR_SUP,0,"Matrices should be of type MATSEQDENSE"); 8873a40ed3dSBarry Smith if (mat1->m != mat2->m) {*flg = PETSC_FALSE; PetscFunctionReturn(0);} 8883a40ed3dSBarry Smith if (mat1->n != mat2->n) {*flg = PETSC_FALSE; PetscFunctionReturn(0);} 889289bc588SBarry Smith for ( i=0; i<mat1->m*mat1->n; i++ ) { 8903a40ed3dSBarry Smith if (*v1 != *v2) {*flg = PETSC_FALSE; PetscFunctionReturn(0);} 891289bc588SBarry Smith v1++; v2++; 892289bc588SBarry Smith } 89377c4ece6SBarry Smith *flg = PETSC_TRUE; 8943a40ed3dSBarry Smith PetscFunctionReturn(0); 895289bc588SBarry Smith } 896289bc588SBarry Smith 8975615d1e5SSatish Balay #undef __FUNC__ 8985615d1e5SSatish Balay #define __FUNC__ "MatGetDiagonal_SeqDense" 8998f6be9afSLois Curfman McInnes int MatGetDiagonal_SeqDense(Mat A,Vec v) 900289bc588SBarry Smith { 901c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 9027a97a34bSBarry Smith int ierr,i, n, len; 90344cd7ae7SLois Curfman McInnes Scalar *x, zero = 0.0; 90444cd7ae7SLois Curfman McInnes 9053a40ed3dSBarry Smith PetscFunctionBegin; 9067a97a34bSBarry Smith ierr = VecSet(&zero,v);CHKERRQ(ierr); 9077a97a34bSBarry Smith ierr = VecGetSize(v,&n);CHKERRQ(ierr); 908600d6d0bSBarry Smith ierr = VecGetArray(v,&x);CHKERRQ(ierr); 90944cd7ae7SLois Curfman McInnes len = PetscMin(mat->m,mat->n); 910e3372554SBarry Smith if (n != mat->m) SETERRQ(PETSC_ERR_ARG_SIZ,0,"Nonconforming mat and vec"); 91144cd7ae7SLois Curfman McInnes for ( i=0; i<len; i++ ) { 912289bc588SBarry Smith x[i] = mat->v[i*mat->m + i]; 913289bc588SBarry Smith } 9147a97a34bSBarry Smith ierr = VecRestoreArray(v,&x);CHKERRQ(ierr); 9153a40ed3dSBarry Smith PetscFunctionReturn(0); 916289bc588SBarry Smith } 917289bc588SBarry Smith 9185615d1e5SSatish Balay #undef __FUNC__ 9195615d1e5SSatish Balay #define __FUNC__ "MatDiagonalScale_SeqDense" 9208f6be9afSLois Curfman McInnes int MatDiagonalScale_SeqDense(Mat A,Vec ll,Vec rr) 921289bc588SBarry Smith { 922c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 923da3a660dSBarry Smith Scalar *l,*r,x,*v; 9247a97a34bSBarry Smith int ierr,i,j,m = mat->m, n = mat->n; 92555659b69SBarry Smith 9263a40ed3dSBarry Smith PetscFunctionBegin; 92728988994SBarry Smith if (ll) { 9287a97a34bSBarry Smith ierr = VecGetSize(ll,&m);CHKERRQ(ierr); 929600d6d0bSBarry Smith ierr = VecGetArray(ll,&l);CHKERRQ(ierr); 930e3372554SBarry Smith if (m != mat->m) SETERRQ(PETSC_ERR_ARG_SIZ,0,"Left scaling vec wrong size"); 931da3a660dSBarry Smith for ( i=0; i<m; i++ ) { 932da3a660dSBarry Smith x = l[i]; 933da3a660dSBarry Smith v = mat->v + i; 934da3a660dSBarry Smith for ( j=0; j<n; j++ ) { (*v) *= x; v+= m;} 935da3a660dSBarry Smith } 9367a97a34bSBarry Smith ierr = VecRestoreArray(ll,&l);CHKERRQ(ierr); 93744cd7ae7SLois Curfman McInnes PLogFlops(n*m); 938da3a660dSBarry Smith } 93928988994SBarry Smith if (rr) { 9407a97a34bSBarry Smith ierr = VecGetSize(rr,&n);CHKERRQ(ierr); 941600d6d0bSBarry Smith ierr = VecGetArray(rr,&r);CHKERRQ(ierr); 942e3372554SBarry Smith if (n != mat->n) SETERRQ(PETSC_ERR_ARG_SIZ,0,"Right scaling vec wrong size"); 943da3a660dSBarry Smith for ( i=0; i<n; i++ ) { 944da3a660dSBarry Smith x = r[i]; 945da3a660dSBarry Smith v = mat->v + i*m; 946da3a660dSBarry Smith for ( j=0; j<m; j++ ) { (*v++) *= x;} 947da3a660dSBarry Smith } 9487a97a34bSBarry Smith ierr = VecRestoreArray(rr,&r);CHKERRQ(ierr); 94944cd7ae7SLois Curfman McInnes PLogFlops(n*m); 950da3a660dSBarry Smith } 9513a40ed3dSBarry Smith PetscFunctionReturn(0); 952289bc588SBarry Smith } 953289bc588SBarry Smith 9545615d1e5SSatish Balay #undef __FUNC__ 9555615d1e5SSatish Balay #define __FUNC__ "MatNorm_SeqDense" 9568f6be9afSLois Curfman McInnes int MatNorm_SeqDense(Mat A,NormType type,double *norm) 957289bc588SBarry Smith { 958c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 959289bc588SBarry Smith Scalar *v = mat->v; 960289bc588SBarry Smith double sum = 0.0; 961289bc588SBarry Smith int i, j; 96255659b69SBarry Smith 9633a40ed3dSBarry Smith PetscFunctionBegin; 964289bc588SBarry Smith if (type == NORM_FROBENIUS) { 965289bc588SBarry Smith for (i=0; i<mat->n*mat->m; i++ ) { 966aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX) 9673f6de6efSSatish Balay sum += PetscReal(PetscConj(*v)*(*v)); v++; 968289bc588SBarry Smith #else 969289bc588SBarry Smith sum += (*v)*(*v); v++; 970289bc588SBarry Smith #endif 971289bc588SBarry Smith } 972289bc588SBarry Smith *norm = sqrt(sum); 97355659b69SBarry Smith PLogFlops(2*mat->n*mat->m); 9743a40ed3dSBarry Smith } else if (type == NORM_1) { 975289bc588SBarry Smith *norm = 0.0; 976289bc588SBarry Smith for ( j=0; j<mat->n; j++ ) { 977289bc588SBarry Smith sum = 0.0; 978289bc588SBarry Smith for ( i=0; i<mat->m; i++ ) { 97933a8263dSBarry Smith sum += PetscAbsScalar(*v); v++; 980289bc588SBarry Smith } 981289bc588SBarry Smith if (sum > *norm) *norm = sum; 982289bc588SBarry Smith } 98355659b69SBarry Smith PLogFlops(mat->n*mat->m); 9843a40ed3dSBarry Smith } else if (type == NORM_INFINITY) { 985289bc588SBarry Smith *norm = 0.0; 986289bc588SBarry Smith for ( j=0; j<mat->m; j++ ) { 987289bc588SBarry Smith v = mat->v + j; 988289bc588SBarry Smith sum = 0.0; 989289bc588SBarry Smith for ( i=0; i<mat->n; i++ ) { 990cddf8d76SBarry Smith sum += PetscAbsScalar(*v); v += mat->m; 991289bc588SBarry Smith } 992289bc588SBarry Smith if (sum > *norm) *norm = sum; 993289bc588SBarry Smith } 99455659b69SBarry Smith PLogFlops(mat->n*mat->m); 9953a40ed3dSBarry Smith } else { 996e3372554SBarry Smith SETERRQ(PETSC_ERR_SUP,0,"No two norm"); 997289bc588SBarry Smith } 9983a40ed3dSBarry Smith PetscFunctionReturn(0); 999289bc588SBarry Smith } 1000289bc588SBarry Smith 10015615d1e5SSatish Balay #undef __FUNC__ 10025615d1e5SSatish Balay #define __FUNC__ "MatSetOption_SeqDense" 10038f6be9afSLois Curfman McInnes int MatSetOption_SeqDense(Mat A,MatOption op) 1004289bc588SBarry Smith { 1005c0bbcb79SLois Curfman McInnes Mat_SeqDense *aij = (Mat_SeqDense *) A->data; 100667e560aaSBarry Smith 10073a40ed3dSBarry Smith PetscFunctionBegin; 10086d4a8577SBarry Smith if (op == MAT_ROW_ORIENTED) aij->roworiented = 1; 10096d4a8577SBarry Smith else if (op == MAT_COLUMN_ORIENTED) aij->roworiented = 0; 10106d4a8577SBarry Smith else if (op == MAT_ROWS_SORTED || 1011219d9a1aSLois Curfman McInnes op == MAT_ROWS_UNSORTED || 10126d4a8577SBarry Smith op == MAT_COLUMNS_SORTED || 1013219d9a1aSLois Curfman McInnes op == MAT_COLUMNS_UNSORTED || 10146d4a8577SBarry Smith op == MAT_SYMMETRIC || 10156d4a8577SBarry Smith op == MAT_STRUCTURALLY_SYMMETRIC || 10166d4a8577SBarry Smith op == MAT_NO_NEW_NONZERO_LOCATIONS || 10176d4a8577SBarry Smith op == MAT_YES_NEW_NONZERO_LOCATIONS || 10184787f768SSatish Balay op == MAT_NEW_NONZERO_LOCATION_ERR || 10196d4a8577SBarry Smith op == MAT_NO_NEW_DIAGONALS || 102090f02eecSBarry Smith op == MAT_YES_NEW_DIAGONALS || 1021b51ba29fSSatish Balay op == MAT_IGNORE_OFF_PROC_ENTRIES || 1022b51ba29fSSatish Balay op == MAT_USE_HASH_TABLE) 1023981c4779SBarry Smith PLogInfo(A,"MatSetOption_SeqDense:Option ignored\n"); 10243a40ed3dSBarry Smith else { 10253a40ed3dSBarry Smith SETERRQ(PETSC_ERR_SUP,0,"unknown option"); 10263a40ed3dSBarry Smith } 10273a40ed3dSBarry Smith PetscFunctionReturn(0); 1028289bc588SBarry Smith } 1029289bc588SBarry Smith 10305615d1e5SSatish Balay #undef __FUNC__ 10315615d1e5SSatish Balay #define __FUNC__ "MatZeroEntries_SeqDense" 10328f6be9afSLois Curfman McInnes int MatZeroEntries_SeqDense(Mat A) 10336f0a148fSBarry Smith { 1034ec8511deSBarry Smith Mat_SeqDense *l = (Mat_SeqDense *) A->data; 1035549d3d68SSatish Balay int ierr; 10363a40ed3dSBarry Smith 10373a40ed3dSBarry Smith PetscFunctionBegin; 1038549d3d68SSatish Balay ierr = PetscMemzero(l->v,l->m*l->n*sizeof(Scalar));CHKERRQ(ierr); 10393a40ed3dSBarry Smith PetscFunctionReturn(0); 10406f0a148fSBarry Smith } 10416f0a148fSBarry Smith 10425615d1e5SSatish Balay #undef __FUNC__ 10435615d1e5SSatish Balay #define __FUNC__ "MatGetBlockSize_SeqDense" 10448f6be9afSLois Curfman McInnes int MatGetBlockSize_SeqDense(Mat A,int *bs) 10454e220ebcSLois Curfman McInnes { 10463a40ed3dSBarry Smith PetscFunctionBegin; 10474e220ebcSLois Curfman McInnes *bs = 1; 10483a40ed3dSBarry Smith PetscFunctionReturn(0); 10494e220ebcSLois Curfman McInnes } 10504e220ebcSLois Curfman McInnes 10515615d1e5SSatish Balay #undef __FUNC__ 10525615d1e5SSatish Balay #define __FUNC__ "MatZeroRows_SeqDense" 10538f6be9afSLois Curfman McInnes int MatZeroRows_SeqDense(Mat A,IS is,Scalar *diag) 10546f0a148fSBarry Smith { 1055ec8511deSBarry Smith Mat_SeqDense *l = (Mat_SeqDense *) A->data; 10566abc6512SBarry Smith int n = l->n, i, j,ierr,N, *rows; 10576f0a148fSBarry Smith Scalar *slot; 105855659b69SBarry Smith 10593a40ed3dSBarry Smith PetscFunctionBegin; 106077c4ece6SBarry Smith ierr = ISGetSize(is,&N);CHKERRQ(ierr); 106178b31e54SBarry Smith ierr = ISGetIndices(is,&rows);CHKERRQ(ierr); 10626f0a148fSBarry Smith for ( i=0; i<N; i++ ) { 10636f0a148fSBarry Smith slot = l->v + rows[i]; 10646f0a148fSBarry Smith for ( j=0; j<n; j++ ) { *slot = 0.0; slot += n;} 10656f0a148fSBarry Smith } 10666f0a148fSBarry Smith if (diag) { 10676f0a148fSBarry Smith for ( i=0; i<N; i++ ) { 10686f0a148fSBarry Smith slot = l->v + (n+1)*rows[i]; 1069260925b8SBarry Smith *slot = *diag; 10706f0a148fSBarry Smith } 10716f0a148fSBarry Smith } 1072260925b8SBarry Smith ISRestoreIndices(is,&rows); 10733a40ed3dSBarry Smith PetscFunctionReturn(0); 10746f0a148fSBarry Smith } 1075557bce09SLois Curfman McInnes 10765615d1e5SSatish Balay #undef __FUNC__ 10775615d1e5SSatish Balay #define __FUNC__ "MatGetSize_SeqDense" 10788f6be9afSLois Curfman McInnes int MatGetSize_SeqDense(Mat A,int *m,int *n) 1079557bce09SLois Curfman McInnes { 1080c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 10813a40ed3dSBarry Smith 10823a40ed3dSBarry Smith PetscFunctionBegin; 1083557bce09SLois Curfman McInnes *m = mat->m; *n = mat->n; 10843a40ed3dSBarry Smith PetscFunctionReturn(0); 1085557bce09SLois Curfman McInnes } 1086557bce09SLois Curfman McInnes 10875615d1e5SSatish Balay #undef __FUNC__ 10885615d1e5SSatish Balay #define __FUNC__ "MatGetOwnershipRange_SeqDense" 10898f6be9afSLois Curfman McInnes int MatGetOwnershipRange_SeqDense(Mat A,int *m,int *n) 1090d3e5ee88SLois Curfman McInnes { 1091c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 10923a40ed3dSBarry Smith 10933a40ed3dSBarry Smith PetscFunctionBegin; 1094d3e5ee88SLois Curfman McInnes *m = 0; *n = mat->m; 10953a40ed3dSBarry Smith PetscFunctionReturn(0); 1096d3e5ee88SLois Curfman McInnes } 1097d3e5ee88SLois Curfman McInnes 10985615d1e5SSatish Balay #undef __FUNC__ 10995615d1e5SSatish Balay #define __FUNC__ "MatGetArray_SeqDense" 11008f6be9afSLois Curfman McInnes int MatGetArray_SeqDense(Mat A,Scalar **array) 110164e87e97SBarry Smith { 1102c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 11033a40ed3dSBarry Smith 11043a40ed3dSBarry Smith PetscFunctionBegin; 110564e87e97SBarry Smith *array = mat->v; 11063a40ed3dSBarry Smith PetscFunctionReturn(0); 110764e87e97SBarry Smith } 11080754003eSLois Curfman McInnes 11095615d1e5SSatish Balay #undef __FUNC__ 11105615d1e5SSatish Balay #define __FUNC__ "MatRestoreArray_SeqDense" 11118f6be9afSLois Curfman McInnes int MatRestoreArray_SeqDense(Mat A,Scalar **array) 1112ff14e315SSatish Balay { 11133a40ed3dSBarry Smith PetscFunctionBegin; 111409b544d4SBarry Smith *array = 0; /* user cannot accidently use the array later */ 11153a40ed3dSBarry Smith PetscFunctionReturn(0); 1116ff14e315SSatish Balay } 11170754003eSLois Curfman McInnes 11185615d1e5SSatish Balay #undef __FUNC__ 11195615d1e5SSatish Balay #define __FUNC__ "MatGetSubMatrix_SeqDense" 11207b2a1423SBarry Smith static int MatGetSubMatrix_SeqDense(Mat A,IS isrow,IS iscol,int cs,MatReuse scall,Mat *B) 11210754003eSLois Curfman McInnes { 1122c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 1123182d2002SSatish Balay int i, j, ierr, m = mat->m, *irow, *icol, nrows, ncols; 1124182d2002SSatish Balay Scalar *av, *bv, *v = mat->v; 11250754003eSLois Curfman McInnes Mat newmat; 11260754003eSLois Curfman McInnes 11273a40ed3dSBarry Smith PetscFunctionBegin; 112878b31e54SBarry Smith ierr = ISGetIndices(isrow,&irow);CHKERRQ(ierr); 112978b31e54SBarry Smith ierr = ISGetIndices(iscol,&icol);CHKERRQ(ierr); 113078b31e54SBarry Smith ierr = ISGetSize(isrow,&nrows);CHKERRQ(ierr); 113178b31e54SBarry Smith ierr = ISGetSize(iscol,&ncols);CHKERRQ(ierr); 11320754003eSLois Curfman McInnes 1133182d2002SSatish Balay /* Check submatrixcall */ 1134182d2002SSatish Balay if (scall == MAT_REUSE_MATRIX) { 1135182d2002SSatish Balay int n_cols,n_rows; 1136182d2002SSatish Balay ierr = MatGetSize(*B,&n_rows,&n_cols);CHKERRQ(ierr); 1137182d2002SSatish Balay if (n_rows != nrows || n_cols != ncols) SETERRQ(PETSC_ERR_ARG_SIZ,0,"Reused submatrix wrong size"); 1138182d2002SSatish Balay newmat = *B; 1139182d2002SSatish Balay } else { 11400754003eSLois Curfman McInnes /* Create and fill new matrix */ 1141b4fd4287SBarry Smith ierr = MatCreateSeqDense(A->comm,nrows,ncols,PETSC_NULL,&newmat);CHKERRQ(ierr); 1142182d2002SSatish Balay } 1143182d2002SSatish Balay 1144182d2002SSatish Balay /* Now extract the data pointers and do the copy, column at a time */ 1145182d2002SSatish Balay bv = ((Mat_SeqDense*)newmat->data)->v; 1146182d2002SSatish Balay 1147182d2002SSatish Balay for ( i=0; i<ncols; i++ ) { 1148182d2002SSatish Balay av = v + m*icol[i]; 1149182d2002SSatish Balay for (j=0; j<nrows; j++ ) { 1150182d2002SSatish Balay *bv++ = av[irow[j]]; 11510754003eSLois Curfman McInnes } 11520754003eSLois Curfman McInnes } 1153182d2002SSatish Balay 1154182d2002SSatish Balay /* Assemble the matrices so that the correct flags are set */ 11556d4a8577SBarry Smith ierr = MatAssemblyBegin(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 11566d4a8577SBarry Smith ierr = MatAssemblyEnd(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 11570754003eSLois Curfman McInnes 11580754003eSLois Curfman McInnes /* Free work space */ 115978b31e54SBarry Smith ierr = ISRestoreIndices(isrow,&irow);CHKERRQ(ierr); 116078b31e54SBarry Smith ierr = ISRestoreIndices(iscol,&icol);CHKERRQ(ierr); 1161182d2002SSatish Balay *B = newmat; 11623a40ed3dSBarry Smith PetscFunctionReturn(0); 11630754003eSLois Curfman McInnes } 11640754003eSLois Curfman McInnes 11655615d1e5SSatish Balay #undef __FUNC__ 11665615d1e5SSatish Balay #define __FUNC__ "MatGetSubMatrices_SeqDense" 11677b2a1423SBarry Smith int MatGetSubMatrices_SeqDense(Mat A,int n, IS *irow,IS *icol,MatReuse scall,Mat **B) 1168905e6a2fSBarry Smith { 1169905e6a2fSBarry Smith int ierr,i; 1170905e6a2fSBarry Smith 11713a40ed3dSBarry Smith PetscFunctionBegin; 1172905e6a2fSBarry Smith if (scall == MAT_INITIAL_MATRIX) { 1173905e6a2fSBarry Smith *B = (Mat *) PetscMalloc( (n+1)*sizeof(Mat) );CHKPTRQ(*B); 1174905e6a2fSBarry Smith } 1175905e6a2fSBarry Smith 1176905e6a2fSBarry Smith for ( i=0; i<n; i++ ) { 11776a6a5d1dSBarry Smith ierr = MatGetSubMatrix_SeqDense(A,irow[i],icol[i],PETSC_DECIDE,scall,&(*B)[i]);CHKERRQ(ierr); 1178905e6a2fSBarry Smith } 11793a40ed3dSBarry Smith PetscFunctionReturn(0); 1180905e6a2fSBarry Smith } 1181905e6a2fSBarry Smith 11825615d1e5SSatish Balay #undef __FUNC__ 11835615d1e5SSatish Balay #define __FUNC__ "MatCopy_SeqDense" 1184cb5b572fSBarry Smith int MatCopy_SeqDense(Mat A, Mat B,MatStructure str) 11854b0e389bSBarry Smith { 11864b0e389bSBarry Smith Mat_SeqDense *a = (Mat_SeqDense *) A->data, *b = (Mat_SeqDense *)B->data; 11873a40ed3dSBarry Smith int ierr; 11883a40ed3dSBarry Smith 11893a40ed3dSBarry Smith PetscFunctionBegin; 11903a40ed3dSBarry Smith if (B->type != MATSEQDENSE) { 1191cb5b572fSBarry Smith ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr); 11923a40ed3dSBarry Smith PetscFunctionReturn(0); 11933a40ed3dSBarry Smith } 1194e3372554SBarry Smith if (a->m != b->m || a->n != b->n) SETERRQ(PETSC_ERR_ARG_SIZ,0,"size(B) != size(A)"); 1195549d3d68SSatish Balay ierr = PetscMemcpy(b->v,a->v,a->m*a->n*sizeof(Scalar));CHKERRQ(ierr); 11963a40ed3dSBarry Smith PetscFunctionReturn(0); 11974b0e389bSBarry Smith } 11984b0e389bSBarry Smith 1199289bc588SBarry Smith /* -------------------------------------------------------------------*/ 1200a5ae1ecdSBarry Smith static struct _MatOps MatOps_Values = {MatSetValues_SeqDense, 1201905e6a2fSBarry Smith MatGetRow_SeqDense, 1202905e6a2fSBarry Smith MatRestoreRow_SeqDense, 1203905e6a2fSBarry Smith MatMult_SeqDense, 1204905e6a2fSBarry Smith MatMultAdd_SeqDense, 1205905e6a2fSBarry Smith MatMultTrans_SeqDense, 1206905e6a2fSBarry Smith MatMultTransAdd_SeqDense, 1207905e6a2fSBarry Smith MatSolve_SeqDense, 1208905e6a2fSBarry Smith MatSolveAdd_SeqDense, 1209905e6a2fSBarry Smith MatSolveTrans_SeqDense, 1210905e6a2fSBarry Smith MatSolveTransAdd_SeqDense, 1211905e6a2fSBarry Smith MatLUFactor_SeqDense, 1212905e6a2fSBarry Smith MatCholeskyFactor_SeqDense, 1213ec8511deSBarry Smith MatRelax_SeqDense, 1214ec8511deSBarry Smith MatTranspose_SeqDense, 1215905e6a2fSBarry Smith MatGetInfo_SeqDense, 1216905e6a2fSBarry Smith MatEqual_SeqDense, 1217905e6a2fSBarry Smith MatGetDiagonal_SeqDense, 1218905e6a2fSBarry Smith MatDiagonalScale_SeqDense, 1219905e6a2fSBarry Smith MatNorm_SeqDense, 1220905e6a2fSBarry Smith 0, 1221905e6a2fSBarry Smith 0, 1222905e6a2fSBarry Smith 0, 1223905e6a2fSBarry Smith MatSetOption_SeqDense, 1224905e6a2fSBarry Smith MatZeroEntries_SeqDense, 1225905e6a2fSBarry Smith MatZeroRows_SeqDense, 1226905e6a2fSBarry Smith MatLUFactorSymbolic_SeqDense, 1227905e6a2fSBarry Smith MatLUFactorNumeric_SeqDense, 1228905e6a2fSBarry Smith MatCholeskyFactorSymbolic_SeqDense, 1229905e6a2fSBarry Smith MatCholeskyFactorNumeric_SeqDense, 1230905e6a2fSBarry Smith MatGetSize_SeqDense, 1231905e6a2fSBarry Smith MatGetSize_SeqDense, 1232905e6a2fSBarry Smith MatGetOwnershipRange_SeqDense, 1233905e6a2fSBarry Smith 0, 1234905e6a2fSBarry Smith 0, 1235905e6a2fSBarry Smith MatGetArray_SeqDense, 1236905e6a2fSBarry Smith MatRestoreArray_SeqDense, 12375609ef8eSBarry Smith MatDuplicate_SeqDense, 1238a5ae1ecdSBarry Smith 0, 1239a5ae1ecdSBarry Smith 0, 1240a5ae1ecdSBarry Smith 0, 1241a5ae1ecdSBarry Smith 0, 1242a5ae1ecdSBarry Smith MatAXPY_SeqDense, 1243a5ae1ecdSBarry Smith MatGetSubMatrices_SeqDense, 1244a5ae1ecdSBarry Smith 0, 12454b0e389bSBarry Smith MatGetValues_SeqDense, 1246a5ae1ecdSBarry Smith MatCopy_SeqDense, 1247a5ae1ecdSBarry Smith 0, 1248a5ae1ecdSBarry Smith MatScale_SeqDense, 1249a5ae1ecdSBarry Smith 0, 1250a5ae1ecdSBarry Smith 0, 1251a5ae1ecdSBarry Smith 0, 1252a5ae1ecdSBarry Smith MatGetBlockSize_SeqDense, 1253a5ae1ecdSBarry Smith 0, 1254a5ae1ecdSBarry Smith 0, 1255a5ae1ecdSBarry Smith 0, 1256a5ae1ecdSBarry Smith 0, 1257a5ae1ecdSBarry Smith 0, 1258a5ae1ecdSBarry Smith 0, 1259a5ae1ecdSBarry Smith 0, 1260a5ae1ecdSBarry Smith 0, 1261a5ae1ecdSBarry Smith 0, 1262a5ae1ecdSBarry Smith 0, 1263a5ae1ecdSBarry Smith 0, 1264a5ae1ecdSBarry Smith 0, 1265a5ae1ecdSBarry Smith MatGetMaps_Petsc}; 126690ace30eSBarry Smith 12675615d1e5SSatish Balay #undef __FUNC__ 12685615d1e5SSatish Balay #define __FUNC__ "MatCreateSeqDense" 12694b828684SBarry Smith /*@C 1270fafbff53SBarry Smith MatCreateSeqDense - Creates a sequential dense matrix that 1271d65003e9SLois Curfman McInnes is stored in column major order (the usual Fortran 77 manner). Many 1272d65003e9SLois Curfman McInnes of the matrix operations use the BLAS and LAPACK routines. 1273289bc588SBarry Smith 1274db81eaa0SLois Curfman McInnes Collective on MPI_Comm 1275db81eaa0SLois Curfman McInnes 127620563c6bSBarry Smith Input Parameters: 1277db81eaa0SLois Curfman McInnes + comm - MPI communicator, set to PETSC_COMM_SELF 12780c775827SLois Curfman McInnes . m - number of rows 127918f449edSLois Curfman McInnes . n - number of columns 1280db81eaa0SLois Curfman McInnes - data - optional location of matrix data. Set data=PETSC_NULL for PETSc 1281dfc5480cSLois Curfman McInnes to control all matrix memory allocation. 128220563c6bSBarry Smith 128320563c6bSBarry Smith Output Parameter: 128444cd7ae7SLois Curfman McInnes . A - the matrix 128520563c6bSBarry Smith 1286b259b22eSLois Curfman McInnes Notes: 128718f449edSLois Curfman McInnes The data input variable is intended primarily for Fortran programmers 128818f449edSLois Curfman McInnes who wish to allocate their own matrix memory space. Most users should 1289b4fd4287SBarry Smith set data=PETSC_NULL. 129018f449edSLois Curfman McInnes 1291027ccd11SLois Curfman McInnes Level: intermediate 1292027ccd11SLois Curfman McInnes 1293dbd7a890SLois Curfman McInnes .keywords: dense, matrix, LAPACK, BLAS 1294d65003e9SLois Curfman McInnes 1295db81eaa0SLois Curfman McInnes .seealso: MatCreate(), MatCreateMPIDense(), MatSetValues() 129620563c6bSBarry Smith @*/ 129744cd7ae7SLois Curfman McInnes int MatCreateSeqDense(MPI_Comm comm,int m,int n,Scalar *data,Mat *A) 1298289bc588SBarry Smith { 129944cd7ae7SLois Curfman McInnes Mat B; 130044cd7ae7SLois Curfman McInnes Mat_SeqDense *b; 13013b2fbd54SBarry Smith int ierr,flg,size; 13023b2fbd54SBarry Smith 13033a40ed3dSBarry Smith PetscFunctionBegin; 1304d132466eSBarry Smith ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr); 1305a8c6a408SBarry Smith if (size > 1) SETERRQ(PETSC_ERR_ARG_WRONG,0,"Comm must be of size 1"); 130655659b69SBarry Smith 130744cd7ae7SLois Curfman McInnes *A = 0; 13083f1db9ecSBarry Smith PetscHeaderCreate(B,_p_Mat,struct _MatOps,MAT_COOKIE,MATSEQDENSE,"Mat",comm,MatDestroy,MatView); 130944cd7ae7SLois Curfman McInnes PLogObjectCreate(B); 131044cd7ae7SLois Curfman McInnes b = (Mat_SeqDense *) PetscMalloc(sizeof(Mat_SeqDense));CHKPTRQ(b); 1311549d3d68SSatish Balay ierr = PetscMemzero(b,sizeof(Mat_SeqDense));CHKERRQ(ierr); 1312549d3d68SSatish Balay ierr = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr); 1313c6e7dd08SBarry Smith B->ops->destroy = MatDestroy_SeqDense; 1314c6e7dd08SBarry Smith B->ops->view = MatView_SeqDense; 131544cd7ae7SLois Curfman McInnes B->factor = 0; 131690f02eecSBarry Smith B->mapping = 0; 1317f09e8eb9SSatish Balay PLogObjectMemory(B,sizeof(struct _p_Mat)); 131844cd7ae7SLois Curfman McInnes B->data = (void *) b; 131918f449edSLois Curfman McInnes 132044cd7ae7SLois Curfman McInnes b->m = m; B->m = m; B->M = m; 132144cd7ae7SLois Curfman McInnes b->n = n; B->n = n; B->N = n; 1322a5ae1ecdSBarry Smith 1323488ecbafSBarry Smith ierr = MapCreateMPI(comm,m,m,&B->rmap);CHKERRQ(ierr); 1324488ecbafSBarry Smith ierr = MapCreateMPI(comm,n,n,&B->cmap);CHKERRQ(ierr); 1325a5ae1ecdSBarry Smith 132644cd7ae7SLois Curfman McInnes b->pivots = 0; 132744cd7ae7SLois Curfman McInnes b->roworiented = 1; 1328b4fd4287SBarry Smith if (data == PETSC_NULL) { 132944cd7ae7SLois Curfman McInnes b->v = (Scalar*) PetscMalloc((m*n+1)*sizeof(Scalar));CHKPTRQ(b->v); 1330549d3d68SSatish Balay ierr = PetscMemzero(b->v,m*n*sizeof(Scalar));CHKERRQ(ierr); 133144cd7ae7SLois Curfman McInnes b->user_alloc = 0; 133244cd7ae7SLois Curfman McInnes PLogObjectMemory(B,n*m*sizeof(Scalar)); 1333273fdc76SBarry Smith } else { /* user-allocated storage */ 133444cd7ae7SLois Curfman McInnes b->v = data; 133544cd7ae7SLois Curfman McInnes b->user_alloc = 1; 13362dd2b441SLois Curfman McInnes } 133725cdf11fSBarry Smith ierr = OptionsHasName(PETSC_NULL,"-help",&flg);CHKERRQ(ierr); 133844cd7ae7SLois Curfman McInnes if (flg) {ierr = MatPrintHelp(B);CHKERRQ(ierr);} 13394e220ebcSLois Curfman McInnes 134044cd7ae7SLois Curfman McInnes *A = B; 13413a40ed3dSBarry Smith PetscFunctionReturn(0); 1342289bc588SBarry Smith } 1343289bc588SBarry Smith 13443b2fbd54SBarry Smith 13453b2fbd54SBarry Smith 13463b2fbd54SBarry Smith 13473b2fbd54SBarry Smith 13483b2fbd54SBarry Smith 1349