From 6bd52e611451f1ce0c6cb4803ac0d310f0f74d0e Mon Sep 17 00:00:00 2001
From: jaimerzp <jaimerz011235813@gmail.com>
Date: Thu, 26 Dec 2024 12:55:26 +0100
Subject: [PATCH] w_type

---
 src/tracers.jl       |  36 ++++++++++++++++++------------------
 test/test_output.npz | Bin 69358 -> 69358 bytes
 2 files changed, 18 insertions(+), 18 deletions(-)

diff --git a/src/tracers.jl b/src/tracers.jl
index 2504134..0bf332e 100755
--- a/src/tracers.jl
+++ b/src/tracers.jl
@@ -70,7 +70,8 @@ WeakLensingTracer(cosmo::Cosmology, z, nz;
     chi = cosmo.chi(z)
     # Compute kernels
     w_itg(chii) = @.(nz*(1-chii/chi))
-    w_arr = zeros(cosmo_type, res)
+    w_type = typeof(integrate(z[2:res], w_itg(chi[2])[2:res], SimpsonEven()))
+    w_arr = zeros(w_type, res)
     @inbounds for i in 1:res-3
         w_arr[i] = integrate(z[i:res], w_itg(chi[i])[i:res], SimpsonEven())
     end
@@ -81,7 +82,7 @@ WeakLensingTracer(cosmo::Cosmology, z, nz;
     w_arr = @. w_arr * chi * lens_prefac * (1+z) / nz_norm
     # IA correction
     hz = Hmpc(cosmo, z)
-    As = @. A_IA*((1 + z)/1.62)^alpha_IA * (0.0134 * cosmo.cpar.Ωm / cosmo.Dz(z))
+    As = get_IA(cosmo, z, A_IA, alpha_IA)
     corr =  @. As * (nz * hz / nz_norm)
     w_arr_ia = @. w_arr - corr
     # Interpolate
@@ -91,6 +92,21 @@ WeakLensingTracer(cosmo::Cosmology, z, nz;
     WeakLensingTracer(wint, F)
 end
 
+"""
+    get_IA(cosmo::Cosmology, zs, IA_params)
+CMB lensing tracer structure. 
+Arguments:
+- `cosmo::Cosmology` : cosmology structure.
+- `zs::Vector{Dual}` : redshift array.
+- `A_IA::Dual` : Intrinsic aligment amplitud.
+- `alpha_IA::Dual` : Intrinsic aligment slope.
+Returns:
+- `IA_corr::Vector{Dual}` : Intrinsic aligment correction to radial kernel.
+"""
+function get_IA(cosmo::Cosmology, zs, A_IA, alpha_IA)
+    return @. A_IA*((1 + zs)/1.62)^alpha_IA * (0.0134 * cosmo.cpar.Ωm / cosmo.Dz(zs))
+end
+
 """
     CMBLensingTracer(warr, chis, wint, lpre)
 CMB lensing tracer structure. 
@@ -128,22 +144,6 @@ CMBLensingTracer(cosmo::Cosmology; res=350) = begin
     CMBLensingTracer(wint, F)
 end
 
-"""
-    get_IA(cosmo::Cosmology, zs, IA_params)
-CMB lensing tracer structure. 
-Arguments:
-- `cosmo::Cosmology` : cosmology structure.
-- `zs::Vector{Dual}` : redshift array.
-- `IA_params::Vector{Dual}` : Intrinsic aligment parameters.
-Returns:
-- `IA_corr::Vector{Dual}` : Intrinsic aligment correction to radial kernel.
-"""
-function get_IA(cosmo::Cosmology, zs, IA_params)
-    A_IA = IA_params[1]
-    alpha_IA = IA_params[2]
-    return @. A_IA*((1 + zs)/1.62)^alpha_IA * (0.0134 * cosmo.cpar.Ωm / cosmo.Dz(zs))
-end
-
 function nz_interpolate(z, nz, res; mode="linear")
     if mode!="none"
         if mode=="linear"
diff --git a/test/test_output.npz b/test/test_output.npz
index b99ebc48e671daac05881286747637b536e7e09b..b96378cf7084f00a5534a93cdfed3a996d9dd67a 100644
GIT binary patch
delta 1887
zcmXw3YfMyE5O&zREDO6^D+=Wer2^7z2oRB{JfwhBt4(VYv85`l+7?CpflXtorD!Bn
zBHSz|P{Bt^(E6b438(~tg(3yT3KeYA_=qOPM8w#J7}!_uojL5EnLFn@Gv9nOXJ?wF
znP$nj@jZ>SbdmadvZJZPH~<>O$`jC}T3+Tc0A7l-d3ya<1#(BXFJC9MBZRxO{p8qm
zHgXQ@uacAPGp@($%a-+EM84y98Ux?>3v7Bo7oE%owqnKRpjIpt>ihX6d24bA-x++8
z5c<$_i0Bk{Ti7Z5mclJ)PlTUEn;&ro?Q{fXTDR-dztl|f=+Gwa6In=&)IGh=<?JH?
zyyC61ls_Vh@Q->j9w;pa%B60Te4%I~&x<ZZ1earE4W?dU#**A-o}dOBl1MPVC0R5U
zmde8rJGmKY@k2$|Da+bVE>l5G@o5#o+lqgulYLuz!?5|-Ek)1dIIBP3aHvv)k?N7M
zpXGOM2eKC)py!SsnkYr%Vbp-9AED&3_UK8h+IE!0DXeb58jq{%i2D9xq+(<(O}4*Q
z@T|B*vIVshtG;%ct6bTEli{(}7D{oWlaPD6NbsP`^xZG)7gxT+noq6_QePLkhly@Q
z577;{Mk>kodXWvU?h}FawidiQV-o@EAkvXP{*8*k5KXD(%h*kF`EW2F9isi<rNe@7
z+{jO;!Pp3Cuy<7Oyg{k@ge|#A=bmFEiIprbzD0C-%x&=uo{wXPynaHc7&S#H$ES$y
zzPr6RZJzk*8tvCVn%qfCJow3`xoBjQJeP`-jJ&81!h_>;(HKyl6nkk<pp$cEgi7;S
zc=D*|??tR!ubgb3kHV9h`4qIn^YLiy^XX_)?J0O;iRIr7bfeBrH@fXfi1^Q*fi}^R
zfk^dz5wg}1g^`l$yQk`%e)5EaZrUB$cq7x9jP_gSO2jp8w;I_?mDQMYIYu4_=Wb*1
zS|zgBMN6OHqAKd!qbkQ-l(oy1huPdN;*;sl$M0%)BHBLpO0=)s397Z2AqQ<aAR9|3
zq+w0(`X2+41v3FGS0OA30CTg00ILvmH6gZ9&b0fnI{=}GnFT07C(8p?qw~GuAXt53
z>R{MC51`3t-Z=1(#d|5CH#HD~=p2oxQKJzx_N4uCia-wqs6hhMoTW&BV`94{V$V7x
zNX0xe5~QJH)&lLt@tB)B<XWW_TwAq}g_x&WK(Sbir{H@3z(5Wv&Z<0t{xhWhXPgAF
zc>>Ml^@tq`e`PLA(h1b6mZf17%2q;{y+kMIp3uP>RO0;3&wOzsnNtV2lFaM{$#`}#
z>DGRnD68}W94R~RCAxdD;3Vor%2K^J`c)6Oxolc5nu_xlO_h0zh;_l4^%g_^aP8Z#
zk+$#qhym>J5$Fydk@30WAhkhTg(8YNu3Q6zU?|An`G>O8?4SW?$sd$`y+REPDgo?%
z+5q_o{V3cJiR1skD8^rPw{3*9YA_1<YL^t`vDK1N;!h*^Q3B&aLnhJmE|XZTb0&EA
Ee_VewjQ{`u

delta 1887
zcmXw3YfMyE5O&zREDO6^D+=YgAQg~qLx6}p<xvVqrP{PM5nHO#R@<V8X|QQbwG@p6
zCBhAyU<I`)LF<FEC!i7p7K#)UD^#$J@exfL6A@z@V%TMOd+*F)|IFMu-<kR5n>jnv
zEX_1a#trXiq@|10-`&EmJB$OMQLH=x$5qSAJO;o^aW+jCewi<Kbo=skQaeJpOWRM5
zO=lzLp#BOuSw7=>yuNH%4@TtMey1_;jlaOA1$5ELY+x%^Yz}J0LZQB&UX-^ah4Agc
zCkdesJ&TA=VYix3;kOjlfc8Y#8MJxfKcSrtr%da1eEgT1Ngf^A$bBLTsFAuS_qm*P
zD1cYIeunagM-u)~Px=F;#Xz~#ZIaIyZQ!|41&H8sw5-9@D=k=(+bk2*U}GW)#y2O4
z#>^?)jM&LdNQ)mRx<*;nesqZnYKl**2;NrwJDu#=(rd=%V>TB(m1C{`eEoq+4MwU*
z%6^vLycNivdw`zL|8SgAG#x|@c-kRKE^7~;#HwwFNt}Y}My&C;x{j#tJxVG@)Y4@8
zYX#4WizHi6JF)6(r@6|N9XJ^tV{M@n*E<QhyNd)5x<udo!hUx7JFNN7<w5G}eD^TX
zt>_`T0ar;S`Cc!w;njU2P-ttxyE8Tsunr;}`NQ9+7!1*rYQBisD3=ch^U)#N4_-Pf
z2*-~6h#HKIkOsR)1<&i0s!!OG8+7hDMv_>`^5UCBmq*_c&*0fOcF5}|go=?<lyZEE
z=<d1Gi__)_FR#*m-PPnwHpGFSY?_NgHc4|SILU|$`XD?wHW!5f^+~Z81_e4fXF;fB
z&%~2QMSm}1<$C30n>`XwYV65qhwX7_9riS|DUM{kv83VO^>m}oK{vV`iHP{mk&ZTD
zJ{^(j`yyoRd?ZFnuI-$vclyZ_^K{cOpM^IvoJnZEb*@5O(>AM-y--<=Iagrhad6I7
z7N=Dri(Rzz@h+;OzCEgP)J0jlT)CLd?IJ!I?mYaib|;|ibFV`C(jBkb1`A}PEeB-N
z&<Uwn)4TpiAhKX4fE6l)B?4e>wjW?Mg095JG|3qbKXw}+6fv^^`RHVNz#4SER~!VZ
zPfQ&QyXOHk8O>|^9<n$uCA3@vA&Ab_h#EB-QDax?FQ*9fP=FdGK+RdQ1UM$PQzG`P
zLxL2{Gb2GNIu<R^UL1?Qp+l}!TEVqd3z>*{q6HL-MSBXa_xBBCqvEW}6X-uf>VL{f
z5Su5^TwV?TT;Z?Gg-JSrTHUhLOrb0#gxO1Ug6;_&yoE}f+y03!ZX|Q+09TS(ydVkB
zE+pREixXv)UVtNI=e$IB&lj8oy+~QA7e~M90XLUT>qS$s-lC~8ZxOLBIJ4ej$nURx
z;}z2OT^})kT|NTc;Uh9WQyio=Xsb{}QOA{IfDjA?`8)qmcAD)s04@20vaeREfk7pJ
z-A@}J51}7~86t4}?-|ATtM0UokXDUGAz$s1{9Lw1QcC=31V2h(TxiH7n%-d&t98}{
GZ~qUIIRUKz