59 m_session->MatchSolverInfo(
"SolverType",
"VelocityCorrectionScheme",
74 if (
m_session->DefinesSolverInfo(
"ModeType") &&
75 (boost::iequals(
m_session->GetSolverInfo(
"ModeType"),
"SingleMode") ||
76 boost::iequals(
m_session->GetSolverInfo(
"ModeType"),
"HalfMode")))
80 m_equ[0]->UpdateFields()[i]->SetWaveSpace(
true);
101 if (!
m_session->DefinesSolverInfo(
"HOMOGENEOUS") &&
102 !
m_session->DefinesSolverInfo(
"ModeType"))
105 "Imaginary shift only supported with HOMOGENEOUS "
106 "expansion and ModeType set to SingleMode");
108 else if (!boost::iequals(
m_session->GetSolverInfo(
"ModeType"),
112 "Imaginary shift only supported with HOMOGENEOUS "
113 "expansion and ModeType set to SingleMode");
124 ASSERTL0(
false,
"Specific version of Arnoldi driver not implemented");
132 if (
m_comm->GetRank() == 0)
134 if (
m_session->DefinesSolverInfo(
"ModeType") &&
135 boost::iequals(
m_session->GetSolverInfo(
"ModeType"),
"SingleMode"))
137 out <<
"\tSingle Fourier mode : true " << std::endl;
139 "Expected a homogeneous expansion to be defined "
144 out <<
"\tSingle Fourier mode : false " << std::endl;
146 if (
m_session->DefinesSolverInfo(
"BetaZero"))
148 out <<
"\tBeta set to Zero : true (overrides LHom)"
153 out <<
"\tBeta set to Zero : false " << std::endl;
158 out <<
"\tEvolution operator : "
159 <<
m_session->GetSolverInfo(
"EvolutionOperator") << std::endl;
163 out <<
"\tShift (Real,Imag) : " <<
m_realShift <<
","
166 out <<
"\tKrylov-space dimension : " <<
m_kdim << std::endl;
167 out <<
"\tNumber of vectors : " <<
m_nvec << std::endl;
168 out <<
"\tMax iterations : " <<
m_nits << std::endl;
169 out <<
"\tEigenvalue tolerance : " <<
m_evtol << std::endl;
170 out <<
"======================================================"
183 m_equ[0]->UpdateFields();
184 int nq = fields[0]->GetNcoeffs();
188 Vmath::Vcopy(nq, &array[k * nq], 1, &fields[k]->UpdateCoeffs()[0], 1);
189 fields[k]->SetPhysState(
false);
207 fields =
m_equ[0]->UpdateFields();
216 int nq = fields[0]->GetNcoeffs();
217 Vmath::Vcopy(nq, &fields[k]->GetCoeffs()[0], 1, &array[k * nq], 1);
218 fields[k]->SetPhysState(
false);
230 "Transient Growth non available for Coupled Solver");
232 fields =
m_equ[0]->UpdateFields();
233 int nq = fields[0]->GetNcoeffs();
238 &
m_equ[1]->UpdateFields()[k]->UpdateCoeffs()[0], 1);
249 std::vector<std::string> variables(
m_nfields);
254 variables[i] =
m_equ[0]->GetVariable(i);
257 m_equ[0]->WriteFld(file,
m_equ[0]->UpdateFields()[0], coeffs, variables);
266 std::vector<std::string> variables(
m_nfields);
267 std::vector<Array<OneD, NekDouble>> fieldcoeffs(
m_nfields);
269 int ncoeffs =
m_equ[0]->UpdateFields()[0]->GetNcoeffs();
271 "coeffs is not of sufficient size");
275 variables[i] =
m_equ[0]->GetVariable(i);
276 fieldcoeffs[i] = coeffs + i * ncoeffs;
279 m_equ[0]->WriteFld(file,
m_equ[0]->UpdateFields()[0], fieldcoeffs,
295 evlout <<
"EV: " << std::setw(2) << i << std::setw(12) << abs_ev
296 << std::setw(12) << ang_ev << std::setw(12)
301 evlout << std::setw(12) << resid;
307 NekDouble invmag = 1.0 / (re_ev * re_ev + im_ev * im_ev);
318 evlout <<
"EV: " << std::setw(2) << i << std::setw(14) <<
sign * re_ev
319 << std::setw(14) <<
sign * im_ev;
329 evlout << std::setw(12) << resid;
339 std::vector<std::vector<LibUtilities::EquationSharedPtr>> &selectedDomains,
340 std::set<int> &unselectedVariables)
342 selectedDomains.clear();
343 std::string domain(
"SelectEVCalcDomain0");
344 std::string condition(
"C0");
345 for (
size_t i = 0; i < 10; ++i)
347 domain[domain.size() - 1] =
'0' + i;
352 for (
size_t j = 0; j < 10; ++j)
354 condition[condition.size() - 1] =
'0' + j;
355 if (!
m_session->DefinesFunction(domain, condition))
361 selectedDomains.push_back(
362 std::vector<LibUtilities::EquationSharedPtr>());
364 selectedDomains[selectedDomains.size() - 1].push_back(
365 m_session->GetFunction(domain, condition));
368 unselectedVariables.clear();
369 std::string funName(
"SelectEVCalcVariables");
370 std::vector<std::string> variables =
m_session->GetVariables();
373 if (!
m_session->DefinesFunction(funName, variables[v]))
375 unselectedVariables.insert(v);
378 if (unselectedVariables.size() ==
m_nfields)
380 unselectedVariables.clear();
389 std::vector<std::vector<LibUtilities::EquationSharedPtr>> selectedDomains;
390 std::set<int> unselectedVariables;
392 if (selectedDomains.size() == 0 && unselectedVariables.size() == 0)
399 int ncoef =
field->GetNcoeffs();
400 int nphys =
field->GetNpoints();
403 for (
size_t i = 0; i <
field->GetExpSize(); ++i)
407 int nv = geom->GetNumVerts();
410 for (
size_t j = 0; j < nv; ++j)
413 vertex->GetCoords(gct[0], gct[1], gct[2]);
419 for (
size_t m = 0; m < selectedDomains.size(); ++m)
422 for (
size_t n = 0; n < selectedDomains[m].size(); ++n)
424 if (selectedDomains[m][n]->Evaluate(gc[0], gc[1], gc[2]) <= 0.)
441 if (unmask == 0 || unselectedVariables.count(j))
444 exp->GetNcoeffs(), 0.,
#define ASSERTL0(condition, msg)
#define NEKERROR(type, msg)
Assert Level 0 – Fundamental assert which is used whether in FULLDEBUG, DEBUG or OPT compilation mode...
#define ASSERTL1(condition, msg)
Assert Level 1 – Debugging which is used whether in FULLDEBUG or DEBUG compilation mode....
#define sign(a, b)
return the sign(b)*a
void CopyFwdToAdj()
Copy the forward field to the adjoint system in transient growth calculations.
void v_InitObject(std::ostream &out=std::cout) override
Virtual function for initialisation implementation.
void WriteFld(std::string file, std::vector< Array< OneD, NekDouble > > coeffs)
Write coefficients to file.
NekDouble m_period
Tolerance of iterations.
int m_infosteps
underlying operator is time stepping
void v_Execute(std::ostream &out=std::cout) override
Virtual function for solve implementation.
Array< OneD, NekDouble > m_maskCoeffs
DriverArnoldi(const LibUtilities::SessionReaderSharedPtr pSession, const SpatialDomains::MeshGraphSharedPtr pGraph)
Constructor.
void CopyFieldToArnoldiArray(Array< OneD, NekDouble > &array)
Copy fields to Arnoldi storage.
int m_nvec
Dimension of Krylov subspace.
bool m_timeSteppingAlgorithm
Period of time stepping algorithm.
int m_nits
Number of vectors to test.
void CopyArnoldiArrayToField(Array< OneD, NekDouble > &array)
Copy Arnoldi storage to fields.
void GetMaskInfo(std::vector< std::vector< LibUtilities::EquationSharedPtr > > &selectedDomains, std::set< int > &unselectedVariables)
Array< OneD, NekDouble > m_maskPhys
NekDouble m_evtol
Maxmum number of iterations.
int m_nfields
interval to dump information if required.
SOLVER_UTILS_EXPORT void ArnoldiSummary(std::ostream &out)
void WriteEvs(std::ostream &evlout, const int k, const NekDouble real, const NekDouble imag, NekDouble resid=NekConstants::kNekUnsetDouble, bool DumpInverse=true)
void MaskInit()
Init mask.
Base class for the development of solvers.
LibUtilities::SessionReaderSharedPtr m_session
Session reader object.
virtual SOLVER_UTILS_EXPORT void v_InitObject(std::ostream &out=std::cout)
Virtual function for initialisation implementation.
LibUtilities::CommSharedPtr m_comm
Communication object.
enum EvolutionOperatorType m_EvolutionOperator
Evolution Operator.
Array< OneD, EquationSystemSharedPtr > m_equ
Equation system to solve.
int m_nequ
number of equations
std::shared_ptr< SessionReader > SessionReaderSharedPtr
std::shared_ptr< Expansion > ExpansionSharedPtr
std::shared_ptr< ExpList > ExpListSharedPtr
Shared pointer to an ExpList object.
static const NekDouble kNekUnsetDouble
std::shared_ptr< MeshGraph > MeshGraphSharedPtr
std::shared_ptr< PointGeom > PointGeomSharedPtr
std::shared_ptr< Geometry > GeometrySharedPtr
void Fill(int n, const T alpha, T *x, const int incx)
Fill a vector with a constant value.
void Vcopy(int n, const T *x, const int incx, T *y, const int incy)
scalarT< T > log(scalarT< T > in)