MHO_ContainerStore.hh

Go to the documentation of this file.

#ifndef MHO_ContainerStore_HH__
#define MHO_ContainerStore_HH__
 
#include <map>
#include <set>
#include <utility>
#include <vector>
 
#include "MHO_ContainerDictionary.hh"
#include "MHO_Message.hh"
#include "MHO_Serializable.hh"
#include "MHO_Types.hh"
#include "MHO_UUID.hh"
#include "MHO_UUIDGenerator.hh"
 
namespace hops
{
 
class MHO_ContainerStore
{
    public:
        MHO_ContainerStore(){};
 
        virtual ~MHO_ContainerStore() { Clear(); };
 
        void Clear();
 
 
        template< typename XClassType > bool AddObject(XClassType* obj);
 
        template< typename XClassType > XClassType* GetObject(const MHO_UUID& obj_id);
 
        template< typename XClassType > XClassType* GetObject(const std::string& shortname);
 
        template< typename XClassType > XClassType* GetObject(std::size_t index);
 
        template< typename XClassType > bool DeleteObject(XClassType* obj_ptr);
 
        template< typename XClassType > MHO_UUID GetTypeUUID();
 
        template< typename XClassType > std::size_t GetNObjects() const;
 
        //check if any object with the given object id is in the store
        bool IsObjectPresent(const MHO_UUID& obj_id) const;
 
        MHO_Serializable* GetObject(const MHO_UUID& obj_id);
        
        bool DeleteObject(const MHO_UUID& obj_id);
 
        void GetAllTypeUUIDs(std::vector< MHO_UUID >& type_ids);
 
        void GetAllObjectUUIDsOfType(MHO_UUID type_id, std::vector< MHO_UUID >& obj_ids);
 
        std::size_t GetNObjects() const { return fObjectsToIds.size(); }
 
        bool SetShortName(const MHO_UUID& obj_id, const std::string& shortname);
 
        MHO_UUID GetObjectUUID(const std::string& shortname);
 
        MHO_UUID GetObjectTypeUUID(const MHO_UUID& obj_id);
 
        std::string GetShortName(const MHO_UUID& obj_id);
 
        void GetAllShortNames(std::vector< std::string >& shortnames);
 
        //this is primarily here to provide a object look-up table for the python interface
        //returns a list of (object_type_uuid, object_item_uuid, shortname)
        std::vector< std::tuple< std::string, std::string, std::string > > GetAllObjectInfo();
 
 
        void RenameObject(const std::string& current_shortname, const std::string& new_shortname);
 
        //debug
        void DumpShortNamesToIds()
        {
            for(auto it = fShortNameToIds.begin(); it != fShortNameToIds.end(); it++)
            {
                std::cout << it->first << " : " << it->second.as_string() << std::endl;
            }
        }
 
    protected:
        using key_pair = std::pair< MHO_UUID, MHO_UUID >;
 
        //object dictionary...currently we only have one dictionary implementation
        //however, we may in the future want to allow the user to pass a custom dictionary implementation
        //if they have additional classes they want to serialize which are not already supported
        MHO_ContainerDictionary fDictionary;
 
        //all objects are stored as pointers to the base-class MHO_Serializable
        //they are cast to the underlying type specified by the type id upon retrieval
 
        //the key pair is <type_uuid, obj_id>, and the value is a pointer to the object
        std::map< key_pair, MHO_Serializable* > fIdsToObjects;
 
        //the key is a pointer to an object, and the value is a pair of <type_uuid, obj_id>
        std::map< MHO_Serializable*, key_pair > fObjectsToIds;
 
