LoopControler.php

Go to the documentation of this file.

<?php
/* vim: set expandtab tabstop=4 shiftwidth=4: */
//  
//  Copyright (c) 2003 Laurent Bedubourg
//  
//  This library is free software; you can redistribute it and/or
//  modify it under the terms of the GNU Lesser General Public
//  License as published by the Free Software Foundation; either
//  version 2.1 of the License, or (at your option) any later version.
//  
//  This library is distributed in the hope that it will be useful,
//  but WITHOUT ANY WARRANTY; without even the implied warranty of
//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
//  Lesser General Public License for more details.
//  
//  You should have received a copy of the GNU Lesser General Public
//  License along with this library; if not, write to the Free Software
//  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
//  
//  Authors: Laurent Bedubourg <laurent.bedubourg@free.fr>
//  

require_once PT_IP . "/Types/OArray.php";
require_once PT_IP . "/Types/OHash.php";
require_once PT_IP . "/Types/Iterator.php";

class PHPTAL_LoopControler
{
    var $_context;
    var $_data;
    var $_data_name;
    var $_iterator;
    var $_error;
    
    function PHPTAL_LoopControler(&$context, $data_name, $data)
    {
        $this->_context   =& $context;
        $this->_data      =& $data;
        $this->_data_name =  $data_name;
        
        // ensure that data is not an error
        if (PEAR::isError($data)) {
            $this->_error =& $data;
            return $data;
        }
        
        // accepted objects
        // 
        // - iteratable implementing getNewIterator() method
        // - iterator implementing next(), isValid() and index() methods
        // - db_result produced by PEAR::DB package
        // 
        if (is_object($data)) {
            if (method_exists($data, "getNewIterator")) {
                
                $this->_iterator =& $data->getNewIterator();
                
            } elseif (is_a("iterator", $data)
                      || (method_exists($data, 'next')
                          && method_exists($data, 'isValid')
                          && method_exists($data, 'index'))) {
                
                $this->_iterator =& $data;
                
            } elseif (get_class($data) == 'db_result') {
                
                $this->_iterator = new PHPTAL_DBResultIterator($data);
                
            } else {
                
                $err = new TypeError("PHPTAL loop controler received a non Iterable object ("
                                     . get_class($data) . ")");
                $this->_error =& $err;
                return PEAR::raiseError($err);
            }
        } elseif (is_array($data)) {
            // 
            // array are accepted thanks to OArrayIterator
            // 
            reset($data);
            if (count($data) > 0 && array_key_exists(0, $data)) {
                // echo "OArray::iterator", RxObject::ClassName($data), endl;
                $this->_data = new OArray($data);
                $this->_iterator =& $this->_data->getNewIterator();
            } else {
                // echo "OHash::iterator", RxObject::ClassName($data), endl;
                $this->_data = new OHash($data);
                $this->_iterator =& $this->_data->getNewIterator();
            }
        } else {
            $err = new TypeError("phptal loop controler received a non Iterable value ("
                                 . gettype($data) . ")");
            $this->_error =& $err;
            return PEAR::raiseError($err);
        }

        // 
        // install loop in repeat context array
        // 
        $repeat =& $this->_context->get("repeat");
        if (array_key_exists($this->_data_name, $repeat)) {
            unset($repeat[$this->_data_name]);
        }
        $repeat[$this->_data_name] =& $this;        

        // $this->_context->setRef($this->_data_name, $temp);
        $temp =& $this->_iterator->value();
        $this->_context->set($this->_data_name, $temp);
        return $temp;
    }

    function index()
    {
        return $this->_iterator->index();
    }

    function key()
    {
        if (method_exists($this->_iterator, "key")) {
            return $this->_iterator->key();
        } else {
            return $this->_iterator->index();
        }
    }

    function number()
    {
        return $this->index() + 1;
    }

    function even()
    {
        return !$this->odd();
    }

    function odd()
    {
        return ($this->index() % 2);
    }

    function start()
    {
        return ($this->index() == 0);
    }

    function end()
    {
        return ($this->length() == $this->number());
    }

    function isValid()
    {
        return $this->_iterator->isValid();
    }

    function length()
    {
        return $this->_data->size();
    }
   
    function &next()
    {
        $temp =& $this->_iterator->next();
        if (!$this->_iterator->isValid()) {
            return false;
        }
        
        // $this->_context->setRef($this->_data_name, $temp);
        $this->_context->set($this->_data_name, $temp);
        return $temp;
    }
};


class PHPTAL_DBResultIterator extends Iterator
{
    var $_src;
    var $_index = -1;
    var $_end   = false;
    var $_value;
    
    function PHPTAL_DBResultIterator(&$result)
    {
        $this->_src =& $result;
        $this->reset();
    }

    function reset()
    {
        if ($this->size() == 0) {
            $this->_end = true;
            return;
        }

        $this->_index = 0;        
        $this->_end = false;
        unset($this->_value);
        $this->_value = $this->_src->fetchRow();        
    }
    
    function size()
    {
        if (!isset($this->_size)) {
            $this->_size = $this->_src->numRows();
        }
        return $this->_size;
    }

    function isValid()
    {
        return !$this->_end;
    }

    function &next()
    {
        if ($this->_end || ++ $this->_index >= $this->size()) {
            $this->_end = true;
            return false;
        }

        unset($this->_value);
        $this->_value = $this->_src->fetchRow();
        return $this->_value;
    }
    
    function &value()
    {
        return $this->_value;
    }

    function index()
    {
        return $this->_index;
    }
}

?>

---