Nektar++
Public Member Functions | Static Public Member Functions | Static Public Attributes | Protected Member Functions | Private Member Functions | Private Attributes | List of all members
Nektar::FieldUtils::ProcessGrad Class Reference

This processing module calculates the gradient and adds it as an extra-field to the output file. More...

#include <ProcessGrad.h>

Inheritance diagram for Nektar::FieldUtils::ProcessGrad:
[legend]

Public Member Functions

 ProcessGrad (FieldSharedPtr f)
 
virtual ~ProcessGrad ()
 
- Public Member Functions inherited from Nektar::FieldUtils::ProcessModule
 ProcessModule ()
 
 ProcessModule (FieldSharedPtr p_f)
 
- Public Member Functions inherited from Nektar::FieldUtils::Module
FIELD_UTILS_EXPORT Module (FieldSharedPtr p_f)
 
virtual ~Module ()=default
 
void Process (po::variables_map &vm)
 
std::string GetModuleName ()
 
std::string GetModuleDescription ()
 
const ConfigOptionGetConfigOption (const std::string &key) const
 
ModulePriority GetModulePriority ()
 
FIELD_UTILS_EXPORT void RegisterConfig (std::string key, std::string value="")
 Register a configuration option with a module. More...
 
FIELD_UTILS_EXPORT void PrintConfig ()
 Print out all configuration options for a module. More...
 
FIELD_UTILS_EXPORT void SetDefaults ()
 Sets default configuration options for those which have not been set. More...
 
FIELD_UTILS_EXPORT void AddFile (std::string fileType, std::string fileName)
 
FIELD_UTILS_EXPORT void EvaluateTriFieldAtEquiSpacedPts (LocalRegions::ExpansionSharedPtr &exp, const Array< OneD, const NekDouble > &infield, Array< OneD, NekDouble > &outfield)
 

Static Public Member Functions

static std::shared_ptr< Modulecreate (FieldSharedPtr f)
 Creates an instance of this class. More...
 

Static Public Attributes

static ModuleKey className
 

Protected Member Functions

virtual void v_Process (po::variables_map &vm) override
 Write mesh to output file. More...
 
virtual std::string v_GetModuleName () override
 
virtual std::string v_GetModuleDescription () override
 
virtual ModulePriority v_GetModulePriority () override
 
- Protected Member Functions inherited from Nektar::FieldUtils::Module
 Module ()
 

Private Member Functions

void ParserOptions (std::set< int > &variables, std::set< int > &directions)
 
void ProcessCartesianFld (Array< OneD, Array< OneD, NekDouble >> &grad)
 
void ProcessMappingFld (Array< OneD, Array< OneD, NekDouble >> &grad)
 

Private Attributes

std::set< int > m_selectedVars
 
std::set< int > m_directions
 

Additional Inherited Members

- Public Attributes inherited from Nektar::FieldUtils::Module
FieldSharedPtr m_f
 Field object. More...
 
- Protected Attributes inherited from Nektar::FieldUtils::Module
std::map< std::string, ConfigOptionm_config
 List of configuration values. More...
 
std::set< std::string > m_allowedFiles
 List of allowed file formats. More...
 

Detailed Description

This processing module calculates the gradient and adds it as an extra-field to the output file.

Definition at line 49 of file ProcessGrad.h.

Constructor & Destructor Documentation

◆ ProcessGrad()

Nektar::FieldUtils::ProcessGrad::ProcessGrad ( FieldSharedPtr  f)

Definition at line 57 of file ProcessGrad.cpp.

57  : ProcessModule(f)
58 {
59  m_config["vars"] = ConfigOption(false, "NotSet", "Select variables");
60  m_config["dirs"] = ConfigOption(false, "NotSet", "Select directions");
61 }
std::map< std::string, ConfigOption > m_config
List of configuration values.
Definition: Module.h:263

References Nektar::FieldUtils::Module::m_config.

◆ ~ProcessGrad()

Nektar::FieldUtils::ProcessGrad::~ProcessGrad ( )
virtual

Definition at line 63 of file ProcessGrad.cpp.

64 {
65 }

