kfbxpropertymap.h

Go to the documentation of this file.

/****************************************************************************************

Copyright (C) 2010 Autodesk, Inc.
All rights reserved.

Use of this software is subject to the terms of the Autodesk license agreement
provided at the time of installation or download, or which otherwise accompanies
this software in either electronic or hard copy form.

****************************************************************************************/

#ifndef FBXFILESDK_FBXCORE_FBXCOLLECTION_KFBXPROPERTYMAP_H
#define FBXFILESDK_FBXCORE_FBXCOLLECTION_KFBXPROPERTYMAP_H

#include <fbxfilesdk/fbxfilesdk_def.h>

#include <fbxfilesdk/components/kbaselib/klib/kmap.h>
#include <fbxfilesdk/kfbxplugins/kfbxproperty.h>
#include <fbxfilesdk/kfbxplugins/kfbxobject.h>

#include <fbxfilesdk/fbxfilesdk_nsbegin.h>

template <class K, class T, class Compare>
class KFBX_DLL KFbxMap
{
//
//  WARNING!
//
//  Anything beyond these lines may not be documented accurately and is 
//  subject to change without notice.
//
#ifndef DOXYGEN_SHOULD_SKIP_THIS
private:
    typedef KMap<K, T, Compare> KMapDef;
    KMapDef mMap;
#endif // DOXYGEN_SHOULD_SKIP_THIS

public:
    typedef typename KMapDef::RecordType* kIterator;

    inline KFbxMap()
    {
    }

    inline void Add(K const &pKey, T const &pValue)
    {
        mMap.Insert(pKey, pValue);
    }

    inline kIterator Find(K const &pKey) const
    {
        return (kIterator)mMap.Find( pKey );
    }

    inline kIterator Find(T const &pValue) const
    {
        kIterator lIterator = GetFirst();
        while (lIterator)
        {
            if (lIterator->GetValue()==pValue)
            {
                return lIterator;
            }
            lIterator = GetNext(lIterator);
        }
        return 0;
    }

    inline void Remove(kIterator pIterator)
    {
        if (pIterator) mMap.Remove( pIterator->GetKey() );
    }

    inline kIterator GetFirst() const
    {
        return (kIterator)mMap.Minimum();
    }

    inline kIterator GetNext(kIterator pIterator) const
    {
        return (kIterator)pIterator ? pIterator->Successor() : 0;
    }

    inline void Clear() 
    {
        mMap.Clear();
    }

    inline void Reserve(int pSize)
    {
        mMap.Reserve( pSize );
    }

    inline int GetCount() const
    {
        return mMap.GetSize();
    }
};

template <class T,class Compare>
class KFBX_DLL KFbxPropertyMap : public KFbxMap<T,KFbxProperty,Compare>
{
public:
    inline KFbxPropertyMap()
    {
    }

    inline KFbxProperty Get(typename KFbxMap<T,KFbxProperty,Compare>::kIterator pIterator)
    {
        return pIterator ? pIterator->GetValue() : KFbxProperty();
    }
};

template <class T, class Compare>
class KFBX_DLL KFbxObjectMap : public KFbxMap<T,KFbxObject*,Compare>
{
public:
    inline KFbxObjectMap()
    {
    }

    inline KFbxObject* Get(typename KFbxMap<T,KFbxObject*,Compare>::kIterator pIterator)
    {
        return pIterator ? pIterator->GetValue() : 0;
    }
};

class KFbxMapKStringCompare
{
public:
    inline int operator()(KString const &pKeyA, KString const &pKeyB) const
    {
        return (pKeyA < pKeyB) ? -1 : ((pKeyB < pKeyA) ? 1 : 0);
    }
};

class KFBX_DLL KFbxPropertyStringMap : public KFbxPropertyMap<KString,KFbxMapKStringCompare>
{
public:
    inline KFbxPropertyStringMap()
    {
    }
};

class KFBX_DLL KFbxObjectStringMap : public KFbxObjectMap<class KString,class KFbxMapKStringCompare>
{
public:
    inline KFbxObjectStringMap()
    {
    }
};

#include <fbxfilesdk/fbxfilesdk_nsend.h>

#endif // FBXFILESDK_FBXCORE_FBXCOLLECTION_KFBXPROPERTYMAP_H