346 lines
9.6 KiB
C++
346 lines
9.6 KiB
C++
/*---------------------------------------------------------------------------*\
|
|
========= |
|
|
\\ / F ield | foam-extend: Open Source CFD
|
|
\\ / O peration | Version: 4.1
|
|
\\ / A nd | Web: http://www.foam-extend.org
|
|
\\/ M anipulation | For copyright notice see file Copyright
|
|
-------------------------------------------------------------------------------
|
|
License
|
|
This file is part of foam-extend.
|
|
|
|
foam-extend 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.
|
|
|
|
foam-extend 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 foam-extend. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
Application
|
|
transformPoints
|
|
|
|
Description
|
|
Transforms the mesh points in the polyMesh directory according to the
|
|
translate, rotate and scale options.
|
|
|
|
Usage
|
|
Options are:
|
|
|
|
-translate vector
|
|
Translates the points by the given vector,
|
|
|
|
-rotate (vector vector)
|
|
Rotates the points from the first vector to the second,
|
|
|
|
or -yawPitchRoll (yawdegrees pitchdegrees rolldegrees)
|
|
or -rollPitchYaw (rolldegrees pitchdegrees yawdegrees)
|
|
or -rotateAlongVector (vector and angle)
|
|
|
|
-scale vector
|
|
Scales the points by the given vector.
|
|
|
|
-cylToCart (vector vector)
|
|
Assumes that constant/points is defined in cylindrical coordinates:
|
|
(radialPosition tangentialPosition axialPosition) for a coordinate
|
|
system with origin: first vec, axis: second vec, direction: third vec
|
|
Radial and axial positions should be in [m].
|
|
Tangential positions should be in [radians].
|
|
Transforms the points to Cartesian positions.
|
|
|
|
The any or all of the three options may be specified and are processed
|
|
in the above order.
|
|
|
|
With -rotateFields (in combination with -rotate/yawPitchRoll/rollPitchYaw)
|
|
it will also read & transform vector & tensor fields.
|
|
|
|
Note:
|
|
yaw (rotation about z)
|
|
pitch (rotation about y)
|
|
roll (rotation about x)
|
|
|
|
\*---------------------------------------------------------------------------*/
|
|
|
|
#include "argList.H"
|
|
#include "objectRegistry.H"
|
|
#include "foamTime.H"
|
|
#include "fvMesh.H"
|
|
#include "volFields.H"
|
|
#include "surfaceFields.H"
|
|
#include "ReadFields.H"
|
|
#include "pointFields.H"
|
|
#include "transformField.H"
|
|
#include "transformGeometricField.H"
|
|
#include "IStringStream.H"
|
|
#include "RodriguesRotation.H"
|
|
#include "cylindricalCS.H"
|
|
|
|
using namespace Foam;
|
|
using namespace Foam::mathematicalConstant;
|
|
|
|
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
|
|
|
|
template<class GeoField>
|
|
void readAndRotateFields
|
|
(
|
|
PtrList<GeoField>& flds,
|
|
const fvMesh& mesh,
|
|
const tensor& T,
|
|
const IOobjectList& objects
|
|
)
|
|
{
|
|
ReadFields(mesh, objects, flds);
|
|
forAll(flds, i)
|
|
{
|
|
Info<< "Transforming " << flds[i].name() << endl;
|
|
dimensionedTensor dimT("t", flds[i].dimensions(), T);
|
|
transform(flds[i], dimT, flds[i]);
|
|
}
|
|
}
|
|
|
|
|
|
void rotateFields(const argList& args, const Time& runTime, const tensor& T)
|
|
{
|
|
# include "createNamedMesh.H"
|
|
|
|
// Read objects in time directory
|
|
IOobjectList objects(mesh, runTime.timeName());
|
|
|
|
// Read vol fields.
|
|
|
|
PtrList<volScalarField> vsFlds;
|
|
readAndRotateFields(vsFlds, mesh, T, objects);
|
|
|
|
PtrList<volVectorField> vvFlds;
|
|
readAndRotateFields(vvFlds, mesh, T, objects);
|
|
|
|
PtrList<volSphericalTensorField> vstFlds;
|
|
readAndRotateFields(vstFlds, mesh, T, objects);
|
|
|
|
PtrList<volSymmTensorField> vsymtFlds;
|
|
readAndRotateFields(vsymtFlds, mesh, T, objects);
|
|
|
|
PtrList<volTensorField> vtFlds;
|
|
readAndRotateFields(vtFlds, mesh, T, objects);
|
|
|
|
// Read surface fields.
|
|
|
|
PtrList<surfaceScalarField> ssFlds;
|
|
readAndRotateFields(ssFlds, mesh, T, objects);
|
|
|
|
PtrList<surfaceVectorField> svFlds;
|
|
readAndRotateFields(svFlds, mesh, T, objects);
|
|
|
|
PtrList<surfaceSphericalTensorField> sstFlds;
|
|
readAndRotateFields(sstFlds, mesh, T, objects);
|
|
|
|
PtrList<surfaceSymmTensorField> ssymtFlds;
|
|
readAndRotateFields(ssymtFlds, mesh, T, objects);
|
|
|
|
PtrList<surfaceTensorField> stFlds;
|
|
readAndRotateFields(stFlds, mesh, T, objects);
|
|
|
|
mesh.write();
|
|
}
|
|
|
|
|
|
// Main program:
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
# include "addRegionOption.H"
|
|
argList::validOptions.insert("translate", "vector");
|
|
argList::validOptions.insert("rotate", "(vector vector)");
|
|
argList::validOptions.insert("rotateAlongVector", "(vector angleInDegree)");
|
|
argList::validOptions.insert("rollPitchYaw", "(roll pitch yaw)");
|
|
argList::validOptions.insert("yawPitchRoll", "(yaw pitch roll)");
|
|
argList::validOptions.insert("rotateFields", "");
|
|
argList::validOptions.insert("scale", "vector");
|
|
argList::validOptions.insert("cylToCart", "(originVec axisVec directionVec)");
|
|
|
|
# include "setRootCase.H"
|
|
# include "createTime.H"
|
|
|
|
if (args.options().empty())
|
|
{
|
|
FatalErrorIn(args.executable())
|
|
<< "No options supplied, please use one or more of "
|
|
"-translate, -rotate, -scale, or -cylToCart options."
|
|
<< exit(FatalError);
|
|
}
|
|
|
|
word regionName = polyMesh::defaultRegion;
|
|
fileName meshDir;
|
|
|
|
if (args.optionReadIfPresent("region", regionName))
|
|
{
|
|
meshDir = regionName/polyMesh::meshSubDir;
|
|
}
|
|
else
|
|
{
|
|
meshDir = polyMesh::meshSubDir;
|
|
}
|
|
|
|
pointIOField points
|
|
(
|
|
IOobject
|
|
(
|
|
"points",
|
|
runTime.findInstance(meshDir, "points"),
|
|
meshDir,
|
|
runTime,
|
|
IOobject::MUST_READ,
|
|
IOobject::NO_WRITE,
|
|
false
|
|
)
|
|
);
|
|
|
|
// Translation options
|
|
|
|
if (args.optionFound("translate"))
|
|
{
|
|
vector transVector(args.optionLookup("translate")());
|
|
|
|
Info<< "Translating points by " << transVector << endl;
|
|
|
|
points += transVector;
|
|
}
|
|
|
|
// Rotation options
|
|
|
|
if (args.optionFound("rotate"))
|
|
{
|
|
Pair<vector> n1n2(args.optionLookup("rotate")());
|
|
n1n2[0] /= mag(n1n2[0]);
|
|
n1n2[1] /= mag(n1n2[1]);
|
|
tensor T = rotationTensor(n1n2[0], n1n2[1]);
|
|
|
|
Info<< "Rotating points by " << T << endl;
|
|
|
|
points = transform(T, points);
|
|
|
|
if (args.optionFound("rotateFields"))
|
|
{
|
|
rotateFields(args, runTime, T);
|
|
}
|
|
}
|
|
else if (args.optionFound("rollPitchYaw"))
|
|
{
|
|
vector v(args.optionLookup("rollPitchYaw")());
|
|
|
|
Info<< "Rotating points by" << nl
|
|
<< " roll " << v.x() << nl
|
|
<< " pitch " << v.y() << nl
|
|
<< " yaw " << v.z() << endl;
|
|
|
|
|
|
// Convert to radians
|
|
v *= pi/180.0;
|
|
|
|
quaternion R(v.x(), v.y(), v.z());
|
|
|
|
Info<< "Rotating points by quaternion " << R << endl;
|
|
points = transform(R, points);
|
|
|
|
if (args.optionFound("rotateFields"))
|
|
{
|
|
rotateFields(args, runTime, R.R());
|
|
}
|
|
}
|
|
else if (args.optionFound("yawPitchRoll"))
|
|
{
|
|
vector v(args.optionLookup("yawPitchRoll")());
|
|
|
|
Info<< "Rotating points by" << nl
|
|
<< " yaw " << v.x() << nl
|
|
<< " pitch " << v.y() << nl
|
|
<< " roll " << v.z() << endl;
|
|
|
|
|
|
// Convert to radians
|
|
v *= pi/180.0;
|
|
|
|
scalar yaw = v.x();
|
|
scalar pitch = v.y();
|
|
scalar roll = v.z();
|
|
|
|
quaternion R = quaternion(vector(0, 0, 1), yaw);
|
|
R *= quaternion(vector(0, 1, 0), pitch);
|
|
R *= quaternion(vector(1, 0, 0), roll);
|
|
|
|
Info<< "Rotating points by quaternion " << R << endl;
|
|
points = transform(R, points);
|
|
|
|
if (args.optionFound("rotateFields"))
|
|
{
|
|
rotateFields(args, runTime, R.R());
|
|
}
|
|
}
|
|
else if (args.optionFound("rotateAlongVector"))
|
|
{
|
|
vector rotationAxis;
|
|
scalar rotationAngle;
|
|
|
|
args.optionLookup("rotateAlongVector")()
|
|
>> rotationAxis
|
|
>> rotationAngle;
|
|
|
|
tensor T = RodriguesRotation(rotationAxis, rotationAngle);
|
|
|
|
Info << "Rotating points by " << T << endl;
|
|
|
|
points = transform(T, points);
|
|
|
|
if (args.options().found("rotateFields"))
|
|
{
|
|
rotateFields(args, runTime, T);
|
|
}
|
|
}
|
|
|
|
// Scale options
|
|
|
|
if (args.optionFound("scale"))
|
|
{
|
|
vector scaleVector(args.optionLookup("scale")());
|
|
|
|
Info<< "Scaling points by " << scaleVector << endl;
|
|
|
|
points.replace(vector::X, scaleVector.x()*points.component(vector::X));
|
|
points.replace(vector::Y, scaleVector.y()*points.component(vector::Y));
|
|
points.replace(vector::Z, scaleVector.z()*points.component(vector::Z));
|
|
}
|
|
|
|
if (args.optionFound("cylToCart"))
|
|
//HN, 140523
|
|
{
|
|
vectorField n1n2(args.optionLookup("cylToCart")());
|
|
n1n2[1] /= mag(n1n2[1]);
|
|
n1n2[2] /= mag(n1n2[2]);
|
|
|
|
cylindricalCS ccs
|
|
(
|
|
"ccs",
|
|
n1n2[0],
|
|
n1n2[1],
|
|
n1n2[2],
|
|
false // Use radians
|
|
);
|
|
|
|
points = ccs.globalPosition(points);
|
|
}
|
|
|
|
// Set the precision of the points data to 10
|
|
IOstream::defaultPrecision(10);
|
|
|
|
Info << "Writing points into directory " << points.path() << nl << endl;
|
|
points.write();
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
// ************************************************************************* //
|