Nektar++
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
ProcessNumModes.cpp
Go to the documentation of this file.
1 ////////////////////////////////////////////////////////////////////////////////
2 //
3 // File: ProcessNumModes.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: Writes the number of modes in each direction.
33 //
34 ////////////////////////////////////////////////////////////////////////////////
35 
36 #include <iostream>
37 #include <string>
38 using namespace std;
39 
40 #include "ProcessNumModes.h"
41 
44 #include <StdRegions/StdQuadExp.h>
45 
46 namespace Nektar
47 {
48 namespace FieldUtils
49 {
50 
51 ModuleKey ProcessNumModes::className =
53  ModuleKey(eProcessModule, "nummodes"),
54  ProcessNumModes::create,
55  "Computes number of modes in each direction for each element.");
56 
57 ProcessNumModes::ProcessNumModes(FieldSharedPtr f) : ProcessModule(f)
58 {
59 }
60 
62 {
63 }
64 
65 void ProcessNumModes::Process(po::variables_map &vm)
66 {
67  if (m_f->m_verbose)
68  {
69  if (m_f->m_comm->TreatAsRankZero())
70  {
71  cout << "ProcessNumModes: Calculating number of modes..." << endl;
72  }
73  }
74 
75  int i, j, s;
76  int expdim = m_f->m_graph->GetMeshDimension();
77  int nfields = m_f->m_fielddef[0]->m_fields.size();
78  int addfields = expdim;
79  int npoints = m_f->m_exp[0]->GetNpoints();
80  Array<OneD, Array<OneD, NekDouble> > outfield(addfields);
81 
82  int nstrips;
83 
84  m_f->m_session->LoadParameter("Strip_Z", nstrips, 1);
85 
86  m_f->m_exp.resize(nfields * nstrips);
87 
88  for (i = 0; i < addfields; ++i)
89  {
90  outfield[i] = Array<OneD, NekDouble>(npoints);
91  }
92 
93  vector<MultiRegions::ExpListSharedPtr> Exp(nstrips * addfields);
94 
95  int nExp, nq, offset;
96  nExp = m_f->m_exp[0]->GetExpSize();
97 
98  for (int n = 0; n < nExp; n++)
99  {
100  offset = m_f->m_exp[0]->GetPhys_Offset(n);
101  nq = m_f->m_exp[0]->GetExp(n)->GetTotPoints();
102 
103  for (i = 0; i < expdim; i++)
104  {
105  int P = m_f->m_exp[0]->GetExp(n)->GetBasis(i)->GetNumModes();
106  Array<OneD, NekDouble> result = outfield[i] + offset;
107  Vmath::Fill(nq, 1.0 * P, result, 1);
108  }
109  }
110 
111  for (s = 0; s < nstrips; ++s)
112  {
113  for (i = 0; i < addfields; ++i)
114  {
115  int n = s * addfields + i;
116  Exp[n] =
117  m_f->AppendExpList(m_f->m_fielddef[0]->m_numHomogeneousDir);
118  Vmath::Vcopy(npoints, outfield[i], 1, Exp[n]->UpdatePhys(), 1);
119  Exp[n]->FwdTrans_IterPerExp(outfield[i], Exp[n]->UpdateCoeffs());
120  }
121  }
122 
124  for (s = 0; s < nstrips; ++s)
125  {
126  for (i = 0; i < addfields; ++i)
127  {
128  it = m_f->m_exp.begin() + s * (nfields + addfields) + nfields + i;
129  m_f->m_exp.insert(it, Exp[s * addfields + i]);
130  }
131  }
132 
133  vector<string> outname;
134  outname.push_back("P1");
135  if (addfields >= 2)
136  {
137  outname.push_back("P2");
138  }
139 
140  if (addfields == 3)
141  {
142  outname.push_back("P3");
143  }
144 
145  std::vector<LibUtilities::FieldDefinitionsSharedPtr> FieldDef =
146  m_f->m_exp[0]->GetFieldDefinitions();
147  std::vector<std::vector<NekDouble> > FieldData(FieldDef.size());
148 
149  // homogeneous strip variant
150  for (s = 0; s < nstrips; ++s)
151  {
152  for (j = 0; j < nfields + addfields; ++j)
153  {
154  for (i = 0; i < FieldDef.size() / nstrips; ++i)
155  {
156  int n = s * FieldDef.size() / nstrips + i;
157 
158  if (j >= nfields)
159  {
160  FieldDef[n]->m_fields.push_back(outname[j - nfields]);
161  }
162  else
163  {
164  FieldDef[n]->m_fields.push_back(
165  m_f->m_fielddef[0]->m_fields[j]);
166  }
167 
168  m_f->m_exp[s * (nfields + addfields) + j]->AppendFieldData(
169  FieldDef[n], FieldData[n]);
170  }
171  }
172  }
173 
174  m_f->m_fielddef = FieldDef;
175  m_f->m_data = FieldData;
176 }
177 }
178 }
virtual void Process(po::variables_map &vm)
Write mesh to output file.
void Fill(int n, const T alpha, T *x, const int incx)
Fill a vector with a constant value.
Definition: Vmath.cpp:46
STL namespace.
pair< ModuleType, string > ModuleKey
boost::shared_ptr< Field > FieldSharedPtr
Definition: Field.hpp:767
StandardMatrixTag boost::call_traits< LhsDataType >::const_reference rhs typedef NekMatrix< LhsDataType, StandardMatrixTag >::iterator iterator
Abstract base class for processing modules.
void Vcopy(int n, const T *x, const int incx, T *y, const int incy)
Definition: Vmath.cpp:1061
ModuleFactory & GetModuleFactory()
FieldSharedPtr m_f
Field object.
tKey RegisterCreatorFunction(tKey idKey, CreatorFunction classCreator, tDescription pDesc="")
Register a class with the factory.
Definition: NekFactory.hpp:215