EdgeNodeFinder.cpp

Go to the documentation of this file.

//=======================================================================================
// ____          ____    __    ______     __________   __      __       __        __
// \    \       |    |  |  |  |   _   \  |___    ___| |  |    |  |     /  \      |  |
//  \    \      |    |  |  |  |  |_)   |     |  |     |  |    |  |    /    \     |  |
//   \    \     |    |  |  |  |   _   /      |  |     |  |    |  |   /  /\  \    |  |
//    \    \    |    |  |  |  |  | \  \      |  |     |   \__/   |  /  ____  \   |  |____
//     \    \   |    |  |__|  |__|  \__\     |__|      \________/  /__/    \__\  |_______|
//      \    \  |    |   ________________________________________________________________
//       \    \ |    |  |  ______________________________________________________________|
//        \    \|    |  |  |         __          __     __     __     ______      _______
//         \         |  |  |_____   |  |        |  |   |  |   |  |   |   _  \    /  _____)
//          \        |  |   _____|  |  |        |  |   |  |   |  |   |  | \  \   \_______
//           \       |  |  |        |  |_____   |   \_/   |   |  |   |  |_/  /    _____  |
//            \ _____|  |__|        |________|   \_______/    |__|   |______/    (_______/
//
//  This file is part of VirtualFluids. VirtualFluids 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.
//
//  VirtualFluids 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.
//
//  SPDX-License-Identifier: GPL-3.0-or-later
//  SPDX-FileCopyrightText: Copyright © VirtualFluids Project contributors, see AUTHORS.md in root folder
//
//=======================================================================================
#include <vector>
#include <optional>
 
#include "EdgeNodeFinder.h"
#include "Parameter.h"
 
namespace vf::gpu
{
void findEdgeNodesXY(LBMSimulationParameter& parameterLB);
void findEdgeNodesXZ(LBMSimulationParameter& parameterLB);
void findEdgeNodesYZ(LBMSimulationParameter& parameterLB);
void findEdgeNodes(const std::vector<ProcessNeighbor27> &recvProcessNeighbor,
                   const std::vector<ProcessNeighbor27> &sendProcessNeighbor,
                   std::vector<LBMSimulationParameter::EdgeNodePositions> &edgeNodes);
std::optional<std::pair<uint, uint>> findIndexInSendNodes(uint nodeIndex,
                                                        const std::vector<ProcessNeighbor27> &sendProcessNeighbor);
 
void findEdgeNodesCommMultiGPU(Parameter& parameter)
{
    for (int level = 0; level <= parameter.getFine(); level++) {
        findEdgeNodesXY(*parameter.getParH(level));
        findEdgeNodesXZ(*parameter.getParH(level));
        findEdgeNodesYZ(*parameter.getParH(level));
    }
}
 
void findEdgeNodesXY(LBMSimulationParameter& parameterLB)
{
    findEdgeNodes(parameterLB.recvProcessNeighborsX, parameterLB.sendProcessNeighborsY,
                  parameterLB.edgeNodesXtoY);
}
 
void findEdgeNodesXZ(LBMSimulationParameter& parameterLB)
{
    findEdgeNodes(parameterLB.recvProcessNeighborsX, parameterLB.sendProcessNeighborsZ,
                  parameterLB.edgeNodesXtoZ);
}
 
void findEdgeNodesYZ(LBMSimulationParameter& parameterLB)
{
    findEdgeNodes(parameterLB.recvProcessNeighborsY, parameterLB.sendProcessNeighborsZ,
                  parameterLB.edgeNodesYtoZ);
}
 
void findEdgeNodes(const std::vector<ProcessNeighbor27> &recvProcessNeighbor,
                   const std::vector<ProcessNeighbor27> &sendProcessNeighbor,
                   std::vector<LBMSimulationParameter::EdgeNodePositions> &edgeNodes)
{
    for (uint neighbor = 0; neighbor < uint(recvProcessNeighbor.size()); neighbor++) {
        for (uint index = 0; index < recvProcessNeighbor[neighbor].numberOfNodes; index++) {
            if (auto sendIndices = findIndexInSendNodes(recvProcessNeighbor[neighbor].index[index], sendProcessNeighbor)) {
                edgeNodes.emplace_back(neighbor, index, sendIndices->first, sendIndices->second);
            }
        }
    }
}
 
std::optional<std::pair<uint, uint>> findIndexInSendNodes(const uint nodeIndex,
                                                          const std::vector<ProcessNeighbor27> &sendProcessNeighbor)
{
    for (uint neighbor = 0; neighbor < uint(sendProcessNeighbor.size()); neighbor++) {
        for (uint node = 0; node < sendProcessNeighbor[neighbor].numberOfNodes; node++) {
            if (sendProcessNeighbor[neighbor].index[node] == nodeIndex) {
                return std::pair<uint, uint>(neighbor, node);
            }
        }
    }
    return std::nullopt;
}
 
} // namespace vf::gpu