fix: small typo and syntax fixes

astro-group-bristol · Sep 9, 2023 · 3484e07 · 3484e07
1 parent 03d4a5f
commit 3484e07
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/src/transfer-functions/transfer-functions-2d.jl b/src/transfer-functions/transfer-functions-2d.jl
@@ -183,6 +183,7 @@ function binflux(
     profile::AbstractDiscProfile;
     redshift = ConstPointFunctions.redshift(tf.emissivity_profile.metric, tf.x),
     E₀ = 6.4,
+    t0 = tf.x[2],
     kwargs...,
 )
     t = coordtime_at(profile, tf.observer_to_disc)
@@ -195,7 +196,7 @@ function binflux(
 
     tb, eb, td = bin_transfer_function(t, g * E₀, F; kwargs...)
     # subtract initial time
-    tb .- tf.x[2], eb, td
+    tb .- t0, eb, td
 end
 
 export bin_transfer_function, lagtransfer, binflux
diff --git a/src/utils.jl b/src/utils.jl
@@ -124,12 +124,10 @@ Lz(m::AbstractMetric, u, v) = Lz(metric(m, u), v)
 Lz(m::AbstractMetric, gp::AbstractGeodesicPoint) = Lz(m, gp.x, gp.v)
 
 _equatorial_project(r, θ) = r * sin(abs(θ))
-_equatorial_project(x::SVector{4}) = _equatorial_project(x[2], x[3])
-_equatorial_project(x::SVector{8}) = _equatorial_project(x[2], x[3])
+_equatorial_project(x::SVector) = _equatorial_project(x[2], x[3])
 
 _spinaxis_project(r, θ) = r * cos(abs(θ))
-_spinaxis_project(x::SVector{4}) = _spinaxis_project(x[2], x[3])
-_spinaxis_project(x::SVector{8}) = _spinaxis_project(x[2], x[3])
+_spinaxis_project(x::SVector) = _spinaxis_project(x[2], x[3])
 
 _rotate_about_spinaxis(n::SVector{3}, ϕ) = SVector(n[1] * cos(ϕ), n[1] * sin(ϕ), n[3])