geometry/grid.h

// -*- c++ -*-

// +--------------------------------------------------------------------+
// | This file is part of MathTL - the Mathematical Template Library    |
// |                                                                    |
// | Copyright (c) 2002-2009                                            |
// | Thorsten Raasch, Manuel Werner                                     |
// +--------------------------------------------------------------------+

#ifndef _MATHTL_GRID_H
#define _MATHTL_GRID_H

#include <geometry/point.h>
#include <utils/array1d.h>
#include <algebra/matrix.h>
#include <io/matrix_io.h>

namespace MathTL
{
  template <unsigned int DIM>
  class Grid
  {
  public:
    Grid() {}

    unsigned int size() const;

    void matlab_output(std::ostream& os) const;
  };

  template <>
  class Grid<1>
  {
  public:
    Grid();

    Grid(const Array1D<double>& grid);

    Grid(const double a, const double b, const unsigned int N);

    inline unsigned int size() const { return grid_.size(); }

    inline const Array1D<double>& points() const { return grid_; }

    void matlab_output(std::ostream& os) const;
    
  protected:
    Array1D<double> grid_;
  };

  template <>
  class Grid<2>
  {
  public:
    Grid();

    Grid(const Matrix<double>& gridx, const Matrix<double>& gridy);

    Grid(const Grid<1>& gridx, const Grid<1>& gridy);

    Grid(const Point<2>& a, const Point<2>& b,
         const unsigned N_x, const unsigned N_y);

    Grid(const Point<2>& a, const Point<2>& b,
         const unsigned int N);

    inline unsigned int size() const { return gridx_.size() * gridy_.size(); }

    Grid<2>& operator = (const Grid<2>& grid);

    void matlab_output(std::ostream& os) const;
    
    void octave_output(std::ostream& os) const;
    
  protected:
    Matrix<double> gridx_, gridy_;
  };
}

#include <geometry/grid.cpp>

#endif