jaxws/src/share/classes/com/sun/xml/internal/stream/buffer/stax/StreamReaderBufferCreator.java - platform/libcore - Git at Google

 /*
  * Copyright 2005-2006 Sun Microsystems, Inc.  All Rights Reserved.
  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
  *
  * This code is free software; you can redistribute it and/or modify it
  * under the terms of the GNU General Public License version 2 only, as
  * published by the Free Software Foundation.  Sun designates this
  * particular file as subject to the "Classpath" exception as provided
  * by Sun in the LICENSE file that accompanied this code.
  *
  * This code 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 GNU General Public License
  * version 2 for more details (a copy is included in the LICENSE file that
  * accompanied this code).
  *
  * You should have received a copy of the GNU General Public License version
  * 2 along with this work; if not, write to the Free Software Foundation,
  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  *
  * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
  * CA 95054 USA or visit www.sun.com if you need additional information or
  * have any questions.
  */
 package com.sun.xml.internal.stream.buffer.stax;

 import com.sun.xml.internal.stream.buffer.MutableXMLStreamBuffer;
 import com.sun.xml.internal.org.jvnet.staxex.Base64Data;
 import com.sun.xml.internal.org.jvnet.staxex.XMLStreamReaderEx;

 import javax.xml.stream.XMLStreamException;
 import javax.xml.stream.XMLStreamReader;
 import java.util.HashMap;
 import java.util.Map;

 /**
  * Create a buffer using an {@link XMLStreamReader}.
  * <p>
  * TODO: Implement the marking the stream on the element when an ID
  * attribute on the element is defined
  */
 public class StreamReaderBufferCreator extends StreamBufferCreator {
     private int _eventType;
     private boolean _storeInScopeNamespacesOnElementFragment;
     private Map<String, Integer> _inScopePrefixes;

     /**
      * Create a stream reader buffer creator.
      * <p>
      * A stream buffer will be created for storing the infoset
      * from a stream reader.
      */
     public StreamReaderBufferCreator() {
     }

     /**
      * Create a stream reader buffer creator using a mutable stream buffer.
      * <p>
      * @param buffer the mutable stream buffer.
      */
     public StreamReaderBufferCreator(MutableXMLStreamBuffer buffer) {
         setBuffer(buffer);
     }

     /**
      * Create the buffer from a stream reader.
      * <p>
      * The stream reader must be positioned at the start of the document
      * or the start of an element.
      * <p>
      * If the stream is positioned at the start of the document then the
      * whole document is stored and after storing the stream will be positioned
      * at the end of the document.
      * <p>
      * If the stream is positioned at the start of an element then the
      * element and all its children will be stored and after storing the stream
      * will be positioned at the next event after the end of the element.
      * <p>
      * @return the mutable stream buffer.
      * @throws XMLStreamException if the stream reader is not positioned at
      *         the start of the document or at an element.
      */
     public MutableXMLStreamBuffer create(XMLStreamReader reader) throws XMLStreamException {
         if (_buffer == null) {
             createBuffer();
         }
         store(reader);

         return getXMLStreamBuffer();
     }

     /**
      * Creates the buffer from a stream reader that is an element fragment.
      * <p>
      * The stream reader will be moved to the position of the next start of
      * an element if the stream reader is not already positioned at the start
      * of an element.
      * <p>
      * The element and all its children will be stored and after storing the stream
      * will be positioned at the next event after the end of the element.
      * <p>
      * @param storeInScopeNamespaces true if in-scope namespaces of the element
      *        fragment should be stored.
      * @return the mutable stream buffer.
      * @throws XMLStreamException if the stream reader cannot be positioned at
      *         the start of an element.
      */
     public MutableXMLStreamBuffer createElementFragment(XMLStreamReader reader,
             boolean storeInScopeNamespaces) throws XMLStreamException {
         if (_buffer == null) {
             createBuffer();
         }

         if (!reader.hasNext()) {
             return _buffer;
         }

         _storeInScopeNamespacesOnElementFragment = storeInScopeNamespaces;

         _eventType = reader.getEventType();
         if (_eventType != XMLStreamReader.START_ELEMENT) {
             do {
                 _eventType = reader.next();
             } while(_eventType != XMLStreamReader.START_ELEMENT && _eventType != XMLStreamReader.END_DOCUMENT);
         }

         if (storeInScopeNamespaces) {
             _inScopePrefixes = new HashMap<String,Integer>();
         }

         storeElementAndChildren(reader);

         return getXMLStreamBuffer();
     }

     private void store(XMLStreamReader reader) throws XMLStreamException {
         if (!reader.hasNext()) {
             return;
         }

         _eventType = reader.getEventType();
         switch (_eventType) {
             case XMLStreamReader.START_DOCUMENT:
                 storeDocumentAndChildren(reader);
                 break;
             case XMLStreamReader.START_ELEMENT:
                 storeElementAndChildren(reader);
                 break;
             default:
                 throw new XMLStreamException("XMLStreamReader not positioned at a document or element");
         }

         increaseTreeCount();
     }

     private void storeDocumentAndChildren(XMLStreamReader reader) throws XMLStreamException {
         storeStructure(T_DOCUMENT);

         _eventType = reader.next();
         while (_eventType != XMLStreamReader.END_DOCUMENT) {
             switch (_eventType) {
                 case XMLStreamReader.START_ELEMENT:
                     storeElementAndChildren(reader);
                     continue;
                 case XMLStreamReader.COMMENT:
                     storeComment(reader);
                     break;
                 case XMLStreamReader.PROCESSING_INSTRUCTION:
                     storeProcessingInstruction(reader);
                     break;
             }
             _eventType = reader.next();
         }

         storeStructure(T_END);
     }

     private void storeElementAndChildren(XMLStreamReader reader) throws XMLStreamException {
         if (reader instanceof XMLStreamReaderEx) {
             storeElementAndChildrenEx((XMLStreamReaderEx)reader);
         } else {
             storeElementAndChildrenNoEx(reader);
         }
     }

     private void storeElementAndChildrenEx(XMLStreamReaderEx reader) throws XMLStreamException {
         int depth = 1;
         if (_storeInScopeNamespacesOnElementFragment) {
             storeElementWithInScopeNamespaces(reader);
         } else {
             storeElement(reader);
         }

         while(depth > 0) {
             _eventType = reader.next();
             switch (_eventType) {
                 case XMLStreamReader.START_ELEMENT:
                     depth++;
                     storeElement(reader);
                     break;
                 case XMLStreamReader.END_ELEMENT:
                     depth--;
                     storeStructure(T_END);
                     break;
                 case XMLStreamReader.NAMESPACE:
                     storeNamespaceAttributes(reader);
                     break;
                 case XMLStreamReader.ATTRIBUTE:
                     storeAttributes(reader);
                     break;
                 case XMLStreamReader.SPACE:
                 case XMLStreamReader.CHARACTERS:
                 case XMLStreamReader.CDATA: {
                     CharSequence c = reader.getPCDATA();
                     if (c instanceof Base64Data) {
                         storeStructure(T_TEXT_AS_OBJECT);
                         storeContentObject(((Base64Data)c).clone());
                     } else {
                         storeContentCharacters(T_TEXT_AS_CHAR_ARRAY,
                                 reader.getTextCharacters(), reader.getTextStart(),
                                 reader.getTextLength());
                     }
                     break;
                 }
                 case XMLStreamReader.COMMENT:
                     storeComment(reader);
                     break;
                 case XMLStreamReader.PROCESSING_INSTRUCTION:
                     storeProcessingInstruction(reader);
                     break;
             }
         }

         /*
          * Move to next item after the end of the element
          * that has been stored
          */
         _eventType = reader.next();
     }

     private void storeElementAndChildrenNoEx(XMLStreamReader reader) throws XMLStreamException {
         int depth = 1;
         if (_storeInScopeNamespacesOnElementFragment) {
             storeElementWithInScopeNamespaces(reader);
         } else {
             storeElement(reader);
         }

         while(depth > 0) {
             _eventType = reader.next();
             switch (_eventType) {
                 case XMLStreamReader.START_ELEMENT:
                     depth++;
                     storeElement(reader);
                     break;
                 case XMLStreamReader.END_ELEMENT:
                     depth--;
                     storeStructure(T_END);
                     break;
                 case XMLStreamReader.NAMESPACE:
                     storeNamespaceAttributes(reader);
                     break;
                 case XMLStreamReader.ATTRIBUTE:
                     storeAttributes(reader);
                     break;
                 case XMLStreamReader.SPACE:
                 case XMLStreamReader.CHARACTERS:
                 case XMLStreamReader.CDATA: {
                     storeContentCharacters(T_TEXT_AS_CHAR_ARRAY,
                             reader.getTextCharacters(), reader.getTextStart(),
                             reader.getTextLength());
                     break;
                 }
                 case XMLStreamReader.COMMENT:
                     storeComment(reader);
                     break;
                 case XMLStreamReader.PROCESSING_INSTRUCTION:
                     storeProcessingInstruction(reader);
                     break;
             }
         }

         /*
          * Move to next item after the end of the element
          * that has been stored
          */
         _eventType = reader.next();
     }

     private void storeElementWithInScopeNamespaces(XMLStreamReader reader) {
         storeQualifiedName(T_ELEMENT_LN,
                 reader.getPrefix(), reader.getNamespaceURI(), reader.getLocalName());

         if (reader.getNamespaceCount() > 0) {
             storeNamespaceAttributes(reader);
         }

         if (reader.getAttributeCount() > 0) {
             storeAttributes(reader);
         }
     }

     private void storeElement(XMLStreamReader reader) {
         storeQualifiedName(T_ELEMENT_LN,
                 reader.getPrefix(), reader.getNamespaceURI(), reader.getLocalName());

         if (reader.getNamespaceCount() > 0) {
             storeNamespaceAttributes(reader);
         }

         if (reader.getAttributeCount() > 0) {
             storeAttributes(reader);
         }
     }

     private void storeNamespaceAttributes(XMLStreamReader reader) {
         int count = reader.getNamespaceCount();
         for (int i = 0; i < count; i++) {
             storeNamespaceAttribute(reader.getNamespacePrefix(i), reader.getNamespaceURI(i));
         }
     }

     private void storeAttributes(XMLStreamReader reader) {
         int count = reader.getAttributeCount();
         for (int i = 0; i < count; i++) {
             storeAttribute(reader.getAttributePrefix(i), reader.getAttributeNamespace(i), reader.getAttributeLocalName(i),
                     reader.getAttributeType(i), reader.getAttributeValue(i));
         }
     }

     private void storeComment(XMLStreamReader reader) {
         storeContentCharacters(T_COMMENT_AS_CHAR_ARRAY,
                 reader.getTextCharacters(), reader.getTextStart(), reader.getTextLength());
     }

     private void storeProcessingInstruction(XMLStreamReader reader) {
         storeProcessingInstruction(reader.getPITarget(), reader.getPIData());
     }
 }
	/*
	* Copyright 2005-2006 Sun Microsystems, Inc. All Rights Reserved.
	* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
	*
	* This code is free software; you can redistribute it and/or modify it
	* under the terms of the GNU General Public License version 2 only, as
	* published by the Free Software Foundation. Sun designates this
	* particular file as subject to the "Classpath" exception as provided
	* by Sun in the LICENSE file that accompanied this code.
	*
	* This code 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 GNU General Public License
	* version 2 for more details (a copy is included in the LICENSE file that
	* accompanied this code).
	*
	* You should have received a copy of the GNU General Public License version
	* 2 along with this work; if not, write to the Free Software Foundation,
	* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
	*
	* Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
	* CA 95054 USA or visit www.sun.com if you need additional information or
	* have any questions.
	*/
	package com.sun.xml.internal.stream.buffer.stax;

	import com.sun.xml.internal.stream.buffer.MutableXMLStreamBuffer;
	import com.sun.xml.internal.org.jvnet.staxex.Base64Data;
	import com.sun.xml.internal.org.jvnet.staxex.XMLStreamReaderEx;

	import javax.xml.stream.XMLStreamException;
	import javax.xml.stream.XMLStreamReader;
	import java.util.HashMap;
	import java.util.Map;

	/**
	* Create a buffer using an {@link XMLStreamReader}.
	* <p>
	* TODO: Implement the marking the stream on the element when an ID
	* attribute on the element is defined
	*/
	public class StreamReaderBufferCreator extends StreamBufferCreator {
	private int _eventType;
	private boolean _storeInScopeNamespacesOnElementFragment;
	private Map<String, Integer> _inScopePrefixes;

	/**
	* Create a stream reader buffer creator.
	* <p>
	* A stream buffer will be created for storing the infoset
	* from a stream reader.
	*/
	public StreamReaderBufferCreator() {
	}

	/**
	* Create a stream reader buffer creator using a mutable stream buffer.
	* <p>
	* @param buffer the mutable stream buffer.
	*/
	public StreamReaderBufferCreator(MutableXMLStreamBuffer buffer) {
	setBuffer(buffer);
	}

	/**
	* Create the buffer from a stream reader.
	* <p>
	* The stream reader must be positioned at the start of the document
	* or the start of an element.
	* <p>
	* If the stream is positioned at the start of the document then the
	* whole document is stored and after storing the stream will be positioned
	* at the end of the document.
	* <p>
	* If the stream is positioned at the start of an element then the
	* element and all its children will be stored and after storing the stream
	* will be positioned at the next event after the end of the element.
	* <p>
	* @return the mutable stream buffer.
	* @throws XMLStreamException if the stream reader is not positioned at
	* the start of the document or at an element.
	*/
	public MutableXMLStreamBuffer create(XMLStreamReader reader) throws XMLStreamException {
	if (_buffer == null) {
	createBuffer();
	}
	store(reader);

	return getXMLStreamBuffer();
	}

	/**
	* Creates the buffer from a stream reader that is an element fragment.
	* <p>
	* The stream reader will be moved to the position of the next start of
	* an element if the stream reader is not already positioned at the start
	* of an element.
	* <p>
	* The element and all its children will be stored and after storing the stream
	* will be positioned at the next event after the end of the element.
	* <p>
	* @param storeInScopeNamespaces true if in-scope namespaces of the element
	* fragment should be stored.
	* @return the mutable stream buffer.
	* @throws XMLStreamException if the stream reader cannot be positioned at
	* the start of an element.
	*/
	public MutableXMLStreamBuffer createElementFragment(XMLStreamReader reader,
	boolean storeInScopeNamespaces) throws XMLStreamException {
	if (_buffer == null) {
	createBuffer();
	}

	if (!reader.hasNext()) {
	return _buffer;
	}

	_storeInScopeNamespacesOnElementFragment = storeInScopeNamespaces;

	_eventType = reader.getEventType();
	if (_eventType != XMLStreamReader.START_ELEMENT) {
	do {
	_eventType = reader.next();
	} while(_eventType != XMLStreamReader.START_ELEMENT && _eventType != XMLStreamReader.END_DOCUMENT);
	}

	if (storeInScopeNamespaces) {
	_inScopePrefixes = new HashMap<String,Integer>();
	}

	storeElementAndChildren(reader);

	return getXMLStreamBuffer();
	}

	private void store(XMLStreamReader reader) throws XMLStreamException {
	if (!reader.hasNext()) {
	return;
	}

	_eventType = reader.getEventType();
	switch (_eventType) {
	case XMLStreamReader.START_DOCUMENT:
	storeDocumentAndChildren(reader);
	break;
	case XMLStreamReader.START_ELEMENT:
	storeElementAndChildren(reader);
	break;
	default:
	throw new XMLStreamException("XMLStreamReader not positioned at a document or element");
	}

	increaseTreeCount();
	}

	private void storeDocumentAndChildren(XMLStreamReader reader) throws XMLStreamException {
	storeStructure(T_DOCUMENT);

	_eventType = reader.next();
	while (_eventType != XMLStreamReader.END_DOCUMENT) {
	switch (_eventType) {
	case XMLStreamReader.START_ELEMENT:
	storeElementAndChildren(reader);
	continue;
	case XMLStreamReader.COMMENT:
	storeComment(reader);
	break;
	case XMLStreamReader.PROCESSING_INSTRUCTION:
	storeProcessingInstruction(reader);
	break;
	}
	_eventType = reader.next();
	}

	storeStructure(T_END);
	}

	private void storeElementAndChildren(XMLStreamReader reader) throws XMLStreamException {
	if (reader instanceof XMLStreamReaderEx) {
	storeElementAndChildrenEx((XMLStreamReaderEx)reader);
	} else {
	storeElementAndChildrenNoEx(reader);
	}
	}

	private void storeElementAndChildrenEx(XMLStreamReaderEx reader) throws XMLStreamException {
	int depth = 1;
	if (_storeInScopeNamespacesOnElementFragment) {
	storeElementWithInScopeNamespaces(reader);
	} else {
	storeElement(reader);
	}

	while(depth > 0) {
	_eventType = reader.next();
	switch (_eventType) {
	case XMLStreamReader.START_ELEMENT:
	depth++;
	storeElement(reader);
	break;
	case XMLStreamReader.END_ELEMENT:
	depth--;
	storeStructure(T_END);
	break;
	case XMLStreamReader.NAMESPACE:
	storeNamespaceAttributes(reader);
	break;
	case XMLStreamReader.ATTRIBUTE:
	storeAttributes(reader);
	break;
	case XMLStreamReader.SPACE:
	case XMLStreamReader.CHARACTERS:
	case XMLStreamReader.CDATA: {
	CharSequence c = reader.getPCDATA();
	if (c instanceof Base64Data) {
	storeStructure(T_TEXT_AS_OBJECT);
	storeContentObject(((Base64Data)c).clone());
	} else {
	storeContentCharacters(T_TEXT_AS_CHAR_ARRAY,
	reader.getTextCharacters(), reader.getTextStart(),
	reader.getTextLength());
	}
	break;
	}
	case XMLStreamReader.COMMENT:
	storeComment(reader);
	break;
	case XMLStreamReader.PROCESSING_INSTRUCTION:
	storeProcessingInstruction(reader);
	break;
	}
	}

	/*
	* Move to next item after the end of the element
	* that has been stored
	*/
	_eventType = reader.next();
	}

	private void storeElementAndChildrenNoEx(XMLStreamReader reader) throws XMLStreamException {
	int depth = 1;
	if (_storeInScopeNamespacesOnElementFragment) {
	storeElementWithInScopeNamespaces(reader);
	} else {
	storeElement(reader);
	}

	while(depth > 0) {
	_eventType = reader.next();
	switch (_eventType) {
	case XMLStreamReader.START_ELEMENT:
	depth++;
	storeElement(reader);
	break;
	case XMLStreamReader.END_ELEMENT:
	depth--;
	storeStructure(T_END);
	break;
	case XMLStreamReader.NAMESPACE:
	storeNamespaceAttributes(reader);
	break;
	case XMLStreamReader.ATTRIBUTE:
	storeAttributes(reader);
	break;
	case XMLStreamReader.SPACE:
	case XMLStreamReader.CHARACTERS:
	case XMLStreamReader.CDATA: {
	storeContentCharacters(T_TEXT_AS_CHAR_ARRAY,
	reader.getTextCharacters(), reader.getTextStart(),
	reader.getTextLength());
	break;
	}
	case XMLStreamReader.COMMENT:
	storeComment(reader);
	break;
	case XMLStreamReader.PROCESSING_INSTRUCTION:
	storeProcessingInstruction(reader);
	break;
	}
	}

	/*
	* Move to next item after the end of the element
	* that has been stored
	*/
	_eventType = reader.next();
	}

	private void storeElementWithInScopeNamespaces(XMLStreamReader reader) {
	storeQualifiedName(T_ELEMENT_LN,
	reader.getPrefix(), reader.getNamespaceURI(), reader.getLocalName());

	if (reader.getNamespaceCount() > 0) {
	storeNamespaceAttributes(reader);
	}

	if (reader.getAttributeCount() > 0) {
	storeAttributes(reader);
	}
	}

	private void storeElement(XMLStreamReader reader) {
	storeQualifiedName(T_ELEMENT_LN,
	reader.getPrefix(), reader.getNamespaceURI(), reader.getLocalName());

	if (reader.getNamespaceCount() > 0) {
	storeNamespaceAttributes(reader);
	}

	if (reader.getAttributeCount() > 0) {
	storeAttributes(reader);
	}
	}

	private void storeNamespaceAttributes(XMLStreamReader reader) {
	int count = reader.getNamespaceCount();
	for (int i = 0; i < count; i++) {
	storeNamespaceAttribute(reader.getNamespacePrefix(i), reader.getNamespaceURI(i));
	}
	}

	private void storeAttributes(XMLStreamReader reader) {
	int count = reader.getAttributeCount();
	for (int i = 0; i < count; i++) {
	storeAttribute(reader.getAttributePrefix(i), reader.getAttributeNamespace(i), reader.getAttributeLocalName(i),
	reader.getAttributeType(i), reader.getAttributeValue(i));
	}
	}

	private void storeComment(XMLStreamReader reader) {
	storeContentCharacters(T_COMMENT_AS_CHAR_ARRAY,
	reader.getTextCharacters(), reader.getTextStart(), reader.getTextLength());
	}

	private void storeProcessingInstruction(XMLStreamReader reader) {
	storeProcessingInstruction(reader.getPITarget(), reader.getPIData());
	}
	}