]> git.openstreetmap.org Git - nominatim.git/blob - lib/PlaceLookup.php
add tests for new polygon_threshold feature
[nominatim.git] / lib / PlaceLookup.php
1 <?php
2         class PlaceLookup
3         {
4                 protected $oDB;
5
6                 protected $iPlaceID;
7
8                 protected $aLangPrefOrder = array();
9
10                 protected $bAddressDetails = false;
11
12                 function PlaceLookup(&$oDB)
13                 {
14                         $this->oDB =& $oDB;
15                 }
16
17                 function setLanguagePreference($aLangPrefOrder)
18                 {
19                         $this->aLangPrefOrder = $aLangPrefOrder;
20                 }
21
22                 function setIncludeAddressDetails($bAddressDetails = true)
23                 {
24                         $this->bAddressDetails = $bAddressDetails;
25                 }
26
27                 function setPlaceID($iPlaceID)
28                 {
29                         $this->iPlaceID = $iPlaceID;
30                 }
31
32                 function setOSMID($sType, $iID)
33                 {
34                         $sSQL = "select place_id from placex where osm_type = '".pg_escape_string($sType)."' and osm_id = ".(int)$iID." order by type = 'postcode' asc";
35                         $this->iPlaceID = $this->oDB->getOne($sSQL);
36                 }
37
38                 function lookup()
39                 {
40                         if (!$this->iPlaceID) return null;
41
42                         $sLanguagePrefArraySQL = "ARRAY[".join(',',array_map("getDBQuoted", $this->aLangPrefOrder))."]";
43
44                         $sSQL = "select placex.place_id, partition, osm_type, osm_id, class, type, admin_level, housenumber, street, isin, postcode, country_code, extratags, parent_place_id, linked_place_id, rank_address, rank_search, ";
45                         $sSQL .= " coalesce(importance,0.75-(rank_search::float/40)) as importance, indexed_status, indexed_date, wikipedia, calculated_country_code, ";
46                         $sSQL .= " get_address_by_language(place_id, $sLanguagePrefArraySQL) as langaddress,";
47                         $sSQL .= " get_name_by_language(name, $sLanguagePrefArraySQL) as placename,";
48                         $sSQL .= " get_name_by_language(name, ARRAY['ref']) as ref,";
49                         $sSQL .= " (case when centroid is null then st_y(st_centroid(geometry)) else st_y(centroid) end) as lat,";
50                         $sSQL .= " (case when centroid is null then st_x(st_centroid(geometry)) else st_x(centroid) end) as lon";
51                         $sSQL .= " from placex where place_id = ".(int)$this->iPlaceID;
52                         $aPlace = $this->oDB->getRow($sSQL);
53
54                         if (PEAR::IsError($aPlace))
55                         {
56                                 failInternalError("Could not lookup place.", $sSQL, $aPlace);
57                         }
58                         if (!$aPlace['place_id']) return null;
59
60                         if ($this->bAddressDetails)
61                         {
62                                 $aAddress = $this->getAddressNames();
63                                 $aPlace['aAddress'] = $aAddress;
64                         }
65
66                         $aClassType = getClassTypes();
67                         $sAddressType = '';
68                         $sClassType = $aPlace['class'].':'.$aPlace['type'].':'.$aPlace['admin_level'];
69                         if (isset($aClassType[$sClassType]) && isset($aClassType[$sClassType]['simplelabel']))
70                         {
71                                 $sAddressType = $aClassType[$aClassType]['simplelabel'];
72                         }
73                         else
74                         {
75                                 $sClassType = $aPlace['class'].':'.$aPlace['type'];
76                                 if (isset($aClassType[$sClassType]) && isset($aClassType[$sClassType]['simplelabel']))
77                                         $sAddressType = $aClassType[$sClassType]['simplelabel'];
78                                 else $sAddressType = $aPlace['class'];
79                         }
80
81                         $aPlace['addresstype'] = $sAddressType;
82
83                         return $aPlace;
84                 }
85
86                 function getAddressDetails($bAll = false)
87                 {
88                         if (!$this->iPlaceID) return null;
89
90                         $sLanguagePrefArraySQL = "ARRAY[".join(',',array_map("getDBQuoted", $this->aLangPrefOrder))."]";
91
92                         $sSQL = "select *,get_name_by_language(name,$sLanguagePrefArraySQL) as localname from get_addressdata(".$this->iPlaceID.")";
93                         if (!$bAll) $sSQL .= " WHERE isaddress OR type = 'country_code'";
94                         $sSQL .= " order by rank_address desc,isaddress desc";
95
96                         $aAddressLines = $this->oDB->getAll($sSQL);
97                         if (PEAR::IsError($aAddressLines))
98                         {
99                                 var_dump($aAddressLines);
100                                 exit;
101                         }
102                         return $aAddressLines;
103                 }
104
105                 function getAddressNames()
106                 {
107                         $aAddressLines = $this->getAddressDetails(false);;
108
109                         $aAddress = array();
110                         $aFallback = array();
111                         $aClassType = getClassTypes();
112                         foreach($aAddressLines as $aLine)
113                         {
114                                 $bFallback = false;
115                                 $aTypeLabel = false;
116                                 if (isset($aClassType[$aLine['class'].':'.$aLine['type'].':'.$aLine['admin_level']])) $aTypeLabel = $aClassType[$aLine['class'].':'.$aLine['type'].':'.$aLine['admin_level']];
117                                 elseif (isset($aClassType[$aLine['class'].':'.$aLine['type']])) $aTypeLabel = $aClassType[$aLine['class'].':'.$aLine['type']];
118                                 elseif (isset($aClassType['boundary:administrative:'.((int)($aLine['rank_address']/2))]))
119                                 {
120                                         $aTypeLabel = $aClassType['boundary:administrative:'.((int)($aLine['rank_address']/2))];
121                                         $bFallback = true;
122                                 }
123                                 else
124                                 {
125                                         $aTypeLabel = array('simplelabel'=>'address'.$aLine['rank_address']);
126                                         $bFallback = true;
127                                 }
128                                 if ($aTypeLabel && ((isset($aLine['localname']) && $aLine['localname']) || (isset($aLine['housenumber']) && $aLine['housenumber'])))
129                                 {
130                                         $sTypeLabel = strtolower(isset($aTypeLabel['simplelabel'])?$aTypeLabel['simplelabel']:$aTypeLabel['label']);
131                                         $sTypeLabel = str_replace(' ','_',$sTypeLabel);
132                                         if (!isset($aAddress[$sTypeLabel]) || (isset($aFallback[$sTypeLabel]) && $aFallback[$sTypeLabel]) || $aLine['class'] == 'place')
133                                         {
134                                                 $aAddress[$sTypeLabel] = $aLine['localname']?$aLine['localname']:$aLine['housenumber'];
135                                         }
136                                         $aFallback[$sTypeLabel] = $bFallback;
137                                 }
138                         }
139                         return $aAddress;
140                 }
141
142         }
143 ?>