hide projection annotations outside of the current visible region

Projection regions that overlap with parts of the file outside of the
current visible region cannot be collapsed hence their annotations
should not be painted.

Author: danthe1st

bundles/org.eclipse.jface.text/projection/org/eclipse/jface/text/source/projection/ProjectionAnnotation.java

@@ -73,6 +73,8 @@ public void run() {
 	/** Indicates whether this annotation should be painted as range */
 	private boolean fIsRangeIndication= false;
 
+	private boolean hidden= false;
+
 	/**
 	 * Creates a new expanded projection annotation.
 	 */
@@ -115,6 +117,9 @@ private void drawRangeIndication(GC gc, Canvas canvas, Rectangle r) {
 
 	@Override
 	public void paint(GC gc, Canvas canvas, Rectangle rectangle) {
+		if (hidden) {
+			return;
+		}
 		Image image= getImage(canvas.getDisplay());
 		if (image != null) {
 			ImageUtilities.drawImage(image, gc, canvas, rectangle, SWT.CENTER, SWT.TOP);
@@ -128,6 +133,10 @@ public void paint(GC gc, Canvas canvas, Rectangle rectangle) {
 		}
 	}
 
+	void setHidden(boolean hidden) {
+		this.hidden= hidden;
+	}
+
 	@Override
 	public int getLayer() {
 		return IAnnotationPresentation.DEFAULT_LAYER;

bundles/org.eclipse.jface.text/projection/org/eclipse/jface/text/source/projection/ProjectionViewer.java

@@ -124,6 +124,7 @@ private void processModelChanged(IAnnotationModel model, AnnotationModelEvent ev
 					fProjectionSummary.updateSummaries();
 				}
 				processCatchupRequest(event);
+				correctChangedAnnotationVisibility(event);
 
 			} else if (model == getAnnotationModel() && fProjectionSummary != null) {
 				fProjectionSummary.updateSummaries();
@@ -773,12 +774,31 @@ public void setVisibleRegion(int start, int length) {
 			expandOutsideCurrentVisibleRegion(document);
 			collapseOutsideOfNewVisibleRegion(start, end, document);
 			fConfiguredVisibleRegion= new Region(start, end - start - 1);
+			hideProjectionAnnotationsOutsideOfVisibleRegion();
 		} catch (BadLocationException e) {
 			ILog log= ILog.of(getClass());
 			log.log(new Status(IStatus.WARNING, getClass(), IStatus.OK, null, e));
 		}
 	}
 
+	private void hideProjectionAnnotationsOutsideOfVisibleRegion() throws BadLocationException {
+		for (Iterator<Annotation> it= fProjectionAnnotationModel.getAnnotationIterator(); it.hasNext();) {
+			Annotation annotation= it.next();
+			hideProjectionAnnotationIfPartsAreOutsideOfVisibleRegion(annotation);
+		}
+	}
+
+	private void hideProjectionAnnotationIfPartsAreOutsideOfVisibleRegion(Annotation annotation) throws BadLocationException {
+		Position position= fProjectionAnnotationModel.getPosition(annotation);
+		if (annotation instanceof ProjectionAnnotation a) {
+			if (overlapsWithNonVisibleRegions(position.getOffset(), position.getLength())) {
+				a.setHidden(true);
+			} else {
+				a.setHidden(false);
+			}
+		}
+	}
+
 	private void expandOutsideCurrentVisibleRegion(IDocument document) throws BadLocationException {
 		if (fConfiguredVisibleRegion != null) {
 			expand(0, fConfiguredVisibleRegion.getOffset(), false, true);
@@ -855,6 +875,12 @@ public void resetVisibleRegion() {
 			super.resetVisibleRegion();
 		}
 		fConfiguredVisibleRegion= null;
+		for (Iterator<Annotation> it= fProjectionAnnotationModel.getAnnotationIterator(); it.hasNext();) {
+			Annotation annotation= it.next();
+			if (annotation instanceof ProjectionAnnotation a) {
+				a.setHidden(false);
+			}
+		}
 	}
 
 	@Override
@@ -1090,6 +1116,20 @@ protected final void postCatchupRequest(final AnnotationModelEvent event) {
 		}
 	}
 
+	private void correctChangedAnnotationVisibility(AnnotationModelEvent event) {
+		try {
+			for (Annotation annotation : event.getAddedAnnotations()) {
+				hideProjectionAnnotationIfPartsAreOutsideOfVisibleRegion(annotation);
+			}
+			for (Annotation annotation : event.getChangedAnnotations()) {
+				hideProjectionAnnotationIfPartsAreOutsideOfVisibleRegion(annotation);
+			}
+		} catch (BadLocationException e) {
+			ILog log= ILog.of(getClass());
+			log.log(new Status(IStatus.WARNING, getClass(), IStatus.OK, null, e));
+		}
+	}
+
 	/**
 	 * Tests whether the visible document's master document
 	 * is identical to this viewer's document.

tests/org.eclipse.jface.text.tests/src/org/eclipse/jface/text/tests/ProjectionViewerTest.java

@@ -11,6 +11,7 @@
 package org.eclipse.jface.text.tests;
 
 import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertThrows;
 
 import org.junit.jupiter.api.Test;
 
@@ -19,6 +20,7 @@
 import org.eclipse.swt.dnd.TextTransfer;
 import org.eclipse.swt.layout.FillLayout;
 import org.eclipse.swt.widgets.Composite;
+import org.eclipse.swt.widgets.Display;
 import org.eclipse.swt.widgets.Shell;
 
 import org.eclipse.jface.text.BadLocationException;
@@ -368,4 +370,88 @@ public void testRemoveEntireVisibleRegion() throws BadLocationException {
 			shell.dispose();
 		}
 	}
+
+	@Test
+	public void testProjectionRegionsShownOnlyInVisibleRegion() {
+		Shell shell= new Shell(Display.getCurrent());
+		shell.setLayout(new FillLayout());
+		TestProjectionViewer viewer= new TestProjectionViewer(shell, null, null, true, SWT.ALL);
+		String documentContent= """
+
+				visible_region_start
+
+				projection_start
+
+				visible_region_end
+
+				projection_end
+
+				""";
+		Document document= new Document(documentContent);
+		viewer.setDocument(document, new AnnotationModel());
+		ProjectionAnnotation annotation= addVisibleRegionAndProjection(viewer, documentContent);
+		try {
+			assertEquals("""
+					visible_region_start
+
+					projection_start
+
+					visible_region_end
+					""", viewer.getVisibleDocument().get());
+
+			annotation.paint(null, null, null); //should exit early and not throw NPE
+		} finally {
+			shell.dispose();
+		}
+	}
+
+	@Test
+	public void testProjectionRegionsShownWithinVisibleRegion() {
+		Shell shell= new Shell(Display.getCurrent());
+		shell.setLayout(new FillLayout());
+		TestProjectionViewer viewer= new TestProjectionViewer(shell, null, null, true, SWT.ALL);
+		String documentContent= """
+
+				visible_region_start
+
+				projection_start
+
+				projection_end
+
+				visible_region_end
+
+				""";
+		Document document= new Document(documentContent);
+		viewer.setDocument(document, new AnnotationModel());
+		ProjectionAnnotation annotation= addVisibleRegionAndProjection(viewer, documentContent);
+		try {
+			assertEquals("""
+					visible_region_start
+
+					projection_start
+
+					projection_end
+
+					visible_region_end
+					""", viewer.getVisibleDocument().get());
+
+			assertThrows(NullPointerException.class, () -> annotation.paint(null, null, null), "expected to run painting logic");
+		} finally {
+			shell.dispose();
+		}
+	}
+
+	private ProjectionAnnotation addVisibleRegionAndProjection(TestProjectionViewer viewer, String documentContent) {
+		int visibleRegionStart= documentContent.indexOf("visible_region_start");
+		int visibleRegionEnd= documentContent.indexOf("\n", documentContent.indexOf("visible_region_end")) + 1;
+
+		int projectionStart= documentContent.indexOf("projection_start");
+		int projectionEnd= documentContent.indexOf("\n", documentContent.indexOf("projection_end")) + 1;
+
+		viewer.setVisibleRegion(visibleRegionStart, visibleRegionEnd - visibleRegionStart);
+		viewer.enableProjection();
+		ProjectionAnnotation annotation= new ProjectionAnnotation();
+		viewer.getProjectionAnnotationModel().addAnnotation(annotation, new Position(projectionStart, projectionEnd - projectionStart));
+		return annotation;
+	}
 }