void (@NotNull PsiStatement switchStatement) { CommentTracker commentTracker = new CommentTracker(); PsiSwitchBlock switchBlock = generateEnhancedSwitch(switchStatement, myExpressionBranches, commentTracker, false); if (switchBlock == null) return; commentTracker.replaceAndRestoreComments(switchStatement, switchBlock); }