Newer
Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
/***************************************************************************
tnlFunctionAdapter.h - description
-------------------
begin : Nov 28, 2014
copyright : (C) 2014 by Tomas Oberhuber
email : tomas.oberhuber@fjfi.cvut.cz
***************************************************************************/
/***************************************************************************
* *
* This program 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. *
* *
***************************************************************************/
#ifndef TNLFUNCTIONADAPTER_H_
#define TNLFUNCTIONADAPTER_H_
#include <functions/tnlConstantFunction.h>
template< typename Mesh,
typename Function >
class tnlFunctionAdapter
{
public:
typedef Mesh MeshType;
typedef Function FunctionType;
typedef typename FunctionType::RealType RealType;
typedef typename MeshType::IndexType IndexType;
template< int MeshEntityDimension >
#ifdef HAVE_CUDA
__device__ __host__
#endif
static RealType getValue( const MeshType& mesh,
const FunctionType& function,
const IndexType index,
const RealType& time = 0.0 )
{
return function.getValue( mesh.template getEntityCenter< MeshEntityDimension >,
time );
}
};
/****
* Specialization for constant function
* - it does not ask the mesh for the mesh entity center
*/
template< typename Mesh,
int FunctionDimensions,
typename Real >
class tnlFunctionAdapter< Mesh, tnlConstantFunction< FunctionDimensions, Real > >
{
public:
typedef Mesh MeshType;
typedef tnlConstantFunction< FunctionDimensions, Real > FunctionType;
typedef typename FunctionType::RealType RealType;
typedef typename MeshType::IndexType IndexType;
typedef typename MeshType::VertexType VertexType;
template< int MeshEntityDimension >
#ifdef HAVE_CUDA
__device__ __host__
#endif
static RealType getValue( const MeshType& mesh,
const FunctionType& function,
const IndexType index,
const RealType& time = 0.0 )
{
VertexType v;
return function.getValue( v, time );
}
};
/****
* Specialization for grids
* - it takes grid coordinates for faster entity center evaluation
*/
template< int Dimensions,
typename Real,
typename Device,
typename Index,
typename Function >
class tnlFunctionAdapter< tnlGrid< Dimensions, Real, Device, Index >, Function >
{
public:
typedef tnlGrid< Dimensions, Real, Device, Index > MeshType;
typedef Function FunctionType;
typedef Real RealType;
typedef Index IndexType;
typedef typename MeshType::VertexType VertexType;
typedef typename MeshType::CoordinatesType CoordinatesType;
#ifdef HAVE_CUDA
__device__ __host__
#endif
static RealType getValue( const MeshType& mesh,
const FunctionType& function,
const IndexType index,
const CoordinatesType& coordinates,
const RealType& time = 0.0 )
return function.getValue( mesh.template getCellCenter< VertexType >( coordinates ),
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
time );
}
};
/****
* Specialization for grids and constant function
* - it takes grid coordinates for faster entity center evaluation
*/
template< int Dimensions,
typename Real,
typename Device,
typename Index >
class tnlFunctionAdapter< tnlGrid< Dimensions, Real, Device, Index >,
tnlConstantFunction< Dimensions, Real > >
{
public:
typedef tnlGrid< Dimensions, Real, Device, Index > MeshType;
typedef tnlConstantFunction< Dimensions, Real > FunctionType;
typedef Real RealType;
typedef Index IndexType;
typedef typename MeshType::VertexType VertexType;
typedef typename MeshType::CoordinatesType CoordinatesType;
#ifdef HAVE_CUDA
__device__ __host__
#endif
static RealType getValue( const MeshType& mesh,
const FunctionType& function,
const IndexType index,
const CoordinatesType& coordinates,
const RealType& time = 0.0 )
{
VertexType v;
return function.getValue( v, time );
}
};
/****
* Specialization for mesh functions
*/
template< typename Mesh,
typename Real,
typename Device,
typename Index >
class tnlFunctionAdapter< Mesh, tnlVector< Real, Device, Index > >
{
public:
typedef Mesh MeshType;
typedef tnlVector< Real, Device, Index > FunctionType;
typedef Real RealType;
typedef Index IndexType;
#ifdef HAVE_CUDA
__device__ __host__
#endif
static RealType getValue( const MeshType& mesh,
const FunctionType& function,
const IndexType index,
const RealType& time = 0.0 )
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
{
return function[ index ];
}
};
/****
* Specialization for mesh functions
*/
template< typename Mesh,
typename Real,
typename Device,
typename Index >
class tnlFunctionAdapter< Mesh, tnlSharedVector< Real, Device, Index > >
: public tnlFunctionAdapter< Mesh, tnlVector< Real, Device, Index > >
{
public:
typedef tnlSharedVector< Real, Device, Index > FunctionType;
};
/****
* Specialization for mesh functions and grids
*/
template< int Dimensions,
typename Real,
typename Device,
typename Index >
class tnlFunctionAdapter< tnlGrid< Dimensions, Real, Device, Index >,
tnlVector< Real, Device, Index > >
{
public:
typedef tnlGrid< Dimensions, Real, Device, Index > MeshType;
typedef tnlVector< Real, Device, Index > FunctionType;
typedef Real RealType;
typedef Index IndexType;
typedef typename MeshType::CoordinatesType CoordinatesType;
#ifdef HAVE_CUDA
__device__ __host__
#endif
static RealType getValue( const MeshType& mesh,
const FunctionType& function,
const IndexType index,
const CoordinatesType& coordinates,
const RealType& time = 0.0 )
{
return function[ index ];
}
};
/****
* Specialization for mesh functions and grids
*/
template< int Dimensions,
typename Real,
typename Device,
typename Index >
class tnlFunctionAdapter< tnlGrid< Dimensions, Real, Device, Index >,
tnlSharedVector< Real, Device, Index > >
: public tnlFunctionAdapter< tnlGrid< Dimensions, Real, Device, Index >,
tnlVector< Real, Device, Index > >
{
public:
typedef tnlSharedVector< Real, Device, Index > FunctionType;
};
#endif /* TNLFUNCTIONADAPTER_H_ */