shapeset.h

// This file is part of Hermes2D.
//
// Hermes2D 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 of the License, or
// (at your option) any later version.
//
// Hermes2D 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 General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with Hermes2D.  If not, see <http://www.gnu.org/licenses/>.

#ifndef __H2D_SHAPESET_H
#define __H2D_SHAPESET_H

#include "../global.h"
namespace Hermes
{
  namespace Hermes2D
  {
    enum FunctionExpansionIndex {
      H2D_FEI_VALUE = 0, 
      H2D_FEI_DX = 1, 
      H2D_FEI_DY = 2, 
      H2D_FEI_DXX = 3, 
      H2D_FEI_DYY = 4, 
      H2D_FEI_DXY = 5 
    };

    namespace RefinementSelectors
    {
      template<typename Scalar> class Selector;
      template<typename Scalar> class POnlySelector;
      template<typename Scalar> class HOnlySelector;
      template<typename Scalar> class OptimumSelector;
      template<typename Scalar> class ProjBasedSelector;
      template<typename Scalar> class H1ProjBasedSelector;
      template<typename Scalar> class L2ProjBasedSelector;
      template<typename Scalar> class HcurlProjBasedSelector;
    };

#ifdef HERMES_FOR_AGROS
#define H2D_MAX_LOCAL_BASIS_SIZE_TRI 78
#define H2D_MAX_LOCAL_BASIS_SIZE_QUAD 137
#define H2D_MAX_LOCAL_BASIS_SIZE 137
#define H2D_NUM_SHAPESETS 2
#else
#define H2D_MAX_LOCAL_BASIS_SIZE_QUAD 308
#define H2D_MAX_LOCAL_BASIS_SIZE_TRI 164
#define H2D_MAX_LOCAL_BASIS_SIZE 308
#define H2D_NUM_SHAPESETS 5
#endif



    class HERMES_API Shapeset : public Hermes::Mixins::Loggable
    {
    public:
      ~Shapeset();

      typedef double(*shape_fn_t)(double, double);

      unsigned short get_order(int index, ElementMode2D mode) const;

      virtual Shapeset* clone() = 0;

      unsigned char get_num_components() const;

      unsigned short get_max_order() const;
      unsigned short get_min_order() const;

      virtual unsigned short get_max_index(ElementMode2D mode) const = 0;

      short get_vertex_index(int vertex, ElementMode2D mode) const;

      short get_edge_index(unsigned char edge, unsigned short ori, unsigned short order, ElementMode2D mode) const;

      virtual SpaceType get_space_type() const = 0;

      virtual unsigned char get_id() const = 0;

      double get_value(int n, int index, double x, double y, unsigned short component, ElementMode2D mode);

      double get_fn_value(int index, double x, double y, unsigned short component, ElementMode2D mode);
      double get_dx_value(int index, double x, double y, unsigned short component, ElementMode2D mode);
      double get_dy_value(int index, double x, double y, unsigned short component, ElementMode2D mode);

      double get_fn_value_0_tri(int index, double x, double y);
      double get_dx_value_0_tri(int index, double x, double y);
      double get_dy_value_0_tri(int index, double x, double y);

      double get_fn_value_0_quad(int index, double x, double y);
      double get_dx_value_0_quad(int index, double x, double y);
      double get_dy_value_0_quad(int index, double x, double y);

      double get_dxx_value(int index, double x, double y, unsigned short component, ElementMode2D mode);
      double get_dyy_value(int index, double x, double y, unsigned short component, ElementMode2D mode);
      double get_dxy_value(int index, double x, double y, unsigned short component, ElementMode2D mode);

      virtual unsigned short get_num_bubbles(unsigned short order, ElementMode2D mode) const;

    protected:
      virtual short* get_bubble_indices(unsigned short order, ElementMode2D mode) const;

      int get_constrained_edge_index(unsigned char edge, unsigned short order, unsigned short ori, unsigned short part, ElementMode2D mode) const;

      double2* get_ref_vertex(int vertex, ElementMode2D mode);

      shape_fn_t*** shape_table[6];

      short**  vertex_indices;
      short*** edge_indices;
      short*** bubble_indices;
      unsigned short**  bubble_count;
      unsigned short**  index_to_order;

      double2 ref_vert[H2D_MAX_SOLUTION_COMPONENTS][H2D_MAX_NUMBER_VERTICES];
      unsigned char max_order, min_order;
      unsigned char num_components;

      unsigned short ebias;

      double** comb_table;
      unsigned short table_size;
      double* calculate_constrained_edge_combination(unsigned short order, unsigned short part, unsigned short ori, ElementMode2D mode);

      double* get_constrained_edge_combination(unsigned short order, unsigned short part, unsigned short ori, unsigned short& nitems, ElementMode2D mode);

      void free_constrained_edge_combinations();

      double get_constrained_value(int n, int index, double x, double y, unsigned short component, ElementMode2D mode);

      template<typename Scalar> friend class DiscreteProblem;
      template<typename Scalar> friend class DiscreteProblemIntegrationOrderCalculator;
      template<typename Scalar> friend class Solution;
      friend class CurvMap;
      friend class CurvMapStatic;
      friend class RefMap;
      template<typename Scalar> friend class RefinementSelectors::H1ProjBasedSelector;
      template<typename Scalar> friend class RefinementSelectors::L2ProjBasedSelector;
      template<typename Scalar> friend class RefinementSelectors::HcurlProjBasedSelector;
      template<typename Scalar> friend class RefinementSelectors::OptimumSelector;
      friend class PrecalcShapeset;
      template<typename Scalar> friend class Form;
      template<typename Scalar> friend class MatrixForm;
      template<typename Scalar> friend class VectorForm;
      template<typename Scalar> friend class Space;
      template<typename Scalar> friend class H1Space;
      template<typename Scalar> friend class L2Space;
      template<typename Scalar> friend class HcurlSpace;
      template<typename Scalar> friend class HdivSpace;
      template<typename Scalar> friend class L2MarkerWiseConstSpace;
    };
  }
}
#endif