#include "lib/self_test.h"

#include "lib/bits.h"

class TestBits : public CxxTest::TestSuite 
{
public:
    void test_is_pow2()
    {
        TS_ASSERT_EQUALS(is_pow2(0u), false);
        TS_ASSERT_EQUALS(is_pow2(~0u), false);
        TS_ASSERT_EQUALS(is_pow2(0x80000001), false);
        TS_ASSERT_EQUALS(is_pow2(1), true);
        TS_ASSERT_EQUALS(is_pow2(1u << 31), true);
    }

    void test_log2_of_pow2()
    {
        TS_ASSERT_EQUALS(log2_of_pow2(0u), -1);
        TS_ASSERT_EQUALS(log2_of_pow2(3u), -1);
        TS_ASSERT_EQUALS(log2_of_pow2(0xffffffffu), -1);
        TS_ASSERT_EQUALS(log2_of_pow2(1u), 0);
        TS_ASSERT_EQUALS(log2_of_pow2(256u), 8);
        TS_ASSERT_EQUALS(log2_of_pow2(0x80000000u), 31);
    }

    void test_ceil_log2()
    {
        TS_ASSERT_EQUALS(ceil_log2(3u), 2u);
        TS_ASSERT_EQUALS(ceil_log2(0xffffffffu), 32u);
        TS_ASSERT_EQUALS(ceil_log2(1u), 0u);
        TS_ASSERT_EQUALS(ceil_log2(256u), 8u);
        TS_ASSERT_EQUALS(ceil_log2(0x80000000u), 31u);
    }

    void test_floor_log2()
    {
        TS_ASSERT_EQUALS(floor_log2(1.f), 0);
        TS_ASSERT_EQUALS(floor_log2(3.f), 1);
        TS_ASSERT_EQUALS(floor_log2(256.f), 8);
    }

    void test_round_up_to_pow2()
    {
        TS_ASSERT_EQUALS(round_up_to_pow2(0u), 1u);
        TS_ASSERT_EQUALS(round_up_to_pow2(1u), 2u);
        TS_ASSERT_EQUALS(round_up_to_pow2(127u), 128u);
        TS_ASSERT_EQUALS(round_up_to_pow2(128u), 256u);
        TS_ASSERT_EQUALS(round_up_to_pow2(129u), 256u);
    }

    void test_round_up()
    {
        TS_ASSERT_EQUALS(round_up( 0u, 16u), 0u);
        TS_ASSERT_EQUALS(round_up( 4u, 16u), 16u);
        TS_ASSERT_EQUALS(round_up(15u, 16u), 16u);
        TS_ASSERT_EQUALS(round_up(20u, 32u), 32u);
        TS_ASSERT_EQUALS(round_up(29u, 32u), 32u);
        TS_ASSERT_EQUALS(round_up(0x1000u, 0x1000u), 0x1000u);
        TS_ASSERT_EQUALS(round_up(0x1001u, 0x1000u), 0x2000u);
        TS_ASSERT_EQUALS(round_up(0x1900u, 0x1000u), 0x2000u);
    }

    void test_round_down()
    {
        TS_ASSERT_EQUALS(round_down( 0u, 16u), 0u);
        TS_ASSERT_EQUALS(round_down( 4u, 16u), 0u);
        TS_ASSERT_EQUALS(round_down(15u, 16u), 0u);
        TS_ASSERT_EQUALS(round_down(20u, 16u), 16u);
        TS_ASSERT_EQUALS(round_down(29u, 16u), 16u);
        TS_ASSERT_EQUALS(round_down(0x1900u, 0x1000u), 0x1000u);
        TS_ASSERT_EQUALS(round_down(0x2001u, 0x2000u), 0x2000u);
    }
};