mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-09-08 13:23:51 +00:00
119 lines
4.6 KiB
C++
119 lines
4.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 <cstdlib>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include <vtkm/cont/DataSet.h>
|
|
#include <vtkm/cont/Initialize.h>
|
|
#include <vtkm/cont/Timer.h>
|
|
|
|
#include <vtkm/filter/Pathline.h>
|
|
|
|
#include <vtkm/io/VTKDataSetReader.h>
|
|
#include <vtkm/io/VTKDataSetWriter.h>
|
|
|
|
int main(int argc, char** argv)
|
|
{
|
|
auto opts = vtkm::cont::InitializeOptions::DefaultAnyDevice;
|
|
auto config = vtkm::cont::Initialize(argc, argv, opts);
|
|
|
|
// Sample data to use this example can be found in the data directory of the
|
|
// VTK-m repo in the location temporal_datasets.
|
|
// These example with these datasets can be used for this example as :
|
|
// ./Temporal_Advection DoubleGyre_0.vtk 0.0 DoubleGyre_5.vtk 5.0
|
|
// velocity 500 0.025 pathlines.vtk
|
|
if (argc < 8)
|
|
{
|
|
std::cout
|
|
<< "Usage:\n"
|
|
<< "./Temporal_Advection slice1 time1 slice2 time2 field num_steps step_size output\n"
|
|
<< "\tslice1 : Time slice 1, sample data in vtk-m/data/data/rectilinear/Double_Gyre0.vtk\n"
|
|
<< "\ttime1 : simulation time for slice 1, for sample data use 0.0\n"
|
|
<< "\tslice2 : Time slice 2, sample data in vtk-m/data/data/rectilinear/Double_Gyre5.vtk\n"
|
|
<< "\ttime2 : simulation time for slice 2, for sample data use 5.0\n"
|
|
<< "\tfield : active velocity field in the data set, for sample data use 'velocity'\n"
|
|
<< "\tnum_steps : maximum number of steps for advection, for sample data use 500\n"
|
|
<< "\tstep_size : the size of a single step during advection, for sample data use 0.025\n"
|
|
<< "\toutput : the name of the output file" << std::endl;
|
|
std::cout << "Example:\n";
|
|
std::cout << "./Temporal_Advection ~/vtk-m/data/data/rectilinear/DoubleGyre_0.vtk 0.0 "
|
|
"~/vtk-m/data/data/rectilinear/DoubleGyre_5.vtk 5.0 velocity 500 0.025 "
|
|
"pathlines.vtk\n";
|
|
exit(EXIT_FAILURE);
|
|
}
|
|
|
|
std::string fieldName, datasetName1, datasetName2, outputName;
|
|
vtkm::FloatDefault time1, time2;
|
|
|
|
vtkm::Id numSteps;
|
|
vtkm::Float32 stepSize;
|
|
|
|
datasetName1 = std::string(argv[1]);
|
|
time1 = static_cast<vtkm::FloatDefault>(atof(argv[2]));
|
|
datasetName2 = std::string(argv[3]);
|
|
time2 = static_cast<vtkm::FloatDefault>(atof(argv[4]));
|
|
fieldName = std::string(argv[5]);
|
|
numSteps = atoi(argv[6]);
|
|
stepSize = static_cast<vtkm::Float32>(atof(argv[7]));
|
|
outputName = std::string(argv[8]);
|
|
|
|
vtkm::io::VTKDataSetReader reader1(datasetName1);
|
|
vtkm::cont::DataSet ds1 = reader1.ReadDataSet();
|
|
|
|
vtkm::io::VTKDataSetReader reader2(datasetName2);
|
|
vtkm::cont::DataSet ds2 = reader2.ReadDataSet();
|
|
|
|
// Use the coordinate system as seeds for performing advection
|
|
vtkm::cont::ArrayHandle<vtkm::Vec3f> pts;
|
|
vtkm::cont::ArrayCopy(ds1.GetCoordinateSystem().GetData(), pts);
|
|
vtkm::cont::ArrayHandle<vtkm::Particle> seeds;
|
|
|
|
vtkm::Id numPts = pts.GetNumberOfValues();
|
|
seeds.Allocate(numPts);
|
|
auto ptsPortal = pts.ReadPortal();
|
|
auto seedPortal = seeds.WritePortal();
|
|
for (vtkm::Id i = 0; i < numPts; i++)
|
|
{
|
|
vtkm::Particle p;
|
|
p.Pos = ptsPortal.Get(i);
|
|
p.ID = i;
|
|
seedPortal.Set(i, p);
|
|
}
|
|
|
|
// Instantiate the filter by providing necessary parameters.
|
|
// Necessary parameters are :
|
|
vtkm::filter::Pathline pathlineFilter;
|
|
pathlineFilter.SetActiveField(fieldName);
|
|
// 1. The current and next time slice. The current time slice is passed
|
|
// through the parameter to the Execute method.
|
|
pathlineFilter.SetNextDataSet(ds2);
|
|
// 2. The current and next times, these times will be used to interpolate
|
|
// the velocities for particle positions in space and time.
|
|
pathlineFilter.SetPreviousTime(time1);
|
|
pathlineFilter.SetNextTime(time2);
|
|
// 3. Maximum number of steps the particle is allowed to take until termination.
|
|
pathlineFilter.SetNumberOfSteps(numSteps);
|
|
// 4. Length for each step.
|
|
pathlineFilter.SetStepSize(stepSize);
|
|
// 5. Seeds for advection.
|
|
pathlineFilter.SetSeeds(seeds);
|
|
|
|
vtkm::cont::DataSet output = pathlineFilter.Execute(ds1);
|
|
|
|
// The way to verify if the code produces correct streamlines
|
|
// is to do a visual test by using VisIt/ParaView to visualize
|
|
// the file written by this method.
|
|
vtkm::io::VTKDataSetWriter writer(outputName);
|
|
writer.WriteDataSet(output);
|
|
return 0;
|
|
}
|