Main.java Source code

Introduction

Here is the source code for Main.java

Source

import java.awt.Component;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import javax.swing.JFrame;
import javax.swing.JScrollPane;
import javax.swing.JTree;
import javax.swing.UIManager;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.DefaultTreeCellRenderer;

public class Main extends JFrame {
    private JTree tree;

    public Main() {
        DefaultMutableTreeNode contacts = new DefaultMutableTreeNode("Contacts");
        createNodes(contacts);
        tree = new JTree(contacts);
        tree.setCellRenderer(new MyTreeCellRenderer());
        JScrollPane treeView = new JScrollPane(tree);
        add(treeView);
        setSize(400, 400);
        setDefaultCloseOperation(EXIT_ON_CLOSE);
        setVisible(true);
    }

    void createNodes(DefaultMutableTreeNode top) {
        DefaultMutableTreeNode contactName;
        List<Contact> contacts = new ArrayList<Contact>();
        contacts.add(new Contact("Me", true));
        contacts.add(new Contact("You"));

        Iterator<Contact> contactItr = contacts.iterator();
        while (contactItr.hasNext()) {
            contactName = new DefaultMutableTreeNode(contactItr.next());
            top.add(contactName);
        }
    }

    public static void main(String[] args) {
        new Main();
    }
}

class MyTreeCellRenderer extends DefaultTreeCellRenderer {

    @Override
    public Component getTreeCellRendererComponent(JTree tree, Object value, boolean sel, boolean expanded,
            boolean leaf, int row, boolean hasFocus) {
        super.getTreeCellRendererComponent(tree, value, sel, expanded, leaf, row, hasFocus);
        if (value instanceof DefaultMutableTreeNode) {
            DefaultMutableTreeNode node = (DefaultMutableTreeNode) value;
            if (node.getUserObject() instanceof String) {
                setIcon(UIManager.getIcon("FileView.computerIcon"));
            } else if (node.getUserObject() instanceof Contact) {
                Contact contact = (Contact) node.getUserObject();
                if (contact.isSomeProperty()) {
                    setIcon(UIManager.getIcon("FileView.hardDriveIcon"));
                } else {
                    setIcon(UIManager.getIcon("FileChooser.homeFolderIcon"));
                }
            }
        }
        return this;
    }
}

class Contact {
    private boolean someProperty;
    private String name;

    public Contact(String name) {
        this(name, false);
    }

    public Contact(String name, boolean property) {
        this.someProperty = property;
        this.name = name;
    }

    public boolean isSomeProperty() {
        return someProperty;
    }

    public String getName() {
        return name;
    }

    @Override
    public String toString() {
        return name;
    }
}