Skip to content

Commit 01473be

Browse files
committed
Update Cone.jl
1 parent 430ed4d commit 01473be

File tree

1 file changed

+6
-7
lines changed

1 file changed

+6
-7
lines changed

src/Cone/Cone.jl

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -91,8 +91,7 @@ checkpoints(::Conic) = conicpolar.(checkpoints(rectspace(DuffyConic())))
9191

9292
function plan_transform(S::DuffyConic, v::AbstractVector{T}) where T
9393
n = length(v)
94-
N = (1 + isqrt(1+8n)) ÷ 4
95-
M = 2N-1
94+
N,M = pointsize(S,length(v))
9695
D = plan_transform!(rectspace(S), Array{T}(undef,N,M))
9796
TransformPlan(S, D, Val{false})
9897
end
@@ -182,8 +181,7 @@ end
182181

183182
function LegendreConicTransformPlan(S::LegendreConic, v::AbstractVector{T}) where T
184183
n = length(v)
185-
N = (1 + isqrt(1+8n)) ÷ 4
186-
M = 2N-1
184+
N,M = pointsize(Conic(), n)
187185
D = plan_transform!(rectspace(DuffyConic()), Array{T}(undef,N,M))
188186
P = c_plan_rottriangle(N, zero(T), zero(T), zero(T))
189187
LegendreConicTransformPlan(D,P)
@@ -428,9 +426,10 @@ struct LegendreConeTransformPlan{DUF,CHEB}
428426
end
429427

430428
function LegendreConeTransformPlan(S::LegendreCone, v::AbstractVector{T}) where T
431-
D = plan_transform(DuffyCone(),v)
432-
F = totensor(rectspace(DuffyCone()), D*v) # wasteful, just use to figure out `size(F,1)`
433-
P = c_plan_rottriangle(size(F,1), zero(T), zero(T), one(T))
429+
n = length(v)
430+
N,M = pointsize(Cone(), n)
431+
D = plan_transform!(rectspace(DuffyCone()), Array{T}(undef,N,M))
432+
P = c_plan_rottriangle(N, zero(T), zero(T), zero(T))
434433
LegendreConeTransformPlan(D,P)
435434
end
436435

0 commit comments

Comments
 (0)