//-------------------------------------------------------------------------------------------------------
// Copyright (C) Microsoft Corporation and contributors. All rights reserved.
// Licensed under the MIT license. See LICENSE.txt file in the project root for full license information.
//-------------------------------------------------------------------------------------------------------


this.WScript.LoadScriptFile("..\\UnitTestFramework\\SimdJsHelpers.js");

function asmModule(stdlib, imports, buffer) {
    "use asm";

    var i4 = stdlib.SIMD.Int32x4;
    var i4check = i4.check;
    var i4splat = i4.splat;
    
    var i4fromFloat32x4 = i4.fromFloat32x4;
    var i4fromFloat32x4Bits = i4.fromFloat32x4Bits;
    //var i4abs = i4.abs;
    var i4neg = i4.neg;
    var i4add = i4.add;
    var i4sub = i4.sub;
    var i4mul = i4.mul;
    var i4swizzle = i4.swizzle;
    var i4shuffle = i4.shuffle;
    var i4lessThan = i4.lessThan;
    var i4equal = i4.equal;
    var i4greaterThan = i4.greaterThan;
    var i4select = i4.select;
    var i4and = i4.and;
    var i4or = i4.or;
    var i4xor = i4.xor;
    var i4not = i4.not;
    var i4load  = i4.load;
    var i4load1 = i4.load1;
    var i4load2 = i4.load2;
    var i4load3 = i4.load3;

    var i4store  = i4.store
    var i4store1 = i4.store1;
    var i4store2 = i4.store2;
    var i4store3 = i4.store3;

    //var i4shiftLeftByScalar = i4.shiftLeftByScalar;
    //var i4shiftRightByScalar = i4.shiftRightByScalar;
    //var i4shiftRightArithmeticByScalar = i4.shiftRightArithmeticByScalar;
    var f4 = stdlib.SIMD.Float32x4;
    var f4check = f4.check;
    var f4splat = f4.splat;
    
    var f4fromInt32x4 = f4.fromInt32x4;
    var f4fromInt32x4Bits = f4.fromInt32x4Bits;
    var f4abs = f4.abs;
    var f4neg = f4.neg;
    var f4add = f4.add;
    var f4sub = f4.sub;
    var f4mul = f4.mul;
    var f4div = f4.div;
    
    var f4min = f4.min;
    var f4max = f4.max;
    
    var f4sqrt = f4.sqrt;
    var f4swizzle = f4.swizzle;
    var f4shuffle = f4.shuffle;
    var f4lessThan = f4.lessThan;
    var f4lessThanOrEqual = f4.lessThanOrEqual;
    var f4equal = f4.equal;
    var f4notEqual = f4.notEqual;
    var f4greaterThan = f4.greaterThan;
    var f4greaterThanOrEqual = f4.greaterThanOrEqual;

    var f4select = f4.select;

    var f4load = f4.load;
    var f4load1 = f4.load1;
    var f4load2 = f4.load2;
    var f4load3 = f4.load3;

    var f4store  = f4.store;
    var f4store1 = f4.store1;
    var f4store2 = f4.store2;
    var f4store3 = f4.store3;

    
    var fround = stdlib.Math.fround;

    var globImportF4 = f4check(imports.g1);       // global var import
    var globImportI4 = i4check(imports.g2);       // global var import
    
    var g1 = f4(-5033.2,-3401.0,665.34,32234.1);          // global var initialized
    var g2 = i4(1065353216, -1073741824, -1077936128, 1082130432);          // global var initialized
    
    var gval = 1234;
    var gval2 = 1234.0;

    var OFFSET_1 = 10;
    var OFFSET_2 = 15;

    var loopCOUNT = 10;

    var Int8Heap = new stdlib.Int8Array (buffer);
    var Uint8Heap = new stdlib.Uint8Array (buffer);

    var Int16Heap = new stdlib.Int16Array(buffer);
    var Uint16Heap = new stdlib.Uint16Array(buffer);
    var Int32Heap = new stdlib.Int32Array(buffer);
    var Uint32Heap = new stdlib.Uint32Array(buffer);
    var Float32Heap = new stdlib.Float32Array(buffer);

    function nestedLoadStoreUint8(idx)
    {
         idx = idx|0;
         idx = idx<<0;
         f4store(Uint8Heap, (idx + 5)|0, f4(9.0,8.0,10.1, -22.121));
         //return f4load(Uint8Heap, (idx + 5)|0);
         f4store(Uint8Heap, (idx + 0)|0, f4load(Uint8Heap, (idx + 5)|0));
         return f4load(Uint8Heap, (idx + 0)|0);
    }

    function scale(fromIdx, toIdx)
    {
        fromIdx = fromIdx | 0;
        toIdx = toIdx | 0;
        var i = 0;
        var xx = f4(0.0, 0.0, 0.0, 0.0);

        //No bound check to enable negative tests.  
         for (i = fromIdx; (i|0) < (toIdx|0); i = (i + 16) | 0)
         {
             xx = f4load(Int8Heap, i | 0);
             xx = f4mul(xx, f4(2.0, 2.0, 2.0, 2.0));
             
             f4store(Int8Heap, toIdx, xx);
             f4store(Int8Heap, (i + 0) | 0, f4load(Int8Heap, (toIdx + 0) | 0));
             //f4store(Int8Heap, (i) | 0, f4load(Int8Heap, (toIdx) | 0));
         }
    }
    function scale2(fromIdx, toIdx)
    {
        fromIdx = fromIdx | 0;
        toIdx = toIdx | 0;
        var i = 0;
        var xx = i4(0, 0, 0, 0);

        //No bound check to enable negative tests.  
         for (i = fromIdx;(i|0) < (toIdx|0); i = (i + 16) | 0)
         {
             xx = i4load(Int8Heap, i | 0);
             xx = i4mul(xx, i4(2, 2, 2, 2));
             
             i4store(Int8Heap, toIdx, xx);
             i4store(Int8Heap, (i + 0) | 0, i4load(Int8Heap, (toIdx + 0) | 0));
             //f4store(Int8Heap, (i) | 0, f4load(Int8Heap, (toIdx) | 0));
         }
    }
    /*
    function scale3(fromIdx, toIdx)
    {
        fromIdx = fromIdx | 0;
        toIdx = toIdx | 0;
        var i = 0;
        var xx = d2(0.0, 0.0);

        //No bound check to enable negative tests.  
         for (i = fromIdx; (i|0) < (toIdx|0); i = (i + 16) | 0)
         {
             xx = d2load(Int8Heap, i | 0);
             xx = d2mul(xx, d2(2.0, 2.0));
             
             d2store(Int8Heap, toIdx, xx);
             d2store(Int8Heap, (i + 0) | 0, d2load(Int8Heap, (toIdx + 0) | 0));
             //f4store(Int8Heap, (i) | 0, f4load(Int8Heap, (toIdx) | 0));
         }
    }
*/
    return {scale: scale
            ,scale2: scale2
            /*,scale3, scale3*/
            ,nestedLoadStoreUint8:nestedLoadStoreUint8};
}

