/**
 ******************************************************************************
 * Xenia : Xbox 360 Emulator Research Project                                 *
 ******************************************************************************
 * Copyright 2014 Ben Vanik. All rights reserved.                             *
 * Released under the BSD license - see LICENSE in the root for more details. *
 ******************************************************************************
 */

#include <tools/alloy-test/util.h>

using namespace alloy;
using namespace alloy::hir;
using namespace alloy::runtime;
using namespace alloy::test;
using alloy::frontend::ppc::PPCContext;

TEST_CASE("SHA_I8", "[instr]") {
  TestFunction test([](hir::HIRBuilder& b) {
    StoreGPR(b, 3, b.ZeroExtend(b.Sha(b.Truncate(LoadGPR(b, 4), INT8_TYPE),
                                      b.Truncate(LoadGPR(b, 5), INT8_TYPE)),
                                INT64_TYPE));
    b.Return();
  });
  test.Run([](PPCContext* ctx) {
             ctx->r[4] = 0xF0;
             ctx->r[5] = 4;
           },
           [](PPCContext* ctx) {
             auto result = static_cast<uint8_t>(ctx->r[3]);
             REQUIRE(result == 0xFF);
           });
  test.Run([](PPCContext* ctx) {
             ctx->r[4] = 0xFF;
             ctx->r[5] = 0;
           },
           [](PPCContext* ctx) {
             auto result = static_cast<uint8_t>(ctx->r[3]);
             REQUIRE(result == 0xFF);
           });
  test.Run([](PPCContext* ctx) {
             ctx->r[4] = 0xFF;
             ctx->r[5] = 1;
           },
           [](PPCContext* ctx) {
             auto result = static_cast<uint8_t>(ctx->r[3]);
             REQUIRE(result == 0xFF);
           });
  test.Run([](PPCContext* ctx) {
             ctx->r[4] = 0x80;
             ctx->r[5] = 8;
           },
           [](PPCContext* ctx) {
             auto result = static_cast<uint8_t>(ctx->r[3]);
             REQUIRE(result == 0xFF);
           });
  test.Run([](PPCContext* ctx) {
             ctx->r[4] = 0x7F;
             ctx->r[5] = 7;
           },
           [](PPCContext* ctx) {
             auto result = static_cast<uint8_t>(ctx->r[3]);
             REQUIRE(result == 0);
           });
}

TEST_CASE("SHA_I16", "[instr]") {
  TestFunction test([](hir::HIRBuilder& b) {
    StoreGPR(b, 3, b.ZeroExtend(b.Sha(b.Truncate(LoadGPR(b, 4), INT16_TYPE),
                                      b.Truncate(LoadGPR(b, 5), INT8_TYPE)),
                                INT64_TYPE));
    b.Return();
  });
  test.Run([](PPCContext* ctx) {
             ctx->r[4] = 0xFF00;
             ctx->r[5] = 8;
           },
           [](PPCContext* ctx) {
             auto result = static_cast<uint16_t>(ctx->r[3]);
             REQUIRE(result == 0xFFFF);
           });
  test.Run([](PPCContext* ctx) {
             ctx->r[4] = 0xFFFF;
             ctx->r[5] = 0;
           },
           [](PPCContext* ctx) {
             auto result = static_cast<uint16_t>(ctx->r[3]);
             REQUIRE(result == 0xFFFF);
           });
  test.Run([](PPCContext* ctx) {
             ctx->r[4] = 0xFFFE;
             ctx->r[5] = 1;
           },
           [](PPCContext* ctx) {
             auto result = static_cast<uint16_t>(ctx->r[3]);
             REQUIRE(result == 0xFFFF);
           });
  test.Run([](PPCContext* ctx) {
             ctx->r[4] = 0x8000;
             ctx->r[5] = 16;
           },
           [](PPCContext* ctx) {
             auto result = static_cast<uint16_t>(ctx->r[3]);
             REQUIRE(result == 0xFFFF);
           });
  test.Run([](PPCContext* ctx) {
             ctx->r[4] = 0x7FFF;
             ctx->r[5] = 15;
           },
           [](PPCContext* ctx) {
             auto result = static_cast<uint16_t>(ctx->r[3]);
             REQUIRE(result == 0);
           });
}

