adds support for incremental model (INRIA#584)

src/main/java/spoon/reflect/CtModel.java

@@ -1,3 +1,19 @@
+/**
+ * Copyright (C) 2006-2015 INRIA and contributors
+ * Spoon - http://spoon.gforge.inria.fr/
+ *
+ * This software is governed by the CeCILL-C License under French law and
+ * abiding by the rules of distribution of free software. You can use, modify
+ * and/or redistribute the software under the terms of the CeCILL-C license as
+ * circulated by CEA, CNRS and INRIA at http://www.cecill.info.
+ *
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE. See the CeCILL-C License for more details.
+ *
+ * The fact that you are presently reading this means that you have had
+ * knowledge of the CeCILL-C license and that you accept its terms.
+ */
 package spoon.reflect;
 
 import java.util.Collection;
@@ -21,6 +37,6 @@ public interface CtModel {
 	Collection<CtPackage> getAllPackages();
 
 	/** process this model with the given processor */
-	void processWith(Processor<?> abstractProcessor);
+	void processWith(Processor<?> processor);
 
 }

src/main/java/spoon/reflect/CtModelImpl.java

@@ -1,3 +1,19 @@
+/**
+ * Copyright (C) 2006-2015 INRIA and contributors
+ * Spoon - http://spoon.gforge.inria.fr/
+ *
+ * This software is governed by the CeCILL-C License under French law and
+ * abiding by the rules of distribution of free software. You can use, modify
+ * and/or redistribute the software under the terms of the CeCILL-C license as
+ * circulated by CEA, CNRS and INRIA at http://www.cecill.info.
+ *
+ * This program is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE. See the CeCILL-C License for more details.
+ *
+ * The fact that you are presently reading this means that you have had
+ * knowledge of the CeCILL-C license and that you accept its terms.
+ */
 package spoon.reflect;
 
 import java.util.Collection;
@@ -78,12 +94,12 @@ public Collection<CtType<?>> getAllTypes() {
 
 	@Override
 	public Collection<CtPackage> getAllPackages() {
-		return Collections.unmodifiableCollection(rootPackage.getElements(new TypeFilter<>(CtPackage.class)));
+		return Collections.unmodifiableCollection(rootPackage.getElements(new TypeFilter<CtPackage>(CtPackage.class)));
 	}
 
 
 	@Override
-	public void processWith(Processor<?> abstractProcessor) {
+	public void processWith(Processor<?> processor) {
 		// processing (consume all the processors)
 		ProcessingManager processing = new QueueProcessingManager(rootPackage.getFactory());
 		processing.process(getRootPackage());

src/main/java/spoon/support/reflect/declaration/CtPackageImpl.java

@@ -16,15 +16,15 @@
  */
 package spoon.support.reflect.declaration;
 
+import java.util.Set;
+import java.util.TreeSet;
+
 import spoon.reflect.cu.SourcePosition;
 import spoon.reflect.declaration.CtPackage;
 import spoon.reflect.declaration.CtType;
 import spoon.reflect.reference.CtPackageReference;
 import spoon.reflect.visitor.CtVisitor;
 
-import java.util.Set;
-import java.util.TreeSet;
-
 /**
  * The implementation for {@link spoon.reflect.declaration.CtPackage}.
  *
@@ -33,7 +33,7 @@
 public class CtPackageImpl extends CtNamedElementImpl implements CtPackage {
 	private static final long serialVersionUID = 1L;
 
-	private Set<CtPackage> packs = new TreeSet<CtPackage>();
+	protected Set<CtPackage> packs = new TreeSet<CtPackage>();
 
 	private Set<CtType<?>> types = new TreeSet<CtType<?>>();
 
@@ -48,11 +48,41 @@ public void accept(CtVisitor v) {
 
 	@Override
 	public <T extends CtPackage> T addPackage(CtPackage pack) {
-		pack.setParent(this);
-		packs.add(pack);
+		// they are the same
+		if (this.getQualifiedName().equals(pack.getQualifiedName())) {
+			addAllTypes(pack, this);
+			addAllPackages(pack, this);
+			return (T) this;
+		}
+
+		// it already exists
+		for (CtPackage p1 : packs) {
+			if (p1.getQualifiedName().equals(pack.getQualifiedName())) {
+				addAllTypes(pack, p1);
+				addAllPackages(pack, p1);
+				return (T) this;
+			}
+		}
+
+		this.packs.add(pack);
+
 		return (T) this;
 	}
 
+	/** add all types of "from" in "to" */
+	private void addAllTypes(CtPackage from, CtPackage to) {
+		for (CtType t : from.getTypes()) {
+			to.addType(t);
+		}
+	}
+
+	/** add all packages of "from" in "to" */
+	private void addAllPackages(CtPackage from, CtPackage to) {
+		for (CtPackage p : from.getPackages()) {
+			to.addPackage(p);
+		}
+	}
+
 	@Override
 	public boolean removePackage(CtPackage pack) {
 		return packs.remove(pack);

src/test/java/spoon/test/factory/FactoryTest.java

@@ -105,4 +105,39 @@ public void testCtModel() throws Exception {
 		// [, spoon, spoon.test, spoon.test.factory, spoon.test.factory.testclasses]
 		assertEquals(5, model.getAllPackages().size());
 	}
+
+	@Test
+	public void testIncrementalModel() throws Exception {
+
+		// Feed some inputResources to a spoon compiler
+		SpoonAPI spoon = new Launcher();
+		spoon.addInputResource("src/test/java/spoon/test/factory/testclasses");
+
+		// Build model
+		spoon.buildModel();
+
+		// Do something with that model..
+		CtModel model = spoon.getModel();
+		model.processWith(new AbstractProcessor<CtMethod>() {
+			@Override
+			public void process(CtMethod element) {
+				element.setDefaultMethod(false);
+			}			
+		});
+
+		// Feed some new inputResources 
+		SpoonAPI spoon2 = new Launcher();
+		spoon2.addInputResource("src/test/java/spoon/test/factory/testclasses2");
+
+		// Build models of newly added classes/packages 
+		spoon2.buildModel();
+
+		// attach them to the existing model.
+		model.getRootPackage().addPackage(spoon2.getModel().getRootPackage());
+
+		// checking the results
+		assertEquals(6, model.getAllPackages().size());		
+		assertEquals(3, model.getAllTypes().size());
+	}
+
 }

src/test/java/spoon/test/factory/testclasses2/Baz.java

@@ -0,0 +1,4 @@
+package spoon.test.factory.testclasses2;
+
+public class Baz {
+}