All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.altova.xml.XmlTreeOperations Maven / Gradle / Ivy

There is a newer version: 2.0.4
Show newest version
// XmlTreeOperations.java
// This file contains generated code and will be overwritten when you rerun code generation.

package com.altova.xml;

import java.io.File;

import org.w3c.dom.Document;
import org.w3c.dom.DocumentType;
import org.w3c.dom.Element;
import org.w3c.dom.Node;

import com.altova.typeinfo.MemberInfo;

public class XmlTreeOperations
{
    private static int staticNSCount = 0;


    
	// TODO who calls this?
    public static boolean exists(Node node)
    {
        return node != null; //?
    }
	
    public static Document loadDocument(com.altova.io.Input input) throws Exception
    {
        if (input instanceof com.altova.io.FileInput)
        {
            return loadDocument(((com.altova.io.FileInput)input).getFilename());
        }
        switch (input.getType())
        {
            case com.altova.io.Input.IO_STREAM:
                return loadDocument(new org.xml.sax.InputSource(input.getStream()));
            
            case com.altova.io.Input.IO_READER:
                return loadDocument(new org.xml.sax.InputSource(input.getReader()));
            
            case com.altova.io.Input.IO_DOM:
                return input.getDocument();
            
            default:
                throw new Exception ("Unknown input type");
        }
    }
    
    public static Document loadDocument(org.xml.sax.InputSource source ) throws Exception
    {
        // turn true to validate onload
        boolean validation = false;
        javax.xml.parsers.DocumentBuilderFactory factory = javax.xml.parsers.DocumentBuilderFactory.newInstance();
        factory.setIgnoringElementContentWhitespace(true);
        factory.setNamespaceAware(true);
        if (validation) 
        {
            factory.setAttribute("http://java.sun.com/xml/jaxp/properties/schemaLanguage", "http://www.w3.org/2001/XMLSchema");
            factory.setValidating(true);
        }
        
        javax.xml.parsers.DocumentBuilder builder = factory.newDocumentBuilder();
        
        try
        {
            Document doc = builder.parse(source);
            return doc;
        }
        catch (Exception e)
        {
            throw new com.altova.SourceInstanceUnavailableException("cannot parse input stream.", e);
        }
    }
    
    public static Document loadDocument( java.io.InputStream stream ) throws Exception
    {
        return loadDocument(new org.xml.sax.InputSource(stream));
    }
    
    public static Document loadDocument( String filename ) throws Exception
    {
        // turn true to validate onload
        boolean validation = false;
        javax.xml.parsers.DocumentBuilderFactory factory = javax.xml.parsers.DocumentBuilderFactory.newInstance();
        factory.setIgnoringElementContentWhitespace(true);
        factory.setNamespaceAware(true);
        if (validation) 
        {
            factory.setAttribute("http://java.sun.com/xml/jaxp/properties/schemaLanguage", "http://www.w3.org/2001/XMLSchema");
            factory.setValidating(true);
        }
        javax.xml.parsers.DocumentBuilder builder = factory.newDocumentBuilder();
        
        try
        {
            Document doc = builder.parse( new File(filename));
            return doc;
        }
        catch (Exception e)
        {
            throw new com.altova.SourceInstanceUnavailableException("cannot parse input file.", e);
        }
    }


    public static Document loadXmlBinary( byte[] xmlTree ) throws Exception
    {
        // turn true to validate onload
        boolean validation = false;
        javax.xml.parsers.DocumentBuilderFactory factory = javax.xml.parsers.DocumentBuilderFactory.newInstance();
        factory.setIgnoringElementContentWhitespace(true);
        factory.setNamespaceAware(true);
        if (validation) 
        {
            factory.setAttribute("http://java.sun.com/xml/jaxp/properties/schemaLanguage", "http://www.w3.org/2001/XMLSchema");
            factory.setValidating(true);
        }

        javax.xml.parsers.DocumentBuilder builder = factory.newDocumentBuilder();
        
        try
        {
            Document doc = builder.parse(new java.io.ByteArrayInputStream(xmlTree));
            return doc;
        }
        catch (Exception e)
        {
            throw new com.altova.SourceInstanceUnavailableException("Cannot parse binary.", e);
        }
    }

