1 files changed, 146 insertions, 0 deletions
diff --git a/core/org.eclipse.cdt.core/parser/org/eclipse/cdt/internal/core/dom/rewrite/changegenerator/ASTModificationHelper.java b/core/org.eclipse.cdt.core/parser/org/eclipse/cdt/internal/core/dom/rewrite/changegenerator/ASTModificationHelper.java
new file mode 100644
index 00000000000..9ffbc001c6f
--- /dev/null
+++ b/core/org.eclipse.cdt.core/parser/org/eclipse/cdt/internal/core/dom/rewrite/changegenerator/ASTModificationHelper.java
@@ -0,0 +1,146 @@
+/*******************************************************************************
+ * Copyright (c) 2008 Institute for Software, HSR Hochschule fuer Technik  
+ * Rapperswil, University of applied sciences 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: 
+ * Institute for Software - initial API and implementation
+ *******************************************************************************/
+package org.eclipse.cdt.internal.core.dom.rewrite.changegenerator;
+
+import java.lang.reflect.Array;
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
+
+import org.eclipse.cdt.core.dom.ast.IASTDeclarator;
+import org.eclipse.cdt.core.dom.ast.IASTInitializer;
+import org.eclipse.cdt.core.dom.ast.IASTNode;
+import org.eclipse.cdt.internal.core.dom.rewrite.ASTModification;
+import org.eclipse.cdt.internal.core.dom.rewrite.ASTModificationMap;
+import org.eclipse.cdt.internal.core.dom.rewrite.ASTModificationStore;
+import org.eclipse.cdt.internal.core.dom.rewrite.ASTModification.ModificationKind;
+import org.eclipse.cdt.internal.core.dom.rewrite.astwriter.ContainerNode;
+
+public class ASTModificationHelper {
+
+	private final ASTModificationStore modificationStore;
+
+	public ASTModificationHelper(ASTModificationStore modificationStore) {
+		this.modificationStore = modificationStore;
+	}
+
+	
+	@SuppressWarnings("unchecked")
+	public <T extends IASTNode> T[] createModifiedChildArray(IASTNode parent, T[] unmodifiedChildren){
+		ArrayList<T> modifiedChildren = new ArrayList<T>(Arrays.asList(unmodifiedChildren));
+		for(T currentChild : unmodifiedChildren){
+			for(ASTModification childModification : modificationsForNode(currentChild)){
+				try{
+					T newNode = (T) childModification.getNewNode();
+					switch(childModification.getKind()){
+					case REPLACE:
+						if(childModification.getNewNode() != null){
+							modifiedChildren.add(modifiedChildren.indexOf(childModification.getTargetNode()), newNode);
+						}
+						modifiedChildren.remove(childModification.getTargetNode());
+						break;
+					case INSERT_BEFORE:
+						modifiedChildren.add(modifiedChildren.indexOf(childModification.getTargetNode()), newNode);
+						break;
+					case APPEND_CHILD:
+						throw new UnhandledASTModificationException(childModification);
+	
+					}
+				}catch(ClassCastException e){
+					throw new UnhandledASTModificationException(childModification);
+				}
+			} 
+		}
+
+		Class<?> componentType = unmodifiedChildren.getClass().getComponentType();
+		for(ASTModification parentModification : modificationsForNode(parent)){
+			if(parentModification.getKind() == ModificationKind.APPEND_CHILD){
+				IASTNode newNode = parentModification.getNewNode();
+				if(componentType.isAssignableFrom(newNode.getClass())){
+					modifiedChildren.add((T) newNode);
+				}
+				else if(newNode instanceof ContainerNode){
+					ContainerNode nodeContainer = (ContainerNode) newNode;
+					for(IASTNode currentNode : nodeContainer.getNodes()){
+						if(componentType.isAssignableFrom(currentNode.getClass())){
+							modifiedChildren.add((T)currentNode);
+						}
+					}
+				}
+			}
+		}
+		
+		return modifiedChildren.toArray((T[]) Array.newInstance(componentType, 0));
+	}
+
+	public List<ASTModification> modificationsForNode(
+			IASTNode targetNode) {
+		ASTModificationMap rootModifications = modificationStore.getRootModifications();
+		if(rootModifications == null){
+			rootModifications = new ASTModificationMap();
+		}
+		List<ASTModification> modificationsForNode = rootModifications.getModificationsForNode(targetNode);
+		return modificationsForNode;
+	}
+	
+	
+	public IASTInitializer getInitializer(IASTDeclarator decl) {
+		IASTInitializer initializer = decl.getInitializer();
+		
+		if(initializer != null){
+			for(ASTModification childModification : modificationsForNode(initializer)){
+				switch(childModification.getKind()){
+				case REPLACE:
+					if(childModification.getNewNode() instanceof IASTInitializer){
+						return (IASTInitializer)childModification.getNewNode();
+					}
+					throw new UnhandledASTModificationException(childModification);
+					
+				case INSERT_BEFORE:
+					throw new UnhandledASTModificationException(childModification);
+					
+				case APPEND_CHILD:
+					throw new UnhandledASTModificationException(childModification);
+				}
+			}
+		}
+		else
+		{
+			for(ASTModification parentModification : modificationsForNode(decl)){
+				if(parentModification.getKind() == ModificationKind.APPEND_CHILD){
+					IASTNode newNode = parentModification.getNewNode();
+					if(newNode instanceof IASTInitializer){
+						return (IASTInitializer) newNode;
+					}
+				}
+			}
+		}
+		return initializer;
+	}
+
+
+	@SuppressWarnings("unchecked")
+	public <T extends IASTNode> T getNodeAfterReplacement(T replacedNode) {
+		List<ASTModification> modifications = modificationsForNode(replacedNode);
+		for(ASTModification currentModification : modifications){
+			try{
+				if(currentModification.getKind() == ModificationKind.REPLACE){
+					return (T) currentModification.getNewNode();
+				}
+			}
+			catch(ClassCastException e){
+				throw new UnhandledASTModificationException(currentModification);
+			}
+		}
+		return replacedNode;
+	}
+}