From 94d3e79a8617f88dc0219cfdeedfa3147833719d Mon Sep 17 00:00:00 2001 From: Adrian Kummerlaender Date: Mon, 24 Jun 2019 14:43:36 +0200 Subject: Initialize at openlb-1-3 --- src/functors/lattice/integral/superIntegralF3D.hh | 200 ++++++++++++++++++++++ 1 file changed, 200 insertions(+) create mode 100644 src/functors/lattice/integral/superIntegralF3D.hh (limited to 'src/functors/lattice/integral/superIntegralF3D.hh') diff --git a/src/functors/lattice/integral/superIntegralF3D.hh b/src/functors/lattice/integral/superIntegralF3D.hh new file mode 100644 index 0000000..20407a2 --- /dev/null +++ b/src/functors/lattice/integral/superIntegralF3D.hh @@ -0,0 +1,200 @@ +/* This file is part of the OpenLB library + * + * Copyright (C) 2018 Adrian Kummerlaender + * E-mail contact: info@openlb.net + * The most recent release of OpenLB can be downloaded at + * + * + * This program 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 2 + * of the License, or (at your option) any later version. + * + * This program 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 this program; if not, write to the Free + * Software Foundation, Inc., 51 Franklin Street, Fifth Floor, + * Boston, MA 02110-1301, USA. + */ + +#ifndef SUPER_INTEGRAL_F_3D_HH +#define SUPER_INTEGRAL_F_3D_HH + +#include +#include + +#include "superIntegralF3D.h" +#include "blockIntegralF3D.h" +#include "functors/analytical/indicator/indicatorBaseF3D.hh" +#include "utilities/functorPtr.hh" + +namespace olb { + + +template +SuperSum3D::SuperSum3D(FunctorPtr>&& f, + FunctorPtr>&& indicatorF) + : SuperF3D(f->getSuperStructure(), f->getTargetDim()+1), + _f(std::move(f)), + _indicatorF(std::move(indicatorF)) +{ + this->getName() = "Sum("+_f->getName()+")"; + + LoadBalancer& load = _f->getSuperStructure().getLoadBalancer(); + + if ( _f->getBlockFSize() == load.size() && + _indicatorF->getBlockFSize() == load.size() ) { + for (int iC = 0; iC < load.size(); ++iC) { + this->_blockF.emplace_back( + new BlockSum3D(_f->getBlockF(iC), + _indicatorF->getBlockIndicatorF(iC)) + ); + } + } +} + +template +SuperSum3D::SuperSum3D(FunctorPtr>&& f, + SuperGeometry3D& superGeometry, + const int material) + : SuperSum3D( + std::forward(f), + superGeometry.getMaterialIndicator(material)) +{ } + +template +bool SuperSum3D::operator() (W output[], const int input[]) +{ + _f->getSuperStructure().communicate(); + CuboidGeometry3D& geometry = _f->getSuperStructure().getCuboidGeometry(); + LoadBalancer& load = _f->getSuperStructure().getLoadBalancer(); + + for (int i = 0; i < this->getTargetDim(); ++i) { + output[i] = W(0); + } + + if (this->_blockF.empty()) { + W outputTmp[_f->getTargetDim()]; + int inputTmp[_f->getSourceDim()]; + std::size_t voxels(0); + + for (int iC = 0; iC < load.size(); ++iC) { + const Cuboid3D cuboid = geometry.get(load.glob(iC)); + inputTmp[0] = load.glob(iC); + for (inputTmp[1] = 0; inputTmp[1] < cuboid.getNx(); ++inputTmp[1]) { + for (inputTmp[2] = 0; inputTmp[2] < cuboid.getNy(); ++inputTmp[2]) { + for (inputTmp[3] = 0; inputTmp[3] < cuboid.getNz(); ++inputTmp[3]) { + if (_indicatorF(inputTmp)) { + _f(outputTmp,inputTmp); + for (int i = 0; i < _f->getTargetDim(); ++i) { + output[i] += outputTmp[i]; + } + voxels += 1; + } + } + } + } + } + output[_f->getTargetDim()] = voxels; + } + else { + for (int iC = 0; iC < load.size(); ++iC) { + this->getBlockF(iC)(output, input); + } + } + +#ifdef PARALLEL_MODE_MPI + for (int i = 0; i < this->getTargetDim(); ++i) { + singleton::mpi().reduceAndBcast(output[i], MPI_SUM); + } +#endif + return true; +} + + +template +SuperIntegral3D::SuperIntegral3D(FunctorPtr>&& f, + FunctorPtr>&& indicatorF) + : SuperF3D(f->getSuperStructure(), f->getTargetDim()), + _f(std::move(f)), + _indicatorF(std::move(indicatorF)) +{ + this->getName() = "Integral("+_f->getName()+")"; + + LoadBalancer& load = _f->getSuperStructure().getLoadBalancer(); + + if ( _f->getBlockFSize() == load.size() && + _indicatorF->getBlockFSize() == load.size() ) { + for (int iC = 0; iC < load.size(); ++iC) { + this->_blockF.emplace_back( + new BlockIntegral3D(_f->getBlockF(iC), + _indicatorF->getBlockIndicatorF(iC)) + ); + } + } +} + +template +SuperIntegral3D::SuperIntegral3D(FunctorPtr>&& f, + SuperGeometry3D& superGeometry, + const int material) + : SuperIntegral3D( + std::forward(f), + superGeometry.getMaterialIndicator(material)) +{ } + +template +bool SuperIntegral3D::operator() (W output[], const int input[]) +{ + _f->getSuperStructure().communicate(); + CuboidGeometry3D& geometry = _f->getSuperStructure().getCuboidGeometry(); + LoadBalancer& load = _f->getSuperStructure().getLoadBalancer(); + + for (int i = 0; i < this->getTargetDim(); ++i) { + output[i] = W(0); + } + + if (this->_blockF.empty()) { + W outputTmp[_f->getTargetDim()]; + int inputTmp[_f->getSourceDim()]; + + for (int iC = 0; iC < load.size(); ++iC) { + const Cuboid3D cuboid = geometry.get(load.glob(iC)); + const W weight = pow(cuboid.getDeltaR(), 3); + inputTmp[0] = load.glob(iC); + for (inputTmp[1] = 0; inputTmp[1] < cuboid.getNx(); ++inputTmp[1]) { + for (inputTmp[2] = 0; inputTmp[2] < cuboid.getNy(); ++inputTmp[2]) { + for (inputTmp[3] = 0; inputTmp[3] < cuboid.getNz(); ++inputTmp[3]) { + if (_indicatorF(inputTmp)) { + _f(outputTmp,inputTmp); + for (int i = 0; i < this->getTargetDim(); ++i) { + output[i] += outputTmp[i] * weight; + } + } + } + } + } + } + } + else { + for (int iC = 0; iC < load.size(); ++iC) { + this->getBlockF(iC)(output, input); + } + } + +#ifdef PARALLEL_MODE_MPI + for (int i = 0; i < this->getTargetDim(); ++i) { + singleton::mpi().reduceAndBcast(output[i], MPI_SUM); + } +#endif + return true; +} + + +} + +#endif -- cgit v1.2.3