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