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

   1 /*---------------------------------------------------------------------------*\
   2   =========                 |
   3   \\      /  F ield         | OpenFOAM: The Open Source CFD Toolbox
   4    \\    /   O peration     |
   5     \\  /    A nd           | Copyright (C) 2011 OpenFOAM Foundation
   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
  13     the Free Software Foundation, either version 3 of the License, or
  14     (at your 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, see <http://www.gnu.org/licenses/>.
  23
  24 \*---------------------------------------------------------------------------*/
  25
  26 #ifndef vtkPV3FoamAddToSelection_H
  27 #define vtkPV3FoamAddToSelection_H
  28
  29 // OpenFOAM includes
  30 #include "IOobjectList.H"
  31 #include "SortableList.H"
  32
  33 // VTK includes
  34 #include "vtkDataArraySelection.h"
  35
  36 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
  37
  38 template<class Type>
  39 Foam::label Foam::vtkPV3Foam::addToSelection
  40 (
  41     vtkDataArraySelection *select,
  42     const IOobjectList& objectLst,
  43     const string& suffix
  44 )
  45 {
  46     SortableList<word> names(objectLst.names(Type::typeName));
  47
  48     forAll(names, nameI)
  49     {
  50         if (suffix.size())
  51         {
  52             select->AddArray
  53             (
  54                 (names[nameI] + suffix).c_str()
  55             );
  56         }
  57         else
  58         {
  59             select->AddArray
  60             (
  61                 (names[nameI]).c_str()
  62             );
  63         }
  64     }
  65
  66     return names.size();
  67 }
  68
  69
  70 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
  71
  72 #endif
  73
  74 // ************************************************************************* //