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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
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
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
/***************************************************************************
tnlFunctionEnumerator.h - description
-------------------
begin : Mar 5, 2015
copyright : (C) 2015 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 SRC_FUNCTIONS_TNLFUNCTIONENUMERATOR_H_
#define SRC_FUNCTIONS_TNLFUNCTIONENUMERATOR_H_
#include <functions/tnlFunctionAdapter.h>
template< typename Function,
typename DofVector >
class tnlFunctionEnumeratorTraversalUserData
{
public:
typedef DofVector::RealType RealType;
const RealType *time;
const Function* function;
DofVector *u;
const RealType* functionCoefficient;
const RealType* dofVectorCoefficient;
tnlFunctionEnumeratorTraversalUserData( const RealType& time,
const Function& function,
DofVector& u,
const RealType& functionCoefficient,
const RealType& dofVectorCoefficient )
: time( &time ),
function( &function ),
u( &u ),
functionCoefficient( &functionCoefficient ),
dofVectorCoefficient( &dofVectorCoefficient )
{};
};
template< typename Mesh,
typename Function,
typename DofVector >
class tnlFunctionEnumerator
{
public:
typedef Mesh MeshType;
typedef typename DofVector::RealType RealType;
typedef typename DofVector::DeviceType DeviceType;
typedef typename DofVector::IndexType IndexType;
typedef tnlFunctionEnumeratorTraversalUserData< Function,
DofVector > TraversalUserData;
template< int EntityDimensions >
void enumerate( const MeshType& mesh,
const Function& function
DofVector& u,
const RealType& functionCoefficient = 1.0,
const RealType& dofVectorCoefficient = 0.0,
const RealType& time = 0.0 ) const;
class TraversalEntitiesProcessor
{
public:
template< int EntityDimensions >
#ifdef HAVE_CUDA
__host__ __device__
#endif
static void processEntity( const MeshType& mesh,
TraversalUserData& userData,
const IndexType index )
{
typedef tnlFunctionAdapter< MeshType, Function > FunctionAdapter;
( *userData.u )[ index ] =
( *userData.dofVectorCoefficient ) * ( *userData.u )[ index ] +
( *userData.functionCoefficient ) * FunctionAdapter::getValue( mesh,
*userData.function,
index,
*userData.time );
}
};
};
template< int Dimensions,
typename Real,
typename Device,
typename Index,
typename Function,
typename DofVector >
class tnlFunctionEnumerator< tnlGrid< Dimensions, Real, Device, Index >,
Function,
DofVector >
{
public:
typedef tnlGrid< Dimensions, Real, Device, Index > MeshType;
typedef typename MeshType::RealType RealType;
typedef typename MeshType::DeviceType DeviceType;
typedef typename MeshType::IndexType IndexType;
typedef typename MeshType::CoordinatesType CoordinatesType;
typedef tnlFunctionEnumeratorTraversalUserData< Function,
DofVector > TraversalUserData;
template< int EntityDimensions >
void enumerate( const MeshType& mesh,
const Function& function,
DofVector& u,
const RealType& time = 0.0 ) const;
class TraversalEntitiesProcessor
{
public:
typedef typename MeshType::VertexType VertexType;
#ifdef HAVE_CUDA
__host__ __device__
#endif
static void processCell( const MeshType& mesh,
TraversalUserData& userData,
const IndexType index,
const CoordinatesType& coordinates )
{
typedef tnlFunctionAdapter< MeshType, Function > FunctionAdapter;
( *userData.u )[ index ] =
( *userData.dofVectorCoefficient ) * ( *userData.u )[ index ] +
( *userData.functionCoefficient ) * FunctionAdapter::getValue( mesh,
*userData.function,
index,
coordinates,
*userData.time );
}
#ifdef HAVE_CUDA
__host__ __device__
#endif
static void processFace( const MeshType& mesh,
TraversalUserData& userData,
const IndexType index,
const CoordinatesType& coordinates )
{
typedef tnlFunctionAdapter< MeshType, Function > FunctionAdapter;
( *userData.u )[ index ] =
( *userData.dofVectorCoefficient ) * ( *userData.u )[ index ] +
( *userData.functionCoefficient ) * FunctionAdapter::getValue( mesh,
*userData.function,
index,
coordinates,
*userData.time );
}
};
};
#include <functions/tnlFunctionEnumerator_impl.h>
#endif /* SRC_FUNCTIONS_TNLFUNCTIONENUMERATOR_H_ */