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