<?php

namespace Nominatim;

class PlaceLookup
{
    protected $oDB;

    protected $aLangPrefOrder = array();

    protected $bAddressDetails = false;
    protected $bExtraTags = false;
    protected $bNameDetails = false;

    protected $bIncludePolygonAsPoints = false;
    protected $bIncludePolygonAsText = false;
    protected $bIncludePolygonAsGeoJSON = false;
    protected $bIncludePolygonAsKML = false;
    protected $bIncludePolygonAsSVG = false;
    protected $fPolygonSimplificationThreshold = 0.0;


    public function __construct(&$oDB)
    {
        $this->oDB =& $oDB;
    }

    public function setLanguagePreference($aLangPrefOrder)
    {
        $this->aLangPrefOrder = $aLangPrefOrder;
    }

    public function setIncludeAddressDetails($bAddressDetails = true)
    {
        $this->bAddressDetails = $bAddressDetails;
    }

    public function setIncludeExtraTags($bExtraTags = false)
    {
        $this->bExtraTags = $bExtraTags;
    }

    public function setIncludeNameDetails($bNameDetails = false)
    {
        $this->bNameDetails = $bNameDetails;
    }


    public function setIncludePolygonAsPoints($b = true)
    {
        $this->bIncludePolygonAsPoints = $b;
    }

    public function getIncludePolygonAsPoints()
    {
        return $this->bIncludePolygonAsPoints;
    }

    public function setIncludePolygonAsText($b = true)
    {
        $this->bIncludePolygonAsText = $b;
    }

    public function getIncludePolygonAsText()
    {
        return $this->bIncludePolygonAsText;
    }

    public function setIncludePolygonAsGeoJSON($b = true)
    {
        $this->bIncludePolygonAsGeoJSON = $b;
    }

    public function setIncludePolygonAsKML($b = true)
    {
        $this->bIncludePolygonAsKML = $b;
    }

    public function setIncludePolygonAsSVG($b = true)
    {
        $this->bIncludePolygonAsSVG = $b;
    }

    public function setPolygonSimplificationThreshold($f)
    {
        $this->fPolygonSimplificationThreshold = $f;
    }

    public function lookupOSMID($sType, $iID)
    {
        $sSQL = "select place_id from placex where osm_type = '".pg_escape_string($sType)."' and osm_id = ".(int)$iID." order by type = 'postcode' asc";
        $iPlaceID = chksql($this->oDB->getOne($sSQL));

        return $this->lookup((int)$iPlaceID);
    }

