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  Timer timer;
94 
95  if(m_f->m_verbose)
96  {
97  if(m_f->m_comm->GetRank() == 0)
98  {
99  cout << "Processing input fld file" << endl;
100  timer.Start();
101  }
102  }
103 
104  int i,j;
105  string fldending;
106  //Determine appropriate field input
107  if(m_f->m_inputfiles.count("fld") != 0)
108  {
109  fldending = "fld";
110  }
111  else if(m_f->m_inputfiles.count("chk") != 0)
112  {
113  fldending = "chk";
114  }
115  else if (m_f->m_inputfiles.count("rst") != 0)
116  {
117  fldending = "rst";
118  }
119  else if (m_f->m_inputfiles.count("bse") != 0)
120  {
121  fldending = "bse";
122  }
123  else
124  {
125  ASSERTL0(false,"no input file found");
126  }
127 
128  if(!m_f->m_fld)
129  {
130  if(m_f->m_session)
131  {
133  ::AllocateSharedPtr(m_f->m_session->GetComm());
134  }
135  else // serial communicator
136  {
140  }
141  }
142 
143 
144  if(m_f->m_graph) // all for restricted expansion defintion when loading field
145  {
146  // currently load all field (possibly could read data from expansion list
147  // but it is re-arranged in expansion)
148 
149  const SpatialDomains::ExpansionMap &expansions = m_f->m_graph->GetExpansions();
150 
151  // if Range has been speficied it is possible to have a
152  // partition which is empty so check this and return if
153  // no elements present.
154 
155  if(!expansions.size())
156  {
157  return;
158  }
159 
160  m_f->m_exp.resize(1);
161 
162  Array<OneD,int> ElementGIDs(expansions.size());
163  SpatialDomains::ExpansionMap::const_iterator expIt;
164 
165  i = 0;
166  for (expIt = expansions.begin(); expIt != expansions.end(); ++expIt)
167  {
168  ElementGIDs[i++] = expIt->second->m_geomShPtr->GetGlobalID();
169  }
170 
171  m_f->m_fielddef.clear();
172  m_f->m_data.clear();
173 
174  m_f->m_fld->Import(m_f->m_inputfiles[fldending][0],
175  m_f->m_fielddef,
176  m_f->m_data,
177  m_f->m_fieldMetaDataMap,
178  ElementGIDs);
179  }
180  else // load all data.
181  {
182  m_f->m_fld->Import(m_f->m_inputfiles[fldending][0],
183  m_f->m_fielddef,
184  m_f->m_data,
185  m_f->m_fieldMetaDataMap);
186  }
187 
188 
189  // if m_exp defined presume we want to load all field into expansions
190  if(m_f->m_exp.size())
191  {
192  int nfields,nstrips;
193 
194  m_f->m_session->LoadParameter("Strip_Z",nstrips,1);
195 
196  if(vm.count("useSessionVariables"))
197  {
198  nfields = m_f->m_session->GetVariables().size();
199  }
200  else
201  {
202  nfields = m_f->m_fielddef[0]->m_fields.size();
203  }
204 
205 
206  m_f->m_exp.resize(nfields*nstrips);
207 
208  vector<string> vars = m_f->m_session->GetVariables();
209 
210 
211  // declare other fields;
212  for (int s = 0; s < nstrips; ++s) //homogeneous strip varient
213  {
214  for (i = 0; i < nfields; ++i)
215  {
216  if(i < vars.size())
217  {
218  // check to see if field already defined
219  if(!m_f->m_exp[s*nfields+i])
220  {
221  m_f->m_exp[s*nfields+i] = m_f->AppendExpList(
222  m_f->m_fielddef[0]->m_numHomogeneousDir,
223  vars[i]);
224  }
225  }
226  else
227  {
228  if(vars.size())
229  {
230  m_f->m_exp[s*nfields+i] =
231  m_f->AppendExpList(
232  m_f->m_fielddef[0]->m_numHomogeneousDir,
233  vars[0]);
234  }
235  else
236  {
237  m_f->m_exp[s*nfields+i] =
238  m_f->AppendExpList(
239  m_f->m_fielddef[0]->m_numHomogeneousDir);
240  }
241  }
242  }
243  }
244 
245  // Extract data to coeffs and bwd transform
246  for(int s = 0; s < nstrips; ++s) //homogeneous strip varient
247  {
248  for (j = 0; j < nfields; ++j)
249  {
250  for (i = 0; i < m_f->m_data.size()/nstrips; ++i)
251  {
252  m_f->m_exp[s*nfields+j]->
253  ExtractDataToCoeffs(m_f->m_fielddef[i*nstrips+s],
254  m_f->m_data[i*nstrips+s],
255  m_f->m_fielddef[i*nstrips+s]
256  ->m_fields[j],
257  m_f->m_exp[s*nfields+j]->UpdateCoeffs());
258  }
259  m_f->m_exp[s*nfields+j]->BwdTrans(m_f->m_exp[s*nfields+j]->GetCoeffs(),
260  m_f->m_exp[s*nfields+j]->UpdatePhys());
261  }
262  }
263 
264  // reset output field in case Import loaded elements that are not
265  // in the expansion (because of range option of partitioning)
266  std::vector<LibUtilities::FieldDefinitionsSharedPtr> FieldDef
267  = m_f->m_exp[0]->GetFieldDefinitions();
268  std::vector<std::vector<NekDouble> > FieldData(FieldDef.size());
269 
270  for (j = 0; j < nfields; ++j)
271  {
272  for (i = 0; i < FieldDef.size(); ++i)
273  {
274  FieldDef[i]->m_fields.push_back(m_f->m_fielddef[0]->m_fields[j]);
275  m_f->m_exp[j]->AppendFieldData(FieldDef[i], FieldData[i]);
276  }
277  }
278  m_f->m_fielddef = FieldDef;
279  m_f->m_data = FieldData;
280  }
281 
282  if(m_f->m_verbose)
283  {
284  if(m_f->m_comm->GetRank() == 0)
285  {
286  timer.Stop();
287  NekDouble cpuTime = timer.TimePerTest(1);
288 
289  stringstream ss;
290  ss << cpuTime << "s";
291  cout << "InputFld CPU Time: " << setw(8) << left
292  << ss.str() << endl;
293  cpuTime = 0.0;
294  }
295 
296  }
297 }
298 
299 }
300 }
#define ASSERTL0(condition, msg)
Definition: ErrorUtil.hpp:161
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.
void Stop()
Definition: Timer.cpp:62
boost::shared_ptr< Comm > CommSharedPtr
Pointer to a Communicator object.
Definition: Comm.h:53
static std::string npts
Definition: InputFld.cpp:43
double NekDouble
boost::shared_ptr< Field > FieldSharedPtr
Definition: Field.hpp:695
void Start()
Definition: Timer.cpp:51
NekDouble TimePerTest(unsigned int n)
Returns amount of seconds per iteration in a test with n iterations.
Definition: Timer.cpp:108
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