/*
 * Decompiled with CFR 0_110.
 * 
 * Could not load the following classes:
 *  javax.swing.tree.DefaultTreeModel
 *  javax.swing.tree.TreeNode
 */
package org.dom4j.swing;

import javax.swing.tree.DefaultTreeModel;
import javax.swing.tree.TreeNode;
import org.dom4j.Branch;
import org.dom4j.Document;
import org.dom4j.swing.BranchTreeNode;

public class DocumentTreeModel
extends DefaultTreeModel {
    protected Document document;

    public DocumentTreeModel(Document document) {
        super((TreeNode)new BranchTreeNode(document));
        this.document = document;
    }

    public Document getDocument() {
        return this.document;
    }

    public void setDocument(Document document) {
        this.document = document;
        this.setRoot((TreeNode)new BranchTreeNode(document));
    }
}

