vectorpipe.util.JsonRobustFeatureCollectionMap.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of vectorpipe_2.11 Show documentation

Import OSM data and output to VectorTiles with GeoTrellis.

The newest version!

package vectorpipe.util

import cats.syntax.either._
import geotrellis.vector._
import _root_.io.circe._
import _root_.io.circe.syntax._

import scala.collection.mutable
import scala.reflect.ClassTag

class JsonRobustFeatureCollectionMap(features: List[Json] = Nil) {
  private val buffer = mutable.ListBuffer(features:_*)

  def add[G <: Geometry: ClassTag, D: Encoder](featureMap: (String, RobustFeature[G, D])) =
    buffer += RobustFeatureFormats.writeRobustFeatureJsonWithID(featureMap)

  def asJson: Json = {
    val bboxOption = getAll[Geometry].map(_._2.extent).reduceOption(_ combine _)
    bboxOption match {
      case Some(bbox) =>
        Json.obj(
          "type" -> "FeatureCollection".asJson,
          "bbox" -> Extent.listEncoder(bbox),
          "features" -> buffer.toVector.asJson
        )
      case _ =>
        Json.obj(
          "type" -> "FeatureCollection".asJson,
          "features" -> buffer.toVector.asJson
        )
    }
  }

  private def getFeatureID(js: Json): String = {
    val c = js.hcursor
    val id = c.downField("id")
    id.as[String] match {
      case Right(i) => i
      case _ =>
        id.as[Int] match {
          case Right(i) => i.toString
          case _ => throw DecodingFailure("Feature expected to have \"ID\" field", c.history)
        }
    }
  }

  def getAll[F: Decoder]: Map[String, F] = {
    var ret = Map[String, F]()
    features.foreach{ f => f.as[F].foreach(ret += getFeatureID(f) -> _) }
    ret
  }
}

object JsonRobustFeatureCollectionMap {
  def apply() = new JsonRobustFeatureCollectionMap()

  def apply[G <: Geometry: ClassTag, D: Encoder](features: Traversable[(String, RobustFeature[G, D])]) = {
    val fc = new JsonRobustFeatureCollectionMap()
    features.foreach(fc.add(_))
    fc
  }

  def apply(features: Traversable[Json])(implicit d: DummyImplicit): JsonRobustFeatureCollectionMap =
    new JsonRobustFeatureCollectionMap(features.toList)
}