2014-02-10 19:53:03 +00:00
|
|
|
//============================================================================
|
|
|
|
// Copyright (c) Kitware, Inc.
|
|
|
|
// All rights reserved.
|
|
|
|
// See LICENSE.txt for details.
|
|
|
|
// This software is distributed WITHOUT ANY WARRANTY; without even
|
|
|
|
// the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
|
|
|
|
// PURPOSE. See the above copyright notice for more information.
|
|
|
|
//
|
2017-09-20 21:33:44 +00:00
|
|
|
// Copyright 2014 National Technology & Engineering Solutions of Sandia, LLC (NTESS).
|
2014-02-10 19:53:03 +00:00
|
|
|
// Copyright 2014 UT-Battelle, LLC.
|
2015-05-21 12:09:22 +00:00
|
|
|
// Copyright 2014 Los Alamos National Security.
|
2014-02-10 19:53:03 +00:00
|
|
|
//
|
2017-09-20 21:33:44 +00:00
|
|
|
// Under the terms of Contract DE-NA0003525 with NTESS,
|
2014-02-10 19:53:03 +00:00
|
|
|
// the U.S. Government retains certain rights in this software.
|
|
|
|
//
|
|
|
|
// Under the terms of Contract DE-AC52-06NA25396 with Los Alamos National
|
|
|
|
// Laboratory (LANL), the U.S. Government retains certain rights in
|
|
|
|
// this software.
|
|
|
|
//============================================================================
|
|
|
|
|
2017-04-05 22:44:15 +00:00
|
|
|
#include <vtkm/cont/ArrayHandle.h>
|
2014-06-23 23:33:04 +00:00
|
|
|
#include <vtkm/cont/StorageBasic.h>
|
2014-02-10 19:53:03 +00:00
|
|
|
|
2014-10-07 23:27:31 +00:00
|
|
|
#include <vtkm/VecTraits.h>
|
2017-05-18 14:51:24 +00:00
|
|
|
#include <vtkm/cont/testing/Testing.h>
|
2014-02-10 19:53:03 +00:00
|
|
|
|
2018-09-14 19:37:38 +00:00
|
|
|
#if defined(VTKM_STORAGE)
|
|
|
|
#undef VTKM_STORAGE
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#define VTKM_STORAGE VTKM_STORAGE_ERROR
|
|
|
|
|
2017-05-18 14:29:41 +00:00
|
|
|
namespace
|
|
|
|
{
|
2014-02-10 19:53:03 +00:00
|
|
|
|
|
|
|
const vtkm::Id ARRAY_SIZE = 10;
|
|
|
|
|
|
|
|
template <typename T>
|
|
|
|
struct TemplatedTests
|
|
|
|
{
|
2017-08-16 15:34:21 +00:00
|
|
|
using StorageType = vtkm::cont::internal::Storage<T, vtkm::cont::StorageTagBasic>;
|
|
|
|
using ValueType = typename StorageType::ValueType;
|
|
|
|
using PortalType = typename StorageType::PortalType;
|
2014-02-10 19:53:03 +00:00
|
|
|
|
2017-05-18 14:29:41 +00:00
|
|
|
void SetStorage(StorageType& array, const ValueType& value)
|
2014-02-10 19:53:03 +00:00
|
|
|
{
|
2014-09-08 20:59:11 +00:00
|
|
|
PortalType portal = array.GetPortal();
|
|
|
|
for (vtkm::Id index = 0; index < portal.GetNumberOfValues(); index++)
|
2014-02-11 21:20:30 +00:00
|
|
|
{
|
2014-09-08 20:59:11 +00:00
|
|
|
portal.Set(index, value);
|
2014-02-11 21:20:30 +00:00
|
|
|
}
|
2014-02-10 19:53:03 +00:00
|
|
|
}
|
|
|
|
|
2017-05-18 14:29:41 +00:00
|
|
|
bool CheckStorage(StorageType& array, const ValueType& value)
|
2014-02-10 19:53:03 +00:00
|
|
|
{
|
2014-09-08 20:59:11 +00:00
|
|
|
PortalType portal = array.GetPortal();
|
|
|
|
for (vtkm::Id index = 0; index < portal.GetNumberOfValues(); index++)
|
2014-02-11 21:20:30 +00:00
|
|
|
{
|
2017-05-18 14:29:41 +00:00
|
|
|
if (!test_equal(portal.Get(index), value))
|
|
|
|
{
|
|
|
|
return false;
|
|
|
|
}
|
2014-02-11 21:20:30 +00:00
|
|
|
}
|
2014-02-10 19:53:03 +00:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2017-05-18 14:29:41 +00:00
|
|
|
typename vtkm::VecTraits<ValueType>::ComponentType STOLEN_ARRAY_VALUE() { return 29; }
|
2014-02-10 19:53:03 +00:00
|
|
|
|
|
|
|
/// Returned value should later be passed to StealArray2. It is best to
|
|
|
|
/// put as much between the two test parts to maximize the chance of a
|
|
|
|
/// deallocated array being overridden (and thus detected).
|
2017-05-18 14:29:41 +00:00
|
|
|
ValueType* StealArray1()
|
2014-02-10 19:53:03 +00:00
|
|
|
{
|
2017-05-18 14:29:41 +00:00
|
|
|
ValueType* stolenArray;
|
2014-02-10 19:53:03 +00:00
|
|
|
|
|
|
|
ValueType stolenArrayValue = ValueType(STOLEN_ARRAY_VALUE());
|
|
|
|
|
2014-06-23 23:33:04 +00:00
|
|
|
StorageType stealMyArray;
|
2014-02-10 19:53:03 +00:00
|
|
|
stealMyArray.Allocate(ARRAY_SIZE);
|
2014-10-23 21:39:05 +00:00
|
|
|
this->SetStorage(stealMyArray, stolenArrayValue);
|
2014-02-10 19:53:03 +00:00
|
|
|
|
|
|
|
VTKM_TEST_ASSERT(stealMyArray.GetNumberOfValues() == ARRAY_SIZE,
|
2014-02-11 21:20:30 +00:00
|
|
|
"Array not properly allocated.");
|
2014-02-10 19:53:03 +00:00
|
|
|
// This call steals the array and prevents deallocation.
|
2017-08-17 12:43:39 +00:00
|
|
|
VTKM_TEST_ASSERT(stealMyArray.WillDeallocate() == true,
|
|
|
|
"Array to be stolen needs to be owned by VTK-m");
|
2014-02-10 19:53:03 +00:00
|
|
|
stolenArray = stealMyArray.StealArray();
|
2017-08-17 12:43:39 +00:00
|
|
|
VTKM_TEST_ASSERT(stealMyArray.WillDeallocate() == false,
|
|
|
|
"Stolen array should not be owned by VTK-m");
|
2014-02-10 19:53:03 +00:00
|
|
|
|
|
|
|
return stolenArray;
|
|
|
|
}
|
2017-05-18 14:29:41 +00:00
|
|
|
void StealArray2(ValueType* stolenArray)
|
2014-02-10 19:53:03 +00:00
|
|
|
{
|
|
|
|
ValueType stolenArrayValue = ValueType(STOLEN_ARRAY_VALUE());
|
|
|
|
|
|
|
|
for (vtkm::Id index = 0; index < ARRAY_SIZE; index++)
|
2014-02-11 21:20:30 +00:00
|
|
|
{
|
2014-02-10 19:53:03 +00:00
|
|
|
VTKM_TEST_ASSERT(test_equal(stolenArray[index], stolenArrayValue),
|
2014-02-11 21:20:30 +00:00
|
|
|
"Stolen array did not retain values.");
|
|
|
|
}
|
2014-10-23 21:39:05 +00:00
|
|
|
typename StorageType::AllocatorType allocator;
|
2018-02-09 21:17:39 +00:00
|
|
|
allocator.deallocate(stolenArray);
|
2014-02-10 19:53:03 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
void BasicAllocation()
|
|
|
|
{
|
2014-06-23 23:33:04 +00:00
|
|
|
StorageType arrayStorage;
|
2017-05-18 14:29:41 +00:00
|
|
|
VTKM_TEST_ASSERT(arrayStorage.GetNumberOfValues() == 0, "New array storage not zero sized.");
|
2014-02-10 19:53:03 +00:00
|
|
|
|
2014-06-23 23:33:04 +00:00
|
|
|
arrayStorage.Allocate(ARRAY_SIZE);
|
|
|
|
VTKM_TEST_ASSERT(arrayStorage.GetNumberOfValues() == ARRAY_SIZE,
|
2014-02-11 21:20:30 +00:00
|
|
|
"Array not properly allocated.");
|
2014-02-10 19:53:03 +00:00
|
|
|
|
2014-10-08 22:56:33 +00:00
|
|
|
const ValueType BASIC_ALLOC_VALUE = ValueType(48);
|
2014-10-23 21:39:05 +00:00
|
|
|
this->SetStorage(arrayStorage, BASIC_ALLOC_VALUE);
|
|
|
|
VTKM_TEST_ASSERT(this->CheckStorage(arrayStorage, BASIC_ALLOC_VALUE),
|
2014-02-11 21:20:30 +00:00
|
|
|
"Array not holding value.");
|
2014-02-10 19:53:03 +00:00
|
|
|
|
2014-06-23 23:33:04 +00:00
|
|
|
arrayStorage.Allocate(ARRAY_SIZE * 2);
|
|
|
|
VTKM_TEST_ASSERT(arrayStorage.GetNumberOfValues() == ARRAY_SIZE * 2,
|
2014-02-11 21:20:30 +00:00
|
|
|
"Array not reallocated correctly.");
|
2014-02-10 19:53:03 +00:00
|
|
|
|
2014-06-23 23:33:04 +00:00
|
|
|
arrayStorage.Shrink(ARRAY_SIZE);
|
|
|
|
VTKM_TEST_ASSERT(arrayStorage.GetNumberOfValues() == ARRAY_SIZE,
|
2014-02-11 21:20:30 +00:00
|
|
|
"Array Shrnk failed to resize.");
|
2014-02-10 19:53:03 +00:00
|
|
|
|
2014-06-23 23:33:04 +00:00
|
|
|
arrayStorage.ReleaseResources();
|
2017-05-18 14:29:41 +00:00
|
|
|
VTKM_TEST_ASSERT(arrayStorage.GetNumberOfValues() == 0, "Array not released correctly.");
|
2014-02-10 19:53:03 +00:00
|
|
|
|
|
|
|
try
|
2014-02-11 21:20:30 +00:00
|
|
|
{
|
2014-06-23 23:33:04 +00:00
|
|
|
arrayStorage.Shrink(ARRAY_SIZE);
|
2017-05-18 14:29:41 +00:00
|
|
|
VTKM_TEST_ASSERT(true == false,
|
2014-02-11 21:20:30 +00:00
|
|
|
"Array shrink do a larger size was possible. This can't be allowed.");
|
|
|
|
}
|
2017-05-18 14:29:41 +00:00
|
|
|
catch (vtkm::cont::ErrorBadValue&)
|
|
|
|
{
|
|
|
|
}
|
2014-02-10 19:53:03 +00:00
|
|
|
}
|
|
|
|
|
2018-04-04 15:27:57 +00:00
|
|
|
void UserFreeFunction()
|
|
|
|
{
|
|
|
|
ValueType* temp = new ValueType[ARRAY_SIZE];
|
|
|
|
StorageType arrayStorage(
|
|
|
|
temp, ARRAY_SIZE, [](void* ptr) { delete[] static_cast<ValueType*>(ptr); });
|
|
|
|
VTKM_TEST_ASSERT(temp == arrayStorage.GetArray(),
|
|
|
|
"improper pointer after telling storage to own user allocated memory");
|
|
|
|
|
|
|
|
const ValueType BASIC_ALLOC_VALUE = ValueType(48);
|
|
|
|
this->SetStorage(arrayStorage, BASIC_ALLOC_VALUE);
|
|
|
|
VTKM_TEST_ASSERT(this->CheckStorage(arrayStorage, BASIC_ALLOC_VALUE),
|
|
|
|
"Array not holding value.");
|
|
|
|
|
|
|
|
arrayStorage.Allocate(ARRAY_SIZE * 2);
|
|
|
|
VTKM_TEST_ASSERT(arrayStorage.GetNumberOfValues() == ARRAY_SIZE * 2,
|
|
|
|
"Array not reallocated correctly.");
|
|
|
|
}
|
|
|
|
|
2014-02-10 19:53:03 +00:00
|
|
|
void operator()()
|
|
|
|
{
|
2017-05-18 14:29:41 +00:00
|
|
|
ValueType* stolenArray = StealArray1();
|
2014-02-10 19:53:03 +00:00
|
|
|
|
|
|
|
BasicAllocation();
|
2018-04-04 15:27:57 +00:00
|
|
|
UserFreeFunction();
|
2014-02-10 19:53:03 +00:00
|
|
|
|
|
|
|
StealArray2(stolenArray);
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
struct TestFunctor
|
|
|
|
{
|
|
|
|
template <typename T>
|
2014-10-08 22:56:33 +00:00
|
|
|
void operator()(T) const
|
2014-02-10 19:53:03 +00:00
|
|
|
{
|
|
|
|
TemplatedTests<T> tests;
|
|
|
|
tests();
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
2014-06-23 23:33:04 +00:00
|
|
|
void TestStorageBasic()
|
2014-02-10 19:53:03 +00:00
|
|
|
{
|
2016-08-24 22:02:18 +00:00
|
|
|
vtkm::testing::Testing::TryTypes(TestFunctor());
|
2014-02-10 19:53:03 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
} // Anonymous namespace
|
|
|
|
|
2019-01-01 22:19:02 +00:00
|
|
|
int UnitTestStorageBasic(int argc, char* argv[])
|
2014-02-10 19:53:03 +00:00
|
|
|
{
|
2019-01-01 22:19:02 +00:00
|
|
|
return vtkm::cont::testing::Testing::Run(TestStorageBasic, argc, argv);
|
2014-02-10 19:53:03 +00:00
|
|
|
}
|