1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
|
#include "FEDataStructures.h"
#include <mpi.h>
#include <string>
#include <vector>
#ifdef USE_CATALYST
#include "FEAdaptor.h"
#endif
#include <cstdlib>
#include <cstring>
// Example of a C++ adaptor for a simulation code
// where the simulation code has a fixed topology
// grid. We treat the grid as an unstructured
// grid even though in the example provided it
// would be best described as a vtkImageData.
// Also, the points are stored in an inconsistent
// manner with respect to the velocity vector.
// This is purposefully done to demonstrate
// the different approaches for getting data
// into Catalyst. The simulation can be run
// from a restarted time step with the
// -- restart <time step> command line argument.
// All other arguments are considered to be input
// script. Note that through configuration
// that the driver can be run without linking
// to Catalyst.
int main(int argc, char* argv[])
{
MPI_Init(&argc, &argv);
Grid grid;
unsigned int numPoints[3] = { 70, 60, 44 };
double spacing[3] = { 1, 1.1, 1.3 };
grid.Initialize(numPoints, spacing);
Attributes attributes;
attributes.Initialize(&grid);
// we are doing a restarted simulation
unsigned int startTimeStep = 0;
std::vector<std::string> scripts;
for (int i = 1; i < argc; i++)
{
if (strcmp(argv[i], "--restart") == 0)
{
if (i + 1 < argc)
{
startTimeStep = static_cast<unsigned int>(atoi(argv[2]));
i++;
}
}
else
{
scripts.push_back(argv[i]);
}
}
#ifdef USE_CATALYST
FEAdaptor::Initialize(scripts);
#endif
unsigned int numberOfTimeSteps = 50;
for (unsigned int timeStep = startTimeStep; timeStep <= startTimeStep + numberOfTimeSteps;
timeStep++)
{
// use a time step length of 0.018
double time = timeStep * 0.018;
attributes.UpdateFields(time);
#ifdef USE_CATALYST
FEAdaptor::CoProcess(
grid, attributes, time, timeStep, timeStep == numberOfTimeSteps + startTimeStep);
#endif
}
#ifdef USE_CATALYST
FEAdaptor::Finalize();
#endif
MPI_Finalize();
return 0;
}
|
