xref: /petsc/src/mat/impls/dense/seq/dense.c (revision a3fa217bfefef3d54995bb82b2b5ea8a3c3e9019)
1be1d678aSKris Buschelman 
267e560aaSBarry Smith /*
367e560aaSBarry Smith      Defines the basic matrix operations for sequential dense.
467e560aaSBarry Smith */
5289bc588SBarry Smith 
6dec5eb66SMatthew G Knepley #include <../src/mat/impls/dense/seq/dense.h> /*I "petscmat.h" I*/
7c6db04a5SJed Brown #include <petscblaslapack.h>
8289bc588SBarry Smith 
96a63e612SBarry Smith #include <../src/mat/impls/aij/seq/aij.h>
10b2573a8aSBarry Smith 
116a63e612SBarry Smith #undef __FUNCT__
126a63e612SBarry Smith #define __FUNCT__ "MatConvert_SeqDense_SeqAIJ"
138cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatConvert_SeqDense_SeqAIJ(Mat A, MatType newtype,MatReuse reuse,Mat *newmat)
146a63e612SBarry Smith {
156a63e612SBarry Smith   Mat            B;
166a63e612SBarry Smith   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
176a63e612SBarry Smith   PetscErrorCode ierr;
189399e1b8SMatthew G. Knepley   PetscInt       i, j;
199399e1b8SMatthew G. Knepley   PetscInt       *rows, *nnz;
209399e1b8SMatthew G. Knepley   MatScalar      *aa = a->v, *vals;
216a63e612SBarry Smith 
226a63e612SBarry Smith   PetscFunctionBegin;
23ce94432eSBarry Smith   ierr = MatCreate(PetscObjectComm((PetscObject)A),&B);CHKERRQ(ierr);
246a63e612SBarry Smith   ierr = MatSetSizes(B,A->rmap->n,A->cmap->n,A->rmap->N,A->cmap->N);CHKERRQ(ierr);
256a63e612SBarry Smith   ierr = MatSetType(B,MATSEQAIJ);CHKERRQ(ierr);
269399e1b8SMatthew G. Knepley   ierr = PetscCalloc3(A->rmap->n,&rows,A->rmap->n,&nnz,A->rmap->n,&vals);CHKERRQ(ierr);
279399e1b8SMatthew G. Knepley   for (j=0; j<A->cmap->n; j++) {
289399e1b8SMatthew G. Knepley     for (i=0; i<A->rmap->n; i++) if (aa[i] != 0.0 || i == j) ++nnz[i];
296a63e612SBarry Smith     aa += a->lda;
306a63e612SBarry Smith   }
319399e1b8SMatthew G. Knepley   ierr = MatSeqAIJSetPreallocation(B,PETSC_DETERMINE,nnz);CHKERRQ(ierr);
329399e1b8SMatthew G. Knepley   aa = a->v;
339399e1b8SMatthew G. Knepley   for (j=0; j<A->cmap->n; j++) {
349399e1b8SMatthew G. Knepley     PetscInt numRows = 0;
359399e1b8SMatthew G. Knepley     for (i=0; i<A->rmap->n; i++) if (aa[i] != 0.0 || i == j) {rows[numRows] = i; vals[numRows++] = aa[i];}
369399e1b8SMatthew G. Knepley     ierr = MatSetValues(B,numRows,rows,1,&j,vals,INSERT_VALUES);CHKERRQ(ierr);
379399e1b8SMatthew G. Knepley     aa  += a->lda;
389399e1b8SMatthew G. Knepley   }
399399e1b8SMatthew G. Knepley   ierr = PetscFree3(rows,nnz,vals);CHKERRQ(ierr);
406a63e612SBarry Smith   ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
416a63e612SBarry Smith   ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
426a63e612SBarry Smith 
436a63e612SBarry Smith   if (reuse == MAT_REUSE_MATRIX) {
446a63e612SBarry Smith     ierr = MatHeaderReplace(A,B);CHKERRQ(ierr);
456a63e612SBarry Smith   } else {
466a63e612SBarry Smith     *newmat = B;
476a63e612SBarry Smith   }
486a63e612SBarry Smith   PetscFunctionReturn(0);
496a63e612SBarry Smith }
506a63e612SBarry Smith 
514a2ae208SSatish Balay #undef __FUNCT__
524a2ae208SSatish Balay #define __FUNCT__ "MatAXPY_SeqDense"
53f4df32b1SMatthew Knepley PetscErrorCode MatAXPY_SeqDense(Mat Y,PetscScalar alpha,Mat X,MatStructure str)
541987afe7SBarry Smith {
551987afe7SBarry Smith   Mat_SeqDense   *x     = (Mat_SeqDense*)X->data,*y = (Mat_SeqDense*)Y->data;
56f4df32b1SMatthew Knepley   PetscScalar    oalpha = alpha;
5713f74950SBarry Smith   PetscInt       j;
580805154bSBarry Smith   PetscBLASInt   N,m,ldax,lday,one = 1;
59efee365bSSatish Balay   PetscErrorCode ierr;
603a40ed3dSBarry Smith 
613a40ed3dSBarry Smith   PetscFunctionBegin;
62c5df96a5SBarry Smith   ierr = PetscBLASIntCast(X->rmap->n*X->cmap->n,&N);CHKERRQ(ierr);
63c5df96a5SBarry Smith   ierr = PetscBLASIntCast(X->rmap->n,&m);CHKERRQ(ierr);
64c5df96a5SBarry Smith   ierr = PetscBLASIntCast(x->lda,&ldax);CHKERRQ(ierr);
65c5df96a5SBarry Smith   ierr = PetscBLASIntCast(y->lda,&lday);CHKERRQ(ierr);
66a5ce6ee0Svictorle   if (ldax>m || lday>m) {
67d0f46423SBarry Smith     for (j=0; j<X->cmap->n; j++) {
688b83055fSJed Brown       PetscStackCallBLAS("BLASaxpy",BLASaxpy_(&m,&oalpha,x->v+j*ldax,&one,y->v+j*lday,&one));
69a5ce6ee0Svictorle     }
70a5ce6ee0Svictorle   } else {
718b83055fSJed Brown     PetscStackCallBLAS("BLASaxpy",BLASaxpy_(&N,&oalpha,x->v,&one,y->v,&one));
72a5ce6ee0Svictorle   }
73*a3fa217bSJose E. Roman   ierr = PetscObjectStateIncrease((PetscObject)Y);CHKERRQ(ierr);
740450473dSBarry Smith   ierr = PetscLogFlops(PetscMax(2*N-1,0));CHKERRQ(ierr);
753a40ed3dSBarry Smith   PetscFunctionReturn(0);
761987afe7SBarry Smith }
771987afe7SBarry Smith 
784a2ae208SSatish Balay #undef __FUNCT__
794a2ae208SSatish Balay #define __FUNCT__ "MatGetInfo_SeqDense"
80dfbe8321SBarry Smith PetscErrorCode MatGetInfo_SeqDense(Mat A,MatInfoType flag,MatInfo *info)
81289bc588SBarry Smith {
82d0f46423SBarry Smith   PetscInt N = A->rmap->n*A->cmap->n;
833a40ed3dSBarry Smith 
843a40ed3dSBarry Smith   PetscFunctionBegin;
854e220ebcSLois Curfman McInnes   info->block_size        = 1.0;
864e220ebcSLois Curfman McInnes   info->nz_allocated      = (double)N;
876de62eeeSBarry Smith   info->nz_used           = (double)N;
886de62eeeSBarry Smith   info->nz_unneeded       = (double)0;
894e220ebcSLois Curfman McInnes   info->assemblies        = (double)A->num_ass;
904e220ebcSLois Curfman McInnes   info->mallocs           = 0;
917adad957SLisandro Dalcin   info->memory            = ((PetscObject)A)->mem;
924e220ebcSLois Curfman McInnes   info->fill_ratio_given  = 0;
934e220ebcSLois Curfman McInnes   info->fill_ratio_needed = 0;
944e220ebcSLois Curfman McInnes   info->factor_mallocs    = 0;
953a40ed3dSBarry Smith   PetscFunctionReturn(0);
96289bc588SBarry Smith }
97289bc588SBarry Smith 
984a2ae208SSatish Balay #undef __FUNCT__
994a2ae208SSatish Balay #define __FUNCT__ "MatScale_SeqDense"
100f4df32b1SMatthew Knepley PetscErrorCode MatScale_SeqDense(Mat A,PetscScalar alpha)
10180cd9d93SLois Curfman McInnes {
102273d9f13SBarry Smith   Mat_SeqDense   *a     = (Mat_SeqDense*)A->data;
103f4df32b1SMatthew Knepley   PetscScalar    oalpha = alpha;
104efee365bSSatish Balay   PetscErrorCode ierr;
105c5df96a5SBarry Smith   PetscBLASInt   one = 1,j,nz,lda;
10680cd9d93SLois Curfman McInnes 
1073a40ed3dSBarry Smith   PetscFunctionBegin;
108c5df96a5SBarry Smith   ierr = PetscBLASIntCast(a->lda,&lda);CHKERRQ(ierr);
109d0f46423SBarry Smith   if (lda>A->rmap->n) {
110c5df96a5SBarry Smith     ierr = PetscBLASIntCast(A->rmap->n,&nz);CHKERRQ(ierr);
111d0f46423SBarry Smith     for (j=0; j<A->cmap->n; j++) {
1128b83055fSJed Brown       PetscStackCallBLAS("BLASscal",BLASscal_(&nz,&oalpha,a->v+j*lda,&one));
113a5ce6ee0Svictorle     }
114a5ce6ee0Svictorle   } else {
115c5df96a5SBarry Smith     ierr = PetscBLASIntCast(A->rmap->n*A->cmap->n,&nz);CHKERRQ(ierr);
1168b83055fSJed Brown     PetscStackCallBLAS("BLASscal",BLASscal_(&nz,&oalpha,a->v,&one));
117a5ce6ee0Svictorle   }
118efee365bSSatish Balay   ierr = PetscLogFlops(nz);CHKERRQ(ierr);
1193a40ed3dSBarry Smith   PetscFunctionReturn(0);
12080cd9d93SLois Curfman McInnes }
12180cd9d93SLois Curfman McInnes 
1221cbb95d3SBarry Smith #undef __FUNCT__
1231cbb95d3SBarry Smith #define __FUNCT__ "MatIsHermitian_SeqDense"
124ace3abfcSBarry Smith PetscErrorCode MatIsHermitian_SeqDense(Mat A,PetscReal rtol,PetscBool  *fl)
1251cbb95d3SBarry Smith {
1261cbb95d3SBarry Smith   Mat_SeqDense *a = (Mat_SeqDense*)A->data;
127d0f46423SBarry Smith   PetscInt     i,j,m = A->rmap->n,N;
1281cbb95d3SBarry Smith   PetscScalar  *v = a->v;
1291cbb95d3SBarry Smith 
1301cbb95d3SBarry Smith   PetscFunctionBegin;
1311cbb95d3SBarry Smith   *fl = PETSC_FALSE;
132d0f46423SBarry Smith   if (A->rmap->n != A->cmap->n) PetscFunctionReturn(0);
1331cbb95d3SBarry Smith   N = a->lda;
1341cbb95d3SBarry Smith 
1351cbb95d3SBarry Smith   for (i=0; i<m; i++) {
1361cbb95d3SBarry Smith     for (j=i+1; j<m; j++) {
1371cbb95d3SBarry Smith       if (PetscAbsScalar(v[i+j*N] - PetscConj(v[j+i*N])) > rtol) PetscFunctionReturn(0);
1381cbb95d3SBarry Smith     }
1391cbb95d3SBarry Smith   }
1401cbb95d3SBarry Smith   *fl = PETSC_TRUE;
1411cbb95d3SBarry Smith   PetscFunctionReturn(0);
1421cbb95d3SBarry Smith }
1431cbb95d3SBarry Smith 
144b24902e0SBarry Smith #undef __FUNCT__
145b24902e0SBarry Smith #define __FUNCT__ "MatDuplicateNoCreate_SeqDense"
146719d5645SBarry Smith PetscErrorCode MatDuplicateNoCreate_SeqDense(Mat newi,Mat A,MatDuplicateOption cpvalues)
147b24902e0SBarry Smith {
148b24902e0SBarry Smith   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data,*l;
149b24902e0SBarry Smith   PetscErrorCode ierr;
150b24902e0SBarry Smith   PetscInt       lda = (PetscInt)mat->lda,j,m;
151b24902e0SBarry Smith 
152b24902e0SBarry Smith   PetscFunctionBegin;
153aa5ea44dSBarry Smith   ierr = PetscLayoutReference(A->rmap,&newi->rmap);CHKERRQ(ierr);
154aa5ea44dSBarry Smith   ierr = PetscLayoutReference(A->cmap,&newi->cmap);CHKERRQ(ierr);
1550298fd71SBarry Smith   ierr = MatSeqDenseSetPreallocation(newi,NULL);CHKERRQ(ierr);
156b24902e0SBarry Smith   if (cpvalues == MAT_COPY_VALUES) {
157b24902e0SBarry Smith     l = (Mat_SeqDense*)newi->data;
158d0f46423SBarry Smith     if (lda>A->rmap->n) {
159d0f46423SBarry Smith       m = A->rmap->n;
160d0f46423SBarry Smith       for (j=0; j<A->cmap->n; j++) {
161b24902e0SBarry Smith         ierr = PetscMemcpy(l->v+j*m,mat->v+j*lda,m*sizeof(PetscScalar));CHKERRQ(ierr);
162b24902e0SBarry Smith       }
163b24902e0SBarry Smith     } else {
164d0f46423SBarry Smith       ierr = PetscMemcpy(l->v,mat->v,A->rmap->n*A->cmap->n*sizeof(PetscScalar));CHKERRQ(ierr);
165b24902e0SBarry Smith     }
166b24902e0SBarry Smith   }
167b24902e0SBarry Smith   newi->assembled = PETSC_TRUE;
168b24902e0SBarry Smith   PetscFunctionReturn(0);
169b24902e0SBarry Smith }
170b24902e0SBarry Smith 
1714a2ae208SSatish Balay #undef __FUNCT__
1724a2ae208SSatish Balay #define __FUNCT__ "MatDuplicate_SeqDense"
173dfbe8321SBarry Smith PetscErrorCode MatDuplicate_SeqDense(Mat A,MatDuplicateOption cpvalues,Mat *newmat)
17402cad45dSBarry Smith {
1756849ba73SBarry Smith   PetscErrorCode ierr;
17602cad45dSBarry Smith 
1773a40ed3dSBarry Smith   PetscFunctionBegin;
178ce94432eSBarry Smith   ierr = MatCreate(PetscObjectComm((PetscObject)A),newmat);CHKERRQ(ierr);
179d0f46423SBarry Smith   ierr = MatSetSizes(*newmat,A->rmap->n,A->cmap->n,A->rmap->n,A->cmap->n);CHKERRQ(ierr);
1805c9eb25fSBarry Smith   ierr = MatSetType(*newmat,((PetscObject)A)->type_name);CHKERRQ(ierr);
181719d5645SBarry Smith   ierr = MatDuplicateNoCreate_SeqDense(*newmat,A,cpvalues);CHKERRQ(ierr);
182b24902e0SBarry Smith   PetscFunctionReturn(0);
183b24902e0SBarry Smith }
184b24902e0SBarry Smith 
1856ee01492SSatish Balay 
1860481f469SBarry Smith extern PetscErrorCode MatLUFactor_SeqDense(Mat,IS,IS,const MatFactorInfo*);
187719d5645SBarry Smith 
1884a2ae208SSatish Balay #undef __FUNCT__
1894a2ae208SSatish Balay #define __FUNCT__ "MatLUFactorNumeric_SeqDense"
1900481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqDense(Mat fact,Mat A,const MatFactorInfo *info_dummy)
191289bc588SBarry Smith {
1924482741eSBarry Smith   MatFactorInfo  info;
193a093e273SMatthew Knepley   PetscErrorCode ierr;
1943a40ed3dSBarry Smith 
1953a40ed3dSBarry Smith   PetscFunctionBegin;
196c3ef05f6SHong Zhang   ierr = MatDuplicateNoCreate_SeqDense(fact,A,MAT_COPY_VALUES);CHKERRQ(ierr);
197719d5645SBarry Smith   ierr = MatLUFactor_SeqDense(fact,0,0,&info);CHKERRQ(ierr);
1983a40ed3dSBarry Smith   PetscFunctionReturn(0);
199289bc588SBarry Smith }
2006ee01492SSatish Balay 
2010b4b3355SBarry Smith #undef __FUNCT__
2024a2ae208SSatish Balay #define __FUNCT__ "MatSolve_SeqDense"
203dfbe8321SBarry Smith PetscErrorCode MatSolve_SeqDense(Mat A,Vec xx,Vec yy)
204289bc588SBarry Smith {
205c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
2066849ba73SBarry Smith   PetscErrorCode ierr;
20787828ca2SBarry Smith   PetscScalar    *x,*y;
208c5df96a5SBarry Smith   PetscBLASInt   one = 1,info,m;
20967e560aaSBarry Smith 
2103a40ed3dSBarry Smith   PetscFunctionBegin;
211c5df96a5SBarry Smith   ierr = PetscBLASIntCast(A->rmap->n,&m);CHKERRQ(ierr);
2121ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
2131ebc52fbSHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
214d0f46423SBarry Smith   ierr = PetscMemcpy(y,x,A->rmap->n*sizeof(PetscScalar));CHKERRQ(ierr);
215d5f3da31SBarry Smith   if (A->factortype == MAT_FACTOR_LU) {
216ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_GETRS)
217e32f2f54SBarry Smith     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"GETRS - Lapack routine is unavailable.");
218ae7cfcebSSatish Balay #else
2198b83055fSJed Brown     PetscStackCallBLAS("LAPACKgetrs",LAPACKgetrs_("N",&m,&one,mat->v,&mat->lda,mat->pivots,y,&m,&info));
220e32f2f54SBarry Smith     if (info) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"GETRS - Bad solve");
221ae7cfcebSSatish Balay #endif
222d5f3da31SBarry Smith   } else if (A->factortype == MAT_FACTOR_CHOLESKY) {
223ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_POTRS)
224e32f2f54SBarry Smith     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"POTRS - Lapack routine is unavailable.");
225ae7cfcebSSatish Balay #else
2268b83055fSJed Brown     PetscStackCallBLAS("LAPACKpotrs",LAPACKpotrs_("L",&m,&one,mat->v,&mat->lda,y,&m,&info));
227e32f2f54SBarry Smith     if (info) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"POTRS Bad solve");
228ae7cfcebSSatish Balay #endif
2292205254eSKarl Rupp   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix must be factored to solve");
2301ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
2311ebc52fbSHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
232dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*A->cmap->n*A->cmap->n - A->cmap->n);CHKERRQ(ierr);
2333a40ed3dSBarry Smith   PetscFunctionReturn(0);
234289bc588SBarry Smith }
2356ee01492SSatish Balay 
2364a2ae208SSatish Balay #undef __FUNCT__
23785e2c93fSHong Zhang #define __FUNCT__ "MatMatSolve_SeqDense"
23885e2c93fSHong Zhang PetscErrorCode MatMatSolve_SeqDense(Mat A,Mat B,Mat X)
23985e2c93fSHong Zhang {
24085e2c93fSHong Zhang   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
24185e2c93fSHong Zhang   PetscErrorCode ierr;
24285e2c93fSHong Zhang   PetscScalar    *b,*x;
243efb80c78SLisandro Dalcin   PetscInt       n;
244783b601eSJed Brown   PetscBLASInt   nrhs,info,m;
245bda8bf91SBarry Smith   PetscBool      flg;
24685e2c93fSHong Zhang 
24785e2c93fSHong Zhang   PetscFunctionBegin;
248c5df96a5SBarry Smith   ierr = PetscBLASIntCast(A->rmap->n,&m);CHKERRQ(ierr);
2490298fd71SBarry Smith   ierr = PetscObjectTypeCompareAny((PetscObject)B,&flg,MATSEQDENSE,MATMPIDENSE,NULL);CHKERRQ(ierr);
250ce94432eSBarry Smith   if (!flg) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONG,"Matrix B must be MATDENSE matrix");
2510298fd71SBarry Smith   ierr = PetscObjectTypeCompareAny((PetscObject)X,&flg,MATSEQDENSE,MATMPIDENSE,NULL);CHKERRQ(ierr);
252ce94432eSBarry Smith   if (!flg) SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONG,"Matrix X must be MATDENSE matrix");
253bda8bf91SBarry Smith 
2540298fd71SBarry Smith   ierr = MatGetSize(B,NULL,&n);CHKERRQ(ierr);
255c5df96a5SBarry Smith   ierr = PetscBLASIntCast(n,&nrhs);CHKERRQ(ierr);
2568c778c55SBarry Smith   ierr = MatDenseGetArray(B,&b);CHKERRQ(ierr);
2578c778c55SBarry Smith   ierr = MatDenseGetArray(X,&x);CHKERRQ(ierr);
25885e2c93fSHong Zhang 
259f272c775SHong Zhang   ierr = PetscMemcpy(x,b,m*nrhs*sizeof(PetscScalar));CHKERRQ(ierr);
26085e2c93fSHong Zhang 
26185e2c93fSHong Zhang   if (A->factortype == MAT_FACTOR_LU) {
26285e2c93fSHong Zhang #if defined(PETSC_MISSING_LAPACK_GETRS)
26385e2c93fSHong Zhang     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"GETRS - Lapack routine is unavailable.");
26485e2c93fSHong Zhang #else
2658b83055fSJed Brown     PetscStackCallBLAS("LAPACKgetrs",LAPACKgetrs_("N",&m,&nrhs,mat->v,&mat->lda,mat->pivots,x,&m,&info));
26685e2c93fSHong Zhang     if (info) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"GETRS - Bad solve");
26785e2c93fSHong Zhang #endif
26885e2c93fSHong Zhang   } else if (A->factortype == MAT_FACTOR_CHOLESKY) {
26985e2c93fSHong Zhang #if defined(PETSC_MISSING_LAPACK_POTRS)
27085e2c93fSHong Zhang     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"POTRS - Lapack routine is unavailable.");
27185e2c93fSHong Zhang #else
2728b83055fSJed Brown     PetscStackCallBLAS("LAPACKpotrs",LAPACKpotrs_("L",&m,&nrhs,mat->v,&mat->lda,x,&m,&info));
27385e2c93fSHong Zhang     if (info) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"POTRS Bad solve");
27485e2c93fSHong Zhang #endif
2752205254eSKarl Rupp   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix must be factored to solve");
27685e2c93fSHong Zhang 
2778c778c55SBarry Smith   ierr = MatDenseRestoreArray(B,&b);CHKERRQ(ierr);
2788c778c55SBarry Smith   ierr = MatDenseRestoreArray(X,&x);CHKERRQ(ierr);
27985e2c93fSHong Zhang   ierr = PetscLogFlops(nrhs*(2.0*m*m - m));CHKERRQ(ierr);
28085e2c93fSHong Zhang   PetscFunctionReturn(0);
28185e2c93fSHong Zhang }
28285e2c93fSHong Zhang 
28385e2c93fSHong Zhang #undef __FUNCT__
2844a2ae208SSatish Balay #define __FUNCT__ "MatSolveTranspose_SeqDense"
285dfbe8321SBarry Smith PetscErrorCode MatSolveTranspose_SeqDense(Mat A,Vec xx,Vec yy)
286da3a660dSBarry Smith {
287c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
288dfbe8321SBarry Smith   PetscErrorCode ierr;
28987828ca2SBarry Smith   PetscScalar    *x,*y;
290c5df96a5SBarry Smith   PetscBLASInt   one = 1,info,m;
29167e560aaSBarry Smith 
2923a40ed3dSBarry Smith   PetscFunctionBegin;
293c5df96a5SBarry Smith   ierr = PetscBLASIntCast(A->rmap->n,&m);CHKERRQ(ierr);
2941ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
2951ebc52fbSHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
296d0f46423SBarry Smith   ierr = PetscMemcpy(y,x,A->rmap->n*sizeof(PetscScalar));CHKERRQ(ierr);
297752f5567SLois Curfman McInnes   /* assume if pivots exist then use LU; else Cholesky */
298da3a660dSBarry Smith   if (mat->pivots) {
299ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_GETRS)
300e32f2f54SBarry Smith     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"GETRS - Lapack routine is unavailable.");
301ae7cfcebSSatish Balay #else
3028b83055fSJed Brown     PetscStackCallBLAS("LAPACKgetrs",LAPACKgetrs_("T",&m,&one,mat->v,&mat->lda,mat->pivots,y,&m,&info));
303e32f2f54SBarry Smith     if (info) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"POTRS - Bad solve");
304ae7cfcebSSatish Balay #endif
3057a97a34bSBarry Smith   } else {
306ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_POTRS)
307e32f2f54SBarry Smith     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"POTRS - Lapack routine is unavailable.");
308ae7cfcebSSatish Balay #else
3098b83055fSJed Brown     PetscStackCallBLAS("LAPACKpotrs",LAPACKpotrs_("L",&m,&one,mat->v,&mat->lda,y,&m,&info));
310e32f2f54SBarry Smith     if (info) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"POTRS - Bad solve");
311ae7cfcebSSatish Balay #endif
312da3a660dSBarry Smith   }
3131ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
3141ebc52fbSHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
315dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*A->cmap->n*A->cmap->n - A->cmap->n);CHKERRQ(ierr);
3163a40ed3dSBarry Smith   PetscFunctionReturn(0);
317da3a660dSBarry Smith }
3186ee01492SSatish Balay 
3194a2ae208SSatish Balay #undef __FUNCT__
3204a2ae208SSatish Balay #define __FUNCT__ "MatSolveAdd_SeqDense"
321dfbe8321SBarry Smith PetscErrorCode MatSolveAdd_SeqDense(Mat A,Vec xx,Vec zz,Vec yy)
322da3a660dSBarry Smith {
323c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
324dfbe8321SBarry Smith   PetscErrorCode ierr;
32587828ca2SBarry Smith   PetscScalar    *x,*y,sone = 1.0;
326da3a660dSBarry Smith   Vec            tmp = 0;
327c5df96a5SBarry Smith   PetscBLASInt   one = 1,info,m;
32867e560aaSBarry Smith 
3293a40ed3dSBarry Smith   PetscFunctionBegin;
330c5df96a5SBarry Smith   ierr = PetscBLASIntCast(A->rmap->n,&m);CHKERRQ(ierr);
3311ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
3321ebc52fbSHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
333d0f46423SBarry Smith   if (!A->rmap->n || !A->cmap->n) PetscFunctionReturn(0);
334da3a660dSBarry Smith   if (yy == zz) {
33578b31e54SBarry Smith     ierr = VecDuplicate(yy,&tmp);CHKERRQ(ierr);
3363bb1ff40SBarry Smith     ierr = PetscLogObjectParent((PetscObject)A,(PetscObject)tmp);CHKERRQ(ierr);
33778b31e54SBarry Smith     ierr = VecCopy(yy,tmp);CHKERRQ(ierr);
338da3a660dSBarry Smith   }
339d0f46423SBarry Smith   ierr = PetscMemcpy(y,x,A->rmap->n*sizeof(PetscScalar));CHKERRQ(ierr);
340752f5567SLois Curfman McInnes   /* assume if pivots exist then use LU; else Cholesky */
341da3a660dSBarry Smith   if (mat->pivots) {
342ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_GETRS)
343e32f2f54SBarry Smith     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"GETRS - Lapack routine is unavailable.");
344ae7cfcebSSatish Balay #else
3458b83055fSJed Brown     PetscStackCallBLAS("LAPACKgetrs",LAPACKgetrs_("N",&m,&one,mat->v,&mat->lda,mat->pivots,y,&m,&info));
346e32f2f54SBarry Smith     if (info) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Bad solve");
347ae7cfcebSSatish Balay #endif
348a8c6a408SBarry Smith   } else {
349ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_POTRS)
350e32f2f54SBarry Smith     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"POTRS - Lapack routine is unavailable.");
351ae7cfcebSSatish Balay #else
3528b83055fSJed Brown     PetscStackCallBLAS("LAPACKpotrs",LAPACKpotrs_("L",&m,&one,mat->v,&mat->lda,y,&m,&info));
353e32f2f54SBarry Smith     if (info) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Bad solve");
354ae7cfcebSSatish Balay #endif
355da3a660dSBarry Smith   }
3566bf464f9SBarry Smith   if (tmp) {
3576bf464f9SBarry Smith     ierr = VecAXPY(yy,sone,tmp);CHKERRQ(ierr);
3586bf464f9SBarry Smith     ierr = VecDestroy(&tmp);CHKERRQ(ierr);
3596bf464f9SBarry Smith   } else {
3606bf464f9SBarry Smith     ierr = VecAXPY(yy,sone,zz);CHKERRQ(ierr);
3616bf464f9SBarry Smith   }
3621ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
3631ebc52fbSHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
364dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*A->cmap->n*A->cmap->n);CHKERRQ(ierr);
3653a40ed3dSBarry Smith   PetscFunctionReturn(0);
366da3a660dSBarry Smith }
36767e560aaSBarry Smith 
3684a2ae208SSatish Balay #undef __FUNCT__
3694a2ae208SSatish Balay #define __FUNCT__ "MatSolveTransposeAdd_SeqDense"
370dfbe8321SBarry Smith PetscErrorCode MatSolveTransposeAdd_SeqDense(Mat A,Vec xx,Vec zz,Vec yy)
371da3a660dSBarry Smith {
372c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
3736849ba73SBarry Smith   PetscErrorCode ierr;
37487828ca2SBarry Smith   PetscScalar    *x,*y,sone = 1.0;
375da3a660dSBarry Smith   Vec            tmp;
376c5df96a5SBarry Smith   PetscBLASInt   one = 1,info,m;
37767e560aaSBarry Smith 
3783a40ed3dSBarry Smith   PetscFunctionBegin;
379c5df96a5SBarry Smith   ierr = PetscBLASIntCast(A->rmap->n,&m);CHKERRQ(ierr);
380d0f46423SBarry Smith   if (!A->rmap->n || !A->cmap->n) PetscFunctionReturn(0);
3811ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
3821ebc52fbSHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
383da3a660dSBarry Smith   if (yy == zz) {
38478b31e54SBarry Smith     ierr = VecDuplicate(yy,&tmp);CHKERRQ(ierr);
3853bb1ff40SBarry Smith     ierr = PetscLogObjectParent((PetscObject)A,(PetscObject)tmp);CHKERRQ(ierr);
38678b31e54SBarry Smith     ierr = VecCopy(yy,tmp);CHKERRQ(ierr);
387da3a660dSBarry Smith   }
388d0f46423SBarry Smith   ierr = PetscMemcpy(y,x,A->rmap->n*sizeof(PetscScalar));CHKERRQ(ierr);
389752f5567SLois Curfman McInnes   /* assume if pivots exist then use LU; else Cholesky */
390da3a660dSBarry Smith   if (mat->pivots) {
391ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_GETRS)
392e32f2f54SBarry Smith     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"GETRS - Lapack routine is unavailable.");
393ae7cfcebSSatish Balay #else
3948b83055fSJed Brown     PetscStackCallBLAS("LAPACKgetrs",LAPACKgetrs_("T",&m,&one,mat->v,&mat->lda,mat->pivots,y,&m,&info));
395e32f2f54SBarry Smith     if (info) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Bad solve");
396ae7cfcebSSatish Balay #endif
3973a40ed3dSBarry Smith   } else {
398ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_POTRS)
399e32f2f54SBarry Smith     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"POTRS - Lapack routine is unavailable.");
400ae7cfcebSSatish Balay #else
4018b83055fSJed Brown     PetscStackCallBLAS("LAPACKpotrs",LAPACKpotrs_("L",&m,&one,mat->v,&mat->lda,y,&m,&info));
402e32f2f54SBarry Smith     if (info) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Bad solve");
403ae7cfcebSSatish Balay #endif
404da3a660dSBarry Smith   }
40590f02eecSBarry Smith   if (tmp) {
4062dcb1b2aSMatthew Knepley     ierr = VecAXPY(yy,sone,tmp);CHKERRQ(ierr);
4076bf464f9SBarry Smith     ierr = VecDestroy(&tmp);CHKERRQ(ierr);
4083a40ed3dSBarry Smith   } else {
4092dcb1b2aSMatthew Knepley     ierr = VecAXPY(yy,sone,zz);CHKERRQ(ierr);
41090f02eecSBarry Smith   }
4111ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
4121ebc52fbSHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
413dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*A->cmap->n*A->cmap->n);CHKERRQ(ierr);
4143a40ed3dSBarry Smith   PetscFunctionReturn(0);
415da3a660dSBarry Smith }
416db4efbfdSBarry Smith 
417db4efbfdSBarry Smith /* ---------------------------------------------------------------*/
418db4efbfdSBarry Smith /* COMMENT: I have chosen to hide row permutation in the pivots,
419db4efbfdSBarry Smith    rather than put it in the Mat->row slot.*/
420db4efbfdSBarry Smith #undef __FUNCT__
421db4efbfdSBarry Smith #define __FUNCT__ "MatLUFactor_SeqDense"
4220481f469SBarry Smith PetscErrorCode MatLUFactor_SeqDense(Mat A,IS row,IS col,const MatFactorInfo *minfo)
423db4efbfdSBarry Smith {
424db4efbfdSBarry Smith #if defined(PETSC_MISSING_LAPACK_GETRF)
425db4efbfdSBarry Smith   PetscFunctionBegin;
426e32f2f54SBarry Smith   SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"GETRF - Lapack routine is unavailable.");
427db4efbfdSBarry Smith #else
428db4efbfdSBarry Smith   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
429db4efbfdSBarry Smith   PetscErrorCode ierr;
430db4efbfdSBarry Smith   PetscBLASInt   n,m,info;
431db4efbfdSBarry Smith 
432db4efbfdSBarry Smith   PetscFunctionBegin;
433c5df96a5SBarry Smith   ierr = PetscBLASIntCast(A->cmap->n,&n);CHKERRQ(ierr);
434c5df96a5SBarry Smith   ierr = PetscBLASIntCast(A->rmap->n,&m);CHKERRQ(ierr);
435db4efbfdSBarry Smith   if (!mat->pivots) {
436785e854fSJed Brown     ierr = PetscMalloc1((A->rmap->n+1),&mat->pivots);CHKERRQ(ierr);
4373bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)A,A->rmap->n*sizeof(PetscBLASInt));CHKERRQ(ierr);
438db4efbfdSBarry Smith   }
439db4efbfdSBarry Smith   if (!A->rmap->n || !A->cmap->n) PetscFunctionReturn(0);
4408e57ea43SSatish Balay   ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr);
4418b83055fSJed Brown   PetscStackCallBLAS("LAPACKgetrf",LAPACKgetrf_(&m,&n,mat->v,&mat->lda,mat->pivots,&info));
4428e57ea43SSatish Balay   ierr = PetscFPTrapPop();CHKERRQ(ierr);
4438e57ea43SSatish Balay 
444e32f2f54SBarry Smith   if (info<0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Bad argument to LU factorization");
445e32f2f54SBarry Smith   if (info>0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Bad LU factorization");
446db4efbfdSBarry Smith   A->ops->solve             = MatSolve_SeqDense;
447db4efbfdSBarry Smith   A->ops->solvetranspose    = MatSolveTranspose_SeqDense;
448db4efbfdSBarry Smith   A->ops->solveadd          = MatSolveAdd_SeqDense;
449db4efbfdSBarry Smith   A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqDense;
450d5f3da31SBarry Smith   A->factortype             = MAT_FACTOR_LU;
451db4efbfdSBarry Smith 
452dc0b31edSSatish Balay   ierr = PetscLogFlops((2.0*A->cmap->n*A->cmap->n*A->cmap->n)/3);CHKERRQ(ierr);
453db4efbfdSBarry Smith #endif
454db4efbfdSBarry Smith   PetscFunctionReturn(0);
455db4efbfdSBarry Smith }
456db4efbfdSBarry Smith 
457db4efbfdSBarry Smith #undef __FUNCT__
458db4efbfdSBarry Smith #define __FUNCT__ "MatCholeskyFactor_SeqDense"
4590481f469SBarry Smith PetscErrorCode MatCholeskyFactor_SeqDense(Mat A,IS perm,const MatFactorInfo *factinfo)
460db4efbfdSBarry Smith {
461db4efbfdSBarry Smith #if defined(PETSC_MISSING_LAPACK_POTRF)
462db4efbfdSBarry Smith   PetscFunctionBegin;
463e32f2f54SBarry Smith   SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"POTRF - Lapack routine is unavailable.");
464db4efbfdSBarry Smith #else
465db4efbfdSBarry Smith   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
466db4efbfdSBarry Smith   PetscErrorCode ierr;
467c5df96a5SBarry Smith   PetscBLASInt   info,n;
468db4efbfdSBarry Smith 
469db4efbfdSBarry Smith   PetscFunctionBegin;
470c5df96a5SBarry Smith   ierr = PetscBLASIntCast(A->cmap->n,&n);CHKERRQ(ierr);
471db4efbfdSBarry Smith   ierr = PetscFree(mat->pivots);CHKERRQ(ierr);
472db4efbfdSBarry Smith 
473db4efbfdSBarry Smith   if (!A->rmap->n || !A->cmap->n) PetscFunctionReturn(0);
4748b83055fSJed Brown   PetscStackCallBLAS("LAPACKpotrf",LAPACKpotrf_("L",&n,mat->v,&mat->lda,&info));
475e32f2f54SBarry Smith   if (info) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_MAT_CH_ZRPVT,"Bad factorization: zero pivot in row %D",(PetscInt)info-1);
476db4efbfdSBarry Smith   A->ops->solve             = MatSolve_SeqDense;
477db4efbfdSBarry Smith   A->ops->solvetranspose    = MatSolveTranspose_SeqDense;
478db4efbfdSBarry Smith   A->ops->solveadd          = MatSolveAdd_SeqDense;
479db4efbfdSBarry Smith   A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqDense;
480d5f3da31SBarry Smith   A->factortype             = MAT_FACTOR_CHOLESKY;
4812205254eSKarl Rupp 
482dc0b31edSSatish Balay   ierr = PetscLogFlops((A->cmap->n*A->cmap->n*A->cmap->n)/3.0);CHKERRQ(ierr);
483db4efbfdSBarry Smith #endif
484db4efbfdSBarry Smith   PetscFunctionReturn(0);
485db4efbfdSBarry Smith }
486db4efbfdSBarry Smith 
487db4efbfdSBarry Smith 
488db4efbfdSBarry Smith #undef __FUNCT__
489db4efbfdSBarry Smith #define __FUNCT__ "MatCholeskyFactorNumeric_SeqDense"
4900481f469SBarry Smith PetscErrorCode MatCholeskyFactorNumeric_SeqDense(Mat fact,Mat A,const MatFactorInfo *info_dummy)
491db4efbfdSBarry Smith {
492db4efbfdSBarry Smith   PetscErrorCode ierr;
493db4efbfdSBarry Smith   MatFactorInfo  info;
494db4efbfdSBarry Smith 
495db4efbfdSBarry Smith   PetscFunctionBegin;
496db4efbfdSBarry Smith   info.fill = 1.0;
4972205254eSKarl Rupp 
498c3ef05f6SHong Zhang   ierr = MatDuplicateNoCreate_SeqDense(fact,A,MAT_COPY_VALUES);CHKERRQ(ierr);
499719d5645SBarry Smith   ierr = MatCholeskyFactor_SeqDense(fact,0,&info);CHKERRQ(ierr);
500db4efbfdSBarry Smith   PetscFunctionReturn(0);
501db4efbfdSBarry Smith }
502db4efbfdSBarry Smith 
503db4efbfdSBarry Smith #undef __FUNCT__
504db4efbfdSBarry Smith #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqDense"
5050481f469SBarry Smith PetscErrorCode MatCholeskyFactorSymbolic_SeqDense(Mat fact,Mat A,IS row,const MatFactorInfo *info)
506db4efbfdSBarry Smith {
507db4efbfdSBarry Smith   PetscFunctionBegin;
508c3ef05f6SHong Zhang   fact->assembled                  = PETSC_TRUE;
5091bbcc794SSatish Balay   fact->preallocated               = PETSC_TRUE;
510719d5645SBarry Smith   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqDense;
511db4efbfdSBarry Smith   PetscFunctionReturn(0);
512db4efbfdSBarry Smith }
513db4efbfdSBarry Smith 
514db4efbfdSBarry Smith #undef __FUNCT__
515db4efbfdSBarry Smith #define __FUNCT__ "MatLUFactorSymbolic_SeqDense"
5160481f469SBarry Smith PetscErrorCode MatLUFactorSymbolic_SeqDense(Mat fact,Mat A,IS row,IS col,const MatFactorInfo *info)
517db4efbfdSBarry Smith {
518db4efbfdSBarry Smith   PetscFunctionBegin;
519b66fe19dSMatthew G Knepley   fact->preallocated         = PETSC_TRUE;
520c3ef05f6SHong Zhang   fact->assembled            = PETSC_TRUE;
521719d5645SBarry Smith   fact->ops->lufactornumeric = MatLUFactorNumeric_SeqDense;
522db4efbfdSBarry Smith   PetscFunctionReturn(0);
523db4efbfdSBarry Smith }
524db4efbfdSBarry Smith 
525db4efbfdSBarry Smith #undef __FUNCT__
526db4efbfdSBarry Smith #define __FUNCT__ "MatGetFactor_seqdense_petsc"
5278cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatGetFactor_seqdense_petsc(Mat A,MatFactorType ftype,Mat *fact)
528db4efbfdSBarry Smith {
529db4efbfdSBarry Smith   PetscErrorCode ierr;
530db4efbfdSBarry Smith 
531db4efbfdSBarry Smith   PetscFunctionBegin;
532ce94432eSBarry Smith   ierr = MatCreate(PetscObjectComm((PetscObject)A),fact);CHKERRQ(ierr);
533db4efbfdSBarry Smith   ierr = MatSetSizes(*fact,A->rmap->n,A->cmap->n,A->rmap->n,A->cmap->n);CHKERRQ(ierr);
534db4efbfdSBarry Smith   ierr = MatSetType(*fact,((PetscObject)A)->type_name);CHKERRQ(ierr);
535db4efbfdSBarry Smith   if (ftype == MAT_FACTOR_LU) {
536db4efbfdSBarry Smith     (*fact)->ops->lufactorsymbolic = MatLUFactorSymbolic_SeqDense;
537db4efbfdSBarry Smith   } else {
538db4efbfdSBarry Smith     (*fact)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqDense;
539db4efbfdSBarry Smith   }
540d5f3da31SBarry Smith   (*fact)->factortype = ftype;
541db4efbfdSBarry Smith   PetscFunctionReturn(0);
542db4efbfdSBarry Smith }
543db4efbfdSBarry Smith 
544289bc588SBarry Smith /* ------------------------------------------------------------------*/
5454a2ae208SSatish Balay #undef __FUNCT__
54641f059aeSBarry Smith #define __FUNCT__ "MatSOR_SeqDense"
54741f059aeSBarry Smith PetscErrorCode MatSOR_SeqDense(Mat A,Vec bb,PetscReal omega,MatSORType flag,PetscReal shift,PetscInt its,PetscInt lits,Vec xx)
548289bc588SBarry Smith {
549c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
55087828ca2SBarry Smith   PetscScalar    *x,*b,*v = mat->v,zero = 0.0,xt;
551dfbe8321SBarry Smith   PetscErrorCode ierr;
552d0f46423SBarry Smith   PetscInt       m = A->rmap->n,i;
553c5df96a5SBarry Smith   PetscBLASInt   o = 1,bm;
554289bc588SBarry Smith 
5553a40ed3dSBarry Smith   PetscFunctionBegin;
556c5df96a5SBarry Smith   ierr = PetscBLASIntCast(m,&bm);CHKERRQ(ierr);
557289bc588SBarry Smith   if (flag & SOR_ZERO_INITIAL_GUESS) {
55871044d3cSBarry Smith     /* this is a hack fix, should have another version without the second BLASdot */
5592dcb1b2aSMatthew Knepley     ierr = VecSet(xx,zero);CHKERRQ(ierr);
560289bc588SBarry Smith   }
5611ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
5621ebc52fbSHong Zhang   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
563b965ef7fSBarry Smith   its  = its*lits;
564e32f2f54SBarry Smith   if (its <= 0) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Relaxation requires global its %D and local its %D both positive",its,lits);
565289bc588SBarry Smith   while (its--) {
566fccaa45eSBarry Smith     if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP) {
567289bc588SBarry Smith       for (i=0; i<m; i++) {
5688b83055fSJed Brown         PetscStackCallBLAS("BLASdot",xt   = b[i] - BLASdot_(&bm,v+i,&bm,x,&o));
56955a1b374SBarry Smith         x[i] = (1. - omega)*x[i] + omega*(xt+v[i + i*m]*x[i])/(v[i + i*m]+shift);
570289bc588SBarry Smith       }
571289bc588SBarry Smith     }
572fccaa45eSBarry Smith     if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP) {
573289bc588SBarry Smith       for (i=m-1; i>=0; i--) {
5748b83055fSJed Brown         PetscStackCallBLAS("BLASdot",xt   = b[i] - BLASdot_(&bm,v+i,&bm,x,&o));
57555a1b374SBarry Smith         x[i] = (1. - omega)*x[i] + omega*(xt+v[i + i*m]*x[i])/(v[i + i*m]+shift);
576289bc588SBarry Smith       }
577289bc588SBarry Smith     }
578289bc588SBarry Smith   }
5791ebc52fbSHong Zhang   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
5801ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
5813a40ed3dSBarry Smith   PetscFunctionReturn(0);
582289bc588SBarry Smith }
583289bc588SBarry Smith 
584289bc588SBarry Smith /* -----------------------------------------------------------------*/
5854a2ae208SSatish Balay #undef __FUNCT__
5864a2ae208SSatish Balay #define __FUNCT__ "MatMultTranspose_SeqDense"
587dfbe8321SBarry Smith PetscErrorCode MatMultTranspose_SeqDense(Mat A,Vec xx,Vec yy)
588289bc588SBarry Smith {
589c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
59087828ca2SBarry Smith   PetscScalar    *v   = mat->v,*x,*y;
591dfbe8321SBarry Smith   PetscErrorCode ierr;
5920805154bSBarry Smith   PetscBLASInt   m, n,_One=1;
593ea709b57SSatish Balay   PetscScalar    _DOne=1.0,_DZero=0.0;
5943a40ed3dSBarry Smith 
5953a40ed3dSBarry Smith   PetscFunctionBegin;
596c5df96a5SBarry Smith   ierr = PetscBLASIntCast(A->rmap->n,&m);CHKERRQ(ierr);
597c5df96a5SBarry Smith   ierr = PetscBLASIntCast(A->cmap->n,&n);CHKERRQ(ierr);
598d0f46423SBarry Smith   if (!A->rmap->n || !A->cmap->n) PetscFunctionReturn(0);
5991ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
6001ebc52fbSHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
6018b83055fSJed Brown   PetscStackCallBLAS("BLASgemv",BLASgemv_("T",&m,&n,&_DOne,v,&mat->lda,x,&_One,&_DZero,y,&_One));
6021ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
6031ebc52fbSHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
604dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*A->rmap->n*A->cmap->n - A->cmap->n);CHKERRQ(ierr);
6053a40ed3dSBarry Smith   PetscFunctionReturn(0);
606289bc588SBarry Smith }
607800995b7SMatthew Knepley 
6084a2ae208SSatish Balay #undef __FUNCT__
6094a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqDense"
610dfbe8321SBarry Smith PetscErrorCode MatMult_SeqDense(Mat A,Vec xx,Vec yy)
611289bc588SBarry Smith {
612c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
61387828ca2SBarry Smith   PetscScalar    *v   = mat->v,*x,*y,_DOne=1.0,_DZero=0.0;
614dfbe8321SBarry Smith   PetscErrorCode ierr;
6150805154bSBarry Smith   PetscBLASInt   m, n, _One=1;
6163a40ed3dSBarry Smith 
6173a40ed3dSBarry Smith   PetscFunctionBegin;
618c5df96a5SBarry Smith   ierr = PetscBLASIntCast(A->rmap->n,&m);CHKERRQ(ierr);
619c5df96a5SBarry Smith   ierr = PetscBLASIntCast(A->cmap->n,&n);CHKERRQ(ierr);
620d0f46423SBarry Smith   if (!A->rmap->n || !A->cmap->n) PetscFunctionReturn(0);
6211ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
6221ebc52fbSHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
6238b83055fSJed Brown   PetscStackCallBLAS("BLASgemv",BLASgemv_("N",&m,&n,&_DOne,v,&(mat->lda),x,&_One,&_DZero,y,&_One));
6241ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
6251ebc52fbSHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
626dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*A->rmap->n*A->cmap->n - A->rmap->n);CHKERRQ(ierr);
6273a40ed3dSBarry Smith   PetscFunctionReturn(0);
628289bc588SBarry Smith }
6296ee01492SSatish Balay 
6304a2ae208SSatish Balay #undef __FUNCT__
6314a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqDense"
632dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqDense(Mat A,Vec xx,Vec zz,Vec yy)
633289bc588SBarry Smith {
634c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
63587828ca2SBarry Smith   PetscScalar    *v   = mat->v,*x,*y,_DOne=1.0;
636dfbe8321SBarry Smith   PetscErrorCode ierr;
6370805154bSBarry Smith   PetscBLASInt   m, n, _One=1;
6383a40ed3dSBarry Smith 
6393a40ed3dSBarry Smith   PetscFunctionBegin;
640c5df96a5SBarry Smith   ierr = PetscBLASIntCast(A->rmap->n,&m);CHKERRQ(ierr);
641c5df96a5SBarry Smith   ierr = PetscBLASIntCast(A->cmap->n,&n);CHKERRQ(ierr);
642d0f46423SBarry Smith   if (!A->rmap->n || !A->cmap->n) PetscFunctionReturn(0);
643600d6d0bSBarry Smith   if (zz != yy) {ierr = VecCopy(zz,yy);CHKERRQ(ierr);}
6441ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
6451ebc52fbSHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
6468b83055fSJed Brown   PetscStackCallBLAS("BLASgemv",BLASgemv_("N",&m,&n,&_DOne,v,&(mat->lda),x,&_One,&_DOne,y,&_One));
6471ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
6481ebc52fbSHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
649dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*A->rmap->n*A->cmap->n);CHKERRQ(ierr);
6503a40ed3dSBarry Smith   PetscFunctionReturn(0);
651289bc588SBarry Smith }
6526ee01492SSatish Balay 
6534a2ae208SSatish Balay #undef __FUNCT__
6544a2ae208SSatish Balay #define __FUNCT__ "MatMultTransposeAdd_SeqDense"
655dfbe8321SBarry Smith PetscErrorCode MatMultTransposeAdd_SeqDense(Mat A,Vec xx,Vec zz,Vec yy)
656289bc588SBarry Smith {
657c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
65887828ca2SBarry Smith   PetscScalar    *v   = mat->v,*x,*y;
659dfbe8321SBarry Smith   PetscErrorCode ierr;
6600805154bSBarry Smith   PetscBLASInt   m, n, _One=1;
66187828ca2SBarry Smith   PetscScalar    _DOne=1.0;
6623a40ed3dSBarry Smith 
6633a40ed3dSBarry Smith   PetscFunctionBegin;
664c5df96a5SBarry Smith   ierr = PetscBLASIntCast(A->rmap->n,&m);CHKERRQ(ierr);
665c5df96a5SBarry Smith   ierr = PetscBLASIntCast(A->cmap->n,&n);CHKERRQ(ierr);
666d0f46423SBarry Smith   if (!A->rmap->n || !A->cmap->n) PetscFunctionReturn(0);
667600d6d0bSBarry Smith   if (zz != yy) {ierr = VecCopy(zz,yy);CHKERRQ(ierr);}
6681ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
6691ebc52fbSHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
6708b83055fSJed Brown   PetscStackCallBLAS("BLASgemv",BLASgemv_("T",&m,&n,&_DOne,v,&(mat->lda),x,&_One,&_DOne,y,&_One));
6711ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
6721ebc52fbSHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
673dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*A->rmap->n*A->cmap->n);CHKERRQ(ierr);
6743a40ed3dSBarry Smith   PetscFunctionReturn(0);
675289bc588SBarry Smith }
676289bc588SBarry Smith 
677289bc588SBarry Smith /* -----------------------------------------------------------------*/
6784a2ae208SSatish Balay #undef __FUNCT__
6794a2ae208SSatish Balay #define __FUNCT__ "MatGetRow_SeqDense"
68013f74950SBarry Smith PetscErrorCode MatGetRow_SeqDense(Mat A,PetscInt row,PetscInt *ncols,PetscInt **cols,PetscScalar **vals)
681289bc588SBarry Smith {
682c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
68387828ca2SBarry Smith   PetscScalar    *v;
6846849ba73SBarry Smith   PetscErrorCode ierr;
68513f74950SBarry Smith   PetscInt       i;
68667e560aaSBarry Smith 
6873a40ed3dSBarry Smith   PetscFunctionBegin;
688d0f46423SBarry Smith   *ncols = A->cmap->n;
689289bc588SBarry Smith   if (cols) {
690785e854fSJed Brown     ierr = PetscMalloc1((A->cmap->n+1),cols);CHKERRQ(ierr);
691d0f46423SBarry Smith     for (i=0; i<A->cmap->n; i++) (*cols)[i] = i;
692289bc588SBarry Smith   }
693289bc588SBarry Smith   if (vals) {
694785e854fSJed Brown     ierr = PetscMalloc1((A->cmap->n+1),vals);CHKERRQ(ierr);
695289bc588SBarry Smith     v    = mat->v + row;
696d0f46423SBarry Smith     for (i=0; i<A->cmap->n; i++) {(*vals)[i] = *v; v += mat->lda;}
697289bc588SBarry Smith   }
6983a40ed3dSBarry Smith   PetscFunctionReturn(0);
699289bc588SBarry Smith }
7006ee01492SSatish Balay 
7014a2ae208SSatish Balay #undef __FUNCT__
7024a2ae208SSatish Balay #define __FUNCT__ "MatRestoreRow_SeqDense"
70313f74950SBarry Smith PetscErrorCode MatRestoreRow_SeqDense(Mat A,PetscInt row,PetscInt *ncols,PetscInt **cols,PetscScalar **vals)
704289bc588SBarry Smith {
705dfbe8321SBarry Smith   PetscErrorCode ierr;
7066e111a19SKarl Rupp 
707606d414cSSatish Balay   PetscFunctionBegin;
708606d414cSSatish Balay   if (cols) {ierr = PetscFree(*cols);CHKERRQ(ierr);}
709606d414cSSatish Balay   if (vals) {ierr = PetscFree(*vals);CHKERRQ(ierr); }
7103a40ed3dSBarry Smith   PetscFunctionReturn(0);
711289bc588SBarry Smith }
712289bc588SBarry Smith /* ----------------------------------------------------------------*/
7134a2ae208SSatish Balay #undef __FUNCT__
7144a2ae208SSatish Balay #define __FUNCT__ "MatSetValues_SeqDense"
71513f74950SBarry Smith PetscErrorCode MatSetValues_SeqDense(Mat A,PetscInt m,const PetscInt indexm[],PetscInt n,const PetscInt indexn[],const PetscScalar v[],InsertMode addv)
716289bc588SBarry Smith {
717c0bbcb79SLois Curfman McInnes   Mat_SeqDense *mat = (Mat_SeqDense*)A->data;
71813f74950SBarry Smith   PetscInt     i,j,idx=0;
719d6dfbf8fSBarry Smith 
7203a40ed3dSBarry Smith   PetscFunctionBegin;
72171fd2e92SBarry Smith   if (v) PetscValidScalarPointer(v,6);
722289bc588SBarry Smith   if (!mat->roworiented) {
723dbb450caSBarry Smith     if (addv == INSERT_VALUES) {
724289bc588SBarry Smith       for (j=0; j<n; j++) {
725cddbea37SSatish Balay         if (indexn[j] < 0) {idx += m; continue;}
7262515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
727e32f2f54SBarry Smith         if (indexn[j] >= A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",indexn[j],A->cmap->n-1);
72858804f6dSBarry Smith #endif
729289bc588SBarry Smith         for (i=0; i<m; i++) {
730cddbea37SSatish Balay           if (indexm[i] < 0) {idx++; continue;}
7312515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
732e32f2f54SBarry Smith           if (indexm[i] >= A->rmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",indexm[i],A->rmap->n-1);
73358804f6dSBarry Smith #endif
734cddbea37SSatish Balay           mat->v[indexn[j]*mat->lda + indexm[i]] = v[idx++];
735289bc588SBarry Smith         }
736289bc588SBarry Smith       }
7373a40ed3dSBarry Smith     } else {
738289bc588SBarry Smith       for (j=0; j<n; j++) {
739cddbea37SSatish Balay         if (indexn[j] < 0) {idx += m; continue;}
7402515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
741e32f2f54SBarry Smith         if (indexn[j] >= A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",indexn[j],A->cmap->n-1);
74258804f6dSBarry Smith #endif
743289bc588SBarry Smith         for (i=0; i<m; i++) {
744cddbea37SSatish Balay           if (indexm[i] < 0) {idx++; continue;}
7452515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
746e32f2f54SBarry Smith           if (indexm[i] >= A->rmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",indexm[i],A->rmap->n-1);
74758804f6dSBarry Smith #endif
748cddbea37SSatish Balay           mat->v[indexn[j]*mat->lda + indexm[i]] += v[idx++];
749289bc588SBarry Smith         }
750289bc588SBarry Smith       }
751289bc588SBarry Smith     }
7523a40ed3dSBarry Smith   } else {
753dbb450caSBarry Smith     if (addv == INSERT_VALUES) {
754e8d4e0b9SBarry Smith       for (i=0; i<m; i++) {
755cddbea37SSatish Balay         if (indexm[i] < 0) { idx += n; continue;}
7562515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
757e32f2f54SBarry Smith         if (indexm[i] >= A->rmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",indexm[i],A->rmap->n-1);
75858804f6dSBarry Smith #endif
759e8d4e0b9SBarry Smith         for (j=0; j<n; j++) {
760cddbea37SSatish Balay           if (indexn[j] < 0) { idx++; continue;}
7612515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
762e32f2f54SBarry Smith           if (indexn[j] >= A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",indexn[j],A->cmap->n-1);
76358804f6dSBarry Smith #endif
764cddbea37SSatish Balay           mat->v[indexn[j]*mat->lda + indexm[i]] = v[idx++];
765e8d4e0b9SBarry Smith         }
766e8d4e0b9SBarry Smith       }
7673a40ed3dSBarry Smith     } else {
768289bc588SBarry Smith       for (i=0; i<m; i++) {
769cddbea37SSatish Balay         if (indexm[i] < 0) { idx += n; continue;}
7702515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
771e32f2f54SBarry Smith         if (indexm[i] >= A->rmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row too large: row %D max %D",indexm[i],A->rmap->n-1);
77258804f6dSBarry Smith #endif
773289bc588SBarry Smith         for (j=0; j<n; j++) {
774cddbea37SSatish Balay           if (indexn[j] < 0) { idx++; continue;}
7752515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
776e32f2f54SBarry Smith           if (indexn[j] >= A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column too large: col %D max %D",indexn[j],A->cmap->n-1);
77758804f6dSBarry Smith #endif
778cddbea37SSatish Balay           mat->v[indexn[j]*mat->lda + indexm[i]] += v[idx++];
779289bc588SBarry Smith         }
780289bc588SBarry Smith       }
781289bc588SBarry Smith     }
782e8d4e0b9SBarry Smith   }
7833a40ed3dSBarry Smith   PetscFunctionReturn(0);
784289bc588SBarry Smith }
785e8d4e0b9SBarry Smith 
7864a2ae208SSatish Balay #undef __FUNCT__
7874a2ae208SSatish Balay #define __FUNCT__ "MatGetValues_SeqDense"
78813f74950SBarry Smith PetscErrorCode MatGetValues_SeqDense(Mat A,PetscInt m,const PetscInt indexm[],PetscInt n,const PetscInt indexn[],PetscScalar v[])
789ae80bb75SLois Curfman McInnes {
790ae80bb75SLois Curfman McInnes   Mat_SeqDense *mat = (Mat_SeqDense*)A->data;
79113f74950SBarry Smith   PetscInt     i,j;
792ae80bb75SLois Curfman McInnes 
7933a40ed3dSBarry Smith   PetscFunctionBegin;
794ae80bb75SLois Curfman McInnes   /* row-oriented output */
795ae80bb75SLois Curfman McInnes   for (i=0; i<m; i++) {
79697e567efSBarry Smith     if (indexm[i] < 0) {v += n;continue;}
797e32f2f54SBarry Smith     if (indexm[i] >= A->rmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row %D requested larger than number rows %D",indexm[i],A->rmap->n);
798ae80bb75SLois Curfman McInnes     for (j=0; j<n; j++) {
7996f31f424SBarry Smith       if (indexn[j] < 0) {v++; continue;}
800e32f2f54SBarry Smith       if (indexn[j] >= A->cmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Column %D requested larger than number columns %D",indexn[j],A->cmap->n);
80197e567efSBarry Smith       *v++ = mat->v[indexn[j]*mat->lda + indexm[i]];
802ae80bb75SLois Curfman McInnes     }
803ae80bb75SLois Curfman McInnes   }
8043a40ed3dSBarry Smith   PetscFunctionReturn(0);
805ae80bb75SLois Curfman McInnes }
806ae80bb75SLois Curfman McInnes 
807289bc588SBarry Smith /* -----------------------------------------------------------------*/
808289bc588SBarry Smith 
8094a2ae208SSatish Balay #undef __FUNCT__
8105bba2384SShri Abhyankar #define __FUNCT__ "MatLoad_SeqDense"
811112444f4SShri Abhyankar PetscErrorCode MatLoad_SeqDense(Mat newmat,PetscViewer viewer)
812aabbc4fbSShri Abhyankar {
813aabbc4fbSShri Abhyankar   Mat_SeqDense   *a;
814aabbc4fbSShri Abhyankar   PetscErrorCode ierr;
815aabbc4fbSShri Abhyankar   PetscInt       *scols,i,j,nz,header[4];
816aabbc4fbSShri Abhyankar   int            fd;
817aabbc4fbSShri Abhyankar   PetscMPIInt    size;
818aabbc4fbSShri Abhyankar   PetscInt       *rowlengths = 0,M,N,*cols,grows,gcols;
819aabbc4fbSShri Abhyankar   PetscScalar    *vals,*svals,*v,*w;
820ce94432eSBarry Smith   MPI_Comm       comm;
821aabbc4fbSShri Abhyankar 
822aabbc4fbSShri Abhyankar   PetscFunctionBegin;
823ce94432eSBarry Smith   ierr = PetscObjectGetComm((PetscObject)viewer,&comm);CHKERRQ(ierr);
824aabbc4fbSShri Abhyankar   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
825aabbc4fbSShri Abhyankar   if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"view must have one processor");
826aabbc4fbSShri Abhyankar   ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr);
827aabbc4fbSShri Abhyankar   ierr = PetscBinaryRead(fd,header,4,PETSC_INT);CHKERRQ(ierr);
828aabbc4fbSShri Abhyankar   if (header[0] != MAT_FILE_CLASSID) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Not matrix object");
829aabbc4fbSShri Abhyankar   M = header[1]; N = header[2]; nz = header[3];
830aabbc4fbSShri Abhyankar 
831aabbc4fbSShri Abhyankar   /* set global size if not set already*/
832aabbc4fbSShri Abhyankar   if (newmat->rmap->n < 0 && newmat->rmap->N < 0 && newmat->cmap->n < 0 && newmat->cmap->N < 0) {
833aabbc4fbSShri Abhyankar     ierr = MatSetSizes(newmat,M,N,M,N);CHKERRQ(ierr);
834aabbc4fbSShri Abhyankar   } else {
835aabbc4fbSShri Abhyankar     /* if sizes and type are already set, check if the vector global sizes are correct */
836aabbc4fbSShri Abhyankar     ierr = MatGetSize(newmat,&grows,&gcols);CHKERRQ(ierr);
837aabbc4fbSShri Abhyankar     if (M != grows ||  N != gcols) SETERRQ4(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED, "Matrix in file of different length (%d, %d) than the input matrix (%d, %d)",M,N,grows,gcols);
838aabbc4fbSShri Abhyankar   }
8390298fd71SBarry Smith   ierr = MatSeqDenseSetPreallocation(newmat,NULL);CHKERRQ(ierr);
840aabbc4fbSShri Abhyankar 
841aabbc4fbSShri Abhyankar   if (nz == MATRIX_BINARY_FORMAT_DENSE) { /* matrix in file is dense */
842aabbc4fbSShri Abhyankar     a = (Mat_SeqDense*)newmat->data;
843aabbc4fbSShri Abhyankar     v = a->v;
844aabbc4fbSShri Abhyankar     /* Allocate some temp space to read in the values and then flip them
845aabbc4fbSShri Abhyankar        from row major to column major */
846785e854fSJed Brown     ierr = PetscMalloc1((M*N > 0 ? M*N : 1),&w);CHKERRQ(ierr);
847aabbc4fbSShri Abhyankar     /* read in nonzero values */
848aabbc4fbSShri Abhyankar     ierr = PetscBinaryRead(fd,w,M*N,PETSC_SCALAR);CHKERRQ(ierr);
849aabbc4fbSShri Abhyankar     /* now flip the values and store them in the matrix*/
850aabbc4fbSShri Abhyankar     for (j=0; j<N; j++) {
851aabbc4fbSShri Abhyankar       for (i=0; i<M; i++) {
852aabbc4fbSShri Abhyankar         *v++ =w[i*N+j];
853aabbc4fbSShri Abhyankar       }
854aabbc4fbSShri Abhyankar     }
855aabbc4fbSShri Abhyankar     ierr = PetscFree(w);CHKERRQ(ierr);
856aabbc4fbSShri Abhyankar   } else {
857aabbc4fbSShri Abhyankar     /* read row lengths */
858785e854fSJed Brown     ierr = PetscMalloc1((M+1),&rowlengths);CHKERRQ(ierr);
859aabbc4fbSShri Abhyankar     ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr);
860aabbc4fbSShri Abhyankar 
861aabbc4fbSShri Abhyankar     a = (Mat_SeqDense*)newmat->data;
862aabbc4fbSShri Abhyankar     v = a->v;
863aabbc4fbSShri Abhyankar 
864aabbc4fbSShri Abhyankar     /* read column indices and nonzeros */
865785e854fSJed Brown     ierr = PetscMalloc1((nz+1),&scols);CHKERRQ(ierr);
866aabbc4fbSShri Abhyankar     cols = scols;
867aabbc4fbSShri Abhyankar     ierr = PetscBinaryRead(fd,cols,nz,PETSC_INT);CHKERRQ(ierr);
868785e854fSJed Brown     ierr = PetscMalloc1((nz+1),&svals);CHKERRQ(ierr);
869aabbc4fbSShri Abhyankar     vals = svals;
870aabbc4fbSShri Abhyankar     ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR);CHKERRQ(ierr);
871aabbc4fbSShri Abhyankar 
872aabbc4fbSShri Abhyankar     /* insert into matrix */
873aabbc4fbSShri Abhyankar     for (i=0; i<M; i++) {
874aabbc4fbSShri Abhyankar       for (j=0; j<rowlengths[i]; j++) v[i+M*scols[j]] = svals[j];
875aabbc4fbSShri Abhyankar       svals += rowlengths[i]; scols += rowlengths[i];
876aabbc4fbSShri Abhyankar     }
877aabbc4fbSShri Abhyankar     ierr = PetscFree(vals);CHKERRQ(ierr);
878aabbc4fbSShri Abhyankar     ierr = PetscFree(cols);CHKERRQ(ierr);
879aabbc4fbSShri Abhyankar     ierr = PetscFree(rowlengths);CHKERRQ(ierr);
880aabbc4fbSShri Abhyankar   }
881aabbc4fbSShri Abhyankar   ierr = MatAssemblyBegin(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
882aabbc4fbSShri Abhyankar   ierr = MatAssemblyEnd(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
883aabbc4fbSShri Abhyankar   PetscFunctionReturn(0);
884aabbc4fbSShri Abhyankar }
885aabbc4fbSShri Abhyankar 
886aabbc4fbSShri Abhyankar #undef __FUNCT__
8874a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqDense_ASCII"
8886849ba73SBarry Smith static PetscErrorCode MatView_SeqDense_ASCII(Mat A,PetscViewer viewer)
889289bc588SBarry Smith {
890932b0c3eSLois Curfman McInnes   Mat_SeqDense      *a = (Mat_SeqDense*)A->data;
891dfbe8321SBarry Smith   PetscErrorCode    ierr;
89213f74950SBarry Smith   PetscInt          i,j;
8932dcb1b2aSMatthew Knepley   const char        *name;
89487828ca2SBarry Smith   PetscScalar       *v;
895f3ef73ceSBarry Smith   PetscViewerFormat format;
8965f481a85SSatish Balay #if defined(PETSC_USE_COMPLEX)
897ace3abfcSBarry Smith   PetscBool allreal = PETSC_TRUE;
8985f481a85SSatish Balay #endif
899932b0c3eSLois Curfman McInnes 
9003a40ed3dSBarry Smith   PetscFunctionBegin;
901b0a32e0cSBarry Smith   ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
902456192e2SBarry Smith   if (format == PETSC_VIEWER_ASCII_INFO || format == PETSC_VIEWER_ASCII_INFO_DETAIL) {
9033a40ed3dSBarry Smith     PetscFunctionReturn(0);  /* do nothing for now */
904fb9695e5SSatish Balay   } else if (format == PETSC_VIEWER_ASCII_COMMON) {
905d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
906dae58748SBarry Smith     ierr = PetscObjectPrintClassNamePrefixType((PetscObject)A,viewer);CHKERRQ(ierr);
907d0f46423SBarry Smith     for (i=0; i<A->rmap->n; i++) {
90844cd7ae7SLois Curfman McInnes       v    = a->v + i;
90977431f27SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i);CHKERRQ(ierr);
910d0f46423SBarry Smith       for (j=0; j<A->cmap->n; j++) {
911aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX)
912329f5518SBarry Smith         if (PetscRealPart(*v) != 0.0 && PetscImaginaryPart(*v) != 0.0) {
91357622a8eSBarry Smith           ierr = PetscViewerASCIIPrintf(viewer," (%D, %g + %g i) ",j,(double)PetscRealPart(*v),(double)PetscImaginaryPart(*v));CHKERRQ(ierr);
914329f5518SBarry Smith         } else if (PetscRealPart(*v)) {
91557622a8eSBarry Smith           ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",j,(double)PetscRealPart(*v));CHKERRQ(ierr);
9166831982aSBarry Smith         }
91780cd9d93SLois Curfman McInnes #else
9186831982aSBarry Smith         if (*v) {
91957622a8eSBarry Smith           ierr = PetscViewerASCIIPrintf(viewer," (%D, %g) ",j,(double)*v);CHKERRQ(ierr);
9206831982aSBarry Smith         }
92180cd9d93SLois Curfman McInnes #endif
9221b807ce4Svictorle         v += a->lda;
92380cd9d93SLois Curfman McInnes       }
924b0a32e0cSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
92580cd9d93SLois Curfman McInnes     }
926d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
9273a40ed3dSBarry Smith   } else {
928d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_FALSE);CHKERRQ(ierr);
929aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX)
93047989497SBarry Smith     /* determine if matrix has all real values */
93147989497SBarry Smith     v = a->v;
932d0f46423SBarry Smith     for (i=0; i<A->rmap->n*A->cmap->n; i++) {
933ffac6cdbSBarry Smith       if (PetscImaginaryPart(v[i])) { allreal = PETSC_FALSE; break;}
93447989497SBarry Smith     }
93547989497SBarry Smith #endif
936fb9695e5SSatish Balay     if (format == PETSC_VIEWER_ASCII_MATLAB) {
9373a7fca6bSBarry Smith       ierr = PetscObjectGetName((PetscObject)A,&name);CHKERRQ(ierr);
938d0f46423SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"%% Size = %D %D \n",A->rmap->n,A->cmap->n);CHKERRQ(ierr);
939d0f46423SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"%s = zeros(%D,%D);\n",name,A->rmap->n,A->cmap->n);CHKERRQ(ierr);
940fb9695e5SSatish Balay       ierr = PetscViewerASCIIPrintf(viewer,"%s = [\n",name);CHKERRQ(ierr);
9417566de4bSShri Abhyankar     } else {
942dae58748SBarry Smith       ierr = PetscObjectPrintClassNamePrefixType((PetscObject)A,viewer);CHKERRQ(ierr);
943ffac6cdbSBarry Smith     }
944ffac6cdbSBarry Smith 
945d0f46423SBarry Smith     for (i=0; i<A->rmap->n; i++) {
946932b0c3eSLois Curfman McInnes       v = a->v + i;
947d0f46423SBarry Smith       for (j=0; j<A->cmap->n; j++) {
948aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX)
94947989497SBarry Smith         if (allreal) {
950c61cd2faSJed Brown           ierr = PetscViewerASCIIPrintf(viewer,"%18.16e ",(double)PetscRealPart(*v));CHKERRQ(ierr);
95147989497SBarry Smith         } else {
952c61cd2faSJed Brown           ierr = PetscViewerASCIIPrintf(viewer,"%18.16e + %18.16ei ",(double)PetscRealPart(*v),(double)PetscImaginaryPart(*v));CHKERRQ(ierr);
95347989497SBarry Smith         }
954289bc588SBarry Smith #else
955c61cd2faSJed Brown         ierr = PetscViewerASCIIPrintf(viewer,"%18.16e ",(double)*v);CHKERRQ(ierr);
956289bc588SBarry Smith #endif
9571b807ce4Svictorle         v += a->lda;
958289bc588SBarry Smith       }
959b0a32e0cSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
960289bc588SBarry Smith     }
961fb9695e5SSatish Balay     if (format == PETSC_VIEWER_ASCII_MATLAB) {
962b0a32e0cSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"];\n");CHKERRQ(ierr);
963ffac6cdbSBarry Smith     }
964d00279f6SBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_TRUE);CHKERRQ(ierr);
965da3a660dSBarry Smith   }
966b0a32e0cSBarry Smith   ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
9673a40ed3dSBarry Smith   PetscFunctionReturn(0);
968289bc588SBarry Smith }
969289bc588SBarry Smith 
9704a2ae208SSatish Balay #undef __FUNCT__
9714a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqDense_Binary"
9726849ba73SBarry Smith static PetscErrorCode MatView_SeqDense_Binary(Mat A,PetscViewer viewer)
973932b0c3eSLois Curfman McInnes {
974932b0c3eSLois Curfman McInnes   Mat_SeqDense      *a = (Mat_SeqDense*)A->data;
9756849ba73SBarry Smith   PetscErrorCode    ierr;
97613f74950SBarry Smith   int               fd;
977d0f46423SBarry Smith   PetscInt          ict,j,n = A->cmap->n,m = A->rmap->n,i,*col_lens,nz = m*n;
978f4403165SShri Abhyankar   PetscScalar       *v,*anonz,*vals;
979f4403165SShri Abhyankar   PetscViewerFormat format;
980932b0c3eSLois Curfman McInnes 
9813a40ed3dSBarry Smith   PetscFunctionBegin;
982b0a32e0cSBarry Smith   ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr);
98390ace30eSBarry Smith 
984f4403165SShri Abhyankar   ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
985f4403165SShri Abhyankar   if (format == PETSC_VIEWER_NATIVE) {
986f4403165SShri Abhyankar     /* store the matrix as a dense matrix */
987785e854fSJed Brown     ierr = PetscMalloc1(4,&col_lens);CHKERRQ(ierr);
9882205254eSKarl Rupp 
989f4403165SShri Abhyankar     col_lens[0] = MAT_FILE_CLASSID;
990f4403165SShri Abhyankar     col_lens[1] = m;
991f4403165SShri Abhyankar     col_lens[2] = n;
992f4403165SShri Abhyankar     col_lens[3] = MATRIX_BINARY_FORMAT_DENSE;
9932205254eSKarl Rupp 
994f4403165SShri Abhyankar     ierr = PetscBinaryWrite(fd,col_lens,4,PETSC_INT,PETSC_TRUE);CHKERRQ(ierr);
995f4403165SShri Abhyankar     ierr = PetscFree(col_lens);CHKERRQ(ierr);
996f4403165SShri Abhyankar 
997f4403165SShri Abhyankar     /* write out matrix, by rows */
998785e854fSJed Brown     ierr = PetscMalloc1((m*n+1),&vals);CHKERRQ(ierr);
999f4403165SShri Abhyankar     v    = a->v;
1000f4403165SShri Abhyankar     for (j=0; j<n; j++) {
1001f4403165SShri Abhyankar       for (i=0; i<m; i++) {
1002f4403165SShri Abhyankar         vals[j + i*n] = *v++;
1003f4403165SShri Abhyankar       }
1004f4403165SShri Abhyankar     }
1005f4403165SShri Abhyankar     ierr = PetscBinaryWrite(fd,vals,n*m,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr);
1006f4403165SShri Abhyankar     ierr = PetscFree(vals);CHKERRQ(ierr);
1007f4403165SShri Abhyankar   } else {
1008785e854fSJed Brown     ierr = PetscMalloc1((4+nz),&col_lens);CHKERRQ(ierr);
10092205254eSKarl Rupp 
10100700a824SBarry Smith     col_lens[0] = MAT_FILE_CLASSID;
1011932b0c3eSLois Curfman McInnes     col_lens[1] = m;
1012932b0c3eSLois Curfman McInnes     col_lens[2] = n;
1013932b0c3eSLois Curfman McInnes     col_lens[3] = nz;
1014932b0c3eSLois Curfman McInnes 
1015932b0c3eSLois Curfman McInnes     /* store lengths of each row and write (including header) to file */
1016932b0c3eSLois Curfman McInnes     for (i=0; i<m; i++) col_lens[4+i] = n;
10176f69ff64SBarry Smith     ierr = PetscBinaryWrite(fd,col_lens,4+m,PETSC_INT,PETSC_TRUE);CHKERRQ(ierr);
1018932b0c3eSLois Curfman McInnes 
1019932b0c3eSLois Curfman McInnes     /* Possibly should write in smaller increments, not whole matrix at once? */
1020932b0c3eSLois Curfman McInnes     /* store column indices (zero start index) */
1021932b0c3eSLois Curfman McInnes     ict = 0;
1022932b0c3eSLois Curfman McInnes     for (i=0; i<m; i++) {
1023932b0c3eSLois Curfman McInnes       for (j=0; j<n; j++) col_lens[ict++] = j;
1024932b0c3eSLois Curfman McInnes     }
10256f69ff64SBarry Smith     ierr = PetscBinaryWrite(fd,col_lens,nz,PETSC_INT,PETSC_FALSE);CHKERRQ(ierr);
1026606d414cSSatish Balay     ierr = PetscFree(col_lens);CHKERRQ(ierr);
1027932b0c3eSLois Curfman McInnes 
1028932b0c3eSLois Curfman McInnes     /* store nonzero values */
1029785e854fSJed Brown     ierr = PetscMalloc1((nz+1),&anonz);CHKERRQ(ierr);
1030932b0c3eSLois Curfman McInnes     ict  = 0;
1031932b0c3eSLois Curfman McInnes     for (i=0; i<m; i++) {
1032932b0c3eSLois Curfman McInnes       v = a->v + i;
1033932b0c3eSLois Curfman McInnes       for (j=0; j<n; j++) {
10341b807ce4Svictorle         anonz[ict++] = *v; v += a->lda;
1035932b0c3eSLois Curfman McInnes       }
1036932b0c3eSLois Curfman McInnes     }
10376f69ff64SBarry Smith     ierr = PetscBinaryWrite(fd,anonz,nz,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr);
1038606d414cSSatish Balay     ierr = PetscFree(anonz);CHKERRQ(ierr);
1039f4403165SShri Abhyankar   }
10403a40ed3dSBarry Smith   PetscFunctionReturn(0);
1041932b0c3eSLois Curfman McInnes }
1042932b0c3eSLois Curfman McInnes 
10439804daf3SBarry Smith #include <petscdraw.h>
10444a2ae208SSatish Balay #undef __FUNCT__
10454a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqDense_Draw_Zoom"
1046dfbe8321SBarry Smith PetscErrorCode MatView_SeqDense_Draw_Zoom(PetscDraw draw,void *Aa)
1047f1af5d2fSBarry Smith {
1048f1af5d2fSBarry Smith   Mat               A  = (Mat) Aa;
1049f1af5d2fSBarry Smith   Mat_SeqDense      *a = (Mat_SeqDense*)A->data;
10506849ba73SBarry Smith   PetscErrorCode    ierr;
1051d0f46423SBarry Smith   PetscInt          m  = A->rmap->n,n = A->cmap->n,color,i,j;
105287828ca2SBarry Smith   PetscScalar       *v = a->v;
1053b0a32e0cSBarry Smith   PetscViewer       viewer;
1054b0a32e0cSBarry Smith   PetscDraw         popup;
1055329f5518SBarry Smith   PetscReal         xl,yl,xr,yr,x_l,x_r,y_l,y_r,scale,maxv = 0.0;
1056f3ef73ceSBarry Smith   PetscViewerFormat format;
1057f1af5d2fSBarry Smith 
1058f1af5d2fSBarry Smith   PetscFunctionBegin;
1059f1af5d2fSBarry Smith   ierr = PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer);CHKERRQ(ierr);
1060b0a32e0cSBarry Smith   ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
1061b0a32e0cSBarry Smith   ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr);
1062f1af5d2fSBarry Smith 
1063f1af5d2fSBarry Smith   /* Loop over matrix elements drawing boxes */
1064fb9695e5SSatish Balay   if (format != PETSC_VIEWER_DRAW_CONTOUR) {
1065f1af5d2fSBarry Smith     /* Blue for negative and Red for positive */
1066b0a32e0cSBarry Smith     color = PETSC_DRAW_BLUE;
1067f1af5d2fSBarry Smith     for (j = 0; j < n; j++) {
1068f1af5d2fSBarry Smith       x_l = j;
1069f1af5d2fSBarry Smith       x_r = x_l + 1.0;
1070f1af5d2fSBarry Smith       for (i = 0; i < m; i++) {
1071f1af5d2fSBarry Smith         y_l = m - i - 1.0;
1072f1af5d2fSBarry Smith         y_r = y_l + 1.0;
1073329f5518SBarry Smith         if (PetscRealPart(v[j*m+i]) >  0.) {
1074b0a32e0cSBarry Smith           color = PETSC_DRAW_RED;
1075329f5518SBarry Smith         } else if (PetscRealPart(v[j*m+i]) <  0.) {
1076b0a32e0cSBarry Smith           color = PETSC_DRAW_BLUE;
1077f1af5d2fSBarry Smith         } else {
1078f1af5d2fSBarry Smith           continue;
1079f1af5d2fSBarry Smith         }
1080b0a32e0cSBarry Smith         ierr = PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);CHKERRQ(ierr);
1081f1af5d2fSBarry Smith       }
1082f1af5d2fSBarry Smith     }
1083f1af5d2fSBarry Smith   } else {
1084f1af5d2fSBarry Smith     /* use contour shading to indicate magnitude of values */
1085f1af5d2fSBarry Smith     /* first determine max of all nonzero values */
1086f1af5d2fSBarry Smith     for (i = 0; i < m*n; i++) {
1087f1af5d2fSBarry Smith       if (PetscAbsScalar(v[i]) > maxv) maxv = PetscAbsScalar(v[i]);
1088f1af5d2fSBarry Smith     }
1089b0a32e0cSBarry Smith     scale = (245.0 - PETSC_DRAW_BASIC_COLORS)/maxv;
1090b0a32e0cSBarry Smith     ierr  = PetscDrawGetPopup(draw,&popup);CHKERRQ(ierr);
1091b0a32e0cSBarry Smith     if (popup) {ierr = PetscDrawScalePopup(popup,0.0,maxv);CHKERRQ(ierr);}
1092f1af5d2fSBarry Smith     for (j = 0; j < n; j++) {
1093f1af5d2fSBarry Smith       x_l = j;
1094f1af5d2fSBarry Smith       x_r = x_l + 1.0;
1095f1af5d2fSBarry Smith       for (i = 0; i < m; i++) {
1096f1af5d2fSBarry Smith         y_l   = m - i - 1.0;
1097f1af5d2fSBarry Smith         y_r   = y_l + 1.0;
1098b0a32e0cSBarry Smith         color = PETSC_DRAW_BASIC_COLORS + (int)(scale*PetscAbsScalar(v[j*m+i]));
1099b0a32e0cSBarry Smith         ierr  = PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);CHKERRQ(ierr);
1100f1af5d2fSBarry Smith       }
1101f1af5d2fSBarry Smith     }
1102f1af5d2fSBarry Smith   }
1103f1af5d2fSBarry Smith   PetscFunctionReturn(0);
1104f1af5d2fSBarry Smith }
1105f1af5d2fSBarry Smith 
11064a2ae208SSatish Balay #undef __FUNCT__
11074a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqDense_Draw"
1108dfbe8321SBarry Smith PetscErrorCode MatView_SeqDense_Draw(Mat A,PetscViewer viewer)
1109f1af5d2fSBarry Smith {
1110b0a32e0cSBarry Smith   PetscDraw      draw;
1111ace3abfcSBarry Smith   PetscBool      isnull;
1112329f5518SBarry Smith   PetscReal      xr,yr,xl,yl,h,w;
1113dfbe8321SBarry Smith   PetscErrorCode ierr;
1114f1af5d2fSBarry Smith 
1115f1af5d2fSBarry Smith   PetscFunctionBegin;
1116b0a32e0cSBarry Smith   ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
1117b0a32e0cSBarry Smith   ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr);
1118abc0a331SBarry Smith   if (isnull) PetscFunctionReturn(0);
1119f1af5d2fSBarry Smith 
1120f1af5d2fSBarry Smith   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer);CHKERRQ(ierr);
1121d0f46423SBarry Smith   xr   = A->cmap->n; yr = A->rmap->n; h = yr/10.0; w = xr/10.0;
1122f1af5d2fSBarry Smith   xr  += w;    yr += h;  xl = -w;     yl = -h;
1123b0a32e0cSBarry Smith   ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr);
1124b0a32e0cSBarry Smith   ierr = PetscDrawZoom(draw,MatView_SeqDense_Draw_Zoom,A);CHKERRQ(ierr);
11250298fd71SBarry Smith   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",NULL);CHKERRQ(ierr);
1126f1af5d2fSBarry Smith   PetscFunctionReturn(0);
1127f1af5d2fSBarry Smith }
1128f1af5d2fSBarry Smith 
11294a2ae208SSatish Balay #undef __FUNCT__
11304a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqDense"
1131dfbe8321SBarry Smith PetscErrorCode MatView_SeqDense(Mat A,PetscViewer viewer)
1132932b0c3eSLois Curfman McInnes {
1133dfbe8321SBarry Smith   PetscErrorCode ierr;
1134ace3abfcSBarry Smith   PetscBool      iascii,isbinary,isdraw;
1135932b0c3eSLois Curfman McInnes 
11363a40ed3dSBarry Smith   PetscFunctionBegin;
1137251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
1138251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr);
1139251f4c67SDmitry Karpeev   ierr = PetscObjectTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr);
11400f5bd95cSBarry Smith 
1141c45a1595SBarry Smith   if (iascii) {
1142c45a1595SBarry Smith     ierr = MatView_SeqDense_ASCII(A,viewer);CHKERRQ(ierr);
11430f5bd95cSBarry Smith   } else if (isbinary) {
11443a40ed3dSBarry Smith     ierr = MatView_SeqDense_Binary(A,viewer);CHKERRQ(ierr);
1145f1af5d2fSBarry Smith   } else if (isdraw) {
1146f1af5d2fSBarry Smith     ierr = MatView_SeqDense_Draw(A,viewer);CHKERRQ(ierr);
1147932b0c3eSLois Curfman McInnes   }
11483a40ed3dSBarry Smith   PetscFunctionReturn(0);
1149932b0c3eSLois Curfman McInnes }
1150289bc588SBarry Smith 
11514a2ae208SSatish Balay #undef __FUNCT__
11524a2ae208SSatish Balay #define __FUNCT__ "MatDestroy_SeqDense"
1153dfbe8321SBarry Smith PetscErrorCode MatDestroy_SeqDense(Mat mat)
1154289bc588SBarry Smith {
1155ec8511deSBarry Smith   Mat_SeqDense   *l = (Mat_SeqDense*)mat->data;
1156dfbe8321SBarry Smith   PetscErrorCode ierr;
115790f02eecSBarry Smith 
11583a40ed3dSBarry Smith   PetscFunctionBegin;
1159aa482453SBarry Smith #if defined(PETSC_USE_LOG)
1160d0f46423SBarry Smith   PetscLogObjectState((PetscObject)mat,"Rows %D Cols %D",mat->rmap->n,mat->cmap->n);
1161a5a9c739SBarry Smith #endif
116205b42c5fSBarry Smith   ierr = PetscFree(l->pivots);CHKERRQ(ierr);
11636857c123SSatish Balay   if (!l->user_alloc) {ierr = PetscFree(l->v);CHKERRQ(ierr);}
1164bf0cc555SLisandro Dalcin   ierr = PetscFree(mat->data);CHKERRQ(ierr);
1165dbd8c25aSHong Zhang 
1166dbd8c25aSHong Zhang   ierr = PetscObjectChangeTypeName((PetscObject)mat,0);CHKERRQ(ierr);
1167bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatDenseGetArray_C",NULL);CHKERRQ(ierr);
1168bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatDenseRestoreArray_C",NULL);CHKERRQ(ierr);
1169bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatSeqDenseSetPreallocation_C",NULL);CHKERRQ(ierr);
1170bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMatMult_seqaij_seqdense_C",NULL);CHKERRQ(ierr);
1171bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMatMultSymbolic_seqaij_seqdense_C",NULL);CHKERRQ(ierr);
1172bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatMatMultNumeric_seqaij_seqdense_C",NULL);CHKERRQ(ierr);
11733bf78175SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatTransposeMatMult_seqaij_seqdense_C",NULL);CHKERRQ(ierr);
11743bf78175SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatTransposeMatMultSymbolic_seqaij_seqdense_C",NULL);CHKERRQ(ierr);
11753bf78175SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)mat,"MatTransposeMatMultNumeric_seqaij_seqdense_C",NULL);CHKERRQ(ierr);
11763a40ed3dSBarry Smith   PetscFunctionReturn(0);
1177289bc588SBarry Smith }
1178289bc588SBarry Smith 
11794a2ae208SSatish Balay #undef __FUNCT__
11804a2ae208SSatish Balay #define __FUNCT__ "MatTranspose_SeqDense"
1181fc4dec0aSBarry Smith PetscErrorCode MatTranspose_SeqDense(Mat A,MatReuse reuse,Mat *matout)
1182289bc588SBarry Smith {
1183c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
11846849ba73SBarry Smith   PetscErrorCode ierr;
118513f74950SBarry Smith   PetscInt       k,j,m,n,M;
118687828ca2SBarry Smith   PetscScalar    *v,tmp;
118748b35521SBarry Smith 
11883a40ed3dSBarry Smith   PetscFunctionBegin;
1189d0f46423SBarry Smith   v = mat->v; m = A->rmap->n; M = mat->lda; n = A->cmap->n;
1190e9695a30SBarry Smith   if (reuse == MAT_REUSE_MATRIX && *matout == A) { /* in place transpose */
1191e7e72b3dSBarry Smith     if (m != n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Can not transpose non-square matrix in place");
1192e7e72b3dSBarry Smith     else {
1193d3e5ee88SLois Curfman McInnes       for (j=0; j<m; j++) {
1194289bc588SBarry Smith         for (k=0; k<j; k++) {
11951b807ce4Svictorle           tmp        = v[j + k*M];
11961b807ce4Svictorle           v[j + k*M] = v[k + j*M];
11971b807ce4Svictorle           v[k + j*M] = tmp;
1198289bc588SBarry Smith         }
1199289bc588SBarry Smith       }
1200d64ed03dSBarry Smith     }
12013a40ed3dSBarry Smith   } else { /* out-of-place transpose */
1202d3e5ee88SLois Curfman McInnes     Mat          tmat;
1203ec8511deSBarry Smith     Mat_SeqDense *tmatd;
120487828ca2SBarry Smith     PetscScalar  *v2;
1205ea709b57SSatish Balay 
1206fc4dec0aSBarry Smith     if (reuse == MAT_INITIAL_MATRIX) {
1207ce94432eSBarry Smith       ierr = MatCreate(PetscObjectComm((PetscObject)A),&tmat);CHKERRQ(ierr);
1208d0f46423SBarry Smith       ierr = MatSetSizes(tmat,A->cmap->n,A->rmap->n,A->cmap->n,A->rmap->n);CHKERRQ(ierr);
12097adad957SLisandro Dalcin       ierr = MatSetType(tmat,((PetscObject)A)->type_name);CHKERRQ(ierr);
12100298fd71SBarry Smith       ierr = MatSeqDenseSetPreallocation(tmat,NULL);CHKERRQ(ierr);
1211fc4dec0aSBarry Smith     } else {
1212fc4dec0aSBarry Smith       tmat = *matout;
1213fc4dec0aSBarry Smith     }
1214ec8511deSBarry Smith     tmatd = (Mat_SeqDense*)tmat->data;
12150de55854SLois Curfman McInnes     v = mat->v; v2 = tmatd->v;
1216d3e5ee88SLois Curfman McInnes     for (j=0; j<n; j++) {
12171b807ce4Svictorle       for (k=0; k<m; k++) v2[j + k*n] = v[k + j*M];
1218d3e5ee88SLois Curfman McInnes     }
12196d4a8577SBarry Smith     ierr = MatAssemblyBegin(tmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
12206d4a8577SBarry Smith     ierr = MatAssemblyEnd(tmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
1221d3e5ee88SLois Curfman McInnes     *matout = tmat;
122248b35521SBarry Smith   }
12233a40ed3dSBarry Smith   PetscFunctionReturn(0);
1224289bc588SBarry Smith }
1225289bc588SBarry Smith 
12264a2ae208SSatish Balay #undef __FUNCT__
12274a2ae208SSatish Balay #define __FUNCT__ "MatEqual_SeqDense"
1228ace3abfcSBarry Smith PetscErrorCode MatEqual_SeqDense(Mat A1,Mat A2,PetscBool  *flg)
1229289bc588SBarry Smith {
1230c0bbcb79SLois Curfman McInnes   Mat_SeqDense *mat1 = (Mat_SeqDense*)A1->data;
1231c0bbcb79SLois Curfman McInnes   Mat_SeqDense *mat2 = (Mat_SeqDense*)A2->data;
123213f74950SBarry Smith   PetscInt     i,j;
1233a2ea699eSBarry Smith   PetscScalar  *v1,*v2;
12349ea5d5aeSSatish Balay 
12353a40ed3dSBarry Smith   PetscFunctionBegin;
1236d0f46423SBarry Smith   if (A1->rmap->n != A2->rmap->n) {*flg = PETSC_FALSE; PetscFunctionReturn(0);}
1237d0f46423SBarry Smith   if (A1->cmap->n != A2->cmap->n) {*flg = PETSC_FALSE; PetscFunctionReturn(0);}
1238d0f46423SBarry Smith   for (i=0; i<A1->rmap->n; i++) {
12391b807ce4Svictorle     v1 = mat1->v+i; v2 = mat2->v+i;
1240d0f46423SBarry Smith     for (j=0; j<A1->cmap->n; j++) {
12413a40ed3dSBarry Smith       if (*v1 != *v2) {*flg = PETSC_FALSE; PetscFunctionReturn(0);}
12421b807ce4Svictorle       v1 += mat1->lda; v2 += mat2->lda;
12431b807ce4Svictorle     }
1244289bc588SBarry Smith   }
124577c4ece6SBarry Smith   *flg = PETSC_TRUE;
12463a40ed3dSBarry Smith   PetscFunctionReturn(0);
1247289bc588SBarry Smith }
1248289bc588SBarry Smith 
12494a2ae208SSatish Balay #undef __FUNCT__
12504a2ae208SSatish Balay #define __FUNCT__ "MatGetDiagonal_SeqDense"
1251dfbe8321SBarry Smith PetscErrorCode MatGetDiagonal_SeqDense(Mat A,Vec v)
1252289bc588SBarry Smith {
1253c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
1254dfbe8321SBarry Smith   PetscErrorCode ierr;
125513f74950SBarry Smith   PetscInt       i,n,len;
125687828ca2SBarry Smith   PetscScalar    *x,zero = 0.0;
125744cd7ae7SLois Curfman McInnes 
12583a40ed3dSBarry Smith   PetscFunctionBegin;
12592dcb1b2aSMatthew Knepley   ierr = VecSet(v,zero);CHKERRQ(ierr);
12607a97a34bSBarry Smith   ierr = VecGetSize(v,&n);CHKERRQ(ierr);
12611ebc52fbSHong Zhang   ierr = VecGetArray(v,&x);CHKERRQ(ierr);
1262d0f46423SBarry Smith   len  = PetscMin(A->rmap->n,A->cmap->n);
1263e32f2f54SBarry Smith   if (n != A->rmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Nonconforming mat and vec");
126444cd7ae7SLois Curfman McInnes   for (i=0; i<len; i++) {
12651b807ce4Svictorle     x[i] = mat->v[i*mat->lda + i];
1266289bc588SBarry Smith   }
12671ebc52fbSHong Zhang   ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
12683a40ed3dSBarry Smith   PetscFunctionReturn(0);
1269289bc588SBarry Smith }
1270289bc588SBarry Smith 
12714a2ae208SSatish Balay #undef __FUNCT__
12724a2ae208SSatish Balay #define __FUNCT__ "MatDiagonalScale_SeqDense"
1273dfbe8321SBarry Smith PetscErrorCode MatDiagonalScale_SeqDense(Mat A,Vec ll,Vec rr)
1274289bc588SBarry Smith {
1275c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
127687828ca2SBarry Smith   PetscScalar    *l,*r,x,*v;
1277dfbe8321SBarry Smith   PetscErrorCode ierr;
1278d0f46423SBarry Smith   PetscInt       i,j,m = A->rmap->n,n = A->cmap->n;
127955659b69SBarry Smith 
12803a40ed3dSBarry Smith   PetscFunctionBegin;
128128988994SBarry Smith   if (ll) {
12827a97a34bSBarry Smith     ierr = VecGetSize(ll,&m);CHKERRQ(ierr);
12831ebc52fbSHong Zhang     ierr = VecGetArray(ll,&l);CHKERRQ(ierr);
1284e32f2f54SBarry Smith     if (m != A->rmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Left scaling vec wrong size");
1285da3a660dSBarry Smith     for (i=0; i<m; i++) {
1286da3a660dSBarry Smith       x = l[i];
1287da3a660dSBarry Smith       v = mat->v + i;
1288da3a660dSBarry Smith       for (j=0; j<n; j++) { (*v) *= x; v+= m;}
1289da3a660dSBarry Smith     }
12901ebc52fbSHong Zhang     ierr = VecRestoreArray(ll,&l);CHKERRQ(ierr);
1291efee365bSSatish Balay     ierr = PetscLogFlops(n*m);CHKERRQ(ierr);
1292da3a660dSBarry Smith   }
129328988994SBarry Smith   if (rr) {
12947a97a34bSBarry Smith     ierr = VecGetSize(rr,&n);CHKERRQ(ierr);
12951ebc52fbSHong Zhang     ierr = VecGetArray(rr,&r);CHKERRQ(ierr);
1296e32f2f54SBarry Smith     if (n != A->cmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Right scaling vec wrong size");
1297da3a660dSBarry Smith     for (i=0; i<n; i++) {
1298da3a660dSBarry Smith       x = r[i];
1299da3a660dSBarry Smith       v = mat->v + i*m;
13002205254eSKarl Rupp       for (j=0; j<m; j++) (*v++) *= x;
1301da3a660dSBarry Smith     }
13021ebc52fbSHong Zhang     ierr = VecRestoreArray(rr,&r);CHKERRQ(ierr);
1303efee365bSSatish Balay     ierr = PetscLogFlops(n*m);CHKERRQ(ierr);
1304da3a660dSBarry Smith   }
13053a40ed3dSBarry Smith   PetscFunctionReturn(0);
1306289bc588SBarry Smith }
1307289bc588SBarry Smith 
13084a2ae208SSatish Balay #undef __FUNCT__
13094a2ae208SSatish Balay #define __FUNCT__ "MatNorm_SeqDense"
1310dfbe8321SBarry Smith PetscErrorCode MatNorm_SeqDense(Mat A,NormType type,PetscReal *nrm)
1311289bc588SBarry Smith {
1312c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
131387828ca2SBarry Smith   PetscScalar    *v   = mat->v;
1314329f5518SBarry Smith   PetscReal      sum  = 0.0;
1315d0f46423SBarry Smith   PetscInt       lda  =mat->lda,m=A->rmap->n,i,j;
1316efee365bSSatish Balay   PetscErrorCode ierr;
131755659b69SBarry Smith 
13183a40ed3dSBarry Smith   PetscFunctionBegin;
1319289bc588SBarry Smith   if (type == NORM_FROBENIUS) {
1320a5ce6ee0Svictorle     if (lda>m) {
1321d0f46423SBarry Smith       for (j=0; j<A->cmap->n; j++) {
1322a5ce6ee0Svictorle         v = mat->v+j*lda;
1323a5ce6ee0Svictorle         for (i=0; i<m; i++) {
1324a5ce6ee0Svictorle           sum += PetscRealPart(PetscConj(*v)*(*v)); v++;
1325a5ce6ee0Svictorle         }
1326a5ce6ee0Svictorle       }
1327a5ce6ee0Svictorle     } else {
1328d0f46423SBarry Smith       for (i=0; i<A->cmap->n*A->rmap->n; i++) {
1329329f5518SBarry Smith         sum += PetscRealPart(PetscConj(*v)*(*v)); v++;
1330289bc588SBarry Smith       }
1331a5ce6ee0Svictorle     }
13328f1a2a5eSBarry Smith     *nrm = PetscSqrtReal(sum);
1333dc0b31edSSatish Balay     ierr = PetscLogFlops(2.0*A->cmap->n*A->rmap->n);CHKERRQ(ierr);
13343a40ed3dSBarry Smith   } else if (type == NORM_1) {
1335064f8208SBarry Smith     *nrm = 0.0;
1336d0f46423SBarry Smith     for (j=0; j<A->cmap->n; j++) {
13371b807ce4Svictorle       v   = mat->v + j*mat->lda;
1338289bc588SBarry Smith       sum = 0.0;
1339d0f46423SBarry Smith       for (i=0; i<A->rmap->n; i++) {
134033a8263dSBarry Smith         sum += PetscAbsScalar(*v);  v++;
1341289bc588SBarry Smith       }
1342064f8208SBarry Smith       if (sum > *nrm) *nrm = sum;
1343289bc588SBarry Smith     }
1344d0f46423SBarry Smith     ierr = PetscLogFlops(A->cmap->n*A->rmap->n);CHKERRQ(ierr);
13453a40ed3dSBarry Smith   } else if (type == NORM_INFINITY) {
1346064f8208SBarry Smith     *nrm = 0.0;
1347d0f46423SBarry Smith     for (j=0; j<A->rmap->n; j++) {
1348289bc588SBarry Smith       v   = mat->v + j;
1349289bc588SBarry Smith       sum = 0.0;
1350d0f46423SBarry Smith       for (i=0; i<A->cmap->n; i++) {
13511b807ce4Svictorle         sum += PetscAbsScalar(*v); v += mat->lda;
1352289bc588SBarry Smith       }
1353064f8208SBarry Smith       if (sum > *nrm) *nrm = sum;
1354289bc588SBarry Smith     }
1355d0f46423SBarry Smith     ierr = PetscLogFlops(A->cmap->n*A->rmap->n);CHKERRQ(ierr);
1356e7e72b3dSBarry Smith   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No two norm");
13573a40ed3dSBarry Smith   PetscFunctionReturn(0);
1358289bc588SBarry Smith }
1359289bc588SBarry Smith 
13604a2ae208SSatish Balay #undef __FUNCT__
13614a2ae208SSatish Balay #define __FUNCT__ "MatSetOption_SeqDense"
1362ace3abfcSBarry Smith PetscErrorCode MatSetOption_SeqDense(Mat A,MatOption op,PetscBool flg)
1363289bc588SBarry Smith {
1364c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *aij = (Mat_SeqDense*)A->data;
136563ba0a88SBarry Smith   PetscErrorCode ierr;
136667e560aaSBarry Smith 
13673a40ed3dSBarry Smith   PetscFunctionBegin;
1368b5a2b587SKris Buschelman   switch (op) {
1369b5a2b587SKris Buschelman   case MAT_ROW_ORIENTED:
13704e0d8c25SBarry Smith     aij->roworiented = flg;
1371b5a2b587SKris Buschelman     break;
1372512a5fc5SBarry Smith   case MAT_NEW_NONZERO_LOCATIONS:
1373b5a2b587SKris Buschelman   case MAT_NEW_NONZERO_LOCATION_ERR:
13743971808eSMatthew Knepley   case MAT_NEW_NONZERO_ALLOCATION_ERR:
13754e0d8c25SBarry Smith   case MAT_NEW_DIAGONALS:
137613fa8e87SLisandro Dalcin   case MAT_KEEP_NONZERO_PATTERN:
1377b5a2b587SKris Buschelman   case MAT_IGNORE_OFF_PROC_ENTRIES:
1378b5a2b587SKris Buschelman   case MAT_USE_HASH_TABLE:
13795021d80fSJed Brown   case MAT_IGNORE_LOWER_TRIANGULAR:
13805021d80fSJed Brown     ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr);
13815021d80fSJed Brown     break;
13825021d80fSJed Brown   case MAT_SPD:
138377e54ba9SKris Buschelman   case MAT_SYMMETRIC:
138477e54ba9SKris Buschelman   case MAT_STRUCTURALLY_SYMMETRIC:
13859a4540c5SBarry Smith   case MAT_HERMITIAN:
13869a4540c5SBarry Smith   case MAT_SYMMETRY_ETERNAL:
13875021d80fSJed Brown     /* These options are handled directly by MatSetOption() */
138877e54ba9SKris Buschelman     break;
1389b5a2b587SKris Buschelman   default:
1390e32f2f54SBarry Smith     SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"unknown option %s",MatOptions[op]);
13913a40ed3dSBarry Smith   }
13923a40ed3dSBarry Smith   PetscFunctionReturn(0);
1393289bc588SBarry Smith }
1394289bc588SBarry Smith 
13954a2ae208SSatish Balay #undef __FUNCT__
13964a2ae208SSatish Balay #define __FUNCT__ "MatZeroEntries_SeqDense"
1397dfbe8321SBarry Smith PetscErrorCode MatZeroEntries_SeqDense(Mat A)
13986f0a148fSBarry Smith {
1399ec8511deSBarry Smith   Mat_SeqDense   *l = (Mat_SeqDense*)A->data;
14006849ba73SBarry Smith   PetscErrorCode ierr;
1401d0f46423SBarry Smith   PetscInt       lda=l->lda,m=A->rmap->n,j;
14023a40ed3dSBarry Smith 
14033a40ed3dSBarry Smith   PetscFunctionBegin;
1404a5ce6ee0Svictorle   if (lda>m) {
1405d0f46423SBarry Smith     for (j=0; j<A->cmap->n; j++) {
1406a5ce6ee0Svictorle       ierr = PetscMemzero(l->v+j*lda,m*sizeof(PetscScalar));CHKERRQ(ierr);
1407a5ce6ee0Svictorle     }
1408a5ce6ee0Svictorle   } else {
1409d0f46423SBarry Smith     ierr = PetscMemzero(l->v,A->rmap->n*A->cmap->n*sizeof(PetscScalar));CHKERRQ(ierr);
1410a5ce6ee0Svictorle   }
14113a40ed3dSBarry Smith   PetscFunctionReturn(0);
14126f0a148fSBarry Smith }
14136f0a148fSBarry Smith 
14144a2ae208SSatish Balay #undef __FUNCT__
14154a2ae208SSatish Balay #define __FUNCT__ "MatZeroRows_SeqDense"
14162b40b63fSBarry Smith PetscErrorCode MatZeroRows_SeqDense(Mat A,PetscInt N,const PetscInt rows[],PetscScalar diag,Vec x,Vec b)
14176f0a148fSBarry Smith {
141897b48c8fSBarry Smith   PetscErrorCode    ierr;
1419ec8511deSBarry Smith   Mat_SeqDense      *l = (Mat_SeqDense*)A->data;
1420b9679d65SBarry Smith   PetscInt          m  = l->lda, n = A->cmap->n, i,j;
142197b48c8fSBarry Smith   PetscScalar       *slot,*bb;
142297b48c8fSBarry Smith   const PetscScalar *xx;
142355659b69SBarry Smith 
14243a40ed3dSBarry Smith   PetscFunctionBegin;
1425b9679d65SBarry Smith #if defined(PETSC_USE_DEBUG)
1426b9679d65SBarry Smith   for (i=0; i<N; i++) {
1427b9679d65SBarry Smith     if (rows[i] < 0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Negative row requested to be zeroed");
1428b9679d65SBarry Smith     if (rows[i] >= A->rmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_OUTOFRANGE,"Row %D requested to be zeroed greater than or equal number of rows %D",rows[i],A->rmap->n);
1429b9679d65SBarry Smith   }
1430b9679d65SBarry Smith #endif
1431b9679d65SBarry Smith 
143297b48c8fSBarry Smith   /* fix right hand side if needed */
143397b48c8fSBarry Smith   if (x && b) {
143497b48c8fSBarry Smith     ierr = VecGetArrayRead(x,&xx);CHKERRQ(ierr);
143597b48c8fSBarry Smith     ierr = VecGetArray(b,&bb);CHKERRQ(ierr);
14362205254eSKarl Rupp     for (i=0; i<N; i++) bb[rows[i]] = diag*xx[rows[i]];
143797b48c8fSBarry Smith     ierr = VecRestoreArrayRead(x,&xx);CHKERRQ(ierr);
143897b48c8fSBarry Smith     ierr = VecRestoreArray(b,&bb);CHKERRQ(ierr);
143997b48c8fSBarry Smith   }
144097b48c8fSBarry Smith 
14416f0a148fSBarry Smith   for (i=0; i<N; i++) {
14426f0a148fSBarry Smith     slot = l->v + rows[i];
1443b9679d65SBarry Smith     for (j=0; j<n; j++) { *slot = 0.0; slot += m;}
14446f0a148fSBarry Smith   }
1445f4df32b1SMatthew Knepley   if (diag != 0.0) {
1446b9679d65SBarry Smith     if (A->rmap->n != A->cmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Only coded for square matrices");
14476f0a148fSBarry Smith     for (i=0; i<N; i++) {
1448b9679d65SBarry Smith       slot  = l->v + (m+1)*rows[i];
1449f4df32b1SMatthew Knepley       *slot = diag;
14506f0a148fSBarry Smith     }
14516f0a148fSBarry Smith   }
14523a40ed3dSBarry Smith   PetscFunctionReturn(0);
14536f0a148fSBarry Smith }
1454557bce09SLois Curfman McInnes 
14554a2ae208SSatish Balay #undef __FUNCT__
14568c778c55SBarry Smith #define __FUNCT__ "MatDenseGetArray_SeqDense"
14578c778c55SBarry Smith PetscErrorCode MatDenseGetArray_SeqDense(Mat A,PetscScalar *array[])
145864e87e97SBarry Smith {
1459c0bbcb79SLois Curfman McInnes   Mat_SeqDense *mat = (Mat_SeqDense*)A->data;
14603a40ed3dSBarry Smith 
14613a40ed3dSBarry Smith   PetscFunctionBegin;
1462e32f2f54SBarry Smith   if (mat->lda != A->rmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot get array for Dense matrices with LDA different from number of rows");
146364e87e97SBarry Smith   *array = mat->v;
14643a40ed3dSBarry Smith   PetscFunctionReturn(0);
146564e87e97SBarry Smith }
14660754003eSLois Curfman McInnes 
14674a2ae208SSatish Balay #undef __FUNCT__
14688c778c55SBarry Smith #define __FUNCT__ "MatDenseRestoreArray_SeqDense"
14698c778c55SBarry Smith PetscErrorCode MatDenseRestoreArray_SeqDense(Mat A,PetscScalar *array[])
1470ff14e315SSatish Balay {
14713a40ed3dSBarry Smith   PetscFunctionBegin;
147209b544d4SBarry Smith   *array = 0; /* user cannot accidently use the array later */
14733a40ed3dSBarry Smith   PetscFunctionReturn(0);
1474ff14e315SSatish Balay }
14750754003eSLois Curfman McInnes 
14764a2ae208SSatish Balay #undef __FUNCT__
14778c778c55SBarry Smith #define __FUNCT__ "MatDenseGetArray"
1478dec5eb66SMatthew G Knepley /*@C
14798c778c55SBarry Smith    MatDenseGetArray - gives access to the array where the data for a SeqDense matrix is stored
148073a71a0fSBarry Smith 
148173a71a0fSBarry Smith    Not Collective
148273a71a0fSBarry Smith 
148373a71a0fSBarry Smith    Input Parameter:
148473a71a0fSBarry Smith .  mat - a MATSEQDENSE matrix
148573a71a0fSBarry Smith 
148673a71a0fSBarry Smith    Output Parameter:
148773a71a0fSBarry Smith .   array - pointer to the data
148873a71a0fSBarry Smith 
148973a71a0fSBarry Smith    Level: intermediate
149073a71a0fSBarry Smith 
14918c778c55SBarry Smith .seealso: MatDenseRestoreArray()
149273a71a0fSBarry Smith @*/
14938c778c55SBarry Smith PetscErrorCode  MatDenseGetArray(Mat A,PetscScalar **array)
149473a71a0fSBarry Smith {
149573a71a0fSBarry Smith   PetscErrorCode ierr;
149673a71a0fSBarry Smith 
149773a71a0fSBarry Smith   PetscFunctionBegin;
14988c778c55SBarry Smith   ierr = PetscUseMethod(A,"MatDenseGetArray_C",(Mat,PetscScalar**),(A,array));CHKERRQ(ierr);
149973a71a0fSBarry Smith   PetscFunctionReturn(0);
150073a71a0fSBarry Smith }
150173a71a0fSBarry Smith 
150273a71a0fSBarry Smith #undef __FUNCT__
15038c778c55SBarry Smith #define __FUNCT__ "MatDenseRestoreArray"
1504dec5eb66SMatthew G Knepley /*@C
15058c778c55SBarry Smith    MatDenseRestoreArray - returns access to the array where the data for a SeqDense matrix is stored obtained by MatDenseGetArray()
150673a71a0fSBarry Smith 
150773a71a0fSBarry Smith    Not Collective
150873a71a0fSBarry Smith 
150973a71a0fSBarry Smith    Input Parameters:
151073a71a0fSBarry Smith .  mat - a MATSEQDENSE matrix
151173a71a0fSBarry Smith .  array - pointer to the data
151273a71a0fSBarry Smith 
151373a71a0fSBarry Smith    Level: intermediate
151473a71a0fSBarry Smith 
15158c778c55SBarry Smith .seealso: MatDenseGetArray()
151673a71a0fSBarry Smith @*/
15178c778c55SBarry Smith PetscErrorCode  MatDenseRestoreArray(Mat A,PetscScalar **array)
151873a71a0fSBarry Smith {
151973a71a0fSBarry Smith   PetscErrorCode ierr;
152073a71a0fSBarry Smith 
152173a71a0fSBarry Smith   PetscFunctionBegin;
15228c778c55SBarry Smith   ierr = PetscUseMethod(A,"MatDenseRestoreArray_C",(Mat,PetscScalar**),(A,array));CHKERRQ(ierr);
152373a71a0fSBarry Smith   PetscFunctionReturn(0);
152473a71a0fSBarry Smith }
152573a71a0fSBarry Smith 
152673a71a0fSBarry Smith #undef __FUNCT__
15274a2ae208SSatish Balay #define __FUNCT__ "MatGetSubMatrix_SeqDense"
152813f74950SBarry Smith static PetscErrorCode MatGetSubMatrix_SeqDense(Mat A,IS isrow,IS iscol,PetscInt cs,MatReuse scall,Mat *B)
15290754003eSLois Curfman McInnes {
1530c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
15316849ba73SBarry Smith   PetscErrorCode ierr;
15325d0c19d7SBarry Smith   PetscInt       i,j,nrows,ncols;
15335d0c19d7SBarry Smith   const PetscInt *irow,*icol;
153487828ca2SBarry Smith   PetscScalar    *av,*bv,*v = mat->v;
15350754003eSLois Curfman McInnes   Mat            newmat;
15360754003eSLois Curfman McInnes 
15373a40ed3dSBarry Smith   PetscFunctionBegin;
153878b31e54SBarry Smith   ierr = ISGetIndices(isrow,&irow);CHKERRQ(ierr);
153978b31e54SBarry Smith   ierr = ISGetIndices(iscol,&icol);CHKERRQ(ierr);
1540e03a110bSBarry Smith   ierr = ISGetLocalSize(isrow,&nrows);CHKERRQ(ierr);
1541e03a110bSBarry Smith   ierr = ISGetLocalSize(iscol,&ncols);CHKERRQ(ierr);
15420754003eSLois Curfman McInnes 
1543182d2002SSatish Balay   /* Check submatrixcall */
1544182d2002SSatish Balay   if (scall == MAT_REUSE_MATRIX) {
154513f74950SBarry Smith     PetscInt n_cols,n_rows;
1546182d2002SSatish Balay     ierr = MatGetSize(*B,&n_rows,&n_cols);CHKERRQ(ierr);
154721a2c019SBarry Smith     if (n_rows != nrows || n_cols != ncols) {
1548f746d493SDmitry Karpeev       /* resize the result matrix to match number of requested rows/columns */
1549c61587bbSBarry Smith       ierr = MatSetSizes(*B,nrows,ncols,nrows,ncols);CHKERRQ(ierr);
155021a2c019SBarry Smith     }
1551182d2002SSatish Balay     newmat = *B;
1552182d2002SSatish Balay   } else {
15530754003eSLois Curfman McInnes     /* Create and fill new matrix */
1554ce94432eSBarry Smith     ierr = MatCreate(PetscObjectComm((PetscObject)A),&newmat);CHKERRQ(ierr);
1555f69a0ea3SMatthew Knepley     ierr = MatSetSizes(newmat,nrows,ncols,nrows,ncols);CHKERRQ(ierr);
15567adad957SLisandro Dalcin     ierr = MatSetType(newmat,((PetscObject)A)->type_name);CHKERRQ(ierr);
15570298fd71SBarry Smith     ierr = MatSeqDenseSetPreallocation(newmat,NULL);CHKERRQ(ierr);
1558182d2002SSatish Balay   }
1559182d2002SSatish Balay 
1560182d2002SSatish Balay   /* Now extract the data pointers and do the copy,column at a time */
1561182d2002SSatish Balay   bv = ((Mat_SeqDense*)newmat->data)->v;
1562182d2002SSatish Balay 
1563182d2002SSatish Balay   for (i=0; i<ncols; i++) {
15646de62eeeSBarry Smith     av = v + mat->lda*icol[i];
15652205254eSKarl Rupp     for (j=0; j<nrows; j++) *bv++ = av[irow[j]];
15660754003eSLois Curfman McInnes   }
1567182d2002SSatish Balay 
1568182d2002SSatish Balay   /* Assemble the matrices so that the correct flags are set */
15696d4a8577SBarry Smith   ierr = MatAssemblyBegin(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
15706d4a8577SBarry Smith   ierr = MatAssemblyEnd(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
15710754003eSLois Curfman McInnes 
15720754003eSLois Curfman McInnes   /* Free work space */
157378b31e54SBarry Smith   ierr = ISRestoreIndices(isrow,&irow);CHKERRQ(ierr);
157478b31e54SBarry Smith   ierr = ISRestoreIndices(iscol,&icol);CHKERRQ(ierr);
1575182d2002SSatish Balay   *B   = newmat;
15763a40ed3dSBarry Smith   PetscFunctionReturn(0);
15770754003eSLois Curfman McInnes }
15780754003eSLois Curfman McInnes 
15794a2ae208SSatish Balay #undef __FUNCT__
15804a2ae208SSatish Balay #define __FUNCT__ "MatGetSubMatrices_SeqDense"
158113f74950SBarry Smith PetscErrorCode MatGetSubMatrices_SeqDense(Mat A,PetscInt n,const IS irow[],const IS icol[],MatReuse scall,Mat *B[])
1582905e6a2fSBarry Smith {
15836849ba73SBarry Smith   PetscErrorCode ierr;
158413f74950SBarry Smith   PetscInt       i;
1585905e6a2fSBarry Smith 
15863a40ed3dSBarry Smith   PetscFunctionBegin;
1587905e6a2fSBarry Smith   if (scall == MAT_INITIAL_MATRIX) {
1588785e854fSJed Brown     ierr = PetscMalloc1((n+1),B);CHKERRQ(ierr);
1589905e6a2fSBarry Smith   }
1590905e6a2fSBarry Smith 
1591905e6a2fSBarry Smith   for (i=0; i<n; i++) {
15926a6a5d1dSBarry Smith     ierr = MatGetSubMatrix_SeqDense(A,irow[i],icol[i],PETSC_DECIDE,scall,&(*B)[i]);CHKERRQ(ierr);
1593905e6a2fSBarry Smith   }
15943a40ed3dSBarry Smith   PetscFunctionReturn(0);
1595905e6a2fSBarry Smith }
1596905e6a2fSBarry Smith 
15974a2ae208SSatish Balay #undef __FUNCT__
1598c0aa2d19SHong Zhang #define __FUNCT__ "MatAssemblyBegin_SeqDense"
1599c0aa2d19SHong Zhang PetscErrorCode MatAssemblyBegin_SeqDense(Mat mat,MatAssemblyType mode)
1600c0aa2d19SHong Zhang {
1601c0aa2d19SHong Zhang   PetscFunctionBegin;
1602c0aa2d19SHong Zhang   PetscFunctionReturn(0);
1603c0aa2d19SHong Zhang }
1604c0aa2d19SHong Zhang 
1605c0aa2d19SHong Zhang #undef __FUNCT__
1606c0aa2d19SHong Zhang #define __FUNCT__ "MatAssemblyEnd_SeqDense"
1607c0aa2d19SHong Zhang PetscErrorCode MatAssemblyEnd_SeqDense(Mat mat,MatAssemblyType mode)
1608c0aa2d19SHong Zhang {
1609c0aa2d19SHong Zhang   PetscFunctionBegin;
1610c0aa2d19SHong Zhang   PetscFunctionReturn(0);
1611c0aa2d19SHong Zhang }
1612c0aa2d19SHong Zhang 
1613c0aa2d19SHong Zhang #undef __FUNCT__
16144a2ae208SSatish Balay #define __FUNCT__ "MatCopy_SeqDense"
1615dfbe8321SBarry Smith PetscErrorCode MatCopy_SeqDense(Mat A,Mat B,MatStructure str)
16164b0e389bSBarry Smith {
16174b0e389bSBarry Smith   Mat_SeqDense   *a = (Mat_SeqDense*)A->data,*b = (Mat_SeqDense*)B->data;
16186849ba73SBarry Smith   PetscErrorCode ierr;
1619d0f46423SBarry Smith   PetscInt       lda1=a->lda,lda2=b->lda, m=A->rmap->n,n=A->cmap->n, j;
16203a40ed3dSBarry Smith 
16213a40ed3dSBarry Smith   PetscFunctionBegin;
162233f4a19fSKris Buschelman   /* If the two matrices don't have the same copy implementation, they aren't compatible for fast copy. */
162333f4a19fSKris Buschelman   if (A->ops->copy != B->ops->copy) {
1624cb5b572fSBarry Smith     ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr);
16253a40ed3dSBarry Smith     PetscFunctionReturn(0);
16263a40ed3dSBarry Smith   }
1627e32f2f54SBarry Smith   if (m != B->rmap->n || n != B->cmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"size(B) != size(A)");
1628a5ce6ee0Svictorle   if (lda1>m || lda2>m) {
16290dbb7854Svictorle     for (j=0; j<n; j++) {
1630a5ce6ee0Svictorle       ierr = PetscMemcpy(b->v+j*lda2,a->v+j*lda1,m*sizeof(PetscScalar));CHKERRQ(ierr);
1631a5ce6ee0Svictorle     }
1632a5ce6ee0Svictorle   } else {
1633d0f46423SBarry Smith     ierr = PetscMemcpy(b->v,a->v,A->rmap->n*A->cmap->n*sizeof(PetscScalar));CHKERRQ(ierr);
1634a5ce6ee0Svictorle   }
1635273d9f13SBarry Smith   PetscFunctionReturn(0);
1636273d9f13SBarry Smith }
1637273d9f13SBarry Smith 
16384a2ae208SSatish Balay #undef __FUNCT__
16394994cf47SJed Brown #define __FUNCT__ "MatSetUp_SeqDense"
16404994cf47SJed Brown PetscErrorCode MatSetUp_SeqDense(Mat A)
1641273d9f13SBarry Smith {
1642dfbe8321SBarry Smith   PetscErrorCode ierr;
1643273d9f13SBarry Smith 
1644273d9f13SBarry Smith   PetscFunctionBegin;
1645273d9f13SBarry Smith   ierr =  MatSeqDenseSetPreallocation(A,0);CHKERRQ(ierr);
16463a40ed3dSBarry Smith   PetscFunctionReturn(0);
16474b0e389bSBarry Smith }
16484b0e389bSBarry Smith 
1649284134d9SBarry Smith #undef __FUNCT__
1650ba337c44SJed Brown #define __FUNCT__ "MatConjugate_SeqDense"
1651ba337c44SJed Brown static PetscErrorCode MatConjugate_SeqDense(Mat A)
1652ba337c44SJed Brown {
1653ba337c44SJed Brown   Mat_SeqDense *a = (Mat_SeqDense*)A->data;
1654ba337c44SJed Brown   PetscInt     i,nz = A->rmap->n*A->cmap->n;
1655ba337c44SJed Brown   PetscScalar  *aa = a->v;
1656ba337c44SJed Brown 
1657ba337c44SJed Brown   PetscFunctionBegin;
1658ba337c44SJed Brown   for (i=0; i<nz; i++) aa[i] = PetscConj(aa[i]);
1659ba337c44SJed Brown   PetscFunctionReturn(0);
1660ba337c44SJed Brown }
1661ba337c44SJed Brown 
1662ba337c44SJed Brown #undef __FUNCT__
1663ba337c44SJed Brown #define __FUNCT__ "MatRealPart_SeqDense"
1664ba337c44SJed Brown static PetscErrorCode MatRealPart_SeqDense(Mat A)
1665ba337c44SJed Brown {
1666ba337c44SJed Brown   Mat_SeqDense *a = (Mat_SeqDense*)A->data;
1667ba337c44SJed Brown   PetscInt     i,nz = A->rmap->n*A->cmap->n;
1668ba337c44SJed Brown   PetscScalar  *aa = a->v;
1669ba337c44SJed Brown 
1670ba337c44SJed Brown   PetscFunctionBegin;
1671ba337c44SJed Brown   for (i=0; i<nz; i++) aa[i] = PetscRealPart(aa[i]);
1672ba337c44SJed Brown   PetscFunctionReturn(0);
1673ba337c44SJed Brown }
1674ba337c44SJed Brown 
1675ba337c44SJed Brown #undef __FUNCT__
1676ba337c44SJed Brown #define __FUNCT__ "MatImaginaryPart_SeqDense"
1677ba337c44SJed Brown static PetscErrorCode MatImaginaryPart_SeqDense(Mat A)
1678ba337c44SJed Brown {
1679ba337c44SJed Brown   Mat_SeqDense *a = (Mat_SeqDense*)A->data;
1680ba337c44SJed Brown   PetscInt     i,nz = A->rmap->n*A->cmap->n;
1681ba337c44SJed Brown   PetscScalar  *aa = a->v;
1682ba337c44SJed Brown 
1683ba337c44SJed Brown   PetscFunctionBegin;
1684ba337c44SJed Brown   for (i=0; i<nz; i++) aa[i] = PetscImaginaryPart(aa[i]);
1685ba337c44SJed Brown   PetscFunctionReturn(0);
1686ba337c44SJed Brown }
1687284134d9SBarry Smith 
1688a9fe9ddaSSatish Balay /* ----------------------------------------------------------------*/
1689a9fe9ddaSSatish Balay #undef __FUNCT__
1690a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMult_SeqDense_SeqDense"
1691a9fe9ddaSSatish Balay PetscErrorCode MatMatMult_SeqDense_SeqDense(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C)
1692a9fe9ddaSSatish Balay {
1693a9fe9ddaSSatish Balay   PetscErrorCode ierr;
1694a9fe9ddaSSatish Balay 
1695a9fe9ddaSSatish Balay   PetscFunctionBegin;
1696a9fe9ddaSSatish Balay   if (scall == MAT_INITIAL_MATRIX) {
16973ff4c91cSHong Zhang     ierr = PetscLogEventBegin(MAT_MatMultSymbolic,A,B,0,0);CHKERRQ(ierr);
1698a9fe9ddaSSatish Balay     ierr = MatMatMultSymbolic_SeqDense_SeqDense(A,B,fill,C);CHKERRQ(ierr);
16993ff4c91cSHong Zhang     ierr = PetscLogEventEnd(MAT_MatMultSymbolic,A,B,0,0);CHKERRQ(ierr);
1700a9fe9ddaSSatish Balay   }
17013ff4c91cSHong Zhang   ierr = PetscLogEventBegin(MAT_MatMultNumeric,A,B,0,0);CHKERRQ(ierr);
1702a9fe9ddaSSatish Balay   ierr = MatMatMultNumeric_SeqDense_SeqDense(A,B,*C);CHKERRQ(ierr);
17033ff4c91cSHong Zhang   ierr = PetscLogEventEnd(MAT_MatMultNumeric,A,B,0,0);CHKERRQ(ierr);
1704a9fe9ddaSSatish Balay   PetscFunctionReturn(0);
1705a9fe9ddaSSatish Balay }
1706a9fe9ddaSSatish Balay 
1707a9fe9ddaSSatish Balay #undef __FUNCT__
1708a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMultSymbolic_SeqDense_SeqDense"
1709a9fe9ddaSSatish Balay PetscErrorCode MatMatMultSymbolic_SeqDense_SeqDense(Mat A,Mat B,PetscReal fill,Mat *C)
1710a9fe9ddaSSatish Balay {
1711ee16a9a1SHong Zhang   PetscErrorCode ierr;
1712d0f46423SBarry Smith   PetscInt       m=A->rmap->n,n=B->cmap->n;
1713ee16a9a1SHong Zhang   Mat            Cmat;
1714a9fe9ddaSSatish Balay 
1715ee16a9a1SHong Zhang   PetscFunctionBegin;
1716e32f2f54SBarry Smith   if (A->cmap->n != B->rmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"A->cmap->n %d != B->rmap->n %d\n",A->cmap->n,B->rmap->n);
1717ee16a9a1SHong Zhang   ierr = MatCreate(PETSC_COMM_SELF,&Cmat);CHKERRQ(ierr);
1718ee16a9a1SHong Zhang   ierr = MatSetSizes(Cmat,m,n,m,n);CHKERRQ(ierr);
1719ee16a9a1SHong Zhang   ierr = MatSetType(Cmat,MATSEQDENSE);CHKERRQ(ierr);
17200298fd71SBarry Smith   ierr = MatSeqDenseSetPreallocation(Cmat,NULL);CHKERRQ(ierr);
1721d73949e8SHong Zhang 
1722ee16a9a1SHong Zhang   *C = Cmat;
1723ee16a9a1SHong Zhang   PetscFunctionReturn(0);
1724ee16a9a1SHong Zhang }
1725a9fe9ddaSSatish Balay 
172698a3b096SSatish Balay #undef __FUNCT__
1727a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMultNumeric_SeqDense_SeqDense"
1728a9fe9ddaSSatish Balay PetscErrorCode MatMatMultNumeric_SeqDense_SeqDense(Mat A,Mat B,Mat C)
1729a9fe9ddaSSatish Balay {
1730a9fe9ddaSSatish Balay   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
1731a9fe9ddaSSatish Balay   Mat_SeqDense   *b = (Mat_SeqDense*)B->data;
1732a9fe9ddaSSatish Balay   Mat_SeqDense   *c = (Mat_SeqDense*)C->data;
17330805154bSBarry Smith   PetscBLASInt   m,n,k;
1734a9fe9ddaSSatish Balay   PetscScalar    _DOne=1.0,_DZero=0.0;
1735c5df96a5SBarry Smith   PetscErrorCode ierr;
1736a9fe9ddaSSatish Balay 
1737a9fe9ddaSSatish Balay   PetscFunctionBegin;
1738c5df96a5SBarry Smith   ierr = PetscBLASIntCast(A->rmap->n,&m);CHKERRQ(ierr);
1739c5df96a5SBarry Smith   ierr = PetscBLASIntCast(B->cmap->n,&n);CHKERRQ(ierr);
1740c5df96a5SBarry Smith   ierr = PetscBLASIntCast(A->cmap->n,&k);CHKERRQ(ierr);
17418b83055fSJed Brown   PetscStackCallBLAS("BLASgemv",BLASgemm_("N","N",&m,&n,&k,&_DOne,a->v,&a->lda,b->v,&b->lda,&_DZero,c->v,&c->lda));
1742a9fe9ddaSSatish Balay   PetscFunctionReturn(0);
1743a9fe9ddaSSatish Balay }
1744a9fe9ddaSSatish Balay 
1745a9fe9ddaSSatish Balay #undef __FUNCT__
174675648e8dSHong Zhang #define __FUNCT__ "MatTransposeMatMult_SeqDense_SeqDense"
174775648e8dSHong Zhang PetscErrorCode MatTransposeMatMult_SeqDense_SeqDense(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C)
1748a9fe9ddaSSatish Balay {
1749a9fe9ddaSSatish Balay   PetscErrorCode ierr;
1750a9fe9ddaSSatish Balay 
1751a9fe9ddaSSatish Balay   PetscFunctionBegin;
1752a9fe9ddaSSatish Balay   if (scall == MAT_INITIAL_MATRIX) {
17533ff4c91cSHong Zhang     ierr = PetscLogEventBegin(MAT_TransposeMatMultSymbolic,A,B,0,0);CHKERRQ(ierr);
175475648e8dSHong Zhang     ierr = MatTransposeMatMultSymbolic_SeqDense_SeqDense(A,B,fill,C);CHKERRQ(ierr);
17553ff4c91cSHong Zhang     ierr = PetscLogEventEnd(MAT_TransposeMatMultSymbolic,A,B,0,0);CHKERRQ(ierr);
1756a9fe9ddaSSatish Balay   }
17573ff4c91cSHong Zhang   ierr = PetscLogEventBegin(MAT_TransposeMatMultNumeric,A,B,0,0);CHKERRQ(ierr);
175875648e8dSHong Zhang   ierr = MatTransposeMatMultNumeric_SeqDense_SeqDense(A,B,*C);CHKERRQ(ierr);
17593ff4c91cSHong Zhang   ierr = PetscLogEventEnd(MAT_TransposeMatMultNumeric,A,B,0,0);CHKERRQ(ierr);
1760a9fe9ddaSSatish Balay   PetscFunctionReturn(0);
1761a9fe9ddaSSatish Balay }
1762a9fe9ddaSSatish Balay 
1763a9fe9ddaSSatish Balay #undef __FUNCT__
176475648e8dSHong Zhang #define __FUNCT__ "MatTransposeMatMultSymbolic_SeqDense_SeqDense"
176575648e8dSHong Zhang PetscErrorCode MatTransposeMatMultSymbolic_SeqDense_SeqDense(Mat A,Mat B,PetscReal fill,Mat *C)
1766a9fe9ddaSSatish Balay {
1767ee16a9a1SHong Zhang   PetscErrorCode ierr;
1768d0f46423SBarry Smith   PetscInt       m=A->cmap->n,n=B->cmap->n;
1769ee16a9a1SHong Zhang   Mat            Cmat;
1770a9fe9ddaSSatish Balay 
1771ee16a9a1SHong Zhang   PetscFunctionBegin;
1772e32f2f54SBarry Smith   if (A->rmap->n != B->rmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"A->rmap->n %d != B->rmap->n %d\n",A->rmap->n,B->rmap->n);
1773ee16a9a1SHong Zhang   ierr = MatCreate(PETSC_COMM_SELF,&Cmat);CHKERRQ(ierr);
1774ee16a9a1SHong Zhang   ierr = MatSetSizes(Cmat,m,n,m,n);CHKERRQ(ierr);
1775ee16a9a1SHong Zhang   ierr = MatSetType(Cmat,MATSEQDENSE);CHKERRQ(ierr);
17760298fd71SBarry Smith   ierr = MatSeqDenseSetPreallocation(Cmat,NULL);CHKERRQ(ierr);
17772205254eSKarl Rupp 
1778ee16a9a1SHong Zhang   Cmat->assembled = PETSC_TRUE;
17792205254eSKarl Rupp 
1780ee16a9a1SHong Zhang   *C = Cmat;
1781ee16a9a1SHong Zhang   PetscFunctionReturn(0);
1782ee16a9a1SHong Zhang }
1783a9fe9ddaSSatish Balay 
1784a9fe9ddaSSatish Balay #undef __FUNCT__
178575648e8dSHong Zhang #define __FUNCT__ "MatTransposeMatMultNumeric_SeqDense_SeqDense"
178675648e8dSHong Zhang PetscErrorCode MatTransposeMatMultNumeric_SeqDense_SeqDense(Mat A,Mat B,Mat C)
1787a9fe9ddaSSatish Balay {
1788a9fe9ddaSSatish Balay   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
1789a9fe9ddaSSatish Balay   Mat_SeqDense   *b = (Mat_SeqDense*)B->data;
1790a9fe9ddaSSatish Balay   Mat_SeqDense   *c = (Mat_SeqDense*)C->data;
17910805154bSBarry Smith   PetscBLASInt   m,n,k;
1792a9fe9ddaSSatish Balay   PetscScalar    _DOne=1.0,_DZero=0.0;
1793c5df96a5SBarry Smith   PetscErrorCode ierr;
1794a9fe9ddaSSatish Balay 
1795a9fe9ddaSSatish Balay   PetscFunctionBegin;
1796c5df96a5SBarry Smith   ierr = PetscBLASIntCast(A->cmap->n,&m);CHKERRQ(ierr);
1797c5df96a5SBarry Smith   ierr = PetscBLASIntCast(B->cmap->n,&n);CHKERRQ(ierr);
1798c5df96a5SBarry Smith   ierr = PetscBLASIntCast(A->rmap->n,&k);CHKERRQ(ierr);
17992fbe02b9SBarry Smith   /*
18002fbe02b9SBarry Smith      Note the m and n arguments below are the number rows and columns of A', not A!
18012fbe02b9SBarry Smith   */
18028b83055fSJed Brown   PetscStackCallBLAS("BLASgemv",BLASgemm_("T","N",&m,&n,&k,&_DOne,a->v,&a->lda,b->v,&b->lda,&_DZero,c->v,&c->lda));
1803a9fe9ddaSSatish Balay   PetscFunctionReturn(0);
1804a9fe9ddaSSatish Balay }
1805985db425SBarry Smith 
1806985db425SBarry Smith #undef __FUNCT__
1807985db425SBarry Smith #define __FUNCT__ "MatGetRowMax_SeqDense"
1808985db425SBarry Smith PetscErrorCode MatGetRowMax_SeqDense(Mat A,Vec v,PetscInt idx[])
1809985db425SBarry Smith {
1810985db425SBarry Smith   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
1811985db425SBarry Smith   PetscErrorCode ierr;
1812d0f46423SBarry Smith   PetscInt       i,j,m = A->rmap->n,n = A->cmap->n,p;
1813985db425SBarry Smith   PetscScalar    *x;
1814985db425SBarry Smith   MatScalar      *aa = a->v;
1815985db425SBarry Smith 
1816985db425SBarry Smith   PetscFunctionBegin;
1817e32f2f54SBarry Smith   if (A->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
1818985db425SBarry Smith 
1819985db425SBarry Smith   ierr = VecSet(v,0.0);CHKERRQ(ierr);
1820985db425SBarry Smith   ierr = VecGetArray(v,&x);CHKERRQ(ierr);
1821985db425SBarry Smith   ierr = VecGetLocalSize(v,&p);CHKERRQ(ierr);
1822e32f2f54SBarry Smith   if (p != A->rmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector");
1823985db425SBarry Smith   for (i=0; i<m; i++) {
1824985db425SBarry Smith     x[i] = aa[i]; if (idx) idx[i] = 0;
1825985db425SBarry Smith     for (j=1; j<n; j++) {
1826985db425SBarry Smith       if (PetscRealPart(x[i]) < PetscRealPart(aa[i+m*j])) {x[i] = aa[i + m*j]; if (idx) idx[i] = j;}
1827985db425SBarry Smith     }
1828985db425SBarry Smith   }
1829985db425SBarry Smith   ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
1830985db425SBarry Smith   PetscFunctionReturn(0);
1831985db425SBarry Smith }
1832985db425SBarry Smith 
1833985db425SBarry Smith #undef __FUNCT__
1834985db425SBarry Smith #define __FUNCT__ "MatGetRowMaxAbs_SeqDense"
1835985db425SBarry Smith PetscErrorCode MatGetRowMaxAbs_SeqDense(Mat A,Vec v,PetscInt idx[])
1836985db425SBarry Smith {
1837985db425SBarry Smith   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
1838985db425SBarry Smith   PetscErrorCode ierr;
1839d0f46423SBarry Smith   PetscInt       i,j,m = A->rmap->n,n = A->cmap->n,p;
1840985db425SBarry Smith   PetscScalar    *x;
1841985db425SBarry Smith   PetscReal      atmp;
1842985db425SBarry Smith   MatScalar      *aa = a->v;
1843985db425SBarry Smith 
1844985db425SBarry Smith   PetscFunctionBegin;
1845e32f2f54SBarry Smith   if (A->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
1846985db425SBarry Smith 
1847985db425SBarry Smith   ierr = VecSet(v,0.0);CHKERRQ(ierr);
1848985db425SBarry Smith   ierr = VecGetArray(v,&x);CHKERRQ(ierr);
1849985db425SBarry Smith   ierr = VecGetLocalSize(v,&p);CHKERRQ(ierr);
1850e32f2f54SBarry Smith   if (p != A->rmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector");
1851985db425SBarry Smith   for (i=0; i<m; i++) {
18529189402eSHong Zhang     x[i] = PetscAbsScalar(aa[i]);
1853985db425SBarry Smith     for (j=1; j<n; j++) {
1854985db425SBarry Smith       atmp = PetscAbsScalar(aa[i+m*j]);
1855985db425SBarry Smith       if (PetscAbsScalar(x[i]) < atmp) {x[i] = atmp; if (idx) idx[i] = j;}
1856985db425SBarry Smith     }
1857985db425SBarry Smith   }
1858985db425SBarry Smith   ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
1859985db425SBarry Smith   PetscFunctionReturn(0);
1860985db425SBarry Smith }
1861985db425SBarry Smith 
1862985db425SBarry Smith #undef __FUNCT__
1863985db425SBarry Smith #define __FUNCT__ "MatGetRowMin_SeqDense"
1864985db425SBarry Smith PetscErrorCode MatGetRowMin_SeqDense(Mat A,Vec v,PetscInt idx[])
1865985db425SBarry Smith {
1866985db425SBarry Smith   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
1867985db425SBarry Smith   PetscErrorCode ierr;
1868d0f46423SBarry Smith   PetscInt       i,j,m = A->rmap->n,n = A->cmap->n,p;
1869985db425SBarry Smith   PetscScalar    *x;
1870985db425SBarry Smith   MatScalar      *aa = a->v;
1871985db425SBarry Smith 
1872985db425SBarry Smith   PetscFunctionBegin;
1873e32f2f54SBarry Smith   if (A->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
1874985db425SBarry Smith 
1875985db425SBarry Smith   ierr = VecSet(v,0.0);CHKERRQ(ierr);
1876985db425SBarry Smith   ierr = VecGetArray(v,&x);CHKERRQ(ierr);
1877985db425SBarry Smith   ierr = VecGetLocalSize(v,&p);CHKERRQ(ierr);
1878e32f2f54SBarry Smith   if (p != A->rmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector");
1879985db425SBarry Smith   for (i=0; i<m; i++) {
1880985db425SBarry Smith     x[i] = aa[i]; if (idx) idx[i] = 0;
1881985db425SBarry Smith     for (j=1; j<n; j++) {
1882985db425SBarry Smith       if (PetscRealPart(x[i]) > PetscRealPart(aa[i+m*j])) {x[i] = aa[i + m*j]; if (idx) idx[i] = j;}
1883985db425SBarry Smith     }
1884985db425SBarry Smith   }
1885985db425SBarry Smith   ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
1886985db425SBarry Smith   PetscFunctionReturn(0);
1887985db425SBarry Smith }
1888985db425SBarry Smith 
18898d0534beSBarry Smith #undef __FUNCT__
18908d0534beSBarry Smith #define __FUNCT__ "MatGetColumnVector_SeqDense"
18918d0534beSBarry Smith PetscErrorCode MatGetColumnVector_SeqDense(Mat A,Vec v,PetscInt col)
18928d0534beSBarry Smith {
18938d0534beSBarry Smith   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
18948d0534beSBarry Smith   PetscErrorCode ierr;
18958d0534beSBarry Smith   PetscScalar    *x;
18968d0534beSBarry Smith 
18978d0534beSBarry Smith   PetscFunctionBegin;
1898e32f2f54SBarry Smith   if (A->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
18998d0534beSBarry Smith 
19008d0534beSBarry Smith   ierr = VecGetArray(v,&x);CHKERRQ(ierr);
1901d0f46423SBarry Smith   ierr = PetscMemcpy(x,a->v+col*a->lda,A->rmap->n*sizeof(PetscScalar));CHKERRQ(ierr);
19028d0534beSBarry Smith   ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
19038d0534beSBarry Smith   PetscFunctionReturn(0);
19048d0534beSBarry Smith }
19058d0534beSBarry Smith 
19060716a85fSBarry Smith 
19070716a85fSBarry Smith #undef __FUNCT__
19080716a85fSBarry Smith #define __FUNCT__ "MatGetColumnNorms_SeqDense"
19090716a85fSBarry Smith PetscErrorCode MatGetColumnNorms_SeqDense(Mat A,NormType type,PetscReal *norms)
19100716a85fSBarry Smith {
19110716a85fSBarry Smith   PetscErrorCode ierr;
19120716a85fSBarry Smith   PetscInt       i,j,m,n;
19130716a85fSBarry Smith   PetscScalar    *a;
19140716a85fSBarry Smith 
19150716a85fSBarry Smith   PetscFunctionBegin;
19160716a85fSBarry Smith   ierr = MatGetSize(A,&m,&n);CHKERRQ(ierr);
19170716a85fSBarry Smith   ierr = PetscMemzero(norms,n*sizeof(PetscReal));CHKERRQ(ierr);
19188c778c55SBarry Smith   ierr = MatDenseGetArray(A,&a);CHKERRQ(ierr);
19190716a85fSBarry Smith   if (type == NORM_2) {
19200716a85fSBarry Smith     for (i=0; i<n; i++) {
19210716a85fSBarry Smith       for (j=0; j<m; j++) {
19220716a85fSBarry Smith         norms[i] += PetscAbsScalar(a[j]*a[j]);
19230716a85fSBarry Smith       }
19240716a85fSBarry Smith       a += m;
19250716a85fSBarry Smith     }
19260716a85fSBarry Smith   } else if (type == NORM_1) {
19270716a85fSBarry Smith     for (i=0; i<n; i++) {
19280716a85fSBarry Smith       for (j=0; j<m; j++) {
19290716a85fSBarry Smith         norms[i] += PetscAbsScalar(a[j]);
19300716a85fSBarry Smith       }
19310716a85fSBarry Smith       a += m;
19320716a85fSBarry Smith     }
19330716a85fSBarry Smith   } else if (type == NORM_INFINITY) {
19340716a85fSBarry Smith     for (i=0; i<n; i++) {
19350716a85fSBarry Smith       for (j=0; j<m; j++) {
19360716a85fSBarry Smith         norms[i] = PetscMax(PetscAbsScalar(a[j]),norms[i]);
19370716a85fSBarry Smith       }
19380716a85fSBarry Smith       a += m;
19390716a85fSBarry Smith     }
1940ce94432eSBarry Smith   } else SETERRQ(PetscObjectComm((PetscObject)A),PETSC_ERR_ARG_WRONG,"Unknown NormType");
19418c778c55SBarry Smith   ierr = MatDenseRestoreArray(A,&a);CHKERRQ(ierr);
19420716a85fSBarry Smith   if (type == NORM_2) {
19438f1a2a5eSBarry Smith     for (i=0; i<n; i++) norms[i] = PetscSqrtReal(norms[i]);
19440716a85fSBarry Smith   }
19450716a85fSBarry Smith   PetscFunctionReturn(0);
19460716a85fSBarry Smith }
19470716a85fSBarry Smith 
194873a71a0fSBarry Smith #undef __FUNCT__
194973a71a0fSBarry Smith #define __FUNCT__ "MatSetRandom_SeqDense"
195073a71a0fSBarry Smith static PetscErrorCode  MatSetRandom_SeqDense(Mat x,PetscRandom rctx)
195173a71a0fSBarry Smith {
195273a71a0fSBarry Smith   PetscErrorCode ierr;
195373a71a0fSBarry Smith   PetscScalar    *a;
195473a71a0fSBarry Smith   PetscInt       m,n,i;
195573a71a0fSBarry Smith 
195673a71a0fSBarry Smith   PetscFunctionBegin;
195773a71a0fSBarry Smith   ierr = MatGetSize(x,&m,&n);CHKERRQ(ierr);
19588c778c55SBarry Smith   ierr = MatDenseGetArray(x,&a);CHKERRQ(ierr);
195973a71a0fSBarry Smith   for (i=0; i<m*n; i++) {
196073a71a0fSBarry Smith     ierr = PetscRandomGetValue(rctx,a+i);CHKERRQ(ierr);
196173a71a0fSBarry Smith   }
19628c778c55SBarry Smith   ierr = MatDenseRestoreArray(x,&a);CHKERRQ(ierr);
196373a71a0fSBarry Smith   PetscFunctionReturn(0);
196473a71a0fSBarry Smith }
196573a71a0fSBarry Smith 
196673a71a0fSBarry Smith 
1967289bc588SBarry Smith /* -------------------------------------------------------------------*/
1968a5ae1ecdSBarry Smith static struct _MatOps MatOps_Values = { MatSetValues_SeqDense,
1969905e6a2fSBarry Smith                                         MatGetRow_SeqDense,
1970905e6a2fSBarry Smith                                         MatRestoreRow_SeqDense,
1971905e6a2fSBarry Smith                                         MatMult_SeqDense,
197297304618SKris Buschelman                                 /*  4*/ MatMultAdd_SeqDense,
19737c922b88SBarry Smith                                         MatMultTranspose_SeqDense,
19747c922b88SBarry Smith                                         MatMultTransposeAdd_SeqDense,
1975db4efbfdSBarry Smith                                         0,
1976db4efbfdSBarry Smith                                         0,
1977db4efbfdSBarry Smith                                         0,
1978db4efbfdSBarry Smith                                 /* 10*/ 0,
1979905e6a2fSBarry Smith                                         MatLUFactor_SeqDense,
1980905e6a2fSBarry Smith                                         MatCholeskyFactor_SeqDense,
198141f059aeSBarry Smith                                         MatSOR_SeqDense,
1982ec8511deSBarry Smith                                         MatTranspose_SeqDense,
198397304618SKris Buschelman                                 /* 15*/ MatGetInfo_SeqDense,
1984905e6a2fSBarry Smith                                         MatEqual_SeqDense,
1985905e6a2fSBarry Smith                                         MatGetDiagonal_SeqDense,
1986905e6a2fSBarry Smith                                         MatDiagonalScale_SeqDense,
1987905e6a2fSBarry Smith                                         MatNorm_SeqDense,
1988c0aa2d19SHong Zhang                                 /* 20*/ MatAssemblyBegin_SeqDense,
1989c0aa2d19SHong Zhang                                         MatAssemblyEnd_SeqDense,
1990905e6a2fSBarry Smith                                         MatSetOption_SeqDense,
1991905e6a2fSBarry Smith                                         MatZeroEntries_SeqDense,
1992d519adbfSMatthew Knepley                                 /* 24*/ MatZeroRows_SeqDense,
1993db4efbfdSBarry Smith                                         0,
1994db4efbfdSBarry Smith                                         0,
1995db4efbfdSBarry Smith                                         0,
1996db4efbfdSBarry Smith                                         0,
19974994cf47SJed Brown                                 /* 29*/ MatSetUp_SeqDense,
1998273d9f13SBarry Smith                                         0,
1999905e6a2fSBarry Smith                                         0,
200073a71a0fSBarry Smith                                         0,
200173a71a0fSBarry Smith                                         0,
2002d519adbfSMatthew Knepley                                 /* 34*/ MatDuplicate_SeqDense,
2003a5ae1ecdSBarry Smith                                         0,
2004a5ae1ecdSBarry Smith                                         0,
2005a5ae1ecdSBarry Smith                                         0,
2006a5ae1ecdSBarry Smith                                         0,
2007d519adbfSMatthew Knepley                                 /* 39*/ MatAXPY_SeqDense,
2008a5ae1ecdSBarry Smith                                         MatGetSubMatrices_SeqDense,
2009a5ae1ecdSBarry Smith                                         0,
20104b0e389bSBarry Smith                                         MatGetValues_SeqDense,
2011a5ae1ecdSBarry Smith                                         MatCopy_SeqDense,
2012d519adbfSMatthew Knepley                                 /* 44*/ MatGetRowMax_SeqDense,
2013a5ae1ecdSBarry Smith                                         MatScale_SeqDense,
2014a5ae1ecdSBarry Smith                                         0,
2015a5ae1ecdSBarry Smith                                         0,
2016a5ae1ecdSBarry Smith                                         0,
201773a71a0fSBarry Smith                                 /* 49*/ MatSetRandom_SeqDense,
2018a5ae1ecdSBarry Smith                                         0,
2019a5ae1ecdSBarry Smith                                         0,
2020a5ae1ecdSBarry Smith                                         0,
2021a5ae1ecdSBarry Smith                                         0,
2022d519adbfSMatthew Knepley                                 /* 54*/ 0,
2023a5ae1ecdSBarry Smith                                         0,
2024a5ae1ecdSBarry Smith                                         0,
2025a5ae1ecdSBarry Smith                                         0,
2026a5ae1ecdSBarry Smith                                         0,
2027d519adbfSMatthew Knepley                                 /* 59*/ 0,
2028e03a110bSBarry Smith                                         MatDestroy_SeqDense,
2029e03a110bSBarry Smith                                         MatView_SeqDense,
2030357abbc8SBarry Smith                                         0,
203197304618SKris Buschelman                                         0,
2032d519adbfSMatthew Knepley                                 /* 64*/ 0,
203397304618SKris Buschelman                                         0,
203497304618SKris Buschelman                                         0,
203597304618SKris Buschelman                                         0,
203697304618SKris Buschelman                                         0,
2037d519adbfSMatthew Knepley                                 /* 69*/ MatGetRowMaxAbs_SeqDense,
203897304618SKris Buschelman                                         0,
203997304618SKris Buschelman                                         0,
204097304618SKris Buschelman                                         0,
204197304618SKris Buschelman                                         0,
2042d519adbfSMatthew Knepley                                 /* 74*/ 0,
204397304618SKris Buschelman                                         0,
204497304618SKris Buschelman                                         0,
204597304618SKris Buschelman                                         0,
204697304618SKris Buschelman                                         0,
2047d519adbfSMatthew Knepley                                 /* 79*/ 0,
204897304618SKris Buschelman                                         0,
204997304618SKris Buschelman                                         0,
205097304618SKris Buschelman                                         0,
20515bba2384SShri Abhyankar                                 /* 83*/ MatLoad_SeqDense,
2052865e5f61SKris Buschelman                                         0,
20531cbb95d3SBarry Smith                                         MatIsHermitian_SeqDense,
2054865e5f61SKris Buschelman                                         0,
2055865e5f61SKris Buschelman                                         0,
2056865e5f61SKris Buschelman                                         0,
2057d519adbfSMatthew Knepley                                 /* 89*/ MatMatMult_SeqDense_SeqDense,
2058a9fe9ddaSSatish Balay                                         MatMatMultSymbolic_SeqDense_SeqDense,
2059a9fe9ddaSSatish Balay                                         MatMatMultNumeric_SeqDense_SeqDense,
2060865e5f61SKris Buschelman                                         0,
2061865e5f61SKris Buschelman                                         0,
2062d519adbfSMatthew Knepley                                 /* 94*/ 0,
20635df89d91SHong Zhang                                         0,
20645df89d91SHong Zhang                                         0,
20655df89d91SHong Zhang                                         0,
2066284134d9SBarry Smith                                         0,
2067d519adbfSMatthew Knepley                                 /* 99*/ 0,
2068284134d9SBarry Smith                                         0,
2069284134d9SBarry Smith                                         0,
2070ba337c44SJed Brown                                         MatConjugate_SeqDense,
2071f73d5cc4SBarry Smith                                         0,
2072ba337c44SJed Brown                                 /*104*/ 0,
2073ba337c44SJed Brown                                         MatRealPart_SeqDense,
2074ba337c44SJed Brown                                         MatImaginaryPart_SeqDense,
2075985db425SBarry Smith                                         0,
2076985db425SBarry Smith                                         0,
207785e2c93fSHong Zhang                                 /*109*/ MatMatSolve_SeqDense,
2078985db425SBarry Smith                                         0,
20798d0534beSBarry Smith                                         MatGetRowMin_SeqDense,
2080aabbc4fbSShri Abhyankar                                         MatGetColumnVector_SeqDense,
2081aabbc4fbSShri Abhyankar                                         0,
2082aabbc4fbSShri Abhyankar                                 /*114*/ 0,
2083aabbc4fbSShri Abhyankar                                         0,
2084aabbc4fbSShri Abhyankar                                         0,
2085aabbc4fbSShri Abhyankar                                         0,
2086aabbc4fbSShri Abhyankar                                         0,
2087aabbc4fbSShri Abhyankar                                 /*119*/ 0,
2088aabbc4fbSShri Abhyankar                                         0,
2089aabbc4fbSShri Abhyankar                                         0,
20900716a85fSBarry Smith                                         0,
20910716a85fSBarry Smith                                         0,
20920716a85fSBarry Smith                                 /*124*/ 0,
20935df89d91SHong Zhang                                         MatGetColumnNorms_SeqDense,
20945df89d91SHong Zhang                                         0,
20955df89d91SHong Zhang                                         0,
20965df89d91SHong Zhang                                         0,
20975df89d91SHong Zhang                                 /*129*/ 0,
209875648e8dSHong Zhang                                         MatTransposeMatMult_SeqDense_SeqDense,
209975648e8dSHong Zhang                                         MatTransposeMatMultSymbolic_SeqDense_SeqDense,
210075648e8dSHong Zhang                                         MatTransposeMatMultNumeric_SeqDense_SeqDense,
21013964eb88SJed Brown                                         0,
21023964eb88SJed Brown                                 /*134*/ 0,
21033964eb88SJed Brown                                         0,
21043964eb88SJed Brown                                         0,
21053964eb88SJed Brown                                         0,
21063964eb88SJed Brown                                         0,
21073964eb88SJed Brown                                 /*139*/ 0,
2108f9426fe0SMark Adams                                         0,
21093964eb88SJed Brown                                         0
2110985db425SBarry Smith };
211190ace30eSBarry Smith 
21124a2ae208SSatish Balay #undef __FUNCT__
21134a2ae208SSatish Balay #define __FUNCT__ "MatCreateSeqDense"
21144b828684SBarry Smith /*@C
2115fafbff53SBarry Smith    MatCreateSeqDense - Creates a sequential dense matrix that
2116d65003e9SLois Curfman McInnes    is stored in column major order (the usual Fortran 77 manner). Many
2117d65003e9SLois Curfman McInnes    of the matrix operations use the BLAS and LAPACK routines.
2118289bc588SBarry Smith 
2119db81eaa0SLois Curfman McInnes    Collective on MPI_Comm
2120db81eaa0SLois Curfman McInnes 
212120563c6bSBarry Smith    Input Parameters:
2122db81eaa0SLois Curfman McInnes +  comm - MPI communicator, set to PETSC_COMM_SELF
21230c775827SLois Curfman McInnes .  m - number of rows
212418f449edSLois Curfman McInnes .  n - number of columns
21250298fd71SBarry Smith -  data - optional location of matrix data in column major order.  Set data=NULL for PETSc
2126dfc5480cSLois Curfman McInnes    to control all matrix memory allocation.
212720563c6bSBarry Smith 
212820563c6bSBarry Smith    Output Parameter:
212944cd7ae7SLois Curfman McInnes .  A - the matrix
213020563c6bSBarry Smith 
2131b259b22eSLois Curfman McInnes    Notes:
213218f449edSLois Curfman McInnes    The data input variable is intended primarily for Fortran programmers
213318f449edSLois Curfman McInnes    who wish to allocate their own matrix memory space.  Most users should
21340298fd71SBarry Smith    set data=NULL.
213518f449edSLois Curfman McInnes 
2136027ccd11SLois Curfman McInnes    Level: intermediate
2137027ccd11SLois Curfman McInnes 
2138dbd7a890SLois Curfman McInnes .keywords: dense, matrix, LAPACK, BLAS
2139d65003e9SLois Curfman McInnes 
214069b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateDense(), MatSetValues()
214120563c6bSBarry Smith @*/
21427087cfbeSBarry Smith PetscErrorCode  MatCreateSeqDense(MPI_Comm comm,PetscInt m,PetscInt n,PetscScalar *data,Mat *A)
2143289bc588SBarry Smith {
2144dfbe8321SBarry Smith   PetscErrorCode ierr;
21453b2fbd54SBarry Smith 
21463a40ed3dSBarry Smith   PetscFunctionBegin;
2147f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,A);CHKERRQ(ierr);
2148f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*A,m,n,m,n);CHKERRQ(ierr);
2149273d9f13SBarry Smith   ierr = MatSetType(*A,MATSEQDENSE);CHKERRQ(ierr);
2150273d9f13SBarry Smith   ierr = MatSeqDenseSetPreallocation(*A,data);CHKERRQ(ierr);
2151273d9f13SBarry Smith   PetscFunctionReturn(0);
2152273d9f13SBarry Smith }
2153273d9f13SBarry Smith 
21544a2ae208SSatish Balay #undef __FUNCT__
2155afc30d2aSLisandro Dalcin #define __FUNCT__ "MatSeqDenseSetPreallocation"
2156273d9f13SBarry Smith /*@C
2157273d9f13SBarry Smith    MatSeqDenseSetPreallocation - Sets the array used for storing the matrix elements
2158273d9f13SBarry Smith 
2159273d9f13SBarry Smith    Collective on MPI_Comm
2160273d9f13SBarry Smith 
2161273d9f13SBarry Smith    Input Parameters:
2162273d9f13SBarry Smith +  A - the matrix
21630298fd71SBarry Smith -  data - the array (or NULL)
2164273d9f13SBarry Smith 
2165273d9f13SBarry Smith    Notes:
2166273d9f13SBarry Smith    The data input variable is intended primarily for Fortran programmers
2167273d9f13SBarry Smith    who wish to allocate their own matrix memory space.  Most users should
2168284134d9SBarry Smith    need not call this routine.
2169273d9f13SBarry Smith 
2170273d9f13SBarry Smith    Level: intermediate
2171273d9f13SBarry Smith 
2172273d9f13SBarry Smith .keywords: dense, matrix, LAPACK, BLAS
2173273d9f13SBarry Smith 
217469b1f4b7SBarry Smith .seealso: MatCreate(), MatCreateDense(), MatSetValues(), MatSeqDenseSetLDA()
2175867c911aSBarry Smith 
2176273d9f13SBarry Smith @*/
21777087cfbeSBarry Smith PetscErrorCode  MatSeqDenseSetPreallocation(Mat B,PetscScalar data[])
2178273d9f13SBarry Smith {
21794ac538c5SBarry Smith   PetscErrorCode ierr;
2180a23d5eceSKris Buschelman 
2181a23d5eceSKris Buschelman   PetscFunctionBegin;
21824ac538c5SBarry Smith   ierr = PetscTryMethod(B,"MatSeqDenseSetPreallocation_C",(Mat,PetscScalar[]),(B,data));CHKERRQ(ierr);
2183a23d5eceSKris Buschelman   PetscFunctionReturn(0);
2184a23d5eceSKris Buschelman }
2185a23d5eceSKris Buschelman 
2186a23d5eceSKris Buschelman #undef __FUNCT__
2187afc30d2aSLisandro Dalcin #define __FUNCT__ "MatSeqDenseSetPreallocation_SeqDense"
21887087cfbeSBarry Smith PetscErrorCode  MatSeqDenseSetPreallocation_SeqDense(Mat B,PetscScalar *data)
2189a23d5eceSKris Buschelman {
2190273d9f13SBarry Smith   Mat_SeqDense   *b;
2191dfbe8321SBarry Smith   PetscErrorCode ierr;
2192273d9f13SBarry Smith 
2193273d9f13SBarry Smith   PetscFunctionBegin;
2194273d9f13SBarry Smith   B->preallocated = PETSC_TRUE;
2195a868139aSShri Abhyankar 
219634ef9618SShri Abhyankar   ierr = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr);
219734ef9618SShri Abhyankar   ierr = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr);
219834ef9618SShri Abhyankar 
2199273d9f13SBarry Smith   b       = (Mat_SeqDense*)B->data;
220086d161a7SShri Abhyankar   b->Mmax = B->rmap->n;
220186d161a7SShri Abhyankar   b->Nmax = B->cmap->n;
220286d161a7SShri Abhyankar   if (b->lda <= 0 || b->changelda) b->lda = B->rmap->n;
220386d161a7SShri Abhyankar 
22049e8f95c4SLisandro Dalcin   if (!data) { /* petsc-allocated storage */
22059e8f95c4SLisandro Dalcin     if (!b->user_alloc) { ierr = PetscFree(b->v);CHKERRQ(ierr); }
2206e92229d0SSatish Balay     ierr = PetscCalloc1((size_t)b->lda*b->Nmax,&b->v);CHKERRQ(ierr);
22073bb1ff40SBarry Smith     ierr = PetscLogObjectMemory((PetscObject)B,b->lda*b->Nmax*sizeof(PetscScalar));CHKERRQ(ierr);
22082205254eSKarl Rupp 
22099e8f95c4SLisandro Dalcin     b->user_alloc = PETSC_FALSE;
2210273d9f13SBarry Smith   } else { /* user-allocated storage */
22119e8f95c4SLisandro Dalcin     if (!b->user_alloc) { ierr = PetscFree(b->v);CHKERRQ(ierr); }
2212273d9f13SBarry Smith     b->v          = data;
2213273d9f13SBarry Smith     b->user_alloc = PETSC_TRUE;
2214273d9f13SBarry Smith   }
22150450473dSBarry Smith   B->assembled = PETSC_TRUE;
2216273d9f13SBarry Smith   PetscFunctionReturn(0);
2217273d9f13SBarry Smith }
2218273d9f13SBarry Smith 
22191b807ce4Svictorle #undef __FUNCT__
22201b807ce4Svictorle #define __FUNCT__ "MatSeqDenseSetLDA"
22211b807ce4Svictorle /*@C
22221b807ce4Svictorle   MatSeqDenseSetLDA - Declare the leading dimension of the user-provided array
22231b807ce4Svictorle 
22241b807ce4Svictorle   Input parameter:
22251b807ce4Svictorle + A - the matrix
22261b807ce4Svictorle - lda - the leading dimension
22271b807ce4Svictorle 
22281b807ce4Svictorle   Notes:
2229867c911aSBarry Smith   This routine is to be used in conjunction with MatSeqDenseSetPreallocation();
22301b807ce4Svictorle   it asserts that the preallocation has a leading dimension (the LDA parameter
22311b807ce4Svictorle   of Blas and Lapack fame) larger than M, the first dimension of the matrix.
22321b807ce4Svictorle 
22331b807ce4Svictorle   Level: intermediate
22341b807ce4Svictorle 
22351b807ce4Svictorle .keywords: dense, matrix, LAPACK, BLAS
22361b807ce4Svictorle 
2237284134d9SBarry Smith .seealso: MatCreate(), MatCreateSeqDense(), MatSeqDenseSetPreallocation(), MatSetMaximumSize()
2238867c911aSBarry Smith 
22391b807ce4Svictorle @*/
22407087cfbeSBarry Smith PetscErrorCode  MatSeqDenseSetLDA(Mat B,PetscInt lda)
22411b807ce4Svictorle {
22421b807ce4Svictorle   Mat_SeqDense *b = (Mat_SeqDense*)B->data;
224321a2c019SBarry Smith 
22441b807ce4Svictorle   PetscFunctionBegin;
2245e32f2f54SBarry Smith   if (lda < B->rmap->n) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"LDA %D must be at least matrix dimension %D",lda,B->rmap->n);
22461b807ce4Svictorle   b->lda       = lda;
224721a2c019SBarry Smith   b->changelda = PETSC_FALSE;
224821a2c019SBarry Smith   b->Mmax      = PetscMax(b->Mmax,lda);
22491b807ce4Svictorle   PetscFunctionReturn(0);
22501b807ce4Svictorle }
22511b807ce4Svictorle 
22520bad9183SKris Buschelman /*MC
2253fafad747SKris Buschelman    MATSEQDENSE - MATSEQDENSE = "seqdense" - A matrix type to be used for sequential dense matrices.
22540bad9183SKris Buschelman 
22550bad9183SKris Buschelman    Options Database Keys:
22560bad9183SKris Buschelman . -mat_type seqdense - sets the matrix type to "seqdense" during a call to MatSetFromOptions()
22570bad9183SKris Buschelman 
22580bad9183SKris Buschelman   Level: beginner
22590bad9183SKris Buschelman 
226089665df3SBarry Smith .seealso: MatCreateSeqDense()
226189665df3SBarry Smith 
22620bad9183SKris Buschelman M*/
22630bad9183SKris Buschelman 
22644a2ae208SSatish Balay #undef __FUNCT__
22654a2ae208SSatish Balay #define __FUNCT__ "MatCreate_SeqDense"
22668cc058d9SJed Brown PETSC_EXTERN PetscErrorCode MatCreate_SeqDense(Mat B)
2267273d9f13SBarry Smith {
2268273d9f13SBarry Smith   Mat_SeqDense   *b;
2269dfbe8321SBarry Smith   PetscErrorCode ierr;
22707c334f02SBarry Smith   PetscMPIInt    size;
2271273d9f13SBarry Smith 
2272273d9f13SBarry Smith   PetscFunctionBegin;
2273ce94432eSBarry Smith   ierr = MPI_Comm_size(PetscObjectComm((PetscObject)B),&size);CHKERRQ(ierr);
2274e32f2f54SBarry Smith   if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Comm must be of size 1");
227555659b69SBarry Smith 
2276b00a9115SJed Brown   ierr    = PetscNewLog(B,&b);CHKERRQ(ierr);
2277549d3d68SSatish Balay   ierr    = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr);
227844cd7ae7SLois Curfman McInnes   B->data = (void*)b;
227918f449edSLois Curfman McInnes 
228044cd7ae7SLois Curfman McInnes   b->pivots      = 0;
2281273d9f13SBarry Smith   b->roworiented = PETSC_TRUE;
2282273d9f13SBarry Smith   b->v           = 0;
228321a2c019SBarry Smith   b->changelda   = PETSC_FALSE;
22844e220ebcSLois Curfman McInnes 
2285bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatDenseGetArray_C",MatDenseGetArray_SeqDense);CHKERRQ(ierr);
2286bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatDenseRestoreArray_C",MatDenseRestoreArray_SeqDense);CHKERRQ(ierr);
2287b24902e0SBarry Smith 
2288bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatConvert_seqdense_seqaij_C",MatConvert_SeqDense_SeqAIJ);CHKERRQ(ierr);
2289bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatGetFactor_petsc_C",MatGetFactor_seqdense_petsc);CHKERRQ(ierr);
2290bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatSeqDenseSetPreallocation_C",MatSeqDenseSetPreallocation_SeqDense);CHKERRQ(ierr);
2291bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatMatMult_seqaij_seqdense_C",MatMatMult_SeqAIJ_SeqDense);CHKERRQ(ierr);
2292bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatMatMultSymbolic_seqaij_seqdense_C",MatMatMultSymbolic_SeqAIJ_SeqDense);CHKERRQ(ierr);
2293bdf89e91SBarry Smith   ierr = PetscObjectComposeFunction((PetscObject)B,"MatMatMultNumeric_seqaij_seqdense_C",MatMatMultNumeric_SeqAIJ_SeqDense);CHKERRQ(ierr);
22943bf78175SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)B,"MatTransposeMatMult_seqaij_seqdense_C",MatTransposeMatMult_SeqAIJ_SeqDense);CHKERRQ(ierr);
22953bf78175SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)B,"MatTransposeMatMultSymbolic_seqaij_seqdense_C",MatTransposeMatMultSymbolic_SeqAIJ_SeqDense);CHKERRQ(ierr);
22963bf78175SHong Zhang   ierr = PetscObjectComposeFunction((PetscObject)B,"MatTransposeMatMultNumeric_seqaij_seqdense_C",MatTransposeMatMultNumeric_SeqAIJ_SeqDense);CHKERRQ(ierr);
229717667f90SBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)B,MATSEQDENSE);CHKERRQ(ierr);
22983a40ed3dSBarry Smith   PetscFunctionReturn(0);
2299289bc588SBarry Smith }
2300