/*
* Copyright 2000-2015 JetBrains s.r.o.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.intellij.vcs.log.data;
import com.intellij.openapi.vfs.VirtualFile;
import com.intellij.util.containers.ContainerUtil;
import com.intellij.vcs.log.VcsLogFilterCollection;
import com.intellij.vcs.log.VcsLogProvider;
import com.intellij.vcs.log.VcsRef;
import com.intellij.vcs.log.graph.GraphColorManagerImpl;
import com.intellij.vcs.log.graph.VisibleGraph;
import com.intellij.vcs.log.graph.api.permanent.PermanentGraphInfo;
import com.intellij.vcs.log.graph.collapsing.CollapsedController;
import com.intellij.vcs.log.graph.impl.facade.BaseController;
import com.intellij.vcs.log.graph.impl.facade.VisibleGraphImpl;
import com.intellij.vcs.log.impl.VcsLogUtil;
import org.jetbrains.annotations.NotNull;
import java.util.Map;
import java.util.Set;
public class FakeVisiblePackBuilder {
@NotNull private final VcsLogStorage myHashMap;
public FakeVisiblePackBuilder(@NotNull VcsLogStorage hashMap) {
myHashMap = hashMap;
}
@NotNull
public VisiblePack build(@NotNull VisiblePack visiblePack) {
if (visiblePack.getVisibleGraph() instanceof VisibleGraphImpl && visiblePack.getVisibleGraph().getVisibleCommitCount() > 0) {
return build(visiblePack.getDataPack(), ((VisibleGraphImpl<Integer>)visiblePack.getVisibleGraph()), visiblePack.getFilters());
}
else {
VisibleGraph<Integer> newGraph = EmptyVisibleGraph.getInstance();
DataPackBase newPack = new DataPackBase(visiblePack.getDataPack().getLogProviders(), createEmptyRefsModel(), false);
return new VisiblePack(newPack, newGraph, true, visiblePack.getFilters());
}
}
@NotNull
private VisiblePack build(@NotNull DataPackBase oldPack,
@NotNull VisibleGraphImpl<Integer> oldGraph,
@NotNull VcsLogFilterCollection filters) {
final PermanentGraphInfo<Integer> info = oldGraph.buildSimpleGraphInfo();
Set<Integer> heads = ContainerUtil.map2Set(info.getPermanentGraphLayout().getHeadNodeIndex(),
integer -> info.getPermanentCommitsInfo().getCommitId(integer));
RefsModel newRefsModel = createRefsModel(oldPack.getRefsModel(), heads, oldGraph, oldPack.getLogProviders());
DataPackBase newPack = new DataPackBase(oldPack.getLogProviders(), newRefsModel, false);
GraphColorManagerImpl colorManager =
new GraphColorManagerImpl(newRefsModel, DataPack.createHashGetter(myHashMap), DataPack.getRefManagerMap(oldPack.getLogProviders()));
VisibleGraph<Integer> newGraph =
new VisibleGraphImpl<>(new CollapsedController(new BaseController(info), info, null), info, colorManager);
return new VisiblePack(newPack, newGraph, true, filters);
}
@NotNull
private RefsModel createEmptyRefsModel() {
return new RefsModel(ContainerUtil.newHashMap(), ContainerUtil.newHashSet(), myHashMap, ContainerUtil.newHashMap());
}
private RefsModel createRefsModel(@NotNull RefsModel refsModel,
@NotNull Set<Integer> heads,
@NotNull VisibleGraph<Integer> visibleGraph,
@NotNull Map<VirtualFile, VcsLogProvider> providers) {
Set<VcsRef> branchesAndHeads = ContainerUtil.newHashSet();
refsModel.getBranches().stream().filter(ref -> {
int index = myHashMap.getCommitIndex(ref.getCommitHash(), ref.getRoot());
Integer row = visibleGraph.getVisibleRowIndex(index);
return row != null && row >= 0;
}).forEach(branchesAndHeads::add);
heads.stream().flatMap(head -> refsModel.refsToCommit(head).stream()).forEach(branchesAndHeads::add);
Map<VirtualFile, Set<VcsRef>> map = VcsLogUtil.groupRefsByRoot(branchesAndHeads);
Map<VirtualFile, CompressedRefs> refs = ContainerUtil.newHashMap();
for (VirtualFile root : providers.keySet()) {
Set<VcsRef> refsForRoot = map.get(root);
refs.put(root, new CompressedRefs(refsForRoot == null ? ContainerUtil.newHashSet() : refsForRoot, myHashMap));
}
return new RefsModel(refs, heads, myHashMap, providers);
}
}