vector_base_view.cpp

// This file is part of Hermes2D.
//
// Copyright 2005-2008 Jakub Cerveny <jakub.cerveny@gmail.com>
// Copyright 2005-2008 Lenka Dubcova <dubcova@gmail.com>
// Copyright 2005-2008 Pavel Solin <solin@unr.edu>
//
// 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 $

#ifndef NOGLUT

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

namespace Hermes
{
  namespace Hermes2D
  {
    namespace Views
    {
      template<typename Scalar>
      void VectorBaseView<Scalar>::show(Space<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 = NULL; sln = NULL; this->lines = false; basic_title.assign(title); }

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

      template<typename Scalar>
      void VectorBaseView<Scalar>::set_title(const char* t) {
        if(basic_title.length() == 0)
          basic_title.assign(t);
        View::set_title(t);
      }

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

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

      template<>
      void VectorBaseView<double>::update_solution()
      {
        double* coeffs = new 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, 0.001, H2D_FN_VAL_0, H2D_FN_VAL_1);
        update_title();

        delete [] coeffs;
      }
      template<>
      void VectorBaseView<std::complex<double> >::update_solution()
      {
        std::complex<double> * coeffs = new std::complex<double>[ndof + 1];
        memset(coeffs, 0, sizeof(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, 0.001, H2D_FN_VAL_0, H2D_FN_VAL_1);
        update_title();

        delete [] 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";
      }

      template class HERMES_API VectorBaseView<double>;
      template class HERMES_API VectorBaseView<std::complex<double> >;
    }
  }
}
#endif