1a5eb4965SSatish Balay #ifdef PETSC_RCS_HEADER 2*4a41a4d3SSatish Balay static char vcid[] = "$Id: dense.c,v 1.157 1998/10/01 18:54:20 bsmith Exp balay $"; 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" 10b16a3bb1SBarry Smith #include "pinclude/pviewer.h" 11289bc588SBarry Smith 125615d1e5SSatish Balay #undef __FUNC__ 135615d1e5SSatish Balay #define __FUNC__ "MatAXPY_SeqDense" 141987afe7SBarry Smith int MatAXPY_SeqDense(Scalar *alpha,Mat X,Mat Y) 151987afe7SBarry Smith { 161987afe7SBarry Smith Mat_SeqDense *x = (Mat_SeqDense*) X->data,*y = (Mat_SeqDense*) Y->data; 171987afe7SBarry Smith int N = x->m*x->n, one = 1; 183a40ed3dSBarry Smith 193a40ed3dSBarry Smith PetscFunctionBegin; 201987afe7SBarry Smith BLaxpy_( &N, alpha, x->v, &one, y->v, &one ); 211987afe7SBarry Smith PLogFlops(2*N-1); 223a40ed3dSBarry Smith PetscFunctionReturn(0); 231987afe7SBarry Smith } 241987afe7SBarry Smith 255615d1e5SSatish Balay #undef __FUNC__ 265615d1e5SSatish Balay #define __FUNC__ "MatGetInfo_SeqDense" 278f6be9afSLois Curfman McInnes int MatGetInfo_SeqDense(Mat A,MatInfoType flag,MatInfo *info) 28289bc588SBarry Smith { 294e220ebcSLois Curfman McInnes Mat_SeqDense *a = (Mat_SeqDense *) A->data; 304e220ebcSLois Curfman McInnes int i,N = a->m*a->n,count = 0; 314e220ebcSLois Curfman McInnes Scalar *v = a->v; 323a40ed3dSBarry Smith 333a40ed3dSBarry Smith PetscFunctionBegin; 34289bc588SBarry Smith for ( i=0; i<N; i++ ) {if (*v != 0.0) count++; v++;} 354e220ebcSLois Curfman McInnes 364e220ebcSLois Curfman McInnes info->rows_global = (double)a->m; 374e220ebcSLois Curfman McInnes info->columns_global = (double)a->n; 384e220ebcSLois Curfman McInnes info->rows_local = (double)a->m; 394e220ebcSLois Curfman McInnes info->columns_local = (double)a->n; 404e220ebcSLois Curfman McInnes info->block_size = 1.0; 414e220ebcSLois Curfman McInnes info->nz_allocated = (double)N; 424e220ebcSLois Curfman McInnes info->nz_used = (double)count; 434e220ebcSLois Curfman McInnes info->nz_unneeded = (double)(N-count); 444e220ebcSLois Curfman McInnes info->assemblies = (double)A->num_ass; 454e220ebcSLois Curfman McInnes info->mallocs = 0; 464e220ebcSLois Curfman McInnes info->memory = A->mem; 474e220ebcSLois Curfman McInnes info->fill_ratio_given = 0; 484e220ebcSLois Curfman McInnes info->fill_ratio_needed = 0; 494e220ebcSLois Curfman McInnes info->factor_mallocs = 0; 504e220ebcSLois Curfman McInnes 513a40ed3dSBarry Smith PetscFunctionReturn(0); 52289bc588SBarry Smith } 53289bc588SBarry Smith 545615d1e5SSatish Balay #undef __FUNC__ 555615d1e5SSatish Balay #define __FUNC__ "MatScale_SeqDense" 568f6be9afSLois Curfman McInnes int MatScale_SeqDense(Scalar *alpha,Mat inA) 5780cd9d93SLois Curfman McInnes { 5880cd9d93SLois Curfman McInnes Mat_SeqDense *a = (Mat_SeqDense *) inA->data; 5980cd9d93SLois Curfman McInnes int one = 1, nz; 6080cd9d93SLois Curfman McInnes 613a40ed3dSBarry Smith PetscFunctionBegin; 6280cd9d93SLois Curfman McInnes nz = a->m*a->n; 6380cd9d93SLois Curfman McInnes BLscal_( &nz, alpha, a->v, &one ); 6480cd9d93SLois Curfman McInnes PLogFlops(nz); 653a40ed3dSBarry Smith PetscFunctionReturn(0); 6680cd9d93SLois Curfman McInnes } 6780cd9d93SLois Curfman McInnes 68289bc588SBarry Smith /* ---------------------------------------------------------------*/ 69289bc588SBarry Smith /* COMMENT: I have chosen to hide column permutation in the pivots, 70289bc588SBarry Smith rather than put it in the Mat->col slot.*/ 715615d1e5SSatish Balay #undef __FUNC__ 725615d1e5SSatish Balay #define __FUNC__ "MatLUFactor_SeqDense" 738f6be9afSLois Curfman McInnes int MatLUFactor_SeqDense(Mat A,IS row,IS col,double f) 74289bc588SBarry Smith { 75c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 766abc6512SBarry Smith int info; 7755659b69SBarry Smith 783a40ed3dSBarry Smith PetscFunctionBegin; 79289bc588SBarry Smith if (!mat->pivots) { 8045d48df9SBarry Smith mat->pivots = (int *) PetscMalloc((mat->m+1)*sizeof(int));CHKPTRQ(mat->pivots); 81c0bbcb79SLois Curfman McInnes PLogObjectMemory(A,mat->m*sizeof(int)); 82289bc588SBarry Smith } 837a97a34bSBarry Smith if (!mat->m || !mat->n) PetscFunctionReturn(0); 84289bc588SBarry Smith LAgetrf_(&mat->m,&mat->n,mat->v,&mat->m,mat->pivots,&info); 85a8c6a408SBarry Smith if (info<0) SETERRQ(PETSC_ERR_LIB,0,"Bad argument to LU factorization"); 86e3372554SBarry Smith if (info>0) SETERRQ(PETSC_ERR_MAT_LU_ZRPVT,0,"Bad LU factorization"); 87c0bbcb79SLois Curfman McInnes A->factor = FACTOR_LU; 8855659b69SBarry Smith PLogFlops((2*mat->n*mat->n*mat->n)/3); 893a40ed3dSBarry Smith PetscFunctionReturn(0); 90289bc588SBarry Smith } 916ee01492SSatish Balay 925615d1e5SSatish Balay #undef __FUNC__ 935609ef8eSBarry Smith #define __FUNC__ "MatDuplicate_SeqDense" 945609ef8eSBarry Smith int MatDuplicate_SeqDense(Mat A,MatDuplicateOption cpvalues,Mat *newmat) 9502cad45dSBarry Smith { 9602cad45dSBarry Smith Mat_SeqDense *mat = (Mat_SeqDense *) A->data, *l; 9702cad45dSBarry Smith int ierr; 9802cad45dSBarry Smith Mat newi; 9902cad45dSBarry Smith 1003a40ed3dSBarry Smith PetscFunctionBegin; 10102cad45dSBarry Smith ierr = MatCreateSeqDense(A->comm,mat->m,mat->n,PETSC_NULL,&newi); CHKERRQ(ierr); 10202cad45dSBarry Smith l = (Mat_SeqDense *) newi->data; 1035609ef8eSBarry Smith if (cpvalues == MAT_COPY_VALUES) { 10402cad45dSBarry Smith PetscMemcpy(l->v,mat->v,mat->m*mat->n*sizeof(Scalar)); 10502cad45dSBarry Smith } 10602cad45dSBarry Smith newi->assembled = PETSC_TRUE; 10702cad45dSBarry Smith *newmat = newi; 1083a40ed3dSBarry Smith PetscFunctionReturn(0); 10902cad45dSBarry Smith } 11002cad45dSBarry Smith 1115615d1e5SSatish Balay #undef __FUNC__ 1125615d1e5SSatish Balay #define __FUNC__ "MatLUFactorSymbolic_SeqDense" 1138f6be9afSLois Curfman McInnes int MatLUFactorSymbolic_SeqDense(Mat A,IS row,IS col,double f,Mat *fact) 114289bc588SBarry Smith { 1153a40ed3dSBarry Smith int ierr; 1163a40ed3dSBarry Smith 1173a40ed3dSBarry Smith PetscFunctionBegin; 1185609ef8eSBarry Smith ierr = MatDuplicate_SeqDense(A,MAT_DO_NOT_COPY_VALUES,fact);CHKERRQ(ierr); 1193a40ed3dSBarry Smith PetscFunctionReturn(0); 120289bc588SBarry Smith } 1216ee01492SSatish Balay 1225615d1e5SSatish Balay #undef __FUNC__ 1235615d1e5SSatish Balay #define __FUNC__ "MatLUFactorNumeric_SeqDense" 1248f6be9afSLois Curfman McInnes int MatLUFactorNumeric_SeqDense(Mat A,Mat *fact) 125289bc588SBarry Smith { 12602cad45dSBarry Smith Mat_SeqDense *mat = (Mat_SeqDense*) A->data, *l = (Mat_SeqDense*) (*fact)->data; 1273a40ed3dSBarry Smith int ierr; 1283a40ed3dSBarry Smith 1293a40ed3dSBarry Smith PetscFunctionBegin; 13002cad45dSBarry Smith /* copy the numerical values */ 13102cad45dSBarry Smith PetscMemcpy(l->v,mat->v,mat->m*mat->n*sizeof(Scalar)); 13202cad45dSBarry Smith (*fact)->factor = 0; 1333a40ed3dSBarry Smith ierr = MatLUFactor(*fact,0,0,1.0);CHKERRQ(ierr); 1343a40ed3dSBarry Smith PetscFunctionReturn(0); 135289bc588SBarry Smith } 1366ee01492SSatish Balay 1375615d1e5SSatish Balay #undef __FUNC__ 1385615d1e5SSatish Balay #define __FUNC__ "MatCholeskyFactorSymbolic_SeqDense" 1398f6be9afSLois Curfman McInnes int MatCholeskyFactorSymbolic_SeqDense(Mat A,IS row,double f,Mat *fact) 140289bc588SBarry Smith { 1413a40ed3dSBarry Smith int ierr; 1423a40ed3dSBarry Smith 1433a40ed3dSBarry Smith PetscFunctionBegin; 1443a40ed3dSBarry Smith ierr = MatConvert(A,MATSAME,fact);CHKERRQ(ierr); 1453a40ed3dSBarry Smith PetscFunctionReturn(0); 146289bc588SBarry Smith } 1476ee01492SSatish Balay 1485615d1e5SSatish Balay #undef __FUNC__ 1495615d1e5SSatish Balay #define __FUNC__ "MatCholeskyFactorNumeric_SeqDense" 1508f6be9afSLois Curfman McInnes int MatCholeskyFactorNumeric_SeqDense(Mat A,Mat *fact) 151289bc588SBarry Smith { 1523a40ed3dSBarry Smith int ierr; 1533a40ed3dSBarry Smith 1543a40ed3dSBarry Smith PetscFunctionBegin; 1553a40ed3dSBarry Smith ierr = MatCholeskyFactor(*fact,0,1.0);CHKERRQ(ierr); 1563a40ed3dSBarry Smith PetscFunctionReturn(0); 157289bc588SBarry Smith } 1586ee01492SSatish Balay 1595615d1e5SSatish Balay #undef __FUNC__ 1605615d1e5SSatish Balay #define __FUNC__ "MatCholeskyFactor_SeqDense" 1618f6be9afSLois Curfman McInnes int MatCholeskyFactor_SeqDense(Mat A,IS perm,double f) 162289bc588SBarry Smith { 163c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 1646abc6512SBarry Smith int info; 16555659b69SBarry Smith 1663a40ed3dSBarry Smith PetscFunctionBegin; 167752f5567SLois Curfman McInnes if (mat->pivots) { 1680452661fSBarry Smith PetscFree(mat->pivots); 169c0bbcb79SLois Curfman McInnes PLogObjectMemory(A,-mat->m*sizeof(int)); 170752f5567SLois Curfman McInnes mat->pivots = 0; 171752f5567SLois Curfman McInnes } 1727a97a34bSBarry Smith if (!mat->m || !mat->n) PetscFunctionReturn(0); 173289bc588SBarry Smith LApotrf_("L",&mat->n,mat->v,&mat->m,&info); 174e3372554SBarry Smith if (info) SETERRQ(PETSC_ERR_MAT_CH_ZRPVT,0,"Bad factorization"); 175c0bbcb79SLois Curfman McInnes A->factor = FACTOR_CHOLESKY; 17655659b69SBarry Smith PLogFlops((mat->n*mat->n*mat->n)/3); 1773a40ed3dSBarry Smith PetscFunctionReturn(0); 178289bc588SBarry Smith } 179289bc588SBarry Smith 1805615d1e5SSatish Balay #undef __FUNC__ 1815615d1e5SSatish Balay #define __FUNC__ "MatSolve_SeqDense" 1828f6be9afSLois Curfman McInnes int MatSolve_SeqDense(Mat A,Vec xx,Vec yy) 183289bc588SBarry Smith { 184c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 185a57ad546SLois Curfman McInnes int one = 1, info, ierr; 1866abc6512SBarry Smith Scalar *x, *y; 18767e560aaSBarry Smith 1883a40ed3dSBarry Smith PetscFunctionBegin; 1897a97a34bSBarry Smith if (!mat->m || !mat->n) PetscFunctionReturn(0); 1907a97a34bSBarry Smith ierr = VecGetArray(xx,&x); CHKERRQ(ierr); 1917a97a34bSBarry Smith ierr = VecGetArray(yy,&y); CHKERRQ(ierr); 192416022c9SBarry Smith PetscMemcpy(y,x,mat->m*sizeof(Scalar)); 193c0bbcb79SLois Curfman McInnes if (A->factor == FACTOR_LU) { 19448d91487SBarry Smith LAgetrs_( "N", &mat->m, &one, mat->v, &mat->m, mat->pivots,y, &mat->m, &info ); 1957a97a34bSBarry Smith } else if (A->factor == FACTOR_CHOLESKY){ 19648d91487SBarry Smith LApotrs_( "L", &mat->m, &one, mat->v, &mat->m,y, &mat->m, &info ); 197289bc588SBarry Smith } 198a8c6a408SBarry Smith else SETERRQ(PETSC_ERR_ARG_WRONGSTATE,0,"Matrix must be factored to solve"); 199a8c6a408SBarry Smith if (info) SETERRQ(PETSC_ERR_LIB,0,"MBad solve"); 2007a97a34bSBarry Smith ierr = VecRestoreArray(xx,&x); CHKERRQ(ierr); 2017a97a34bSBarry Smith ierr = VecRestoreArray(yy,&y); CHKERRQ(ierr); 20255659b69SBarry Smith PLogFlops(mat->n*mat->n - mat->n); 2033a40ed3dSBarry Smith PetscFunctionReturn(0); 204289bc588SBarry Smith } 2056ee01492SSatish Balay 2065615d1e5SSatish Balay #undef __FUNC__ 2075615d1e5SSatish Balay #define __FUNC__ "MatSolveTrans_SeqDense" 2088f6be9afSLois Curfman McInnes int MatSolveTrans_SeqDense(Mat A,Vec xx,Vec yy) 209da3a660dSBarry Smith { 210c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 2117a97a34bSBarry Smith int ierr,one = 1, info; 2126abc6512SBarry Smith Scalar *x, *y; 21367e560aaSBarry Smith 2143a40ed3dSBarry Smith PetscFunctionBegin; 2157a97a34bSBarry Smith if (!mat->m || !mat->n) PetscFunctionReturn(0); 2167a97a34bSBarry Smith ierr = VecGetArray(xx,&x); CHKERRQ(ierr); 2177a97a34bSBarry Smith ierr = VecGetArray(yy,&y); CHKERRQ(ierr); 218416022c9SBarry Smith PetscMemcpy(y,x,mat->m*sizeof(Scalar)); 219752f5567SLois Curfman McInnes /* assume if pivots exist then use LU; else Cholesky */ 220da3a660dSBarry Smith if (mat->pivots) { 22148d91487SBarry Smith LAgetrs_( "T", &mat->m, &one, mat->v, &mat->m, mat->pivots,y, &mat->m, &info ); 2227a97a34bSBarry Smith } else { 22348d91487SBarry Smith LApotrs_( "L", &mat->m, &one, mat->v, &mat->m,y, &mat->m, &info ); 224da3a660dSBarry Smith } 225a8c6a408SBarry Smith if (info) SETERRQ(PETSC_ERR_LIB,0,"Bad solve"); 2267a97a34bSBarry Smith ierr = VecRestoreArray(xx,&x); CHKERRQ(ierr); 2277a97a34bSBarry Smith ierr = VecRestoreArray(yy,&y); CHKERRQ(ierr); 22855659b69SBarry Smith PLogFlops(mat->n*mat->n - mat->n); 2293a40ed3dSBarry Smith PetscFunctionReturn(0); 230da3a660dSBarry Smith } 2316ee01492SSatish Balay 2325615d1e5SSatish Balay #undef __FUNC__ 2335615d1e5SSatish Balay #define __FUNC__ "MatSolveAdd_SeqDense" 2348f6be9afSLois Curfman McInnes int MatSolveAdd_SeqDense(Mat A,Vec xx,Vec zz,Vec yy) 235da3a660dSBarry Smith { 236c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 2377a97a34bSBarry Smith int ierr,one = 1, info; 2386abc6512SBarry Smith Scalar *x, *y, sone = 1.0; 239da3a660dSBarry Smith Vec tmp = 0; 24067e560aaSBarry Smith 2413a40ed3dSBarry Smith PetscFunctionBegin; 2427a97a34bSBarry Smith ierr = VecGetArray(xx,&x); CHKERRQ(ierr); 2437a97a34bSBarry Smith ierr = VecGetArray(yy,&y); CHKERRQ(ierr); 2447a97a34bSBarry Smith if (!mat->m || !mat->n) PetscFunctionReturn(0); 245da3a660dSBarry Smith if (yy == zz) { 24678b31e54SBarry Smith ierr = VecDuplicate(yy,&tmp); CHKERRQ(ierr); 247c0bbcb79SLois Curfman McInnes PLogObjectParent(A,tmp); 24878b31e54SBarry Smith ierr = VecCopy(yy,tmp); CHKERRQ(ierr); 249da3a660dSBarry Smith } 250416022c9SBarry Smith PetscMemcpy(y,x,mat->m*sizeof(Scalar)); 251752f5567SLois Curfman McInnes /* assume if pivots exist then use LU; else Cholesky */ 252da3a660dSBarry Smith if (mat->pivots) { 25348d91487SBarry Smith LAgetrs_( "N", &mat->m, &one, mat->v, &mat->m, mat->pivots,y, &mat->m, &info ); 254a8c6a408SBarry Smith } else { 25548d91487SBarry Smith LApotrs_( "L", &mat->m, &one, mat->v, &mat->m,y, &mat->m, &info ); 256da3a660dSBarry Smith } 257a8c6a408SBarry Smith if (info) SETERRQ(PETSC_ERR_LIB,0,"Bad solve"); 258a8c6a408SBarry Smith if (tmp) {ierr = VecAXPY(&sone,tmp,yy); CHKERRQ(ierr); ierr = VecDestroy(tmp);CHKERRQ(ierr);} 259a8c6a408SBarry Smith else {ierr = VecAXPY(&sone,zz,yy); CHKERRQ(ierr);} 2607a97a34bSBarry Smith ierr = VecRestoreArray(xx,&x); CHKERRQ(ierr); 2617a97a34bSBarry Smith ierr = VecRestoreArray(yy,&y); CHKERRQ(ierr); 26255659b69SBarry Smith PLogFlops(mat->n*mat->n - mat->n); 2633a40ed3dSBarry Smith PetscFunctionReturn(0); 264da3a660dSBarry Smith } 26567e560aaSBarry Smith 2665615d1e5SSatish Balay #undef __FUNC__ 2675615d1e5SSatish Balay #define __FUNC__ "MatSolveTransAdd_SeqDense" 2688f6be9afSLois Curfman McInnes int MatSolveTransAdd_SeqDense(Mat A,Vec xx,Vec zz, Vec yy) 269da3a660dSBarry Smith { 270c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 2716abc6512SBarry Smith int one = 1, info,ierr; 2726abc6512SBarry Smith Scalar *x, *y, sone = 1.0; 273da3a660dSBarry Smith Vec tmp; 27467e560aaSBarry Smith 2753a40ed3dSBarry Smith PetscFunctionBegin; 2767a97a34bSBarry Smith if (!mat->m || !mat->n) PetscFunctionReturn(0); 2777a97a34bSBarry Smith ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 2787a97a34bSBarry Smith ierr = VecGetArray(yy,&y);CHKERRQ(ierr); 279da3a660dSBarry Smith if (yy == zz) { 28078b31e54SBarry Smith ierr = VecDuplicate(yy,&tmp); CHKERRQ(ierr); 281c0bbcb79SLois Curfman McInnes PLogObjectParent(A,tmp); 28278b31e54SBarry Smith ierr = VecCopy(yy,tmp); CHKERRQ(ierr); 283da3a660dSBarry Smith } 284416022c9SBarry Smith PetscMemcpy(y,x,mat->m*sizeof(Scalar)); 285752f5567SLois Curfman McInnes /* assume if pivots exist then use LU; else Cholesky */ 286da3a660dSBarry Smith if (mat->pivots) { 28748d91487SBarry Smith LAgetrs_( "T", &mat->m, &one, mat->v, &mat->m, mat->pivots,y, &mat->m, &info ); 2883a40ed3dSBarry Smith } else { 28948d91487SBarry Smith LApotrs_( "L", &mat->m, &one, mat->v, &mat->m,y, &mat->m, &info ); 290da3a660dSBarry Smith } 291a8c6a408SBarry Smith if (info) SETERRQ(PETSC_ERR_LIB,0,"Bad solve"); 29290f02eecSBarry Smith if (tmp) { 29390f02eecSBarry Smith ierr = VecAXPY(&sone,tmp,yy); CHKERRQ(ierr); 29490f02eecSBarry Smith ierr = VecDestroy(tmp); CHKERRQ(ierr); 2953a40ed3dSBarry Smith } else { 29690f02eecSBarry Smith ierr = VecAXPY(&sone,zz,yy); CHKERRQ(ierr); 29790f02eecSBarry Smith } 2987a97a34bSBarry Smith ierr = VecRestoreArray(xx,&x); CHKERRQ(ierr); 2997a97a34bSBarry Smith ierr = VecRestoreArray(yy,&y); CHKERRQ(ierr); 30055659b69SBarry Smith PLogFlops(mat->n*mat->n - mat->n); 3013a40ed3dSBarry Smith PetscFunctionReturn(0); 302da3a660dSBarry Smith } 303289bc588SBarry Smith /* ------------------------------------------------------------------*/ 3045615d1e5SSatish Balay #undef __FUNC__ 3055615d1e5SSatish Balay #define __FUNC__ "MatRelax_SeqDense" 3068f6be9afSLois Curfman McInnes int MatRelax_SeqDense(Mat A,Vec bb,double omega,MatSORType flag, 30720e1a0d4SLois Curfman McInnes double shift,int its,Vec xx) 308289bc588SBarry Smith { 309c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 310289bc588SBarry Smith Scalar *x, *b, *v = mat->v, zero = 0.0, xt; 311bc1b551cSSatish Balay int ierr, m = mat->m, i; 312bc1b551cSSatish Balay #if !defined(USE_PETSC_COMPLEX) 313bc1b551cSSatish Balay int o = 1; 314bc1b551cSSatish Balay #endif 315289bc588SBarry Smith 3163a40ed3dSBarry Smith PetscFunctionBegin; 317289bc588SBarry Smith if (flag & SOR_ZERO_INITIAL_GUESS) { 3183a40ed3dSBarry Smith /* this is a hack fix, should have another version without the second BLdot */ 319bbb6d6a8SBarry Smith ierr = VecSet(&zero,xx); CHKERRQ(ierr); 320289bc588SBarry Smith } 3217a97a34bSBarry Smith ierr = VecGetArray(xx,&x);CHKERRQ(ierr); 3227a97a34bSBarry Smith ierr = VecGetArray(bb,&b);CHKERRQ(ierr); 323289bc588SBarry Smith while (its--) { 324289bc588SBarry Smith if (flag & SOR_FORWARD_SWEEP){ 325289bc588SBarry Smith for ( i=0; i<m; i++ ) { 3263a40ed3dSBarry Smith #if defined(USE_PETSC_COMPLEX) 327f1747703SBarry Smith /* cannot use BLAS dot for complex because compiler/linker is 328f1747703SBarry Smith not happy about returning a double complex */ 329f1747703SBarry Smith int _i; 330f1747703SBarry Smith Scalar sum = b[i]; 331f1747703SBarry Smith for ( _i=0; _i<m; _i++ ) { 3323f6de6efSSatish Balay sum -= PetscConj(v[i+_i*m])*x[_i]; 333f1747703SBarry Smith } 334f1747703SBarry Smith xt = sum; 335f1747703SBarry Smith #else 336289bc588SBarry Smith xt = b[i]-BLdot_(&m,v+i,&m,x,&o); 337f1747703SBarry Smith #endif 33855a1b374SBarry Smith x[i] = (1. - omega)*x[i] + omega*(xt+v[i + i*m]*x[i])/(v[i + i*m]+shift); 339289bc588SBarry Smith } 340289bc588SBarry Smith } 341289bc588SBarry Smith if (flag & SOR_BACKWARD_SWEEP) { 342289bc588SBarry Smith for ( i=m-1; i>=0; i-- ) { 3433a40ed3dSBarry Smith #if defined(USE_PETSC_COMPLEX) 344f1747703SBarry Smith /* cannot use BLAS dot for complex because compiler/linker is 345f1747703SBarry Smith not happy about returning a double complex */ 346f1747703SBarry Smith int _i; 347f1747703SBarry Smith Scalar sum = b[i]; 348f1747703SBarry Smith for ( _i=0; _i<m; _i++ ) { 3493f6de6efSSatish Balay sum -= PetscConj(v[i+_i*m])*x[_i]; 350f1747703SBarry Smith } 351f1747703SBarry Smith xt = sum; 352f1747703SBarry Smith #else 353289bc588SBarry Smith xt = b[i]-BLdot_(&m,v+i,&m,x,&o); 354f1747703SBarry Smith #endif 35555a1b374SBarry Smith x[i] = (1. - omega)*x[i] + omega*(xt+v[i + i*m]*x[i])/(v[i + i*m]+shift); 356289bc588SBarry Smith } 357289bc588SBarry Smith } 358289bc588SBarry Smith } 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; 407289bc588SBarry Smith Scalar *v = mat->v, *x, *y, *z; 4087a97a34bSBarry Smith int ierr,_One=1; Scalar _DOne=1.0; 4093a40ed3dSBarry Smith 4103a40ed3dSBarry Smith PetscFunctionBegin; 4117a97a34bSBarry Smith if (!mat->m || !mat->n) PetscFunctionReturn(0); 4127a97a34bSBarry Smith ierr = VecGetArray(xx,&x); CHKERRQ(ierr); 4137a97a34bSBarry Smith ierr = VecGetArray(yy,&y); CHKERRQ(ierr); 4147a97a34bSBarry Smith ierr = VecGetArray(zz,&z);CHKERRQ(ierr); 415416022c9SBarry Smith if (zz != yy) PetscMemcpy(y,z,mat->m*sizeof(Scalar)); 41648d91487SBarry Smith LAgemv_( "N", &(mat->m), &(mat->n),&_DOne,v,&(mat->m),x,&_One,&_DOne,y,&_One); 4177a97a34bSBarry Smith ierr = VecRestoreArray(xx,&x); CHKERRQ(ierr); 4187a97a34bSBarry Smith ierr = VecRestoreArray(yy,&y); CHKERRQ(ierr); 4197a97a34bSBarry Smith ierr = VecRestoreArray(zz,&z); CHKERRQ(ierr); 42055659b69SBarry Smith PLogFlops(2*mat->m*mat->n); 4213a40ed3dSBarry Smith PetscFunctionReturn(0); 422289bc588SBarry Smith } 4236ee01492SSatish Balay 4245615d1e5SSatish Balay #undef __FUNC__ 4255615d1e5SSatish Balay #define __FUNC__ "MatMultTransAdd_SeqDense" 42644cd7ae7SLois Curfman McInnes int MatMultTransAdd_SeqDense(Mat A,Vec xx,Vec zz,Vec yy) 427289bc588SBarry Smith { 428c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 429289bc588SBarry Smith Scalar *v = mat->v, *x, *y, *z; 4307a97a34bSBarry Smith int ierr,_One=1; 4316abc6512SBarry Smith Scalar _DOne=1.0; 4323a40ed3dSBarry Smith 4333a40ed3dSBarry Smith PetscFunctionBegin; 4347a97a34bSBarry Smith if (!mat->m || !mat->n) PetscFunctionReturn(0); 4357a97a34bSBarry Smith ierr = VecGetArray(xx,&x); CHKERRQ(ierr); 4367a97a34bSBarry Smith ierr = VecGetArray(yy,&y); CHKERRQ(ierr); 4377a97a34bSBarry Smith ierr = VecGetArray(zz,&z);CHKERRQ(ierr); 438717eeb74SLois Curfman McInnes if (zz != yy) PetscMemcpy(y,z,mat->n*sizeof(Scalar)); 43948d91487SBarry Smith LAgemv_( "T", &(mat->m), &(mat->n), &_DOne, v, &(mat->m),x,&_One,&_DOne,y,&_One); 4407a97a34bSBarry Smith ierr = VecRestoreArray(xx,&x); CHKERRQ(ierr); 4417a97a34bSBarry Smith ierr = VecRestoreArray(yy,&y); CHKERRQ(ierr); 4427a97a34bSBarry Smith ierr = VecRestoreArray(zz,&z); CHKERRQ(ierr); 44355659b69SBarry Smith PLogFlops(2*mat->m*mat->n); 4443a40ed3dSBarry Smith PetscFunctionReturn(0); 445289bc588SBarry Smith } 446289bc588SBarry Smith 447289bc588SBarry Smith /* -----------------------------------------------------------------*/ 4485615d1e5SSatish Balay #undef __FUNC__ 4495615d1e5SSatish Balay #define __FUNC__ "MatGetRow_SeqDense" 4508f6be9afSLois Curfman McInnes int MatGetRow_SeqDense(Mat A,int row,int *ncols,int **cols,Scalar **vals) 451289bc588SBarry Smith { 452c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 453289bc588SBarry Smith Scalar *v; 454289bc588SBarry Smith int i; 45567e560aaSBarry Smith 4563a40ed3dSBarry Smith PetscFunctionBegin; 457289bc588SBarry Smith *ncols = mat->n; 458289bc588SBarry Smith if (cols) { 45945d48df9SBarry Smith *cols = (int *) PetscMalloc((mat->n+1)*sizeof(int)); CHKPTRQ(*cols); 46073c3ce57SBarry Smith for ( i=0; i<mat->n; i++ ) (*cols)[i] = i; 461289bc588SBarry Smith } 462289bc588SBarry Smith if (vals) { 46345d48df9SBarry Smith *vals = (Scalar *) PetscMalloc((mat->n+1)*sizeof(Scalar)); CHKPTRQ(*vals); 464289bc588SBarry Smith v = mat->v + row; 46573c3ce57SBarry Smith for ( i=0; i<mat->n; i++ ) {(*vals)[i] = *v; v += mat->m;} 466289bc588SBarry Smith } 4673a40ed3dSBarry Smith PetscFunctionReturn(0); 468289bc588SBarry Smith } 4696ee01492SSatish Balay 4705615d1e5SSatish Balay #undef __FUNC__ 4715615d1e5SSatish Balay #define __FUNC__ "MatRestoreRow_SeqDense" 4728f6be9afSLois Curfman McInnes int MatRestoreRow_SeqDense(Mat A,int row,int *ncols,int **cols,Scalar **vals) 473289bc588SBarry Smith { 4740452661fSBarry Smith if (cols) { PetscFree(*cols); } 4750452661fSBarry Smith if (vals) { PetscFree(*vals); } 4763a40ed3dSBarry Smith PetscFunctionReturn(0); 477289bc588SBarry Smith } 478289bc588SBarry Smith /* ----------------------------------------------------------------*/ 4795615d1e5SSatish Balay #undef __FUNC__ 4805615d1e5SSatish Balay #define __FUNC__ "MatSetValues_SeqDense" 4818f6be9afSLois Curfman McInnes int MatSetValues_SeqDense(Mat A,int m,int *indexm,int n, 482e8d4e0b9SBarry Smith int *indexn,Scalar *v,InsertMode addv) 483289bc588SBarry Smith { 484c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 485289bc588SBarry Smith int i,j; 486d6dfbf8fSBarry Smith 4873a40ed3dSBarry Smith PetscFunctionBegin; 488289bc588SBarry Smith if (!mat->roworiented) { 489dbb450caSBarry Smith if (addv == INSERT_VALUES) { 490289bc588SBarry Smith for ( j=0; j<n; j++ ) { 4913a40ed3dSBarry Smith #if defined(USE_PETSC_BOPT_g) 492a8c6a408SBarry Smith if (indexn[j] < 0) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Negative column"); 493a8c6a408SBarry Smith if (indexn[j] >= A->n) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Column too large"); 49458804f6dSBarry Smith #endif 495289bc588SBarry Smith for ( i=0; i<m; i++ ) { 4963a40ed3dSBarry Smith #if defined(USE_PETSC_BOPT_g) 497a8c6a408SBarry Smith if (indexm[i] < 0) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Negative row"); 498a8c6a408SBarry Smith if (indexm[i] >= A->m) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Row too large"); 49958804f6dSBarry Smith #endif 500289bc588SBarry Smith mat->v[indexn[j]*mat->m + indexm[i]] = *v++; 501289bc588SBarry Smith } 502289bc588SBarry Smith } 5033a40ed3dSBarry Smith } else { 504289bc588SBarry Smith for ( j=0; j<n; j++ ) { 5053a40ed3dSBarry Smith #if defined(USE_PETSC_BOPT_g) 506a8c6a408SBarry Smith if (indexn[j] < 0) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Negative column"); 507a8c6a408SBarry Smith if (indexn[j] >= A->n) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Column too large"); 50858804f6dSBarry Smith #endif 509289bc588SBarry Smith for ( i=0; i<m; i++ ) { 5103a40ed3dSBarry Smith #if defined(USE_PETSC_BOPT_g) 511a8c6a408SBarry Smith if (indexm[i] < 0) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Negative row"); 512a8c6a408SBarry Smith if (indexm[i] >= A->m) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Row too large"); 51358804f6dSBarry Smith #endif 514289bc588SBarry Smith mat->v[indexn[j]*mat->m + indexm[i]] += *v++; 515289bc588SBarry Smith } 516289bc588SBarry Smith } 517289bc588SBarry Smith } 5183a40ed3dSBarry Smith } else { 519dbb450caSBarry Smith if (addv == INSERT_VALUES) { 520e8d4e0b9SBarry Smith for ( i=0; i<m; i++ ) { 5213a40ed3dSBarry Smith #if defined(USE_PETSC_BOPT_g) 522a8c6a408SBarry Smith if (indexm[i] < 0) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Negative row"); 523a8c6a408SBarry Smith if (indexm[i] >= A->m) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Row too large"); 52458804f6dSBarry Smith #endif 525e8d4e0b9SBarry Smith for ( j=0; j<n; j++ ) { 5263a40ed3dSBarry Smith #if defined(USE_PETSC_BOPT_g) 527a8c6a408SBarry Smith if (indexn[j] < 0) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Negative column"); 528a8c6a408SBarry Smith if (indexn[j] >= A->n) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Column too large"); 52958804f6dSBarry Smith #endif 530e8d4e0b9SBarry Smith mat->v[indexn[j]*mat->m + indexm[i]] = *v++; 531e8d4e0b9SBarry Smith } 532e8d4e0b9SBarry Smith } 5333a40ed3dSBarry Smith } else { 534289bc588SBarry Smith for ( i=0; i<m; i++ ) { 5353a40ed3dSBarry Smith #if defined(USE_PETSC_BOPT_g) 536a8c6a408SBarry Smith if (indexm[i] < 0) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Negative row"); 537a8c6a408SBarry Smith if (indexm[i] >= A->m) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Row too large"); 53858804f6dSBarry Smith #endif 539289bc588SBarry Smith for ( j=0; j<n; j++ ) { 5403a40ed3dSBarry Smith #if defined(USE_PETSC_BOPT_g) 541a8c6a408SBarry Smith if (indexn[j] < 0) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Negative column"); 542a8c6a408SBarry Smith if (indexn[j] >= A->n) SETERRQ(PETSC_ERR_ARG_OUTOFRANGE,0,"Column too large"); 54358804f6dSBarry Smith #endif 544289bc588SBarry Smith mat->v[indexn[j]*mat->m + indexm[i]] += *v++; 545289bc588SBarry Smith } 546289bc588SBarry Smith } 547289bc588SBarry Smith } 548e8d4e0b9SBarry Smith } 5493a40ed3dSBarry Smith PetscFunctionReturn(0); 550289bc588SBarry Smith } 551e8d4e0b9SBarry Smith 5525615d1e5SSatish Balay #undef __FUNC__ 5535615d1e5SSatish Balay #define __FUNC__ "MatGetValues_SeqDense" 5548f6be9afSLois Curfman McInnes int MatGetValues_SeqDense(Mat A,int m,int *indexm,int n,int *indexn,Scalar *v) 555ae80bb75SLois Curfman McInnes { 556ae80bb75SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 557ae80bb75SLois Curfman McInnes int i, j; 558ae80bb75SLois Curfman McInnes Scalar *vpt = v; 559ae80bb75SLois Curfman McInnes 5603a40ed3dSBarry Smith PetscFunctionBegin; 561ae80bb75SLois Curfman McInnes /* row-oriented output */ 562ae80bb75SLois Curfman McInnes for ( i=0; i<m; i++ ) { 563ae80bb75SLois Curfman McInnes for ( j=0; j<n; j++ ) { 564ae80bb75SLois Curfman McInnes *vpt++ = mat->v[indexn[j]*mat->m + indexm[i]]; 565ae80bb75SLois Curfman McInnes } 566ae80bb75SLois Curfman McInnes } 5673a40ed3dSBarry Smith PetscFunctionReturn(0); 568ae80bb75SLois Curfman McInnes } 569ae80bb75SLois Curfman McInnes 570289bc588SBarry Smith /* -----------------------------------------------------------------*/ 571289bc588SBarry Smith 57277c4ece6SBarry Smith #include "sys.h" 57388e32edaSLois Curfman McInnes 5745615d1e5SSatish Balay #undef __FUNC__ 5755615d1e5SSatish Balay #define __FUNC__ "MatLoad_SeqDense" 57619bcc07fSBarry Smith int MatLoad_SeqDense(Viewer viewer,MatType type,Mat *A) 57788e32edaSLois Curfman McInnes { 57888e32edaSLois Curfman McInnes Mat_SeqDense *a; 57988e32edaSLois Curfman McInnes Mat B; 58088e32edaSLois Curfman McInnes int *scols, i, j, nz, ierr, fd, header[4], size; 58188e32edaSLois Curfman McInnes int *rowlengths = 0, M, N, *cols; 582*4a41a4d3SSatish Balay Scalar *vals, *svals, *v,*w; 58319bcc07fSBarry Smith MPI_Comm comm = ((PetscObject)viewer)->comm; 58488e32edaSLois Curfman McInnes 5853a40ed3dSBarry Smith PetscFunctionBegin; 58688e32edaSLois Curfman McInnes MPI_Comm_size(comm,&size); 587a8c6a408SBarry Smith if (size > 1) SETERRQ(PETSC_ERR_ARG_WRONG,0,"view must have one processor"); 58890ace30eSBarry Smith ierr = ViewerBinaryGetDescriptor(viewer,&fd); CHKERRQ(ierr); 5890752156aSBarry Smith ierr = PetscBinaryRead(fd,header,4,PETSC_INT); CHKERRQ(ierr); 590a8c6a408SBarry Smith if (header[0] != MAT_COOKIE) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,0,"Not matrix object"); 59188e32edaSLois Curfman McInnes M = header[1]; N = header[2]; nz = header[3]; 59288e32edaSLois Curfman McInnes 59390ace30eSBarry Smith if (nz == MATRIX_BINARY_FORMAT_DENSE) { /* matrix in file is dense */ 59490ace30eSBarry Smith ierr = MatCreateSeqDense(comm,M,N,PETSC_NULL,A); CHKERRQ(ierr); 59590ace30eSBarry Smith B = *A; 59690ace30eSBarry Smith a = (Mat_SeqDense *) B->data; 597*4a41a4d3SSatish Balay v = a->v; 598*4a41a4d3SSatish Balay /* Allocate some temp space to read in the values and then flip them 599*4a41a4d3SSatish Balay from row major to column major */ 600*4a41a4d3SSatish Balay w = (Scalar *) PetscMalloc((M*N+1)*sizeof(Scalar));CHKPTRQ(w); 60190ace30eSBarry Smith /* read in nonzero values */ 602*4a41a4d3SSatish Balay ierr = PetscBinaryRead(fd,w,M*N,PETSC_SCALAR); CHKERRQ(ierr); 603*4a41a4d3SSatish Balay /* now flip the values and store them in the matrix*/ 604*4a41a4d3SSatish Balay for ( j=0; j<N; j++ ) { 605*4a41a4d3SSatish Balay for ( i=0; i<M; i++ ) { 606*4a41a4d3SSatish Balay *v++ =w[i*N+j]; 607*4a41a4d3SSatish Balay } 608*4a41a4d3SSatish Balay } 609*4a41a4d3SSatish Balay PetscFree(w); 6106d4a8577SBarry Smith ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 6116d4a8577SBarry Smith ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 61290ace30eSBarry Smith } else { 61388e32edaSLois Curfman McInnes /* read row lengths */ 61445d48df9SBarry Smith rowlengths = (int*) PetscMalloc( (M+1)*sizeof(int) ); CHKPTRQ(rowlengths); 6150752156aSBarry Smith ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT); CHKERRQ(ierr); 61688e32edaSLois Curfman McInnes 61788e32edaSLois Curfman McInnes /* create our matrix */ 618b4fd4287SBarry Smith ierr = MatCreateSeqDense(comm,M,N,PETSC_NULL,A); CHKERRQ(ierr); 61988e32edaSLois Curfman McInnes B = *A; 62088e32edaSLois Curfman McInnes a = (Mat_SeqDense *) B->data; 62188e32edaSLois Curfman McInnes v = a->v; 62288e32edaSLois Curfman McInnes 62388e32edaSLois Curfman McInnes /* read column indices and nonzeros */ 62445d48df9SBarry Smith cols = scols = (int *) PetscMalloc( (nz+1)*sizeof(int) ); CHKPTRQ(cols); 6250752156aSBarry Smith ierr = PetscBinaryRead(fd,cols,nz,PETSC_INT); CHKERRQ(ierr); 62645d48df9SBarry Smith vals = svals = (Scalar *) PetscMalloc( (nz+1)*sizeof(Scalar) ); CHKPTRQ(vals); 6270752156aSBarry Smith ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR); CHKERRQ(ierr); 62888e32edaSLois Curfman McInnes 62988e32edaSLois Curfman McInnes /* insert into matrix */ 63088e32edaSLois Curfman McInnes for ( i=0; i<M; i++ ) { 63188e32edaSLois Curfman McInnes for ( j=0; j<rowlengths[i]; j++ ) v[i+M*scols[j]] = svals[j]; 63288e32edaSLois Curfman McInnes svals += rowlengths[i]; scols += rowlengths[i]; 63388e32edaSLois Curfman McInnes } 6340452661fSBarry Smith PetscFree(vals); PetscFree(cols); PetscFree(rowlengths); 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 642932b0c3eSLois Curfman McInnes #include "pinclude/pviewer.h" 64377c4ece6SBarry Smith #include "sys.h" 644932b0c3eSLois Curfman McInnes 6455615d1e5SSatish Balay #undef __FUNC__ 6465615d1e5SSatish Balay #define __FUNC__ "MatView_SeqDense_ASCII" 647932b0c3eSLois Curfman McInnes static int MatView_SeqDense_ASCII(Mat A,Viewer viewer) 648289bc588SBarry Smith { 649932b0c3eSLois Curfman McInnes Mat_SeqDense *a = (Mat_SeqDense *) A->data; 650932b0c3eSLois Curfman McInnes int ierr, i, j, format; 651d636dbe3SBarry Smith FILE *fd; 652932b0c3eSLois Curfman McInnes char *outputname; 653932b0c3eSLois Curfman McInnes Scalar *v; 654932b0c3eSLois Curfman McInnes 6553a40ed3dSBarry Smith PetscFunctionBegin; 65690ace30eSBarry Smith ierr = ViewerASCIIGetPointer(viewer,&fd); CHKERRQ(ierr); 657932b0c3eSLois Curfman McInnes ierr = ViewerFileGetOutputname_Private(viewer,&outputname); CHKERRQ(ierr); 65890ace30eSBarry Smith ierr = ViewerGetFormat(viewer,&format); 659639f9d9dSBarry Smith if (format == VIEWER_FORMAT_ASCII_INFO || format == VIEWER_FORMAT_ASCII_INFO_LONG) { 6603a40ed3dSBarry Smith PetscFunctionReturn(0); /* do nothing for now */ 661f72585f2SLois Curfman McInnes } 66202cad45dSBarry Smith else if (format == VIEWER_FORMAT_ASCII_COMMON) { 66380cd9d93SLois Curfman McInnes for ( i=0; i<a->m; i++ ) { 66444cd7ae7SLois Curfman McInnes v = a->v + i; 66580cd9d93SLois Curfman McInnes fprintf(fd,"row %d:",i); 66680cd9d93SLois Curfman McInnes for ( j=0; j<a->n; j++ ) { 6673a40ed3dSBarry Smith #if defined(USE_PETSC_COMPLEX) 668e20fef11SSatish Balay if (PetscReal(*v) != 0.0 && PetscImaginary(*v) != 0.0) fprintf(fd," %d %g + %g i",j,PetscReal(*v),PetscImaginary(*v)); 6693f6de6efSSatish Balay else if (PetscReal(*v)) fprintf(fd," %d %g ",j,PetscReal(*v)); 67080cd9d93SLois Curfman McInnes #else 67180cd9d93SLois Curfman McInnes if (*v) fprintf(fd," %d %g ",j,*v); 67280cd9d93SLois Curfman McInnes #endif 673d64ed03dSBarry Smith v += a->m; 67480cd9d93SLois Curfman McInnes } 67580cd9d93SLois Curfman McInnes fprintf(fd,"\n"); 67680cd9d93SLois Curfman McInnes } 6773a40ed3dSBarry Smith } else { 6783a40ed3dSBarry Smith #if defined(USE_PETSC_COMPLEX) 67947989497SBarry Smith int allreal = 1; 68047989497SBarry Smith /* determine if matrix has all real values */ 68147989497SBarry Smith v = a->v; 68247989497SBarry Smith for ( i=0; i<a->m*a->n; i++ ) { 683e20fef11SSatish Balay if (PetscImaginary(v[i])) { allreal = 0; break ;} 68447989497SBarry Smith } 68547989497SBarry Smith #endif 686932b0c3eSLois Curfman McInnes for ( i=0; i<a->m; i++ ) { 687932b0c3eSLois Curfman McInnes v = a->v + i; 688932b0c3eSLois Curfman McInnes for ( j=0; j<a->n; j++ ) { 6893a40ed3dSBarry Smith #if defined(USE_PETSC_COMPLEX) 69047989497SBarry Smith if (allreal) { 6913f6de6efSSatish Balay fprintf(fd,"%6.4e ",PetscReal(*v)); v += a->m; 69247989497SBarry Smith } else { 693e20fef11SSatish Balay fprintf(fd,"%6.4e + %6.4e i ",PetscReal(*v),PetscImaginary(*v)); v += a->m; 69447989497SBarry Smith } 695289bc588SBarry Smith #else 696932b0c3eSLois Curfman McInnes fprintf(fd,"%6.4e ",*v); v += a->m; 697289bc588SBarry Smith #endif 698289bc588SBarry Smith } 6998e37dc09SBarry Smith fprintf(fd,"\n"); 700289bc588SBarry Smith } 701da3a660dSBarry Smith } 702932b0c3eSLois Curfman McInnes fflush(fd); 7033a40ed3dSBarry Smith PetscFunctionReturn(0); 704289bc588SBarry Smith } 705289bc588SBarry Smith 7065615d1e5SSatish Balay #undef __FUNC__ 7075615d1e5SSatish Balay #define __FUNC__ "MatView_SeqDense_Binary" 708932b0c3eSLois Curfman McInnes static int MatView_SeqDense_Binary(Mat A,Viewer viewer) 709932b0c3eSLois Curfman McInnes { 710932b0c3eSLois Curfman McInnes Mat_SeqDense *a = (Mat_SeqDense *) A->data; 711932b0c3eSLois Curfman McInnes int ict, j, n = a->n, m = a->m, i, fd, *col_lens, ierr, nz = m*n; 71290ace30eSBarry Smith int format; 71390ace30eSBarry Smith Scalar *v, *anonz,*vals; 714932b0c3eSLois Curfman McInnes 7153a40ed3dSBarry Smith PetscFunctionBegin; 71690ace30eSBarry Smith ierr = ViewerBinaryGetDescriptor(viewer,&fd); CHKERRQ(ierr); 71790ace30eSBarry Smith 71890ace30eSBarry Smith ierr = ViewerGetFormat(viewer,&format); CHKERRQ(ierr); 71902cad45dSBarry Smith if (format == VIEWER_FORMAT_BINARY_NATIVE) { 72090ace30eSBarry Smith /* store the matrix as a dense matrix */ 72190ace30eSBarry Smith col_lens = (int *) PetscMalloc( 4*sizeof(int) ); CHKPTRQ(col_lens); 72290ace30eSBarry Smith col_lens[0] = MAT_COOKIE; 72390ace30eSBarry Smith col_lens[1] = m; 72490ace30eSBarry Smith col_lens[2] = n; 72590ace30eSBarry Smith col_lens[3] = MATRIX_BINARY_FORMAT_DENSE; 7260752156aSBarry Smith ierr = PetscBinaryWrite(fd,col_lens,4,PETSC_INT,1); CHKERRQ(ierr); 72790ace30eSBarry Smith PetscFree(col_lens); 72890ace30eSBarry Smith 72990ace30eSBarry Smith /* write out matrix, by rows */ 73045d48df9SBarry Smith vals = (Scalar *) PetscMalloc((m*n+1)*sizeof(Scalar)); CHKPTRQ(vals); 73190ace30eSBarry Smith v = a->v; 73290ace30eSBarry Smith for ( i=0; i<m; i++ ) { 73390ace30eSBarry Smith for ( j=0; j<n; j++ ) { 73490ace30eSBarry Smith vals[i + j*m] = *v++; 73590ace30eSBarry Smith } 73690ace30eSBarry Smith } 7370752156aSBarry Smith ierr = PetscBinaryWrite(fd,vals,n*m,PETSC_SCALAR,0); CHKERRQ(ierr); 73890ace30eSBarry Smith PetscFree(vals); 73990ace30eSBarry Smith } else { 7400452661fSBarry Smith col_lens = (int *) PetscMalloc( (4+nz)*sizeof(int) ); CHKPTRQ(col_lens); 741932b0c3eSLois Curfman McInnes col_lens[0] = MAT_COOKIE; 742932b0c3eSLois Curfman McInnes col_lens[1] = m; 743932b0c3eSLois Curfman McInnes col_lens[2] = n; 744932b0c3eSLois Curfman McInnes col_lens[3] = nz; 745932b0c3eSLois Curfman McInnes 746932b0c3eSLois Curfman McInnes /* store lengths of each row and write (including header) to file */ 747932b0c3eSLois Curfman McInnes for ( i=0; i<m; i++ ) col_lens[4+i] = n; 7480752156aSBarry Smith ierr = PetscBinaryWrite(fd,col_lens,4+m,PETSC_INT,1); CHKERRQ(ierr); 749932b0c3eSLois Curfman McInnes 750932b0c3eSLois Curfman McInnes /* Possibly should write in smaller increments, not whole matrix at once? */ 751932b0c3eSLois Curfman McInnes /* store column indices (zero start index) */ 752932b0c3eSLois Curfman McInnes ict = 0; 753932b0c3eSLois Curfman McInnes for ( i=0; i<m; i++ ) { 754932b0c3eSLois Curfman McInnes for ( j=0; j<n; j++ ) col_lens[ict++] = j; 755932b0c3eSLois Curfman McInnes } 7560752156aSBarry Smith ierr = PetscBinaryWrite(fd,col_lens,nz,PETSC_INT,0); CHKERRQ(ierr); 7570452661fSBarry Smith PetscFree(col_lens); 758932b0c3eSLois Curfman McInnes 759932b0c3eSLois Curfman McInnes /* store nonzero values */ 76045d48df9SBarry Smith anonz = (Scalar *) PetscMalloc((nz+1)*sizeof(Scalar)); CHKPTRQ(anonz); 761932b0c3eSLois Curfman McInnes ict = 0; 762932b0c3eSLois Curfman McInnes for ( i=0; i<m; i++ ) { 763932b0c3eSLois Curfman McInnes v = a->v + i; 764932b0c3eSLois Curfman McInnes for ( j=0; j<n; j++ ) { 765932b0c3eSLois Curfman McInnes anonz[ict++] = *v; v += a->m; 766932b0c3eSLois Curfman McInnes } 767932b0c3eSLois Curfman McInnes } 7680752156aSBarry Smith ierr = PetscBinaryWrite(fd,anonz,nz,PETSC_SCALAR,0); CHKERRQ(ierr); 7690452661fSBarry Smith PetscFree(anonz); 77090ace30eSBarry Smith } 7713a40ed3dSBarry Smith PetscFunctionReturn(0); 772932b0c3eSLois Curfman McInnes } 773932b0c3eSLois Curfman McInnes 7745615d1e5SSatish Balay #undef __FUNC__ 7755615d1e5SSatish Balay #define __FUNC__ "MatView_SeqDense" 776c6e7dd08SBarry Smith int MatView_SeqDense(Mat A,Viewer viewer) 777932b0c3eSLois Curfman McInnes { 778932b0c3eSLois Curfman McInnes Mat_SeqDense *a = (Mat_SeqDense*) A->data; 779bcd2baecSBarry Smith ViewerType vtype; 780bcd2baecSBarry Smith int ierr; 781932b0c3eSLois Curfman McInnes 7823a40ed3dSBarry Smith PetscFunctionBegin; 783bcd2baecSBarry Smith ierr = ViewerGetType(viewer,&vtype); CHKERRQ(ierr); 784bcd2baecSBarry Smith 785bcd2baecSBarry Smith if (vtype == MATLAB_VIEWER) { 7861a0f753aSSatish Balay ierr = ViewerMatlabPutScalar_Private(viewer,a->m,a->n,a->v); CHKERRQ(ierr); 7873a40ed3dSBarry Smith } else if (vtype == ASCII_FILE_VIEWER || vtype == ASCII_FILES_VIEWER) { 7883a40ed3dSBarry Smith ierr = MatView_SeqDense_ASCII(A,viewer);CHKERRQ(ierr); 7893a40ed3dSBarry Smith } else if (vtype == BINARY_FILE_VIEWER) { 7903a40ed3dSBarry Smith ierr = MatView_SeqDense_Binary(A,viewer);CHKERRQ(ierr); 7915cd90555SBarry Smith } else { 7925cd90555SBarry Smith SETERRQ(1,1,"Viewer type not supported by PETSc object"); 793932b0c3eSLois Curfman McInnes } 7943a40ed3dSBarry Smith PetscFunctionReturn(0); 795932b0c3eSLois Curfman McInnes } 796289bc588SBarry Smith 7975615d1e5SSatish Balay #undef __FUNC__ 7985615d1e5SSatish Balay #define __FUNC__ "MatDestroy_SeqDense" 799c6e7dd08SBarry Smith int MatDestroy_SeqDense(Mat mat) 800289bc588SBarry Smith { 801ec8511deSBarry Smith Mat_SeqDense *l = (Mat_SeqDense *) mat->data; 80290f02eecSBarry Smith int ierr; 80390f02eecSBarry Smith 8043a40ed3dSBarry Smith PetscFunctionBegin; 80594d884c6SBarry Smith if (--mat->refct > 0) PetscFunctionReturn(0); 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 } 8133a40ed3dSBarry Smith #if defined(USE_PETSC_LOG) 814c6e7dd08SBarry Smith PLogObjectState((PetscObject)mat,"Rows %d Cols %d",l->m,l->n); 815a5a9c739SBarry Smith #endif 8160452661fSBarry Smith if (l->pivots) PetscFree(l->pivots); 8173439631bSBarry Smith if (!l->user_alloc) PetscFree(l->v); 8180452661fSBarry Smith PetscFree(l); 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; 835d3e5ee88SLois Curfman McInnes int k, j, m, n; 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 } 848d64ed03dSBarry Smith PetscMemcpy(v,w,m*n*sizeof(Scalar)); 849d64ed03dSBarry Smith PetscFree(w); 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 int ierr; 861d3e5ee88SLois Curfman McInnes Mat tmat; 862ec8511deSBarry Smith Mat_SeqDense *tmatd; 863d3e5ee88SLois Curfman McInnes Scalar *v2; 864b4fd4287SBarry Smith ierr = MatCreateSeqDense(A->comm,mat->n,mat->m,PETSC_NULL,&tmat); CHKERRQ(ierr); 865ec8511deSBarry Smith tmatd = (Mat_SeqDense *) tmat->data; 8660de55854SLois Curfman McInnes v = mat->v; v2 = tmatd->v; 867d3e5ee88SLois Curfman McInnes for ( j=0; j<n; j++ ) { 8680de55854SLois Curfman McInnes for ( k=0; k<m; k++ ) v2[j + k*n] = v[k + j*m]; 869d3e5ee88SLois Curfman McInnes } 8706d4a8577SBarry Smith ierr = MatAssemblyBegin(tmat,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 8716d4a8577SBarry Smith ierr = MatAssemblyEnd(tmat,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 872d3e5ee88SLois Curfman McInnes *matout = tmat; 87348b35521SBarry Smith } 8743a40ed3dSBarry Smith PetscFunctionReturn(0); 875289bc588SBarry Smith } 876289bc588SBarry Smith 8775615d1e5SSatish Balay #undef __FUNC__ 8785615d1e5SSatish Balay #define __FUNC__ "MatEqual_SeqDense" 8798f6be9afSLois Curfman McInnes int MatEqual_SeqDense(Mat A1,Mat A2, PetscTruth *flg) 880289bc588SBarry Smith { 881c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat1 = (Mat_SeqDense *) A1->data; 882c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat2 = (Mat_SeqDense *) A2->data; 883289bc588SBarry Smith int i; 884289bc588SBarry Smith Scalar *v1 = mat1->v, *v2 = mat2->v; 8859ea5d5aeSSatish Balay 8863a40ed3dSBarry Smith PetscFunctionBegin; 8873a40ed3dSBarry Smith if (A2->type != MATSEQDENSE) SETERRQ(PETSC_ERR_SUP,0,"Matrices should be of type MATSEQDENSE"); 8883a40ed3dSBarry Smith if (mat1->m != mat2->m) {*flg = PETSC_FALSE; PetscFunctionReturn(0);} 8893a40ed3dSBarry Smith if (mat1->n != mat2->n) {*flg = PETSC_FALSE; PetscFunctionReturn(0);} 890289bc588SBarry Smith for ( i=0; i<mat1->m*mat1->n; i++ ) { 8913a40ed3dSBarry Smith if (*v1 != *v2) {*flg = PETSC_FALSE; PetscFunctionReturn(0);} 892289bc588SBarry Smith v1++; v2++; 893289bc588SBarry Smith } 89477c4ece6SBarry Smith *flg = PETSC_TRUE; 8953a40ed3dSBarry Smith PetscFunctionReturn(0); 896289bc588SBarry Smith } 897289bc588SBarry Smith 8985615d1e5SSatish Balay #undef __FUNC__ 8995615d1e5SSatish Balay #define __FUNC__ "MatGetDiagonal_SeqDense" 9008f6be9afSLois Curfman McInnes int MatGetDiagonal_SeqDense(Mat A,Vec v) 901289bc588SBarry Smith { 902c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 9037a97a34bSBarry Smith int ierr,i, n, len; 90444cd7ae7SLois Curfman McInnes Scalar *x, zero = 0.0; 90544cd7ae7SLois Curfman McInnes 9063a40ed3dSBarry Smith PetscFunctionBegin; 9077a97a34bSBarry Smith ierr = VecSet(&zero,v);CHKERRQ(ierr); 9087a97a34bSBarry Smith ierr = VecGetArray(v,&x); CHKERRQ(ierr); 9097a97a34bSBarry Smith ierr = VecGetSize(v,&n);CHKERRQ(ierr); 91044cd7ae7SLois Curfman McInnes len = PetscMin(mat->m,mat->n); 911e3372554SBarry Smith if (n != mat->m) SETERRQ(PETSC_ERR_ARG_SIZ,0,"Nonconforming mat and vec"); 91244cd7ae7SLois Curfman McInnes for ( i=0; i<len; i++ ) { 913289bc588SBarry Smith x[i] = mat->v[i*mat->m + i]; 914289bc588SBarry Smith } 9157a97a34bSBarry Smith ierr = VecRestoreArray(v,&x); CHKERRQ(ierr); 9163a40ed3dSBarry Smith PetscFunctionReturn(0); 917289bc588SBarry Smith } 918289bc588SBarry Smith 9195615d1e5SSatish Balay #undef __FUNC__ 9205615d1e5SSatish Balay #define __FUNC__ "MatDiagonalScale_SeqDense" 9218f6be9afSLois Curfman McInnes int MatDiagonalScale_SeqDense(Mat A,Vec ll,Vec rr) 922289bc588SBarry Smith { 923c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 924da3a660dSBarry Smith Scalar *l,*r,x,*v; 9257a97a34bSBarry Smith int ierr,i,j,m = mat->m, n = mat->n; 92655659b69SBarry Smith 9273a40ed3dSBarry Smith PetscFunctionBegin; 92828988994SBarry Smith if (ll) { 9297a97a34bSBarry Smith ierr = VecGetArray(ll,&l);CHKERRQ(ierr); 9307a97a34bSBarry Smith ierr = VecGetSize(ll,&m);CHKERRQ(ierr); 931e3372554SBarry Smith if (m != mat->m) SETERRQ(PETSC_ERR_ARG_SIZ,0,"Left scaling vec wrong size"); 932da3a660dSBarry Smith for ( i=0; i<m; i++ ) { 933da3a660dSBarry Smith x = l[i]; 934da3a660dSBarry Smith v = mat->v + i; 935da3a660dSBarry Smith for ( j=0; j<n; j++ ) { (*v) *= x; v+= m;} 936da3a660dSBarry Smith } 9377a97a34bSBarry Smith ierr = VecRestoreArray(ll,&l);CHKERRQ(ierr); 93844cd7ae7SLois Curfman McInnes PLogFlops(n*m); 939da3a660dSBarry Smith } 94028988994SBarry Smith if (rr) { 9417a97a34bSBarry Smith ierr = VecGetArray(rr,&r);CHKERRQ(ierr); 9427a97a34bSBarry Smith ierr = VecGetSize(rr,&n);CHKERRQ(ierr); 943e3372554SBarry Smith if (n != mat->n) SETERRQ(PETSC_ERR_ARG_SIZ,0,"Right scaling vec wrong size"); 944da3a660dSBarry Smith for ( i=0; i<n; i++ ) { 945da3a660dSBarry Smith x = r[i]; 946da3a660dSBarry Smith v = mat->v + i*m; 947da3a660dSBarry Smith for ( j=0; j<m; j++ ) { (*v++) *= x;} 948da3a660dSBarry Smith } 9497a97a34bSBarry Smith ierr = VecRestoreArray(rr,&r);CHKERRQ(ierr); 95044cd7ae7SLois Curfman McInnes PLogFlops(n*m); 951da3a660dSBarry Smith } 9523a40ed3dSBarry Smith PetscFunctionReturn(0); 953289bc588SBarry Smith } 954289bc588SBarry Smith 9555615d1e5SSatish Balay #undef __FUNC__ 9565615d1e5SSatish Balay #define __FUNC__ "MatNorm_SeqDense" 9578f6be9afSLois Curfman McInnes int MatNorm_SeqDense(Mat A,NormType type,double *norm) 958289bc588SBarry Smith { 959c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 960289bc588SBarry Smith Scalar *v = mat->v; 961289bc588SBarry Smith double sum = 0.0; 962289bc588SBarry Smith int i, j; 96355659b69SBarry Smith 9643a40ed3dSBarry Smith PetscFunctionBegin; 965289bc588SBarry Smith if (type == NORM_FROBENIUS) { 966289bc588SBarry Smith for (i=0; i<mat->n*mat->m; i++ ) { 9673a40ed3dSBarry Smith #if defined(USE_PETSC_COMPLEX) 9683f6de6efSSatish Balay sum += PetscReal(PetscConj(*v)*(*v)); v++; 969289bc588SBarry Smith #else 970289bc588SBarry Smith sum += (*v)*(*v); v++; 971289bc588SBarry Smith #endif 972289bc588SBarry Smith } 973289bc588SBarry Smith *norm = sqrt(sum); 97455659b69SBarry Smith PLogFlops(2*mat->n*mat->m); 9753a40ed3dSBarry Smith } else if (type == NORM_1) { 976289bc588SBarry Smith *norm = 0.0; 977289bc588SBarry Smith for ( j=0; j<mat->n; j++ ) { 978289bc588SBarry Smith sum = 0.0; 979289bc588SBarry Smith for ( i=0; i<mat->m; i++ ) { 98033a8263dSBarry Smith sum += PetscAbsScalar(*v); v++; 981289bc588SBarry Smith } 982289bc588SBarry Smith if (sum > *norm) *norm = sum; 983289bc588SBarry Smith } 98455659b69SBarry Smith PLogFlops(mat->n*mat->m); 9853a40ed3dSBarry Smith } else if (type == NORM_INFINITY) { 986289bc588SBarry Smith *norm = 0.0; 987289bc588SBarry Smith for ( j=0; j<mat->m; j++ ) { 988289bc588SBarry Smith v = mat->v + j; 989289bc588SBarry Smith sum = 0.0; 990289bc588SBarry Smith for ( i=0; i<mat->n; i++ ) { 991cddf8d76SBarry Smith sum += PetscAbsScalar(*v); v += mat->m; 992289bc588SBarry Smith } 993289bc588SBarry Smith if (sum > *norm) *norm = sum; 994289bc588SBarry Smith } 99555659b69SBarry Smith PLogFlops(mat->n*mat->m); 9963a40ed3dSBarry Smith } else { 997e3372554SBarry Smith SETERRQ(PETSC_ERR_SUP,0,"No two norm"); 998289bc588SBarry Smith } 9993a40ed3dSBarry Smith PetscFunctionReturn(0); 1000289bc588SBarry Smith } 1001289bc588SBarry Smith 10025615d1e5SSatish Balay #undef __FUNC__ 10035615d1e5SSatish Balay #define __FUNC__ "MatSetOption_SeqDense" 10048f6be9afSLois Curfman McInnes int MatSetOption_SeqDense(Mat A,MatOption op) 1005289bc588SBarry Smith { 1006c0bbcb79SLois Curfman McInnes Mat_SeqDense *aij = (Mat_SeqDense *) A->data; 100767e560aaSBarry Smith 10083a40ed3dSBarry Smith PetscFunctionBegin; 10096d4a8577SBarry Smith if (op == MAT_ROW_ORIENTED) aij->roworiented = 1; 10106d4a8577SBarry Smith else if (op == MAT_COLUMN_ORIENTED) aij->roworiented = 0; 10116d4a8577SBarry Smith else if (op == MAT_ROWS_SORTED || 1012219d9a1aSLois Curfman McInnes op == MAT_ROWS_UNSORTED || 10136d4a8577SBarry Smith op == MAT_COLUMNS_SORTED || 1014219d9a1aSLois Curfman McInnes op == MAT_COLUMNS_UNSORTED || 10156d4a8577SBarry Smith op == MAT_SYMMETRIC || 10166d4a8577SBarry Smith op == MAT_STRUCTURALLY_SYMMETRIC || 10176d4a8577SBarry Smith op == MAT_NO_NEW_NONZERO_LOCATIONS || 10186d4a8577SBarry Smith op == MAT_YES_NEW_NONZERO_LOCATIONS || 1019c2653b3dSLois Curfman McInnes op == MAT_NEW_NONZERO_LOCATION_ERROR || 10206d4a8577SBarry Smith op == MAT_NO_NEW_DIAGONALS || 102190f02eecSBarry Smith op == MAT_YES_NEW_DIAGONALS || 1022b51ba29fSSatish Balay op == MAT_IGNORE_OFF_PROC_ENTRIES || 1023b51ba29fSSatish Balay op == MAT_USE_HASH_TABLE) 1024981c4779SBarry Smith PLogInfo(A,"MatSetOption_SeqDense:Option ignored\n"); 10253a40ed3dSBarry Smith else { 10263a40ed3dSBarry Smith SETERRQ(PETSC_ERR_SUP,0,"unknown option"); 10273a40ed3dSBarry Smith } 10283a40ed3dSBarry Smith PetscFunctionReturn(0); 1029289bc588SBarry Smith } 1030289bc588SBarry Smith 10315615d1e5SSatish Balay #undef __FUNC__ 10325615d1e5SSatish Balay #define __FUNC__ "MatZeroEntries_SeqDense" 10338f6be9afSLois Curfman McInnes int MatZeroEntries_SeqDense(Mat A) 10346f0a148fSBarry Smith { 1035ec8511deSBarry Smith Mat_SeqDense *l = (Mat_SeqDense *) A->data; 10363a40ed3dSBarry Smith 10373a40ed3dSBarry Smith PetscFunctionBegin; 1038cddf8d76SBarry Smith PetscMemzero(l->v,l->m*l->n*sizeof(Scalar)); 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; 11143a40ed3dSBarry Smith PetscFunctionReturn(0); 1115ff14e315SSatish Balay } 11160754003eSLois Curfman McInnes 11175615d1e5SSatish Balay #undef __FUNC__ 11185615d1e5SSatish Balay #define __FUNC__ "MatGetSubMatrix_SeqDense" 1119182d2002SSatish Balay static int MatGetSubMatrix_SeqDense(Mat A,IS isrow,IS iscol,int cs,MatGetSubMatrixCall scall,Mat *B) 11200754003eSLois Curfman McInnes { 1121c0bbcb79SLois Curfman McInnes Mat_SeqDense *mat = (Mat_SeqDense *) A->data; 1122182d2002SSatish Balay int i, j, ierr, m = mat->m, *irow, *icol, nrows, ncols; 1123182d2002SSatish Balay Scalar *av, *bv, *v = mat->v; 11240754003eSLois Curfman McInnes Mat newmat; 11250754003eSLois Curfman McInnes 11263a40ed3dSBarry Smith PetscFunctionBegin; 112778b31e54SBarry Smith ierr = ISGetIndices(isrow,&irow); CHKERRQ(ierr); 112878b31e54SBarry Smith ierr = ISGetIndices(iscol,&icol); CHKERRQ(ierr); 112978b31e54SBarry Smith ierr = ISGetSize(isrow,&nrows); CHKERRQ(ierr); 113078b31e54SBarry Smith ierr = ISGetSize(iscol,&ncols); CHKERRQ(ierr); 11310754003eSLois Curfman McInnes 1132182d2002SSatish Balay /* Check submatrixcall */ 1133182d2002SSatish Balay if (scall == MAT_REUSE_MATRIX) { 1134182d2002SSatish Balay int n_cols,n_rows; 1135182d2002SSatish Balay ierr = MatGetSize(*B,&n_rows,&n_cols); CHKERRQ(ierr); 1136182d2002SSatish Balay if (n_rows != nrows || n_cols != ncols) SETERRQ(PETSC_ERR_ARG_SIZ,0,"Reused submatrix wrong size"); 1137182d2002SSatish Balay newmat = *B; 1138182d2002SSatish Balay } else { 11390754003eSLois Curfman McInnes /* Create and fill new matrix */ 1140b4fd4287SBarry Smith ierr = MatCreateSeqDense(A->comm,nrows,ncols,PETSC_NULL,&newmat); CHKERRQ(ierr); 1141182d2002SSatish Balay } 1142182d2002SSatish Balay 1143182d2002SSatish Balay /* Now extract the data pointers and do the copy, column at a time */ 1144182d2002SSatish Balay bv = ((Mat_SeqDense*)newmat->data)->v; 1145182d2002SSatish Balay 1146182d2002SSatish Balay for ( i=0; i<ncols; i++ ) { 1147182d2002SSatish Balay av = v + m*icol[i]; 1148182d2002SSatish Balay for (j=0; j<nrows; j++ ) { 1149182d2002SSatish Balay *bv++ = av[irow[j]]; 11500754003eSLois Curfman McInnes } 11510754003eSLois Curfman McInnes } 1152182d2002SSatish Balay 1153182d2002SSatish Balay /* Assemble the matrices so that the correct flags are set */ 11546d4a8577SBarry Smith ierr = MatAssemblyBegin(newmat,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 11556d4a8577SBarry Smith ierr = MatAssemblyEnd(newmat,MAT_FINAL_ASSEMBLY); CHKERRQ(ierr); 11560754003eSLois Curfman McInnes 11570754003eSLois Curfman McInnes /* Free work space */ 115878b31e54SBarry Smith ierr = ISRestoreIndices(isrow,&irow); CHKERRQ(ierr); 115978b31e54SBarry Smith ierr = ISRestoreIndices(iscol,&icol); CHKERRQ(ierr); 1160182d2002SSatish Balay *B = newmat; 11613a40ed3dSBarry Smith PetscFunctionReturn(0); 11620754003eSLois Curfman McInnes } 11630754003eSLois Curfman McInnes 11645615d1e5SSatish Balay #undef __FUNC__ 11655615d1e5SSatish Balay #define __FUNC__ "MatGetSubMatrices_SeqDense" 11666a6a5d1dSBarry Smith int MatGetSubMatrices_SeqDense(Mat A,int n, IS *irow,IS *icol,MatGetSubMatrixCall scall,Mat **B) 1167905e6a2fSBarry Smith { 1168905e6a2fSBarry Smith int ierr,i; 1169905e6a2fSBarry Smith 11703a40ed3dSBarry Smith PetscFunctionBegin; 1171905e6a2fSBarry Smith if (scall == MAT_INITIAL_MATRIX) { 1172905e6a2fSBarry Smith *B = (Mat *) PetscMalloc( (n+1)*sizeof(Mat) ); CHKPTRQ(*B); 1173905e6a2fSBarry Smith } 1174905e6a2fSBarry Smith 1175905e6a2fSBarry Smith for ( i=0; i<n; i++ ) { 11766a6a5d1dSBarry Smith ierr = MatGetSubMatrix_SeqDense(A,irow[i],icol[i],PETSC_DECIDE,scall,&(*B)[i]);CHKERRQ(ierr); 1177905e6a2fSBarry Smith } 11783a40ed3dSBarry Smith PetscFunctionReturn(0); 1179905e6a2fSBarry Smith } 1180905e6a2fSBarry Smith 11815615d1e5SSatish Balay #undef __FUNC__ 11825615d1e5SSatish Balay #define __FUNC__ "MatCopy_SeqDense" 1183cb5b572fSBarry Smith int MatCopy_SeqDense(Mat A, Mat B,MatStructure str) 11844b0e389bSBarry Smith { 11854b0e389bSBarry Smith Mat_SeqDense *a = (Mat_SeqDense *) A->data, *b = (Mat_SeqDense *)B->data; 11863a40ed3dSBarry Smith int ierr; 11873a40ed3dSBarry Smith 11883a40ed3dSBarry Smith PetscFunctionBegin; 11893a40ed3dSBarry Smith if (B->type != MATSEQDENSE) { 1190cb5b572fSBarry Smith ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr); 11913a40ed3dSBarry Smith PetscFunctionReturn(0); 11923a40ed3dSBarry Smith } 1193e3372554SBarry Smith if (a->m != b->m || a->n != b->n) SETERRQ(PETSC_ERR_ARG_SIZ,0,"size(B) != size(A)"); 11944b0e389bSBarry Smith PetscMemcpy(b->v,a->v,a->m*a->n*sizeof(Scalar)); 11953a40ed3dSBarry Smith PetscFunctionReturn(0); 11964b0e389bSBarry Smith } 11974b0e389bSBarry Smith 1198289bc588SBarry Smith /* -------------------------------------------------------------------*/ 1199a5ae1ecdSBarry Smith static struct _MatOps MatOps_Values = {MatSetValues_SeqDense, 1200905e6a2fSBarry Smith MatGetRow_SeqDense, 1201905e6a2fSBarry Smith MatRestoreRow_SeqDense, 1202905e6a2fSBarry Smith MatMult_SeqDense, 1203905e6a2fSBarry Smith MatMultAdd_SeqDense, 1204905e6a2fSBarry Smith MatMultTrans_SeqDense, 1205905e6a2fSBarry Smith MatMultTransAdd_SeqDense, 1206905e6a2fSBarry Smith MatSolve_SeqDense, 1207905e6a2fSBarry Smith MatSolveAdd_SeqDense, 1208905e6a2fSBarry Smith MatSolveTrans_SeqDense, 1209905e6a2fSBarry Smith MatSolveTransAdd_SeqDense, 1210905e6a2fSBarry Smith MatLUFactor_SeqDense, 1211905e6a2fSBarry Smith MatCholeskyFactor_SeqDense, 1212ec8511deSBarry Smith MatRelax_SeqDense, 1213ec8511deSBarry Smith MatTranspose_SeqDense, 1214905e6a2fSBarry Smith MatGetInfo_SeqDense, 1215905e6a2fSBarry Smith MatEqual_SeqDense, 1216905e6a2fSBarry Smith MatGetDiagonal_SeqDense, 1217905e6a2fSBarry Smith MatDiagonalScale_SeqDense, 1218905e6a2fSBarry Smith MatNorm_SeqDense, 1219905e6a2fSBarry Smith 0, 1220905e6a2fSBarry Smith 0, 1221905e6a2fSBarry Smith 0, 1222905e6a2fSBarry Smith MatSetOption_SeqDense, 1223905e6a2fSBarry Smith MatZeroEntries_SeqDense, 1224905e6a2fSBarry Smith MatZeroRows_SeqDense, 1225905e6a2fSBarry Smith MatLUFactorSymbolic_SeqDense, 1226905e6a2fSBarry Smith MatLUFactorNumeric_SeqDense, 1227905e6a2fSBarry Smith MatCholeskyFactorSymbolic_SeqDense, 1228905e6a2fSBarry Smith MatCholeskyFactorNumeric_SeqDense, 1229905e6a2fSBarry Smith MatGetSize_SeqDense, 1230905e6a2fSBarry Smith MatGetSize_SeqDense, 1231905e6a2fSBarry Smith MatGetOwnershipRange_SeqDense, 1232905e6a2fSBarry Smith 0, 1233905e6a2fSBarry Smith 0, 1234905e6a2fSBarry Smith MatGetArray_SeqDense, 1235905e6a2fSBarry Smith MatRestoreArray_SeqDense, 12365609ef8eSBarry Smith MatDuplicate_SeqDense, 1237a5ae1ecdSBarry Smith 0, 1238a5ae1ecdSBarry Smith 0, 1239a5ae1ecdSBarry Smith 0, 1240a5ae1ecdSBarry Smith 0, 1241a5ae1ecdSBarry Smith MatAXPY_SeqDense, 1242a5ae1ecdSBarry Smith MatGetSubMatrices_SeqDense, 1243a5ae1ecdSBarry Smith 0, 12444b0e389bSBarry Smith MatGetValues_SeqDense, 1245a5ae1ecdSBarry Smith MatCopy_SeqDense, 1246a5ae1ecdSBarry Smith 0, 1247a5ae1ecdSBarry Smith MatScale_SeqDense, 1248a5ae1ecdSBarry Smith 0, 1249a5ae1ecdSBarry Smith 0, 1250a5ae1ecdSBarry Smith 0, 1251a5ae1ecdSBarry Smith MatGetBlockSize_SeqDense, 1252a5ae1ecdSBarry Smith 0, 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 MatGetMaps_Petsc}; 126590ace30eSBarry Smith 12665615d1e5SSatish Balay #undef __FUNC__ 12675615d1e5SSatish Balay #define __FUNC__ "MatCreateSeqDense" 12684b828684SBarry Smith /*@C 1269fafbff53SBarry Smith MatCreateSeqDense - Creates a sequential dense matrix that 1270d65003e9SLois Curfman McInnes is stored in column major order (the usual Fortran 77 manner). Many 1271d65003e9SLois Curfman McInnes of the matrix operations use the BLAS and LAPACK routines. 1272289bc588SBarry Smith 1273db81eaa0SLois Curfman McInnes Collective on MPI_Comm 1274db81eaa0SLois Curfman McInnes 127520563c6bSBarry Smith Input Parameters: 1276db81eaa0SLois Curfman McInnes + comm - MPI communicator, set to PETSC_COMM_SELF 12770c775827SLois Curfman McInnes . m - number of rows 127818f449edSLois Curfman McInnes . n - number of columns 1279db81eaa0SLois Curfman McInnes - data - optional location of matrix data. Set data=PETSC_NULL for PETSc 1280dfc5480cSLois Curfman McInnes to control all matrix memory allocation. 128120563c6bSBarry Smith 128220563c6bSBarry Smith Output Parameter: 128344cd7ae7SLois Curfman McInnes . A - the matrix 128420563c6bSBarry Smith 1285b259b22eSLois Curfman McInnes Notes: 128618f449edSLois Curfman McInnes The data input variable is intended primarily for Fortran programmers 128718f449edSLois Curfman McInnes who wish to allocate their own matrix memory space. Most users should 1288b4fd4287SBarry Smith set data=PETSC_NULL. 128918f449edSLois Curfman McInnes 1290dbd7a890SLois Curfman McInnes .keywords: dense, matrix, LAPACK, BLAS 1291d65003e9SLois Curfman McInnes 1292db81eaa0SLois Curfman McInnes .seealso: MatCreate(), MatCreateMPIDense(), MatSetValues() 129320563c6bSBarry Smith @*/ 129444cd7ae7SLois Curfman McInnes int MatCreateSeqDense(MPI_Comm comm,int m,int n,Scalar *data,Mat *A) 1295289bc588SBarry Smith { 129644cd7ae7SLois Curfman McInnes Mat B; 129744cd7ae7SLois Curfman McInnes Mat_SeqDense *b; 12983b2fbd54SBarry Smith int ierr,flg,size; 12993b2fbd54SBarry Smith 13003a40ed3dSBarry Smith PetscFunctionBegin; 13013b2fbd54SBarry Smith MPI_Comm_size(comm,&size); 1302a8c6a408SBarry Smith if (size > 1) SETERRQ(PETSC_ERR_ARG_WRONG,0,"Comm must be of size 1"); 130355659b69SBarry Smith 130444cd7ae7SLois Curfman McInnes *A = 0; 1305f830108cSBarry Smith PetscHeaderCreate(B,_p_Mat,struct _MatOps,MAT_COOKIE,MATSEQDENSE,comm,MatDestroy,MatView); 130644cd7ae7SLois Curfman McInnes PLogObjectCreate(B); 130744cd7ae7SLois Curfman McInnes b = (Mat_SeqDense *) PetscMalloc(sizeof(Mat_SeqDense)); CHKPTRQ(b); 130844cd7ae7SLois Curfman McInnes PetscMemzero(b,sizeof(Mat_SeqDense)); 1309a5ae1ecdSBarry Smith PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps)); 1310c6e7dd08SBarry Smith B->ops->destroy = MatDestroy_SeqDense; 1311c6e7dd08SBarry Smith B->ops->view = MatView_SeqDense; 131244cd7ae7SLois Curfman McInnes B->factor = 0; 131390f02eecSBarry Smith B->mapping = 0; 1314f09e8eb9SSatish Balay PLogObjectMemory(B,sizeof(struct _p_Mat)); 131544cd7ae7SLois Curfman McInnes B->data = (void *) b; 131618f449edSLois Curfman McInnes 131744cd7ae7SLois Curfman McInnes b->m = m; B->m = m; B->M = m; 131844cd7ae7SLois Curfman McInnes b->n = n; B->n = n; B->N = n; 1319a5ae1ecdSBarry Smith 1320488ecbafSBarry Smith ierr = MapCreateMPI(comm,m,m,&B->rmap);CHKERRQ(ierr); 1321488ecbafSBarry Smith ierr = MapCreateMPI(comm,n,n,&B->cmap);CHKERRQ(ierr); 1322a5ae1ecdSBarry Smith 132344cd7ae7SLois Curfman McInnes b->pivots = 0; 132444cd7ae7SLois Curfman McInnes b->roworiented = 1; 1325b4fd4287SBarry Smith if (data == PETSC_NULL) { 132644cd7ae7SLois Curfman McInnes b->v = (Scalar*) PetscMalloc((m*n+1)*sizeof(Scalar)); CHKPTRQ(b->v); 132744cd7ae7SLois Curfman McInnes PetscMemzero(b->v,m*n*sizeof(Scalar)); 132844cd7ae7SLois Curfman McInnes b->user_alloc = 0; 132944cd7ae7SLois Curfman McInnes PLogObjectMemory(B,n*m*sizeof(Scalar)); 1330273fdc76SBarry Smith } else { /* user-allocated storage */ 133144cd7ae7SLois Curfman McInnes b->v = data; 133244cd7ae7SLois Curfman McInnes b->user_alloc = 1; 13332dd2b441SLois Curfman McInnes } 133425cdf11fSBarry Smith ierr = OptionsHasName(PETSC_NULL,"-help",&flg); CHKERRQ(ierr); 133544cd7ae7SLois Curfman McInnes if (flg) {ierr = MatPrintHelp(B); CHKERRQ(ierr);} 13364e220ebcSLois Curfman McInnes 133744cd7ae7SLois Curfman McInnes *A = B; 13383a40ed3dSBarry Smith PetscFunctionReturn(0); 1339289bc588SBarry Smith } 1340289bc588SBarry Smith 13413b2fbd54SBarry Smith 13423b2fbd54SBarry Smith 13433b2fbd54SBarry Smith 13443b2fbd54SBarry Smith 13453b2fbd54SBarry Smith 1346