precalc.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_PRECALC_H
#define __H2D_PRECALC_H

#include "../function/function.h"
#include "../shapeset/shapeset.h"

namespace Hermes
{
  namespace Hermes2D
  {
    enum SpaceType;
    class HERMES_API PrecalcShapeset : public Function < double >
    {
    public:
      SpaceType get_space_type() const;

      PrecalcShapeset(Shapeset* shapeset);

      virtual ~PrecalcShapeset();

      virtual void set_active_shape(int index);

    protected:
      virtual void set_quad_2d(Quad2D* quad_2d);

      virtual void free();

      int get_active_shape() const;

      Shapeset* get_shapeset() const;

      virtual unsigned short get_edge_fn_order(int edge);

      Shapeset* shapeset;

      int index;

      unsigned short max_index[H2D_NUM_MODES];

      double2 ref_points[H2D_MAX_INTEGRATION_POINTS_COUNT];

      virtual void precalculate(unsigned short order, unsigned short mask);

      void update_max_index();

      friend class RefMap;
      template<typename T> friend class KellyTypeAdapt;
      template<typename T> friend class Adapt;
      template<typename T> friend class Func;
      template<typename T> friend class Solution;
      template<typename T> friend class DiscontinuousFunc;
      template<typename T> friend class DiscreteProblem;
      template<typename T> friend class DiscreteProblemDGAssembler;
      template<typename T> friend class DiscreteProblemThreadAssembler;
      template<typename T> friend class NeighborSearch;
      friend class CurvMap;
    };

    class HERMES_API PrecalcShapesetAssemblingStorage
    {
    public:
      PrecalcShapesetAssemblingStorage(Shapeset* shapeset);
      ~PrecalcShapesetAssemblingStorage();
      unsigned char shapeset_id;
      unsigned short max_index[2];
      unsigned short ref_count;

    private:
      double*** PrecalculatedValues[H2D_NUM_MODES][H2D_NUM_FUNCTION_VALUES];
      bool** PrecalculatedInfo[H2D_NUM_MODES];
      friend class PrecalcShapesetAssembling;
    };

    class HERMES_API PrecalcShapesetAssembling : public PrecalcShapeset
    {
    public:
      PrecalcShapesetAssembling(Shapeset* shapeset);

      PrecalcShapesetAssembling(const PrecalcShapesetAssembling& other);

      virtual ~PrecalcShapesetAssembling();

      const double* get_fn_values(int component = 0) const;

      const double* get_dx_values(int component = 0) const;

      const double* get_dy_values(int component = 0) const;

#ifdef H2D_USE_SECOND_DERIVATIVES
      const double* get_dxx_values(int component = 0) const;

      const double* get_dyy_values(int component = 0) const;

      const double* get_dxy_values(int component = 0) const;
#endif

      const double* get_values(int component, unsigned short item) const;

    private:
      virtual void precalculate(unsigned short order, unsigned short mask);

      PrecalcShapesetAssemblingStorage* storage;

      bool attempt_to_reuse(unsigned short order) const;
      bool reuse_possible() const;
    };

    class PrecalcShapesetAssemblingInternal : public PrecalcShapesetAssembling
    {
    public:
      PrecalcShapesetAssemblingInternal(Shapeset* shapeset);
      virtual ~PrecalcShapesetAssemblingInternal();
    };
  }
}
#endif