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;
    };

    enum SpaceType {
      HERMES_H1_SPACE = 0,
      HERMES_HCURL_SPACE = 1,
      HERMES_HDIV_SPACE = 2,
      HERMES_L2_SPACE = 3,
      HERMES_INVALID_SPACE = -9999
    };

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

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

      int get_order(int index, ElementMode2D mode) const;

      virtual Shapeset* clone() = 0;

      int get_num_components() const;

      int get_max_order() const;

      virtual int get_max_index(ElementMode2D mode) = 0;

      int get_vertex_index(int vertex, ElementMode2D mode) const;

      int get_edge_index(int edge, int ori, int order, ElementMode2D mode) const;

      virtual SpaceType get_space_type() const = 0;

    protected:
      int* get_bubble_indices(int order, ElementMode2D mode) const;

      int get_num_bubbles(int order, ElementMode2D mode) const;

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

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

      double get_fn_value (int index, double x, double y, int component, ElementMode2D mode);
      double get_dx_value (int index, double x, double y, int component, ElementMode2D mode);
      double get_dy_value (int index, double x, double y, int component, ElementMode2D mode);
      double get_dxx_value(int index, double x, double y, int component, ElementMode2D mode);
      double get_dyy_value(int index, double x, double y, int component, ElementMode2D mode);
      double get_dxy_value(int index, double x, double y, int component, ElementMode2D mode);

      double2* get_ref_vertex(int vertex, ElementMode2D mode);

      virtual int get_id() const = 0;

      shape_fn_t*** shape_table[6];

      int**  vertex_indices;
      int*** edge_indices;
      int*** bubble_indices;
      int**  bubble_count;
      int**  index_to_order;

      double2 ref_vert[H2D_MAX_SOLUTION_COMPONENTS][H2D_MAX_NUMBER_VERTICES];
      int max_order;
      int num_components;

      int ebias; 


      double** comb_table;
      int table_size;





      double* calculate_constrained_edge_combination(int order, int part, int ori, ElementMode2D mode);

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

      void free_constrained_edge_combinations();

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

      template<typename Scalar> friend class DiscreteProblem;
      template<typename Scalar> friend class Solution;
      friend class CurvMap; 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;
      friend void check_leg_tri(Shapeset* shapeset);
      friend void check_gradleg_tri(Shapeset* shapeset);
      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;
    };
  }
}
#endif