newmark_solver.h

Go to the documentation of this file.

// The libMesh Finite Element Library.
// Copyright (C) 2002-2018 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



#ifndef LIBMESH_NEWMARK_SOLVER_H
#define LIBMESH_NEWMARK_SOLVER_H

// Local includes
#include "libmesh/second_order_unsteady_solver.h"

namespace libMesh
{
class NewmarkSolver : public SecondOrderUnsteadySolver
{
public:
  typedef UnsteadySolver Parent;

  explicit
  NewmarkSolver (sys_type & s);

  virtual ~NewmarkSolver ();

  virtual void advance_timestep () override;

  virtual void adjoint_advance_timestep () override;

  virtual void compute_initial_accel();

  void project_initial_accel (FunctionBase<Number> * f,
                              FunctionBase<Gradient> * g = nullptr);

  void set_initial_accel_avail (bool initial_accel_set);

  virtual Real error_order() const override;

  virtual void solve () override;

  virtual bool element_residual (bool request_jacobian,
                                 DiffContext &) override;

  virtual bool side_residual (bool request_jacobian,
                              DiffContext &) override;

  virtual bool nonlocal_residual (bool request_jacobian,
                                  DiffContext &) override;


  void set_beta ( Real beta )
  { _beta = beta; }

  void set_gamma ( Real gamma )
  { _gamma = gamma; }

protected:

  Real _beta;

  Real _gamma;

  bool _is_accel_solve;


  bool _initial_accel_set;

  virtual bool _general_residual (bool request_jacobian,
                                  DiffContext &,
                                  ResFuncType mass,
                                  ResFuncType damping,
                                  ResFuncType time_deriv,
                                  ResFuncType constraint,
                                  ReinitFuncType reinit);
};

} // namespace libMesh

#endif // LIBMESH_NEWMARK_SOLVER_H