FnPickingTypes.h

// Copyright (c) 2017 The Foundry Visionmongers Ltd. All Rights Reserved.

#ifndef FNVIEWER_PICKINGTYPES_H
#define FNVIEWER_PICKINGTYPES_H



#include <FnViewer/plugin/FnMathTypes.h>
#include <FnAttribute/FnAttribute.h>
#include <FnAttribute/FnGroupBuilder.h>

#include <stdint.h>
#include <sstream>
#include <map>

namespace Foundry
{
namespace Katana
{
namespace ViewerAPI
{

typedef uint64_t FnPickId;


class PickedAttrsMap : public std::map<FnPickId, FnAttribute::Attribute>
{
public:

    FnAttribute::GroupAttribute toGroupAttribute()
    {
        if (size() == 0)
        {
            return FnAttribute::GroupAttribute();
        }

        FnAttribute::GroupBuilder gb;
        for (const_iterator it = begin(); it != end(); ++it)
        {
            // Convert the \c FnPickId into a string representation
            std::ostringstream idStr;
            idStr << it->first;
            gb.set(idStr.str(), it->second);
        }

        return gb.build();
    }

    void fromGroupAttribute(FnAttribute::GroupAttribute groupAttr)
    {
        clear();

        if (groupAttr.isValid())
        {
            for (unsigned int i = 0; i < groupAttr.getNumberOfChildren(); ++i)
            {
                // Get the \c FnPickId by parsing the entry name
                FnPickId pickId;
                std::istringstream istr(groupAttr.getChildName(i));
                istr >> pickId;
                (*this)[pickId] = groupAttr.getChildByIndex(i);
            }
        }
    }
};

inline void pickIdToColor(FnPickId pickId, Vec4i& color)
{
    color.x = (pickId >> 24 & 0xff);
    color.y = (pickId >> 16 & 0xff);
    color.z = (pickId >> 8  & 0xff);
    color.w = (pickId       & 0xff);
}

inline void pickIdToColor(FnPickId pickId, Vec4f& color)
{
    color.x = (float)(pickId >> 24 & 0xff) / 0xff;
    color.y = (float)(pickId >> 16 & 0xff) / 0xff;
    color.z = (float)(pickId >> 8  & 0xff) / 0xff;
    color.w = (float)(pickId       & 0xff) / 0xff;
}

} // ViewerAPI
} // Katana
} // Foundry

#endif /* FNVIEWER_PICKINGTYPES_H */