com.sun.org.apache.xalan.internal.xsltc.trax.SAX2StAXEventWriter Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of jaxp-ri Show documentation
Show all versions of jaxp-ri Show documentation
Java API for XML Processing Reference Implementation
The newest version!
/*
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
*
* Copyright (c) 1997-2010 Oracle and/or its affiliates. All rights reserved.
*
* The contents of this file are subject to the terms of either the GNU
* General Public License Version 2 only ("GPL") or the Common Development
* and Distribution License("CDDL") (collectively, the "License"). You
* may not use this file except in compliance with the License. You can
* obtain a copy of the License at
* https://glassfish.dev.java.net/public/CDDL+GPL_1_1.html
* or packager/legal/LICENSE.txt. See the License for the specific
* language governing permissions and limitations under the License.
*
* When distributing the software, include this License Header Notice in each
* file and include the License file at packager/legal/LICENSE.txt.
*
* GPL Classpath Exception:
* Oracle designates this particular file as subject to the "Classpath"
* exception as provided by Oracle in the GPL Version 2 section of the License
* file that accompanied this code.
*
* Modifications:
* If applicable, add the following below the License Header, with the fields
* enclosed by brackets [] replaced by your own identifying information:
* "Portions Copyright [year] [name of copyright owner]"
*
* Contributor(s):
* If you wish your version of this file to be governed by only the CDDL or
* only the GPL Version 2, indicate your decision by adding "[Contributor]
* elects to include this software in this distribution under the [CDDL or GPL
* Version 2] license." If you don't indicate a single choice of license, a
* recipient has the option to distribute your version of this file under
* either the CDDL, the GPL Version 2 or to extend the choice of license to
* its licensees as provided above. However, if you add GPL Version 2 code
* and therefore, elected the GPL Version 2 license, then the option applies
* only if the new code is made subject to such option by the copyright
* holder.
*/
/*
* $Id: SAX2StAXEventWriter.java,v 1.11 2010-11-01 04:34:32 joehw Exp $
* %W% %E%
*/
package com.sun.org.apache.xalan.internal.xsltc.trax;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import javax.xml.stream.XMLEventFactory;
import javax.xml.stream.XMLStreamException;
import javax.xml.stream.events.*;
import javax.xml.stream.XMLEventWriter;
import org.xml.sax.Attributes;
import org.xml.sax.SAXException;
import org.xml.sax.ext.Locator2;
/**
* @author Sunitha Reddy
*/
public class SAX2StAXEventWriter extends SAX2StAXBaseWriter {
private XMLEventWriter writer;
private XMLEventFactory eventFactory;
private List namespaceStack = new ArrayList();
private boolean needToCallStartDocument = false;
public SAX2StAXEventWriter() {
eventFactory = XMLEventFactory.newInstance();
}
public SAX2StAXEventWriter(XMLEventWriter writer) {
this.writer = writer;
eventFactory = XMLEventFactory.newInstance();
}
public SAX2StAXEventWriter(XMLEventWriter writer,
XMLEventFactory factory) {
this.writer = writer;
if (factory != null) {
this.eventFactory = factory;
} else {
eventFactory = XMLEventFactory.newInstance();
}
}
public XMLEventWriter getEventWriter() {
return writer;
}
public void setEventWriter(XMLEventWriter writer) {
this.writer = writer;
}
public XMLEventFactory getEventFactory() {
return eventFactory;
}
public void setEventFactory(XMLEventFactory factory) {
this.eventFactory = factory;
}
public void startDocument() throws SAXException {
super.startDocument();
namespaceStack.clear();
eventFactory.setLocation(getCurrentLocation());
// Encoding and version info will be available only after startElement
// is called for first time. So, defer START_DOCUMENT event of StAX till
// that point of time.
needToCallStartDocument = true;
}
private void writeStartDocument() throws SAXException {
try {
if (docLocator == null)
writer.add(eventFactory.createStartDocument());
else {
try{
writer.add(eventFactory.createStartDocument(((Locator2)docLocator).getEncoding(),((Locator2)docLocator).getXMLVersion()));
} catch(ClassCastException e){
writer.add(eventFactory.createStartDocument());
}
}
} catch (XMLStreamException e) {
throw new SAXException(e);
}
needToCallStartDocument = false;
}
public void endDocument() throws SAXException {
eventFactory.setLocation(getCurrentLocation());
try {
writer.add(eventFactory.createEndDocument());
} catch (XMLStreamException e) {
throw new SAXException(e);
}
super.endDocument();
// clear the namespaces
namespaceStack.clear();
}
public void startElement(String uri, String localName, String qName, Attributes attributes) throws SAXException {
if (needToCallStartDocument) {
writeStartDocument();
}
// set document location
eventFactory.setLocation(getCurrentLocation());
// create attribute and namespace events
Collection[] events = {null, null};
createStartEvents(attributes, events);
namespaceStack.add(events[0]);
try {
String[] qname = {null, null};
parseQName(qName, qname);
writer.add(eventFactory.createStartElement(qname[0], uri,
qname[1], events[1].iterator(), events[0].iterator()));
} catch (XMLStreamException e) {
throw new SAXException(e);
} finally {
super.startElement(uri, localName, qName, attributes);
}
}
public void endElement(String uri, String localName, String qName)
throws SAXException {
super.endElement(uri, localName, qName);
eventFactory.setLocation(getCurrentLocation());
// parse name
String[] qname = {null, null};
parseQName(qName, qname);
// get namespaces
Collection nsList = (Collection) namespaceStack.remove(namespaceStack.size() - 1);
Iterator nsIter = nsList.iterator();
try {
writer.add(eventFactory.createEndElement(qname[0], uri, qname[1],
nsIter));
} catch (XMLStreamException e) {
throw new SAXException(e);
}
}
public void comment(char[] ch, int start, int length) throws SAXException {
if (needToCallStartDocument) {
// Drat. We were trying to postpone this until the first element so that we could get
// the locator, but we can't output a comment before the start document, so we're just
// going to have to do without the locator if it hasn't been set yet.
writeStartDocument();
}
super.comment(ch, start, length);
eventFactory.setLocation(getCurrentLocation());
try {
writer.add(eventFactory.createComment(new String(ch, start,
length)));
} catch (XMLStreamException e) {
throw new SAXException(e);
}
}
public void characters(char[] ch, int start, int length)
throws SAXException {
super.characters(ch, start, length);
try {
if (!isCDATA) {
eventFactory.setLocation(getCurrentLocation());
writer.add(eventFactory.createCharacters(new String(ch,
start, length)));
}
} catch (XMLStreamException e) {
throw new SAXException(e);
}
}
public void ignorableWhitespace(char[] ch, int start, int length)
throws SAXException {
super.ignorableWhitespace(ch, start, length);
characters(ch, start, length);
}
public void processingInstruction(String target, String data)
throws SAXException {
if (needToCallStartDocument) {
// Drat. We were trying to postpone this until the first element so that we could get
// the locator, but we can't output a PI before the start document, so we're just
// going to have to do without the locator if it hasn't been set yet.
writeStartDocument();
}
super.processingInstruction(target, data);
try {
writer.add(eventFactory.createProcessingInstruction(target, data));
} catch (XMLStreamException e) {
throw new SAXException(e);
}
}
public void endCDATA() throws SAXException {
eventFactory.setLocation(getCurrentLocation());
try {
writer.add(eventFactory.createCData(CDATABuffer.toString()));
} catch (XMLStreamException e) {
throw new SAXException(e);
}
super.endCDATA();
}
protected void createStartEvents(Attributes attributes, Collection[] events) {
Map nsMap = null;
List attrs = null;
// create namespaces
if (namespaces != null) {
final int nDecls = namespaces.size();
for (int i = 0; i < nDecls; i++) {
final String prefix = (String) namespaces.elementAt(i++);
String uri = (String) namespaces.elementAt(i);
Namespace ns = createNamespace(prefix, uri);
if (nsMap == null) {
nsMap = new HashMap();
}
nsMap.put(prefix, ns);
}
}
// create attributes
String[] qname = {null, null};
for (int i = 0, s = attributes.getLength(); i < s; i++) {
parseQName(attributes.getQName(i), qname);
String attrPrefix = qname[0];
String attrLocal = qname[1];
String attrQName = attributes.getQName(i);
String attrValue = attributes.getValue(i);
String attrURI = attributes.getURI(i);
if ("xmlns".equals(attrQName) || "xmlns".equals(attrPrefix)) {
// namespace declaration disguised as an attribute. If the
// namespace has already been declared, skip it, otherwise
// write it as an namespace
if (nsMap == null) {
nsMap = new HashMap();
}
if (!nsMap.containsKey(attrLocal)) {
Namespace ns = createNamespace(attrLocal, attrValue);
nsMap.put(attrLocal, ns);
}
} else {
Attribute attribute;
if (attrPrefix.length() > 0) {
attribute = eventFactory.createAttribute(attrPrefix,
attrURI, attrLocal, attrValue);
} else {
attribute = eventFactory.createAttribute(attrLocal,
attrValue);
}
if (attrs == null) {
attrs = new ArrayList();
}
attrs.add(attribute);
}
}
events[0] = (nsMap == null ? Collections.EMPTY_LIST : nsMap.values());
events[1] = (attrs == null ? Collections.EMPTY_LIST : attrs);
}
protected Namespace createNamespace(String prefix, String uri) {
if (prefix == null || prefix.length() == 0) {
return eventFactory.createNamespace(uri);
} else {
return eventFactory.createNamespace(prefix, uri);
}
}
}