Member Function Documentation

◆ create()

static std::shared_ptr<Module> Nektar::FieldUtils::ProcessGrad::create ( FieldSharedPtr  f)
inlinestatic

Creates an instance of this class.

Definition at line 53 of file ProcessGrad.h.

54  {
56  }
static std::shared_ptr< DataType > AllocateSharedPtr(const Args &...args)
Allocate a shared pointer from the memory pool.

References Nektar::MemoryManager< DataType >::AllocateSharedPtr().

◆ ParserOptions()

void Nektar::FieldUtils::ProcessGrad::ParserOptions ( std::set< int > &  variables,
std::set< int > &  directions 
)
private

Definition at line 67 of file ProcessGrad.cpp.

69 {
70  if (m_config["vars"].as<string>().compare("NotSet"))
71  {
72  ParseUtils::GenerateVariableSet(m_config["vars"].as<string>(),
73  m_f->m_variables, variables);
74  }
75  else
76  {
77  for (int v = 0; v < m_f->m_variables.size(); ++v)
78  {
79  variables.insert(v);
80  }
81  }
82  vector<string> coords = {"x", "y", "z"};
83  int spacedim = m_f->m_numHomogeneousDir + m_f->m_graph->GetMeshDimension();
84  coords.resize(spacedim);
85  if (m_config["dirs"].as<string>().compare("NotSet"))
86  {
87  ParseUtils::GenerateVariableSet(m_config["dirs"].as<string>(), coords,
88  directions);
89  }
90  else
91  {
92  for (int d = 0; d < spacedim; ++d)
93  {
94  directions.insert(d);
95  }
96  }
97 }
FieldSharedPtr m_f
Field object.
Definition: Module.h:234
static bool GenerateVariableSet(const std::string &str, const std::vector< std::string > &variables, std::set< int > &out)
Generate a set of variable locations.
Definition: ParseUtils.cpp:166

References Nektar::ParseUtils::GenerateVariableSet(), Nektar::FieldUtils::Module::m_config, and Nektar::FieldUtils::Module::m_f.

Referenced by v_Process().

◆ ProcessCartesianFld()

void Nektar::FieldUtils::ProcessGrad::ProcessCartesianFld ( Array< OneD, Array< OneD, NekDouble >> &  grad)
private

Definition at line 99 of file ProcessGrad.cpp.

100 {
101  // Calculate Gradient
102  int n = 0;
103  for (int i : m_selectedVars)
104  {
105  for (int j : m_directions)
106  {
107  m_f->m_exp[i]->PhysDeriv(MultiRegions::DirCartesianMap[j],
108  m_f->m_exp[i]->GetPhys(), grad[n]);
109  ++n;
110  }
111  }
112 }
std::set< int > m_selectedVars
Definition: ProcessGrad.h:85
MultiRegions::Direction const DirCartesianMap[]
Definition: ExpList.h:91

References Nektar::MultiRegions::DirCartesianMap, m_directions, Nektar::FieldUtils::Module::m_f, and m_selectedVars.

Referenced by v_Process().

◆ ProcessMappingFld()

void Nektar::FieldUtils::ProcessGrad::ProcessMappingFld ( Array< OneD, Array< OneD, NekDouble >> &  grad)
private

Definition at line 114 of file ProcessGrad.cpp.

