<?php
/* vim: set expandtab sw=4 ts=4 sts=4: */
/**
 *
 * @version $Id$
 * @package phpMyAdmin
 */

include_once("Export_Relation_Schema.class.php");

/**
 * This Class inherits the XMLwriter class and
 * helps in developing structure of SVG Schema
 * 
 *
 * @name PMA_SVG
 * @author Muhammad Adnan <hiddenpearls@gmail.com>
 * @copyright
 * @license
 * @see http://php.net/manual/en/book.xmlwriter.php
 */

class PMA_SVG extends XMLWriter
{
    public $title;
    public $author;
    public $font;
    public $fontSize;

    function __construct()
    {
        $this->openMemory();
        /*
         * Set indenting using three spaces,
         * so output is formatted
         */

        $this->setIndent(TRUE);
        $this->setIndentString('   ');
        /*
         * Create the XML document
         */

        $this->startDocument('1.0','UTF-8');
        $this->startDtd('svg','-//W3C//DTD SVG 1.1//EN','http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd');
        $this->endDtd();        
    }

    function setTitle($title)
    {
        $this->title=$title;
    }
    
    function setAuthor($author)
    {
        $this->author=$author;
    }

    function setFont($font)
    {
        $this->font=$font;
    }
    function getFont()
    {
        return $this->font;
    }

    function setFontSize($fontSize)
    {
        $this->fontSize=$fontSize;
    }

    function getFontSize()
    {
        return $this->fontSize;
    }

    function startSvgDoc($width,$height)
    {
        $this->startElement('svg');
        $this->writeAttribute('width', $width);
        $this->writeAttribute('height', $height);
        $this->writeAttribute('xmlns', 'http://www.w3.org/2000/svg');
        $this->writeAttribute('version', '1.1');
    }

    function endSvgDoc()
    {
        $this->endElement();
        $this->endDocument();
    }

    function showOutput()
    {
        if(ob_end_clean()){
            ob_end_clean();
            //ob_start();
        }
        header('Content-type: image/svg+xml');
        $output = $this->flush();
        print $output;
    }

    function printElement($name,$x,$y,$width='',$height='',$text='',$styles='')
    {
        $this->startElement($name);
        $this->writeAttribute('width',$width);
        $this->writeAttribute('height',$height);
        $this->writeAttribute('x', $x);
        $this->writeAttribute('y', $y);
        $this->writeAttribute('style', $styles);
        if(isset($text)){
            $this->writeAttribute('font-family', $this->font);
            $this->writeAttribute('font-size', $this->fontSize);
            $this->text($text);
        }
        $this->endElement();
    }

    function printElementLine($name,$x1,$y1,$x2,$y2,$styles)
    {
        $this->startElement($name);
        $this->writeAttribute('x1',$x1);
        $this->writeAttribute('y1',$y1);
        $this->writeAttribute('x2', $x2);
        $this->writeAttribute('y2', $y2);
        $this->writeAttribute('style', $styles);
        $this->endElement();
    }

    function getStringWidth($text,$font,$fontSize)
    {
        /*
         * Start by counting the width, giving each character a modifying value
         */
        $count = 0;
        $count = $count + ((strlen($text) - strlen(str_replace(array("i","j","l"),"",$text)))*0.23);//ijl
        $count = $count + ((strlen($text) - strlen(str_replace(array("f"),"",$text)))*0.27);//f
        $count = $count + ((strlen($text) - strlen(str_replace(array("t","I"),"",$text)))*0.28);//tI
        $count = $count + ((strlen($text) - strlen(str_replace(array("r"),"",$text)))*0.34);//r
        $count = $count + ((strlen($text) - strlen(str_replace(array("1"),"",$text)))*0.49);//1
        $count = $count + ((strlen($text) - strlen(str_replace(array("c","k","s","v","x","y","z","J"),"",$text)))*0.5);//cksvxyzJ
        $count = $count + ((strlen($text) - strlen(str_replace(array("a","b","d","e","g","h","n","o","p","q","u","L","0","2","3","4","5","6","7","8","9"),"",$text)))*0.56);//abdeghnopquL023456789
        $count = $count + ((strlen($text) - strlen(str_replace(array("F","T","Z"),"",$text)))*0.61);//FTZ
        $count = $count + ((strlen($text) - strlen(str_replace(array("A","B","E","K","P","S","V","X","Y"),"",$text)))*0.67);//ABEKPSVXY
        $count = $count + ((strlen($text) - strlen(str_replace(array("w","C","D","H","N","R","U"),"",$text)))*0.73);//wCDHNRU
        $count = $count + ((strlen($text) - strlen(str_replace(array("G","O","Q"),"",$text)))*0.78);//GOQ
        $count = $count + ((strlen($text) - strlen(str_replace(array("m","M"),"",$text)))*0.84);//mM
        $count = $count + ((strlen($text) - strlen(str_replace("W","",$text)))*.95);//W
        $count = $count + ((strlen($text) - strlen(str_replace(" ","",$text)))*.28);//" "
        $text = str_replace(" ","",$text);//remove the " "'s
        $count = $count + (strlen(preg_replace("/[a-z0-9]/i","",$text))*0.3); //all other chrs

        $modifier = 1;
        $font = strtolower($font);
        switch($font){
        /*
         * no modifier for arial and sans-serif
         */
        case 'arial':
        case 'sans-serif':
        break;
        /*
         * .92 modifer for time, serif, brushscriptstd, and californian fb
         */
        case 'times':
        case 'serif':
        case 'brushscriptstd':
        case 'californian fb':
            $modifier = .92;
        break;
        /*
         * 1.23 modifier for broadway
         */
        case 'broadway':
            $modifier = 1.23;
        break;
        }
        $textWidth = $count*$fontSize;
        return ceil($textWidth*$modifier);
    }
}
/**
 * Draws tables schema
 */
class Table_Stats {
    /**
     * Defines properties
     */

    private $_tableName;
    private $_showInfo = false;

    public $width = 0;
    public $height;
    public $fields = array();
    public $heightCell = 0;
    public $x, $y;
    public $primary = array();

    
    /**
     * The "Table_Stats" constructor
     *
     * @param string table_name The table name
     * @param integer ff The font size
     * @param integer samewidth The max. with among tables
     * @param boolean show_keys Whether to display keys or not
     * @param boolean show_info Whether to display table position or not
     * @global object    The current SVG image document
     * @global integer   The current page number (from the
     *                     $cfg['Servers'][$i]['table_coords'] table)
     * @global array     The relations settings
     * @global string    The current db name
     * @access private
     * @see PMA_PDF, Table_Stats::Table_Stats_setWidth,
            Table_Stats::Table_Stats_setHeight
     */
    function __construct($tableName, $font, $fontSize, &$same_wide_width, $showKeys = false, $showInfo = false)
    {
        global $svg, $pdf_page_number, $cfgRelation, $db;

        $this->_tableName = $tableName;
        $sql = 'DESCRIBE ' . PMA_backquote($tableName);
        $result = PMA_DBI_try_query($sql, null, PMA_DBI_QUERY_STORE);
        if (!$result || !PMA_DBI_num_rows($result)) {
            $svg->dieSchema(sprintf(__('The %s table doesn\'t exist!'), $tableName));
        }

        /*
        * load fields
        * check to see if it will load all fields or only the foreign keys
        */
        
        if ($showKeys) {
            $indexes = PMA_Index::getFromTable($this->_tableName, $db);
            $all_columns = array();
            foreach ($indexes as $index) {
            $all_columns = array_merge($all_columns, array_flip(array_keys($index->getColumns())));
            }
            $this->fields = array_keys($all_columns);
        } else {
            while ($row = PMA_DBI_fetch_row($result)) {
                $this->fields[] = $row[0];
            }
        }

        $this->_showInfo = $showInfo;

        // height and width
        $this->_setHeightTable($fontSize);

        // setWidth must me after setHeight, because title
        // can include table height which changes table width
        $this->_setWidthTable($font,$fontSize);
        if ($same_wide_width < $this->width) {
            $same_wide_width = $this->width;
        }

        // x and y
        $sql = 'SELECT x, y FROM '
         . PMA_backquote($GLOBALS['cfgRelation']['db']) . '.' . PMA_backquote($cfgRelation['table_coords'])
         . ' WHERE db_name = \'' . PMA_sqlAddslashes($db) . '\''
         . ' AND   table_name = \'' . PMA_sqlAddslashes($tableName) . '\''
         . ' AND   pdf_page_number = ' . $pdf_page_number;
        $result = PMA_query_as_controluser($sql, false, PMA_DBI_QUERY_STORE);

        if (!$result || !PMA_DBI_num_rows($result)) {
            //$svg->dieSchema(sprintf(__('Please configure the coordinates for table %s'), $tableName));
        }
        list($this->x, $this->y) = PMA_DBI_fetch_row($result);
        $this->x = (double) $this->x;
        $this->y = (double) $this->y;
        // displayfield
        $this->displayfield = PMA_getDisplayField($db, $tableName);
        // index
        $result = PMA_DBI_query('SHOW INDEX FROM ' . PMA_backquote($tableName) . ';', null, PMA_DBI_QUERY_STORE);
        if (PMA_DBI_num_rows($result) > 0) {
            while ($row = PMA_DBI_fetch_assoc($result)) {
                if ($row['Key_name'] == 'PRIMARY') {
                    $this->primary[] = $row['Column_name'];
                }
            }
        }
    }

    /**
     * Returns title of the current table,
     * title can have the dimensions/co-ordinates of the table
     *
     * @access private
     */
    function _getTitle()
    {
        return ($this->_showInfo ? sprintf('%.0f', $this->width) . 'x' . sprintf('%.0f', $this->heightCell) : '') . ' ' . $this->_tableName;
    }

    /**
     * Sets the width of the table
     *
     * @param string font The font size
     * @param integer fontSize The font size
     * @global object    The current SVG image document
     * @access private
     * @see PMA_SVG
     */
    function _setWidthTable($font,$fontSize)
    {
        global $svg;

        foreach ($this->fields as $field) {
            $this->width = max($this->width, $svg->getStringWidth($field,$font,$fontSize));
        }
        $this->width += $svg->getStringWidth('  ',$font,$fontSize);
        /*
         * it is unknown what value must be added, because
         * table title is affected by the tabe width value
         */ 
        while ($this->width < $svg->GetStringWidth($this->_getTitle(),$font,$fontSize)) {
            $this->width += 7;
        }
    }

    /**
     * Sets the height of the table
     *
     * @access private
     */
    function _setHeightTable($fontSize)
    {
        $this->heightCell = $fontSize + 4;
        $this->height = (count($this->fields) + 1) * $this->heightCell;
    }

    /**
     * draw the table
     *
     * @param boolean showColor Whether to display color
     * @global object The current SVG image document
     * @access public
     * @see PMA_SVG,PMA_SVG::printElement
     */
    public function tableDraw($showColor)
    {
        global $svg;
        //echo $this->_tableName.'<br />';
        $svg->printElement('rect',$this->x,$this->y,
            $this->width,$this->heightCell,
            NULL,'fill:red;stroke:black;'
            );
        $svg->printElement('text',$this->x + 5,$this->y+ 14,
            $this->width,$this->heightCell,
            $this->_getTitle(),
            'fill:none;stroke:black;'
            );
        foreach ($this->fields as $field) {
                $currentCell+=$this->heightCell;
                $showColor='none';
                if ($showColor) {
                    if (in_array($field, $this->primary)) {
                        $showColor='#0c0';
                    }
                    if ($field == $this->displayfield) {
                        $showColor='none';
                    }
                }
                $svg->printElement('rect', $this->x,$this->y  + $currentCell,
                    $this->width, $this->heightCell,
                    NULL,
                    'fill:'.$showColor.';stroke:black;'
                    );
                $svg->printElement('text', $this->x + 5, $this->y + 14 + $currentCell,
                    $this->width, $this->heightCell,
                    $field,
                    'fill:none;stroke:black;'
                    );
        //echo $field.'<br />';
        }
    }
}