TEST_CASE("SHA_I32", "[instr]") {
  TestFunction test([](hir::HIRBuilder& b) {
    StoreGPR(b, 3, b.ZeroExtend(b.Sha(b.Truncate(LoadGPR(b, 4), INT32_TYPE),
                                      b.Truncate(LoadGPR(b, 5), INT8_TYPE)),
                                INT64_TYPE));
    b.Return();
  });
  test.Run([](PPCContext* ctx) {
             ctx->r[4] = 0xFFFF0000;
             ctx->r[5] = 16;
           },
           [](PPCContext* ctx) {
             auto result = static_cast<uint32_t>(ctx->r[3]);
             REQUIRE(result == 0xFFFFFFFF);
           });
  test.Run([](PPCContext* ctx) {
             ctx->r[4] = 0xFFFFFFFF;
             ctx->r[5] = 0;
           },
           [](PPCContext* ctx) {
             auto result = static_cast<uint32_t>(ctx->r[3]);
             REQUIRE(result == 0xFFFFFFFF);
           });
  test.Run([](PPCContext* ctx) {
             ctx->r[4] = 0xFFFFFFFE;
             ctx->r[5] = 1;
           },
           [](PPCContext* ctx) {
             auto result = static_cast<uint32_t>(ctx->r[3]);
             REQUIRE(result == 0xFFFFFFFF);
           });
  test.Run([](PPCContext* ctx) {
             ctx->r[4] = 0x80000000;
             ctx->r[5] = 32;
           },
           [](PPCContext* ctx) {
             auto result = static_cast<uint32_t>(ctx->r[3]);
             REQUIRE(result == 0x80000000);
           });
  test.Run([](PPCContext* ctx) {
             ctx->r[4] = 0x7FFFFFFF;
             ctx->r[5] = 31;
           },
           [](PPCContext* ctx) {
             auto result = static_cast<uint32_t>(ctx->r[3]);
             REQUIRE(result == 0);
           });
}

TEST_CASE("SHA_I64", "[instr]") {
  TestFunction test([](hir::HIRBuilder& b) {
    StoreGPR(b, 3, b.Sha(b.Truncate(LoadGPR(b, 4), INT64_TYPE),
                         b.Truncate(LoadGPR(b, 5), INT8_TYPE)));
    b.Return();
  });
  test.Run([](PPCContext* ctx) {
             ctx->r[4] = 0xFFFFFFFF00000000ull;
             ctx->r[5] = 32;
           },
           [](PPCContext* ctx) {
             auto result = static_cast<uint64_t>(ctx->r[3]);
             REQUIRE(result == 0xFFFFFFFFFFFFFFFFull);
           });
  test.Run([](PPCContext* ctx) {
             ctx->r[4] = 0xFFFFFFFFFFFFFFFFull;
             ctx->r[5] = 0;
           },
           [](PPCContext* ctx) {
             auto result = static_cast<uint64_t>(ctx->r[3]);
             REQUIRE(result == 0xFFFFFFFFFFFFFFFFull);
           });
  test.Run([](PPCContext* ctx) {
             ctx->r[4] = 0xFFFFFFFFFFFFFFFEull;
             ctx->r[5] = 1;
           },
           [](PPCContext* ctx) {
             auto result = static_cast<uint64_t>(ctx->r[3]);
             REQUIRE(result == 0xFFFFFFFFFFFFFFFFull);
           });
  test.Run([](PPCContext* ctx) {
             ctx->r[4] = 0x8000000000000000ull;
             ctx->r[5] = 64;
           },
           [](PPCContext* ctx) {
             auto result = static_cast<uint64_t>(ctx->r[3]);
             REQUIRE(result == 0x8000000000000000ull);
           });
  test.Run([](PPCContext* ctx) {
             ctx->r[4] = 0x7FFFFFFFFFFFFFFFull;
             ctx->r[5] = 63;
           },
           [](PPCContext* ctx) {
             auto result = static_cast<uint64_t>(ctx->r[3]);
             REQUIRE(result == 0);
           });
}