/*
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
*
* Copyright (c) 2004-2011 Oracle and/or its affiliates. All rights reserved.
*
* Oracle licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.sun.xml.fastinfoset.tools;
import com.sun.xml.fastinfoset.QualifiedName;
import com.sun.xml.fastinfoset.util.CharArray;
import com.sun.xml.fastinfoset.util.DuplicateAttributeVerifier;
import com.sun.xml.fastinfoset.util.KeyIntMap;
import com.sun.xml.fastinfoset.util.LocalNameQualifiedNamesMap;
import com.sun.xml.fastinfoset.util.PrefixArray;
import com.sun.xml.fastinfoset.util.QualifiedNameArray;
import com.sun.xml.fastinfoset.util.StringArray;
import com.sun.xml.fastinfoset.util.StringIntMap;
import com.sun.xml.fastinfoset.vocab.ParserVocabulary;
import com.sun.xml.fastinfoset.vocab.SerializerVocabulary;
import org.xml.sax.Attributes;
import org.xml.sax.SAXException;
import org.xml.sax.ext.LexicalHandler;
import org.xml.sax.helpers.DefaultHandler;
import com.sun.xml.fastinfoset.CommonResourceBundle;
import java.util.Set;
import org.jvnet.fastinfoset.FastInfosetSerializer;
public class VocabularyGenerator extends DefaultHandler implements LexicalHandler {
protected SerializerVocabulary _serializerVocabulary;
protected ParserVocabulary _parserVocabulary;
protected org.jvnet.fastinfoset.Vocabulary _v;
protected int attributeValueSizeConstraint = FastInfosetSerializer.MAX_ATTRIBUTE_VALUE_SIZE;
protected int characterContentChunkSizeContraint = FastInfosetSerializer.MAX_CHARACTER_CONTENT_CHUNK_SIZE;
/** Creates a new instance of VocabularyGenerator */
public VocabularyGenerator() {
_serializerVocabulary = new SerializerVocabulary();
_parserVocabulary = new ParserVocabulary();
_v = new org.jvnet.fastinfoset.Vocabulary();
}
public VocabularyGenerator(SerializerVocabulary serializerVocabulary) {
_serializerVocabulary = serializerVocabulary;
_parserVocabulary = new ParserVocabulary();
_v = new org.jvnet.fastinfoset.Vocabulary();
}
public VocabularyGenerator(ParserVocabulary parserVocabulary) {
_serializerVocabulary = new SerializerVocabulary();
_parserVocabulary = parserVocabulary;
_v = new org.jvnet.fastinfoset.Vocabulary();
}
/** Creates a new instance of VocabularyGenerator */
public VocabularyGenerator(SerializerVocabulary serializerVocabulary, ParserVocabulary parserVocabulary) {
_serializerVocabulary = serializerVocabulary;
_parserVocabulary = parserVocabulary;
_v = new org.jvnet.fastinfoset.Vocabulary();
}
public org.jvnet.fastinfoset.Vocabulary getVocabulary() {
return _v;
}
public void setCharacterContentChunkSizeLimit(int size) {
if (size < 0 ) {
size = 0;
}
characterContentChunkSizeContraint = size;
}
public int getCharacterContentChunkSizeLimit() {
return characterContentChunkSizeContraint;
}
public void setAttributeValueSizeLimit(int size) {
if (size < 0 ) {
size = 0;
}
attributeValueSizeConstraint = size;
}
public int getAttributeValueSizeLimit() {
return attributeValueSizeConstraint;
}
// ContentHandler
public void startDocument() throws SAXException {
}
public void endDocument() throws SAXException {
}
public void startPrefixMapping(String prefix, String uri) throws SAXException {
addToTable(prefix, _v.prefixes, _serializerVocabulary.prefix, _parserVocabulary.prefix);
addToTable(uri, _v.namespaceNames, _serializerVocabulary.namespaceName, _parserVocabulary.namespaceName);
}
public void endPrefixMapping(String prefix) throws SAXException {
}
public void startElement(String namespaceURI, String localName, String qName, Attributes atts) throws SAXException {
addToNameTable(namespaceURI, qName, localName,
_v.elements, _serializerVocabulary.elementName, _parserVocabulary.elementName, false);
for (int a = 0; a < atts.getLength(); a++) {
addToNameTable(atts.getURI(a), atts.getQName(a), atts.getLocalName(a),
_v.attributes, _serializerVocabulary.attributeName, _parserVocabulary.attributeName, true);
String value = atts.getValue(a);
if (value.length() < attributeValueSizeConstraint) {
addToTable(value, _v.attributeValues, _serializerVocabulary.attributeValue, _parserVocabulary.attributeValue);
}
}
}
public void endElement(String namespaceURI, String localName, String qName) throws SAXException {
}
public void characters(char[] ch, int start, int length) throws SAXException {
if (length < characterContentChunkSizeContraint) {
addToCharArrayTable(new CharArray(ch, start, length, true));
}
}
public void ignorableWhitespace(char[] ch, int start, int length) throws SAXException {
}
public void processingInstruction(String target, String data) throws SAXException {
}
public void setDocumentLocator(org.xml.sax.Locator locator) {
}
public void skippedEntity(String name) throws SAXException {
}
// LexicalHandler
public void comment(char[] ch, int start, int length) throws SAXException {
}
public void startCDATA() throws SAXException {
}
public void endCDATA() throws SAXException {
}
public void startDTD(String name, String publicId, String systemId) throws SAXException {
}
public void endDTD() throws SAXException {
}
public void startEntity(String name) throws SAXException {
}
public void endEntity(String name) throws SAXException {
}
public void addToTable(String s, Set v, StringIntMap m, StringArray a) {
if (s.length() == 0) {
return;
}
if (m.obtainIndex(s) == KeyIntMap.NOT_PRESENT) {
a.add(s);
}
v.add(s);
}
public void addToTable(String s, Set v, StringIntMap m, PrefixArray a) {
if (s.length() == 0) {
return;
}
if (m.obtainIndex(s) == KeyIntMap.NOT_PRESENT) {
a.add(s);
}
v.add(s);
}
public void addToCharArrayTable(CharArray c) {
if (_serializerVocabulary.characterContentChunk.obtainIndex(c.ch, c.start, c.length, false) == KeyIntMap.NOT_PRESENT) {
_parserVocabulary.characterContentChunk.add(c.ch, c.length);
}
_v.characterContentChunks.add(c.toString());
}
public void addToNameTable(String namespaceURI, String qName, String localName,
Set v, LocalNameQualifiedNamesMap m, QualifiedNameArray a,
boolean isAttribute) throws SAXException {
LocalNameQualifiedNamesMap.Entry entry = m.obtainEntry(qName);
if (entry._valueIndex > 0) {
QualifiedName[] names = entry._value;
for (int i = 0; i < entry._valueIndex; i++) {
if ((namespaceURI == names[i].namespaceName || namespaceURI.equals(names[i].namespaceName))) {
return;
}
}
}
String prefix = getPrefixFromQualifiedName(qName);
int namespaceURIIndex = -1;
int prefixIndex = -1;
int localNameIndex = -1;
if (namespaceURI.length() > 0) {
namespaceURIIndex = _serializerVocabulary.namespaceName.get(namespaceURI);
if (namespaceURIIndex == KeyIntMap.NOT_PRESENT) {
throw new SAXException(CommonResourceBundle.getInstance().
getString("message.namespaceURINotIndexed", new Object[]{Integer.valueOf(namespaceURIIndex)}));
}
if (prefix.length() > 0) {
prefixIndex = _serializerVocabulary.prefix.get(prefix);
if (prefixIndex == KeyIntMap.NOT_PRESENT) {
throw new SAXException(CommonResourceBundle.getInstance().
getString("message.prefixNotIndexed", new Object[]{Integer.valueOf(prefixIndex)}));
}
}
}
localNameIndex = _serializerVocabulary.localName.obtainIndex(localName);
if (localNameIndex == KeyIntMap.NOT_PRESENT) {
_parserVocabulary.localName.add(localName);
localNameIndex = _parserVocabulary.localName.getSize() - 1;
}
QualifiedName name = new QualifiedName(prefix, namespaceURI, localName, m.getNextIndex(),
prefixIndex, namespaceURIIndex, localNameIndex);
if (isAttribute) {
name.createAttributeValues(DuplicateAttributeVerifier.MAP_SIZE);
}
entry.addQualifiedName(name);
a.add(name);
v.add(name.getQName());
}
public static String getPrefixFromQualifiedName(String qName) {
int i = qName.indexOf(':');
String prefix = "";
if (i != -1) {
prefix = qName.substring(0, i);
}
return prefix;
}
}