--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/WebServiceToolWorkflow/lib/SAWADLParser/src/edu/uga/cs/lsdis/meteors/wadls/GrammarsImpl.java	Tue Jun 07 18:00:50 2011 -0400
@@ -0,0 +1,272 @@
+/*
+ * (c) Copyright IBM Corp 2001, 2005 
+ */
+
+package edu.uga.cs.lsdis.meteors.wadls;
+
+import java.net.URISyntaxException;
+import java.util.ArrayList;
+import java.util.Iterator;
+import java.util.List;
+import java.util.Map;
+import java.util.Vector;
+
+import javax.wadls.Application;
+import javax.wadls.ModelReference;
+import javax.wadls.Grammars;
+import javax.wadls.WADLSException;
+import javax.wadls.extensions.schema.Schema;
+import javax.wadls.extensions.ExtensibilityElement;
+import javax.xml.namespace.QName;
+
+import org.w3c.dom.Attr;
+import org.w3c.dom.Element;
+
+import edu.uga.cs.lsdis.meteors.wadls.util.SchemaUtils;
+
+/**
+ * This class represents the <types> section of a WSDL document.
+ *
+ * @author Zixin Wu (wuzixin@uga.edu)
+ * @author Matthew J. Duftler (duftler@us.ibm.com)
+ */
+public class GrammarsImpl implements Grammars
+{
+	protected Application app = null;		//WADLS Definition
+	protected List<ModelReference> modelReferences = null;
+	protected Element docEl = null;
+	protected List extElements = new Vector();
+	protected Map allSchemas = null;
+	
+	
+	public static final long serialVersionUID = 1;
+	
+	public GrammarsImpl(Application app){
+		this.app = app;
+	}
+	
+	public List getTopLevelSchemas(){
+		return SchemaUtils.getSchemas(this.extElements);		
+	}
+	
+	/**
+	 * Get the DOM elements of all the schemas in this Types.
+	 * @return A list of DOM elements of all the schemas in this Types.
+	 */
+	public Map getSchemas(){
+		return this.allSchemas;
+	}
+	
+	public void setSchemas(Map allSchemas){
+		this.allSchemas = allSchemas;
+	}
+	
+	
+	/**
+	 * Get the DOM element of the first schema in this Types.
+	 * @return The DOM element of the first schema in this Types.
+	 */
+	public Schema getFirstSchema(){
+		return SchemaUtils.getFirstSchema(this.extElements);
+	}
+	public ModelReference getModelReference(){
+		if(modelReferences == null)
+			return null;
+		return modelReferences.get(0);
+	}
+	
+	public List<ModelReference> getModelReferences(Element startElement, String path, Application app) throws WADLSException, URISyntaxException {
+		Element el = getXSDEle(startElement, path);
+		if (el == null)
+			return null;
+		String attrModelReference = el.getAttributeNS(Constants.NS_URI_WADLS, Constants.ATTR_MODELREF);
+		if (attrModelReference.equals(""))
+			return null;
+		List<ModelReference> mrefs = ModelReferenceImpl.getModelReferences(attrModelReference, app);
+		if(mrefs.size() == 0)
+			return null;
+		return mrefs;
+	}
+	public ModelReference getModelReference(Element startElement, String path, Application app) throws WADLSException, URISyntaxException{
+		List<ModelReference> mrefs = getModelReferences(startElement, path, app);
+		return mrefs.get(0);
+	}
+	
+	
+	private Element getXSDEle(Element startElement, String path) throws WADLSException{
+		if (path == null || path == "")
+			return startElement;
+		String eleName = startElement.getLocalName();
+		if (eleName.equals("element")){					//search the target from <element ...
+			return SchemaUtils.findXSDEleOnEle(startElement, path, this);
+		}
+		else if (eleName.equals("complexType")){			//search the target from <complexType ...
+			return SchemaUtils.findXSDEleOnComplexType(startElement, path, this);
+		}
+		else{												//error
+			WADLSException wsdlsExc = new WADLSException(WADLSException.PATH_ERROR, 
+			"simpleType cannot has path");
+			throw wsdlsExc;
+		}
+	}
+	
+
+	
+	/**
+	 * Get all the extensibility elements defined here.
+	 */
+
+	/**
+	 * Set the modelReference.
+	 *
+	 * @param modelReference The desired modelReference.
+	 */
+	public void addModelReference(ModelReference modelReference){
+		if(modelReferences == null)
+			modelReferences = new ArrayList<ModelReference>();
+		modelReferences.add(0, modelReference);
+	}
+	
+	public List<ModelReference> getModelReferences() {
+		return modelReferences;
+	}
+	
+	public void setModelReferences(List<ModelReference> refs) {
+		modelReferences = refs;		
+	}
+	
+	/**
+	 * Set the documentation element for this document. This dependency
+	 * on org.w3c.dom.Element should eventually be removed when a more
+	 * appropriate way of representing this information is employed.
+	 *
+	 * @param docEl the documentation element
+	 */
+	public void setDocumentationElement(Element docEl)
+	{
+		this.docEl = docEl;
+	}
+	
+	public void setModelReferences(Element startElement, String path, List<ModelReference> refs) throws WADLSException {
+		for(ModelReference ref : refs) {
+			addModelReference(startElement, path, ref);
+		}
+	}
+	
+	public void addModelReference(Element startElement, String path, ModelReference modelReference) throws WADLSException{
+		Element el = getXSDEle(startElement, path);
+		
+		if (modelReference != null){
+			String strModelReference = modelReference.value();
+			Attr attr = el.getAttributeNodeNS(
+					Constants.NS_URI_WADLS, Constants.ATTR_MODELREF);
+			if(attr == null) {
+				attr = el.getOwnerDocument().createAttributeNS(Constants.NS_URI_WADLS, Constants.ATTR_MODELREF);
+				attr.setPrefix(Constants.PREFIX_WSDLS);
+				el.setAttributeNodeNS(attr);
+			}
+			String value = attr.getValue();
+			if(value != null) {
+				value += " " + strModelReference;
+			} else {
+				value = strModelReference;
+			}
+			attr.setValue(value);
+			modelReference.setParent(el);
+		}
+		else
+			el.removeAttributeNS(Constants.NS_URI_WADLS, Constants.ATTR_MODELREF);
+	}
+	
+	
+	/**
+	 * Get the documentation element. This dependency on org.w3c.dom.Element
+	 * should eventually be removed when a more appropriate way of
+	 * representing this information is employed.
+	 *
+	 * @return the documentation element
+	 */
+	public Element getDocumentationElement()
+	{
+		return docEl;
+	}
+	
+	public void setLoweringSchemaMapping(Element startElement, String path, String schemaMapping) throws WADLSException{
+		Element el = getXSDEle(startElement, path);
+		if (schemaMapping != null){
+			Attr attr = el.getOwnerDocument().createAttributeNS(Constants.NS_URI_WADLS, Constants.ATTR_LOWERINGSCHEMAMAPPING);
+			attr.setPrefix(Constants.PREFIX_WSDLS);
+			attr.setValue(schemaMapping);
+			el.setAttributeNodeNS(attr);
+		}
+		else
+			el.removeAttributeNS(Constants.NS_URI_WADLS, Constants.ATTR_LOWERINGSCHEMAMAPPING);
+	}
+	
+	public void setLiftingSchemaMapping(Element startElement, String path, String schemaMapping) throws WADLSException{
+		Element el = getXSDEle(startElement, path);
+		if (schemaMapping != null){
+			Attr attr = el.getOwnerDocument().createAttributeNS(Constants.NS_URI_WADLS, Constants.ATTR_LIFTINGSCHEMAMAPPING);
+			attr.setPrefix(Constants.PREFIX_WSDLS);
+			attr.setValue(schemaMapping);
+			el.setAttributeNodeNS(attr);
+		}
+		else
+			el.removeAttributeNS(Constants.NS_URI_WADLS, Constants.ATTR_LIFTINGSCHEMAMAPPING);
+	}
+	public String getLoweringSchemaMapping(Element startElement, String path) throws WADLSException{
+		Element el = getXSDEle(startElement, path);
+		if (el == null)
+			return null;
+		String attrSchemaMapping = el.getAttributeNS(Constants.NS_URI_WADLS, Constants.ATTR_LOWERINGSCHEMAMAPPING);
+		if (attrSchemaMapping == "")
+			return null;
+		return attrSchemaMapping;
+		
+	}
+	
+	public String getLiftingSchemaMapping(Element startElement, String path) throws WADLSException{
+		Element el = getXSDEle(startElement, path);
+		if (el == null)
+			return null;
+		String attrSchemaMapping = el.getAttributeNS(Constants.NS_URI_WADLS, Constants.ATTR_LIFTINGSCHEMAMAPPING);
+		if (attrSchemaMapping == "")
+			return null;
+		return attrSchemaMapping;
+		
+	}
+	
+
+	/**
+	 * Add an extensibility element.
+	 *
+	 * @param extElement the extensibility element to be added
+	 */
+	
+	
+	/**
+	 * Get all the extensibility elements defined here.
+	 */
+
+	
+	public String toString()
+	{
+		StringBuffer strBuf = new StringBuffer();
+		
+		strBuf.append("Types:");
+		
+		if (extElements != null)
+		{
+			Iterator extIterator = extElements.iterator();
+			
+			while (extIterator.hasNext())
+			{
+				strBuf.append("\n" + extIterator.next());
+			}
+		}
+		
+		return strBuf.toString();
+	}
+	
+	
+}