/**
 * Draws relation links
 *
 * @access public
 * @see PMA_SVG,PMA_SVG::printElementLine
 */
class Relation_Stats {
    /**
     * Defines properties
     */
    public $xSrc, $ySrc;
    public $srcDir ;
    public $destDir;
    public $xDest, $yDest;
    public $wTick = 10;

    /**
     * The "Relation_Stats" constructor
     *
     * @param string master_table The master table name
     * @param string master_field The relation field in the master table
     * @param string foreign_table The foreign table name
     * @param string foreigh_field The relation field in the foreign table
     * @access private
     * @see Relation_Stats::_getXy
     */
    function __construct($master_table, $master_field, $foreign_table, $foreign_field)
    {
        $src_pos = $this->_getXy($master_table, $master_field);
        $dest_pos = $this->_getXy($foreign_table, $foreign_field);
        /*
        * [0] is x-left
        * [1] is x-right
        * [2] is y
        */
        $src_left = $src_pos[0] - $this->wTick;
        $src_right = $src_pos[1] + $this->wTick;
        $dest_left = $dest_pos[0] - $this->wTick;
        $dest_right = $dest_pos[1] + $this->wTick;

        $d1 = abs($src_left - $dest_left);
        $d2 = abs($src_right - $dest_left);
        $d3 = abs($src_left - $dest_right);
        $d4 = abs($src_right - $dest_right);
        $d = min($d1, $d2, $d3, $d4);

        if ($d == $d1) {
            $this->xSrc = $src_pos[0];
            $this->srcDir = -1;
            $this->xDest = $dest_pos[0];
            $this->destDir = -1;
        } elseif ($d == $d2) {
            $this->xSrc = $src_pos[1];
            $this->srcDir = 1;
            $this->xDest = $dest_pos[0];
            $this->destDir = -1;
        } elseif ($d == $d3) {
            $this->xSrc = $src_pos[0];
            $this->srcDir = -1;
            $this->xDest = $dest_pos[1];
            $this->destDir = 1;
        } else {
            $this->xSrc = $src_pos[1];
            $this->srcDir = 1;
            $this->xDest = $dest_pos[1];
            $this->destDir = 1;
        }
        $this->ySrc = $src_pos[2];
        $this->y_dest = $dest_pos[2];
    }

    /**
     * Gets arrows coordinates
     *
     * @param string table The current table name
     * @param string column The relation column name
     * @return array Arrows coordinates
     * @access private
     */
    function _getXy($table, $column)
    {
        $pos = array_search($column, $table->fields);
        // x_left, x_right, y
        return array($table->x, $table->x + $table->width, $table->y + ($pos + 1.5) * $table->heightCell);
    }

