geotrellis.server.ogc.wmts.WmtsModel.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of geotrellis-server-ogc_2.12 Show documentation

GeoTrellis Server is a set of components designed to simplify viewing, processing, and serving raster data from arbitrary sources with an emphasis on doing so in a functional style.

The newest version!

/*
 * Copyright 2020 Azavea
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package geotrellis.server.ogc.wmts

import geotrellis.server.ogc._
import geotrellis.server.ogc.style._
import geotrellis.server.ogc.wmts.WmtsParams.GetTile
import geotrellis.layer._
import geotrellis.proj4._
import geotrellis.store.query.RepositoryM

import cats.Monad
import cats.syntax.functor._
import cats.syntax.apply._
import cats.instances.option._

/**
 * This class holds all the information necessary to construct a response to a WMTS request
 */
case class WmtsModel[F[_]: Monad](
  serviceMetadata: ows.ServiceMetadata,
  matrices: List[GeotrellisTileMatrixSet],
  sources: RepositoryM[F, List, OgcSource]
) {

  val matrixSetLookup: Map[String, GeotrellisTileMatrixSet] =
    matrices.map(tileMatrixSet => tileMatrixSet.identifier -> tileMatrixSet).toMap

  /**
   * Take a specific request for a map and combine it with the relevant [[OgcSource]] to produce a [[TiledOgcLayer]]
   */
  def getLayer(p: GetTile): F[List[TiledOgcLayer]] = {
    (
      getMatrixCrs(p.tileMatrixSet),
      getMatrixLayoutDefinition(p.tileMatrixSet, p.tileMatrix)
    ).traverseN { (crs, layout) =>
      sources.find(p.toQuery).map { sources =>
        sources.map { source =>
          val style: Option[OgcStyle] = source.styles.find(_.name == p.style)

          source match {
            case mas: MapAlgebraSource =>
              val (name, title, algebra, resampleMethod, overviewStrategy) =
                (mas.name, mas.title, mas.algebra, mas.resampleMethod, mas.overviewStrategy)

              val simpleLayers = mas.sources.map { case (key, rs) =>
                key -> SimpleTiledOgcLayer(name, title, crs, layout, rs, style, resampleMethod, overviewStrategy)
              }
              MapAlgebraTiledOgcLayer(name, title, crs, layout, simpleLayers, algebra, style, resampleMethod, overviewStrategy)
            case ss: SimpleSource =>
              SimpleTiledOgcLayer(ss.name, ss.title, crs, layout, ss.source, style, ss.resampleMethod, ss.overviewStrategy)
            case gos: GeoTrellisOgcSource =>
              SimpleTiledOgcLayer(gos.name, gos.title, crs, layout, gos.source, style, gos.resampleMethod, gos.overviewStrategy)
          }
        }
      }
    }
  }.map { _.getOrElse(Nil) }

  def getMatrixLayoutDefinition(tileMatrixSetId: String, tileMatrixId: String): Option[LayoutDefinition] =
    for {
      matrixSet <- matrixSetLookup.get(tileMatrixSetId)
      matrix <- matrixSet.tileMatrix.find(_.identifier == tileMatrixId)
    } yield matrix.layout

  def getMatrixCrs(tileMatrixSetId: String): Option[CRS] =
    matrixSetLookup.get(tileMatrixSetId).map(_.supportedCrs)
}