From af048fa2b41a8ea8c03e2df82dc288870fc0d982 Mon Sep 17 00:00:00 2001 From: "Rustam.Vishnyakov" Date: Thu, 2 Feb 2012 14:41:05 +0400 Subject: [PATCH] Java custom folding enabled --- .../codeInsight/folding/impl/JavaFoldingBuilder.java | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/java/java-impl/src/com/intellij/codeInsight/folding/impl/JavaFoldingBuilder.java b/java/java-impl/src/com/intellij/codeInsight/folding/impl/JavaFoldingBuilder.java index 5809eb59a3d8..207b53242598 100644 --- a/java/java-impl/src/com/intellij/codeInsight/folding/impl/JavaFoldingBuilder.java +++ b/java/java-impl/src/com/intellij/codeInsight/folding/impl/JavaFoldingBuilder.java @@ -33,6 +33,7 @@ import com.intellij.psi.impl.source.PsiClassReferenceType; import com.intellij.psi.impl.source.SourceTreeToPsiMap; import com.intellij.psi.impl.source.jsp.jspJava.JspHolderMethod; import com.intellij.psi.impl.source.tree.JavaDocElementType; +import com.intellij.psi.impl.source.tree.JavaElementType; import com.intellij.psi.javadoc.PsiDocComment; import com.intellij.psi.tree.IElementType; import com.intellij.psi.util.PropertyUtil; @@ -740,5 +741,11 @@ public class JavaFoldingBuilder extends CustomFoldingBuilder implements DumbAwar protected boolean isCustomFoldingCandidate(ASTNode node) { return node.getElementType() == JavaTokenType.END_OF_LINE_COMMENT; } + + @Override + protected boolean mayContainCustomFoldings(ASTNode node) { + IElementType nodeType = node.getElementType(); + return nodeType == JavaElementType.CLASS || nodeType == JavaElementType.CODE_BLOCK; + } }