PsiStatement (PsiElement container, PsiElement anchorStatement) { if(container instanceof PsiLoopStatement) { return ((PsiLoopStatement) container).getBody(); } else if (container instanceof PsiIfStatement) { final PsiStatement thenBranch = ((PsiIfStatement)container).getThenBranch(); if (thenBranch != null && PsiTreeUtil.isAncestor(thenBranch, anchorStatement, false)) { return thenBranch; } final PsiStatement elseBranch = ((PsiIfStatement)container).getElseBranch(); if (elseBranch != null && PsiTreeUtil.isAncestor(elseBranch, anchorStatement, false)) { return elseBranch; } LOG.assertTrue(false); } LOG.assertTrue(false); return null; }