/*******************************************************************************
 * Copyright IBM Corp. and others 2016
 *
 * This program and the accompanying materials are made available under
 * the terms of the Eclipse Public License 2.0 which accompanies this
 * distribution and is available at https://www.eclipse.org/legal/epl-2.0/
 * or the Apache License, Version 2.0 which accompanies this distribution and
 * is available at https://www.apache.org/licenses/LICENSE-2.0.
 *
 * This Source Code may also be made available under the following
 * Secondary Licenses when the conditions for such availability set
 * forth in the Eclipse Public License, v. 2.0 are satisfied: GNU
 * General Public License, version 2 with the GNU Classpath
 * Exception [1] and GNU General Public License, version 2 with the
 * OpenJDK Assembly Exception [2].
 *
 * [1] https://www.gnu.org/software/classpath/license.html
 * [2] https://openjdk.org/legal/assembly-exception.html
 *
 * SPDX-License-Identifier: EPL-2.0 OR Apache-2.0 OR GPL-2.0-only WITH Classpath-exception-2.0 OR GPL-2.0-only WITH OpenJDK-assembly-exception-1.0
 *******************************************************************************/

#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <errno.h>

#include "NestedLoop.hpp"

NestedLoopMethod::NestedLoopMethod(OMR::JitBuilder::TypeDictionary *types)
   : OMR::JitBuilder::MethodBuilder(types)
   {
   DefineLine(LINETOSTR(__LINE__));
   DefineFile(__FILE__);

   DefineName("nested_loop");
   DefineParameter("n", Int32);
   DefineReturnType(Int32);
   }

bool
NestedLoopMethod::buildIL()
   {
   Store("x",
      ConstInt32(0));

   OMR::JitBuilder::IlBuilder *aLoop=NULL;
   ForLoopUp("a", &aLoop,
             ConstInt32(0),
             Load("n"),
             ConstInt32(1));

   OMR::JitBuilder::IlBuilder *bLoop=NULL;
   aLoop->ForLoopUp("b", &bLoop,
   aLoop->          ConstInt32(0),
   aLoop->          Load("n"),
   aLoop->          ConstInt32(1));

   OMR::JitBuilder::IlBuilder *cLoop=NULL;
   bLoop->ForLoopUp("c", &cLoop,
   bLoop->          ConstInt32(0),
   bLoop->          Load("n"),
   bLoop->          ConstInt32(1));

   OMR::JitBuilder::IlBuilder *dLoop=NULL;
   cLoop->ForLoopUp("d", &dLoop,
   cLoop->          ConstInt32(0),
   cLoop->          Load("n"),
   cLoop->          ConstInt32(1));

   OMR::JitBuilder::IlBuilder *eLoop=NULL;
   dLoop->ForLoopUp("e", &eLoop,
   dLoop->          ConstInt32(0),
   dLoop->          Load("n"),
   dLoop->          ConstInt32(1));

   OMR::JitBuilder::IlBuilder *fLoop=NULL;
   eLoop->ForLoopUp("f", &fLoop,
   eLoop->          ConstInt32(0),
   eLoop->          Load("n"),
   eLoop->          ConstInt32(1));

   fLoop->Store("x",
   fLoop->      Add(
   fLoop->         Load("x"),
   fLoop->         ConstInt32(1)));

   Return(
      Load("x"));

   return true;
   }


int
main(int argc, char *argv[])
   {
   printf("Step 1: initialize JIT\n");
   bool initialized = initializeJit();
   if (!initialized)
      {
      fprintf(stderr, "FAIL: could not initialize JIT\n");
      exit(-1);
      }

   printf("Step 2: define relevant types\n");
   OMR::JitBuilder::TypeDictionary types;

   printf("Step 3: compile method builder\n");
   NestedLoopMethod nestedLoop(&types);
   void *entry=0;
   int32_t rc = compileMethodBuilder(&nestedLoop, &entry);
   if (rc != 0)
      {
      fprintf(stderr,"FAIL: compilation error %d\n", rc);
      exit(-2);
      }

   printf("Step 4: invoke compiled code\n");
   NestedLoopFunctionType *nested_loop = (NestedLoopFunctionType *)entry;
   for (int32_t n=0;n < 20;n++)
      printf("nested_loop(%2d) = %d\n", n, nested_loop(n));

   printf ("Step 5: shutdown JIT\n");
   shutdownJit();

   printf("PASS\n");
   }
