assemble_stokes.h

Go to the documentation of this file.

// Parallel Finite Element-General Geometry Ewald-like Method.
// Copyright (C) 2015-2016 Xujun Zhao, Jiyuan Li, Xikai Jiang

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


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


#pragma once

// Local includes
#include "assemble_system.h"

class AssembleStokes : public AssembleSystem
{
public:
  AssembleStokes(EquationSystems& es);


  ~AssembleStokes();


  void assemble_global_K(const std::string& system_name,
                         const std::string& option) override;


  void assemble_global_F(const std::string& system_name,
                         const std::string& option) override;


  void assemble_element_KIJ(const std::vector<Real>& JxW,
                            const std::vector<std::vector<RealGradient> >& dphi,
                            const unsigned int n_u_dofs,
                            const unsigned int I,
                            const unsigned int J,
                            DenseMatrix<Number>& Kij) override;


  void assemble_element_QI(const std::vector<Real>& JxW,
                           const std::vector<std::vector<RealGradient> >& dphi,
                           const std::vector<std::vector<Real> >& psi,
                           const unsigned int n_v_dofs,
                           const unsigned int n_p_dofs,
                           const unsigned int I,
                           DenseMatrix<Number>& Qi);


  void assemble_element_MIJ(const std::vector<Real>& JxW,
                            const std::vector<std::vector<Real> >& phi,
                            const unsigned int n_v_dofs,
                            DenseMatrix<Number>& Mij);


  void compute_element_rhs(const Elem* elem,
                           const unsigned int n_u_dofs,
                           FEBase& fe_v,
                           const std::vector<std::size_t> n_list,
                           const bool& pf_flag,
                           const std::string& option,
                           const Real& alpha,
                           DenseVector<Number>& Fe) override;

    void select_boundary_side(const Elem* elem) override;               

  void apply_bc_by_penalty(const Elem* elem,
                           const std::string& matrix_or_vector,
                           DenseMatrix<Number>& Ke,
                           DenseVector<Number>& Fe,
                           const std::string& option) override;


  Real boundary_pressure_jump(const std::string& which_side) const;


  Real boundary_traction(const std::string& which_side) const;
};