578 lines
13 KiB
C
578 lines
13 KiB
C
/*BHEADER**********************************************************************
|
|
* Copyright (c) 2008, Lawrence Livermore National Security, LLC.
|
|
* Produced at the Lawrence Livermore National Laboratory.
|
|
* This file is part of HYPRE. See file COPYRIGHT for details.
|
|
*
|
|
* HYPRE is free software; you can redistribute it and/or modify it under the
|
|
* terms of the GNU Lesser General Public License (as published by the Free
|
|
* Software Foundation) version 2.1 dated February 1999.
|
|
*
|
|
* $Revision$
|
|
***********************************************************************EHEADER*/
|
|
|
|
|
|
|
|
|
|
#include <assert.h>
|
|
#include <math.h>
|
|
|
|
#include "temp_multivector.h"
|
|
|
|
void*
|
|
hypre_TempMultiVectorCreateFromSampleVector( void* ii_, HYPRE_Int n, void* sample ) {
|
|
|
|
HYPRE_Int i;
|
|
hypre_TempMultiVector* data;
|
|
HYPRE_InterfaceInterpreter* ii = (HYPRE_InterfaceInterpreter*)ii_;
|
|
|
|
data = (hypre_TempMultiVector*) malloc(sizeof(hypre_TempMultiVector));
|
|
hypre_assert( data != NULL );
|
|
|
|
data->interpreter = ii;
|
|
data->numVectors = n;
|
|
|
|
data->vector = (void**) calloc( n, sizeof(void*) );
|
|
hypre_assert( data->vector != NULL );
|
|
|
|
data->ownsVectors = 1;
|
|
data->mask = NULL;
|
|
data->ownsMask = 0;
|
|
|
|
for ( i = 0; i < n; i++ )
|
|
data->vector[i] = (ii->CreateVector)(sample);
|
|
|
|
return data;
|
|
|
|
}
|
|
|
|
void*
|
|
hypre_TempMultiVectorCreateCopy( void* src_, HYPRE_Int copyValues ) {
|
|
|
|
HYPRE_Int i, n;
|
|
|
|
hypre_TempMultiVector* src;
|
|
hypre_TempMultiVector* dest;
|
|
|
|
src = (hypre_TempMultiVector*)src_;
|
|
hypre_assert( src != NULL );
|
|
|
|
n = src->numVectors;
|
|
|
|
dest = hypre_TempMultiVectorCreateFromSampleVector( src->interpreter,
|
|
n, src->vector[0] );
|
|
if ( copyValues )
|
|
for ( i = 0; i < n; i++ ) {
|
|
(dest->interpreter->CopyVector)(src->vector[i],dest->vector[i]);
|
|
}
|
|
|
|
return dest;
|
|
}
|
|
|
|
void
|
|
hypre_TempMultiVectorDestroy( void* v_ ) {
|
|
|
|
HYPRE_Int i;
|
|
hypre_TempMultiVector* data = (hypre_TempMultiVector*)v_;
|
|
|
|
if ( data == NULL )
|
|
return;
|
|
|
|
if ( data->ownsVectors && data->vector != NULL ) {
|
|
for ( i = 0; i < data->numVectors; i++ )
|
|
(data->interpreter->DestroyVector)(data->vector[i]);
|
|
free(data->vector);
|
|
}
|
|
if ( data->mask && data->ownsMask )
|
|
free(data->mask);
|
|
free(data);
|
|
}
|
|
|
|
HYPRE_Int
|
|
hypre_TempMultiVectorWidth( void* v ) {
|
|
|
|
hypre_TempMultiVector* data = (hypre_TempMultiVector*)v;
|
|
|
|
if ( data == NULL )
|
|
return 0;
|
|
|
|
return data->numVectors;
|
|
}
|
|
|
|
HYPRE_Int
|
|
hypre_TempMultiVectorHeight( void* v ) {
|
|
|
|
return 0;
|
|
}
|
|
|
|
void
|
|
hypre_TempMultiVectorSetMask( void* v, HYPRE_Int* mask ) {
|
|
|
|
hypre_TempMultiVector* data = (hypre_TempMultiVector*)v;
|
|
|
|
hypre_assert( data != NULL );
|
|
data->mask = mask;
|
|
data->ownsMask = 0;
|
|
}
|
|
|
|
void
|
|
hypre_TempMultiVectorClear( void* v ) {
|
|
|
|
HYPRE_Int i;
|
|
hypre_TempMultiVector* data = (hypre_TempMultiVector*)v;
|
|
|
|
hypre_assert( data != NULL );
|
|
|
|
for ( i = 0; i < data->numVectors; i++ )
|
|
if ( data->mask == NULL || (data->mask)[i] )
|
|
(data->interpreter->ClearVector)(data->vector[i]);
|
|
}
|
|
|
|
void
|
|
hypre_TempMultiVectorSetRandom( void* v, HYPRE_Int seed ) {
|
|
|
|
HYPRE_Int i;
|
|
hypre_TempMultiVector* data = (hypre_TempMultiVector*)v;
|
|
|
|
hypre_assert( data != NULL );
|
|
|
|
srand( seed );
|
|
for ( i = 0; i < data->numVectors; i++ ) {
|
|
if ( data->mask == NULL || (data->mask)[i] ) {
|
|
seed = rand();
|
|
(data->interpreter->SetRandomValues)(data->vector[i],seed);
|
|
}
|
|
}
|
|
}
|
|
|
|
void
|
|
hypre_collectVectorPtr( HYPRE_Int* mask, hypre_TempMultiVector* x, void** px ) {
|
|
|
|
HYPRE_Int ix, jx;
|
|
|
|
if ( mask != NULL ) {
|
|
for ( ix = 0, jx = 0; ix < x->numVectors; ix++ )
|
|
if ( mask[ix] )
|
|
px[jx++] = x->vector[ix];
|
|
}
|
|
else
|
|
for ( ix = 0; ix < x->numVectors; ix++ )
|
|
px[ix] = x->vector[ix];
|
|
|
|
}
|
|
|
|
void
|
|
hypre_TempMultiVectorCopy( void* src, void* dest ) {
|
|
|
|
HYPRE_Int i, ms, md;
|
|
void** ps;
|
|
void** pd;
|
|
hypre_TempMultiVector* srcData = (hypre_TempMultiVector*)src;
|
|
hypre_TempMultiVector* destData = (hypre_TempMultiVector*)dest;
|
|
|
|
hypre_assert( srcData != NULL && destData != NULL );
|
|
|
|
ms = aux_maskCount( srcData->numVectors, srcData->mask );
|
|
md = aux_maskCount( destData->numVectors, destData->mask );
|
|
hypre_assert( ms == md );
|
|
|
|
ps = (void**) calloc( ms, sizeof(void*) );
|
|
hypre_assert( ps != NULL );
|
|
pd = (void**) calloc( md, sizeof(void*) );
|
|
hypre_assert( pd != NULL );
|
|
|
|
hypre_collectVectorPtr( srcData->mask, srcData, ps );
|
|
hypre_collectVectorPtr( destData->mask, destData, pd );
|
|
|
|
for ( i = 0; i < ms; i++ )
|
|
(srcData->interpreter->CopyVector)(ps[i],pd[i]);
|
|
|
|
free(ps);
|
|
free(pd);
|
|
}
|
|
|
|
void
|
|
hypre_TempMultiVectorAxpy( double a, void* x_, void* y_ ) {
|
|
|
|
HYPRE_Int i, mx, my;
|
|
void** px;
|
|
void** py;
|
|
hypre_TempMultiVector* xData;
|
|
hypre_TempMultiVector* yData;
|
|
|
|
xData = (hypre_TempMultiVector*)x_;
|
|
yData = (hypre_TempMultiVector*)y_;
|
|
hypre_assert( xData != NULL && yData != NULL );
|
|
|
|
mx = aux_maskCount( xData->numVectors, xData->mask );
|
|
my = aux_maskCount( yData->numVectors, yData->mask );
|
|
hypre_assert( mx == my );
|
|
|
|
px = (void**) calloc( mx, sizeof(void*) );
|
|
hypre_assert( px != NULL );
|
|
py = (void**) calloc( my, sizeof(void*) );
|
|
hypre_assert( py != NULL );
|
|
|
|
hypre_collectVectorPtr( xData->mask, xData, px );
|
|
hypre_collectVectorPtr( yData->mask, yData, py );
|
|
|
|
for ( i = 0; i < mx; i++ )
|
|
(xData->interpreter->Axpy)(a,px[i],py[i]);
|
|
|
|
free(px);
|
|
free(py);
|
|
}
|
|
|
|
void
|
|
hypre_TempMultiVectorByMultiVector( void* x_, void* y_,
|
|
HYPRE_Int xyGHeight, HYPRE_Int xyHeight,
|
|
HYPRE_Int xyWidth, double* xyVal ) {
|
|
/* xy = x'*y */
|
|
|
|
HYPRE_Int ix, iy, mx, my, jxy;
|
|
double* p;
|
|
void** px;
|
|
void** py;
|
|
hypre_TempMultiVector* xData;
|
|
hypre_TempMultiVector* yData;
|
|
|
|
xData = (hypre_TempMultiVector*)x_;
|
|
yData = (hypre_TempMultiVector*)y_;
|
|
hypre_assert( xData != NULL && yData != NULL );
|
|
|
|
mx = aux_maskCount( xData->numVectors, xData->mask );
|
|
hypre_assert( mx == xyHeight );
|
|
|
|
my = aux_maskCount( yData->numVectors, yData->mask );
|
|
hypre_assert( my == xyWidth );
|
|
|
|
px = (void**) calloc( mx, sizeof(void*) );
|
|
hypre_assert( px != NULL );
|
|
py = (void**) calloc( my, sizeof(void*) );
|
|
hypre_assert( py != NULL );
|
|
|
|
hypre_collectVectorPtr( xData->mask, xData, px );
|
|
hypre_collectVectorPtr( yData->mask, yData, py );
|
|
|
|
jxy = xyGHeight - xyHeight;
|
|
for ( iy = 0, p = xyVal; iy < my; iy++ ) {
|
|
for ( ix = 0; ix < mx; ix++, p++ )
|
|
*p = (xData->interpreter->InnerProd)(px[ix],py[iy]);
|
|
p += jxy;
|
|
}
|
|
|
|
free(px);
|
|
free(py);
|
|
|
|
}
|
|
|
|
void
|
|
hypre_TempMultiVectorByMultiVectorDiag( void* x_, void* y_,
|
|
HYPRE_Int* mask, HYPRE_Int n, double* diag ) {
|
|
/* diag = diag(x'*y) */
|
|
|
|
HYPRE_Int i, mx, my, m;
|
|
void** px;
|
|
void** py;
|
|
HYPRE_Int* index;
|
|
hypre_TempMultiVector* xData;
|
|
hypre_TempMultiVector* yData;
|
|
|
|
xData = (hypre_TempMultiVector*)x_;
|
|
yData = (hypre_TempMultiVector*)y_;
|
|
hypre_assert( xData != NULL && yData != NULL );
|
|
|
|
mx = aux_maskCount( xData->numVectors, xData->mask );
|
|
my = aux_maskCount( yData->numVectors, yData->mask );
|
|
m = aux_maskCount( n, mask );
|
|
hypre_assert( mx == my && mx == m );
|
|
|
|
px = (void**) calloc( mx, sizeof(void*) );
|
|
hypre_assert( px != NULL );
|
|
py = (void**) calloc( my, sizeof(void*) );
|
|
hypre_assert( py != NULL );
|
|
|
|
hypre_collectVectorPtr( xData->mask, xData, px );
|
|
hypre_collectVectorPtr( yData->mask, yData, py );
|
|
|
|
index = (HYPRE_Int*)calloc( m, sizeof(HYPRE_Int) );
|
|
aux_indexFromMask( n, mask, index );
|
|
|
|
for ( i = 0; i < m; i++ )
|
|
*(diag+index[i]-1) = (xData->interpreter->InnerProd)(px[i],py[i]);
|
|
|
|
free(index);
|
|
free(px);
|
|
free(py);
|
|
|
|
}
|
|
|
|
void
|
|
hypre_TempMultiVectorByMatrix( void* x_,
|
|
HYPRE_Int rGHeight, HYPRE_Int rHeight,
|
|
HYPRE_Int rWidth, double* rVal,
|
|
void* y_ ) {
|
|
|
|
HYPRE_Int i, j, jump;
|
|
HYPRE_Int mx, my;
|
|
double* p;
|
|
void** px;
|
|
void** py;
|
|
hypre_TempMultiVector* xData;
|
|
hypre_TempMultiVector* yData;
|
|
|
|
xData = (hypre_TempMultiVector*)x_;
|
|
yData = (hypre_TempMultiVector*)y_;
|
|
hypre_assert( xData != NULL && yData != NULL );
|
|
|
|
mx = aux_maskCount( xData->numVectors, xData->mask );
|
|
my = aux_maskCount( yData->numVectors, yData->mask );
|
|
|
|
hypre_assert( mx == rHeight && my == rWidth );
|
|
|
|
px = (void**) calloc( mx, sizeof(void*) );
|
|
hypre_assert( px != NULL );
|
|
py = (void**) calloc( my, sizeof(void*) );
|
|
hypre_assert( py != NULL );
|
|
|
|
hypre_collectVectorPtr( xData->mask, xData, px );
|
|
hypre_collectVectorPtr( yData->mask, yData, py );
|
|
|
|
jump = rGHeight - rHeight;
|
|
for ( j = 0, p = rVal; j < my; j++ ) {
|
|
(xData->interpreter->ClearVector)( py[j] );
|
|
for ( i = 0; i < mx; i++, p++ )
|
|
(xData->interpreter->Axpy)(*p,px[i],py[j]);
|
|
p += jump;
|
|
}
|
|
|
|
free(px);
|
|
free(py);
|
|
}
|
|
|
|
void
|
|
hypre_TempMultiVectorXapy( void* x_,
|
|
HYPRE_Int rGHeight, HYPRE_Int rHeight,
|
|
HYPRE_Int rWidth, double* rVal,
|
|
void* y_ ) {
|
|
|
|
HYPRE_Int i, j, jump;
|
|
HYPRE_Int mx, my;
|
|
double* p;
|
|
void** px;
|
|
void** py;
|
|
hypre_TempMultiVector* xData;
|
|
hypre_TempMultiVector* yData;
|
|
|
|
xData = (hypre_TempMultiVector*)x_;
|
|
yData = (hypre_TempMultiVector*)y_;
|
|
hypre_assert( xData != NULL && yData != NULL );
|
|
|
|
mx = aux_maskCount( xData->numVectors, xData->mask );
|
|
my = aux_maskCount( yData->numVectors, yData->mask );
|
|
|
|
hypre_assert( mx == rHeight && my == rWidth );
|
|
|
|
px = (void**) calloc( mx, sizeof(void*) );
|
|
hypre_assert( px != NULL );
|
|
py = (void**) calloc( my, sizeof(void*) );
|
|
hypre_assert( py != NULL );
|
|
|
|
hypre_collectVectorPtr( xData->mask, xData, px );
|
|
hypre_collectVectorPtr( yData->mask, yData, py );
|
|
|
|
jump = rGHeight - rHeight;
|
|
for ( j = 0, p = rVal; j < my; j++ ) {
|
|
for ( i = 0; i < mx; i++, p++ )
|
|
(xData->interpreter->Axpy)(*p,px[i],py[j]);
|
|
p += jump;
|
|
}
|
|
|
|
free(px);
|
|
free(py);
|
|
}
|
|
|
|
void
|
|
hypre_TempMultiVectorByDiagonal( void* x_,
|
|
HYPRE_Int* mask, HYPRE_Int n, double* diag,
|
|
void* y_ ) {
|
|
|
|
HYPRE_Int j;
|
|
HYPRE_Int mx, my, m;
|
|
void** px;
|
|
void** py;
|
|
HYPRE_Int* index;
|
|
hypre_TempMultiVector* xData;
|
|
hypre_TempMultiVector* yData;
|
|
|
|
xData = (hypre_TempMultiVector*)x_;
|
|
yData = (hypre_TempMultiVector*)y_;
|
|
hypre_assert( xData != NULL && yData != NULL );
|
|
|
|
mx = aux_maskCount( xData->numVectors, xData->mask );
|
|
my = aux_maskCount( yData->numVectors, yData->mask );
|
|
m = aux_maskCount( n, mask );
|
|
|
|
hypre_assert( mx == m && my == m );
|
|
|
|
if ( m < 1 )
|
|
return;
|
|
|
|
px = (void**) calloc( mx, sizeof(void*) );
|
|
hypre_assert( px != NULL );
|
|
py = (void**) calloc( my, sizeof(void*) );
|
|
hypre_assert( py != NULL );
|
|
|
|
index = (HYPRE_Int*)calloc( m, sizeof(HYPRE_Int) );
|
|
aux_indexFromMask( n, mask, index );
|
|
|
|
hypre_collectVectorPtr( xData->mask, xData, px );
|
|
hypre_collectVectorPtr( yData->mask, yData, py );
|
|
|
|
for ( j = 0; j < my; j++ ) {
|
|
(xData->interpreter->ClearVector)(py[j]);
|
|
(xData->interpreter->Axpy)(diag[index[j]-1],px[j],py[j]);
|
|
}
|
|
|
|
free(px);
|
|
free(py);
|
|
free( index );
|
|
}
|
|
|
|
void
|
|
hypre_TempMultiVectorEval( void (*f)( void*, void*, void* ), void* par,
|
|
void* x_, void* y_ ) {
|
|
|
|
HYPRE_Int i, mx, my;
|
|
void** px;
|
|
void** py;
|
|
hypre_TempMultiVector* x;
|
|
hypre_TempMultiVector* y;
|
|
|
|
x = (hypre_TempMultiVector*)x_;
|
|
y = (hypre_TempMultiVector*)y_;
|
|
hypre_assert( x != NULL && y != NULL );
|
|
|
|
if ( f == NULL ) {
|
|
hypre_TempMultiVectorCopy( x, y );
|
|
return;
|
|
}
|
|
|
|
mx = aux_maskCount( x->numVectors, x->mask );
|
|
my = aux_maskCount( y->numVectors, y->mask );
|
|
hypre_assert( mx == my );
|
|
|
|
px = (void**) calloc( mx, sizeof(void*) );
|
|
hypre_assert( px != NULL );
|
|
py = (void**) calloc( my, sizeof(void*) );
|
|
hypre_assert( py != NULL );
|
|
|
|
hypre_collectVectorPtr( x->mask, x, px );
|
|
hypre_collectVectorPtr( y->mask, y, py );
|
|
|
|
for ( i = 0; i < mx; i++ )
|
|
f( par, (void*)px[i], (void*)py[i] );
|
|
|
|
free(px);
|
|
free(py);
|
|
}
|
|
|
|
HYPRE_Int
|
|
hypre_TempMultiVectorPrint( void* x_, const char* fileName ) {
|
|
|
|
HYPRE_Int i, ierr;
|
|
hypre_TempMultiVector* x;
|
|
char fullName[128];
|
|
|
|
x = (hypre_TempMultiVector*)x_;
|
|
hypre_assert( x != NULL );
|
|
if ( x->interpreter->PrintVector == NULL )
|
|
return 1;
|
|
|
|
ierr = 0;
|
|
for ( i = 0; i < x->numVectors; i++ ) {
|
|
hypre_sprintf( fullName, "%s.%d", fileName, i );
|
|
ierr = ierr ||
|
|
(x->interpreter->PrintVector)( x->vector[i], fullName );
|
|
}
|
|
return ierr;
|
|
}
|
|
|
|
void*
|
|
hypre_TempMultiVectorRead( MPI_Comm comm, void* ii_, const char* fileName ) {
|
|
|
|
HYPRE_Int i, n, id;
|
|
FILE* fp;
|
|
char fullName[128];
|
|
hypre_TempMultiVector* x;
|
|
HYPRE_InterfaceInterpreter* ii = (HYPRE_InterfaceInterpreter*)ii_;
|
|
|
|
if ( ii->ReadVector == NULL )
|
|
return NULL;
|
|
|
|
hypre_MPI_Comm_rank( comm, &id );
|
|
|
|
n = 0;
|
|
do {
|
|
hypre_sprintf( fullName, "%s.%d.%d", fileName, n, id );
|
|
if ( (fp = fopen(fullName, "r")) ) {
|
|
n++;
|
|
fclose( fp );
|
|
}
|
|
} while ( fp );
|
|
|
|
x = (hypre_TempMultiVector*) malloc(sizeof(hypre_TempMultiVector));
|
|
hypre_assert( x != NULL );
|
|
|
|
x->interpreter = ii;
|
|
|
|
x->numVectors = n;
|
|
|
|
x->vector = (void**) calloc( n, sizeof(void*) );
|
|
hypre_assert( x->vector != NULL );
|
|
|
|
x->ownsVectors = 1;
|
|
|
|
for ( i = 0; i < n; i++ ) {
|
|
hypre_sprintf( fullName, "%s.%d", fileName, i );
|
|
x->vector[i] = (ii->ReadVector)( comm, fullName );
|
|
}
|
|
|
|
x->mask = NULL;
|
|
x->ownsMask = 0;
|
|
|
|
return x;
|
|
}
|
|
|
|
HYPRE_Int
|
|
aux_maskCount( HYPRE_Int n, HYPRE_Int* mask ) {
|
|
|
|
HYPRE_Int i, m;
|
|
|
|
if ( mask == NULL )
|
|
return n;
|
|
|
|
for ( i = m = 0; i < n; i++ )
|
|
if ( mask[i] )
|
|
m++;
|
|
|
|
return m;
|
|
}
|
|
|
|
void
|
|
aux_indexFromMask( HYPRE_Int n, HYPRE_Int* mask, HYPRE_Int* index ) {
|
|
|
|
HYPRE_Int i, j;
|
|
|
|
if ( mask != NULL ) {
|
|
for ( i = 0, j = 0; i < n; i++ )
|
|
if ( mask[i] )
|
|
index[j++] = i + 1;
|
|
}
|
|
else
|
|
for ( i = 0; i < n; i++ )
|
|
index[i] = i + 1;
|
|
|
|
}
|
|
|
|
|