partitionCells.hpp

/*
  Copyright 2021 Total SE

  This file is part of the Open Porous Media project (OPM).

  OPM 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 3 of the License, or
  (at your option) any later version.

  OPM 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 OPM.  If not, see <http://www.gnu.org/licenses/>.
*/

#ifndef OPM_ASPINPARTITION_HEADER_INCLUDED
#define OPM_ASPINPARTITION_HEADER_INCLUDED

#include <functional>
#include <set>
#include <string>
#include <utility>
#include <vector>

namespace Opm {

class Well;

} // namespace Opm

namespace Opm {

template <typename Element>
struct ZoltanPartitioningControl
{
    double domain_imbalance{1.03};

    std::function<int(const Element&)> index;

    std::function<int(int)> local_to_global;
};

template <class GridView, class Element>
std::pair<std::vector<int>, int>
partitionCells(const std::string&                                    method,
               const int                                             num_local_domains,
               const GridView&                                       grid_view,
               const std::vector<Well>&                              wells,
               const std::unordered_map<std::string, std::set<int>>& possibleFutureConnections,
               const ZoltanPartitioningControl<Element>&             zoltan_ctrl,
               const int                                             num_neighbor_levels);

std::pair<std::vector<int>, int> partitionCellsFromFile(const std::string& filename, const int num_cells);

std::pair<std::vector<int>, int> partitionCellsSimple(const int num_cells, const int num_domains);

} // namespace Opm

#endif // OPM_ASPINPARTITION_HEADER_INCLUDED