add recipes for features and feature collections (#72)

Author: rafaqz

GeoInterfaceRecipes/src/GeoInterfaceRecipes.jl

@@ -19,15 +19,15 @@ macro enable_geo_plots(typ)
               @nospecialize
               series_list = RecipesBase.RecipeData[]
               RecipesBase.is_explicit(plotattributes, :label) || (plotattributes[:label] = :none)
-              Base.push!(series_list, RecipesBase.RecipeData(plotattributes, (GeoInterface.geomtrait(geom), geom)))
+              Base.push!(series_list, RecipesBase.RecipeData(plotattributes, (GeoInterface.trait(geom), geom)))
               return series_list
         end
         function RecipesBase.apply_recipe(plotattributes::Base.AbstractDict{Base.Symbol, Base.Any}, geom::Base.AbstractVector{<:Base.Union{Base.Missing,<:($(esc(typ)))}})
               @nospecialize
               series_list = RecipesBase.RecipeData[]
               RecipesBase.is_explicit(plotattributes, :label) || (plotattributes[:label] = :none)
               for g in Base.skipmissing(geom)
-                  Base.push!(series_list, RecipesBase.RecipeData(plotattributes, (GeoInterface.geomtrait(g), g)))
+                  Base.push!(series_list, RecipesBase.RecipeData(plotattributes, (GeoInterface.trait(g), g)))
               end
               return series_list
         end
@@ -51,6 +51,11 @@ end
 
 RecipesBase.@recipe f(::GI.GeometryCollectionTrait, collection) = collect(getgeom(collection))
 
+# Features
+RecipesBase.@recipe f(t::GI.FeatureTrait, feature) = GI.geometry(feature)
+
+RecipesBase.@recipe f(t::GI.FeatureCollectionTrait, fc) = collect(GI.getfeature(fc))
+
 # Convert coordinates to the form used by Plots.jl
 _coordvecs(::GI.PointTrait, geom) = [tuple(GI.coordinates(geom)...)]
 function _coordvecs(::GI.MultiPointTrait, geom)
@@ -126,6 +131,7 @@ function _coordvecs(::GI.MultiPolygonTrait, geom)
     end
 end
 
+
 _coordvec(n) = Array{Float64}(undef, n)
 
 function _geom2coordvecs!(xs, ys, geom)

GeoInterfaceRecipes/test/runtests.jl

@@ -14,8 +14,12 @@ struct MyMultiPoint{N} <: MyAbstractGeom{N} end
 struct MyMultiCurve{N} <: MyAbstractGeom{N} end
 struct MyMultiPolygon{N} <: MyAbstractGeom{N} end
 struct MyCollection{N} <: MyAbstractGeom{N} end
+struct MyFeature end
+struct MyFeatureCollection end
 
 GeoInterfaceRecipes.@enable_geo_plots MyAbstractGeom
+GeoInterfaceRecipes.@enable_geo_plots MyFeature
+GeoInterfaceRecipes.@enable_geo_plots MyFeatureCollection
 
 GeoInterface.isgeometry(::MyAbstractGeom) = true
 GeoInterface.is3d(::GeoInterface.AbstractGeometryTrait, ::MyAbstractGeom{N}) where {N} = N == 3
@@ -54,6 +58,13 @@ GeoInterface.ncoord(::GeoInterface.GeometryCollectionTrait, geom::MyCollection{N
 GeoInterface.ngeom(::GeoInterface.GeometryCollectionTrait, geom::MyCollection) = 4
 GeoInterface.getgeom(::GeoInterface.GeometryCollectionTrait, geom::MyCollection{N}, i) where {N} = MyMultiPolygon{N}()
 
+GeoInterface.trait(::MyFeature) = FeatureTrait()
+GeoInterface.trait(::MyFeatureCollection) = FeatureCollectionTrait()
+GeoInterface.getfeature(::GeoInterface.FeatureCollectionTrait, geom::MyFeatureCollection, i) = MyFeature()
+GeoInterface.getfeature(::GeoInterface.FeatureCollectionTrait, geom::MyFeatureCollection) = [MyFeature(), MyFeature()]
+GeoInterface.geometry(geom::MyFeature) = rand((MyPolygon{2}(), MyMultiPolygon{2}()))
+GeoInterface.nfeature(::GeoInterface.FeatureTrait, geom::MyFeature) = 1
+
 @testset "plot" begin
     # We just check if they actually run
     # 2d
@@ -72,4 +83,6 @@ GeoInterface.getgeom(::GeoInterface.GeometryCollectionTrait, geom::MyCollection{
     plot(MyPolygon{3}())
     plot(MyMultiPolygon{3}())
     plot(MyCollection{3}())
+    plot(MyFeature())
+    plot(MyFeatureCollection())
 end