CD_LoadBalancing.H

Go to the documentation of this file.

/* chombo-discharge
 * Copyright © 2021 SINTEF Energy Research.
 * Please refer to Copyright.txt and LICENSE in the chombo-discharge root directory.
 */
 
#ifndef CD_LoadBalancing_H
#define CD_LoadBalancing_H
 
// Our includes
#include <CD_MultiFluidIndexSpace.H>
#include <CD_BoxSorting.H>
#include <CD_Loads.H>
#include <CD_NamespaceHeader.H>
 
class LoadBalancing
{
public:
  template <class T>
  static void
  makeBalance(Vector<int>& a_ranks, const Vector<T>& a_loads, const Vector<Box>& a_boxes);
 
  template <class T>
  static void
  makeBalance(Vector<int>& a_ranks, Loads& a_rankLoads, const Vector<T>& a_boxLoads, const Vector<Box>& a_boxes);
 
  template <typename T>
  static void
  sort(Vector<Vector<Box>>& a_boxes, Vector<Vector<T>>& a_loads, const BoxSorting a_whichSorting);
 
  template <typename T>
  static void
  sort(Vector<Box>& a_boxes, Vector<T>& a_loads, const BoxSorting a_which);
 
  static void
  sort(Vector<Box>& a_boxes, const BoxSorting a_which);
 
  static void
  gatherBoxes(Vector<Box>& a_boxes);
 
  static void
  gatherLoads(Vector<Real>& a_loads);
 
  static void
  gatherLoads(Vector<int>& a_loads);
 
  static void
  gatherLoads(Vector<long>& a_loads);
 
  static void
  gatherBoxesAndLoads(Vector<Box>& a_boxes, Vector<int>& a_loads);
 
protected:
  template <class T>
  static std::vector<std::pair<Box, T>>
  packPairs(const Vector<Box>& a_boxes, const Vector<T>& a_loads);
 
  template <class T>
  static void
  unpackPairs(Vector<Box>& a_boxes, Vector<T>& a_loads, const std::vector<std::pair<Box, T>>& a_pairs);
 
  template <class T>
  static void
  standardSort(Vector<Box>& a_boxes, Vector<T>& a_loads);
 
  template <class T>
  static void
  shuffleSort(Vector<Box>& a_boxes, Vector<T>& a_loads);
 
  template <class T>
  static void
  mortonSort(Vector<Box>& a_boxes, Vector<T>& a_loads);
 
  template <class T>
  static bool
  mortonComparator(const int a_maxBits, const std::pair<Box, T>& a_lhs, const std::pair<Box, T>& a_rhs);
 
  static int
  maxBits(std::vector<Box>::iterator a_first, std::vector<Box>::iterator a_last);
};
 
#include <CD_NamespaceFooter.H>
 
#include <CD_LoadBalancingImplem.H>
 
#endif