Nektar++
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
Functions
SplitModes.cpp File Reference
#include <cstdio>
#include <cstdlib>
#include <StdRegions/StdTriExp.h>
#include <SpatialDomains/MeshGraph.h>
Include dependency graph for SplitModes.cpp:

Go to the source code of this file.

Functions

int main (int argc, char *argv[])
 

Function Documentation

int main ( int  argc,
char *  argv[] 
)

Definition at line 10 of file SplitModes.cpp.

References ASSERTL0, Nektar::LibUtilities::eQuadrilateral, Nektar::LibUtilities::eTriangle, Nektar::LibUtilities::StdSegData::getNumberOfCoefficients(), Nektar::LibUtilities::Import(), Nektar::iterator, and Nektar::LibUtilities::Write().

11 {
12  if(argc != 2)
13  {
14  fprintf(stderr,"Usage: Splitmodes fieldfile \n");
15  exit(1);
16  }
17 
18  //default meshgraph
20 
21 
22  //----------------------------------------------
23  // Import fieldfile2.
24  string fieldfile(argv[argc-1]);
25  vector<LibUtilities::FieldDefinitionsSharedPtr> fielddef;
26  vector<vector<NekDouble> > fielddata;
27  LibUtilities::Import(fieldfile,fielddef,fielddata);
28  //----------------------------------------------
29 
30 
31  ASSERTL0(fielddef[0]->m_numModes[2] > 1,"Expected Fourier field to have at least 2 modes");
32 
33  ASSERTL0(fielddef[0]->m_numHomogeneousDir == 1,"Expected second fld to have one homogeneous direction");
34 
35  int nmodes = fielddef[0]->m_numModes[2];
36 
37  // take off modes from fielddef
38  vector<unsigned int> newNumModes;
39  newNumModes.push_back(fielddef[0]->m_numModes[0]);
40  newNumModes.push_back(fielddef[0]->m_numModes[1]);
41  vector<LibUtilities::BasisType> newBasis;
42  newBasis.push_back(fielddef[0]->m_basis[0]);
43  newBasis.push_back(fielddef[0]->m_basis[1]);
44  for(int i = 0; i < fielddata.size(); ++i)
45  {
46  fielddef[i]->m_numModes = newNumModes;
47  fielddef[i]->m_basis = newBasis;
48  fielddef[i]->m_numHomogeneousDir = 0;
49  }
50 
51 
52  for(int m = 0; m < nmodes; ++m)
53  {
54  vector<vector<NekDouble> > writedata;
55 
56  // outfile
57  string outfile(argv[argc-1]);
58  string out = outfile.substr(0, outfile.find_last_of("."));
59  char num[16] ="";
60  sprintf(num,"%d",(m/2));
61 
62  if(m%2 == 0)
63  {
64  out = out + "_mode_" + num + "_real.fld";
65  }
66  else
67  {
68  out = out + "_mode_" + num + "_imag.fld";
69  }
70 
71  for(int i = 0; i < fielddata.size(); ++i)
72  {
73  vector<NekDouble> newdata;
74 
75  // Determine the number of coefficients per element
76  int ncoeffs;
77  switch(fielddef[i]->m_shapeType)
78  {
80  ncoeffs = LibUtilities::StdTriData::getNumberOfCoefficients(fielddef[i]->m_numModes[0], fielddef[i]->m_numModes[1]);
81  break;
83  ncoeffs = fielddef[i]->m_numModes[0]*fielddef[i]->m_numModes[1];
84  break;
85  default:
86  ASSERTL0(false,"Shape not recognised");
87  break;
88  }
89 
91  vec_iter = fielddata[i].begin();
92 
93  for(int k = 0; k < fielddef[i]->m_fields.size(); ++k)
94  {
95  for(int n = 0; n < fielddef[i]->m_elementIDs.size(); ++n)
96  {
97  // Put orginal mode in here.
98  vec_iter += m*ncoeffs;
99  newdata.insert(newdata.end(),vec_iter,vec_iter+ncoeffs);
100  vec_iter += (nmodes-m)*ncoeffs;
101  }
102  }
103  writedata.push_back(newdata);
104  }
105 
106  //-----------------------------------------------
107  // Write out datafile.
108  LibUtilities::Write(out.c_str(), fielddef, writedata);
109  //-----------------------------------------------
110 
111  }
112  //----------------------------------------------
113 
114 
115  return 0;
116 }
#define ASSERTL0(condition, msg)
Definition: ErrorUtil.hpp:198
void Import(const std::string &infilename, std::vector< FieldDefinitionsSharedPtr > &fielddefs, std::vector< std::vector< NekDouble > > &fielddata, FieldMetaDataMap &fieldinfomap, const Array< OneD, int > &ElementIDs)
This function allows for data to be imported from an FLD file when a session and/or communicator is n...
Definition: FieldIO.cpp:279
void Write(const std::string &outFile, std::vector< FieldDefinitionsSharedPtr > &fielddefs, std::vector< std::vector< NekDouble > > &fielddata, const FieldMetaDataMap &fieldinfomap, const bool backup)
This function allows for data to be written to an FLD file when a session and/or communicator is not ...
Definition: FieldIO.cpp:235
StandardMatrixTag boost::call_traits< LhsDataType >::const_reference rhs typedef NekMatrix< LhsDataType, StandardMatrixTag >::iterator iterator
Base class for a spectral/hp element mesh.
Definition: MeshGraph.h:186