generateCollationData.php

Go to the documentation of this file.

<?php
// @codeCoverageIgnoreStart
require_once __DIR__ . '/../Maintenance.php';
// @codeCoverageIgnoreEnd
 
use MediaWiki\Collation\IcuCollation;
use MediaWiki\Maintenance\Maintenance;
use Wikimedia\StaticArrayWriter;
use Wikimedia\StringUtils\StringUtils;
 
class GenerateCollationData extends Maintenance {
    public $dataDir;
 
    public $weights;
 
    public $mappedChars;
 
    public $debugOutFile;
 
    private $groups;
 
    public function __construct() {
        parent::__construct();
        $this->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" );
 
        if ( !$allkeysPresent || !$ucdallPresent ) {
            $icuVersion = INTL_ICU_VERSION;
            $unicodeVersion = implode( '.', array_slice( IntlChar::getUnicodeVersion(), 0, 3 ) );
 
            $error = "";
 
            if ( !$allkeysPresent ) {
                $error .= "Unable to find allkeys.txt. "
                    . "Download it and specify its location with --data-dir=<DIR>. "
                    . "\n\n";
            }
            if ( !$ucdallPresent ) {
                $error .= "Unable to find ucd.all.grouped.xml. "
                    . "Download it, unzip, and specify its location with --data-dir=<DIR>. "
                    . "\n\n";
            }
 
            $error .= "You are using ICU $icuVersion, intended for Unicode $unicodeVersion. "
                . "Appropriate file(s) should be available at:\n";
 
            $allkeysURL = "https://www.unicode.org/Public/UCA/$unicodeVersion/allkeys.txt";
            $ucdallURL = "https://www.unicode.org/Public/$unicodeVersion/ucdxml/ucd.all.grouped.zip";
 
            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( array $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 );
        // @phan-suppress-next-line PhanParamSuspiciousOrder False positive
        if ( !str_contains( 'LNPS', $category )
            && $data['cp'] !== '0020'
        ) {
            return;
        }
        $cp = (int)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;
        }
 
        // Skip characters that mapped to a single character we skipped above.
        // e.g. U+2329 -> U+3008 (from CJK Symbols and Punctuation)
        if ( $data['dm'] !== '#' && !str_contains( $data['dm'], ' ' ) &&
            !isset( $this->weights[ hexdec( $data['dm'] ) ] )
        ) {
            return;
        }
 
        // Calculate implicit weight per UTS #10 v6.0.0, sec 7.1.3
        $a = 0xFBC0 + ( $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
        // phpcs:ignore Generic.CodeAnalysis.AssignmentInCondition.FoundInWhileCondition
        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 = (int)hexdec( $m[1] );
            $allWeights = trim( $m[2] );
            $primary = '';
            $tertiary = '';
 
            if ( !isset( $this->weights[$cp] ) ) {
                // Non-printable, ignore
                continue;
            }
            foreach ( StringUtils::explode( '[', $allWeights ) as $weightStr ) {
                if ( preg_match_all( '/[*.]([0-9A-F]+)/', $weightStr, $m ) ) {
                    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::class, 'codepointToUtf8' ], $group ) ) ) );
            }
        }
 
        print "Out of order: $numOutOfOrder / " . count( $headerChars ) . "\n";
 
        $writer = new StaticArrayWriter();
        file_put_contents(
            MW_INSTALL_PATH . '/languages/data/first-letters-root.php',
            $writer->create(
                $headerChars,
                "File created by maintenance/language/generateCollationData.php\n"
                    . "@codeCoverageIgnore"
            )
        );
        echo "first-letters-root: file written.\n";
    }
}
 
class UcdXmlReader {
    public $fileName;
    public $callback;
    public $groupAttrs;
    public $xml;
    public $blocks = [];
    public $currentBlock;
 
    public function __construct( string $fileName ) {
        $this->fileName = $fileName;
    }
 
    public function readChars( callable $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(): XMLReader {
        $this->xml = new XMLReader;
        if ( !$this->xml->open( $this->fileName ) ) {
            throw new RuntimeException( __METHOD__ . ": unable to open {$this->fileName}" );
        }
        while ( $this->xml->name !== 'ucd' && $this->xml->read() );
        $this->xml->read();
 
        return $this->xml;
    }
 
    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;
            ( $this->callback )( $attrs );
        }
    }
 
    public function getBlocks(): array {
        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;
    }
}
 
// @codeCoverageIgnoreStart
$maintClass = GenerateCollationData::class;
require_once RUN_MAINTENANCE_IF_MAIN;
// @codeCoverageIgnoreEnd