applications/test/primitivePatch/testPrimitivePatch.C

   1 /*---------------------------------------------------------------------------*\
   2   =========                 |
   3   \\      /  F ield         | OpenFOAM: The Open Source CFD Toolbox
   4    \\    /   O peration     |
   5     \\  /    A nd           | Copyright (C) 1991-2008 OpenCFD Ltd.
   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     Test new primitive patches.
  27
  28 \*---------------------------------------------------------------------------*/
  29
  30 #include "argList.H"
  31 #include "Time.H"
  32 #include "polyMesh.H"
  33 #include "primitivePatch.H"
  34 #include "faceIOList.H"
  35 #include "IFstream.H"
  36 #include "OFstream.H"
  37
  38 using namespace Foam;
  39
  40 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
  41
  42 typedef PrimitivePatch<face, List, const pointField&> myPrimitivePatch;
  43
  44
  45 void writeObj(Ostream& os,const pointField& points)
  46 {
  47     forAll(points, pointI)
  48     {
  49         const point& pt = points[pointI];
  50
  51         os << "v " << pt.x() << ' ' << pt.y()
  52             << ' ' << pt.z() << endl;
  53     }
  54 }
  55
  56
  57 void checkFaceEdges
  58 (
  59     const faceList& localFaces,
  60     const edgeList& edges,
  61     const labelListList& faceEdges
  62 )
  63 {
  64     forAll(faceEdges, faceI)
  65     {
  66         const face& f = localFaces[faceI];
  67
  68         const labelList& myEdges = faceEdges[faceI];
  69
  70         forAll(f, fp)
  71         {
  72             label fp1 = (fp + 1) % f.size();
  73
  74             if (edges[myEdges[fp]] != edge(f[fp], f[fp1]))
  75             {
  76                 FatalErrorIn("checkFaceEdges")
  77                     << "Edges of face not in face point order:"
  78                     << "face:" << faceI << " localF:" << f
  79                     << " faceEdges:" << myEdges
  80                     << abort(FatalError);
  81             }
  82         }
  83     }
  84 }
  85
  86
  87 void writeEdges
  88 (
  89     const pointField& localPoints,
  90     const edgeList& edges,
  91     const label nInternalEdges
  92 )
  93 {
  94     Info<< "Writing internal edges to internalEdges.obj" << nl << endl;
  95
  96     OFstream intStream("internalEdges.obj");
  97
  98     writeObj(intStream, localPoints);
  99
 100     for (label edgeI = 0; edgeI < nInternalEdges; edgeI++)
 101     {
 102         const edge& e = edges[edgeI];
 103
 104         intStream << "l " << e.start()+1 << ' ' << e.end()+1 << endl;
 105     }
 106
 107     Info<< "Writing boundary edges to boundaryEdges.obj" << nl << endl;
 108
 109     OFstream bndStream("boundaryEdges.obj");
 110
 111     writeObj(bndStream, localPoints);
 112
 113     for (label edgeI = nInternalEdges; edgeI < edges.size(); edgeI++)
 114     {
 115         const edge& e = edges[edgeI];
 116
 117         bndStream << "l " << e.start()+1 << ' ' << e.end()+1 << endl;
 118     }
 119 }
 120
 121
 122 void writeFaceEdges
 123 (
 124     const pointField& localPoints,
 125     const edgeList& edges,
 126     const labelListList& faceEdges
 127 )
 128 {
 129     Info<< "Writing faceEdges per face to faceEdges.obj" << nl << endl;
 130
 131     OFstream feStream("faceEdges.obj");
 132
 133     writeObj(feStream, localPoints);
 134
 135     forAll(faceEdges, faceI)
 136     {
 137         const labelList& myEdges = faceEdges[faceI];
 138
 139         forAll(myEdges, i)
 140         {
 141             const edge& e = edges[myEdges[i]];
 142
 143             feStream<< "l " << e.start()+1 << ' ' << e.end()+1 << endl;
 144         }
 145     }
 146 }
 147
 148
 149 void writeEdgeFaces
 150 (
 151     const pointField& localPoints,
 152     const faceList& localFaces,
 153     const edgeList& edges,
 154     const labelListList& edgeFaces
 155 )
 156 {
 157     Info<< "Writing edgeFaces per face to edgeFaces.obj" << nl << endl;
 158
 159     OFstream efStream("edgeFaces.obj");
 160
 161     pointField ctrs(localFaces.size(), vector::zero);
 162
 163     forAll(localFaces, faceI)
 164     {
 165         ctrs[faceI] = localFaces[faceI].centre(localPoints);
 166     }
 167     writeObj(efStream, ctrs);
 168
 169
 170     forAll(edgeFaces, edgeI)
 171     {
 172         const labelList& myFaces = edgeFaces[edgeI];
 173
 174         forAll(myFaces, i)
 175         {
 176             efStream<< "l " << myFaces[0]+1 << ' ' << myFaces[i]+1 << endl;
 177         }
 178     }
 179 }
 180
 181
 182 void writeFaceFaces
 183 (
 184     const pointField& localPoints,
 185     const faceList& localFaces,
 186     const labelListList& faceFaces
 187 )
 188 {
 189     Info<< "Writing faceFaces per face to faceFaces.obj" << nl << endl;
 190
 191     OFstream ffStream("faceFaces.obj");
 192
 193     pointField ctrs(localFaces.size(), vector::zero);
 194
 195     forAll(localFaces, faceI)
 196     {
 197         ctrs[faceI] = localFaces[faceI].centre(localPoints);
 198     }
 199     writeObj(ffStream, ctrs);
 200
 201     forAll(faceFaces, faceI)
 202     {
 203         const labelList& nbrs = faceFaces[faceI];
 204
 205         forAll(nbrs, nbI)
 206         {
 207             ffStream << "l " << faceI+1 << ' ' << nbrs[nbI]+1 << endl;
 208         }
 209     }
 210 }
 211
 212
 213 // Main program:
 214
 215 int main(int argc, char *argv[])
 216 {
 217     argList::noParallel();
 218     argList::validArgs.append("patch");
 219
 220 #   include "setRootCase.H"
 221 #   include "createTime.H"
 222 #   include "createPolyMesh.H"
 223
 224     word patchName(args.additionalArgs()[0]);
 225
 226     label patchI = mesh.boundaryMesh().findPatchID(patchName);
 227
 228     const polyPatch& patch = mesh.boundaryMesh()[patchI];
 229
 230     Info<< "Patch:" << patch.name() << endl;
 231
 232     PrimitivePatch<face, List, const pointField&> pp(patch, patch.points());
 233
 234     const pointField& localPoints = pp.localPoints();
 235     const faceList& localFaces = pp.localFaces();
 236     const labelListList& faceFaces = pp.faceFaces();
 237     const edgeList& edges = pp.edges();
 238     const labelListList& edgeFaces = pp.edgeFaces();
 239     const labelListList& faceEdges = pp.faceEdges();
 240
 241
 242     checkFaceEdges(localFaces, edges, faceEdges);
 243
 244     writeEdges(localPoints, edges, pp.nInternalEdges());
 245
 246     writeFaceEdges(localPoints, edges, faceEdges);
 247
 248     writeEdgeFaces(localPoints, localFaces, edges, edgeFaces);
 249
 250     writeFaceFaces(localPoints, localFaces, faceFaces);
 251
 252     Info << "End\n" << endl;
 253
 254     return 0;
 255 }
 256
 257
 258 // ************************************************************************* //