View Javadoc

1   /*
2    * Licensed to the Apache Software Foundation (ASF) under one or more
3    * contributor license agreements.  See the NOTICE file distributed with
4    * this work for additional information regarding copyright ownership.
5    * The ASF licenses this file to You under the Apache License, Version 2.0
6    * (the "License"); you may not use this file except in compliance with
7    * the License.  You may obtain a copy of the License at
8    *
9    *      http://www.apache.org/licenses/LICENSE-2.0
10   *
11   * Unless required by applicable law or agreed to in writing, software
12   * distributed under the License is distributed on an "AS IS" BASIS,
13   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14   * See the License for the specific language governing permissions and
15   * limitations under the License.
16   */
17  package org.apache.commons.nabla.forward.arithmetic;
18  
19  import org.apache.commons.nabla.DifferentiationException;
20  import org.apache.commons.nabla.forward.analysis.InstructionsTransformer;
21  import org.apache.commons.nabla.forward.analysis.MethodDifferentiator;
22  import org.objectweb.asm.Opcodes;
23  import org.objectweb.asm.Type;
24  import org.objectweb.asm.tree.AbstractInsnNode;
25  import org.objectweb.asm.tree.InsnList;
26  import org.objectweb.asm.tree.InsnNode;
27  import org.objectweb.asm.tree.MethodInsnNode;
28  
29  /** Differentiation transformer for DMUL instructions.
30   * @version $Id$
31   */
32  public class DMulTransformer implements InstructionsTransformer {
33  
34      /** Name of the {@link org.apache.commons.math3.analysis.differentiation.DerivativeStructure} method corresponding to the multiplication. */
35      private static final String MULTIPLY_METHOD = "multiply";
36  
37      /** Indicator for top stack element conversion. */
38      private final boolean stack0Converted;
39  
40      /** Indicator for next to top stack element conversion. */
41      private final boolean stack1Converted;
42  
43      /** Simple constructor.
44       * @param stack0Converted if true, the top level stack element has already been converted
45       * @param stack1Converted if true, the next to top level stack element has already been converted
46       */
47      public DMulTransformer(final boolean stack0Converted, final boolean stack1Converted) {
48          this.stack0Converted = stack0Converted;
49          this.stack1Converted = stack1Converted;
50      }
51  
52      /** {@inheritDoc} */
53      public InsnList getReplacement(final AbstractInsnNode insn,
54                                     final MethodDifferentiator methodDifferentiator)
55          throws DifferentiationException {
56  
57          final InsnList list = new InsnList();
58  
59          if (stack1Converted) {
60              if (stack0Converted) {
61                  list.add(new MethodInsnNode(Opcodes.INVOKEVIRTUAL, DS_TYPE.getInternalName(),
62                                              MULTIPLY_METHOD,
63                                              Type.getMethodDescriptor(DS_TYPE, DS_TYPE)));
64              } else {
65                  list.add(new MethodInsnNode(Opcodes.INVOKEVIRTUAL, DS_TYPE.getInternalName(),
66                                              MULTIPLY_METHOD,
67                                              Type.getMethodDescriptor(DS_TYPE, Type.DOUBLE_TYPE)));
68              }
69          } else {
70              list.add(new InsnNode(Opcodes.DUP_X2));
71              list.add(new InsnNode(Opcodes.POP));
72              list.add(new MethodInsnNode(Opcodes.INVOKEVIRTUAL, DS_TYPE.getInternalName(),
73                                          MULTIPLY_METHOD,
74                                          Type.getMethodDescriptor(DS_TYPE, Type.DOUBLE_TYPE)));
75          }
76  
77          return list;
78  
79      }
80  
81  }