- /*
- * The Apache Software License, Version 1.1
- *
- *
- * Copyright (c) 1999 The Apache Software Foundation. All rights
- * reserved.
- *
- * Redistribution and use in source and binary forms, with or without
- * modification, are permitted provided that the following conditions
- * are met:
- *
- * 1. Redistributions of source code must retain the above copyright
- * notice, this list of conditions and the following disclaimer.
- *
- * 2. Redistributions in binary form must reproduce the above copyright
- * notice, this list of conditions and the following disclaimer in
- * the documentation and/or other materials provided with the
- * distribution.
- *
- * 3. The end-user documentation included with the redistribution,
- * if any, must include the following acknowledgment:
- * "This product includes software developed by the
- * Apache Software Foundation (http://www.apache.org/)."
- * Alternately, this acknowledgment may appear in the software itself,
- * if and wherever such third-party acknowledgments normally appear.
- *
- * 4. The names "Xalan" and "Apache Software Foundation" must
- * not be used to endorse or promote products derived from this
- * software without prior written permission. For written
- * permission, please contact apache@apache.org.
- *
- * 5. Products derived from this software may not be called "Apache",
- * nor may "Apache" appear in their name, without prior written
- * permission of the Apache Software Foundation.
- *
- * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED
- * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
- * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
- * DISCLAIMED. IN NO EVENT SHALL THE APACHE SOFTWARE FOUNDATION OR
- * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
- * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
- * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
- * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
- * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
- * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
- * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
- * SUCH DAMAGE.
- * ====================================================================
- *
- * This software consists of voluntary contributions made by many
- * individuals on behalf of the Apache Software Foundation and was
- * originally based on software copyright (c) 1999, Lotus
- * Development Corporation., http://www.lotus.com. For more
- * information on the Apache Software Foundation, please see
- * <http://www.apache.org/>.
- */
- package org.apache.xalan.templates;
-
- import java.util.Vector;
- import java.util.Hashtable;
-
- import org.apache.xpath.functions.Function;
- import org.apache.xpath.functions.Function2Args;
- import org.apache.xpath.XPath;
- import org.apache.xpath.objects.XObject;
- import org.apache.xpath.objects.XNodeSet;
- import org.apache.xpath.XPathContext;
- import org.apache.xpath.axes.LocPathIterator;
- import org.apache.xpath.axes.UnionPathIterator;
- import org.apache.xml.utils.QName;
- import org.apache.xalan.transformer.TransformerImpl;
- import org.apache.xalan.transformer.KeyManager;
- import org.apache.xpath.res.XPATHErrorResources;
- import org.apache.xpath.XPathContext;
-
- import org.apache.xml.dtm.DTM;
- import org.apache.xml.dtm.DTMIterator;
-
- import org.apache.xml.utils.XMLString;
-
- /**
- * <meta name="usage" content="advanced"/>
- * Execute the Key() function.
- */
- public class FuncKey extends Function2Args
- {
-
- /** Dummy value to be used in usedrefs hashtable */
- static private Boolean ISTRUE = new Boolean(true);
-
- /**
- * Execute the function. The function must return
- * a valid object.
- * @param xctxt The current execution context.
- * @return A valid XObject.
- *
- * @throws javax.xml.transform.TransformerException
- */
- public XObject execute(XPathContext xctxt) throws javax.xml.transform.TransformerException
- {
-
- // TransformerImpl transformer = (TransformerImpl)xctxt;
- TransformerImpl transformer = (TransformerImpl) xctxt.getOwnerObject();
- XNodeSet nodes = null;
- int context = xctxt.getCurrentNode();
- DTM dtm = xctxt.getDTM(context);
- int docContext = dtm.getDocumentRoot(context);
-
- if (DTM.NULL == docContext)
- {
-
- // path.error(context, XPATHErrorResources.ER_CONTEXT_HAS_NO_OWNERDOC); //"context does not have an owner document!");
- }
-
- String xkeyname = getArg0().execute(xctxt).str();
- QName keyname = new QName(xkeyname, xctxt.getNamespaceContext());
- XObject arg = getArg1().execute(xctxt);
- boolean argIsNodeSetDTM = (XObject.CLASS_NODESET == arg.getType());
- KeyManager kmgr = transformer.getKeyManager();
-
- // Don't bother with nodeset logic if the thing is only one node.
- if(argIsNodeSetDTM)
- {
- XNodeSet ns = (XNodeSet)arg;
- ns.setShouldCacheNodes(true);
- int len = ns.getLength();
- if(len <= 1)
- argIsNodeSetDTM = false;
- }
-
- if (argIsNodeSetDTM)
- {
- Hashtable usedrefs = null;
- DTMIterator ni = arg.iter();
- int pos;
- UnionPathIterator upi = new UnionPathIterator();
- upi.exprSetParent(this);
-
- while (DTM.NULL != (pos = ni.nextNode()))
- {
- dtm = xctxt.getDTM(pos);
- XMLString ref = dtm.getStringValue(pos);
-
- if (null == ref)
- continue;
-
- if (null == usedrefs)
- usedrefs = new Hashtable();
-
- if (usedrefs.get(ref) != null)
- {
- continue; // We already have 'em.
- }
- else
- {
-
- // ISTRUE being used as a dummy value.
- usedrefs.put(ref, ISTRUE);
- }
-
- XNodeSet nl =
- kmgr.getNodeSetDTMByKey(xctxt, docContext, keyname, ref,
- xctxt.getNamespaceContext());
-
- nl.setRoot(xctxt.getCurrentNode(), xctxt);
-
- // try
- // {
- upi.addIterator(nl);
- // }
- // catch(CloneNotSupportedException cnse)
- // {
- // // will never happen.
- // }
- //mnodeset.addNodesInDocOrder(nl, xctxt); needed??
- }
-
- int current = xctxt.getCurrentNode();
- upi.setRoot(current, xctxt);
-
- nodes = new XNodeSet(upi);
- }
- else
- {
- XMLString ref = arg.xstr();
- nodes = kmgr.getNodeSetDTMByKey(xctxt, docContext, keyname,
- ref,
- xctxt.getNamespaceContext());
- nodes.setRoot(xctxt.getCurrentNode(), xctxt);
- }
-
- return nodes;
- }
- }