AMIWeights.C

Go to the documentation of this file.

/*---------------------------------------------------------------------------*\
  =========                 |
  \\      /  F ield         | OpenFOAM: The Open Source CFD Toolbox
   \\    /   O peration     |
    \\  /    A nd           | www.openfoam.com
     \\/     M anipulation  |
-------------------------------------------------------------------------------
    Copyright (C) 2018-2021 OpenCFD Ltd.
-------------------------------------------------------------------------------
License
    This file is part of OpenFOAM.

    OpenFOAM 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.

    OpenFOAM 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 OpenFOAM.  If not, see <http://www.gnu.org/licenses/>.

\*---------------------------------------------------------------------------*/

#include "AMIWeights.H"
#include "fvMesh.H"
#include "foamVtkSurfaceWriter.H"
#include "PatchTools.H"
#include "cyclicACMIPolyPatch.H"
#include "addToRunTimeSelectionTable.H"

// * * * * * * * * * * * * * * Static Data Members * * * * * * * * * * * * * //

namespace Foam
{
namespace functionObjects
{
    defineTypeNameAndDebug(AMIWeights, 0);
    addToRunTimeSelectionTable(functionObject, AMIWeights, dictionary);
}
}

// * * * * * * * * * * * * Protected Member Functions  * * * * * * * * * * * //

void Foam::functionObjects::AMIWeights::writeFileHeader(Ostream& os)
{
    writeHeader(os, "AMI");

    writeCommented(os, "Time");
    forAll(patchIDs_, patchi)
    {
        writeTabbed(os, "Patch");
        writeTabbed(os, "nbr_patch");

        if (UPstream::parRun())
        {
            writeTabbed(os, "distributed");
        }

        writeTabbed(os, "src_min_weight");
        writeTabbed(os, "src_max_weight");
        writeTabbed(os, "src_average_weight");
        writeTabbed(os, "src_min_neighbours");
        writeTabbed(os, "src_max_neighbours");
        writeTabbed(os, "src_average_neighbours");
        writeTabbed(os, "tgt_min_weight");
        writeTabbed(os, "tgt_max_weight");
        writeTabbed(os, "tgt_average_weight");
        writeTabbed(os, "tgt_min_neighbours");
        writeTabbed(os, "tgt_max_neighbours");
        writeTabbed(os, "tgt_average_neighbours");
    }

    os  << endl;
}


void Foam::functionObjects::AMIWeights::reportPatch
(
    const cyclicAMIPolyPatch& pp
)
{
    const word& nbrPatchName = pp.neighbPatchName();

    const Switch distributed = pp.AMI().distributed();

    const scalarField& srcWeightsSum = pp.AMI().srcWeightsSum();
    const auto srcWeightLimits = gMinMax(srcWeightsSum);
    const auto srcWeightAvg = gAverage(srcWeightsSum);

    const labelListList& srcAddress = pp.AMI().srcAddress();

    labelMinMax srcNbrLimits(labelMax, labelMin);
    scalar srcNbrAvg(0);
    for (const labelList& srcFace : srcAddress)
    {
        const label n = srcFace.size();

        srcNbrAvg += n;
        srcNbrLimits.add(n);
    }

    {
        reduce(srcNbrLimits, sumOp<labelMinMax>());

        label count = srcAddress.size();
        sumReduce(srcNbrAvg, count);
        srcNbrAvg /= (count + ROOTVSMALL);
    }

    const scalarField& tgtWeightsSum = pp.AMI().tgtWeightsSum();
    const auto tgtWeightLimits = gMinMax(tgtWeightsSum);
    const auto tgtWeightAvg = gAverage(tgtWeightsSum);

    const labelListList& tgtAddress = pp.AMI().tgtAddress();

    labelMinMax tgtNbrLimits(labelMax, labelMin);
    scalar tgtNbrAvg(0);
    for (const labelList& tgtFace : tgtAddress)
    {
        const label n = tgtFace.size();

        tgtNbrAvg += n;
        tgtNbrLimits.add(n);
    }

    {
        reduce(tgtNbrLimits, sumOp<labelMinMax>());

        label count = tgtAddress.size();
        sumReduce(tgtNbrAvg, count);
        tgtNbrAvg /= (count + ROOTVSMALL);
    }

    file()
        << mesh_.time().timeName() << tab
        << pp.name() << tab
        << nbrPatchName << tab;


    if (UPstream::parRun())
    {
        file() << distributed << tab;
    }

    file()
        << srcWeightLimits.min() << tab
        << srcWeightLimits.max() << tab
        << srcWeightAvg << tab
        << srcNbrLimits.min() << tab
        << srcNbrLimits.max() << tab
        << srcNbrAvg << tab
        << tgtWeightLimits.min() << tab
        << tgtWeightLimits.max() << tab
        << tgtWeightAvg << tab
        << tgtNbrLimits.min() << tab
        << tgtNbrLimits.max() << tab
        << tgtNbrAvg
        << endl;

    Log << "    Patches: " << nl
        << "        Source: " << pp.name() << nl
        << "        Target: " << nbrPatchName << nl;

    if (UPstream::parRun())
    {
        Log << "        Parallel distributed: " << distributed << nl;
    }

    Log << nl;

    const label w = IOstream::defaultPrecision() + 8;

    Log << "                     | " << setw(w) << pp.name()
        << " | " << setw(w) << nbrPatchName << " | " << nl
        << "        min(weight)  | " << setw(w) << srcWeightLimits.min()
        << " | " << setw(w) << tgtWeightLimits.min() << " | " << nl
        << "        max(weight)  | " << setw(w) << srcWeightLimits.max()
        << " | " << setw(w) << tgtWeightLimits.max() << " | " << nl
        << "        ave(weight)  | " << setw(w) << srcWeightAvg
        << " | " << setw(w) << tgtWeightAvg << " | " << nl
        << "        min(address) | " << setw(w) << srcNbrLimits.min()
        << " | " << setw(w) << tgtNbrLimits.min() << " | " << nl
        << "        max(address) | " << setw(w) << srcNbrLimits.max()
        << " | " << setw(w) << tgtNbrLimits.max() << " | " << nl
        << "        ave(address) | " << setw(w) << srcNbrAvg
        << " | " << setw(w) << tgtNbrAvg << " | " << nl
        << endl;

    setResult(pp.name() + ":src", pp.name());
    setResult(pp.name() + ":tgt", nbrPatchName);
    setResult(pp.name() + ":src:min(weight)", srcWeightLimits.min());
    setResult(pp.name() + ":src:max(weight)", srcWeightLimits.max());
    setResult(pp.name() + ":src:ave(weight)", srcWeightAvg);
    setResult(pp.name() + ":src:min(address)", srcNbrLimits.min());
    setResult(pp.name() + ":src:max(address)", srcNbrLimits.max());
    setResult(pp.name() + ":src:ave(address)", srcNbrAvg);
    setResult(pp.name() + ":tgt:min(weight)", tgtWeightLimits.min());
    setResult(pp.name() + ":tgt:max(weight)", tgtWeightLimits.max());
    setResult(pp.name() + ":tgt:ave(weight)", tgtWeightAvg);
    setResult(pp.name() + ":tgt:min(address)", tgtNbrLimits.min());
    setResult(pp.name() + ":tgt:max(address)", tgtNbrLimits.max());
    setResult(pp.name() + ":tgt:ave(address)", tgtNbrAvg);
}


