1 // **************************************************************************
2 // * This file is property of and copyright by the ALICE HLT Project *
3 // * All rights reserved. *
5 // * Primary Authors: *
6 // * Copyright 2009 Matthias Kretz <kretz@kde.org> *
8 // * Permission to use, copy, modify and distribute this software and its *
9 // * documentation strictly for non-commercial purposes is hereby granted *
10 // * without fee, provided that the above copyright notice appears in all *
11 // * copies and that both the copyright notice and this permission notice *
12 // * appear in the supporting documentation. The authors make no claims *
13 // * about the suitability of this software for any purpose. It is *
14 // * provided "as is" without express or implied warranty. *
15 // **************************************************************************
17 #ifndef MEMORYASSIGNMENTHELPERS_H
18 #define MEMORYASSIGNMENTHELPERS_H
24 template<unsigned int X>
25 GPUhd() static inline void AlignTo( char *&mem )
27 STATIC_ASSERT( ( X & ( X - 1 ) ) == 0, X_needs_to_be_a_multiple_of_2 );
28 const int offset = reinterpret_cast<unsigned long long>( mem ) & ( X - 1 );
30 mem += ( X - offset );
32 //assert( ( reinterpret_cast<unsigned long>( mem ) & ( X - 1 ) ) == 0 );
35 template<unsigned int X>
36 GPUhd() static inline unsigned int NextMultipleOf( unsigned int value )
38 STATIC_ASSERT( ( X & ( X - 1 ) ) == 0, X_needs_to_be_a_multiple_of_2 );
39 const int offset = value & ( X - 1 );
41 return value + X - offset;
46 template<typename T, unsigned int Alignment> static T *AssignMemory( char *&mem, unsigned int size )
48 STATIC_ASSERT( ( Alignment & ( Alignment - 1 ) ) == 0, Alignment_needs_to_be_a_multiple_of_2 );
49 AlignTo<Alignment> ( mem );
50 T *r = reinterpret_cast<T *>( mem );
51 mem += size * sizeof( T );
55 template<typename T, unsigned int Alignment> static inline T *AssignMemory( char *&mem, unsigned int stride, unsigned int count )
57 assert( 0 == ( stride & ( Alignment - 1 ) ) );
58 return AssignMemory<T, Alignment>( mem, stride * count );
61 template<typename T, unsigned int Alignment> GPUhd() static T *_assignMemory( char *&mem, unsigned int size )
63 STATIC_ASSERT( ( Alignment & ( Alignment - 1 ) ) == 0, Alignment_needs_to_be_a_multiple_of_2 );
64 AlignTo<Alignment>( mem );
65 T *r = reinterpret_cast<T *>( mem );
66 mem += size * sizeof( T );
70 template<typename T> GPUhd() static inline void AssignMemory( T *&dst, char *&mem, int count )
72 dst = _assignMemory < T, ( sizeof( T ) & ( sizeof( T ) - 1 ) ) == 0 && sizeof( T ) <= 16 ? sizeof( T ) : sizeof( void * ) > ( mem, count );
75 #endif // MEMORYASSIGNMENTHELPERS_H