AbstractPdeSolver.cpp

Go to the documentation of this file.

/*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
   This file is part of CEPS.
 
   CEPS is free software: you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation, either version 3 of the License, or
   (at your option) any later version.
 
   CEPS 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 General Public License for more details.
 
   You should have received a copy of the GNU General Public License
   along with CEPS (see file LICENSE at root of project).
   If not, see <https://www.gnu.org/licenses/>.
 
 
   Copyright 2019-2024 Inria, Universite de Bordeaux
 
   Authors, in alphabetical order:
 
     Pierre-Elliott BECUE, Florian CARO, Yves COUDIERE(*), Andjela DAVIDOVIC, 
     Charlie DOUANLA-LONTSI, Marc FUENTES, Mehdi JUHOOR, Michael LEGUEBE(*), 
     Pauline MIGERDITICHAN, Valentin PANNETIER(*), Nejib ZEMZEMI.
     * : currently active authors
 
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*/
#include "pde/solver/AbstractPdeSolver.hpp"
#include "pde/problem/AbstractPdeProblem.hpp"
#include "pde/assemblers/AbstractAssembler.hpp"
 
AbstractPdeSolver::AbstractPdeSolver(AbstractPdeProblem* pb):
  CepsObject(),
  m_problem       (pb),
  m_discretization(nullptr),
  m_linearSystem  (ceps::getNew<LinearSystem>()),
  m_opAsb         (nullptr),
  m_bcAsb         (nullptr),
  m_lapSrcAsb     (nullptr),
  m_hasRegSrc     (false),
  m_hasLapSrc     (false),
  m_useMassLumping(false),
  m_doOutput      (true),
  m_writer        (nullptr),
  m_timeWriter    (nullptr),
  m_doError       (false),
  m_errors        (nullptr)
{
  CEPS_ABORT_IF(ceps::isNullPtr(pb),
    "Passed pointer on PDE problem is null"
  );
 
  m_discretization = pb->getSpatialDiscretization();
 
  if (pb->getParameters())
  {
    AbstractPdeSolver::setupWithParameters(pb->getParameters());
    m_linearSystem->setupWithParameters(pb->getParameters());
  }
 
  m_hasRegSrc      = m_problem->getSourceTermManager()->getNbOfSourceTermsOf(
      CepsSourceTermFlag::Default | CepsSourceTermFlag::Stimulation) != 0;
  m_hasLapSrc = m_problem->getSourceTermManager()->getNbOfSourceTermsOf(
      CepsSourceTermFlag::Laplace) != 0;
 
  // Link matrix and vector to linear system
  m_rhs = m_discretization->newDofVector();
  m_lhs = m_discretization->newDofMatrix();
  m_linearSystem->setRhsVector(m_rhs);
  m_linearSystem->setLhsMatrix(m_lhs);
 
  // Matrix of the operator
  m_opMat = m_discretization->newDofMatrix();
  // Vector of boundary conditions
  m_bcMat = m_discretization->newDofMatrix();
  m_bcVec = m_discretization->newDofHaloVector();
  // Matrix for laplace source terms
  m_lapSrcMat = m_discretization->newDofMatrix();
}
 
AbstractPdeSolver::~AbstractPdeSolver()
{
  ceps::destroyObject(m_linearSystem);
  ceps::destroyObject(m_writer);
  ceps::destroyObject(m_errors);
  ceps::destroyObject(m_opAsb);
  ceps::destroyObject(m_bcAsb);
  ceps::destroyObject(m_lapSrcAsb);
  ceps::destroyObject(m_timeWriter);
}
 
 
// -----------------------------------------------------------------------------------------
 
void
AbstractPdeSolver::disableOutput()
{
  m_doOutput = false;
}
 
void
AbstractPdeSolver::enableOutput()
{
  m_doOutput = true;
}
 
TimeWriter*
AbstractPdeSolver::getTimeWriter() const
{
  return m_timeWriter;
}
 
// -----------------------------------------------------------------------------------------
 
void
AbstractPdeSolver::disableErrorComputation()
{
  m_doError = false;
}
 
 
// -----------------------------------------------------------------------------------------
 
void
AbstractPdeSolver::setupWithParameters(InputParameters* params)
{
  m_useMassLumping = params->isActiveOption("use mass lumping");
  m_doOutput       = not params->isActiveOption("disable output");
}
 
void
AbstractPdeSolver::initializeWriter()
{
  m_writer = ceps::getNew<VtkWriter>(
    m_problem->getSpatialDiscretization(),
    m_problem->getOutputFileBase(),
    getExpectedNumberOfOutputs(),
    m_problem->getOutputFormat(),
    m_problem->writesGlobalIndices()
  );
}