Advertisement
Guest User

Untitled

a guest
Jun 26th, 2019
91
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 2.15 KB | None | 0 0
  1. import multiprocessing as mp
  2.  
  3. def MAIN_LOOP(lll):
  4. global aa
  5. global bb
  6. aa, bb = 0,0
  7. if paramo == 3:
  8. C_ij = np.zeros((len(zrange), len(zrange)))
  9. C_ij_up = np.zeros((len(zrange), len(zrange)))
  10. C_ij_dw = np.zeros((len(zrange), len(zrange)))
  11. while aa < len(zrange):
  12. while bb <= aa:
  13. if paramo == 3:
  14. C_ij[aa][bb] = Pobs_C(zpm, zrange[aa], zrange[bb], h[2], Omega_m[2], Omega_DE[2], w0[2], wa[2], E_tab, R_tab, DG_tab, W_tab, l[lll], P_dd_C, R_tab(z_pk))
  15. C_ij_up[aa][bb] = Pobs_C(zpm, zrange[aa], zrange[bb], h[0], Omega_m[0], Omega_DE[0], w0[0], wa[0], E_tab_up, R_tab_up, DG_tab_up, W_tab_up, l[lll], P_dd_C_up, R_tab_up(z_pk))
  16. C_ij_dw[aa][bb] = Pobs_C(zpm, zrange[aa], zrange[bb], h[3], Omega_m[3], Omega_DE[3], w0[3], wa[3], E_tab_dw, R_tab_dw, DG_tab_dw, W_tab_dw, l[lll], P_dd_C_dw, R_tab_dw(z_pk))
  17.  
  18. if aa != bb:
  19. if paramo == 3:
  20. C_ij[bb][aa] = C_ij[aa][bb]
  21. C_ij_up[bb][aa] = C_ij_up[aa][bb]
  22. C_ij_dw[bb][aa] = C_ij_dw[aa][bb]
  23. bb=bb+1
  24. bb=0
  25. aa=aa+1
  26.  
  27. if paramo == 3:
  28. aa, bb = 0,0
  29. outF=open(CC_path[2]+"/COVAR_fid_"+str(l[lll]),'w')
  30. while aa < len(C_ij):
  31. while bb < len(C_ij):
  32. outF.write(str("%.16e" % C_ij[aa][bb]))
  33. outF.write(str(' '))
  34. bb=bb+1
  35. outF.write(str('n'))
  36. bb=0
  37. aa=aa+1
  38. outF.close()
  39.  
  40. aa, bb = 0,0
  41. outU=open(CC_path[0]+"/COVAR_up_"+str(l[lll]),'w')
  42. outD=open(CC_path[3]+"/COVAR_dw_"+str(l[lll]),'w')
  43. while aa < len(C_ij_up):
  44. while bb < len(C_ij_up):
  45. outU.write(str("%.16e" % C_ij_up[aa][bb]))
  46. outU.write(str(' '))
  47. outD.write(str("%.16e" % C_ij_dw[aa][bb]))
  48. outD.write(str(' '))
  49. bb=bb+1
  50. outU.write(str('n'))
  51. outD.write(str('n'))
  52. bb=0
  53. aa=aa+1
  54. outU.close()
  55. outD.close()
  56. lll=lll+1
  57.  
  58. lll = range(len(l))
  59.  
  60. if __name__ == '__main__':
  61. pool = mp.Pool(12)
  62. pool.map(MAIN_LOOP, lll)
  63. pool.close()
  64. pool.join()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement