Nektar++
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Pages
CalcDivergence.cpp
Go to the documentation of this file.
1 /**
2  * This function calculate the divergence starting from an .fld file.
3  * It is meant to be used with solutions produced by the
4  * incompressible Navier-Stokes solver. To use it with solutions
5  * coming form another solver further generalisations are required.
6  */
7 #include <cstdio>
8 #include <cstdlib>
9 
10 #include <MultiRegions/ExpList.h>
11 #include <MultiRegions/ExpList1D.h>
12 #include <MultiRegions/ExpList2D.h>
13 #include <MultiRegions/ExpList3D.h>
14 #include <MultiRegions/ExpList2DHomogeneous1D.h>
15 #include <MultiRegions/ExpList3DHomogeneous1D.h>
16 #include <MultiRegions/ExpList3DHomogeneous2D.h>
17 
18 using namespace Nektar;
19 
20 int main(int argc, char *argv[])
21 {
22  int i,j;
23 
24  if(argc != 3)
25  {
26  fprintf(stderr,"Usage: ./CalcDivergence file.xml file.fld\n");
27  exit(1);
28  }
29 
30  LibUtilities::SessionReaderSharedPtr vSession
31  = LibUtilities::SessionReader::CreateInstance(argc, argv);
32 
33 
34  //----------------------------------------------
35  // Read in mesh from input file
36  string meshfile(argv[argc-2]);
37  SpatialDomains::MeshGraphSharedPtr graphShPt = SpatialDomains::MeshGraph::Read(vSession);//meshfile);
38  //----------------------------------------------
39 
40  //----------------------------------------------
41  // Import field file.
42  string fieldfile(argv[argc-1]);
43  vector<LibUtilities::FieldDefinitionsSharedPtr> fielddef;
44  vector<vector<NekDouble> > fielddata;
45  LibUtilities::Import(fieldfile,fielddef,fielddata);
46  bool useFFT = false;
47  bool dealiasing = false;
48  //----------------------------------------------
49 
50  //----------------------------------------------
51  // Define Expansion
52  int expdim = graphShPt->GetMeshDimension();
53  int nfields = fielddef[0]->m_fields.size();
54 
55  Array<OneD, MultiRegions::ExpListSharedPtr> Exp(nfields + 1);
56 
57  switch(expdim)
58  {
59  case 1:
60  {
61  ASSERTL0(fielddef[0]->m_numHomogeneousDir <= 2,"Quasi-3D approach is only set up for 1 or 2 homogeneous directions");
62 
63  if(fielddef[0]->m_numHomogeneousDir == 1)
64  {
65  MultiRegions::ExpList2DHomogeneous1DSharedPtr Exp2DH1;
66 
67  // Define Homogeneous expansion
68  //int nplanes = fielddef[0]->m_numModes[1];
69 
70  int nplanes;
71  vSession->LoadParameter("HomModesZ",nplanes,fielddef[0]->m_numModes[1]);
72 
73  // choose points to be at evenly spaced points at
74  const LibUtilities::PointsKey Pkey(nplanes+1,LibUtilities::ePolyEvenlySpaced);
75  const LibUtilities::BasisKey Bkey(fielddef[0]->m_basis[1],nplanes,Pkey);
76  NekDouble ly = fielddef[0]->m_homogeneousLengths[0];
77 
78  Exp2DH1 = MemoryManager<MultiRegions::ExpList2DHomogeneous1D>::AllocateSharedPtr(vSession,Bkey,ly,useFFT,dealiasing,graphShPt);
79  Exp[0] = Exp2DH1;
80 
81  for(i = 1; i < nfields + 1; ++i)
82  {
83  Exp[i] = MemoryManager<MultiRegions::ExpList2DHomogeneous1D>::AllocateSharedPtr(*Exp2DH1);
84  }
85  }
86  else if(fielddef[0]->m_numHomogeneousDir == 2)
87  {
88  MultiRegions::ExpList3DHomogeneous2DSharedPtr Exp3DH2;
89 
90  // Define Homogeneous expansion
91  //int nylines = fielddef[0]->m_numModes[1];
92  //int nzlines = fielddef[0]->m_numModes[2];
93 
94  int nylines;
95  int nzlines;
96  vSession->LoadParameter("HomModesY",nylines,fielddef[0]->m_numModes[1]);
97  vSession->LoadParameter("HomModesZ",nzlines,fielddef[0]->m_numModes[2]);
98 
99  // choose points to be at evenly spaced points at
100  const LibUtilities::PointsKey PkeyY(nylines+1,LibUtilities::ePolyEvenlySpaced);
101  const LibUtilities::BasisKey BkeyY(fielddef[0]->m_basis[1],nylines,PkeyY);
102 
103  const LibUtilities::PointsKey PkeyZ(nzlines+1,LibUtilities::ePolyEvenlySpaced);
104  const LibUtilities::BasisKey BkeyZ(fielddef[0]->m_basis[2],nzlines,PkeyZ);
105 
106  NekDouble ly = fielddef[0]->m_homogeneousLengths[0];
107  NekDouble lz = fielddef[0]->m_homogeneousLengths[1];
108 
109  Exp3DH2 = MemoryManager<MultiRegions::ExpList3DHomogeneous2D>::AllocateSharedPtr(vSession,BkeyY,BkeyZ,ly,lz,useFFT,dealiasing,graphShPt);
110  Exp[0] = Exp3DH2;
111 
112  for(i = 1; i < nfields + 1; ++i)
113  {
114  Exp[i] = MemoryManager<MultiRegions::ExpList3DHomogeneous2D>::AllocateSharedPtr(*Exp3DH2);
115  }
116  }
117  else
118  {
119  MultiRegions::ExpList1DSharedPtr Exp1D;
120  Exp1D = MemoryManager<MultiRegions::ExpList1D>
121  ::AllocateSharedPtr(vSession,graphShPt);
122  Exp[0] = Exp1D;
123  for(i = 1; i < nfields + 1; ++i)
124  {
125  Exp[i] = MemoryManager<MultiRegions::ExpList1D>
126  ::AllocateSharedPtr(*Exp1D);
127  }
128  }
129  }
130  break;
131  case 2:
132  {
133  ASSERTL0(fielddef[0]->m_numHomogeneousDir <= 1,"NumHomogeneousDir is only set up for 1");
134 
135  if(fielddef[0]->m_numHomogeneousDir == 1)
136  {
137  MultiRegions::ExpList3DHomogeneous1DSharedPtr Exp3DH1;
138 
139  // Define Homogeneous expansion
140  //int nplanes = fielddef[0]->m_numModes[2];
141 
142  int nplanes;
143  vSession->LoadParameter("HomModesZ",nplanes,fielddef[0]->m_numModes[2]);
144 
145  // choose points to be at evenly spaced points at
146  // nplanes + 1 points
147  const LibUtilities::PointsKey Pkey(nplanes+1,LibUtilities::ePolyEvenlySpaced);
148  const LibUtilities::BasisKey Bkey(fielddef[0]->m_basis[2],nplanes,Pkey);
149  NekDouble lz = fielddef[0]->m_homogeneousLengths[0];
150 
151  Exp3DH1 = MemoryManager<MultiRegions::ExpList3DHomogeneous1D>::AllocateSharedPtr(vSession,Bkey,lz,useFFT,dealiasing,graphShPt);
152  Exp[0] = Exp3DH1;
153 
154  for(i = 1; i < nfields + 1; ++i)
155  {
156  Exp[i] = MemoryManager<MultiRegions::ExpList3DHomogeneous1D>::AllocateSharedPtr(*Exp3DH1);
157  }
158  }
159  else
160  {
161  MultiRegions::ExpList2DSharedPtr Exp2D;
162  Exp2D = MemoryManager<MultiRegions::ExpList2D>
163  ::AllocateSharedPtr(vSession,graphShPt);
164  Exp[0] = Exp2D;
165 
166  for(i = 1; i < nfields + 1; ++i)
167  {
168  Exp[i] = MemoryManager<MultiRegions::ExpList2D>
169  ::AllocateSharedPtr(*Exp2D);
170  }
171  }
172  }
173  break;
174  case 3:
175  {
176  MultiRegions::ExpList3DSharedPtr Exp3D;
177  Exp3D = MemoryManager<MultiRegions::ExpList3D>
178  ::AllocateSharedPtr(vSession,graphShPt);
179  Exp[0] = Exp3D;
180 
181  for(i = 1; i < nfields + 1; ++i)
182  {
183  Exp[i] = MemoryManager<MultiRegions::ExpList3D>
184  ::AllocateSharedPtr(*Exp3D);
185  }
186  }
187  break;
188  default:
189  ASSERTL0(false,"Expansion dimension not recognised");
190  break;
191  }
192 
193  //----------------------------------------------
194  // Copy data from field file
195  for(j = 0; j < nfields; ++j)
196  {
197  for(unsigned int i = 0; i < fielddata.size(); ++i)
198  {
199  Exp[j]->ExtractDataToCoeffs(fielddef [i],
200  fielddata[i],
201  fielddef [i]->m_fields[j],
202  Exp[j]->UpdateCoeffs());
203  }
204  Exp[j]->BwdTrans(Exp[j]->GetCoeffs(),Exp[j]->UpdatePhys());
205  }
206  //----------------------------------------------
207 
208  int nq = Exp[0]->GetNpoints();
209 
210  Array<OneD, NekDouble> Ux(nq);
211  Array<OneD, NekDouble> Vy(nq);
212  Array<OneD, NekDouble> Div(nq);
213 
214  switch(expdim)
215  {
216  case 1:
217  {
218  if(fielddef[0]->m_numHomogeneousDir == 1)
219  {
220  ASSERTL0(false,"Not implemented yet");
221  }
222  else if(fielddef[0]->m_numHomogeneousDir == 2)
223  {
224  ASSERTL0(false,"Not implemented yet");
225  }
226  else
227  {
228  ASSERTL0(false,"Not implemented yet");
229  }
230  }
231  break;
232  case 2:
233  {
234  if(fielddef[0]->m_numHomogeneousDir == 1)
235  {
236  ASSERTL0(false,"Not implemented yet");
237  }
238  else
239  {
240  Exp[0]->PhysDeriv(MultiRegions::DirCartesianMap[0],Exp[0]->GetPhys(),Ux);
241  Exp[0]->PhysDeriv(MultiRegions::DirCartesianMap[1],Exp[1]->GetPhys(),Vy);
242 
243  Vmath::Vsub(nq,Ux,1,Vy,1,Div,1);
244 
245  Exp[3]->FwdTrans(Div,Exp[3]->UpdateCoeffs());
246  }
247  }
248  break;
249  case 3:
250  {
251  ASSERTL0(false,"Not implemented yet");
252  }
253  break;
254  default:
255  {
256  ASSERTL0(false,"Expansion dimension not recognised");
257  }
258  break;
259  }
260 
261  //-----------------------------------------------
262  // Write solution to file with additional computed fields
263  string fldfilename(argv[2]);
264  string out = fldfilename.substr(0, fldfilename.find_last_of("."));
265  string endfile("_with_divergence.fld");
266  out += endfile;
267  std::vector<LibUtilities::FieldDefinitionsSharedPtr> FieldDef
268  = Exp[0]->GetFieldDefinitions();
269  std::vector<std::vector<NekDouble> > FieldData(FieldDef.size());
270 
271  for(j = 0; j < nfields + 1; ++j)
272  {
273  for(i = 0; i < FieldDef.size(); ++i)
274  {
275  if (j >= nfields)
276  {
277  FieldDef[i]->m_fields.push_back("Div");
278  }
279  else
280  {
281  FieldDef[i]->m_fields.push_back(fielddef[i]->m_fields[j]);
282  }
283  Exp[j]->AppendFieldData(FieldDef[i], FieldData[i]);
284  }
285  }
286  LibUtilities::Write(out, FieldDef, FieldData);
287  //-----------------------------------------------
288 
289  return 0;
290 }
291 
Generated on Sun Mar 15 2015 00:11:29 for Nektar++ by   doxygen 1.8.1.2