blob: fa0fe0739a60fa92e0eadf106987e513ab852750 (
plain)
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
105
|
package com.intellij.coverage.view;
import com.intellij.coverage.CoverageEngine;
import com.intellij.coverage.CoverageSuitesBundle;
import com.intellij.ide.commander.AbstractListBuilder;
import com.intellij.openapi.project.Project;
import com.intellij.util.ArrayUtil;
import com.intellij.util.ui.ColumnInfo;
import com.intellij.util.ui.SortableColumnModel;
import org.jetbrains.annotations.NotNull;
import javax.swing.*;
import javax.swing.table.AbstractTableModel;
import java.util.ArrayList;
import java.util.List;
/**
* User: anna
* Date: 1/3/12
*/
class CoverageTableModel extends AbstractTableModel implements AbstractListBuilder.Model, SortableColumnModel {
private final ColumnInfo[] COLUMN_INFOS;
final List myElements = new ArrayList();
public CoverageTableModel(@NotNull CoverageSuitesBundle suitesBundle, CoverageViewManager.StateBean stateBean, Project project) {
final CoverageEngine coverageEngine = suitesBundle.getCoverageEngine();
COLUMN_INFOS = coverageEngine.createCoverageViewExtension(project, suitesBundle, stateBean).createColumnInfos();
}
public void removeAllElements() {
myElements.clear();
fireTableDataChanged();
}
public void addElement(final Object obj) {
myElements.add(obj);
fireTableDataChanged();
}
public void replaceElements(final List newElements) {
removeAllElements();
myElements.addAll(newElements);
fireTableDataChanged();
}
public Object[] toArray() {
return ArrayUtil.toObjectArray(myElements);
}
public int indexOf(final Object o) {
return myElements.indexOf(o);
}
public int getSize() {
return myElements.size();
}
public Object getElementAt(final int index) {
return myElements.get(index);
}
public int getRowCount() {
return myElements.size();
}
public int getColumnCount() {
return COLUMN_INFOS.length;
}
@Override
public String getColumnName(int column) {
return COLUMN_INFOS[column].getName();
}
public Object getValueAt(int rowIndex, int columnIndex) {
final Object element = getElementAt(rowIndex);
if (columnIndex == 0) {
return element;
}
else if (element instanceof CoverageListNode) {
return COLUMN_INFOS[columnIndex].valueOf(element);
}
return element;
}
public ColumnInfo[] getColumnInfos() {
return COLUMN_INFOS;
}
public void setSortable(boolean aBoolean) {
}
public boolean isSortable() {
return true;
}
public Object getRowValue(int row) {
return getElementAt(row);
}
public RowSorter.SortKey getDefaultSortKey() {
return null;
}
}
|
