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
92
93
94
95
96
97
98
99
100
101
102
103
104
|
package com.vaadin.tests.components.treegrid;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.List;
import com.vaadin.annotations.Widgetset;
import com.vaadin.server.VaadinRequest;
import com.vaadin.tests.components.AbstractTestUI;
import com.vaadin.ui.Button;
import com.vaadin.ui.HorizontalLayout;
import com.vaadin.ui.RadioButtonGroup;
import com.vaadin.ui.TreeGrid;
@Widgetset("com.vaadin.DefaultWidgetSet")
public class TreeGridExpandCollapseRecursively extends AbstractTestUI {
private static class Directory {
private String name;
private Directory parent;
private List<Directory> subDirectories = new ArrayList<>();
public Directory(String name, Directory parent) {
this.name = name;
this.parent = parent;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
public Directory getParent() {
return parent;
}
public void setParent(Directory parent) {
this.parent = parent;
}
public List<Directory> getSubDirectories() {
return subDirectories;
}
public void setSubDirectories(List<Directory> subDirectories) {
this.subDirectories = subDirectories;
}
}
private static final int DEPTH = 4;
private static final int CHILDREN = 5;
@Override
protected void setup(VaadinRequest request) {
Collection<Directory> roots = generateDirectoryStructure(DEPTH);
TreeGrid<Directory> grid = new TreeGrid<>();
grid.addColumn(item -> "Item" + item.getName());
grid.setItems(roots, Directory::getSubDirectories);
RadioButtonGroup<Integer> depthSelector = new RadioButtonGroup<>(
"Depth", Arrays.asList(0, 1, 2, 3));
depthSelector.addStyleName("horizontal");
depthSelector.setValue(3);
HorizontalLayout buttons = new HorizontalLayout();
buttons.addComponent(new Button("Expand recursively",
e -> grid.expandRecursively(roots, depthSelector.getValue())));
buttons.addComponent(new Button("Collapse recursively", e -> grid
.collapseRecursively(roots, depthSelector.getValue())));
addComponents(depthSelector, buttons, grid);
}
private Collection<Directory> generateDirectoryStructure(int depth) {
return generateDirectories(depth, null, CHILDREN);
}
private Collection<Directory> generateDirectories(int depth,
Directory parent, int childCount) {
Collection<Directory> dirs = new ArrayList<>();
if (depth >= 0) {
for (int i = 0; i < childCount; i++) {
String name = parent != null ? parent.getName() + "-" + i
: "-" + i;
Directory dir = new Directory(name, parent);
if (parent != null) {
parent.getSubDirectories().add(dir);
}
dirs.add(dir);
generateDirectories(depth - 1, dir, childCount);
}
}
return dirs;
}
}
|
