MeshTopology.h

// Copyright (C) 2006-2009 Anders Logg
//
// 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/>.
//
// First added:  2006-05-08
// Last changed: 2014-07-02

#ifndef __MESH_TOPOLOGY_H
#define __MESH_TOPOLOGY_H

#include <cstdint>
#include <map>
#include <utility>
#include <vector>

#include <dolfin/common/Variable.h>
#include "MeshConnectivity.h"

namespace dolfin
{


  class MeshTopology : public Variable
  {
  public:

    MeshTopology();

    MeshTopology(const MeshTopology& topology);

    ~MeshTopology();

    MeshTopology& operator= (const MeshTopology& topology);

    std::size_t dim() const;

    std::size_t size(std::size_t dim) const;

    std::size_t size_global(std::size_t dim) const;

    std::size_t ghost_offset(std::size_t dim) const;

    void clear();

    void clear(std::size_t d0, std::size_t d1);

    void init(std::size_t dim);

    void init(std::size_t dim, std::size_t local_size, std::size_t global_size);

    void init_global_indices(std::size_t dim, std::size_t size);

    void init_ghost(std::size_t dim, std::size_t index);

    void set_global_index(std::size_t dim, std::size_t local_index,
                          std::int64_t global_index)
    {
      dolfin_assert(dim < _global_indices.size());
      dolfin_assert(local_index < _global_indices[dim].size());
      _global_indices[dim][local_index] = global_index;
    }

    const std::vector<std::int64_t>& global_indices(std::size_t d) const
    {
      dolfin_assert(d < _global_indices.size());
      return _global_indices[d];
    }

    bool have_global_indices(std::size_t dim) const
    {
      dolfin_assert(dim < _global_indices.size());
      return !_global_indices[dim].empty();
    }

    bool have_shared_entities(unsigned int dim) const
    { return (_shared_entities.find(dim) != _shared_entities.end()); }

    std::map<std::int32_t, std::set<unsigned int> >&
      shared_entities(unsigned int dim);

    const std::map<std::int32_t, std::set<unsigned int> >&
      shared_entities(unsigned int dim) const;

    std::vector<unsigned int>& cell_owner()
    { return _cell_owner;  }

    const std::vector<unsigned int>& cell_owner() const
    { return _cell_owner;  }

    dolfin::MeshConnectivity& operator() (std::size_t d0, std::size_t d1);

    const dolfin::MeshConnectivity& operator() (std::size_t d0,
                                                std::size_t d1) const;

    size_t hash() const;

    std::string str(bool verbose) const;

    // Developer note: std::vector is used in place of a MeshFunction
    //                 to avoid circular dependencies in the header files
    std::map<std::vector<std::size_t>,
      std::pair<std::vector<std::size_t>,
      std::vector<std::vector<std::size_t>>>> coloring;

  private:

    // Number of mesh entities for each topological dimension
    std::vector<unsigned int> num_entities;

    // Number of ghost indices for each topological dimension
    std::vector<std::size_t> ghost_offset_index;

    // Global number of mesh entities for each topological dimension
    std::vector<std::size_t> global_num_entities;

    // Global indices for mesh entities (empty if not set)
    std::vector<std::vector<std::int64_t> > _global_indices;

    // For entities of a given dimension d , maps each shared entity
    // (local index) to a list of the processes sharing the vertex
    std::map<unsigned int, std::map<std::int32_t, std::set<unsigned int>>>
      _shared_entities;

    // For cells which are "ghosted", locate the owning process,
    // using a vector rather than a map,
    // since ghost cells are always at the end of the range.
    std::vector<unsigned int> _cell_owner;

    // Connectivity for pairs of topological dimensions
    std::vector<std::vector<MeshConnectivity> > connectivity;

  };

}

#endif