HYPRE_Int
HYPRE_StructSMGSolve( HYPRE_StructSolver solver,
HYPRE_StructMatrix A,
HYPRE_StructVector b,
HYPRE_StructVector x )
{
return( hypre_SMGSolve( (void *) solver,
(hypre_StructMatrix *) A,
(hypre_StructVector *) b,
(hypre_StructVector *) x ) );
}
int
hypre_SMGRelax( void *relax_vdata,
hypre_StructMatrix *A,
hypre_StructVector *b,
hypre_StructVector *x )
{
hypre_SMGRelaxData *relax_data = relax_vdata;
int zero_guess;
int stencil_dim;
hypre_StructVector *temp_vec;
hypre_StructMatrix *A_sol;
hypre_StructMatrix *A_rem;
void **residual_data;
void **solve_data;
hypre_IndexRef base_stride;
hypre_BoxArray *base_box_a;
double zero = 0.0;
int max_iter;
int num_spaces;
int *space_ranks;
int i, j, k, is;
int ierr = 0;
/*----------------------------------------------------------
* Note: The zero_guess stuff is not handled correctly
* for general relaxation parameters. It is correct when
* the spaces are independent sets in the direction of
* relaxation.
*----------------------------------------------------------*/
hypre_BeginTiming(relax_data -> time_index);
/*----------------------------------------------------------
* Set up the solver
*----------------------------------------------------------*/
/* insure that the solver memory gets fully set up */
if ((relax_data -> setup_a_sol) > 0)
{
(relax_data -> setup_a_sol) = 2;
}
hypre_SMGRelaxSetup(relax_vdata, A, b, x);
zero_guess = (relax_data -> zero_guess);
stencil_dim = (relax_data -> stencil_dim);
temp_vec = (relax_data -> temp_vec);
A_sol = (relax_data -> A_sol);
A_rem = (relax_data -> A_rem);
residual_data = (relax_data -> residual_data);
solve_data = (relax_data -> solve_data);
/*----------------------------------------------------------
* Set zero values
*----------------------------------------------------------*/
if (zero_guess)
{
base_stride = (relax_data -> base_stride);
base_box_a = (relax_data -> base_box_array);
ierr = hypre_SMGSetStructVectorConstantValues(x, zero, base_box_a,
base_stride);
}
/*----------------------------------------------------------
* Iterate
*----------------------------------------------------------*/
for (k = 0; k < 2; k++)
{
switch(k)
{
/* Do pre-relaxation iterations */
case 0:
max_iter = 1;
num_spaces = (relax_data -> num_pre_spaces);
space_ranks = (relax_data -> pre_space_ranks);
break;
/* Do regular relaxation iterations */
case 1:
max_iter = (relax_data -> max_iter);
num_spaces = (relax_data -> num_reg_spaces);
space_ranks = (relax_data -> reg_space_ranks);
break;
}
for (i = 0; i < max_iter; i++)
{
for (j = 0; j < num_spaces; j++)
{
is = space_ranks[j];
hypre_SMGResidual(residual_data[is], A_rem, x, b, temp_vec);
if (stencil_dim > 2)
hypre_SMGSolve(solve_data[is], A_sol, temp_vec, x);
else
hypre_CyclicReduction(solve_data[is], A_sol, temp_vec, x);
}
(relax_data -> num_iterations) = (i + 1);
}
}
/*----------------------------------------------------------
* Free up memory according to memory_use parameter
*----------------------------------------------------------*/
if ((stencil_dim - 1) <= (relax_data -> memory_use))
{
hypre_SMGRelaxDestroyASol(relax_vdata);
}
hypre_EndTiming(relax_data -> time_index);
return ierr;
}
