SemanticResultFormats/Math/SRF_Math.php

Go to the documentation of this file.

<?php

class SRFMath extends SMWResultPrinter {

        public function getName() {
                return wfMsg( 'srf_printername_' . $this->mFormat );
        }

        public function getResult( SMWQueryResult $results, array $params, $outputmode ) {
                $this->handleParameters( $params, $outputmode );
                global $wgLang;
                return $wgLang->formatNum( $this->getResultText( $results, SMW_OUTPUT_HTML ) );
        }

        protected function getResultText( SMWQueryResult $res, $outputmode ) {
                $numbers = $this->getNumbers( $res );
                
                if ( count( $numbers ) == 0 ) {
                        return $this->params['default'];
                }
                
                switch ( $this->mFormat ) {
                        case 'max':
                                return max( $numbers );
                                break;
                        case 'min':
                                return min( $numbers );
                                break;
                        case 'sum':
                                return array_sum( $numbers );
                                break;
                        case 'product':
                                return array_product( $numbers );
                                break;
                        case 'average':
                                return array_sum( $numbers ) / count( $numbers );
                                break;
                        case 'median':
                                sort( $numbers, SORT_NUMERIC );
                                $position = ( count( $numbers ) + 1 ) / 2 - 1;
                                return ( $numbers[ceil( $position )] + $numbers[floor( $position )] ) / 2; 
                                break;
                }
        }
        
        protected function getNumbers( SMWQueryResult $res ) {
                $numbers = array();

                while ( $row = $res->getNext() ) {
                        foreach( $row as /* SMWResultArray */ $resultArray ) {
                                foreach ( $resultArray->getContent() as /* SMWDataItem */ $dataItem ) {
                                        self::addNumbersForDataItem( $dataItem, $numbers );
                                }
                        }
                }

                return $numbers;
        }
        
        public static function addNumbersForDataItem( SMWDataItem $dataItem, array &$numbers ) {
                switch ( $dataItem->getDIType() ) {
                        case SMWDataItem::TYPE_NUMBER:
                                $numbers[] = $dataItem->getNumber();
                                break;
                        case SMWDataItem::TYPE_CONTAINER:
                                foreach ( $dataItem->getDataItems() as $di ) {
                                        self::addNumbersForDataItem( $di, $numbers );
                                }
                                break;
                        default:
                }
        }

        public function getParameters() {
                $params = parent::getParameters();
                
                $params['limit'] = new Parameter( 'limit', Parameter::TYPE_INTEGER );
                $params['limit']->setMessage( 'srf_paramdesc_limit' );
                $params['limit']->setDefault( 1000 );

                $params['default'] = new Parameter( 'default' );
                $params['default']->setMessage( 'srf-paramdesc-default' );
                $params['default']->setDefault( '' );
                
                return $params;
        }

}

---