axisym_linear_elasticity_elements.h

Go to the documentation of this file.

// LIC// ====================================================================
// LIC// This file forms part of oomph-lib, the object-oriented,
// LIC// multi-physics finite-element library, available
// LIC// at http://www.oomph-lib.org.
// LIC//
// LIC// Copyright (C) 2006-2022 Matthias Heil and Andrew Hazel
// LIC//
// LIC// This library is free software; you can redistribute it and/or
// LIC// modify it under the terms of the GNU Lesser General Public
// LIC// License as published by the Free Software Foundation; either
// LIC// version 2.1 of the License, or (at your option) any later version.
// LIC//
// LIC// This library is distributed in the hope that it will be useful,
// LIC// but WITHOUT ANY WARRANTY; without even the implied warranty of
// LIC// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
// LIC// Lesser General Public License for more details.
// LIC//
// LIC// You should have received a copy of the GNU Lesser General Public
// LIC// License along with this library; if not, write to the Free Software
// LIC// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
// LIC// 02110-1301  USA.
// LIC//
// LIC// The authors may be contacted at oomph-lib@maths.man.ac.uk.
// LIC//
// LIC//====================================================================
 
// Include guards to prevent multiple inclusion of the header
#ifndef OOMPH_AXISYMMETRIC_LINEAR_ELASTICITY_ELEMENTS_HEADER
#define OOMPH_AXISYMMETRIC_LINEAR_ELASTICITY_ELEMENTS_HEADER
 
// Config header generated by autoconfig
#ifdef HAVE_CONFIG_H
#include <oomph-lib-config.h>
#endif
 
 
#ifdef OOMPH_HAS_MPI
#include "mpi.h"
#endif
 
// OOMPH-LIB headers
#include "src/generic/Qelements.h"
#include "src/generic/Telements.h"
#include "src/generic/projection.h"
 
 
namespace oomph
{
  //=======================================================================
  //=======================================================================
  class AxisymmetricLinearElasticityEquationsBase : public virtual FiniteElement
  {
  public:
    virtual inline unsigned u_index_axisymmetric_linear_elasticity(
      const unsigned& i) const
    {
      return i;
    }
 
    double d2u_dt2_axisymmetric_linear_elasticity(const unsigned& n,
                                                  const unsigned& i) const
    {
      // Get the timestepper
      TimeStepper* time_stepper_pt = node_pt(n)->time_stepper_pt();
 
      // Storage for the derivative - initialise to 0
      double d2u_dt2 = 0.0;
 
      // If we are doing an unsteady solve then calculate the derivative
      if (!time_stepper_pt->is_steady())
      {
        // Get the nodal index
        const unsigned u_nodal_index =
          u_index_axisymmetric_linear_elasticity(i);
 
        // Get the number of values required to represent history
        const unsigned n_time = time_stepper_pt->ntstorage();
 
        // Loop over history values
        for (unsigned t = 0; t < n_time; t++)
        {
          // Add the contribution to the derivative
          d2u_dt2 +=
            time_stepper_pt->weight(2, t) * nodal_value(t, n, u_nodal_index);
        }
      }
 
      return d2u_dt2;
    }
 
 
    double du_dt_axisymmetric_linear_elasticity(const unsigned& n,
                                                const unsigned& i) const
    {
      // Get the timestepper
      TimeStepper* time_stepper_pt = node_pt(n)->time_stepper_pt();
 
      // Storage for the derivative - initialise to 0
      double du_dt = 0.0;
 
      // If we are doing an unsteady solve then calculate the derivative
      if (!time_stepper_pt->is_steady())
      {
        // Get the nodal index
        const unsigned u_nodal_index =
          u_index_axisymmetric_linear_elasticity(i);
 
        // Get the number of values required to represent history
        const unsigned n_time = time_stepper_pt->ntstorage();
 
        // Loop over history values
        for (unsigned t = 0; t < n_time; t++)
        {
          // Add the contribution to the derivative
          du_dt +=
            time_stepper_pt->weight(1, t) * nodal_value(t, n, u_nodal_index);
        }
      }
      return du_dt;
    }
 
