From 54507d94369655b42a747b631daf434d23af6884 Mon Sep 17 00:00:00 2001
From: Samuel Teuber <teuber@kit.edu>
Date: Fri, 24 May 2024 00:49:49 +0200
Subject: [PATCH] Fixed jml commenting for loop invariants

---
 .../github/javaparser/printer/DefaultPrettyPrinterVisitor.java | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/javaparser-core/src/main/java/com/github/javaparser/printer/DefaultPrettyPrinterVisitor.java b/javaparser-core/src/main/java/com/github/javaparser/printer/DefaultPrettyPrinterVisitor.java
index d2cb8c74a2..6e64ccf10c 100644
--- a/javaparser-core/src/main/java/com/github/javaparser/printer/DefaultPrettyPrinterVisitor.java
+++ b/javaparser-core/src/main/java/com/github/javaparser/printer/DefaultPrettyPrinterVisitor.java
@@ -936,10 +936,13 @@ public void visit(JmlClauseIf n, Void arg) {
     @Override
     public void visit(JmlClassExprDeclaration n, Void arg) {
         printOrphanCommentsBeforeThisChildNode(n);
+        boolean openedJml = inJmlComment();
+        if (!openedJml) startJmlComment(false, n.getJmlTags());
         printModifiers(n.getModifiers());
         printer.print("invariant ");
         n.getInvariant().accept(this, arg);
         printer.print(";");
+        if (!openedJml) endJmlComment();
     }
 
     @Override