Homec4science
Diffusion tamaas (features/netcdf-mpi) Browse

tamaas/src/model/westergaard.hh
e722836a68b2features/netcdf-mpi

Last Change
Raw File

westergaard.hh
View Options

/**
* @file
*
* @author Lucas Frérot <lucas.frerot@epfl.ch>
*
* @section LICENSE
*
* Copyright (©) 2017 EPFL (Ecole Polytechnique Fédérale de
* Lausanne) Laboratory (LSMS - Laboratoire de Simulation en Mécanique des
* Solides)
*
* Tamaas 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, either version 3 of the License, or (at your option) any
* later version.
*
* Tamaas is distributed in the hope that it will be useful, but WITHOUT ANY
* WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
* A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
* details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with Tamaas. If not, see <http://www.gnu.org/licenses/>.
*
*/
/* -------------------------------------------------------------------------- */
#ifndef __WESTERGAARD_HH__
#define __WESTERGAARD_HH__
/* -------------------------------------------------------------------------- */
#include "be_engine.hh"
#include "grid_hermitian.hh"
#include "fft_plan_manager.hh"
#include "model_type.hh"
#include "tamaas.hh"
/* -------------------------------------------------------------------------- */
__BEGIN_TAMAAS__
/**
* @brief Engine based on Westergaard solution and the Dicrete Fourier Transofrm
* This class is templated with model type to allow efficient storage of the
* influence coefficients.
*/
template <model_type type>
class Westergaard : public BEEngine {
using trait = model_type_traits<type>;
public:
/// Constructor
Westergaard(Model* model);
/// Destructor
virtual ~Westergaard() = default;
public:
/// Initialize influence coefficients
void initInfluence() override;
/// Solve Neumann problem
void solveNeumann(GridBase<Real>& neumann,
GridBase<Real>& dirichlet) const override;
/// Solve Dirichlet problem
void solveDirichlet(GridBase<Real>& dirichlet,
GridBase<Real>& neumann) const override;
/// Get influence coefficients
const GridHermitian<Real, trait::dimension>& getInfluence() const {
return influence;
}
template <typename Functor>
void fourierApply(Functor func, GridBase<Real>& in,
GridBase<Real>& out) const;
public:
GridHermitian<Real, trait::dimension> influence;
mutable GridHermitian<Real, trait::boundary_dimension> buffer;
};
/* -------------------------------------------------------------------------- */
/* Template function implementation */
/* -------------------------------------------------------------------------- */
template <model_type type>
template <typename Functor>
void Westergaard<type>::fourierApply(Functor func, GridBase<Real>& in,
GridBase<Real>& out) const try {
Grid<Real, trait::boundary_dimension>& i = dynamic_cast<decltype(i)>(in);
Grid<Real, trait::boundary_dimension>& o = dynamic_cast<decltype(o)>(out);
auto& fwd = FFTPlanManager::get().createPlan(i, buffer);
auto& bwd = FFTPlanManager::get().createPlan(o, buffer);
fwd.forwardTransform();
func(buffer, influence);
bwd.backwardTransform();
} catch (const std::bad_cast& e) {
TAMAAS_EXCEPTION("Neumann and dirichlet types do not match model type");
}
__END_TAMAAS__
#endif // __WESTERGAARD_HH__
c4science · Help