115 {
116  int expdim = m_f->m_graph->GetMeshDimension();
117  int spacedim = m_f->m_numHomogeneousDir + expdim;
118  int nfields = m_f->m_variables.size();
119  bool hasvel = m_selectedVars.size() && (*m_selectedVars.begin() < spacedim);
120 
121  int npoints = m_f->m_exp[0]->GetNpoints();
122 
123  Array<OneD, Array<OneD, NekDouble>> tmp(spacedim);
124  for (int i = 0; i < spacedim; i++)
125  {
126  tmp[i] = Array<OneD, NekDouble>(npoints);
127  }
128 
129  // Get mapping
131 
132  // Get velocity and convert to Cartesian system,
133  // if it is still in transformed system
134  Array<OneD, Array<OneD, NekDouble>> vel(spacedim);
135  if (hasvel && m_f->m_fieldMetaDataMap.count("MappingCartesianVel"))
136  {
137  if (m_f->m_fieldMetaDataMap["MappingCartesianVel"] == "False")
138  {
139  // Initialize arrays and copy velocity
140  for (int i = 0; i < spacedim; ++i)
141  {
142  vel[i] = Array<OneD, NekDouble>(npoints);
143  if (m_f->m_exp[0]->GetWaveSpace())
144  {
145  m_f->m_exp[0]->HomogeneousBwdTrans(
146  npoints, m_f->m_exp[i]->GetPhys(), vel[i]);
147  }
148  else
149  {
150  Vmath::Vcopy(npoints, m_f->m_exp[i]->GetPhys(), 1, vel[i],
151  1);
152  }
153  }
154  // Convert velocity to cartesian system
155  mapping->ContravarToCartesian(vel, vel);
156  // Convert back to wavespace if necessary
157  if (m_f->m_exp[0]->GetWaveSpace())
158  {
159  for (int i = 0; i < spacedim; ++i)
160  {
161  m_f->m_exp[0]->HomogeneousFwdTrans(npoints, vel[i], vel[i]);
162  }
163  }
164  }
165  else
166  {
167  for (int i = 0; i < spacedim; ++i)
168  {
169  vel[i] = Array<OneD, NekDouble>(npoints);
170  Vmath::Vcopy(npoints, m_f->m_exp[i]->GetPhys(), 1, vel[i], 1);
171  }
172  }
173  }
174  else if (hasvel)
175  {
176  for (int i = 0; i < spacedim && i < nfields; ++i)
177  {
178  vel[i] = Array<OneD, NekDouble>(npoints);
179  Vmath::Vcopy(npoints, m_f->m_exp[i]->GetPhys(), 1, vel[i], 1);
180  }
181  }
182 
183  // Calculate Gradient
184  int n = 0;
185  for (int i : m_selectedVars)
186  {
187  for (int j = 0; j < spacedim; ++j)
188  {
189  if (i < spacedim)
190  {
191  m_f->m_exp[i]->PhysDeriv(MultiRegions::DirCartesianMap[j],
192  vel[i], tmp[j]);
193  }
194  else
195  {
196  m_f->m_exp[i]->PhysDeriv(MultiRegions::DirCartesianMap[j],
197  m_f->m_exp[i]->GetPhys(), tmp[j]);
198  }
199  }
200  mapping->CovarToCartesian(tmp, tmp);
201  for (int j : m_directions)
202  {
203  Vmath::Vcopy(npoints, tmp[j], 1, grad[n], 1);
204  ++n;
205  }
206  }
207 }
static GlobalMapping::MappingSharedPtr GetMapping(FieldSharedPtr f)
GLOBAL_MAPPING_EXPORT typedef std::shared_ptr< Mapping > MappingSharedPtr
A shared pointer to a Mapping object.
Definition: Mapping.h:50
void Vcopy(int n, const T *x, const int incx, T *y, const int incy)
Definition: Vmath.cpp:1255

References Nektar::MultiRegions::DirCartesianMap, Nektar::FieldUtils::ProcessMapping::GetMapping(), m_directions, Nektar::FieldUtils::Module::m_f, m_selectedVars, Nektar::GlobalMapping::MappingSharedPtr, and Vmath::Vcopy().

Referenced by v_Process().

◆ v_GetModuleDescription()

virtual std::string Nektar::FieldUtils::ProcessGrad::v_GetModuleDescription ( )
inlineoverrideprotectedvirtual

Reimplemented from Nektar::FieldUtils::Module.

Definition at line 71 of file ProcessGrad.h.

72  {
73  return "Calculating gradients";
74  }

◆ v_GetModuleName()

virtual std::string Nektar::FieldUtils::ProcessGrad::v_GetModuleName ( )
inlineoverrideprotectedvirtual

Reimplemented from Nektar::FieldUtils::Module.

Definition at line 66 of file ProcessGrad.h.

67  {
68  return "ProcessGrad";
69  }

