Veselin Dobrev
27ffe35369
general/globals.[ch]. Renamed class WrappedOStream to OutStream and use a different implementation in order to support the case when the OutStream is passed as a parameter of type std::ostream. In class MPI_Session, added an MPI_Comm parameter to both constructors, with a default value of MPI_COMM_WORLD. Hide the global variable MFEM_COMM_WORLD and allow access (read and write) with global functions: GetGlobalMPI_Comm() and SetGlobalMPI_Comm() - in the future, we may want to perform some tasks when changing the "global" MPI communicator. In mfem_error(), use GetGlobalMPI_Comm() instead of MPI_COMM_WORLD. Minor formatting changes and #include reordering.
112 lines
2.3 KiB
C++
112 lines
2.3 KiB
C++
// Copyright (c) 2010, Lawrence Livermore National Security, LLC. Produced at
|
|
// the Lawrence Livermore National Laboratory. LLNL-CODE-443211. All Rights
|
|
// reserved. See file COPYRIGHT for details.
|
|
//
|
|
// This file is part of the MFEM library. For more information and source code
|
|
// availability see http://mfem.org.
|
|
//
|
|
// MFEM 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.
|
|
|
|
// Implementation of data types for sparse matrix smoothers
|
|
|
|
#include "vector.hpp"
|
|
#include "matrix.hpp"
|
|
#include "sparsemat.hpp"
|
|
#include "sparsesmoothers.hpp"
|
|
#include <iostream>
|
|
|
|
namespace mfem
|
|
{
|
|
|
|
void SparseSmoother::SetOperator(const Operator &a)
|
|
{
|
|
oper = dynamic_cast<const SparseMatrix*>(&a);
|
|
if (oper == NULL)
|
|
{
|
|
mfem_error("SparseSmoother::SetOperator : not a SparseMatrix!");
|
|
}
|
|
height = oper->Height();
|
|
width = oper->Width();
|
|
}
|
|
|
|
/// Matrix vector multiplication with GS Smoother.
|
|
void GSSmoother::Mult(const Vector &x, Vector &y) const
|
|
{
|
|
if (!iterative_mode)
|
|
{
|
|
y = 0.0;
|
|
}
|
|
for (int i = 0; i < iterations; i++)
|
|
{
|
|
if (type != 2)
|
|
{
|
|
oper->Gauss_Seidel_forw(x, y);
|
|
}
|
|
if (type != 1)
|
|
{
|
|
oper->Gauss_Seidel_back(x, y);
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Create the Jacobi smoother.
|
|
DSmoother::DSmoother(const SparseMatrix &a, int t, double s, int it)
|
|
: SparseSmoother(a)
|
|
{
|
|
type = t;
|
|
scale = s;
|
|
iterations = it;
|
|
}
|
|
|
|
/// Matrix vector multiplication with Jacobi smoother.
|
|
void DSmoother::Mult(const Vector &x, Vector &y) const
|
|
{
|
|
if (!iterative_mode && type == 0 && iterations == 1)
|
|
{
|
|
oper->DiagScale(x, y, scale);
|
|
return;
|
|
}
|
|
|
|
z.SetSize(width);
|
|
|
|
Vector *r = &y, *p = &z;
|
|
|
|
if (iterations % 2 == 0)
|
|
{
|
|
Swap<Vector*>(r, p);
|
|
}
|
|
|
|
if (!iterative_mode)
|
|
{
|
|
*p = 0.0;
|
|
}
|
|
else if (iterations % 2)
|
|
{
|
|
*p = y;
|
|
}
|
|
for (int i = 0; i < iterations; i++)
|
|
{
|
|
if (type == 0)
|
|
{
|
|
oper->Jacobi(x, *p, *r, scale);
|
|
}
|
|
else if (type == 1)
|
|
{
|
|
oper->Jacobi2(x, *p, *r, scale);
|
|
}
|
|
else if (type == 2)
|
|
{
|
|
oper->Jacobi3(x, *p, *r, scale);
|
|
}
|
|
else
|
|
{
|
|
mfem_error("DSmoother::Mult wrong type");
|
|
}
|
|
Swap<Vector*>(r, p);
|
|
}
|
|
}
|
|
|
|
}
|