    void interpolated_u_axisymmetric_linear_elasticity(
      const Vector<double>& s, Vector<double>& disp) const
    {
      // Find number of nodes
      unsigned n_node = nnode();
 
      // Local shape function
      Shape psi(n_node);
 
      // Find values of shape function
      shape(s, psi);
 
      for (unsigned i = 0; i < 3; i++)
      {
        // Index at which the nodal value is stored
        unsigned u_nodal_index = u_index_axisymmetric_linear_elasticity(i);
 
        // Initialise value of u
        disp[i] = 0.0;
 
        // Loop over the local nodes and sum
        for (unsigned l = 0; l < n_node; l++)
        {
          const double u_value = nodal_value(l, u_nodal_index);
 
          disp[i] += u_value * psi[l];
        }
      }
    }
 
    double interpolated_u_axisymmetric_linear_elasticity(
      const Vector<double>& s, const unsigned& i) const
    {
      // Find number of nodes
      unsigned n_node = nnode();
 
      // Local shape function
      Shape psi(n_node);
 
      // Find values of shape function
      shape(s, psi);
 
      // Get nodal index at which i-th velocity is stored
      unsigned u_nodal_index = u_index_axisymmetric_linear_elasticity(i);
 
      // Initialise value of u
      double interpolated_u = 0.0;
 
      // Loop over the local nodes and sum
      for (unsigned l = 0; l < n_node; l++)
      {
        const double u_value = nodal_value(l, u_nodal_index);
 
        interpolated_u += u_value * psi[l];
      }
 
      return (interpolated_u);
    }
 
 
    void interpolated_du_dt_axisymmetric_linear_elasticity(
      const Vector<double>& s, Vector<double>& du_dt) const
    {
      // Find number of nodes
      unsigned n_node = nnode();
 
      // Local shape function
      Shape psi(n_node);
 
      // Find values of shape function
      shape(s, psi);
 
      // Loop over directions
      for (unsigned i = 0; i < 3; i++)
      {
        // Initialise value of u
        du_dt[i] = 0.0;
 
        // Loop over the local nodes and sum
        for (unsigned l = 0; l < n_node; l++)
        {
          du_dt[i] += du_dt_axisymmetric_linear_elasticity(l, i) * psi[l];
        }
      }
    }
 
    void interpolated_d2u_dt2_axisymmetric_linear_elasticity(
      const Vector<double>& s, Vector<double>& d2u_dt2) const
    {
      // Find number of nodes
      unsigned n_node = nnode();
 
      // Local shape function
      Shape psi(n_node);
 
      // Find values of shape function
      shape(s, psi);
 
      // Loop over directions
      for (unsigned i = 0; i < 3; i++)
      {
        // Initialise value of u
        d2u_dt2[i] = 0.0;
 
        // Loop over the local nodes and sum
        for (unsigned l = 0; l < n_node; l++)
        {
          d2u_dt2[i] += d2u_dt2_axisymmetric_linear_elasticity(l, i) * psi[l];
        }
      }
    }
 
    typedef void (*BodyForceFctPt)(const double& time,
                                   const Vector<double>& x,
                                   Vector<double>& b);
 
    AxisymmetricLinearElasticityEquationsBase()
      : Youngs_modulus_pt(&Default_youngs_modulus_value),
        Nu_pt(0),
        Lambda_sq_pt(&Default_lambda_sq_value),
        Body_force_fct_pt(0)
    {
    }
 
    double*& youngs_modulus_pt()
    {
      return Youngs_modulus_pt;
    }
 
    inline double youngs_modulus() const
    {
      return (*Youngs_modulus_pt);
    }
 
    double& nu() const
    {
#ifdef PARANOID
      if (Nu_pt == 0)
      {
        std::ostringstream error_message;
        error_message << "No pointer to Poisson's ratio set. Please set one!\n";
        throw OomphLibError(error_message.str(),
                            OOMPH_CURRENT_FUNCTION,
                            OOMPH_EXCEPTION_LOCATION);
      }
#endif
      return *Nu_pt;
    }
 
