lib-php/admin/warm.php

   1 <?php
   2 @define('CONST_LibDir', dirname(dirname(__FILE__)));
   3
   4 require_once(CONST_LibDir.'/init-cmd.php');
   5 require_once(CONST_LibDir.'/log.php');
   6 require_once(CONST_LibDir.'/Geocode.php');
   7 require_once(CONST_LibDir.'/PlaceLookup.php');
   8 require_once(CONST_LibDir.'/ReverseGeocode.php');
   9
  10 ini_set('memory_limit', '800M');
  11
  12 $aCMDOptions = array(
  13                 'Tools to warm nominatim db',
  14                 array('help', 'h', 0, 1, 0, 0, false, 'Show Help'),
  15                 array('quiet', 'q', 0, 1, 0, 0, 'bool', 'Quiet output'),
  16                 array('verbose', 'v', 0, 1, 0, 0, 'bool', 'Verbose output'),
  17                 array('reverse-only', '', 0, 1, 0, 0, 'bool', 'Warm reverse only'),
  18                 array('search-only', '', 0, 1, 0, 0, 'bool', 'Warm search only'),
  19                 array('project-dir', '', 0, 1, 1, 1, 'realpath', 'Base directory of the Nominatim installation (default: .)'),
  20                );
  21 getCmdOpt($_SERVER['argv'], $aCMDOptions, $aResult, true, true);
  22
  23 loadSettings($aCMDResult['project-dir'] ?? getcwd());
  24
  25 @define('CONST_Database_DSN', getSetting('DATABASE_DSN'));
  26 @define('CONST_Default_Language', getSetting('DEFAULT_LANGUAGE', false));
  27 @define('CONST_Log_DB', getSettingBool('LOG_DB'));
  28 @define('CONST_Log_File', getSetting('LOG_FILE', false));
  29 @define('CONST_Max_Word_Frequency', getSetting('MAX_WORD_FREQUENCY'));
  30 @define('CONST_NoAccessControl', getSettingBool('CORS_NOACCESSCONTROL'));
  31 @define('CONST_Places_Max_ID_count', getSetting('LOOKUP_MAX_COUNT'));
  32 @define('CONST_PolygonOutput_MaximumTypes', getSetting('POLYGON_OUTPUT_MAX_TYPES'));
  33 @define('CONST_Search_BatchMode', getSettingBool('SEARCH_BATCH_MODE'));
  34 @define('CONST_Search_NameOnlySearchFrequencyThreshold', getSetting('SEARCH_NAME_ONLY_THRESHOLD'));
  35 @define('CONST_Term_Normalization_Rules', getSetting('TERM_NORMALIZATION'));
  36 @define('CONST_Use_US_Tiger_Data', getSettingBool('USE_US_TIGER_DATA'));
  37 @define('CONST_MapIcon_URL', getSetting('MAPICON_URL', false));
  38
  39
  40 $oDB = new Nominatim\DB();
  41 $oDB->connect();
  42
  43 $bVerbose = $aResult['verbose'];
  44
  45 function print_results($aResults, $bVerbose)
  46 {
  47     if ($bVerbose) {
  48         if ($aResults && count($aResults)) {
  49             echo $aResults[0]['langaddress']."\n";
  50         } else {
  51             echo "<not found>\n";
  52         }
  53     } else {
  54         echo '.';
  55     }
  56 }
  57
  58 if (!$aResult['search-only']) {
  59     $oReverseGeocode = new Nominatim\ReverseGeocode($oDB);
  60     $oReverseGeocode->setZoom(20);
  61     $oPlaceLookup = new Nominatim\PlaceLookup($oDB);
  62     $oPlaceLookup->setIncludeAddressDetails(true);
  63     $oPlaceLookup->setLanguagePreference(array('en'));
  64
  65     echo 'Warm reverse: ';
  66     if ($bVerbose) echo "\n";
  67     for ($i = 0; $i < 1000; $i++) {
  68         $fLat = rand(-9000, 9000) / 100;
  69         $fLon = rand(-18000, 18000) / 100;
  70         if ($bVerbose) echo "$fLat, $fLon = ";
  71
  72         $oLookup = $oReverseGeocode->lookup($fLat, $fLon);
  73         $aSearchResults = $oLookup ? $oPlaceLookup->lookup(array($oLookup->iId => $oLookup)) : null;
  74         print_results($aSearchResults, $bVerbose);
  75     }
  76     echo "\n";
  77 }
  78
  79 if (!$aResult['reverse-only']) {
  80     $oGeocode = new Nominatim\Geocode($oDB);
  81
  82     echo 'Warm search: ';
  83     if ($bVerbose) echo "\n";
  84     $sSQL = 'SELECT word FROM word WHERE word is not null ORDER BY search_name_count DESC LIMIT 1000';
  85     foreach ($oDB->getCol($sSQL) as $sWord) {
  86         if ($bVerbose) echo "$sWord = ";
  87
  88         $oGeocode->setLanguagePreference(array('en'));
  89         $oGeocode->setQuery($sWord);
  90         $aSearchResults = $oGeocode->lookup();
  91         print_results($aSearchResults, $bVerbose);
  92     }
  93     echo "\n";
  94 }