arrayred = thetarun[rank * sizered:(rank + 1) * sizered]
freepar = Grid.freepar
print("lenrun, sizered", lenrun, sizered)
allPlin = []
allPloop = []
for i, theta in enumerate(arrayred):
parameters = copy.deepcopy(Grid.parref)
truetheta = Grid.valueref + theta * Grid.delta
idx = nrun * lenrun + rank * sizered + i
print("nrun, rank, i", nrun, rank, i)
parameters["PathToOutput"] = os.path.join(
OUTPATH, 'output' + str(nrun) + str(rank) + str(i))
for k, var in enumerate(freepar):
parameters[var] = truetheta[k]
# parameters['h'] = 1/parameters['invh']
Plin, Ploop = Grid.CompPterms(parameters)
idxcol = np.full([Plin.shape[0], 1], idx)
allPlin.append(np.hstack([Plin, idxcol]))
allPloop.append(np.hstack([Ploop, idxcol]))
if (i == 0) or ((i + 1) % 100 == 0):
print("theta check: ", Grid.flattenedgrid[idx], theta, truetheta)
# np.save(os.path.join(outpk, "temp", "Plin_run%s_rank%si%s.npy" % (str(nrun), str(rank), str(i))), np.array(allPlin))
# np.save(os.path.join(outpk, "temp", "Ploop_run%s_rank%si%s.npy" % (str(nrun), str(rank), str(i))), np.array(allPloop))
np.save(
os.path.join(outpk, "Plin_run%s_rank%s.npy" % (str(nrun), str(rank))),
np.array(allPlin))
np.save(
os.path.join(outpk, "Ploop_run%s_rank%s.npy" % (str(nrun), str(rank))),
np.array(allPloop))
allPlin = []
allPloop = []
for i, theta in enumerate(arrayred):
parameters = copy.deepcopy(Grid.parref)
truetheta = Grid.valueref + theta * Grid.delta
#idx = nrun * lenrun + rank * sizered + i
#print("nrun, rank, i", nrun, rank, i)
idx = i
print ("i on tot", i, sizearray)
parameters["PathToOutput"] = os.path.join(OUTPATH, 'output' + str(nrun) + str(rank) + str(i))
for k, var in enumerate(freepar):
parameters[var] = truetheta[k]
# parameters['h'] = 1/parameters['invh']
kin, Plin, z, Omega_m = Grid.CompPterms(parameters)
bird = pybird.Bird(kin, Plin, Omega_m, z, full=False)
nonlinear.PsCf(bird)
bird.setPsCfl()
resum.Ps(bird, full=False)
bird.subtractShotNoise()
Plin, Ploop = bird.formatTaylor()
idxcol = np.full([Plin.shape[0], 1], idx)
allPlin.append(np.hstack([Plin, idxcol]))
allPloop.append(np.hstack([Ploop, idxcol]))
if (i == 0) or ((i+1) % 100 == 0):
print("theta check: ", Grid.flattenedgrid[idx], theta, truetheta)
# np.save(os.path.join(outpk, "temp", "Plin_run%s_rank%si%s.npy" % (str(nrun), str(rank), str(i))), np.array(allPlin))
# np.save(os.path.join(outpk, "temp", "Ploop_run%s_rank%si%s.npy" % (str(nrun), str(rank), str(i))), np.array(allPloop))
np.save(os.path.join(outpk, "Plin_run%s_rank%s.npy" % (str(nrun), str(rank))), np.array(allPlin))