var buffer = new ArrayBuffer(0x10000); //16mb min 2^12

//Reset or flush the buffer
function initF32(buffer) {
    var values = new Float32Array( buffer );
    for( var i=0; i < values.length ; ++i ) {
        values[i] = i * 10;
    }
    return values.length;
}
function initI32(buffer) {
    var values = new Int32Array( buffer );
    for( var i=0; i < values.length ; ++i ) {
        values[i] = i * 10;
    }
    return values.length;
}
function validateBuffer(buffer, count)
{
    var f4;
    var data = [
    SIMD.Float32x4(0.0,20.0,40.0,60.0),
    SIMD.Float32x4(80.0,100.0,120.0,140.0),
    SIMD.Float32x4(160.0,180.0,200.0,220.0),
    SIMD.Float32x4(240.0,260.0,280.0,300.0),
    SIMD.Float32x4(320.0,340.0,360.0,380.0),
    SIMD.Float32x4(400.0,420.0,440.0,460.0),
    SIMD.Float32x4(480.0,500.0,520.0,540.0),
    SIMD.Float32x4(560.0,580.0,600.0,620.0),
    SIMD.Float32x4(640.0,660.0,680.0,700.0),
    SIMD.Float32x4(720.0,740.0,760.0,780.0)
    ];
    for (var i = 0; i < count; i += 4)
    {
        f4 = SIMD.Float32x4.load(buffer, i);

        equalSimd(data[i/4], f4, SIMD.Float32x4, "validateBuffer");

    }
}
function validateBuffer2(buffer, count)
{
    var i4;
    var data = [
        SIMD.Int32x4(0,20,40,60),
        SIMD.Int32x4(80,100,120,140),
        SIMD.Int32x4(160,180,200,220),
        SIMD.Int32x4(240,260,280,300),
        SIMD.Int32x4(320,340,360,380),
        SIMD.Int32x4(400,420,440,460),
        SIMD.Int32x4(480,500,520,540),
        SIMD.Int32x4(560,580,600,620),
        SIMD.Int32x4(640,660,680,700),
        SIMD.Int32x4(720,740,760,780)
    ];

    for (var i = 0; i < count; i += 4)
    {
        i4 = SIMD.Int32x4.load(buffer, i);
        equalSimd(data[i/4], i4, SIMD.Int32x4, "validateBuffer2");
    }
}

function printResults(res)
{
    print(typeof(res));
    print(res.toString());
}

inputLength = initF32(buffer);

//Module initialization
var m = asmModule(this, {g0:initF32(buffer),g1:SIMD.Float32x4(9,9,9,9), g2:SIMD.Int32x4(1, 2, 3, 4)}, buffer);
var values = new Float32Array(buffer);

m.scale(0, 16 * 10); //Scale
validateBuffer(values, 4 * 10);

initI32(buffer);

m.scale2(0, 16 * 10); //Scale
validateBuffer2(values, 4 * 10);

print("PASS");