9 int main(
int argc,
char *argv[])
17 "Usage: AddModeTo2DFld scal1 scal2 2Dfieldfile1 fieldfile2 "
19 "\t produces scal1*2Dfieldfiel1 + scal2*fieldfile2 in outfield\n");
23 scal1 = boost::lexical_cast<double>(argv[argc - 5]);
24 scal2 = boost::lexical_cast<double>(argv[argc - 4]);
28 string fieldfile1(argv[argc - 3]);
29 vector<LibUtilities::FieldDefinitionsSharedPtr> fielddef1;
30 vector<vector<NekDouble>> fielddata1;
36 string fieldfile2(argv[argc - 2]);
37 vector<LibUtilities::FieldDefinitionsSharedPtr> fielddef2;
38 vector<vector<NekDouble>> fielddata2;
42 vector<vector<NekDouble>> combineddata;
44 ASSERTL0(fielddata1.size() == fielddata2.size(),
45 "Inner has different size");
54 for (i = 0; i < fielddata2.size(); ++i)
56 ASSERTL0(fielddef2[i]->m_numHomogeneousDir == 1,
57 "Expected second fld to have one homogeneous direction");
58 ASSERTL0(fielddef2[i]->m_numModes[2] == 2,
59 "Expected Fourier field to have 2 modes");
61 int datalen1 = fielddata1[i].size() / fielddef1[i]->m_fields.size();
62 int datalen2 = fielddata2[i].size() / fielddef2[i]->m_fields.size();
65 "Data per fields is note compatible");
69 switch (fielddef2[i]->m_shapeType)
73 fielddef2[i]->m_numModes[0], fielddef2[i]->m_numModes[1]);
77 fielddef2[i]->m_numModes[0] * fielddef2[i]->m_numModes[1];
80 ASSERTL0(
false,
"Shape not recognised");
88 Vmath::Smul(fielddata1[i].size(), scal1, &fielddata1[i][0], 1,
89 &fielddata1[i][0], 1);
90 Vmath::Smul(fielddata2[i].size(), scal2, &fielddata2[i][0], 1,
91 &fielddata2[i][0], 1);
93 vector<NekDouble> newdata;
94 auto vec_iter = fielddata2[i].begin();
96 for (k = 0; k < fielddef2[i]->m_fields.size(); ++k)
100 for (j = 0; j < fielddef1[i]->m_fields.size(); ++j)
102 if (fielddef1[i]->m_fields[j] == fielddef2[i]->m_fields[k])
109 if (j != fielddef1[i]->m_fields.size())
111 for (n = 0; n < fielddef2[i]->m_elementIDs.size(); ++n)
115 newdata.end(), &(fielddata1[i][offset + n * ncoeffs]),
116 &(fielddata1[i][offset + n * ncoeffs]) + ncoeffs);
119 newdata.insert(newdata.end(), &
Zero[0], &
Zero[0] + ncoeffs);
122 newdata.insert(newdata.end(), vec_iter,
123 vec_iter + 2 * ncoeffs);
124 vec_iter += 2 * ncoeffs;
130 for (n = 0; n < fielddef2[i]->m_elementIDs.size(); ++n)
133 newdata.insert(newdata.end(), &
Zero[0], &
Zero[0] + ncoeffs);
134 newdata.insert(newdata.end(), &
Zero[0], &
Zero[0] + ncoeffs);
137 newdata.insert(newdata.end(), vec_iter,
138 vec_iter + 2 * ncoeffs);
139 vec_iter += 2 * ncoeffs;
143 combineddata.push_back(newdata);
144 fielddef2[i]->m_numModes[2] += 2;
145 fielddef2[i]->m_homogeneousZIDs.push_back(2);
146 fielddef2[i]->m_homogeneousZIDs.push_back(3);
150 for (k = 0; k < fielddef1[i]->m_fields.size(); ++k)
152 for (j = 0; j < fielddef2[i]->m_fields.size(); ++j)
154 if (fielddef1[i]->m_fields[k] == fielddef2[i]->m_fields[j])
160 if (j == fielddef2[i]->m_fields.size())
162 cout <<
"Warning: Field \'" << fielddef1[i]->m_fields[k]
163 <<
"\' was not included in output " << endl;
int main(int argc, char *argv[])
#define ASSERTL0(condition, msg)
int getNumberOfCoefficients(int Na)
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 ...
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...
The above copyright notice and this permission notice shall be included.
void Smul(int n, const T alpha, const T *x, const int incx, T *y, const int incy)
Scalar multiply y = alpha*x.
void Zero(int n, T *x, const int incx)
Zero vector.