    public function lookup($iPlaceID, $sType = '', $fInterpolFraction = 0.0)
    {
        if (!$iPlaceID) return null;

        $sLanguagePrefArraySQL = "ARRAY[".join(',', array_map("getDBQuoted", $this->aLangPrefOrder))."]";
        $bIsTiger = CONST_Use_US_Tiger_Data && $sType == 'tiger';
        $bIsInterpolation = $sType == 'interpolation';

        if ($bIsTiger) {
            $sSQL = "select place_id,partition, 'T' as osm_type, place_id as osm_id, 'place' as class, 'house' as type, null as admin_level, housenumber, postcode,";
            $sSQL .= " 'us' as country_code, parent_place_id, null as linked_place_id, 30 as rank_address, 30 as rank_search,";
            $sSQL .= " coalesce(null,0.75-(30::float/40)) as importance, null as indexed_status, null as indexed_date, null as wikipedia, 'us' as country_code, ";
            $sSQL .= " get_address_by_language(place_id, housenumber, $sLanguagePrefArraySQL) as langaddress,";
            $sSQL .= " null as placename,";
            $sSQL .= " null as ref,";
            if ($this->bExtraTags) $sSQL .= " null as extra,";
            if ($this->bNameDetails) $sSQL .= " null as names,";
            $sSQL .= " ST_X(point) as lon, ST_Y(point) as lat from (select *, ST_LineInterpolatePoint(linegeo, (housenumber-startnumber::float)/(endnumber-startnumber)::float) as point from ";
            $sSQL .= " (select *, ";
            $sSQL .= " CASE WHEN interpolationtype='odd' THEN floor((".$fInterpolFraction."*(endnumber-startnumber)+startnumber)/2)::int*2+1";
            $sSQL .= " WHEN interpolationtype='even' THEN ((".$fInterpolFraction."*(endnumber-startnumber)+startnumber)/2)::int*2";
            $sSQL .= " WHEN interpolationtype='all' THEN (".$fInterpolFraction."*(endnumber-startnumber)+startnumber)::int";
            $sSQL .= " END as housenumber";
            $sSQL .= " from location_property_tiger where place_id = ".$iPlaceID.") as blub1) as blub2";
        } elseif ($bIsInterpolation) {
            $sSQL = "select place_id, partition, 'W' as osm_type, osm_id, 'place' as class, 'house' as type, null admin_level, housenumber, postcode,";
            $sSQL .= " country_code, parent_place_id, null as linked_place_id, 30 as rank_address, 30 as rank_search,";
            $sSQL .= " (0.75-(30::float/40)) as importance, null as indexed_status, null as indexed_date, null as wikipedia, country_code, ";
            $sSQL .= " get_address_by_language(place_id, housenumber, $sLanguagePrefArraySQL) as langaddress,";
            $sSQL .= " null as placename,";
            $sSQL .= " null as ref,";
            if ($this->bExtraTags) $sSQL .= " null as extra,";
            if ($this->bNameDetails) $sSQL .= " null as names,";
            $sSQL .= " ST_X(point) as lon, ST_Y(point) as lat from (select *, ST_LineInterpolatePoint(linegeo, (housenumber-startnumber::float)/(endnumber-startnumber)::float) as point from ";
            $sSQL .= " (select *, ";
            $sSQL .= " CASE WHEN interpolationtype='odd' THEN floor((".$fInterpolFraction."*(endnumber-startnumber)+startnumber)/2)::int*2+1";
            $sSQL .= " WHEN interpolationtype='even' THEN ((".$fInterpolFraction."*(endnumber-startnumber)+startnumber)/2)::int*2";
            $sSQL .= " WHEN interpolationtype='all' THEN (".$fInterpolFraction."*(endnumber-startnumber)+startnumber)::int";
            $sSQL .= " END as housenumber";
            $sSQL .= " from location_property_osmline where place_id = ".$iPlaceID.") as blub1) as blub2";
            // testcase: interpolationtype=odd, startnumber=1000, endnumber=1006, fInterpolFraction=1 => housenumber=1007 => error in st_lineinterpolatepoint
            // but this will never happen, because if the searched point is that close to the endnumber, the endnumber house will be directly taken from placex (in ReverseGeocode.php line 220)
            // and not interpolated
        } else {
            $sSQL = "select placex.place_id, partition, osm_type, osm_id, class,";
            $sSQL .= " type, admin_level, housenumber, postcode, country_code,";
            $sSQL .= " parent_place_id, linked_place_id, rank_address, rank_search, ";
            $sSQL .= " coalesce(importance,0.75-(rank_search::float/40)) as importance, indexed_status, indexed_date, wikipedia, country_code, ";
            $sSQL .= " get_address_by_language(place_id, -1, $sLanguagePrefArraySQL) as langaddress,";
            $sSQL .= " get_name_by_language(name, $sLanguagePrefArraySQL) as placename,";
            $sSQL .= " get_name_by_language(name, ARRAY['ref']) as ref,";
            if ($this->bExtraTags) $sSQL .= " hstore_to_json(extratags) as extra,";
            if ($this->bNameDetails) $sSQL .= " hstore_to_json(name) as names,";
            $sSQL .= " (case when centroid is null then st_y(st_centroid(geometry)) else st_y(centroid) end) as lat,";
            $sSQL .= " (case when centroid is null then st_x(st_centroid(geometry)) else st_x(centroid) end) as lon";
            $sSQL .= " from placex where place_id = ".$iPlaceID;
        }

        $aPlace = chksql($this->oDB->getRow($sSQL), "Could not lookup place");

        if (!$aPlace['place_id']) return null;

        if ($this->bAddressDetails) {
            // to get addressdetails for tiger data, the housenumber is needed
            $iHousenumber = ($bIsTiger || $bIsInterpolation) ? $aPlace['housenumber'] : -1;
            $aPlace['aAddress'] = $this->getAddressNames($aPlace['place_id'], $iHousenumber);
        }

        if ($this->bExtraTags) {
            if ($aPlace['extra']) {
                $aPlace['sExtraTags'] = json_decode($aPlace['extra']);
            } else {
                $aPlace['sExtraTags'] = (object) array();
            }
        }

        if ($this->bNameDetails) {
            if ($aPlace['names']) {
                $aPlace['sNameDetails'] = json_decode($aPlace['names']);
            } else {
                $aPlace['sNameDetails'] = (object) array();
            }
        }

        $aClassType = getClassTypes();
        $sAddressType = '';
        $sClassType = $aPlace['class'].':'.$aPlace['type'].':'.$aPlace['admin_level'];
        if (isset($aClassType[$sClassType]) && isset($aClassType[$sClassType]['simplelabel'])) {
            $sAddressType = $aClassType[$aClassType]['simplelabel'];
        } else {
            $sClassType = $aPlace['class'].':'.$aPlace['type'];
            if (isset($aClassType[$sClassType]) && isset($aClassType[$sClassType]['simplelabel']))
                $sAddressType = $aClassType[$sClassType]['simplelabel'];
            else $sAddressType = $aPlace['class'];
        }

        $aPlace['addresstype'] = $sAddressType;

        return $aPlace;
    }