    public static Document loadXml( String xmlString ) throws Exception
    {
        return loadDocument(new org.xml.sax.InputSource(new java.io.StringReader(xmlString)));
    }
    
    public static void saveDocument( Document doc, String filename, boolean prettyPrint, boolean omitXmlDecl ) throws Exception
    {
        saveDocument( doc, filename, "UTF-8", false, false, prettyPrint, omitXmlDecl );
    }
    
    public static void saveDocument( Document doc, com.altova.io.Output output, String encoding, boolean bBigEndian, boolean bBOM, boolean prettyPrint, boolean omitXmlDecl ) throws Exception
    {
        switch (output.getType())
        {
        case com.altova.io.Output.IO_STREAM:
            saveDocument(doc, output.getStream(), encoding, bBigEndian, bBOM, prettyPrint, omitXmlDecl);
            break;
            
        case com.altova.io.Output.IO_WRITER:
            saveXml(doc, prettyPrint, omitXmlDecl, output.getWriter());
            break;
            
        case com.altova.io.Output.IO_DOM:
            break;
            
        default:
            throw new Exception ("Unknown output type");
        }
    }
    
    public static void saveDocument( Document doc, java.io.OutputStream targetStream, String encoding, boolean bBigEndian, boolean bBOM, boolean prettyPrint, boolean omitXmlDecl ) throws Exception
    {
        if(prettyPrint)
            indentNode(doc.getDocumentElement(), 0);
    
        try {
            java.io.ByteArrayOutputStream result = new java.io.ByteArrayOutputStream();
            internalSave(new javax.xml.transform.stream.StreamResult(result), doc, encoding, prettyPrint, omitXmlDecl);

            correctByteOrderAndBOM( targetStream, result.toByteArray(), encoding, bBigEndian, bBOM );
        }
        catch (Exception e)
        {
            throw new com.altova.TargetInstanceUnavailableException("Cannot write to output stream", e);
        }
    }
    
    public static void saveDocument( Document doc, String filename, String encoding, boolean bBigEndian, boolean bBOM, boolean prettyPrint, boolean omitXmlDecl ) throws Exception
    {
        java.io.FileOutputStream outStream = new java.io.FileOutputStream( filename, false );
        saveDocument(doc, outStream, encoding, bBigEndian, bBOM, prettyPrint, omitXmlDecl);
        outStream.close();
    }
    
    public static byte[] saveXmlBinary( Document doc, String encoding, boolean bBigEndian, boolean bBOM, boolean prettyPrint ) throws Exception
    {
        if(prettyPrint)
            indentNode(doc.getDocumentElement(), 0);

        try
        {
            java.io.ByteArrayOutputStream result = new java.io.ByteArrayOutputStream();
            internalSave(new javax.xml.transform.stream.StreamResult(result), doc, encoding, prettyPrint, false);

            byte[] transformedData = result.toByteArray();
            result.reset();

            correctByteOrderAndBOM( result, transformedData, encoding, bBigEndian, bBOM );
            return result.toByteArray();
        }
        catch (Exception e)
        {
            throw new com.altova.TargetInstanceUnavailableException("Cannot serialize.", e);
        }
    }
    
    private static void saveXml( Document doc, boolean prettyPrint, boolean omitXmlDecl, java.io.Writer writer ) throws Exception
    {
        if(prettyPrint)
            indentNode(doc.getDocumentElement(), 0);

        try 
        {
            internalSave(new javax.xml.transform.stream.StreamResult(writer), doc, null, prettyPrint, omitXmlDecl);
            writer.flush();
        }
        catch (Exception e)
        {
            throw new com.altova.TargetInstanceUnavailableException("Cannot serialize.", e);
        }
    }
    
    public static String saveXml( Document doc, boolean prettyPrint, boolean omitXmlDecl ) throws Exception
    {
        java.io.StringWriter result = new java.io.StringWriter();
        saveXml(doc, prettyPrint, omitXmlDecl, result);
        return result.toString();
    }
	