    double*& nu_pt()
    {
      return Nu_pt;
    }
 
    double*& lambda_sq_pt()
    {
      return Lambda_sq_pt;
    }
 
    const double& lambda_sq() const
    {
      return *Lambda_sq_pt;
    }
 
    BodyForceFctPt& body_force_fct_pt()
    {
      return Body_force_fct_pt;
    }
 
    BodyForceFctPt body_force_fct_pt() const
    {
      return Body_force_fct_pt;
    }
 
    inline void body_force(const double& time,
                           const Vector<double>& x,
                           Vector<double>& b) const
    {
      // If no function has been set, return zero vector
      if (Body_force_fct_pt == 0)
      {
        // Get spatial dimension of element
        unsigned n = dim();
        for (unsigned i = 0; i < n; i++)
        {
          b[i] = 0.0;
        }
      }
      else
      {
        (*Body_force_fct_pt)(time, x, b);
      }
    }
 
    unsigned ndof_types() const
    {
      return 1;
    }
 
    void get_dof_numbers_for_unknowns(
      std::list<std::pair<unsigned long, unsigned>>& dof_lookup_list) const
    {
      // temporary pair (used to store DOF lookup prior to being added
      // to list)
      std::pair<unsigned long, unsigned> dof_lookup;
 
      // number of nodes
      const unsigned n_node = this->nnode();
 
      // Integer storage for local unknown
      int local_unknown = 0;
 
      // Loop over the nodes
      for (unsigned n = 0; n < n_node; n++)
      {
        // Loop over dimension
        for (unsigned i = 0; i < 3; i++)
        {
          // If the variable is free
          local_unknown = nodal_local_eqn(n, i);
 
          // ignore pinned values
          if (local_unknown >= 0)
          {
            // store DOF type lookup in temporary pair: First entry in pair
            // is global equation number; second entry is DOF type
            dof_lookup.first = this->eqn_number(local_unknown);
            dof_lookup.second = 0;
 
            // add to list
            dof_lookup_list.push_front(dof_lookup);
          }
        }
      }
    }
 
 
  protected:
    double* Youngs_modulus_pt;
 
    double* Nu_pt;
 
    double* Lambda_sq_pt;
 
    BodyForceFctPt Body_force_fct_pt;
 
    static double Default_youngs_modulus_value;
 
