linearizer.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_LINEARIZER_H
#define __H2D_LINEARIZER_H

#include "../global.h"
#include "../function/solution.h"
#include "linearizer_base.h"

namespace Hermes
{
  namespace Hermes2D
  {
    namespace Views
    {
      class HERMES_API Linearizer : public LinearizerBase
      {
      public:

        Linearizer(bool auto_max = true);
        ~Linearizer();

        void process_solution(MeshFunction<double>* sln, int item = H2D_FN_VAL_0, double eps = HERMES_EPS_NORMAL);

        void save_solution_vtk(MeshFunction<double>* sln, const char* filename, const char* quantity_name,
          bool mode_3D = true, int item = H2D_FN_VAL_0,
          double eps = HERMES_EPS_NORMAL);

        void set_displacement(MeshFunction<double>* xdisp, MeshFunction<double>* ydisp, double dmult = 1.0);

        void calc_vertices_aabb(double* min_x, double* max_x,
          double* min_y, double* max_y) const; 

        int get_num_vertices();
        double3* get_vertices();

        int3* get_contour_triangles();
        int get_num_contour_triangles();

        void set_curvature_epsilon(double curvature_epsilon);

        double get_curvature_epsilon();

        void free();

      protected:
        double curvature_epsilon;

        bool user_xdisp, user_ydisp;
        MeshFunction<double> *xdisp, *ydisp;
        double dmult;

        int3* tris_contours;      
        int triangle_contours_count;
        double3* verts;  

        int item, component, value_type;

        int add_vertex();
        int get_vertex(int p1, int p2, double x, double y, double value);

        void process_triangle(MeshFunction<double>** fns, int iv0, int iv1, int iv2, int level,
          double* val, double* phx, double* phy, int* indices, bool curved);

        void process_quad(MeshFunction<double>** fns, int iv0, int iv1, int iv2, int iv3, int level,
          double* val, double* phx, double* phy, int* indices, bool curved);

        void find_min_max();
      };
    }
  }
}
#endif