    /**
     * draws relation links and arrows
     * shows foreign key relations
     *
     * @param boolean changeColor Whether to use one color per relation or not
     * @global object    The current SVG image document
     * @access public
     * @see PMA_SVG
     */
    public function relationDraw($changeColor)
    {
        global $svg;

        if ($changeColor) {
            $listOfColors=array(
                'red',
                'grey',
                'black',
                'yellow',
                'green',
                'cyan',
            '    orange'
            );
            shuffle($listOfColors);
            $color =  $listOfColors[0];
        } else {
            $color = 'black';
        }

        $svg->printElementLine('line',$this->xSrc,$this->ySrc,
            $this->xSrc + $this->srcDir * $this->wTick,$this->ySrc,
            'fill:'.$color.';stroke:black;stroke-width:2;'
            );
        $svg->printElementLine('line',$this->xDest + $this->destDir * $this->wTick, $this->y_dest,
            $this->xDest, $this->y_dest,
            'fill:'.$color.';stroke:black;stroke-width:2;'
            );
        $svg->printElementLine('line',$this->xSrc + $this->srcDir * $this->wTick,$this->ySrc,
            $this->xDest + $this->destDir * $this->wTick, $this->y_dest,
            'fill:'.$color.';stroke:'.$color.';stroke-width:1;'
            );
        $root2 = 2 * sqrt(2);
        $svg->printElementLine('line',$this->xSrc + $this->srcDir * $this->wTick * 0.75, $this->ySrc,
            $this->xSrc + $this->srcDir * (0.75 - 1 / $root2) * $this->wTick ,
            $this->ySrc + $this->wTick / $root2 ,
            'fill:'.$color.';stroke:black;stroke-width:2;'
            );
        $svg->printElementLine('line',$this->xSrc + $this->srcDir * $this->wTick * 0.75, $this->ySrc,
            $this->xSrc + $this->srcDir * (0.75 - 1 / $root2) * $this->wTick ,
            $this->ySrc - $this->wTick / $root2 ,
            'fill:'.$color.';stroke:black;stroke-width:2;'
            );
        $svg->printElementLine('line',$this->xDest + $this->destDir * $this->wTick / 2 , $this->y_dest ,
            $this->xDest + $this->destDir * (0.5 + 1 / $root2) * $this->wTick,
            $this->y_dest + $this->wTick / $root2 ,
            'fill:'.$color.';stroke:black;stroke-width:2;');
        $svg->printElementLine('line',$this->xDest + $this->destDir * $this->wTick / 2 ,
            $this->y_dest , $this->xDest + $this->destDir * (0.5 + 1 / $root2) * $this->wTick ,
            $this->y_dest - $this->wTick / $root2 ,
            'fill:'.$color.';stroke:black;stroke-width:2;'
            );
    }
} 
/*
* end of the "Relation_Stats" class
*/


class PMA_Svg_Relation_Schema extends PMA_Export_Relation_Schema
{

    private $tables = array();
    private $_relations = array();
    private $_xMax = 0;
    private $_yMax = 0;
    private $scale;
    private $_xMin = 100000;
    private $_yMin = 100000;
    private $t_marg = 10;
    private $b_marg = 10;
    private $l_marg = 10;
    private $r_marg = 10;
    private $_tablewidth;
    
