xref: /petsc/src/mat/impls/dense/seq/dense.c (revision 6f75cfa4e1f97e2176ba4dac097d9fdaf232f5b9)
1cb512458SBarry Smith #ifndef lint
2*6f75cfa4SBarry Smith static char vcid[] = "$Id: dense.c,v 1.27 1995/04/27 20:15:27 curfman Exp bsmith $";
3cb512458SBarry Smith #endif
4289bc588SBarry Smith 
5289bc588SBarry Smith /*
6289bc588SBarry Smith     Standard Fortran style matrices
7289bc588SBarry Smith */
8289bc588SBarry Smith #include "ptscimpl.h"
9289bc588SBarry Smith #include "plapack.h"
10289bc588SBarry Smith #include "matimpl.h"
11289bc588SBarry Smith #include "math.h"
12289bc588SBarry Smith #include "vec/vecimpl.h"
13289bc588SBarry Smith 
14289bc588SBarry Smith typedef struct {
15289bc588SBarry Smith   Scalar *v;
16289bc588SBarry Smith   int    roworiented;
17289bc588SBarry Smith   int    m,n,pad;
18289bc588SBarry Smith   int    *pivots;   /* pivots in LU factorization */
1944a69424SLois Curfman McInnes } Mat_Dense;
20289bc588SBarry Smith 
21289bc588SBarry Smith 
22fa9ec3f1SLois Curfman McInnes static int MatGetInfo_Dense(Mat matin,int flag,int *nz,int *nzalloc,int *mem)
23289bc588SBarry Smith {
2444a69424SLois Curfman McInnes   Mat_Dense *mat = (Mat_Dense *) matin->data;
25289bc588SBarry Smith   int    i,N = mat->m*mat->n,count = 0;
26289bc588SBarry Smith   Scalar *v = mat->v;
27289bc588SBarry Smith   for ( i=0; i<N; i++ ) {if (*v != 0.0) count++; v++;}
28fa9ec3f1SLois Curfman McInnes   *nz = count; *nzalloc = N; *mem = N*sizeof(Scalar);
29fa9ec3f1SLois Curfman McInnes   return 0;
30289bc588SBarry Smith }
31289bc588SBarry Smith 
32289bc588SBarry Smith /* ---------------------------------------------------------------*/
33289bc588SBarry Smith /* COMMENT: I have chosen to hide column permutation in the pivots,
34289bc588SBarry Smith    rather than put it in the Mat->col slot.*/
3544a69424SLois Curfman McInnes static int MatLUFactor_Dense(Mat matin,IS row,IS col)
36289bc588SBarry Smith {
3744a69424SLois Curfman McInnes   Mat_Dense *mat = (Mat_Dense *) matin->data;
386abc6512SBarry Smith   int    info;
39289bc588SBarry Smith   if (!mat->pivots) {
40289bc588SBarry Smith     mat->pivots = (int *) MALLOC( mat->m*sizeof(int) );
41289bc588SBarry Smith     CHKPTR(mat->pivots);
42289bc588SBarry Smith   }
43289bc588SBarry Smith   LAgetrf_(&mat->m,&mat->n,mat->v,&mat->m,mat->pivots,&info);
44289bc588SBarry Smith   if (info) SETERR(1,"Bad LU factorization");
45289bc588SBarry Smith   matin->factor = FACTOR_LU;
46289bc588SBarry Smith   return 0;
47289bc588SBarry Smith }
4844a69424SLois Curfman McInnes static int MatLUFactorSymbolic_Dense(Mat matin,IS row,IS col,Mat *fact)
49289bc588SBarry Smith {
50289bc588SBarry Smith   int ierr;
516abc6512SBarry Smith   if ((ierr = MatCopy(matin,fact))) SETERR(ierr,0);
52289bc588SBarry Smith   return 0;
53289bc588SBarry Smith }
5444a69424SLois Curfman McInnes static int MatLUFactorNumeric_Dense(Mat matin,Mat *fact)
55289bc588SBarry Smith {
5620563c6bSBarry Smith   return MatLUFactor(*fact,0,0);
57289bc588SBarry Smith }
5844a69424SLois Curfman McInnes static int MatChFactorSymbolic_Dense(Mat matin,IS row,Mat *fact)
59289bc588SBarry Smith {
60289bc588SBarry Smith   int ierr;
616abc6512SBarry Smith   if ((ierr = MatCopy(matin,fact))) SETERR(ierr,0);
62289bc588SBarry Smith   return 0;
63289bc588SBarry Smith }
6444a69424SLois Curfman McInnes static int MatChFactorNumeric_Dense(Mat matin,Mat *fact)
65289bc588SBarry Smith {
6620563c6bSBarry Smith   return MatCholeskyFactor(*fact,0);
67289bc588SBarry Smith }
6844a69424SLois Curfman McInnes static int MatChFactor_Dense(Mat matin,IS perm)
69289bc588SBarry Smith {
7044a69424SLois Curfman McInnes   Mat_Dense    *mat = (Mat_Dense *) matin->data;
716abc6512SBarry Smith   int       info;
72289bc588SBarry Smith   if (mat->pivots) {FREE(mat->pivots); mat->pivots = 0;}
73289bc588SBarry Smith   LApotrf_("L",&mat->n,mat->v,&mat->m,&info);
74289bc588SBarry Smith   if (info) SETERR(1,"Bad Cholesky factorization");
75289bc588SBarry Smith   matin->factor = FACTOR_CHOLESKY;
76289bc588SBarry Smith   return 0;
77289bc588SBarry Smith }
78289bc588SBarry Smith 
7944a69424SLois Curfman McInnes static int MatSolve_Dense(Mat matin,Vec xx,Vec yy)
80289bc588SBarry Smith {
8144a69424SLois Curfman McInnes   Mat_Dense *mat = (Mat_Dense *) matin->data;
826abc6512SBarry Smith   int    one = 1, info;
836abc6512SBarry Smith   Scalar *x, *y;
84289bc588SBarry Smith   VecGetArray(xx,&x); VecGetArray(yy,&y);
85289bc588SBarry Smith   MEMCPY(y,x,mat->m*sizeof(Scalar));
869e25ed09SBarry Smith   if (matin->factor == FACTOR_LU) {
87289bc588SBarry Smith     LAgetrs_( "N", &mat->m, &one, mat->v, &mat->m, mat->pivots,
88289bc588SBarry Smith               y, &mat->m, &info );
89289bc588SBarry Smith   }
909e25ed09SBarry Smith   else if (matin->factor == FACTOR_CHOLESKY){
91289bc588SBarry Smith     LApotrs_( "L", &mat->m, &one, mat->v, &mat->m,
92289bc588SBarry Smith               y, &mat->m, &info );
93289bc588SBarry Smith   }
949e25ed09SBarry Smith   else SETERR(1,"Matrix must be factored to solve");
95289bc588SBarry Smith   if (info) SETERR(1,"Bad solve");
96289bc588SBarry Smith   return 0;
97289bc588SBarry Smith }
9844a69424SLois Curfman McInnes static int MatSolveTrans_Dense(Mat matin,Vec xx,Vec yy)
99da3a660dSBarry Smith {
10044a69424SLois Curfman McInnes   Mat_Dense *mat = (Mat_Dense *) matin->data;
1016abc6512SBarry Smith   int    one = 1, info;
1026abc6512SBarry Smith   Scalar *x, *y;
103da3a660dSBarry Smith   VecGetArray(xx,&x); VecGetArray(yy,&y);
104da3a660dSBarry Smith   MEMCPY(y,x,mat->m*sizeof(Scalar));
105da3a660dSBarry Smith   /* assume if pivots exist then LU else Cholesky */
106da3a660dSBarry Smith   if (mat->pivots) {
107da3a660dSBarry Smith     LAgetrs_( "T", &mat->m, &one, mat->v, &mat->m, mat->pivots,
108da3a660dSBarry Smith               y, &mat->m, &info );
109da3a660dSBarry Smith   }
110da3a660dSBarry Smith   else {
111da3a660dSBarry Smith     LApotrs_( "L", &mat->m, &one, mat->v, &mat->m,
112da3a660dSBarry Smith               y, &mat->m, &info );
113da3a660dSBarry Smith   }
114da3a660dSBarry Smith   if (info) SETERR(1,"Bad solve");
115da3a660dSBarry Smith   return 0;
116da3a660dSBarry Smith }
11744a69424SLois Curfman McInnes static int MatSolveAdd_Dense(Mat matin,Vec xx,Vec zz,Vec yy)
118da3a660dSBarry Smith {
11944a69424SLois Curfman McInnes   Mat_Dense *mat = (Mat_Dense *) matin->data;
1206abc6512SBarry Smith   int    one = 1, info,ierr;
1216abc6512SBarry Smith   Scalar *x, *y, sone = 1.0;
122da3a660dSBarry Smith   Vec    tmp = 0;
123da3a660dSBarry Smith   VecGetArray(xx,&x); VecGetArray(yy,&y);
124da3a660dSBarry Smith   if (yy == zz) {
125da3a660dSBarry Smith     ierr = VecCreate(yy,&tmp); CHKERR(ierr);
126da3a660dSBarry Smith     ierr = VecCopy(yy,tmp); CHKERR(ierr);
127da3a660dSBarry Smith   }
128da3a660dSBarry Smith   MEMCPY(y,x,mat->m*sizeof(Scalar));
129da3a660dSBarry Smith   /* assume if pivots exist then LU else Cholesky */
130da3a660dSBarry Smith   if (mat->pivots) {
131da3a660dSBarry Smith     LAgetrs_( "N", &mat->m, &one, mat->v, &mat->m, mat->pivots,
132da3a660dSBarry Smith               y, &mat->m, &info );
133da3a660dSBarry Smith   }
134da3a660dSBarry Smith   else {
135da3a660dSBarry Smith     LApotrs_( "L", &mat->m, &one, mat->v, &mat->m,
136da3a660dSBarry Smith               y, &mat->m, &info );
137da3a660dSBarry Smith   }
138da3a660dSBarry Smith   if (info) SETERR(1,"Bad solve");
139da3a660dSBarry Smith   if (tmp) {VecAXPY(&sone,tmp,yy); VecDestroy(tmp);}
140da3a660dSBarry Smith   else VecAXPY(&sone,zz,yy);
141da3a660dSBarry Smith   return 0;
142da3a660dSBarry Smith }
14344a69424SLois Curfman McInnes static int MatSolveTransAdd_Dense(Mat matin,Vec xx,Vec zz, Vec yy)
144da3a660dSBarry Smith {
14544a69424SLois Curfman McInnes   Mat_Dense  *mat = (Mat_Dense *) matin->data;
1466abc6512SBarry Smith   int     one = 1, info,ierr;
1476abc6512SBarry Smith   Scalar  *x, *y, sone = 1.0;
148da3a660dSBarry Smith   Vec     tmp;
149da3a660dSBarry Smith   VecGetArray(xx,&x); VecGetArray(yy,&y);
150da3a660dSBarry Smith   if (yy == zz) {
151da3a660dSBarry Smith     ierr = VecCreate(yy,&tmp); CHKERR(ierr);
152da3a660dSBarry Smith     ierr = VecCopy(yy,tmp); CHKERR(ierr);
153da3a660dSBarry Smith   }
154da3a660dSBarry Smith   MEMCPY(y,x,mat->m*sizeof(Scalar));
155da3a660dSBarry Smith   /* assume if pivots exist then LU else Cholesky */
156da3a660dSBarry Smith   if (mat->pivots) {
157da3a660dSBarry Smith     LAgetrs_( "T", &mat->m, &one, mat->v, &mat->m, mat->pivots,
158da3a660dSBarry Smith               y, &mat->m, &info );
159da3a660dSBarry Smith   }
160da3a660dSBarry Smith   else {
161da3a660dSBarry Smith     LApotrs_( "L", &mat->m, &one, mat->v, &mat->m,
162da3a660dSBarry Smith               y, &mat->m, &info );
163da3a660dSBarry Smith   }
164da3a660dSBarry Smith   if (info) SETERR(1,"Bad solve");
165da3a660dSBarry Smith   if (tmp) {VecAXPY(&sone,tmp,yy); VecDestroy(tmp);}
166da3a660dSBarry Smith   else VecAXPY(&sone,zz,yy);
167da3a660dSBarry Smith   return 0;
168da3a660dSBarry Smith }
169289bc588SBarry Smith /* ------------------------------------------------------------------*/
17044a69424SLois Curfman McInnes static int MatRelax_Dense(Mat matin,Vec bb,double omega,int flag,double shift,
171289bc588SBarry Smith                        int its,Vec xx)
172289bc588SBarry Smith {
17344a69424SLois Curfman McInnes   Mat_Dense *mat = (Mat_Dense *) matin->data;
174289bc588SBarry Smith   Scalar *x, *b, *v = mat->v, zero = 0.0, xt;
1756abc6512SBarry Smith   int    o = 1,ierr, m = mat->m, i;
176289bc588SBarry Smith 
177289bc588SBarry Smith   if (flag & SOR_ZERO_INITIAL_GUESS) {
178289bc588SBarry Smith     /* this is a hack fix, should have another version without
179289bc588SBarry Smith        the second BLdot */
1806abc6512SBarry Smith     if ((ierr = VecSet(&zero,xx))) SETERR(ierr,0);
181289bc588SBarry Smith   }
182289bc588SBarry Smith   VecGetArray(xx,&x); VecGetArray(bb,&b);
183289bc588SBarry Smith   while (its--) {
184289bc588SBarry Smith     if (flag & SOR_FORWARD_SWEEP){
185289bc588SBarry Smith       for ( i=0; i<m; i++ ) {
186289bc588SBarry Smith         xt = b[i]-BLdot_(&m,v+i,&m,x,&o);
187d6dfbf8fSBarry Smith         x[i] = (1. - omega)*x[i] + omega*(xt/(v[i + i*m]+shift) + x[i]);
188289bc588SBarry Smith       }
189289bc588SBarry Smith     }
190289bc588SBarry Smith     if (flag & SOR_BACKWARD_SWEEP) {
191289bc588SBarry Smith       for ( i=m-1; i>=0; i-- ) {
192289bc588SBarry Smith         xt = b[i]-BLdot_(&m,v+i,&m,x,&o);
193d6dfbf8fSBarry Smith         x[i] = (1. - omega)*x[i] + omega*(xt/(v[i + i*m]+shift) + x[i]);
194289bc588SBarry Smith       }
195289bc588SBarry Smith     }
196289bc588SBarry Smith   }
197289bc588SBarry Smith   return 0;
198289bc588SBarry Smith }
199289bc588SBarry Smith 
200289bc588SBarry Smith /* -----------------------------------------------------------------*/
20144a69424SLois Curfman McInnes static int MatMultTrans_Dense(Mat matin,Vec xx,Vec yy)
202289bc588SBarry Smith {
20344a69424SLois Curfman McInnes   Mat_Dense *mat = (Mat_Dense *) matin->data;
204289bc588SBarry Smith   Scalar *v = mat->v, *x, *y;
205289bc588SBarry Smith   int _One=1;Scalar _DOne=1.0, _DZero=0.0;
206289bc588SBarry Smith   VecGetArray(xx,&x), VecGetArray(yy,&y);
207289bc588SBarry Smith   LAgemv_( "T", &(mat->m), &(mat->n), &_DOne, v, &(mat->m),
208289bc588SBarry Smith          x, &_One, &_DZero, y, &_One );
209289bc588SBarry Smith   return 0;
210289bc588SBarry Smith }
21144a69424SLois Curfman McInnes static int MatMult_Dense(Mat matin,Vec xx,Vec yy)
212289bc588SBarry Smith {
21344a69424SLois Curfman McInnes   Mat_Dense *mat = (Mat_Dense *) matin->data;
214289bc588SBarry Smith   Scalar *v = mat->v, *x, *y;
215289bc588SBarry Smith   int _One=1;Scalar _DOne=1.0, _DZero=0.0;
216289bc588SBarry Smith   VecGetArray(xx,&x); VecGetArray(yy,&y);
217289bc588SBarry Smith   LAgemv_( "N", &(mat->m), &(mat->n), &_DOne, v, &(mat->m),
218289bc588SBarry Smith          x, &_One, &_DZero, y, &_One );
219289bc588SBarry Smith   return 0;
220289bc588SBarry Smith }
22144a69424SLois Curfman McInnes static int MatMultAdd_Dense(Mat matin,Vec xx,Vec zz,Vec yy)
222289bc588SBarry Smith {
22344a69424SLois Curfman McInnes   Mat_Dense *mat = (Mat_Dense *) matin->data;
224289bc588SBarry Smith   Scalar *v = mat->v, *x, *y, *z;
2256abc6512SBarry Smith   int    _One=1; Scalar _DOne=1.0;
226289bc588SBarry Smith   VecGetArray(xx,&x); VecGetArray(yy,&y); VecGetArray(zz,&z);
227289bc588SBarry Smith   if (zz != yy) MEMCPY(y,z,mat->m*sizeof(Scalar));
228289bc588SBarry Smith   LAgemv_( "N", &(mat->m), &(mat->n), &_DOne, v, &(mat->m),
229289bc588SBarry Smith          x, &_One, &_DOne, y, &_One );
230289bc588SBarry Smith   return 0;
231289bc588SBarry Smith }
23244a69424SLois Curfman McInnes static int MatMultTransAdd_Dense(Mat matin,Vec xx,Vec zz,Vec yy)
233289bc588SBarry Smith {
23444a69424SLois Curfman McInnes   Mat_Dense *mat = (Mat_Dense *) matin->data;
235289bc588SBarry Smith   Scalar *v = mat->v, *x, *y, *z;
236289bc588SBarry Smith   int    _One=1;
2376abc6512SBarry Smith   Scalar _DOne=1.0;
238289bc588SBarry Smith   VecGetArray(xx,&x); VecGetArray(yy,&y);
239289bc588SBarry Smith   VecGetArray(zz,&z);
240289bc588SBarry Smith   if (zz != yy) MEMCPY(y,z,mat->m*sizeof(Scalar));
241289bc588SBarry Smith   LAgemv_( "T", &(mat->m), &(mat->n), &_DOne, v, &(mat->m),
242289bc588SBarry Smith          x, &_One, &_DOne, y, &_One );
243289bc588SBarry Smith   return 0;
244289bc588SBarry Smith }
245289bc588SBarry Smith 
246289bc588SBarry Smith /* -----------------------------------------------------------------*/
24744a69424SLois Curfman McInnes static int MatGetRow_Dense(Mat matin,int row,int *ncols,int **cols,
248289bc588SBarry Smith                         Scalar **vals)
249289bc588SBarry Smith {
25044a69424SLois Curfman McInnes   Mat_Dense *mat = (Mat_Dense *) matin->data;
251289bc588SBarry Smith   Scalar *v;
252289bc588SBarry Smith   int    i;
253289bc588SBarry Smith   *ncols = mat->n;
254289bc588SBarry Smith   if (cols) {
255289bc588SBarry Smith     *cols = (int *) MALLOC(mat->n*sizeof(int)); CHKPTR(*cols);
256289bc588SBarry Smith     for ( i=0; i<mat->n; i++ ) *cols[i] = i;
257289bc588SBarry Smith   }
258289bc588SBarry Smith   if (vals) {
259289bc588SBarry Smith     *vals = (Scalar *) MALLOC(mat->n*sizeof(Scalar)); CHKPTR(*vals);
260289bc588SBarry Smith     v = mat->v + row;
261289bc588SBarry Smith     for ( i=0; i<mat->n; i++ ) {*vals[i] = *v; v += mat->m;}
262289bc588SBarry Smith   }
263289bc588SBarry Smith   return 0;
264289bc588SBarry Smith }
26544a69424SLois Curfman McInnes static int MatRestoreRow_Dense(Mat matin,int row,int *ncols,int **cols,
266289bc588SBarry Smith                             Scalar **vals)
267289bc588SBarry Smith {
268289bc588SBarry Smith   if (cols) { FREE(*cols); }
269289bc588SBarry Smith   if (vals) { FREE(*vals); }
270289bc588SBarry Smith   return 0;
271289bc588SBarry Smith }
272289bc588SBarry Smith /* ----------------------------------------------------------------*/
27344a69424SLois Curfman McInnes static int MatInsert_Dense(Mat matin,int m,int *indexm,int n,
274e8d4e0b9SBarry Smith                         int *indexn,Scalar *v,InsertMode addv)
275289bc588SBarry Smith {
27644a69424SLois Curfman McInnes   Mat_Dense *mat = (Mat_Dense *) matin->data;
277289bc588SBarry Smith   int    i,j;
278d6dfbf8fSBarry Smith 
279289bc588SBarry Smith   if (!mat->roworiented) {
280e8d4e0b9SBarry Smith     if (addv == InsertValues) {
281289bc588SBarry Smith       for ( j=0; j<n; j++ ) {
282d6dfbf8fSBarry Smith         if (indexn[j] < 0) {*v += m; continue;}
283289bc588SBarry Smith         for ( i=0; i<m; i++ ) {
284d6dfbf8fSBarry Smith           if (indexm[i] < 0) {*v++; continue;}
285289bc588SBarry Smith           mat->v[indexn[j]*mat->m + indexm[i]] = *v++;
286289bc588SBarry Smith         }
287289bc588SBarry Smith       }
288289bc588SBarry Smith     }
289289bc588SBarry Smith     else {
290289bc588SBarry Smith       for ( j=0; j<n; j++ ) {
291d6dfbf8fSBarry Smith         if (indexn[j] < 0) {*v += m; continue;}
292289bc588SBarry Smith         for ( i=0; i<m; i++ ) {
293d6dfbf8fSBarry Smith           if (indexm[i] < 0) {*v++; continue;}
294289bc588SBarry Smith           mat->v[indexn[j]*mat->m + indexm[i]] += *v++;
295289bc588SBarry Smith         }
296289bc588SBarry Smith       }
297289bc588SBarry Smith     }
298e8d4e0b9SBarry Smith   }
299e8d4e0b9SBarry Smith   else {
300e8d4e0b9SBarry Smith     if (addv == InsertValues) {
301e8d4e0b9SBarry Smith       for ( i=0; i<m; i++ ) {
302d6dfbf8fSBarry Smith         if (indexm[i] < 0) {*v += n; continue;}
303e8d4e0b9SBarry Smith         for ( j=0; j<n; j++ ) {
304d6dfbf8fSBarry Smith           if (indexn[j] < 0) {*v++; continue;}
305e8d4e0b9SBarry Smith           mat->v[indexn[j]*mat->m + indexm[i]] = *v++;
306e8d4e0b9SBarry Smith         }
307e8d4e0b9SBarry Smith       }
308e8d4e0b9SBarry Smith     }
309289bc588SBarry Smith     else {
310289bc588SBarry Smith       for ( i=0; i<m; i++ ) {
311d6dfbf8fSBarry Smith         if (indexm[i] < 0) {*v += n; continue;}
312289bc588SBarry Smith         for ( j=0; j<n; j++ ) {
313d6dfbf8fSBarry Smith           if (indexn[j] < 0) {*v++; continue;}
314289bc588SBarry Smith           mat->v[indexn[j]*mat->m + indexm[i]] += *v++;
315289bc588SBarry Smith         }
316289bc588SBarry Smith       }
317289bc588SBarry Smith     }
318e8d4e0b9SBarry Smith   }
319289bc588SBarry Smith   return 0;
320289bc588SBarry Smith }
321e8d4e0b9SBarry Smith 
322289bc588SBarry Smith /* -----------------------------------------------------------------*/
32344a69424SLois Curfman McInnes static int MatCopy_Dense(Mat matin,Mat *newmat)
324289bc588SBarry Smith {
32544a69424SLois Curfman McInnes   Mat_Dense *mat = (Mat_Dense *) matin->data;
326289bc588SBarry Smith   int ierr;
327289bc588SBarry Smith   Mat newi;
32844a69424SLois Curfman McInnes   Mat_Dense *l;
3296b5873e3SBarry Smith   if ((ierr = MatCreateSequentialDense(matin->comm,mat->m,mat->n,&newi)))
3306b5873e3SBarry Smith                                                           SETERR(ierr,0);
33144a69424SLois Curfman McInnes   l = (Mat_Dense *) newi->data;
332289bc588SBarry Smith   MEMCPY(l->v,mat->v,mat->m*mat->n*sizeof(Scalar));
333289bc588SBarry Smith   *newmat = newi;
334289bc588SBarry Smith   return 0;
335289bc588SBarry Smith }
336da3a660dSBarry Smith #include "viewer.h"
337289bc588SBarry Smith 
33844a69424SLois Curfman McInnes int MatView_Dense(PetscObject obj,Viewer ptr)
339289bc588SBarry Smith {
340289bc588SBarry Smith   Mat         matin = (Mat) obj;
34144a69424SLois Curfman McInnes   Mat_Dense      *mat = (Mat_Dense *) matin->data;
342289bc588SBarry Smith   Scalar      *v;
343289bc588SBarry Smith   int         i,j;
344da3a660dSBarry Smith   PetscObject ojb = (PetscObject) ptr;
345da3a660dSBarry Smith 
3466abc6512SBarry Smith   if (ojb && ojb->cookie == VIEWER_COOKIE && ojb->type == MATLAB_VIEWER) {
347*6f75cfa4SBarry Smith     return ViewerMatlabPutArray_Private(ptr,mat->m,mat->n,mat->v);
348da3a660dSBarry Smith   }
349da3a660dSBarry Smith   else {
3504a0ce102SLois Curfman McInnes     FILE *fd = ViewerFileGetPointer_Private(ptr);
351289bc588SBarry Smith     for ( i=0; i<mat->m; i++ ) {
352289bc588SBarry Smith       v = mat->v + i;
353289bc588SBarry Smith       for ( j=0; j<mat->n; j++ ) {
354289bc588SBarry Smith #if defined(PETSC_COMPLEX)
3558e37dc09SBarry Smith         fprintf(fd,"%6.4e + %6.4e i ",real(*v),imag(*v)); v += mat->m;
356289bc588SBarry Smith #else
3578e37dc09SBarry Smith         fprintf(fd,"%6.4e ",*v); v += mat->m;
358289bc588SBarry Smith #endif
359289bc588SBarry Smith       }
3608e37dc09SBarry Smith       fprintf(fd,"\n");
361289bc588SBarry Smith     }
362da3a660dSBarry Smith   }
363289bc588SBarry Smith   return 0;
364289bc588SBarry Smith }
365289bc588SBarry Smith 
366289bc588SBarry Smith 
36744a69424SLois Curfman McInnes static int MatDestroy_Dense(PetscObject obj)
368289bc588SBarry Smith {
369289bc588SBarry Smith   Mat    mat = (Mat) obj;
37044a69424SLois Curfman McInnes   Mat_Dense *l = (Mat_Dense *) mat->data;
371a5a9c739SBarry Smith #if defined(PETSC_LOG)
372a5a9c739SBarry Smith   PLogObjectState(obj,"Rows %d Cols %d",l->m,l->n);
373a5a9c739SBarry Smith #endif
374289bc588SBarry Smith   if (l->pivots) FREE(l->pivots);
375289bc588SBarry Smith   FREE(l);
376a5a9c739SBarry Smith   PLogObjectDestroy(mat);
3779e25ed09SBarry Smith   PETSCHEADERDESTROY(mat);
378289bc588SBarry Smith   return 0;
379289bc588SBarry Smith }
380289bc588SBarry Smith 
38144a69424SLois Curfman McInnes static int MatTrans_Dense(Mat matin)
382289bc588SBarry Smith {
38344a69424SLois Curfman McInnes   Mat_Dense *mat = (Mat_Dense *) matin->data;
384289bc588SBarry Smith   int    k,j;
385289bc588SBarry Smith   Scalar *v = mat->v, tmp;
386289bc588SBarry Smith   if (mat->m != mat->n) {
387289bc588SBarry Smith     SETERR(1,"Cannot transpose rectangular dense matrix");
388289bc588SBarry Smith   }
389289bc588SBarry Smith   for ( j=0; j<mat->m; j++ ) {
390289bc588SBarry Smith     for ( k=0; k<j; k++ ) {
391289bc588SBarry Smith       tmp = v[j + k*mat->n];
392289bc588SBarry Smith       v[j + k*mat->n] = v[k + j*mat->n];
393289bc588SBarry Smith       v[k + j*mat->n] = tmp;
394289bc588SBarry Smith     }
395289bc588SBarry Smith   }
396289bc588SBarry Smith   return 0;
397289bc588SBarry Smith }
398289bc588SBarry Smith 
39944a69424SLois Curfman McInnes static int MatEqual_Dense(Mat matin1,Mat matin2)
400289bc588SBarry Smith {
40144a69424SLois Curfman McInnes   Mat_Dense *mat1 = (Mat_Dense *) matin1->data;
40244a69424SLois Curfman McInnes   Mat_Dense *mat2 = (Mat_Dense *) matin2->data;
403289bc588SBarry Smith   int    i;
404289bc588SBarry Smith   Scalar *v1 = mat1->v, *v2 = mat2->v;
405289bc588SBarry Smith   if (mat1->m != mat2->m) return 0;
406289bc588SBarry Smith   if (mat1->n != mat2->n) return 0;
407289bc588SBarry Smith   for ( i=0; i<mat1->m*mat1->n; i++ ) {
408289bc588SBarry Smith     if (*v1 != *v2) return 0;
409289bc588SBarry Smith     v1++; v2++;
410289bc588SBarry Smith   }
411289bc588SBarry Smith   return 1;
412289bc588SBarry Smith }
413289bc588SBarry Smith 
41444a69424SLois Curfman McInnes static int MatGetDiag_Dense(Mat matin,Vec v)
415289bc588SBarry Smith {
41644a69424SLois Curfman McInnes   Mat_Dense *mat = (Mat_Dense *) matin->data;
4176abc6512SBarry Smith   int    i, n;
4186abc6512SBarry Smith   Scalar *x;
419289bc588SBarry Smith   CHKTYPE(v,SEQVECTOR);
420289bc588SBarry Smith   VecGetArray(v,&x); VecGetSize(v,&n);
421289bc588SBarry Smith   if (n != mat->m) SETERR(1,"Nonconforming matrix and vector");
422289bc588SBarry Smith   for ( i=0; i<mat->m; i++ ) {
423289bc588SBarry Smith     x[i] = mat->v[i*mat->m + i];
424289bc588SBarry Smith   }
425289bc588SBarry Smith   return 0;
426289bc588SBarry Smith }
427289bc588SBarry Smith 
42844a69424SLois Curfman McInnes static int MatScale_Dense(Mat matin,Vec ll,Vec rr)
429289bc588SBarry Smith {
43044a69424SLois Curfman McInnes   Mat_Dense *mat = (Mat_Dense *) matin->data;
431da3a660dSBarry Smith   Scalar *l,*r,x,*v;
432da3a660dSBarry Smith   int    i,j,m = mat->m, n = mat->n;
43328988994SBarry Smith   if (ll) {
434da3a660dSBarry Smith     VecGetArray(ll,&l); VecGetSize(ll,&m);
435da3a660dSBarry Smith     if (m != mat->m) SETERR(1,"Left scaling vector wrong length");
436da3a660dSBarry Smith     for ( i=0; i<m; i++ ) {
437da3a660dSBarry Smith       x = l[i];
438da3a660dSBarry Smith       v = mat->v + i;
439da3a660dSBarry Smith       for ( j=0; j<n; j++ ) { (*v) *= x; v+= m;}
440da3a660dSBarry Smith     }
441da3a660dSBarry Smith   }
44228988994SBarry Smith   if (rr) {
443da3a660dSBarry Smith     VecGetArray(rr,&r); VecGetSize(rr,&n);
444da3a660dSBarry Smith     if (n != mat->n) SETERR(1,"Right scaling vector wrong length");
445da3a660dSBarry Smith     for ( i=0; i<n; i++ ) {
446da3a660dSBarry Smith       x = r[i];
447da3a660dSBarry Smith       v = mat->v + i*m;
448da3a660dSBarry Smith       for ( j=0; j<m; j++ ) { (*v++) *= x;}
449da3a660dSBarry Smith     }
450da3a660dSBarry Smith   }
451289bc588SBarry Smith   return 0;
452289bc588SBarry Smith }
453289bc588SBarry Smith 
454da3a660dSBarry Smith 
45544a69424SLois Curfman McInnes static int MatNorm_Dense(Mat matin,int type,double *norm)
456289bc588SBarry Smith {
45744a69424SLois Curfman McInnes   Mat_Dense *mat = (Mat_Dense *) matin->data;
458289bc588SBarry Smith   Scalar *v = mat->v;
459289bc588SBarry Smith   double sum = 0.0;
460289bc588SBarry Smith   int    i, j;
461289bc588SBarry Smith   if (type == NORM_FROBENIUS) {
462289bc588SBarry Smith     for (i=0; i<mat->n*mat->m; i++ ) {
463289bc588SBarry Smith #if defined(PETSC_COMPLEX)
464289bc588SBarry Smith       sum += real(conj(*v)*(*v)); v++;
465289bc588SBarry Smith #else
466289bc588SBarry Smith       sum += (*v)*(*v); v++;
467289bc588SBarry Smith #endif
468289bc588SBarry Smith     }
469289bc588SBarry Smith     *norm = sqrt(sum);
470289bc588SBarry Smith   }
471289bc588SBarry Smith   else if (type == NORM_1) {
472289bc588SBarry Smith     *norm = 0.0;
473289bc588SBarry Smith     for ( j=0; j<mat->n; j++ ) {
474289bc588SBarry Smith       sum = 0.0;
475289bc588SBarry Smith       for ( i=0; i<mat->m; i++ ) {
476289bc588SBarry Smith #if defined(PETSC_COMPLEX)
477289bc588SBarry Smith         sum += abs(*v++);
478289bc588SBarry Smith #else
479289bc588SBarry Smith         sum += fabs(*v++);
480289bc588SBarry Smith #endif
481289bc588SBarry Smith       }
482289bc588SBarry Smith       if (sum > *norm) *norm = sum;
483289bc588SBarry Smith     }
484289bc588SBarry Smith   }
485289bc588SBarry Smith   else if (type == NORM_INFINITY) {
486289bc588SBarry Smith     *norm = 0.0;
487289bc588SBarry Smith     for ( j=0; j<mat->m; j++ ) {
488289bc588SBarry Smith       v = mat->v + j;
489289bc588SBarry Smith       sum = 0.0;
490289bc588SBarry Smith       for ( i=0; i<mat->n; i++ ) {
491289bc588SBarry Smith #if defined(PETSC_COMPLEX)
492289bc588SBarry Smith         sum += abs(*v); v += mat->m;
493289bc588SBarry Smith #else
494289bc588SBarry Smith         sum += fabs(*v); v += mat->m;
495289bc588SBarry Smith #endif
496289bc588SBarry Smith       }
497289bc588SBarry Smith       if (sum > *norm) *norm = sum;
498289bc588SBarry Smith     }
499289bc588SBarry Smith   }
500289bc588SBarry Smith   else {
501289bc588SBarry Smith     SETERR(1,"No support for the two norm yet");
502289bc588SBarry Smith   }
503289bc588SBarry Smith   return 0;
504289bc588SBarry Smith }
505289bc588SBarry Smith 
506681d1451SLois Curfman McInnes static int MatSetOption_Dense(Mat aijin,int op)
507289bc588SBarry Smith {
50844a69424SLois Curfman McInnes   Mat_Dense *aij = (Mat_Dense *) aijin->data;
509289bc588SBarry Smith   if (op == ROW_ORIENTED)            aij->roworiented = 1;
510289bc588SBarry Smith   else if (op == COLUMN_ORIENTED)    aij->roworiented = 0;
511289bc588SBarry Smith   /* doesn't care about sorted rows or columns */
512289bc588SBarry Smith   return 0;
513289bc588SBarry Smith }
514289bc588SBarry Smith 
51544a69424SLois Curfman McInnes static int MatZero_Dense(Mat A)
5166f0a148fSBarry Smith {
51744a69424SLois Curfman McInnes   Mat_Dense *l = (Mat_Dense *) A->data;
5186f0a148fSBarry Smith   MEMSET(l->v,0,l->m*l->n*sizeof(Scalar));
5196f0a148fSBarry Smith   return 0;
5206f0a148fSBarry Smith }
5216f0a148fSBarry Smith 
52244a69424SLois Curfman McInnes static int MatZeroRows_Dense(Mat A,IS is,Scalar *diag)
5236f0a148fSBarry Smith {
52444a69424SLois Curfman McInnes   Mat_Dense *l = (Mat_Dense *) A->data;
5256abc6512SBarry Smith   int    n = l->n, i, j,ierr,N, *rows;
5266f0a148fSBarry Smith   Scalar *slot;
527260925b8SBarry Smith   ierr = ISGetLocalSize(is,&N); CHKERR(ierr);
528260925b8SBarry Smith   ierr = ISGetIndices(is,&rows); CHKERR(ierr);
5296f0a148fSBarry Smith   for ( i=0; i<N; i++ ) {
5306f0a148fSBarry Smith     slot = l->v + rows[i];
5316f0a148fSBarry Smith     for ( j=0; j<n; j++ ) { *slot = 0.0; slot += n;}
5326f0a148fSBarry Smith   }
5336f0a148fSBarry Smith   if (diag) {
5346f0a148fSBarry Smith     for ( i=0; i<N; i++ ) {
5356f0a148fSBarry Smith       slot = l->v + (n+1)*rows[i];
536260925b8SBarry Smith       *slot = *diag;
5376f0a148fSBarry Smith     }
5386f0a148fSBarry Smith   }
539260925b8SBarry Smith   ISRestoreIndices(is,&rows);
5406f0a148fSBarry Smith   return 0;
5416f0a148fSBarry Smith }
542557bce09SLois Curfman McInnes 
543fa9ec3f1SLois Curfman McInnes static int MatGetSize_Dense(Mat matin,int *m,int *n)
544557bce09SLois Curfman McInnes {
54544a69424SLois Curfman McInnes   Mat_Dense *mat = (Mat_Dense *) matin->data;
546557bce09SLois Curfman McInnes   *m = mat->m; *n = mat->n;
547557bce09SLois Curfman McInnes   return 0;
548557bce09SLois Curfman McInnes }
549557bce09SLois Curfman McInnes 
55044a69424SLois Curfman McInnes static int MatGetArray_Dense(Mat matin,Scalar **array)
55164e87e97SBarry Smith {
55244a69424SLois Curfman McInnes   Mat_Dense *mat = (Mat_Dense *) matin->data;
55364e87e97SBarry Smith   *array = mat->v;
55464e87e97SBarry Smith   return 0;
55564e87e97SBarry Smith }
5560754003eSLois Curfman McInnes 
5570754003eSLois Curfman McInnes 
5580754003eSLois Curfman McInnes static int MatGetSubMatrixInPlace_Dense(Mat matin,IS isrow,IS iscol)
5590754003eSLois Curfman McInnes {
5600754003eSLois Curfman McInnes   SETERR(1,"MatGetSubMatrixInPlace_Dense not yet done.");
5610754003eSLois Curfman McInnes }
5620754003eSLois Curfman McInnes 
5630754003eSLois Curfman McInnes static int MatGetSubMatrix_Dense(Mat matin,IS isrow,IS iscol,Mat *submat)
5640754003eSLois Curfman McInnes {
5650754003eSLois Curfman McInnes   Mat_Dense *mat = (Mat_Dense *) matin->data;
5660754003eSLois Curfman McInnes   int     nznew, *smap, i, j, ierr, oldcols = mat->n;
5670754003eSLois Curfman McInnes   int     *irow, *icol, nrows, ncols, *cwork, jstart;
5680754003eSLois Curfman McInnes   Scalar  *vwork, *val;
5690754003eSLois Curfman McInnes   Mat     newmat;
5700754003eSLois Curfman McInnes 
5710754003eSLois Curfman McInnes   ierr = ISGetIndices(isrow,&irow); CHKERR(ierr);
5720754003eSLois Curfman McInnes   ierr = ISGetIndices(iscol,&icol); CHKERR(ierr);
5730754003eSLois Curfman McInnes   ierr = ISGetSize(isrow,&nrows); CHKERR(ierr);
5740754003eSLois Curfman McInnes   ierr = ISGetSize(iscol,&ncols); CHKERR(ierr);
5750754003eSLois Curfman McInnes 
5760754003eSLois Curfman McInnes   smap = (int *) MALLOC(oldcols*sizeof(int)); CHKPTR(smap);
5770754003eSLois Curfman McInnes   cwork = (int *) MALLOC(ncols*sizeof(int)); CHKPTR(cwork);
5780754003eSLois Curfman McInnes   vwork = (Scalar *) MALLOC(ncols*sizeof(Scalar)); CHKPTR(vwork);
5790754003eSLois Curfman McInnes   memset((char*)smap,0,oldcols*sizeof(int));
5800754003eSLois Curfman McInnes   for ( i=0; i<ncols; i++ ) smap[icol[i]] = i+1;
5810754003eSLois Curfman McInnes 
5820754003eSLois Curfman McInnes   /* Create and fill new matrix */
5830754003eSLois Curfman McInnes   ierr = MatCreateSequentialDense(matin->comm,nrows,ncols,&newmat);
5840754003eSLois Curfman McInnes          CHKERR(ierr);
5850754003eSLois Curfman McInnes   for (i=0; i<nrows; i++) {
5860754003eSLois Curfman McInnes     nznew = 0;
5870754003eSLois Curfman McInnes     val   = mat->v + irow[i];
5880754003eSLois Curfman McInnes     for (j=0; j<oldcols; j++) {
5890754003eSLois Curfman McInnes       if (smap[j]) {
5900754003eSLois Curfman McInnes         cwork[nznew]   = smap[j] - 1;
5910754003eSLois Curfman McInnes         vwork[nznew++] = val[j * mat->m];
5920754003eSLois Curfman McInnes       }
5930754003eSLois Curfman McInnes     }
5940754003eSLois Curfman McInnes     ierr = MatSetValues(newmat,1,&i,nznew,cwork,vwork,InsertValues);
5950754003eSLois Curfman McInnes            CHKERR(ierr);
5960754003eSLois Curfman McInnes   }
5970754003eSLois Curfman McInnes   ierr = MatBeginAssembly(newmat,FINAL_ASSEMBLY); CHKERR(ierr);
5980754003eSLois Curfman McInnes   ierr = MatEndAssembly(newmat,FINAL_ASSEMBLY); CHKERR(ierr);
5990754003eSLois Curfman McInnes 
6000754003eSLois Curfman McInnes   /* Free work space */
6010754003eSLois Curfman McInnes   FREE(smap); FREE(cwork); FREE(vwork);
6020754003eSLois Curfman McInnes   ierr = ISRestoreIndices(isrow,&irow); CHKERR(ierr);
6030754003eSLois Curfman McInnes   ierr = ISRestoreIndices(iscol,&icol); CHKERR(ierr);
6040754003eSLois Curfman McInnes   *submat = newmat;
6050754003eSLois Curfman McInnes   return 0;
6060754003eSLois Curfman McInnes }
6070754003eSLois Curfman McInnes 
608289bc588SBarry Smith /* -------------------------------------------------------------------*/
60944a69424SLois Curfman McInnes static struct _MatOps MatOps = {MatInsert_Dense,
61044a69424SLois Curfman McInnes        MatGetRow_Dense, MatRestoreRow_Dense,
61144a69424SLois Curfman McInnes        MatMult_Dense, MatMultAdd_Dense,
61244a69424SLois Curfman McInnes        MatMultTrans_Dense, MatMultTransAdd_Dense,
61344a69424SLois Curfman McInnes        MatSolve_Dense,MatSolveAdd_Dense,
61444a69424SLois Curfman McInnes        MatSolveTrans_Dense,MatSolveTransAdd_Dense,
61544a69424SLois Curfman McInnes        MatLUFactor_Dense,MatChFactor_Dense,
61644a69424SLois Curfman McInnes        MatRelax_Dense,
61744a69424SLois Curfman McInnes        MatTrans_Dense,
618fa9ec3f1SLois Curfman McInnes        MatGetInfo_Dense,MatEqual_Dense,
61944a69424SLois Curfman McInnes        MatCopy_Dense,
62044a69424SLois Curfman McInnes        MatGetDiag_Dense,MatScale_Dense,MatNorm_Dense,
621289bc588SBarry Smith        0,0,
622681d1451SLois Curfman McInnes        0, MatSetOption_Dense,MatZero_Dense,MatZeroRows_Dense,0,
62344a69424SLois Curfman McInnes        MatLUFactorSymbolic_Dense,MatLUFactorNumeric_Dense,
62444a69424SLois Curfman McInnes        MatChFactorSymbolic_Dense,MatChFactorNumeric_Dense,
625fa9ec3f1SLois Curfman McInnes        MatGetSize_Dense,MatGetSize_Dense,0,
6260754003eSLois Curfman McInnes        0,0,MatGetArray_Dense,0,
6270754003eSLois Curfman McInnes        MatGetSubMatrix_Dense,MatGetSubMatrixInPlace_Dense};
6280754003eSLois Curfman McInnes 
62920563c6bSBarry Smith /*@
63020563c6bSBarry Smith    MatCreateSequentialDense - Creates a sequential dense matrix that
631d65003e9SLois Curfman McInnes    is stored in column major order (the usual Fortran 77 manner). Many
632d65003e9SLois Curfman McInnes    of the matrix operations use the BLAS and LAPACK routines.
633289bc588SBarry Smith 
63420563c6bSBarry Smith    Input Parameters:
6350c775827SLois Curfman McInnes .  comm - MPI communicator, set to MPI_COMM_SELF
6360c775827SLois Curfman McInnes .  m - number of rows
6370c775827SLois Curfman McInnes .  n - number of column
63820563c6bSBarry Smith 
63920563c6bSBarry Smith    Output Parameter:
6400c775827SLois Curfman McInnes .  newmat - the matrix
64120563c6bSBarry Smith 
642d65003e9SLois Curfman McInnes .keywords: Mat, dense, matrix, LAPACK, BLAS
643d65003e9SLois Curfman McInnes 
644d65003e9SLois Curfman McInnes .seealso: MatCreateSequentialAIJ()
64520563c6bSBarry Smith @*/
6466b5873e3SBarry Smith int MatCreateSequentialDense(MPI_Comm comm,int m,int n,Mat *newmat)
647289bc588SBarry Smith {
64844a69424SLois Curfman McInnes   int       size = sizeof(Mat_Dense) + m*n*sizeof(Scalar);
649289bc588SBarry Smith   Mat mat;
65044a69424SLois Curfman McInnes   Mat_Dense    *l;
651289bc588SBarry Smith   *newmat        = 0;
6526b5873e3SBarry Smith   PETSCHEADERCREATE(mat,_Mat,MAT_COOKIE,MATDENSE,comm);
653a5a9c739SBarry Smith   PLogObjectCreate(mat);
65444a69424SLois Curfman McInnes   l              = (Mat_Dense *) MALLOC(size); CHKPTR(l);
655289bc588SBarry Smith   mat->ops       = &MatOps;
65644a69424SLois Curfman McInnes   mat->destroy   = MatDestroy_Dense;
65744a69424SLois Curfman McInnes   mat->view      = MatView_Dense;
658289bc588SBarry Smith   mat->data      = (void *) l;
659289bc588SBarry Smith   mat->factor    = 0;
660d6dfbf8fSBarry Smith 
661289bc588SBarry Smith   l->m           = m;
662289bc588SBarry Smith   l->n           = n;
663289bc588SBarry Smith   l->v           = (Scalar *) (l + 1);
664289bc588SBarry Smith   l->pivots      = 0;
665289bc588SBarry Smith   l->roworiented = 1;
666d6dfbf8fSBarry Smith 
667289bc588SBarry Smith   MEMSET(l->v,0,m*n*sizeof(Scalar));
668289bc588SBarry Smith   *newmat = mat;
669289bc588SBarry Smith   return 0;
670289bc588SBarry Smith }
671289bc588SBarry Smith 
67244a69424SLois Curfman McInnes int MatCreate_Dense(Mat matin,Mat *newmat)
673289bc588SBarry Smith {
67444a69424SLois Curfman McInnes   Mat_Dense *m = (Mat_Dense *) matin->data;
6756b5873e3SBarry Smith   return MatCreateSequentialDense(matin->comm,m->m,m->n,newmat);
676289bc588SBarry Smith }
677