        //maps string names to object uuids
        std::map< std::string, MHO_UUID > fShortNameToIds;
        std::set< std::string > fShortNameSet;
};
 
template< typename XClassType > bool MHO_ContainerStore::AddObject(XClassType* obj)
{
    if(obj == nullptr)
    {
        return false;
    }
 
    //attempt to cast to our storage/serialization type
    auto ptr = static_cast< MHO_Serializable* >(obj);
    if(ptr == nullptr)
    {
        return false;
    }
 
    MHO_UUID obj_id = obj->GetObjectUUID();
    MHO_UUID type_id = obj->GetTypeUUID();
 
    key_pair kp;
    kp.first = type_id;
    kp.second = obj_id;
 
    fIdsToObjects[kp] = ptr;
    fObjectsToIds[ptr] = kp;
    return true;
}
 
template< typename XClassType > XClassType* MHO_ContainerStore::GetObject(const MHO_UUID& obj_id)
{
    MHO_UUID type_id = fDictionary.GetUUIDFor< XClassType >();
 
    key_pair kp;
    kp.first = type_id;
    kp.second = obj_id;
 
    XClassType* ptr = nullptr;
    auto it = fIdsToObjects.find(kp);
    if(it != fIdsToObjects.end())
    {
        MHO_Serializable* obj = it->second;
        ptr = dynamic_cast< XClassType* >(obj);
    }
    return ptr;
}
 
template< typename XClassType > XClassType* MHO_ContainerStore::GetObject(const std::string& shortname)
{
    XClassType* ptr = nullptr;
    MHO_UUID obj_id = GetObjectUUID(shortname);
    if(!obj_id.is_empty())
    {
        ptr = GetObject< XClassType >(obj_id);
    }
    return ptr;
}
 
template< typename XClassType > XClassType* MHO_ContainerStore::GetObject(std::size_t index)
{
    XClassType* ptr = nullptr;
    std::size_t n_objects = GetNObjects< XClassType >();
    std::size_t count = 0;
    if(index < n_objects)
    {
        MHO_UUID type_id = fDictionary.GetUUIDFor< XClassType >();
        for(auto it = fIdsToObjects.begin(); it != fIdsToObjects.end(); it++)
        {
            key_pair item_ids = it->first;
            MHO_UUID item_type_id = item_ids.first;
            if(type_id == item_type_id)
            {
                if(count == index)
                {
                    MHO_Serializable* obj = it->second;
                    ptr = dynamic_cast< XClassType* >(obj);
                    break;
                }
                count++;
            }
        }
    }
 
    return ptr;
}
 
template< typename XClassType > bool MHO_ContainerStore::DeleteObject(XClassType* obj_ptr)
{
    MHO_Serializable* ptr = static_cast< MHO_Serializable* >(obj_ptr);
    if(ptr == nullptr)
    {
        return false;
    }
 
    auto it = fObjectsToIds.find(ptr);
    if(it == fObjectsToIds.end())
    {
        return false;
    }
 
    key_pair kp = it->second;
    MHO_UUID obj_id = kp.second;
    auto it2 = fIdsToObjects.find(kp);
    if(it2 == fIdsToObjects.end())
    {
        return false;
    }
 
    //remove entries related to this object
    fObjectsToIds.erase(it);
    fIdsToObjects.erase(it2);
    delete obj_ptr;
 
    //remove labels and short name associated with this object
    std::string shortname = "";
    for(auto it = fShortNameToIds.begin(); it != fShortNameToIds.end(); it++)
    {
        if(it->second == obj_id)
        {
            shortname = it->first;
            fShortNameToIds.erase(it);
            break;
        }
    }
    fShortNameSet.erase(shortname);
    return true;
}
 
template< typename XClassType > MHO_UUID MHO_ContainerStore::GetTypeUUID()
{
    MHO_UUID type_id = fDictionary.GetUUIDFor< XClassType >();
    return type_id;
}
 
template< typename XClassType > std::size_t MHO_ContainerStore::GetNObjects() const
{
    MHO_UUID type_id = fDictionary.GetUUIDFor< XClassType >();
    std::size_t count = 0;
    for(auto it = fIdsToObjects.begin(); it != fIdsToObjects.end(); it++)
    {
        key_pair item_ids = it->first;
        MHO_UUID item_type_id = item_ids.first;
        if(type_id == item_type_id)
        {
            count++;
        }
    }
    return count;
}
 
inline void MHO_ContainerStore::RenameObject(const std::string& current_shortname, const std::string& new_shortname)
{
    MHO_UUID obj_uuid = GetObjectUUID(current_shortname);
    if(obj_uuid.as_long() != 0)
    {
        fShortNameSet.erase(current_shortname);
        fShortNameToIds.erase(current_shortname);
        SetShortName(obj_uuid, new_shortname);
    }
}
 
} // namespace hops
 
#endif