]> git.openstreetmap.org Git - nominatim.git/blob - utils/warm.php
move psqlRunScript implementation into cmd lib
[nominatim.git] / utils / warm.php
1 #!/usr/bin/php -Cq
2 <?php
3
4 require_once(dirname(dirname(__FILE__)).'/settings/settings.php');
5 require_once(CONST_BasePath.'/lib/init-cmd.php');
6 ini_set('memory_limit', '800M');
7
8 $aCMDOptions = array(
9                 "Tools to warm nominatim db",
10                 array('help', 'h', 0, 1, 0, 0, false, 'Show Help'),
11                 array('quiet', 'q', 0, 1, 0, 0, 'bool', 'Quiet output'),
12                 array('verbose', 'v', 0, 1, 0, 0, 'bool', 'Verbose output'),
13                 array('reverse-only', '', 0, 1, 0, 0, 'bool', 'Warm reverse only'),
14                 array('search-only', '', 0, 1, 0, 0, 'bool', 'Warm search only'),
15                );
16 getCmdOpt($_SERVER['argv'], $aCMDOptions, $aResult, true, true);
17
18 require_once(CONST_BasePath.'/lib/log.php');
19 require_once(CONST_BasePath.'/lib/Geocode.php');
20 require_once(CONST_BasePath.'/lib/PlaceLookup.php');
21 require_once(CONST_BasePath.'/lib/ReverseGeocode.php');
22
23 $oDB =& getDB();
24
25 $bVerbose = $aResult['verbose'];
26
27 if (!$aResult['search-only']) {
28     $oReverseGeocode = new Nominatim\ReverseGeocode($oDB);
29     $oReverseGeocode->setZoom(20);
30     $oPlaceLookup = new Nominatim\PlaceLookup($oDB);
31     $oPlaceLookup->setIncludeAddressDetails(true);
32     $oPlaceLookup->setLanguagePreference(array('en'));
33
34     echo "Warm reverse: ";
35     if ($bVerbose) echo "\n";
36     for ($i = 0; $i < 1000; $i++) {
37         $fLat = rand(-9000, 9000) / 100;
38         $fLon = rand(-18000, 18000) / 100;
39         if ($bVerbose) echo "$fLat, $fLon = ";
40         $aLookup = $oReverseGeocode->lookup($fLat, $fLon);
41         if ($aLookup && $aLookup['place_id']) {
42             $aDetails = $oPlaceLookup->lookup(
43                 (int)$aLookup['place_id'],
44                 $aLookup['type'],
45                 $aLookup['fraction']
46             );
47             if ($bVerbose) echo $aDetails['langaddress']."\n";
48         } else {
49             echo ".";
50         }
51     }
52     echo "\n";
53 }
54
55 if (!$aResult['reverse-only']) {
56     $oGeocode = new Nominatim\Geocode($oDB);
57
58     echo "Warm search: ";
59     if ($bVerbose) echo "\n";
60     $sSQL = 'select word from word where word is not null order by search_name_count desc limit 1000';
61     foreach ($oDB->getCol($sSQL) as $sWord) {
62         if ($bVerbose) echo "$sWord = ";
63         $oGeocode->setLanguagePreference(array('en'));
64         $oGeocode->setQuery($sWord);
65         $aSearchResults = $oGeocode->lookup();
66         if ($bVerbose) echo $aSearchResults[0]['langaddress']."\n";
67         else echo ".";
68     }
69 }