#ifndef OUTPUT_VTKHDF_HELPERS_POPULATE_H
#define OUTPUT_VTKHDF_HELPERS_POPULATE_H
#ifdef HAVE_HDF5
/** ### Populate points_dset_box for vtkhdf (marker stores local indices) */
void populate_points_dset_box_vtkhdf(vertex scalar mask, vertex scalar marker, double **points_dset, long num_points, long *offset_points, hsize_t *count, hsize_t *offset) {
// Each process defines dataset in memory and writes to an hyperslab
count[0] = num_points;
count[1] = 3;
offset[0] = 0;
offset[1] = 0;
if (pid() != 0){
for (int i = 1; i <= pid(); ++i){
offset[0] += offset_points[i - 1];
}
}
// Allocate memory for points_dset
*points_dset = (double *)malloc(count[0] * count[1] * sizeof(double));
// Iterate over each vertex
foreach_vertex(serial, noauto){
if (mask[] >= 0.5){
long ii = marker[] * 3;
// Store coordinates
(*points_dset)[ii + 0] = x;
(*points_dset)[ii + 1] = y;
#if dimension == 2
(*points_dset)[ii + 2] = 0.;
#else
(*points_dset)[ii + 2] = z;
#endif
}
}
}
/** ### Populate types_dset */
void populate_types_dset(char **types_dset, char type, long num_cells, long *offset_cells, hsize_t *count, hsize_t *offset) {
// Each process defines dataset in memory and writes to an hyperslab
count[0] = num_cells;
count[1] = 1;
offset[0] = 0;
offset[1] = 0;
if (pid() != 0){
for (int i = 1; i <= pid(); ++i){
offset[0] += offset_cells[i - 1];
}
}
// Allocate memory for types_dset
*types_dset = (char *)malloc(count[0] * count[1] * sizeof(char));
for (long i = 0; i < num_cells; ++i){
(*types_dset)[i] = type;
}
}
/** ### Populate offsets_dset */
void populate_offsets_dset(long **offsets_dset, char noffset, long num_cells, long *offset_cells, hsize_t *count, hsize_t *offset) {
// Each process defines dataset in memory and writes to an hyperslab
count[0] = num_cells;
count[1] = 1;
offset[0] = 0;
offset[1] = 0;
if (pid() != 0){
for (int i = 1; i <= pid(); ++i){
offset[0] += offset_cells[i - 1];
}
}
// Allocate memory for topo_dset
*offsets_dset = (long *)malloc(count[0] * count[1] * sizeof(long));
for (long i = 0; i < num_cells; ++i){
(*offsets_dset)[i] = (long)i * (long)noffset;
}
}
/** ### Populate topo_dset for vtkhdf (flattened 1D connectivity) */
void populate_topo_dset_vtkhdf(long **topo_dset, long num_cells, long *offset_cells, hsize_t *count, hsize_t *offset, scalar per_mask, vertex scalar marker) {
// Each process defines dataset in memory and writes to an hyperslab
count[0] = num_cells;
count[1] = pow(2, dimension);
offset[0] = 0;
offset[1] = 0;
if (pid() != 0){
for (int i = 1; i <= pid(); ++i){
offset[0] += offset_cells[i - 1];
}
}
// Allocate memory for topo_dset
*topo_dset = (long *)malloc(count[0] * count[1] * sizeof(long));
// Iterate over each cell
long ii = 0;
foreach (serial, noauto){
if (per_mask[]){
// Calculate starting index for topo_dset
long ij = ii * count[1];
// Assign marker values to topo_dset
(*topo_dset)[ij + 0] = (long)marker[];
(*topo_dset)[ij + 1] = (long)marker[1, 0];
(*topo_dset)[ij + 2] = (long)marker[1, 1];
(*topo_dset)[ij + 3] = (long)marker[0, 1];
#if dimension == 3
// Additional assignments for 3D
(*topo_dset)[ij + 4] = (long)marker[0, 0, 1];
(*topo_dset)[ij + 5] = (long)marker[1, 0, 1];
(*topo_dset)[ij + 6] = (long)marker[1, 1, 1];
(*topo_dset)[ij + 7] = (long)marker[0, 1, 1];
#endif
ii++;
}
}
count[0] = num_cells*(long)pow(2, dimension);
count[1] = 1;
}
void populate_topo_dset_slice_vtkhdf(long **topo_dset, long num_cells, long *offset_cells, hsize_t *count,
hsize_t *offset, scalar per_mask, vertex scalar marker, coord n = {0, 0, 1}, double _alpha = 0)
{
// Each process defines dataset in memory and writes to an hyperslab
count[0] = num_cells;
count[1] = pow(2, dimension - 1);
offset[0] = 0;
offset[1] = 0;
if (pid() != 0){
for (int i = 1; i <= pid(); ++i){
offset[0] += offset_cells[i - 1];
}
}
// Allocate memory for topo_dset
*topo_dset = (long *)malloc(count[0] * count[1] * sizeof(long));
// Iterate over each cell
long ii = 0;
foreach (serial, noauto){
if (per_mask[]){
// Calculate index
long ij = ii * count[1];
if (n.x == 1){
(*topo_dset)[ij + 0] = (long)marker[1, 0, 0];
(*topo_dset)[ij + 1] = (long)marker[1, 1, 0];
(*topo_dset)[ij + 2] = (long)marker[1, 1, 1];
(*topo_dset)[ij + 3] = (long)marker[1, 0, 1];
}
else if (n.y == 1){
(*topo_dset)[ij + 0] = (long)marker[0, 1, 0];
(*topo_dset)[ij + 1] = (long)marker[1, 1, 0];
(*topo_dset)[ij + 2] = (long)marker[1, 1, 1];
(*topo_dset)[ij + 3] = (long)marker[0, 1, 1];
}
else{
(*topo_dset)[ij + 0] = (long)marker[0, 0, 1];
(*topo_dset)[ij + 1] = (long)marker[1, 0, 1];
(*topo_dset)[ij + 2] = (long)marker[1, 1, 1];
(*topo_dset)[ij + 3] = (long)marker[0, 1, 1];
}
ii++;
}
}
count[0] = ii*(long)pow(2, dimension - 1);
count[1] = 1;
}
#endif // HAVE_HDF5
#endif // OUTPUT_VTKHDF_HELPERS_POPULATE_H
