sdrtrunk/src/controller/system/SystemNode.java

/*******************************************************************************
 *     SDR Trunk 
 *     Copyright (C) 2014 Dennis Sheirer
 * 
 *     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 3 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 General Public License for more details.
 * 
 *     You should have received a copy of the GNU General Public License
 *     along with this program.  If not, see <http://www.gnu.org/licenses/>
 ******************************************************************************/
package controller.system;


import java.awt.Color;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.util.Enumeration;

import javax.swing.JMenuItem;
import javax.swing.JPanel;
import javax.swing.JPopupMenu;
import javax.swing.JSeparator;
import javax.swing.tree.DefaultMutableTreeNode;

import controller.ConfigurableNode;
import controller.site.Site;
import controller.site.SiteNode;

public class SystemNode extends ConfigurableNode
{
    private static final long serialVersionUID = 1L;

	public SystemNode( System system )
	{
		super( system );
	}
	
	public void init()
	{
		for( Site site: getSystem().getSite() )
		{
			SiteNode node = new SiteNode( site );
			
			getModel().insertNodeInto( node, SystemNode.this, getChildCount() );
			
			node.init();
		}
		
		sort();
	}
	
	@Override
    public String getIconPath()
    {
		return "images/system.png";
    }
	
	@Override
	public Color getBackgroundColor()
	{
    	@SuppressWarnings( "unchecked" )
        Enumeration<DefaultMutableTreeNode> nodes = children();

    	while( nodes.hasMoreElements() )
    	{
    		SiteNode child = (SiteNode)nodes.nextElement();

    		if( child.getBackgroundColor() != null )
    		{
    			return Color.CYAN;
    		}
    	}

    	return null;
	}

	@Override
	public JPanel getEditor()
	{
	    return new SystemEditor( this );
	}
	
	public System getSystem()
	{
		return (System)getUserObject();
	}
	
	public String toString()
	{
		return getSystem().getName();
	}
	
	public void delete()
	{
		for( int x = 0; x < getChildCount(); x++ )
		{
			((SiteNode)getChildAt( x )).delete();
		}
		
		SystemListNode parent = (SystemListNode)getParent();
		
		parent.getSystemList().removeSystem( getSystem() );

		save();

		getModel().removeNodeFromParent( SystemNode.this );

		/* Show the first child, if one exists so that we don't have a 
		 * collapsed list */
		if( parent.getChildCount() > 0 )
		{
			((ConfigurableNode)parent.getFirstChild()).show();
		}
	}
	
	public JPopupMenu getContextMenu()
	{
		JPopupMenu retVal = new JPopupMenu();
		
		JMenuItem addSiteItem = new JMenuItem( "Add Site" );
		addSiteItem.addActionListener( new ActionListener() 
		{
			@Override
            public void actionPerformed( ActionEvent e )
            {
			    Site site = new Site();
			    getSystem().addSite( site );
			    
			    SiteNode node = new SiteNode( site );
			    
				getModel().addNode( node, 
									SystemNode.this, 
									SystemNode.this.getChildCount() );
				
				sort();
				
				node.show();
            }
		} );
		
		retVal.add( addSiteItem );
		
		retVal.add( new JSeparator() );

		JMenuItem deleteItem = new JMenuItem( "Delete" );
		deleteItem.addActionListener( new ActionListener() 
		{
			@Override
            public void actionPerformed( ActionEvent e )
            {
				delete();
            }
		} );
		
		retVal.add( deleteItem );
		
		return retVal;
	}
}