Multiplication of Fun{<:ContinuousSpace} (#86)

* improve inference in PiecewiseSpace

* undo type assertion

* version bump to v0.4.14

Author: jishnub

Project.toml

@@ -1,6 +1,6 @@
 name = "ApproxFunOrthogonalPolynomials"
 uuid = "b70543e2-c0d9-56b8-a290-0d4d6d4de211"
-version = "0.4.13"
+version = "0.4.14"
 
 [deps]
 AbstractFFTs = "621f4979-c628-5d54-868e-fcf4e3e8185c"

src/Spaces/Ultraspherical/ContinuousSpace.jl

@@ -94,7 +94,7 @@ function *(P::TransformPlan{T,SS,false},vals::AbstractVector{T}) where {T,SS<:Co
     end
 end
 
-components(S::ContinuousSpace) = map(ChebyshevDirichlet{1,1},components(domain(S)))
+components(S::ContinuousSpace) = map(ChebyshevDirichlet{1,1}, components(domain(S)))
 canonicalspace(S::ContinuousSpace) = PiecewiseSpace(components(S))
 convert(::Type{PiecewiseSpace}, S::ContinuousSpace) = canonicalspace(S)
 
@@ -105,11 +105,11 @@ block(C::ContinuousSpace,k) = Block((k-1)÷ncomponents(C.domain)+1)
 
 ## components
 
-components(f::Fun{CS},j::Integer) where {CS<:ContinuousSpace} = components(Fun(f,canonicalspace(f)),j)
-components(f::Fun{CS}) where {CS<:ContinuousSpace} = components(Fun(f,canonicalspace(space(f))))
+components(f::Fun{<:ContinuousSpace},j::Integer) = components(Fun(f,canonicalspace(f)),j)
+components(f::Fun{<:ContinuousSpace}) = components(Fun(f,canonicalspace(space(f))))
 
 
-function points(f::Fun{CS}) where {CS<:ContinuousSpace}
+function points(f::Fun{<:ContinuousSpace})
     n=ncoefficients(f)
     d=domain(f)
     K=ncomponents(d)

src/specialfunctions.jl

@@ -7,6 +7,10 @@
 ^(f::Fun{S},c::Integer) where {S<:ContinuousSpace} = Fun(map(f->f^c,components(f)),PiecewiseSpace)
 ^(f::Fun{S},c::Number) where {S<:ContinuousSpace} = Fun(map(f->f^c,components(f)),PiecewiseSpace)
 
+function *(f::Fun{<:ContinuousSpace}, g::Fun{<:ContinuousSpace})
+    @assert components(domain(f)) == components(domain(g))
+    Fun(map(*, components(f), components(g)), PiecewiseSpace)
+end
 
 ^(f::Fun{<:PolynomialSpace},k::Integer) = intpow(f,k)
 

test/SpacesTest.jl

@@ -60,6 +60,8 @@ import ApproxFunBase: space, SpaceOperator,
         c=Fun(cos,-2..2)|>abs
         sc=Fun(x -> abs(sin(x))+abs(cos(x)), ContinuousSpace(PiecewiseSegment([-2,(-π/2),0,π/2,2])))
         @test norm(sc-(c+s))<100eps()
+
+        @test sc * sc == sc^2
     end
 
     @testset "max/min creates breakpoints" begin