Commit 02b6d0c6 authored by Andrew Heather's avatar Andrew Heather
Browse files

adding new particle tracks app - needs work...

parent 449be34d
particleTracks.C
EXE = $(FOAM_APPBIN)/particleTracks
EXE_INC = \
-I$(LIB_SRC)/finiteVolume/lnInclude \
-I$(LIB_SRC)/lagrangian/basic/lnInclude
EXE_LIBS = \
-lfiniteVolume \
-llagrangian
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"));
/*---------------------------------------------------------------------------*\
========= |
\\ / 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;
}
// ************************************************************************* //
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment