gparted/src/TreeView_Detail.cc

/* Copyright (C) 2004 Bart
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  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 Library 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 "../include/TreeView_Detail.h"
 
namespace GParted
{ 

TreeView_Detail::TreeView_Detail( )
{
	treestore_detail = Gtk::TreeStore::create( treeview_detail_columns );
	this->set_model( treestore_detail );
	this->set_rules_hint(true);
	this->treeselection = this->get_selection();
		
	//append columns
	this->append_column( _("Partition"), treeview_detail_columns.partition );
	this->append_column( _("Type"), treeview_detail_columns.type_square );
	this->append_column( _("Size(MB)"), treeview_detail_columns.size );
	this->append_column( _("Used(MB)"), treeview_detail_columns.used );
	this->append_column( _("Unused(MB)"), treeview_detail_columns.unused );
	this->append_column( _("Flags"), treeview_detail_columns.flags );
	
		
	//status_icon
	this->get_column( 0 ) ->pack_start( treeview_detail_columns.status_icon,false );
	
	//colored text in Partition column (used for darkgrey unallocated)
	Gtk::CellRendererText *cell_renderer_text = dynamic_cast<Gtk::CellRendererText*>(  this->get_column( 0 ) ->get_first_cell_renderer()   );
	this->get_column( 0 ) ->add_attribute(cell_renderer_text->property_foreground(), treeview_detail_columns.text_color);
	
	//colored square in Type column
	cell_renderer_text = dynamic_cast<Gtk::CellRendererText*>( this->get_column( 1 ) ->get_first_cell_renderer() );
	this->get_column( 1 ) ->add_attribute(cell_renderer_text->property_foreground(), treeview_detail_columns. color);
	
	//colored text in Type column
	this->get_column( 1 ) ->pack_start( treeview_detail_columns .type, false );
	
	//this sucks :) get_cell_renderers doesn't return them in some order, so i have to check manually...
	std::vector <Gtk::CellRenderer *> renderers = this->get_column( 1 ) ->get_cell_renderers( ) ;
	
	if ( renderers .front() != this->get_column( 1 ) ->get_first_cell_renderer() )	
		cell_renderer_text = dynamic_cast<Gtk::CellRendererText*>( renderers .front() ) ;
	else 
		cell_renderer_text = dynamic_cast<Gtk::CellRendererText*>( renderers .back() ) ;
	
	this->get_column( 1 ) ->add_attribute(cell_renderer_text->property_foreground(), treeview_detail_columns. text_color);
	
	
	//set alignment of numeric columns to right
	for( short t=2;t<5;t++)
	{
		cell_renderer_text = dynamic_cast<Gtk::CellRendererText*>( this->get_column( t ) ->get_first_cell_renderer() );
		cell_renderer_text->property_xalign () = 1;
	}
	
}

void TreeView_Detail::Load_Partitions( const std::vector<Partition> & partitions ) 
{
	treestore_detail ->clear( ) ;
	
	for ( unsigned int i = 0 ; i < partitions .size( ) ; i++ ) 
	{
		row = *( treestore_detail ->append( ) );
		Create_Row( row, partitions[ i ] );
		
		if ( partitions[ i ] .type == GParted::EXTENDED )
		{
			for ( unsigned int t = 0 ; t < partitions[ i ] .logicals .size( ) ; t++ ) 
			{
				childrow = *( treestore_detail ->append( row.children( ) ) );
				Create_Row( childrow, partitions[ i ] .logicals[ t ] );
			}
			
		}
	}
	
	//show logical partitions ( if any )
	this ->expand_all( );
}

void TreeView_Detail::Set_Selected( const Partition & partition )
{ 
	//look for appropiate row
	for( iter = treestore_detail ->children( ) .begin( ) ; iter != treestore_detail ->children( ) .end( ) ; iter++ )
	{
		row = *iter;
		partition_temp = row[ treeview_detail_columns.partition_struct ] ;
		//primary's
		if (	partition .sector_start >= partition_temp .sector_start &&
			partition .sector_end <=partition_temp .sector_end &&
			partition.inside_extended == partition_temp.inside_extended )
		{
			this ->set_cursor( (Gtk::TreePath) row );
			return;
		}
		
		//logicals
		if ( row .children( ) .size( ) > 0 ) //this is the row with the extended partition, search it's childrows...
		{
			for( iter_child = row .children( ) .begin( ) ; iter_child != row.children( ) .end( ) ; iter_child++ )
			{
				childrow = *iter_child;
				partition_temp = childrow[ treeview_detail_columns.partition_struct ] ;
						
				if ( partition .sector_start >= partition_temp .sector_start && partition .sector_end <= partition_temp .sector_end )
				{
					this ->expand_all( );
					this ->set_cursor( (Gtk::TreePath) childrow );
					return;
				}
			}
		}

	}
	
	
}

void TreeView_Detail::Create_Row( const Gtk::TreeRow & treerow, const Partition & partition )
{
	//hereby i assume these 2 are mutual exclusive. is this wise?? Time (and bugreports) will tell :)
	if ( partition .busy )
		treerow[ treeview_detail_columns .status_icon ] = render_icon( Gtk::Stock::DIALOG_AUTHENTICATION, Gtk::ICON_SIZE_BUTTON );
	else if ( partition .filesystem == "unknown" || partition .error != "" )
		treerow[ treeview_detail_columns .status_icon ] = render_icon( Gtk::Stock::DIALOG_WARNING, Gtk::ICON_SIZE_BUTTON );

	treerow[ treeview_detail_columns .partition ] = partition .partition;
	treerow[ treeview_detail_columns .color ] = Get_Color( partition .filesystem ) ;

	treerow[ treeview_detail_columns .text_color ] = ( partition .type == GParted::UNALLOCATED ) ? "darkgrey" : "black" ;
	treerow[ treeview_detail_columns .type ] = partition .filesystem ;
	treerow[ treeview_detail_columns .type_square ] = "██" ;
	
	//size
	treerow[ treeview_detail_columns .size ] = num_to_str( partition .Get_Length_MB( ) ) ;

	//used
	if ( partition .sectors_used != -1 )
		treerow[ treeview_detail_columns .used ] = num_to_str( partition .Get_Used_MB( ) ) ;
	else
		treerow[ treeview_detail_columns .used ] = "---" ;

	//unused
	if ( partition .sectors_unused != -1 )
		treerow[ treeview_detail_columns .unused ] = num_to_str( partition .Get_Unused_MB( ) ) ;
	else
		treerow[ treeview_detail_columns .unused ] = "---" ;
	
	//flags	
	treerow[ treeview_detail_columns .flags ] = " " + partition .flags ;
	
	//hidden column (partition object)
	treerow[ treeview_detail_columns .partition_struct ] = partition;
}

bool TreeView_Detail::on_button_press_event( GdkEventButton* event )
{ 
	//Call base class, to allow normal handling,
   	bool return_value = TreeView::on_button_press_event( event );

	iter = treeselection ->get_selected( );
		
	if ( *iter != 0 )
	{
		row = *iter;
		signal_mouse_click .emit( event, row[ treeview_detail_columns .partition_struct ] );
	}
		
	return return_value;
}

} //GParted