--- /dev/null
+/*******************************************************************************
+ * Copyright (c) 2000, 2011 IBM Corporation and others.
+ * All rights reserved. This program and the accompanying materials
+ * are made available under the terms of the Eclipse Public License v1.0
+ * which accompanies this distribution, and is available at
+ * http://www.eclipse.org/legal/epl-v10.html
+ *
+ * Contributors:
+ * IBM Corporation - initial API and implementation
+ *******************************************************************************/
+package org.eclipse.jdt.internal.corext.dom;
+
+import java.util.Arrays;
+import java.util.HashMap;
+import java.util.Iterator;
+import java.util.List;
+import java.util.Map;
+
+import org.eclipse.core.runtime.Assert;
+
+import org.eclipse.text.edits.TextEditGroup;
+
+import org.eclipse.jdt.core.dom.AST;
+import org.eclipse.jdt.core.dom.ASTNode;
+import org.eclipse.jdt.core.dom.AbstractTypeDeclaration;
+import org.eclipse.jdt.core.dom.AnonymousClassDeclaration;
+import org.eclipse.jdt.core.dom.Block;
+import org.eclipse.jdt.core.dom.FieldDeclaration;
+import org.eclipse.jdt.core.dom.IExtendedModifier;
+import org.eclipse.jdt.core.dom.Modifier;
+import org.eclipse.jdt.core.dom.SingleVariableDeclaration;
+import org.eclipse.jdt.core.dom.SwitchStatement;
+import org.eclipse.jdt.core.dom.Type;
+import org.eclipse.jdt.core.dom.VariableDeclarationExpression;
+import org.eclipse.jdt.core.dom.VariableDeclarationFragment;
+import org.eclipse.jdt.core.dom.VariableDeclarationStatement;
+import org.eclipse.jdt.core.dom.rewrite.ASTRewrite;
+import org.eclipse.jdt.core.dom.rewrite.ListRewrite;
+
+
+public class VariableDeclarationRewrite {
+
+ public static void rewriteModifiers(final SingleVariableDeclaration declarationNode, final int includedModifiers, final int excludedModifiers, final ASTRewrite rewrite, final TextEditGroup group) {
+ ModifierRewrite listRewrite= ModifierRewrite.create(rewrite, declarationNode);
+ listRewrite.setModifiers(includedModifiers, excludedModifiers, group);
+ }
+
+ public static void rewriteModifiers(final VariableDeclarationExpression declarationNode, final int includedModifiers, final int excludedModifiers, final ASTRewrite rewrite, final TextEditGroup group) {
+ ModifierRewrite listRewrite= ModifierRewrite.create(rewrite, declarationNode);
+ listRewrite.setModifiers(includedModifiers, excludedModifiers, group);
+ }
+
+ public static void rewriteModifiers(final FieldDeclaration declarationNode, final VariableDeclarationFragment[] toChange, final int includedModifiers, final int excludedModifiers, final ASTRewrite rewrite, final TextEditGroup group) {
+ final List<VariableDeclarationFragment> fragmentsToChange= Arrays.asList(toChange);
+ AST ast= declarationNode.getAST();
+/*
+ * Problem: Same declarationNode can be the subject of multiple calls to this method.
+ * For the 2nd++ calls, the original declarationNode has already been rewritten, and this has to be taken into account.
+ *
+ * Assumption:
+ * - Modifiers for each VariableDeclarationFragment are modified at most once.
+ *
+ * Solution:
+ * - Maintain a map from original VariableDeclarationFragments to their new FieldDeclaration.
+ * - Original modifiers in declarationNode belong to the first fragment.
+ * - When a later fragment needs different modifiers, we create a new FieldDeclaration and move all successive fragments into that declaration
+ * - When a fragment has been moved to a new declaration, make sure we don't create a new move target again, but instead use the already created one
+ */
+ List<VariableDeclarationFragment> fragments= declarationNode.fragments();
+ Iterator<VariableDeclarationFragment> iter= fragments.iterator();
+
+ ListRewrite blockRewrite;
+ if (declarationNode.getParent() instanceof AbstractTypeDeclaration) {
+ blockRewrite= rewrite.getListRewrite(declarationNode.getParent(), ((AbstractTypeDeclaration)declarationNode.getParent()).getBodyDeclarationsProperty());
+ } else {
+ blockRewrite= rewrite.getListRewrite(declarationNode.getParent(), AnonymousClassDeclaration.BODY_DECLARATIONS_PROPERTY);
+ }
+
+ VariableDeclarationFragment lastFragment= iter.next();
+ ASTNode lastStatement= declarationNode;
+
+ if (fragmentsToChange.contains(lastFragment)) {
+ ModifierRewrite modifierRewrite= ModifierRewrite.create(rewrite, declarationNode);
+ modifierRewrite.setModifiers(includedModifiers, excludedModifiers, group);
+ }
+
+ ListRewrite fragmentsRewrite= null;
+ while (iter.hasNext()) {
+ VariableDeclarationFragment currentFragment= iter.next();
+
+ Map<VariableDeclarationFragment, MovedFragment> lookup= (Map<VariableDeclarationFragment, MovedFragment>) rewrite.getProperty(MovedFragment.class.getName());
+ if (lookup == null) {
+ lookup= new HashMap<VariableDeclarationFragment, MovedFragment>();
+ rewrite.setProperty(MovedFragment.class.getName(), lookup);
+ }
+ MovedFragment currentMovedFragment= lookup.get(currentFragment);
+
+ boolean changeLast= fragmentsToChange.contains(lastFragment);
+ boolean changeCurrent= fragmentsToChange.contains(currentFragment);
+ if (changeLast != changeCurrent || lookup.containsKey(lastFragment)) {
+ ModifierRewrite modifierRewrite= null;
+ if (currentMovedFragment != null) {
+ // Current fragment has already been moved.
+
+ if (currentMovedFragment.fUsesOriginalModifiers) {
+ // Need to put in the right modifiers (removing any existing ones).
+ modifierRewrite= ModifierRewrite.create(rewrite, currentMovedFragment.fDeclaration);
+ ListRewrite listRewrite= rewrite.getListRewrite(currentMovedFragment.fDeclaration, FieldDeclaration.MODIFIERS2_PROPERTY);
+ List<IExtendedModifier> extendedList= listRewrite.getRewrittenList();
+ for (int i= 0; i < extendedList.size(); i++) {
+ ASTNode curr= (ASTNode)extendedList.get(i);
+ if (curr instanceof Modifier)
+ rewrite.remove(curr, group);
+ }
+ }
+ // otherwise, don't need to touch the modifiers, so leave modifierRewrite null
+
+ } else { // need to split an existing field declaration
+ VariableDeclarationFragment moveTarget;
+ moveTarget= (VariableDeclarationFragment)rewrite.createMoveTarget(currentFragment);
+
+ FieldDeclaration newStatement= (FieldDeclaration)ast.createInstance(FieldDeclaration.class);
+ rewrite.getListRewrite(newStatement, FieldDeclaration.FRAGMENTS_PROPERTY).insertLast(moveTarget, group);
+ lookup.put(currentFragment, new MovedFragment(moveTarget, newStatement, !changeCurrent));
+ rewrite.set(newStatement, FieldDeclaration.TYPE_PROPERTY, rewrite.createCopyTarget(declarationNode.getType()), group);
+
+ modifierRewrite= ModifierRewrite.create(rewrite, newStatement);
+ modifierRewrite.copyAllAnnotations(declarationNode, group);
+ blockRewrite.insertAfter(newStatement, lastStatement, group);
+
+ fragmentsRewrite= rewrite.getListRewrite(newStatement, FieldDeclaration.FRAGMENTS_PROPERTY);
+ lastStatement= newStatement;
+ }
+
+ if (modifierRewrite != null) {
+ if (changeCurrent) {
+ int newModifiers= (declarationNode.getModifiers() & ~excludedModifiers) | includedModifiers;
+ modifierRewrite.setModifiers(newModifiers, excludedModifiers, group);
+ } else {
+ int newModifiers= declarationNode.getModifiers();
+ modifierRewrite.setModifiers(newModifiers, Modifier.NONE, group);
+ }
+ }
+
+ } else if (fragmentsRewrite != null) {
+ VariableDeclarationFragment fragment0;
+ boolean usesOriginalModifiers= true;
+ if (currentMovedFragment != null) {
+ fragment0= currentMovedFragment.fMoveTarget;
+ usesOriginalModifiers= currentMovedFragment.fUsesOriginalModifiers;
+ rewrite.getListRewrite(currentMovedFragment.fDeclaration, FieldDeclaration.FRAGMENTS_PROPERTY).remove(fragment0, group);
+ } else {
+ fragment0= (VariableDeclarationFragment)rewrite.createMoveTarget(currentFragment);
+ }
+ lookup.put(currentFragment, new MovedFragment(fragment0, lastStatement, usesOriginalModifiers));
+ fragmentsRewrite.insertLast(fragment0, group);
+ }
+ lastFragment= currentFragment;
+ }
+ }
+
+ private static class MovedFragment {
+ final VariableDeclarationFragment fMoveTarget;
+ final ASTNode fDeclaration;
+ boolean fUsesOriginalModifiers;
+
+ public MovedFragment(VariableDeclarationFragment moveTarget, ASTNode declaration, boolean usesOriginalModifiers) {
+ fMoveTarget= moveTarget;
+ fDeclaration= declaration;
+ fUsesOriginalModifiers= usesOriginalModifiers;
+ }
+ }
+
+ public static void rewriteModifiers(final VariableDeclarationStatement declarationNode, final VariableDeclarationFragment[] toChange, final int includedModifiers, final int excludedModifiers, ASTRewrite rewrite, final TextEditGroup group) {
+ final List<VariableDeclarationFragment> fragmentsToChange= Arrays.asList(toChange);
+ AST ast= declarationNode.getAST();
+
+ List<VariableDeclarationFragment> fragments= declarationNode.fragments();
+ Iterator<VariableDeclarationFragment> iter= fragments.iterator();
+
+ ListRewrite blockRewrite= null;
+ ASTNode parentStatement= declarationNode.getParent();
+ if (parentStatement instanceof SwitchStatement) {
+ blockRewrite= rewrite.getListRewrite(parentStatement, SwitchStatement.STATEMENTS_PROPERTY);
+ } else if (parentStatement instanceof Block) {
+ blockRewrite= rewrite.getListRewrite(parentStatement, Block.STATEMENTS_PROPERTY);
+ } else {
+ // should not happen. VariableDeclaration's can not be in a control statement body
+ Assert.isTrue(false);
+ }
+
+ VariableDeclarationFragment lastFragment= iter.next();
+ ASTNode lastStatement= declarationNode;
+
+ boolean modifiersModified= false;
+ if (fragmentsToChange.contains(lastFragment)) {
+ ModifierRewrite modifierRewrite= ModifierRewrite.create(rewrite, declarationNode);
+ modifierRewrite.setModifiers(includedModifiers, excludedModifiers, group);
+ modifiersModified= true;
+ }
+
+ ListRewrite fragmentsRewrite= null;
+ while (iter.hasNext()) {
+ VariableDeclarationFragment currentFragment= iter.next();
+
+ if (fragmentsToChange.contains(lastFragment) != fragmentsToChange.contains(currentFragment)) {
+
+ VariableDeclarationStatement newStatement= ast.newVariableDeclarationStatement((VariableDeclarationFragment)rewrite.createMoveTarget(currentFragment));
+ newStatement.setType((Type)rewrite.createCopyTarget(declarationNode.getType()));
+
+ ModifierRewrite modifierRewrite= ModifierRewrite.create(rewrite, newStatement);
+ if (fragmentsToChange.contains(currentFragment)) {
+ modifierRewrite.copyAllAnnotations(declarationNode, group);
+ int newModifiers= (declarationNode.getModifiers() & ~excludedModifiers) | includedModifiers;
+ modifierRewrite.setModifiers(newModifiers, excludedModifiers, group);
+ } else {
+ modifierRewrite.copyAllModifiers(declarationNode, group, modifiersModified);
+ }
+ blockRewrite.insertAfter(newStatement, lastStatement, group);
+
+ fragmentsRewrite= rewrite.getListRewrite(newStatement, VariableDeclarationStatement.FRAGMENTS_PROPERTY);
+ lastStatement= newStatement;
+ } else if (fragmentsRewrite != null) {
+ ASTNode fragment0= rewrite.createMoveTarget(currentFragment);
+ fragmentsRewrite.insertLast(fragment0, group);
+ }
+ lastFragment= currentFragment;
+ }
+ }
+}