xac_compute_scene_tool.h

Go to the documentation of this file.

/******************************************************************************
 * Copyright 2023 NVIDIA Corporation. All rights reserved.
 *****************************************************************************/
#ifndef EXAMPLES_XAC_COMPUTE_SCENE_TOOL_H
#define EXAMPLES_XAC_COMPUTE_SCENE_TOOL_H
 
#include <mi/dice.h>
#define USE_NVINDEX_ACCESS
#include "utility/example_shared.h"
 
 
namespace xac_compute
{
 
// Helper tool to setup scenes.
struct Scene_tool
{
    mi::base::Handle<const nv::index::ISession>         session;
    mi::base::Handle<nv::index::IScene>                 scene;
    mi::base::Handle<nv::index::IScene_group>           current_group;
    mi::neuraylib::IDice_transaction*                   transaction;
 
    Scene_tool(
        mi::neuraylib::Tag                session_tag_,
        mi::neuraylib::IDice_transaction* transaction_);
 
    // Add a static or transformed group to scene.
    mi::neuraylib::Tag add_new_group(bool transformed_group = false);
    
    // Set the current group.
    void set_group(mi::neuraylib::Tag group_);
 
 
    template<class T>
    mi::neuraylib::Tag store_and_add_to_group(const mi::base::Handle<T>& element)
    {
        if (element) {
            const mi::neuraylib::Tag tag = transaction->store_for_reference_counting(element.get());
            check_success(tag.is_valid());
            current_group->append(tag, transaction);
            return tag;
        }
        return mi::neuraylib::Tag();
    }
 
    template<class T>
    void create_attribute(mi::base::Handle<T>& element)
    {
        element = scene->create_attribute<T>();
        check_success(element.is_valid_interface());
    }
};
 
 
 
struct Program_source
{
    const char* program_begin;
    const char* program_end;
    const char* color_program;
    const char* compute_program;
};
 
enum Sample_program_kind {
    Surface_sample_program,
    Volume_sample_program
};
 
// nv::index::IVolume_sample_program or nv::index::ISurface_sample_program
std::pair<mi::neuraylib::Tag, std::string>
add_user_program(
    Sample_program_kind     prg_kind,
    int                     user_prg,
    const std::string&      user_text,
    const Program_source&   prg_text,
    Scene_tool&             scene_tool,
    const char*             extra_pgr_methods=0);
 
}//xac_compute
 
 
#endif