src/sampling/sampledSet/cloud/cloudSet.C

   1 /*---------------------------------------------------------------------------*\
   2   =========                 |
   3   \\      /  F ield         | OpenFOAM: The Open Source CFD Toolbox
   4    \\    /   O peration     |
   5     \\  /    A nd           | Copyright (C) 1991-2009 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 \*---------------------------------------------------------------------------*/
  26
  27 #include "cloudSet.H"
  28 #include <sampling/sampledSet.H>
  29 #include <meshTools/meshSearch.H>
  30 #include <OpenFOAM/DynamicList.H>
  31 #include <OpenFOAM/polyMesh.H>
  32 #include <OpenFOAM/addToRunTimeSelectionTable.H>
  33 #include <OpenFOAM/word.H>
  34
  35 // * * * * * * * * * * * * * * Static Data Members * * * * * * * * * * * * * //
  36
  37 namespace Foam
  38 {
  39     defineTypeNameAndDebug(cloudSet, 0);
  40     addToRunTimeSelectionTable(sampledSet, cloudSet, word);
  41 }
  42
  43
  44 // * * * * * * * * * * * * * Private Member Functions  * * * * * * * * * * * //
  45
  46 void Foam::cloudSet::calcSamples
  47 (
  48     DynamicList<point>& samplingPts,
  49     DynamicList<label>& samplingCells,
  50     DynamicList<label>& samplingFaces,
  51     DynamicList<label>& samplingSegments,
  52     DynamicList<scalar>& samplingCurveDist
  53 ) const
  54 {
  55     forAll(sampleCoords_, sampleI)
  56     {
  57         label cellI = searchEngine().findCell(sampleCoords_[sampleI]);
  58
  59         if (cellI != -1)
  60         {
  61             samplingPts.append(sampleCoords_[sampleI]);
  62             samplingCells.append(cellI);
  63             samplingFaces.append(-1);
  64             samplingSegments.append(0);
  65             samplingCurveDist.append(1.0 * sampleI);
  66         }
  67     }
  68 }
  69
  70
  71 void Foam::cloudSet::genSamples()
  72 {
  73     // Storage for sample points
  74     DynamicList<point> samplingPts;
  75     DynamicList<label> samplingCells;
  76     DynamicList<label> samplingFaces;
  77     DynamicList<label> samplingSegments;
  78     DynamicList<scalar> samplingCurveDist;
  79
  80     calcSamples
  81     (
  82         samplingPts,
  83         samplingCells,
  84         samplingFaces,
  85         samplingSegments,
  86         samplingCurveDist
  87     );
  88
  89     samplingPts.shrink();
  90     samplingCells.shrink();
  91     samplingFaces.shrink();
  92     samplingSegments.shrink();
  93     samplingCurveDist.shrink();
  94
  95     setSamples
  96     (
  97         samplingPts,
  98         samplingCells,
  99         samplingFaces,
 100         samplingSegments,
 101         samplingCurveDist
 102     );
 103 }
 104
 105
 106 // * * * * * * * * * * * * * * * * Constructors  * * * * * * * * * * * * * * //
 107
 108 Foam::cloudSet::cloudSet
 109 (
 110     const word& name,
 111     const polyMesh& mesh,
 112     meshSearch& searchEngine,
 113     const word& axis,
 114     const List<point>& sampleCoords
 115 )
 116 :
 117     sampledSet(name, mesh, searchEngine, axis),
 118     sampleCoords_(sampleCoords)
 119 {
 120     genSamples();
 121
 122     if (debug)
 123     {
 124         write(Info);
 125     }
 126 }
 127
 128
 129 Foam::cloudSet::cloudSet
 130 (
 131     const word& name,
 132     const polyMesh& mesh,
 133     meshSearch& searchEngine,
 134     const dictionary& dict
 135 )
 136 :
 137     sampledSet(name, mesh, searchEngine, dict),
 138     sampleCoords_(dict.lookup("points"))
 139 {
 140     genSamples();
 141
 142     if (debug)
 143     {
 144         write(Info);
 145     }
 146 }
 147
 148
 149 // * * * * * * * * * * * * * * * * Destructor  * * * * * * * * * * * * * * * //
 150
 151 Foam::cloudSet::~cloudSet()
 152 {}
 153
 154
 155 // * * * * * * * * * * * * * * * Member Functions  * * * * * * * * * * * * * //
 156
 157 Foam::point Foam::cloudSet::getRefPoint(const List<point>& pts) const
 158 {
 159     if (pts.size())
 160     {
 161         // Use first samplePt as starting point
 162         return pts[0];
 163     }
 164     else
 165     {
 166         return vector::zero;
 167     }
 168 }
 169
 170
 171 // ************************ vim: set sw=4 sts=4 et: ************************ //