ba50dc7dfbe37f4dcd2837c262459ab2fee4c348
[idea/community.git] / platform / analysis-impl / src / com / intellij / codeHighlighting / RainbowHighlighter.java
1 /*
2  * Copyright 2000-2016 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.codeHighlighting;
17
18 import com.intellij.codeInsight.daemon.impl.HighlightInfo;
19 import com.intellij.codeInsight.daemon.impl.HighlightInfoType;
20 import com.intellij.lang.annotation.HighlightSeverity;
21 import com.intellij.openapi.application.ApplicationManager;
22 import com.intellij.openapi.editor.DefaultLanguageHighlighterColors;
23 import com.intellij.openapi.editor.colors.EditorColorsManager;
24 import com.intellij.openapi.editor.colors.TextAttributesKey;
25 import com.intellij.openapi.editor.colors.TextAttributesScheme;
26 import com.intellij.openapi.editor.markup.TextAttributes;
27 import com.intellij.openapi.util.registry.Registry;
28 import com.intellij.openapi.util.text.StringUtil;
29 import com.intellij.psi.PsiElement;
30 import com.intellij.ui.ColorUtil;
31 import com.intellij.ui.JBColor;
32 import org.jetbrains.annotations.NotNull;
33 import org.jetbrains.annotations.Nullable;
34
35 import java.awt.*;
36 import java.util.ArrayList;
37 import java.util.List;
38 import java.util.stream.Collectors;
39
40 import static com.intellij.util.ObjectUtils.notNull;
41
42 public class RainbowHighlighter {
43   private final static String UNIT_TEST_COLORS = "#000001,#000002,#000003,#000004"; // Do not modify!
44   private final static int COLOR_COUNT = 16;
45
46   @NotNull private final TextAttributesScheme myColorsScheme;
47   @NotNull private final List<Color> myRainbowColors;
48
49   public RainbowHighlighter(@Nullable TextAttributesScheme colorsScheme) {
50     myColorsScheme = colorsScheme != null ? colorsScheme : EditorColorsManager.getInstance().getGlobalScheme();
51     myRainbowColors = generateColorSequence(myColorsScheme);
52   }
53
54   public static final HighlightInfoType RAINBOW_ELEMENT =
55     new HighlightInfoType.HighlightInfoTypeImpl(HighlightSeverity.INFORMATION, DefaultLanguageHighlighterColors.CONSTANT);
56
57   public static boolean isRainbowEnabled() {
58     return Registry.is("editor.rainbow.identifiers", false);
59   }
60
61   @NotNull
62   public Color calculateForeground(int colorIndex) {
63     return myRainbowColors.get(Math.abs(colorIndex) % myRainbowColors.size());
64   }
65
66   public int getColorsCount() {
67     return myRainbowColors.size();
68   }
69
70   private static List<Color> generateColorSequence(@NotNull TextAttributesScheme colorsScheme) {
71     String colorDump = ApplicationManager.getApplication().isUnitTestMode()
72       ? UNIT_TEST_COLORS
73       : Registry.get("rainbow.highlighter.colors").asString();
74
75     final List<String> registryColors = StringUtil.split(colorDump, ",");
76     if (!registryColors.isEmpty()) {
77       return registryColors.stream().map((s -> ColorUtil.fromHex(s.trim()))).collect(Collectors.toList());
78     }
79
80     List<Color> colors = new ArrayList<Color>(COLOR_COUNT);
81     TextAttributes attributes = colorsScheme.getAttributes(DefaultLanguageHighlighterColors.CONSTANT);
82     Color foregroundColor = notNull(attributes != null ? attributes.getForegroundColor() : null, JBColor.gray);
83     float[] floats = Color.RGBtoHSB(foregroundColor.getRed(), foregroundColor.getGreen(), foregroundColor.getBlue(), null);
84     for (int i = 0; i < COLOR_COUNT; ++i) {
85       float factor = ((float)i) / COLOR_COUNT;
86       Color color = Color.getHSBColor(factor, .7f, floats[2] * 1.3f);
87       //noinspection UseJBColor
88       colors.add(new Color(color.getRed()/2, color.getGreen(), color.getBlue())); // to highlight errors we reduce Red color
89     }
90     return colors;
91   }
92
93   public HighlightInfo getInfo(int colorIndex, @Nullable PsiElement id, @Nullable TextAttributesKey colorKey) {
94     if (id == null) {
95       return null;
96     }
97     if (colorKey == null) {
98       colorKey = DefaultLanguageHighlighterColors.LOCAL_VARIABLE;
99     }
100     return HighlightInfo
101       .newHighlightInfo(RAINBOW_ELEMENT)
102       .textAttributes(TextAttributes
103                         .fromFlyweight(myColorsScheme
104                                          .getAttributes(colorKey)
105                                          .getFlyweight()
106                                          .withForeground(calculateForeground(colorIndex))))
107       .range(id)
108       .create();
109   }
110 }