core folder

Normand Briere

2018-07-07 09ddd38fd4a8a7100c834a5e976f4796fae53541

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
package mocap.gui;
 
import java.awt.BorderLayout;
import java.awt.Dimension;
 
import javax.swing.BorderFactory;
import javax.swing.JLabel;
import javax.swing.JPanel;
import javax.swing.JScrollPane;
import javax.swing.JTree;
import javax.swing.event.TreeSelectionEvent;
import javax.swing.event.TreeSelectionListener;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.DefaultTreeModel;
import javax.swing.tree.TreePath;
 
import mocap.figure.Bone;
 
/**
 * Shows the skeletal hierarchy in a tree.
 * 
 * @author Michael Kipp
 */
public class BoneHierarachyTree extends JPanel implements TreeSelectionListener {
 
    private JTree _tree;
    private JLabel _info;
    private BoneInfoPanel _boneInfo;
    private Bone _prevSelected;
 
    public BoneHierarachyTree(BoneInfoPanel bip) {
        setLayout(new BorderLayout());
        _boneInfo = bip;
        _info = new JLabel("--");
        _info.setBorder(BorderFactory.createEmptyBorder(5, 5, 10, 5));
        _tree = new JTree();
        _tree.addTreeSelectionListener(this);
        JScrollPane sp = new JScrollPane(_tree);
        add(BorderLayout.NORTH, _info);
        add(BorderLayout.CENTER, sp);
        sp.setPreferredSize(new Dimension(200, 800));
    }
 
    public void loadSkeleton(Bone skel) {
        _tree.removeAll();
        DefaultMutableTreeNode r = new DefaultMutableTreeNode("skeleton");
        r.add(createNodes(skel));
        _tree.setModel(new DefaultTreeModel(r));
 
        for (int i = 0; i < _tree.getRowCount(); i++) {
            _tree.expandRow(i);
        }
    }
 
    private DefaultMutableTreeNode createNodes(Bone b) {
        final Bone b1 = b;
        DefaultMutableTreeNode n = new DefaultMutableTreeNode(b) {
 
            @Override
            public String toString() {
                return b1.getName();
            }
        };
        for (int i = 0; i < b.getChildren().length; i++) {
            n.add(createNodes(b.getChildren()[i]));
        }
        return n;
    }
 
    public Bone getSelectedBone() {
        TreePath p = _tree.getSelectionPath();
        if (p != null) {
            DefaultMutableTreeNode n = (DefaultMutableTreeNode) p.getLastPathComponent();
            return n.getUserObject() instanceof Bone ? (Bone) n.getUserObject() : null;
        }
        return null;
    }
 
    public void valueChanged(TreeSelectionEvent e) {
        Bone b = getSelectedBone();
        if (b != null) {
            _info.setText(b.getName());
            _boneInfo.setBone(b);
            b.setSelected(true);
            if (_prevSelected != null) {
                _prevSelected.setSelected(false);
            }
            _prevSelected = b;
        }
    }
}