    public function getAddressDetails($iPlaceID, $bAll = false, $housenumber = -1)
    {
        $sLanguagePrefArraySQL = "ARRAY[".join(',', array_map("getDBQuoted", $this->aLangPrefOrder))."]";

        $sSQL = "select *,get_name_by_language(name,$sLanguagePrefArraySQL) as localname from get_addressdata(".$iPlaceID.",".$housenumber.")";
        if (!$bAll) $sSQL .= " WHERE isaddress OR type = 'country_code'";
        $sSQL .= " order by rank_address desc,isaddress desc";

        return chksql($this->oDB->getAll($sSQL));
    }

    public function getAddressNames($iPlaceID, $housenumber = -1)
    {
        $aAddressLines = $this->getAddressDetails($iPlaceID, false, $housenumber);

        $aAddress = array();
        $aFallback = array();
        $aClassType = getClassTypes();
        foreach ($aAddressLines as $aLine) {
            $bFallback = false;
            $aTypeLabel = false;
            if (isset($aClassType[$aLine['class'].':'.$aLine['type'].':'.$aLine['admin_level']])) {
                $aTypeLabel = $aClassType[$aLine['class'].':'.$aLine['type'].':'.$aLine['admin_level']];
            } elseif (isset($aClassType[$aLine['class'].':'.$aLine['type']])) {
                $aTypeLabel = $aClassType[$aLine['class'].':'.$aLine['type']];
            } elseif (isset($aClassType['boundary:administrative:'.((int)($aLine['rank_address']/2))])) {
                $aTypeLabel = $aClassType['boundary:administrative:'.((int)($aLine['rank_address']/2))];
                $bFallback = true;
            } else {
                $aTypeLabel = array('simplelabel' => 'address'.$aLine['rank_address']);
                $bFallback = true;
            }
            if ($aTypeLabel && ((isset($aLine['localname']) && $aLine['localname']) || (isset($aLine['housenumber']) && $aLine['housenumber']))) {
                $sTypeLabel = strtolower(isset($aTypeLabel['simplelabel'])?$aTypeLabel['simplelabel']:$aTypeLabel['label']);
                $sTypeLabel = str_replace(' ', '_', $sTypeLabel);
                if (!isset($aAddress[$sTypeLabel]) || (isset($aFallback[$sTypeLabel]) && $aFallback[$sTypeLabel]) || $aLine['class'] == 'place') {
                    $aAddress[$sTypeLabel] = $aLine['localname']?$aLine['localname']:$aLine['housenumber'];
                }
                $aFallback[$sTypeLabel] = $bFallback;
            }
        }
        return $aAddress;
    }


    /* returns an array which will contain the keys
     *   aBoundingBox
     * and may also contain one or more of the keys
     *   asgeojson
     *   askml
     *   assvg
     *   astext
     *   lat
     *   lon
     */


