RankMap.h

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://mooseframework.inl.gov
//*
//* All rights reserved, see COPYRIGHT for full restrictions
//* https://github.com/idaholab/moose/blob/master/COPYRIGHT
//*
//* Licensed under LGPL 2.1, please see LICENSE for details
//* https://www.gnu.org/licenses/lgpl-2.1.html

#pragma once

#include "PerfGraphInterface.h"

#include "libmesh/parallel_object.h"

class RankMap : libMesh::ParallelObject, PerfGraphInterface
{
public:
  RankMap(const libMesh::Parallel::Communicator & comm, PerfGraph & perf_graph);

  unsigned int hardwareID(processor_id_type pid) const
  {
    mooseAssert(pid < _communicator.size(), "PID out of range");
    return _rank_to_hardware_id[pid];
  }

  const std::vector<processor_id_type> & ranks(unsigned int hardware_id) const
  {
    auto item = _hardware_id_to_ranks.find(hardware_id);
    if (item == _hardware_id_to_ranks.end())
      mooseError("hardware_id not found");

    return item->second;
  }

  const std::vector<unsigned int> & rankHardwareIds() const { return _rank_to_hardware_id; }

protected:
  std::unordered_map<unsigned int, std::vector<processor_id_type>> _hardware_id_to_ranks;

  std::vector<unsigned int> _rank_to_hardware_id;
};