applications/utilities/postProcessing/graphics/PVFoamReader/vtkFoam/vtkFoamConvertPatchPointField.H

   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 InClass
  26     Foam::vtkFoam
  27
  28 \*---------------------------------------------------------------------------*/
  29
  30 #ifndef vtkFoamConvertPatchPointField_H
  31 #define vtkFoamConvertPatchPointField_H
  32
  33 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
  34
  35 template<class Type>
  36 void Foam::vtkFoam::convertPatchPointField
  37 (
  38     const word& name,
  39     const Field<Type>& pptf,
  40     const label regioni
  41 )
  42 {
  43     vtkUnstructuredGrid *vtkMesh =
  44         vtkUnstructuredGrid::SafeDownCast(reader_->GetOutput(regioni));
  45
  46     vtkFloatArray *pointTypes = vtkFloatArray::New();
  47     pointTypes->SetNumberOfTuples(pptf.size());
  48     pointTypes->SetNumberOfComponents(Type::nComponents);
  49     pointTypes->Allocate(Type::nComponents*pptf.size());
  50     pointTypes->SetName(name.c_str());
  51
  52     float vec[Type::nComponents];
  53
  54     forAll(pptf, i)
  55     {
  56         for (direction d=0; d<Type::nComponents; d++)
  57         {
  58             vec[d] = pptf[i][d];
  59         }
  60
  61         pointTypes->InsertTuple(i, vec);
  62     }
  63
  64     vtkMesh->GetPointData()->AddArray(pointTypes);
  65     pointTypes->Delete();
  66 }
  67
  68
  69 template<>
  70 void Foam::vtkFoam::convertPatchPointField
  71 (
  72     const word& name,
  73     const Field<scalar>& ppsf,
  74     const label regioni
  75 )
  76 {
  77     vtkUnstructuredGrid *vtkMesh =
  78         vtkUnstructuredGrid::SafeDownCast(reader_->GetOutput(regioni));
  79
  80     vtkFloatArray *pointScalars = vtkFloatArray::New();
  81     pointScalars->SetNumberOfTuples(ppsf.size());
  82     pointScalars->SetNumberOfComponents(1);
  83     pointScalars->Allocate(ppsf.size());
  84     pointScalars->SetName(name.c_str());
  85
  86     for (int i=0; i<ppsf.size(); i++)
  87     {
  88         pointScalars->InsertComponent(i, 0, ppsf[i]);
  89     }
  90
  91     vtkMesh->GetPointData()->AddArray(pointScalars);
  92     if (!vtkMesh->GetPointData()->GetScalars())
  93     {
  94         vtkMesh->GetPointData()->SetScalars(pointScalars);
  95     }
  96
  97     pointScalars->Delete();
  98 }
  99
 100
 101 // * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //
 102
 103 #endif
 104
 105 // ************************************************************************* //