/** * @return \ArrayIterator */ protected function getDataRows() { if (TreeView::EXPANDED === $this->mode) { if ($this->dataRows === NULL) { $this->dataRows = $this->dataSource->fetchAssoc($this->primaryKey); } return $this->dataRows; } return parent::getDataRows(); }
/** * Generates columns from data source */ protected function generateColumns() { foreach ($this->dataSource->getColumns() as $name) { $this->addColumn($name); } }