vector_base_view.cpp

// 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/>.
// $Id: view4.cpp 1086 2008-10-21 09:05:44Z jakub $

#include "vector_base_view.h"

#ifndef NOGLUT

#include <GL/freeglut.h>
#include "global.h"
#include "space.h"
#include "precalc.h"
#include "filter.h"

namespace Hermes
{
  namespace Hermes2D
  {
    namespace Views
    {
      template<typename Scalar>
      void VectorBaseView<Scalar>::show(SpaceSharedPtr<Scalar> space)
      {
        free();
        pss = new PrecalcShapeset(space->shapeset);
        sln = new Solution<Scalar>();
        this->space = space;
        ndof = space->get_num_dofs();
        base_index = 0;
        update_solution();
      }

      template<typename Scalar>
      VectorBaseView<Scalar>::VectorBaseView(const char* title, WinGeom* wg)
        : VectorView(title, wg) {
        pss = nullptr; sln = nullptr; this->lines = false; basic_title.assign(title);
      }

      template<typename Scalar>
      VectorBaseView<Scalar>::VectorBaseView(char* title, WinGeom* wg)
        : VectorView(title, wg) {
        pss = nullptr; sln = nullptr; this->lines = false; basic_title.assign(title);
      }

      template<typename Scalar>
      VectorBaseView<Scalar>::~VectorBaseView()
      {
        free();
      }

      template<typename Scalar>
      void VectorBaseView<Scalar>::free()
      {
        if (pss != nullptr) { delete pss; pss = nullptr; }
      }

      template<>
      void VectorBaseView<double>::update_solution()
      {
        double* coeffs = malloc_with_check<double>(ndof + 1);
        memset(coeffs, 0, sizeof(double)* (ndof + 1));
        if (base_index >= -1 && base_index < ndof)
          coeffs[base_index + 1] = 1.0;

        Solution<double>::vector_to_solution(coeffs, space, sln, pss);

        VectorView::show(sln, sln, H2D_FN_VAL_0, H2D_FN_VAL_1);
        update_title();

        free_with_check(coeffs);
      }
      template<>
      void VectorBaseView<std::complex<double> >::update_solution()
      {
        std::complex<double>* coeffs = calloc_with_check<std::complex<double> >(ndof + 1);
        if (base_index >= -1 && base_index < ndof)
          coeffs[base_index + 1] = 1.0;
        Solution<std::complex<double> >::vector_to_solution(coeffs, space, sln, pss);

        Hermes::Hermes2D::RealFilter filter(sln);

        this->VectorView::show(&filter, &filter, H2D_FN_VAL_0, H2D_FN_VAL_1);
        update_title();

        free_with_check(coeffs);
      }

      template<typename Scalar>
      void VectorBaseView<Scalar>::update_title()
      {
        std::stringstream str;
        str << basic_title << " - dof = " << base_index;
        if (base_index < 0)
          str << " (Dirichlet lift)";
        View::set_title(str.str().c_str());
      }

      template<typename Scalar>
      void VectorBaseView<Scalar>::on_special_key(int key, int x, int y)
      {
        switch (key)
        {
        case GLUT_KEY_LEFT:
          if (base_index > -1) base_index--;
          update_solution();
          break;

        case GLUT_KEY_RIGHT:
          if (base_index < ndof - 1) base_index++;
          update_solution();
          break;

        default:
          VectorView::on_special_key(key, x, y);
        }
      }

      template<typename Scalar>
      const char* VectorBaseView<Scalar>::get_help_text() const
      {
        return
          "VectorBaseView\n\n"
          "Controls:\n"
          "  Left mouse - pan\n"
          "  Right mouse - zoom\n"
          "  Left arrow - previous basis function\n"
          "  Right arrow - next basis function\n"
          "  C - center image\n"
          "  F - toggle smooth palette\n"
          "  X - toggle hexagonal grid\n"
          "  H - render high-quality frame\n"
          "  M - toggle mesh\n"
          "  P - cycle palettes\n"
          "  S - save screenshot\n"
          "  F1 - this help\n"
          "  Esc, Q - quit";
      }
    }
  }
}
#endif

namespace Hermes
{
  namespace Hermes2D
  {
    namespace Views
    {
      template class HERMES_API VectorBaseView < double > ;
      template class HERMES_API VectorBaseView < std::complex<double> > ;
    }
  }
}