MapBeamLine::MapBeamLine(string filename, string filenameerr, int order, int nbthreads, int fmultipole, bool strpl) {
Twiss t = Twiss(filename);
Twiss terr = Twiss(filenameerr);
if (strpl) {
t.stripLine();
terr.stripLine();
}
omp_set_num_threads(nbthreads);
vector<vector<Polynom<double>>> v = separateComplexList(EQ(4, order));
Polynom<double> x = X<double>(order);
Polynom<double> px = PX<double>(order);
Polynom<double> y = Y<double>(order);
Polynom<double> py = PY<double>(order);
Polynom<double> d = D<double>(order);
Polynom<double> s = S<double>(order);
Polmap<double> R = generateDefaultMap(x, px, y, py, d, s);
Polmap<double>* Res = new Polmap<double>[nbthreads];
for (int i = 0; i < nbthreads; i ++)
Res[i] = R;
int size = t.elems.size();
if (strpl) {
#pragma omp parallel for shared(Res) schedule(static)
for (int i = 0; i < size; i ++) {
int index = omp_get_thread_num();
Polmap<double> mp = mapForElement(t.elems[i], v, x, px, y, py, d, s, fmultipole);
double dx = atof(terr.elems[i][DX].c_str());
double dy = atof(terr.elems[i][DY].c_str());
mp = mp.eval("x", Polynom<double>(order, 1E-18, "x", 1) + dx);
mp = mp.eval("y", Polynom<double>(order, 1E-18, "y", 1) + dy);
Res[index] = mp * Res[index];
}
}
else {
#pragma omp parallel for shared(Res) schedule(static)
for (int i = 0; i < size; i ++) {
int index = omp_get_thread_num();
Polmap<double> mp = mapForElement(t.elems[i], v, x, px, y, py, d, s, fmultipole);
double dx = atof(terr.elems[i][DX].c_str());
double dy = atof(terr.elems[i][DY].c_str());
mp = mp.eval("x", Polynom<double>(order, 1E-18, "x", 1) + dx);
mp = mp.eval("y", Polynom<double>(order, 1E-18, "y", 1) + dy);
if (mp.pols.size() != 0)
Res[index] = mp * Res[index];
}
}
R = Res[0];
for (int i = 1; i < nbthreads; i ++)
R = Res[i] * R;
polmap = R.getMap();
for (unordered_map<string, Polynom<double>>:: iterator it = R.pols.begin(); it != R.pols.end(); it ++)
pols[it->first] = it->second;
delete [] Res;
}
