mortar_assembly.hh

Go to the documentation of this file.

/*
 * mortar_assembly.hh
 *
 *  Created on: Feb 22, 2017
 *      Author: jb
 */
 
#ifndef SRC_FLOW_MORTAR_ASSEMBLY_HH_
#define SRC_FLOW_MORTAR_ASSEMBLY_HH_
 
#include "system/index_types.hh"
#include "mesh/mesh.h"
#include "quadrature/intersection_quadrature.hh"
#include "flow/darcy_flow_mh.hh"
#include "la/local_system.hh"
#include "fem/dh_cell_accessor.hh"
#include <vector>
 
 
typedef std::shared_ptr<DarcyMH::EqFields> AssemblyFieldsPtr;
typedef std::shared_ptr<DarcyMH::EqData> AssemblyDataPtr;
 
 
class MortarAssemblyBase {
public:
    typedef vector<unsigned int> IsecList;
 
    MortarAssemblyBase(AssemblyFieldsPtr eq_fields, AssemblyDataPtr eq_data)
    : eq_fields_(eq_fields),
      eq_data_(eq_data),
      mixed_mesh_(eq_data->mesh->mixed_intersections()),
      fix_velocity_flag(false)
    {
 
    }
 
    virtual ~MortarAssemblyBase() {};
 
    // Default assembly is empty to allow dummy implementation for dimensions without coupling.
    virtual void assembly(FMT_UNUSED const DHCellAccessor& dh_cell) {};
 
    void fix_velocity(const DHCellAccessor& dh_cell) {
        fix_velocity_flag = true;
        this->assembly(dh_cell);
        fix_velocity_flag = false;
    }
 
protected:
    AssemblyFieldsPtr eq_fields_;
    AssemblyDataPtr eq_data_;
    MixedMeshIntersections &mixed_mesh_;
    LocalSystem loc_system_;
    bool fix_velocity_flag;
 
};
 
 
struct IsecData {
    LocDofVec vel_dofs;
    LocDofVec dofs;
    unsigned int dim;
    double delta;
    double ele_z_coord_;
    LocDofVec dirichlet_dofs;
    arma::vec dirichlet_sol;
    unsigned int n_dirichlet;
};
 
 
class P0_CouplingAssembler :public MortarAssemblyBase {
public:
    P0_CouplingAssembler(AssemblyFieldsPtr eq_fields, AssemblyDataPtr eq_data);
    void assembly(const DHCellAccessor& dh_cell);
    void pressure_diff(const DHCellAccessor& dh_cell, double delta);
    void fix_velocity_local(const IsecData & row_ele, const IsecData &col_ele);
private:
    inline arma::mat & tensor_average(unsigned int row_dim, unsigned int col_dim) {
        return tensor_average_[4*row_dim + col_dim];
    }
 
    void add_to_linsys(double scale);
 
    vector<IsecData> isec_data_list;
 
    /// Row matrices to compute element pressure as average of boundary pressures
    std::vector< arma::mat > tensor_average_;
    std::vector< arma::vec > col_average_;
    IntersectionQuadratureP0 quadrature_;
    arma::mat product_;
 
};
 
 
 
class P1_CouplingAssembler :public MortarAssemblyBase {
public:
    P1_CouplingAssembler(AssemblyFieldsPtr eq_fields, AssemblyDataPtr eq_data)
: MortarAssemblyBase(eq_fields, eq_data),
      rhs(5),
      dofs(5),
      dirichlet(5)
    {
        rhs.zeros();
    }
 
    void assembly(const DHCellAccessor& dh_cell);
    void add_sides(const DHCellAccessor& dh_cell, unsigned int shift, vector<int> &dofs, vector<double> &dirichlet);
private:
 
    arma::vec rhs;
    vector<int> dofs;
    vector<double> dirichlet;
};
 
 
 
#endif /* SRC_FLOW_MORTAR_ASSEMBLY_HH_ */