hpc-datastore-cpp/samples/common.hpp at main · fiji-hpc/hpc-datastore-cpp · 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
#pragma once
#include <i3d/image3d.h>
#include <i3d/vector3d.h>
#include <random>
#include <string>
#include <utility>

/** DO NOT FORGET TO CHANGE THESE TO MATCH YOUR SERVER **/

/* SERVER PROPERTIES  */
const std::string SERVER_IP = "127.0.0.1";
constexpr int SERVER_PORT = 9080;

/* DATASET PROPERTIES */
const std::string DS_UUID = "c44f8dcf-d26b-46b8-90a7-0d820651af35";

/* IMAGE PROPERTIES */
constexpr int IMG_CHANNEL = 0;
constexpr int IMG_ANGLE = 0;
constexpr int IMG_TIMEPOINT = 0;
const i3d::Vector3d<int> IMG_RESOLUTION = {1, 1, 1};
const std::string IMG_VERSION = "latest";

/** Some functionality shared between samples **/

/**
 * @brief Fill image with random values
 *
 * @tparam T Scalar image element type
 * @param img image
 */
template <typename T>
void fill_random(i3d::Image3d<T>& img) {
	std::mt19937_64 gen{std::random_device()()};
	std::uniform_int_distribution<long long> dist;

	for (auto& voxel : img)
		voxel = T(dist(gen));
}

template <typename T>
bool operator==(const i3d::Image3d<T>& rhs, const i3d::Image3d<T>& lhs) {
	if (rhs.GetSize() != lhs.GetSize())
		return false;

	for (std::size_t i = 0; i < rhs.GetImageSize(); ++i)
		if (rhs.GetVoxel(i) != lhs.GetVoxel(i))
			return false;

	return true;
}

#define SELECT_TYPE(type, func)                                                \
	if (type == "uint8")                                                       \
		func<uint8_t>();                                                       \
	else if (type == "uint16")                                                 \
		func<uint16_t>();                                                      \
	else if (type == "uint64")                                                 \
		func<uint64_t>();                                                      \
	else if (type == "int32")                                                  \
		func<int32_t>();                                                       \
	else if (type == "float32")                                                \
		func<float>();                                                         \
	else if (type == "float64")                                                \
		func<double>();                                                        \
	else                                                                       \
		std::cout << "Image type is not supported !! \n";