applications/utilities/surface/surfaceTransformPoints/surfaceTransformPoints.C

   1 /*---------------------------------------------------------------------------*\
   2   =========                 |
   3   \\      /  F ield         | OpenFOAM: The Open Source CFD Toolbox
   4    \\    /   O peration     |
   5     \\  /    A nd           | Copyright held by original author
   6      \\/     M anipulation  |
   7 -------------------------------------------------------------------------------
   8 License
   9     This file is part of OpenFOAM.
  10
  11     OpenFOAM is free software; you can redistribute it and/or modify it
  12     under the terms of the GNU General Public License as published by the
  13     Free Software Foundation; either version 2 of the License, or (at your
  14     option) any later version.
  15
  16     OpenFOAM is distributed in the hope that it will be useful, but WITHOUT
  17     ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  18     FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  19     for more details.
  20
  21     You should have received a copy of the GNU General Public License
  22     along with OpenFOAM; if not, write to the Free Software Foundation,
  23     Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
  24
  25 Description
  26     Transform (scale/rotate) a surface. Like transformPoints but then for
  27     surfaces.
  28
  29     The rollPitchYaw option takes three angles (degrees):
  30     - roll (rotation about x) followed by
  31     - pitch (rotation about y) followed by
  32     - yaw (rotation about z)
  33
  34     The yawPitchRoll does yaw followed by pitch followed by roll.
  35
  36 \*---------------------------------------------------------------------------*/
  37
  38 #include "triSurface.H"
  39 #include "argList.H"
  40 #include "OFstream.H"
  41 #include "IFstream.H"
  42 #include "boundBox.H"
  43 #include "transformField.H"
  44 #include "Pair.H"
  45 #include "quaternion.H"
  46
  47 using namespace Foam;
  48 using namespace Foam::mathematicalConstant;
  49
  50
  51 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
  52 // Main program:
  53
  54 int main(int argc, char *argv[])
  55 {
  56     argList::noParallel();
  57     argList::validArgs.clear();
  58
  59     argList::validArgs.append("surface file");
  60     argList::validArgs.append("output surface file");
  61     argList::validOptions.insert("translate", "vector");
  62     argList::validOptions.insert("rotate", "(vector vector)");
  63     argList::validOptions.insert("scale", "vector");
  64     argList::validOptions.insert("rollPitchYaw", "(roll pitch yaw)");
  65     argList::validOptions.insert("yawPitchRoll", "(yaw pitch roll)");
  66     argList args(argc, argv);
  67
  68     fileName surfFileName(args.additionalArgs()[0]);
  69
  70     Info<< "Reading surf from " << surfFileName << " ..." << endl;
  71
  72     fileName outFileName(args.additionalArgs()[1]);
  73
  74     Info<< "Writing surf to " << outFileName << " ..." << endl;
  75
  76
  77     if (args.options().empty())
  78     {
  79         FatalErrorIn(args.executable())
  80             << "No options supplied, please use one or more of "
  81                "-translate, -rotate or -scale options."
  82             << exit(FatalError);
  83     }
  84
  85     triSurface surf1(surfFileName);
  86
  87     pointField points(surf1.points());
  88
  89     if (args.optionFound("translate"))
  90     {
  91         vector transVector(args.optionLookup("translate")());
  92
  93         Info<< "Translating points by " << transVector << endl;
  94
  95         points += transVector;
  96     }
  97
  98     if (args.optionFound("rotate"))
  99     {
 100         Pair<vector> n1n2(args.optionLookup("rotate")());
 101         n1n2[0] /= mag(n1n2[0]);
 102         n1n2[1] /= mag(n1n2[1]);
 103
 104         tensor T = rotationTensor(n1n2[0], n1n2[1]);
 105
 106         Info<< "Rotating points by " << T << endl;
 107
 108         points = transform(T, points);
 109     }
 110     else if (args.optionFound("rollPitchYaw"))
 111     {
 112         vector v(args.optionLookup("rollPitchYaw")());
 113
 114         Info<< "Rotating points by" << nl
 115             << "    roll  " << v.x() << nl
 116             << "    pitch " << v.y() << nl
 117             << "    yaw   " << v.z() << endl;
 118
 119
 120         // Convert to radians
 121         v *= pi/180.0;
 122
 123         quaternion R(v.x(), v.y(), v.z());
 124
 125         Info<< "Rotating points by quaternion " << R << endl;
 126         points = transform(R, points);
 127     }
 128     else if (args.optionFound("yawPitchRoll"))
 129     {
 130         vector v(args.optionLookup("yawPitchRoll")());
 131
 132         Info<< "Rotating points by" << nl
 133             << "    yaw   " << v.x() << nl
 134             << "    pitch " << v.y() << nl
 135             << "    roll  " << v.z() << endl;
 136
 137
 138         // Convert to radians
 139         v *= pi/180.0;
 140
 141         scalar yaw = v.x();
 142         scalar pitch = v.y();
 143         scalar roll = v.z();
 144
 145         quaternion R = quaternion(vector(0, 0, 1), yaw);
 146         R *= quaternion(vector(0, 1, 0), pitch);
 147         R *= quaternion(vector(1, 0, 0), roll);
 148
 149         Info<< "Rotating points by quaternion " << R << endl;
 150         points = transform(R, points);
 151     }
 152
 153     if (args.optionFound("scale"))
 154     {
 155         vector scaleVector(args.optionLookup("scale")());
 156
 157         Info<< "Scaling points by " << scaleVector << endl;
 158
 159         points.replace(vector::X, scaleVector.x()*points.component(vector::X));
 160         points.replace(vector::Y, scaleVector.y()*points.component(vector::Y));
 161         points.replace(vector::Z, scaleVector.z()*points.component(vector::Z));
 162     }
 163
 164     triSurface surf2(surf1, surf1.patches(), points);
 165
 166     surf2.write(outFileName);
 167
 168     Info << "End\n" << endl;
 169
 170     return 0;
 171 }
 172
 173
 174 // ************************************************************************* //