    protected static void internalSave(javax.xml.transform.Result result, Node node, String encoding, boolean bIndent, boolean omitXmlDecl) throws Exception
    {
        try 
        {
            javax.xml.transform.Source source
                    = new javax.xml.transform.dom.DOMSource(node);
            javax.xml.transform.Transformer transformer
                    = javax.xml.transform.TransformerFactory.newInstance().newTransformer();
            if (encoding != null)
                transformer.setOutputProperty("encoding", encoding);
            Document doc = node.getNodeType() == Node.DOCUMENT_NODE ? (Document) node : node.getOwnerDocument();
            if (doc.getDoctype() != null) 
            {
                if (doc.getDoctype().getPublicId() != null)
                    transformer.setOutputProperty(javax.xml.transform.OutputKeys.DOCTYPE_PUBLIC, doc.getDoctype().getPublicId());
                if (doc.getDoctype().getSystemId() != null)
                    transformer.setOutputProperty(javax.xml.transform.OutputKeys.DOCTYPE_SYSTEM, doc.getDoctype().getSystemId());
            }
            transformer.setOutputProperty("indent", bIndent ? "yes" : "no");
			if (omitXmlDecl)
				transformer.setOutputProperty("omit-xml-declaration", "yes");
            transformer.transform(source, result);
        } 
        catch (javax.xml.transform.TransformerConfigurationException e) 
        {
            throw new Exception(e);
        } 
        catch (javax.xml.transform.TransformerException e) 
        {
            throw new Exception(e);
        }
    }

    private static void correctByteOrderAndBOM( java.io.OutputStream outStream, byte[] inData, String encoding, boolean bBigEndian, boolean bBOM ) throws Exception
    {
        if( inData == null || outStream == null )
            return;

        if( encoding == null )
        {
            outStream.write( inData );
            return;
        }

        int unisize = getUnicodeSizeFromEncodingName( encoding );
        if( unisize == 0 )
        {
            outStream.write( inData );
            return;
        }

        if( unisize < 2 )
            bBigEndian = false;

        int encbo = getEncodingAndByteOrderFromXml( inData );

        boolean bHasBOM = (encbo & 0x10) != 0;
        boolean bIsBigEndian = (encbo & 0x20) != 0;

        if( bBOM == bHasBOM && bBigEndian == bIsBigEndian )
        {
            outStream.write( inData );
            return;
        }

        if( unisize > 0 && bBOM && !bHasBOM )
        {
            if( unisize == 1 )
            {
                outStream.write( 0xef );
                outStream.write( 0xbb );
                outStream.write( 0xbf );
            }
            else
            if( unisize == 2 )
            {
                if( bBigEndian ) {
                    outStream.write( 0xfe );
                    outStream.write( 0xff );
                } else {
                    outStream.write( 0xff );
                    outStream.write( 0xfe );
                }
            }
        }

        int off = 0;
        int len = inData.length;

        if( !bBOM && bHasBOM )
        {
            int bomsize = (encbo & 0xf00) >> 8;
            off += bomsize;
            len -= bomsize;
        }

        if( bBigEndian != bIsBigEndian && unisize >= 2 )
        {
            int count = (len / unisize) * unisize;
            for( int i = 0 ; i < count ; i += unisize )
            {
                outStream.write( inData[off + i + 1] );
                outStream.write( inData[off + i    ] );
            }
        }
        else
            outStream.write( inData, off, len );
    }

    // Returns
    //	bits 0-3:		encoding char size in bytes (1 = UTF-8, 2 = UTF-16 or UCS-2)
    //	bit  4:			if BOM is included
    //	bit  5:			if BigEndian (set) or LittleEndian (cleared)
    //  bits 8-11:		BOM size in bytes
    private static int getEncodingAndByteOrderFromXml( byte[] source )
    {
        if( source == null ) return 0;
        if( source.length < 2 ) return 0;

        if( ((source[0] & 0xff) == 0xfe) && ((source[1] & 0xff) == 0xff) )
            return 0x232;	// UTF-16BE with BOM

        if( ((source[0] & 0xff) == 0xff) && ((source[1] & 0xff) == 0xfe) )
            return 0x212;	// UTF-16LE with BOM

        if( ((source[0] & 0xff) == 0x00) && ((source[1] & 0xff) == 0x3c) )	// checks for "<" character
            return 0x022;	// UTF-16BE without BOM

        if( ((source[0] & 0xff) == 0x3c) && ((source[1] & 0xff) == 0x00) )	// checks for "<" character
            return 0x002;	// UTF-16LE without BOM

        if( source.length < 3 ) return 0;

        if( ((source[0] & 0xff) == 0xef) // & is because chars are signed!
                && ((source[1] & 0xff) == 0xbb) && ((source[2] & 0xff) == 0xbf) )
            return 0x311;	// UTF-8 with BOM

        return 0;
    }

