vtkPV3FoamPatchField.H 3.46 KB
Newer Older
1
2
3
4
/*---------------------------------------------------------------------------*\
  =========                 |
  \\      /  F ield         | OpenFOAM: The Open Source CFD Toolbox
   \\    /   O peration     |
Mark Olesen's avatar
Mark Olesen committed
5
    \\  /    A nd           | Copyright (C) 1991-2009 OpenCFD Ltd.
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
     \\/     M anipulation  |
-------------------------------------------------------------------------------
License
    This file is part of OpenFOAM.

    OpenFOAM is free software; you can redistribute it and/or modify it
    under the terms of the GNU General Public License as published by the
    Free Software Foundation; either version 2 of the License, or (at your
    option) any later version.

    OpenFOAM is distributed in the hope that it will be useful, but WITHOUT
    ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
    FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
    for more details.

    You should have received a copy of the GNU General Public License
    along with OpenFOAM; if not, write to the Free Software Foundation,
    Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA

InClass
    vtkPV3Foam

\*---------------------------------------------------------------------------*/

Mark Olesen's avatar
Mark Olesen committed
30
31
#ifndef vtkPV3FoamPatchField_H
#define vtkPV3FoamPatchField_H
32
33
34
35
36

// VTK includes
#include "vtkCellData.h"
#include "vtkFloatArray.h"
#include "vtkMultiBlockDataSet.h"
Mark Olesen's avatar
Mark Olesen committed
37
#include "vtkPointData.h"
38
39
40
41
42
#include "vtkPolyData.h"

// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //

template<class Type>
Mark Olesen's avatar
Mark Olesen committed
43
void Foam::vtkPV3Foam::convertPatchField
44
45
46
47
(
    const word& name,
    const Field<Type>& ptf,
    vtkMultiBlockDataSet* output,
Mark Olesen's avatar
Mark Olesen committed
48
    const partInfo& selector,
49
50
51
    const label datasetNo
)
{
52
53
    const label nComp = pTraits<Type>::nComponents;

Mark Olesen's avatar
Mark Olesen committed
54
55
56
57
58
    vtkFloatArray* cellData = vtkFloatArray::New();
    cellData->SetNumberOfTuples( ptf.size() );
    cellData->SetNumberOfComponents( nComp );
    cellData->Allocate( nComp*ptf.size() );
    cellData->SetName( name.c_str() );
59

60
    float vec[nComp];
61
62
63
    forAll(ptf, i)
    {
        const Type& t = ptf[i];
64
        for (direction d=0; d<nComp; d++)
65
        {
66
            vec[d] = component(t, d);
67
68
69
70
        }
        cellData->InsertTuple(i, vec);
    }

Mark Olesen's avatar
Mark Olesen committed
71
72
73
74
75
76
    vtkPolyData::SafeDownCast
    (
        GetDataSetFromBlock(output, selector, datasetNo)
    )   ->GetCellData()
        ->AddArray(cellData);

77
78
79
80
    cellData->Delete();
}


Mark Olesen's avatar
Mark Olesen committed
81
82
83
84
85
86
87
// as above, but with PointData()
template<class Type>
void Foam::vtkPV3Foam::convertPatchPointField
(
    const word& name,
    const Field<Type>& pptf,
    vtkMultiBlockDataSet* output,
Mark Olesen's avatar
Mark Olesen committed
88
    const partInfo& selector,
Mark Olesen's avatar
Mark Olesen committed
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
    const label datasetNo
)
{
    const label nComp = pTraits<Type>::nComponents;

    vtkFloatArray *pointData = vtkFloatArray::New();
    pointData->SetNumberOfTuples( pptf.size() );
    pointData->SetNumberOfComponents( nComp );
    pointData->Allocate( nComp*pptf.size() );
    pointData->SetName( name.c_str() );

    float vec[nComp];
    forAll(pptf, i)
    {
        const Type& t = pptf[i];
        for (direction d=0; d<nComp; d++)
        {
            vec[d] = component(t, d);
        }

        pointData->InsertTuple(i, vec);
    }

Mark Olesen's avatar
Mark Olesen committed
112
113
114
115
116
117
    vtkPolyData::SafeDownCast
    (
        GetDataSetFromBlock(output, selector, datasetNo)
    )   ->GetPointData()
        ->AddArray(pointData);

Mark Olesen's avatar
Mark Olesen committed
118
119
120
    pointData->Delete();
}

121
122
123
124
125
// * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * //

#endif

// ************************************************************************* //