lucuma.catalog.votable.ADQLGaiaQuery.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of lucuma-catalog_3 Show documentation

lucuma-catalog

There is a newer version: 0.48.7

// Copyright (c) 2016-2023 Association of Universities for Research in Astronomy, Inc. (AURA)
// For license information see LICENSE or https://opensource.org/licenses/BSD-3-Clause

package lucuma.catalog.votable

import cats.syntax.all.*

sealed trait ADQLGaiaQuery {
  val gaia = CatalogAdapter.Gaia

  /**
   * Builds an adql query for gaia taking input from the adapter and the query itself
   */
  def adql(gaia: CatalogAdapter.Gaia, cs: ADQLQuery)(using ci: ADQLInterpreter): String = {
    //
    val fields           = ci.allFields.map(_.id.value.toLowerCase).mkString(",")
    val extraFields      = ci.extraFields(cs.base)
    val extraFieldsStr   =
      if (extraFields.isEmpty) "" else extraFields.mkString(",", ",", "")
    val shapeAdql        = cs.adqlGeom
    val brightnessFields = cs.adqlBrightness
    val brightnessAdql   =
      if (brightnessFields.isEmpty) "" else brightnessFields.mkString("and (", " or ", ")")
    val orderBy          = ci.orderBy.foldMap(s => s"ORDER BY $s")
    val extraConstraints =
      if (ci.extraConstraints.isEmpty) "" else ci.extraConstraints.mkString("and (", " and ", ")")

    val query =
      f"""|SELECT TOP ${ci.MaxCount} $fields $extraFieldsStr
        |     FROM ${gaia.gaiaDB}
        |     WHERE CONTAINS(POINT('ICRS',${gaia.raField.id},${gaia.decField.id}),$shapeAdql)=1
        |     $brightnessAdql
        |     $extraConstraints
        |     $orderBy
      """.stripMargin
    query
  }

}
object ADQLGaiaQuery extends ADQLGaiaQuery