voxelization_toolkit/tests/test_hdf.cpp at 163d991cf25a1d049875a1b89806173925a752c8 · IfcOpenShell/voxelization_toolkit · GitHub

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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
#include "../voxelizer.h"
#include "../writer.h"
#include "../processor.h"

#include "H5Cpp.h"
#include <gtest/gtest.h>

#include <BRepBuilderAPI_MakePolygon.hxx>
#include <BRepBuilderAPI_MakeFace.hxx>
#include <BRepPrimAPI_MakeBox.hxx>
#include <BRepPrimAPI_MakeSphere.hxx>
#include <BRep_Builder.hxx>
#include <BRepBndLib.hxx>
#include <Bnd_Box.hxx>
#include <BRepMesh_IncrementalMesh.hxx>
#include <BRepTools.hxx>

#ifdef WITH_IFC
#include <ifcparse/IfcFile.h>
#ifdef IFCOPENSHELL_05
#include <ifcgeom/IfcGeomIterator.h>
using namespace Ifc2x3;
#else
#include <ifcgeom_schema_agnostic/IfcGeomIterator.h>
#endif
#endif


#ifdef WIN32
#define DIRSEP "\\"
#else
#define DIRSEP "/"
#endif


TEST(HdfFileName, HDF) {


	BRepPrimAPI_MakeBox mb(gp_Pnt(0, 0, 0), gp_Pnt(10, 10, 10));
	BRepPrimAPI_MakeSphere s(gp_Pnt(10, 10, 10), 7);

	auto x = s.Solid();

	auto storage2 = new chunked_voxel_storage<bit_t>( 0, 0, 0, 1, 32, 32, 32, 32);
	BRepTools breptools;

	breptools.Write(x, "sphere.brep");

	BRepMesh_IncrementalMesh(x, 0.001);
	auto vox = voxelizer(x, storage2);
	vox.Convert();

	auto storage = new chunked_voxel_storage<bit_t>(0., 0., 0., 0.1, 200, 150, 10, 32);

	{
		BRepBuilderAPI_MakePolygon mp(gp_Pnt(1, 1, 2), gp_Pnt(16, 1, 2), gp_Pnt(16, 9.8, 2), gp_Pnt(1, 9.8, 2), true);
		BRepBuilderAPI_MakeFace mf(mp.Wire());
		TopoDS_Face face = mf.Face();

		auto vox = voxelizer(face, storage);
		vox.Convert();
	}

	const double dim = 0.1;
	auto storage3 = new chunked_voxel_storage<bit_t>(-dim, -dim, -dim, dim, 100, 100, 100, 16);
	BRepPrimAPI_MakeBox make_box(8., 8., 8.);
	auto vox3 = voxelizer(make_box.Solid(), storage3);
	vox3.Convert();

	hdf_writer writer;
	writer.SetVoxels(storage3);
	writer.Write("multi_dim_vox.h5");

	//std::ofstream fs("voxobj.obj");
	//obj_export_helper oeh{ fs };
	//storage2->obj_export(oeh, false, false);

	// Write a 4D dataset

	const H5std_string  FILE_NAME("multidim.h5");
	const H5std_string  DATASET_NAME("continuous_chunks");
	const int   NX = 32;                    // dataset dimensions
	const int   NY = 32;
	const int   NZ = 32;
	const int   NC = 3;

	const int   RANK = 4;
	H5::H5File file(FILE_NAME, H5F_ACC_TRUNC);

	hsize_t     dimsf[4];              // dataset dimensions
	dimsf[0] = NC;
	dimsf[1] = NX;
	dimsf[2] = NY;
	dimsf[3] = NZ;

	H5::DataSpace dataspace(RANK, dimsf);

	H5::IntType datatype(H5::PredType::NATIVE_INT);
	datatype.setOrder(H5T_ORDER_LE);

	H5::DataSet dataset = file.createDataSet(DATASET_NAME, datatype, dataspace);

	//std::vector<int> data;

	//for (int i = 0; i < NX*NY*NZ*NC; i++) {
	//	data.push_back(0);
	//}

	//dataset.write(data.data(), H5::PredType::NATIVE_INT);

}