Merge pull request #3445 from jojo2357/equation-folding

Fold Equations
Hannah-Sten · Feb 6, 2024 · 029edc8 · 029edc8
2 parents 2eb6d85 + 35cb8ab
commit 029edc8
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 2 deletions.
diff --git a/src/nl/hannahsten/texifyidea/editor/folding/LatexMathSymbolFoldingBuilder.kt b/src/nl/hannahsten/texifyidea/editor/folding/LatexMathSymbolFoldingBuilder.kt
@@ -9,8 +9,10 @@ import com.intellij.openapi.project.DumbAware
 import com.intellij.psi.PsiElement
 import nl.hannahsten.texifyidea.lang.commands.LatexMathCommand
 import nl.hannahsten.texifyidea.psi.LatexCommands
+import nl.hannahsten.texifyidea.psi.LatexEnvironmentContent
 import nl.hannahsten.texifyidea.psi.LatexMathEnvironment
 import nl.hannahsten.texifyidea.util.parser.childrenOfType
+import nl.hannahsten.texifyidea.util.parser.inMathContext
 
 /**
  * @author Sten Wessel
@@ -19,9 +21,10 @@ class LatexMathSymbolFoldingBuilder : FoldingBuilderEx(), DumbAware {
 
     override fun buildFoldRegions(root: PsiElement, document: Document, quick: Boolean): Array<FoldingDescriptor> {
         val descriptors = listOf<FoldingDescriptor>().toMutableList()
-        val mathEnvironments = root.childrenOfType(LatexMathEnvironment::class)
+        val inlineOrDisplayMath = root.childrenOfType(LatexMathEnvironment::class)
+        val mathEnvironments = root.childrenOfType(LatexEnvironmentContent::class).filter { it.inMathContext() }
 
-        for (mathEnvironment in mathEnvironments) {
+        for (mathEnvironment in inlineOrDisplayMath + mathEnvironments) {
             val group = FoldingGroup.newGroup("latexMathSymbol")
             val commands = mathEnvironment.childrenOfType(LatexCommands::class)
 

diff --git a/test/nl/hannahsten/texifyidea/editor/LatexFoldingTest.kt b/test/nl/hannahsten/texifyidea/editor/LatexFoldingTest.kt
@@ -34,6 +34,13 @@ class LatexFoldingTest : BasePlatformTestCase() {
         }
     }
 
+    fun testMathSymbolFoldingInEnvironment() {
+        // Unicode issues on windows
+        if (!SystemInfo.isWindows) {
+            myFixture.testFolding("$testDataPath/math-symbols-environment.tex")
+        }
+    }
+
     fun testSectionFolding() {
         myFixture.testFolding("$testDataPath/sections.tex")
     }

diff --git a/test/resources/editor/folding/math-symbols-environment.tex b/test/resources/editor/folding/math-symbols-environment.tex
@@ -0,0 +1,3 @@
+\begin{equation}<fold text='...'>
+    <fold text='ξ'>\xi</fold>^<fold text='α'>\alpha</fold>
+</fold>\end{equation}