/*

  * Licensed to the Apache Software Foundation (ASF) under one or more

  * contributor license agreements.  See the NOTICE file distributed with

  * this work for additional information regarding copyright ownership.

  * The ASF 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 org.apache.commons.nabla.automatic.functions;

 import org.apache.commons.nabla.automatic.analysis.MethodDifferentiator;

 import org.apache.commons.nabla.core.DifferentiationException;

 import org.objectweb.asm.Opcodes;

 import org.objectweb.asm.tree.InsnList;

 import org.objectweb.asm.tree.InsnNode;

 import org.objectweb.asm.tree.MethodInsnNode;

 import org.objectweb.asm.tree.VarInsnNode;

 /** Differentiation transformer for the atan2 function invocation instructions.

  */

 public class Atan2Transformer12 implements MathInvocationTransformer {

     /** {@inheritDoc} */

     public InsnList getReplacementList(final String owner, final MethodDifferentiator methodDifferentiator)

         throws DifferentiationException {

         final int tmp1 = methodDifferentiator.getTmp(1);

         final int tmp2 = methodDifferentiator.getTmp(2);

         final int tmp3 = methodDifferentiator.getTmp(3);

         final int tmp4 = methodDifferentiator.getTmp(4);

         // generate differential code

         // ... y0, y1, x0, x1  --> ...  atan2(y0, x0), (x0*y1x0*y1-x1*y00)/(x0^2+y0^2)

         final InsnList list = new InsnList();

         list.add(new VarInsnNode(Opcodes.DSTORE, tmp4)); // => y0, y1, x0

         list.add(new VarInsnNode(Opcodes.DSTORE, tmp3)); // => y0, y1

         list.add(new VarInsnNode(Opcodes.DSTORE, tmp2)); // => y0

         list.add(new VarInsnNode(Opcodes.DSTORE, tmp1)); // =>

         list.add(new VarInsnNode(Opcodes.DLOAD,  tmp1)); // => y0

         list.add(new VarInsnNode(Opcodes.DLOAD,  tmp3)); // => y0, x0

         list.add(new MethodInsnNode(Opcodes.INVOKESTATIC, owner, "atan2", DD_RETURN_D_DESCRIPTOR)); // => atan2(y0,x0)

         list.add(new VarInsnNode(Opcodes.DLOAD,  tmp3)); // => atan2(y0,x0), x0

         list.add(new VarInsnNode(Opcodes.DLOAD,  tmp2)); // => atan2(y0,x0), x0, y1

         list.add(new InsnNode(Opcodes.DMUL));            // => atan2(y0,x0), x0*y1

         list.add(new VarInsnNode(Opcodes.DLOAD,  tmp4)); // => atan2(y0,x0), x0*y1, x1

         list.add(new VarInsnNode(Opcodes.DLOAD,  tmp1)); // => atan2(y0,x0), x0*y1, x1, y0

         list.add(new InsnNode(Opcodes.DMUL));            // => atan2(y0,x0), x0*y1, x1*y0

         list.add(new InsnNode(Opcodes.DSUB));            // => atan2(y0,x0), x0*y1-x1*y0

         list.add(new VarInsnNode(Opcodes.DLOAD,  tmp3)); // => atan2(y0,x0), x0*y1-x1*y0, x0

         list.add(new InsnNode(Opcodes.DUP2));            // => atan2(y0,x0), x0*y1-x1*y0, x0, x0

         list.add(new InsnNode(Opcodes.DMUL));            // => atan2(y0,x0), x0*y1-x1*y0, x0^2

         list.add(new VarInsnNode(Opcodes.DLOAD,  tmp1)); // => atan2(y0,x0), x0*y1-x1*y0, x0^2, y0

         list.add(new InsnNode(Opcodes.DUP2));            // => atan2(y0,x0), x0*y1-x1*y0, x0^2, y0, y0

         list.add(new InsnNode(Opcodes.DMUL));            // => atan2(y0,x0), x0*y1-x1*y0, x0^2, y0^2

         list.add(new InsnNode(Opcodes.DADD));            // => atan2(y0,x0), x0*y1-x1*y0, x0^2+y0^2

         list.add(new InsnNode(Opcodes.DDIV));            // => atan2(y0,x0), (x0*y1-x1*y0)/(x0^2+y0^2)

         return list;

     }

 }