Nektar++
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
ProcessHomogeneousPlane.cpp
Go to the documentation of this file.
1 ////////////////////////////////////////////////////////////////////////////////
2 //
3 // File: ProcessHomogeneousPlane.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: Extract a single plane of a 3DH1D field.
33 //
34 ////////////////////////////////////////////////////////////////////////////////
35 
36 #include <iostream>
37 #include <string>
38 using namespace std;
39 
41 
44 
45 namespace Nektar
46 {
47 namespace FieldUtils
48 {
49 
50 ModuleKey ProcessHomogeneousPlane::className =
52  ModuleKey(eProcessModule, "homplane"),
53  ProcessHomogeneousPlane::create,
54  "Extracts a plane from a 3DH1D expansion, requires planeid to be "
55  "defined.");
56 
57 ProcessHomogeneousPlane::ProcessHomogeneousPlane(FieldSharedPtr f)
58  : ProcessModule(f)
59 {
60  m_config["planeid"] = ConfigOption(false, "NotSet", "plane id to extract");
61  m_config["wavespace"] =
62  ConfigOption(true, "NotSet", "Extract plane in Fourier space");
63 }
64 
66 {
67 }
68 
69 void ProcessHomogeneousPlane::Process(po::variables_map &vm)
70 {
71  if (m_f->m_verbose)
72  {
73  if (m_f->m_comm->TreatAsRankZero())
74  {
75  cout << "ProcessHomogeneousPlane: Extracting plane..." << endl;
76  }
77  }
78 
79  if ((m_f->m_fielddef[0]->m_numHomogeneousDir) != 1)
80  {
81  ASSERTL0(false,
82  "ProcessHomogeneousPlane only works for Homogeneous1D.");
83  }
84 
85  ASSERTL0(m_config["planeid"].m_beenSet,
86  "Missing parameter planeid for ProcessHomogeneousPlane");
87 
88  int planeid = m_config["planeid"].as<int>();
89  int nfields = m_f->m_fielddef[0]->m_fields.size();
90 
91  int nstrips;
92  m_f->m_session->LoadParameter("Strip_Z", nstrips, 1);
93 
94  // Look for correct plane (because of parallel case)
95  int plane = -1;
96  for (int i = 0; i < m_f->m_fielddef[0]->m_homogeneousZIDs.size(); ++i)
97  {
98  if (m_f->m_fielddef[0]->m_homogeneousZIDs[i] == planeid)
99  {
100  plane = i;
101  }
102  }
103 
104  if (plane != -1)
105  {
106  for (int s = 0; s < nstrips; ++s)
107  {
108  for (int i = 0; i < nfields; ++i)
109  {
110  int n = s * nfields + i;
111  m_f->m_exp[n] = m_f->m_exp[n]->GetPlane(plane);
112 
113  if (m_config["wavespace"].m_beenSet)
114  {
115  m_f->m_exp[n]->BwdTrans(m_f->m_exp[n]->GetCoeffs(),
116  m_f->m_exp[n]->UpdatePhys());
117  }
118  else
119  {
120  m_f->m_exp[n]->FwdTrans(m_f->m_exp[n]->GetPhys(),
121  m_f->m_exp[n]->UpdateCoeffs());
122  }
123  }
124  }
125  std::vector<LibUtilities::FieldDefinitionsSharedPtr> FieldDef =
126  m_f->m_exp[0]->GetFieldDefinitions();
127  std::vector<std::vector<NekDouble> > FieldData(FieldDef.size());
128 
129  for (int s = 0; s < nstrips; ++s)
130  {
131  for (int j = 0; j < nfields; ++j)
132  {
133  for (int i = 0; i < FieldDef.size() / nstrips; ++i)
134  {
135  int n = s * FieldDef.size() / nstrips + i;
136 
137  FieldDef[n]->m_fields.push_back(
138  m_f->m_fielddef[0]->m_fields[j]);
139  m_f->m_exp[s * nfields + j]->AppendFieldData(FieldDef[n],
140  FieldData[n]);
141  }
142  }
143  }
144 
145  m_f->m_fielddef = FieldDef;
146  m_f->m_data = FieldData;
147  }
148  else
149  {
150  for (int s = 0; s < nstrips; ++s)
151  {
152  for (int i = 0; i < nfields; ++i)
153  {
154  int n = s * nfields + i;
155  m_f->m_exp[n] =
157  }
158  }
159  m_f->m_fielddef =
160  std::vector<LibUtilities::FieldDefinitionsSharedPtr>();
161  m_f->m_data = std::vector<std::vector<NekDouble> >();
162  }
163 }
164 }
165 }
map< string, ConfigOption > m_config
List of configuration values.
#define ASSERTL0(condition, msg)
Definition: ErrorUtil.hpp:198
virtual void Process(po::variables_map &vm)
Write mesh to output file.
static boost::shared_ptr< DataType > AllocateSharedPtr()
Allocate a shared pointer from the memory pool.
Represents a command-line configuration option.
STL namespace.
pair< ModuleType, string > ModuleKey
boost::shared_ptr< Field > FieldSharedPtr
Definition: Field.hpp:767
Abstract base class for processing modules.
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