mirror of
https://gitlab.kitware.com/vtk/vtk-m
synced 2024-09-08 13:23:51 +00:00
With the major revision 2.0 of VTK-m, many items previously marked as deprecated were removed. If updating to a new version of VTK-m, it is recommended to first update to VTK-m 1.9, which will include the deprecated features but provide warnings (with the right compiler) that will point to the replacement code. Once the deprecations have been fixed, updating to 2.0 should be smoother.
332 lines
11 KiB
C++
332 lines
11 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 <algorithm>
|
|
#include <cctype>
|
|
#include <iostream>
|
|
|
|
#include <vtkm/Math.h>
|
|
#include <vtkm/cont/ArrayHandle.h>
|
|
#include <vtkm/cont/DataSetBuilderUniform.h>
|
|
#include <vtkm/cont/Initialize.h>
|
|
|
|
#include <vtkm/source/Oscillator.h>
|
|
|
|
#if !defined(_WIN32) || defined(__CYGWIN__)
|
|
#include <unistd.h> /* unlink */
|
|
#else
|
|
#include <io.h> /* unlink */
|
|
#endif
|
|
|
|
//Suppress warnings about glut being deprecated on OSX
|
|
#if (defined(VTKM_GCC) || defined(VTKM_CLANG))
|
|
#pragma GCC diagnostic push
|
|
#pragma GCC diagnostic ignored "-Wdeprecated-declarations"
|
|
#endif
|
|
|
|
|
|
// trim() from http://stackoverflow.com/a/217605/44738
|
|
static inline std::string& ltrim(std::string& s)
|
|
{
|
|
s.erase(s.begin(), std::find_if(s.begin(), s.end(), [](int ch) { return !std::isspace(ch); }));
|
|
return s;
|
|
}
|
|
static inline std::string& rtrim(std::string& s)
|
|
{
|
|
s.erase(std::find_if(s.rbegin(), s.rend(), [](int ch) { return !std::isspace(ch); }).base(),
|
|
s.end());
|
|
return s;
|
|
}
|
|
static inline std::string& trim(std::string& s)
|
|
{
|
|
return ltrim(rtrim(s));
|
|
}
|
|
|
|
// ----------------------------------------------------------------------------
|
|
|
|
void read_oscillators(std::string filePath, vtkm::source::Oscillator& source)
|
|
{
|
|
std::ifstream in(filePath);
|
|
if (!in)
|
|
throw std::runtime_error("Unable to open " + filePath);
|
|
std::string line;
|
|
while (std::getline(in, line))
|
|
{
|
|
line = trim(line);
|
|
if (line.empty() || line[0] == '#')
|
|
continue;
|
|
std::istringstream iss(line);
|
|
|
|
std::string stype;
|
|
iss >> stype;
|
|
|
|
int x, y, z;
|
|
iss >> x >> y >> z;
|
|
|
|
float r, omega0, zeta = 0.0f;
|
|
iss >> r >> omega0;
|
|
|
|
if (stype == "damped")
|
|
{
|
|
iss >> zeta;
|
|
}
|
|
|
|
if (stype == "damped")
|
|
{
|
|
source.AddDamped(x, y, z, r, omega0, zeta);
|
|
}
|
|
else if (stype == "decaying")
|
|
{
|
|
source.AddDecaying(x, y, z, r, omega0, zeta);
|
|
}
|
|
else if (stype == "periodic")
|
|
{
|
|
source.AddPeriodic(x, y, z, r, omega0, zeta);
|
|
}
|
|
}
|
|
}
|
|
|
|
// ----------------------------------------------------------------------------
|
|
// ArcticViewer helper
|
|
// ----------------------------------------------------------------------------
|
|
|
|
void writeData(std::string& basePath,
|
|
int timestep,
|
|
int iSize,
|
|
int jSize,
|
|
int kSize,
|
|
const double* values)
|
|
{
|
|
int size = iSize * jSize * kSize;
|
|
std::ostringstream timeValues;
|
|
for (int t = 0; t <= timestep; t++)
|
|
{
|
|
if (t > 0)
|
|
{
|
|
timeValues << ", ";
|
|
}
|
|
timeValues << "\"" << t << "\"";
|
|
}
|
|
|
|
// Root metadata file
|
|
std::ostringstream metaFilePath;
|
|
metaFilePath << basePath.c_str() << "/index.json";
|
|
std::ofstream metaFilePathPointer(metaFilePath.str().c_str(), std::ios::out);
|
|
if (metaFilePathPointer.fail())
|
|
{
|
|
std::cout << "Unable to open file: " << metaFilePath.str().c_str() << std::endl;
|
|
}
|
|
else
|
|
{
|
|
metaFilePathPointer << "{\n"
|
|
<< " \"metadata\": {\"backgroundColor\": \"#000000\"},\n"
|
|
<< " \"type\": [\"tonic-query-data-model\", \"vtk-volume\"],\n"
|
|
<< " \"arguments_order\": [\"time\"],\n"
|
|
<< " \"arguments\": {\n"
|
|
<< " \"time\": {\n"
|
|
<< " \"loop\": \"modulo\",\n"
|
|
<< " \"ui\": \"slider\",\n"
|
|
<< " \"values\": [" << timeValues.str().c_str() << "]\n"
|
|
<< " }\n"
|
|
<< " },\n"
|
|
<< " \"data\": [{\n"
|
|
<< " \"pattern\": \"{time}.json\",\n"
|
|
<< " \"rootFile\": true,\n"
|
|
<< " \"name\": \"scene\",\n"
|
|
<< " \"type\": \"json\"\n"
|
|
<< " }]\n"
|
|
<< "}\n";
|
|
metaFilePathPointer.flush();
|
|
metaFilePathPointer.close();
|
|
}
|
|
|
|
// Current data file
|
|
std::ostringstream dataFilePath;
|
|
dataFilePath << basePath.c_str() << "/" << timestep << ".data";
|
|
std::ofstream dataFilePathPointer(dataFilePath.str().c_str(), std::ios::out | std::ios::binary);
|
|
if (dataFilePathPointer.fail())
|
|
{
|
|
std::cout << "Unable to open file: " << dataFilePath.str().c_str() << std::endl;
|
|
}
|
|
else
|
|
{
|
|
int stackSize = size * 8;
|
|
dataFilePathPointer.write(reinterpret_cast<const char*>(values), stackSize);
|
|
dataFilePathPointer.flush();
|
|
dataFilePathPointer.close();
|
|
}
|
|
|
|
// Current dataset meta file
|
|
std::ostringstream dataMetaFilePath;
|
|
dataMetaFilePath << basePath.c_str() << "/" << timestep << ".json";
|
|
std::ofstream dataMetaFilePathPointer(dataMetaFilePath.str().c_str(), std::ios::out);
|
|
if (dataMetaFilePathPointer.fail())
|
|
{
|
|
std::cout << "Unable to open file: " << dataMetaFilePath.str().c_str() << std::endl;
|
|
}
|
|
else
|
|
{
|
|
dataMetaFilePathPointer << "{\n"
|
|
<< " \"origin\": [0,0,0],\n"
|
|
<< " \"spacing\": [1,1,1],\n"
|
|
<< " \"extent\": [\n"
|
|
<< " 0, " << (iSize - 1) << ",\n"
|
|
<< " 0, " << (jSize - 1) << ",\n"
|
|
<< " 0, " << (kSize - 1) << "],\n"
|
|
<< " \"vtkClass\": \"vtkImageData\",\n"
|
|
<< " \"pointData\": {\n"
|
|
<< " \"vtkClass\": \"vtkDataSetAttributes\",\n"
|
|
<< " \"arrays\": [{\n"
|
|
<< " \"data\": {\n"
|
|
<< " \"numberOfComponents\": 1,\n"
|
|
<< " \"name\": \"oscillation\",\n"
|
|
<< " \"vtkClass\": \"vtkDataArray\",\n"
|
|
<< " \"dataType\": \"Float64Array\",\n"
|
|
<< " \"ref\": {\n"
|
|
<< " \"registration\": \"setScalars\",\n"
|
|
<< " \"encode\": \"LittleEndian\",\n"
|
|
<< " \"basepath\": \"\",\n"
|
|
<< " \"id\": \"" << timestep << ".data\"\n"
|
|
<< " },\n"
|
|
<< " \"size\": " << size << std::endl
|
|
<< " }\n"
|
|
<< " }]\n"
|
|
<< " }\n"
|
|
<< "}\n";
|
|
dataMetaFilePathPointer.flush();
|
|
dataMetaFilePathPointer.close();
|
|
}
|
|
}
|
|
|
|
// ----------------------------------------------------------------------------
|
|
|
|
void printUsage(const std::string& vtkm_options)
|
|
{
|
|
std::cout << "Usage: Oscillator [options]\n\n"
|
|
<< "Options:\n\n"
|
|
<< " -s, --shape POINT domain shape [default: 64x64x64]\n"
|
|
<< " -t, --dt FLOAT time step [default: 0.01]\n"
|
|
<< " -f, --config STRING oscillator file (required). Available in "
|
|
"vtk-m/examples/oscillator/inputs\n"
|
|
<< " --t-end FLOAT end time [default: 10]\n"
|
|
<< " -o, --output STRING directory to output data\n"
|
|
<< "General VTK-m Options:\n\n"
|
|
<< vtkm_options << std::endl;
|
|
}
|
|
|
|
// ----------------------------------------------------------------------------
|
|
|
|
int main(int argc, char** argv)
|
|
{
|
|
std::string oscillatorConfigFile = "";
|
|
std::string outputDirectory = "";
|
|
int sizeX = 64;
|
|
int sizeY = 64;
|
|
int sizeZ = 64;
|
|
float startTime = 0.0f;
|
|
float endTime = 10.0f;
|
|
float deltaTime = 0.01f;
|
|
float currentTime = startTime;
|
|
bool generateOutput = false;
|
|
|
|
// Process vtk-m general args
|
|
auto opts = vtkm::cont::InitializeOptions::DefaultAnyDevice;
|
|
auto initializeResults = vtkm::cont::Initialize(argc, argv, opts);
|
|
// Process args
|
|
int nbOptions = argc - 1;
|
|
for (int i = 1; i < nbOptions; i += 2)
|
|
{
|
|
// Config (REQUIRED)
|
|
if (!strcmp(argv[i], "-f") || !strcmp(argv[i], "--config"))
|
|
{
|
|
oscillatorConfigFile = argv[i + 1];
|
|
}
|
|
// Output
|
|
if (!strcmp(argv[i], "-o") || !strcmp(argv[i], "--output"))
|
|
{
|
|
generateOutput = true;
|
|
outputDirectory = argv[i + 1];
|
|
}
|
|
|
|
// Shape
|
|
if (!strcmp(argv[i], "-s") || !strcmp(argv[i], "--shape"))
|
|
{
|
|
std::string shape = argv[i + 1];
|
|
size_t found = shape.find("x");
|
|
sizeX = sizeY = sizeZ = std::stoi(shape);
|
|
if (found != std::string::npos)
|
|
{
|
|
shape = shape.substr(found + 1);
|
|
sizeY = sizeZ = std::stoi(shape);
|
|
found = shape.find("x");
|
|
if (found != std::string::npos)
|
|
{
|
|
shape = shape.substr(found + 1);
|
|
sizeZ = std::stoi(shape);
|
|
found = shape.find("x");
|
|
}
|
|
}
|
|
}
|
|
|
|
// Time
|
|
if (!strcmp(argv[i], "-t") || !strcmp(argv[i], "--dt"))
|
|
{
|
|
deltaTime = float(std::atof(argv[i + 1]));
|
|
}
|
|
|
|
// End-Time
|
|
if (!strcmp(argv[i], "--t-end"))
|
|
{
|
|
endTime = float(std::atof(argv[i + 1]));
|
|
}
|
|
}
|
|
|
|
if (oscillatorConfigFile.size() < 2)
|
|
{
|
|
printUsage(initializeResults.Usage);
|
|
return 0;
|
|
}
|
|
|
|
std::cout << "\n=========== configuration ============" << std::endl;
|
|
std::cout << " oscillator config: " << oscillatorConfigFile.c_str() << std::endl;
|
|
std::cout << " mesh size: " << sizeX << "x" << sizeY << "x" << sizeZ << std::endl;
|
|
std::cout << " time handling:" << std::endl;
|
|
std::cout << " - dt: " << deltaTime << std::endl;
|
|
std::cout << " - end: " << endTime << std::endl;
|
|
std::cout << "=======================================\n" << std::endl;
|
|
|
|
vtkm::source::Oscillator source(vtkm::Id3{ sizeX, sizeY, sizeZ });
|
|
read_oscillators(oscillatorConfigFile, source);
|
|
|
|
std::cout << "=========== start computation ============" << std::endl;
|
|
int count = 0;
|
|
while (currentTime < endTime)
|
|
{
|
|
source.SetTime(currentTime);
|
|
vtkm::cont::DataSet rdata = source.Execute();
|
|
if (generateOutput)
|
|
{
|
|
vtkm::cont::ArrayHandleBasic<vtkm::Float64> tmp;
|
|
rdata.GetField("oscillating", vtkm::cont::Field::Association::Points)
|
|
.GetData()
|
|
.AsArrayHandle(tmp);
|
|
const double* values = tmp.GetReadPointer();
|
|
writeData(outputDirectory, count++, sizeX, sizeY, sizeZ, values);
|
|
}
|
|
|
|
std::cout << "Compute time " << currentTime << std::endl;
|
|
currentTime += deltaTime;
|
|
}
|
|
std::cout << "=========== computation done ============" << std::endl;
|
|
}
|
|
|
|
#if (defined(VTKM_GCC) || defined(VTKM_CLANG))
|
|
#pragma GCC diagnostic pop
|
|
#endif
|