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
|
package com.intellij.vcs.log.data;
import com.intellij.openapi.util.Condition;
import com.intellij.openapi.vfs.VirtualFile;
import com.intellij.util.NotNullFunction;
import com.intellij.util.SmartList;
import com.intellij.util.containers.ContainerUtil;
import com.intellij.util.containers.MultiMap;
import com.intellij.vcs.log.Hash;
import com.intellij.vcs.log.VcsLogRefs;
import com.intellij.vcs.log.VcsRef;
import gnu.trove.TIntObjectHashMap;
import org.jetbrains.annotations.NotNull;
import java.util.*;
public class RefsModel implements VcsLogRefs {
@NotNull private final Map<VirtualFile, Collection<VcsRef>> myRefs;
@NotNull private final NotNullFunction<Hash, Integer> myIndexGetter;
@NotNull private final Collection<VcsRef> myBranches;
@NotNull private final MultiMap<Hash, VcsRef> myRefsToHashes;
@NotNull private final TIntObjectHashMap<SmartList<VcsRef>> myRefsToIndices;
public RefsModel(@NotNull Map<VirtualFile, Collection<VcsRef>> refsByRoot, @NotNull NotNullFunction<Hash, Integer> indexGetter) {
myRefs = refsByRoot;
myIndexGetter = indexGetter;
List<VcsRef> allRefs = ContainerUtil.concat(refsByRoot.values());
myBranches = ContainerUtil.filter(allRefs, new Condition<VcsRef>() {
@Override
public boolean value(VcsRef ref) {
return ref.getType().isBranch();
}
});
myRefsToHashes = prepareRefsMap(allRefs);
myRefsToIndices = prepareRefsToIndicesMap(allRefs);
}
@NotNull
private TIntObjectHashMap<SmartList<VcsRef>> prepareRefsToIndicesMap(@NotNull Collection<VcsRef> refs) {
TIntObjectHashMap<SmartList<VcsRef>> map = new TIntObjectHashMap<SmartList<VcsRef>>();
for (VcsRef ref : refs) {
int index = myIndexGetter.fun(ref.getCommitHash());
SmartList<VcsRef> list = map.get(index);
if (list == null) map.put(index, list = new SmartList<VcsRef>());
list.add(ref);
}
return map;
}
@NotNull
private static MultiMap<Hash, VcsRef> prepareRefsMap(@NotNull Collection<VcsRef> refs) {
MultiMap<Hash, VcsRef> map = MultiMap.createSmartList();
for (VcsRef ref : refs) {
map.putValue(ref.getCommitHash(), ref);
}
return map;
}
@NotNull
public Collection<VcsRef> refsToCommit(@NotNull Hash hash) {
if (myRefsToHashes.containsKey(hash)) {
return myRefsToHashes.get(hash);
}
return Collections.emptyList();
}
@NotNull
public Collection<VcsRef> refsToCommit(int index) {
return myRefsToIndices.containsKey(index) ? myRefsToIndices.get(index) : Collections.<VcsRef>emptyList();
}
@Override
@NotNull
public Collection<VcsRef> getBranches() {
return myBranches;
}
@NotNull
public Collection<VcsRef> getAllRefs() {
return new ArrayList<VcsRef>(myRefsToHashes.values());
}
@NotNull
public Map<VirtualFile, Collection<VcsRef>> getAllRefsByRoot() {
return myRefs;
}
}
|