X-Git-Url: https://git.openstreetmap.org./nominatim.git/blobdiff_plain/d594bb238b974afa638bcb8f92c069c97914f9b9..b133f2bc4cd5e467ff6ef10124416b1307cfdb62:/utils/warm.php?ds=sidebyside

diff --git a/utils/warm.php b/utils/warm.php
old mode 100755
new mode 100644
index 18121eb6..0d8260f1
--- a/utils/warm.php
+++ b/utils/warm.php
@@ -1,69 +1,78 @@
-#!/usr/bin/php -Cq
 <?php
 
-	require_once(dirname(dirname(__FILE__)).'/settings/settings.php');
-	require_once(CONST_BasePath.'/lib/init-cmd.php');
-	ini_set('memory_limit', '800M');
+require_once(CONST_LibDir.'/init-cmd.php');
+require_once(CONST_LibDir.'/log.php');
+require_once(CONST_LibDir.'/Geocode.php');
+require_once(CONST_LibDir.'/PlaceLookup.php');
+require_once(CONST_LibDir.'/ReverseGeocode.php');
 
-	$aCMDOptions = array(
-		"Tools to warm nominatim db",
-		array('help', 'h', 0, 1, 0, 0, false, 'Show Help'),
-		array('quiet', 'q', 0, 1, 0, 0, 'bool', 'Quiet output'),
-		array('verbose', 'v', 0, 1, 0, 0, 'bool', 'Verbose output'),
-		array('reverse-only', '', 0, 1, 0, 0, 'bool', 'Warm reverse only'),
-		array('search-only', '', 0, 1, 0, 0, 'bool', 'Warm reverse only'),
-	);
-	getCmdOpt($_SERVER['argv'], $aCMDOptions, $aResult, true, true);
+ini_set('memory_limit', '800M');
 
-	require_once(CONST_BasePath.'/lib/log.php');
-	require_once(CONST_BasePath.'/lib/Geocode.php');
-	require_once(CONST_BasePath.'/lib/PlaceLookup.php');
-	require_once(CONST_BasePath.'/lib/ReverseGeocode.php');
+$aCMDOptions = array(
+                'Tools to warm nominatim db',
+                array('help', 'h', 0, 1, 0, 0, false, 'Show Help'),
+                array('quiet', 'q', 0, 1, 0, 0, 'bool', 'Quiet output'),
+                array('verbose', 'v', 0, 1, 0, 0, 'bool', 'Verbose output'),
+                array('reverse-only', '', 0, 1, 0, 0, 'bool', 'Warm reverse only'),
+                array('search-only', '', 0, 1, 0, 0, 'bool', 'Warm search only'),
+                array('project-dir', '', 0, 1, 1, 1, 'realpath', 'Base directory of the Nominatim installation (default: .)'),
+               );
+getCmdOpt($_SERVER['argv'], $aCMDOptions, $aResult, true, true);
 
-	$oDB =& getDB();
+loadSettings($aCMDResult['project-dir'] ?? getcwd());
 
-	$bVerbose = $aResult['verbose'];
+$oDB = new Nominatim\DB();
+$oDB->connect();
 
-	if (!$aResult['search-only']) {
+$bVerbose = $aResult['verbose'];
 
-		$oReverseGeocode = new ReverseGeocode($oDB);
-		$oReverseGeocode->setZoom(20);
-		$oPlaceLookup = new PlaceLookup($oDB);
-		$oPlaceLookup->setIncludeAddressDetails(true);
-		$oPlaceLookup->setLanguagePreference(array('en'));
+function print_results($aResults, $bVerbose)
+{
+    if ($bVerbose) {
+        if ($aResults && count($aResults)) {
+            echo $aResults[0]['langaddress']."\n";
+        } else {
+            echo "<not found>\n";
+        }
+    } else {
+        echo '.';
+    }
+}
 
-		echo "Warm reverse: ";
-		if ($bVerbose) echo "\n";
-		for($i = 0; $i < 1000; $i++) {
-			$fLat = rand(-9000, 9000) / 100;
-			$fLon = rand(-18000, 18000) / 100;
-			if ($bVerbose) echo "$fLat, $fLon = ";
-			$aLookup = $oReverseGeocode->lookup($fLat, $fLon);
-			if ($aLookup && $aLookup['place_id'])
-			{
-				$aDetails = $oPlaceLookup->lookup((int)$aLookup['place_id'],
-				                                  $aLookup['type'], $aLookup['fraction']);
-				if ($bVerbose) echo $aDetails['langaddress']."\n";
-			}
-			else echo ".";
-		}
-		echo "\n";
-	}
+if (!$aResult['search-only']) {
+    $oReverseGeocode = new Nominatim\ReverseGeocode($oDB);
+    $oReverseGeocode->setZoom(20);
+    $oPlaceLookup = new Nominatim\PlaceLookup($oDB);
+    $oPlaceLookup->setIncludeAddressDetails(true);
+    $oPlaceLookup->setLanguagePreference(array('en'));
 
-	if (!$aResult['reverse-only']) {
+    echo 'Warm reverse: ';
+    if ($bVerbose) echo "\n";
+    for ($i = 0; $i < 1000; $i++) {
+        $fLat = rand(-9000, 9000) / 100;
+        $fLon = rand(-18000, 18000) / 100;
+        if ($bVerbose) echo "$fLat, $fLon = ";
 
-		$oGeocode =& new Geocode($oDB);
+        $oLookup = $oReverseGeocode->lookup($fLat, $fLon);
+        $aSearchResults = $oLookup ? $oPlaceLookup->lookup(array($oLookup->iId => $oLookup)) : null;
+        print_results($aSearchResults, $bVerbose);
+    }
+    echo "\n";
+}
 
-		echo "Warm search: ";
-		if ($bVerbose) echo "\n";
-		$sSQL = 'select word from word where word is not null order by search_name_count desc limit 1000';
-		foreach($oDB->getCol($sSQL) as $sWord) {
-			if ($bVerbose) echo "$sWord = ";
-			$oGeocode->setLanguagePreference(array('en'));
-			$oGeocode->setQuery($sWord);
-			$aSearchResults = $oGeocode->lookup();
-			if ($bVerbose) echo $aSearchResults[0]['langaddress']."\n";
-			else echo ".";
-		}
-	}
+if (!$aResult['reverse-only']) {
+    $oGeocode = new Nominatim\Geocode($oDB);
 
+    echo 'Warm search: ';
+    if ($bVerbose) echo "\n";
+    $sSQL = 'SELECT word FROM word WHERE word is not null ORDER BY search_name_count DESC LIMIT 1000';
+    foreach ($oDB->getCol($sSQL) as $sWord) {
+        if ($bVerbose) echo "$sWord = ";
+
+        $oGeocode->setLanguagePreference(array('en'));
+        $oGeocode->setQuery($sWord);
+        $aSearchResults = $oGeocode->lookup();
+        print_results($aSearchResults, $bVerbose);
+    }
+    echo "\n";
+}