Initial code check in continued

src/controller/system/SystemNode.java

@@ -0,0 +1,165 @@
+/*******************************************************************************
+ *     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.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()
+    {
+    	String retVal = "images/system.png";
+
+    	@SuppressWarnings( "unchecked" )
+        Enumeration<DefaultMutableTreeNode> nodes = children();
+
+    	while( nodes.hasMoreElements() )
+    	{
+    		SiteNode child = (SiteNode)nodes.nextElement();
+    		
+    		if( child.getIconPath().contains( "green.png" ) )
+    		{
+    			return "images/system_green.png";
+    		}
+    		else if( child.getIconPath().contains( "red.png" ) )
+    		{
+    			retVal = "images/system_red.png";
+    		}
+    	}
+    	
+    	return retVal;
+    }
+	
+	@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;
+	}
+}