mirror of
https://develop.openfoam.com/Development/openfoam.git
synced 2025-11-28 03:28:01 +00:00
adding new particle tracks app - needs work...
This commit is contained in:
andy
@ -0,0 +1,3 @@
|
||||
particleTracks.C
|
||||
|
||||
EXE = $(FOAM_APPBIN)/particleTracks
|
||||
@ -0,0 +1,7 @@
|
||||
EXE_INC = \
|
||||
-I$(LIB_SRC)/finiteVolume/lnInclude \
|
||||
-I$(LIB_SRC)/lagrangian/basic/lnInclude
|
||||
|
||||
EXE_LIBS = \
|
||||
-lfiniteVolume \
|
||||
-llagrangian
|
||||
@ -0,0 +1,22 @@
|
||||
IOdictionary propsDict
|
||||
(
|
||||
IOobject
|
||||
(
|
||||
"particleTrackProperties",
|
||||
runTime.constant(),
|
||||
mesh,
|
||||
IOobject::MUST_READ
|
||||
)
|
||||
);
|
||||
|
||||
word cloudName(propsDict.lookup("cloudName"));
|
||||
|
||||
label sampleFrequency(readLabel(propsDict.lookup("sampleFrequency")));
|
||||
|
||||
// outputMode: compositeFile, filePerTrack
|
||||
//word outputmode(propsDict.lookup("outputMode"))
|
||||
|
||||
label maxPositions(readLabel(propsDict.lookup("maxPositions")));
|
||||
|
||||
// outputFormat: raw, vtk
|
||||
//word outputFormat(propsDict.lookup("outputFormat"));
|
||||
@ -0,0 +1,337 @@
|
||||
/*---------------------------------------------------------------------------*\
|
||||
========= |
|
||||
\\ / F ield | OpenFOAM: The Open Source CFD Toolbox
|
||||
\\ / O peration |
|
||||
\\ / A nd | Copyright (C) 1991-2009 OpenCFD Ltd.
|
||||
\\/ 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 2 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, write to the Free Software Foundation,
|
||||
Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
|
||||
|
||||
Application
|
||||
particleTracks
|
||||
|
||||
Description
|
||||
|
||||
\*---------------------------------------------------------------------------*/
|
||||
|
||||
#include "argList.H"
|
||||
#include "Cloud.H"
|
||||
#include "IOdictionary.H"
|
||||
#include "fvMesh.H"
|
||||
#include "Time.H"
|
||||
#include "timeSelector.H"
|
||||
#include "OFstream.H"
|
||||
#include "passiveParticle.H"
|
||||
#include "SortableList.H"
|
||||
|
||||
using namespace Foam;
|
||||
|
||||
Foam::label checkHeader(IOobject& io)
|
||||
{
|
||||
if (io.headerOk())
|
||||
{
|
||||
return 0;
|
||||
}
|
||||
else
|
||||
{
|
||||
Info<< " no " << io.name() << " field" << endl;
|
||||
return 1;
|
||||
}
|
||||
}
|
||||
|
||||
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
|
||||
|
||||
int main(int argc, char *argv[])
|
||||
{
|
||||
# include "setRootCase.H"
|
||||
|
||||
# include "createTime.H"
|
||||
instantList timeDirs = timeSelector::select0(runTime, args);
|
||||
# include "createMesh.H"
|
||||
# include "createFields.H"
|
||||
|
||||
|
||||
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
|
||||
|
||||
/*
|
||||
Parallel mode
|
||||
- nProcs known
|
||||
- scan through times to determine max id for each proc
|
||||
|
||||
- scan through times to populate for each particle (indexed to procId):
|
||||
- id
|
||||
- time
|
||||
- construct single flat list from all procs of ids and times:
|
||||
- starting id = sum of ids of lower procs
|
||||
- sort particles according to time
|
||||
- join the dots...
|
||||
|
||||
Serial mode
|
||||
- don't know if case was run in serial mode, or is a reconstructed
|
||||
parallel run
|
||||
- scan times to determine max origProcId and max id for each origProc
|
||||
|
||||
- scan through times to populate for each particle (indexed to procId):
|
||||
- id
|
||||
- time
|
||||
- construct single flat list from all procs of ids and times:
|
||||
- starting id = sum of ids of lower procs
|
||||
- sort particles according to time
|
||||
- join the dots...
|
||||
|
||||
|
||||
*/
|
||||
|
||||
DynamicList<label> maxIds;
|
||||
labelHashSet validIds;
|
||||
if (Pstream::parRun())
|
||||
{
|
||||
maxIds(Pstream::myProcNo()) = -1;
|
||||
forAll(timeDirs, timeI)
|
||||
{
|
||||
runTime.setTime(timeDirs[timeI], timeI);
|
||||
Info<< "Time = " << runTime.timeName() << endl;
|
||||
|
||||
IOobject idHeader
|
||||
(
|
||||
"id",
|
||||
runTime.timeName(),
|
||||
cloud::prefix/cloudName,
|
||||
mesh,
|
||||
IOobject::MUST_READ
|
||||
);
|
||||
if (idHeader.headerOk())
|
||||
{
|
||||
IOField<label> id(idHeader);
|
||||
maxIds[Pstream::myProcNo()]
|
||||
= max(maxIds[Pstream::myProcNo()], max(id));
|
||||
}
|
||||
}
|
||||
Pstream::listCombineGather(maxIds, maxOp<label>());
|
||||
// reduce(maxIds, maxOp<DynamicList<label> >());
|
||||
}
|
||||
else
|
||||
{
|
||||
Info<< "Assuming case is a reconstructed parallel run" << nl;
|
||||
Info<< "Scanning time folders for tracking information" << nl << endl;
|
||||
forAll(timeDirs, timeI)
|
||||
{
|
||||
runTime.setTime(timeDirs[timeI], timeI);
|
||||
Info<< "Time = " << runTime.timeName() << endl;
|
||||
|
||||
label proceed = 1;
|
||||
IOobject origProcHeader
|
||||
(
|
||||
"origProc",
|
||||
runTime.timeName(),
|
||||
cloud::prefix/cloudName,
|
||||
mesh,
|
||||
IOobject::MUST_READ
|
||||
);
|
||||
proceed -= checkHeader(origProcHeader);
|
||||
|
||||
IOobject idHeader
|
||||
(
|
||||
"id",
|
||||
runTime.timeName(),
|
||||
cloud::prefix/cloudName,
|
||||
mesh,
|
||||
IOobject::MUST_READ
|
||||
);
|
||||
proceed -= checkHeader(idHeader);
|
||||
|
||||
if (proceed == 1)
|
||||
{
|
||||
IOField<label> origProc(origProcHeader);
|
||||
IOField<label> id(idHeader);
|
||||
forAll(origProc, i)
|
||||
{
|
||||
validIds.insert(origProc[i]);
|
||||
if (maxIds.size() > origProc[i])
|
||||
{
|
||||
// proc group already exists
|
||||
maxIds[origProc[i]] = max(maxIds[origProc[i]], id[i]);
|
||||
}
|
||||
else
|
||||
{
|
||||
maxIds(origProc[i]) = id[i];
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
maxIds.shrink();
|
||||
|
||||
// determine number of unique particles
|
||||
label nParticle = 0;
|
||||
forAll(maxIds, procI)
|
||||
{
|
||||
if (validIds.found(procI))
|
||||
{
|
||||
nParticle += maxIds[procI] + 1;
|
||||
}
|
||||
}
|
||||
|
||||
// calc starting ids for particles on each processor
|
||||
List<label> startIds(validIds.size(), 0);
|
||||
for (label i = 0; i < validIds.size(); i++)
|
||||
{
|
||||
if (validIds.found(i))
|
||||
{
|
||||
startIds[i+1] += startIds[i] + maxIds[i];
|
||||
}
|
||||
}
|
||||
|
||||
// number of tracks to generate
|
||||
label nTracks = nParticle/sampleFrequency;
|
||||
|
||||
// storage for all particle tracks
|
||||
List<DynamicList<vector> > allTracks(nTracks);
|
||||
|
||||
// storage for all particle times
|
||||
List<DynamicList<scalar> > allTimes(nTracks);
|
||||
|
||||
Info<< "\nGenerating " << nTracks << " particle tracks" << nl << endl;
|
||||
forAll(timeDirs, timeI)
|
||||
{
|
||||
runTime.setTime(timeDirs[timeI], timeI);
|
||||
Info<< "Time = " << runTime.timeName() << endl;
|
||||
|
||||
label proceed = 1;
|
||||
IOobject positionsHeader
|
||||
(
|
||||
"positions",
|
||||
runTime.timeName(),
|
||||
cloud::prefix/cloudName,
|
||||
mesh,
|
||||
IOobject::MUST_READ
|
||||
);
|
||||
proceed -= checkHeader(positionsHeader);
|
||||
|
||||
IOobject idHeader
|
||||
(
|
||||
"id",
|
||||
runTime.timeName(),
|
||||
cloud::prefix/cloudName,
|
||||
mesh,
|
||||
IOobject::MUST_READ
|
||||
);
|
||||
proceed -= checkHeader(idHeader);
|
||||
|
||||
if (proceed == 1)
|
||||
{
|
||||
Info<< " Reading particle positions" << endl;
|
||||
Cloud<passiveParticle> positions(mesh, cloudName, false);
|
||||
|
||||
if (positions.size() > 0)
|
||||
{
|
||||
IOField<label> id(idHeader);
|
||||
Info<< " Reading particle ids" << nl << endl;
|
||||
|
||||
label i = 0;
|
||||
forAllConstIter(Cloud<passiveParticle>, positions, iter)
|
||||
{
|
||||
label globalId = startIds[Pstream::myProcNo()] + id[i++];
|
||||
if (globalId % sampleFrequency == 0)
|
||||
{
|
||||
label trackId = globalId/sampleFrequency;
|
||||
if (allTracks[trackId].size() < maxPositions)
|
||||
{
|
||||
allTracks[trackId].append(iter().position());
|
||||
allTimes[trackId].append(timeDirs[timeI].value());
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
Info<< " No particles read" << nl << endl;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (Pstream::master)
|
||||
{
|
||||
Info<< "Writing particle tracks" << nl << endl;
|
||||
|
||||
forAll(allTracks, trackI)
|
||||
{
|
||||
Pstream::listCombineGather(allTracks[trackI], plusEqOp<vector>());
|
||||
Pstream::listCombineGather(allTimes[trackI], plusEqOp<scalar>());
|
||||
}
|
||||
|
||||
OFstream vtkTracks("particleTracks.vtk");
|
||||
|
||||
// Total number of points in tracks + 1 per track
|
||||
label nPoints = 0;
|
||||
forAll(allTracks, trackI)
|
||||
{
|
||||
nPoints += allTracks[trackI].size();
|
||||
}
|
||||
reduce(nPoints, sumOp<label>());
|
||||
|
||||
vtkTracks
|
||||
<< "# vtk DataFile Version 2.0\n"
|
||||
<< "particleTracks" << nl
|
||||
<< "ASCII\n"
|
||||
<< "DATASET POLYDATA\n"
|
||||
<< "POINTS " << nPoints << " float\n";
|
||||
|
||||
// Write track points to file
|
||||
forAll(allTracks, trackI)
|
||||
{
|
||||
SortableList<scalar> sortedTimes(allTimes[trackI]);
|
||||
|
||||
const DynamicList<vector>& trackPts = allTracks[trackI];
|
||||
forAll(trackPts, i)
|
||||
{
|
||||
label ptI = sortedTimes.indices()[i];
|
||||
const vector& pt = trackPts[ptI];
|
||||
vtkTracks << pt.x() << ' ' << pt.y() << ' ' << pt.z() << nl;
|
||||
}
|
||||
}
|
||||
|
||||
// write track (line) connectivity to file
|
||||
vtkTracks << "LINES " << nTracks << ' ' << nPoints+nTracks << nl;
|
||||
|
||||
// Write track points to file
|
||||
label globalPtI = 0;
|
||||
forAll(allTracks, trackI)
|
||||
{
|
||||
const DynamicList<vector>& trackPts = allTracks[trackI];
|
||||
|
||||
vtkTracks << trackPts.size();
|
||||
|
||||
forAll(trackPts, i)
|
||||
{
|
||||
vtkTracks << ' ' << globalPtI;
|
||||
globalPtI++;
|
||||
}
|
||||
|
||||
vtkTracks << nl;
|
||||
}
|
||||
|
||||
Info<< "end" << endl;
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
// ************************************************************************* //
|
||||
Reference in New Issue
Block a user