    static double Default_lambda_sq_value;
  };
 
 
 
 
  //=======================================================================
  //=======================================================================
  class AxisymmetricLinearElasticityEquations
    : public AxisymmetricLinearElasticityEquationsBase
  {
  public:
    AxisymmetricLinearElasticityEquations() {}
 
    unsigned required_nvalue(const unsigned& n) const
    {
      return 3;
    }
 
    void fill_in_contribution_to_residuals(Vector<double>& residuals)
    {
      fill_in_generic_contribution_to_residuals_axisymmetric_linear_elasticity(
        residuals, GeneralisedElement::Dummy_matrix, 0);
    }
 
 
    void fill_in_contribution_to_jacobian(Vector<double>& residuals,
                                          DenseMatrix<double>& jacobian)
    {
      // Add the contribution to the residuals
      this
        ->fill_in_generic_contribution_to_residuals_axisymmetric_linear_elasticity(
          residuals, jacobian, 1);
    }
 
 
    void get_strain(const Vector<double>& s, DenseMatrix<double>& strain);
 
    void output_fct(std::ostream& outfile,
                    const unsigned& nplot,
                    FiniteElement::SteadyExactSolutionFctPt exact_soln_pt);
 
    void output_fct(std::ostream& outfile,
                    const unsigned& nplot,
                    const double& time,
                    FiniteElement::UnsteadyExactSolutionFctPt exact_soln_pt);
 
    void output(std::ostream& outfile)
    {
      unsigned n_plot = 5;
      output(outfile, n_plot);
    }
 
    void output(std::ostream& outfile, const unsigned& n_plot);
 
    void output(FILE* file_pt)
    {
      unsigned n_plot = 5;
      output(file_pt, n_plot);
    }
 
    void output(FILE* file_pt, const unsigned& n_plot);
 
    void compute_error(std::ostream& outfile,
                       FiniteElement::SteadyExactSolutionFctPt exact_soln_pt,
                       double& error,
                       double& norm);
 
    void compute_error(std::ostream& outfile,
                       FiniteElement::UnsteadyExactSolutionFctPt exact_soln_pt,
                       const double& time,
                       double& error,
                       double& norm);
 
 
  protected:
    virtual void fill_in_generic_contribution_to_residuals_axisymmetric_linear_elasticity(
      Vector<double>& residuals, DenseMatrix<double>& jacobian, unsigned flag);
  };
 
 
 
 
  //===========================================================================
  //============================================================================
  template<unsigned NNODE_1D>
  class QAxisymmetricLinearElasticityElement
    : public virtual QElement<2, NNODE_1D>,
      public virtual AxisymmetricLinearElasticityEquations
  {
  public:
    QAxisymmetricLinearElasticityElement()
      : QElement<2, NNODE_1D>(), AxisymmetricLinearElasticityEquations()
    {
    }
 
    void output(std::ostream& outfile)
    {
      AxisymmetricLinearElasticityEquations::output(outfile);
    }
 
    void output(std::ostream& outfile, const unsigned& n_plot)
    {
      AxisymmetricLinearElasticityEquations::output(outfile, n_plot);
    }
 
 
    void output(FILE* file_pt)
    {
      AxisymmetricLinearElasticityEquations::output(file_pt);
    }
 
    void output(FILE* file_pt, const unsigned& n_plot)
    {
      AxisymmetricLinearElasticityEquations::output(file_pt, n_plot);
    }
  };
 
 
  //============================================================================
  //============================================================================
  template<unsigned NNODE_1D>
  class FaceGeometry<QAxisymmetricLinearElasticityElement<NNODE_1D>>
    : public virtual QElement<1, NNODE_1D>
  {
  public:
    FaceGeometry() : QElement<1, NNODE_1D>() {}
  };
 
 
  /* //////////////////////////////////////////////////////////////////////// */
  /* //////////////////////////////////////////////////////////////////////// */
  /* //////////////////////////////////////////////////////////////////////// */
 
 
  /* //===========================================================================
   */
  /* /// An Element that solves the equations of axisymmetric (in cylindrical */
  /* /// polars) linear elasticity, using TElements for the geometry. */
  /* //============================================================================
   */
  /*   template<unsigned NNODE_1D> */
  /*    class TAxisymmetricLinearElasticityElement :  */
  /*   public virtual TElement<2,NNODE_1D>, */
  /*    public virtual AxisymmetricLinearElasticityEquations */
  /*    { */
  /*     public: */
 
  /*    /// Constructor */
  /*    TAxisymmetricLinearElasticityElement() :  */
  /*     TElement<2,NNODE_1D>(),  */
  /*     AxisymmetricLinearElasticityEquations() { } */
 
  /*     /// Output function */
  /*    void output(std::ostream &outfile)  */
  /*    {AxisymmetricLinearElasticityEquations::output(outfile);} */
 
  /*    /// Output function */
  /*    void output(std::ostream &outfile, const unsigned &n_plot) */
  /*    {AxisymmetricLinearElasticityEquations:: */
  /*      output(outfile,n_plot);} */
 
  /*    /// C-style output function */
  /*    void output(FILE* file_pt)  */
  /*    {AxisymmetricLinearElasticityEquations::output(file_pt);} */
 
  /*    /// C-style output function */
  /*    void output(FILE* file_pt, const unsigned &n_plot) */
  /*    {AxisymmetricLinearElasticityEquations:: */
  /*      output(file_pt,n_plot);} */
 
