c4445c4ab5d225041f9b1c4a33fd0221a04e309c
[idea/community.git] / platform / lang-impl / src / com / intellij / find / actions / ShowUsagesTableCellRenderer.java
1 /*
2  * Copyright 2000-2014 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
17 package com.intellij.find.actions;
18
19 import com.intellij.openapi.project.Project;
20 import com.intellij.openapi.vfs.VirtualFile;
21 import com.intellij.psi.PsiFile;
22 import com.intellij.psi.PsiManager;
23 import com.intellij.psi.search.SearchScope;
24 import com.intellij.ui.FileColorManager;
25 import com.intellij.ui.SimpleColoredComponent;
26 import com.intellij.ui.SimpleTextAttributes;
27 import com.intellij.usages.TextChunk;
28 import com.intellij.usages.Usage;
29 import com.intellij.usages.UsageGroup;
30 import com.intellij.usages.UsagePresentation;
31 import com.intellij.usages.impl.GroupNode;
32 import com.intellij.usages.impl.UsageNode;
33 import com.intellij.usages.impl.UsageViewImpl;
34 import com.intellij.usages.impl.UsageViewManagerImpl;
35 import com.intellij.usages.rules.UsageInFile;
36 import com.intellij.util.ui.EmptyIcon;
37 import com.intellij.util.ui.UIUtil;
38 import org.jetbrains.annotations.NotNull;
39
40 import javax.swing.*;
41 import javax.swing.table.TableCellRenderer;
42 import javax.swing.table.TableColumnModel;
43 import java.awt.*;
44 import java.util.concurrent.atomic.AtomicInteger;
45
46 /**
47 * @author cdr
48 */
49 class ShowUsagesTableCellRenderer implements TableCellRenderer {
50   private final UsageViewImpl myUsageView;
51   @NotNull private final AtomicInteger myOutOfScopeUsages;
52   @NotNull private final SearchScope mySearchScope;
53
54   ShowUsagesTableCellRenderer(@NotNull UsageViewImpl usageView, @NotNull AtomicInteger outOfScopeUsages, @NotNull SearchScope searchScope) {
55     myUsageView = usageView;
56     myOutOfScopeUsages = outOfScopeUsages;
57     mySearchScope = searchScope;
58   }
59
60   @Override
61   public Component getTableCellRendererComponent(JTable list, Object value, boolean isSelected, boolean hasFocus, int row, int column) {
62     UsageNode usageNode = value instanceof UsageNode ? (UsageNode)value : null;
63
64     Usage usage = usageNode == null ? null : usageNode.getUsage();
65
66     JPanel panel = new JPanel(new FlowLayout(FlowLayout.LEFT, 0,0));
67     Color fileBgColor = getBackgroundColor(isSelected, usage);
68     final Color bg = UIUtil.getListSelectionBackground();
69     final Color fg = UIUtil.getListSelectionForeground();
70     Color panelBackground = isSelected ? bg : fileBgColor == null ? list.getBackground() : fileBgColor;
71     Color panelForeground = isSelected ? fg : list.getForeground();
72     panel.setBackground(panelBackground);
73     panel.setForeground(panelForeground);
74
75     SimpleColoredComponent textChunks = new SimpleColoredComponent();
76     textChunks.setIpad(new Insets(0, 0, 0, 0));
77     textChunks.setBorder(null);
78
79     if (usage == null || usageNode instanceof ShowUsagesAction.StringNode) {
80       textChunks.append(value != null ? value.toString() : "", SimpleTextAttributes.REGULAR_BOLD_ATTRIBUTES);
81       return textComponentSpanningWholeRow(textChunks, panelBackground, panelForeground, column, list, row);
82     }
83     if (usage == ShowUsagesAction.MORE_USAGES_SEPARATOR) {
84       textChunks.append("...<");
85       textChunks.append("more usages", SimpleTextAttributes.REGULAR_BOLD_ATTRIBUTES);
86       textChunks.append(">...");
87       return textComponentSpanningWholeRow(textChunks, panelBackground, panelForeground, column, list, row);
88     }
89     else if (usage == ShowUsagesAction.USAGES_OUTSIDE_SCOPE_SEPARATOR) {
90       textChunks.append("...<");
91       textChunks.append(UsageViewManagerImpl.outOfScopeMessage(myOutOfScopeUsages.get(), mySearchScope), SimpleTextAttributes.REGULAR_BOLD_ATTRIBUTES);
92       textChunks.append(">...");
93       return textComponentSpanningWholeRow(textChunks, panelBackground, panelForeground, column, list, row);
94     }
95
96     if (column == 0) {
97       GroupNode parent = (GroupNode)usageNode.getParent();
98       appendGroupText(parent, panel, fileBgColor);
99     }
100     else if (usage != ShowUsagesAction.MORE_USAGES_SEPARATOR && usage != ShowUsagesAction.USAGES_OUTSIDE_SCOPE_SEPARATOR) {
101       UsagePresentation presentation = usage.getPresentation();
102       TextChunk[] text = presentation.getText();
103
104       if (column == 1) {
105         final Icon icon = presentation.getIcon();
106         textChunks.setIcon(icon == null ? EmptyIcon.ICON_16 : icon);
107         if (text.length != 0) {
108           SimpleTextAttributes attributes = isSelected ?
109             new SimpleTextAttributes(bg, fg, fg, SimpleTextAttributes.STYLE_ITALIC) :
110             deriveAttributesWithColor(text[0].getSimpleAttributesIgnoreBackground(), fileBgColor);
111           textChunks.append(text[0].getText(), attributes);
112         }
113       }
114       else if (column == 2) {
115         for (int i = 1; i < text.length; i++) {
116           TextChunk textChunk = text[i];
117           final SimpleTextAttributes attrs = textChunk.getSimpleAttributesIgnoreBackground();
118           SimpleTextAttributes attributes = isSelected ?
119             new SimpleTextAttributes(bg, fg, fg, attrs.getStyle()) : deriveAttributesWithColor(attrs, fileBgColor);
120           textChunks.append(textChunk.getText(), attributes);
121         }
122       }
123       else {
124         assert false : column;
125       }
126     }
127     panel.add(textChunks);
128     return panel;
129   }
130
131   @NotNull
132   private static Component textComponentSpanningWholeRow(@NotNull SimpleColoredComponent chunks,
133                                                          Color panelBackground,
134                                                          Color panelForeground,
135                                                          final int column,
136                                                          @NotNull final JTable table, int row) {
137     final SimpleColoredComponent component = new SimpleColoredComponent() {
138       @Override
139       protected void doPaint(Graphics2D g) {
140         int offset = 0;
141         int i = 0;
142         final TableColumnModel columnModel = table.getColumnModel();
143         while (i < column) {
144           offset += columnModel.getColumn(i).getWidth();
145           i++;
146         }
147         g.translate(-offset, 0);
148
149         //if (column == columnModel.getColumnCount()-1) {
150         //}
151         setSize(getWidth()+offset, getHeight()); // should increase the column width so that selection background will be visible even after offset translation
152
153         super.doPaint(g);
154
155         g.translate(+offset, 0);
156       }
157
158       @NotNull
159       @Override
160       public Dimension getPreferredSize() {
161         //return super.getPreferredSize();
162         return column == table.getColumnModel().getColumnCount()-1 ? super.getPreferredSize() : new Dimension(0,0);
163         // it should span the whole row, so we can't return any specific value here,
164         // because otherwise it would be used in the "max width" calculation in com.intellij.find.actions.ShowUsagesAction.calcMaxWidth
165       }
166     };
167
168     component.setIpad(new Insets(0,0,0,0));
169     component.setBorder(null);
170     component.setBackground(panelBackground);
171     component.setForeground(panelForeground);
172
173     for (SimpleColoredComponent.ColoredIterator iterator = chunks.iterator(); iterator.hasNext(); ) {
174       iterator.next();
175       String fragment = iterator.getFragment();
176       SimpleTextAttributes attributes = iterator.getTextAttributes();
177       attributes = attributes.derive(attributes.getStyle(), panelForeground, panelBackground, attributes.getWaveColor());
178       component.append(fragment, attributes);
179     }
180
181     return component;
182   }
183
184   private static SimpleTextAttributes deriveAttributesWithColor(SimpleTextAttributes attributes, Color fileBgColor) {
185     if (fileBgColor != null) {
186       attributes = attributes.derive(-1,null, fileBgColor,null);
187     }
188     return attributes;
189   }
190
191   private Color getBackgroundColor(boolean isSelected, Usage usage) {
192     Color fileBgColor = null;
193     if (isSelected) {
194       fileBgColor = UIUtil.getListSelectionBackground();
195     }
196     else {
197       VirtualFile virtualFile = usage instanceof UsageInFile ? ((UsageInFile)usage).getFile() : null;
198       if (virtualFile != null) {
199         Project project = myUsageView.getProject();
200         PsiFile psiFile = PsiManager.getInstance(project).findFile(virtualFile);
201         if (psiFile != null && psiFile.isValid()) {
202           final Color color = FileColorManager.getInstance(project).getRendererBackground(psiFile);
203           if (color != null) fileBgColor = color;
204         }
205       }
206     }
207     return fileBgColor;
208   }
209
210   private void appendGroupText(final GroupNode node, JPanel panel, Color fileBgColor) {
211     UsageGroup group = node == null ? null : node.getGroup();
212     if (group == null) return;
213     GroupNode parentGroup = (GroupNode)node.getParent();
214     appendGroupText(parentGroup, panel, fileBgColor);
215     if (node.canNavigateToSource()) {
216       SimpleColoredComponent renderer = new SimpleColoredComponent();
217
218       renderer.setIcon(group.getIcon(false));
219       SimpleTextAttributes attributes = deriveAttributesWithColor(SimpleTextAttributes.REGULAR_ATTRIBUTES, fileBgColor);
220       renderer.append(group.getText(myUsageView), attributes);
221       renderer.append(" ", attributes);
222       renderer.setIpad(new Insets(0,0,0,0));
223       renderer.setBorder(null);
224       panel.add(renderer);
225     }
226   }
227 }