//$Id: exampletreemodel.cc,v 1.1 2004/01/16 13:49:39 murrayc Exp $ -*- c++ -*-

/* gtkmm example Copyright (C) 2002 gtkmm development team
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2
 * as published by the Free Software Foundation.
 *
 * This program 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
 */

#include <iostream>
#include "exampletreemodel.h"


ExampleTreeModel::ExampleTreeModel()
: Glib::ObjectBase( typeid(ExampleTreeModel) ), //register a custom GType.
  Glib::Object(), //The custom GType is actually registered here.
  m_stamp(1) //When the model's stamp != the iterator's stamp then that iterator is invalid and should be ignored. Also, 0=invalid.
{
  
  //We need to specify a particular get_type() from one of the virtual base classes, but they should
  //both return the same piece of data.
  Gtk::TreeModel::add_interface( Glib::Object::get_type() );
   
  //Initialize our underlying data:
  const typeListOfRows::size_type rows_count = 100;
  const typeRow::size_type columns_count = 10;
  m_rows.resize(rows_count); //100 rows.
  for(unsigned int row_number = 0; row_number < rows_count; ++row_number)
  {
    //Create the row:
    m_rows[row_number].resize(columns_count, "abc"); //Note that all 10 columns here are of the same type.
  }

  //The Column information that can be used with TreeView::append(), TreeModel::iterator[], etc.
  m_listModelColumns.resize(columns_count);
  for(unsigned int column_number = 0; column_number < columns_count; ++column_number)
  {
    m_column_record.add( m_listModelColumns[column_number] );
  }
}

ExampleTreeModel::~ExampleTreeModel()
{
}

//static:
Glib::RefPtr<ExampleTreeModel> ExampleTreeModel::create()
{
  return Glib::RefPtr<ExampleTreeModel>( new ExampleTreeModel );
}

Gtk::TreeModelFlags ExampleTreeModel::get_flags_vfunc()
{
   return Gtk::TreeModelFlags(0);
}

int ExampleTreeModel::get_n_columns_vfunc()
{
   return m_rows[0].size(); //The number of columns in the first (same as every one) row.
}

GType ExampleTreeModel::get_column_type_vfunc(int index)
{
  if(index <= (int)m_listModelColumns.size())
    return m_listModelColumns[index].type();
  else
    return 0;
}

void ExampleTreeModel::get_value_vfunc(const TreeModel::iterator& iter, int column, GValue* value)
{
  if(check_treeiter_validity(iter.gobj()))
  {  
    GType gtype = 0;
    if(column <= (int)m_listModelColumns.size())
      gtype =  m_listModelColumns[column].type();

    g_value_init(value, gtype);

    typeListOfRows::iterator dataRowIter = get_data_row_iter_from_tree_row_iter(iter.gobj());
    if(dataRowIter != m_rows.end())
    {
      typeRow& dataRow = *dataRowIter;

      Glib::ustring result = dataRow[column];

      g_value_set_string(value, result.c_str());
    }
  }
}

bool ExampleTreeModel::iter_next_vfunc(GtkTreeIter* iter)
{
  if(!check_treeiter_validity(iter))
    return false;
    
  //Make the GtkTreeIter represent the next row:
  typeListOfRows::size_type row_index = (typeListOfRows::size_type)iter->user_data;
  row_index++;
  if( row_index < m_rows.size() )
  {
    //Put the index of the next row in the iter, replacing the previous row index:
    iter->user_data = (void*)row_index;
    return true; //success
  }

  return false; //There is no next row.
}

bool ExampleTreeModel::iter_children_vfunc(GtkTreeIter* iter, const GtkTreeIter* /* parent */)
{
  if(!check_treeiter_validity(iter))
    return false;
    
  return false; //There are no children.
}

bool ExampleTreeModel::iter_has_child_vfunc(const GtkTreeIter* iter)
{
  if(!check_treeiter_validity(iter))
    return false;
    
  return false; //There are no children.
}

int ExampleTreeModel::iter_n_children_vfunc(const GtkTreeIter* iter)
{
  if(!check_treeiter_validity(iter))
    return 0;
    
  return 0; //There are no children
}

bool ExampleTreeModel::iter_nth_child_vfunc(GtkTreeIter* iter, const GtkTreeIter* /* parent */, int /* n */)
{
  if(!check_treeiter_validity(iter))
    return false;
    
  return false; //There are no children.
}

bool ExampleTreeModel::iter_parent_vfunc(GtkTreeIter* iter, const GtkTreeIter* /* child */)
{
  if(!check_treeiter_validity(iter))
    return false;
    
  return false; //There are no children, so no parents.
}

Gtk::TreeModel::Path ExampleTreeModel::get_path_vfunc(const Gtk::TreeModel::iterator& /* iter */)
{
   //TODO:
   return Gtk::TreeModel::Path();
}

bool ExampleTreeModel::get_iter_vfunc(GtkTreeIter* iter, const Gtk::TreeModel::Path& path)
{
   unsigned sz = path.size();
   if(!sz)
     return false;

   if(sz > 1)
     return false; //There are no children.

   //This is a new GtkTreeIter, so it needs the current stamp value.
   //See the comment in the constructor.
   iter->stamp = m_stamp;
   
   //Store the row_index in the GtkTreeIter:
   //See also iter_next_vfunc()
   //TODO: Store a pointer to some more complex data type such as a typeListOfRows::iterator.
   
   unsigned row_index = path[0];
   iter->user_data = (void*)row_index;
   
   return true;
}

Gtk::TreeModelColumn< Glib::ustring >& ExampleTreeModel::get_model_column(int column)
{
  return m_listModelColumns[column];
}

ExampleTreeModel::typeListOfRows::iterator ExampleTreeModel::get_data_row_iter_from_tree_row_iter(const GtkTreeIter* iter)
{
  typeListOfRows::size_type row_index = (typeListOfRows::size_type)iter->user_data;
  if( row_index > m_rows.size() )
    return m_rows.end();
  else
    return m_rows.begin() + row_index; //TODO: Performance.
}

bool ExampleTreeModel::check_treeiter_validity(const GtkTreeIter* iter) const
{
  // Anything that modifies the model's structure should change the model's stamp,
  // so that old iters are ignored.
  return m_stamp == iter->stamp;
}