65c7324ffe
This is a convenience method to do a deep copy of an array. This comes up a lot, but can be a pain if you don't have a specific device adapter on which to do the copy.
70 lines
1.9 KiB
C++
70 lines
1.9 KiB
C++
//============================================================================
|
|
// 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.
|
|
//
|
|
// Copyright 2017 National Technology & Engineering Solutions of Sandia, LLC (NTESS).
|
|
// Copyright 2017 UT-Battelle, LLC.
|
|
// Copyright 2017 Los Alamos National Security.
|
|
//
|
|
// Under the terms of Contract DE-NA0003525 with NTESS,
|
|
// 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.
|
|
//============================================================================
|
|
|
|
#include <vtkm/cont/ArrayCopy.h>
|
|
#include <vtkm/cont/ArrayHandleIndex.h>
|
|
|
|
#include <vtkm/TypeTraits.h>
|
|
|
|
#include <vtkm/cont/testing/Testing.h>
|
|
|
|
namespace
|
|
{
|
|
|
|
static const vtkm::Id ARRAY_SIZE = 10;
|
|
|
|
template <typename PortalType>
|
|
void TestValues(const PortalType& portal)
|
|
{
|
|
VTKM_TEST_ASSERT(portal.GetNumberOfValues() == ARRAY_SIZE, "Wrong array size.");
|
|
|
|
for (vtkm::Id index = 0; index < ARRAY_SIZE; ++index)
|
|
{
|
|
VTKM_TEST_ASSERT(test_equal(portal.Get(index), index), "Got bad value.");
|
|
}
|
|
}
|
|
|
|
template <typename ValueType>
|
|
void TryCopy()
|
|
{
|
|
std::cout << "Trying type: " << vtkm::testing::TypeName<ValueType>::Name() << std::endl;
|
|
|
|
vtkm::cont::ArrayHandleIndex input(ARRAY_SIZE);
|
|
vtkm::cont::ArrayHandle<ValueType> output;
|
|
|
|
vtkm::cont::ArrayCopy(input, output);
|
|
|
|
TestValues(output.GetPortalConstControl());
|
|
}
|
|
|
|
void TestArrayCopy()
|
|
{
|
|
TryCopy<vtkm::Id>();
|
|
TryCopy<vtkm::IdComponent>();
|
|
TryCopy<vtkm::Float32>();
|
|
}
|
|
|
|
} // anonymous namespace
|
|
|
|
int UnitTestArrayCopy(int, char* [])
|
|
{
|
|
return vtkm::cont::testing::Testing::Run(TestArrayCopy);
|
|
}
|