    public function getOutlines($iPlaceID, $fLon = null, $fLat = null, $fRadius = null)
    {

        $aOutlineResult = array();
        if (!$iPlaceID) return $aOutlineResult;

        if (CONST_Search_AreaPolygons) {
            // Get the bounding box and outline polygon
            $sSQL  = "select place_id,0 as numfeatures,st_area(geometry) as area,";
            $sSQL .= "ST_Y(centroid) as centrelat,ST_X(centroid) as centrelon,";
            $sSQL .= "ST_YMin(geometry) as minlat,ST_YMax(geometry) as maxlat,";
            $sSQL .= "ST_XMin(geometry) as minlon,ST_XMax(geometry) as maxlon";
            if ($this->bIncludePolygonAsGeoJSON) $sSQL .= ",ST_AsGeoJSON(geometry) as asgeojson";
            if ($this->bIncludePolygonAsKML) $sSQL .= ",ST_AsKML(geometry) as askml";
            if ($this->bIncludePolygonAsSVG) $sSQL .= ",ST_AsSVG(geometry) as assvg";
            if ($this->bIncludePolygonAsText || $this->bIncludePolygonAsPoints) $sSQL .= ",ST_AsText(geometry) as astext";
            $sFrom = " from placex where place_id = ".$iPlaceID;
            if ($this->fPolygonSimplificationThreshold > 0) {
                $sSQL .= " from (select place_id,centroid,ST_SimplifyPreserveTopology(geometry,".$this->fPolygonSimplificationThreshold.") as geometry".$sFrom.") as plx";
            } else {
                $sSQL .= $sFrom;
            }

            $aPointPolygon = chksql($this->oDB->getRow($sSQL), "Could not get outline");

            if ($aPointPolygon['place_id']) {
                if ($aPointPolygon['centrelon'] !== null && $aPointPolygon['centrelat'] !== null) {
                    $aOutlineResult['lat'] = $aPointPolygon['centrelat'];
                    $aOutlineResult['lon'] = $aPointPolygon['centrelon'];
                }

                if ($this->bIncludePolygonAsGeoJSON) $aOutlineResult['asgeojson'] = $aPointPolygon['asgeojson'];
                if ($this->bIncludePolygonAsKML) $aOutlineResult['askml'] = $aPointPolygon['askml'];
                if ($this->bIncludePolygonAsSVG) $aOutlineResult['assvg'] = $aPointPolygon['assvg'];
                if ($this->bIncludePolygonAsText) $aOutlineResult['astext'] = $aPointPolygon['astext'];
                if ($this->bIncludePolygonAsPoints) $aOutlineResult['aPolyPoints'] = geometryText2Points($aPointPolygon['astext'], $fRadius);


                if (abs($aPointPolygon['minlat'] - $aPointPolygon['maxlat']) < 0.0000001) {
                    $aPointPolygon['minlat'] = $aPointPolygon['minlat'] - $fRadius;
                    $aPointPolygon['maxlat'] = $aPointPolygon['maxlat'] + $fRadius;
                }

                if (abs($aPointPolygon['minlon'] - $aPointPolygon['maxlon']) < 0.0000001) {
                    $aPointPolygon['minlon'] = $aPointPolygon['minlon'] - $fRadius;
                    $aPointPolygon['maxlon'] = $aPointPolygon['maxlon'] + $fRadius;
                }

                $aOutlineResult['aBoundingBox'] = array(
                                                   (string)$aPointPolygon['minlat'],
                                                   (string)$aPointPolygon['maxlat'],
                                                   (string)$aPointPolygon['minlon'],
                                                   (string)$aPointPolygon['maxlon']
                                                  );
            }
        }

        // as a fallback we generate a bounding box without knowing the size of the geometry
        if ((!isset($aOutlineResult['aBoundingBox'])) && isset($fLon)) {
            //
            if ($this->bIncludePolygonAsPoints) {
                $sGeometryText = 'POINT('.$fLon.','.$fLat.')';
                $aOutlineResult['aPolyPoints'] = geometryText2Points($sGeometryText, $fRadius);
            }

            $aBounds = array();
            $aBounds['minlat'] = $fLat - $fRadius;
            $aBounds['maxlat'] = $fLat + $fRadius;
            $aBounds['minlon'] = $fLon - $fRadius;
            $aBounds['maxlon'] = $fLon + $fRadius;

            $aOutlineResult['aBoundingBox'] = array(
                                               (string)$aBounds['minlat'],
                                               (string)$aBounds['maxlat'],
                                               (string)$aBounds['minlon'],
                                               (string)$aBounds['maxlon']
                                              );
        }
        return $aOutlineResult;
    }
}