83c30ce71bce689451b75553edfe34526ebbc126
[idea/community.git] / platform / vcs-api / src / com / intellij / openapi / vcs / changes / SortByVcsRoots.java
1 /*
2  * Copyright 2000-2010 JetBrains s.r.o.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 package com.intellij.openapi.vcs.changes;
17
18 import com.intellij.openapi.project.Project;
19 import com.intellij.openapi.vcs.FilePath;
20 import com.intellij.openapi.vcs.ProjectLevelVcsManager;
21 import com.intellij.openapi.vcs.VcsRoot;
22 import com.intellij.util.containers.Convertor;
23 import com.intellij.util.containers.MultiMap;
24
25 import java.util.Collection;
26
27 public class SortByVcsRoots<T> {
28   private final Project myProject;
29   private final Convertor<T, FilePath> myConvertor;
30   private ProjectLevelVcsManager myVcsManager;
31   public static final VcsRoot ourFictiveValue = new VcsRoot(null, null);
32
33   public SortByVcsRoots(Project project, final Convertor<T, FilePath> convertor) {
34     myProject = project;
35     myVcsManager = ProjectLevelVcsManager.getInstance(project);
36     myConvertor = convertor;
37   }
38
39   public MultiMap<VcsRoot, T> sort(final Collection<T> in) {
40     final MultiMap<VcsRoot, T> result = new MultiMap<>();
41     for (T t : in) {
42       final VcsRoot root = myVcsManager.getVcsRootObjectFor(myConvertor.convert(t));
43       if (root != null) {
44         result.putValue(root, t);
45       } else {
46         result.putValue(ourFictiveValue, t);
47       }
48     }
49     return result;
50   }
51 }