* Code cleanup;

[Persons_Comparator.git] / src / RegionView.java

import javax.swing.*;
import javax.swing.tree.DefaultMutableTreeNode;
import java.util.ListIterator;

public class RegionView extends JPanel {
    private Region regionObj;
    private JTree tree;

    RegionView(Region regionObj) {
        setRegionObj(regionObj);
        DefaultMutableTreeNode top = new DefaultMutableTreeNode("Region");

        for (String continent : regionObj.getContinents()) {
            DefaultMutableTreeNode topContinent = new DefaultMutableTreeNode(continent);
            if (continent.equals("Europe")) {
                ListIterator<Country> iter = regionObj.getCountryArrayListEurope().listIterator();
                while (iter.hasNext()) {
                    Country countryCursor = iter.next();
                    topContinent.add(new DefaultMutableTreeNode(countryCursor.getName()));
                    top.add(topContinent);
                }
            } else if (continent.equals("Africa")) {
                ListIterator<Country> iter = regionObj.getCountryArrayListAfrica().listIterator();
                while (iter.hasNext()) {
                    Country countryCursor = iter.next();
                    topContinent.add(new DefaultMutableTreeNode(countryCursor.getName()));
                    top.add(topContinent);
                }
            } else if (continent.equals("Americas")) {
                ListIterator<Country> iter = regionObj.getCountryArrayListAmericas().listIterator();
                while (iter.hasNext()) {
                    Country countryCursor = iter.next();
                    topContinent.add(new DefaultMutableTreeNode(countryCursor.getName()));
                    top.add(topContinent);
                }
            } else if (continent.equals("Asia")) {
                ListIterator<Country> iter = regionObj.getCountryArrayListAsia().listIterator();
                while (iter.hasNext()) {
                    Country countryCursor = iter.next();
                    topContinent.add(new DefaultMutableTreeNode(countryCursor.getName()));
                    top.add(topContinent);
                }
            } else if (continent.equals("Oceania")) {
                ListIterator<Country> iter = regionObj.getCountryArrayListOceania().listIterator();
                while (iter.hasNext()) {
                    Country countryCursor = iter.next();
                    topContinent.add(new DefaultMutableTreeNode(countryCursor.getName()));
                    top.add(topContinent);
                }
            }
        }
        tree = new JTree(top);
        add(tree);
    }

    public void setRegionObj(Region regionObj) {
        this.regionObj = regionObj;
    }
}