public static TreeItem getTreeItem(Tree tree, Object data) {
for (TreeItem item : tree.getItems()) {
if (item.getData() == data) {
return item;
}
}
return null;
}
public static void packColumns(@NotNull Tree tree, boolean fit, @Nullable float[] ratios) {
tree.setRedraw(false);
try {
// Check for disposed items
// TODO: it looks like SWT error. Sometimes tree items are disposed and NPE is thrown from
// column.pack
for (TreeItem item : tree.getItems()) {
if (item.isDisposed()) {
return;
}
}
int totalWidth = 0;
final TreeColumn[] columns = tree.getColumns();
for (TreeColumn column : columns) {
column.pack();
totalWidth += column.getWidth();
}
Rectangle clientArea = tree.getClientArea();
if (clientArea.isEmpty()) {
return;
}
if (fit) {
int areaWidth = clientArea.width;
if (tree.getVerticalBar() != null) {
areaWidth -= tree.getVerticalBar().getSize().x;
}
if (totalWidth > areaWidth) {
int extraSpace = totalWidth - areaWidth;
for (TreeColumn tc : columns) {
double ratio = (double) tc.getWidth() / totalWidth;
tc.setWidth((int) (tc.getWidth() - extraSpace * ratio));
}
} else if (totalWidth < areaWidth) {
float extraSpace = areaWidth - totalWidth;
if (columns.length > 0) {
if (ratios == null || ratios.length < columns.length) {
extraSpace /= columns.length;
extraSpace--;
for (TreeColumn tc : columns) {
tc.setWidth((int) (tc.getWidth() + extraSpace));
}
} else {
for (int i = 0; i < columns.length; i++) {
TreeColumn tc = columns[i];
tc.setWidth((int) (tc.getWidth() + extraSpace * ratios[i]));
}
}
}
}
}
} finally {
tree.setRedraw(true);
}
}