Utils.h

#pragma once
#include "AL/maya/SIMD.h"
#include "AL/usdmaya/Common.h"

#include "maya/MObject.h"
#include "maya/MString.h"
#include "maya/MUuid.h"
#include "maya/MFnDependencyNode.h"

#include <map>
#include <string>

namespace AL {
namespace usdmaya {

//----------------------------------------------------------------------------------------------------------------------
//----------------------------------------------------------------------------------------------------------------------
void mapUsdPrimToMayaNode(const UsdPrim& usdPrim, const MObject& mayaObject, const MDagPath* const proxyShapeNode = nullptr);

//----------------------------------------------------------------------------------------------------------------------
//----------------------------------------------------------------------------------------------------------------------
void matrixToSRT(GfMatrix4d& value, double S[3], MEulerRotation& R, double T[3]);

//----------------------------------------------------------------------------------------------------------------------
//----------------------------------------------------------------------------------------------------------------------
inline MString convert(const std::string& str)
{
  return MString(str.data(), str.size());
}

//----------------------------------------------------------------------------------------------------------------------
//----------------------------------------------------------------------------------------------------------------------
inline std::string convert(const MString& str)
{
  return std::string(str.asChar(), str.length());
}

//----------------------------------------------------------------------------------------------------------------------
// code to speed up comparisons of MObject guids
//----------------------------------------------------------------------------------------------------------------------

struct guid
{
  uint8_t uuid[16]; 
};

#if AL_MAYA_ENABLE_SIMD

struct guid_compare
{
  inline bool operator () (const i128 a, const i128 b) const
  {
    const uint32_t lt_mask = movemask16i8(cmplt16i8(a, b));
    const uint32_t eq_mask = 0xFFFF & (~movemask16i8(cmpeq16i8(a, b)));
    if(!eq_mask) return false;

    // find first bit value that is not equal
    const uint32_t index = __builtin_ctz(eq_mask);
    // now see whether that bit has been set
    return (lt_mask & (1 << index)) != 0;
  }
};

#else

struct guid_compare
{
  inline bool operator () (const guid& a, const guid& b) const
  {
    for(int i = 0; i < 16; ++i)
    {
      if(a.uuid[i] < b.uuid[i]) return true;
      if(a.uuid[i] > b.uuid[i]) return false;
    }
    return false;
  }
};
#endif

//----------------------------------------------------------------------------------------------------------------------
//----------------------------------------------------------------------------------------------------------------------
struct MObjectMap
{
  inline bool insert(const MFnDependencyNode& fn)
  {
    #if AL_MAYA_ENABLE_SIMD
    union
    {
      __m128i sse;
      guid uuid;
    };
    fn.uuid().get(uuid.uuid);
    bool contains = m_nodeMap.find(sse) != m_nodeMap.end();
    if(!contains)
      m_nodeMap.insert(std::make_pair(sse, fn.object()));
    #else
    guid uuid;
    fn.uuid().get(uuid.uuid);
    bool contains = m_nodeMap.find(uuid) != m_nodeMap.end();
    if(!contains)
      m_nodeMap.insert(std::make_pair(uuid, fn.object()));
    #endif
    return contains;
  };

  inline bool contains(const MFnDependencyNode& fn)
  {
    #if AL_MAYA_ENABLE_SIMD
    union
    {
      __m128i sse;
      guid uuid;
    };
    fn.uuid().get(uuid.uuid);
    bool contains = m_nodeMap.find(sse) != m_nodeMap.end();
    #else
    guid uuid;
    fn.uuid().get(uuid.uuid);
    bool contains = m_nodeMap.find(uuid) != m_nodeMap.end();
    #endif
    return contains;
  };

private:
  #if AL_MAYA_ENABLE_SIMD
  std::map<i128, MObject, guid_compare> m_nodeMap;
  #else
  std::map<guid, MObject, guid_compare> m_nodeMap;
  #endif
};

} // usdmaya
//----------------------------------------------------------------------------------------------------------------------
} // AL
//----------------------------------------------------------------------------------------------------------------------