Open3D (C++ API)  0.15.1
Dispatch.h
Go to the documentation of this file.
1// ----------------------------------------------------------------------------
2// - Open3D: www.open3d.org -
3// ----------------------------------------------------------------------------
4// The MIT License (MIT)
5//
6// Copyright (c) 2018-2021 www.open3d.org
7//
8// Permission is hereby granted, free of charge, to any person obtaining a copy
9// of this software and associated documentation files (the "Software"), to deal
10// in the Software without restriction, including without limitation the rights
11// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
12// copies of the Software, and to permit persons to whom the Software is
13// furnished to do so, subject to the following conditions:
14//
15// The above copyright notice and this permission notice shall be included in
16// all copies or substantial portions of the Software.
17//
18// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
19// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
20// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
21// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
22// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
23// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
24// IN THE SOFTWARE.
25// ----------------------------------------------------------------------------
26
27#pragma once
28
29#include "open3d/core/Dtype.h"
31
49#define DISPATCH_DTYPE_TO_TEMPLATE(DTYPE, ...) \
50 [&] { \
51 if (DTYPE == open3d::core::Float32) { \
52 using scalar_t = float; \
53 return __VA_ARGS__(); \
54 } else if (DTYPE == open3d::core::Float64) { \
55 using scalar_t = double; \
56 return __VA_ARGS__(); \
57 } else if (DTYPE == open3d::core::Int8) { \
58 using scalar_t = int8_t; \
59 return __VA_ARGS__(); \
60 } else if (DTYPE == open3d::core::Int16) { \
61 using scalar_t = int16_t; \
62 return __VA_ARGS__(); \
63 } else if (DTYPE == open3d::core::Int32) { \
64 using scalar_t = int32_t; \
65 return __VA_ARGS__(); \
66 } else if (DTYPE == open3d::core::Int64) { \
67 using scalar_t = int64_t; \
68 return __VA_ARGS__(); \
69 } else if (DTYPE == open3d::core::UInt8) { \
70 using scalar_t = uint8_t; \
71 return __VA_ARGS__(); \
72 } else if (DTYPE == open3d::core::UInt16) { \
73 using scalar_t = uint16_t; \
74 return __VA_ARGS__(); \
75 } else if (DTYPE == open3d::core::UInt32) { \
76 using scalar_t = uint32_t; \
77 return __VA_ARGS__(); \
78 } else if (DTYPE == open3d::core::UInt64) { \
79 using scalar_t = uint64_t; \
80 return __VA_ARGS__(); \
81 } else { \
82 utility::LogError("Unsupported data type."); \
83 } \
84 }()
85
86#define DISPATCH_DTYPE_TO_TEMPLATE_WITH_BOOL(DTYPE, ...) \
87 [&] { \
88 if (DTYPE == open3d::core::Bool) { \
89 using scalar_t = bool; \
90 return __VA_ARGS__(); \
91 } else { \
92 DISPATCH_DTYPE_TO_TEMPLATE(DTYPE, __VA_ARGS__); \
93 } \
94 }()
95
96#define DISPATCH_FLOAT_DTYPE_TO_TEMPLATE(DTYPE, ...) \
97 [&] { \
98 if (DTYPE == open3d::core::Float32) { \
99 using scalar_t = float; \
100 return __VA_ARGS__(); \
101 } else if (DTYPE == open3d::core::Float64) { \
102 using scalar_t = double; \
103 return __VA_ARGS__(); \
104 } else { \
105 utility::LogError("Unsupported data type."); \
106 } \
107 }()