transient_system.C

Go to the documentation of this file.

// The libMesh Finite Element Library.
// Copyright (C) 2002-2024 Benjamin S. Kirk, John W. Peterson, Roy H. Stogner

// This library 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 2.1 of the License, or (at your option) any later version.

// This library 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 this library; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA



// C++ includes

// Local includes
#include "libmesh/transient_system.h"
#include "libmesh/explicit_system.h"
#include "libmesh/linear_implicit_system.h"
#include "libmesh/nonlinear_implicit_system.h"
#include "libmesh/dof_map.h"
#include "libmesh/numeric_vector.h"
#include "libmesh/rb_construction.h"
#include "libmesh/eigen_system.h"

// Nvidia C++ whining about destroying incomplete unique_ptr<T> Base::foo types
#include "libmesh/diff_solver.h"
#include "libmesh/nonlinear_solver.h"
#include "libmesh/shell_matrix.h"
#include "libmesh/sparse_matrix.h"

namespace libMesh
{


// ------------------------------------------------------------
// TransientSystem implementation
template <class Base>
TransientSystem<Base>::TransientSystem (EquationSystems & es,
                                        const std::string & name_in,
                                        const unsigned int number_in) :

  Base(es, name_in, number_in),
  old_local_solution(nullptr),
  older_local_solution(nullptr)
{
  this->add_old_vectors();
}



template <class Base>
TransientSystem<Base>::~TransientSystem () = default;



template <class Base>
void TransientSystem<Base>::clear ()
{
  // clear the parent data
  Base::clear();

  // Restore us to a "basic" state
  this->add_old_vectors();
}



template <class Base>
void TransientSystem<Base>::re_update ()
{
  // re_update the parent system
  Base::re_update ();

  const std::vector<dof_id_type> & send_list = this->get_dof_map().get_send_list ();

  const dof_id_type first_local_dof = Base::get_dof_map().first_dof();
  const dof_id_type end_local_dof  = Base::get_dof_map().end_dof();

  // Check sizes
  libmesh_assert_greater_equal (end_local_dof, first_local_dof);
  libmesh_assert_greater_equal (older_local_solution->size(), send_list.size());
  libmesh_assert_greater_equal (old_local_solution->size(), send_list.size());

  // Even if we don't have to do anything ourselves, localize() may
  // use parallel_only tools
  // if (first_local_dof == end_local_dof)
  //   return;

  // Update the old & older solutions with the send_list,
  // which may have changed since their last update.
  older_local_solution->localize (first_local_dof,
                                  end_local_dof-1,
                                  send_list);

  old_local_solution->localize (first_local_dof,
                                end_local_dof-1,
                                send_list);
}




template <class Base>
Number TransientSystem<Base>::old_solution (const dof_id_type global_dof_number) const
{
  // Check the sizes
  libmesh_assert_less (global_dof_number, this->get_dof_map().n_dofs());
  libmesh_assert_less (global_dof_number, old_local_solution->size());

  return (*old_local_solution)(global_dof_number);
}



template <class Base>
Number TransientSystem<Base>::older_solution (const dof_id_type global_dof_number) const
{
  // Check the sizes
  libmesh_assert_less (global_dof_number, this->get_dof_map().n_dofs());
  libmesh_assert_less (global_dof_number, older_local_solution->size());

  return (*older_local_solution)(global_dof_number);
}



template <class Base>
void TransientSystem<Base>::add_old_vectors()
{
  ParallelType type =
#ifdef LIBMESH_ENABLE_GHOSTED
    GHOSTED;
#else
    SERIAL;
#endif

  old_local_solution = &(this->add_vector("_transient_old_local_solution", true, type));
  older_local_solution = &(this->add_vector("_transient_older_local_solution", true, type));
}

// ------------------------------------------------------------
// TransientSystem instantiations
template class LIBMESH_EXPORT TransientSystem<LinearImplicitSystem>;
template class LIBMESH_EXPORT TransientSystem<NonlinearImplicitSystem>;
template class LIBMESH_EXPORT TransientSystem<ExplicitSystem>;
template class LIBMESH_EXPORT TransientSystem<System>;
template class LIBMESH_EXPORT TransientSystem<RBConstruction>;
#ifdef LIBMESH_HAVE_SLEPC
template class LIBMESH_EXPORT TransientSystem<EigenSystem>;
#endif

} // namespace libMesh