/*
 * ScalaCL - putting Scala on the GPU with JavaCL / OpenCL
 * http://scalacl.googlecode.com/
 *
 * Copyright (c) 2009-2010, Olivier Chafik (http://ochafik.free.fr/)
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 *     * Redistributions of source code must retain the above copyright
 *       notice, this list of conditions and the following disclaimer.
 *     * 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.
 *     * Neither the name of Olivier Chafik nor the
 *       names of its contributors may be used to endorse or promote products
 *       derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY OLIVIER CHAFIK AND CONTRIBUTORS ``AS IS'' AND ANY
 * EXPRESS 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 REGENTS AND 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.
 */
package com.nativelibs4java.scalaxy ; package test
import plugin._

import java.io.File
import org.junit._
import Assert._
import Function.{tupled, untupled}

class ArrayMap2WhileTest extends ScalaxyTestUtils with TypeUtils {

  @Test
  def simplePrimitiveArrayTabulate =
    ensurePluginCompilesSnippetsToSameByteCode(primValues map tupled { simpleArrayTabulate })
  
  @Test
  def simpleRefArrayTabulate =
    ensurePluginCompilesSnippetsToSameByteCode(refValues map tupled { simpleArrayTabulate })
  
  def simpleArrayTabulate(typeStr: String, valStr: String) = {
    (
      """
          val length = 10
          val a = Array.tabulate(length)(i => """ + valStr + """)
          println(a)
      """,
      """
          val length = 10
          val a = {
            val n = length
            val m = Array.ofDim[""" + typeStr + """](n)
            var i = 0
            while (i < n)
            {
              m(i) = """ + valStr + """
              i += 1
            }
            m
          }
          println(a)
      """
    )
  }
  
  @Test
  def nestedRefArrayTabulate =
    ensurePluginCompilesSnippetsToSameByteCode(refValues map tupled { nestedArrayTabulate })

  @Test
  def nestedPrimitiveArrayTabulate =
    ensurePluginCompilesSnippetsToSameByteCode(primValues map tupled { nestedArrayTabulate })

  def nestedArrayTabulate(typeStr: String, valStr: String) = {
    (
      """
          val length = 10
          val a = Array.tabulate(length, length * 2)((i, j) => """ + valStr + """)
          println(a)
      """,
      """
          val length = 10
          val a = {
            val n1 = length
            val n2 = length * 2
            val multiArray = Array.ofDim[""" + typeStr + """](n1, n2)
            var i1 = 0
            while (i1 < n1)
            {
              val iVal1 = i1
              val subArray1 = multiArray(iVal1);
              {
                var i2 = 0
                while (i2 < n2)
                {
                  val iVal2 = i2
                  subArray1(i2) = """ + valStr + """
                  i2 += 1
                }
              }
              i1 += 1
            }
            multiArray
          }
          println(a)
      """
    )
  }

  @Test
  def simplePrimitiveArrayMap =
    ensurePluginCompilesSnippetsToSameByteCode(primTypeNames map simpleArrayMap)

  @Test
  def simpleRefArrayMap =
    ensurePluginCompilesSnippetsToSameByteCode(refTypeNames map simpleArrayMap)

  def simpleArrayMap(typeStr: String) = {
    (
      """
          val a = new Array[""" + typeStr + """](10)
          val m = a.map(_ + "...")
      """,
      """
          val a = new Array[""" + typeStr + """](10)
          val m = {
            val aa = a
            val n = aa.length
            var i = 0
            val mm = new Array[String](n)
            while (i < n)
            {
              val item = aa(i)
              val mapped = item + "..."
              mm(i) = mapped
              i += 1
            }
            mm
          }
      """
    )
  }
  
  @Test
  def inlinePrimitiveArrayMap =
    ensurePluginCompilesSnippetsToSameByteCode(primTypeNames map inlineArrayMap)

  @Test
  def inlineRefArrayMap =
    ensurePluginCompilesSnippetsToSameByteCode(refTypeNames map inlineArrayMap)

  def inlineArrayMap(typeStr: String) = {
    (
      """
          val m = new Array[""" + typeStr + """](10).map(_ + "...")
      """,
      """
          val m = {
            val aa = new Array[""" + typeStr + """](10)
            val n = aa.length
            var i = 0
            val mm = new Array[String](n)
            while (i < n)
            {
              val item = aa(i)
              val mapped = item + "..."
              mm(i) = mapped
              i += 1
            }
            mm
          }
      """
    )
  }
}
