Question

I am trying to write a class that implements the TreeModel class. I was hoping someone could guide me in the right direction. Below is my class. The problem is when I bind it to a jTree Component, the second level keeps being added over and over. So I suspect my reference to the parent object is wrong:

public class PMEntry implements TreeModel{

private String title;
private List<PMEntry> pmEntryCollection;
private String pmId;
private String href;
private PMEntry root;
private ModuleType type;

public PMEntry (PMEntry root){

  this.root = root;
}

@Override
public Object getRoot() {        

   return ((PMEntry)this.root);
}

@Override
public Object getChild(Object o, int i) {


    if(getPmEntryCollection().isEmpty()){

        return null;

    }else {

      return (PMEntry) getPmEntryCollection().get(i);

    }
}

@Override
public int getChildCount(Object o) {

   if(getPmEntryCollection().isEmpty()){

        return 0;

    }else {

      return getPmEntryCollection().size();

    }
}

@Override
public boolean isLeaf(Object o) {
    PMEntry pmentry = (PMEntry)o;
    return (pmentry.getType() == ModuleType.DM) ? true : false;
 }

@Override
public void valueForPathChanged(TreePath tp, Object o) {
   //todo
}

@Override
public int getIndexOfChild(Object parent, Object child) {

    if (!(parent instanceof PMEntry)){

        System.out.println("Returning -1");
        return -1;
    }           

    PMEntry pParent = (PMEntry) parent;

    List<PMEntry> children = pParent.getPmEntryCollection();

    if (children == null) {
        System.out.println("children = null, Returning -1");
        return -1;

    }

    for (int i = 0; i < children.size(); i++) {

        System.out.println("Child:" + child);

        if (children.get(i) == child) {

            return i;
        }

    }

    return -1;        
}

@Override
public void addTreeModelListener(TreeModelListener tl) {
   //todo
}

@Override
public void removeTreeModelListener(TreeModelListener tl) {
    //todo
}

@Override
public String toString(){

    return this.getTitle();
}
public enum ModuleType {

    PM,
    DM

}

// getters and setters here....

And here is a snippet of how I am binding the data

PMEntry tm = new PMEntry(null);
tm.setTitle("Root");

PMEntry pmRoot = new PMEntry((PMEntry)(tm));
pmRoot.setTitle("Project");

PMEntry pm1 = new PMEntry(pmRoot);
pm1.setType(PMEntry.ModuleType.DM);
pm1.setTitle("Publication Module");

PMEntry pm2 = new PMEntry(pmRoot);
pm2.setType(PMEntry.ModuleType.PM);
pm2.setTitle("Chapter");     

List<PMEntry> pmCollection = new ArrayList<PMEntry>();      
List<PMEntry> pmCollection1 = new ArrayList<PMEntry>();

PMEntry pm3 = new PMEntry(null);
pm3.setType(PMEntry.ModuleType.DM);
pm3.setTitle("Data Module");

PMEntry pm4 = new PMEntry(null);
pm4.setType(PMEntry.ModuleType.DM);
pm4.setTitle("Data Module");

pmCollection1.add(pm3);
pmCollection1.add(pm4); 

pm2.setPmEntryCollection(pmCollection1);

pmCollection.add(pm1);
pmCollection.add(pm2);                     

pmRoot.setPmEntryCollection(pmCollection);

this.jTree1.setModel(pmRoot);

Answer 1

I’d wonder why you think you need to implement TreeModel. Have you looked into DefaultTreeModel? What new behavior over and above that class do you plan to implement?