-
Notifications
You must be signed in to change notification settings - Fork 8
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Implemented test data generation using C++ and libE57format
- Loading branch information
Showing
7 changed files
with
228 additions
and
0 deletions.
There are no files selected for viewing
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,2 @@ | ||
/vcpkg | ||
/build |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,5 @@ | ||
cmake_minimum_required(VERSION 3.27) | ||
project(e57-testdata-generator) | ||
find_package(E57Format CONFIG REQUIRED) | ||
add_executable(main main.cpp) | ||
target_link_libraries(main PRIVATE E57Format) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,4 @@ | ||
This folder contains some C++ code that uses the libE57Format library. | ||
The code was used to generate test data for the Rust integration tests. | ||
Its using CMake for building and VCPKG for resolving all dependencies. | ||
A simple Powershell build script for Windows is also included. |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,8 @@ | ||
if (Test-Path build) { Remove-Item -Recurse -Force build } | ||
if (Test-Path vcpkg) { Remove-Item -Recurse -Force vcpkg } | ||
git clone https://github.com/Microsoft/vcpkg.git | ||
./vcpkg/bootstrap-vcpkg.bat | ||
mkdir build | out-null | ||
cmake -B ./build -S . -DCMAKE_TOOLCHAIN_FILE="./vcpkg/scripts/buildsystems/vcpkg.cmake" | ||
cmake --build ./build --config Release | ||
./build/Release/main.exe |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,201 @@ | ||
#include <iostream> | ||
#include <fstream> | ||
|
||
#include <E57SimpleReader.h> | ||
#include <E57SimpleWriter.h> | ||
|
||
void empty() { | ||
e57::Writer writer("empty.e57"); | ||
if (!writer.Close()) throw std::string("Failed to close empty.e57"); | ||
} | ||
|
||
void tiny_pc() { | ||
e57::Writer writer("tiny_pc.e57"); | ||
e57::DateTime time; | ||
time.isAtomicClockReferenced = 1; | ||
time.dateTimeValue = 1.23; | ||
e57::Data3D header; | ||
header.guid = "guid"; | ||
header.name = "name"; | ||
header.description = "desc"; | ||
header.sensorFirmwareVersion = "fw"; | ||
header.sensorHardwareVersion = "hw"; | ||
header.sensorSoftwareVersion = "sw"; | ||
header.sensorModel = "model"; | ||
header.sensorVendor = "vendor"; | ||
header.sensorSerialNumber = "serial"; | ||
header.relativeHumidity = 99; | ||
header.temperature = 20; | ||
header.acquisitionStart = time; | ||
header.acquisitionEnd = time; | ||
header.pointCount = 1; | ||
header.pointFields.cartesianXField = true; | ||
header.pointFields.cartesianYField = true; | ||
header.pointFields.cartesianZField = true; | ||
e57::Data3DPointsFloat buffers(header); | ||
buffers.cartesianX[0] = 1; | ||
buffers.cartesianY[0] = 2; | ||
buffers.cartesianZ[0] = 3; | ||
writer.WriteData3DData(header, buffers); | ||
if (!writer.Close()) throw std::string("Failed to close tiny_pc.e57"); | ||
} | ||
|
||
void tiny_pc_with_extension() { | ||
e57::Writer writer("tiny_pc_with_extension.e57"); | ||
e57::Data3D header; | ||
header.pointCount = 1; | ||
header.pointFields.cartesianXField = true; | ||
header.pointFields.cartesianYField = true; | ||
header.pointFields.cartesianZField = true; | ||
header.pointFields.pointRangeNodeType = e57::NumericalNodeType::Double; | ||
header.pointFields.normalXField = true; | ||
header.pointFields.normalYField = true; | ||
header.pointFields.normalZField = true; | ||
e57::Data3DPointsDouble buffers(header); | ||
buffers.cartesianX[0] = 1; | ||
buffers.cartesianY[0] = 2; | ||
buffers.cartesianZ[0] = 3; | ||
buffers.normalX[0] = 1; | ||
buffers.normalY[0] = 0; | ||
buffers.normalZ[0] = 0; | ||
writer.WriteData3DData(header, buffers); | ||
if (!writer.Close()) throw std::string("Failed to close tiny_pc_with_extension.e57"); | ||
} | ||
|
||
void empty_pc() { | ||
e57::Writer writer("empty_pc.e57"); | ||
e57::Data3D header; | ||
header.pointCount = 0; | ||
header.pointFields.cartesianXField = true; | ||
header.pointFields.cartesianYField = true; | ||
header.pointFields.cartesianZField = true; | ||
e57::Data3DPointsFloat buffers; | ||
buffers.cartesianX = new float[0]; | ||
buffers.cartesianY = new float[0]; | ||
buffers.cartesianZ = new float[0]; | ||
writer.WriteData3DData(header, buffers); | ||
if (!writer.Close()) throw std::string("Failed to close empty_pc.e57"); | ||
} | ||
|
||
void tiny_pc_and_images() { | ||
e57::Writer writer("tiny_pc_and_images.e57"); | ||
|
||
e57::Data3D header; | ||
header.pointCount = 2; | ||
header.pointFields.cartesianXField = true; | ||
header.pointFields.cartesianYField = true; | ||
header.pointFields.cartesianZField = true; | ||
e57::Data3DPointsFloat buffers(header); | ||
buffers.cartesianX[0] = 0; | ||
buffers.cartesianY[0] = 0; | ||
buffers.cartesianZ[0] = 0; | ||
buffers.cartesianX[1] = 1; | ||
buffers.cartesianY[1] = 1; | ||
buffers.cartesianZ[1] = 1; | ||
writer.WriteData3DData(header, buffers); | ||
|
||
std::vector<uint8_t> jpegData; | ||
{ | ||
std::ifstream ifs("../castle.jpg", std::ios::in | std::ios::binary); | ||
if (!ifs) throw std::string("Cannot open JPEG file"); | ||
ifs.seekg(0, ifs.end); | ||
auto length = ifs.tellg(); | ||
ifs.seekg(0, ifs.beg); | ||
jpegData.resize(length); | ||
ifs.read((char*)jpegData.data(), length); | ||
} | ||
|
||
std::vector<uint8_t> pngData; | ||
{ | ||
std::ifstream ifs("../square.png", std::ios::in | std::ios::binary); | ||
if (!ifs) throw std::string("Cannot open PNG file"); | ||
ifs.seekg(0, ifs.end); | ||
auto length = ifs.tellg(); | ||
ifs.seekg(0, ifs.beg); | ||
pngData.resize(length); | ||
ifs.read((char*)pngData.data(), length); | ||
} | ||
|
||
e57::Image2D visImg; | ||
visImg.name = "visual"; | ||
visImg.visualReferenceRepresentation.imageHeight = 100; | ||
visImg.visualReferenceRepresentation.imageWidth = 100; | ||
visImg.visualReferenceRepresentation.jpegImageSize = jpegData.size(); | ||
writer.WriteImage2DData(visImg, e57::Image2DType::ImageJPEG, e57::Image2DProjection::ProjectionVisual, 0, jpegData.data(), jpegData.size()); | ||
|
||
e57::Image2D sphImg; | ||
sphImg.name = "spherical"; | ||
sphImg.sensorModel = "sensor"; | ||
sphImg.sensorSerialNumber = "serial"; | ||
sphImg.sensorVendor = "vendor"; | ||
sphImg.associatedData3DGuid = header.guid; | ||
sphImg.description = "desc"; | ||
sphImg.pose.rotation.x = 1; | ||
sphImg.pose.rotation.y = 0; | ||
sphImg.pose.rotation.z = 0; | ||
sphImg.pose.rotation.w = 0.5; | ||
sphImg.pose.translation.x = 1; | ||
sphImg.pose.translation.y = 2; | ||
sphImg.pose.translation.z = 3; | ||
sphImg.sphericalRepresentation.imageHeight = 100; | ||
sphImg.sphericalRepresentation.imageWidth = 100; | ||
sphImg.sphericalRepresentation.pixelHeight = 0.0314; | ||
sphImg.sphericalRepresentation.pixelWidth = 0.0314; | ||
sphImg.sphericalRepresentation.pngImageSize = pngData.size(); | ||
writer.WriteImage2DData(sphImg, e57::Image2DType::ImagePNG, e57::Image2DProjection::ProjectionSpherical, 0, pngData.data(), pngData.size()); | ||
|
||
e57::Image2D pinImg; | ||
pinImg.name = "pinhole"; | ||
pinImg.pinholeRepresentation.imageHeight = 100; | ||
pinImg.pinholeRepresentation.imageWidth = 100; | ||
pinImg.pinholeRepresentation.pixelHeight = 0.033; | ||
pinImg.pinholeRepresentation.pixelWidth = 0.044; | ||
pinImg.pinholeRepresentation.focalLength = 123; | ||
pinImg.pinholeRepresentation.principalPointX = 23; | ||
pinImg.pinholeRepresentation.principalPointY = 42; | ||
pinImg.pinholeRepresentation.pngImageSize = pngData.size(); | ||
writer.WriteImage2DData(pinImg, e57::Image2DType::ImageJPEG, e57::Image2DProjection::ProjectionPinhole, 0, jpegData.data(), jpegData.size()); | ||
|
||
e57::Image2D cylImg; | ||
cylImg.name = "cylindrical"; | ||
cylImg.cylindricalRepresentation.imageHeight = 100; | ||
cylImg.cylindricalRepresentation.imageWidth = 100; | ||
cylImg.cylindricalRepresentation.pixelHeight = 0.033; | ||
cylImg.cylindricalRepresentation.pixelWidth = 0.044; | ||
cylImg.cylindricalRepresentation.principalPointY = 42; | ||
cylImg.cylindricalRepresentation.radius = 666; | ||
cylImg.cylindricalRepresentation.pngImageSize = pngData.size(); | ||
writer.WriteImage2DData(cylImg, e57::Image2DType::ImageJPEG, e57::Image2DProjection::ProjectionCylindrical, 0, jpegData.data(), jpegData.size()); | ||
|
||
if (!writer.Close()) throw std::string("Failed to close tiny_pc_and_images.e57"); | ||
} | ||
|
||
void tiny_spherical() { | ||
e57::Writer writer("tiny_spherical.e57"); | ||
e57::Data3D header; | ||
header.pointCount = 360; | ||
header.pointFields.sphericalAzimuthField = true; | ||
header.pointFields.sphericalElevationField = true; | ||
header.pointFields.sphericalRangeField = true; | ||
header.pointFields.sphericalInvalidStateField = true; | ||
e57::Data3DPointsDouble buffers(header); | ||
for (auto i = 0; i < header.pointCount; i++) { | ||
buffers.sphericalAzimuth[i] = i * (3.14 / 360.0); | ||
buffers.sphericalElevation[i] = i * (3.14 / 360.0); | ||
buffers.sphericalRange[i] = 1.0; | ||
buffers.sphericalInvalidState[i] = i % 2 ? 1 : 0; | ||
} | ||
writer.WriteData3DData(header, buffers); | ||
if (!writer.Close()) throw std::string("Failed to close tiny_spherical.e57"); | ||
} | ||
|
||
int main() { | ||
empty(); | ||
tiny_pc(); | ||
tiny_pc_with_extension(); | ||
empty_pc(); | ||
tiny_pc_and_images(); | ||
tiny_spherical(); | ||
|
||
std::cout << "Finished!\n"; | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,8 @@ | ||
{ | ||
"$schema": "https://raw.githubusercontent.com/microsoft/vcpkg-tool/main/docs/vcpkg.schema.json", | ||
"name": "e57-testdata-generator", | ||
"version": "0.1.0", | ||
"dependencies": [ | ||
"libe57format" | ||
] | ||
} |