4c223b8aee
This reference represents unnecessary storage. The mesh can be obtained from tracking data or passed to the particle evolution functions by argument. In addition, removing the mesh reference makes it possible to construct as particle from an Istream without the need for an iNew class. This simplifies stream-based transfer, and makes it possible for particles to be communicated by a polyDistributionMap.
228 lines
7.0 KiB
C++
228 lines
7.0 KiB
C++
/*---------------------------------------------------------------------------*\
|
|
========= |
|
|
\\ / F ield | OpenFOAM: The Open Source CFD Toolbox
|
|
\\ / O peration | Website: https://openfoam.org
|
|
\\ / A nd | Copyright (C) 2011-2022 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
|
|
|
|
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 "setWriter.H"
|
|
#include "systemDict.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"
|
|
|
|
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
|
|
|
|
const fileName vtkPath(runTime.rootPath()/runTime.globalCaseName()/"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.userTimeName() << endl;
|
|
|
|
mesh.readUpdate();
|
|
|
|
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())
|
|
{
|
|
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;
|
|
|
|
|
|
// Calculate 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.userTimeName() << endl;
|
|
|
|
mesh.readUpdate();
|
|
|
|
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(mesh);
|
|
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 (allTracks.size() && Pstream::master())
|
|
{
|
|
DynamicList<point> allTrack;
|
|
DynamicList<label> allTrackIDs;
|
|
forAll(allTracks, trackI)
|
|
{
|
|
allTrack.append(allTracks[trackI]);
|
|
allTrackIDs.append(labelList(allTracks[trackI].size(), trackI));
|
|
}
|
|
|
|
Info<< "\nWriting particle tracks in " << setFormat
|
|
<< " format to " << vtkPath << nl << endl;
|
|
|
|
setWriter::New(setFormat, propsDict)->write
|
|
(
|
|
vtkPath,
|
|
"tracks",
|
|
coordSet(allTrackIDs, word::null, pointField(allTrack))
|
|
);
|
|
}
|
|
|
|
Info<< "End\n" << endl;
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
// ************************************************************************* //
|