platform/lang-impl/src/com/intellij/application/options/colors/RainbowAttributeDescriptor.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.application.options.colors;
  17
  18 import com.intellij.codeHighlighting.RainbowHighlighter;
  19 import com.intellij.openapi.editor.colors.EditorColorsScheme;
  20 import com.intellij.openapi.editor.colors.EditorSchemeAttributeDescriptorWithPath;
  21 import com.intellij.openapi.editor.colors.TextAttributesKey;
  22 import com.intellij.openapi.editor.markup.TextAttributes;
  23 import com.intellij.openapi.util.Pair;
  24 import org.jetbrains.annotations.NotNull;
  25
  26 import java.awt.*;
  27 import java.util.ArrayList;
  28 import java.util.List;
  29
  30 class RainbowAttributeDescriptor implements EditorSchemeAttributeDescriptorWithPath {
  31   private final String myGroup;
  32   private final String myDisplayName;
  33   private final EditorColorsScheme myScheme;
  34
  35   // can be shared between instances
  36   private final RainbowInSchemeState myCurState;
  37   private final RainbowInSchemeState myInitState;
  38
  39   public RainbowAttributeDescriptor(@NotNull String group,
  40                                     @NotNull String displayNameWithPath,
  41                                     @NotNull EditorColorsScheme scheme,
  42                                     RainbowInSchemeState initState,
  43                                     RainbowInSchemeState curState) {
  44     myDisplayName = displayNameWithPath;
  45     myInitState = initState;
  46     myCurState = curState;
  47     myScheme = scheme;
  48     myGroup = group;
  49   }
  50
  51   @Override
  52   public String toString() {
  53     return myDisplayName;
  54   }
  55
  56   @Override
  57   public String getGroup() {
  58     return myGroup;
  59   }
  60
  61   @Override
  62   public String getType() {
  63     return RainbowHighlighter.RAINBOW_TYPE;
  64   }
  65
  66   @Override
  67   public EditorColorsScheme getScheme() {
  68     return myScheme;
  69   }
  70
  71   @Override
  72   public void apply(@NotNull EditorColorsScheme scheme) {
  73     myCurState.apply(scheme);
  74   }
  75
  76   @Override
  77   public boolean isModified() {
  78     return !myInitState.equals(myCurState);
  79   }
  80
  81   public List<Pair<Boolean, Color>> getRainbowCurState() {
  82     return myCurState.myRainbowState;
  83   }
  84
  85   public Color getDefaultColor(int index) {
  86     return RainbowHighlighter.RAINBOW_COLOR_KEYS[index].getDefaultAttributes().getForegroundColor();
  87   }
  88
  89   public static class RainbowInSchemeState {
  90     private final List<Pair<Boolean, Color>> myRainbowState = new ArrayList<>();
  91
  92     public RainbowInSchemeState(@NotNull EditorColorsScheme scheme) {
  93       for (TextAttributesKey rainbowKey : RainbowHighlighter.RAINBOW_COLOR_KEYS) {
  94         myRainbowState.add(getColorStateFromScheme(scheme, rainbowKey));
  95       }
  96     }
  97
  98     public RainbowInSchemeState(@NotNull RainbowInSchemeState state) {
  99       copyFrom(state);
 100     }
 101
 102     public void copyFrom(@NotNull RainbowInSchemeState state) {
 103       assert this != state;
 104
 105       myRainbowState.clear();
 106       myRainbowState.addAll(state.myRainbowState);
 107     }
 108
 109     public void apply(@NotNull EditorColorsScheme scheme) {
 110       int i = 0;
 111       for (TextAttributesKey rainbowKey : RainbowHighlighter.RAINBOW_COLOR_KEYS) {
 112         Pair<Boolean, Color> pair = myRainbowState.get(i);
 113         scheme.setAttributes(rainbowKey, pair.first ? new TextAttributes(pair.second, null, null, null, Font.PLAIN)
 114                                                     : rainbowKey.getDefaultAttributes());
 115         ++i;
 116       }
 117     }
 118
 119     @NotNull
 120     private static Pair<Boolean, Color> getColorStateFromScheme(@NotNull EditorColorsScheme scheme, TextAttributesKey rainbowKey) {
 121       TextAttributes schemeAttributes = scheme.getAttributes(rainbowKey);
 122       @NotNull Color defaultRainbow = rainbowKey.getDefaultAttributes().getForegroundColor();
 123       Pair<Boolean, Color> pair;
 124       if (schemeAttributes == null) {
 125         pair = Pair.create(false, defaultRainbow);
 126       }
 127       else {
 128         Color schemeColor = schemeAttributes.getForegroundColor();
 129         if (schemeColor == null) {
 130           pair = Pair.create(false, defaultRainbow);
 131         }
 132         else {
 133           pair = Pair.create(!defaultRainbow.equals(schemeColor), schemeColor);
 134         }
 135       }
 136       return pair;
 137     }
 138
 139     @Override
 140     public boolean equals(Object o) {
 141       if (this == o) return true;
 142       if (o == null || getClass() != o.getClass()) return false;
 143
 144       RainbowInSchemeState state = (RainbowInSchemeState)o;
 145       return myRainbowState.equals(state.myRainbowState);
 146     }
 147
 148     @Override
 149     public int hashCode() {
 150       return myRainbowState.hashCode();
 151     }
 152   }
 153 }