ivdb_subset.h

Go to the documentation of this file.

/******************************************************************************
 * Copyright 2023 NVIDIA Corporation. All rights reserved.
 *****************************************************************************/
 
#ifndef NVIDIA_INDEX_IVDB_SUBSET_H
#define NVIDIA_INDEX_IVDB_SUBSET_H
 
#include <mi/dice.h>
#include <mi/base/interface_declare.h>
#include <mi/math/vector.h>
 
#include <nv/index/idistributed_data_subset.h>
 
namespace nv {
namespace index {
 
class IVDB_subset;
class IVDB_subset_device;
 
class IVDB_subset :
    public mi::base::Interface_declare<0xe0555782,0x2353,0x4d3b,0xaf,0x9d,0xa4,0xbb,0xbc,0x10,0xa9,0xec,
                                       IDistributed_data_subset>
{
public:
    struct NVDB_version {
        mi::Uint64 magic;           
        mi::Uint32 major;           
        mi::Uint32 minor;           
    };
 
    struct NVDB_grid_storage {
        mi::Size    size;
        void*       data;
    };
 
public:
    virtual bool                    generate_grid_storage(
                                        mi::Uint32      attribute_index,
                                        mi::Size        grid_memory_size,
                                        NVDB_version    nvdb_version) = 0;
 
    virtual NVDB_grid_storage       get_grid_storage(
                                        mi::Uint32      attribute_index) const = 0;
 
    virtual IVDB_subset_device*     get_device_subset() const = 0;
};
 
class IVDB_subset_device :
    public mi::base::Interface_declare<0xee4883a1,0xf5d3,0x4a42,0xaa,0x81,0x37,0xdd,0xe3,0x6c,0x8e,0xc8,
                                       IDistributed_data_subset_device>
{
public:
    typedef IVDB_subset::NVDB_version   NVDB_version;
 
public:
    virtual NVDB_version        nvdb_version() const = 0;
 
    virtual void*               grid_buffer(
                                    mi::Uint32      attribute_index) = 0;
 
    virtual mi::Size            grid_buffer_size(
                                    mi::Uint32      attribute_index) const = 0;
    virtual bool                grid_buffer_resize(
                                    mi::Uint32      attribute_index,
                                    const mi::Size  new_size) = 0;
 
    // #todo take ownership?
    // - set grid_data_size to ~0x00000000u for unknown size.
    virtual bool                adopt_grid_buffer(
                                    mi::Uint32  attribute_index,
                                    void*       grid_data,
                                    mi::Size    grid_data_size) = 0;
};
 
} // namespace index
} // namespace nv
 
#endif // NVIDIA_INDEX_IVDB_SUBSET_H