smoothSurfaceData.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) 2022 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/>.
 
Application
    smoothSurfaceData
 
Description
    Pre-processing, filtering of surface field data.
    Currently this is just used to test filter settings
    (for MappedFile) and only generates vtk output, which can be
    easily loaded in ParaView.
 
\*---------------------------------------------------------------------------*/
 
#include "argList.H"
#include "Time.H"
#include "clockTime.H"
#include "primitiveFields.H"
#include "surfaceReader.H"
#include "surfaceWriter.H"
#include "foamVtkSurfaceWriter.H"
#include "MappedFileFilterField.H"
 
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
 
using namespace Foam;
 
 
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
 
int main(int argc, char *argv[])
{
    argList::addNote
    (
        "Testing, pre-processing, filtering of surface field data"
    );
 
    argList::noCheckProcessorDirectories();
    argList::addVerboseOption();
 
    argList::addArgument("input", "The input surface file");
 
    argList::addOption("radius", "m", "Specify filter radius [metres]");
    argList::addOption("sweeps", "N", "Number of median filter stages");
    argList::addOption
    (
        "field",
        "name",
        "Field <scalar> to process (default: T)"
    );
 
    argList::addOption
    (
        "read-format",
        "type",
        "Input format (default: ensight)"
    );
 
    argList args(argc, argv);
    // #include "setRootCase.H"
 
    const int optVerbose = args.verbose();
 
    const word readFileType
    (
        args.getOrDefault<word>("read-format", "ensight")
    );
 
    // Constant radius searching
    label filterSweeps_(1);
    scalar filterRadius_(0);
    args.readIfPresent("sweeps", filterSweeps_);
    args.readIfPresent("radius", filterRadius_);
 
    Info<< nl
        << "Filter: radius=" << filterRadius_
        << " sweeps=" << filterSweeps_ << endl;
 
    // Simple sanity check
    if ((filterSweeps_ < 1) || (filterRadius_ <= VSMALL))
    {
        Info<< nl << "Nothing to do. Exiting..." << nl << endl;
        return 0;
    }
 
    word fieldName("T");
    args.readIfPresent("field", fieldName);
 
 
    const fileName importName = args.get<fileName>(1);
 
    auto readerPtr_ = surfaceReader::New(readFileType, importName);
 
    auto& reader = readerPtr_();
 
    const label fieldIndex = reader.fieldNames(0).find(fieldName);
    if (fieldIndex == -1)
    {
        FatalErrorInFunction
            << "Unable to find field name: " << fieldName
            << " in list of available fields: " << reader.fieldNames(0)
            << exit(FatalError);
    }
 
    clockTime timing;
 
    const meshedSurface& geom = reader.geometry(0);
 
    Info<< nl << "Read " << geom.nFaces() << " faces and "
        << geom.nPoints() << " points in "
        << timing.timeIncrement() << "s" << endl;
 
    PatchFunction1Types::FilterField fieldFilter;
 
    PatchFunction1Types::FilterField::debug = optVerbose;
 
    fieldFilter.reset(geom, filterRadius_);
 
    Info<< nl << "Built weights/addressing "
        << timing.timeIncrement() << "s" << endl;
 
    Info<< nl << "Processing " << reader.times().size() << " times" << nl;
 
    const instantList times(reader.times());
 
    forAll(times, timeIndex)
    {
        tmp<scalarField> tfield
        (
            reader.field(timeIndex, fieldIndex, pTraits<scalar>::zero)
        );
 
        tfield = fieldFilter.evaluate(tfield, filterSweeps_);
 
        vtk::surfaceWriter writer
        (
            geom.points(),
            geom,
            word::printf("filtered_%06d", timeIndex)
        );
 
        writer.piece(geom.points(), geom);
 
        writer.writeGeometry();
        writer.beginCellData();
        writer.writeCellData(fieldName, tfield());
    }
 
    Info<< nl << "Smoothing/writing "
            << timing.timeIncrement() << "s" << endl;
 
    Info<< nl << "End\n" << endl;
 
    return 0;
}
 
 
// ************************************************************************* //