/* Copyright 2014 Samsung Electronics Co., Ltd.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

/**
 * Unit test for pool manager.
 */

#define JERRY_MEM_POOL_INTERNAL

#include "globals.h"
#include "jerry-libc.h"
#include "mem-allocator.h"
#include "mem-pool.h"
#include "mem-poolman.h"

extern void srand (unsigned int __seed);
extern int rand (void);
extern long int time (long int *__timer);

// Heap size is 8K
const size_t test_heap_size = 8 * 1024;

// Iterations count
const uint32_t test_iters = 16384;

// Subiterations count
const uint32_t test_max_sub_iters = 32;

int
main( int __unused argc,
      char __unused **argv)
{
    uint8_t heap[test_heap_size] __attribute__((aligned(MEM_ALIGNMENT)));

    mem_heap_init( heap, sizeof (heap));
    mem_pools_init();

    srand((unsigned int) time(NULL));
    unsigned int seed = (unsigned int)rand();
    __printf("seed=%u\n", seed);
    srand(seed);

    for ( uint32_t i = 0; i < test_iters; i++ )
    {
        const size_t subiters = ( (size_t) rand() % test_max_sub_iters ) + 1;

        uint8_t * ptrs[subiters];
        mem_pool_chunk_type_t types[subiters];

        for ( size_t j = 0; j < subiters; j++ )
        {
            mem_pool_chunk_type_t type = (mem_pool_chunk_type_t) (rand() % MEM_POOL_CHUNK_TYPE__COUNT);
            const size_t chunk_size = mem_get_chunk_size( type);

            types[j] = type;
            ptrs[j] = mem_pools_alloc( type);
            // JERRY_ASSERT(ptrs[j] != NULL);

            if ( ptrs[j] != NULL )
            {
                __memset(ptrs[j], 0, chunk_size);
            }
        }

        // mem_heap_print( false);
        
        for ( size_t j = 0; j < subiters; j++ )
        {
            if ( ptrs[j] != NULL )
            {
                mem_pool_chunk_type_t type = types[j];
                const size_t chunk_size = mem_get_chunk_size( type);

                for ( size_t k = 0; k < chunk_size; k++ )
                {
                    JERRY_ASSERT( ((uint8_t*) ptrs[j])[k] == 0 );
                }
                
                mem_pools_free( type, ptrs[j]);
            }
        }
    }

#ifdef MEM_STATS
    mem_pools_stats_t stats;
    mem_pools_get_stats( &stats);
#endif /* MEM_STATS */

    __printf("Pools stats:\n");
    for(mem_pool_chunk_type_t type = 0;
        type < MEM_POOL_CHUNK_TYPE__COUNT;
        type++)
    {
      __printf(" Chunk size: %u\n"
                  "  Pools: %lu\n"
                  "  Allocated chunks: %lu\n"
                  "  Free chunks: %lu\n"
                  "  Peak pools: %lu\n"
                  "  Peak allocated chunks: %lu\n",
                  mem_get_chunk_size( type),
                  stats.pools_count[ type ],
                  stats.allocated_chunks[ type ],
                  stats.free_chunks[ type ],
                  stats.peak_pools_count[ type ],
                  stats.peak_allocated_chunks[ type ]);
    }
    __printf("\n");

    return 0;
} /* main */