TemplateHashmap.h

Go to the documentation of this file.

// ----------------------------------------------------------------------------
// -                        Open3D: www.open3d.org                            -
// ----------------------------------------------------------------------------
// The MIT License (MIT)
//
// Copyright (c) 2018 www.open3d.org
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
// IN THE SOFTWARE.
// ----------------------------------------------------------------------------

#pragma once

// Low level hashmap interface for advanced usages who wish to modify hash
// function and/or equal function.
// Users must create a .cu file and compile with
// nvcc to use customized GPU hashmap.
//
// APIs are available in DeviceHashmap.h.
// Include path: TemplatedHashmap.h -> HashmapCPU.h  -> DeviceHashmap.h
//                                |                    ^
//                                |--> HashmapCUDA.h --|
//                                        (CUDA code)
//
// .cpp targets only include CPU part that can be compiled by non-nvcc
// compilers even if BUILD_CUDA_MODULE is enabled.
// .cu targets include both.

#include "open3d/core/hashmap/CPU/TemplateHashmapCPU.hpp"

#if defined(BUILD_CUDA_MODULE) && defined(__CUDACC__)
#include "open3d/core/hashmap/CUDA/TemplateHashmapCUDA.cuh"
#endif

#include <unordered_map>

namespace open3d {
namespace core {

template <typename Hash, typename KeyEq>
std::shared_ptr<DeviceHashmap<Hash, KeyEq>> CreateTemplateDeviceHashmap(
        int64_t init_buckets,
        int64_t init_capacity,
        int64_t dsize_key,
        int64_t dsize_value,
        const Device &device) {
    if (device.GetType() == Device::DeviceType::CPU) {
        return CreateTemplateCPUHashmap<Hash, KeyEq>(
                init_buckets, init_capacity, dsize_key, dsize_value, device);
    }
#if defined(BUILD_CUDA_MODULE) && defined(__CUDACC__)
    else if (device.GetType() == Device::DeviceType::CUDA) {
        return CreateTemplateCUDAHashmap<Hash, KeyEq>(
                init_buckets, init_capacity, dsize_key, dsize_value, device);
    }
#endif
    else {
        utility::LogError("[CreateTemplateHashmap]: Unimplemented device");
    }
}
}  // namespace core
}  // namespace open3d