如何在JDialog中显示JTree？

import javax.swing.*;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.DefaultTreeModel;

public class TreeDialog extends JDialog {
    public TreeDialog(JFrame parent) {
        super(parent, "JTree Dialog", true);

        // 创建根节点
        DefaultMutableTreeNode root = new DefaultMutableTreeNode("Root");

        // 创建子节点
        DefaultMutableTreeNode node1 = new DefaultMutableTreeNode("Node 1");
        DefaultMutableTreeNode node2 = new DefaultMutableTreeNode("Node 2");

        // 将子节点添加到根节点下
        root.add(node1);
        root.add(node2);

        // 创建树模型
        DefaultTreeModel treeModel = new DefaultTreeModel(root);

        // 创建树组件
        JTree tree = new JTree(treeModel);

        // 将树组件添加到滚动面板中
        JScrollPane scrollPane = new JScrollPane(tree);

        // 将滚动面板添加到对话框中
        getContentPane().add(scrollPane);

        // 设置对话框大小和位置
        setSize(300, 400);
        setLocationRelativeTo(parent);
    }

    public static void main(String[] args) {
        JFrame frame = new JFrame("Main Frame");
        frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
        frame.setSize(800, 600);
        frame.setLocationRelativeTo(null);

        JButton button = new JButton("Open Dialog");
        button.addActionListener(e -> {
            TreeDialog dialog = new TreeDialog(frame);
            dialog.setVisible(true);
        });

        frame.getContentPane().add(button);
        frame.setVisible(true);
    }
}