rework XOrgDB -> XDB (shorter, easier to use).
[platal.git] / include / geoloc.inc.php
CommitLineData
0337d704 1<?php
2/***************************************************************************
2b105fb6 3 * Copyright (C) 2003-2006 Polytechnique.org *
0337d704 4 * http://opensource.polytechnique.org/ *
5 * *
6 * This program is free software; you can redistribute it and/or modify *
7 * it under the terms of the GNU General Public License as published by *
8 * the Free Software Foundation; either version 2 of the License, or *
9 * (at your option) any later version. *
10 * *
11 * This program is distributed in the hope that it will be useful, *
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of *
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
14 * GNU General Public License for more details. *
15 * *
16 * You should have received a copy of the GNU General Public License *
17 * along with this program; if not, write to the Free Software *
18 * Foundation, Inc., *
19 * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA *
20 ***************************************************************************/
21
22
23// {{{ liste les pays ou les régions d'un pays
24/** donne la liste déroulante des pays
25 * @param $current pays actuellement selectionné
26 */
27function geoloc_country($current) {
28 global $globals;
08cce2ff 29 $res = XDB::iterRow('SELECT a2,pays FROM geoloc_pays ORDER BY pays');
0337d704 30 $html = "";
31 while (list($my_id, $my_pays) = $res->next()) {
32 $html .= sprintf("<option value=\"%s\" %s>%s</option>\n",
33 $my_id, ($current==$my_id?"selected='selected'":""), $my_pays);
34 }
35 return $html;
36}
37
38function _geoloc_country_smarty($params){
39 if(!isset($params['country']))
40 return;
41 return geoloc_country($params['country']);
42}
80ca1b4e 43$GLOBALS['page']->register_function('geoloc_country', '_geoloc_country_smarty');
0337d704 44
45/** donne la liste deroulante des regions pour un pays
46 * @param $pays le pays dont on veut afficher les regions
47 * @param $current la region actuellement selectionnee
48 */
49function geoloc_region($country,$current) {
50 global $globals;
08cce2ff 51 $res = XDB::iterRow('SELECT region,name FROM geoloc_region where a2={?} ORDER BY name', $country);
0337d704 52 $html = "<option value=\"\"></option>";
53 while (list($regid, $regname) = $res->next()) {
54 $html .= sprintf("<option value=\"%s\" %s>%s</option>\n",
55 $regid, ($current==$regid?"selected='selected'":""), $regname);
56 }
57 return $html;
58
59}
60function _geoloc_region_smarty($params){
61 if(!isset($params['country']))
62 return;
63 if(!isset($params['region']))
64 return;
65 return geoloc_region($params['country'], $params['region']);
66}
80ca1b4e 67$GLOBALS['page']->register_function('geoloc_region', '_geoloc_region_smarty');
0337d704 68// }}}
69
70// {{{ get_address_infos($txt)
71/** retrieve the infos on a text address
72 * store on the fly the info of the city concerned
73 * @param $txt the raw text of an address
74 */
75function get_address_infos($txt) {
56670b6a 76 global $globals;
c41e72df 77 $url = $globals->geoloc->webservice_url."address.php?precise=1&txt=".urlencode(utf8_encode($txt));
0337d704 78 if (!($f = @fopen($url, 'r'))) return false;
79 $keys = explode('|',fgets($f));
80 $vals = explode('|',fgets($f));
81 $infos = array();
82 foreach ($keys as $i=>$key) if($vals[$i]) $infos[$key] = ($key == 'sql')?$vals[$i]:utf8_decode(strtr($vals[$i], array(chr(197).chr(147) => "&oelig;")));
0337d704 83 if ($infos['sql'])
08cce2ff 84 XDB::execute("REPLACE INTO geoloc_city VALUES ".$infos['sql']);
80ca1b4e 85 if ($infos['display'])
08cce2ff 86 XDB::execute("UPDATE geoloc_pays SET display = {?} WHERE a2 = {?}", $infos['display'], $infos['country']);
0337d704 87 return $infos;
88}
89// }}}
90
56670b6a 91// {{{ get_cities_maps($array)
92/* get all the maps id of the cities contained in an array */
93function get_cities_maps($array)
94{
95 global $globals;
96 implode("\n",$array);
97 $url = $globals->geoloc->webservice_url."findMaps.php?datatext=".urlencode(utf8_encode(implode("\n", $array)));
98 if (!($f = @fopen($url, 'r'))) return false;
99 $maps = array();
100 while (!feof($f))
101 {
102 $l = trim(fgets($f));
103 $tab = explode(';', $l);
104 $i = $tab[0];
105 unset($tab[0]);
106 $maps[$i] = $tab;
107 }
108 return $maps;
109}
110// }}}
111
112// {{{ get_new_maps($url)
113/** set new maps from url **/
114function get_new_maps($url)
115{
116 if (!($f = @fopen($url, 'r'))) return false;
117 global $globals;
08cce2ff 118 XDB::query('TRUNCATE TABLE geoloc_maps');
56670b6a 119 $s = '';
120 while (!feof($f)) {
121 $l = fgetcsv($f, 1024, ';', '"');
122 foreach ($l as $i => $val) if ($val != 'NULL') $l[$i] = '\''.addslashes($val).'\'';
123 $s .= ',('.implode(',',$l).')';
124 }
08cce2ff 125 XDB::execute('INSERT INTO geoloc_maps VALUES '.substr($s, 1));
56670b6a 126 return true;
127}
128
0337d704 129// {{{ get_address_text($adr)
130/** make the text of an address that can be read by a mailman
131 * @param $adr an array with all the usual fields
132 */
133function get_address_text($adr) {
134 $t = "";
135 if ($adr['adr1']) $t.= $adr['adr1'];
136 if ($adr['adr2']) $t.= "\n".$adr['adr2'];
137 if ($adr['adr3']) $t.= "\n".$adr['adr3'];
138 $l = "";
80ca1b4e 139 if ($adr['display']) {
140 $keys = explode(' ', $adr['display']);
141 foreach ($keys as $key) {
142 if (isset($adr[$key]))
143 $l .= " ".$adr[$key];
144 else
145 $l .= " ".$key;
146 }
147 if ($l) $l = substr($l, 1);
148 }
149 else
150 {
151 if ($adr['country'] == 'US' || $adr['country'] == 'CA' || $adr['country'] == 'GB') {
152 if ($adr['city']) $l .= $adr['city'].",\n";
153 if ($adr['region']) $l .= $adr['region']." ";
154 if ($adr['postcode']) $l .= $adr['postcode'];
155 } else {
156 if ($adr['postcode']) $l .= $adr['postcode']." ";
157 if ($adr['city']) $l .= $adr['city'];
158 }
0337d704 159 }
160 if ($l) $t .= "\n".trim($l);
161 if ($adr['country'] != '00' && (!$adr['countrytxt'] || $adr['countrytxt'] == strtoupper($adr['countrytxt']))) {
162 global $globals;
08cce2ff 163 $res = XDB::query("SELECT pays FROM geoloc_pays WHERE a2 = {?}", $adr['country']);
0337d704 164 $adr['countrytxt'] = $res->fetchOneCell();
165 }
166 if ($adr['countrytxt']) $t .= "\n".$adr['countrytxt'];
167 return trim($t);
168}
169// }}}
170
171// {{{ compare_addresses_text($a, $b)
172/** compares if two address matches
173 * @param $a the raw text of an address
174 * @param $b the raw text of a complete valid address
175 */
176function compare_addresses_text($a, $b) {
177 $ta = strtoupper(preg_replace(array("/[0-9,\"'#~:;_\- ]/", "/\r\n/"), array("", "\n"), $a));
178 $tb = strtoupper(preg_replace(array("/[0-9,\"'#~:;_\- ]/", "/\r\n/"), array("", "\n"), $b));
179
180 $la = explode("\n", $ta);
181 $lb = explode("\n", $tb);
182
183 if (count($lb) > count($la) + 1) return false;
184 foreach ($la as $i=>$l) if (levenshtein($l, $lb[$i]) > 3) return false;
185 return true;
186}
187
188// }}}
189
190function empty_address() {
191 return Array(
192 "adr1" => "",
193 "adr2" => "",
194 "adr3" => "",
195 "cityid" => NULL,
196 "city" => "",
197 "postcode" => "",
198 "region" => "",
80ca1b4e 199 "regiontxt" => "",
0337d704 200 "country" => "00",
201 "countrytxt" => "");
202}
203
204// create a simple address from a text without geoloc
205function cut_address($txt) {
206 $txt = str_replace("\r\n", "\n", $txt);
207 ereg("^([^\n]*)(\n([^\n]*)(\n(.*))?)?$", trim($txt), $a);
208 return array("adr1" => trim($a[1]), "adr2" => trim($a[3]), "adr3" => trim(str_replace("\n", " ", $a[5])));
209}
210
211// {{{ localize_addresses($uid)
212/* localize all the address of a user and modify the database
213 * if the new address match with the old one
214 * @param $uid the id of the user
215 */
216function localize_addresses($uid) {
217 global $globals;
08cce2ff 218 $res = XDB::iterator("SELECT * FROM adresses WHERE uid = {?} and (cityid IS NULL OR cityid = 0)", $uid);
0337d704 219 $erreur = Array();
220
221 while ($a = $res->next()) {
222 $new = get_address_infos($ta = get_address_text($a));
223 if (compare_addresses_text($ta, get_address_text($new))) {
08cce2ff 224 XDB::execute("UPDATE adresses SET
0337d704 225 adr1 = {?}, adr2 = {?}, adr3 = {?},
226 cityid = {?}, city = {?}, postcode = {?},
56670b6a 227 region = {?}, regiontxt = {?}, country = {?},
228 glat = {?}, glng = {?}
0337d704 229 WHERE uid = {?} AND adrid = {?}",
230 $new['adr1'], $new['adr2'], $new['adr3'],
231 $new['cityid'], $new['city'], $new['postcode'],
80ca1b4e 232 $new['region'], $new['regiontxt'], $new['country'],
56670b6a 233 $new['precise_lat'], $new['precise_lon'],
0337d704 234 $uid, $a['adrid']);
235 $new['store'] = true;
236 if (!$new['cityid']) $erreur[$a['adrid']] = $new;
237 } else {
238 $new['store'] = false;
239 $erreur[$a['adrid']] = $new;
240 }
241 }
242 return $erreur;
243}
244// }}}
245
246// {{{ synchro_city($id)
247/** synchronise the local geoloc_city base to geoloc.org
248 * @param $id the id of the city to synchronize
249 */
250 function synchro_city($id) {
56670b6a 251 global $globals;
252 $url = $globals->geoloc->webservice_url."cityFinder.php?method=id&id=".$id."&out=sql";
0337d704 253 if (!($f = @fopen($url, 'r'))) return false;
254 $s = fgets($f);
0337d704 255 if ($s)
08cce2ff 256 return XDB::execute("REPLACE INTO geoloc_city VALUES ".$s) > 0;
0337d704 257 }
258 // }}}
259
56670b6a 260// {{{ function fix_cities_not_on_map($limit)
261function fix_cities_not_on_map($limit=false)
262{
263 global $globals;
08cce2ff 264 $missing = XDB::query("SELECT c.id FROM geoloc_city AS c LEFT JOIN geoloc_city_in_maps AS m ON(c.id = m.city_id) WHERE m.city_id IS NULL".($limit?" LIMIT $limit":""));
56670b6a 265 $maps = get_cities_maps($missing->fetchColumn());
266 if ($maps)
267 {
268 $values = "";
269 foreach ($maps as $cityid => $maps_c)
270 foreach ($maps_c as $map_id)
271 $values .= ",($cityid, $map_id, '')";
08cce2ff 272 XDB::execute("REPLACE INTO geoloc_city_in_maps VALUES ".substr($values, 1));
56670b6a 273 }
274 else
275 return false;
014c8464 276 return true;
277}
56670b6a 278
014c8464 279function set_smallest_levels() {
280 global $globals;
08cce2ff 281 $maxlengths = XDB::iterRow("SELECT MAX(LENGTH(gm.path)), gcim.city_id
014c8464 282 FROM geoloc_city_in_maps AS gcim
283 INNER JOIN geoloc_maps AS gm
284 USING ( map_id )
285 GROUP BY gcim.city_id
286 ");
287 while (list($length, $id) = $maxlengths->next()) {
08cce2ff 288 XDB::execute("UPDATE geoloc_city_in_maps AS gcim
014c8464 289 INNER JOIN geoloc_maps AS gm USING(map_id)
290 SET gcim.infos = IF(LENGTH(gm.path) = {?}, 'smallest', '')
291 WHERE gcim.city_id = {?}", $length, $id);
292 }
56670b6a 293 return true;
294}
295// }}}
296
297
298function geoloc_to_x($lon, $lat) { return deg2rad(1) * $lon *100; }
299
300function geoloc_to_y($lon, $lat) {
301 if ($lat < -75) return latToY(-75);
302 if ($lat > 75) return latToY(75);
303 return -100 * log(tan(pi()/4 + deg2rad(1)/2*$lat));
304}
305
306function size_of_city($nb) { $s = round(log($nb + 1)*2,2); if ($s < 1) return 1; return $s; }
307function size_of_territory($nb) { return size_of_city($nb); }
308
2b105fb6 309function geoloc_getData_subcities($mapid, $SFields, &$cities, $direct=true) {
310 global $globals;
311 for ($i_mapfield=0; $i_mapfield < count($SFields) ; $i_mapfield++) if ($SFields[$i_mapfield]->fieldFormName == 'mapid') break;
312 $SFields[$i_mapfield] = new MapSField('mapid', array('gcim.map_id'), array('adresses','geoloc_city_in_maps'), array('am','gcim'), array(getadr_join('am'), 'am.cityid = gcim.city_id'), $mapid);
313
314 $fields = new SFieldGroup(true, $SFields);
315 $where = $fields->get_where_statement();
316 if ($where) $where = " AND ".$where;
317
08cce2ff 318 $cityres = XDB::iterator("
2b105fb6 319 SELECT gc.id,
320 gc.lon / 100000 AS x, gc.lat/100000 AS y,
321 gc.name,
322 COUNT(u.user_id) AS pop,
323 SUM(u.promo % 2) AS yellow
324 FROM auth_user_md5 AS u
325 INNER JOIN auth_user_quick AS q ON(u.user_id = q.user_id)
326 ".$fields->get_select_statement()."
327 LEFT JOIN geoloc_city AS gc ON(gcim.city_id = gc.id)
328 WHERE ".($direct?"gcim.infos = 'smallest'":"1")."
329 $where
330 GROUP BY gc.id,gc.alias ORDER BY pop DESC");
331 while ($c = $cityres->next())
332 if ($c['pop'] > 0)
333 {
334 $city = $c;
335 $city['x'] = geoloc_to_x($c['x'], $c['y']);
336 $city['y'] = geoloc_to_y($c['x'], $c['y']);
337 $city['size'] = size_of_city($c['pop']);
338 $cities[$c['id']] = $city;
339 }
340}
341
342function geoloc_getData_subcountries($mapid, $SFields, $minentities) {
343 global $globals;
344 $countries = array();
345 $cities = array();
346
347 if ($mapid === false)
348 $wheremapid = "WHERE gm.parent IS NULL";
349 else
350 $wheremapid = "WHERE gm.parent = {?}";
08cce2ff 351 $submapres = XDB::iterator(
2b105fb6 352 "SELECT gm.map_id AS id, gm.name, gm.x, gm.y, gm.xclip, gm.yclip,
353 gm.width, gm.height, gm.scale, 1 AS rat
354 FROM geoloc_maps AS gm
355 ".$wheremapid, Env::get('mapid',''));
356
357 while ($c = $submapres->next())
358 {
359 $country = $c;
360 $country['name'] = utf8_decode($country['name']);
361 $country['color'] = 0xFFFFFF;
362 $country['swf'] = $globals->geoloc->webservice_url."maps/mercator/map_".$c['id'].".swf";
363 $countries[$c['id']] = $country;
364 }
365
366 if ($mapid === false) return array($countries, $cities);
367
368 geoloc_getData_subcities(Env::getInt('mapid'), $SFields, $cities);
369 $nbcities = count($cities);
370 $nocity = $nbcities == 0;
371
372 for ($i_mapfield=0; $i_mapfield < count($SFields) ; $i_mapfield++) if ($SFields[$i_mapfield]->fieldFormName == 'mapid') break;
373 $SFields[$i_mapfield] = new MapSField('mapid', array('map.parent'), array('adresses','geoloc_city_in_maps','geoloc_maps'), array('am','gcim','map'), array(getadr_join('am'), 'am.cityid = gcim.city_id', 'map.map_id = gcim.map_id'));
374
375 $fields = new SFieldGroup(true, $SFields);
376 $where = $fields->get_where_statement();
377 if ($where) $where = " WHERE ".$where;
378
08cce2ff 379 $countryres = XDB::iterator("
2b105fb6 380 SELECT map.map_id AS id,
381 COUNT(u.user_id) AS nbPop,
382 SUM(u.promo % 2) AS yellow,
383 COUNT(DISTINCT gcim.city_id) AS nbCities,
384 SUM(IF(u.user_id IS NULL,0,am.glng)) AS lonPop,
385 SUM(IF(u.user_id IS NULL, 0,am.glat)) AS latPop
386 FROM auth_user_md5 AS u
387 INNER JOIN auth_user_quick AS q ON(u.user_id = q.user_id)
388 ".$fields->get_select_statement()."
389 $where
390 GROUP BY map.map_id ORDER BY NULL", $hierarchy);
391
392 $maxpop = 0;
393 $nbentities = $nbcities + $countryres->total();
394 while ($c = $countryres->next())
395 {
396 $c['latPop'] /= $c['nbPop'];
397 $c['lonPop'] /= $c['nbPop'];
398 $c['rad'] = size_of_territory($c['nbPop']);
399 if ($maxpop < $c['nbPop']) $maxpop = $c['nbPop'];
400 $c['xPop'] = geoloc_to_x($c['lonPop'], $c['latPop']);
401 $c['yPop'] = geoloc_to_y($c['lonPop'], $c['latPop']);
402 $countries[$c['id']] = array_merge($countries[$c['id']], $c);
403
404 $nbcities += $c['nbCities'];
405 }
406
407 if ($nocity && $nbcities < $minentities)
408 {
409 foreach($countries as $i => $c)
410 {
411 $countries[$i]['nbPop'] = 0;
412 if ($c['nbCities'] > 0)
413 geoloc_getData_subcities($c['id'], $SFields, $cities, false);
414 }
415 }
416
417 foreach ($countries as $i => $c) if ($c['nbPop'] > 0)
418 {
419 $lambda = pow($c['nbPop'] / $maxpop,0.3);
420 $countries[$i]['color'] = 0x0000FF + round((1-$lambda) * 0xFF)*0x010100;
421 }
422
423 return array($countries, $cities);
424}
0337d704 425// vim:set et sw=4 sts=4 sws=4 foldmethod=marker:
426?>