graph.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_GRAPH_H
#define __H2D_GRAPH_H

#include "global.h"
namespace Hermes
{
  namespace Hermes2D
  {
    class HERMES_API Graph : public Hermes::Mixins::Loggable
    {
    public:

      Graph(const char* title = nullptr, const char* x_axis_name = nullptr, const char* y_axis_name = nullptr);

      void set_captions(const char* title = nullptr, const char* x_axis_name = nullptr, const char* y_axis_name = nullptr);

      void set_log_x(bool log = true);
      void set_log_y(bool log = true);

      void show_legend(bool show = true);
      void show_grid(bool show = true);

      int  add_row(const char* name = nullptr, const char* color = "k", const char* line = "-", const char* marker = "");
      void set_row_style(int row, const char* color = "k", const char* line = "-", const char* marker = "");

      void add_values(int row, double x, double y);
      // same as previous but uses row = 0
      void add_values(double x, double y);
      void add_values(int row, int n, double* x, double* y);
      void add_values(int row, int n, double2* xy);

      virtual void save(const char* filename) = 0;
      void save_numbered(const char* filename, int number);

    protected:

      std::string title, xname, yname;
      bool logx, logy, legend, grid;

      struct Values
      {
        double x, y;
      };

      struct Row
      {
        std::string name, color, line, marker;
        std::vector<Values> data;
      };

      std::vector<Row> rows;
    };

    class HERMES_API SimpleGraph : public Graph
    {
    public:

      SimpleGraph(const char* title = nullptr, const char* x_axis_name = nullptr, const char* y_axis_name = nullptr)
        : Graph(title, x_axis_name, y_axis_name) {}

      virtual void save(const char* filename);
    };

    class HERMES_API MatlabGraph : public Graph
    {
    public:

      MatlabGraph(const char* title = nullptr, const char* x_axis_name = nullptr, const char* y_axis_name = nullptr)
        : Graph(title, x_axis_name, y_axis_name) {}

      virtual void save(const char* filename);
    };


    static const char* default_terminal = "set terminal postscript eps enhanced\n";

    class HERMES_API GnuplotGraph : public Graph
    {
    public:

      GnuplotGraph(const char* title = nullptr, const char* x_axis_name = nullptr, const
        char* y_axis_name = nullptr, double lines_width = 1.0, const
        std::string& terminal_str = default_terminal) :
        Graph(title, x_axis_name, y_axis_name),
        legend_pos(),
        terminal_str(terminal_str),
        lw(lines_width) {
      }

      virtual void save(const char* filename);

      void set_legend_pos(const char* posspec);

    protected:
      std::string legend_pos;

      std::string terminal_str;

      double lw;
    };

    class HERMES_API PNGGraph : public GnuplotGraph
    {
    public:

      PNGGraph(const char* title = nullptr, const char* x_axis_name = nullptr, const char* y_axis_name = nullptr,
        double lines_width = 1.0, double plot_width = 800, double plot_height = 600);
    };
  }
}
#endif