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
|
// SPDX-FileCopyrightText: Copyright (c) Kitware Inc.
// SPDX-License-Identifier: BSD-3-Clause
/**
The code for the toy simulation itself is here. In this example we
make a configurable number of random sized spheres, and make them
bounce around in a cube. At each timestep we move the spheres, and then
sample them onto a configurable sized volume. The points and the volume
are available to Catalyst pipelines, but in this example only the
volume is temporally cached.
Note: Interaction between spheres is left as an exercise to the reader.
*/
#ifndef FEDATASTRUCTURES_HEADER
#define FEDATASTRUCTURES_HEADER
#include <cstddef>
#include <vector>
class particle;
class region;
class Grid
{
// computational domain
public:
~Grid();
Grid();
void Initialize(const unsigned int numPoints[3], const double spacing[3]);
unsigned int GetNumberOfLocalPoints();
unsigned int GetNumberOfLocalCells();
void GetLocalPoint(unsigned int pointId, double* point);
unsigned int* GetNumPoints();
unsigned int* GetExtent();
double* GetSpacing();
region* GetMyRegion() { return MyRegion; };
private:
unsigned int NumPoints[3];
unsigned int Extent[6];
double Spacing[3];
region* MyRegion;
};
class Attributes
{
// array of results, updated every timestep
public:
~Attributes();
Attributes(int numParticles);
void Initialize(Grid* grid);
void UpdateFields(double time);
double* GetOccupancyArray();
const std::vector<double>& GetParticles();
private:
std::vector<double> Occupancy;
std::vector<double> Particles;
Grid* GridPtr;
particle** MyParticles;
int NumParticles;
};
#endif
|
