// This class is a runtime engine for parsers generated by Grammar-Kit. // Do not modify or refactor without complete investigation and/or review. package com.intellij.lang.parser; import com.intellij.codeInsight.completion.impl.CamelHumpMatcher; import com.intellij.lang.*; import com.intellij.lang.impl.PsiBuilderAdapter; import com.intellij.lang.impl.PsiBuilderImpl; import com.intellij.lexer.Lexer; import com.intellij.openapi.diagnostic.Logger; import com.intellij.openapi.util.Comparing; import com.intellij.openapi.util.Key; import com.intellij.openapi.util.Pair; import com.intellij.openapi.util.text.StringHash; import com.intellij.openapi.util.text.StringUtil; import com.intellij.psi.PsiFile; import com.intellij.psi.PsiReference; import com.intellij.psi.TokenType; import com.intellij.psi.impl.source.resolve.FileContextUtil; import com.intellij.psi.impl.source.tree.CompositePsiElement; import com.intellij.psi.tree.ICompositeElementType; import com.intellij.psi.tree.IElementType; import com.intellij.psi.tree.TokenSet; import com.intellij.util.Function; import com.intellij.util.PairProcessor; import com.intellij.util.containers.ContainerUtil; import com.intellij.util.containers.LimitedPool; import org.jetbrains.annotations.NotNull; import org.jetbrains.annotations.Nullable; import java.util.ArrayList; import java.util.Arrays; import java.util.Collection; import java.util.LinkedList; /** * @author gregsh */ public class GeneratedParserUtilBase { private static final Logger LOG = Logger.getInstance("org.intellij.grammar.parser.GeneratedParserUtilBase"); private static final int MAX_RECURSION_LEVEL = 1000; private static final int MAX_VARIANTS_SIZE = 10000; private static final int MAX_VARIANTS_TO_DISPLAY = 50; private static final int INITIAL_VARIANTS_SIZE = 1000; private static final int VARIANTS_POOL_SIZE = 10000; private static final int FRAMES_POOL_SIZE = 500; public static final IElementType DUMMY_BLOCK = new DummyBlockElementType(); public interface Parser { boolean parse(PsiBuilder builder, int level); } public static final Parser TOKEN_ADVANCER = new Parser() { @Override public boolean parse(PsiBuilder builder, int level) { if (builder.eof()) return false; builder.advanceLexer(); return true; } }; public static final Parser TRUE_CONDITION = new Parser() { @Override public boolean parse(PsiBuilder builder, int level) { return true; } }; public static boolean eof(PsiBuilder builder_, int level_) { return builder_.eof(); } public static int current_position_(PsiBuilder builder_) { return builder_.rawTokenIndex(); } public static boolean recursion_guard_(PsiBuilder builder_, int level_, String funcName_) { if (level_ > MAX_RECURSION_LEVEL) { builder_.error("Maximum recursion level (" + MAX_RECURSION_LEVEL + ") reached in '" + funcName_ + "'"); return false; } return true; } public static boolean empty_element_parsed_guard_(PsiBuilder builder_, String funcName_, int prev_position_) { if (prev_position_ == current_position_(builder_)) { builder_.error("Empty element parsed in '" + funcName_ + "' at offset " + builder_.getCurrentOffset()); return false; } return true; } public static boolean invalid_left_marker_guard_(PsiBuilder builder_, PsiBuilder.Marker marker_, String funcName_) { //builder_.error("Invalid left marker encountered in " + funcName_ +" at offset " + builder_.getCurrentOffset()); boolean goodMarker = marker_ != null; // && ((LighterASTNode)marker_).getTokenType() != TokenType.ERROR_ELEMENT; if (!goodMarker) return false; ErrorState state = ErrorState.get(builder_); Frame frame = state.frameStack.peekLast(); return frame == null || frame.errorReportedAt <= builder_.rawTokenIndex(); } public static TokenSet create_token_set_(IElementType... tokenTypes_) { return TokenSet.create(tokenTypes_); } private static boolean consumeTokens(PsiBuilder builder_, boolean smart, int pin, IElementType... tokens) { ErrorState state = ErrorState.get(builder_); if (state.completionState != null && state.predicateCount == 0) { addCompletionVariant(builder_, state.completionState, tokens); } // suppress single token completion CompletionState completionState = state.completionState; state.completionState = null; boolean result_ = true; boolean pinned_ = false; for (int i = 0, tokensLength = tokens.length; i < tokensLength; i++) { if (pin > 0 && i == pin) pinned_ = result_; if (result_ || pinned_) { boolean fast = smart && i == 0; if (!(fast ? consumeTokenFast(builder_, tokens[i]) : consumeToken(builder_, tokens[i]))) { result_ = false; if (pin < 0 || pinned_) report_error_(builder_, state, false); } } } state.completionState = completionState; return pinned_ || result_; } public static boolean consumeTokens(PsiBuilder builder_, int pin_, IElementType... token) { return consumeTokens(builder_, false, pin_, token); } public static boolean consumeTokensSmart(PsiBuilder builder_, int pin_, IElementType... token) { return consumeTokens(builder_, true, pin_, token); } public static boolean parseTokens(PsiBuilder builder_, int pin_, IElementType... tokens) { return parseTokens(builder_, false, pin_, tokens); } public static boolean parseTokensSmart(PsiBuilder builder_, int pin_, IElementType... tokens) { return parseTokens(builder_, true, pin_, tokens); } public static boolean parseTokens(PsiBuilder builder_, boolean smart, int pin_, IElementType... tokens) { PsiBuilder.Marker marker_ = builder_.mark(); boolean result_ = consumeTokens(builder_, smart, pin_, tokens); if (!result_) { marker_.rollbackTo(); } else { marker_.drop(); } return result_; } public static boolean consumeTokenSmart(PsiBuilder builder_, IElementType token) { addCompletionVariantSmart(builder_, token); return consumeTokenFast(builder_, token); } public static boolean consumeTokenSmart(PsiBuilder builder_, String token) { addCompletionVariantSmart(builder_, token); return consumeTokenFast(builder_, token); } public static boolean consumeToken(PsiBuilder builder_, IElementType token) { addVariantSmart(builder_, token, true); if (nextTokenIsFast(builder_, token)) { builder_.advanceLexer(); return true; } return false; } public static boolean consumeTokenFast(PsiBuilder builder_, IElementType token) { if (nextTokenIsFast(builder_, token)) { builder_.advanceLexer(); return true; } return false; } public static boolean consumeToken(PsiBuilder builder_, String text) { return consumeToken(builder_, text, ErrorState.get(builder_).caseSensitive); } public static boolean consumeToken(PsiBuilder builder_, String text, boolean caseSensitive) { addVariantSmart(builder_, text, true); int count = nextTokenIsFast(builder_, text, caseSensitive); if (count > 0) { while (count-- > 0) builder_.advanceLexer(); return true; } return false; } public static boolean consumeTokenFast(PsiBuilder builder_, String text) { int count = nextTokenIsFast(builder_, text, ErrorState.get(builder_).caseSensitive); if (count > 0) { while (count-- > 0) builder_.advanceLexer(); return true; } return false; } public static boolean nextTokenIsFast(PsiBuilder builder_, IElementType token) { return builder_.getTokenType() == token; } public static boolean nextTokenIsFast(PsiBuilder builder_, IElementType... tokens) { IElementType tokenType = builder_.getTokenType(); for (IElementType token : tokens) { if (token == tokenType) return true; } return false; } public static boolean nextTokenIs(PsiBuilder builder_, String frameName, IElementType... tokens) { ErrorState state = ErrorState.get(builder_); if (state.completionState != null) return true; boolean track = !state.suppressErrors && state.predicateCount < 2 && state.predicateSign; if (!track) return nextTokenIsFast(builder_, tokens); IElementType tokenType = builder_.getTokenType(); if (StringUtil.isNotEmpty(frameName)) { addVariantInner(state, builder_.rawTokenIndex(), frameName); } else { for (IElementType token : tokens) { addVariant(builder_, state, token); } } if (tokenType == null) return false; for (IElementType token : tokens) { if (tokenType == token) return true; } return false; } public static boolean nextTokenIs(PsiBuilder builder_, IElementType token) { if (!addVariantSmart(builder_, token, false)) return true; return nextTokenIsFast(builder_, token); } public static boolean nextTokenIs(PsiBuilder builder_, String tokenText) { if (!addVariantSmart(builder_, tokenText, false)) return true; return nextTokenIsFast(builder_, tokenText, ErrorState.get(builder_).caseSensitive) > 0; } public static boolean nextTokenIsFast(PsiBuilder builder_, String tokenText) { return nextTokenIsFast(builder_, tokenText, ErrorState.get(builder_).caseSensitive) > 0; } public static int nextTokenIsFast(PsiBuilder builder_, String tokenText, boolean caseSensitive) { CharSequence sequence = builder_.getOriginalText(); int offset = builder_.getCurrentOffset(); int endOffset = offset + tokenText.length(); CharSequence subSequence = sequence.subSequence(offset, Math.min(endOffset, sequence.length())); if (!Comparing.equal(subSequence, tokenText, caseSensitive)) return 0; int count = 0; while (true) { int nextOffset = builder_.rawTokenTypeStart(++count); if (nextOffset > endOffset) { return -count; } else if (nextOffset == endOffset) { break; } } return count; } private static void addCompletionVariantSmart(PsiBuilder builder_, Object token) { ErrorState state = ErrorState.get(builder_); CompletionState completionState = state.completionState; if (completionState != null && state.predicateCount == 0) { addCompletionVariant(builder_, completionState, token); } } private static boolean addVariantSmart(PsiBuilder builder_, Object token, boolean force) { ErrorState state = ErrorState.get(builder_); // skip FIRST check in completion mode if (state.completionState != null && !force) return false; builder_.eof(); if (!state.suppressErrors && state.predicateCount < 2) { addVariant(builder_, state, token); } return true; } public static void addVariant(PsiBuilder builder_, String text) { addVariant(builder_, ErrorState.get(builder_), text); } private static void addVariant(PsiBuilder builder_, ErrorState state, Object o) { builder_.eof(); // skip whitespaces addVariantInner(state, builder_.rawTokenIndex(), o); CompletionState completionState = state.completionState; if (completionState != null && state.predicateSign) { addCompletionVariant(builder_, completionState, o); } } private static void addVariantInner(ErrorState state, int pos, Object o) { Variant variant = state.VARIANTS.alloc().init(pos, o); if (state.predicateSign) { state.variants.add(variant); if (state.lastExpectedVariantPos < variant.position) { state.lastExpectedVariantPos = variant.position; } } else { state.unexpected.add(variant); } } private static void addCompletionVariant(@NotNull PsiBuilder builder_, @NotNull CompletionState completionState, Object o) { int offset = builder_.getCurrentOffset(); if (!builder_.eof() && offset == builder_.rawTokenTypeStart(1)) return; // suppress for zero-length tokens boolean add = false; int diff = completionState.offset - offset; String text = completionState.convertItem(o); int length = text == null? 0 : text.length(); if (length == 0) return; if (diff == 0) { add = true; } else if (diff > 0 && diff <= length) { CharSequence fragment = builder_.getOriginalText().subSequence(offset, completionState.offset); add = completionState.prefixMatches(fragment.toString(), text); } else if (diff < 0) { for (int i=-1; ; i--) { IElementType type = builder_.rawLookup(i); int tokenStart = builder_.rawTokenTypeStart(i); if (isWhitespaceOrComment(builder_, type)) { diff = completionState.offset - tokenStart; } else if (type != null && tokenStart < completionState.offset) { CharSequence fragment = builder_.getOriginalText().subSequence(tokenStart, completionState.offset); if (completionState.prefixMatches(fragment.toString(), text)) { diff = completionState.offset - tokenStart; } break; } else break; } add = diff >= 0 && diff < length; } add = add && length > 1 && !(text.charAt(0) == '<' && text.charAt(length - 1) == '>') && !(text.charAt(0) == '\'' && text.charAt(length - 1) == '\'' && length < 5); if (add) { completionState.addItem(builder_, text); } } public static boolean isWhitespaceOrComment(@NotNull PsiBuilder builder_, @Nullable IElementType type) { return ((PsiBuilderImpl)((Builder)builder_).getDelegate()).whitespaceOrComment(type); } // here's the new section API for compact parsers & less IntelliJ platform API exposure public static final int _NONE_ = 0x0; public static final int _COLLAPSE_ = 0x1; public static final int _LEFT_ = 0x2; public static final int _LEFT_INNER_ = 0x4; public static final int _AND_ = 0x8; public static final int _NOT_ = 0x10; // simple enter/exit methods pair that doesn't require frame object public static PsiBuilder.Marker enter_section_(PsiBuilder builder_) { return builder_.mark(); } public static void exit_section_(PsiBuilder builder_, PsiBuilder.Marker marker, @Nullable IElementType elementType, boolean result) { close_marker_impl_(ErrorState.get(builder_).frameStack.peekLast(), marker, elementType, result); } // complex enter/exit methods pair with frame object public static PsiBuilder.Marker enter_section_(PsiBuilder builder_, int level, int modifiers, @Nullable String frameName) { PsiBuilder.Marker marker = builder_.mark(); enter_section_impl_(builder_, level, modifiers, frameName); return marker; } private static void enter_section_impl_(PsiBuilder builder_, int level, int modifiers, @Nullable String frameName) { ErrorState state = ErrorState.get(builder_); Frame frame = state.FRAMES.alloc().init(builder_, state, level, modifiers, frameName); if (((frame.modifiers & _LEFT_) | (frame.modifiers & _LEFT_INNER_)) != 0) { PsiBuilder.Marker left = (PsiBuilder.Marker)builder_.getLatestDoneMarker(); if (invalid_left_marker_guard_(builder_, left, frameName)) { frame.leftMarker = left; } } state.frameStack.add(frame); if ((modifiers & _AND_) != 0) { if (state.predicateCount == 0 && !state.predicateSign) { throw new AssertionError("Incorrect false predicate sign"); } state.predicateCount++; } else if ((modifiers & _NOT_) != 0) { if (state.predicateCount == 0) { state.predicateSign = false; } else { state.predicateSign = !state.predicateSign; } state.predicateCount++; } } public static void exit_section_(PsiBuilder builder_, int level, PsiBuilder.Marker marker, @Nullable IElementType elementType, boolean result, boolean pinned, @Nullable Parser eatMore) { ErrorState state = ErrorState.get(builder_); Frame frame = state.frameStack.pollLast(); if (frame == null || level != frame.level) { LOG.error("Unbalanced error section: got " + frame + ", expected level " + level); if (frame != null) state.FRAMES.recycle(frame); close_marker_impl_(frame, marker, elementType, result); return; } if (((frame.modifiers & _AND_) | (frame.modifiers & _NOT_)) != 0) { close_marker_impl_(frame, marker, null, false); state.predicateCount--; if ((frame.modifiers & _NOT_) != 0) state.predicateSign = !state.predicateSign; state.FRAMES.recycle(frame); return; } exit_section_impl_(state, frame, builder_, marker, elementType, result, pinned); int initialPos = builder_.rawTokenIndex(); boolean willFail = !result && !pinned; if (willFail && initialPos == frame.position && state.lastExpectedVariantPos == frame.position && frame.name != null && state.variants.size() - frame.variantCount > 1) { state.clearVariants(true, frame.variantCount); addVariantInner(state, initialPos, frame.name); } if (!state.suppressErrors && eatMore != null) { state.suppressErrors = true; final boolean eatMoreFlagOnce = !builder_.eof() && eatMore.parse(builder_, frame.level + 1); final int lastErrorPos = getLastVariantPos(state, initialPos); boolean eatMoreFlag = eatMoreFlagOnce || !result && frame.position == initialPos && lastErrorPos > frame.position; PsiBuilderImpl.ProductionMarker latestDoneMarker = (pinned || result) && (state.altMode || elementType != null) && eatMoreFlagOnce ? (PsiBuilderImpl.ProductionMarker)builder_.getLatestDoneMarker() : null; PsiBuilder.Marker extensionMarker = null; IElementType extensionTokenType = null; // whitespace prefix makes the very first frame offset bigger than marker start offset which is always 0 if (latestDoneMarker instanceof PsiBuilder.Marker && frame.position >= latestDoneMarker.getStartIndex() && frame.position <= latestDoneMarker.getEndIndex()) { extensionMarker = ((PsiBuilder.Marker)latestDoneMarker).precede(); extensionTokenType = latestDoneMarker.getTokenType(); ((PsiBuilder.Marker)latestDoneMarker).drop(); } // advance to the last error pos // skip tokens until lastErrorPos. parseAsTree might look better here... int parenCount = 0; while ((eatMoreFlag || parenCount > 0) && builder_.rawTokenIndex() < lastErrorPos) { if (state.braces != null) { if (builder_.getTokenType() == state.braces[0].getLeftBraceType()) parenCount ++; else if (builder_.getTokenType() == state.braces[0].getRightBraceType()) parenCount --; } builder_.advanceLexer(); eatMoreFlag = eatMore.parse(builder_, frame.level + 1); } boolean errorReported = frame.errorReportedAt == initialPos || !result && frame.errorReportedAt >= frame.position; if (errorReported) { if (eatMoreFlag) { builder_.advanceLexer(); parseAsTree(state, builder_, frame.level + 1, DUMMY_BLOCK, true, TOKEN_ADVANCER, eatMore); } } else if (eatMoreFlag) { errorReported = reportError(builder_, state, frame, true, true); parseAsTree(state, builder_, frame.level + 1, DUMMY_BLOCK, true, TOKEN_ADVANCER, eatMore); } else if (eatMoreFlagOnce || (!result && frame.position != builder_.rawTokenIndex())) { errorReported = reportError(builder_, state, frame, true, false); } if (extensionMarker != null) { extensionMarker.done(extensionTokenType); } state.suppressErrors = false; if (errorReported || result) { state.clearVariants(true, 0); state.clearVariants(false, 0); state.lastExpectedVariantPos = -1; } } else if (!result && pinned && frame.errorReportedAt < 0) { // do not report if there are errors beyond current position if (getLastVariantPos(state, initialPos) == initialPos) { // do not force, inner recoverRoot might have skipped some tokens reportError(builder_, state, frame, false, false); } } // propagate errorReportedAt up the stack to avoid duplicate reporting Frame prevFrame = willFail && eatMore == null ? null : state.frameStack.peekLast(); if (prevFrame != null && prevFrame.errorReportedAt < frame.errorReportedAt) { prevFrame.errorReportedAt = frame.errorReportedAt; } state.FRAMES.recycle(frame); } private static void exit_section_impl_(ErrorState state, Frame frame, PsiBuilder builder_, PsiBuilder.Marker marker, IElementType elementType, boolean result, boolean pinned) { if (elementType != null && marker != null) { if ((frame.modifiers & _COLLAPSE_) != 0) { PsiBuilderImpl.ProductionMarker last = result || pinned? (PsiBuilderImpl.ProductionMarker)builder_.getLatestDoneMarker() : null; if (last != null && last.getStartIndex() == frame.position && state.typeExtends(last.getTokenType(), elementType)) { IElementType resultType = last.getTokenType(); ((PsiBuilder.Marker)last).drop(); marker.done(resultType); return; } } if (result || pinned) { if ((frame.modifiers & _LEFT_INNER_) != 0 && frame.leftMarker != null) { marker.done(elementType); frame.leftMarker.precede().done(((LighterASTNode)frame.leftMarker).getTokenType()); frame.leftMarker.drop(); } else if ((frame.modifiers & _LEFT_) != 0 && frame.leftMarker != null) { marker.drop(); frame.leftMarker.precede().done(elementType); } else { if (frame.level == 0) builder_.eof(); // skip whitespaces marker.done(elementType); } } else { close_marker_impl_(frame, marker, null, false); } } else if (result || pinned) { if (marker != null) marker.drop(); if ((frame.modifiers & _LEFT_INNER_) != 0 && frame.leftMarker != null) { frame.leftMarker.precede().done(((LighterASTNode)frame.leftMarker).getTokenType()); frame.leftMarker.drop(); } } else { close_marker_impl_(frame, marker, null, false); } } private static void close_marker_impl_(Frame frame, PsiBuilder.Marker marker, IElementType elementType, boolean result) { if (marker == null) return; if (result) { if (elementType != null) { marker.done(elementType); } else { marker.drop(); } } else { if (frame != null) { int position = ((PsiBuilderImpl.ProductionMarker)marker).getStartIndex(); if (frame.errorReportedAt > position) { frame.errorReportedAt = frame.errorReportedAtPrev; } } marker.rollbackTo(); } } public static boolean report_error_(PsiBuilder builder_, boolean result_) { if (!result_) report_error_(builder_, ErrorState.get(builder_), false); return result_; } public static void report_error_(PsiBuilder builder_, ErrorState state, boolean advance) { Frame frame = state.frameStack.isEmpty()? null : state.frameStack.getLast(); if (frame == null) { LOG.error("unbalanced enter/exit section call: got null"); return; } int position = builder_.rawTokenIndex(); if (frame.errorReportedAt < position && getLastVariantPos(state, position) <= position) { reportError(builder_, state, frame, true, advance); } } private static int getLastVariantPos(ErrorState state, int defValue) { return state.lastExpectedVariantPos < 0? defValue : state.lastExpectedVariantPos; } private static boolean reportError(PsiBuilder builder_, ErrorState state, Frame frame, boolean force, boolean advance) { String expectedText = state.getExpectedText(builder_); boolean notEmpty = StringUtil.isNotEmpty(expectedText); if (force || notEmpty || advance) { String gotText = builder_.eof()? "unexpected end of file" : notEmpty? "got '" + builder_.getTokenText() +"'" : "'" + builder_.getTokenText() +"' unexpected"; String message = expectedText + gotText; if (advance) { PsiBuilder.Marker mark = builder_.mark(); builder_.advanceLexer(); mark.error(message); } else { builder_.error(message); } builder_.eof(); // skip whitespaces frame.errorReportedAt = builder_.rawTokenIndex(); return true; } return false; } public static final Key COMPLETION_STATE_KEY = Key.create("COMPLETION_STATE_KEY"); public static class CompletionState implements Function { public final int offset; public final Collection items = ContainerUtil.newTroveSet(); public CompletionState(int offset_) { offset = offset_; } @Nullable public String convertItem(Object o) { return o instanceof Object[] ? StringUtil.join((Object[]) o, this, " ") : o.toString(); } @Override public String fun(Object o) { return o.toString(); } public void addItem(@NotNull PsiBuilder builder, @NotNull String text) { items.add(text); } public boolean prefixMatches(@NotNull String prefix, @NotNull String variant) { boolean matches = new CamelHumpMatcher(prefix, false).prefixMatches(variant.replace(' ', '_')); if (matches && StringUtil.isWhiteSpace(prefix.charAt(prefix.length() - 1))) { return StringUtil.startsWithIgnoreCase(variant, prefix); } return matches; } } public static class Builder extends PsiBuilderAdapter { public final ErrorState state; public final PsiParser parser; public Builder(PsiBuilder builder_, ErrorState state_, PsiParser parser_) { super(builder_); state = state_; parser = parser_; } public Lexer getLexer() { return ((PsiBuilderImpl)myDelegate).getLexer(); } } public static PsiBuilder adapt_builder_(IElementType root, PsiBuilder builder, PsiParser parser) { return adapt_builder_(root, builder, parser, null); } public static PsiBuilder adapt_builder_(IElementType root, PsiBuilder builder, PsiParser parser, TokenSet[] extendsSets) { ErrorState state = new ErrorState(); ErrorState.initState(state, builder, root, extendsSets); return new Builder(builder, state, parser); } public static class ErrorState { TokenSet[] extendsSets; public PairProcessor altExtendsChecker; int predicateCount; boolean predicateSign = true; boolean suppressErrors; public final LinkedList frameStack = new LinkedList(); public CompletionState completionState; private boolean caseSensitive; public BracePair[] braces; public boolean altMode; int lastExpectedVariantPos = -1; MyList variants = new MyList(INITIAL_VARIANTS_SIZE); MyList unexpected = new MyList(INITIAL_VARIANTS_SIZE / 10); final LimitedPool VARIANTS = new LimitedPool(VARIANTS_POOL_SIZE, new LimitedPool.ObjectFactory() { @Override public Variant create() { return new Variant(); } @Override public void cleanup(final Variant o) { } }); final LimitedPool FRAMES = new LimitedPool(FRAMES_POOL_SIZE, new LimitedPool.ObjectFactory() { @Override public Frame create() { return new Frame(); } @Override public void cleanup(final Frame o) { } }); public static ErrorState get(PsiBuilder builder) { return ((Builder)builder).state; } public static void initState(ErrorState state, PsiBuilder builder, IElementType root, TokenSet[] extendsSets) { state.extendsSets = extendsSets; PsiFile file = builder.getUserDataUnprotected(FileContextUtil.CONTAINING_FILE_KEY); state.completionState = file == null? null: file.getUserData(COMPLETION_STATE_KEY); Language language = file == null? root.getLanguage() : file.getLanguage(); state.caseSensitive = language.isCaseSensitive(); PairedBraceMatcher matcher = LanguageBraceMatching.INSTANCE.forLanguage(language); state.braces = matcher == null ? null : matcher.getPairs(); if (state.braces != null && state.braces.length == 0) state.braces = null; } public String getExpectedText(PsiBuilder builder_) { int position = builder_.rawTokenIndex(); StringBuilder sb = new StringBuilder(); if (addExpected(sb, position, true)) { sb.append(" expected, "); } else if (addExpected(sb, position, false)) sb.append(" unexpected, "); return sb.toString(); } private boolean addExpected(StringBuilder sb, int position, boolean expected) { MyList list = expected ? variants : unexpected; String[] strings = new String[list.size()]; long[] hashes = new long[strings.length]; Arrays.fill(strings, ""); int count = 0; loop: for (Variant variant : list) { if (position == variant.position) { String text = variant.object.toString(); long hash = StringHash.calc(text); for (int i=0; i 0) { if (count > MAX_VARIANTS_TO_DISPLAY) { sb.append(" and ..."); break; } else { sb.append(", "); } } char c = s.charAt(0); String displayText = c == '<' || StringUtil.isJavaIdentifierStart(c) ? s : '\'' + s + '\''; sb.append(displayText); } if (count > 1 && count < MAX_VARIANTS_TO_DISPLAY) { int idx = sb.lastIndexOf(", "); sb.replace(idx, idx + 1, " or"); } return count > 0; } public void clearVariants(boolean expected, int start) { MyList list = expected? variants : unexpected; if (start < 0 || start >= list.size()) return; for (int i = start, len = list.size(); i < len; i ++) { VARIANTS.recycle(list.get(i)); } list.setSize(start); } boolean typeExtends(IElementType child_, IElementType parent_) { if (child_ == parent_) return true; if (extendsSets != null) { for (TokenSet set : extendsSets) { if (set.contains(child_) && set.contains(parent_)) return true; } } return altExtendsChecker != null && altExtendsChecker.process(child_, parent_); } } public static class Frame { public int offset; public int position; public int level; public int modifiers; public String name; public int variantCount; public int errorReportedAt; public int errorReportedAtPrev; public PsiBuilder.Marker leftMarker; public Frame() { } public Frame init(PsiBuilder builder_, ErrorState state, int level_, int modifiers_, String name_) { offset = builder_.getCurrentOffset(); position = builder_.rawTokenIndex(); level = level_; modifiers = modifiers_; name = name_; variantCount = state.variants.size(); errorReportedAt = -1; Frame prev = state.frameStack.peekLast(); errorReportedAtPrev = prev == null? -1 : prev.errorReportedAt; leftMarker = null; return this; } @Override public String toString() { String mod = modifiers == _NONE_ ? "_NONE_, " : ((modifiers & _COLLAPSE_) != 0? "_CAN_COLLAPSE_, ": "") + ((modifiers & _LEFT_) != 0? "_LEFT_, ": "") + ((modifiers & _LEFT_INNER_) != 0? "_LEFT_INNER_, ": "") + ((modifiers & _AND_) != 0? "_AND_, ": "") + ((modifiers & _NOT_) != 0? "_NOT_, ": ""); return "<" + offset + ", " + mod + level + (errorReportedAt > -1 ? ", [" + errorReportedAt + "]" : "") + ">"; } } private static class Variant { int position; Object object; public Variant init(int pos, Object o) { position = pos; object = o; return this; } @Override public String toString() { return "<" + position + ", " + object + ">"; } @Override public boolean equals(Object o) { if (this == o) return true; if (o == null || getClass() != o.getClass()) return false; Variant variant = (Variant)o; if (position != variant.position) return false; if (!this.object.equals(variant.object)) return false; return true; } @Override public int hashCode() { int result = position; result = 31 * result + object.hashCode(); return result; } } private static final int MAX_CHILDREN_IN_TREE = 10; public static boolean parseAsTree(ErrorState state, final PsiBuilder builder_, int level, final IElementType chunkType, boolean checkBraces, final Parser parser, final Parser eatMoreCondition) { final LinkedList> parenList = new LinkedList>(); final LinkedList> siblingList = new LinkedList>(); PsiBuilder.Marker marker = null; final Runnable checkSiblingsRunnable = new Runnable() { @Override public void run() { main: while (!siblingList.isEmpty()) { final Pair parenPair = parenList.peek(); final int rating = siblingList.getFirst().second; int count = 0; for (Pair pair : siblingList) { if (pair.second != rating || parenPair != null && pair.first == parenPair.second) break main; if (++count >= MAX_CHILDREN_IN_TREE) { final PsiBuilder.Marker parentMarker = pair.first.precede(); while (count-- > 0) { siblingList.removeFirst(); } parentMarker.done(chunkType); siblingList.addFirst(Pair.create(parentMarker, rating + 1)); continue main; } } break; } } }; boolean checkParens = state.braces != null && checkBraces; int totalCount = 0; int tokenCount = 0; if (checkParens) { int tokenIdx = -1; while (builder_.rawLookup(tokenIdx) == TokenType.WHITE_SPACE) tokenIdx --; LighterASTNode doneMarker = builder_.rawLookup(tokenIdx) == state.braces[0].getLeftBraceType() ? builder_.getLatestDoneMarker() : null; if (doneMarker != null && doneMarker.getStartOffset() == builder_.rawTokenTypeStart(tokenIdx) && doneMarker.getTokenType() == TokenType.ERROR_ELEMENT) { parenList.add(Pair.create(((PsiBuilder.Marker)doneMarker).precede(), (PsiBuilder.Marker)null)); } } while (true) { final IElementType tokenType = builder_.getTokenType(); if (checkParens && (tokenType == state.braces[0].getLeftBraceType() || tokenType == state.braces[0].getRightBraceType() && !parenList.isEmpty())) { if (marker != null) { marker.done(chunkType); siblingList.addFirst(Pair.create(marker, 1)); marker = null; tokenCount = 0; } if (tokenType == state.braces[0].getLeftBraceType()) { final Pair prev = siblingList.peek(); parenList.addFirst(Pair.create(builder_.mark(), prev == null ? null : prev.first)); } checkSiblingsRunnable.run(); builder_.advanceLexer(); if (tokenType == state.braces[0].getRightBraceType()) { final Pair pair = parenList.removeFirst(); pair.first.done(chunkType); // drop all markers inside parens while (!siblingList.isEmpty() && siblingList.getFirst().first != pair.second) { siblingList.removeFirst(); } siblingList.addFirst(Pair.create(pair.first, 1)); checkSiblingsRunnable.run(); } } else { if (marker == null) { marker = builder_.mark(); } final boolean result = (!parenList.isEmpty() || eatMoreCondition.parse(builder_, level + 1)) && parser.parse(builder_, level + 1); if (result) { tokenCount++; totalCount++; } if (!result) { break; } } if (tokenCount >= MAX_CHILDREN_IN_TREE && marker != null) { marker.done(chunkType); siblingList.addFirst(Pair.create(marker, 1)); checkSiblingsRunnable.run(); marker = null; tokenCount = 0; } } if (marker != null) { marker.drop(); } for (Pair pair : parenList) { pair.first.drop(); } return totalCount != 0; } private static class DummyBlockElementType extends IElementType implements ICompositeElementType{ DummyBlockElementType() { super("DUMMY_BLOCK", Language.ANY); } @NotNull @Override public ASTNode createCompositeNode() { return new DummyBlock(); } } public static class DummyBlock extends CompositePsiElement { DummyBlock() { super(DUMMY_BLOCK); } @NotNull @Override public PsiReference[] getReferences() { return PsiReference.EMPTY_ARRAY; } @NotNull @Override public Language getLanguage() { return getParent().getLanguage(); } } private static class MyList extends ArrayList { MyList(int initialCapacity) { super(initialCapacity); } protected void setSize(int fromIndex) { removeRange(fromIndex, size()); } @Override public boolean add(E e) { int size = size(); if (size >= MAX_VARIANTS_SIZE) { removeRange(MAX_VARIANTS_SIZE / 4, size - MAX_VARIANTS_SIZE / 4); } return super.add(e); } } }