source: trunk/src/gfx/skeleton_instance.cc @ 496

// Copyright (C) 2011-2015 ChaosForge Ltd
// http://chaosforge.org/
//
// This file is part of Nova libraries. 
// For conditions of distribution and use, see copying.txt file in root folder.

#include "nv/gfx/skeleton_instance.hh"

#include "nv/core/profiler.hh"
#include "nv/interface/interpolate.hh"

void nv::skeleton_binding::assign( const skeleton_binding& other )
{
        m_indices.assign( other.m_indices );
        m_key        = other.m_key;
        m_bone_count = other.m_bone_count;
}

void nv::skeleton_binding::prepare( const mesh_nodes_data* node_data, const data_node_list& bone_data )
{
        if ( m_indices.empty() )
        {
                // TODO: either fixed size struct or static allocator
                hash_store< shash64, uint16 > bone_names;
                m_indices.resize( node_data->size() );

                for ( nv::uint16 bi = 0; bi < bone_data.size(); ++bi )
                        bone_names[bone_data[bi].name] = bi;

                for ( uint32 n = 0; n < node_data->size(); ++n )
                {
                        sint16 bone_id = -1;
                        auto bi = bone_names.find( node_data->get_info( n ).name );
                        if ( bi != bone_names.end() )
                        {
                                bone_id = sint16( bi->second );
                        }
                        m_indices[n] = bone_id;

                }
                m_bone_count = bone_data.size();
        }

        if ( m_key.size() == 0 )
        {
                for ( uint32 n = 0; n < node_data->size(); ++n )
                        if ( ( *node_data )[n]->size() > 0 )
                        {
                                m_key = ( *node_data )[n]->get_interpolation_key();
                                break;
                        }
        }
}

void nv::skeleton_binding::prepare( const data_node_list& pose_data, const data_node_list& bone_data )
{
        if ( m_indices.empty() )
        {
                // TODO: either fixed size struct or static allocator
                hash_store< shash64, uint16 > bone_names;
                m_indices.resize( pose_data.size() );

                for ( nv::uint16 bi = 0; bi < bone_data.size(); ++bi )
                        bone_names[bone_data[bi].name] = bi;

                for ( uint32 n = 0; n < pose_data.size(); ++n )
                {
                        sint16 bone_id = -1;
                        auto bi = bone_names.find( pose_data[ n ].name );
                        if ( bi != bone_names.end() )
                        {
                                bone_id = sint16( bi->second );
                        }
                        m_indices[n] = bone_id;

                }
                m_bone_count = bone_data.size();
        }
}

void nv::skeleton_transforms::assign( const data_node_list* node_data )
{
        NV_ASSERT( node_data, "!!!" );
        if ( m_transforms.size() != node_data->size() )
                m_transforms.resize( node_data->size() );
        for ( uint32 n = 0; n < node_data->size(); ++n )
        {
                m_transforms[n] = transform( ( *node_data )[n].transform );
        }
}

void nv::skeleton_transforms::assign( const skeleton_transforms& other, const array_view< bool >& mask )
{
        if ( m_transforms.size() != other.size() ) m_transforms.resize( other.size() );
        if ( mask.size() == 0 )
                m_transforms.assign( other.m_transforms );
        else
                for ( uint32 i = 0; i < other.size(); ++i )
                        if ( mask[i] )
                                m_transforms[i] = other.m_transforms[i];
}

void nv::skeleton_transforms::interpolate( const skeleton_transforms& a, const skeleton_transforms& b, float t, interpolation i )
{
        if ( m_transforms.size() != a.size() ) m_transforms.resize( a.size() );
        ::nv::interpolate( m_transforms, t, i, a.m_transforms, b.m_transforms );
}

void nv::skeleton_transforms::interpolate( const skeleton_transforms& a, const skeleton_transforms& b, float t, interpolation i, const array_view< bool >& mask )
{
        if ( m_transforms.size() != a.size() ) m_transforms.resize( a.size() );
        if ( mask.size() > 0 )
                ::nv::interpolate( m_transforms, t, i, mask, a.m_transforms, b.m_transforms );
        else
                ::nv::interpolate( m_transforms, t, i, a.m_transforms, b.m_transforms );
}

