applications/utilities/postProcessing/graphics/PV3FoamReader/vtkPV3Foam/vtkPV3FoamAddToSelection.H

   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 \*---------------------------------------------------------------------------*/
  26
  27 #ifndef vtkPV3FoamAddToSelection_H
  28 #define vtkPV3FoamAddToSelection_H
  29
  30 // FOAM includes
  31 #include "IOobjectList.H"
  32 #include "SortableList.H"
  33
  34 // VTK includes
  35 #include "vtkDataArraySelection.h"
  36
  37 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
  38
  39 template<class Type>
  40 Foam::label Foam::vtkPV3Foam::addToSelection
  41 (
  42     vtkDataArraySelection *select,
  43     const IOobjectList& objectLst,
  44     const string& suffix
  45 )
  46 {
  47     SortableList<word> names(objectLst.names(Type::typeName));
  48
  49     forAll(names, nameI)
  50     {
  51         if (suffix.size())
  52         {
  53             select->AddArray((names[nameI] + suffix).c_str());
  54         }
  55         else
  56         {
  57             select->AddArray(names[nameI].c_str());
  58         }
  59     }
  60
  61     return names.size();
  62 }
  63
  64
  65 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
  66
  67 #endif
  68
  69 // ************************************************************************* //