Function DistributeAMRBoxes

IO/CatalystConduit/vtkConduitToDataObject.cxx:251–329

distribute AMRBoxes to all processes

// distribute AMRBoxes to all processes
void DistributeAMRBoxes(
  const LocalInfo& rankInfo, const GlobalInfo& globalInfo, vtkOverlappingAMR* amr)
{
  vtkMultiProcessController* controller = vtkMultiProcessController::GetGlobalController();

  if (globalInfo.NbOfProcesses == 1 || !controller)
  {
    return;
  }
  std::vector<vtkIdType> boxBoundsOffsets(globalInfo.NbOfProcesses, 0);
  std::vector<vtkIdType> boxBoundsCounts(globalInfo.NbOfProcesses);
  std::vector<int> boxExtentsLocal(8 * rankInfo.NbOfBlocks, 0);
  std::vector<int> boxExtentsGlobal(8 * globalInfo.NbOfBlocks, 0);

  for (int rank = 0; rank < globalInfo.NbOfProcesses; ++rank)
  {
    int num_blocks = 0;
    for (int level = 0; level < globalInfo.NbOfLevels; level++)
    {
      num_blocks += globalInfo.BlocksPerLevelAndRank[level + rank * globalInfo.NbOfLevels];
    }
    boxBoundsCounts[rank] = num_blocks * 8;
    if (rank > 0)
    {
      boxBoundsOffsets[rank] = boxBoundsCounts[rank - 1] + boxBoundsOffsets[rank - 1];
    }
  }

  int local_index = 0;
  for (std::map<int, std::pair<int, int>>::const_iterator it = rankInfo.DomainBlockLevelIds.begin();
       it != rankInfo.DomainBlockLevelIds.end(); ++it)
  {
    int level = it->second.first;
    int id = it->second.second + rankInfo.BlockOffsets[level];

    vtkAMRBox box = amr->GetAMRBox(level, id);
    const int* loCorner = box.GetLoCorner();
    const int* hiCorner = box.GetHiCorner();
    int offset = 8 * local_index;
    boxExtentsLocal[offset + 0] = level;
    boxExtentsLocal[offset + 1] = id;
    boxExtentsLocal[offset + 2] = loCorner[0];
    boxExtentsLocal[offset + 3] = loCorner[1];
    boxExtentsLocal[offset + 4] = loCorner[2];
    boxExtentsLocal[offset + 5] = hiCorner[0];
    boxExtentsLocal[offset + 6] = hiCorner[1];
    boxExtentsLocal[offset + 7] = hiCorner[2];
    ++local_index;
  }

  controller->AllGatherV(boxExtentsLocal.data(), boxExtentsGlobal.data(), boxExtentsLocal.size(),
    boxBoundsCounts.data(), boxBoundsOffsets.data());
  for (int block = 0; block < globalInfo.NbOfBlocks; ++block)
  {
    int level = boxExtentsGlobal[8 * block];
    int id = boxExtentsGlobal[8 * block + 1];
    int* dims = &boxExtentsGlobal[8 * block + 2];
    vtkAMRBox box(dims[0], dims[1], dims[2], dims[3], dims[4], dims[5]);