void nv::skeleton_transforms::interpolate( const skeleton_transforms& s1, const skeleton_transforms& v1, const skeleton_transforms& v2, const skeleton_transforms& s2, float t, interpolation i )
{
        if ( m_transforms.size() != s1.size() ) m_transforms.resize( s1.size() );
        ::nv::interpolate( m_transforms, t, i, s1.m_transforms, v1.m_transforms, v2.m_transforms, s2.m_transforms );
}

void nv::skeleton_transforms::interpolate( const skeleton_transforms& s1, const skeleton_transforms& v1, const skeleton_transforms& v2, const skeleton_transforms& s2, float t, interpolation i, const array_view< bool >& mask )
{
        if ( m_transforms.size() != s1.size() ) m_transforms.resize( s1.size() );
        if ( mask.size() > 0 )
                ::nv::interpolate( m_transforms, t, i, mask, s1.m_transforms, v1.m_transforms, v2.m_transforms, s2.m_transforms );
        else
                ::nv::interpolate( m_transforms, t, i, s1.m_transforms, v1.m_transforms, v2.m_transforms, s2.m_transforms );
}

void nv::skeleton_transforms::blend( const skeleton_transforms& a, const skeleton_transforms& b, float t, interpolation i, float blend, interpolation bi )
{
        NV_ASSERT( a.size() == b.size(), "!!!" );
        if ( m_transforms.size() != a.size() )
                m_transforms.resize( a.size() );
        ::nv::interpolate( m_transforms, t, i, blend, bi, a.m_transforms, b.m_transforms );
}

void nv::skeleton_transforms::blend( const skeleton_transforms& a, const skeleton_transforms& b, float t, interpolation i, float blend, interpolation bi, const array_view< bool >& mask )
{
        NV_ASSERT( a.size() == b.size(), "!!!" );
        if ( m_transforms.size() != a.size() )
                m_transforms.resize( a.size() );
        if ( mask.size() > 0 )
                ::nv::interpolate( m_transforms, t, i, blend, bi, mask, a.m_transforms, b.m_transforms );
        else
                ::nv::interpolate( m_transforms, t, i, blend, bi, a.m_transforms, b.m_transforms );
}

void nv::skeleton_transforms::blend( const skeleton_transforms& s1, const skeleton_transforms& v1, const skeleton_transforms& v2, const skeleton_transforms& s2, float t, interpolation i, float blend, interpolation bi )
{
        if ( m_transforms.size() != s1.size() ) m_transforms.resize( s1.size() );
        ::nv::interpolate( m_transforms, t, i, blend, bi, s1.m_transforms, v1.m_transforms, v2.m_transforms, s2.m_transforms );
}

void nv::skeleton_transforms::blend( const skeleton_transforms& s1, const skeleton_transforms& v1, const skeleton_transforms& v2, const skeleton_transforms& s2, float t, interpolation i, float blend, interpolation bi, const array_view< bool >& mask )
{
        if ( m_transforms.size() != s1.size() ) m_transforms.resize( s1.size() );
        if ( mask.size() > 0 )
                ::nv::interpolate( m_transforms, t, i, blend, bi, mask, s1.m_transforms, v1.m_transforms, v2.m_transforms, s2.m_transforms );
        else
                ::nv::interpolate( m_transforms, t, i, blend, bi, s1.m_transforms, v1.m_transforms, v2.m_transforms, s2.m_transforms );
}

void nv::skeleton_transforms::assign( const skeleton_transforms& other )
{
        m_transforms.assign( other.m_transforms );
}


void nv::skeleton_transforms::delocalize_rec( const data_node_tree& node_data, uint32 id, const transform& parent )
{
        transform global_mat = parent;
        global_mat *= m_transforms[id];
        m_transforms[id] = global_mat;
        for ( auto child : node_data.children( id ) )
        {
                delocalize_rec( node_data, child, global_mat );
        }
}