mirror of
https://github.com/paboyle/Grid.git
synced 2025-04-06 20:25:56 +01:00
Restructured the class that computes and returns the A2A vectors.
This commit is contained in:
parent
ae37fda699
commit
7a0abfac89
@ -13,9 +13,79 @@ BEGIN_HADRONS_NAMESPACE
|
||||
template <class Field, class Matrix>
|
||||
class A2AModesSchurDiagTwo
|
||||
{
|
||||
private:
|
||||
const std::vector<Field> *evec;
|
||||
const std::vector<RealD> *eval;
|
||||
Matrix &action;
|
||||
std::function<void(Field &, const Field &)> &Solver;
|
||||
const int Nl, Nh;
|
||||
const bool return_5d;
|
||||
std::vector<Field> w_high_5d, v_high_5d, w_high_4d, v_high_4d;
|
||||
|
||||
public:
|
||||
A2AModesSchurDiagTwo(void) = default;
|
||||
virtual ~A2AModesSchurDiagTwo(void) = default;
|
||||
A2AModesSchurDiagTwo(const std::vector<Field> *_evec, const std::vector<RealD> *_eval,
|
||||
Matrix &_action,
|
||||
std::function<void(Field &, const Field &)> &_Solver,
|
||||
const int _Nl, const int _Nh,
|
||||
const bool _return_5d)
|
||||
: evec(_evec), eval(_eval),
|
||||
action(_action),
|
||||
Solver(_Solver),
|
||||
Nl(_Nl), Nh(_Nh),
|
||||
return_5d(_return_5d)
|
||||
{
|
||||
init_resize(1, Nh);
|
||||
if (return_5d) init_resize(Nh, Nh);
|
||||
};
|
||||
|
||||
void init_resize(const size_t size_5d, const size_t size_4d)
|
||||
{
|
||||
GridBase *grid_5d = action.Grid();
|
||||
GridBase *grid_4d = action.GaugeGrid();
|
||||
|
||||
w_high_5d.resize(size_5d, grid_5d);
|
||||
v_high_5d.resize(size_5d, grid_5d);
|
||||
|
||||
w_high_4d.resize(size_4d, grid_4d);
|
||||
v_high_4d.resize(size_4d, grid_4d);
|
||||
}
|
||||
|
||||
void high_modes(Field &source_5d, Field &source_4d, int i)
|
||||
{
|
||||
int i5d;
|
||||
LOG(Message) << "A2A high modes for i = " << i << std::endl;
|
||||
i5d = 0;
|
||||
if (return_5d) i5d = i;
|
||||
this->high_mode_v(action, Solver, source_5d, v_high_5d[i5d], v_high_4d[i]);
|
||||
this->high_mode_w(source_5d, source_4d, w_high_5d[i5d], w_high_4d[i]);
|
||||
}
|
||||
|
||||
void return_v(int i, Field &vout_5d, Field &vout_4d)
|
||||
{
|
||||
if (i < Nl)
|
||||
{
|
||||
this->low_mode_v(action, evec->at(i), eval->at(i), vout_5d, vout_4d);
|
||||
}
|
||||
else
|
||||
{
|
||||
vout_4d = v_high_4d[i - Nl];
|
||||
if (!(return_5d)) i = Nl;
|
||||
vout_5d = v_high_5d[i - Nl];
|
||||
}
|
||||
}
|
||||
void return_w(int i, Field &wout_5d, Field &wout_4d)
|
||||
{
|
||||
if (i < Nl)
|
||||
{
|
||||
this->low_mode_w(action, evec->at(i), eval->at(i), wout_5d, wout_5d);
|
||||
}
|
||||
else
|
||||
{
|
||||
wout_4d = w_high_4d[i - Nl];
|
||||
if (!(return_5d)) i = Nl;
|
||||
wout_5d = w_high_5d[i - Nl];
|
||||
}
|
||||
}
|
||||
|
||||
void Doo(Matrix &action, const Field &in, Field &out)
|
||||
{
|
||||
@ -26,10 +96,10 @@ class A2AModesSchurDiagTwo
|
||||
action.MooeeInv(tmp, out);
|
||||
action.Meooe(out, tmp);
|
||||
|
||||
axpy(out, -1.0, tmp, in);
|
||||
axpy(out,-1.0, tmp, in);
|
||||
}
|
||||
|
||||
void low_mode_v(Matrix &action, const Field &evec, const RealD &eval, Field &vout, bool return_5d = true)
|
||||
void low_mode_v(Matrix &action, const Field &evec, const RealD &eval, Field &vout_5d, Field &vout_4d)
|
||||
{
|
||||
|
||||
GridBase *grid = action.RedBlackGrid();
|
||||
@ -38,13 +108,10 @@ class A2AModesSchurDiagTwo
|
||||
Field sol_o(grid);
|
||||
Field tmp(grid);
|
||||
|
||||
GridBase *fgrid = action.Grid();
|
||||
Field tmp_out(fgrid);
|
||||
|
||||
src_o = evec;
|
||||
src_o.checkerboard = Odd;
|
||||
pickCheckerboard(Even, sol_e, tmp_out);
|
||||
pickCheckerboard(Odd, sol_o, tmp_out);
|
||||
pickCheckerboard(Even, sol_e, vout_5d);
|
||||
pickCheckerboard(Odd, sol_o, vout_5d);
|
||||
|
||||
/////////////////////////////////////////////////////
|
||||
// v_ie = -(1/eval_i) * MeeInv Meo MooInv evec_i
|
||||
@ -66,15 +133,15 @@ class A2AModesSchurDiagTwo
|
||||
sol_o = (1.0 / eval) * tmp;
|
||||
assert(sol_o.checkerboard == Odd);
|
||||
|
||||
setCheckerboard(tmp_out, sol_e);
|
||||
setCheckerboard(vout_5d, sol_e);
|
||||
assert(sol_e.checkerboard == Even);
|
||||
setCheckerboard(tmp_out, sol_o);
|
||||
setCheckerboard(vout_5d, sol_o);
|
||||
assert(sol_o.checkerboard == Odd);
|
||||
|
||||
this->return_dim(action, tmp_out, vout, return_5d);
|
||||
action.ExportPhysicalFermionSolution(vout_5d, vout_4d);
|
||||
}
|
||||
|
||||
void low_mode_w(Matrix &action, const Field &evec, const RealD &eval, Field &wout, bool return_5d = true)
|
||||
void low_mode_w(Matrix &action, const Field &evec, const RealD &eval, Field &wout_5d, Field &wout_4d)
|
||||
{
|
||||
GridBase *grid = action.RedBlackGrid();
|
||||
SchurDiagTwoOperator<Matrix, Field> _HermOpEO(action);
|
||||
@ -85,7 +152,6 @@ class A2AModesSchurDiagTwo
|
||||
Field tmp(grid);
|
||||
|
||||
GridBase *fgrid = action.Grid();
|
||||
Field tmp_out(fgrid);
|
||||
Field tmp_wout(fgrid);
|
||||
|
||||
src_o = evec;
|
||||
@ -115,41 +181,24 @@ class A2AModesSchurDiagTwo
|
||||
setCheckerboard(tmp_wout, sol_o);
|
||||
assert(sol_o.checkerboard == Odd);
|
||||
|
||||
action.DminusDag(tmp_wout, tmp_out);
|
||||
this->return_dim(action, tmp_out, wout, return_5d);
|
||||
action.DminusDag(tmp_wout, wout_5d);
|
||||
action.ExportPhysicalFermionSolution(wout_5d, wout_4d);
|
||||
}
|
||||
|
||||
void high_mode_v(Matrix &action, std::function<void(Field &, const Field &)> &Solver, const Field &source, Field &vout, bool return_5d = true)
|
||||
void high_mode_v(Matrix &action, std::function<void(Field &, const Field &)> &Solver, const Field &source, Field &vout_5d, Field &vout_4d)
|
||||
{
|
||||
GridBase *fgrid = action.Grid();
|
||||
Field tmp(fgrid);
|
||||
Field tmp_out(fgrid);
|
||||
|
||||
action.Dminus(source, tmp);
|
||||
Solver(tmp_out, source); // Note: Solver is Solver(out, in)
|
||||
this->return_dim(action, tmp_out, vout, return_5d);
|
||||
Solver(vout_5d, source); // Note: Solver is Solver(out, in)
|
||||
action.ExportPhysicalFermionSolution(vout_5d, vout_4d);
|
||||
}
|
||||
|
||||
void high_mode_w(Matrix &action, const Field &source4d, Field &wout, bool return_5d = true)
|
||||
void high_mode_w(const Field &source_5d, const Field &source_4d, Field &wout_5d, Field &wout_4d)
|
||||
{
|
||||
// GridBase *fgrid = action.Grid();
|
||||
// Field tmp_out(fgrid);
|
||||
|
||||
// tmp_out = source;
|
||||
// this->return_dim(action, tmp_out, wout, return_5d);
|
||||
wout = source4d;
|
||||
}
|
||||
|
||||
void return_dim(Matrix &action, const Field &in, Field &out, bool return_5d)
|
||||
{
|
||||
if (return_5d)
|
||||
{
|
||||
out = in;
|
||||
}
|
||||
else
|
||||
{
|
||||
action.ExportPhysicalFermionSolution(in, out);
|
||||
}
|
||||
wout_5d = source_5d;
|
||||
wout_4d = source_4d;
|
||||
}
|
||||
};
|
||||
|
||||
@ -211,75 +260,6 @@ class A2AHMSchurDiagTwo : virtual public A2AModesSchurDiagTwo<Field, Matrix>
|
||||
}
|
||||
};
|
||||
|
||||
////////////////////////////////
|
||||
// Both Modes
|
||||
////////////////////////////////
|
||||
|
||||
template <class Field, class Matrix>
|
||||
class A2AVectorsReturn : public A2AModesSchurDiagTwo<Field, Matrix>
|
||||
{
|
||||
private:
|
||||
const std::vector<Field> *evec;
|
||||
const std::vector<RealD> *eval;
|
||||
Matrix &action;
|
||||
std::function<void(Field &, const Field &)> &Solver;
|
||||
const int Nl, Nh;
|
||||
const bool return_5d;
|
||||
std::vector<Field> w_high, v_high;
|
||||
|
||||
public:
|
||||
A2AVectorsReturn(const std::vector<Field> *_evec, const std::vector<RealD> *_eval,
|
||||
Matrix &_action,
|
||||
std::function<void(Field &, const Field &)> &_Solver,
|
||||
const int _Nl, const int _Nh,
|
||||
const bool _return_5d)
|
||||
: evec(_evec), eval(_eval),
|
||||
action(_action),
|
||||
Solver(_Solver),
|
||||
Nl(_Nl), Nh(_Nh),
|
||||
return_5d(_return_5d)
|
||||
{
|
||||
GridBase *grid;
|
||||
if (return_5d)
|
||||
{
|
||||
grid = action.Grid();
|
||||
}
|
||||
else
|
||||
{
|
||||
grid = action.GaugeGrid();
|
||||
}
|
||||
resize(Nh, grid);
|
||||
};
|
||||
|
||||
void resize(const size_t size, GridBase *grid)
|
||||
{
|
||||
w_high.resize(size, grid);
|
||||
v_high.resize(size, grid);
|
||||
}
|
||||
|
||||
void high_modes(Field &source5d, Field &source4d, int i)
|
||||
{
|
||||
LOG(Message) << "A2A high modes for i = " << i << std::endl;
|
||||
this->high_mode_v(action, Solver, source5d, v_high[i], return_5d);
|
||||
this->high_mode_w(action, source4d, w_high[i], return_5d);
|
||||
}
|
||||
|
||||
void operator()(int i, Field &vout, Field &wout)
|
||||
{
|
||||
if (i < Nl)
|
||||
{
|
||||
LOG(Message) << "A2A low modes for i = " << i << std::endl;
|
||||
this->low_mode_v(action, evec->at(i), eval->at(i), vout, return_5d);
|
||||
this->low_mode_w(action, evec->at(i), eval->at(i), wout, return_5d);
|
||||
}
|
||||
else
|
||||
{
|
||||
vout = v_high[i - Nl];
|
||||
wout = w_high[i - Nl];
|
||||
}
|
||||
}
|
||||
};
|
||||
|
||||
END_HADRONS_NAMESPACE
|
||||
|
||||
#endif // A2A_Vectors_hpp_
|
Loading…
x
Reference in New Issue
Block a user