refineable_advection_diffusion_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//====================================================================
// Header file for elements that solve the advection diffusion equation
// and that can be refined.
 
#ifndef OOMPH_REFINEABLE_ADVECTION_DIFFUSION_ELEMENTS_HEADER
#define OOMPH_REFINEABLE_ADVECTION_DIFFUSION_ELEMENTS_HEADER
 
// Config header generated by autoconfig
#ifdef HAVE_CONFIG_H
#include <oomph-lib-config.h>
#endif
 
// oomph-lib headers
#include "../generic/refineable_quad_element.h"
#include "../generic/refineable_brick_element.h"
#include "../generic/error_estimator.h"
#include "advection_diffusion_elements.h"
 
namespace oomph
{
  //======================================================================
  //======================================================================
  template<unsigned DIM>
  class RefineableAdvectionDiffusionEquations
    : public virtual AdvectionDiffusionEquations<DIM>,
      public virtual RefineableElement,
      public virtual ElementWithZ2ErrorEstimator
  {
  public:
    RefineableAdvectionDiffusionEquations()
      : AdvectionDiffusionEquations<DIM>(),
        RefineableElement(),
        ElementWithZ2ErrorEstimator()
    {
    }
 
 
    RefineableAdvectionDiffusionEquations(
      const RefineableAdvectionDiffusionEquations<DIM>& dummy) = delete;
 
    void operator=(const RefineableAdvectionDiffusionEquations<DIM>&) = delete;
 
    unsigned num_Z2_flux_terms()
    {
      return DIM;
    }
 
    void get_Z2_flux(const Vector<double>& s, Vector<double>& flux)
    {
      this->get_flux(s, flux);
    }
 
 
    void get_interpolated_values(const Vector<double>& s,
                                 Vector<double>& values)
    {
      // Set size of Vector: u
      values.resize(1);
 
      // Find number of nodes
      unsigned n_node = nnode();
 
      // Find the index at which the unknown is stored
      unsigned u_nodal_index = this->u_index_adv_diff();
 
      // Local shape function
      Shape psi(n_node);
 
      // Find values of shape function
      shape(s, psi);
 
      // Initialise value of u
      values[0] = 0.0;
 
      // Loop over the local nodes and sum
      for (unsigned l = 0; l < n_node; l++)
      {
        values[0] += this->nodal_value(l, u_nodal_index) * psi[l];
      }
    }
 
    void get_interpolated_values(const unsigned& t,
                                 const Vector<double>& s,
                                 Vector<double>& values)
    {
      // Set size of Vector: u
      values.resize(1);
 
      // Find number of nodes
      const unsigned n_node = nnode();
 
      // Find the index at which the unknown is stored
      const unsigned u_nodal_index = this->u_index_adv_diff();
 
      // Local shape function
      Shape psi(n_node);
 
      // Find values of shape function
      shape(s, psi);
 
      // Initialise value of u
      values[0] = 0.0;
 
      // Loop over the local nodes and sum
      for (unsigned l = 0; l < n_node; l++)
      {
        values[0] += this->nodal_value(t, l, u_nodal_index) * psi[l];
      }
    }
 
    /*if (t!=0)
       {
        std::string error_message =
         "Time-dependent version of get_interpolated_values() ";
        error_message += "not implemented for this element \n";
        throw
         OomphLibError(
          error_message,
          "RefineableAdvectionDiffusionEquations::get_interpolated_values()",
          OOMPH_EXCEPTION_LOCATION);
       }
      else
       {
        //Make sure that we call the appropriate steady version
        //(the entire function might be overloaded lower down)
        RefineableAdvectionDiffusionEquations<DIM>::
         get_interpolated_values(s,values);
       }
       }*/
 
 
    void further_build()
    {
      RefineableAdvectionDiffusionEquations<DIM>* cast_father_element_pt =
        dynamic_cast<RefineableAdvectionDiffusionEquations<DIM>*>(
          this->father_element_pt());
 
      // Set the values of the pointers from the father
      this->Source_fct_pt = cast_father_element_pt->source_fct_pt();
      this->Wind_fct_pt = cast_father_element_pt->wind_fct_pt();
      this->Pe_pt = cast_father_element_pt->pe_pt();
      this->PeSt_pt = cast_father_element_pt->pe_st_pt();
 
      // Set the ALE status
      this->ALE_is_disabled = cast_father_element_pt->ALE_is_disabled;
    }
 
    void dinterpolated_u_adv_diff_ddata(const Vector<double>& s,
                                        Vector<double>& du_ddata,
                                        Vector<unsigned>& global_eqn_number)
    {
      // Find number of nodes
      unsigned n_node = this->nnode();
      // Local shape function
      Shape psi(n_node);
      // Find values of shape function at the given local coordinate
      this->shape(s, psi);
 
      // Find the index at which the velocity component is stored
      const unsigned u_nodal_index = this->u_index_adv_diff();
 
      // Storage for hang info pointer
      HangInfo* hang_info_pt = 0;
      // Storage for global equation
      int global_eqn = 0;
 
      // Find the number of dofs associated with interpolated u
      unsigned n_u_dof = 0;
      for (unsigned l = 0; l < n_node; l++)
      {
        unsigned n_master = 1;
 
        // Local bool (is the node hanging)
        bool is_node_hanging = this->node_pt(l)->is_hanging();
 
        // If the node is hanging, get the number of master nodes
        if (is_node_hanging)
        {
          hang_info_pt = this->node_pt(l)->hanging_pt();
          n_master = hang_info_pt->nmaster();
        }
        // Otherwise there is just one master node, the node itself
        else
        {
          n_master = 1;
        }
 
        // Loop over the master nodes
        for (unsigned m = 0; m < n_master; m++)
        {
          // Get the equation number
          if (is_node_hanging)
          {
            // Get the equation number from the master node
            global_eqn =
              hang_info_pt->master_node_pt(m)->eqn_number(u_nodal_index);
          }
          else
          {
            // Global equation number
            global_eqn = this->node_pt(l)->eqn_number(u_nodal_index);
          }
 
          // If it's positive add to the count
          if (global_eqn >= 0)
          {
            ++n_u_dof;
          }
        }
      }
 
      // Now resize the storage schemes
      du_ddata.resize(n_u_dof, 0.0);
      global_eqn_number.resize(n_u_dof, 0);
 
      // Loop over th nodes again and set the derivatives
      unsigned count = 0;
      // Loop over the local nodes and sum
      for (unsigned l = 0; l < n_node; l++)
      {
        unsigned n_master = 1;
        double hang_weight = 1.0;
 
        // Local bool (is the node hanging)
        bool is_node_hanging = this->node_pt(l)->is_hanging();
 
        // If the node is hanging, get the number of master nodes
        if (is_node_hanging)
        {
          hang_info_pt = this->node_pt(l)->hanging_pt();
          n_master = hang_info_pt->nmaster();
        }
        // Otherwise there is just one master node, the node itself
        else
        {
          n_master = 1;
        }
 
        // Loop over the master nodes
        for (unsigned m = 0; m < n_master; m++)
        {
          // If the node is hanging get weight from master node
          if (is_node_hanging)
          {
            // Get the hang weight from the master node
            hang_weight = hang_info_pt->master_weight(m);
          }
          else
          {
            // Node contributes with full weight
            hang_weight = 1.0;
          }
 
          // Get the equation number
          if (is_node_hanging)
          {
            // Get the equation number from the master node
            global_eqn =
              hang_info_pt->master_node_pt(m)->eqn_number(u_nodal_index);
          }
          else
          {
            // Global equation number
            global_eqn = this->node_pt(l)->eqn_number(u_nodal_index);
          }
 
          if (global_eqn >= 0)
          {
            // Set the global equation number
            global_eqn_number[count] = global_eqn;
            // Set the derivative with respect to the unknown
            du_ddata[count] = psi[l] * hang_weight;
            // Increase the counter
            ++count;
          }
        }
      }
    }
 
 
  protected:
    void fill_in_generic_residual_contribution_adv_diff(
      Vector<double>& residuals,
      DenseMatrix<double>& jacobian,
      DenseMatrix<double>& mass_matrix,
      unsigned flag);
  };
 
 
  //======================================================================
  //======================================================================
  template<unsigned DIM, unsigned NNODE_1D>
  class RefineableQAdvectionDiffusionElement
    : public QAdvectionDiffusionElement<DIM, NNODE_1D>,
      public virtual RefineableAdvectionDiffusionEquations<DIM>,
      public virtual RefineableQElement<DIM>
  {
  public:
    RefineableQAdvectionDiffusionElement()
      : RefineableElement(),
        RefineableAdvectionDiffusionEquations<DIM>(),
        RefineableQElement<DIM>(),
        QAdvectionDiffusionElement<DIM, NNODE_1D>()
    {
    }
 
 
    RefineableQAdvectionDiffusionElement(
      const RefineableQAdvectionDiffusionElement<DIM, NNODE_1D>& dummy) =
      delete;
 
    void operator=(const RefineableQAdvectionDiffusionElement<DIM, NNODE_1D>&) =
      delete;
 
    unsigned ncont_interpolated_values() const
    {
      return 1;
    }
 
    unsigned nvertex_node() const
    {
      return QAdvectionDiffusionElement<DIM, NNODE_1D>::nvertex_node();
    }
 
    Node* vertex_node_pt(const unsigned& j) const
    {
      return QAdvectionDiffusionElement<DIM, NNODE_1D>::vertex_node_pt(j);
    }
 
    void rebuild_from_sons(Mesh*& mesh_pt) {}
 
    unsigned nrecovery_order()
    {
      return (NNODE_1D - 1);
    }
 
    void further_setup_hanging_nodes() {}
  };
 
 
 
  //=======================================================================
  //=======================================================================
  template<unsigned DIM, unsigned NNODE_1D>
  class FaceGeometry<RefineableQAdvectionDiffusionElement<DIM, NNODE_1D>>
    : public virtual QElement<DIM - 1, NNODE_1D>
  {
  public:
    FaceGeometry() : QElement<DIM - 1, NNODE_1D>() {}
  };
 
} // namespace oomph
 
#endif