xref: /petsc/src/mat/impls/dense/seq/dense.c (revision 170fe5c81eeebf19534f729defaf7de405b3be81)
1be1d678aSKris Buschelman #define PETSCMAT_DLL
2be1d678aSKris Buschelman 
367e560aaSBarry Smith /*
467e560aaSBarry Smith      Defines the basic matrix operations for sequential dense.
567e560aaSBarry Smith */
6289bc588SBarry Smith 
770f55243SBarry Smith #include "src/mat/impls/dense/seq/dense.h"
8b4c862e0SSatish Balay #include "petscblaslapack.h"
9289bc588SBarry Smith 
104a2ae208SSatish Balay #undef __FUNCT__
114a2ae208SSatish Balay #define __FUNCT__ "MatAXPY_SeqDense"
12f4df32b1SMatthew Knepley PetscErrorCode MatAXPY_SeqDense(Mat Y,PetscScalar alpha,Mat X,MatStructure str)
131987afe7SBarry Smith {
141987afe7SBarry Smith   Mat_SeqDense   *x = (Mat_SeqDense*)X->data,*y = (Mat_SeqDense*)Y->data;
15f4df32b1SMatthew Knepley   PetscScalar    oalpha = alpha;
1613f74950SBarry Smith   PetscInt       j;
170805154bSBarry Smith   PetscBLASInt   N,m,ldax,lday,one = 1;
18efee365bSSatish Balay   PetscErrorCode ierr;
193a40ed3dSBarry Smith 
203a40ed3dSBarry Smith   PetscFunctionBegin;
210805154bSBarry Smith   N    = PetscBLASIntCast(X->rmap.n*X->cmap.n);
220805154bSBarry Smith   m    = PetscBLASIntCast(X->rmap.n);
230805154bSBarry Smith   ldax = PetscBLASIntCast(x->lda);
240805154bSBarry Smith   lday = PetscBLASIntCast(y->lda);
25a5ce6ee0Svictorle   if (ldax>m || lday>m) {
26899cda47SBarry Smith     for (j=0; j<X->cmap.n; j++) {
27f4df32b1SMatthew Knepley       BLASaxpy_(&m,&oalpha,x->v+j*ldax,&one,y->v+j*lday,&one);
28a5ce6ee0Svictorle     }
29a5ce6ee0Svictorle   } else {
30f4df32b1SMatthew Knepley     BLASaxpy_(&N,&oalpha,x->v,&one,y->v,&one);
31a5ce6ee0Svictorle   }
32efee365bSSatish Balay   ierr = PetscLogFlops(2*N-1);CHKERRQ(ierr);
333a40ed3dSBarry Smith   PetscFunctionReturn(0);
341987afe7SBarry Smith }
351987afe7SBarry Smith 
364a2ae208SSatish Balay #undef __FUNCT__
374a2ae208SSatish Balay #define __FUNCT__ "MatGetInfo_SeqDense"
38dfbe8321SBarry Smith PetscErrorCode MatGetInfo_SeqDense(Mat A,MatInfoType flag,MatInfo *info)
39289bc588SBarry Smith {
40899cda47SBarry Smith   PetscInt     N = A->rmap.n*A->cmap.n;
413a40ed3dSBarry Smith 
423a40ed3dSBarry Smith   PetscFunctionBegin;
43899cda47SBarry Smith   info->rows_global       = (double)A->rmap.n;
44899cda47SBarry Smith   info->columns_global    = (double)A->cmap.n;
45899cda47SBarry Smith   info->rows_local        = (double)A->rmap.n;
46899cda47SBarry Smith   info->columns_local     = (double)A->cmap.n;
474e220ebcSLois Curfman McInnes   info->block_size        = 1.0;
484e220ebcSLois Curfman McInnes   info->nz_allocated      = (double)N;
496de62eeeSBarry Smith   info->nz_used           = (double)N;
506de62eeeSBarry Smith   info->nz_unneeded       = (double)0;
514e220ebcSLois Curfman McInnes   info->assemblies        = (double)A->num_ass;
524e220ebcSLois Curfman McInnes   info->mallocs           = 0;
537adad957SLisandro Dalcin   info->memory            = ((PetscObject)A)->mem;
544e220ebcSLois Curfman McInnes   info->fill_ratio_given  = 0;
554e220ebcSLois Curfman McInnes   info->fill_ratio_needed = 0;
564e220ebcSLois Curfman McInnes   info->factor_mallocs    = 0;
573a40ed3dSBarry Smith   PetscFunctionReturn(0);
58289bc588SBarry Smith }
59289bc588SBarry Smith 
604a2ae208SSatish Balay #undef __FUNCT__
614a2ae208SSatish Balay #define __FUNCT__ "MatScale_SeqDense"
62f4df32b1SMatthew Knepley PetscErrorCode MatScale_SeqDense(Mat A,PetscScalar alpha)
6380cd9d93SLois Curfman McInnes {
64273d9f13SBarry Smith   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
65f4df32b1SMatthew Knepley   PetscScalar    oalpha = alpha;
66efee365bSSatish Balay   PetscErrorCode ierr;
670805154bSBarry Smith   PetscBLASInt   one = 1,j,nz,lda = PetscBLASIntCast(a->lda);
6880cd9d93SLois Curfman McInnes 
693a40ed3dSBarry Smith   PetscFunctionBegin;
70899cda47SBarry Smith   if (lda>A->rmap.n) {
710805154bSBarry Smith     nz = PetscBLASIntCast(A->rmap.n);
72899cda47SBarry Smith     for (j=0; j<A->cmap.n; j++) {
73f4df32b1SMatthew Knepley       BLASscal_(&nz,&oalpha,a->v+j*lda,&one);
74a5ce6ee0Svictorle     }
75a5ce6ee0Svictorle   } else {
760805154bSBarry Smith     nz = PetscBLASIntCast(A->rmap.n*A->cmap.n);
77f4df32b1SMatthew Knepley     BLASscal_(&nz,&oalpha,a->v,&one);
78a5ce6ee0Svictorle   }
79efee365bSSatish Balay   ierr = PetscLogFlops(nz);CHKERRQ(ierr);
803a40ed3dSBarry Smith   PetscFunctionReturn(0);
8180cd9d93SLois Curfman McInnes }
8280cd9d93SLois Curfman McInnes 
831cbb95d3SBarry Smith #undef __FUNCT__
841cbb95d3SBarry Smith #define __FUNCT__ "MatIsHermitian_SeqDense"
851cbb95d3SBarry Smith PetscErrorCode MatIsHermitian_SeqDense(Mat A,PetscReal rtol,PetscTruth *fl)
861cbb95d3SBarry Smith {
871cbb95d3SBarry Smith   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
881cbb95d3SBarry Smith   PetscInt       i,j,m = A->rmap.n,N;
891cbb95d3SBarry Smith   PetscScalar    *v = a->v;
901cbb95d3SBarry Smith 
911cbb95d3SBarry Smith   PetscFunctionBegin;
921cbb95d3SBarry Smith   *fl = PETSC_FALSE;
931cbb95d3SBarry Smith   if (A->rmap.n != A->cmap.n) PetscFunctionReturn(0);
941cbb95d3SBarry Smith   N = a->lda;
951cbb95d3SBarry Smith 
961cbb95d3SBarry Smith   for (i=0; i<m; i++) {
971cbb95d3SBarry Smith     for (j=i+1; j<m; j++) {
981cbb95d3SBarry Smith       if (PetscAbsScalar(v[i+j*N] - PetscConj(v[j+i*N])) > rtol) PetscFunctionReturn(0);
991cbb95d3SBarry Smith     }
1001cbb95d3SBarry Smith   }
1011cbb95d3SBarry Smith   *fl = PETSC_TRUE;
1021cbb95d3SBarry Smith   PetscFunctionReturn(0);
1031cbb95d3SBarry Smith }
1041cbb95d3SBarry Smith 
105289bc588SBarry Smith /* ---------------------------------------------------------------*/
1066831982aSBarry Smith /* COMMENT: I have chosen to hide row permutation in the pivots,
1076831982aSBarry Smith    rather than put it in the Mat->row slot.*/
1084a2ae208SSatish Balay #undef __FUNCT__
1094a2ae208SSatish Balay #define __FUNCT__ "MatLUFactor_SeqDense"
110dfbe8321SBarry Smith PetscErrorCode MatLUFactor_SeqDense(Mat A,IS row,IS col,MatFactorInfo *minfo)
111289bc588SBarry Smith {
112a07ab388SBarry Smith #if defined(PETSC_MISSING_LAPACK_GETRF)
11381f479a6Svictorle   PetscFunctionBegin;
114a07ab388SBarry Smith   SETERRQ(PETSC_ERR_SUP,"GETRF - Lapack routine is unavailable.");
115a07ab388SBarry Smith #else
116c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
1176849ba73SBarry Smith   PetscErrorCode ierr;
1180805154bSBarry Smith   PetscBLASInt   n,m,info;
11955659b69SBarry Smith 
1203a40ed3dSBarry Smith   PetscFunctionBegin;
1210805154bSBarry Smith   n = PetscBLASIntCast(A->cmap.n);
1220805154bSBarry Smith   m = PetscBLASIntCast(A->rmap.n);
123289bc588SBarry Smith   if (!mat->pivots) {
124899cda47SBarry Smith     ierr = PetscMalloc((A->rmap.n+1)*sizeof(PetscBLASInt),&mat->pivots);CHKERRQ(ierr);
125899cda47SBarry Smith     ierr = PetscLogObjectMemory(A,A->rmap.n*sizeof(PetscBLASInt));CHKERRQ(ierr);
126289bc588SBarry Smith   }
127f1af5d2fSBarry Smith   A->factor = FACTOR_LU;
128899cda47SBarry Smith   if (!A->rmap.n || !A->cmap.n) PetscFunctionReturn(0);
12971044d3cSBarry Smith   LAPACKgetrf_(&m,&n,mat->v,&mat->lda,mat->pivots,&info);
13029bbc08cSBarry Smith   if (info<0) SETERRQ(PETSC_ERR_LIB,"Bad argument to LU factorization");
13129bbc08cSBarry Smith   if (info>0) SETERRQ(PETSC_ERR_MAT_LU_ZRPVT,"Bad LU factorization");
132899cda47SBarry Smith   ierr = PetscLogFlops((2*A->cmap.n*A->cmap.n*A->cmap.n)/3);CHKERRQ(ierr);
133a07ab388SBarry Smith #endif
1343a40ed3dSBarry Smith   PetscFunctionReturn(0);
135289bc588SBarry Smith }
1366ee01492SSatish Balay 
1374a2ae208SSatish Balay #undef __FUNCT__
1384a2ae208SSatish Balay #define __FUNCT__ "MatDuplicate_SeqDense"
139dfbe8321SBarry Smith PetscErrorCode MatDuplicate_SeqDense(Mat A,MatDuplicateOption cpvalues,Mat *newmat)
14002cad45dSBarry Smith {
14102cad45dSBarry Smith   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data,*l;
1426849ba73SBarry Smith   PetscErrorCode ierr;
14313f74950SBarry Smith   PetscInt       lda = (PetscInt)mat->lda,j,m;
14402cad45dSBarry Smith   Mat            newi;
14502cad45dSBarry Smith 
1463a40ed3dSBarry Smith   PetscFunctionBegin;
1477adad957SLisandro Dalcin   ierr = MatCreate(((PetscObject)A)->comm,&newi);CHKERRQ(ierr);
148899cda47SBarry Smith   ierr = MatSetSizes(newi,A->rmap.n,A->cmap.n,A->rmap.n,A->cmap.n);CHKERRQ(ierr);
1497adad957SLisandro Dalcin   ierr = MatSetType(newi,((PetscObject)A)->type_name);CHKERRQ(ierr);
1505c5985e7SKris Buschelman   ierr = MatSeqDenseSetPreallocation(newi,PETSC_NULL);CHKERRQ(ierr);
1515609ef8eSBarry Smith   if (cpvalues == MAT_COPY_VALUES) {
152a5ce6ee0Svictorle     l = (Mat_SeqDense*)newi->data;
153899cda47SBarry Smith     if (lda>A->rmap.n) {
154899cda47SBarry Smith       m = A->rmap.n;
155899cda47SBarry Smith       for (j=0; j<A->cmap.n; j++) {
156a5ce6ee0Svictorle 	ierr = PetscMemcpy(l->v+j*m,mat->v+j*lda,m*sizeof(PetscScalar));CHKERRQ(ierr);
157a5ce6ee0Svictorle       }
158a5ce6ee0Svictorle     } else {
159899cda47SBarry Smith       ierr = PetscMemcpy(l->v,mat->v,A->rmap.n*A->cmap.n*sizeof(PetscScalar));CHKERRQ(ierr);
16002cad45dSBarry Smith     }
161a5ce6ee0Svictorle   }
16202cad45dSBarry Smith   newi->assembled = PETSC_TRUE;
16302cad45dSBarry Smith   *newmat = newi;
1643a40ed3dSBarry Smith   PetscFunctionReturn(0);
16502cad45dSBarry Smith }
16602cad45dSBarry Smith 
1674a2ae208SSatish Balay #undef __FUNCT__
1684a2ae208SSatish Balay #define __FUNCT__ "MatLUFactorSymbolic_SeqDense"
169dfbe8321SBarry Smith PetscErrorCode MatLUFactorSymbolic_SeqDense(Mat A,IS row,IS col,MatFactorInfo *info,Mat *fact)
170289bc588SBarry Smith {
171dfbe8321SBarry Smith   PetscErrorCode ierr;
1723a40ed3dSBarry Smith 
1733a40ed3dSBarry Smith   PetscFunctionBegin;
1745609ef8eSBarry Smith   ierr = MatDuplicate_SeqDense(A,MAT_DO_NOT_COPY_VALUES,fact);CHKERRQ(ierr);
1753a40ed3dSBarry Smith   PetscFunctionReturn(0);
176289bc588SBarry Smith }
1776ee01492SSatish Balay 
1784a2ae208SSatish Balay #undef __FUNCT__
1794a2ae208SSatish Balay #define __FUNCT__ "MatLUFactorNumeric_SeqDense"
180af281ebdSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqDense(Mat A,MatFactorInfo *info_dummy,Mat *fact)
181289bc588SBarry Smith {
18202cad45dSBarry Smith   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data,*l = (Mat_SeqDense*)(*fact)->data;
1836849ba73SBarry Smith   PetscErrorCode ierr;
184899cda47SBarry Smith   PetscInt       lda1=mat->lda,lda2=l->lda, m=A->rmap.n,n=A->cmap.n, j;
1854482741eSBarry Smith   MatFactorInfo  info;
1863a40ed3dSBarry Smith 
1873a40ed3dSBarry Smith   PetscFunctionBegin;
18802cad45dSBarry Smith   /* copy the numerical values */
1891b807ce4Svictorle   if (lda1>m || lda2>m ) {
1901b807ce4Svictorle     for (j=0; j<n; j++) {
1911b807ce4Svictorle       ierr = PetscMemcpy(l->v+j*lda2,mat->v+j*lda1,m*sizeof(PetscScalar));CHKERRQ(ierr);
1921b807ce4Svictorle     }
1931b807ce4Svictorle   } else {
194899cda47SBarry Smith     ierr = PetscMemcpy(l->v,mat->v,A->rmap.n*A->cmap.n*sizeof(PetscScalar));CHKERRQ(ierr);
1951b807ce4Svictorle   }
19602cad45dSBarry Smith   (*fact)->factor = 0;
1974482741eSBarry Smith   ierr = MatLUFactor(*fact,0,0,&info);CHKERRQ(ierr);
1983a40ed3dSBarry Smith   PetscFunctionReturn(0);
199289bc588SBarry Smith }
2006ee01492SSatish Balay 
2014a2ae208SSatish Balay #undef __FUNCT__
2024a2ae208SSatish Balay #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqDense"
203dfbe8321SBarry Smith PetscErrorCode MatCholeskyFactorSymbolic_SeqDense(Mat A,IS row,MatFactorInfo *info,Mat *fact)
204289bc588SBarry Smith {
205dfbe8321SBarry Smith   PetscErrorCode ierr;
2063a40ed3dSBarry Smith 
2073a40ed3dSBarry Smith   PetscFunctionBegin;
208ceb03754SKris Buschelman   ierr = MatConvert(A,MATSAME,MAT_INITIAL_MATRIX,fact);CHKERRQ(ierr);
2093a40ed3dSBarry Smith   PetscFunctionReturn(0);
210289bc588SBarry Smith }
2116ee01492SSatish Balay 
2124a2ae208SSatish Balay #undef __FUNCT__
2134a2ae208SSatish Balay #define __FUNCT__ "MatCholeskyFactor_SeqDense"
214dfbe8321SBarry Smith PetscErrorCode MatCholeskyFactor_SeqDense(Mat A,IS perm,MatFactorInfo *factinfo)
215289bc588SBarry Smith {
216a07ab388SBarry Smith #if defined(PETSC_MISSING_LAPACK_POTRF)
217a07ab388SBarry Smith   PetscFunctionBegin;
218a07ab388SBarry Smith   SETERRQ(PETSC_ERR_SUP,"POTRF - Lapack routine is unavailable.");
219a07ab388SBarry Smith #else
220c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
2216849ba73SBarry Smith   PetscErrorCode ierr;
2220805154bSBarry Smith   PetscBLASInt   info,n = PetscBLASIntCast(A->cmap.n);
22355659b69SBarry Smith 
2243a40ed3dSBarry Smith   PetscFunctionBegin;
225606d414cSSatish Balay   ierr = PetscFree(mat->pivots);CHKERRQ(ierr);
226752f5567SLois Curfman McInnes   mat->pivots = 0;
22705b42c5fSBarry Smith 
228899cda47SBarry Smith   if (!A->rmap.n || !A->cmap.n) PetscFunctionReturn(0);
22971044d3cSBarry Smith   LAPACKpotrf_("L",&n,mat->v,&mat->lda,&info);
23077431f27SBarry Smith   if (info) SETERRQ1(PETSC_ERR_MAT_CH_ZRPVT,"Bad factorization: zero pivot in row %D",(PetscInt)info-1);
231c0bbcb79SLois Curfman McInnes   A->factor = FACTOR_CHOLESKY;
232899cda47SBarry Smith   ierr = PetscLogFlops((A->cmap.n*A->cmap.n*A->cmap.n)/3);CHKERRQ(ierr);
233a07ab388SBarry Smith #endif
2343a40ed3dSBarry Smith   PetscFunctionReturn(0);
235289bc588SBarry Smith }
236289bc588SBarry Smith 
2374a2ae208SSatish Balay #undef __FUNCT__
2380b4b3355SBarry Smith #define __FUNCT__ "MatCholeskyFactorNumeric_SeqDense"
239af281ebdSHong Zhang PetscErrorCode MatCholeskyFactorNumeric_SeqDense(Mat A,MatFactorInfo *info_dummy,Mat *fact)
2400b4b3355SBarry Smith {
241dfbe8321SBarry Smith   PetscErrorCode ierr;
24215e8a5b3SHong Zhang   MatFactorInfo  info;
2430b4b3355SBarry Smith 
2440b4b3355SBarry Smith   PetscFunctionBegin;
24515e8a5b3SHong Zhang   info.fill = 1.0;
24615e8a5b3SHong Zhang   ierr = MatCholeskyFactor_SeqDense(*fact,0,&info);CHKERRQ(ierr);
2470b4b3355SBarry Smith   PetscFunctionReturn(0);
2480b4b3355SBarry Smith }
2490b4b3355SBarry Smith 
2500b4b3355SBarry Smith #undef __FUNCT__
2514a2ae208SSatish Balay #define __FUNCT__ "MatSolve_SeqDense"
252dfbe8321SBarry Smith PetscErrorCode MatSolve_SeqDense(Mat A,Vec xx,Vec yy)
253289bc588SBarry Smith {
254c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
2556849ba73SBarry Smith   PetscErrorCode ierr;
25687828ca2SBarry Smith   PetscScalar    *x,*y;
2570805154bSBarry Smith   PetscBLASInt   one = 1,info,m = PetscBLASIntCast(A->rmap.n);
25867e560aaSBarry Smith 
2593a40ed3dSBarry Smith   PetscFunctionBegin;
2601ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
2611ebc52fbSHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
262899cda47SBarry Smith   ierr = PetscMemcpy(y,x,A->rmap.n*sizeof(PetscScalar));CHKERRQ(ierr);
263c0bbcb79SLois Curfman McInnes   if (A->factor == FACTOR_LU) {
264ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_GETRS)
26580444007SBarry Smith     SETERRQ(PETSC_ERR_SUP,"GETRS - Lapack routine is unavailable.");
266ae7cfcebSSatish Balay #else
26771044d3cSBarry Smith     LAPACKgetrs_("N",&m,&one,mat->v,&mat->lda,mat->pivots,y,&m,&info);
2684ce68768SBarry Smith     if (info) SETERRQ(PETSC_ERR_LIB,"GETRS - Bad solve");
269ae7cfcebSSatish Balay #endif
2707a97a34bSBarry Smith   } else if (A->factor == FACTOR_CHOLESKY){
271ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_POTRS)
27280444007SBarry Smith     SETERRQ(PETSC_ERR_SUP,"POTRS - Lapack routine is unavailable.");
273ae7cfcebSSatish Balay #else
27471044d3cSBarry Smith     LAPACKpotrs_("L",&m,&one,mat->v,&mat->lda,y,&m,&info);
2754ce68768SBarry Smith     if (info) SETERRQ(PETSC_ERR_LIB,"POTRS Bad solve");
276ae7cfcebSSatish Balay #endif
277289bc588SBarry Smith   }
27829bbc08cSBarry Smith   else SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Matrix must be factored to solve");
2791ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
2801ebc52fbSHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
281899cda47SBarry Smith   ierr = PetscLogFlops(2*A->cmap.n*A->cmap.n - A->cmap.n);CHKERRQ(ierr);
2823a40ed3dSBarry Smith   PetscFunctionReturn(0);
283289bc588SBarry Smith }
2846ee01492SSatish Balay 
2854a2ae208SSatish Balay #undef __FUNCT__
2864a2ae208SSatish Balay #define __FUNCT__ "MatSolveTranspose_SeqDense"
287dfbe8321SBarry Smith PetscErrorCode MatSolveTranspose_SeqDense(Mat A,Vec xx,Vec yy)
288da3a660dSBarry Smith {
289c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
290dfbe8321SBarry Smith   PetscErrorCode ierr;
29187828ca2SBarry Smith   PetscScalar    *x,*y;
2920805154bSBarry Smith   PetscBLASInt   one = 1,info,m = PetscBLASIntCast(A->rmap.n);
29367e560aaSBarry Smith 
2943a40ed3dSBarry Smith   PetscFunctionBegin;
2951ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
2961ebc52fbSHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
297899cda47SBarry Smith   ierr = PetscMemcpy(y,x,A->rmap.n*sizeof(PetscScalar));CHKERRQ(ierr);
298752f5567SLois Curfman McInnes   /* assume if pivots exist then use LU; else Cholesky */
299da3a660dSBarry Smith   if (mat->pivots) {
300ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_GETRS)
30180444007SBarry Smith     SETERRQ(PETSC_ERR_SUP,"GETRS - Lapack routine is unavailable.");
302ae7cfcebSSatish Balay #else
30371044d3cSBarry Smith     LAPACKgetrs_("T",&m,&one,mat->v,&mat->lda,mat->pivots,y,&m,&info);
3044ce68768SBarry Smith     if (info) SETERRQ(PETSC_ERR_LIB,"POTRS - Bad solve");
305ae7cfcebSSatish Balay #endif
3067a97a34bSBarry Smith   } else {
307ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_POTRS)
30880444007SBarry Smith     SETERRQ(PETSC_ERR_SUP,"POTRS - Lapack routine is unavailable.");
309ae7cfcebSSatish Balay #else
31071044d3cSBarry Smith     LAPACKpotrs_("L",&m,&one,mat->v,&mat->lda,y,&m,&info);
3114ce68768SBarry Smith     if (info) SETERRQ(PETSC_ERR_LIB,"POTRS - Bad solve");
312ae7cfcebSSatish Balay #endif
313da3a660dSBarry Smith   }
3141ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
3151ebc52fbSHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
316899cda47SBarry Smith   ierr = PetscLogFlops(2*A->cmap.n*A->cmap.n - A->cmap.n);CHKERRQ(ierr);
3173a40ed3dSBarry Smith   PetscFunctionReturn(0);
318da3a660dSBarry Smith }
3196ee01492SSatish Balay 
3204a2ae208SSatish Balay #undef __FUNCT__
3214a2ae208SSatish Balay #define __FUNCT__ "MatSolveAdd_SeqDense"
322dfbe8321SBarry Smith PetscErrorCode MatSolveAdd_SeqDense(Mat A,Vec xx,Vec zz,Vec yy)
323da3a660dSBarry Smith {
324c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
325dfbe8321SBarry Smith   PetscErrorCode ierr;
32687828ca2SBarry Smith   PetscScalar    *x,*y,sone = 1.0;
327da3a660dSBarry Smith   Vec            tmp = 0;
3280805154bSBarry Smith   PetscBLASInt   one = 1,info,m = PetscBLASIntCast(A->rmap.n);
32967e560aaSBarry Smith 
3303a40ed3dSBarry Smith   PetscFunctionBegin;
3311ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
3321ebc52fbSHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
333899cda47SBarry Smith   if (!A->rmap.n || !A->cmap.n) PetscFunctionReturn(0);
334da3a660dSBarry Smith   if (yy == zz) {
33578b31e54SBarry Smith     ierr = VecDuplicate(yy,&tmp);CHKERRQ(ierr);
33652e6d16bSBarry Smith     ierr = PetscLogObjectParent(A,tmp);CHKERRQ(ierr);
33778b31e54SBarry Smith     ierr = VecCopy(yy,tmp);CHKERRQ(ierr);
338da3a660dSBarry Smith   }
339899cda47SBarry Smith   ierr = PetscMemcpy(y,x,A->rmap.n*sizeof(PetscScalar));CHKERRQ(ierr);
340752f5567SLois Curfman McInnes   /* assume if pivots exist then use LU; else Cholesky */
341da3a660dSBarry Smith   if (mat->pivots) {
342ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_GETRS)
34380444007SBarry Smith     SETERRQ(PETSC_ERR_SUP,"GETRS - Lapack routine is unavailable.");
344ae7cfcebSSatish Balay #else
34571044d3cSBarry Smith     LAPACKgetrs_("N",&m,&one,mat->v,&mat->lda,mat->pivots,y,&m,&info);
3462ffef20aSBarry Smith     if (info) SETERRQ(PETSC_ERR_LIB,"Bad solve");
347ae7cfcebSSatish Balay #endif
348a8c6a408SBarry Smith   } else {
349ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_POTRS)
35080444007SBarry Smith     SETERRQ(PETSC_ERR_SUP,"POTRS - Lapack routine is unavailable.");
351ae7cfcebSSatish Balay #else
35271044d3cSBarry Smith     LAPACKpotrs_("L",&m,&one,mat->v,&mat->lda,y,&m,&info);
3532ffef20aSBarry Smith     if (info) SETERRQ(PETSC_ERR_LIB,"Bad solve");
354ae7cfcebSSatish Balay #endif
355da3a660dSBarry Smith   }
3562dcb1b2aSMatthew Knepley   if (tmp) {ierr = VecAXPY(yy,sone,tmp);CHKERRQ(ierr); ierr = VecDestroy(tmp);CHKERRQ(ierr);}
3572dcb1b2aSMatthew Knepley   else     {ierr = VecAXPY(yy,sone,zz);CHKERRQ(ierr);}
3581ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
3591ebc52fbSHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
360899cda47SBarry Smith   ierr = PetscLogFlops(2*A->cmap.n*A->cmap.n);CHKERRQ(ierr);
3613a40ed3dSBarry Smith   PetscFunctionReturn(0);
362da3a660dSBarry Smith }
36367e560aaSBarry Smith 
3644a2ae208SSatish Balay #undef __FUNCT__
3654a2ae208SSatish Balay #define __FUNCT__ "MatSolveTransposeAdd_SeqDense"
366dfbe8321SBarry Smith PetscErrorCode MatSolveTransposeAdd_SeqDense(Mat A,Vec xx,Vec zz,Vec yy)
367da3a660dSBarry Smith {
368c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
3696849ba73SBarry Smith   PetscErrorCode ierr;
37087828ca2SBarry Smith   PetscScalar    *x,*y,sone = 1.0;
371da3a660dSBarry Smith   Vec            tmp;
3720805154bSBarry Smith   PetscBLASInt   one = 1,info,m = PetscBLASIntCast(A->rmap.n);
37367e560aaSBarry Smith 
3743a40ed3dSBarry Smith   PetscFunctionBegin;
375899cda47SBarry Smith   if (!A->rmap.n || !A->cmap.n) PetscFunctionReturn(0);
3761ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
3771ebc52fbSHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
378da3a660dSBarry Smith   if (yy == zz) {
37978b31e54SBarry Smith     ierr = VecDuplicate(yy,&tmp);CHKERRQ(ierr);
38052e6d16bSBarry Smith     ierr = PetscLogObjectParent(A,tmp);CHKERRQ(ierr);
38178b31e54SBarry Smith     ierr = VecCopy(yy,tmp);CHKERRQ(ierr);
382da3a660dSBarry Smith   }
383899cda47SBarry Smith   ierr = PetscMemcpy(y,x,A->rmap.n*sizeof(PetscScalar));CHKERRQ(ierr);
384752f5567SLois Curfman McInnes   /* assume if pivots exist then use LU; else Cholesky */
385da3a660dSBarry Smith   if (mat->pivots) {
386ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_GETRS)
38780444007SBarry Smith     SETERRQ(PETSC_ERR_SUP,"GETRS - Lapack routine is unavailable.");
388ae7cfcebSSatish Balay #else
38971044d3cSBarry Smith     LAPACKgetrs_("T",&m,&one,mat->v,&mat->lda,mat->pivots,y,&m,&info);
3902ffef20aSBarry Smith     if (info) SETERRQ(PETSC_ERR_LIB,"Bad solve");
391ae7cfcebSSatish Balay #endif
3923a40ed3dSBarry Smith   } else {
393ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_POTRS)
39480444007SBarry Smith     SETERRQ(PETSC_ERR_SUP,"POTRS - Lapack routine is unavailable.");
395ae7cfcebSSatish Balay #else
39671044d3cSBarry Smith     LAPACKpotrs_("L",&m,&one,mat->v,&mat->lda,y,&m,&info);
3972ffef20aSBarry Smith     if (info) SETERRQ(PETSC_ERR_LIB,"Bad solve");
398ae7cfcebSSatish Balay #endif
399da3a660dSBarry Smith   }
40090f02eecSBarry Smith   if (tmp) {
4012dcb1b2aSMatthew Knepley     ierr = VecAXPY(yy,sone,tmp);CHKERRQ(ierr);
40290f02eecSBarry Smith     ierr = VecDestroy(tmp);CHKERRQ(ierr);
4033a40ed3dSBarry Smith   } else {
4042dcb1b2aSMatthew Knepley     ierr = VecAXPY(yy,sone,zz);CHKERRQ(ierr);
40590f02eecSBarry Smith   }
4061ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
4071ebc52fbSHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
408899cda47SBarry Smith   ierr = PetscLogFlops(2*A->cmap.n*A->cmap.n);CHKERRQ(ierr);
4093a40ed3dSBarry Smith   PetscFunctionReturn(0);
410da3a660dSBarry Smith }
411289bc588SBarry Smith /* ------------------------------------------------------------------*/
4124a2ae208SSatish Balay #undef __FUNCT__
4134a2ae208SSatish Balay #define __FUNCT__ "MatRelax_SeqDense"
41413f74950SBarry Smith PetscErrorCode MatRelax_SeqDense(Mat A,Vec bb,PetscReal omega,MatSORType flag,PetscReal shift,PetscInt its,PetscInt lits,Vec xx)
415289bc588SBarry Smith {
416c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
41787828ca2SBarry Smith   PetscScalar    *x,*b,*v = mat->v,zero = 0.0,xt;
418dfbe8321SBarry Smith   PetscErrorCode ierr;
419899cda47SBarry Smith   PetscInt       m = A->rmap.n,i;
420aa482453SBarry Smith #if !defined(PETSC_USE_COMPLEX)
4210805154bSBarry Smith   PetscBLASInt   o = 1,bm = PetscBLASIntCast(m);
422bc1b551cSSatish Balay #endif
423289bc588SBarry Smith 
4243a40ed3dSBarry Smith   PetscFunctionBegin;
425289bc588SBarry Smith   if (flag & SOR_ZERO_INITIAL_GUESS) {
42671044d3cSBarry Smith     /* this is a hack fix, should have another version without the second BLASdot */
4272dcb1b2aSMatthew Knepley     ierr = VecSet(xx,zero);CHKERRQ(ierr);
428289bc588SBarry Smith   }
4291ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
4301ebc52fbSHong Zhang   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
431b965ef7fSBarry Smith   its  = its*lits;
43277431f27SBarry Smith   if (its <= 0) SETERRQ2(PETSC_ERR_ARG_WRONG,"Relaxation requires global its %D and local its %D both positive",its,lits);
433289bc588SBarry Smith   while (its--) {
434fccaa45eSBarry Smith     if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){
435289bc588SBarry Smith       for (i=0; i<m; i++) {
436aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX)
437f1747703SBarry Smith         /* cannot use BLAS dot for complex because compiler/linker is
438f1747703SBarry Smith            not happy about returning a double complex */
43913f74950SBarry Smith         PetscInt    _i;
44087828ca2SBarry Smith         PetscScalar sum = b[i];
441f1747703SBarry Smith         for (_i=0; _i<m; _i++) {
4423f6de6efSSatish Balay           sum -= PetscConj(v[i+_i*m])*x[_i];
443f1747703SBarry Smith         }
444f1747703SBarry Smith         xt = sum;
445f1747703SBarry Smith #else
44671044d3cSBarry Smith         xt = b[i] - BLASdot_(&bm,v+i,&bm,x,&o);
447f1747703SBarry Smith #endif
44855a1b374SBarry Smith         x[i] = (1. - omega)*x[i] + omega*(xt+v[i + i*m]*x[i])/(v[i + i*m]+shift);
449289bc588SBarry Smith       }
450289bc588SBarry Smith     }
451fccaa45eSBarry Smith     if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){
452289bc588SBarry Smith       for (i=m-1; i>=0; i--) {
453aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX)
454f1747703SBarry Smith         /* cannot use BLAS dot for complex because compiler/linker is
455f1747703SBarry Smith            not happy about returning a double complex */
45613f74950SBarry Smith         PetscInt    _i;
45787828ca2SBarry Smith         PetscScalar sum = b[i];
458f1747703SBarry Smith         for (_i=0; _i<m; _i++) {
4593f6de6efSSatish Balay           sum -= PetscConj(v[i+_i*m])*x[_i];
460f1747703SBarry Smith         }
461f1747703SBarry Smith         xt = sum;
462f1747703SBarry Smith #else
46371044d3cSBarry Smith         xt = b[i] - BLASdot_(&bm,v+i,&bm,x,&o);
464f1747703SBarry Smith #endif
46555a1b374SBarry Smith         x[i] = (1. - omega)*x[i] + omega*(xt+v[i + i*m]*x[i])/(v[i + i*m]+shift);
466289bc588SBarry Smith       }
467289bc588SBarry Smith     }
468289bc588SBarry Smith   }
4691ebc52fbSHong Zhang   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
4701ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
4713a40ed3dSBarry Smith   PetscFunctionReturn(0);
472289bc588SBarry Smith }
473289bc588SBarry Smith 
474289bc588SBarry Smith /* -----------------------------------------------------------------*/
4754a2ae208SSatish Balay #undef __FUNCT__
4764a2ae208SSatish Balay #define __FUNCT__ "MatMultTranspose_SeqDense"
477dfbe8321SBarry Smith PetscErrorCode MatMultTranspose_SeqDense(Mat A,Vec xx,Vec yy)
478289bc588SBarry Smith {
479c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
48087828ca2SBarry Smith   PetscScalar    *v = mat->v,*x,*y;
481dfbe8321SBarry Smith   PetscErrorCode ierr;
4820805154bSBarry Smith   PetscBLASInt   m, n,_One=1;
483ea709b57SSatish Balay   PetscScalar    _DOne=1.0,_DZero=0.0;
4843a40ed3dSBarry Smith 
4853a40ed3dSBarry Smith   PetscFunctionBegin;
4860805154bSBarry Smith   m = PetscBLASIntCast(A->rmap.n);
4870805154bSBarry Smith   n = PetscBLASIntCast(A->cmap.n);
488899cda47SBarry Smith   if (!A->rmap.n || !A->cmap.n) PetscFunctionReturn(0);
4891ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
4901ebc52fbSHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
49171044d3cSBarry Smith   BLASgemv_("T",&m,&n,&_DOne,v,&mat->lda,x,&_One,&_DZero,y,&_One);
4921ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
4931ebc52fbSHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
494899cda47SBarry Smith   ierr = PetscLogFlops(2*A->rmap.n*A->cmap.n - A->cmap.n);CHKERRQ(ierr);
4953a40ed3dSBarry Smith   PetscFunctionReturn(0);
496289bc588SBarry Smith }
4976ee01492SSatish Balay 
4984a2ae208SSatish Balay #undef __FUNCT__
4994a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqDense"
500dfbe8321SBarry Smith PetscErrorCode MatMult_SeqDense(Mat A,Vec xx,Vec yy)
501289bc588SBarry Smith {
502c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
50387828ca2SBarry Smith   PetscScalar    *v = mat->v,*x,*y,_DOne=1.0,_DZero=0.0;
504dfbe8321SBarry Smith   PetscErrorCode ierr;
5050805154bSBarry Smith   PetscBLASInt   m, n, _One=1;
5063a40ed3dSBarry Smith 
5073a40ed3dSBarry Smith   PetscFunctionBegin;
5080805154bSBarry Smith   m = PetscBLASIntCast(A->rmap.n);
5090805154bSBarry Smith   n = PetscBLASIntCast(A->cmap.n);
510899cda47SBarry Smith   if (!A->rmap.n || !A->cmap.n) PetscFunctionReturn(0);
5111ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
5121ebc52fbSHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
51371044d3cSBarry Smith   BLASgemv_("N",&m,&n,&_DOne,v,&(mat->lda),x,&_One,&_DZero,y,&_One);
5141ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
5151ebc52fbSHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
516899cda47SBarry Smith   ierr = PetscLogFlops(2*A->rmap.n*A->cmap.n - A->rmap.n);CHKERRQ(ierr);
5173a40ed3dSBarry Smith   PetscFunctionReturn(0);
518289bc588SBarry Smith }
5196ee01492SSatish Balay 
5204a2ae208SSatish Balay #undef __FUNCT__
5214a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqDense"
522dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqDense(Mat A,Vec xx,Vec zz,Vec yy)
523289bc588SBarry Smith {
524c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
52587828ca2SBarry Smith   PetscScalar    *v = mat->v,*x,*y,_DOne=1.0;
526dfbe8321SBarry Smith   PetscErrorCode ierr;
5270805154bSBarry Smith   PetscBLASInt   m, n, _One=1;
5283a40ed3dSBarry Smith 
5293a40ed3dSBarry Smith   PetscFunctionBegin;
5300805154bSBarry Smith   m = PetscBLASIntCast(A->rmap.n);
5310805154bSBarry Smith   n = PetscBLASIntCast(A->cmap.n);
532899cda47SBarry Smith   if (!A->rmap.n || !A->cmap.n) PetscFunctionReturn(0);
533600d6d0bSBarry Smith   if (zz != yy) {ierr = VecCopy(zz,yy);CHKERRQ(ierr);}
5341ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
5351ebc52fbSHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
53671044d3cSBarry Smith   BLASgemv_("N",&m,&n,&_DOne,v,&(mat->lda),x,&_One,&_DOne,y,&_One);
5371ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
5381ebc52fbSHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
539899cda47SBarry Smith   ierr = PetscLogFlops(2*A->rmap.n*A->cmap.n);CHKERRQ(ierr);
5403a40ed3dSBarry Smith   PetscFunctionReturn(0);
541289bc588SBarry Smith }
5426ee01492SSatish Balay 
5434a2ae208SSatish Balay #undef __FUNCT__
5444a2ae208SSatish Balay #define __FUNCT__ "MatMultTransposeAdd_SeqDense"
545dfbe8321SBarry Smith PetscErrorCode MatMultTransposeAdd_SeqDense(Mat A,Vec xx,Vec zz,Vec yy)
546289bc588SBarry Smith {
547c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
54887828ca2SBarry Smith   PetscScalar    *v = mat->v,*x,*y;
549dfbe8321SBarry Smith   PetscErrorCode ierr;
5500805154bSBarry Smith   PetscBLASInt   m, n, _One=1;
55187828ca2SBarry Smith   PetscScalar    _DOne=1.0;
5523a40ed3dSBarry Smith 
5533a40ed3dSBarry Smith   PetscFunctionBegin;
5540805154bSBarry Smith   m = PetscBLASIntCast(A->rmap.n);
5550805154bSBarry Smith   n = PetscBLASIntCast(A->cmap.n);
556899cda47SBarry Smith   if (!A->rmap.n || !A->cmap.n) PetscFunctionReturn(0);
557600d6d0bSBarry Smith   if (zz != yy) {ierr = VecCopy(zz,yy);CHKERRQ(ierr);}
5581ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
5591ebc52fbSHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
56071044d3cSBarry Smith   BLASgemv_("T",&m,&n,&_DOne,v,&(mat->lda),x,&_One,&_DOne,y,&_One);
5611ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
5621ebc52fbSHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
563899cda47SBarry Smith   ierr = PetscLogFlops(2*A->rmap.n*A->cmap.n);CHKERRQ(ierr);
5643a40ed3dSBarry Smith   PetscFunctionReturn(0);
565289bc588SBarry Smith }
566289bc588SBarry Smith 
567289bc588SBarry Smith /* -----------------------------------------------------------------*/
5684a2ae208SSatish Balay #undef __FUNCT__
5694a2ae208SSatish Balay #define __FUNCT__ "MatGetRow_SeqDense"
57013f74950SBarry Smith PetscErrorCode MatGetRow_SeqDense(Mat A,PetscInt row,PetscInt *ncols,PetscInt **cols,PetscScalar **vals)
571289bc588SBarry Smith {
572c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
57387828ca2SBarry Smith   PetscScalar    *v;
5746849ba73SBarry Smith   PetscErrorCode ierr;
57513f74950SBarry Smith   PetscInt       i;
57667e560aaSBarry Smith 
5773a40ed3dSBarry Smith   PetscFunctionBegin;
578899cda47SBarry Smith   *ncols = A->cmap.n;
579289bc588SBarry Smith   if (cols) {
580899cda47SBarry Smith     ierr = PetscMalloc((A->cmap.n+1)*sizeof(PetscInt),cols);CHKERRQ(ierr);
581899cda47SBarry Smith     for (i=0; i<A->cmap.n; i++) (*cols)[i] = i;
582289bc588SBarry Smith   }
583289bc588SBarry Smith   if (vals) {
584899cda47SBarry Smith     ierr = PetscMalloc((A->cmap.n+1)*sizeof(PetscScalar),vals);CHKERRQ(ierr);
585289bc588SBarry Smith     v    = mat->v + row;
586899cda47SBarry Smith     for (i=0; i<A->cmap.n; i++) {(*vals)[i] = *v; v += mat->lda;}
587289bc588SBarry Smith   }
5883a40ed3dSBarry Smith   PetscFunctionReturn(0);
589289bc588SBarry Smith }
5906ee01492SSatish Balay 
5914a2ae208SSatish Balay #undef __FUNCT__
5924a2ae208SSatish Balay #define __FUNCT__ "MatRestoreRow_SeqDense"
59313f74950SBarry Smith PetscErrorCode MatRestoreRow_SeqDense(Mat A,PetscInt row,PetscInt *ncols,PetscInt **cols,PetscScalar **vals)
594289bc588SBarry Smith {
595dfbe8321SBarry Smith   PetscErrorCode ierr;
596606d414cSSatish Balay   PetscFunctionBegin;
597606d414cSSatish Balay   if (cols) {ierr = PetscFree(*cols);CHKERRQ(ierr);}
598606d414cSSatish Balay   if (vals) {ierr = PetscFree(*vals);CHKERRQ(ierr); }
5993a40ed3dSBarry Smith   PetscFunctionReturn(0);
600289bc588SBarry Smith }
601289bc588SBarry Smith /* ----------------------------------------------------------------*/
6024a2ae208SSatish Balay #undef __FUNCT__
6034a2ae208SSatish Balay #define __FUNCT__ "MatSetValues_SeqDense"
60413f74950SBarry Smith PetscErrorCode MatSetValues_SeqDense(Mat A,PetscInt m,const PetscInt indexm[],PetscInt n,const PetscInt indexn[],const PetscScalar v[],InsertMode addv)
605289bc588SBarry Smith {
606c0bbcb79SLois Curfman McInnes   Mat_SeqDense *mat = (Mat_SeqDense*)A->data;
60713f74950SBarry Smith   PetscInt     i,j,idx=0;
608d6dfbf8fSBarry Smith 
6093a40ed3dSBarry Smith   PetscFunctionBegin;
610289bc588SBarry Smith   if (!mat->roworiented) {
611dbb450caSBarry Smith     if (addv == INSERT_VALUES) {
612289bc588SBarry Smith       for (j=0; j<n; j++) {
613cddbea37SSatish Balay         if (indexn[j] < 0) {idx += m; continue;}
6142515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
615899cda47SBarry Smith         if (indexn[j] >= A->cmap.n) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",indexn[j],A->cmap.n-1);
61658804f6dSBarry Smith #endif
617289bc588SBarry Smith         for (i=0; i<m; i++) {
618cddbea37SSatish Balay           if (indexm[i] < 0) {idx++; continue;}
6192515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
620899cda47SBarry Smith           if (indexm[i] >= A->rmap.n) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",indexm[i],A->rmap.n-1);
62158804f6dSBarry Smith #endif
622cddbea37SSatish Balay           mat->v[indexn[j]*mat->lda + indexm[i]] = v[idx++];
623289bc588SBarry Smith         }
624289bc588SBarry Smith       }
6253a40ed3dSBarry Smith     } else {
626289bc588SBarry Smith       for (j=0; j<n; j++) {
627cddbea37SSatish Balay         if (indexn[j] < 0) {idx += m; continue;}
6282515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
629899cda47SBarry Smith         if (indexn[j] >= A->cmap.n) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",indexn[j],A->cmap.n-1);
63058804f6dSBarry Smith #endif
631289bc588SBarry Smith         for (i=0; i<m; i++) {
632cddbea37SSatish Balay           if (indexm[i] < 0) {idx++; continue;}
6332515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
634899cda47SBarry Smith           if (indexm[i] >= A->rmap.n) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",indexm[i],A->rmap.n-1);
63558804f6dSBarry Smith #endif
636cddbea37SSatish Balay           mat->v[indexn[j]*mat->lda + indexm[i]] += v[idx++];
637289bc588SBarry Smith         }
638289bc588SBarry Smith       }
639289bc588SBarry Smith     }
6403a40ed3dSBarry Smith   } else {
641dbb450caSBarry Smith     if (addv == INSERT_VALUES) {
642e8d4e0b9SBarry Smith       for (i=0; i<m; i++) {
643cddbea37SSatish Balay         if (indexm[i] < 0) { idx += n; continue;}
6442515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
645899cda47SBarry Smith         if (indexm[i] >= A->rmap.n) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",indexm[i],A->rmap.n-1);
64658804f6dSBarry Smith #endif
647e8d4e0b9SBarry Smith         for (j=0; j<n; j++) {
648cddbea37SSatish Balay           if (indexn[j] < 0) { idx++; continue;}
6492515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
650899cda47SBarry Smith           if (indexn[j] >= A->cmap.n) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",indexn[j],A->cmap.n-1);
65158804f6dSBarry Smith #endif
652cddbea37SSatish Balay           mat->v[indexn[j]*mat->lda + indexm[i]] = v[idx++];
653e8d4e0b9SBarry Smith         }
654e8d4e0b9SBarry Smith       }
6553a40ed3dSBarry Smith     } else {
656289bc588SBarry Smith       for (i=0; i<m; i++) {
657cddbea37SSatish Balay         if (indexm[i] < 0) { idx += n; continue;}
6582515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
659899cda47SBarry Smith         if (indexm[i] >= A->rmap.n) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",indexm[i],A->rmap.n-1);
66058804f6dSBarry Smith #endif
661289bc588SBarry Smith         for (j=0; j<n; j++) {
662cddbea37SSatish Balay           if (indexn[j] < 0) { idx++; continue;}
6632515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
664899cda47SBarry Smith           if (indexn[j] >= A->cmap.n) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",indexn[j],A->cmap.n-1);
66558804f6dSBarry Smith #endif
666cddbea37SSatish Balay           mat->v[indexn[j]*mat->lda + indexm[i]] += v[idx++];
667289bc588SBarry Smith         }
668289bc588SBarry Smith       }
669289bc588SBarry Smith     }
670e8d4e0b9SBarry Smith   }
6713a40ed3dSBarry Smith   PetscFunctionReturn(0);
672289bc588SBarry Smith }
673e8d4e0b9SBarry Smith 
6744a2ae208SSatish Balay #undef __FUNCT__
6754a2ae208SSatish Balay #define __FUNCT__ "MatGetValues_SeqDense"
67613f74950SBarry Smith PetscErrorCode MatGetValues_SeqDense(Mat A,PetscInt m,const PetscInt indexm[],PetscInt n,const PetscInt indexn[],PetscScalar v[])
677ae80bb75SLois Curfman McInnes {
678ae80bb75SLois Curfman McInnes   Mat_SeqDense *mat = (Mat_SeqDense*)A->data;
67913f74950SBarry Smith   PetscInt     i,j;
680ae80bb75SLois Curfman McInnes 
6813a40ed3dSBarry Smith   PetscFunctionBegin;
682ae80bb75SLois Curfman McInnes   /* row-oriented output */
683ae80bb75SLois Curfman McInnes   for (i=0; i<m; i++) {
68497e567efSBarry Smith     if (indexm[i] < 0) {v += n;continue;}
685f12f8aa1SBarry Smith     if (indexm[i] >= A->rmap.n) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Row %D requested larger than number rows %D",indexm[i],A->rmap.n);
686ae80bb75SLois Curfman McInnes     for (j=0; j<n; j++) {
6876f31f424SBarry Smith       if (indexn[j] < 0) {v++; continue;}
6886f31f424SBarry Smith       if (indexn[j] >= A->cmap.n) SETERRQ2(PETSC_ERR_ARG_OUTOFRANGE,"Column %D requested larger than number columns %D",indexn[j],A->cmap.n);
68997e567efSBarry Smith       *v++ = mat->v[indexn[j]*mat->lda + indexm[i]];
690ae80bb75SLois Curfman McInnes     }
691ae80bb75SLois Curfman McInnes   }
6923a40ed3dSBarry Smith   PetscFunctionReturn(0);
693ae80bb75SLois Curfman McInnes }
694ae80bb75SLois Curfman McInnes 
695289bc588SBarry Smith /* -----------------------------------------------------------------*/
696289bc588SBarry Smith 
697e090d566SSatish Balay #include "petscsys.h"
69888e32edaSLois Curfman McInnes 
6994a2ae208SSatish Balay #undef __FUNCT__
7004a2ae208SSatish Balay #define __FUNCT__ "MatLoad_SeqDense"
701f69a0ea3SMatthew Knepley PetscErrorCode MatLoad_SeqDense(PetscViewer viewer, MatType type,Mat *A)
70288e32edaSLois Curfman McInnes {
70388e32edaSLois Curfman McInnes   Mat_SeqDense   *a;
70488e32edaSLois Curfman McInnes   Mat            B;
7056849ba73SBarry Smith   PetscErrorCode ierr;
70613f74950SBarry Smith   PetscInt       *scols,i,j,nz,header[4];
70713f74950SBarry Smith   int            fd;
70813f74950SBarry Smith   PetscMPIInt    size;
70913f74950SBarry Smith   PetscInt       *rowlengths = 0,M,N,*cols;
71087828ca2SBarry Smith   PetscScalar    *vals,*svals,*v,*w;
71119bcc07fSBarry Smith   MPI_Comm       comm = ((PetscObject)viewer)->comm;
71288e32edaSLois Curfman McInnes 
7133a40ed3dSBarry Smith   PetscFunctionBegin;
714d132466eSBarry Smith   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
71529bbc08cSBarry Smith   if (size > 1) SETERRQ(PETSC_ERR_ARG_WRONG,"view must have one processor");
716b0a32e0cSBarry Smith   ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr);
7170752156aSBarry Smith   ierr = PetscBinaryRead(fd,header,4,PETSC_INT);CHKERRQ(ierr);
718552e946dSBarry Smith   if (header[0] != MAT_FILE_COOKIE) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"Not matrix object");
71988e32edaSLois Curfman McInnes   M = header[1]; N = header[2]; nz = header[3];
72088e32edaSLois Curfman McInnes 
72190ace30eSBarry Smith   if (nz == MATRIX_BINARY_FORMAT_DENSE) { /* matrix in file is dense */
722f69a0ea3SMatthew Knepley     ierr = MatCreate(comm,A);CHKERRQ(ierr);
723f69a0ea3SMatthew Knepley     ierr = MatSetSizes(*A,M,N,M,N);CHKERRQ(ierr);
724be5d1d56SKris Buschelman     ierr = MatSetType(*A,type);CHKERRQ(ierr);
7255c5985e7SKris Buschelman     ierr = MatSeqDenseSetPreallocation(*A,PETSC_NULL);CHKERRQ(ierr);
72690ace30eSBarry Smith     B    = *A;
72790ace30eSBarry Smith     a    = (Mat_SeqDense*)B->data;
7284a41a4d3SSatish Balay     v    = a->v;
7294a41a4d3SSatish Balay     /* Allocate some temp space to read in the values and then flip them
7304a41a4d3SSatish Balay        from row major to column major */
731b72907f3SBarry Smith     ierr = PetscMalloc((M*N > 0 ? M*N : 1)*sizeof(PetscScalar),&w);CHKERRQ(ierr);
73290ace30eSBarry Smith     /* read in nonzero values */
7334a41a4d3SSatish Balay     ierr = PetscBinaryRead(fd,w,M*N,PETSC_SCALAR);CHKERRQ(ierr);
7344a41a4d3SSatish Balay     /* now flip the values and store them in the matrix*/
7354a41a4d3SSatish Balay     for (j=0; j<N; j++) {
7364a41a4d3SSatish Balay       for (i=0; i<M; i++) {
7374a41a4d3SSatish Balay         *v++ =w[i*N+j];
7384a41a4d3SSatish Balay       }
7394a41a4d3SSatish Balay     }
740606d414cSSatish Balay     ierr = PetscFree(w);CHKERRQ(ierr);
7416d4a8577SBarry Smith     ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
7426d4a8577SBarry Smith     ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
74390ace30eSBarry Smith   } else {
74488e32edaSLois Curfman McInnes     /* read row lengths */
74513f74950SBarry Smith     ierr = PetscMalloc((M+1)*sizeof(PetscInt),&rowlengths);CHKERRQ(ierr);
7460752156aSBarry Smith     ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr);
74788e32edaSLois Curfman McInnes 
74888e32edaSLois Curfman McInnes     /* create our matrix */
749f69a0ea3SMatthew Knepley     ierr = MatCreate(comm,A);CHKERRQ(ierr);
750f69a0ea3SMatthew Knepley     ierr = MatSetSizes(*A,M,N,M,N);CHKERRQ(ierr);
751be5d1d56SKris Buschelman     ierr = MatSetType(*A,type);CHKERRQ(ierr);
7525c5985e7SKris Buschelman     ierr = MatSeqDenseSetPreallocation(*A,PETSC_NULL);CHKERRQ(ierr);
75388e32edaSLois Curfman McInnes     B = *A;
75488e32edaSLois Curfman McInnes     a = (Mat_SeqDense*)B->data;
75588e32edaSLois Curfman McInnes     v = a->v;
75688e32edaSLois Curfman McInnes 
75788e32edaSLois Curfman McInnes     /* read column indices and nonzeros */
75813f74950SBarry Smith     ierr = PetscMalloc((nz+1)*sizeof(PetscInt),&scols);CHKERRQ(ierr);
759b0a32e0cSBarry Smith     cols = scols;
7600752156aSBarry Smith     ierr = PetscBinaryRead(fd,cols,nz,PETSC_INT);CHKERRQ(ierr);
76187828ca2SBarry Smith     ierr = PetscMalloc((nz+1)*sizeof(PetscScalar),&svals);CHKERRQ(ierr);
762b0a32e0cSBarry Smith     vals = svals;
7630752156aSBarry Smith     ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR);CHKERRQ(ierr);
76488e32edaSLois Curfman McInnes 
76588e32edaSLois Curfman McInnes     /* insert into matrix */
76688e32edaSLois Curfman McInnes     for (i=0; i<M; i++) {
76788e32edaSLois Curfman McInnes       for (j=0; j<rowlengths[i]; j++) v[i+M*scols[j]] = svals[j];
76888e32edaSLois Curfman McInnes       svals += rowlengths[i]; scols += rowlengths[i];
76988e32edaSLois Curfman McInnes     }
770606d414cSSatish Balay     ierr = PetscFree(vals);CHKERRQ(ierr);
771606d414cSSatish Balay     ierr = PetscFree(cols);CHKERRQ(ierr);
772606d414cSSatish Balay     ierr = PetscFree(rowlengths);CHKERRQ(ierr);
77388e32edaSLois Curfman McInnes 
7746d4a8577SBarry Smith     ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
7756d4a8577SBarry Smith     ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
77690ace30eSBarry Smith   }
7773a40ed3dSBarry Smith   PetscFunctionReturn(0);
77888e32edaSLois Curfman McInnes }
77988e32edaSLois Curfman McInnes 
780e090d566SSatish Balay #include "petscsys.h"
781932b0c3eSLois Curfman McInnes 
7824a2ae208SSatish Balay #undef __FUNCT__
7834a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqDense_ASCII"
7846849ba73SBarry Smith static PetscErrorCode MatView_SeqDense_ASCII(Mat A,PetscViewer viewer)
785289bc588SBarry Smith {
786932b0c3eSLois Curfman McInnes   Mat_SeqDense      *a = (Mat_SeqDense*)A->data;
787dfbe8321SBarry Smith   PetscErrorCode    ierr;
78813f74950SBarry Smith   PetscInt          i,j;
7892dcb1b2aSMatthew Knepley   const char        *name;
79087828ca2SBarry Smith   PetscScalar       *v;
791f3ef73ceSBarry Smith   PetscViewerFormat format;
7925f481a85SSatish Balay #if defined(PETSC_USE_COMPLEX)
7935f481a85SSatish Balay   PetscTruth        allreal = PETSC_TRUE;
7945f481a85SSatish Balay #endif
795932b0c3eSLois Curfman McInnes 
7963a40ed3dSBarry Smith   PetscFunctionBegin;
797435da068SBarry Smith   ierr = PetscObjectGetName((PetscObject)A,&name);CHKERRQ(ierr);
798b0a32e0cSBarry Smith   ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
799456192e2SBarry Smith   if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
8003a40ed3dSBarry Smith     PetscFunctionReturn(0);  /* do nothing for now */
801fb9695e5SSatish Balay   } else if (format == PETSC_VIEWER_ASCII_COMMON) {
802b0a32e0cSBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_NO);CHKERRQ(ierr);
803899cda47SBarry Smith     for (i=0; i<A->rmap.n; i++) {
80444cd7ae7SLois Curfman McInnes       v = a->v + i;
80577431f27SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i);CHKERRQ(ierr);
806899cda47SBarry Smith       for (j=0; j<A->cmap.n; j++) {
807aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX)
808329f5518SBarry Smith         if (PetscRealPart(*v) != 0.0 && PetscImaginaryPart(*v) != 0.0) {
809a83599f4SBarry Smith           ierr = PetscViewerASCIIPrintf(viewer," (%D, %G + %G i) ",j,PetscRealPart(*v),PetscImaginaryPart(*v));CHKERRQ(ierr);
810329f5518SBarry Smith         } else if (PetscRealPart(*v)) {
811a83599f4SBarry Smith           ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",j,PetscRealPart(*v));CHKERRQ(ierr);
8126831982aSBarry Smith         }
81380cd9d93SLois Curfman McInnes #else
8146831982aSBarry Smith         if (*v) {
815a83599f4SBarry Smith           ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",j,*v);CHKERRQ(ierr);
8166831982aSBarry Smith         }
81780cd9d93SLois Curfman McInnes #endif
8181b807ce4Svictorle         v += a->lda;
81980cd9d93SLois Curfman McInnes       }
820b0a32e0cSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
82180cd9d93SLois Curfman McInnes     }
822b0a32e0cSBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_YES);CHKERRQ(ierr);
8233a40ed3dSBarry Smith   } else {
824b0a32e0cSBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_NO);CHKERRQ(ierr);
825aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX)
82647989497SBarry Smith     /* determine if matrix has all real values */
82747989497SBarry Smith     v = a->v;
828899cda47SBarry Smith     for (i=0; i<A->rmap.n*A->cmap.n; i++) {
829ffac6cdbSBarry Smith 	if (PetscImaginaryPart(v[i])) { allreal = PETSC_FALSE; break ;}
83047989497SBarry Smith     }
83147989497SBarry Smith #endif
832fb9695e5SSatish Balay     if (format == PETSC_VIEWER_ASCII_MATLAB) {
8333a7fca6bSBarry Smith       ierr = PetscObjectGetName((PetscObject)A,&name);CHKERRQ(ierr);
834899cda47SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"%% Size = %D %D \n",A->rmap.n,A->cmap.n);CHKERRQ(ierr);
835899cda47SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"%s = zeros(%D,%D);\n",name,A->rmap.n,A->cmap.n);CHKERRQ(ierr);
836fb9695e5SSatish Balay       ierr = PetscViewerASCIIPrintf(viewer,"%s = [\n",name);CHKERRQ(ierr);
837ffac6cdbSBarry Smith     }
838ffac6cdbSBarry Smith 
839899cda47SBarry Smith     for (i=0; i<A->rmap.n; i++) {
840932b0c3eSLois Curfman McInnes       v = a->v + i;
841899cda47SBarry Smith       for (j=0; j<A->cmap.n; j++) {
842aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX)
84347989497SBarry Smith         if (allreal) {
844f32d5d43SBarry Smith           ierr = PetscViewerASCIIPrintf(viewer,"%18.16e ",PetscRealPart(*v));CHKERRQ(ierr);
84547989497SBarry Smith         } else {
846f32d5d43SBarry Smith           ierr = PetscViewerASCIIPrintf(viewer,"%18.16e + %18.16e i ",PetscRealPart(*v),PetscImaginaryPart(*v));CHKERRQ(ierr);
84747989497SBarry Smith         }
848289bc588SBarry Smith #else
849f32d5d43SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"%18.16e ",*v);CHKERRQ(ierr);
850289bc588SBarry Smith #endif
8511b807ce4Svictorle 	v += a->lda;
852289bc588SBarry Smith       }
853b0a32e0cSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
854289bc588SBarry Smith     }
855fb9695e5SSatish Balay     if (format == PETSC_VIEWER_ASCII_MATLAB) {
856b0a32e0cSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"];\n");CHKERRQ(ierr);
857ffac6cdbSBarry Smith     }
858b0a32e0cSBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_YES);CHKERRQ(ierr);
859da3a660dSBarry Smith   }
860b0a32e0cSBarry Smith   ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
8613a40ed3dSBarry Smith   PetscFunctionReturn(0);
862289bc588SBarry Smith }
863289bc588SBarry Smith 
8644a2ae208SSatish Balay #undef __FUNCT__
8654a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqDense_Binary"
8666849ba73SBarry Smith static PetscErrorCode MatView_SeqDense_Binary(Mat A,PetscViewer viewer)
867932b0c3eSLois Curfman McInnes {
868932b0c3eSLois Curfman McInnes   Mat_SeqDense      *a = (Mat_SeqDense*)A->data;
8696849ba73SBarry Smith   PetscErrorCode    ierr;
87013f74950SBarry Smith   int               fd;
871899cda47SBarry Smith   PetscInt          ict,j,n = A->cmap.n,m = A->rmap.n,i,*col_lens,nz = m*n;
87287828ca2SBarry Smith   PetscScalar       *v,*anonz,*vals;
873f3ef73ceSBarry Smith   PetscViewerFormat format;
874932b0c3eSLois Curfman McInnes 
8753a40ed3dSBarry Smith   PetscFunctionBegin;
876b0a32e0cSBarry Smith   ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr);
87790ace30eSBarry Smith 
878b0a32e0cSBarry Smith   ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
879fb9695e5SSatish Balay   if (format == PETSC_VIEWER_BINARY_NATIVE) {
88090ace30eSBarry Smith     /* store the matrix as a dense matrix */
88113f74950SBarry Smith     ierr = PetscMalloc(4*sizeof(PetscInt),&col_lens);CHKERRQ(ierr);
882552e946dSBarry Smith     col_lens[0] = MAT_FILE_COOKIE;
88390ace30eSBarry Smith     col_lens[1] = m;
88490ace30eSBarry Smith     col_lens[2] = n;
88590ace30eSBarry Smith     col_lens[3] = MATRIX_BINARY_FORMAT_DENSE;
8866f69ff64SBarry Smith     ierr = PetscBinaryWrite(fd,col_lens,4,PETSC_INT,PETSC_TRUE);CHKERRQ(ierr);
887606d414cSSatish Balay     ierr = PetscFree(col_lens);CHKERRQ(ierr);
88890ace30eSBarry Smith 
88990ace30eSBarry Smith     /* write out matrix, by rows */
89087828ca2SBarry Smith     ierr = PetscMalloc((m*n+1)*sizeof(PetscScalar),&vals);CHKERRQ(ierr);
89190ace30eSBarry Smith     v    = a->v;
89290ace30eSBarry Smith     for (j=0; j<n; j++) {
893578230a0SSatish Balay       for (i=0; i<m; i++) {
894578230a0SSatish Balay         vals[j + i*n] = *v++;
89590ace30eSBarry Smith       }
89690ace30eSBarry Smith     }
8976f69ff64SBarry Smith     ierr = PetscBinaryWrite(fd,vals,n*m,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr);
898606d414cSSatish Balay     ierr = PetscFree(vals);CHKERRQ(ierr);
89990ace30eSBarry Smith   } else {
90013f74950SBarry Smith     ierr = PetscMalloc((4+nz)*sizeof(PetscInt),&col_lens);CHKERRQ(ierr);
901552e946dSBarry Smith     col_lens[0] = MAT_FILE_COOKIE;
902932b0c3eSLois Curfman McInnes     col_lens[1] = m;
903932b0c3eSLois Curfman McInnes     col_lens[2] = n;
904932b0c3eSLois Curfman McInnes     col_lens[3] = nz;
905932b0c3eSLois Curfman McInnes 
906932b0c3eSLois Curfman McInnes     /* store lengths of each row and write (including header) to file */
907932b0c3eSLois Curfman McInnes     for (i=0; i<m; i++) col_lens[4+i] = n;
9086f69ff64SBarry Smith     ierr = PetscBinaryWrite(fd,col_lens,4+m,PETSC_INT,PETSC_TRUE);CHKERRQ(ierr);
909932b0c3eSLois Curfman McInnes 
910932b0c3eSLois Curfman McInnes     /* Possibly should write in smaller increments, not whole matrix at once? */
911932b0c3eSLois Curfman McInnes     /* store column indices (zero start index) */
912932b0c3eSLois Curfman McInnes     ict = 0;
913932b0c3eSLois Curfman McInnes     for (i=0; i<m; i++) {
914932b0c3eSLois Curfman McInnes       for (j=0; j<n; j++) col_lens[ict++] = j;
915932b0c3eSLois Curfman McInnes     }
9166f69ff64SBarry Smith     ierr = PetscBinaryWrite(fd,col_lens,nz,PETSC_INT,PETSC_FALSE);CHKERRQ(ierr);
917606d414cSSatish Balay     ierr = PetscFree(col_lens);CHKERRQ(ierr);
918932b0c3eSLois Curfman McInnes 
919932b0c3eSLois Curfman McInnes     /* store nonzero values */
92087828ca2SBarry Smith     ierr = PetscMalloc((nz+1)*sizeof(PetscScalar),&anonz);CHKERRQ(ierr);
921932b0c3eSLois Curfman McInnes     ict  = 0;
922932b0c3eSLois Curfman McInnes     for (i=0; i<m; i++) {
923932b0c3eSLois Curfman McInnes       v = a->v + i;
924932b0c3eSLois Curfman McInnes       for (j=0; j<n; j++) {
9251b807ce4Svictorle         anonz[ict++] = *v; v += a->lda;
926932b0c3eSLois Curfman McInnes       }
927932b0c3eSLois Curfman McInnes     }
9286f69ff64SBarry Smith     ierr = PetscBinaryWrite(fd,anonz,nz,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr);
929606d414cSSatish Balay     ierr = PetscFree(anonz);CHKERRQ(ierr);
93090ace30eSBarry Smith   }
9313a40ed3dSBarry Smith   PetscFunctionReturn(0);
932932b0c3eSLois Curfman McInnes }
933932b0c3eSLois Curfman McInnes 
9344a2ae208SSatish Balay #undef __FUNCT__
9354a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqDense_Draw_Zoom"
936dfbe8321SBarry Smith PetscErrorCode MatView_SeqDense_Draw_Zoom(PetscDraw draw,void *Aa)
937f1af5d2fSBarry Smith {
938f1af5d2fSBarry Smith   Mat               A = (Mat) Aa;
939f1af5d2fSBarry Smith   Mat_SeqDense      *a = (Mat_SeqDense*)A->data;
9406849ba73SBarry Smith   PetscErrorCode    ierr;
941899cda47SBarry Smith   PetscInt          m = A->rmap.n,n = A->cmap.n,color,i,j;
94287828ca2SBarry Smith   PetscScalar       *v = a->v;
943b0a32e0cSBarry Smith   PetscViewer       viewer;
944b0a32e0cSBarry Smith   PetscDraw         popup;
945329f5518SBarry Smith   PetscReal         xl,yl,xr,yr,x_l,x_r,y_l,y_r,scale,maxv = 0.0;
946f3ef73ceSBarry Smith   PetscViewerFormat format;
947f1af5d2fSBarry Smith 
948f1af5d2fSBarry Smith   PetscFunctionBegin;
949f1af5d2fSBarry Smith 
950f1af5d2fSBarry Smith   ierr = PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer);CHKERRQ(ierr);
951b0a32e0cSBarry Smith   ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
952b0a32e0cSBarry Smith   ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr);
953f1af5d2fSBarry Smith 
954f1af5d2fSBarry Smith   /* Loop over matrix elements drawing boxes */
955fb9695e5SSatish Balay   if (format != PETSC_VIEWER_DRAW_CONTOUR) {
956f1af5d2fSBarry Smith     /* Blue for negative and Red for positive */
957b0a32e0cSBarry Smith     color = PETSC_DRAW_BLUE;
958f1af5d2fSBarry Smith     for(j = 0; j < n; j++) {
959f1af5d2fSBarry Smith       x_l = j;
960f1af5d2fSBarry Smith       x_r = x_l + 1.0;
961f1af5d2fSBarry Smith       for(i = 0; i < m; i++) {
962f1af5d2fSBarry Smith         y_l = m - i - 1.0;
963f1af5d2fSBarry Smith         y_r = y_l + 1.0;
964f1af5d2fSBarry Smith #if defined(PETSC_USE_COMPLEX)
965329f5518SBarry Smith         if (PetscRealPart(v[j*m+i]) >  0.) {
966b0a32e0cSBarry Smith           color = PETSC_DRAW_RED;
967329f5518SBarry Smith         } else if (PetscRealPart(v[j*m+i]) <  0.) {
968b0a32e0cSBarry Smith           color = PETSC_DRAW_BLUE;
969f1af5d2fSBarry Smith         } else {
970f1af5d2fSBarry Smith           continue;
971f1af5d2fSBarry Smith         }
972f1af5d2fSBarry Smith #else
973f1af5d2fSBarry Smith         if (v[j*m+i] >  0.) {
974b0a32e0cSBarry Smith           color = PETSC_DRAW_RED;
975f1af5d2fSBarry Smith         } else if (v[j*m+i] <  0.) {
976b0a32e0cSBarry Smith           color = PETSC_DRAW_BLUE;
977f1af5d2fSBarry Smith         } else {
978f1af5d2fSBarry Smith           continue;
979f1af5d2fSBarry Smith         }
980f1af5d2fSBarry Smith #endif
981b0a32e0cSBarry Smith         ierr = PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);CHKERRQ(ierr);
982f1af5d2fSBarry Smith       }
983f1af5d2fSBarry Smith     }
984f1af5d2fSBarry Smith   } else {
985f1af5d2fSBarry Smith     /* use contour shading to indicate magnitude of values */
986f1af5d2fSBarry Smith     /* first determine max of all nonzero values */
987f1af5d2fSBarry Smith     for(i = 0; i < m*n; i++) {
988f1af5d2fSBarry Smith       if (PetscAbsScalar(v[i]) > maxv) maxv = PetscAbsScalar(v[i]);
989f1af5d2fSBarry Smith     }
990b0a32e0cSBarry Smith     scale = (245.0 - PETSC_DRAW_BASIC_COLORS)/maxv;
991b0a32e0cSBarry Smith     ierr  = PetscDrawGetPopup(draw,&popup);CHKERRQ(ierr);
992b0a32e0cSBarry Smith     if (popup) {ierr  = PetscDrawScalePopup(popup,0.0,maxv);CHKERRQ(ierr);}
993f1af5d2fSBarry Smith     for(j = 0; j < n; j++) {
994f1af5d2fSBarry Smith       x_l = j;
995f1af5d2fSBarry Smith       x_r = x_l + 1.0;
996f1af5d2fSBarry Smith       for(i = 0; i < m; i++) {
997f1af5d2fSBarry Smith         y_l   = m - i - 1.0;
998f1af5d2fSBarry Smith         y_r   = y_l + 1.0;
999b0a32e0cSBarry Smith         color = PETSC_DRAW_BASIC_COLORS + (int)(scale*PetscAbsScalar(v[j*m+i]));
1000b0a32e0cSBarry Smith         ierr  = PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);CHKERRQ(ierr);
1001f1af5d2fSBarry Smith       }
1002f1af5d2fSBarry Smith     }
1003f1af5d2fSBarry Smith   }
1004f1af5d2fSBarry Smith   PetscFunctionReturn(0);
1005f1af5d2fSBarry Smith }
1006f1af5d2fSBarry Smith 
10074a2ae208SSatish Balay #undef __FUNCT__
10084a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqDense_Draw"
1009dfbe8321SBarry Smith PetscErrorCode MatView_SeqDense_Draw(Mat A,PetscViewer viewer)
1010f1af5d2fSBarry Smith {
1011b0a32e0cSBarry Smith   PetscDraw      draw;
1012f1af5d2fSBarry Smith   PetscTruth     isnull;
1013329f5518SBarry Smith   PetscReal      xr,yr,xl,yl,h,w;
1014dfbe8321SBarry Smith   PetscErrorCode ierr;
1015f1af5d2fSBarry Smith 
1016f1af5d2fSBarry Smith   PetscFunctionBegin;
1017b0a32e0cSBarry Smith   ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
1018b0a32e0cSBarry Smith   ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr);
1019abc0a331SBarry Smith   if (isnull) PetscFunctionReturn(0);
1020f1af5d2fSBarry Smith 
1021f1af5d2fSBarry Smith   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer);CHKERRQ(ierr);
1022899cda47SBarry Smith   xr  = A->cmap.n; yr = A->rmap.n; h = yr/10.0; w = xr/10.0;
1023f1af5d2fSBarry Smith   xr += w;    yr += h;  xl = -w;     yl = -h;
1024b0a32e0cSBarry Smith   ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr);
1025b0a32e0cSBarry Smith   ierr = PetscDrawZoom(draw,MatView_SeqDense_Draw_Zoom,A);CHKERRQ(ierr);
1026f1af5d2fSBarry Smith   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",PETSC_NULL);CHKERRQ(ierr);
1027f1af5d2fSBarry Smith   PetscFunctionReturn(0);
1028f1af5d2fSBarry Smith }
1029f1af5d2fSBarry Smith 
10304a2ae208SSatish Balay #undef __FUNCT__
10314a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqDense"
1032dfbe8321SBarry Smith PetscErrorCode MatView_SeqDense(Mat A,PetscViewer viewer)
1033932b0c3eSLois Curfman McInnes {
1034dfbe8321SBarry Smith   PetscErrorCode ierr;
10356805f65bSBarry Smith   PetscTruth     iascii,isbinary,isdraw;
1036932b0c3eSLois Curfman McInnes 
10373a40ed3dSBarry Smith   PetscFunctionBegin;
103832077d6dSBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr);
1039fb9695e5SSatish Balay   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_BINARY,&isbinary);CHKERRQ(ierr);
1040fb9695e5SSatish Balay   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_DRAW,&isdraw);CHKERRQ(ierr);
10410f5bd95cSBarry Smith 
1042c45a1595SBarry Smith   if (iascii) {
1043c45a1595SBarry Smith     ierr = MatView_SeqDense_ASCII(A,viewer);CHKERRQ(ierr);
10440f5bd95cSBarry Smith   } else if (isbinary) {
10453a40ed3dSBarry Smith     ierr = MatView_SeqDense_Binary(A,viewer);CHKERRQ(ierr);
1046f1af5d2fSBarry Smith   } else if (isdraw) {
1047f1af5d2fSBarry Smith     ierr = MatView_SeqDense_Draw(A,viewer);CHKERRQ(ierr);
10485cd90555SBarry Smith   } else {
1049958c9bccSBarry Smith     SETERRQ1(PETSC_ERR_SUP,"Viewer type %s not supported by dense matrix",((PetscObject)viewer)->type_name);
1050932b0c3eSLois Curfman McInnes   }
10513a40ed3dSBarry Smith   PetscFunctionReturn(0);
1052932b0c3eSLois Curfman McInnes }
1053289bc588SBarry Smith 
10544a2ae208SSatish Balay #undef __FUNCT__
10554a2ae208SSatish Balay #define __FUNCT__ "MatDestroy_SeqDense"
1056dfbe8321SBarry Smith PetscErrorCode MatDestroy_SeqDense(Mat mat)
1057289bc588SBarry Smith {
1058ec8511deSBarry Smith   Mat_SeqDense   *l = (Mat_SeqDense*)mat->data;
1059dfbe8321SBarry Smith   PetscErrorCode ierr;
106090f02eecSBarry Smith 
10613a40ed3dSBarry Smith   PetscFunctionBegin;
1062aa482453SBarry Smith #if defined(PETSC_USE_LOG)
1063899cda47SBarry Smith   PetscLogObjectState((PetscObject)mat,"Rows %D Cols %D",mat->rmap.n,mat->cmap.n);
1064a5a9c739SBarry Smith #endif
106505b42c5fSBarry Smith   ierr = PetscFree(l->pivots);CHKERRQ(ierr);
10666857c123SSatish Balay   if (!l->user_alloc) {ierr = PetscFree(l->v);CHKERRQ(ierr);}
1067606d414cSSatish Balay   ierr = PetscFree(l);CHKERRQ(ierr);
1068dbd8c25aSHong Zhang 
1069dbd8c25aSHong Zhang   ierr = PetscObjectChangeTypeName((PetscObject)mat,0);CHKERRQ(ierr);
1070901853e0SKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatSeqDenseSetPreallocation_C","",PETSC_NULL);CHKERRQ(ierr);
10714ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMult_seqaij_seqdense_C","",PETSC_NULL);CHKERRQ(ierr);
10724ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMultSymbolic_seqaij_seqdense_C","",PETSC_NULL);CHKERRQ(ierr);
10734ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMultNumeric_seqaij_seqdense_C","",PETSC_NULL);CHKERRQ(ierr);
10743a40ed3dSBarry Smith   PetscFunctionReturn(0);
1075289bc588SBarry Smith }
1076289bc588SBarry Smith 
10774a2ae208SSatish Balay #undef __FUNCT__
10784a2ae208SSatish Balay #define __FUNCT__ "MatTranspose_SeqDense"
1079dfbe8321SBarry Smith PetscErrorCode MatTranspose_SeqDense(Mat A,Mat *matout)
1080289bc588SBarry Smith {
1081c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
10826849ba73SBarry Smith   PetscErrorCode ierr;
108313f74950SBarry Smith   PetscInt       k,j,m,n,M;
108487828ca2SBarry Smith   PetscScalar    *v,tmp;
108548b35521SBarry Smith 
10863a40ed3dSBarry Smith   PetscFunctionBegin;
1087899cda47SBarry Smith   v = mat->v; m = A->rmap.n; M = mat->lda; n = A->cmap.n;
10887c922b88SBarry Smith   if (!matout) { /* in place transpose */
1089a5ce6ee0Svictorle     if (m != n) {
1090634064b4SBarry Smith       SETERRQ(PETSC_ERR_SUP,"Can not transpose non-square matrix in place");
1091d64ed03dSBarry Smith     } else {
1092d3e5ee88SLois Curfman McInnes       for (j=0; j<m; j++) {
1093289bc588SBarry Smith         for (k=0; k<j; k++) {
10941b807ce4Svictorle           tmp = v[j + k*M];
10951b807ce4Svictorle           v[j + k*M] = v[k + j*M];
10961b807ce4Svictorle           v[k + j*M] = tmp;
1097289bc588SBarry Smith         }
1098289bc588SBarry Smith       }
1099d64ed03dSBarry Smith     }
11003a40ed3dSBarry Smith   } else { /* out-of-place transpose */
1101d3e5ee88SLois Curfman McInnes     Mat          tmat;
1102ec8511deSBarry Smith     Mat_SeqDense *tmatd;
110387828ca2SBarry Smith     PetscScalar  *v2;
1104ea709b57SSatish Balay 
11057adad957SLisandro Dalcin     ierr  = MatCreate(((PetscObject)A)->comm,&tmat);CHKERRQ(ierr);
1106899cda47SBarry Smith     ierr  = MatSetSizes(tmat,A->cmap.n,A->rmap.n,A->cmap.n,A->rmap.n);CHKERRQ(ierr);
11077adad957SLisandro Dalcin     ierr  = MatSetType(tmat,((PetscObject)A)->type_name);CHKERRQ(ierr);
11085c5985e7SKris Buschelman     ierr  = MatSeqDenseSetPreallocation(tmat,PETSC_NULL);CHKERRQ(ierr);
1109ec8511deSBarry Smith     tmatd = (Mat_SeqDense*)tmat->data;
11100de55854SLois Curfman McInnes     v = mat->v; v2 = tmatd->v;
1111d3e5ee88SLois Curfman McInnes     for (j=0; j<n; j++) {
11121b807ce4Svictorle       for (k=0; k<m; k++) v2[j + k*n] = v[k + j*M];
1113d3e5ee88SLois Curfman McInnes     }
11146d4a8577SBarry Smith     ierr = MatAssemblyBegin(tmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
11156d4a8577SBarry Smith     ierr = MatAssemblyEnd(tmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
1116d3e5ee88SLois Curfman McInnes     *matout = tmat;
111748b35521SBarry Smith   }
11183a40ed3dSBarry Smith   PetscFunctionReturn(0);
1119289bc588SBarry Smith }
1120289bc588SBarry Smith 
11214a2ae208SSatish Balay #undef __FUNCT__
11224a2ae208SSatish Balay #define __FUNCT__ "MatEqual_SeqDense"
1123dfbe8321SBarry Smith PetscErrorCode MatEqual_SeqDense(Mat A1,Mat A2,PetscTruth *flg)
1124289bc588SBarry Smith {
1125c0bbcb79SLois Curfman McInnes   Mat_SeqDense *mat1 = (Mat_SeqDense*)A1->data;
1126c0bbcb79SLois Curfman McInnes   Mat_SeqDense *mat2 = (Mat_SeqDense*)A2->data;
112713f74950SBarry Smith   PetscInt     i,j;
112887828ca2SBarry Smith   PetscScalar  *v1 = mat1->v,*v2 = mat2->v;
11299ea5d5aeSSatish Balay 
11303a40ed3dSBarry Smith   PetscFunctionBegin;
1131899cda47SBarry Smith   if (A1->rmap.n != A2->rmap.n) {*flg = PETSC_FALSE; PetscFunctionReturn(0);}
1132899cda47SBarry Smith   if (A1->cmap.n != A2->cmap.n) {*flg = PETSC_FALSE; PetscFunctionReturn(0);}
1133899cda47SBarry Smith   for (i=0; i<A1->rmap.n; i++) {
11341b807ce4Svictorle     v1 = mat1->v+i; v2 = mat2->v+i;
1135899cda47SBarry Smith     for (j=0; j<A1->cmap.n; j++) {
11363a40ed3dSBarry Smith       if (*v1 != *v2) {*flg = PETSC_FALSE; PetscFunctionReturn(0);}
11371b807ce4Svictorle       v1 += mat1->lda; v2 += mat2->lda;
11381b807ce4Svictorle     }
1139289bc588SBarry Smith   }
114077c4ece6SBarry Smith   *flg = PETSC_TRUE;
11413a40ed3dSBarry Smith   PetscFunctionReturn(0);
1142289bc588SBarry Smith }
1143289bc588SBarry Smith 
11444a2ae208SSatish Balay #undef __FUNCT__
11454a2ae208SSatish Balay #define __FUNCT__ "MatGetDiagonal_SeqDense"
1146dfbe8321SBarry Smith PetscErrorCode MatGetDiagonal_SeqDense(Mat A,Vec v)
1147289bc588SBarry Smith {
1148c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
1149dfbe8321SBarry Smith   PetscErrorCode ierr;
115013f74950SBarry Smith   PetscInt       i,n,len;
115187828ca2SBarry Smith   PetscScalar    *x,zero = 0.0;
115244cd7ae7SLois Curfman McInnes 
11533a40ed3dSBarry Smith   PetscFunctionBegin;
11542dcb1b2aSMatthew Knepley   ierr = VecSet(v,zero);CHKERRQ(ierr);
11557a97a34bSBarry Smith   ierr = VecGetSize(v,&n);CHKERRQ(ierr);
11561ebc52fbSHong Zhang   ierr = VecGetArray(v,&x);CHKERRQ(ierr);
1157899cda47SBarry Smith   len = PetscMin(A->rmap.n,A->cmap.n);
1158899cda47SBarry Smith   if (n != A->rmap.n) SETERRQ(PETSC_ERR_ARG_SIZ,"Nonconforming mat and vec");
115944cd7ae7SLois Curfman McInnes   for (i=0; i<len; i++) {
11601b807ce4Svictorle     x[i] = mat->v[i*mat->lda + i];
1161289bc588SBarry Smith   }
11621ebc52fbSHong Zhang   ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
11633a40ed3dSBarry Smith   PetscFunctionReturn(0);
1164289bc588SBarry Smith }
1165289bc588SBarry Smith 
11664a2ae208SSatish Balay #undef __FUNCT__
11674a2ae208SSatish Balay #define __FUNCT__ "MatDiagonalScale_SeqDense"
1168dfbe8321SBarry Smith PetscErrorCode MatDiagonalScale_SeqDense(Mat A,Vec ll,Vec rr)
1169289bc588SBarry Smith {
1170c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
117187828ca2SBarry Smith   PetscScalar    *l,*r,x,*v;
1172dfbe8321SBarry Smith   PetscErrorCode ierr;
1173899cda47SBarry Smith   PetscInt       i,j,m = A->rmap.n,n = A->cmap.n;
117455659b69SBarry Smith 
11753a40ed3dSBarry Smith   PetscFunctionBegin;
117628988994SBarry Smith   if (ll) {
11777a97a34bSBarry Smith     ierr = VecGetSize(ll,&m);CHKERRQ(ierr);
11781ebc52fbSHong Zhang     ierr = VecGetArray(ll,&l);CHKERRQ(ierr);
1179899cda47SBarry Smith     if (m != A->rmap.n) SETERRQ(PETSC_ERR_ARG_SIZ,"Left scaling vec wrong size");
1180da3a660dSBarry Smith     for (i=0; i<m; i++) {
1181da3a660dSBarry Smith       x = l[i];
1182da3a660dSBarry Smith       v = mat->v + i;
1183da3a660dSBarry Smith       for (j=0; j<n; j++) { (*v) *= x; v+= m;}
1184da3a660dSBarry Smith     }
11851ebc52fbSHong Zhang     ierr = VecRestoreArray(ll,&l);CHKERRQ(ierr);
1186efee365bSSatish Balay     ierr = PetscLogFlops(n*m);CHKERRQ(ierr);
1187da3a660dSBarry Smith   }
118828988994SBarry Smith   if (rr) {
11897a97a34bSBarry Smith     ierr = VecGetSize(rr,&n);CHKERRQ(ierr);
11901ebc52fbSHong Zhang     ierr = VecGetArray(rr,&r);CHKERRQ(ierr);
1191899cda47SBarry Smith     if (n != A->cmap.n) SETERRQ(PETSC_ERR_ARG_SIZ,"Right scaling vec wrong size");
1192da3a660dSBarry Smith     for (i=0; i<n; i++) {
1193da3a660dSBarry Smith       x = r[i];
1194da3a660dSBarry Smith       v = mat->v + i*m;
1195da3a660dSBarry Smith       for (j=0; j<m; j++) { (*v++) *= x;}
1196da3a660dSBarry Smith     }
11971ebc52fbSHong Zhang     ierr = VecRestoreArray(rr,&r);CHKERRQ(ierr);
1198efee365bSSatish Balay     ierr = PetscLogFlops(n*m);CHKERRQ(ierr);
1199da3a660dSBarry Smith   }
12003a40ed3dSBarry Smith   PetscFunctionReturn(0);
1201289bc588SBarry Smith }
1202289bc588SBarry Smith 
12034a2ae208SSatish Balay #undef __FUNCT__
12044a2ae208SSatish Balay #define __FUNCT__ "MatNorm_SeqDense"
1205dfbe8321SBarry Smith PetscErrorCode MatNorm_SeqDense(Mat A,NormType type,PetscReal *nrm)
1206289bc588SBarry Smith {
1207c0bbcb79SLois Curfman McInnes   Mat_SeqDense *mat = (Mat_SeqDense*)A->data;
120887828ca2SBarry Smith   PetscScalar  *v = mat->v;
1209329f5518SBarry Smith   PetscReal    sum = 0.0;
1210899cda47SBarry Smith   PetscInt     lda=mat->lda,m=A->rmap.n,i,j;
1211efee365bSSatish Balay   PetscErrorCode ierr;
121255659b69SBarry Smith 
12133a40ed3dSBarry Smith   PetscFunctionBegin;
1214289bc588SBarry Smith   if (type == NORM_FROBENIUS) {
1215a5ce6ee0Svictorle     if (lda>m) {
1216899cda47SBarry Smith       for (j=0; j<A->cmap.n; j++) {
1217a5ce6ee0Svictorle 	v = mat->v+j*lda;
1218a5ce6ee0Svictorle 	for (i=0; i<m; i++) {
1219a5ce6ee0Svictorle #if defined(PETSC_USE_COMPLEX)
1220a5ce6ee0Svictorle 	  sum += PetscRealPart(PetscConj(*v)*(*v)); v++;
1221a5ce6ee0Svictorle #else
1222a5ce6ee0Svictorle 	  sum += (*v)*(*v); v++;
1223a5ce6ee0Svictorle #endif
1224a5ce6ee0Svictorle 	}
1225a5ce6ee0Svictorle       }
1226a5ce6ee0Svictorle     } else {
1227899cda47SBarry Smith       for (i=0; i<A->cmap.n*A->rmap.n; i++) {
1228aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX)
1229329f5518SBarry Smith 	sum += PetscRealPart(PetscConj(*v)*(*v)); v++;
1230289bc588SBarry Smith #else
1231289bc588SBarry Smith 	sum += (*v)*(*v); v++;
1232289bc588SBarry Smith #endif
1233289bc588SBarry Smith       }
1234a5ce6ee0Svictorle     }
1235064f8208SBarry Smith     *nrm = sqrt(sum);
1236899cda47SBarry Smith     ierr = PetscLogFlops(2*A->cmap.n*A->rmap.n);CHKERRQ(ierr);
12373a40ed3dSBarry Smith   } else if (type == NORM_1) {
1238064f8208SBarry Smith     *nrm = 0.0;
1239899cda47SBarry Smith     for (j=0; j<A->cmap.n; j++) {
12401b807ce4Svictorle       v = mat->v + j*mat->lda;
1241289bc588SBarry Smith       sum = 0.0;
1242899cda47SBarry Smith       for (i=0; i<A->rmap.n; i++) {
124333a8263dSBarry Smith         sum += PetscAbsScalar(*v);  v++;
1244289bc588SBarry Smith       }
1245064f8208SBarry Smith       if (sum > *nrm) *nrm = sum;
1246289bc588SBarry Smith     }
1247899cda47SBarry Smith     ierr = PetscLogFlops(A->cmap.n*A->rmap.n);CHKERRQ(ierr);
12483a40ed3dSBarry Smith   } else if (type == NORM_INFINITY) {
1249064f8208SBarry Smith     *nrm = 0.0;
1250899cda47SBarry Smith     for (j=0; j<A->rmap.n; j++) {
1251289bc588SBarry Smith       v = mat->v + j;
1252289bc588SBarry Smith       sum = 0.0;
1253899cda47SBarry Smith       for (i=0; i<A->cmap.n; i++) {
12541b807ce4Svictorle         sum += PetscAbsScalar(*v); v += mat->lda;
1255289bc588SBarry Smith       }
1256064f8208SBarry Smith       if (sum > *nrm) *nrm = sum;
1257289bc588SBarry Smith     }
1258899cda47SBarry Smith     ierr = PetscLogFlops(A->cmap.n*A->rmap.n);CHKERRQ(ierr);
12593a40ed3dSBarry Smith   } else {
126029bbc08cSBarry Smith     SETERRQ(PETSC_ERR_SUP,"No two norm");
1261289bc588SBarry Smith   }
12623a40ed3dSBarry Smith   PetscFunctionReturn(0);
1263289bc588SBarry Smith }
1264289bc588SBarry Smith 
12654a2ae208SSatish Balay #undef __FUNCT__
12664a2ae208SSatish Balay #define __FUNCT__ "MatSetOption_SeqDense"
12674e0d8c25SBarry Smith PetscErrorCode MatSetOption_SeqDense(Mat A,MatOption op,PetscTruth flg)
1268289bc588SBarry Smith {
1269c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *aij = (Mat_SeqDense*)A->data;
127063ba0a88SBarry Smith   PetscErrorCode ierr;
127167e560aaSBarry Smith 
12723a40ed3dSBarry Smith   PetscFunctionBegin;
1273b5a2b587SKris Buschelman   switch (op) {
1274b5a2b587SKris Buschelman   case MAT_ROW_ORIENTED:
12754e0d8c25SBarry Smith     aij->roworiented = flg;
1276b5a2b587SKris Buschelman     break;
1277512a5fc5SBarry Smith   case MAT_NEW_NONZERO_LOCATIONS:
1278b5a2b587SKris Buschelman   case MAT_NEW_NONZERO_LOCATION_ERR:
12793971808eSMatthew Knepley   case MAT_NEW_NONZERO_ALLOCATION_ERR:
12804e0d8c25SBarry Smith   case MAT_NEW_DIAGONALS:
1281b5a2b587SKris Buschelman   case MAT_IGNORE_OFF_PROC_ENTRIES:
1282b5a2b587SKris Buschelman   case MAT_USE_HASH_TABLE:
128377e54ba9SKris Buschelman   case MAT_SYMMETRIC:
128477e54ba9SKris Buschelman   case MAT_STRUCTURALLY_SYMMETRIC:
12859a4540c5SBarry Smith   case MAT_HERMITIAN:
12869a4540c5SBarry Smith   case MAT_SYMMETRY_ETERNAL:
1287290bbb0aSBarry Smith     ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr);
128877e54ba9SKris Buschelman     break;
1289b5a2b587SKris Buschelman   default:
1290ad86a440SBarry Smith     SETERRQ1(PETSC_ERR_SUP,"unknown option %d",op);
12913a40ed3dSBarry Smith   }
12923a40ed3dSBarry Smith   PetscFunctionReturn(0);
1293289bc588SBarry Smith }
1294289bc588SBarry Smith 
12954a2ae208SSatish Balay #undef __FUNCT__
12964a2ae208SSatish Balay #define __FUNCT__ "MatZeroEntries_SeqDense"
1297dfbe8321SBarry Smith PetscErrorCode MatZeroEntries_SeqDense(Mat A)
12986f0a148fSBarry Smith {
1299ec8511deSBarry Smith   Mat_SeqDense   *l = (Mat_SeqDense*)A->data;
13006849ba73SBarry Smith   PetscErrorCode ierr;
1301899cda47SBarry Smith   PetscInt       lda=l->lda,m=A->rmap.n,j;
13023a40ed3dSBarry Smith 
13033a40ed3dSBarry Smith   PetscFunctionBegin;
1304a5ce6ee0Svictorle   if (lda>m) {
1305899cda47SBarry Smith     for (j=0; j<A->cmap.n; j++) {
1306a5ce6ee0Svictorle       ierr = PetscMemzero(l->v+j*lda,m*sizeof(PetscScalar));CHKERRQ(ierr);
1307a5ce6ee0Svictorle     }
1308a5ce6ee0Svictorle   } else {
1309899cda47SBarry Smith     ierr = PetscMemzero(l->v,A->rmap.n*A->cmap.n*sizeof(PetscScalar));CHKERRQ(ierr);
1310a5ce6ee0Svictorle   }
13113a40ed3dSBarry Smith   PetscFunctionReturn(0);
13126f0a148fSBarry Smith }
13136f0a148fSBarry Smith 
13144a2ae208SSatish Balay #undef __FUNCT__
13154a2ae208SSatish Balay #define __FUNCT__ "MatZeroRows_SeqDense"
1316f4df32b1SMatthew Knepley PetscErrorCode MatZeroRows_SeqDense(Mat A,PetscInt N,const PetscInt rows[],PetscScalar diag)
13176f0a148fSBarry Smith {
1318ec8511deSBarry Smith   Mat_SeqDense   *l = (Mat_SeqDense*)A->data;
1319899cda47SBarry Smith   PetscInt       n = A->cmap.n,i,j;
132087828ca2SBarry Smith   PetscScalar    *slot;
132155659b69SBarry Smith 
13223a40ed3dSBarry Smith   PetscFunctionBegin;
13236f0a148fSBarry Smith   for (i=0; i<N; i++) {
13246f0a148fSBarry Smith     slot = l->v + rows[i];
13256f0a148fSBarry Smith     for (j=0; j<n; j++) { *slot = 0.0; slot += n;}
13266f0a148fSBarry Smith   }
1327f4df32b1SMatthew Knepley   if (diag != 0.0) {
13286f0a148fSBarry Smith     for (i=0; i<N; i++) {
13296f0a148fSBarry Smith       slot = l->v + (n+1)*rows[i];
1330f4df32b1SMatthew Knepley       *slot = diag;
13316f0a148fSBarry Smith     }
13326f0a148fSBarry Smith   }
13333a40ed3dSBarry Smith   PetscFunctionReturn(0);
13346f0a148fSBarry Smith }
1335557bce09SLois Curfman McInnes 
13364a2ae208SSatish Balay #undef __FUNCT__
13374a2ae208SSatish Balay #define __FUNCT__ "MatGetArray_SeqDense"
1338dfbe8321SBarry Smith PetscErrorCode MatGetArray_SeqDense(Mat A,PetscScalar *array[])
133964e87e97SBarry Smith {
1340c0bbcb79SLois Curfman McInnes   Mat_SeqDense *mat = (Mat_SeqDense*)A->data;
13413a40ed3dSBarry Smith 
13423a40ed3dSBarry Smith   PetscFunctionBegin;
1343899cda47SBarry Smith   if (mat->lda != A->rmap.n) SETERRQ(PETSC_ERR_SUP,"Cannot get array for Dense matrices with LDA different from number of rows");
134464e87e97SBarry Smith   *array = mat->v;
13453a40ed3dSBarry Smith   PetscFunctionReturn(0);
134664e87e97SBarry Smith }
13470754003eSLois Curfman McInnes 
13484a2ae208SSatish Balay #undef __FUNCT__
13494a2ae208SSatish Balay #define __FUNCT__ "MatRestoreArray_SeqDense"
1350dfbe8321SBarry Smith PetscErrorCode MatRestoreArray_SeqDense(Mat A,PetscScalar *array[])
1351ff14e315SSatish Balay {
13523a40ed3dSBarry Smith   PetscFunctionBegin;
135309b544d4SBarry Smith   *array = 0; /* user cannot accidently use the array later */
13543a40ed3dSBarry Smith   PetscFunctionReturn(0);
1355ff14e315SSatish Balay }
13560754003eSLois Curfman McInnes 
13574a2ae208SSatish Balay #undef __FUNCT__
13584a2ae208SSatish Balay #define __FUNCT__ "MatGetSubMatrix_SeqDense"
135913f74950SBarry Smith static PetscErrorCode MatGetSubMatrix_SeqDense(Mat A,IS isrow,IS iscol,PetscInt cs,MatReuse scall,Mat *B)
13600754003eSLois Curfman McInnes {
1361c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
13626849ba73SBarry Smith   PetscErrorCode ierr;
136321a2c019SBarry Smith   PetscInt       i,j,*irow,*icol,nrows,ncols;
136487828ca2SBarry Smith   PetscScalar    *av,*bv,*v = mat->v;
13650754003eSLois Curfman McInnes   Mat            newmat;
13660754003eSLois Curfman McInnes 
13673a40ed3dSBarry Smith   PetscFunctionBegin;
136878b31e54SBarry Smith   ierr = ISGetIndices(isrow,&irow);CHKERRQ(ierr);
136978b31e54SBarry Smith   ierr = ISGetIndices(iscol,&icol);CHKERRQ(ierr);
1370e03a110bSBarry Smith   ierr = ISGetLocalSize(isrow,&nrows);CHKERRQ(ierr);
1371e03a110bSBarry Smith   ierr = ISGetLocalSize(iscol,&ncols);CHKERRQ(ierr);
13720754003eSLois Curfman McInnes 
1373182d2002SSatish Balay   /* Check submatrixcall */
1374182d2002SSatish Balay   if (scall == MAT_REUSE_MATRIX) {
137513f74950SBarry Smith     PetscInt n_cols,n_rows;
1376182d2002SSatish Balay     ierr = MatGetSize(*B,&n_rows,&n_cols);CHKERRQ(ierr);
137721a2c019SBarry Smith     if (n_rows != nrows || n_cols != ncols) {
137821a2c019SBarry Smith       /* resize the result result matrix to match number of requested rows/columns */
137921a2c019SBarry Smith       ierr = MatSetSizes(*B,nrows,nrows,nrows,nrows);CHKERRQ(ierr);
138021a2c019SBarry Smith     }
1381182d2002SSatish Balay     newmat = *B;
1382182d2002SSatish Balay   } else {
13830754003eSLois Curfman McInnes     /* Create and fill new matrix */
13847adad957SLisandro Dalcin     ierr = MatCreate(((PetscObject)A)->comm,&newmat);CHKERRQ(ierr);
1385f69a0ea3SMatthew Knepley     ierr = MatSetSizes(newmat,nrows,ncols,nrows,ncols);CHKERRQ(ierr);
13867adad957SLisandro Dalcin     ierr = MatSetType(newmat,((PetscObject)A)->type_name);CHKERRQ(ierr);
13875c5985e7SKris Buschelman     ierr = MatSeqDenseSetPreallocation(newmat,PETSC_NULL);CHKERRQ(ierr);
1388182d2002SSatish Balay   }
1389182d2002SSatish Balay 
1390182d2002SSatish Balay   /* Now extract the data pointers and do the copy,column at a time */
1391182d2002SSatish Balay   bv = ((Mat_SeqDense*)newmat->data)->v;
1392182d2002SSatish Balay 
1393182d2002SSatish Balay   for (i=0; i<ncols; i++) {
13946de62eeeSBarry Smith     av = v + mat->lda*icol[i];
1395182d2002SSatish Balay     for (j=0; j<nrows; j++) {
1396182d2002SSatish Balay       *bv++ = av[irow[j]];
13970754003eSLois Curfman McInnes     }
13980754003eSLois Curfman McInnes   }
1399182d2002SSatish Balay 
1400182d2002SSatish Balay   /* Assemble the matrices so that the correct flags are set */
14016d4a8577SBarry Smith   ierr = MatAssemblyBegin(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
14026d4a8577SBarry Smith   ierr = MatAssemblyEnd(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
14030754003eSLois Curfman McInnes 
14040754003eSLois Curfman McInnes   /* Free work space */
140578b31e54SBarry Smith   ierr = ISRestoreIndices(isrow,&irow);CHKERRQ(ierr);
140678b31e54SBarry Smith   ierr = ISRestoreIndices(iscol,&icol);CHKERRQ(ierr);
1407182d2002SSatish Balay   *B = newmat;
14083a40ed3dSBarry Smith   PetscFunctionReturn(0);
14090754003eSLois Curfman McInnes }
14100754003eSLois Curfman McInnes 
14114a2ae208SSatish Balay #undef __FUNCT__
14124a2ae208SSatish Balay #define __FUNCT__ "MatGetSubMatrices_SeqDense"
141313f74950SBarry Smith PetscErrorCode MatGetSubMatrices_SeqDense(Mat A,PetscInt n,const IS irow[],const IS icol[],MatReuse scall,Mat *B[])
1414905e6a2fSBarry Smith {
14156849ba73SBarry Smith   PetscErrorCode ierr;
141613f74950SBarry Smith   PetscInt       i;
1417905e6a2fSBarry Smith 
14183a40ed3dSBarry Smith   PetscFunctionBegin;
1419905e6a2fSBarry Smith   if (scall == MAT_INITIAL_MATRIX) {
1420b0a32e0cSBarry Smith     ierr = PetscMalloc((n+1)*sizeof(Mat),B);CHKERRQ(ierr);
1421905e6a2fSBarry Smith   }
1422905e6a2fSBarry Smith 
1423905e6a2fSBarry Smith   for (i=0; i<n; i++) {
14246a6a5d1dSBarry Smith     ierr = MatGetSubMatrix_SeqDense(A,irow[i],icol[i],PETSC_DECIDE,scall,&(*B)[i]);CHKERRQ(ierr);
1425905e6a2fSBarry Smith   }
14263a40ed3dSBarry Smith   PetscFunctionReturn(0);
1427905e6a2fSBarry Smith }
1428905e6a2fSBarry Smith 
14294a2ae208SSatish Balay #undef __FUNCT__
1430c0aa2d19SHong Zhang #define __FUNCT__ "MatAssemblyBegin_SeqDense"
1431c0aa2d19SHong Zhang PetscErrorCode MatAssemblyBegin_SeqDense(Mat mat,MatAssemblyType mode)
1432c0aa2d19SHong Zhang {
1433c0aa2d19SHong Zhang   PetscFunctionBegin;
1434c0aa2d19SHong Zhang   PetscFunctionReturn(0);
1435c0aa2d19SHong Zhang }
1436c0aa2d19SHong Zhang 
1437c0aa2d19SHong Zhang #undef __FUNCT__
1438c0aa2d19SHong Zhang #define __FUNCT__ "MatAssemblyEnd_SeqDense"
1439c0aa2d19SHong Zhang PetscErrorCode MatAssemblyEnd_SeqDense(Mat mat,MatAssemblyType mode)
1440c0aa2d19SHong Zhang {
1441c0aa2d19SHong Zhang   PetscFunctionBegin;
1442c0aa2d19SHong Zhang   PetscFunctionReturn(0);
1443c0aa2d19SHong Zhang }
1444c0aa2d19SHong Zhang 
1445c0aa2d19SHong Zhang #undef __FUNCT__
14464a2ae208SSatish Balay #define __FUNCT__ "MatCopy_SeqDense"
1447dfbe8321SBarry Smith PetscErrorCode MatCopy_SeqDense(Mat A,Mat B,MatStructure str)
14484b0e389bSBarry Smith {
14494b0e389bSBarry Smith   Mat_SeqDense   *a = (Mat_SeqDense*)A->data,*b = (Mat_SeqDense *)B->data;
14506849ba73SBarry Smith   PetscErrorCode ierr;
1451899cda47SBarry Smith   PetscInt       lda1=a->lda,lda2=b->lda, m=A->rmap.n,n=A->cmap.n, j;
14523a40ed3dSBarry Smith 
14533a40ed3dSBarry Smith   PetscFunctionBegin;
145433f4a19fSKris Buschelman   /* If the two matrices don't have the same copy implementation, they aren't compatible for fast copy. */
145533f4a19fSKris Buschelman   if (A->ops->copy != B->ops->copy) {
1456cb5b572fSBarry Smith     ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr);
14573a40ed3dSBarry Smith     PetscFunctionReturn(0);
14583a40ed3dSBarry Smith   }
1459899cda47SBarry Smith   if (m != B->rmap.n || n != B->cmap.n) SETERRQ(PETSC_ERR_ARG_SIZ,"size(B) != size(A)");
1460a5ce6ee0Svictorle   if (lda1>m || lda2>m) {
14610dbb7854Svictorle     for (j=0; j<n; j++) {
1462a5ce6ee0Svictorle       ierr = PetscMemcpy(b->v+j*lda2,a->v+j*lda1,m*sizeof(PetscScalar));CHKERRQ(ierr);
1463a5ce6ee0Svictorle     }
1464a5ce6ee0Svictorle   } else {
1465899cda47SBarry Smith     ierr = PetscMemcpy(b->v,a->v,A->rmap.n*A->cmap.n*sizeof(PetscScalar));CHKERRQ(ierr);
1466a5ce6ee0Svictorle   }
1467273d9f13SBarry Smith   PetscFunctionReturn(0);
1468273d9f13SBarry Smith }
1469273d9f13SBarry Smith 
14704a2ae208SSatish Balay #undef __FUNCT__
14714a2ae208SSatish Balay #define __FUNCT__ "MatSetUpPreallocation_SeqDense"
1472dfbe8321SBarry Smith PetscErrorCode MatSetUpPreallocation_SeqDense(Mat A)
1473273d9f13SBarry Smith {
1474dfbe8321SBarry Smith   PetscErrorCode ierr;
1475273d9f13SBarry Smith 
1476273d9f13SBarry Smith   PetscFunctionBegin;
1477273d9f13SBarry Smith   ierr =  MatSeqDenseSetPreallocation(A,0);CHKERRQ(ierr);
14783a40ed3dSBarry Smith   PetscFunctionReturn(0);
14794b0e389bSBarry Smith }
14804b0e389bSBarry Smith 
1481284134d9SBarry Smith #undef __FUNCT__
1482284134d9SBarry Smith #define __FUNCT__ "MatSetSizes_SeqDense"
1483284134d9SBarry Smith PetscErrorCode MatSetSizes_SeqDense(Mat A,PetscInt m,PetscInt n,PetscInt M,PetscInt N)
1484284134d9SBarry Smith {
1485284134d9SBarry Smith   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
148621a2c019SBarry Smith   PetscErrorCode ierr;
1487284134d9SBarry Smith   PetscFunctionBegin;
148821a2c019SBarry Smith   /* this will not be called before lda, Mmax,  and Nmax have been set */
1489284134d9SBarry Smith   m = PetscMax(m,M);
1490284134d9SBarry Smith   n = PetscMax(n,N);
149121a2c019SBarry Smith   if (m > a->Mmax) SETERRQ2(PETSC_ERR_SUP,"Cannot yet resize number rows of dense matrix larger then its initial size %d, requested %d",a->lda,(int)m);
1492284134d9SBarry Smith   if (n > a->Nmax) SETERRQ2(PETSC_ERR_SUP,"Cannot yet resize number columns of dense matrix larger then its initial size %d, requested %d",a->Nmax,(int)n);
1493899cda47SBarry Smith   A->rmap.n = A->rmap.n = m;
1494899cda47SBarry Smith   A->cmap.n = A->cmap.N = n;
149521a2c019SBarry Smith   if (a->changelda) a->lda = m;
149621a2c019SBarry Smith   ierr = PetscMemzero(a->v,m*n*sizeof(PetscScalar));CHKERRQ(ierr);
1497284134d9SBarry Smith   PetscFunctionReturn(0);
1498284134d9SBarry Smith }
1499*170fe5c8SBarry Smith 
1500284134d9SBarry Smith 
1501a9fe9ddaSSatish Balay /* ----------------------------------------------------------------*/
1502a9fe9ddaSSatish Balay #undef __FUNCT__
1503a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMult_SeqDense_SeqDense"
1504a9fe9ddaSSatish Balay PetscErrorCode MatMatMult_SeqDense_SeqDense(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C)
1505a9fe9ddaSSatish Balay {
1506a9fe9ddaSSatish Balay   PetscErrorCode ierr;
1507a9fe9ddaSSatish Balay 
1508a9fe9ddaSSatish Balay   PetscFunctionBegin;
1509a9fe9ddaSSatish Balay   if (scall == MAT_INITIAL_MATRIX){
1510a9fe9ddaSSatish Balay     ierr = MatMatMultSymbolic_SeqDense_SeqDense(A,B,fill,C);CHKERRQ(ierr);
1511a9fe9ddaSSatish Balay   }
1512a9fe9ddaSSatish Balay   ierr = MatMatMultNumeric_SeqDense_SeqDense(A,B,*C);CHKERRQ(ierr);
1513a9fe9ddaSSatish Balay   PetscFunctionReturn(0);
1514a9fe9ddaSSatish Balay }
1515a9fe9ddaSSatish Balay 
1516a9fe9ddaSSatish Balay #undef __FUNCT__
1517a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMultSymbolic_SeqDense_SeqDense"
1518a9fe9ddaSSatish Balay PetscErrorCode MatMatMultSymbolic_SeqDense_SeqDense(Mat A,Mat B,PetscReal fill,Mat *C)
1519a9fe9ddaSSatish Balay {
1520ee16a9a1SHong Zhang   PetscErrorCode ierr;
1521899cda47SBarry Smith   PetscInt       m=A->rmap.n,n=B->cmap.n;
1522ee16a9a1SHong Zhang   Mat            Cmat;
1523a9fe9ddaSSatish Balay 
1524ee16a9a1SHong Zhang   PetscFunctionBegin;
1525899cda47SBarry Smith   if (A->cmap.n != B->rmap.n) SETERRQ2(PETSC_ERR_ARG_SIZ,"A->cmap.n %d != B->rmap.n %d\n",A->cmap.n,B->rmap.n);
1526ee16a9a1SHong Zhang   ierr = MatCreate(PETSC_COMM_SELF,&Cmat);CHKERRQ(ierr);
1527ee16a9a1SHong Zhang   ierr = MatSetSizes(Cmat,m,n,m,n);CHKERRQ(ierr);
1528ee16a9a1SHong Zhang   ierr = MatSetType(Cmat,MATSEQDENSE);CHKERRQ(ierr);
1529ee16a9a1SHong Zhang   ierr = MatSeqDenseSetPreallocation(Cmat,PETSC_NULL);CHKERRQ(ierr);
1530ee16a9a1SHong Zhang   Cmat->assembled = PETSC_TRUE;
1531ee16a9a1SHong Zhang   *C = Cmat;
1532ee16a9a1SHong Zhang   PetscFunctionReturn(0);
1533ee16a9a1SHong Zhang }
1534a9fe9ddaSSatish Balay 
1535a9fe9ddaSSatish Balay #undef __FUNCT__
1536a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMultNumeric_SeqDense_SeqDense"
1537a9fe9ddaSSatish Balay PetscErrorCode MatMatMultNumeric_SeqDense_SeqDense(Mat A,Mat B,Mat C)
1538a9fe9ddaSSatish Balay {
1539a9fe9ddaSSatish Balay   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
1540a9fe9ddaSSatish Balay   Mat_SeqDense   *b = (Mat_SeqDense*)B->data;
1541a9fe9ddaSSatish Balay   Mat_SeqDense   *c = (Mat_SeqDense*)C->data;
15420805154bSBarry Smith   PetscBLASInt   m,n,k;
1543a9fe9ddaSSatish Balay   PetscScalar    _DOne=1.0,_DZero=0.0;
1544a9fe9ddaSSatish Balay 
1545a9fe9ddaSSatish Balay   PetscFunctionBegin;
15460805154bSBarry Smith   m = PetscBLASIntCast(A->rmap.n);
15470805154bSBarry Smith   n = PetscBLASIntCast(B->cmap.n);
15480805154bSBarry Smith   k = PetscBLASIntCast(A->cmap.n);
1549a9fe9ddaSSatish Balay   BLASgemm_("N","N",&m,&n,&k,&_DOne,a->v,&a->lda,b->v,&b->lda,&_DZero,c->v,&c->lda);
1550a9fe9ddaSSatish Balay   PetscFunctionReturn(0);
1551a9fe9ddaSSatish Balay }
1552a9fe9ddaSSatish Balay 
1553a9fe9ddaSSatish Balay #undef __FUNCT__
1554a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMultTranspose_SeqDense_SeqDense"
1555a9fe9ddaSSatish Balay PetscErrorCode MatMatMultTranspose_SeqDense_SeqDense(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C)
1556a9fe9ddaSSatish Balay {
1557a9fe9ddaSSatish Balay   PetscErrorCode ierr;
1558a9fe9ddaSSatish Balay 
1559a9fe9ddaSSatish Balay   PetscFunctionBegin;
1560a9fe9ddaSSatish Balay   if (scall == MAT_INITIAL_MATRIX){
1561a9fe9ddaSSatish Balay     ierr = MatMatMultTransposeSymbolic_SeqDense_SeqDense(A,B,fill,C);CHKERRQ(ierr);
1562a9fe9ddaSSatish Balay   }
1563a9fe9ddaSSatish Balay   ierr = MatMatMultTransposeNumeric_SeqDense_SeqDense(A,B,*C);CHKERRQ(ierr);
1564a9fe9ddaSSatish Balay   PetscFunctionReturn(0);
1565a9fe9ddaSSatish Balay }
1566a9fe9ddaSSatish Balay 
1567a9fe9ddaSSatish Balay #undef __FUNCT__
1568a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMultTransposeSymbolic_SeqDense_SeqDense"
1569a9fe9ddaSSatish Balay PetscErrorCode MatMatMultTransposeSymbolic_SeqDense_SeqDense(Mat A,Mat B,PetscReal fill,Mat *C)
1570a9fe9ddaSSatish Balay {
1571ee16a9a1SHong Zhang   PetscErrorCode ierr;
1572899cda47SBarry Smith   PetscInt       m=A->cmap.n,n=B->cmap.n;
1573ee16a9a1SHong Zhang   Mat            Cmat;
1574a9fe9ddaSSatish Balay 
1575ee16a9a1SHong Zhang   PetscFunctionBegin;
1576899cda47SBarry Smith   if (A->rmap.n != B->rmap.n) SETERRQ2(PETSC_ERR_ARG_SIZ,"A->rmap.n %d != B->rmap.n %d\n",A->rmap.n,B->rmap.n);
1577ee16a9a1SHong Zhang   ierr = MatCreate(PETSC_COMM_SELF,&Cmat);CHKERRQ(ierr);
1578ee16a9a1SHong Zhang   ierr = MatSetSizes(Cmat,m,n,m,n);CHKERRQ(ierr);
1579ee16a9a1SHong Zhang   ierr = MatSetType(Cmat,MATSEQDENSE);CHKERRQ(ierr);
1580ee16a9a1SHong Zhang   ierr = MatSeqDenseSetPreallocation(Cmat,PETSC_NULL);CHKERRQ(ierr);
1581ee16a9a1SHong Zhang   Cmat->assembled = PETSC_TRUE;
1582ee16a9a1SHong Zhang   *C = Cmat;
1583ee16a9a1SHong Zhang   PetscFunctionReturn(0);
1584ee16a9a1SHong Zhang }
1585a9fe9ddaSSatish Balay 
1586a9fe9ddaSSatish Balay #undef __FUNCT__
1587a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMultTransposeNumeric_SeqDense_SeqDense"
1588a9fe9ddaSSatish Balay PetscErrorCode MatMatMultTransposeNumeric_SeqDense_SeqDense(Mat A,Mat B,Mat C)
1589a9fe9ddaSSatish Balay {
1590a9fe9ddaSSatish Balay   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
1591a9fe9ddaSSatish Balay   Mat_SeqDense   *b = (Mat_SeqDense*)B->data;
1592a9fe9ddaSSatish Balay   Mat_SeqDense   *c = (Mat_SeqDense*)C->data;
15930805154bSBarry Smith   PetscBLASInt   m,n,k;
1594a9fe9ddaSSatish Balay   PetscScalar    _DOne=1.0,_DZero=0.0;
1595a9fe9ddaSSatish Balay 
1596a9fe9ddaSSatish Balay   PetscFunctionBegin;
15970805154bSBarry Smith   m = PetscBLASIntCast(A->rmap.n);
15980805154bSBarry Smith   n = PetscBLASIntCast(B->cmap.n);
15990805154bSBarry Smith   k = PetscBLASIntCast(A->cmap.n);
1600a9fe9ddaSSatish Balay   BLASgemm_("T","N",&m,&n,&k,&_DOne,a->v,&a->lda,b->v,&b->lda,&_DZero,c->v,&c->lda);
1601a9fe9ddaSSatish Balay   PetscFunctionReturn(0);
1602a9fe9ddaSSatish Balay }
1603985db425SBarry Smith 
1604985db425SBarry Smith #undef __FUNCT__
1605985db425SBarry Smith #define __FUNCT__ "MatGetRowMax_SeqDense"
1606985db425SBarry Smith PetscErrorCode MatGetRowMax_SeqDense(Mat A,Vec v,PetscInt idx[])
1607985db425SBarry Smith {
1608985db425SBarry Smith   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
1609985db425SBarry Smith   PetscErrorCode ierr;
1610985db425SBarry Smith   PetscInt       i,j,m = A->rmap.n,n = A->cmap.n,p;
1611985db425SBarry Smith   PetscScalar    *x;
1612985db425SBarry Smith   MatScalar      *aa = a->v;
1613985db425SBarry Smith 
1614985db425SBarry Smith   PetscFunctionBegin;
1615985db425SBarry Smith   if (A->factor) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
1616985db425SBarry Smith 
1617985db425SBarry Smith   ierr = VecSet(v,0.0);CHKERRQ(ierr);
1618985db425SBarry Smith   ierr = VecGetArray(v,&x);CHKERRQ(ierr);
1619985db425SBarry Smith   ierr = VecGetLocalSize(v,&p);CHKERRQ(ierr);
1620985db425SBarry Smith   if (p != A->rmap.n) SETERRQ(PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector");
1621985db425SBarry Smith   for (i=0; i<m; i++) {
1622985db425SBarry Smith     x[i] = aa[i]; if (idx) idx[i] = 0;
1623985db425SBarry Smith     for (j=1; j<n; j++){
1624985db425SBarry Smith       if (PetscRealPart(x[i]) < PetscRealPart(aa[i+m*j])) {x[i] = aa[i + m*j]; if (idx) idx[i] = j;}
1625985db425SBarry Smith     }
1626985db425SBarry Smith   }
1627985db425SBarry Smith   ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
1628985db425SBarry Smith   PetscFunctionReturn(0);
1629985db425SBarry Smith }
1630985db425SBarry Smith 
1631985db425SBarry Smith #undef __FUNCT__
1632985db425SBarry Smith #define __FUNCT__ "MatGetRowMaxAbs_SeqDense"
1633985db425SBarry Smith PetscErrorCode MatGetRowMaxAbs_SeqDense(Mat A,Vec v,PetscInt idx[])
1634985db425SBarry Smith {
1635985db425SBarry Smith   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
1636985db425SBarry Smith   PetscErrorCode ierr;
1637985db425SBarry Smith   PetscInt       i,j,m = A->rmap.n,n = A->cmap.n,p;
1638985db425SBarry Smith   PetscScalar    *x;
1639985db425SBarry Smith   PetscReal      atmp;
1640985db425SBarry Smith   MatScalar      *aa = a->v;
1641985db425SBarry Smith 
1642985db425SBarry Smith   PetscFunctionBegin;
1643985db425SBarry Smith   if (A->factor) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
1644985db425SBarry Smith 
1645985db425SBarry Smith   ierr = VecSet(v,0.0);CHKERRQ(ierr);
1646985db425SBarry Smith   ierr = VecGetArray(v,&x);CHKERRQ(ierr);
1647985db425SBarry Smith   ierr = VecGetLocalSize(v,&p);CHKERRQ(ierr);
1648985db425SBarry Smith   if (p != A->rmap.n) SETERRQ(PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector");
1649985db425SBarry Smith   for (i=0; i<m; i++) {
1650985db425SBarry Smith     x[i] = PetscAbsScalar(aa[i]); if (idx) idx[i] = 0;
1651985db425SBarry Smith     for (j=1; j<n; j++){
1652985db425SBarry Smith       atmp = PetscAbsScalar(aa[i+m*j]);
1653985db425SBarry Smith       if (PetscAbsScalar(x[i]) < atmp) {x[i] = atmp; if (idx) idx[i] = j;}
1654985db425SBarry Smith     }
1655985db425SBarry Smith   }
1656985db425SBarry Smith   ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
1657985db425SBarry Smith   PetscFunctionReturn(0);
1658985db425SBarry Smith }
1659985db425SBarry Smith 
1660985db425SBarry Smith #undef __FUNCT__
1661985db425SBarry Smith #define __FUNCT__ "MatGetRowMin_SeqDense"
1662985db425SBarry Smith PetscErrorCode MatGetRowMin_SeqDense(Mat A,Vec v,PetscInt idx[])
1663985db425SBarry Smith {
1664985db425SBarry Smith   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
1665985db425SBarry Smith   PetscErrorCode ierr;
1666985db425SBarry Smith   PetscInt       i,j,m = A->rmap.n,n = A->cmap.n,p;
1667985db425SBarry Smith   PetscScalar    *x;
1668985db425SBarry Smith   MatScalar      *aa = a->v;
1669985db425SBarry Smith 
1670985db425SBarry Smith   PetscFunctionBegin;
1671985db425SBarry Smith   if (A->factor) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
1672985db425SBarry Smith 
1673985db425SBarry Smith   ierr = VecSet(v,0.0);CHKERRQ(ierr);
1674985db425SBarry Smith   ierr = VecGetArray(v,&x);CHKERRQ(ierr);
1675985db425SBarry Smith   ierr = VecGetLocalSize(v,&p);CHKERRQ(ierr);
1676985db425SBarry Smith   if (p != A->rmap.n) SETERRQ(PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector");
1677985db425SBarry Smith   for (i=0; i<m; i++) {
1678985db425SBarry Smith     x[i] = aa[i]; if (idx) idx[i] = 0;
1679985db425SBarry Smith     for (j=1; j<n; j++){
1680985db425SBarry Smith       if (PetscRealPart(x[i]) > PetscRealPart(aa[i+m*j])) {x[i] = aa[i + m*j]; if (idx) idx[i] = j;}
1681985db425SBarry Smith     }
1682985db425SBarry Smith   }
1683985db425SBarry Smith   ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
1684985db425SBarry Smith   PetscFunctionReturn(0);
1685985db425SBarry Smith }
1686985db425SBarry Smith 
16878d0534beSBarry Smith #undef __FUNCT__
16888d0534beSBarry Smith #define __FUNCT__ "MatGetColumnVector_SeqDense"
16898d0534beSBarry Smith PetscErrorCode MatGetColumnVector_SeqDense(Mat A,Vec v,PetscInt col)
16908d0534beSBarry Smith {
16918d0534beSBarry Smith   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
16928d0534beSBarry Smith   PetscErrorCode ierr;
16938d0534beSBarry Smith   PetscScalar    *x;
16948d0534beSBarry Smith 
16958d0534beSBarry Smith   PetscFunctionBegin;
16968d0534beSBarry Smith   if (A->factor) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
16978d0534beSBarry Smith 
16988d0534beSBarry Smith   ierr = VecGetArray(v,&x);CHKERRQ(ierr);
16998d0534beSBarry Smith   ierr = PetscMemcpy(x,a->v+col*a->lda,A->rmap.n*sizeof(PetscScalar));CHKERRQ(ierr);
17008d0534beSBarry Smith   ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
17018d0534beSBarry Smith   PetscFunctionReturn(0);
17028d0534beSBarry Smith }
17038d0534beSBarry Smith 
1704289bc588SBarry Smith /* -------------------------------------------------------------------*/
1705a5ae1ecdSBarry Smith static struct _MatOps MatOps_Values = {MatSetValues_SeqDense,
1706905e6a2fSBarry Smith        MatGetRow_SeqDense,
1707905e6a2fSBarry Smith        MatRestoreRow_SeqDense,
1708905e6a2fSBarry Smith        MatMult_SeqDense,
170997304618SKris Buschelman /* 4*/ MatMultAdd_SeqDense,
17107c922b88SBarry Smith        MatMultTranspose_SeqDense,
17117c922b88SBarry Smith        MatMultTransposeAdd_SeqDense,
1712905e6a2fSBarry Smith        MatSolve_SeqDense,
1713905e6a2fSBarry Smith        MatSolveAdd_SeqDense,
17147c922b88SBarry Smith        MatSolveTranspose_SeqDense,
171597304618SKris Buschelman /*10*/ MatSolveTransposeAdd_SeqDense,
1716905e6a2fSBarry Smith        MatLUFactor_SeqDense,
1717905e6a2fSBarry Smith        MatCholeskyFactor_SeqDense,
1718ec8511deSBarry Smith        MatRelax_SeqDense,
1719ec8511deSBarry Smith        MatTranspose_SeqDense,
172097304618SKris Buschelman /*15*/ MatGetInfo_SeqDense,
1721905e6a2fSBarry Smith        MatEqual_SeqDense,
1722905e6a2fSBarry Smith        MatGetDiagonal_SeqDense,
1723905e6a2fSBarry Smith        MatDiagonalScale_SeqDense,
1724905e6a2fSBarry Smith        MatNorm_SeqDense,
1725c0aa2d19SHong Zhang /*20*/ MatAssemblyBegin_SeqDense,
1726c0aa2d19SHong Zhang        MatAssemblyEnd_SeqDense,
1727905e6a2fSBarry Smith        0,
1728905e6a2fSBarry Smith        MatSetOption_SeqDense,
1729905e6a2fSBarry Smith        MatZeroEntries_SeqDense,
173097304618SKris Buschelman /*25*/ MatZeroRows_SeqDense,
1731905e6a2fSBarry Smith        MatLUFactorSymbolic_SeqDense,
1732905e6a2fSBarry Smith        MatLUFactorNumeric_SeqDense,
1733905e6a2fSBarry Smith        MatCholeskyFactorSymbolic_SeqDense,
1734905e6a2fSBarry Smith        MatCholeskyFactorNumeric_SeqDense,
173597304618SKris Buschelman /*30*/ MatSetUpPreallocation_SeqDense,
1736273d9f13SBarry Smith        0,
1737905e6a2fSBarry Smith        0,
1738905e6a2fSBarry Smith        MatGetArray_SeqDense,
1739905e6a2fSBarry Smith        MatRestoreArray_SeqDense,
174097304618SKris Buschelman /*35*/ MatDuplicate_SeqDense,
1741a5ae1ecdSBarry Smith        0,
1742a5ae1ecdSBarry Smith        0,
1743a5ae1ecdSBarry Smith        0,
1744a5ae1ecdSBarry Smith        0,
174597304618SKris Buschelman /*40*/ MatAXPY_SeqDense,
1746a5ae1ecdSBarry Smith        MatGetSubMatrices_SeqDense,
1747a5ae1ecdSBarry Smith        0,
17484b0e389bSBarry Smith        MatGetValues_SeqDense,
1749a5ae1ecdSBarry Smith        MatCopy_SeqDense,
1750985db425SBarry Smith /*45*/ MatGetRowMax_SeqDense,
1751a5ae1ecdSBarry Smith        MatScale_SeqDense,
1752a5ae1ecdSBarry Smith        0,
1753a5ae1ecdSBarry Smith        0,
1754a5ae1ecdSBarry Smith        0,
1755521d7252SBarry Smith /*50*/ 0,
1756a5ae1ecdSBarry Smith        0,
1757a5ae1ecdSBarry Smith        0,
1758a5ae1ecdSBarry Smith        0,
1759a5ae1ecdSBarry Smith        0,
176097304618SKris Buschelman /*55*/ 0,
1761a5ae1ecdSBarry Smith        0,
1762a5ae1ecdSBarry Smith        0,
1763a5ae1ecdSBarry Smith        0,
1764a5ae1ecdSBarry Smith        0,
176597304618SKris Buschelman /*60*/ 0,
1766e03a110bSBarry Smith        MatDestroy_SeqDense,
1767e03a110bSBarry Smith        MatView_SeqDense,
1768357abbc8SBarry Smith        0,
176997304618SKris Buschelman        0,
177097304618SKris Buschelman /*65*/ 0,
177197304618SKris Buschelman        0,
177297304618SKris Buschelman        0,
177397304618SKris Buschelman        0,
177497304618SKris Buschelman        0,
1775985db425SBarry Smith /*70*/ MatGetRowMaxAbs_SeqDense,
177697304618SKris Buschelman        0,
177797304618SKris Buschelman        0,
177897304618SKris Buschelman        0,
177997304618SKris Buschelman        0,
178097304618SKris Buschelman /*75*/ 0,
178197304618SKris Buschelman        0,
178297304618SKris Buschelman        0,
178397304618SKris Buschelman        0,
178497304618SKris Buschelman        0,
178597304618SKris Buschelman /*80*/ 0,
178697304618SKris Buschelman        0,
178797304618SKris Buschelman        0,
178897304618SKris Buschelman        0,
1789865e5f61SKris Buschelman /*84*/ MatLoad_SeqDense,
1790865e5f61SKris Buschelman        0,
17911cbb95d3SBarry Smith        MatIsHermitian_SeqDense,
1792865e5f61SKris Buschelman        0,
1793865e5f61SKris Buschelman        0,
1794865e5f61SKris Buschelman        0,
1795a9fe9ddaSSatish Balay /*90*/ MatMatMult_SeqDense_SeqDense,
1796a9fe9ddaSSatish Balay        MatMatMultSymbolic_SeqDense_SeqDense,
1797a9fe9ddaSSatish Balay        MatMatMultNumeric_SeqDense_SeqDense,
1798865e5f61SKris Buschelman        0,
1799865e5f61SKris Buschelman        0,
1800865e5f61SKris Buschelman /*95*/ 0,
1801a9fe9ddaSSatish Balay        MatMatMultTranspose_SeqDense_SeqDense,
1802a9fe9ddaSSatish Balay        MatMatMultTransposeSymbolic_SeqDense_SeqDense,
1803a9fe9ddaSSatish Balay        MatMatMultTransposeNumeric_SeqDense_SeqDense,
1804284134d9SBarry Smith        0,
1805284134d9SBarry Smith /*100*/0,
1806284134d9SBarry Smith        0,
1807284134d9SBarry Smith        0,
1808284134d9SBarry Smith        0,
1809985db425SBarry Smith        MatSetSizes_SeqDense,
1810985db425SBarry Smith        0,
1811985db425SBarry Smith        0,
1812985db425SBarry Smith        0,
1813985db425SBarry Smith        0,
1814985db425SBarry Smith        0,
1815985db425SBarry Smith /*110*/0,
1816985db425SBarry Smith        0,
18178d0534beSBarry Smith        MatGetRowMin_SeqDense,
18188d0534beSBarry Smith        MatGetColumnVector_SeqDense
1819985db425SBarry Smith };
182090ace30eSBarry Smith 
18214a2ae208SSatish Balay #undef __FUNCT__
18224a2ae208SSatish Balay #define __FUNCT__ "MatCreateSeqDense"
18234b828684SBarry Smith /*@C
1824fafbff53SBarry Smith    MatCreateSeqDense - Creates a sequential dense matrix that
1825d65003e9SLois Curfman McInnes    is stored in column major order (the usual Fortran 77 manner). Many
1826d65003e9SLois Curfman McInnes    of the matrix operations use the BLAS and LAPACK routines.
1827289bc588SBarry Smith 
1828db81eaa0SLois Curfman McInnes    Collective on MPI_Comm
1829db81eaa0SLois Curfman McInnes 
183020563c6bSBarry Smith    Input Parameters:
1831db81eaa0SLois Curfman McInnes +  comm - MPI communicator, set to PETSC_COMM_SELF
18320c775827SLois Curfman McInnes .  m - number of rows
183318f449edSLois Curfman McInnes .  n - number of columns
1834db81eaa0SLois Curfman McInnes -  data - optional location of matrix data.  Set data=PETSC_NULL for PETSc
1835dfc5480cSLois Curfman McInnes    to control all matrix memory allocation.
183620563c6bSBarry Smith 
183720563c6bSBarry Smith    Output Parameter:
183844cd7ae7SLois Curfman McInnes .  A - the matrix
183920563c6bSBarry Smith 
1840b259b22eSLois Curfman McInnes    Notes:
184118f449edSLois Curfman McInnes    The data input variable is intended primarily for Fortran programmers
184218f449edSLois Curfman McInnes    who wish to allocate their own matrix memory space.  Most users should
1843b4fd4287SBarry Smith    set data=PETSC_NULL.
184418f449edSLois Curfman McInnes 
1845027ccd11SLois Curfman McInnes    Level: intermediate
1846027ccd11SLois Curfman McInnes 
1847dbd7a890SLois Curfman McInnes .keywords: dense, matrix, LAPACK, BLAS
1848d65003e9SLois Curfman McInnes 
1849db81eaa0SLois Curfman McInnes .seealso: MatCreate(), MatCreateMPIDense(), MatSetValues()
185020563c6bSBarry Smith @*/
1851be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreateSeqDense(MPI_Comm comm,PetscInt m,PetscInt n,PetscScalar *data,Mat *A)
1852289bc588SBarry Smith {
1853dfbe8321SBarry Smith   PetscErrorCode ierr;
18543b2fbd54SBarry Smith 
18553a40ed3dSBarry Smith   PetscFunctionBegin;
1856f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,A);CHKERRQ(ierr);
1857f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*A,m,n,m,n);CHKERRQ(ierr);
1858273d9f13SBarry Smith   ierr = MatSetType(*A,MATSEQDENSE);CHKERRQ(ierr);
1859273d9f13SBarry Smith   ierr = MatSeqDenseSetPreallocation(*A,data);CHKERRQ(ierr);
1860273d9f13SBarry Smith   PetscFunctionReturn(0);
1861273d9f13SBarry Smith }
1862273d9f13SBarry Smith 
18634a2ae208SSatish Balay #undef __FUNCT__
1864afc30d2aSLisandro Dalcin #define __FUNCT__ "MatSeqDenseSetPreallocation"
1865273d9f13SBarry Smith /*@C
1866273d9f13SBarry Smith    MatSeqDenseSetPreallocation - Sets the array used for storing the matrix elements
1867273d9f13SBarry Smith 
1868273d9f13SBarry Smith    Collective on MPI_Comm
1869273d9f13SBarry Smith 
1870273d9f13SBarry Smith    Input Parameters:
1871273d9f13SBarry Smith +  A - the matrix
1872273d9f13SBarry Smith -  data - the array (or PETSC_NULL)
1873273d9f13SBarry Smith 
1874273d9f13SBarry Smith    Notes:
1875273d9f13SBarry Smith    The data input variable is intended primarily for Fortran programmers
1876273d9f13SBarry Smith    who wish to allocate their own matrix memory space.  Most users should
1877284134d9SBarry Smith    need not call this routine.
1878273d9f13SBarry Smith 
1879273d9f13SBarry Smith    Level: intermediate
1880273d9f13SBarry Smith 
1881273d9f13SBarry Smith .keywords: dense, matrix, LAPACK, BLAS
1882273d9f13SBarry Smith 
1883273d9f13SBarry Smith .seealso: MatCreate(), MatCreateMPIDense(), MatSetValues()
1884273d9f13SBarry Smith @*/
1885be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatSeqDenseSetPreallocation(Mat B,PetscScalar data[])
1886273d9f13SBarry Smith {
1887dfbe8321SBarry Smith   PetscErrorCode ierr,(*f)(Mat,PetscScalar[]);
1888a23d5eceSKris Buschelman 
1889a23d5eceSKris Buschelman   PetscFunctionBegin;
1890a23d5eceSKris Buschelman   ierr = PetscObjectQueryFunction((PetscObject)B,"MatSeqDenseSetPreallocation_C",(void (**)(void))&f);CHKERRQ(ierr);
1891a23d5eceSKris Buschelman   if (f) {
1892a23d5eceSKris Buschelman     ierr = (*f)(B,data);CHKERRQ(ierr);
1893a23d5eceSKris Buschelman   }
1894a23d5eceSKris Buschelman   PetscFunctionReturn(0);
1895a23d5eceSKris Buschelman }
1896a23d5eceSKris Buschelman 
1897a23d5eceSKris Buschelman EXTERN_C_BEGIN
1898a23d5eceSKris Buschelman #undef __FUNCT__
1899afc30d2aSLisandro Dalcin #define __FUNCT__ "MatSeqDenseSetPreallocation_SeqDense"
1900be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatSeqDenseSetPreallocation_SeqDense(Mat B,PetscScalar *data)
1901a23d5eceSKris Buschelman {
1902273d9f13SBarry Smith   Mat_SeqDense   *b;
1903dfbe8321SBarry Smith   PetscErrorCode ierr;
1904273d9f13SBarry Smith 
1905273d9f13SBarry Smith   PetscFunctionBegin;
1906273d9f13SBarry Smith   B->preallocated = PETSC_TRUE;
1907273d9f13SBarry Smith   b               = (Mat_SeqDense*)B->data;
19089e8f95c4SLisandro Dalcin   if (!data) { /* petsc-allocated storage */
19099e8f95c4SLisandro Dalcin     if (!b->user_alloc) { ierr = PetscFree(b->v);CHKERRQ(ierr); }
1910284134d9SBarry Smith     ierr = PetscMalloc((b->lda*b->Nmax+1)*sizeof(PetscScalar),&b->v);CHKERRQ(ierr);
1911284134d9SBarry Smith     ierr = PetscMemzero(b->v,b->lda*b->Nmax*sizeof(PetscScalar));CHKERRQ(ierr);
1912284134d9SBarry Smith     ierr = PetscLogObjectMemory(B,b->lda*b->Nmax*sizeof(PetscScalar));CHKERRQ(ierr);
19139e8f95c4SLisandro Dalcin     b->user_alloc = PETSC_FALSE;
1914273d9f13SBarry Smith   } else { /* user-allocated storage */
19159e8f95c4SLisandro Dalcin     if (!b->user_alloc) { ierr = PetscFree(b->v);CHKERRQ(ierr); }
1916273d9f13SBarry Smith     b->v          = data;
1917273d9f13SBarry Smith     b->user_alloc = PETSC_TRUE;
1918273d9f13SBarry Smith   }
1919273d9f13SBarry Smith   PetscFunctionReturn(0);
1920273d9f13SBarry Smith }
1921a23d5eceSKris Buschelman EXTERN_C_END
1922273d9f13SBarry Smith 
19231b807ce4Svictorle #undef __FUNCT__
19241b807ce4Svictorle #define __FUNCT__ "MatSeqDenseSetLDA"
19251b807ce4Svictorle /*@C
19261b807ce4Svictorle   MatSeqDenseSetLDA - Declare the leading dimension of the user-provided array
19271b807ce4Svictorle 
19281b807ce4Svictorle   Input parameter:
19291b807ce4Svictorle + A - the matrix
19301b807ce4Svictorle - lda - the leading dimension
19311b807ce4Svictorle 
19321b807ce4Svictorle   Notes:
19331b807ce4Svictorle   This routine is to be used in conjunction with MatSeqDenseSetPreallocation;
19341b807ce4Svictorle   it asserts that the preallocation has a leading dimension (the LDA parameter
19351b807ce4Svictorle   of Blas and Lapack fame) larger than M, the first dimension of the matrix.
19361b807ce4Svictorle 
19371b807ce4Svictorle   Level: intermediate
19381b807ce4Svictorle 
19391b807ce4Svictorle .keywords: dense, matrix, LAPACK, BLAS
19401b807ce4Svictorle 
1941284134d9SBarry Smith .seealso: MatCreate(), MatCreateSeqDense(), MatSeqDenseSetPreallocation(), MatSetMaximumSize()
19421b807ce4Svictorle @*/
1943be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatSeqDenseSetLDA(Mat B,PetscInt lda)
19441b807ce4Svictorle {
19451b807ce4Svictorle   Mat_SeqDense *b = (Mat_SeqDense*)B->data;
194621a2c019SBarry Smith 
19471b807ce4Svictorle   PetscFunctionBegin;
1948899cda47SBarry Smith   if (lda < B->rmap.n) SETERRQ2(PETSC_ERR_ARG_SIZ,"LDA %D must be at least matrix dimension %D",lda,B->rmap.n);
19491b807ce4Svictorle   b->lda       = lda;
195021a2c019SBarry Smith   b->changelda = PETSC_FALSE;
195121a2c019SBarry Smith   b->Mmax      = PetscMax(b->Mmax,lda);
19521b807ce4Svictorle   PetscFunctionReturn(0);
19531b807ce4Svictorle }
19541b807ce4Svictorle 
19550bad9183SKris Buschelman /*MC
1956fafad747SKris Buschelman    MATSEQDENSE - MATSEQDENSE = "seqdense" - A matrix type to be used for sequential dense matrices.
19570bad9183SKris Buschelman 
19580bad9183SKris Buschelman    Options Database Keys:
19590bad9183SKris Buschelman . -mat_type seqdense - sets the matrix type to "seqdense" during a call to MatSetFromOptions()
19600bad9183SKris Buschelman 
19610bad9183SKris Buschelman   Level: beginner
19620bad9183SKris Buschelman 
196389665df3SBarry Smith .seealso: MatCreateSeqDense()
196489665df3SBarry Smith 
19650bad9183SKris Buschelman M*/
19660bad9183SKris Buschelman 
1967273d9f13SBarry Smith EXTERN_C_BEGIN
19684a2ae208SSatish Balay #undef __FUNCT__
19694a2ae208SSatish Balay #define __FUNCT__ "MatCreate_SeqDense"
1970be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreate_SeqDense(Mat B)
1971273d9f13SBarry Smith {
1972273d9f13SBarry Smith   Mat_SeqDense   *b;
1973dfbe8321SBarry Smith   PetscErrorCode ierr;
19747c334f02SBarry Smith   PetscMPIInt    size;
1975273d9f13SBarry Smith 
1976273d9f13SBarry Smith   PetscFunctionBegin;
19777adad957SLisandro Dalcin   ierr = MPI_Comm_size(((PetscObject)B)->comm,&size);CHKERRQ(ierr);
197829bbc08cSBarry Smith   if (size > 1) SETERRQ(PETSC_ERR_ARG_WRONG,"Comm must be of size 1");
197955659b69SBarry Smith 
1980899cda47SBarry Smith   B->rmap.bs = B->cmap.bs = 1;
19816148ca0dSBarry Smith   ierr = PetscMapSetUp(&B->rmap);CHKERRQ(ierr);
19826148ca0dSBarry Smith   ierr = PetscMapSetUp(&B->cmap);CHKERRQ(ierr);
1983273d9f13SBarry Smith 
198438f2d2fdSLisandro Dalcin   ierr            = PetscNewLog(B,Mat_SeqDense,&b);CHKERRQ(ierr);
1985549d3d68SSatish Balay   ierr            = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr);
198644cd7ae7SLois Curfman McInnes   B->factor       = 0;
198790f02eecSBarry Smith   B->mapping      = 0;
198844cd7ae7SLois Curfman McInnes   B->data         = (void*)b;
198918f449edSLois Curfman McInnes 
1990a5ae1ecdSBarry Smith 
199144cd7ae7SLois Curfman McInnes   b->pivots       = 0;
1992273d9f13SBarry Smith   b->roworiented  = PETSC_TRUE;
1993273d9f13SBarry Smith   b->v            = 0;
1994899cda47SBarry Smith   b->lda          = B->rmap.n;
199521a2c019SBarry Smith   b->changelda    = PETSC_FALSE;
1996899cda47SBarry Smith   b->Mmax         = B->rmap.n;
1997899cda47SBarry Smith   b->Nmax         = B->cmap.n;
19984e220ebcSLois Curfman McInnes 
1999a23d5eceSKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatSeqDenseSetPreallocation_C",
2000a23d5eceSKris Buschelman                                     "MatSeqDenseSetPreallocation_SeqDense",
2001a23d5eceSKris Buschelman                                      MatSeqDenseSetPreallocation_SeqDense);CHKERRQ(ierr);
20024ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatMatMult_seqaij_seqdense_C",
20034ae313f4SHong Zhang                                      "MatMatMult_SeqAIJ_SeqDense",
20044ae313f4SHong Zhang                                       MatMatMult_SeqAIJ_SeqDense);CHKERRQ(ierr);
20054ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatMatMultSymbolic_seqaij_seqdense_C",
20064ae313f4SHong Zhang                                      "MatMatMultSymbolic_SeqAIJ_SeqDense",
20074ae313f4SHong Zhang                                       MatMatMultSymbolic_SeqAIJ_SeqDense);CHKERRQ(ierr);
20084ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatMatMultNumeric_seqaij_seqdense_C",
20094ae313f4SHong Zhang                                      "MatMatMultNumeric_SeqAIJ_SeqDense",
20104ae313f4SHong Zhang                                       MatMatMultNumeric_SeqAIJ_SeqDense);CHKERRQ(ierr);
201117667f90SBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)B,MATSEQDENSE);CHKERRQ(ierr);
20123a40ed3dSBarry Smith   PetscFunctionReturn(0);
2013289bc588SBarry Smith }
2014c0aa2d19SHong Zhang 
2015c0aa2d19SHong Zhang 
2016273d9f13SBarry Smith EXTERN_C_END
2017