mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-09-19 10:35:42 +00:00
76 lines
2.6 KiB
C++
76 lines
2.6 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.
|
|
//============================================================================
|
|
|
|
#include <vtkm/cont/ArrayHandleConcatenate.h>
|
|
#include <vtkm/cont/ArrayHandleIndex.h>
|
|
|
|
#include <vtkm/cont/testing/Testing.h>
|
|
|
|
namespace UnitTestArrayHandleConcatenateNamespace
|
|
{
|
|
|
|
const vtkm::Id ARRAY_SIZE = 5;
|
|
|
|
void TestArrayHandleConcatenate()
|
|
{
|
|
vtkm::cont::ArrayHandleIndex array1(ARRAY_SIZE);
|
|
vtkm::cont::ArrayHandleIndex array2(2 * ARRAY_SIZE);
|
|
|
|
vtkm::cont::ArrayHandleConcatenate<vtkm::cont::ArrayHandleIndex, vtkm::cont::ArrayHandleIndex>
|
|
array3(array1, array2);
|
|
|
|
vtkm::cont::ArrayHandleIndex array4(ARRAY_SIZE);
|
|
vtkm::cont::ArrayHandleConcatenate<
|
|
vtkm::cont::ArrayHandleConcatenate<vtkm::cont::ArrayHandleIndex, // 1st
|
|
vtkm::cont::ArrayHandleIndex>, // ArrayHandle
|
|
vtkm::cont::ArrayHandleIndex> // 2nd ArrayHandle
|
|
array5;
|
|
{
|
|
array5 = vtkm::cont::make_ArrayHandleConcatenate(array3, array4);
|
|
}
|
|
|
|
auto array5Portal = array5.ReadPortal();
|
|
for (vtkm::Id index = 0; index < array5.GetNumberOfValues(); index++)
|
|
{
|
|
std::cout << array5Portal.Get(index) << std::endl;
|
|
}
|
|
}
|
|
|
|
void TestConcatenateEmptyArray()
|
|
{
|
|
std::vector<vtkm::Float64> vec;
|
|
for (vtkm::Id i = 0; i < ARRAY_SIZE; i++)
|
|
vec.push_back(vtkm::Float64(i) * 1.5);
|
|
|
|
using CoeffValueType = vtkm::Float64;
|
|
using CoeffArrayTypeTmp = vtkm::cont::ArrayHandle<CoeffValueType>;
|
|
using ArrayConcat = vtkm::cont::ArrayHandleConcatenate<CoeffArrayTypeTmp, CoeffArrayTypeTmp>;
|
|
using ArrayConcat2 = vtkm::cont::ArrayHandleConcatenate<ArrayConcat, CoeffArrayTypeTmp>;
|
|
|
|
CoeffArrayTypeTmp arr1 = vtkm::cont::make_ArrayHandle(vec);
|
|
CoeffArrayTypeTmp arr2, arr3;
|
|
|
|
ArrayConcat arrConc(arr2, arr1);
|
|
ArrayConcat2 arrConc2(arrConc, arr3);
|
|
|
|
auto arrConc2Portal = arrConc2.ReadPortal();
|
|
for (vtkm::Id i = 0; i < arrConc2.GetNumberOfValues(); i++)
|
|
std::cout << arrConc2Portal.Get(i) << std::endl;
|
|
}
|
|
|
|
} // namespace UnitTestArrayHandleIndexNamespace
|
|
|
|
int UnitTestArrayHandleConcatenate(int argc, char* argv[])
|
|
{
|
|
using namespace UnitTestArrayHandleConcatenateNamespace;
|
|
//TestConcatenateEmptyArray();
|
|
return vtkm::cont::testing::Testing::Run(TestArrayHandleConcatenate, argc, argv);
|
|
}
|