void Foam::functionObjects::AMIWeights::writeWeightField
(
    const cyclicAMIPolyPatch& cpp,
    const scalarField& weightSum,
    const word& side
) const
{
    // Collect geometry
    labelList pointToGlobal;
    labelList uniqueMeshPointLabels;
    autoPtr<globalIndex> globalPoints;
    autoPtr<globalIndex> globalFaces;
    faceList mergedFaces;
    pointField mergedPoints;
    Foam::PatchTools::gatherAndMerge
    (
        mesh_,
        cpp.localFaces(),
        cpp.meshPoints(),
        cpp.meshPointMap(),

        pointToGlobal,
        uniqueMeshPointLabels,
        globalPoints,
        globalFaces,

        mergedFaces,
        mergedPoints
    );

    // Collect field
    scalarField mergedWeights = globalFaces().gather(weightSum);

    const bool isACMI = isA<cyclicACMIPolyPatch>(cpp);

    scalarField mergedMask;
    if (isACMI)
    {
        const cyclicACMIPolyPatch& pp = refCast<const cyclicACMIPolyPatch>(cpp);

        mergedMask = globalFaces().gather(pp.mask());
    }

    if (Pstream::master())
    {
        instant inst(mesh_.time().value(), mesh_.time().timeName());

        vtk::surfaceWriter writer
        (
            mergedPoints,
            mergedFaces,
            (baseTimeDir()/cpp.name() + "_" + side),
            false // serial: master-only
        );

        writer.setTime(inst);
        writer.writeTimeValue();
        writer.writeGeometry();

        writer.beginCellData(1 + (isACMI ? 1 : 0));
        writer.write("weightsSum", mergedWeights);

        if (isACMI)
        {
            writer.write("mask", mergedMask);
        }
    }
}


void Foam::functionObjects::AMIWeights::writeWeightFields
(
    const cyclicAMIPolyPatch& cpp
) const
{
    if (cpp.owner())
    {
        writeWeightField(cpp, cpp.AMI().srcWeightsSum(), "src");
        writeWeightField(cpp.neighbPatch(), cpp.AMI().tgtWeightsSum(), "tgt");
    }
}


// * * * * * * * * * * * * * * * * Constructors  * * * * * * * * * * * * * * //

Foam::functionObjects::AMIWeights::AMIWeights
(
    const word& name,
    const Time& runTime,
    const dictionary& dict
)
:
    fvMeshFunctionObject(name, runTime, dict),
    writeFile(mesh_, name, typeName, dict),
    writeFields_(false),
    patchIDs_()
{
    read(dict);
}


// * * * * * * * * * * * * * * * Member Functions  * * * * * * * * * * * * * //

bool Foam::functionObjects::AMIWeights::read(const dictionary& dict)
{
    if (fvMeshFunctionObject::read(dict) && writeFile::read(dict))
    {
        patchIDs_.clear();
        labelHashSet ids;

        for (const polyPatch& pp : mesh_.boundaryMesh())
        {
            const auto* amicpp = isA<cyclicAMIPolyPatch>(pp);

            if (amicpp && amicpp->owner())
            {
                ids.insert(pp.index());
            }
        }

        patchIDs_ = ids.sortedToc();

        writeFileHeader(file());

        writeFields_ = dict.get<bool>("writeFields");

        return true;
    }

    return false;
}


bool Foam::functionObjects::AMIWeights::execute()
{
    return true;
}


bool Foam::functionObjects::AMIWeights::write()
{
    Log << type() << " " << name() <<  " write:" << nl;

    const polyBoundaryMesh& pbm = mesh_.boundaryMesh();
    for (const label patchi : patchIDs_)
    {
        const polyPatch& pp = pbm[patchi];
        const auto& cpp = static_cast<const cyclicAMIPolyPatch&>(pp);

        reportPatch(cpp);

        if (writeFields_)
        {
            writeWeightFields(cpp);
        }
    }

    return true;
}


// ************************************************************************* //