@Override
  public TextAttributesKey @NotNull [] getTokenHighlights(final IElementType tokenType) {
    return pack(map.get(tokenType));
  }
@Override
  public TextAttributesKey @NotNull [] getTokenHighlights(final IElementType tokenType)