    function __construct()
    {
        global $svg,$db,$pdf_page_number,$show_info,$show_table_dimension,$show_color,$change_color,$all_table_same_wide;

        $pdf_page_number            =  isset($pdf_page_number) ? $pdf_page_number : 1;
        $show_info              = (isset($show_table_dimension) && $show_table_dimension == 'on') ? 1 : 0;
        $all_table_same_wide    = (isset($all_table_same_wide) && $all_table_same_wide == 'on') ? 1 : 0;
        $change_color             = (isset($show_color) && $show_color == 'on') ? 1 : 0;
        $show_keys              = (isset($show_keys) && $show_keys == 'on') ? 1 : 0;
        $svg = new PMA_SVG();
        $this->setSameWidthTables($all_table_same_wide);
        $svg->setTitle(sprintf(__('Schema of the %s database - Page %s'), $db, $pdf_page_number));
        $svg->SetAuthor('phpMyAdmin ' . PMA_VERSION);
        $svg->setFont('Arial');
        $svg->setFontSize('16px');
        $svg->startSvgDoc('1000px','600px');
        $alltables=$this->getAllTables($db,$pdf_page_number);

        foreach ($alltables AS $table) {
            if (!isset($this->tables[$table])) {
                $this->tables[$table] = new Table_Stats($table,$svg->getFont(),$svg->getFontSize(), $this->_tablewidth, $show_keys, $show_info);
            }

            if ($this->sameWide) {
                $this->tables[$table]->width = $this->_tablewidth;
            }
           $this->_setMinMax($this->tables[$table]);
        }
        $seen_a_relation = false;
        foreach ($alltables as $one_table) {
            $exist_rel = PMA_getForeigners($db, $one_table, '', 'both');
            if ($exist_rel) {
                $seen_a_relation = true;
                foreach ($exist_rel as $master_field => $rel) {
                    /* put the foreign table on the schema only if selected
                    * by the user
                    * (do not use array_search() because we would have to
                    * to do a === FALSE and this is not PHP3 compatible)
                    */
                    if (in_array($rel['foreign_table'], $alltables)) {
                        $this->_addRelation($one_table,$svg->getFont(),$svg->getFontSize(), $master_field, $rel['foreign_table'], $rel['foreign_field'], $show_info);
                    }
                }
            }
        }

        if ($seen_a_relation) {
            $this->_drawRelations($change_color);
        }

        $this->_drawTables($change_color);
        $svg->endSvgDoc();
        $svg->showOutput();
      //  print '<pre>';
        //print_r(get_object_vars($svg));
        //print_r($alltables);
        //print_r($this);
        //print '</pre>';
        //exit();
    }

    /**
     * Sets X and Y minimum and maximum for a table cell
     *
     * @param string table The table name
     * @access private
     */
    function _setMinMax($table)
    {
        $this->_xMax = max($this->_xMax, $table->x + $table->width);
        $this->_yMax = max($this->_yMax, $table->y + $table->height);
        $this->_xMin = min($this->_xMin, $table->x);
        $this->_yMin = min($this->_yMin, $table->y);
    }

    /**
     * Defines relation objects
     *
     * @param string masterTable The master table name
     * @param string masterField The relation field in the master table
     * @param string foreignTable The foreign table name
     * @param string foreignField The relation field in the foreign table
     * @param boolean showInfo Whether to display table position or not
     * @access private
     * @see _setMinMax,Table_Stats::__construct(),Relation_Stats::__construct()
     */
    function _addRelation($masterTable,$font,$fontSize, $masterField, $foreignTable, $foreignField, $showInfo)
    {
        if (!isset($this->tables[$masterTable])) {
            $this->tables[$masterTable] = new Table_Stats($masterTable,$font,$fontSize, $this->_tablewidth, false, $showInfo);
            $this->_setMinMax($this->tables[$masterTable]);
        }
        if (!isset($this->tables[$foreignTable])) {
            $this->tables[$foreignTable] = new Table_Stats($foreignTable,$font,$fontSize, $this->_tablewidth, false, $showInfo);
            $this->_setMinMax($this->tables[$foreignTable]);
        }
        $this->_relations[] = new Relation_Stats($this->tables[$masterTable], $masterField, $this->tables[$foreignTable], $foreignField);
    }
    
    /**
     * Draws relation arrows and lines
     * connects master table's master field to 
     * foreign table's forein field
     *
     * @param boolean changeColor Whether to use one color per relation or not
     * @access private
     * @see Relation_Stats::relationDraw()
     */
    private function _drawRelations($changeColor)
    {
        foreach ($this->_relations as $relation) {
            $relation->relationDraw($changeColor);
        }
    }
    
    /**
     * Draws tables
     *
     * @param boolean showColor Whether to show color for primary fields or not
     * @access private
     * @see Table_Stats::Table_Stats_tableDraw()
     */
    private function _drawTables($changeColor)
    {
        global $svg;
        foreach ($this->tables as $table) {
            $table->tableDraw($changeColor);
        }
    }
}
?>