  /*   }; */
 
 
  /* //============================================================================
   */
  /* /// FaceGeometry of a linear  */
  /* /// TAxisymmetricLinearElasticityElement element */
  /* //============================================================================
   */
  /*  template<unsigned NNODE_1D> */
  /*   class FaceGeometry<TAxisymmetricLinearElasticityElement<NNODE_1D> > : */
  /*  public virtual TElement<1,NNODE_1D> */
  /*   { */
  /*     public: */
  /*    /// Constructor must call the constructor of the underlying element */
  /*     FaceGeometry() : TElement<1,NNODE_1D>() {} */
  /*   }; */
 
 
 
 
  //==========================================================
  //==========================================================
  template<class AXISYM_LINEAR_ELAST_ELEMENT>
  class ProjectableAxisymLinearElasticityElement
    : public virtual ProjectableElement<AXISYM_LINEAR_ELAST_ELEMENT>
  {
  public:
    ProjectableAxisymLinearElasticityElement() {}
 
 
    Vector<std::pair<Data*, unsigned>> data_values_of_field(const unsigned& fld)
    {
      // Create the vector
      Vector<std::pair<Data*, unsigned>> data_values;
 
      // Loop over all nodes and extract the fld-th nodal value
      unsigned nnod = this->nnode();
      for (unsigned j = 0; j < nnod; j++)
      {
        // Add the data value associated with the displacement components
        data_values.push_back(std::make_pair(this->node_pt(j), fld));
      }
 
      // Return the vector
      return data_values;
    }
 
    unsigned nfields_for_projection()
    {
      return 3;
    }
 
    unsigned nhistory_values_for_projection(const unsigned& fld)
    {
#ifdef PARANOID
      if (fld > 2)
      {
        std::stringstream error_stream;
        error_stream << "Elements only store two fields so fld can't be"
                     << " " << fld << std::endl;
        throw OomphLibError(
          error_stream.str(), OOMPH_CURRENT_FUNCTION, OOMPH_EXCEPTION_LOCATION);
      }
#endif
      return this->node_pt(0)->ntstorage();
    }
 
    unsigned nhistory_values_for_coordinate_projection()
    {
      return this->node_pt(0)->position_time_stepper_pt()->ntstorage();
    }
 
    double jacobian_and_shape_of_field(const unsigned& fld,
                                       const Vector<double>& s,
                                       Shape& psi)
    {
      unsigned n_dim = this->dim();
      unsigned n_node = this->nnode();
      DShape dpsidx(n_node, n_dim);
 
      // Call the derivatives of the shape functions and return
      // the Jacobian
      return this->dshape_eulerian(s, psi, dpsidx);
    }
 
 
    double get_field(const unsigned& t,
                     const unsigned& fld,
                     const Vector<double>& s)
    {
      unsigned n_node = this->nnode();
 
      // Local shape function
      Shape psi(n_node);
 
      // Find values of shape function
      this->shape(s, psi);
 
      // Initialise value of u
      double interpolated_u = 0.0;
 
      // Sum over the local nodes at that time
      for (unsigned l = 0; l < n_node; l++)
      {
        interpolated_u += this->nodal_value(t, l, fld) * psi[l];
      }
      return interpolated_u;
    }
 
 
    unsigned nvalue_of_field(const unsigned& fld)
    {
      return this->nnode();
    }
 
 
    int local_equation(const unsigned& fld, const unsigned& j)
    {
      return this->nodal_local_eqn(j, fld);
    }
  };
 
 
  //=======================================================================
  //=======================================================================
  template<class ELEMENT>
  class FaceGeometry<ProjectableAxisymLinearElasticityElement<ELEMENT>>
    : public virtual FaceGeometry<ELEMENT>
  {
  public:
    FaceGeometry() : FaceGeometry<ELEMENT>() {}
  };
 
 
  //=======================================================================
  //=======================================================================
  template<class ELEMENT>
  class FaceGeometry<
    FaceGeometry<ProjectableAxisymLinearElasticityElement<ELEMENT>>>
    : public virtual FaceGeometry<FaceGeometry<ELEMENT>>
  {
  public:
    FaceGeometry() : FaceGeometry<FaceGeometry<ELEMENT>>() {}
  };
 
 
} // namespace oomph
 
 
#endif