blob: ad4ee431025d2b29a10f119df50170159c124d79 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
|
// File: Standard_MMgrTBBalloc.cxx
// Created: Tue Mar 15 12:09:38 2010
// Author: Sergey KUUL
#include <Standard_MMgrTBBalloc.hxx>
#include <Standard_OutOfMemory.hxx>
// NOTE: replaced by more correct check
//#if defined(WNT) || defined(LIN)
//#define HAVE_TBB 1
//#endif
// paralleling with Intel TBB
#ifdef HAVE_TBB
#include <tbb/scalable_allocator.h>
using namespace tbb;
#else
#define scalable_malloc malloc
#define scalable_calloc calloc
#define scalable_realloc realloc
#define scalable_free free
#endif
//=======================================================================
//function : Standard_MMgrTBBalloc
//purpose :
//=======================================================================
Standard_MMgrTBBalloc::Standard_MMgrTBBalloc(const Standard_Boolean aClear)
{
myClear = aClear;
}
//=======================================================================
//function : Allocate
//purpose :
//=======================================================================
Standard_Address Standard_MMgrTBBalloc::Allocate(const Standard_Size aSize)
{
// the size is rounded up to 4 since some OCC classes
// (e.g. TCollection_AsciiString) assume memory to be double word-aligned
const Standard_Size aRoundSize = (aSize + 3) & ~0x3;
// we use ?: operator instead of if() since it is faster :-)
Standard_Address aPtr = ( myClear ? scalable_calloc(aRoundSize, sizeof(char)) :
scalable_malloc(aRoundSize) );
if ( ! aPtr )
Standard_OutOfMemory::Raise("Standard_MMgrTBBalloc::Allocate(): malloc failed");
return aPtr;
}
//=======================================================================
//function : Free
//purpose :
//=======================================================================
void Standard_MMgrTBBalloc::Free(Standard_Address& aStorage)
{
scalable_free(aStorage);
aStorage=NULL;
}
//=======================================================================
//function : Reallocate
//purpose :
//=======================================================================
Standard_Address Standard_MMgrTBBalloc::Reallocate(Standard_Address& aStorage,
const Standard_Size newSize)
{
// the size is rounded up to 4 since some OCC classes
// (e.g. TCollection_AsciiString) assume memory to be double word-aligned
const Standard_Size aRoundSize = (newSize + 3) & ~0x3;
Standard_Address newStorage = (Standard_Address)scalable_realloc(aStorage, aRoundSize);
if ( ! newStorage )
Standard_OutOfMemory::Raise("Standard_MMgrTBBalloc::Reallocate(): realloc failed");
// Note that it is not possible to ensure that additional memory
// allocated by realloc will be cleared (so as to satisfy myClear mode);
// in order to do that we would need using memset...
aStorage = NULL;
return newStorage;
}
|
