work on cone

Author: dlfivefifty

src/Cone/Cone.jl

@@ -175,14 +175,33 @@ tensorizer(K::LegendreCone) = BallTensorizer()
 
 rectspace(::DuffyCone) = NormalizedJacobi(0,1,Segment(1,0))*ZernikeDisk()
 
-
+function points(sp::Cone,n)
+    pts=Array{float(eltype(domain(sp)))}(undef,0)
+    a,b = sp.spaces
+    if isfinite(dimension(a)) && isfinite(dimension(b))
+        N,M=dimension(a),dimension(b)
+    elseif isfinite(dimension(a))
+        N=dimension(a)
+        M=n÷N
+    elseif isfinite(dimension(b))
+        M=dimension(b)
+        N=n÷M
+    else
+        N=M=round(Int,sqrt(n))
+    end
+
+    for y in points(b,M), x in points(a,N)
+        push!(pts,Vec(x...,y...))
+    end
+    pts
+end
 
 
 points(::Cone, n) = conemap.(points(rectspace(DuffyCone()), n))
 checkpoints(::Cone) = conemap.(checkpoints(rectspace(DuffyCone())))
 
 
-function plan_transform(S::DuffyCone, n::AbstractVector) 
+function plan_transform(sp::DuffyCone, n::AbstractVector) 
     rs = rectspace(S)
     P = TransformPlan(sp,((plan_transform(sp.spaces[1],T,N),N), (plan_transform(sp.spaces[2],T,M),M)), Val{false})
     TransformPlan(S, P, Val{false})

src/Disk/Disk.jl

@@ -42,7 +42,8 @@ DiskTensorizer() = Tensorizer((Ones{Int}(∞),Ones{Int}(∞)))
 tensorizer(K::ChebyshevDisk) = DiskTensorizer()
 
 # we have each polynomial
-blocklengths(K::ChebyshevDisk) = 1:∞
+blocklengths(K::ChebyshevDisk) = Base.OneTo(∞)
+blocklengths(K::ZernikeDisk) = Base.OneTo(∞)
 
 for OP in (:block,:blockstart,:blockstop)
     @eval begin

test/test_cone.jl

@@ -1,4 +1,5 @@
 using ApproxFun, MultivariateOrthogonalPolynomials, StaticArrays, Test
+import ApproxFunBase: checkpoints
 import MultivariateOrthogonalPolynomials: rectspace, totensor, duffy2legendreconic!, legendre2duffyconic!, c_plan_rottriangle, plan_transform
 
 @testset "Conic" begin

test/test_disk.jl

@@ -1,7 +1,7 @@
 using Revise
-using  ApproxFun, MultivariateOrthogonalPolynomials
+using  ApproxFun, MultivariateOrthogonalPolynomials, InfiniteArrays
 import MultivariateOrthogonalPolynomials: checkerboard, icheckerboard, CDisk2CxfPlan
-import ApproxFunBase: totensor
+import ApproxFunBase: totensor, blocklengths
 import ApproxFunOrthogonalPolynomials: jacobip
 
 
@@ -112,4 +112,6 @@ end
     @test f(0.1,0.2) ≈ 1.0
     f = Fun((x,y) -> exp(x*cos(y-0.1)), ZernikeDisk())
     @test f(0.1,0.2) ≈ exp(0.1*cos(0.1))
+
+    @test blocklengths(ZernikeDisk()) == Base.OneTo(∞)
 end