summaryrefslogtreecommitdiffstats
path: root/src
diff options
context:
space:
mode:
authorWillem Jan Palenstijn <WillemJan.Palenstijn@uantwerpen.be>2014-04-22 14:15:59 +0000
committerwpalenst <WillemJan.Palenstijn@uantwerpen.be>2014-04-22 14:15:59 +0000
commitb4324c3d3ee5e27c271a4965680d3bf1fee81827 (patch)
tree3feea7c5f844bd4430174c3bd574a7d295ab0caa /src
parent3cc3de7f681c2d8c4c604dac40e92d3e7356d5c7 (diff)
downloadastra-b4324c3d3ee5e27c271a4965680d3bf1fee81827.tar.gz
astra-b4324c3d3ee5e27c271a4965680d3bf1fee81827.tar.bz2
astra-b4324c3d3ee5e27c271a4965680d3bf1fee81827.tar.xz
astra-b4324c3d3ee5e27c271a4965680d3bf1fee81827.zip
Add CustomMemory interface to Float32Data2D
Diffstat (limited to 'src')
-rw-r--r--src/Float32Data2D.cpp75
1 files changed, 67 insertions, 8 deletions
diff --git a/src/Float32Data2D.cpp b/src/Float32Data2D.cpp
index ae84716..151f58d 100644
--- a/src/Float32Data2D.cpp
+++ b/src/Float32Data2D.cpp
@@ -39,6 +39,10 @@ $Id$
namespace astra {
+CFloat32CustomMemory::~CFloat32CustomMemory() {
+
+}
+
//----------------------------------------------------------------------------------------
// Constructors
@@ -77,6 +81,14 @@ CFloat32Data2D::CFloat32Data2D(int _iWidth, int _iHeight, float32 _fScalar)
}
//----------------------------------------------------------------------------------------
+// Create an instance of the CFloat32Data2D class with pre-allocated memory.
+CFloat32Data2D::CFloat32Data2D(int _iWidth, int _iHeight, CFloat32CustomMemory *_pCustomMemory)
+{
+ m_bInitialized = false;
+ _initialize(_iWidth, _iHeight, _pCustomMemory);
+}
+
+//----------------------------------------------------------------------------------------
// Copy constructor
CFloat32Data2D::CFloat32Data2D(const CFloat32Data2D& _other)
{
@@ -103,6 +115,11 @@ CFloat32Data2D& CFloat32Data2D::operator=(const CFloat32Data2D& _dataIn)
memcpy(m_pfData, _dataIn.m_pfData, m_iSize * sizeof(float32));
} else {
+ if (m_pCustomMemory) {
+ // Can't re-allocate custom data
+ ASTRA_ASSERT(false);
+ return *(CFloat32Data2D*)0;
+ }
// Re-allocate data
_unInit();
_initialize(_dataIn.getWidth(), _dataIn.getHeight(), _dataIn.getDataConst());
@@ -224,6 +241,36 @@ bool CFloat32Data2D::_initialize(int _iWidth, int _iHeight, float32 _fScalar)
return true;
}
+//----------------------------------------------------------------------------------------
+// Initializes an instance of the CFloat32Data2D class with pre-allocated memory
+bool CFloat32Data2D::_initialize(int _iWidth, int _iHeight, CFloat32CustomMemory* _pCustomMemory)
+{
+ // basic checks
+ ASTRA_ASSERT(_iWidth > 0);
+ ASTRA_ASSERT(_iHeight > 0);
+ ASTRA_ASSERT(_pCustomMemory != NULL);
+
+ if (m_bInitialized)
+ {
+ _unInit();
+ }
+
+ // calculate size
+ m_iWidth = _iWidth;
+ m_iHeight = _iHeight;
+ m_iSize = (size_t)m_iWidth * m_iHeight;
+
+ // initialize the data pointers
+ m_pCustomMemory = _pCustomMemory;
+ m_pfData = 0;
+ m_ppfData2D = 0;
+ _allocateData();
+
+ // initialization complete
+ return true;
+}
+
+
//----------------------------------------------------------------------------------------
// Memory Allocation
@@ -241,14 +288,20 @@ void CFloat32Data2D::_allocateData()
ASTRA_ASSERT(m_pfData == NULL);
ASTRA_ASSERT(m_ppfData2D == NULL);
- // allocate contiguous block
+ if (!m_pCustomMemory) {
+
+ // allocate contiguous block
#ifdef _MSC_VER
- m_pfData = (float32*)_aligned_malloc(m_iSize * sizeof(float32), 16);
+ m_pfData = (float32*)_aligned_malloc(m_iSize * sizeof(float32), 16);
#else
- int ret = posix_memalign((void**)&m_pfData, 16, m_iSize * sizeof(float32));
- ASTRA_ASSERT(ret == 0);
+ int ret = posix_memalign((void**)&m_pfData, 16, m_iSize * sizeof(float32));
+ ASTRA_ASSERT(ret == 0);
#endif
+ } else {
+ m_pfData = m_pCustomMemory->m_fPtr;
+ }
+
// create array of pointers to each row of the data block
m_ppfData2D = new float32*[m_iHeight];
for (int iy = 0; iy < m_iHeight; iy++)
@@ -264,15 +317,20 @@ void CFloat32Data2D::_freeData()
// basic checks
ASTRA_ASSERT(m_pfData != NULL);
ASTRA_ASSERT(m_ppfData2D != NULL);
-
// free memory for index table
delete[] m_ppfData2D;
- // free memory for data block
+
+ if (!m_pCustomMemory) {
+ // free memory for data block
#ifdef _MSC_VER
- _aligned_free(m_pfData);
+ _aligned_free(m_pfData);
#else
- free(m_pfData);
+ free(m_pfData);
#endif
+ } else {
+ delete m_pCustomMemory;
+ m_pCustomMemory = 0;
+ }
}
@@ -286,6 +344,7 @@ void CFloat32Data2D::_clear()
m_pfData = NULL;
m_ppfData2D = NULL;
+ m_pCustomMemory = NULL;
m_fGlobalMin = 0.0f;
m_fGlobalMax = 0.0f;