260 lines
7.5 KiB
C
260 lines
7.5 KiB
C
/*---------------------------------------------------------------------------*\
|
|
========= |
|
|
\\ / F ield | OpenFOAM: The Open Source CFD Toolbox
|
|
\\ / O peration |
|
|
\\ / A nd | Copyright (C) 2011-2016 OpenFOAM Foundation
|
|
\\/ M anipulation |
|
|
-------------------------------------------------------------------------------
|
|
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
|
|
particleTracks
|
|
|
|
Group
|
|
grpPostProcessingUtilities
|
|
|
|
Description
|
|
Generates a VTK file of particle tracks for cases that were computed using
|
|
a tracked-parcel-type cloud.
|
|
|
|
\*---------------------------------------------------------------------------*/
|
|
|
|
#include "argList.H"
|
|
#include "Cloud.H"
|
|
#include "IOdictionary.H"
|
|
#include "fvMesh.H"
|
|
#include "Time.H"
|
|
#include "timeSelector.H"
|
|
#include "OFstream.H"
|
|
#include "passiveParticleCloud.H"
|
|
#include "writer.H"
|
|
|
|
using namespace Foam;
|
|
|
|
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
timeSelector::addOptions();
|
|
#include "addRegionOption.H"
|
|
|
|
#include "setRootCase.H"
|
|
|
|
#include "createTime.H"
|
|
instantList timeDirs = timeSelector::select0(runTime, args);
|
|
#include "createNamedMesh.H"
|
|
#include "createFields.H"
|
|
|
|
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
|
|
|
|
fileName vtkPath(runTime.path()/"VTK");
|
|
mkDir(vtkPath);
|
|
|
|
Info<< "Scanning times to determine track data for cloud " << cloudName
|
|
<< nl << endl;
|
|
|
|
labelList maxIds(Pstream::nProcs(), -1);
|
|
forAll(timeDirs, timeI)
|
|
{
|
|
runTime.setTime(timeDirs[timeI], timeI);
|
|
Info<< "Time = " << runTime.timeName() << endl;
|
|
|
|
Info<< " Reading particle positions" << endl;
|
|
passiveParticleCloud myCloud(mesh, cloudName);
|
|
Info<< " Read " << returnReduce(myCloud.size(), sumOp<label>())
|
|
<< " particles" << endl;
|
|
|
|
forAllConstIter(passiveParticleCloud, myCloud, iter)
|
|
{
|
|
label origId = iter().origId();
|
|
label origProc = iter().origProc();
|
|
|
|
if (origProc >= maxIds.size())
|
|
{
|
|
// Expand size
|
|
maxIds.setSize(origProc+1, -1);
|
|
}
|
|
|
|
maxIds[origProc] = max(maxIds[origProc], origId);
|
|
}
|
|
}
|
|
|
|
label maxNProcs = returnReduce(maxIds.size(), maxOp<label>());
|
|
|
|
Info<< "Detected particles originating from " << maxNProcs
|
|
<< " processors." << nl << endl;
|
|
|
|
maxIds.setSize(maxNProcs, -1);
|
|
|
|
Pstream::listCombineGather(maxIds, maxEqOp<label>());
|
|
Pstream::listCombineScatter(maxIds);
|
|
|
|
labelList numIds = maxIds + 1;
|
|
|
|
Info<< nl << "Particle statistics:" << endl;
|
|
forAll(maxIds, procI)
|
|
{
|
|
Info<< " Found " << numIds[procI] << " particles originating"
|
|
<< " from processor " << procI << endl;
|
|
}
|
|
Info<< nl << endl;
|
|
|
|
|
|
// calc starting ids for particles on each processor
|
|
List<label> startIds(numIds.size(), 0);
|
|
for (label i = 0; i < numIds.size()-1; i++)
|
|
{
|
|
startIds[i+1] += startIds[i] + numIds[i];
|
|
}
|
|
label nParticle = startIds.last() + numIds[startIds.size()-1];
|
|
|
|
|
|
|
|
// number of tracks to generate
|
|
label nTracks = nParticle/sampleFrequency;
|
|
|
|
// storage for all particle tracks
|
|
List<DynamicList<vector>> allTracks(nTracks);
|
|
|
|
Info<< "\nGenerating " << nTracks << " particle tracks for cloud "
|
|
<< cloudName << nl << endl;
|
|
|
|
forAll(timeDirs, timeI)
|
|
{
|
|
runTime.setTime(timeDirs[timeI], timeI);
|
|
Info<< "Time = " << runTime.timeName() << endl;
|
|
|
|
List<pointField> allPositions(Pstream::nProcs());
|
|
List<labelField> allOrigIds(Pstream::nProcs());
|
|
List<labelField> allOrigProcs(Pstream::nProcs());
|
|
|
|
// Read particles. Will be size 0 if no particles.
|
|
Info<< " Reading particle positions" << endl;
|
|
passiveParticleCloud myCloud(mesh, cloudName);
|
|
|
|
// collect the track data on all processors that have positions
|
|
allPositions[Pstream::myProcNo()].setSize
|
|
(
|
|
myCloud.size(),
|
|
point::zero
|
|
);
|
|
allOrigIds[Pstream::myProcNo()].setSize(myCloud.size(), 0);
|
|
allOrigProcs[Pstream::myProcNo()].setSize(myCloud.size(), 0);
|
|
|
|
label i = 0;
|
|
forAllConstIter(passiveParticleCloud, myCloud, iter)
|
|
{
|
|
allPositions[Pstream::myProcNo()][i] = iter().position();
|
|
allOrigIds[Pstream::myProcNo()][i] = iter().origId();
|
|
allOrigProcs[Pstream::myProcNo()][i] = iter().origProc();
|
|
i++;
|
|
}
|
|
|
|
// collect the track data on the master processor
|
|
Pstream::gatherList(allPositions);
|
|
Pstream::gatherList(allOrigIds);
|
|
Pstream::gatherList(allOrigProcs);
|
|
|
|
Info<< " Constructing tracks" << nl << endl;
|
|
if (Pstream::master())
|
|
{
|
|
forAll(allPositions, procI)
|
|
{
|
|
forAll(allPositions[procI], i)
|
|
{
|
|
label globalId =
|
|
startIds[allOrigProcs[procI][i]]
|
|
+ allOrigIds[procI][i];
|
|
|
|
if (globalId % sampleFrequency == 0)
|
|
{
|
|
label trackId = globalId/sampleFrequency;
|
|
if (allTracks[trackId].size() < maxPositions)
|
|
{
|
|
allTracks[trackId].append
|
|
(
|
|
allPositions[procI][i]
|
|
);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
if (Pstream::master())
|
|
{
|
|
PtrList<coordSet> tracks(allTracks.size());
|
|
forAll(allTracks, trackI)
|
|
{
|
|
tracks.set
|
|
(
|
|
trackI,
|
|
new coordSet
|
|
(
|
|
"track" + Foam::name(trackI),
|
|
"distance"
|
|
)
|
|
);
|
|
tracks[trackI].transfer(allTracks[trackI]);
|
|
}
|
|
|
|
autoPtr<writer<scalar>> scalarFormatterPtr = writer<scalar>::New
|
|
(
|
|
setFormat
|
|
);
|
|
|
|
//OFstream vtkTracks(vtkPath/"particleTracks.vtk");
|
|
fileName vtkFile
|
|
(
|
|
scalarFormatterPtr().getFileName
|
|
(
|
|
tracks[0],
|
|
wordList(0)
|
|
)
|
|
);
|
|
|
|
OFstream vtkTracks
|
|
(
|
|
vtkPath
|
|
/ "particleTracks." + vtkFile.ext()
|
|
);
|
|
|
|
Info<< "\nWriting particle tracks in " << setFormat
|
|
<< " format to " << vtkTracks.name()
|
|
<< nl << endl;
|
|
|
|
|
|
scalarFormatterPtr().write
|
|
(
|
|
true, // writeTracks
|
|
tracks,
|
|
wordList(0),
|
|
List<List<scalarField>>(0),
|
|
vtkTracks
|
|
);
|
|
}
|
|
|
|
Info<< "End\n" << endl;
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
// ************************************************************************* //
|