BoxMesh.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 __BOX_H
#define __BOX_H

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

namespace dolfin
{


  class BoxMesh : public Mesh
  {
  public:

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

    static Mesh create(MPI_Comm comm, const std::array<Point, 2>& p,
                       std::array<std::size_t, 3> n, CellType::Type cell_type)
    {
      Mesh mesh(comm);
      if (cell_type == CellType::Type::tetrahedron)
        build_tet(mesh, p, n);
      else if (cell_type == CellType::Type::hexahedron)
        build_hex(mesh, p, n);
      else
      {
        dolfin_error("BoxMesh.h",
                     "generate box mesh",
                     "Wrong cell type '%d'", cell_type);
      }

      return mesh;
    }

    BoxMesh(const Point& p0, const Point& p1,
            std::size_t nx, std::size_t ny, std::size_t nz);

    BoxMesh(MPI_Comm comm, const Point& p0, const Point& p1,
            std::size_t nx, std::size_t ny, std::size_t nz);

  private:

    // Build mesh
    static void build_tet(Mesh& mesh, const std::array<Point, 2>& p,
                          std::array<std::size_t, 3> n);

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

  };

}

#endif