aoqi@0: /*
aoqi@0:  * Copyright (c) 1997, 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: 
aoqi@0: package com.sun.xml.internal.bind.v2.runtime.output;
aoqi@0: 
aoqi@0: import java.io.IOException;
aoqi@0: 
aoqi@0: import javax.xml.stream.XMLEventFactory;
aoqi@0: import javax.xml.stream.XMLEventWriter;
aoqi@0: import javax.xml.stream.XMLStreamException;
aoqi@0: import javax.xml.stream.events.Attribute;
aoqi@0: import javax.xml.stream.events.Characters;
aoqi@0: 
aoqi@0: import com.sun.xml.internal.bind.v2.runtime.XMLSerializer;
aoqi@0: 
aoqi@0: import org.xml.sax.SAXException;
aoqi@0: 
aoqi@0: /**
aoqi@0:  * {@link XmlOutput} that writes to StAX {@link XMLEventWriter}.
aoqi@0:  *
aoqi@0:  * @author Kohsuke Kawaguchi
aoqi@0:  */
aoqi@0: public class XMLEventWriterOutput extends XmlOutputAbstractImpl {
aoqi@0:     private final XMLEventWriter out;
aoqi@0:     private final XMLEventFactory ef;
aoqi@0: 
aoqi@0:     /** One whitespace. */
aoqi@0:     private final Characters sp;
aoqi@0: 
aoqi@0:     public XMLEventWriterOutput(XMLEventWriter out) {
aoqi@0:         this.out = out;
aoqi@0:         ef = XMLEventFactory.newInstance();
aoqi@0:         sp = ef.createCharacters(" ");
aoqi@0:     }
aoqi@0: 
aoqi@0:     // not called if we are generating fragments
aoqi@0:     @Override
aoqi@0:     public void startDocument(XMLSerializer serializer, boolean fragment, int[] nsUriIndex2prefixIndex, NamespaceContextImpl nsContext) throws IOException, SAXException, XMLStreamException {
aoqi@0:         super.startDocument(serializer, fragment,nsUriIndex2prefixIndex,nsContext);
aoqi@0:         if(!fragment)
aoqi@0:             out.add(ef.createStartDocument());
aoqi@0:     }
aoqi@0: 
aoqi@0:     public void endDocument(boolean fragment) throws IOException, SAXException, XMLStreamException {
aoqi@0:         if(!fragment) {
aoqi@0:             out.add(ef.createEndDocument());
aoqi@0:             out.flush();
aoqi@0:         }
aoqi@0:         super.endDocument(fragment);
aoqi@0:     }
aoqi@0: 
aoqi@0:     public void beginStartTag(int prefix, String localName) throws IOException, XMLStreamException {
aoqi@0:         out.add(
aoqi@0:             ef.createStartElement(
aoqi@0:                 nsContext.getPrefix(prefix),
aoqi@0:                 nsContext.getNamespaceURI(prefix),
aoqi@0:                 localName));
aoqi@0: 
aoqi@0:         NamespaceContextImpl.Element nse = nsContext.getCurrent();
aoqi@0:         if(nse.count()>0) {
aoqi@0:             for( int i=nse.count()-1; i>=0; i-- ) {
aoqi@0:                 String uri = nse.getNsUri(i);
aoqi@0:                 if(uri.length()==0 && nse.getBase()==1)
aoqi@0:                     continue;   // no point in definint xmlns='' on the root
aoqi@0:                 out.add(ef.createNamespace(nse.getPrefix(i),uri));
aoqi@0:             }
aoqi@0:         }
aoqi@0:     }
aoqi@0: 
aoqi@0:     public void attribute(int prefix, String localName, String value) throws IOException, XMLStreamException {
aoqi@0:         Attribute att;
aoqi@0:         if(prefix==-1)
aoqi@0:             att = ef.createAttribute(localName,value);
aoqi@0:         else
aoqi@0:             att = ef.createAttribute(
aoqi@0:                     nsContext.getPrefix(prefix),
aoqi@0:                     nsContext.getNamespaceURI(prefix),
aoqi@0:                     localName, value);
aoqi@0: 
aoqi@0:         out.add(att);
aoqi@0:     }
aoqi@0: 
aoqi@0:     public void endStartTag() throws IOException, SAXException {
aoqi@0:         // noop
aoqi@0:     }
aoqi@0: 
aoqi@0:     public void endTag(int prefix, String localName) throws IOException, SAXException, XMLStreamException {
aoqi@0:         out.add(
aoqi@0:             ef.createEndElement(
aoqi@0:                 nsContext.getPrefix(prefix),
aoqi@0:                 nsContext.getNamespaceURI(prefix),
aoqi@0:                 localName));
aoqi@0:     }
aoqi@0: 
aoqi@0:     public void text(String value, boolean needsSeparatingWhitespace) throws IOException, SAXException, XMLStreamException {
aoqi@0:         if(needsSeparatingWhitespace)
aoqi@0:             out.add(sp);
aoqi@0:         out.add(ef.createCharacters(value));
aoqi@0:     }
aoqi@0: 
aoqi@0:     public void text(Pcdata value, boolean needsSeparatingWhitespace) throws IOException, SAXException, XMLStreamException {
aoqi@0:         text(value.toString(),needsSeparatingWhitespace);
aoqi@0:     }
aoqi@0: }