mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-09-20 11:05:44 +00:00
242 lines
8.8 KiB
C++
242 lines
8.8 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 2014 Sandia Corporation.
|
|
// Copyright 2014 UT-Battelle, LLC.
|
|
// Copyright 2014 Los Alamos National Security.
|
|
//
|
|
// Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
|
|
// 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/worklet/StreamLineUniformGrid.h>
|
|
#include <vtkm/cont/ArrayHandle.h>
|
|
#include <vtkm/cont/DataSet.h>
|
|
#include <vtkm/cont/testing/Testing.h>
|
|
|
|
#include <fstream>
|
|
#include <vector>
|
|
#include <math.h>
|
|
|
|
namespace {
|
|
|
|
template <typename T>
|
|
VTKM_EXEC_CONT_EXPORT
|
|
vtkm::Vec<T,3> Normalize(vtkm::Vec<T,3> v)
|
|
{
|
|
T magnitude = static_cast<T>(sqrt(vtkm::dot(v, v)));
|
|
T zero = static_cast<T>(0.0);
|
|
T one = static_cast<T>(1.0);
|
|
if (magnitude == zero)
|
|
return vtkm::make_Vec(zero, zero, zero);
|
|
else
|
|
return one / magnitude * v;
|
|
}
|
|
|
|
float data[125*3] =
|
|
{
|
|
-0.00603248f, -0.0966396f, -0.000732792f,
|
|
0.000530014f, -0.0986189f, -0.000806706f,
|
|
0.00684929f, -0.100098f, -0.000876566f,
|
|
0.0129235f, -0.101102f, -0.000942341f,
|
|
0.0187515f, -0.101656f, -0.00100401f,
|
|
0.0706091f, -0.083023f, -0.00144278f,
|
|
0.0736404f, -0.0801616f, -0.00145784f,
|
|
0.0765194f, -0.0772063f, -0.00147036f,
|
|
0.0792559f, -0.0741751f, -0.00148051f,
|
|
0.0818589f, -0.071084f, -0.00148843f,
|
|
0.103585f, -0.0342287f, -0.001425f,
|
|
0.104472f, -0.0316147f, -0.00140433f,
|
|
0.105175f, -0.0291574f, -0.00138057f,
|
|
0.105682f, -0.0268808f, -0.00135357f,
|
|
0.105985f, -0.0248099f, -0.00132315f,
|
|
-0.00244603f, -0.0989576f, -0.000821705f,
|
|
0.00389525f, -0.100695f, -0.000894513f,
|
|
0.00999301f, -0.10193f, -0.000963114f,
|
|
0.0158452f, -0.102688f, -0.00102747f,
|
|
0.0214509f, -0.102995f, -0.00108757f,
|
|
0.0708166f, -0.081799f, -0.00149941f,
|
|
0.0736939f, -0.0787879f, -0.00151236f,
|
|
0.0764359f, -0.0756944f, -0.00152297f,
|
|
0.0790546f, -0.0725352f, -0.00153146f,
|
|
0.0815609f, -0.0693255f, -0.001538f,
|
|
-0.00914287f, -0.104658f, -0.001574f,
|
|
-0.00642891f, -0.10239f, -0.00159659f,
|
|
-0.00402289f, -0.0994835f, -0.00160731f,
|
|
-0.00194792f, -0.0959752f, -0.00160528f,
|
|
-0.00022818f, -0.0919077f, -0.00158957f,
|
|
-0.0134913f, -0.0274735f, -9.50056e-05f,
|
|
-0.0188683f, -0.023273f, 0.000194107f,
|
|
-0.0254516f, -0.0197589f, 0.000529693f,
|
|
-0.0312798f, -0.0179514f, 0.00083619f,
|
|
-0.0360426f, -0.0177537f, 0.00110164f,
|
|
0.0259929f, -0.0204479f, -0.000304646f,
|
|
0.033336f, -0.0157385f, -0.000505569f,
|
|
0.0403427f, -0.0104637f, -0.000693529f,
|
|
0.0469371f, -0.00477766f, -0.000865609f,
|
|
0.0530722f, 0.0011701f, -0.00102f,
|
|
-0.0121869f, -0.10317f, -0.0015868f,
|
|
-0.0096549f, -0.100606f, -0.00160377f,
|
|
-0.00743038f, -0.0973796f, -0.00160783f,
|
|
-0.00553901f, -0.0935261f, -0.00159792f,
|
|
-0.00400821f, -0.0890871f, -0.00157287f,
|
|
-0.0267803f, -0.0165823f, 0.000454173f,
|
|
-0.0348303f, -0.011642f, 0.000881271f,
|
|
-0.0424964f, -0.00870761f, 0.00129226f,
|
|
-0.049437f, -0.00781358f, 0.0016728f,
|
|
-0.0552635f, -0.00888708f, 0.00200659f,
|
|
-0.0629746f, -0.0721524f, -0.00160475f,
|
|
-0.0606813f, -0.0677576f, -0.00158427f,
|
|
-0.0582203f, -0.0625009f, -0.00154304f,
|
|
-0.0555686f, -0.0563905f, -0.00147822f,
|
|
-0.0526988f, -0.0494369f, -0.00138643f,
|
|
0.0385695f, 0.115704f, 0.00674413f,
|
|
0.056434f, 0.128273f, 0.00869052f,
|
|
0.0775564f, 0.137275f, 0.0110399f,
|
|
0.102515f, 0.140823f, 0.0138637f,
|
|
0.131458f, 0.136024f, 0.0171804f,
|
|
0.0595175f, -0.0845927f, 0.00512454f,
|
|
0.0506615f, -0.0680369f, 0.00376604f,
|
|
0.0434904f, -0.0503557f, 0.00261592f,
|
|
0.0376711f, -0.0318716f, 0.00163301f,
|
|
0.0329454f, -0.0128019f, 0.000785352f,
|
|
-0.0664062f, -0.0701094f, -0.00160644f,
|
|
-0.0641074f, -0.0658893f, -0.00158969f,
|
|
-0.0616054f, -0.0608302f, -0.00155303f,
|
|
-0.0588734f, -0.0549447f, -0.00149385f,
|
|
-0.0558797f, -0.0482482f, -0.00140906f,
|
|
0.0434062f, 0.102969f, 0.00581269f,
|
|
0.0619547f, 0.112838f, 0.00742057f,
|
|
0.0830229f, 0.118752f, 0.00927516f,
|
|
0.106603f, 0.119129f, 0.0113757f,
|
|
0.132073f, 0.111946f, 0.0136613f,
|
|
-0.0135758f, -0.0934604f, -0.000533868f,
|
|
-0.00690763f, -0.0958773f, -0.000598878f,
|
|
-0.000475275f, -0.0977838f, -0.000660985f,
|
|
0.00571866f, -0.0992032f, -0.0007201f,
|
|
0.0116724f, -0.10016f, -0.000776144f,
|
|
0.0651428f, -0.0850475f, -0.00120243f,
|
|
0.0682895f, -0.0823666f, -0.00121889f,
|
|
0.0712792f, -0.0795772f, -0.00123291f,
|
|
0.0741224f, -0.0766981f, -0.00124462f,
|
|
0.076829f, -0.0737465f, -0.00125416f,
|
|
0.10019f, -0.0375515f, -0.00121866f,
|
|
0.101296f, -0.0348723f, -0.00120216f,
|
|
0.102235f, -0.0323223f, -0.00118309f,
|
|
0.102994f, -0.0299234f, -0.00116131f,
|
|
0.103563f, -0.0276989f, -0.0011367f,
|
|
-0.00989236f, -0.0958821f, -0.000608883f,
|
|
-0.00344154f, -0.0980645f, -0.000673641f,
|
|
0.00277318f, -0.0997337f, -0.000735354f,
|
|
0.00874908f, -0.100914f, -0.000793927f,
|
|
0.0144843f, -0.101629f, -0.000849279f,
|
|
0.0654428f, -0.0839355f, -0.00125739f,
|
|
0.0684225f, -0.0810989f, -0.00127208f,
|
|
0.0712599f, -0.0781657f, -0.00128444f,
|
|
0.0739678f, -0.0751541f, -0.00129465f,
|
|
0.076558f, -0.0720804f, -0.00130286f,
|
|
-0.0132841f, -0.103948f, -0.00131159f,
|
|
-0.010344f, -0.102328f, -0.0013452f,
|
|
-0.00768637f, -0.100054f, -0.00136938f,
|
|
-0.00533293f, -0.0971572f, -0.00138324f,
|
|
-0.00330643f, -0.0936735f, -0.00138586f,
|
|
-0.0116984f, -0.0303752f, -0.000229102f,
|
|
-0.0149879f, -0.0265231f, -3.43823e-05f,
|
|
-0.0212917f, -0.0219544f, 0.000270283f,
|
|
-0.0277756f, -0.0186879f, 0.000582781f,
|
|
-0.0335115f, -0.0171098f, 0.00086919f,
|
|
0.0170095f, -0.025299f, -3.73557e-05f,
|
|
0.024552f, -0.0214351f, -0.000231975f,
|
|
0.0318714f, -0.0168568f, -0.000417463f,
|
|
0.0388586f, -0.0117131f, -0.000589883f,
|
|
0.0454388f, -0.00615626f, -0.000746594f,
|
|
-0.0160785f, -0.102675f, -0.00132891f,
|
|
-0.0133174f, -0.100785f, -0.00135859f,
|
|
-0.0108365f, -0.0982184f, -0.00137801f,
|
|
-0.00865931f, -0.0950053f, -0.00138614f,
|
|
-0.00681126f, -0.0911806f, -0.00138185f,
|
|
-0.0208973f, -0.0216631f, 0.000111231f,
|
|
-0.0289373f, -0.0151081f, 0.000512553f,
|
|
-0.0368736f, -0.0104306f, 0.000911793f,
|
|
-0.0444294f, -0.00773838f, 0.00129762f,
|
|
-0.0512663f, -0.00706554f, 0.00165611f
|
|
};
|
|
|
|
}
|
|
|
|
void TestStreamLineUniformGrid()
|
|
{
|
|
std::cout << "Testing StreamLineUniformGrid Filter" << std::endl;
|
|
|
|
typedef VTKM_DEFAULT_DEVICE_ADAPTER_TAG DeviceAdapter;
|
|
|
|
// Parameters for streamlines
|
|
vtkm::Id numSeeds = 5;
|
|
vtkm::Id maxSteps = 50;
|
|
vtkm::Float32 timeStep = 0.5f;
|
|
|
|
// Size of the dataset
|
|
const vtkm::Id3 vdims(5, 5, 5);
|
|
|
|
// Read vector data at each point of the uniform grid and store
|
|
vtkm::Id nElements = vdims[0] * vdims[1] * vdims[2] * 3;
|
|
|
|
std::vector<vtkm::Vec<vtkm::Float32, 3> > field;
|
|
for (vtkm::Id i = 0; i < nElements; i++)
|
|
{
|
|
vtkm::Float32 x = data[i];
|
|
vtkm::Float32 y = data[++i];
|
|
vtkm::Float32 z = data[++i];
|
|
vtkm::Vec<vtkm::Float32, 3> vecData(x, y, z);
|
|
field.push_back(Normalize(vecData));
|
|
}
|
|
vtkm::cont::ArrayHandle<vtkm::Vec<vtkm::Float32, 3> > fieldArray;
|
|
fieldArray = vtkm::cont::make_ArrayHandle(&field[0], nElements);
|
|
|
|
// Construct the input dataset (uniform) to hold the input and set vector data
|
|
vtkm::cont::DataSet inDataSet;
|
|
vtkm::cont::ArrayHandleUniformPointCoordinates coordinates(vdims);
|
|
inDataSet.AddCoordinateSystem(vtkm::cont::CoordinateSystem("coordinates", 1, coordinates));
|
|
inDataSet.AddField(vtkm::cont::Field("vecData", 1, vtkm::cont::Field::ASSOC_POINTS, fieldArray));
|
|
|
|
vtkm::cont::CellSetStructured<3> inCellSet("cells");
|
|
inCellSet.SetPointDimensions(vtkm::make_Vec(vdims[0], vdims[1], vdims[2]));
|
|
inDataSet.AddCellSet(inCellSet);
|
|
|
|
// Create and run the filter
|
|
vtkm::worklet::StreamLineFilterUniformGrid<vtkm::Float32, DeviceAdapter> streamLineFilter;
|
|
vtkm::cont::DataSet outDataSet = streamLineFilter.Run(inDataSet,
|
|
vtkm::worklet::internal::BOTH,
|
|
numSeeds,
|
|
maxSteps,
|
|
timeStep);
|
|
|
|
// Check output
|
|
vtkm::cont::CellSetExplicit<> &outCellSet =
|
|
outDataSet.GetCellSet(0).CastTo<vtkm::cont::CellSetExplicit<> >();
|
|
const vtkm::cont::DynamicArrayHandleCoordinateSystem &coordArray =
|
|
outDataSet.GetCoordinateSystem(0).GetData();
|
|
|
|
vtkm::Id numberOfCells = outCellSet.GetNumberOfCells();
|
|
vtkm::Id numberOfPoints = coordArray.GetNumberOfValues();
|
|
std::cout << "Number of polylines " << numberOfCells << std::endl;
|
|
std::cout << "Number of coordinates " << numberOfPoints << std::endl;
|
|
|
|
VTKM_TEST_ASSERT(test_equal(numberOfCells, numSeeds * 2),
|
|
"Wrong number of cells for stream lines");
|
|
}
|
|
|
|
int UnitTestStreamLineUniformGrid(int, char *[])
|
|
{
|
|
return vtkm::cont::testing::Testing::Run(TestStreamLineUniformGrid);
|
|
}
|