jaxp/src/com/sun/org/apache/xalan/internal/xsltc/compiler/WithParam.java

*** 1,8 ****
  /*
!  * reserved comment block
!  * DO NOT REMOVE OR ALTER!
   */
  /*
   * Copyright 2001-2004 The Apache Software Foundation.
   *
   * Licensed under the Apache License, Version 2.0 (the "License");
--- 1,7 ----
  /*
!  * Copyright (c) 2016, Oracle and/or its affiliates. All rights reserved.
   */
  /*
   * Copyright 2001-2004 The Apache Software Foundation.
   *
   * Licensed under the Apache License, Version 2.0 (the "License");
*** 15,33 ****
   * 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.
   */
- /*
-  * $Id: WithParam.java,v 1.2.4.1 2005/09/12 11:38:01 pvedula Exp $
-  */
  
  package com.sun.org.apache.xalan.internal.xsltc.compiler;
  
  import com.sun.org.apache.bcel.internal.generic.ConstantPoolGen;
  import com.sun.org.apache.bcel.internal.generic.INVOKEVIRTUAL;
  import com.sun.org.apache.bcel.internal.generic.InstructionList;
  import com.sun.org.apache.bcel.internal.generic.PUSH;
  import com.sun.org.apache.xalan.internal.xsltc.compiler.util.ClassGenerator;
  import com.sun.org.apache.xalan.internal.xsltc.compiler.util.ErrorMsg;
  import com.sun.org.apache.xalan.internal.xsltc.compiler.util.MethodGenerator;
  import com.sun.org.apache.xalan.internal.xsltc.compiler.util.ReferenceType;
--- 14,34 ----
   * 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.org.apache.xalan.internal.xsltc.compiler;
  
+ import com.sun.org.apache.bcel.internal.generic.ALOAD;
+ import com.sun.org.apache.bcel.internal.generic.ASTORE;
+ import com.sun.org.apache.bcel.internal.generic.CHECKCAST;
  import com.sun.org.apache.bcel.internal.generic.ConstantPoolGen;
+ import com.sun.org.apache.bcel.internal.generic.INVOKEINTERFACE;
  import com.sun.org.apache.bcel.internal.generic.INVOKEVIRTUAL;
  import com.sun.org.apache.bcel.internal.generic.InstructionList;
+ import com.sun.org.apache.bcel.internal.generic.LocalVariableGen;
  import com.sun.org.apache.bcel.internal.generic.PUSH;
  import com.sun.org.apache.xalan.internal.xsltc.compiler.util.ClassGenerator;
  import com.sun.org.apache.xalan.internal.xsltc.compiler.util.ErrorMsg;
  import com.sun.org.apache.xalan.internal.xsltc.compiler.util.MethodGenerator;
  import com.sun.org.apache.xalan.internal.xsltc.compiler.util.ReferenceType;
*** 58,67 ****
--- 59,73 ----
       * Parameter's default value.
       */
      private Expression _select;
  
      /**
+      * Reference to JVM variable holding temporary result tree.
+      */
+     private LocalVariableGen _domAdapter;
+ 
+     /**
       * %OPT% This is set to true when the WithParam is used in a CallTemplate
       * for a simple named template. If this is true, the parameters are
       * passed to the named template through method arguments rather than
       * using the expensive Translet.addParameter() call.
       */
*** 162,173 ****
--- 168,184 ----
          if (_select != null) {
              _select.translate(classGen, methodGen);
              _select.startIterator(classGen, methodGen);
          }
          // If not, compile result tree from parameter body if present.
+         // Store result tree into local variable for releasing it later
          else if (hasContents()) {
+             final InstructionList il = methodGen.getInstructionList();
              compileResultTree(classGen, methodGen);
+             _domAdapter = methodGen.addLocalVariable2("@" + _escapedName, Type.ResultTree.toJCType(), il.getEnd());
+             il.append(DUP);
+             il.append(new ASTORE(_domAdapter.getIndex()));
          }
          // If neither are present then store empty string in parameter slot
          else {
              final ConstantPoolGen cpg = classGen.getConstantPool();
              final InstructionList il = methodGen.getInstructionList();
*** 206,211 ****
--- 217,244 ----
          il.append(new INVOKEVIRTUAL(cpg.addMethodref(TRANSLET_CLASS,
                                                       ADD_PARAMETER,
                                                       ADD_PARAMETER_SIG)));
          il.append(POP); // cleanup stack
      }
+ 
+     /**
+      * Release the compiled result tree.
+      */
+     public void releaseResultTree(ClassGenerator classGen, MethodGenerator methodGen) {
+         if (_domAdapter != null) {
+             final ConstantPoolGen cpg = classGen.getConstantPool();
+             final InstructionList il = methodGen.getInstructionList();
+             if (classGen.getStylesheet().callsNodeset() && classGen.getDOMClass().equals(MULTI_DOM_CLASS)) {
+                 final int removeDA = cpg.addMethodref(MULTI_DOM_CLASS, "removeDOMAdapter", "(" + DOM_ADAPTER_SIG + ")V");
+                 il.append(methodGen.loadDOM());
+                 il.append(new CHECKCAST(cpg.addClass(MULTI_DOM_CLASS)));
+                 il.append(new ALOAD(_domAdapter.getIndex()));
+                 il.append(new CHECKCAST(cpg.addClass(DOM_ADAPTER_CLASS)));
+                 il.append(new INVOKEVIRTUAL(removeDA));
+             }
+             final int release = cpg.addInterfaceMethodref(DOM_IMPL_CLASS, "release", "()V");
+             il.append(new ALOAD(_domAdapter.getIndex()));
+             il.append(new INVOKEINTERFACE(release, 1));
+             _domAdapter = null;
+          }
+      }
  }