Rearrange code.

[Persons_Comparator.git] / src / RegionView.java

diff --git a/src/RegionView.java b/src/RegionView.java
index d44cd0494f9f10dd063eb4b59ec09ad4d230691d..c5921d6a881b6ac4a59297a2463c4bcf7fbced1a 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,10 +52,11 @@ 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) {
+    private void sortTree(DefaultMutableTreeNode root) {
         {
             for (int i = 0; i < root.getChildCount() - 1; i++) {
                 DefaultMutableTreeNode node = (DefaultMutableTreeNode) root.getChildAt(i);
@@ -70,13 +72,20 @@ public class RegionView extends JPanel {
                     }
                 }
                 if (node.getChildCount() > 0) {
-                    node = sortTree(node);
+                    this.sortTree(node);
                 }
             }
-            return root;
         }
     }
 
+    public JTree getTree() {
+        return tree;
+    }
+
+    public Region getRegionObj() {
+        return regionObj;
+    }
+
     public void setRegionObj(Region regionObj) {
         this.regionObj = regionObj;
     }