1 /*

     2  * mempool.h: memory pool

     3  *

     4  * This file is part of Pyrit Ray Tracer.

     5  *

     6  * Copyright 2008  Radek Brich

     7  *

     8  * Permission is hereby granted, free of charge, to any person obtaining a copy

     9  * of this software and associated documentation files (the "Software"), to deal

    10  * in the Software without restriction, including without limitation the rights

    11  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell

    12  * copies of the Software, and to permit persons to whom the Software is

    13  * furnished to do so, subject to the following conditions:

    14  *

    15  * The above copyright notice and this permission notice shall be included in

    16  * all copies or substantial portions of the Software.

    17  *

    18  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR

    19  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,

    20  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE

    21  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER

    22  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,

    23  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN

    24  * THE SOFTWARE.

    25  */

    26 

    27 #ifndef MEMPOOL_H

    28 #define MEMPOOL_H

    29 

    30 #include "common.h"

    31 

    32 template <typename Type>

    33 class MemoryPool

    34 {

    35 	void *mem;

    36 	size_t cur, size, typesize, align;

    37 

    38 	void init()

    39 	{

    40 		typesize = ((sizeof(Type)-1)/align+1)*align;

    41 #ifndef NO_SIMD

    42 		mem = (Type *)_mm_malloc(size * typesize, align);

    43 #else

    44 		mem = (Type *)malloc(inisize * typesize);

    45 #endif

    46 	};

    47 public:

    48 	MemoryPool(const size_t inisize):

    49 		cur(0), size(inisize), align(16) { init(); };

    50 	MemoryPool(const size_t inisize, const size_t inialign):

    51 		cur(0), size(inisize), align(inialign) { init(); };

    52 	~MemoryPool()

    53 	{

    54 #ifndef NO_SIMD

    55 		_mm_free(mem);

    56 #else

    57 		free(mem);

    58 #endif

    59 	};

    60 

    61 	void *alloc()

    62 	{

    63 		if (cur == size)

    64 		{

    65 #ifndef NO_SIMD

    66 			void *newmem = _mm_malloc(2 * size * typesize, align);

    67 #else

    68 			void *newmem = malloc(2 * size * typesize);

    69 #endif

    70 			memcpy(newmem, mem, size * typesize);

    71 			mem = newmem;

    72 			size *= 2;

    73 		}

    74 		return (void*)((size_t)mem + typesize*(cur++));

    75 	};

    76 };

    77 

    78 #endif