Switch person class usage of Origin class to Country class and use it for distance...

[Persons_Comparator.git] / src / RegionView.java

diff --git a/src/RegionView.java b/src/RegionView.java
index 3c79ea6f4a294fb5c446c977b2c2a40807303d3d..42941ac2d6f27247f749a309994f98fdd3854172 100644 (file)
--- a/src/RegionView.java
+++ b/src/RegionView.java
@@ -1,5 +1,6 @@
 import javax.swing.*;
 import javax.swing.tree.DefaultMutableTreeNode;
+import javax.swing.tree.TreeSelectionModel;
 import java.util.ListIterator;
 
 public class RegionView extends JPanel {
@@ -51,38 +52,42 @@ public class RegionView extends JPanel {
         }
         sortTree(top);
         tree = new JTree(top);
+        tree.getSelectionModel().setSelectionMode(TreeSelectionModel.SINGLE_TREE_SELECTION);
         add(new JScrollPane(tree));
     }
 
     public static DefaultMutableTreeNode sortTree(DefaultMutableTreeNode root) {
         {
             for (int i = 0; i < root.getChildCount() - 1; i++) {
-                DefaultMutableTreeNode node = (DefaultMutableTreeNode) root
-                        .getChildAt(i);
+                DefaultMutableTreeNode node = (DefaultMutableTreeNode) root.getChildAt(i);
                 String nt = node.getUserObject().toString();
 
                 for (int j = i + 1; j <= root.getChildCount() - 1; j++) {
                     DefaultMutableTreeNode prevNode = (DefaultMutableTreeNode) root
                             .getChildAt(j);
                     String np = prevNode.getUserObject().toString();
-
-                    System.out.println(nt + " " + np);
                     if (nt.compareToIgnoreCase(np) > 0) {
-
                         root.insert(node, j);
                         break;
                     }
                 }
                 if (node.getChildCount() > 0) {
-                    node = sortTree(node);
+                    sortTree(node);
                 }
             }
-
             return root;
         }
     }
 
+    public JTree getTree() {
+        return tree;
+    }
+
     public void setRegionObj(Region regionObj) {
         this.regionObj = regionObj;
     }
+
+    public Region getRegionObj() {
+        return regionObj;
+    }
 }