xref: /petsc/src/mat/impls/dense/seq/dense.c (revision 9f4db6b06762a458a58be34e7fcbb95d38774e56)
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   }
1275c9eb25fSBarry Smith   A->factor = MAT_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 
137b24902e0SBarry Smith 
138b24902e0SBarry Smith #undef __FUNCT__
139b24902e0SBarry Smith #define __FUNCT__ "MatDuplicateNoCreate_SeqDense"
140b24902e0SBarry Smith PetscErrorCode MatDuplicateNoCreate_SeqDense(Mat A,MatDuplicateOption cpvalues,Mat *newmat)
141b24902e0SBarry Smith {
142b24902e0SBarry Smith   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data,*l;
143b24902e0SBarry Smith   PetscErrorCode ierr;
144b24902e0SBarry Smith   PetscInt       lda = (PetscInt)mat->lda,j,m;
145b24902e0SBarry Smith   Mat            newi = *newmat;
146b24902e0SBarry Smith 
147b24902e0SBarry Smith   PetscFunctionBegin;
1485c9eb25fSBarry Smith   ierr = MatSeqDenseSetPreallocation(*newmat,PETSC_NULL);CHKERRQ(ierr);
149b24902e0SBarry Smith   if (cpvalues == MAT_COPY_VALUES) {
150b24902e0SBarry Smith     l = (Mat_SeqDense*)newi->data;
151b24902e0SBarry Smith     if (lda>A->rmap.n) {
152b24902e0SBarry Smith       m = A->rmap.n;
153b24902e0SBarry Smith       for (j=0; j<A->cmap.n; j++) {
154b24902e0SBarry Smith 	ierr = PetscMemcpy(l->v+j*m,mat->v+j*lda,m*sizeof(PetscScalar));CHKERRQ(ierr);
155b24902e0SBarry Smith       }
156b24902e0SBarry Smith     } else {
157b24902e0SBarry Smith       ierr = PetscMemcpy(l->v,mat->v,A->rmap.n*A->cmap.n*sizeof(PetscScalar));CHKERRQ(ierr);
158b24902e0SBarry Smith     }
159b24902e0SBarry Smith   }
160b24902e0SBarry Smith   newi->assembled = PETSC_TRUE;
161b24902e0SBarry Smith   PetscFunctionReturn(0);
162b24902e0SBarry Smith }
163b24902e0SBarry Smith 
1644a2ae208SSatish Balay #undef __FUNCT__
1654a2ae208SSatish Balay #define __FUNCT__ "MatDuplicate_SeqDense"
166dfbe8321SBarry Smith PetscErrorCode MatDuplicate_SeqDense(Mat A,MatDuplicateOption cpvalues,Mat *newmat)
16702cad45dSBarry Smith {
1686849ba73SBarry Smith   PetscErrorCode ierr;
16902cad45dSBarry Smith 
1703a40ed3dSBarry Smith   PetscFunctionBegin;
1715c9eb25fSBarry Smith   ierr = MatCreate(((PetscObject)A)->comm,newmat);CHKERRQ(ierr);
1725c9eb25fSBarry Smith   ierr = MatSetSizes(*newmat,A->rmap.n,A->cmap.n,A->rmap.n,A->cmap.n);CHKERRQ(ierr);
1735c9eb25fSBarry Smith   ierr = MatSetType(*newmat,((PetscObject)A)->type_name);CHKERRQ(ierr);
174b24902e0SBarry Smith   ierr = MatDuplicateNoCreate_SeqDense(A,cpvalues,newmat);CHKERRQ(ierr);
175b24902e0SBarry Smith   PetscFunctionReturn(0);
176b24902e0SBarry Smith }
177b24902e0SBarry Smith 
178b24902e0SBarry Smith #undef __FUNCT__
179b24902e0SBarry Smith #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqDense"
180b24902e0SBarry Smith PetscErrorCode MatCholeskyFactorSymbolic_SeqDense(Mat A,IS row,MatFactorInfo *info,Mat *fact)
181b24902e0SBarry Smith {
182b24902e0SBarry Smith   PetscErrorCode ierr;
183b24902e0SBarry Smith 
184b24902e0SBarry Smith   PetscFunctionBegin;
1855c9eb25fSBarry Smith   ierr = MatDuplicateNoCreate_SeqDense(A,MAT_COPY_VALUES,fact);CHKERRQ(ierr);
1863a40ed3dSBarry Smith   PetscFunctionReturn(0);
18702cad45dSBarry Smith }
18802cad45dSBarry Smith 
1894a2ae208SSatish Balay #undef __FUNCT__
1904a2ae208SSatish Balay #define __FUNCT__ "MatLUFactorSymbolic_SeqDense"
191dfbe8321SBarry Smith PetscErrorCode MatLUFactorSymbolic_SeqDense(Mat A,IS row,IS col,MatFactorInfo *info,Mat *fact)
192289bc588SBarry Smith {
193dfbe8321SBarry Smith   PetscErrorCode ierr;
1943a40ed3dSBarry Smith 
1953a40ed3dSBarry Smith   PetscFunctionBegin;
196b24902e0SBarry Smith   ierr = MatDuplicateNoCreate_SeqDense(A,MAT_DO_NOT_COPY_VALUES,fact);CHKERRQ(ierr);
197b24902e0SBarry Smith   PetscFunctionReturn(0);
198b24902e0SBarry Smith }
199b24902e0SBarry Smith 
200b24902e0SBarry Smith #undef __FUNCT__
201b24902e0SBarry Smith #define __FUNCT__ "MatGetFactor_seqdense_petsc"
2025c9eb25fSBarry Smith PetscErrorCode MatGetFactor_seqdense_petsc(Mat A,MatFactorType ftype,Mat *fact)
203b24902e0SBarry Smith {
204b24902e0SBarry Smith   PetscErrorCode ierr;
205b24902e0SBarry Smith 
206b24902e0SBarry Smith   PetscFunctionBegin;
207b24902e0SBarry Smith   ierr = MatCreate(((PetscObject)A)->comm,fact);CHKERRQ(ierr);
208b24902e0SBarry Smith   ierr = MatSetSizes(*fact,A->rmap.n,A->cmap.n,A->rmap.n,A->cmap.n);CHKERRQ(ierr);
209b24902e0SBarry Smith   ierr = MatSetType(*fact,((PetscObject)A)->type_name);CHKERRQ(ierr);
210*9f4db6b0SBarry Smith   (*fact)->factor = ftype;
2113a40ed3dSBarry Smith   PetscFunctionReturn(0);
212289bc588SBarry Smith }
2136ee01492SSatish Balay 
2144a2ae208SSatish Balay #undef __FUNCT__
2154a2ae208SSatish Balay #define __FUNCT__ "MatLUFactorNumeric_SeqDense"
216af281ebdSHong Zhang PetscErrorCode MatLUFactorNumeric_SeqDense(Mat A,MatFactorInfo *info_dummy,Mat *fact)
217289bc588SBarry Smith {
21802cad45dSBarry Smith   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data,*l = (Mat_SeqDense*)(*fact)->data;
2196849ba73SBarry Smith   PetscErrorCode ierr;
220899cda47SBarry Smith   PetscInt       lda1=mat->lda,lda2=l->lda, m=A->rmap.n,n=A->cmap.n, j;
2214482741eSBarry Smith   MatFactorInfo  info;
2223a40ed3dSBarry Smith 
2233a40ed3dSBarry Smith   PetscFunctionBegin;
22402cad45dSBarry Smith   /* copy the numerical values */
2251b807ce4Svictorle   if (lda1>m || lda2>m ) {
2261b807ce4Svictorle     for (j=0; j<n; j++) {
2271b807ce4Svictorle       ierr = PetscMemcpy(l->v+j*lda2,mat->v+j*lda1,m*sizeof(PetscScalar));CHKERRQ(ierr);
2281b807ce4Svictorle     }
2291b807ce4Svictorle   } else {
230899cda47SBarry Smith     ierr = PetscMemcpy(l->v,mat->v,A->rmap.n*A->cmap.n*sizeof(PetscScalar));CHKERRQ(ierr);
2311b807ce4Svictorle   }
232*9f4db6b0SBarry Smith   (*fact)->factor = MAT_FACTOR_NONE;
2334482741eSBarry Smith   ierr = MatLUFactor(*fact,0,0,&info);CHKERRQ(ierr);
2343a40ed3dSBarry Smith   PetscFunctionReturn(0);
235289bc588SBarry Smith }
2366ee01492SSatish Balay 
2374a2ae208SSatish Balay #undef __FUNCT__
2384a2ae208SSatish Balay #define __FUNCT__ "MatCholeskyFactor_SeqDense"
239dfbe8321SBarry Smith PetscErrorCode MatCholeskyFactor_SeqDense(Mat A,IS perm,MatFactorInfo *factinfo)
240289bc588SBarry Smith {
241a07ab388SBarry Smith #if defined(PETSC_MISSING_LAPACK_POTRF)
242a07ab388SBarry Smith   PetscFunctionBegin;
243a07ab388SBarry Smith   SETERRQ(PETSC_ERR_SUP,"POTRF - Lapack routine is unavailable.");
244a07ab388SBarry Smith #else
245c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
2466849ba73SBarry Smith   PetscErrorCode ierr;
2470805154bSBarry Smith   PetscBLASInt   info,n = PetscBLASIntCast(A->cmap.n);
24855659b69SBarry Smith 
2493a40ed3dSBarry Smith   PetscFunctionBegin;
250606d414cSSatish Balay   ierr = PetscFree(mat->pivots);CHKERRQ(ierr);
251752f5567SLois Curfman McInnes   mat->pivots = 0;
25205b42c5fSBarry Smith 
253899cda47SBarry Smith   if (!A->rmap.n || !A->cmap.n) PetscFunctionReturn(0);
25471044d3cSBarry Smith   LAPACKpotrf_("L",&n,mat->v,&mat->lda,&info);
25577431f27SBarry Smith   if (info) SETERRQ1(PETSC_ERR_MAT_CH_ZRPVT,"Bad factorization: zero pivot in row %D",(PetscInt)info-1);
2565c9eb25fSBarry Smith   A->factor = MAT_FACTOR_CHOLESKY;
257899cda47SBarry Smith   ierr = PetscLogFlops((A->cmap.n*A->cmap.n*A->cmap.n)/3);CHKERRQ(ierr);
258a07ab388SBarry Smith #endif
2593a40ed3dSBarry Smith   PetscFunctionReturn(0);
260289bc588SBarry Smith }
261289bc588SBarry Smith 
2624a2ae208SSatish Balay #undef __FUNCT__
2630b4b3355SBarry Smith #define __FUNCT__ "MatCholeskyFactorNumeric_SeqDense"
264af281ebdSHong Zhang PetscErrorCode MatCholeskyFactorNumeric_SeqDense(Mat A,MatFactorInfo *info_dummy,Mat *fact)
2650b4b3355SBarry Smith {
266dfbe8321SBarry Smith   PetscErrorCode ierr;
26715e8a5b3SHong Zhang   MatFactorInfo  info;
2680b4b3355SBarry Smith 
2690b4b3355SBarry Smith   PetscFunctionBegin;
27015e8a5b3SHong Zhang   info.fill = 1.0;
27115e8a5b3SHong Zhang   ierr = MatCholeskyFactor_SeqDense(*fact,0,&info);CHKERRQ(ierr);
2720b4b3355SBarry Smith   PetscFunctionReturn(0);
2730b4b3355SBarry Smith }
2740b4b3355SBarry Smith 
2750b4b3355SBarry Smith #undef __FUNCT__
2764a2ae208SSatish Balay #define __FUNCT__ "MatSolve_SeqDense"
277dfbe8321SBarry Smith PetscErrorCode MatSolve_SeqDense(Mat A,Vec xx,Vec yy)
278289bc588SBarry Smith {
279c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
2806849ba73SBarry Smith   PetscErrorCode ierr;
28187828ca2SBarry Smith   PetscScalar    *x,*y;
2820805154bSBarry Smith   PetscBLASInt   one = 1,info,m = PetscBLASIntCast(A->rmap.n);
28367e560aaSBarry Smith 
2843a40ed3dSBarry Smith   PetscFunctionBegin;
2851ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
2861ebc52fbSHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
287899cda47SBarry Smith   ierr = PetscMemcpy(y,x,A->rmap.n*sizeof(PetscScalar));CHKERRQ(ierr);
2885c9eb25fSBarry Smith   if (A->factor == MAT_FACTOR_LU) {
289ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_GETRS)
29080444007SBarry Smith     SETERRQ(PETSC_ERR_SUP,"GETRS - Lapack routine is unavailable.");
291ae7cfcebSSatish Balay #else
29271044d3cSBarry Smith     LAPACKgetrs_("N",&m,&one,mat->v,&mat->lda,mat->pivots,y,&m,&info);
2934ce68768SBarry Smith     if (info) SETERRQ(PETSC_ERR_LIB,"GETRS - Bad solve");
294ae7cfcebSSatish Balay #endif
2955c9eb25fSBarry Smith   } else if (A->factor == MAT_FACTOR_CHOLESKY){
296ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_POTRS)
29780444007SBarry Smith     SETERRQ(PETSC_ERR_SUP,"POTRS - Lapack routine is unavailable.");
298ae7cfcebSSatish Balay #else
29971044d3cSBarry Smith     LAPACKpotrs_("L",&m,&one,mat->v,&mat->lda,y,&m,&info);
3004ce68768SBarry Smith     if (info) SETERRQ(PETSC_ERR_LIB,"POTRS Bad solve");
301ae7cfcebSSatish Balay #endif
302289bc588SBarry Smith   }
30329bbc08cSBarry Smith   else SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Matrix must be factored to solve");
3041ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
3051ebc52fbSHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
306899cda47SBarry Smith   ierr = PetscLogFlops(2*A->cmap.n*A->cmap.n - A->cmap.n);CHKERRQ(ierr);
3073a40ed3dSBarry Smith   PetscFunctionReturn(0);
308289bc588SBarry Smith }
3096ee01492SSatish Balay 
3104a2ae208SSatish Balay #undef __FUNCT__
3114a2ae208SSatish Balay #define __FUNCT__ "MatSolveTranspose_SeqDense"
312dfbe8321SBarry Smith PetscErrorCode MatSolveTranspose_SeqDense(Mat A,Vec xx,Vec yy)
313da3a660dSBarry Smith {
314c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
315dfbe8321SBarry Smith   PetscErrorCode ierr;
31687828ca2SBarry Smith   PetscScalar    *x,*y;
3170805154bSBarry Smith   PetscBLASInt   one = 1,info,m = PetscBLASIntCast(A->rmap.n);
31867e560aaSBarry Smith 
3193a40ed3dSBarry Smith   PetscFunctionBegin;
3201ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
3211ebc52fbSHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
322899cda47SBarry Smith   ierr = PetscMemcpy(y,x,A->rmap.n*sizeof(PetscScalar));CHKERRQ(ierr);
323752f5567SLois Curfman McInnes   /* assume if pivots exist then use LU; else Cholesky */
324da3a660dSBarry Smith   if (mat->pivots) {
325ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_GETRS)
32680444007SBarry Smith     SETERRQ(PETSC_ERR_SUP,"GETRS - Lapack routine is unavailable.");
327ae7cfcebSSatish Balay #else
32871044d3cSBarry Smith     LAPACKgetrs_("T",&m,&one,mat->v,&mat->lda,mat->pivots,y,&m,&info);
3294ce68768SBarry Smith     if (info) SETERRQ(PETSC_ERR_LIB,"POTRS - Bad solve");
330ae7cfcebSSatish Balay #endif
3317a97a34bSBarry Smith   } else {
332ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_POTRS)
33380444007SBarry Smith     SETERRQ(PETSC_ERR_SUP,"POTRS - Lapack routine is unavailable.");
334ae7cfcebSSatish Balay #else
33571044d3cSBarry Smith     LAPACKpotrs_("L",&m,&one,mat->v,&mat->lda,y,&m,&info);
3364ce68768SBarry Smith     if (info) SETERRQ(PETSC_ERR_LIB,"POTRS - Bad solve");
337ae7cfcebSSatish Balay #endif
338da3a660dSBarry Smith   }
3391ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
3401ebc52fbSHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
341899cda47SBarry Smith   ierr = PetscLogFlops(2*A->cmap.n*A->cmap.n - A->cmap.n);CHKERRQ(ierr);
3423a40ed3dSBarry Smith   PetscFunctionReturn(0);
343da3a660dSBarry Smith }
3446ee01492SSatish Balay 
3454a2ae208SSatish Balay #undef __FUNCT__
3464a2ae208SSatish Balay #define __FUNCT__ "MatSolveAdd_SeqDense"
347dfbe8321SBarry Smith PetscErrorCode MatSolveAdd_SeqDense(Mat A,Vec xx,Vec zz,Vec yy)
348da3a660dSBarry Smith {
349c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
350dfbe8321SBarry Smith   PetscErrorCode ierr;
35187828ca2SBarry Smith   PetscScalar    *x,*y,sone = 1.0;
352da3a660dSBarry Smith   Vec            tmp = 0;
3530805154bSBarry Smith   PetscBLASInt   one = 1,info,m = PetscBLASIntCast(A->rmap.n);
35467e560aaSBarry Smith 
3553a40ed3dSBarry Smith   PetscFunctionBegin;
3561ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
3571ebc52fbSHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
358899cda47SBarry Smith   if (!A->rmap.n || !A->cmap.n) PetscFunctionReturn(0);
359da3a660dSBarry Smith   if (yy == zz) {
36078b31e54SBarry Smith     ierr = VecDuplicate(yy,&tmp);CHKERRQ(ierr);
36152e6d16bSBarry Smith     ierr = PetscLogObjectParent(A,tmp);CHKERRQ(ierr);
36278b31e54SBarry Smith     ierr = VecCopy(yy,tmp);CHKERRQ(ierr);
363da3a660dSBarry Smith   }
364899cda47SBarry Smith   ierr = PetscMemcpy(y,x,A->rmap.n*sizeof(PetscScalar));CHKERRQ(ierr);
365752f5567SLois Curfman McInnes   /* assume if pivots exist then use LU; else Cholesky */
366da3a660dSBarry Smith   if (mat->pivots) {
367ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_GETRS)
36880444007SBarry Smith     SETERRQ(PETSC_ERR_SUP,"GETRS - Lapack routine is unavailable.");
369ae7cfcebSSatish Balay #else
37071044d3cSBarry Smith     LAPACKgetrs_("N",&m,&one,mat->v,&mat->lda,mat->pivots,y,&m,&info);
3712ffef20aSBarry Smith     if (info) SETERRQ(PETSC_ERR_LIB,"Bad solve");
372ae7cfcebSSatish Balay #endif
373a8c6a408SBarry Smith   } else {
374ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_POTRS)
37580444007SBarry Smith     SETERRQ(PETSC_ERR_SUP,"POTRS - Lapack routine is unavailable.");
376ae7cfcebSSatish Balay #else
37771044d3cSBarry Smith     LAPACKpotrs_("L",&m,&one,mat->v,&mat->lda,y,&m,&info);
3782ffef20aSBarry Smith     if (info) SETERRQ(PETSC_ERR_LIB,"Bad solve");
379ae7cfcebSSatish Balay #endif
380da3a660dSBarry Smith   }
3812dcb1b2aSMatthew Knepley   if (tmp) {ierr = VecAXPY(yy,sone,tmp);CHKERRQ(ierr); ierr = VecDestroy(tmp);CHKERRQ(ierr);}
3822dcb1b2aSMatthew Knepley   else     {ierr = VecAXPY(yy,sone,zz);CHKERRQ(ierr);}
3831ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
3841ebc52fbSHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
385899cda47SBarry Smith   ierr = PetscLogFlops(2*A->cmap.n*A->cmap.n);CHKERRQ(ierr);
3863a40ed3dSBarry Smith   PetscFunctionReturn(0);
387da3a660dSBarry Smith }
38867e560aaSBarry Smith 
3894a2ae208SSatish Balay #undef __FUNCT__
3904a2ae208SSatish Balay #define __FUNCT__ "MatSolveTransposeAdd_SeqDense"
391dfbe8321SBarry Smith PetscErrorCode MatSolveTransposeAdd_SeqDense(Mat A,Vec xx,Vec zz,Vec yy)
392da3a660dSBarry Smith {
393c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
3946849ba73SBarry Smith   PetscErrorCode ierr;
39587828ca2SBarry Smith   PetscScalar    *x,*y,sone = 1.0;
396da3a660dSBarry Smith   Vec            tmp;
3970805154bSBarry Smith   PetscBLASInt   one = 1,info,m = PetscBLASIntCast(A->rmap.n);
39867e560aaSBarry Smith 
3993a40ed3dSBarry Smith   PetscFunctionBegin;
400899cda47SBarry Smith   if (!A->rmap.n || !A->cmap.n) PetscFunctionReturn(0);
4011ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
4021ebc52fbSHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
403da3a660dSBarry Smith   if (yy == zz) {
40478b31e54SBarry Smith     ierr = VecDuplicate(yy,&tmp);CHKERRQ(ierr);
40552e6d16bSBarry Smith     ierr = PetscLogObjectParent(A,tmp);CHKERRQ(ierr);
40678b31e54SBarry Smith     ierr = VecCopy(yy,tmp);CHKERRQ(ierr);
407da3a660dSBarry Smith   }
408899cda47SBarry Smith   ierr = PetscMemcpy(y,x,A->rmap.n*sizeof(PetscScalar));CHKERRQ(ierr);
409752f5567SLois Curfman McInnes   /* assume if pivots exist then use LU; else Cholesky */
410da3a660dSBarry Smith   if (mat->pivots) {
411ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_GETRS)
41280444007SBarry Smith     SETERRQ(PETSC_ERR_SUP,"GETRS - Lapack routine is unavailable.");
413ae7cfcebSSatish Balay #else
41471044d3cSBarry Smith     LAPACKgetrs_("T",&m,&one,mat->v,&mat->lda,mat->pivots,y,&m,&info);
4152ffef20aSBarry Smith     if (info) SETERRQ(PETSC_ERR_LIB,"Bad solve");
416ae7cfcebSSatish Balay #endif
4173a40ed3dSBarry Smith   } else {
418ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_POTRS)
41980444007SBarry Smith     SETERRQ(PETSC_ERR_SUP,"POTRS - Lapack routine is unavailable.");
420ae7cfcebSSatish Balay #else
42171044d3cSBarry Smith     LAPACKpotrs_("L",&m,&one,mat->v,&mat->lda,y,&m,&info);
4222ffef20aSBarry Smith     if (info) SETERRQ(PETSC_ERR_LIB,"Bad solve");
423ae7cfcebSSatish Balay #endif
424da3a660dSBarry Smith   }
42590f02eecSBarry Smith   if (tmp) {
4262dcb1b2aSMatthew Knepley     ierr = VecAXPY(yy,sone,tmp);CHKERRQ(ierr);
42790f02eecSBarry Smith     ierr = VecDestroy(tmp);CHKERRQ(ierr);
4283a40ed3dSBarry Smith   } else {
4292dcb1b2aSMatthew Knepley     ierr = VecAXPY(yy,sone,zz);CHKERRQ(ierr);
43090f02eecSBarry Smith   }
4311ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
4321ebc52fbSHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
433899cda47SBarry Smith   ierr = PetscLogFlops(2*A->cmap.n*A->cmap.n);CHKERRQ(ierr);
4343a40ed3dSBarry Smith   PetscFunctionReturn(0);
435da3a660dSBarry Smith }
436289bc588SBarry Smith /* ------------------------------------------------------------------*/
4374a2ae208SSatish Balay #undef __FUNCT__
4384a2ae208SSatish Balay #define __FUNCT__ "MatRelax_SeqDense"
43913f74950SBarry Smith PetscErrorCode MatRelax_SeqDense(Mat A,Vec bb,PetscReal omega,MatSORType flag,PetscReal shift,PetscInt its,PetscInt lits,Vec xx)
440289bc588SBarry Smith {
441c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
44287828ca2SBarry Smith   PetscScalar    *x,*b,*v = mat->v,zero = 0.0,xt;
443dfbe8321SBarry Smith   PetscErrorCode ierr;
444899cda47SBarry Smith   PetscInt       m = A->rmap.n,i;
445aa482453SBarry Smith #if !defined(PETSC_USE_COMPLEX)
4460805154bSBarry Smith   PetscBLASInt   o = 1,bm = PetscBLASIntCast(m);
447bc1b551cSSatish Balay #endif
448289bc588SBarry Smith 
4493a40ed3dSBarry Smith   PetscFunctionBegin;
450289bc588SBarry Smith   if (flag & SOR_ZERO_INITIAL_GUESS) {
45171044d3cSBarry Smith     /* this is a hack fix, should have another version without the second BLASdot */
4522dcb1b2aSMatthew Knepley     ierr = VecSet(xx,zero);CHKERRQ(ierr);
453289bc588SBarry Smith   }
4541ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
4551ebc52fbSHong Zhang   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
456b965ef7fSBarry Smith   its  = its*lits;
45777431f27SBarry Smith   if (its <= 0) SETERRQ2(PETSC_ERR_ARG_WRONG,"Relaxation requires global its %D and local its %D both positive",its,lits);
458289bc588SBarry Smith   while (its--) {
459fccaa45eSBarry Smith     if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){
460289bc588SBarry Smith       for (i=0; i<m; i++) {
461aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX)
462f1747703SBarry Smith         /* cannot use BLAS dot for complex because compiler/linker is
463f1747703SBarry Smith            not happy about returning a double complex */
46413f74950SBarry Smith         PetscInt    _i;
46587828ca2SBarry Smith         PetscScalar sum = b[i];
466f1747703SBarry Smith         for (_i=0; _i<m; _i++) {
4673f6de6efSSatish Balay           sum -= PetscConj(v[i+_i*m])*x[_i];
468f1747703SBarry Smith         }
469f1747703SBarry Smith         xt = sum;
470f1747703SBarry Smith #else
47171044d3cSBarry Smith         xt = b[i] - BLASdot_(&bm,v+i,&bm,x,&o);
472f1747703SBarry Smith #endif
47355a1b374SBarry Smith         x[i] = (1. - omega)*x[i] + omega*(xt+v[i + i*m]*x[i])/(v[i + i*m]+shift);
474289bc588SBarry Smith       }
475289bc588SBarry Smith     }
476fccaa45eSBarry Smith     if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){
477289bc588SBarry Smith       for (i=m-1; i>=0; i--) {
478aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX)
479f1747703SBarry Smith         /* cannot use BLAS dot for complex because compiler/linker is
480f1747703SBarry Smith            not happy about returning a double complex */
48113f74950SBarry Smith         PetscInt    _i;
48287828ca2SBarry Smith         PetscScalar sum = b[i];
483f1747703SBarry Smith         for (_i=0; _i<m; _i++) {
4843f6de6efSSatish Balay           sum -= PetscConj(v[i+_i*m])*x[_i];
485f1747703SBarry Smith         }
486f1747703SBarry Smith         xt = sum;
487f1747703SBarry Smith #else
48871044d3cSBarry Smith         xt = b[i] - BLASdot_(&bm,v+i,&bm,x,&o);
489f1747703SBarry Smith #endif
49055a1b374SBarry Smith         x[i] = (1. - omega)*x[i] + omega*(xt+v[i + i*m]*x[i])/(v[i + i*m]+shift);
491289bc588SBarry Smith       }
492289bc588SBarry Smith     }
493289bc588SBarry Smith   }
4941ebc52fbSHong Zhang   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
4951ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
4963a40ed3dSBarry Smith   PetscFunctionReturn(0);
497289bc588SBarry Smith }
498289bc588SBarry Smith 
499289bc588SBarry Smith /* -----------------------------------------------------------------*/
5004a2ae208SSatish Balay #undef __FUNCT__
5014a2ae208SSatish Balay #define __FUNCT__ "MatMultTranspose_SeqDense"
502dfbe8321SBarry Smith PetscErrorCode MatMultTranspose_SeqDense(Mat A,Vec xx,Vec yy)
503289bc588SBarry Smith {
504c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
50587828ca2SBarry Smith   PetscScalar    *v = mat->v,*x,*y;
506dfbe8321SBarry Smith   PetscErrorCode ierr;
5070805154bSBarry Smith   PetscBLASInt   m, n,_One=1;
508ea709b57SSatish Balay   PetscScalar    _DOne=1.0,_DZero=0.0;
5093a40ed3dSBarry Smith 
5103a40ed3dSBarry Smith   PetscFunctionBegin;
5110805154bSBarry Smith   m = PetscBLASIntCast(A->rmap.n);
5120805154bSBarry Smith   n = PetscBLASIntCast(A->cmap.n);
513899cda47SBarry Smith   if (!A->rmap.n || !A->cmap.n) PetscFunctionReturn(0);
5141ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
5151ebc52fbSHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
51671044d3cSBarry Smith   BLASgemv_("T",&m,&n,&_DOne,v,&mat->lda,x,&_One,&_DZero,y,&_One);
5171ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
5181ebc52fbSHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
519899cda47SBarry Smith   ierr = PetscLogFlops(2*A->rmap.n*A->cmap.n - A->cmap.n);CHKERRQ(ierr);
5203a40ed3dSBarry Smith   PetscFunctionReturn(0);
521289bc588SBarry Smith }
5226ee01492SSatish Balay 
5234a2ae208SSatish Balay #undef __FUNCT__
5244a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqDense"
525dfbe8321SBarry Smith PetscErrorCode MatMult_SeqDense(Mat A,Vec xx,Vec yy)
526289bc588SBarry Smith {
527c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
52887828ca2SBarry Smith   PetscScalar    *v = mat->v,*x,*y,_DOne=1.0,_DZero=0.0;
529dfbe8321SBarry Smith   PetscErrorCode ierr;
5300805154bSBarry Smith   PetscBLASInt   m, n, _One=1;
5313a40ed3dSBarry Smith 
5323a40ed3dSBarry Smith   PetscFunctionBegin;
5330805154bSBarry Smith   m = PetscBLASIntCast(A->rmap.n);
5340805154bSBarry Smith   n = PetscBLASIntCast(A->cmap.n);
535899cda47SBarry Smith   if (!A->rmap.n || !A->cmap.n) PetscFunctionReturn(0);
5361ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
5371ebc52fbSHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
53871044d3cSBarry Smith   BLASgemv_("N",&m,&n,&_DOne,v,&(mat->lda),x,&_One,&_DZero,y,&_One);
5391ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
5401ebc52fbSHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
541899cda47SBarry Smith   ierr = PetscLogFlops(2*A->rmap.n*A->cmap.n - A->rmap.n);CHKERRQ(ierr);
5423a40ed3dSBarry Smith   PetscFunctionReturn(0);
543289bc588SBarry Smith }
5446ee01492SSatish Balay 
5454a2ae208SSatish Balay #undef __FUNCT__
5464a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqDense"
547dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqDense(Mat A,Vec xx,Vec zz,Vec yy)
548289bc588SBarry Smith {
549c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
55087828ca2SBarry Smith   PetscScalar    *v = mat->v,*x,*y,_DOne=1.0;
551dfbe8321SBarry Smith   PetscErrorCode ierr;
5520805154bSBarry Smith   PetscBLASInt   m, n, _One=1;
5533a40ed3dSBarry Smith 
5543a40ed3dSBarry Smith   PetscFunctionBegin;
5550805154bSBarry Smith   m = PetscBLASIntCast(A->rmap.n);
5560805154bSBarry Smith   n = PetscBLASIntCast(A->cmap.n);
557899cda47SBarry Smith   if (!A->rmap.n || !A->cmap.n) PetscFunctionReturn(0);
558600d6d0bSBarry Smith   if (zz != yy) {ierr = VecCopy(zz,yy);CHKERRQ(ierr);}
5591ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
5601ebc52fbSHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
56171044d3cSBarry Smith   BLASgemv_("N",&m,&n,&_DOne,v,&(mat->lda),x,&_One,&_DOne,y,&_One);
5621ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
5631ebc52fbSHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
564899cda47SBarry Smith   ierr = PetscLogFlops(2*A->rmap.n*A->cmap.n);CHKERRQ(ierr);
5653a40ed3dSBarry Smith   PetscFunctionReturn(0);
566289bc588SBarry Smith }
5676ee01492SSatish Balay 
5684a2ae208SSatish Balay #undef __FUNCT__
5694a2ae208SSatish Balay #define __FUNCT__ "MatMultTransposeAdd_SeqDense"
570dfbe8321SBarry Smith PetscErrorCode MatMultTransposeAdd_SeqDense(Mat A,Vec xx,Vec zz,Vec yy)
571289bc588SBarry Smith {
572c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
57387828ca2SBarry Smith   PetscScalar    *v = mat->v,*x,*y;
574dfbe8321SBarry Smith   PetscErrorCode ierr;
5750805154bSBarry Smith   PetscBLASInt   m, n, _One=1;
57687828ca2SBarry Smith   PetscScalar    _DOne=1.0;
5773a40ed3dSBarry Smith 
5783a40ed3dSBarry Smith   PetscFunctionBegin;
5790805154bSBarry Smith   m = PetscBLASIntCast(A->rmap.n);
5800805154bSBarry Smith   n = PetscBLASIntCast(A->cmap.n);
581899cda47SBarry Smith   if (!A->rmap.n || !A->cmap.n) PetscFunctionReturn(0);
582600d6d0bSBarry Smith   if (zz != yy) {ierr = VecCopy(zz,yy);CHKERRQ(ierr);}
5831ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
5841ebc52fbSHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
58571044d3cSBarry Smith   BLASgemv_("T",&m,&n,&_DOne,v,&(mat->lda),x,&_One,&_DOne,y,&_One);
5861ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
5871ebc52fbSHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
588899cda47SBarry Smith   ierr = PetscLogFlops(2*A->rmap.n*A->cmap.n);CHKERRQ(ierr);
5893a40ed3dSBarry Smith   PetscFunctionReturn(0);
590289bc588SBarry Smith }
591289bc588SBarry Smith 
592289bc588SBarry Smith /* -----------------------------------------------------------------*/
5934a2ae208SSatish Balay #undef __FUNCT__
5944a2ae208SSatish Balay #define __FUNCT__ "MatGetRow_SeqDense"
59513f74950SBarry Smith PetscErrorCode MatGetRow_SeqDense(Mat A,PetscInt row,PetscInt *ncols,PetscInt **cols,PetscScalar **vals)
596289bc588SBarry Smith {
597c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
59887828ca2SBarry Smith   PetscScalar    *v;
5996849ba73SBarry Smith   PetscErrorCode ierr;
60013f74950SBarry Smith   PetscInt       i;
60167e560aaSBarry Smith 
6023a40ed3dSBarry Smith   PetscFunctionBegin;
603899cda47SBarry Smith   *ncols = A->cmap.n;
604289bc588SBarry Smith   if (cols) {
605899cda47SBarry Smith     ierr = PetscMalloc((A->cmap.n+1)*sizeof(PetscInt),cols);CHKERRQ(ierr);
606899cda47SBarry Smith     for (i=0; i<A->cmap.n; i++) (*cols)[i] = i;
607289bc588SBarry Smith   }
608289bc588SBarry Smith   if (vals) {
609899cda47SBarry Smith     ierr = PetscMalloc((A->cmap.n+1)*sizeof(PetscScalar),vals);CHKERRQ(ierr);
610289bc588SBarry Smith     v    = mat->v + row;
611899cda47SBarry Smith     for (i=0; i<A->cmap.n; i++) {(*vals)[i] = *v; v += mat->lda;}
612289bc588SBarry Smith   }
6133a40ed3dSBarry Smith   PetscFunctionReturn(0);
614289bc588SBarry Smith }
6156ee01492SSatish Balay 
6164a2ae208SSatish Balay #undef __FUNCT__
6174a2ae208SSatish Balay #define __FUNCT__ "MatRestoreRow_SeqDense"
61813f74950SBarry Smith PetscErrorCode MatRestoreRow_SeqDense(Mat A,PetscInt row,PetscInt *ncols,PetscInt **cols,PetscScalar **vals)
619289bc588SBarry Smith {
620dfbe8321SBarry Smith   PetscErrorCode ierr;
621606d414cSSatish Balay   PetscFunctionBegin;
622606d414cSSatish Balay   if (cols) {ierr = PetscFree(*cols);CHKERRQ(ierr);}
623606d414cSSatish Balay   if (vals) {ierr = PetscFree(*vals);CHKERRQ(ierr); }
6243a40ed3dSBarry Smith   PetscFunctionReturn(0);
625289bc588SBarry Smith }
626289bc588SBarry Smith /* ----------------------------------------------------------------*/
6274a2ae208SSatish Balay #undef __FUNCT__
6284a2ae208SSatish Balay #define __FUNCT__ "MatSetValues_SeqDense"
62913f74950SBarry Smith PetscErrorCode MatSetValues_SeqDense(Mat A,PetscInt m,const PetscInt indexm[],PetscInt n,const PetscInt indexn[],const PetscScalar v[],InsertMode addv)
630289bc588SBarry Smith {
631c0bbcb79SLois Curfman McInnes   Mat_SeqDense *mat = (Mat_SeqDense*)A->data;
63213f74950SBarry Smith   PetscInt     i,j,idx=0;
633d6dfbf8fSBarry Smith 
6343a40ed3dSBarry Smith   PetscFunctionBegin;
635289bc588SBarry Smith   if (!mat->roworiented) {
636dbb450caSBarry Smith     if (addv == INSERT_VALUES) {
637289bc588SBarry Smith       for (j=0; j<n; j++) {
638cddbea37SSatish Balay         if (indexn[j] < 0) {idx += m; continue;}
6392515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
640899cda47SBarry 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);
64158804f6dSBarry Smith #endif
642289bc588SBarry Smith         for (i=0; i<m; i++) {
643cddbea37SSatish Balay           if (indexm[i] < 0) {idx++; 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
647cddbea37SSatish Balay           mat->v[indexn[j]*mat->lda + indexm[i]] = v[idx++];
648289bc588SBarry Smith         }
649289bc588SBarry Smith       }
6503a40ed3dSBarry Smith     } else {
651289bc588SBarry Smith       for (j=0; j<n; j++) {
652cddbea37SSatish Balay         if (indexn[j] < 0) {idx += m; continue;}
6532515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
654899cda47SBarry 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);
65558804f6dSBarry Smith #endif
656289bc588SBarry Smith         for (i=0; i<m; i++) {
657cddbea37SSatish Balay           if (indexm[i] < 0) {idx++; 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
661cddbea37SSatish Balay           mat->v[indexn[j]*mat->lda + indexm[i]] += v[idx++];
662289bc588SBarry Smith         }
663289bc588SBarry Smith       }
664289bc588SBarry Smith     }
6653a40ed3dSBarry Smith   } else {
666dbb450caSBarry Smith     if (addv == INSERT_VALUES) {
667e8d4e0b9SBarry Smith       for (i=0; i<m; i++) {
668cddbea37SSatish Balay         if (indexm[i] < 0) { idx += n; continue;}
6692515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
670899cda47SBarry 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);
67158804f6dSBarry Smith #endif
672e8d4e0b9SBarry Smith         for (j=0; j<n; j++) {
673cddbea37SSatish Balay           if (indexn[j] < 0) { idx++; continue;}
6742515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
675899cda47SBarry 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);
67658804f6dSBarry Smith #endif
677cddbea37SSatish Balay           mat->v[indexn[j]*mat->lda + indexm[i]] = v[idx++];
678e8d4e0b9SBarry Smith         }
679e8d4e0b9SBarry Smith       }
6803a40ed3dSBarry Smith     } else {
681289bc588SBarry Smith       for (i=0; i<m; i++) {
682cddbea37SSatish Balay         if (indexm[i] < 0) { idx += n; continue;}
6832515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
684899cda47SBarry 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);
68558804f6dSBarry Smith #endif
686289bc588SBarry Smith         for (j=0; j<n; j++) {
687cddbea37SSatish Balay           if (indexn[j] < 0) { idx++; continue;}
6882515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
689899cda47SBarry 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);
69058804f6dSBarry Smith #endif
691cddbea37SSatish Balay           mat->v[indexn[j]*mat->lda + indexm[i]] += v[idx++];
692289bc588SBarry Smith         }
693289bc588SBarry Smith       }
694289bc588SBarry Smith     }
695e8d4e0b9SBarry Smith   }
6963a40ed3dSBarry Smith   PetscFunctionReturn(0);
697289bc588SBarry Smith }
698e8d4e0b9SBarry Smith 
6994a2ae208SSatish Balay #undef __FUNCT__
7004a2ae208SSatish Balay #define __FUNCT__ "MatGetValues_SeqDense"
70113f74950SBarry Smith PetscErrorCode MatGetValues_SeqDense(Mat A,PetscInt m,const PetscInt indexm[],PetscInt n,const PetscInt indexn[],PetscScalar v[])
702ae80bb75SLois Curfman McInnes {
703ae80bb75SLois Curfman McInnes   Mat_SeqDense *mat = (Mat_SeqDense*)A->data;
70413f74950SBarry Smith   PetscInt     i,j;
705ae80bb75SLois Curfman McInnes 
7063a40ed3dSBarry Smith   PetscFunctionBegin;
707ae80bb75SLois Curfman McInnes   /* row-oriented output */
708ae80bb75SLois Curfman McInnes   for (i=0; i<m; i++) {
70997e567efSBarry Smith     if (indexm[i] < 0) {v += n;continue;}
710f12f8aa1SBarry 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);
711ae80bb75SLois Curfman McInnes     for (j=0; j<n; j++) {
7126f31f424SBarry Smith       if (indexn[j] < 0) {v++; continue;}
7136f31f424SBarry 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);
71497e567efSBarry Smith       *v++ = mat->v[indexn[j]*mat->lda + indexm[i]];
715ae80bb75SLois Curfman McInnes     }
716ae80bb75SLois Curfman McInnes   }
7173a40ed3dSBarry Smith   PetscFunctionReturn(0);
718ae80bb75SLois Curfman McInnes }
719ae80bb75SLois Curfman McInnes 
720289bc588SBarry Smith /* -----------------------------------------------------------------*/
721289bc588SBarry Smith 
722e090d566SSatish Balay #include "petscsys.h"
72388e32edaSLois Curfman McInnes 
7244a2ae208SSatish Balay #undef __FUNCT__
7254a2ae208SSatish Balay #define __FUNCT__ "MatLoad_SeqDense"
726a313700dSBarry Smith PetscErrorCode MatLoad_SeqDense(PetscViewer viewer, const MatType type,Mat *A)
72788e32edaSLois Curfman McInnes {
72888e32edaSLois Curfman McInnes   Mat_SeqDense   *a;
72988e32edaSLois Curfman McInnes   Mat            B;
7306849ba73SBarry Smith   PetscErrorCode ierr;
73113f74950SBarry Smith   PetscInt       *scols,i,j,nz,header[4];
73213f74950SBarry Smith   int            fd;
73313f74950SBarry Smith   PetscMPIInt    size;
73413f74950SBarry Smith   PetscInt       *rowlengths = 0,M,N,*cols;
73587828ca2SBarry Smith   PetscScalar    *vals,*svals,*v,*w;
73619bcc07fSBarry Smith   MPI_Comm       comm = ((PetscObject)viewer)->comm;
73788e32edaSLois Curfman McInnes 
7383a40ed3dSBarry Smith   PetscFunctionBegin;
739d132466eSBarry Smith   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
74029bbc08cSBarry Smith   if (size > 1) SETERRQ(PETSC_ERR_ARG_WRONG,"view must have one processor");
741b0a32e0cSBarry Smith   ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr);
7420752156aSBarry Smith   ierr = PetscBinaryRead(fd,header,4,PETSC_INT);CHKERRQ(ierr);
743552e946dSBarry Smith   if (header[0] != MAT_FILE_COOKIE) SETERRQ(PETSC_ERR_FILE_UNEXPECTED,"Not matrix object");
74488e32edaSLois Curfman McInnes   M = header[1]; N = header[2]; nz = header[3];
74588e32edaSLois Curfman McInnes 
74690ace30eSBarry Smith   if (nz == MATRIX_BINARY_FORMAT_DENSE) { /* matrix in file is dense */
747f69a0ea3SMatthew Knepley     ierr = MatCreate(comm,A);CHKERRQ(ierr);
748f69a0ea3SMatthew Knepley     ierr = MatSetSizes(*A,M,N,M,N);CHKERRQ(ierr);
749be5d1d56SKris Buschelman     ierr = MatSetType(*A,type);CHKERRQ(ierr);
7505c5985e7SKris Buschelman     ierr = MatSeqDenseSetPreallocation(*A,PETSC_NULL);CHKERRQ(ierr);
75190ace30eSBarry Smith     B    = *A;
75290ace30eSBarry Smith     a    = (Mat_SeqDense*)B->data;
7534a41a4d3SSatish Balay     v    = a->v;
7544a41a4d3SSatish Balay     /* Allocate some temp space to read in the values and then flip them
7554a41a4d3SSatish Balay        from row major to column major */
756b72907f3SBarry Smith     ierr = PetscMalloc((M*N > 0 ? M*N : 1)*sizeof(PetscScalar),&w);CHKERRQ(ierr);
75790ace30eSBarry Smith     /* read in nonzero values */
7584a41a4d3SSatish Balay     ierr = PetscBinaryRead(fd,w,M*N,PETSC_SCALAR);CHKERRQ(ierr);
7594a41a4d3SSatish Balay     /* now flip the values and store them in the matrix*/
7604a41a4d3SSatish Balay     for (j=0; j<N; j++) {
7614a41a4d3SSatish Balay       for (i=0; i<M; i++) {
7624a41a4d3SSatish Balay         *v++ =w[i*N+j];
7634a41a4d3SSatish Balay       }
7644a41a4d3SSatish Balay     }
765606d414cSSatish Balay     ierr = PetscFree(w);CHKERRQ(ierr);
7666d4a8577SBarry Smith     ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
7676d4a8577SBarry Smith     ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
76890ace30eSBarry Smith   } else {
76988e32edaSLois Curfman McInnes     /* read row lengths */
77013f74950SBarry Smith     ierr = PetscMalloc((M+1)*sizeof(PetscInt),&rowlengths);CHKERRQ(ierr);
7710752156aSBarry Smith     ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr);
77288e32edaSLois Curfman McInnes 
77388e32edaSLois Curfman McInnes     /* create our matrix */
774f69a0ea3SMatthew Knepley     ierr = MatCreate(comm,A);CHKERRQ(ierr);
775f69a0ea3SMatthew Knepley     ierr = MatSetSizes(*A,M,N,M,N);CHKERRQ(ierr);
776be5d1d56SKris Buschelman     ierr = MatSetType(*A,type);CHKERRQ(ierr);
7775c5985e7SKris Buschelman     ierr = MatSeqDenseSetPreallocation(*A,PETSC_NULL);CHKERRQ(ierr);
77888e32edaSLois Curfman McInnes     B = *A;
77988e32edaSLois Curfman McInnes     a = (Mat_SeqDense*)B->data;
78088e32edaSLois Curfman McInnes     v = a->v;
78188e32edaSLois Curfman McInnes 
78288e32edaSLois Curfman McInnes     /* read column indices and nonzeros */
78313f74950SBarry Smith     ierr = PetscMalloc((nz+1)*sizeof(PetscInt),&scols);CHKERRQ(ierr);
784b0a32e0cSBarry Smith     cols = scols;
7850752156aSBarry Smith     ierr = PetscBinaryRead(fd,cols,nz,PETSC_INT);CHKERRQ(ierr);
78687828ca2SBarry Smith     ierr = PetscMalloc((nz+1)*sizeof(PetscScalar),&svals);CHKERRQ(ierr);
787b0a32e0cSBarry Smith     vals = svals;
7880752156aSBarry Smith     ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR);CHKERRQ(ierr);
78988e32edaSLois Curfman McInnes 
79088e32edaSLois Curfman McInnes     /* insert into matrix */
79188e32edaSLois Curfman McInnes     for (i=0; i<M; i++) {
79288e32edaSLois Curfman McInnes       for (j=0; j<rowlengths[i]; j++) v[i+M*scols[j]] = svals[j];
79388e32edaSLois Curfman McInnes       svals += rowlengths[i]; scols += rowlengths[i];
79488e32edaSLois Curfman McInnes     }
795606d414cSSatish Balay     ierr = PetscFree(vals);CHKERRQ(ierr);
796606d414cSSatish Balay     ierr = PetscFree(cols);CHKERRQ(ierr);
797606d414cSSatish Balay     ierr = PetscFree(rowlengths);CHKERRQ(ierr);
79888e32edaSLois Curfman McInnes 
7996d4a8577SBarry Smith     ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
8006d4a8577SBarry Smith     ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
80190ace30eSBarry Smith   }
8023a40ed3dSBarry Smith   PetscFunctionReturn(0);
80388e32edaSLois Curfman McInnes }
80488e32edaSLois Curfman McInnes 
805e090d566SSatish Balay #include "petscsys.h"
806932b0c3eSLois Curfman McInnes 
8074a2ae208SSatish Balay #undef __FUNCT__
8084a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqDense_ASCII"
8096849ba73SBarry Smith static PetscErrorCode MatView_SeqDense_ASCII(Mat A,PetscViewer viewer)
810289bc588SBarry Smith {
811932b0c3eSLois Curfman McInnes   Mat_SeqDense      *a = (Mat_SeqDense*)A->data;
812dfbe8321SBarry Smith   PetscErrorCode    ierr;
81313f74950SBarry Smith   PetscInt          i,j;
8142dcb1b2aSMatthew Knepley   const char        *name;
81587828ca2SBarry Smith   PetscScalar       *v;
816f3ef73ceSBarry Smith   PetscViewerFormat format;
8175f481a85SSatish Balay #if defined(PETSC_USE_COMPLEX)
8185f481a85SSatish Balay   PetscTruth        allreal = PETSC_TRUE;
8195f481a85SSatish Balay #endif
820932b0c3eSLois Curfman McInnes 
8213a40ed3dSBarry Smith   PetscFunctionBegin;
822435da068SBarry Smith   ierr = PetscObjectGetName((PetscObject)A,&name);CHKERRQ(ierr);
823b0a32e0cSBarry Smith   ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
824456192e2SBarry Smith   if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
8253a40ed3dSBarry Smith     PetscFunctionReturn(0);  /* do nothing for now */
826fb9695e5SSatish Balay   } else if (format == PETSC_VIEWER_ASCII_COMMON) {
827b0a32e0cSBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_NO);CHKERRQ(ierr);
828899cda47SBarry Smith     for (i=0; i<A->rmap.n; i++) {
82944cd7ae7SLois Curfman McInnes       v = a->v + i;
83077431f27SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i);CHKERRQ(ierr);
831899cda47SBarry Smith       for (j=0; j<A->cmap.n; j++) {
832aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX)
833329f5518SBarry Smith         if (PetscRealPart(*v) != 0.0 && PetscImaginaryPart(*v) != 0.0) {
834a83599f4SBarry Smith           ierr = PetscViewerASCIIPrintf(viewer," (%D, %G + %G i) ",j,PetscRealPart(*v),PetscImaginaryPart(*v));CHKERRQ(ierr);
835329f5518SBarry Smith         } else if (PetscRealPart(*v)) {
836a83599f4SBarry Smith           ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",j,PetscRealPart(*v));CHKERRQ(ierr);
8376831982aSBarry Smith         }
83880cd9d93SLois Curfman McInnes #else
8396831982aSBarry Smith         if (*v) {
840a83599f4SBarry Smith           ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",j,*v);CHKERRQ(ierr);
8416831982aSBarry Smith         }
84280cd9d93SLois Curfman McInnes #endif
8431b807ce4Svictorle         v += a->lda;
84480cd9d93SLois Curfman McInnes       }
845b0a32e0cSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
84680cd9d93SLois Curfman McInnes     }
847b0a32e0cSBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_YES);CHKERRQ(ierr);
8483a40ed3dSBarry Smith   } else {
849b0a32e0cSBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_NO);CHKERRQ(ierr);
850aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX)
85147989497SBarry Smith     /* determine if matrix has all real values */
85247989497SBarry Smith     v = a->v;
853899cda47SBarry Smith     for (i=0; i<A->rmap.n*A->cmap.n; i++) {
854ffac6cdbSBarry Smith 	if (PetscImaginaryPart(v[i])) { allreal = PETSC_FALSE; break ;}
85547989497SBarry Smith     }
85647989497SBarry Smith #endif
857fb9695e5SSatish Balay     if (format == PETSC_VIEWER_ASCII_MATLAB) {
8583a7fca6bSBarry Smith       ierr = PetscObjectGetName((PetscObject)A,&name);CHKERRQ(ierr);
859899cda47SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"%% Size = %D %D \n",A->rmap.n,A->cmap.n);CHKERRQ(ierr);
860899cda47SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"%s = zeros(%D,%D);\n",name,A->rmap.n,A->cmap.n);CHKERRQ(ierr);
861fb9695e5SSatish Balay       ierr = PetscViewerASCIIPrintf(viewer,"%s = [\n",name);CHKERRQ(ierr);
862ffac6cdbSBarry Smith     }
863ffac6cdbSBarry Smith 
864899cda47SBarry Smith     for (i=0; i<A->rmap.n; i++) {
865932b0c3eSLois Curfman McInnes       v = a->v + i;
866899cda47SBarry Smith       for (j=0; j<A->cmap.n; j++) {
867aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX)
86847989497SBarry Smith         if (allreal) {
869f32d5d43SBarry Smith           ierr = PetscViewerASCIIPrintf(viewer,"%18.16e ",PetscRealPart(*v));CHKERRQ(ierr);
87047989497SBarry Smith         } else {
871f32d5d43SBarry Smith           ierr = PetscViewerASCIIPrintf(viewer,"%18.16e + %18.16e i ",PetscRealPart(*v),PetscImaginaryPart(*v));CHKERRQ(ierr);
87247989497SBarry Smith         }
873289bc588SBarry Smith #else
874f32d5d43SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"%18.16e ",*v);CHKERRQ(ierr);
875289bc588SBarry Smith #endif
8761b807ce4Svictorle 	v += a->lda;
877289bc588SBarry Smith       }
878b0a32e0cSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
879289bc588SBarry Smith     }
880fb9695e5SSatish Balay     if (format == PETSC_VIEWER_ASCII_MATLAB) {
881b0a32e0cSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"];\n");CHKERRQ(ierr);
882ffac6cdbSBarry Smith     }
883b0a32e0cSBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_YES);CHKERRQ(ierr);
884da3a660dSBarry Smith   }
885b0a32e0cSBarry Smith   ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
8863a40ed3dSBarry Smith   PetscFunctionReturn(0);
887289bc588SBarry Smith }
888289bc588SBarry Smith 
8894a2ae208SSatish Balay #undef __FUNCT__
8904a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqDense_Binary"
8916849ba73SBarry Smith static PetscErrorCode MatView_SeqDense_Binary(Mat A,PetscViewer viewer)
892932b0c3eSLois Curfman McInnes {
893932b0c3eSLois Curfman McInnes   Mat_SeqDense      *a = (Mat_SeqDense*)A->data;
8946849ba73SBarry Smith   PetscErrorCode    ierr;
89513f74950SBarry Smith   int               fd;
896899cda47SBarry Smith   PetscInt          ict,j,n = A->cmap.n,m = A->rmap.n,i,*col_lens,nz = m*n;
89787828ca2SBarry Smith   PetscScalar       *v,*anonz,*vals;
898f3ef73ceSBarry Smith   PetscViewerFormat format;
899932b0c3eSLois Curfman McInnes 
9003a40ed3dSBarry Smith   PetscFunctionBegin;
901b0a32e0cSBarry Smith   ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr);
90290ace30eSBarry Smith 
903b0a32e0cSBarry Smith   ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
904fb9695e5SSatish Balay   if (format == PETSC_VIEWER_BINARY_NATIVE) {
90590ace30eSBarry Smith     /* store the matrix as a dense matrix */
90613f74950SBarry Smith     ierr = PetscMalloc(4*sizeof(PetscInt),&col_lens);CHKERRQ(ierr);
907552e946dSBarry Smith     col_lens[0] = MAT_FILE_COOKIE;
90890ace30eSBarry Smith     col_lens[1] = m;
90990ace30eSBarry Smith     col_lens[2] = n;
91090ace30eSBarry Smith     col_lens[3] = MATRIX_BINARY_FORMAT_DENSE;
9116f69ff64SBarry Smith     ierr = PetscBinaryWrite(fd,col_lens,4,PETSC_INT,PETSC_TRUE);CHKERRQ(ierr);
912606d414cSSatish Balay     ierr = PetscFree(col_lens);CHKERRQ(ierr);
91390ace30eSBarry Smith 
91490ace30eSBarry Smith     /* write out matrix, by rows */
91587828ca2SBarry Smith     ierr = PetscMalloc((m*n+1)*sizeof(PetscScalar),&vals);CHKERRQ(ierr);
91690ace30eSBarry Smith     v    = a->v;
91790ace30eSBarry Smith     for (j=0; j<n; j++) {
918578230a0SSatish Balay       for (i=0; i<m; i++) {
919578230a0SSatish Balay         vals[j + i*n] = *v++;
92090ace30eSBarry Smith       }
92190ace30eSBarry Smith     }
9226f69ff64SBarry Smith     ierr = PetscBinaryWrite(fd,vals,n*m,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr);
923606d414cSSatish Balay     ierr = PetscFree(vals);CHKERRQ(ierr);
92490ace30eSBarry Smith   } else {
92513f74950SBarry Smith     ierr = PetscMalloc((4+nz)*sizeof(PetscInt),&col_lens);CHKERRQ(ierr);
926552e946dSBarry Smith     col_lens[0] = MAT_FILE_COOKIE;
927932b0c3eSLois Curfman McInnes     col_lens[1] = m;
928932b0c3eSLois Curfman McInnes     col_lens[2] = n;
929932b0c3eSLois Curfman McInnes     col_lens[3] = nz;
930932b0c3eSLois Curfman McInnes 
931932b0c3eSLois Curfman McInnes     /* store lengths of each row and write (including header) to file */
932932b0c3eSLois Curfman McInnes     for (i=0; i<m; i++) col_lens[4+i] = n;
9336f69ff64SBarry Smith     ierr = PetscBinaryWrite(fd,col_lens,4+m,PETSC_INT,PETSC_TRUE);CHKERRQ(ierr);
934932b0c3eSLois Curfman McInnes 
935932b0c3eSLois Curfman McInnes     /* Possibly should write in smaller increments, not whole matrix at once? */
936932b0c3eSLois Curfman McInnes     /* store column indices (zero start index) */
937932b0c3eSLois Curfman McInnes     ict = 0;
938932b0c3eSLois Curfman McInnes     for (i=0; i<m; i++) {
939932b0c3eSLois Curfman McInnes       for (j=0; j<n; j++) col_lens[ict++] = j;
940932b0c3eSLois Curfman McInnes     }
9416f69ff64SBarry Smith     ierr = PetscBinaryWrite(fd,col_lens,nz,PETSC_INT,PETSC_FALSE);CHKERRQ(ierr);
942606d414cSSatish Balay     ierr = PetscFree(col_lens);CHKERRQ(ierr);
943932b0c3eSLois Curfman McInnes 
944932b0c3eSLois Curfman McInnes     /* store nonzero values */
94587828ca2SBarry Smith     ierr = PetscMalloc((nz+1)*sizeof(PetscScalar),&anonz);CHKERRQ(ierr);
946932b0c3eSLois Curfman McInnes     ict  = 0;
947932b0c3eSLois Curfman McInnes     for (i=0; i<m; i++) {
948932b0c3eSLois Curfman McInnes       v = a->v + i;
949932b0c3eSLois Curfman McInnes       for (j=0; j<n; j++) {
9501b807ce4Svictorle         anonz[ict++] = *v; v += a->lda;
951932b0c3eSLois Curfman McInnes       }
952932b0c3eSLois Curfman McInnes     }
9536f69ff64SBarry Smith     ierr = PetscBinaryWrite(fd,anonz,nz,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr);
954606d414cSSatish Balay     ierr = PetscFree(anonz);CHKERRQ(ierr);
95590ace30eSBarry Smith   }
9563a40ed3dSBarry Smith   PetscFunctionReturn(0);
957932b0c3eSLois Curfman McInnes }
958932b0c3eSLois Curfman McInnes 
9594a2ae208SSatish Balay #undef __FUNCT__
9604a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqDense_Draw_Zoom"
961dfbe8321SBarry Smith PetscErrorCode MatView_SeqDense_Draw_Zoom(PetscDraw draw,void *Aa)
962f1af5d2fSBarry Smith {
963f1af5d2fSBarry Smith   Mat               A = (Mat) Aa;
964f1af5d2fSBarry Smith   Mat_SeqDense      *a = (Mat_SeqDense*)A->data;
9656849ba73SBarry Smith   PetscErrorCode    ierr;
966899cda47SBarry Smith   PetscInt          m = A->rmap.n,n = A->cmap.n,color,i,j;
96787828ca2SBarry Smith   PetscScalar       *v = a->v;
968b0a32e0cSBarry Smith   PetscViewer       viewer;
969b0a32e0cSBarry Smith   PetscDraw         popup;
970329f5518SBarry Smith   PetscReal         xl,yl,xr,yr,x_l,x_r,y_l,y_r,scale,maxv = 0.0;
971f3ef73ceSBarry Smith   PetscViewerFormat format;
972f1af5d2fSBarry Smith 
973f1af5d2fSBarry Smith   PetscFunctionBegin;
974f1af5d2fSBarry Smith 
975f1af5d2fSBarry Smith   ierr = PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer);CHKERRQ(ierr);
976b0a32e0cSBarry Smith   ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
977b0a32e0cSBarry Smith   ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr);
978f1af5d2fSBarry Smith 
979f1af5d2fSBarry Smith   /* Loop over matrix elements drawing boxes */
980fb9695e5SSatish Balay   if (format != PETSC_VIEWER_DRAW_CONTOUR) {
981f1af5d2fSBarry Smith     /* Blue for negative and Red for positive */
982b0a32e0cSBarry Smith     color = PETSC_DRAW_BLUE;
983f1af5d2fSBarry Smith     for(j = 0; j < n; j++) {
984f1af5d2fSBarry Smith       x_l = j;
985f1af5d2fSBarry Smith       x_r = x_l + 1.0;
986f1af5d2fSBarry Smith       for(i = 0; i < m; i++) {
987f1af5d2fSBarry Smith         y_l = m - i - 1.0;
988f1af5d2fSBarry Smith         y_r = y_l + 1.0;
989f1af5d2fSBarry Smith #if defined(PETSC_USE_COMPLEX)
990329f5518SBarry Smith         if (PetscRealPart(v[j*m+i]) >  0.) {
991b0a32e0cSBarry Smith           color = PETSC_DRAW_RED;
992329f5518SBarry Smith         } else if (PetscRealPart(v[j*m+i]) <  0.) {
993b0a32e0cSBarry Smith           color = PETSC_DRAW_BLUE;
994f1af5d2fSBarry Smith         } else {
995f1af5d2fSBarry Smith           continue;
996f1af5d2fSBarry Smith         }
997f1af5d2fSBarry Smith #else
998f1af5d2fSBarry Smith         if (v[j*m+i] >  0.) {
999b0a32e0cSBarry Smith           color = PETSC_DRAW_RED;
1000f1af5d2fSBarry Smith         } else if (v[j*m+i] <  0.) {
1001b0a32e0cSBarry Smith           color = PETSC_DRAW_BLUE;
1002f1af5d2fSBarry Smith         } else {
1003f1af5d2fSBarry Smith           continue;
1004f1af5d2fSBarry Smith         }
1005f1af5d2fSBarry Smith #endif
1006b0a32e0cSBarry Smith         ierr = PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);CHKERRQ(ierr);
1007f1af5d2fSBarry Smith       }
1008f1af5d2fSBarry Smith     }
1009f1af5d2fSBarry Smith   } else {
1010f1af5d2fSBarry Smith     /* use contour shading to indicate magnitude of values */
1011f1af5d2fSBarry Smith     /* first determine max of all nonzero values */
1012f1af5d2fSBarry Smith     for(i = 0; i < m*n; i++) {
1013f1af5d2fSBarry Smith       if (PetscAbsScalar(v[i]) > maxv) maxv = PetscAbsScalar(v[i]);
1014f1af5d2fSBarry Smith     }
1015b0a32e0cSBarry Smith     scale = (245.0 - PETSC_DRAW_BASIC_COLORS)/maxv;
1016b0a32e0cSBarry Smith     ierr  = PetscDrawGetPopup(draw,&popup);CHKERRQ(ierr);
1017b0a32e0cSBarry Smith     if (popup) {ierr  = PetscDrawScalePopup(popup,0.0,maxv);CHKERRQ(ierr);}
1018f1af5d2fSBarry Smith     for(j = 0; j < n; j++) {
1019f1af5d2fSBarry Smith       x_l = j;
1020f1af5d2fSBarry Smith       x_r = x_l + 1.0;
1021f1af5d2fSBarry Smith       for(i = 0; i < m; i++) {
1022f1af5d2fSBarry Smith         y_l   = m - i - 1.0;
1023f1af5d2fSBarry Smith         y_r   = y_l + 1.0;
1024b0a32e0cSBarry Smith         color = PETSC_DRAW_BASIC_COLORS + (int)(scale*PetscAbsScalar(v[j*m+i]));
1025b0a32e0cSBarry Smith         ierr  = PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);CHKERRQ(ierr);
1026f1af5d2fSBarry Smith       }
1027f1af5d2fSBarry Smith     }
1028f1af5d2fSBarry Smith   }
1029f1af5d2fSBarry Smith   PetscFunctionReturn(0);
1030f1af5d2fSBarry Smith }
1031f1af5d2fSBarry Smith 
10324a2ae208SSatish Balay #undef __FUNCT__
10334a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqDense_Draw"
1034dfbe8321SBarry Smith PetscErrorCode MatView_SeqDense_Draw(Mat A,PetscViewer viewer)
1035f1af5d2fSBarry Smith {
1036b0a32e0cSBarry Smith   PetscDraw      draw;
1037f1af5d2fSBarry Smith   PetscTruth     isnull;
1038329f5518SBarry Smith   PetscReal      xr,yr,xl,yl,h,w;
1039dfbe8321SBarry Smith   PetscErrorCode ierr;
1040f1af5d2fSBarry Smith 
1041f1af5d2fSBarry Smith   PetscFunctionBegin;
1042b0a32e0cSBarry Smith   ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
1043b0a32e0cSBarry Smith   ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr);
1044abc0a331SBarry Smith   if (isnull) PetscFunctionReturn(0);
1045f1af5d2fSBarry Smith 
1046f1af5d2fSBarry Smith   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer);CHKERRQ(ierr);
1047899cda47SBarry Smith   xr  = A->cmap.n; yr = A->rmap.n; h = yr/10.0; w = xr/10.0;
1048f1af5d2fSBarry Smith   xr += w;    yr += h;  xl = -w;     yl = -h;
1049b0a32e0cSBarry Smith   ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr);
1050b0a32e0cSBarry Smith   ierr = PetscDrawZoom(draw,MatView_SeqDense_Draw_Zoom,A);CHKERRQ(ierr);
1051f1af5d2fSBarry Smith   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",PETSC_NULL);CHKERRQ(ierr);
1052f1af5d2fSBarry Smith   PetscFunctionReturn(0);
1053f1af5d2fSBarry Smith }
1054f1af5d2fSBarry Smith 
10554a2ae208SSatish Balay #undef __FUNCT__
10564a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqDense"
1057dfbe8321SBarry Smith PetscErrorCode MatView_SeqDense(Mat A,PetscViewer viewer)
1058932b0c3eSLois Curfman McInnes {
1059dfbe8321SBarry Smith   PetscErrorCode ierr;
10606805f65bSBarry Smith   PetscTruth     iascii,isbinary,isdraw;
1061932b0c3eSLois Curfman McInnes 
10623a40ed3dSBarry Smith   PetscFunctionBegin;
106332077d6dSBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_ASCII,&iascii);CHKERRQ(ierr);
1064fb9695e5SSatish Balay   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_BINARY,&isbinary);CHKERRQ(ierr);
1065fb9695e5SSatish Balay   ierr = PetscTypeCompare((PetscObject)viewer,PETSC_VIEWER_DRAW,&isdraw);CHKERRQ(ierr);
10660f5bd95cSBarry Smith 
1067c45a1595SBarry Smith   if (iascii) {
1068c45a1595SBarry Smith     ierr = MatView_SeqDense_ASCII(A,viewer);CHKERRQ(ierr);
10690f5bd95cSBarry Smith   } else if (isbinary) {
10703a40ed3dSBarry Smith     ierr = MatView_SeqDense_Binary(A,viewer);CHKERRQ(ierr);
1071f1af5d2fSBarry Smith   } else if (isdraw) {
1072f1af5d2fSBarry Smith     ierr = MatView_SeqDense_Draw(A,viewer);CHKERRQ(ierr);
10735cd90555SBarry Smith   } else {
1074958c9bccSBarry Smith     SETERRQ1(PETSC_ERR_SUP,"Viewer type %s not supported by dense matrix",((PetscObject)viewer)->type_name);
1075932b0c3eSLois Curfman McInnes   }
10763a40ed3dSBarry Smith   PetscFunctionReturn(0);
1077932b0c3eSLois Curfman McInnes }
1078289bc588SBarry Smith 
10794a2ae208SSatish Balay #undef __FUNCT__
10804a2ae208SSatish Balay #define __FUNCT__ "MatDestroy_SeqDense"
1081dfbe8321SBarry Smith PetscErrorCode MatDestroy_SeqDense(Mat mat)
1082289bc588SBarry Smith {
1083ec8511deSBarry Smith   Mat_SeqDense   *l = (Mat_SeqDense*)mat->data;
1084dfbe8321SBarry Smith   PetscErrorCode ierr;
108590f02eecSBarry Smith 
10863a40ed3dSBarry Smith   PetscFunctionBegin;
1087aa482453SBarry Smith #if defined(PETSC_USE_LOG)
1088899cda47SBarry Smith   PetscLogObjectState((PetscObject)mat,"Rows %D Cols %D",mat->rmap.n,mat->cmap.n);
1089a5a9c739SBarry Smith #endif
109005b42c5fSBarry Smith   ierr = PetscFree(l->pivots);CHKERRQ(ierr);
10916857c123SSatish Balay   if (!l->user_alloc) {ierr = PetscFree(l->v);CHKERRQ(ierr);}
1092606d414cSSatish Balay   ierr = PetscFree(l);CHKERRQ(ierr);
1093dbd8c25aSHong Zhang 
1094dbd8c25aSHong Zhang   ierr = PetscObjectChangeTypeName((PetscObject)mat,0);CHKERRQ(ierr);
1095901853e0SKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatSeqDenseSetPreallocation_C","",PETSC_NULL);CHKERRQ(ierr);
10964ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMult_seqaij_seqdense_C","",PETSC_NULL);CHKERRQ(ierr);
10974ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMultSymbolic_seqaij_seqdense_C","",PETSC_NULL);CHKERRQ(ierr);
10984ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMultNumeric_seqaij_seqdense_C","",PETSC_NULL);CHKERRQ(ierr);
10993a40ed3dSBarry Smith   PetscFunctionReturn(0);
1100289bc588SBarry Smith }
1101289bc588SBarry Smith 
11024a2ae208SSatish Balay #undef __FUNCT__
11034a2ae208SSatish Balay #define __FUNCT__ "MatTranspose_SeqDense"
1104fc4dec0aSBarry Smith PetscErrorCode MatTranspose_SeqDense(Mat A,MatReuse reuse,Mat *matout)
1105289bc588SBarry Smith {
1106c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
11076849ba73SBarry Smith   PetscErrorCode ierr;
110813f74950SBarry Smith   PetscInt       k,j,m,n,M;
110987828ca2SBarry Smith   PetscScalar    *v,tmp;
111048b35521SBarry Smith 
11113a40ed3dSBarry Smith   PetscFunctionBegin;
1112899cda47SBarry Smith   v = mat->v; m = A->rmap.n; M = mat->lda; n = A->cmap.n;
1113e9695a30SBarry Smith   if (reuse == MAT_REUSE_MATRIX && *matout == A) { /* in place transpose */
1114a5ce6ee0Svictorle     if (m != n) {
1115634064b4SBarry Smith       SETERRQ(PETSC_ERR_SUP,"Can not transpose non-square matrix in place");
1116d64ed03dSBarry Smith     } else {
1117d3e5ee88SLois Curfman McInnes       for (j=0; j<m; j++) {
1118289bc588SBarry Smith         for (k=0; k<j; k++) {
11191b807ce4Svictorle           tmp = v[j + k*M];
11201b807ce4Svictorle           v[j + k*M] = v[k + j*M];
11211b807ce4Svictorle           v[k + j*M] = tmp;
1122289bc588SBarry Smith         }
1123289bc588SBarry Smith       }
1124d64ed03dSBarry Smith     }
11253a40ed3dSBarry Smith   } else { /* out-of-place transpose */
1126d3e5ee88SLois Curfman McInnes     Mat          tmat;
1127ec8511deSBarry Smith     Mat_SeqDense *tmatd;
112887828ca2SBarry Smith     PetscScalar  *v2;
1129ea709b57SSatish Balay 
1130fc4dec0aSBarry Smith     if (reuse == MAT_INITIAL_MATRIX) {
11317adad957SLisandro Dalcin       ierr  = MatCreate(((PetscObject)A)->comm,&tmat);CHKERRQ(ierr);
1132899cda47SBarry Smith       ierr  = MatSetSizes(tmat,A->cmap.n,A->rmap.n,A->cmap.n,A->rmap.n);CHKERRQ(ierr);
11337adad957SLisandro Dalcin       ierr  = MatSetType(tmat,((PetscObject)A)->type_name);CHKERRQ(ierr);
11345c5985e7SKris Buschelman       ierr  = MatSeqDenseSetPreallocation(tmat,PETSC_NULL);CHKERRQ(ierr);
1135fc4dec0aSBarry Smith     } else {
1136fc4dec0aSBarry Smith       tmat = *matout;
1137fc4dec0aSBarry Smith     }
1138ec8511deSBarry Smith     tmatd = (Mat_SeqDense*)tmat->data;
11390de55854SLois Curfman McInnes     v = mat->v; v2 = tmatd->v;
1140d3e5ee88SLois Curfman McInnes     for (j=0; j<n; j++) {
11411b807ce4Svictorle       for (k=0; k<m; k++) v2[j + k*n] = v[k + j*M];
1142d3e5ee88SLois Curfman McInnes     }
11436d4a8577SBarry Smith     ierr = MatAssemblyBegin(tmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
11446d4a8577SBarry Smith     ierr = MatAssemblyEnd(tmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
1145d3e5ee88SLois Curfman McInnes     *matout = tmat;
114648b35521SBarry Smith   }
11473a40ed3dSBarry Smith   PetscFunctionReturn(0);
1148289bc588SBarry Smith }
1149289bc588SBarry Smith 
11504a2ae208SSatish Balay #undef __FUNCT__
11514a2ae208SSatish Balay #define __FUNCT__ "MatEqual_SeqDense"
1152dfbe8321SBarry Smith PetscErrorCode MatEqual_SeqDense(Mat A1,Mat A2,PetscTruth *flg)
1153289bc588SBarry Smith {
1154c0bbcb79SLois Curfman McInnes   Mat_SeqDense *mat1 = (Mat_SeqDense*)A1->data;
1155c0bbcb79SLois Curfman McInnes   Mat_SeqDense *mat2 = (Mat_SeqDense*)A2->data;
115613f74950SBarry Smith   PetscInt     i,j;
115787828ca2SBarry Smith   PetscScalar  *v1 = mat1->v,*v2 = mat2->v;
11589ea5d5aeSSatish Balay 
11593a40ed3dSBarry Smith   PetscFunctionBegin;
1160899cda47SBarry Smith   if (A1->rmap.n != A2->rmap.n) {*flg = PETSC_FALSE; PetscFunctionReturn(0);}
1161899cda47SBarry Smith   if (A1->cmap.n != A2->cmap.n) {*flg = PETSC_FALSE; PetscFunctionReturn(0);}
1162899cda47SBarry Smith   for (i=0; i<A1->rmap.n; i++) {
11631b807ce4Svictorle     v1 = mat1->v+i; v2 = mat2->v+i;
1164899cda47SBarry Smith     for (j=0; j<A1->cmap.n; j++) {
11653a40ed3dSBarry Smith       if (*v1 != *v2) {*flg = PETSC_FALSE; PetscFunctionReturn(0);}
11661b807ce4Svictorle       v1 += mat1->lda; v2 += mat2->lda;
11671b807ce4Svictorle     }
1168289bc588SBarry Smith   }
116977c4ece6SBarry Smith   *flg = PETSC_TRUE;
11703a40ed3dSBarry Smith   PetscFunctionReturn(0);
1171289bc588SBarry Smith }
1172289bc588SBarry Smith 
11734a2ae208SSatish Balay #undef __FUNCT__
11744a2ae208SSatish Balay #define __FUNCT__ "MatGetDiagonal_SeqDense"
1175dfbe8321SBarry Smith PetscErrorCode MatGetDiagonal_SeqDense(Mat A,Vec v)
1176289bc588SBarry Smith {
1177c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
1178dfbe8321SBarry Smith   PetscErrorCode ierr;
117913f74950SBarry Smith   PetscInt       i,n,len;
118087828ca2SBarry Smith   PetscScalar    *x,zero = 0.0;
118144cd7ae7SLois Curfman McInnes 
11823a40ed3dSBarry Smith   PetscFunctionBegin;
11832dcb1b2aSMatthew Knepley   ierr = VecSet(v,zero);CHKERRQ(ierr);
11847a97a34bSBarry Smith   ierr = VecGetSize(v,&n);CHKERRQ(ierr);
11851ebc52fbSHong Zhang   ierr = VecGetArray(v,&x);CHKERRQ(ierr);
1186899cda47SBarry Smith   len = PetscMin(A->rmap.n,A->cmap.n);
1187899cda47SBarry Smith   if (n != A->rmap.n) SETERRQ(PETSC_ERR_ARG_SIZ,"Nonconforming mat and vec");
118844cd7ae7SLois Curfman McInnes   for (i=0; i<len; i++) {
11891b807ce4Svictorle     x[i] = mat->v[i*mat->lda + i];
1190289bc588SBarry Smith   }
11911ebc52fbSHong Zhang   ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
11923a40ed3dSBarry Smith   PetscFunctionReturn(0);
1193289bc588SBarry Smith }
1194289bc588SBarry Smith 
11954a2ae208SSatish Balay #undef __FUNCT__
11964a2ae208SSatish Balay #define __FUNCT__ "MatDiagonalScale_SeqDense"
1197dfbe8321SBarry Smith PetscErrorCode MatDiagonalScale_SeqDense(Mat A,Vec ll,Vec rr)
1198289bc588SBarry Smith {
1199c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
120087828ca2SBarry Smith   PetscScalar    *l,*r,x,*v;
1201dfbe8321SBarry Smith   PetscErrorCode ierr;
1202899cda47SBarry Smith   PetscInt       i,j,m = A->rmap.n,n = A->cmap.n;
120355659b69SBarry Smith 
12043a40ed3dSBarry Smith   PetscFunctionBegin;
120528988994SBarry Smith   if (ll) {
12067a97a34bSBarry Smith     ierr = VecGetSize(ll,&m);CHKERRQ(ierr);
12071ebc52fbSHong Zhang     ierr = VecGetArray(ll,&l);CHKERRQ(ierr);
1208899cda47SBarry Smith     if (m != A->rmap.n) SETERRQ(PETSC_ERR_ARG_SIZ,"Left scaling vec wrong size");
1209da3a660dSBarry Smith     for (i=0; i<m; i++) {
1210da3a660dSBarry Smith       x = l[i];
1211da3a660dSBarry Smith       v = mat->v + i;
1212da3a660dSBarry Smith       for (j=0; j<n; j++) { (*v) *= x; v+= m;}
1213da3a660dSBarry Smith     }
12141ebc52fbSHong Zhang     ierr = VecRestoreArray(ll,&l);CHKERRQ(ierr);
1215efee365bSSatish Balay     ierr = PetscLogFlops(n*m);CHKERRQ(ierr);
1216da3a660dSBarry Smith   }
121728988994SBarry Smith   if (rr) {
12187a97a34bSBarry Smith     ierr = VecGetSize(rr,&n);CHKERRQ(ierr);
12191ebc52fbSHong Zhang     ierr = VecGetArray(rr,&r);CHKERRQ(ierr);
1220899cda47SBarry Smith     if (n != A->cmap.n) SETERRQ(PETSC_ERR_ARG_SIZ,"Right scaling vec wrong size");
1221da3a660dSBarry Smith     for (i=0; i<n; i++) {
1222da3a660dSBarry Smith       x = r[i];
1223da3a660dSBarry Smith       v = mat->v + i*m;
1224da3a660dSBarry Smith       for (j=0; j<m; j++) { (*v++) *= x;}
1225da3a660dSBarry Smith     }
12261ebc52fbSHong Zhang     ierr = VecRestoreArray(rr,&r);CHKERRQ(ierr);
1227efee365bSSatish Balay     ierr = PetscLogFlops(n*m);CHKERRQ(ierr);
1228da3a660dSBarry Smith   }
12293a40ed3dSBarry Smith   PetscFunctionReturn(0);
1230289bc588SBarry Smith }
1231289bc588SBarry Smith 
12324a2ae208SSatish Balay #undef __FUNCT__
12334a2ae208SSatish Balay #define __FUNCT__ "MatNorm_SeqDense"
1234dfbe8321SBarry Smith PetscErrorCode MatNorm_SeqDense(Mat A,NormType type,PetscReal *nrm)
1235289bc588SBarry Smith {
1236c0bbcb79SLois Curfman McInnes   Mat_SeqDense *mat = (Mat_SeqDense*)A->data;
123787828ca2SBarry Smith   PetscScalar  *v = mat->v;
1238329f5518SBarry Smith   PetscReal    sum = 0.0;
1239899cda47SBarry Smith   PetscInt     lda=mat->lda,m=A->rmap.n,i,j;
1240efee365bSSatish Balay   PetscErrorCode ierr;
124155659b69SBarry Smith 
12423a40ed3dSBarry Smith   PetscFunctionBegin;
1243289bc588SBarry Smith   if (type == NORM_FROBENIUS) {
1244a5ce6ee0Svictorle     if (lda>m) {
1245899cda47SBarry Smith       for (j=0; j<A->cmap.n; j++) {
1246a5ce6ee0Svictorle 	v = mat->v+j*lda;
1247a5ce6ee0Svictorle 	for (i=0; i<m; i++) {
1248a5ce6ee0Svictorle #if defined(PETSC_USE_COMPLEX)
1249a5ce6ee0Svictorle 	  sum += PetscRealPart(PetscConj(*v)*(*v)); v++;
1250a5ce6ee0Svictorle #else
1251a5ce6ee0Svictorle 	  sum += (*v)*(*v); v++;
1252a5ce6ee0Svictorle #endif
1253a5ce6ee0Svictorle 	}
1254a5ce6ee0Svictorle       }
1255a5ce6ee0Svictorle     } else {
1256899cda47SBarry Smith       for (i=0; i<A->cmap.n*A->rmap.n; i++) {
1257aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX)
1258329f5518SBarry Smith 	sum += PetscRealPart(PetscConj(*v)*(*v)); v++;
1259289bc588SBarry Smith #else
1260289bc588SBarry Smith 	sum += (*v)*(*v); v++;
1261289bc588SBarry Smith #endif
1262289bc588SBarry Smith       }
1263a5ce6ee0Svictorle     }
1264064f8208SBarry Smith     *nrm = sqrt(sum);
1265899cda47SBarry Smith     ierr = PetscLogFlops(2*A->cmap.n*A->rmap.n);CHKERRQ(ierr);
12663a40ed3dSBarry Smith   } else if (type == NORM_1) {
1267064f8208SBarry Smith     *nrm = 0.0;
1268899cda47SBarry Smith     for (j=0; j<A->cmap.n; j++) {
12691b807ce4Svictorle       v = mat->v + j*mat->lda;
1270289bc588SBarry Smith       sum = 0.0;
1271899cda47SBarry Smith       for (i=0; i<A->rmap.n; i++) {
127233a8263dSBarry Smith         sum += PetscAbsScalar(*v);  v++;
1273289bc588SBarry Smith       }
1274064f8208SBarry Smith       if (sum > *nrm) *nrm = sum;
1275289bc588SBarry Smith     }
1276899cda47SBarry Smith     ierr = PetscLogFlops(A->cmap.n*A->rmap.n);CHKERRQ(ierr);
12773a40ed3dSBarry Smith   } else if (type == NORM_INFINITY) {
1278064f8208SBarry Smith     *nrm = 0.0;
1279899cda47SBarry Smith     for (j=0; j<A->rmap.n; j++) {
1280289bc588SBarry Smith       v = mat->v + j;
1281289bc588SBarry Smith       sum = 0.0;
1282899cda47SBarry Smith       for (i=0; i<A->cmap.n; i++) {
12831b807ce4Svictorle         sum += PetscAbsScalar(*v); v += mat->lda;
1284289bc588SBarry Smith       }
1285064f8208SBarry Smith       if (sum > *nrm) *nrm = sum;
1286289bc588SBarry Smith     }
1287899cda47SBarry Smith     ierr = PetscLogFlops(A->cmap.n*A->rmap.n);CHKERRQ(ierr);
12883a40ed3dSBarry Smith   } else {
128929bbc08cSBarry Smith     SETERRQ(PETSC_ERR_SUP,"No two norm");
1290289bc588SBarry Smith   }
12913a40ed3dSBarry Smith   PetscFunctionReturn(0);
1292289bc588SBarry Smith }
1293289bc588SBarry Smith 
12944a2ae208SSatish Balay #undef __FUNCT__
12954a2ae208SSatish Balay #define __FUNCT__ "MatSetOption_SeqDense"
12964e0d8c25SBarry Smith PetscErrorCode MatSetOption_SeqDense(Mat A,MatOption op,PetscTruth flg)
1297289bc588SBarry Smith {
1298c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *aij = (Mat_SeqDense*)A->data;
129963ba0a88SBarry Smith   PetscErrorCode ierr;
130067e560aaSBarry Smith 
13013a40ed3dSBarry Smith   PetscFunctionBegin;
1302b5a2b587SKris Buschelman   switch (op) {
1303b5a2b587SKris Buschelman   case MAT_ROW_ORIENTED:
13044e0d8c25SBarry Smith     aij->roworiented = flg;
1305b5a2b587SKris Buschelman     break;
1306512a5fc5SBarry Smith   case MAT_NEW_NONZERO_LOCATIONS:
1307b5a2b587SKris Buschelman   case MAT_NEW_NONZERO_LOCATION_ERR:
13083971808eSMatthew Knepley   case MAT_NEW_NONZERO_ALLOCATION_ERR:
13094e0d8c25SBarry Smith   case MAT_NEW_DIAGONALS:
1310b5a2b587SKris Buschelman   case MAT_IGNORE_OFF_PROC_ENTRIES:
1311b5a2b587SKris Buschelman   case MAT_USE_HASH_TABLE:
131277e54ba9SKris Buschelman   case MAT_SYMMETRIC:
131377e54ba9SKris Buschelman   case MAT_STRUCTURALLY_SYMMETRIC:
13149a4540c5SBarry Smith   case MAT_HERMITIAN:
13159a4540c5SBarry Smith   case MAT_SYMMETRY_ETERNAL:
1316600fe468SBarry Smith   case MAT_IGNORE_LOWER_TRIANGULAR:
1317290bbb0aSBarry Smith     ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr);
131877e54ba9SKris Buschelman     break;
1319b5a2b587SKris Buschelman   default:
1320600fe468SBarry Smith     SETERRQ1(PETSC_ERR_SUP,"unknown option %s",MatOptions[op]);
13213a40ed3dSBarry Smith   }
13223a40ed3dSBarry Smith   PetscFunctionReturn(0);
1323289bc588SBarry Smith }
1324289bc588SBarry Smith 
13254a2ae208SSatish Balay #undef __FUNCT__
13264a2ae208SSatish Balay #define __FUNCT__ "MatZeroEntries_SeqDense"
1327dfbe8321SBarry Smith PetscErrorCode MatZeroEntries_SeqDense(Mat A)
13286f0a148fSBarry Smith {
1329ec8511deSBarry Smith   Mat_SeqDense   *l = (Mat_SeqDense*)A->data;
13306849ba73SBarry Smith   PetscErrorCode ierr;
1331899cda47SBarry Smith   PetscInt       lda=l->lda,m=A->rmap.n,j;
13323a40ed3dSBarry Smith 
13333a40ed3dSBarry Smith   PetscFunctionBegin;
1334a5ce6ee0Svictorle   if (lda>m) {
1335899cda47SBarry Smith     for (j=0; j<A->cmap.n; j++) {
1336a5ce6ee0Svictorle       ierr = PetscMemzero(l->v+j*lda,m*sizeof(PetscScalar));CHKERRQ(ierr);
1337a5ce6ee0Svictorle     }
1338a5ce6ee0Svictorle   } else {
1339899cda47SBarry Smith     ierr = PetscMemzero(l->v,A->rmap.n*A->cmap.n*sizeof(PetscScalar));CHKERRQ(ierr);
1340a5ce6ee0Svictorle   }
13413a40ed3dSBarry Smith   PetscFunctionReturn(0);
13426f0a148fSBarry Smith }
13436f0a148fSBarry Smith 
13444a2ae208SSatish Balay #undef __FUNCT__
13454a2ae208SSatish Balay #define __FUNCT__ "MatZeroRows_SeqDense"
1346f4df32b1SMatthew Knepley PetscErrorCode MatZeroRows_SeqDense(Mat A,PetscInt N,const PetscInt rows[],PetscScalar diag)
13476f0a148fSBarry Smith {
1348ec8511deSBarry Smith   Mat_SeqDense   *l = (Mat_SeqDense*)A->data;
1349899cda47SBarry Smith   PetscInt       n = A->cmap.n,i,j;
135087828ca2SBarry Smith   PetscScalar    *slot;
135155659b69SBarry Smith 
13523a40ed3dSBarry Smith   PetscFunctionBegin;
13536f0a148fSBarry Smith   for (i=0; i<N; i++) {
13546f0a148fSBarry Smith     slot = l->v + rows[i];
13556f0a148fSBarry Smith     for (j=0; j<n; j++) { *slot = 0.0; slot += n;}
13566f0a148fSBarry Smith   }
1357f4df32b1SMatthew Knepley   if (diag != 0.0) {
13586f0a148fSBarry Smith     for (i=0; i<N; i++) {
13596f0a148fSBarry Smith       slot = l->v + (n+1)*rows[i];
1360f4df32b1SMatthew Knepley       *slot = diag;
13616f0a148fSBarry Smith     }
13626f0a148fSBarry Smith   }
13633a40ed3dSBarry Smith   PetscFunctionReturn(0);
13646f0a148fSBarry Smith }
1365557bce09SLois Curfman McInnes 
13664a2ae208SSatish Balay #undef __FUNCT__
13674a2ae208SSatish Balay #define __FUNCT__ "MatGetArray_SeqDense"
1368dfbe8321SBarry Smith PetscErrorCode MatGetArray_SeqDense(Mat A,PetscScalar *array[])
136964e87e97SBarry Smith {
1370c0bbcb79SLois Curfman McInnes   Mat_SeqDense *mat = (Mat_SeqDense*)A->data;
13713a40ed3dSBarry Smith 
13723a40ed3dSBarry Smith   PetscFunctionBegin;
1373899cda47SBarry Smith   if (mat->lda != A->rmap.n) SETERRQ(PETSC_ERR_SUP,"Cannot get array for Dense matrices with LDA different from number of rows");
137464e87e97SBarry Smith   *array = mat->v;
13753a40ed3dSBarry Smith   PetscFunctionReturn(0);
137664e87e97SBarry Smith }
13770754003eSLois Curfman McInnes 
13784a2ae208SSatish Balay #undef __FUNCT__
13794a2ae208SSatish Balay #define __FUNCT__ "MatRestoreArray_SeqDense"
1380dfbe8321SBarry Smith PetscErrorCode MatRestoreArray_SeqDense(Mat A,PetscScalar *array[])
1381ff14e315SSatish Balay {
13823a40ed3dSBarry Smith   PetscFunctionBegin;
138309b544d4SBarry Smith   *array = 0; /* user cannot accidently use the array later */
13843a40ed3dSBarry Smith   PetscFunctionReturn(0);
1385ff14e315SSatish Balay }
13860754003eSLois Curfman McInnes 
13874a2ae208SSatish Balay #undef __FUNCT__
13884a2ae208SSatish Balay #define __FUNCT__ "MatGetSubMatrix_SeqDense"
138913f74950SBarry Smith static PetscErrorCode MatGetSubMatrix_SeqDense(Mat A,IS isrow,IS iscol,PetscInt cs,MatReuse scall,Mat *B)
13900754003eSLois Curfman McInnes {
1391c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
13926849ba73SBarry Smith   PetscErrorCode ierr;
139321a2c019SBarry Smith   PetscInt       i,j,*irow,*icol,nrows,ncols;
139487828ca2SBarry Smith   PetscScalar    *av,*bv,*v = mat->v;
13950754003eSLois Curfman McInnes   Mat            newmat;
13960754003eSLois Curfman McInnes 
13973a40ed3dSBarry Smith   PetscFunctionBegin;
139878b31e54SBarry Smith   ierr = ISGetIndices(isrow,&irow);CHKERRQ(ierr);
139978b31e54SBarry Smith   ierr = ISGetIndices(iscol,&icol);CHKERRQ(ierr);
1400e03a110bSBarry Smith   ierr = ISGetLocalSize(isrow,&nrows);CHKERRQ(ierr);
1401e03a110bSBarry Smith   ierr = ISGetLocalSize(iscol,&ncols);CHKERRQ(ierr);
14020754003eSLois Curfman McInnes 
1403182d2002SSatish Balay   /* Check submatrixcall */
1404182d2002SSatish Balay   if (scall == MAT_REUSE_MATRIX) {
140513f74950SBarry Smith     PetscInt n_cols,n_rows;
1406182d2002SSatish Balay     ierr = MatGetSize(*B,&n_rows,&n_cols);CHKERRQ(ierr);
140721a2c019SBarry Smith     if (n_rows != nrows || n_cols != ncols) {
140821a2c019SBarry Smith       /* resize the result result matrix to match number of requested rows/columns */
140921a2c019SBarry Smith       ierr = MatSetSizes(*B,nrows,nrows,nrows,nrows);CHKERRQ(ierr);
141021a2c019SBarry Smith     }
1411182d2002SSatish Balay     newmat = *B;
1412182d2002SSatish Balay   } else {
14130754003eSLois Curfman McInnes     /* Create and fill new matrix */
14147adad957SLisandro Dalcin     ierr = MatCreate(((PetscObject)A)->comm,&newmat);CHKERRQ(ierr);
1415f69a0ea3SMatthew Knepley     ierr = MatSetSizes(newmat,nrows,ncols,nrows,ncols);CHKERRQ(ierr);
14167adad957SLisandro Dalcin     ierr = MatSetType(newmat,((PetscObject)A)->type_name);CHKERRQ(ierr);
14175c5985e7SKris Buschelman     ierr = MatSeqDenseSetPreallocation(newmat,PETSC_NULL);CHKERRQ(ierr);
1418182d2002SSatish Balay   }
1419182d2002SSatish Balay 
1420182d2002SSatish Balay   /* Now extract the data pointers and do the copy,column at a time */
1421182d2002SSatish Balay   bv = ((Mat_SeqDense*)newmat->data)->v;
1422182d2002SSatish Balay 
1423182d2002SSatish Balay   for (i=0; i<ncols; i++) {
14246de62eeeSBarry Smith     av = v + mat->lda*icol[i];
1425182d2002SSatish Balay     for (j=0; j<nrows; j++) {
1426182d2002SSatish Balay       *bv++ = av[irow[j]];
14270754003eSLois Curfman McInnes     }
14280754003eSLois Curfman McInnes   }
1429182d2002SSatish Balay 
1430182d2002SSatish Balay   /* Assemble the matrices so that the correct flags are set */
14316d4a8577SBarry Smith   ierr = MatAssemblyBegin(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
14326d4a8577SBarry Smith   ierr = MatAssemblyEnd(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
14330754003eSLois Curfman McInnes 
14340754003eSLois Curfman McInnes   /* Free work space */
143578b31e54SBarry Smith   ierr = ISRestoreIndices(isrow,&irow);CHKERRQ(ierr);
143678b31e54SBarry Smith   ierr = ISRestoreIndices(iscol,&icol);CHKERRQ(ierr);
1437182d2002SSatish Balay   *B = newmat;
14383a40ed3dSBarry Smith   PetscFunctionReturn(0);
14390754003eSLois Curfman McInnes }
14400754003eSLois Curfman McInnes 
14414a2ae208SSatish Balay #undef __FUNCT__
14424a2ae208SSatish Balay #define __FUNCT__ "MatGetSubMatrices_SeqDense"
144313f74950SBarry Smith PetscErrorCode MatGetSubMatrices_SeqDense(Mat A,PetscInt n,const IS irow[],const IS icol[],MatReuse scall,Mat *B[])
1444905e6a2fSBarry Smith {
14456849ba73SBarry Smith   PetscErrorCode ierr;
144613f74950SBarry Smith   PetscInt       i;
1447905e6a2fSBarry Smith 
14483a40ed3dSBarry Smith   PetscFunctionBegin;
1449905e6a2fSBarry Smith   if (scall == MAT_INITIAL_MATRIX) {
1450b0a32e0cSBarry Smith     ierr = PetscMalloc((n+1)*sizeof(Mat),B);CHKERRQ(ierr);
1451905e6a2fSBarry Smith   }
1452905e6a2fSBarry Smith 
1453905e6a2fSBarry Smith   for (i=0; i<n; i++) {
14546a6a5d1dSBarry Smith     ierr = MatGetSubMatrix_SeqDense(A,irow[i],icol[i],PETSC_DECIDE,scall,&(*B)[i]);CHKERRQ(ierr);
1455905e6a2fSBarry Smith   }
14563a40ed3dSBarry Smith   PetscFunctionReturn(0);
1457905e6a2fSBarry Smith }
1458905e6a2fSBarry Smith 
14594a2ae208SSatish Balay #undef __FUNCT__
1460c0aa2d19SHong Zhang #define __FUNCT__ "MatAssemblyBegin_SeqDense"
1461c0aa2d19SHong Zhang PetscErrorCode MatAssemblyBegin_SeqDense(Mat mat,MatAssemblyType mode)
1462c0aa2d19SHong Zhang {
1463c0aa2d19SHong Zhang   PetscFunctionBegin;
1464c0aa2d19SHong Zhang   PetscFunctionReturn(0);
1465c0aa2d19SHong Zhang }
1466c0aa2d19SHong Zhang 
1467c0aa2d19SHong Zhang #undef __FUNCT__
1468c0aa2d19SHong Zhang #define __FUNCT__ "MatAssemblyEnd_SeqDense"
1469c0aa2d19SHong Zhang PetscErrorCode MatAssemblyEnd_SeqDense(Mat mat,MatAssemblyType mode)
1470c0aa2d19SHong Zhang {
1471c0aa2d19SHong Zhang   PetscFunctionBegin;
1472c0aa2d19SHong Zhang   PetscFunctionReturn(0);
1473c0aa2d19SHong Zhang }
1474c0aa2d19SHong Zhang 
1475c0aa2d19SHong Zhang #undef __FUNCT__
14764a2ae208SSatish Balay #define __FUNCT__ "MatCopy_SeqDense"
1477dfbe8321SBarry Smith PetscErrorCode MatCopy_SeqDense(Mat A,Mat B,MatStructure str)
14784b0e389bSBarry Smith {
14794b0e389bSBarry Smith   Mat_SeqDense   *a = (Mat_SeqDense*)A->data,*b = (Mat_SeqDense *)B->data;
14806849ba73SBarry Smith   PetscErrorCode ierr;
1481899cda47SBarry Smith   PetscInt       lda1=a->lda,lda2=b->lda, m=A->rmap.n,n=A->cmap.n, j;
14823a40ed3dSBarry Smith 
14833a40ed3dSBarry Smith   PetscFunctionBegin;
148433f4a19fSKris Buschelman   /* If the two matrices don't have the same copy implementation, they aren't compatible for fast copy. */
148533f4a19fSKris Buschelman   if (A->ops->copy != B->ops->copy) {
1486cb5b572fSBarry Smith     ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr);
14873a40ed3dSBarry Smith     PetscFunctionReturn(0);
14883a40ed3dSBarry Smith   }
1489899cda47SBarry Smith   if (m != B->rmap.n || n != B->cmap.n) SETERRQ(PETSC_ERR_ARG_SIZ,"size(B) != size(A)");
1490a5ce6ee0Svictorle   if (lda1>m || lda2>m) {
14910dbb7854Svictorle     for (j=0; j<n; j++) {
1492a5ce6ee0Svictorle       ierr = PetscMemcpy(b->v+j*lda2,a->v+j*lda1,m*sizeof(PetscScalar));CHKERRQ(ierr);
1493a5ce6ee0Svictorle     }
1494a5ce6ee0Svictorle   } else {
1495899cda47SBarry Smith     ierr = PetscMemcpy(b->v,a->v,A->rmap.n*A->cmap.n*sizeof(PetscScalar));CHKERRQ(ierr);
1496a5ce6ee0Svictorle   }
1497273d9f13SBarry Smith   PetscFunctionReturn(0);
1498273d9f13SBarry Smith }
1499273d9f13SBarry Smith 
15004a2ae208SSatish Balay #undef __FUNCT__
15014a2ae208SSatish Balay #define __FUNCT__ "MatSetUpPreallocation_SeqDense"
1502dfbe8321SBarry Smith PetscErrorCode MatSetUpPreallocation_SeqDense(Mat A)
1503273d9f13SBarry Smith {
1504dfbe8321SBarry Smith   PetscErrorCode ierr;
1505273d9f13SBarry Smith 
1506273d9f13SBarry Smith   PetscFunctionBegin;
1507273d9f13SBarry Smith   ierr =  MatSeqDenseSetPreallocation(A,0);CHKERRQ(ierr);
15083a40ed3dSBarry Smith   PetscFunctionReturn(0);
15094b0e389bSBarry Smith }
15104b0e389bSBarry Smith 
1511284134d9SBarry Smith #undef __FUNCT__
1512284134d9SBarry Smith #define __FUNCT__ "MatSetSizes_SeqDense"
1513284134d9SBarry Smith PetscErrorCode MatSetSizes_SeqDense(Mat A,PetscInt m,PetscInt n,PetscInt M,PetscInt N)
1514284134d9SBarry Smith {
1515284134d9SBarry Smith   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
151621a2c019SBarry Smith   PetscErrorCode ierr;
1517284134d9SBarry Smith   PetscFunctionBegin;
151821a2c019SBarry Smith   /* this will not be called before lda, Mmax,  and Nmax have been set */
1519284134d9SBarry Smith   m = PetscMax(m,M);
1520284134d9SBarry Smith   n = PetscMax(n,N);
152121a2c019SBarry 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);
1522284134d9SBarry 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);
1523899cda47SBarry Smith   A->rmap.n = A->rmap.n = m;
1524899cda47SBarry Smith   A->cmap.n = A->cmap.N = n;
152521a2c019SBarry Smith   if (a->changelda) a->lda = m;
152621a2c019SBarry Smith   ierr = PetscMemzero(a->v,m*n*sizeof(PetscScalar));CHKERRQ(ierr);
1527284134d9SBarry Smith   PetscFunctionReturn(0);
1528284134d9SBarry Smith }
1529170fe5c8SBarry Smith 
1530284134d9SBarry Smith 
1531a9fe9ddaSSatish Balay /* ----------------------------------------------------------------*/
1532a9fe9ddaSSatish Balay #undef __FUNCT__
1533a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMult_SeqDense_SeqDense"
1534a9fe9ddaSSatish Balay PetscErrorCode MatMatMult_SeqDense_SeqDense(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C)
1535a9fe9ddaSSatish Balay {
1536a9fe9ddaSSatish Balay   PetscErrorCode ierr;
1537a9fe9ddaSSatish Balay 
1538a9fe9ddaSSatish Balay   PetscFunctionBegin;
1539a9fe9ddaSSatish Balay   if (scall == MAT_INITIAL_MATRIX){
1540a9fe9ddaSSatish Balay     ierr = MatMatMultSymbolic_SeqDense_SeqDense(A,B,fill,C);CHKERRQ(ierr);
1541a9fe9ddaSSatish Balay   }
1542a9fe9ddaSSatish Balay   ierr = MatMatMultNumeric_SeqDense_SeqDense(A,B,*C);CHKERRQ(ierr);
1543a9fe9ddaSSatish Balay   PetscFunctionReturn(0);
1544a9fe9ddaSSatish Balay }
1545a9fe9ddaSSatish Balay 
1546a9fe9ddaSSatish Balay #undef __FUNCT__
1547a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMultSymbolic_SeqDense_SeqDense"
1548a9fe9ddaSSatish Balay PetscErrorCode MatMatMultSymbolic_SeqDense_SeqDense(Mat A,Mat B,PetscReal fill,Mat *C)
1549a9fe9ddaSSatish Balay {
1550ee16a9a1SHong Zhang   PetscErrorCode ierr;
1551899cda47SBarry Smith   PetscInt       m=A->rmap.n,n=B->cmap.n;
1552ee16a9a1SHong Zhang   Mat            Cmat;
1553a9fe9ddaSSatish Balay 
1554ee16a9a1SHong Zhang   PetscFunctionBegin;
1555899cda47SBarry 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);
1556ee16a9a1SHong Zhang   ierr = MatCreate(PETSC_COMM_SELF,&Cmat);CHKERRQ(ierr);
1557ee16a9a1SHong Zhang   ierr = MatSetSizes(Cmat,m,n,m,n);CHKERRQ(ierr);
1558ee16a9a1SHong Zhang   ierr = MatSetType(Cmat,MATSEQDENSE);CHKERRQ(ierr);
1559ee16a9a1SHong Zhang   ierr = MatSeqDenseSetPreallocation(Cmat,PETSC_NULL);CHKERRQ(ierr);
1560ee16a9a1SHong Zhang   Cmat->assembled = PETSC_TRUE;
1561ee16a9a1SHong Zhang   *C = Cmat;
1562ee16a9a1SHong Zhang   PetscFunctionReturn(0);
1563ee16a9a1SHong Zhang }
1564a9fe9ddaSSatish Balay 
156598a3b096SSatish Balay #undef __FUNCT__
1566a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMultNumeric_SeqDense_SeqDense"
1567a9fe9ddaSSatish Balay PetscErrorCode MatMatMultNumeric_SeqDense_SeqDense(Mat A,Mat B,Mat C)
1568a9fe9ddaSSatish Balay {
1569a9fe9ddaSSatish Balay   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
1570a9fe9ddaSSatish Balay   Mat_SeqDense   *b = (Mat_SeqDense*)B->data;
1571a9fe9ddaSSatish Balay   Mat_SeqDense   *c = (Mat_SeqDense*)C->data;
15720805154bSBarry Smith   PetscBLASInt   m,n,k;
1573a9fe9ddaSSatish Balay   PetscScalar    _DOne=1.0,_DZero=0.0;
1574a9fe9ddaSSatish Balay 
1575a9fe9ddaSSatish Balay   PetscFunctionBegin;
15760805154bSBarry Smith   m = PetscBLASIntCast(A->rmap.n);
15770805154bSBarry Smith   n = PetscBLASIntCast(B->cmap.n);
15780805154bSBarry Smith   k = PetscBLASIntCast(A->cmap.n);
1579a9fe9ddaSSatish Balay   BLASgemm_("N","N",&m,&n,&k,&_DOne,a->v,&a->lda,b->v,&b->lda,&_DZero,c->v,&c->lda);
1580a9fe9ddaSSatish Balay   PetscFunctionReturn(0);
1581a9fe9ddaSSatish Balay }
1582a9fe9ddaSSatish Balay 
1583a9fe9ddaSSatish Balay #undef __FUNCT__
1584a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMultTranspose_SeqDense_SeqDense"
1585a9fe9ddaSSatish Balay PetscErrorCode MatMatMultTranspose_SeqDense_SeqDense(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C)
1586a9fe9ddaSSatish Balay {
1587a9fe9ddaSSatish Balay   PetscErrorCode ierr;
1588a9fe9ddaSSatish Balay 
1589a9fe9ddaSSatish Balay   PetscFunctionBegin;
1590a9fe9ddaSSatish Balay   if (scall == MAT_INITIAL_MATRIX){
1591a9fe9ddaSSatish Balay     ierr = MatMatMultTransposeSymbolic_SeqDense_SeqDense(A,B,fill,C);CHKERRQ(ierr);
1592a9fe9ddaSSatish Balay   }
1593a9fe9ddaSSatish Balay   ierr = MatMatMultTransposeNumeric_SeqDense_SeqDense(A,B,*C);CHKERRQ(ierr);
1594a9fe9ddaSSatish Balay   PetscFunctionReturn(0);
1595a9fe9ddaSSatish Balay }
1596a9fe9ddaSSatish Balay 
1597a9fe9ddaSSatish Balay #undef __FUNCT__
1598a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMultTransposeSymbolic_SeqDense_SeqDense"
1599a9fe9ddaSSatish Balay PetscErrorCode MatMatMultTransposeSymbolic_SeqDense_SeqDense(Mat A,Mat B,PetscReal fill,Mat *C)
1600a9fe9ddaSSatish Balay {
1601ee16a9a1SHong Zhang   PetscErrorCode ierr;
1602899cda47SBarry Smith   PetscInt       m=A->cmap.n,n=B->cmap.n;
1603ee16a9a1SHong Zhang   Mat            Cmat;
1604a9fe9ddaSSatish Balay 
1605ee16a9a1SHong Zhang   PetscFunctionBegin;
1606899cda47SBarry 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);
1607ee16a9a1SHong Zhang   ierr = MatCreate(PETSC_COMM_SELF,&Cmat);CHKERRQ(ierr);
1608ee16a9a1SHong Zhang   ierr = MatSetSizes(Cmat,m,n,m,n);CHKERRQ(ierr);
1609ee16a9a1SHong Zhang   ierr = MatSetType(Cmat,MATSEQDENSE);CHKERRQ(ierr);
1610ee16a9a1SHong Zhang   ierr = MatSeqDenseSetPreallocation(Cmat,PETSC_NULL);CHKERRQ(ierr);
1611ee16a9a1SHong Zhang   Cmat->assembled = PETSC_TRUE;
1612ee16a9a1SHong Zhang   *C = Cmat;
1613ee16a9a1SHong Zhang   PetscFunctionReturn(0);
1614ee16a9a1SHong Zhang }
1615a9fe9ddaSSatish Balay 
1616a9fe9ddaSSatish Balay #undef __FUNCT__
1617a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMultTransposeNumeric_SeqDense_SeqDense"
1618a9fe9ddaSSatish Balay PetscErrorCode MatMatMultTransposeNumeric_SeqDense_SeqDense(Mat A,Mat B,Mat C)
1619a9fe9ddaSSatish Balay {
1620a9fe9ddaSSatish Balay   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
1621a9fe9ddaSSatish Balay   Mat_SeqDense   *b = (Mat_SeqDense*)B->data;
1622a9fe9ddaSSatish Balay   Mat_SeqDense   *c = (Mat_SeqDense*)C->data;
16230805154bSBarry Smith   PetscBLASInt   m,n,k;
1624a9fe9ddaSSatish Balay   PetscScalar    _DOne=1.0,_DZero=0.0;
1625a9fe9ddaSSatish Balay 
1626a9fe9ddaSSatish Balay   PetscFunctionBegin;
16272fbe02b9SBarry Smith   m = PetscBLASIntCast(A->cmap.n);
16280805154bSBarry Smith   n = PetscBLASIntCast(B->cmap.n);
16292fbe02b9SBarry Smith   k = PetscBLASIntCast(A->rmap.n);
16302fbe02b9SBarry Smith   /*
16312fbe02b9SBarry Smith      Note the m and n arguments below are the number rows and columns of A', not A!
16322fbe02b9SBarry Smith   */
1633a9fe9ddaSSatish Balay   BLASgemm_("T","N",&m,&n,&k,&_DOne,a->v,&a->lda,b->v,&b->lda,&_DZero,c->v,&c->lda);
1634a9fe9ddaSSatish Balay   PetscFunctionReturn(0);
1635a9fe9ddaSSatish Balay }
1636985db425SBarry Smith 
1637985db425SBarry Smith #undef __FUNCT__
1638985db425SBarry Smith #define __FUNCT__ "MatGetRowMax_SeqDense"
1639985db425SBarry Smith PetscErrorCode MatGetRowMax_SeqDense(Mat A,Vec v,PetscInt idx[])
1640985db425SBarry Smith {
1641985db425SBarry Smith   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
1642985db425SBarry Smith   PetscErrorCode ierr;
1643985db425SBarry Smith   PetscInt       i,j,m = A->rmap.n,n = A->cmap.n,p;
1644985db425SBarry Smith   PetscScalar    *x;
1645985db425SBarry Smith   MatScalar      *aa = a->v;
1646985db425SBarry Smith 
1647985db425SBarry Smith   PetscFunctionBegin;
1648985db425SBarry Smith   if (A->factor) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
1649985db425SBarry Smith 
1650985db425SBarry Smith   ierr = VecSet(v,0.0);CHKERRQ(ierr);
1651985db425SBarry Smith   ierr = VecGetArray(v,&x);CHKERRQ(ierr);
1652985db425SBarry Smith   ierr = VecGetLocalSize(v,&p);CHKERRQ(ierr);
1653985db425SBarry Smith   if (p != A->rmap.n) SETERRQ(PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector");
1654985db425SBarry Smith   for (i=0; i<m; i++) {
1655985db425SBarry Smith     x[i] = aa[i]; if (idx) idx[i] = 0;
1656985db425SBarry Smith     for (j=1; j<n; j++){
1657985db425SBarry Smith       if (PetscRealPart(x[i]) < PetscRealPart(aa[i+m*j])) {x[i] = aa[i + m*j]; if (idx) idx[i] = j;}
1658985db425SBarry Smith     }
1659985db425SBarry Smith   }
1660985db425SBarry Smith   ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
1661985db425SBarry Smith   PetscFunctionReturn(0);
1662985db425SBarry Smith }
1663985db425SBarry Smith 
1664985db425SBarry Smith #undef __FUNCT__
1665985db425SBarry Smith #define __FUNCT__ "MatGetRowMaxAbs_SeqDense"
1666985db425SBarry Smith PetscErrorCode MatGetRowMaxAbs_SeqDense(Mat A,Vec v,PetscInt idx[])
1667985db425SBarry Smith {
1668985db425SBarry Smith   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
1669985db425SBarry Smith   PetscErrorCode ierr;
1670985db425SBarry Smith   PetscInt       i,j,m = A->rmap.n,n = A->cmap.n,p;
1671985db425SBarry Smith   PetscScalar    *x;
1672985db425SBarry Smith   PetscReal      atmp;
1673985db425SBarry Smith   MatScalar      *aa = a->v;
1674985db425SBarry Smith 
1675985db425SBarry Smith   PetscFunctionBegin;
1676985db425SBarry Smith   if (A->factor) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
1677985db425SBarry Smith 
1678985db425SBarry Smith   ierr = VecSet(v,0.0);CHKERRQ(ierr);
1679985db425SBarry Smith   ierr = VecGetArray(v,&x);CHKERRQ(ierr);
1680985db425SBarry Smith   ierr = VecGetLocalSize(v,&p);CHKERRQ(ierr);
1681985db425SBarry Smith   if (p != A->rmap.n) SETERRQ(PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector");
1682985db425SBarry Smith   for (i=0; i<m; i++) {
16839189402eSHong Zhang     x[i] = PetscAbsScalar(aa[i]);
1684985db425SBarry Smith     for (j=1; j<n; j++){
1685985db425SBarry Smith       atmp = PetscAbsScalar(aa[i+m*j]);
1686985db425SBarry Smith       if (PetscAbsScalar(x[i]) < atmp) {x[i] = atmp; if (idx) idx[i] = j;}
1687985db425SBarry Smith     }
1688985db425SBarry Smith   }
1689985db425SBarry Smith   ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
1690985db425SBarry Smith   PetscFunctionReturn(0);
1691985db425SBarry Smith }
1692985db425SBarry Smith 
1693985db425SBarry Smith #undef __FUNCT__
1694985db425SBarry Smith #define __FUNCT__ "MatGetRowMin_SeqDense"
1695985db425SBarry Smith PetscErrorCode MatGetRowMin_SeqDense(Mat A,Vec v,PetscInt idx[])
1696985db425SBarry Smith {
1697985db425SBarry Smith   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
1698985db425SBarry Smith   PetscErrorCode ierr;
1699985db425SBarry Smith   PetscInt       i,j,m = A->rmap.n,n = A->cmap.n,p;
1700985db425SBarry Smith   PetscScalar    *x;
1701985db425SBarry Smith   MatScalar      *aa = a->v;
1702985db425SBarry Smith 
1703985db425SBarry Smith   PetscFunctionBegin;
1704985db425SBarry Smith   if (A->factor) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
1705985db425SBarry Smith 
1706985db425SBarry Smith   ierr = VecSet(v,0.0);CHKERRQ(ierr);
1707985db425SBarry Smith   ierr = VecGetArray(v,&x);CHKERRQ(ierr);
1708985db425SBarry Smith   ierr = VecGetLocalSize(v,&p);CHKERRQ(ierr);
1709985db425SBarry Smith   if (p != A->rmap.n) SETERRQ(PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector");
1710985db425SBarry Smith   for (i=0; i<m; i++) {
1711985db425SBarry Smith     x[i] = aa[i]; if (idx) idx[i] = 0;
1712985db425SBarry Smith     for (j=1; j<n; j++){
1713985db425SBarry Smith       if (PetscRealPart(x[i]) > PetscRealPart(aa[i+m*j])) {x[i] = aa[i + m*j]; if (idx) idx[i] = j;}
1714985db425SBarry Smith     }
1715985db425SBarry Smith   }
1716985db425SBarry Smith   ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
1717985db425SBarry Smith   PetscFunctionReturn(0);
1718985db425SBarry Smith }
1719985db425SBarry Smith 
17208d0534beSBarry Smith #undef __FUNCT__
17218d0534beSBarry Smith #define __FUNCT__ "MatGetColumnVector_SeqDense"
17228d0534beSBarry Smith PetscErrorCode MatGetColumnVector_SeqDense(Mat A,Vec v,PetscInt col)
17238d0534beSBarry Smith {
17248d0534beSBarry Smith   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
17258d0534beSBarry Smith   PetscErrorCode ierr;
17268d0534beSBarry Smith   PetscScalar    *x;
17278d0534beSBarry Smith 
17288d0534beSBarry Smith   PetscFunctionBegin;
17298d0534beSBarry Smith   if (A->factor) SETERRQ(PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
17308d0534beSBarry Smith 
17318d0534beSBarry Smith   ierr = VecGetArray(v,&x);CHKERRQ(ierr);
17328d0534beSBarry Smith   ierr = PetscMemcpy(x,a->v+col*a->lda,A->rmap.n*sizeof(PetscScalar));CHKERRQ(ierr);
17338d0534beSBarry Smith   ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
17348d0534beSBarry Smith   PetscFunctionReturn(0);
17358d0534beSBarry Smith }
17368d0534beSBarry Smith 
1737289bc588SBarry Smith /* -------------------------------------------------------------------*/
1738a5ae1ecdSBarry Smith static struct _MatOps MatOps_Values = {MatSetValues_SeqDense,
1739905e6a2fSBarry Smith        MatGetRow_SeqDense,
1740905e6a2fSBarry Smith        MatRestoreRow_SeqDense,
1741905e6a2fSBarry Smith        MatMult_SeqDense,
174297304618SKris Buschelman /* 4*/ MatMultAdd_SeqDense,
17437c922b88SBarry Smith        MatMultTranspose_SeqDense,
17447c922b88SBarry Smith        MatMultTransposeAdd_SeqDense,
1745905e6a2fSBarry Smith        MatSolve_SeqDense,
1746905e6a2fSBarry Smith        MatSolveAdd_SeqDense,
17477c922b88SBarry Smith        MatSolveTranspose_SeqDense,
174897304618SKris Buschelman /*10*/ MatSolveTransposeAdd_SeqDense,
1749905e6a2fSBarry Smith        MatLUFactor_SeqDense,
1750905e6a2fSBarry Smith        MatCholeskyFactor_SeqDense,
1751ec8511deSBarry Smith        MatRelax_SeqDense,
1752ec8511deSBarry Smith        MatTranspose_SeqDense,
175397304618SKris Buschelman /*15*/ MatGetInfo_SeqDense,
1754905e6a2fSBarry Smith        MatEqual_SeqDense,
1755905e6a2fSBarry Smith        MatGetDiagonal_SeqDense,
1756905e6a2fSBarry Smith        MatDiagonalScale_SeqDense,
1757905e6a2fSBarry Smith        MatNorm_SeqDense,
1758c0aa2d19SHong Zhang /*20*/ MatAssemblyBegin_SeqDense,
1759c0aa2d19SHong Zhang        MatAssemblyEnd_SeqDense,
1760905e6a2fSBarry Smith        0,
1761905e6a2fSBarry Smith        MatSetOption_SeqDense,
1762905e6a2fSBarry Smith        MatZeroEntries_SeqDense,
176397304618SKris Buschelman /*25*/ MatZeroRows_SeqDense,
1764905e6a2fSBarry Smith        MatLUFactorSymbolic_SeqDense,
1765905e6a2fSBarry Smith        MatLUFactorNumeric_SeqDense,
1766905e6a2fSBarry Smith        MatCholeskyFactorSymbolic_SeqDense,
1767905e6a2fSBarry Smith        MatCholeskyFactorNumeric_SeqDense,
176897304618SKris Buschelman /*30*/ MatSetUpPreallocation_SeqDense,
1769273d9f13SBarry Smith        0,
1770905e6a2fSBarry Smith        0,
1771905e6a2fSBarry Smith        MatGetArray_SeqDense,
1772905e6a2fSBarry Smith        MatRestoreArray_SeqDense,
177397304618SKris Buschelman /*35*/ MatDuplicate_SeqDense,
1774a5ae1ecdSBarry Smith        0,
1775a5ae1ecdSBarry Smith        0,
1776a5ae1ecdSBarry Smith        0,
1777a5ae1ecdSBarry Smith        0,
177897304618SKris Buschelman /*40*/ MatAXPY_SeqDense,
1779a5ae1ecdSBarry Smith        MatGetSubMatrices_SeqDense,
1780a5ae1ecdSBarry Smith        0,
17814b0e389bSBarry Smith        MatGetValues_SeqDense,
1782a5ae1ecdSBarry Smith        MatCopy_SeqDense,
1783985db425SBarry Smith /*45*/ MatGetRowMax_SeqDense,
1784a5ae1ecdSBarry Smith        MatScale_SeqDense,
1785a5ae1ecdSBarry Smith        0,
1786a5ae1ecdSBarry Smith        0,
1787a5ae1ecdSBarry Smith        0,
1788521d7252SBarry Smith /*50*/ 0,
1789a5ae1ecdSBarry Smith        0,
1790a5ae1ecdSBarry Smith        0,
1791a5ae1ecdSBarry Smith        0,
1792a5ae1ecdSBarry Smith        0,
179397304618SKris Buschelman /*55*/ 0,
1794a5ae1ecdSBarry Smith        0,
1795a5ae1ecdSBarry Smith        0,
1796a5ae1ecdSBarry Smith        0,
1797a5ae1ecdSBarry Smith        0,
179897304618SKris Buschelman /*60*/ 0,
1799e03a110bSBarry Smith        MatDestroy_SeqDense,
1800e03a110bSBarry Smith        MatView_SeqDense,
1801357abbc8SBarry Smith        0,
180297304618SKris Buschelman        0,
180397304618SKris Buschelman /*65*/ 0,
180497304618SKris Buschelman        0,
180597304618SKris Buschelman        0,
180697304618SKris Buschelman        0,
180797304618SKris Buschelman        0,
1808985db425SBarry Smith /*70*/ MatGetRowMaxAbs_SeqDense,
180997304618SKris Buschelman        0,
181097304618SKris Buschelman        0,
181197304618SKris Buschelman        0,
181297304618SKris Buschelman        0,
181397304618SKris Buschelman /*75*/ 0,
181497304618SKris Buschelman        0,
181597304618SKris Buschelman        0,
181697304618SKris Buschelman        0,
181797304618SKris Buschelman        0,
181897304618SKris Buschelman /*80*/ 0,
181997304618SKris Buschelman        0,
182097304618SKris Buschelman        0,
182197304618SKris Buschelman        0,
1822865e5f61SKris Buschelman /*84*/ MatLoad_SeqDense,
1823865e5f61SKris Buschelman        0,
18241cbb95d3SBarry Smith        MatIsHermitian_SeqDense,
1825865e5f61SKris Buschelman        0,
1826865e5f61SKris Buschelman        0,
1827865e5f61SKris Buschelman        0,
1828a9fe9ddaSSatish Balay /*90*/ MatMatMult_SeqDense_SeqDense,
1829a9fe9ddaSSatish Balay        MatMatMultSymbolic_SeqDense_SeqDense,
1830a9fe9ddaSSatish Balay        MatMatMultNumeric_SeqDense_SeqDense,
1831865e5f61SKris Buschelman        0,
1832865e5f61SKris Buschelman        0,
1833865e5f61SKris Buschelman /*95*/ 0,
1834a9fe9ddaSSatish Balay        MatMatMultTranspose_SeqDense_SeqDense,
1835a9fe9ddaSSatish Balay        MatMatMultTransposeSymbolic_SeqDense_SeqDense,
1836a9fe9ddaSSatish Balay        MatMatMultTransposeNumeric_SeqDense_SeqDense,
1837284134d9SBarry Smith        0,
1838284134d9SBarry Smith /*100*/0,
1839284134d9SBarry Smith        0,
1840284134d9SBarry Smith        0,
1841284134d9SBarry Smith        0,
1842985db425SBarry Smith        MatSetSizes_SeqDense,
1843985db425SBarry Smith        0,
1844985db425SBarry Smith        0,
1845985db425SBarry Smith        0,
1846985db425SBarry Smith        0,
1847985db425SBarry Smith        0,
1848985db425SBarry Smith /*110*/0,
1849985db425SBarry Smith        0,
18508d0534beSBarry Smith        MatGetRowMin_SeqDense,
18518d0534beSBarry Smith        MatGetColumnVector_SeqDense
1852985db425SBarry Smith };
185390ace30eSBarry Smith 
18544a2ae208SSatish Balay #undef __FUNCT__
18554a2ae208SSatish Balay #define __FUNCT__ "MatCreateSeqDense"
18564b828684SBarry Smith /*@C
1857fafbff53SBarry Smith    MatCreateSeqDense - Creates a sequential dense matrix that
1858d65003e9SLois Curfman McInnes    is stored in column major order (the usual Fortran 77 manner). Many
1859d65003e9SLois Curfman McInnes    of the matrix operations use the BLAS and LAPACK routines.
1860289bc588SBarry Smith 
1861db81eaa0SLois Curfman McInnes    Collective on MPI_Comm
1862db81eaa0SLois Curfman McInnes 
186320563c6bSBarry Smith    Input Parameters:
1864db81eaa0SLois Curfman McInnes +  comm - MPI communicator, set to PETSC_COMM_SELF
18650c775827SLois Curfman McInnes .  m - number of rows
186618f449edSLois Curfman McInnes .  n - number of columns
1867db81eaa0SLois Curfman McInnes -  data - optional location of matrix data.  Set data=PETSC_NULL for PETSc
1868dfc5480cSLois Curfman McInnes    to control all matrix memory allocation.
186920563c6bSBarry Smith 
187020563c6bSBarry Smith    Output Parameter:
187144cd7ae7SLois Curfman McInnes .  A - the matrix
187220563c6bSBarry Smith 
1873b259b22eSLois Curfman McInnes    Notes:
187418f449edSLois Curfman McInnes    The data input variable is intended primarily for Fortran programmers
187518f449edSLois Curfman McInnes    who wish to allocate their own matrix memory space.  Most users should
1876b4fd4287SBarry Smith    set data=PETSC_NULL.
187718f449edSLois Curfman McInnes 
1878027ccd11SLois Curfman McInnes    Level: intermediate
1879027ccd11SLois Curfman McInnes 
1880dbd7a890SLois Curfman McInnes .keywords: dense, matrix, LAPACK, BLAS
1881d65003e9SLois Curfman McInnes 
1882db81eaa0SLois Curfman McInnes .seealso: MatCreate(), MatCreateMPIDense(), MatSetValues()
188320563c6bSBarry Smith @*/
1884be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreateSeqDense(MPI_Comm comm,PetscInt m,PetscInt n,PetscScalar *data,Mat *A)
1885289bc588SBarry Smith {
1886dfbe8321SBarry Smith   PetscErrorCode ierr;
18873b2fbd54SBarry Smith 
18883a40ed3dSBarry Smith   PetscFunctionBegin;
1889f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,A);CHKERRQ(ierr);
1890f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*A,m,n,m,n);CHKERRQ(ierr);
1891273d9f13SBarry Smith   ierr = MatSetType(*A,MATSEQDENSE);CHKERRQ(ierr);
1892273d9f13SBarry Smith   ierr = MatSeqDenseSetPreallocation(*A,data);CHKERRQ(ierr);
1893273d9f13SBarry Smith   PetscFunctionReturn(0);
1894273d9f13SBarry Smith }
1895273d9f13SBarry Smith 
18964a2ae208SSatish Balay #undef __FUNCT__
1897afc30d2aSLisandro Dalcin #define __FUNCT__ "MatSeqDenseSetPreallocation"
1898273d9f13SBarry Smith /*@C
1899273d9f13SBarry Smith    MatSeqDenseSetPreallocation - Sets the array used for storing the matrix elements
1900273d9f13SBarry Smith 
1901273d9f13SBarry Smith    Collective on MPI_Comm
1902273d9f13SBarry Smith 
1903273d9f13SBarry Smith    Input Parameters:
1904273d9f13SBarry Smith +  A - the matrix
1905273d9f13SBarry Smith -  data - the array (or PETSC_NULL)
1906273d9f13SBarry Smith 
1907273d9f13SBarry Smith    Notes:
1908273d9f13SBarry Smith    The data input variable is intended primarily for Fortran programmers
1909273d9f13SBarry Smith    who wish to allocate their own matrix memory space.  Most users should
1910284134d9SBarry Smith    need not call this routine.
1911273d9f13SBarry Smith 
1912273d9f13SBarry Smith    Level: intermediate
1913273d9f13SBarry Smith 
1914273d9f13SBarry Smith .keywords: dense, matrix, LAPACK, BLAS
1915273d9f13SBarry Smith 
1916273d9f13SBarry Smith .seealso: MatCreate(), MatCreateMPIDense(), MatSetValues()
1917273d9f13SBarry Smith @*/
1918be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatSeqDenseSetPreallocation(Mat B,PetscScalar data[])
1919273d9f13SBarry Smith {
1920dfbe8321SBarry Smith   PetscErrorCode ierr,(*f)(Mat,PetscScalar[]);
1921a23d5eceSKris Buschelman 
1922a23d5eceSKris Buschelman   PetscFunctionBegin;
1923a23d5eceSKris Buschelman   ierr = PetscObjectQueryFunction((PetscObject)B,"MatSeqDenseSetPreallocation_C",(void (**)(void))&f);CHKERRQ(ierr);
1924a23d5eceSKris Buschelman   if (f) {
1925a23d5eceSKris Buschelman     ierr = (*f)(B,data);CHKERRQ(ierr);
1926a23d5eceSKris Buschelman   }
1927a23d5eceSKris Buschelman   PetscFunctionReturn(0);
1928a23d5eceSKris Buschelman }
1929a23d5eceSKris Buschelman 
1930a23d5eceSKris Buschelman EXTERN_C_BEGIN
1931a23d5eceSKris Buschelman #undef __FUNCT__
1932afc30d2aSLisandro Dalcin #define __FUNCT__ "MatSeqDenseSetPreallocation_SeqDense"
1933be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatSeqDenseSetPreallocation_SeqDense(Mat B,PetscScalar *data)
1934a23d5eceSKris Buschelman {
1935273d9f13SBarry Smith   Mat_SeqDense   *b;
1936dfbe8321SBarry Smith   PetscErrorCode ierr;
1937273d9f13SBarry Smith 
1938273d9f13SBarry Smith   PetscFunctionBegin;
1939273d9f13SBarry Smith   B->preallocated = PETSC_TRUE;
1940273d9f13SBarry Smith   b               = (Mat_SeqDense*)B->data;
19415afd5e0cSBarry Smith   if (b->lda <= 0) b->lda = B->rmap.n;
19429e8f95c4SLisandro Dalcin   if (!data) { /* petsc-allocated storage */
19439e8f95c4SLisandro Dalcin     if (!b->user_alloc) { ierr = PetscFree(b->v);CHKERRQ(ierr); }
19445afd5e0cSBarry Smith     ierr = PetscMalloc(b->lda*b->Nmax*sizeof(PetscScalar),&b->v);CHKERRQ(ierr);
1945284134d9SBarry Smith     ierr = PetscMemzero(b->v,b->lda*b->Nmax*sizeof(PetscScalar));CHKERRQ(ierr);
1946284134d9SBarry Smith     ierr = PetscLogObjectMemory(B,b->lda*b->Nmax*sizeof(PetscScalar));CHKERRQ(ierr);
19479e8f95c4SLisandro Dalcin     b->user_alloc = PETSC_FALSE;
1948273d9f13SBarry Smith   } else { /* user-allocated storage */
19499e8f95c4SLisandro Dalcin     if (!b->user_alloc) { ierr = PetscFree(b->v);CHKERRQ(ierr); }
1950273d9f13SBarry Smith     b->v          = data;
1951273d9f13SBarry Smith     b->user_alloc = PETSC_TRUE;
1952273d9f13SBarry Smith   }
1953273d9f13SBarry Smith   PetscFunctionReturn(0);
1954273d9f13SBarry Smith }
1955a23d5eceSKris Buschelman EXTERN_C_END
1956273d9f13SBarry Smith 
19571b807ce4Svictorle #undef __FUNCT__
19581b807ce4Svictorle #define __FUNCT__ "MatSeqDenseSetLDA"
19591b807ce4Svictorle /*@C
19601b807ce4Svictorle   MatSeqDenseSetLDA - Declare the leading dimension of the user-provided array
19611b807ce4Svictorle 
19621b807ce4Svictorle   Input parameter:
19631b807ce4Svictorle + A - the matrix
19641b807ce4Svictorle - lda - the leading dimension
19651b807ce4Svictorle 
19661b807ce4Svictorle   Notes:
19671b807ce4Svictorle   This routine is to be used in conjunction with MatSeqDenseSetPreallocation;
19681b807ce4Svictorle   it asserts that the preallocation has a leading dimension (the LDA parameter
19691b807ce4Svictorle   of Blas and Lapack fame) larger than M, the first dimension of the matrix.
19701b807ce4Svictorle 
19711b807ce4Svictorle   Level: intermediate
19721b807ce4Svictorle 
19731b807ce4Svictorle .keywords: dense, matrix, LAPACK, BLAS
19741b807ce4Svictorle 
1975284134d9SBarry Smith .seealso: MatCreate(), MatCreateSeqDense(), MatSeqDenseSetPreallocation(), MatSetMaximumSize()
19761b807ce4Svictorle @*/
1977be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatSeqDenseSetLDA(Mat B,PetscInt lda)
19781b807ce4Svictorle {
19791b807ce4Svictorle   Mat_SeqDense *b = (Mat_SeqDense*)B->data;
198021a2c019SBarry Smith 
19811b807ce4Svictorle   PetscFunctionBegin;
1982899cda47SBarry Smith   if (lda < B->rmap.n) SETERRQ2(PETSC_ERR_ARG_SIZ,"LDA %D must be at least matrix dimension %D",lda,B->rmap.n);
19831b807ce4Svictorle   b->lda       = lda;
198421a2c019SBarry Smith   b->changelda = PETSC_FALSE;
198521a2c019SBarry Smith   b->Mmax      = PetscMax(b->Mmax,lda);
19861b807ce4Svictorle   PetscFunctionReturn(0);
19871b807ce4Svictorle }
19881b807ce4Svictorle 
19890bad9183SKris Buschelman /*MC
1990fafad747SKris Buschelman    MATSEQDENSE - MATSEQDENSE = "seqdense" - A matrix type to be used for sequential dense matrices.
19910bad9183SKris Buschelman 
19920bad9183SKris Buschelman    Options Database Keys:
19930bad9183SKris Buschelman . -mat_type seqdense - sets the matrix type to "seqdense" during a call to MatSetFromOptions()
19940bad9183SKris Buschelman 
19950bad9183SKris Buschelman   Level: beginner
19960bad9183SKris Buschelman 
199789665df3SBarry Smith .seealso: MatCreateSeqDense()
199889665df3SBarry Smith 
19990bad9183SKris Buschelman M*/
20000bad9183SKris Buschelman 
2001273d9f13SBarry Smith EXTERN_C_BEGIN
20024a2ae208SSatish Balay #undef __FUNCT__
20034a2ae208SSatish Balay #define __FUNCT__ "MatCreate_SeqDense"
2004be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreate_SeqDense(Mat B)
2005273d9f13SBarry Smith {
2006273d9f13SBarry Smith   Mat_SeqDense   *b;
2007dfbe8321SBarry Smith   PetscErrorCode ierr;
20087c334f02SBarry Smith   PetscMPIInt    size;
2009273d9f13SBarry Smith 
2010273d9f13SBarry Smith   PetscFunctionBegin;
20117adad957SLisandro Dalcin   ierr = MPI_Comm_size(((PetscObject)B)->comm,&size);CHKERRQ(ierr);
201229bbc08cSBarry Smith   if (size > 1) SETERRQ(PETSC_ERR_ARG_WRONG,"Comm must be of size 1");
201355659b69SBarry Smith 
2014899cda47SBarry Smith   B->rmap.bs = B->cmap.bs = 1;
20156148ca0dSBarry Smith   ierr = PetscMapSetUp(&B->rmap);CHKERRQ(ierr);
20166148ca0dSBarry Smith   ierr = PetscMapSetUp(&B->cmap);CHKERRQ(ierr);
2017273d9f13SBarry Smith 
201838f2d2fdSLisandro Dalcin   ierr            = PetscNewLog(B,Mat_SeqDense,&b);CHKERRQ(ierr);
2019549d3d68SSatish Balay   ierr            = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr);
202090f02eecSBarry Smith   B->mapping      = 0;
202144cd7ae7SLois Curfman McInnes   B->data         = (void*)b;
202218f449edSLois Curfman McInnes 
2023a5ae1ecdSBarry Smith 
202444cd7ae7SLois Curfman McInnes   b->pivots       = 0;
2025273d9f13SBarry Smith   b->roworiented  = PETSC_TRUE;
2026273d9f13SBarry Smith   b->v            = 0;
2027899cda47SBarry Smith   b->lda          = B->rmap.n;
202821a2c019SBarry Smith   b->changelda    = PETSC_FALSE;
2029899cda47SBarry Smith   b->Mmax         = B->rmap.n;
2030899cda47SBarry Smith   b->Nmax         = B->cmap.n;
20314e220ebcSLois Curfman McInnes 
2032b24902e0SBarry Smith 
2033b24902e0SBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatGetFactor_seqdense_petsc_C",
2034b24902e0SBarry Smith                                      "MatGetFactor_seqdense_petsc",
2035b24902e0SBarry Smith                                       MatGetFactor_seqdense_petsc);CHKERRQ(ierr);
2036a23d5eceSKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatSeqDenseSetPreallocation_C",
2037a23d5eceSKris Buschelman                                     "MatSeqDenseSetPreallocation_SeqDense",
2038a23d5eceSKris Buschelman                                      MatSeqDenseSetPreallocation_SeqDense);CHKERRQ(ierr);
20394ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatMatMult_seqaij_seqdense_C",
20404ae313f4SHong Zhang                                      "MatMatMult_SeqAIJ_SeqDense",
20414ae313f4SHong Zhang                                       MatMatMult_SeqAIJ_SeqDense);CHKERRQ(ierr);
20424ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatMatMultSymbolic_seqaij_seqdense_C",
20434ae313f4SHong Zhang                                      "MatMatMultSymbolic_SeqAIJ_SeqDense",
20444ae313f4SHong Zhang                                       MatMatMultSymbolic_SeqAIJ_SeqDense);CHKERRQ(ierr);
20454ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatMatMultNumeric_seqaij_seqdense_C",
20464ae313f4SHong Zhang                                      "MatMatMultNumeric_SeqAIJ_SeqDense",
20474ae313f4SHong Zhang                                       MatMatMultNumeric_SeqAIJ_SeqDense);CHKERRQ(ierr);
204817667f90SBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)B,MATSEQDENSE);CHKERRQ(ierr);
20493a40ed3dSBarry Smith   PetscFunctionReturn(0);
2050289bc588SBarry Smith }
2051c0aa2d19SHong Zhang 
2052c0aa2d19SHong Zhang 
2053273d9f13SBarry Smith EXTERN_C_END
2054