xref: /petsc/src/mat/impls/dense/seq/dense.c (revision aabbc4fba255ddf684b806b03041ff741c689cc8)
1be1d678aSKris Buschelman #define PETSCMAT_DLL
2be1d678aSKris Buschelman 
367e560aaSBarry Smith /*
467e560aaSBarry Smith      Defines the basic matrix operations for sequential dense.
567e560aaSBarry Smith */
6289bc588SBarry Smith 
77c4f633dSBarry Smith #include "../src/mat/impls/dense/seq/dense.h"
8f3da1532SBarry Smith #include "petscblaslapack.h"
9289bc588SBarry Smith 
104a2ae208SSatish Balay #undef __FUNCT__
114a2ae208SSatish Balay #define __FUNCT__ "MatAXPY_SeqDense"
12f4df32b1SMatthew Knepley PetscErrorCode MatAXPY_SeqDense(Mat Y,PetscScalar alpha,Mat X,MatStructure str)
131987afe7SBarry Smith {
141987afe7SBarry Smith   Mat_SeqDense   *x = (Mat_SeqDense*)X->data,*y = (Mat_SeqDense*)Y->data;
15f4df32b1SMatthew Knepley   PetscScalar    oalpha = alpha;
1613f74950SBarry Smith   PetscInt       j;
170805154bSBarry Smith   PetscBLASInt   N,m,ldax,lday,one = 1;
18efee365bSSatish Balay   PetscErrorCode ierr;
193a40ed3dSBarry Smith 
203a40ed3dSBarry Smith   PetscFunctionBegin;
21d0f46423SBarry Smith   N    = PetscBLASIntCast(X->rmap->n*X->cmap->n);
22d0f46423SBarry Smith   m    = PetscBLASIntCast(X->rmap->n);
230805154bSBarry Smith   ldax = PetscBLASIntCast(x->lda);
240805154bSBarry Smith   lday = PetscBLASIntCast(y->lda);
25a5ce6ee0Svictorle   if (ldax>m || lday>m) {
26d0f46423SBarry Smith     for (j=0; j<X->cmap->n; j++) {
27f4df32b1SMatthew Knepley       BLASaxpy_(&m,&oalpha,x->v+j*ldax,&one,y->v+j*lday,&one);
28a5ce6ee0Svictorle     }
29a5ce6ee0Svictorle   } else {
30f4df32b1SMatthew Knepley     BLASaxpy_(&N,&oalpha,x->v,&one,y->v,&one);
31a5ce6ee0Svictorle   }
320450473dSBarry Smith   ierr = PetscLogFlops(PetscMax(2*N-1,0));CHKERRQ(ierr);
333a40ed3dSBarry Smith   PetscFunctionReturn(0);
341987afe7SBarry Smith }
351987afe7SBarry Smith 
364a2ae208SSatish Balay #undef __FUNCT__
374a2ae208SSatish Balay #define __FUNCT__ "MatGetInfo_SeqDense"
38dfbe8321SBarry Smith PetscErrorCode MatGetInfo_SeqDense(Mat A,MatInfoType flag,MatInfo *info)
39289bc588SBarry Smith {
40d0f46423SBarry Smith   PetscInt     N = A->rmap->n*A->cmap->n;
413a40ed3dSBarry Smith 
423a40ed3dSBarry Smith   PetscFunctionBegin;
434e220ebcSLois Curfman McInnes   info->block_size        = 1.0;
444e220ebcSLois Curfman McInnes   info->nz_allocated      = (double)N;
456de62eeeSBarry Smith   info->nz_used           = (double)N;
466de62eeeSBarry Smith   info->nz_unneeded       = (double)0;
474e220ebcSLois Curfman McInnes   info->assemblies        = (double)A->num_ass;
484e220ebcSLois Curfman McInnes   info->mallocs           = 0;
497adad957SLisandro Dalcin   info->memory            = ((PetscObject)A)->mem;
504e220ebcSLois Curfman McInnes   info->fill_ratio_given  = 0;
514e220ebcSLois Curfman McInnes   info->fill_ratio_needed = 0;
524e220ebcSLois Curfman McInnes   info->factor_mallocs    = 0;
533a40ed3dSBarry Smith   PetscFunctionReturn(0);
54289bc588SBarry Smith }
55289bc588SBarry Smith 
564a2ae208SSatish Balay #undef __FUNCT__
574a2ae208SSatish Balay #define __FUNCT__ "MatScale_SeqDense"
58f4df32b1SMatthew Knepley PetscErrorCode MatScale_SeqDense(Mat A,PetscScalar alpha)
5980cd9d93SLois Curfman McInnes {
60273d9f13SBarry Smith   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
61f4df32b1SMatthew Knepley   PetscScalar    oalpha = alpha;
62efee365bSSatish Balay   PetscErrorCode ierr;
630805154bSBarry Smith   PetscBLASInt   one = 1,j,nz,lda = PetscBLASIntCast(a->lda);
6480cd9d93SLois Curfman McInnes 
653a40ed3dSBarry Smith   PetscFunctionBegin;
66d0f46423SBarry Smith   if (lda>A->rmap->n) {
67d0f46423SBarry Smith     nz = PetscBLASIntCast(A->rmap->n);
68d0f46423SBarry Smith     for (j=0; j<A->cmap->n; j++) {
69f4df32b1SMatthew Knepley       BLASscal_(&nz,&oalpha,a->v+j*lda,&one);
70a5ce6ee0Svictorle     }
71a5ce6ee0Svictorle   } else {
72d0f46423SBarry Smith     nz = PetscBLASIntCast(A->rmap->n*A->cmap->n);
73f4df32b1SMatthew Knepley     BLASscal_(&nz,&oalpha,a->v,&one);
74a5ce6ee0Svictorle   }
75efee365bSSatish Balay   ierr = PetscLogFlops(nz);CHKERRQ(ierr);
763a40ed3dSBarry Smith   PetscFunctionReturn(0);
7780cd9d93SLois Curfman McInnes }
7880cd9d93SLois Curfman McInnes 
791cbb95d3SBarry Smith #undef __FUNCT__
801cbb95d3SBarry Smith #define __FUNCT__ "MatIsHermitian_SeqDense"
811cbb95d3SBarry Smith PetscErrorCode MatIsHermitian_SeqDense(Mat A,PetscReal rtol,PetscTruth *fl)
821cbb95d3SBarry Smith {
831cbb95d3SBarry Smith   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
84d0f46423SBarry Smith   PetscInt       i,j,m = A->rmap->n,N;
851cbb95d3SBarry Smith   PetscScalar    *v = a->v;
861cbb95d3SBarry Smith 
871cbb95d3SBarry Smith   PetscFunctionBegin;
881cbb95d3SBarry Smith   *fl = PETSC_FALSE;
89d0f46423SBarry Smith   if (A->rmap->n != A->cmap->n) PetscFunctionReturn(0);
901cbb95d3SBarry Smith   N = a->lda;
911cbb95d3SBarry Smith 
921cbb95d3SBarry Smith   for (i=0; i<m; i++) {
931cbb95d3SBarry Smith     for (j=i+1; j<m; j++) {
941cbb95d3SBarry Smith       if (PetscAbsScalar(v[i+j*N] - PetscConj(v[j+i*N])) > rtol) PetscFunctionReturn(0);
951cbb95d3SBarry Smith     }
961cbb95d3SBarry Smith   }
971cbb95d3SBarry Smith   *fl = PETSC_TRUE;
981cbb95d3SBarry Smith   PetscFunctionReturn(0);
991cbb95d3SBarry Smith }
1001cbb95d3SBarry Smith 
101b24902e0SBarry Smith #undef __FUNCT__
102b24902e0SBarry Smith #define __FUNCT__ "MatDuplicateNoCreate_SeqDense"
103719d5645SBarry Smith PetscErrorCode MatDuplicateNoCreate_SeqDense(Mat newi,Mat A,MatDuplicateOption cpvalues)
104b24902e0SBarry Smith {
105b24902e0SBarry Smith   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data,*l;
106b24902e0SBarry Smith   PetscErrorCode ierr;
107b24902e0SBarry Smith   PetscInt       lda = (PetscInt)mat->lda,j,m;
108b24902e0SBarry Smith 
109b24902e0SBarry Smith   PetscFunctionBegin;
110719d5645SBarry Smith   ierr = MatSeqDenseSetPreallocation(newi,PETSC_NULL);CHKERRQ(ierr);
111b24902e0SBarry Smith   if (cpvalues == MAT_COPY_VALUES) {
112b24902e0SBarry Smith     l = (Mat_SeqDense*)newi->data;
113d0f46423SBarry Smith     if (lda>A->rmap->n) {
114d0f46423SBarry Smith       m = A->rmap->n;
115d0f46423SBarry Smith       for (j=0; j<A->cmap->n; j++) {
116b24902e0SBarry Smith 	ierr = PetscMemcpy(l->v+j*m,mat->v+j*lda,m*sizeof(PetscScalar));CHKERRQ(ierr);
117b24902e0SBarry Smith       }
118b24902e0SBarry Smith     } else {
119d0f46423SBarry Smith       ierr = PetscMemcpy(l->v,mat->v,A->rmap->n*A->cmap->n*sizeof(PetscScalar));CHKERRQ(ierr);
120b24902e0SBarry Smith     }
121b24902e0SBarry Smith   }
122b24902e0SBarry Smith   newi->assembled = PETSC_TRUE;
123b24902e0SBarry Smith   PetscFunctionReturn(0);
124b24902e0SBarry Smith }
125b24902e0SBarry Smith 
1264a2ae208SSatish Balay #undef __FUNCT__
1274a2ae208SSatish Balay #define __FUNCT__ "MatDuplicate_SeqDense"
128dfbe8321SBarry Smith PetscErrorCode MatDuplicate_SeqDense(Mat A,MatDuplicateOption cpvalues,Mat *newmat)
12902cad45dSBarry Smith {
1306849ba73SBarry Smith   PetscErrorCode ierr;
13102cad45dSBarry Smith 
1323a40ed3dSBarry Smith   PetscFunctionBegin;
1335c9eb25fSBarry Smith   ierr = MatCreate(((PetscObject)A)->comm,newmat);CHKERRQ(ierr);
134d0f46423SBarry Smith   ierr = MatSetSizes(*newmat,A->rmap->n,A->cmap->n,A->rmap->n,A->cmap->n);CHKERRQ(ierr);
1355c9eb25fSBarry Smith   ierr = MatSetType(*newmat,((PetscObject)A)->type_name);CHKERRQ(ierr);
136719d5645SBarry Smith   ierr = MatDuplicateNoCreate_SeqDense(*newmat,A,cpvalues);CHKERRQ(ierr);
137b24902e0SBarry Smith   PetscFunctionReturn(0);
138b24902e0SBarry Smith }
139b24902e0SBarry Smith 
1406ee01492SSatish Balay 
1410481f469SBarry Smith extern PetscErrorCode MatLUFactor_SeqDense(Mat,IS,IS,const MatFactorInfo*);
142719d5645SBarry Smith 
1434a2ae208SSatish Balay #undef __FUNCT__
1444a2ae208SSatish Balay #define __FUNCT__ "MatLUFactorNumeric_SeqDense"
1450481f469SBarry Smith PetscErrorCode MatLUFactorNumeric_SeqDense(Mat fact,Mat A,const MatFactorInfo *info_dummy)
146289bc588SBarry Smith {
1474482741eSBarry Smith   MatFactorInfo  info;
148a093e273SMatthew Knepley   PetscErrorCode ierr;
1493a40ed3dSBarry Smith 
1503a40ed3dSBarry Smith   PetscFunctionBegin;
151c3ef05f6SHong Zhang   ierr = MatDuplicateNoCreate_SeqDense(fact,A,MAT_COPY_VALUES);CHKERRQ(ierr);
152719d5645SBarry Smith   ierr = MatLUFactor_SeqDense(fact,0,0,&info);CHKERRQ(ierr);
1533a40ed3dSBarry Smith   PetscFunctionReturn(0);
154289bc588SBarry Smith }
1556ee01492SSatish Balay 
1560b4b3355SBarry Smith #undef __FUNCT__
1574a2ae208SSatish Balay #define __FUNCT__ "MatSolve_SeqDense"
158dfbe8321SBarry Smith PetscErrorCode MatSolve_SeqDense(Mat A,Vec xx,Vec yy)
159289bc588SBarry Smith {
160c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
1616849ba73SBarry Smith   PetscErrorCode ierr;
16287828ca2SBarry Smith   PetscScalar    *x,*y;
163d0f46423SBarry Smith   PetscBLASInt   one = 1,info,m = PetscBLASIntCast(A->rmap->n);
16467e560aaSBarry Smith 
1653a40ed3dSBarry Smith   PetscFunctionBegin;
1661ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
1671ebc52fbSHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
168d0f46423SBarry Smith   ierr = PetscMemcpy(y,x,A->rmap->n*sizeof(PetscScalar));CHKERRQ(ierr);
169d5f3da31SBarry Smith   if (A->factortype == MAT_FACTOR_LU) {
170ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_GETRS)
171e32f2f54SBarry Smith     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"GETRS - Lapack routine is unavailable.");
172ae7cfcebSSatish Balay #else
17371044d3cSBarry Smith     LAPACKgetrs_("N",&m,&one,mat->v,&mat->lda,mat->pivots,y,&m,&info);
174e32f2f54SBarry Smith     if (info) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"GETRS - Bad solve");
175ae7cfcebSSatish Balay #endif
176d5f3da31SBarry Smith   } else if (A->factortype == MAT_FACTOR_CHOLESKY){
177ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_POTRS)
178e32f2f54SBarry Smith     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"POTRS - Lapack routine is unavailable.");
179ae7cfcebSSatish Balay #else
18071044d3cSBarry Smith     LAPACKpotrs_("L",&m,&one,mat->v,&mat->lda,y,&m,&info);
181e32f2f54SBarry Smith     if (info) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"POTRS Bad solve");
182ae7cfcebSSatish Balay #endif
183289bc588SBarry Smith   }
184e32f2f54SBarry Smith   else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Matrix must be factored to solve");
1851ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
1861ebc52fbSHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
187dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*A->cmap->n*A->cmap->n - A->cmap->n);CHKERRQ(ierr);
1883a40ed3dSBarry Smith   PetscFunctionReturn(0);
189289bc588SBarry Smith }
1906ee01492SSatish Balay 
1914a2ae208SSatish Balay #undef __FUNCT__
1924a2ae208SSatish Balay #define __FUNCT__ "MatSolveTranspose_SeqDense"
193dfbe8321SBarry Smith PetscErrorCode MatSolveTranspose_SeqDense(Mat A,Vec xx,Vec yy)
194da3a660dSBarry Smith {
195c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
196dfbe8321SBarry Smith   PetscErrorCode ierr;
19787828ca2SBarry Smith   PetscScalar    *x,*y;
198d0f46423SBarry Smith   PetscBLASInt   one = 1,info,m = PetscBLASIntCast(A->rmap->n);
19967e560aaSBarry Smith 
2003a40ed3dSBarry Smith   PetscFunctionBegin;
2011ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
2021ebc52fbSHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
203d0f46423SBarry Smith   ierr = PetscMemcpy(y,x,A->rmap->n*sizeof(PetscScalar));CHKERRQ(ierr);
204752f5567SLois Curfman McInnes   /* assume if pivots exist then use LU; else Cholesky */
205da3a660dSBarry Smith   if (mat->pivots) {
206ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_GETRS)
207e32f2f54SBarry Smith     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"GETRS - Lapack routine is unavailable.");
208ae7cfcebSSatish Balay #else
20971044d3cSBarry Smith     LAPACKgetrs_("T",&m,&one,mat->v,&mat->lda,mat->pivots,y,&m,&info);
210e32f2f54SBarry Smith     if (info) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"POTRS - Bad solve");
211ae7cfcebSSatish Balay #endif
2127a97a34bSBarry Smith   } else {
213ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_POTRS)
214e32f2f54SBarry Smith     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"POTRS - Lapack routine is unavailable.");
215ae7cfcebSSatish Balay #else
21671044d3cSBarry Smith     LAPACKpotrs_("L",&m,&one,mat->v,&mat->lda,y,&m,&info);
217e32f2f54SBarry Smith     if (info) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"POTRS - Bad solve");
218ae7cfcebSSatish Balay #endif
219da3a660dSBarry Smith   }
2201ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
2211ebc52fbSHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
222dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*A->cmap->n*A->cmap->n - A->cmap->n);CHKERRQ(ierr);
2233a40ed3dSBarry Smith   PetscFunctionReturn(0);
224da3a660dSBarry Smith }
2256ee01492SSatish Balay 
2264a2ae208SSatish Balay #undef __FUNCT__
2274a2ae208SSatish Balay #define __FUNCT__ "MatSolveAdd_SeqDense"
228dfbe8321SBarry Smith PetscErrorCode MatSolveAdd_SeqDense(Mat A,Vec xx,Vec zz,Vec yy)
229da3a660dSBarry Smith {
230c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
231dfbe8321SBarry Smith   PetscErrorCode ierr;
23287828ca2SBarry Smith   PetscScalar    *x,*y,sone = 1.0;
233da3a660dSBarry Smith   Vec            tmp = 0;
234d0f46423SBarry Smith   PetscBLASInt   one = 1,info,m = PetscBLASIntCast(A->rmap->n);
23567e560aaSBarry Smith 
2363a40ed3dSBarry Smith   PetscFunctionBegin;
2371ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
2381ebc52fbSHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
239d0f46423SBarry Smith   if (!A->rmap->n || !A->cmap->n) PetscFunctionReturn(0);
240da3a660dSBarry Smith   if (yy == zz) {
24178b31e54SBarry Smith     ierr = VecDuplicate(yy,&tmp);CHKERRQ(ierr);
24252e6d16bSBarry Smith     ierr = PetscLogObjectParent(A,tmp);CHKERRQ(ierr);
24378b31e54SBarry Smith     ierr = VecCopy(yy,tmp);CHKERRQ(ierr);
244da3a660dSBarry Smith   }
245d0f46423SBarry Smith   ierr = PetscMemcpy(y,x,A->rmap->n*sizeof(PetscScalar));CHKERRQ(ierr);
246752f5567SLois Curfman McInnes   /* assume if pivots exist then use LU; else Cholesky */
247da3a660dSBarry Smith   if (mat->pivots) {
248ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_GETRS)
249e32f2f54SBarry Smith     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"GETRS - Lapack routine is unavailable.");
250ae7cfcebSSatish Balay #else
25171044d3cSBarry Smith     LAPACKgetrs_("N",&m,&one,mat->v,&mat->lda,mat->pivots,y,&m,&info);
252e32f2f54SBarry Smith     if (info) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Bad solve");
253ae7cfcebSSatish Balay #endif
254a8c6a408SBarry Smith   } else {
255ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_POTRS)
256e32f2f54SBarry Smith     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"POTRS - Lapack routine is unavailable.");
257ae7cfcebSSatish Balay #else
25871044d3cSBarry Smith     LAPACKpotrs_("L",&m,&one,mat->v,&mat->lda,y,&m,&info);
259e32f2f54SBarry Smith     if (info) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Bad solve");
260ae7cfcebSSatish Balay #endif
261da3a660dSBarry Smith   }
2622dcb1b2aSMatthew Knepley   if (tmp) {ierr = VecAXPY(yy,sone,tmp);CHKERRQ(ierr); ierr = VecDestroy(tmp);CHKERRQ(ierr);}
2632dcb1b2aSMatthew Knepley   else     {ierr = VecAXPY(yy,sone,zz);CHKERRQ(ierr);}
2641ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
2651ebc52fbSHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
266dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*A->cmap->n*A->cmap->n);CHKERRQ(ierr);
2673a40ed3dSBarry Smith   PetscFunctionReturn(0);
268da3a660dSBarry Smith }
26967e560aaSBarry Smith 
2704a2ae208SSatish Balay #undef __FUNCT__
2714a2ae208SSatish Balay #define __FUNCT__ "MatSolveTransposeAdd_SeqDense"
272dfbe8321SBarry Smith PetscErrorCode MatSolveTransposeAdd_SeqDense(Mat A,Vec xx,Vec zz,Vec yy)
273da3a660dSBarry Smith {
274c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
2756849ba73SBarry Smith   PetscErrorCode ierr;
27687828ca2SBarry Smith   PetscScalar    *x,*y,sone = 1.0;
277da3a660dSBarry Smith   Vec            tmp;
278d0f46423SBarry Smith   PetscBLASInt   one = 1,info,m = PetscBLASIntCast(A->rmap->n);
27967e560aaSBarry Smith 
2803a40ed3dSBarry Smith   PetscFunctionBegin;
281d0f46423SBarry Smith   if (!A->rmap->n || !A->cmap->n) PetscFunctionReturn(0);
2821ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
2831ebc52fbSHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
284da3a660dSBarry Smith   if (yy == zz) {
28578b31e54SBarry Smith     ierr = VecDuplicate(yy,&tmp);CHKERRQ(ierr);
28652e6d16bSBarry Smith     ierr = PetscLogObjectParent(A,tmp);CHKERRQ(ierr);
28778b31e54SBarry Smith     ierr = VecCopy(yy,tmp);CHKERRQ(ierr);
288da3a660dSBarry Smith   }
289d0f46423SBarry Smith   ierr = PetscMemcpy(y,x,A->rmap->n*sizeof(PetscScalar));CHKERRQ(ierr);
290752f5567SLois Curfman McInnes   /* assume if pivots exist then use LU; else Cholesky */
291da3a660dSBarry Smith   if (mat->pivots) {
292ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_GETRS)
293e32f2f54SBarry Smith     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"GETRS - Lapack routine is unavailable.");
294ae7cfcebSSatish Balay #else
29571044d3cSBarry Smith     LAPACKgetrs_("T",&m,&one,mat->v,&mat->lda,mat->pivots,y,&m,&info);
296e32f2f54SBarry Smith     if (info) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Bad solve");
297ae7cfcebSSatish Balay #endif
2983a40ed3dSBarry Smith   } else {
299ae7cfcebSSatish Balay #if defined(PETSC_MISSING_LAPACK_POTRS)
300e32f2f54SBarry Smith     SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"POTRS - Lapack routine is unavailable.");
301ae7cfcebSSatish Balay #else
30271044d3cSBarry Smith     LAPACKpotrs_("L",&m,&one,mat->v,&mat->lda,y,&m,&info);
303e32f2f54SBarry Smith     if (info) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Bad solve");
304ae7cfcebSSatish Balay #endif
305da3a660dSBarry Smith   }
30690f02eecSBarry Smith   if (tmp) {
3072dcb1b2aSMatthew Knepley     ierr = VecAXPY(yy,sone,tmp);CHKERRQ(ierr);
30890f02eecSBarry Smith     ierr = VecDestroy(tmp);CHKERRQ(ierr);
3093a40ed3dSBarry Smith   } else {
3102dcb1b2aSMatthew Knepley     ierr = VecAXPY(yy,sone,zz);CHKERRQ(ierr);
31190f02eecSBarry Smith   }
3121ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
3131ebc52fbSHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
314dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*A->cmap->n*A->cmap->n);CHKERRQ(ierr);
3153a40ed3dSBarry Smith   PetscFunctionReturn(0);
316da3a660dSBarry Smith }
317db4efbfdSBarry Smith 
318db4efbfdSBarry Smith /* ---------------------------------------------------------------*/
319db4efbfdSBarry Smith /* COMMENT: I have chosen to hide row permutation in the pivots,
320db4efbfdSBarry Smith    rather than put it in the Mat->row slot.*/
321db4efbfdSBarry Smith #undef __FUNCT__
322db4efbfdSBarry Smith #define __FUNCT__ "MatLUFactor_SeqDense"
3230481f469SBarry Smith PetscErrorCode MatLUFactor_SeqDense(Mat A,IS row,IS col,const MatFactorInfo *minfo)
324db4efbfdSBarry Smith {
325db4efbfdSBarry Smith #if defined(PETSC_MISSING_LAPACK_GETRF)
326db4efbfdSBarry Smith   PetscFunctionBegin;
327e32f2f54SBarry Smith   SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"GETRF - Lapack routine is unavailable.");
328db4efbfdSBarry Smith #else
329db4efbfdSBarry Smith   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
330db4efbfdSBarry Smith   PetscErrorCode ierr;
331db4efbfdSBarry Smith   PetscBLASInt   n,m,info;
332db4efbfdSBarry Smith 
333db4efbfdSBarry Smith   PetscFunctionBegin;
334db4efbfdSBarry Smith   n = PetscBLASIntCast(A->cmap->n);
335db4efbfdSBarry Smith   m = PetscBLASIntCast(A->rmap->n);
336db4efbfdSBarry Smith   if (!mat->pivots) {
337db4efbfdSBarry Smith     ierr = PetscMalloc((A->rmap->n+1)*sizeof(PetscBLASInt),&mat->pivots);CHKERRQ(ierr);
338db4efbfdSBarry Smith     ierr = PetscLogObjectMemory(A,A->rmap->n*sizeof(PetscBLASInt));CHKERRQ(ierr);
339db4efbfdSBarry Smith   }
340db4efbfdSBarry Smith   if (!A->rmap->n || !A->cmap->n) PetscFunctionReturn(0);
341db4efbfdSBarry Smith   LAPACKgetrf_(&m,&n,mat->v,&mat->lda,mat->pivots,&info);
342e32f2f54SBarry Smith   if (info<0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_LIB,"Bad argument to LU factorization");
343e32f2f54SBarry Smith   if (info>0) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_MAT_LU_ZRPVT,"Bad LU factorization");
344db4efbfdSBarry Smith   A->ops->solve             = MatSolve_SeqDense;
345db4efbfdSBarry Smith   A->ops->solvetranspose    = MatSolveTranspose_SeqDense;
346db4efbfdSBarry Smith   A->ops->solveadd          = MatSolveAdd_SeqDense;
347db4efbfdSBarry Smith   A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqDense;
348d5f3da31SBarry Smith   A->factortype             = MAT_FACTOR_LU;
349db4efbfdSBarry Smith 
350dc0b31edSSatish Balay   ierr = PetscLogFlops((2.0*A->cmap->n*A->cmap->n*A->cmap->n)/3);CHKERRQ(ierr);
351db4efbfdSBarry Smith #endif
352db4efbfdSBarry Smith   PetscFunctionReturn(0);
353db4efbfdSBarry Smith }
354db4efbfdSBarry Smith 
355db4efbfdSBarry Smith #undef __FUNCT__
356db4efbfdSBarry Smith #define __FUNCT__ "MatCholeskyFactor_SeqDense"
3570481f469SBarry Smith PetscErrorCode MatCholeskyFactor_SeqDense(Mat A,IS perm,const MatFactorInfo *factinfo)
358db4efbfdSBarry Smith {
359db4efbfdSBarry Smith #if defined(PETSC_MISSING_LAPACK_POTRF)
360db4efbfdSBarry Smith   PetscFunctionBegin;
361e32f2f54SBarry Smith   SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"POTRF - Lapack routine is unavailable.");
362db4efbfdSBarry Smith #else
363db4efbfdSBarry Smith   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
364db4efbfdSBarry Smith   PetscErrorCode ierr;
365db4efbfdSBarry Smith   PetscBLASInt   info,n = PetscBLASIntCast(A->cmap->n);
366db4efbfdSBarry Smith 
367db4efbfdSBarry Smith   PetscFunctionBegin;
368db4efbfdSBarry Smith   ierr = PetscFree(mat->pivots);CHKERRQ(ierr);
369db4efbfdSBarry Smith   mat->pivots = 0;
370db4efbfdSBarry Smith 
371db4efbfdSBarry Smith   if (!A->rmap->n || !A->cmap->n) PetscFunctionReturn(0);
372db4efbfdSBarry Smith   LAPACKpotrf_("L",&n,mat->v,&mat->lda,&info);
373e32f2f54SBarry Smith   if (info) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_MAT_CH_ZRPVT,"Bad factorization: zero pivot in row %D",(PetscInt)info-1);
374db4efbfdSBarry Smith   A->ops->solve             = MatSolve_SeqDense;
375db4efbfdSBarry Smith   A->ops->solvetranspose    = MatSolveTranspose_SeqDense;
376db4efbfdSBarry Smith   A->ops->solveadd          = MatSolveAdd_SeqDense;
377db4efbfdSBarry Smith   A->ops->solvetransposeadd = MatSolveTransposeAdd_SeqDense;
378d5f3da31SBarry Smith   A->factortype             = MAT_FACTOR_CHOLESKY;
379dc0b31edSSatish Balay   ierr = PetscLogFlops((A->cmap->n*A->cmap->n*A->cmap->n)/3.0);CHKERRQ(ierr);
380db4efbfdSBarry Smith #endif
381db4efbfdSBarry Smith   PetscFunctionReturn(0);
382db4efbfdSBarry Smith }
383db4efbfdSBarry Smith 
384db4efbfdSBarry Smith 
385db4efbfdSBarry Smith #undef __FUNCT__
386db4efbfdSBarry Smith #define __FUNCT__ "MatCholeskyFactorNumeric_SeqDense"
3870481f469SBarry Smith PetscErrorCode MatCholeskyFactorNumeric_SeqDense(Mat fact,Mat A,const MatFactorInfo *info_dummy)
388db4efbfdSBarry Smith {
389db4efbfdSBarry Smith   PetscErrorCode ierr;
390db4efbfdSBarry Smith   MatFactorInfo  info;
391db4efbfdSBarry Smith 
392db4efbfdSBarry Smith   PetscFunctionBegin;
393db4efbfdSBarry Smith   info.fill = 1.0;
394c3ef05f6SHong Zhang   ierr = MatDuplicateNoCreate_SeqDense(fact,A,MAT_COPY_VALUES);CHKERRQ(ierr);
395719d5645SBarry Smith   ierr = MatCholeskyFactor_SeqDense(fact,0,&info);CHKERRQ(ierr);
396db4efbfdSBarry Smith   PetscFunctionReturn(0);
397db4efbfdSBarry Smith }
398db4efbfdSBarry Smith 
399db4efbfdSBarry Smith #undef __FUNCT__
400db4efbfdSBarry Smith #define __FUNCT__ "MatCholeskyFactorSymbolic_SeqDense"
4010481f469SBarry Smith PetscErrorCode MatCholeskyFactorSymbolic_SeqDense(Mat fact,Mat A,IS row,const MatFactorInfo *info)
402db4efbfdSBarry Smith {
403db4efbfdSBarry Smith   PetscFunctionBegin;
404c3ef05f6SHong Zhang   fact->assembled                  = PETSC_TRUE;
405719d5645SBarry Smith   fact->ops->choleskyfactornumeric = MatCholeskyFactorNumeric_SeqDense;
406db4efbfdSBarry Smith   PetscFunctionReturn(0);
407db4efbfdSBarry Smith }
408db4efbfdSBarry Smith 
409db4efbfdSBarry Smith #undef __FUNCT__
410db4efbfdSBarry Smith #define __FUNCT__ "MatLUFactorSymbolic_SeqDense"
4110481f469SBarry Smith PetscErrorCode MatLUFactorSymbolic_SeqDense(Mat fact,Mat A,IS row,IS col,const MatFactorInfo *info)
412db4efbfdSBarry Smith {
413db4efbfdSBarry Smith   PetscFunctionBegin;
414c3ef05f6SHong Zhang   fact->assembled            = PETSC_TRUE;
415719d5645SBarry Smith   fact->ops->lufactornumeric = MatLUFactorNumeric_SeqDense;
416db4efbfdSBarry Smith   PetscFunctionReturn(0);
417db4efbfdSBarry Smith }
418db4efbfdSBarry Smith 
419bb5747d9SMatthew Knepley EXTERN_C_BEGIN
420db4efbfdSBarry Smith #undef __FUNCT__
421db4efbfdSBarry Smith #define __FUNCT__ "MatGetFactor_seqdense_petsc"
422db4efbfdSBarry Smith PetscErrorCode MatGetFactor_seqdense_petsc(Mat A,MatFactorType ftype,Mat *fact)
423db4efbfdSBarry Smith {
424db4efbfdSBarry Smith   PetscErrorCode ierr;
425db4efbfdSBarry Smith 
426db4efbfdSBarry Smith   PetscFunctionBegin;
427db4efbfdSBarry Smith   ierr = MatCreate(((PetscObject)A)->comm,fact);CHKERRQ(ierr);
428db4efbfdSBarry Smith   ierr = MatSetSizes(*fact,A->rmap->n,A->cmap->n,A->rmap->n,A->cmap->n);CHKERRQ(ierr);
429db4efbfdSBarry Smith   ierr = MatSetType(*fact,((PetscObject)A)->type_name);CHKERRQ(ierr);
430db4efbfdSBarry Smith   if (ftype == MAT_FACTOR_LU){
431db4efbfdSBarry Smith     (*fact)->ops->lufactorsymbolic = MatLUFactorSymbolic_SeqDense;
432db4efbfdSBarry Smith   } else {
433db4efbfdSBarry Smith     (*fact)->ops->choleskyfactorsymbolic = MatCholeskyFactorSymbolic_SeqDense;
434db4efbfdSBarry Smith   }
435d5f3da31SBarry Smith   (*fact)->factortype = ftype;
436db4efbfdSBarry Smith   PetscFunctionReturn(0);
437db4efbfdSBarry Smith }
438bb5747d9SMatthew Knepley EXTERN_C_END
439db4efbfdSBarry Smith 
440289bc588SBarry Smith /* ------------------------------------------------------------------*/
4414a2ae208SSatish Balay #undef __FUNCT__
44241f059aeSBarry Smith #define __FUNCT__ "MatSOR_SeqDense"
44341f059aeSBarry Smith PetscErrorCode MatSOR_SeqDense(Mat A,Vec bb,PetscReal omega,MatSORType flag,PetscReal shift,PetscInt its,PetscInt lits,Vec xx)
444289bc588SBarry Smith {
445c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
44687828ca2SBarry Smith   PetscScalar    *x,*b,*v = mat->v,zero = 0.0,xt;
447dfbe8321SBarry Smith   PetscErrorCode ierr;
448d0f46423SBarry Smith   PetscInt       m = A->rmap->n,i;
449aa482453SBarry Smith #if !defined(PETSC_USE_COMPLEX)
4500805154bSBarry Smith   PetscBLASInt   o = 1,bm = PetscBLASIntCast(m);
451bc1b551cSSatish Balay #endif
452289bc588SBarry Smith 
4533a40ed3dSBarry Smith   PetscFunctionBegin;
454289bc588SBarry Smith   if (flag & SOR_ZERO_INITIAL_GUESS) {
45571044d3cSBarry Smith     /* this is a hack fix, should have another version without the second BLASdot */
4562dcb1b2aSMatthew Knepley     ierr = VecSet(xx,zero);CHKERRQ(ierr);
457289bc588SBarry Smith   }
4581ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
4591ebc52fbSHong Zhang   ierr = VecGetArray(bb,&b);CHKERRQ(ierr);
460b965ef7fSBarry Smith   its  = its*lits;
461e32f2f54SBarry 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);
462289bc588SBarry Smith   while (its--) {
463fccaa45eSBarry Smith     if (flag & SOR_FORWARD_SWEEP || flag & SOR_LOCAL_FORWARD_SWEEP){
464289bc588SBarry Smith       for (i=0; i<m; i++) {
465aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX)
466f1747703SBarry Smith         /* cannot use BLAS dot for complex because compiler/linker is
467f1747703SBarry Smith            not happy about returning a double complex */
46813f74950SBarry Smith         PetscInt    _i;
46987828ca2SBarry Smith         PetscScalar sum = b[i];
470f1747703SBarry Smith         for (_i=0; _i<m; _i++) {
4713f6de6efSSatish Balay           sum -= PetscConj(v[i+_i*m])*x[_i];
472f1747703SBarry Smith         }
473f1747703SBarry Smith         xt = sum;
474f1747703SBarry Smith #else
47571044d3cSBarry Smith         xt = b[i] - BLASdot_(&bm,v+i,&bm,x,&o);
476f1747703SBarry Smith #endif
47755a1b374SBarry Smith         x[i] = (1. - omega)*x[i] + omega*(xt+v[i + i*m]*x[i])/(v[i + i*m]+shift);
478289bc588SBarry Smith       }
479289bc588SBarry Smith     }
480fccaa45eSBarry Smith     if (flag & SOR_BACKWARD_SWEEP || flag & SOR_LOCAL_BACKWARD_SWEEP){
481289bc588SBarry Smith       for (i=m-1; i>=0; i--) {
482aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX)
483f1747703SBarry Smith         /* cannot use BLAS dot for complex because compiler/linker is
484f1747703SBarry Smith            not happy about returning a double complex */
48513f74950SBarry Smith         PetscInt    _i;
48687828ca2SBarry Smith         PetscScalar sum = b[i];
487f1747703SBarry Smith         for (_i=0; _i<m; _i++) {
4883f6de6efSSatish Balay           sum -= PetscConj(v[i+_i*m])*x[_i];
489f1747703SBarry Smith         }
490f1747703SBarry Smith         xt = sum;
491f1747703SBarry Smith #else
49271044d3cSBarry Smith         xt = b[i] - BLASdot_(&bm,v+i,&bm,x,&o);
493f1747703SBarry Smith #endif
49455a1b374SBarry Smith         x[i] = (1. - omega)*x[i] + omega*(xt+v[i + i*m]*x[i])/(v[i + i*m]+shift);
495289bc588SBarry Smith       }
496289bc588SBarry Smith     }
497289bc588SBarry Smith   }
4981ebc52fbSHong Zhang   ierr = VecRestoreArray(bb,&b);CHKERRQ(ierr);
4991ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
5003a40ed3dSBarry Smith   PetscFunctionReturn(0);
501289bc588SBarry Smith }
502289bc588SBarry Smith 
503289bc588SBarry Smith /* -----------------------------------------------------------------*/
5044a2ae208SSatish Balay #undef __FUNCT__
5054a2ae208SSatish Balay #define __FUNCT__ "MatMultTranspose_SeqDense"
506dfbe8321SBarry Smith PetscErrorCode MatMultTranspose_SeqDense(Mat A,Vec xx,Vec yy)
507289bc588SBarry Smith {
508c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
50987828ca2SBarry Smith   PetscScalar    *v = mat->v,*x,*y;
510dfbe8321SBarry Smith   PetscErrorCode ierr;
5110805154bSBarry Smith   PetscBLASInt   m, n,_One=1;
512ea709b57SSatish Balay   PetscScalar    _DOne=1.0,_DZero=0.0;
5133a40ed3dSBarry Smith 
5143a40ed3dSBarry Smith   PetscFunctionBegin;
515d0f46423SBarry Smith   m = PetscBLASIntCast(A->rmap->n);
516d0f46423SBarry Smith   n = PetscBLASIntCast(A->cmap->n);
517d0f46423SBarry Smith   if (!A->rmap->n || !A->cmap->n) PetscFunctionReturn(0);
5181ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
5191ebc52fbSHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
52071044d3cSBarry Smith   BLASgemv_("T",&m,&n,&_DOne,v,&mat->lda,x,&_One,&_DZero,y,&_One);
5211ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
5221ebc52fbSHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
523dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*A->rmap->n*A->cmap->n - A->cmap->n);CHKERRQ(ierr);
5243a40ed3dSBarry Smith   PetscFunctionReturn(0);
525289bc588SBarry Smith }
526800995b7SMatthew Knepley 
5274a2ae208SSatish Balay #undef __FUNCT__
5284a2ae208SSatish Balay #define __FUNCT__ "MatMult_SeqDense"
529dfbe8321SBarry Smith PetscErrorCode MatMult_SeqDense(Mat A,Vec xx,Vec yy)
530289bc588SBarry Smith {
531c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
53287828ca2SBarry Smith   PetscScalar    *v = mat->v,*x,*y,_DOne=1.0,_DZero=0.0;
533dfbe8321SBarry Smith   PetscErrorCode ierr;
5340805154bSBarry Smith   PetscBLASInt   m, n, _One=1;
5353a40ed3dSBarry Smith 
5363a40ed3dSBarry Smith   PetscFunctionBegin;
537d0f46423SBarry Smith   m = PetscBLASIntCast(A->rmap->n);
538d0f46423SBarry Smith   n = PetscBLASIntCast(A->cmap->n);
539d0f46423SBarry Smith   if (!A->rmap->n || !A->cmap->n) PetscFunctionReturn(0);
5401ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
5411ebc52fbSHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
54271044d3cSBarry Smith   BLASgemv_("N",&m,&n,&_DOne,v,&(mat->lda),x,&_One,&_DZero,y,&_One);
5431ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
5441ebc52fbSHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
545dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*A->rmap->n*A->cmap->n - A->rmap->n);CHKERRQ(ierr);
5463a40ed3dSBarry Smith   PetscFunctionReturn(0);
547289bc588SBarry Smith }
5486ee01492SSatish Balay 
5494a2ae208SSatish Balay #undef __FUNCT__
5504a2ae208SSatish Balay #define __FUNCT__ "MatMultAdd_SeqDense"
551dfbe8321SBarry Smith PetscErrorCode MatMultAdd_SeqDense(Mat A,Vec xx,Vec zz,Vec yy)
552289bc588SBarry Smith {
553c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
55487828ca2SBarry Smith   PetscScalar    *v = mat->v,*x,*y,_DOne=1.0;
555dfbe8321SBarry Smith   PetscErrorCode ierr;
5560805154bSBarry Smith   PetscBLASInt   m, n, _One=1;
5573a40ed3dSBarry Smith 
5583a40ed3dSBarry Smith   PetscFunctionBegin;
559d0f46423SBarry Smith   m = PetscBLASIntCast(A->rmap->n);
560d0f46423SBarry Smith   n = PetscBLASIntCast(A->cmap->n);
561d0f46423SBarry Smith   if (!A->rmap->n || !A->cmap->n) PetscFunctionReturn(0);
562600d6d0bSBarry Smith   if (zz != yy) {ierr = VecCopy(zz,yy);CHKERRQ(ierr);}
5631ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
5641ebc52fbSHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
56571044d3cSBarry Smith   BLASgemv_("N",&m,&n,&_DOne,v,&(mat->lda),x,&_One,&_DOne,y,&_One);
5661ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
5671ebc52fbSHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
568dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*A->rmap->n*A->cmap->n);CHKERRQ(ierr);
5693a40ed3dSBarry Smith   PetscFunctionReturn(0);
570289bc588SBarry Smith }
5716ee01492SSatish Balay 
5724a2ae208SSatish Balay #undef __FUNCT__
5734a2ae208SSatish Balay #define __FUNCT__ "MatMultTransposeAdd_SeqDense"
574dfbe8321SBarry Smith PetscErrorCode MatMultTransposeAdd_SeqDense(Mat A,Vec xx,Vec zz,Vec yy)
575289bc588SBarry Smith {
576c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
57787828ca2SBarry Smith   PetscScalar    *v = mat->v,*x,*y;
578dfbe8321SBarry Smith   PetscErrorCode ierr;
5790805154bSBarry Smith   PetscBLASInt   m, n, _One=1;
58087828ca2SBarry Smith   PetscScalar    _DOne=1.0;
5813a40ed3dSBarry Smith 
5823a40ed3dSBarry Smith   PetscFunctionBegin;
583d0f46423SBarry Smith   m = PetscBLASIntCast(A->rmap->n);
584d0f46423SBarry Smith   n = PetscBLASIntCast(A->cmap->n);
585d0f46423SBarry Smith   if (!A->rmap->n || !A->cmap->n) PetscFunctionReturn(0);
586600d6d0bSBarry Smith   if (zz != yy) {ierr = VecCopy(zz,yy);CHKERRQ(ierr);}
5871ebc52fbSHong Zhang   ierr = VecGetArray(xx,&x);CHKERRQ(ierr);
5881ebc52fbSHong Zhang   ierr = VecGetArray(yy,&y);CHKERRQ(ierr);
58971044d3cSBarry Smith   BLASgemv_("T",&m,&n,&_DOne,v,&(mat->lda),x,&_One,&_DOne,y,&_One);
5901ebc52fbSHong Zhang   ierr = VecRestoreArray(xx,&x);CHKERRQ(ierr);
5911ebc52fbSHong Zhang   ierr = VecRestoreArray(yy,&y);CHKERRQ(ierr);
592dc0b31edSSatish Balay   ierr = PetscLogFlops(2.0*A->rmap->n*A->cmap->n);CHKERRQ(ierr);
5933a40ed3dSBarry Smith   PetscFunctionReturn(0);
594289bc588SBarry Smith }
595289bc588SBarry Smith 
596289bc588SBarry Smith /* -----------------------------------------------------------------*/
5974a2ae208SSatish Balay #undef __FUNCT__
5984a2ae208SSatish Balay #define __FUNCT__ "MatGetRow_SeqDense"
59913f74950SBarry Smith PetscErrorCode MatGetRow_SeqDense(Mat A,PetscInt row,PetscInt *ncols,PetscInt **cols,PetscScalar **vals)
600289bc588SBarry Smith {
601c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
60287828ca2SBarry Smith   PetscScalar    *v;
6036849ba73SBarry Smith   PetscErrorCode ierr;
60413f74950SBarry Smith   PetscInt       i;
60567e560aaSBarry Smith 
6063a40ed3dSBarry Smith   PetscFunctionBegin;
607d0f46423SBarry Smith   *ncols = A->cmap->n;
608289bc588SBarry Smith   if (cols) {
609d0f46423SBarry Smith     ierr = PetscMalloc((A->cmap->n+1)*sizeof(PetscInt),cols);CHKERRQ(ierr);
610d0f46423SBarry Smith     for (i=0; i<A->cmap->n; i++) (*cols)[i] = i;
611289bc588SBarry Smith   }
612289bc588SBarry Smith   if (vals) {
613d0f46423SBarry Smith     ierr = PetscMalloc((A->cmap->n+1)*sizeof(PetscScalar),vals);CHKERRQ(ierr);
614289bc588SBarry Smith     v    = mat->v + row;
615d0f46423SBarry Smith     for (i=0; i<A->cmap->n; i++) {(*vals)[i] = *v; v += mat->lda;}
616289bc588SBarry Smith   }
6173a40ed3dSBarry Smith   PetscFunctionReturn(0);
618289bc588SBarry Smith }
6196ee01492SSatish Balay 
6204a2ae208SSatish Balay #undef __FUNCT__
6214a2ae208SSatish Balay #define __FUNCT__ "MatRestoreRow_SeqDense"
62213f74950SBarry Smith PetscErrorCode MatRestoreRow_SeqDense(Mat A,PetscInt row,PetscInt *ncols,PetscInt **cols,PetscScalar **vals)
623289bc588SBarry Smith {
624dfbe8321SBarry Smith   PetscErrorCode ierr;
625606d414cSSatish Balay   PetscFunctionBegin;
626606d414cSSatish Balay   if (cols) {ierr = PetscFree(*cols);CHKERRQ(ierr);}
627606d414cSSatish Balay   if (vals) {ierr = PetscFree(*vals);CHKERRQ(ierr); }
6283a40ed3dSBarry Smith   PetscFunctionReturn(0);
629289bc588SBarry Smith }
630289bc588SBarry Smith /* ----------------------------------------------------------------*/
6314a2ae208SSatish Balay #undef __FUNCT__
6324a2ae208SSatish Balay #define __FUNCT__ "MatSetValues_SeqDense"
63313f74950SBarry Smith PetscErrorCode MatSetValues_SeqDense(Mat A,PetscInt m,const PetscInt indexm[],PetscInt n,const PetscInt indexn[],const PetscScalar v[],InsertMode addv)
634289bc588SBarry Smith {
635c0bbcb79SLois Curfman McInnes   Mat_SeqDense *mat = (Mat_SeqDense*)A->data;
63613f74950SBarry Smith   PetscInt     i,j,idx=0;
637d6dfbf8fSBarry Smith 
6383a40ed3dSBarry Smith   PetscFunctionBegin;
63971fd2e92SBarry Smith   if (v) PetscValidScalarPointer(v,6);
640289bc588SBarry Smith   if (!mat->roworiented) {
641dbb450caSBarry Smith     if (addv == INSERT_VALUES) {
642289bc588SBarry Smith       for (j=0; j<n; j++) {
643cddbea37SSatish Balay         if (indexn[j] < 0) {idx += m; continue;}
6442515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
645e32f2f54SBarry 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);
64658804f6dSBarry Smith #endif
647289bc588SBarry Smith         for (i=0; i<m; i++) {
648cddbea37SSatish Balay           if (indexm[i] < 0) {idx++; continue;}
6492515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
650e32f2f54SBarry 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);
65158804f6dSBarry Smith #endif
652cddbea37SSatish Balay           mat->v[indexn[j]*mat->lda + indexm[i]] = v[idx++];
653289bc588SBarry Smith         }
654289bc588SBarry Smith       }
6553a40ed3dSBarry Smith     } else {
656289bc588SBarry Smith       for (j=0; j<n; j++) {
657cddbea37SSatish Balay         if (indexn[j] < 0) {idx += m; continue;}
6582515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
659e32f2f54SBarry 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);
66058804f6dSBarry Smith #endif
661289bc588SBarry Smith         for (i=0; i<m; i++) {
662cddbea37SSatish Balay           if (indexm[i] < 0) {idx++; continue;}
6632515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
664e32f2f54SBarry 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);
66558804f6dSBarry Smith #endif
666cddbea37SSatish Balay           mat->v[indexn[j]*mat->lda + indexm[i]] += v[idx++];
667289bc588SBarry Smith         }
668289bc588SBarry Smith       }
669289bc588SBarry Smith     }
6703a40ed3dSBarry Smith   } else {
671dbb450caSBarry Smith     if (addv == INSERT_VALUES) {
672e8d4e0b9SBarry Smith       for (i=0; i<m; i++) {
673cddbea37SSatish Balay         if (indexm[i] < 0) { idx += n; continue;}
6742515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
675e32f2f54SBarry 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);
67658804f6dSBarry Smith #endif
677e8d4e0b9SBarry Smith         for (j=0; j<n; j++) {
678cddbea37SSatish Balay           if (indexn[j] < 0) { idx++; continue;}
6792515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
680e32f2f54SBarry 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);
68158804f6dSBarry Smith #endif
682cddbea37SSatish Balay           mat->v[indexn[j]*mat->lda + indexm[i]] = v[idx++];
683e8d4e0b9SBarry Smith         }
684e8d4e0b9SBarry Smith       }
6853a40ed3dSBarry Smith     } else {
686289bc588SBarry Smith       for (i=0; i<m; i++) {
687cddbea37SSatish Balay         if (indexm[i] < 0) { idx += n; continue;}
6882515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
689e32f2f54SBarry 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);
69058804f6dSBarry Smith #endif
691289bc588SBarry Smith         for (j=0; j<n; j++) {
692cddbea37SSatish Balay           if (indexn[j] < 0) { idx++; continue;}
6932515c552SBarry Smith #if defined(PETSC_USE_DEBUG)
694e32f2f54SBarry 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);
69558804f6dSBarry Smith #endif
696cddbea37SSatish Balay           mat->v[indexn[j]*mat->lda + indexm[i]] += v[idx++];
697289bc588SBarry Smith         }
698289bc588SBarry Smith       }
699289bc588SBarry Smith     }
700e8d4e0b9SBarry Smith   }
7013a40ed3dSBarry Smith   PetscFunctionReturn(0);
702289bc588SBarry Smith }
703e8d4e0b9SBarry Smith 
7044a2ae208SSatish Balay #undef __FUNCT__
7054a2ae208SSatish Balay #define __FUNCT__ "MatGetValues_SeqDense"
70613f74950SBarry Smith PetscErrorCode MatGetValues_SeqDense(Mat A,PetscInt m,const PetscInt indexm[],PetscInt n,const PetscInt indexn[],PetscScalar v[])
707ae80bb75SLois Curfman McInnes {
708ae80bb75SLois Curfman McInnes   Mat_SeqDense *mat = (Mat_SeqDense*)A->data;
70913f74950SBarry Smith   PetscInt     i,j;
710ae80bb75SLois Curfman McInnes 
7113a40ed3dSBarry Smith   PetscFunctionBegin;
712ae80bb75SLois Curfman McInnes   /* row-oriented output */
713ae80bb75SLois Curfman McInnes   for (i=0; i<m; i++) {
71497e567efSBarry Smith     if (indexm[i] < 0) {v += n;continue;}
715e32f2f54SBarry 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);
716ae80bb75SLois Curfman McInnes     for (j=0; j<n; j++) {
7176f31f424SBarry Smith       if (indexn[j] < 0) {v++; continue;}
718e32f2f54SBarry 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);
71997e567efSBarry Smith       *v++ = mat->v[indexn[j]*mat->lda + indexm[i]];
720ae80bb75SLois Curfman McInnes     }
721ae80bb75SLois Curfman McInnes   }
7223a40ed3dSBarry Smith   PetscFunctionReturn(0);
723ae80bb75SLois Curfman McInnes }
724ae80bb75SLois Curfman McInnes 
725289bc588SBarry Smith /* -----------------------------------------------------------------*/
726289bc588SBarry Smith 
7274a2ae208SSatish Balay #undef __FUNCT__
7284a2ae208SSatish Balay #define __FUNCT__ "MatLoad_SeqDense"
729a313700dSBarry Smith PetscErrorCode MatLoad_SeqDense(PetscViewer viewer, const MatType type,Mat *A)
73088e32edaSLois Curfman McInnes {
73188e32edaSLois Curfman McInnes   Mat_SeqDense   *a;
73288e32edaSLois Curfman McInnes   Mat            B;
7336849ba73SBarry Smith   PetscErrorCode ierr;
73413f74950SBarry Smith   PetscInt       *scols,i,j,nz,header[4];
73513f74950SBarry Smith   int            fd;
73613f74950SBarry Smith   PetscMPIInt    size;
73713f74950SBarry Smith   PetscInt       *rowlengths = 0,M,N,*cols;
73887828ca2SBarry Smith   PetscScalar    *vals,*svals,*v,*w;
73919bcc07fSBarry Smith   MPI_Comm       comm = ((PetscObject)viewer)->comm;
74088e32edaSLois Curfman McInnes 
7413a40ed3dSBarry Smith   PetscFunctionBegin;
742d132466eSBarry Smith   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
743e32f2f54SBarry Smith   if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"view must have one processor");
744b0a32e0cSBarry Smith   ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr);
7450752156aSBarry Smith   ierr = PetscBinaryRead(fd,header,4,PETSC_INT);CHKERRQ(ierr);
746e32f2f54SBarry Smith   if (header[0] != MAT_FILE_CLASSID) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Not matrix object");
74788e32edaSLois Curfman McInnes   M = header[1]; N = header[2]; nz = header[3];
74888e32edaSLois Curfman McInnes 
74990ace30eSBarry Smith   if (nz == MATRIX_BINARY_FORMAT_DENSE) { /* matrix in file is dense */
750f69a0ea3SMatthew Knepley     ierr = MatCreate(comm,A);CHKERRQ(ierr);
751f69a0ea3SMatthew Knepley     ierr = MatSetSizes(*A,M,N,M,N);CHKERRQ(ierr);
752be5d1d56SKris Buschelman     ierr = MatSetType(*A,type);CHKERRQ(ierr);
7535c5985e7SKris Buschelman     ierr = MatSeqDenseSetPreallocation(*A,PETSC_NULL);CHKERRQ(ierr);
75490ace30eSBarry Smith     B    = *A;
75590ace30eSBarry Smith     a    = (Mat_SeqDense*)B->data;
7564a41a4d3SSatish Balay     v    = a->v;
7574a41a4d3SSatish Balay     /* Allocate some temp space to read in the values and then flip them
7584a41a4d3SSatish Balay        from row major to column major */
759b72907f3SBarry Smith     ierr = PetscMalloc((M*N > 0 ? M*N : 1)*sizeof(PetscScalar),&w);CHKERRQ(ierr);
76090ace30eSBarry Smith     /* read in nonzero values */
7614a41a4d3SSatish Balay     ierr = PetscBinaryRead(fd,w,M*N,PETSC_SCALAR);CHKERRQ(ierr);
7624a41a4d3SSatish Balay     /* now flip the values and store them in the matrix*/
7634a41a4d3SSatish Balay     for (j=0; j<N; j++) {
7644a41a4d3SSatish Balay       for (i=0; i<M; i++) {
7654a41a4d3SSatish Balay         *v++ =w[i*N+j];
7664a41a4d3SSatish Balay       }
7674a41a4d3SSatish Balay     }
768606d414cSSatish Balay     ierr = PetscFree(w);CHKERRQ(ierr);
7696d4a8577SBarry Smith     ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
7706d4a8577SBarry Smith     ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
77190ace30eSBarry Smith   } else {
77288e32edaSLois Curfman McInnes     /* read row lengths */
77313f74950SBarry Smith     ierr = PetscMalloc((M+1)*sizeof(PetscInt),&rowlengths);CHKERRQ(ierr);
7740752156aSBarry Smith     ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr);
77588e32edaSLois Curfman McInnes 
77688e32edaSLois Curfman McInnes     /* create our matrix */
777f69a0ea3SMatthew Knepley     ierr = MatCreate(comm,A);CHKERRQ(ierr);
778f69a0ea3SMatthew Knepley     ierr = MatSetSizes(*A,M,N,M,N);CHKERRQ(ierr);
779be5d1d56SKris Buschelman     ierr = MatSetType(*A,type);CHKERRQ(ierr);
7805c5985e7SKris Buschelman     ierr = MatSeqDenseSetPreallocation(*A,PETSC_NULL);CHKERRQ(ierr);
78188e32edaSLois Curfman McInnes     B = *A;
78288e32edaSLois Curfman McInnes     a = (Mat_SeqDense*)B->data;
78388e32edaSLois Curfman McInnes     v = a->v;
78488e32edaSLois Curfman McInnes 
78588e32edaSLois Curfman McInnes     /* read column indices and nonzeros */
78613f74950SBarry Smith     ierr = PetscMalloc((nz+1)*sizeof(PetscInt),&scols);CHKERRQ(ierr);
787b0a32e0cSBarry Smith     cols = scols;
7880752156aSBarry Smith     ierr = PetscBinaryRead(fd,cols,nz,PETSC_INT);CHKERRQ(ierr);
78987828ca2SBarry Smith     ierr = PetscMalloc((nz+1)*sizeof(PetscScalar),&svals);CHKERRQ(ierr);
790b0a32e0cSBarry Smith     vals = svals;
7910752156aSBarry Smith     ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR);CHKERRQ(ierr);
79288e32edaSLois Curfman McInnes 
79388e32edaSLois Curfman McInnes     /* insert into matrix */
79488e32edaSLois Curfman McInnes     for (i=0; i<M; i++) {
79588e32edaSLois Curfman McInnes       for (j=0; j<rowlengths[i]; j++) v[i+M*scols[j]] = svals[j];
79688e32edaSLois Curfman McInnes       svals += rowlengths[i]; scols += rowlengths[i];
79788e32edaSLois Curfman McInnes     }
798606d414cSSatish Balay     ierr = PetscFree(vals);CHKERRQ(ierr);
799606d414cSSatish Balay     ierr = PetscFree(cols);CHKERRQ(ierr);
800606d414cSSatish Balay     ierr = PetscFree(rowlengths);CHKERRQ(ierr);
80188e32edaSLois Curfman McInnes 
8026d4a8577SBarry Smith     ierr = MatAssemblyBegin(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
8036d4a8577SBarry Smith     ierr = MatAssemblyEnd(B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
80490ace30eSBarry Smith   }
8053a40ed3dSBarry Smith   PetscFunctionReturn(0);
80688e32edaSLois Curfman McInnes }
80788e32edaSLois Curfman McInnes 
8084a2ae208SSatish Balay #undef __FUNCT__
809*aabbc4fbSShri Abhyankar #define __FUNCT__ "MatLoadnew_SeqDense"
810*aabbc4fbSShri Abhyankar PetscErrorCode MatLoadnew_SeqDense(PetscViewer viewer, Mat newmat)
811*aabbc4fbSShri Abhyankar {
812*aabbc4fbSShri Abhyankar   Mat_SeqDense   *a;
813*aabbc4fbSShri Abhyankar   PetscErrorCode ierr;
814*aabbc4fbSShri Abhyankar   PetscInt       *scols,i,j,nz,header[4];
815*aabbc4fbSShri Abhyankar   int            fd;
816*aabbc4fbSShri Abhyankar   PetscMPIInt    size;
817*aabbc4fbSShri Abhyankar   PetscInt       *rowlengths = 0,M,N,*cols,grows,gcols;
818*aabbc4fbSShri Abhyankar   PetscScalar    *vals,*svals,*v,*w;
819*aabbc4fbSShri Abhyankar   MPI_Comm       comm = ((PetscObject)viewer)->comm;
820*aabbc4fbSShri Abhyankar 
821*aabbc4fbSShri Abhyankar   PetscFunctionBegin;
822*aabbc4fbSShri Abhyankar   ierr = MPI_Comm_size(comm,&size);CHKERRQ(ierr);
823*aabbc4fbSShri Abhyankar   if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"view must have one processor");
824*aabbc4fbSShri Abhyankar   ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr);
825*aabbc4fbSShri Abhyankar   ierr = PetscBinaryRead(fd,header,4,PETSC_INT);CHKERRQ(ierr);
826*aabbc4fbSShri Abhyankar   if (header[0] != MAT_FILE_CLASSID) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_FILE_UNEXPECTED,"Not matrix object");
827*aabbc4fbSShri Abhyankar   M = header[1]; N = header[2]; nz = header[3];
828*aabbc4fbSShri Abhyankar 
829*aabbc4fbSShri Abhyankar   /* set global size if not set already*/
830*aabbc4fbSShri Abhyankar   if (newmat->rmap->n < 0 && newmat->rmap->N < 0 && newmat->cmap->n < 0 && newmat->cmap->N < 0) {
831*aabbc4fbSShri Abhyankar     ierr = MatSetSizes(newmat,M,N,M,N);CHKERRQ(ierr);
832*aabbc4fbSShri Abhyankar   } else {
833*aabbc4fbSShri Abhyankar     /* if sizes and type are already set, check if the vector global sizes are correct */
834*aabbc4fbSShri Abhyankar     ierr = MatGetSize(newmat,&grows,&gcols);CHKERRQ(ierr);
835*aabbc4fbSShri 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);
836*aabbc4fbSShri Abhyankar   }
837*aabbc4fbSShri Abhyankar   ierr = MatSeqDenseSetPreallocation(newmat,PETSC_NULL);CHKERRQ(ierr);
838*aabbc4fbSShri Abhyankar 
839*aabbc4fbSShri Abhyankar   if (nz == MATRIX_BINARY_FORMAT_DENSE) { /* matrix in file is dense */
840*aabbc4fbSShri Abhyankar     a    = (Mat_SeqDense*)newmat->data;
841*aabbc4fbSShri Abhyankar     v    = a->v;
842*aabbc4fbSShri Abhyankar     /* Allocate some temp space to read in the values and then flip them
843*aabbc4fbSShri Abhyankar        from row major to column major */
844*aabbc4fbSShri Abhyankar     ierr = PetscMalloc((M*N > 0 ? M*N : 1)*sizeof(PetscScalar),&w);CHKERRQ(ierr);
845*aabbc4fbSShri Abhyankar     /* read in nonzero values */
846*aabbc4fbSShri Abhyankar     ierr = PetscBinaryRead(fd,w,M*N,PETSC_SCALAR);CHKERRQ(ierr);
847*aabbc4fbSShri Abhyankar     /* now flip the values and store them in the matrix*/
848*aabbc4fbSShri Abhyankar     for (j=0; j<N; j++) {
849*aabbc4fbSShri Abhyankar       for (i=0; i<M; i++) {
850*aabbc4fbSShri Abhyankar         *v++ =w[i*N+j];
851*aabbc4fbSShri Abhyankar       }
852*aabbc4fbSShri Abhyankar     }
853*aabbc4fbSShri Abhyankar     ierr = PetscFree(w);CHKERRQ(ierr);
854*aabbc4fbSShri Abhyankar   } else {
855*aabbc4fbSShri Abhyankar     /* read row lengths */
856*aabbc4fbSShri Abhyankar     ierr = PetscMalloc((M+1)*sizeof(PetscInt),&rowlengths);CHKERRQ(ierr);
857*aabbc4fbSShri Abhyankar     ierr = PetscBinaryRead(fd,rowlengths,M,PETSC_INT);CHKERRQ(ierr);
858*aabbc4fbSShri Abhyankar 
859*aabbc4fbSShri Abhyankar     a = (Mat_SeqDense*)newmat->data;
860*aabbc4fbSShri Abhyankar     v = a->v;
861*aabbc4fbSShri Abhyankar 
862*aabbc4fbSShri Abhyankar     /* read column indices and nonzeros */
863*aabbc4fbSShri Abhyankar     ierr = PetscMalloc((nz+1)*sizeof(PetscInt),&scols);CHKERRQ(ierr);
864*aabbc4fbSShri Abhyankar     cols = scols;
865*aabbc4fbSShri Abhyankar     ierr = PetscBinaryRead(fd,cols,nz,PETSC_INT);CHKERRQ(ierr);
866*aabbc4fbSShri Abhyankar     ierr = PetscMalloc((nz+1)*sizeof(PetscScalar),&svals);CHKERRQ(ierr);
867*aabbc4fbSShri Abhyankar     vals = svals;
868*aabbc4fbSShri Abhyankar     ierr = PetscBinaryRead(fd,vals,nz,PETSC_SCALAR);CHKERRQ(ierr);
869*aabbc4fbSShri Abhyankar 
870*aabbc4fbSShri Abhyankar     /* insert into matrix */
871*aabbc4fbSShri Abhyankar     for (i=0; i<M; i++) {
872*aabbc4fbSShri Abhyankar       for (j=0; j<rowlengths[i]; j++) v[i+M*scols[j]] = svals[j];
873*aabbc4fbSShri Abhyankar       svals += rowlengths[i]; scols += rowlengths[i];
874*aabbc4fbSShri Abhyankar     }
875*aabbc4fbSShri Abhyankar     ierr = PetscFree(vals);CHKERRQ(ierr);
876*aabbc4fbSShri Abhyankar     ierr = PetscFree(cols);CHKERRQ(ierr);
877*aabbc4fbSShri Abhyankar     ierr = PetscFree(rowlengths);CHKERRQ(ierr);
878*aabbc4fbSShri Abhyankar   }
879*aabbc4fbSShri Abhyankar   ierr = MatAssemblyBegin(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
880*aabbc4fbSShri Abhyankar   ierr = MatAssemblyEnd(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
881*aabbc4fbSShri Abhyankar 
882*aabbc4fbSShri Abhyankar   PetscFunctionReturn(0);
883*aabbc4fbSShri Abhyankar }
884*aabbc4fbSShri Abhyankar 
885*aabbc4fbSShri Abhyankar #undef __FUNCT__
8864a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqDense_ASCII"
8876849ba73SBarry Smith static PetscErrorCode MatView_SeqDense_ASCII(Mat A,PetscViewer viewer)
888289bc588SBarry Smith {
889932b0c3eSLois Curfman McInnes   Mat_SeqDense      *a = (Mat_SeqDense*)A->data;
890dfbe8321SBarry Smith   PetscErrorCode    ierr;
89113f74950SBarry Smith   PetscInt          i,j;
8922dcb1b2aSMatthew Knepley   const char        *name;
89387828ca2SBarry Smith   PetscScalar       *v;
894f3ef73ceSBarry Smith   PetscViewerFormat format;
8955f481a85SSatish Balay #if defined(PETSC_USE_COMPLEX)
8965f481a85SSatish Balay   PetscTruth        allreal = PETSC_TRUE;
8975f481a85SSatish Balay #endif
898932b0c3eSLois Curfman McInnes 
8993a40ed3dSBarry Smith   PetscFunctionBegin;
900435da068SBarry Smith   ierr = PetscObjectGetName((PetscObject)A,&name);CHKERRQ(ierr);
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) {
905b0a32e0cSBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_NO);CHKERRQ(ierr);
906d0f46423SBarry Smith     for (i=0; i<A->rmap->n; i++) {
90744cd7ae7SLois Curfman McInnes       v = a->v + i;
90877431f27SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"row %D:",i);CHKERRQ(ierr);
909d0f46423SBarry Smith       for (j=0; j<A->cmap->n; j++) {
910aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX)
911329f5518SBarry Smith         if (PetscRealPart(*v) != 0.0 && PetscImaginaryPart(*v) != 0.0) {
912a83599f4SBarry Smith           ierr = PetscViewerASCIIPrintf(viewer," (%D, %G + %G i) ",j,PetscRealPart(*v),PetscImaginaryPart(*v));CHKERRQ(ierr);
913329f5518SBarry Smith         } else if (PetscRealPart(*v)) {
914a83599f4SBarry Smith           ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",j,PetscRealPart(*v));CHKERRQ(ierr);
9156831982aSBarry Smith         }
91680cd9d93SLois Curfman McInnes #else
9176831982aSBarry Smith         if (*v) {
918a83599f4SBarry Smith           ierr = PetscViewerASCIIPrintf(viewer," (%D, %G) ",j,*v);CHKERRQ(ierr);
9196831982aSBarry Smith         }
92080cd9d93SLois Curfman McInnes #endif
9211b807ce4Svictorle         v += a->lda;
92280cd9d93SLois Curfman McInnes       }
923b0a32e0cSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
92480cd9d93SLois Curfman McInnes     }
925b0a32e0cSBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_YES);CHKERRQ(ierr);
9263a40ed3dSBarry Smith   } else {
927b0a32e0cSBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_NO);CHKERRQ(ierr);
928aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX)
92947989497SBarry Smith     /* determine if matrix has all real values */
93047989497SBarry Smith     v = a->v;
931d0f46423SBarry Smith     for (i=0; i<A->rmap->n*A->cmap->n; i++) {
932ffac6cdbSBarry Smith 	if (PetscImaginaryPart(v[i])) { allreal = PETSC_FALSE; break ;}
93347989497SBarry Smith     }
93447989497SBarry Smith #endif
935fb9695e5SSatish Balay     if (format == PETSC_VIEWER_ASCII_MATLAB) {
9363a7fca6bSBarry Smith       ierr = PetscObjectGetName((PetscObject)A,&name);CHKERRQ(ierr);
937d0f46423SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"%% Size = %D %D \n",A->rmap->n,A->cmap->n);CHKERRQ(ierr);
938d0f46423SBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"%s = zeros(%D,%D);\n",name,A->rmap->n,A->cmap->n);CHKERRQ(ierr);
939fb9695e5SSatish Balay       ierr = PetscViewerASCIIPrintf(viewer,"%s = [\n",name);CHKERRQ(ierr);
940ffac6cdbSBarry Smith     }
941ffac6cdbSBarry Smith 
942d0f46423SBarry Smith     for (i=0; i<A->rmap->n; i++) {
943932b0c3eSLois Curfman McInnes       v = a->v + i;
944d0f46423SBarry Smith       for (j=0; j<A->cmap->n; j++) {
945aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX)
94647989497SBarry Smith         if (allreal) {
947f32d5d43SBarry Smith           ierr = PetscViewerASCIIPrintf(viewer,"%18.16e ",PetscRealPart(*v));CHKERRQ(ierr);
94847989497SBarry Smith         } else {
949f32d5d43SBarry Smith           ierr = PetscViewerASCIIPrintf(viewer,"%18.16e + %18.16e i ",PetscRealPart(*v),PetscImaginaryPart(*v));CHKERRQ(ierr);
95047989497SBarry Smith         }
951289bc588SBarry Smith #else
952f32d5d43SBarry Smith         ierr = PetscViewerASCIIPrintf(viewer,"%18.16e ",*v);CHKERRQ(ierr);
953289bc588SBarry Smith #endif
9541b807ce4Svictorle 	v += a->lda;
955289bc588SBarry Smith       }
956b0a32e0cSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"\n");CHKERRQ(ierr);
957289bc588SBarry Smith     }
958fb9695e5SSatish Balay     if (format == PETSC_VIEWER_ASCII_MATLAB) {
959b0a32e0cSBarry Smith       ierr = PetscViewerASCIIPrintf(viewer,"];\n");CHKERRQ(ierr);
960ffac6cdbSBarry Smith     }
961b0a32e0cSBarry Smith     ierr = PetscViewerASCIIUseTabs(viewer,PETSC_YES);CHKERRQ(ierr);
962da3a660dSBarry Smith   }
963b0a32e0cSBarry Smith   ierr = PetscViewerFlush(viewer);CHKERRQ(ierr);
9643a40ed3dSBarry Smith   PetscFunctionReturn(0);
965289bc588SBarry Smith }
966289bc588SBarry Smith 
9674a2ae208SSatish Balay #undef __FUNCT__
9684a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqDense_Binary"
9696849ba73SBarry Smith static PetscErrorCode MatView_SeqDense_Binary(Mat A,PetscViewer viewer)
970932b0c3eSLois Curfman McInnes {
971932b0c3eSLois Curfman McInnes   Mat_SeqDense      *a = (Mat_SeqDense*)A->data;
9726849ba73SBarry Smith   PetscErrorCode    ierr;
97313f74950SBarry Smith   int               fd;
974d0f46423SBarry Smith   PetscInt          ict,j,n = A->cmap->n,m = A->rmap->n,i,*col_lens,nz = m*n;
97587828ca2SBarry Smith   PetscScalar       *v,*anonz,*vals;
976f3ef73ceSBarry Smith   PetscViewerFormat format;
977932b0c3eSLois Curfman McInnes 
9783a40ed3dSBarry Smith   PetscFunctionBegin;
979b0a32e0cSBarry Smith   ierr = PetscViewerBinaryGetDescriptor(viewer,&fd);CHKERRQ(ierr);
98090ace30eSBarry Smith 
981b0a32e0cSBarry Smith   ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
9827973ad04SBarry Smith   if (format == PETSC_VIEWER_NATIVE) {
98390ace30eSBarry Smith     /* store the matrix as a dense matrix */
98413f74950SBarry Smith     ierr = PetscMalloc(4*sizeof(PetscInt),&col_lens);CHKERRQ(ierr);
9850700a824SBarry Smith     col_lens[0] = MAT_FILE_CLASSID;
98690ace30eSBarry Smith     col_lens[1] = m;
98790ace30eSBarry Smith     col_lens[2] = n;
98890ace30eSBarry Smith     col_lens[3] = MATRIX_BINARY_FORMAT_DENSE;
9896f69ff64SBarry Smith     ierr = PetscBinaryWrite(fd,col_lens,4,PETSC_INT,PETSC_TRUE);CHKERRQ(ierr);
990606d414cSSatish Balay     ierr = PetscFree(col_lens);CHKERRQ(ierr);
99190ace30eSBarry Smith 
99290ace30eSBarry Smith     /* write out matrix, by rows */
99387828ca2SBarry Smith     ierr = PetscMalloc((m*n+1)*sizeof(PetscScalar),&vals);CHKERRQ(ierr);
99490ace30eSBarry Smith     v    = a->v;
99590ace30eSBarry Smith     for (j=0; j<n; j++) {
996578230a0SSatish Balay       for (i=0; i<m; i++) {
997578230a0SSatish Balay         vals[j + i*n] = *v++;
99890ace30eSBarry Smith       }
99990ace30eSBarry Smith     }
10006f69ff64SBarry Smith     ierr = PetscBinaryWrite(fd,vals,n*m,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr);
1001606d414cSSatish Balay     ierr = PetscFree(vals);CHKERRQ(ierr);
100290ace30eSBarry Smith   } else {
100313f74950SBarry Smith     ierr = PetscMalloc((4+nz)*sizeof(PetscInt),&col_lens);CHKERRQ(ierr);
10040700a824SBarry Smith     col_lens[0] = MAT_FILE_CLASSID;
1005932b0c3eSLois Curfman McInnes     col_lens[1] = m;
1006932b0c3eSLois Curfman McInnes     col_lens[2] = n;
1007932b0c3eSLois Curfman McInnes     col_lens[3] = nz;
1008932b0c3eSLois Curfman McInnes 
1009932b0c3eSLois Curfman McInnes     /* store lengths of each row and write (including header) to file */
1010932b0c3eSLois Curfman McInnes     for (i=0; i<m; i++) col_lens[4+i] = n;
10116f69ff64SBarry Smith     ierr = PetscBinaryWrite(fd,col_lens,4+m,PETSC_INT,PETSC_TRUE);CHKERRQ(ierr);
1012932b0c3eSLois Curfman McInnes 
1013932b0c3eSLois Curfman McInnes     /* Possibly should write in smaller increments, not whole matrix at once? */
1014932b0c3eSLois Curfman McInnes     /* store column indices (zero start index) */
1015932b0c3eSLois Curfman McInnes     ict = 0;
1016932b0c3eSLois Curfman McInnes     for (i=0; i<m; i++) {
1017932b0c3eSLois Curfman McInnes       for (j=0; j<n; j++) col_lens[ict++] = j;
1018932b0c3eSLois Curfman McInnes     }
10196f69ff64SBarry Smith     ierr = PetscBinaryWrite(fd,col_lens,nz,PETSC_INT,PETSC_FALSE);CHKERRQ(ierr);
1020606d414cSSatish Balay     ierr = PetscFree(col_lens);CHKERRQ(ierr);
1021932b0c3eSLois Curfman McInnes 
1022932b0c3eSLois Curfman McInnes     /* store nonzero values */
102387828ca2SBarry Smith     ierr = PetscMalloc((nz+1)*sizeof(PetscScalar),&anonz);CHKERRQ(ierr);
1024932b0c3eSLois Curfman McInnes     ict  = 0;
1025932b0c3eSLois Curfman McInnes     for (i=0; i<m; i++) {
1026932b0c3eSLois Curfman McInnes       v = a->v + i;
1027932b0c3eSLois Curfman McInnes       for (j=0; j<n; j++) {
10281b807ce4Svictorle         anonz[ict++] = *v; v += a->lda;
1029932b0c3eSLois Curfman McInnes       }
1030932b0c3eSLois Curfman McInnes     }
10316f69ff64SBarry Smith     ierr = PetscBinaryWrite(fd,anonz,nz,PETSC_SCALAR,PETSC_FALSE);CHKERRQ(ierr);
1032606d414cSSatish Balay     ierr = PetscFree(anonz);CHKERRQ(ierr);
103390ace30eSBarry Smith   }
10343a40ed3dSBarry Smith   PetscFunctionReturn(0);
1035932b0c3eSLois Curfman McInnes }
1036932b0c3eSLois Curfman McInnes 
10374a2ae208SSatish Balay #undef __FUNCT__
10384a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqDense_Draw_Zoom"
1039dfbe8321SBarry Smith PetscErrorCode MatView_SeqDense_Draw_Zoom(PetscDraw draw,void *Aa)
1040f1af5d2fSBarry Smith {
1041f1af5d2fSBarry Smith   Mat               A = (Mat) Aa;
1042f1af5d2fSBarry Smith   Mat_SeqDense      *a = (Mat_SeqDense*)A->data;
10436849ba73SBarry Smith   PetscErrorCode    ierr;
1044d0f46423SBarry Smith   PetscInt          m = A->rmap->n,n = A->cmap->n,color,i,j;
104587828ca2SBarry Smith   PetscScalar       *v = a->v;
1046b0a32e0cSBarry Smith   PetscViewer       viewer;
1047b0a32e0cSBarry Smith   PetscDraw         popup;
1048329f5518SBarry Smith   PetscReal         xl,yl,xr,yr,x_l,x_r,y_l,y_r,scale,maxv = 0.0;
1049f3ef73ceSBarry Smith   PetscViewerFormat format;
1050f1af5d2fSBarry Smith 
1051f1af5d2fSBarry Smith   PetscFunctionBegin;
1052f1af5d2fSBarry Smith 
1053f1af5d2fSBarry Smith   ierr = PetscObjectQuery((PetscObject)A,"Zoomviewer",(PetscObject*)&viewer);CHKERRQ(ierr);
1054b0a32e0cSBarry Smith   ierr = PetscViewerGetFormat(viewer,&format);CHKERRQ(ierr);
1055b0a32e0cSBarry Smith   ierr = PetscDrawGetCoordinates(draw,&xl,&yl,&xr,&yr);CHKERRQ(ierr);
1056f1af5d2fSBarry Smith 
1057f1af5d2fSBarry Smith   /* Loop over matrix elements drawing boxes */
1058fb9695e5SSatish Balay   if (format != PETSC_VIEWER_DRAW_CONTOUR) {
1059f1af5d2fSBarry Smith     /* Blue for negative and Red for positive */
1060b0a32e0cSBarry Smith     color = PETSC_DRAW_BLUE;
1061f1af5d2fSBarry Smith     for(j = 0; j < n; j++) {
1062f1af5d2fSBarry Smith       x_l = j;
1063f1af5d2fSBarry Smith       x_r = x_l + 1.0;
1064f1af5d2fSBarry Smith       for(i = 0; i < m; i++) {
1065f1af5d2fSBarry Smith         y_l = m - i - 1.0;
1066f1af5d2fSBarry Smith         y_r = y_l + 1.0;
1067f1af5d2fSBarry Smith #if defined(PETSC_USE_COMPLEX)
1068329f5518SBarry Smith         if (PetscRealPart(v[j*m+i]) >  0.) {
1069b0a32e0cSBarry Smith           color = PETSC_DRAW_RED;
1070329f5518SBarry Smith         } else if (PetscRealPart(v[j*m+i]) <  0.) {
1071b0a32e0cSBarry Smith           color = PETSC_DRAW_BLUE;
1072f1af5d2fSBarry Smith         } else {
1073f1af5d2fSBarry Smith           continue;
1074f1af5d2fSBarry Smith         }
1075f1af5d2fSBarry Smith #else
1076f1af5d2fSBarry Smith         if (v[j*m+i] >  0.) {
1077b0a32e0cSBarry Smith           color = PETSC_DRAW_RED;
1078f1af5d2fSBarry Smith         } else if (v[j*m+i] <  0.) {
1079b0a32e0cSBarry Smith           color = PETSC_DRAW_BLUE;
1080f1af5d2fSBarry Smith         } else {
1081f1af5d2fSBarry Smith           continue;
1082f1af5d2fSBarry Smith         }
1083f1af5d2fSBarry Smith #endif
1084b0a32e0cSBarry Smith         ierr = PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);CHKERRQ(ierr);
1085f1af5d2fSBarry Smith       }
1086f1af5d2fSBarry Smith     }
1087f1af5d2fSBarry Smith   } else {
1088f1af5d2fSBarry Smith     /* use contour shading to indicate magnitude of values */
1089f1af5d2fSBarry Smith     /* first determine max of all nonzero values */
1090f1af5d2fSBarry Smith     for(i = 0; i < m*n; i++) {
1091f1af5d2fSBarry Smith       if (PetscAbsScalar(v[i]) > maxv) maxv = PetscAbsScalar(v[i]);
1092f1af5d2fSBarry Smith     }
1093b0a32e0cSBarry Smith     scale = (245.0 - PETSC_DRAW_BASIC_COLORS)/maxv;
1094b0a32e0cSBarry Smith     ierr  = PetscDrawGetPopup(draw,&popup);CHKERRQ(ierr);
1095b0a32e0cSBarry Smith     if (popup) {ierr  = PetscDrawScalePopup(popup,0.0,maxv);CHKERRQ(ierr);}
1096f1af5d2fSBarry Smith     for(j = 0; j < n; j++) {
1097f1af5d2fSBarry Smith       x_l = j;
1098f1af5d2fSBarry Smith       x_r = x_l + 1.0;
1099f1af5d2fSBarry Smith       for(i = 0; i < m; i++) {
1100f1af5d2fSBarry Smith         y_l   = m - i - 1.0;
1101f1af5d2fSBarry Smith         y_r   = y_l + 1.0;
1102b0a32e0cSBarry Smith         color = PETSC_DRAW_BASIC_COLORS + (int)(scale*PetscAbsScalar(v[j*m+i]));
1103b0a32e0cSBarry Smith         ierr  = PetscDrawRectangle(draw,x_l,y_l,x_r,y_r,color,color,color,color);CHKERRQ(ierr);
1104f1af5d2fSBarry Smith       }
1105f1af5d2fSBarry Smith     }
1106f1af5d2fSBarry Smith   }
1107f1af5d2fSBarry Smith   PetscFunctionReturn(0);
1108f1af5d2fSBarry Smith }
1109f1af5d2fSBarry Smith 
11104a2ae208SSatish Balay #undef __FUNCT__
11114a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqDense_Draw"
1112dfbe8321SBarry Smith PetscErrorCode MatView_SeqDense_Draw(Mat A,PetscViewer viewer)
1113f1af5d2fSBarry Smith {
1114b0a32e0cSBarry Smith   PetscDraw      draw;
1115f1af5d2fSBarry Smith   PetscTruth     isnull;
1116329f5518SBarry Smith   PetscReal      xr,yr,xl,yl,h,w;
1117dfbe8321SBarry Smith   PetscErrorCode ierr;
1118f1af5d2fSBarry Smith 
1119f1af5d2fSBarry Smith   PetscFunctionBegin;
1120b0a32e0cSBarry Smith   ierr = PetscViewerDrawGetDraw(viewer,0,&draw);CHKERRQ(ierr);
1121b0a32e0cSBarry Smith   ierr = PetscDrawIsNull(draw,&isnull);CHKERRQ(ierr);
1122abc0a331SBarry Smith   if (isnull) PetscFunctionReturn(0);
1123f1af5d2fSBarry Smith 
1124f1af5d2fSBarry Smith   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",(PetscObject)viewer);CHKERRQ(ierr);
1125d0f46423SBarry Smith   xr  = A->cmap->n; yr = A->rmap->n; h = yr/10.0; w = xr/10.0;
1126f1af5d2fSBarry Smith   xr += w;    yr += h;  xl = -w;     yl = -h;
1127b0a32e0cSBarry Smith   ierr = PetscDrawSetCoordinates(draw,xl,yl,xr,yr);CHKERRQ(ierr);
1128b0a32e0cSBarry Smith   ierr = PetscDrawZoom(draw,MatView_SeqDense_Draw_Zoom,A);CHKERRQ(ierr);
1129f1af5d2fSBarry Smith   ierr = PetscObjectCompose((PetscObject)A,"Zoomviewer",PETSC_NULL);CHKERRQ(ierr);
1130f1af5d2fSBarry Smith   PetscFunctionReturn(0);
1131f1af5d2fSBarry Smith }
1132f1af5d2fSBarry Smith 
11334a2ae208SSatish Balay #undef __FUNCT__
11344a2ae208SSatish Balay #define __FUNCT__ "MatView_SeqDense"
1135dfbe8321SBarry Smith PetscErrorCode MatView_SeqDense(Mat A,PetscViewer viewer)
1136932b0c3eSLois Curfman McInnes {
1137dfbe8321SBarry Smith   PetscErrorCode ierr;
11386805f65bSBarry Smith   PetscTruth     iascii,isbinary,isdraw;
1139932b0c3eSLois Curfman McInnes 
11403a40ed3dSBarry Smith   PetscFunctionBegin;
11412692d6eeSBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSCVIEWERASCII,&iascii);CHKERRQ(ierr);
11422692d6eeSBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSCVIEWERBINARY,&isbinary);CHKERRQ(ierr);
11432692d6eeSBarry Smith   ierr = PetscTypeCompare((PetscObject)viewer,PETSCVIEWERDRAW,&isdraw);CHKERRQ(ierr);
11440f5bd95cSBarry Smith 
1145c45a1595SBarry Smith   if (iascii) {
1146c45a1595SBarry Smith     ierr = MatView_SeqDense_ASCII(A,viewer);CHKERRQ(ierr);
11470f5bd95cSBarry Smith   } else if (isbinary) {
11483a40ed3dSBarry Smith     ierr = MatView_SeqDense_Binary(A,viewer);CHKERRQ(ierr);
1149f1af5d2fSBarry Smith   } else if (isdraw) {
1150f1af5d2fSBarry Smith     ierr = MatView_SeqDense_Draw(A,viewer);CHKERRQ(ierr);
11515cd90555SBarry Smith   } else {
1152e32f2f54SBarry Smith     SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"Viewer type %s not supported by dense matrix",((PetscObject)viewer)->type_name);
1153932b0c3eSLois Curfman McInnes   }
11543a40ed3dSBarry Smith   PetscFunctionReturn(0);
1155932b0c3eSLois Curfman McInnes }
1156289bc588SBarry Smith 
11574a2ae208SSatish Balay #undef __FUNCT__
11584a2ae208SSatish Balay #define __FUNCT__ "MatDestroy_SeqDense"
1159dfbe8321SBarry Smith PetscErrorCode MatDestroy_SeqDense(Mat mat)
1160289bc588SBarry Smith {
1161ec8511deSBarry Smith   Mat_SeqDense   *l = (Mat_SeqDense*)mat->data;
1162dfbe8321SBarry Smith   PetscErrorCode ierr;
116390f02eecSBarry Smith 
11643a40ed3dSBarry Smith   PetscFunctionBegin;
1165aa482453SBarry Smith #if defined(PETSC_USE_LOG)
1166d0f46423SBarry Smith   PetscLogObjectState((PetscObject)mat,"Rows %D Cols %D",mat->rmap->n,mat->cmap->n);
1167a5a9c739SBarry Smith #endif
116805b42c5fSBarry Smith   ierr = PetscFree(l->pivots);CHKERRQ(ierr);
11696857c123SSatish Balay   if (!l->user_alloc) {ierr = PetscFree(l->v);CHKERRQ(ierr);}
1170606d414cSSatish Balay   ierr = PetscFree(l);CHKERRQ(ierr);
1171dbd8c25aSHong Zhang 
1172dbd8c25aSHong Zhang   ierr = PetscObjectChangeTypeName((PetscObject)mat,0);CHKERRQ(ierr);
1173901853e0SKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatSeqDenseSetPreallocation_C","",PETSC_NULL);CHKERRQ(ierr);
11744ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMult_seqaij_seqdense_C","",PETSC_NULL);CHKERRQ(ierr);
11754ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMultSymbolic_seqaij_seqdense_C","",PETSC_NULL);CHKERRQ(ierr);
11764ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)mat,"MatMatMultNumeric_seqaij_seqdense_C","",PETSC_NULL);CHKERRQ(ierr);
11773a40ed3dSBarry Smith   PetscFunctionReturn(0);
1178289bc588SBarry Smith }
1179289bc588SBarry Smith 
11804a2ae208SSatish Balay #undef __FUNCT__
11814a2ae208SSatish Balay #define __FUNCT__ "MatTranspose_SeqDense"
1182fc4dec0aSBarry Smith PetscErrorCode MatTranspose_SeqDense(Mat A,MatReuse reuse,Mat *matout)
1183289bc588SBarry Smith {
1184c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
11856849ba73SBarry Smith   PetscErrorCode ierr;
118613f74950SBarry Smith   PetscInt       k,j,m,n,M;
118787828ca2SBarry Smith   PetscScalar    *v,tmp;
118848b35521SBarry Smith 
11893a40ed3dSBarry Smith   PetscFunctionBegin;
1190d0f46423SBarry Smith   v = mat->v; m = A->rmap->n; M = mat->lda; n = A->cmap->n;
1191e9695a30SBarry Smith   if (reuse == MAT_REUSE_MATRIX && *matout == A) { /* in place transpose */
1192e7e72b3dSBarry Smith     if (m != n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"Can not transpose non-square matrix in place");
1193e7e72b3dSBarry Smith     else {
1194d3e5ee88SLois Curfman McInnes       for (j=0; j<m; j++) {
1195289bc588SBarry Smith         for (k=0; k<j; k++) {
11961b807ce4Svictorle           tmp = v[j + k*M];
11971b807ce4Svictorle           v[j + k*M] = v[k + j*M];
11981b807ce4Svictorle           v[k + j*M] = tmp;
1199289bc588SBarry Smith         }
1200289bc588SBarry Smith       }
1201d64ed03dSBarry Smith     }
12023a40ed3dSBarry Smith   } else { /* out-of-place transpose */
1203d3e5ee88SLois Curfman McInnes     Mat          tmat;
1204ec8511deSBarry Smith     Mat_SeqDense *tmatd;
120587828ca2SBarry Smith     PetscScalar  *v2;
1206ea709b57SSatish Balay 
1207fc4dec0aSBarry Smith     if (reuse == MAT_INITIAL_MATRIX) {
12087adad957SLisandro Dalcin       ierr  = MatCreate(((PetscObject)A)->comm,&tmat);CHKERRQ(ierr);
1209d0f46423SBarry Smith       ierr  = MatSetSizes(tmat,A->cmap->n,A->rmap->n,A->cmap->n,A->rmap->n);CHKERRQ(ierr);
12107adad957SLisandro Dalcin       ierr  = MatSetType(tmat,((PetscObject)A)->type_name);CHKERRQ(ierr);
12115c5985e7SKris Buschelman       ierr  = MatSeqDenseSetPreallocation(tmat,PETSC_NULL);CHKERRQ(ierr);
1212fc4dec0aSBarry Smith     } else {
1213fc4dec0aSBarry Smith       tmat = *matout;
1214fc4dec0aSBarry Smith     }
1215ec8511deSBarry Smith     tmatd = (Mat_SeqDense*)tmat->data;
12160de55854SLois Curfman McInnes     v = mat->v; v2 = tmatd->v;
1217d3e5ee88SLois Curfman McInnes     for (j=0; j<n; j++) {
12181b807ce4Svictorle       for (k=0; k<m; k++) v2[j + k*n] = v[k + j*M];
1219d3e5ee88SLois Curfman McInnes     }
12206d4a8577SBarry Smith     ierr = MatAssemblyBegin(tmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
12216d4a8577SBarry Smith     ierr = MatAssemblyEnd(tmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
1222d3e5ee88SLois Curfman McInnes     *matout = tmat;
122348b35521SBarry Smith   }
12243a40ed3dSBarry Smith   PetscFunctionReturn(0);
1225289bc588SBarry Smith }
1226289bc588SBarry Smith 
12274a2ae208SSatish Balay #undef __FUNCT__
12284a2ae208SSatish Balay #define __FUNCT__ "MatEqual_SeqDense"
1229dfbe8321SBarry Smith PetscErrorCode MatEqual_SeqDense(Mat A1,Mat A2,PetscTruth *flg)
1230289bc588SBarry Smith {
1231c0bbcb79SLois Curfman McInnes   Mat_SeqDense *mat1 = (Mat_SeqDense*)A1->data;
1232c0bbcb79SLois Curfman McInnes   Mat_SeqDense *mat2 = (Mat_SeqDense*)A2->data;
123313f74950SBarry Smith   PetscInt     i,j;
123487828ca2SBarry Smith   PetscScalar  *v1 = mat1->v,*v2 = mat2->v;
12359ea5d5aeSSatish Balay 
12363a40ed3dSBarry Smith   PetscFunctionBegin;
1237d0f46423SBarry Smith   if (A1->rmap->n != A2->rmap->n) {*flg = PETSC_FALSE; PetscFunctionReturn(0);}
1238d0f46423SBarry Smith   if (A1->cmap->n != A2->cmap->n) {*flg = PETSC_FALSE; PetscFunctionReturn(0);}
1239d0f46423SBarry Smith   for (i=0; i<A1->rmap->n; i++) {
12401b807ce4Svictorle     v1 = mat1->v+i; v2 = mat2->v+i;
1241d0f46423SBarry Smith     for (j=0; j<A1->cmap->n; j++) {
12423a40ed3dSBarry Smith       if (*v1 != *v2) {*flg = PETSC_FALSE; PetscFunctionReturn(0);}
12431b807ce4Svictorle       v1 += mat1->lda; v2 += mat2->lda;
12441b807ce4Svictorle     }
1245289bc588SBarry Smith   }
124677c4ece6SBarry Smith   *flg = PETSC_TRUE;
12473a40ed3dSBarry Smith   PetscFunctionReturn(0);
1248289bc588SBarry Smith }
1249289bc588SBarry Smith 
12504a2ae208SSatish Balay #undef __FUNCT__
12514a2ae208SSatish Balay #define __FUNCT__ "MatGetDiagonal_SeqDense"
1252dfbe8321SBarry Smith PetscErrorCode MatGetDiagonal_SeqDense(Mat A,Vec v)
1253289bc588SBarry Smith {
1254c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
1255dfbe8321SBarry Smith   PetscErrorCode ierr;
125613f74950SBarry Smith   PetscInt       i,n,len;
125787828ca2SBarry Smith   PetscScalar    *x,zero = 0.0;
125844cd7ae7SLois Curfman McInnes 
12593a40ed3dSBarry Smith   PetscFunctionBegin;
12602dcb1b2aSMatthew Knepley   ierr = VecSet(v,zero);CHKERRQ(ierr);
12617a97a34bSBarry Smith   ierr = VecGetSize(v,&n);CHKERRQ(ierr);
12621ebc52fbSHong Zhang   ierr = VecGetArray(v,&x);CHKERRQ(ierr);
1263d0f46423SBarry Smith   len = PetscMin(A->rmap->n,A->cmap->n);
1264e32f2f54SBarry Smith   if (n != A->rmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Nonconforming mat and vec");
126544cd7ae7SLois Curfman McInnes   for (i=0; i<len; i++) {
12661b807ce4Svictorle     x[i] = mat->v[i*mat->lda + i];
1267289bc588SBarry Smith   }
12681ebc52fbSHong Zhang   ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
12693a40ed3dSBarry Smith   PetscFunctionReturn(0);
1270289bc588SBarry Smith }
1271289bc588SBarry Smith 
12724a2ae208SSatish Balay #undef __FUNCT__
12734a2ae208SSatish Balay #define __FUNCT__ "MatDiagonalScale_SeqDense"
1274dfbe8321SBarry Smith PetscErrorCode MatDiagonalScale_SeqDense(Mat A,Vec ll,Vec rr)
1275289bc588SBarry Smith {
1276c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
127787828ca2SBarry Smith   PetscScalar    *l,*r,x,*v;
1278dfbe8321SBarry Smith   PetscErrorCode ierr;
1279d0f46423SBarry Smith   PetscInt       i,j,m = A->rmap->n,n = A->cmap->n;
128055659b69SBarry Smith 
12813a40ed3dSBarry Smith   PetscFunctionBegin;
128228988994SBarry Smith   if (ll) {
12837a97a34bSBarry Smith     ierr = VecGetSize(ll,&m);CHKERRQ(ierr);
12841ebc52fbSHong Zhang     ierr = VecGetArray(ll,&l);CHKERRQ(ierr);
1285e32f2f54SBarry Smith     if (m != A->rmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Left scaling vec wrong size");
1286da3a660dSBarry Smith     for (i=0; i<m; i++) {
1287da3a660dSBarry Smith       x = l[i];
1288da3a660dSBarry Smith       v = mat->v + i;
1289da3a660dSBarry Smith       for (j=0; j<n; j++) { (*v) *= x; v+= m;}
1290da3a660dSBarry Smith     }
12911ebc52fbSHong Zhang     ierr = VecRestoreArray(ll,&l);CHKERRQ(ierr);
1292efee365bSSatish Balay     ierr = PetscLogFlops(n*m);CHKERRQ(ierr);
1293da3a660dSBarry Smith   }
129428988994SBarry Smith   if (rr) {
12957a97a34bSBarry Smith     ierr = VecGetSize(rr,&n);CHKERRQ(ierr);
12961ebc52fbSHong Zhang     ierr = VecGetArray(rr,&r);CHKERRQ(ierr);
1297e32f2f54SBarry Smith     if (n != A->cmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Right scaling vec wrong size");
1298da3a660dSBarry Smith     for (i=0; i<n; i++) {
1299da3a660dSBarry Smith       x = r[i];
1300da3a660dSBarry Smith       v = mat->v + i*m;
1301da3a660dSBarry Smith       for (j=0; j<m; j++) { (*v++) *= x;}
1302da3a660dSBarry Smith     }
13031ebc52fbSHong Zhang     ierr = VecRestoreArray(rr,&r);CHKERRQ(ierr);
1304efee365bSSatish Balay     ierr = PetscLogFlops(n*m);CHKERRQ(ierr);
1305da3a660dSBarry Smith   }
13063a40ed3dSBarry Smith   PetscFunctionReturn(0);
1307289bc588SBarry Smith }
1308289bc588SBarry Smith 
13094a2ae208SSatish Balay #undef __FUNCT__
13104a2ae208SSatish Balay #define __FUNCT__ "MatNorm_SeqDense"
1311dfbe8321SBarry Smith PetscErrorCode MatNorm_SeqDense(Mat A,NormType type,PetscReal *nrm)
1312289bc588SBarry Smith {
1313c0bbcb79SLois Curfman McInnes   Mat_SeqDense *mat = (Mat_SeqDense*)A->data;
131487828ca2SBarry Smith   PetscScalar  *v = mat->v;
1315329f5518SBarry Smith   PetscReal    sum = 0.0;
1316d0f46423SBarry Smith   PetscInt     lda=mat->lda,m=A->rmap->n,i,j;
1317efee365bSSatish Balay   PetscErrorCode ierr;
131855659b69SBarry Smith 
13193a40ed3dSBarry Smith   PetscFunctionBegin;
1320289bc588SBarry Smith   if (type == NORM_FROBENIUS) {
1321a5ce6ee0Svictorle     if (lda>m) {
1322d0f46423SBarry Smith       for (j=0; j<A->cmap->n; j++) {
1323a5ce6ee0Svictorle 	v = mat->v+j*lda;
1324a5ce6ee0Svictorle 	for (i=0; i<m; i++) {
1325a5ce6ee0Svictorle #if defined(PETSC_USE_COMPLEX)
1326a5ce6ee0Svictorle 	  sum += PetscRealPart(PetscConj(*v)*(*v)); v++;
1327a5ce6ee0Svictorle #else
1328a5ce6ee0Svictorle 	  sum += (*v)*(*v); v++;
1329a5ce6ee0Svictorle #endif
1330a5ce6ee0Svictorle 	}
1331a5ce6ee0Svictorle       }
1332a5ce6ee0Svictorle     } else {
1333d0f46423SBarry Smith       for (i=0; i<A->cmap->n*A->rmap->n; i++) {
1334aa482453SBarry Smith #if defined(PETSC_USE_COMPLEX)
1335329f5518SBarry Smith 	sum += PetscRealPart(PetscConj(*v)*(*v)); v++;
1336289bc588SBarry Smith #else
1337289bc588SBarry Smith 	sum += (*v)*(*v); v++;
1338289bc588SBarry Smith #endif
1339289bc588SBarry Smith       }
1340a5ce6ee0Svictorle     }
1341064f8208SBarry Smith     *nrm = sqrt(sum);
1342dc0b31edSSatish Balay     ierr = PetscLogFlops(2.0*A->cmap->n*A->rmap->n);CHKERRQ(ierr);
13433a40ed3dSBarry Smith   } else if (type == NORM_1) {
1344064f8208SBarry Smith     *nrm = 0.0;
1345d0f46423SBarry Smith     for (j=0; j<A->cmap->n; j++) {
13461b807ce4Svictorle       v = mat->v + j*mat->lda;
1347289bc588SBarry Smith       sum = 0.0;
1348d0f46423SBarry Smith       for (i=0; i<A->rmap->n; i++) {
134933a8263dSBarry Smith         sum += PetscAbsScalar(*v);  v++;
1350289bc588SBarry Smith       }
1351064f8208SBarry Smith       if (sum > *nrm) *nrm = sum;
1352289bc588SBarry Smith     }
1353d0f46423SBarry Smith     ierr = PetscLogFlops(A->cmap->n*A->rmap->n);CHKERRQ(ierr);
13543a40ed3dSBarry Smith   } else if (type == NORM_INFINITY) {
1355064f8208SBarry Smith     *nrm = 0.0;
1356d0f46423SBarry Smith     for (j=0; j<A->rmap->n; j++) {
1357289bc588SBarry Smith       v = mat->v + j;
1358289bc588SBarry Smith       sum = 0.0;
1359d0f46423SBarry Smith       for (i=0; i<A->cmap->n; i++) {
13601b807ce4Svictorle         sum += PetscAbsScalar(*v); v += mat->lda;
1361289bc588SBarry Smith       }
1362064f8208SBarry Smith       if (sum > *nrm) *nrm = sum;
1363289bc588SBarry Smith     }
1364d0f46423SBarry Smith     ierr = PetscLogFlops(A->cmap->n*A->rmap->n);CHKERRQ(ierr);
1365e7e72b3dSBarry Smith   } else SETERRQ(PETSC_COMM_SELF,PETSC_ERR_SUP,"No two norm");
13663a40ed3dSBarry Smith   PetscFunctionReturn(0);
1367289bc588SBarry Smith }
1368289bc588SBarry Smith 
13694a2ae208SSatish Balay #undef __FUNCT__
13704a2ae208SSatish Balay #define __FUNCT__ "MatSetOption_SeqDense"
13714e0d8c25SBarry Smith PetscErrorCode MatSetOption_SeqDense(Mat A,MatOption op,PetscTruth flg)
1372289bc588SBarry Smith {
1373c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *aij = (Mat_SeqDense*)A->data;
137463ba0a88SBarry Smith   PetscErrorCode ierr;
137567e560aaSBarry Smith 
13763a40ed3dSBarry Smith   PetscFunctionBegin;
1377b5a2b587SKris Buschelman   switch (op) {
1378b5a2b587SKris Buschelman   case MAT_ROW_ORIENTED:
13794e0d8c25SBarry Smith     aij->roworiented = flg;
1380b5a2b587SKris Buschelman     break;
1381512a5fc5SBarry Smith   case MAT_NEW_NONZERO_LOCATIONS:
1382b5a2b587SKris Buschelman   case MAT_NEW_NONZERO_LOCATION_ERR:
13833971808eSMatthew Knepley   case MAT_NEW_NONZERO_ALLOCATION_ERR:
13844e0d8c25SBarry Smith   case MAT_NEW_DIAGONALS:
1385b5a2b587SKris Buschelman   case MAT_IGNORE_OFF_PROC_ENTRIES:
1386b5a2b587SKris Buschelman   case MAT_USE_HASH_TABLE:
138777e54ba9SKris Buschelman   case MAT_SYMMETRIC:
138877e54ba9SKris Buschelman   case MAT_STRUCTURALLY_SYMMETRIC:
13899a4540c5SBarry Smith   case MAT_HERMITIAN:
13909a4540c5SBarry Smith   case MAT_SYMMETRY_ETERNAL:
1391600fe468SBarry Smith   case MAT_IGNORE_LOWER_TRIANGULAR:
1392290bbb0aSBarry Smith     ierr = PetscInfo1(A,"Option %s ignored\n",MatOptions[op]);CHKERRQ(ierr);
139377e54ba9SKris Buschelman     break;
1394b5a2b587SKris Buschelman   default:
1395e32f2f54SBarry Smith     SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_SUP,"unknown option %s",MatOptions[op]);
13963a40ed3dSBarry Smith   }
13973a40ed3dSBarry Smith   PetscFunctionReturn(0);
1398289bc588SBarry Smith }
1399289bc588SBarry Smith 
14004a2ae208SSatish Balay #undef __FUNCT__
14014a2ae208SSatish Balay #define __FUNCT__ "MatZeroEntries_SeqDense"
1402dfbe8321SBarry Smith PetscErrorCode MatZeroEntries_SeqDense(Mat A)
14036f0a148fSBarry Smith {
1404ec8511deSBarry Smith   Mat_SeqDense   *l = (Mat_SeqDense*)A->data;
14056849ba73SBarry Smith   PetscErrorCode ierr;
1406d0f46423SBarry Smith   PetscInt       lda=l->lda,m=A->rmap->n,j;
14073a40ed3dSBarry Smith 
14083a40ed3dSBarry Smith   PetscFunctionBegin;
1409a5ce6ee0Svictorle   if (lda>m) {
1410d0f46423SBarry Smith     for (j=0; j<A->cmap->n; j++) {
1411a5ce6ee0Svictorle       ierr = PetscMemzero(l->v+j*lda,m*sizeof(PetscScalar));CHKERRQ(ierr);
1412a5ce6ee0Svictorle     }
1413a5ce6ee0Svictorle   } else {
1414d0f46423SBarry Smith     ierr = PetscMemzero(l->v,A->rmap->n*A->cmap->n*sizeof(PetscScalar));CHKERRQ(ierr);
1415a5ce6ee0Svictorle   }
14163a40ed3dSBarry Smith   PetscFunctionReturn(0);
14176f0a148fSBarry Smith }
14186f0a148fSBarry Smith 
14194a2ae208SSatish Balay #undef __FUNCT__
14204a2ae208SSatish Balay #define __FUNCT__ "MatZeroRows_SeqDense"
1421f4df32b1SMatthew Knepley PetscErrorCode MatZeroRows_SeqDense(Mat A,PetscInt N,const PetscInt rows[],PetscScalar diag)
14226f0a148fSBarry Smith {
1423ec8511deSBarry Smith   Mat_SeqDense   *l = (Mat_SeqDense*)A->data;
1424d0f46423SBarry Smith   PetscInt       n = A->cmap->n,i,j;
142587828ca2SBarry Smith   PetscScalar    *slot;
142655659b69SBarry Smith 
14273a40ed3dSBarry Smith   PetscFunctionBegin;
14286f0a148fSBarry Smith   for (i=0; i<N; i++) {
14296f0a148fSBarry Smith     slot = l->v + rows[i];
14306f0a148fSBarry Smith     for (j=0; j<n; j++) { *slot = 0.0; slot += n;}
14316f0a148fSBarry Smith   }
1432f4df32b1SMatthew Knepley   if (diag != 0.0) {
14336f0a148fSBarry Smith     for (i=0; i<N; i++) {
14346f0a148fSBarry Smith       slot = l->v + (n+1)*rows[i];
1435f4df32b1SMatthew Knepley       *slot = diag;
14366f0a148fSBarry Smith     }
14376f0a148fSBarry Smith   }
14383a40ed3dSBarry Smith   PetscFunctionReturn(0);
14396f0a148fSBarry Smith }
1440557bce09SLois Curfman McInnes 
14414a2ae208SSatish Balay #undef __FUNCT__
14424a2ae208SSatish Balay #define __FUNCT__ "MatGetArray_SeqDense"
1443dfbe8321SBarry Smith PetscErrorCode MatGetArray_SeqDense(Mat A,PetscScalar *array[])
144464e87e97SBarry Smith {
1445c0bbcb79SLois Curfman McInnes   Mat_SeqDense *mat = (Mat_SeqDense*)A->data;
14463a40ed3dSBarry Smith 
14473a40ed3dSBarry Smith   PetscFunctionBegin;
1448e32f2f54SBarry 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");
144964e87e97SBarry Smith   *array = mat->v;
14503a40ed3dSBarry Smith   PetscFunctionReturn(0);
145164e87e97SBarry Smith }
14520754003eSLois Curfman McInnes 
14534a2ae208SSatish Balay #undef __FUNCT__
14544a2ae208SSatish Balay #define __FUNCT__ "MatRestoreArray_SeqDense"
1455dfbe8321SBarry Smith PetscErrorCode MatRestoreArray_SeqDense(Mat A,PetscScalar *array[])
1456ff14e315SSatish Balay {
14573a40ed3dSBarry Smith   PetscFunctionBegin;
145809b544d4SBarry Smith   *array = 0; /* user cannot accidently use the array later */
14593a40ed3dSBarry Smith   PetscFunctionReturn(0);
1460ff14e315SSatish Balay }
14610754003eSLois Curfman McInnes 
14624a2ae208SSatish Balay #undef __FUNCT__
14634a2ae208SSatish Balay #define __FUNCT__ "MatGetSubMatrix_SeqDense"
146413f74950SBarry Smith static PetscErrorCode MatGetSubMatrix_SeqDense(Mat A,IS isrow,IS iscol,PetscInt cs,MatReuse scall,Mat *B)
14650754003eSLois Curfman McInnes {
1466c0bbcb79SLois Curfman McInnes   Mat_SeqDense   *mat = (Mat_SeqDense*)A->data;
14676849ba73SBarry Smith   PetscErrorCode ierr;
14685d0c19d7SBarry Smith   PetscInt       i,j,nrows,ncols;
14695d0c19d7SBarry Smith   const PetscInt *irow,*icol;
147087828ca2SBarry Smith   PetscScalar    *av,*bv,*v = mat->v;
14710754003eSLois Curfman McInnes   Mat            newmat;
14720754003eSLois Curfman McInnes 
14733a40ed3dSBarry Smith   PetscFunctionBegin;
147478b31e54SBarry Smith   ierr = ISGetIndices(isrow,&irow);CHKERRQ(ierr);
147578b31e54SBarry Smith   ierr = ISGetIndices(iscol,&icol);CHKERRQ(ierr);
1476e03a110bSBarry Smith   ierr = ISGetLocalSize(isrow,&nrows);CHKERRQ(ierr);
1477e03a110bSBarry Smith   ierr = ISGetLocalSize(iscol,&ncols);CHKERRQ(ierr);
14780754003eSLois Curfman McInnes 
1479182d2002SSatish Balay   /* Check submatrixcall */
1480182d2002SSatish Balay   if (scall == MAT_REUSE_MATRIX) {
148113f74950SBarry Smith     PetscInt n_cols,n_rows;
1482182d2002SSatish Balay     ierr = MatGetSize(*B,&n_rows,&n_cols);CHKERRQ(ierr);
148321a2c019SBarry Smith     if (n_rows != nrows || n_cols != ncols) {
148421a2c019SBarry Smith       /* resize the result result matrix to match number of requested rows/columns */
1485c61587bbSBarry Smith       ierr = MatSetSizes(*B,nrows,ncols,nrows,ncols);CHKERRQ(ierr);
148621a2c019SBarry Smith     }
1487182d2002SSatish Balay     newmat = *B;
1488182d2002SSatish Balay   } else {
14890754003eSLois Curfman McInnes     /* Create and fill new matrix */
14907adad957SLisandro Dalcin     ierr = MatCreate(((PetscObject)A)->comm,&newmat);CHKERRQ(ierr);
1491f69a0ea3SMatthew Knepley     ierr = MatSetSizes(newmat,nrows,ncols,nrows,ncols);CHKERRQ(ierr);
14927adad957SLisandro Dalcin     ierr = MatSetType(newmat,((PetscObject)A)->type_name);CHKERRQ(ierr);
14935c5985e7SKris Buschelman     ierr = MatSeqDenseSetPreallocation(newmat,PETSC_NULL);CHKERRQ(ierr);
1494182d2002SSatish Balay   }
1495182d2002SSatish Balay 
1496182d2002SSatish Balay   /* Now extract the data pointers and do the copy,column at a time */
1497182d2002SSatish Balay   bv = ((Mat_SeqDense*)newmat->data)->v;
1498182d2002SSatish Balay 
1499182d2002SSatish Balay   for (i=0; i<ncols; i++) {
15006de62eeeSBarry Smith     av = v + mat->lda*icol[i];
1501182d2002SSatish Balay     for (j=0; j<nrows; j++) {
1502182d2002SSatish Balay       *bv++ = av[irow[j]];
15030754003eSLois Curfman McInnes     }
15040754003eSLois Curfman McInnes   }
1505182d2002SSatish Balay 
1506182d2002SSatish Balay   /* Assemble the matrices so that the correct flags are set */
15076d4a8577SBarry Smith   ierr = MatAssemblyBegin(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
15086d4a8577SBarry Smith   ierr = MatAssemblyEnd(newmat,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr);
15090754003eSLois Curfman McInnes 
15100754003eSLois Curfman McInnes   /* Free work space */
151178b31e54SBarry Smith   ierr = ISRestoreIndices(isrow,&irow);CHKERRQ(ierr);
151278b31e54SBarry Smith   ierr = ISRestoreIndices(iscol,&icol);CHKERRQ(ierr);
1513182d2002SSatish Balay   *B = newmat;
15143a40ed3dSBarry Smith   PetscFunctionReturn(0);
15150754003eSLois Curfman McInnes }
15160754003eSLois Curfman McInnes 
15174a2ae208SSatish Balay #undef __FUNCT__
15184a2ae208SSatish Balay #define __FUNCT__ "MatGetSubMatrices_SeqDense"
151913f74950SBarry Smith PetscErrorCode MatGetSubMatrices_SeqDense(Mat A,PetscInt n,const IS irow[],const IS icol[],MatReuse scall,Mat *B[])
1520905e6a2fSBarry Smith {
15216849ba73SBarry Smith   PetscErrorCode ierr;
152213f74950SBarry Smith   PetscInt       i;
1523905e6a2fSBarry Smith 
15243a40ed3dSBarry Smith   PetscFunctionBegin;
1525905e6a2fSBarry Smith   if (scall == MAT_INITIAL_MATRIX) {
1526b0a32e0cSBarry Smith     ierr = PetscMalloc((n+1)*sizeof(Mat),B);CHKERRQ(ierr);
1527905e6a2fSBarry Smith   }
1528905e6a2fSBarry Smith 
1529905e6a2fSBarry Smith   for (i=0; i<n; i++) {
15306a6a5d1dSBarry Smith     ierr = MatGetSubMatrix_SeqDense(A,irow[i],icol[i],PETSC_DECIDE,scall,&(*B)[i]);CHKERRQ(ierr);
1531905e6a2fSBarry Smith   }
15323a40ed3dSBarry Smith   PetscFunctionReturn(0);
1533905e6a2fSBarry Smith }
1534905e6a2fSBarry Smith 
15354a2ae208SSatish Balay #undef __FUNCT__
1536c0aa2d19SHong Zhang #define __FUNCT__ "MatAssemblyBegin_SeqDense"
1537c0aa2d19SHong Zhang PetscErrorCode MatAssemblyBegin_SeqDense(Mat mat,MatAssemblyType mode)
1538c0aa2d19SHong Zhang {
1539c0aa2d19SHong Zhang   PetscFunctionBegin;
1540c0aa2d19SHong Zhang   PetscFunctionReturn(0);
1541c0aa2d19SHong Zhang }
1542c0aa2d19SHong Zhang 
1543c0aa2d19SHong Zhang #undef __FUNCT__
1544c0aa2d19SHong Zhang #define __FUNCT__ "MatAssemblyEnd_SeqDense"
1545c0aa2d19SHong Zhang PetscErrorCode MatAssemblyEnd_SeqDense(Mat mat,MatAssemblyType mode)
1546c0aa2d19SHong Zhang {
1547c0aa2d19SHong Zhang   PetscFunctionBegin;
1548c0aa2d19SHong Zhang   PetscFunctionReturn(0);
1549c0aa2d19SHong Zhang }
1550c0aa2d19SHong Zhang 
1551c0aa2d19SHong Zhang #undef __FUNCT__
15524a2ae208SSatish Balay #define __FUNCT__ "MatCopy_SeqDense"
1553dfbe8321SBarry Smith PetscErrorCode MatCopy_SeqDense(Mat A,Mat B,MatStructure str)
15544b0e389bSBarry Smith {
15554b0e389bSBarry Smith   Mat_SeqDense   *a = (Mat_SeqDense*)A->data,*b = (Mat_SeqDense *)B->data;
15566849ba73SBarry Smith   PetscErrorCode ierr;
1557d0f46423SBarry Smith   PetscInt       lda1=a->lda,lda2=b->lda, m=A->rmap->n,n=A->cmap->n, j;
15583a40ed3dSBarry Smith 
15593a40ed3dSBarry Smith   PetscFunctionBegin;
156033f4a19fSKris Buschelman   /* If the two matrices don't have the same copy implementation, they aren't compatible for fast copy. */
156133f4a19fSKris Buschelman   if (A->ops->copy != B->ops->copy) {
1562cb5b572fSBarry Smith     ierr = MatCopy_Basic(A,B,str);CHKERRQ(ierr);
15633a40ed3dSBarry Smith     PetscFunctionReturn(0);
15643a40ed3dSBarry Smith   }
1565e32f2f54SBarry Smith   if (m != B->rmap->n || n != B->cmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"size(B) != size(A)");
1566a5ce6ee0Svictorle   if (lda1>m || lda2>m) {
15670dbb7854Svictorle     for (j=0; j<n; j++) {
1568a5ce6ee0Svictorle       ierr = PetscMemcpy(b->v+j*lda2,a->v+j*lda1,m*sizeof(PetscScalar));CHKERRQ(ierr);
1569a5ce6ee0Svictorle     }
1570a5ce6ee0Svictorle   } else {
1571d0f46423SBarry Smith     ierr = PetscMemcpy(b->v,a->v,A->rmap->n*A->cmap->n*sizeof(PetscScalar));CHKERRQ(ierr);
1572a5ce6ee0Svictorle   }
1573273d9f13SBarry Smith   PetscFunctionReturn(0);
1574273d9f13SBarry Smith }
1575273d9f13SBarry Smith 
15764a2ae208SSatish Balay #undef __FUNCT__
15774a2ae208SSatish Balay #define __FUNCT__ "MatSetUpPreallocation_SeqDense"
1578dfbe8321SBarry Smith PetscErrorCode MatSetUpPreallocation_SeqDense(Mat A)
1579273d9f13SBarry Smith {
1580dfbe8321SBarry Smith   PetscErrorCode ierr;
1581273d9f13SBarry Smith 
1582273d9f13SBarry Smith   PetscFunctionBegin;
1583273d9f13SBarry Smith   ierr =  MatSeqDenseSetPreallocation(A,0);CHKERRQ(ierr);
15843a40ed3dSBarry Smith   PetscFunctionReturn(0);
15854b0e389bSBarry Smith }
15864b0e389bSBarry Smith 
1587284134d9SBarry Smith #undef __FUNCT__
1588284134d9SBarry Smith #define __FUNCT__ "MatSetSizes_SeqDense"
1589284134d9SBarry Smith PetscErrorCode MatSetSizes_SeqDense(Mat A,PetscInt m,PetscInt n,PetscInt M,PetscInt N)
1590284134d9SBarry Smith {
1591284134d9SBarry Smith   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
1592284134d9SBarry Smith   PetscFunctionBegin;
159321a2c019SBarry Smith   /* this will not be called before lda, Mmax,  and Nmax have been set */
1594284134d9SBarry Smith   m = PetscMax(m,M);
1595284134d9SBarry Smith   n = PetscMax(n,N);
1596e32f2f54SBarry Smith   if (m > a->Mmax) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot yet resize number rows of dense matrix larger then its initial size %d, requested %d",a->lda,(int)m);
1597e32f2f54SBarry Smith   if (n > a->Nmax) SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Cannot yet resize number columns of dense matrix larger then its initial size %d, requested %d",a->Nmax,(int)n);
1598dc5cefdeSJed Brown   A->rmap->n = A->rmap->N = m;
1599d0f46423SBarry Smith   A->cmap->n = A->cmap->N = n;
160021a2c019SBarry Smith   if (a->changelda) a->lda = m;
1601284134d9SBarry Smith   PetscFunctionReturn(0);
1602284134d9SBarry Smith }
1603170fe5c8SBarry Smith 
1604ba337c44SJed Brown #undef __FUNCT__
1605ba337c44SJed Brown #define __FUNCT__ "MatConjugate_SeqDense"
1606ba337c44SJed Brown static PetscErrorCode MatConjugate_SeqDense(Mat A)
1607ba337c44SJed Brown {
1608ba337c44SJed Brown   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
1609ba337c44SJed Brown   PetscInt       i,nz = A->rmap->n*A->cmap->n;
1610ba337c44SJed Brown   PetscScalar    *aa = a->v;
1611ba337c44SJed Brown 
1612ba337c44SJed Brown   PetscFunctionBegin;
1613ba337c44SJed Brown   for (i=0; i<nz; i++) aa[i] = PetscConj(aa[i]);
1614ba337c44SJed Brown   PetscFunctionReturn(0);
1615ba337c44SJed Brown }
1616ba337c44SJed Brown 
1617ba337c44SJed Brown #undef __FUNCT__
1618ba337c44SJed Brown #define __FUNCT__ "MatRealPart_SeqDense"
1619ba337c44SJed Brown static PetscErrorCode MatRealPart_SeqDense(Mat A)
1620ba337c44SJed Brown {
1621ba337c44SJed Brown   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
1622ba337c44SJed Brown   PetscInt       i,nz = A->rmap->n*A->cmap->n;
1623ba337c44SJed Brown   PetscScalar    *aa = a->v;
1624ba337c44SJed Brown 
1625ba337c44SJed Brown   PetscFunctionBegin;
1626ba337c44SJed Brown   for (i=0; i<nz; i++) aa[i] = PetscRealPart(aa[i]);
1627ba337c44SJed Brown   PetscFunctionReturn(0);
1628ba337c44SJed Brown }
1629ba337c44SJed Brown 
1630ba337c44SJed Brown #undef __FUNCT__
1631ba337c44SJed Brown #define __FUNCT__ "MatImaginaryPart_SeqDense"
1632ba337c44SJed Brown static PetscErrorCode MatImaginaryPart_SeqDense(Mat A)
1633ba337c44SJed Brown {
1634ba337c44SJed Brown   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
1635ba337c44SJed Brown   PetscInt       i,nz = A->rmap->n*A->cmap->n;
1636ba337c44SJed Brown   PetscScalar    *aa = a->v;
1637ba337c44SJed Brown 
1638ba337c44SJed Brown   PetscFunctionBegin;
1639ba337c44SJed Brown   for (i=0; i<nz; i++) aa[i] = PetscImaginaryPart(aa[i]);
1640ba337c44SJed Brown   PetscFunctionReturn(0);
1641ba337c44SJed Brown }
1642284134d9SBarry Smith 
1643a9fe9ddaSSatish Balay /* ----------------------------------------------------------------*/
1644a9fe9ddaSSatish Balay #undef __FUNCT__
1645a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMult_SeqDense_SeqDense"
1646a9fe9ddaSSatish Balay PetscErrorCode MatMatMult_SeqDense_SeqDense(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C)
1647a9fe9ddaSSatish Balay {
1648a9fe9ddaSSatish Balay   PetscErrorCode ierr;
1649a9fe9ddaSSatish Balay 
1650a9fe9ddaSSatish Balay   PetscFunctionBegin;
1651a9fe9ddaSSatish Balay   if (scall == MAT_INITIAL_MATRIX){
1652a9fe9ddaSSatish Balay     ierr = MatMatMultSymbolic_SeqDense_SeqDense(A,B,fill,C);CHKERRQ(ierr);
1653a9fe9ddaSSatish Balay   }
1654a9fe9ddaSSatish Balay   ierr = MatMatMultNumeric_SeqDense_SeqDense(A,B,*C);CHKERRQ(ierr);
1655a9fe9ddaSSatish Balay   PetscFunctionReturn(0);
1656a9fe9ddaSSatish Balay }
1657a9fe9ddaSSatish Balay 
1658a9fe9ddaSSatish Balay #undef __FUNCT__
1659a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMultSymbolic_SeqDense_SeqDense"
1660a9fe9ddaSSatish Balay PetscErrorCode MatMatMultSymbolic_SeqDense_SeqDense(Mat A,Mat B,PetscReal fill,Mat *C)
1661a9fe9ddaSSatish Balay {
1662ee16a9a1SHong Zhang   PetscErrorCode ierr;
1663d0f46423SBarry Smith   PetscInt       m=A->rmap->n,n=B->cmap->n;
1664ee16a9a1SHong Zhang   Mat            Cmat;
1665a9fe9ddaSSatish Balay 
1666ee16a9a1SHong Zhang   PetscFunctionBegin;
1667e32f2f54SBarry 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);
1668ee16a9a1SHong Zhang   ierr = MatCreate(PETSC_COMM_SELF,&Cmat);CHKERRQ(ierr);
1669ee16a9a1SHong Zhang   ierr = MatSetSizes(Cmat,m,n,m,n);CHKERRQ(ierr);
1670ee16a9a1SHong Zhang   ierr = MatSetType(Cmat,MATSEQDENSE);CHKERRQ(ierr);
1671ee16a9a1SHong Zhang   ierr = MatSeqDenseSetPreallocation(Cmat,PETSC_NULL);CHKERRQ(ierr);
1672ee16a9a1SHong Zhang   Cmat->assembled = PETSC_TRUE;
1673ee16a9a1SHong Zhang   *C = Cmat;
1674ee16a9a1SHong Zhang   PetscFunctionReturn(0);
1675ee16a9a1SHong Zhang }
1676a9fe9ddaSSatish Balay 
167798a3b096SSatish Balay #undef __FUNCT__
1678a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMultNumeric_SeqDense_SeqDense"
1679a9fe9ddaSSatish Balay PetscErrorCode MatMatMultNumeric_SeqDense_SeqDense(Mat A,Mat B,Mat C)
1680a9fe9ddaSSatish Balay {
1681a9fe9ddaSSatish Balay   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
1682a9fe9ddaSSatish Balay   Mat_SeqDense   *b = (Mat_SeqDense*)B->data;
1683a9fe9ddaSSatish Balay   Mat_SeqDense   *c = (Mat_SeqDense*)C->data;
16840805154bSBarry Smith   PetscBLASInt   m,n,k;
1685a9fe9ddaSSatish Balay   PetscScalar    _DOne=1.0,_DZero=0.0;
1686a9fe9ddaSSatish Balay 
1687a9fe9ddaSSatish Balay   PetscFunctionBegin;
1688d0f46423SBarry Smith   m = PetscBLASIntCast(A->rmap->n);
1689d0f46423SBarry Smith   n = PetscBLASIntCast(B->cmap->n);
1690d0f46423SBarry Smith   k = PetscBLASIntCast(A->cmap->n);
1691a9fe9ddaSSatish Balay   BLASgemm_("N","N",&m,&n,&k,&_DOne,a->v,&a->lda,b->v,&b->lda,&_DZero,c->v,&c->lda);
1692a9fe9ddaSSatish Balay   PetscFunctionReturn(0);
1693a9fe9ddaSSatish Balay }
1694a9fe9ddaSSatish Balay 
1695a9fe9ddaSSatish Balay #undef __FUNCT__
1696a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMultTranspose_SeqDense_SeqDense"
1697a9fe9ddaSSatish Balay PetscErrorCode MatMatMultTranspose_SeqDense_SeqDense(Mat A,Mat B,MatReuse scall,PetscReal fill,Mat *C)
1698a9fe9ddaSSatish Balay {
1699a9fe9ddaSSatish Balay   PetscErrorCode ierr;
1700a9fe9ddaSSatish Balay 
1701a9fe9ddaSSatish Balay   PetscFunctionBegin;
1702a9fe9ddaSSatish Balay   if (scall == MAT_INITIAL_MATRIX){
1703a9fe9ddaSSatish Balay     ierr = MatMatMultTransposeSymbolic_SeqDense_SeqDense(A,B,fill,C);CHKERRQ(ierr);
1704a9fe9ddaSSatish Balay   }
1705a9fe9ddaSSatish Balay   ierr = MatMatMultTransposeNumeric_SeqDense_SeqDense(A,B,*C);CHKERRQ(ierr);
1706a9fe9ddaSSatish Balay   PetscFunctionReturn(0);
1707a9fe9ddaSSatish Balay }
1708a9fe9ddaSSatish Balay 
1709a9fe9ddaSSatish Balay #undef __FUNCT__
1710a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMultTransposeSymbolic_SeqDense_SeqDense"
1711a9fe9ddaSSatish Balay PetscErrorCode MatMatMultTransposeSymbolic_SeqDense_SeqDense(Mat A,Mat B,PetscReal fill,Mat *C)
1712a9fe9ddaSSatish Balay {
1713ee16a9a1SHong Zhang   PetscErrorCode ierr;
1714d0f46423SBarry Smith   PetscInt       m=A->cmap->n,n=B->cmap->n;
1715ee16a9a1SHong Zhang   Mat            Cmat;
1716a9fe9ddaSSatish Balay 
1717ee16a9a1SHong Zhang   PetscFunctionBegin;
1718e32f2f54SBarry 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);
1719ee16a9a1SHong Zhang   ierr = MatCreate(PETSC_COMM_SELF,&Cmat);CHKERRQ(ierr);
1720ee16a9a1SHong Zhang   ierr = MatSetSizes(Cmat,m,n,m,n);CHKERRQ(ierr);
1721ee16a9a1SHong Zhang   ierr = MatSetType(Cmat,MATSEQDENSE);CHKERRQ(ierr);
1722ee16a9a1SHong Zhang   ierr = MatSeqDenseSetPreallocation(Cmat,PETSC_NULL);CHKERRQ(ierr);
1723ee16a9a1SHong Zhang   Cmat->assembled = PETSC_TRUE;
1724ee16a9a1SHong Zhang   *C = Cmat;
1725ee16a9a1SHong Zhang   PetscFunctionReturn(0);
1726ee16a9a1SHong Zhang }
1727a9fe9ddaSSatish Balay 
1728a9fe9ddaSSatish Balay #undef __FUNCT__
1729a9fe9ddaSSatish Balay #define __FUNCT__ "MatMatMultTransposeNumeric_SeqDense_SeqDense"
1730a9fe9ddaSSatish Balay PetscErrorCode MatMatMultTransposeNumeric_SeqDense_SeqDense(Mat A,Mat B,Mat C)
1731a9fe9ddaSSatish Balay {
1732a9fe9ddaSSatish Balay   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
1733a9fe9ddaSSatish Balay   Mat_SeqDense   *b = (Mat_SeqDense*)B->data;
1734a9fe9ddaSSatish Balay   Mat_SeqDense   *c = (Mat_SeqDense*)C->data;
17350805154bSBarry Smith   PetscBLASInt   m,n,k;
1736a9fe9ddaSSatish Balay   PetscScalar    _DOne=1.0,_DZero=0.0;
1737a9fe9ddaSSatish Balay 
1738a9fe9ddaSSatish Balay   PetscFunctionBegin;
1739d0f46423SBarry Smith   m = PetscBLASIntCast(A->cmap->n);
1740d0f46423SBarry Smith   n = PetscBLASIntCast(B->cmap->n);
1741d0f46423SBarry Smith   k = PetscBLASIntCast(A->rmap->n);
17422fbe02b9SBarry Smith   /*
17432fbe02b9SBarry Smith      Note the m and n arguments below are the number rows and columns of A', not A!
17442fbe02b9SBarry Smith   */
1745a9fe9ddaSSatish Balay   BLASgemm_("T","N",&m,&n,&k,&_DOne,a->v,&a->lda,b->v,&b->lda,&_DZero,c->v,&c->lda);
1746a9fe9ddaSSatish Balay   PetscFunctionReturn(0);
1747a9fe9ddaSSatish Balay }
1748985db425SBarry Smith 
1749985db425SBarry Smith #undef __FUNCT__
1750985db425SBarry Smith #define __FUNCT__ "MatGetRowMax_SeqDense"
1751985db425SBarry Smith PetscErrorCode MatGetRowMax_SeqDense(Mat A,Vec v,PetscInt idx[])
1752985db425SBarry Smith {
1753985db425SBarry Smith   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
1754985db425SBarry Smith   PetscErrorCode ierr;
1755d0f46423SBarry Smith   PetscInt       i,j,m = A->rmap->n,n = A->cmap->n,p;
1756985db425SBarry Smith   PetscScalar    *x;
1757985db425SBarry Smith   MatScalar      *aa = a->v;
1758985db425SBarry Smith 
1759985db425SBarry Smith   PetscFunctionBegin;
1760e32f2f54SBarry Smith   if (A->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
1761985db425SBarry Smith 
1762985db425SBarry Smith   ierr = VecSet(v,0.0);CHKERRQ(ierr);
1763985db425SBarry Smith   ierr = VecGetArray(v,&x);CHKERRQ(ierr);
1764985db425SBarry Smith   ierr = VecGetLocalSize(v,&p);CHKERRQ(ierr);
1765e32f2f54SBarry Smith   if (p != A->rmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector");
1766985db425SBarry Smith   for (i=0; i<m; i++) {
1767985db425SBarry Smith     x[i] = aa[i]; if (idx) idx[i] = 0;
1768985db425SBarry Smith     for (j=1; j<n; j++){
1769985db425SBarry Smith       if (PetscRealPart(x[i]) < PetscRealPart(aa[i+m*j])) {x[i] = aa[i + m*j]; if (idx) idx[i] = j;}
1770985db425SBarry Smith     }
1771985db425SBarry Smith   }
1772985db425SBarry Smith   ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
1773985db425SBarry Smith   PetscFunctionReturn(0);
1774985db425SBarry Smith }
1775985db425SBarry Smith 
1776985db425SBarry Smith #undef __FUNCT__
1777985db425SBarry Smith #define __FUNCT__ "MatGetRowMaxAbs_SeqDense"
1778985db425SBarry Smith PetscErrorCode MatGetRowMaxAbs_SeqDense(Mat A,Vec v,PetscInt idx[])
1779985db425SBarry Smith {
1780985db425SBarry Smith   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
1781985db425SBarry Smith   PetscErrorCode ierr;
1782d0f46423SBarry Smith   PetscInt       i,j,m = A->rmap->n,n = A->cmap->n,p;
1783985db425SBarry Smith   PetscScalar    *x;
1784985db425SBarry Smith   PetscReal      atmp;
1785985db425SBarry Smith   MatScalar      *aa = a->v;
1786985db425SBarry Smith 
1787985db425SBarry Smith   PetscFunctionBegin;
1788e32f2f54SBarry Smith   if (A->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
1789985db425SBarry Smith 
1790985db425SBarry Smith   ierr = VecSet(v,0.0);CHKERRQ(ierr);
1791985db425SBarry Smith   ierr = VecGetArray(v,&x);CHKERRQ(ierr);
1792985db425SBarry Smith   ierr = VecGetLocalSize(v,&p);CHKERRQ(ierr);
1793e32f2f54SBarry Smith   if (p != A->rmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector");
1794985db425SBarry Smith   for (i=0; i<m; i++) {
17959189402eSHong Zhang     x[i] = PetscAbsScalar(aa[i]);
1796985db425SBarry Smith     for (j=1; j<n; j++){
1797985db425SBarry Smith       atmp = PetscAbsScalar(aa[i+m*j]);
1798985db425SBarry Smith       if (PetscAbsScalar(x[i]) < atmp) {x[i] = atmp; if (idx) idx[i] = j;}
1799985db425SBarry Smith     }
1800985db425SBarry Smith   }
1801985db425SBarry Smith   ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
1802985db425SBarry Smith   PetscFunctionReturn(0);
1803985db425SBarry Smith }
1804985db425SBarry Smith 
1805985db425SBarry Smith #undef __FUNCT__
1806985db425SBarry Smith #define __FUNCT__ "MatGetRowMin_SeqDense"
1807985db425SBarry Smith PetscErrorCode MatGetRowMin_SeqDense(Mat A,Vec v,PetscInt idx[])
1808985db425SBarry Smith {
1809985db425SBarry Smith   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
1810985db425SBarry Smith   PetscErrorCode ierr;
1811d0f46423SBarry Smith   PetscInt       i,j,m = A->rmap->n,n = A->cmap->n,p;
1812985db425SBarry Smith   PetscScalar    *x;
1813985db425SBarry Smith   MatScalar      *aa = a->v;
1814985db425SBarry Smith 
1815985db425SBarry Smith   PetscFunctionBegin;
1816e32f2f54SBarry Smith   if (A->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
1817985db425SBarry Smith 
1818985db425SBarry Smith   ierr = VecSet(v,0.0);CHKERRQ(ierr);
1819985db425SBarry Smith   ierr = VecGetArray(v,&x);CHKERRQ(ierr);
1820985db425SBarry Smith   ierr = VecGetLocalSize(v,&p);CHKERRQ(ierr);
1821e32f2f54SBarry Smith   if (p != A->rmap->n) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_SIZ,"Nonconforming matrix and vector");
1822985db425SBarry Smith   for (i=0; i<m; i++) {
1823985db425SBarry Smith     x[i] = aa[i]; if (idx) idx[i] = 0;
1824985db425SBarry Smith     for (j=1; j<n; j++){
1825985db425SBarry Smith       if (PetscRealPart(x[i]) > PetscRealPart(aa[i+m*j])) {x[i] = aa[i + m*j]; if (idx) idx[i] = j;}
1826985db425SBarry Smith     }
1827985db425SBarry Smith   }
1828985db425SBarry Smith   ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
1829985db425SBarry Smith   PetscFunctionReturn(0);
1830985db425SBarry Smith }
1831985db425SBarry Smith 
18328d0534beSBarry Smith #undef __FUNCT__
18338d0534beSBarry Smith #define __FUNCT__ "MatGetColumnVector_SeqDense"
18348d0534beSBarry Smith PetscErrorCode MatGetColumnVector_SeqDense(Mat A,Vec v,PetscInt col)
18358d0534beSBarry Smith {
18368d0534beSBarry Smith   Mat_SeqDense   *a = (Mat_SeqDense*)A->data;
18378d0534beSBarry Smith   PetscErrorCode ierr;
18388d0534beSBarry Smith   PetscScalar    *x;
18398d0534beSBarry Smith 
18408d0534beSBarry Smith   PetscFunctionBegin;
1841e32f2f54SBarry Smith   if (A->factortype) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONGSTATE,"Not for factored matrix");
18428d0534beSBarry Smith 
18438d0534beSBarry Smith   ierr = VecGetArray(v,&x);CHKERRQ(ierr);
1844d0f46423SBarry Smith   ierr = PetscMemcpy(x,a->v+col*a->lda,A->rmap->n*sizeof(PetscScalar));CHKERRQ(ierr);
18458d0534beSBarry Smith   ierr = VecRestoreArray(v,&x);CHKERRQ(ierr);
18468d0534beSBarry Smith   PetscFunctionReturn(0);
18478d0534beSBarry Smith }
18488d0534beSBarry Smith 
1849289bc588SBarry Smith /* -------------------------------------------------------------------*/
1850a5ae1ecdSBarry Smith static struct _MatOps MatOps_Values = {MatSetValues_SeqDense,
1851905e6a2fSBarry Smith        MatGetRow_SeqDense,
1852905e6a2fSBarry Smith        MatRestoreRow_SeqDense,
1853905e6a2fSBarry Smith        MatMult_SeqDense,
185497304618SKris Buschelman /* 4*/ MatMultAdd_SeqDense,
18557c922b88SBarry Smith        MatMultTranspose_SeqDense,
18567c922b88SBarry Smith        MatMultTransposeAdd_SeqDense,
1857db4efbfdSBarry Smith        0,
1858db4efbfdSBarry Smith        0,
1859db4efbfdSBarry Smith        0,
1860db4efbfdSBarry Smith /*10*/ 0,
1861905e6a2fSBarry Smith        MatLUFactor_SeqDense,
1862905e6a2fSBarry Smith        MatCholeskyFactor_SeqDense,
186341f059aeSBarry Smith        MatSOR_SeqDense,
1864ec8511deSBarry Smith        MatTranspose_SeqDense,
186597304618SKris Buschelman /*15*/ MatGetInfo_SeqDense,
1866905e6a2fSBarry Smith        MatEqual_SeqDense,
1867905e6a2fSBarry Smith        MatGetDiagonal_SeqDense,
1868905e6a2fSBarry Smith        MatDiagonalScale_SeqDense,
1869905e6a2fSBarry Smith        MatNorm_SeqDense,
1870c0aa2d19SHong Zhang /*20*/ MatAssemblyBegin_SeqDense,
1871c0aa2d19SHong Zhang        MatAssemblyEnd_SeqDense,
1872905e6a2fSBarry Smith        MatSetOption_SeqDense,
1873905e6a2fSBarry Smith        MatZeroEntries_SeqDense,
1874d519adbfSMatthew Knepley /*24*/ MatZeroRows_SeqDense,
1875db4efbfdSBarry Smith        0,
1876db4efbfdSBarry Smith        0,
1877db4efbfdSBarry Smith        0,
1878db4efbfdSBarry Smith        0,
1879d519adbfSMatthew Knepley /*29*/ MatSetUpPreallocation_SeqDense,
1880273d9f13SBarry Smith        0,
1881905e6a2fSBarry Smith        0,
1882905e6a2fSBarry Smith        MatGetArray_SeqDense,
1883905e6a2fSBarry Smith        MatRestoreArray_SeqDense,
1884d519adbfSMatthew Knepley /*34*/ MatDuplicate_SeqDense,
1885a5ae1ecdSBarry Smith        0,
1886a5ae1ecdSBarry Smith        0,
1887a5ae1ecdSBarry Smith        0,
1888a5ae1ecdSBarry Smith        0,
1889d519adbfSMatthew Knepley /*39*/ MatAXPY_SeqDense,
1890a5ae1ecdSBarry Smith        MatGetSubMatrices_SeqDense,
1891a5ae1ecdSBarry Smith        0,
18924b0e389bSBarry Smith        MatGetValues_SeqDense,
1893a5ae1ecdSBarry Smith        MatCopy_SeqDense,
1894d519adbfSMatthew Knepley /*44*/ MatGetRowMax_SeqDense,
1895a5ae1ecdSBarry Smith        MatScale_SeqDense,
1896a5ae1ecdSBarry Smith        0,
1897a5ae1ecdSBarry Smith        0,
1898a5ae1ecdSBarry Smith        0,
1899d519adbfSMatthew Knepley /*49*/ 0,
1900a5ae1ecdSBarry Smith        0,
1901a5ae1ecdSBarry Smith        0,
1902a5ae1ecdSBarry Smith        0,
1903a5ae1ecdSBarry Smith        0,
1904d519adbfSMatthew Knepley /*54*/ 0,
1905a5ae1ecdSBarry Smith        0,
1906a5ae1ecdSBarry Smith        0,
1907a5ae1ecdSBarry Smith        0,
1908a5ae1ecdSBarry Smith        0,
1909d519adbfSMatthew Knepley /*59*/ 0,
1910e03a110bSBarry Smith        MatDestroy_SeqDense,
1911e03a110bSBarry Smith        MatView_SeqDense,
1912357abbc8SBarry Smith        0,
191397304618SKris Buschelman        0,
1914d519adbfSMatthew Knepley /*64*/ 0,
191597304618SKris Buschelman        0,
191697304618SKris Buschelman        0,
191797304618SKris Buschelman        0,
191897304618SKris Buschelman        0,
1919d519adbfSMatthew Knepley /*69*/ MatGetRowMaxAbs_SeqDense,
192097304618SKris Buschelman        0,
192197304618SKris Buschelman        0,
192297304618SKris Buschelman        0,
192397304618SKris Buschelman        0,
1924d519adbfSMatthew Knepley /*74*/ 0,
192597304618SKris Buschelman        0,
192697304618SKris Buschelman        0,
192797304618SKris Buschelman        0,
192897304618SKris Buschelman        0,
1929d519adbfSMatthew Knepley /*79*/ 0,
193097304618SKris Buschelman        0,
193197304618SKris Buschelman        0,
193297304618SKris Buschelman        0,
1933d519adbfSMatthew Knepley /*83*/ MatLoad_SeqDense,
1934865e5f61SKris Buschelman        0,
19351cbb95d3SBarry Smith        MatIsHermitian_SeqDense,
1936865e5f61SKris Buschelman        0,
1937865e5f61SKris Buschelman        0,
1938865e5f61SKris Buschelman        0,
1939d519adbfSMatthew Knepley /*89*/ MatMatMult_SeqDense_SeqDense,
1940a9fe9ddaSSatish Balay        MatMatMultSymbolic_SeqDense_SeqDense,
1941a9fe9ddaSSatish Balay        MatMatMultNumeric_SeqDense_SeqDense,
1942865e5f61SKris Buschelman        0,
1943865e5f61SKris Buschelman        0,
1944d519adbfSMatthew Knepley /*94*/ 0,
1945a9fe9ddaSSatish Balay        MatMatMultTranspose_SeqDense_SeqDense,
1946a9fe9ddaSSatish Balay        MatMatMultTransposeSymbolic_SeqDense_SeqDense,
1947a9fe9ddaSSatish Balay        MatMatMultTransposeNumeric_SeqDense_SeqDense,
1948284134d9SBarry Smith        0,
1949d519adbfSMatthew Knepley /*99*/ 0,
1950284134d9SBarry Smith        0,
1951284134d9SBarry Smith        0,
1952ba337c44SJed Brown        MatConjugate_SeqDense,
1953985db425SBarry Smith        MatSetSizes_SeqDense,
1954ba337c44SJed Brown /*104*/0,
1955ba337c44SJed Brown        MatRealPart_SeqDense,
1956ba337c44SJed Brown        MatImaginaryPart_SeqDense,
1957985db425SBarry Smith        0,
1958985db425SBarry Smith        0,
1959d519adbfSMatthew Knepley /*109*/0,
1960985db425SBarry Smith        0,
19618d0534beSBarry Smith        MatGetRowMin_SeqDense,
1962*aabbc4fbSShri Abhyankar        MatGetColumnVector_SeqDense,
1963*aabbc4fbSShri Abhyankar        0,
1964*aabbc4fbSShri Abhyankar /*114*/0,
1965*aabbc4fbSShri Abhyankar        0,
1966*aabbc4fbSShri Abhyankar        0,
1967*aabbc4fbSShri Abhyankar        0,
1968*aabbc4fbSShri Abhyankar        0,
1969*aabbc4fbSShri Abhyankar /*119*/0,
1970*aabbc4fbSShri Abhyankar        0,
1971*aabbc4fbSShri Abhyankar        0,
1972*aabbc4fbSShri Abhyankar        0,
1973*aabbc4fbSShri Abhyankar        MatLoadnew_SeqDense
1974985db425SBarry Smith };
197590ace30eSBarry Smith 
19764a2ae208SSatish Balay #undef __FUNCT__
19774a2ae208SSatish Balay #define __FUNCT__ "MatCreateSeqDense"
19784b828684SBarry Smith /*@C
1979fafbff53SBarry Smith    MatCreateSeqDense - Creates a sequential dense matrix that
1980d65003e9SLois Curfman McInnes    is stored in column major order (the usual Fortran 77 manner). Many
1981d65003e9SLois Curfman McInnes    of the matrix operations use the BLAS and LAPACK routines.
1982289bc588SBarry Smith 
1983db81eaa0SLois Curfman McInnes    Collective on MPI_Comm
1984db81eaa0SLois Curfman McInnes 
198520563c6bSBarry Smith    Input Parameters:
1986db81eaa0SLois Curfman McInnes +  comm - MPI communicator, set to PETSC_COMM_SELF
19870c775827SLois Curfman McInnes .  m - number of rows
198818f449edSLois Curfman McInnes .  n - number of columns
1989c0235b3cSMatthew Knepley -  data - optional location of matrix data in column major order.  Set data=PETSC_NULL for PETSc
1990dfc5480cSLois Curfman McInnes    to control all matrix memory allocation.
199120563c6bSBarry Smith 
199220563c6bSBarry Smith    Output Parameter:
199344cd7ae7SLois Curfman McInnes .  A - the matrix
199420563c6bSBarry Smith 
1995b259b22eSLois Curfman McInnes    Notes:
199618f449edSLois Curfman McInnes    The data input variable is intended primarily for Fortran programmers
199718f449edSLois Curfman McInnes    who wish to allocate their own matrix memory space.  Most users should
1998b4fd4287SBarry Smith    set data=PETSC_NULL.
199918f449edSLois Curfman McInnes 
2000027ccd11SLois Curfman McInnes    Level: intermediate
2001027ccd11SLois Curfman McInnes 
2002dbd7a890SLois Curfman McInnes .keywords: dense, matrix, LAPACK, BLAS
2003d65003e9SLois Curfman McInnes 
2004db81eaa0SLois Curfman McInnes .seealso: MatCreate(), MatCreateMPIDense(), MatSetValues()
200520563c6bSBarry Smith @*/
2006be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreateSeqDense(MPI_Comm comm,PetscInt m,PetscInt n,PetscScalar *data,Mat *A)
2007289bc588SBarry Smith {
2008dfbe8321SBarry Smith   PetscErrorCode ierr;
20093b2fbd54SBarry Smith 
20103a40ed3dSBarry Smith   PetscFunctionBegin;
2011f69a0ea3SMatthew Knepley   ierr = MatCreate(comm,A);CHKERRQ(ierr);
2012f69a0ea3SMatthew Knepley   ierr = MatSetSizes(*A,m,n,m,n);CHKERRQ(ierr);
2013273d9f13SBarry Smith   ierr = MatSetType(*A,MATSEQDENSE);CHKERRQ(ierr);
2014273d9f13SBarry Smith   ierr = MatSeqDenseSetPreallocation(*A,data);CHKERRQ(ierr);
2015273d9f13SBarry Smith   PetscFunctionReturn(0);
2016273d9f13SBarry Smith }
2017273d9f13SBarry Smith 
20184a2ae208SSatish Balay #undef __FUNCT__
2019afc30d2aSLisandro Dalcin #define __FUNCT__ "MatSeqDenseSetPreallocation"
2020273d9f13SBarry Smith /*@C
2021273d9f13SBarry Smith    MatSeqDenseSetPreallocation - Sets the array used for storing the matrix elements
2022273d9f13SBarry Smith 
2023273d9f13SBarry Smith    Collective on MPI_Comm
2024273d9f13SBarry Smith 
2025273d9f13SBarry Smith    Input Parameters:
2026273d9f13SBarry Smith +  A - the matrix
2027273d9f13SBarry Smith -  data - the array (or PETSC_NULL)
2028273d9f13SBarry Smith 
2029273d9f13SBarry Smith    Notes:
2030273d9f13SBarry Smith    The data input variable is intended primarily for Fortran programmers
2031273d9f13SBarry Smith    who wish to allocate their own matrix memory space.  Most users should
2032284134d9SBarry Smith    need not call this routine.
2033273d9f13SBarry Smith 
2034273d9f13SBarry Smith    Level: intermediate
2035273d9f13SBarry Smith 
2036273d9f13SBarry Smith .keywords: dense, matrix, LAPACK, BLAS
2037273d9f13SBarry Smith 
2038867c911aSBarry Smith .seealso: MatCreate(), MatCreateMPIDense(), MatSetValues(), MatSeqDenseSetLDA()
2039867c911aSBarry Smith 
2040273d9f13SBarry Smith @*/
2041be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatSeqDenseSetPreallocation(Mat B,PetscScalar data[])
2042273d9f13SBarry Smith {
2043dfbe8321SBarry Smith   PetscErrorCode ierr,(*f)(Mat,PetscScalar[]);
2044a23d5eceSKris Buschelman 
2045a23d5eceSKris Buschelman   PetscFunctionBegin;
2046a23d5eceSKris Buschelman   ierr = PetscObjectQueryFunction((PetscObject)B,"MatSeqDenseSetPreallocation_C",(void (**)(void))&f);CHKERRQ(ierr);
2047a23d5eceSKris Buschelman   if (f) {
2048a23d5eceSKris Buschelman     ierr = (*f)(B,data);CHKERRQ(ierr);
2049a23d5eceSKris Buschelman   }
2050a23d5eceSKris Buschelman   PetscFunctionReturn(0);
2051a23d5eceSKris Buschelman }
2052a23d5eceSKris Buschelman 
2053a23d5eceSKris Buschelman EXTERN_C_BEGIN
2054a23d5eceSKris Buschelman #undef __FUNCT__
2055afc30d2aSLisandro Dalcin #define __FUNCT__ "MatSeqDenseSetPreallocation_SeqDense"
2056be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatSeqDenseSetPreallocation_SeqDense(Mat B,PetscScalar *data)
2057a23d5eceSKris Buschelman {
2058273d9f13SBarry Smith   Mat_SeqDense   *b;
2059dfbe8321SBarry Smith   PetscErrorCode ierr;
2060273d9f13SBarry Smith 
2061273d9f13SBarry Smith   PetscFunctionBegin;
2062273d9f13SBarry Smith   B->preallocated = PETSC_TRUE;
2063273d9f13SBarry Smith   b               = (Mat_SeqDense*)B->data;
2064d0f46423SBarry Smith   if (b->lda <= 0) b->lda = B->rmap->n;
20659e8f95c4SLisandro Dalcin   if (!data) { /* petsc-allocated storage */
20669e8f95c4SLisandro Dalcin     if (!b->user_alloc) { ierr = PetscFree(b->v);CHKERRQ(ierr); }
20675afd5e0cSBarry Smith     ierr = PetscMalloc(b->lda*b->Nmax*sizeof(PetscScalar),&b->v);CHKERRQ(ierr);
2068284134d9SBarry Smith     ierr = PetscMemzero(b->v,b->lda*b->Nmax*sizeof(PetscScalar));CHKERRQ(ierr);
2069284134d9SBarry Smith     ierr = PetscLogObjectMemory(B,b->lda*b->Nmax*sizeof(PetscScalar));CHKERRQ(ierr);
20709e8f95c4SLisandro Dalcin     b->user_alloc = PETSC_FALSE;
2071273d9f13SBarry Smith   } else { /* user-allocated storage */
20729e8f95c4SLisandro Dalcin     if (!b->user_alloc) { ierr = PetscFree(b->v);CHKERRQ(ierr); }
2073273d9f13SBarry Smith     b->v          = data;
2074273d9f13SBarry Smith     b->user_alloc = PETSC_TRUE;
2075273d9f13SBarry Smith   }
20760450473dSBarry Smith   B->assembled = PETSC_TRUE;
2077273d9f13SBarry Smith   PetscFunctionReturn(0);
2078273d9f13SBarry Smith }
2079a23d5eceSKris Buschelman EXTERN_C_END
2080273d9f13SBarry Smith 
20811b807ce4Svictorle #undef __FUNCT__
20821b807ce4Svictorle #define __FUNCT__ "MatSeqDenseSetLDA"
20831b807ce4Svictorle /*@C
20841b807ce4Svictorle   MatSeqDenseSetLDA - Declare the leading dimension of the user-provided array
20851b807ce4Svictorle 
20861b807ce4Svictorle   Input parameter:
20871b807ce4Svictorle + A - the matrix
20881b807ce4Svictorle - lda - the leading dimension
20891b807ce4Svictorle 
20901b807ce4Svictorle   Notes:
2091867c911aSBarry Smith   This routine is to be used in conjunction with MatSeqDenseSetPreallocation();
20921b807ce4Svictorle   it asserts that the preallocation has a leading dimension (the LDA parameter
20931b807ce4Svictorle   of Blas and Lapack fame) larger than M, the first dimension of the matrix.
20941b807ce4Svictorle 
20951b807ce4Svictorle   Level: intermediate
20961b807ce4Svictorle 
20971b807ce4Svictorle .keywords: dense, matrix, LAPACK, BLAS
20981b807ce4Svictorle 
2099284134d9SBarry Smith .seealso: MatCreate(), MatCreateSeqDense(), MatSeqDenseSetPreallocation(), MatSetMaximumSize()
2100867c911aSBarry Smith 
21011b807ce4Svictorle @*/
2102be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatSeqDenseSetLDA(Mat B,PetscInt lda)
21031b807ce4Svictorle {
21041b807ce4Svictorle   Mat_SeqDense *b = (Mat_SeqDense*)B->data;
210521a2c019SBarry Smith 
21061b807ce4Svictorle   PetscFunctionBegin;
2107e32f2f54SBarry 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);
21081b807ce4Svictorle   b->lda       = lda;
210921a2c019SBarry Smith   b->changelda = PETSC_FALSE;
211021a2c019SBarry Smith   b->Mmax      = PetscMax(b->Mmax,lda);
21111b807ce4Svictorle   PetscFunctionReturn(0);
21121b807ce4Svictorle }
21131b807ce4Svictorle 
21140bad9183SKris Buschelman /*MC
2115fafad747SKris Buschelman    MATSEQDENSE - MATSEQDENSE = "seqdense" - A matrix type to be used for sequential dense matrices.
21160bad9183SKris Buschelman 
21170bad9183SKris Buschelman    Options Database Keys:
21180bad9183SKris Buschelman . -mat_type seqdense - sets the matrix type to "seqdense" during a call to MatSetFromOptions()
21190bad9183SKris Buschelman 
21200bad9183SKris Buschelman   Level: beginner
21210bad9183SKris Buschelman 
212289665df3SBarry Smith .seealso: MatCreateSeqDense()
212389665df3SBarry Smith 
21240bad9183SKris Buschelman M*/
21250bad9183SKris Buschelman 
2126273d9f13SBarry Smith EXTERN_C_BEGIN
21274a2ae208SSatish Balay #undef __FUNCT__
21284a2ae208SSatish Balay #define __FUNCT__ "MatCreate_SeqDense"
2129be1d678aSKris Buschelman PetscErrorCode PETSCMAT_DLLEXPORT MatCreate_SeqDense(Mat B)
2130273d9f13SBarry Smith {
2131273d9f13SBarry Smith   Mat_SeqDense   *b;
2132dfbe8321SBarry Smith   PetscErrorCode ierr;
21337c334f02SBarry Smith   PetscMPIInt    size;
2134273d9f13SBarry Smith 
2135273d9f13SBarry Smith   PetscFunctionBegin;
21367adad957SLisandro Dalcin   ierr = MPI_Comm_size(((PetscObject)B)->comm,&size);CHKERRQ(ierr);
2137e32f2f54SBarry Smith   if (size > 1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_ARG_WRONG,"Comm must be of size 1");
213855659b69SBarry Smith 
213926283091SBarry Smith   ierr = PetscLayoutSetBlockSize(B->rmap,1);CHKERRQ(ierr);
214026283091SBarry Smith   ierr = PetscLayoutSetBlockSize(B->cmap,1);CHKERRQ(ierr);
214126283091SBarry Smith   ierr = PetscLayoutSetUp(B->rmap);CHKERRQ(ierr);
214226283091SBarry Smith   ierr = PetscLayoutSetUp(B->cmap);CHKERRQ(ierr);
2143273d9f13SBarry Smith 
214438f2d2fdSLisandro Dalcin   ierr            = PetscNewLog(B,Mat_SeqDense,&b);CHKERRQ(ierr);
2145549d3d68SSatish Balay   ierr            = PetscMemcpy(B->ops,&MatOps_Values,sizeof(struct _MatOps));CHKERRQ(ierr);
214690f02eecSBarry Smith   B->mapping      = 0;
214744cd7ae7SLois Curfman McInnes   B->data         = (void*)b;
214818f449edSLois Curfman McInnes 
2149a5ae1ecdSBarry Smith 
215044cd7ae7SLois Curfman McInnes   b->pivots       = 0;
2151273d9f13SBarry Smith   b->roworiented  = PETSC_TRUE;
2152273d9f13SBarry Smith   b->v            = 0;
2153d0f46423SBarry Smith   b->lda          = B->rmap->n;
215421a2c019SBarry Smith   b->changelda    = PETSC_FALSE;
2155d0f46423SBarry Smith   b->Mmax         = B->rmap->n;
2156d0f46423SBarry Smith   b->Nmax         = B->cmap->n;
21574e220ebcSLois Curfman McInnes 
2158b24902e0SBarry Smith 
2159ec1065edSBarry Smith   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatGetFactor_petsc_C",
2160b24902e0SBarry Smith                                      "MatGetFactor_seqdense_petsc",
2161b24902e0SBarry Smith                                       MatGetFactor_seqdense_petsc);CHKERRQ(ierr);
2162a23d5eceSKris Buschelman   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatSeqDenseSetPreallocation_C",
2163a23d5eceSKris Buschelman                                     "MatSeqDenseSetPreallocation_SeqDense",
2164a23d5eceSKris Buschelman                                      MatSeqDenseSetPreallocation_SeqDense);CHKERRQ(ierr);
21654ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatMatMult_seqaij_seqdense_C",
21664ae313f4SHong Zhang                                      "MatMatMult_SeqAIJ_SeqDense",
21674ae313f4SHong Zhang                                       MatMatMult_SeqAIJ_SeqDense);CHKERRQ(ierr);
21684ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatMatMultSymbolic_seqaij_seqdense_C",
21694ae313f4SHong Zhang                                      "MatMatMultSymbolic_SeqAIJ_SeqDense",
21704ae313f4SHong Zhang                                       MatMatMultSymbolic_SeqAIJ_SeqDense);CHKERRQ(ierr);
21714ae313f4SHong Zhang   ierr = PetscObjectComposeFunctionDynamic((PetscObject)B,"MatMatMultNumeric_seqaij_seqdense_C",
21724ae313f4SHong Zhang                                      "MatMatMultNumeric_SeqAIJ_SeqDense",
21734ae313f4SHong Zhang                                       MatMatMultNumeric_SeqAIJ_SeqDense);CHKERRQ(ierr);
217417667f90SBarry Smith   ierr = PetscObjectChangeTypeName((PetscObject)B,MATSEQDENSE);CHKERRQ(ierr);
21753a40ed3dSBarry Smith   PetscFunctionReturn(0);
2176289bc588SBarry Smith }
2177c0aa2d19SHong Zhang 
2178c0aa2d19SHong Zhang 
2179273d9f13SBarry Smith EXTERN_C_END
2180