2007-01-15 23:12:22 +03:00
|
|
|
<?php
|
|
|
|
/*
|
|
|
|
* $Id$
|
|
|
|
*
|
|
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
|
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
|
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
|
|
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
|
|
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
|
|
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
|
|
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
|
|
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
|
|
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
|
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
|
|
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
*
|
|
|
|
* This software consists of voluntary contributions made by many individuals
|
|
|
|
* and is licensed under the LGPL. For more information, see
|
2008-01-23 01:52:53 +03:00
|
|
|
* <http://www.phpdoctrine.org>.
|
2007-01-15 23:12:22 +03:00
|
|
|
*/
|
2007-10-21 10:23:59 +04:00
|
|
|
|
2007-01-15 23:12:22 +03:00
|
|
|
/**
|
|
|
|
* Doctrine_Node_NestedSet_PreOrderIterator
|
|
|
|
*
|
|
|
|
* @package Doctrine
|
2007-10-04 01:43:22 +04:00
|
|
|
* @subpackage Node
|
2007-01-15 23:12:22 +03:00
|
|
|
* @license http://www.opensource.org/licenses/lgpl-license.php LGPL
|
2008-02-22 21:11:35 +03:00
|
|
|
* @link www.phpdoctrine.org
|
2007-01-15 23:12:22 +03:00
|
|
|
* @since 1.0
|
|
|
|
* @version $Revision$
|
|
|
|
* @author Joe Simms <joe.simms@websites4.com>
|
|
|
|
*/
|
|
|
|
class Doctrine_Node_NestedSet_PreOrderIterator implements Iterator
|
|
|
|
{
|
|
|
|
/**
|
|
|
|
* @var Doctrine_Collection $collection
|
|
|
|
*/
|
|
|
|
protected $collection;
|
2007-10-21 10:23:59 +04:00
|
|
|
|
2007-01-15 23:12:22 +03:00
|
|
|
/**
|
|
|
|
* @var array $keys
|
|
|
|
*/
|
|
|
|
protected $keys;
|
2007-10-21 10:23:59 +04:00
|
|
|
|
2007-01-15 23:12:22 +03:00
|
|
|
/**
|
|
|
|
* @var mixed $key
|
|
|
|
*/
|
|
|
|
protected $key;
|
2007-10-21 10:23:59 +04:00
|
|
|
|
2007-01-15 23:12:22 +03:00
|
|
|
/**
|
|
|
|
* @var integer $index
|
|
|
|
*/
|
|
|
|
protected $index;
|
2007-10-21 10:23:59 +04:00
|
|
|
|
2007-01-15 23:12:22 +03:00
|
|
|
/**
|
|
|
|
* @var integer $index
|
|
|
|
*/
|
|
|
|
protected $prevIndex;
|
2007-10-21 10:23:59 +04:00
|
|
|
|
2007-01-15 23:12:22 +03:00
|
|
|
/**
|
|
|
|
* @var integer $index
|
|
|
|
*/
|
|
|
|
protected $traverseLevel;
|
2007-10-21 10:23:59 +04:00
|
|
|
|
2007-01-15 23:12:22 +03:00
|
|
|
/**
|
|
|
|
* @var integer $count
|
|
|
|
*/
|
|
|
|
protected $count;
|
|
|
|
|
|
|
|
public function __construct($record, $opts)
|
|
|
|
{
|
|
|
|
$componentName = $record->getTable()->getComponentName();
|
|
|
|
|
|
|
|
$q = $record->getTable()->createQuery();
|
|
|
|
|
|
|
|
$params = array($record->get('lft'), $record->get('rgt'));
|
|
|
|
if (isset($opts['include_record']) && $opts['include_record']) {
|
2007-02-08 15:53:32 +03:00
|
|
|
$query = $q->where("$componentName.lft >= ? AND $componentName.rgt <= ?", $params)->orderBy("$componentName.lft asc");
|
2007-01-15 23:12:22 +03:00
|
|
|
} else {
|
2007-02-08 15:53:32 +03:00
|
|
|
$query = $q->where("$componentName.lft > ? AND $componentName.rgt < ?", $params)->orderBy("$componentName.lft asc");
|
2007-01-15 23:12:22 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
$query = $record->getTable()->getTree()->returnQueryWithRootId($query, $record->getNode()->getRootValue());
|
|
|
|
|
|
|
|
$this->maxLevel = isset($opts['depth']) ? ($opts['depth'] + $record->getNode()->getLevel()) : 0;
|
|
|
|
$this->options = $opts;
|
|
|
|
$this->collection = isset($opts['collection']) ? $opts['collection'] : $query->execute();
|
|
|
|
$this->keys = $this->collection->getKeys();
|
|
|
|
$this->count = $this->collection->count();
|
|
|
|
$this->index = -1;
|
|
|
|
$this->level = $record->getNode()->getLevel();
|
|
|
|
$this->prevLeft = $record->getNode()->getLeftValue();
|
|
|
|
|
|
|
|
// clear the table identity cache
|
|
|
|
$record->getTable()->clear();
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* rewinds the iterator
|
|
|
|
*
|
|
|
|
* @return void
|
|
|
|
*/
|
|
|
|
public function rewind()
|
|
|
|
{
|
|
|
|
$this->index = -1;
|
|
|
|
$this->key = null;
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* returns the current key
|
|
|
|
*
|
|
|
|
* @return integer
|
|
|
|
*/
|
|
|
|
public function key()
|
|
|
|
{
|
|
|
|
return $this->key;
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* returns the current record
|
|
|
|
*
|
|
|
|
* @return Doctrine_Record
|
|
|
|
*/
|
|
|
|
public function current()
|
|
|
|
{
|
|
|
|
$record = $this->collection->get($this->key);
|
|
|
|
$record->getNode()->setLevel($this->level);
|
|
|
|
return $record;
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* advances the internal pointer
|
|
|
|
*
|
|
|
|
* @return void
|
|
|
|
*/
|
|
|
|
public function next()
|
|
|
|
{
|
|
|
|
while ($current = $this->advanceIndex()) {
|
|
|
|
if ($this->maxLevel && ($this->level > $this->maxLevel)) {
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
return $current;
|
|
|
|
}
|
|
|
|
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @return boolean whether or not the iteration will continue
|
|
|
|
*/
|
|
|
|
public function valid()
|
|
|
|
{
|
|
|
|
return ($this->index < $this->count);
|
|
|
|
}
|
|
|
|
|
|
|
|
public function count()
|
|
|
|
{
|
|
|
|
return $this->count;
|
|
|
|
}
|
|
|
|
|
|
|
|
private function updateLevel()
|
|
|
|
{
|
2007-09-03 18:57:18 +04:00
|
|
|
if ( ! (isset($this->options['include_record']) && $this->options['include_record'] && $this->index == 0)) {
|
2007-01-15 23:12:22 +03:00
|
|
|
$left = $this->collection->get($this->key)->getNode()->getLeftValue();
|
|
|
|
$this->level += $this->prevLeft - $left + 2;
|
|
|
|
$this->prevLeft = $left;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
private function advanceIndex()
|
|
|
|
{
|
|
|
|
$this->index++;
|
|
|
|
$i = $this->index;
|
|
|
|
if (isset($this->keys[$i])) {
|
|
|
|
$this->key = $this->keys[$i];
|
|
|
|
$this->updateLevel();
|
|
|
|
return $this->current();
|
|
|
|
}
|
|
|
|
|
|
|
|
return false;
|
|
|
|
}
|
2007-10-04 01:43:22 +04:00
|
|
|
}
|