    private static int getUnicodeSizeFromEncodingName( String encoding )
    {
        if( encoding == null ) return 0;
        encoding = encoding.toUpperCase();

        if( encoding.indexOf("UTF-8") >= 0 )
            return 1;

        if( encoding.indexOf("UTF-16") >= 0 || encoding.indexOf("UCS-2") >= 0 )
            return 2;

        return 0;
    }

    private static void indentNode(org.w3c.dom.Node node, int nIndent) {
        if (node == null)
            return;
        
        if (nIndent > 0 && (node.getNodeType() == org.w3c.dom.Node.ELEMENT_NODE || node.getNodeType() == org.w3c.dom.Node.COMMENT_NODE))
        {
            if (node.getPreviousSibling() == null || 
                (node.getPreviousSibling().getNodeType() != org.w3c.dom.Node.CDATA_SECTION_NODE && 
                 node.getPreviousSibling().getNodeType() != org.w3c.dom.Node.TEXT_NODE))
            {
                String sIndent = "\n";
                for (int i=0; i 0)
            name = name.substring(name.indexOf(':')+1);
        
        if (nameNsUri != null && nameNsUri.length() > 0)
        {
            namePrefix = lookupPrefix(element, nameNsUri);
            if (namePrefix == null)
            {
                if (inlinedNamePrefix.length() > 0)
                    namePrefix = inlinedNamePrefix;
                else
                    namePrefix = "ns" + staticNSCount++;
                element.setAttributeNS("http://www.w3.org/2000/xmlns/", "xmlns:" + namePrefix, nameNsUri);
            }
            namePrefix += ":";
        }
        else
            namePrefix = "";
        
        if (valueNsUri != null && valueNsUri.length() > 0)
        {	
            valuePrefix = lookupPrefix(element, valueNsUri);
            if (valuePrefix == null)
            {
                valuePrefix = "ns" + staticNSCount++;
                element.setAttributeNS("http://www.w3.org/2000/xmlns/", "xmlns:" + valuePrefix, valueNsUri);
            }
            valuePrefix += ":";
        }
        else 
            valuePrefix = "";
        
        element.setAttributeNS(nameNsUri, namePrefix + name, valuePrefix + value);
    }

	public static String lookupNamespaceURI( Node node, String prefix )
	{
		if ( prefix != null && prefix.equals("xml") )
			return "http://www.w3.org/XML/1998/namespace";

		return node.lookupNamespaceURI( prefix );
	}

	public static String lookupPrefix( Node node, String uri )
	{
		if ( uri != null && uri.equals("http://www.w3.org/XML/1998/namespace") )
			return "xml";

		return node.lookupPrefix( uri );
	}

    public static String findUnusedPrefix(Node node, String prefixHint)
    {
        String pp = prefixHint;
        if (pp == null || pp.equals(javax.xml.XMLConstants.DEFAULT_NS_PREFIX))
            pp = "n";
        else
        {
            if (lookupNamespaceURI(node, pp) == null)
                return pp;
        }
        
        int n = 1;
        while (true)
        {
            String s = pp + String.valueOf(n);
            if (lookupNamespaceURI(node, s) == null)
                return s;
            ++n;
        }
    }
        
    public static Node dereference(Node node) throws Exception
    {
        if (node.getNodeType() == Node.ELEMENT_NODE && ((Element) node).hasAttribute("href"))
        {
            String href = ((Element) node).getAttribute("href");
            if (href.length() == 0 || href.charAt(0) != '#')
                throw new Exception("Cannot dereference external references.");
                
            href = href.substring(1);
            
            for (Node ref = node.getOwnerDocument().getDocumentElement().getFirstChild().getNextSibling(); ref != null; ref = ref.getNextSibling())
                if (ref.getNodeType() == Node.ELEMENT_NODE && ((Element)ref).getAttribute("id").equals(href))
                    return ref;
        }
        return node;
    }
    
    public static Node dereference12(Node node) throws Exception
    {
        if (node.getNodeType() == Node.ELEMENT_NODE && ((Element) node).hasAttributeNS("http://www.w3.org/2003/05/soap-encoding", "ref"))
        {
            String ref = ((Element) node).getAttributeNS("http://www.w3.org/2003/05/soap-encoding", "ref");
            if (ref.length() == 0)
                throw new Exception("Cannot dereference null references.");
            
            Node refNode = derefNode12(node.getOwnerDocument().getDocumentElement(), ref);
            if (refNode != null)
                return refNode;
        }
        return node;
    }
    
    private static Node derefNode12(Node node, String ref) throws Exception
    {
        if (node.getNodeType() == Node.ELEMENT_NODE && ((Element)node).getAttributeNS("http://www.w3.org/2003/05/soap-encoding", "id").equals(ref))
            return node;
        
        for (Node n = node.getFirstChild(); n!= null; n=n.getNextSibling())
        {
            Node refNode = derefNode12(n, ref);
            if (refNode != null)
                return refNode;
        }
        
        return null;
    }
    
    public static Element getNextElementSibling(Node node)
    {
        Node n = node.getNextSibling();
                    
        if (n == null || n.getNodeType() == Node.ELEMENT_NODE)
            return (Element) n;
        
        return getNextElementSibling(n);
    }
    
    public static Element getFirstElementChild(Node node)
    {
        for (Node n=node.getFirstChild(); n!=null; n=n.getNextSibling())
            if (n.getNodeType() == Node.ELEMENT_NODE)
                return (Element) n;
        return null;
    }
    
    public static Element findChildNode(Node node, String localName)
    {
        for (Element e = getFirstElementChild(node); e != null; e = getNextElementSibling(e))
        {
            if (e.getLocalName().equals(localName))
                return e;
        }
            
        return null;
    }
    
    public static Element appendElement(Node parent, String localName)
    {
        return appendElement(parent, null, localName);
    }
    
    public static Element appendElement(Node parent, String nsURI, String localName)
    {
        if (parent instanceof Document)
            return (Element) parent.appendChild(((Document) parent).createElementNS(nsURI, localName));
        return (Element) parent.appendChild(parent.getOwnerDocument().createElementNS(nsURI, localName));
    }

    private static String getTextContent(Node node)
    {
        short nodeType = node.getNodeType();
        if( nodeType == Node.ATTRIBUTE_NODE ||
            nodeType == Node.PROCESSING_INSTRUCTION_NODE ||
            nodeType == Node.COMMENT_NODE ||
            nodeType == Node.TEXT_NODE ||
            nodeType == Node.CDATA_SECTION_NODE )
        {
            return com.altova.CoreTypes.castToString(node.getNodeValue());
        }
        else
        if( nodeType == Node.DOCUMENT_NODE ||
            nodeType == Node.DOCUMENT_FRAGMENT_NODE ||
            nodeType == Node.ELEMENT_NODE ||
            nodeType == Node.ENTITY_REFERENCE_NODE )
        {
            String result = "";
            Node child = node.getFirstChild();
            for( ; child != null ; child = child.getNextSibling() )
            {
                short childNodeType = child.getNodeType();
                if( childNodeType == Node.TEXT_NODE ||
                    childNodeType == Node.CDATA_SECTION_NODE ||
                    childNodeType == Node.ELEMENT_NODE ||
                    childNodeType == Node.ENTITY_REFERENCE_NODE )
                {
                    result += getTextContent( child );
                }
            }
            return result;
        }
        else
            return "";
    }
	
	
	public static String getTextValue(Node node)
    {
        return getTextContent( node );
    }

	
	public static javax.xml.namespace.QName castToQName(Node node, MemberInfo member)
    {
        String value = getTextContent(node);
        int i = value.indexOf(':');
        if (i == -1)
            return new javax.xml.namespace.QName(lookupNamespaceURI(node, null), value);
        
        String prefix = value.substring(0, i);
        String local = value.substring(i+1);
        
        String uri = lookupNamespaceURI(node, prefix);
        
        return new javax.xml.namespace.QName(uri, local, prefix);
    }


}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy