1 files changed, 19 insertions, 0 deletions
diff --git a/jaxb/plugins/org.eclipse.jpt.jaxb.core/src/org/eclipse/jpt/jaxb/core/internal/context/GenericContextRoot.java b/jaxb/plugins/org.eclipse.jpt.jaxb.core/src/org/eclipse/jpt/jaxb/core/internal/context/GenericContextRoot.java
index 7e5e78f567..fd37fec96d 100644
--- a/jaxb/plugins/org.eclipse.jpt.jaxb.core/src/org/eclipse/jpt/jaxb/core/internal/context/GenericContextRoot.java
+++ b/jaxb/plugins/org.eclipse.jpt.jaxb.core/src/org/eclipse/jpt/jaxb/core/internal/context/GenericContextRoot.java
@@ -26,6 +26,7 @@ import org.eclipse.jpt.jaxb.core.resource.java.JAXB;
 import org.eclipse.jpt.jaxb.core.resource.java.JavaResourcePackage;
 import org.eclipse.jpt.jaxb.core.resource.java.JavaResourceType;
 import org.eclipse.jpt.utility.internal.CollectionTools;
+import org.eclipse.jpt.utility.internal.StringTools;
 import org.eclipse.jpt.utility.internal.iterables.FilteringIterable;
 import org.eclipse.jpt.utility.internal.iterables.LiveCloneIterable;
 import org.eclipse.jpt.utility.internal.iterables.SubIterableWrapper;
@@ -268,6 +269,15 @@ public class GenericContextRoot
 	public int getPackagesSize() {
 		return this.packages.size();
 	}
+
+	public JaxbPackage getPackage(String packageName) {
+		for (JaxbPackage jaxbPackage : this.getPackages()) {
+			if (StringTools.stringsAreEqual(jaxbPackage.getName(), packageName)) {
+				return jaxbPackage;
+			}
+		}
+		return null;
+	}
 	
 	protected JaxbPackage addPackage(JaxbPackage contextPackage) {
 		if (this.packages.containsKey(contextPackage.getName())) {
@@ -390,6 +400,15 @@ public class GenericContextRoot
 		};
 	}
 
+	public JaxbPersistentClass getPersistentClass(String className) {
+		for (JaxbPersistentClass jaxbClass : this.getPersistentClasses()) {
+			if (StringTools.stringsAreEqual(jaxbClass.getFullyQualifiedName(), className)) {
+				return jaxbClass;
+			}
+		}
+		return null;
+	}
+
 	@Override
     public void stateChanged() {
 		super.stateChanged();