attribute_point_set.h

Go to the documentation of this file.

/******************************************************************************
 * Copyright 2023 NVIDIA Corporation. All rights reserved.
 *****************************************************************************/
 
#ifndef NVIDIA_INDEX_EXAMPLE_ATTRIBUTE_POINT_SET_H
#define NVIDIA_INDEX_EXAMPLE_ATTRIBUTE_POINT_SET_H
 
#include <mi/dice.h>
#include <mi/base/interface_implement.h>
#include <mi/base/handle.h>
#include <mi/base/uuid.h>
#include <mi/math/vector.h>
 
#include <nv/index/ipoint_set.h>
 
#include <vector>
 
 
class Attribute_point_set :
    public mi::neuraylib::Element<0x37d4157e,0x7ac0,0x4812,0xb8,0xe2,0x7a,0x45,0x26,0x4b,0xa3,0x4c,
                                  nv::index::IPoint_set>
{
public:
    Attribute_point_set(
        const std::vector< mi::math::Vector_struct< mi::Float32, 3 > >& vertex_vec,
        const std::vector< mi::Float32 > &                              attribute_vec,
        nv::index::IPoint_set::Point_style                              point_style);
 
    Attribute_point_set()
        :
        m_vertex_vec(),
        m_per_vertex_color_values(),
        m_per_vertex_radii(),
        m_point_style(nv::index::IPoint_set::FLAT_CIRCLE),
        m_rendering_enabled(true),
        m_pickable(true)
    {
        // empty
    }
 
    virtual ~Attribute_point_set() {}
 
    void set_attribute(std::vector< mi::Float32 > const & attribute_vec);
 
    mi::math::Vector_struct<mi::Float32, 3> * peek_vertices();
 
 
public:
    virtual mi::Size get_nb_vertices() const { return m_vertex_vec.size(); }
 
    virtual const mi::math::Vector_struct<mi::Float32, 3>* get_vertices() const;
 
    virtual mi::Size get_nb_colors() const { return m_per_vertex_color_values.size(); }
 
    virtual const mi::math::Color_struct* get_colors() const;
 
    virtual mi::Size get_nb_radii() const { return m_per_vertex_radii.size(); }
 
    virtual const mi::Float32* get_radii() const;
 
    virtual Point_style get_point_style() const { return m_point_style; }
    virtual void set_point_style(Point_style style) { m_point_style = style; }
 
    virtual void set_enabled(bool enable) { m_rendering_enabled = enable;}
    virtual bool get_enabled() const { return m_rendering_enabled;}
 
    virtual void set_meta_data(mi::neuraylib::Tag_struct tag) { m_meta_data = tag; }
    
    virtual mi::neuraylib::Tag_struct get_meta_data() const { return m_meta_data; }
 
public:
    // Implemented IShape
 
    virtual bool get_pickable() const { return m_pickable; }
    virtual void set_pickable(bool pickable) { m_pickable = pickable; }
 
public:
    // Implemented IElement
 
    virtual mi::neuraylib::IElement* copy() const;
    virtual const char* get_class_name() const;
    virtual void serialize(
        mi::neuraylib::ISerializer *serializer) const;
    virtual void deserialize(
        mi::neuraylib::IDeserializer* deserializer);
    virtual void get_references(mi::neuraylib::ITag_set* result) const
    {
        if(m_meta_data.is_valid())
            result->add_tag(m_meta_data);
    }
 
protected:
    mi::math::Color_struct get_color_from_attribute(mi::Float32 attrib);
 
    mi::Float32 get_radius_from_attribute(mi::Float32 attrib);
 
private:
    virtual void set_vertices(mi::math::Vector_struct<mi::Float32, 3>* vertices, mi::Size nb_vertices) { }
    virtual void set_radii(mi::Float32* radii, mi::Size nb_radii) { }
    virtual void set_colors(mi::math::Color_struct* colors, mi::Size nb_colors) { }
 
    // vertices coordinates
    std::vector<mi::math::Vector_struct<mi::Float32, 3> >   m_vertex_vec;
    // vertices colors
    std::vector<mi::math::Color_struct>                     m_per_vertex_color_values;
    // vertices radii
    std::vector<mi::Float32>                                m_per_vertex_radii;
    // style of the point set
    nv::index::IPoint_set::Point_style                      m_point_style;
    // rendering enable/disable status
    bool                                                    m_rendering_enabled;
    // status of enabled pick-ability
    bool                                                    m_pickable;
    // Meta data
    mi::neuraylib::Tag                                      m_meta_data;
};
 
#endif // NVIDIA_INDEX_EXAMPLE_ATTRIBUTE_POINT_SET_H