diff --git a/org.eclipse.jgit/src/org/eclipse/jgit/revplot/PlotWalk.java b/org.eclipse.jgit/src/org/eclipse/jgit/revplot/PlotWalk.java index ea2437ff7..de181b103 100644 --- a/org.eclipse.jgit/src/org/eclipse/jgit/revplot/PlotWalk.java +++ b/org.eclipse.jgit/src/org/eclipse/jgit/revplot/PlotWalk.java @@ -51,7 +51,9 @@ import static org.eclipse.jgit.lib.Constants.R_TAGS; import java.io.IOException; import java.util.Arrays; import java.util.Collection; +import java.util.Collections; import java.util.Comparator; +import java.util.HashSet; import java.util.Map; import java.util.Set; @@ -91,6 +93,27 @@ public class PlotWalk extends RevWalk { reverseRefMap = repo.getAllRefsByPeeledObjectId(); } + /** + * Add additional refs to the walk + * + * @param refs + * additional refs + * + * @throws IOException + */ + public void addAdditionalRefs(Iterable refs) throws IOException { + for (Ref ref : refs) { + Set set = reverseRefMap.get(ref.getObjectId()); + if (set == null) + set = Collections.singleton(ref); + else { + set = new HashSet(set); + set.add(ref); + } + reverseRefMap.put(ref.getObjectId(), set); + } + } + @Override public void sort(final RevSort s, final boolean use) { if (s == RevSort.TOPO && !use)