RectangleMesh.h

// Copyright (C) 2005-2017 Anders Logg and Garth N. Wells
//
// This file is part of DOLFIN.
//
// DOLFIN is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// DOLFIN 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 Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with DOLFIN. If not, see <http://www.gnu.org/licenses/>.

#ifndef __RECTANGLE_MESH_H
#define __RECTANGLE_MESH_H

#include <array>
#include <string>
#include <dolfin/common/MPI.h>
#include <dolfin/log/log.h>
#include <dolfin/mesh/CellType.h>
#include <dolfin/mesh/Mesh.h>

namespace dolfin
{


  class RectangleMesh : public Mesh
  {
  public:

    static Mesh create(const std::array<Point, 2>& p,
                       std::array<std::size_t, 2> n,
                       CellType::Type cell_type,
                       std::string diagonal="right")
    { return create(MPI_COMM_WORLD, p, n, cell_type, diagonal); }

    static Mesh create(MPI_Comm comm, const std::array<Point, 2>& p,
                       std::array<std::size_t, 2> n,
                       CellType::Type cell_type,
                       std::string diagonal="right")
    {
      Mesh mesh(comm);
      if (cell_type == CellType::Type::triangle)
        build_tri(mesh, p, n, diagonal);
      else if (cell_type == CellType::Type::quadrilateral)
        build_quad(mesh, p, n);
      else
      {
        dolfin_error("RectangleMesh.h",
                     "generate rectangle mesh",
                     "Wrong cell type '%d'", cell_type);
      }

      return mesh;
    }

    // Deprecated
    RectangleMesh(const Point& p0, const Point& p1,
                  std::size_t nx, std::size_t ny,
                  std::string diagonal="right");

    // Deprecated
    RectangleMesh(MPI_Comm comm,
                  const Point& p0, const Point& p1,
                  std::size_t nx, std::size_t ny,
                  std::string diagonal="right");

  private:

    // Build mesh
    static void build_tri(Mesh& mesh, const std::array<Point, 2>& p,
                          std::array<std::size_t, 2> n,
                          std::string diagonal="right");


    static void build_quad(Mesh& mesh, const std::array<Point, 2>& p,
                           std::array<std::size_t, 2> n);
  };

}

#endif