115 lst lmom = ex_to<lst>(aio.
Internal);
116 lst emom = ex_to<lst>(aio.
External);
120 if(IBPmethod==1) wdir = aio.
SaveDir +
"/FIRE";
121 else if(IBPmethod==2) wdir = aio.
SaveDir +
"/KIRA";
122 else if(IBPmethod==3) wdir = aio.
SaveDir +
"/UKIRA";
124 wdir = to_string(getpid());
125 if(IBPmethod==1) wdir = wdir +
"_FIRE";
126 else if(IBPmethod==2) wdir = wdir +
"_KIRA";
127 else if(IBPmethod==3) wdir = wdir +
"_UKIRA";
131 vector<IBP*> ibp_vec;
133 if(
Verbose > 1) cout <<
PRE <<
"\\--Reading AIR2F" << flush;
134 AIR2F_Get(aio.
SaveDir, air_vec, IntFs, ibp_vec, IBPmethod);
135 for(
auto ibp : ibp_vec) ibp->WorkingDir = wdir;
136 if(
Verbose > 1) cout <<
" @ " <<
now(
false) << endl;
142 if(
Verbose > 1) cout <<
PRE <<
"\\--Reading AP.gar" << flush;
144 if(
Verbose > 1) cout <<
" @ " <<
now(
false) << endl;
146 }
else rc = system((
"mkdir -p "+aio.
SaveDir).c_str());
150 int av_size = air_vec.size();
156 for(
int i=0; i<av_size; i++) {
157 for(
auto cv : ex_to<lst>(air_vec[i])) vset.insert(cv.op(1));
159 exvector vvec(vset.begin(), vset.end());
163 for(
int i=0; i<vvec.size(); i++) v2api[vvec[i]] = i;
164 for(
int i=0; i<av_size; i++) {
166 lst cvs = ex_to<lst>(air_vec[i]);
167 air_vec[i] = av_item;
168 for(
auto & cv : cvs) av_item.append(lst{cv.op(0), v2api[cv.op(1)]});
170 air_vec[i] = av_item;
174 auto ap_vec =
GiNaC_Parallel(vvec.size(), [&vvec,lmom,emom,aio] (
int idx) {
175 auto air = vvec[idx];
176 air = Apart(air,lmom,emom,aio.smap);
177 air = air.subs(ApartIR(1,w)==aio.apart1);
178 air = collect_lst(air, ApartIR(w1,w2), o_flintf);
184 for(
auto cvs : ap_vec)
for(
auto cv : cvs)
if(is_a<matrix>(cv.op(1).op(0))) ap_set.insert(cv.op(1));
185 exvector ap_ir_vec(ap_set.begin(), ap_set.end());
192 ap_vec =
GiNaC_Parallel(ap_vec.size(), [&aio,&ap_vec,&ap_rules] (
int idx) {
193 auto const & cvs = ap_vec[idx];
196 for(auto const & cv : cvs) {
197 auto fi = ap_rules.find(cv.op(1));
198 if(fi==ap_rules.end()) res += cv.op(0) * cv.op(1);
199 else res += cv.op(0) * fi->second;
202 for(auto cv : cvs) res += cv.op(0) * ApartIRC(cv.op(1));
207 int vn = ap_vec.size();
208 for(
int idx=0; idx<vn; idx++) {
209 lst cvs = ex_to<lst>(ap_vec[idx]);
212 for(
auto const & cv : cvs) {
213 auto fi = ap_rules.find(cv.op(1));
214 if(fi==ap_rules.end()) res += cv.op(0) * cv.op(1);
215 else res += cv.op(0) * fi->second;
218 for(
auto cv : cvs) res += cv.op(0) *
ApartIRC(cv.op(1));
228 lst cvs = ex_to<lst>(air_vec[idx]);
230 for(
auto const & cv : cvs) {
231 int idx = ex_to<numeric>(cv.op(1)).to_int();
232 res += cv.op(0) * ap_vec[idx];
239 if(aio.SaveDir !=
"") {
240 garWrite(air_vec,aio.SaveDir+
"/AP.gar");
241 garRead(air_vec,aio.SaveDir+
"/AP.gar");
250 auto ret =
GiNaC_Parallel(air_vec.size(), [&air_vec](
int idx)->ex {
251 auto air = air_vec[idx];
253 find(air, ApartIR(w1,w2), airs);
255 for(auto item : airs) ret.append(item);
259 for(
auto airs : ret) for(auto air : ex_to<lst>(airs))
intg.insert(air);
260 AIR = exvector(
intg.begin(),
intg.end());
263 for(
auto sp : aio.CSP)
SP_map.erase(
sp);
269 repls.append(w*kv.first == w*kv.second);
270 repls.append(kv.first == kv.second);
274 for(
auto li : lmom) {
275 if(is_a<Vector>(li)) loops.append(ex_to<Vector>(li).name);
276 else loops.append(li);
278 for(
auto li : emom) {
279 if(is_a<Vector>(li)) exts.append(ex_to<Vector>(li).name);
280 else exts.append(li);
283 if(Verbose>0) cout <<
PRE <<
"\\--Prepare " <<
WHITE <<
"IBP" <<
RESET <<
" reduction @ " <<
now(
false) << flush;
286 std::map<ex, IBP*, ex_is_less> p2IBP;
288 int ntot = AIR.size();
289 for(
int i=0; i<ntot; i++) {
290 if(Verbose>0 && (((i+1)%1000)==0 || i+1==ntot)) {
291 cout <<
"\r \r" << flush;
292 cout <<
PRE <<
"\\--Prepare " <<
WHITE <<
"IBP" <<
RESET <<
" reduction [" << (i+1) <<
"/" << ntot <<
"] @ " <<
now(
false) << flush;
294 auto const & ir = AIR[i];
295 auto mat = ex_to<matrix>(ir.op(0));
296 auto vars = ex_to<lst>(ir.op(1));
298 int nrow = mat.rows();
300 for(
int c=0; c<mat.cols(); c++) {
302 for(
int r=0; r<nrow-2; r++) pc += mat(r,c) * vars.op(r);
305 ex nc = ex(0)-mat(nrow-1,c);
309 if(ncn==-1) den_tot++;
310 pns.append(lst{ ncn, pc, nc });
312 bool pn_sector =
false;
313 if(aio.pn_sector>0 && den_tot>=aio.pn_sector) pn_sector =
true;
315 for(
int i=0; i<pns.nops(); i++) pns[i] = lst{ pns.op(i).op(1), pns.op(i).op(2) };
319 for(
int i=0; i<pns.nops(); i++) pns[i] = lst{ pns.op(i).op(1), pns.op(i).op(2) };
322 int nCut = aio.Cut.nops();
325 cuts = cuts.subs(SP_map,nopat);
326 if(aio.CutFirst)
for(
auto cut : cuts) pns.prepend(lst{
SP2sp(cut), 1 });
327 else for(
auto cut : cuts) pns.append(lst{
SP2sp(cut), 1 });
331 for(
auto item : pns) {
332 props.append(item.op(0));
333 ns.append(item.op(1));
339 for(
int i=0; i<ns.nops(); i++) nss.append(ns.op(i)>0 ? 1 : 0);
340 key = lst{props,nss};
343 auto kv = p2IBP.find(key);
344 if(kv==p2IBP.end()) {
346 if(IBPmethod==0) ibp =
new IBP();
347 else if(IBPmethod==1) ibp =
new FIRE();
348 else if(IBPmethod==2) ibp =
new KIRA();
349 else if(IBPmethod==3) ibp =
new UKIRA();
355 p2IBP.insert(make_pair(key,ibp));
356 ibp->Propagator = props;
357 ibp->Internal = loops;
358 ibp->External = exts;
359 ibp->Replacement = repls;
360 if(aio.ISP.nops()>0)
for(
auto item : aio.ISP) ibp->ISP.append(
SP2sp(item));
361 if(aio.DSP.nops()>0) {
362 for(
auto item : aio.DSP) {
363 lst
sp = ex_to<lst>(item);
364 if(is_a<Vector>(
sp.op(0)))
sp[0] = (ex_to<Vector>(
sp.op(0)).name);
365 if(is_a<Vector>(
sp.op(1))) sp[1] = (ex_to<Vector>(
sp.op(1)).name);
371 for(
auto const & item : ns) {
372 if(item>0) sector.append(1);
373 else sector.append(0);
375 ibp->SECTOR = sector;
377 ibp->WorkingDir = wdir;
378 ibp->ProblemNumber =
pn;
381 if(aio.CutFirst)
for(
int i=0; i<nCut; i++) ibp->Cut.append(i+1);
382 else for(
int i=0; i<nCut; i++) ibp->Cut.append(nCut-i);
384 ibp_vec.push_back(ibp);
385 ibp->Integral.append(ns);
386 AIR2F[AIR[i]] = F(ibp->ProblemNumber, ns);
388 IBP* ibp = kv->second;
389 ibp->Integral.append(ns);
390 AIR2F[AIR[i]] = F(ibp->ProblemNumber, ns);
393 if(Verbose>0) cout << endl;
395 if(Verbose>0) cout <<
PRE <<
"\\--Total Ints/Pros: " <<
WHITE << ntot <<
"/" << ibp_vec.size() <<
RESET <<
" @ " <<
now(
false) << endl;
400 auto int_fr =
FindRules(ibp_vec,
false, aio.UF);
401 IntFs = int_fr.second;
404 air_vec =
GiNaC_Parallel(air_vec.size(), [&air_vec,&AIR2F,&int_fr] (
int idx) {
405 auto air = air_vec[idx];
406 air = air.subs(AIR2F,nopat);
407 air = air.subs(int_fr.first,nopat);
408 air = collect_ex(air, F(w1,w2));
411 if(aio.SaveDir !=
"") AIR2F_Save(aio.SaveDir, air_vec, IntFs, ibp_vec);
416 MapFunction _F2ex([&ibp_vec,aio](
const ex &e, MapFunction &self)->ex {
417 if(!e.has(F(w1,w2)))
return e;
418 else if(e.match(F(w1,w2))) {
419 int pn = ex_to<numeric>(e.op(0)).to_int();
420 auto pso = ex_to<lst>(ibp_vec[
pn-1]->Propagator);
421 auto nso = ex_to<lst>(e.op(1));
423 for(
int i=0; i<pso.nops(); i++) {
424 if(!aio.keep0F && nso.op(i).is_zero())
continue;
425 ps.append(pso.op(i));
426 ns.append(nso.op(i));
429 }
else return e.map(self);
433 air_vec =
GiNaC_Parallel(air_vec.size(), [&air_vec,&_F2ex](
int idx)->ex {
434 auto res = air_vec[idx];
437 for(
auto fp : ibp_vec) delete fp;
438 if(aio.SaveDir ==
"") rc = system((
"rm -rf "+wdir).c_str());
443 if(aio.SaveDir !=
"" &&
file_exists(aio.SaveDir+
"/Rules.gar")) {
447 vector<IBP*> ibp_vec_re;
449 map<int,lst> pn_ints_map;
450 for(
auto item : IntFs) {
451 int pn = ex_to<numeric>(item.op(0)).to_int();
452 pn_ints_map[
pn].append(item.op(1));
456 for(
auto pi : pn_ints_map) {
457 auto ibp = ibp_vec[pi.first-1];
458 ibp->Integral = pi.second;
459 nints += ibp->Integral.nops();
460 ibp_vec_re.push_back(ibp);
463 if(Verbose>0) cout <<
PRE <<
"\\--Refined Ints/Pros: " <<
WHITE << nints <<
"/" << ibp_vec_re.size() <<
RESET <<
" @ " <<
now(
false) << endl;
468 auto pRes =
GiNaC_Parallel(ibp_vec_re.size(), [&ibp_vec_re](
int idx)->ex {
469 ibp_vec_re[idx]->Export();
470 auto ret = lst{ ibp_vec_re[idx]->IsAlwaysZero ? 1 : 0, ibp_vec_re[idx]->Rules };
473 for(
int i=0; i<ibp_vec_re.size(); i++) {
474 ibp_vec_re[i]->IsAlwaysZero = (pRes[i].op(0)==1 ? true :
false);
475 ibp_vec_re[i]->Rules = ex_to<lst>(pRes[i].op(1));
478 int nproc = aio.NIBP;
479 if(nproc<1) nproc = 8;
481 if(nproc<1) nproc = 1;
482 size_t nibp = ibp_vec_re.size();
486 #pragma omp parallel for num_threads(nproc) schedule(dynamic, 1)
487 for(
int pi=0; pi<nibp; pi++) {
491 cout <<
"\r \r" << flush;
492 cout <<
PRE <<
"\\--" <<
WHITE <<
"FIRE" <<
RESET <<
" Reduction [" << (++cproc) <<
"/" << nibp <<
"] " << flush;
495 ibp_vec_re[pi]->Run();
497 if(Verbose>1 && nibp>0) cout <<
"@" <<
now(
false) << endl;
503 ibp_vec_re[idx]->Run();
507 for(
int pi=0; pi<nibp; pi++) {
508 if(Verbose>1) cout <<
"\r \r" <<
PRE <<
"\\--" <<
WHITE <<
"FIRE" <<
RESET <<
" Reduction [" << (++cproc) <<
"/" << nibp <<
"] " << flush;
509 ibp_vec_re[pi]->Run();
511 if(Verbose>1 && nibp>0) cout <<
"@" <<
now(
false) << endl;
515 if(ibp_vec_re.size()>100) {
516 auto ret =
GiNaC_Parallel(ibp_vec_re.size(), [&ibp_vec_re,wdir](
int idx)->ex {
517 ibp_vec_re[idx]->Import();
518 return ibp_vec_re[idx]->TO();
520 for(
int i=0; i<ibp_vec_re.size(); i++) ibp_vec_re[i]->FROM(ret[i]);
523 for(
auto item : ibp_vec_re) {
524 if(Verbose>1) cout <<
"\r \r" <<
PRE <<
"\\--" <<
WHITE <<
"FIRE" <<
RESET <<
" Import [" << (++cproc) <<
"/" << ibp_vec_re.size() <<
"] " << flush;
527 if(Verbose>1 && ibp_vec_re.size()>0) cout <<
"@" <<
now(
false) << endl;
531 if(aio.SaveDir ==
"") rc = system((
"rm -rf "+wdir).c_str());
532 }
else if(IBPmethod==2 || IBPmethod==3) {
533 for(
auto ibp : ibp_vec_re) ibp->Reduce();
534 if(aio.SaveDir ==
"") rc = system((
"rm -rf "+wdir).c_str());
538 exmap miRules =
FindRules(ibp_vec_re,
true, aio.UF).first;
540 if(aio.SaveDir !=
"") rc = system((
"mkdir -p "+aio.SaveDir+
"/Rules").c_str());
541 auto rules_vec =
GiNaC_Parallel(ibp_vec_re.size(), [&ibp_vec_re,&miRules,&aio](
int idx)->ex {
542 lst rules = ex_to<lst>(ibp_vec_re[idx]->Rules);
544 for(auto ri : rules) res.append(lst {
546 collect_ex(ri.op(1).subs(miRules,nopat),F(w1,w2),o_flint)
548 for(
auto mi : ibp_vec_re[idx]->MIntegral) {
549 auto fi = miRules.find(mi);
550 if(fi!=miRules.end()) res.append(lst{ mi, fi->second });
552 auto pn = ibp_vec_re[idx]->ProblemNumber;
553 if(aio.SaveDir !=
"") {
554 garWrite(aio.SaveDir+
"/Rules/"+to_string(
pn)+
".gar", res);
558 if(aio.SaveDir !=
"") {
559 garWrite(aio.SaveDir+
"/Rules.gar", 1);
561 for(
auto rs : rules_vec) {
562 for(
auto ri : rs) if(ri.op(0)!=ri.op(1)) ibpRules[ri.op(0)] = ri.op(1);
571 GiNaC_Parallel(air_vec.size(), [&air_vec,&ibpRules,&_F2ex,&aio](
int idx)->ex {
572 ex res = air_vec[idx];
574 if(aio.SaveDir !=
"") {
576 find(res, F(w1,w2), fs);
578 for(auto fi : fs) pns.insert(fi.op(0));
580 auto rs = ex_to<lst>(garRead(aio.SaveDir+
"/Rules/"+ex2str(pn)+
".gar"));
581 for(auto ri : rs) if(ri.op(0)!=ri.op(1)) rules[ri.op(0)] = ri.op(1);
583 }
else rules = ibpRules;
584 res = res.subs(rules,nopat);
585 if(aio.pat.nops()>0) {
591 if(aio.cv!=
nullptr) {
592 auto _cv = aio.cv(c,v);
602 for(
auto fp : ibp_vec) delete fp;