//-------------------------------------------------------------------------------------------------------
// 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) {
    "use asm";

    var i4 = stdlib.SIMD.Int32x4;
    var i4check = i4.check;
    
    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 f4 = stdlib.SIMD.Float32x4;
    var f4check = f4.check;
    
    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 f4extractLane = f4.extractLane;

    var f4select = f4.select;

    
    var fround = stdlib.Math.fround;

    var globImportF4 = f4check(imports.g1);       // global var import
    var globImportI4 = i4check(imports.g2);       // global var import
    
    var g1 = f4(1.0, 2.0, 3.0, -0.0);          // global var initialized
    var g2 = f4(-5.3, -0.0, 7.332, 8.0);          // global var initialized
    var g3 = i4(1, 2, 3, 4);          // global var initialized
    var g4 = i4(5, 6, 7, 8);          // global var initialized
    
    var gval = 1234;
    var gval2 = 1234.0;

    var f4splat = f4.splat;

    var sqrt = stdlib.Math.sqrt;
    var pow = stdlib.Math.pow;

    var loopCOUNT = 3;

    function swizzle1() {
        var xyxy = f4(0.0, 0.0, 0.0, 0.0);
        var zwzw = f4(0.0, 0.0, 0.0, 0.0);
        var xxxx = f4(0.0, 0.0, 0.0, 0.0);
        var xxyy = f4(0.0, 0.0, 0.0, 0.0);

        var x = 0.0, y = 0.0, z = 0.0, w = 0.0;

        var loopIndex = 0;
        while ((loopIndex | 0) < (loopCOUNT | 0)) {

            xyxy = f4swizzle(g1, 0, 1, 0, 1);
            zwzw = f4swizzle(g1, 2, 3, 2, 3);
            xxxx = f4swizzle(g1, 0, 0, 0, 0);
            xxyy = f4swizzle(g1, 0, 0, 1, 1);

            loopIndex = (loopIndex + 1) | 0;
        }

        x = +sqrt((+fround(f4extractLane(xyxy, 0)) * +fround(f4extractLane(xyxy, 0))) 
                + (+fround(f4extractLane(xyxy, 1)) * +fround(f4extractLane(xyxy, 1))) 
                + (+fround(f4extractLane(xyxy, 2)) * +fround(f4extractLane(xyxy, 2))) 
                + (+fround(f4extractLane(xyxy, 3)) * +fround(f4extractLane(xyxy, 3))));
        y = +sqrt((+fround(f4extractLane(zwzw, 0)) * +fround(f4extractLane(zwzw, 0))) 
                + (+fround(f4extractLane(zwzw, 1)) * +fround(f4extractLane(zwzw, 1)))
                + (+fround(f4extractLane(zwzw, 2)) * +fround(f4extractLane(zwzw, 2)))
                + (+fround(f4extractLane(zwzw, 3)) * +fround(f4extractLane(zwzw, 3))));
        z = +sqrt((+fround(f4extractLane(xxxx, 0)) * +fround(f4extractLane(xxxx, 0)))
                + (+fround(f4extractLane(xxxx, 1)) * +fround(f4extractLane(xxxx, 1))) 
                + (+fround(f4extractLane(xxxx, 2)) * +fround(f4extractLane(xxxx, 2)))
                + (+fround(f4extractLane(xxxx, 3)) * +fround(f4extractLane(xxxx, 3))));
        w = +sqrt((+fround(f4extractLane(xxyy, 0)) * +fround(f4extractLane(xxyy, 0))) 
                + (+fround(f4extractLane(xxyy, 1)) * +fround(f4extractLane(xxyy, 1))) 
                + (+fround(f4extractLane(xxyy, 2)) * +fround(f4extractLane(xxyy, 2)))
                + (+fround(f4extractLane(xxyy, 3)) * +fround(f4extractLane(xxyy, 3))));


        return f4check(f4(fround(x), fround(y), fround(z), fround(w)));
    }

    function swizzle2() {
        var xyxy = f4(0.0, 0.0, 0.0, 0.0);
        var zwzw = f4(0.0, 0.0, 0.0, 0.0);
        var xxxx = f4(0.0, 0.0, 0.0, 0.0);
        var xxyy = f4(0.0, 0.0, 0.0, 0.0);
        var v1 = f4(122.0, -0.0, 334.0, -9500.231);
        var v2 = f4(102.0, -33.13, -1.0, 233.000001);
        var x = 0.0, y = 0.0, z = 0.0, w = 0.0;

        var loopIndex = 0;
        for (loopIndex = 0; (loopIndex | 0) < (loopCOUNT | 0) ; loopIndex = (loopIndex + 1) | 0) {
            xyxy = f4swizzle(f4add(v1, v2), 0, 1, 0, 1);
            zwzw = f4swizzle(f4mul(v1, v2), 2, 3, 2, 3);
            xxxx = f4swizzle(f4sub(v1, v2), 0, 0, 0, 0);
            xxyy = f4swizzle(v2, 0, 0, 1, 1);
        }

        x = +sqrt((+fround(f4extractLane(xyxy, 0)) * +fround(f4extractLane(xyxy, 0)))
                + (+fround(f4extractLane(xyxy, 1)) * +fround(f4extractLane(xyxy, 1))) 
                + (+fround(f4extractLane(xyxy, 2)) * +fround(f4extractLane(xyxy, 2))) 
                + (+fround(f4extractLane(xyxy, 3)) * +fround(f4extractLane(xyxy, 3))));
        y = +sqrt((+fround(f4extractLane(zwzw, 0)) * +fround(f4extractLane(zwzw, 0))) 
                + (+fround(f4extractLane(zwzw, 1)) * +fround(f4extractLane(zwzw, 1)))
                + (+fround(f4extractLane(zwzw, 2)) * +fround(f4extractLane(zwzw, 2)))
                + (+fround(f4extractLane(zwzw, 3)) * +fround(f4extractLane(zwzw, 3))));
        z = +sqrt((+fround(f4extractLane(xxxx, 0)) * +fround(f4extractLane(xxxx, 0))) 
                + (+fround(f4extractLane(xxxx, 1)) * +fround(f4extractLane(xxxx, 1))) 
                + (+fround(f4extractLane(xxxx, 2)) * +fround(f4extractLane(xxxx, 2))) 
                + (+fround(f4extractLane(xxxx, 3)) * +fround(f4extractLane(xxxx, 3))));
        w = +sqrt((+fround(f4extractLane(xxyy, 0)) * +fround(f4extractLane(xxyy, 0))) 
                + (+fround(f4extractLane(xxyy, 1)) * +fround(f4extractLane(xxyy, 1))) 
                + (+fround(f4extractLane(xxyy, 2)) * +fround(f4extractLane(xxyy, 2)))
                + (+fround(f4extractLane(xxyy, 3)) * +fround(f4extractLane(xxyy, 3))));

        return f4check(f4(fround(x), fround(y), fround(z), fround(w)));
    }

    function swizzle3() {
        var xyxy = f4(0.0, 0.0, 0.0, 0.0);
        var zwzw = f4(0.0, 0.0, 0.0, 0.0);
        var xxxx = f4(0.0, 0.0, 0.0, 0.0);
        var xxyy = f4(0.0, 0.0, 0.0, 0.0);

        var x = 0.0, y = 0.0, z = 0.0, w = 0.0;
        var v1 = f4(122.0, -0.0, 334.0, -9500.231);
        var loopIndex = 0;
        loopIndex = loopCOUNT | 0;
        do {

            xyxy = f4swizzle(f4add(v1, g2), 0, 1, 0, 1);
            zwzw = f4swizzle(f4mul(v1, g2), 2, 3, 2, 3);
            xxxx = f4swizzle(f4sub(v1, g2), 0, 0, 0, 0);
            xxyy = f4swizzle(g2, 0, 0, 1, 1);

            loopIndex = (loopIndex - 1) | 0;
        }
        while ((loopIndex | 0) > 0);

        x = +sqrt((+fround(f4extractLane(xyxy, 0)) * +fround(f4extractLane(xyxy, 0)))
                + (+fround(f4extractLane(xyxy, 1)) * +fround(f4extractLane(xyxy, 1)))
                + (+fround(f4extractLane(xyxy, 2)) * +fround(f4extractLane(xyxy, 2)))
                + (+fround(f4extractLane(xyxy, 3)) * +fround(f4extractLane(xyxy, 3))));
        y = +sqrt((+fround(f4extractLane(zwzw, 0)) * +fround(f4extractLane(zwzw, 0)))
                + (+fround(f4extractLane(zwzw, 1)) * +fround(f4extractLane(zwzw, 1)))
                + (+fround(f4extractLane(zwzw, 2)) * +fround(f4extractLane(zwzw, 2)))
                + (+fround(f4extractLane(zwzw, 3)) * +fround(f4extractLane(zwzw, 3))));
        z = +sqrt((+fround(f4extractLane(xxxx, 0)) * +fround(f4extractLane(xxxx, 0)))
                + (+fround(f4extractLane(xxxx, 1)) * +fround(f4extractLane(xxxx, 1)))
                + (+fround(f4extractLane(xxxx, 2)) * +fround(f4extractLane(xxxx, 2)))
                + (+fround(f4extractLane(xxxx, 3)) * +fround(f4extractLane(xxxx, 3))));
        w = +sqrt((+fround(f4extractLane(xxyy, 0)) * +fround(f4extractLane(xxyy, 0)))
                + (+fround(f4extractLane(xxyy, 1)) * +fround(f4extractLane(xxyy, 1)))
                + (+fround(f4extractLane(xxyy, 2)) * +fround(f4extractLane(xxyy, 2)))
                + (+fround(f4extractLane(xxyy, 3)) * +fround(f4extractLane(xxyy, 3))));

        return f4check(f4(fround(x), fround(y), fround(z), fround(w)));
    }

    return { func1: swizzle1, func2: swizzle2, func3: swizzle3 };
}


var m = asmModule(this, { g1: SIMD.Float32x4(9, 9, 9, 9), g2: SIMD.Int32x4(1, 2, 3, 4)});

var ret1 = m.func1();
var ret2 = m.func2();
var ret3 = m.func3();


equalSimd([3.1622776985168457, 4.242640495300293, 2, 3.1622776985168457], ret1, SIMD.Float32x4, "Test Swizzle");
equalSimd([320.22991943359375, 3130438, 40, 151.6680450439453], ret2, SIMD.Float32x4, "Test Swizzle");
equalSimd([165.03871154785156, 107538.6328125, 254.60000610351562, 7.49533224105835], ret3, SIMD.Float32x4, "Test Swizzle");

print("PASS");