addOption( 'data-dir', 'A directory on the local filesystem ' . 'containing allkeys.txt and ucd.all.grouped.xml from unicode.org', false, true ); $this->addOption( 'debug-output', 'Filename for sending debug output to', false, true ); } public function execute() { $this->dataDir = $this->getOption( 'data-dir', '.' ); $allkeysPresent = file_exists( "{$this->dataDir}/allkeys.txt" ); $ucdallPresent = file_exists( "{$this->dataDir}/ucd.all.grouped.xml" ); // As of January 2013, these links work for all versions of Unicode // between 5.1 and 6.2, inclusive. $allkeysURL = "https://www.unicode.org/Public/UCA//allkeys.txt"; $ucdallURL = "https://www.unicode.org/Public//ucdxml/ucd.all.grouped.zip"; if ( !$allkeysPresent || !$ucdallPresent ) { $icuVersion = INTL_ICU_VERSION; $unicodeVersion = IcuCollation::getUnicodeVersionForICU(); $error = ""; if ( !$allkeysPresent ) { $error .= "Unable to find allkeys.txt. " . "Download it and specify its location with --data-dir=. " . "\n\n"; } if ( !$ucdallPresent ) { $error .= "Unable to find ucd.all.grouped.xml. " . "Download it, unzip, and specify its location with --data-dir=. " . "\n\n"; } $versionKnown = false; if ( version_compare( $icuVersion, "4.0", "<" ) ) { // Extra old version $error .= "You are using outdated version of ICU ($icuVersion), intended for " . ( $unicodeVersion ? "Unicode $unicodeVersion" : "an unknown version of Unicode" ) . "; this file might not be available for it, and it's not supported by MediaWiki. " . " You are on your own; consider upgrading PHP's intl extension or try " . "one of the files available at:"; } elseif ( version_compare( $icuVersion, "51.0", ">=" ) ) { // Extra recent version $error .= "You are using ICU $icuVersion, released after this script was last updated. " . "Check what is the Unicode version it is using at http://site.icu-project.org/download . " . "It can't be guaranteed everything will work, but appropriate file(s) should " . "be available at:"; } else { // ICU 4.0 to 50.x $versionKnown = true; $error .= "You are using ICU $icuVersion, intended for " . ( $unicodeVersion ? "Unicode $unicodeVersion" : "an unknown version of Unicode" ) . ". Appropriate file(s) should be available at:"; } $error .= "\n"; if ( $versionKnown && $unicodeVersion ) { $allkeysURL = str_replace( "", "$unicodeVersion.0", $allkeysURL ); $ucdallURL = str_replace( "", "$unicodeVersion.0", $ucdallURL ); } if ( !$allkeysPresent ) { $error .= "* $allkeysURL\n"; } if ( !$ucdallPresent ) { $error .= "* $ucdallURL\n"; } $this->fatalError( $error ); } $debugOutFileName = $this->getOption( 'debug-output' ); if ( $debugOutFileName ) { $this->debugOutFile = fopen( $debugOutFileName, 'w' ); if ( !$this->debugOutFile ) { $this->fatalError( "Unable to open debug output file for writing" ); } } $this->loadUcd(); $this->generateFirstChars(); } private function loadUcd() { $uxr = new UcdXmlReader( "{$this->dataDir}/ucd.all.grouped.xml" ); $uxr->readChars( [ $this, 'charCallback' ] ); } private function charCallback( $data ) { // Skip non-printable characters, // but do not skip a normal space (U+0020) since // people like to use that as a fake no header symbol. $category = substr( $data['gc'], 0, 1 ); if ( strpos( 'LNPS', $category ) === false && $data['cp'] !== '0020' ) { return; } $cp = hexdec( $data['cp'] ); // Skip the CJK ideograph blocks, as an optimisation measure. // UCA doesn't sort them properly anyway, without tailoring. if ( IcuCollation::isCjk( $cp ) ) { return; } // Skip the composed Hangul syllables, we will use the bare Jamo // as first letters if ( $data['block'] == 'Hangul Syllables' ) { return; } // Calculate implicit weight per UTS #10 v6.0.0, sec 7.1.3 if ( $data['UIdeo'] === 'Y' ) { if ( $data['block'] == 'CJK Unified Ideographs' || $data['block'] == 'CJK Compatibility Ideographs' ) { $base = 0xFB40; } else { $base = 0xFB80; } } else { $base = 0xFBC0; } $a = $base + ( $cp >> 15 ); $b = ( $cp & 0x7fff ) | 0x8000; $this->weights[$cp] = sprintf( ".%04X.%04X", $a, $b ); if ( $data['dm'] !== '#' ) { $this->mappedChars[$cp] = true; } if ( $cp % 4096 == 0 ) { print "{$data['cp']}\n"; } } private function generateFirstChars() { $file = fopen( "{$this->dataDir}/allkeys.txt", 'r' ); if ( !$file ) { $this->fatalError( "Unable to open allkeys.txt" ); } $goodTertiaryChars = []; // For each character with an entry in allkeys.txt, overwrite the implicit // entry in $this->weights that came from the UCD. // Also gather a list of tertiary weights, for use in selecting the group header while ( ( $line = fgets( $file ) ) !== false ) { // We're only interested in single-character weights, pick them out with a regex $line = trim( $line ); if ( !preg_match( '/^([0-9A-F]+)\s*;\s*([^#]*)/', $line, $m ) ) { continue; } $cp = hexdec( $m[1] ); $allWeights = trim( $m[2] ); $primary = ''; $tertiary = ''; if ( !isset( $this->weights[$cp] ) ) { // Non-printable, ignore continue; } foreach ( StringUtils::explode( '[', $allWeights ) as $weightStr ) { preg_match_all( '/[*.]([0-9A-F]+)/', $weightStr, $m ); if ( !empty( $m[1] ) ) { if ( $m[1][0] !== '0000' ) { $primary .= '.' . $m[1][0]; } if ( $m[1][2] !== '0000' ) { $tertiary .= '.' . $m[1][2]; } } } $this->weights[$cp] = $primary; if ( $tertiary === '.0008' || $tertiary === '.000E' ) { $goodTertiaryChars[$cp] = true; } } fclose( $file ); // Identify groups of characters with the same primary weight $this->groups = []; asort( $this->weights, SORT_STRING ); $prevWeight = reset( $this->weights ); $group = []; foreach ( $this->weights as $cp => $weight ) { if ( $weight !== $prevWeight ) { $this->groups[$prevWeight] = $group; $prevWeight = $weight; $group = $this->groups[$weight] ?? []; } $group[] = $cp; } if ( $group ) { $this->groups[$prevWeight] = $group; } // If one character has a given primary weight sequence, and a second // character has a longer primary weight sequence with an initial // portion equal to the first character, then remove the second // character. This avoids having characters like U+A732 (double A) // polluting the basic Latin sort area. foreach ( $this->groups as $weight => $group ) { if ( preg_match( '/(\.[0-9A-F]*)\./', $weight, $m ) ) { if ( isset( $this->groups[$m[1]] ) ) { unset( $this->groups[$weight] ); } } } ksort( $this->groups, SORT_STRING ); // Identify the header character in each group $headerChars = []; $prevChar = "\000"; $tertiaryCollator = new Collator( 'root' ); $primaryCollator = new Collator( 'root' ); $primaryCollator->setStrength( Collator::PRIMARY ); $numOutOfOrder = 0; foreach ( $this->groups as $weight => $group ) { $uncomposedChars = []; $goodChars = []; foreach ( $group as $cp ) { if ( isset( $goodTertiaryChars[$cp] ) ) { $goodChars[] = $cp; } if ( !isset( $this->mappedChars[$cp] ) ) { $uncomposedChars[] = $cp; } } $x = array_intersect( $goodChars, $uncomposedChars ); if ( !$x ) { $x = $uncomposedChars; if ( !$x ) { $x = $group; } } // Use ICU to pick the lowest sorting character in the selection $tertiaryCollator->sort( $x ); $cp = $x[0]; $char = UtfNormal\Utils::codepointToUtf8( $cp ); $headerChars[] = $char; if ( $primaryCollator->compare( $char, $prevChar ) <= 0 ) { $numOutOfOrder++; } $prevChar = $char; if ( $this->debugOutFile ) { fwrite( $this->debugOutFile, sprintf( "%05X %s %s (%s)\n", $cp, $weight, $char, implode( ' ', array_map( 'UtfNormal\Utils::codepointToUtf8', $group ) ) ) ); } } print "Out of order: $numOutOfOrder / " . count( $headerChars ) . "\n"; global $IP; $writer = new StaticArrayWriter(); file_put_contents( "$IP/includes/collation/data/first-letters-root.php", $writer->create( $headerChars, 'File created by generateCollationData.php' ) ); echo "first-letters-root: file written.\n"; } } class UcdXmlReader { public $fileName; public $callback; public $groupAttrs; public $xml; public $blocks = []; public $currentBlock; public function __construct( $fileName ) { $this->fileName = $fileName; } public function readChars( $callback ) { $this->getBlocks(); $this->currentBlock = reset( $this->blocks ); $xml = $this->open(); $this->callback = $callback; while ( $xml->name !== 'repertoire' && $xml->next() ); while ( $xml->read() ) { if ( $xml->nodeType == XMLReader::ELEMENT ) { if ( $xml->name === 'group' ) { $this->groupAttrs = $this->readAttributes(); } elseif ( $xml->name === 'char' ) { $this->handleChar(); } } elseif ( $xml->nodeType === XMLReader::END_ELEMENT ) { if ( $xml->name === 'group' ) { $this->groupAttrs = []; } } } $xml->close(); } protected function open() { $this->xml = new XMLReader; $this->xml->open( $this->fileName ); if ( !$this->xml ) { throw new MWException( __METHOD__ . ": unable to open {$this->fileName}" ); } while ( $this->xml->name !== 'ucd' && $this->xml->read() ); $this->xml->read(); return $this->xml; } /** * Read the attributes of the current element node and return them * as an array * @return array */ protected function readAttributes() { $attrs = []; while ( $this->xml->moveToNextAttribute() ) { $attrs[$this->xml->name] = $this->xml->value; } return $attrs; } protected function handleChar() { $attrs = $this->readAttributes() + $this->groupAttrs; if ( isset( $attrs['cp'] ) ) { $first = $last = hexdec( $attrs['cp'] ); } else { $first = hexdec( $attrs['first-cp'] ); $last = hexdec( $attrs['last-cp'] ); unset( $attrs['first-cp'] ); unset( $attrs['last-cp'] ); } for ( $cp = $first; $cp <= $last; $cp++ ) { $hexCp = sprintf( "%04X", $cp ); foreach ( [ 'na', 'na1' ] as $nameProp ) { if ( isset( $attrs[$nameProp] ) ) { $attrs[$nameProp] = str_replace( '#', $hexCp, $attrs[$nameProp] ); } } while ( $this->currentBlock ) { if ( $cp < $this->currentBlock[0] ) { break; } elseif ( $cp <= $this->currentBlock[1] ) { $attrs['block'] = key( $this->blocks ); break; } else { $this->currentBlock = next( $this->blocks ); } } $attrs['cp'] = $hexCp; call_user_func( $this->callback, $attrs ); } } public function getBlocks() { if ( $this->blocks ) { return $this->blocks; } $xml = $this->open(); while ( $xml->name !== 'blocks' && $xml->read() ); while ( $xml->read() ) { if ( $xml->nodeType == XMLReader::ELEMENT ) { if ( $xml->name === 'block' ) { $attrs = $this->readAttributes(); $first = hexdec( $attrs['first-cp'] ); $last = hexdec( $attrs['last-cp'] ); $this->blocks[$attrs['name']] = [ $first, $last ]; } } } $xml->close(); return $this->blocks; } } $maintClass = GenerateCollationData::class; require_once RUN_MAINTENANCE_IF_MAIN;