Nektar++
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Pages
InputFld.cpp
Go to the documentation of this file.
1 ////////////////////////////////////////////////////////////////////////////////
2 //
3 // File: InputFld.cpp
4 //
5 // For more information, please see: http://www.nektar.info/
6 //
7 // The MIT License
8 //
9 // Copyright (c) 2006 Division of Applied Mathematics, Brown University (USA),
10 // Department of Aeronautics, Imperial College London (UK), and Scientific
11 // Computing and Imaging Institute, University of Utah (USA).
12 //
13 // License for the specific language governing rights and limitations under
14 // Permission is hereby granted, free of charge, to any person obtaining a
15 // copy of this software and associated documentation files (the "Software"),
16 // to deal in the Software without restriction, including without limitation
17 // the rights to use, copy, modify, merge, publish, distribute, sublicense,
18 // and/or sell copies of the Software, and to permit persons to whom the
19 // Software is furnished to do so, subject to the following conditions:
20 //
21 // The above copyright notice and this permission notice shall be included
22 // in all copies or substantial portions of the Software.
23 //
24 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
25 // OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
26 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
27 // THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
28 // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
29 // FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
30 // DEALINGS IN THE SOFTWARE.
31 //
32 // Description: Reads a Nektar++ FLD file.
33 //
34 ////////////////////////////////////////////////////////////////////////////////
35 
36 #include <string>
37 #include <iostream>
38 using namespace std;
39 
40 #include "InputFld.h"
41 using namespace Nektar;
42 
43 static std::string npts = LibUtilities::SessionReader::RegisterCmdLineArgument(
44  "NumberOfPoints","n","Define number of points to dump output");
45 
46 namespace Nektar
47 {
48 namespace Utilities
49 {
50 
51 ModuleKey InputFld::m_className[4] = {
53  ModuleKey(eInputModule, "fld"), InputFld::create,
54  "Reads Fld file."),
56  ModuleKey(eInputModule, "chk"), InputFld::create,
57  "Reads checkpoint file."),
59  ModuleKey(eInputModule, "rst"), InputFld::create,
60  "Reads restart file."),
62  ModuleKey(eInputModule, "bse"), InputFld::create,
63  "Reads stability base-flow file.")
64 };
65 
66 
67 /**
68  * @brief Set up InputFld object.
69  *
70  */
71 InputFld::InputFld(FieldSharedPtr f) : InputModule(f)
72 {
73  m_allowedFiles.insert("fld");
74  m_allowedFiles.insert("chk");
75  m_allowedFiles.insert("rst");
76  m_allowedFiles.insert("bse");
77 }
78 
79 
80 /**
81  *
82  */
84 {
85 }
86 
87 
88 /**
89  *
90  */
91 void InputFld::Process(po::variables_map &vm)
92 {
93 
94  if(m_f->m_verbose)
95  {
96  if(m_f->m_comm->GetRank() == 0)
97  {
98  cout << "Processing input fld file" << endl;
99  }
100  }
101 
102  int i,j;
103  string fldending;
104  //Determine appropriate field input
105  if(m_f->m_inputfiles.count("fld") != 0)
106  {
107  fldending = "fld";
108  }
109  else if(m_f->m_inputfiles.count("chk") != 0)
110  {
111  fldending = "chk";
112  }
113  else if (m_f->m_inputfiles.count("rst") != 0)
114  {
115  fldending = "rst";
116  }
117  else if (m_f->m_inputfiles.count("bse") != 0)
118  {
119  fldending = "bse";
120  }
121  else
122  {
123  ASSERTL0(false,"no input file found");
124  }
125 
126  if(!m_f->m_fld)
127  {
128  if(m_f->m_session)
129  {
131  ::AllocateSharedPtr(m_f->m_session->GetComm());
132  }
133  else // serial communicator
134  {
138  }
139  }
140 
141 
142  if(m_f->m_graph) // all for restricted expansion defintion when loading field
143  {
144  // currently load all field (possibly could read data from expansion list
145  // but it is re-arranged in expansion)
146 
147  const SpatialDomains::ExpansionMap &expansions = m_f->m_graph->GetExpansions();
148 
149  // if Range has been speficied it is possible to have a
150  // partition which is empty so check this and return if
151  // no elements present.
152 
153  if(!expansions.size())
154  {
155  return;
156  }
157 
158  m_f->m_exp.resize(1);
159 
160  Array<OneD,int> ElementGIDs(expansions.size());
161  SpatialDomains::ExpansionMap::const_iterator expIt;
162 
163  i = 0;
164  for (expIt = expansions.begin(); expIt != expansions.end(); ++expIt)
165  {
166  ElementGIDs[i++] = expIt->second->m_geomShPtr->GetGlobalID();
167  }
168 
169  m_f->m_fielddef.clear();
170  m_f->m_data.clear();
171 
172  m_f->m_fld->Import(m_f->m_inputfiles[fldending][0],
173  m_f->m_fielddef,
174  m_f->m_data,
175  m_f->m_fieldMetaDataMap,
176  ElementGIDs);
177  }
178  else // load all data.
179  {
180  m_f->m_fld->Import(m_f->m_inputfiles[fldending][0],
181  m_f->m_fielddef,
182  m_f->m_data,
183  m_f->m_fieldMetaDataMap);
184  }
185 
186 
187  // if m_exp defined presume we want to load all field into expansions
188  if(m_f->m_exp.size())
189  {
190  int nfields,nstrips;
191 
192  m_f->m_session->LoadParameter("Strip_Z",nstrips,1);
193 
194  if(vm.count("useSessionVariables"))
195  {
196  nfields = m_f->m_session->GetVariables().size();
197  }
198  else
199  {
200  nfields = m_f->m_fielddef[0]->m_fields.size();
201  }
202 
203 
204  m_f->m_exp.resize(nfields*nstrips);
205 
206  vector<string> vars = m_f->m_session->GetVariables();
207 
208 
209  // declare other fields;
210  for (int s = 0; s < nstrips; ++s) //homogeneous strip varient
211  {
212  for (i = 0; i < nfields; ++i)
213  {
214  if(i < vars.size())
215  {
216  // check to see if field already defined
217  if(!m_f->m_exp[s*nfields+i])
218  {
219  m_f->m_exp[s*nfields+i] = m_f->AppendExpList(
220  m_f->m_fielddef[0]->m_numHomogeneousDir,
221  vars[i]);
222  }
223  }
224  else
225  {
226  if(vars.size())
227  {
228  m_f->m_exp[s*nfields+i] =
229  m_f->AppendExpList(
230  m_f->m_fielddef[0]->m_numHomogeneousDir,
231  vars[0]);
232  }
233  else
234  {
235  m_f->m_exp[s*nfields+i] =
236  m_f->AppendExpList(
237  m_f->m_fielddef[0]->m_numHomogeneousDir);
238  }
239  }
240  }
241  }
242 
243  // Extract data to coeffs and bwd transform
244  for(int s = 0; s < nstrips; ++s) //homogeneous strip varient
245  {
246  for (j = 0; j < nfields; ++j)
247  {
248  for (i = 0; i < m_f->m_data.size()/nstrips; ++i)
249  {
250  m_f->m_exp[s*nfields+j]->
251  ExtractDataToCoeffs(m_f->m_fielddef[i*nstrips+s],
252  m_f->m_data[i*nstrips+s],
253  m_f->m_fielddef[i*nstrips+s]
254  ->m_fields[j],
255  m_f->m_exp[s*nfields+j]->UpdateCoeffs());
256  }
257  m_f->m_exp[s*nfields+j]->BwdTrans(m_f->m_exp[s*nfields+j]->GetCoeffs(),
258  m_f->m_exp[s*nfields+j]->UpdatePhys());
259  }
260  }
261 
262  // reset output field in case Import loaded elements that are not
263  // in the expansion (because of range option of partitioning)
264  std::vector<LibUtilities::FieldDefinitionsSharedPtr> FieldDef
265  = m_f->m_exp[0]->GetFieldDefinitions();
266  std::vector<std::vector<NekDouble> > FieldData(FieldDef.size());
267 
268  for (j = 0; j < nfields; ++j)
269  {
270  for (i = 0; i < FieldDef.size(); ++i)
271  {
272  FieldDef[i]->m_fields.push_back(m_f->m_fielddef[0]->m_fields[j]);
273  m_f->m_exp[j]->AppendFieldData(FieldDef[i], FieldData[i]);
274  }
275  }
276  m_f->m_fielddef = FieldDef;
277  m_f->m_data = FieldData;
278  }
279 }
280 
281 }
282 }
#define ASSERTL0(condition, msg)
Definition: ErrorUtil.hpp:188
pair< ModuleType, string > ModuleKey
tBaseSharedPtr CreateInstance(tKey idKey BOOST_PP_COMMA_IF(MAX_PARAM) BOOST_PP_ENUM_BINARY_PARAMS(MAX_PARAM, tParam, x))
Create an instance of the class referred to by idKey.
Definition: NekFactory.hpp:162
static boost::shared_ptr< DataType > AllocateSharedPtr()
Allocate a shared pointer from the memory pool.
virtual void Process()=0
STL namespace.
CommFactory & GetCommFactory()
Definition: Comm.cpp:64
FieldSharedPtr m_f
Field object.
boost::shared_ptr< Comm > CommSharedPtr
Pointer to a Communicator object.
Definition: Comm.h:53
static std::string npts
Definition: InputFld.cpp:43
boost::shared_ptr< Field > FieldSharedPtr
Definition: Field.hpp:695
std::map< int, ExpansionShPtr > ExpansionMap
Definition: MeshGraph.h:174
ModuleFactory & GetModuleFactory()
Abstract base class for input modules.
tKey RegisterCreatorFunction(tKey idKey, CreatorFunction classCreator, tDescription pDesc="")
Register a class with the factory.
Definition: NekFactory.hpp:215