◆ v_GetModulePriority()

virtual ModulePriority Nektar::FieldUtils::ProcessGrad::v_GetModulePriority ( )
inlineoverrideprotectedvirtual

Reimplemented from Nektar::FieldUtils::Module.

Definition at line 76 of file ProcessGrad.h.

77  {
78  return eModifyExp;
79  }

References Nektar::FieldUtils::eModifyExp.

◆ v_Process()

void Nektar::FieldUtils::ProcessGrad::v_Process ( po::variables_map &  vm)
overrideprotectedvirtual

Write mesh to output file.

Reimplemented from Nektar::FieldUtils::Module.

Definition at line 209 of file ProcessGrad.cpp.

210 {
211  m_f->SetUpExp(vm);
213 
214  int nfields = m_f->m_variables.size();
215  int addfields = m_selectedVars.size() * m_directions.size();
216  m_f->m_exp.resize(nfields + addfields);
217 
218  vector<string> coords = {"x", "y", "z"};
219  for (int i : m_selectedVars)
220  {
221  for (int j : m_directions)
222  {
223  m_f->m_variables.push_back(m_f->m_variables[i] + "_" + coords[j]);
224  }
225  }
226 
227  // Skip in case of empty partition
228  if (m_f->m_exp[0]->GetNumElmts() == 0)
229  {
230  return;
231  }
232 
233  int npoints = m_f->m_exp[0]->GetNpoints();
234  Array<OneD, Array<OneD, NekDouble>> grad(addfields);
235  for (int i = 0; i < addfields; ++i)
236  {
237  grad[i] = Array<OneD, NekDouble>(npoints);
238  }
239 
240  if (m_f->m_fieldMetaDataMap.count("MappingType"))
241  {
242  ProcessMappingFld(grad);
243  }
244  else
245  {
246  ProcessCartesianFld(grad);
247  }
248 
249  for (int i = 0; i < addfields; ++i)
250  {
251  m_f->m_exp[nfields + i] = m_f->AppendExpList(m_f->m_numHomogeneousDir);
252 
253  Vmath::Vcopy(npoints, grad[i], 1, m_f->m_exp[nfields + i]->UpdatePhys(),
254  1);
255  m_f->m_exp[nfields + i]->FwdTransLocalElmt(
256  grad[i], m_f->m_exp[nfields + i]->UpdateCoeffs());
257  }
258 }
void ProcessCartesianFld(Array< OneD, Array< OneD, NekDouble >> &grad)
Definition: ProcessGrad.cpp:99
void ProcessMappingFld(Array< OneD, Array< OneD, NekDouble >> &grad)
void ParserOptions(std::set< int > &variables, std::set< int > &directions)
Definition: ProcessGrad.cpp:67

References m_directions, Nektar::FieldUtils::Module::m_f, m_selectedVars, ParserOptions(), ProcessCartesianFld(), ProcessMappingFld(), and Vmath::Vcopy().

Member Data Documentation

◆ className

ModuleKey Nektar::FieldUtils::ProcessGrad::className
static
Initial value:
"Computes gradient of fields.")
static std::shared_ptr< Module > create(FieldSharedPtr f)
Creates an instance of this class.
Definition: ProcessGrad.h:53
tKey RegisterCreatorFunction(tKey idKey, CreatorFunction classCreator, std::string pDesc="")
Register a class with the factory.
Definition: NekFactory.hpp:198
std::pair< ModuleType, std::string > ModuleKey
Definition: Module.h:317
ModuleFactory & GetModuleFactory()
Definition: Module.cpp:49

Definition at line 57 of file ProcessGrad.h.

◆ m_directions

std::set<int> Nektar::FieldUtils::ProcessGrad::m_directions
private

Definition at line 86 of file ProcessGrad.h.

Referenced by ProcessCartesianFld(), ProcessMappingFld(), and v_Process().

◆ m_selectedVars

std::set<int> Nektar::FieldUtils::ProcessGrad::m_selectedVars
private

Definition at line 85 of file ProcessGrad.h.

Referenced by ProcessCartesianFld(), ProcessMappingFld(), and v_Process().