aoqi@0: /*
aoqi@0:  * Copyright (c) 2004, 2011, Oracle and/or its affiliates. All rights reserved.
aoqi@0:  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
aoqi@0:  *
aoqi@0:  * This code is free software; you can redistribute it and/or modify it
aoqi@0:  * under the terms of the GNU General Public License version 2 only, as
aoqi@0:  * published by the Free Software Foundation.  Oracle designates this
aoqi@0:  * particular file as subject to the "Classpath" exception as provided
aoqi@0:  * by Oracle in the LICENSE file that accompanied this code.
aoqi@0:  *
aoqi@0:  * This code is distributed in the hope that it will be useful, but WITHOUT
aoqi@0:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
aoqi@0:  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
aoqi@0:  * version 2 for more details (a copy is included in the LICENSE file that
aoqi@0:  * accompanied this code).
aoqi@0:  *
aoqi@0:  * You should have received a copy of the GNU General Public License version
aoqi@0:  * 2 along with this work; if not, write to the Free Software Foundation,
aoqi@0:  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
aoqi@0:  *
aoqi@0:  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
aoqi@0:  * or visit www.oracle.com if you need additional information or have any
aoqi@0:  * questions.
aoqi@0:  *
aoqi@0:  * THIS FILE WAS MODIFIED BY SUN MICROSYSTEMS, INC.
aoqi@0:  */
aoqi@0: 
aoqi@0: package com.sun.xml.internal.fastinfoset.stax.events;
aoqi@0: 
aoqi@0: import com.sun.xml.internal.fastinfoset.stax.*;
aoqi@0: import java.util.NoSuchElementException;
aoqi@0: import javax.xml.stream.XMLInputFactory;
aoqi@0: import javax.xml.stream.XMLStreamException;
aoqi@0: import javax.xml.stream.XMLStreamReader;
aoqi@0: import javax.xml.stream.events.XMLEvent;
aoqi@0: import javax.xml.stream.util.XMLEventAllocator;
aoqi@0: import com.sun.xml.internal.fastinfoset.CommonResourceBundle;
aoqi@0: 
aoqi@0: 
aoqi@0: 
aoqi@0: public class StAXEventReader implements javax.xml.stream.XMLEventReader{
aoqi@0:     protected XMLStreamReader _streamReader ;
aoqi@0:     protected XMLEventAllocator _eventAllocator;
aoqi@0:     private XMLEvent _currentEvent;     //the current event
aoqi@0:     private XMLEvent[] events = new XMLEvent[3];
aoqi@0:     private int size = 3;
aoqi@0:     private int currentIndex = 0;
aoqi@0:     private boolean hasEvent = false;   //true when current event exists, false initially & at end
aoqi@0: 
aoqi@0:     //only constructor will do because we delegate everything to underlying XMLStreamReader
aoqi@0:     public StAXEventReader(XMLStreamReader reader) throws  XMLStreamException {
aoqi@0:         _streamReader = reader ;
aoqi@0:         _eventAllocator = (XMLEventAllocator)reader.getProperty(XMLInputFactory.ALLOCATOR);
aoqi@0:         if(_eventAllocator == null){
aoqi@0:             _eventAllocator = new StAXEventAllocatorBase();
aoqi@0:         }
aoqi@0:         //initialize
aoqi@0:         if (_streamReader.hasNext())
aoqi@0:         {
aoqi@0:             _streamReader.next();
aoqi@0:             _currentEvent =_eventAllocator.allocate(_streamReader);
aoqi@0:             events[0] = _currentEvent;
aoqi@0:             hasEvent = true;
aoqi@0:         } else {
aoqi@0:             throw new XMLStreamException(CommonResourceBundle.getInstance().getString("message.noElement"));
aoqi@0:         }
aoqi@0:     }
aoqi@0: 
aoqi@0:     public boolean hasNext() {
aoqi@0:         return hasEvent;
aoqi@0:     }
aoqi@0: 
aoqi@0:     public XMLEvent nextEvent() throws XMLStreamException {
aoqi@0:         XMLEvent event = null;
aoqi@0:         XMLEvent nextEvent = null;
aoqi@0:         if (hasEvent)
aoqi@0:         {
aoqi@0:             event = events[currentIndex];
aoqi@0:             events[currentIndex] = null;
aoqi@0:             if (_streamReader.hasNext())
aoqi@0:             {
aoqi@0:                 //advance and read the next
aoqi@0:                 _streamReader.next();
aoqi@0:                 nextEvent = _eventAllocator.allocate(_streamReader);
aoqi@0:                 if (++currentIndex==size)
aoqi@0:                     currentIndex = 0;
aoqi@0:                 events[currentIndex] = nextEvent;
aoqi@0:                 hasEvent = true;
aoqi@0:             } else {
aoqi@0:                 _currentEvent = null;
aoqi@0:                 hasEvent = false;
aoqi@0:             }
aoqi@0:             return event;
aoqi@0:         }
aoqi@0:         else{
aoqi@0:             throw new NoSuchElementException();
aoqi@0:         }
aoqi@0:     }
aoqi@0: 
aoqi@0:     public void remove(){
aoqi@0:         //stream reader is read-only.
aoqi@0:         throw new java.lang.UnsupportedOperationException();
aoqi@0:     }
aoqi@0: 
aoqi@0: 
aoqi@0:     public void close() throws XMLStreamException {
aoqi@0:         _streamReader.close();
aoqi@0:     }
aoqi@0: 
aoqi@0:     /** Reads the content of a text-only element. Precondition:
aoqi@0:      * the current event is START_ELEMENT. Postcondition:
aoqi@0:      * The current event is the corresponding END_ELEMENT.
aoqi@0:      * @throws XMLStreamException if the current event is not a START_ELEMENT
aoqi@0:      * or if a non text element is encountered
aoqi@0:      */
aoqi@0:     public String getElementText() throws XMLStreamException {
aoqi@0:         if(!hasEvent) {
aoqi@0:             throw new NoSuchElementException();
aoqi@0:         }
aoqi@0: 
aoqi@0:         if(!_currentEvent.isStartElement()) {
aoqi@0:             StAXDocumentParser parser = (StAXDocumentParser)_streamReader;
aoqi@0:             return parser.getElementText(true);
aoqi@0:         } else {
aoqi@0:             return _streamReader.getElementText();
aoqi@0:         }
aoqi@0:     }
aoqi@0: 
aoqi@0:     /** Get the value of a feature/property from the underlying implementation
aoqi@0:      * @param name The name of the property
aoqi@0:      * @return The value of the property
aoqi@0:      * @throws IllegalArgumentException if the property is not supported
aoqi@0:      */
aoqi@0:     public Object getProperty(java.lang.String name) throws java.lang.IllegalArgumentException {
aoqi@0:         return _streamReader.getProperty(name) ;
aoqi@0:     }
aoqi@0: 
aoqi@0:     /** Skips any insignificant space events until a START_ELEMENT or
aoqi@0:      * END_ELEMENT is reached. If anything other than space characters are
aoqi@0:      * encountered, an exception is thrown. This method should
aoqi@0:      * be used when processing element-only content because
aoqi@0:      * the parser is not able to recognize ignorable whitespace if
aoqi@0:      * the DTD is missing or not interpreted.
aoqi@0:      * @throws XMLStreamException if anything other than space characters are encountered
aoqi@0:      */
aoqi@0:     public XMLEvent nextTag() throws XMLStreamException {
aoqi@0:         if(!hasEvent) {
aoqi@0:             throw new NoSuchElementException();
aoqi@0:         }
aoqi@0:         StAXDocumentParser parser = (StAXDocumentParser)_streamReader;
aoqi@0:         parser.nextTag(true);
aoqi@0:         return _eventAllocator.allocate(_streamReader);
aoqi@0:     }
aoqi@0: 
aoqi@0:     //XMLEventReader extends Iterator;
aoqi@0:     public Object next() {
aoqi@0:         try{
aoqi@0:             return nextEvent();
aoqi@0:         }catch(XMLStreamException streamException){
aoqi@0:             return null;
aoqi@0:         }
aoqi@0:     }
aoqi@0: 
aoqi@0:     public XMLEvent peek() throws XMLStreamException{
aoqi@0:         if (!hasEvent)
aoqi@0:              throw new XMLStreamException(CommonResourceBundle.getInstance().getString("message.noElement"));
aoqi@0:         _currentEvent = events[currentIndex];
aoqi@0:         return _currentEvent;
aoqi@0:     }
aoqi@0: 
aoqi@0:     public void setAllocator(XMLEventAllocator allocator) {
aoqi@0:         if (allocator == null)
aoqi@0:             throw new IllegalArgumentException(CommonResourceBundle.getInstance().getString("message.nullXMLEventAllocator"));
aoqi@0: 
aoqi@0:         _eventAllocator = allocator;
aoqi@0:     }
aoqi@0: 
aoqi@0: 
aoqi@0: }